CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-06-24 10:37:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

working on issue #12

Browse source
This commit is contained in:
Bernardo Damele 2012-07-26 23:11:07 +01:00
parent 57f2fccc24
commit d492291744
36 changed files with 72 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

2
lib/core/agent.py
View file

@ -46,7 +46,7 @@ class Agent:
if kb.tamperFunctions:
for function in kb.tamperFunctions:
query = function(query)
query, _ = function(payload=query, headers=None)
return query

6
lib/core/option.py
View file

@ -802,7 +802,7 @@ def __setTamperingFunctions():
priority = PRIORITY.NORMAL if not hasattr(module, '__priority__') else module.__priority__
for name, function in inspect.getmembers(module, inspect.isfunction):
if name == "tamper" and function.func_code.co_argcount == 1:
if name == "tamper" and function.func_code.co_argcount == 2:
found = True
kb.tamperFunctions.append(function)
@ -829,7 +829,9 @@ def __setTamperingFunctions():
function()
if not found:
raise sqlmapGenericException, "missing function 'tamper(value)' in tamper script '%s'" % tfile
errMsg = "missing function 'tamper(payload, headers)' "
errMsg += "in tamper script '%s'" % tfile
raise sqlmapGenericException, errMsg
if resolve_priorities and priorities:
priorities.sort(reverse=True)

2
lib/request/connect.py
View file

@ -550,7 +550,7 @@ class Connect:
if payload:
if kb.tamperFunctions:
for function in kb.tamperFunctions:
payload = function(payload)
payload, auxHeaders = function(payload=payload, headers=auxHeaders)
value = agent.replacePayload(value, payload)