3 from vpp_object import VppObject
6 class VppLispLocatorSet(VppObject):
7 """ Represents LISP locator set in VPP """
9 def __init__(self, test, ls_name):
11 self._ls_name = ls_name
21 def add_vpp_config(self):
22 self.test.vapi.lisp_add_del_locator_set(ls_name=self._ls_name)
23 self._test.registry.register(self, self.test.logger)
25 def get_lisp_locator_sets_dump_entry(self):
26 result = self.test.vapi.lisp_locator_set_dump()
28 if ls.ls_name.strip(b'\x00') == self._ls_name:
32 def query_vpp_config(self):
33 return self.get_lisp_locator_sets_dump_entry() is not None
35 def remove_vpp_config(self):
36 self.test.vapi.lisp_add_del_locator_set(ls_name=self._ls_name,
40 return 'lisp-locator-set-%s' % self._ls_name
43 class VppLispLocator(VppObject):
44 """ Represents LISP locator in VPP """
46 def __init__(self, test, sw_if_index, ls_name, priority=1, weight=1):
48 self._sw_if_index = sw_if_index
49 self._ls_name = ls_name
50 self._priority = priority
55 """ Test which created this locator """
60 """ Locator set name """
64 def sw_if_index(self):
65 return self._sw_if_index
75 def add_vpp_config(self):
76 self.test.vapi.lisp_add_del_locator(ls_name=self._ls_name,
77 sw_if_index=self._sw_if_index,
78 priority=self._priority,
80 self._test.registry.register(self, self.test.logger)
82 def get_lisp_locator_dump_entry(self):
83 locators = self.test.vapi.lisp_locator_dump(
84 is_index_set=0, ls_name=self._ls_name)
85 for locator in locators:
86 if locator.sw_if_index == self._sw_if_index:
90 def query_vpp_config(self):
91 locator = self.get_lisp_locator_dump_entry()
92 return locator is not None
94 def remove_vpp_config(self):
95 self.test.vapi.lisp_add_del_locator(
96 ls_name=self._ls_name, sw_if_index=self._sw_if_index,
97 priority=self._priority, weight=self._weight, is_add=0)
98 self._test.registry.register(self, self.test.logger)
101 return 'lisp-locator-%s-%d' % (self._ls_name, self._sw_if_index)
104 class LispEIDType(object):
110 class LispKeyIdType(object):
116 class LispEID(object):
117 """ Lisp endpoint identifier """
118 def __init__(self, eid):
121 # find out whether EID is ip4 prefix, ip6 prefix or MAC
122 if self.eid.find("/") != -1:
123 if self.eid.find(":") == -1:
124 self.eid_type = LispEIDType.IP4
127 self.eid_type = LispEIDType.IP6
128 self.data_length = 16
130 self.eid_address = self.eid.split("/")[0]
131 self.prefix_length = int(self.eid.split("/")[1])
132 elif self.eid.count(":") == 5: # MAC address
133 self.eid_type = LispEIDType.MAC
134 self.eid_address = self.eid
135 self.prefix_length = 0
138 raise Exception('Unsupported EID format {!s}!'.format(eid))
142 if self.eid_type == LispEIDType.IP4:
143 return socket.inet_pton(socket.AF_INET, self.eid_address)
144 elif self.eid_type == LispEIDType.IP6:
145 return socket.inet_pton(socket.AF_INET6, self.eid_address)
146 elif self.eid_type == LispEIDType.MAC:
147 return Exception('Unimplemented')
148 raise Exception('Unknown EID type {!s}!'.format(self.eid_type))
151 class VppLispMapping(VppObject):
152 """ Represents common features for remote and local LISP mapping in VPP """
154 def __init__(self, test, eid, vni=0, priority=1, weight=1):
155 self._eid = LispEID(eid)
157 self._priority = priority
158 self._weight = weight
175 return self._priority
181 def get_lisp_mapping_dump_entry(self):
182 return self.test.vapi.lisp_eid_table_dump(
183 eid_set=1, prefix_length=self._eid.prefix_length,
184 vni=self._vni, eid_type=self._eid.eid_type,
185 eid=self._eid.packed)
187 def query_vpp_config(self):
188 mapping = self.get_lisp_mapping_dump_entry()
192 return 'lisp-mapping-[%s]-%s-%s-%s' % (
193 self.vni, self.eid, self.priority, self.weight)
196 class VppLocalMapping(VppLispMapping):
197 """ LISP Local mapping """
198 def __init__(self, test, eid, ls_name, vni=0, priority=1, weight=1,
199 key_id=LispKeyIdType.NONE, key=''):
200 super(VppLocalMapping, self).__init__(test, eid, vni, priority, weight)
201 self._ls_name = ls_name
202 self._key_id = key_id
217 def add_vpp_config(self):
218 self.test.vapi.lisp_add_del_local_eid(
219 ls_name=self._ls_name, eid_type=self._eid.eid_type,
220 eid=self._eid.packed, prefix_len=self._eid.prefix_length,
221 vni=self._vni, key_id=self._key_id, key=self._key)
222 self._test.registry.register(self, self.test.logger)
224 def remove_vpp_config(self):
225 self.test.vapi.lisp_add_del_local_eid(
226 ls_name=self._ls_name, eid_type=self._eid.eid_type,
227 eid=self._eid.packed, prefix_len=self._eid.prefix_length,
228 vni=self._vni, is_add=0)
231 return 'lisp-eid-local-mapping-%s[%d]' % (self._eid, self._vni)
234 class VppRemoteMapping(VppLispMapping):
236 def __init__(self, test, eid, rlocs=None, vni=0, priority=1, weight=1):
237 super(VppRemoteMapping, self).__init__(test, eid, vni, priority,
245 def add_vpp_config(self):
246 self.test.vapi.lisp_add_del_remote_mapping(
247 rlocs=self._rlocs, eid_type=self._eid.eid_type,
248 eid=self._eid.packed, eid_prefix_len=self._eid.prefix_length,
249 vni=self._vni, rlocs_num=len(self._rlocs))
250 self._test.registry.register(self, self.test.logger)
252 def remove_vpp_config(self):
253 self.test.vapi.lisp_add_del_remote_mapping(
254 eid_type=self._eid.eid_type, eid=self._eid.packed,
255 eid_prefix_len=self._eid.prefix_length, vni=self._vni,
256 is_add=0, rlocs_num=0)
259 return 'lisp-eid-remote-mapping-%s[%d]' % (self._eid, self._vni)
262 class VppLispAdjacency(VppObject):
263 """ Represents LISP adjacency in VPP """
265 def __init__(self, test, leid, reid, vni=0):
266 self._leid = LispEID(leid)
267 self._reid = LispEID(reid)
268 if self._leid.eid_type != self._reid.eid_type:
269 raise Exception('remote and local EID are different types!')
289 def add_vpp_config(self):
290 self.test.vapi.lisp_add_del_adjacency(
291 leid=self._leid.packed,
292 reid=self._reid.packed, eid_type=self._leid.eid_type,
293 leid_len=self._leid.prefix_length,
294 reid_len=self._reid.prefix_length, vni=self._vni)
295 self._test.registry.register(self, self.test.logger)
298 def eid_equal(eid, eid_type, eid_data, prefix_len):
299 if eid.eid_type != eid_type:
302 if eid_type == LispEIDType.IP4 or eid_type == LispEIDType.IP6:
303 if eid.prefix_length != prefix_len:
306 if eid.packed != eid_data[0:eid.data_length]:
311 def query_vpp_config(self):
312 res = self.test.vapi.lisp_adjacencies_get(vni=self._vni)
313 for adj in res.adjacencies:
314 if self.eid_equal(self._leid, adj.eid_type, adj.leid,
315 adj.leid_prefix_len) and \
316 self.eid_equal(self._reid, adj.eid_type, adj.reid,
317 adj.reid_prefix_len):
321 def remove_vpp_config(self):
322 self.test.vapi.lisp_add_del_adjacency(
323 leid=self._leid.packed,
324 reid=self._reid.packed, eid_type=self._leid.eid_type,
325 leid_len=self._leid.prefix_length,
326 reid_len=self._reid.prefix_length, vni=self._vni, is_add=0)
329 return 'lisp-adjacency-%s-%s[%d]' % (self._leid, self._reid, self._vni)