Merge pull request #34 from MiaoZhangAWS/master

Add support for tunnel multi-port feature

Author: MiaoZhangAWS

.github/workflows/ci.yml

@@ -51,7 +51,7 @@ jobs:
       run: |
         mkdir build
         cd build
-        cmake .. -DOPENSSL_ROOT_DIR=/usr/local/Cellar/[email protected]/1.1.1g/ -DOPENSSL_LIBRARIES=/usr/local/Cellar/[email protected]/1.1.1g/lib/
+        cmake .. -DOPENSSL_ROOT_DIR=/usr/local/Cellar/[email protected]/1.1.1h/ -DOPENSSL_LIBRARIES=/usr/local/Cellar/[email protected]/1.1.1h/lib/
         make
   ubuntu:
     runs-on: ubuntu-latest

CMakeLists.txt

@@ -66,10 +66,10 @@ find_package(Catch2 REQUIRED)
 #########################################
 # Boost dependencies                    #
 #########################################
-set_property(GLOBAL PROPERTY Boost_USE_STATIC_LIBS ON) 
+set_property(GLOBAL PROPERTY Boost_USE_STATIC_LIBS ON)
 set_property(GLOBAL PROPERTY Boost_USE_DEBUG_RUNTIME OFF)
 #set_property(GLOBAL PROPERTY Boost_USE_MULTITHREADED ON)
-find_package(Boost 1.68.0 REQUIRED COMPONENTS system log log_setup thread program_options date_time)
+find_package(Boost 1.68.0 REQUIRED COMPONENTS system log log_setup thread program_options date_time filesystem)
 include_directories(${Boost_INCLUDE_DIRS})
 foreach(BOOST_LIB ${Boost_LIBRARIES})
     string(REPLACE ${CMAKE_SHARED_LIBRARY_SUFFIX} ${CMAKE_STATIC_LIBRARY_SUFFIX} BOOST_STATIC_LIB ${BOOST_LIB})
@@ -81,9 +81,10 @@ endforeach()
 #########################################
 file(GLOB ALL_SOURCES ${PROJECT_SOURCE_DIR}/src/*.cpp)
 
+set(UTIL_SOURCE ${PROJECT_SOURCE_DIR}/src/config/ConfigFile.cpp)
 set(CORE_SOURCES ${PROJECT_SOURCE_DIR}/src/TcpAdapterProxy.cpp ${PROJECT_SOURCE_DIR}/src/ProxySettings.cpp ${PROTO_HDRS} ${PROTO_SRCS})
-set(MAIN_SOURCES ${PROJECT_SOURCE_DIR}/src/main.cpp ${CORE_SOURCES})
-set(TEST_SOURCES ${PROJECT_SOURCE_DIR}/test/AdapterTests.cpp ${CORE_SOURCES} ${PROJECT_SOURCE_DIR}/test/TestWebsocketServer.cpp)
+set(MAIN_SOURCES ${PROJECT_SOURCE_DIR}/src/main.cpp ${CORE_SOURCES} ${UTIL_SOURCE})
+set(TEST_SOURCES ${PROJECT_SOURCE_DIR}/test/AdapterTests.cpp ${CORE_SOURCES} ${UTIL_SOURCE} ${PROJECT_SOURCE_DIR}/test/TestWebsocketServer.cpp)
 
 add_executable(${AWS_TUNNEL_LOCAL_PROXY_TARGET_NAME} ${MAIN_SOURCES})
 add_executable(${AWS_TUNNEL_LOCAL_PROXY_TEST_NAME} ${TEST_SOURCES})

README.md

@@ -1,8 +1,8 @@
-The reference implementation of the local proxy provides features that may require OS facilities not available on all device runtime environments in the industry. This guide provides details about the communication that occurs between the service and client to enable integration without or beyond the local proxy reference implementation choices.
+The reference implementation of the local proxy provides features that may require OS facilities not available on all device runtime environments in the industry. This guide provides details about the communication that occurs between the service and client to enable integration without or beyond the local proxy reference implementation choices. This protocol guide is only applicable for v1 local proxy. 
 
 ## Core implementation requirements
 
-In order to properly connect with and interpret messages from the AWS IoT Secure Tunneling service, the bare minimum is required:
+In order to properly connect with and interpret messages from the AWS IoT Secure Tunneling service, the following is required:
 
 **Communications Protocols:**
 * Websocket protocol ([RFC6455](https://tools.ietf.org/html/rfc6455)) over TCP/IP

WebsocketProtocolGuide.md renamed to V1WebSocketProtocolGuide.md

@@ -12,12 +12,15 @@ message Message {
     int32   streamId     = 2;
     bool    ignorable    = 3;
     bytes   payload      = 4;
+    string  serviceId = 5;
+    repeated string availableServiceIds = 6;
 
     enum Type {
         UNKNOWN = 0;
         DATA = 1;
         STREAM_START = 2;
         STREAM_RESET = 3;
         SESSION_RESET = 4;
+        SERVICE_IDS = 5;
     }
 }

V2WebSocketProtocolGuide.md

@@ -48,7 +48,7 @@ namespace aws { namespace iot { namespace securedtunneling { namespace settings
     bool const DEFAULT_WEB_SOCKET_DATA_ERROR_RETRY = true;
 
     char const * const KEY_WEB_SOCKET_SUBPROTOCOL = "tunneling.proxy.websocket.subprotocol";
-    std::string const DEFAULT_WEB_SOCKET_SUBPROTOCOL = "aws.iot.securetunneling-1.0";
+    std::string const DEFAULT_WEB_SOCKET_SUBPROTOCOL = "aws.iot.securetunneling-2.0";
 
     char const * const KEY_WEB_SOCKET_MAX_FRAME_SIZE = "tunneling.proxy.websocket.max_frame_size";
     std::size_t const DEFAULT_WEB_SOCKET_MAX_FRAME_SIZE = DEFAULT_MAX_DATA_FRAME_SIZE * 2;

resources/Message.proto

@@ -0,0 +1,32 @@
+// Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+#pragma once
+#include <boost/beast/core/flat_buffer.hpp>
+#include <boost/asio.hpp>
+#include <boost/asio/ip/tcp.hpp>
+#include "Message.pb.h"
+
+namespace aws { namespace iot { namespace securedtunneling { namespace connection {
+    class tcp_client
+    {
+    public:
+        typedef boost::shared_ptr<tcp_client> pointer;
+        tcp_client(boost::asio::io_context & io_context, std::size_t write_buf_size, std::size_t read_buf_size,  std::size_t ws_write_buf_size)
+                : resolver_(io_context)
+        {
+            connection_ =
+                    tcp_connection::create(io_context, write_buf_size, read_buf_size, ws_write_buf_size);
+        }
+        static pointer create(boost::asio::io_context& io_context, std::size_t const & write_buf_size, std::size_t const & read_buf_size, std::size_t const & ws_write_buf_size)
+        {
+            return pointer(new tcp_client(io_context, write_buf_size, read_buf_size, ws_write_buf_size));
+        }
+
+        tcp_connection::pointer                 connection_;
+        tcp::resolver                           resolver_;
+        // function object defines what to do after set up a tcp socket
+        std::function<void()>                   after_setup_tcp_socket = nullptr;
+        // function object defines what to do receiving control message: stream start
+        std::function<void()>                   on_receive_stream_start = nullptr;
+    };
+}}}}