1 from scapy.fields import *
2 from vpp_object import *
5 class VppLispLocatorSet(VppObject):
6 """ Represents LISP locator set in VPP """
8 def __init__(self, test, ls_name):
10 self._ls_name = ls_name
20 def add_vpp_config(self):
21 self.test.vapi.lisp_locator_set(ls_name=self._ls_name)
22 self._test.registry.register(self, self.test.logger)
24 def get_lisp_locator_sets_dump_entry(self):
25 result = self.test.vapi.lisp_locator_set_dump()
27 if ls.ls_name.strip('\x00') == self._ls_name:
31 def query_vpp_config(self):
32 return self.get_lisp_locator_sets_dump_entry() is not None
34 def remove_vpp_config(self):
35 self.test.vapi.lisp_locator_set(ls_name=self._ls_name, is_add=0)
38 return 'lisp-locator-set-%s' % self._ls_name
41 class VppLispLocator(VppObject):
42 """ Represents LISP locator in VPP """
44 def __init__(self, test, sw_if_index, ls_name, priority=1, weight=1):
46 self._sw_if_index = sw_if_index
47 self._ls_name = ls_name
48 self._priority = priority
53 """ Test which created this locator """
58 """ Locator set name """
62 def sw_if_index(self):
63 return self._sw_if_index
73 def add_vpp_config(self):
74 self.test.vapi.lisp_locator(ls_name=self._ls_name,
75 sw_if_index=self._sw_if_index,
76 priority=self._priority,
78 self._test.registry.register(self, self.test.logger)
80 def get_lisp_locator_dump_entry(self):
81 locators = self.test.vapi.lisp_locator_dump(
82 is_index_set=0, ls_name=self._ls_name)
83 for locator in locators:
84 if locator.sw_if_index == self._sw_if_index:
88 def query_vpp_config(self):
89 locator = self.get_lisp_locator_dump_entry()
90 return locator is not None
92 def remove_vpp_config(self):
93 self.test.vapi.lisp_locator(
94 ls_name=self._ls_name, sw_if_index=self._sw_if_index,
95 priority=self._priority, weight=self._weight, is_add=0)
96 self._test.registry.register(self, self.test.logger)
99 return 'lisp-locator-%s-%d' % (self._ls_name, self._sw_if_index)
102 class LispEIDType(object):
108 class LispKeyIdType(object):
114 class LispEID(object):
115 """ Lisp endpoint identifier """
116 def __init__(self, eid):
119 # find out whether EID is ip4 prefix, ip6 prefix or MAC
120 if self.eid.find("/") != -1:
121 if self.eid.find(":") == -1:
122 self.eid_type = LispEIDType.IP4
125 self.eid_type = LispEIDType.IP6
126 self.data_length = 16
128 self.eid_address = self.eid.split("/")[0]
129 self.prefix_length = int(self.eid.split("/")[1])
130 elif self.eid.count(":") == 5: # MAC address
131 self.eid_type = LispEIDType.MAC
132 self.eid_address = self.eid
133 self.prefix_length = 0
136 raise Exception('Unsupported EID format {}!'.format(eid))
139 if self.eid_type == LispEIDType.IP4:
140 return socket.inet_pton(socket.AF_INET, self.eid_address)
141 elif self.eid_type == LispEIDType.IP6:
142 return socket.inet_pton(socket.AF_INET6, self.eid_address)
143 elif self.eid_type == LispEIDType.MAC:
144 return Exception('Unimplemented')
145 raise Exception('Unknown EID type {}!'.format(self.eid_type))
148 class VppLispMapping(VppObject):
149 """ Represents common features for remote and local LISP mapping in VPP """
151 def __init__(self, test, eid, vni=0, priority=1, weight=1):
152 self._eid = LispEID(eid)
154 self._priority = priority
155 self._weight = weight
172 return self._priority
178 def get_lisp_mapping_dump_entry(self):
179 return self.test.vapi.lisp_eid_table_dump(
180 eid_set=1, prefix_length=self._eid.prefix_length,
181 vni=self._vni, eid_type=self._eid.eid_type, eid=str(self._eid))
183 def query_vpp_config(self):
184 mapping = self.get_lisp_mapping_dump_entry()
188 return 'lisp-mapping-[%s]-%s-%s-%s' % (
189 self.vni, self.eid, self.priority, self.weight)
192 class VppLocalMapping(VppLispMapping):
193 """ LISP Local mapping """
194 def __init__(self, test, eid, ls_name, vni=0, priority=1, weight=1,
195 key_id=LispKeyIdType.NONE, key=''):
196 super(VppLocalMapping, self).__init__(test, eid, vni, priority, weight)
197 self._ls_name = ls_name
198 self._key_id = key_id
213 def add_vpp_config(self):
214 self.test.vapi.lisp_local_mapping(
215 ls_name=self._ls_name, eid_type=self._eid.eid_type,
216 eid=str(self._eid), prefix_len=self._eid.prefix_length,
217 vni=self._vni, key_id=self._key_id, key=self._key)
218 self._test.registry.register(self, self.test.logger)
220 def remove_vpp_config(self):
221 self.test.vapi.lisp_local_mapping(
222 ls_name=self._ls_name, eid_type=self._eid.eid_type,
223 eid=str(self._eid), prefix_len=self._eid.prefix_length,
224 vni=self._vni, is_add=0)
227 return 'lisp-eid-local-mapping-%s[%d]' % (self._eid, self._vni)
230 class VppRemoteMapping(VppLispMapping):
232 def __init__(self, test, eid, rlocs=None, vni=0, priority=1, weight=1):
233 super(VppRemoteMapping, self).__init__(test, eid, vni, priority,
241 def add_vpp_config(self):
242 self.test.vapi.lisp_remote_mapping(
243 rlocs=self._rlocs, eid_type=self._eid.eid_type,
244 eid=str(self._eid), eid_prefix_len=self._eid.prefix_length,
245 vni=self._vni, rlocs_num=len(self._rlocs))
246 self._test.registry.register(self, self.test.logger)
248 def remove_vpp_config(self):
249 self.test.vapi.lisp_remote_mapping(
250 eid_type=self._eid.eid_type, eid=str(self._eid),
251 eid_prefix_len=self._eid.prefix_length, vni=self._vni,
252 is_add=0, rlocs_num=0)
255 return 'lisp-eid-remote-mapping-%s[%d]' % (self._eid, self._vni)
258 class VppLispAdjacency(VppObject):
259 """ Represents LISP adjacency in VPP """
261 def __init__(self, test, leid, reid, vni=0):
262 self._leid = LispEID(leid)
263 self._reid = LispEID(reid)
264 if self._leid.eid_type != self._reid.eid_type:
265 raise Exception('remote and local EID are different types!')
285 def add_vpp_config(self):
286 self.test.vapi.lisp_adjacency(
287 leid=str(self._leid),
288 reid=str(self._reid), eid_type=self._leid.eid_type,
289 leid_len=self._leid.prefix_length,
290 reid_len=self._reid.prefix_length, vni=self._vni)
291 self._test.registry.register(self, self.test.logger)
294 def eid_equal(eid, eid_type, eid_data, prefix_len):
295 if eid.eid_type != eid_type:
298 if eid_type == LispEIDType.IP4 or eid_type == LispEIDType.IP6:
299 if eid.prefix_length != prefix_len:
302 if str(eid) != eid_data[0:eid.data_length]:
307 def query_vpp_config(self):
308 res = self.test.vapi.lisp_adjacencies_get(vni=self._vni)
309 for adj in res.adjacencies:
310 if self.eid_equal(self._leid, adj.eid_type, adj.leid,
311 adj.leid_prefix_len) and \
312 self.eid_equal(self._reid, adj.eid_type, adj.reid,
313 adj.reid_prefix_len):
317 def remove_vpp_config(self):
318 self.test.vapi.lisp_adjacency(
319 leid=str(self._leid),
320 reid=str(self._reid), eid_type=self._leid.eid_type,
321 leid_len=self._leid.prefix_length,
322 reid_len=self._reid.prefix_length, vni=self._vni, is_add=0)
325 return 'lisp-adjacency-%s-%s[%d]' % (self._leid, self._reid, self._vni)