src/vppinfra/vector/test/ip_csum.c

   1 /* SPDX-License-Identifier: Apache-2.0
   2  * Copyright(c) 2021 Cisco Systems, Inc.
   3  */
   4
   5 #include <vppinfra/format.h>
   6 #include <vppinfra/vector/test/test.h>
   7 #include <vppinfra/vector/ip_csum.h>
   8
   9 typedef struct
  10 {
  11   struct
  12   {
  13     u8 *src;
  14     u32 count;
  15   } chunk[5];
  16   u16 result;
  17 } ip_csum_test_t;
  18
  19 static u8 test1[] = { 0x45, 0x00, 0x00, 0x73, 0x00, 0x00, 0x40,
  20                       0x00, 0x40, 0x11, 0x00, 0x00, 0xc0, 0xa8,
  21                       0x00, 0x01, 0xc0, 0xa8, 0x00, 0xc7, 0x00 };
  22 #define TEST_LEN(x) (ARRAY_LEN (x) - 1)
  23
  24 static ip_csum_test_t tests[] = { {
  25                                     .chunk[0].src = test1,
  26                                     .chunk[0].count = TEST_LEN (test1),
  27                                     .result = 0x61b8,
  28                                   },
  29                                   {
  30                                     .chunk[0].src = test1,
  31                                     .chunk[0].count = 1,
  32                                     .chunk[1].src = test1 + 1,
  33                                     .chunk[1].count = 2,
  34                                     .chunk[2].src = test1 + 3,
  35                                     .chunk[2].count = 3,
  36                                     .chunk[3].src = test1 + 6,
  37                                     .chunk[3].count = 4,
  38                                     .chunk[4].src = test1 + 10,
  39                                     .chunk[4].count = TEST_LEN (test1) - 10,
  40                                     .result = 0x61b8,
  41                                   },
  42                                   {
  43                                     .chunk[0].count = 1,
  44                                     .result = 0xff0f,
  45                                   },
  46                                   {
  47                                     .chunk[0].count = 2,
  48                                     .result = 0x080f,
  49                                   },
  50                                   {
  51                                     .chunk[0].count = 3,
  52                                     .result = 0x0711,
  53                                   },
  54                                   {
  55                                     .chunk[0].count = 4,
  56                                     .result = 0x1210,
  57                                   },
  58                                   {
  59                                     .chunk[0].count = 63,
  60                                     .result = 0xda01,
  61                                   },
  62                                   {
  63                                     .chunk[0].count = 64,
  64                                     .result = 0xe100,
  65                                   },
  66                                   {
  67                                     .chunk[0].count = 65,
  68                                     .result = 0xe010,
  69                                   },
  70                                   {
  71                                     .chunk[0].count = 65535,
  72                                     .result = 0xfc84,
  73                                   },
  74                                   {
  75                                     .chunk[0].count = 65536,
  76                                     .result = 0xffff,
  77                                   } };
  78
  79 static clib_error_t *
  80 test_clib_ip_csum (clib_error_t *err)
  81 {
  82   u8 *buf;
  83   buf = clib_mem_alloc_aligned (65536, CLIB_CACHE_LINE_BYTES);
  84   for (int i = 0; i < 65536; i++)
  85     buf[i] = 0xf0 + ((i * 7) & 0xf);
  86
  87   for (int i = 0; i < ARRAY_LEN (tests); i++)
  88     {
  89       clib_ip_csum_t c = {};
  90       ip_csum_test_t *t = tests + i;
  91       u16 rv;
  92
  93       for (int j = 0; j < ARRAY_LEN (((ip_csum_test_t *) 0)->chunk); j++)
  94         if (t->chunk[j].count > 0)
  95           {
  96             if (t->chunk[j].src == 0)
  97               clib_ip_csum_chunk (&c, buf, t->chunk[j].count);
  98             else
  99               clib_ip_csum_chunk (&c, t->chunk[j].src, t->chunk[j].count);
 100           }
 101       rv = clib_ip_csum_fold (&c);
 102
 103       if (rv != tests[i].result)
 104         {
 105           err = clib_error_return (err,
 106                                    "bad checksum in test case %u (expected "
 107                                    "0x%04x, calculated 0x%04x)",
 108                                    i, tests[i].result, rv);
 109           goto done;
 110         }
 111     }
 112 done:
 113   clib_mem_free (buf);
 114   return err;
 115 }
 116
 117 REGISTER_TEST (clib_ip_csum) = {
 118   .name = "clib_ip_csum",
 119   .fn = test_clib_ip_csum,
 120 };