git.oblomov.eu Git - linux-2.6/blob - drivers/staging/rt2870/common/cmm_sanity.c

   1 /*
   2  *************************************************************************
   3  * Ralink Tech Inc.
   4  * 5F., No.36, Taiyuan St., Jhubei City,
   5  * Hsinchu County 302,
   6  * Taiwan, R.O.C.
   7  *
   8  * (c) Copyright 2002-2007, Ralink Technology, Inc.
   9  *
  10  * This program is free software; you can redistribute it and/or modify  *
  11  * it under the terms of the GNU General Public License as published by  *
  12  * the Free Software Foundation; either version 2 of the License, or     *
  13  * (at your option) any later version.                                   *
  14  *                                                                       *
  15  * This program is distributed in the hope that it will be useful,       *
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  18  * GNU General Public License for more details.                          *
  19  *                                                                       *
  20  * You should have received a copy of the GNU General Public License     *
  21  * along with this program; if not, write to the                         *
  22  * Free Software Foundation, Inc.,                                       *
  23  * 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  24  *                                                                       *
  25  *************************************************************************
  26
  27         Module Name:
  28         sanity.c
  29
  30         Abstract:
  31
  32         Revision History:
  33         Who                     When                    What
  34         --------        ----------              ----------------------------------------------
  35         John Chang  2004-09-01      add WMM support
  36 */
  37 #include "../rt_config.h"
  38
  39
  40 extern UCHAR    CISCO_OUI[];
  41
  42 extern UCHAR    WPA_OUI[];
  43 extern UCHAR    RSN_OUI[];
  44 extern UCHAR    WME_INFO_ELEM[];
  45 extern UCHAR    WME_PARM_ELEM[];
  46 extern UCHAR    Ccx2QosInfo[];
  47 extern UCHAR    RALINK_OUI[];
  48 extern UCHAR    BROADCOM_OUI[];
  49 extern UCHAR    WPS_OUI[];
  50
  51 /*
  52     ==========================================================================
  53     Description:
  54         MLME message sanity check
  55     Return:
  56         TRUE if all parameters are OK, FALSE otherwise
  57
  58         IRQL = DISPATCH_LEVEL
  59
  60     ==========================================================================
  61  */
  62 BOOLEAN MlmeAddBAReqSanity(
  63     IN PRTMP_ADAPTER pAd,
  64     IN VOID *Msg,
  65     IN ULONG MsgLen,
  66     OUT PUCHAR pAddr2)
  67 {
  68     PMLME_ADDBA_REQ_STRUCT   pInfo;
  69
  70     pInfo = (MLME_ADDBA_REQ_STRUCT *)Msg;
  71
  72     if ((MsgLen != sizeof(MLME_ADDBA_REQ_STRUCT)))
  73     {
  74         DBGPRINT(RT_DEBUG_TRACE, ("MlmeAddBAReqSanity fail - message lenght not correct.\n"));
  75         return FALSE;
  76     }
  77
  78     if ((pInfo->Wcid >= MAX_LEN_OF_MAC_TABLE))
  79     {
  80         DBGPRINT(RT_DEBUG_TRACE, ("MlmeAddBAReqSanity fail - The peer Mac is not associated yet.\n"));
  81         return FALSE;
  82     }
  83
  84         /*
  85     if ((pInfo->BaBufSize > MAX_RX_REORDERBUF) || (pInfo->BaBufSize < 2))
  86     {
  87         DBGPRINT(RT_DEBUG_TRACE, ("MlmeAddBAReqSanity fail - Rx Reordering buffer too big or too small\n"));
  88         return FALSE;
  89     }
  90         */
  91
  92     if ((pInfo->pAddr[0]&0x01) == 0x01)
  93     {
  94         DBGPRINT(RT_DEBUG_TRACE, ("MlmeAddBAReqSanity fail - broadcast address not support BA\n"));
  95         return FALSE;
  96     }
  97
  98     return TRUE;
  99 }
 100
 101 /*
 102     ==========================================================================
 103     Description:
 104         MLME message sanity check
 105     Return:
 106         TRUE if all parameters are OK, FALSE otherwise
 107
 108         IRQL = DISPATCH_LEVEL
 109
 110     ==========================================================================
 111  */
 112 BOOLEAN MlmeDelBAReqSanity(
 113     IN PRTMP_ADAPTER pAd,
 114     IN VOID *Msg,
 115     IN ULONG MsgLen)
 116 {
 117         MLME_DELBA_REQ_STRUCT *pInfo;
 118         pInfo = (MLME_DELBA_REQ_STRUCT *)Msg;
 119
 120     if ((MsgLen != sizeof(MLME_DELBA_REQ_STRUCT)))
 121     {
 122         DBGPRINT(RT_DEBUG_ERROR, ("MlmeDelBAReqSanity fail - message lenght not correct.\n"));
 123         return FALSE;
 124     }
 125
 126     if ((pInfo->Wcid >= MAX_LEN_OF_MAC_TABLE))
 127     {
 128         DBGPRINT(RT_DEBUG_ERROR, ("MlmeDelBAReqSanity fail - The peer Mac is not associated yet.\n"));
 129         return FALSE;
 130     }
 131
 132     if ((pInfo->TID & 0xf0))
 133     {
 134         DBGPRINT(RT_DEBUG_ERROR, ("MlmeDelBAReqSanity fail - The peer TID is incorrect.\n"));
 135         return FALSE;
 136     }
 137
 138         if (NdisEqualMemory(pAd->MacTab.Content[pInfo->Wcid].Addr, pInfo->Addr, MAC_ADDR_LEN) == 0)
 139     {
 140         DBGPRINT(RT_DEBUG_ERROR, ("MlmeDelBAReqSanity fail - the peer addr dosen't exist.\n"));
 141         return FALSE;
 142     }
 143
 144     return TRUE;
 145 }
 146
 147 BOOLEAN PeerAddBAReqActionSanity(
 148     IN PRTMP_ADAPTER pAd,
 149     IN VOID *pMsg,
 150     IN ULONG MsgLen,
 151         OUT PUCHAR pAddr2)
 152 {
 153         PFRAME_802_11 pFrame = (PFRAME_802_11)pMsg;
 154         PFRAME_ADDBA_REQ pAddFrame;
 155         pAddFrame = (PFRAME_ADDBA_REQ)(pMsg);
 156         if (MsgLen < (sizeof(FRAME_ADDBA_REQ)))
 157         {
 158                 DBGPRINT(RT_DEBUG_ERROR,("PeerAddBAReqActionSanity: ADDBA Request frame length size = %ld incorrect\n", MsgLen));
 159                 return FALSE;
 160         }
 161         // we support immediate BA.
 162         *(USHORT *)(&pAddFrame->BaParm) = cpu2le16(*(USHORT *)(&pAddFrame->BaParm));
 163         pAddFrame->TimeOutValue = cpu2le16(pAddFrame->TimeOutValue);
 164         pAddFrame->BaStartSeq.word = cpu2le16(pAddFrame->BaStartSeq.word);
 165
 166         if (pAddFrame->BaParm.BAPolicy != IMMED_BA)
 167         {
 168                 DBGPRINT(RT_DEBUG_ERROR,("PeerAddBAReqActionSanity: ADDBA Request Ba Policy[%d] not support\n", pAddFrame->BaParm.BAPolicy));
 169                 DBGPRINT(RT_DEBUG_ERROR,("ADDBA Request. tid=%x, Bufsize=%x, AMSDUSupported=%x \n", pAddFrame->BaParm.TID, pAddFrame->BaParm.BufSize, pAddFrame->BaParm.AMSDUSupported));
 170                 return FALSE;
 171         }
 172
 173         // we support immediate BA.
 174         if (pAddFrame->BaParm.TID &0xfff0)
 175         {
 176                 DBGPRINT(RT_DEBUG_ERROR,("PeerAddBAReqActionSanity: ADDBA Request incorrect TID = %d\n", pAddFrame->BaParm.TID));
 177                 return FALSE;
 178         }
 179         COPY_MAC_ADDR(pAddr2, pFrame->Hdr.Addr2);
 180         return TRUE;
 181 }
 182
 183 BOOLEAN PeerAddBARspActionSanity(
 184     IN PRTMP_ADAPTER pAd,
 185     IN VOID *pMsg,
 186     IN ULONG MsgLen)
 187 {
 188         //PFRAME_802_11 pFrame = (PFRAME_802_11)pMsg;
 189         PFRAME_ADDBA_RSP pAddFrame;
 190
 191         pAddFrame = (PFRAME_ADDBA_RSP)(pMsg);
 192         if (MsgLen < (sizeof(FRAME_ADDBA_RSP)))
 193         {
 194                 DBGPRINT(RT_DEBUG_ERROR,("PeerAddBARspActionSanity: ADDBA Response frame length size = %ld incorrect\n", MsgLen));
 195                 return FALSE;
 196         }
 197         // we support immediate BA.
 198         *(USHORT *)(&pAddFrame->BaParm) = cpu2le16(*(USHORT *)(&pAddFrame->BaParm));
 199         pAddFrame->StatusCode = cpu2le16(pAddFrame->StatusCode);
 200         pAddFrame->TimeOutValue = cpu2le16(pAddFrame->TimeOutValue);
 201
 202         if (pAddFrame->BaParm.BAPolicy != IMMED_BA)
 203         {
 204                 DBGPRINT(RT_DEBUG_ERROR,("PeerAddBAReqActionSanity: ADDBA Response Ba Policy[%d] not support\n", pAddFrame->BaParm.BAPolicy));
 205                 return FALSE;
 206         }
 207
 208         // we support immediate BA.
 209         if (pAddFrame->BaParm.TID &0xfff0)
 210         {
 211                 DBGPRINT(RT_DEBUG_ERROR,("PeerAddBARspActionSanity: ADDBA Response incorrect TID = %d\n", pAddFrame->BaParm.TID));
 212                 return FALSE;
 213         }
 214         return TRUE;
 215
 216 }
 217
 218 BOOLEAN PeerDelBAActionSanity(
 219     IN PRTMP_ADAPTER pAd,
 220     IN UCHAR Wcid,
 221     IN VOID *pMsg,
 222     IN ULONG MsgLen )
 223 {
 224         //PFRAME_802_11 pFrame = (PFRAME_802_11)pMsg;
 225         PFRAME_DELBA_REQ  pDelFrame;
 226         if (MsgLen != (sizeof(FRAME_DELBA_REQ)))
 227                 return FALSE;
 228
 229         if (Wcid >= MAX_LEN_OF_MAC_TABLE)
 230                 return FALSE;
 231
 232         pDelFrame = (PFRAME_DELBA_REQ)(pMsg);
 233
 234         *(USHORT *)(&pDelFrame->DelbaParm) = cpu2le16(*(USHORT *)(&pDelFrame->DelbaParm));
 235         pDelFrame->ReasonCode = cpu2le16(pDelFrame->ReasonCode);
 236
 237         if (pDelFrame->DelbaParm.TID &0xfff0)
 238                 return FALSE;
 239
 240         return TRUE;
 241 }
 242
 243 /*
 244     ==========================================================================
 245     Description:
 246         MLME message sanity check
 247     Return:
 248         TRUE if all parameters are OK, FALSE otherwise
 249
 250         IRQL = DISPATCH_LEVEL
 251
 252     ==========================================================================
 253  */
 254 BOOLEAN PeerBeaconAndProbeRspSanity(
 255     IN PRTMP_ADAPTER pAd,
 256     IN VOID *Msg,
 257     IN ULONG MsgLen,
 258     IN UCHAR  MsgChannel,
 259     OUT PUCHAR pAddr2,
 260     OUT PUCHAR pBssid,
 261     OUT CHAR Ssid[],
 262     OUT UCHAR *pSsidLen,
 263     OUT UCHAR *pBssType,
 264     OUT USHORT *pBeaconPeriod,
 265     OUT UCHAR *pChannel,
 266     OUT UCHAR *pNewChannel,
 267     OUT LARGE_INTEGER *pTimestamp,
 268     OUT CF_PARM *pCfParm,
 269     OUT USHORT *pAtimWin,
 270     OUT USHORT *pCapabilityInfo,
 271     OUT UCHAR *pErp,
 272     OUT UCHAR *pDtimCount,
 273     OUT UCHAR *pDtimPeriod,
 274     OUT UCHAR *pBcastFlag,
 275     OUT UCHAR *pMessageToMe,
 276     OUT UCHAR SupRate[],
 277     OUT UCHAR *pSupRateLen,
 278     OUT UCHAR ExtRate[],
 279     OUT UCHAR *pExtRateLen,
 280     OUT UCHAR *pCkipFlag,
 281     OUT UCHAR *pAironetCellPowerLimit,
 282     OUT PEDCA_PARM       pEdcaParm,
 283     OUT PQBSS_LOAD_PARM  pQbssLoad,
 284     OUT PQOS_CAPABILITY_PARM pQosCapability,
 285     OUT ULONG *pRalinkIe,
 286     OUT UCHAR            *pHtCapabilityLen,
 287 #ifdef CONFIG_STA_SUPPORT
 288     OUT UCHAR            *pPreNHtCapabilityLen,
 289 #endif // CONFIG_STA_SUPPORT //
 290     OUT HT_CAPABILITY_IE *pHtCapability,
 291         OUT UCHAR                *AddHtInfoLen,
 292         OUT ADD_HT_INFO_IE *AddHtInfo,
 293         OUT UCHAR *NewExtChannelOffset,         // Ht extension channel offset(above or below)
 294     OUT USHORT *LengthVIE,
 295     OUT PNDIS_802_11_VARIABLE_IEs pVIE)
 296 {
 297     CHAR                                *Ptr;
 298 #ifdef CONFIG_STA_SUPPORT
 299         CHAR                            TimLen;
 300 #endif // CONFIG_STA_SUPPORT //
 301     PFRAME_802_11               pFrame;
 302     PEID_STRUCT         pEid;
 303     UCHAR                               SubType;
 304     UCHAR                               Sanity;
 305     //UCHAR                             ECWMin, ECWMax;
 306     //MAC_CSR9_STRUC            Csr9;
 307     ULONG                               Length = 0;
 308
 309         // For some 11a AP which didn't have DS_IE, we use two conditions to decide the channel
 310         //      1. If the AP is 11n enabled, then check the control channel.
 311         //      2. If the AP didn't have any info about channel, use the channel we received this frame as the channel. (May inaccuracy!!)
 312         UCHAR                   CtrlChannel = 0;
 313
 314     // Add for 3 necessary EID field check
 315     Sanity = 0;
 316
 317     *pAtimWin = 0;
 318     *pErp = 0;
 319     *pDtimCount = 0;
 320     *pDtimPeriod = 0;
 321     *pBcastFlag = 0;
 322     *pMessageToMe = 0;
 323     *pExtRateLen = 0;
 324     *pCkipFlag = 0;                             // Default of CkipFlag is 0
 325     *pAironetCellPowerLimit = 0xFF;  // Default of AironetCellPowerLimit is 0xFF
 326     *LengthVIE = 0;                                     // Set the length of VIE to init value 0
 327     *pHtCapabilityLen = 0;                                      // Set the length of VIE to init value 0
 328 #ifdef CONFIG_STA_SUPPORT
 329         if (pAd->OpMode == OPMODE_STA)
 330                 *pPreNHtCapabilityLen = 0;                                      // Set the length of VIE to init value 0
 331 #endif // CONFIG_STA_SUPPORT //
 332     *AddHtInfoLen = 0;                                  // Set the length of VIE to init value 0
 333     *pRalinkIe = 0;
 334     *pNewChannel = 0;
 335     *NewExtChannelOffset = 0xff;        //Default 0xff means no such IE
 336     pCfParm->bValid = FALSE;        // default: no IE_CF found
 337     pQbssLoad->bValid = FALSE;      // default: no IE_QBSS_LOAD found
 338     pEdcaParm->bValid = FALSE;      // default: no IE_EDCA_PARAMETER found
 339     pQosCapability->bValid = FALSE; // default: no IE_QOS_CAPABILITY found
 340
 341     pFrame = (PFRAME_802_11)Msg;
 342
 343     // get subtype from header
 344     SubType = (UCHAR)pFrame->Hdr.FC.SubType;
 345
 346     // get Addr2 and BSSID from header
 347     COPY_MAC_ADDR(pAddr2, pFrame->Hdr.Addr2);
 348     COPY_MAC_ADDR(pBssid, pFrame->Hdr.Addr3);
 349
 350 //      hex_dump("Beacon", Msg, MsgLen);
 351
 352     Ptr = pFrame->Octet;
 353     Length += LENGTH_802_11;
 354
 355     // get timestamp from payload and advance the pointer
 356     NdisMoveMemory(pTimestamp, Ptr, TIMESTAMP_LEN);
 357
 358         pTimestamp->u.LowPart = cpu2le32(pTimestamp->u.LowPart);
 359         pTimestamp->u.HighPart = cpu2le32(pTimestamp->u.HighPart);
 360
 361     Ptr += TIMESTAMP_LEN;
 362     Length += TIMESTAMP_LEN;
 363
 364     // get beacon interval from payload and advance the pointer
 365     NdisMoveMemory(pBeaconPeriod, Ptr, 2);
 366     Ptr += 2;
 367     Length += 2;
 368
 369     // get capability info from payload and advance the pointer
 370     NdisMoveMemory(pCapabilityInfo, Ptr, 2);
 371     Ptr += 2;
 372     Length += 2;
 373
 374     if (CAP_IS_ESS_ON(*pCapabilityInfo))
 375         *pBssType = BSS_INFRA;
 376     else
 377         *pBssType = BSS_ADHOC;
 378
 379     pEid = (PEID_STRUCT) Ptr;
 380
 381     // get variable fields from payload and advance the pointer
 382     while ((Length + 2 + pEid->Len) <= MsgLen)
 383     {
 384         //
 385         // Secure copy VIE to VarIE[MAX_VIE_LEN] didn't overflow.
 386         //
 387         if ((*LengthVIE + pEid->Len + 2) >= MAX_VIE_LEN)
 388         {
 389             DBGPRINT(RT_DEBUG_WARN, ("PeerBeaconAndProbeRspSanity - Variable IEs out of resource [len(=%d) > MAX_VIE_LEN(=%d)]\n",
 390                     (*LengthVIE + pEid->Len + 2), MAX_VIE_LEN));
 391             break;
 392         }
 393
 394         switch(pEid->Eid)
 395         {
 396             case IE_SSID:
 397                 // Already has one SSID EID in this beacon, ignore the second one
 398                 if (Sanity & 0x1)
 399                     break;
 400                 if(pEid->Len <= MAX_LEN_OF_SSID)
 401                 {
 402                     NdisMoveMemory(Ssid, pEid->Octet, pEid->Len);
 403                     *pSsidLen = pEid->Len;
 404                     Sanity |= 0x1;
 405                 }
 406                 else
 407                 {
 408                     DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity - wrong IE_SSID (len=%d)\n",pEid->Len));
 409                     return FALSE;
 410                 }
 411                 break;
 412
 413             case IE_SUPP_RATES:
 414                 if(pEid->Len <= MAX_LEN_OF_SUPPORTED_RATES)
 415                 {
 416                     Sanity |= 0x2;
 417                     NdisMoveMemory(SupRate, pEid->Octet, pEid->Len);
 418                     *pSupRateLen = pEid->Len;
 419
 420                     // TODO: 2004-09-14 not a good design here, cause it exclude extra rates
 421                     // from ScanTab. We should report as is. And filter out unsupported
 422                     // rates in MlmeAux.
 423                     // Check against the supported rates
 424                     // RTMPCheckRates(pAd, SupRate, pSupRateLen);
 425                 }
 426                 else
 427                 {
 428                     DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity - wrong IE_SUPP_RATES (len=%d)\n",pEid->Len));
 429                     return FALSE;
 430                 }
 431                 break;
 432
 433             case IE_HT_CAP:
 434                         if (pEid->Len >= SIZE_HT_CAP_IE)  //Note: allow extension.!!
 435                         {
 436                                 NdisMoveMemory(pHtCapability, pEid->Octet, sizeof(HT_CAPABILITY_IE));
 437                                 *pHtCapabilityLen = SIZE_HT_CAP_IE;     // Nnow we only support 26 bytes.
 438
 439                                 *(USHORT *)(&pHtCapability->HtCapInfo) = cpu2le16(*(USHORT *)(&pHtCapability->HtCapInfo));
 440                                 *(USHORT *)(&pHtCapability->ExtHtCapInfo) = cpu2le16(*(USHORT *)(&pHtCapability->ExtHtCapInfo));
 441
 442 #ifdef CONFIG_STA_SUPPORT
 443                                 IF_DEV_CONFIG_OPMODE_ON_STA(pAd)
 444                                 {
 445                                         *pPreNHtCapabilityLen = 0;      // Nnow we only support 26 bytes.
 446
 447                                         Ptr = (PUCHAR) pVIE;
 448                                         NdisMoveMemory(Ptr + *LengthVIE, &pEid->Eid, pEid->Len + 2);
 449                                         *LengthVIE += (pEid->Len + 2);
 450                                 }
 451 #endif // CONFIG_STA_SUPPORT //
 452                         }
 453                         else
 454                         {
 455                                 DBGPRINT(RT_DEBUG_WARN, ("PeerBeaconAndProbeRspSanity - wrong IE_HT_CAP. pEid->Len = %d\n", pEid->Len));
 456                         }
 457
 458                 break;
 459             case IE_ADD_HT:
 460                         if (pEid->Len >= sizeof(ADD_HT_INFO_IE))
 461                         {
 462                                 // This IE allows extension, but we can ignore extra bytes beyond our knowledge , so only
 463                                 // copy first sizeof(ADD_HT_INFO_IE)
 464                                 NdisMoveMemory(AddHtInfo, pEid->Octet, sizeof(ADD_HT_INFO_IE));
 465                                 *AddHtInfoLen = SIZE_ADD_HT_INFO_IE;
 466
 467                                 CtrlChannel = AddHtInfo->ControlChan;
 468
 469                                 *(USHORT *)(&AddHtInfo->AddHtInfo2) = cpu2le16(*(USHORT *)(&AddHtInfo->AddHtInfo2));
 470                                 *(USHORT *)(&AddHtInfo->AddHtInfo3) = cpu2le16(*(USHORT *)(&AddHtInfo->AddHtInfo3));
 471
 472 #ifdef CONFIG_STA_SUPPORT
 473                                 IF_DEV_CONFIG_OPMODE_ON_STA(pAd)
 474                                 {
 475                                         Ptr = (PUCHAR) pVIE;
 476                                         NdisMoveMemory(Ptr + *LengthVIE, &pEid->Eid, pEid->Len + 2);
 477                                         *LengthVIE += (pEid->Len + 2);
 478                                 }
 479 #endif // CONFIG_STA_SUPPORT //
 480                         }
 481                         else
 482                         {
 483                                 DBGPRINT(RT_DEBUG_WARN, ("PeerBeaconAndProbeRspSanity - wrong IE_ADD_HT. \n"));
 484                         }
 485
 486                 break;
 487             case IE_SECONDARY_CH_OFFSET:
 488                         if (pEid->Len == 1)
 489                         {
 490                                 *NewExtChannelOffset = pEid->Octet[0];
 491                         }
 492                         else
 493                         {
 494                                 DBGPRINT(RT_DEBUG_WARN, ("PeerBeaconAndProbeRspSanity - wrong IE_SECONDARY_CH_OFFSET. \n"));
 495                         }
 496
 497                 break;
 498             case IE_FH_PARM:
 499                 DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity(IE_FH_PARM) \n"));
 500                 break;
 501
 502             case IE_DS_PARM:
 503                 if(pEid->Len == 1)
 504                 {
 505                     *pChannel = *pEid->Octet;
 506 #ifdef CONFIG_STA_SUPPORT
 507                                         IF_DEV_CONFIG_OPMODE_ON_STA(pAd)
 508                                         {
 509                                                 if (ChannelSanity(pAd, *pChannel) == 0)
 510                                                 {
 511
 512                                                         return FALSE;
 513                                                 }
 514                                         }
 515 #endif // CONFIG_STA_SUPPORT //
 516                     Sanity |= 0x4;
 517                 }
 518                 else
 519                 {
 520                     DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity - wrong IE_DS_PARM (len=%d)\n",pEid->Len));
 521                     return FALSE;
 522                 }
 523                 break;
 524
 525             case IE_CF_PARM:
 526                 if(pEid->Len == 6)
 527                 {
 528                     pCfParm->bValid = TRUE;
 529                     pCfParm->CfpCount = pEid->Octet[0];
 530                     pCfParm->CfpPeriod = pEid->Octet[1];
 531                     pCfParm->CfpMaxDuration = pEid->Octet[2] + 256 * pEid->Octet[3];
 532                     pCfParm->CfpDurRemaining = pEid->Octet[4] + 256 * pEid->Octet[5];
 533                 }
 534                 else
 535                 {
 536                     DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity - wrong IE_CF_PARM\n"));
 537                     return FALSE;
 538                 }
 539                 break;
 540
 541             case IE_IBSS_PARM:
 542                 if(pEid->Len == 2)
 543                 {
 544                     NdisMoveMemory(pAtimWin, pEid->Octet, pEid->Len);
 545                 }
 546                 else
 547                 {
 548                     DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity - wrong IE_IBSS_PARM\n"));
 549                     return FALSE;
 550                 }
 551                 break;
 552
 553 #ifdef CONFIG_STA_SUPPORT
 554             case IE_TIM:
 555                 if(INFRA_ON(pAd) && SubType == SUBTYPE_BEACON)
 556                 {
 557                     GetTimBit((PUCHAR)pEid, pAd->StaActive.Aid, &TimLen, pBcastFlag, pDtimCount, pDtimPeriod, pMessageToMe);
 558                 }
 559                 break;
 560 #endif // CONFIG_STA_SUPPORT //
 561             case IE_CHANNEL_SWITCH_ANNOUNCEMENT:
 562                 if(pEid->Len == 3)
 563                 {
 564                         *pNewChannel = pEid->Octet[1];  //extract new channel number
 565                 }
 566                 break;
 567
 568             // New for WPA
 569             // CCX v2 has the same IE, we need to parse that too
 570             // Wifi WMM use the same IE vale, need to parse that too
 571             // case IE_WPA:
 572             case IE_VENDOR_SPECIFIC:
 573                 // Check Broadcom/Atheros 802.11n OUI version, for HT Capability IE.
 574                 // This HT IE is before IEEE draft set HT IE value.2006-09-28 by Jan.
 575                 /*if (NdisEqualMemory(pEid->Octet, BROADCOM_OUI, 3) && (pEid->Len >= 4))
 576                 {
 577                         if ((pEid->Octet[3] == OUI_BROADCOM_HT) && (pEid->Len >= 30))
 578                         {
 579                                 {
 580                                         NdisMoveMemory(pHtCapability, &pEid->Octet[4], sizeof(HT_CAPABILITY_IE));
 581                                         *pHtCapabilityLen = SIZE_HT_CAP_IE;     // Nnow we only support 26 bytes.
 582                                 }
 583                         }
 584                         if ((pEid->Octet[3] == OUI_BROADCOM_HT) && (pEid->Len >= 26))
 585                         {
 586                                 {
 587                                         NdisMoveMemory(AddHtInfo, &pEid->Octet[4], sizeof(ADD_HT_INFO_IE));
 588                                         *AddHtInfoLen = SIZE_ADD_HT_INFO_IE;    // Nnow we only support 26 bytes.
 589                                 }
 590                         }
 591                 }
 592                                 */
 593                 // Check the OUI version, filter out non-standard usage
 594                 if (NdisEqualMemory(pEid->Octet, RALINK_OUI, 3) && (pEid->Len == 7))
 595                 {
 596                     //*pRalinkIe = pEid->Octet[3];
 597                     if (pEid->Octet[3] != 0)
 598                                         *pRalinkIe = pEid->Octet[3];
 599                                 else
 600                                         *pRalinkIe = 0xf0000000; // Set to non-zero value (can't set bit0-2) to represent this is Ralink Chip. So at linkup, we will set ralinkchip flag.
 601                 }
 602 #ifdef CONFIG_STA_SUPPORT
 603 #ifdef DOT11_N_SUPPORT
 604                 // This HT IE is before IEEE draft set HT IE value.2006-09-28 by Jan.
 605
 606                 // Other vendors had production before IE_HT_CAP value is assigned. To backward support those old-firmware AP,
 607                 // Check broadcom-defiend pre-802.11nD1.0 OUI for HT related IE, including HT Capatilities IE and HT Information IE
 608                 else if ((*pHtCapabilityLen == 0) && NdisEqualMemory(pEid->Octet, PRE_N_HT_OUI, 3) && (pEid->Len >= 4) && (pAd->OpMode == OPMODE_STA))
 609                 {
 610                     if ((pEid->Octet[3] == OUI_PREN_HT_CAP) && (pEid->Len >= 30) && (*pHtCapabilityLen == 0))
 611                     {
 612                         NdisMoveMemory(pHtCapability, &pEid->Octet[4], sizeof(HT_CAPABILITY_IE));
 613                         *pPreNHtCapabilityLen = SIZE_HT_CAP_IE;
 614                     }
 615
 616                     if ((pEid->Octet[3] == OUI_PREN_ADD_HT) && (pEid->Len >= 26))
 617                     {
 618                         NdisMoveMemory(AddHtInfo, &pEid->Octet[4], sizeof(ADD_HT_INFO_IE));
 619                         *AddHtInfoLen = SIZE_ADD_HT_INFO_IE;
 620                     }
 621                 }
 622 #endif // DOT11_N_SUPPORT //
 623 #endif // CONFIG_STA_SUPPORT //
 624                 else if (NdisEqualMemory(pEid->Octet, WPA_OUI, 4))
 625                 {
 626                     // Copy to pVIE which will report to microsoft bssid list.
 627                     Ptr = (PUCHAR) pVIE;
 628                     NdisMoveMemory(Ptr + *LengthVIE, &pEid->Eid, pEid->Len + 2);
 629                     *LengthVIE += (pEid->Len + 2);
 630                 }
 631                 else if (NdisEqualMemory(pEid->Octet, WME_PARM_ELEM, 6) && (pEid->Len == 24))
 632                 {
 633                     PUCHAR ptr;
 634                     int i;
 635
 636                     // parsing EDCA parameters
 637                     pEdcaParm->bValid          = TRUE;
 638                     pEdcaParm->bQAck           = FALSE; // pEid->Octet[0] & 0x10;
 639                     pEdcaParm->bQueueRequest   = FALSE; // pEid->Octet[0] & 0x20;
 640                     pEdcaParm->bTxopRequest    = FALSE; // pEid->Octet[0] & 0x40;
 641                     pEdcaParm->EdcaUpdateCount = pEid->Octet[6] & 0x0f;
 642                     pEdcaParm->bAPSDCapable    = (pEid->Octet[6] & 0x80) ? 1 : 0;
 643                     ptr = &pEid->Octet[8];
 644                     for (i=0; i<4; i++)
 645                     {
 646                         UCHAR aci = (*ptr & 0x60) >> 5; // b5~6 is AC INDEX
 647                         pEdcaParm->bACM[aci]  = (((*ptr) & 0x10) == 0x10);   // b5 is ACM
 648                         pEdcaParm->Aifsn[aci] = (*ptr) & 0x0f;               // b0~3 is AIFSN
 649                         pEdcaParm->Cwmin[aci] = *(ptr+1) & 0x0f;             // b0~4 is Cwmin
 650                         pEdcaParm->Cwmax[aci] = *(ptr+1) >> 4;               // b5~8 is Cwmax
 651                         pEdcaParm->Txop[aci]  = *(ptr+2) + 256 * (*(ptr+3)); // in unit of 32-us
 652                         ptr += 4; // point to next AC
 653                     }
 654                 }
 655                 else if (NdisEqualMemory(pEid->Octet, WME_INFO_ELEM, 6) && (pEid->Len == 7))
 656                 {
 657                     // parsing EDCA parameters
 658                     pEdcaParm->bValid          = TRUE;
 659                     pEdcaParm->bQAck           = FALSE; // pEid->Octet[0] & 0x10;
 660                     pEdcaParm->bQueueRequest   = FALSE; // pEid->Octet[0] & 0x20;
 661                     pEdcaParm->bTxopRequest    = FALSE; // pEid->Octet[0] & 0x40;
 662                     pEdcaParm->EdcaUpdateCount = pEid->Octet[6] & 0x0f;
 663                     pEdcaParm->bAPSDCapable    = (pEid->Octet[6] & 0x80) ? 1 : 0;
 664
 665                     // use default EDCA parameter
 666                     pEdcaParm->bACM[QID_AC_BE]  = 0;
 667                     pEdcaParm->Aifsn[QID_AC_BE] = 3;
 668                     pEdcaParm->Cwmin[QID_AC_BE] = CW_MIN_IN_BITS;
 669                     pEdcaParm->Cwmax[QID_AC_BE] = CW_MAX_IN_BITS;
 670                     pEdcaParm->Txop[QID_AC_BE]  = 0;
 671
 672                     pEdcaParm->bACM[QID_AC_BK]  = 0;
 673                     pEdcaParm->Aifsn[QID_AC_BK] = 7;
 674                     pEdcaParm->Cwmin[QID_AC_BK] = CW_MIN_IN_BITS;
 675                     pEdcaParm->Cwmax[QID_AC_BK] = CW_MAX_IN_BITS;
 676                     pEdcaParm->Txop[QID_AC_BK]  = 0;
 677
 678                     pEdcaParm->bACM[QID_AC_VI]  = 0;
 679                     pEdcaParm->Aifsn[QID_AC_VI] = 2;
 680                     pEdcaParm->Cwmin[QID_AC_VI] = CW_MIN_IN_BITS-1;
 681                     pEdcaParm->Cwmax[QID_AC_VI] = CW_MAX_IN_BITS;
 682                     pEdcaParm->Txop[QID_AC_VI]  = 96;   // AC_VI: 96*32us ~= 3ms
 683
 684                     pEdcaParm->bACM[QID_AC_VO]  = 0;
 685                     pEdcaParm->Aifsn[QID_AC_VO] = 2;
 686                     pEdcaParm->Cwmin[QID_AC_VO] = CW_MIN_IN_BITS-2;
 687                     pEdcaParm->Cwmax[QID_AC_VO] = CW_MAX_IN_BITS-1;
 688                     pEdcaParm->Txop[QID_AC_VO]  = 48;   // AC_VO: 48*32us ~= 1.5ms
 689                 }
 690                 break;
 691
 692             case IE_EXT_SUPP_RATES:
 693                 if (pEid->Len <= MAX_LEN_OF_SUPPORTED_RATES)
 694                 {
 695                     NdisMoveMemory(ExtRate, pEid->Octet, pEid->Len);
 696                     *pExtRateLen = pEid->Len;
 697
 698                     // TODO: 2004-09-14 not a good design here, cause it exclude extra rates
 699                     // from ScanTab. We should report as is. And filter out unsupported
 700                     // rates in MlmeAux.
 701                     // Check against the supported rates
 702                     // RTMPCheckRates(pAd, ExtRate, pExtRateLen);
 703                 }
 704                 break;
 705
 706             case IE_ERP:
 707                 if (pEid->Len == 1)
 708                 {
 709                     *pErp = (UCHAR)pEid->Octet[0];
 710                 }
 711                 break;
 712
 713             case IE_AIRONET_CKIP:
 714                 // 0. Check Aironet IE length, it must be larger or equal to 28
 715                 // Cisco AP350 used length as 28
 716                 // Cisco AP12XX used length as 30
 717                 if (pEid->Len < (CKIP_NEGOTIATION_LENGTH - 2))
 718                     break;
 719
 720                 // 1. Copy CKIP flag byte to buffer for process
 721                 *pCkipFlag = *(pEid->Octet + 8);
 722                 break;
 723
 724             case IE_AP_TX_POWER:
 725                 // AP Control of Client Transmit Power
 726                 //0. Check Aironet IE length, it must be 6
 727                 if (pEid->Len != 0x06)
 728                     break;
 729
 730                 // Get cell power limit in dBm
 731                 if (NdisEqualMemory(pEid->Octet, CISCO_OUI, 3) == 1)
 732                     *pAironetCellPowerLimit = *(pEid->Octet + 4);
 733                 break;
 734
 735             // WPA2 & 802.11i RSN
 736             case IE_RSN:
 737                 // There is no OUI for version anymore, check the group cipher OUI before copying
 738                 if (RTMPEqualMemory(pEid->Octet + 2, RSN_OUI, 3))
 739                 {
 740                     // Copy to pVIE which will report to microsoft bssid list.
 741                     Ptr = (PUCHAR) pVIE;
 742                     NdisMoveMemory(Ptr + *LengthVIE, &pEid->Eid, pEid->Len + 2);
 743                     *LengthVIE += (pEid->Len + 2);
 744                 }
 745                 break;
 746 #ifdef CONFIG_STA_SUPPORT
 747 #ifdef EXT_BUILD_CHANNEL_LIST
 748                         case IE_COUNTRY:
 749                                 Ptr = (PUCHAR) pVIE;
 750                 NdisMoveMemory(Ptr + *LengthVIE, &pEid->Eid, pEid->Len + 2);
 751                 *LengthVIE += (pEid->Len + 2);
 752                                 break;
 753 #endif // EXT_BUILD_CHANNEL_LIST //
 754 #endif // CONFIG_STA_SUPPORT //
 755             default:
 756                 break;
 757         }
 758
 759         Length = Length + 2 + pEid->Len;  // Eid[1] + Len[1]+ content[Len]
 760         pEid = (PEID_STRUCT)((UCHAR*)pEid + 2 + pEid->Len);
 761     }
 762
 763     // For some 11a AP. it did not have the channel EID, patch here
 764 #ifdef CONFIG_STA_SUPPORT
 765         IF_DEV_CONFIG_OPMODE_ON_STA(pAd)
 766         {
 767                 UCHAR LatchRfChannel = MsgChannel;
 768                 if ((pAd->LatchRfRegs.Channel > 14) && ((Sanity & 0x4) == 0))
 769                 {
 770                         if (CtrlChannel != 0)
 771                                 *pChannel = CtrlChannel;
 772                         else
 773                                 *pChannel = LatchRfChannel;
 774                         Sanity |= 0x4;
 775                 }
 776         }
 777 #endif // CONFIG_STA_SUPPORT //
 778
 779         if (Sanity != 0x7)
 780         {
 781                 DBGPRINT(RT_DEBUG_WARN, ("PeerBeaconAndProbeRspSanity - missing field, Sanity=0x%02x\n", Sanity));
 782                 return FALSE;
 783         }
 784         else
 785         {
 786                 return TRUE;
 787         }
 788
 789 }
 790
 791 #ifdef DOT11N_DRAFT3
 792 /*
 793         ==========================================================================
 794         Description:
 795                 MLME message sanity check for some IE addressed  in 802.11n d3.03.
 796         Return:
 797                 TRUE if all parameters are OK, FALSE otherwise
 798
 799         IRQL = DISPATCH_LEVEL
 800
 801         ==========================================================================
 802  */
 803 BOOLEAN PeerBeaconAndProbeRspSanity2(
 804         IN PRTMP_ADAPTER pAd,
 805         IN VOID *Msg,
 806         IN ULONG MsgLen,
 807         OUT UCHAR       *RegClass)
 808 {
 809         CHAR                            *Ptr;
 810         PFRAME_802_11           pFrame;
 811         PEID_STRUCT                     pEid;
 812         ULONG                           Length = 0;
 813
 814         pFrame = (PFRAME_802_11)Msg;
 815
 816         *RegClass = 0;
 817         Ptr = pFrame->Octet;
 818         Length += LENGTH_802_11;
 819
 820         // get timestamp from payload and advance the pointer
 821         Ptr += TIMESTAMP_LEN;
 822         Length += TIMESTAMP_LEN;
 823
 824         // get beacon interval from payload and advance the pointer
 825         Ptr += 2;
 826         Length += 2;
 827
 828         // get capability info from payload and advance the pointer
 829         Ptr += 2;
 830         Length += 2;
 831
 832         pEid = (PEID_STRUCT) Ptr;
 833
 834         // get variable fields from payload and advance the pointer
 835         while ((Length + 2 + pEid->Len) <= MsgLen)
 836         {
 837                 switch(pEid->Eid)
 838                 {
 839                         case IE_SUPP_REG_CLASS:
 840                                 if(pEid->Len > 0)
 841                                 {
 842                                         *RegClass = *pEid->Octet;
 843                                 }
 844                                 else
 845                                 {
 846                                         DBGPRINT(RT_DEBUG_TRACE, ("PeerBeaconAndProbeRspSanity - wrong IE_SSID (len=%d)\n",pEid->Len));
 847                                         return FALSE;
 848                                 }
 849                                 break;
 850                 }
 851
 852                 Length = Length + 2 + pEid->Len;  // Eid[1] + Len[1]+ content[Len]
 853                 pEid = (PEID_STRUCT)((UCHAR*)pEid + 2 + pEid->Len);
 854         }
 855
 856         return TRUE;
 857
 858 }
 859 #endif // DOT11N_DRAFT3 //
 860
 861 /*
 862     ==========================================================================
 863     Description:
 864         MLME message sanity check
 865     Return:
 866         TRUE if all parameters are OK, FALSE otherwise
 867     ==========================================================================
 868  */
 869 BOOLEAN MlmeScanReqSanity(
 870         IN PRTMP_ADAPTER pAd,
 871         IN VOID *Msg,
 872         IN ULONG MsgLen,
 873         OUT UCHAR *pBssType,
 874         OUT CHAR Ssid[],
 875         OUT UCHAR *pSsidLen,
 876         OUT UCHAR *pScanType)
 877 {
 878         MLME_SCAN_REQ_STRUCT *Info;
 879
 880         Info = (MLME_SCAN_REQ_STRUCT *)(Msg);
 881         *pBssType = Info->BssType;
 882         *pSsidLen = Info->SsidLen;
 883         NdisMoveMemory(Ssid, Info->Ssid, *pSsidLen);
 884         *pScanType = Info->ScanType;
 885
 886         if ((*pBssType == BSS_INFRA || *pBssType == BSS_ADHOC || *pBssType == BSS_ANY)
 887                 && (*pScanType == SCAN_ACTIVE || *pScanType == SCAN_PASSIVE
 888 #ifdef CONFIG_STA_SUPPORT
 889                 || *pScanType == SCAN_CISCO_PASSIVE || *pScanType == SCAN_CISCO_ACTIVE
 890                 || *pScanType == SCAN_CISCO_CHANNEL_LOAD || *pScanType == SCAN_CISCO_NOISE
 891 #endif // CONFIG_STA_SUPPORT //
 892                 ))
 893         {
 894                 return TRUE;
 895         }
 896         else
 897         {
 898                 DBGPRINT(RT_DEBUG_TRACE, ("MlmeScanReqSanity fail - wrong BssType or ScanType\n"));
 899                 return FALSE;
 900         }
 901 }
 902
 903 // IRQL = DISPATCH_LEVEL
 904 UCHAR ChannelSanity(
 905     IN PRTMP_ADAPTER pAd,
 906     IN UCHAR channel)
 907 {
 908     int i;
 909
 910     for (i = 0; i < pAd->ChannelListNum; i ++)
 911     {
 912         if (channel == pAd->ChannelList[i].Channel)
 913             return 1;
 914     }
 915     return 0;
 916 }
 917
 918 /*
 919     ==========================================================================
 920     Description:
 921         MLME message sanity check
 922     Return:
 923         TRUE if all parameters are OK, FALSE otherwise
 924
 925         IRQL = DISPATCH_LEVEL
 926
 927     ==========================================================================
 928  */
 929 BOOLEAN PeerDeauthSanity(
 930     IN PRTMP_ADAPTER pAd,
 931     IN VOID *Msg,
 932     IN ULONG MsgLen,
 933     OUT PUCHAR pAddr2,
 934     OUT USHORT *pReason)
 935 {
 936     PFRAME_802_11 pFrame = (PFRAME_802_11)Msg;
 937
 938     COPY_MAC_ADDR(pAddr2, pFrame->Hdr.Addr2);
 939     NdisMoveMemory(pReason, &pFrame->Octet[0], 2);
 940
 941     return TRUE;
 942 }
 943
 944 /*
 945     ==========================================================================
 946     Description:
 947         MLME message sanity check
 948     Return:
 949         TRUE if all parameters are OK, FALSE otherwise
 950
 951         IRQL = DISPATCH_LEVEL
 952
 953     ==========================================================================
 954  */
 955 BOOLEAN PeerAuthSanity(
 956     IN PRTMP_ADAPTER pAd,
 957     IN VOID *Msg,
 958     IN ULONG MsgLen,
 959     OUT PUCHAR pAddr,
 960     OUT USHORT *pAlg,
 961     OUT USHORT *pSeq,
 962     OUT USHORT *pStatus,
 963     CHAR *pChlgText)
 964 {
 965     PFRAME_802_11 pFrame = (PFRAME_802_11)Msg;
 966
 967     COPY_MAC_ADDR(pAddr,   pFrame->Hdr.Addr2);
 968     NdisMoveMemory(pAlg,    &pFrame->Octet[0], 2);
 969     NdisMoveMemory(pSeq,    &pFrame->Octet[2], 2);
 970     NdisMoveMemory(pStatus, &pFrame->Octet[4], 2);
 971
 972     if ((*pAlg == Ndis802_11AuthModeOpen)
 973 #ifdef LEAP_SUPPORT
 974       || (*pAlg == CISCO_AuthModeLEAP)
 975 #endif // LEAP_SUPPORT //
 976       )
 977     {
 978         if (*pSeq == 1 || *pSeq == 2)
 979         {
 980             return TRUE;
 981         }
 982         else
 983         {
 984             DBGPRINT(RT_DEBUG_TRACE, ("PeerAuthSanity fail - wrong Seg#\n"));
 985             return FALSE;
 986         }
 987     }
 988     else if (*pAlg == Ndis802_11AuthModeShared)
 989     {
 990         if (*pSeq == 1 || *pSeq == 4)
 991         {
 992             return TRUE;
 993         }
 994         else if (*pSeq == 2 || *pSeq == 3)
 995         {
 996             NdisMoveMemory(pChlgText, &pFrame->Octet[8], CIPHER_TEXT_LEN);
 997             return TRUE;
 998         }
 999         else
1000         {
1001             DBGPRINT(RT_DEBUG_TRACE, ("PeerAuthSanity fail - wrong Seg#\n"));
1002             return FALSE;
1003         }
1004     }
1005     else
1006     {
1007         DBGPRINT(RT_DEBUG_TRACE, ("PeerAuthSanity fail - wrong algorithm\n"));
1008         return FALSE;
1009     }
1010 }
1011
1012 /*
1013     ==========================================================================
1014     Description:
1015         MLME message sanity check
1016     Return:
1017         TRUE if all parameters are OK, FALSE otherwise
1018     ==========================================================================
1019  */
1020 BOOLEAN MlmeAuthReqSanity(
1021     IN PRTMP_ADAPTER pAd,
1022     IN VOID *Msg,
1023     IN ULONG MsgLen,
1024     OUT PUCHAR pAddr,
1025     OUT ULONG *pTimeout,
1026     OUT USHORT *pAlg)
1027 {
1028     MLME_AUTH_REQ_STRUCT *pInfo;
1029
1030     pInfo  = (MLME_AUTH_REQ_STRUCT *)Msg;
1031     COPY_MAC_ADDR(pAddr, pInfo->Addr);
1032     *pTimeout = pInfo->Timeout;
1033     *pAlg = pInfo->Alg;
1034
1035     if (((*pAlg == Ndis802_11AuthModeShared) ||(*pAlg == Ndis802_11AuthModeOpen)
1036 #ifdef LEAP_SUPPORT
1037      || (*pAlg == CISCO_AuthModeLEAP)
1038 #endif // LEAP_SUPPORT //
1039         ) &&
1040         ((*pAddr & 0x01) == 0))
1041     {
1042         return TRUE;
1043     }
1044     else
1045     {
1046         DBGPRINT(RT_DEBUG_TRACE, ("MlmeAuthReqSanity fail - wrong algorithm\n"));
1047         return FALSE;
1048     }
1049 }
1050
1051 /*
1052     ==========================================================================
1053     Description:
1054         MLME message sanity check
1055     Return:
1056         TRUE if all parameters are OK, FALSE otherwise
1057
1058         IRQL = DISPATCH_LEVEL
1059
1060     ==========================================================================
1061  */
1062 BOOLEAN MlmeAssocReqSanity(
1063     IN PRTMP_ADAPTER pAd,
1064     IN VOID *Msg,
1065     IN ULONG MsgLen,
1066     OUT PUCHAR pApAddr,
1067     OUT USHORT *pCapabilityInfo,
1068     OUT ULONG *pTimeout,
1069     OUT USHORT *pListenIntv)
1070 {
1071     MLME_ASSOC_REQ_STRUCT *pInfo;
1072
1073     pInfo = (MLME_ASSOC_REQ_STRUCT *)Msg;
1074     *pTimeout = pInfo->Timeout;                             // timeout
1075     COPY_MAC_ADDR(pApAddr, pInfo->Addr);                   // AP address
1076     *pCapabilityInfo = pInfo->CapabilityInfo;               // capability info
1077     *pListenIntv = pInfo->ListenIntv;
1078
1079     return TRUE;
1080 }
1081
1082 /*
1083     ==========================================================================
1084     Description:
1085         MLME message sanity check
1086     Return:
1087         TRUE if all parameters are OK, FALSE otherwise
1088
1089         IRQL = DISPATCH_LEVEL
1090
1091     ==========================================================================
1092  */
1093 BOOLEAN PeerDisassocSanity(
1094     IN PRTMP_ADAPTER pAd,
1095     IN VOID *Msg,
1096     IN ULONG MsgLen,
1097     OUT PUCHAR pAddr2,
1098     OUT USHORT *pReason)
1099 {
1100     PFRAME_802_11 pFrame = (PFRAME_802_11)Msg;
1101
1102     COPY_MAC_ADDR(pAddr2, pFrame->Hdr.Addr2);
1103     NdisMoveMemory(pReason, &pFrame->Octet[0], 2);
1104
1105     return TRUE;
1106 }
1107
1108 /*
1109         ========================================================================
1110         Routine Description:
1111                 Sanity check NetworkType (11b, 11g or 11a)
1112
1113         Arguments:
1114                 pBss - Pointer to BSS table.
1115
1116         Return Value:
1117         Ndis802_11DS .......(11b)
1118         Ndis802_11OFDM24....(11g)
1119         Ndis802_11OFDM5.....(11a)
1120
1121         IRQL = DISPATCH_LEVEL
1122
1123         ========================================================================
1124 */
1125 NDIS_802_11_NETWORK_TYPE NetworkTypeInUseSanity(
1126     IN PBSS_ENTRY pBss)
1127 {
1128         NDIS_802_11_NETWORK_TYPE        NetWorkType;
1129         UCHAR                                           rate, i;
1130
1131         NetWorkType = Ndis802_11DS;
1132
1133         if (pBss->Channel <= 14)
1134         {
1135                 //
1136                 // First check support Rate.
1137                 //
1138                 for (i = 0; i < pBss->SupRateLen; i++)
1139                 {
1140                         rate = pBss->SupRate[i] & 0x7f; // Mask out basic rate set bit
1141                         if ((rate == 2) || (rate == 4) || (rate == 11) || (rate == 22))
1142                         {
1143                                 continue;
1144                         }
1145                         else
1146                         {
1147                                 //
1148                                 // Otherwise (even rate > 108) means Ndis802_11OFDM24
1149                                 //
1150                                 NetWorkType = Ndis802_11OFDM24;
1151                                 break;
1152                         }
1153                 }
1154
1155                 //
1156                 // Second check Extend Rate.
1157                 //
1158                 if (NetWorkType != Ndis802_11OFDM24)
1159                 {
1160                         for (i = 0; i < pBss->ExtRateLen; i++)
1161                         {
1162                                 rate = pBss->SupRate[i] & 0x7f; // Mask out basic rate set bit
1163                                 if ((rate == 2) || (rate == 4) || (rate == 11) || (rate == 22))
1164                                 {
1165                                         continue;
1166                                 }
1167                                 else
1168                                 {
1169                                         //
1170                                         // Otherwise (even rate > 108) means Ndis802_11OFDM24
1171                                         //
1172                                         NetWorkType = Ndis802_11OFDM24;
1173                                         break;
1174                                 }
1175                         }
1176                 }
1177         }
1178         else
1179         {
1180                 NetWorkType = Ndis802_11OFDM5;
1181         }
1182
1183     if (pBss->HtCapabilityLen != 0)
1184     {
1185         if (NetWorkType == Ndis802_11OFDM5)
1186             NetWorkType = Ndis802_11OFDM5_N;
1187         else
1188             NetWorkType = Ndis802_11OFDM24_N;
1189     }
1190
1191         return NetWorkType;
1192 }
1193
1194 /*
1195     ==========================================================================
1196     Description:
1197         WPA message sanity check
1198     Return:
1199         TRUE if all parameters are OK, FALSE otherwise
1200     ==========================================================================
1201  */
1202 BOOLEAN PeerWpaMessageSanity(
1203     IN  PRTMP_ADAPTER           pAd,
1204     IN  PEAPOL_PACKET           pMsg,
1205     IN  ULONG                           MsgLen,
1206     IN  UCHAR                           MsgType,
1207     IN  MAC_TABLE_ENTRY         *pEntry)
1208 {
1209         UCHAR                   mic[LEN_KEY_DESC_MIC], digest[80], KEYDATA[MAX_LEN_OF_RSNIE];
1210         BOOLEAN                 bReplayDiff = FALSE;
1211         BOOLEAN                 bWPA2 = FALSE;
1212         KEY_INFO                EapolKeyInfo;
1213         UCHAR                   GroupKeyIndex = 0;
1214
1215
1216         NdisZeroMemory(mic, sizeof(mic));
1217         NdisZeroMemory(digest, sizeof(digest));
1218         NdisZeroMemory(KEYDATA, sizeof(KEYDATA));
1219         NdisZeroMemory((PUCHAR)&EapolKeyInfo, sizeof(EapolKeyInfo));
1220
1221         NdisMoveMemory((PUCHAR)&EapolKeyInfo, (PUCHAR)&pMsg->KeyDesc.KeyInfo, sizeof(KEY_INFO));
1222
1223         *((USHORT *)&EapolKeyInfo) = cpu2le16(*((USHORT *)&EapolKeyInfo));
1224
1225         // Choose WPA2 or not
1226         if ((pEntry->AuthMode == Ndis802_11AuthModeWPA2) || (pEntry->AuthMode == Ndis802_11AuthModeWPA2PSK))
1227                 bWPA2 = TRUE;
1228
1229         // 0. Check MsgType
1230         if ((MsgType > EAPOL_GROUP_MSG_2) || (MsgType < EAPOL_PAIR_MSG_1))
1231         {
1232                 DBGPRINT(RT_DEBUG_ERROR, ("The message type is invalid(%d)! \n", MsgType));
1233                 return FALSE;
1234         }
1235
1236         // 1. Replay counter check
1237         if (MsgType == EAPOL_PAIR_MSG_1 || MsgType == EAPOL_PAIR_MSG_3 || MsgType == EAPOL_GROUP_MSG_1) // For supplicant
1238     {
1239         // First validate replay counter, only accept message with larger replay counter.
1240                 // Let equal pass, some AP start with all zero replay counter
1241                 UCHAR   ZeroReplay[LEN_KEY_DESC_REPLAY];
1242
1243         NdisZeroMemory(ZeroReplay, LEN_KEY_DESC_REPLAY);
1244                 if ((RTMPCompareMemory(pMsg->KeyDesc.ReplayCounter, pEntry->R_Counter, LEN_KEY_DESC_REPLAY) != 1) &&
1245                         (RTMPCompareMemory(pMsg->KeyDesc.ReplayCounter, ZeroReplay, LEN_KEY_DESC_REPLAY) != 0))
1246         {
1247                         bReplayDiff = TRUE;
1248         }
1249         }
1250         else if (MsgType == EAPOL_PAIR_MSG_2 || MsgType == EAPOL_PAIR_MSG_4 || MsgType == EAPOL_GROUP_MSG_2)    // For authenticator
1251         {
1252                 // check Replay Counter coresponds to MSG from authenticator, otherwise discard
1253         if (!NdisEqualMemory(pMsg->KeyDesc.ReplayCounter, pEntry->R_Counter, LEN_KEY_DESC_REPLAY))
1254         {
1255                         bReplayDiff = TRUE;
1256         }
1257         }
1258
1259         // Replay Counter different condition
1260         if (bReplayDiff)
1261         {
1262                 // send wireless event - for replay counter different
1263                 if (pAd->CommonCfg.bWirelessEvent)
1264                         RTMPSendWirelessEvent(pAd, IW_REPLAY_COUNTER_DIFF_EVENT_FLAG, pEntry->Addr, pEntry->apidx, 0);
1265
1266                 if (MsgType < EAPOL_GROUP_MSG_1)
1267                 {
1268                 DBGPRINT(RT_DEBUG_ERROR, ("Replay Counter Different in pairwise msg %d of 4-way handshake!\n", MsgType));
1269                 }
1270                 else
1271                 {
1272                         DBGPRINT(RT_DEBUG_ERROR, ("Replay Counter Different in group msg %d of 2-way handshake!\n", (MsgType - EAPOL_PAIR_MSG_4)));
1273                 }
1274
1275                 hex_dump("Receive replay counter ", pMsg->KeyDesc.ReplayCounter, LEN_KEY_DESC_REPLAY);
1276                 hex_dump("Current replay counter ", pEntry->R_Counter, LEN_KEY_DESC_REPLAY);
1277         return FALSE;
1278         }
1279
1280         // 2. Verify MIC except Pairwise Msg1
1281         if (MsgType != EAPOL_PAIR_MSG_1)
1282         {
1283                 UCHAR                   rcvd_mic[LEN_KEY_DESC_MIC];
1284
1285                 // Record the received MIC for check later
1286                 NdisMoveMemory(rcvd_mic, pMsg->KeyDesc.KeyMic, LEN_KEY_DESC_MIC);
1287                 NdisZeroMemory(pMsg->KeyDesc.KeyMic, LEN_KEY_DESC_MIC);
1288
1289         if (pEntry->WepStatus == Ndis802_11Encryption2Enabled)  // TKIP
1290         {
1291             hmac_md5(pEntry->PTK, LEN_EAP_MICK, (PUCHAR)pMsg, MsgLen, mic);
1292         }
1293         else if (pEntry->WepStatus == Ndis802_11Encryption3Enabled)     // AES
1294         {
1295             HMAC_SHA1((PUCHAR)pMsg, MsgLen, pEntry->PTK, LEN_EAP_MICK, digest);
1296             NdisMoveMemory(mic, digest, LEN_KEY_DESC_MIC);
1297         }
1298
1299         if (!NdisEqualMemory(rcvd_mic, mic, LEN_KEY_DESC_MIC))
1300         {
1301                         // send wireless event - for MIC different
1302                         if (pAd->CommonCfg.bWirelessEvent)
1303                                 RTMPSendWirelessEvent(pAd, IW_MIC_DIFF_EVENT_FLAG, pEntry->Addr, pEntry->apidx, 0);
1304
1305                         if (MsgType < EAPOL_GROUP_MSG_1)
1306                         {
1307                 DBGPRINT(RT_DEBUG_ERROR, ("MIC Different in pairwise msg %d of 4-way handshake!\n", MsgType));
1308                         }
1309                         else
1310                         {
1311                                 DBGPRINT(RT_DEBUG_ERROR, ("MIC Different in group msg %d of 2-way handshake!\n", (MsgType - EAPOL_PAIR_MSG_4)));
1312                         }
1313
1314                         hex_dump("Received MIC", rcvd_mic, LEN_KEY_DESC_MIC);
1315                         hex_dump("Desired  MIC", mic, LEN_KEY_DESC_MIC);
1316
1317                         return FALSE;
1318         }
1319         }
1320
1321         // Extract the context of the Key Data field if it exist
1322         // The field in pairwise_msg_2_WPA1(WPA2) & pairwise_msg_3_WPA1 is un-encrypted.
1323         // The field in group_msg_1_WPA1(WPA2) & pairwise_msg_3_WPA2 is encrypted.
1324         if (pMsg->KeyDesc.KeyDataLen[1] > 0)
1325         {
1326                 // Decrypt this field
1327                 if ((MsgType == EAPOL_PAIR_MSG_3 && bWPA2) || (MsgType == EAPOL_GROUP_MSG_1))
1328                 {
1329                         if(pEntry->WepStatus == Ndis802_11Encryption3Enabled)
1330                         {
1331                                 // AES
1332                                 AES_GTK_KEY_UNWRAP(&pEntry->PTK[16], KEYDATA, pMsg->KeyDesc.KeyDataLen[1],pMsg->KeyDesc.KeyData);
1333                         }
1334                         else
1335                         {
1336                                 INT     i;
1337                                 UCHAR   Key[32];
1338                                 // Decrypt TKIP GTK
1339                                 // Construct 32 bytes RC4 Key
1340                                 NdisMoveMemory(Key, pMsg->KeyDesc.KeyIv, 16);
1341                                 NdisMoveMemory(&Key[16], &pEntry->PTK[16], 16);
1342                                 ARCFOUR_INIT(&pAd->PrivateInfo.WEPCONTEXT, Key, 32);
1343                                 //discard first 256 bytes
1344                                 for(i = 0; i < 256; i++)
1345                                         ARCFOUR_BYTE(&pAd->PrivateInfo.WEPCONTEXT);
1346                                 // Decrypt GTK. Becareful, there is no ICV to check the result is correct or not
1347                                 ARCFOUR_DECRYPT(&pAd->PrivateInfo.WEPCONTEXT, KEYDATA, pMsg->KeyDesc.KeyData, pMsg->KeyDesc.KeyDataLen[1]);
1348                         }
1349
1350                         if (!bWPA2 && (MsgType == EAPOL_GROUP_MSG_1))
1351                                 GroupKeyIndex = EapolKeyInfo.KeyIndex;
1352
1353                 }
1354                 else if ((MsgType == EAPOL_PAIR_MSG_2) || (MsgType == EAPOL_PAIR_MSG_3 && !bWPA2))
1355                 {
1356                         NdisMoveMemory(KEYDATA, pMsg->KeyDesc.KeyData, pMsg->KeyDesc.KeyDataLen[1]);
1357                 }
1358                 else
1359                 {
1360
1361                         return TRUE;
1362                 }
1363
1364                 // Parse Key Data field to
1365                 // 1. verify RSN IE for pairwise_msg_2_WPA1(WPA2) ,pairwise_msg_3_WPA1(WPA2)
1366                 // 2. verify KDE format for pairwise_msg_3_WPA2, group_msg_1_WPA2
1367                 // 3. update shared key for pairwise_msg_3_WPA2, group_msg_1_WPA1(WPA2)
1368                 if (!RTMPParseEapolKeyData(pAd, KEYDATA, pMsg->KeyDesc.KeyDataLen[1], GroupKeyIndex, MsgType, bWPA2, pEntry))
1369                 {
1370                         return FALSE;
1371                 }
1372         }
1373
1374         return TRUE;
1375
1376 }