Staging: Add pristine upstream vt6656 driver sources to drivers/staging/vt6656.
[pandora-kernel.git] / drivers / staging / vt6656 / wpa.c
1 /*
2  * Copyright (c) 1996, 2003 VIA Networking Technologies, Inc.
3  * All rights reserved.
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License along
16  * with this program; if not, write to the Free Software Foundation, Inc.,
17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  *
20  * File: wpa.c
21  *
22  * Purpose: Handles the Basic Service Set & Node Database functions
23  *
24  * Functions:
25  *      WPA_ParseRSN - Parse RSN IE.
26  *
27  * Revision History:
28  *
29  * Author: Kyle Hsu
30  *
31  * Date: July 14, 2003
32  *
33  */
34
35
36 #if !defined(__TTYPE_H__)
37 #include "ttype.h"
38 #endif
39 #if !defined(__UMEM_H__)
40 #include "umem.h"
41 #endif
42 #if !defined(__TMACRO_H__)
43 #include "tmacro.h"
44 #endif
45 #if !defined(__TETHER_H__)
46 #include "tether.h"
47 #endif
48 #if !defined(__DEVICE_H__)
49 #include "device.h"
50 #endif
51 #if !defined(__80211HDR_H__)
52 #include "80211hdr.h"
53 #endif
54 #if !defined(__BSSDB_H__)
55 #include "bssdb.h"
56 #endif
57 #if !defined(__WMGR_H__)
58 #include "wmgr.h"
59 #endif
60 #if !defined(__WPA_H__)
61 #include "wpa.h"
62 #endif
63 #if !defined(__80211MGR_H__)
64 #include "80211mgr.h"
65 #endif
66
67
68 /*---------------------  Static Variables  --------------------------*/
69 static int          msglevel                =MSG_LEVEL_INFO;
70
71 const BYTE abyOUI00[4] = { 0x00, 0x50, 0xf2, 0x00 };
72 const BYTE abyOUI01[4] = { 0x00, 0x50, 0xf2, 0x01 };
73 const BYTE abyOUI02[4] = { 0x00, 0x50, 0xf2, 0x02 };
74 const BYTE abyOUI03[4] = { 0x00, 0x50, 0xf2, 0x03 };
75 const BYTE abyOUI04[4] = { 0x00, 0x50, 0xf2, 0x04 };
76 const BYTE abyOUI05[4] = { 0x00, 0x50, 0xf2, 0x05 };
77
78
79 /*+
80  *
81  * Description:
82  *    Clear RSN information in BSSList.
83  *
84  * Parameters:
85  *  In:
86  *      pBSSList - BSS list.
87  *  Out:
88  *      none
89  *
90  * Return Value: none.
91  *
92 -*/
93
94 VOID
95 WPA_ClearRSN (
96     IN PKnownBSS        pBSSList
97     )
98 {
99     int ii;
100     pBSSList->byGKType = WPA_TKIP;
101     for (ii=0; ii < 4; ii ++)
102         pBSSList->abyPKType[ii] = WPA_TKIP;
103     pBSSList->wPKCount = 0;
104     for (ii=0; ii < 4; ii ++)
105         pBSSList->abyAuthType[ii] = WPA_AUTH_IEEE802_1X;
106     pBSSList->wAuthCount = 0;
107     pBSSList->byDefaultK_as_PK = 0;
108     pBSSList->byReplayIdx = 0;
109     pBSSList->sRSNCapObj.bRSNCapExist = FALSE;
110     pBSSList->sRSNCapObj.wRSNCap = 0;
111     pBSSList->bWPAValid = FALSE;
112 }
113
114
115 /*+
116  *
117  * Description:
118  *    Parse RSN IE.
119  *
120  * Parameters:
121  *  In:
122  *      pBSSList - BSS list.
123  *      pRSN - Pointer to the RSN IE.
124  *  Out:
125  *      none
126  *
127  * Return Value: none.
128  *
129 -*/
130 VOID
131 WPA_ParseRSN (
132     IN PKnownBSS        pBSSList,
133     IN PWLAN_IE_RSN_EXT pRSN
134     )
135 {
136     PWLAN_IE_RSN_AUTH  pIE_RSN_Auth = NULL;
137     int                i, j, m, n = 0;
138     PBYTE              pbyCaps;
139
140     WPA_ClearRSN(pBSSList);
141
142     DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"WPA_ParseRSN: [%d]\n", pRSN->len);
143
144     // information element header makes sense
145     if ((pRSN->len >= 6) // oui1(4)+ver(2)
146          && (pRSN->byElementID == WLAN_EID_RSN_WPA) && MEMEqualMemory(pRSN->abyOUI, abyOUI01, 4)
147          && (pRSN->wVersion == 1)) {
148
149         DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"Legal RSN\n");
150         // update each variable if pRSN is long enough to contain the variable
151         if (pRSN->len >= 10) //oui1(4)+ver(2)+GKSuite(4)
152         {
153             if (MEMEqualMemory(pRSN->abyMulticast, abyOUI01, 4))
154                 pBSSList->byGKType = WPA_WEP40;
155             else if (MEMEqualMemory(pRSN->abyMulticast, abyOUI02, 4))
156                 pBSSList->byGKType = WPA_TKIP;
157             else if (MEMEqualMemory(pRSN->abyMulticast, abyOUI03, 4))
158                 pBSSList->byGKType = WPA_AESWRAP;
159             else if (MEMEqualMemory(pRSN->abyMulticast, abyOUI04, 4))
160                 pBSSList->byGKType = WPA_AESCCMP;
161             else if (MEMEqualMemory(pRSN->abyMulticast, abyOUI05, 4))
162                 pBSSList->byGKType = WPA_WEP104;
163             else
164                 // any vendor checks here
165                 pBSSList->byGKType = WPA_NONE;
166
167             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"byGKType: %x\n", pBSSList->byGKType);
168         }
169
170         if (pRSN->len >= 12) //oui1(4)+ver(2)+GKS(4)+PKSCnt(2)
171         {
172             j = 0;
173             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"wPKCount: %d, sizeof(pBSSList->abyPKType): %d\n", pRSN->wPKCount, sizeof(pBSSList->abyPKType));
174             for(i = 0; (i < pRSN->wPKCount) && (j < sizeof(pBSSList->abyPKType)/sizeof(BYTE)); i++) {
175                 if(pRSN->len >= 12+i*4+4) { //oui1(4)+ver(2)+GKS(4)+PKSCnt(2)+PKS(4*i)
176                     if (MEMEqualMemory(pRSN->PKSList[i].abyOUI, abyOUI00, 4))
177                         pBSSList->abyPKType[j++] = WPA_NONE;
178                     else if (MEMEqualMemory(pRSN->PKSList[i].abyOUI, abyOUI02, 4))
179                         pBSSList->abyPKType[j++] = WPA_TKIP;
180                     else if (MEMEqualMemory(pRSN->PKSList[i].abyOUI, abyOUI03, 4))
181                         pBSSList->abyPKType[j++] = WPA_AESWRAP;
182                     else if (MEMEqualMemory(pRSN->PKSList[i].abyOUI, abyOUI04, 4))
183                         pBSSList->abyPKType[j++] = WPA_AESCCMP;
184                     else
185                         // any vendor checks here
186                         ;
187                 }
188                 else
189                     break;
190                 //DBG_PRN_GRP14(("abyPKType[%d]: %X\n", j-1, pBSSList->abyPKType[j-1]));
191             } //for
192             pBSSList->wPKCount = (WORD)j;
193             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"wPKCount: %d\n", pBSSList->wPKCount);
194         }
195
196         m = pRSN->wPKCount;
197         DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"m: %d\n", m);
198         DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"14+m*4: %d\n", 14+m*4);
199
200         if (pRSN->len >= 14+m*4) { //oui1(4)+ver(2)+GKS(4)+PKSCnt(2)+PKS(4*m)+AKC(2)
201             // overlay IE_RSN_Auth structure into correct place
202             pIE_RSN_Auth = (PWLAN_IE_RSN_AUTH) pRSN->PKSList[m].abyOUI;
203             j = 0;
204             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"wAuthCount: %d, sizeof(pBSSList->abyAuthType): %d\n",
205                           pIE_RSN_Auth->wAuthCount, sizeof(pBSSList->abyAuthType));
206             for(i = 0; (i < pIE_RSN_Auth->wAuthCount) && (j < sizeof(pBSSList->abyAuthType)/sizeof(BYTE)); i++) {
207                 if(pRSN->len >= 14+4+(m+i)*4) { //oui1(4)+ver(2)+GKS(4)+PKSCnt(2)+PKS(4*m)+AKC(2)+AKS(4*i)
208                     if (MEMEqualMemory(pIE_RSN_Auth->AuthKSList[i].abyOUI, abyOUI01, 4))
209                         pBSSList->abyAuthType[j++] = WPA_AUTH_IEEE802_1X;
210                     else if (MEMEqualMemory(pIE_RSN_Auth->AuthKSList[i].abyOUI, abyOUI02, 4))
211                         pBSSList->abyAuthType[j++] = WPA_AUTH_PSK;
212                     else
213                     // any vendor checks here
214                     ;
215                 }
216                 else
217                     break;
218                 //DBG_PRN_GRP14(("abyAuthType[%d]: %X\n", j-1, pBSSList->abyAuthType[j-1]));
219             }
220             if(j > 0)
221                 pBSSList->wAuthCount = (WORD)j;
222             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"wAuthCount: %d\n", pBSSList->wAuthCount);
223         }
224
225         if (pIE_RSN_Auth != NULL) {
226
227             n = pIE_RSN_Auth->wAuthCount;
228
229             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"n: %d\n", n);
230             DBG_PRT(MSG_LEVEL_DEBUG, KERN_INFO"14+4+(m+n)*4: %d\n", 14+4+(m+n)*4);
231
232             if(pRSN->len+2 >= 14+4+(m+n)*4) { //oui1(4)+ver(2)+GKS(4)+PKSCnt(2)+PKS(4*m)+AKC(2)+AKS(4*n)+Cap(2)
233                 pbyCaps = (PBYTE)pIE_RSN_Auth->AuthKSList[n].abyOUI;
234                 pBSSList->byDefaultK_as_PK = (*pbyCaps) & WPA_GROUPFLAG;
235                 pBSSList->byReplayIdx = 2 << ((*pbyCaps >> WPA_REPLAYBITSSHIFT) & WPA_REPLAYBITS);
236                 pBSSList->sRSNCapObj.bRSNCapExist = TRUE;
237                 pBSSList->sRSNCapObj.wRSNCap = *(PWORD)pbyCaps;
238                 //DBG_PRN_GRP14(("pbyCaps: %X\n", *pbyCaps));
239                 //DBG_PRN_GRP14(("byDefaultK_as_PK: %X\n", pBSSList->byDefaultK_as_PK));
240                 //DBG_PRN_GRP14(("byReplayIdx: %X\n", pBSSList->byReplayIdx));
241             }
242         }
243         pBSSList->bWPAValid = TRUE;
244     }
245 }
246
247 /*+
248  *
249  * Description:
250  *    Search RSN information in BSSList.
251  *
252  * Parameters:
253  *  In:
254  *      byCmd    - Search type
255  *      byEncrypt- Encrcypt Type
256  *      pBSSList - BSS list
257  *  Out:
258  *      none
259  *
260  * Return Value: none.
261  *
262 -*/
263 BOOL
264 WPA_SearchRSN (
265     BYTE                byCmd,
266     BYTE                byEncrypt,
267     IN PKnownBSS        pBSSList
268     )
269 {
270     int ii;
271     BYTE byPKType = WPA_NONE;
272
273     if (pBSSList->bWPAValid == FALSE)
274         return FALSE;
275
276     switch(byCmd) {
277     case 0:
278
279         if (byEncrypt != pBSSList->byGKType)
280             return FALSE;
281
282         if (pBSSList->wPKCount > 0) {
283             for (ii = 0; ii < pBSSList->wPKCount; ii ++) {
284                 if (pBSSList->abyPKType[ii] == WPA_AESCCMP)
285                     byPKType = WPA_AESCCMP;
286                 else if ((pBSSList->abyPKType[ii] == WPA_TKIP) && (byPKType != WPA_AESCCMP))
287                      byPKType = WPA_TKIP;
288                 else if ((pBSSList->abyPKType[ii] == WPA_WEP40) && (byPKType != WPA_AESCCMP) && (byPKType != WPA_TKIP))
289                      byPKType = WPA_WEP40;
290                 else if ((pBSSList->abyPKType[ii] == WPA_WEP104) && (byPKType != WPA_AESCCMP) && (byPKType != WPA_TKIP))
291                      byPKType = WPA_WEP104;
292             }
293             if (byEncrypt != byPKType)
294                 return FALSE;
295         }
296         return TRUE;
297 //        if (pBSSList->wAuthCount > 0)
298 //            for (ii=0; ii < pBSSList->wAuthCount; ii ++)
299 //                if (byAuth == pBSSList->abyAuthType[ii])
300 //                    break;
301         break;
302
303     default:
304         break;
305     }
306     return FALSE;
307 }
308
309 /*+
310  *
311  * Description:
312  *    Check if RSN IE makes sense.
313  *
314  * Parameters:
315  *  In:
316  *      pRSN - Pointer to the RSN IE.
317  *  Out:
318  *      none
319  *
320  * Return Value: none.
321  *
322 -*/
323 BOOL
324 WPAb_Is_RSN (
325     IN PWLAN_IE_RSN_EXT pRSN
326     )
327 {
328     if (pRSN == NULL)
329         return FALSE;
330
331     if ((pRSN->len >= 6) && // oui1(4)+ver(2)
332         (pRSN->byElementID == WLAN_EID_RSN_WPA) && MEMEqualMemory(pRSN->abyOUI, abyOUI01, 4) &&
333         (pRSN->wVersion == 1)) {
334         return TRUE;
335     }
336     else
337         return FALSE;
338 }
339