client-mbedtls example added

Author: lganzzzo

.gitignore

@@ -33,8 +33,8 @@
 
 # Idea
 
-.idea/
+**/.idea/
+**/cmake-build-debug/
+**/build/
 
-async-server/cmake-build-debug/
-server/cmake-build-debug/
 

client-mbedtls/CMakeLists.txt

@@ -0,0 +1,59 @@
+cmake_minimum_required(VERSION 3.1)
+
+set(project_name websocket-client-mbedtls) ## rename your project here
+
+project(${project_name})
+
+set(CMAKE_CXX_STANDARD 11)
+
+include_directories(src)
+
+add_library(${project_name}-lib
+        src/WSListener.cpp
+        src/WSListener.hpp
+)
+
+## link libs
+
+find_package(oatpp              0.19.4 REQUIRED)
+find_package(oatpp-websocket    0.19.4 REQUIRED)
+find_package(oatpp-mbedtls      0.19.4 REQUIRED)
+
+target_link_libraries(${project_name}-lib
+        PUBLIC oatpp::oatpp
+        PUBLIC oatpp::oatpp-test
+        PUBLIC oatpp::oatpp-websocket
+        PUBLIC oatpp::oatpp-mbedtls
+)
+
+#################################################################
+## link mbedtls
+
+link_directories(
+        /usr/local/lib/
+)
+
+#################################################################
+
+## add executables
+
+add_executable(${project_name}-exe
+        src/App.cpp
+)
+target_link_libraries(${project_name}-exe ${project_name}-lib)
+add_dependencies(${project_name}-exe ${project_name}-lib)
+
+add_executable(${project_name}-test
+        test/tests.cpp
+        test/WSTest.cpp
+        test/WSTest.hpp
+)
+target_link_libraries(${project_name}-test ${project_name}-lib)
+add_dependencies(${project_name}-test ${project_name}-lib)
+
+set_target_properties(${project_name}-lib ${project_name}-exe ${project_name}-test PROPERTIES
+        CXX_STANDARD 11
+        CXX_EXTENSIONS OFF
+        CXX_STANDARD_REQUIRED ON
+        LINKER_LANGUAGE CXX
+)

client-mbedtls/src/App.cpp

@@ -0,0 +1,64 @@
+
+#include "WSListener.hpp"
+
+#include "oatpp-websocket/WebSocket.hpp"
+#include "oatpp-websocket/Connector.hpp"
+
+#include "oatpp-mbedtls/client/ConnectionProvider.hpp"
+#include "oatpp-mbedtls/Config.hpp"
+
+#include <thread>
+
+namespace {
+
+  bool finished = false;
+
+  void socketTask(const std::shared_ptr<oatpp::websocket::WebSocket>& websocket) {
+    websocket->listen();
+    OATPP_LOGD("websocket-client", "SOCKET CLOSED!!!");
+    finished = true;
+  }
+
+}
+
+void run() {
+
+  OATPP_LOGI("websocket-client", "Application Started");
+
+  auto config = oatpp::mbedtls::Config::createDefaultClientConfigShared();
+
+  auto connectionProvider = oatpp::mbedtls::client::ConnectionProvider::createShared(config, "echo.websocket.org", 443);
+
+  auto connector = oatpp::websocket::Connector::createShared(connectionProvider);
+
+  auto connection = connector->connect("/");
+
+  OATPP_LOGI("websocket-client", "Connected");
+
+  auto socket = oatpp::websocket::WebSocket::createShared(connection, true /* maskOutgoingMessages must be true for clients */);
+
+  std::mutex socketWriteMutex;
+
+  socket->setListener(std::make_shared<WSListener>(socketWriteMutex));
+
+  std::thread thread(socketTask, socket);
+
+  while(!finished) {
+    {
+      OATPP_LOGD("websocket-client", "sending message...");
+      std::lock_guard<std::mutex> lock(socketWriteMutex);
+      socket->sendOneFrameText("hello");
+    }
+    std::this_thread::sleep_for(std::chrono::milliseconds(1000));
+  }
+
+  thread.join();
+
+}
+
+int main() {
+  oatpp::base::Environment::init();
+  run();
+  oatpp::base::Environment::destroy();
+  return 0;
+}

client-mbedtls/src/WSListener.cpp

@@ -0,0 +1,41 @@
+//
+// Created by Leonid  on 2019-06-07.
+//
+
+#include "WSListener.hpp"
+
+////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+// WSListener
+
+void WSListener::onPing(const WebSocket& socket, const oatpp::String& message) {
+  OATPP_LOGD(TAG, "onPing");
+  std::lock_guard<std::mutex> lock(m_writeMutex);
+  socket.sendPong(message);
+}
+
+void WSListener::onPong(const WebSocket& socket, const oatpp::String& message) {
+  OATPP_LOGD(TAG, "onPong");
+}
+
+void WSListener::onClose(const WebSocket& socket, v_word16 code, const oatpp::String& message) {
+  OATPP_LOGD(TAG, "onClose code=%d", code);
+}
+
+void WSListener::readMessage(const WebSocket& socket, p_char8 data, oatpp::data::v_io_size size) {
+
+  if(size == 0) { // message transfer finished
+
+    auto wholeMessage = m_messageBuffer.toString();
+    m_messageBuffer.clear();
+
+    OATPP_LOGD(TAG, "on message received '%s'", wholeMessage->c_str());
+
+    /* Send message in reply */
+    //std::lock_guard<std::mutex> lock(m_writeMutex);
+    //socket.sendOneFrameText( "Hello from oatpp!: " + wholeMessage);
+
+  } else if(size > 0) { // message frame received
+    m_messageBuffer.write(data, size);
+  }
+
+}

client-mbedtls/src/WSListener.hpp

@@ -0,0 +1,51 @@
+//
+// Created by Leonid  on 2019-06-07.
+//
+
+#ifndef WSListener_hpp
+#define WSListener_hpp
+
+#include "oatpp-websocket/ConnectionHandler.hpp"
+#include "oatpp-websocket/WebSocket.hpp"
+
+/**
+ * WebSocket listener listens on incoming WebSocket events.
+ */
+class WSListener : public oatpp::websocket::WebSocket::Listener {
+private:
+  static constexpr const char* TAG = "Client_WSListener";
+private:
+  std::mutex& m_writeMutex;
+  /**
+   * Buffer for messages. Needed for multi-frame messages.
+   */
+  oatpp::data::stream::ChunkedBuffer m_messageBuffer;
+public:
+
+  WSListener(std::mutex& writeMutex)
+    : m_writeMutex(writeMutex)
+  {}
+
+  /**
+   * Called on "ping" frame.
+   */
+  void onPing(const WebSocket& socket, const oatpp::String& message) override;
+
+  /**
+   * Called on "pong" frame
+   */
+  void onPong(const WebSocket& socket, const oatpp::String& message) override;
+
+  /**
+   * Called on "close" frame
+   */
+  void onClose(const WebSocket& socket, v_word16 code, const oatpp::String& message) override;
+
+  /**
+   * Called on each message frame. After the last message will be called once-again with size == 0 to designate end of the message.
+   */
+  void readMessage(const WebSocket& socket, p_char8 data, oatpp::data::v_io_size size) override;
+
+};
+
+#endif // WSListener_hpp

client-mbedtls/test/WSTest.cpp

@@ -0,0 +1,11 @@
+//
+// Created by Leonid  on 2019-03-25.
+//
+
+#include "WSTest.hpp"
+
+void WSTest::onRun() {
+
+  OATPP_LOGD(TAG, "TODO - write tests");
+
+}

client-mbedtls/test/WSTest.hpp

@@ -0,0 +1,20 @@
+//
+// Created by Leonid  on 2019-03-25.
+//
+
+#ifndef MY_PROJECT_WEBSOCKETTEST_HPP
+#define MY_PROJECT_WEBSOCKETTEST_HPP
+
+
+#include "oatpp-test/UnitTest.hpp"
+
+class WSTest : public oatpp::test::UnitTest {
+public:
+
+  WSTest():UnitTest("TEST[WSTest]"){}
+  void onRun() override;
+
+};
+
+
+#endif //MY_PROJECT_WEBSOCKETTEST_HPP

client-mbedtls/test/tests.cpp

@@ -0,0 +1,29 @@
+
+#include "WSTest.hpp"
+
+#include "oatpp-test/UnitTest.hpp"
+#include <iostream>
+
+
+void runTests() {
+  OATPP_RUN_TEST(WSTest);
+}
+
+int main() {
+
+  oatpp::base::Environment::init();
+
+  runTests();
+
+  /* Print how much objects were created during app running, and what have left-probably leaked */
+  /* Disable object counting for release builds using '-D OATPP_DISABLE_ENV_OBJECT_COUNTERS' flag for better performance */
+  std::cout << "\nEnvironment:\n";
+  std::cout << "objectsCount = " << oatpp::base::Environment::getObjectsCount() << "\n";
+  std::cout << "objectsCreated = " << oatpp::base::Environment::getObjectsCreated() << "\n\n";
+
+  OATPP_ASSERT(oatpp::base::Environment::getObjectsCount() == 0);
+
+  oatpp::base::Environment::destroy();
+
+  return 0;
+}