test/vpp_ipsec_tun_interface.py

   1 from vpp_tunnel_interface import VppTunnelInterface
   2
   3
   4 class VppIpsecTunInterface(VppTunnelInterface):
   5     """
   6     VPP IPsec Tunnel interface
   7     """
   8
   9     def __init__(self, test, parent_if, local_spi,
  10                  remote_spi, crypto_alg, local_crypto_key, remote_crypto_key,
  11                  integ_alg, local_integ_key, remote_integ_key, salt=0,
  12                  udp_encap=False,
  13                  is_ip6=False,
  14                  dst=None):
  15         super(VppIpsecTunInterface, self).__init__(test, parent_if)
  16         self.local_spi = local_spi
  17         self.remote_spi = remote_spi
  18         self.crypto_alg = crypto_alg
  19         self.local_crypto_key = local_crypto_key
  20         self.remote_crypto_key = remote_crypto_key
  21         self.integ_alg = integ_alg
  22         self.local_integ_key = local_integ_key
  23         self.remote_integ_key = remote_integ_key
  24         self.salt = salt
  25         if is_ip6:
  26             self.local_ip = self.parent_if.local_ip6
  27             self.remote_ip = self.parent_if.remote_ip6
  28         else:
  29             self.local_ip = self.parent_if.local_ip4
  30             self.remote_ip = self.parent_if.remote_ip4
  31         if dst:
  32             self.remote_ip = dst
  33         self.udp_encap = udp_encap
  34
  35     def add_vpp_config(self):
  36         r = self.test.vapi.ipsec_tunnel_if_add_del(
  37             self.local_ip, self.remote_ip,
  38             self.remote_spi, self.local_spi,
  39             self.crypto_alg, self.local_crypto_key, self.remote_crypto_key,
  40             self.integ_alg, self.local_integ_key, self.remote_integ_key,
  41             salt=self.salt,
  42             udp_encap=self.udp_encap)
  43         self.set_sw_if_index(r.sw_if_index)
  44         self.generate_remote_hosts()
  45         self.test.registry.register(self, self.test.logger)
  46
  47     def remove_vpp_config(self):
  48         self.test.vapi.ipsec_tunnel_if_add_del(
  49             self.local_ip, self.remote_ip,
  50             self.remote_spi, self.local_spi,
  51             self.crypto_alg, self.local_crypto_key, self.remote_crypto_key,
  52             self.integ_alg, self.local_integ_key, self.remote_integ_key,
  53             is_add=0)
  54
  55     def object_id(self):
  56         return "ipsec-tun-if-%d" % self._sw_if_index