After the storm, a restore..

lib/request/proxy.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python
+
+"""
+$Id: proxy.py 322 2008-08-27 00:21:22Z inquisb $
+
+This file is part of the sqlmap project, http://sqlmap.sourceforge.net.
+
+Copyright (c) 2006-2008 Bernardo Damele A. G. <bernardo.damele@gmail.com>
+                        and Daniele Bellucci <daniele.bellucci@gmail.com>
+
+sqlmap is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation version 2 of the License.
+
+sqlmap is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+details.
+
+You should have received a copy of the GNU General Public License along
+with sqlmap; if not, write to the Free Software Foundation, Inc., 51
+Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+"""
+
+
+
+import httplib
+import socket
+import urllib
+import urllib2
+
+
+class ProxyHTTPConnection(httplib.HTTPConnection):
+    _ports = {"http" : 80, "https" : 443}
+
+    def request(self, method, url, body=None, headers={}):
+        # Request is called before connect, so can interpret url and get
+        # real host/port to be used to make CONNECT request to proxy
+        proto, rest = urllib.splittype(url)
+
+        if proto is None:
+            raise ValueError, "unknown URL type: %s" % url
+
+        # Get host
+        host, rest = urllib.splithost(rest)
+
+        # Try to get port
+        host, port = urllib.splitport(host)
+
+        # If port is not defined try to get from proto
+        if port is None:
+            try:
+                port = self._ports[proto]
+            except KeyError:
+                raise ValueError, "unknown protocol for: %s" % url
+
+        self._real_host = host
+        self._real_port = int(port)
+
+        httplib.HTTPConnection.request(self, method, url, body, headers)
+
+
+    def connect(self):
+        httplib.HTTPConnection.connect(self)
+
+        # Send proxy CONNECT request
+        self.send("CONNECT %s:%d HTTP/1.0\r\n\r\n" % (self._real_host, self._real_port))
+
+        # Expect a HTTP/1.0 200 Connection established
+        response = self.response_class(self.sock, strict=self.strict, method=self._method)
+        (version, code, message) = response._read_status()
+
+        # Probably here we can handle auth requests...
+        if code != 200:
+            # Proxy returned and error, abort connection, and raise exception
+            self.close()
+
+            raise socket.error, "Proxy connection failed: %d %s" % (code, message.strip())
+
+        # Eat up header block from proxy
+        while True:
+            # Should not use directly fp probably
+            line = response.fp.readline()
+
+            if line == "\r\n":
+                break
+
+
+class ProxyHTTPSConnection(ProxyHTTPConnection):
+    default_port = 443
+
+    def __init__(self, host, port=None, key_file=None, cert_file=None, strict=None):
+        ProxyHTTPConnection.__init__(self, host, port)
+        self.key_file = key_file
+        self.cert_file = cert_file
+
+    def connect(self):
+        ProxyHTTPConnection.connect(self)
+
+        # Make the sock ssl-aware
+        ssl = socket.ssl(self.sock, self.key_file, self.cert_file)
+        self.sock = httplib.FakeSocket(self.sock, ssl)
+
+
+class ProxyHTTPHandler(urllib2.HTTPHandler):
+    def __init__(self, proxy=None, debuglevel=0):
+        self.proxy = proxy
+
+        urllib2.HTTPHandler.__init__(self, debuglevel)
+
+    def do_open(self, http_class, req):
+        if self.proxy is not None:
+            req.set_proxy(self.proxy, "http")
+
+        return urllib2.HTTPHandler.do_open(self, ProxyHTTPConnection, req)
+
+
+class ProxyHTTPSHandler(urllib2.HTTPSHandler):
+    def __init__(self, proxy=None, debuglevel=0):
+        self.proxy = proxy
+
+        urllib2.HTTPSHandler.__init__(self, debuglevel)
+
+    def do_open(self, http_class, req):
+        if self.proxy is not None:
+            req.set_proxy(self.proxy, "https")
+
+        return urllib2.HTTPSHandler.do_open(self, ProxyHTTPSConnection, req)