drivers/staging/rt2860/sta/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 extern UCHAR    CISCO_OUI[];
  40
  41 extern UCHAR    WPA_OUI[];
  42 extern UCHAR    RSN_OUI[];
  43 extern UCHAR    WME_INFO_ELEM[];
  44 extern UCHAR    WME_PARM_ELEM[];
  45 extern UCHAR    Ccx2QosInfo[];
  46 extern UCHAR    RALINK_OUI[];
  47 extern UCHAR    BROADCOM_OUI[];
  48
  49 /*
  50     ==========================================================================
  51     Description:
  52         MLME message sanity check
  53     Return:
  54         TRUE if all parameters are OK, FALSE otherwise
  55     ==========================================================================
  56  */
  57 BOOLEAN MlmeStartReqSanity(
  58     IN PRTMP_ADAPTER pAd,
  59     IN VOID *Msg,
  60     IN ULONG MsgLen,
  61     OUT CHAR Ssid[],
  62     OUT UCHAR *pSsidLen)
  63 {
  64     MLME_START_REQ_STRUCT *Info;
  65
  66     Info = (MLME_START_REQ_STRUCT *)(Msg);
  67
  68     if (Info->SsidLen > MAX_LEN_OF_SSID)
  69     {
  70         DBGPRINT(RT_DEBUG_TRACE, ("MlmeStartReqSanity fail - wrong SSID length\n"));
  71         return FALSE;
  72     }
  73
  74     *pSsidLen = Info->SsidLen;
  75     NdisMoveMemory(Ssid, Info->Ssid, *pSsidLen);
  76
  77     return TRUE;
  78 }
  79
  80 /*
  81     ==========================================================================
  82     Description:
  83         MLME message sanity check
  84     Return:
  85         TRUE if all parameters are OK, FALSE otherwise
  86
  87     IRQL = DISPATCH_LEVEL
  88
  89     ==========================================================================
  90  */
  91 BOOLEAN PeerAssocRspSanity(
  92     IN PRTMP_ADAPTER pAd,
  93     IN VOID *pMsg,
  94     IN ULONG MsgLen,
  95     OUT PUCHAR pAddr2,
  96     OUT USHORT *pCapabilityInfo,
  97     OUT USHORT *pStatus,
  98     OUT USHORT *pAid,
  99     OUT UCHAR SupRate[],
 100     OUT UCHAR *pSupRateLen,
 101     OUT UCHAR ExtRate[],
 102     OUT UCHAR *pExtRateLen,
 103     OUT HT_CAPABILITY_IE                *pHtCapability,
 104     OUT ADD_HT_INFO_IE          *pAddHtInfo,    // AP might use this additional ht info IE
 105     OUT UCHAR                   *pHtCapabilityLen,
 106     OUT UCHAR                   *pAddHtInfoLen,
 107     OUT UCHAR                   *pNewExtChannelOffset,
 108     OUT PEDCA_PARM pEdcaParm,
 109     OUT UCHAR *pCkipFlag)
 110 {
 111     CHAR          IeType, *Ptr;
 112     PFRAME_802_11 pFrame = (PFRAME_802_11)pMsg;
 113     PEID_STRUCT   pEid;
 114     ULONG         Length = 0;
 115
 116         *pNewExtChannelOffset = 0xff;
 117         *pHtCapabilityLen = 0;
 118         *pAddHtInfoLen = 0;
 119     COPY_MAC_ADDR(pAddr2, pFrame->Hdr.Addr2);
 120     Ptr = (PCHAR)pFrame->Octet;
 121     Length += LENGTH_802_11;
 122
 123     NdisMoveMemory(pCapabilityInfo, &pFrame->Octet[0], 2);
 124     Length += 2;
 125     NdisMoveMemory(pStatus,         &pFrame->Octet[2], 2);
 126     Length += 2;
 127     *pCkipFlag = 0;
 128     *pExtRateLen = 0;
 129     pEdcaParm->bValid = FALSE;
 130
 131     if (*pStatus != MLME_SUCCESS)
 132         return TRUE;
 133
 134     NdisMoveMemory(pAid, &pFrame->Octet[4], 2);
 135     Length += 2;
 136
 137     // Aid already swaped byte order in RTMPFrameEndianChange() for big endian platform
 138     *pAid = (*pAid) & 0x3fff; // AID is low 14-bit
 139
 140     // -- get supported rates from payload and advance the pointer
 141     IeType = pFrame->Octet[6];
 142     *pSupRateLen = pFrame->Octet[7];
 143     if ((IeType != IE_SUPP_RATES) || (*pSupRateLen > MAX_LEN_OF_SUPPORTED_RATES))
 144     {
 145         DBGPRINT(RT_DEBUG_TRACE, ("PeerAssocRspSanity fail - wrong SupportedRates IE\n"));
 146         return FALSE;
 147     }
 148     else
 149         NdisMoveMemory(SupRate, &pFrame->Octet[8], *pSupRateLen);
 150
 151     Length = Length + 2 + *pSupRateLen;
 152
 153     // many AP implement proprietary IEs in non-standard order, we'd better
 154     // tolerate mis-ordered IEs to get best compatibility
 155     pEid = (PEID_STRUCT) &pFrame->Octet[8 + (*pSupRateLen)];
 156
 157     // get variable fields from payload and advance the pointer
 158     while ((Length + 2 + pEid->Len) <= MsgLen)
 159     {
 160         switch (pEid->Eid)
 161         {
 162             case IE_EXT_SUPP_RATES:
 163                 if (pEid->Len <= MAX_LEN_OF_SUPPORTED_RATES)
 164                 {
 165                     NdisMoveMemory(ExtRate, pEid->Octet, pEid->Len);
 166                     *pExtRateLen = pEid->Len;
 167                 }
 168                 break;
 169
 170              case IE_HT_CAP:
 171             case IE_HT_CAP2:
 172                         if (pEid->Len >= SIZE_HT_CAP_IE)  //Note: allow extension.!!
 173                         {
 174                                 NdisMoveMemory(pHtCapability, pEid->Octet, SIZE_HT_CAP_IE);
 175
 176                                 *(USHORT *)(&pHtCapability->HtCapInfo) = cpu2le16(*(USHORT *)(&pHtCapability->HtCapInfo));
 177                                 *(USHORT *)(&pHtCapability->ExtHtCapInfo) = cpu2le16(*(USHORT *)(&pHtCapability->ExtHtCapInfo));
 178
 179                                 *pHtCapabilityLen = SIZE_HT_CAP_IE;
 180                         }
 181                         else
 182                         {
 183                                 DBGPRINT(RT_DEBUG_WARN, ("PeerAssocRspSanity - wrong IE_HT_CAP. \n"));
 184                         }
 185
 186                 break;
 187             case IE_ADD_HT:
 188             case IE_ADD_HT2:
 189                         if (pEid->Len >= sizeof(ADD_HT_INFO_IE))
 190                         {
 191                                 // This IE allows extension, but we can ignore extra bytes beyond our knowledge , so only
 192                                 // copy first sizeof(ADD_HT_INFO_IE)
 193                                 NdisMoveMemory(pAddHtInfo, pEid->Octet, sizeof(ADD_HT_INFO_IE));
 194
 195                                 *(USHORT *)(&pAddHtInfo->AddHtInfo2) = cpu2le16(*(USHORT *)(&pAddHtInfo->AddHtInfo2));
 196                                 *(USHORT *)(&pAddHtInfo->AddHtInfo3) = cpu2le16(*(USHORT *)(&pAddHtInfo->AddHtInfo3));
 197
 198                                 *pAddHtInfoLen = SIZE_ADD_HT_INFO_IE;
 199                         }
 200                         else
 201                         {
 202                                 DBGPRINT(RT_DEBUG_WARN, ("PeerAssocRspSanity - wrong IE_ADD_HT. \n"));
 203                         }
 204
 205                 break;
 206             case IE_SECONDARY_CH_OFFSET:
 207                         if (pEid->Len == 1)
 208                         {
 209                                 *pNewExtChannelOffset = pEid->Octet[0];
 210                         }
 211                         else
 212                         {
 213                                 DBGPRINT(RT_DEBUG_WARN, ("PeerAssocRspSanity - wrong IE_SECONDARY_CH_OFFSET. \n"));
 214                         }
 215                 break;
 216
 217             case IE_VENDOR_SPECIFIC:
 218                 // handle WME PARAMTER ELEMENT
 219                 if (NdisEqualMemory(pEid->Octet, WME_PARM_ELEM, 6) && (pEid->Len == 24))
 220                 {
 221                     PUCHAR ptr;
 222                     int i;
 223
 224                     // parsing EDCA parameters
 225                     pEdcaParm->bValid          = TRUE;
 226                     pEdcaParm->bQAck           = FALSE; // pEid->Octet[0] & 0x10;
 227                     pEdcaParm->bQueueRequest   = FALSE; // pEid->Octet[0] & 0x20;
 228                     pEdcaParm->bTxopRequest    = FALSE; // pEid->Octet[0] & 0x40;
 229                     //pEdcaParm->bMoreDataAck    = FALSE; // pEid->Octet[0] & 0x80;
 230                     pEdcaParm->EdcaUpdateCount = pEid->Octet[6] & 0x0f;
 231                     pEdcaParm->bAPSDCapable    = (pEid->Octet[6] & 0x80) ? 1 : 0;
 232                     ptr = (PUCHAR)&pEid->Octet[8];
 233                     for (i=0; i<4; i++)
 234                     {
 235                         UCHAR aci = (*ptr & 0x60) >> 5; // b5~6 is AC INDEX
 236                         pEdcaParm->bACM[aci]  = (((*ptr) & 0x10) == 0x10);   // b5 is ACM
 237                         pEdcaParm->Aifsn[aci] = (*ptr) & 0x0f;               // b0~3 is AIFSN
 238                         pEdcaParm->Cwmin[aci] = *(ptr+1) & 0x0f;             // b0~4 is Cwmin
 239                         pEdcaParm->Cwmax[aci] = *(ptr+1) >> 4;               // b5~8 is Cwmax
 240                         pEdcaParm->Txop[aci]  = *(ptr+2) + 256 * (*(ptr+3)); // in unit of 32-us
 241                         ptr += 4; // point to next AC
 242                     }
 243                 }
 244                 break;
 245             default:
 246                 DBGPRINT(RT_DEBUG_TRACE, ("PeerAssocRspSanity - ignore unrecognized EID = %d\n", pEid->Eid));
 247                 break;
 248         }
 249
 250         Length = Length + 2 + pEid->Len;
 251         pEid = (PEID_STRUCT)((UCHAR*)pEid + 2 + pEid->Len);
 252     }
 253
 254
 255     return TRUE;
 256 }
 257
 258 /*
 259     ==========================================================================
 260     Description:
 261         MLME message sanity check
 262     Return:
 263         TRUE if all parameters are OK, FALSE otherwise
 264
 265         IRQL = DISPATCH_LEVEL
 266
 267     ==========================================================================
 268  */
 269 BOOLEAN PeerProbeReqSanity(
 270     IN PRTMP_ADAPTER pAd,
 271     IN VOID *Msg,
 272     IN ULONG MsgLen,
 273     OUT PUCHAR pAddr2,
 274     OUT CHAR Ssid[],
 275     OUT UCHAR *pSsidLen)
 276 {
 277     UCHAR         Idx;
 278     UCHAR             RateLen;
 279     CHAR          IeType;
 280     PFRAME_802_11 pFrame = (PFRAME_802_11)Msg;
 281
 282     COPY_MAC_ADDR(pAddr2, pFrame->Hdr.Addr2);
 283
 284     if ((pFrame->Octet[0] != IE_SSID) || (pFrame->Octet[1] > MAX_LEN_OF_SSID))
 285     {
 286         DBGPRINT(RT_DEBUG_TRACE, ("PeerProbeReqSanity fail - wrong SSID IE(Type=%d,Len=%d)\n",pFrame->Octet[0],pFrame->Octet[1]));
 287         return FALSE;
 288     }
 289
 290     *pSsidLen = pFrame->Octet[1];
 291     NdisMoveMemory(Ssid, &pFrame->Octet[2], *pSsidLen);
 292
 293     Idx = *pSsidLen + 2;
 294
 295     // -- get supported rates from payload and advance the pointer
 296     IeType = pFrame->Octet[Idx];
 297     RateLen = pFrame->Octet[Idx + 1];
 298     if (IeType != IE_SUPP_RATES)
 299     {
 300         DBGPRINT(RT_DEBUG_TRACE, ("PeerProbeReqSanity fail - wrong SupportRates IE(Type=%d,Len=%d)\n",pFrame->Octet[Idx],pFrame->Octet[Idx+1]));
 301         return FALSE;
 302     }
 303     else
 304     {
 305         if ((pAd->CommonCfg.PhyMode == PHY_11G) && (RateLen < 8))
 306             return (FALSE);
 307     }
 308
 309     return TRUE;
 310 }
 311
 312 /*
 313     ==========================================================================
 314     Description:
 315
 316         IRQL = DISPATCH_LEVEL
 317
 318     ==========================================================================
 319  */
 320 BOOLEAN GetTimBit(
 321     IN CHAR *Ptr,
 322     IN USHORT Aid,
 323     OUT UCHAR *TimLen,
 324     OUT UCHAR *BcastFlag,
 325     OUT UCHAR *DtimCount,
 326     OUT UCHAR *DtimPeriod,
 327     OUT UCHAR *MessageToMe)
 328 {
 329     UCHAR          BitCntl, N1, N2, MyByte, MyBit;
 330     CHAR          *IdxPtr;
 331
 332     IdxPtr = Ptr;
 333
 334     IdxPtr ++;
 335     *TimLen = *IdxPtr;
 336
 337     // get DTIM Count from TIM element
 338     IdxPtr ++;
 339     *DtimCount = *IdxPtr;
 340
 341     // get DTIM Period from TIM element
 342     IdxPtr++;
 343     *DtimPeriod = *IdxPtr;
 344
 345     // get Bitmap Control from TIM element
 346     IdxPtr++;
 347     BitCntl = *IdxPtr;
 348
 349     if ((*DtimCount == 0) && (BitCntl & 0x01))
 350         *BcastFlag = TRUE;
 351     else
 352         *BcastFlag = FALSE;
 353
 354     // Parse Partial Virtual Bitmap from TIM element
 355     N1 = BitCntl & 0xfe;    // N1 is the first bitmap byte#
 356     N2 = *TimLen - 4 + N1;  // N2 is the last bitmap byte#
 357
 358     if ((Aid < (N1 << 3)) || (Aid >= ((N2 + 1) << 3)))
 359         *MessageToMe = FALSE;
 360     else
 361     {
 362         MyByte = (Aid >> 3) - N1;                       // my byte position in the bitmap byte-stream
 363         MyBit = Aid % 16 - ((MyByte & 0x01)? 8:0);
 364
 365         IdxPtr += (MyByte + 1);
 366
 367         //if (*IdxPtr)
 368         //    DBGPRINT(RT_DEBUG_WARN, ("TIM bitmap = 0x%02x\n", *IdxPtr));
 369
 370         if (*IdxPtr & (0x01 << MyBit))
 371             *MessageToMe = TRUE;
 372         else
 373             *MessageToMe = FALSE;
 374     }
 375
 376     return TRUE;
 377 }
 378