diff options
| -rw-r--r-- | tools/sploit/sploit/comm.py | 3 | ||||
| -rw-r--r-- | tools/sploit/sploit/main.py | 1 |
2 files changed, 4 insertions, 0 deletions
diff --git a/tools/sploit/sploit/comm.py b/tools/sploit/sploit/comm.py index 789a77e..604045c 100644 --- a/tools/sploit/sploit/comm.py +++ b/tools/sploit/sploit/comm.py @@ -15,6 +15,9 @@ class Comm: def __init__(self, backend): self.back = backend + def shutdown(self): + self.back.stdout.close() + def read(self, size): data = os.read(self.back.stdin.fileno(), size) if(data == b''): diff --git a/tools/sploit/sploit/main.py b/tools/sploit/sploit/main.py index e6c9bb3..b0fe3eb 100644 --- a/tools/sploit/sploit/main.py +++ b/tools/sploit/sploit/main.py @@ -48,6 +48,7 @@ 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: |