aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries/python')
-rw-r--r--resources/libraries/python/LispSetup.py403
1 files changed, 268 insertions, 135 deletions
diff --git a/resources/libraries/python/LispSetup.py b/resources/libraries/python/LispSetup.py
index 5ef8bc06a9..5fee5db87d 100644
--- a/resources/libraries/python/LispSetup.py
+++ b/resources/libraries/python/LispSetup.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2016 Cisco and/or its affiliates.
+# Copyright (c) 2016-2019 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
@@ -13,9 +13,11 @@
"""Library to set up Lisp in topology."""
-from resources.libraries.python.topology import NodeType
-from resources.libraries.python.VatExecutor import VatExecutor
+from ipaddress import ip_address
+from resources.libraries.python.topology import NodeType
+from resources.libraries.python.PapiExecutor import PapiSocketExecutor
+from resources.libraries.python.L2Util import L2Util
class LispStatus(object):
"""Class for lisp API."""
@@ -33,10 +35,14 @@ class LispStatus(object):
:type state: str
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/lisp_status.vat',
- state=state)
+ args = dict(is_en=0 if state == 'disable' else 1)
+
+ cmd = 'lisp_enable_disable'
+ err_msg = "Failed to set LISP status on host {host}".format(
+ host=node['host'])
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispRemoteMapping(object):
"""Class for lisp remote mapping API."""
@@ -66,20 +72,40 @@ class LispRemoteMapping(object):
:type rloc: str
:type is_mac: bool
"""
- if is_mac:
- deid_prefix = ''
- seid_prefix = ''
+
+ if not is_mac:
+ eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1
+ eid_packed = ip_address(unicode(deid)).packed
+ seid_packed = ip_address(unicode(seid)).packed
+ eid_len = deid_prefix
+ seid_len = seid_prefix
else:
- deid_prefix = '/{}'.format(deid_prefix)
- seid_prefix = '/{}'.format(seid_prefix)
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_remote_mapping.vat',
- vni=vni,
- deid=deid,
- deid_prefix=deid_prefix,
- seid=seid,
- seid_prefix=seid_prefix,
- rloc=rloc)
+ eid_type = 2
+ eid_packed = L2Util.mac_to_bin(deid)
+ seid_packed = L2Util.mac_to_bin(seid)
+ eid_len = 0
+ seid_len = 0
+
+ rlocs = [dict(is_ip4=1 if ip_address(unicode(rloc)).version == 4 else 0,
+ addr=ip_address(unicode(rloc)).packed)]
+
+ args = dict(is_add=1,
+ is_src_dst=1,
+ vni=int(vni),
+ eid_type=eid_type,
+ eid=eid_packed,
+ eid_len=eid_len,
+ seid=seid_packed,
+ seid_len=seid_len,
+ rloc_num=1,
+ rlocs=rlocs)
+
+ cmd = 'lisp_add_del_remote_mapping'
+ err_msg = "Failed to add remote mapping on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
@staticmethod
def vpp_del_lisp_remote_mapping(node, vni, deid, deid_prefix, seid,
@@ -102,15 +128,42 @@ class LispRemoteMapping(object):
:type rloc: str
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_remote_mapping.vat',
- vni=vni,
- deid=deid,
- deid_prefix=deid_prefix,
- seid=seid,
- seid_prefix=seid_prefix,
- rloc=rloc)
+ # used only with IPs
+ is_mac = False
+ if not is_mac:
+ eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1
+ eid_packed = ip_address(unicode(deid)).packed
+ seid_packed = ip_address(unicode(seid)).packed
+ eid_len = deid_prefix
+ seid_len = seid_prefix
+ else:
+ eid_type = 2
+ eid_packed = L2Util.mac_to_bin(deid)
+ seid_packed = L2Util.mac_to_bin(seid)
+ eid_len = 0
+ seid_len = 0
+
+ rlocs = [dict(is_ip4=1 if ip_address(unicode(rloc)).version == 4 else 0,
+ addr=ip_address(unicode(rloc)).packed)]
+
+ args = dict(is_add=0,
+ is_src_dst=1,
+ vni=int(vni),
+ eid_type=eid_type,
+ eid=eid_packed,
+ eid_len=eid_len,
+ seid=seid_packed,
+ seid_len=seid_len,
+ rloc_num=1,
+ rlocs=rlocs)
+
+ cmd = 'lisp_add_del_remote_mapping'
+ err_msg = "Failed to delete remote mapping on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispAdjacency(object):
"""Class for lisp adjacency API."""
@@ -138,19 +191,34 @@ class LispAdjacency(object):
:type seid_prefix: int
:type is_mac: bool
"""
- if is_mac:
- deid_prefix = ''
- seid_prefix = ''
+
+ if not is_mac:
+ eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1
+ reid = ip_address(unicode(deid)).packed
+ leid = ip_address(unicode(seid)).packed
+ reid_len = deid_prefix
+ leid_len = seid_prefix
else:
- deid_prefix = '/{}'.format(deid_prefix)
- seid_prefix = '/{}'.format(seid_prefix)
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_adjacency.vat',
- vni=vni,
- deid=deid,
- deid_prefix=deid_prefix,
- seid=seid,
- seid_prefix=seid_prefix)
+ eid_type = 2
+ reid = L2Util.mac_to_bin(deid)
+ leid = L2Util.mac_to_bin(seid)
+ reid_len = 0
+ leid_len = 0
+
+ args = dict(is_add=1,
+ vni=int(vni),
+ eid_type=eid_type,
+ reid=reid,
+ reid_len=reid_len,
+ leid=leid,
+ leid_len=leid_len)
+
+ cmd = 'lisp_add_del_adjacency'
+ err_msg = "Failed to add lisp adjacency on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
@staticmethod
def vpp_del_lisp_adjacency(node, vni, deid, deid_prefix, seid,
@@ -171,14 +239,36 @@ class LispAdjacency(object):
:type seid_prefix: int
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_adjacency.vat',
- vni=vni,
- deid=deid,
- deid_prefix=deid_prefix,
- seid=seid,
- seid_prefix=seid_prefix)
+ # used only with IPs
+ is_mac = False
+ if not is_mac:
+ eid_type = 0 if ip_address(unicode(deid)).version == 4 else 1
+ reid = ip_address(unicode(deid)).packed
+ leid = ip_address(unicode(seid)).packed
+ reid_len = deid_prefix
+ leid_len = seid_prefix
+ else:
+ eid_type = 2
+ reid = L2Util.mac_to_bin(deid)
+ leid = L2Util.mac_to_bin(seid)
+ reid_len = 0
+ leid_len = 0
+
+ args = dict(is_add=0,
+ vni=int(vni),
+ eid_type=eid_type,
+ reid=reid,
+ reid_len=reid_len,
+ leid=leid,
+ leid_len=leid_len)
+
+ cmd = 'lisp_add_del_adjacency'
+ err_msg = "Failed to delete lisp adjacency on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispGpeStatus(object):
"""Clas for LISP GPE status manipulation."""
@@ -196,29 +286,14 @@ class LispGpeStatus(object):
:type state: str
"""
- VatExecutor.cmd_from_template(node, 'lisp/lisp_gpe_status.vat',
- state=state)
-
-
-class LispGpeIface(object):
- """Class for Lisp gpe interface API."""
-
- def __init__(self):
- pass
-
- @staticmethod
- def vpp_lisp_gpe_iface(node, state):
- """Set lisp gpe interface up or down on the VPP node in topology.
+ args = dict(is_en=0 if state == 'disable' else 1)
- :param node: VPP node.
- :param state: State of the gpe iface, up or down.
- :type node: dict
- :type state: str
- """
-
- VatExecutor.cmd_from_template(node, 'lisp/lisp_gpe_iface.vat',
- state=state)
+ cmd = 'gpe_enable_disable'
+ err_msg = "Failed to set LISP GPE status on host {host}".format(
+ host=node['host'])
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispGpeForwardEntry(object):
"""The functionality needed for these methods is not implemented in VPP
@@ -259,9 +334,17 @@ class LispMapResolver(object):
:type map_resolver_ip: str
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_map_resolver.vat',
- address=map_resolver_ip)
+ args = dict(is_add=1,
+ is_ipv6=0 if ip_address(unicode(map_resolver_ip)).version \
+ == 4 else 1,
+ ip_address=ip_address(unicode(map_resolver_ip)).packed)
+
+ cmd = 'lisp_add_del_map_resolver'
+ err_msg = "Failed to add map resolver on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
@staticmethod
def vpp_del_map_resolver(node, map_resolver_ip):
@@ -273,10 +356,17 @@ class LispMapResolver(object):
:type map_resolver_ip: str
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_map_resolver.vat',
- address=map_resolver_ip)
+ args = dict(is_add=0,
+ is_ipv6=0 if ip_address(unicode(map_resolver_ip)).version \
+ == 4 else 1,
+ ip_address=ip_address(unicode(map_resolver_ip)).packed)
+
+ cmd = 'lisp_add_del_map_resolver'
+ err_msg = "Failed to delete map resolver on host {host}".format(
+ host=node['host'])
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispLocalEid(object):
"""Class for Lisp local eid API."""
@@ -301,19 +391,26 @@ class LispLocalEid(object):
:type prefix_len: int
"""
- if prefix_len is not None:
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_local_eid.vat',
- vni=vni,
- eid=eid,
- eid_prefix=prefix_len,
- locator_name=locator_set_name)
+ if prefix_len:
+ eid_type = 0 if ip_address(unicode(eid)).version == 4 else 1
+ eid_packed = ip_address(unicode(eid)).packed
else:
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_local_eid_mac.vat',
- vni=vni,
- eid=eid,
- locator_name=locator_set_name)
+ eid_type = 2
+ eid_packed = L2Util.mac_to_bin(eid)
+
+ args = dict(is_add=1,
+ eid_type=eid_type,
+ eid=eid_packed,
+ prefix_len=prefix_len,
+ locator_set_name=locator_set_name,
+ vni=int(vni))
+
+ cmd = 'lisp_add_del_local_eid'
+ err_msg = "Failed to add local eid on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
@staticmethod
def vpp_del_lisp_local_eid(node, locator_set_name, vni, eid,
@@ -332,20 +429,26 @@ class LispLocalEid(object):
:type prefix_len: int
"""
- if prefix_len is not None:
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_local_eid.vat',
- vni=vni,
- eid=eid,
- eid_prefix=prefix_len,
- locator_name=locator_set_name)
+ if prefix_len:
+ eid_type = 0 if ip_address(unicode(eid)).version == 4 else 1
+ eid_packed = ip_address(unicode(eid)).packed
else:
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_local_eid_mac.vat',
- vni=vni,
- eid=eid,
- locator_name=locator_set_name)
+ eid_type = 2
+ eid_packed = L2Util.mac_to_bin(eid)
+ args = dict(is_add=0,
+ eid_type=eid_type,
+ eid=eid_packed,
+ prefix_len=prefix_len,
+ locator_set_name=locator_set_name,
+ vni=int(vni))
+
+ cmd = 'lisp_add_del_local_eid'
+ err_msg = "Failed to delete local eid on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispLocator(object):
"""Class for the Lisp Locator API."""
@@ -369,12 +472,18 @@ class LispLocator(object):
:type weight: int
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_locator.vat',
- lisp_name=locator_name,
- sw_if_index=sw_if_index,
- priority=priority,
- weight=weight)
+ args = dict(is_add=1,
+ locator_set_name=locator_name,
+ sw_if_index=sw_if_index,
+ priority=priority,
+ weight=weight)
+
+ cmd = 'lisp_add_del_locator'
+ err_msg = "Failed to add locator on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
@staticmethod
def vpp_del_lisp_locator(node, locator_name, sw_if_index, priority, weight):
@@ -392,13 +501,18 @@ class LispLocator(object):
:type weight: int
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_locator.vat',
- lisp_name=locator_name,
- sw_if_index=sw_if_index,
- priority=priority,
- weight=weight)
+ args = dict(is_add=0,
+ locator_set_name=locator_name,
+ sw_if_index=sw_if_index,
+ priority=priority,
+ weight=weight)
+
+ cmd = 'lisp_add_del_locator'
+ err_msg = "Failed to delete locator on host {host}".format(
+ host=node['host'])
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispLocatorSet(object):
"""Class for Lisp Locator Set API."""
@@ -416,9 +530,17 @@ class LispLocatorSet(object):
:type name: str
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/add_lisp_locator_set.vat',
- lisp_name=name)
+ args = dict(is_add=1,
+ locator_set_name=name,
+ locator_num=0,
+ locators=[])
+
+ cmd = 'lisp_add_del_locator_set'
+ err_msg = "Failed to add locator set on host {host}".format(
+ host=node['host'])
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
@staticmethod
def vpp_del_lisp_locator_set(node, name):
@@ -430,10 +552,17 @@ class LispLocatorSet(object):
:type name: str
"""
- VatExecutor.cmd_from_template(node,
- 'lisp/del_lisp_locator_set.vat',
- lisp_name=name)
+ args = dict(is_add=0,
+ locator_set_name=name,
+ locator_num=0,
+ locators=[])
+
+ cmd = 'lisp_add_del_locator_set'
+ err_msg = "Failed to delete locator set on host {host}".format(
+ host=node['host'])
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
class LispSetup(object):
"""Lisp setup in topology."""
@@ -590,20 +719,6 @@ class LispSetup(object):
for map_ip in map_resolver:
lisp_map_res.vpp_del_map_resolver(node, map_ip.get('map resolver'))
- @staticmethod
- def vpp_lisp_gpe_interface_status(node, state):
- """Set lisp gpe interface status on VPP node in topology.
-
- :param node: VPP node.
- :param state: State of the gpe iface, up or down
- :type node: dict
- :type state: str
- """
-
- lgi = LispGpeIface()
- lgi.vpp_lisp_gpe_iface(node, state)
-
-
class LispEidTableMap(object):
"""
Class for EID table map.
@@ -623,11 +738,29 @@ class LispEidTableMap(object):
:type bd_id: int
:type vrf: int
"""
+
+ # adding default mapping vni=0, vrf=0 needs to be skipped
+ skip = False
+
if bd_id:
- bd_or_vrf = 'bd_index {}'.format(bd_id)
+ is_l2 = 1
+ dp_table = bd_id
else:
- bd_or_vrf = 'vrf {}'.format(vrf)
- VatExecutor.cmd_from_template(node,
- 'lisp/lisp_eid_table_add_del_map.vat',
- vni=vni,
- bd_or_vrf=bd_or_vrf)
+ is_l2 = 0
+ dp_table = vrf
+ # skip adding default mapping
+ if (int(vrf) == 0) and (int(vni) == 0):
+ skip = True
+
+ args = dict(is_add=1,
+ vni=int(vni),
+ dp_table=int(dp_table),
+ is_l2=is_l2)
+
+ cmd = 'lisp_eid_table_add_del_map'
+ err_msg = "Failed to add eid table map on host {host}".format(
+ host=node['host'])
+
+ if not skip:
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)