src/vnet/crypto/crypto.c

   1 /*
   2  * Copyright (c) 2018 Cisco 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
  16 #include <stdbool.h>
  17 #include <vlib/vlib.h>
  18 #include <vnet/crypto/crypto.h>
  19
  20 vnet_crypto_main_t crypto_main;
  21
  22 static_always_inline void
  23 crypto_set_op_status (vnet_crypto_op_t * ops[], u32 n_ops, int status)
  24 {
  25   while (n_ops--)
  26     {
  27       ops[0]->status = status;
  28       ops++;
  29     }
  30 }
  31
  32 static_always_inline u32
  33 vnet_crypto_process_ops_call_handler (vlib_main_t * vm,
  34                                       vnet_crypto_main_t * cm,
  35                                       vnet_crypto_op_id_t opt,
  36                                       vnet_crypto_op_t * ops[],
  37                                       vnet_crypto_op_chunk_t * chunks,
  38                                       u32 n_ops)
  39 {
  40   u32 rv = 0;
  41   if (n_ops == 0)
  42     return 0;
  43
  44   if (chunks)
  45     {
  46
  47       if (cm->chained_ops_handlers[opt] == 0)
  48         crypto_set_op_status (ops, n_ops,
  49                               VNET_CRYPTO_OP_STATUS_FAIL_NO_HANDLER);
  50       else
  51         rv = (cm->chained_ops_handlers[opt]) (vm, ops, chunks, n_ops);
  52     }
  53   else
  54     {
  55       if (cm->ops_handlers[opt] == 0)
  56         crypto_set_op_status (ops, n_ops,
  57                               VNET_CRYPTO_OP_STATUS_FAIL_NO_HANDLER);
  58       else
  59         rv = (cm->ops_handlers[opt]) (vm, ops, n_ops);
  60     }
  61   return rv;
  62 }
  63
  64 static_always_inline u32
  65 vnet_crypto_process_ops_inline (vlib_main_t * vm, vnet_crypto_op_t ops[],
  66                                 vnet_crypto_op_chunk_t * chunks, u32 n_ops)
  67 {
  68   vnet_crypto_main_t *cm = &crypto_main;
  69   const int op_q_size = VLIB_FRAME_SIZE;
  70   vnet_crypto_op_t *op_queue[op_q_size];
  71   vnet_crypto_op_id_t opt, current_op_type = ~0;
  72   u32 n_op_queue = 0;
  73   u32 rv = 0, i;
  74
  75   ASSERT (n_ops >= 1);
  76
  77   for (i = 0; i < n_ops; i++)
  78     {
  79       opt = ops[i].op;
  80
  81       if (current_op_type != opt || n_op_queue >= op_q_size)
  82         {
  83           rv += vnet_crypto_process_ops_call_handler (vm, cm, current_op_type,
  84                                                       op_queue, chunks,
  85                                                       n_op_queue);
  86           n_op_queue = 0;
  87           current_op_type = opt;
  88         }
  89
  90       op_queue[n_op_queue++] = &ops[i];
  91     }
  92
  93   rv += vnet_crypto_process_ops_call_handler (vm, cm, current_op_type,
  94                                               op_queue, chunks, n_op_queue);
  95   return rv;
  96 }
  97
  98 u32
  99 vnet_crypto_process_ops (vlib_main_t * vm, vnet_crypto_op_t ops[], u32 n_ops)
 100 {
 101   return vnet_crypto_process_ops_inline (vm, ops, 0, n_ops);
 102 }
 103
 104 u32
 105 vnet_crypto_process_chained_ops (vlib_main_t * vm, vnet_crypto_op_t ops[],
 106                                  vnet_crypto_op_chunk_t * chunks, u32 n_ops)
 107 {
 108   return vnet_crypto_process_ops_inline (vm, ops, chunks, n_ops);
 109 }
 110
 111 u32
 112 vnet_crypto_register_engine (vlib_main_t * vm, char *name, int prio,
 113                              char *desc)
 114 {
 115   vnet_crypto_main_t *cm = &crypto_main;
 116   vnet_crypto_engine_t *p;
 117
 118   vec_add2 (cm->engines, p, 1);
 119   p->name = name;
 120   p->desc = desc;
 121   p->priority = prio;
 122
 123   hash_set_mem (cm->engine_index_by_name, p->name, p - cm->engines);
 124
 125   return p - cm->engines;
 126 }
 127
 128 static_always_inline void
 129 crypto_set_active_engine (vnet_crypto_op_data_t * od,
 130                           vnet_crypto_op_id_t id, u32 ei,
 131                           crypto_op_class_type_t oct)
 132 {
 133   vnet_crypto_main_t *cm = &crypto_main;
 134   vnet_crypto_engine_t *ce = vec_elt_at_index (cm->engines, ei);
 135
 136   if (oct == CRYPTO_OP_BOTH || oct == CRYPTO_OP_CHAINED)
 137     {
 138       if (ce->chained_ops_handlers[id])
 139         {
 140           od->active_engine_index_chained = ei;
 141           cm->chained_ops_handlers[id] = ce->chained_ops_handlers[id];
 142         }
 143     }
 144
 145   if (oct == CRYPTO_OP_BOTH || oct == CRYPTO_OP_SIMPLE)
 146     {
 147       if (ce->ops_handlers[id])
 148         {
 149           od->active_engine_index_simple = ei;
 150           cm->ops_handlers[id] = ce->ops_handlers[id];
 151         }
 152     }
 153 }
 154
 155 int
 156 vnet_crypto_set_handler2 (char *alg_name, char *engine,
 157                           crypto_op_class_type_t oct)
 158 {
 159   uword *p;
 160   vnet_crypto_main_t *cm = &crypto_main;
 161   vnet_crypto_alg_data_t *ad;
 162   int i;
 163
 164   p = hash_get_mem (cm->alg_index_by_name, alg_name);
 165   if (!p)
 166     return -1;
 167
 168   ad = vec_elt_at_index (cm->algs, p[0]);
 169
 170   p = hash_get_mem (cm->engine_index_by_name, engine);
 171   if (!p)
 172     return -1;
 173
 174   for (i = 0; i < VNET_CRYPTO_OP_N_TYPES; i++)
 175     {
 176       vnet_crypto_op_data_t *od;
 177       vnet_crypto_op_id_t id = ad->op_by_type[i];
 178       if (id == 0)
 179         continue;
 180
 181       od = cm->opt_data + id;
 182       crypto_set_active_engine (od, id, p[0], oct);
 183     }
 184
 185   return 0;
 186 }
 187
 188 int
 189 vnet_crypto_is_set_handler (vnet_crypto_alg_t alg)
 190 {
 191   vnet_crypto_main_t *cm = &crypto_main;
 192
 193   return (alg < vec_len (cm->ops_handlers) && NULL != cm->ops_handlers[alg]);
 194 }
 195
 196 void
 197 vnet_crypto_register_ops_handler_inline (vlib_main_t * vm, u32 engine_index,
 198                                          vnet_crypto_op_id_t opt,
 199                                          vnet_crypto_ops_handler_t * fn,
 200                                          vnet_crypto_chained_ops_handler_t *
 201                                          cfn)
 202 {
 203   vnet_crypto_main_t *cm = &crypto_main;
 204   vnet_crypto_engine_t *ae, *e = vec_elt_at_index (cm->engines, engine_index);
 205   vnet_crypto_op_data_t *otd = cm->opt_data + opt;
 206   vec_validate_aligned (cm->ops_handlers, VNET_CRYPTO_N_OP_IDS - 1,
 207                         CLIB_CACHE_LINE_BYTES);
 208   vec_validate_aligned (cm->chained_ops_handlers, VNET_CRYPTO_N_OP_IDS - 1,
 209                         CLIB_CACHE_LINE_BYTES);
 210
 211   if (fn)
 212     {
 213       e->ops_handlers[opt] = fn;
 214       if (otd->active_engine_index_simple == ~0)
 215         {
 216           otd->active_engine_index_simple = engine_index;
 217           cm->ops_handlers[opt] = fn;
 218         }
 219
 220       ae = vec_elt_at_index (cm->engines, otd->active_engine_index_simple);
 221       if (ae->priority < e->priority)
 222         crypto_set_active_engine (otd, opt, engine_index, CRYPTO_OP_SIMPLE);
 223     }
 224
 225   if (cfn)
 226     {
 227       e->chained_ops_handlers[opt] = cfn;
 228       if (otd->active_engine_index_chained == ~0)
 229         {
 230           otd->active_engine_index_chained = engine_index;
 231           cm->chained_ops_handlers[opt] = cfn;
 232         }
 233
 234       ae = vec_elt_at_index (cm->engines, otd->active_engine_index_chained);
 235       if (ae->priority < e->priority)
 236         crypto_set_active_engine (otd, opt, engine_index, CRYPTO_OP_CHAINED);
 237     }
 238
 239   return;
 240 }
 241
 242 void
 243 vnet_crypto_register_ops_handler (vlib_main_t * vm, u32 engine_index,
 244                                   vnet_crypto_op_id_t opt,
 245                                   vnet_crypto_ops_handler_t * fn)
 246 {
 247   vnet_crypto_register_ops_handler_inline (vm, engine_index, opt, fn, 0);
 248 }
 249
 250 void
 251 vnet_crypto_register_chained_ops_handler (vlib_main_t * vm, u32 engine_index,
 252                                           vnet_crypto_op_id_t opt,
 253                                           vnet_crypto_chained_ops_handler_t *
 254                                           fn)
 255 {
 256   vnet_crypto_register_ops_handler_inline (vm, engine_index, opt, 0, fn);
 257 }
 258
 259 void
 260 vnet_crypto_register_ops_handlers (vlib_main_t * vm, u32 engine_index,
 261                                    vnet_crypto_op_id_t opt,
 262                                    vnet_crypto_ops_handler_t * fn,
 263                                    vnet_crypto_chained_ops_handler_t * cfn)
 264 {
 265   vnet_crypto_register_ops_handler_inline (vm, engine_index, opt, fn, cfn);
 266 }
 267
 268 void
 269 vnet_crypto_register_async_handler (vlib_main_t * vm, u32 engine_index,
 270                                     vnet_crypto_async_op_id_t opt,
 271                                     vnet_crypto_frame_enqueue_t * enqueue_hdl,
 272                                     vnet_crypto_frame_dequeue_t * dequeue_hdl)
 273 {
 274   vnet_crypto_main_t *cm = &crypto_main;
 275   vnet_crypto_engine_t *ae, *e = vec_elt_at_index (cm->engines, engine_index);
 276   vnet_crypto_async_op_data_t *otd = cm->async_opt_data + opt;
 277   vec_validate_aligned (cm->enqueue_handlers, VNET_CRYPTO_ASYNC_OP_N_IDS - 1,
 278                         CLIB_CACHE_LINE_BYTES);
 279   vec_validate_aligned (cm->dequeue_handlers, VNET_CRYPTO_ASYNC_OP_N_IDS - 1,
 280                         CLIB_CACHE_LINE_BYTES);
 281
 282   /* both enqueue hdl and dequeue hdl should present */
 283   if (!enqueue_hdl && !dequeue_hdl)
 284     return;
 285
 286   e->enqueue_handlers[opt] = enqueue_hdl;
 287   e->dequeue_handlers[opt] = dequeue_hdl;
 288   if (otd->active_engine_index_async == ~0)
 289     {
 290       otd->active_engine_index_async = engine_index;
 291       cm->enqueue_handlers[opt] = enqueue_hdl;
 292       cm->dequeue_handlers[opt] = dequeue_hdl;
 293     }
 294
 295   ae = vec_elt_at_index (cm->engines, otd->active_engine_index_async);
 296   if (ae->priority < e->priority)
 297     {
 298       otd->active_engine_index_async = engine_index;
 299       cm->enqueue_handlers[opt] = enqueue_hdl;
 300       cm->dequeue_handlers[opt] = dequeue_hdl;
 301     }
 302
 303   return;
 304 }
 305
 306 void
 307 vnet_crypto_register_key_handler (vlib_main_t * vm, u32 engine_index,
 308                                   vnet_crypto_key_handler_t * key_handler)
 309 {
 310   vnet_crypto_main_t *cm = &crypto_main;
 311   vnet_crypto_engine_t *e = vec_elt_at_index (cm->engines, engine_index);
 312   e->key_op_handler = key_handler;
 313   return;
 314 }
 315
 316 static int
 317 vnet_crypto_key_len_check (vnet_crypto_alg_t alg, u16 length)
 318 {
 319   switch (alg)
 320     {
 321     case VNET_CRYPTO_N_ALGS:
 322       return 0;
 323     case VNET_CRYPTO_ALG_NONE:
 324       return 1;
 325
 326 #define _(n, s, l) \
 327       case VNET_CRYPTO_ALG_##n: \
 328         if ((l) == length) \
 329           return 1;        \
 330         break;
 331       foreach_crypto_cipher_alg foreach_crypto_aead_alg
 332 #undef _
 333         /* HMAC allows any key length */
 334 #define _(n, s) \
 335       case VNET_CRYPTO_ALG_HMAC_##n: \
 336         return 1;
 337         foreach_crypto_hmac_alg
 338 #undef _
 339     }
 340
 341   return 0;
 342 }
 343
 344 u32
 345 vnet_crypto_key_add (vlib_main_t * vm, vnet_crypto_alg_t alg, u8 * data,
 346                      u16 length)
 347 {
 348   u32 index;
 349   vnet_crypto_main_t *cm = &crypto_main;
 350   vnet_crypto_engine_t *engine;
 351   vnet_crypto_key_t *key;
 352
 353   if (!vnet_crypto_key_len_check (alg, length))
 354     return ~0;
 355
 356   pool_get_zero (cm->keys, key);
 357   index = key - cm->keys;
 358   key->type = VNET_CRYPTO_KEY_TYPE_DATA;
 359   key->alg = alg;
 360   vec_validate_aligned (key->data, length - 1, CLIB_CACHE_LINE_BYTES);
 361   clib_memcpy (key->data, data, length);
 362   /* *INDENT-OFF* */
 363   vec_foreach (engine, cm->engines)
 364     if (engine->key_op_handler)
 365       engine->key_op_handler (vm, VNET_CRYPTO_KEY_OP_ADD, index);
 366   /* *INDENT-ON* */
 367   return index;
 368 }
 369
 370 void
 371 vnet_crypto_key_del (vlib_main_t * vm, vnet_crypto_key_index_t index)
 372 {
 373   vnet_crypto_main_t *cm = &crypto_main;
 374   vnet_crypto_engine_t *engine;
 375   vnet_crypto_key_t *key = pool_elt_at_index (cm->keys, index);
 376
 377   /* *INDENT-OFF* */
 378   vec_foreach (engine, cm->engines)
 379     if (engine->key_op_handler)
 380       engine->key_op_handler (vm, VNET_CRYPTO_KEY_OP_DEL, index);
 381   /* *INDENT-ON* */
 382
 383   if (key->type == VNET_CRYPTO_KEY_TYPE_DATA)
 384     {
 385       clib_memset (key->data, 0, vec_len (key->data));
 386       vec_free (key->data);
 387     }
 388   else if (key->type == VNET_CRYPTO_KEY_TYPE_LINK)
 389     {
 390       key->index_crypto = key->index_integ = 0;
 391     }
 392
 393   pool_put (cm->keys, key);
 394 }
 395
 396 vnet_crypto_async_alg_t
 397 vnet_crypto_link_algs (vnet_crypto_alg_t crypto_alg,
 398                        vnet_crypto_alg_t integ_alg)
 399 {
 400 #define _(c, h, s, k ,d) \
 401   if (crypto_alg == VNET_CRYPTO_ALG_##c && \
 402       integ_alg == VNET_CRYPTO_ALG_HMAC_##h) \
 403     return VNET_CRYPTO_ALG_##c##_##h##_TAG##d;
 404   foreach_crypto_link_async_alg
 405 #undef _
 406     return ~0;
 407 }
 408
 409 u32
 410 vnet_crypto_key_add_linked (vlib_main_t * vm,
 411                             vnet_crypto_key_index_t index_crypto,
 412                             vnet_crypto_key_index_t index_integ)
 413 {
 414   u32 index;
 415   vnet_crypto_main_t *cm = &crypto_main;
 416   vnet_crypto_engine_t *engine;
 417   vnet_crypto_key_t *key_crypto, *key_integ, *key;
 418   vnet_crypto_async_alg_t linked_alg;
 419
 420   key_crypto = pool_elt_at_index (cm->keys, index_crypto);
 421   key_integ = pool_elt_at_index (cm->keys, index_integ);
 422
 423   if (!key_crypto || !key_integ)
 424     return ~0;
 425
 426   linked_alg = vnet_crypto_link_algs (key_crypto->alg, key_integ->alg);
 427   if (linked_alg == ~0)
 428     return ~0;
 429
 430   pool_get_zero (cm->keys, key);
 431   index = key - cm->keys;
 432   key->type = VNET_CRYPTO_KEY_TYPE_LINK;
 433   key->index_crypto = index_crypto;
 434   key->index_integ = index_integ;
 435   key->async_alg = linked_alg;
 436
 437   /* *INDENT-OFF* */
 438   vec_foreach (engine, cm->engines)
 439     if (engine->key_op_handler)
 440       engine->key_op_handler (vm, VNET_CRYPTO_KEY_OP_ADD, index);
 441   /* *INDENT-ON* */
 442
 443   return index;
 444 }
 445
 446 clib_error_t *
 447 crypto_dispatch_enable_disable (int is_enable)
 448 {
 449   vnet_crypto_main_t *cm = &crypto_main;
 450   vlib_thread_main_t *tm = vlib_get_thread_main ();
 451   u32 skip_master = vlib_num_workers () > 0, i;
 452   vlib_node_state_t state = VLIB_NODE_STATE_DISABLED;
 453   u8 state_change = 0;
 454
 455   CLIB_MEMORY_STORE_BARRIER ();
 456   if (is_enable && cm->async_refcnt > 0)
 457     {
 458       state_change = 1;
 459       state =
 460         cm->dispatch_mode ==
 461         VNET_CRYPTO_ASYNC_DISPATCH_POLLING ? VLIB_NODE_STATE_POLLING :
 462         VLIB_NODE_STATE_INTERRUPT;
 463     }
 464
 465   if (!is_enable && cm->async_refcnt == 0)
 466     {
 467       state_change = 1;
 468       state = VLIB_NODE_STATE_DISABLED;
 469     }
 470
 471   if (state_change)
 472     for (i = skip_master; i < tm->n_vlib_mains; i++)
 473       {
 474         if (state !=
 475             vlib_node_get_state (vlib_mains[i], cm->crypto_node_index))
 476           vlib_node_set_state (vlib_mains[i], cm->crypto_node_index, state);
 477       }
 478   return 0;
 479 }
 480
 481 static_always_inline void
 482 crypto_set_active_async_engine (vnet_crypto_async_op_data_t * od,
 483                                 vnet_crypto_async_op_id_t id, u32 ei)
 484 {
 485   vnet_crypto_main_t *cm = &crypto_main;
 486   vnet_crypto_engine_t *ce = vec_elt_at_index (cm->engines, ei);
 487
 488   if (ce->enqueue_handlers[id] && ce->dequeue_handlers[id])
 489     {
 490       od->active_engine_index_async = ei;
 491       cm->enqueue_handlers[id] = ce->enqueue_handlers[id];
 492       cm->dequeue_handlers[id] = ce->dequeue_handlers[id];
 493     }
 494 }
 495
 496 int
 497 vnet_crypto_set_async_handler2 (char *alg_name, char *engine)
 498 {
 499   uword *p;
 500   vnet_crypto_main_t *cm = &crypto_main;
 501   vnet_crypto_async_alg_data_t *ad;
 502   int i;
 503
 504   p = hash_get_mem (cm->async_alg_index_by_name, alg_name);
 505   if (!p)
 506     return -1;
 507
 508   ad = vec_elt_at_index (cm->async_algs, p[0]);
 509
 510   p = hash_get_mem (cm->engine_index_by_name, engine);
 511   if (!p)
 512     return -1;
 513
 514   for (i = 0; i < VNET_CRYPTO_ASYNC_OP_N_TYPES; i++)
 515     {
 516       vnet_crypto_async_op_data_t *od;
 517       vnet_crypto_async_op_id_t id = ad->op_by_type[i];
 518       if (id == 0)
 519         continue;
 520
 521       od = cm->async_opt_data + id;
 522       crypto_set_active_async_engine (od, id, p[0]);
 523     }
 524
 525   return 0;
 526 }
 527
 528 u32
 529 vnet_crypto_register_post_node (vlib_main_t * vm, char *post_node_name)
 530 {
 531   vnet_crypto_main_t *cm = &crypto_main;
 532   vnet_crypto_async_next_node_t *nn = 0;
 533   vlib_node_t *cc, *pn;
 534   uword index = vec_len (cm->next_nodes);
 535
 536   pn = vlib_get_node_by_name (vm, (u8 *) post_node_name);
 537   if (!pn)
 538     return ~0;
 539
 540   /* *INDENT-OFF* */
 541   vec_foreach (cm->next_nodes, nn)
 542   {
 543     if (nn->node_idx == pn->index)
 544       return nn->next_idx;
 545   }
 546   /* *INDENT-ON* */
 547
 548   vec_validate (cm->next_nodes, index);
 549   nn = vec_elt_at_index (cm->next_nodes, index);
 550
 551   cc = vlib_get_node_by_name (vm, (u8 *) "crypto-dispatch");
 552   nn->next_idx = vlib_node_add_named_next (vm, cc->index, post_node_name);
 553   nn->node_idx = pn->index;
 554
 555   return nn->next_idx;
 556 }
 557
 558 void
 559 vnet_crypto_request_async_mode (int is_enable)
 560 {
 561   vnet_crypto_main_t *cm = &crypto_main;
 562   vlib_thread_main_t *tm = vlib_get_thread_main ();
 563   u32 skip_master = vlib_num_workers () > 0, i;
 564   vlib_node_state_t state = VLIB_NODE_STATE_DISABLED;
 565   u8 state_change = 0;
 566
 567   CLIB_MEMORY_STORE_BARRIER ();
 568   if (is_enable && cm->async_refcnt == 0)
 569     {
 570       state_change = 1;
 571       state =
 572         cm->dispatch_mode == VNET_CRYPTO_ASYNC_DISPATCH_POLLING ?
 573         VLIB_NODE_STATE_POLLING : VLIB_NODE_STATE_INTERRUPT;
 574     }
 575   if (!is_enable && cm->async_refcnt == 1)
 576     {
 577       state_change = 1;
 578       state = VLIB_NODE_STATE_DISABLED;
 579     }
 580
 581   if (state_change)
 582     for (i = skip_master; i < tm->n_vlib_mains; i++)
 583       {
 584         if (state !=
 585             vlib_node_get_state (vlib_mains[i], cm->crypto_node_index))
 586           vlib_node_set_state (vlib_mains[i], cm->crypto_node_index, state);
 587       }
 588
 589   if (is_enable)
 590     cm->async_refcnt += 1;
 591   else if (cm->async_refcnt > 0)
 592     cm->async_refcnt -= 1;
 593 }
 594
 595 void
 596 vnet_crypto_set_async_dispatch_mode (u8 mode)
 597 {
 598   vnet_crypto_main_t *cm = &crypto_main;
 599   u32 skip_master = vlib_num_workers () > 0, i;
 600   vlib_thread_main_t *tm = vlib_get_thread_main ();
 601   vlib_node_state_t state = VLIB_NODE_STATE_DISABLED;
 602
 603   CLIB_MEMORY_STORE_BARRIER ();
 604   cm->dispatch_mode = mode;
 605   if (mode == VNET_CRYPTO_ASYNC_DISPATCH_INTERRUPT)
 606     {
 607       state =
 608         cm->async_refcnt == 0 ?
 609         VLIB_NODE_STATE_DISABLED : VLIB_NODE_STATE_INTERRUPT;
 610     }
 611   else if (mode == VNET_CRYPTO_ASYNC_DISPATCH_POLLING)
 612     {
 613       state =
 614         cm->async_refcnt == 0 ?
 615         VLIB_NODE_STATE_DISABLED : VLIB_NODE_STATE_POLLING;
 616     }
 617
 618   for (i = skip_master; i < tm->n_vlib_mains; i++)
 619     {
 620       if (state != vlib_node_get_state (vlib_mains[i], cm->crypto_node_index))
 621         vlib_node_set_state (vlib_mains[i], cm->crypto_node_index, state);
 622     }
 623   clib_warning ("Switching dispatch mode might not work is some situations.");
 624   clib_warning
 625     ("Use 'show crypto async status' to verify that the nodes' states were set");
 626   clib_warning ("and if not, set 'crypto async dispatch' mode again.");
 627 }
 628
 629 int
 630 vnet_crypto_is_set_async_handler (vnet_crypto_async_op_id_t op)
 631 {
 632   vnet_crypto_main_t *cm = &crypto_main;
 633
 634   return (op < vec_len (cm->enqueue_handlers) &&
 635           NULL != cm->enqueue_handlers[op]);
 636 }
 637
 638 static void
 639 vnet_crypto_init_cipher_data (vnet_crypto_alg_t alg, vnet_crypto_op_id_t eid,
 640                               vnet_crypto_op_id_t did, char *name, u8 is_aead)
 641 {
 642   vnet_crypto_op_type_t eopt, dopt;
 643   vnet_crypto_main_t *cm = &crypto_main;
 644
 645   cm->algs[alg].name = name;
 646   cm->opt_data[eid].alg = cm->opt_data[did].alg = alg;
 647   cm->opt_data[eid].active_engine_index_simple = ~0;
 648   cm->opt_data[did].active_engine_index_simple = ~0;
 649   cm->opt_data[eid].active_engine_index_chained = ~0;
 650   cm->opt_data[did].active_engine_index_chained = ~0;
 651   if (is_aead)
 652     {
 653       eopt = VNET_CRYPTO_OP_TYPE_AEAD_ENCRYPT;
 654       dopt = VNET_CRYPTO_OP_TYPE_AEAD_DECRYPT;
 655     }
 656   else
 657     {
 658       eopt = VNET_CRYPTO_OP_TYPE_ENCRYPT;
 659       dopt = VNET_CRYPTO_OP_TYPE_DECRYPT;
 660     }
 661   cm->opt_data[eid].type = eopt;
 662   cm->opt_data[did].type = dopt;
 663   cm->algs[alg].op_by_type[eopt] = eid;
 664   cm->algs[alg].op_by_type[dopt] = did;
 665   hash_set_mem (cm->alg_index_by_name, name, alg);
 666 }
 667
 668 static void
 669 vnet_crypto_init_hmac_data (vnet_crypto_alg_t alg,
 670                             vnet_crypto_op_id_t id, char *name)
 671 {
 672   vnet_crypto_main_t *cm = &crypto_main;
 673   cm->algs[alg].name = name;
 674   cm->algs[alg].op_by_type[VNET_CRYPTO_OP_TYPE_HMAC] = id;
 675   cm->opt_data[id].alg = alg;
 676   cm->opt_data[id].active_engine_index_simple = ~0;
 677   cm->opt_data[id].active_engine_index_chained = ~0;
 678   cm->opt_data[id].type = VNET_CRYPTO_OP_TYPE_HMAC;
 679   hash_set_mem (cm->alg_index_by_name, name, alg);
 680 }
 681
 682 static void
 683 vnet_crypto_init_async_data (vnet_crypto_async_alg_t alg,
 684                              vnet_crypto_async_op_id_t eid,
 685                              vnet_crypto_async_op_id_t did, char *name)
 686 {
 687   vnet_crypto_main_t *cm = &crypto_main;
 688
 689   cm->async_algs[alg].name = name;
 690   cm->async_algs[alg].op_by_type[VNET_CRYPTO_ASYNC_OP_TYPE_ENCRYPT] = eid;
 691   cm->async_algs[alg].op_by_type[VNET_CRYPTO_ASYNC_OP_TYPE_DECRYPT] = did;
 692   cm->async_opt_data[eid].type = VNET_CRYPTO_ASYNC_OP_TYPE_ENCRYPT;
 693   cm->async_opt_data[eid].alg = alg;
 694   cm->async_opt_data[eid].active_engine_index_async = ~0;
 695   cm->async_opt_data[eid].active_engine_index_async = ~0;
 696   cm->async_opt_data[did].type = VNET_CRYPTO_ASYNC_OP_TYPE_DECRYPT;
 697   cm->async_opt_data[did].alg = alg;
 698   cm->async_opt_data[did].active_engine_index_async = ~0;
 699   cm->async_opt_data[did].active_engine_index_async = ~0;
 700   hash_set_mem (cm->async_alg_index_by_name, name, alg);
 701 }
 702
 703 clib_error_t *
 704 vnet_crypto_init (vlib_main_t * vm)
 705 {
 706   vnet_crypto_main_t *cm = &crypto_main;
 707   vlib_thread_main_t *tm = vlib_get_thread_main ();
 708   vnet_crypto_thread_t *ct = 0;
 709
 710   cm->dispatch_mode = VNET_CRYPTO_ASYNC_DISPATCH_POLLING;
 711   cm->engine_index_by_name = hash_create_string ( /* size */ 0,
 712                                                  sizeof (uword));
 713   cm->alg_index_by_name = hash_create_string (0, sizeof (uword));
 714   cm->async_alg_index_by_name = hash_create_string (0, sizeof (uword));
 715   vec_validate_aligned (cm->threads, tm->n_vlib_mains, CLIB_CACHE_LINE_BYTES);
 716   vec_foreach (ct, cm->threads)
 717     pool_alloc_aligned (ct->frame_pool, 256, CLIB_CACHE_LINE_BYTES);
 718   vec_validate (cm->algs, VNET_CRYPTO_N_ALGS);
 719   vec_validate (cm->async_algs, VNET_CRYPTO_N_ASYNC_ALGS);
 720   clib_bitmap_validate (cm->async_active_ids, VNET_CRYPTO_ASYNC_OP_N_IDS - 1);
 721
 722 #define _(n, s, l) \
 723   vnet_crypto_init_cipher_data (VNET_CRYPTO_ALG_##n, \
 724                                 VNET_CRYPTO_OP_##n##_ENC, \
 725                                 VNET_CRYPTO_OP_##n##_DEC, s, 0);
 726   foreach_crypto_cipher_alg;
 727 #undef _
 728 #define _(n, s, l) \
 729   vnet_crypto_init_cipher_data (VNET_CRYPTO_ALG_##n, \
 730                                 VNET_CRYPTO_OP_##n##_ENC, \
 731                                 VNET_CRYPTO_OP_##n##_DEC, s, 1);
 732   foreach_crypto_aead_alg;
 733 #undef _
 734 #define _(n, s) \
 735   vnet_crypto_init_hmac_data (VNET_CRYPTO_ALG_HMAC_##n, \
 736                               VNET_CRYPTO_OP_##n##_HMAC, "hmac-" s);
 737   foreach_crypto_hmac_alg;
 738 #undef _
 739 #define _(n, s, k, t, a) \
 740   vnet_crypto_init_async_data (VNET_CRYPTO_ALG_##n##_TAG##t##_AAD##a, \
 741                                VNET_CRYPTO_OP_##n##_TAG##t##_AAD##a##_ENC, \
 742                                VNET_CRYPTO_OP_##n##_TAG##t##_AAD##a##_DEC, \
 743                                s);
 744   foreach_crypto_aead_async_alg
 745 #undef _
 746 #define _(c, h, s, k ,d) \
 747   vnet_crypto_init_async_data (VNET_CRYPTO_ALG_##c##_##h##_TAG##d, \
 748                                VNET_CRYPTO_OP_##c##_##h##_TAG##d##_ENC, \
 749                                VNET_CRYPTO_OP_##c##_##h##_TAG##d##_DEC, \
 750                                s);
 751     foreach_crypto_link_async_alg
 752 #undef _
 753     cm->crypto_node_index =
 754     vlib_get_node_by_name (vm, (u8 *) "crypto-dispatch")->index;
 755
 756   return 0;
 757 }
 758
 759 VLIB_INIT_FUNCTION (vnet_crypto_init);
 760
 761 /*
 762  * fd.io coding-style-patch-verification: ON
 763  *
 764  * Local Variables:
 765  * eval: (c-set-style "gnu")
 766  * End:
 767  */