73 static void iwlmld_kunit_init_link(struct ieee80211_vif *vif,  in IWL_MLD_ALLOC_FN()
79 	struct iwl_mld_vif *mld_vif = iwl_mld_vif_from_mac80211(vif);  in IWL_MLD_ALLOC_FN()
83 	rcu_assign_pointer(vif->link_conf[link_id], link);  in IWL_MLD_ALLOC_FN()
85 	link->vif = vif;  in IWL_MLD_ALLOC_FN()
94 	rcu_assign_pointer(vif->link_conf[link_id], link);  in IWL_MLD_ALLOC_FN()
97 IWL_MLD_ALLOC_FN(vif, vif)  in IWL_MLD_ALLOC_FN()  argument
99 /* Helper function to add and initialize a VIF for KUnit tests */  in IWL_MLD_ALLOC_FN()
104 	struct ieee80211_vif *vif;  in IWL_MLD_ALLOC_FN()  local
111 	KUNIT_ALLOC_AND_ASSERT_SIZE(test, vif,  in IWL_MLD_ALLOC_FN()
112 				    sizeof(*vif) + sizeof(*mld_vif));  in IWL_MLD_ALLOC_FN()
114 	vif->type = type;  in IWL_MLD_ALLOC_FN()
115 	mld_vif = iwl_mld_vif_from_mac80211(vif);  in IWL_MLD_ALLOC_FN()
118 	ret = iwl_mld_allocate_vif_fw_id(mld, &mld_vif->fw_id, vif);  in IWL_MLD_ALLOC_FN()
123 		return vif;  in IWL_MLD_ALLOC_FN()
126 	iwlmld_kunit_init_link(vif, &vif->bss_conf, &mld_vif->deflink, 0);  in IWL_MLD_ALLOC_FN()
128 	return vif;  in IWL_MLD_ALLOC_FN()
131 /* Use only for MLO vif */
133 iwlmld_kunit_add_link(struct ieee80211_vif *vif, int link_id)  in iwlmld_kunit_add_link()  argument
142 	iwlmld_kunit_init_link(vif, link, mld_link, link_id);  in iwlmld_kunit_add_link()
143 	vif->valid_links |= BIT(link_id);  in iwlmld_kunit_add_link()
177 void iwlmld_kunit_assign_chanctx_to_link(struct ieee80211_vif *vif,  in iwlmld_kunit_assign_chanctx_to_link()  argument
198 	if (ieee80211_vif_is_mld(vif))  in iwlmld_kunit_assign_chanctx_to_link()
199 		vif->active_links |= BIT(link->link_id);  in iwlmld_kunit_assign_chanctx_to_link()
251 iwlmld_kunit_add_sta(struct ieee80211_vif *vif, int link_id)  in iwlmld_kunit_add_sta()  argument
263 	mld_sta->vif = vif;  in iwlmld_kunit_add_sta()
281 static void iwlmld_kunit_move_sta_state(struct ieee80211_vif *vif,  in iwlmld_kunit_move_sta_state()  argument
295 	mld_vif = iwl_mld_vif_from_mac80211(mld_sta->vif);  in iwlmld_kunit_move_sta_state()
298 	if (vif->type == NL80211_IFTYPE_STATION && !sta->tdls)  in iwlmld_kunit_move_sta_state()
302 struct ieee80211_sta *iwlmld_kunit_setup_sta(struct ieee80211_vif *vif,  in iwlmld_kunit_setup_sta()  argument
313 	sta = iwlmld_kunit_add_sta(vif, link_id);  in iwlmld_kunit_setup_sta()
318 		iwlmld_kunit_move_sta_state(vif, sta, state);  in iwlmld_kunit_setup_sta()
323 static void iwlmld_kunit_set_vif_associated(struct ieee80211_vif *vif)  in iwlmld_kunit_set_vif_associated()  argument
328 	vif->cfg.assoc = 1;  in iwlmld_kunit_set_vif_associated()
336 	struct ieee80211_vif *vif;  in iwlmld_kunit_setup_assoc()  local
342 	vif = iwlmld_kunit_add_vif(mlo, NL80211_IFTYPE_STATION);  in iwlmld_kunit_setup_assoc()
345 		link = iwlmld_kunit_add_link(vif, assoc_link->id);  in iwlmld_kunit_setup_assoc()
347 		link = &vif->bss_conf;  in iwlmld_kunit_setup_assoc()
352 	iwlmld_kunit_assign_chanctx_to_link(vif, link, chan_ctx);  in iwlmld_kunit_setup_assoc()
356 	iwlmld_kunit_setup_sta(vif, IEEE80211_STA_AUTHORIZED, assoc_link->id);  in iwlmld_kunit_setup_assoc()
358 	iwlmld_kunit_set_vif_associated(vif);  in iwlmld_kunit_setup_assoc()
360 	return vif;  in iwlmld_kunit_setup_assoc()
368 	struct ieee80211_vif *vif;  in iwlmld_kunit_setup_mlo_assoc()  local
375 	vif = iwlmld_kunit_setup_assoc(true, assoc_link);  in iwlmld_kunit_setup_mlo_assoc()
381 		iwlmld_kunit_add_link(vif, other_link_id);  in iwlmld_kunit_setup_mlo_assoc()
384 	return vif;  in iwlmld_kunit_setup_mlo_assoc()
412 	struct ieee80211_vif *vif;  in iwlmld_kunit_assoc_emlsr()  local
421 	vif = iwlmld_kunit_setup_mlo_assoc(valid_links, link1);  in iwlmld_kunit_assoc_emlsr()
422 	mld_vif = iwl_mld_vif_from_mac80211(vif);  in iwlmld_kunit_assoc_emlsr()
427 	link = wiphy_dereference(mld->wiphy, vif->link_conf[link2->id]);  in iwlmld_kunit_assoc_emlsr()
431 	iwlmld_kunit_assign_chanctx_to_link(vif, link, chan_ctx);  in iwlmld_kunit_assoc_emlsr()
441 	return vif;  in iwlmld_kunit_assoc_emlsr()
458 struct iwl_mld_phy *iwlmld_kunit_get_phy_of_link(struct ieee80211_vif *vif,  in iwlmld_kunit_get_phy_of_link()  argument
465 		wiphy_dereference(mld->wiphy, vif->link_conf[link_id]);  in iwlmld_kunit_get_phy_of_link()