minor fixes

This commit is contained in:
Miroslav Stampar 2012-04-03 10:10:33 +00:00
parent e05109812f
commit 2504f4edb8
3 changed files with 4 additions and 11 deletions

View File

@ -57,18 +57,11 @@ class DNSServer:
self._socket.bind(("", 53)) self._socket.bind(("", 53))
self._running = False self._running = False
def pop(self): def pop(self, prefix=None, suffix=None):
retVal = None
with self._lock:
if len(self._requests):
retVal = self._requests.pop(0)
return retVal
def pop(self, prefix, suffix):
retVal = None retVal = None
with self._lock: with self._lock:
for _ in self._requests: for _ in self._requests:
if re.search("%s\..+\.%s" % (prefix, suffix), _, re.I): if prefix is None and suffix is None or re.search("%s\..+\.%s" % (prefix, suffix), _, re.I):
retVal = _ retVal = _
self._requests.remove(_) self._requests.remove(_)
break break

View File

@ -86,7 +86,7 @@ def __goDns(payload, expression):
if conf.dnsDomain and kb.dnsTest is not False: if conf.dnsDomain and kb.dnsTest is not False:
if kb.dnsTest is None: if kb.dnsTest is None:
randInt = randomInt() randInt = randomInt()
kb.dnsTest = dnsUse(payload, "SELECT %d" % randInt) == str(randInt) kb.dnsTest = dnsUse(payload, "SELECT %d%s" % (randInt, FROM_DUMMY_TABLE.get(Backend.getIdentifiedDbms(), ""))) == str(randInt)
if not kb.dnsTest: if not kb.dnsTest:
errMsg = "test for data retrieval through DNS channel failed. Turning off DNS exfiltration support" errMsg = "test for data retrieval through DNS channel failed. Turning off DNS exfiltration support"
logger.error(errMsg) logger.error(errMsg)

View File

@ -1,3 +1,3 @@
Files in this folder represent SQL (Procedural Language) snippets used Files in this folder represent SPL/SQL snippets used
by sqlmap on the target system. They are licensed under the terms of by sqlmap on the target system. They are licensed under the terms of
the GNU Lesser General Public License. the GNU Lesser General Public License.