Initialize the info for web sockets

Author: zeroSteiner

lib/msf/core/handler/bind_aws_ssm.rb

@@ -271,9 +271,10 @@ def start_handler
             # Configure Channel
             chan._start_ssm_keepalive if datastore['SSM_KEEP_ALIVE']
             chan.params.comm = Rex::Socket::Comm::Local unless chan.params.comm
+            chan.params.peerhostname = peer_info['ComputerName']
             chan.params.peerhost = peer_info['IpAddress']
             chan.params.peerport = 0
-            chan.params.peerhostname = peer_info['ComputerName']
+            chan.lsock.initinfo(Rex::Socket.to_authority(peer_info['IpAddress'], 0), chan.lsock.localinfo)
             chan.update_term_size
           rescue => e
             print_error("AWS SSM handler failed: #{e.message}")

lib/rex/proto/http/web_socket.rb

@@ -44,8 +44,6 @@ def type?
       #   binary and text)
       # @param [Symbol] write_type the data type to write to the WebSocket
       def initialize(websocket, read_type: nil, write_type: :binary)
-        initialize_abstraction
-
         # a read type of nil will handle both binary and text frames that are received
         raise ArgumentError, 'read_type must be nil, :binary or :text' unless [nil, :binary, :text].include?(read_type)
         raise ArgumentError, 'write_type must be :binary or :text' unless %i[binary text].include?(write_type)
@@ -66,6 +64,10 @@ def initialize(websocket, read_type: nil, write_type: :binary)
           'SSL' => websocket.respond_to?(:sslctx) && !websocket.sslctx.nil?
         })
 
+        initialize_abstraction
+
+        lsock.initinfo(Rex::Socket.to_authority(peerhost, peerport), Rex::Socket.to_authority(localhost, localport))
+
         @thread = Rex::ThreadFactory.spawn("WebSocketChannel(#{localhost}->#{peerhost})", false) do
           websocket.wsloop do |data, data_type|
             next unless @read_type.nil? || data_type == @read_type