From 59ec642f0b70f2ab7b1d28070c5675b0cb3bc061 Mon Sep 17 00:00:00 2001
From: Mister-Joe <joseph.tredeau@outlook.com>
Date: Tue, 17 Jun 2025 08:16:03 -0700
Subject: [PATCH] add support for setup.py installation

---
 ntlmscan.py | 6 ++++--
 setup.py    | 8 ++++++++
 2 files changed, 12 insertions(+), 2 deletions(-)
 create mode 100644 setup.py

diff --git a/ntlmscan.py b/ntlmscan.py
index f700ddb..98509c6 100755
--- a/ntlmscan.py
+++ b/ntlmscan.py
@@ -92,8 +92,7 @@ def makeRequests(url_data):
         # print("Unexpected error:", sys.exc_info()[0])
         pass
 
-
-if __name__ == "__main__":
+def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--url", help="full url path to test")
     parser.add_argument("--host", help="a single host to search for ntlm dirs on")
@@ -193,3 +192,6 @@ def makeRequests(url_data):
 
     if args.nmap:
         nmapScanner(foundURLs)
+
+if __name__ == "__main__":
+    main()
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..7ba333d
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,8 @@
+from setuptools import setup
+
+setup(
+    name="ntlmscan",
+    version="1.0",
+    install_requires=[ "requests" ],
+    entry_points={ "console_scripts": [ "ntlmscan=ntlmscan:main" ] }
+)