mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-30 13:33:49 +03:00
101 lines
2.6 KiB
Python
101 lines
2.6 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2024 sqlmap developers (https://sqlmap.org/)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
import functools
|
|
import hashlib
|
|
import threading
|
|
|
|
from lib.core.datatype import LRUDict
|
|
from lib.core.settings import MAX_CACHE_ITEMS
|
|
from lib.core.settings import UNICODE_ENCODING
|
|
from lib.core.threads import getCurrentThreadData
|
|
|
|
_cache = {}
|
|
_cache_lock = threading.Lock()
|
|
_method_locks = {}
|
|
|
|
def cachedmethod(f):
|
|
"""
|
|
Method with a cached content
|
|
|
|
>>> __ = cachedmethod(lambda _: _)
|
|
>>> __(1)
|
|
1
|
|
>>> __(1)
|
|
1
|
|
>>> __ = cachedmethod(lambda *args, **kwargs: args[0])
|
|
>>> __(2)
|
|
2
|
|
>>> __ = cachedmethod(lambda *args, **kwargs: next(iter(kwargs.values())))
|
|
>>> __(foobar=3)
|
|
3
|
|
|
|
Reference: http://code.activestate.com/recipes/325205-cache-decorator-in-python-24/
|
|
"""
|
|
|
|
_cache[f] = LRUDict(capacity=MAX_CACHE_ITEMS)
|
|
|
|
@functools.wraps(f)
|
|
def _f(*args, **kwargs):
|
|
try:
|
|
key = int(hashlib.md5("|".join(str(_) for _ in (f, args, kwargs)).encode(UNICODE_ENCODING)).hexdigest(), 16) & 0x7fffffffffffffff
|
|
except ValueError: # https://github.com/sqlmapproject/sqlmap/issues/4281 (NOTE: non-standard Python behavior where hexdigest returns binary value)
|
|
result = f(*args, **kwargs)
|
|
else:
|
|
try:
|
|
with _cache_lock:
|
|
result = _cache[f][key]
|
|
except KeyError:
|
|
result = f(*args, **kwargs)
|
|
|
|
with _cache_lock:
|
|
_cache[f][key] = result
|
|
|
|
return result
|
|
|
|
return _f
|
|
|
|
def stackedmethod(f):
|
|
"""
|
|
Method using pushValue/popValue functions (fallback function for stack realignment)
|
|
|
|
>>> threadData = getCurrentThreadData()
|
|
>>> original = len(threadData.valueStack)
|
|
>>> __ = stackedmethod(lambda _: threadData.valueStack.append(_))
|
|
>>> __(1)
|
|
>>> len(threadData.valueStack) == original
|
|
True
|
|
"""
|
|
|
|
@functools.wraps(f)
|
|
def _(*args, **kwargs):
|
|
threadData = getCurrentThreadData()
|
|
originalLevel = len(threadData.valueStack)
|
|
|
|
try:
|
|
result = f(*args, **kwargs)
|
|
finally:
|
|
if len(threadData.valueStack) > originalLevel:
|
|
threadData.valueStack = threadData.valueStack[:originalLevel]
|
|
|
|
return result
|
|
|
|
return _
|
|
|
|
def lockedmethod(f):
|
|
@functools.wraps(f)
|
|
def _(*args, **kwargs):
|
|
if f not in _method_locks:
|
|
_method_locks[f] = threading.RLock()
|
|
|
|
with _method_locks[f]:
|
|
result = f(*args, **kwargs)
|
|
|
|
return result
|
|
|
|
return _
|