src/plugins/wireguard/wireguard.h

   1 /*
   2  * Copyright (c) 2020 Doc.ai and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15 #ifndef __included_wg_h__
  16 #define __included_wg_h__
  17
  18 #include <wireguard/wireguard_index_table.h>
  19 #include <wireguard/wireguard_messages.h>
  20 #include <wireguard/wireguard_timer.h>
  21 #include <vnet/buffer.h>
  22
  23 #define WG_DEFAULT_DATA_SIZE 2048
  24
  25 extern vlib_node_registration_t wg4_input_node;
  26 extern vlib_node_registration_t wg6_input_node;
  27 extern vlib_node_registration_t wg4_output_tun_node;
  28 extern vlib_node_registration_t wg6_output_tun_node;
  29
  30 typedef struct wg_per_thread_data_t_
  31 {
  32   CLIB_CACHE_LINE_ALIGN_MARK (cacheline0);
  33   vnet_crypto_op_t *crypto_ops;
  34   vnet_crypto_async_frame_t **async_frames;
  35   u8 data[WG_DEFAULT_DATA_SIZE];
  36 } wg_per_thread_data_t;
  37 typedef struct
  38 {
  39   /* convenience */
  40   vlib_main_t *vlib_main;
  41
  42   u16 msg_id_base;
  43
  44   wg_index_table_t index_table;
  45
  46   u32 in4_fq_index;
  47   u32 in6_fq_index;
  48   u32 out4_fq_index;
  49   u32 out6_fq_index;
  50
  51   wg_per_thread_data_t *per_thread_data;
  52   u8 feature_init;
  53
  54   tw_timer_wheel_16t_2w_512sl_t timer_wheel;
  55
  56   /* operation mode flags (e.g. async) */
  57   u8 op_mode_flags;
  58 } wg_main_t;
  59
  60 typedef struct
  61 {
  62   /* wg post node index for async crypto */
  63   u32 wg4_post_next;
  64   u32 wg6_post_next;
  65 } wg_async_post_next_t;
  66
  67 extern wg_async_post_next_t wg_encrypt_async_next;
  68 extern wg_async_post_next_t wg_decrypt_async_next;
  69 extern wg_main_t wg_main;
  70
  71 /**
  72  * Wireguard operation mode
  73  **/
  74 #define foreach_wg_op_mode_flags _ (0, ASYNC, "async")
  75
  76 /**
  77  * Helper function to set/unset and check op modes
  78  **/
  79 typedef enum wg_op_mode_flags_t_
  80 {
  81 #define _(v, f, s) WG_OP_MODE_FLAG_##f = 1 << v,
  82   foreach_wg_op_mode_flags
  83 #undef _
  84 } __clib_packed wg_op_mode_flags_t;
  85
  86 #define _(a, v, s)                                                            \
  87   always_inline int wg_op_mode_set_##v (void)                                 \
  88   {                                                                           \
  89     return (wg_main.op_mode_flags |= WG_OP_MODE_FLAG_##v);                    \
  90   }                                                                           \
  91   always_inline int wg_op_mode_unset_##v (void)                               \
  92   {                                                                           \
  93     return (wg_main.op_mode_flags &= ~WG_OP_MODE_FLAG_##v);                   \
  94   }                                                                           \
  95   always_inline int wg_op_mode_is_set_##v (void)                              \
  96   {                                                                           \
  97     return (wg_main.op_mode_flags & WG_OP_MODE_FLAG_##v);                     \
  98   }
  99 foreach_wg_op_mode_flags
 100 #undef _
 101
 102   typedef struct
 103 {
 104   u8 __pad[22];
 105   u16 next_index;
 106 } wg_post_data_t;
 107
 108 STATIC_ASSERT (sizeof (wg_post_data_t) <=
 109                  STRUCT_SIZE_OF (vnet_buffer_opaque_t, unused),
 110                "Custom meta-data too large for vnet_buffer_opaque_t");
 111
 112 #define wg_post_data(b)                                                       \
 113   ((wg_post_data_t *) ((u8 *) ((b)->opaque) +                                 \
 114                        STRUCT_OFFSET_OF (vnet_buffer_opaque_t, unused)))
 115
 116 #define WG_START_EVENT  1
 117 void wg_feature_init (wg_main_t * wmp);
 118 void wg_set_async_mode (u32 is_enabled);
 119
 120 void wg_secure_zero_memory (void *v, size_t n);
 121
 122 #endif /* __included_wg_h__ */
 123
 124 /*
 125  * fd.io coding-style-patch-verification: ON
 126  *
 127  * Local Variables:
 128  * eval: (c-set-style "gnu")
 129  * End:
 130  */