1 /******************************************************************************
2 *
3 * Copyright(c) 2016 - 2017 Realtek Corporation.
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of version 2 of the GNU General Public License as
7 * published by the Free Software Foundation.
8 *
9 * This program is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
12 * more details.
13 *
14 *****************************************************************************/
15 #ifdef CONFIG_WAPI_SUPPORT
16
17 #include <linux/unistd.h>
18 #include <linux/etherdevice.h>
19 #include <drv_types.h>
20 #include <rtw_wapi.h>
21
22
23 u32 wapi_debug_component =
24 /* WAPI_INIT |
25 * WAPI_API |
26 * WAPI_TX |
27 * WAPI_RX | */
28 WAPI_ERR ; /* always open err flags on */
29
WapiFreeAllStaInfo(_adapter * padapter)30 void WapiFreeAllStaInfo(_adapter *padapter)
31 {
32 PRT_WAPI_T pWapiInfo;
33 PRT_WAPI_STA_INFO pWapiStaInfo;
34 PRT_WAPI_BKID pWapiBkid;
35
36 WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);
37 pWapiInfo = &padapter->wapiInfo;
38
39 /* Pust to Idle List */
40 rtw_wapi_return_all_sta_info(padapter);
41
42 /* Sta Info List */
43 while (!list_empty(&(pWapiInfo->wapiSTAIdleList))) {
44 pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAIdleList.next, RT_WAPI_STA_INFO, list);
45 list_del_init(&pWapiStaInfo->list);
46 }
47
48 /* BKID List */
49 while (!list_empty(&(pWapiInfo->wapiBKIDIdleList))) {
50 pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDIdleList.next, RT_WAPI_BKID, list);
51 list_del_init(&pWapiBkid->list);
52 }
53 WAPI_TRACE(WAPI_INIT, "<=========== %s\n", __FUNCTION__);
54 return;
55 }
56
WapiSetIE(_adapter * padapter)57 void WapiSetIE(_adapter *padapter)
58 {
59 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
60 /* PRT_WAPI_BKID pWapiBkid; */
61 u16 protocolVer = 1;
62 u16 akmCnt = 1;
63 u16 suiteCnt = 1;
64 u16 capability = 0;
65 u8 OUI[3];
66
67 OUI[0] = 0x00;
68 OUI[1] = 0x14;
69 OUI[2] = 0x72;
70
71 pWapiInfo->wapiIELength = 0;
72 /* protocol version */
73 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, &protocolVer, 2);
74 pWapiInfo->wapiIELength += 2;
75 /* akm */
76 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, &akmCnt, 2);
77 pWapiInfo->wapiIELength += 2;
78
79 if (pWapiInfo->bWapiPSK) {
80 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, OUI, 3);
81 pWapiInfo->wapiIELength += 3;
82 pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x2;
83 pWapiInfo->wapiIELength += 1;
84 } else {
85 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, OUI, 3);
86 pWapiInfo->wapiIELength += 3;
87 pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x1;
88 pWapiInfo->wapiIELength += 1;
89 }
90
91 /* usk */
92 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, &suiteCnt, 2);
93 pWapiInfo->wapiIELength += 2;
94 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, OUI, 3);
95 pWapiInfo->wapiIELength += 3;
96 pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x1;
97 pWapiInfo->wapiIELength += 1;
98
99 /* msk */
100 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, OUI, 3);
101 pWapiInfo->wapiIELength += 3;
102 pWapiInfo->wapiIE[pWapiInfo->wapiIELength] = 0x1;
103 pWapiInfo->wapiIELength += 1;
104
105 /* Capbility */
106 _rtw_memcpy(pWapiInfo->wapiIE + pWapiInfo->wapiIELength, &capability, 2);
107 pWapiInfo->wapiIELength += 2;
108 }
109
110
111 /* PN1 > PN2, return 1,
112 * else return 0.
113 */
WapiComparePN(u8 * PN1,u8 * PN2)114 u32 WapiComparePN(u8 *PN1, u8 *PN2)
115 {
116 char i;
117
118 if ((NULL == PN1) || (NULL == PN2))
119 return 1;
120
121 /* overflow case */
122 if ((PN2[15] - PN1[15]) & 0x80)
123 return 1;
124
125 for (i = 16; i > 0; i--) {
126 if (PN1[i - 1] == PN2[i - 1])
127 continue;
128 else if (PN1[i - 1] > PN2[i - 1])
129 return 1;
130 else
131 return 0;
132 }
133
134 return 0;
135 }
136
137 u8
WapiGetEntryForCamWrite(_adapter * padapter,u8 * pMacAddr,u8 KID,BOOLEAN IsMsk)138 WapiGetEntryForCamWrite(_adapter *padapter, u8 *pMacAddr, u8 KID, BOOLEAN IsMsk)
139 {
140 PRT_WAPI_T pWapiInfo = NULL;
141 /* PRT_WAPI_CAM_ENTRY pEntry=NULL; */
142 u8 i = 0;
143 u8 ret = 0xff;
144
145 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
146
147 pWapiInfo = &padapter->wapiInfo;
148
149 /* exist? */
150 for (i = 0; i < WAPI_CAM_ENTRY_NUM; i++) {
151 if (pWapiInfo->wapiCamEntry[i].IsUsed
152 && (_rtw_memcmp(pMacAddr, pWapiInfo->wapiCamEntry[i].PeerMacAddr, ETH_ALEN) == _TRUE)
153 && pWapiInfo->wapiCamEntry[i].keyidx == KID
154 && pWapiInfo->wapiCamEntry[i].type == IsMsk) {
155 ret = pWapiInfo->wapiCamEntry[i].entry_idx; /* cover it */
156 break;
157 }
158 }
159
160 if (i == WAPI_CAM_ENTRY_NUM) { /* not found */
161 for (i = 0; i < WAPI_CAM_ENTRY_NUM; i++) {
162 if (pWapiInfo->wapiCamEntry[i].IsUsed == 0) {
163 pWapiInfo->wapiCamEntry[i].IsUsed = 1;
164 pWapiInfo->wapiCamEntry[i].type = IsMsk;
165 pWapiInfo->wapiCamEntry[i].keyidx = KID;
166 _rtw_memcpy(pWapiInfo->wapiCamEntry[i].PeerMacAddr, pMacAddr, ETH_ALEN);
167 ret = pWapiInfo->wapiCamEntry[i].entry_idx;
168 break;
169 }
170 }
171 }
172
173 WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);
174 return ret;
175
176 /*
177 if(RTIsListEmpty(&pWapiInfo->wapiCamIdleList)) {
178 return 0;
179 }
180
181 pEntry = (PRT_WAPI_CAM_ENTRY)RTRemoveHeadList(&pWapiInfo->wapiCamIdleList);
182 RTInsertTailList(&pWapiInfo->wapiCamUsedList, &pEntry->list);
183
184
185 return pEntry->entry_idx;*/
186 }
187
WapiGetEntryForCamClear(_adapter * padapter,u8 * pPeerMac,u8 keyid,u8 IsMsk)188 u8 WapiGetEntryForCamClear(_adapter *padapter, u8 *pPeerMac, u8 keyid, u8 IsMsk)
189 {
190 PRT_WAPI_T pWapiInfo = NULL;
191 u8 i = 0;
192
193 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
194
195 pWapiInfo = &padapter->wapiInfo;
196
197 for (i = 0; i < WAPI_CAM_ENTRY_NUM; i++) {
198 if (pWapiInfo->wapiCamEntry[i].IsUsed
199 && (_rtw_memcmp(pPeerMac, pWapiInfo->wapiCamEntry[i].PeerMacAddr, ETH_ALEN) == _TRUE)
200 && pWapiInfo->wapiCamEntry[i].keyidx == keyid
201 && pWapiInfo->wapiCamEntry[i].type == IsMsk) {
202 pWapiInfo->wapiCamEntry[i].IsUsed = 0;
203 pWapiInfo->wapiCamEntry[i].keyidx = 2;
204 _rtw_memset(pWapiInfo->wapiCamEntry[i].PeerMacAddr, 0, ETH_ALEN);
205
206 WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);
207 return pWapiInfo->wapiCamEntry[i].entry_idx;
208 }
209 }
210
211 WAPI_TRACE(WAPI_API, "<====WapiGetReturnCamEntry(), No this cam entry.\n");
212 return 0xff;
213 /*
214 if(RTIsListEmpty(&pWapiInfo->wapiCamUsedList)) {
215 return FALSE;
216 }
217
218 pList = &pWapiInfo->wapiCamUsedList;
219 while(pList->Flink != &pWapiInfo->wapiCamUsedList)
220 {
221 pEntry = (PRT_WAPI_CAM_ENTRY)pList->Flink;
222 if(PlatformCompareMemory(pPeerMac,pEntry->PeerMacAddr, ETHER_ADDRLEN)== 0
223 && keyid == pEntry->keyidx)
224 {
225 RTRemoveEntryList(pList);
226 RTInsertHeadList(&pWapiInfo->wapiCamIdleList, pList);
227 return pEntry->entry_idx;
228 }
229 pList = pList->Flink;
230 }
231
232 return 0;
233 */
234 }
235
236 void
WapiResetAllCamEntry(_adapter * padapter)237 WapiResetAllCamEntry(_adapter *padapter)
238 {
239 PRT_WAPI_T pWapiInfo;
240 int i;
241
242 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
243
244 pWapiInfo = &padapter->wapiInfo;
245
246 for (i = 0; i < WAPI_CAM_ENTRY_NUM; i++) {
247 _rtw_memset(pWapiInfo->wapiCamEntry[i].PeerMacAddr, 0, ETH_ALEN);
248 pWapiInfo->wapiCamEntry[i].IsUsed = 0;
249 pWapiInfo->wapiCamEntry[i].keyidx = 2; /* invalid */
250 pWapiInfo->wapiCamEntry[i].entry_idx = 4 + i * 2;
251 }
252
253 WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);
254
255 return;
256 }
257
WapiWriteOneCamEntry(_adapter * padapter,u8 * pMacAddr,u8 KeyId,u8 EntryId,u8 EncAlg,u8 bGroupKey,u8 * pKey)258 u8 WapiWriteOneCamEntry(
259 _adapter *padapter,
260 u8 *pMacAddr,
261 u8 KeyId,
262 u8 EntryId,
263 u8 EncAlg,
264 u8 bGroupKey,
265 u8 *pKey
266 )
267 {
268 u8 retVal = 0;
269 u16 usConfig = 0;
270
271 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
272
273 if (EntryId >= 32) {
274 WAPI_TRACE(WAPI_ERR, "<=== CamAddOneEntry(): ulKeyId exceed!\n");
275 return retVal;
276 }
277
278 usConfig = usConfig | (0x01 << 15) | ((u16)(EncAlg) << 2) | (KeyId);
279
280 if (EncAlg == _SMS4_) {
281 if (bGroupKey == 1)
282 usConfig |= (0x01 << 6);
283 if ((EntryId % 2) == 1) /* ==0 sec key; == 1mic key */
284 usConfig |= (0x01 << 5);
285 }
286
287 write_cam(padapter, EntryId, usConfig, pMacAddr, pKey);
288
289 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
290 return 1;
291 }
292
rtw_wapi_init(_adapter * padapter)293 void rtw_wapi_init(_adapter *padapter)
294 {
295 PRT_WAPI_T pWapiInfo;
296 int i;
297
298 WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);
299 RT_ASSERT_RET(padapter);
300
301 if (!padapter->WapiSupport) {
302 WAPI_TRACE(WAPI_INIT, "<========== %s, WAPI not supported!\n", __FUNCTION__);
303 return;
304 }
305
306 pWapiInfo = &padapter->wapiInfo;
307 pWapiInfo->bWapiEnable = false;
308
309 /* Init BKID List */
310 INIT_LIST_HEAD(&pWapiInfo->wapiBKIDIdleList);
311 INIT_LIST_HEAD(&pWapiInfo->wapiBKIDStoreList);
312 for (i = 0; i < WAPI_MAX_BKID_NUM; i++)
313 list_add_tail(&pWapiInfo->wapiBKID[i].list, &pWapiInfo->wapiBKIDIdleList);
314
315 /* Init STA List */
316 INIT_LIST_HEAD(&pWapiInfo->wapiSTAIdleList);
317 INIT_LIST_HEAD(&pWapiInfo->wapiSTAUsedList);
318 for (i = 0; i < WAPI_MAX_STAINFO_NUM; i++)
319 list_add_tail(&pWapiInfo->wapiSta[i].list, &pWapiInfo->wapiSTAIdleList);
320
321 for (i = 0; i < WAPI_CAM_ENTRY_NUM; i++) {
322 pWapiInfo->wapiCamEntry[i].IsUsed = 0;
323 pWapiInfo->wapiCamEntry[i].keyidx = 2; /* invalid */
324 pWapiInfo->wapiCamEntry[i].entry_idx = 4 + i * 2;
325 }
326
327 WAPI_TRACE(WAPI_INIT, "<========== %s\n", __FUNCTION__);
328 }
329
rtw_wapi_free(_adapter * padapter)330 void rtw_wapi_free(_adapter *padapter)
331 {
332 WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);
333 RT_ASSERT_RET(padapter);
334
335 if (!padapter->WapiSupport) {
336 WAPI_TRACE(WAPI_INIT, "<========== %s, WAPI not supported!\n", __FUNCTION__);
337 return;
338 }
339
340 WapiFreeAllStaInfo(padapter);
341
342 WAPI_TRACE(WAPI_INIT, "<========== %s\n", __FUNCTION__);
343 }
344
rtw_wapi_disable_tx(_adapter * padapter)345 void rtw_wapi_disable_tx(_adapter *padapter)
346 {
347 WAPI_TRACE(WAPI_INIT, "===========> %s\n", __FUNCTION__);
348 RT_ASSERT_RET(padapter);
349
350 if (!padapter->WapiSupport) {
351 WAPI_TRACE(WAPI_INIT, "<========== %s, WAPI not supported!\n", __FUNCTION__);
352 return;
353 }
354
355 padapter->wapiInfo.wapiTxMsk.bTxEnable = false;
356 padapter->wapiInfo.wapiTxMsk.bSet = false;
357
358 WAPI_TRACE(WAPI_INIT, "<========== %s\n", __FUNCTION__);
359 }
360
rtw_wapi_is_wai_packet(_adapter * padapter,u8 * pkt_data)361 u8 rtw_wapi_is_wai_packet(_adapter *padapter, u8 *pkt_data)
362 {
363 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
364 struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
365 struct security_priv *psecuritypriv = &padapter->securitypriv;
366 PRT_WAPI_STA_INFO pWapiSta = NULL;
367 u8 WaiPkt = 0, *pTaddr, bFind = false;
368 u8 Offset_TypeWAI = 0 ; /* (mac header len + llc length) */
369
370 WAPI_TRACE(WAPI_TX | WAPI_RX, "===========> %s\n", __FUNCTION__);
371
372 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
373 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
374 return 0;
375 }
376
377 Offset_TypeWAI = 24 + 6 ;
378
379 /* YJ,add,091103. Data frame may also have skb->data[30]=0x88 and skb->data[31]=0xb4. */
380 if ((pkt_data[1] & 0x40) != 0) {
381 /* RTW_INFO("data is privacy\n"); */
382 return 0;
383 }
384
385 pTaddr = get_addr2_ptr(pkt_data);
386 if (list_empty(&pWapiInfo->wapiSTAUsedList))
387 bFind = false;
388 else {
389 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
390 if (_rtw_memcmp(pTaddr, pWapiSta->PeerMacAddr, 6) == _TRUE) {
391 bFind = true;
392 break;
393 }
394 }
395 }
396
397 WAPI_TRACE(WAPI_TX | WAPI_RX, "%s: bFind=%d pTaddr="MAC_FMT"\n", __FUNCTION__, bFind, MAC_ARG(pTaddr));
398
399 if (pkt_data[0] == WIFI_QOS_DATA_TYPE)
400 Offset_TypeWAI += 2;
401
402 /* 88b4? */
403 if ((pkt_data[Offset_TypeWAI] == 0x88) && (pkt_data[Offset_TypeWAI + 1] == 0xb4)) {
404 WaiPkt = pkt_data[Offset_TypeWAI + 5];
405
406 psecuritypriv->hw_decrypted = _TRUE;
407 } else
408 WAPI_TRACE(WAPI_TX | WAPI_RX, "%s(): non wai packet\n", __FUNCTION__);
409
410 WAPI_TRACE(WAPI_TX | WAPI_RX, "%s(): Recvd WAI frame. IsWAIPkt(%d)\n", __FUNCTION__, WaiPkt);
411
412 return WaiPkt;
413 }
414
415
rtw_wapi_update_info(_adapter * padapter,union recv_frame * precv_frame)416 void rtw_wapi_update_info(_adapter *padapter, union recv_frame *precv_frame)
417 {
418 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
419 struct recv_frame_hdr *precv_hdr;
420 u8 *ptr;
421 u8 *pTA;
422 u8 *pRecvPN;
423
424
425 WAPI_TRACE(WAPI_RX, "===========> %s\n", __FUNCTION__);
426
427 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
428 WAPI_TRACE(WAPI_RX, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
429 return;
430 }
431
432 precv_hdr = &precv_frame->u.hdr;
433 ptr = precv_hdr->rx_data;
434
435 if (precv_hdr->attrib.qos == 1)
436 precv_hdr->UserPriority = GetTid(ptr);
437 else
438 precv_hdr->UserPriority = 0;
439
440 pTA = get_addr2_ptr(ptr);
441 _rtw_memcpy((u8 *)precv_hdr->WapiSrcAddr, pTA, 6);
442 pRecvPN = ptr + precv_hdr->attrib.hdrlen + 2;
443 _rtw_memcpy((u8 *)precv_hdr->WapiTempPN, pRecvPN, 16);
444
445 WAPI_TRACE(WAPI_RX, "<========== %s\n", __FUNCTION__);
446 }
447
448 /****************************************************************************
449 TRUE-----------------Drop
450 FALSE---------------- handle
451 add to support WAPI to N-mode
452 *****************************************************************************/
rtw_wapi_check_for_drop(_adapter * padapter,union recv_frame * precv_frame,u8 * ehdr_ops)453 u8 rtw_wapi_check_for_drop(
454 _adapter *padapter,
455 union recv_frame *precv_frame,
456 u8 *ehdr_ops
457 )
458 {
459 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
460 u8 *pLastRecvPN = NULL;
461 u8 bFind = false;
462 PRT_WAPI_STA_INFO pWapiSta = NULL;
463 u8 bDrop = false;
464 struct recv_frame_hdr *precv_hdr = &precv_frame->u.hdr;
465 u8 WapiAEPNInitialValueSrc[16] = {0x37, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
466 u8 WapiAEMultiCastPNInitialValueSrc[16] = {0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
467 u8 *ptr = ehdr_ops;
468 int i;
469
470 WAPI_TRACE(WAPI_RX, "===========> %s\n", __FUNCTION__);
471
472 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
473 WAPI_TRACE(WAPI_RX, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
474 return false;
475 }
476
477 if (precv_hdr->bIsWaiPacket != 0) {
478 if (precv_hdr->bIsWaiPacket == 0x8) {
479
480 RTW_INFO("rtw_wapi_check_for_drop: dump packet\n");
481 for (i = 0; i < 50; i++) {
482 RTW_INFO("%02X ", ptr[i]);
483 if ((i + 1) % 8 == 0)
484 RTW_INFO("\n");
485 }
486 RTW_INFO("\n rtw_wapi_check_for_drop: dump packet\n");
487
488 for (i = 0; i < 16; i++) {
489 if (ptr[i + 27] != 0)
490 break;
491 }
492
493 if (i == 16) {
494 WAPI_TRACE(WAPI_RX, "rtw_wapi_check_for_drop: drop with zero BKID\n");
495 return true;
496 } else
497 return false;
498 } else
499 return false;
500 }
501
502 if (list_empty(&pWapiInfo->wapiSTAUsedList))
503 bFind = false;
504 else {
505 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
506 if (_rtw_memcmp(precv_hdr->WapiSrcAddr, pWapiSta->PeerMacAddr, ETH_ALEN) == _TRUE) {
507 bFind = true;
508 break;
509 }
510 }
511 }
512 WAPI_TRACE(WAPI_RX, "%s: bFind=%d prxb->WapiSrcAddr="MAC_FMT"\n", __FUNCTION__, bFind, MAC_ARG(precv_hdr->WapiSrcAddr));
513
514 if (bFind) {
515 if (IS_MCAST(precv_hdr->attrib.ra)) {
516 WAPI_TRACE(WAPI_RX, "rtw_wapi_check_for_drop: multicast case\n");
517 pLastRecvPN = pWapiSta->lastRxMulticastPN;
518 } else {
519 WAPI_TRACE(WAPI_RX, "rtw_wapi_check_for_drop: unicast case\n");
520 switch (precv_hdr->UserPriority) {
521 case 0:
522 case 3:
523 pLastRecvPN = pWapiSta->lastRxUnicastPNBEQueue;
524 break;
525 case 1:
526 case 2:
527 pLastRecvPN = pWapiSta->lastRxUnicastPNBKQueue;
528 break;
529 case 4:
530 case 5:
531 pLastRecvPN = pWapiSta->lastRxUnicastPNVIQueue;
532 break;
533 case 6:
534 case 7:
535 pLastRecvPN = pWapiSta->lastRxUnicastPNVOQueue;
536 break;
537 default:
538 WAPI_TRACE(WAPI_ERR, "%s: Unknown TID\n", __FUNCTION__);
539 break;
540 }
541 }
542
543 if (!WapiComparePN(precv_hdr->WapiTempPN, pLastRecvPN)) {
544 WAPI_TRACE(WAPI_RX, "%s: Equal PN!!\n", __FUNCTION__);
545 if (IS_MCAST(precv_hdr->attrib.ra))
546 _rtw_memcpy(pLastRecvPN, WapiAEMultiCastPNInitialValueSrc, 16);
547 else
548 _rtw_memcpy(pLastRecvPN, WapiAEPNInitialValueSrc, 16);
549 bDrop = true;
550 } else
551 _rtw_memcpy(pLastRecvPN, precv_hdr->WapiTempPN, 16);
552 }
553
554 WAPI_TRACE(WAPI_RX, "<========== %s\n", __FUNCTION__);
555 return bDrop;
556 }
557
rtw_build_probe_resp_wapi_ie(_adapter * padapter,unsigned char * pframe,struct pkt_attrib * pattrib)558 void rtw_build_probe_resp_wapi_ie(_adapter *padapter, unsigned char *pframe, struct pkt_attrib *pattrib)
559 {
560 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
561 u8 WapiIELength = 0;
562
563 WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);
564
565 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
566 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported!\n", __FUNCTION__);
567 return;
568 }
569
570 WapiSetIE(padapter);
571 WapiIELength = pWapiInfo->wapiIELength;
572 pframe[0] = _WAPI_IE_;
573 pframe[1] = WapiIELength;
574 _rtw_memcpy(pframe + 2, pWapiInfo->wapiIE, WapiIELength);
575 pframe += WapiIELength + 2;
576 pattrib->pktlen += WapiIELength + 2;
577
578 WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);
579 }
580
rtw_build_beacon_wapi_ie(_adapter * padapter,unsigned char * pframe,struct pkt_attrib * pattrib)581 void rtw_build_beacon_wapi_ie(_adapter *padapter, unsigned char *pframe, struct pkt_attrib *pattrib)
582 {
583 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
584 u8 WapiIELength = 0;
585 WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);
586
587 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
588 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported!\n", __FUNCTION__);
589 return;
590 }
591
592 WapiSetIE(padapter);
593 WapiIELength = pWapiInfo->wapiIELength;
594 pframe[0] = _WAPI_IE_;
595 pframe[1] = WapiIELength;
596 _rtw_memcpy(pframe + 2, pWapiInfo->wapiIE, WapiIELength);
597 pframe += WapiIELength + 2;
598 pattrib->pktlen += WapiIELength + 2;
599
600 WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);
601 }
602
rtw_build_assoc_req_wapi_ie(_adapter * padapter,unsigned char * pframe,struct pkt_attrib * pattrib)603 void rtw_build_assoc_req_wapi_ie(_adapter *padapter, unsigned char *pframe, struct pkt_attrib *pattrib)
604 {
605 PRT_WAPI_BKID pWapiBKID;
606 u16 bkidNum;
607 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
608 u8 WapiIELength = 0;
609
610 WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);
611
612 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
613 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported!\n", __FUNCTION__);
614 return;
615 }
616
617 WapiSetIE(padapter);
618 WapiIELength = pWapiInfo->wapiIELength;
619 bkidNum = 0;
620 if (!list_empty(&(pWapiInfo->wapiBKIDStoreList))) {
621 list_for_each_entry(pWapiBKID, &pWapiInfo->wapiBKIDStoreList, list) {
622 bkidNum++;
623 _rtw_memcpy(pWapiInfo->wapiIE + WapiIELength + 2, pWapiBKID->bkid, 16);
624 WapiIELength += 16;
625 }
626 }
627 _rtw_memcpy(pWapiInfo->wapiIE + WapiIELength, &bkidNum, 2);
628 WapiIELength += 2;
629
630 pframe[0] = _WAPI_IE_;
631 pframe[1] = WapiIELength;
632 _rtw_memcpy(pframe + 2, pWapiInfo->wapiIE, WapiIELength);
633 pframe += WapiIELength + 2;
634 pattrib->pktlen += WapiIELength + 2;
635 WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);
636 }
637
rtw_wapi_on_assoc_ok(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)638 void rtw_wapi_on_assoc_ok(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
639 {
640 PRT_WAPI_T pWapiInfo = &(padapter->wapiInfo);
641 PRT_WAPI_STA_INFO pWapiSta;
642 u8 WapiAEPNInitialValueSrc[16] = {0x37, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
643 /* u8 WapiASUEPNInitialValueSrc[16] = {0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C,0x36,0x5C} ; */
644 u8 WapiAEMultiCastPNInitialValueSrc[16] = {0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
645
646 WAPI_TRACE(WAPI_MLME, "===========> %s\n", __FUNCTION__);
647
648 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
649 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
650 return;
651 }
652
653 pWapiSta = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAIdleList.next, RT_WAPI_STA_INFO, list);
654 list_del_init(&pWapiSta->list);
655 list_add_tail(&pWapiSta->list, &pWapiInfo->wapiSTAUsedList);
656 _rtw_memcpy(pWapiSta->PeerMacAddr, padapter->mlmeextpriv.mlmext_info.network.MacAddress, 6);
657 _rtw_memcpy(pWapiSta->lastRxMulticastPN, WapiAEMultiCastPNInitialValueSrc, 16);
658 _rtw_memcpy(pWapiSta->lastRxUnicastPN, WapiAEPNInitialValueSrc, 16);
659
660 /* For chenk PN error with Qos Data after s3: add by ylb 20111114 */
661 _rtw_memcpy(pWapiSta->lastRxUnicastPNBEQueue, WapiAEPNInitialValueSrc, 16);
662 _rtw_memcpy(pWapiSta->lastRxUnicastPNBKQueue, WapiAEPNInitialValueSrc, 16);
663 _rtw_memcpy(pWapiSta->lastRxUnicastPNVIQueue, WapiAEPNInitialValueSrc, 16);
664 _rtw_memcpy(pWapiSta->lastRxUnicastPNVOQueue, WapiAEPNInitialValueSrc, 16);
665
666 WAPI_TRACE(WAPI_MLME, "<========== %s\n", __FUNCTION__);
667 }
668
669
rtw_wapi_return_one_sta_info(_adapter * padapter,u8 * MacAddr)670 void rtw_wapi_return_one_sta_info(_adapter *padapter, u8 *MacAddr)
671 {
672 PRT_WAPI_T pWapiInfo;
673 PRT_WAPI_STA_INFO pWapiStaInfo = NULL;
674 PRT_WAPI_BKID pWapiBkid = NULL;
675 struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
676
677 pWapiInfo = &padapter->wapiInfo;
678
679 WAPI_TRACE(WAPI_API, "==========> %s\n", __FUNCTION__);
680
681 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
682 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
683 return;
684 }
685
686 if (MLME_IS_STA(padapter)) {
687 while (!list_empty(&(pWapiInfo->wapiBKIDStoreList))) {
688 pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDStoreList.next, RT_WAPI_BKID, list);
689 list_del_init(&pWapiBkid->list);
690 _rtw_memset(pWapiBkid->bkid, 0, 16);
691 list_add_tail(&pWapiBkid->list, &pWapiInfo->wapiBKIDIdleList);
692 }
693 }
694
695
696 WAPI_TRACE(WAPI_API, " %s: after clear bkid\n", __FUNCTION__);
697
698
699 /* Remove STA info */
700 if (list_empty(&(pWapiInfo->wapiSTAUsedList))) {
701 WAPI_TRACE(WAPI_API, " %s: wapiSTAUsedList is null\n", __FUNCTION__);
702 return;
703 } else {
704
705 WAPI_TRACE(WAPI_API, " %s: wapiSTAUsedList is not null\n", __FUNCTION__);
706 #if 0
707 pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry((pWapiInfo->wapiSTAUsedList.next), RT_WAPI_STA_INFO, list);
708
709 list_for_each_entry(pWapiStaInfo, &(pWapiInfo->wapiSTAUsedList), list) {
710
711 RTW_INFO("MAC Addr %02x-%02x-%02x-%02x-%02x-%02x\n", MacAddr[0], MacAddr[1], MacAddr[2], MacAddr[3], MacAddr[4], MacAddr[5]);
712
713
714 RTW_INFO("peer Addr %02x-%02x-%02x-%02x-%02x-%02x\n", pWapiStaInfo->PeerMacAddr[0], pWapiStaInfo->PeerMacAddr[1], pWapiStaInfo->PeerMacAddr[2], pWapiStaInfo->PeerMacAddr[3],
715 pWapiStaInfo->PeerMacAddr[4], pWapiStaInfo->PeerMacAddr[5]);
716
717 if (pWapiStaInfo == NULL) {
718 WAPI_TRACE(WAPI_API, " %s: pWapiStaInfo == NULL Case\n", __FUNCTION__);
719 return;
720 }
721
722 if (pWapiStaInfo->PeerMacAddr == NULL) {
723 WAPI_TRACE(WAPI_API, " %s: pWapiStaInfo->PeerMacAddr == NULL Case\n", __FUNCTION__);
724 return;
725 }
726
727 if (MacAddr == NULL) {
728 WAPI_TRACE(WAPI_API, " %s: MacAddr == NULL Case\n", __FUNCTION__);
729 return;
730 }
731
732 if (_rtw_memcmp(pWapiStaInfo->PeerMacAddr, MacAddr, ETH_ALEN) == _TRUE) {
733 pWapiStaInfo->bAuthenticateInProgress = false;
734 pWapiStaInfo->bSetkeyOk = false;
735 _rtw_memset(pWapiStaInfo->PeerMacAddr, 0, ETH_ALEN);
736 list_del_init(&pWapiStaInfo->list);
737 list_add_tail(&pWapiStaInfo->list, &pWapiInfo->wapiSTAIdleList);
738 break;
739 }
740
741 }
742 #endif
743
744 while (!list_empty(&(pWapiInfo->wapiSTAUsedList))) {
745 pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAUsedList.next, RT_WAPI_STA_INFO, list);
746
747 RTW_INFO("peer Addr %02x-%02x-%02x-%02x-%02x-%02x\n", pWapiStaInfo->PeerMacAddr[0], pWapiStaInfo->PeerMacAddr[1], pWapiStaInfo->PeerMacAddr[2], pWapiStaInfo->PeerMacAddr[3],
748 pWapiStaInfo->PeerMacAddr[4], pWapiStaInfo->PeerMacAddr[5]);
749
750 list_del_init(&pWapiStaInfo->list);
751 _rtw_memset(pWapiStaInfo->PeerMacAddr, 0, ETH_ALEN);
752 pWapiStaInfo->bSetkeyOk = 0;
753 list_add_tail(&pWapiStaInfo->list, &pWapiInfo->wapiSTAIdleList);
754 }
755
756 }
757
758 WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);
759 return;
760 }
761
rtw_wapi_return_all_sta_info(_adapter * padapter)762 void rtw_wapi_return_all_sta_info(_adapter *padapter)
763 {
764 PRT_WAPI_T pWapiInfo;
765 PRT_WAPI_STA_INFO pWapiStaInfo;
766 PRT_WAPI_BKID pWapiBkid;
767 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
768
769 pWapiInfo = &padapter->wapiInfo;
770
771 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable)) {
772 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
773 return;
774 }
775
776 /* Sta Info List */
777 while (!list_empty(&(pWapiInfo->wapiSTAUsedList))) {
778 pWapiStaInfo = (PRT_WAPI_STA_INFO)list_entry(pWapiInfo->wapiSTAUsedList.next, RT_WAPI_STA_INFO, list);
779 list_del_init(&pWapiStaInfo->list);
780 _rtw_memset(pWapiStaInfo->PeerMacAddr, 0, ETH_ALEN);
781 pWapiStaInfo->bSetkeyOk = 0;
782 list_add_tail(&pWapiStaInfo->list, &pWapiInfo->wapiSTAIdleList);
783 }
784
785 /* BKID List */
786 while (!list_empty(&(pWapiInfo->wapiBKIDStoreList))) {
787 pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDStoreList.next, RT_WAPI_BKID, list);
788 list_del_init(&pWapiBkid->list);
789 _rtw_memset(pWapiBkid->bkid, 0, 16);
790 list_add_tail(&pWapiBkid->list, &pWapiInfo->wapiBKIDIdleList);
791 }
792 WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);
793 }
794
CAM_empty_entry(_adapter * adapter,u8 ucIndex)795 void CAM_empty_entry(
796 _adapter *adapter,
797 u8 ucIndex
798 )
799 {
800 rtw_hal_set_hwreg(adapter, HW_VAR_CAM_EMPTY_ENTRY, (u8 *)(&ucIndex));
801 }
802
rtw_wapi_clear_cam_entry(_adapter * padapter,u8 * pMacAddr)803 void rtw_wapi_clear_cam_entry(_adapter *padapter, u8 *pMacAddr)
804 {
805 u8 UcIndex = 0;
806
807 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
808
809 if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable)) {
810 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
811 return;
812 }
813
814 UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 0, 0);
815 if (UcIndex != 0xff) {
816 /* CAM_mark_invalid(adapter, UcIndex); */
817 CAM_empty_entry(padapter, UcIndex);
818 }
819
820 UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 1, 0);
821 if (UcIndex != 0xff) {
822 /* CAM_mark_invalid(adapter, UcIndex); */
823 CAM_empty_entry(padapter, UcIndex);
824 }
825
826 UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 0, 1);
827 if (UcIndex != 0xff) {
828 /* CAM_mark_invalid(adapter, UcIndex); */
829 CAM_empty_entry(padapter, UcIndex);
830 }
831
832 UcIndex = WapiGetEntryForCamClear(padapter, pMacAddr, 1, 1);
833 if (UcIndex != 0xff) {
834 /* CAM_mark_invalid(padapter, UcIndex); */
835 CAM_empty_entry(padapter, UcIndex);
836 }
837
838 WAPI_TRACE(WAPI_API, "<========== %s\n", __FUNCTION__);
839 }
840
rtw_wapi_clear_all_cam_entry(_adapter * padapter)841 void rtw_wapi_clear_all_cam_entry(_adapter *padapter)
842 {
843 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
844
845 if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable)) {
846 WAPI_TRACE(WAPI_MLME, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
847 return;
848 }
849
850 invalidate_cam_all(padapter); /* is this ok? */
851 WapiResetAllCamEntry(padapter);
852
853 WAPI_TRACE(WAPI_API, "<=========== %s\n", __FUNCTION__);
854 }
855
rtw_wapi_set_key(_adapter * padapter,RT_WAPI_KEY * pWapiKey,RT_WAPI_STA_INFO * pWapiSta,u8 bGroupKey)856 void rtw_wapi_set_key(_adapter *padapter, RT_WAPI_KEY *pWapiKey, RT_WAPI_STA_INFO *pWapiSta, u8 bGroupKey)
857 {
858 struct setkey_parm *psetkeyparm;
859 struct set_stakey_parm *psetstakeyparm;
860 PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;
861 u8 *pMacAddr = pWapiSta->PeerMacAddr;
862 u32 EntryId = 0;
863 BOOLEAN IsPairWise = false ;
864 u8 keylen;
865
866 WAPI_TRACE(WAPI_API, "===========> %s\n", __FUNCTION__);
867
868 if ((!padapter->WapiSupport) || (!padapter->wapiInfo.bWapiEnable)) {
869 WAPI_TRACE(WAPI_API, "<========== %s, WAPI not supported or not enabled!\n", __FUNCTION__);
870 return;
871 }
872
873 psetstakeyparm = (struct set_stakey_parm *)rtw_zmalloc(sizeof(struct set_stakey_parm));
874 if (NULL == psetstakeyparm) {
875 WAPI_TRACE(WAPI_API, "<========== new %s, could not zmalloc psetstakeyparm at line %d\n", __FUNCTION__, __LINE__);
876 return;
877 }
878 psetstakeyparm->algorithm = _SMS4_;
879 keylen = 16;
880
881 _rtw_memcpy(psetstakeyparm->addr, pMacAddr, ETH_ALEN);
882 _rtw_memcpy(&(psetstakeyparm->key[0]), &pWapiKey->dataKey, keylen);
883 _rtw_memcpy(&(psetstakeyparm->key[16]), &pWapiKey->micKey, keylen);
884 psetstakeyparm->keyid = pWapiKey->keyId;
885
886 if(bGroupKey) {
887 /* Group Key */
888 psetstakeyparm->gk = 1;
889 WAPI_TRACE(WAPI_API, "new %s: group keyid = %d, gk = %d, algorithm = %d\n", __FUNCTION__, psetstakeyparm->keyid, psetstakeyparm->gk, psetstakeyparm->algorithm);
890 } else {
891 /* Pairwise Key */
892 psetstakeyparm->gk = 0;
893 WAPI_TRACE(WAPI_API, "new %s: pairwise keyid = %d, gk = %d, algorithm = %d\n", __FUNCTION__, psetstakeyparm->keyid, psetstakeyparm->gk, psetstakeyparm->algorithm);
894 }
895
896 WAPI_DATA(WAPI_API, "new %s: ", psetstakeyparm->key, 32);
897 #ifdef CONFIG_CMD_DISP
898 set_stakey_hdl(padapter, psetstakeyparm, PHL_CMD_DIRECTLY, 0);
899 #else
900 set_stakey_hdl(padapter, (u8 *)psetstakeyparm);
901 #endif
902 rtw_mfree((u8 *) psetstakeyparm, sizeof(struct setkey_parm));
903 WAPI_TRACE(WAPI_API, "Set Wapi Key :KeyId:%d,EntryId:%d,PairwiseKey:%d.\n", pWapiKey->keyId, EntryId, !bGroupKey);
904 WAPI_TRACE(WAPI_API, "<=========== %s\n", __FUNCTION__);
905 }
906
907 #if 0
908 /* YJ,test,091013 */
909 void wapi_test_set_key(_adapter *padapter, u8 *buf)
910 {
911 /*Data: keyType(1) + bTxEnable(1) + bAuthenticator(1) + bUpdate(1) + PeerAddr(6) + DataKey(16) + MicKey(16) + KeyId(1)*/
912 PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;
913 PRT_WAPI_BKID pWapiBkid;
914 PRT_WAPI_STA_INFO pWapiSta;
915 u8 data[43];
916 bool bTxEnable;
917 bool bUpdate;
918 bool bAuthenticator;
919 u8 PeerAddr[6];
920 u8 WapiAEPNInitialValueSrc[16] = {0x37, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
921 u8 WapiASUEPNInitialValueSrc[16] = {0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
922 u8 WapiAEMultiCastPNInitialValueSrc[16] = {0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
923
924 WAPI_TRACE(WAPI_INIT, "===========>%s\n", __FUNCTION__);
925
926 if (!padapter->WapiSupport)
927 return;
928
929 copy_from_user(data, buf, 43);
930 bTxEnable = data[1];
931 bAuthenticator = data[2];
932 bUpdate = data[3];
933 _rtw_memcpy(PeerAddr, data + 4, 6);
934
935 if (data[0] == 0x3) {
936 if (!list_empty(&(pWapiInfo->wapiBKIDIdleList))) {
937 pWapiBkid = (PRT_WAPI_BKID)list_entry(pWapiInfo->wapiBKIDIdleList.next, RT_WAPI_BKID, list);
938 list_del_init(&pWapiBkid->list);
939 _rtw_memcpy(pWapiBkid->bkid, data + 10, 16);
940 WAPI_DATA(WAPI_INIT, "SetKey - BKID", pWapiBkid->bkid, 16);
941 list_add_tail(&pWapiBkid->list, &pWapiInfo->wapiBKIDStoreList);
942 }
943 } else {
944 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
945 if (_rtw_memcmp(pWapiSta->PeerMacAddr, PeerAddr, 6)) {
946 pWapiSta->bAuthenticatorInUpdata = false;
947 switch (data[0]) {
948 case 1: /* usk */
949 if (bAuthenticator) { /* authenticator */
950 _rtw_memcpy(pWapiSta->lastTxUnicastPN, WapiAEPNInitialValueSrc, 16);
951 if (!bUpdate) { /* first */
952 WAPI_TRACE(WAPI_INIT, "AE fisrt set usk\n");
953 pWapiSta->wapiUsk.bSet = true;
954 _rtw_memcpy(pWapiSta->wapiUsk.dataKey, data + 10, 16);
955 _rtw_memcpy(pWapiSta->wapiUsk.micKey, data + 26, 16);
956 pWapiSta->wapiUsk.keyId = *(data + 42);
957 pWapiSta->wapiUsk.bTxEnable = true;
958 WAPI_DATA(WAPI_INIT, "SetKey - AE USK Data Key", pWapiSta->wapiUsk.dataKey, 16);
959 WAPI_DATA(WAPI_INIT, "SetKey - AE USK Mic Key", pWapiSta->wapiUsk.micKey, 16);
960 } else { /* update */
961 WAPI_TRACE(WAPI_INIT, "AE update usk\n");
962 pWapiSta->wapiUskUpdate.bSet = true;
963 pWapiSta->bAuthenticatorInUpdata = true;
964 _rtw_memcpy(pWapiSta->wapiUskUpdate.dataKey, data + 10, 16);
965 _rtw_memcpy(pWapiSta->wapiUskUpdate.micKey, data + 26, 16);
966 _rtw_memcpy(pWapiSta->lastRxUnicastPNBEQueue, WapiASUEPNInitialValueSrc, 16);
967 _rtw_memcpy(pWapiSta->lastRxUnicastPNBKQueue, WapiASUEPNInitialValueSrc, 16);
968 _rtw_memcpy(pWapiSta->lastRxUnicastPNVIQueue, WapiASUEPNInitialValueSrc, 16);
969 _rtw_memcpy(pWapiSta->lastRxUnicastPNVOQueue, WapiASUEPNInitialValueSrc, 16);
970 _rtw_memcpy(pWapiSta->lastRxUnicastPN, WapiASUEPNInitialValueSrc, 16);
971 pWapiSta->wapiUskUpdate.keyId = *(data + 42);
972 pWapiSta->wapiUskUpdate.bTxEnable = true;
973 }
974 } else {
975 if (!bUpdate) {
976 WAPI_TRACE(WAPI_INIT, "ASUE fisrt set usk\n");
977 if (bTxEnable) {
978 pWapiSta->wapiUsk.bTxEnable = true;
979 _rtw_memcpy(pWapiSta->lastTxUnicastPN, WapiASUEPNInitialValueSrc, 16);
980 } else {
981 pWapiSta->wapiUsk.bSet = true;
982 _rtw_memcpy(pWapiSta->wapiUsk.dataKey, data + 10, 16);
983 _rtw_memcpy(pWapiSta->wapiUsk.micKey, data + 26, 16);
984 pWapiSta->wapiUsk.keyId = *(data + 42);
985 pWapiSta->wapiUsk.bTxEnable = false;
986 }
987 } else {
988 WAPI_TRACE(WAPI_INIT, "ASUE update usk\n");
989 if (bTxEnable) {
990 pWapiSta->wapiUskUpdate.bTxEnable = true;
991 if (pWapiSta->wapiUskUpdate.bSet) {
992 _rtw_memcpy(pWapiSta->wapiUsk.dataKey, pWapiSta->wapiUskUpdate.dataKey, 16);
993 _rtw_memcpy(pWapiSta->wapiUsk.micKey, pWapiSta->wapiUskUpdate.micKey, 16);
994 pWapiSta->wapiUsk.keyId = pWapiSta->wapiUskUpdate.keyId;
995 _rtw_memcpy(pWapiSta->lastRxUnicastPNBEQueue, WapiASUEPNInitialValueSrc, 16);
996 _rtw_memcpy(pWapiSta->lastRxUnicastPNBKQueue, WapiASUEPNInitialValueSrc, 16);
997 _rtw_memcpy(pWapiSta->lastRxUnicastPNVIQueue, WapiASUEPNInitialValueSrc, 16);
998 _rtw_memcpy(pWapiSta->lastRxUnicastPNVOQueue, WapiASUEPNInitialValueSrc, 16);
999 _rtw_memcpy(pWapiSta->lastRxUnicastPN, WapiASUEPNInitialValueSrc, 16);
1000 pWapiSta->wapiUskUpdate.bTxEnable = false;
1001 pWapiSta->wapiUskUpdate.bSet = false;
1002 }
1003 _rtw_memcpy(pWapiSta->lastTxUnicastPN, WapiASUEPNInitialValueSrc, 16);
1004 } else {
1005 pWapiSta->wapiUskUpdate.bSet = true;
1006 _rtw_memcpy(pWapiSta->wapiUskUpdate.dataKey, data + 10, 16);
1007 _rtw_memcpy(pWapiSta->wapiUskUpdate.micKey, data + 26, 16);
1008 pWapiSta->wapiUskUpdate.keyId = *(data + 42);
1009 pWapiSta->wapiUskUpdate.bTxEnable = false;
1010 }
1011 }
1012 }
1013 break;
1014 case 2: /* msk */
1015 if (bAuthenticator) { /* authenticator */
1016 pWapiInfo->wapiTxMsk.bSet = true;
1017 _rtw_memcpy(pWapiInfo->wapiTxMsk.dataKey, data + 10, 16);
1018 _rtw_memcpy(pWapiInfo->wapiTxMsk.micKey, data + 26, 16);
1019 pWapiInfo->wapiTxMsk.keyId = *(data + 42);
1020 pWapiInfo->wapiTxMsk.bTxEnable = true;
1021 _rtw_memcpy(pWapiInfo->lastTxMulticastPN, WapiAEMultiCastPNInitialValueSrc, 16);
1022
1023 if (!bUpdate) { /* first */
1024 WAPI_TRACE(WAPI_INIT, "AE fisrt set msk\n");
1025 if (!pWapiSta->bSetkeyOk)
1026 pWapiSta->bSetkeyOk = true;
1027 pWapiInfo->bFirstAuthentiateInProgress = false;
1028 } else /* update */
1029 WAPI_TRACE(WAPI_INIT, "AE update msk\n");
1030
1031 WAPI_DATA(WAPI_INIT, "SetKey - AE MSK Data Key", pWapiInfo->wapiTxMsk.dataKey, 16);
1032 WAPI_DATA(WAPI_INIT, "SetKey - AE MSK Mic Key", pWapiInfo->wapiTxMsk.micKey, 16);
1033 } else {
1034 if (!bUpdate) {
1035 WAPI_TRACE(WAPI_INIT, "ASUE fisrt set msk\n");
1036 pWapiSta->wapiMsk.bSet = true;
1037 _rtw_memcpy(pWapiSta->wapiMsk.dataKey, data + 10, 16);
1038 _rtw_memcpy(pWapiSta->wapiMsk.micKey, data + 26, 16);
1039 pWapiSta->wapiMsk.keyId = *(data + 42);
1040 pWapiSta->wapiMsk.bTxEnable = false;
1041 if (!pWapiSta->bSetkeyOk)
1042 pWapiSta->bSetkeyOk = true;
1043 pWapiInfo->bFirstAuthentiateInProgress = false;
1044 WAPI_DATA(WAPI_INIT, "SetKey - ASUE MSK Data Key", pWapiSta->wapiMsk.dataKey, 16);
1045 WAPI_DATA(WAPI_INIT, "SetKey - ASUE MSK Mic Key", pWapiSta->wapiMsk.micKey, 16);
1046 } else {
1047 WAPI_TRACE(WAPI_INIT, "ASUE update msk\n");
1048 pWapiSta->wapiMskUpdate.bSet = true;
1049 _rtw_memcpy(pWapiSta->wapiMskUpdate.dataKey, data + 10, 16);
1050 _rtw_memcpy(pWapiSta->wapiMskUpdate.micKey, data + 26, 16);
1051 pWapiSta->wapiMskUpdate.keyId = *(data + 42);
1052 pWapiSta->wapiMskUpdate.bTxEnable = false;
1053 }
1054 }
1055 break;
1056 default:
1057 WAPI_TRACE(WAPI_ERR, "Unknown Flag\n");
1058 break;
1059 }
1060 }
1061 }
1062 }
1063 WAPI_TRACE(WAPI_INIT, "<===========%s\n", __FUNCTION__);
1064 }
1065
1066
1067 void wapi_test_init(_adapter *padapter)
1068 {
1069 u8 keybuf[100];
1070 u8 mac_addr[ETH_ALEN] = {0x00, 0xe0, 0x4c, 0x72, 0x04, 0x70};
1071 u8 UskDataKey[16] = {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f};
1072 u8 UskMicKey[16] = {0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f};
1073 u8 UskId = 0;
1074 u8 MskDataKey[16] = {0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f};
1075 u8 MskMicKey[16] = {0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f};
1076 u8 MskId = 0;
1077
1078 WAPI_TRACE(WAPI_INIT, "===========>%s\n", __FUNCTION__);
1079
1080 /* Enable Wapi */
1081 WAPI_TRACE(WAPI_INIT, "%s: Enable wapi!!!!\n", __FUNCTION__);
1082 padapter->wapiInfo.bWapiEnable = true;
1083 padapter->pairwise_key_type = KEY_TYPE_SMS4;
1084 ieee->group_key_type = KEY_TYPE_SMS4;
1085 padapter->wapiInfo.extra_prefix_len = WAPI_EXT_LEN;
1086 padapter->wapiInfo.extra_postfix_len = SMS4_MIC_LEN;
1087
1088 /* set usk */
1089 WAPI_TRACE(WAPI_INIT, "%s: Set USK!!!!\n", __FUNCTION__);
1090 _rtw_memset(keybuf, 0, 100);
1091 keybuf[0] = 1; /* set usk */
1092 keybuf[1] = 1; /* enable tx */
1093 keybuf[2] = 1; /* AE */
1094 keybuf[3] = 0; /* not update */
1095
1096 _rtw_memcpy(keybuf + 4, mac_addr, ETH_ALEN);
1097 _rtw_memcpy(keybuf + 10, UskDataKey, 16);
1098 _rtw_memcpy(keybuf + 26, UskMicKey, 16);
1099 keybuf[42] = UskId;
1100 wapi_test_set_key(padapter, keybuf);
1101
1102 _rtw_memset(keybuf, 0, 100);
1103 keybuf[0] = 1; /* set usk */
1104 keybuf[1] = 1; /* enable tx */
1105 keybuf[2] = 0; /* AE */
1106 keybuf[3] = 0; /* not update */
1107
1108 _rtw_memcpy(keybuf + 4, mac_addr, ETH_ALEN);
1109 _rtw_memcpy(keybuf + 10, UskDataKey, 16);
1110 _rtw_memcpy(keybuf + 26, UskMicKey, 16);
1111 keybuf[42] = UskId;
1112 wapi_test_set_key(padapter, keybuf);
1113
1114 /* set msk */
1115 WAPI_TRACE(WAPI_INIT, "%s: Set MSK!!!!\n", __FUNCTION__);
1116 _rtw_memset(keybuf, 0, 100);
1117 keybuf[0] = 2; /* set msk */
1118 keybuf[1] = 1; /* Enable TX */
1119 keybuf[2] = 1; /* AE */
1120 keybuf[3] = 0; /* not update */
1121 _rtw_memcpy(keybuf + 4, mac_addr, ETH_ALEN);
1122 _rtw_memcpy(keybuf + 10, MskDataKey, 16);
1123 _rtw_memcpy(keybuf + 26, MskMicKey, 16);
1124 keybuf[42] = MskId;
1125 wapi_test_set_key(padapter, keybuf);
1126
1127 _rtw_memset(keybuf, 0, 100);
1128 keybuf[0] = 2; /* set msk */
1129 keybuf[1] = 1; /* Enable TX */
1130 keybuf[2] = 0; /* AE */
1131 keybuf[3] = 0; /* not update */
1132 _rtw_memcpy(keybuf + 4, mac_addr, ETH_ALEN);
1133 _rtw_memcpy(keybuf + 10, MskDataKey, 16);
1134 _rtw_memcpy(keybuf + 26, MskMicKey, 16);
1135 keybuf[42] = MskId;
1136 wapi_test_set_key(padapter, keybuf);
1137 WAPI_TRACE(WAPI_INIT, "<===========%s\n", __FUNCTION__);
1138 }
1139 #endif
1140
rtw_wapi_get_iv(_adapter * padapter,u8 * pRA,u8 * IV)1141 void rtw_wapi_get_iv(_adapter *padapter, u8 *pRA, u8 *IV)
1142 {
1143 PWLAN_HEADER_WAPI_EXTENSION pWapiExt = NULL;
1144 PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;
1145 bool bPNOverflow = false;
1146 bool bFindMatchPeer = false;
1147 PRT_WAPI_STA_INFO pWapiSta = NULL;
1148
1149 pWapiExt = (PWLAN_HEADER_WAPI_EXTENSION)IV;
1150
1151 WAPI_DATA(WAPI_RX, "wapi_get_iv: pra", pRA, 6);
1152
1153 if (IS_MCAST(pRA)) {
1154 if (!pWapiInfo->wapiTxMsk.bTxEnable) {
1155 WAPI_TRACE(WAPI_ERR, "%s: bTxEnable = 0!!\n", __FUNCTION__);
1156 return;
1157 }
1158
1159 if (pWapiInfo->wapiTxMsk.keyId <= 1) {
1160 pWapiExt->KeyIdx = pWapiInfo->wapiTxMsk.keyId;
1161 pWapiExt->Reserved = 0;
1162 bPNOverflow = WapiIncreasePN(pWapiInfo->lastTxMulticastPN, 1);
1163 _rtw_memcpy(pWapiExt->PN, pWapiInfo->lastTxMulticastPN, 16);
1164 }
1165 } else {
1166 if (list_empty(&pWapiInfo->wapiSTAUsedList)) {
1167 WAPI_TRACE(WAPI_RX, "rtw_wapi_get_iv: list is empty\n");
1168 _rtw_memset(IV, 10, 18);
1169 return;
1170 } else {
1171 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
1172 WAPI_DATA(WAPI_RX, "rtw_wapi_get_iv: peermacaddr ", pWapiSta->PeerMacAddr, 6);
1173 if (_rtw_memcmp((u8 *)pWapiSta->PeerMacAddr, pRA, 6) == _TRUE) {
1174 bFindMatchPeer = true;
1175 break;
1176 }
1177 }
1178
1179 WAPI_TRACE(WAPI_RX, "bFindMatchPeer: %d\n", bFindMatchPeer);
1180 WAPI_DATA(WAPI_RX, "Addr", pRA, 6);
1181
1182 if (bFindMatchPeer) {
1183 if ((!pWapiSta->wapiUskUpdate.bTxEnable) && (!pWapiSta->wapiUsk.bTxEnable))
1184 return;
1185
1186 if (pWapiSta->wapiUsk.keyId <= 1) {
1187 if (pWapiSta->wapiUskUpdate.bTxEnable)
1188 pWapiExt->KeyIdx = pWapiSta->wapiUskUpdate.keyId;
1189 else
1190 pWapiExt->KeyIdx = pWapiSta->wapiUsk.keyId;
1191
1192 pWapiExt->Reserved = 0;
1193 bPNOverflow = WapiIncreasePN(pWapiSta->lastTxUnicastPN, 2);
1194 _rtw_memcpy(pWapiExt->PN, pWapiSta->lastTxUnicastPN, 16);
1195
1196 }
1197 }
1198 }
1199
1200 }
1201
1202 }
1203
rtw_wapi_drop_for_key_absent(_adapter * padapter,u8 * pRA)1204 bool rtw_wapi_drop_for_key_absent(_adapter *padapter, u8 *pRA)
1205 {
1206 PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;
1207 bool bFindMatchPeer = false;
1208 bool bDrop = false;
1209 PRT_WAPI_STA_INFO pWapiSta = NULL;
1210 struct security_priv *psecuritypriv = &padapter->securitypriv;
1211
1212 WAPI_DATA(WAPI_RX, "rtw_wapi_drop_for_key_absent: ra ", pRA, 6);
1213
1214 if (psecuritypriv->dot11PrivacyAlgrthm == _SMS4_) {
1215 if ((!padapter->WapiSupport) || (!pWapiInfo->bWapiEnable))
1216 return true;
1217
1218 if (IS_MCAST(pRA)) {
1219 if (!pWapiInfo->wapiTxMsk.bTxEnable) {
1220 bDrop = true;
1221 WAPI_TRACE(WAPI_RX, "rtw_wapi_drop_for_key_absent: multicast key is absent\n");
1222 return bDrop;
1223 }
1224 } else {
1225 if (!list_empty(&pWapiInfo->wapiSTAUsedList)) {
1226 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
1227 WAPI_DATA(WAPI_RX, "rtw_wapi_drop_for_key_absent: pWapiSta->PeerMacAddr ", pWapiSta->PeerMacAddr, 6);
1228 if (_rtw_memcmp(pRA, pWapiSta->PeerMacAddr, 6) == _TRUE) {
1229 bFindMatchPeer = true;
1230 break;
1231 }
1232 }
1233 if (bFindMatchPeer) {
1234 if (!pWapiSta->wapiUsk.bTxEnable) {
1235 bDrop = true;
1236 WAPI_TRACE(WAPI_RX, "rtw_wapi_drop_for_key_absent: unicast key is absent\n");
1237 return bDrop;
1238 }
1239 } else {
1240 bDrop = true;
1241 WAPI_TRACE(WAPI_RX, "rtw_wapi_drop_for_key_absent: no peer find\n");
1242 return bDrop;
1243 }
1244
1245 } else {
1246 bDrop = true;
1247 WAPI_TRACE(WAPI_RX, "rtw_wapi_drop_for_key_absent: no sta exist\n");
1248 return bDrop;
1249 }
1250 }
1251 } else
1252 return bDrop;
1253
1254 return bDrop;
1255 }
1256
rtw_wapi_set_set_encryption(_adapter * padapter,struct ieee_param * param)1257 void rtw_wapi_set_set_encryption(_adapter *padapter, struct ieee_param *param)
1258 {
1259 struct security_priv *psecuritypriv = &padapter->securitypriv;
1260 struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
1261 PRT_WAPI_T pWapiInfo = &padapter->wapiInfo;
1262 PRT_WAPI_STA_INFO pWapiSta;
1263 u8 WapiASUEPNInitialValueSrc[16] = {0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
1264 u8 WapiAEPNInitialValueSrc[16] = {0x37, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
1265 u8 WapiAEMultiCastPNInitialValueSrc[16] = {0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C, 0x36, 0x5C} ;
1266
1267 if (param->u.crypt.set_tx == 1) {
1268 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
1269 if (_rtw_memcmp(pWapiSta->PeerMacAddr, param->sta_addr, 6)) {
1270 _rtw_memcpy(pWapiSta->lastTxUnicastPN, WapiASUEPNInitialValueSrc, 16);
1271
1272 pWapiSta->wapiUsk.bSet = true;
1273 _rtw_memcpy(pWapiSta->wapiUsk.dataKey, param->u.crypt.key, 16);
1274 _rtw_memcpy(pWapiSta->wapiUsk.micKey, param->u.crypt.key + 16, 16);
1275 pWapiSta->wapiUsk.keyId = param->u.crypt.idx;
1276 psecuritypriv->dot11PrivacyKeyIndex = param->u.crypt.idx;
1277 pWapiSta->wapiUsk.bTxEnable = true;
1278
1279 _rtw_memcpy(pWapiSta->lastRxUnicastPNBEQueue, WapiAEPNInitialValueSrc, 16);
1280 _rtw_memcpy(pWapiSta->lastRxUnicastPNBKQueue, WapiAEPNInitialValueSrc, 16);
1281 _rtw_memcpy(pWapiSta->lastRxUnicastPNVIQueue, WapiAEPNInitialValueSrc, 16);
1282 _rtw_memcpy(pWapiSta->lastRxUnicastPNVOQueue, WapiAEPNInitialValueSrc, 16);
1283 _rtw_memcpy(pWapiSta->lastRxUnicastPN, WapiAEPNInitialValueSrc, 16);
1284 pWapiSta->wapiUskUpdate.bTxEnable = false;
1285 pWapiSta->wapiUskUpdate.bSet = false;
1286
1287 if (psecuritypriv->sw_encrypt == false || psecuritypriv->sw_decrypt == false) {
1288 /* set unicast key for ASUE */
1289 rtw_wapi_set_key(padapter, &pWapiSta->wapiUsk, pWapiSta, false);
1290 }
1291 }
1292 }
1293 } else {
1294 list_for_each_entry(pWapiSta, &pWapiInfo->wapiSTAUsedList, list) {
1295 if (_rtw_memcmp(pWapiSta->PeerMacAddr, get_bssid(pmlmepriv), 6)) {
1296 pWapiSta->wapiMsk.bSet = true;
1297 _rtw_memcpy(pWapiSta->wapiMsk.dataKey, param->u.crypt.key, 16);
1298 _rtw_memcpy(pWapiSta->wapiMsk.micKey, param->u.crypt.key + 16, 16);
1299 pWapiSta->wapiMsk.keyId = param->u.crypt.idx;
1300 psecuritypriv->dot118021XGrpKeyid = param->u.crypt.idx;
1301 pWapiSta->wapiMsk.bTxEnable = false;
1302 if (!pWapiSta->bSetkeyOk)
1303 pWapiSta->bSetkeyOk = true;
1304 pWapiSta->bAuthenticateInProgress = false;
1305
1306 _rtw_memcpy(pWapiSta->lastRxMulticastPN, WapiAEMultiCastPNInitialValueSrc, 16);
1307
1308 if (psecuritypriv->sw_decrypt == false) {
1309 /* set rx broadcast key for ASUE */
1310 rtw_wapi_set_key(padapter, &pWapiSta->wapiMsk, pWapiSta, true);
1311 }
1312 }
1313 }
1314 }
1315 }
1316 #endif
1317