package_analysis.py

import pyshark
import sys

def protocol_capture(source):
    cap = pyshark.FileCapture(file_name, display_filter='udp')
    protocols = {}
    for pkt in cap:
        protocol = pkt.transport_layer
        if protocol not in protocols:
            protocols[protocol] = 0
        protocols[protocol] += 1

    for protocol, count in protocols.items():
        print(f"Protocolo {protocol}: {count} veces.")


def main():
    pcap = pyshark.FileCapture(sys.argv[1], display_filter='tcp')
    count = 0
    types_layers = []
    for package in pcap:
        count += 1
        

    print(count)  
    #print(f"The number of packages captured is : {count}")
if __name__ == '__main__':
    main()