summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--sploit/main.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/sploit/main.py b/sploit/main.py
index b0fe3eb..0f949f9 100644
--- a/sploit/main.py
+++ b/sploit/main.py
@@ -48,13 +48,14 @@ def runscript(script, comm):
code = compile(open(script).read(), script, 'exec')
exec(code, {'io': comm, 'print': elog})
ilog("Script Finished!")
- comm.shutdown()
- comm.readall()
return
except KeyboardInterrupt:
pass
except:
ilog(traceback.format_exc(), end='', color=ERROR)
finally:
+ comm.shutdown()
+ comm.readall()
gc.collect()
+
ilog("Script Ended Early!", color=WARNING)