diff --git a/aftermath.py b/aftermath.py index d240ed4..678d443 100644 --- a/aftermath.py +++ b/aftermath.py @@ -64,7 +64,7 @@ def scan_target(target, args): r = requests.get(target, headers=headers, - timeout=10, + timeout=args.timeout, allow_redirects=False, verify=False) @@ -118,6 +118,14 @@ def main(): metavar='', help='number of threads to use') + parser.add_argument('-T', + '--timeout', + type=int, + dest='timeout', + default=10, + metavar='', + help='the request timeout in seconds') + args = parser.parse_args() print( @@ -148,7 +156,6 @@ def main(): logging.info(f'scanning completed. valid webshells: {valid}') sys.exit(-1) - # scan_targets(args) except KeyboardInterrupt: print(f'{ui.prompt()} user interrupt detected - exiting')