ipsec: T2816: IPSec python rework, includes DMVPN and VTI support

1 files changed, 149 insertions, 0 deletions

diff --git a/smoketest/scripts/cli/test_vpn_ipsec.py b/smoketest/scripts/cli/test_vpn_ipsec.py
new file mode 100644
index 000000000..4a3340ffb
--- /dev/null
+++ b/smoketest/scripts/cli/test_vpn_ipsec.py
@@ -0,0 +1,149 @@
+#!/usr/bin/env python3
+#
+# Copyright (C) 2021 VyOS maintainers and contributors
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 or later as
+# published by the Free Software Foundation.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import unittest
+
+from base_vyostest_shim import VyOSUnitTestSHIM
+
+from vyos.util import call, process_named_running, read_file
+
+tunnel_path = ['interfaces', 'tunnel']
+nhrp_path = ['protocols', 'nhrp']
+base_path = ['vpn', 'ipsec']
+
+class TestVPNIPsec(VyOSUnitTestSHIM.TestCase):
+    def tearDown(self):
+        self.cli_delete(base_path)
+        self.cli_delete(nhrp_path)
+        self.cli_delete(tunnel_path)
+        self.cli_commit()
+
+    def test_site_to_site(self):
+        self.cli_delete(base_path)
+
+        # IKE/ESP Groups
+        self.cli_set(base_path + ["esp-group", "MyESPGroup", "proposal", "1", "encryption", "aes128"])
+        self.cli_set(base_path + ["esp-group", "MyESPGroup", "proposal", "1", "hash", "sha1"])
+        self.cli_set(base_path + ["ike-group", "MyIKEGroup", "proposal", "1", "dh-group", "2"])
+        self.cli_set(base_path + ["ike-group", "MyIKEGroup", "proposal", "1", "encryption", "aes128"])
+        self.cli_set(base_path + ["ike-group", "MyIKEGroup", "proposal", "1", "hash", "sha1"])
+
+        # Site to site
+        self.cli_set(base_path + ["ipsec-interfaces", "interface", "eth0"])
+        self.cli_set(base_path + ["site-to-site", "peer", "203.0.113.45", "authentication", "mode", "pre-shared-secret"])
+        self.cli_set(base_path + ["site-to-site", "peer", "203.0.113.45", "authentication", "pre-shared-secret", "MYSECRETKEY"])
+        self.cli_set(base_path + ["site-to-site", "peer", "203.0.113.45", "ike-group", "MyIKEGroup"])
+        self.cli_set(base_path + ["site-to-site", "peer", "203.0.113.45", "default-esp-group", "MyESPGroup"])
+        self.cli_set(base_path + ["site-to-site", "peer", "203.0.113.45", "local-address", "192.0.2.10"])
+        self.cli_set(base_path + ["site-to-site", "peer", "203.0.113.45", "tunnel", "1", "protocol", "gre"])
+
+        self.cli_commit()
+
+        ipsec_conf_lines = [
+            'authby = secret',
+            'ike = aes128-sha1-modp1024!',
+            'esp = aes128-sha1-modp1024!',
+            'left = 192.0.2.10',
+            'right = 203.0.113.45',
+            'type = tunnel'
+        ]
+
+        ipsec_secrets_lines = [
+            '192.0.2.10 203.0.113.45 : PSK "MYSECRETKEY" # dhcp:no'
+        ]
+
+        tmp_ipsec_conf = read_file('/etc/ipsec.conf')
+
+        for line in ipsec_conf_lines:
+            self.assertIn(line, tmp_ipsec_conf)
+
+        call('sudo chmod 644 /etc/ipsec.secrets') # Needs to be readable
+        tmp_ipsec_secrets = read_file('/etc/ipsec.secrets')
+
+        for line in ipsec_secrets_lines:
+            self.assertIn(line, tmp_ipsec_secrets)
+
+        # Check for running process
+        self.assertTrue(process_named_running('charon'))
+
+    def test_dmvpn(self):
+        self.cli_delete(base_path)
+        self.cli_delete(nhrp_path)
+        self.cli_delete(tunnel_path)
+
+        # Tunnel
+        self.cli_set(tunnel_path + ["tun100", "address", "172.16.253.134/29"])
+        self.cli_set(tunnel_path + ["tun100", "encapsulation", "gre"])
+        self.cli_set(tunnel_path + ["tun100", "source-address", "192.0.2.1"])
+        self.cli_set(tunnel_path + ["tun100", "multicast", "enable"])
+        self.cli_set(tunnel_path + ["tun100", "parameters", "ip", "key", "1"])
+
+        # NHRP
+        self.cli_set(nhrp_path + ["tunnel", "tun100", "cisco-authentication", "secret"])
+        self.cli_set(nhrp_path + ["tunnel", "tun100", "holding-time", "300"])
+        self.cli_set(nhrp_path + ["tunnel", "tun100", "multicast", "dynamic"])
+        self.cli_set(nhrp_path + ["tunnel", "tun100", "redirect"])
+        self.cli_set(nhrp_path + ["tunnel", "tun100", "shortcut"])
+
+        # IKE/ESP Groups
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "compression", "disable"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "lifetime", "1800"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "mode", "transport"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "pfs", "dh-group2"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "proposal", "1", "encryption", "aes256"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "proposal", "1", "hash", "sha1"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "proposal", "2", "encryption", "3des"])
+        self.cli_set(base_path + ["esp-group", "ESP-HUB", "proposal", "2", "hash", "md5"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "ikev2-reauth", "no"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "key-exchange", "ikev1"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "lifetime", "3600"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "proposal", "1", "dh-group", "2"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "proposal", "1", "encryption", "aes256"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "proposal", "1", "hash", "sha1"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "proposal", "2", "dh-group", "2"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "proposal", "2", "encryption", "aes128"])
+        self.cli_set(base_path + ["ike-group", "IKE-HUB", "proposal", "2", "hash", "sha1"])
+
+        # Profile
+        self.cli_set(base_path + ["ipsec-interfaces", "interface", "eth0"])
+        self.cli_set(base_path + ["profile", "NHRPVPN", "authentication", "mode", "pre-shared-secret"])
+        self.cli_set(base_path + ["profile", "NHRPVPN", "authentication", "pre-shared-secret", "secret"])
+        self.cli_set(base_path + ["profile", "NHRPVPN", "bind", "tunnel", "tun100"])
+        self.cli_set(base_path + ["profile", "NHRPVPN", "esp-group", "ESP-HUB"])
+        self.cli_set(base_path + ["profile", "NHRPVPN", "ike-group", "IKE-HUB"])
+
+        self.cli_commit()
+
+        swanctl_lines = [
+            'proposals = aes256-sha1-modp1024,aes128-sha1-modp1024',
+            'version = 1',
+            'rekey_time = 3600s',
+            'esp_proposals = aes256-sha1-modp1024,3des-md5-modp1024',
+            'local_ts = dynamic[gre]',
+            'remote_ts = dynamic[gre]',
+            'mode = transport',
+            'secret = secret'
+        ]
+
+        tmp_swanctl_conf = read_file('/etc/swanctl/swanctl.conf')
+
+        for line in swanctl_lines:
+            self.assertIn(line, tmp_swanctl_conf)
+
+        self.assertTrue(process_named_running('charon'))
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)