qcacmn: Check if opclass is supported by country

When operating class 137 (320 MHz width) is filled in the AFC partial
request on a radio that supports upto 160 MHz bandwidth, there is a
crash observed.

The 160 MHz platform on which the issue is reported does not have
enough memory to accommodate a 320 MHz AFC response. Therefore, the
crash happens.

To fix this issue, fill the operating class in the AFC partial request
based on the max_5g_bw sent in the WMI_REG_CHAN_LIST_CC_EXT_EVENT.

Change-Id: I3be0e0aa7dc2a41f987a6725bc74444311408e51
CRs-Fixed: 3238405

umac/regulatory/core/src/reg_opclass.c

@@ -667,6 +667,63 @@ void reg_dmn_free_6g_opclasses_and_channels(struct wlan_objmgr_pdev *pdev,
 	qdf_mem_free(opclass_lst);
 }
 
+/* The array of bandwidth to 6GHz operating class mapping */
+static const struct bw_opclass_pair bw_opclass_pair_map[] = {
+	{BW_20_MHZ,  131},
+	{BW_40_MHZ,  132},
+	{BW_80_MHZ,  133},
+	{BW_160_MHZ, 134},
+#ifdef WLAN_FEATURE_11BE
+	{BW_320_MHZ, 137},
+#endif
+};
+
+/**
+ * reg_convert_bw_to_opclass() - Convert bandwidth to operating class
+ *
+ * @bw: Bandwidth in MHz
+ */
+static uint8_t reg_convert_bw_to_opclass(uint16_t bw)
+{
+	uint8_t i, num_bws;
+
+	num_bws = QDF_ARRAY_SIZE(bw_opclass_pair_map);
+	for (i = 0; i < num_bws; i++)
+		if (bw == bw_opclass_pair_map[i].bw)
+			return bw_opclass_pair_map[i].opclass;
+
+	return 0;
+}
+
+/**
+ * reg_is_6ghz_op_class_supported() - Check whether the 6GHz opclass is
+ * supported. It is determined by the max bandwidth of the chip.
+ *
+ * @pdev: Pointer to pdev
+ * @op_class: oper class
+ */
+static bool reg_is_6ghz_op_class_supported(struct wlan_objmgr_pdev *pdev,
+					   uint8_t op_class)
+{
+	uint16_t max_5g_bw_supported;
+	uint8_t max_opclass;
+	struct wlan_objmgr_psoc *psoc;
+
+	psoc = wlan_pdev_get_psoc(pdev);
+
+	if (!reg_is_6ghz_op_class(pdev, op_class))
+		return false;
+
+	max_5g_bw_supported = reg_get_max_bw_5G_for_fo(pdev);
+	if (op_class == 136 && max_5g_bw_supported >= BW_20_MHZ &&
+	    reg_is_lower_6g_edge_ch_supp(psoc))
+		return true;
+
+	max_opclass = reg_convert_bw_to_opclass(max_5g_bw_supported);
+
+	return op_class <= max_opclass;
+}
+
 /**
  * reg_dmn_get_num_6g_opclasses() - Calculate the number of opclasses in the
  * 6GHz band.
@@ -685,7 +742,8 @@ static uint8_t reg_dmn_get_num_6g_opclasses(struct wlan_objmgr_pdev *pdev)
 
 		p_lst = op_class_tbl->p_cfi_lst_obj;
 		if (p_lst &&
-		    reg_is_6ghz_op_class(pdev, op_class_tbl->op_class))
+		    reg_is_6ghz_op_class_supported(pdev,
+						   op_class_tbl->op_class))
 			count++;
 
 		op_class_tbl++;
@@ -720,7 +778,8 @@ static void reg_dmn_fill_6g_opcls_chan_lists(struct wlan_objmgr_pdev *pdev,
 
 		p_lst = op_class_tbl->p_cfi_lst_obj;
 		if (p_lst &&
-		    reg_is_6ghz_op_class(pdev, op_class_tbl->op_class)) {
+		    reg_is_6ghz_op_class_supported(pdev,
+						   op_class_tbl->op_class)) {
 			uint8_t j;
 			uint8_t cfi_idx = 0;
 			uint8_t *dst;
@@ -809,7 +868,8 @@ QDF_STATUS reg_dmn_get_6g_opclasses_and_channels(struct wlan_objmgr_pdev *pdev,
 
 		p_lst = op_class_tbl->p_cfi_lst_obj;
 		if (p_lst &&
-		    reg_is_6ghz_op_class(pdev, op_class_tbl->op_class)) {
+		    reg_is_6ghz_op_class_supported(pdev,
+						   op_class_tbl->op_class)) {
 			uint8_t n_supp_cfis = 0;
 			uint8_t j;
 

umac/regulatory/core/src/reg_opclass.h

@@ -483,6 +483,15 @@ uint16_t reg_dmn_get_chanwidth_from_opclass_auto(uint8_t *country,
 						 uint8_t opclass);
 
 #ifdef CONFIG_AFC_SUPPORT
+/**
+ * struct bw_opclass_pair - The bandwidth and operating class pair.
+ * @bw: The input bandwidth
+ * @opclass: The output operating class
+ */
+struct bw_opclass_pair {
+	uint16_t bw;
+	uint8_t opclass;
+};
 
 /**
  * reg_dmn_get_6g_opclasses_and_channels()- Get the following from the