tests/sys/netinet/carp.py

   1 import pytest
   2 from atf_python.sys.net.tools import ToolsHelper
   3 from atf_python.sys.net.vnet import VnetTestTemplate
   4
   5 sc = None
   6
   7
   8 def filter_f(x):
   9     ip = x.getlayer(sc.IP)
  10     if not ip:
  11         return False
  12
  13     return ip.proto == 112
  14
  15
  16 class 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)