1 files changed, 68 insertions, 0 deletions
diff --git a/tools/sploit/sploit/comm.py b/tools/sploit/sploit/comm.py
new file mode 100644
index 0000000..009f193
--- /dev/null
+++ b/tools/sploit/sploit/comm.py
@@ -0,0 +1,68 @@
+import subprocess
+import tempfile
+import os
+
+from sploit.log import log
+
+class Comm:
+    def __init__(self, backend):
+        self.back = backend
+
+    def __del__(self):
+        for line in self.back.stdin:
+            log(line)
+
+    def read(self, size):
+        data = self.back.stdin.read(size)
+        log(data)
+        return data
+
+    def readline(self):
+        data = self.back.stdin.readline()
+        log(data)
+        return data
+
+    def write(self, data):
+        self.back.stdout.write(data)
+        self.back.stdout.flush()
+
+    def writeline(self, data):
+        self.write(data + b'\n')
+
+class Process:
+    def __init__(self, args):
+        print(f"Running: {' '.join(args)}")
+        self.proc = subprocess.Popen(args,
+                stdin=subprocess.PIPE,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.STDOUT)
+        print(f"PID: {self.proc.pid}")
+        self.stdin = self.proc.stdout
+        self.stdout = self.proc.stdin
+
+    def __del__(self):
+        self.proc.wait()
+
+class Pipes:
+    def __init__(self,tmp=None):
+        if(tmp == None):
+            self.dir = tempfile.TemporaryDirectory()
+            dirname = self.dir.name
+        else:
+            dirname =  os.path.join("/tmp",tmp)
+        self.pathin = os.path.join(dirname,"in")
+        self.pathout = os.path.join(dirname,"out")
+        os.mkfifo(self.pathin)
+        os.mkfifo(self.pathout)
+        print("Waiting on Target to Connect...")
+        print("<"+self.pathin+" >"+self.pathout)
+        self.stdout = open(self.pathin,"wb")
+        self.stdin = open(self.pathout, "rb")
+        print("Connected!")
+
+    def __del__(self):
+        self.stdout.close()
+        self.stdin.close()
+        os.unlink(self.pathin)
+        os.unlink(self.pathout)
+