/*
========================================================================
Description:
This routine frees all packets in PSQ that's destined to a specific DA.
BCAST/MCAST in DTIMCount=0 case is also handled here, just like a PS-POLL
is received from a WSTA which has MAC address FF:FF:FF:FF:FF:FF
========================================================================
*/
VOID RtmpHandleRxPsPoll(RTMP_ADAPTER *pAd, UCHAR *pAddr, USHORT wcid, BOOLEAN isActive)
{
QUEUE_ENTRY *pQEntry;
MAC_TABLE_ENTRY *pMacEntry;
unsigned long IrqFlags;
/*
DBGPRINT(RT_DEBUG_TRACE, ("rcv PS-POLL (AID=%d) from %02x:%02x:%02x:%02x:%02x:%02x\n",
Aid, PRINT_MAC(pAddr)));
*/
pMacEntry = &pAd->MacTab.Content[wcid];
if (RTMPEqualMemory(pMacEntry->Addr, pAddr, MAC_ADDR_LEN))
{
#ifdef DROP_MASK_SUPPORT
/* Disable Drop Mask */
set_drop_mask_per_client(pAd, pMacEntry, 2, 0);
#endif /* DROP_MASK_SUPPORT */
#ifdef PS_ENTRY_MAITENANCE
pMacEntry->continuous_ps_count = 0;
#endif /* PS_ENTRY_MAITENANCE */
/* Sta is change to Power Active stat. Reset ContinueTxFailCnt */
pMacEntry->ContinueTxFailCnt = 0;
#ifdef UAPSD_SUPPORT
if (UAPSD_MR_IS_ALL_AC_UAPSD(isActive, pMacEntry))
{
/*
IEEE802.11e spec.
11.2.1.7 Receive operation for STAs in PS mode during the CP
When a non-AP QSTA that is using U-APSD and has all ACs
delivery-enabled detects that the bit corresponding to its AID
is set in the TIM, the non-AP QSTA shall issue a trigger frame
or a PS-Poll frame to retrieve the buffered MSDU or management
frames.
WMM Spec. v1.1a 070601
3.6.2 U-APSD STA Operation
3.6.2.3 In case one or more ACs are not
delivery-enabled ACs, the WMM STA may retrieve MSDUs and
MMPDUs belonging to those ACs by sending PS-Polls to the WMM AP.
In case all ACs are delivery enabled ACs, WMM STA should only
use trigger frames to retrieve MSDUs and MMPDUs belonging to
those ACs, and it should not send PS-Poll frames.
Different definitions in IEEE802.11e and WMM spec.
But we follow the WiFi WMM Spec.
*/
DBGPRINT(RT_DEBUG_TRACE, ("All AC are UAPSD, can not use PS-Poll\n"));
return; /* all AC are U-APSD, can not use PS-Poll */
}
#endif /* UAPSD_SUPPORT */
RTMP_IRQ_LOCK(&pAd->irq_lock, IrqFlags);
if (isActive == FALSE)
{
if (pMacEntry->PsQueue.Head)
{
#ifdef UAPSD_SUPPORT
UINT32 NumOfOldPsPkt;
NumOfOldPsPkt = pAd->TxSwQueue[QID_AC_BE].Number;
#endif /* UAPSD_SUPPORT */
pQEntry = RemoveHeadQueue(&pMacEntry->PsQueue);
if ( pMacEntry->PsQueue.Number >=1 )
RTMP_SET_PACKET_MOREDATA(RTPKT_TO_OSPKT(pQEntry), TRUE);
InsertTailQueueAc(pAd, pMacEntry, &pAd->TxSwQueue[QID_AC_BE], pQEntry);
#ifdef UAPSD_SUPPORT
/* we need to call RTMPDeQueuePacket() immediately as below */
if (NumOfOldPsPkt != pAd->TxSwQueue[QID_AC_BE].Number)
{
if (RTMP_GET_PACKET_DHCP(RTPKT_TO_OSPKT(pQEntry)) ||
RTMP_GET_PACKET_EAPOL(RTPKT_TO_OSPKT(pQEntry)) ||
RTMP_GET_PACKET_WAI(RTPKT_TO_OSPKT(pQEntry)))
{
/*
These packets will use 1M/6M rate to send.
If you use 1M(2.4G)/6M(5G) to send, no statistics
count in NICUpdateFifoStaCounters().
So we can not count it for UAPSD; Or the SP will
not closed until timeout.
*/
}
else
UAPSD_MR_MIX_PS_POLL_RCV(pAd, pMacEntry);
}
#endif /* UAPSD_SUPPORT */
}
else
{
/*
or transmit a (QoS) Null Frame;
In addtion, in Station Keep Alive mechanism, we need to
send a QoS Null frame to detect the station live status.
*/
BOOLEAN bQosNull = FALSE;
if (CLIENT_STATUS_TEST_FLAG(pMacEntry, fCLIENT_STATUS_WMM_CAPABLE))
bQosNull = TRUE;
RtmpEnqueueNullFrame(pAd, pMacEntry->Addr, pMacEntry->CurrTxRate,
pMacEntry->Aid, pMacEntry->apidx,
bQosNull, TRUE, 0);
}
}
else
{
#ifdef UAPSD_SUPPORT
/* deliver all queued UAPSD packets */
UAPSD_AllPacketDeliver(pAd, pMacEntry);
/* end the SP if exists */
UAPSD_MR_ENTRY_RESET(pAd, pMacEntry);
#endif /* UAPSD_SUPPORT */
while(pMacEntry->PsQueue.Head)
{
pQEntry = RemoveHeadQueue(&pMacEntry->PsQueue);
InsertTailQueueAc(pAd, pMacEntry, &pAd->TxSwQueue[QID_AC_BE], pQEntry);
}
}
if ((pMacEntry->Aid > 0) && (pMacEntry->Aid < MAX_LEN_OF_MAC_TABLE) &&
(pMacEntry->PsQueue.Number == 0))
{
/* clear corresponding TIM bit because no any PS packet */
#ifdef CONFIG_AP_SUPPORT
if(pMacEntry->wdev->wdev_type == WDEV_TYPE_AP)
{
WLAN_MR_TIM_BIT_CLEAR(pAd, pMacEntry->apidx, pMacEntry->Aid);
}
#endif /* CONFIG_AP_SUPPORT */
pMacEntry->PsQIdleCount = 0;
}
RTMP_IRQ_UNLOCK(&pAd->irq_lock, IrqFlags);
/*
Dequeue outgoing frames from TxSwQueue0..3 queue and process it
TODO: 2004-12-27 it's not a good idea to handle "More Data" bit here.
because the RTMPDeQueue process doesn't guarantee to de-queue the
desired MSDU from the corresponding TxSwQueue/PsQueue when QOS
in-used. We should consider "HardTransmt" this MPDU using MGMT
queue or things like that.
*/
RTMPDeQueuePacket(pAd, FALSE, NUM_OF_TX_RING, MAX_TX_PROCESS);
}
else
{
DBGPRINT(RT_DEBUG_ERROR,("rcv PS-POLL (AID=%d not match) from %02x:%02x:%02x:%02x:%02x:%02x\n",
pMacEntry->Aid, PRINT_MAC(pAddr)));
}
}
/*
==========================================================================
Description:
Update the station current power save mode. Calling this routine also
prove the specified client is still alive. Otherwise AP will age-out
this client once IdleCount exceeds a threshold.
==========================================================================
*/
BOOLEAN RtmpPsIndicate(RTMP_ADAPTER *pAd, UCHAR *pAddr, UCHAR wcid, UCHAR Psm)
{
MAC_TABLE_ENTRY *pEntry;
UCHAR old_psmode;
if (wcid >= MAX_LEN_OF_MAC_TABLE)
return PWR_ACTIVE;
pEntry = &pAd->MacTab.Content[wcid];
old_psmode = pEntry->PsMode;
/*
Change power save mode first because we will call
RTMPDeQueuePacket() in RtmpHandleRxPsPoll().
Or when Psm = PWR_ACTIVE, we will not do Aggregation in
RTMPDeQueuePacket().
*/
pEntry->NoDataIdleCount = 0;
pEntry->PsMode = Psm;
if (old_psmode != Psm) {
DBGPRINT(RT_DEBUG_INFO, ("%s():%02x:%02x:%02x:%02x:%02x:%02x %s!\n",
__FUNCTION__, PRINT_MAC(pAddr),
(Psm == PWR_SAVE ? "Sleep" : "wakes up, act like rx PS-POLL")));
}
if ((old_psmode == PWR_SAVE) && (Psm == PWR_ACTIVE))
{
#ifdef DROP_MASK_SUPPORT
/* Disable Drop Mask */
set_drop_mask_per_client(pAd, pEntry, 2, 0);
#endif /* DROP_MASK_SUPPORT */
#ifdef PS_ENTRY_MAITENANCE
pEntry->continuous_ps_count = 0;
#endif /* PS_ENTRY_MAITENANCE */
#ifdef RTMP_MAC_PCI
#ifdef DOT11_N_SUPPORT
/*
When sta wake up, we send BAR to refresh the BA sequence.
TODO:
For RT2870, how to handle BA when STA in PS mode?
*/
SendRefreshBAR(pAd, pEntry);
#endif /* DOT11_N_SUPPORT */
#endif /* RTMP_MAC_PCI */
/* sleep station awakes, move all pending frames from PSQ to TXQ if any */
RtmpHandleRxPsPoll(pAd, pAddr, pEntry->wcid, TRUE);
}
#ifdef DROP_MASK_SUPPORT
else if ((old_psmode == PWR_ACTIVE) && (Psm == PWR_SAVE)) {
/* Enable Drop Mask */
set_drop_mask_per_client(pAd, pEntry, 2, 1);
}
#endif /* DROP_MASK_SUPPORT */
return old_psmode;
}
/*
==========================================================================
Description:
Timer execution function for periodically updating group key.
Return:
==========================================================================
*/
VOID GREKEYPeriodicExec(
IN PVOID SystemSpecific1,
IN PVOID FunctionContext,
IN PVOID SystemSpecific2,
IN PVOID SystemSpecific3)
{
UINT i, apidx;
ULONG temp_counter = 0;
struct rtmp_adapter *pAd = (struct rtmp_adapter *)FunctionContext;
PRALINK_TIMER_STRUCT pTimer = (PRALINK_TIMER_STRUCT) SystemSpecific3;
MULTISSID_STRUCT *pMbss = NULL;
struct rtmp_wifi_dev *wdev;
for (apidx=0; apidx<pAd->ApCfg.BssidNum; apidx++)
{
if (&pAd->ApCfg.MBSSID[apidx].REKEYTimer == pTimer)
break;
}
if (apidx == pAd->ApCfg.BssidNum)
return;
else
pMbss = &pAd->ApCfg.MBSSID[apidx];
wdev = &pMbss->wdev;
if (wdev->AuthMode < Ndis802_11AuthModeWPA ||
wdev->AuthMode > Ndis802_11AuthModeWPA1PSKWPA2PSK)
return;
if ((pMbss->WPAREKEY.ReKeyMethod == TIME_REKEY) && (pMbss->REKEYCOUNTER < 0xffffffff))
temp_counter = (++pMbss->REKEYCOUNTER);
/* REKEYCOUNTER is incremented every MCAST packets transmitted, */
/* But the unit of Rekeyinterval is 1K packets */
else if (pMbss->WPAREKEY.ReKeyMethod == PKT_REKEY)
temp_counter = pMbss->REKEYCOUNTER/1000;
else
{
return;
}
if (temp_counter > (pMbss->WPAREKEY.ReKeyInterval))
{
pMbss->REKEYCOUNTER = 0;
pMbss->RekeyCountDown = 3;
DBGPRINT(RT_DEBUG_TRACE, ("Rekey Interval Excess, GKeyDoneStations=%d\n", pMbss->StaCount));
/* take turn updating different groupkey index, */
if ((pMbss->StaCount) > 0)
{
/* change key index */
wdev->DefaultKeyId = (wdev->DefaultKeyId == 1) ? 2 : 1;
/* Generate GNonce randomly */
GenRandom(pAd, wdev->bssid, pMbss->GNonce);
/* Update GTK */
WpaDeriveGTK(pMbss->GMK,
(u8 *)pMbss->GNonce,
wdev->bssid, pMbss->GTK, LEN_TKIP_GTK);
/* Process 2-way handshaking */
for (i = 0; i < MAX_LEN_OF_MAC_TABLE; i++)
{
MAC_TABLE_ENTRY *pEntry;
pEntry = &pAd->MacTab.Content[i];
if (IS_ENTRY_CLIENT(pEntry) &&
(pEntry->WpaState == AS_PTKINITDONE) &&
(pEntry->apidx == apidx))
{
pEntry->GTKState = REKEY_NEGOTIATING;
#ifdef DROP_MASK_SUPPORT
/* Disable Drop Mask */
set_drop_mask_per_client(pAd, pEntry, 1, 0);
set_drop_mask_per_client(pAd, pEntry, 2, 0);
#endif /* DROP_MASK_SUPPORT */
WPAStart2WayGroupHS(pAd, pEntry);
DBGPRINT(RT_DEBUG_TRACE, ("Rekey interval excess, Update Group Key for %x %x %x %x %x %x , DefaultKeyId= %x \n",\
PRINT_MAC(pEntry->Addr), wdev->DefaultKeyId));
}
}
}
}
/* Use countdown to ensure the 2-way handshaking had completed */
if (pMbss->RekeyCountDown > 0)
{
pMbss->RekeyCountDown--;
if (pMbss->RekeyCountDown == 0)
{
unsigned short Wcid;
/* Get a specific WCID to record this MBSS key attribute */
GET_GroupKey_WCID(pAd, Wcid, apidx);
/* Install shared key table */
WPAInstallSharedKey(pAd,
wdev->GroupKeyWepStatus,
apidx,
wdev->DefaultKeyId,
Wcid,
true,
pMbss->GTK,
LEN_TKIP_GTK);
}
}
}