[linux-block.git] / net / mac80211 / link.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * MLO link handling
 *
 * Copyright (C) 2022 Intel Corporation
 */
#include <linux/slab.h>
#include <linux/kernel.h>
#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "driver-ops.h"
#include "key.h"

void ieee80211_link_setup(struct ieee80211_link_data *link)
{
	if (link->sdata->vif.type == NL80211_IFTYPE_STATION)
		ieee80211_mgd_setup_link(link);
}

void ieee80211_link_init(struct ieee80211_sub_if_data *sdata,
			 int link_id,
			 struct ieee80211_link_data *link,
			 struct ieee80211_bss_conf *link_conf)
{
	bool deflink = link_id < 0;

	if (link_id < 0)
		link_id = 0;

	rcu_assign_pointer(sdata->vif.link_conf[link_id], link_conf);
	rcu_assign_pointer(sdata->link[link_id], link);

	link->sdata = sdata;
	link->link_id = link_id;
	link->conf = link_conf;
	link_conf->link_id = link_id;

	INIT_WORK(&link->csa_finalize_work,
		  ieee80211_csa_finalize_work);
	INIT_WORK(&link->color_change_finalize_work,
		  ieee80211_color_change_finalize_work);
	INIT_LIST_HEAD(&link->assigned_chanctx_list);
	INIT_LIST_HEAD(&link->reserved_chanctx_list);
	INIT_DELAYED_WORK(&link->dfs_cac_timer_work,
			  ieee80211_dfs_cac_timer_work);

	if (!deflink) {
		switch (sdata->vif.type) {
		case NL80211_IFTYPE_AP:
			ether_addr_copy(link_conf->addr,
					sdata->wdev.links[link_id].addr);
			link_conf->bssid = link_conf->addr;
			WARN_ON(!(sdata->wdev.valid_links & BIT(link_id)));
			break;
		case NL80211_IFTYPE_STATION:
			/* station sets the bssid in ieee80211_mgd_setup_link */
			break;
		default:
			WARN_ON(1);
		}
	}
}

void ieee80211_link_stop(struct ieee80211_link_data *link)
{
	if (link->sdata->vif.type == NL80211_IFTYPE_STATION)
		ieee80211_mgd_stop_link(link);

	ieee80211_link_release_channel(link);
}

struct link_container {
	struct ieee80211_link_data data;
	struct ieee80211_bss_conf conf;
};

static void ieee80211_tear_down_links(struct ieee80211_sub_if_data *sdata,
				      struct link_container **links, u16 mask)
{
	struct ieee80211_link_data *link;
	LIST_HEAD(keys);
	unsigned int link_id;

	for (link_id = 0; link_id < IEEE80211_MLD_MAX_NUM_LINKS; link_id++) {
		if (!(mask & BIT(link_id)))
			continue;
		link = &links[link_id]->data;
		if (link_id == 0 && !link)
			link = &sdata->deflink;
		if (WARN_ON(!link))
			continue;
		ieee80211_remove_link_keys(link, &keys);
		ieee80211_link_stop(link);
	}

	synchronize_rcu();

	ieee80211_free_key_list(sdata->local, &keys);
}

static void ieee80211_free_links(struct ieee80211_sub_if_data *sdata,
				 struct link_container **links)
{
	unsigned int link_id;

	for (link_id = 0; link_id < IEEE80211_MLD_MAX_NUM_LINKS; link_id++)
		kfree(links[link_id]);
}

static int ieee80211_check_dup_link_addrs(struct ieee80211_sub_if_data *sdata)
{
	unsigned int i, j;

	for (i = 0; i < IEEE80211_MLD_MAX_NUM_LINKS; i++) {
		struct ieee80211_link_data *link1;

		link1 = sdata_dereference(sdata->link[i], sdata);
		if (!link1)
			continue;
		for (j = i + 1; j < IEEE80211_MLD_MAX_NUM_LINKS; j++) {
			struct ieee80211_link_data *link2;

			link2 = sdata_dereference(sdata->link[j], sdata);
			if (!link2)
				continue;

			if (ether_addr_equal(link1->conf->addr,
					     link2->conf->addr))
				return -EALREADY;
		}
	}

	return 0;
}

static void ieee80211_set_vif_links_bitmaps(struct ieee80211_sub_if_data *sdata,
					    u16 links)
{
	sdata->vif.valid_links = links;

	if (!links) {
		sdata->vif.active_links = 0;
		return;
	}

	switch (sdata->vif.type) {
	case NL80211_IFTYPE_AP:
		/* in an AP all links are always active */
		sdata->vif.active_links = links;
		break;
	case NL80211_IFTYPE_STATION:
		if (sdata->vif.active_links)
			break;
		WARN_ON(hweight16(links) > 1);
		sdata->vif.active_links = links;
		break;
	default:
		WARN_ON(1);
	}
}

static int ieee80211_vif_update_links(struct ieee80211_sub_if_data *sdata,
				      struct link_container **to_free,
				      u16 new_links)
{
	u16 old_links = sdata->vif.valid_links;
	u16 old_active = sdata->vif.active_links;
	unsigned long add = new_links & ~old_links;
	unsigned long rem = old_links & ~new_links;
	unsigned int link_id;
	int ret;
	struct link_container *links[IEEE80211_MLD_MAX_NUM_LINKS] = {}, *link;
	struct ieee80211_bss_conf *old[IEEE80211_MLD_MAX_NUM_LINKS];
	struct ieee80211_link_data *old_data[IEEE80211_MLD_MAX_NUM_LINKS];
	bool use_deflink = old_links == 0; /* set for error case */

	sdata_assert_lock(sdata);

	memset(to_free, 0, sizeof(links));

	if (old_links == new_links)
		return 0;

	/* if there were no old links, need to clear the pointers to deflink */
	if (!old_links)
		rem |= BIT(0);

	/* allocate new link structures first */
	for_each_set_bit(link_id, &add, IEEE80211_MLD_MAX_NUM_LINKS) {
		link = kzalloc(sizeof(*link), GFP_KERNEL);
		if (!link) {
			ret = -ENOMEM;
			goto free;
		}
		links[link_id] = link;
	}

	/* keep track of the old pointers for the driver */
	BUILD_BUG_ON(sizeof(old) != sizeof(sdata->vif.link_conf));
	memcpy(old, sdata->vif.link_conf, sizeof(old));
	/* and for us in error cases */
	BUILD_BUG_ON(sizeof(old_data) != sizeof(sdata->link));
	memcpy(old_data, sdata->link, sizeof(old_data));

	/* grab old links to free later */
	for_each_set_bit(link_id, &rem, IEEE80211_MLD_MAX_NUM_LINKS) {
		if (rcu_access_pointer(sdata->link[link_id]) != &sdata->deflink) {
			/*
			 * we must have allocated the data through this path so
			 * we know we can free both at the same time
			 */
			to_free[link_id] = container_of(rcu_access_pointer(sdata->link[link_id]),
							typeof(*links[link_id]),
							data);
		}

		RCU_INIT_POINTER(sdata->link[link_id], NULL);
		RCU_INIT_POINTER(sdata->vif.link_conf[link_id], NULL);
	}

	/* link them into data structures */
	for_each_set_bit(link_id, &add, IEEE80211_MLD_MAX_NUM_LINKS) {
		WARN_ON(!use_deflink &&
			rcu_access_pointer(sdata->link[link_id]) == &sdata->deflink);

		link = links[link_id];
		ieee80211_link_init(sdata, link_id, &link->data, &link->conf);
		ieee80211_link_setup(&link->data);
	}

	if (new_links == 0)
		ieee80211_link_init(sdata, -1, &sdata->deflink,
				    &sdata->vif.bss_conf);

	ret = ieee80211_check_dup_link_addrs(sdata);
	if (!ret) {
		/* for keys we will not be able to undo this */
		ieee80211_tear_down_links(sdata, to_free, rem);

		ieee80211_set_vif_links_bitmaps(sdata, new_links);

		/* tell the driver */
		ret = drv_change_vif_links(sdata->local, sdata,
					   old_links & old_active,
					   new_links & sdata->vif.active_links,
					   old);
	}

	if (ret) {
		/* restore config */
		memcpy(sdata->link, old_data, sizeof(old_data));
		memcpy(sdata->vif.link_conf, old, sizeof(old));
		ieee80211_set_vif_links_bitmaps(sdata, old_links);
		/* and free (only) the newly allocated links */
		memset(to_free, 0, sizeof(links));
		goto free;
	}

	/* use deflink/bss_conf again if and only if there are no more links */
	use_deflink = new_links == 0;

	goto deinit;
free:
	/* if we failed during allocation, only free all */
	for (link_id = 0; link_id < IEEE80211_MLD_MAX_NUM_LINKS; link_id++) {
		kfree(links[link_id]);
		links[link_id] = NULL;
	}
deinit:
	if (use_deflink)
		ieee80211_link_init(sdata, -1, &sdata->deflink,
				    &sdata->vif.bss_conf);
	return ret;
}

int ieee80211_vif_set_links(struct ieee80211_sub_if_data *sdata,
			    u16 new_links)
{
	struct link_container *links[IEEE80211_MLD_MAX_NUM_LINKS];
	int ret;

	ret = ieee80211_vif_update_links(sdata, links, new_links);
	ieee80211_free_links(sdata, links);

	return ret;
}

void ieee80211_vif_clear_links(struct ieee80211_sub_if_data *sdata)
{
	struct link_container *links[IEEE80211_MLD_MAX_NUM_LINKS];

	/*
	 * The locking here is different because when we free links
	 * in the station case we need to be able to cancel_work_sync()
	 * something that also takes the lock.
	 */

	sdata_lock(sdata);
	ieee80211_vif_update_links(sdata, links, 0);
	sdata_unlock(sdata);

	ieee80211_free_links(sdata, links);
}

static int _ieee80211_set_active_links(struct ieee80211_sub_if_data *sdata,
				       u16 active_links)
{
	struct ieee80211_bss_conf *link_confs[IEEE80211_MLD_MAX_NUM_LINKS];
	struct ieee80211_local *local = sdata->local;
	u16 old_active = sdata->vif.active_links;
	unsigned long rem = old_active & ~active_links;
	unsigned long add = active_links & ~old_active;
	struct sta_info *sta;
	unsigned int link_id;
	int ret, i;

	if (!ieee80211_sdata_running(sdata))
		return -ENETDOWN;

	if (sdata->vif.type != NL80211_IFTYPE_STATION)
		return -EINVAL;

	/* cannot activate links that don't exist */
	if (active_links & ~sdata->vif.valid_links)
		return -EINVAL;

	/* nothing to do */
	if (old_active == active_links)
		return 0;

	for (i = 0; i < IEEE80211_MLD_MAX_NUM_LINKS; i++)
		link_confs[i] = sdata_dereference(sdata->vif.link_conf[i],
						  sdata);

	if (add) {
		sdata->vif.active_links |= active_links;
		ret = drv_change_vif_links(local, sdata,
					   old_active,
					   sdata->vif.active_links,
					   link_confs);
		if (ret) {
			sdata->vif.active_links = old_active;
			return ret;
		}
	}

	for_each_set_bit(link_id, &rem, IEEE80211_MLD_MAX_NUM_LINKS) {
		struct ieee80211_link_data *link;

		link = sdata_dereference(sdata->link[link_id], sdata);

		/* FIXME: kill TDLS connections on the link */

		ieee80211_link_release_channel(link);
	}

	list_for_each_entry(sta, &local->sta_list, list) {
		if (sdata != sta->sdata)
			continue;

		/* this is very temporary, but do it anyway */
		__ieee80211_sta_recalc_aggregates(sta,
						  old_active | active_links);

		ret = drv_change_sta_links(local, sdata, &sta->sta,
					   old_active,
					   old_active | active_links);
		WARN_ON_ONCE(ret);
	}

	ret = ieee80211_key_switch_links(sdata, rem, add);
	WARN_ON_ONCE(ret);

	list_for_each_entry(sta, &local->sta_list, list) {
		if (sdata != sta->sdata)
			continue;

		__ieee80211_sta_recalc_aggregates(sta, active_links);

		ret = drv_change_sta_links(local, sdata, &sta->sta,
					   old_active | active_links,
					   active_links);
		WARN_ON_ONCE(ret);

		/*
		 * Do it again, just in case - the driver might very
		 * well have called ieee80211_sta_recalc_aggregates()
		 * from there when filling in the new links, which
		 * would set it wrong since the vif's active links are
		 * not switched yet...
		 */
		__ieee80211_sta_recalc_aggregates(sta, active_links);
	}

	for_each_set_bit(link_id, &add, IEEE80211_MLD_MAX_NUM_LINKS) {
		struct ieee80211_link_data *link;

		link = sdata_dereference(sdata->link[link_id], sdata);

		ret = ieee80211_link_use_channel(link, &link->conf->chandef,
						 IEEE80211_CHANCTX_SHARED);
		WARN_ON_ONCE(ret);

		ieee80211_link_info_change_notify(sdata, link,
						  BSS_CHANGED_ERP_CTS_PROT |
						  BSS_CHANGED_ERP_PREAMBLE |
						  BSS_CHANGED_ERP_SLOT |
						  BSS_CHANGED_HT |
						  BSS_CHANGED_BASIC_RATES |
						  BSS_CHANGED_BSSID |
						  BSS_CHANGED_CQM |
						  BSS_CHANGED_QOS |
						  BSS_CHANGED_TXPOWER |
						  BSS_CHANGED_BANDWIDTH |
						  BSS_CHANGED_TWT |
						  BSS_CHANGED_HE_OBSS_PD |
						  BSS_CHANGED_HE_BSS_COLOR);
		ieee80211_mgd_set_link_qos_params(link);
	}

	old_active = sdata->vif.active_links;
	sdata->vif.active_links = active_links;

	if (rem) {
		ret = drv_change_vif_links(local, sdata, old_active,
					   active_links, link_confs);
		WARN_ON_ONCE(ret);
	}

	return 0;
}

int ieee80211_set_active_links(struct ieee80211_vif *vif, u16 active_links)
{
	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	struct ieee80211_local *local = sdata->local;
	u16 old_active;
	int ret;

	sdata_lock(sdata);
	mutex_lock(&local->sta_mtx);
	mutex_lock(&local->mtx);
	mutex_lock(&local->key_mtx);
	old_active = sdata->vif.active_links;
	if (old_active & active_links) {
		/*
		 * if there's at least one link that stays active across
		 * the change then switch to it (to those) first, and
		 * then enable the additional links
		 */
		ret = _ieee80211_set_active_links(sdata,
						  old_active & active_links);
		if (!ret)
			ret = _ieee80211_set_active_links(sdata, active_links);
	} else {
		/* otherwise switch directly */
		ret = _ieee80211_set_active_links(sdata, active_links);
	}
	mutex_unlock(&local->key_mtx);
	mutex_unlock(&local->mtx);
	mutex_unlock(&local->sta_mtx);
	sdata_unlock(sdata);

	return ret;
}
EXPORT_SYMBOL_GPL(ieee80211_set_active_links);

void ieee80211_set_active_links_async(struct ieee80211_vif *vif,
				      u16 active_links)
{
	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);

	if (!ieee80211_sdata_running(sdata))
		return;

	if (sdata->vif.type != NL80211_IFTYPE_STATION)
		return;

	/* cannot activate links that don't exist */
	if (active_links & ~sdata->vif.valid_links)
		return;

	/* nothing to do */
	if (sdata->vif.active_links == active_links)
		return;

	sdata->desired_active_links = active_links;
	schedule_work(&sdata->activate_links_work);
}
EXPORT_SYMBOL_GPL(ieee80211_set_active_links_async);
Commit	Line	Data
e73b5e51 JB	1	// SPDX-License-Identifier: GPL-2.0-only
	2	/*
	3	* MLO link handling
	4	*
	5	* Copyright (C) 2022 Intel Corporation
	6	*/
	7	#include <linux/slab.h>
	8	#include <linux/kernel.h>
	9	#include <net/mac80211.h>
	10	#include "ieee80211_i.h"
	11	#include "driver-ops.h"
3d901102	12	#include "key.h"
e73b5e51 JB	13
	14	void ieee80211_link_setup(struct ieee80211_link_data *link)
	15	{
	16	if (link->sdata->vif.type == NL80211_IFTYPE_STATION)
	17	ieee80211_mgd_setup_link(link);
	18	}
	19
	20	void ieee80211_link_init(struct ieee80211_sub_if_data *sdata,
	21	int link_id,
	22	struct ieee80211_link_data *link,
	23	struct ieee80211_bss_conf *link_conf)
	24	{
	25	bool deflink = link_id < 0;
	26
	27	if (link_id < 0)
	28	link_id = 0;
	29
	30	rcu_assign_pointer(sdata->vif.link_conf[link_id], link_conf);
	31	rcu_assign_pointer(sdata->link[link_id], link);
	32
	33	link->sdata = sdata;
	34	link->link_id = link_id;
	35	link->conf = link_conf;
	36	link_conf->link_id = link_id;
	37
	38	INIT_WORK(&link->csa_finalize_work,
	39	ieee80211_csa_finalize_work);
	40	INIT_WORK(&link->color_change_finalize_work,
	41	ieee80211_color_change_finalize_work);
	42	INIT_LIST_HEAD(&link->assigned_chanctx_list);
	43	INIT_LIST_HEAD(&link->reserved_chanctx_list);
	44	INIT_DELAYED_WORK(&link->dfs_cac_timer_work,
	45	ieee80211_dfs_cac_timer_work);
	46
	47	if (!deflink) {
	48	switch (sdata->vif.type) {
	49	case NL80211_IFTYPE_AP:
	50	ether_addr_copy(link_conf->addr,
	51	sdata->wdev.links[link_id].addr);
	52	link_conf->bssid = link_conf->addr;
	53	WARN_ON(!(sdata->wdev.valid_links & BIT(link_id)));
	54	break;
	55	case NL80211_IFTYPE_STATION:
	56	/* station sets the bssid in ieee80211_mgd_setup_link */
	57	break;
	58	default:
	59	WARN_ON(1);
	60	}
	61	}
	62	}
	63
	64	void ieee80211_link_stop(struct ieee80211_link_data *link)
	65	{
	66	if (link->sdata->vif.type == NL80211_IFTYPE_STATION)
	67	ieee80211_mgd_stop_link(link);
	68
	69	ieee80211_link_release_channel(link);
	70	}
	71
	72	struct link_container {
	73	struct ieee80211_link_data data;
	74	struct ieee80211_bss_conf conf;
	75	};
	76
efe9c2bf JB	77	static void ieee80211_tear_down_links(struct ieee80211_sub_if_data *sdata,
efe9c2bf JB	78	struct link_container **links, u16 mask)
e73b5e51	79	{
efe9c2bf	80	struct ieee80211_link_data *link;
e73b5e51 JB	81	LIST_HEAD(keys);
	82	unsigned int link_id;
	83
	84	for (link_id = 0; link_id < IEEE80211_MLD_MAX_NUM_LINKS; link_id++) {
efe9c2bf JB	85	if (!(mask & BIT(link_id)))
	86	continue;
	87	link = &links[link_id]->data;
	88	if (link_id == 0 && !link)
	89	link = &sdata->deflink;
	90	if (WARN_ON(!link))
e73b5e51	91	continue;
efe9c2bf JB	92	ieee80211_remove_link_keys(link, &keys);
efe9c2bf JB	93	ieee80211_link_stop(link);
e73b5e51 JB	94	}
	95
	96	synchronize_rcu();
	97
	98	ieee80211_free_key_list(sdata->local, &keys);
efe9c2bf	99	}
e73b5e51	100
efe9c2bf JB	101	static void ieee80211_free_links(struct ieee80211_sub_if_data *sdata,
	102	struct link_container **links)
	103	{
	104	unsigned int link_id;
	105
	106	for (link_id = 0; link_id < IEEE80211_MLD_MAX_NUM_LINKS; link_id++)
e73b5e51	107	kfree(links[link_id]);
e73b5e51 JB	108	}
	109
	110	static int ieee80211_check_dup_link_addrs(struct ieee80211_sub_if_data *sdata)
	111	{
	112	unsigned int i, j;
	113
	114	for (i = 0; i < IEEE80211_MLD_MAX_NUM_LINKS; i++) {
	115	struct ieee80211_link_data *link1;
	116
	117	link1 = sdata_dereference(sdata->link[i], sdata);
	118	if (!link1)
	119	continue;
	120	for (j = i + 1; j < IEEE80211_MLD_MAX_NUM_LINKS; j++) {
	121	struct ieee80211_link_data *link2;
	122
	123	link2 = sdata_dereference(sdata->link[j], sdata);
	124	if (!link2)
	125	continue;
	126
	127	if (ether_addr_equal(link1->conf->addr,
	128	link2->conf->addr))
	129	return -EALREADY;
	130	}
	131	}
	132
	133	return 0;
	134	}
	135
efe9c2bf JB	136	static void ieee80211_set_vif_links_bitmaps(struct ieee80211_sub_if_data *sdata,
	137	u16 links)
	138	{
	139	sdata->vif.valid_links = links;
	140
	141	if (!links) {
	142	sdata->vif.active_links = 0;
	143	return;
	144	}
	145
	146	switch (sdata->vif.type) {
	147	case NL80211_IFTYPE_AP:
	148	/* in an AP all links are always active */
	149	sdata->vif.active_links = links;
	150	break;
	151	case NL80211_IFTYPE_STATION:
	152	if (sdata->vif.active_links)
	153	break;
	154	WARN_ON(hweight16(links) > 1);
	155	sdata->vif.active_links = links;
	156	break;
	157	default:
	158	WARN_ON(1);
	159	}
	160	}
	161
e73b5e51 JB	162	static int ieee80211_vif_update_links(struct ieee80211_sub_if_data *sdata,
	163	struct link_container **to_free,
	164	u16 new_links)
	165	{
	166	u16 old_links = sdata->vif.valid_links;
efe9c2bf	167	u16 old_active = sdata->vif.active_links;
e73b5e51 JB	168	unsigned long add = new_links & ~old_links;
	169	unsigned long rem = old_links & ~new_links;
	170	unsigned int link_id;
	171	int ret;
	172	struct link_container links[IEEE80211_MLD_MAX_NUM_LINKS] = {}, link;
	173	struct ieee80211_bss_conf *old[IEEE80211_MLD_MAX_NUM_LINKS];
	174	struct ieee80211_link_data *old_data[IEEE80211_MLD_MAX_NUM_LINKS];
	175	bool use_deflink = old_links == 0; /* set for error case */
	176
	177	sdata_assert_lock(sdata);
	178
	179	memset(to_free, 0, sizeof(links));
	180
	181	if (old_links == new_links)
	182	return 0;
	183
	184	/* if there were no old links, need to clear the pointers to deflink */
	185	if (!old_links)
	186	rem \|= BIT(0);
	187
	188	/* allocate new link structures first */
	189	for_each_set_bit(link_id, &add, IEEE80211_MLD_MAX_NUM_LINKS) {
	190	link = kzalloc(sizeof(*link), GFP_KERNEL);
	191	if (!link) {
	192	ret = -ENOMEM;
	193	goto free;
	194	}
	195	links[link_id] = link;
	196	}
	197
	198	/* keep track of the old pointers for the driver */
	199	BUILD_BUG_ON(sizeof(old) != sizeof(sdata->vif.link_conf));
	200	memcpy(old, sdata->vif.link_conf, sizeof(old));
	201	/* and for us in error cases */
	202	BUILD_BUG_ON(sizeof(old_data) != sizeof(sdata->link));
	203	memcpy(old_data, sdata->link, sizeof(old_data));
	204
	205	/* grab old links to free later */
	206	for_each_set_bit(link_id, &rem, IEEE80211_MLD_MAX_NUM_LINKS) {
	207	if (rcu_access_pointer(sdata->link[link_id]) != &sdata->deflink) {
	208	/*
	209	* we must have allocated the data through this path so
	210	* we know we can free both at the same time
	211	*/
	212	to_free[link_id] = container_of(rcu_access_pointer(sdata->link[link_id]),
	213	typeof(*links[link_id]),
	214	data);
	215	}
	216
	217	RCU_INIT_POINTER(sdata->link[link_id], NULL);
	218	RCU_INIT_POINTER(sdata->vif.link_conf[link_id], NULL);
	219	}
	220
	221	/* link them into data structures */
	222	for_each_set_bit(link_id, &add, IEEE80211_MLD_MAX_NUM_LINKS) {
	223	WARN_ON(!use_deflink &&
	224	rcu_access_pointer(sdata->link[link_id]) == &sdata->deflink);
	225
	226	link = links[link_id];
	227	ieee80211_link_init(sdata, link_id, &link->data, &link->conf);
	228	ieee80211_link_setup(&link->data);
	229	}
	230
	231	if (new_links == 0)
232	ieee80211_link_init(sdata, -1, &sdata->deflink,
233	&sdata->vif.bss_conf);
234
e73b5e51 JB	235	ret = ieee80211_check_dup_link_addrs(sdata);
e73b5e51 JB	236	if (!ret) {
efe9c2bf JB	237	/* for keys we will not be able to undo this */
	238	ieee80211_tear_down_links(sdata, to_free, rem);
	239
	240	ieee80211_set_vif_links_bitmaps(sdata, new_links);
	241
e73b5e51 JB	242	/* tell the driver */
e73b5e51 JB	243	ret = drv_change_vif_links(sdata->local, sdata,
efe9c2bf JB	244	old_links & old_active,
efe9c2bf JB	245	new_links & sdata->vif.active_links,
e73b5e51 JB	246	old);
	247	}
	248
	249	if (ret) {
	250	/* restore config */
	251	memcpy(sdata->link, old_data, sizeof(old_data));
	252	memcpy(sdata->vif.link_conf, old, sizeof(old));
efe9c2bf	253	ieee80211_set_vif_links_bitmaps(sdata, old_links);
e73b5e51 JB	254	/* and free (only) the newly allocated links */
	255	memset(to_free, 0, sizeof(links));
	256	goto free;
	257	}
	258
	259	/* use deflink/bss_conf again if and only if there are no more links */
	260	use_deflink = new_links == 0;
	261
	262	goto deinit;
	263	free:
	264	/* if we failed during allocation, only free all */
	265	for (link_id = 0; link_id < IEEE80211_MLD_MAX_NUM_LINKS; link_id++) {
	266	kfree(links[link_id]);
	267	links[link_id] = NULL;
	268	}
	269	deinit:
	270	if (use_deflink)
	271	ieee80211_link_init(sdata, -1, &sdata->deflink,
	272	&sdata->vif.bss_conf);
	273	return ret;
	274	}
	275
	276	int ieee80211_vif_set_links(struct ieee80211_sub_if_data *sdata,
	277	u16 new_links)
	278	{
	279	struct link_container *links[IEEE80211_MLD_MAX_NUM_LINKS];
	280	int ret;
	281
	282	ret = ieee80211_vif_update_links(sdata, links, new_links);
	283	ieee80211_free_links(sdata, links);
	284
	285	return ret;
	286	}
	287
	288	void ieee80211_vif_clear_links(struct ieee80211_sub_if_data *sdata)
	289	{
	290	struct link_container *links[IEEE80211_MLD_MAX_NUM_LINKS];
	291
	292	/*
	293	* The locking here is different because when we free links
	294	* in the station case we need to be able to cancel_work_sync()
	295	* something that also takes the lock.
	296	*/
	297
	298	sdata_lock(sdata);
	299	ieee80211_vif_update_links(sdata, links, 0);
	300	sdata_unlock(sdata);
	301
	302	ieee80211_free_links(sdata, links);
	303	}
3d901102 JB	304
	305	static int _ieee80211_set_active_links(struct ieee80211_sub_if_data *sdata,
	306	u16 active_links)
	307	{
	308	struct ieee80211_bss_conf *link_confs[IEEE80211_MLD_MAX_NUM_LINKS];
	309	struct ieee80211_local *local = sdata->local;
	310	u16 old_active = sdata->vif.active_links;
	311	unsigned long rem = old_active & ~active_links;
	312	unsigned long add = active_links & ~old_active;
	313	struct sta_info *sta;
	314	unsigned int link_id;
	315	int ret, i;
	316
	317	if (!ieee80211_sdata_running(sdata))
	318	return -ENETDOWN;
	319
	320	if (sdata->vif.type != NL80211_IFTYPE_STATION)
	321	return -EINVAL;
	322
	323	/* cannot activate links that don't exist */
	324	if (active_links & ~sdata->vif.valid_links)
	325	return -EINVAL;
	326
	327	/* nothing to do */
	328	if (old_active == active_links)
	329	return 0;
	330
	331	for (i = 0; i < IEEE80211_MLD_MAX_NUM_LINKS; i++)
	332	link_confs[i] = sdata_dereference(sdata->vif.link_conf[i],
	333	sdata);
	334
	335	if (add) {
	336	sdata->vif.active_links \|= active_links;
	337	ret = drv_change_vif_links(local, sdata,
	338	old_active,
	339	sdata->vif.active_links,
	340	link_confs);
	341	if (ret) {
	342	sdata->vif.active_links = old_active;
	343	return ret;
	344	}
	345	}
	346
	347	for_each_set_bit(link_id, &rem, IEEE80211_MLD_MAX_NUM_LINKS) {
	348	struct ieee80211_link_data *link;
	349
	350	link = sdata_dereference(sdata->link[link_id], sdata);
	351
	352	/* FIXME: kill TDLS connections on the link */
	353
	354	ieee80211_link_release_channel(link);
	355	}
	356
	357	list_for_each_entry(sta, &local->sta_list, list) {
	358	if (sdata != sta->sdata)
	359	continue;
9b41a9d7 JB	360
	361	/* this is very temporary, but do it anyway */
	362	__ieee80211_sta_recalc_aggregates(sta,
	363	old_active \| active_links);
	364
3d901102 JB	365	ret = drv_change_sta_links(local, sdata, &sta->sta,
	366	old_active,
	367	old_active \| active_links);
	368	WARN_ON_ONCE(ret);
	369	}
	370
	371	ret = ieee80211_key_switch_links(sdata, rem, add);
	372	WARN_ON_ONCE(ret);
	373
	374	list_for_each_entry(sta, &local->sta_list, list) {
	375	if (sdata != sta->sdata)
	376	continue;
9b41a9d7 JB	377
	378	__ieee80211_sta_recalc_aggregates(sta, active_links);
	379
3d901102 JB	380	ret = drv_change_sta_links(local, sdata, &sta->sta,
	381	old_active \| active_links,
	382	active_links);
	383	WARN_ON_ONCE(ret);
9b41a9d7 JB	384
	385	/*
	386	* Do it again, just in case - the driver might very
	387	* well have called ieee80211_sta_recalc_aggregates()
	388	* from there when filling in the new links, which
	389	* would set it wrong since the vif's active links are
	390	* not switched yet...
	391	*/
	392	__ieee80211_sta_recalc_aggregates(sta, active_links);
3d901102 JB	393	}
	394
	395	for_each_set_bit(link_id, &add, IEEE80211_MLD_MAX_NUM_LINKS) {
	396	struct ieee80211_link_data *link;
	397
	398	link = sdata_dereference(sdata->link[link_id], sdata);
	399
	400	ret = ieee80211_link_use_channel(link, &link->conf->chandef,
	401	IEEE80211_CHANCTX_SHARED);
	402	WARN_ON_ONCE(ret);
	403
	404	ieee80211_link_info_change_notify(sdata, link,
	405	BSS_CHANGED_ERP_CTS_PROT \|
	406	BSS_CHANGED_ERP_PREAMBLE \|
	407	BSS_CHANGED_ERP_SLOT \|
	408	BSS_CHANGED_HT \|
	409	BSS_CHANGED_BASIC_RATES \|
	410	BSS_CHANGED_BSSID \|
	411	BSS_CHANGED_CQM \|
	412	BSS_CHANGED_QOS \|
	413	BSS_CHANGED_TXPOWER \|
	414	BSS_CHANGED_BANDWIDTH \|
	415	BSS_CHANGED_TWT \|
	416	BSS_CHANGED_HE_OBSS_PD \|
	417	BSS_CHANGED_HE_BSS_COLOR);
	418	ieee80211_mgd_set_link_qos_params(link);
	419	}
	420
	421	old_active = sdata->vif.active_links;
	422	sdata->vif.active_links = active_links;
	423
	424	if (rem) {
	425	ret = drv_change_vif_links(local, sdata, old_active,
	426	active_links, link_confs);
	427	WARN_ON_ONCE(ret);
	428	}
	429
	430	return 0;
	431	}
	432
	433	int ieee80211_set_active_links(struct ieee80211_vif *vif, u16 active_links)
	434	{
	435	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	436	struct ieee80211_local *local = sdata->local;
	437	u16 old_active;
	438	int ret;
	439
	440	sdata_lock(sdata);
	441	mutex_lock(&local->sta_mtx);
	442	mutex_lock(&local->mtx);
	443	mutex_lock(&local->key_mtx);
	444	old_active = sdata->vif.active_links;
	445	if (old_active & active_links) {
	446	/*
	447	* if there's at least one link that stays active across
	448	* the change then switch to it (to those) first, and
	449	* then enable the additional links
	450	*/
	451	ret = _ieee80211_set_active_links(sdata,
	452	old_active & active_links);
	453	if (!ret)
	454	ret = _ieee80211_set_active_links(sdata, active_links);
	455	} else {
	456	/* otherwise switch directly */
457	ret = _ieee80211_set_active_links(sdata, active_links);
458	}
459	mutex_unlock(&local->key_mtx);
460	mutex_unlock(&local->mtx);
461	mutex_unlock(&local->sta_mtx);
462	sdata_unlock(sdata);
463
464	return ret;
465	}
466	EXPORT_SYMBOL_GPL(ieee80211_set_active_links);
467
468	void ieee80211_set_active_links_async(struct ieee80211_vif *vif,
469	u16 active_links)
470	{
471	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
472
473	if (!ieee80211_sdata_running(sdata))
474	return;
475
476	if (sdata->vif.type != NL80211_IFTYPE_STATION)
477	return;
478
479	/* cannot activate links that don't exist */
480	if (active_links & ~sdata->vif.valid_links)
481	return;
482
483	/* nothing to do */
484	if (sdata->vif.active_links == active_links)
485	return;
486
487	sdata->desired_active_links = active_links;
488	schedule_work(&sdata->activate_links_work);
489	}
490	EXPORT_SYMBOL_GPL(ieee80211_set_active_links_async);