net: wireless: move regulatory timeout work to power efficient workqueue
[firefly-linux-kernel-4.4.55.git] / net / wireless / reg.c
1 /*
2  * Copyright 2002-2005, Instant802 Networks, Inc.
3  * Copyright 2005-2006, Devicescape Software, Inc.
4  * Copyright 2007       Johannes Berg <johannes@sipsolutions.net>
5  * Copyright 2008-2011  Luis R. Rodriguez <mcgrof@qca.qualcomm.com>
6  *
7  * Permission to use, copy, modify, and/or distribute this software for any
8  * purpose with or without fee is hereby granted, provided that the above
9  * copyright notice and this permission notice appear in all copies.
10  *
11  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18  */
19
20
21 /**
22  * DOC: Wireless regulatory infrastructure
23  *
24  * The usual implementation is for a driver to read a device EEPROM to
25  * determine which regulatory domain it should be operating under, then
26  * looking up the allowable channels in a driver-local table and finally
27  * registering those channels in the wiphy structure.
28  *
29  * Another set of compliance enforcement is for drivers to use their
30  * own compliance limits which can be stored on the EEPROM. The host
31  * driver or firmware may ensure these are used.
32  *
33  * In addition to all this we provide an extra layer of regulatory
34  * conformance. For drivers which do not have any regulatory
35  * information CRDA provides the complete regulatory solution.
36  * For others it provides a community effort on further restrictions
37  * to enhance compliance.
38  *
39  * Note: When number of rules --> infinity we will not be able to
40  * index on alpha2 any more, instead we'll probably have to
41  * rely on some SHA1 checksum of the regdomain for example.
42  *
43  */
44
45 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
46
47 #include <linux/kernel.h>
48 #include <linux/export.h>
49 #include <linux/slab.h>
50 #include <linux/list.h>
51 #include <linux/ctype.h>
52 #include <linux/nl80211.h>
53 #include <linux/platform_device.h>
54 #include <linux/moduleparam.h>
55 #include <net/cfg80211.h>
56 #include "core.h"
57 #include "reg.h"
58 #include "regdb.h"
59 #include "nl80211.h"
60
61 #ifdef CONFIG_CFG80211_REG_DEBUG
62 #define REG_DBG_PRINT(format, args...)                  \
63         printk(KERN_DEBUG pr_fmt(format), ##args)
64 #else
65 #define REG_DBG_PRINT(args...)
66 #endif
67
68 enum reg_request_treatment {
69         REG_REQ_OK,
70         REG_REQ_IGNORE,
71         REG_REQ_INTERSECT,
72         REG_REQ_ALREADY_SET,
73 };
74
75 static struct regulatory_request core_request_world = {
76         .initiator = NL80211_REGDOM_SET_BY_CORE,
77         .alpha2[0] = '0',
78         .alpha2[1] = '0',
79         .intersect = false,
80         .processed = true,
81         .country_ie_env = ENVIRON_ANY,
82 };
83
84 /*
85  * Receipt of information from last regulatory request,
86  * protected by RTNL (and can be accessed with RCU protection)
87  */
88 static struct regulatory_request __rcu *last_request =
89         (void __rcu *)&core_request_world;
90
91 /* To trigger userspace events */
92 static struct platform_device *reg_pdev;
93
94 static const struct device_type reg_device_type = {
95         .uevent = reg_device_uevent,
96 };
97
98 /*
99  * Central wireless core regulatory domains, we only need two,
100  * the current one and a world regulatory domain in case we have no
101  * information to give us an alpha2.
102  * (protected by RTNL, can be read under RCU)
103  */
104 const struct ieee80211_regdomain __rcu *cfg80211_regdomain;
105
106 /*
107  * Number of devices that registered to the core
108  * that support cellular base station regulatory hints
109  * (protected by RTNL)
110  */
111 static int reg_num_devs_support_basehint;
112
113 static const struct ieee80211_regdomain *get_cfg80211_regdom(void)
114 {
115         return rtnl_dereference(cfg80211_regdomain);
116 }
117
118 static const struct ieee80211_regdomain *get_wiphy_regdom(struct wiphy *wiphy)
119 {
120         return rtnl_dereference(wiphy->regd);
121 }
122
123 static const char *reg_dfs_region_str(enum nl80211_dfs_regions dfs_region)
124 {
125         switch (dfs_region) {
126         case NL80211_DFS_UNSET:
127                 return "unset";
128         case NL80211_DFS_FCC:
129                 return "FCC";
130         case NL80211_DFS_ETSI:
131                 return "ETSI";
132         case NL80211_DFS_JP:
133                 return "JP";
134         }
135         return "Unknown";
136 }
137
138 enum nl80211_dfs_regions reg_get_dfs_region(struct wiphy *wiphy)
139 {
140         const struct ieee80211_regdomain *regd = NULL;
141         const struct ieee80211_regdomain *wiphy_regd = NULL;
142
143         regd = get_cfg80211_regdom();
144         if (!wiphy)
145                 goto out;
146
147         wiphy_regd = get_wiphy_regdom(wiphy);
148         if (!wiphy_regd)
149                 goto out;
150
151         if (wiphy_regd->dfs_region == regd->dfs_region)
152                 goto out;
153
154         REG_DBG_PRINT("%s: device specific dfs_region "
155                       "(%s) disagrees with cfg80211's "
156                       "central dfs_region (%s)\n",
157                       dev_name(&wiphy->dev),
158                       reg_dfs_region_str(wiphy_regd->dfs_region),
159                       reg_dfs_region_str(regd->dfs_region));
160
161 out:
162         return regd->dfs_region;
163 }
164
165 static void rcu_free_regdom(const struct ieee80211_regdomain *r)
166 {
167         if (!r)
168                 return;
169         kfree_rcu((struct ieee80211_regdomain *)r, rcu_head);
170 }
171
172 static struct regulatory_request *get_last_request(void)
173 {
174         return rcu_dereference_rtnl(last_request);
175 }
176
177 /* Used to queue up regulatory hints */
178 static LIST_HEAD(reg_requests_list);
179 static spinlock_t reg_requests_lock;
180
181 /* Used to queue up beacon hints for review */
182 static LIST_HEAD(reg_pending_beacons);
183 static spinlock_t reg_pending_beacons_lock;
184
185 /* Used to keep track of processed beacon hints */
186 static LIST_HEAD(reg_beacon_list);
187
188 struct reg_beacon {
189         struct list_head list;
190         struct ieee80211_channel chan;
191 };
192
193 static void reg_todo(struct work_struct *work);
194 static DECLARE_WORK(reg_work, reg_todo);
195
196 static void reg_timeout_work(struct work_struct *work);
197 static DECLARE_DELAYED_WORK(reg_timeout, reg_timeout_work);
198
199 /* We keep a static world regulatory domain in case of the absence of CRDA */
200 static const struct ieee80211_regdomain world_regdom = {
201         .n_reg_rules = 6,
202         .alpha2 =  "00",
203         .reg_rules = {
204                 /* IEEE 802.11b/g, channels 1..11 */
205                 REG_RULE(2412-10, 2462+10, 40, 6, 20, 0),
206                 /* IEEE 802.11b/g, channels 12..13. */
207                 REG_RULE(2467-10, 2472+10, 40, 6, 20,
208                         NL80211_RRF_NO_IR),
209                 /* IEEE 802.11 channel 14 - Only JP enables
210                  * this and for 802.11b only */
211                 REG_RULE(2484-10, 2484+10, 20, 6, 20,
212                         NL80211_RRF_NO_IR |
213                         NL80211_RRF_NO_OFDM),
214                 /* IEEE 802.11a, channel 36..48 */
215                 REG_RULE(5180-10, 5240+10, 160, 6, 20,
216                         NL80211_RRF_NO_IR),
217
218                 /* IEEE 802.11a, channel 52..64 - DFS required */
219                 REG_RULE(5260-10, 5320+10, 160, 6, 20,
220                         NL80211_RRF_NO_IR |
221                         NL80211_RRF_DFS),
222
223                 /* IEEE 802.11a, channel 100..144 - DFS required */
224                 REG_RULE(5500-10, 5720+10, 160, 6, 20,
225                         NL80211_RRF_NO_IR |
226                         NL80211_RRF_DFS),
227
228                 /* IEEE 802.11a, channel 149..165 */
229                 REG_RULE(5745-10, 5825+10, 80, 6, 20,
230                         NL80211_RRF_NO_IR),
231
232                 /* IEEE 802.11ad (60gHz), channels 1..3 */
233                 REG_RULE(56160+2160*1-1080, 56160+2160*3+1080, 2160, 0, 0, 0),
234         }
235 };
236
237 /* protected by RTNL */
238 static const struct ieee80211_regdomain *cfg80211_world_regdom =
239         &world_regdom;
240
241 static char *ieee80211_regdom = "00";
242 static char user_alpha2[2];
243
244 module_param(ieee80211_regdom, charp, 0444);
245 MODULE_PARM_DESC(ieee80211_regdom, "IEEE 802.11 regulatory domain code");
246
247 static void reg_kfree_last_request(void)
248 {
249         struct regulatory_request *lr;
250
251         lr = get_last_request();
252
253         if (lr != &core_request_world && lr)
254                 kfree_rcu(lr, rcu_head);
255 }
256
257 static void reg_update_last_request(struct regulatory_request *request)
258 {
259         reg_kfree_last_request();
260         rcu_assign_pointer(last_request, request);
261 }
262
263 static void reset_regdomains(bool full_reset,
264                              const struct ieee80211_regdomain *new_regdom)
265 {
266         const struct ieee80211_regdomain *r;
267
268         ASSERT_RTNL();
269
270         r = get_cfg80211_regdom();
271
272         /* avoid freeing static information or freeing something twice */
273         if (r == cfg80211_world_regdom)
274                 r = NULL;
275         if (cfg80211_world_regdom == &world_regdom)
276                 cfg80211_world_regdom = NULL;
277         if (r == &world_regdom)
278                 r = NULL;
279
280         rcu_free_regdom(r);
281         rcu_free_regdom(cfg80211_world_regdom);
282
283         cfg80211_world_regdom = &world_regdom;
284         rcu_assign_pointer(cfg80211_regdomain, new_regdom);
285
286         if (!full_reset)
287                 return;
288
289         reg_update_last_request(&core_request_world);
290 }
291
292 /*
293  * Dynamic world regulatory domain requested by the wireless
294  * core upon initialization
295  */
296 static void update_world_regdomain(const struct ieee80211_regdomain *rd)
297 {
298         struct regulatory_request *lr;
299
300         lr = get_last_request();
301
302         WARN_ON(!lr);
303
304         reset_regdomains(false, rd);
305
306         cfg80211_world_regdom = rd;
307 }
308
309 bool is_world_regdom(const char *alpha2)
310 {
311         if (!alpha2)
312                 return false;
313         return alpha2[0] == '0' && alpha2[1] == '0';
314 }
315
316 static bool is_alpha2_set(const char *alpha2)
317 {
318         if (!alpha2)
319                 return false;
320         return alpha2[0] && alpha2[1];
321 }
322
323 static bool is_unknown_alpha2(const char *alpha2)
324 {
325         if (!alpha2)
326                 return false;
327         /*
328          * Special case where regulatory domain was built by driver
329          * but a specific alpha2 cannot be determined
330          */
331         return alpha2[0] == '9' && alpha2[1] == '9';
332 }
333
334 static bool is_intersected_alpha2(const char *alpha2)
335 {
336         if (!alpha2)
337                 return false;
338         /*
339          * Special case where regulatory domain is the
340          * result of an intersection between two regulatory domain
341          * structures
342          */
343         return alpha2[0] == '9' && alpha2[1] == '8';
344 }
345
346 static bool is_an_alpha2(const char *alpha2)
347 {
348         if (!alpha2)
349                 return false;
350         return isalpha(alpha2[0]) && isalpha(alpha2[1]);
351 }
352
353 static bool alpha2_equal(const char *alpha2_x, const char *alpha2_y)
354 {
355         if (!alpha2_x || !alpha2_y)
356                 return false;
357         return alpha2_x[0] == alpha2_y[0] && alpha2_x[1] == alpha2_y[1];
358 }
359
360 static bool regdom_changes(const char *alpha2)
361 {
362         const struct ieee80211_regdomain *r = get_cfg80211_regdom();
363
364         if (!r)
365                 return true;
366         return !alpha2_equal(r->alpha2, alpha2);
367 }
368
369 /*
370  * The NL80211_REGDOM_SET_BY_USER regdom alpha2 is cached, this lets
371  * you know if a valid regulatory hint with NL80211_REGDOM_SET_BY_USER
372  * has ever been issued.
373  */
374 static bool is_user_regdom_saved(void)
375 {
376         if (user_alpha2[0] == '9' && user_alpha2[1] == '7')
377                 return false;
378
379         /* This would indicate a mistake on the design */
380         if (WARN(!is_world_regdom(user_alpha2) && !is_an_alpha2(user_alpha2),
381                  "Unexpected user alpha2: %c%c\n",
382                  user_alpha2[0], user_alpha2[1]))
383                 return false;
384
385         return true;
386 }
387
388 static const struct ieee80211_regdomain *
389 reg_copy_regd(const struct ieee80211_regdomain *src_regd)
390 {
391         struct ieee80211_regdomain *regd;
392         int size_of_regd;
393         unsigned int i;
394
395         size_of_regd =
396                 sizeof(struct ieee80211_regdomain) +
397                 src_regd->n_reg_rules * sizeof(struct ieee80211_reg_rule);
398
399         regd = kzalloc(size_of_regd, GFP_KERNEL);
400         if (!regd)
401                 return ERR_PTR(-ENOMEM);
402
403         memcpy(regd, src_regd, sizeof(struct ieee80211_regdomain));
404
405         for (i = 0; i < src_regd->n_reg_rules; i++)
406                 memcpy(&regd->reg_rules[i], &src_regd->reg_rules[i],
407                        sizeof(struct ieee80211_reg_rule));
408
409         return regd;
410 }
411
412 #ifdef CONFIG_CFG80211_INTERNAL_REGDB
413 struct reg_regdb_search_request {
414         char alpha2[2];
415         struct list_head list;
416 };
417
418 static LIST_HEAD(reg_regdb_search_list);
419 static DEFINE_MUTEX(reg_regdb_search_mutex);
420
421 static void reg_regdb_search(struct work_struct *work)
422 {
423         struct reg_regdb_search_request *request;
424         const struct ieee80211_regdomain *curdom, *regdom = NULL;
425         int i;
426
427         rtnl_lock();
428
429         mutex_lock(&reg_regdb_search_mutex);
430         while (!list_empty(&reg_regdb_search_list)) {
431                 request = list_first_entry(&reg_regdb_search_list,
432                                            struct reg_regdb_search_request,
433                                            list);
434                 list_del(&request->list);
435
436                 for (i = 0; i < reg_regdb_size; i++) {
437                         curdom = reg_regdb[i];
438
439                         if (alpha2_equal(request->alpha2, curdom->alpha2)) {
440                                 regdom = reg_copy_regd(curdom);
441                                 break;
442                         }
443                 }
444
445                 kfree(request);
446         }
447         mutex_unlock(&reg_regdb_search_mutex);
448
449         if (!IS_ERR_OR_NULL(regdom))
450                 set_regdom(regdom);
451
452         rtnl_unlock();
453 }
454
455 static DECLARE_WORK(reg_regdb_work, reg_regdb_search);
456
457 static void reg_regdb_query(const char *alpha2)
458 {
459         struct reg_regdb_search_request *request;
460
461         if (!alpha2)
462                 return;
463
464         request = kzalloc(sizeof(struct reg_regdb_search_request), GFP_KERNEL);
465         if (!request)
466                 return;
467
468         memcpy(request->alpha2, alpha2, 2);
469
470         mutex_lock(&reg_regdb_search_mutex);
471         list_add_tail(&request->list, &reg_regdb_search_list);
472         mutex_unlock(&reg_regdb_search_mutex);
473
474         schedule_work(&reg_regdb_work);
475 }
476
477 /* Feel free to add any other sanity checks here */
478 static void reg_regdb_size_check(void)
479 {
480         /* We should ideally BUILD_BUG_ON() but then random builds would fail */
481         WARN_ONCE(!reg_regdb_size, "db.txt is empty, you should update it...");
482 }
483 #else
484 static inline void reg_regdb_size_check(void) {}
485 static inline void reg_regdb_query(const char *alpha2) {}
486 #endif /* CONFIG_CFG80211_INTERNAL_REGDB */
487
488 /*
489  * This lets us keep regulatory code which is updated on a regulatory
490  * basis in userspace. Country information is filled in by
491  * reg_device_uevent
492  */
493 static int call_crda(const char *alpha2)
494 {
495         if (!is_world_regdom((char *) alpha2))
496                 pr_info("Calling CRDA for country: %c%c\n",
497                         alpha2[0], alpha2[1]);
498         else
499                 pr_info("Calling CRDA to update world regulatory domain\n");
500
501         /* query internal regulatory database (if it exists) */
502         reg_regdb_query(alpha2);
503
504         return kobject_uevent(&reg_pdev->dev.kobj, KOBJ_CHANGE);
505 }
506
507 static enum reg_request_treatment
508 reg_call_crda(struct regulatory_request *request)
509 {
510         if (call_crda(request->alpha2))
511                 return REG_REQ_IGNORE;
512         return REG_REQ_OK;
513 }
514
515 bool reg_is_valid_request(const char *alpha2)
516 {
517         struct regulatory_request *lr = get_last_request();
518
519         if (!lr || lr->processed)
520                 return false;
521
522         return alpha2_equal(lr->alpha2, alpha2);
523 }
524
525 static const struct ieee80211_regdomain *reg_get_regdomain(struct wiphy *wiphy)
526 {
527         struct regulatory_request *lr = get_last_request();
528
529         /*
530          * Follow the driver's regulatory domain, if present, unless a country
531          * IE has been processed or a user wants to help complaince further
532          */
533         if (lr->initiator != NL80211_REGDOM_SET_BY_COUNTRY_IE &&
534             lr->initiator != NL80211_REGDOM_SET_BY_USER &&
535             wiphy->regd)
536                 return get_wiphy_regdom(wiphy);
537
538         return get_cfg80211_regdom();
539 }
540
541 /* Sanity check on a regulatory rule */
542 static bool is_valid_reg_rule(const struct ieee80211_reg_rule *rule)
543 {
544         const struct ieee80211_freq_range *freq_range = &rule->freq_range;
545         u32 freq_diff;
546
547         if (freq_range->start_freq_khz <= 0 || freq_range->end_freq_khz <= 0)
548                 return false;
549
550         if (freq_range->start_freq_khz > freq_range->end_freq_khz)
551                 return false;
552
553         freq_diff = freq_range->end_freq_khz - freq_range->start_freq_khz;
554
555         if (freq_range->end_freq_khz <= freq_range->start_freq_khz ||
556             freq_range->max_bandwidth_khz > freq_diff)
557                 return false;
558
559         return true;
560 }
561
562 static bool is_valid_rd(const struct ieee80211_regdomain *rd)
563 {
564         const struct ieee80211_reg_rule *reg_rule = NULL;
565         unsigned int i;
566
567         if (!rd->n_reg_rules)
568                 return false;
569
570         if (WARN_ON(rd->n_reg_rules > NL80211_MAX_SUPP_REG_RULES))
571                 return false;
572
573         for (i = 0; i < rd->n_reg_rules; i++) {
574                 reg_rule = &rd->reg_rules[i];
575                 if (!is_valid_reg_rule(reg_rule))
576                         return false;
577         }
578
579         return true;
580 }
581
582 static bool reg_does_bw_fit(const struct ieee80211_freq_range *freq_range,
583                             u32 center_freq_khz, u32 bw_khz)
584 {
585         u32 start_freq_khz, end_freq_khz;
586
587         start_freq_khz = center_freq_khz - (bw_khz/2);
588         end_freq_khz = center_freq_khz + (bw_khz/2);
589
590         if (start_freq_khz >= freq_range->start_freq_khz &&
591             end_freq_khz <= freq_range->end_freq_khz)
592                 return true;
593
594         return false;
595 }
596
597 /**
598  * freq_in_rule_band - tells us if a frequency is in a frequency band
599  * @freq_range: frequency rule we want to query
600  * @freq_khz: frequency we are inquiring about
601  *
602  * This lets us know if a specific frequency rule is or is not relevant to
603  * a specific frequency's band. Bands are device specific and artificial
604  * definitions (the "2.4 GHz band", the "5 GHz band" and the "60GHz band"),
605  * however it is safe for now to assume that a frequency rule should not be
606  * part of a frequency's band if the start freq or end freq are off by more
607  * than 2 GHz for the 2.4 and 5 GHz bands, and by more than 10 GHz for the
608  * 60 GHz band.
609  * This resolution can be lowered and should be considered as we add
610  * regulatory rule support for other "bands".
611  **/
612 static bool freq_in_rule_band(const struct ieee80211_freq_range *freq_range,
613                               u32 freq_khz)
614 {
615 #define ONE_GHZ_IN_KHZ  1000000
616         /*
617          * From 802.11ad: directional multi-gigabit (DMG):
618          * Pertaining to operation in a frequency band containing a channel
619          * with the Channel starting frequency above 45 GHz.
620          */
621         u32 limit = freq_khz > 45 * ONE_GHZ_IN_KHZ ?
622                         10 * ONE_GHZ_IN_KHZ : 2 * ONE_GHZ_IN_KHZ;
623         if (abs(freq_khz - freq_range->start_freq_khz) <= limit)
624                 return true;
625         if (abs(freq_khz - freq_range->end_freq_khz) <= limit)
626                 return true;
627         return false;
628 #undef ONE_GHZ_IN_KHZ
629 }
630
631 /*
632  * Later on we can perhaps use the more restrictive DFS
633  * region but we don't have information for that yet so
634  * for now simply disallow conflicts.
635  */
636 static enum nl80211_dfs_regions
637 reg_intersect_dfs_region(const enum nl80211_dfs_regions dfs_region1,
638                          const enum nl80211_dfs_regions dfs_region2)
639 {
640         if (dfs_region1 != dfs_region2)
641                 return NL80211_DFS_UNSET;
642         return dfs_region1;
643 }
644
645 /*
646  * Helper for regdom_intersect(), this does the real
647  * mathematical intersection fun
648  */
649 static int reg_rules_intersect(const struct ieee80211_reg_rule *rule1,
650                                const struct ieee80211_reg_rule *rule2,
651                                struct ieee80211_reg_rule *intersected_rule)
652 {
653         const struct ieee80211_freq_range *freq_range1, *freq_range2;
654         struct ieee80211_freq_range *freq_range;
655         const struct ieee80211_power_rule *power_rule1, *power_rule2;
656         struct ieee80211_power_rule *power_rule;
657         u32 freq_diff;
658
659         freq_range1 = &rule1->freq_range;
660         freq_range2 = &rule2->freq_range;
661         freq_range = &intersected_rule->freq_range;
662
663         power_rule1 = &rule1->power_rule;
664         power_rule2 = &rule2->power_rule;
665         power_rule = &intersected_rule->power_rule;
666
667         freq_range->start_freq_khz = max(freq_range1->start_freq_khz,
668                                          freq_range2->start_freq_khz);
669         freq_range->end_freq_khz = min(freq_range1->end_freq_khz,
670                                        freq_range2->end_freq_khz);
671         freq_range->max_bandwidth_khz = min(freq_range1->max_bandwidth_khz,
672                                             freq_range2->max_bandwidth_khz);
673
674         freq_diff = freq_range->end_freq_khz - freq_range->start_freq_khz;
675         if (freq_range->max_bandwidth_khz > freq_diff)
676                 freq_range->max_bandwidth_khz = freq_diff;
677
678         power_rule->max_eirp = min(power_rule1->max_eirp,
679                 power_rule2->max_eirp);
680         power_rule->max_antenna_gain = min(power_rule1->max_antenna_gain,
681                 power_rule2->max_antenna_gain);
682
683         intersected_rule->flags = rule1->flags | rule2->flags;
684
685         if (!is_valid_reg_rule(intersected_rule))
686                 return -EINVAL;
687
688         return 0;
689 }
690
691 /**
692  * regdom_intersect - do the intersection between two regulatory domains
693  * @rd1: first regulatory domain
694  * @rd2: second regulatory domain
695  *
696  * Use this function to get the intersection between two regulatory domains.
697  * Once completed we will mark the alpha2 for the rd as intersected, "98",
698  * as no one single alpha2 can represent this regulatory domain.
699  *
700  * Returns a pointer to the regulatory domain structure which will hold the
701  * resulting intersection of rules between rd1 and rd2. We will
702  * kzalloc() this structure for you.
703  */
704 static struct ieee80211_regdomain *
705 regdom_intersect(const struct ieee80211_regdomain *rd1,
706                  const struct ieee80211_regdomain *rd2)
707 {
708         int r, size_of_regd;
709         unsigned int x, y;
710         unsigned int num_rules = 0, rule_idx = 0;
711         const struct ieee80211_reg_rule *rule1, *rule2;
712         struct ieee80211_reg_rule *intersected_rule;
713         struct ieee80211_regdomain *rd;
714         /* This is just a dummy holder to help us count */
715         struct ieee80211_reg_rule dummy_rule;
716
717         if (!rd1 || !rd2)
718                 return NULL;
719
720         /*
721          * First we get a count of the rules we'll need, then we actually
722          * build them. This is to so we can malloc() and free() a
723          * regdomain once. The reason we use reg_rules_intersect() here
724          * is it will return -EINVAL if the rule computed makes no sense.
725          * All rules that do check out OK are valid.
726          */
727
728         for (x = 0; x < rd1->n_reg_rules; x++) {
729                 rule1 = &rd1->reg_rules[x];
730                 for (y = 0; y < rd2->n_reg_rules; y++) {
731                         rule2 = &rd2->reg_rules[y];
732                         if (!reg_rules_intersect(rule1, rule2, &dummy_rule))
733                                 num_rules++;
734                 }
735         }
736
737         if (!num_rules)
738                 return NULL;
739
740         size_of_regd = sizeof(struct ieee80211_regdomain) +
741                        num_rules * sizeof(struct ieee80211_reg_rule);
742
743         rd = kzalloc(size_of_regd, GFP_KERNEL);
744         if (!rd)
745                 return NULL;
746
747         for (x = 0; x < rd1->n_reg_rules && rule_idx < num_rules; x++) {
748                 rule1 = &rd1->reg_rules[x];
749                 for (y = 0; y < rd2->n_reg_rules && rule_idx < num_rules; y++) {
750                         rule2 = &rd2->reg_rules[y];
751                         /*
752                          * This time around instead of using the stack lets
753                          * write to the target rule directly saving ourselves
754                          * a memcpy()
755                          */
756                         intersected_rule = &rd->reg_rules[rule_idx];
757                         r = reg_rules_intersect(rule1, rule2, intersected_rule);
758                         /*
759                          * No need to memset here the intersected rule here as
760                          * we're not using the stack anymore
761                          */
762                         if (r)
763                                 continue;
764                         rule_idx++;
765                 }
766         }
767
768         if (rule_idx != num_rules) {
769                 kfree(rd);
770                 return NULL;
771         }
772
773         rd->n_reg_rules = num_rules;
774         rd->alpha2[0] = '9';
775         rd->alpha2[1] = '8';
776         rd->dfs_region = reg_intersect_dfs_region(rd1->dfs_region,
777                                                   rd2->dfs_region);
778
779         return rd;
780 }
781
782 /*
783  * XXX: add support for the rest of enum nl80211_reg_rule_flags, we may
784  * want to just have the channel structure use these
785  */
786 static u32 map_regdom_flags(u32 rd_flags)
787 {
788         u32 channel_flags = 0;
789         if (rd_flags & NL80211_RRF_NO_IR_ALL)
790                 channel_flags |= IEEE80211_CHAN_NO_IR;
791         if (rd_flags & NL80211_RRF_DFS)
792                 channel_flags |= IEEE80211_CHAN_RADAR;
793         if (rd_flags & NL80211_RRF_NO_OFDM)
794                 channel_flags |= IEEE80211_CHAN_NO_OFDM;
795         return channel_flags;
796 }
797
798 static const struct ieee80211_reg_rule *
799 freq_reg_info_regd(struct wiphy *wiphy, u32 center_freq,
800                    const struct ieee80211_regdomain *regd)
801 {
802         int i;
803         bool band_rule_found = false;
804         bool bw_fits = false;
805
806         if (!regd)
807                 return ERR_PTR(-EINVAL);
808
809         for (i = 0; i < regd->n_reg_rules; i++) {
810                 const struct ieee80211_reg_rule *rr;
811                 const struct ieee80211_freq_range *fr = NULL;
812
813                 rr = &regd->reg_rules[i];
814                 fr = &rr->freq_range;
815
816                 /*
817                  * We only need to know if one frequency rule was
818                  * was in center_freq's band, that's enough, so lets
819                  * not overwrite it once found
820                  */
821                 if (!band_rule_found)
822                         band_rule_found = freq_in_rule_band(fr, center_freq);
823
824                 bw_fits = reg_does_bw_fit(fr, center_freq, MHZ_TO_KHZ(20));
825
826                 if (band_rule_found && bw_fits)
827                         return rr;
828         }
829
830         if (!band_rule_found)
831                 return ERR_PTR(-ERANGE);
832
833         return ERR_PTR(-EINVAL);
834 }
835
836 const struct ieee80211_reg_rule *freq_reg_info(struct wiphy *wiphy,
837                                                u32 center_freq)
838 {
839         const struct ieee80211_regdomain *regd;
840
841         regd = reg_get_regdomain(wiphy);
842
843         return freq_reg_info_regd(wiphy, center_freq, regd);
844 }
845 EXPORT_SYMBOL(freq_reg_info);
846
847 const char *reg_initiator_name(enum nl80211_reg_initiator initiator)
848 {
849         switch (initiator) {
850         case NL80211_REGDOM_SET_BY_CORE:
851                 return "core";
852         case NL80211_REGDOM_SET_BY_USER:
853                 return "user";
854         case NL80211_REGDOM_SET_BY_DRIVER:
855                 return "driver";
856         case NL80211_REGDOM_SET_BY_COUNTRY_IE:
857                 return "country IE";
858         default:
859                 WARN_ON(1);
860                 return "bug";
861         }
862 }
863 EXPORT_SYMBOL(reg_initiator_name);
864
865 #ifdef CONFIG_CFG80211_REG_DEBUG
866 static void chan_reg_rule_print_dbg(struct ieee80211_channel *chan,
867                                     const struct ieee80211_reg_rule *reg_rule)
868 {
869         const struct ieee80211_power_rule *power_rule;
870         const struct ieee80211_freq_range *freq_range;
871         char max_antenna_gain[32];
872
873         power_rule = &reg_rule->power_rule;
874         freq_range = &reg_rule->freq_range;
875
876         if (!power_rule->max_antenna_gain)
877                 snprintf(max_antenna_gain, 32, "N/A");
878         else
879                 snprintf(max_antenna_gain, 32, "%d", power_rule->max_antenna_gain);
880
881         REG_DBG_PRINT("Updating information on frequency %d MHz with regulatory rule:\n",
882                       chan->center_freq);
883
884         REG_DBG_PRINT("%d KHz - %d KHz @ %d KHz), (%s mBi, %d mBm)\n",
885                       freq_range->start_freq_khz, freq_range->end_freq_khz,
886                       freq_range->max_bandwidth_khz, max_antenna_gain,
887                       power_rule->max_eirp);
888 }
889 #else
890 static void chan_reg_rule_print_dbg(struct ieee80211_channel *chan,
891                                     const struct ieee80211_reg_rule *reg_rule)
892 {
893         return;
894 }
895 #endif
896
897 /*
898  * Note that right now we assume the desired channel bandwidth
899  * is always 20 MHz for each individual channel (HT40 uses 20 MHz
900  * per channel, the primary and the extension channel).
901  */
902 static void handle_channel(struct wiphy *wiphy,
903                            enum nl80211_reg_initiator initiator,
904                            struct ieee80211_channel *chan)
905 {
906         u32 flags, bw_flags = 0;
907         const struct ieee80211_reg_rule *reg_rule = NULL;
908         const struct ieee80211_power_rule *power_rule = NULL;
909         const struct ieee80211_freq_range *freq_range = NULL;
910         struct wiphy *request_wiphy = NULL;
911         struct regulatory_request *lr = get_last_request();
912
913         request_wiphy = wiphy_idx_to_wiphy(lr->wiphy_idx);
914
915         flags = chan->orig_flags;
916
917         reg_rule = freq_reg_info(wiphy, MHZ_TO_KHZ(chan->center_freq));
918         if (IS_ERR(reg_rule)) {
919                 /*
920                  * We will disable all channels that do not match our
921                  * received regulatory rule unless the hint is coming
922                  * from a Country IE and the Country IE had no information
923                  * about a band. The IEEE 802.11 spec allows for an AP
924                  * to send only a subset of the regulatory rules allowed,
925                  * so an AP in the US that only supports 2.4 GHz may only send
926                  * a country IE with information for the 2.4 GHz band
927                  * while 5 GHz is still supported.
928                  */
929                 if (initiator == NL80211_REGDOM_SET_BY_COUNTRY_IE &&
930                     PTR_ERR(reg_rule) == -ERANGE)
931                         return;
932
933                 if (lr->initiator == NL80211_REGDOM_SET_BY_DRIVER &&
934                     request_wiphy && request_wiphy == wiphy &&
935                     request_wiphy->regulatory_flags & REGULATORY_STRICT_REG) {
936                         REG_DBG_PRINT("Disabling freq %d MHz for good\n",
937                                       chan->center_freq);
938                         chan->orig_flags |= IEEE80211_CHAN_DISABLED;
939                         chan->flags = chan->orig_flags;
940                 } else {
941                         REG_DBG_PRINT("Disabling freq %d MHz\n",
942                                       chan->center_freq);
943                         chan->flags |= IEEE80211_CHAN_DISABLED;
944                 }
945                 return;
946         }
947
948         chan_reg_rule_print_dbg(chan, reg_rule);
949
950         power_rule = &reg_rule->power_rule;
951         freq_range = &reg_rule->freq_range;
952
953         if (freq_range->max_bandwidth_khz < MHZ_TO_KHZ(40))
954                 bw_flags = IEEE80211_CHAN_NO_HT40;
955         if (freq_range->max_bandwidth_khz < MHZ_TO_KHZ(80))
956                 bw_flags |= IEEE80211_CHAN_NO_80MHZ;
957         if (freq_range->max_bandwidth_khz < MHZ_TO_KHZ(160))
958                 bw_flags |= IEEE80211_CHAN_NO_160MHZ;
959
960         if (lr->initiator == NL80211_REGDOM_SET_BY_DRIVER &&
961             request_wiphy && request_wiphy == wiphy &&
962             request_wiphy->regulatory_flags & REGULATORY_STRICT_REG) {
963                 /*
964                  * This guarantees the driver's requested regulatory domain
965                  * will always be used as a base for further regulatory
966                  * settings
967                  */
968                 chan->flags = chan->orig_flags =
969                         map_regdom_flags(reg_rule->flags) | bw_flags;
970                 chan->max_antenna_gain = chan->orig_mag =
971                         (int) MBI_TO_DBI(power_rule->max_antenna_gain);
972                 chan->max_reg_power = chan->max_power = chan->orig_mpwr =
973                         (int) MBM_TO_DBM(power_rule->max_eirp);
974                 return;
975         }
976
977         chan->dfs_state = NL80211_DFS_USABLE;
978         chan->dfs_state_entered = jiffies;
979
980         chan->beacon_found = false;
981         chan->flags = flags | bw_flags | map_regdom_flags(reg_rule->flags);
982         chan->max_antenna_gain =
983                 min_t(int, chan->orig_mag,
984                       MBI_TO_DBI(power_rule->max_antenna_gain));
985         chan->max_reg_power = (int) MBM_TO_DBM(power_rule->max_eirp);
986         if (chan->orig_mpwr) {
987                 /*
988                  * Devices that use REGULATORY_COUNTRY_IE_FOLLOW_POWER
989                  * will always follow the passed country IE power settings.
990                  */
991                 if (initiator == NL80211_REGDOM_SET_BY_COUNTRY_IE &&
992                     wiphy->regulatory_flags & REGULATORY_COUNTRY_IE_FOLLOW_POWER)
993                         chan->max_power = chan->max_reg_power;
994                 else
995                         chan->max_power = min(chan->orig_mpwr,
996                                               chan->max_reg_power);
997         } else
998                 chan->max_power = chan->max_reg_power;
999 }
1000
1001 static void handle_band(struct wiphy *wiphy,
1002                         enum nl80211_reg_initiator initiator,
1003                         struct ieee80211_supported_band *sband)
1004 {
1005         unsigned int i;
1006
1007         if (!sband)
1008                 return;
1009
1010         for (i = 0; i < sband->n_channels; i++)
1011                 handle_channel(wiphy, initiator, &sband->channels[i]);
1012 }
1013
1014 static bool reg_request_cell_base(struct regulatory_request *request)
1015 {
1016         if (request->initiator != NL80211_REGDOM_SET_BY_USER)
1017                 return false;
1018         return request->user_reg_hint_type == NL80211_USER_REG_HINT_CELL_BASE;
1019 }
1020
1021 bool reg_last_request_cell_base(void)
1022 {
1023         return reg_request_cell_base(get_last_request());
1024 }
1025
1026 #ifdef CONFIG_CFG80211_CERTIFICATION_ONUS
1027 /* Core specific check */
1028 static enum reg_request_treatment
1029 reg_ignore_cell_hint(struct regulatory_request *pending_request)
1030 {
1031         struct regulatory_request *lr = get_last_request();
1032
1033         if (!reg_num_devs_support_basehint)
1034                 return REG_REQ_IGNORE;
1035
1036         if (reg_request_cell_base(lr) &&
1037             !regdom_changes(pending_request->alpha2))
1038                 return REG_REQ_ALREADY_SET;
1039
1040         return REG_REQ_OK;
1041 }
1042
1043 /* Device specific check */
1044 static bool reg_dev_ignore_cell_hint(struct wiphy *wiphy)
1045 {
1046         return !(wiphy->features & NL80211_FEATURE_CELL_BASE_REG_HINTS);
1047 }
1048 #else
1049 static int reg_ignore_cell_hint(struct regulatory_request *pending_request)
1050 {
1051         return REG_REQ_IGNORE;
1052 }
1053
1054 static bool reg_dev_ignore_cell_hint(struct wiphy *wiphy)
1055 {
1056         return true;
1057 }
1058 #endif
1059
1060 static bool wiphy_strict_alpha2_regd(struct wiphy *wiphy)
1061 {
1062         if (wiphy->regulatory_flags & REGULATORY_STRICT_REG &&
1063             !(wiphy->regulatory_flags & REGULATORY_CUSTOM_REG))
1064                 return true;
1065         return false;
1066 }
1067
1068 static bool ignore_reg_update(struct wiphy *wiphy,
1069                               enum nl80211_reg_initiator initiator)
1070 {
1071         struct regulatory_request *lr = get_last_request();
1072
1073         if (!lr) {
1074                 REG_DBG_PRINT("Ignoring regulatory request set by %s "
1075                               "since last_request is not set\n",
1076                               reg_initiator_name(initiator));
1077                 return true;
1078         }
1079
1080         if (initiator == NL80211_REGDOM_SET_BY_CORE &&
1081             wiphy->regulatory_flags & REGULATORY_CUSTOM_REG) {
1082                 REG_DBG_PRINT("Ignoring regulatory request set by %s "
1083                               "since the driver uses its own custom "
1084                               "regulatory domain\n",
1085                               reg_initiator_name(initiator));
1086                 return true;
1087         }
1088
1089         /*
1090          * wiphy->regd will be set once the device has its own
1091          * desired regulatory domain set
1092          */
1093         if (wiphy_strict_alpha2_regd(wiphy) && !wiphy->regd &&
1094             initiator != NL80211_REGDOM_SET_BY_COUNTRY_IE &&
1095             !is_world_regdom(lr->alpha2)) {
1096                 REG_DBG_PRINT("Ignoring regulatory request set by %s "
1097                               "since the driver requires its own regulatory "
1098                               "domain to be set first\n",
1099                               reg_initiator_name(initiator));
1100                 return true;
1101         }
1102
1103         if (reg_request_cell_base(lr))
1104                 return reg_dev_ignore_cell_hint(wiphy);
1105
1106         return false;
1107 }
1108
1109 static bool reg_is_world_roaming(struct wiphy *wiphy)
1110 {
1111         const struct ieee80211_regdomain *cr = get_cfg80211_regdom();
1112         const struct ieee80211_regdomain *wr = get_wiphy_regdom(wiphy);
1113         struct regulatory_request *lr = get_last_request();
1114
1115         if (is_world_regdom(cr->alpha2) || (wr && is_world_regdom(wr->alpha2)))
1116                 return true;
1117
1118         if (lr && lr->initiator != NL80211_REGDOM_SET_BY_COUNTRY_IE &&
1119             wiphy->regulatory_flags & REGULATORY_CUSTOM_REG)
1120                 return true;
1121
1122         return false;
1123 }
1124
1125 static void handle_reg_beacon(struct wiphy *wiphy, unsigned int chan_idx,
1126                               struct reg_beacon *reg_beacon)
1127 {
1128         struct ieee80211_supported_band *sband;
1129         struct ieee80211_channel *chan;
1130         bool channel_changed = false;
1131         struct ieee80211_channel chan_before;
1132
1133         sband = wiphy->bands[reg_beacon->chan.band];
1134         chan = &sband->channels[chan_idx];
1135
1136         if (likely(chan->center_freq != reg_beacon->chan.center_freq))
1137                 return;
1138
1139         if (chan->beacon_found)
1140                 return;
1141
1142         chan->beacon_found = true;
1143
1144         if (!reg_is_world_roaming(wiphy))
1145                 return;
1146
1147         if (wiphy->regulatory_flags & REGULATORY_DISABLE_BEACON_HINTS)
1148                 return;
1149
1150         chan_before.center_freq = chan->center_freq;
1151         chan_before.flags = chan->flags;
1152
1153         if (chan->flags & IEEE80211_CHAN_NO_IR) {
1154                 chan->flags &= ~IEEE80211_CHAN_NO_IR;
1155                 channel_changed = true;
1156         }
1157
1158         if (channel_changed)
1159                 nl80211_send_beacon_hint_event(wiphy, &chan_before, chan);
1160 }
1161
1162 /*
1163  * Called when a scan on a wiphy finds a beacon on
1164  * new channel
1165  */
1166 static void wiphy_update_new_beacon(struct wiphy *wiphy,
1167                                     struct reg_beacon *reg_beacon)
1168 {
1169         unsigned int i;
1170         struct ieee80211_supported_band *sband;
1171
1172         if (!wiphy->bands[reg_beacon->chan.band])
1173                 return;
1174
1175         sband = wiphy->bands[reg_beacon->chan.band];
1176
1177         for (i = 0; i < sband->n_channels; i++)
1178                 handle_reg_beacon(wiphy, i, reg_beacon);
1179 }
1180
1181 /*
1182  * Called upon reg changes or a new wiphy is added
1183  */
1184 static void wiphy_update_beacon_reg(struct wiphy *wiphy)
1185 {
1186         unsigned int i;
1187         struct ieee80211_supported_band *sband;
1188         struct reg_beacon *reg_beacon;
1189
1190         list_for_each_entry(reg_beacon, &reg_beacon_list, list) {
1191                 if (!wiphy->bands[reg_beacon->chan.band])
1192                         continue;
1193                 sband = wiphy->bands[reg_beacon->chan.band];
1194                 for (i = 0; i < sband->n_channels; i++)
1195                         handle_reg_beacon(wiphy, i, reg_beacon);
1196         }
1197 }
1198
1199 /* Reap the advantages of previously found beacons */
1200 static void reg_process_beacons(struct wiphy *wiphy)
1201 {
1202         /*
1203          * Means we are just firing up cfg80211, so no beacons would
1204          * have been processed yet.
1205          */
1206         if (!last_request)
1207                 return;
1208         wiphy_update_beacon_reg(wiphy);
1209 }
1210
1211 static bool is_ht40_allowed(struct ieee80211_channel *chan)
1212 {
1213         if (!chan)
1214                 return false;
1215         if (chan->flags & IEEE80211_CHAN_DISABLED)
1216                 return false;
1217         /* This would happen when regulatory rules disallow HT40 completely */
1218         if ((chan->flags & IEEE80211_CHAN_NO_HT40) == IEEE80211_CHAN_NO_HT40)
1219                 return false;
1220         return true;
1221 }
1222
1223 static void reg_process_ht_flags_channel(struct wiphy *wiphy,
1224                                          struct ieee80211_channel *channel)
1225 {
1226         struct ieee80211_supported_band *sband = wiphy->bands[channel->band];
1227         struct ieee80211_channel *channel_before = NULL, *channel_after = NULL;
1228         unsigned int i;
1229
1230         if (!is_ht40_allowed(channel)) {
1231                 channel->flags |= IEEE80211_CHAN_NO_HT40;
1232                 return;
1233         }
1234
1235         /*
1236          * We need to ensure the extension channels exist to
1237          * be able to use HT40- or HT40+, this finds them (or not)
1238          */
1239         for (i = 0; i < sband->n_channels; i++) {
1240                 struct ieee80211_channel *c = &sband->channels[i];
1241
1242                 if (c->center_freq == (channel->center_freq - 20))
1243                         channel_before = c;
1244                 if (c->center_freq == (channel->center_freq + 20))
1245                         channel_after = c;
1246         }
1247
1248         /*
1249          * Please note that this assumes target bandwidth is 20 MHz,
1250          * if that ever changes we also need to change the below logic
1251          * to include that as well.
1252          */
1253         if (!is_ht40_allowed(channel_before))
1254                 channel->flags |= IEEE80211_CHAN_NO_HT40MINUS;
1255         else
1256                 channel->flags &= ~IEEE80211_CHAN_NO_HT40MINUS;
1257
1258         if (!is_ht40_allowed(channel_after))
1259                 channel->flags |= IEEE80211_CHAN_NO_HT40PLUS;
1260         else
1261                 channel->flags &= ~IEEE80211_CHAN_NO_HT40PLUS;
1262 }
1263
1264 static void reg_process_ht_flags_band(struct wiphy *wiphy,
1265                                       struct ieee80211_supported_band *sband)
1266 {
1267         unsigned int i;
1268
1269         if (!sband)
1270                 return;
1271
1272         for (i = 0; i < sband->n_channels; i++)
1273                 reg_process_ht_flags_channel(wiphy, &sband->channels[i]);
1274 }
1275
1276 static void reg_process_ht_flags(struct wiphy *wiphy)
1277 {
1278         enum ieee80211_band band;
1279
1280         if (!wiphy)
1281                 return;
1282
1283         for (band = 0; band < IEEE80211_NUM_BANDS; band++)
1284                 reg_process_ht_flags_band(wiphy, wiphy->bands[band]);
1285 }
1286
1287 static void reg_call_notifier(struct wiphy *wiphy,
1288                               struct regulatory_request *request)
1289 {
1290         if (wiphy->reg_notifier)
1291                 wiphy->reg_notifier(wiphy, request);
1292 }
1293
1294 static void wiphy_update_regulatory(struct wiphy *wiphy,
1295                                     enum nl80211_reg_initiator initiator)
1296 {
1297         enum ieee80211_band band;
1298         struct regulatory_request *lr = get_last_request();
1299
1300         if (ignore_reg_update(wiphy, initiator)) {
1301                 /*
1302                  * Regulatory updates set by CORE are ignored for custom
1303                  * regulatory cards. Let us notify the changes to the driver,
1304                  * as some drivers used this to restore its orig_* reg domain.
1305                  */
1306                 if (initiator == NL80211_REGDOM_SET_BY_CORE &&
1307                     wiphy->regulatory_flags & REGULATORY_CUSTOM_REG)
1308                         reg_call_notifier(wiphy, lr);
1309                 return;
1310         }
1311
1312         lr->dfs_region = get_cfg80211_regdom()->dfs_region;
1313
1314         for (band = 0; band < IEEE80211_NUM_BANDS; band++)
1315                 handle_band(wiphy, initiator, wiphy->bands[band]);
1316
1317         reg_process_beacons(wiphy);
1318         reg_process_ht_flags(wiphy);
1319         reg_call_notifier(wiphy, lr);
1320 }
1321
1322 static void update_all_wiphy_regulatory(enum nl80211_reg_initiator initiator)
1323 {
1324         struct cfg80211_registered_device *rdev;
1325         struct wiphy *wiphy;
1326
1327         ASSERT_RTNL();
1328
1329         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
1330                 wiphy = &rdev->wiphy;
1331                 wiphy_update_regulatory(wiphy, initiator);
1332         }
1333 }
1334
1335 static void handle_channel_custom(struct wiphy *wiphy,
1336                                   struct ieee80211_channel *chan,
1337                                   const struct ieee80211_regdomain *regd)
1338 {
1339         u32 bw_flags = 0;
1340         const struct ieee80211_reg_rule *reg_rule = NULL;
1341         const struct ieee80211_power_rule *power_rule = NULL;
1342         const struct ieee80211_freq_range *freq_range = NULL;
1343
1344         reg_rule = freq_reg_info_regd(wiphy, MHZ_TO_KHZ(chan->center_freq),
1345                                       regd);
1346
1347         if (IS_ERR(reg_rule)) {
1348                 REG_DBG_PRINT("Disabling freq %d MHz as custom regd has no rule that fits it\n",
1349                               chan->center_freq);
1350                 chan->orig_flags |= IEEE80211_CHAN_DISABLED;
1351                 chan->flags = chan->orig_flags;
1352                 return;
1353         }
1354
1355         chan_reg_rule_print_dbg(chan, reg_rule);
1356
1357         power_rule = &reg_rule->power_rule;
1358         freq_range = &reg_rule->freq_range;
1359
1360         if (freq_range->max_bandwidth_khz < MHZ_TO_KHZ(40))
1361                 bw_flags = IEEE80211_CHAN_NO_HT40;
1362         if (freq_range->max_bandwidth_khz < MHZ_TO_KHZ(80))
1363                 bw_flags |= IEEE80211_CHAN_NO_80MHZ;
1364         if (freq_range->max_bandwidth_khz < MHZ_TO_KHZ(160))
1365                 bw_flags |= IEEE80211_CHAN_NO_160MHZ;
1366
1367         chan->flags |= map_regdom_flags(reg_rule->flags) | bw_flags;
1368         chan->max_antenna_gain = (int) MBI_TO_DBI(power_rule->max_antenna_gain);
1369         chan->max_reg_power = chan->max_power =
1370                 (int) MBM_TO_DBM(power_rule->max_eirp);
1371 }
1372
1373 static void handle_band_custom(struct wiphy *wiphy,
1374                                struct ieee80211_supported_band *sband,
1375                                const struct ieee80211_regdomain *regd)
1376 {
1377         unsigned int i;
1378
1379         if (!sband)
1380                 return;
1381
1382         for (i = 0; i < sband->n_channels; i++)
1383                 handle_channel_custom(wiphy, &sband->channels[i], regd);
1384 }
1385
1386 /* Used by drivers prior to wiphy registration */
1387 void wiphy_apply_custom_regulatory(struct wiphy *wiphy,
1388                                    const struct ieee80211_regdomain *regd)
1389 {
1390         enum ieee80211_band band;
1391         unsigned int bands_set = 0;
1392
1393         WARN(!(wiphy->regulatory_flags & REGULATORY_CUSTOM_REG),
1394              "wiphy should have REGULATORY_CUSTOM_REG\n");
1395         wiphy->regulatory_flags |= REGULATORY_CUSTOM_REG;
1396
1397         for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
1398                 if (!wiphy->bands[band])
1399                         continue;
1400                 handle_band_custom(wiphy, wiphy->bands[band], regd);
1401                 bands_set++;
1402         }
1403
1404         /*
1405          * no point in calling this if it won't have any effect
1406          * on your device's supported bands.
1407          */
1408         WARN_ON(!bands_set);
1409 }
1410 EXPORT_SYMBOL(wiphy_apply_custom_regulatory);
1411
1412 static void reg_set_request_processed(void)
1413 {
1414         bool need_more_processing = false;
1415         struct regulatory_request *lr = get_last_request();
1416
1417         lr->processed = true;
1418
1419         spin_lock(&reg_requests_lock);
1420         if (!list_empty(&reg_requests_list))
1421                 need_more_processing = true;
1422         spin_unlock(&reg_requests_lock);
1423
1424         if (lr->initiator == NL80211_REGDOM_SET_BY_USER)
1425                 cancel_delayed_work(&reg_timeout);
1426
1427         if (need_more_processing)
1428                 schedule_work(&reg_work);
1429 }
1430
1431 /**
1432  * reg_process_hint_core - process core regulatory requests
1433  * @pending_request: a pending core regulatory request
1434  *
1435  * The wireless subsystem can use this function to process
1436  * a regulatory request issued by the regulatory core.
1437  *
1438  * Returns one of the different reg request treatment values.
1439  */
1440 static enum reg_request_treatment
1441 reg_process_hint_core(struct regulatory_request *core_request)
1442 {
1443
1444         core_request->intersect = false;
1445         core_request->processed = false;
1446
1447         reg_update_last_request(core_request);
1448
1449         return reg_call_crda(core_request);
1450 }
1451
1452 static enum reg_request_treatment
1453 __reg_process_hint_user(struct regulatory_request *user_request)
1454 {
1455         struct regulatory_request *lr = get_last_request();
1456
1457         if (reg_request_cell_base(user_request))
1458                 return reg_ignore_cell_hint(user_request);
1459
1460         if (reg_request_cell_base(lr))
1461                 return REG_REQ_IGNORE;
1462
1463         if (lr->initiator == NL80211_REGDOM_SET_BY_COUNTRY_IE)
1464                 return REG_REQ_INTERSECT;
1465         /*
1466          * If the user knows better the user should set the regdom
1467          * to their country before the IE is picked up
1468          */
1469         if (lr->initiator == NL80211_REGDOM_SET_BY_USER &&
1470             lr->intersect)
1471                 return REG_REQ_IGNORE;
1472         /*
1473          * Process user requests only after previous user/driver/core
1474          * requests have been processed
1475          */
1476         if ((lr->initiator == NL80211_REGDOM_SET_BY_CORE ||
1477              lr->initiator == NL80211_REGDOM_SET_BY_DRIVER ||
1478              lr->initiator == NL80211_REGDOM_SET_BY_USER) &&
1479             regdom_changes(lr->alpha2))
1480                 return REG_REQ_IGNORE;
1481
1482         if (!regdom_changes(user_request->alpha2))
1483                 return REG_REQ_ALREADY_SET;
1484
1485         return REG_REQ_OK;
1486 }
1487
1488 /**
1489  * reg_process_hint_user - process user regulatory requests
1490  * @user_request: a pending user regulatory request
1491  *
1492  * The wireless subsystem can use this function to process
1493  * a regulatory request initiated by userspace.
1494  *
1495  * Returns one of the different reg request treatment values.
1496  */
1497 static enum reg_request_treatment
1498 reg_process_hint_user(struct regulatory_request *user_request)
1499 {
1500         enum reg_request_treatment treatment;
1501
1502         treatment = __reg_process_hint_user(user_request);
1503         if (treatment == REG_REQ_IGNORE ||
1504             treatment == REG_REQ_ALREADY_SET) {
1505                 kfree(user_request);
1506                 return treatment;
1507         }
1508
1509         user_request->intersect = treatment == REG_REQ_INTERSECT;
1510         user_request->processed = false;
1511
1512         reg_update_last_request(user_request);
1513
1514         user_alpha2[0] = user_request->alpha2[0];
1515         user_alpha2[1] = user_request->alpha2[1];
1516
1517         return reg_call_crda(user_request);
1518 }
1519
1520 static enum reg_request_treatment
1521 __reg_process_hint_driver(struct regulatory_request *driver_request)
1522 {
1523         struct regulatory_request *lr = get_last_request();
1524
1525         if (lr->initiator == NL80211_REGDOM_SET_BY_CORE) {
1526                 if (regdom_changes(driver_request->alpha2))
1527                         return REG_REQ_OK;
1528                 return REG_REQ_ALREADY_SET;
1529         }
1530
1531         /*
1532          * This would happen if you unplug and plug your card
1533          * back in or if you add a new device for which the previously
1534          * loaded card also agrees on the regulatory domain.
1535          */
1536         if (lr->initiator == NL80211_REGDOM_SET_BY_DRIVER &&
1537             !regdom_changes(driver_request->alpha2))
1538                 return REG_REQ_ALREADY_SET;
1539
1540         return REG_REQ_INTERSECT;
1541 }
1542
1543 /**
1544  * reg_process_hint_driver - process driver regulatory requests
1545  * @driver_request: a pending driver regulatory request
1546  *
1547  * The wireless subsystem can use this function to process
1548  * a regulatory request issued by an 802.11 driver.
1549  *
1550  * Returns one of the different reg request treatment values.
1551  */
1552 static enum reg_request_treatment
1553 reg_process_hint_driver(struct wiphy *wiphy,
1554                         struct regulatory_request *driver_request)
1555 {
1556         const struct ieee80211_regdomain *regd;
1557         enum reg_request_treatment treatment;
1558
1559         treatment = __reg_process_hint_driver(driver_request);
1560
1561         switch (treatment) {
1562         case REG_REQ_OK:
1563                 break;
1564         case REG_REQ_IGNORE:
1565                 kfree(driver_request);
1566                 return treatment;
1567         case REG_REQ_INTERSECT:
1568                 /* fall through */
1569         case REG_REQ_ALREADY_SET:
1570                 regd = reg_copy_regd(get_cfg80211_regdom());
1571                 if (IS_ERR(regd)) {
1572                         kfree(driver_request);
1573                         return REG_REQ_IGNORE;
1574                 }
1575                 rcu_assign_pointer(wiphy->regd, regd);
1576         }
1577
1578
1579         driver_request->intersect = treatment == REG_REQ_INTERSECT;
1580         driver_request->processed = false;
1581
1582         reg_update_last_request(driver_request);
1583
1584         /*
1585          * Since CRDA will not be called in this case as we already
1586          * have applied the requested regulatory domain before we just
1587          * inform userspace we have processed the request
1588          */
1589         if (treatment == REG_REQ_ALREADY_SET) {
1590                 nl80211_send_reg_change_event(driver_request);
1591                 reg_set_request_processed();
1592                 return treatment;
1593         }
1594
1595         return reg_call_crda(driver_request);
1596 }
1597
1598 static enum reg_request_treatment
1599 __reg_process_hint_country_ie(struct wiphy *wiphy,
1600                               struct regulatory_request *country_ie_request)
1601 {
1602         struct wiphy *last_wiphy = NULL;
1603         struct regulatory_request *lr = get_last_request();
1604
1605         if (reg_request_cell_base(lr)) {
1606                 /* Trust a Cell base station over the AP's country IE */
1607                 if (regdom_changes(country_ie_request->alpha2))
1608                         return REG_REQ_IGNORE;
1609                 return REG_REQ_ALREADY_SET;
1610         } else {
1611                 if (wiphy->regulatory_flags & REGULATORY_COUNTRY_IE_IGNORE)
1612                         return REG_REQ_IGNORE;
1613         }
1614
1615         if (unlikely(!is_an_alpha2(country_ie_request->alpha2)))
1616                 return -EINVAL;
1617
1618         if (lr->initiator != NL80211_REGDOM_SET_BY_COUNTRY_IE)
1619                 return REG_REQ_OK;
1620
1621         last_wiphy = wiphy_idx_to_wiphy(lr->wiphy_idx);
1622
1623         if (last_wiphy != wiphy) {
1624                 /*
1625                  * Two cards with two APs claiming different
1626                  * Country IE alpha2s. We could
1627                  * intersect them, but that seems unlikely
1628                  * to be correct. Reject second one for now.
1629                  */
1630                 if (regdom_changes(country_ie_request->alpha2))
1631                         return REG_REQ_IGNORE;
1632                 return REG_REQ_ALREADY_SET;
1633         }
1634         /*
1635          * Two consecutive Country IE hints on the same wiphy.
1636          * This should be picked up early by the driver/stack
1637          */
1638         if (WARN_ON(regdom_changes(country_ie_request->alpha2)))
1639                 return REG_REQ_OK;
1640         return REG_REQ_ALREADY_SET;
1641 }
1642
1643 /**
1644  * reg_process_hint_country_ie - process regulatory requests from country IEs
1645  * @country_ie_request: a regulatory request from a country IE
1646  *
1647  * The wireless subsystem can use this function to process
1648  * a regulatory request issued by a country Information Element.
1649  *
1650  * Returns one of the different reg request treatment values.
1651  */
1652 static enum reg_request_treatment
1653 reg_process_hint_country_ie(struct wiphy *wiphy,
1654                             struct regulatory_request *country_ie_request)
1655 {
1656         enum reg_request_treatment treatment;
1657
1658         treatment = __reg_process_hint_country_ie(wiphy, country_ie_request);
1659
1660         switch (treatment) {
1661         case REG_REQ_OK:
1662                 break;
1663         case REG_REQ_IGNORE:
1664                 /* fall through */
1665         case REG_REQ_ALREADY_SET:
1666                 kfree(country_ie_request);
1667                 return treatment;
1668         case REG_REQ_INTERSECT:
1669                 kfree(country_ie_request);
1670                 /*
1671                  * This doesn't happen yet, not sure we
1672                  * ever want to support it for this case.
1673                  */
1674                 WARN_ONCE(1, "Unexpected intersection for country IEs");
1675                 return REG_REQ_IGNORE;
1676         }
1677
1678         country_ie_request->intersect = false;
1679         country_ie_request->processed = false;
1680
1681         reg_update_last_request(country_ie_request);
1682
1683         return reg_call_crda(country_ie_request);
1684 }
1685
1686 /* This processes *all* regulatory hints */
1687 static void reg_process_hint(struct regulatory_request *reg_request)
1688 {
1689         struct wiphy *wiphy = NULL;
1690         enum reg_request_treatment treatment;
1691
1692         if (reg_request->wiphy_idx != WIPHY_IDX_INVALID)
1693                 wiphy = wiphy_idx_to_wiphy(reg_request->wiphy_idx);
1694
1695         switch (reg_request->initiator) {
1696         case NL80211_REGDOM_SET_BY_CORE:
1697                 reg_process_hint_core(reg_request);
1698                 return;
1699         case NL80211_REGDOM_SET_BY_USER:
1700                 treatment = reg_process_hint_user(reg_request);
1701                 if (treatment == REG_REQ_OK ||
1702                     treatment == REG_REQ_ALREADY_SET)
1703                         return;
1704                 queue_delayed_work(system_power_efficient_wq,
1705                                    &reg_timeout, msecs_to_jiffies(3142));
1706                 return;
1707         case NL80211_REGDOM_SET_BY_DRIVER:
1708                 if (!wiphy)
1709                         goto out_free;
1710                 treatment = reg_process_hint_driver(wiphy, reg_request);
1711                 break;
1712         case NL80211_REGDOM_SET_BY_COUNTRY_IE:
1713                 if (!wiphy)
1714                         goto out_free;
1715                 treatment = reg_process_hint_country_ie(wiphy, reg_request);
1716                 break;
1717         default:
1718                 WARN(1, "invalid initiator %d\n", reg_request->initiator);
1719                 goto out_free;
1720         }
1721
1722         /* This is required so that the orig_* parameters are saved */
1723         if (treatment == REG_REQ_ALREADY_SET && wiphy &&
1724             wiphy->regulatory_flags & REGULATORY_STRICT_REG)
1725                 wiphy_update_regulatory(wiphy, reg_request->initiator);
1726
1727         return;
1728
1729 out_free:
1730         kfree(reg_request);
1731 }
1732
1733 /*
1734  * Processes regulatory hints, this is all the NL80211_REGDOM_SET_BY_*
1735  * Regulatory hints come on a first come first serve basis and we
1736  * must process each one atomically.
1737  */
1738 static void reg_process_pending_hints(void)
1739 {
1740         struct regulatory_request *reg_request, *lr;
1741
1742         lr = get_last_request();
1743
1744         /* When last_request->processed becomes true this will be rescheduled */
1745         if (lr && !lr->processed) {
1746                 REG_DBG_PRINT("Pending regulatory request, waiting for it to be processed...\n");
1747                 return;
1748         }
1749
1750         spin_lock(&reg_requests_lock);
1751
1752         if (list_empty(&reg_requests_list)) {
1753                 spin_unlock(&reg_requests_lock);
1754                 return;
1755         }
1756
1757         reg_request = list_first_entry(&reg_requests_list,
1758                                        struct regulatory_request,
1759                                        list);
1760         list_del_init(&reg_request->list);
1761
1762         spin_unlock(&reg_requests_lock);
1763
1764         reg_process_hint(reg_request);
1765 }
1766
1767 /* Processes beacon hints -- this has nothing to do with country IEs */
1768 static void reg_process_pending_beacon_hints(void)
1769 {
1770         struct cfg80211_registered_device *rdev;
1771         struct reg_beacon *pending_beacon, *tmp;
1772
1773         /* This goes through the _pending_ beacon list */
1774         spin_lock_bh(&reg_pending_beacons_lock);
1775
1776         list_for_each_entry_safe(pending_beacon, tmp,
1777                                  &reg_pending_beacons, list) {
1778                 list_del_init(&pending_beacon->list);
1779
1780                 /* Applies the beacon hint to current wiphys */
1781                 list_for_each_entry(rdev, &cfg80211_rdev_list, list)
1782                         wiphy_update_new_beacon(&rdev->wiphy, pending_beacon);
1783
1784                 /* Remembers the beacon hint for new wiphys or reg changes */
1785                 list_add_tail(&pending_beacon->list, &reg_beacon_list);
1786         }
1787
1788         spin_unlock_bh(&reg_pending_beacons_lock);
1789 }
1790
1791 static void reg_todo(struct work_struct *work)
1792 {
1793         rtnl_lock();
1794         reg_process_pending_hints();
1795         reg_process_pending_beacon_hints();
1796         rtnl_unlock();
1797 }
1798
1799 static void queue_regulatory_request(struct regulatory_request *request)
1800 {
1801         request->alpha2[0] = toupper(request->alpha2[0]);
1802         request->alpha2[1] = toupper(request->alpha2[1]);
1803
1804         spin_lock(&reg_requests_lock);
1805         list_add_tail(&request->list, &reg_requests_list);
1806         spin_unlock(&reg_requests_lock);
1807
1808         schedule_work(&reg_work);
1809 }
1810
1811 /*
1812  * Core regulatory hint -- happens during cfg80211_init()
1813  * and when we restore regulatory settings.
1814  */
1815 static int regulatory_hint_core(const char *alpha2)
1816 {
1817         struct regulatory_request *request;
1818
1819         request = kzalloc(sizeof(struct regulatory_request), GFP_KERNEL);
1820         if (!request)
1821                 return -ENOMEM;
1822
1823         request->alpha2[0] = alpha2[0];
1824         request->alpha2[1] = alpha2[1];
1825         request->initiator = NL80211_REGDOM_SET_BY_CORE;
1826
1827         queue_regulatory_request(request);
1828
1829         return 0;
1830 }
1831
1832 /* User hints */
1833 int regulatory_hint_user(const char *alpha2,
1834                          enum nl80211_user_reg_hint_type user_reg_hint_type)
1835 {
1836         struct regulatory_request *request;
1837
1838         if (WARN_ON(!alpha2))
1839                 return -EINVAL;
1840
1841         request = kzalloc(sizeof(struct regulatory_request), GFP_KERNEL);
1842         if (!request)
1843                 return -ENOMEM;
1844
1845         request->wiphy_idx = WIPHY_IDX_INVALID;
1846         request->alpha2[0] = alpha2[0];
1847         request->alpha2[1] = alpha2[1];
1848         request->initiator = NL80211_REGDOM_SET_BY_USER;
1849         request->user_reg_hint_type = user_reg_hint_type;
1850
1851         queue_regulatory_request(request);
1852
1853         return 0;
1854 }
1855
1856 /* Driver hints */
1857 int regulatory_hint(struct wiphy *wiphy, const char *alpha2)
1858 {
1859         struct regulatory_request *request;
1860
1861         if (WARN_ON(!alpha2 || !wiphy))
1862                 return -EINVAL;
1863
1864         wiphy->regulatory_flags &= ~REGULATORY_CUSTOM_REG;
1865
1866         request = kzalloc(sizeof(struct regulatory_request), GFP_KERNEL);
1867         if (!request)
1868                 return -ENOMEM;
1869
1870         request->wiphy_idx = get_wiphy_idx(wiphy);
1871
1872         request->alpha2[0] = alpha2[0];
1873         request->alpha2[1] = alpha2[1];
1874         request->initiator = NL80211_REGDOM_SET_BY_DRIVER;
1875
1876         queue_regulatory_request(request);
1877
1878         return 0;
1879 }
1880 EXPORT_SYMBOL(regulatory_hint);
1881
1882 void regulatory_hint_country_ie(struct wiphy *wiphy, enum ieee80211_band band,
1883                                 const u8 *country_ie, u8 country_ie_len)
1884 {
1885         char alpha2[2];
1886         enum environment_cap env = ENVIRON_ANY;
1887         struct regulatory_request *request = NULL, *lr;
1888
1889         /* IE len must be evenly divisible by 2 */
1890         if (country_ie_len & 0x01)
1891                 return;
1892
1893         if (country_ie_len < IEEE80211_COUNTRY_IE_MIN_LEN)
1894                 return;
1895
1896         request = kzalloc(sizeof(*request), GFP_KERNEL);
1897         if (!request)
1898                 return;
1899
1900         alpha2[0] = country_ie[0];
1901         alpha2[1] = country_ie[1];
1902
1903         if (country_ie[2] == 'I')
1904                 env = ENVIRON_INDOOR;
1905         else if (country_ie[2] == 'O')
1906                 env = ENVIRON_OUTDOOR;
1907
1908         rcu_read_lock();
1909         lr = get_last_request();
1910
1911         if (unlikely(!lr))
1912                 goto out;
1913
1914         /*
1915          * We will run this only upon a successful connection on cfg80211.
1916          * We leave conflict resolution to the workqueue, where can hold
1917          * the RTNL.
1918          */
1919         if (lr->initiator == NL80211_REGDOM_SET_BY_COUNTRY_IE &&
1920             lr->wiphy_idx != WIPHY_IDX_INVALID)
1921                 goto out;
1922
1923         request->wiphy_idx = get_wiphy_idx(wiphy);
1924         request->alpha2[0] = alpha2[0];
1925         request->alpha2[1] = alpha2[1];
1926         request->initiator = NL80211_REGDOM_SET_BY_COUNTRY_IE;
1927         request->country_ie_env = env;
1928
1929         queue_regulatory_request(request);
1930         request = NULL;
1931 out:
1932         kfree(request);
1933         rcu_read_unlock();
1934 }
1935
1936 static void restore_alpha2(char *alpha2, bool reset_user)
1937 {
1938         /* indicates there is no alpha2 to consider for restoration */
1939         alpha2[0] = '9';
1940         alpha2[1] = '7';
1941
1942         /* The user setting has precedence over the module parameter */
1943         if (is_user_regdom_saved()) {
1944                 /* Unless we're asked to ignore it and reset it */
1945                 if (reset_user) {
1946                         REG_DBG_PRINT("Restoring regulatory settings including user preference\n");
1947                         user_alpha2[0] = '9';
1948                         user_alpha2[1] = '7';
1949
1950                         /*
1951                          * If we're ignoring user settings, we still need to
1952                          * check the module parameter to ensure we put things
1953                          * back as they were for a full restore.
1954                          */
1955                         if (!is_world_regdom(ieee80211_regdom)) {
1956                                 REG_DBG_PRINT("Keeping preference on module parameter ieee80211_regdom: %c%c\n",
1957                                               ieee80211_regdom[0], ieee80211_regdom[1]);
1958                                 alpha2[0] = ieee80211_regdom[0];
1959                                 alpha2[1] = ieee80211_regdom[1];
1960                         }
1961                 } else {
1962                         REG_DBG_PRINT("Restoring regulatory settings while preserving user preference for: %c%c\n",
1963                                       user_alpha2[0], user_alpha2[1]);
1964                         alpha2[0] = user_alpha2[0];
1965                         alpha2[1] = user_alpha2[1];
1966                 }
1967         } else if (!is_world_regdom(ieee80211_regdom)) {
1968                 REG_DBG_PRINT("Keeping preference on module parameter ieee80211_regdom: %c%c\n",
1969                               ieee80211_regdom[0], ieee80211_regdom[1]);
1970                 alpha2[0] = ieee80211_regdom[0];
1971                 alpha2[1] = ieee80211_regdom[1];
1972         } else
1973                 REG_DBG_PRINT("Restoring regulatory settings\n");
1974 }
1975
1976 static void restore_custom_reg_settings(struct wiphy *wiphy)
1977 {
1978         struct ieee80211_supported_band *sband;
1979         enum ieee80211_band band;
1980         struct ieee80211_channel *chan;
1981         int i;
1982
1983         for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
1984                 sband = wiphy->bands[band];
1985                 if (!sband)
1986                         continue;
1987                 for (i = 0; i < sband->n_channels; i++) {
1988                         chan = &sband->channels[i];
1989                         chan->flags = chan->orig_flags;
1990                         chan->max_antenna_gain = chan->orig_mag;
1991                         chan->max_power = chan->orig_mpwr;
1992                         chan->beacon_found = false;
1993                 }
1994         }
1995 }
1996
1997 /*
1998  * Restoring regulatory settings involves ingoring any
1999  * possibly stale country IE information and user regulatory
2000  * settings if so desired, this includes any beacon hints
2001  * learned as we could have traveled outside to another country
2002  * after disconnection. To restore regulatory settings we do
2003  * exactly what we did at bootup:
2004  *
2005  *   - send a core regulatory hint
2006  *   - send a user regulatory hint if applicable
2007  *
2008  * Device drivers that send a regulatory hint for a specific country
2009  * keep their own regulatory domain on wiphy->regd so that does does
2010  * not need to be remembered.
2011  */
2012 static void restore_regulatory_settings(bool reset_user)
2013 {
2014         char alpha2[2];
2015         char world_alpha2[2];
2016         struct reg_beacon *reg_beacon, *btmp;
2017         struct regulatory_request *reg_request, *tmp;
2018         LIST_HEAD(tmp_reg_req_list);
2019         struct cfg80211_registered_device *rdev;
2020
2021         ASSERT_RTNL();
2022
2023         reset_regdomains(true, &world_regdom);
2024         restore_alpha2(alpha2, reset_user);
2025
2026         /*
2027          * If there's any pending requests we simply
2028          * stash them to a temporary pending queue and
2029          * add then after we've restored regulatory
2030          * settings.
2031          */
2032         spin_lock(&reg_requests_lock);
2033         list_for_each_entry_safe(reg_request, tmp, &reg_requests_list, list) {
2034                 if (reg_request->initiator != NL80211_REGDOM_SET_BY_USER)
2035                         continue;
2036                 list_move_tail(&reg_request->list, &tmp_reg_req_list);
2037         }
2038         spin_unlock(&reg_requests_lock);
2039
2040         /* Clear beacon hints */
2041         spin_lock_bh(&reg_pending_beacons_lock);
2042         list_for_each_entry_safe(reg_beacon, btmp, &reg_pending_beacons, list) {
2043                 list_del(&reg_beacon->list);
2044                 kfree(reg_beacon);
2045         }
2046         spin_unlock_bh(&reg_pending_beacons_lock);
2047
2048         list_for_each_entry_safe(reg_beacon, btmp, &reg_beacon_list, list) {
2049                 list_del(&reg_beacon->list);
2050                 kfree(reg_beacon);
2051         }
2052
2053         /* First restore to the basic regulatory settings */
2054         world_alpha2[0] = cfg80211_world_regdom->alpha2[0];
2055         world_alpha2[1] = cfg80211_world_regdom->alpha2[1];
2056
2057         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
2058                 if (rdev->wiphy.regulatory_flags & REGULATORY_CUSTOM_REG)
2059                         restore_custom_reg_settings(&rdev->wiphy);
2060         }
2061
2062         regulatory_hint_core(world_alpha2);
2063
2064         /*
2065          * This restores the ieee80211_regdom module parameter
2066          * preference or the last user requested regulatory
2067          * settings, user regulatory settings takes precedence.
2068          */
2069         if (is_an_alpha2(alpha2))
2070                 regulatory_hint_user(user_alpha2, NL80211_USER_REG_HINT_USER);
2071
2072         spin_lock(&reg_requests_lock);
2073         list_splice_tail_init(&tmp_reg_req_list, &reg_requests_list);
2074         spin_unlock(&reg_requests_lock);
2075
2076         REG_DBG_PRINT("Kicking the queue\n");
2077
2078         schedule_work(&reg_work);
2079 }
2080
2081 void regulatory_hint_disconnect(void)
2082 {
2083         REG_DBG_PRINT("All devices are disconnected, going to restore regulatory settings\n");
2084         restore_regulatory_settings(false);
2085 }
2086
2087 static bool freq_is_chan_12_13_14(u16 freq)
2088 {
2089         if (freq == ieee80211_channel_to_frequency(12, IEEE80211_BAND_2GHZ) ||
2090             freq == ieee80211_channel_to_frequency(13, IEEE80211_BAND_2GHZ) ||
2091             freq == ieee80211_channel_to_frequency(14, IEEE80211_BAND_2GHZ))
2092                 return true;
2093         return false;
2094 }
2095
2096 static bool pending_reg_beacon(struct ieee80211_channel *beacon_chan)
2097 {
2098         struct reg_beacon *pending_beacon;
2099
2100         list_for_each_entry(pending_beacon, &reg_pending_beacons, list)
2101                 if (beacon_chan->center_freq ==
2102                     pending_beacon->chan.center_freq)
2103                         return true;
2104         return false;
2105 }
2106
2107 int regulatory_hint_found_beacon(struct wiphy *wiphy,
2108                                  struct ieee80211_channel *beacon_chan,
2109                                  gfp_t gfp)
2110 {
2111         struct reg_beacon *reg_beacon;
2112         bool processing;
2113
2114         if (beacon_chan->beacon_found ||
2115             beacon_chan->flags & IEEE80211_CHAN_RADAR ||
2116             (beacon_chan->band == IEEE80211_BAND_2GHZ &&
2117              !freq_is_chan_12_13_14(beacon_chan->center_freq)))
2118                 return 0;
2119
2120         spin_lock_bh(&reg_pending_beacons_lock);
2121         processing = pending_reg_beacon(beacon_chan);
2122         spin_unlock_bh(&reg_pending_beacons_lock);
2123
2124         if (processing)
2125                 return 0;
2126
2127         reg_beacon = kzalloc(sizeof(struct reg_beacon), gfp);
2128         if (!reg_beacon)
2129                 return -ENOMEM;
2130
2131         REG_DBG_PRINT("Found new beacon on frequency: %d MHz (Ch %d) on %s\n",
2132                       beacon_chan->center_freq,
2133                       ieee80211_frequency_to_channel(beacon_chan->center_freq),
2134                       wiphy_name(wiphy));
2135
2136         memcpy(&reg_beacon->chan, beacon_chan,
2137                sizeof(struct ieee80211_channel));
2138
2139         /*
2140          * Since we can be called from BH or and non-BH context
2141          * we must use spin_lock_bh()
2142          */
2143         spin_lock_bh(&reg_pending_beacons_lock);
2144         list_add_tail(&reg_beacon->list, &reg_pending_beacons);
2145         spin_unlock_bh(&reg_pending_beacons_lock);
2146
2147         schedule_work(&reg_work);
2148
2149         return 0;
2150 }
2151
2152 static void print_rd_rules(const struct ieee80211_regdomain *rd)
2153 {
2154         unsigned int i;
2155         const struct ieee80211_reg_rule *reg_rule = NULL;
2156         const struct ieee80211_freq_range *freq_range = NULL;
2157         const struct ieee80211_power_rule *power_rule = NULL;
2158
2159         pr_info("  (start_freq - end_freq @ bandwidth), (max_antenna_gain, max_eirp)\n");
2160
2161         for (i = 0; i < rd->n_reg_rules; i++) {
2162                 reg_rule = &rd->reg_rules[i];
2163                 freq_range = &reg_rule->freq_range;
2164                 power_rule = &reg_rule->power_rule;
2165
2166                 /*
2167                  * There may not be documentation for max antenna gain
2168                  * in certain regions
2169                  */
2170                 if (power_rule->max_antenna_gain)
2171                         pr_info("  (%d KHz - %d KHz @ %d KHz), (%d mBi, %d mBm)\n",
2172                                 freq_range->start_freq_khz,
2173                                 freq_range->end_freq_khz,
2174                                 freq_range->max_bandwidth_khz,
2175                                 power_rule->max_antenna_gain,
2176                                 power_rule->max_eirp);
2177                 else
2178                         pr_info("  (%d KHz - %d KHz @ %d KHz), (N/A, %d mBm)\n",
2179                                 freq_range->start_freq_khz,
2180                                 freq_range->end_freq_khz,
2181                                 freq_range->max_bandwidth_khz,
2182                                 power_rule->max_eirp);
2183         }
2184 }
2185
2186 bool reg_supported_dfs_region(enum nl80211_dfs_regions dfs_region)
2187 {
2188         switch (dfs_region) {
2189         case NL80211_DFS_UNSET:
2190         case NL80211_DFS_FCC:
2191         case NL80211_DFS_ETSI:
2192         case NL80211_DFS_JP:
2193                 return true;
2194         default:
2195                 REG_DBG_PRINT("Ignoring uknown DFS master region: %d\n",
2196                               dfs_region);
2197                 return false;
2198         }
2199 }
2200
2201 static void print_regdomain(const struct ieee80211_regdomain *rd)
2202 {
2203         struct regulatory_request *lr = get_last_request();
2204
2205         if (is_intersected_alpha2(rd->alpha2)) {
2206                 if (lr->initiator == NL80211_REGDOM_SET_BY_COUNTRY_IE) {
2207                         struct cfg80211_registered_device *rdev;
2208                         rdev = cfg80211_rdev_by_wiphy_idx(lr->wiphy_idx);
2209                         if (rdev) {
2210                                 pr_info("Current regulatory domain updated by AP to: %c%c\n",
2211                                         rdev->country_ie_alpha2[0],
2212                                         rdev->country_ie_alpha2[1]);
2213                         } else
2214                                 pr_info("Current regulatory domain intersected:\n");
2215                 } else
2216                         pr_info("Current regulatory domain intersected:\n");
2217         } else if (is_world_regdom(rd->alpha2)) {
2218                 pr_info("World regulatory domain updated:\n");
2219         } else {
2220                 if (is_unknown_alpha2(rd->alpha2))
2221                         pr_info("Regulatory domain changed to driver built-in settings (unknown country)\n");
2222                 else {
2223                         if (reg_request_cell_base(lr))
2224                                 pr_info("Regulatory domain changed to country: %c%c by Cell Station\n",
2225                                         rd->alpha2[0], rd->alpha2[1]);
2226                         else
2227                                 pr_info("Regulatory domain changed to country: %c%c\n",
2228                                         rd->alpha2[0], rd->alpha2[1]);
2229                 }
2230         }
2231
2232         pr_info(" DFS Master region: %s", reg_dfs_region_str(rd->dfs_region));
2233         print_rd_rules(rd);
2234 }
2235
2236 static void print_regdomain_info(const struct ieee80211_regdomain *rd)
2237 {
2238         pr_info("Regulatory domain: %c%c\n", rd->alpha2[0], rd->alpha2[1]);
2239         print_rd_rules(rd);
2240 }
2241
2242 static int reg_set_rd_core(const struct ieee80211_regdomain *rd)
2243 {
2244         if (!is_world_regdom(rd->alpha2))
2245                 return -EINVAL;
2246         update_world_regdomain(rd);
2247         return 0;
2248 }
2249
2250 static int reg_set_rd_user(const struct ieee80211_regdomain *rd,
2251                            struct regulatory_request *user_request)
2252 {
2253         const struct ieee80211_regdomain *intersected_rd = NULL;
2254
2255         if (is_world_regdom(rd->alpha2))
2256                 return -EINVAL;
2257
2258         if (!regdom_changes(rd->alpha2))
2259                 return -EALREADY;
2260
2261         if (!is_valid_rd(rd)) {
2262                 pr_err("Invalid regulatory domain detected:\n");
2263                 print_regdomain_info(rd);
2264                 return -EINVAL;
2265         }
2266
2267         if (!user_request->intersect) {
2268                 reset_regdomains(false, rd);
2269                 return 0;
2270         }
2271
2272         intersected_rd = regdom_intersect(rd, get_cfg80211_regdom());
2273         if (!intersected_rd)
2274                 return -EINVAL;
2275
2276         kfree(rd);
2277         rd = NULL;
2278         reset_regdomains(false, intersected_rd);
2279
2280         return 0;
2281 }
2282
2283 static int reg_set_rd_driver(const struct ieee80211_regdomain *rd,
2284                              struct regulatory_request *driver_request)
2285 {
2286         const struct ieee80211_regdomain *regd;
2287         const struct ieee80211_regdomain *intersected_rd = NULL;
2288         const struct ieee80211_regdomain *tmp;
2289         struct wiphy *request_wiphy;
2290
2291         if (is_world_regdom(rd->alpha2))
2292                 return -EINVAL;
2293
2294         if (!regdom_changes(rd->alpha2))
2295                 return -EALREADY;
2296
2297         if (!is_valid_rd(rd)) {
2298                 pr_err("Invalid regulatory domain detected:\n");
2299                 print_regdomain_info(rd);
2300                 return -EINVAL;
2301         }
2302
2303         request_wiphy = wiphy_idx_to_wiphy(driver_request->wiphy_idx);
2304         if (!request_wiphy) {
2305                 queue_delayed_work(system_power_efficient_wq,
2306                                    &reg_timeout, 0);
2307                 return -ENODEV;
2308         }
2309
2310         if (!driver_request->intersect) {
2311                 if (request_wiphy->regd)
2312                         return -EALREADY;
2313
2314                 regd = reg_copy_regd(rd);
2315                 if (IS_ERR(regd))
2316                         return PTR_ERR(regd);
2317
2318                 rcu_assign_pointer(request_wiphy->regd, regd);
2319                 reset_regdomains(false, rd);
2320                 return 0;
2321         }
2322
2323         intersected_rd = regdom_intersect(rd, get_cfg80211_regdom());
2324         if (!intersected_rd)
2325                 return -EINVAL;
2326
2327         /*
2328          * We can trash what CRDA provided now.
2329          * However if a driver requested this specific regulatory
2330          * domain we keep it for its private use
2331          */
2332         tmp = get_wiphy_regdom(request_wiphy);
2333         rcu_assign_pointer(request_wiphy->regd, rd);
2334         rcu_free_regdom(tmp);
2335
2336         rd = NULL;
2337
2338         reset_regdomains(false, intersected_rd);
2339
2340         return 0;
2341 }
2342
2343 static int reg_set_rd_country_ie(const struct ieee80211_regdomain *rd,
2344                                  struct regulatory_request *country_ie_request)
2345 {
2346         struct wiphy *request_wiphy;
2347
2348         if (!is_alpha2_set(rd->alpha2) && !is_an_alpha2(rd->alpha2) &&
2349             !is_unknown_alpha2(rd->alpha2))
2350                 return -EINVAL;
2351
2352         /*
2353          * Lets only bother proceeding on the same alpha2 if the current
2354          * rd is non static (it means CRDA was present and was used last)
2355          * and the pending request came in from a country IE
2356          */
2357
2358         if (!is_valid_rd(rd)) {
2359                 pr_err("Invalid regulatory domain detected:\n");
2360                 print_regdomain_info(rd);
2361                 return -EINVAL;
2362         }
2363
2364         request_wiphy = wiphy_idx_to_wiphy(country_ie_request->wiphy_idx);
2365         if (!request_wiphy) {
2366                 queue_delayed_work(system_power_efficient_wq,
2367                                    &reg_timeout, 0);
2368                 return -ENODEV;
2369         }
2370
2371         if (country_ie_request->intersect)
2372                 return -EINVAL;
2373
2374         reset_regdomains(false, rd);
2375         return 0;
2376 }
2377
2378 /*
2379  * Use this call to set the current regulatory domain. Conflicts with
2380  * multiple drivers can be ironed out later. Caller must've already
2381  * kmalloc'd the rd structure.
2382  */
2383 int set_regdom(const struct ieee80211_regdomain *rd)
2384 {
2385         struct regulatory_request *lr;
2386         int r;
2387
2388         if (!reg_is_valid_request(rd->alpha2)) {
2389                 kfree(rd);
2390                 return -EINVAL;
2391         }
2392
2393         lr = get_last_request();
2394
2395         /* Note that this doesn't update the wiphys, this is done below */
2396         switch (lr->initiator) {
2397         case NL80211_REGDOM_SET_BY_CORE:
2398                 r = reg_set_rd_core(rd);
2399                 break;
2400         case NL80211_REGDOM_SET_BY_USER:
2401                 r = reg_set_rd_user(rd, lr);
2402                 break;
2403         case NL80211_REGDOM_SET_BY_DRIVER:
2404                 r = reg_set_rd_driver(rd, lr);
2405                 break;
2406         case NL80211_REGDOM_SET_BY_COUNTRY_IE:
2407                 r = reg_set_rd_country_ie(rd, lr);
2408                 break;
2409         default:
2410                 WARN(1, "invalid initiator %d\n", lr->initiator);
2411                 return -EINVAL;
2412         }
2413
2414         if (r) {
2415                 if (r == -EALREADY)
2416                         reg_set_request_processed();
2417
2418                 kfree(rd);
2419                 return r;
2420         }
2421
2422         /* This would make this whole thing pointless */
2423         if (WARN_ON(!lr->intersect && rd != get_cfg80211_regdom()))
2424                 return -EINVAL;
2425
2426         /* update all wiphys now with the new established regulatory domain */
2427         update_all_wiphy_regulatory(lr->initiator);
2428
2429         print_regdomain(get_cfg80211_regdom());
2430
2431         nl80211_send_reg_change_event(lr);
2432
2433         reg_set_request_processed();
2434
2435         return 0;
2436 }
2437
2438 int reg_device_uevent(struct device *dev, struct kobj_uevent_env *env)
2439 {
2440         struct regulatory_request *lr;
2441         u8 alpha2[2];
2442         bool add = false;
2443
2444         rcu_read_lock();
2445         lr = get_last_request();
2446         if (lr && !lr->processed) {
2447                 memcpy(alpha2, lr->alpha2, 2);
2448                 add = true;
2449         }
2450         rcu_read_unlock();
2451
2452         if (add)
2453                 return add_uevent_var(env, "COUNTRY=%c%c",
2454                                       alpha2[0], alpha2[1]);
2455         return 0;
2456 }
2457
2458 void wiphy_regulatory_register(struct wiphy *wiphy)
2459 {
2460         struct regulatory_request *lr;
2461
2462         if (!reg_dev_ignore_cell_hint(wiphy))
2463                 reg_num_devs_support_basehint++;
2464
2465         lr = get_last_request();
2466         wiphy_update_regulatory(wiphy, lr->initiator);
2467 }
2468
2469 void wiphy_regulatory_deregister(struct wiphy *wiphy)
2470 {
2471         struct wiphy *request_wiphy = NULL;
2472         struct regulatory_request *lr;
2473
2474         lr = get_last_request();
2475
2476         if (!reg_dev_ignore_cell_hint(wiphy))
2477                 reg_num_devs_support_basehint--;
2478
2479         rcu_free_regdom(get_wiphy_regdom(wiphy));
2480         rcu_assign_pointer(wiphy->regd, NULL);
2481
2482         if (lr)
2483                 request_wiphy = wiphy_idx_to_wiphy(lr->wiphy_idx);
2484
2485         if (!request_wiphy || request_wiphy != wiphy)
2486                 return;
2487
2488         lr->wiphy_idx = WIPHY_IDX_INVALID;
2489         lr->country_ie_env = ENVIRON_ANY;
2490 }
2491
2492 static void reg_timeout_work(struct work_struct *work)
2493 {
2494         REG_DBG_PRINT("Timeout while waiting for CRDA to reply, restoring regulatory settings\n");
2495         rtnl_lock();
2496         restore_regulatory_settings(true);
2497         rtnl_unlock();
2498 }
2499
2500 int __init regulatory_init(void)
2501 {
2502         int err = 0;
2503
2504         reg_pdev = platform_device_register_simple("regulatory", 0, NULL, 0);
2505         if (IS_ERR(reg_pdev))
2506                 return PTR_ERR(reg_pdev);
2507
2508         reg_pdev->dev.type = &reg_device_type;
2509
2510         spin_lock_init(&reg_requests_lock);
2511         spin_lock_init(&reg_pending_beacons_lock);
2512
2513         reg_regdb_size_check();
2514
2515         rcu_assign_pointer(cfg80211_regdomain, cfg80211_world_regdom);
2516
2517         user_alpha2[0] = '9';
2518         user_alpha2[1] = '7';
2519
2520         /* We always try to get an update for the static regdomain */
2521         err = regulatory_hint_core(cfg80211_world_regdom->alpha2);
2522         if (err) {
2523                 if (err == -ENOMEM)
2524                         return err;
2525                 /*
2526                  * N.B. kobject_uevent_env() can fail mainly for when we're out
2527                  * memory which is handled and propagated appropriately above
2528                  * but it can also fail during a netlink_broadcast() or during
2529                  * early boot for call_usermodehelper(). For now treat these
2530                  * errors as non-fatal.
2531                  */
2532                 pr_err("kobject_uevent_env() was unable to call CRDA during init\n");
2533         }
2534
2535         /*
2536          * Finally, if the user set the module parameter treat it
2537          * as a user hint.
2538          */
2539         if (!is_world_regdom(ieee80211_regdom))
2540                 regulatory_hint_user(ieee80211_regdom,
2541                                      NL80211_USER_REG_HINT_USER);
2542
2543         return 0;
2544 }
2545
2546 void regulatory_exit(void)
2547 {
2548         struct regulatory_request *reg_request, *tmp;
2549         struct reg_beacon *reg_beacon, *btmp;
2550
2551         cancel_work_sync(&reg_work);
2552         cancel_delayed_work_sync(&reg_timeout);
2553
2554         /* Lock to suppress warnings */
2555         rtnl_lock();
2556         reset_regdomains(true, NULL);
2557         rtnl_unlock();
2558
2559         dev_set_uevent_suppress(&reg_pdev->dev, true);
2560
2561         platform_device_unregister(reg_pdev);
2562
2563         list_for_each_entry_safe(reg_beacon, btmp, &reg_pending_beacons, list) {
2564                 list_del(&reg_beacon->list);
2565                 kfree(reg_beacon);
2566         }
2567
2568         list_for_each_entry_safe(reg_beacon, btmp, &reg_beacon_list, list) {
2569                 list_del(&reg_beacon->list);
2570                 kfree(reg_beacon);
2571         }
2572
2573         list_for_each_entry_safe(reg_request, tmp, &reg_requests_list, list) {
2574                 list_del(&reg_request->list);
2575                 kfree(reg_request);
2576         }
2577 }