1import pytest 2from atf_python.sys.net.tools import ToolsHelper 3from atf_python.sys.net.vnet import VnetTestTemplate 4 5sc = None 6 7 8def filter_f(x): 9 ip = x.getlayer(sc.IP) 10 if not ip: 11 return False 12 13 return ip.proto == 112 14 15 16class TestCarp(VnetTestTemplate): 17 REQUIRED_MODULES = ["carp"] 18 TOPOLOGY = { 19 "vnet1": {"ifaces": ["if1"]}, 20 "if1": {"prefixes4": [("192.0.2.1/24", "192.0.2.2/24")]}, 21 } 22 23 def setup_method(self, method): 24 global sc 25 if sc is None: 26 import scapy.all as _sc 27 28 sc = _sc 29 super().setup_method(method) 30 31 @classmethod 32 def check_carp_src_mac(self, pkts): 33 for p in pkts: 34 if not filter_f(p): 35 continue 36 37 print("Packet src mac {}".format(p.src)) 38 39 if p.src != "00:00:5e:00:01:01": 40 raise 41 42 def test_source_mac(self): 43 "Test carp packets source address" 44 45 if1 = self.vnet.iface_alias_map["if1"] 46 47 ToolsHelper.print_output( 48 "ifconfig {} add vhid 1 192.0.2.203/24".format(if1.name) 49 ) 50 51 carp_pkts = sc.sniff(iface=if1.name, stop_filter=filter_f, timeout=5) 52 53 self.check_carp_src_mac(carp_pkts) 54