Module: Rex::Proto::Proxy::Socks4a::Client::Relay

Defined in:
lib/rex/proto/proxy/socks4a.rb

Overview

A mixin for a socket to perform a relay to another socket.

Instance Method Summary collapse

Instance Method Details

#relay(relay_client, relay_sock) ⇒ Object

Relay data coming in from relay_sock to this socket.



165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
 
# File 'lib/rex/proto/proxy/socks4a.rb', line 165

def relay( relay_client, relay_sock )
  @relay_client = relay_client
  @relay_sock   = relay_sock
  # start the relay thread (modified from Rex::IO::StreamAbstraction)
  @relay_thread = Rex::ThreadFactory.spawn("SOCKS4AProxyServerRelay", false) do
    loop do
      closed = false
      buf    = nil

      begin
        s = Rex::ThreadSafe.select( [ @relay_sock ], nil, nil, 0.2 )
        if( s == nil || s[0] == nil )
          next
        end
      rescue
        closed = true
      end

      if( closed == false )
        begin
          buf = @relay_sock.sysread( 32768 )
          closed = true if( buf == nil )
        rescue
          closed = true
        end
      end

      if( closed == false )
        total_sent   = 0
        total_length = buf.length
        while( total_sent < total_length )
          begin
            data = buf[total_sent, buf.length]
            sent = self.write( data )
            if( sent > 0 )
              total_sent += sent
            end
          rescue
            closed = true
            break
          end
        end
      end

      if( closed )
        @relay_client.stop
        ::Thread.exit
      end
    end
  end

end