Update port_scanner.py

Author: batubyte

port_scanner.py

@@ -2,17 +2,23 @@
 
 """
 Project: Port Scanner
-Description: Installs and runs nmap with colored scan results
+Description: Async port scanner with Nmap support
 Author: batubyte
-Date: 2025-15-06
+Date: 2025-06-17
 """
 
 import subprocess
 import platform
+import argparse
+import asyncio
 import shutil
+import socket
 import sys
 import re
 
+DESCRIPTION = "Async port scanner with Nmap support"
+VERSION = "0.1.1"
+
 RED = "\033[31m"
 GREEN = "\033[32m"
 YELLOW = "\033[33m"
@@ -22,7 +28,7 @@
 
 
 def install_nmap():
-    if shutil.which("nmap") is not None:
+    if shutil.which("nmap"):
         return
 
     answer = input("Nmap not found. Install? [Y/n]: ").strip().lower()
@@ -34,16 +40,20 @@ def install_nmap():
         subprocess.run(["sudo", "apt-get", "update"], check=True)
         subprocess.run(["sudo", "apt-get", "install", "-y", "nmap"], check=True)
     elif system == "Windows":
-        print("Install nmap from https://nmap.org/download.html")
-        print("Then restart your system.")
+        print("Install nmap from https://nmap.org/download.html#windows")
+        print("Restart system after installation")
         sys.exit(1)
 
     if shutil.which("nmap") is None:
         print("Installation failed. Install nmap manually.")
         sys.exit(1)
 
 
-def print_output(output):
+def run_nmap(args):
+    cmd = ["nmap"] + args
+    result = subprocess.run(cmd, capture_output=True, text=True, check=True)
+    output = result.stdout
+
     ports_section = False
     for line in output.splitlines():
         line = line.strip()
@@ -73,24 +83,63 @@ def print_output(output):
             print(line)
 
 
-def nmap(args):
-    cmd = ["nmap"] + args
-    result = subprocess.run(cmd, capture_output=True, text=True, check=True)
-    print_output(result.stdout)
+async def scan_port(host, port, sem):
+    async with sem:
+        try:
+            reader, writer = await asyncio.open_connection(host, port)
+            print(f"{port}/tcp open")
+            writer.close()
+            await writer.wait_closed()
+        except:
+            pass
 
 
-def main():
+async def full_scan(host):
+    sem = asyncio.Semaphore(500)
+    tasks = [scan_port(host, port, sem) for port in range(1, 65536)]
     try:
+        await asyncio.gather(*tasks)
+    except asyncio.CancelledError:
+        pass
+
+
+def valid_host(host):
+    try:
+        socket.gethostbyname(host)
+        return True
+    except socket.error:
+        return False
+    
+
+def parse_args(parser):
+    parser.add_argument("-v", "--version", action="version", version=VERSION)
+    parser.add_argument("-n", nargs=argparse.REMAINDER, help="Run nmap")
+    parser.add_argument("target", nargs="?", help="IP address, hostname or domain")
+    return parser.parse_args()
+
+
+def main():
+    parser = argparse.ArgumentParser(description=DESCRIPTION)
+    args = parse_args(parser)
+        
+    if args.n:
         install_nmap()
-        if len(sys.argv) < 2:
+        if len(args.n) == 0:
             subprocess.run(["nmap", "-h"])
-            sys.exit(0)
-
-        nmap(sys.argv[1:])
-    except Exception as e:
-        print(f"Error: {e}")
+        else:
+            run_nmap(args.n)
+    elif args.target and valid_host(args.target):
+        asyncio.run(full_scan(args.target))
+    else:
+        parser.print_help()
         sys.exit(1)
 
 
 if __name__ == "__main__":
-    main()
+    try:
+        main()
+    except KeyboardInterrupt:
+        sys.exit(130)
+    except Exception as e:
+        print(f"Error: {e}")
+        sys.exit(1)