blob: 997d0cd27b2da986e7159620bc2a06b6c6cb6fc6 [file] [log] [blame]
// SPDX-License-Identifier: GPL-2.0-only
/*
* KUnit tests for element parsing
*
* Copyright (C) 2023 Intel Corporation
*/
#include <kunit/test.h>
#include "../ieee80211_i.h"
MODULE_IMPORT_NS(EXPORTED_FOR_KUNIT_TESTING);
static void mle_defrag(struct kunit *test)
{
struct ieee80211_elems_parse_params parse_params = {
.link_id = 12,
.from_ap = true,
};
struct ieee802_11_elems *parsed;
struct sk_buff *skb;
u8 *len_mle, *len_prof;
int i;
skb = alloc_skb(1024, GFP_KERNEL);
KUNIT_ASSERT_NOT_NULL(test, skb);
if (skb_pad(skb, skb_tailroom(skb))) {
KUNIT_FAIL(test, "failed to pad skb");
return;
}
/* build a multi-link element */
skb_put_u8(skb, WLAN_EID_EXTENSION);
len_mle = skb_put(skb, 1);
skb_put_u8(skb, WLAN_EID_EXT_EHT_MULTI_LINK);
put_unaligned_le16(IEEE80211_ML_CONTROL_TYPE_BASIC,
skb_put(skb, 2));
/* struct ieee80211_mle_basic_common_info */
skb_put_u8(skb, 7); /* includes len field */
skb_put_data(skb, "\x00\x00\x00\x00\x00\x00", ETH_ALEN); /* MLD addr */
/* with a STA profile inside */
skb_put_u8(skb, IEEE80211_MLE_SUBELEM_PER_STA_PROFILE);
len_prof = skb_put(skb, 1);
put_unaligned_le16(IEEE80211_MLE_STA_CONTROL_COMPLETE_PROFILE |
parse_params.link_id,
skb_put(skb, 2));
skb_put_u8(skb, 1); /* fake sta_info_len - includes itself */
/* put a bunch of useless elements into it */
for (i = 0; i < 20; i++) {
skb_put_u8(skb, WLAN_EID_SSID);
skb_put_u8(skb, 20);
skb_put(skb, 20);
}
/* fragment STA profile */
ieee80211_fragment_element(skb, len_prof,
IEEE80211_MLE_SUBELEM_FRAGMENT);
/* fragment MLE */
ieee80211_fragment_element(skb, len_mle, WLAN_EID_FRAGMENT);
parse_params.start = skb->data;
parse_params.len = skb->len;
parsed = ieee802_11_parse_elems_full(&parse_params);
/* should return ERR_PTR or valid, not NULL */
KUNIT_EXPECT_NOT_NULL(test, parsed);
if (IS_ERR_OR_NULL(parsed))
goto free_skb;
KUNIT_EXPECT_NOT_NULL(test, parsed->ml_basic_elem);
KUNIT_EXPECT_EQ(test,
parsed->ml_basic_len,
2 /* control */ +
7 /* common info */ +
2 /* sta profile element header */ +
3 /* sta profile header */ +
20 * 22 /* sta profile data */ +
2 /* sta profile fragment element */);
KUNIT_EXPECT_NOT_NULL(test, parsed->prof);
KUNIT_EXPECT_EQ(test,
parsed->sta_prof_len,
3 /* sta profile header */ +
20 * 22 /* sta profile data */);
kfree(parsed);
free_skb:
kfree_skb(skb);
}
static struct kunit_case element_parsing_test_cases[] = {
KUNIT_CASE(mle_defrag),
{}
};
static struct kunit_suite element_parsing = {
.name = "mac80211-element-parsing",
.test_cases = element_parsing_test_cases,
};
kunit_test_suite(element_parsing);