Fix ssl copy bug (cmu-db#1436)

* Fix ssl copy bug * Update network_io_wrapper_factory.cpp
mbutrovich · Jun 28, 2018 · 40acdf6 · 40acdf6
1 parent d22bd24
commit 40acdf6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/network/network_io_wrapper_factory.cpp b/src/network/network_io_wrapper_factory.cpp
@@ -24,8 +24,7 @@ std::shared_ptr<NetworkIoWrapper> NetworkIoWrapperFactory::NewNetworkIoWrapper(
         conn_fd, std::make_shared<ReadBuffer>(),
         std::make_shared<WriteBuffer>());
     reusable_wrappers_[conn_fd] =
-        std::static_pointer_cast<NetworkIoWrapper, PosixSocketIoWrapper>(
-            wrapper);
+        std::static_pointer_cast<NetworkIoWrapper, PosixSocketIoWrapper>(wrapper);
     return wrapper;
   }
 
@@ -52,6 +51,7 @@ Transition NetworkIoWrapperFactory::PerformSslHandshake(
       throw NetworkProcessException("Failed to set ssl fd");
     io_wrapper =
         std::make_shared<SslSocketIoWrapper>(std::move(*io_wrapper), context);
+    reusable_wrappers_[io_wrapper->sock_fd_] = io_wrapper;
   } else {
     auto ptr = std::dynamic_pointer_cast<SslSocketIoWrapper, NetworkIoWrapper>(
         io_wrapper);
@@ -75,4 +75,4 @@ Transition NetworkIoWrapperFactory::PerformSslHandshake(
   }
 }
 }  // namespace network
-}  // namespace peloton
+}  // namespace peloton