trx_toolkit/trx_sniff.py: add support for reading from PCAP file

Change-Id: I8a3481fbea5f6c917ae5684d0b5b806f4a76ff78
This commit is contained in:
Vadim Yanitskiy 2019-07-24 02:35:14 +07:00
parent 7cde195e30
commit e92c9cd923
1 changed files with 23 additions and 7 deletions

View File

@ -61,18 +61,29 @@ class Application(ApplicationBase):
pkt_filter = "udp and (port %d or port %d)" \
% (self.argv.base_port + 2, self.argv.base_port + 102)
log.info("Listening on interface '%s'..." % self.argv.sniff_if)
# Arguments to be passed to scapy.all.sniff()
sniff_args = {
"prn" : self.pkt_handler,
"filter" : pkt_filter,
"store" : 0,
}
if self.argv.cap_file is not None:
log.info("Reading packets from '%s'..." % self.argv.cap_file)
sniff_args["offline"] = self.argv.cap_file
else:
log.info("Listening on interface '%s'..." % self.argv.sniff_if)
sniff_args["iface"] = self.argv.sniff_if
# Start sniffing...
scapy.all.sniff(iface = self.argv.sniff_if, store = 0,
filter = pkt_filter, prn = self.pkt_handler)
scapy.all.sniff(**sniff_args)
# Scapy registers its own signal handler
self.shutdown()
def pkt_handler(self, ether):
# Prevent loopback packet duplication
if self.argv.sniff_if == "lo":
if self.argv.sniff_if == "lo" and self.argv.cap_file is None:
self.lo_trigger = not self.lo_trigger
if not self.lo_trigger:
return
@ -197,9 +208,6 @@ class Application(ApplicationBase):
self.app_reg_logging_options(parser)
trx_group = parser.add_argument_group("TRX interface")
trx_group.add_argument("-i", "--sniff-interface",
dest = "sniff_if", type = str, default = "lo", metavar = "IF",
help = "Set network interface (default '%(default)s')")
trx_group.add_argument("-p", "--base-port",
dest = "base_port", type = int, default = 6700,
help = "Set base port number (default %(default)s)")
@ -207,6 +215,14 @@ class Application(ApplicationBase):
dest = "output_file", type = str,
help = "Write bursts to a capture file")
input_group = trx_group.add_mutually_exclusive_group()
input_group.add_argument("-i", "--sniff-interface",
dest = "sniff_if", type = str, default = "lo", metavar = "IF",
help = "Set network interface (default '%(default)s')")
input_group.add_argument("-r", "--capture-file",
dest = "cap_file", type = str, metavar = "FILE",
help = "Read packets from a PCAP file")
cnt_group = parser.add_argument_group("Count limitations (optional)")
cnt_group.add_argument("--frame-count", metavar = "N",
dest = "frame_count", type = int,