1 /*- 2 * Copyright (c) 2019 Mellanox Technologies. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS `AS IS' AND 14 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 16 * ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE 17 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 18 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 19 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 20 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 21 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 22 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 23 * SUCH DAMAGE. 24 * 25 * $FreeBSD$ 26 */ 27 28 #include "opt_kern_tls.h" 29 30 #include "en.h" 31 32 #include <dev/mlx5/tls.h> 33 34 #include <linux/delay.h> 35 #include <sys/ktls.h> 36 #include <opencrypto/cryptodev.h> 37 38 #ifdef KERN_TLS 39 40 MALLOC_DEFINE(M_MLX5E_TLS, "MLX5E_TLS", "MLX5 ethernet HW TLS"); 41 42 /* software TLS context */ 43 struct mlx5_ifc_sw_tls_cntx_bits { 44 struct mlx5_ifc_tls_static_params_bits param; 45 struct mlx5_ifc_tls_progress_params_bits progress; 46 struct { 47 uint8_t key_data[8][0x20]; 48 uint8_t key_len[0x20]; 49 } key; 50 }; 51 52 CTASSERT(MLX5_ST_SZ_BYTES(sw_tls_cntx) <= sizeof(((struct mlx5e_tls_tag *)0)->crypto_params)); 53 CTASSERT(MLX5_ST_SZ_BYTES(mkc) == sizeof(((struct mlx5e_tx_umr_wqe *)0)->mkc)); 54 55 static const char *mlx5e_tls_stats_desc[] = { 56 MLX5E_TLS_STATS(MLX5E_STATS_DESC) 57 }; 58 59 static void mlx5e_tls_work(struct work_struct *); 60 61 static int 62 mlx5e_tls_tag_zinit(void *mem, int size, int flags) 63 { 64 struct mlx5e_tls_tag *ptag = mem; 65 66 MPASS(size == sizeof(*ptag)); 67 68 memset(ptag, 0, sizeof(*ptag)); 69 mtx_init(&ptag->mtx, "mlx5-tls-tag-mtx", NULL, MTX_DEF); 70 INIT_WORK(&ptag->work, mlx5e_tls_work); 71 72 return (0); 73 } 74 75 static void 76 mlx5e_tls_tag_zfini(void *mem, int size) 77 { 78 struct mlx5e_tls_tag *ptag = mem; 79 struct mlx5e_priv *priv; 80 struct mlx5e_tls *ptls; 81 82 ptls = ptag->tls; 83 priv = container_of(ptls, struct mlx5e_priv, tls); 84 85 flush_work(&ptag->work); 86 87 if (ptag->tisn != 0) { 88 mlx5_tls_close_tis(priv->mdev, ptag->tisn); 89 atomic_add_32(&ptls->num_resources, -1U); 90 } 91 92 mtx_destroy(&ptag->mtx); 93 } 94 95 static void 96 mlx5e_tls_tag_zfree(struct mlx5e_tls_tag *ptag) 97 { 98 99 /* reset some variables */ 100 ptag->state = MLX5E_TLS_ST_INIT; 101 ptag->dek_index = 0; 102 ptag->dek_index_ok = 0; 103 104 /* avoid leaking keys */ 105 memset(ptag->crypto_params, 0, sizeof(ptag->crypto_params)); 106 107 /* update number of TIS contexts */ 108 if (ptag->tisn == 0) 109 atomic_add_32(&ptag->tls->num_resources, -1U); 110 111 /* return tag to UMA */ 112 uma_zfree(ptag->tls->zone, ptag); 113 } 114 115 int 116 mlx5e_tls_init(struct mlx5e_priv *priv) 117 { 118 struct mlx5e_tls *ptls = &priv->tls; 119 struct sysctl_oid *node; 120 uint32_t x; 121 122 if (MLX5_CAP_GEN(priv->mdev, tls_tx) == 0) 123 return (0); 124 125 ptls->wq = create_singlethread_workqueue("mlx5-tls-wq"); 126 if (ptls->wq == NULL) 127 return (ENOMEM); 128 129 sysctl_ctx_init(&ptls->ctx); 130 131 snprintf(ptls->zname, sizeof(ptls->zname), 132 "mlx5_%u_tls", device_get_unit(priv->mdev->pdev->dev.bsddev)); 133 134 ptls->zone = uma_zcreate(ptls->zname, sizeof(struct mlx5e_tls_tag), 135 NULL, NULL, mlx5e_tls_tag_zinit, mlx5e_tls_tag_zfini, UMA_ALIGN_CACHE, 0); 136 137 ptls->max_resources = 1U << MLX5_CAP_GEN(priv->mdev, log_max_dek); 138 139 for (x = 0; x != MLX5E_TLS_STATS_NUM; x++) 140 ptls->stats.arg[x] = counter_u64_alloc(M_WAITOK); 141 142 ptls->init = 1; 143 144 node = SYSCTL_ADD_NODE(&priv->sysctl_ctx, 145 SYSCTL_CHILDREN(priv->sysctl_ifnet), OID_AUTO, 146 "tls", CTLFLAG_RW | CTLFLAG_MPSAFE, NULL, "Hardware TLS offload"); 147 if (node == NULL) 148 return (0); 149 150 mlx5e_create_counter_stats(&ptls->ctx, 151 SYSCTL_CHILDREN(node), "stats", 152 mlx5e_tls_stats_desc, MLX5E_TLS_STATS_NUM, 153 ptls->stats.arg); 154 155 return (0); 156 } 157 158 void 159 mlx5e_tls_cleanup(struct mlx5e_priv *priv) 160 { 161 struct mlx5e_tls *ptls = &priv->tls; 162 uint32_t x; 163 164 if (MLX5_CAP_GEN(priv->mdev, tls_tx) == 0) 165 return; 166 167 ptls->init = 0; 168 flush_workqueue(ptls->wq); 169 sysctl_ctx_free(&ptls->ctx); 170 uma_zdestroy(ptls->zone); 171 destroy_workqueue(ptls->wq); 172 173 /* check if all resources are freed */ 174 MPASS(priv->tls.num_resources == 0); 175 176 for (x = 0; x != MLX5E_TLS_STATS_NUM; x++) 177 counter_u64_free(ptls->stats.arg[x]); 178 } 179 180 static void 181 mlx5e_tls_work(struct work_struct *work) 182 { 183 struct mlx5e_tls_tag *ptag; 184 struct mlx5e_priv *priv; 185 int err; 186 187 ptag = container_of(work, struct mlx5e_tls_tag, work); 188 priv = container_of(ptag->tls, struct mlx5e_priv, tls); 189 190 switch (ptag->state) { 191 case MLX5E_TLS_ST_INIT: 192 /* try to open TIS, if not present */ 193 if (ptag->tisn == 0) { 194 err = mlx5_tls_open_tis(priv->mdev, 0, priv->tdn, 195 priv->pdn, &ptag->tisn); 196 if (err) { 197 MLX5E_TLS_STAT_INC(ptag, tx_error, 1); 198 break; 199 } 200 } 201 MLX5_SET(sw_tls_cntx, ptag->crypto_params, progress.pd, ptag->tisn); 202 203 /* try to allocate a DEK context ID */ 204 err = mlx5_encryption_key_create(priv->mdev, priv->pdn, 205 MLX5_ADDR_OF(sw_tls_cntx, ptag->crypto_params, key.key_data), 206 MLX5_GET(sw_tls_cntx, ptag->crypto_params, key.key_len), 207 &ptag->dek_index); 208 if (err) { 209 MLX5E_TLS_STAT_INC(ptag, tx_error, 1); 210 break; 211 } 212 213 MLX5_SET(sw_tls_cntx, ptag->crypto_params, param.dek_index, ptag->dek_index); 214 215 ptag->dek_index_ok = 1; 216 217 MLX5E_TLS_TAG_LOCK(ptag); 218 if (ptag->state == MLX5E_TLS_ST_INIT) 219 ptag->state = MLX5E_TLS_ST_SETUP; 220 MLX5E_TLS_TAG_UNLOCK(ptag); 221 break; 222 223 case MLX5E_TLS_ST_FREED: 224 /* wait for all refs to go away */ 225 while (ptag->refs != 0) 226 msleep(1); 227 228 /* try to destroy DEK context by ID */ 229 if (ptag->dek_index_ok) 230 err = mlx5_encryption_key_destroy(priv->mdev, ptag->dek_index); 231 232 /* free tag */ 233 mlx5e_tls_tag_zfree(ptag); 234 break; 235 236 default: 237 break; 238 } 239 } 240 241 static int 242 mlx5e_tls_set_params(void *ctx, const struct tls_session_params *en) 243 { 244 245 MLX5_SET(sw_tls_cntx, ctx, param.const_2, 2); 246 if (en->tls_vminor == TLS_MINOR_VER_TWO) 247 MLX5_SET(sw_tls_cntx, ctx, param.tls_version, 2); /* v1.2 */ 248 else 249 MLX5_SET(sw_tls_cntx, ctx, param.tls_version, 3); /* v1.3 */ 250 MLX5_SET(sw_tls_cntx, ctx, param.const_1, 1); 251 MLX5_SET(sw_tls_cntx, ctx, param.encryption_standard, 1); /* TLS */ 252 253 /* copy the initial vector in place */ 254 switch (en->iv_len) { 255 case MLX5_FLD_SZ_BYTES(sw_tls_cntx, param.gcm_iv): 256 case MLX5_FLD_SZ_BYTES(sw_tls_cntx, param.gcm_iv) + 257 MLX5_FLD_SZ_BYTES(sw_tls_cntx, param.implicit_iv): 258 memcpy(MLX5_ADDR_OF(sw_tls_cntx, ctx, param.gcm_iv), 259 en->iv, en->iv_len); 260 break; 261 default: 262 return (EINVAL); 263 } 264 265 if (en->cipher_key_len <= MLX5_FLD_SZ_BYTES(sw_tls_cntx, key.key_data)) { 266 memcpy(MLX5_ADDR_OF(sw_tls_cntx, ctx, key.key_data), 267 en->cipher_key, en->cipher_key_len); 268 MLX5_SET(sw_tls_cntx, ctx, key.key_len, en->cipher_key_len); 269 } else { 270 return (EINVAL); 271 } 272 return (0); 273 } 274 275 /* Verify zero default */ 276 CTASSERT(MLX5E_TLS_ST_INIT == 0); 277 278 int 279 mlx5e_tls_snd_tag_alloc(struct ifnet *ifp, 280 union if_snd_tag_alloc_params *params, 281 struct m_snd_tag **ppmt) 282 { 283 union if_snd_tag_alloc_params rl_params; 284 struct mlx5e_priv *priv; 285 struct mlx5e_tls_tag *ptag; 286 const struct tls_session_params *en; 287 int error; 288 289 priv = ifp->if_softc; 290 291 if (priv->gone != 0 || priv->tls.init == 0) 292 return (EOPNOTSUPP); 293 294 /* allocate new tag from zone, if any */ 295 ptag = uma_zalloc(priv->tls.zone, M_NOWAIT); 296 if (ptag == NULL) 297 return (ENOMEM); 298 299 /* sanity check default values */ 300 MPASS(ptag->state == MLX5E_TLS_ST_INIT); 301 MPASS(ptag->dek_index == 0); 302 MPASS(ptag->dek_index_ok == 0); 303 304 /* setup TLS tag */ 305 ptag->tls = &priv->tls; 306 307 /* check if there is no TIS context */ 308 if (ptag->tisn == 0) { 309 uint32_t value; 310 311 value = atomic_fetchadd_32(&priv->tls.num_resources, 1U); 312 313 /* check resource limits */ 314 if (value >= priv->tls.max_resources) { 315 error = ENOMEM; 316 goto failure; 317 } 318 } 319 320 en = ¶ms->tls.tls->params; 321 322 /* only TLS v1.2 and v1.3 is currently supported */ 323 if (en->tls_vmajor != TLS_MAJOR_VER_ONE || 324 (en->tls_vminor != TLS_MINOR_VER_TWO 325 #ifdef TLS_MINOR_VER_THREE 326 && en->tls_vminor != TLS_MINOR_VER_THREE 327 #endif 328 )) { 329 error = EPROTONOSUPPORT; 330 goto failure; 331 } 332 333 switch (en->cipher_algorithm) { 334 case CRYPTO_AES_NIST_GCM_16: 335 switch (en->cipher_key_len) { 336 case 128 / 8: 337 if (en->tls_vminor == TLS_MINOR_VER_TWO) { 338 if (MLX5_CAP_TLS(priv->mdev, tls_1_2_aes_gcm_128) == 0) { 339 error = EPROTONOSUPPORT; 340 goto failure; 341 } 342 } else { 343 if (MLX5_CAP_TLS(priv->mdev, tls_1_3_aes_gcm_128) == 0) { 344 error = EPROTONOSUPPORT; 345 goto failure; 346 } 347 } 348 error = mlx5e_tls_set_params(ptag->crypto_params, en); 349 if (error) 350 goto failure; 351 break; 352 353 case 256 / 8: 354 if (en->tls_vminor == TLS_MINOR_VER_TWO) { 355 if (MLX5_CAP_TLS(priv->mdev, tls_1_2_aes_gcm_256) == 0) { 356 error = EPROTONOSUPPORT; 357 goto failure; 358 } 359 } else { 360 if (MLX5_CAP_TLS(priv->mdev, tls_1_3_aes_gcm_256) == 0) { 361 error = EPROTONOSUPPORT; 362 goto failure; 363 } 364 } 365 error = mlx5e_tls_set_params(ptag->crypto_params, en); 366 if (error) 367 goto failure; 368 break; 369 370 default: 371 error = EINVAL; 372 goto failure; 373 } 374 break; 375 default: 376 error = EPROTONOSUPPORT; 377 goto failure; 378 } 379 380 memset(&rl_params, 0, sizeof(rl_params)); 381 rl_params.hdr = params->hdr; 382 switch (params->hdr.type) { 383 #ifdef RATELIMIT 384 case IF_SND_TAG_TYPE_TLS_RATE_LIMIT: 385 rl_params.hdr.type = IF_SND_TAG_TYPE_RATE_LIMIT; 386 rl_params.rate_limit.max_rate = params->tls_rate_limit.max_rate; 387 break; 388 #endif 389 case IF_SND_TAG_TYPE_TLS: 390 rl_params.hdr.type = IF_SND_TAG_TYPE_UNLIMITED; 391 break; 392 default: 393 error = EOPNOTSUPP; 394 goto failure; 395 } 396 397 error = m_snd_tag_alloc(ifp, &rl_params, &ptag->rl_tag); 398 if (error) 399 goto failure; 400 401 /* store pointer to mbuf tag */ 402 MPASS(ptag->tag.refcount == 0); 403 m_snd_tag_init(&ptag->tag, ifp, params->hdr.type); 404 *ppmt = &ptag->tag; 405 406 queue_work(priv->tls.wq, &ptag->work); 407 flush_work(&ptag->work); 408 409 return (0); 410 411 failure: 412 mlx5e_tls_tag_zfree(ptag); 413 return (error); 414 } 415 416 int 417 mlx5e_tls_snd_tag_modify(struct m_snd_tag *pmt, union if_snd_tag_modify_params *params) 418 { 419 #ifdef RATELIMIT 420 union if_snd_tag_modify_params rl_params; 421 struct mlx5e_tls_tag *ptag = 422 container_of(pmt, struct mlx5e_tls_tag, tag); 423 int error; 424 #endif 425 426 switch (pmt->type) { 427 #ifdef RATELIMIT 428 case IF_SND_TAG_TYPE_TLS_RATE_LIMIT: 429 memset(&rl_params, 0, sizeof(rl_params)); 430 rl_params.rate_limit.max_rate = params->tls_rate_limit.max_rate; 431 error = ptag->rl_tag->ifp->if_snd_tag_modify(ptag->rl_tag, 432 &rl_params); 433 return (error); 434 #endif 435 default: 436 return (EOPNOTSUPP); 437 } 438 } 439 440 int 441 mlx5e_tls_snd_tag_query(struct m_snd_tag *pmt, union if_snd_tag_query_params *params) 442 { 443 struct mlx5e_tls_tag *ptag = 444 container_of(pmt, struct mlx5e_tls_tag, tag); 445 int error; 446 447 switch (pmt->type) { 448 #ifdef RATELIMIT 449 case IF_SND_TAG_TYPE_TLS_RATE_LIMIT: 450 #endif 451 case IF_SND_TAG_TYPE_TLS: 452 error = ptag->rl_tag->ifp->if_snd_tag_query(ptag->rl_tag, 453 params); 454 break; 455 default: 456 error = EOPNOTSUPP; 457 break; 458 } 459 return (error); 460 } 461 462 void 463 mlx5e_tls_snd_tag_free(struct m_snd_tag *pmt) 464 { 465 struct mlx5e_tls_tag *ptag = 466 container_of(pmt, struct mlx5e_tls_tag, tag); 467 struct mlx5e_priv *priv; 468 469 m_snd_tag_rele(ptag->rl_tag); 470 471 MLX5E_TLS_TAG_LOCK(ptag); 472 ptag->state = MLX5E_TLS_ST_FREED; 473 MLX5E_TLS_TAG_UNLOCK(ptag); 474 475 priv = ptag->tag.ifp->if_softc; 476 queue_work(priv->tls.wq, &ptag->work); 477 } 478 479 CTASSERT((MLX5_FLD_SZ_BYTES(sw_tls_cntx, param) % 16) == 0); 480 481 static void 482 mlx5e_tls_send_static_parameters(struct mlx5e_sq *sq, struct mlx5e_tls_tag *ptag) 483 { 484 const u32 ds_cnt = DIV_ROUND_UP(sizeof(struct mlx5e_tx_umr_wqe) + 485 MLX5_FLD_SZ_BYTES(sw_tls_cntx, param), MLX5_SEND_WQE_DS); 486 struct mlx5e_tx_umr_wqe *wqe; 487 u16 pi; 488 489 pi = sq->pc & sq->wq.sz_m1; 490 wqe = mlx5_wq_cyc_get_wqe(&sq->wq, pi); 491 492 memset(wqe, 0, sizeof(*wqe)); 493 494 wqe->ctrl.opmod_idx_opcode = cpu_to_be32((sq->pc << 8) | 495 MLX5_OPCODE_UMR | (MLX5_OPCODE_MOD_UMR_TLS_TIS_STATIC_PARAMS << 24)); 496 wqe->ctrl.qpn_ds = cpu_to_be32((sq->sqn << 8) | ds_cnt); 497 wqe->ctrl.imm = cpu_to_be32(ptag->tisn << 8); 498 499 if (mlx5e_do_send_cqe(sq)) 500 wqe->ctrl.fm_ce_se = MLX5_WQE_CTRL_CQ_UPDATE | MLX5_FENCE_MODE_INITIATOR_SMALL; 501 else 502 wqe->ctrl.fm_ce_se = MLX5_FENCE_MODE_INITIATOR_SMALL; 503 504 /* fill out UMR control segment */ 505 wqe->umr.flags = 0x80; /* inline data */ 506 wqe->umr.bsf_octowords = cpu_to_be16(MLX5_FLD_SZ_BYTES(sw_tls_cntx, param) / 16); 507 508 /* copy in the static crypto parameters */ 509 memcpy(wqe + 1, MLX5_ADDR_OF(sw_tls_cntx, ptag->crypto_params, param), 510 MLX5_FLD_SZ_BYTES(sw_tls_cntx, param)); 511 512 /* copy data for doorbell */ 513 memcpy(sq->doorbell.d32, &wqe->ctrl, sizeof(sq->doorbell.d32)); 514 515 sq->mbuf[pi].mbuf = NULL; 516 sq->mbuf[pi].num_bytes = 0; 517 sq->mbuf[pi].num_wqebbs = DIV_ROUND_UP(ds_cnt, MLX5_SEND_WQEBB_NUM_DS); 518 sq->mbuf[pi].p_refcount = &ptag->refs; 519 atomic_add_int(&ptag->refs, 1); 520 sq->pc += sq->mbuf[pi].num_wqebbs; 521 } 522 523 CTASSERT(MLX5_FLD_SZ_BYTES(sw_tls_cntx, progress) == 524 sizeof(((struct mlx5e_tx_psv_wqe *)0)->psv)); 525 526 static void 527 mlx5e_tls_send_progress_parameters(struct mlx5e_sq *sq, struct mlx5e_tls_tag *ptag) 528 { 529 const u32 ds_cnt = DIV_ROUND_UP(sizeof(struct mlx5e_tx_psv_wqe), 530 MLX5_SEND_WQE_DS); 531 struct mlx5e_tx_psv_wqe *wqe; 532 u16 pi; 533 534 pi = sq->pc & sq->wq.sz_m1; 535 wqe = mlx5_wq_cyc_get_wqe(&sq->wq, pi); 536 537 memset(wqe, 0, sizeof(*wqe)); 538 539 wqe->ctrl.opmod_idx_opcode = cpu_to_be32((sq->pc << 8) | 540 MLX5_OPCODE_SET_PSV | (MLX5_OPCODE_MOD_PSV_TLS_TIS_PROGRESS_PARAMS << 24)); 541 wqe->ctrl.qpn_ds = cpu_to_be32((sq->sqn << 8) | ds_cnt); 542 543 if (mlx5e_do_send_cqe(sq)) 544 wqe->ctrl.fm_ce_se = MLX5_WQE_CTRL_CQ_UPDATE; 545 546 /* copy in the PSV control segment */ 547 memcpy(&wqe->psv, MLX5_ADDR_OF(sw_tls_cntx, ptag->crypto_params, progress), 548 sizeof(wqe->psv)); 549 550 /* copy data for doorbell */ 551 memcpy(sq->doorbell.d32, &wqe->ctrl, sizeof(sq->doorbell.d32)); 552 553 sq->mbuf[pi].mbuf = NULL; 554 sq->mbuf[pi].num_bytes = 0; 555 sq->mbuf[pi].num_wqebbs = DIV_ROUND_UP(ds_cnt, MLX5_SEND_WQEBB_NUM_DS); 556 sq->mbuf[pi].p_refcount = &ptag->refs; 557 atomic_add_int(&ptag->refs, 1); 558 sq->pc += sq->mbuf[pi].num_wqebbs; 559 } 560 561 static void 562 mlx5e_tls_send_nop(struct mlx5e_sq *sq, struct mlx5e_tls_tag *ptag) 563 { 564 const u32 ds_cnt = MLX5_SEND_WQEBB_NUM_DS; 565 struct mlx5e_tx_wqe *wqe; 566 u16 pi; 567 568 pi = sq->pc & sq->wq.sz_m1; 569 wqe = mlx5_wq_cyc_get_wqe(&sq->wq, pi); 570 571 memset(&wqe->ctrl, 0, sizeof(wqe->ctrl)); 572 573 wqe->ctrl.opmod_idx_opcode = cpu_to_be32((sq->pc << 8) | MLX5_OPCODE_NOP); 574 wqe->ctrl.qpn_ds = cpu_to_be32((sq->sqn << 8) | ds_cnt); 575 if (mlx5e_do_send_cqe(sq)) 576 wqe->ctrl.fm_ce_se = MLX5_WQE_CTRL_CQ_UPDATE | MLX5_FENCE_MODE_INITIATOR_SMALL; 577 else 578 wqe->ctrl.fm_ce_se = MLX5_FENCE_MODE_INITIATOR_SMALL; 579 580 /* Copy data for doorbell */ 581 memcpy(sq->doorbell.d32, &wqe->ctrl, sizeof(sq->doorbell.d32)); 582 583 sq->mbuf[pi].mbuf = NULL; 584 sq->mbuf[pi].num_bytes = 0; 585 sq->mbuf[pi].num_wqebbs = DIV_ROUND_UP(ds_cnt, MLX5_SEND_WQEBB_NUM_DS); 586 sq->mbuf[pi].p_refcount = &ptag->refs; 587 atomic_add_int(&ptag->refs, 1); 588 sq->pc += sq->mbuf[pi].num_wqebbs; 589 } 590 591 #define SBTLS_MBUF_NO_DATA ((struct mbuf *)1) 592 593 static struct mbuf * 594 sbtls_recover_record(struct mbuf *mb, int wait, uint32_t tcp_old, uint32_t *ptcp_seq, bool *pis_start) 595 { 596 struct mbuf *mr, *top; 597 uint32_t offset; 598 uint32_t delta; 599 600 /* check format of incoming mbuf */ 601 if (mb->m_next == NULL || 602 (mb->m_next->m_flags & (M_EXTPG | M_EXT)) != (M_EXTPG | M_EXT)) { 603 top = NULL; 604 goto done; 605 } 606 607 /* get unmapped data offset */ 608 offset = mtod(mb->m_next, uintptr_t); 609 610 /* check if we don't need to re-transmit anything */ 611 if (offset == 0) { 612 top = SBTLS_MBUF_NO_DATA; 613 *pis_start = true; 614 goto done; 615 } 616 617 /* try to get a new packet header */ 618 top = m_gethdr(wait, MT_DATA); 619 if (top == NULL) 620 goto done; 621 622 mr = m_get(wait, MT_DATA); 623 if (mr == NULL) { 624 m_free(top); 625 top = NULL; 626 goto done; 627 } 628 629 top->m_next = mr; 630 631 mb_dupcl(mr, mb->m_next); 632 633 /* the beginning of the TLS record */ 634 mr->m_data = NULL; 635 636 /* setup packet header length */ 637 top->m_pkthdr.len = mr->m_len = offset; 638 top->m_len = 0; 639 640 /* check for partial re-transmit */ 641 delta = *ptcp_seq - tcp_old; 642 643 if (delta < offset) { 644 m_adj(top, offset - delta); 645 offset = delta; 646 647 /* continue where we left off */ 648 *pis_start = false; 649 } else { 650 *pis_start = true; 651 } 652 653 /* 654 * Rewind the TCP sequence number by the amount of data 655 * retransmitted: 656 */ 657 *ptcp_seq -= offset; 658 done: 659 return (top); 660 } 661 662 static int 663 mlx5e_sq_tls_populate(struct mbuf *mb, uint64_t *pseq) 664 { 665 666 for (; mb != NULL; mb = mb->m_next) { 667 if (!(mb->m_flags & M_EXTPG)) 668 continue; 669 *pseq = mb->m_epg_seqno; 670 return (1); 671 } 672 return (0); 673 } 674 675 int 676 mlx5e_sq_tls_xmit(struct mlx5e_sq *sq, struct mlx5e_xmit_args *parg, struct mbuf **ppmb) 677 { 678 struct mlx5e_tls_tag *ptls_tag; 679 struct m_snd_tag *ptag; 680 const struct tcphdr *th; 681 struct mbuf *mb = *ppmb; 682 u64 rcd_sn; 683 u32 header_size; 684 u32 mb_seq; 685 686 if ((mb->m_pkthdr.csum_flags & CSUM_SND_TAG) == 0) 687 return (MLX5E_TLS_CONTINUE); 688 689 ptag = mb->m_pkthdr.snd_tag; 690 691 if ( 692 #ifdef RATELIMIT 693 ptag->type != IF_SND_TAG_TYPE_TLS_RATE_LIMIT && 694 #endif 695 ptag->type != IF_SND_TAG_TYPE_TLS) 696 return (MLX5E_TLS_CONTINUE); 697 698 ptls_tag = container_of(ptag, struct mlx5e_tls_tag, tag); 699 700 header_size = mlx5e_get_full_header_size(mb, &th); 701 if (unlikely(header_size == 0 || th == NULL)) 702 return (MLX5E_TLS_FAILURE); 703 704 /* 705 * Send non-TLS TCP packets AS-IS: 706 */ 707 if (header_size == mb->m_pkthdr.len || 708 mlx5e_sq_tls_populate(mb, &rcd_sn) == 0) { 709 parg->tisn = 0; 710 parg->ihs = header_size; 711 return (MLX5E_TLS_CONTINUE); 712 } 713 714 mb_seq = ntohl(th->th_seq); 715 716 MLX5E_TLS_TAG_LOCK(ptls_tag); 717 switch (ptls_tag->state) { 718 case MLX5E_TLS_ST_INIT: 719 MLX5E_TLS_TAG_UNLOCK(ptls_tag); 720 return (MLX5E_TLS_FAILURE); 721 case MLX5E_TLS_ST_SETUP: 722 ptls_tag->state = MLX5E_TLS_ST_TXRDY; 723 ptls_tag->expected_seq = ~mb_seq; /* force setup */ 724 default: 725 MLX5E_TLS_TAG_UNLOCK(ptls_tag); 726 break; 727 } 728 729 if (unlikely(ptls_tag->expected_seq != mb_seq)) { 730 bool is_start; 731 struct mbuf *r_mb; 732 uint32_t tcp_seq = mb_seq; 733 734 r_mb = sbtls_recover_record(mb, M_NOWAIT, ptls_tag->expected_seq, &tcp_seq, &is_start); 735 if (r_mb == NULL) { 736 MLX5E_TLS_STAT_INC(ptls_tag, tx_error, 1); 737 return (MLX5E_TLS_FAILURE); 738 } 739 740 MLX5E_TLS_STAT_INC(ptls_tag, tx_packets_ooo, 1); 741 742 /* check if this is the first fragment of a TLS record */ 743 if (is_start) { 744 /* setup TLS static parameters */ 745 MLX5_SET64(sw_tls_cntx, ptls_tag->crypto_params, 746 param.initial_record_number, rcd_sn); 747 748 /* 749 * NOTE: The sendqueue should have enough room to 750 * carry both the static and the progress parameters 751 * when we get here! 752 */ 753 mlx5e_tls_send_static_parameters(sq, ptls_tag); 754 mlx5e_tls_send_progress_parameters(sq, ptls_tag); 755 756 if (r_mb == SBTLS_MBUF_NO_DATA) { 757 mlx5e_tls_send_nop(sq, ptls_tag); 758 ptls_tag->expected_seq = mb_seq; 759 return (MLX5E_TLS_LOOP); 760 } 761 } 762 763 MLX5E_TLS_STAT_INC(ptls_tag, tx_bytes_ooo, r_mb->m_pkthdr.len); 764 765 /* setup transmit arguments */ 766 parg->tisn = ptls_tag->tisn; 767 parg->pref = &ptls_tag->refs; 768 769 /* try to send DUMP data */ 770 if (mlx5e_sq_dump_xmit(sq, parg, &r_mb) != 0) { 771 m_freem(r_mb); 772 ptls_tag->expected_seq = tcp_seq; 773 return (MLX5E_TLS_FAILURE); 774 } else { 775 ptls_tag->expected_seq = mb_seq; 776 return (MLX5E_TLS_LOOP); 777 } 778 } else { 779 MLX5E_TLS_STAT_INC(ptls_tag, tx_packets, 1); 780 MLX5E_TLS_STAT_INC(ptls_tag, tx_bytes, mb->m_pkthdr.len); 781 } 782 ptls_tag->expected_seq += mb->m_pkthdr.len - header_size; 783 784 parg->tisn = ptls_tag->tisn; 785 parg->ihs = header_size; 786 parg->pref = &ptls_tag->refs; 787 return (MLX5E_TLS_CONTINUE); 788 } 789 790 #else 791 792 int 793 mlx5e_tls_init(struct mlx5e_priv *priv) 794 { 795 796 return (0); 797 } 798 799 void 800 mlx5e_tls_cleanup(struct mlx5e_priv *priv) 801 { 802 /* NOP */ 803 } 804 805 #endif /* KERN_TLS */ 806