src/vppinfra/vector/test/count_equal.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/count_equal.h>
   8
   9 #define foreach_clib_count_equal(type)                                        \
  10   typedef uword (wrapper_fn_##type) (type * a, uword maxcount);               \
  11                                                                               \
  12   __clib_test_fn uword clib_count_equal_##type##_wrapper (type *a,            \
  13                                                           uword maxcount)     \
  14   {                                                                           \
  15     return clib_count_equal_##type (a, maxcount);                             \
  16   }                                                                           \
  17                                                                               \
  18   static wrapper_fn_##type *wfn_##type = &clib_count_equal_##type##_wrapper;  \
  19   static clib_error_t *test_clib_count_equal_##type (clib_error_t *err)       \
  20   {                                                                           \
  21     u32 ps = clib_mem_get_log2_page_size ();                                  \
  22     void *map;                                                                \
  23                                                                               \
  24     u16 lengths[] = {                                                         \
  25       1, 2, 3, 5, 7, 9, 15, 16, 17, 31, 32, 33, 255, 256, 257                 \
  26     };                                                                        \
  27     type *data;                                                               \
  28                                                                               \
  29     map = clib_mem_vm_map (0, 2ULL << ps, ps, "test");                        \
  30     if (map == CLIB_MEM_VM_MAP_FAILED)                                        \
  31       return clib_error_return (err, "clib_mem_vm_map failed");               \
  32                                                                               \
  33     data = ((type *) (map + (1ULL << ps)));                                   \
  34     data[-1] = 0xfe;                                                          \
  35                                                                               \
  36     mprotect (data, 1ULL < ps, PROT_NONE);                                    \
  37                                                                               \
  38     for (int i = 1; i <= (1 << ps) / sizeof (data[0]); i++)                   \
  39       data[-i] = 7;                                                           \
  40                                                                               \
  41     for (int i = 0; i < ARRAY_LEN (lengths); i++)                             \
  42       {                                                                       \
  43         uword rv, len = lengths[i];                                           \
  44                                                                               \
  45         if ((rv = wfn_##type (data - len, len)) != len)                       \
  46           {                                                                   \
  47             err = clib_error_return (                                         \
  48               err, "testcase 1 failed for len %u (rv %u)", len, rv);          \
  49             goto done;                                                        \
  50           }                                                                   \
  51                                                                               \
  52         data[-1] = 8;                                                         \
  53         if (len > 1 && ((rv = wfn_##type (data - len, len)) != len - 1))      \
  54           {                                                                   \
  55             err = clib_error_return (                                         \
  56               err, "testcase 2 failed for len %u (rv %u)", len, rv);          \
  57             goto done;                                                        \
  58           }                                                                   \
  59         data[-1] = 7;                                                         \
  60                                                                               \
  61         data[-2] = 8;                                                         \
  62         if (len > 2 && ((rv = wfn_##type (data - len, len)) != len - 2))      \
  63           {                                                                   \
  64             err = clib_error_return (                                         \
  65               err, "testcase 3 failed for len %u (rv %u)", len, rv);          \
  66             goto done;                                                        \
  67           }                                                                   \
  68         data[-2] = 7;                                                         \
  69       }                                                                       \
  70                                                                               \
  71   done:                                                                       \
  72     clib_mem_vm_unmap (map);                                                  \
  73     return err;                                                               \
  74   }
  75
  76 foreach_clib_count_equal (u8);
  77 foreach_clib_count_equal (u16);
  78 foreach_clib_count_equal (u32);
  79 foreach_clib_count_equal (u64);
  80
  81 REGISTER_TEST (clib_count_equal_u8) = {
  82   .name = "clib_count_equal_u8",
  83   .fn = test_clib_count_equal_u8,
  84 };
  85
  86 REGISTER_TEST (clib_count_equal_u16) = {
  87   .name = "clib_count_equal_u16",
  88   .fn = test_clib_count_equal_u16,
  89 };
  90
  91 REGISTER_TEST (clib_count_equal_u32) = {
  92   .name = "clib_count_equal_u32",
  93   .fn = test_clib_count_equal_u32,
  94 };
  95
  96 REGISTER_TEST (clib_count_equal_u64) = {
  97   .name = "clib_count_equal_u64",
  98   .fn = test_clib_count_equal_u64,
  99 };