1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * KUnit tests for element parsing 4 * 5 * Copyright (C) 2023-2024 Intel Corporation 6 */ 7 #include <kunit/test.h> 8 #include "../ieee80211_i.h" 9 10 MODULE_IMPORT_NS(EXPORTED_FOR_KUNIT_TESTING); 11 12 static void mle_defrag(struct kunit *test) 13 { 14 struct ieee80211_elems_parse_params parse_params = { 15 .link_id = 12, 16 .from_ap = true, 17 .mode = IEEE80211_CONN_MODE_EHT, 18 }; 19 struct ieee802_11_elems *parsed; 20 struct sk_buff *skb; 21 u8 *len_mle, *len_prof; 22 int i; 23 24 skb = alloc_skb(1024, GFP_KERNEL); 25 KUNIT_ASSERT_NOT_NULL(test, skb); 26 27 if (skb_pad(skb, skb_tailroom(skb))) { 28 KUNIT_FAIL(test, "failed to pad skb"); 29 return; 30 } 31 32 /* build a multi-link element */ 33 skb_put_u8(skb, WLAN_EID_EXTENSION); 34 len_mle = skb_put(skb, 1); 35 skb_put_u8(skb, WLAN_EID_EXT_EHT_MULTI_LINK); 36 37 put_unaligned_le16(IEEE80211_ML_CONTROL_TYPE_BASIC, 38 skb_put(skb, 2)); 39 /* struct ieee80211_mle_basic_common_info */ 40 skb_put_u8(skb, 7); /* includes len field */ 41 skb_put_data(skb, "\x00\x00\x00\x00\x00\x00", ETH_ALEN); /* MLD addr */ 42 43 /* with a STA profile inside */ 44 skb_put_u8(skb, IEEE80211_MLE_SUBELEM_PER_STA_PROFILE); 45 len_prof = skb_put(skb, 1); 46 put_unaligned_le16(IEEE80211_MLE_STA_CONTROL_COMPLETE_PROFILE | 47 parse_params.link_id, 48 skb_put(skb, 2)); 49 skb_put_u8(skb, 1); /* fake sta_info_len - includes itself */ 50 /* put a bunch of useless elements into it */ 51 for (i = 0; i < 20; i++) { 52 skb_put_u8(skb, WLAN_EID_SSID); 53 skb_put_u8(skb, 20); 54 skb_put(skb, 20); 55 } 56 57 /* fragment STA profile */ 58 ieee80211_fragment_element(skb, len_prof, 59 IEEE80211_MLE_SUBELEM_FRAGMENT); 60 /* fragment MLE */ 61 ieee80211_fragment_element(skb, len_mle, WLAN_EID_FRAGMENT); 62 63 parse_params.start = skb->data; 64 parse_params.len = skb->len; 65 parsed = ieee802_11_parse_elems_full(&parse_params); 66 /* should return ERR_PTR or valid, not NULL */ 67 KUNIT_EXPECT_NOT_NULL(test, parsed); 68 69 if (IS_ERR_OR_NULL(parsed)) 70 goto free_skb; 71 72 KUNIT_EXPECT_NOT_NULL(test, parsed->ml_basic); 73 KUNIT_EXPECT_EQ(test, 74 parsed->ml_basic_len, 75 2 /* control */ + 76 7 /* common info */ + 77 2 /* sta profile element header */ + 78 3 /* sta profile header */ + 79 20 * 22 /* sta profile data */ + 80 2 /* sta profile fragment element */); 81 KUNIT_EXPECT_NOT_NULL(test, parsed->prof); 82 KUNIT_EXPECT_EQ(test, 83 parsed->sta_prof_len, 84 3 /* sta profile header */ + 85 20 * 22 /* sta profile data */); 86 87 kfree(parsed); 88 free_skb: 89 kfree_skb(skb); 90 } 91 92 static struct kunit_case element_parsing_test_cases[] = { 93 KUNIT_CASE(mle_defrag), 94 {} 95 }; 96 97 static struct kunit_suite element_parsing = { 98 .name = "mac80211-element-parsing", 99 .test_cases = element_parsing_test_cases, 100 }; 101 102 kunit_test_suite(element_parsing); 103