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