xref: /linux/drivers/net/wireless/ath/ath12k/reg.c (revision b803c4a4f78834b31ebfbbcea350473333760559)
1 // SPDX-License-Identifier: BSD-3-Clause-Clear
2 /*
3  * Copyright (c) 2018-2021 The Linux Foundation. All rights reserved.
4  * Copyright (c) 2021-2025 Qualcomm Innovation Center, Inc. All rights reserved.
5  */
6 #include <linux/rtnetlink.h>
7 #include "core.h"
8 #include "debug.h"
9 #include "mac.h"
10 
11 /* World regdom to be used in case default regd from fw is unavailable */
12 #define ATH12K_2GHZ_CH01_11      REG_RULE(2412 - 10, 2462 + 10, 40, 0, 20, 0)
13 #define ATH12K_5GHZ_5150_5350    REG_RULE(5150 - 10, 5350 + 10, 80, 0, 30,\
14 					  NL80211_RRF_NO_IR)
15 #define ATH12K_5GHZ_5725_5850    REG_RULE(5725 - 10, 5850 + 10, 80, 0, 30,\
16 					  NL80211_RRF_NO_IR)
17 
18 #define ETSI_WEATHER_RADAR_BAND_LOW		5590
19 #define ETSI_WEATHER_RADAR_BAND_HIGH		5650
20 #define ETSI_WEATHER_RADAR_BAND_CAC_TIMEOUT	600000
21 
22 static const struct ieee80211_regdomain ath12k_world_regd = {
23 	.n_reg_rules = 3,
24 	.alpha2 = "00",
25 	.reg_rules = {
26 		ATH12K_2GHZ_CH01_11,
27 		ATH12K_5GHZ_5150_5350,
28 		ATH12K_5GHZ_5725_5850,
29 	}
30 };
31 
32 static bool ath12k_regdom_changes(struct ieee80211_hw *hw, char *alpha2)
33 {
34 	const struct ieee80211_regdomain *regd;
35 
36 	regd = rcu_dereference_rtnl(hw->wiphy->regd);
37 	/* This can happen during wiphy registration where the previous
38 	 * user request is received before we update the regd received
39 	 * from firmware.
40 	 */
41 	if (!regd)
42 		return true;
43 
44 	return memcmp(regd->alpha2, alpha2, 2) != 0;
45 }
46 
47 static void
48 ath12k_reg_notifier(struct wiphy *wiphy, struct regulatory_request *request)
49 {
50 	struct ieee80211_hw *hw = wiphy_to_ieee80211_hw(wiphy);
51 	struct ath12k_wmi_init_country_arg arg;
52 	struct wmi_set_current_country_arg current_arg = {};
53 	struct ath12k_hw *ah = ath12k_hw_to_ah(hw);
54 	struct ath12k *ar = ath12k_ah_to_ar(ah, 0);
55 	int ret, i;
56 
57 	ath12k_dbg(ar->ab, ATH12K_DBG_REG,
58 		   "Regulatory Notification received for %s\n", wiphy_name(wiphy));
59 
60 	if (request->initiator == NL80211_REGDOM_SET_BY_DRIVER) {
61 		ath12k_dbg(ar->ab, ATH12K_DBG_REG,
62 			   "driver initiated regd update\n");
63 		if (ah->state != ATH12K_HW_STATE_ON)
64 			return;
65 
66 		for_each_ar(ah, ar, i) {
67 			ret = ath12k_reg_update_chan_list(ar, true);
68 			if (ret) {
69 				ath12k_warn(ar->ab,
70 					    "failed to update chan list for pdev %u, ret %d\n",
71 					    i, ret);
72 				break;
73 			}
74 		}
75 		return;
76 	}
77 
78 	/* Currently supporting only General User Hints. Cell base user
79 	 * hints to be handled later.
80 	 * Hints from other sources like Core, Beacons are not expected for
81 	 * self managed wiphy's
82 	 */
83 	if (!(request->initiator == NL80211_REGDOM_SET_BY_USER &&
84 	      request->user_reg_hint_type == NL80211_USER_REG_HINT_USER)) {
85 		ath12k_warn(ar->ab, "Unexpected Regulatory event for this wiphy\n");
86 		return;
87 	}
88 
89 	if (!IS_ENABLED(CONFIG_ATH_REG_DYNAMIC_USER_REG_HINTS)) {
90 		ath12k_dbg(ar->ab, ATH12K_DBG_REG,
91 			   "Country Setting is not allowed\n");
92 		return;
93 	}
94 
95 	if (!ath12k_regdom_changes(hw, request->alpha2)) {
96 		ath12k_dbg(ar->ab, ATH12K_DBG_REG, "Country is already set\n");
97 		return;
98 	}
99 
100 	/* Allow fresh updates to wiphy regd */
101 	ah->regd_updated = false;
102 
103 	/* Send the reg change request to all the radios */
104 	for_each_ar(ah, ar, i) {
105 		if (ar->ab->hw_params->current_cc_support) {
106 			memcpy(&current_arg.alpha2, request->alpha2, 2);
107 			memcpy(&ar->alpha2, &current_arg.alpha2, 2);
108 			ret = ath12k_wmi_send_set_current_country_cmd(ar, &current_arg);
109 			if (ret)
110 				ath12k_warn(ar->ab,
111 					    "failed set current country code: %d\n", ret);
112 		} else {
113 			arg.flags = ALPHA_IS_SET;
114 			memcpy(&arg.cc_info.alpha2, request->alpha2, 2);
115 			arg.cc_info.alpha2[2] = 0;
116 
117 			ret = ath12k_wmi_send_init_country_cmd(ar, &arg);
118 			if (ret)
119 				ath12k_warn(ar->ab,
120 					    "failed set INIT Country code: %d\n", ret);
121 		}
122 
123 		wiphy_lock(wiphy);
124 		ath12k_mac_11d_scan_stop(ar);
125 		wiphy_unlock(wiphy);
126 
127 		ar->regdom_set_by_user = true;
128 	}
129 }
130 
131 int ath12k_reg_update_chan_list(struct ath12k *ar, bool wait)
132 {
133 	struct ieee80211_supported_band **bands;
134 	struct ath12k_wmi_scan_chan_list_arg *arg;
135 	struct ieee80211_channel *channel;
136 	struct ieee80211_hw *hw = ath12k_ar_to_hw(ar);
137 	struct ath12k_wmi_channel_arg *ch;
138 	enum nl80211_band band;
139 	int num_channels = 0;
140 	int i, ret, left;
141 
142 	if (wait && ar->state_11d != ATH12K_11D_IDLE) {
143 		left = wait_for_completion_timeout(&ar->completed_11d_scan,
144 						   ATH12K_SCAN_TIMEOUT_HZ);
145 		if (!left) {
146 			ath12k_dbg(ar->ab, ATH12K_DBG_REG,
147 				   "failed to receive 11d scan complete: timed out\n");
148 			ar->state_11d = ATH12K_11D_IDLE;
149 		}
150 		ath12k_dbg(ar->ab, ATH12K_DBG_REG,
151 			   "reg 11d scan wait left time %d\n", left);
152 	}
153 
154 	if (wait &&
155 	    (ar->scan.state == ATH12K_SCAN_STARTING ||
156 	    ar->scan.state == ATH12K_SCAN_RUNNING)) {
157 		left = wait_for_completion_timeout(&ar->scan.completed,
158 						   ATH12K_SCAN_TIMEOUT_HZ);
159 		if (!left)
160 			ath12k_dbg(ar->ab, ATH12K_DBG_REG,
161 				   "failed to receive hw scan complete: timed out\n");
162 
163 		ath12k_dbg(ar->ab, ATH12K_DBG_REG,
164 			   "reg hw scan wait left time %d\n", left);
165 	}
166 
167 	if (ar->ah->state == ATH12K_HW_STATE_RESTARTING)
168 		return 0;
169 
170 	bands = hw->wiphy->bands;
171 	for (band = 0; band < NUM_NL80211_BANDS; band++) {
172 		if (!(ar->mac.sbands[band].channels && bands[band]))
173 			continue;
174 
175 		for (i = 0; i < bands[band]->n_channels; i++) {
176 			if (bands[band]->channels[i].flags &
177 			    IEEE80211_CHAN_DISABLED)
178 				continue;
179 
180 			num_channels++;
181 		}
182 	}
183 
184 	if (WARN_ON(!num_channels))
185 		return -EINVAL;
186 
187 	arg = kzalloc(struct_size(arg, channel, num_channels), GFP_KERNEL);
188 
189 	if (!arg)
190 		return -ENOMEM;
191 
192 	arg->pdev_id = ar->pdev->pdev_id;
193 	arg->nallchans = num_channels;
194 
195 	ch = arg->channel;
196 
197 	for (band = 0; band < NUM_NL80211_BANDS; band++) {
198 		if (!(ar->mac.sbands[band].channels && bands[band]))
199 			continue;
200 
201 		for (i = 0; i < bands[band]->n_channels; i++) {
202 			channel = &bands[band]->channels[i];
203 
204 			if (channel->flags & IEEE80211_CHAN_DISABLED)
205 				continue;
206 
207 			/* TODO: Set to true/false based on some condition? */
208 			ch->allow_ht = true;
209 			ch->allow_vht = true;
210 			ch->allow_he = true;
211 
212 			ch->dfs_set =
213 				!!(channel->flags & IEEE80211_CHAN_RADAR);
214 			ch->is_chan_passive = !!(channel->flags &
215 						IEEE80211_CHAN_NO_IR);
216 			ch->is_chan_passive |= ch->dfs_set;
217 			ch->mhz = channel->center_freq;
218 			ch->cfreq1 = channel->center_freq;
219 			ch->minpower = 0;
220 			ch->maxpower = channel->max_power * 2;
221 			ch->maxregpower = channel->max_reg_power * 2;
222 			ch->antennamax = channel->max_antenna_gain * 2;
223 
224 			/* TODO: Use appropriate phymodes */
225 			if (channel->band == NL80211_BAND_2GHZ)
226 				ch->phy_mode = MODE_11G;
227 			else
228 				ch->phy_mode = MODE_11A;
229 
230 			if (channel->band == NL80211_BAND_6GHZ &&
231 			    cfg80211_channel_is_psc(channel))
232 				ch->psc_channel = true;
233 
234 			ath12k_dbg(ar->ab, ATH12K_DBG_WMI,
235 				   "mac channel [%d/%d] freq %d maxpower %d regpower %d antenna %d mode %d\n",
236 				   i, arg->nallchans,
237 				   ch->mhz, ch->maxpower, ch->maxregpower,
238 				   ch->antennamax, ch->phy_mode);
239 
240 			ch++;
241 			/* TODO: use quarrter/half rate, cfreq12, dfs_cfreq2
242 			 * set_agile, reg_class_idx
243 			 */
244 		}
245 	}
246 
247 	ret = ath12k_wmi_send_scan_chan_list_cmd(ar, arg);
248 	kfree(arg);
249 
250 	return ret;
251 }
252 
253 static void ath12k_copy_regd(struct ieee80211_regdomain *regd_orig,
254 			     struct ieee80211_regdomain *regd_copy)
255 {
256 	u8 i;
257 
258 	/* The caller should have checked error conditions */
259 	memcpy(regd_copy, regd_orig, sizeof(*regd_orig));
260 
261 	for (i = 0; i < regd_orig->n_reg_rules; i++)
262 		memcpy(&regd_copy->reg_rules[i], &regd_orig->reg_rules[i],
263 		       sizeof(struct ieee80211_reg_rule));
264 }
265 
266 int ath12k_regd_update(struct ath12k *ar, bool init)
267 {
268 	u32 phy_id, freq_low = 0, freq_high = 0, supported_bands, band;
269 	struct ath12k_wmi_hal_reg_capabilities_ext_arg *reg_cap;
270 	struct ath12k_hw *ah = ath12k_ar_to_ah(ar);
271 	struct ieee80211_hw *hw = ah->hw;
272 	struct ieee80211_regdomain *regd, *regd_copy = NULL;
273 	int ret, regd_len, pdev_id;
274 	struct ath12k_base *ab;
275 
276 	ab = ar->ab;
277 
278 	supported_bands = ar->pdev->cap.supported_bands;
279 	if (supported_bands & WMI_HOST_WLAN_2GHZ_CAP) {
280 		band = NL80211_BAND_2GHZ;
281 	} else if (supported_bands & WMI_HOST_WLAN_5GHZ_CAP && !ar->supports_6ghz) {
282 		band = NL80211_BAND_5GHZ;
283 	} else if (supported_bands & WMI_HOST_WLAN_5GHZ_CAP && ar->supports_6ghz) {
284 		band = NL80211_BAND_6GHZ;
285 	} else {
286 		/* This condition is not expected.
287 		 */
288 		WARN_ON(1);
289 		ret = -EINVAL;
290 		goto err;
291 	}
292 
293 	reg_cap = &ab->hal_reg_cap[ar->pdev_idx];
294 
295 	if (ab->hw_params->single_pdev_only && !ar->supports_6ghz) {
296 		phy_id = ar->pdev->cap.band[band].phy_id;
297 		reg_cap = &ab->hal_reg_cap[phy_id];
298 	}
299 
300 	/* Possible that due to reg change, current limits for supported
301 	 * frequency changed. Update that
302 	 */
303 	if (supported_bands & WMI_HOST_WLAN_2GHZ_CAP) {
304 		freq_low = max(reg_cap->low_2ghz_chan, ab->reg_freq_2ghz.start_freq);
305 		freq_high = min(reg_cap->high_2ghz_chan, ab->reg_freq_2ghz.end_freq);
306 	} else if (supported_bands & WMI_HOST_WLAN_5GHZ_CAP && !ar->supports_6ghz) {
307 		freq_low = max(reg_cap->low_5ghz_chan, ab->reg_freq_5ghz.start_freq);
308 		freq_high = min(reg_cap->high_5ghz_chan, ab->reg_freq_5ghz.end_freq);
309 	} else if (supported_bands & WMI_HOST_WLAN_5GHZ_CAP && ar->supports_6ghz) {
310 		freq_low = max(reg_cap->low_5ghz_chan, ab->reg_freq_6ghz.start_freq);
311 		freq_high = min(reg_cap->high_5ghz_chan, ab->reg_freq_6ghz.end_freq);
312 	}
313 
314 	ath12k_mac_update_freq_range(ar, freq_low, freq_high);
315 
316 	ath12k_dbg(ab, ATH12K_DBG_REG, "pdev %u reg updated freq limits %u->%u MHz\n",
317 		   ar->pdev->pdev_id, freq_low, freq_high);
318 
319 	/* If one of the radios within ah has already updated the regd for
320 	 * the wiphy, then avoid setting regd again
321 	 */
322 	if (ah->regd_updated)
323 		return 0;
324 
325 	/* firmware provides reg rules which are similar for 2 GHz and 5 GHz
326 	 * pdev but 6 GHz pdev has superset of all rules including rules for
327 	 * all bands, we prefer 6 GHz pdev's rules to be used for setup of
328 	 * the wiphy regd.
329 	 * If 6 GHz pdev was part of the ath12k_hw, wait for the 6 GHz pdev,
330 	 * else pick the first pdev which calls this function and use its
331 	 * regd to update global hw regd.
332 	 * The regd_updated flag set at the end will not allow any further
333 	 * updates.
334 	 */
335 	if (ah->use_6ghz_regd && !ar->supports_6ghz)
336 		return 0;
337 
338 	pdev_id = ar->pdev_idx;
339 
340 	spin_lock_bh(&ab->base_lock);
341 
342 	if (init) {
343 		/* Apply the regd received during init through
344 		 * WMI_REG_CHAN_LIST_CC event. In case of failure to
345 		 * receive the regd, initialize with a default world
346 		 * regulatory.
347 		 */
348 		if (ab->default_regd[pdev_id]) {
349 			regd = ab->default_regd[pdev_id];
350 		} else {
351 			ath12k_warn(ab,
352 				    "failed to receive default regd during init\n");
353 			regd = (struct ieee80211_regdomain *)&ath12k_world_regd;
354 		}
355 	} else {
356 		regd = ab->new_regd[pdev_id];
357 	}
358 
359 	if (!regd) {
360 		ret = -EINVAL;
361 		spin_unlock_bh(&ab->base_lock);
362 		goto err;
363 	}
364 
365 	regd_len = sizeof(*regd) + (regd->n_reg_rules *
366 		sizeof(struct ieee80211_reg_rule));
367 
368 	regd_copy = kzalloc(regd_len, GFP_ATOMIC);
369 	if (regd_copy)
370 		ath12k_copy_regd(regd, regd_copy);
371 
372 	spin_unlock_bh(&ab->base_lock);
373 
374 	if (!regd_copy) {
375 		ret = -ENOMEM;
376 		goto err;
377 	}
378 
379 	ret = regulatory_set_wiphy_regd(hw->wiphy, regd_copy);
380 
381 	kfree(regd_copy);
382 
383 	if (ret)
384 		goto err;
385 
386 	if (ah->state != ATH12K_HW_STATE_ON)
387 		goto skip;
388 
389 	ah->regd_updated = true;
390 
391 skip:
392 	return 0;
393 err:
394 	ath12k_warn(ab, "failed to perform regd update : %d\n", ret);
395 	return ret;
396 }
397 
398 static enum nl80211_dfs_regions
399 ath12k_map_fw_dfs_region(enum ath12k_dfs_region dfs_region)
400 {
401 	switch (dfs_region) {
402 	case ATH12K_DFS_REG_FCC:
403 	case ATH12K_DFS_REG_CN:
404 		return NL80211_DFS_FCC;
405 	case ATH12K_DFS_REG_ETSI:
406 	case ATH12K_DFS_REG_KR:
407 		return NL80211_DFS_ETSI;
408 	case ATH12K_DFS_REG_MKK:
409 	case ATH12K_DFS_REG_MKK_N:
410 		return NL80211_DFS_JP;
411 	default:
412 		return NL80211_DFS_UNSET;
413 	}
414 }
415 
416 static u32 ath12k_map_fw_reg_flags(u16 reg_flags)
417 {
418 	u32 flags = 0;
419 
420 	if (reg_flags & REGULATORY_CHAN_NO_IR)
421 		flags = NL80211_RRF_NO_IR;
422 
423 	if (reg_flags & REGULATORY_CHAN_RADAR)
424 		flags |= NL80211_RRF_DFS;
425 
426 	if (reg_flags & REGULATORY_CHAN_NO_OFDM)
427 		flags |= NL80211_RRF_NO_OFDM;
428 
429 	if (reg_flags & REGULATORY_CHAN_INDOOR_ONLY)
430 		flags |= NL80211_RRF_NO_OUTDOOR;
431 
432 	if (reg_flags & REGULATORY_CHAN_NO_HT40)
433 		flags |= NL80211_RRF_NO_HT40;
434 
435 	if (reg_flags & REGULATORY_CHAN_NO_80MHZ)
436 		flags |= NL80211_RRF_NO_80MHZ;
437 
438 	if (reg_flags & REGULATORY_CHAN_NO_160MHZ)
439 		flags |= NL80211_RRF_NO_160MHZ;
440 
441 	return flags;
442 }
443 
444 static u32 ath12k_map_fw_phy_flags(u32 phy_flags)
445 {
446 	u32 flags = 0;
447 
448 	if (phy_flags & ATH12K_REG_PHY_BITMAP_NO11AX)
449 		flags |= NL80211_RRF_NO_HE;
450 
451 	if (phy_flags & ATH12K_REG_PHY_BITMAP_NO11BE)
452 		flags |= NL80211_RRF_NO_EHT;
453 
454 	return flags;
455 }
456 
457 static bool
458 ath12k_reg_can_intersect(struct ieee80211_reg_rule *rule1,
459 			 struct ieee80211_reg_rule *rule2)
460 {
461 	u32 start_freq1, end_freq1;
462 	u32 start_freq2, end_freq2;
463 
464 	start_freq1 = rule1->freq_range.start_freq_khz;
465 	start_freq2 = rule2->freq_range.start_freq_khz;
466 
467 	end_freq1 = rule1->freq_range.end_freq_khz;
468 	end_freq2 = rule2->freq_range.end_freq_khz;
469 
470 	if ((start_freq1 >= start_freq2 &&
471 	     start_freq1 < end_freq2) ||
472 	    (start_freq2 > start_freq1 &&
473 	     start_freq2 < end_freq1))
474 		return true;
475 
476 	/* TODO: Should we restrict intersection feasibility
477 	 *  based on min bandwidth of the intersected region also,
478 	 *  say the intersected rule should have a  min bandwidth
479 	 * of 20MHz?
480 	 */
481 
482 	return false;
483 }
484 
485 static void ath12k_reg_intersect_rules(struct ieee80211_reg_rule *rule1,
486 				       struct ieee80211_reg_rule *rule2,
487 				       struct ieee80211_reg_rule *new_rule)
488 {
489 	u32 start_freq1, end_freq1;
490 	u32 start_freq2, end_freq2;
491 	u32 freq_diff, max_bw;
492 
493 	start_freq1 = rule1->freq_range.start_freq_khz;
494 	start_freq2 = rule2->freq_range.start_freq_khz;
495 
496 	end_freq1 = rule1->freq_range.end_freq_khz;
497 	end_freq2 = rule2->freq_range.end_freq_khz;
498 
499 	new_rule->freq_range.start_freq_khz = max_t(u32, start_freq1,
500 						    start_freq2);
501 	new_rule->freq_range.end_freq_khz = min_t(u32, end_freq1, end_freq2);
502 
503 	freq_diff = new_rule->freq_range.end_freq_khz -
504 			new_rule->freq_range.start_freq_khz;
505 	max_bw = min_t(u32, rule1->freq_range.max_bandwidth_khz,
506 		       rule2->freq_range.max_bandwidth_khz);
507 	new_rule->freq_range.max_bandwidth_khz = min_t(u32, max_bw, freq_diff);
508 
509 	new_rule->power_rule.max_antenna_gain =
510 		min_t(u32, rule1->power_rule.max_antenna_gain,
511 		      rule2->power_rule.max_antenna_gain);
512 
513 	new_rule->power_rule.max_eirp = min_t(u32, rule1->power_rule.max_eirp,
514 					      rule2->power_rule.max_eirp);
515 
516 	/* Use the flags of both the rules */
517 	new_rule->flags = rule1->flags | rule2->flags;
518 
519 	/* To be safe, lts use the max cac timeout of both rules */
520 	new_rule->dfs_cac_ms = max_t(u32, rule1->dfs_cac_ms,
521 				     rule2->dfs_cac_ms);
522 }
523 
524 static struct ieee80211_regdomain *
525 ath12k_regd_intersect(struct ieee80211_regdomain *default_regd,
526 		      struct ieee80211_regdomain *curr_regd)
527 {
528 	u8 num_old_regd_rules, num_curr_regd_rules, num_new_regd_rules;
529 	struct ieee80211_reg_rule *old_rule, *curr_rule, *new_rule;
530 	struct ieee80211_regdomain *new_regd = NULL;
531 	u8 i, j, k;
532 
533 	num_old_regd_rules = default_regd->n_reg_rules;
534 	num_curr_regd_rules = curr_regd->n_reg_rules;
535 	num_new_regd_rules = 0;
536 
537 	/* Find the number of intersecting rules to allocate new regd memory */
538 	for (i = 0; i < num_old_regd_rules; i++) {
539 		old_rule = default_regd->reg_rules + i;
540 		for (j = 0; j < num_curr_regd_rules; j++) {
541 			curr_rule = curr_regd->reg_rules + j;
542 
543 			if (ath12k_reg_can_intersect(old_rule, curr_rule))
544 				num_new_regd_rules++;
545 		}
546 	}
547 
548 	if (!num_new_regd_rules)
549 		return NULL;
550 
551 	new_regd = kzalloc(sizeof(*new_regd) + (num_new_regd_rules *
552 			sizeof(struct ieee80211_reg_rule)),
553 			GFP_ATOMIC);
554 
555 	if (!new_regd)
556 		return NULL;
557 
558 	/* We set the new country and dfs region directly and only trim
559 	 * the freq, power, antenna gain by intersecting with the
560 	 * default regdomain. Also MAX of the dfs cac timeout is selected.
561 	 */
562 	new_regd->n_reg_rules = num_new_regd_rules;
563 	memcpy(new_regd->alpha2, curr_regd->alpha2, sizeof(new_regd->alpha2));
564 	new_regd->dfs_region = curr_regd->dfs_region;
565 	new_rule = new_regd->reg_rules;
566 
567 	for (i = 0, k = 0; i < num_old_regd_rules; i++) {
568 		old_rule = default_regd->reg_rules + i;
569 		for (j = 0; j < num_curr_regd_rules; j++) {
570 			curr_rule = curr_regd->reg_rules + j;
571 
572 			if (ath12k_reg_can_intersect(old_rule, curr_rule))
573 				ath12k_reg_intersect_rules(old_rule, curr_rule,
574 							   (new_rule + k++));
575 		}
576 	}
577 	return new_regd;
578 }
579 
580 static const char *
581 ath12k_reg_get_regdom_str(enum nl80211_dfs_regions dfs_region)
582 {
583 	switch (dfs_region) {
584 	case NL80211_DFS_FCC:
585 		return "FCC";
586 	case NL80211_DFS_ETSI:
587 		return "ETSI";
588 	case NL80211_DFS_JP:
589 		return "JP";
590 	default:
591 		return "UNSET";
592 	}
593 }
594 
595 static u16
596 ath12k_reg_adjust_bw(u16 start_freq, u16 end_freq, u16 max_bw)
597 {
598 	u16 bw;
599 
600 	bw = end_freq - start_freq;
601 	bw = min_t(u16, bw, max_bw);
602 
603 	if (bw >= 80 && bw < 160)
604 		bw = 80;
605 	else if (bw >= 40 && bw < 80)
606 		bw = 40;
607 	else if (bw < 40)
608 		bw = 20;
609 
610 	return bw;
611 }
612 
613 static void
614 ath12k_reg_update_rule(struct ieee80211_reg_rule *reg_rule, u32 start_freq,
615 		       u32 end_freq, u32 bw, u32 ant_gain, u32 reg_pwr,
616 		       u32 reg_flags)
617 {
618 	reg_rule->freq_range.start_freq_khz = MHZ_TO_KHZ(start_freq);
619 	reg_rule->freq_range.end_freq_khz = MHZ_TO_KHZ(end_freq);
620 	reg_rule->freq_range.max_bandwidth_khz = MHZ_TO_KHZ(bw);
621 	reg_rule->power_rule.max_antenna_gain = DBI_TO_MBI(ant_gain);
622 	reg_rule->power_rule.max_eirp = DBM_TO_MBM(reg_pwr);
623 	reg_rule->flags = reg_flags;
624 }
625 
626 static void
627 ath12k_reg_update_weather_radar_band(struct ath12k_base *ab,
628 				     struct ieee80211_regdomain *regd,
629 				     struct ath12k_reg_rule *reg_rule,
630 				     u8 *rule_idx, u32 flags, u16 max_bw)
631 {
632 	u32 end_freq;
633 	u16 bw;
634 	u8 i;
635 
636 	i = *rule_idx;
637 
638 	bw = ath12k_reg_adjust_bw(reg_rule->start_freq,
639 				  ETSI_WEATHER_RADAR_BAND_LOW, max_bw);
640 
641 	ath12k_reg_update_rule(regd->reg_rules + i, reg_rule->start_freq,
642 			       ETSI_WEATHER_RADAR_BAND_LOW, bw,
643 			       reg_rule->ant_gain, reg_rule->reg_power,
644 			       flags);
645 
646 	ath12k_dbg(ab, ATH12K_DBG_REG,
647 		   "\t%d. (%d - %d @ %d) (%d, %d) (%d ms) (FLAGS %d)\n",
648 		   i + 1, reg_rule->start_freq, ETSI_WEATHER_RADAR_BAND_LOW,
649 		   bw, reg_rule->ant_gain, reg_rule->reg_power,
650 		   regd->reg_rules[i].dfs_cac_ms,
651 		   flags);
652 
653 	if (reg_rule->end_freq > ETSI_WEATHER_RADAR_BAND_HIGH)
654 		end_freq = ETSI_WEATHER_RADAR_BAND_HIGH;
655 	else
656 		end_freq = reg_rule->end_freq;
657 
658 	bw = ath12k_reg_adjust_bw(ETSI_WEATHER_RADAR_BAND_LOW, end_freq,
659 				  max_bw);
660 
661 	i++;
662 
663 	ath12k_reg_update_rule(regd->reg_rules + i,
664 			       ETSI_WEATHER_RADAR_BAND_LOW, end_freq, bw,
665 			       reg_rule->ant_gain, reg_rule->reg_power,
666 			       flags);
667 
668 	regd->reg_rules[i].dfs_cac_ms = ETSI_WEATHER_RADAR_BAND_CAC_TIMEOUT;
669 
670 	ath12k_dbg(ab, ATH12K_DBG_REG,
671 		   "\t%d. (%d - %d @ %d) (%d, %d) (%d ms) (FLAGS %d)\n",
672 		   i + 1, ETSI_WEATHER_RADAR_BAND_LOW, end_freq,
673 		   bw, reg_rule->ant_gain, reg_rule->reg_power,
674 		   regd->reg_rules[i].dfs_cac_ms,
675 		   flags);
676 
677 	if (end_freq == reg_rule->end_freq) {
678 		regd->n_reg_rules--;
679 		*rule_idx = i;
680 		return;
681 	}
682 
683 	bw = ath12k_reg_adjust_bw(ETSI_WEATHER_RADAR_BAND_HIGH,
684 				  reg_rule->end_freq, max_bw);
685 
686 	i++;
687 
688 	ath12k_reg_update_rule(regd->reg_rules + i, ETSI_WEATHER_RADAR_BAND_HIGH,
689 			       reg_rule->end_freq, bw,
690 			       reg_rule->ant_gain, reg_rule->reg_power,
691 			       flags);
692 
693 	ath12k_dbg(ab, ATH12K_DBG_REG,
694 		   "\t%d. (%d - %d @ %d) (%d, %d) (%d ms) (FLAGS %d)\n",
695 		   i + 1, ETSI_WEATHER_RADAR_BAND_HIGH, reg_rule->end_freq,
696 		   bw, reg_rule->ant_gain, reg_rule->reg_power,
697 		   regd->reg_rules[i].dfs_cac_ms,
698 		   flags);
699 
700 	*rule_idx = i;
701 }
702 
703 static void ath12k_reg_update_freq_range(struct ath12k_reg_freq *reg_freq,
704 					 struct ath12k_reg_rule *reg_rule)
705 {
706 	if (reg_freq->start_freq > reg_rule->start_freq)
707 		reg_freq->start_freq = reg_rule->start_freq;
708 
709 	if (reg_freq->end_freq < reg_rule->end_freq)
710 		reg_freq->end_freq = reg_rule->end_freq;
711 }
712 
713 struct ieee80211_regdomain *
714 ath12k_reg_build_regd(struct ath12k_base *ab,
715 		      struct ath12k_reg_info *reg_info, bool intersect)
716 {
717 	struct ieee80211_regdomain *tmp_regd, *default_regd, *new_regd = NULL;
718 	struct ath12k_reg_rule *reg_rule;
719 	u8 i = 0, j = 0, k = 0;
720 	u8 num_rules;
721 	u16 max_bw;
722 	u32 flags;
723 	char alpha2[3];
724 
725 	num_rules = reg_info->num_5g_reg_rules + reg_info->num_2g_reg_rules;
726 
727 	/* FIXME: Currently taking reg rules for 6G only from Indoor AP mode list.
728 	 * This can be updated to choose the combination dynamically based on AP
729 	 * type and client type, after complete 6G regulatory support is added.
730 	 */
731 	if (reg_info->is_ext_reg_event)
732 		num_rules += reg_info->num_6g_reg_rules_ap[WMI_REG_INDOOR_AP];
733 
734 	if (!num_rules)
735 		goto ret;
736 
737 	/* Add max additional rules to accommodate weather radar band */
738 	if (reg_info->dfs_region == ATH12K_DFS_REG_ETSI)
739 		num_rules += 2;
740 
741 	tmp_regd = kzalloc(sizeof(*tmp_regd) +
742 			   (num_rules * sizeof(struct ieee80211_reg_rule)),
743 			   GFP_ATOMIC);
744 	if (!tmp_regd)
745 		goto ret;
746 
747 	memcpy(tmp_regd->alpha2, reg_info->alpha2, REG_ALPHA2_LEN + 1);
748 	memcpy(alpha2, reg_info->alpha2, REG_ALPHA2_LEN + 1);
749 	alpha2[2] = '\0';
750 	tmp_regd->dfs_region = ath12k_map_fw_dfs_region(reg_info->dfs_region);
751 
752 	ath12k_dbg(ab, ATH12K_DBG_REG,
753 		   "\r\nCountry %s, CFG Regdomain %s FW Regdomain %d, num_reg_rules %d\n",
754 		   alpha2, ath12k_reg_get_regdom_str(tmp_regd->dfs_region),
755 		   reg_info->dfs_region, num_rules);
756 
757 	/* Reset start and end frequency for each band
758 	 */
759 	ab->reg_freq_5ghz.start_freq = INT_MAX;
760 	ab->reg_freq_5ghz.end_freq = 0;
761 	ab->reg_freq_2ghz.start_freq = INT_MAX;
762 	ab->reg_freq_2ghz.end_freq = 0;
763 	ab->reg_freq_6ghz.start_freq = INT_MAX;
764 	ab->reg_freq_6ghz.end_freq = 0;
765 
766 	/* Update reg_rules[] below. Firmware is expected to
767 	 * send these rules in order(2G rules first and then 5G)
768 	 */
769 	for (; i < num_rules; i++) {
770 		if (reg_info->num_2g_reg_rules &&
771 		    (i < reg_info->num_2g_reg_rules)) {
772 			reg_rule = reg_info->reg_rules_2g_ptr + i;
773 			max_bw = min_t(u16, reg_rule->max_bw,
774 				       reg_info->max_bw_2g);
775 			flags = 0;
776 			ath12k_reg_update_freq_range(&ab->reg_freq_2ghz, reg_rule);
777 		} else if (reg_info->num_5g_reg_rules &&
778 			   (j < reg_info->num_5g_reg_rules)) {
779 			reg_rule = reg_info->reg_rules_5g_ptr + j++;
780 			max_bw = min_t(u16, reg_rule->max_bw,
781 				       reg_info->max_bw_5g);
782 
783 			/* FW doesn't pass NL80211_RRF_AUTO_BW flag for
784 			 * BW Auto correction, we can enable this by default
785 			 * for all 5G rules here. The regulatory core performs
786 			 * BW correction if required and applies flags as
787 			 * per other BW rule flags we pass from here
788 			 */
789 			flags = NL80211_RRF_AUTO_BW;
790 			ath12k_reg_update_freq_range(&ab->reg_freq_5ghz, reg_rule);
791 		} else if (reg_info->is_ext_reg_event &&
792 			   reg_info->num_6g_reg_rules_ap[WMI_REG_INDOOR_AP] &&
793 			(k < reg_info->num_6g_reg_rules_ap[WMI_REG_INDOOR_AP])) {
794 			reg_rule = reg_info->reg_rules_6g_ap_ptr[WMI_REG_INDOOR_AP] + k++;
795 			max_bw = min_t(u16, reg_rule->max_bw,
796 				       reg_info->max_bw_6g_ap[WMI_REG_INDOOR_AP]);
797 			flags = NL80211_RRF_AUTO_BW;
798 			ath12k_reg_update_freq_range(&ab->reg_freq_6ghz, reg_rule);
799 		} else {
800 			break;
801 		}
802 
803 		flags |= ath12k_map_fw_reg_flags(reg_rule->flags);
804 		flags |= ath12k_map_fw_phy_flags(reg_info->phybitmap);
805 
806 		ath12k_reg_update_rule(tmp_regd->reg_rules + i,
807 				       reg_rule->start_freq,
808 				       reg_rule->end_freq, max_bw,
809 				       reg_rule->ant_gain, reg_rule->reg_power,
810 				       flags);
811 
812 		/* Update dfs cac timeout if the dfs domain is ETSI and the
813 		 * new rule covers weather radar band.
814 		 * Default value of '0' corresponds to 60s timeout, so no
815 		 * need to update that for other rules.
816 		 */
817 		if (flags & NL80211_RRF_DFS &&
818 		    reg_info->dfs_region == ATH12K_DFS_REG_ETSI &&
819 		    (reg_rule->end_freq > ETSI_WEATHER_RADAR_BAND_LOW &&
820 		    reg_rule->start_freq < ETSI_WEATHER_RADAR_BAND_HIGH)){
821 			ath12k_reg_update_weather_radar_band(ab, tmp_regd,
822 							     reg_rule, &i,
823 							     flags, max_bw);
824 			continue;
825 		}
826 
827 		if (reg_info->is_ext_reg_event) {
828 			ath12k_dbg(ab, ATH12K_DBG_REG, "\t%d. (%d - %d @ %d) (%d, %d) (%d ms) (FLAGS %d) (%d, %d)\n",
829 				   i + 1, reg_rule->start_freq, reg_rule->end_freq,
830 				   max_bw, reg_rule->ant_gain, reg_rule->reg_power,
831 				   tmp_regd->reg_rules[i].dfs_cac_ms,
832 				   flags, reg_rule->psd_flag, reg_rule->psd_eirp);
833 		} else {
834 			ath12k_dbg(ab, ATH12K_DBG_REG,
835 				   "\t%d. (%d - %d @ %d) (%d, %d) (%d ms) (FLAGS %d)\n",
836 				   i + 1, reg_rule->start_freq, reg_rule->end_freq,
837 				   max_bw, reg_rule->ant_gain, reg_rule->reg_power,
838 				   tmp_regd->reg_rules[i].dfs_cac_ms,
839 				   flags);
840 		}
841 	}
842 
843 	tmp_regd->n_reg_rules = i;
844 
845 	if (intersect) {
846 		default_regd = ab->default_regd[reg_info->phy_id];
847 
848 		/* Get a new regd by intersecting the received regd with
849 		 * our default regd.
850 		 */
851 		new_regd = ath12k_regd_intersect(default_regd, tmp_regd);
852 		kfree(tmp_regd);
853 		if (!new_regd) {
854 			ath12k_warn(ab, "Unable to create intersected regdomain\n");
855 			goto ret;
856 		}
857 	} else {
858 		new_regd = tmp_regd;
859 	}
860 
861 ret:
862 	return new_regd;
863 }
864 
865 void ath12k_regd_update_work(struct work_struct *work)
866 {
867 	struct ath12k *ar = container_of(work, struct ath12k,
868 					 regd_update_work);
869 	int ret;
870 
871 	ret = ath12k_regd_update(ar, false);
872 	if (ret) {
873 		/* Firmware has already moved to the new regd. We need
874 		 * to maintain channel consistency across FW, Host driver
875 		 * and userspace. Hence as a fallback mechanism we can set
876 		 * the prev or default country code to the firmware.
877 		 */
878 		/* TODO: Implement Fallback Mechanism */
879 	}
880 }
881 
882 void ath12k_reg_init(struct ieee80211_hw *hw)
883 {
884 	hw->wiphy->regulatory_flags = REGULATORY_WIPHY_SELF_MANAGED;
885 	hw->wiphy->flags |= WIPHY_FLAG_NOTIFY_REGDOM_BY_DRIVER;
886 	hw->wiphy->reg_notifier = ath12k_reg_notifier;
887 }
888 
889 void ath12k_reg_free(struct ath12k_base *ab)
890 {
891 	int i;
892 
893 	mutex_lock(&ab->core_lock);
894 	for (i = 0; i < ab->hw_params->max_radios; i++) {
895 		kfree(ab->default_regd[i]);
896 		kfree(ab->new_regd[i]);
897 		ab->default_regd[i] = NULL;
898 		ab->new_regd[i] = NULL;
899 	}
900 	mutex_unlock(&ab->core_lock);
901 }
902