Packet_Sniffer

import socket
from Capture_Packets import PCAPFile
from nettypes import EthernetFrame, IPHeader, TCPSegment, UDPSegment
from utils import mac_addr



def main():
    conn = socket.socket(socket.AF_PACKET, socket.SOCK_RAW, socket.ntohs(3))
    pcap = PCAPFile('packets.pcap')
    while True:
        raw_data, addr = conn.recvfrom(65535)
        ethframe = EthernetFrame(raw_data)
        print(ethframe)
        if ethframe.protocol == 8:
            ipheader = IPHeader(ethframe.leftover_data)
            print(ipheader)
            if ipheader.protocol == 6:
                tcp = TCPSegment(ipheader.leftover_data)
                print(tcp)
            elif ipheader.protocol == 17:
                udp = UDPSegment(ipheader.leftover_data)
                print(udp)
        #print(ipheader.__dict__)
        pcap.write(raw_data)
        #print(raw_data)

    pcap.close()




if __name__ == '__main__':
    main()