Updated code to shutdown on tunnel close callback and updated zlib de…

…pendency version(#426) * updated code to shutdown on tunnel close callback * addressed lint errors * addressed lint errors * addressed lint errors * updated zlib version * updated zlib version * updated comment
awslabs · Sep 29, 2023 · 129f662 · 129f662
1 parent 95db8c9
commit 129f662
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/.github/docker-images/integration-tests/ubuntu/Dockerfile b/.github/docker-images/integration-tests/ubuntu/Dockerfile
@@ -14,9 +14,9 @@ RUN apt update && apt upgrade -y && \
 RUN mkdir /home/dependencies
 WORKDIR /home/dependencies
 
-RUN wget https://www.zlib.net/zlib-1.2.13.tar.gz -O /tmp/zlib-1.2.13.tar.gz && \
-	tar xzvf /tmp/zlib-1.2.13.tar.gz && \
-	cd zlib-1.2.13 && \
+RUN wget https://www.zlib.net/zlib-1.3.tar.gz -O /tmp/zlib-1.3.tar.gz && \
+	tar xzvf /tmp/zlib-1.3.tar.gz && \
+	cd zlib-1.3 && \
 	./configure && \
 	make && \
 	make install && \

diff --git a/source/main.cpp b/source/main.cpp
@@ -237,6 +237,10 @@ namespace Aws
                         case ClientBaseEventNotification::FEATURE_STOPPED:
                         {
                             LOGM_INFO(TAG, "%s has stopped", feature->getName().c_str());
+// Stopping DC instance if secure tunnel is closed for a ST component
+#if defined(DISABLE_MQTT)
+                            shutdown();
+#endif
                             break;
                         }
                         default:

diff --git a/source/tunneling/SecureTunnelingFeature.cpp b/source/tunneling/SecureTunnelingFeature.cpp
@@ -272,12 +272,15 @@ namespace Aws
                 void SecureTunnelingFeature::OnConnectionShutdown(SecureTunnelingContext *contextToRemove)
                 {
                     LOG_DEBUG(TAG, "SecureTunnelingFeature::OnConnectionShutdown");
-
+#if defined(DISABLE_MQTT)
+                    this->stop();
+#else
                     auto it =
                         find_if(mContexts.begin(), mContexts.end(), [&](const unique_ptr<SecureTunnelingContext> &c) {
                             return c.get() == contextToRemove;
                         });
                     mContexts.erase(std::remove(mContexts.begin(), mContexts.end(), *it));
+#endif
                 }
 
             } // namespace SecureTunneling