import socket
from concurrent.futures import ThreadPoolExecutor

# Configurações do Módulo
ALVO = "scanme.nmap.org"  # Host de teste permitido
TIMEOUT = 1.0             # Tempo de espera por resposta (segundos)
THREADS = 50              # Número de conexões simultâneas

def scan_port(ip, port):
    """
    Tenta estabelecer uma conexão TCP com o IP e porta especificados.
    """
    # Cria o objeto socket: AF_INET (IPv4) e SOCK_STREAM (TCP)
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.settimeout(TIMEOUT)

    try:
        # connect_ex retorna 0 para sucesso (porta aberta)
        result = sock.connect_ex((ip, port))
        if result == 0:
            print(f"[+] Porta {port:5} [ABERTA]")
        sock.close()
    except Exception:
        pass

def main():
    print("-" * 60)
    print(f"Módulo: TCP Port Scanner | Projeto: 30 Lessons in Python")
    print("-" * 60)

    try:
        target_ip = socket.gethostbyname(ALVO)
        print(f"[*] Escaneando: {ALVO} ({target_ip})")
        print(f"[*] Utilizando {THREADS} threads...\n")

        # Lista de portas mais comuns para o exemplo
        common_ports = [21, 22, 23, 25, 53, 80, 110, 135, 139, 443, 445, 3306, 3389, 8080]

        # Execução paralela para otimização de tempo
        with ThreadPoolExecutor(max_workers=THREADS) as executor:
            for port in common_ports:
                executor.submit(scan_port, target_ip, port)

    except socket.gaierror:
        print("[!] Erro: Não foi possível resolver o domínio alvo.")
    except KeyboardInterrupt:
        print("\n[!] Escaneamento interrompido pelo usuário.")

    print("\n[*] Varredura finalizada.")

if __name__ == "__main__":
    main()