src/plugins/crypto_native/main.c

   1 /*
   2  *------------------------------------------------------------------
   3  * Copyright (c) 2019 Cisco and/or its affiliates.
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  *------------------------------------------------------------------
  16  */
  17
  18 #include <vlib/vlib.h>
  19 #include <vnet/plugin/plugin.h>
  20 #include <vnet/crypto/crypto.h>
  21 #include <crypto_native/crypto_native.h>
  22
  23 crypto_native_main_t crypto_native_main;
  24
  25 static void
  26 crypto_native_key_handler (vlib_main_t * vm, vnet_crypto_key_op_t kop,
  27                            vnet_crypto_key_index_t idx)
  28 {
  29   vnet_crypto_key_t *key = vnet_crypto_get_key (idx);
  30   crypto_native_main_t *cm = &crypto_native_main;
  31
  32   if (cm->key_fn[key->alg] == 0)
  33     return;
  34
  35   if (kop == VNET_CRYPTO_KEY_OP_DEL)
  36     {
  37       if (idx >= vec_len (cm->key_data))
  38         return;
  39
  40       if (cm->key_data[idx] == 0)
  41         return;
  42
  43       clib_mem_free_s (cm->key_data[idx]);
  44       cm->key_data[idx] = 0;
  45       return;
  46     }
  47
  48   /** TODO: add linked alg support **/
  49   if (key->type == VNET_CRYPTO_KEY_TYPE_LINK)
  50     return;
  51
  52   vec_validate_aligned (cm->key_data, idx, CLIB_CACHE_LINE_BYTES);
  53
  54   if (kop == VNET_CRYPTO_KEY_OP_MODIFY && cm->key_data[idx])
  55     {
  56       clib_mem_free_s (cm->key_data[idx]);
  57     }
  58
  59   cm->key_data[idx] = cm->key_fn[key->alg] (key);
  60 }
  61
  62 clib_error_t *
  63 crypto_native_init (vlib_main_t * vm)
  64 {
  65   crypto_native_main_t *cm = &crypto_native_main;
  66   vlib_thread_main_t *tm = vlib_get_thread_main ();
  67   clib_error_t *error = 0;
  68
  69   if (clib_cpu_supports_x86_aes () == 0 &&
  70       clib_cpu_supports_aarch64_aes () == 0)
  71     return 0;
  72
  73   vec_validate_aligned (cm->per_thread_data, tm->n_vlib_mains - 1,
  74                         CLIB_CACHE_LINE_BYTES);
  75
  76   cm->crypto_engine_index =
  77     vnet_crypto_register_engine (vm, "native", 100,
  78                                  "Native ISA Optimized Crypto");
  79
  80 #if __x86_64__
  81   if (clib_cpu_supports_vaes ())
  82     error = crypto_native_aes_cbc_init_vaes (vm);
  83   else if (clib_cpu_supports_avx512f ())
  84     error = crypto_native_aes_cbc_init_avx512 (vm);
  85   else if (clib_cpu_supports_avx2 ())
  86     error = crypto_native_aes_cbc_init_avx2 (vm);
  87   else
  88     error = crypto_native_aes_cbc_init_sse42 (vm);
  89
  90   if (error)
  91     goto error;
  92
  93   if (clib_cpu_supports_pclmulqdq ())
  94     {
  95       if (clib_cpu_supports_vaes ())
  96         error = crypto_native_aes_gcm_init_vaes (vm);
  97       else if (clib_cpu_supports_avx512f ())
  98         error = crypto_native_aes_gcm_init_avx512 (vm);
  99       else if (clib_cpu_supports_avx2 ())
 100         error = crypto_native_aes_gcm_init_avx2 (vm);
 101       else
 102         error = crypto_native_aes_gcm_init_sse42 (vm);
 103
 104       if (error)
 105         goto error;
 106     }
 107 #endif
 108 #if __aarch64__
 109   if ((error = crypto_native_aes_cbc_init_neon (vm)))
 110     goto error;
 111
 112   if ((error = crypto_native_aes_gcm_init_neon (vm)))
 113     goto error;
 114 #endif
 115
 116   vnet_crypto_register_key_handler (vm, cm->crypto_engine_index,
 117                                     crypto_native_key_handler);
 118
 119
 120 error:
 121   if (error)
 122     vec_free (cm->per_thread_data);
 123
 124   return error;
 125 }
 126
 127 /* *INDENT-OFF* */
 128 VLIB_INIT_FUNCTION (crypto_native_init) =
 129 {
 130   .runs_after = VLIB_INITS ("vnet_crypto_init"),
 131 };
 132 /* *INDENT-ON* */
 133
 134 #include <vpp/app/version.h>
 135
 136 /* *INDENT-OFF* */
 137 VLIB_PLUGIN_REGISTER () = {
 138   .version = VPP_BUILD_VER,
 139   .description = "Intel IA32 Software Crypto Engine",
 140 };
 141 /* *INDENT-ON* */
 142
 143 /*
 144  * fd.io coding-style-patch-verification: ON
 145  *
 146  * Local Variables:
 147  * eval: (c-set-style "gnu")
 148  * End:
 149  */