39624c493643e4e8a9478744909044dc4bf384e6
[deb_dpdk.git] / examples / ipsec-secgw / sa.c
1 /*-
2  *   BSD LICENSE
3  *
4  *   Copyright(c) 2016 Intel Corporation. All rights reserved.
5  *   All rights reserved.
6  *
7  *   Redistribution and use in source and binary forms, with or without
8  *   modification, are permitted provided that the following conditions
9  *   are met:
10  *
11  *     * Redistributions of source code must retain the above copyright
12  *       notice, this list of conditions and the following disclaimer.
13  *     * Redistributions in binary form must reproduce the above copyright
14  *       notice, this list of conditions and the following disclaimer in
15  *       the documentation and/or other materials provided with the
16  *       distribution.
17  *     * Neither the name of Intel Corporation nor the names of its
18  *       contributors may be used to endorse or promote products derived
19  *       from this software without specific prior written permission.
20  *
21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32  */
33
34 /*
35  * Security Associations
36  */
37 #include <sys/types.h>
38 #include <netinet/in.h>
39 #include <netinet/ip.h>
40 #include <netinet/ip6.h>
41
42 #include <rte_memzone.h>
43 #include <rte_crypto.h>
44 #include <rte_cryptodev.h>
45 #include <rte_byteorder.h>
46 #include <rte_errno.h>
47 #include <rte_ip.h>
48 #include <rte_random.h>
49
50 #include "ipsec.h"
51 #include "esp.h"
52 #include "parser.h"
53
54 struct supported_cipher_algo {
55         const char *keyword;
56         enum rte_crypto_cipher_algorithm algo;
57         uint16_t iv_len;
58         uint16_t block_size;
59         uint16_t key_len;
60 };
61
62 struct supported_auth_algo {
63         const char *keyword;
64         enum rte_crypto_auth_algorithm algo;
65         uint16_t digest_len;
66         uint16_t key_len;
67         uint8_t aad_len;
68         uint8_t key_not_req;
69 };
70
71 const struct supported_cipher_algo cipher_algos[] = {
72         {
73                 .keyword = "null",
74                 .algo = RTE_CRYPTO_CIPHER_NULL,
75                 .iv_len = 0,
76                 .block_size = 4,
77                 .key_len = 0
78         },
79         {
80                 .keyword = "aes-128-cbc",
81                 .algo = RTE_CRYPTO_CIPHER_AES_CBC,
82                 .iv_len = 16,
83                 .block_size = 16,
84                 .key_len = 16
85         },
86         {
87                 .keyword = "aes-128-gcm",
88                 .algo = RTE_CRYPTO_CIPHER_AES_GCM,
89                 .iv_len = 8,
90                 .block_size = 4,
91                 .key_len = 20
92         },
93         {
94                 .keyword = "aes-128-ctr",
95                 .algo = RTE_CRYPTO_CIPHER_AES_CTR,
96                 .iv_len = 8,
97                 .block_size = 16, /* XXX AESNI MB limition, should be 4 */
98                 .key_len = 20
99         }
100 };
101
102 const struct supported_auth_algo auth_algos[] = {
103         {
104                 .keyword = "null",
105                 .algo = RTE_CRYPTO_AUTH_NULL,
106                 .digest_len = 0,
107                 .key_len = 0,
108                 .key_not_req = 1
109         },
110         {
111                 .keyword = "sha1-hmac",
112                 .algo = RTE_CRYPTO_AUTH_SHA1_HMAC,
113                 .digest_len = 12,
114                 .key_len = 20
115         },
116         {
117                 .keyword = "sha256-hmac",
118                 .algo = RTE_CRYPTO_AUTH_SHA256_HMAC,
119                 .digest_len = 12,
120                 .key_len = 32
121         },
122         {
123                 .keyword = "aes-128-gcm",
124                 .algo = RTE_CRYPTO_AUTH_AES_GCM,
125                 .digest_len = 16,
126                 .aad_len = 8,
127                 .key_not_req = 1
128         }
129 };
130
131 struct ipsec_sa sa_out[IPSEC_SA_MAX_ENTRIES];
132 uint32_t nb_sa_out;
133
134 struct ipsec_sa sa_in[IPSEC_SA_MAX_ENTRIES];
135 uint32_t nb_sa_in;
136
137 static const struct supported_cipher_algo *
138 find_match_cipher_algo(const char *cipher_keyword)
139 {
140         size_t i;
141
142         for (i = 0; i < RTE_DIM(cipher_algos); i++) {
143                 const struct supported_cipher_algo *algo =
144                         &cipher_algos[i];
145
146                 if (strcmp(cipher_keyword, algo->keyword) == 0)
147                         return algo;
148         }
149
150         return NULL;
151 }
152
153 static const struct supported_auth_algo *
154 find_match_auth_algo(const char *auth_keyword)
155 {
156         size_t i;
157
158         for (i = 0; i < RTE_DIM(auth_algos); i++) {
159                 const struct supported_auth_algo *algo =
160                         &auth_algos[i];
161
162                 if (strcmp(auth_keyword, algo->keyword) == 0)
163                         return algo;
164         }
165
166         return NULL;
167 }
168
169 /** parse_key_string
170  *  parse x:x:x:x.... hex number key string into uint8_t *key
171  *  return:
172  *  > 0: number of bytes parsed
173  *  0:   failed
174  */
175 static uint32_t
176 parse_key_string(const char *key_str, uint8_t *key)
177 {
178         const char *pt_start = key_str, *pt_end = key_str;
179         uint32_t nb_bytes = 0;
180
181         while (pt_end != NULL) {
182                 char sub_str[3] = {0};
183
184                 pt_end = strchr(pt_start, ':');
185
186                 if (pt_end == NULL) {
187                         if (strlen(pt_start) > 2)
188                                 return 0;
189                         strncpy(sub_str, pt_start, 2);
190                 } else {
191                         if (pt_end - pt_start > 2)
192                                 return 0;
193
194                         strncpy(sub_str, pt_start, pt_end - pt_start);
195                         pt_start = pt_end + 1;
196                 }
197
198                 key[nb_bytes++] = strtol(sub_str, NULL, 16);
199         }
200
201         return nb_bytes;
202 }
203
204 void
205 parse_sa_tokens(char **tokens, uint32_t n_tokens,
206         struct parse_status *status)
207 {
208         struct ipsec_sa *rule = NULL;
209         uint32_t ti; /*token index*/
210         uint32_t *ri /*rule index*/;
211         uint32_t cipher_algo_p = 0;
212         uint32_t auth_algo_p = 0;
213         uint32_t src_p = 0;
214         uint32_t dst_p = 0;
215         uint32_t mode_p = 0;
216
217         if (strcmp(tokens[0], "in") == 0) {
218                 ri = &nb_sa_in;
219
220                 APP_CHECK(*ri <= IPSEC_SA_MAX_ENTRIES - 1, status,
221                         "too many sa rules, abort insertion\n");
222                 if (status->status < 0)
223                         return;
224
225                 rule = &sa_in[*ri];
226         } else {
227                 ri = &nb_sa_out;
228
229                 APP_CHECK(*ri <= IPSEC_SA_MAX_ENTRIES - 1, status,
230                         "too many sa rules, abort insertion\n");
231                 if (status->status < 0)
232                         return;
233
234                 rule = &sa_out[*ri];
235         }
236
237         /* spi number */
238         APP_CHECK_TOKEN_IS_NUM(tokens, 1, status);
239         if (status->status < 0)
240                 return;
241         rule->spi = atoi(tokens[1]);
242
243         for (ti = 2; ti < n_tokens; ti++) {
244                 if (strcmp(tokens[ti], "mode") == 0) {
245                         APP_CHECK_PRESENCE(mode_p, tokens[ti], status);
246                         if (status->status < 0)
247                                 return;
248
249                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
250                         if (status->status < 0)
251                                 return;
252
253                         if (strcmp(tokens[ti], "ipv4-tunnel") == 0)
254                                 rule->flags = IP4_TUNNEL;
255                         else if (strcmp(tokens[ti], "ipv6-tunnel") == 0)
256                                 rule->flags = IP6_TUNNEL;
257                         else if (strcmp(tokens[ti], "transport") == 0)
258                                 rule->flags = TRANSPORT;
259                         else {
260                                 APP_CHECK(0, status, "unrecognized "
261                                         "input \"%s\"", tokens[ti]);
262                                 return;
263                         }
264
265                         mode_p = 1;
266                         continue;
267                 }
268
269                 if (strcmp(tokens[ti], "cipher_algo") == 0) {
270                         const struct supported_cipher_algo *algo;
271                         uint32_t key_len;
272
273                         APP_CHECK_PRESENCE(cipher_algo_p, tokens[ti],
274                                 status);
275                         if (status->status < 0)
276                                 return;
277
278                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
279                         if (status->status < 0)
280                                 return;
281
282                         algo = find_match_cipher_algo(tokens[ti]);
283
284                         APP_CHECK(algo != NULL, status, "unrecognized "
285                                 "input \"%s\"", tokens[ti]);
286
287                         rule->cipher_algo = algo->algo;
288                         rule->block_size = algo->block_size;
289                         rule->iv_len = algo->iv_len;
290                         rule->cipher_key_len = algo->key_len;
291
292                         /* for NULL algorithm, no cipher key required */
293                         if (rule->cipher_algo == RTE_CRYPTO_CIPHER_NULL) {
294                                 cipher_algo_p = 1;
295                                 continue;
296                         }
297
298                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
299                         if (status->status < 0)
300                                 return;
301
302                         APP_CHECK(strcmp(tokens[ti], "cipher_key") == 0,
303                                 status, "unrecognized input \"%s\", "
304                                 "expect \"cipher_key\"", tokens[ti]);
305                         if (status->status < 0)
306                                 return;
307
308                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
309                         if (status->status < 0)
310                                 return;
311
312                         key_len = parse_key_string(tokens[ti],
313                                 rule->cipher_key);
314                         APP_CHECK(key_len == rule->cipher_key_len, status,
315                                 "unrecognized input \"%s\"", tokens[ti]);
316                         if (status->status < 0)
317                                 return;
318
319                         if (algo->algo == RTE_CRYPTO_CIPHER_AES_CBC)
320                                 rule->salt = (uint32_t)rte_rand();
321
322                         if ((algo->algo == RTE_CRYPTO_CIPHER_AES_CTR) ||
323                                 (algo->algo == RTE_CRYPTO_CIPHER_AES_GCM)) {
324                                 key_len -= 4;
325                                 rule->cipher_key_len = key_len;
326                                 memcpy(&rule->salt,
327                                         &rule->cipher_key[key_len], 4);
328                         }
329
330                         cipher_algo_p = 1;
331                         continue;
332                 }
333
334                 if (strcmp(tokens[ti], "auth_algo") == 0) {
335                         const struct supported_auth_algo *algo;
336                         uint32_t key_len;
337
338                         APP_CHECK_PRESENCE(auth_algo_p, tokens[ti],
339                                 status);
340                         if (status->status < 0)
341                                 return;
342
343                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
344                         if (status->status < 0)
345                                 return;
346
347                         algo = find_match_auth_algo(tokens[ti]);
348                         APP_CHECK(algo != NULL, status, "unrecognized "
349                                 "input \"%s\"", tokens[ti]);
350
351                         rule->auth_algo = algo->algo;
352                         rule->auth_key_len = algo->key_len;
353                         rule->digest_len = algo->digest_len;
354                         rule->aad_len = algo->key_len;
355
356                         /* NULL algorithm and combined algos do not
357                          * require auth key
358                          */
359                         if (algo->key_not_req) {
360                                 auth_algo_p = 1;
361                                 continue;
362                         }
363
364                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
365                         if (status->status < 0)
366                                 return;
367
368                         APP_CHECK(strcmp(tokens[ti], "auth_key") == 0,
369                                 status, "unrecognized input \"%s\", "
370                                 "expect \"auth_key\"", tokens[ti]);
371                         if (status->status < 0)
372                                 return;
373
374                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
375                         if (status->status < 0)
376                                 return;
377
378                         key_len = parse_key_string(tokens[ti],
379                                 rule->auth_key);
380                         APP_CHECK(key_len == rule->auth_key_len, status,
381                                 "unrecognized input \"%s\"", tokens[ti]);
382                         if (status->status < 0)
383                                 return;
384
385                         auth_algo_p = 1;
386                         continue;
387                 }
388
389                 if (strcmp(tokens[ti], "src") == 0) {
390                         APP_CHECK_PRESENCE(src_p, tokens[ti], status);
391                         if (status->status < 0)
392                                 return;
393
394                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
395                         if (status->status < 0)
396                                 return;
397
398                         if (rule->flags == IP4_TUNNEL) {
399                                 struct in_addr ip;
400
401                                 APP_CHECK(parse_ipv4_addr(tokens[ti],
402                                         &ip, NULL) == 0, status,
403                                         "unrecognized input \"%s\", "
404                                         "expect valid ipv4 addr",
405                                         tokens[ti]);
406                                 if (status->status < 0)
407                                         return;
408                                 rule->src.ip.ip4 = rte_bswap32(
409                                         (uint32_t)ip.s_addr);
410                         } else if (rule->flags == IP6_TUNNEL) {
411                                 struct in6_addr ip;
412
413                                 APP_CHECK(parse_ipv6_addr(tokens[ti], &ip,
414                                         NULL) == 0, status,
415                                         "unrecognized input \"%s\", "
416                                         "expect valid ipv6 addr",
417                                         tokens[ti]);
418                                 if (status->status < 0)
419                                         return;
420                                 memcpy(rule->src.ip.ip6.ip6_b,
421                                         ip.s6_addr, 16);
422                         } else if (rule->flags == TRANSPORT) {
423                                 APP_CHECK(0, status, "unrecognized input "
424                                         "\"%s\"", tokens[ti]);
425                                 return;
426                         }
427
428                         src_p = 1;
429                         continue;
430                 }
431
432                 if (strcmp(tokens[ti], "dst") == 0) {
433                         APP_CHECK_PRESENCE(dst_p, tokens[ti], status);
434                         if (status->status < 0)
435                                 return;
436
437                         INCREMENT_TOKEN_INDEX(ti, n_tokens, status);
438                         if (status->status < 0)
439                                 return;
440
441                         if (rule->flags == IP4_TUNNEL) {
442                                 struct in_addr ip;
443
444                                 APP_CHECK(parse_ipv4_addr(tokens[ti],
445                                         &ip, NULL) == 0, status,
446                                         "unrecognized input \"%s\", "
447                                         "expect valid ipv4 addr",
448                                         tokens[ti]);
449                                 if (status->status < 0)
450                                         return;
451                                 rule->dst.ip.ip4 = rte_bswap32(
452                                         (uint32_t)ip.s_addr);
453                         } else if (rule->flags == IP6_TUNNEL) {
454                                 struct in6_addr ip;
455
456                                 APP_CHECK(parse_ipv6_addr(tokens[ti], &ip,
457                                         NULL) == 0, status,
458                                         "unrecognized input \"%s\", "
459                                         "expect valid ipv6 addr",
460                                         tokens[ti]);
461                                 if (status->status < 0)
462                                         return;
463                                 memcpy(rule->dst.ip.ip6.ip6_b, ip.s6_addr, 16);
464                         } else if (rule->flags == TRANSPORT) {
465                                 APP_CHECK(0, status, "unrecognized "
466                                         "input \"%s\"", tokens[ti]);
467                                 return;
468                         }
469
470                         dst_p = 1;
471                         continue;
472                 }
473
474                 /* unrecognizeable input */
475                 APP_CHECK(0, status, "unrecognized input \"%s\"",
476                         tokens[ti]);
477                 return;
478         }
479
480         APP_CHECK(cipher_algo_p == 1, status, "missing cipher options");
481         if (status->status < 0)
482                 return;
483
484         APP_CHECK(auth_algo_p == 1, status, "missing auth options");
485         if (status->status < 0)
486                 return;
487
488         APP_CHECK(mode_p == 1, status, "missing mode option");
489         if (status->status < 0)
490                 return;
491
492         *ri = *ri + 1;
493 }
494
495 static inline void
496 print_one_sa_rule(const struct ipsec_sa *sa, int inbound)
497 {
498         uint32_t i;
499         uint8_t a, b, c, d;
500
501         printf("\tspi_%s(%3u):", inbound?"in":"out", sa->spi);
502
503         for (i = 0; i < RTE_DIM(cipher_algos); i++) {
504                 if (cipher_algos[i].algo == sa->cipher_algo) {
505                         printf("%s ", cipher_algos[i].keyword);
506                         break;
507                 }
508         }
509
510         for (i = 0; i < RTE_DIM(auth_algos); i++) {
511                 if (auth_algos[i].algo == sa->auth_algo) {
512                         printf("%s ", auth_algos[i].keyword);
513                         break;
514                 }
515         }
516
517         printf("mode:");
518
519         switch (sa->flags) {
520         case IP4_TUNNEL:
521                 printf("IP4Tunnel ");
522                 uint32_t_to_char(sa->src.ip.ip4, &a, &b, &c, &d);
523                 printf("%hhu.%hhu.%hhu.%hhu ", d, c, b, a);
524                 uint32_t_to_char(sa->dst.ip.ip4, &a, &b, &c, &d);
525                 printf("%hhu.%hhu.%hhu.%hhu", d, c, b, a);
526                 break;
527         case IP6_TUNNEL:
528                 printf("IP6Tunnel ");
529                 for (i = 0; i < 16; i++) {
530                         if (i % 2 && i != 15)
531                                 printf("%.2x:", sa->src.ip.ip6.ip6_b[i]);
532                         else
533                                 printf("%.2x", sa->src.ip.ip6.ip6_b[i]);
534                 }
535                 printf(" ");
536                 for (i = 0; i < 16; i++) {
537                         if (i % 2 && i != 15)
538                                 printf("%.2x:", sa->dst.ip.ip6.ip6_b[i]);
539                         else
540                                 printf("%.2x", sa->dst.ip.ip6.ip6_b[i]);
541                 }
542                 break;
543         case TRANSPORT:
544                 printf("Transport");
545                 break;
546         }
547         printf("\n");
548 }
549
550 struct sa_ctx {
551         struct ipsec_sa sa[IPSEC_SA_MAX_ENTRIES];
552         struct {
553                 struct rte_crypto_sym_xform a;
554                 struct rte_crypto_sym_xform b;
555         } xf[IPSEC_SA_MAX_ENTRIES];
556 };
557
558 static struct sa_ctx *
559 sa_create(const char *name, int32_t socket_id)
560 {
561         char s[PATH_MAX];
562         struct sa_ctx *sa_ctx;
563         uint32_t mz_size;
564         const struct rte_memzone *mz;
565
566         snprintf(s, sizeof(s), "%s_%u", name, socket_id);
567
568         /* Create SA array table */
569         printf("Creating SA context with %u maximum entries\n",
570                         IPSEC_SA_MAX_ENTRIES);
571
572         mz_size = sizeof(struct sa_ctx);
573         mz = rte_memzone_reserve(s, mz_size, socket_id,
574                         RTE_MEMZONE_1GB | RTE_MEMZONE_SIZE_HINT_ONLY);
575         if (mz == NULL) {
576                 printf("Failed to allocate SA DB memory\n");
577                 rte_errno = -ENOMEM;
578                 return NULL;
579         }
580
581         sa_ctx = (struct sa_ctx *)mz->addr;
582
583         return sa_ctx;
584 }
585
586 static int
587 sa_add_rules(struct sa_ctx *sa_ctx, const struct ipsec_sa entries[],
588                 uint32_t nb_entries, uint32_t inbound)
589 {
590         struct ipsec_sa *sa;
591         uint32_t i, idx;
592
593         for (i = 0; i < nb_entries; i++) {
594                 idx = SPI2IDX(entries[i].spi);
595                 sa = &sa_ctx->sa[idx];
596                 if (sa->spi != 0) {
597                         printf("Index %u already in use by SPI %u\n",
598                                         idx, sa->spi);
599                         return -EINVAL;
600                 }
601                 *sa = entries[i];
602                 sa->seq = 0;
603
604                 switch (sa->flags) {
605                 case IP4_TUNNEL:
606                         sa->src.ip.ip4 = rte_cpu_to_be_32(sa->src.ip.ip4);
607                         sa->dst.ip.ip4 = rte_cpu_to_be_32(sa->dst.ip.ip4);
608                 }
609
610                 if (inbound) {
611                         sa_ctx->xf[idx].b.type = RTE_CRYPTO_SYM_XFORM_CIPHER;
612                         sa_ctx->xf[idx].b.cipher.algo = sa->cipher_algo;
613                         sa_ctx->xf[idx].b.cipher.key.data = sa->cipher_key;
614                         sa_ctx->xf[idx].b.cipher.key.length =
615                                 sa->cipher_key_len;
616                         sa_ctx->xf[idx].b.cipher.op =
617                                 RTE_CRYPTO_CIPHER_OP_DECRYPT;
618                         sa_ctx->xf[idx].b.next = NULL;
619
620                         sa_ctx->xf[idx].a.type = RTE_CRYPTO_SYM_XFORM_AUTH;
621                         sa_ctx->xf[idx].a.auth.algo = sa->auth_algo;
622                         sa_ctx->xf[idx].a.auth.add_auth_data_length =
623                                 sa->aad_len;
624                         sa_ctx->xf[idx].a.auth.key.data = sa->auth_key;
625                         sa_ctx->xf[idx].a.auth.key.length =
626                                 sa->auth_key_len;
627                         sa_ctx->xf[idx].a.auth.digest_length =
628                                 sa->digest_len;
629                         sa_ctx->xf[idx].a.auth.op =
630                                 RTE_CRYPTO_AUTH_OP_VERIFY;
631
632                 } else { /* outbound */
633                         sa_ctx->xf[idx].a.type = RTE_CRYPTO_SYM_XFORM_CIPHER;
634                         sa_ctx->xf[idx].a.cipher.algo = sa->cipher_algo;
635                         sa_ctx->xf[idx].a.cipher.key.data = sa->cipher_key;
636                         sa_ctx->xf[idx].a.cipher.key.length =
637                                 sa->cipher_key_len;
638                         sa_ctx->xf[idx].a.cipher.op =
639                                 RTE_CRYPTO_CIPHER_OP_ENCRYPT;
640                         sa_ctx->xf[idx].a.next = NULL;
641
642                         sa_ctx->xf[idx].b.type = RTE_CRYPTO_SYM_XFORM_AUTH;
643                         sa_ctx->xf[idx].b.auth.algo = sa->auth_algo;
644                         sa_ctx->xf[idx].b.auth.add_auth_data_length =
645                                 sa->aad_len;
646                         sa_ctx->xf[idx].b.auth.key.data = sa->auth_key;
647                         sa_ctx->xf[idx].b.auth.key.length =
648                                 sa->auth_key_len;
649                         sa_ctx->xf[idx].b.auth.digest_length =
650                                 sa->digest_len;
651                         sa_ctx->xf[idx].b.auth.op =
652                                 RTE_CRYPTO_AUTH_OP_GENERATE;
653                 }
654
655                 sa_ctx->xf[idx].a.next = &sa_ctx->xf[idx].b;
656                 sa_ctx->xf[idx].b.next = NULL;
657                 sa->xforms = &sa_ctx->xf[idx].a;
658
659                 print_one_sa_rule(sa, inbound);
660         }
661
662         return 0;
663 }
664
665 static inline int
666 sa_out_add_rules(struct sa_ctx *sa_ctx, const struct ipsec_sa entries[],
667                 uint32_t nb_entries)
668 {
669         return sa_add_rules(sa_ctx, entries, nb_entries, 0);
670 }
671
672 static inline int
673 sa_in_add_rules(struct sa_ctx *sa_ctx, const struct ipsec_sa entries[],
674                 uint32_t nb_entries)
675 {
676         return sa_add_rules(sa_ctx, entries, nb_entries, 1);
677 }
678
679 void
680 sa_init(struct socket_ctx *ctx, int32_t socket_id)
681 {
682         const char *name;
683
684         if (ctx == NULL)
685                 rte_exit(EXIT_FAILURE, "NULL context.\n");
686
687         if (ctx->sa_in != NULL)
688                 rte_exit(EXIT_FAILURE, "Inbound SA DB for socket %u already "
689                                 "initialized\n", socket_id);
690
691         if (ctx->sa_out != NULL)
692                 rte_exit(EXIT_FAILURE, "Outbound SA DB for socket %u already "
693                                 "initialized\n", socket_id);
694
695         if (nb_sa_in > 0) {
696                 name = "sa_in";
697                 ctx->sa_in = sa_create(name, socket_id);
698                 if (ctx->sa_in == NULL)
699                         rte_exit(EXIT_FAILURE, "Error [%d] creating SA "
700                                 "context %s in socket %d\n", rte_errno,
701                                 name, socket_id);
702
703                 sa_in_add_rules(ctx->sa_in, sa_in, nb_sa_in);
704         } else
705                 RTE_LOG(WARNING, IPSEC, "No SA Inbound rule specified\n");
706
707         if (nb_sa_out > 0) {
708                 name = "sa_out";
709                 ctx->sa_out = sa_create(name, socket_id);
710                 if (ctx->sa_out == NULL)
711                         rte_exit(EXIT_FAILURE, "Error [%d] creating SA "
712                                 "context %s in socket %d\n", rte_errno,
713                                 name, socket_id);
714
715                 sa_out_add_rules(ctx->sa_out, sa_out, nb_sa_out);
716         } else
717                 RTE_LOG(WARNING, IPSEC, "No SA Outbound rule "
718                         "specified\n");
719 }
720
721 int
722 inbound_sa_check(struct sa_ctx *sa_ctx, struct rte_mbuf *m, uint32_t sa_idx)
723 {
724         struct ipsec_mbuf_metadata *priv;
725
726         priv = RTE_PTR_ADD(m, sizeof(struct rte_mbuf));
727
728         return (sa_ctx->sa[sa_idx].spi == priv->sa->spi);
729 }
730
731 static inline void
732 single_inbound_lookup(struct ipsec_sa *sadb, struct rte_mbuf *pkt,
733                 struct ipsec_sa **sa_ret)
734 {
735         struct esp_hdr *esp;
736         struct ip *ip;
737         uint32_t *src4_addr;
738         uint8_t *src6_addr;
739         struct ipsec_sa *sa;
740
741         *sa_ret = NULL;
742
743         ip = rte_pktmbuf_mtod(pkt, struct ip *);
744         if (ip->ip_v == IPVERSION)
745                 esp = (struct esp_hdr *)(ip + 1);
746         else
747                 esp = (struct esp_hdr *)(((struct ip6_hdr *)ip) + 1);
748
749         if (esp->spi == INVALID_SPI)
750                 return;
751
752         sa = &sadb[SPI2IDX(rte_be_to_cpu_32(esp->spi))];
753         if (rte_be_to_cpu_32(esp->spi) != sa->spi)
754                 return;
755
756         switch (sa->flags) {
757         case IP4_TUNNEL:
758                 src4_addr = RTE_PTR_ADD(ip, offsetof(struct ip, ip_src));
759                 if ((ip->ip_v == IPVERSION) &&
760                                 (sa->src.ip.ip4 == *src4_addr) &&
761                                 (sa->dst.ip.ip4 == *(src4_addr + 1)))
762                         *sa_ret = sa;
763                 break;
764         case IP6_TUNNEL:
765                 src6_addr = RTE_PTR_ADD(ip, offsetof(struct ip6_hdr, ip6_src));
766                 if ((ip->ip_v == IP6_VERSION) &&
767                                 !memcmp(&sa->src.ip.ip6.ip6, src6_addr, 16) &&
768                                 !memcmp(&sa->dst.ip.ip6.ip6, src6_addr + 16, 16))
769                         *sa_ret = sa;
770                 break;
771         case TRANSPORT:
772                 *sa_ret = sa;
773         }
774 }
775
776 void
777 inbound_sa_lookup(struct sa_ctx *sa_ctx, struct rte_mbuf *pkts[],
778                 struct ipsec_sa *sa[], uint16_t nb_pkts)
779 {
780         uint32_t i;
781
782         for (i = 0; i < nb_pkts; i++)
783                 single_inbound_lookup(sa_ctx->sa, pkts[i], &sa[i]);
784 }
785
786 void
787 outbound_sa_lookup(struct sa_ctx *sa_ctx, uint32_t sa_idx[],
788                 struct ipsec_sa *sa[], uint16_t nb_pkts)
789 {
790         uint32_t i;
791
792         for (i = 0; i < nb_pkts; i++)
793                 sa[i] = &sa_ctx->sa[sa_idx[i]];
794 }