remove @EAP_LDFLAGS@, no longer exists
[mech_eap.orig] / libeap / src / drivers / driver_ndis.c
1 /*
2  * WPA Supplicant - Windows/NDIS driver interface
3  * Copyright (c) 2004-2007, Jouni Malinen <j@w1.fi>
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 version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #ifdef __CYGWIN__
16 /* Avoid some header file conflicts by not including standard headers for
17  * cygwin builds when Packet32.h is included. */
18 #include "build_config.h"
19 int close(int fd);
20 #else /* __CYGWIN__ */
21 #include "includes.h"
22 #endif /* __CYGWIN__ */
23 #ifdef CONFIG_USE_NDISUIO
24 #include <winsock2.h>
25 #else /* CONFIG_USE_NDISUIO */
26 #include <Packet32.h>
27 #endif /* CONFIG_USE_NDISUIO */
28 #ifdef __MINGW32_VERSION
29 #include <ddk/ntddndis.h>
30 #else /* __MINGW32_VERSION */
31 #include <ntddndis.h>
32 #endif /* __MINGW32_VERSION */
33
34 #ifdef _WIN32_WCE
35 #include <winioctl.h>
36 #include <nuiouser.h>
37 #include <devload.h>
38 #endif /* _WIN32_WCE */
39
40 #include "common.h"
41 #include "driver.h"
42 #include "eloop.h"
43 #include "common/ieee802_11_defs.h"
44 #include "driver_ndis.h"
45
46 int wpa_driver_register_event_cb(struct wpa_driver_ndis_data *drv);
47 #ifdef CONFIG_NDIS_EVENTS_INTEGRATED
48 void wpa_driver_ndis_event_pipe_cb(void *eloop_data, void *user_data);
49 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
50
51 static void wpa_driver_ndis_deinit(void *priv);
52 static void wpa_driver_ndis_poll(void *drv);
53 static void wpa_driver_ndis_poll_timeout(void *eloop_ctx, void *timeout_ctx);
54 static int wpa_driver_ndis_adapter_init(struct wpa_driver_ndis_data *drv);
55 static int wpa_driver_ndis_adapter_open(struct wpa_driver_ndis_data *drv);
56 static void wpa_driver_ndis_adapter_close(struct wpa_driver_ndis_data *drv);
57
58
59 static const u8 pae_group_addr[ETH_ALEN] =
60 { 0x01, 0x80, 0xc2, 0x00, 0x00, 0x03 };
61
62
63 /* FIX: to be removed once this can be compiled with the complete NDIS
64  * header files */
65 #ifndef OID_802_11_BSSID
66 #define OID_802_11_BSSID                        0x0d010101
67 #define OID_802_11_SSID                         0x0d010102
68 #define OID_802_11_INFRASTRUCTURE_MODE          0x0d010108
69 #define OID_802_11_ADD_WEP                      0x0D010113
70 #define OID_802_11_REMOVE_WEP                   0x0D010114
71 #define OID_802_11_DISASSOCIATE                 0x0D010115
72 #define OID_802_11_BSSID_LIST                   0x0d010217
73 #define OID_802_11_AUTHENTICATION_MODE          0x0d010118
74 #define OID_802_11_PRIVACY_FILTER               0x0d010119
75 #define OID_802_11_BSSID_LIST_SCAN              0x0d01011A
76 #define OID_802_11_WEP_STATUS                   0x0d01011B
77 #define OID_802_11_ENCRYPTION_STATUS OID_802_11_WEP_STATUS
78 #define OID_802_11_ADD_KEY                      0x0d01011D
79 #define OID_802_11_REMOVE_KEY                   0x0d01011E
80 #define OID_802_11_ASSOCIATION_INFORMATION      0x0d01011F
81 #define OID_802_11_TEST                         0x0d010120
82 #define OID_802_11_CAPABILITY                   0x0d010122
83 #define OID_802_11_PMKID                        0x0d010123
84
85 #define NDIS_802_11_LENGTH_SSID 32
86 #define NDIS_802_11_LENGTH_RATES 8
87 #define NDIS_802_11_LENGTH_RATES_EX 16
88
89 typedef UCHAR NDIS_802_11_MAC_ADDRESS[6];
90
91 typedef struct NDIS_802_11_SSID {
92         ULONG SsidLength;
93         UCHAR Ssid[NDIS_802_11_LENGTH_SSID];
94 } NDIS_802_11_SSID;
95
96 typedef LONG NDIS_802_11_RSSI;
97
98 typedef enum NDIS_802_11_NETWORK_TYPE {
99         Ndis802_11FH,
100         Ndis802_11DS,
101         Ndis802_11OFDM5,
102         Ndis802_11OFDM24,
103         Ndis802_11NetworkTypeMax
104 } NDIS_802_11_NETWORK_TYPE;
105
106 typedef struct NDIS_802_11_CONFIGURATION_FH {
107         ULONG Length;
108         ULONG HopPattern;
109         ULONG HopSet;
110         ULONG DwellTime;
111 } NDIS_802_11_CONFIGURATION_FH;
112
113 typedef struct NDIS_802_11_CONFIGURATION {
114         ULONG Length;
115         ULONG BeaconPeriod;
116         ULONG ATIMWindow;
117         ULONG DSConfig;
118         NDIS_802_11_CONFIGURATION_FH FHConfig;
119 } NDIS_802_11_CONFIGURATION;
120
121 typedef enum NDIS_802_11_NETWORK_INFRASTRUCTURE {
122         Ndis802_11IBSS,
123         Ndis802_11Infrastructure,
124         Ndis802_11AutoUnknown,
125         Ndis802_11InfrastructureMax
126 } NDIS_802_11_NETWORK_INFRASTRUCTURE;
127
128 typedef enum NDIS_802_11_AUTHENTICATION_MODE {
129         Ndis802_11AuthModeOpen,
130         Ndis802_11AuthModeShared,
131         Ndis802_11AuthModeAutoSwitch,
132         Ndis802_11AuthModeWPA,
133         Ndis802_11AuthModeWPAPSK,
134         Ndis802_11AuthModeWPANone,
135         Ndis802_11AuthModeWPA2,
136         Ndis802_11AuthModeWPA2PSK,
137         Ndis802_11AuthModeMax
138 } NDIS_802_11_AUTHENTICATION_MODE;
139
140 typedef enum NDIS_802_11_WEP_STATUS {
141         Ndis802_11WEPEnabled,
142         Ndis802_11Encryption1Enabled = Ndis802_11WEPEnabled,
143         Ndis802_11WEPDisabled,
144         Ndis802_11EncryptionDisabled = Ndis802_11WEPDisabled,
145         Ndis802_11WEPKeyAbsent,
146         Ndis802_11Encryption1KeyAbsent = Ndis802_11WEPKeyAbsent,
147         Ndis802_11WEPNotSupported,
148         Ndis802_11EncryptionNotSupported = Ndis802_11WEPNotSupported,
149         Ndis802_11Encryption2Enabled,
150         Ndis802_11Encryption2KeyAbsent,
151         Ndis802_11Encryption3Enabled,
152         Ndis802_11Encryption3KeyAbsent
153 } NDIS_802_11_WEP_STATUS, NDIS_802_11_ENCRYPTION_STATUS;
154
155 typedef enum NDIS_802_11_PRIVACY_FILTER {
156         Ndis802_11PrivFilterAcceptAll,
157         Ndis802_11PrivFilter8021xWEP
158 } NDIS_802_11_PRIVACY_FILTER;
159
160 typedef UCHAR NDIS_802_11_RATES[NDIS_802_11_LENGTH_RATES];
161 typedef UCHAR NDIS_802_11_RATES_EX[NDIS_802_11_LENGTH_RATES_EX];
162
163 typedef struct NDIS_WLAN_BSSID_EX {
164         ULONG Length;
165         NDIS_802_11_MAC_ADDRESS MacAddress; /* BSSID */
166         UCHAR Reserved[2];
167         NDIS_802_11_SSID Ssid;
168         ULONG Privacy;
169         NDIS_802_11_RSSI Rssi;
170         NDIS_802_11_NETWORK_TYPE NetworkTypeInUse;
171         NDIS_802_11_CONFIGURATION Configuration;
172         NDIS_802_11_NETWORK_INFRASTRUCTURE InfrastructureMode;
173         NDIS_802_11_RATES_EX SupportedRates;
174         ULONG IELength;
175         UCHAR IEs[1];
176 } NDIS_WLAN_BSSID_EX;
177
178 typedef struct NDIS_802_11_BSSID_LIST_EX {
179         ULONG NumberOfItems;
180         NDIS_WLAN_BSSID_EX Bssid[1];
181 } NDIS_802_11_BSSID_LIST_EX;
182
183 typedef struct NDIS_802_11_FIXED_IEs {
184         UCHAR Timestamp[8];
185         USHORT BeaconInterval;
186         USHORT Capabilities;
187 } NDIS_802_11_FIXED_IEs;
188
189 typedef struct NDIS_802_11_WEP {
190         ULONG Length;
191         ULONG KeyIndex;
192         ULONG KeyLength;
193         UCHAR KeyMaterial[1];
194 } NDIS_802_11_WEP;
195
196 typedef ULONG NDIS_802_11_KEY_INDEX;
197 typedef ULONGLONG NDIS_802_11_KEY_RSC;
198
199 typedef struct NDIS_802_11_KEY {
200         ULONG Length;
201         ULONG KeyIndex;
202         ULONG KeyLength;
203         NDIS_802_11_MAC_ADDRESS BSSID;
204         NDIS_802_11_KEY_RSC KeyRSC;
205         UCHAR KeyMaterial[1];
206 } NDIS_802_11_KEY;
207
208 typedef struct NDIS_802_11_REMOVE_KEY {
209         ULONG Length;
210         ULONG KeyIndex;
211         NDIS_802_11_MAC_ADDRESS BSSID;
212 } NDIS_802_11_REMOVE_KEY;
213
214 typedef struct NDIS_802_11_AI_REQFI {
215         USHORT Capabilities;
216         USHORT ListenInterval;
217         NDIS_802_11_MAC_ADDRESS CurrentAPAddress;
218 } NDIS_802_11_AI_REQFI;
219
220 typedef struct NDIS_802_11_AI_RESFI {
221         USHORT Capabilities;
222         USHORT StatusCode;
223         USHORT AssociationId;
224 } NDIS_802_11_AI_RESFI;
225
226 typedef struct NDIS_802_11_ASSOCIATION_INFORMATION {
227         ULONG Length;
228         USHORT AvailableRequestFixedIEs;
229         NDIS_802_11_AI_REQFI RequestFixedIEs;
230         ULONG RequestIELength;
231         ULONG OffsetRequestIEs;
232         USHORT AvailableResponseFixedIEs;
233         NDIS_802_11_AI_RESFI ResponseFixedIEs;
234         ULONG ResponseIELength;
235         ULONG OffsetResponseIEs;
236 } NDIS_802_11_ASSOCIATION_INFORMATION;
237
238 typedef struct NDIS_802_11_AUTHENTICATION_ENCRYPTION {
239         NDIS_802_11_AUTHENTICATION_MODE AuthModeSupported;
240         NDIS_802_11_ENCRYPTION_STATUS EncryptStatusSupported;
241 } NDIS_802_11_AUTHENTICATION_ENCRYPTION;
242
243 typedef struct NDIS_802_11_CAPABILITY {
244         ULONG Length;
245         ULONG Version;
246         ULONG NoOfPMKIDs;
247         ULONG NoOfAuthEncryptPairsSupported;
248         NDIS_802_11_AUTHENTICATION_ENCRYPTION
249                 AuthenticationEncryptionSupported[1];
250 } NDIS_802_11_CAPABILITY;
251
252 typedef UCHAR NDIS_802_11_PMKID_VALUE[16];
253
254 typedef struct BSSID_INFO {
255         NDIS_802_11_MAC_ADDRESS BSSID;
256         NDIS_802_11_PMKID_VALUE PMKID;
257 } BSSID_INFO;
258
259 typedef struct NDIS_802_11_PMKID {
260         ULONG Length;
261         ULONG BSSIDInfoCount;
262         BSSID_INFO BSSIDInfo[1];
263 } NDIS_802_11_PMKID;
264
265 typedef enum NDIS_802_11_STATUS_TYPE {
266         Ndis802_11StatusType_Authentication,
267         Ndis802_11StatusType_PMKID_CandidateList = 2,
268         Ndis802_11StatusTypeMax
269 } NDIS_802_11_STATUS_TYPE;
270
271 typedef struct NDIS_802_11_STATUS_INDICATION {
272         NDIS_802_11_STATUS_TYPE StatusType;
273 } NDIS_802_11_STATUS_INDICATION;
274
275 typedef struct PMKID_CANDIDATE {
276         NDIS_802_11_MAC_ADDRESS BSSID;
277         ULONG Flags;
278 } PMKID_CANDIDATE;
279
280 #define NDIS_802_11_PMKID_CANDIDATE_PREAUTH_ENABLED 0x01
281
282 typedef struct NDIS_802_11_PMKID_CANDIDATE_LIST {
283         ULONG Version;
284         ULONG NumCandidates;
285         PMKID_CANDIDATE CandidateList[1];
286 } NDIS_802_11_PMKID_CANDIDATE_LIST;
287
288 typedef struct NDIS_802_11_AUTHENTICATION_REQUEST {
289         ULONG Length;
290         NDIS_802_11_MAC_ADDRESS Bssid;
291         ULONG Flags;
292 } NDIS_802_11_AUTHENTICATION_REQUEST;
293
294 #define NDIS_802_11_AUTH_REQUEST_REAUTH                 0x01
295 #define NDIS_802_11_AUTH_REQUEST_KEYUPDATE              0x02
296 #define NDIS_802_11_AUTH_REQUEST_PAIRWISE_ERROR         0x06
297 #define NDIS_802_11_AUTH_REQUEST_GROUP_ERROR            0x0E
298
299 #endif /* OID_802_11_BSSID */
300
301
302 #ifndef OID_802_11_PMKID
303 /* Platform SDK for XP did not include WPA2, so add needed definitions */
304
305 #define OID_802_11_CAPABILITY                   0x0d010122
306 #define OID_802_11_PMKID                        0x0d010123
307
308 #define Ndis802_11AuthModeWPA2 6
309 #define Ndis802_11AuthModeWPA2PSK 7
310
311 #define Ndis802_11StatusType_PMKID_CandidateList 2
312
313 typedef struct NDIS_802_11_AUTHENTICATION_ENCRYPTION {
314         NDIS_802_11_AUTHENTICATION_MODE AuthModeSupported;
315         NDIS_802_11_ENCRYPTION_STATUS EncryptStatusSupported;
316 } NDIS_802_11_AUTHENTICATION_ENCRYPTION;
317
318 typedef struct NDIS_802_11_CAPABILITY {
319         ULONG Length;
320         ULONG Version;
321         ULONG NoOfPMKIDs;
322         ULONG NoOfAuthEncryptPairsSupported;
323         NDIS_802_11_AUTHENTICATION_ENCRYPTION
324                 AuthenticationEncryptionSupported[1];
325 } NDIS_802_11_CAPABILITY;
326
327 typedef UCHAR NDIS_802_11_PMKID_VALUE[16];
328
329 typedef struct BSSID_INFO {
330         NDIS_802_11_MAC_ADDRESS BSSID;
331         NDIS_802_11_PMKID_VALUE PMKID;
332 } BSSID_INFO;
333
334 typedef struct NDIS_802_11_PMKID {
335         ULONG Length;
336         ULONG BSSIDInfoCount;
337         BSSID_INFO BSSIDInfo[1];
338 } NDIS_802_11_PMKID;
339
340 typedef struct PMKID_CANDIDATE {
341         NDIS_802_11_MAC_ADDRESS BSSID;
342         ULONG Flags;
343 } PMKID_CANDIDATE;
344
345 #define NDIS_802_11_PMKID_CANDIDATE_PREAUTH_ENABLED 0x01
346
347 typedef struct NDIS_802_11_PMKID_CANDIDATE_LIST {
348         ULONG Version;
349         ULONG NumCandidates;
350         PMKID_CANDIDATE CandidateList[1];
351 } NDIS_802_11_PMKID_CANDIDATE_LIST;
352
353 #endif /* OID_802_11_CAPABILITY */
354
355
356 #ifndef OID_DOT11_CURRENT_OPERATION_MODE
357 /* Native 802.11 OIDs */
358 #define OID_DOT11_NDIS_START 0x0D010300
359 #define OID_DOT11_CURRENT_OPERATION_MODE (OID_DOT11_NDIS_START + 8)
360 #define OID_DOT11_SCAN_REQUEST (OID_DOT11_NDIS_START + 11)
361
362 typedef enum _DOT11_BSS_TYPE {
363         dot11_BSS_type_infrastructure = 1,
364         dot11_BSS_type_independent = 2,
365         dot11_BSS_type_any = 3
366 } DOT11_BSS_TYPE, * PDOT11_BSS_TYPE;
367
368 typedef UCHAR DOT11_MAC_ADDRESS[6];
369 typedef DOT11_MAC_ADDRESS * PDOT11_MAC_ADDRESS;
370
371 typedef enum _DOT11_SCAN_TYPE {
372         dot11_scan_type_active = 1,
373         dot11_scan_type_passive = 2,
374         dot11_scan_type_auto = 3,
375         dot11_scan_type_forced = 0x80000000
376 } DOT11_SCAN_TYPE, * PDOT11_SCAN_TYPE;
377
378 typedef struct _DOT11_SCAN_REQUEST_V2 {
379         DOT11_BSS_TYPE dot11BSSType;
380         DOT11_MAC_ADDRESS dot11BSSID;
381         DOT11_SCAN_TYPE dot11ScanType;
382         BOOLEAN bRestrictedScan;
383         ULONG udot11SSIDsOffset;
384         ULONG uNumOfdot11SSIDs;
385         BOOLEAN bUseRequestIE;
386         ULONG uRequestIDsOffset;
387         ULONG uNumOfRequestIDs;
388         ULONG uPhyTypeInfosOffset;
389         ULONG uNumOfPhyTypeInfos;
390         ULONG uIEsOffset;
391         ULONG uIEsLength;
392         UCHAR ucBuffer[1];
393 } DOT11_SCAN_REQUEST_V2, * PDOT11_SCAN_REQUEST_V2;
394
395 #endif /* OID_DOT11_CURRENT_OPERATION_MODE */
396
397 #ifdef CONFIG_USE_NDISUIO
398 #ifndef _WIN32_WCE
399 #ifdef __MINGW32_VERSION
400 typedef ULONG NDIS_OID;
401 #endif /* __MINGW32_VERSION */
402 /* from nuiouser.h */
403 #define FSCTL_NDISUIO_BASE      FILE_DEVICE_NETWORK
404
405 #define _NDISUIO_CTL_CODE(_Function, _Method, _Access) \
406         CTL_CODE(FSCTL_NDISUIO_BASE, _Function, _Method, _Access)
407
408 #define IOCTL_NDISUIO_OPEN_DEVICE \
409         _NDISUIO_CTL_CODE(0x200, METHOD_BUFFERED, \
410                           FILE_READ_ACCESS | FILE_WRITE_ACCESS)
411
412 #define IOCTL_NDISUIO_QUERY_OID_VALUE \
413         _NDISUIO_CTL_CODE(0x201, METHOD_BUFFERED, \
414                           FILE_READ_ACCESS | FILE_WRITE_ACCESS)
415
416 #define IOCTL_NDISUIO_SET_OID_VALUE \
417         _NDISUIO_CTL_CODE(0x205, METHOD_BUFFERED, \
418                           FILE_READ_ACCESS | FILE_WRITE_ACCESS)
419
420 #define IOCTL_NDISUIO_SET_ETHER_TYPE \
421         _NDISUIO_CTL_CODE(0x202, METHOD_BUFFERED, \
422                           FILE_READ_ACCESS | FILE_WRITE_ACCESS)
423
424 #define IOCTL_NDISUIO_QUERY_BINDING \
425         _NDISUIO_CTL_CODE(0x203, METHOD_BUFFERED, \
426                           FILE_READ_ACCESS | FILE_WRITE_ACCESS)
427
428 #define IOCTL_NDISUIO_BIND_WAIT \
429         _NDISUIO_CTL_CODE(0x204, METHOD_BUFFERED, \
430                           FILE_READ_ACCESS | FILE_WRITE_ACCESS)
431
432 typedef struct _NDISUIO_QUERY_OID
433 {
434     NDIS_OID Oid;
435     UCHAR Data[sizeof(ULONG)];
436 } NDISUIO_QUERY_OID, *PNDISUIO_QUERY_OID;
437
438 typedef struct _NDISUIO_SET_OID
439 {
440     NDIS_OID Oid;
441     UCHAR Data[sizeof(ULONG)];
442 } NDISUIO_SET_OID, *PNDISUIO_SET_OID;
443
444 typedef struct _NDISUIO_QUERY_BINDING
445 {
446         ULONG BindingIndex;
447         ULONG DeviceNameOffset;
448         ULONG DeviceNameLength;
449         ULONG DeviceDescrOffset;
450         ULONG DeviceDescrLength;
451 } NDISUIO_QUERY_BINDING, *PNDISUIO_QUERY_BINDING;
452 #endif /* _WIN32_WCE */
453 #endif /* CONFIG_USE_NDISUIO */
454
455
456 static int ndis_get_oid(struct wpa_driver_ndis_data *drv, unsigned int oid,
457                         char *data, size_t len)
458 {
459 #ifdef CONFIG_USE_NDISUIO
460         NDISUIO_QUERY_OID *o;
461         size_t buflen = sizeof(*o) + len;
462         DWORD written;
463         int ret;
464         size_t hdrlen;
465
466         o = os_zalloc(buflen);
467         if (o == NULL)
468                 return -1;
469         o->Oid = oid;
470 #ifdef _WIN32_WCE
471         o->ptcDeviceName = drv->adapter_name;
472 #endif /* _WIN32_WCE */
473         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_QUERY_OID_VALUE,
474                              o, sizeof(NDISUIO_QUERY_OID), o, buflen, &written,
475                              NULL)) {
476                 wpa_printf(MSG_DEBUG, "NDIS: IOCTL_NDISUIO_QUERY_OID_VALUE "
477                            "failed (oid=%08x): %d", oid, (int) GetLastError());
478                 os_free(o);
479                 return -1;
480         }
481         hdrlen = sizeof(NDISUIO_QUERY_OID) - sizeof(o->Data);
482         if (written < hdrlen) {
483                 wpa_printf(MSG_DEBUG, "NDIS: query oid=%08x written (%d); "
484                            "too short", oid, (unsigned int) written);
485                 os_free(o);
486                 return -1;
487         }
488         written -= hdrlen;
489         if (written > len) {
490                 wpa_printf(MSG_DEBUG, "NDIS: query oid=%08x written (%d) > "
491                            "len (%d)",oid, (unsigned int) written, len);
492                 os_free(o);
493                 return -1;
494         }
495         os_memcpy(data, o->Data, written);
496         ret = written;
497         os_free(o);
498         return ret;
499 #else /* CONFIG_USE_NDISUIO */
500         char *buf;
501         PACKET_OID_DATA *o;
502         int ret;
503
504         buf = os_zalloc(sizeof(*o) + len);
505         if (buf == NULL)
506                 return -1;
507         o = (PACKET_OID_DATA *) buf;
508         o->Oid = oid;
509         o->Length = len;
510
511         if (!PacketRequest(drv->adapter, FALSE, o)) {
512                 wpa_printf(MSG_DEBUG, "%s: oid=0x%x len (%d) failed",
513                            __func__, oid, len);
514                 os_free(buf);
515                 return -1;
516         }
517         if (o->Length > len) {
518                 wpa_printf(MSG_DEBUG, "%s: oid=0x%x Length (%d) > len (%d)",
519                            __func__, oid, (unsigned int) o->Length, len);
520                 os_free(buf);
521                 return -1;
522         }
523         os_memcpy(data, o->Data, o->Length);
524         ret = o->Length;
525         os_free(buf);
526         return ret;
527 #endif /* CONFIG_USE_NDISUIO */
528 }
529
530
531 static int ndis_set_oid(struct wpa_driver_ndis_data *drv, unsigned int oid,
532                         const char *data, size_t len)
533 {
534 #ifdef CONFIG_USE_NDISUIO
535         NDISUIO_SET_OID *o;
536         size_t buflen, reallen;
537         DWORD written;
538         char txt[50];
539
540         os_snprintf(txt, sizeof(txt), "NDIS: Set OID %08x", oid);
541         wpa_hexdump_key(MSG_MSGDUMP, txt, (const u8 *) data, len);
542
543         buflen = sizeof(*o) + len;
544         reallen = buflen - sizeof(o->Data);
545         o = os_zalloc(buflen);
546         if (o == NULL)
547                 return -1;
548         o->Oid = oid;
549 #ifdef _WIN32_WCE
550         o->ptcDeviceName = drv->adapter_name;
551 #endif /* _WIN32_WCE */
552         if (data)
553                 os_memcpy(o->Data, data, len);
554         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_SET_OID_VALUE,
555                              o, reallen, NULL, 0, &written, NULL)) {
556                 wpa_printf(MSG_DEBUG, "NDIS: IOCTL_NDISUIO_SET_OID_VALUE "
557                            "(oid=%08x) failed: %d", oid, (int) GetLastError());
558                 os_free(o);
559                 return -1;
560         }
561         os_free(o);
562         return 0;
563 #else /* CONFIG_USE_NDISUIO */
564         char *buf;
565         PACKET_OID_DATA *o;
566         char txt[50];
567
568         os_snprintf(txt, sizeof(txt), "NDIS: Set OID %08x", oid);
569         wpa_hexdump_key(MSG_MSGDUMP, txt, (const u8 *) data, len);
570
571         buf = os_zalloc(sizeof(*o) + len);
572         if (buf == NULL)
573                 return -1;
574         o = (PACKET_OID_DATA *) buf;
575         o->Oid = oid;
576         o->Length = len;
577         if (data)
578                 os_memcpy(o->Data, data, len);
579
580         if (!PacketRequest(drv->adapter, TRUE, o)) {
581                 wpa_printf(MSG_DEBUG, "%s: oid=0x%x len (%d) failed",
582                            __func__, oid, len);
583                 os_free(buf);
584                 return -1;
585         }
586         os_free(buf);
587         return 0;
588 #endif /* CONFIG_USE_NDISUIO */
589 }
590
591
592 static int ndis_set_auth_mode(struct wpa_driver_ndis_data *drv, int mode)
593 {
594         u32 auth_mode = mode;
595         if (ndis_set_oid(drv, OID_802_11_AUTHENTICATION_MODE,
596                          (char *) &auth_mode, sizeof(auth_mode)) < 0) {
597                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
598                            "OID_802_11_AUTHENTICATION_MODE (%d)",
599                            (int) auth_mode);
600                 return -1;
601         }
602         return 0;
603 }
604
605
606 static int ndis_get_auth_mode(struct wpa_driver_ndis_data *drv)
607 {
608         u32 auth_mode;
609         int res;
610         res = ndis_get_oid(drv, OID_802_11_AUTHENTICATION_MODE,
611                            (char *) &auth_mode, sizeof(auth_mode));
612         if (res != sizeof(auth_mode)) {
613                 wpa_printf(MSG_DEBUG, "NDIS: Failed to get "
614                            "OID_802_11_AUTHENTICATION_MODE");
615                 return -1;
616         }
617         return auth_mode;
618 }
619
620
621 static int ndis_set_encr_status(struct wpa_driver_ndis_data *drv, int encr)
622 {
623         u32 encr_status = encr;
624         if (ndis_set_oid(drv, OID_802_11_ENCRYPTION_STATUS,
625                          (char *) &encr_status, sizeof(encr_status)) < 0) {
626                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
627                            "OID_802_11_ENCRYPTION_STATUS (%d)", encr);
628                 return -1;
629         }
630         return 0;
631 }
632
633
634 static int ndis_get_encr_status(struct wpa_driver_ndis_data *drv)
635 {
636         u32 encr;
637         int res;
638         res = ndis_get_oid(drv, OID_802_11_ENCRYPTION_STATUS,
639                            (char *) &encr, sizeof(encr));
640         if (res != sizeof(encr)) {
641                 wpa_printf(MSG_DEBUG, "NDIS: Failed to get "
642                            "OID_802_11_ENCRYPTION_STATUS");
643                 return -1;
644         }
645         return encr;
646 }
647
648
649 static int wpa_driver_ndis_get_bssid(void *priv, u8 *bssid)
650 {
651         struct wpa_driver_ndis_data *drv = priv;
652
653         if (drv->wired) {
654                 /*
655                  * Report PAE group address as the "BSSID" for wired
656                  * connection.
657                  */
658                 os_memcpy(bssid, pae_group_addr, ETH_ALEN);
659                 return 0;
660         }
661
662         return ndis_get_oid(drv, OID_802_11_BSSID, (char *) bssid, ETH_ALEN) <
663                 0 ? -1 : 0;
664 }
665
666
667 static int wpa_driver_ndis_get_ssid(void *priv, u8 *ssid)
668 {
669         struct wpa_driver_ndis_data *drv = priv;
670         NDIS_802_11_SSID buf;
671         int res;
672
673         res = ndis_get_oid(drv, OID_802_11_SSID, (char *) &buf, sizeof(buf));
674         if (res < 4) {
675                 wpa_printf(MSG_DEBUG, "NDIS: Failed to get SSID");
676                 if (drv->wired) {
677                         wpa_printf(MSG_DEBUG, "NDIS: Allow get_ssid failure "
678                                    "with a wired interface");
679                         return 0;
680                 }
681                 return -1;
682         }
683         os_memcpy(ssid, buf.Ssid, buf.SsidLength);
684         return buf.SsidLength;
685 }
686
687
688 static int wpa_driver_ndis_set_ssid(struct wpa_driver_ndis_data *drv,
689                                     const u8 *ssid, size_t ssid_len)
690 {
691         NDIS_802_11_SSID buf;
692
693         os_memset(&buf, 0, sizeof(buf));
694         buf.SsidLength = ssid_len;
695         os_memcpy(buf.Ssid, ssid, ssid_len);
696         /*
697          * Make sure radio is marked enabled here so that scan request will not
698          * force SSID to be changed to a random one in order to enable radio at
699          * that point.
700          */
701         drv->radio_enabled = 1;
702         return ndis_set_oid(drv, OID_802_11_SSID, (char *) &buf, sizeof(buf));
703 }
704
705
706 /* Disconnect using OID_802_11_DISASSOCIATE. This will also turn the radio off.
707  */
708 static int wpa_driver_ndis_radio_off(struct wpa_driver_ndis_data *drv)
709 {
710         drv->radio_enabled = 0;
711         return ndis_set_oid(drv, OID_802_11_DISASSOCIATE, "    ", 4);
712 }
713
714
715 /* Disconnect by setting SSID to random (i.e., likely not used). */
716 static int wpa_driver_ndis_disconnect(struct wpa_driver_ndis_data *drv)
717 {
718         char ssid[32];
719         int i;
720         for (i = 0; i < 32; i++)
721                 ssid[i] = rand() & 0xff;
722         return wpa_driver_ndis_set_ssid(drv, (u8 *) ssid, 32);
723 }
724
725
726 static int wpa_driver_ndis_deauthenticate(void *priv, const u8 *addr,
727                                           int reason_code)
728 {
729         struct wpa_driver_ndis_data *drv = priv;
730         return wpa_driver_ndis_disconnect(drv);
731 }
732
733
734 static int wpa_driver_ndis_disassociate(void *priv, const u8 *addr,
735                                         int reason_code)
736 {
737         struct wpa_driver_ndis_data *drv = priv;
738         return wpa_driver_ndis_disconnect(drv);
739 }
740
741
742 static void wpa_driver_ndis_scan_timeout(void *eloop_ctx, void *timeout_ctx)
743 {
744         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
745         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
746 }
747
748
749 static int wpa_driver_ndis_scan_native80211(
750         struct wpa_driver_ndis_data *drv,
751         struct wpa_driver_scan_params *params)
752 {
753         DOT11_SCAN_REQUEST_V2 req;
754         int res;
755
756         os_memset(&req, 0, sizeof(req));
757         req.dot11BSSType = dot11_BSS_type_any;
758         os_memset(req.dot11BSSID, 0xff, ETH_ALEN);
759         req.dot11ScanType = dot11_scan_type_auto;
760         res = ndis_set_oid(drv, OID_DOT11_SCAN_REQUEST, (char *) &req,
761                            sizeof(req));
762         eloop_cancel_timeout(wpa_driver_ndis_scan_timeout, drv, drv->ctx);
763         eloop_register_timeout(7, 0, wpa_driver_ndis_scan_timeout, drv,
764                                drv->ctx);
765         return res;
766 }
767
768
769 static int wpa_driver_ndis_scan(void *priv,
770                                 struct wpa_driver_scan_params *params)
771 {
772         struct wpa_driver_ndis_data *drv = priv;
773         int res;
774
775         if (drv->native80211)
776                 return wpa_driver_ndis_scan_native80211(drv, params);
777
778         if (!drv->radio_enabled) {
779                 wpa_printf(MSG_DEBUG, "NDIS: turning radio on before the first"
780                            " scan");
781                 if (wpa_driver_ndis_disconnect(drv) < 0) {
782                         wpa_printf(MSG_DEBUG, "NDIS: failed to enable radio");
783                 }
784                 drv->radio_enabled = 1;
785         }
786
787         res = ndis_set_oid(drv, OID_802_11_BSSID_LIST_SCAN, "    ", 4);
788         eloop_cancel_timeout(wpa_driver_ndis_scan_timeout, drv, drv->ctx);
789         eloop_register_timeout(7, 0, wpa_driver_ndis_scan_timeout, drv,
790                                drv->ctx);
791         return res;
792 }
793
794
795 static const u8 * wpa_scan_get_ie(const struct wpa_scan_res *res, u8 ie)
796 {
797         const u8 *end, *pos;
798
799         pos = (const u8 *) (res + 1);
800         end = pos + res->ie_len;
801
802         while (pos + 1 < end) {
803                 if (pos + 2 + pos[1] > end)
804                         break;
805                 if (pos[0] == ie)
806                         return pos;
807                 pos += 2 + pos[1];
808         }
809
810         return NULL;
811 }
812
813
814 static struct wpa_scan_res * wpa_driver_ndis_add_scan_ssid(
815         struct wpa_scan_res *r, NDIS_802_11_SSID *ssid)
816 {
817         struct wpa_scan_res *nr;
818         u8 *pos;
819
820         if (wpa_scan_get_ie(r, WLAN_EID_SSID))
821                 return r; /* SSID IE already present */
822
823         if (ssid->SsidLength == 0 || ssid->SsidLength > 32)
824                 return r; /* No valid SSID inside scan data */
825
826         nr = os_realloc(r, sizeof(*r) + r->ie_len + 2 + ssid->SsidLength);
827         if (nr == NULL)
828                 return r;
829
830         pos = ((u8 *) (nr + 1)) + nr->ie_len;
831         *pos++ = WLAN_EID_SSID;
832         *pos++ = ssid->SsidLength;
833         os_memcpy(pos, ssid->Ssid, ssid->SsidLength);
834         nr->ie_len += 2 + ssid->SsidLength;
835
836         return nr;
837 }
838
839
840 static struct wpa_scan_results * wpa_driver_ndis_get_scan_results(void *priv)
841 {
842         struct wpa_driver_ndis_data *drv = priv;
843         NDIS_802_11_BSSID_LIST_EX *b;
844         size_t blen, count, i;
845         int len;
846         char *pos;
847         struct wpa_scan_results *results;
848         struct wpa_scan_res *r;
849
850         blen = 65535;
851         b = os_zalloc(blen);
852         if (b == NULL)
853                 return NULL;
854         len = ndis_get_oid(drv, OID_802_11_BSSID_LIST, (char *) b, blen);
855         if (len < 0) {
856                 wpa_printf(MSG_DEBUG, "NDIS: failed to get scan results");
857                 os_free(b);
858                 return NULL;
859         }
860         count = b->NumberOfItems;
861
862         results = os_zalloc(sizeof(*results));
863         if (results == NULL) {
864                 os_free(b);
865                 return NULL;
866         }
867         results->res = os_zalloc(count * sizeof(struct wpa_scan_res *));
868         if (results->res == NULL) {
869                 os_free(results);
870                 os_free(b);
871                 return NULL;
872         }
873
874         pos = (char *) &b->Bssid[0];
875         for (i = 0; i < count; i++) {
876                 NDIS_WLAN_BSSID_EX *bss = (NDIS_WLAN_BSSID_EX *) pos;
877                 NDIS_802_11_FIXED_IEs *fixed;
878
879                 if (bss->IELength < sizeof(NDIS_802_11_FIXED_IEs)) {
880                         wpa_printf(MSG_DEBUG, "NDIS: too small IELength=%d",
881                                    (int) bss->IELength);
882                         break;
883                 }
884                 if (((char *) bss->IEs) + bss->IELength  > (char *) b + blen) {
885                         /*
886                          * Some NDIS drivers have been reported to include an
887                          * entry with an invalid IELength in scan results and
888                          * this has crashed wpa_supplicant, so validate the
889                          * returned value before using it.
890                          */
891                         wpa_printf(MSG_DEBUG, "NDIS: skipped invalid scan "
892                                    "result IE (BSSID=" MACSTR ") IELength=%d",
893                                    MAC2STR(bss->MacAddress),
894                                    (int) bss->IELength);
895                         break;
896                 }
897
898                 r = os_zalloc(sizeof(*r) + bss->IELength -
899                               sizeof(NDIS_802_11_FIXED_IEs));
900                 if (r == NULL)
901                         break;
902
903                 os_memcpy(r->bssid, bss->MacAddress, ETH_ALEN);
904                 r->level = (int) bss->Rssi;
905                 r->freq = bss->Configuration.DSConfig / 1000;
906                 fixed = (NDIS_802_11_FIXED_IEs *) bss->IEs;
907                 r->beacon_int = WPA_GET_LE16((u8 *) &fixed->BeaconInterval);
908                 r->caps = WPA_GET_LE16((u8 *) &fixed->Capabilities);
909                 r->tsf = WPA_GET_LE64(fixed->Timestamp);
910                 os_memcpy(r + 1, bss->IEs + sizeof(NDIS_802_11_FIXED_IEs),
911                           bss->IELength - sizeof(NDIS_802_11_FIXED_IEs));
912                 r->ie_len = bss->IELength - sizeof(NDIS_802_11_FIXED_IEs);
913                 r = wpa_driver_ndis_add_scan_ssid(r, &bss->Ssid);
914
915                 results->res[results->num++] = r;
916
917                 pos += bss->Length;
918                 if (pos > (char *) b + blen)
919                         break;
920         }
921
922         os_free(b);
923
924         return results;
925 }
926
927
928 static int wpa_driver_ndis_remove_key(struct wpa_driver_ndis_data *drv,
929                                       int key_idx, const u8 *addr,
930                                       const u8 *bssid, int pairwise)
931 {
932         NDIS_802_11_REMOVE_KEY rkey;
933         NDIS_802_11_KEY_INDEX index;
934         int res, res2;
935
936         os_memset(&rkey, 0, sizeof(rkey));
937
938         rkey.Length = sizeof(rkey);
939         rkey.KeyIndex = key_idx;
940         if (pairwise)
941                 rkey.KeyIndex |= 1 << 30;
942         os_memcpy(rkey.BSSID, bssid, ETH_ALEN);
943
944         res = ndis_set_oid(drv, OID_802_11_REMOVE_KEY, (char *) &rkey,
945                            sizeof(rkey));
946         if (!pairwise) {
947                 index = key_idx;
948                 res2 = ndis_set_oid(drv, OID_802_11_REMOVE_WEP,
949                                     (char *) &index, sizeof(index));
950         } else
951                 res2 = 0;
952
953         if (res < 0 && res2 < 0)
954                 return -1;
955         return 0;
956 }
957
958
959 static int wpa_driver_ndis_add_wep(struct wpa_driver_ndis_data *drv,
960                                    int pairwise, int key_idx, int set_tx,
961                                    const u8 *key, size_t key_len)
962 {
963         NDIS_802_11_WEP *wep;
964         size_t len;
965         int res;
966
967         len = 12 + key_len;
968         wep = os_zalloc(len);
969         if (wep == NULL)
970                 return -1;
971         wep->Length = len;
972         wep->KeyIndex = key_idx;
973         if (set_tx)
974                 wep->KeyIndex |= 1 << 31;
975 #if 0 /* Setting bit30 does not seem to work with some NDIS drivers */
976         if (pairwise)
977                 wep->KeyIndex |= 1 << 30;
978 #endif
979         wep->KeyLength = key_len;
980         os_memcpy(wep->KeyMaterial, key, key_len);
981
982         wpa_hexdump_key(MSG_MSGDUMP, "NDIS: OID_802_11_ADD_WEP",
983                         (u8 *) wep, len);
984         res = ndis_set_oid(drv, OID_802_11_ADD_WEP, (char *) wep, len);
985
986         os_free(wep);
987
988         return res;
989 }
990
991
992 static int wpa_driver_ndis_set_key(const char *ifname, void *priv,
993                                    enum wpa_alg alg, const u8 *addr,
994                                    int key_idx, int set_tx,
995                                    const u8 *seq, size_t seq_len,
996                                    const u8 *key, size_t key_len)
997 {
998         struct wpa_driver_ndis_data *drv = priv;
999         size_t len, i;
1000         NDIS_802_11_KEY *nkey;
1001         int res, pairwise;
1002         u8 bssid[ETH_ALEN];
1003
1004         if (addr == NULL || os_memcmp(addr, "\xff\xff\xff\xff\xff\xff",
1005                                       ETH_ALEN) == 0) {
1006                 /* Group Key */
1007                 pairwise = 0;
1008                 if (wpa_driver_ndis_get_bssid(drv, bssid) < 0)
1009                         os_memset(bssid, 0xff, ETH_ALEN);
1010         } else {
1011                 /* Pairwise Key */
1012                 pairwise = 1;
1013                 os_memcpy(bssid, addr, ETH_ALEN);
1014         }
1015
1016         if (alg == WPA_ALG_NONE || key_len == 0) {
1017                 return wpa_driver_ndis_remove_key(drv, key_idx, addr, bssid,
1018                                                   pairwise);
1019         }
1020
1021         if (alg == WPA_ALG_WEP) {
1022                 return wpa_driver_ndis_add_wep(drv, pairwise, key_idx, set_tx,
1023                                                key, key_len);
1024         }
1025
1026         len = 12 + 6 + 6 + 8 + key_len;
1027
1028         nkey = os_zalloc(len);
1029         if (nkey == NULL)
1030                 return -1;
1031
1032         nkey->Length = len;
1033         nkey->KeyIndex = key_idx;
1034         if (set_tx)
1035                 nkey->KeyIndex |= 1 << 31;
1036         if (pairwise)
1037                 nkey->KeyIndex |= 1 << 30;
1038         if (seq && seq_len)
1039                 nkey->KeyIndex |= 1 << 29;
1040         nkey->KeyLength = key_len;
1041         os_memcpy(nkey->BSSID, bssid, ETH_ALEN);
1042         if (seq && seq_len) {
1043                 for (i = 0; i < seq_len; i++)
1044                         nkey->KeyRSC |= (ULONGLONG) seq[i] << (i * 8);
1045         }
1046         if (alg == WPA_ALG_TKIP && key_len == 32) {
1047                 os_memcpy(nkey->KeyMaterial, key, 16);
1048                 os_memcpy(nkey->KeyMaterial + 16, key + 24, 8);
1049                 os_memcpy(nkey->KeyMaterial + 24, key + 16, 8);
1050         } else {
1051                 os_memcpy(nkey->KeyMaterial, key, key_len);
1052         }
1053
1054         wpa_hexdump_key(MSG_MSGDUMP, "NDIS: OID_802_11_ADD_KEY",
1055                         (u8 *) nkey, len);
1056         res = ndis_set_oid(drv, OID_802_11_ADD_KEY, (char *) nkey, len);
1057         os_free(nkey);
1058
1059         return res;
1060 }
1061
1062
1063 static int
1064 wpa_driver_ndis_associate(void *priv,
1065                           struct wpa_driver_associate_params *params)
1066 {
1067         struct wpa_driver_ndis_data *drv = priv;
1068         u32 auth_mode, encr, priv_mode, mode;
1069         u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
1070
1071         drv->mode = params->mode;
1072
1073         /* Note: Setting OID_802_11_INFRASTRUCTURE_MODE clears current keys,
1074          * so static WEP keys needs to be set again after this. */
1075         if (params->mode == IEEE80211_MODE_IBSS) {
1076                 mode = Ndis802_11IBSS;
1077                 /* Need to make sure that BSSID polling is enabled for
1078                  * IBSS mode. */
1079                 eloop_cancel_timeout(wpa_driver_ndis_poll_timeout, drv, NULL);
1080                 eloop_register_timeout(1, 0, wpa_driver_ndis_poll_timeout,
1081                                        drv, NULL);
1082         } else
1083                 mode = Ndis802_11Infrastructure;
1084         if (ndis_set_oid(drv, OID_802_11_INFRASTRUCTURE_MODE,
1085                          (char *) &mode, sizeof(mode)) < 0) {
1086                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
1087                            "OID_802_11_INFRASTRUCTURE_MODE (%d)",
1088                            (int) mode);
1089                 /* Try to continue anyway */
1090         }
1091
1092         if (params->key_mgmt_suite == KEY_MGMT_NONE ||
1093             params->key_mgmt_suite == KEY_MGMT_802_1X_NO_WPA) {
1094                 /* Re-set WEP keys if static WEP configuration is used. */
1095                 int i;
1096                 for (i = 0; i < 4; i++) {
1097                         if (!params->wep_key[i])
1098                                 continue;
1099                         wpa_printf(MSG_DEBUG, "NDIS: Re-setting static WEP "
1100                                    "key %d", i);
1101                         wpa_driver_ndis_set_key(drv->ifname, drv, WPA_ALG_WEP,
1102                                                 bcast, i,
1103                                                 i == params->wep_tx_keyidx,
1104                                                 NULL, 0, params->wep_key[i],
1105                                                 params->wep_key_len[i]);
1106                 }
1107         }
1108
1109         if (params->wpa_ie == NULL || params->wpa_ie_len == 0) {
1110                 if (params->auth_alg & WPA_AUTH_ALG_SHARED) {
1111                         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
1112                                 auth_mode = Ndis802_11AuthModeAutoSwitch;
1113                         else
1114                                 auth_mode = Ndis802_11AuthModeShared;
1115                 } else
1116                         auth_mode = Ndis802_11AuthModeOpen;
1117                 priv_mode = Ndis802_11PrivFilterAcceptAll;
1118         } else if (params->wpa_ie[0] == WLAN_EID_RSN) {
1119                 priv_mode = Ndis802_11PrivFilter8021xWEP;
1120                 if (params->key_mgmt_suite == KEY_MGMT_PSK)
1121                         auth_mode = Ndis802_11AuthModeWPA2PSK;
1122                 else
1123                         auth_mode = Ndis802_11AuthModeWPA2;
1124 #ifdef CONFIG_WPS
1125         } else if (params->key_mgmt_suite == KEY_MGMT_WPS) {
1126                 auth_mode = Ndis802_11AuthModeOpen;
1127                 priv_mode = Ndis802_11PrivFilterAcceptAll;
1128                 if (params->wps == WPS_MODE_PRIVACY) {
1129                         u8 dummy_key[5] = { 0x11, 0x22, 0x33, 0x44, 0x55 };
1130                         /*
1131                          * Some NDIS drivers refuse to associate in open mode
1132                          * configuration due to Privacy field mismatch, so use
1133                          * a workaround to make the configuration look like
1134                          * matching one for WPS provisioning.
1135                          */
1136                         wpa_printf(MSG_DEBUG, "NDIS: Set dummy WEP key as a "
1137                                    "workaround to allow driver to associate "
1138                                    "for WPS");
1139                         wpa_driver_ndis_set_key(drv->ifname, drv, WPA_ALG_WEP,
1140                                                 bcast, 0, 1,
1141                                                 NULL, 0, dummy_key,
1142                                                 sizeof(dummy_key));
1143                 }
1144 #endif /* CONFIG_WPS */
1145         } else {
1146                 priv_mode = Ndis802_11PrivFilter8021xWEP;
1147                 if (params->key_mgmt_suite == KEY_MGMT_WPA_NONE)
1148                         auth_mode = Ndis802_11AuthModeWPANone;
1149                 else if (params->key_mgmt_suite == KEY_MGMT_PSK)
1150                         auth_mode = Ndis802_11AuthModeWPAPSK;
1151                 else
1152                         auth_mode = Ndis802_11AuthModeWPA;
1153         }
1154
1155         switch (params->pairwise_suite) {
1156         case CIPHER_CCMP:
1157                 encr = Ndis802_11Encryption3Enabled;
1158                 break;
1159         case CIPHER_TKIP:
1160                 encr = Ndis802_11Encryption2Enabled;
1161                 break;
1162         case CIPHER_WEP40:
1163         case CIPHER_WEP104:
1164                 encr = Ndis802_11Encryption1Enabled;
1165                 break;
1166         case CIPHER_NONE:
1167 #ifdef CONFIG_WPS
1168                 if (params->wps == WPS_MODE_PRIVACY) {
1169                         encr = Ndis802_11Encryption1Enabled;
1170                         break;
1171                 }
1172 #endif /* CONFIG_WPS */
1173                 if (params->group_suite == CIPHER_CCMP)
1174                         encr = Ndis802_11Encryption3Enabled;
1175                 else if (params->group_suite == CIPHER_TKIP)
1176                         encr = Ndis802_11Encryption2Enabled;
1177                 else
1178                         encr = Ndis802_11EncryptionDisabled;
1179                 break;
1180         default:
1181 #ifdef CONFIG_WPS
1182                 if (params->wps == WPS_MODE_PRIVACY) {
1183                         encr = Ndis802_11Encryption1Enabled;
1184                         break;
1185                 }
1186 #endif /* CONFIG_WPS */
1187                 encr = Ndis802_11EncryptionDisabled;
1188                 break;
1189         };
1190
1191         if (ndis_set_oid(drv, OID_802_11_PRIVACY_FILTER,
1192                          (char *) &priv_mode, sizeof(priv_mode)) < 0) {
1193                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
1194                            "OID_802_11_PRIVACY_FILTER (%d)",
1195                            (int) priv_mode);
1196                 /* Try to continue anyway */
1197         }
1198
1199         ndis_set_auth_mode(drv, auth_mode);
1200         ndis_set_encr_status(drv, encr);
1201
1202         if (params->bssid) {
1203                 ndis_set_oid(drv, OID_802_11_BSSID, (char *) params->bssid,
1204                              ETH_ALEN);
1205                 drv->oid_bssid_set = 1;
1206         } else if (drv->oid_bssid_set) {
1207                 ndis_set_oid(drv, OID_802_11_BSSID, "\xff\xff\xff\xff\xff\xff",
1208                              ETH_ALEN);
1209                 drv->oid_bssid_set = 0;
1210         }
1211
1212         return wpa_driver_ndis_set_ssid(drv, params->ssid, params->ssid_len);
1213 }
1214
1215
1216 static int wpa_driver_ndis_set_pmkid(struct wpa_driver_ndis_data *drv)
1217 {
1218         int len, count, i, ret;
1219         struct ndis_pmkid_entry *entry;
1220         NDIS_802_11_PMKID *p;
1221
1222         count = 0;
1223         entry = drv->pmkid;
1224         while (entry) {
1225                 count++;
1226                 if (count >= drv->no_of_pmkid)
1227                         break;
1228                 entry = entry->next;
1229         }
1230         len = 8 + count * sizeof(BSSID_INFO);
1231         p = os_zalloc(len);
1232         if (p == NULL)
1233                 return -1;
1234
1235         p->Length = len;
1236         p->BSSIDInfoCount = count;
1237         entry = drv->pmkid;
1238         for (i = 0; i < count; i++) {
1239                 os_memcpy(&p->BSSIDInfo[i].BSSID, entry->bssid, ETH_ALEN);
1240                 os_memcpy(&p->BSSIDInfo[i].PMKID, entry->pmkid, 16);
1241                 entry = entry->next;
1242         }
1243         wpa_hexdump(MSG_MSGDUMP, "NDIS: OID_802_11_PMKID", (u8 *) p, len);
1244         ret = ndis_set_oid(drv, OID_802_11_PMKID, (char *) p, len);
1245         os_free(p);
1246         return ret;
1247 }
1248
1249
1250 static int wpa_driver_ndis_add_pmkid(void *priv, const u8 *bssid,
1251                                      const u8 *pmkid)
1252 {
1253         struct wpa_driver_ndis_data *drv = priv;
1254         struct ndis_pmkid_entry *entry, *prev;
1255
1256         if (drv->no_of_pmkid == 0)
1257                 return 0;
1258
1259         prev = NULL;
1260         entry = drv->pmkid;
1261         while (entry) {
1262                 if (os_memcmp(entry->bssid, bssid, ETH_ALEN) == 0)
1263                         break;
1264                 prev = entry;
1265                 entry = entry->next;
1266         }
1267
1268         if (entry) {
1269                 /* Replace existing entry for this BSSID and move it into the
1270                  * beginning of the list. */
1271                 os_memcpy(entry->pmkid, pmkid, 16);
1272                 if (prev) {
1273                         prev->next = entry->next;
1274                         entry->next = drv->pmkid;
1275                         drv->pmkid = entry;
1276                 }
1277         } else {
1278                 entry = os_malloc(sizeof(*entry));
1279                 if (entry) {
1280                         os_memcpy(entry->bssid, bssid, ETH_ALEN);
1281                         os_memcpy(entry->pmkid, pmkid, 16);
1282                         entry->next = drv->pmkid;
1283                         drv->pmkid = entry;
1284                 }
1285         }
1286
1287         return wpa_driver_ndis_set_pmkid(drv);
1288 }
1289
1290
1291 static int wpa_driver_ndis_remove_pmkid(void *priv, const u8 *bssid,
1292                                         const u8 *pmkid)
1293 {
1294         struct wpa_driver_ndis_data *drv = priv;
1295         struct ndis_pmkid_entry *entry, *prev;
1296
1297         if (drv->no_of_pmkid == 0)
1298                 return 0;
1299
1300         entry = drv->pmkid;
1301         prev = NULL;
1302         while (entry) {
1303                 if (os_memcmp(entry->bssid, bssid, ETH_ALEN) == 0 &&
1304                     os_memcmp(entry->pmkid, pmkid, 16) == 0) {
1305                         if (prev)
1306                                 prev->next = entry->next;
1307                         else
1308                                 drv->pmkid = entry->next;
1309                         os_free(entry);
1310                         break;
1311                 }
1312                 prev = entry;
1313                 entry = entry->next;
1314         }
1315         return wpa_driver_ndis_set_pmkid(drv);
1316 }
1317
1318
1319 static int wpa_driver_ndis_flush_pmkid(void *priv)
1320 {
1321         struct wpa_driver_ndis_data *drv = priv;
1322         NDIS_802_11_PMKID p;
1323         struct ndis_pmkid_entry *pmkid, *prev;
1324         int prev_authmode, ret;
1325
1326         if (drv->no_of_pmkid == 0)
1327                 return 0;
1328
1329         pmkid = drv->pmkid;
1330         drv->pmkid = NULL;
1331         while (pmkid) {
1332                 prev = pmkid;
1333                 pmkid = pmkid->next;
1334                 os_free(prev);
1335         }
1336
1337         /*
1338          * Some drivers may refuse OID_802_11_PMKID if authMode is not set to
1339          * WPA2, so change authMode temporarily, if needed.
1340          */
1341         prev_authmode = ndis_get_auth_mode(drv);
1342         if (prev_authmode != Ndis802_11AuthModeWPA2)
1343                 ndis_set_auth_mode(drv, Ndis802_11AuthModeWPA2);
1344
1345         os_memset(&p, 0, sizeof(p));
1346         p.Length = 8;
1347         p.BSSIDInfoCount = 0;
1348         wpa_hexdump(MSG_MSGDUMP, "NDIS: OID_802_11_PMKID (flush)",
1349                     (u8 *) &p, 8);
1350         ret = ndis_set_oid(drv, OID_802_11_PMKID, (char *) &p, 8);
1351
1352         if (prev_authmode != Ndis802_11AuthModeWPA2)
1353                 ndis_set_auth_mode(drv, prev_authmode);
1354
1355         return ret;
1356 }
1357
1358
1359 static int wpa_driver_ndis_get_associnfo(struct wpa_driver_ndis_data *drv)
1360 {
1361         char buf[512], *pos;
1362         NDIS_802_11_ASSOCIATION_INFORMATION *ai;
1363         int len;
1364         union wpa_event_data data;
1365         NDIS_802_11_BSSID_LIST_EX *b;
1366         size_t blen, i;
1367
1368         len = ndis_get_oid(drv, OID_802_11_ASSOCIATION_INFORMATION, buf,
1369                            sizeof(buf));
1370         if (len < 0) {
1371                 wpa_printf(MSG_DEBUG, "NDIS: failed to get association "
1372                            "information");
1373                 return -1;
1374         }
1375         if (len > sizeof(buf)) {
1376                 /* Some drivers seem to be producing incorrect length for this
1377                  * data. Limit the length to the current buffer size to avoid
1378                  * crashing in hexdump. The data seems to be otherwise valid,
1379                  * so better try to use it. */
1380                 wpa_printf(MSG_DEBUG, "NDIS: ignored bogus association "
1381                            "information length %d", len);
1382                 len = ndis_get_oid(drv, OID_802_11_ASSOCIATION_INFORMATION,
1383                                    buf, sizeof(buf));
1384                 if (len < -1) {
1385                         wpa_printf(MSG_DEBUG, "NDIS: re-reading association "
1386                                    "information failed");
1387                         return -1;
1388                 }
1389                 if (len > sizeof(buf)) {
1390                         wpa_printf(MSG_DEBUG, "NDIS: ignored bogus association"
1391                                    " information length %d (re-read)", len);
1392                         len = sizeof(buf);
1393                 }
1394         }
1395         wpa_hexdump(MSG_MSGDUMP, "NDIS: association information",
1396                     (u8 *) buf, len);
1397         if (len < sizeof(*ai)) {
1398                 wpa_printf(MSG_DEBUG, "NDIS: too short association "
1399                            "information");
1400                 return -1;
1401         }
1402         ai = (NDIS_802_11_ASSOCIATION_INFORMATION *) buf;
1403         wpa_printf(MSG_DEBUG, "NDIS: ReqFixed=0x%x RespFixed=0x%x off_req=%d "
1404                    "off_resp=%d len_req=%d len_resp=%d",
1405                    ai->AvailableRequestFixedIEs, ai->AvailableResponseFixedIEs,
1406                    (int) ai->OffsetRequestIEs, (int) ai->OffsetResponseIEs,
1407                    (int) ai->RequestIELength, (int) ai->ResponseIELength);
1408
1409         if (ai->OffsetRequestIEs + ai->RequestIELength > (unsigned) len ||
1410             ai->OffsetResponseIEs + ai->ResponseIELength > (unsigned) len) {
1411                 wpa_printf(MSG_DEBUG, "NDIS: association information - "
1412                            "IE overflow");
1413                 return -1;
1414         }
1415
1416         wpa_hexdump(MSG_MSGDUMP, "NDIS: Request IEs",
1417                     (u8 *) buf + ai->OffsetRequestIEs, ai->RequestIELength);
1418         wpa_hexdump(MSG_MSGDUMP, "NDIS: Response IEs",
1419                     (u8 *) buf + ai->OffsetResponseIEs, ai->ResponseIELength);
1420
1421         os_memset(&data, 0, sizeof(data));
1422         data.assoc_info.req_ies = (u8 *) buf + ai->OffsetRequestIEs;
1423         data.assoc_info.req_ies_len = ai->RequestIELength;
1424         data.assoc_info.resp_ies = (u8 *) buf + ai->OffsetResponseIEs;
1425         data.assoc_info.resp_ies_len = ai->ResponseIELength;
1426
1427         blen = 65535;
1428         b = os_zalloc(blen);
1429         if (b == NULL)
1430                 goto skip_scan_results;
1431         len = ndis_get_oid(drv, OID_802_11_BSSID_LIST, (char *) b, blen);
1432         if (len < 0) {
1433                 wpa_printf(MSG_DEBUG, "NDIS: failed to get scan results");
1434                 os_free(b);
1435                 b = NULL;
1436                 goto skip_scan_results;
1437         }
1438         wpa_printf(MSG_DEBUG, "NDIS: %d BSSID items to process for AssocInfo",
1439                    (unsigned int) b->NumberOfItems);
1440
1441         pos = (char *) &b->Bssid[0];
1442         for (i = 0; i < b->NumberOfItems; i++) {
1443                 NDIS_WLAN_BSSID_EX *bss = (NDIS_WLAN_BSSID_EX *) pos;
1444                 if (os_memcmp(drv->bssid, bss->MacAddress, ETH_ALEN) == 0 &&
1445                     bss->IELength > sizeof(NDIS_802_11_FIXED_IEs)) {
1446                         data.assoc_info.beacon_ies =
1447                                 ((u8 *) bss->IEs) +
1448                                 sizeof(NDIS_802_11_FIXED_IEs);
1449                         data.assoc_info.beacon_ies_len =
1450                                 bss->IELength - sizeof(NDIS_802_11_FIXED_IEs);
1451                         wpa_hexdump(MSG_MSGDUMP, "NDIS: Beacon IEs",
1452                                     data.assoc_info.beacon_ies,
1453                                     data.assoc_info.beacon_ies_len);
1454                         break;
1455                 }
1456                 pos += bss->Length;
1457                 if (pos > (char *) b + blen)
1458                         break;
1459         }
1460
1461 skip_scan_results:
1462         wpa_supplicant_event(drv->ctx, EVENT_ASSOCINFO, &data);
1463
1464         os_free(b);
1465
1466         return 0;
1467 }
1468
1469
1470 static void wpa_driver_ndis_poll_timeout(void *eloop_ctx, void *timeout_ctx)
1471 {
1472         struct wpa_driver_ndis_data *drv = eloop_ctx;
1473         u8 bssid[ETH_ALEN];
1474         int poll;
1475
1476         if (drv->wired)
1477                 return;
1478
1479         if (wpa_driver_ndis_get_bssid(drv, bssid)) {
1480                 /* Disconnected */
1481                 if (!is_zero_ether_addr(drv->bssid)) {
1482                         os_memset(drv->bssid, 0, ETH_ALEN);
1483                         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
1484                 }
1485         } else {
1486                 /* Connected */
1487                 if (os_memcmp(drv->bssid, bssid, ETH_ALEN) != 0) {
1488                         os_memcpy(drv->bssid, bssid, ETH_ALEN);
1489                         wpa_driver_ndis_get_associnfo(drv);
1490                         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
1491                 }
1492         }
1493
1494         /* When using integrated NDIS event receiver, we can skip BSSID
1495          * polling when using infrastructure network. However, when using
1496          * IBSS mode, many driver do not seem to generate connection event,
1497          * so we need to enable BSSID polling to figure out when IBSS network
1498          * has been formed.
1499          */
1500         poll = drv->mode == IEEE80211_MODE_IBSS;
1501 #ifndef CONFIG_NDIS_EVENTS_INTEGRATED
1502 #ifndef _WIN32_WCE
1503         poll = 1;
1504 #endif /* _WIN32_WCE */
1505 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
1506
1507         if (poll) {
1508                 eloop_register_timeout(1, 0, wpa_driver_ndis_poll_timeout,
1509                                         drv, NULL);
1510         }
1511 }
1512
1513
1514 static void wpa_driver_ndis_poll(void *priv)
1515 {
1516         struct wpa_driver_ndis_data *drv = priv;
1517         eloop_cancel_timeout(wpa_driver_ndis_poll_timeout, drv, NULL);
1518         wpa_driver_ndis_poll_timeout(drv, NULL);
1519 }
1520
1521
1522 /* Called when driver generates Media Connect Event by calling
1523  * NdisMIndicateStatus() with NDIS_STATUS_MEDIA_CONNECT */
1524 void wpa_driver_ndis_event_connect(struct wpa_driver_ndis_data *drv)
1525 {
1526         wpa_printf(MSG_DEBUG, "NDIS: Media Connect Event");
1527         if (wpa_driver_ndis_get_bssid(drv, drv->bssid) == 0) {
1528                 wpa_driver_ndis_get_associnfo(drv);
1529                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
1530         }
1531 }
1532
1533
1534 /* Called when driver generates Media Disconnect Event by calling
1535  * NdisMIndicateStatus() with NDIS_STATUS_MEDIA_DISCONNECT */
1536 void wpa_driver_ndis_event_disconnect(struct wpa_driver_ndis_data *drv)
1537 {
1538         wpa_printf(MSG_DEBUG, "NDIS: Media Disconnect Event");
1539         os_memset(drv->bssid, 0, ETH_ALEN);
1540         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
1541 }
1542
1543
1544 static void wpa_driver_ndis_event_auth(struct wpa_driver_ndis_data *drv,
1545                                        const u8 *data, size_t data_len)
1546 {
1547         NDIS_802_11_AUTHENTICATION_REQUEST *req;
1548         int pairwise = 0, group = 0;
1549         union wpa_event_data event;
1550
1551         if (data_len < sizeof(*req)) {
1552                 wpa_printf(MSG_DEBUG, "NDIS: Too short Authentication Request "
1553                            "Event (len=%d)", data_len);
1554                 return;
1555         }
1556         req = (NDIS_802_11_AUTHENTICATION_REQUEST *) data;
1557
1558         wpa_printf(MSG_DEBUG, "NDIS: Authentication Request Event: "
1559                    "Bssid " MACSTR " Flags 0x%x",
1560                    MAC2STR(req->Bssid), (int) req->Flags);
1561
1562         if ((req->Flags & NDIS_802_11_AUTH_REQUEST_PAIRWISE_ERROR) ==
1563             NDIS_802_11_AUTH_REQUEST_PAIRWISE_ERROR)
1564                 pairwise = 1;
1565         else if ((req->Flags & NDIS_802_11_AUTH_REQUEST_GROUP_ERROR) ==
1566             NDIS_802_11_AUTH_REQUEST_GROUP_ERROR)
1567                 group = 1;
1568
1569         if (pairwise || group) {
1570                 os_memset(&event, 0, sizeof(event));
1571                 event.michael_mic_failure.unicast = pairwise;
1572                 wpa_supplicant_event(drv->ctx, EVENT_MICHAEL_MIC_FAILURE,
1573                                      &event);
1574         }
1575 }
1576
1577
1578 static void wpa_driver_ndis_event_pmkid(struct wpa_driver_ndis_data *drv,
1579                                         const u8 *data, size_t data_len)
1580 {
1581         NDIS_802_11_PMKID_CANDIDATE_LIST *pmkid;
1582         size_t i;
1583         union wpa_event_data event;
1584
1585         if (data_len < 8) {
1586                 wpa_printf(MSG_DEBUG, "NDIS: Too short PMKID Candidate List "
1587                            "Event (len=%d)", data_len);
1588                 return;
1589         }
1590         pmkid = (NDIS_802_11_PMKID_CANDIDATE_LIST *) data;
1591         wpa_printf(MSG_DEBUG, "NDIS: PMKID Candidate List Event - Version %d "
1592                    "NumCandidates %d",
1593                    (int) pmkid->Version, (int) pmkid->NumCandidates);
1594
1595         if (pmkid->Version != 1) {
1596                 wpa_printf(MSG_DEBUG, "NDIS: Unsupported PMKID Candidate List "
1597                            "Version %d", (int) pmkid->Version);
1598                 return;
1599         }
1600
1601         if (data_len < 8 + pmkid->NumCandidates * sizeof(PMKID_CANDIDATE)) {
1602                 wpa_printf(MSG_DEBUG, "NDIS: PMKID Candidate List underflow");
1603                 return;
1604         }
1605
1606         os_memset(&event, 0, sizeof(event));
1607         for (i = 0; i < pmkid->NumCandidates; i++) {
1608                 PMKID_CANDIDATE *p = &pmkid->CandidateList[i];
1609                 wpa_printf(MSG_DEBUG, "NDIS: %d: " MACSTR " Flags 0x%x",
1610                            i, MAC2STR(p->BSSID), (int) p->Flags);
1611                 os_memcpy(event.pmkid_candidate.bssid, p->BSSID, ETH_ALEN);
1612                 event.pmkid_candidate.index = i;
1613                 event.pmkid_candidate.preauth =
1614                         p->Flags & NDIS_802_11_PMKID_CANDIDATE_PREAUTH_ENABLED;
1615                 wpa_supplicant_event(drv->ctx, EVENT_PMKID_CANDIDATE,
1616                                      &event);
1617         }
1618 }
1619
1620
1621 /* Called when driver calls NdisMIndicateStatus() with
1622  * NDIS_STATUS_MEDIA_SPECIFIC_INDICATION */
1623 void wpa_driver_ndis_event_media_specific(struct wpa_driver_ndis_data *drv,
1624                                           const u8 *data, size_t data_len)
1625 {
1626         NDIS_802_11_STATUS_INDICATION *status;
1627
1628         if (data == NULL || data_len < sizeof(*status))
1629                 return;
1630
1631         wpa_hexdump(MSG_DEBUG, "NDIS: Media Specific Indication",
1632                     data, data_len);
1633
1634         status = (NDIS_802_11_STATUS_INDICATION *) data;
1635         data += sizeof(status);
1636         data_len -= sizeof(status);
1637
1638         switch (status->StatusType) {
1639         case Ndis802_11StatusType_Authentication:
1640                 wpa_driver_ndis_event_auth(drv, data, data_len);
1641                 break;
1642         case Ndis802_11StatusType_PMKID_CandidateList:
1643                 wpa_driver_ndis_event_pmkid(drv, data, data_len);
1644                 break;
1645         default:
1646                 wpa_printf(MSG_DEBUG, "NDIS: Unknown StatusType %d",
1647                            (int) status->StatusType);
1648                 break;
1649         }
1650 }
1651
1652
1653 /* Called when an adapter is added */
1654 void wpa_driver_ndis_event_adapter_arrival(struct wpa_driver_ndis_data *drv)
1655 {
1656         union wpa_event_data event;
1657         int i;
1658
1659         wpa_printf(MSG_DEBUG, "NDIS: Notify Adapter Arrival");
1660
1661         for (i = 0; i < 30; i++) {
1662                 /* Re-open Packet32/NDISUIO connection */
1663                 wpa_driver_ndis_adapter_close(drv);
1664                 if (wpa_driver_ndis_adapter_init(drv) < 0 ||
1665                     wpa_driver_ndis_adapter_open(drv) < 0) {
1666                         wpa_printf(MSG_DEBUG, "NDIS: Driver re-initialization "
1667                                    "(%d) failed", i);
1668                         os_sleep(1, 0);
1669                 } else {
1670                         wpa_printf(MSG_DEBUG, "NDIS: Driver re-initialized");
1671                         break;
1672                 }
1673         }
1674
1675         os_memset(&event, 0, sizeof(event));
1676         os_strlcpy(event.interface_status.ifname, drv->ifname,
1677                    sizeof(event.interface_status.ifname));
1678         event.interface_status.ievent = EVENT_INTERFACE_ADDED;
1679         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_STATUS, &event);
1680 }
1681
1682
1683 /* Called when an adapter is removed */
1684 void wpa_driver_ndis_event_adapter_removal(struct wpa_driver_ndis_data *drv)
1685 {
1686         union wpa_event_data event;
1687
1688         wpa_printf(MSG_DEBUG, "NDIS: Notify Adapter Removal");
1689         os_memset(&event, 0, sizeof(event));
1690         os_strlcpy(event.interface_status.ifname, drv->ifname,
1691                    sizeof(event.interface_status.ifname));
1692         event.interface_status.ievent = EVENT_INTERFACE_REMOVED;
1693         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_STATUS, &event);
1694 }
1695
1696
1697 static void
1698 wpa_driver_ndis_get_wpa_capability(struct wpa_driver_ndis_data *drv)
1699 {
1700         wpa_printf(MSG_DEBUG, "NDIS: verifying driver WPA capability");
1701
1702         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeWPA) == 0 &&
1703             ndis_get_auth_mode(drv) == Ndis802_11AuthModeWPA) {
1704                 wpa_printf(MSG_DEBUG, "NDIS: WPA key management supported");
1705                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA;
1706         }
1707
1708         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeWPAPSK) == 0 &&
1709             ndis_get_auth_mode(drv) == Ndis802_11AuthModeWPAPSK) {
1710                 wpa_printf(MSG_DEBUG, "NDIS: WPA-PSK key management "
1711                            "supported");
1712                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK;
1713         }
1714
1715         if (ndis_set_encr_status(drv, Ndis802_11Encryption3Enabled) == 0 &&
1716             ndis_get_encr_status(drv) == Ndis802_11Encryption3KeyAbsent) {
1717                 wpa_printf(MSG_DEBUG, "NDIS: CCMP encryption supported");
1718                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_CCMP;
1719         }
1720
1721         if (ndis_set_encr_status(drv, Ndis802_11Encryption2Enabled) == 0 &&
1722             ndis_get_encr_status(drv) == Ndis802_11Encryption2KeyAbsent) {
1723                 wpa_printf(MSG_DEBUG, "NDIS: TKIP encryption supported");
1724                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_TKIP;
1725         }
1726
1727         if (ndis_set_encr_status(drv, Ndis802_11Encryption1Enabled) == 0 &&
1728             ndis_get_encr_status(drv) == Ndis802_11Encryption1KeyAbsent) {
1729                 wpa_printf(MSG_DEBUG, "NDIS: WEP encryption supported");
1730                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP40 |
1731                         WPA_DRIVER_CAPA_ENC_WEP104;
1732         }
1733
1734         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeShared) == 0 &&
1735             ndis_get_auth_mode(drv) == Ndis802_11AuthModeShared) {
1736                 drv->capa.auth |= WPA_DRIVER_AUTH_SHARED;
1737         }
1738
1739         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeOpen) == 0 &&
1740             ndis_get_auth_mode(drv) == Ndis802_11AuthModeOpen) {
1741                 drv->capa.auth |= WPA_DRIVER_AUTH_OPEN;
1742         }
1743
1744         ndis_set_encr_status(drv, Ndis802_11EncryptionDisabled);
1745
1746         /* Could also verify OID_802_11_ADD_KEY error reporting and
1747          * support for OID_802_11_ASSOCIATION_INFORMATION. */
1748
1749         if (drv->capa.key_mgmt & WPA_DRIVER_CAPA_KEY_MGMT_WPA &&
1750             drv->capa.enc & (WPA_DRIVER_CAPA_ENC_TKIP |
1751                              WPA_DRIVER_CAPA_ENC_CCMP)) {
1752                 wpa_printf(MSG_DEBUG, "NDIS: driver supports WPA");
1753                 drv->has_capability = 1;
1754         } else {
1755                 wpa_printf(MSG_DEBUG, "NDIS: no WPA support found");
1756         }
1757
1758         wpa_printf(MSG_DEBUG, "NDIS: driver capabilities: key_mgmt 0x%x "
1759                    "enc 0x%x auth 0x%x",
1760                    drv->capa.key_mgmt, drv->capa.enc, drv->capa.auth);
1761 }
1762
1763
1764 static void wpa_driver_ndis_get_capability(struct wpa_driver_ndis_data *drv)
1765 {
1766         char buf[512];
1767         int len;
1768         size_t i;
1769         NDIS_802_11_CAPABILITY *c;
1770
1771         drv->capa.flags = WPA_DRIVER_FLAGS_DRIVER_IE;
1772
1773         len = ndis_get_oid(drv, OID_802_11_CAPABILITY, buf, sizeof(buf));
1774         if (len < 0) {
1775                 wpa_driver_ndis_get_wpa_capability(drv);
1776                 return;
1777         }
1778
1779         wpa_hexdump(MSG_MSGDUMP, "OID_802_11_CAPABILITY", (u8 *) buf, len);
1780         c = (NDIS_802_11_CAPABILITY *) buf;
1781         if (len < sizeof(*c) || c->Version != 2) {
1782                 wpa_printf(MSG_DEBUG, "NDIS: unsupported "
1783                            "OID_802_11_CAPABILITY data");
1784                 return;
1785         }
1786         wpa_printf(MSG_DEBUG, "NDIS: Driver supports OID_802_11_CAPABILITY - "
1787                    "NoOfPMKIDs %d NoOfAuthEncrPairs %d",
1788                    (int) c->NoOfPMKIDs,
1789                    (int) c->NoOfAuthEncryptPairsSupported);
1790         drv->has_capability = 1;
1791         drv->no_of_pmkid = c->NoOfPMKIDs;
1792         for (i = 0; i < c->NoOfAuthEncryptPairsSupported; i++) {
1793                 NDIS_802_11_AUTHENTICATION_ENCRYPTION *ae;
1794                 ae = &c->AuthenticationEncryptionSupported[i];
1795                 if ((char *) (ae + 1) > buf + len) {
1796                         wpa_printf(MSG_DEBUG, "NDIS: auth/encr pair list "
1797                                    "overflow");
1798                         break;
1799                 }
1800                 wpa_printf(MSG_MSGDUMP, "NDIS: %d - auth %d encr %d",
1801                            i, (int) ae->AuthModeSupported,
1802                            (int) ae->EncryptStatusSupported);
1803                 switch (ae->AuthModeSupported) {
1804                 case Ndis802_11AuthModeOpen:
1805                         drv->capa.auth |= WPA_DRIVER_AUTH_OPEN;
1806                         break;
1807                 case Ndis802_11AuthModeShared:
1808                         drv->capa.auth |= WPA_DRIVER_AUTH_SHARED;
1809                         break;
1810                 case Ndis802_11AuthModeWPA:
1811                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA;
1812                         break;
1813                 case Ndis802_11AuthModeWPAPSK:
1814                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK;
1815                         break;
1816                 case Ndis802_11AuthModeWPA2:
1817                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA2;
1818                         break;
1819                 case Ndis802_11AuthModeWPA2PSK:
1820                         drv->capa.key_mgmt |=
1821                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
1822                         break;
1823                 case Ndis802_11AuthModeWPANone:
1824                         drv->capa.key_mgmt |=
1825                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_NONE;
1826                         break;
1827                 default:
1828                         break;
1829                 }
1830                 switch (ae->EncryptStatusSupported) {
1831                 case Ndis802_11Encryption1Enabled:
1832                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP40;
1833                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP104;
1834                         break;
1835                 case Ndis802_11Encryption2Enabled:
1836                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_TKIP;
1837                         break;
1838                 case Ndis802_11Encryption3Enabled:
1839                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_CCMP;
1840                         break;
1841                 default:
1842                         break;
1843                 }
1844         }
1845
1846         wpa_printf(MSG_DEBUG, "NDIS: driver capabilities: key_mgmt 0x%x "
1847                    "enc 0x%x auth 0x%x",
1848                    drv->capa.key_mgmt, drv->capa.enc, drv->capa.auth);
1849 }
1850
1851
1852 static int wpa_driver_ndis_get_capa(void *priv, struct wpa_driver_capa *capa)
1853 {
1854         struct wpa_driver_ndis_data *drv = priv;
1855         if (!drv->has_capability)
1856                 return -1;
1857         os_memcpy(capa, &drv->capa, sizeof(*capa));
1858         return 0;
1859 }
1860
1861
1862 static const char * wpa_driver_ndis_get_ifname(void *priv)
1863 {
1864         struct wpa_driver_ndis_data *drv = priv;
1865         return drv->ifname;
1866 }
1867
1868
1869 static const u8 * wpa_driver_ndis_get_mac_addr(void *priv)
1870 {
1871         struct wpa_driver_ndis_data *drv = priv;
1872         return drv->own_addr;
1873 }
1874
1875
1876 #ifdef _WIN32_WCE
1877
1878 #define NDISUIO_MSG_SIZE (sizeof(NDISUIO_DEVICE_NOTIFICATION) + 512)
1879
1880 static void ndisuio_notification_receive(void *eloop_data, void *user_ctx)
1881 {
1882         struct wpa_driver_ndis_data *drv = eloop_data;
1883         NDISUIO_DEVICE_NOTIFICATION *hdr;
1884         u8 buf[NDISUIO_MSG_SIZE];
1885         DWORD len, flags;
1886
1887         if (!ReadMsgQueue(drv->event_queue, buf, NDISUIO_MSG_SIZE, &len, 0,
1888                           &flags)) {
1889                 wpa_printf(MSG_DEBUG, "ndisuio_notification_receive: "
1890                            "ReadMsgQueue failed: %d", (int) GetLastError());
1891                 return;
1892         }
1893
1894         if (len < sizeof(NDISUIO_DEVICE_NOTIFICATION)) {
1895                 wpa_printf(MSG_DEBUG, "ndisuio_notification_receive: "
1896                            "Too short message (len=%d)", (int) len);
1897                 return;
1898         }
1899
1900         hdr = (NDISUIO_DEVICE_NOTIFICATION *) buf;
1901         wpa_printf(MSG_DEBUG, "NDIS: Notification received: len=%d type=0x%x",
1902                    (int) len, hdr->dwNotificationType);
1903
1904         switch (hdr->dwNotificationType) {
1905 #ifdef NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL
1906         case NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL:
1907                 wpa_printf(MSG_DEBUG, "NDIS: ADAPTER_ARRIVAL");
1908                 wpa_driver_ndis_event_adapter_arrival(drv);
1909                 break;
1910 #endif
1911 #ifdef NDISUIO_NOTIFICATION_ADAPTER_REMOVAL
1912         case NDISUIO_NOTIFICATION_ADAPTER_REMOVAL:
1913                 wpa_printf(MSG_DEBUG, "NDIS: ADAPTER_REMOVAL");
1914                 wpa_driver_ndis_event_adapter_removal(drv);
1915                 break;
1916 #endif
1917         case NDISUIO_NOTIFICATION_MEDIA_CONNECT:
1918                 wpa_printf(MSG_DEBUG, "NDIS: MEDIA_CONNECT");
1919                 SetEvent(drv->connected_event);
1920                 wpa_driver_ndis_event_connect(drv);
1921                 break;
1922         case NDISUIO_NOTIFICATION_MEDIA_DISCONNECT:
1923                 ResetEvent(drv->connected_event);
1924                 wpa_printf(MSG_DEBUG, "NDIS: MEDIA_DISCONNECT");
1925                 wpa_driver_ndis_event_disconnect(drv);
1926                 break;
1927         case NDISUIO_NOTIFICATION_MEDIA_SPECIFIC_NOTIFICATION:
1928                 wpa_printf(MSG_DEBUG, "NDIS: MEDIA_SPECIFIC_NOTIFICATION");
1929 #if _WIN32_WCE == 420 || _WIN32_WCE == 0x420
1930                 wpa_driver_ndis_event_media_specific(
1931                         drv, hdr->pvStatusBuffer, hdr->uiStatusBufferSize);
1932 #else
1933                 wpa_driver_ndis_event_media_specific(
1934                         drv, ((const u8 *) hdr) + hdr->uiOffsetToStatusBuffer,
1935                         (size_t) hdr->uiStatusBufferSize);
1936 #endif
1937                 break;
1938         default:
1939                 wpa_printf(MSG_DEBUG, "NDIS: Unknown notification type 0x%x",
1940                            hdr->dwNotificationType);
1941                 break;
1942         }
1943 }
1944
1945
1946 static void ndisuio_notification_deinit(struct wpa_driver_ndis_data *drv)
1947 {
1948         NDISUIO_REQUEST_NOTIFICATION req;
1949
1950         memset(&req, 0, sizeof(req));
1951         req.hMsgQueue = drv->event_queue;
1952         req.dwNotificationTypes = 0;
1953
1954         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_REQUEST_NOTIFICATION,
1955                              &req, sizeof(req), NULL, 0, NULL, NULL)) {
1956                 wpa_printf(MSG_INFO, "ndisuio_notification_deinit: "
1957                            "IOCTL_NDISUIO_REQUEST_NOTIFICATION failed: %d",
1958                            (int) GetLastError());
1959         }
1960
1961         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_CANCEL_NOTIFICATION,
1962                              NULL, 0, NULL, 0, NULL, NULL)) {
1963                 wpa_printf(MSG_INFO, "ndisuio_notification_deinit: "
1964                            "IOCTL_NDISUIO_CANCEL_NOTIFICATION failed: %d",
1965                            (int) GetLastError());
1966         }
1967
1968         if (drv->event_queue) {
1969                 eloop_unregister_event(drv->event_queue,
1970                                        sizeof(drv->event_queue));
1971                 CloseHandle(drv->event_queue);
1972                 drv->event_queue = NULL;
1973         }
1974
1975         if (drv->connected_event) {
1976                 CloseHandle(drv->connected_event);
1977                 drv->connected_event = NULL;
1978         }
1979 }
1980
1981
1982 static int ndisuio_notification_init(struct wpa_driver_ndis_data *drv)
1983 {
1984         MSGQUEUEOPTIONS opt;
1985         NDISUIO_REQUEST_NOTIFICATION req;
1986
1987         drv->connected_event =
1988                 CreateEvent(NULL, TRUE, FALSE, TEXT("WpaSupplicantConnected"));
1989         if (drv->connected_event == NULL) {
1990                 wpa_printf(MSG_INFO, "ndisuio_notification_init: "
1991                            "CreateEvent failed: %d",
1992                            (int) GetLastError());
1993                 return -1;
1994         }
1995
1996         memset(&opt, 0, sizeof(opt));
1997         opt.dwSize = sizeof(opt);
1998         opt.dwMaxMessages = 5;
1999         opt.cbMaxMessage = NDISUIO_MSG_SIZE;
2000         opt.bReadAccess = TRUE;
2001
2002         drv->event_queue = CreateMsgQueue(NULL, &opt);
2003         if (drv->event_queue == NULL) {
2004                 wpa_printf(MSG_INFO, "ndisuio_notification_init: "
2005                            "CreateMsgQueue failed: %d",
2006                            (int) GetLastError());
2007                 ndisuio_notification_deinit(drv);
2008                 return -1;
2009         }
2010
2011         memset(&req, 0, sizeof(req));
2012         req.hMsgQueue = drv->event_queue;
2013         req.dwNotificationTypes =
2014 #ifdef NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL
2015                 NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL |
2016 #endif
2017 #ifdef NDISUIO_NOTIFICATION_ADAPTER_REMOVAL
2018                 NDISUIO_NOTIFICATION_ADAPTER_REMOVAL |
2019 #endif
2020                 NDISUIO_NOTIFICATION_MEDIA_CONNECT |
2021                 NDISUIO_NOTIFICATION_MEDIA_DISCONNECT |
2022                 NDISUIO_NOTIFICATION_MEDIA_SPECIFIC_NOTIFICATION;
2023
2024         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_REQUEST_NOTIFICATION,
2025                              &req, sizeof(req), NULL, 0, NULL, NULL)) {
2026                 wpa_printf(MSG_INFO, "ndisuio_notification_init: "
2027                            "IOCTL_NDISUIO_REQUEST_NOTIFICATION failed: %d",
2028                            (int) GetLastError());
2029                 ndisuio_notification_deinit(drv);
2030                 return -1;
2031         }
2032
2033         eloop_register_event(drv->event_queue, sizeof(drv->event_queue),
2034                              ndisuio_notification_receive, drv, NULL);
2035
2036         return 0;
2037 }
2038 #endif /* _WIN32_WCE */
2039
2040
2041 static int wpa_driver_ndis_get_names(struct wpa_driver_ndis_data *drv)
2042 {
2043 #ifdef CONFIG_USE_NDISUIO
2044         NDISUIO_QUERY_BINDING *b;
2045         size_t blen = sizeof(*b) + 1024;
2046         int i, error, found = 0;
2047         DWORD written;
2048         char name[256], desc[256], *dpos;
2049         WCHAR *pos;
2050         size_t j, len, dlen;
2051
2052         b = os_malloc(blen);
2053         if (b == NULL)
2054                 return -1;
2055
2056         for (i = 0; ; i++) {
2057                 os_memset(b, 0, blen);
2058                 b->BindingIndex = i;
2059                 if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_QUERY_BINDING,
2060                                      b, sizeof(NDISUIO_QUERY_BINDING), b, blen,
2061                                      &written, NULL)) {
2062                         error = (int) GetLastError();
2063                         if (error == ERROR_NO_MORE_ITEMS)
2064                                 break;
2065                         wpa_printf(MSG_DEBUG, "IOCTL_NDISUIO_QUERY_BINDING "
2066                                    "failed: %d", error);
2067                         break;
2068                 }
2069
2070                 pos = (WCHAR *) ((char *) b + b->DeviceNameOffset);
2071                 len = b->DeviceNameLength;
2072                 if (len >= sizeof(name))
2073                         len = sizeof(name) - 1;
2074                 for (j = 0; j < len; j++)
2075                         name[j] = (char) pos[j];
2076                 name[len] = '\0';
2077
2078                 pos = (WCHAR *) ((char *) b + b->DeviceDescrOffset);
2079                 len = b->DeviceDescrLength;
2080                 if (len >= sizeof(desc))
2081                         len = sizeof(desc) - 1;
2082                 for (j = 0; j < len; j++)
2083                         desc[j] = (char) pos[j];
2084                 desc[len] = '\0';
2085
2086                 wpa_printf(MSG_DEBUG, "NDIS: %d - %s - %s", i, name, desc);
2087
2088                 if (os_strstr(name, drv->ifname)) {
2089                         wpa_printf(MSG_DEBUG, "NDIS: Interface name match");
2090                         found = 1;
2091                         break;
2092                 }
2093
2094                 if (os_strncmp(desc, drv->ifname, os_strlen(drv->ifname)) == 0)
2095                 {
2096                         wpa_printf(MSG_DEBUG, "NDIS: Interface description "
2097                                    "match");
2098                         found = 1;
2099                         break;
2100                 }
2101         }
2102
2103         if (!found) {
2104                 wpa_printf(MSG_DEBUG, "NDIS: Could not find interface '%s'",
2105                            drv->ifname);
2106                 os_free(b);
2107                 return -1;
2108         }
2109
2110         os_strlcpy(drv->ifname,
2111                    os_strncmp(name, "\\DEVICE\\", 8) == 0 ? name + 8 : name,
2112                    sizeof(drv->ifname));
2113 #ifdef _WIN32_WCE
2114         drv->adapter_name = wpa_strdup_tchar(drv->ifname);
2115         if (drv->adapter_name == NULL) {
2116                 wpa_printf(MSG_ERROR, "NDIS: Failed to allocate memory for "
2117                            "adapter name");
2118                 os_free(b);
2119                 return -1;
2120         }
2121 #endif /* _WIN32_WCE */
2122
2123         dpos = os_strstr(desc, " - ");
2124         if (dpos)
2125                 dlen = dpos - desc;
2126         else
2127                 dlen = os_strlen(desc);
2128         drv->adapter_desc = os_malloc(dlen + 1);
2129         if (drv->adapter_desc) {
2130                 os_memcpy(drv->adapter_desc, desc, dlen);
2131                 drv->adapter_desc[dlen] = '\0';
2132         }
2133
2134         os_free(b);
2135
2136         if (drv->adapter_desc == NULL)
2137                 return -1;
2138
2139         wpa_printf(MSG_DEBUG, "NDIS: Adapter description prefix '%s'",
2140                    drv->adapter_desc);
2141
2142         return 0;
2143 #else /* CONFIG_USE_NDISUIO */
2144         PTSTR _names;
2145         char *names, *pos, *pos2;
2146         ULONG len;
2147         BOOLEAN res;
2148 #define MAX_ADAPTERS 32
2149         char *name[MAX_ADAPTERS];
2150         char *desc[MAX_ADAPTERS];
2151         int num_name, num_desc, i, found_name, found_desc;
2152         size_t dlen;
2153
2154         wpa_printf(MSG_DEBUG, "NDIS: Packet.dll version: %s",
2155                    PacketGetVersion());
2156
2157         len = 8192;
2158         _names = os_zalloc(len);
2159         if (_names == NULL)
2160                 return -1;
2161
2162         res = PacketGetAdapterNames(_names, &len);
2163         if (!res && len > 8192) {
2164                 os_free(_names);
2165                 _names = os_zalloc(len);
2166                 if (_names == NULL)
2167                         return -1;
2168                 res = PacketGetAdapterNames(_names, &len);
2169         }
2170
2171         if (!res) {
2172                 wpa_printf(MSG_ERROR, "NDIS: Failed to get adapter list "
2173                            "(PacketGetAdapterNames)");
2174                 os_free(_names);
2175                 return -1;
2176         }
2177
2178         names = (char *) _names;
2179         if (names[0] && names[1] == '\0' && names[2] && names[3] == '\0') {
2180                 wpa_printf(MSG_DEBUG, "NDIS: Looks like adapter names are in "
2181                            "UNICODE");
2182                 /* Convert to ASCII */
2183                 pos2 = pos = names;
2184                 while (pos2 < names + len) {
2185                         if (pos2[0] == '\0' && pos2[1] == '\0' &&
2186                             pos2[2] == '\0' && pos2[3] == '\0') {
2187                                 pos2 += 4;
2188                                 break;
2189                         }
2190                         *pos++ = pos2[0];
2191                         pos2 += 2;
2192                 }
2193                 os_memcpy(pos + 2, names, pos - names);
2194                 pos += 2;
2195         } else
2196                 pos = names;
2197
2198         num_name = 0;
2199         while (pos < names + len) {
2200                 name[num_name] = pos;
2201                 while (*pos && pos < names + len)
2202                         pos++;
2203                 if (pos + 1 >= names + len) {
2204                         os_free(names);
2205                         return -1;
2206                 }
2207                 pos++;
2208                 num_name++;
2209                 if (num_name >= MAX_ADAPTERS) {
2210                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapters");
2211                         os_free(names);
2212                         return -1;
2213                 }
2214                 if (*pos == '\0') {
2215                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter names found",
2216                                    num_name);
2217                         pos++;
2218                         break;
2219                 }
2220         }
2221
2222         num_desc = 0;
2223         while (pos < names + len) {
2224                 desc[num_desc] = pos;
2225                 while (*pos && pos < names + len)
2226                         pos++;
2227                 if (pos + 1 >= names + len) {
2228                         os_free(names);
2229                         return -1;
2230                 }
2231                 pos++;
2232                 num_desc++;
2233                 if (num_desc >= MAX_ADAPTERS) {
2234                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapter "
2235                                    "descriptions");
2236                         os_free(names);
2237                         return -1;
2238                 }
2239                 if (*pos == '\0') {
2240                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter descriptions "
2241                                    "found", num_name);
2242                         pos++;
2243                         break;
2244                 }
2245         }
2246
2247         /*
2248          * Windows 98 with Packet.dll 3.0 alpha3 does not include adapter
2249          * descriptions. Fill in dummy descriptors to work around this.
2250          */
2251         while (num_desc < num_name)
2252                 desc[num_desc++] = "dummy description";
2253
2254         if (num_name != num_desc) {
2255                 wpa_printf(MSG_DEBUG, "NDIS: mismatch in adapter name and "
2256                            "description counts (%d != %d)",
2257                            num_name, num_desc);
2258                 os_free(names);
2259                 return -1;
2260         }
2261
2262         found_name = found_desc = -1;
2263         for (i = 0; i < num_name; i++) {
2264                 wpa_printf(MSG_DEBUG, "NDIS: %d - %s - %s",
2265                            i, name[i], desc[i]);
2266                 if (found_name == -1 && os_strstr(name[i], drv->ifname))
2267                         found_name = i;
2268                 if (found_desc == -1 &&
2269                     os_strncmp(desc[i], drv->ifname, os_strlen(drv->ifname)) ==
2270                     0)
2271                         found_desc = i;
2272         }
2273
2274         if (found_name < 0 && found_desc >= 0) {
2275                 wpa_printf(MSG_DEBUG, "NDIS: Matched interface '%s' based on "
2276                            "description '%s'",
2277                            name[found_desc], desc[found_desc]);
2278                 found_name = found_desc;
2279                 os_strlcpy(drv->ifname,
2280                            os_strncmp(name[found_desc], "\\Device\\NPF_", 12)
2281                            == 0 ? name[found_desc] + 12 : name[found_desc],
2282                            sizeof(drv->ifname));
2283         }
2284
2285         if (found_name < 0) {
2286                 wpa_printf(MSG_DEBUG, "NDIS: Could not find interface '%s'",
2287                            drv->ifname);
2288                 os_free(names);
2289                 return -1;
2290         }
2291
2292         i = found_name;
2293         pos = os_strrchr(desc[i], '(');
2294         if (pos) {
2295                 dlen = pos - desc[i];
2296                 pos--;
2297                 if (pos > desc[i] && *pos == ' ')
2298                         dlen--;
2299         } else {
2300                 dlen = os_strlen(desc[i]);
2301         }
2302         drv->adapter_desc = os_malloc(dlen + 1);
2303         if (drv->adapter_desc) {
2304                 os_memcpy(drv->adapter_desc, desc[i], dlen);
2305                 drv->adapter_desc[dlen] = '\0';
2306         }
2307
2308         os_free(names);
2309
2310         if (drv->adapter_desc == NULL)
2311                 return -1;
2312
2313         wpa_printf(MSG_DEBUG, "NDIS: Adapter description prefix '%s'",
2314                    drv->adapter_desc);
2315
2316         return 0;
2317 #endif /* CONFIG_USE_NDISUIO */
2318 }
2319
2320
2321 #if defined(CONFIG_NATIVE_WINDOWS) || defined(__CYGWIN__)
2322 #ifndef _WIN32_WCE
2323 /*
2324  * These structures are undocumented for WinXP; only WinCE version is
2325  * documented. These would be included wzcsapi.h if it were available. Some
2326  * changes here have been needed to make the structures match with WinXP SP2.
2327  * It is unclear whether these work with any other version.
2328  */
2329
2330 typedef struct {
2331         LPWSTR wszGuid;
2332 } INTF_KEY_ENTRY, *PINTF_KEY_ENTRY;
2333
2334 typedef struct {
2335         DWORD dwNumIntfs;
2336         PINTF_KEY_ENTRY pIntfs;
2337 } INTFS_KEY_TABLE, *PINTFS_KEY_TABLE;
2338
2339 typedef struct {
2340         DWORD dwDataLen;
2341         LPBYTE pData;
2342 } RAW_DATA, *PRAW_DATA;
2343
2344 typedef struct {
2345         LPWSTR wszGuid;
2346         LPWSTR wszDescr;
2347         ULONG ulMediaState;
2348         ULONG ulMediaType;
2349         ULONG ulPhysicalMediaType;
2350         INT nInfraMode;
2351         INT nAuthMode;
2352         INT nWepStatus;
2353 #ifndef _WIN32_WCE
2354         u8 pad[2]; /* why is this needed? */
2355 #endif /* _WIN32_WCE */
2356         DWORD dwCtlFlags;
2357         DWORD dwCapabilities; /* something added for WinXP SP2(?) */
2358         RAW_DATA rdSSID;
2359         RAW_DATA rdBSSID;
2360         RAW_DATA rdBSSIDList;
2361         RAW_DATA rdStSSIDList;
2362         RAW_DATA rdCtrlData;
2363 #ifdef UNDER_CE
2364         BOOL bInitialized;
2365 #endif
2366         DWORD nWPAMCastCipher;
2367         /* add some extra buffer for later additions since this interface is
2368          * far from stable */
2369         u8 later_additions[100];
2370 } INTF_ENTRY, *PINTF_ENTRY;
2371
2372 #define INTF_ALL 0xffffffff
2373 #define INTF_ALL_FLAGS 0x0000ffff
2374 #define INTF_CTLFLAGS 0x00000010
2375 #define INTFCTL_ENABLED 0x8000
2376 #endif /* _WIN32_WCE */
2377
2378
2379 #ifdef _WIN32_WCE
2380 static int wpa_driver_ndis_rebind_adapter(struct wpa_driver_ndis_data *drv)
2381 {
2382         HANDLE ndis;
2383         TCHAR multi[100];
2384         int len;
2385
2386         len = _tcslen(drv->adapter_name);
2387         if (len > 80)
2388                 return -1;
2389
2390         ndis = CreateFile(DD_NDIS_DEVICE_NAME, GENERIC_READ | GENERIC_WRITE,
2391                           0, NULL, OPEN_EXISTING, 0, NULL);
2392         if (ndis == INVALID_HANDLE_VALUE) {
2393                 wpa_printf(MSG_DEBUG, "NDIS: Failed to open file to NDIS "
2394                            "device: %d", (int) GetLastError());
2395                 return -1;
2396         }
2397
2398         len++;
2399         memcpy(multi, drv->adapter_name, len * sizeof(TCHAR));
2400         memcpy(&multi[len], TEXT("NDISUIO\0"), 9 * sizeof(TCHAR));
2401         len += 9;
2402
2403         if (!DeviceIoControl(ndis, IOCTL_NDIS_REBIND_ADAPTER,
2404                              multi, len * sizeof(TCHAR), NULL, 0, NULL, NULL))
2405         {
2406                 wpa_printf(MSG_DEBUG, "NDIS: IOCTL_NDIS_REBIND_ADAPTER "
2407                            "failed: 0x%x", (int) GetLastError());
2408                 wpa_hexdump_ascii(MSG_DEBUG, "NDIS: rebind multi_sz",
2409                                   (u8 *) multi, len * sizeof(TCHAR));
2410                 CloseHandle(ndis);
2411                 return -1;
2412         }
2413
2414         CloseHandle(ndis);
2415
2416         wpa_printf(MSG_DEBUG, "NDIS: Requested NDIS rebind of NDISUIO "
2417                    "protocol");
2418
2419         return 0;
2420 }
2421 #endif /* _WIN32_WCE */
2422
2423
2424 static int wpa_driver_ndis_set_wzc(struct wpa_driver_ndis_data *drv,
2425                                    int enable)
2426 {
2427 #ifdef _WIN32_WCE
2428         HKEY hk, hk2;
2429         LONG ret;
2430         DWORD i, hnd, len;
2431         TCHAR keyname[256], devname[256];
2432
2433 #define WZC_DRIVER TEXT("Drivers\\BuiltIn\\ZeroConfig")
2434
2435         if (enable) {
2436                 HANDLE h;
2437                 h = ActivateDeviceEx(WZC_DRIVER, NULL, 0, NULL);
2438                 if (h == INVALID_HANDLE_VALUE || h == 0) {
2439                         wpa_printf(MSG_DEBUG, "NDIS: Failed to re-enable WZC "
2440                                    "- ActivateDeviceEx failed: %d",
2441                                    (int) GetLastError());
2442                         return -1;
2443                 }
2444
2445                 wpa_printf(MSG_DEBUG, "NDIS: WZC re-enabled");
2446                 return wpa_driver_ndis_rebind_adapter(drv);
2447         }
2448
2449         /*
2450          * Unfortunately, just disabling the WZC for an interface is not enough
2451          * to free NDISUIO for us, so need to disable and unload WZC completely
2452          * for now when using WinCE with NDISUIO. In addition, must request
2453          * NDISUIO protocol to be rebound to the adapter in order to free the
2454          * NDISUIO binding that WZC hold before us.
2455          */
2456
2457         /* Enumerate HKLM\Drivers\Active\* to find a handle to WZC. */
2458         ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, DEVLOAD_ACTIVE_KEY, 0, 0, &hk);
2459         if (ret != ERROR_SUCCESS) {
2460                 wpa_printf(MSG_DEBUG, "NDIS: RegOpenKeyEx(DEVLOAD_ACTIVE_KEY) "
2461                            "failed: %d %d", (int) ret, (int) GetLastError());
2462                 return -1;
2463         }
2464
2465         for (i = 0; ; i++) {
2466                 len = sizeof(keyname);
2467                 ret = RegEnumKeyEx(hk, i, keyname, &len, NULL, NULL, NULL,
2468                                    NULL);
2469                 if (ret != ERROR_SUCCESS) {
2470                         wpa_printf(MSG_DEBUG, "NDIS: Could not find active "
2471                                    "WZC - assuming it is not running.");
2472                         RegCloseKey(hk);
2473                         return -1;
2474                 }
2475
2476                 ret = RegOpenKeyEx(hk, keyname, 0, 0, &hk2);
2477                 if (ret != ERROR_SUCCESS) {
2478                         wpa_printf(MSG_DEBUG, "NDIS: RegOpenKeyEx(active dev) "
2479                                    "failed: %d %d",
2480                                    (int) ret, (int) GetLastError());
2481                         continue;
2482                 }
2483
2484                 len = sizeof(devname);
2485                 ret = RegQueryValueEx(hk2, DEVLOAD_DEVKEY_VALNAME, NULL, NULL,
2486                                       (LPBYTE) devname, &len);
2487                 if (ret != ERROR_SUCCESS) {
2488                         wpa_printf(MSG_DEBUG, "NDIS: RegQueryValueEx("
2489                                    "DEVKEY_VALNAME) failed: %d %d",
2490                                    (int) ret, (int) GetLastError());
2491                         RegCloseKey(hk2);
2492                         continue;
2493                 }
2494
2495                 if (_tcscmp(devname, WZC_DRIVER) == 0)
2496                         break;
2497
2498                 RegCloseKey(hk2);
2499         }
2500
2501         RegCloseKey(hk);
2502
2503         /* Found WZC - get handle to it. */
2504         len = sizeof(hnd);
2505         ret = RegQueryValueEx(hk2, DEVLOAD_HANDLE_VALNAME, NULL, NULL,
2506                               (PUCHAR) &hnd, &len);
2507         if (ret != ERROR_SUCCESS) {
2508                 wpa_printf(MSG_DEBUG, "NDIS: RegQueryValueEx(HANDLE_VALNAME) "
2509                            "failed: %d %d", (int) ret, (int) GetLastError());
2510                 RegCloseKey(hk2);
2511                 return -1;
2512         }
2513
2514         RegCloseKey(hk2);
2515
2516         /* Deactivate WZC */
2517         if (!DeactivateDevice((HANDLE) hnd)) {
2518                 wpa_printf(MSG_DEBUG, "NDIS: DeactivateDevice failed: %d",
2519                            (int) GetLastError());
2520                 return -1;
2521         }
2522
2523         wpa_printf(MSG_DEBUG, "NDIS: Disabled WZC temporarily");
2524         drv->wzc_disabled = 1;
2525         return wpa_driver_ndis_rebind_adapter(drv);
2526
2527 #else /* _WIN32_WCE */
2528
2529         HMODULE hm;
2530         DWORD (WINAPI *wzc_enum_interf)(LPWSTR pSrvAddr,
2531                                         PINTFS_KEY_TABLE pIntfs);
2532         DWORD (WINAPI *wzc_query_interf)(LPWSTR pSrvAddr, DWORD dwInFlags,
2533                                          PINTF_ENTRY pIntf,
2534                                          LPDWORD pdwOutFlags);
2535         DWORD (WINAPI *wzc_set_interf)(LPWSTR pSrvAddr, DWORD dwInFlags,
2536                                        PINTF_ENTRY pIntf, LPDWORD pdwOutFlags);
2537         int ret = -1, j;
2538         DWORD res;
2539         INTFS_KEY_TABLE guids;
2540         INTF_ENTRY intf;
2541         char guid[128];
2542         WCHAR *pos;
2543         DWORD flags, i;
2544
2545         hm = LoadLibrary(TEXT("wzcsapi.dll"));
2546         if (hm == NULL) {
2547                 wpa_printf(MSG_DEBUG, "NDIS: Failed to load wzcsapi.dll (%u) "
2548                            "- WZC probably not running",
2549                            (unsigned int) GetLastError());
2550                 return -1;
2551         }
2552
2553 #ifdef _WIN32_WCE
2554         wzc_enum_interf = (void *) GetProcAddressA(hm, "WZCEnumInterfaces");
2555         wzc_query_interf = (void *) GetProcAddressA(hm, "WZCQueryInterface");
2556         wzc_set_interf = (void *) GetProcAddressA(hm, "WZCSetInterface");
2557 #else /* _WIN32_WCE */
2558         wzc_enum_interf = (void *) GetProcAddress(hm, "WZCEnumInterfaces");
2559         wzc_query_interf = (void *) GetProcAddress(hm, "WZCQueryInterface");
2560         wzc_set_interf = (void *) GetProcAddress(hm, "WZCSetInterface");
2561 #endif /* _WIN32_WCE */
2562
2563         if (wzc_enum_interf == NULL || wzc_query_interf == NULL ||
2564             wzc_set_interf == NULL) {
2565                 wpa_printf(MSG_DEBUG, "NDIS: WZCEnumInterfaces, "
2566                            "WZCQueryInterface, or WZCSetInterface not found "
2567                            "in wzcsapi.dll");
2568                 goto fail;
2569         }
2570
2571         os_memset(&guids, 0, sizeof(guids));
2572         res = wzc_enum_interf(NULL, &guids);
2573         if (res != 0) {
2574                 wpa_printf(MSG_DEBUG, "NDIS: WZCEnumInterfaces failed: %d; "
2575                            "WZC service is apparently not running",
2576                            (int) res);
2577                 goto fail;
2578         }
2579
2580         wpa_printf(MSG_DEBUG, "NDIS: WZCEnumInterfaces: %d interfaces",
2581                    (int) guids.dwNumIntfs);
2582
2583         for (i = 0; i < guids.dwNumIntfs; i++) {
2584                 pos = guids.pIntfs[i].wszGuid;
2585                 for (j = 0; j < sizeof(guid); j++) {
2586                         guid[j] = (char) *pos;
2587                         if (*pos == 0)
2588                                 break;
2589                         pos++;
2590                 }
2591                 guid[sizeof(guid) - 1] = '\0';
2592                 wpa_printf(MSG_DEBUG, "NDIS: intfs %d GUID '%s'",
2593                            (int) i, guid);
2594                 if (os_strstr(drv->ifname, guid) == NULL)
2595                         continue;
2596
2597                 wpa_printf(MSG_DEBUG, "NDIS: Current interface found from "
2598                            "WZC");
2599                 break;
2600         }
2601
2602         if (i >= guids.dwNumIntfs) {
2603                 wpa_printf(MSG_DEBUG, "NDIS: Current interface not found from "
2604                            "WZC");
2605                 goto fail;
2606         }
2607
2608         os_memset(&intf, 0, sizeof(intf));
2609         intf.wszGuid = guids.pIntfs[i].wszGuid;
2610         /* Set flags to verify that the structure has not changed. */
2611         intf.dwCtlFlags = -1;
2612         flags = 0;
2613         res = wzc_query_interf(NULL, INTFCTL_ENABLED, &intf, &flags);
2614         if (res != 0) {
2615                 wpa_printf(MSG_DEBUG, "NDIS: Could not query flags for the "
2616                            "WZC interface: %d (0x%x)",
2617                            (int) res, (int) res);
2618                 wpa_printf(MSG_DEBUG, "NDIS: GetLastError: %u",
2619                            (unsigned int) GetLastError());
2620                 goto fail;
2621         }
2622
2623         wpa_printf(MSG_DEBUG, "NDIS: WZC interface flags 0x%x dwCtlFlags 0x%x",
2624                    (int) flags, (int) intf.dwCtlFlags);
2625
2626         if (intf.dwCtlFlags == -1) {
2627                 wpa_printf(MSG_DEBUG, "NDIS: Looks like wzcsapi has changed "
2628                            "again - could not disable WZC");
2629                 wpa_hexdump(MSG_MSGDUMP, "NDIS: intf",
2630                             (u8 *) &intf, sizeof(intf));
2631                 goto fail;
2632         }
2633
2634         if (enable) {
2635                 if (!(intf.dwCtlFlags & INTFCTL_ENABLED)) {
2636                         wpa_printf(MSG_DEBUG, "NDIS: Enabling WZC for this "
2637                                    "interface");
2638                         intf.dwCtlFlags |= INTFCTL_ENABLED;
2639                         res = wzc_set_interf(NULL, INTFCTL_ENABLED, &intf,
2640                                              &flags);
2641                         if (res != 0) {
2642                                 wpa_printf(MSG_DEBUG, "NDIS: Failed to enable "
2643                                            "WZC: %d (0x%x)",
2644                                            (int) res, (int) res);
2645                                 wpa_printf(MSG_DEBUG, "NDIS: GetLastError: %u",
2646                                            (unsigned int) GetLastError());
2647                                 goto fail;
2648                         }
2649                         wpa_printf(MSG_DEBUG, "NDIS: Re-enabled WZC for this "
2650                                    "interface");
2651                         drv->wzc_disabled = 0;
2652                 }
2653         } else {
2654                 if (intf.dwCtlFlags & INTFCTL_ENABLED) {
2655                         wpa_printf(MSG_DEBUG, "NDIS: Disabling WZC for this "
2656                                    "interface");
2657                         intf.dwCtlFlags &= ~INTFCTL_ENABLED;
2658                         res = wzc_set_interf(NULL, INTFCTL_ENABLED, &intf,
2659                                              &flags);
2660                         if (res != 0) {
2661                                 wpa_printf(MSG_DEBUG, "NDIS: Failed to "
2662                                            "disable WZC: %d (0x%x)",
2663                                            (int) res, (int) res);
2664                                 wpa_printf(MSG_DEBUG, "NDIS: GetLastError: %u",
2665                                            (unsigned int) GetLastError());
2666                                 goto fail;
2667                         }
2668                         wpa_printf(MSG_DEBUG, "NDIS: Disabled WZC temporarily "
2669                                    "for this interface");
2670                         drv->wzc_disabled = 1;
2671                 } else {
2672                         wpa_printf(MSG_DEBUG, "NDIS: WZC was not enabled for "
2673                                    "this interface");
2674                 }
2675         }
2676
2677         ret = 0;
2678
2679 fail:
2680         FreeLibrary(hm);
2681
2682         return ret;
2683 #endif /* _WIN32_WCE */
2684 }
2685
2686 #else /* CONFIG_NATIVE_WINDOWS || __CYGWIN__ */
2687
2688 static int wpa_driver_ndis_set_wzc(struct wpa_driver_ndis_data *drv,
2689                                    int enable)
2690 {
2691         return 0;
2692 }
2693
2694 #endif /* CONFIG_NATIVE_WINDOWS || __CYGWIN__ */
2695
2696
2697 #ifdef CONFIG_USE_NDISUIO
2698 /*
2699  * l2_packet_ndis.c is sharing the same handle to NDISUIO, so we must be able
2700  * to export this handle. This is somewhat ugly, but there is no better
2701  * mechanism available to pass data from driver interface to l2_packet wrapper.
2702  */
2703 static HANDLE driver_ndis_ndisuio_handle = INVALID_HANDLE_VALUE;
2704
2705 HANDLE driver_ndis_get_ndisuio_handle(void)
2706 {
2707         return driver_ndis_ndisuio_handle;
2708 }
2709 #endif /* CONFIG_USE_NDISUIO */
2710
2711
2712 static int wpa_driver_ndis_adapter_init(struct wpa_driver_ndis_data *drv)
2713 {
2714 #ifdef CONFIG_USE_NDISUIO
2715 #ifndef _WIN32_WCE
2716 #define NDISUIO_DEVICE_NAME TEXT("\\\\.\\\\Ndisuio")
2717         DWORD written;
2718 #endif /* _WIN32_WCE */
2719         drv->ndisuio = CreateFile(NDISUIO_DEVICE_NAME,
2720                                   GENERIC_READ | GENERIC_WRITE, 0, NULL,
2721                                   OPEN_EXISTING,
2722                                   FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED,
2723                                   INVALID_HANDLE_VALUE);
2724         if (drv->ndisuio == INVALID_HANDLE_VALUE) {
2725                 wpa_printf(MSG_ERROR, "NDIS: Failed to open connection to "
2726                            "NDISUIO: %d", (int) GetLastError());
2727                 return -1;
2728         }
2729         driver_ndis_ndisuio_handle = drv->ndisuio;
2730
2731 #ifndef _WIN32_WCE
2732         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_BIND_WAIT, NULL, 0,
2733                              NULL, 0, &written, NULL)) {
2734                 wpa_printf(MSG_ERROR, "NDIS: IOCTL_NDISUIO_BIND_WAIT failed: "
2735                            "%d", (int) GetLastError());
2736                 CloseHandle(drv->ndisuio);
2737                 drv->ndisuio = INVALID_HANDLE_VALUE;
2738                 return -1;
2739         }
2740 #endif /* _WIN32_WCE */
2741
2742         return 0;
2743 #else /* CONFIG_USE_NDISUIO */
2744         return 0;
2745 #endif /* CONFIG_USE_NDISUIO */
2746 }
2747
2748
2749 static int wpa_driver_ndis_adapter_open(struct wpa_driver_ndis_data *drv)
2750 {
2751 #ifdef CONFIG_USE_NDISUIO
2752         DWORD written;
2753 #define MAX_NDIS_DEVICE_NAME_LEN 256
2754         WCHAR ifname[MAX_NDIS_DEVICE_NAME_LEN];
2755         size_t len, i, pos;
2756         const char *prefix = "\\DEVICE\\";
2757
2758 #ifdef _WIN32_WCE
2759         pos = 0;
2760 #else /* _WIN32_WCE */
2761         pos = 8;
2762 #endif /* _WIN32_WCE */
2763         len = pos + os_strlen(drv->ifname);
2764         if (len >= MAX_NDIS_DEVICE_NAME_LEN)
2765                 return -1;
2766         for (i = 0; i < pos; i++)
2767                 ifname[i] = (WCHAR) prefix[i];
2768         for (i = pos; i < len; i++)
2769                 ifname[i] = (WCHAR) drv->ifname[i - pos];
2770         ifname[i] = L'\0';
2771
2772         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_OPEN_DEVICE,
2773                              ifname, len * sizeof(WCHAR), NULL, 0, &written,
2774                              NULL)) {
2775                 wpa_printf(MSG_ERROR, "NDIS: IOCTL_NDISUIO_OPEN_DEVICE "
2776                            "failed: %d", (int) GetLastError());
2777                 wpa_hexdump_ascii(MSG_DEBUG, "NDIS: ifname",
2778                                   (const u8 *) ifname, len * sizeof(WCHAR));
2779                 CloseHandle(drv->ndisuio);
2780                 drv->ndisuio = INVALID_HANDLE_VALUE;
2781                 return -1;
2782         }
2783
2784         wpa_printf(MSG_DEBUG, "NDIS: Opened NDISUIO device successfully");
2785
2786         return 0;
2787 #else /* CONFIG_USE_NDISUIO */
2788         char ifname[128];
2789         os_snprintf(ifname, sizeof(ifname), "\\Device\\NPF_%s", drv->ifname);
2790         drv->adapter = PacketOpenAdapter(ifname);
2791         if (drv->adapter == NULL) {
2792                 wpa_printf(MSG_DEBUG, "NDIS: PacketOpenAdapter failed for "
2793                            "'%s'", ifname);
2794                 return -1;
2795         }
2796         return 0;
2797 #endif /* CONFIG_USE_NDISUIO */
2798 }
2799
2800
2801 static void wpa_driver_ndis_adapter_close(struct wpa_driver_ndis_data *drv)
2802 {
2803 #ifdef CONFIG_USE_NDISUIO
2804         driver_ndis_ndisuio_handle = INVALID_HANDLE_VALUE;
2805         if (drv->ndisuio != INVALID_HANDLE_VALUE)
2806                 CloseHandle(drv->ndisuio);
2807 #else /* CONFIG_USE_NDISUIO */
2808         if (drv->adapter)
2809                 PacketCloseAdapter(drv->adapter);
2810 #endif /* CONFIG_USE_NDISUIO */
2811 }
2812
2813
2814 static int ndis_add_multicast(struct wpa_driver_ndis_data *drv)
2815 {
2816         if (ndis_set_oid(drv, OID_802_3_MULTICAST_LIST,
2817                          (const char *) pae_group_addr, ETH_ALEN) < 0) {
2818                 wpa_printf(MSG_DEBUG, "NDIS: Failed to add PAE group address "
2819                            "to the multicast list");
2820                 return -1;
2821         }
2822
2823         return 0;
2824 }
2825
2826
2827 static void * wpa_driver_ndis_init(void *ctx, const char *ifname)
2828 {
2829         struct wpa_driver_ndis_data *drv;
2830         u32 mode;
2831
2832         drv = os_zalloc(sizeof(*drv));
2833         if (drv == NULL)
2834                 return NULL;
2835         drv->ctx = ctx;
2836         /*
2837          * Compatibility code to strip possible prefix from the GUID. Previous
2838          * versions include \Device\NPF_ prefix for all names, but the internal
2839          * interface name is now only the GUI. Both Packet32 and NDISUIO
2840          * prefixes are supported.
2841          */
2842         if (os_strncmp(ifname, "\\Device\\NPF_", 12) == 0)
2843                 ifname += 12;
2844         else if (os_strncmp(ifname, "\\DEVICE\\", 8) == 0)
2845                 ifname += 8;
2846         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
2847
2848         if (wpa_driver_ndis_adapter_init(drv) < 0) {
2849                 os_free(drv);
2850                 return NULL;
2851         }
2852
2853         if (wpa_driver_ndis_get_names(drv) < 0) {
2854                 wpa_driver_ndis_adapter_close(drv);
2855                 os_free(drv);
2856                 return NULL;
2857         }
2858
2859         wpa_driver_ndis_set_wzc(drv, 0);
2860
2861         if (wpa_driver_ndis_adapter_open(drv) < 0) {
2862                 wpa_driver_ndis_adapter_close(drv);
2863                 os_free(drv);
2864                 return NULL;
2865         }
2866
2867         if (ndis_get_oid(drv, OID_802_3_CURRENT_ADDRESS,
2868                          (char *) drv->own_addr, ETH_ALEN) < 0) {
2869                 wpa_printf(MSG_DEBUG, "NDIS: Get OID_802_3_CURRENT_ADDRESS "
2870                            "failed");
2871                 wpa_driver_ndis_adapter_close(drv);
2872                 os_free(drv);
2873                 return NULL;
2874         }
2875         wpa_driver_ndis_get_capability(drv);
2876
2877         /* Make sure that the driver does not have any obsolete PMKID entries.
2878          */
2879         wpa_driver_ndis_flush_pmkid(drv);
2880
2881         /*
2882          * Disconnect to make sure that driver re-associates if it was
2883          * connected.
2884          */
2885         wpa_driver_ndis_disconnect(drv);
2886
2887         eloop_register_timeout(1, 0, wpa_driver_ndis_poll_timeout, drv, NULL);
2888
2889 #ifdef CONFIG_NDIS_EVENTS_INTEGRATED
2890         drv->events = ndis_events_init(&drv->events_pipe, &drv->event_avail,
2891                                        drv->ifname, drv->adapter_desc);
2892         if (drv->events == NULL) {
2893                 wpa_driver_ndis_deinit(drv);
2894                 return NULL;
2895         }
2896         eloop_register_event(drv->event_avail, sizeof(drv->event_avail),
2897                              wpa_driver_ndis_event_pipe_cb, drv, NULL);
2898 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
2899
2900 #ifdef _WIN32_WCE
2901         if (ndisuio_notification_init(drv) < 0) {
2902                 wpa_driver_ndis_deinit(drv);
2903                 return NULL;
2904         }
2905 #endif /* _WIN32_WCE */
2906
2907         /* Set mode here in case card was configured for ad-hoc mode
2908          * previously. */
2909         mode = Ndis802_11Infrastructure;
2910         if (ndis_set_oid(drv, OID_802_11_INFRASTRUCTURE_MODE,
2911                          (char *) &mode, sizeof(mode)) < 0) {
2912                 char buf[8];
2913                 int res;
2914                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
2915                            "OID_802_11_INFRASTRUCTURE_MODE (%d)",
2916                            (int) mode);
2917                 /* Try to continue anyway */
2918
2919                 res = ndis_get_oid(drv, OID_DOT11_CURRENT_OPERATION_MODE, buf,
2920                                    sizeof(buf));
2921                 if (res > 0) {
2922                         wpa_printf(MSG_INFO, "NDIS: The driver seems to use "
2923                                    "Native 802.11 OIDs. These are not yet "
2924                                    "fully supported.");
2925                         drv->native80211 = 1;
2926                 } else if (!drv->has_capability || drv->capa.enc == 0) {
2927                         /*
2928                          * Note: This will also happen with NDIS 6 drivers with
2929                          * Vista.
2930                          */
2931                         wpa_printf(MSG_DEBUG, "NDIS: Driver did not provide "
2932                                    "any wireless capabilities - assume it is "
2933                                    "a wired interface");
2934                         drv->wired = 1;
2935                         drv->capa.flags |= WPA_DRIVER_FLAGS_WIRED;
2936                         drv->has_capability = 1;
2937                         ndis_add_multicast(drv);
2938                 }
2939         }
2940
2941         return drv;
2942 }
2943
2944
2945 static void wpa_driver_ndis_deinit(void *priv)
2946 {
2947         struct wpa_driver_ndis_data *drv = priv;
2948
2949 #ifdef CONFIG_NDIS_EVENTS_INTEGRATED
2950         if (drv->events) {
2951                 eloop_unregister_event(drv->event_avail,
2952                                        sizeof(drv->event_avail));
2953                 ndis_events_deinit(drv->events);
2954         }
2955 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
2956
2957 #ifdef _WIN32_WCE
2958         ndisuio_notification_deinit(drv);
2959 #endif /* _WIN32_WCE */
2960
2961         eloop_cancel_timeout(wpa_driver_ndis_scan_timeout, drv, drv->ctx);
2962         eloop_cancel_timeout(wpa_driver_ndis_poll_timeout, drv, NULL);
2963         wpa_driver_ndis_flush_pmkid(drv);
2964         wpa_driver_ndis_disconnect(drv);
2965         if (wpa_driver_ndis_radio_off(drv) < 0) {
2966                 wpa_printf(MSG_DEBUG, "NDIS: failed to disassociate and turn "
2967                            "radio off");
2968         }
2969
2970         wpa_driver_ndis_adapter_close(drv);
2971
2972         if (drv->wzc_disabled)
2973                 wpa_driver_ndis_set_wzc(drv, 1);
2974
2975 #ifdef _WIN32_WCE
2976         os_free(drv->adapter_name);
2977 #endif /* _WIN32_WCE */
2978         os_free(drv->adapter_desc);
2979         os_free(drv);
2980 }
2981
2982
2983 static struct wpa_interface_info *
2984 wpa_driver_ndis_get_interfaces(void *global_priv)
2985 {
2986         struct wpa_interface_info *iface = NULL, *niface;
2987
2988 #ifdef CONFIG_USE_NDISUIO
2989         NDISUIO_QUERY_BINDING *b;
2990         size_t blen = sizeof(*b) + 1024;
2991         int i, error;
2992         DWORD written;
2993         char name[256], desc[256];
2994         WCHAR *pos;
2995         size_t j, len;
2996         HANDLE ndisuio;
2997
2998         ndisuio = CreateFile(NDISUIO_DEVICE_NAME,
2999                              GENERIC_READ | GENERIC_WRITE, 0, NULL,
3000                              OPEN_EXISTING,
3001                              FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED,
3002                              INVALID_HANDLE_VALUE);
3003         if (ndisuio == INVALID_HANDLE_VALUE) {
3004                 wpa_printf(MSG_ERROR, "NDIS: Failed to open connection to "
3005                            "NDISUIO: %d", (int) GetLastError());
3006                 return NULL;
3007         }
3008
3009 #ifndef _WIN32_WCE
3010         if (!DeviceIoControl(ndisuio, IOCTL_NDISUIO_BIND_WAIT, NULL, 0,
3011                              NULL, 0, &written, NULL)) {
3012                 wpa_printf(MSG_ERROR, "NDIS: IOCTL_NDISUIO_BIND_WAIT failed: "
3013                            "%d", (int) GetLastError());
3014                 CloseHandle(ndisuio);
3015                 return NULL;
3016         }
3017 #endif /* _WIN32_WCE */
3018
3019         b = os_malloc(blen);
3020         if (b == NULL) {
3021                 CloseHandle(ndisuio);
3022                 return NULL;
3023         }
3024
3025         for (i = 0; ; i++) {
3026                 os_memset(b, 0, blen);
3027                 b->BindingIndex = i;
3028                 if (!DeviceIoControl(ndisuio, IOCTL_NDISUIO_QUERY_BINDING,
3029                                      b, sizeof(NDISUIO_QUERY_BINDING), b, blen,
3030                                      &written, NULL)) {
3031                         error = (int) GetLastError();
3032                         if (error == ERROR_NO_MORE_ITEMS)
3033                                 break;
3034                         wpa_printf(MSG_DEBUG, "IOCTL_NDISUIO_QUERY_BINDING "
3035                                    "failed: %d", error);
3036                         break;
3037                 }
3038
3039                 pos = (WCHAR *) ((char *) b + b->DeviceNameOffset);
3040                 len = b->DeviceNameLength;
3041                 if (len >= sizeof(name))
3042                         len = sizeof(name) - 1;
3043                 for (j = 0; j < len; j++)
3044                         name[j] = (char) pos[j];
3045                 name[len] = '\0';
3046
3047                 pos = (WCHAR *) ((char *) b + b->DeviceDescrOffset);
3048                 len = b->DeviceDescrLength;
3049                 if (len >= sizeof(desc))
3050                         len = sizeof(desc) - 1;
3051                 for (j = 0; j < len; j++)
3052                         desc[j] = (char) pos[j];
3053                 desc[len] = '\0';
3054
3055                 wpa_printf(MSG_DEBUG, "NDIS: %d - %s - %s", i, name, desc);
3056
3057                 niface = os_zalloc(sizeof(*niface));
3058                 if (niface == NULL)
3059                         break;
3060                 niface->drv_name = "ndis";
3061                 if (os_strncmp(name, "\\DEVICE\\", 8) == 0)
3062                         niface->ifname = os_strdup(name + 8);
3063                 else
3064                         niface->ifname = os_strdup(name);
3065                 if (niface->ifname == NULL) {
3066                         os_free(niface);
3067                         break;
3068                 }
3069                 niface->desc = os_strdup(desc);
3070                 niface->next = iface;
3071                 iface = niface;
3072         }
3073
3074         os_free(b);
3075         CloseHandle(ndisuio);
3076 #else /* CONFIG_USE_NDISUIO */
3077         PTSTR _names;
3078         char *names, *pos, *pos2;
3079         ULONG len;
3080         BOOLEAN res;
3081         char *name[MAX_ADAPTERS];
3082         char *desc[MAX_ADAPTERS];
3083         int num_name, num_desc, i;
3084
3085         wpa_printf(MSG_DEBUG, "NDIS: Packet.dll version: %s",
3086                    PacketGetVersion());
3087
3088         len = 8192;
3089         _names = os_zalloc(len);
3090         if (_names == NULL)
3091                 return NULL;
3092
3093         res = PacketGetAdapterNames(_names, &len);
3094         if (!res && len > 8192) {
3095                 os_free(_names);
3096                 _names = os_zalloc(len);
3097                 if (_names == NULL)
3098                         return NULL;
3099                 res = PacketGetAdapterNames(_names, &len);
3100         }
3101
3102         if (!res) {
3103                 wpa_printf(MSG_ERROR, "NDIS: Failed to get adapter list "
3104                            "(PacketGetAdapterNames)");
3105                 os_free(_names);
3106                 return NULL;
3107         }
3108
3109         names = (char *) _names;
3110         if (names[0] && names[1] == '\0' && names[2] && names[3] == '\0') {
3111                 wpa_printf(MSG_DEBUG, "NDIS: Looks like adapter names are in "
3112                            "UNICODE");
3113                 /* Convert to ASCII */
3114                 pos2 = pos = names;
3115                 while (pos2 < names + len) {
3116                         if (pos2[0] == '\0' && pos2[1] == '\0' &&
3117                             pos2[2] == '\0' && pos2[3] == '\0') {
3118                                 pos2 += 4;
3119                                 break;
3120                         }
3121                         *pos++ = pos2[0];
3122                         pos2 += 2;
3123                 }
3124                 os_memcpy(pos + 2, names, pos - names);
3125                 pos += 2;
3126         } else
3127                 pos = names;
3128
3129         num_name = 0;
3130         while (pos < names + len) {
3131                 name[num_name] = pos;
3132                 while (*pos && pos < names + len)
3133                         pos++;
3134                 if (pos + 1 >= names + len) {
3135                         os_free(names);
3136                         return NULL;
3137                 }
3138                 pos++;
3139                 num_name++;
3140                 if (num_name >= MAX_ADAPTERS) {
3141                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapters");
3142                         os_free(names);
3143                         return NULL;
3144                 }
3145                 if (*pos == '\0') {
3146                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter names found",
3147                                    num_name);
3148                         pos++;
3149                         break;
3150                 }
3151         }
3152
3153         num_desc = 0;
3154         while (pos < names + len) {
3155                 desc[num_desc] = pos;
3156                 while (*pos && pos < names + len)
3157                         pos++;
3158                 if (pos + 1 >= names + len) {
3159                         os_free(names);
3160                         return NULL;
3161                 }
3162                 pos++;
3163                 num_desc++;
3164                 if (num_desc >= MAX_ADAPTERS) {
3165                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapter "
3166                                    "descriptions");
3167                         os_free(names);
3168                         return NULL;
3169                 }
3170                 if (*pos == '\0') {
3171                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter descriptions "
3172                                    "found", num_name);
3173                         pos++;
3174                         break;
3175                 }
3176         }
3177
3178         /*
3179          * Windows 98 with Packet.dll 3.0 alpha3 does not include adapter
3180          * descriptions. Fill in dummy descriptors to work around this.
3181          */
3182         while (num_desc < num_name)
3183                 desc[num_desc++] = "dummy description";
3184
3185         if (num_name != num_desc) {
3186                 wpa_printf(MSG_DEBUG, "NDIS: mismatch in adapter name and "
3187                            "description counts (%d != %d)",
3188                            num_name, num_desc);
3189                 os_free(names);
3190                 return NULL;
3191         }
3192
3193         for (i = 0; i < num_name; i++) {
3194                 niface = os_zalloc(sizeof(*niface));
3195                 if (niface == NULL)
3196                         break;
3197                 niface->drv_name = "ndis";
3198                 if (os_strncmp(name[i], "\\Device\\NPF_", 12) == 0)
3199                         niface->ifname = os_strdup(name[i] + 12);
3200                 else
3201                         niface->ifname = os_strdup(name[i]);
3202                 if (niface->ifname == NULL) {
3203                         os_free(niface);
3204                         break;
3205                 }
3206                 niface->desc = os_strdup(desc[i]);
3207                 niface->next = iface;
3208                 iface = niface;
3209         }
3210
3211 #endif /* CONFIG_USE_NDISUIO */
3212
3213         return iface;
3214 }
3215
3216
3217 const struct wpa_driver_ops wpa_driver_ndis_ops = {
3218         "ndis",
3219         "Windows NDIS driver",
3220         wpa_driver_ndis_get_bssid,
3221         wpa_driver_ndis_get_ssid,
3222         wpa_driver_ndis_set_key,
3223         wpa_driver_ndis_init,
3224         wpa_driver_ndis_deinit,
3225         NULL /* set_param */,
3226         NULL /* set_countermeasures */,
3227         wpa_driver_ndis_deauthenticate,
3228         wpa_driver_ndis_disassociate,
3229         wpa_driver_ndis_associate,
3230         wpa_driver_ndis_add_pmkid,
3231         wpa_driver_ndis_remove_pmkid,
3232         wpa_driver_ndis_flush_pmkid,
3233         wpa_driver_ndis_get_capa,
3234         wpa_driver_ndis_poll,
3235         wpa_driver_ndis_get_ifname,
3236         wpa_driver_ndis_get_mac_addr,
3237         NULL /* send_eapol */,
3238         NULL /* set_operstate */,
3239         NULL /* mlme_setprotection */,
3240         NULL /* get_hw_feature_data */,
3241         NULL /* set_channel */,
3242         NULL /* set_ssid */,
3243         NULL /* set_bssid */,
3244         NULL /* send_mlme */,
3245         NULL /* mlme_add_sta */,
3246         NULL /* mlme_remove_sta */,
3247         NULL /* update_ft_ies */,
3248         NULL /* send_ft_action */,
3249         wpa_driver_ndis_get_scan_results,
3250         NULL /* set_country */,
3251         NULL /* global_init */,
3252         NULL /* global_deinit */,
3253         NULL /* init2 */,
3254         wpa_driver_ndis_get_interfaces,
3255         wpa_driver_ndis_scan,
3256         NULL /* authenticate */,
3257         NULL /* set_beacon */,
3258         NULL /* hapd_init */,
3259         NULL /* hapd_deinit */,
3260         NULL /* set_ieee8021x */,
3261         NULL /* set_privacy */,
3262         NULL /* get_seqnum */,
3263         NULL /* flush */,
3264         NULL /* set_generic_elem */,
3265         NULL /* read_sta_data */,
3266         NULL /* hapd_send_eapol */,
3267         NULL /* sta_deauth */,
3268         NULL /* sta_disassoc */,
3269         NULL /* sta_remove */,
3270         NULL /* hapd_get_ssid */,
3271         NULL /* hapd_set_ssid */,
3272         NULL /* hapd_set_countermeasures */,
3273         NULL /* sta_add */,
3274         NULL /* get_inact_sec */,
3275         NULL /* sta_clear_stats */,
3276         NULL /* set_freq */,
3277         NULL /* set_rts */,
3278         NULL /* set_frag */,
3279         NULL /* sta_set_flags */,
3280         NULL /* set_rate_sets */,
3281         NULL /* set_cts_protect */,
3282         NULL /* set_preamble */,
3283         NULL /* set_short_slot_time */,
3284         NULL /* set_tx_queue_params */,
3285         NULL /* valid_bss_mask */,
3286         NULL /* if_add */,
3287         NULL /* if_remove */,
3288         NULL /* set_sta_vlan */,
3289         NULL /* commit */,
3290         NULL /* send_ether */,
3291         NULL /* set_radius_acl_auth */,
3292         NULL /* set_radius_acl_expire */,
3293         NULL /* set_ht_params */,
3294         NULL /* set_ap_wps_ie */,
3295         NULL /* set_supp_port */,
3296         NULL /* set_wds_sta */,
3297         NULL /* send_action */,
3298         NULL /* remain_on_channel */,
3299         NULL /* cancel_remain_on_channel */,
3300         NULL /* probe_req_report */,
3301         NULL /* disable_11b_rates */,
3302         NULL /* deinit_ap */,
3303         NULL /* suspend */,
3304         NULL /* resume */,
3305         NULL /* signal_monitor */,
3306         NULL /* send_frame */,
3307         NULL /* shared_freq */,
3308         NULL /* get_noa */,
3309         NULL /* set_noa */,
3310         NULL /* set_p2p_powersave */,
3311         NULL /* ampdu */,
3312         NULL /* set_intra_bss */
3313 };