/*
  * Copyright (c) 2016 MediaTek Inc.  All rights reserved.
  *
  * This software is available to you under a choice of one of two
  * licenses.  You may choose to be licensed under the terms of the GNU
  * General Public License (GPL) Version 2, available from the file
  * COPYING in the main directory of this source tree, or the
  * BSD license below:
  *
  *     Redistribution and use in source and binary forms, with or
  *     without modification, are permitted provided that the following
  *     conditions are met:
  *
  *      - Redistributions of source code must retain the above
  *        copyright notice, this list of conditions and the following
  *        disclaimer.
  *
  *      - Redistributions in binary form must reproduce the above
  *        copyright notice, this list of conditions and the following
  *        disclaimer in the documentation and/or other materials
  *        provided with the distribution.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  * SOFTWARE.
  */


#include "rt_config.h"

UCHAR PowerConstraintIE[3] = { IE_POWER_CONSTRAINT, 1, 3 };

/*
	==========================================================================
	Description:
		Used to check the necessary to send Beancon.
	return value
		0: mean no necessary.
		0: mean need to send Beacon for the service.
	==========================================================================
*/
BOOLEAN BeaconTransmitRequired(RTMP_ADAPTER *pAd, INT apidx, MULTISSID_STRUCT *pMbss)
{
#ifdef WDS_SUPPORT
	UCHAR idx;
#endif /* WDS_SUPPORT */
	BOOLEAN result = FALSE;

	do {
#ifdef WDS_SUPPORT
		if (pAd->WdsTab.Mode == WDS_BRIDGE_MODE)
			break;
#endif /* WDS_SUPPORT */

#ifdef CARRIER_DETECTION_SUPPORT
		if (isCarrierDetectExist(pAd) == TRUE)
			break;
#endif /* CARRIER_DETECTION_SUPPORT */


		if (pMbss->BcnBufIdx >= HW_BEACON_MAX_NUM)
			break;
		if (pAd->BeaconOffset[pMbss->BcnBufIdx] == 0)
			break;

		if (apidx == MAIN_MBSSID) {
			if (pMbss->bBcnSntReq == TRUE) {
				result = TRUE;
				break;
			}
#ifdef WDS_SUPPORT
			for (idx = 0; idx < MAX_WDS_ENTRY; idx++) {
				if ((pAd->WdsTab.WdsEntry[idx].wdev.if_dev != NULL)
				    &&
				    (RTMP_OS_NETDEV_STATE_RUNNING
				     (pAd->WdsTab.WdsEntry[idx].wdev.if_dev))) {
					result = TRUE;
					break;
				}
			}
#endif /* WDS_SUPPORT */
		} else {
			if (pMbss->bBcnSntReq == TRUE)
				result = TRUE;
		}
	} while (FALSE);

	return result;
}

/*
	==========================================================================
	Description:
		Pre-build a BEACON frame in the shared memory
	==========================================================================
*/
VOID APMakeBssBeacon(RTMP_ADAPTER *pAd, INT apidx)
{
	UCHAR DsLen = 1;
	HEADER_802_11 BcnHdr;
	LARGE_INTEGER FakeTimestamp;
	ULONG FrameLen = 0;
	PUCHAR pBeaconFrame = (PUCHAR) pAd->ApCfg.MBSSID[apidx].BeaconBuf;
	/* MGMT frame PHY rate setting when operatin at Ht rate. */
	HTTRANSMIT_SETTING BeaconTransmit = {.word = 0 };
	UCHAR PhyMode, SupRateLen;
	MULTISSID_STRUCT *pMbss = &pAd->ApCfg.MBSSID[apidx];
#ifdef SPECIFIC_TX_POWER_SUPPORT
	UCHAR TxPwrAdj = 0;
#endif /* SPECIFIC_TX_POWER_SUPPORT */

	if (!BeaconTransmitRequired(pAd, apidx, pMbss))
		return;

	PhyMode = pMbss->wdev.PhyMode;

	MgtMacHeaderInit(pAd, &BcnHdr, SUBTYPE_BEACON, 0, BROADCAST_ADDR,
			 pMbss->wdev.if_addr, pMbss->wdev.bssid);

	/* for update framelen to TxWI later. */
	SupRateLen = pAd->CommonCfg.SupRateLen;
	if (PhyMode == WMODE_B)
		SupRateLen = 4;

		MakeOutgoingFrame(pBeaconFrame, &FrameLen,
				  sizeof(HEADER_802_11), &BcnHdr,
				  TIMESTAMP_LEN, &FakeTimestamp,
				  2, &pAd->CommonCfg.BeaconPeriod,
				  2, &pMbss->CapabilityInfo,
				  1, &SsidIe,
				  1, &pMbss->SsidLen,
				  pMbss->SsidLen, pMbss->Ssid,
				  1, &SupRateIe,
				  1, &SupRateLen,
				  SupRateLen, pAd->CommonCfg.SupRate,
				  1, &DsIe, 1, &DsLen, 1, &pAd->CommonCfg.Channel, END_OF_ARGS);

	if ((pAd->CommonCfg.ExtRateLen) && (PhyMode != WMODE_B)) {
		ULONG TmpLen;

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &ExtRateIe,
				  1, &pAd->CommonCfg.ExtRateLen,
				  pAd->CommonCfg.ExtRateLen, pAd->CommonCfg.ExtRate, END_OF_ARGS);
		FrameLen += TmpLen;
	}

	/* add country IE, power constraint IE */
	if (pAd->CommonCfg.bCountryFlag) {
		ULONG TmpLen, TmpLen2 = 0;
		UCHAR *TmpFrame = NULL;
		UCHAR CountryIe = IE_COUNTRY;

		os_alloc_mem(NULL, (UCHAR **) &TmpFrame, 256);
		if (TmpFrame != NULL) {
			NdisZeroMemory(TmpFrame, 256);

			/* prepare channel information */
#ifdef EXT_BUILD_CHANNEL_LIST
			BuildBeaconChList(pAd, TmpFrame, &TmpLen2);
#else
			{
				UCHAR MaxTxPower = GetCuntryMaxTxPwr(pAd, pAd->CommonCfg.Channel);

				MakeOutgoingFrame(TmpFrame + TmpLen2, &TmpLen,
						  1, &pAd->ChannelList[0].Channel,
						  1, &pAd->ChannelListNum,
						  1, &MaxTxPower, END_OF_ARGS);
				TmpLen2 += TmpLen;
			}
#endif /* EXT_BUILD_CHANNEL_LIST */


			/* need to do the padding bit check, and concatenate it */
			if ((TmpLen2 % 2) == 0) {
				UCHAR TmpLen3 = TmpLen2 + 4;

				MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
						  1, &CountryIe,
						  1, &TmpLen3,
						  3, pAd->CommonCfg.CountryCode,
						  TmpLen2 + 1, TmpFrame, END_OF_ARGS);
			} else {
				UCHAR TmpLen3 = TmpLen2 + 3;

				MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
						  1, &CountryIe,
						  1, &TmpLen3,
						  3, pAd->CommonCfg.CountryCode,
						  TmpLen2, TmpFrame, END_OF_ARGS);
			}
			FrameLen += TmpLen;

			os_free_mem(NULL, TmpFrame);
		} else
			DBGPRINT(RT_DEBUG_ERROR, ("%s: Allocate memory fail!!!\n", __func__));
	}

	/* AP Channel Report */
	{
		UCHAR APChannelReportIe = IE_AP_CHANNEL_REPORT;
		ULONG TmpLen;

		/*
		   802.11n D2.0 Annex J, USA regulatory
		   class 32, channel set 1~7
		   class 33, channel set 5-11
		 */
		UCHAR rclass32[] = { 32, 1, 2, 3, 4, 5, 6, 7 };
		UCHAR rclass33[] = { 33, 5, 6, 7, 8, 9, 10, 11 };
		UCHAR rclasslen = 8;	/*sizeof(rclass32); */

		if (PhyMode == (WMODE_B | WMODE_G | WMODE_GN)) {
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &APChannelReportIe,
					  1, &rclasslen,
					  rclasslen, rclass32,
					  1, &APChannelReportIe,
					  1, &rclasslen, rclasslen, rclass33, END_OF_ARGS);
			FrameLen += TmpLen;
		}
	}


	BeaconTransmit.word = 0;

#ifdef SPECIFIC_TX_POWER_SUPPORT
	/* Specific Power for Long-Range Beacon */
	if ((pAd->ApCfg.MBSSID[apidx].TxPwrAdj != -1)	/* &&
							   (BeaconTransmit.field.MODE == MODE_CCK) */) {
		TxPwrAdj = pAd->ApCfg.MBSSID[apidx].TxPwrAdj;
	}
#endif /* SPECIFIC_TX_POWER_SUPPORT */

	RTMPWriteTxWI(pAd, &pAd->BeaconTxWI, FALSE, FALSE, TRUE, FALSE, FALSE, TRUE, 0,
		      BSS0Mcast_WCID, FrameLen, PID_MGMT, 0, 0, IFS_HTTXOP, &BeaconTransmit);

#ifdef SPECIFIC_TX_POWER_SUPPORT
#ifdef RTMP_MAC
	if ((IS_RT6352(pAd) || IS_MT76x2(pAd)) && (pAd->chipCap.hif_type == HIF_RTMP))
		pAd->BeaconTxWI.TXWI_O.TxPwrAdj = TxPwrAdj;
#endif /* RTMP_MAC */
#ifdef RLT_MAC
	if ((IS_RT6352(pAd) || IS_MT76x2(pAd)) && (pAd->chipCap.hif_type == HIF_RLT))
		pAd->BeaconTxWI.TXWI_N.TxPwrAdj = TxPwrAdj;
#endif /* RLT_MAC */
#endif /* SPECIFIC_TX_POWER_SUPPORT */

	/*
	   step 6. move BEACON TXD and frame content to on-chip memory
	 */
	updateAllBeacon(pAd, apidx, FrameLen);

	pMbss->TimIELocationInBeacon = (UCHAR) FrameLen;
	pMbss->CapabilityInfoLocationInBeacon = sizeof(HEADER_802_11) + TIMESTAMP_LEN + 2;
}

void updateAllBeacon(RTMP_ADAPTER *pAd, INT apidx, ULONG FrameLen)
{
	UCHAR *ptr = NULL;
	MULTISSID_STRUCT *pMbss = &pAd->ApCfg.MBSSID[apidx];
	UINT32 longValue, reg_base;
	UINT i = 0;
	UINT8 TXWISize = pAd->chipCap.TXWISize;

	ptr = (PUCHAR) &pAd->BeaconTxWI;
#ifdef RT_BIG_ENDIAN
	RTMPWIEndianChange(pAd, ptr, TYPE_TXWI);
#endif /* endif */

	reg_base = pAd->BeaconOffset[apidx];
	for (i = 0; i < TXWISize; i += 4) {
		longValue = *ptr + (*(ptr + 1) << 8) + (*(ptr + 2) << 16) + (*(ptr + 3) << 24);
		RTMP_CHIP_UPDATE_BEACON(pAd, reg_base + i, longValue, 4);
		ptr += 4;
	}

	/* update BEACON frame content. start right after the TXWI field. */
	ptr = (PUCHAR) pMbss->BeaconBuf;
#ifdef RT_BIG_ENDIAN
	RTMPFrameEndianChange(pAd, ptr, DIR_WRITE, FALSE);
#endif /* endif */

	reg_base = pAd->BeaconOffset[apidx] + TXWISize;
	for (i = 0; i < FrameLen; i += 4) {
		longValue = *ptr + (*(ptr + 1) << 8) + (*(ptr + 2) << 16) + (*(ptr + 3) << 24);
		RTMP_CHIP_UPDATE_BEACON(pAd, reg_base + i, longValue, 4);
		ptr += 4;
	}

#ifdef RTMP_MAC_USB
	if (IS_USB_INF(pAd)) {
		/* WCNCR00017102:pBeaconSync buf is only available for USB */
		BEACON_SYNC_STRUCT *pBeaconSync = pAd->CommonCfg.pBeaconSync;
		if (pBeaconSync == NULL) {
			DBGPRINT(RT_DEBUG_ERROR, ("BEACON_SYNC_STRUCT is NULL\n"));
			return;
			}
		else {
			ptr = pMbss->BeaconBuf + pMbss->TimIELocationInBeacon;
			if ((*(ptr + 4)) & 0x01)
				pBeaconSync->DtimBitOn |= (1 << apidx);
			else
				pBeaconSync->DtimBitOn &= ~(1 << apidx);
			}
	}
#endif /* RTMP_MAC_USB */
}

/*
	==========================================================================
	Description:
		Update the BEACON frame in the shared memory. Because TIM IE is variable
		length. other IEs after TIM has to shift and total frame length may change
		for each BEACON period.
	Output:
		pAd->ApCfg.MBSSID[apidx].CapabilityInfo
		pAd->ApCfg.ErpIeContent
	==========================================================================
*/
VOID APUpdateBeaconFrame(RTMP_ADAPTER *pAd, INT apidx)
{
	UCHAR *pBeaconFrame;
	UCHAR *ptr;
	ULONG FrameLen;
	ULONG UpdatePos;
	UCHAR RSNIe = IE_WPA, RSNIe2 = IE_WPA2;
	UCHAR ID_1B, TimFirst, TimLast, *pTim;
	MULTISSID_STRUCT *pMbss;
	COMMON_CONFIG *pComCfg;
	UCHAR PhyMode;
#if (defined HOSTAPD_SUPPORT) || (defined WSC_AP_SUPPORT)
	BOOLEAN bHasWpsIE = FALSE;
#endif
	UINT i;
	/* MGMT frame PHY rate setting when operatin at Ht rate. */
	HTTRANSMIT_SETTING BeaconTransmit = {.word = 0 };
	struct wifi_dev *wdev;
#ifdef SPECIFIC_TX_POWER_SUPPORT
	UCHAR TxPwrAdj = 0;
#endif /* SPECIFIC_TX_POWER_SUPPORT */

	pComCfg = &pAd->CommonCfg;
	pMbss = &pAd->ApCfg.MBSSID[apidx];
	wdev = &pMbss->wdev;

	pBeaconFrame = (UCHAR *) &pMbss->BeaconBuf[0];
	FrameLen = UpdatePos = pMbss->TimIELocationInBeacon;
	PhyMode = wdev->PhyMode;

	if (!BeaconTransmitRequired(pAd, apidx, pMbss))
		return;

	/*
	   step 1 - update BEACON's Capability
	 */
	ptr = pBeaconFrame + pMbss->CapabilityInfoLocationInBeacon;
	*ptr = (UCHAR) (pMbss->CapabilityInfo & 0x00ff);
	*(ptr + 1) = (UCHAR) ((pMbss->CapabilityInfo & 0xff00) >> 8);

	/*
	   step 2 - update TIM IE
	   TODO: enlarge TIM bitmap to support up to 64 STAs
	   TODO: re-measure if RT2600 TBTT interrupt happens faster than BEACON sent out time
	 */
	ptr = pBeaconFrame + pMbss->TimIELocationInBeacon;
	*ptr = IE_TIM;
	*(ptr + 2) = pAd->ApCfg.DtimCount[apidx];
	*(ptr + 3) = pAd->ApCfg.DtimPeriod[apidx];

	/* find the smallest AID (PS mode) */
	TimFirst = 0;		/* record first TIM byte != 0x00 */
	TimLast = 0;		/* record last  TIM byte != 0x00 */
	pTim = pMbss->TimBitmaps;

	for (ID_1B = 0; ID_1B < WLAN_MAX_NUM_OF_TIM; ID_1B++) {
		/* get the TIM indicating PS packets for 8 stations */
		UCHAR tim_1B = pTim[ID_1B];

		if (ID_1B == 0)
			tim_1B &= 0xfe;	/* skip bit0 bc/mc */

		if (tim_1B == 0)
			continue;	/* find next 1B */

		if (TimFirst == 0)
			TimFirst = ID_1B;

		TimLast = ID_1B;
	}

	/* fill TIM content to beacon buffer */
	if (TimFirst & 0x01)
		TimFirst--;	/* find the even offset byte */

	*(ptr + 1) = 3 + (TimLast - TimFirst + 1);	/* TIM IE length */
	*(ptr + 4) = TimFirst;

	for (i = TimFirst; i <= TimLast; i++)
		*(ptr + 5 + i - TimFirst) = pTim[i];

	/* bit0 means backlogged mcast/bcast */
	if (pAd->ApCfg.DtimCount[apidx] == 0)
		*(ptr + 4) |= (pMbss->TimBitmaps[WLAN_CT_TIM_BCMC_OFFSET] & 0x01);

	/* adjust BEACON length according to the new TIM */
	FrameLen += (2 + *(ptr + 1));

	/* move RSN IE from below to here for Ralink Win7 v3.0.0.61 version parse beacon issue. */
	/* sync the order with BRCM's AP. */
	if ((wdev->AuthMode == Ndis802_11AuthModeWPA) ||
	    (wdev->AuthMode == Ndis802_11AuthModeWPAPSK))
		RSNIe = IE_WPA;
	else if ((wdev->AuthMode == Ndis802_11AuthModeWPA2) ||
		 (wdev->AuthMode == Ndis802_11AuthModeWPA2PSK))
		RSNIe = IE_WPA2;
#ifdef WAPI_SUPPORT
	else if ((wdev->AuthMode == Ndis802_11AuthModeWAICERT) ||
		 (wdev->AuthMode == Ndis802_11AuthModeWAIPSK))
		RSNIe = IE_WAPI;
#endif /* WAPI_SUPPORT */

	/* Append RSN_IE when  WPA OR WPAPSK, */
	if ((wdev->AuthMode == Ndis802_11AuthModeWPA1WPA2) ||
	    (wdev->AuthMode == Ndis802_11AuthModeWPA1PSKWPA2PSK)) {
		ULONG TmpLen;

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &RSNIe,
				  1, &pMbss->RSNIE_Len[0],
				  pMbss->RSNIE_Len[0], pMbss->RSN_IE[0],
				  1, &RSNIe2,
				  1, &pMbss->RSNIE_Len[1],
				  pMbss->RSNIE_Len[1], pMbss->RSN_IE[1], END_OF_ARGS);
		FrameLen += TmpLen;
	} else if (wdev->AuthMode >= Ndis802_11AuthModeWPA) {
		ULONG TmpLen;

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &RSNIe,
				  1, &pMbss->RSNIE_Len[0],
				  pMbss->RSNIE_Len[0], pMbss->RSN_IE[0], END_OF_ARGS);
		FrameLen += TmpLen;
	}
#ifdef HOSTAPD_SUPPORT
	if (pMbss->HostapdWPS && (pMbss->WscIEBeacon.ValueLen))
		bHasWpsIE = TRUE;
#endif /* endif */

#ifdef WSC_AP_SUPPORT
	/* add Simple Config Information Element */
	if (((pMbss->WscControl.WscConfMode >= 1) && (pMbss->WscIEBeacon.ValueLen)))
		bHasWpsIE = TRUE;
#endif /* WSC_AP_SUPPORT */

#if (defined HOSTAPD_SUPPORT) || (defined WSC_AP_SUPPORT)
	if (bHasWpsIE) {
		ULONG WscTmpLen = 0;

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &WscTmpLen,
				  pMbss->WscIEBeacon.ValueLen, pMbss->WscIEBeacon.Value,
				  END_OF_ARGS);
		FrameLen += WscTmpLen;
	}
#endif

#ifdef WSC_AP_SUPPORT
	if ((pMbss->WscControl.WscConfMode != WSC_DISABLE) &&
#ifdef DOT1X_SUPPORT
	    (pMbss->wdev.IEEE8021X == FALSE) &&
#endif /* DOT1X_SUPPORT */
	    (pMbss->wdev.WepStatus == Ndis802_11WEPEnabled)) {
		/*
		   Non-WPS Windows XP and Vista PCs are unable to determine if a WEP enalbed network
		   is static key based or 802.1X based. If the legacy station gets an EAP-Request/Identity
		   from the AP, it assume the WEP network is 802.1X enabled & will prompt the user for
		   802.1X credentials. If the legacy station doesn't receive anything after sending an
		   EAPOL-Start, it will assume the WEP network is static key based and prompt user
		   for the WEP key. <<from "WPS and Static Key WEP Networks">> A WPS enabled AP
		   should include this IE in the beacon when the AP is hosting a static WEP key network.
		   The IE would be 7 bytes long with the Extended Capability field set to 0 (all bits zero)
		   http:msdn.microsoft.com/library/default.asp?url=/library/en-us/randz/protocol/
		   securing_public_wi-fi_hotspots.asp
		 */
		ULONG TempLen = 0;
		UCHAR PROVISION_SERVICE_IE[7] = { 0xDD, 0x05, 0x00, 0x50, 0xF2, 0x05, 0x00 };

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TempLen,
				  7, PROVISION_SERVICE_IE, END_OF_ARGS);
		FrameLen += TempLen;
	}
#endif /* WSC_AP_SUPPORT */


	/* Update ERP */
	if ((pComCfg->ExtRateLen) && (PhyMode != WMODE_B)) {
		/* fill ERP IE */
		ptr = (UCHAR *) pBeaconFrame + FrameLen;	/* pTxD->DataByteCnt; */
		*ptr = IE_ERP;
		*(ptr + 1) = 1;
		*(ptr + 2) = pAd->ApCfg.ErpIeContent;
		FrameLen += 3;
	}
#ifdef A_BAND_SUPPORT
	/* fill up Channel Switch Announcement Element */
	if ((pComCfg->Channel > 14)
	    && (pComCfg->bIEEE80211H == 1)
	    && (pAd->Dot11_H.RDMode == RD_SWITCHING_MODE)) {
		ptr = pBeaconFrame + FrameLen;
		*ptr = IE_CHANNEL_SWITCH_ANNOUNCEMENT;
		*(ptr + 1) = 3;
		*(ptr + 2) = 1;
		*(ptr + 3) = pComCfg->Channel;
		*(ptr + 4) = (pAd->Dot11_H.CSPeriod - pAd->Dot11_H.CSCount - 1);
		ptr += 5;
		FrameLen += 5;

		/* Extended Channel Switch Announcement Element */
		if (pComCfg->bExtChannelSwitchAnnouncement) {
			HT_EXT_CHANNEL_SWITCH_ANNOUNCEMENT_IE HtExtChannelSwitchIe;

			build_ext_channel_switch_ie(pAd, &HtExtChannelSwitchIe);
			NdisMoveMemory(ptr, &HtExtChannelSwitchIe,
				       sizeof(HT_EXT_CHANNEL_SWITCH_ANNOUNCEMENT_IE));
			ptr += sizeof(HT_EXT_CHANNEL_SWITCH_ANNOUNCEMENT_IE);
			FrameLen += sizeof(HT_EXT_CHANNEL_SWITCH_ANNOUNCEMENT_IE);
		}
#ifdef DOT11_VHT_AC
		if (WMODE_CAP_AC(PhyMode)) {
			INT tp_len, wb_len = 0;
			UCHAR *ch_sw_wrapper;
			VHT_TXPWR_ENV_IE txpwr_env;

			*ptr = IE_CH_SWITCH_WRAPPER;
			ch_sw_wrapper = (UCHAR *) (ptr + 1);	/* reserve for length */
			ptr += 2;	/* skip len */

			if (pComCfg->RegTransmitSetting.field.BW == BW_40) {
				WIDE_BW_CH_SWITCH_ELEMENT wb_info;

				*ptr = IE_WIDE_BW_CH_SWITCH;
				*(ptr + 1) = sizeof(WIDE_BW_CH_SWITCH_ELEMENT);
				ptr += 2;
				NdisZeroMemory(&wb_info, sizeof(WIDE_BW_CH_SWITCH_ELEMENT));
				if (pComCfg->vht_bw == VHT_BW_2040)
					wb_info.new_ch_width = 0;
				else
					wb_info.new_ch_width = 1;

				if (pComCfg->vht_bw == VHT_BW_80) {
					wb_info.center_freq_1 =
					    vht_cent_ch_freq(pAd, pComCfg->Channel);
					wb_info.center_freq_2 = 0;
				}
				NdisMoveMemory(ptr, &wb_info, sizeof(WIDE_BW_CH_SWITCH_ELEMENT));
				wb_len = sizeof(WIDE_BW_CH_SWITCH_ELEMENT);
				ptr += wb_len;
				wb_len += 2;
			}

			*ptr = IE_VHT_TXPWR_ENV;
			NdisZeroMemory(&txpwr_env, sizeof(VHT_TXPWR_ENV_IE));
			tp_len = build_vht_txpwr_envelope(pAd, (UCHAR *) &txpwr_env);
			*(ptr + 1) = tp_len;
			ptr += 2;
			NdisMoveMemory(ptr, &txpwr_env, tp_len);
			ptr += tp_len;
			tp_len += 2;
			*ch_sw_wrapper = wb_len + tp_len;

			FrameLen += (2 + wb_len + tp_len);
		}
#endif /* DOT11_VHT_AC */

	}
#endif /* A_BAND_SUPPORT */

	/* step 5. Update HT. Since some fields might change in the same BSS. */
	if (WMODE_CAP_N(PhyMode) && (wdev->DesiredHtPhyInfo.bHtEnable)) {
		ULONG TmpLen;
		UCHAR HtLen, HtLen1;
		/*UCHAR i; */

#ifdef RT_BIG_ENDIAN
		HT_CAPABILITY_IE HtCapabilityTmp;
		ADD_HT_INFO_IE addHTInfoTmp;
/*		USHORT	b2lTmp, b2lTmp2; // no use */
#endif /* endif */

		/* add HT Capability IE */
		HtLen = sizeof(pComCfg->HtCapability);
		HtLen1 = sizeof(pComCfg->AddHTInfo);
#ifndef RT_BIG_ENDIAN
		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &HtCapIe,
				  1, &HtLen,
				  HtLen, &pComCfg->HtCapability,
				  1, &AddHtInfoIe,
				  1, &HtLen1, HtLen1, &pComCfg->AddHTInfo, END_OF_ARGS);
#else
		NdisMoveMemory(&HtCapabilityTmp, &pComCfg->HtCapability, HtLen);
		*(USHORT *) (&HtCapabilityTmp.HtCapInfo) =
		    SWAP16(*(USHORT *) (&HtCapabilityTmp.HtCapInfo));
#ifdef UNALIGNMENT_SUPPORT
		{
			EXT_HT_CAP_INFO extHtCapInfo;

			NdisMoveMemory((PUCHAR) (&extHtCapInfo),
				       (PUCHAR) (&HtCapabilityTmp.ExtHtCapInfo),
				       sizeof(EXT_HT_CAP_INFO));
			*(USHORT *) (&extHtCapInfo) = cpu2le16(*(USHORT *) (&extHtCapInfo));
			NdisMoveMemory((PUCHAR) (&HtCapabilityTmp.ExtHtCapInfo),
				       (PUCHAR) (&extHtCapInfo), sizeof(EXT_HT_CAP_INFO));
		}
#else
		*(USHORT *) (&HtCapabilityTmp.ExtHtCapInfo) =
		    SWAP16(*(USHORT *) (&HtCapabilityTmp.ExtHtCapInfo));
#endif /* UNALIGNMENT_SUPPORT */

		NdisMoveMemory(&addHTInfoTmp, &pComCfg->AddHTInfo, HtLen1);
		*(USHORT *) (&addHTInfoTmp.AddHtInfo2) =
		    SWAP16(*(USHORT *) (&addHTInfoTmp.AddHtInfo2));
		*(USHORT *) (&addHTInfoTmp.AddHtInfo3) =
		    SWAP16(*(USHORT *) (&addHTInfoTmp.AddHtInfo3));

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &HtCapIe,
				  1, &HtLen,
				  HtLen, &HtCapabilityTmp,
				  1, &AddHtInfoIe, 1, &HtLen1, HtLen1, &addHTInfoTmp, END_OF_ARGS);
#endif /* endif */
		FrameLen += TmpLen;

		/*
		   P802.11n_D3.03, 7.3.2.60 Overlapping BSS Scan Parameters IE
		 */
		if ((pComCfg->Channel <= 14) && (pComCfg->HtCapability.HtCapInfo.ChannelWidth == 1)) {
			OVERLAP_BSS_SCAN_IE OverlapScanParam;
			ULONG TmpLen;
			UCHAR OverlapScanIE, ScanIELen;

			OverlapScanIE = IE_OVERLAPBSS_SCAN_PARM;
			ScanIELen = 14;
			OverlapScanParam.ScanPassiveDwell =
			    cpu2le16(pComCfg->Dot11OBssScanPassiveDwell);
			OverlapScanParam.ScanActiveDwell =
			    cpu2le16(pComCfg->Dot11OBssScanActiveDwell);
			OverlapScanParam.TriggerScanInt =
			    cpu2le16(pComCfg->Dot11BssWidthTriggerScanInt);
			OverlapScanParam.PassiveTalPerChannel =
			    cpu2le16(pComCfg->Dot11OBssScanPassiveTotalPerChannel);
			OverlapScanParam.ActiveTalPerChannel =
			    cpu2le16(pComCfg->Dot11OBssScanActiveTotalPerChannel);
			OverlapScanParam.DelayFactor =
			    cpu2le16(pComCfg->Dot11BssWidthChanTranDelayFactor);
			OverlapScanParam.ScanActThre = cpu2le16(pComCfg->Dot11OBssScanActivityThre);

			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &OverlapScanIE,
					  1, &ScanIELen, ScanIELen, &OverlapScanParam, END_OF_ARGS);

			FrameLen += TmpLen;
		}
#ifdef CONFIG_HOTSPOT
		if (pMbss->HotSpotCtrl.HotSpotEnable) {
			ULONG TmpLen;

			/* Indication element */
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  pMbss->HotSpotCtrl.HSIndicationIELen,
					  pMbss->HotSpotCtrl.HSIndicationIE, END_OF_ARGS);

			FrameLen += TmpLen;

			/* Interworking element */
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  pMbss->HotSpotCtrl.InterWorkingIELen,
					  pMbss->HotSpotCtrl.InterWorkingIE, END_OF_ARGS);

			FrameLen += TmpLen;

			/* Advertisement Protocol element */
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  pMbss->HotSpotCtrl.AdvertisementProtoIELen,
					  pMbss->HotSpotCtrl.AdvertisementProtoIE, END_OF_ARGS);

			FrameLen += TmpLen;

			/* Roaming Consortium element */
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  pMbss->HotSpotCtrl.RoamingConsortiumIELen,
					  pMbss->HotSpotCtrl.RoamingConsortiumIE, END_OF_ARGS);

			FrameLen += TmpLen;

			/* P2P element */
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  pMbss->HotSpotCtrl.P2PIELen,
					  pMbss->HotSpotCtrl.P2PIE, END_OF_ARGS);

			FrameLen += TmpLen;

		}
#endif /* endif */

#ifdef DOT11_VHT_AC
		if (WMODE_CAP_AC(PhyMode) && (pComCfg->Channel > 14)) {
			int _len =
			    build_vht_ies(pAd, (UCHAR *) (pBeaconFrame + FrameLen), SUBTYPE_BEACON);
			FrameLen += _len;
		}
#endif /* DOT11_VHT_AC */
	}

	/* 7.3.2.27 Extended Capabilities IE */
	{
		ULONG TmpLen, infoPos;
		PUCHAR pInfo;
		UCHAR extInfoLen;
		BOOLEAN bNeedAppendExtIE = FALSE;
		EXT_CAP_INFO_ELEMENT extCapInfo;

		extInfoLen = sizeof(EXT_CAP_INFO_ELEMENT);
		NdisZeroMemory(&extCapInfo, extInfoLen);

		/* P802.11n_D1.10, HT Information Exchange Support */
		if (WMODE_CAP_N(PhyMode) && (pComCfg->Channel <= 14) &&
		    (pMbss->wdev.DesiredHtPhyInfo.bHtEnable) && (pComCfg->bBssCoexEnable == TRUE)
		    ) {
			extCapInfo.BssCoexistMgmtSupport = 1;
		}
#ifdef CONFIG_DOT11V_WNM
		if (pMbss->WNMCtrl.ProxyARPEnable)
			extCapInfo.proxy_arp = 1;
#endif /* endif */

#ifdef CONFIG_HOTSPOT
		if (pMbss->HotSpotCtrl.HotSpotEnable)
			extCapInfo.interworking = 1;
#endif /* endif */

#ifdef DOT11_VHT_AC
		if (WMODE_CAP_AC(PhyMode) && (pAd->CommonCfg.Channel > 14))
			extCapInfo.operating_mode_notification = 1;
#endif /* DOT11_VHT_AC */

		pInfo = (PUCHAR) (&extCapInfo);
		for (infoPos = 0; infoPos < extInfoLen; infoPos++) {
			if (pInfo[infoPos] != 0) {
				bNeedAppendExtIE = TRUE;
				break;
			}
		}

		if (bNeedAppendExtIE == TRUE) {
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &ExtCapIe,
					  1, &extInfoLen, extInfoLen, &extCapInfo, END_OF_ARGS);
			FrameLen += TmpLen;
		}
	}

#ifdef WFA_VHT_PF
	if (pAd->force_vht_op_mode == TRUE) {
		ULONG TmpLen;
		UCHAR operating_ie = IE_OPERATING_MODE_NOTIFY, operating_len = 1;
		OPERATING_MODE operating_mode;

		operating_mode.rx_nss_type = 0;
		operating_mode.rx_nss = (pAd->vht_pf_op_ss - 1);
		operating_mode.ch_width = pAd->vht_pf_op_bw;

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &operating_ie,
				  1, &operating_len, 1, &operating_mode, END_OF_ARGS);
		FrameLen += TmpLen;
	}
#endif /* WFA_VHT_PF */

	/* add WMM IE here */
	if (pMbss->wdev.bWmmCapable) {
		ULONG TmpLen;
		UCHAR i;
		UCHAR WmeParmIe[26] = {
		     IE_VENDOR_SPECIFIC, 24, 0x00, 0x50, 0xf2, 0x02, 0x01, 0x01, 0, 0 };
		UINT8 AIFSN[4];

		WmeParmIe[8] = pAd->ApCfg.BssEdcaParm.EdcaUpdateCount & 0x0f;

#ifdef UAPSD_SUPPORT
		UAPSD_MR_IE_FILL(WmeParmIe[8], &pMbss->UapsdInfo);
#endif /* UAPSD_SUPPORT */

		NdisMoveMemory(AIFSN, pAd->ApCfg.BssEdcaParm.Aifsn, sizeof(AIFSN));


		for (i = QID_AC_BE; i <= QID_AC_VO; i++) {
			WmeParmIe[10 + (i * 4)] = (i << 5) +	/* b5-6 is ACI */
			    ((UCHAR) pAd->ApCfg.BssEdcaParm.bACM[i] << 4) +	/* b4 is ACM */
			    (AIFSN[i] & 0x0f);	/* b0-3 is AIFSN */
			WmeParmIe[11 + (i * 4)] = (pAd->ApCfg.BssEdcaParm.Cwmax[i] << 4) +	/* b5-8 is CWMAX */
			    (pAd->ApCfg.BssEdcaParm.Cwmin[i] & 0x0f);	/* b0-3 is CWMIN */
			/* low byte of TXOP */
			WmeParmIe[12 + (i * 4)] = (UCHAR) (pAd->ApCfg.BssEdcaParm.Txop[i] & 0xff);
			/* high byte of TXOP */
			WmeParmIe[13 + (i * 4)] = (UCHAR) (pAd->ApCfg.BssEdcaParm.Txop[i] >> 8);
		}

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen, 26, WmeParmIe, END_OF_ARGS);
		FrameLen += TmpLen;
	}
#ifdef AP_QLOAD_SUPPORT
	if (pAd->phy_ctrl.FlgQloadEnable != 0)
		FrameLen += QBSS_LoadElementAppend(pAd, pBeaconFrame + FrameLen);
#endif /* AP_QLOAD_SUPPORT */

#ifdef A_BAND_SUPPORT
	/*
	   Only 802.11a APs that comply with 802.11h are required to include a
	   Power Constrint Element(IE=32) in beacons and probe response frames
	 */
	if (((pComCfg->Channel > 14) && pComCfg->bIEEE80211H == TRUE)
	    ) {
		ULONG TmpLen;
		UINT8 PwrConstraintIE = IE_POWER_CONSTRAINT;
		UINT8 PwrConstraintLen = 1;
		UINT8 PwrConstraint = pComCfg->PwrConstraint;

		/* prepare power constraint IE */
		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  1, &PwrConstraintIE,
				  1, &PwrConstraintLen, 1, &PwrConstraint, END_OF_ARGS);
		FrameLen += TmpLen;

#ifdef DOT11_VHT_AC
		if (WMODE_CAP_AC(PhyMode)) {
			ULONG TmpLen;
			UINT8 vht_txpwr_env_ie = IE_VHT_TXPWR_ENV;
			UINT8 ie_len;
			VHT_TXPWR_ENV_IE txpwr_env;

			ie_len = build_vht_txpwr_envelope(pAd, (UCHAR *) &txpwr_env);
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &vht_txpwr_env_ie,
					  1, &ie_len, ie_len, &txpwr_env, END_OF_ARGS);
			FrameLen += TmpLen;
		}
#endif /* DOT11_VHT_AC */

	}
#endif /* A_BAND_SUPPORT */


	if (WMODE_CAP_N(PhyMode) && (wdev->DesiredHtPhyInfo.bHtEnable)) {
		ULONG TmpLen;
		UCHAR HtLen, HtLen1;
#ifdef RT_BIG_ENDIAN
		HT_CAPABILITY_IE HtCapabilityTmp;
		ADD_HT_INFO_IE addHTInfoTmp;
#endif /* endif */
		/* add HT Capability IE */
		HtLen = sizeof(pComCfg->HtCapability);
		HtLen1 = sizeof(pComCfg->AddHTInfo);

		if (pAd->bBroadComHT == TRUE) {
			UCHAR epigram_ie_len;
			UCHAR BROADCOM_HTC[4] = { 0x0, 0x90, 0x4c, 0x33 };
			UCHAR BROADCOM_AHTINFO[4] = { 0x0, 0x90, 0x4c, 0x34 };

			epigram_ie_len = HtLen + 4;
#ifndef RT_BIG_ENDIAN
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &WpaIe,
					  1, &epigram_ie_len,
					  4, &BROADCOM_HTC[0],
					  HtLen, &pComCfg->HtCapability, END_OF_ARGS);
#else
			NdisMoveMemory(&HtCapabilityTmp, &pComCfg->HtCapability, HtLen);
			*(USHORT *) (&HtCapabilityTmp.HtCapInfo) =
			    SWAP16(*(USHORT *) (&HtCapabilityTmp.HtCapInfo));
#ifdef UNALIGNMENT_SUPPORT
			{
				EXT_HT_CAP_INFO extHtCapInfo;

				NdisMoveMemory((PUCHAR) (&extHtCapInfo),
					       (PUCHAR) (&HtCapabilityTmp.ExtHtCapInfo),
					       sizeof(EXT_HT_CAP_INFO));
				*(USHORT *) (&extHtCapInfo) = cpu2le16(*(USHORT *) (&extHtCapInfo));
				NdisMoveMemory((PUCHAR) (&HtCapabilityTmp.ExtHtCapInfo),
					       (PUCHAR) (&extHtCapInfo), sizeof(EXT_HT_CAP_INFO));
			}
#else
			*(USHORT *) (&HtCapabilityTmp.ExtHtCapInfo) =
			    SWAP16(*(USHORT *) (&HtCapabilityTmp.ExtHtCapInfo));
#endif /* UNALIGNMENT_SUPPORT */

			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &WpaIe,
					  1, &epigram_ie_len,
					  4, &BROADCOM_HTC[0],
					  HtLen, &HtCapabilityTmp, END_OF_ARGS);
#endif /* endif */

			FrameLen += TmpLen;

			epigram_ie_len = HtLen1 + 4;
#ifndef RT_BIG_ENDIAN
			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &WpaIe,
					  1, &epigram_ie_len,
					  4, &BROADCOM_AHTINFO[0],
					  HtLen1, &pComCfg->AddHTInfo, END_OF_ARGS);
#else
			NdisMoveMemory(&addHTInfoTmp, &pComCfg->AddHTInfo, HtLen1);
			*(USHORT *) (&addHTInfoTmp.AddHtInfo2) =
			    SWAP16(*(USHORT *) (&addHTInfoTmp.AddHtInfo2));
			*(USHORT *) (&addHTInfoTmp.AddHtInfo3) =
			    SWAP16(*(USHORT *) (&addHTInfoTmp.AddHtInfo3));

			MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
					  1, &WpaIe,
					  1, &epigram_ie_len,
					  4, &BROADCOM_AHTINFO[0],
					  HtLen1, &addHTInfoTmp, END_OF_ARGS);
#endif /* endif */
			FrameLen += TmpLen;
		}
	}

	/* add Ralink-specific IE here - Byte0.b0=1 for aggregation, Byte0.b1=1 for piggy-back */
	{
		ULONG TmpLen;
		UCHAR RalinkSpecificIe[9] = {
		     IE_VENDOR_SPECIFIC, 7, 0x00, 0x0c, 0x43, 0x00, 0x00, 0x00, 0x00 };

		if (pComCfg->bAggregationCapable)
			RalinkSpecificIe[5] |= 0x1;
		if (pComCfg->bPiggyBackCapable)
			RalinkSpecificIe[5] |= 0x2;
		if (pComCfg->bRdg)
			RalinkSpecificIe[5] |= 0x4;

#ifdef DOT11_VHT_AC
		if (pComCfg->b256QAM_2G && WMODE_2G_ONLY(pComCfg->PhyMode))
			RalinkSpecificIe[5] |= 0x8;
#endif /* DOT11_VHT_AC */

		MakeOutgoingFrame(pBeaconFrame + FrameLen, &TmpLen,
				  9, RalinkSpecificIe, END_OF_ARGS);
		FrameLen += TmpLen;

	}


	/* step 6. Since FrameLen may change, update TXWI. */
#ifdef A_BAND_SUPPORT
	if (pAd->CommonCfg.Channel > 14) {
		BeaconTransmit.field.MODE = MODE_OFDM;
		BeaconTransmit.field.MCS = MCS_RATE_6;
	}
#endif /* A_BAND_SUPPORT */

#ifdef SPECIFIC_TX_POWER_SUPPORT
	/* Specific Power for Long-Range Beacon */
	if ((pAd->ApCfg.MBSSID[apidx].TxPwrAdj != -1)	/* &&
							   (BeaconTransmit.field.MODE == MODE_CCK) */) {
		TxPwrAdj = pAd->ApCfg.MBSSID[apidx].TxPwrAdj;
	}
#endif /* SPECIFIC_TX_POWER_SUPPORT */

	RTMPWriteTxWI(pAd, &pAd->BeaconTxWI, FALSE, FALSE, TRUE, FALSE, FALSE, TRUE, 0,
		      RESERVED_WCID, FrameLen, PID_MGMT, 0 /*QID_MGMT */ , 0, IFS_HTTXOP,
		      &BeaconTransmit);

#ifdef SPECIFIC_TX_POWER_SUPPORT
#ifdef RTMP_MAC
	if ((IS_RT6352(pAd) || IS_MT76x2(pAd)) && (pAd->chipCap.hif_type == HIF_RTMP))
		pAd->BeaconTxWI.TXWI_O.TxPwrAdj = TxPwrAdj;
#endif /* RTMP_MAC */
#ifdef RLT_MAC
	if ((IS_RT6352(pAd) || IS_MT76x2(pAd)) && (pAd->chipCap.hif_type == HIF_RLT))
		pAd->BeaconTxWI.TXWI_N.TxPwrAdj = TxPwrAdj;
#endif /* RLT_MAC */
#endif /* SPECIFIC_TX_POWER_SUPPORT */

	/* step 7. move BEACON TXWI and frame content to on-chip memory */
	RT28xx_UpdateBeaconToAsic(pAd, apidx, FrameLen, UpdatePos);

}

/*
    ==========================================================================
    Description:
	Pre-build All BEACON frame in the shared memory
    ==========================================================================
*/
static UCHAR GetBcnNum(RTMP_ADAPTER *pAd)
{
	int i;
	int NumBcn;

	NumBcn = 0;
	for (i = 0; i < pAd->ApCfg.BssidNum; i++) {
		if (pAd->ApCfg.MBSSID[i].bBcnSntReq) {
			pAd->ApCfg.MBSSID[i].BcnBufIdx = NumBcn;
			NumBcn++;
		}
	}


	return NumBcn;
}

VOID APMakeAllBssBeacon(RTMP_ADAPTER *pAd)
{
	INT i, j;
	UINT32 regValue;
	UCHAR NumOfMacs;
	UCHAR NumOfBcns;
	UINT8 TXWISize = pAd->chipCap.TXWISize;

	/* before MakeBssBeacon, clear all beacon TxD's valid bit */
	/* Note: can not use MAX_MBSSID_NUM here, or
	   1. when MBSS_SUPPORT is enabled;
	   2. MAX_MBSSID_NUM will be 8;
	   3. if HW_BEACON_OFFSET is 0x0200,
	   we will overwrite other shared memory SRAM of chip */
	/* use pAd->ApCfg.BssidNum to avoid the case is best */

	/* choose the Beacon number */
	NumOfBcns = GetBcnNum(pAd);

	for (i = 0; i < HW_BEACON_MAX_COUNT(pAd); i++) {
		for (j = 0; j < TXWISize; j += 4)
			RTMP_CHIP_UPDATE_BEACON(pAd, pAd->BeaconOffset[i] + j, 0, 4);
	}

#ifdef RTMP_MAC_USB
	RTUSBBssBeaconStop(pAd);
#endif /* RTMP_MAC_USB */

	for (i = 0; i < pAd->ApCfg.BssidNum; i++)
		APMakeBssBeacon(pAd, i);

	RTMP_IO_READ32(pAd, MAC_BSSID_DW1, &regValue);
	regValue &= 0x0000FFFF;

	/*
	   Note:
	   1.The MAC address of Mesh and AP-Client link are different from Main BSSID.
	   2.If the Mesh link is included, its MAC address shall follow the last MBSSID's MAC by increasing 1.
	   3.If the AP-Client link is included, its MAC address shall follow the Mesh interface MAC by increasing 1.
	 */
	NumOfMacs = pAd->ApCfg.BssidNum + MAX_MESH_NUM + MAX_APCLI_NUM;


	/* set Multiple BSSID mode */
	if (NumOfMacs <= 1) {
		pAd->ApCfg.MacMask = ~(1 - 1);
		/*regValue |= 0x0; */
	} else if (NumOfMacs <= 2) {
		if ((pAd->CurrentAddress[5] % 2 != 0)
		    )
			DBGPRINT(RT_DEBUG_ERROR,
				 ("The 2-BSSID mode is enabled, the BSSID byte5 MUST be the multiple of 2\n"));

		regValue |= (1 << 16);
		pAd->ApCfg.MacMask = ~(2 - 1);
	} else if (NumOfMacs <= 4) {
		if (pAd->CurrentAddress[5] % 4 != 0)
			DBGPRINT(RT_DEBUG_ERROR,
				 ("The 4-BSSID mode is enabled, the BSSID byte5 MUST be the multiple of 4\n"));

		regValue |= (2 << 16);
		pAd->ApCfg.MacMask = ~(4 - 1);
	} else if (NumOfMacs <= 8) {
		if (pAd->CurrentAddress[5] % 8 != 0)
			DBGPRINT(RT_DEBUG_ERROR,
				 ("The 8-BSSID mode is enabled, the BSSID byte5 MUST be the multiple of 8\n"));

		regValue |= (3 << 16);
		pAd->ApCfg.MacMask = ~(8 - 1);
	} else if (NumOfMacs <= 16) {
		/* Set MULTI_BSSID_MODE_BIT4 in MAC register 0x1014 */
		regValue |= (1 << 22);
		pAd->ApCfg.MacMask = ~(16 - 1);
	}

	/* set Multiple BSSID Beacon number */
	if (NumOfBcns > 1) {
		if (NumOfBcns > 8)
			regValue |= (((NumOfBcns - 1) >> 3) << 23);
		regValue |= (((NumOfBcns - 1) & 0x7) << 18);
	}

	/*      set as 0/1 bit-21 of MAC_BSSID_DW1(offset: 0x1014)
	   to disable/enable the new MAC address assignment.  */
	if (pAd->chipCap.MBSSIDMode >= MBSSID_MODE1) {
		regValue |= (1 << 21);
#ifdef ENHANCE_NEW_MBSSID_MODE
		if (pAd->chipCap.MBSSIDMode == MBSSID_MODE2)
			regValue |= (1 << 24);
		else if (pAd->chipCap.MBSSIDMode == MBSSID_MODE3)
			regValue |= (2 << 24);
		else if (pAd->chipCap.MBSSIDMode == MBSSID_MODE4)
			regValue |= (3 << 24);
		else if (pAd->chipCap.MBSSIDMode == MBSSID_MODE5)
			regValue |= (4 << 24);
		else if (pAd->chipCap.MBSSIDMode == MBSSID_MODE6)
			regValue |= (5 << 24);
#endif /* ENHANCE_NEW_MBSSID_MODE */
	}

	RTMP_IO_WRITE32(pAd, MAC_BSSID_DW1, regValue);

#ifdef HDR_TRANS_SUPPORT
	/*
	   point WCID MAC table to 0x1800
	   This is for debug.
	   But HDR_TRANS doesn't work if you remove it.
	   Check after IC formal release.
	 */
	regValue |= 0x18000000;
	RTMP_IO_WRITE32(pAd, HT_MAC_BSSID_DW1, regValue);
#endif /* HDR_TRANS_SUPPORT */

#ifdef RTMP_MAC_USB
#ifndef RT_CFG80211_SUPPORT
	RTUSBBssBeaconStart(pAd);
#endif /* RT_CFG80211_SUPPORT */
#endif /* RTMP_MAC_USB */

}

/*
    ==========================================================================
    Description:
	Pre-build All BEACON frame in the shared memory
    ==========================================================================
*/
VOID APUpdateAllBeaconFrame(RTMP_ADAPTER *pAd)
{
	INT i;
	BOOLEAN FlgQloadIsAlarmIssued = FALSE;

	if (pAd->ApCfg.DtimCount[MAIN_MBSSID] == 0)
		pAd->ApCfg.DtimCount[MAIN_MBSSID] =
			pAd->ApCfg.DtimPeriod[MAIN_MBSSID] - 1;
	else
		pAd->ApCfg.DtimCount[MAIN_MBSSID] -= 1;
	/* QLOAD ALARM */
#ifdef AP_QLOAD_SUPPORT
	FlgQloadIsAlarmIssued = QBSS_LoadIsAlarmIssued(pAd);
#endif /* AP_QLOAD_SUPPORT */

	if ((pAd->ApCfg.DtimCount[MAIN_MBSSID] == 0) &&
	    (((pAd->CommonCfg.Bss2040CoexistFlag & BSS_2040_COEXIST_INFO_SYNC) &&
	      (pAd->CommonCfg.bForty_Mhz_Intolerant == FALSE)) ||
	     (FlgQloadIsAlarmIssued == TRUE))) {
		UCHAR prevBW, prevExtChOffset;

		DBGPRINT(RT_DEBUG_TRACE,
			 ("DTIM Period reached, BSS20WidthReq=%d, Intolerant40=%d!\n",
			  pAd->CommonCfg.LastBSSCoexist2040.field.BSS20WidthReq,
			  pAd->CommonCfg.LastBSSCoexist2040.field.Intolerant40));
		pAd->CommonCfg.Bss2040CoexistFlag &= (~BSS_2040_COEXIST_INFO_SYNC);

		prevBW = pAd->CommonCfg.AddHTInfo.AddHtInfo.RecomWidth;
		prevExtChOffset = pAd->CommonCfg.AddHTInfo.AddHtInfo.ExtChanOffset;

		if (pAd->CommonCfg.LastBSSCoexist2040.field.BSS20WidthReq ||
		    pAd->CommonCfg.LastBSSCoexist2040.field.Intolerant40 ||
		    (pAd->MacTab.fAnyStaFortyIntolerant == TRUE) ||
		    (FlgQloadIsAlarmIssued == TRUE)) {
			pAd->CommonCfg.AddHTInfo.AddHtInfo.RecomWidth = 0;
			pAd->CommonCfg.AddHTInfo.AddHtInfo.ExtChanOffset = 0;
		} else {
			pAd->CommonCfg.AddHTInfo.AddHtInfo.RecomWidth =
			    pAd->CommonCfg.RegTransmitSetting.field.BW;
			pAd->CommonCfg.AddHTInfo.AddHtInfo.ExtChanOffset =
			    pAd->CommonCfg.RegTransmitSetting.field.EXTCHA;
		}
		DBGPRINT(RT_DEBUG_TRACE,
			 ("\tNow RecomWidth=%d, ExtChanOffset=%d, prevBW=%d, prevExtOffset=%d\n",
			  pAd->CommonCfg.AddHTInfo.AddHtInfo.RecomWidth,
			  pAd->CommonCfg.AddHTInfo.AddHtInfo.ExtChanOffset, prevBW,
			  prevExtChOffset));
		pAd->CommonCfg.Bss2040CoexistFlag |= BSS_2040_COEXIST_INFO_NOTIFY;
	}

	for (i = 0; i < pAd->ApCfg.BssidNum; i++)
		APUpdateBeaconFrame(pAd, i);
}
