Handle Exceptions in Daemon Mode

If the user presses Ctrl+C while waiting on a connection, we want to
gracefully exit.
If the user presses Ctrl+C during the script, we want to stop executing
the script and restart the loop.
If any other exception happens during the script, we want to print out
the stacktrace as normal, but continue the loop.

Signed-off-by: dusoleil <howcansocksbereal@gmail.com>

1 files changed, 12 insertions, 1 deletions

diff --git a/sploit/main.py b/sploit/main.py
index 6c96104..bd01993 100644
--- a/sploit/main.py
+++ b/sploit/main.py
@@ -1,5 +1,6 @@
 import argparse
 import tempfile
+import traceback
 
 from sploit.comm import *
 
@@ -30,7 +31,17 @@ def daemon(script):
     print("Running in Pipe Daemon Mode...")
     with tempfile.TemporaryDirectory() as tmpdir:
         while(True):
-            runscript(script,Comm(Pipes(tmpdir)));
+            try:
+                p = Pipes(tmpdir)
+            except KeyboardInterrupt:
+                break
+            try:
+                runscript(script,Comm(p));
+            except KeyboardInterrupt:
+                pass
+            except:
+                traceback.print_exc()
+            del p
 
 def pipe(script):
     print("Running in Pipe Mode...");