1 files changed, 152 insertions, 0 deletions
diff --git a/tools/sploit/sploit/comm.py b/tools/sploit/sploit/comm.py
new file mode 100644
index 0000000..7d6cd8c
--- /dev/null
+++ b/tools/sploit/sploit/comm.py
@@ -0,0 +1,152 @@
+import subprocess
+import tempfile
+import os
+import sys
+import select
+
+from sploit.log import log
+from sploit.until import bind
+
+class Comm:
+    logonread = True
+    flushonwrite = True
+
+    def __init__(self, backend):
+        self.back = backend
+
+    def read(self, size):
+        data = os.read(self.back.stdin.fileno(), size)
+        if(data == b''):
+            raise BrokenPipeError('Tried to read on broken pipe')
+        if self.logonread : log(data)
+        return data
+
+    def readline(self):
+        data = self.back.stdin.readline()
+        if(data == b''):
+            raise BrokenPipeError('Tried to read on broken pipe')
+        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)
+        l = self.logonread
+        self.logonread = False
+        while(True):
+            data += self.read(1)
+            if(pred(data)):
+                break
+        self.logonread = l
+        if self.logonread : log(data)
+        return data
+
+    def readlineuntil(self, pred, /, *args, **kwargs):
+        dataarr = []
+        pred = bind(pred, *args, **kwargs)
+        while(True):
+            dataarr.append(self.readline())
+            if(pred(dataarr)):
+                break
+        return dataarr
+
+    def write(self, data):
+        self.back.stdout.write(data)
+        if self.flushonwrite : self.back.stdout.flush()
+
+    def writeline(self, data):
+        self.write(data + b'\n')
+
+    def interact(self):
+        print("<--Interact Mode-->")
+        stdin = sys.stdin.buffer
+        os.set_blocking(self.back.stdin.fileno(), False)
+        os.set_blocking(stdin.fileno(), False)
+        poll = select.poll()
+        poll.register(self.back.stdin, select.POLLIN)
+        poll.register(stdin, select.POLLIN)
+        brk = False
+        def readall(read, write):
+            while(True):
+                data = read()
+                if(data == b''):
+                    break
+                write(data)
+        readtable = {
+                stdin.fileno() : lambda : readall(stdin.readline, self.write),
+                self.back.stdin.fileno() : lambda : readall(self.back.stdin.readline, log)
+        }
+        readtable[self.back.stdin.fileno()]()
+        while(not brk):
+            try:
+                ioevents = poll.poll(100)
+                for ev in ioevents:
+                    if(ev[1] & select.POLLIN):
+                        readtable[ev[0]]()
+                    else:
+                        brk = True
+                        break
+            except KeyboardInterrupt:
+                break
+        os.set_blocking(self.back.stdin.fileno(), True)
+        os.set_blocking(stdin.fileno(), True)
+        print("<--Interact Mode Done-->")
+
+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,
+                preexec_fn=lambda : os.setpgrp())
+        print(f"PID: {self.proc.pid}")
+        self.stdin = self.proc.stdout
+        self.stdout = self.proc.stdin
+
+    def __del__(self):
+        if getattr(self, 'proc', None) == None : return
+        if(self.proc.poll() != None):
+            return
+        try:
+            print("Waiting on Target Program to End...")
+            print("Press Ctrl+C to Forcefully Kill It...")
+            self.proc.wait()
+        except KeyboardInterrupt:
+            self.proc.kill()
+
+class Pipes:
+    def __init__(self, tmp=None):
+        if(tmp == None):
+            self.dir = tempfile.TemporaryDirectory()
+            dirname = self.dir.name
+        else:
+            if(not os.path.exists(tmp)):
+                os.mkdir(tmp)
+            dirname = 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):
+        try:
+            if getattr(self,'stdout',None) : self.stdout.close()
+            if getattr(self,'stdin',None) : self.stdin.close()
+        except BrokenPipeError:
+            pass
+        if getattr(self,'pathin',None) and os.path.exists(self.pathin) : os.unlink(self.pathin)
+        if getattr(self,'pathout',None) and os.path.exists(self.pathout) : os.unlink(self.pathout)
+