from scapy.all import sniff, IP, TCP, Raw
import logging

# Oculta mensagens de aviso do Scapy
logging.getLogger("scapy.runtime").setLevel(logging.ERROR)

def process_packet(packet):
    """
    Callback executada para cada pacote capturado.
    Filtra pacotes TCP que contenham dados (camada Raw).
    """
    if packet.haslayer(TCP) and packet.haslayer(Raw):
        payload = packet[Raw].load.decode(errors='ignore')

        # Filtramos palavras-chave comuns em ataques de interceptação
        keywords = ["user", "pass", "login", "username", "password", "set-cookie"]

        if any(key in payload.lower() for key in keywords):
            src_ip = packet[IP].src
            dst_ip = packet[IP].dst
            print(f"\n[!] POSSÍVEL DADO SENSÍVEL DETECTADO!")
            print(f"[*] Origem: {src_ip} -> Destino: {dst_ip}")
            print(f"[*] Conteúdo: {payload[:200]}...") # Exibe apenas os primeiros 200 caracteres
            print("-" * 50)

def main():
    print("=" * 60)
    print("Módulo: Network Sniffer | Projeto: 30 Lessons in Python")
    print("=" * 60)
    print("[*] Iniciando captura de tráfego (Apenas portas 80, 21, 23)...")
    print("[*] Pressione Ctrl+C para parar.\n")

    # sniff: Função principal de captura
    # filter: Sintaxe BPF para capturar apenas tráfego comum sem SSL/TLS
    # prn: Função que será chamada para cada pacote capturado
    # store: 0 para não guardar pacotes na memória (evita crash em capturas longas)
    try:
        sniff(filter="tcp port 80 or tcp port 21 or tcp port 23", 
              prn=process_packet, 
              store=0)
    except PermissionError:
        print("[!] Erro: Você precisa de privilégios de Administrador/Root para cheirar a rede.")
    except Exception as e:
        print(f"[!] Erro inesperado: {e}")

if __name__ == "__main__":
    main()