Changeset 20017 for trunk/FACT++/src/ConnectionSSL.cc

Timestamp:

12/28/20 17:25:10 (4 years ago)

Author:

tbretz

Message:

Made ready for boost 1.70 by replacing get_io_context; the SSL connection has to be properly shutdown, but this required a new socket.. therefore switched to boost::optional to keep the socket.

File:

• trunk/FACT++/src/ConnectionSSL.cc (modified) (12 diffs)

trunk/FACT++/src/ConnectionSSL.cc

@@ -18 +18 @@
 namespace dummy = ba::placeholders;
 
-typedef ssl::stream<boost::asio::ip::tcp::socket> stream;
-
 using ba::ip::tcp;
 
@@ -34 +32 @@
 void ConnectionSSL::AsyncRead(const ba::mutable_buffers_1 buffers, int type)
 {
-    ba::async_read(*this, buffers,
+    ba::async_read(*stream, buffers,
                    boost::bind(&ConnectionSSL::HandleReceivedData, this,
                                dummy::error, dummy::bytes_transferred, type));
@@ -41 +39 @@
 void ConnectionSSL::AsyncWrite(const ba::const_buffers_1 &buffers)
 {
-    ba::async_write(*this, buffers,
+    ba::async_write(*stream, buffers,
                     boost::bind(&ConnectionSSL::HandleSentData, this,
                                 dummy::error, dummy::bytes_transferred));
@@ -69 +67 @@
 
     // AsyncConnect + Deadline
-     lowest_layer().async_connect(endpoint,
+    stream->lowest_layer().async_connect(endpoint,
                   boost::bind(&ConnectionSSL::ConnectIter,
                               this, iterator, ba::placeholders::error));
@@ -80 +78 @@
 {
     // AsyncConnect + Deadline
-     lowest_layer().async_connect(fEndpoint,
+     stream->lowest_layer().async_connect(fEndpoint,
                   boost::bind(&ConnectionSSL::ConnectAddr,
                               this, fEndpoint, ba::placeholders::error));
@@ -87 +85 @@
     //AsyncWait(fConnectTimeout, 5, &ConnectionSSL::HandleConnectTimeout);
 }
+
+void ConnectionSSL::HandleAsyncShutdown(const bs::error_code &error)
+{
+    stream->lowest_layer().close();
+
+    if (!error || error==error.default_error_condition())
+        return;
+
+    ostringstream str;
+    str << "Shutdown of SSL connection to " << URL() << "[" << error.value() << "]: " << error.message() << " (" << error << ")";// << endl;
+    Error(str);
+}
+
 
 // ------------------------ close --------------------------
@@ -103 +114 @@
 
     // Close possible open connections
-    lowest_layer().close();
+    if (!IsClosed())
+    {
+        // Adapted from
+        // https://stackoverflow.com/questions/49122521/cant-implement-boostasiosslstreamboostasioiptcpsocket-reconnect
+
+#if BOOST_VERSION < 107000
+        auto& io_context = stream->get_io_service();
+#else
+        const auto& io_context = stream->get_executor();
+#endif
+
+        stream->lowest_layer().cancel(); // Remove all pending stuff in the queue
+        stream->async_shutdown(boost::bind(&ConnectionSSL::HandleAsyncShutdown, this, dummy::error));
+
+        stream.emplace(io_context, *this);
+    }
 
     // Reset the connection status
@@ -128 +154 @@
 void ConnectionSSL::PostClose(bool restart)
 {
-    get_io_service().post(boost::bind(&ConnectionSSL::CloseImp, this, restart));
+#if BOOST_VERSION < 107000
+    stream->get_io_service().post(boost::bind(&ConnectionSSL::CloseImp, this, restart));
+#else
+    ba::post(boost::bind(&ConnectionSSL::CloseImp, this, restart));
+#endif
 }
 
@@ -265 +295 @@
     if (!error)
     {
-        lowest_layer().set_option(boost::asio::socket_base::keep_alive(true));
+        stream->lowest_layer().set_option(boost::asio::socket_base::keep_alive(true));
 
         const int optval = 30;
 
-#if BOOST_VERSION <107000
+#if BOOST_VERSION < 107000
         // First keep alive after 30s
-        setsockopt(lowest_layer().native(), SOL_TCP, TCP_KEEPIDLE, &optval, sizeof(optval));
+        setsockopt(stream->lowest_layer().native(), SOL_TCP, TCP_KEEPIDLE, &optval, sizeof(optval));
         // New keep alive after 30s
-        setsockopt(lowest_layer().native(), SOL_TCP, TCP_KEEPINTVL, &optval, sizeof(optval));
+        setsockopt(stream->lowest_layer().native(), SOL_TCP, TCP_KEEPINTVL, &optval, sizeof(optval));
 #else
         // First keep alive after 30s
-        setsockopt(lowest_layer().native_handle(), SOL_TCP, TCP_KEEPIDLE, &optval, sizeof(optval));
+        setsockopt(stream->lowest_layer().native_handle(), SOL_TCP, TCP_KEEPIDLE, &optval, sizeof(optval));
         // New keep alive after 30s
-        setsockopt(lowest_layer().native_handle(), SOL_TCP, TCP_KEEPINTVL, &optval, sizeof(optval));
+        setsockopt(stream->lowest_layer().native_handle(), SOL_TCP, TCP_KEEPINTVL, &optval, sizeof(optval));
 #endif
 
@@ -331 +361 @@
         //set_default_verify_paths();
 
-        async_handshake(stream::client,
+        stream->async_handshake(boost::asio::ssl::stream<boost::asio::ip::tcp::socket>::client,
                         boost::bind(&ConnectionSSL::HandleHandshake, this,
                                     endpoint, dummy::error));
@@ -454 +484 @@
     else
     {
-        tcp::resolver resolver(get_io_service());
+#if BOOST_VERSION < 107000
+        tcp::resolver resolver(stream->get_io_service());
+#else
+        tcp::resolver resolver(stream->get_executor());
+#endif
 
         tcp::resolver::query query(fAddress, fPort);
@@ -529 +563 @@
 
 ConnectionSSL::ConnectionSSL(ba::io_service& ioservice, ostream &out) : MessageImp(out),
+#if BOOST_VERSION < 107000
 ssl::context(ioservice, boost::asio::ssl::context::method::sslv23_client),
-stream(ioservice, *this), fLog(0), fVerbose(true), fDebugTx(false),
+#else
+ssl::context(boost::asio::ssl::context::method::sslv23_client),
+#endif
+stream(boost::in_place_init, ioservice, *this),
+fLog(0), fVerbose(true), fDebugTx(false),
 fInTimeout(ioservice), fOutTimeout(ioservice), fConnectionTimer(ioservice),
 fQueueSize(0), fConnectionStatus(kDisconnected)