1 // SPDX-License-Identifier: GPL-2.0-only
3 * KUnit tests for element fragmentation
5 * Copyright (C) 2023 Intel Corporation
7 #include <linux/ieee80211.h>
8 #include <net/cfg80211.h>
9 #include <kunit/test.h>
11 static void defragment_0(struct kunit *test)
14 static const u8 input[] = {
15 [0] = WLAN_EID_EXTENSION,
17 [2] = WLAN_EID_EXT_EHT_MULTI_LINK,
20 [254 + 2] = WLAN_EID_FRAGMENT,
22 [254 + 3 + 7] = 0, /* for size */
24 u8 *data = kunit_kzalloc(test, sizeof(input), GFP_KERNEL);
26 KUNIT_ASSERT_NOT_NULL(test, data);
28 ret = cfg80211_defragment_element((void *)input,
32 KUNIT_EXPECT_EQ(test, ret, 253);
33 KUNIT_EXPECT_MEMEQ(test, data, input + 3, 253);
36 static void defragment_1(struct kunit *test)
39 static const u8 input[] = {
40 [0] = WLAN_EID_EXTENSION,
42 [2] = WLAN_EID_EXT_EHT_MULTI_LINK,
45 [255 + 2] = WLAN_EID_FRAGMENT,
48 [255 + 3 + 8] = WLAN_EID_FRAGMENT, /* not used */
50 [255 + 3 + 10] = 0, /* for size */
52 u8 *data = kunit_kzalloc(test, sizeof(input), GFP_KERNEL);
53 const struct element *elem;
56 KUNIT_ASSERT_NOT_NULL(test, data);
58 for_each_element(elem, input, sizeof(input))
61 /* check the elements are right */
62 KUNIT_ASSERT_EQ(test, count, 3);
64 ret = cfg80211_defragment_element((void *)input,
68 /* this means the last fragment was not used */
69 KUNIT_EXPECT_EQ(test, ret, 254 + 7);
70 KUNIT_EXPECT_MEMEQ(test, data, input + 3, 254);
71 KUNIT_EXPECT_MEMEQ(test, data + 254, input + 255 + 4, 7);
74 static void defragment_2(struct kunit *test)
77 static const u8 input[] = {
78 [0] = WLAN_EID_EXTENSION,
80 [2] = WLAN_EID_EXT_EHT_MULTI_LINK,
84 [257 + 0] = WLAN_EID_FRAGMENT,
88 [2 * 257 + 0] = WLAN_EID_FRAGMENT,
91 [2 * 257 + 3] = WLAN_EID_FRAGMENT, /* not used */
93 [2 * 257 + 5] = 0, /* for size */
95 u8 *data = kunit_kzalloc(test, sizeof(input), GFP_KERNEL);
96 const struct element *elem;
99 KUNIT_ASSERT_NOT_NULL(test, data);
101 for_each_element(elem, input, sizeof(input))
104 /* check the elements are right */
105 KUNIT_ASSERT_EQ(test, count, 4);
107 ret = cfg80211_defragment_element((void *)input,
108 input, sizeof(input),
111 /* this means the last fragment was not used */
112 KUNIT_EXPECT_EQ(test, ret, 254 + 255 + 1);
113 KUNIT_EXPECT_MEMEQ(test, data, input + 3, 254);
114 KUNIT_EXPECT_MEMEQ(test, data + 254, input + 257 + 2, 255);
115 KUNIT_EXPECT_MEMEQ(test, data + 254 + 255, input + 2 * 257 + 2, 1);
118 static void defragment_at_end(struct kunit *test)
121 static const u8 input[] = {
122 [0] = WLAN_EID_EXTENSION,
124 [2] = WLAN_EID_EXT_EHT_MULTI_LINK,
127 [255 + 2] = WLAN_EID_FRAGMENT,
129 [255 + 3 + 7] = 0, /* for size */
131 u8 *data = kunit_kzalloc(test, sizeof(input), GFP_KERNEL);
133 KUNIT_ASSERT_NOT_NULL(test, data);
135 ret = cfg80211_defragment_element((void *)input,
136 input, sizeof(input),
139 KUNIT_EXPECT_EQ(test, ret, 254 + 7);
140 KUNIT_EXPECT_MEMEQ(test, data, input + 3, 254);
141 KUNIT_EXPECT_MEMEQ(test, data + 254, input + 255 + 4, 7);
144 static struct kunit_case element_fragmentation_test_cases[] = {
145 KUNIT_CASE(defragment_0),
146 KUNIT_CASE(defragment_1),
147 KUNIT_CASE(defragment_2),
148 KUNIT_CASE(defragment_at_end),
152 static struct kunit_suite element_fragmentation = {
153 .name = "cfg80211-element-defragmentation",
154 .test_cases = element_fragmentation_test_cases,
157 kunit_test_suite(element_fragmentation);