diff options
Diffstat (limited to 'lib/net/ssh/proxy/http.rb')
-rw-r--r-- | lib/net/ssh/proxy/http.rb | 34 |
1 files changed, 17 insertions, 17 deletions
diff --git a/lib/net/ssh/proxy/http.rb b/lib/net/ssh/proxy/http.rb index cf94634..e7dc3d0 100644 --- a/lib/net/ssh/proxy/http.rb +++ b/lib/net/ssh/proxy/http.rb @@ -1,8 +1,8 @@ require 'socket' require 'net/ssh/proxy/errors' -module Net - module SSH +module Net + module SSH module Proxy # An implementation of an HTTP proxy. To use it, instantiate it, then # pass the instantiated object via the :proxy key to Net::SSH.start: @@ -25,14 +25,14 @@ module Net class HTTP # The hostname or IP address of the HTTP proxy. attr_reader :proxy_host - + # The port number of the proxy. attr_reader :proxy_port - + # The map of additional options that were given to the object at # initialization. attr_reader :options - + # Create a new socket factory that tunnels via the given host and # port. The +options+ parameter is a hash of additional settings that # can be used to tweak this proxy connection. Specifically, the following @@ -45,47 +45,47 @@ module Net @proxy_port = proxy_port @options = options end - + # Return a new socket connected to the given host and port via the # proxy that was requested when the socket factory was instantiated. def open(host, port, connection_options) socket = establish_connection(connection_options[:timeout]) socket.write "CONNECT #{host}:#{port} HTTP/1.1\r\n" socket.write "Host: #{host}:#{port}\r\n" - + if options[:user] credentials = ["#{options[:user]}:#{options[:password]}"].pack("m*").gsub(/\s/, "") socket.write "Proxy-Authorization: Basic #{credentials}\r\n" end - + socket.write "\r\n" - + resp = parse_response(socket) - + return socket if resp[:code] == 200 - + socket.close raise ConnectError, resp.inspect end - + protected - + def establish_connection(connect_timeout) Socket.tcp(proxy_host, proxy_port, nil, nil, connect_timeout: connect_timeout) end - + def parse_response(socket) version, code, reason = socket.gets.chomp.split(/ /, 3) headers = {} - + while (line = socket.gets) && (line.chomp! != "") name, value = line.split(/:/, 2) headers[name.strip] = value.strip end - + body = socket.read(headers["Content-Length"].to_i) if headers["Content-Length"] - + return { version: version, code: code.to_i, reason: reason, |