1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
@media only all and (prefers-color-scheme: dark) {
.highlight .hll { background-color: #49483e }
.highlight .c { color: #75715e } /* Comment */
.highlight .err { color: #960050; background-color: #1e0010 } /* Error */
.highlight .k { color: #66d9ef } /* Keyword */
.highlight .l { color: #ae81ff } /* Literal */
.highlight .n { color: #f8f8f2 } /* Name */
.highlight .o { color: #f92672 } /* Operator */
.highlight .p { color: #f8f8f2 } /* Punctuation */
.highlight .ch { color: #75715e } /* Comment.Hashbang */
.highlight .cm { color: #75715e } /* Comment.Multiline */
.highlight .cp { color: #75715e } /* Comment.Preproc */
.highlight .cpf { color: #75715e } /* Comment.PreprocFile */
.highlight .c1 { color: #75715e } /* Comment.Single */
.highlight .cs { color: #75715e } /* Comment.Special */
.highlight .gd { color: #f92672 } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .gi { color: #a6e22e } /* Generic.Inserted */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #75715e } /* Generic.Subheading */
.highlight .kc { color: #66d9ef } /* Keyword.Constant */
.highlight .kd { color: #66d9ef } /* Keyword.Declaration */
.highlight .kn { color: #f92672 } /* Keyword.Namespace */
.highlight .kp { color: #66d9ef } /* Keyword.Pseudo */
.highlight .kr { color: #66d9ef } /* Keyword.Reserved */
.highlight .kt { color: #66d9ef } /* Keyword.Type */
.highlight .ld { color: #e6db74 } /* Literal.Date */
.highlight .m { color: #ae81ff } /* Literal.Number */
.highlight .s { color: #e6db74 } /* Literal.String */
.highlight .na { color: #a6e22e } /* Name.Attribute */
.highlight .nb { color: #f8f8f2 } /* Name.Builtin */
.highlight .nc { color: #a6e22e } /* Name.Class */
.highlight .no { color: #66d9ef } /* Name.Constant */
.highlight .nd { color: #a6e22e } /* Name.Decorator */
.highlight .ni { color: #f8f8f2 } /* Name.Entity */
.highlight .ne { color: #a6e22e } /* Name.Exception */
.highlight .nf { color: #a6e22e } /* Name.Function */
.highlight .nl { color: #f8f8f2 } /* Name.Label */
.highlight .nn { color: #f8f8f2 } /* Name.Namespace */
.highlight .nx { color: #a6e22e } /* Name.Other */
.highlight .py { color: #f8f8f2 } /* Name.Property */
.highlight .nt { color: #f92672 } /* Name.Tag */
.highlight .nv { color: #f8f8f2 } /* Name.Variable */
.highlight .ow { color: #f92672 } /* Operator.Word */
.highlight .w { color: #f8f8f2 } /* Text.Whitespace */
.highlight .mb { color: #ae81ff } /* Literal.Number.Bin */
.highlight .mf { color: #ae81ff } /* Literal.Number.Float */
.highlight .mh { color: #ae81ff } /* Literal.Number.Hex */
.highlight .mi { color: #ae81ff } /* Literal.Number.Integer */
.highlight .mo { color: #ae81ff } /* Literal.Number.Oct */
.highlight .sa { color: #e6db74 } /* Literal.String.Affix */
.highlight .sb { color: #e6db74 } /* Literal.String.Backtick */
.highlight .sc { color: #e6db74 } /* Literal.String.Char */
.highlight .dl { color: #e6db74 } /* Literal.String.Delimiter */
.highlight .sd { color: #e6db74 } /* Literal.String.Doc */
.highlight .s2 { color: #e6db74 } /* Literal.String.Double */
.highlight .se { color: #ae81ff } /* Literal.String.Escape */
.highlight .sh { color: #e6db74 } /* Literal.String.Heredoc */
.highlight .si { color: #e6db74 } /* Literal.String.Interpol */
.highlight .sx { color: #e6db74 } /* Literal.String.Other */
.highlight .sr { color: #e6db74 } /* Literal.String.Regex */
.highlight .s1 { color: #e6db74 } /* Literal.String.Single */
.highlight .ss { color: #e6db74 } /* Literal.String.Symbol */
.highlight .bp { color: #f8f8f2 } /* Name.Builtin.Pseudo */
.highlight .fm { color: #a6e22e } /* Name.Function.Magic */
.highlight .vc { color: #f8f8f2 } /* Name.Variable.Class */
.highlight .vg { color: #f8f8f2 } /* Name.Variable.Global */
.highlight .vi { color: #f8f8f2 } /* Name.Variable.Instance */
.highlight .vm { color: #f8f8f2 } /* Name.Variable.Magic */
.highlight .il { color: #ae81ff } /* Literal.Number.Integer.Long */
}
@media (prefers-color-scheme: light) {
.highlight .hll { background-color: #ffffcc }
.highlight .c { color: #888888 } /* Comment */
.highlight .err { color: #a61717; background-color: #e3d2d2 } /* Error */
.highlight .k { color: #008800; font-weight: bold } /* Keyword */
.highlight .ch { color: #888888 } /* Comment.Hashbang */
.highlight .cm { color: #888888 } /* Comment.Multiline */
.highlight .cp { color: #cc0000; font-weight: bold } /* Comment.Preproc */
.highlight .cpf { color: #888888 } /* Comment.PreprocFile */
.highlight .c1 { color: #888888 } /* Comment.Single */
.highlight .cs { color: #cc0000; font-weight: bold; background-color: #fff0f0 } /* Comment.Special */
.highlight .gd { color: #000000; background-color: #ffdddd } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .gr { color: #aa0000 } /* Generic.Error */
.highlight .gh { color: #333333 } /* Generic.Heading */
.highlight .gi { color: #000000; background-color: #ddffdd } /* Generic.Inserted */
.highlight .go { color: #888888 } /* Generic.Output */
.highlight .gp { color: #555555 } /* Generic.Prompt */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #666666 } /* Generic.Subheading */
.highlight .gt { color: #aa0000 } /* Generic.Traceback */
.highlight .kc { color: #008800; font-weight: bold } /* Keyword.Constant */
.highlight .kd { color: #008800; font-weight: bold } /* Keyword.Declaration */
.highlight .kn { color: #008800; font-weight: bold } /* Keyword.Namespace */
.highlight .kp { color: #008800 } /* Keyword.Pseudo */
.highlight .kr { color: #008800; font-weight: bold } /* Keyword.Reserved */
.highlight .kt { color: #888888; font-weight: bold } /* Keyword.Type */
.highlight .m { color: #0000DD; font-weight: bold } /* Literal.Number */
.highlight .s { color: #dd2200; background-color: #fff0f0 } /* Literal.String */
.highlight .na { color: #336699 } /* Name.Attribute */
.highlight .nb { color: #003388 } /* Name.Builtin */
.highlight .nc { color: #bb0066; font-weight: bold } /* Name.Class */
.highlight .no { color: #003366; font-weight: bold } /* Name.Constant */
.highlight .nd { color: #555555 } /* Name.Decorator */
.highlight .ne { color: #bb0066; font-weight: bold } /* Name.Exception */
.highlight .nf { color: #0066bb; font-weight: bold } /* Name.Function */
.highlight .nl { color: #336699; font-style: italic } /* Name.Label */
.highlight .nn { color: #bb0066; font-weight: bold } /* Name.Namespace */
.highlight .py { color: #336699; font-weight: bold } /* Name.Property */
.highlight .nt { color: #bb0066; font-weight: bold } /* Name.Tag */
.highlight .nv { color: #336699 } /* Name.Variable */
.highlight .ow { color: #008800 } /* Operator.Word */
.highlight .w { color: #bbbbbb } /* Text.Whitespace */
.highlight .mb { color: #0000DD; font-weight: bold } /* Literal.Number.Bin */
.highlight .mf { color: #0000DD; font-weight: bold } /* Literal.Number.Float */
.highlight .mh { color: #0000DD; font-weight: bold } /* Literal.Number.Hex */
.highlight .mi { color: #0000DD; font-weight: bold } /* Literal.Number.Integer */
.highlight .mo { color: #0000DD; font-weight: bold } /* Literal.Number.Oct */
.highlight .sa { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Affix */
.highlight .sb { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Backtick */
.highlight .sc { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Char */
.highlight .dl { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Delimiter */
.highlight .sd { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Doc */
.highlight .s2 { color: #dd2200; background-color: #fff0f0 } /* Literal.Strifrom random import randint
from socket import AF_INET, AF_INET6
from scapy.all import *
from scapy.packet import *
from scapy.fields import *
from lisp import *
from framework import *
from vpp_object import *
class VppLispLocatorSet(VppObject):
""" Represents LISP locator set in VPP """
def __init__(self, test, ls_name):
self._test = test
self._ls_name = ls_name
@property
def test(self):
return self._test
@property
def ls_name(self):
return self._ls_name
def add_vpp_config(self):
self.test.vapi.lisp_locator_set(ls_name=self._ls_name)
self._test.registry.register(self, self.test.logger)
def get_lisp_locator_sets_dump_entry(self):
result = self.test.vapi.lisp_locator_set_dump()
for ls in result:
if ls.ls_name.strip('\x00') == self._ls_name:
return ls
return None
def query_vpp_config(self):
return self.get_lisp_locator_sets_dump_entry() is not None
def remove_vpp_config(self):
self.test.vapi.lisp_locator_set(ls_name=self._ls_name, is_add=0)
def object_id(self):
return 'lisp-locator-set-%s' % self._ls_name
class VppLispLocator(VppObject):
""" Represents LISP locator in VPP """
def __init__(self, test, sw_if_index, ls_name, priority=1, weight=1):
self._test = test
self._sw_if_index = sw_if_index
self._ls_name = ls_name
self._priority = priority
self._weight = weight
@property
def test(self):
""" Test which created this locator """
return self._test
@property
def ls_name(self):
""" Locator set name """
return self._ls_name
@property
def sw_if_index(self):
return self._sw_if_index
@property
def priority(self):
return self.priority
@property
def weight(self):
return self._weight
def add_vpp_config(self):
self.test.vapi.lisp_locator(ls_name=self._ls_name,
sw_if_index=self._sw_if_index,
priority=self._priority,
weight=self._weight)
self._test.registry.register(self, self.test.logger)
def get_lisp_locator_dump_entry(self):
locators = self.test.vapi.lisp_locator_dump(
is_index_set=0, ls_name=self._ls_name)
for locator in locators:
if locator.sw_if_index == self._sw_if_index:
return locator
return None
def query_vpp_config(self):
locator = self.get_lisp_locator_dump_entry()
return locator is not None
def remove_vpp_config(self):
self.test.vapi.lisp_locator(
ls_name=self._ls_name, sw_if_index=self._sw_if_index,
priority=self._priority, weight=self._weight, is_add=0)
self._test.registry.register(self, self.test.logger)
def object_id(self):
return 'lisp-locator-%s-%d' % (self._ls_name, self._sw_if_index)
class LispEIDType(object):
IP4 = 0
IP6 = 1
MAC = 2
class LispKeyIdType(object):
NONE = 0
SHA1 = 1
SHA256 = 2
class LispEID(object):
""" Lisp endpoint identifier """
def __init__(self, eid):
self.eid = eid
# find out whether EID is ip4 prefix, ip6 prefix or MAC
if self.eid.find("/") != -1:
if self.eid.find(":") == -1:
self.eid_type = LispEIDType.IP4
self.data_length = 4
else:
self.eid_type = LispEIDType.IP6
self.data_length = 16
self.eid_address = self.eid.split("/")[0]
self.prefix_length = int(self.eid.split("/")[1])
elif self.eid.count(":") == 5: # MAC address
self.eid_type = LispEIDType.MAC
self.eid_address = self.eid
self.prefix_length = 0
self.data_length = 6
else:
raise Exception('Unsupported EID format {}!'.format(eid))
def __str__(self):
if self.eid_type == LispEIDType.IP4:
return socket.inet_pton(socket.AF_INET, self.eid_address)
elif self.eid_type == LispEIDType.IP6:
return socket.inet_pton(socket.AF_INET6, self.eid_address)
elif self.eid_type == LispEIDType.MAC:
return Exception('Unimplemented')
raise Exception('Unknown EID type {}!'.format(self.eid_type))
class VppLispMapping(VppObject):
""" Represents common features for remote and local LISP mapping in VPP """
def __init__(self, test, eid, vni=0, priority=1, weight=1):
self._eid = LispEID(eid)
self._test = test
self._priority = priority
self._weight = weight
self._vni = vni
@property
def test(self):
return self._test
@property
def vni(self):
return self._vni
@property
def eid(self):
return self._eid
@property
def priority(self):
return self._priority
@property
def weight(self):
return self._weight
def get_lisp_mapping_dump_entry(self):
return self.test.vapi.lisp_eid_table_dump(
eid_set=1, prefix_length=self._eid.prefix_length,
vni=self._vni, eid_type=self._eid.eid_type, eid=str(self._eid))
def query_vpp_config(self):
mapping = self.get_lisp_mapping_dump_entry()
return mapping
class VppLocalMapping(VppLispMapping):
""" LISP Local mapping """
def __init__(self, test, eid, ls_name, vni=0, priority=1, weight=1,
key_id=LispKeyIdType.NONE, key=''):
super(VppLocalMapping, self).__init__(test, eid, vni, priority, weight)
self._ls_name = ls_name
self._key_id = key_id
self._key = key
@property
def ls_name(self):
return self._ls_name
@property
def key_id(self):
return self._key_id
@property
def key(self):
return self._key
def add_vpp_config(self):
self.test.vapi.lisp_local_mapping(
ls_name=self._ls_name, eid_type=self._eid.eid_type,
eid=str(self._eid), prefix_len=self._eid.prefix_length,
vni=self._vni, key_id=self._key_id, key=self._key)
self._test.registry.register(self, self.test.logger)
def remove_vpp_config(self):
self.test.vapi.lisp_local_mapping(
ls_name=self._ls_name, eid_type=self._eid.eid_type,
eid=str(self._eid), prefix_len=self._eid.prefix_length,
vni=self._vni, is_add=0)
def object_id(self):
return 'lisp-eid-local-mapping-%s[%d]' % (self._eid, self._vni)
class VppRemoteMapping(VppLispMapping):
def __init__(self, test, eid, rlocs=None, vni=0, priority=1, weight=1):
super(VppRemoteMapping, self).__init__(test, eid, vni, priority,
weight)
self._rlocs = rlocs
@property
def rlocs(self):
return self._rlocs
def add_vpp_config(self):
self.test.vapi.lisp_remote_mapping(
rlocs=self._rlocs, eid_type=self._eid.eid_type,
eid=str(self._eid), eid_prefix_len=self._eid.prefix_length,
vni=self._vni, rlocs_num=len(self._rlocs))
self._test.registry.register(self, self.test.logger)
def remove_vpp_config(self):
self.test.vapi.lisp_remote_mapping(
eid_type=self._eid.eid_type, eid=str(self._eid),
eid_prefix_len=self._eid.prefix_length, vni=self._vni,
is_add=0, rlocs_num=0)
def object_id(self):
return 'lisp-eid-remote-mapping-%s[%d]' % (self._eid, self._vni)
class VppLispAdjacency(VppObject):
""" Represents LISP adjacency in VPP """
def __init__(self, test, leid, reid, vni=0):
self._leid = LispEID(leid)
self._reid = LispEID(reid)
if self._leid.eid_type != self._reid.eid_type:
raise Exception('remote and local EID are different types!')
self._vni = vni
self._test = test
@property
def test(self):
return self._test
@property
def leid(self):
return self._leid
@property
def reid(self):
return self._reid
@property
def vni(self):
return self._vni
def add_vpp_config(self):
self.test.vapi.lisp_adjacency(
leid=str(self._leid),
reid=str(self._reid), eid_type=self._leid.eid_type,
leid_len=self._leid.prefix_length,
reid_len=self._reid.prefix_length, vni=self._vni)
self._test.registry.register(self, self.test.logger)
def eid_equal(self, eid, eid_type, eid_data, prefix_len):
if eid.eid_type != eid_type:
return False
if eid_type == LispEIDType.IP4 or eid_type == LispEIDType.IP6:
if eid.prefix_length != prefix_len:
return False
if str(eid) != eid_data[0:eid.data_length]:
return False
return True
def query_vpp_config(self):
res = self.test.vapi.lisp_adjacencies_get(vni=self._vni)
for adj in res.adjacencies:
if self.eid_equal(self._leid, adj.eid_type, adj.leid,
adj.leid_prefix_len) and \
self.eid_equal(self._reid, adj.eid_type, adj.reid,
adj.reid_prefix_len):
return True
return False
def remove_vpp_config(self):
self.test.vapi.lisp_adjacency(
leid=str(self._leid),
reid=str(self._reid), eid_type=self._leid.eid_type,
leid_len=self._leid.prefix_length,
reid_len=self._reid.prefix_length, vni=self._vni, is_add=0)
def object_id(self):
return 'lisp-adjacency-%s-%s[%d]' % (self._leid, self._reid, self._vni)
|