summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--sploit/main.py19
1 files 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...")