sqlmap/lib/utils/timeout.py

#!/usr/bin/env python

import threading

from lib.core.data import logger

def timeout(func, args=(), kwargs={}, duration=1, default=None):
    class InterruptableThread(threading.Thread):
        def __init__(self):
            threading.Thread.__init__(self)
            self.result = None

        def run(self):
            try:
                self.result = func(*args, **kwargs)
            except Exception, msg:
                logger.log(8, msg)
                self.result = default

    thread = InterruptableThread()
    thread.start()
    thread.join(duration)
    if thread.isAlive():
        return default
    else:
        return thread.result