mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2025-12-06 12:41:30 +00:00
minor update regarding DNS data retrieval task
This commit is contained in:
@@ -8,6 +8,7 @@ See the file 'doc/COPYING' for copying permission
|
||||
"""
|
||||
|
||||
import os
|
||||
import re
|
||||
import socket
|
||||
import threading
|
||||
import time
|
||||
@@ -63,6 +64,16 @@ class DNSServer:
|
||||
retVal = self._requests.pop(0)
|
||||
return retVal
|
||||
|
||||
def pop(self, prefix, suffix):
|
||||
retVal = None
|
||||
with self._lock:
|
||||
for _ in self._requests:
|
||||
if re.search("%s\..+\.%s" % (prefix, suffix), _, re.I):
|
||||
retVal = _
|
||||
self._requests.remove(_)
|
||||
break
|
||||
return retVal
|
||||
|
||||
def run(self):
|
||||
def _():
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user