From 641c4a5f2dbb483f7d8d1c23a585afb21ba4d3bc Mon Sep 17 00:00:00 2001 From: Mathy Vanhoef Date: Fri, 18 Dec 2020 21:18:51 +0400 Subject: [PATCH] fragattack: add --stay-up argument --- research/fragattack.py | 1 + research/fraginternals.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/research/fragattack.py b/research/fragattack.py index de27fc8cc..838a92804 100755 --- a/research/fragattack.py +++ b/research/fragattack.py @@ -204,6 +204,7 @@ if __name__ == "__main__": parser.add_argument('--connected-delay', type=float, default=1, help="Second to wait after AfterAuth before triggering Connected event") parser.add_argument('--to-self', default=False, action='store_true', help="Send ARP/DHCP/ICMP with same src and dst MAC address.") parser.add_argument('--no-drivercheck', default=False, action='store_true', help="Don't check if patched drivers are being used.") + parser.add_argument('--stay-up', default=False, action='store_true', help="Don't quit when test has finished.") options = parser.parse_args() # Default value for options that should not be command line parameters diff --git a/research/fraginternals.py b/research/fraginternals.py index 51115b031..949d0ded9 100644 --- a/research/fraginternals.py +++ b/research/fraginternals.py @@ -333,7 +333,8 @@ class Station(): def stop_test(self, failed=True): self.test = None - quit(failed) + if not self.options.stay_up: + quit(failed) def reset_keys(self): self.tk = None