53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
|
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)
|