src/plugins/npt66/npt66.c

   1 // SPDX-License-Identifier: Apache-2.0
   2 // Copyright(c) 2023 Cisco Systems, Inc.
   3
   4 /*
   5  * npt66.c: NPT66 plugin
   6  * An implementation of Network Prefix Translation for IPv6-to-IPv6 (NPTv6) as
   7  * specified in RFC6296.
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdint.h>
  12 #include <inttypes.h>
  13 #include <vlib/vlib.h>
  14 #include <vnet/feature/feature.h>
  15 #include <vppinfra/pool.h>
  16 #include "npt66.h"
  17
  18 static int
  19 npt66_feature_enable_disable (u32 sw_if_index, bool is_add)
  20 {
  21   if (vnet_feature_enable_disable ("ip6-unicast", "npt66-input", sw_if_index,
  22                                    is_add, 0, 0) != 0)
  23     return -1;
  24   if (vnet_feature_enable_disable ("ip6-output", "npt66-output", sw_if_index,
  25                                    is_add, 0, 0) != 0)
  26     return -1;
  27   return 0;
  28 }
  29
  30 static void
  31 ipv6_prefix_zero (ip6_address_t *address, int prefix_len)
  32 {
  33   int byte_index = prefix_len / 8;
  34   int bit_offset = prefix_len % 8;
  35   uint8_t mask = (1 << (8 - bit_offset)) - 1;
  36   if (byte_index < 16)
  37     {
  38       address->as_u8[byte_index] &= mask;
  39       for (int i = byte_index + 1; i < 16; i++)
  40         {
  41           address->as_u8[i] = 0;
  42         }
  43     }
  44 }
  45
  46 int
  47 npt66_binding_add_del (u32 sw_if_index, ip6_address_t *internal,
  48                        int internal_plen, ip6_address_t *external,
  49                        int external_plen, bool is_add)
  50 {
  51   npt66_main_t *nm = &npt66_main;
  52
  53   if (is_add)
  54     {
  55
  56       /* Ensure prefix lengths are less than or equal to a /64 */
  57       if (internal_plen > 64 || external_plen > 64)
  58         return VNET_API_ERROR_INVALID_VALUE;
  59
  60       /* Create a binding entry */
  61       npt66_binding_t *b;
  62       pool_get_zero (nm->bindings, b);
  63       b->internal = *internal;
  64       b->internal_plen = internal_plen;
  65       b->external = *external;
  66       b->external_plen = external_plen;
  67       b->sw_if_index = sw_if_index;
  68
  69       ipv6_prefix_zero (&b->internal, internal_plen);
  70       ipv6_prefix_zero (&b->external, external_plen);
  71       vec_validate_init_empty (nm->interface_by_sw_if_index, sw_if_index, ~0);
  72       nm->interface_by_sw_if_index[sw_if_index] = b - nm->bindings;
  73
  74       uword delta = 0;
  75       delta = ip_csum_add_even (delta, b->external.as_u64[0]);
  76       delta = ip_csum_add_even (delta, b->external.as_u64[1]);
  77       delta = ip_csum_sub_even (delta, b->internal.as_u64[0]);
  78       delta = ip_csum_sub_even (delta, b->internal.as_u64[1]);
  79       delta = ip_csum_fold (delta);
  80       b->delta = delta;
  81     }
  82   else
  83     {
  84       /* Delete a binding entry */
  85       npt66_binding_t *b = npt66_interface_by_sw_if_index (sw_if_index);
  86       if (!b)
  87         return VNET_API_ERROR_NO_SUCH_ENTRY;
  88       nm->interface_by_sw_if_index[sw_if_index] = ~0;
  89       pool_put (nm->bindings, b);
  90     }
  91
  92   /* Enable feature on interface */
  93   int rv = npt66_feature_enable_disable (sw_if_index, is_add);
  94
  95   return rv;
  96 }
  97
  98 /*
  99  * Do a lookup in the interface vector (interface_by_sw_if_index)
 100  * and return pool entry.
 101  */
 102 npt66_binding_t *
 103 npt66_interface_by_sw_if_index (u32 sw_if_index)
 104 {
 105   npt66_main_t *nm = &npt66_main;
 106
 107   if (!nm->interface_by_sw_if_index ||
 108       sw_if_index > (vec_len (nm->interface_by_sw_if_index) - 1))
 109     return 0;
 110   u32 index = nm->interface_by_sw_if_index[sw_if_index];
 111   if (index == ~0)
 112     return 0;
 113   if (pool_is_free_index (nm->bindings, index))
 114     return 0;
 115   return pool_elt_at_index (nm->bindings, index);
 116 }