diff options
| -rw-r--r-- | tools/sploit/sploit/comm.py | 7 | ||||
| -rw-r--r-- | tools/sploit/sploit/main.py | 3 |
2 files changed, 8 insertions, 2 deletions
diff --git a/tools/sploit/sploit/comm.py b/tools/sploit/sploit/comm.py index 1b8ba7f..223b639 100644 --- a/tools/sploit/sploit/comm.py +++ b/tools/sploit/sploit/comm.py @@ -28,6 +28,13 @@ class Comm: if self.logonread : log(data) return data + def readall(self): + data = b'' + for line in self.back.stdin: + log(line) + data += line + return data + def readuntil(self, pred, /, *args, **kwargs): data = b'' pred = bind(pred, *args, **kwargs) diff --git a/tools/sploit/sploit/main.py b/tools/sploit/sploit/main.py index ee0495f..0bc799e 100644 --- a/tools/sploit/sploit/main.py +++ b/tools/sploit/sploit/main.py @@ -55,6 +55,5 @@ def runscript(script, comm): print("Running Script...") exec(open(script).read()) print("Script Finished!") - for line in comm.back.stdin: - log(line) + comm.readall() |