File Server POST

fileserver.py

@@ -4,15 +4,21 @@ from hackingscripts import util
 from http.server import BaseHTTPRequestHandler, HTTPServer
 import threading
 import sys
+import os
+import ssl
 
 class FileServerRequestHandler(BaseHTTPRequestHandler):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
+    def do_POST(self):
+        self.do_GET()
+
     def do_GET(self):
-        if self.path in self.server.files:
+        path = self.path if "?" not in self.path else self.path[0:self.path.find("?")]
-            data = self.server.files[self.path]
+        if path in self.server.files:
+            data = self.server.files[path]
             self.send_response(200)
             self.end_headers()
             self.wfile.write(data)
@@ -20,14 +26,30 @@ class FileServerRequestHandler(BaseHTTPRequestHandler):
             self.send_response(404)
             self.end_headers()
 
+        if path in self.server.dumpRequests:
+            contentLength = self.headers.get('Content-Length')
+            body = None
+
+            if contentLength and int(contentLength) > 0:
+                body = self.rfile.read(int(contentLength))
+
+            print("==========")
+            print(str(self.headers).strip())
+            if body:
+                print()
+                print(body)
+            print("==========")
+
     def log_message(self, format, *args):
         if self.server.logRequests:
-            BaseHTTPRequestHandler.log_message(format, *args)
+            # BaseHTTPRequestHandler.log_message(format, *args)
+            super().log_message(format, *args)
 
 class HttpFileServer(HTTPServer):
     def __init__(self, addr, port):
         super().__init__((addr, port), FileServerRequestHandler)
         self.logRequests = False
+        self.dumpRequests = []
         self.files = { }
 
     def addFile(self, name, data):
@@ -44,6 +66,34 @@ class HttpFileServer(HTTPServer):
             name = "/" + name
         self.files[name.strip()] = data
 
+    def dumpRequest(self, name):
+        if not name.startswith("/"):
+            name = "/" + name
+        self.dumpRequests.append(name)
+
+    def enableLogging(self):
+        self.logRequests = True
+
+    def enableSSL(self, keyFile=None, certFile=None):
+        if keyFile is None:
+            print("Generating certificate…")
+            os.system("openssl req -new -x509 -keyout private.key -out server.crt -days 365 -nodes")
+            certFile = "server.crt"
+            keyFile = "private.key"
+
+        self.socket = ssl.wrap_socket(self.socket,
+            server_side=True,
+            certfile=certFile,
+            keyfile=keyFile,
+            ssl_version=ssl.PROTOCOL_TLS,
+            cert_reqs=ssl.CERT_NONE)
+
+        # try:
+        #     ssl._create_default_https_context = ssl._create_unverified_context
+        # except AttributeError:
+        #     print("Legacy Python that doesn't verify HTTPS certificates by default")
+        #     pass
+
     def startBackground(self):
         t = threading.Thread(target=self.serve_forever)
         t.start()