From eb7e05d64c8df4ca30b60727b7846ee621e0d920 Mon Sep 17 00:00:00 2001 From: dusoleil Date: Wed, 1 Sep 2021 00:24:38 -0400 Subject: Allow Exit with Ctrl+C without a Stacktrace Signed-off-by: dusoleil --- sploit/main.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/sploit/main.py b/sploit/main.py index b1ddf13..6c96104 100644 --- a/sploit/main.py +++ b/sploit/main.py @@ -13,15 +13,18 @@ def main(): help='target program to exploit') args = parser.parse_args() - if(len(args.target)>0): - if(args.daemon): - print("Target Given. Ignoring Daemon Flag...") - target(args.script,args.target) - else: - if(args.daemon): - daemon(args.script) + try: + if(len(args.target)>0): + if(args.daemon): + print("Target Given. Ignoring Daemon Flag...") + target(args.script,args.target) else: - pipe(args.script) + if(args.daemon): + daemon(args.script) + else: + pipe(args.script) + except KeyboardInterrupt: + pass def daemon(script): print("Running in Pipe Daemon Mode...") -- cgit v1.2.3