From 773eec3e966f449e23790b56dd6f0cdb8e737393 Mon Sep 17 00:00:00 2001
From: Jia Ding <quic_jiad@quicinc.com>
Date: Thu, 7 Apr 2022 19:39:32 +0800
Subject: [PATCH] qcacld-3.0: Properly update EHT caps support for 320MHz

320MHz bandwidth is supported only for single PHY mode. Therefore populate
320MHz support from HW_MODE_SINGLE mac_phy_caps_ext2 entry.

Change-Id: I298b13a82ca0532aa6da76d12edd12a57c74121a
CRs-Fixed: 3165631
---
 core/wma/src/wma_eht.c | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/core/wma/src/wma_eht.c b/core/wma/src/wma_eht.c
index a6e9e43c39..a00b9ccd4e 100644
--- a/core/wma/src/wma_eht.c
+++ b/core/wma/src/wma_eht.c
@@ -125,6 +125,24 @@ void wma_eht_update_tgt_services(struct wmi_unified *wmi_handle,
 	}
 }
 
+static void
+wma_update_eht_cap_support_for_320mhz(struct target_psoc_info *tgt_hdl,
+				      tDot11fIEeht_cap *eht_cap)
+{
+	struct wlan_psoc_host_mac_phy_caps_ext2 *cap;
+
+	cap = target_psoc_get_mac_phy_cap_ext2_for_mode(
+			tgt_hdl, WMI_HOST_HW_MODE_SINGLE);
+	if (!cap) {
+		wma_debug("HW_MODE_SINGLE does not exist");
+		return;
+	}
+
+	eht_cap->support_320mhz_6ghz = WMI_EHTCAP_PHY_320MHZIN6GHZ_GET(
+			cap->eht_cap_phy_info_5G);
+	wma_debug("Support for 320MHz 0x%01x", eht_cap->support_320mhz_6ghz);
+}
+
 void wma_update_target_ext_eht_cap(struct target_psoc_info *tgt_hdl,
 				   struct wma_tgt_cfg *tgt_cfg)
 {
@@ -185,6 +203,9 @@ void wma_update_target_ext_eht_cap(struct target_psoc_info *tgt_hdl,
 		}
 	}
 	qdf_mem_copy(eht_cap, &eht_cap_mac, sizeof(tDot11fIEeht_cap));
+
+	wma_update_eht_cap_support_for_320mhz(tgt_hdl, eht_cap);
+
 	wma_print_eht_cap(eht_cap);
 }