net: wireless: rockchip_wlan: add rtl8188fu support
[firefly-linux-kernel-4.4.55.git] / drivers / net / wireless / rockchip_wlan / rtl8188fu / core / rtw_wapi.c
diff --git a/drivers/net/wireless/rockchip_wlan/rtl8188fu/core/rtw_wapi.c b/drivers/net/wireless/rockchip_wlan/rtl8188fu/core/rtw_wapi.c
new file mode 100644 (file)
index 0000000..d4835da
--- /dev/null
@@ -0,0 +1,1326 @@
+#ifdef CONFIG_WAPI_SUPPORT\r
+\r
+#include <linux/unistd.h>\r
+#include <linux/etherdevice.h>\r
+#include <drv_types.h>\r
+#include <rtw_wapi.h>\r
+\r
+\r
+u32 wapi_debug_component =\r
+//                             WAPI_INIT       |\r
+//                             WAPI_API        |\r
+//                             WAPI_TX         |\r
+//                             WAPI_RX         |\r
+                               WAPI_ERR ; //always open err flags on\r
+\r
+void WapiFreeAllStaInfo(_adapter *padapter)\r
+{\r
+       PRT_WAPI_T                              pWapiInfo;\r
+       PRT_WAPI_STA_INFO               pWapiStaInfo;\r
+       PRT_WAPI_BKID                   pWapiBkid;\r
+\r
+       WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);\r
+       pWapiInfo = &padapter->wapiInfo;\r
+\r
+       //Pust to Idle List\r
+       rtw_wapi_return_all_sta_info(padapter);\r
+\r
+       //Sta Info List\r
+       while(!list_empty(&(pWapiInfo->wapiSTAIdleList)))\r
+       {\r
+               pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAIdleList.next, RT_WAPI_STA_INFO, list);\r
+               list_del_init(&pWapiStaInfo->list);\r
+       }\r
+\r
+       //BKID List\r
+       while(!list_empty(&(pWapiInfo->wapiBKIDIdleList)))\r
+       {\r
+               pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDIdleList.next, RT_WAPI_BKID, list);\r
+               list_del_init(&pWapiBkid->list);\r
+       }\r
+       WAPI_TRACE(WAPI_INIT, "<=========== %s\n", __FUNCTION__);\r
+       return;\r
+}\r
+\r
+void WapiSetIE(_adapter *padapter)\r
+{\r
+       PRT_WAPI_T              pWapiInfo = &(padapter->wapiInfo);\r
+       //PRT_WAPI_BKID pWapiBkid;\r
+       u16             protocolVer = 1;\r
+       u16             akmCnt = 1;\r
+       u16             suiteCnt = 1;\r
+       u16             capability = 0;\r
+       u8              OUI[3];\r
+\r
+       OUI[0] = 0x00;\r
+       OUI[1] = 0x14;\r
+       OUI[2] = 0x72;\r
+\r
+       pWapiInfo->wapiIELength = 0;\r
+//protocol version\r
+       memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength, &protocolVer, 2);\r
+       pWapiInfo->wapiIELength +=2;\r
+//akm\r
+       memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength, &akmCnt, 2);\r
+       pWapiInfo->wapiIELength +=2;\r
+\r
+       if(pWapiInfo->bWapiPSK){\r
+               memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength,OUI, 3);\r
+               pWapiInfo->wapiIELength +=3;\r
+               pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x2;\r
+               pWapiInfo->wapiIELength +=1;\r
+       }else{\r
+               memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength,OUI, 3);\r
+               pWapiInfo->wapiIELength +=3;\r
+               pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x1;\r
+               pWapiInfo->wapiIELength +=1;\r
+       }\r
+\r
+//usk\r
+       memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength, &suiteCnt, 2);\r
+       pWapiInfo->wapiIELength +=2;\r
+       memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength,OUI, 3);\r
+       pWapiInfo->wapiIELength +=3;\r
+       pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x1;\r
+       pWapiInfo->wapiIELength +=1;\r
+\r
+//msk\r
+       memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength,OUI, 3);\r
+       pWapiInfo->wapiIELength +=3;\r
+       pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x1;\r
+       pWapiInfo->wapiIELength +=1;\r
+\r
+//Capbility\r
+       memcpy(pWapiInfo->wapiIE+pWapiInfo->wapiIELength, &capability, 2);\r
+       pWapiInfo->wapiIELength +=2;\r
+}\r
+\r
+\r
+/*  PN1 > PN2, return 1,\r
+ *  else return 0.\r
+ */\r
+u32 WapiComparePN(u8 *PN1, u8 *PN2)\r
+{\r
+       char i;\r
+\r
+       if ((NULL == PN1) || (NULL == PN2))\r
+               return 1;\r
+\r
+       // overflow case\r
+       if ((PN2[15] - PN1[15]) & 0x80)\r
+               return 1;\r
+\r
+       for (i=16; i>0; i--)\r
+       {\r
+               if(PN1[i-1] == PN2[i-1])\r
+                       continue;\r
+               else if(PN1[i-1] > PN2[i-1])\r
+                       return 1;\r
+               else\r
+                       return 0;\r
+       }\r
+\r
+       return 0;\r
+}\r
+\r
+u8\r
+WapiGetEntryForCamWrite(_adapter *padapter,u8 *pMacAddr,u8 KID,BOOLEAN IsMsk)\r
+{\r
+       PRT_WAPI_T              pWapiInfo=NULL;\r
+       //PRT_WAPI_CAM_ENTRY    pEntry=NULL;\r
+       u8 i=0;\r
+       u8 ret = 0xff;\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       pWapiInfo =  &padapter->wapiInfo;\r
+\r
+       //exist?\r
+       for(i=0;i<WAPI_CAM_ENTRY_NUM;i++)\r
+       {\r
+               if(pWapiInfo->wapiCamEntry[i].IsUsed\r
+                       && (_rtw_memcmp(pMacAddr, pWapiInfo->wapiCamEntry[i].PeerMacAddr, ETH_ALEN) == _TRUE)\r
+                       && pWapiInfo->wapiCamEntry[i].keyidx == KID\r
+                       && pWapiInfo->wapiCamEntry[i].type == IsMsk)\r
+               {\r
+                       ret = pWapiInfo->wapiCamEntry[i].entry_idx; //cover it\r
+                       break;\r
+               }\r
+       }\r
+\r
+       if(i == WAPI_CAM_ENTRY_NUM) //not found\r
+       {\r
+               for(i=0;i<WAPI_CAM_ENTRY_NUM;i++)\r
+               {\r
+                       if(pWapiInfo->wapiCamEntry[i].IsUsed == 0)\r
+                       {\r
+                               pWapiInfo->wapiCamEntry[i].IsUsed = 1;\r
+                               pWapiInfo->wapiCamEntry[i].type = IsMsk;\r
+                               pWapiInfo->wapiCamEntry[i].keyidx = KID;\r
+                               _rtw_memcpy(pWapiInfo->wapiCamEntry[i].PeerMacAddr, pMacAddr,ETH_ALEN);\r
+                               ret = pWapiInfo->wapiCamEntry[i].entry_idx;\r
+                               break;\r
+                       }\r
+               }\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);\r
+       return ret;\r
+\r
+/*\r
+       if(RTIsListEmpty(&pWapiInfo->wapiCamIdleList)){\r
+               RT_TRACE(COMP_SEC,DBG_LOUD,("No Entry for wapi!!!\n"));\r
+               return 0;\r
+       }\r
+\r
+       pEntry = (PRT_WAPI_CAM_ENTRY)RTRemoveHeadList(&pWapiInfo->wapiCamIdleList);\r
+       RTInsertTailList(&pWapiInfo->wapiCamUsedList, &pEntry->list);\r
+\r
+       RT_TRACE(COMP_SEC,DBG_LOUD,("<====WapiGetCamEntry(),Get Entry Idx:%d.but we just return 4 for test\n",pEntry->entry_idx));\r
+\r
+       return pEntry->entry_idx;*/\r
+}\r
+\r
+u8 WapiGetEntryForCamClear(_adapter *padapter,u8 *pPeerMac,u8 keyid,u8 IsMsk)\r
+{\r
+       PRT_WAPI_T              pWapiInfo=NULL;\r
+       u8              i=0;\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       pWapiInfo =  &padapter->wapiInfo;\r
+\r
+       for(i=0;i<WAPI_CAM_ENTRY_NUM;i++)\r
+       {\r
+               if(pWapiInfo->wapiCamEntry[i].IsUsed\r
+                       && (_rtw_memcmp(pPeerMac, pWapiInfo->wapiCamEntry[i].PeerMacAddr, ETH_ALEN) == _TRUE)\r
+                       && pWapiInfo->wapiCamEntry[i].keyidx == keyid\r
+                       && pWapiInfo->wapiCamEntry[i].type == IsMsk)\r
+               {\r
+                               pWapiInfo->wapiCamEntry[i].IsUsed = 0;\r
+                               pWapiInfo->wapiCamEntry[i].keyidx = 2;\r
+                               _rtw_memset(pWapiInfo->wapiCamEntry[i].PeerMacAddr,0,ETH_ALEN);\r
+\r
+                               WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);\r
+                               return pWapiInfo->wapiCamEntry[i].entry_idx;\r
+               }\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_API,"<====WapiGetReturnCamEntry(), No this cam entry.\n");\r
+       return 0xff;\r
+/*\r
+       if(RTIsListEmpty(&pWapiInfo->wapiCamUsedList)){\r
+               RT_TRACE(COMP_SEC,DBG_LOUD,("No Entry for wapi!!!\n"));\r
+               return FALSE;\r
+       }\r
+\r
+       pList = &pWapiInfo->wapiCamUsedList;\r
+       while(pList->Flink != &pWapiInfo->wapiCamUsedList)\r
+       {\r
+               pEntry = (PRT_WAPI_CAM_ENTRY)pList->Flink;\r
+               if(PlatformCompareMemory(pPeerMac,pEntry->PeerMacAddr, ETHER_ADDRLEN)== 0\r
+                       && keyid == pEntry->keyidx)\r
+               {\r
+                       RTRemoveEntryList(pList);\r
+                       RTInsertHeadList(&pWapiInfo->wapiCamIdleList, pList);\r
+                       return pEntry->entry_idx;\r
+               }\r
+               pList = pList->Flink;\r
+       }\r
+\r
+       return 0;\r
+*/\r
+}\r
+\r
+void\r
+WapiResetAllCamEntry(_adapter *padapter)\r
+{\r
+       PRT_WAPI_T              pWapiInfo;\r
+       int                             i;\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       pWapiInfo =  &padapter->wapiInfo;\r
+\r
+       for (i=0;i<WAPI_CAM_ENTRY_NUM;i++)\r
+       {\r
+               _rtw_memset(pWapiInfo->wapiCamEntry[i].PeerMacAddr, 0, ETH_ALEN);\r
+               pWapiInfo->wapiCamEntry[i].IsUsed = 0;\r
+               pWapiInfo->wapiCamEntry[i].keyidx = 2; //invalid\r
+               pWapiInfo->wapiCamEntry[i].entry_idx = 4+i*2;\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);\r
+\r
+       return;\r
+}\r
+\r
+u8 WapiWriteOneCamEntry(\r
+       _adapter        *padapter,\r
+       u8                      *pMacAddr,\r
+       u8                      KeyId,\r
+       u8                      EntryId,\r
+       u8                      EncAlg,\r
+       u8                      bGroupKey,\r
+       u8                      *pKey\r
+)\r
+{\r
+       u8 retVal = 0;\r
+       u16 usConfig = 0;\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       if(EntryId >= 32)\r
+       {\r
+               WAPI_TRACE(WAPI_ERR, "<=== CamAddOneEntry(): ulKeyId exceed!\n");\r
+               return retVal;\r
+       }\r
+\r
+       usConfig=usConfig|(0x01<<15)|((u16)(EncAlg)<<2)|(KeyId);\r
+\r
+       if(EncAlg == _SMS4_ )\r
+       {\r
+               if(bGroupKey == 1)\r
+                       usConfig |= (0x01<<6);\r
+               if((EntryId % 2)==1)  // ==0 sec key; == 1mic key\r
+                       usConfig |= (0x01<<5);\r
+       }\r
+\r
+       write_cam(padapter, EntryId, usConfig, pMacAddr, pKey);\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+       return 1;\r
+}\r
+\r
+void rtw_wapi_init(_adapter *padapter)\r
+{\r
+       PRT_WAPI_T              pWapiInfo;\r
+       int                             i;\r
+\r
+       WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);\r
+       RT_ASSERT_RET(padapter);\r
+\r
+       if (!padapter->WapiSupport)\r
+       {\r
+               WAPI_TRACE(WAPI_INIT, "<========== %s, WAPI not supported!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       pWapiInfo =  &padapter->wapiInfo;\r
+       pWapiInfo->bWapiEnable = false;\r
+\r
+       //Init BKID List\r
+       INIT_LIST_HEAD(&pWapiInfo->wapiBKIDIdleList);\r
+       INIT_LIST_HEAD(&pWapiInfo->wapiBKIDStoreList);\r
+       for(i=0;i<WAPI_MAX_BKID_NUM;i++)\r
+       {\r
+               list_add_tail(&pWapiInfo->wapiBKID[i].list, &pWapiInfo->wapiBKIDIdleList);\r
+       }\r
+\r
+       //Init STA List\r
+       INIT_LIST_HEAD(&pWapiInfo->wapiSTAIdleList);\r
+       INIT_LIST_HEAD(&pWapiInfo->wapiSTAUsedList);\r
+       for(i=0;i<WAPI_MAX_STAINFO_NUM;i++)\r
+       {\r
+               list_add_tail(&pWapiInfo->wapiSta[i].list, &pWapiInfo->wapiSTAIdleList);\r
+       }\r
+\r
+        for (i=0;i<WAPI_CAM_ENTRY_NUM;i++)\r
+        {\r
+               pWapiInfo->wapiCamEntry[i].IsUsed = 0;\r
+               pWapiInfo->wapiCamEntry[i].keyidx = 2; //invalid\r
+               pWapiInfo->wapiCamEntry[i].entry_idx = 4+i*2;\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_INIT, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_wapi_free(_adapter *padapter)\r
+{\r
+       WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);\r
+       RT_ASSERT_RET(padapter);\r
+\r
+       if (!padapter->WapiSupport)\r
+       {\r
+               WAPI_TRACE(WAPI_INIT, "<========== %s, WAPI not supported!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       WapiFreeAllStaInfo(padapter);\r
+\r
+       WAPI_TRACE(WAPI_INIT, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_wapi_disable_tx(_adapter *padapter)\r
+{\r
+       WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);\r
+       RT_ASSERT_RET(padapter);\r
+\r
+       if (!padapter->WapiSupport)\r
+       {\r
+               WAPI_TRACE(WAPI_INIT, "<========== %s, WAPI not supported!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       padapter->wapiInfo.wapiTxMsk.bTxEnable = false;\r
+       padapter->wapiInfo.wapiTxMsk.bSet = false;\r
+\r
+       WAPI_TRACE(WAPI_INIT, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+u8 rtw_wapi_is_wai_packet(_adapter* padapter,u8 *pkt_data)\r
+{\r
+       PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);\r
+       struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;\r
+       struct security_priv   *psecuritypriv = &padapter->securitypriv;\r
+       PRT_WAPI_STA_INFO pWapiSta = NULL;\r
+       u8 WaiPkt = 0, *pTaddr, bFind = false;\r
+       u8 Offset_TypeWAI = 0 ; // (mac header len + llc length)\r
+       \r
+       WAPI_TRACE(WAPI_TX|WAPI_RX, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return 0;\r
+       }\r
+\r
+        Offset_TypeWAI = 24 + 6 ;\r
+\r
+       //YJ,add,091103. Data frame may also have skb->data[30]=0x88 and skb->data[31]=0xb4.\r
+       if ((pkt_data[1]&0x40) !=0)\r
+       {\r
+               //DBG_871X("data is privacy \n");\r
+               return 0;\r
+       }\r
+       \r
+       pTaddr = GetAddr2Ptr(pkt_data);\r
+       if(list_empty(&pWapiInfo->wapiSTAUsedList)){\r
+               bFind = false;\r
+       }else{\r
+               list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list){\r
+                       if (_rtw_memcmp(pTaddr, pWapiSta->PeerMacAddr, 6) == _TRUE) {\r
+                               bFind = true;\r
+                               break;\r
+                       }\r
+               }\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_TX|WAPI_RX, "%s: bFind=%d pTaddr="MAC_FMT"\n", __FUNCTION__, bFind, MAC_ARG(pTaddr));\r
+\r
+       if (pkt_data[0] == WIFI_QOS_DATA_TYPE)\r
+       {\r
+               Offset_TypeWAI += 2;\r
+       }\r
+\r
+       // 88b4?\r
+       if( (pkt_data[Offset_TypeWAI]==0x88) && (pkt_data[Offset_TypeWAI+1]==0xb4) ){\r
+               WaiPkt = pkt_data[Offset_TypeWAI+5];\r
+\r
+               psecuritypriv->hw_decrypted = _TRUE;\r
+       }else{\r
+               WAPI_TRACE(WAPI_TX|WAPI_RX, "%s(): non wai packet\n",__FUNCTION__);\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_TX|WAPI_RX, "%s(): Recvd WAI frame. IsWAIPkt(%d)\n",__FUNCTION__, WaiPkt);\r
+\r
+       return  WaiPkt;\r
+}\r
+\r
+\r
+void rtw_wapi_update_info(_adapter *padapter, union recv_frame *precv_frame)\r
+{\r
+       PRT_WAPI_T     pWapiInfo = &(padapter->wapiInfo);\r
+       struct recv_frame_hdr *precv_hdr;\r
+       u8      *ptr;\r
+       u8      *pTA;\r
+       u8      *pRecvPN;\r
+\r
+\r
+       WAPI_TRACE(WAPI_RX, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_RX, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       precv_hdr = &precv_frame->u.hdr;\r
+       ptr = precv_hdr->rx_data;\r
+\r
+       if (precv_hdr->attrib.qos == 1)\r
+       {\r
+               precv_hdr->UserPriority = GetTid(ptr);\r
+       }\r
+       else\r
+       {\r
+               precv_hdr->UserPriority = 0;\r
+       }\r
+\r
+       pTA = GetAddr2Ptr(ptr);\r
+       _rtw_memcpy((u8 *)precv_hdr->WapiSrcAddr, pTA, 6);\r
+       pRecvPN = ptr + precv_hdr->attrib.hdrlen + 2;\r
+       _rtw_memcpy((u8 *)precv_hdr->WapiTempPN, pRecvPN, 16);\r
+\r
+       WAPI_TRACE(WAPI_RX, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+/****************************************************************************\r
+TRUE-----------------Drop\r
+FALSE---------------- handle\r
+add to support WAPI to N-mode\r
+*****************************************************************************/\r
+u8 rtw_wapi_check_for_drop(\r
+       _adapter *padapter,\r
+       union recv_frame *precv_frame\r
+)\r
+{\r
+       PRT_WAPI_T     pWapiInfo = &(padapter->wapiInfo);\r
+       u8                      *pLastRecvPN = NULL;\r
+       u8                      bFind = false;\r
+       PRT_WAPI_STA_INFO       pWapiSta = NULL;\r
+       u8                      bDrop = false;\r
+       struct recv_frame_hdr *precv_hdr = &precv_frame->u.hdr;\r
+       u8                                      WapiAEPNInitialValueSrc[16] = {0x37,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+       u8                                      WapiAEMultiCastPNInitialValueSrc[16] = {0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+       u8                                      *ptr = precv_frame->u.hdr.rx_data;\r
+       int                                     i;\r
+\r
+       WAPI_TRACE(WAPI_RX, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_RX, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return false;\r
+       }\r
+\r
+       if(precv_hdr->bIsWaiPacket !=0)\r
+       {\r
+               if(precv_hdr->bIsWaiPacket== 0x8)\r
+               {\r
+\r
+                       DBG_871X("rtw_wapi_check_for_drop: dump packet \n");\r
+                       for(i=0;i<50;i++)\r
+                       {\r
+                               DBG_871X("%02X  ",ptr[i]);\r
+                               if((i+1) %8 ==0)\r
+                                       DBG_871X("\n");\r
+                       }\r
+                       DBG_871X("\n rtw_wapi_check_for_drop: dump packet \n");\r
+\r
+                       for(i=0;i<16;i++)\r
+                       {\r
+                               if(ptr[i+27] !=0)\r
+                                       break;\r
+                       }\r
+\r
+                       if(i== 16)\r
+                       {\r
+                               WAPI_TRACE(WAPI_RX,"rtw_wapi_check_for_drop: drop with zero BKID \n");\r
+                               return true;\r
+                       }\r
+                       else\r
+                       {\r
+                               return false;\r
+                       }\r
+               }\r
+                       else\r
+                               return false;\r
+       }\r
+\r
+       if(list_empty(&pWapiInfo->wapiSTAUsedList)){\r
+               bFind = false;\r
+       }else{\r
+               list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {\r
+                       if (_rtw_memcmp(precv_hdr->WapiSrcAddr, pWapiSta->PeerMacAddr, ETH_ALEN) == _TRUE) {\r
+                               bFind = true;\r
+                               break;\r
+                       }\r
+               }\r
+       }\r
+       WAPI_TRACE(WAPI_RX, "%s: bFind=%d prxb->WapiSrcAddr="MAC_FMT"\n", __FUNCTION__, bFind, MAC_ARG(precv_hdr->WapiSrcAddr));\r
+\r
+       if(bFind)\r
+       {\r
+               if(IS_MCAST(precv_hdr->attrib.ra))\r
+               {\r
+                       WAPI_TRACE(WAPI_RX,"rtw_wapi_check_for_drop: multicast case \n");\r
+                       pLastRecvPN = pWapiSta->lastRxMulticastPN;\r
+               }\r
+               else\r
+               {\r
+                       WAPI_TRACE(WAPI_RX,"rtw_wapi_check_for_drop: unicast case \n");\r
+                       switch(precv_hdr->UserPriority)\r
+                       {\r
+                               case 0:\r
+                               case 3:\r
+                                       pLastRecvPN = pWapiSta->lastRxUnicastPNBEQueue;\r
+                                       break;\r
+                               case 1:\r
+                               case 2:\r
+                                       pLastRecvPN = pWapiSta->lastRxUnicastPNBKQueue;\r
+                                       break;\r
+                               case 4:\r
+                               case 5:\r
+                                       pLastRecvPN = pWapiSta->lastRxUnicastPNVIQueue;\r
+                                       break;\r
+                               case 6:\r
+                               case 7:\r
+                                       pLastRecvPN = pWapiSta->lastRxUnicastPNVOQueue;\r
+                                       break;\r
+                               default:\r
+                                       WAPI_TRACE(WAPI_ERR,"%s: Unknown TID \n",__FUNCTION__);\r
+                                       break;\r
+                       }\r
+               }\r
+\r
+               if(!WapiComparePN(precv_hdr->WapiTempPN,pLastRecvPN))\r
+               {\r
+                       WAPI_TRACE(WAPI_RX,"%s: Equal PN!!\n",__FUNCTION__);\r
+                       if(IS_MCAST(precv_hdr->attrib.ra))\r
+                               _rtw_memcpy(pLastRecvPN,WapiAEMultiCastPNInitialValueSrc,16);\r
+                       else\r
+                               _rtw_memcpy(pLastRecvPN,WapiAEPNInitialValueSrc,16);\r
+                       bDrop = true;\r
+               }\r
+               else\r
+               {\r
+                       _rtw_memcpy(pLastRecvPN,precv_hdr->WapiTempPN,16);\r
+               }\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_RX, "<========== %s\n", __FUNCTION__);\r
+       return bDrop;\r
+}\r
+\r
+void rtw_build_probe_resp_wapi_ie(_adapter *padapter, unsigned char *pframe, struct pkt_attrib *pattrib)\r
+{\r
+       PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);\r
+       u8 WapiIELength = 0;\r
+\r
+       WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport)  || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       WapiSetIE(padapter);\r
+       WapiIELength = pWapiInfo->wapiIELength;\r
+       pframe[0] = _WAPI_IE_;\r
+       pframe[1] = WapiIELength;\r
+       _rtw_memcpy(pframe+2, pWapiInfo->wapiIE, WapiIELength);\r
+       pframe += WapiIELength+2;\r
+       pattrib->pktlen += WapiIELength+2;\r
+\r
+       WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_build_beacon_wapi_ie(_adapter *padapter, unsigned char *pframe, struct pkt_attrib *pattrib)\r
+{\r
+       PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);\r
+       u8 WapiIELength = 0;\r
+       WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport)  || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       WapiSetIE(padapter);\r
+       WapiIELength = pWapiInfo->wapiIELength;\r
+       pframe[0] = _WAPI_IE_;\r
+       pframe[1] = WapiIELength;\r
+       _rtw_memcpy(pframe+2, pWapiInfo->wapiIE, WapiIELength);\r
+       pframe += WapiIELength+2;\r
+       pattrib->pktlen += WapiIELength+2;\r
+\r
+       WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_build_assoc_req_wapi_ie(_adapter *padapter, unsigned char *pframe, struct pkt_attrib *pattrib)\r
+{\r
+       PRT_WAPI_BKID           pWapiBKID;\r
+       u16                                     bkidNum;\r
+       PRT_WAPI_T                      pWapiInfo = &(padapter->wapiInfo);\r
+       u8                                      WapiIELength = 0;\r
+\r
+       WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       WapiSetIE(padapter);\r
+       WapiIELength = pWapiInfo->wapiIELength;\r
+       bkidNum = 0;\r
+       if(!list_empty(&(pWapiInfo->wapiBKIDStoreList))){\r
+               list_for_each_entry(pWapiBKID, &pWapiInfo->wapiBKIDStoreList, list) {\r
+                       bkidNum ++;\r
+                       _rtw_memcpy(pWapiInfo->wapiIE+WapiIELength+2, pWapiBKID->bkid,16);\r
+                       WapiIELength += 16;\r
+               }\r
+       }\r
+       _rtw_memcpy(pWapiInfo->wapiIE+WapiIELength, &bkidNum, 2);\r
+       WapiIELength += 2;\r
+\r
+        pframe[0] = _WAPI_IE_;\r
+        pframe[1] = WapiIELength;\r
+       _rtw_memcpy(pframe+2, pWapiInfo->wapiIE, WapiIELength);\r
+       pframe += WapiIELength+2;\r
+       pattrib->pktlen += WapiIELength+2;\r
+       WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_wapi_on_assoc_ok(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)\r
+{\r
+       PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);\r
+       PRT_WAPI_STA_INFO pWapiSta;\r
+       u8 WapiAEPNInitialValueSrc[16] = {0x37,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+       //u8 WapiASUEPNInitialValueSrc[16] = {0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+       u8 WapiAEMultiCastPNInitialValueSrc[16] = {0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+\r
+       WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       pWapiSta =(PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAIdleList.next, RT_WAPI_STA_INFO, list);\r
+       list_del_init(&pWapiSta->list);\r
+       list_add_tail(&pWapiSta->list, &pWapiInfo->wapiSTAUsedList);\r
+       _rtw_memcpy(pWapiSta->PeerMacAddr,padapter->mlmeextpriv.mlmext_info.network.MacAddress,6);\r
+       _rtw_memcpy(pWapiSta->lastRxMulticastPN, WapiAEMultiCastPNInitialValueSrc, 16);\r
+       _rtw_memcpy(pWapiSta->lastRxUnicastPN, WapiAEPNInitialValueSrc, 16);\r
+\r
+       //For chenk PN error with Qos Data after s3: add by ylb 20111114\r
+       _rtw_memcpy(pWapiSta->lastRxUnicastPNBEQueue,WapiAEPNInitialValueSrc,16);\r
+       _rtw_memcpy(pWapiSta->lastRxUnicastPNBKQueue,WapiAEPNInitialValueSrc,16);\r
+       _rtw_memcpy(pWapiSta->lastRxUnicastPNVIQueue,WapiAEPNInitialValueSrc,16);\r
+       _rtw_memcpy(pWapiSta->lastRxUnicastPNVOQueue,WapiAEPNInitialValueSrc,16);\r
+\r
+       WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+\r
+void rtw_wapi_return_one_sta_info(_adapter *padapter, u8 *MacAddr)\r
+{\r
+       PRT_WAPI_T                              pWapiInfo;\r
+       PRT_WAPI_STA_INFO               pWapiStaInfo = NULL;\r
+       PRT_WAPI_BKID                   pWapiBkid = NULL;\r
+       struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;\r
+\r
+       pWapiInfo = &padapter->wapiInfo;\r
+\r
+       WAPI_TRACE(WAPI_API, "==========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       if(check_fwstate(pmlmepriv, WIFI_STATION_STATE))\r
+       {\r
+             while(!list_empty(&(pWapiInfo->wapiBKIDStoreList)))\r
+                 {\r
+                        pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDStoreList.next, RT_WAPI_BKID, list);\r
+                           list_del_init(&pWapiBkid->list);\r
+                           _rtw_memset(pWapiBkid->bkid,0,16);\r
+                        list_add_tail(&pWapiBkid->list, &pWapiInfo->wapiBKIDIdleList);\r
+                 }\r
+        }\r
+\r
+\r
+       WAPI_TRACE(WAPI_API, " %s: after clear bkid \n", __FUNCTION__);\r
+\r
+\r
+       //Remove STA info\r
+       if(list_empty(&(pWapiInfo->wapiSTAUsedList))){\r
+               WAPI_TRACE(WAPI_API, " %s: wapiSTAUsedList is null \n", __FUNCTION__);\r
+               return;\r
+       }else{\r
+\r
+               WAPI_TRACE(WAPI_API, " %s: wapiSTAUsedList is not null \n", __FUNCTION__);\r
+#if 0\r
+               pWapiStaInfo=(PRT_WAPI_STA_INFO)list_entry((pWapiInfo->wapiSTAUsedList.next),RT_WAPI_STA_INFO,list);\r
+\r
+               list_for_each_entry(pWapiStaInfo, &(pWapiInfo->wapiSTAUsedList), list) {\r
+\r
+                       DBG_871X("MAC Addr %02x-%02x-%02x-%02x-%02x-%02x \n",MacAddr[0],MacAddr[1],MacAddr[2],MacAddr[3],MacAddr[4],MacAddr[5]);\r
+\r
+\r
+                       DBG_871X("peer Addr %02x-%02x-%02x-%02x-%02x-%02x \n",pWapiStaInfo->PeerMacAddr[0],pWapiStaInfo->PeerMacAddr[1],pWapiStaInfo->PeerMacAddr[2],pWapiStaInfo->PeerMacAddr[3],pWapiStaInfo->PeerMacAddr[4],pWapiStaInfo->PeerMacAddr[5]);\r
+\r
+                       if(pWapiStaInfo == NULL)\r
+                       {\r
+                               WAPI_TRACE(WAPI_API, " %s: pWapiStaInfo == NULL Case \n", __FUNCTION__);\r
+                               return;\r
+                       }\r
+\r
+                       if(pWapiStaInfo->PeerMacAddr == NULL)\r
+                       {\r
+                               WAPI_TRACE(WAPI_API, " %s: pWapiStaInfo->PeerMacAddr == NULL Case \n", __FUNCTION__);\r
+                               return;\r
+                       }\r
+\r
+                       if(MacAddr == NULL)\r
+                       {\r
+                               WAPI_TRACE(WAPI_API, " %s: MacAddr == NULL Case \n", __FUNCTION__);\r
+                               return;\r
+                       }\r
+\r
+                       if (_rtw_memcmp(pWapiStaInfo->PeerMacAddr, MacAddr, ETH_ALEN) == _TRUE) {\r
+                               pWapiStaInfo->bAuthenticateInProgress = false;\r
+                               pWapiStaInfo->bSetkeyOk = false;\r
+                               _rtw_memset(pWapiStaInfo->PeerMacAddr,0,ETH_ALEN);\r
+                               list_del_init(&pWapiStaInfo->list);\r
+                               list_add_tail(&pWapiStaInfo->list, &pWapiInfo->wapiSTAIdleList);\r
+                               break;\r
+                       }\r
+\r
+               }\r
+#endif\r
+\r
+               while(!list_empty(&(pWapiInfo->wapiSTAUsedList)))\r
+               {\r
+                       pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAUsedList.next, RT_WAPI_STA_INFO, list);\r
+\r
+                       DBG_871X("peer Addr %02x-%02x-%02x-%02x-%02x-%02x \n",pWapiStaInfo->PeerMacAddr[0],pWapiStaInfo->PeerMacAddr[1],pWapiStaInfo->PeerMacAddr[2],pWapiStaInfo->PeerMacAddr[3],pWapiStaInfo->PeerMacAddr[4],pWapiStaInfo->PeerMacAddr[5]);\r
+\r
+                       list_del_init(&pWapiStaInfo->list);\r
+                       memset(pWapiStaInfo->PeerMacAddr,0,ETH_ALEN);\r
+                       pWapiStaInfo->bSetkeyOk = 0;\r
+                       list_add_tail(&pWapiStaInfo->list, &pWapiInfo->wapiSTAIdleList);\r
+               }\r
+\r
+       }\r
+\r
+        WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);\r
+       return;\r
+}\r
+\r
+void rtw_wapi_return_all_sta_info(_adapter *padapter)\r
+{\r
+       PRT_WAPI_T                              pWapiInfo;\r
+       PRT_WAPI_STA_INFO               pWapiStaInfo;\r
+       PRT_WAPI_BKID                   pWapiBkid;\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       pWapiInfo = &padapter->wapiInfo;\r
+\r
+       if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       //Sta Info List\r
+       while(!list_empty(&(pWapiInfo->wapiSTAUsedList)))\r
+       {\r
+               pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAUsedList.next, RT_WAPI_STA_INFO, list);\r
+               list_del_init(&pWapiStaInfo->list);\r
+               memset(pWapiStaInfo->PeerMacAddr,0,ETH_ALEN);\r
+               pWapiStaInfo->bSetkeyOk = 0;\r
+               list_add_tail(&pWapiStaInfo->list, &pWapiInfo->wapiSTAIdleList);\r
+       }\r
+\r
+       //BKID List\r
+       while(!list_empty(&(pWapiInfo->wapiBKIDStoreList)))\r
+       {\r
+               pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDStoreList.next, RT_WAPI_BKID, list);\r
+               list_del_init(&pWapiBkid->list);\r
+               memset(pWapiBkid->bkid,0,16);\r
+               list_add_tail(&pWapiBkid->list, &pWapiInfo->wapiBKIDIdleList);\r
+       }\r
+       WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_wapi_clear_cam_entry(_adapter *padapter, u8 *pMacAddr)\r
+{\r
+       u8 UcIndex = 0;\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 0, 0);\r
+       if(UcIndex != 0xff){\r
+               //CAM_mark_invalid(Adapter, UcIndex);\r
+               CAM_empty_entry(padapter, UcIndex);\r
+       }\r
+\r
+       UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 1, 0);\r
+       if(UcIndex != 0xff){\r
+               //CAM_mark_invalid(Adapter, UcIndex);\r
+               CAM_empty_entry(padapter, UcIndex);\r
+       }\r
+\r
+       UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 0, 1);\r
+       if(UcIndex != 0xff){\r
+               //CAM_mark_invalid(Adapter, UcIndex);\r
+               CAM_empty_entry(padapter, UcIndex);\r
+       }\r
+\r
+       UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 1, 1);\r
+       if(UcIndex != 0xff){\r
+               //CAM_mark_invalid(padapter, UcIndex);\r
+               CAM_empty_entry(padapter, UcIndex);\r
+       }\r
+\r
+       WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_wapi_clear_all_cam_entry(_adapter *padapter)\r
+{\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       invalidate_cam_all(padapter); // is this ok?\r
+       WapiResetAllCamEntry(padapter);\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+}\r
+\r
+void rtw_wapi_set_key(_adapter *padapter, RT_WAPI_KEY *pWapiKey, RT_WAPI_STA_INFO *pWapiSta, u8 bGroupKey, u8 bUseDefaultKey)\r
+{\r
+       PRT_WAPI_T              pWapiInfo =  &padapter->wapiInfo;\r
+       u8                              *pMacAddr = pWapiSta->PeerMacAddr;\r
+       u32 EntryId = 0;\r
+       BOOLEAN IsPairWise = false ;\r
+       u8 EncAlgo;\r
+\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+       if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable))\r
+       {\r
+               WAPI_TRACE(WAPI_API, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);\r
+               return;\r
+       }\r
+\r
+       EncAlgo = _SMS4_;\r
+\r
+       //For Tx bc/mc pkt,use defualt key entry\r
+       if(bUseDefaultKey)\r
+       {\r
+               // when WAPI update key, keyid will be 0 or 1 by turns.\r
+               if (pWapiKey->keyId == 0)\r
+                       EntryId = 0;\r
+               else\r
+                       EntryId = 2;\r
+       }\r
+       else\r
+       {\r
+               // tx/rx unicast pkt, or rx broadcast, find the key entry by peer's MacAddr\r
+               EntryId = WapiGetEntryForCamWrite(padapter,pMacAddr,pWapiKey->keyId,bGroupKey);\r
+       }\r
+\r
+       if(EntryId == 0xff){\r
+               WAPI_TRACE(WAPI_API, "===>No entry for WAPI setkey! !!\n");\r
+               return;\r
+       }\r
+\r
+       //EntryId is also used to diff Sec key and Mic key\r
+       //Sec Key\r
+       WapiWriteOneCamEntry(padapter,\r
+                                       pMacAddr,\r
+                                       pWapiKey->keyId, //keyid\r
+                                       EntryId,        //entry\r
+                                       EncAlgo, //type\r
+                                       bGroupKey, //pairwise or group key\r
+                                       pWapiKey->dataKey);\r
+       //MIC key\r
+       WapiWriteOneCamEntry(padapter,\r
+                                       pMacAddr,\r
+                                       pWapiKey->keyId, //keyid\r
+                                       EntryId+1,      //entry\r
+                                       EncAlgo, //type\r
+                                       bGroupKey, //pairwise or group key\r
+                                       pWapiKey->micKey);\r
+\r
+       WAPI_TRACE(WAPI_API, "Set Wapi Key :KeyId:%d,EntryId:%d,PairwiseKey:%d.\n",pWapiKey->keyId,EntryId,!bGroupKey);\r
+       WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);\r
+\r
+}\r
+\r
+#if 0\r
+//YJ,test,091013\r
+void wapi_test_set_key(struct _adapter *padapter, u8* buf)\r
+{ /*Data: keyType(1) + bTxEnable(1) + bAuthenticator(1) + bUpdate(1) + PeerAddr(6) + DataKey(16) + MicKey(16) + KeyId(1)*/\r
+       PRT_WAPI_T                      pWapiInfo = &padapter->wapiInfo;\r
+       PRT_WAPI_BKID           pWapiBkid;\r
+       PRT_WAPI_STA_INFO       pWapiSta;\r
+       u8                                      data[43];\r
+       bool                                    bTxEnable;\r
+       bool                                    bUpdate;\r
+       bool                                    bAuthenticator;\r
+       u8                                      PeerAddr[6];\r
+       u8                                      WapiAEPNInitialValueSrc[16] = {0x37,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+       u8                                      WapiASUEPNInitialValueSrc[16] = {0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+       u8                                      WapiAEMultiCastPNInitialValueSrc[16] = {0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ;\r
+\r
+       WAPI_TRACE(WAPI_INIT, "===========>%s\n", __FUNCTION__);\r
+\r
+       if (!padapter->WapiSupport){\r
+           return;\r
+       }\r
+\r
+       copy_from_user(data, buf, 43);\r
+       bTxEnable = data[1];\r
+       bAuthenticator = data[2];\r
+       bUpdate = data[3];\r
+       memcpy(PeerAddr,data+4,6);\r
+\r
+       if(data[0] == 0x3){\r
+               if(!list_empty(&(pWapiInfo->wapiBKIDIdleList))){\r
+                       pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDIdleList.next, RT_WAPI_BKID, list);\r
+                       list_del_init(&pWapiBkid->list);\r
+                       memcpy(pWapiBkid->bkid, data+10, 16);\r
+                       WAPI_DATA(WAPI_INIT, "SetKey - BKID", pWapiBkid->bkid, 16);\r
+                       list_add_tail(&pWapiBkid->list, &pWapiInfo->wapiBKIDStoreList);\r
+               }\r
+       }else{\r
+               list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {\r
+                       if(!memcmp(pWapiSta->PeerMacAddr,PeerAddr,6)){\r
+                               pWapiSta->bAuthenticatorInUpdata = false;\r
+                               switch(data[0]){\r
+                               case 1:              //usk\r
+                                       if(bAuthenticator){         //authenticator\r
+                                               memcpy(pWapiSta->lastTxUnicastPN,WapiAEPNInitialValueSrc,16);\r
+                                               if(!bUpdate) {     //first\r
+                                                       WAPI_TRACE(WAPI_INIT,"AE fisrt set usk \n");\r
+                                                       pWapiSta->wapiUsk.bSet = true;\r
+                                                       memcpy(pWapiSta->wapiUsk.dataKey,data+10,16);\r
+                                                       memcpy(pWapiSta->wapiUsk.micKey,data+26,16);\r
+                                                       pWapiSta->wapiUsk.keyId = *(data+42);\r
+                                                       pWapiSta->wapiUsk.bTxEnable = true;\r
+                                                       WAPI_DATA(WAPI_INIT, "SetKey - AE USK Data Key", pWapiSta->wapiUsk.dataKey, 16);\r
+                                                       WAPI_DATA(WAPI_INIT, "SetKey - AE USK Mic Key", pWapiSta->wapiUsk.micKey, 16);\r
+                                               }\r
+                                               else               //update\r
+                                               {\r
+                                                       WAPI_TRACE(WAPI_INIT, "AE update usk \n");\r
+                                                       pWapiSta->wapiUskUpdate.bSet = true;\r
+                                                       pWapiSta->bAuthenticatorInUpdata = true;\r
+                                                       memcpy(pWapiSta->wapiUskUpdate.dataKey,data+10,16);\r
+                                                       memcpy(pWapiSta->wapiUskUpdate.micKey,data+26,16);\r
+                                                       memcpy(pWapiSta->lastRxUnicastPNBEQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                       memcpy(pWapiSta->lastRxUnicastPNBKQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                       memcpy(pWapiSta->lastRxUnicastPNVIQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                       memcpy(pWapiSta->lastRxUnicastPNVOQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                       memcpy(pWapiSta->lastRxUnicastPN,WapiASUEPNInitialValueSrc,16);\r
+                                                       pWapiSta->wapiUskUpdate.keyId = *(data+42);\r
+                                                       pWapiSta->wapiUskUpdate.bTxEnable = true;\r
+                                               }\r
+                                       }\r
+                                       else{\r
+                                               if(!bUpdate){\r
+                                                       WAPI_TRACE(WAPI_INIT,"ASUE fisrt set usk \n");\r
+                                                       if(bTxEnable){\r
+                                                               pWapiSta->wapiUsk.bTxEnable = true;\r
+                                                               memcpy(pWapiSta->lastTxUnicastPN,WapiASUEPNInitialValueSrc,16);\r
+                                                       }else{\r
+                                                               pWapiSta->wapiUsk.bSet = true;\r
+                                                               memcpy(pWapiSta->wapiUsk.dataKey,data+10,16);\r
+                                                               memcpy(pWapiSta->wapiUsk.micKey,data+26,16);\r
+                                                               pWapiSta->wapiUsk.keyId = *(data+42);\r
+                                                               pWapiSta->wapiUsk.bTxEnable = false;\r
+                                                       }\r
+                                               }else{\r
+                                                       WAPI_TRACE(WAPI_INIT,"ASUE update usk \n");\r
+                                                       if(bTxEnable){\r
+                                                               pWapiSta->wapiUskUpdate.bTxEnable = true;\r
+                                                               if(pWapiSta->wapiUskUpdate.bSet){\r
+                                                                       memcpy(pWapiSta->wapiUsk.dataKey,pWapiSta->wapiUskUpdate.dataKey,16);\r
+                                                                       memcpy(pWapiSta->wapiUsk.micKey,pWapiSta->wapiUskUpdate.micKey,16);\r
+                                                                       pWapiSta->wapiUsk.keyId=pWapiSta->wapiUskUpdate.keyId;\r
+                                                                       memcpy(pWapiSta->lastRxUnicastPNBEQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                                       memcpy(pWapiSta->lastRxUnicastPNBKQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                                       memcpy(pWapiSta->lastRxUnicastPNVIQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                                       memcpy(pWapiSta->lastRxUnicastPNVOQueue,WapiASUEPNInitialValueSrc,16);\r
+                                                                       memcpy(pWapiSta->lastRxUnicastPN,WapiASUEPNInitialValueSrc,16);\r
+                                                                       pWapiSta->wapiUskUpdate.bTxEnable = false;\r
+                                                                       pWapiSta->wapiUskUpdate.bSet = false;\r
+                                                               }\r
+                                                               memcpy(pWapiSta->lastTxUnicastPN,WapiASUEPNInitialValueSrc,16);\r
+                                                       }else{\r
+                                                               pWapiSta->wapiUskUpdate.bSet = true;\r
+                                                               memcpy(pWapiSta->wapiUskUpdate.dataKey,data+10,16);\r
+                                                               memcpy(pWapiSta->wapiUskUpdate.micKey,data+26,16);\r
+                                                               pWapiSta->wapiUskUpdate.keyId = *(data+42);\r
+                                                               pWapiSta->wapiUskUpdate.bTxEnable = false;\r
+                                                       }\r
+                                               }\r
+                                       }\r
+                                       break;\r
+                               case 2:         //msk\r
+                                       if(bAuthenticator){          //authenticator\r
+                                               pWapiInfo->wapiTxMsk.bSet = true;\r
+                                               memcpy(pWapiInfo->wapiTxMsk.dataKey,data+10,16);\r
+                                               memcpy(pWapiInfo->wapiTxMsk.micKey,data+26,16);\r
+                                               pWapiInfo->wapiTxMsk.keyId = *(data+42);\r
+                                               pWapiInfo->wapiTxMsk.bTxEnable = true;\r
+                                               memcpy(pWapiInfo->lastTxMulticastPN,WapiAEMultiCastPNInitialValueSrc,16);\r
+\r
+                                               if(!bUpdate){      //first\r
+                                                       WAPI_TRACE(WAPI_INIT, "AE fisrt set msk \n");\r
+                                                       if(!pWapiSta->bSetkeyOk)\r
+                                                               pWapiSta->bSetkeyOk = true;\r
+                                                       pWapiInfo->bFirstAuthentiateInProgress= false;\r
+                                               }else{               //update\r
+                                                       WAPI_TRACE(WAPI_INIT,"AE update msk \n");\r
+                                               }\r
+\r
+                                               WAPI_DATA(WAPI_INIT, "SetKey - AE MSK Data Key", pWapiInfo->wapiTxMsk.dataKey, 16);\r
+                                               WAPI_DATA(WAPI_INIT, "SetKey - AE MSK Mic Key", pWapiInfo->wapiTxMsk.micKey, 16);\r
+                                       }\r
+                                       else{\r
+                                               if(!bUpdate){\r
+                                                       WAPI_TRACE(WAPI_INIT,"ASUE fisrt set msk \n");\r
+                                                       pWapiSta->wapiMsk.bSet = true;\r
+                                                       memcpy(pWapiSta->wapiMsk.dataKey,data+10,16);\r
+                                                       memcpy(pWapiSta->wapiMsk.micKey,data+26,16);\r
+                                                       pWapiSta->wapiMsk.keyId = *(data+42);\r
+                                                       pWapiSta->wapiMsk.bTxEnable = false;\r
+                                                       if(!pWapiSta->bSetkeyOk)\r
+                                                               pWapiSta->bSetkeyOk = true;\r
+                                                       pWapiInfo->bFirstAuthentiateInProgress= false;\r
+                                                       WAPI_DATA(WAPI_INIT, "SetKey - ASUE MSK Data Key", pWapiSta->wapiMsk.dataKey, 16);\r
+                                                       WAPI_DATA(WAPI_INIT, "SetKey - ASUE MSK Mic Key", pWapiSta->wapiMsk.micKey, 16);\r
+                                               }else{\r
+                                                       WAPI_TRACE(WAPI_INIT,"ASUE update msk \n");\r
+                                                       pWapiSta->wapiMskUpdate.bSet = true;\r
+                                                       memcpy(pWapiSta->wapiMskUpdate.dataKey,data+10,16);\r
+                                                       memcpy(pWapiSta->wapiMskUpdate.micKey,data+26,16);\r
+                                                       pWapiSta->wapiMskUpdate.keyId = *(data+42);\r
+                                                       pWapiSta->wapiMskUpdate.bTxEnable = false;\r
+                                               }\r
+                                       }\r
+                                       break;\r
+                               default:\r
+                                       WAPI_TRACE(WAPI_ERR,"Unknown Flag \n");\r
+                                       break;\r
+                               }\r
+                       }\r
+               }\r
+       }\r
+       WAPI_TRACE(WAPI_INIT, "<===========%s\n", __FUNCTION__);\r
+}\r
+\r
+\r
+void wapi_test_init(struct _adapter *padapter)\r
+{\r
+       u8 keybuf[100];\r
+       u8 mac_addr[6]={0x00,0xe0,0x4c,0x72,0x04,0x70};\r
+       u8 UskDataKey[16]={0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f};\r
+       u8 UskMicKey[16]={0x10,0x11,0x12,0x13,0x14,0x15,0x16,0x17,0x18,0x19,0x1a,0x1b,0x1c,0x1d,0x1e,0x1f};\r
+       u8 UskId = 0;\r
+       u8 MskDataKey[16]={0x20,0x21,0x22,0x23,0x24,0x25,0x26,0x27,0x28,0x29,0x2a,0x2b,0x2c,0x2d,0x2e,0x2f};\r
+       u8 MskMicKey[16]={0x30,0x31,0x32,0x33,0x34,0x35,0x36,0x37,0x38,0x39,0x3a,0x3b,0x3c,0x3d,0x3e,0x3f};\r
+       u8 MskId = 0;\r
+\r
+       WAPI_TRACE(WAPI_INIT, "===========>%s\n", __FUNCTION__);\r
+\r
+       //Enable Wapi\r
+       WAPI_TRACE(WAPI_INIT, "%s: Enable wapi!!!!\n", __FUNCTION__);\r
+       padapter->wapiInfo.bWapiEnable = true;\r
+       padapter->pairwise_key_type = KEY_TYPE_SMS4;\r
+       ieee->group_key_type = KEY_TYPE_SMS4;\r
+       padapter->wapiInfo.extra_prefix_len = WAPI_EXT_LEN;\r
+       padapter->wapiInfo.extra_postfix_len = SMS4_MIC_LEN;\r
+\r
+       //set usk\r
+       WAPI_TRACE(WAPI_INIT, "%s: Set USK!!!!\n", __FUNCTION__);\r
+       memset(keybuf,0,100);\r
+       keybuf[0] = 1;                           //set usk\r
+       keybuf[1] = 1;                          //enable tx\r
+       keybuf[2] = 1;                          //AE\r
+       keybuf[3] = 0;                          //not update\r
+\r
+       memcpy(keybuf+4,mac_addr,6);\r
+       memcpy(keybuf+10,UskDataKey,16);\r
+       memcpy(keybuf+26,UskMicKey,16);\r
+       keybuf[42]=UskId;\r
+       wapi_test_set_key(padapter, keybuf);\r
+\r
+       memset(keybuf,0,100);\r
+       keybuf[0] = 1;                           //set usk\r
+       keybuf[1] = 1;                          //enable tx\r
+       keybuf[2] = 0;                          //AE\r
+       keybuf[3] = 0;                          //not update\r
+\r
+       memcpy(keybuf+4,mac_addr,6);\r
+       memcpy(keybuf+10,UskDataKey,16);\r
+       memcpy(keybuf+26,UskMicKey,16);\r
+       keybuf[42]=UskId;\r
+       wapi_test_set_key(padapter, keybuf);\r
+\r
+       //set msk\r
+       WAPI_TRACE(WAPI_INIT, "%s: Set MSK!!!!\n", __FUNCTION__);\r
+       memset(keybuf,0,100);\r
+       keybuf[0] = 2;                                //set msk\r
+       keybuf[1] = 1;                               //Enable TX\r
+       keybuf[2] = 1;                          //AE\r
+       keybuf[3] = 0;                              //not update\r
+       memcpy(keybuf+4,mac_addr,6);\r
+       memcpy(keybuf+10,MskDataKey,16);\r
+       memcpy(keybuf+26,MskMicKey,16);\r
+       keybuf[42] = MskId;\r
+       wapi_test_set_key(padapter, keybuf);\r
+\r
+       memset(keybuf,0,100);\r
+       keybuf[0] = 2;                                //set msk\r
+       keybuf[1] = 1;                               //Enable TX\r
+       keybuf[2] = 0;                          //AE\r
+       keybuf[3] = 0;                              //not update\r
+       memcpy(keybuf+4,mac_addr,6);\r
+       memcpy(keybuf+10,MskDataKey,16);\r
+       memcpy(keybuf+26,MskMicKey,16);\r
+       keybuf[42] = MskId;\r
+       wapi_test_set_key(padapter, keybuf);\r
+       WAPI_TRACE(WAPI_INIT, "<===========%s\n", __FUNCTION__);\r
+}\r
+#endif\r
+\r
+void rtw_wapi_get_iv(_adapter *padapter,u8 *pRA, u8*IV)\r
+{\r
+       PWLAN_HEADER_WAPI_EXTENSION pWapiExt = NULL;\r
+       PRT_WAPI_T         pWapiInfo = &padapter->wapiInfo;\r
+       bool    bPNOverflow = false;\r
+       bool    bFindMatchPeer = false;\r
+       PRT_WAPI_STA_INFO  pWapiSta = NULL;\r
+\r
+       pWapiExt = (PWLAN_HEADER_WAPI_EXTENSION)IV;\r
+\r
+       WAPI_DATA(WAPI_RX,"wapi_get_iv: pra",pRA,6);\r
+\r
+       if(IS_MCAST(pRA)){\r
+               if(!pWapiInfo->wapiTxMsk.bTxEnable){\r
+                        WAPI_TRACE(WAPI_ERR,"%s: bTxEnable = 0!!\n",__FUNCTION__);\r
+                        return;\r
+               }\r
+\r
+               if(pWapiInfo->wapiTxMsk.keyId <= 1){\r
+                        pWapiExt->KeyIdx = pWapiInfo->wapiTxMsk.keyId;\r
+                        pWapiExt->Reserved = 0;\r
+                        bPNOverflow = WapiIncreasePN(pWapiInfo->lastTxMulticastPN, 1);\r
+                        memcpy(pWapiExt->PN, pWapiInfo->lastTxMulticastPN, 16);\r
+               }\r
+       }\r
+       else\r
+       {\r
+               if(list_empty(&pWapiInfo->wapiSTAUsedList)){\r
+                       WAPI_TRACE(WAPI_RX,"rtw_wapi_get_iv: list is empty \n");\r
+                       _rtw_memset(IV,10,18);\r
+                       return;\r
+               }\r
+               else{\r
+                               list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list){\r
+                                       WAPI_DATA(WAPI_RX,"rtw_wapi_get_iv: peermacaddr ",pWapiSta->PeerMacAddr,6);\r
+                                       if (_rtw_memcmp((u8*)pWapiSta->PeerMacAddr, pRA, 6) == _TRUE) {\r
+                                               bFindMatchPeer = true;\r
+                                               break;\r
+                                       }\r
+                               }\r
+\r
+                               WAPI_TRACE(WAPI_RX,"bFindMatchPeer: %d \n",bFindMatchPeer);\r
+                               WAPI_DATA(WAPI_RX,"Addr",pRA,6);\r
+\r
+                               if (bFindMatchPeer){\r
+                                       if((!pWapiSta->wapiUskUpdate.bTxEnable) && (!pWapiSta->wapiUsk.bTxEnable))\r
+                                               return;\r
+\r
+                                       if (pWapiSta->wapiUsk.keyId <= 1){\r
+                                               if(pWapiSta->wapiUskUpdate.bTxEnable)\r
+                                                       pWapiExt->KeyIdx = pWapiSta->wapiUskUpdate.keyId;\r
+                                               else\r
+                                                       pWapiExt->KeyIdx = pWapiSta->wapiUsk.keyId;\r
+\r
+                                               pWapiExt->Reserved = 0;\r
+                                               bPNOverflow = WapiIncreasePN(pWapiSta->lastTxUnicastPN, 2);\r
+                                               _rtw_memcpy(pWapiExt->PN, pWapiSta->lastTxUnicastPN, 16);\r
+\r
+                                       }\r
+                               }\r
+               }\r
+\r
+       }\r
+\r
+}\r
+\r
+bool rtw_wapi_drop_for_key_absent(_adapter *padapter,u8 *pRA)\r
+{\r
+       PRT_WAPI_T         pWapiInfo = &padapter->wapiInfo;\r
+       bool                            bFindMatchPeer = false;\r
+       bool                            bDrop = false;\r
+       PRT_WAPI_STA_INFO  pWapiSta = NULL;\r
+       struct security_priv            *psecuritypriv = &padapter->securitypriv;\r
+\r
+       WAPI_DATA(WAPI_RX,"rtw_wapi_drop_for_key_absent: ra ",pRA,6);\r
+\r
+       if(psecuritypriv->dot11PrivacyAlgrthm == _SMS4_)\r
+       {\r
+               if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))\r
+                       return true;\r
+\r
+               if(IS_MCAST(pRA)){\r
+                       if(!pWapiInfo->wapiTxMsk.bTxEnable){\r
+                               bDrop = true;\r
+                               WAPI_TRACE(WAPI_RX,"rtw_wapi_drop_for_key_absent: multicast key is absent \n");\r
+                               return bDrop;\r
+                       }\r
+               }\r
+               else{\r
+                               if(!list_empty(&pWapiInfo->wapiSTAUsedList)){\r
+                                       list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list){\r
+                                               WAPI_DATA(WAPI_RX,"rtw_wapi_drop_for_key_absent: pWapiSta->PeerMacAddr ",pWapiSta->PeerMacAddr,6);\r
+                                               if (_rtw_memcmp(pRA, pWapiSta->PeerMacAddr, 6) == _TRUE){\r
+                                                       bFindMatchPeer = true;\r
+                                                       break;\r
+                                               }\r
+                                       }\r
+                                       if (bFindMatchPeer)     {\r
+                                               if (!pWapiSta->wapiUsk.bTxEnable){\r
+                                                       bDrop = true;\r
+                                                       WAPI_TRACE(WAPI_RX,"rtw_wapi_drop_for_key_absent: unicast key is absent \n");\r
+                                                       return bDrop;\r
+                                               }\r
+                                       }\r
+                                       else{\r
+                                               bDrop = true;\r
+                                               WAPI_TRACE(WAPI_RX,"rtw_wapi_drop_for_key_absent: no peer find \n");\r
+                                               return bDrop;\r
+                                       }\r
+\r
+                               }\r
+                               else{\r
+                                               bDrop = true;\r
+                                               WAPI_TRACE(WAPI_RX,"rtw_wapi_drop_for_key_absent: no sta  exist \n");\r
+                                               return bDrop;\r
+                               }\r
+               }\r
+       }\r
+       else\r
+       {\r
+               return bDrop;\r
+       }\r
+\r
+       return bDrop;\r
+}\r
+\r
+#endif\r