examples/ipsec-secgw: add target queues in flow actions
[dpdk.git] / examples / ipsec-secgw / ipsec.c
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright(c) 2016-2017 Intel Corporation
3  */
4 #include <sys/types.h>
5 #include <netinet/in.h>
6 #include <netinet/ip.h>
7
8 #include <rte_branch_prediction.h>
9 #include <rte_log.h>
10 #include <rte_crypto.h>
11 #include <rte_security.h>
12 #include <rte_cryptodev.h>
13 #include <rte_ethdev.h>
14 #include <rte_mbuf.h>
15 #include <rte_hash.h>
16
17 #include "ipsec.h"
18 #include "esp.h"
19
20 static inline void
21 set_ipsec_conf(struct ipsec_sa *sa, struct rte_security_ipsec_xform *ipsec)
22 {
23         if (ipsec->mode == RTE_SECURITY_IPSEC_SA_MODE_TUNNEL) {
24                 struct rte_security_ipsec_tunnel_param *tunnel =
25                                 &ipsec->tunnel;
26                 if (sa->flags == IP4_TUNNEL) {
27                         tunnel->type =
28                                 RTE_SECURITY_IPSEC_TUNNEL_IPV4;
29                         tunnel->ipv4.ttl = IPDEFTTL;
30
31                         memcpy((uint8_t *)&tunnel->ipv4.src_ip,
32                                 (uint8_t *)&sa->src.ip.ip4, 4);
33
34                         memcpy((uint8_t *)&tunnel->ipv4.dst_ip,
35                                 (uint8_t *)&sa->dst.ip.ip4, 4);
36                 }
37                 /* TODO support for Transport and IPV6 tunnel */
38         }
39 }
40
41 static inline int
42 create_session(struct ipsec_ctx *ipsec_ctx, struct ipsec_sa *sa)
43 {
44         struct rte_cryptodev_info cdev_info;
45         unsigned long cdev_id_qp = 0;
46         int32_t ret = 0;
47         struct cdev_key key = { 0 };
48
49         key.lcore_id = (uint8_t)rte_lcore_id();
50
51         key.cipher_algo = (uint8_t)sa->cipher_algo;
52         key.auth_algo = (uint8_t)sa->auth_algo;
53         key.aead_algo = (uint8_t)sa->aead_algo;
54
55         if (sa->type == RTE_SECURITY_ACTION_TYPE_NONE) {
56                 ret = rte_hash_lookup_data(ipsec_ctx->cdev_map, &key,
57                                 (void **)&cdev_id_qp);
58                 if (ret < 0) {
59                         RTE_LOG(ERR, IPSEC,
60                                 "No cryptodev: core %u, cipher_algo %u, "
61                                 "auth_algo %u, aead_algo %u\n",
62                                 key.lcore_id,
63                                 key.cipher_algo,
64                                 key.auth_algo,
65                                 key.aead_algo);
66                         return -1;
67                 }
68         }
69
70         RTE_LOG_DP(DEBUG, IPSEC, "Create session for SA spi %u on cryptodev "
71                         "%u qp %u\n", sa->spi,
72                         ipsec_ctx->tbl[cdev_id_qp].id,
73                         ipsec_ctx->tbl[cdev_id_qp].qp);
74
75         if (sa->type != RTE_SECURITY_ACTION_TYPE_NONE) {
76                 struct rte_security_session_conf sess_conf = {
77                         .action_type = sa->type,
78                         .protocol = RTE_SECURITY_PROTOCOL_IPSEC,
79                         {.ipsec = {
80                                 .spi = sa->spi,
81                                 .salt = sa->salt,
82                                 .options = { 0 },
83                                 .direction = sa->direction,
84                                 .proto = RTE_SECURITY_IPSEC_SA_PROTO_ESP,
85                                 .mode = (sa->flags == IP4_TUNNEL ||
86                                                 sa->flags == IP6_TUNNEL) ?
87                                         RTE_SECURITY_IPSEC_SA_MODE_TUNNEL :
88                                         RTE_SECURITY_IPSEC_SA_MODE_TRANSPORT,
89                         } },
90                         .crypto_xform = sa->xforms,
91                         .userdata = NULL,
92
93                 };
94
95                 if (sa->type == RTE_SECURITY_ACTION_TYPE_LOOKASIDE_PROTOCOL) {
96                         struct rte_security_ctx *ctx = (struct rte_security_ctx *)
97                                                         rte_cryptodev_get_sec_ctx(
98                                                         ipsec_ctx->tbl[cdev_id_qp].id);
99
100                         /* Set IPsec parameters in conf */
101                         set_ipsec_conf(sa, &(sess_conf.ipsec));
102
103                         sa->sec_session = rte_security_session_create(ctx,
104                                         &sess_conf, ipsec_ctx->session_pool);
105                         if (sa->sec_session == NULL) {
106                                 RTE_LOG(ERR, IPSEC,
107                                 "SEC Session init failed: err: %d\n", ret);
108                                 return -1;
109                         }
110                 } else if (sa->type == RTE_SECURITY_ACTION_TYPE_INLINE_CRYPTO) {
111                         struct rte_flow_error err;
112                         struct rte_security_ctx *ctx = (struct rte_security_ctx *)
113                                                         rte_eth_dev_get_sec_ctx(
114                                                         sa->portid);
115                         const struct rte_security_capability *sec_cap;
116                         int ret = 0;
117
118                         sa->sec_session = rte_security_session_create(ctx,
119                                         &sess_conf, ipsec_ctx->session_pool);
120                         if (sa->sec_session == NULL) {
121                                 RTE_LOG(ERR, IPSEC,
122                                 "SEC Session init failed: err: %d\n", ret);
123                                 return -1;
124                         }
125
126                         sec_cap = rte_security_capabilities_get(ctx);
127
128                         /* iterate until ESP tunnel*/
129                         while (sec_cap->action !=
130                                         RTE_SECURITY_ACTION_TYPE_NONE) {
131
132                                 if (sec_cap->action == sa->type &&
133                                     sec_cap->protocol ==
134                                         RTE_SECURITY_PROTOCOL_IPSEC &&
135                                     sec_cap->ipsec.mode ==
136                                         RTE_SECURITY_IPSEC_SA_MODE_TUNNEL &&
137                                     sec_cap->ipsec.direction == sa->direction)
138                                         break;
139                                 sec_cap++;
140                         }
141
142                         if (sec_cap->action == RTE_SECURITY_ACTION_TYPE_NONE) {
143                                 RTE_LOG(ERR, IPSEC,
144                                 "No suitable security capability found\n");
145                                 return -1;
146                         }
147
148                         sa->ol_flags = sec_cap->ol_flags;
149                         sa->security_ctx = ctx;
150                         sa->pattern[0].type = RTE_FLOW_ITEM_TYPE_ETH;
151
152                         sa->pattern[1].type = RTE_FLOW_ITEM_TYPE_IPV4;
153                         sa->pattern[1].mask = &rte_flow_item_ipv4_mask;
154                         if (sa->flags & IP6_TUNNEL) {
155                                 sa->pattern[1].spec = &sa->ipv6_spec;
156                                 memcpy(sa->ipv6_spec.hdr.dst_addr,
157                                         sa->dst.ip.ip6.ip6_b, 16);
158                                 memcpy(sa->ipv6_spec.hdr.src_addr,
159                                        sa->src.ip.ip6.ip6_b, 16);
160                         } else {
161                                 sa->pattern[1].spec = &sa->ipv4_spec;
162                                 sa->ipv4_spec.hdr.dst_addr = sa->dst.ip.ip4;
163                                 sa->ipv4_spec.hdr.src_addr = sa->src.ip.ip4;
164                         }
165
166                         sa->pattern[2].type = RTE_FLOW_ITEM_TYPE_ESP;
167                         sa->pattern[2].spec = &sa->esp_spec;
168                         sa->pattern[2].mask = &rte_flow_item_esp_mask;
169                         sa->esp_spec.hdr.spi = sa->spi;
170
171                         sa->pattern[3].type = RTE_FLOW_ITEM_TYPE_END;
172
173                         sa->action[0].type = RTE_FLOW_ACTION_TYPE_SECURITY;
174                         sa->action[0].conf = sa->sec_session;
175
176                         sa->action[1].type = RTE_FLOW_ACTION_TYPE_END;
177
178                         sa->attr.egress = (sa->direction ==
179                                         RTE_SECURITY_IPSEC_SA_DIR_EGRESS);
180                         sa->attr.ingress = (sa->direction ==
181                                         RTE_SECURITY_IPSEC_SA_DIR_INGRESS);
182                         if (sa->attr.ingress) {
183                                 uint8_t rss_key[40];
184                                 struct rte_eth_rss_conf rss_conf = {
185                                         .rss_key = rss_key,
186                                         .rss_key_len = 40,
187                                 };
188                                 struct rte_eth_dev *eth_dev;
189                                 union {
190                                         struct rte_flow_action_rss rss;
191                                         struct {
192                                         const struct rte_eth_rss_conf *rss_conf;
193                                         uint16_t num;
194                                         uint16_t queue[RTE_MAX_QUEUES_PER_PORT];
195                                         } local;
196                                 } action_rss;
197                                 unsigned int i;
198                                 unsigned int j;
199
200                                 sa->action[2].type = RTE_FLOW_ACTION_TYPE_END;
201                                 /* Try RSS. */
202                                 sa->action[1].type = RTE_FLOW_ACTION_TYPE_RSS;
203                                 sa->action[1].conf = &action_rss;
204                                 eth_dev = ctx->device;
205                                 rte_eth_dev_rss_hash_conf_get(sa->portid,
206                                                               &rss_conf);
207                                 for (i = 0, j = 0;
208                                      i < eth_dev->data->nb_rx_queues; ++i)
209                                         if (eth_dev->data->rx_queues[i])
210                                                 action_rss.local.queue[j++] = i;
211                                 action_rss.local.num = j;
212                                 action_rss.local.rss_conf = &rss_conf;
213                                 ret = rte_flow_validate(sa->portid, &sa->attr,
214                                                         sa->pattern, sa->action,
215                                                         &err);
216                                 if (!ret)
217                                         goto flow_create;
218                                 /* Try Queue. */
219                                 sa->action[1].type = RTE_FLOW_ACTION_TYPE_QUEUE;
220                                 sa->action[1].conf =
221                                         &(struct rte_flow_action_queue){
222                                         .index = 0,
223                                 };
224                                 ret = rte_flow_validate(sa->portid, &sa->attr,
225                                                         sa->pattern, sa->action,
226                                                         &err);
227                                 if (ret)
228                                         goto flow_create_failure;
229                         }
230 flow_create:
231                         sa->flow = rte_flow_create(sa->portid,
232                                 &sa->attr, sa->pattern, sa->action, &err);
233                         if (sa->flow == NULL) {
234 flow_create_failure:
235                                 RTE_LOG(ERR, IPSEC,
236                                         "Failed to create ipsec flow msg: %s\n",
237                                         err.message);
238                                 return -1;
239                         }
240                 } else if (sa->type ==
241                                 RTE_SECURITY_ACTION_TYPE_INLINE_PROTOCOL) {
242                         struct rte_security_ctx *ctx =
243                                         (struct rte_security_ctx *)
244                                         rte_eth_dev_get_sec_ctx(sa->portid);
245                         const struct rte_security_capability *sec_cap;
246
247                         if (ctx == NULL) {
248                                 RTE_LOG(ERR, IPSEC,
249                                 "Ethernet device doesn't have security features registered\n");
250                                 return -1;
251                         }
252
253                         /* Set IPsec parameters in conf */
254                         set_ipsec_conf(sa, &(sess_conf.ipsec));
255
256                         /* Save SA as userdata for the security session. When
257                          * the packet is received, this userdata will be
258                          * retrieved using the metadata from the packet.
259                          *
260                          * This is required only for inbound SAs.
261                          */
262
263                         if (sa->direction == RTE_SECURITY_IPSEC_SA_DIR_INGRESS)
264                                 sess_conf.userdata = (void *) sa;
265
266                         sa->sec_session = rte_security_session_create(ctx,
267                                         &sess_conf, ipsec_ctx->session_pool);
268                         if (sa->sec_session == NULL) {
269                                 RTE_LOG(ERR, IPSEC,
270                                 "SEC Session init failed: err: %d\n", ret);
271                                 return -1;
272                         }
273
274                         sec_cap = rte_security_capabilities_get(ctx);
275
276                         if (sec_cap == NULL) {
277                                 RTE_LOG(ERR, IPSEC,
278                                 "No capabilities registered\n");
279                                 return -1;
280                         }
281
282                         /* iterate until ESP tunnel*/
283                         while (sec_cap->action !=
284                                         RTE_SECURITY_ACTION_TYPE_NONE) {
285
286                                 if (sec_cap->action == sa->type &&
287                                     sec_cap->protocol ==
288                                         RTE_SECURITY_PROTOCOL_IPSEC &&
289                                     sec_cap->ipsec.mode ==
290                                         RTE_SECURITY_IPSEC_SA_MODE_TUNNEL &&
291                                     sec_cap->ipsec.direction == sa->direction)
292                                         break;
293                                 sec_cap++;
294                         }
295
296                         if (sec_cap->action == RTE_SECURITY_ACTION_TYPE_NONE) {
297                                 RTE_LOG(ERR, IPSEC,
298                                 "No suitable security capability found\n");
299                                 return -1;
300                         }
301
302                         sa->ol_flags = sec_cap->ol_flags;
303                         sa->security_ctx = ctx;
304                 }
305         } else {
306                 sa->crypto_session = rte_cryptodev_sym_session_create(
307                                 ipsec_ctx->session_pool);
308                 rte_cryptodev_sym_session_init(ipsec_ctx->tbl[cdev_id_qp].id,
309                                 sa->crypto_session, sa->xforms,
310                                 ipsec_ctx->session_pool);
311
312                 rte_cryptodev_info_get(ipsec_ctx->tbl[cdev_id_qp].id,
313                                 &cdev_info);
314                 if (cdev_info.sym.max_nb_sessions_per_qp > 0) {
315                         ret = rte_cryptodev_queue_pair_attach_sym_session(
316                                         ipsec_ctx->tbl[cdev_id_qp].id,
317                                         ipsec_ctx->tbl[cdev_id_qp].qp,
318                                         sa->crypto_session);
319                         if (ret < 0) {
320                                 RTE_LOG(ERR, IPSEC,
321                                         "Session cannot be attached to qp %u\n",
322                                         ipsec_ctx->tbl[cdev_id_qp].qp);
323                                 return -1;
324                         }
325                 }
326         }
327         sa->cdev_id_qp = cdev_id_qp;
328
329         return 0;
330 }
331
332 static inline void
333 enqueue_cop(struct cdev_qp *cqp, struct rte_crypto_op *cop)
334 {
335         int32_t ret, i;
336
337         cqp->buf[cqp->len++] = cop;
338
339         if (cqp->len == MAX_PKT_BURST) {
340                 ret = rte_cryptodev_enqueue_burst(cqp->id, cqp->qp,
341                                 cqp->buf, cqp->len);
342                 if (ret < cqp->len) {
343                         RTE_LOG_DP(DEBUG, IPSEC, "Cryptodev %u queue %u:"
344                                         " enqueued %u crypto ops out of %u\n",
345                                          cqp->id, cqp->qp,
346                                          ret, cqp->len);
347                         for (i = ret; i < cqp->len; i++)
348                                 rte_pktmbuf_free(cqp->buf[i]->sym->m_src);
349                 }
350                 cqp->in_flight += ret;
351                 cqp->len = 0;
352         }
353 }
354
355 static inline void
356 ipsec_enqueue(ipsec_xform_fn xform_func, struct ipsec_ctx *ipsec_ctx,
357                 struct rte_mbuf *pkts[], struct ipsec_sa *sas[],
358                 uint16_t nb_pkts)
359 {
360         int32_t ret = 0, i;
361         struct ipsec_mbuf_metadata *priv;
362         struct rte_crypto_sym_op *sym_cop;
363         struct ipsec_sa *sa;
364         struct cdev_qp *cqp;
365
366         for (i = 0; i < nb_pkts; i++) {
367                 if (unlikely(sas[i] == NULL)) {
368                         rte_pktmbuf_free(pkts[i]);
369                         continue;
370                 }
371
372                 rte_prefetch0(sas[i]);
373                 rte_prefetch0(pkts[i]);
374
375                 priv = get_priv(pkts[i]);
376                 sa = sas[i];
377                 priv->sa = sa;
378
379                 switch (sa->type) {
380                 case RTE_SECURITY_ACTION_TYPE_LOOKASIDE_PROTOCOL:
381                         priv->cop.type = RTE_CRYPTO_OP_TYPE_SYMMETRIC;
382                         priv->cop.status = RTE_CRYPTO_OP_STATUS_NOT_PROCESSED;
383
384                         rte_prefetch0(&priv->sym_cop);
385
386                         if ((unlikely(sa->sec_session == NULL)) &&
387                                         create_session(ipsec_ctx, sa)) {
388                                 rte_pktmbuf_free(pkts[i]);
389                                 continue;
390                         }
391
392                         sym_cop = get_sym_cop(&priv->cop);
393                         sym_cop->m_src = pkts[i];
394
395                         rte_security_attach_session(&priv->cop,
396                                         sa->sec_session);
397                         break;
398                 case RTE_SECURITY_ACTION_TYPE_NONE:
399
400                         priv->cop.type = RTE_CRYPTO_OP_TYPE_SYMMETRIC;
401                         priv->cop.status = RTE_CRYPTO_OP_STATUS_NOT_PROCESSED;
402
403                         rte_prefetch0(&priv->sym_cop);
404
405                         if ((unlikely(sa->crypto_session == NULL)) &&
406                                         create_session(ipsec_ctx, sa)) {
407                                 rte_pktmbuf_free(pkts[i]);
408                                 continue;
409                         }
410
411                         rte_crypto_op_attach_sym_session(&priv->cop,
412                                         sa->crypto_session);
413
414                         ret = xform_func(pkts[i], sa, &priv->cop);
415                         if (unlikely(ret)) {
416                                 rte_pktmbuf_free(pkts[i]);
417                                 continue;
418                         }
419                         break;
420                 case RTE_SECURITY_ACTION_TYPE_INLINE_PROTOCOL:
421                         if ((unlikely(sa->sec_session == NULL)) &&
422                                         create_session(ipsec_ctx, sa)) {
423                                 rte_pktmbuf_free(pkts[i]);
424                                 continue;
425                         }
426
427                         cqp = &ipsec_ctx->tbl[sa->cdev_id_qp];
428                         cqp->ol_pkts[cqp->ol_pkts_cnt++] = pkts[i];
429                         if (sa->ol_flags & RTE_SECURITY_TX_OLOAD_NEED_MDATA)
430                                 rte_security_set_pkt_metadata(
431                                                 sa->security_ctx,
432                                                 sa->sec_session, pkts[i], NULL);
433                         continue;
434                 case RTE_SECURITY_ACTION_TYPE_INLINE_CRYPTO:
435                         priv->cop.type = RTE_CRYPTO_OP_TYPE_SYMMETRIC;
436                         priv->cop.status = RTE_CRYPTO_OP_STATUS_NOT_PROCESSED;
437
438                         rte_prefetch0(&priv->sym_cop);
439
440                         if ((unlikely(sa->sec_session == NULL)) &&
441                                         create_session(ipsec_ctx, sa)) {
442                                 rte_pktmbuf_free(pkts[i]);
443                                 continue;
444                         }
445
446                         rte_security_attach_session(&priv->cop,
447                                         sa->sec_session);
448
449                         ret = xform_func(pkts[i], sa, &priv->cop);
450                         if (unlikely(ret)) {
451                                 rte_pktmbuf_free(pkts[i]);
452                                 continue;
453                         }
454
455                         cqp = &ipsec_ctx->tbl[sa->cdev_id_qp];
456                         cqp->ol_pkts[cqp->ol_pkts_cnt++] = pkts[i];
457                         if (sa->ol_flags & RTE_SECURITY_TX_OLOAD_NEED_MDATA)
458                                 rte_security_set_pkt_metadata(
459                                                 sa->security_ctx,
460                                                 sa->sec_session, pkts[i], NULL);
461                         continue;
462                 }
463
464                 RTE_ASSERT(sa->cdev_id_qp < ipsec_ctx->nb_qps);
465                 enqueue_cop(&ipsec_ctx->tbl[sa->cdev_id_qp], &priv->cop);
466         }
467 }
468
469 static inline int
470 ipsec_dequeue(ipsec_xform_fn xform_func, struct ipsec_ctx *ipsec_ctx,
471               struct rte_mbuf *pkts[], uint16_t max_pkts)
472 {
473         int32_t nb_pkts = 0, ret = 0, i, j, nb_cops;
474         struct ipsec_mbuf_metadata *priv;
475         struct rte_crypto_op *cops[max_pkts];
476         struct ipsec_sa *sa;
477         struct rte_mbuf *pkt;
478
479         for (i = 0; i < ipsec_ctx->nb_qps && nb_pkts < max_pkts;) {
480                 struct cdev_qp *cqp;
481
482                 cqp = &ipsec_ctx->tbl[ipsec_ctx->last_qp];
483
484                 while (cqp->ol_pkts_cnt > 0 && nb_pkts < max_pkts) {
485                         pkt = cqp->ol_pkts[--cqp->ol_pkts_cnt];
486                         rte_prefetch0(pkt);
487                         priv = get_priv(pkt);
488                         sa = priv->sa;
489                         ret = xform_func(pkt, sa, &priv->cop);
490                         if (unlikely(ret)) {
491                                 rte_pktmbuf_free(pkt);
492                                 continue;
493                         }
494                         pkts[nb_pkts++] = pkt;
495                 }
496
497                 if (cqp->in_flight == 0) {
498                         ipsec_ctx->last_qp++;
499                         if (ipsec_ctx->last_qp == ipsec_ctx->nb_qps)
500                                 ipsec_ctx->last_qp %= ipsec_ctx->nb_qps;
501                         i++;
502                         continue;
503                 }
504
505                 nb_cops = rte_cryptodev_dequeue_burst(cqp->id, cqp->qp,
506                                 cops, max_pkts - nb_pkts);
507
508                 cqp->in_flight -= nb_cops;
509
510                 for (j = 0; j < nb_cops; j++) {
511                         pkt = cops[j]->sym->m_src;
512                         rte_prefetch0(pkt);
513
514                         priv = get_priv(pkt);
515                         sa = priv->sa;
516
517                         RTE_ASSERT(sa != NULL);
518
519                         if (sa->type == RTE_SECURITY_ACTION_TYPE_NONE) {
520                                 ret = xform_func(pkt, sa, cops[j]);
521                                 if (unlikely(ret)) {
522                                         rte_pktmbuf_free(pkt);
523                                         continue;
524                                 }
525                         }
526                         pkts[nb_pkts++] = pkt;
527                         if (cqp->in_flight < max_pkts) {
528                                 ipsec_ctx->last_qp++;
529                                 if (ipsec_ctx->last_qp == ipsec_ctx->nb_qps)
530                                         ipsec_ctx->last_qp %= ipsec_ctx->nb_qps;
531                                 i++;
532                         }
533                 }
534         }
535
536         /* return packets */
537         return nb_pkts;
538 }
539
540 uint16_t
541 ipsec_inbound(struct ipsec_ctx *ctx, struct rte_mbuf *pkts[],
542                 uint16_t nb_pkts, uint16_t len)
543 {
544         struct ipsec_sa *sas[nb_pkts];
545
546         inbound_sa_lookup(ctx->sa_ctx, pkts, sas, nb_pkts);
547
548         ipsec_enqueue(esp_inbound, ctx, pkts, sas, nb_pkts);
549
550         return ipsec_dequeue(esp_inbound_post, ctx, pkts, len);
551 }
552
553 uint16_t
554 ipsec_outbound(struct ipsec_ctx *ctx, struct rte_mbuf *pkts[],
555                 uint32_t sa_idx[], uint16_t nb_pkts, uint16_t len)
556 {
557         struct ipsec_sa *sas[nb_pkts];
558
559         outbound_sa_lookup(ctx->sa_ctx, sa_idx, sas, nb_pkts);
560
561         ipsec_enqueue(esp_outbound, ctx, pkts, sas, nb_pkts);
562
563         return ipsec_dequeue(esp_outbound_post, ctx, pkts, len);
564 }