diff --git a/pocsuite3/lib/yaml/nuclei/protocols/common/expressions/__init__.py b/pocsuite3/lib/yaml/nuclei/protocols/common/expressions/__init__.py index c9afd97f..4f87aeac 100644 --- a/pocsuite3/lib/yaml/nuclei/protocols/common/expressions/__init__.py +++ b/pocsuite3/lib/yaml/nuclei/protocols/common/expressions/__init__.py @@ -31,6 +31,12 @@ class Marker: # ParenthesisClose marker - end of a placeholder ParenthesisClose = "}}" + def extract_timeout_value(raw_timeout: str) -> int: + match = re.search(r'@timeout:?(\d+)s', raw_timeout, re.IGNORECASE) + if match: + return int(match.group(1)) + return None + def auto_convert_types(func): @wraps(func)