--- /dev/null
+#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
+#ifdef CONFIG_WAPI_SW_SMS4\r
+\r
+#define WAPI_LITTLE_ENDIAN\r
+//#define BIG_ENDIAN\r
+#define ENCRYPT 0\r
+#define DECRYPT 1\r
+\r
+\r
+/**********************************************************\r
+ **********************************************************/\r
+const u8 Sbox[256] = {\r
+0xd6,0x90,0xe9,0xfe,0xcc,0xe1,0x3d,0xb7,0x16,0xb6,0x14,0xc2,0x28,0xfb,0x2c,0x05,\r
+0x2b,0x67,0x9a,0x76,0x2a,0xbe,0x04,0xc3,0xaa,0x44,0x13,0x26,0x49,0x86,0x06,0x99,\r
+0x9c,0x42,0x50,0xf4,0x91,0xef,0x98,0x7a,0x33,0x54,0x0b,0x43,0xed,0xcf,0xac,0x62,\r
+0xe4,0xb3,0x1c,0xa9,0xc9,0x08,0xe8,0x95,0x80,0xdf,0x94,0xfa,0x75,0x8f,0x3f,0xa6,\r
+0x47,0x07,0xa7,0xfc,0xf3,0x73,0x17,0xba,0x83,0x59,0x3c,0x19,0xe6,0x85,0x4f,0xa8,\r
+0x68,0x6b,0x81,0xb2,0x71,0x64,0xda,0x8b,0xf8,0xeb,0x0f,0x4b,0x70,0x56,0x9d,0x35,\r
+0x1e,0x24,0x0e,0x5e,0x63,0x58,0xd1,0xa2,0x25,0x22,0x7c,0x3b,0x01,0x21,0x78,0x87,\r
+0xd4,0x00,0x46,0x57,0x9f,0xd3,0x27,0x52,0x4c,0x36,0x02,0xe7,0xa0,0xc4,0xc8,0x9e,\r
+0xea,0xbf,0x8a,0xd2,0x40,0xc7,0x38,0xb5,0xa3,0xf7,0xf2,0xce,0xf9,0x61,0x15,0xa1,\r
+0xe0,0xae,0x5d,0xa4,0x9b,0x34,0x1a,0x55,0xad,0x93,0x32,0x30,0xf5,0x8c,0xb1,0xe3,\r
+0x1d,0xf6,0xe2,0x2e,0x82,0x66,0xca,0x60,0xc0,0x29,0x23,0xab,0x0d,0x53,0x4e,0x6f,\r
+0xd5,0xdb,0x37,0x45,0xde,0xfd,0x8e,0x2f,0x03,0xff,0x6a,0x72,0x6d,0x6c,0x5b,0x51,\r
+0x8d,0x1b,0xaf,0x92,0xbb,0xdd,0xbc,0x7f,0x11,0xd9,0x5c,0x41,0x1f,0x10,0x5a,0xd8,\r
+0x0a,0xc1,0x31,0x88,0xa5,0xcd,0x7b,0xbd,0x2d,0x74,0xd0,0x12,0xb8,0xe5,0xb4,0xb0,\r
+0x89,0x69,0x97,0x4a,0x0c,0x96,0x77,0x7e,0x65,0xb9,0xf1,0x09,0xc5,0x6e,0xc6,0x84,\r
+0x18,0xf0,0x7d,0xec,0x3a,0xdc,0x4d,0x20,0x79,0xee,0x5f,0x3e,0xd7,0xcb,0x39,0x48\r
+};\r
+\r
+const u32 CK[32] = {\r
+ 0x00070e15, 0x1c232a31, 0x383f464d, 0x545b6269,\r
+ 0x70777e85, 0x8c939aa1, 0xa8afb6bd, 0xc4cbd2d9,\r
+ 0xe0e7eef5, 0xfc030a11, 0x181f262d, 0x343b4249,\r
+ 0x50575e65, 0x6c737a81, 0x888f969d, 0xa4abb2b9,\r
+ 0xc0c7ced5, 0xdce3eaf1, 0xf8ff060d, 0x141b2229,\r
+ 0x30373e45, 0x4c535a61, 0x686f767d, 0x848b9299,\r
+ 0xa0a7aeb5, 0xbcc3cad1, 0xd8dfe6ed, 0xf4fb0209,\r
+ 0x10171e25, 0x2c333a41, 0x484f565d, 0x646b7279 };\r
+\r
+#define Rotl(_x, _y) (((_x) << (_y)) | ((_x) >> (32 - (_y))))\r
+\r
+#define ByteSub(_A) (Sbox[(_A) >> 24 & 0xFF] << 24 | \\r
+ Sbox[(_A) >> 16 & 0xFF] << 16 | \\r
+ Sbox[(_A) >> 8 & 0xFF] << 8 | \\r
+ Sbox[(_A) & 0xFF])\r
+\r
+#define L1(_B) ((_B) ^ Rotl(_B, 2) ^ Rotl(_B, 10) ^ Rotl(_B, 18) ^ Rotl(_B, 24))\r
+#define L2(_B) ((_B) ^ Rotl(_B, 13) ^ Rotl(_B, 23))\r
+\r
+static void\r
+xor_block(void *dst, void *src1, void *src2)\r
+/* 128-bit xor: *dst = *src1 xor *src2. Pointers must be 32-bit aligned */\r
+{\r
+ ((u32 *)dst)[0] = ((u32 *)src1)[0] ^ ((u32 *)src2)[0];\r
+ ((u32 *)dst)[1] = ((u32 *)src1)[1] ^ ((u32 *)src2)[1];\r
+ ((u32 *)dst)[2] = ((u32 *)src1)[2] ^ ((u32 *)src2)[2];\r
+ ((u32 *)dst)[3] = ((u32 *)src1)[3] ^ ((u32 *)src2)[3];\r
+}\r
+\r
+\r
+void SMS4Crypt(u8 *Input, u8 *Output, u32 *rk)\r
+{\r
+ u32 r, mid, x0, x1, x2, x3, *p;\r
+ p = (u32 *)Input;\r
+ x0 = p[0];\r
+ x1 = p[1];\r
+ x2 = p[2];\r
+ x3 = p[3];\r
+#ifdef WAPI_LITTLE_ENDIAN\r
+ x0 = Rotl(x0, 16); x0 = ((x0 & 0x00FF00FF) << 8) | ((x0 & 0xFF00FF00) >> 8);\r
+ x1 = Rotl(x1, 16); x1 = ((x1 & 0x00FF00FF) << 8) | ((x1 & 0xFF00FF00) >> 8);\r
+ x2 = Rotl(x2, 16); x2 = ((x2 & 0x00FF00FF) << 8) | ((x2 & 0xFF00FF00) >> 8);\r
+ x3 = Rotl(x3, 16); x3 = ((x3 & 0x00FF00FF) << 8) | ((x3 & 0xFF00FF00) >> 8);\r
+#endif\r
+ for (r = 0; r < 32; r += 4)\r
+ {\r
+ mid = x1 ^ x2 ^ x3 ^ rk[r + 0];\r
+ mid = ByteSub(mid);\r
+ x0 ^= L1(mid);\r
+ mid = x2 ^ x3 ^ x0 ^ rk[r + 1];\r
+ mid = ByteSub(mid);\r
+ x1 ^= L1(mid);\r
+ mid = x3 ^ x0 ^ x1 ^ rk[r + 2];\r
+ mid = ByteSub(mid);\r
+ x2 ^= L1(mid);\r
+ mid = x0 ^ x1 ^ x2 ^ rk[r + 3];\r
+ mid = ByteSub(mid);\r
+ x3 ^= L1(mid);\r
+ }\r
+#ifdef WAPI_LITTLE_ENDIAN\r
+ x0 = Rotl(x0, 16); x0 = ((x0 & 0x00FF00FF) << 8) | ((x0 & 0xFF00FF00) >> 8);\r
+ x1 = Rotl(x1, 16); x1 = ((x1 & 0x00FF00FF) << 8) | ((x1 & 0xFF00FF00) >> 8);\r
+ x2 = Rotl(x2, 16); x2 = ((x2 & 0x00FF00FF) << 8) | ((x2 & 0xFF00FF00) >> 8);\r
+ x3 = Rotl(x3, 16); x3 = ((x3 & 0x00FF00FF) << 8) | ((x3 & 0xFF00FF00) >> 8);\r
+#endif\r
+ p = (u32 *)Output;\r
+ p[0] = x3;\r
+ p[1] = x2;\r
+ p[2] = x1;\r
+ p[3] = x0;\r
+}\r
+\r
+\r
+\r
+void SMS4KeyExt(u8 *Key, u32 *rk, u32 CryptFlag)\r
+{\r
+ u32 r, mid, x0, x1, x2, x3, *p;\r
+\r
+ p = (u32 *)Key;\r
+ x0 = p[0];\r
+ x1 = p[1];\r
+ x2 = p[2];\r
+ x3 = p[3];\r
+#ifdef WAPI_LITTLE_ENDIAN\r
+ x0 = Rotl(x0, 16); x0 = ((x0 & 0xFF00FF) << 8) | ((x0 & 0xFF00FF00) >> 8);\r
+ x1 = Rotl(x1, 16); x1 = ((x1 & 0xFF00FF) << 8) | ((x1 & 0xFF00FF00) >> 8);\r
+ x2 = Rotl(x2, 16); x2 = ((x2 & 0xFF00FF) << 8) | ((x2 & 0xFF00FF00) >> 8);\r
+ x3 = Rotl(x3, 16); x3 = ((x3 & 0xFF00FF) << 8) | ((x3 & 0xFF00FF00) >> 8);\r
+#endif\r
+\r
+ x0 ^= 0xa3b1bac6;\r
+ x1 ^= 0x56aa3350;\r
+ x2 ^= 0x677d9197;\r
+ x3 ^= 0xb27022dc;\r
+ for (r = 0; r < 32; r += 4)\r
+ {\r
+ mid = x1 ^ x2 ^ x3 ^ CK[r + 0];\r
+ mid = ByteSub(mid);\r
+ rk[r + 0] = x0 ^= L2(mid);\r
+ mid = x2 ^ x3 ^ x0 ^ CK[r + 1];\r
+ mid = ByteSub(mid);\r
+ rk[r + 1] = x1 ^= L2(mid);\r
+ mid = x3 ^ x0 ^ x1 ^ CK[r + 2];\r
+ mid = ByteSub(mid);\r
+ rk[r + 2] = x2 ^= L2(mid);\r
+ mid = x0 ^ x1 ^ x2 ^ CK[r + 3];\r
+ mid = ByteSub(mid);\r
+ rk[r + 3] = x3 ^= L2(mid);\r
+ }\r
+ if (CryptFlag == DECRYPT)\r
+ {\r
+ for (r = 0; r < 16; r++)\r
+ mid = rk[r], rk[r] = rk[31 - r], rk[31 - r] = mid;\r
+ }\r
+}\r
+\r
+\r
+void WapiSMS4Cryption(u8 *Key, u8 *IV, u8 *Input, u16 InputLength,\r
+ u8 *Output, u16 *OutputLength, u32 CryptFlag)\r
+{\r
+ u32 blockNum,i,j, rk[32];\r
+ u16 remainder;\r
+ u8 blockIn[16],blockOut[16], tempIV[16], k;\r
+\r
+ *OutputLength = 0;\r
+ remainder = InputLength & 0x0F;\r
+ blockNum = InputLength >> 4;\r
+ if(remainder !=0)\r
+ blockNum++;\r
+ else\r
+ remainder = 16;\r
+\r
+ for(k=0;k<16;k++)\r
+ tempIV[k] = IV[15-k];\r
+\r
+ memcpy(blockIn, tempIV, 16);\r
+\r
+ SMS4KeyExt((u8 *)Key, rk,CryptFlag);\r
+\r
+ for(i=0; i<blockNum-1; i++)\r
+ {\r
+ SMS4Crypt((u8 *)blockIn, blockOut, rk);\r
+ xor_block(&Output[i*16], &Input[i*16], blockOut);\r
+ memcpy(blockIn,blockOut,16);\r
+ }\r
+\r
+ *OutputLength = i*16;\r
+\r
+ SMS4Crypt((u8 *)blockIn, blockOut, rk);\r
+\r
+ for(j=0; j<remainder; j++)\r
+ {\r
+ Output[i*16+j] = Input[i*16+j] ^ blockOut[j];\r
+ }\r
+ *OutputLength += remainder;\r
+\r
+}\r
+\r
+void WapiSMS4Encryption(u8 *Key, u8 *IV, u8 *Input, u16 InputLength,\r
+ u8 *Output, u16 *OutputLength)\r
+{\r
+\r
+ WapiSMS4Cryption(Key, IV, Input, InputLength, Output, OutputLength, ENCRYPT);\r
+}\r
+\r
+void WapiSMS4Decryption(u8 *Key, u8 *IV, u8 *Input, u16 InputLength,\r
+ u8 *Output, u16 *OutputLength)\r
+{\r
+ // OFB mode: is also ENCRYPT flag\r
+ WapiSMS4Cryption(Key, IV, Input, InputLength, Output, OutputLength, ENCRYPT);\r
+}\r
+\r
+void WapiSMS4CalculateMic(u8 *Key, u8 *IV, u8 *Input1, u8 Input1Length,\r
+ u8 *Input2, u16 Input2Length, u8 *Output, u8 *OutputLength)\r
+{\r
+ u32 blockNum, i, remainder, rk[32];\r
+ u8 BlockIn[16], BlockOut[16], TempBlock[16], tempIV[16], k;\r
+\r
+ *OutputLength = 0;\r
+ remainder = Input1Length & 0x0F;\r
+ blockNum = Input1Length >> 4;\r
+\r
+ for(k=0;k<16;k++)\r
+ tempIV[k] = IV[15-k];\r
+\r
+ memcpy(BlockIn, tempIV, 16);\r
+\r
+ SMS4KeyExt((u8 *)Key, rk, ENCRYPT);\r
+\r
+ SMS4Crypt((u8 *)BlockIn, BlockOut, rk);\r
+\r
+ for(i=0; i<blockNum; i++){\r
+ xor_block(BlockIn, (Input1+i*16), BlockOut);\r
+ SMS4Crypt((u8 *)BlockIn, BlockOut, rk);\r
+ }\r
+\r
+ if(remainder !=0){\r
+ memset(TempBlock, 0, 16);\r
+ memcpy(TempBlock, (Input1+blockNum*16), remainder);\r
+\r
+ xor_block(BlockIn, TempBlock, BlockOut);\r
+ SMS4Crypt((u8 *)BlockIn, BlockOut, rk);\r
+ }\r
+\r
+ remainder = Input2Length & 0x0F;\r
+ blockNum = Input2Length >> 4;\r
+\r
+ for(i=0; i<blockNum; i++){\r
+ xor_block(BlockIn, (Input2+i*16), BlockOut);\r
+ SMS4Crypt((u8 *)BlockIn, BlockOut, rk);\r
+ }\r
+\r
+ if(remainder !=0){\r
+ memset(TempBlock, 0, 16);\r
+ memcpy(TempBlock, (Input2+blockNum*16), remainder);\r
+\r
+ xor_block(BlockIn, TempBlock, BlockOut);\r
+ SMS4Crypt((u8 *)BlockIn, BlockOut, rk);\r
+ }\r
+\r
+ memcpy(Output, BlockOut, 16);\r
+ *OutputLength = 16;\r
+}\r
+\r
+void SecCalculateMicSMS4(\r
+ u8 KeyIdx,\r
+ u8 *MicKey,\r
+ u8 *pHeader,\r
+ u8 *pData,\r
+ u16 DataLen,\r
+ u8 *MicBuffer\r
+ )\r
+{\r
+#if 0\r
+ struct ieee80211_hdr_3addr_qos *header;\r
+ u8 TempBuf[34], TempLen = 32, MicLen, QosOffset, *IV;\r
+ u16 *pTemp, fc;\r
+\r
+ WAPI_TRACE(WAPI_TX|WAPI_RX, "=========>%s\n", __FUNCTION__);\r
+\r
+ header = (struct ieee80211_hdr_3addr_qos *)pHeader;\r
+ memset(TempBuf, 0, 34);\r
+ memcpy(TempBuf, pHeader, 2); //FrameCtrl\r
+ pTemp = (u16*)TempBuf;\r
+ *pTemp &= 0xc78f; //bit4,5,6,11,12,13\r
+\r
+ memcpy((TempBuf+2), (pHeader+4), 12); //Addr1, Addr2\r
+ memcpy((TempBuf+14), (pHeader+22), 2); // SeqCtrl\r
+ pTemp = (u16*)(TempBuf + 14);\r
+ *pTemp &= 0x000f;\r
+\r
+ memcpy((TempBuf+16), (pHeader+16), 6); //Addr3\r
+\r
+ fc = le16_to_cpu(header->frame_ctl);\r
+\r
+\r
+\r
+ if (GetFrDs((u16*)&fc) && GetToDs((u16 *)&fc))\r
+ {\r
+ memcpy((TempBuf+22), (pHeader+24), 6);\r
+ QosOffset = 30;\r
+ }else{\r
+ memset((TempBuf+22), 0, 6);\r
+ QosOffset = 24;\r
+ }\r
+\r
+ if((fc & 0x0088) == 0x0088){\r
+ memcpy((TempBuf+28), (pHeader+QosOffset), 2);\r
+ TempLen += 2;\r
+ //IV = pHeader + QosOffset + 2 + SNAP_SIZE + sizeof(u16) + 2;\r
+ IV = pHeader + QosOffset + 2 + 2;\r
+ }else{\r
+ IV = pHeader + QosOffset + 2;\r
+ //IV = pHeader + QosOffset + SNAP_SIZE + sizeof(u16) + 2;\r
+ }\r
+\r
+ TempBuf[TempLen-1] = (u8)(DataLen & 0xff);\r
+ TempBuf[TempLen-2] = (u8)((DataLen & 0xff00)>>8);\r
+ TempBuf[TempLen-4] = KeyIdx;\r
+\r
+ WAPI_DATA(WAPI_TX, "CalculateMic - KEY", MicKey, 16);\r
+ WAPI_DATA(WAPI_TX, "CalculateMic - IV", IV, 16);\r
+ WAPI_DATA(WAPI_TX, "CalculateMic - TempBuf", TempBuf, TempLen);\r
+ WAPI_DATA(WAPI_TX, "CalculateMic - pData", pData, DataLen);\r
+\r
+ WapiSMS4CalculateMic(MicKey, IV, TempBuf, TempLen,\r
+ pData, DataLen, MicBuffer, &MicLen);\r
+\r
+ if (MicLen != 16)\r
+ WAPI_TRACE(WAPI_ERR,"%s: MIC Length Error!!\n",__FUNCTION__);\r
+\r
+ WAPI_TRACE(WAPI_TX|WAPI_RX, "<=========%s\n", __FUNCTION__);\r
+#endif\r
+}\r
+\r
+/* AddCount: 1 or 2.\r
+ * If overflow, return 1,\r
+ * else return 0.\r
+ */\r
+u8 WapiIncreasePN(u8 *PN, u8 AddCount)\r
+{\r
+ u8 i;\r
+\r
+ if (NULL == PN)\r
+ return 1;\r
+ //YJ,test,091102\r
+ /*\r
+ if(AddCount == 2){\r
+ DBG_8192C("############################%s(): PN[0]=0x%x\n", __FUNCTION__, PN[0]);\r
+ if(PN[0] == 0x48){\r
+ PN[0] += AddCount;\r
+ return 1;\r
+ }else{\r
+ PN[0] += AddCount;\r
+ return 0;\r
+ }\r
+ }\r
+ */\r
+ //YJ,test,091102,end\r
+\r
+ for (i=0; i<16; i++)\r
+ {\r
+ if (PN[i] + AddCount <= 0xff)\r
+ {\r
+ PN[i] += AddCount;\r
+ return 0;\r
+ }\r
+ else\r
+ {\r
+ PN[i] += AddCount;\r
+ AddCount = 1;\r
+ }\r
+ }\r
+ return 1;\r
+}\r
+\r
+\r
+void WapiGetLastRxUnicastPNForQoSData(\r
+ u8 UserPriority,\r
+ PRT_WAPI_STA_INFO pWapiStaInfo,\r
+ u8 *PNOut\r
+)\r
+{\r
+ WAPI_TRACE(WAPI_RX, "===========> %s\n", __FUNCTION__);\r
+ switch(UserPriority)\r
+ {\r
+ case 0:\r
+ case 3:\r
+ memcpy(PNOut,pWapiStaInfo->lastRxUnicastPNBEQueue,16);\r
+ break;\r
+ case 1:\r
+ case 2:\r
+ memcpy(PNOut,pWapiStaInfo->lastRxUnicastPNBKQueue,16);\r
+ break;\r
+ case 4:\r
+ case 5:\r
+ memcpy(PNOut,pWapiStaInfo->lastRxUnicastPNVIQueue,16);\r
+ break;\r
+ case 6:\r
+ case 7:\r
+ memcpy(PNOut,pWapiStaInfo->lastRxUnicastPNVOQueue,16);\r
+ break;\r
+ default:\r
+ WAPI_TRACE(WAPI_ERR, "%s: Unknown TID \n", __FUNCTION__);\r
+ break;\r
+ }\r
+ WAPI_TRACE(WAPI_RX, "<=========== %s\n", __FUNCTION__);\r
+}\r
+\r
+\r
+void WapiSetLastRxUnicastPNForQoSData(\r
+ u8 UserPriority,\r
+ u8 *PNIn,\r
+ PRT_WAPI_STA_INFO pWapiStaInfo\r
+)\r
+{\r
+ WAPI_TRACE(WAPI_RX, "===========> %s\n", __FUNCTION__);\r
+ switch(UserPriority)\r
+ {\r
+ case 0:\r
+ case 3:\r
+ memcpy(pWapiStaInfo->lastRxUnicastPNBEQueue,PNIn,16);\r
+ break;\r
+ case 1:\r
+ case 2:\r
+ memcpy(pWapiStaInfo->lastRxUnicastPNBKQueue,PNIn,16);\r
+ break;\r
+ case 4:\r
+ case 5:\r
+ memcpy(pWapiStaInfo->lastRxUnicastPNVIQueue,PNIn,16);\r
+ break;\r
+ case 6:\r
+ case 7:\r
+ memcpy(pWapiStaInfo->lastRxUnicastPNVOQueue,PNIn,16);\r
+ break;\r
+ default:\r
+ WAPI_TRACE(WAPI_ERR, "%s: Unknown TID \n", __FUNCTION__);\r
+ break;\r
+ }\r
+ WAPI_TRACE(WAPI_RX, "<=========== %s\n", __FUNCTION__);\r
+}\r
+\r
+\r
+/****************************************************************************\r
+ FALSE not RX-Reorder\r
+ TRUE do RX Reorder\r
+add to support WAPI to N-mode\r
+*****************************************************************************/\r
+u8 WapiCheckPnInSwDecrypt(\r
+ _adapter *padapter,\r
+ struct sk_buff *pskb\r
+)\r
+{\r
+ u8 ret = false;\r
+\r
+#if 0\r
+ struct ieee80211_hdr_3addr_qos *header;\r
+ u16 fc;\r
+ u8 *pDaddr, *pTaddr, *pRaddr;\r
+\r
+ header = (struct ieee80211_hdr_3addr_qos *)pskb->data;\r
+ pTaddr = header->addr2;\r
+ pRaddr = header->addr1;\r
+ fc = le16_to_cpu(header->frame_ctl);\r
+\r
+ if(GetToDs(&fc))\r
+ pDaddr = header->addr3;\r
+ else\r
+ pDaddr = header->addr1;\r
+\r
+ if ((_rtw_memcmp(pRaddr, padapter->pnetdev->dev_addr, ETH_ALEN) == 0)\r
+ && ! (pDaddr)\r
+ && (GetFrameType(&fc) == WIFI_QOS_DATA_TYPE))\r
+ //&& ieee->pHTInfo->bCurrentHTSupport &&\r
+ //ieee->pHTInfo->bCurRxReorderEnable)\r
+ ret = false;\r
+ else\r
+ ret = true;\r
+#endif\r
+ WAPI_TRACE(WAPI_RX, "%s: return %d\n", __FUNCTION__, ret);\r
+ return ret;\r
+}\r
+\r
+int SecSMS4HeaderFillIV(_adapter *padapter, u8 *pxmitframe)\r
+{\r
+ struct pkt_attrib *pattrib = &((struct xmit_frame*)pxmitframe)->attrib;\r
+ u8 * frame = ((struct xmit_frame *)pxmitframe)->buf_addr + TXDESC_OFFSET;\r
+ u8 *pSecHeader = NULL, *pos = NULL, *pRA = NULL;\r
+ u8 bPNOverflow = false, bFindMatchPeer = false, hdr_len = 0;\r
+ PWLAN_HEADER_WAPI_EXTENSION pWapiExt = NULL;\r
+ PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;\r
+ PRT_WAPI_STA_INFO pWapiSta = NULL;\r
+ int ret = 0;\r
+\r
+ WAPI_TRACE(WAPI_TX, "=========>%s\n", __FUNCTION__);\r
+\r
+ return ret;\r
+#if 0\r
+ hdr_len = sMacHdrLng;\r
+ if (GetFrameType(pskb->data) == WIFI_QOS_DATA_TYPE)\r
+ {\r
+ hdr_len += 2;\r
+ }\r
+ //hdr_len += SNAP_SIZE + sizeof(u16);\r
+\r
+ pos = skb_push(pskb, padapter->wapiInfo.extra_prefix_len);\r
+ memmove(pos, pos+padapter->wapiInfo.extra_prefix_len, hdr_len);\r
+\r
+ pSecHeader = pskb->data + hdr_len;\r
+ pWapiExt = (PWLAN_HEADER_WAPI_EXTENSION)pSecHeader;\r
+ pRA = pskb->data + 4;\r
+\r
+ WAPI_DATA(WAPI_TX, "FillIV - Before Fill IV", pskb->data, pskb->len);\r
+\r
+ //Address 1 is always receiver's address\r
+ if( IS_MCAST(pRA) ){\r
+ if(!pWapiInfo->wapiTxMsk.bTxEnable){\r
+ WAPI_TRACE(WAPI_ERR,"%s: bTxEnable = 0!!\n",__FUNCTION__);\r
+ return -2;\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
+ if (bPNOverflow){\r
+ // Update MSK Notification.\r
+ WAPI_TRACE(WAPI_ERR,"===============>%s():multicast PN overflow\n",__FUNCTION__);\r
+ rtw_wapi_app_event_handler(padapter,NULL,0,pRA, false, false, true, 0, false);\r
+ }\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR,"%s: Invalid Wapi Multicast KeyIdx!!\n",__FUNCTION__);\r
+ ret = -3;\r
+ }\r
+ }\r
+ else{\r
+ list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {\r
+ if(!memcmp(pWapiSta->PeerMacAddr,pRA,6)){\r
+ bFindMatchPeer = true;\r
+ break;\r
+ }\r
+ }\r
+ if (bFindMatchPeer){\r
+ if((!pWapiSta->wapiUskUpdate.bTxEnable) && (!pWapiSta->wapiUsk.bTxEnable)){\r
+ WAPI_TRACE(WAPI_ERR,"%s: bTxEnable = 0!!\n",__FUNCTION__);\r
+ return -4;\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
+ memcpy(pWapiExt->PN, pWapiSta->lastTxUnicastPN, 16);\r
+ if (bPNOverflow){\r
+ // Update USK Notification.\r
+ WAPI_TRACE(WAPI_ERR,"===============>%s():unicast PN overflow\n",__FUNCTION__);\r
+ rtw_wapi_app_event_handler(padapter,NULL,0,pWapiSta->PeerMacAddr, false, true, false, 0, false);\r
+ }\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR,"%s: Invalid Wapi Unicast KeyIdx!!\n",__FUNCTION__);\r
+ ret = -5;\r
+ }\r
+ }\r
+ else{\r
+ WAPI_TRACE(WAPI_ERR,"%s: Can not find Peer Sta "MAC_FMT"!!\n",__FUNCTION__, MAC_ARG(pRA));\r
+ ret = -6;\r
+ }\r
+ }\r
+\r
+ WAPI_DATA(WAPI_TX, "FillIV - After Fill IV", pskb->data, pskb->len);\r
+ WAPI_TRACE(WAPI_TX, "<=========%s\n", __FUNCTION__);\r
+ return ret;\r
+#endif\r
+}\r
+\r
+// WAPI SW Enc: must have done Coalesce!\r
+void SecSWSMS4Encryption(\r
+ _adapter *padapter,\r
+ u8 * pxmitframe\r
+ )\r
+{\r
+ PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;\r
+ PRT_WAPI_STA_INFO pWapiSta = NULL;\r
+ u8 *pframe = ((struct xmit_frame*)pxmitframe)->buf_addr + TXDESC_SIZE;\r
+ struct pkt_attrib *pattrib = &((struct xmit_frame*)pxmitframe)->attrib;\r
+\r
+ u8 *SecPtr = NULL, *pRA, *pMicKey = NULL, *pDataKey = NULL, *pIV = NULL;\r
+ u8 IVOffset, DataOffset, bFindMatchPeer = false, KeyIdx = 0, MicBuffer[16];\r
+ u16 OutputLength;\r
+\r
+ WAPI_TRACE(WAPI_TX, "=========>%s\n", __FUNCTION__);\r
+\r
+ WAPI_TRACE(WAPI_TX,"hdrlen: %d \n",pattrib->hdrlen);\r
+\r
+ return;\r
+\r
+ DataOffset = pattrib->hdrlen + pattrib->iv_len;\r
+\r
+ pRA = pframe + 4;\r
+\r
+\r
+ if( IS_MCAST(pRA) ){\r
+ KeyIdx = pWapiInfo->wapiTxMsk.keyId;\r
+ pIV = pWapiInfo->lastTxMulticastPN;\r
+ pMicKey = pWapiInfo->wapiTxMsk.micKey;\r
+ pDataKey = pWapiInfo->wapiTxMsk.dataKey;\r
+ }else{\r
+ if (!list_empty(&(pWapiInfo->wapiSTAUsedList))){\r
+ list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {\r
+ if (0 == memcmp(pWapiSta->PeerMacAddr, pRA, 6)){\r
+ bFindMatchPeer = true;\r
+ break;\r
+ }\r
+ }\r
+\r
+ if (bFindMatchPeer){\r
+ if (pWapiSta->wapiUskUpdate.bTxEnable){\r
+ KeyIdx = pWapiSta->wapiUskUpdate.keyId;\r
+ WAPI_TRACE(WAPI_TX, "%s(): Use update USK!! KeyIdx=%d\n", __FUNCTION__, KeyIdx);\r
+ pIV = pWapiSta->lastTxUnicastPN;\r
+ pMicKey = pWapiSta->wapiUskUpdate.micKey;\r
+ pDataKey = pWapiSta->wapiUskUpdate.dataKey;\r
+ }else{\r
+ KeyIdx = pWapiSta->wapiUsk.keyId;\r
+ WAPI_TRACE(WAPI_TX, "%s(): Use USK!! KeyIdx=%d\n", __FUNCTION__, KeyIdx);\r
+ pIV = pWapiSta->lastTxUnicastPN;\r
+ pMicKey = pWapiSta->wapiUsk.micKey;\r
+ pDataKey = pWapiSta->wapiUsk.dataKey;\r
+ }\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR,"%s: Can not find Peer Sta!!\n",__FUNCTION__);\r
+ return;\r
+ }\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR,"%s: wapiSTAUsedList is empty!!\n",__FUNCTION__);\r
+ return;\r
+ }\r
+ }\r
+\r
+ SecPtr = pframe;\r
+ SecCalculateMicSMS4(KeyIdx, pMicKey, SecPtr, (SecPtr+DataOffset), pattrib->pktlen, MicBuffer);\r
+\r
+ WAPI_DATA(WAPI_TX, "Encryption - MIC", MicBuffer, padapter->wapiInfo.extra_postfix_len);\r
+\r
+ memcpy(pframe+pattrib->hdrlen+pattrib->iv_len+pattrib->pktlen-pattrib->icv_len,\r
+ (u8 *)MicBuffer,\r
+ padapter->wapiInfo.extra_postfix_len\r
+ );\r
+\r
+\r
+ WapiSMS4Encryption(pDataKey, pIV, (SecPtr+DataOffset),pattrib->pktlen+pattrib->icv_len, (SecPtr+DataOffset), &OutputLength);\r
+\r
+ WAPI_DATA(WAPI_TX, "Encryption - After SMS4 encryption",pframe,pattrib->hdrlen+pattrib->iv_len+pattrib->pktlen);\r
+\r
+ WAPI_TRACE(WAPI_TX, "<=========%s\n", __FUNCTION__);\r
+}\r
+\r
+u8 SecSWSMS4Decryption(\r
+ _adapter *padapter,\r
+ u8 *precv_frame,\r
+ struct recv_priv *precv_priv\r
+ )\r
+{\r
+ PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;\r
+ struct recv_frame_hdr *precv_hdr;\r
+ PRT_WAPI_STA_INFO pWapiSta = NULL;\r
+ u8 IVOffset, DataOffset, bFindMatchPeer = false, bUseUpdatedKey = false;\r
+ u8 KeyIdx, MicBuffer[16], lastRxPNforQoS[16];\r
+ u8 *pRA, *pTA, *pMicKey, *pDataKey, *pLastRxPN, *pRecvPN, *pSecData, *pRecvMic, *pos;\r
+ u8 TID = 0;\r
+ u16 OutputLength, DataLen;\r
+ u8 bQosData;\r
+ struct sk_buff * pskb;\r
+\r
+ WAPI_TRACE(WAPI_RX, "=========>%s\n", __FUNCTION__);\r
+\r
+ return 0;\r
+\r
+ precv_hdr = &((union recv_frame*)precv_frame)->u.hdr;\r
+ pskb = (struct sk_buff *)(precv_hdr->rx_data);\r
+ precv_hdr->bWapiCheckPNInDecrypt = WapiCheckPnInSwDecrypt(padapter, pskb);\r
+ WAPI_TRACE(WAPI_RX, "=========>%s: check PN %d\n", __FUNCTION__,precv_hdr->bWapiCheckPNInDecrypt);\r
+ WAPI_DATA(WAPI_RX, "Decryption - Before decryption", pskb->data, pskb->len);\r
+\r
+ IVOffset = sMacHdrLng;\r
+ bQosData = GetFrameType(pskb->data) == WIFI_QOS_DATA_TYPE;\r
+ if (bQosData){\r
+ IVOffset += 2;\r
+ }\r
+\r
+ //if(GetHTC())\r
+ // IVOffset += 4;\r
+\r
+ //IVOffset += SNAP_SIZE + sizeof(u16);\r
+\r
+ DataOffset = IVOffset + padapter->wapiInfo.extra_prefix_len;\r
+\r
+ pRA = pskb->data + 4;\r
+ pTA = pskb->data + 10;\r
+ KeyIdx = *(pskb->data + IVOffset);\r
+ pRecvPN = pskb->data + IVOffset + 2;\r
+ pSecData = pskb->data + DataOffset;\r
+ DataLen = pskb->len - DataOffset;\r
+ pRecvMic = pskb->data + pskb->len - padapter->wapiInfo.extra_postfix_len;\r
+ TID = GetTid(pskb->data);\r
+\r
+ if (!list_empty(&(pWapiInfo->wapiSTAUsedList))){\r
+ list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {\r
+ if (0 == memcmp(pWapiSta->PeerMacAddr, pTA, 6)){\r
+ bFindMatchPeer = true;\r
+ break;\r
+ }\r
+ }\r
+ }\r
+\r
+ if (!bFindMatchPeer){\r
+ WAPI_TRACE(WAPI_ERR, "%s: Can not find Peer Sta "MAC_FMT" for Key Info!!!\n", __FUNCTION__, MAC_ARG(pTA));\r
+ return false;\r
+ }\r
+\r
+ if( IS_MCAST(pRA) ){\r
+ WAPI_TRACE(WAPI_RX, "%s: Multicast decryption !!!\n", __FUNCTION__);\r
+ if (pWapiSta->wapiMsk.keyId == KeyIdx && pWapiSta->wapiMsk.bSet){\r
+ pLastRxPN = pWapiSta->lastRxMulticastPN;\r
+ if (!WapiComparePN(pRecvPN, pLastRxPN)){\r
+ WAPI_TRACE(WAPI_ERR, "%s: MSK PN is not larger than last, Dropped!!!\n", __FUNCTION__);\r
+ WAPI_DATA(WAPI_ERR, "pRecvPN:", pRecvPN, 16);\r
+ WAPI_DATA(WAPI_ERR, "pLastRxPN:", pLastRxPN, 16);\r
+ return false;\r
+ }\r
+\r
+ memcpy(pLastRxPN, pRecvPN, 16);\r
+ pMicKey = pWapiSta->wapiMsk.micKey;\r
+ pDataKey = pWapiSta->wapiMsk.dataKey;\r
+ }else if (pWapiSta->wapiMskUpdate.keyId == KeyIdx && pWapiSta->wapiMskUpdate.bSet){\r
+ WAPI_TRACE(WAPI_RX, "%s: Use Updated MSK for Decryption !!!\n", __FUNCTION__);\r
+ bUseUpdatedKey = true;\r
+ memcpy(pWapiSta->lastRxMulticastPN, pRecvPN, 16);\r
+ pMicKey = pWapiSta->wapiMskUpdate.micKey;\r
+ pDataKey = pWapiSta->wapiMskUpdate.dataKey;\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR, "%s: Can not find MSK with matched KeyIdx(%d), Dropped !!!\n", __FUNCTION__,KeyIdx);\r
+ return false;\r
+ }\r
+ }\r
+ else{\r
+ WAPI_TRACE(WAPI_RX, "%s: Unicast decryption !!!\n", __FUNCTION__);\r
+ if (pWapiSta->wapiUsk.keyId == KeyIdx && pWapiSta->wapiUsk.bSet){\r
+ WAPI_TRACE(WAPI_RX, "%s: Use USK for Decryption!!!\n", __FUNCTION__);\r
+ if(precv_hdr->bWapiCheckPNInDecrypt){\r
+ if(GetFrameType(pskb->data) == WIFI_QOS_DATA_TYPE){\r
+ WapiGetLastRxUnicastPNForQoSData(TID, pWapiSta, lastRxPNforQoS);\r
+ pLastRxPN = lastRxPNforQoS;\r
+ }else{\r
+ pLastRxPN = pWapiSta->lastRxUnicastPN;\r
+ }\r
+ if (!WapiComparePN(pRecvPN, pLastRxPN)){\r
+ return false;\r
+ }\r
+ if(bQosData){\r
+ WapiSetLastRxUnicastPNForQoSData(TID, pRecvPN, pWapiSta);\r
+ }else{\r
+ memcpy(pWapiSta->lastRxUnicastPN, pRecvPN, 16);\r
+ }\r
+ }else{\r
+ memcpy(precv_hdr->WapiTempPN,pRecvPN,16);\r
+ }\r
+\r
+ if (check_fwstate(&padapter->mlmepriv, WIFI_STATION_STATE))\r
+ {\r
+ if ((pRecvPN[0] & 0x1) == 0){\r
+ WAPI_TRACE(WAPI_ERR, "%s: Rx USK PN is not odd when Infra STA mode, Dropped !!!\n", __FUNCTION__);\r
+ return false;\r
+ }\r
+ }\r
+\r
+ pMicKey = pWapiSta->wapiUsk.micKey;\r
+ pDataKey = pWapiSta->wapiUsk.dataKey;\r
+ }\r
+ else if (pWapiSta->wapiUskUpdate.keyId == KeyIdx && pWapiSta->wapiUskUpdate.bSet ){\r
+ WAPI_TRACE(WAPI_RX, "%s: Use Updated USK for Decryption!!!\n", __FUNCTION__);\r
+ if(pWapiSta->bAuthenticatorInUpdata)\r
+ bUseUpdatedKey = true;\r
+ else\r
+ bUseUpdatedKey = false;\r
+\r
+ if(bQosData){\r
+ WapiSetLastRxUnicastPNForQoSData(TID, pRecvPN, pWapiSta);\r
+ }else{\r
+ memcpy(pWapiSta->lastRxUnicastPN, pRecvPN, 16);\r
+ }\r
+ pMicKey = pWapiSta->wapiUskUpdate.micKey;\r
+ pDataKey = pWapiSta->wapiUskUpdate.dataKey;\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR, "%s: No valid USK!!!KeyIdx=%d pWapiSta->wapiUsk.keyId=%d pWapiSta->wapiUskUpdate.keyId=%d\n", __FUNCTION__, KeyIdx, pWapiSta->wapiUsk.keyId, pWapiSta->wapiUskUpdate.keyId);\r
+ //dump_buf(pskb->data,pskb->len);\r
+ return false;\r
+ }\r
+ }\r
+\r
+ WAPI_DATA(WAPI_RX, "Decryption - DataKey", pDataKey, 16);\r
+ WAPI_DATA(WAPI_RX, "Decryption - IV", pRecvPN, 16);\r
+ WapiSMS4Decryption(pDataKey, pRecvPN, pSecData, DataLen, pSecData, &OutputLength);\r
+\r
+ if (OutputLength != DataLen)\r
+ WAPI_TRACE(WAPI_ERR, "%s: Output Length Error!!!!\n", __FUNCTION__);\r
+\r
+ WAPI_DATA(WAPI_RX, "Decryption - After decryption", pskb->data, pskb->len);\r
+\r
+ DataLen -= padapter->wapiInfo.extra_postfix_len;\r
+\r
+ SecCalculateMicSMS4(KeyIdx, pMicKey, pskb->data, pSecData, DataLen, MicBuffer);\r
+\r
+ WAPI_DATA(WAPI_RX, "Decryption - MIC received", pRecvMic, SMS4_MIC_LEN);\r
+ WAPI_DATA(WAPI_RX, "Decryption - MIC calculated", MicBuffer, SMS4_MIC_LEN);\r
+\r
+ if (0 == memcmp(MicBuffer, pRecvMic, padapter->wapiInfo.extra_postfix_len)){\r
+ WAPI_TRACE(WAPI_RX, "%s: Check MIC OK!!\n", __FUNCTION__);\r
+ if (bUseUpdatedKey){\r
+ // delete the old key\r
+ if ( IS_MCAST(pRA) ){\r
+ WAPI_TRACE(WAPI_API, "%s(): AE use new update MSK!!\n", __FUNCTION__);\r
+ pWapiSta->wapiMsk.keyId = pWapiSta->wapiMskUpdate.keyId;\r
+ memcpy(pWapiSta->wapiMsk.dataKey, pWapiSta->wapiMskUpdate.dataKey, 16);\r
+ memcpy(pWapiSta->wapiMsk.micKey, pWapiSta->wapiMskUpdate.micKey, 16);\r
+ pWapiSta->wapiMskUpdate.bTxEnable = pWapiSta->wapiMskUpdate.bSet = false;\r
+ }else{\r
+ WAPI_TRACE(WAPI_API, "%s(): AE use new update USK!!\n", __FUNCTION__);\r
+ pWapiSta->wapiUsk.keyId = pWapiSta->wapiUskUpdate.keyId;\r
+ memcpy(pWapiSta->wapiUsk.dataKey, pWapiSta->wapiUskUpdate.dataKey, 16);\r
+ memcpy(pWapiSta->wapiUsk.micKey, pWapiSta->wapiUskUpdate.micKey, 16);\r
+ pWapiSta->wapiUskUpdate.bTxEnable = pWapiSta->wapiUskUpdate.bSet = false;\r
+ }\r
+ }\r
+ }else{\r
+ WAPI_TRACE(WAPI_ERR, "%s: Check MIC Error, Dropped !!!!\n", __FUNCTION__);\r
+ return false;\r
+ }\r
+\r
+ pos = pskb->data;\r
+ memmove(pos+padapter->wapiInfo.extra_prefix_len, pos, IVOffset);\r
+ skb_pull(pskb, padapter->wapiInfo.extra_prefix_len);\r
+\r
+ WAPI_TRACE(WAPI_RX, "<=========%s\n", __FUNCTION__);\r
+\r
+ return true;\r
+}\r
+\r
+u32 rtw_sms4_encrypt(_adapter *padapter, u8 *pxmitframe)\r
+{\r
+\r
+ u8 *pframe;\r
+ u32 res = _SUCCESS;\r
+\r
+ WAPI_TRACE(WAPI_TX, "=========>%s\n", __FUNCTION__);\r
+\r
+ if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable))\r
+ {\r
+ WAPI_TRACE(WAPI_TX, "<========== %s, WAPI not supported or enabled!\n", __FUNCTION__);\r
+ return _FAIL;\r
+ }\r
+\r
+ if(((struct xmit_frame*)pxmitframe)->buf_addr==NULL)\r
+ return _FAIL;\r
+\r
+ pframe = ((struct xmit_frame*)pxmitframe)->buf_addr + TXDESC_OFFSET;\r
+\r
+ SecSWSMS4Encryption(padapter, pxmitframe);\r
+\r
+ WAPI_TRACE(WAPI_TX, "<=========%s\n", __FUNCTION__);\r
+ return res;\r
+}\r
+\r
+u32 rtw_sms4_decrypt(_adapter *padapter, u8 *precvframe)\r
+{\r
+ u8 *pframe;\r
+ u32 res = _SUCCESS;\r
+\r
+ WAPI_TRACE(WAPI_RX, "=========>%s\n", __FUNCTION__);\r
+\r
+ if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable))\r
+ {\r
+ WAPI_TRACE(WAPI_RX, "<========== %s, WAPI not supported or enabled!\n", __FUNCTION__);\r
+ return _FAIL;\r
+ }\r
+\r
+\r
+ //drop packet when hw decrypt fail\r
+ //return tempraily\r
+ return _FAIL;\r
+\r
+ //pframe=(unsigned char *)((union recv_frame*)precvframe)->u.hdr.rx_data;\r
+\r
+ if (false == SecSWSMS4Decryption(padapter, precvframe, &padapter->recvpriv))\r
+ {\r
+ WAPI_TRACE(WAPI_ERR, "%s():SMS4 decrypt frame error\n",__FUNCTION__);\r
+ return _FAIL;\r
+ }\r
+\r
+ WAPI_TRACE(WAPI_RX, "<=========%s\n", __FUNCTION__);\r
+ return res;\r
+}\r
+\r
+#else\r
+\r
+u32 rtw_sms4_encrypt(_adapter *padapter, u8 *pxmitframe)\r
+{\r
+ WAPI_TRACE(WAPI_TX, "=========>Dummy %s\n", __FUNCTION__);\r
+ WAPI_TRACE(WAPI_TX, "<=========Dummy %s\n", __FUNCTION__);\r
+ return _SUCCESS;\r
+}\r
+\r
+u32 rtw_sms4_decrypt(_adapter *padapter, u8 *precvframe)\r
+{\r
+ WAPI_TRACE(WAPI_RX, "=========>Dummy %s\n", __FUNCTION__);\r
+ WAPI_TRACE(WAPI_RX, "<=========Dummy %s\n", __FUNCTION__);\r
+ return _SUCCESS;\r
+}\r
+\r
+#endif\r
+\r
+#endif\r