Add Secure Tunnel Sample to CI tests (#353)

* Add Secure Tunnel Sample to CI tests

Author: sbSteveK

codebuild/samples/linux-smoke-tests.yml

@@ -16,6 +16,7 @@ phases:
       - echo Build started on `date`
       - $CODEBUILD_SRC_DIR/codebuild/samples/setup-linux.sh
       - $CODEBUILD_SRC_DIR/codebuild/samples/pubsub-linux.sh
+      - $CODEBUILD_SRC_DIR/codebuild/samples/securetunnel-linux.sh
   post_build:
     commands:
       - echo Build completed on `date`

codebuild/samples/securetunnel-linux.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+
+set -e
+
+env
+
+pushd $CODEBUILD_SRC_DIR/samples/secure_tunneling/secure_tunnel
+
+mkdir _build
+cd _build
+cmake -DCMAKE_PREFIX_PATH=/tmp/install ..
+make -j
+
+echo "aws iotsecuretunneling open-tunnel"
+RESPONSE=$(aws iotsecuretunneling open-tunnel)
+array=(${RESPONSE//:/ })
+
+echo "Assigning SOURCETOKEN"
+SOURCETOKEN="${array[11]}"
+
+echo "Assigning DESTINATIONTOKEN"
+DESTINATIONTOKEN="${array[13]}"
+
+echo "Secure Tunnel Destination test"
+./secure-tunnel --test --region us-east-1 --access_token $DESTINATIONTOKEN &
+sleep 10 &
+
+echo "Secure Tunnel Source test"
+./secure-tunnel --localProxyModeSource --region us-east-1 --access_token $SOURCETOKEN &

samples/secure_tunneling/secure_tunnel/main.cpp

@@ -78,6 +78,12 @@ int main(int argc, char *argv[])
     string proxyUserName;
     string proxyPassword;
 
+    /*
+     * For internal testing
+     */
+    bool isTest = s_cmdOptionExists(argv, argv + argc, "--test");
+    int expectedMessageCount = 5;
+
     std::shared_ptr<SecureTunnel> secureTunnel;
 
     /*********************** Parse Arguments ***************************/
@@ -267,6 +273,14 @@ int main(int argc, char *argv[])
                 fprintf(stdout, "Data Receive Complete in Destination\n");
                 fprintf(stdout, "Sending response message:\"%s\"\n", returnMessage.c_str());
                 secureTunnel->SendData(ByteCursorFromCString(returnMessage.c_str()));
+                if (isTest)
+                {
+                    expectedMessageCount--;
+                    if (expectedMessageCount == 0)
+                    {
+                        exit(0);
+                    }
+                }
                 break;
             case AWS_SECURE_TUNNELING_SOURCE_MODE:
                 fprintf(stdout, "Data Receive Complete in Source\n");