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 refactoring

Browse Source
This commit is contained in:
Miroslav Stampar
2011-11-29 19:17:07 +00:00
parent 3cd8f47686
commit 872a73f631
4 changed files with 22 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/request/basic.py
View File

@@ -26,6 +26,7 @@ from lib.core.common import singleTimeLogMessage
from lib.core.data import conf
from lib.core.data import kb
from lib.core.data import logger
from lib.core.enums import HTTPHEADER
from lib.core.exception import sqlmapDataException
from lib.core.settings import ML
from lib.core.settings import META_CHARSET_REGEX
@@ -43,20 +44,20 @@ def forgeHeaders(cookie, ua, referer):
headers = {}
for header, value in conf.httpHeaders:
if cookie and header == "Cookie":
if cookie and header == HTTPHEADER.COOKIE:
headers[header] = cookie
elif ua and header == "User-Agent":
elif ua and header == HTTPHEADER.USER_AGENT:
headers[header] = ua
elif referer and header == "Referer":
elif referer and header == HTTPHEADER.REFERER:
headers[header] = referer
else:
headers[header] = value
if kb.redirectSetCookie and not conf.dropSetCookie:
if "Cookie" in headers:
headers["Cookie"] = "%s; %s" % (headers["Cookie"], kb.redirectSetCookie)
if HTTPHEADER.COOKIE in headers:
headers[HTTPHEADER.COOKIE] = "%s; %s" % (headers[HTTPHEADER.COOKIE], kb.redirectSetCookie)
else:
headers["Cookie"] = kb.redirectSetCookie
headers[HTTPHEADER.COOKIE] = kb.redirectSetCookie
return headers