From 4ad55bd63722303ae642dd955d9496c41c8fb681 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 --- tools/sploit/sploit/main.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) (limited to 'tools/sploit') diff --git a/tools/sploit/sploit/main.py b/tools/sploit/sploit/main.py index b1ddf13..6c96104 100644 --- a/tools/sploit/sploit/main.py +++ b/tools/sploit/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