vpp/test/vpp_neighbor.py
Neale Ranns 39f9d8bd22 [Proxy] ARP tests
Change-Id: I40d6d763b55a26cdee0afef85d1acdd19dd10dd6
Signed-off-by: Neale Ranns <nranns@cisco.com>
2017-02-27 12:10:50 +00:00

78 lines
2.1 KiB
Python

"""
Neighbour Entries
object abstractions for ARP and ND
"""
from socket import inet_pton, inet_ntop, AF_INET, AF_INET6
from vpp_object import *
from util import mactobinary
def find_nbr(test, sw_if_index, ip_addr, is_static=0, inet=AF_INET):
nbrs = test.vapi.ip_neighbor_dump(sw_if_index,
is_ipv6=1 if AF_INET6 == inet else 0)
if inet == AF_INET:
s = 4
else:
s = 16
nbr_addr = inet_pton(inet, ip_addr)
for n in nbrs:
if nbr_addr == n.ip_address[:s] \
and is_static == n.is_static:
return True
return False
class VppNeighbor(VppObject):
"""
ARP Entry
"""
def __init__(self, test, sw_if_index, mac_addr, nbr_addr,
af=AF_INET, is_static=0):
self._test = test
self.sw_if_index = sw_if_index
self.mac_addr = mactobinary(mac_addr)
self.af = af
self.is_static = is_static
self.nbr_addr = inet_pton(af, nbr_addr)
def add_vpp_config(self):
self._test.vapi.ip_neighbor_add_del(
self.sw_if_index,
self.mac_addr,
self.nbr_addr,
is_add=1,
is_ipv6=1 if AF_INET6 == self.af else 0,
is_static=self.is_static)
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_ipv6=1 if AF_INET6 == self.af else 0,
is_add=0,
is_static=self.is_static)
def query_vpp_config(self):
dump = self._test.vapi.ip_neighbor_dump(
self.sw_if_index,
is_ipv6=1 if AF_INET6 == self.af else 0)
for n in dump:
if self.nbr_addr == n.ip_address \
and self.is_static == n.is_static:
return True
return False
def __str__(self):
return self.object_id()
def object_id(self):
return ("%d:%s"
% (self.sw_if_index,
inet_ntop(self.af, self.nbr_addr)))