added Range handler (dealing with 206 HTTP messages)

stamparm · stamparm · commit eb999de0f1dc · 2010-11-08T12:26:13.000Z
diff --git a/lib/core/option.py b/lib/core/option.py
@@ -64,13 +64,15 @@
 from lib.parse.configfile import configFileParser
 from lib.request.proxy import ProxyHTTPSHandler
 from lib.request.certhandler import HTTPSCertAuthHandler
+from lib.request.rangehandler import HTTPRangeHandler
 from lib.request.redirecthandler import SmartRedirectHandler
 from lib.utils.google import Google
 
 authHandler = urllib2.BaseHandler()
 keepAliveHandler = keepalive.HTTPHandler()
 proxyHandler = urllib2.BaseHandler()
 redirectHandler = SmartRedirectHandler()
+rangeHandler = HTTPRangeHandler()
 
 def __urllib2Opener():
     """
@@ -80,12 +82,13 @@ def __urllib2Opener():
     global authHandler
     global keepAliveHandler
     global proxyHandler
+    global rangeHandler
     global redirectHandler
 
     debugMsg = "creating HTTP requests opener object"
     logger.debug(debugMsg)
 
-    handlers = [proxyHandler, authHandler, redirectHandler]
+    handlers = [proxyHandler, authHandler, redirectHandler, rangeHandler]
 
     if not conf.dropSetCookie:
         conf.cj = cookielib.LWPCookieJar()
diff --git a/lib/request/rangehandler.py b/lib/request/rangehandler.py
@@ -0,0 +1,52 @@
+#!/usr/bin/env python
+
+"""
+$Id$
+
+Copyright (c) 2006-2010 sqlmap developers (http://sqlmap.sourceforge.net/)
+See the file 'doc/COPYING' for copying permission
+"""
+
+import urllib
+import urllib2
+
+from lib.core.exception import sqlmapConnectionException
+
+class HTTPRangeHandler(urllib2.BaseHandler):
+    """
+    Handler that enables HTTP Range headers.
+
+    Reference: http://stackoverflow.com/questions/1971240/python-seek-on-remote-file
+
+    This was extremely simple. The Range header is a HTTP feature to
+    begin with so all this class does is tell urllib2 that the 
+    "206 Partial Content" reponse from the HTTP server is what we 
+    expected.
+
+    Example:
+        import urllib2
+        import byterange
+
+        range_handler = range.HTTPRangeHandler()
+        opener = urllib2.build_opener(range_handler)
+
+        # install it
+        urllib2.install_opener(opener)
+
+        # create Request and set Range header
+        req = urllib2.Request('http://www.python.org/')
+        req.header['Range'] = 'bytes=30-50'
+        f = urllib2.urlopen(req)
+    """
+
+    def http_error_206(self, req, fp, code, msg, hdrs):
+        # 206 Partial Content Response
+        r = urllib.addinfourl(fp, hdrs, req.get_full_url())
+        r.code = code
+        r.msg = msg
+        return r
+
+    def http_error_416(self, req, fp, code, msg, hdrs):
+        # HTTP's Range Not Satisfiable error
+        errMsg = "Invalid range"
+        raise sqlmapConnectionException, errMsg
