]> git.karo-electronics.de Git - karo-tx-linux.git/blob - net/mac80211/chan.c
Merge remote-tracking branch 'usb-chipidea-next/ci-for-usb-next'
[karo-tx-linux.git] / net / mac80211 / chan.c
1 /*
2  * mac80211 - channel management
3  */
4
5 #include <linux/nl80211.h>
6 #include <linux/export.h>
7 #include <linux/rtnetlink.h>
8 #include <net/cfg80211.h>
9 #include "ieee80211_i.h"
10 #include "driver-ops.h"
11
12 static int ieee80211_chanctx_num_assigned(struct ieee80211_local *local,
13                                           struct ieee80211_chanctx *ctx)
14 {
15         struct ieee80211_sub_if_data *sdata;
16         int num = 0;
17
18         lockdep_assert_held(&local->chanctx_mtx);
19
20         list_for_each_entry(sdata, &ctx->assigned_vifs, assigned_chanctx_list)
21                 num++;
22
23         return num;
24 }
25
26 static int ieee80211_chanctx_num_reserved(struct ieee80211_local *local,
27                                           struct ieee80211_chanctx *ctx)
28 {
29         struct ieee80211_sub_if_data *sdata;
30         int num = 0;
31
32         lockdep_assert_held(&local->chanctx_mtx);
33
34         list_for_each_entry(sdata, &ctx->reserved_vifs, reserved_chanctx_list)
35                 num++;
36
37         return num;
38 }
39
40 int ieee80211_chanctx_refcount(struct ieee80211_local *local,
41                                struct ieee80211_chanctx *ctx)
42 {
43         return ieee80211_chanctx_num_assigned(local, ctx) +
44                ieee80211_chanctx_num_reserved(local, ctx);
45 }
46
47 static int ieee80211_num_chanctx(struct ieee80211_local *local)
48 {
49         struct ieee80211_chanctx *ctx;
50         int num = 0;
51
52         lockdep_assert_held(&local->chanctx_mtx);
53
54         list_for_each_entry(ctx, &local->chanctx_list, list)
55                 num++;
56
57         return num;
58 }
59
60 static bool ieee80211_can_create_new_chanctx(struct ieee80211_local *local)
61 {
62         lockdep_assert_held(&local->chanctx_mtx);
63         return ieee80211_num_chanctx(local) < ieee80211_max_num_channels(local);
64 }
65
66 static struct ieee80211_chanctx *
67 ieee80211_vif_get_chanctx(struct ieee80211_sub_if_data *sdata)
68 {
69         struct ieee80211_local *local __maybe_unused = sdata->local;
70         struct ieee80211_chanctx_conf *conf;
71
72         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
73                                          lockdep_is_held(&local->chanctx_mtx));
74         if (!conf)
75                 return NULL;
76
77         return container_of(conf, struct ieee80211_chanctx, conf);
78 }
79
80 static const struct cfg80211_chan_def *
81 ieee80211_chanctx_reserved_chandef(struct ieee80211_local *local,
82                                    struct ieee80211_chanctx *ctx,
83                                    const struct cfg80211_chan_def *compat)
84 {
85         struct ieee80211_sub_if_data *sdata;
86
87         lockdep_assert_held(&local->chanctx_mtx);
88
89         list_for_each_entry(sdata, &ctx->reserved_vifs,
90                             reserved_chanctx_list) {
91                 if (!compat)
92                         compat = &sdata->reserved_chandef;
93
94                 compat = cfg80211_chandef_compatible(&sdata->reserved_chandef,
95                                                      compat);
96                 if (!compat)
97                         break;
98         }
99
100         return compat;
101 }
102
103 static const struct cfg80211_chan_def *
104 ieee80211_chanctx_non_reserved_chandef(struct ieee80211_local *local,
105                                        struct ieee80211_chanctx *ctx,
106                                        const struct cfg80211_chan_def *compat)
107 {
108         struct ieee80211_sub_if_data *sdata;
109
110         lockdep_assert_held(&local->chanctx_mtx);
111
112         list_for_each_entry(sdata, &ctx->assigned_vifs,
113                             assigned_chanctx_list) {
114                 if (sdata->reserved_chanctx != NULL)
115                         continue;
116
117                 if (!compat)
118                         compat = &sdata->vif.bss_conf.chandef;
119
120                 compat = cfg80211_chandef_compatible(
121                                 &sdata->vif.bss_conf.chandef, compat);
122                 if (!compat)
123                         break;
124         }
125
126         return compat;
127 }
128
129 static const struct cfg80211_chan_def *
130 ieee80211_chanctx_combined_chandef(struct ieee80211_local *local,
131                                    struct ieee80211_chanctx *ctx,
132                                    const struct cfg80211_chan_def *compat)
133 {
134         lockdep_assert_held(&local->chanctx_mtx);
135
136         compat = ieee80211_chanctx_reserved_chandef(local, ctx, compat);
137         if (!compat)
138                 return NULL;
139
140         compat = ieee80211_chanctx_non_reserved_chandef(local, ctx, compat);
141         if (!compat)
142                 return NULL;
143
144         return compat;
145 }
146
147 static bool
148 ieee80211_chanctx_can_reserve_chandef(struct ieee80211_local *local,
149                                       struct ieee80211_chanctx *ctx,
150                                       const struct cfg80211_chan_def *def)
151 {
152         lockdep_assert_held(&local->chanctx_mtx);
153
154         if (ieee80211_chanctx_combined_chandef(local, ctx, def))
155                 return true;
156
157         if (!list_empty(&ctx->reserved_vifs) &&
158             ieee80211_chanctx_reserved_chandef(local, ctx, def))
159                 return true;
160
161         return false;
162 }
163
164 static struct ieee80211_chanctx *
165 ieee80211_find_reservation_chanctx(struct ieee80211_local *local,
166                                    const struct cfg80211_chan_def *chandef,
167                                    enum ieee80211_chanctx_mode mode)
168 {
169         struct ieee80211_chanctx *ctx;
170
171         lockdep_assert_held(&local->chanctx_mtx);
172
173         if (mode == IEEE80211_CHANCTX_EXCLUSIVE)
174                 return NULL;
175
176         list_for_each_entry(ctx, &local->chanctx_list, list) {
177                 if (ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED)
178                         continue;
179
180                 if (ctx->mode == IEEE80211_CHANCTX_EXCLUSIVE)
181                         continue;
182
183                 if (!ieee80211_chanctx_can_reserve_chandef(local, ctx,
184                                                            chandef))
185                         continue;
186
187                 return ctx;
188         }
189
190         return NULL;
191 }
192
193 enum nl80211_chan_width ieee80211_get_sta_bw(struct ieee80211_sta *sta)
194 {
195         switch (sta->bandwidth) {
196         case IEEE80211_STA_RX_BW_20:
197                 if (sta->ht_cap.ht_supported)
198                         return NL80211_CHAN_WIDTH_20;
199                 else
200                         return NL80211_CHAN_WIDTH_20_NOHT;
201         case IEEE80211_STA_RX_BW_40:
202                 return NL80211_CHAN_WIDTH_40;
203         case IEEE80211_STA_RX_BW_80:
204                 return NL80211_CHAN_WIDTH_80;
205         case IEEE80211_STA_RX_BW_160:
206                 /*
207                  * This applied for both 160 and 80+80. since we use
208                  * the returned value to consider degradation of
209                  * ctx->conf.min_def, we have to make sure to take
210                  * the bigger one (NL80211_CHAN_WIDTH_160).
211                  * Otherwise we might try degrading even when not
212                  * needed, as the max required sta_bw returned (80+80)
213                  * might be smaller than the configured bw (160).
214                  */
215                 return NL80211_CHAN_WIDTH_160;
216         default:
217                 WARN_ON(1);
218                 return NL80211_CHAN_WIDTH_20;
219         }
220 }
221
222 static enum nl80211_chan_width
223 ieee80211_get_max_required_bw(struct ieee80211_sub_if_data *sdata)
224 {
225         enum nl80211_chan_width max_bw = NL80211_CHAN_WIDTH_20_NOHT;
226         struct sta_info *sta;
227
228         rcu_read_lock();
229         list_for_each_entry_rcu(sta, &sdata->local->sta_list, list) {
230                 if (sdata != sta->sdata &&
231                     !(sta->sdata->bss && sta->sdata->bss == sdata->bss))
232                         continue;
233
234                 if (!sta->uploaded || !test_sta_flag(sta, WLAN_STA_ASSOC))
235                         continue;
236
237                 max_bw = max(max_bw, ieee80211_get_sta_bw(&sta->sta));
238         }
239         rcu_read_unlock();
240
241         return max_bw;
242 }
243
244 static enum nl80211_chan_width
245 ieee80211_get_chanctx_max_required_bw(struct ieee80211_local *local,
246                                       struct ieee80211_chanctx_conf *conf)
247 {
248         struct ieee80211_sub_if_data *sdata;
249         enum nl80211_chan_width max_bw = NL80211_CHAN_WIDTH_20_NOHT;
250
251         rcu_read_lock();
252         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
253                 struct ieee80211_vif *vif = &sdata->vif;
254                 enum nl80211_chan_width width = NL80211_CHAN_WIDTH_20_NOHT;
255
256                 if (!ieee80211_sdata_running(sdata))
257                         continue;
258
259                 if (rcu_access_pointer(sdata->vif.chanctx_conf) != conf)
260                         continue;
261
262                 switch (vif->type) {
263                 case NL80211_IFTYPE_AP:
264                 case NL80211_IFTYPE_AP_VLAN:
265                         width = ieee80211_get_max_required_bw(sdata);
266                         break;
267                 case NL80211_IFTYPE_STATION:
268                         /*
269                          * The ap's sta->bandwidth is not set yet at this
270                          * point, so take the width from the chandef, but
271                          * account also for TDLS peers
272                          */
273                         width = max(vif->bss_conf.chandef.width,
274                                     ieee80211_get_max_required_bw(sdata));
275                         break;
276                 case NL80211_IFTYPE_P2P_DEVICE:
277                         continue;
278                 case NL80211_IFTYPE_ADHOC:
279                 case NL80211_IFTYPE_WDS:
280                 case NL80211_IFTYPE_MESH_POINT:
281                 case NL80211_IFTYPE_OCB:
282                         width = vif->bss_conf.chandef.width;
283                         break;
284                 case NL80211_IFTYPE_UNSPECIFIED:
285                 case NUM_NL80211_IFTYPES:
286                 case NL80211_IFTYPE_MONITOR:
287                 case NL80211_IFTYPE_P2P_CLIENT:
288                 case NL80211_IFTYPE_P2P_GO:
289                         WARN_ON_ONCE(1);
290                 }
291                 max_bw = max(max_bw, width);
292         }
293
294         /* use the configured bandwidth in case of monitor interface */
295         sdata = rcu_dereference(local->monitor_sdata);
296         if (sdata && rcu_access_pointer(sdata->vif.chanctx_conf) == conf)
297                 max_bw = max(max_bw, conf->def.width);
298
299         rcu_read_unlock();
300
301         return max_bw;
302 }
303
304 /*
305  * recalc the min required chan width of the channel context, which is
306  * the max of min required widths of all the interfaces bound to this
307  * channel context.
308  */
309 void ieee80211_recalc_chanctx_min_def(struct ieee80211_local *local,
310                                       struct ieee80211_chanctx *ctx)
311 {
312         enum nl80211_chan_width max_bw;
313         struct cfg80211_chan_def min_def;
314
315         lockdep_assert_held(&local->chanctx_mtx);
316
317         /* don't optimize 5MHz, 10MHz, and radar_enabled confs */
318         if (ctx->conf.def.width == NL80211_CHAN_WIDTH_5 ||
319             ctx->conf.def.width == NL80211_CHAN_WIDTH_10 ||
320             ctx->conf.radar_enabled) {
321                 ctx->conf.min_def = ctx->conf.def;
322                 return;
323         }
324
325         max_bw = ieee80211_get_chanctx_max_required_bw(local, &ctx->conf);
326
327         /* downgrade chandef up to max_bw */
328         min_def = ctx->conf.def;
329         while (min_def.width > max_bw)
330                 ieee80211_chandef_downgrade(&min_def);
331
332         if (cfg80211_chandef_identical(&ctx->conf.min_def, &min_def))
333                 return;
334
335         ctx->conf.min_def = min_def;
336         if (!ctx->driver_present)
337                 return;
338
339         drv_change_chanctx(local, ctx, IEEE80211_CHANCTX_CHANGE_MIN_WIDTH);
340 }
341
342 static void ieee80211_change_chanctx(struct ieee80211_local *local,
343                                      struct ieee80211_chanctx *ctx,
344                                      const struct cfg80211_chan_def *chandef)
345 {
346         if (cfg80211_chandef_identical(&ctx->conf.def, chandef))
347                 return;
348
349         WARN_ON(!cfg80211_chandef_compatible(&ctx->conf.def, chandef));
350
351         ctx->conf.def = *chandef;
352         drv_change_chanctx(local, ctx, IEEE80211_CHANCTX_CHANGE_WIDTH);
353         ieee80211_recalc_chanctx_min_def(local, ctx);
354
355         if (!local->use_chanctx) {
356                 local->_oper_chandef = *chandef;
357                 ieee80211_hw_config(local, 0);
358         }
359 }
360
361 static struct ieee80211_chanctx *
362 ieee80211_find_chanctx(struct ieee80211_local *local,
363                        const struct cfg80211_chan_def *chandef,
364                        enum ieee80211_chanctx_mode mode)
365 {
366         struct ieee80211_chanctx *ctx;
367
368         lockdep_assert_held(&local->chanctx_mtx);
369
370         if (mode == IEEE80211_CHANCTX_EXCLUSIVE)
371                 return NULL;
372
373         list_for_each_entry(ctx, &local->chanctx_list, list) {
374                 const struct cfg80211_chan_def *compat;
375
376                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACE_NONE)
377                         continue;
378
379                 if (ctx->mode == IEEE80211_CHANCTX_EXCLUSIVE)
380                         continue;
381
382                 compat = cfg80211_chandef_compatible(&ctx->conf.def, chandef);
383                 if (!compat)
384                         continue;
385
386                 compat = ieee80211_chanctx_reserved_chandef(local, ctx,
387                                                             compat);
388                 if (!compat)
389                         continue;
390
391                 ieee80211_change_chanctx(local, ctx, compat);
392
393                 return ctx;
394         }
395
396         return NULL;
397 }
398
399 bool ieee80211_is_radar_required(struct ieee80211_local *local)
400 {
401         struct ieee80211_sub_if_data *sdata;
402
403         lockdep_assert_held(&local->mtx);
404
405         rcu_read_lock();
406         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
407                 if (sdata->radar_required) {
408                         rcu_read_unlock();
409                         return true;
410                 }
411         }
412         rcu_read_unlock();
413
414         return false;
415 }
416
417 static bool
418 ieee80211_chanctx_radar_required(struct ieee80211_local *local,
419                                  struct ieee80211_chanctx *ctx)
420 {
421         struct ieee80211_chanctx_conf *conf = &ctx->conf;
422         struct ieee80211_sub_if_data *sdata;
423         bool required = false;
424
425         lockdep_assert_held(&local->chanctx_mtx);
426         lockdep_assert_held(&local->mtx);
427
428         rcu_read_lock();
429         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
430                 if (!ieee80211_sdata_running(sdata))
431                         continue;
432                 if (rcu_access_pointer(sdata->vif.chanctx_conf) != conf)
433                         continue;
434                 if (!sdata->radar_required)
435                         continue;
436
437                 required = true;
438                 break;
439         }
440         rcu_read_unlock();
441
442         return required;
443 }
444
445 static struct ieee80211_chanctx *
446 ieee80211_alloc_chanctx(struct ieee80211_local *local,
447                         const struct cfg80211_chan_def *chandef,
448                         enum ieee80211_chanctx_mode mode)
449 {
450         struct ieee80211_chanctx *ctx;
451
452         lockdep_assert_held(&local->chanctx_mtx);
453
454         ctx = kzalloc(sizeof(*ctx) + local->hw.chanctx_data_size, GFP_KERNEL);
455         if (!ctx)
456                 return NULL;
457
458         INIT_LIST_HEAD(&ctx->assigned_vifs);
459         INIT_LIST_HEAD(&ctx->reserved_vifs);
460         ctx->conf.def = *chandef;
461         ctx->conf.rx_chains_static = 1;
462         ctx->conf.rx_chains_dynamic = 1;
463         ctx->mode = mode;
464         ctx->conf.radar_enabled = false;
465         ieee80211_recalc_chanctx_min_def(local, ctx);
466
467         return ctx;
468 }
469
470 static int ieee80211_add_chanctx(struct ieee80211_local *local,
471                                  struct ieee80211_chanctx *ctx)
472 {
473         u32 changed;
474         int err;
475
476         lockdep_assert_held(&local->mtx);
477         lockdep_assert_held(&local->chanctx_mtx);
478
479         if (!local->use_chanctx)
480                 local->hw.conf.radar_enabled = ctx->conf.radar_enabled;
481
482         /* turn idle off *before* setting channel -- some drivers need that */
483         changed = ieee80211_idle_off(local);
484         if (changed)
485                 ieee80211_hw_config(local, changed);
486
487         if (!local->use_chanctx) {
488                 local->_oper_chandef = ctx->conf.def;
489                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_CHANNEL);
490         } else {
491                 err = drv_add_chanctx(local, ctx);
492                 if (err) {
493                         ieee80211_recalc_idle(local);
494                         return err;
495                 }
496         }
497
498         return 0;
499 }
500
501 static struct ieee80211_chanctx *
502 ieee80211_new_chanctx(struct ieee80211_local *local,
503                       const struct cfg80211_chan_def *chandef,
504                       enum ieee80211_chanctx_mode mode)
505 {
506         struct ieee80211_chanctx *ctx;
507         int err;
508
509         lockdep_assert_held(&local->mtx);
510         lockdep_assert_held(&local->chanctx_mtx);
511
512         ctx = ieee80211_alloc_chanctx(local, chandef, mode);
513         if (!ctx)
514                 return ERR_PTR(-ENOMEM);
515
516         err = ieee80211_add_chanctx(local, ctx);
517         if (err) {
518                 kfree(ctx);
519                 return ERR_PTR(err);
520         }
521
522         list_add_rcu(&ctx->list, &local->chanctx_list);
523         return ctx;
524 }
525
526 static void ieee80211_del_chanctx(struct ieee80211_local *local,
527                                   struct ieee80211_chanctx *ctx)
528 {
529         lockdep_assert_held(&local->chanctx_mtx);
530
531         if (!local->use_chanctx) {
532                 struct cfg80211_chan_def *chandef = &local->_oper_chandef;
533                 chandef->width = NL80211_CHAN_WIDTH_20_NOHT;
534                 chandef->center_freq1 = chandef->chan->center_freq;
535                 chandef->center_freq2 = 0;
536
537                 /* NOTE: Disabling radar is only valid here for
538                  * single channel context. To be sure, check it ...
539                  */
540                 WARN_ON(local->hw.conf.radar_enabled &&
541                         !list_empty(&local->chanctx_list));
542
543                 local->hw.conf.radar_enabled = false;
544
545                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_CHANNEL);
546         } else {
547                 drv_remove_chanctx(local, ctx);
548         }
549
550         ieee80211_recalc_idle(local);
551 }
552
553 static void ieee80211_free_chanctx(struct ieee80211_local *local,
554                                    struct ieee80211_chanctx *ctx)
555 {
556         lockdep_assert_held(&local->chanctx_mtx);
557
558         WARN_ON_ONCE(ieee80211_chanctx_refcount(local, ctx) != 0);
559
560         list_del_rcu(&ctx->list);
561         ieee80211_del_chanctx(local, ctx);
562         kfree_rcu(ctx, rcu_head);
563 }
564
565 void ieee80211_recalc_chanctx_chantype(struct ieee80211_local *local,
566                                        struct ieee80211_chanctx *ctx)
567 {
568         struct ieee80211_chanctx_conf *conf = &ctx->conf;
569         struct ieee80211_sub_if_data *sdata;
570         const struct cfg80211_chan_def *compat = NULL;
571         struct sta_info *sta;
572
573         lockdep_assert_held(&local->chanctx_mtx);
574
575         rcu_read_lock();
576         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
577
578                 if (!ieee80211_sdata_running(sdata))
579                         continue;
580                 if (rcu_access_pointer(sdata->vif.chanctx_conf) != conf)
581                         continue;
582                 if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
583                         continue;
584
585                 if (!compat)
586                         compat = &sdata->vif.bss_conf.chandef;
587
588                 compat = cfg80211_chandef_compatible(
589                                 &sdata->vif.bss_conf.chandef, compat);
590                 if (WARN_ON_ONCE(!compat))
591                         break;
592         }
593
594         /* TDLS peers can sometimes affect the chandef width */
595         list_for_each_entry_rcu(sta, &local->sta_list, list) {
596                 if (!sta->uploaded ||
597                     !test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW) ||
598                     !test_sta_flag(sta, WLAN_STA_AUTHORIZED) ||
599                     !sta->tdls_chandef.chan)
600                         continue;
601
602                 compat = cfg80211_chandef_compatible(&sta->tdls_chandef,
603                                                      compat);
604                 if (WARN_ON_ONCE(!compat))
605                         break;
606         }
607         rcu_read_unlock();
608
609         if (!compat)
610                 return;
611
612         ieee80211_change_chanctx(local, ctx, compat);
613 }
614
615 static void ieee80211_recalc_radar_chanctx(struct ieee80211_local *local,
616                                            struct ieee80211_chanctx *chanctx)
617 {
618         bool radar_enabled;
619
620         lockdep_assert_held(&local->chanctx_mtx);
621         /* for ieee80211_is_radar_required */
622         lockdep_assert_held(&local->mtx);
623
624         radar_enabled = ieee80211_chanctx_radar_required(local, chanctx);
625
626         if (radar_enabled == chanctx->conf.radar_enabled)
627                 return;
628
629         chanctx->conf.radar_enabled = radar_enabled;
630
631         if (!local->use_chanctx) {
632                 local->hw.conf.radar_enabled = chanctx->conf.radar_enabled;
633                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_CHANNEL);
634         }
635
636         drv_change_chanctx(local, chanctx, IEEE80211_CHANCTX_CHANGE_RADAR);
637 }
638
639 static int ieee80211_assign_vif_chanctx(struct ieee80211_sub_if_data *sdata,
640                                         struct ieee80211_chanctx *new_ctx)
641 {
642         struct ieee80211_local *local = sdata->local;
643         struct ieee80211_chanctx_conf *conf;
644         struct ieee80211_chanctx *curr_ctx = NULL;
645         int ret = 0;
646
647         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
648                                          lockdep_is_held(&local->chanctx_mtx));
649
650         if (conf) {
651                 curr_ctx = container_of(conf, struct ieee80211_chanctx, conf);
652
653                 drv_unassign_vif_chanctx(local, sdata, curr_ctx);
654                 conf = NULL;
655                 list_del(&sdata->assigned_chanctx_list);
656         }
657
658         if (new_ctx) {
659                 ret = drv_assign_vif_chanctx(local, sdata, new_ctx);
660                 if (ret)
661                         goto out;
662
663                 conf = &new_ctx->conf;
664                 list_add(&sdata->assigned_chanctx_list,
665                          &new_ctx->assigned_vifs);
666         }
667
668 out:
669         rcu_assign_pointer(sdata->vif.chanctx_conf, conf);
670
671         sdata->vif.bss_conf.idle = !conf;
672
673         if (curr_ctx && ieee80211_chanctx_num_assigned(local, curr_ctx) > 0) {
674                 ieee80211_recalc_chanctx_chantype(local, curr_ctx);
675                 ieee80211_recalc_smps_chanctx(local, curr_ctx);
676                 ieee80211_recalc_radar_chanctx(local, curr_ctx);
677                 ieee80211_recalc_chanctx_min_def(local, curr_ctx);
678         }
679
680         if (new_ctx && ieee80211_chanctx_num_assigned(local, new_ctx) > 0) {
681                 ieee80211_recalc_txpower(sdata, false);
682                 ieee80211_recalc_chanctx_min_def(local, new_ctx);
683         }
684
685         if (sdata->vif.type != NL80211_IFTYPE_P2P_DEVICE &&
686             sdata->vif.type != NL80211_IFTYPE_MONITOR)
687                 ieee80211_bss_info_change_notify(sdata,
688                                                  BSS_CHANGED_IDLE);
689
690         ieee80211_check_fast_xmit_iface(sdata);
691
692         return ret;
693 }
694
695 void ieee80211_recalc_smps_chanctx(struct ieee80211_local *local,
696                                    struct ieee80211_chanctx *chanctx)
697 {
698         struct ieee80211_sub_if_data *sdata;
699         u8 rx_chains_static, rx_chains_dynamic;
700
701         lockdep_assert_held(&local->chanctx_mtx);
702
703         rx_chains_static = 1;
704         rx_chains_dynamic = 1;
705
706         rcu_read_lock();
707         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
708                 u8 needed_static, needed_dynamic;
709
710                 if (!ieee80211_sdata_running(sdata))
711                         continue;
712
713                 if (rcu_access_pointer(sdata->vif.chanctx_conf) !=
714                                                 &chanctx->conf)
715                         continue;
716
717                 switch (sdata->vif.type) {
718                 case NL80211_IFTYPE_P2P_DEVICE:
719                         continue;
720                 case NL80211_IFTYPE_STATION:
721                         if (!sdata->u.mgd.associated)
722                                 continue;
723                         break;
724                 case NL80211_IFTYPE_AP_VLAN:
725                         continue;
726                 case NL80211_IFTYPE_AP:
727                 case NL80211_IFTYPE_ADHOC:
728                 case NL80211_IFTYPE_WDS:
729                 case NL80211_IFTYPE_MESH_POINT:
730                 case NL80211_IFTYPE_OCB:
731                         break;
732                 default:
733                         WARN_ON_ONCE(1);
734                 }
735
736                 switch (sdata->smps_mode) {
737                 default:
738                         WARN_ONCE(1, "Invalid SMPS mode %d\n",
739                                   sdata->smps_mode);
740                         /* fall through */
741                 case IEEE80211_SMPS_OFF:
742                         needed_static = sdata->needed_rx_chains;
743                         needed_dynamic = sdata->needed_rx_chains;
744                         break;
745                 case IEEE80211_SMPS_DYNAMIC:
746                         needed_static = 1;
747                         needed_dynamic = sdata->needed_rx_chains;
748                         break;
749                 case IEEE80211_SMPS_STATIC:
750                         needed_static = 1;
751                         needed_dynamic = 1;
752                         break;
753                 }
754
755                 rx_chains_static = max(rx_chains_static, needed_static);
756                 rx_chains_dynamic = max(rx_chains_dynamic, needed_dynamic);
757         }
758
759         /* Disable SMPS for the monitor interface */
760         sdata = rcu_dereference(local->monitor_sdata);
761         if (sdata &&
762             rcu_access_pointer(sdata->vif.chanctx_conf) == &chanctx->conf)
763                 rx_chains_dynamic = rx_chains_static = local->rx_chains;
764
765         rcu_read_unlock();
766
767         if (!local->use_chanctx) {
768                 if (rx_chains_static > 1)
769                         local->smps_mode = IEEE80211_SMPS_OFF;
770                 else if (rx_chains_dynamic > 1)
771                         local->smps_mode = IEEE80211_SMPS_DYNAMIC;
772                 else
773                         local->smps_mode = IEEE80211_SMPS_STATIC;
774                 ieee80211_hw_config(local, 0);
775         }
776
777         if (rx_chains_static == chanctx->conf.rx_chains_static &&
778             rx_chains_dynamic == chanctx->conf.rx_chains_dynamic)
779                 return;
780
781         chanctx->conf.rx_chains_static = rx_chains_static;
782         chanctx->conf.rx_chains_dynamic = rx_chains_dynamic;
783         drv_change_chanctx(local, chanctx, IEEE80211_CHANCTX_CHANGE_RX_CHAINS);
784 }
785
786 static void
787 __ieee80211_vif_copy_chanctx_to_vlans(struct ieee80211_sub_if_data *sdata,
788                                       bool clear)
789 {
790         struct ieee80211_local *local __maybe_unused = sdata->local;
791         struct ieee80211_sub_if_data *vlan;
792         struct ieee80211_chanctx_conf *conf;
793
794         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_AP))
795                 return;
796
797         lockdep_assert_held(&local->mtx);
798
799         /* Check that conf exists, even when clearing this function
800          * must be called with the AP's channel context still there
801          * as it would otherwise cause VLANs to have an invalid
802          * channel context pointer for a while, possibly pointing
803          * to a channel context that has already been freed.
804          */
805         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
806                                          lockdep_is_held(&local->chanctx_mtx));
807         WARN_ON(!conf);
808
809         if (clear)
810                 conf = NULL;
811
812         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
813                 rcu_assign_pointer(vlan->vif.chanctx_conf, conf);
814 }
815
816 void ieee80211_vif_copy_chanctx_to_vlans(struct ieee80211_sub_if_data *sdata,
817                                          bool clear)
818 {
819         struct ieee80211_local *local = sdata->local;
820
821         mutex_lock(&local->chanctx_mtx);
822
823         __ieee80211_vif_copy_chanctx_to_vlans(sdata, clear);
824
825         mutex_unlock(&local->chanctx_mtx);
826 }
827
828 int ieee80211_vif_unreserve_chanctx(struct ieee80211_sub_if_data *sdata)
829 {
830         struct ieee80211_chanctx *ctx = sdata->reserved_chanctx;
831
832         lockdep_assert_held(&sdata->local->chanctx_mtx);
833
834         if (WARN_ON(!ctx))
835                 return -EINVAL;
836
837         list_del(&sdata->reserved_chanctx_list);
838         sdata->reserved_chanctx = NULL;
839
840         if (ieee80211_chanctx_refcount(sdata->local, ctx) == 0) {
841                 if (ctx->replace_state == IEEE80211_CHANCTX_REPLACES_OTHER) {
842                         if (WARN_ON(!ctx->replace_ctx))
843                                 return -EINVAL;
844
845                         WARN_ON(ctx->replace_ctx->replace_state !=
846                                 IEEE80211_CHANCTX_WILL_BE_REPLACED);
847                         WARN_ON(ctx->replace_ctx->replace_ctx != ctx);
848
849                         ctx->replace_ctx->replace_ctx = NULL;
850                         ctx->replace_ctx->replace_state =
851                                         IEEE80211_CHANCTX_REPLACE_NONE;
852
853                         list_del_rcu(&ctx->list);
854                         kfree_rcu(ctx, rcu_head);
855                 } else {
856                         ieee80211_free_chanctx(sdata->local, ctx);
857                 }
858         }
859
860         return 0;
861 }
862
863 int ieee80211_vif_reserve_chanctx(struct ieee80211_sub_if_data *sdata,
864                                   const struct cfg80211_chan_def *chandef,
865                                   enum ieee80211_chanctx_mode mode,
866                                   bool radar_required)
867 {
868         struct ieee80211_local *local = sdata->local;
869         struct ieee80211_chanctx *new_ctx, *curr_ctx, *ctx;
870
871         lockdep_assert_held(&local->chanctx_mtx);
872
873         curr_ctx = ieee80211_vif_get_chanctx(sdata);
874         if (curr_ctx && local->use_chanctx && !local->ops->switch_vif_chanctx)
875                 return -ENOTSUPP;
876
877         new_ctx = ieee80211_find_reservation_chanctx(local, chandef, mode);
878         if (!new_ctx) {
879                 if (ieee80211_can_create_new_chanctx(local)) {
880                         new_ctx = ieee80211_new_chanctx(local, chandef, mode);
881                         if (IS_ERR(new_ctx))
882                                 return PTR_ERR(new_ctx);
883                 } else {
884                         if (!curr_ctx ||
885                             (curr_ctx->replace_state ==
886                              IEEE80211_CHANCTX_WILL_BE_REPLACED) ||
887                             !list_empty(&curr_ctx->reserved_vifs)) {
888                                 /*
889                                  * Another vif already requested this context
890                                  * for a reservation. Find another one hoping
891                                  * all vifs assigned to it will also switch
892                                  * soon enough.
893                                  *
894                                  * TODO: This needs a little more work as some
895                                  * cases (more than 2 chanctx capable devices)
896                                  * may fail which could otherwise succeed
897                                  * provided some channel context juggling was
898                                  * performed.
899                                  *
900                                  * Consider ctx1..3, vif1..6, each ctx has 2
901                                  * vifs. vif1 and vif2 from ctx1 request new
902                                  * different chandefs starting 2 in-place
903                                  * reserations with ctx4 and ctx5 replacing
904                                  * ctx1 and ctx2 respectively. Next vif5 and
905                                  * vif6 from ctx3 reserve ctx4. If vif3 and
906                                  * vif4 remain on ctx2 as they are then this
907                                  * fails unless `replace_ctx` from ctx5 is
908                                  * replaced with ctx3.
909                                  */
910                                 list_for_each_entry(ctx, &local->chanctx_list,
911                                                     list) {
912                                         if (ctx->replace_state !=
913                                             IEEE80211_CHANCTX_REPLACE_NONE)
914                                                 continue;
915
916                                         if (!list_empty(&ctx->reserved_vifs))
917                                                 continue;
918
919                                         curr_ctx = ctx;
920                                         break;
921                                 }
922                         }
923
924                         /*
925                          * If that's true then all available contexts already
926                          * have reservations and cannot be used.
927                          */
928                         if (!curr_ctx ||
929                             (curr_ctx->replace_state ==
930                              IEEE80211_CHANCTX_WILL_BE_REPLACED) ||
931                             !list_empty(&curr_ctx->reserved_vifs))
932                                 return -EBUSY;
933
934                         new_ctx = ieee80211_alloc_chanctx(local, chandef, mode);
935                         if (!new_ctx)
936                                 return -ENOMEM;
937
938                         new_ctx->replace_ctx = curr_ctx;
939                         new_ctx->replace_state =
940                                         IEEE80211_CHANCTX_REPLACES_OTHER;
941
942                         curr_ctx->replace_ctx = new_ctx;
943                         curr_ctx->replace_state =
944                                         IEEE80211_CHANCTX_WILL_BE_REPLACED;
945
946                         list_add_rcu(&new_ctx->list, &local->chanctx_list);
947                 }
948         }
949
950         list_add(&sdata->reserved_chanctx_list, &new_ctx->reserved_vifs);
951         sdata->reserved_chanctx = new_ctx;
952         sdata->reserved_chandef = *chandef;
953         sdata->reserved_radar_required = radar_required;
954         sdata->reserved_ready = false;
955
956         return 0;
957 }
958
959 static void
960 ieee80211_vif_chanctx_reservation_complete(struct ieee80211_sub_if_data *sdata)
961 {
962         switch (sdata->vif.type) {
963         case NL80211_IFTYPE_ADHOC:
964         case NL80211_IFTYPE_AP:
965         case NL80211_IFTYPE_MESH_POINT:
966         case NL80211_IFTYPE_OCB:
967                 ieee80211_queue_work(&sdata->local->hw,
968                                      &sdata->csa_finalize_work);
969                 break;
970         case NL80211_IFTYPE_STATION:
971                 ieee80211_queue_work(&sdata->local->hw,
972                                      &sdata->u.mgd.chswitch_work);
973                 break;
974         case NL80211_IFTYPE_UNSPECIFIED:
975         case NL80211_IFTYPE_AP_VLAN:
976         case NL80211_IFTYPE_WDS:
977         case NL80211_IFTYPE_MONITOR:
978         case NL80211_IFTYPE_P2P_CLIENT:
979         case NL80211_IFTYPE_P2P_GO:
980         case NL80211_IFTYPE_P2P_DEVICE:
981         case NUM_NL80211_IFTYPES:
982                 WARN_ON(1);
983                 break;
984         }
985 }
986
987 static void
988 ieee80211_vif_update_chandef(struct ieee80211_sub_if_data *sdata,
989                              const struct cfg80211_chan_def *chandef)
990 {
991         struct ieee80211_sub_if_data *vlan;
992
993         sdata->vif.bss_conf.chandef = *chandef;
994
995         if (sdata->vif.type != NL80211_IFTYPE_AP)
996                 return;
997
998         list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
999                 vlan->vif.bss_conf.chandef = *chandef;
1000 }
1001
1002 static int
1003 ieee80211_vif_use_reserved_reassign(struct ieee80211_sub_if_data *sdata)
1004 {
1005         struct ieee80211_local *local = sdata->local;
1006         struct ieee80211_vif_chanctx_switch vif_chsw[1] = {};
1007         struct ieee80211_chanctx *old_ctx, *new_ctx;
1008         const struct cfg80211_chan_def *chandef;
1009         u32 changed = 0;
1010         int err;
1011
1012         lockdep_assert_held(&local->mtx);
1013         lockdep_assert_held(&local->chanctx_mtx);
1014
1015         new_ctx = sdata->reserved_chanctx;
1016         old_ctx = ieee80211_vif_get_chanctx(sdata);
1017
1018         if (WARN_ON(!sdata->reserved_ready))
1019                 return -EBUSY;
1020
1021         if (WARN_ON(!new_ctx))
1022                 return -EINVAL;
1023
1024         if (WARN_ON(!old_ctx))
1025                 return -EINVAL;
1026
1027         if (WARN_ON(new_ctx->replace_state ==
1028                     IEEE80211_CHANCTX_REPLACES_OTHER))
1029                 return -EINVAL;
1030
1031         chandef = ieee80211_chanctx_non_reserved_chandef(local, new_ctx,
1032                                 &sdata->reserved_chandef);
1033         if (WARN_ON(!chandef))
1034                 return -EINVAL;
1035
1036         ieee80211_change_chanctx(local, new_ctx, chandef);
1037
1038         vif_chsw[0].vif = &sdata->vif;
1039         vif_chsw[0].old_ctx = &old_ctx->conf;
1040         vif_chsw[0].new_ctx = &new_ctx->conf;
1041
1042         list_del(&sdata->reserved_chanctx_list);
1043         sdata->reserved_chanctx = NULL;
1044
1045         err = drv_switch_vif_chanctx(local, vif_chsw, 1,
1046                                      CHANCTX_SWMODE_REASSIGN_VIF);
1047         if (err) {
1048                 if (ieee80211_chanctx_refcount(local, new_ctx) == 0)
1049                         ieee80211_free_chanctx(local, new_ctx);
1050
1051                 goto out;
1052         }
1053
1054         list_move(&sdata->assigned_chanctx_list, &new_ctx->assigned_vifs);
1055         rcu_assign_pointer(sdata->vif.chanctx_conf, &new_ctx->conf);
1056
1057         if (sdata->vif.type == NL80211_IFTYPE_AP)
1058                 __ieee80211_vif_copy_chanctx_to_vlans(sdata, false);
1059
1060         ieee80211_check_fast_xmit_iface(sdata);
1061
1062         if (ieee80211_chanctx_refcount(local, old_ctx) == 0)
1063                 ieee80211_free_chanctx(local, old_ctx);
1064
1065         if (sdata->vif.bss_conf.chandef.width != sdata->reserved_chandef.width)
1066                 changed = BSS_CHANGED_BANDWIDTH;
1067
1068         ieee80211_vif_update_chandef(sdata, &sdata->reserved_chandef);
1069
1070         ieee80211_recalc_smps_chanctx(local, new_ctx);
1071         ieee80211_recalc_radar_chanctx(local, new_ctx);
1072         ieee80211_recalc_chanctx_min_def(local, new_ctx);
1073
1074         if (changed)
1075                 ieee80211_bss_info_change_notify(sdata, changed);
1076
1077 out:
1078         ieee80211_vif_chanctx_reservation_complete(sdata);
1079         return err;
1080 }
1081
1082 static int
1083 ieee80211_vif_use_reserved_assign(struct ieee80211_sub_if_data *sdata)
1084 {
1085         struct ieee80211_local *local = sdata->local;
1086         struct ieee80211_chanctx *old_ctx, *new_ctx;
1087         const struct cfg80211_chan_def *chandef;
1088         int err;
1089
1090         old_ctx = ieee80211_vif_get_chanctx(sdata);
1091         new_ctx = sdata->reserved_chanctx;
1092
1093         if (WARN_ON(!sdata->reserved_ready))
1094                 return -EINVAL;
1095
1096         if (WARN_ON(old_ctx))
1097                 return -EINVAL;
1098
1099         if (WARN_ON(!new_ctx))
1100                 return -EINVAL;
1101
1102         if (WARN_ON(new_ctx->replace_state ==
1103                     IEEE80211_CHANCTX_REPLACES_OTHER))
1104                 return -EINVAL;
1105
1106         chandef = ieee80211_chanctx_non_reserved_chandef(local, new_ctx,
1107                                 &sdata->reserved_chandef);
1108         if (WARN_ON(!chandef))
1109                 return -EINVAL;
1110
1111         ieee80211_change_chanctx(local, new_ctx, chandef);
1112
1113         list_del(&sdata->reserved_chanctx_list);
1114         sdata->reserved_chanctx = NULL;
1115
1116         err = ieee80211_assign_vif_chanctx(sdata, new_ctx);
1117         if (err) {
1118                 if (ieee80211_chanctx_refcount(local, new_ctx) == 0)
1119                         ieee80211_free_chanctx(local, new_ctx);
1120
1121                 goto out;
1122         }
1123
1124 out:
1125         ieee80211_vif_chanctx_reservation_complete(sdata);
1126         return err;
1127 }
1128
1129 static bool
1130 ieee80211_vif_has_in_place_reservation(struct ieee80211_sub_if_data *sdata)
1131 {
1132         struct ieee80211_chanctx *old_ctx, *new_ctx;
1133
1134         lockdep_assert_held(&sdata->local->chanctx_mtx);
1135
1136         new_ctx = sdata->reserved_chanctx;
1137         old_ctx = ieee80211_vif_get_chanctx(sdata);
1138
1139         if (!old_ctx)
1140                 return false;
1141
1142         if (WARN_ON(!new_ctx))
1143                 return false;
1144
1145         if (old_ctx->replace_state != IEEE80211_CHANCTX_WILL_BE_REPLACED)
1146                 return false;
1147
1148         if (new_ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1149                 return false;
1150
1151         return true;
1152 }
1153
1154 static int ieee80211_chsw_switch_hwconf(struct ieee80211_local *local,
1155                                         struct ieee80211_chanctx *new_ctx)
1156 {
1157         const struct cfg80211_chan_def *chandef;
1158
1159         lockdep_assert_held(&local->mtx);
1160         lockdep_assert_held(&local->chanctx_mtx);
1161
1162         chandef = ieee80211_chanctx_reserved_chandef(local, new_ctx, NULL);
1163         if (WARN_ON(!chandef))
1164                 return -EINVAL;
1165
1166         local->hw.conf.radar_enabled = new_ctx->conf.radar_enabled;
1167         local->_oper_chandef = *chandef;
1168         ieee80211_hw_config(local, 0);
1169
1170         return 0;
1171 }
1172
1173 static int ieee80211_chsw_switch_vifs(struct ieee80211_local *local,
1174                                       int n_vifs)
1175 {
1176         struct ieee80211_vif_chanctx_switch *vif_chsw;
1177         struct ieee80211_sub_if_data *sdata;
1178         struct ieee80211_chanctx *ctx, *old_ctx;
1179         int i, err;
1180
1181         lockdep_assert_held(&local->mtx);
1182         lockdep_assert_held(&local->chanctx_mtx);
1183
1184         vif_chsw = kzalloc(sizeof(vif_chsw[0]) * n_vifs, GFP_KERNEL);
1185         if (!vif_chsw)
1186                 return -ENOMEM;
1187
1188         i = 0;
1189         list_for_each_entry(ctx, &local->chanctx_list, list) {
1190                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1191                         continue;
1192
1193                 if (WARN_ON(!ctx->replace_ctx)) {
1194                         err = -EINVAL;
1195                         goto out;
1196                 }
1197
1198                 list_for_each_entry(sdata, &ctx->reserved_vifs,
1199                                     reserved_chanctx_list) {
1200                         if (!ieee80211_vif_has_in_place_reservation(
1201                                         sdata))
1202                                 continue;
1203
1204                         old_ctx = ieee80211_vif_get_chanctx(sdata);
1205                         vif_chsw[i].vif = &sdata->vif;
1206                         vif_chsw[i].old_ctx = &old_ctx->conf;
1207                         vif_chsw[i].new_ctx = &ctx->conf;
1208
1209                         i++;
1210                 }
1211         }
1212
1213         err = drv_switch_vif_chanctx(local, vif_chsw, n_vifs,
1214                                      CHANCTX_SWMODE_SWAP_CONTEXTS);
1215
1216 out:
1217         kfree(vif_chsw);
1218         return err;
1219 }
1220
1221 static int ieee80211_chsw_switch_ctxs(struct ieee80211_local *local)
1222 {
1223         struct ieee80211_chanctx *ctx;
1224         int err;
1225
1226         lockdep_assert_held(&local->mtx);
1227         lockdep_assert_held(&local->chanctx_mtx);
1228
1229         list_for_each_entry(ctx, &local->chanctx_list, list) {
1230                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1231                         continue;
1232
1233                 if (!list_empty(&ctx->replace_ctx->assigned_vifs))
1234                         continue;
1235
1236                 ieee80211_del_chanctx(local, ctx->replace_ctx);
1237                 err = ieee80211_add_chanctx(local, ctx);
1238                 if (err)
1239                         goto err;
1240         }
1241
1242         return 0;
1243
1244 err:
1245         WARN_ON(ieee80211_add_chanctx(local, ctx));
1246         list_for_each_entry_continue_reverse(ctx, &local->chanctx_list, list) {
1247                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1248                         continue;
1249
1250                 if (!list_empty(&ctx->replace_ctx->assigned_vifs))
1251                         continue;
1252
1253                 ieee80211_del_chanctx(local, ctx);
1254                 WARN_ON(ieee80211_add_chanctx(local, ctx->replace_ctx));
1255         }
1256
1257         return err;
1258 }
1259
1260 static int ieee80211_vif_use_reserved_switch(struct ieee80211_local *local)
1261 {
1262         struct ieee80211_sub_if_data *sdata, *sdata_tmp;
1263         struct ieee80211_chanctx *ctx, *ctx_tmp, *old_ctx;
1264         struct ieee80211_chanctx *new_ctx = NULL;
1265         int i, err, n_assigned, n_reserved, n_ready;
1266         int n_ctx = 0, n_vifs_switch = 0, n_vifs_assign = 0, n_vifs_ctxless = 0;
1267
1268         lockdep_assert_held(&local->mtx);
1269         lockdep_assert_held(&local->chanctx_mtx);
1270
1271         /*
1272          * If there are 2 independent pairs of channel contexts performing
1273          * cross-switch of their vifs this code will still wait until both are
1274          * ready even though it could be possible to switch one before the
1275          * other is ready.
1276          *
1277          * For practical reasons and code simplicity just do a single huge
1278          * switch.
1279          */
1280
1281         /*
1282          * Verify if the reservation is still feasible.
1283          *  - if it's not then disconnect
1284          *  - if it is but not all vifs necessary are ready then defer
1285          */
1286
1287         list_for_each_entry(ctx, &local->chanctx_list, list) {
1288                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1289                         continue;
1290
1291                 if (WARN_ON(!ctx->replace_ctx)) {
1292                         err = -EINVAL;
1293                         goto err;
1294                 }
1295
1296                 if (!local->use_chanctx)
1297                         new_ctx = ctx;
1298
1299                 n_ctx++;
1300
1301                 n_assigned = 0;
1302                 n_reserved = 0;
1303                 n_ready = 0;
1304
1305                 list_for_each_entry(sdata, &ctx->replace_ctx->assigned_vifs,
1306                                     assigned_chanctx_list) {
1307                         n_assigned++;
1308                         if (sdata->reserved_chanctx) {
1309                                 n_reserved++;
1310                                 if (sdata->reserved_ready)
1311                                         n_ready++;
1312                         }
1313                 }
1314
1315                 if (n_assigned != n_reserved) {
1316                         if (n_ready == n_reserved) {
1317                                 wiphy_info(local->hw.wiphy,
1318                                            "channel context reservation cannot be finalized because some interfaces aren't switching\n");
1319                                 err = -EBUSY;
1320                                 goto err;
1321                         }
1322
1323                         return -EAGAIN;
1324                 }
1325
1326                 ctx->conf.radar_enabled = false;
1327                 list_for_each_entry(sdata, &ctx->reserved_vifs,
1328                                     reserved_chanctx_list) {
1329                         if (ieee80211_vif_has_in_place_reservation(sdata) &&
1330                             !sdata->reserved_ready)
1331                                 return -EAGAIN;
1332
1333                         old_ctx = ieee80211_vif_get_chanctx(sdata);
1334                         if (old_ctx) {
1335                                 if (old_ctx->replace_state ==
1336                                     IEEE80211_CHANCTX_WILL_BE_REPLACED)
1337                                         n_vifs_switch++;
1338                                 else
1339                                         n_vifs_assign++;
1340                         } else {
1341                                 n_vifs_ctxless++;
1342                         }
1343
1344                         if (sdata->reserved_radar_required)
1345                                 ctx->conf.radar_enabled = true;
1346                 }
1347         }
1348
1349         if (WARN_ON(n_ctx == 0) ||
1350             WARN_ON(n_vifs_switch == 0 &&
1351                     n_vifs_assign == 0 &&
1352                     n_vifs_ctxless == 0) ||
1353             WARN_ON(n_ctx > 1 && !local->use_chanctx) ||
1354             WARN_ON(!new_ctx && !local->use_chanctx)) {
1355                 err = -EINVAL;
1356                 goto err;
1357         }
1358
1359         /*
1360          * All necessary vifs are ready. Perform the switch now depending on
1361          * reservations and driver capabilities.
1362          */
1363
1364         if (local->use_chanctx) {
1365                 if (n_vifs_switch > 0) {
1366                         err = ieee80211_chsw_switch_vifs(local, n_vifs_switch);
1367                         if (err)
1368                                 goto err;
1369                 }
1370
1371                 if (n_vifs_assign > 0 || n_vifs_ctxless > 0) {
1372                         err = ieee80211_chsw_switch_ctxs(local);
1373                         if (err)
1374                                 goto err;
1375                 }
1376         } else {
1377                 err = ieee80211_chsw_switch_hwconf(local, new_ctx);
1378                 if (err)
1379                         goto err;
1380         }
1381
1382         /*
1383          * Update all structures, values and pointers to point to new channel
1384          * context(s).
1385          */
1386
1387         i = 0;
1388         list_for_each_entry(ctx, &local->chanctx_list, list) {
1389                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1390                         continue;
1391
1392                 if (WARN_ON(!ctx->replace_ctx)) {
1393                         err = -EINVAL;
1394                         goto err;
1395                 }
1396
1397                 list_for_each_entry(sdata, &ctx->reserved_vifs,
1398                                     reserved_chanctx_list) {
1399                         u32 changed = 0;
1400
1401                         if (!ieee80211_vif_has_in_place_reservation(sdata))
1402                                 continue;
1403
1404                         rcu_assign_pointer(sdata->vif.chanctx_conf, &ctx->conf);
1405
1406                         if (sdata->vif.type == NL80211_IFTYPE_AP)
1407                                 __ieee80211_vif_copy_chanctx_to_vlans(sdata,
1408                                                                       false);
1409
1410                         ieee80211_check_fast_xmit_iface(sdata);
1411
1412                         sdata->radar_required = sdata->reserved_radar_required;
1413
1414                         if (sdata->vif.bss_conf.chandef.width !=
1415                             sdata->reserved_chandef.width)
1416                                 changed = BSS_CHANGED_BANDWIDTH;
1417
1418                         ieee80211_vif_update_chandef(sdata, &sdata->reserved_chandef);
1419                         if (changed)
1420                                 ieee80211_bss_info_change_notify(sdata,
1421                                                                  changed);
1422
1423                         ieee80211_recalc_txpower(sdata, false);
1424                 }
1425
1426                 ieee80211_recalc_chanctx_chantype(local, ctx);
1427                 ieee80211_recalc_smps_chanctx(local, ctx);
1428                 ieee80211_recalc_radar_chanctx(local, ctx);
1429                 ieee80211_recalc_chanctx_min_def(local, ctx);
1430
1431                 list_for_each_entry_safe(sdata, sdata_tmp, &ctx->reserved_vifs,
1432                                          reserved_chanctx_list) {
1433                         if (ieee80211_vif_get_chanctx(sdata) != ctx)
1434                                 continue;
1435
1436                         list_del(&sdata->reserved_chanctx_list);
1437                         list_move(&sdata->assigned_chanctx_list,
1438                                   &ctx->assigned_vifs);
1439                         sdata->reserved_chanctx = NULL;
1440
1441                         ieee80211_vif_chanctx_reservation_complete(sdata);
1442                 }
1443
1444                 /*
1445                  * This context might have been a dependency for an already
1446                  * ready re-assign reservation interface that was deferred. Do
1447                  * not propagate error to the caller though. The in-place
1448                  * reservation for originally requested interface has already
1449                  * succeeded at this point.
1450                  */
1451                 list_for_each_entry_safe(sdata, sdata_tmp, &ctx->reserved_vifs,
1452                                          reserved_chanctx_list) {
1453                         if (WARN_ON(ieee80211_vif_has_in_place_reservation(
1454                                         sdata)))
1455                                 continue;
1456
1457                         if (WARN_ON(sdata->reserved_chanctx != ctx))
1458                                 continue;
1459
1460                         if (!sdata->reserved_ready)
1461                                 continue;
1462
1463                         if (ieee80211_vif_get_chanctx(sdata))
1464                                 err = ieee80211_vif_use_reserved_reassign(
1465                                                 sdata);
1466                         else
1467                                 err = ieee80211_vif_use_reserved_assign(sdata);
1468
1469                         if (err) {
1470                                 sdata_info(sdata,
1471                                            "failed to finalize (re-)assign reservation (err=%d)\n",
1472                                            err);
1473                                 ieee80211_vif_unreserve_chanctx(sdata);
1474                                 cfg80211_stop_iface(local->hw.wiphy,
1475                                                     &sdata->wdev,
1476                                                     GFP_KERNEL);
1477                         }
1478                 }
1479         }
1480
1481         /*
1482          * Finally free old contexts
1483          */
1484
1485         list_for_each_entry_safe(ctx, ctx_tmp, &local->chanctx_list, list) {
1486                 if (ctx->replace_state != IEEE80211_CHANCTX_WILL_BE_REPLACED)
1487                         continue;
1488
1489                 ctx->replace_ctx->replace_ctx = NULL;
1490                 ctx->replace_ctx->replace_state =
1491                                 IEEE80211_CHANCTX_REPLACE_NONE;
1492
1493                 list_del_rcu(&ctx->list);
1494                 kfree_rcu(ctx, rcu_head);
1495         }
1496
1497         return 0;
1498
1499 err:
1500         list_for_each_entry(ctx, &local->chanctx_list, list) {
1501                 if (ctx->replace_state != IEEE80211_CHANCTX_REPLACES_OTHER)
1502                         continue;
1503
1504                 list_for_each_entry_safe(sdata, sdata_tmp, &ctx->reserved_vifs,
1505                                          reserved_chanctx_list) {
1506                         ieee80211_vif_unreserve_chanctx(sdata);
1507                         ieee80211_vif_chanctx_reservation_complete(sdata);
1508                 }
1509         }
1510
1511         return err;
1512 }
1513
1514 static void __ieee80211_vif_release_channel(struct ieee80211_sub_if_data *sdata)
1515 {
1516         struct ieee80211_local *local = sdata->local;
1517         struct ieee80211_chanctx_conf *conf;
1518         struct ieee80211_chanctx *ctx;
1519         bool use_reserved_switch = false;
1520
1521         lockdep_assert_held(&local->chanctx_mtx);
1522
1523         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1524                                          lockdep_is_held(&local->chanctx_mtx));
1525         if (!conf)
1526                 return;
1527
1528         ctx = container_of(conf, struct ieee80211_chanctx, conf);
1529
1530         if (sdata->reserved_chanctx) {
1531                 if (sdata->reserved_chanctx->replace_state ==
1532                     IEEE80211_CHANCTX_REPLACES_OTHER &&
1533                     ieee80211_chanctx_num_reserved(local,
1534                                                    sdata->reserved_chanctx) > 1)
1535                         use_reserved_switch = true;
1536
1537                 ieee80211_vif_unreserve_chanctx(sdata);
1538         }
1539
1540         ieee80211_assign_vif_chanctx(sdata, NULL);
1541         if (ieee80211_chanctx_refcount(local, ctx) == 0)
1542                 ieee80211_free_chanctx(local, ctx);
1543
1544         sdata->radar_required = false;
1545
1546         /* Unreserving may ready an in-place reservation. */
1547         if (use_reserved_switch)
1548                 ieee80211_vif_use_reserved_switch(local);
1549 }
1550
1551 int ieee80211_vif_use_channel(struct ieee80211_sub_if_data *sdata,
1552                               const struct cfg80211_chan_def *chandef,
1553                               enum ieee80211_chanctx_mode mode)
1554 {
1555         struct ieee80211_local *local = sdata->local;
1556         struct ieee80211_chanctx *ctx;
1557         u8 radar_detect_width = 0;
1558         int ret;
1559
1560         lockdep_assert_held(&local->mtx);
1561
1562         WARN_ON(sdata->dev && netif_carrier_ok(sdata->dev));
1563
1564         mutex_lock(&local->chanctx_mtx);
1565
1566         ret = cfg80211_chandef_dfs_required(local->hw.wiphy,
1567                                             chandef,
1568                                             sdata->wdev.iftype);
1569         if (ret < 0)
1570                 goto out;
1571         if (ret > 0)
1572                 radar_detect_width = BIT(chandef->width);
1573
1574         sdata->radar_required = ret;
1575
1576         ret = ieee80211_check_combinations(sdata, chandef, mode,
1577                                            radar_detect_width);
1578         if (ret < 0)
1579                 goto out;
1580
1581         __ieee80211_vif_release_channel(sdata);
1582
1583         ctx = ieee80211_find_chanctx(local, chandef, mode);
1584         if (!ctx)
1585                 ctx = ieee80211_new_chanctx(local, chandef, mode);
1586         if (IS_ERR(ctx)) {
1587                 ret = PTR_ERR(ctx);
1588                 goto out;
1589         }
1590
1591         ieee80211_vif_update_chandef(sdata, chandef);
1592
1593         ret = ieee80211_assign_vif_chanctx(sdata, ctx);
1594         if (ret) {
1595                 /* if assign fails refcount stays the same */
1596                 if (ieee80211_chanctx_refcount(local, ctx) == 0)
1597                         ieee80211_free_chanctx(local, ctx);
1598                 goto out;
1599         }
1600
1601         ieee80211_recalc_smps_chanctx(local, ctx);
1602         ieee80211_recalc_radar_chanctx(local, ctx);
1603  out:
1604         if (ret)
1605                 sdata->radar_required = false;
1606
1607         mutex_unlock(&local->chanctx_mtx);
1608         return ret;
1609 }
1610
1611 int ieee80211_vif_use_reserved_context(struct ieee80211_sub_if_data *sdata)
1612 {
1613         struct ieee80211_local *local = sdata->local;
1614         struct ieee80211_chanctx *new_ctx;
1615         struct ieee80211_chanctx *old_ctx;
1616         int err;
1617
1618         lockdep_assert_held(&local->mtx);
1619         lockdep_assert_held(&local->chanctx_mtx);
1620
1621         new_ctx = sdata->reserved_chanctx;
1622         old_ctx = ieee80211_vif_get_chanctx(sdata);
1623
1624         if (WARN_ON(!new_ctx))
1625                 return -EINVAL;
1626
1627         if (WARN_ON(new_ctx->replace_state ==
1628                     IEEE80211_CHANCTX_WILL_BE_REPLACED))
1629                 return -EINVAL;
1630
1631         if (WARN_ON(sdata->reserved_ready))
1632                 return -EINVAL;
1633
1634         sdata->reserved_ready = true;
1635
1636         if (new_ctx->replace_state == IEEE80211_CHANCTX_REPLACE_NONE) {
1637                 if (old_ctx)
1638                         err = ieee80211_vif_use_reserved_reassign(sdata);
1639                 else
1640                         err = ieee80211_vif_use_reserved_assign(sdata);
1641
1642                 if (err)
1643                         return err;
1644         }
1645
1646         /*
1647          * In-place reservation may need to be finalized now either if:
1648          *  a) sdata is taking part in the swapping itself and is the last one
1649          *  b) sdata has switched with a re-assign reservation to an existing
1650          *     context readying in-place switching of old_ctx
1651          *
1652          * In case of (b) do not propagate the error up because the requested
1653          * sdata already switched successfully. Just spill an extra warning.
1654          * The ieee80211_vif_use_reserved_switch() already stops all necessary
1655          * interfaces upon failure.
1656          */
1657         if ((old_ctx &&
1658              old_ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED) ||
1659             new_ctx->replace_state == IEEE80211_CHANCTX_REPLACES_OTHER) {
1660                 err = ieee80211_vif_use_reserved_switch(local);
1661                 if (err && err != -EAGAIN) {
1662                         if (new_ctx->replace_state ==
1663                             IEEE80211_CHANCTX_REPLACES_OTHER)
1664                                 return err;
1665
1666                         wiphy_info(local->hw.wiphy,
1667                                    "depending in-place reservation failed (err=%d)\n",
1668                                    err);
1669                 }
1670         }
1671
1672         return 0;
1673 }
1674
1675 int ieee80211_vif_change_bandwidth(struct ieee80211_sub_if_data *sdata,
1676                                    const struct cfg80211_chan_def *chandef,
1677                                    u32 *changed)
1678 {
1679         struct ieee80211_local *local = sdata->local;
1680         struct ieee80211_chanctx_conf *conf;
1681         struct ieee80211_chanctx *ctx;
1682         const struct cfg80211_chan_def *compat;
1683         int ret;
1684
1685         if (!cfg80211_chandef_usable(sdata->local->hw.wiphy, chandef,
1686                                      IEEE80211_CHAN_DISABLED))
1687                 return -EINVAL;
1688
1689         mutex_lock(&local->chanctx_mtx);
1690         if (cfg80211_chandef_identical(chandef, &sdata->vif.bss_conf.chandef)) {
1691                 ret = 0;
1692                 goto out;
1693         }
1694
1695         if (chandef->width == NL80211_CHAN_WIDTH_20_NOHT ||
1696             sdata->vif.bss_conf.chandef.width == NL80211_CHAN_WIDTH_20_NOHT) {
1697                 ret = -EINVAL;
1698                 goto out;
1699         }
1700
1701         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1702                                          lockdep_is_held(&local->chanctx_mtx));
1703         if (!conf) {
1704                 ret = -EINVAL;
1705                 goto out;
1706         }
1707
1708         ctx = container_of(conf, struct ieee80211_chanctx, conf);
1709
1710         compat = cfg80211_chandef_compatible(&conf->def, chandef);
1711         if (!compat) {
1712                 ret = -EINVAL;
1713                 goto out;
1714         }
1715
1716         switch (ctx->replace_state) {
1717         case IEEE80211_CHANCTX_REPLACE_NONE:
1718                 if (!ieee80211_chanctx_reserved_chandef(local, ctx, compat)) {
1719                         ret = -EBUSY;
1720                         goto out;
1721                 }
1722                 break;
1723         case IEEE80211_CHANCTX_WILL_BE_REPLACED:
1724                 /* TODO: Perhaps the bandwidth change could be treated as a
1725                  * reservation itself? */
1726                 ret = -EBUSY;
1727                 goto out;
1728         case IEEE80211_CHANCTX_REPLACES_OTHER:
1729                 /* channel context that is going to replace another channel
1730                  * context doesn't really exist and shouldn't be assigned
1731                  * anywhere yet */
1732                 WARN_ON(1);
1733                 break;
1734         }
1735
1736         ieee80211_vif_update_chandef(sdata, chandef);
1737
1738         ieee80211_recalc_chanctx_chantype(local, ctx);
1739
1740         *changed |= BSS_CHANGED_BANDWIDTH;
1741         ret = 0;
1742  out:
1743         mutex_unlock(&local->chanctx_mtx);
1744         return ret;
1745 }
1746
1747 void ieee80211_vif_release_channel(struct ieee80211_sub_if_data *sdata)
1748 {
1749         WARN_ON(sdata->dev && netif_carrier_ok(sdata->dev));
1750
1751         lockdep_assert_held(&sdata->local->mtx);
1752
1753         mutex_lock(&sdata->local->chanctx_mtx);
1754         __ieee80211_vif_release_channel(sdata);
1755         mutex_unlock(&sdata->local->chanctx_mtx);
1756 }
1757
1758 void ieee80211_vif_vlan_copy_chanctx(struct ieee80211_sub_if_data *sdata)
1759 {
1760         struct ieee80211_local *local = sdata->local;
1761         struct ieee80211_sub_if_data *ap;
1762         struct ieee80211_chanctx_conf *conf;
1763
1764         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_AP_VLAN || !sdata->bss))
1765                 return;
1766
1767         ap = container_of(sdata->bss, struct ieee80211_sub_if_data, u.ap);
1768
1769         mutex_lock(&local->chanctx_mtx);
1770
1771         conf = rcu_dereference_protected(ap->vif.chanctx_conf,
1772                                          lockdep_is_held(&local->chanctx_mtx));
1773         rcu_assign_pointer(sdata->vif.chanctx_conf, conf);
1774         mutex_unlock(&local->chanctx_mtx);
1775 }
1776
1777 void ieee80211_iter_chan_contexts_atomic(
1778         struct ieee80211_hw *hw,
1779         void (*iter)(struct ieee80211_hw *hw,
1780                      struct ieee80211_chanctx_conf *chanctx_conf,
1781                      void *data),
1782         void *iter_data)
1783 {
1784         struct ieee80211_local *local = hw_to_local(hw);
1785         struct ieee80211_chanctx *ctx;
1786
1787         rcu_read_lock();
1788         list_for_each_entry_rcu(ctx, &local->chanctx_list, list)
1789                 if (ctx->driver_present)
1790                         iter(hw, &ctx->conf, iter_data);
1791         rcu_read_unlock();
1792 }
1793 EXPORT_SYMBOL_GPL(ieee80211_iter_chan_contexts_atomic);