PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-06 12:41:30 +00:00
Code Issues Projects Releases Wiki Activity

minor update

Browse Source
This commit is contained in:
Miroslav Stampar
2011-06-24 17:15:15 +00:00
parent b2e6cf3ed9
commit 96190cf594
2 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lib/core/option.py
View File

@@ -101,6 +101,7 @@ from lib.core.settings import BURP_SPLITTER
from lib.core.settings import LOCALHOST
from lib.core.settings import MAX_NUMBER_OF_THREADS
from lib.core.settings import TIME_DELAY_CANDIDATES
from lib.core.settings import RAW_IP_ADDR_INFO
from lib.core.settings import UNKNOWN_DBMS_VERSION
from lib.core.settings import WEBSCARAB_SPLITTER
from lib.core.update import update
@@ -933,13 +934,18 @@ def __setHTTPProxy():
else:
proxyHandler = urllib2.ProxyHandler({"http": __proxyString})
# Patch for DNS leakage
# Just in case patch for eventual "DNS leakage"
if conf.proxy:
try:
addrinfo = socket.getaddrinfo(__hostname, __port)
except:
errMsg = "proxy host '%s' does not exist" % __hostname
raise sqlmapConnectionException, errMsg
if re.match(GENERAL_IP_ADDRESS_REGEX, __hostname):
addrinfo = RAW_IP_ADDR_INFO
for item in addrinfo:
item[-1] = (__hostname, __port)
else:
try:
addrinfo = socket.getaddrinfo(__hostname, __port)
except:
errMsg = "proxy host '%s' does not exist" % __hostname
raise sqlmapConnectionException, errMsg
conf.proxyDNSResponse = addrinfo