qcacld-3.0: Update bss rate flags after CSA

Currently the driver does not update the peer
rate flags after CSA and hence the rates and
link speed are wrong.

Fix is to update the rate flags after CSA so
that the reported link speed now is correct.

Change-Id: Idb51447b98acc281625d1ed0dae983fe2da2426a
CRs-Fixed: 2669792

core/wma/src/wma_dev_if.c

@@ -86,6 +86,7 @@
 #include "../../core/src/vdev_mgr_ops.h"
 #include "wlan_utility.h"
 #include "wlan_coex_ucfg_api.h"
+#include <wlan_cp_stats_mc_ucfg_api.h>
 
 #ifdef DCS_INTERFERENCE_DETECTION
 #include <wlan_dcs_ucfg_api.h>
@@ -1077,6 +1078,63 @@ void wma_update_peer_phymode_after_vdev_restart(tp_wma_handle wma,
 		wma_update_peer_phymode_sta(wma, iface);
 }
 
+static
+void wma_update_rate_flags_after_vdev_restart(tp_wma_handle wma,
+						struct wma_txrx_node *iface)
+{
+	struct vdev_mlme_obj *vdev_mlme;
+	enum tx_rate_info *rate_flags;
+	enum wlan_phymode bss_phymode;
+	struct wlan_channel *des_chan;
+
+	if (iface->type != WMI_VDEV_TYPE_STA)
+		return;
+
+	vdev_mlme = wlan_vdev_mlme_get_cmpt_obj(iface->vdev);
+	if (!vdev_mlme)
+		return;
+
+	rate_flags = &vdev_mlme->mgmt.rate_info.rate_flags;
+
+	des_chan = wlan_vdev_mlme_get_des_chan(iface->vdev);
+	bss_phymode = des_chan->ch_phymode;
+
+	if (IS_WLAN_PHYMODE_HE(bss_phymode)) {
+		if (des_chan->ch_width == CH_WIDTH_160MHZ ||
+		    des_chan->ch_width == CH_WIDTH_80P80MHZ)
+			*rate_flags |= TX_RATE_HE160;
+		else if (des_chan->ch_width == CH_WIDTH_80MHZ)
+			*rate_flags |= TX_RATE_HE80;
+		else if (des_chan->ch_width)
+			*rate_flags |= TX_RATE_HE40;
+		else
+			*rate_flags |= TX_RATE_HE20;
+	} else if (IS_WLAN_PHYMODE_VHT(bss_phymode)) {
+		if (des_chan->ch_width == CH_WIDTH_80P80MHZ)
+			*rate_flags |= TX_RATE_VHT160;
+		if (des_chan->ch_width == CH_WIDTH_160MHZ)
+			*rate_flags |= TX_RATE_VHT160;
+		if (des_chan->ch_width == CH_WIDTH_80MHZ)
+			*rate_flags |= TX_RATE_VHT80;
+		else if (des_chan->ch_width)
+			*rate_flags |= TX_RATE_VHT40;
+		else
+			*rate_flags |= TX_RATE_VHT20;
+	} else if (IS_WLAN_PHYMODE_HT(bss_phymode)) {
+		if (des_chan->ch_width)
+			*rate_flags |= TX_RATE_HT40;
+		else
+			*rate_flags |= TX_RATE_HT20;
+	} else {
+		*rate_flags = TX_RATE_LEGACY;
+	}
+
+	wma_debug("bss phymode %d rate_flags %x, ch_width %d",
+		  bss_phymode, *rate_flags, des_chan->ch_width);
+
+	ucfg_mc_cp_stats_set_rate_flags(iface->vdev, *rate_flags);
+}
+
 QDF_STATUS wma_handle_channel_switch_resp(tp_wma_handle wma,
 					  struct vdev_start_response *rsp)
 {
@@ -1098,9 +1156,10 @@ QDF_STATUS wma_handle_channel_switch_resp(tp_wma_handle wma,
 	}
 
 	if (QDF_IS_STATUS_SUCCESS(rsp->status) &&
-	    rsp->resp_type == WMI_VDEV_RESTART_RESP_EVENT)
+	    rsp->resp_type == WMI_VDEV_RESTART_RESP_EVENT) {
 		wma_update_peer_phymode_after_vdev_restart(wma, iface);
-
+		wma_update_rate_flags_after_vdev_restart(wma, iface);
+	}
 	if (wma_is_vdev_in_ap_mode(wma, rsp->vdev_id) ||
 	    mlme_is_chan_switch_in_progress(iface->vdev))
 		event = WLAN_VDEV_SM_EV_RESTART_RESP;