5be4b869a4
BPF filter w/o optimization can take x2 - x3 more instructions, causing significant slow down in fast path. Enable pcap optimization by default via cli and introduce api v2 with pcap optimization control, keep v1 for a while as it exists in previous release already. Intriduce bpf filter cli dump, similar to tcpdump -d. Also fix memleak, function name typo, cli pcap format hint and add related tests. Type: improvement Signed-off-by: Vladislav Grishenko <themiron@yandex-team.ru> Change-Id: I92b2b519e92326f1b8e1a4dda6a3e3edc52f87ad
137 lines
4.4 KiB
Python
137 lines
4.4 KiB
Python
from framework import VppTestCase
|
|
from asfframework import VppTestRunner
|
|
import unittest
|
|
from config import config
|
|
from scapy.layers.l2 import Ether
|
|
from scapy.layers.inet import IP, UDP
|
|
from random import randint
|
|
|
|
|
|
@unittest.skipIf(
|
|
"bpf_trace_filter" in config.excluded_plugins,
|
|
"Exclude BPF Trace Filter plugin tests",
|
|
)
|
|
class TestBpfTraceFilter(VppTestCase):
|
|
"""BPF Trace filter test"""
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(TestBpfTraceFilter, cls).setUpClass()
|
|
try:
|
|
cls.create_pg_interfaces(range(2))
|
|
for i in cls.pg_interfaces:
|
|
i.config_ip4()
|
|
i.resolve_arp()
|
|
i.admin_up()
|
|
except Exception:
|
|
cls.tearDownClass()
|
|
raise
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
for i in cls.pg_interfaces:
|
|
i.unconfig_ip4()
|
|
i.admin_down()
|
|
super(TestBpfTraceFilter, cls).tearDownClass()
|
|
|
|
# reset trace filter before each test
|
|
def setUp(self):
|
|
super(TestBpfTraceFilter, self).setUp()
|
|
self.vapi.cli("set trace filter function vnet_is_packet_traced")
|
|
self.vapi.cli("clear trace")
|
|
|
|
def create_stream(self, src_if, dst_if, count):
|
|
packets = []
|
|
for i in range(count):
|
|
info = self.create_packet_info(src_if, dst_if)
|
|
payload = self.info_to_payload(info)
|
|
p = (
|
|
Ether(dst=src_if.local_mac, src=src_if.remote_mac)
|
|
/ IP(src=src_if.remote_ip4, dst=dst_if.remote_ip4)
|
|
/ UDP(sport=randint(49152, 65535), dport=5678 + i)
|
|
)
|
|
info.data = p.copy()
|
|
packets.append(p)
|
|
return packets
|
|
|
|
def test_bpf_trace_filter_cli(self):
|
|
"""BPF Trace filter test [CLI]"""
|
|
self.vapi.cli("set bpf trace filter {{tcp}}")
|
|
self.vapi.cli("set trace filter function bpf_trace_filter")
|
|
|
|
packets = self.create_stream(self.pg0, self.pg1, 3)
|
|
self.pg0.add_stream(packets)
|
|
self.pg_start(traceFilter=True)
|
|
|
|
# verify that bpf trace filter has been selected
|
|
reply = self.vapi.cli("show trace filter function")
|
|
self.assertIn(
|
|
"(*) name:bpf_trace_filter", reply, "BPF Trace filter is not selected"
|
|
)
|
|
|
|
# verify that trace is empty
|
|
reply = self.vapi.cli("show trace")
|
|
self.assertIn(
|
|
"No packets in trace buffer",
|
|
reply,
|
|
"Unexpected packets in the trace buffer",
|
|
)
|
|
|
|
reply = self.vapi.cli("show bpf trace filter")
|
|
self.assertIn("(000)", reply, "Unexpected bpf filter dump")
|
|
|
|
def test_bpf_trace_filter_vapi(self):
|
|
"""BPF Trace filter test [VAPI]"""
|
|
self.vapi.bpf_trace_filter_set(filter="tcp")
|
|
self.vapi.trace_set_filter_function(filter_function_name="bpf_trace_filter")
|
|
|
|
packets = self.create_stream(self.pg0, self.pg1, 3)
|
|
self.pg0.add_stream(packets)
|
|
self.pg_start(traceFilter=True)
|
|
|
|
# verify that bpf trace filter has been selected
|
|
reply = self.vapi.cli("show trace filter function")
|
|
self.assertIn(
|
|
"(*) name:bpf_trace_filter", reply, "BPF Trace filter is not selected"
|
|
)
|
|
|
|
# verify that trace is empty
|
|
reply = self.vapi.cli("show trace")
|
|
self.assertIn(
|
|
"No packets in trace buffer",
|
|
reply,
|
|
"Unexpected packets in the trace buffer",
|
|
)
|
|
|
|
def test_bpf_trace_filter_vapi_v2(self):
|
|
"""BPF Trace filter test [VAPI v2]"""
|
|
self.vapi.bpf_trace_filter_set_v2(filter="tcp or dst port 5678")
|
|
self.vapi.trace_set_filter_function(filter_function_name="bpf_trace_filter")
|
|
|
|
packets = self.create_stream(self.pg0, self.pg1, 3)
|
|
self.pg0.add_stream(packets)
|
|
self.pg_start(traceFilter=True)
|
|
|
|
# verify that bpf trace filter has been selected
|
|
reply = self.vapi.cli("show trace filter function")
|
|
self.assertIn(
|
|
"(*) name:bpf_trace_filter", reply, "BPF Trace filter is not selected"
|
|
)
|
|
|
|
# verify that trace is filtered
|
|
reply = self.vapi.cli("show trace")
|
|
self.assertIn(
|
|
"Packet 1\n",
|
|
reply,
|
|
"No expected packets in the trace buffer",
|
|
)
|
|
self.assertNotIn(
|
|
"Packet 2\n",
|
|
reply,
|
|
"Unexpected packets in the trace buffer",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(testRunner=VppTestRunner)
|