Merge pull request #620 from jack9603301/T3030

tunnel: T3030: Add erspan protocol support

1 files changed, 190 insertions, 0 deletions

diff --git a/python/vyos/ifconfig/erspan.py b/python/vyos/ifconfig/erspan.py
new file mode 100755
index 000000000..50230e14a
--- /dev/null
+++ b/python/vyos/ifconfig/erspan.py
@@ -0,0 +1,190 @@
+# Copyright 2019-2020 VyOS maintainers and contributors <maintainers@vyos.io>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library.  If not, see <http://www.gnu.org/licenses/>.
+
+# https://developers.redhat.com/blog/2019/05/17/an-introduction-to-linux-virtual-interfaces-tunnels/#erspan
+# http://vger.kernel.org/lpc_net2018_talks/erspan-linux-presentation.pdf
+
+from copy import deepcopy
+
+from netaddr import EUI
+from netaddr import mac_unix_expanded
+from random import getrandbits
+
+from vyos.util import dict_search
+from vyos.ifconfig.interface import Interface
+from vyos.validate import assert_list
+
+@Interface.register
+class _ERSpan(Interface):
+    """
+    _ERSpan: private base class for ERSPAN tunnels
+    """
+    default = {
+        **Interface.default,
+        **{
+            'type': 'erspan',
+        }
+    }
+    definition = {
+        **Interface.definition,
+        **{
+            'section': 'erspan',
+            'prefixes': ['ersp',],
+        },
+    }
+    
+    options = ['local_ip','remote_ip','encapsulation','parameters']
+    
+    def __init__(self,ifname,**config):
+        self.config = deepcopy(config) if config else {}
+        super().__init__(ifname, **self.config)
+    
+    def change_options(self):
+        pass
+    
+    def update(self, config):
+        
+        # Enable/Disable of an interface must always be done at the end of the
+        # derived class to make use of the ref-counting set_admin_state()
+        # function. We will only enable the interface if 'up' was called as
+        # often as 'down'. This is required by some interface implementations
+        # as certain parameters can only be changed when the interface is
+        # in admin-down state. This ensures the link does not flap during
+        # reconfiguration.
+        super().update(config)
+        state = 'down' if 'disable' in config else 'up'
+        self.set_admin_state(state)
+    
+    def _create(self):
+        pass
+
+class ERSpanIf(_ERSpan):
+    """
+    ERSpanIf: private base class for ERSPAN Over GRE and IPv4 tunnels
+    """
+    
+    def _create(self):
+        ifname = self.config['ifname']
+        local_ip = self.config['local_ip']
+        remote_ip = self.config['remote_ip']
+        key = self.config['parameters']['ip']['key']
+        version = self.config['parameters']['version']
+        command = f'ip link add dev {ifname} type erspan local {local_ip} remote {remote_ip} seq key {key} erspan_ver {version}'
+        
+        if int(version) == 1:
+            idx=dict_search('parameters.erspan.idx',self.config)
+            if idx:
+                command += f' erspan {idx}'
+        elif int(version) == 2:
+            direction=dict_search('parameters.erspan.direction',self.config)
+            if direction:
+                command += f' erspan_dir {direction}'
+            hwid=dict_search('parameters.erspan.hwid',self.config)
+            if hwid:
+                command += f' erspan_hwid {hwid}'
+        
+        ttl = dict_search('parameters.ip.ttl',self.config)
+        if ttl:
+            command += f' ttl {ttl}'
+        tos = dict_search('parameters.ip.tos',self.config)
+        if tos:
+            command += f' tos {tos}'
+                
+        self._cmd(command)
+    
+    def change_options(self):
+        ifname = self.config['ifname']
+        local_ip = self.config['local_ip']
+        remote_ip = self.config['remote_ip']
+        key = self.config['parameters']['ip']['key']
+        version = self.config['parameters']['version']
+        command = f'ip link set dev {ifname} type erspan local {local_ip} remote {remote_ip} seq key {key} erspan_ver {version}'
+        
+        if int(version) == 1:
+            idx=dict_search('parameters.erspan.idx',self.config)
+            if idx:
+                command += f' erspan {idx}'
+        elif int(version) == 2:
+            direction=dict_search('parameters.erspan.direction',self.config)
+            if direction:
+                command += f' erspan_dir {direction}'
+            hwid=dict_search('parameters.erspan.hwid',self.config)
+            if hwid:
+                command += f' erspan_hwid {hwid}'
+        
+        ttl = dict_search('parameters.ip.ttl',self.config)
+        if ttl:
+            command += f' ttl {ttl}'
+        tos = dict_search('parameters.ip.tos',self.config)
+        if tos:
+            command += f' tos {tos}'
+                
+        self._cmd(command)
+
+class ER6SpanIf(_ERSpan):
+    """
+    ER6SpanIf: private base class for ERSPAN Over GRE and IPv6 tunnels
+    """
+    
+    def _create(self):
+        ifname = self.config['ifname']
+        local_ip = self.config['local_ip']
+        remote_ip = self.config['remote_ip']
+        key = self.config['parameters']['ip']['key']
+        version = self.config['parameters']['version']
+        command = f'ip link add dev {ifname} type ip6erspan local {local_ip} remote {remote_ip} seq key {key} erspan_ver {version}'
+        
+        if int(version) == 1:
+            idx=dict_search('parameters.erspan.idx',self.config)
+            if idx:
+                command += f' erspan {idx}'
+        elif int(version) == 2:
+            direction=dict_search('parameters.erspan.direction',self.config)
+            if direction:
+                command += f' erspan_dir {direction}'
+            hwid=dict_search('parameters.erspan.hwid',self.config)
+            if hwid:
+                command += f' erspan_hwid {hwid}'
+        
+        ttl = dict_search('parameters.ip.ttl',self.config)
+        if ttl:
+            command += f' ttl {ttl}'
+        tos = dict_search('parameters.ip.tos',self.config)
+        if tos:
+            command += f' tos {tos}'
+                
+        self._cmd(command)
+    
+    def change_options(self):
+        ifname = self.config['ifname']
+        local_ip = self.config['local_ip']
+        remote_ip = self.config['remote_ip']
+        key = self.config['parameters']['ip']['key']
+        version = self.config['parameters']['version']
+        command = f'ip link set dev {ifname} type ip6erspan local {local_ip} remote {remote_ip} seq key {key} erspan_ver {version}'
+        
+        if int(version) == 1:
+            idx=dict_search('parameters.erspan.idx',self.config)
+            if idx:
+                command += f' erspan {idx}'
+        elif int(version) == 2:
+            direction=dict_search('parameters.erspan.direction',self.config)
+            if direction:
+                command += f' erspan_dir {direction}'
+            hwid=dict_search('parameters.erspan.hwid',self.config)
+            if hwid:
+                command += f' erspan_hwid {hwid}'
+                
+        self._cmd(command)