SERVER-86694 Fix crash when AsioTransportLayer is shut down between setup and start (#19073) (#19114)

GitOrigin-RevId: 5eff1aa2d23b15324f92a8f596175ae17c56a409

Author: patrickfreed

src/mongo/transport/asio/asio_transport_layer.cpp

@@ -1215,18 +1215,17 @@ Status AsioTransportLayer::start() {
         return ShutdownStatus;
     }
 
-    if (_sessionManager) {
-        uassertStatusOK(_sessionManager->start());
-    }
-
-    if (_listenerOptions.isIngress() && _listener.state == Listener::State::kNew) {
-        invariant(_sessionManager);
-        _listener.thread = stdx::thread([this] { _runListener(); });
-        _listener.cv.wait(lk, [&] { return _listener.state != Listener::State::kNew; });
-        return Status::OK();
+    if (_listenerOptions.isIngress()) {
+        // Only start the listener thread if the TL wasn't shut down before start() was invoked.
+        if (_listener.state == Listener::State::kNew) {
+            invariant(_sessionManager);
+            _listener.thread = stdx::thread([this] { _runListener(); });
+            _listener.cv.wait(lk, [&] { return _listener.state != Listener::State::kNew; });
+        }
+    } else {
+        invariant(_acceptorRecords.empty());
     }
 
-    invariant(_acceptorRecords.empty());
     return Status::OK();
 }
 

src/mongo/transport/asio/asio_transport_layer_test.cpp

@@ -342,6 +342,16 @@ TEST(AsioTransportLayer, TCPResetAfterConnectionIsSilentlySwallowed) {
     ASSERT_EQ(sessionsCreated.load(), 0);
 }
 
+TEST(AsioTransportLayer, StopAcceptingSessionsBeforeStart) {
+    auto sm = std::make_unique<test::MockSessionManager>();
+    auto tla = std::make_unique<AsioTransportLayer>(defaultTLAOptions(), std::move(sm));
+    ON_BLOCK_EXIT([&] { tla->shutdown(); });
+
+    ASSERT_OK(tla->setup());
+    tla->stopAcceptingSessions();
+    ASSERT_OK(tla->start());
+}
+
 #ifdef __linux__
 /**
  * Test that the server successfully captures the TCP socket queue depth, and places the value both
@@ -969,10 +979,6 @@ class AsioNetworkingBatonTest : public ServiceContextTest {
             _join();
         }
 
-        Status start() override {
-            return Status::OK();
-        }
-
         void startSession(std::shared_ptr<Session> session) override {
             stdx::lock_guard lk{_mutex};
             _sessions.push_back(session);

src/mongo/transport/grpc/grpc_transport_layer_impl.cpp

@@ -192,10 +192,6 @@ Status GRPCTransportLayerImpl::start() {
                 "tlsFIPSMode is not supported when gRPC mode is enabled",
                 !sslGlobalParams.sslFIPSMode);
 
-        if (_sessionManager) {
-            uassertStatusOK(_sessionManager->start());
-        }
-
         if (_server) {
             invariant(_sessionManager);
             _server->start();

src/mongo/transport/session_manager.h

@@ -77,11 +77,6 @@ class SessionManager {
      */
     virtual void endAllSessions(Client::TagMask tags) = 0;
 
-    /**
-     * Starts the session manager.
-     */
-    virtual Status start() = 0;
-
     /**
      * Shuts down the session manager.
      */

src/mongo/transport/session_manager_common.h

@@ -60,9 +60,6 @@ class SessionManagerCommon : public SessionManager {
     void endSessionByClient(Client* client) override;
     void endAllSessionsNoTagMask();
 
-    Status start() override {
-        return Status::OK();
-    }
     bool shutdown(Milliseconds timeout) override;
     bool shutdownAndWait(Milliseconds timeout);
     bool waitForNoSessions(Milliseconds timeout);

src/mongo/transport/session_manager_noop.h

@@ -41,9 +41,6 @@ class SessionManagerNoop : public SessionManager {
     void startSession(std::shared_ptr<transport::Session> session) override {}
     void endAllSessions(Client::TagMask tags) override {}
     void endSessionByClient(Client* client) override {}
-    Status start() override {
-        return Status::OK();
-    }
     bool shutdown(Milliseconds timeout) override {
         return true;
     }

src/mongo/transport/test_fixtures.h

@@ -183,10 +183,6 @@ class MockSessionManager : public SessionManager {
         _join();
     }
 
-    Status start() override {
-        return Status::OK();
-    }
-
     void startSession(std::shared_ptr<transport::Session> session) override {
         LOGV2(6109510, "Accepted connection", "remote"_attr = session->remote());
         auto& newSession = [&]() -> SessionThread& {

src/mongo/util/net/ssl_manager_test.cpp

@@ -77,10 +77,6 @@ class SessionManagerUtil : public transport::SessionManager {
         old_sessions.clear();
     }
 
-    Status start() override {
-        return Status::OK();
-    }
-
     bool shutdown(Milliseconds timeout) override {
         return true;
     }