add pyproject.toml

.gitignore

@@ -0,0 +1,2 @@
+__pycache__
+.venv

pyproject.toml

@@ -0,0 +1,19 @@
+[tool.poetry]
+name = "winpwnage"
+version = "0.0.0"
+description = "UAC bypass, Elevate, Persistence methods"
+authors = ["rootm0s"]
+license = "Proprietary"
+readme = "README.md"
+repository = "https://github.com/rootm0s/WinPwnage"
+
+[tool.poetry.dependencies]
+python = "^3.9"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+winpwnage = 'winpwnage.__main__:main'

main.py → winpwnage/__main__.py

@@ -1,52 +1,52 @@
-from winpwnage.core.prints import print_info
-from winpwnage.core.scanner import scanner, function
-from winpwnage.core.utils import *
-import argparse
-import sys
-
-print("""
-        _
-  _ _ _|_|___ ___ _ _ _ ___ ___ ___ ___
- | | | | |   | . | | | |   | .'| . | -_|
- |_____|_|_|_|  _|_____|_|_|__,|_  |___|
-             |_|               |___|
-""")
-
-print_info("UAC level: {}".format(information().uac_level()))
-print_info("Build number: {}".format(information().build_number()))
-print_info("Running elevated: {}".format(information().admin()))
-print_info("Python version: {}.{}.{}\n".format(*sys.version_info))
-
-def main():
-	scan_cmds = ["uac", "persist", "elevate"]
-
-	parser = argparse.ArgumentParser()
-	parser.add_argument("-s", "--scan", nargs="+", required=False, help="Scan for either uac, persist or elevate method")
-	parser.add_argument("-u", "--use", nargs="+", required=False, help="Use either uac, persist or elevate method")
-	parser.add_argument("-i", "--id", nargs="+", required=False, help="Id of method")
-	parser.add_argument("-p", "--payload", nargs="+", required=False, help="Full path to payload, can include params")
-	parser.add_argument("-r", "--remove", action="store_true", required=False, help="Removes installed persistence")
-	args = parser.parse_args()
-
-	if args.scan:
-		if not all([_ in scan_cmds for _ in args.scan]):
-			parser.print_help()
-
-		scanner(**{scan_cmds[_]: scan_cmds[_] in args.scan for _ in range(3)}).start()
-
-	if args.use and args.id:
-		if not all([_ in scan_cmds for _ in args.use]):
-			parser.print_help()
-
-		if scan_cmds[0] in args.use and args.payload:
-			function(uac=True, persist=False, elevate=False).run(id=args.id[0], payload=args.payload)
-
-		if scan_cmds[1] in args.use:		
-			function(uac=False, persist=True, elevate=False).run(id=args.id[0], payload=args.payload,
-			add=(False if args.remove else True))
-
-		if scan_cmds[2] in args.use and args.payload:
-			function(uac=False, persist=False, elevate=True).run(id=args.id[0], payload=args.payload)
-
-if __name__ == "__main__":
+from winpwnage.core.prints import print_info
+from winpwnage.core.scanner import scanner, function
+from winpwnage.core.utils import *
+import argparse
+import sys
+
+print("""
+        _
+  _ _ _|_|___ ___ _ _ _ ___ ___ ___ ___
+ | | | | |   | . | | | |   | .'| . | -_|
+ |_____|_|_|_|  _|_____|_|_|__,|_  |___|
+             |_|               |___|
+""")
+
+print_info("UAC level: {}".format(information().uac_level()))
+print_info("Build number: {}".format(information().build_number()))
+print_info("Running elevated: {}".format(information().admin()))
+print_info("Python version: {}.{}.{}\n".format(*sys.version_info))
+
+def main():
+	scan_cmds = ["uac", "persist", "elevate"]
+
+	parser = argparse.ArgumentParser()
+	parser.add_argument("-s", "--scan", nargs="+", required=False, help="Scan for either uac, persist or elevate method")
+	parser.add_argument("-u", "--use", nargs="+", required=False, help="Use either uac, persist or elevate method")
+	parser.add_argument("-i", "--id", nargs="+", required=False, help="Id of method")
+	parser.add_argument("-p", "--payload", nargs="+", required=False, help="Full path to payload, can include params")
+	parser.add_argument("-r", "--remove", action="store_true", required=False, help="Removes installed persistence")
+	args = parser.parse_args()
+
+	if args.scan:
+		if not all([_ in scan_cmds for _ in args.scan]):
+			parser.print_help()
+
+		scanner(**{scan_cmds[_]: scan_cmds[_] in args.scan for _ in range(3)}).start()
+
+	if args.use and args.id:
+		if not all([_ in scan_cmds for _ in args.use]):
+			parser.print_help()
+
+		if scan_cmds[0] in args.use and args.payload:
+			function(uac=True, persist=False, elevate=False).run(id=args.id[0], payload=args.payload)
+
+		if scan_cmds[1] in args.use:		
+			function(uac=False, persist=True, elevate=False).run(id=args.id[0], payload=args.payload,
+			add=(False if args.remove else True))
+
+		if scan_cmds[2] in args.use and args.payload:
+			function(uac=False, persist=False, elevate=True).run(id=args.id[0], payload=args.payload)
+
+if __name__ == "__main__":
 	main()