sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-02 20:54:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Minor update (consistency patch)

Browse Source
This commit is contained in:
Miroslav Stampar 2014-03-30 16:51:31 +02:00
parent 76b9fad24a
commit 3e024ac8e6
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lib/core/agent.py
View File

@ -953,35 +953,35 @@ class Agent(object):
return caseExpression return caseExpression
def addPayloadDelimiters(self, inpStr): def addPayloadDelimiters(self, value):
""" """
Adds payload delimiters around the input string Adds payload delimiters around the input string
""" """
return "%s%s%s" % (PAYLOAD_DELIMITER, inpStr, PAYLOAD_DELIMITER) if inpStr else inpStr return "%s%s%s" % (PAYLOAD_DELIMITER, value, PAYLOAD_DELIMITER) if value else value
def removePayloadDelimiters(self, inpStr): def removePayloadDelimiters(self, value):
""" """
Removes payload delimiters from inside the input string Removes payload delimiters from inside the input string
""" """
return inpStr.replace(PAYLOAD_DELIMITER, '') if inpStr else inpStr return value.replace(PAYLOAD_DELIMITER, '') if value else value
def extractPayload(self, inpStr): def extractPayload(self, value):
""" """
Extracts payload from inside of the input string Extracts payload from inside of the input string
""" """
_ = re.escape(PAYLOAD_DELIMITER) _ = re.escape(PAYLOAD_DELIMITER)
return extractRegexResult("(?s)%s(?P<result>.*?)%s" % (_, _), inpStr) return extractRegexResult("(?s)%s(?P<result>.*?)%s" % (_, _), value)
def replacePayload(self, inpStr, payload): def replacePayload(self, value, payload):
""" """
Replaces payload inside the input string with a given payload Replaces payload inside the input string with a given payload
""" """
_ = re.escape(PAYLOAD_DELIMITER) _ = re.escape(PAYLOAD_DELIMITER)
return re.sub("(%s.*?%s)" % (_, _), ("%s%s%s" % (PAYLOAD_DELIMITER, payload, PAYLOAD_DELIMITER)).replace("\\", r"\\"), inpStr) if inpStr else inpStr return re.sub("(%s.*?%s)" % (_, _), ("%s%s%s" % (PAYLOAD_DELIMITER, payload, PAYLOAD_DELIMITER)).replace("\\", r"\\"), value) if value else value
def runAsDBMSUser(self, query): def runAsDBMSUser(self, query):
if conf.dbmsCred and "Ad Hoc Distributed Queries" not in query: if conf.dbmsCred and "Ad Hoc Distributed Queries" not in query: