Add port-sniffer task

submissions/AQuaZar/port-sniffer/sniffer.py

@@ -0,0 +1,65 @@
+import sys
+import argparse
+import socket
+
+
+def range_type(astr, minimal=0, maximal=65535):
+    value = int(astr)
+    if minimal <= value <= maximal:
+        return value
+    else:
+        raise argparse.ArgumentTypeError(f"value not in range {minimal}-{maximal}")
+
+
+parser = argparse.ArgumentParser(
+    description="""Output a list of opened TCP ports on specified host. 
+    By default all ports in range (0..65535) will be checked"""
+)
+
+parser.add_argument(
+    "--host",
+    metavar="Host",
+    type=str,
+    help="IPv4 address or domain name, which ports to check e.g. www.google.com, 192.168.0.1, e.t.c ",
+    required=True,
+)
+parser.add_argument(
+    "--ports",
+    metavar=("First", "Last"),
+    type=range_type,
+    nargs=2,
+    default=[0, 65535],
+    help="To check specific range of ports, define first and last port from range, e.g. --ports 69 420",
+)
+
+args = parser.parse_args()
+
+
+def port_scan(host, first, last):
+    print("Searching")
+    open_ports = []
+    for port in range(first, last + 1):
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.settimeout(0.3)
+        result = sock.connect_ex((host, port))
+
+        if result == 0:
+            open_ports.append(port)
+            print(".", end="")
+            sys.stdout.flush()
+        sock.close()
+    print("\n")
+    return open_ports
+
+
+if __name__ == "__main__":
+    try:
+        ports = port_scan(args.host, args.ports[0], args.ports[1])
+    except (KeyboardInterrupt, socket.gaierror, socket.error) as error:
+        print(error)
+        sys.exit()
+
+    if ports:
+        print("Opened: " + ", ".join(map(str, ports)))
+    else:
+        print("Not found any opened ports on desired host")