blob: 9bbcb76586c0cf91e2755bbb06f4ba5eba6b868a [file] [log] [blame]
from unittest import TestCase, main
from scapy.layers.l2 import Ether, Dot1Q
from common.frameio.frameio import BpfProgramFilter
class TestBpf(TestCase):
def test_bpf1(self):
vid = 4090
pcp = 7
frame_match = 'ether[14:2] = 0x{:01x}{:03x}'.format(pcp << 1, vid)
filter = BpfProgramFilter(frame_match)
self.assertTrue(filter(str(Ether()/Dot1Q(prio=pcp, vlan=vid))))
self.assertFalse(filter(str(Ether()/Dot1Q(prio=pcp, vlan=4000))))
def test_bpf2(self):
vid1 = 4090
pcp1 = 7
frame_match_case1 = 'ether[14:2] = 0x{:01x}{:03x}'.format(
pcp1 << 1, vid1)
vid2 = 4000
frame_match_case2 = '(ether[14:2] & 0xfff) = 0x{:03x}'.format(vid2)
filter = BpfProgramFilter('{} or {}'.format(
frame_match_case1, frame_match_case2))
self.assertTrue(filter(str(Ether()/Dot1Q(prio=pcp1, vlan=vid1))))
self.assertTrue(filter(str(Ether()/Dot1Q(vlan=vid2))))
self.assertFalse(filter(str(Ether()/Dot1Q(vlan=4001))))
if __name__ == '__main__':
main()