vppinfra: add array mask func
[vpp.git] / src / vppinfra / vector / test / array_mask.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/array_mask.h>
8
9 __clib_test_fn void
10 clib_array_mask_u32_wrapper (u32 *src, u32 mask, u32 n_elts)
11 {
12   clib_array_mask_u32 (src, mask, n_elts);
13 }
14
15 typedef struct
16 {
17   u32 mask;
18   u32 expected[256];
19 } array_mask_test_t;
20
21 static array_mask_test_t tests[] = {
22   /* mask values 0x1, output array of alternating 0 1 0 1 .. */
23   { .mask = 1,
24     .expected = { 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
25                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
26                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
27                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
28                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
29                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
30                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
31                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
32                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
33                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
34                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
35                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1,
36                   0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1 } },
37   /* mask values 0xFFFFFFFF, output array of 0, 1, 2, .., 255 */
38   { .mask = ~0U,
39     .expected = { 0,   1,   2,   3,   4,   5,   6,   7,   8,   9,   10,  11,
40                   12,  13,  14,  15,  16,  17,  18,  19,  20,  21,  22,  23,
41                   24,  25,  26,  27,  28,  29,  30,  31,  32,  33,  34,  35,
42                   36,  37,  38,  39,  40,  41,  42,  43,  44,  45,  46,  47,
43                   48,  49,  50,  51,  52,  53,  54,  55,  56,  57,  58,  59,
44                   60,  61,  62,  63,  64,  65,  66,  67,  68,  69,  70,  71,
45                   72,  73,  74,  75,  76,  77,  78,  79,  80,  81,  82,  83,
46                   84,  85,  86,  87,  88,  89,  90,  91,  92,  93,  94,  95,
47                   96,  97,  98,  99,  100, 101, 102, 103, 104, 105, 106, 107,
48                   108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119,
49                   120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131,
50                   132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143,
51                   144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155,
52                   156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167,
53                   168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179,
54                   180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191,
55                   192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203,
56                   204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215,
57                   216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227,
58                   228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239,
59                   240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251,
60                   252, 253, 254, 255 } },
61   /* mask values 0xF, output array of 0, .., 15, 0, .., 15, 0, .., 15 */
62   { .mask = 15,
63     .expected = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
64                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
65                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
66                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
67                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
68                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
69                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
70                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
71                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
72                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
73                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
74                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
75                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
76                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
77                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
78                   0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 } },
79 };
80
81 static clib_error_t *
82 test_clib_array_mask_u32 (clib_error_t *err)
83 {
84   u32 i, j;
85   for (i = 0; i < ARRAY_LEN (tests); i++)
86     {
87       u32 src[256];
88       for (j = 0; j < ARRAY_LEN (src); j++)
89         src[j] = j;
90
91       array_mask_test_t *t = tests + i;
92       clib_array_mask_u32_wrapper (src, t->mask, ARRAY_LEN (src));
93       for (j = 0; j < ARRAY_LEN (src); j++)
94         {
95           if (src[j] != t->expected[j])
96             return clib_error_return (err,
97                                       "testcase %u failed at "
98                                       "(src[%u] = 0x%x, expected 0x%x)",
99                                       i, j, src[j], t->expected[j]);
100         }
101     }
102   return err;
103 }
104
105 REGISTER_TEST (clib_array_mask_u32) = {
106   .name = "clib_array_mask_u32",
107   .fn = test_clib_array_mask_u32,
108 };