| """ |
| Neighbour Entries |
| |
| object abstractions for ARP and ND |
| """ |
| |
| from ipaddress import ip_address |
| from vpp_object import * |
| from vpp_papi import mac_pton, VppEnum |
| try: |
| text_type = unicode |
| except NameError: |
| text_type = str |
| |
| |
| def find_nbr(test, sw_if_index, nbr_addr, is_static=0, mac=None): |
| ip_addr = ip_address(text_type(nbr_addr)) |
| e = VppEnum.vl_api_ip_neighbor_flags_t |
| nbrs = test.vapi.ip_neighbor_dump(sw_if_index, |
| is_ipv6=(6 is ip_addr.version)) |
| |
| for n in nbrs: |
| if ip_addr == n.neighbor.ip_address and \ |
| is_static == (n.neighbor.flags & e.IP_API_NEIGHBOR_FLAG_STATIC): |
| if mac: |
| if mac == str(n.neighbor.mac_address): |
| return True |
| else: |
| return True |
| return False |
| |
| |
| class VppNeighbor(VppObject): |
| """ |
| ARP Entry |
| """ |
| |
| def __init__(self, test, sw_if_index, mac_addr, nbr_addr, |
| is_static=False, is_no_fib_entry=False): |
| self._test = test |
| self.sw_if_index = sw_if_index |
| self.mac_addr = mac_addr |
| self.nbr_addr = nbr_addr |
| |
| e = VppEnum.vl_api_ip_neighbor_flags_t |
| self.flags = e.IP_API_NEIGHBOR_FLAG_NONE |
| if is_static: |
| self.flags |= e.IP_API_NEIGHBOR_FLAG_STATIC |
| if is_no_fib_entry: |
| self.flags |= e.IP_API_NEIGHBOR_FLAG_NO_FIB_ENTRY |
| |
| def add_vpp_config(self): |
| r = self._test.vapi.ip_neighbor_add_del( |
| self.sw_if_index, |
| self.mac_addr, |
| self.nbr_addr, |
| is_add=1, |
| flags=self.flags) |
| self.stats_index = r.stats_index |
| self._test.registry.register(self, self._test.logger) |
| |
| def remove_vpp_config(self): |
| self._test.vapi.ip_neighbor_add_del( |
| self.sw_if_index, |
| self.mac_addr, |
| self.nbr_addr, |
| is_add=0, |
| flags=self.flags) |
| |
| def is_static(self): |
| e = VppEnum.vl_api_ip_neighbor_flags_t |
| return (self.flags & e.IP_API_NEIGHBOR_FLAG_STATIC) |
| |
| def query_vpp_config(self): |
| return find_nbr(self._test, |
| self.sw_if_index, |
| self.nbr_addr, |
| self.is_static()) |
| |
| def __str__(self): |
| return self.object_id() |
| |
| def object_id(self): |
| return ("%d:%s" % (self.sw_if_index, self.nbr_addr)) |
| |
| def get_stats(self): |
| c = self._test.statistics.get_counter("/net/adjacency") |
| return c[0][self.stats_index] |