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

some update

Browse Source
This commit is contained in:
Miroslav Stampar
2010-04-06 14:59:31 +00:00
parent 60f04f0a41
commit c24f1cc07c
8 changed files with 45 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lib/utils/timeout.py
View File

@@ -2,22 +2,26 @@
import threading
def timeout(func, args=(), kwargs={}, timeout_duration=1, default=None):
def timeout(func, args=(), kwargs={}, duration=1, default=None):
class InterruptableThread(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
self.exceeded = False
self.exceptionMsg = None
self.result = None
def run(self):
try:
self.result = func(*args, **kwargs)
except:
except Exception, msg:
self.exceptionMsg = msg
self.result = default
thread = InterruptableThread()
thread.start()
thread.join(timeout_duration)
if thread.isAlive():
thread.join(duration)
self.exceeded = thread.isAlive()
if self.exceeded:
return default
else:
return thread.result