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 class VppLocalMapping(VppLispMapping):
189 """ LISP Local mapping """
190 def __init__(self, test, eid, ls_name, vni=0, priority=1, weight=1,
191 key_id=LispKeyIdType.NONE, key=''):
192 super(VppLocalMapping, self).__init__(test, eid, vni, priority, weight)
193 self._ls_name = ls_name
194 self._key_id = key_id
209 def add_vpp_config(self):
210 self.test.vapi.lisp_local_mapping(
211 ls_name=self._ls_name, eid_type=self._eid.eid_type,
212 eid=str(self._eid), prefix_len=self._eid.prefix_length,
213 vni=self._vni, key_id=self._key_id, key=self._key)
214 self._test.registry.register(self, self.test.logger)
216 def remove_vpp_config(self):
217 self.test.vapi.lisp_local_mapping(
218 ls_name=self._ls_name, eid_type=self._eid.eid_type,
219 eid=str(self._eid), prefix_len=self._eid.prefix_length,
220 vni=self._vni, is_add=0)
223 return 'lisp-eid-local-mapping-%s[%d]' % (self._eid, self._vni)
226 class VppRemoteMapping(VppLispMapping):
228 def __init__(self, test, eid, rlocs=None, vni=0, priority=1, weight=1):
229 super(VppRemoteMapping, self).__init__(test, eid, vni, priority,
237 def add_vpp_config(self):
238 self.test.vapi.lisp_remote_mapping(
239 rlocs=self._rlocs, eid_type=self._eid.eid_type,
240 eid=str(self._eid), eid_prefix_len=self._eid.prefix_length,
241 vni=self._vni, rlocs_num=len(self._rlocs))
242 self._test.registry.register(self, self.test.logger)
244 def remove_vpp_config(self):
245 self.test.vapi.lisp_remote_mapping(
246 eid_type=self._eid.eid_type, eid=str(self._eid),
247 eid_prefix_len=self._eid.prefix_length, vni=self._vni,
248 is_add=0, rlocs_num=0)
251 return 'lisp-eid-remote-mapping-%s[%d]' % (self._eid, self._vni)
254 class VppLispAdjacency(VppObject):
255 """ Represents LISP adjacency in VPP """
257 def __init__(self, test, leid, reid, vni=0):
258 self._leid = LispEID(leid)
259 self._reid = LispEID(reid)
260 if self._leid.eid_type != self._reid.eid_type:
261 raise Exception('remote and local EID are different types!')
281 def add_vpp_config(self):
282 self.test.vapi.lisp_adjacency(
283 leid=str(self._leid),
284 reid=str(self._reid), eid_type=self._leid.eid_type,
285 leid_len=self._leid.prefix_length,
286 reid_len=self._reid.prefix_length, vni=self._vni)
287 self._test.registry.register(self, self.test.logger)
290 def eid_equal(eid, eid_type, eid_data, prefix_len):
291 if eid.eid_type != eid_type:
294 if eid_type == LispEIDType.IP4 or eid_type == LispEIDType.IP6:
295 if eid.prefix_length != prefix_len:
298 if str(eid) != eid_data[0:eid.data_length]:
303 def query_vpp_config(self):
304 res = self.test.vapi.lisp_adjacencies_get(vni=self._vni)
305 for adj in res.adjacencies:
306 if self.eid_equal(self._leid, adj.eid_type, adj.leid,
307 adj.leid_prefix_len) and \
308 self.eid_equal(self._reid, adj.eid_type, adj.reid,
309 adj.reid_prefix_len):
313 def remove_vpp_config(self):
314 self.test.vapi.lisp_adjacency(
315 leid=str(self._leid),
316 reid=str(self._reid), eid_type=self._leid.eid_type,
317 leid_len=self._leid.prefix_length,
318 reid_len=self._reid.prefix_length, vni=self._vni, is_add=0)
321 return 'lisp-adjacency-%s-%s[%d]' % (self._leid, self._reid, self._vni)