summaryrefslogtreecommitdiffstats
path: root/test/test_ipsec_tun_if_esp.py
blob: c260f03b63da71bdd9cb3d0ab736068a752686c7 (plain)
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
import unittest
import socket
from scapy.layers.ipsec import ESP
from framework import VppTestRunner
from template_ipsec import TemplateIpsec, IpsecTunTests, IpsecTcpTests
from vpp_ipsec_tun_interface import VppIpsecTunInterface


class TemplateIpsecTunIfEsp(TemplateIpsec):
    """ IPsec tunnel interface tests """

    encryption_type = ESP

    @classmethod
    def setUpClass(cls):
        super(TemplateIpsecTunIfEsp, cls).setUpClass()
        cls.tun_if = cls.pg0

    def setUp(self):
        self.ipsec_tun_if = VppIpsecTunInterface(self, self.pg0,
                                                 self.vpp_tun_spi,
                                                 self.scapy_tun_spi,
                                                 self.crypt_algo_vpp_id,
                                                 self.crypt_key,
                                                 self.crypt_key,
                                                 self.auth_algo_vpp_id,
                                                 self.auth_key,
                                                 self.auth_key)
        self.ipsec_tun_if.add_vpp_config()
        self.ipsec_tun_if.admin_up()
        self.ipsec_tun_if.config_ip4()
        src4 = socket.inet_pton(socket.AF_INET, self.remote_tun_if_host)
        self.vapi.ip_add_del_route(src4, 32, self.ipsec_tun_if.remote_ip4n)

    def tearDown(self):
        if not self.vpp_dead:
            self.vapi.cli("show hardware")
        super(TemplateIpsecTunIfEsp, self).tearDown()


class TestIpsecTunIfEsp1(TemplateIpsecTunIfEsp, IpsecTunTests):
    """ Ipsec ESP - TUN tests """
    pass


class TestIpsecTunIfEsp2(TemplateIpsecTunIfEsp, IpsecTcpTests):
    """ Ipsec ESP - TCP tests """
    pass


if __name__ == '__main__':
    unittest.main(testRunner=VppTestRunner)