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('\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))
141 if self.eid_type == LispEIDType.IP4:
142 return socket.inet_pton(socket.AF_INET, self.eid_address)
143 elif self.eid_type == LispEIDType.IP6:
144 return socket.inet_pton(socket.AF_INET6, self.eid_address)
145 elif self.eid_type == LispEIDType.MAC:
146 return Exception('Unimplemented')
147 raise Exception('Unknown EID type {!s}!'.format(self.eid_type))
150 class VppLispMapping(VppObject):
151 """ Represents common features for remote and local LISP mapping in VPP """
153 def __init__(self, test, eid, vni=0, priority=1, weight=1):
154 self._eid = LispEID(eid)
156 self._priority = priority
157 self._weight = weight
174 return self._priority
180 def get_lisp_mapping_dump_entry(self):
181 return self.test.vapi.lisp_eid_table_dump(
182 eid_set=1, prefix_length=self._eid.prefix_length,
183 vni=self._vni, eid_type=self._eid.eid_type, eid=str(self._eid))
185 def query_vpp_config(self):
186 mapping = self.get_lisp_mapping_dump_entry()
190 return 'lisp-mapping-[%s]-%s-%s-%s' % (
191 self.vni, self.eid, self.priority, self.weight)
194 class VppLocalMapping(VppLispMapping):
195 """ LISP Local mapping """
196 def __init__(self, test, eid, ls_name, vni=0, priority=1, weight=1,
197 key_id=LispKeyIdType.NONE, key=''):
198 super(VppLocalMapping, self).__init__(test, eid, vni, priority, weight)
199 self._ls_name = ls_name
200 self._key_id = key_id
215 def add_vpp_config(self):
216 self.test.vapi.lisp_add_del_local_eid(
217 ls_name=self._ls_name, eid_type=self._eid.eid_type,
218 eid=str(self._eid), prefix_len=self._eid.prefix_length,
219 vni=self._vni, key_id=self._key_id, key=self._key)
220 self._test.registry.register(self, self.test.logger)
222 def remove_vpp_config(self):
223 self.test.vapi.lisp_add_del_local_eid(
224 ls_name=self._ls_name, eid_type=self._eid.eid_type,
225 eid=str(self._eid), prefix_len=self._eid.prefix_length,
226 vni=self._vni, is_add=0)
229 return 'lisp-eid-local-mapping-%s[%d]' % (self._eid, self._vni)
232 class VppRemoteMapping(VppLispMapping):
234 def __init__(self, test, eid, rlocs=None, vni=0, priority=1, weight=1):
235 super(VppRemoteMapping, self).__init__(test, eid, vni, priority,
243 def add_vpp_config(self):
244 self.test.vapi.lisp_add_del_remote_mapping(
245 rlocs=self._rlocs, eid_type=self._eid.eid_type,
246 eid=str(self._eid), eid_prefix_len=self._eid.prefix_length,
247 vni=self._vni, rlocs_num=len(self._rlocs))
248 self._test.registry.register(self, self.test.logger)
250 def remove_vpp_config(self):
251 self.test.vapi.lisp_add_del_remote_mapping(
252 eid_type=self._eid.eid_type, eid=str(self._eid),
253 eid_prefix_len=self._eid.prefix_length, vni=self._vni,
254 is_add=0, rlocs_num=0)
257 return 'lisp-eid-remote-mapping-%s[%d]' % (self._eid, self._vni)
260 class VppLispAdjacency(VppObject):
261 """ Represents LISP adjacency in VPP """
263 def __init__(self, test, leid, reid, vni=0):
264 self._leid = LispEID(leid)
265 self._reid = LispEID(reid)
266 if self._leid.eid_type != self._reid.eid_type:
267 raise Exception('remote and local EID are different types!')
287 def add_vpp_config(self):
288 self.test.vapi.lisp_add_del_adjacency(
289 leid=str(self._leid),
290 reid=str(self._reid), eid_type=self._leid.eid_type,
291 leid_len=self._leid.prefix_length,
292 reid_len=self._reid.prefix_length, vni=self._vni)
293 self._test.registry.register(self, self.test.logger)
296 def eid_equal(eid, eid_type, eid_data, prefix_len):
297 if eid.eid_type != eid_type:
300 if eid_type == LispEIDType.IP4 or eid_type == LispEIDType.IP6:
301 if eid.prefix_length != prefix_len:
304 if str(eid) != eid_data[0:eid.data_length]:
309 def query_vpp_config(self):
310 res = self.test.vapi.lisp_adjacencies_get(vni=self._vni)
311 for adj in res.adjacencies:
312 if self.eid_equal(self._leid, adj.eid_type, adj.leid,
313 adj.leid_prefix_len) and \
314 self.eid_equal(self._reid, adj.eid_type, adj.reid,
315 adj.reid_prefix_len):
319 def remove_vpp_config(self):
320 self.test.vapi.lisp_add_del_adjacency(
321 leid=str(self._leid),
322 reid=str(self._reid), eid_type=self._leid.eid_type,
323 leid_len=self._leid.prefix_length,
324 reid_len=self._reid.prefix_length, vni=self._vni, is_add=0)
327 return 'lisp-adjacency-%s-%s[%d]' % (self._leid, self._reid, self._vni)