From 8a009c5b481cd53c113ee05bcda357e3fc8629c0 Mon Sep 17 00:00:00 2001
From: James Lee <James_Lee@rapid7.com>
Date: Sun, 22 Jun 2008 22:55:38 +0000
Subject: [PATCH] initial support for SOCKS5

git-svn-id: file:///home/svn/framework3/trunk@5537 4d416f70-5f16-0410-b530-b9f4589650da
---
 lib/rex/socket/comm/local.rb | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/lib/rex/socket/comm/local.rb b/lib/rex/socket/comm/local.rb
index 33c9d56d77..c2378e2575 100644
--- a/lib/rex/socket/comm/local.rb
+++ b/lib/rex/socket/comm/local.rb
@@ -233,6 +233,36 @@ class Rex::Socket::Comm::Local
 			if ret[1] != 90
 				raise "SOCKS4 server responded with error code #{ret[0]}"
 			end
+		when 'socks5'
+			# TODO: add dns lookups through socks5
+			auth_methods = [5,1,0].pack('CCC')
+			size = sock.put(auth_methods)
+			if (size != auth_methods.length)
+				raise ArgumentError, "Wrote less data than expected to the socks proxy"
+			end
+			response = sock.get_once(2,30)
+			if (0xff == response[1,1])
+				raise ArgumentError, "Proxy requires authentication"
+			end
+
+			setup = [5,1,0,1].pack('CCCC') + Socket.gethostbyname(host)[3] + [port.to_i].pack('n')
+			size = sock.put(setup)
+			if (size != setup.length)
+				raise ArgumentError, "Wrote less data than expected to the socks proxy"
+			end
+	
+			begin
+				response = sock.get_once(10, 30)
+			rescue IOError
+				raise Rex::ConnectionRefused.new(host, port), caller
+			end
+	
+			if (response.nil? or response.length < 8)
+				raise ArgumentError, 'SOCKS5 server did not respond with a proper response'
+			end
+			if response[1] != 0
+				raise "SOCKS5 server responded with error code #{response[1]}"
+			end
 		else
 			raise ArgumentError, 'Unsupported proxy type and/or version', caller
 		end