[WCNCR00155320] misc: Fix incorrect bandwidth of Tx MCS distribution

[Description]
1. Fix incorrect bandwidth of Tx MCS distribution
2. Fix crash issue in Rx MCS collection timeout function

Change-Id: I5e1f9f5f8c600e88948db0bc1746bdde6cb2cba4
Signed-off-by: Glenn Tung <glenn.tung@mediatek.com>
CR-Id: WCNCR00155320
Feature: misc
Reviewed-on: http://gerrit.mediatek.inc:8080/1252507
CheckPatch: Check Patch <srv_checkpatch@mediatek.com>
Reviewed-by: George Kuo <george.kuo@mediatek.com>
Reviewed-by: ZD Hu <zd.hu@mediatek.com>
Build: srv_neptune_adm <srv_neptune_adm@mediatek.com>
diff --git a/mgmt/ais_fsm.c b/mgmt/ais_fsm.c
index fce992f..ccaa072 100644
--- a/mgmt/ais_fsm.c
+++ b/mgmt/ais_fsm.c
@@ -3137,7 +3137,12 @@
 VOID aisRxMcsCollectionTimeout(IN P_ADAPTER_T prAdapter, ULONG ulParamPtr)
 {
 	static UINT_8 ucSmapleCnt;
-	UINT_8 ucStaIdx = prAdapter->prAisBssInfo->prStaRecOfAP->ucIndex;
+	UINT_8 ucStaIdx = 0;
+
+	if (prAdapter->prAisBssInfo->prStaRecOfAP == NULL)
+		return;
+
+	ucStaIdx = prAdapter->prAisBssInfo->prStaRecOfAP->ucIndex;
 
 	if (prAdapter->arStaRec[ucStaIdx].fgIsValid && prAdapter->arStaRec[ucStaIdx].fgIsInUse) {
 		prAdapter->arStaRec[ucStaIdx].au4RxVect0Que[ucSmapleCnt] = prAdapter->arStaRec[ucStaIdx].u4RxVector0;
diff --git a/os/linux/gl_wext_priv.c b/os/linux/gl_wext_priv.c
index 241f047..fa3571e 100644
--- a/os/linux/gl_wext_priv.c
+++ b/os/linux/gl_wext_priv.c
@@ -8839,6 +8839,7 @@
 	if (!prHwWlanInfo)
 		return -1;
 
+	prHwWlanInfo->u4Index = prGlueInfo->prAdapter->prAisBssInfo->prStaRecOfAP->ucWlanIndex;
 	rStatus = kalIoctl(prGlueInfo,
 				   wlanoidQueryWlanInfo,
 				   prHwWlanInfo, sizeof(PARAM_HW_WLAN_INFO_T), TRUE, TRUE, TRUE, &u4BufLen);