Use IP address types on UDP encap API
Change-Id: I3c714c519b6d0009329b50947ce250c18ee2a85a Signed-off-by: Neale Ranns <nranns@cisco.com>
This commit is contained in:
@ -5,13 +5,21 @@
|
||||
|
||||
from vpp_object import *
|
||||
from socket import inet_pton, inet_ntop, AF_INET, AF_INET6
|
||||
from vpp_ip import *
|
||||
|
||||
|
||||
def find_udp_encap(test, id):
|
||||
def find_udp_encap(test, ue):
|
||||
encaps = test.vapi.udp_encap_dump()
|
||||
for e in encaps:
|
||||
if id == e.id:
|
||||
if ue.id == e.udp_encap.id \
|
||||
and compare_ip_address(e.udp_encap.src_ip.un,
|
||||
ue.src_ip.addr.ip_addr) \
|
||||
and compare_ip_address(e.udp_encap.dst_ip.un,
|
||||
ue.dst_ip.addr.ip_addr) \
|
||||
and e.udp_encap.dst_port == ue.dst_port \
|
||||
and e.udp_encap.src_port == ue.src_port:
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
|
||||
@ -24,48 +32,32 @@ class VppUdpEncap(VppObject):
|
||||
dst_ip,
|
||||
src_port,
|
||||
dst_port,
|
||||
table_id=0,
|
||||
is_ip6=0):
|
||||
table_id=0):
|
||||
self._test = test
|
||||
self.id = id
|
||||
self.table_id = table_id
|
||||
self.is_ip6 = is_ip6
|
||||
self.src_ip_s = src_ip
|
||||
self.dst_ip_s = dst_ip
|
||||
if is_ip6:
|
||||
self.src_ip = inet_pton(AF_INET6, src_ip)
|
||||
self.dst_ip = inet_pton(AF_INET6, dst_ip)
|
||||
else:
|
||||
self.src_ip = inet_pton(AF_INET, src_ip)
|
||||
self.dst_ip = inet_pton(AF_INET, dst_ip)
|
||||
self.src_ip = VppIpAddress(src_ip)
|
||||
self.dst_ip = VppIpAddress(dst_ip)
|
||||
self.src_port = src_port
|
||||
self.dst_port = dst_port
|
||||
|
||||
def add_vpp_config(self):
|
||||
self._test.vapi.udp_encap_add_del(
|
||||
self._test.vapi.udp_encap_add(
|
||||
self.id,
|
||||
self.src_ip,
|
||||
self.dst_ip,
|
||||
self.src_ip.encode(),
|
||||
self.dst_ip.encode(),
|
||||
self.src_port,
|
||||
self.dst_port,
|
||||
self.table_id,
|
||||
is_ip6=self.is_ip6,
|
||||
is_add=1)
|
||||
self.table_id)
|
||||
self._test.registry.register(self, self._test.logger)
|
||||
|
||||
def remove_vpp_config(self):
|
||||
self._test.vapi.udp_encap_add_del(
|
||||
self.id,
|
||||
self.src_ip,
|
||||
self.dst_ip,
|
||||
self.src_port,
|
||||
self.dst_port,
|
||||
self.table_id,
|
||||
is_ip6=self.is_ip6,
|
||||
is_add=0)
|
||||
self._test.vapi.udp_encap_del(self.id)
|
||||
|
||||
def query_vpp_config(self):
|
||||
return find_udp_encap(self._test, self.id)
|
||||
return find_udp_encap(self._test, self)
|
||||
|
||||
def __str__(self):
|
||||
return self.object_id()
|
||||
|
Reference in New Issue
Block a user