import socket import sys host = sys.argv[1] mode = sys.argv[2] def check_port(host, port): sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.settimeout(5) result = sock.connect_ex((host, port)) if result == 0: print (f "{port} is open" ) else : print (f "{port} is not open" ) sock.close() def check_ports(host, start_port, end_port): open_ports = [] for port in range(start_port, end_port + 1): sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.settimeout(1) result = sock.connect_ex((host, port)) if result == 0: open_ports.append(port) sock.close() return open_ports if mode== "all" : open_ports = check_ports(host, 1, 65535) print ( "Open ports on {}: {}" .format(host, open_ports)) else : port = int(mode) check_port(host, port) |