SERVER-29165: Create and populate RestrictionEnvironment class

1 files changed, 4 insertions, 0 deletions

diff --git a/src/mongo/transport/transport_layer_legacy.cpp b/src/mongo/transport/transport_layer_legacy.cpp
index 2e3d39b1cd4..6aeb6f89695 100644
--- a/src/mongo/transport/transport_layer_legacy.cpp
+++ b/src/mongo/transport/transport_layer_legacy.cpp
@@ -38,6 +38,7 @@
 
 #include "mongo/base/checked_cast.h"
 #include "mongo/config.h"
+#include "mongo/db/auth/restriction_environment.h"
 #include "mongo/db/service_context.h"
 #include "mongo/db/stats/counters.h"
 #include "mongo/stdx/functional.h"
@@ -339,7 +340,10 @@ void TransportLayerLegacy::_handleNewConnection(std::unique_ptr<AbstractMessagin
     }
 
     amp->setLogLevel(logger::LogSeverity::Debug(1));
+    auto restrictionEnvironment =
+        stdx::make_unique<RestrictionEnvironment>(amp->remoteAddr(), amp->localAddr());
     auto session = LegacySession::create(std::move(amp), this);
+    RestrictionEnvironment::set(session, std::move(restrictionEnvironment));
 
     stdx::list<std::weak_ptr<LegacySession>> list;
     auto it = list.emplace(list.begin(), session);