Merge pull request #19718 from sjanusz-r7/add-rpc-token

Keep track of RPC token per-thread
rapid7 · Dec 16, 2024 · 0068857 · 0068857
2 parents ebf7379 + 686a463
commit 0068857
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/msf/core/rpc/v10/service.rb b/lib/msf/core/rpc/v10/service.rb
@@ -140,11 +140,16 @@ def process(req)
         end
       end
 
-      ::Timeout.timeout(self.dispatcher_timeout) { self.handlers[group].send(mname, *msg) }
+      ::Timeout.timeout(self.dispatcher_timeout) do
+        Thread.current[:rpc_token] = token
+        self.handlers[group].send(mname, *msg)
+      end
 
     rescue ::Exception => e
       elog('RPC Exception', error: e)
       process_exception(e)
+    ensure
+      Thread.current[:rpc_token] = nil
     end
   end