Add files via upload

Final Version CW2.py

@@ -1,4 +1,5 @@
 def menu():
+    while True:
         print("The code will now check what operating system you are using")
         from sys import platform
         if platform == "linux" or platform == "linux2":
@@ -10,7 +11,7 @@ def menu():
                 input("Press the enter key to continue")
                 print("What Method would you like to use")
                 option = (input(
-                    "\n1. IP Address \n2. Username\n3. Hostname\n4. Port Scan \n5. UID 5\n6. Method 6\n"))
+                    "\n1. IP Address \n2. Username\n3. Hostname\n4. Port Scan \n5. UID 5\n6. Process Id\n"))
                 if option == "1":
                     import socket
                     hostname = socket.gethostname()
@@ -72,6 +73,9 @@ def menu():
                     elif option == "n":
                         exit()
                 elif option == "6":
+                    import os
+                    pid = os.getpid()
+                    print(pid)
 
                     print("Do you want to quit or return to menu?")
                     option = (input("y or n\n"))
@@ -85,8 +89,18 @@ def menu():
                 print("Privilege Escalation")
                 input("Press the enter key to continue")
                 print("What Method would you like to use")
-                option = (input("\n1. Method 1\n2. Method 2\n3. Method 3\n"))
+                option = (input("\n1. Admin CMD\n2. Method 2\n3. Method 3\n"))
                 if option == "1":
+                    import os
+                    import sys
+                    import win32com.shell.shell as shell
+                    ASADMIN = 'asadmin'
+
+                    if sys.argv[-1] != ASADMIN:
+                        script = os.path.abspath(sys.argv[0])
+                        params = ' '.join([script] + sys.argv[1:] + [ASADMIN])
+                        shell.ShellExecuteEx(lpVerb='runas', lpFile=sys.executable, lpParameters=params)
+                        sys.exit(0)
                     print("Do you want to quit or return to menu?")
                     option = (input("y or n\n"))
                     if option == "y":
@@ -124,7 +138,7 @@ def menu():
                 input("Press the enter key to continue")
                 print("What Method would you like to use")
                 option = (input(
-                    "\n1. IP Address \n2. Username\n3. Hostname\n4. Firewall Status\n5. Port Scan\n6. Method 6\n"))
+                    "\n1. IP Address \n2. Username\n3. Hostname\n4. Firewall Status\n5. Port Scan\n6. Process ID\n"))
                 if option == "1":
                     import socket
                     hostname = socket.gethostname()
@@ -185,7 +199,9 @@ def menu():
                     elif option == "n":
                         exit()
                 elif option == "6":
-
+                    import os
+                    pid = os.getpid()
+                    print(pid)
                     print("Do you want to quit or return to menu?")
                     option = (input("y or n\n"))
                     if option == "y":