Add testing for BPF module.

Author: justinemarie

bessctl/conf/testing/module_tests/bpf.py

@@ -0,0 +1,60 @@
+# CRASH TEST ##
+
+filters = [
+        "tcp src port 92",
+        "len <= 1000",
+        "ether proto 0x800",
+        "ip proto 47 or ip6 proto 47",
+        "ip host 22.22.22.22"
+        ]
+
+filter0 = {"priority": 0, "filter": filters[0], "gate": 1}
+bpf0::BPF()
+bpf0.add(filters=[filter0])
+CRASH_TEST_INPUTS.append([bpf0, 1, 2])
+
+bpf1::BPF()
+for i, exp in enumerate(filters):
+    bpf1.add(filters=[{"priority": i, "filter": exp, "gate": i}])
+
+CRASH_TEST_INPUTS.append([bpf1, 1, len(filters)])
+
+# OUTPUT TEST ##
+
+# Test basic output/steering with single rule
+bpf2::BPF()
+bpf2.add(filters=[filter0])
+packet1 = gen_packet(scapy.UDP, '12.34.56.78', '12.34.56.78')
+packet2 = gen_packet(scapy.TCP, '12.34.56.78', '12.34.56.78', srcport=92)
+
+OUTPUT_TEST_INPUTS.append([bpf2, 1, 2,
+    [{'input_port': 0,
+        'input_packet': packet1,
+        'output_port': 0,
+        'output_packet': packet1},
+     {'input_port': 0,
+         'input_packet': packet2,
+         'output_port': 1,
+         'output_packet': packet2}]])
+
+# Test multiple rules with priorities
+bpf3::BPF()
+bpf3.add(filters=[{"priority": 2, "filter": filters[0], "gate": 1}])
+bpf3.add(filters=[{"priority": 1, "filter": filters[4], "gate": 2}])
+packet1 = gen_packet(scapy.UDP, '22.22.22.22', '12.34.56.78', srcport=700)
+packet2 = gen_packet(scapy.TCP, '12.34.56.78', '22.22.22.22', srcport=92)
+packet3 = gen_packet(scapy.TCP, '12.34.56.78', '12.34.56.78', srcport=700)
+
+OUTPUT_TEST_INPUTS.append([bpf3, 1, 3,
+    [{'input_port': 0,
+        'input_packet': packet1,
+        'output_port': 2,
+        'output_packet': packet1},
+     {'input_port': 0,
+         'input_packet': packet2,
+         'output_port': 1,
+         'output_packet': packet2},
+     {'input_port': 0,
+         'input_packet': packet3,
+         'output_port': 0,
+         'output_packet': packet3}]])

bessctl/conf/testing/run_module_tests.bess

@@ -31,10 +31,10 @@ def gen_socket_and_port(sockname):
 # All the other fields -- Ether, ports -- are dummy values.
 
 
-def gen_packet(proto, src_ip, dst_ip, ip_ttl=64):
+def gen_packet(proto, src_ip, dst_ip, ip_ttl=64, srcport=1001, dstport=1002):
     eth = scapy.Ether(src='02:1e:67:9f:4d:ae', dst='06:16:3e:1b:72:32')
     ip = scapy.IP(src=src_ip, dst=dst_ip, ttl=ip_ttl)
-    udp = proto(sport=10001, dport=10002)
+    udp = proto(sport=srcport, dport=dstport)
     payload = 'helloworld'
     pkt = eth / ip / udp / payload
     return pkt