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

Bug fix (reconnecting in case of timeouted direct connection)

Browse Source
This commit is contained in:
Miroslav Stampar
2016-10-17 22:55:07 +02:00
parent 24eaf55dc8
commit 5b14eecd25
5 changed files with 24 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/utils/timeout.py
View File

@@ -9,25 +9,29 @@ import threading
from lib.core.data import logger
from lib.core.enums import CUSTOM_LOGGING
from lib.core.enums import TIMEOUT_STATE
def timeout(func, args=(), kwargs={}, duration=1, default=None):
class InterruptableThread(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
self.result = None
self.timeout_state = None
def run(self):
try:
self.result = func(*args, **kwargs)
self.timeout_state = TIMEOUT_STATE.NORMAL
except Exception, msg:
logger.log(CUSTOM_LOGGING.TRAFFIC_IN, msg)
self.result = default
self.timeout_state = TIMEOUT_STATE.EXCEPTION
thread = InterruptableThread()
thread.start()
thread.join(duration)
if thread.isAlive():
return default
return default, TIMEOUT_STATE.TIMEOUT
else:
return thread.result
return thread.result, thread.timeout_state