mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 17:46:37 +03:00
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
"""
|
||
|
|
Copyright (c) 2006-2019 sqlmap developers (http://sqlmap.org/)
|
||
|
|
See the file 'LICENSE' for copying permission
|
||
|
|
"""
|
||
|
|
|
||
|
|
import urllib2
|
||
|
|
import httplib
|
||
|
|
from lib.core.data import conf
|
||
|
|
|
||
|
|
|
||
|
|
class HTTPHandler(urllib2.HTTPHandler):
|
||
|
|
"""
|
||
|
|
The hook http_requests function ensures that the chunk function is working properly.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def _hook(self, request):
|
||
|
|
host = request.get_host()
|
||
|
|
if not host:
|
||
|
|
raise urllib2.URLError('no host given')
|
||
|
|
|
||
|
|
if request.has_data(): # POST
|
||
|
|
data = request.get_data()
|
||
|
|
if not request.has_header('Content-type'):
|
||
|
|
request.add_unredirected_header(
|
||
|
|
'Content-type',
|
||
|
|
'application/x-www-form-urlencoded')
|
||
|
|
if not request.has_header('Content-length') and not conf.chunk:
|
||
|
|
request.add_unredirected_header(
|
||
|
|
'Content-length', '%d' % len(data))
|
||
|
|
|
||
|
|
sel_host = host
|
||
|
|
if request.has_proxy():
|
||
|
|
scheme, sel = urllib2.splittype(request.get_selector())
|
||
|
|
sel_host, sel_path = urllib2.splithost(sel)
|
||
|
|
|
||
|
|
if not request.has_header('Host'):
|
||
|
|
request.add_unredirected_header('Host', sel_host)
|
||
|
|
for name, value in self.parent.addheaders:
|
||
|
|
name = name.capitalize()
|
||
|
|
if not request.has_header(name):
|
||
|
|
request.add_unredirected_header(name, value)
|
||
|
|
return request
|
||
|
|
|
||
|
|
http_request = _hook
|