Remove deprecated driver_ops handlers
[libeap.git] / 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 "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(struct wpa_driver_ndis_data *drv,
750                                             const u8 *ssid, size_t ssid_len)
751 {
752         DOT11_SCAN_REQUEST_V2 req;
753         int res;
754
755         os_memset(&req, 0, sizeof(req));
756         req.dot11BSSType = dot11_BSS_type_any;
757         os_memset(req.dot11BSSID, 0xff, ETH_ALEN);
758         req.dot11ScanType = dot11_scan_type_auto;
759         res = ndis_set_oid(drv, OID_DOT11_SCAN_REQUEST, (char *) &req,
760                            sizeof(req));
761         eloop_cancel_timeout(wpa_driver_ndis_scan_timeout, drv, drv->ctx);
762         eloop_register_timeout(7, 0, wpa_driver_ndis_scan_timeout, drv,
763                                drv->ctx);
764         return res;
765 }
766
767
768 static int wpa_driver_ndis_scan(void *priv, const u8 *ssid, size_t ssid_len)
769 {
770         struct wpa_driver_ndis_data *drv = priv;
771         int res;
772
773         if (drv->native80211)
774                 return wpa_driver_ndis_scan_native80211(drv, ssid, ssid_len);
775
776         if (!drv->radio_enabled) {
777                 wpa_printf(MSG_DEBUG, "NDIS: turning radio on before the first"
778                            " scan");
779                 if (wpa_driver_ndis_disconnect(drv) < 0) {
780                         wpa_printf(MSG_DEBUG, "NDIS: failed to enable radio");
781                 }
782                 drv->radio_enabled = 1;
783         }
784
785         res = ndis_set_oid(drv, OID_802_11_BSSID_LIST_SCAN, "    ", 4);
786         eloop_cancel_timeout(wpa_driver_ndis_scan_timeout, drv, drv->ctx);
787         eloop_register_timeout(7, 0, wpa_driver_ndis_scan_timeout, drv,
788                                drv->ctx);
789         return res;
790 }
791
792
793 static struct wpa_scan_res * wpa_driver_ndis_add_scan_ssid(
794         struct wpa_scan_res *r, NDIS_802_11_SSID *ssid)
795 {
796         struct wpa_scan_res *nr;
797         u8 *pos;
798
799         if (wpa_scan_get_ie(r, WLAN_EID_SSID))
800                 return r; /* SSID IE already present */
801
802         if (ssid->SsidLength == 0 || ssid->SsidLength > 32)
803                 return r; /* No valid SSID inside scan data */
804
805         nr = os_realloc(r, sizeof(*r) + r->ie_len + 2 + ssid->SsidLength);
806         if (nr == NULL)
807                 return r;
808
809         pos = ((u8 *) (nr + 1)) + nr->ie_len;
810         *pos++ = WLAN_EID_SSID;
811         *pos++ = ssid->SsidLength;
812         os_memcpy(pos, ssid->Ssid, ssid->SsidLength);
813         nr->ie_len += 2 + ssid->SsidLength;
814
815         return nr;
816 }
817
818
819 static struct wpa_scan_results * wpa_driver_ndis_get_scan_results(void *priv)
820 {
821         struct wpa_driver_ndis_data *drv = priv;
822         NDIS_802_11_BSSID_LIST_EX *b;
823         size_t blen, count, i;
824         int len;
825         char *pos;
826         struct wpa_scan_results *results;
827         struct wpa_scan_res *r;
828
829         blen = 65535;
830         b = os_zalloc(blen);
831         if (b == NULL)
832                 return NULL;
833         len = ndis_get_oid(drv, OID_802_11_BSSID_LIST, (char *) b, blen);
834         if (len < 0) {
835                 wpa_printf(MSG_DEBUG, "NDIS: failed to get scan results");
836                 os_free(b);
837                 return NULL;
838         }
839         count = b->NumberOfItems;
840
841         results = os_zalloc(sizeof(*results));
842         if (results == NULL) {
843                 os_free(b);
844                 return NULL;
845         }
846         results->res = os_zalloc(count * sizeof(struct wpa_scan_res *));
847         if (results->res == NULL) {
848                 os_free(results);
849                 os_free(b);
850                 return NULL;
851         }
852
853         pos = (char *) &b->Bssid[0];
854         for (i = 0; i < count; i++) {
855                 NDIS_WLAN_BSSID_EX *bss = (NDIS_WLAN_BSSID_EX *) pos;
856                 NDIS_802_11_FIXED_IEs *fixed;
857
858                 if (bss->IELength < sizeof(NDIS_802_11_FIXED_IEs)) {
859                         wpa_printf(MSG_DEBUG, "NDIS: too small IELength=%d",
860                                    (int) bss->IELength);
861                         break;
862                 }
863                 if (((char *) bss->IEs) + bss->IELength  > (char *) b + blen) {
864                         /*
865                          * Some NDIS drivers have been reported to include an
866                          * entry with an invalid IELength in scan results and
867                          * this has crashed wpa_supplicant, so validate the
868                          * returned value before using it.
869                          */
870                         wpa_printf(MSG_DEBUG, "NDIS: skipped invalid scan "
871                                    "result IE (BSSID=" MACSTR ") IELength=%d",
872                                    MAC2STR(bss->MacAddress),
873                                    (int) bss->IELength);
874                         break;
875                 }
876
877                 r = os_zalloc(sizeof(*r) + bss->IELength -
878                               sizeof(NDIS_802_11_FIXED_IEs));
879                 if (r == NULL)
880                         break;
881
882                 os_memcpy(r->bssid, bss->MacAddress, ETH_ALEN);
883                 r->level = (int) bss->Rssi;
884                 r->freq = bss->Configuration.DSConfig / 1000;
885                 fixed = (NDIS_802_11_FIXED_IEs *) bss->IEs;
886                 r->beacon_int = WPA_GET_LE16((u8 *) &fixed->BeaconInterval);
887                 r->caps = WPA_GET_LE16((u8 *) &fixed->Capabilities);
888                 r->tsf = WPA_GET_LE64(fixed->Timestamp);
889                 os_memcpy(r + 1, bss->IEs + sizeof(NDIS_802_11_FIXED_IEs),
890                           bss->IELength - sizeof(NDIS_802_11_FIXED_IEs));
891                 r->ie_len = bss->IELength - sizeof(NDIS_802_11_FIXED_IEs);
892                 r = wpa_driver_ndis_add_scan_ssid(r, &bss->Ssid);
893
894                 results->res[results->num++] = r;
895
896                 pos += bss->Length;
897                 if (pos > (char *) b + blen)
898                         break;
899         }
900
901         os_free(b);
902
903         return results;
904 }
905
906
907 static int wpa_driver_ndis_remove_key(struct wpa_driver_ndis_data *drv,
908                                       int key_idx, const u8 *addr,
909                                       const u8 *bssid, int pairwise)
910 {
911         NDIS_802_11_REMOVE_KEY rkey;
912         NDIS_802_11_KEY_INDEX index;
913         int res, res2;
914
915         os_memset(&rkey, 0, sizeof(rkey));
916
917         rkey.Length = sizeof(rkey);
918         rkey.KeyIndex = key_idx;
919         if (pairwise)
920                 rkey.KeyIndex |= 1 << 30;
921         os_memcpy(rkey.BSSID, bssid, ETH_ALEN);
922
923         res = ndis_set_oid(drv, OID_802_11_REMOVE_KEY, (char *) &rkey,
924                            sizeof(rkey));
925         if (!pairwise) {
926                 index = key_idx;
927                 res2 = ndis_set_oid(drv, OID_802_11_REMOVE_WEP,
928                                     (char *) &index, sizeof(index));
929         } else
930                 res2 = 0;
931
932         if (res < 0 && res2 < 0)
933                 return -1;
934         return 0;
935 }
936
937
938 static int wpa_driver_ndis_add_wep(struct wpa_driver_ndis_data *drv,
939                                    int pairwise, int key_idx, int set_tx,
940                                    const u8 *key, size_t key_len)
941 {
942         NDIS_802_11_WEP *wep;
943         size_t len;
944         int res;
945
946         len = 12 + key_len;
947         wep = os_zalloc(len);
948         if (wep == NULL)
949                 return -1;
950         wep->Length = len;
951         wep->KeyIndex = key_idx;
952         if (set_tx)
953                 wep->KeyIndex |= 1 << 31;
954 #if 0 /* Setting bit30 does not seem to work with some NDIS drivers */
955         if (pairwise)
956                 wep->KeyIndex |= 1 << 30;
957 #endif
958         wep->KeyLength = key_len;
959         os_memcpy(wep->KeyMaterial, key, key_len);
960
961         wpa_hexdump_key(MSG_MSGDUMP, "NDIS: OID_802_11_ADD_WEP",
962                         (u8 *) wep, len);
963         res = ndis_set_oid(drv, OID_802_11_ADD_WEP, (char *) wep, len);
964
965         os_free(wep);
966
967         return res;
968 }
969
970
971 static int wpa_driver_ndis_set_key(const char *ifname, void *priv, wpa_alg alg,
972                                    const u8 *addr, int key_idx, int set_tx,
973                                    const u8 *seq, size_t seq_len,
974                                    const u8 *key, size_t key_len)
975 {
976         struct wpa_driver_ndis_data *drv = priv;
977         size_t len, i;
978         NDIS_802_11_KEY *nkey;
979         int res, pairwise;
980         u8 bssid[ETH_ALEN];
981
982         if (addr == NULL || os_memcmp(addr, "\xff\xff\xff\xff\xff\xff",
983                                       ETH_ALEN) == 0) {
984                 /* Group Key */
985                 pairwise = 0;
986                 if (wpa_driver_ndis_get_bssid(drv, bssid) < 0)
987                         os_memset(bssid, 0xff, ETH_ALEN);
988         } else {
989                 /* Pairwise Key */
990                 pairwise = 1;
991                 os_memcpy(bssid, addr, ETH_ALEN);
992         }
993
994         if (alg == WPA_ALG_NONE || key_len == 0) {
995                 return wpa_driver_ndis_remove_key(drv, key_idx, addr, bssid,
996                                                   pairwise);
997         }
998
999         if (alg == WPA_ALG_WEP) {
1000                 return wpa_driver_ndis_add_wep(drv, pairwise, key_idx, set_tx,
1001                                                key, key_len);
1002         }
1003
1004         len = 12 + 6 + 6 + 8 + key_len;
1005
1006         nkey = os_zalloc(len);
1007         if (nkey == NULL)
1008                 return -1;
1009
1010         nkey->Length = len;
1011         nkey->KeyIndex = key_idx;
1012         if (set_tx)
1013                 nkey->KeyIndex |= 1 << 31;
1014         if (pairwise)
1015                 nkey->KeyIndex |= 1 << 30;
1016         if (seq && seq_len)
1017                 nkey->KeyIndex |= 1 << 29;
1018         nkey->KeyLength = key_len;
1019         os_memcpy(nkey->BSSID, bssid, ETH_ALEN);
1020         if (seq && seq_len) {
1021                 for (i = 0; i < seq_len; i++)
1022                         nkey->KeyRSC |= (ULONGLONG) seq[i] << (i * 8);
1023         }
1024         if (alg == WPA_ALG_TKIP && key_len == 32) {
1025                 os_memcpy(nkey->KeyMaterial, key, 16);
1026                 os_memcpy(nkey->KeyMaterial + 16, key + 24, 8);
1027                 os_memcpy(nkey->KeyMaterial + 24, key + 16, 8);
1028         } else {
1029                 os_memcpy(nkey->KeyMaterial, key, key_len);
1030         }
1031
1032         wpa_hexdump_key(MSG_MSGDUMP, "NDIS: OID_802_11_ADD_KEY",
1033                         (u8 *) nkey, len);
1034         res = ndis_set_oid(drv, OID_802_11_ADD_KEY, (char *) nkey, len);
1035         os_free(nkey);
1036
1037         return res;
1038 }
1039
1040
1041 static int
1042 wpa_driver_ndis_associate(void *priv,
1043                           struct wpa_driver_associate_params *params)
1044 {
1045         struct wpa_driver_ndis_data *drv = priv;
1046         u32 auth_mode, encr, priv_mode, mode;
1047
1048         drv->mode = params->mode;
1049
1050         /* Note: Setting OID_802_11_INFRASTRUCTURE_MODE clears current keys,
1051          * so static WEP keys needs to be set again after this. */
1052         if (params->mode == IEEE80211_MODE_IBSS) {
1053                 mode = Ndis802_11IBSS;
1054                 /* Need to make sure that BSSID polling is enabled for
1055                  * IBSS mode. */
1056                 eloop_cancel_timeout(wpa_driver_ndis_poll_timeout, drv, NULL);
1057                 eloop_register_timeout(1, 0, wpa_driver_ndis_poll_timeout,
1058                                        drv, NULL);
1059         } else
1060                 mode = Ndis802_11Infrastructure;
1061         if (ndis_set_oid(drv, OID_802_11_INFRASTRUCTURE_MODE,
1062                          (char *) &mode, sizeof(mode)) < 0) {
1063                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
1064                            "OID_802_11_INFRASTRUCTURE_MODE (%d)",
1065                            (int) mode);
1066                 /* Try to continue anyway */
1067         }
1068
1069         if (params->key_mgmt_suite == KEY_MGMT_NONE ||
1070             params->key_mgmt_suite == KEY_MGMT_802_1X_NO_WPA) {
1071                 /* Re-set WEP keys if static WEP configuration is used. */
1072                 u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
1073                 int i;
1074                 for (i = 0; i < 4; i++) {
1075                         if (!params->wep_key[i])
1076                                 continue;
1077                         wpa_printf(MSG_DEBUG, "NDIS: Re-setting static WEP "
1078                                    "key %d", i);
1079                         wpa_driver_ndis_set_key(drv->ifname, drv, WPA_ALG_WEP,
1080                                                 bcast, i,
1081                                                 i == params->wep_tx_keyidx,
1082                                                 NULL, 0, params->wep_key[i],
1083                                                 params->wep_key_len[i]);
1084                 }
1085         }
1086
1087         if (params->wpa_ie == NULL || params->wpa_ie_len == 0) {
1088                 if (params->auth_alg & AUTH_ALG_SHARED_KEY) {
1089                         if (params->auth_alg & AUTH_ALG_OPEN_SYSTEM)
1090                                 auth_mode = Ndis802_11AuthModeAutoSwitch;
1091                         else
1092                                 auth_mode = Ndis802_11AuthModeShared;
1093                 } else
1094                         auth_mode = Ndis802_11AuthModeOpen;
1095                 priv_mode = Ndis802_11PrivFilterAcceptAll;
1096         } else if (params->wpa_ie[0] == WLAN_EID_RSN) {
1097                 priv_mode = Ndis802_11PrivFilter8021xWEP;
1098                 if (params->key_mgmt_suite == KEY_MGMT_PSK)
1099                         auth_mode = Ndis802_11AuthModeWPA2PSK;
1100                 else
1101                         auth_mode = Ndis802_11AuthModeWPA2;
1102 #ifdef CONFIG_WPS
1103         } else if (params->key_mgmt_suite == KEY_MGMT_WPS) {
1104                 auth_mode = Ndis802_11AuthModeOpen;
1105                 priv_mode = Ndis802_11PrivFilterAcceptAll;
1106 #endif /* CONFIG_WPS */
1107         } else {
1108                 priv_mode = Ndis802_11PrivFilter8021xWEP;
1109                 if (params->key_mgmt_suite == KEY_MGMT_WPA_NONE)
1110                         auth_mode = Ndis802_11AuthModeWPANone;
1111                 else if (params->key_mgmt_suite == KEY_MGMT_PSK)
1112                         auth_mode = Ndis802_11AuthModeWPAPSK;
1113                 else
1114                         auth_mode = Ndis802_11AuthModeWPA;
1115         }
1116
1117         switch (params->pairwise_suite) {
1118         case CIPHER_CCMP:
1119                 encr = Ndis802_11Encryption3Enabled;
1120                 break;
1121         case CIPHER_TKIP:
1122                 encr = Ndis802_11Encryption2Enabled;
1123                 break;
1124         case CIPHER_WEP40:
1125         case CIPHER_WEP104:
1126                 encr = Ndis802_11Encryption1Enabled;
1127                 break;
1128         case CIPHER_NONE:
1129                 if (params->group_suite == CIPHER_CCMP)
1130                         encr = Ndis802_11Encryption3Enabled;
1131                 else if (params->group_suite == CIPHER_TKIP)
1132                         encr = Ndis802_11Encryption2Enabled;
1133                 else
1134                         encr = Ndis802_11EncryptionDisabled;
1135                 break;
1136         default:
1137                 encr = Ndis802_11EncryptionDisabled;
1138         };
1139
1140         if (ndis_set_oid(drv, OID_802_11_PRIVACY_FILTER,
1141                          (char *) &priv_mode, sizeof(priv_mode)) < 0) {
1142                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
1143                            "OID_802_11_PRIVACY_FILTER (%d)",
1144                            (int) priv_mode);
1145                 /* Try to continue anyway */
1146         }
1147
1148         ndis_set_auth_mode(drv, auth_mode);
1149         ndis_set_encr_status(drv, encr);
1150
1151         if (params->bssid) {
1152                 ndis_set_oid(drv, OID_802_11_BSSID, (char *) params->bssid,
1153                              ETH_ALEN);
1154                 drv->oid_bssid_set = 1;
1155         } else if (drv->oid_bssid_set) {
1156                 ndis_set_oid(drv, OID_802_11_BSSID, "\xff\xff\xff\xff\xff\xff",
1157                              ETH_ALEN);
1158                 drv->oid_bssid_set = 0;
1159         }
1160
1161         return wpa_driver_ndis_set_ssid(drv, params->ssid, params->ssid_len);
1162 }
1163
1164
1165 static int wpa_driver_ndis_set_pmkid(struct wpa_driver_ndis_data *drv)
1166 {
1167         int len, count, i, ret;
1168         struct ndis_pmkid_entry *entry;
1169         NDIS_802_11_PMKID *p;
1170
1171         count = 0;
1172         entry = drv->pmkid;
1173         while (entry) {
1174                 count++;
1175                 if (count >= drv->no_of_pmkid)
1176                         break;
1177                 entry = entry->next;
1178         }
1179         len = 8 + count * sizeof(BSSID_INFO);
1180         p = os_zalloc(len);
1181         if (p == NULL)
1182                 return -1;
1183
1184         p->Length = len;
1185         p->BSSIDInfoCount = count;
1186         entry = drv->pmkid;
1187         for (i = 0; i < count; i++) {
1188                 os_memcpy(&p->BSSIDInfo[i].BSSID, entry->bssid, ETH_ALEN);
1189                 os_memcpy(&p->BSSIDInfo[i].PMKID, entry->pmkid, 16);
1190                 entry = entry->next;
1191         }
1192         wpa_hexdump(MSG_MSGDUMP, "NDIS: OID_802_11_PMKID", (u8 *) p, len);
1193         ret = ndis_set_oid(drv, OID_802_11_PMKID, (char *) p, len);
1194         os_free(p);
1195         return ret;
1196 }
1197
1198
1199 static int wpa_driver_ndis_add_pmkid(void *priv, const u8 *bssid,
1200                                      const u8 *pmkid)
1201 {
1202         struct wpa_driver_ndis_data *drv = priv;
1203         struct ndis_pmkid_entry *entry, *prev;
1204
1205         if (drv->no_of_pmkid == 0)
1206                 return 0;
1207
1208         prev = NULL;
1209         entry = drv->pmkid;
1210         while (entry) {
1211                 if (os_memcmp(entry->bssid, bssid, ETH_ALEN) == 0)
1212                         break;
1213                 prev = entry;
1214                 entry = entry->next;
1215         }
1216
1217         if (entry) {
1218                 /* Replace existing entry for this BSSID and move it into the
1219                  * beginning of the list. */
1220                 os_memcpy(entry->pmkid, pmkid, 16);
1221                 if (prev) {
1222                         prev->next = entry->next;
1223                         entry->next = drv->pmkid;
1224                         drv->pmkid = entry;
1225                 }
1226         } else {
1227                 entry = os_malloc(sizeof(*entry));
1228                 if (entry) {
1229                         os_memcpy(entry->bssid, bssid, ETH_ALEN);
1230                         os_memcpy(entry->pmkid, pmkid, 16);
1231                         entry->next = drv->pmkid;
1232                         drv->pmkid = entry;
1233                 }
1234         }
1235
1236         return wpa_driver_ndis_set_pmkid(drv);
1237 }
1238
1239
1240 static int wpa_driver_ndis_remove_pmkid(void *priv, const u8 *bssid,
1241                                         const u8 *pmkid)
1242 {
1243         struct wpa_driver_ndis_data *drv = priv;
1244         struct ndis_pmkid_entry *entry, *prev;
1245
1246         if (drv->no_of_pmkid == 0)
1247                 return 0;
1248
1249         entry = drv->pmkid;
1250         prev = NULL;
1251         while (entry) {
1252                 if (os_memcmp(entry->bssid, bssid, ETH_ALEN) == 0 &&
1253                     os_memcmp(entry->pmkid, pmkid, 16) == 0) {
1254                         if (prev)
1255                                 prev->next = entry->next;
1256                         else
1257                                 drv->pmkid = entry->next;
1258                         os_free(entry);
1259                         break;
1260                 }
1261                 prev = entry;
1262                 entry = entry->next;
1263         }
1264         return wpa_driver_ndis_set_pmkid(drv);
1265 }
1266
1267
1268 static int wpa_driver_ndis_flush_pmkid(void *priv)
1269 {
1270         struct wpa_driver_ndis_data *drv = priv;
1271         NDIS_802_11_PMKID p;
1272         struct ndis_pmkid_entry *pmkid, *prev;
1273         int prev_authmode, ret;
1274
1275         if (drv->no_of_pmkid == 0)
1276                 return 0;
1277
1278         pmkid = drv->pmkid;
1279         drv->pmkid = NULL;
1280         while (pmkid) {
1281                 prev = pmkid;
1282                 pmkid = pmkid->next;
1283                 os_free(prev);
1284         }
1285
1286         /*
1287          * Some drivers may refuse OID_802_11_PMKID if authMode is not set to
1288          * WPA2, so change authMode temporarily, if needed.
1289          */
1290         prev_authmode = ndis_get_auth_mode(drv);
1291         if (prev_authmode != Ndis802_11AuthModeWPA2)
1292                 ndis_set_auth_mode(drv, Ndis802_11AuthModeWPA2);
1293
1294         os_memset(&p, 0, sizeof(p));
1295         p.Length = 8;
1296         p.BSSIDInfoCount = 0;
1297         wpa_hexdump(MSG_MSGDUMP, "NDIS: OID_802_11_PMKID (flush)",
1298                     (u8 *) &p, 8);
1299         ret = ndis_set_oid(drv, OID_802_11_PMKID, (char *) &p, 8);
1300
1301         if (prev_authmode != Ndis802_11AuthModeWPA2)
1302                 ndis_set_auth_mode(drv, prev_authmode);
1303
1304         return ret;
1305 }
1306
1307
1308 static int wpa_driver_ndis_get_associnfo(struct wpa_driver_ndis_data *drv)
1309 {
1310         char buf[512], *pos;
1311         NDIS_802_11_ASSOCIATION_INFORMATION *ai;
1312         int len;
1313         union wpa_event_data data;
1314         NDIS_802_11_BSSID_LIST_EX *b;
1315         size_t blen, i;
1316
1317         len = ndis_get_oid(drv, OID_802_11_ASSOCIATION_INFORMATION, buf,
1318                            sizeof(buf));
1319         if (len < 0) {
1320                 wpa_printf(MSG_DEBUG, "NDIS: failed to get association "
1321                            "information");
1322                 return -1;
1323         }
1324         if (len > sizeof(buf)) {
1325                 /* Some drivers seem to be producing incorrect length for this
1326                  * data. Limit the length to the current buffer size to avoid
1327                  * crashing in hexdump. The data seems to be otherwise valid,
1328                  * so better try to use it. */
1329                 wpa_printf(MSG_DEBUG, "NDIS: ignored bogus association "
1330                            "information length %d", len);
1331                 len = ndis_get_oid(drv, OID_802_11_ASSOCIATION_INFORMATION,
1332                                    buf, sizeof(buf));
1333                 if (len < -1) {
1334                         wpa_printf(MSG_DEBUG, "NDIS: re-reading association "
1335                                    "information failed");
1336                         return -1;
1337                 }
1338                 if (len > sizeof(buf)) {
1339                         wpa_printf(MSG_DEBUG, "NDIS: ignored bogus association"
1340                                    " information length %d (re-read)", len);
1341                         len = sizeof(buf);
1342                 }
1343         }
1344         wpa_hexdump(MSG_MSGDUMP, "NDIS: association information",
1345                     (u8 *) buf, len);
1346         if (len < sizeof(*ai)) {
1347                 wpa_printf(MSG_DEBUG, "NDIS: too short association "
1348                            "information");
1349                 return -1;
1350         }
1351         ai = (NDIS_802_11_ASSOCIATION_INFORMATION *) buf;
1352         wpa_printf(MSG_DEBUG, "NDIS: ReqFixed=0x%x RespFixed=0x%x off_req=%d "
1353                    "off_resp=%d len_req=%d len_resp=%d",
1354                    ai->AvailableRequestFixedIEs, ai->AvailableResponseFixedIEs,
1355                    (int) ai->OffsetRequestIEs, (int) ai->OffsetResponseIEs,
1356                    (int) ai->RequestIELength, (int) ai->ResponseIELength);
1357
1358         if (ai->OffsetRequestIEs + ai->RequestIELength > (unsigned) len ||
1359             ai->OffsetResponseIEs + ai->ResponseIELength > (unsigned) len) {
1360                 wpa_printf(MSG_DEBUG, "NDIS: association information - "
1361                            "IE overflow");
1362                 return -1;
1363         }
1364
1365         wpa_hexdump(MSG_MSGDUMP, "NDIS: Request IEs",
1366                     (u8 *) buf + ai->OffsetRequestIEs, ai->RequestIELength);
1367         wpa_hexdump(MSG_MSGDUMP, "NDIS: Response IEs",
1368                     (u8 *) buf + ai->OffsetResponseIEs, ai->ResponseIELength);
1369
1370         os_memset(&data, 0, sizeof(data));
1371         data.assoc_info.req_ies = (u8 *) buf + ai->OffsetRequestIEs;
1372         data.assoc_info.req_ies_len = ai->RequestIELength;
1373         data.assoc_info.resp_ies = (u8 *) buf + ai->OffsetResponseIEs;
1374         data.assoc_info.resp_ies_len = ai->ResponseIELength;
1375
1376         blen = 65535;
1377         b = os_zalloc(blen);
1378         if (b == NULL)
1379                 goto skip_scan_results;
1380         len = ndis_get_oid(drv, OID_802_11_BSSID_LIST, (char *) b, blen);
1381         if (len < 0) {
1382                 wpa_printf(MSG_DEBUG, "NDIS: failed to get scan results");
1383                 os_free(b);
1384                 b = NULL;
1385                 goto skip_scan_results;
1386         }
1387         wpa_printf(MSG_DEBUG, "NDIS: %d BSSID items to process for AssocInfo",
1388                    (unsigned int) b->NumberOfItems);
1389
1390         pos = (char *) &b->Bssid[0];
1391         for (i = 0; i < b->NumberOfItems; i++) {
1392                 NDIS_WLAN_BSSID_EX *bss = (NDIS_WLAN_BSSID_EX *) pos;
1393                 if (os_memcmp(drv->bssid, bss->MacAddress, ETH_ALEN) == 0 &&
1394                     bss->IELength > sizeof(NDIS_802_11_FIXED_IEs)) {
1395                         data.assoc_info.beacon_ies =
1396                                 ((u8 *) bss->IEs) +
1397                                 sizeof(NDIS_802_11_FIXED_IEs);
1398                         data.assoc_info.beacon_ies_len =
1399                                 bss->IELength - sizeof(NDIS_802_11_FIXED_IEs);
1400                         wpa_hexdump(MSG_MSGDUMP, "NDIS: Beacon IEs",
1401                                     data.assoc_info.beacon_ies,
1402                                     data.assoc_info.beacon_ies_len);
1403                         break;
1404                 }
1405                 pos += bss->Length;
1406                 if (pos > (char *) b + blen)
1407                         break;
1408         }
1409
1410 skip_scan_results:
1411         wpa_supplicant_event(drv->ctx, EVENT_ASSOCINFO, &data);
1412
1413         os_free(b);
1414
1415         return 0;
1416 }
1417
1418
1419 static void wpa_driver_ndis_poll_timeout(void *eloop_ctx, void *timeout_ctx)
1420 {
1421         struct wpa_driver_ndis_data *drv = eloop_ctx;
1422         u8 bssid[ETH_ALEN];
1423         int poll;
1424
1425         if (drv->wired)
1426                 return;
1427
1428         if (wpa_driver_ndis_get_bssid(drv, bssid)) {
1429                 /* Disconnected */
1430                 if (!is_zero_ether_addr(drv->bssid)) {
1431                         os_memset(drv->bssid, 0, ETH_ALEN);
1432                         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
1433                 }
1434         } else {
1435                 /* Connected */
1436                 if (os_memcmp(drv->bssid, bssid, ETH_ALEN) != 0) {
1437                         os_memcpy(drv->bssid, bssid, ETH_ALEN);
1438                         wpa_driver_ndis_get_associnfo(drv);
1439                         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
1440                 }
1441         }
1442
1443         /* When using integrated NDIS event receiver, we can skip BSSID
1444          * polling when using infrastructure network. However, when using
1445          * IBSS mode, many driver do not seem to generate connection event,
1446          * so we need to enable BSSID polling to figure out when IBSS network
1447          * has been formed.
1448          */
1449         poll = drv->mode == IEEE80211_MODE_IBSS;
1450 #ifndef CONFIG_NDIS_EVENTS_INTEGRATED
1451 #ifndef _WIN32_WCE
1452         poll = 1;
1453 #endif /* _WIN32_WCE */
1454 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
1455
1456         if (poll) {
1457                 eloop_register_timeout(1, 0, wpa_driver_ndis_poll_timeout,
1458                                         drv, NULL);
1459         }
1460 }
1461
1462
1463 static void wpa_driver_ndis_poll(void *priv)
1464 {
1465         struct wpa_driver_ndis_data *drv = priv;
1466         eloop_cancel_timeout(wpa_driver_ndis_poll_timeout, drv, NULL);
1467         wpa_driver_ndis_poll_timeout(drv, NULL);
1468 }
1469
1470
1471 /* Called when driver generates Media Connect Event by calling
1472  * NdisMIndicateStatus() with NDIS_STATUS_MEDIA_CONNECT */
1473 void wpa_driver_ndis_event_connect(struct wpa_driver_ndis_data *drv)
1474 {
1475         wpa_printf(MSG_DEBUG, "NDIS: Media Connect Event");
1476         if (wpa_driver_ndis_get_bssid(drv, drv->bssid) == 0) {
1477                 wpa_driver_ndis_get_associnfo(drv);
1478                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
1479         }
1480 }
1481
1482
1483 /* Called when driver generates Media Disconnect Event by calling
1484  * NdisMIndicateStatus() with NDIS_STATUS_MEDIA_DISCONNECT */
1485 void wpa_driver_ndis_event_disconnect(struct wpa_driver_ndis_data *drv)
1486 {
1487         wpa_printf(MSG_DEBUG, "NDIS: Media Disconnect Event");
1488         os_memset(drv->bssid, 0, ETH_ALEN);
1489         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
1490 }
1491
1492
1493 static void wpa_driver_ndis_event_auth(struct wpa_driver_ndis_data *drv,
1494                                        const u8 *data, size_t data_len)
1495 {
1496         NDIS_802_11_AUTHENTICATION_REQUEST *req;
1497         int pairwise = 0, group = 0;
1498         union wpa_event_data event;
1499
1500         if (data_len < sizeof(*req)) {
1501                 wpa_printf(MSG_DEBUG, "NDIS: Too short Authentication Request "
1502                            "Event (len=%d)", data_len);
1503                 return;
1504         }
1505         req = (NDIS_802_11_AUTHENTICATION_REQUEST *) data;
1506
1507         wpa_printf(MSG_DEBUG, "NDIS: Authentication Request Event: "
1508                    "Bssid " MACSTR " Flags 0x%x",
1509                    MAC2STR(req->Bssid), (int) req->Flags);
1510
1511         if ((req->Flags & NDIS_802_11_AUTH_REQUEST_PAIRWISE_ERROR) ==
1512             NDIS_802_11_AUTH_REQUEST_PAIRWISE_ERROR)
1513                 pairwise = 1;
1514         else if ((req->Flags & NDIS_802_11_AUTH_REQUEST_GROUP_ERROR) ==
1515             NDIS_802_11_AUTH_REQUEST_GROUP_ERROR)
1516                 group = 1;
1517
1518         if (pairwise || group) {
1519                 os_memset(&event, 0, sizeof(event));
1520                 event.michael_mic_failure.unicast = pairwise;
1521                 wpa_supplicant_event(drv->ctx, EVENT_MICHAEL_MIC_FAILURE,
1522                                      &event);
1523         }
1524 }
1525
1526
1527 static void wpa_driver_ndis_event_pmkid(struct wpa_driver_ndis_data *drv,
1528                                         const u8 *data, size_t data_len)
1529 {
1530         NDIS_802_11_PMKID_CANDIDATE_LIST *pmkid;
1531         size_t i;
1532         union wpa_event_data event;
1533
1534         if (data_len < 8) {
1535                 wpa_printf(MSG_DEBUG, "NDIS: Too short PMKID Candidate List "
1536                            "Event (len=%d)", data_len);
1537                 return;
1538         }
1539         pmkid = (NDIS_802_11_PMKID_CANDIDATE_LIST *) data;
1540         wpa_printf(MSG_DEBUG, "NDIS: PMKID Candidate List Event - Version %d "
1541                    "NumCandidates %d",
1542                    (int) pmkid->Version, (int) pmkid->NumCandidates);
1543
1544         if (pmkid->Version != 1) {
1545                 wpa_printf(MSG_DEBUG, "NDIS: Unsupported PMKID Candidate List "
1546                            "Version %d", (int) pmkid->Version);
1547                 return;
1548         }
1549
1550         if (data_len < 8 + pmkid->NumCandidates * sizeof(PMKID_CANDIDATE)) {
1551                 wpa_printf(MSG_DEBUG, "NDIS: PMKID Candidate List underflow");
1552                 return;
1553         }
1554
1555         os_memset(&event, 0, sizeof(event));
1556         for (i = 0; i < pmkid->NumCandidates; i++) {
1557                 PMKID_CANDIDATE *p = &pmkid->CandidateList[i];
1558                 wpa_printf(MSG_DEBUG, "NDIS: %d: " MACSTR " Flags 0x%x",
1559                            i, MAC2STR(p->BSSID), (int) p->Flags);
1560                 os_memcpy(event.pmkid_candidate.bssid, p->BSSID, ETH_ALEN);
1561                 event.pmkid_candidate.index = i;
1562                 event.pmkid_candidate.preauth =
1563                         p->Flags & NDIS_802_11_PMKID_CANDIDATE_PREAUTH_ENABLED;
1564                 wpa_supplicant_event(drv->ctx, EVENT_PMKID_CANDIDATE,
1565                                      &event);
1566         }
1567 }
1568
1569
1570 /* Called when driver calls NdisMIndicateStatus() with
1571  * NDIS_STATUS_MEDIA_SPECIFIC_INDICATION */
1572 void wpa_driver_ndis_event_media_specific(struct wpa_driver_ndis_data *drv,
1573                                           const u8 *data, size_t data_len)
1574 {
1575         NDIS_802_11_STATUS_INDICATION *status;
1576
1577         if (data == NULL || data_len < sizeof(*status))
1578                 return;
1579
1580         wpa_hexdump(MSG_DEBUG, "NDIS: Media Specific Indication",
1581                     data, data_len);
1582
1583         status = (NDIS_802_11_STATUS_INDICATION *) data;
1584         data += sizeof(status);
1585         data_len -= sizeof(status);
1586
1587         switch (status->StatusType) {
1588         case Ndis802_11StatusType_Authentication:
1589                 wpa_driver_ndis_event_auth(drv, data, data_len);
1590                 break;
1591         case Ndis802_11StatusType_PMKID_CandidateList:
1592                 wpa_driver_ndis_event_pmkid(drv, data, data_len);
1593                 break;
1594         default:
1595                 wpa_printf(MSG_DEBUG, "NDIS: Unknown StatusType %d",
1596                            (int) status->StatusType);
1597                 break;
1598         }
1599 }
1600
1601
1602 /* Called when an adapter is added */
1603 void wpa_driver_ndis_event_adapter_arrival(struct wpa_driver_ndis_data *drv)
1604 {
1605         union wpa_event_data event;
1606         int i;
1607
1608         wpa_printf(MSG_DEBUG, "NDIS: Notify Adapter Arrival");
1609
1610         for (i = 0; i < 30; i++) {
1611                 /* Re-open Packet32/NDISUIO connection */
1612                 wpa_driver_ndis_adapter_close(drv);
1613                 if (wpa_driver_ndis_adapter_init(drv) < 0 ||
1614                     wpa_driver_ndis_adapter_open(drv) < 0) {
1615                         wpa_printf(MSG_DEBUG, "NDIS: Driver re-initialization "
1616                                    "(%d) failed", i);
1617                         os_sleep(1, 0);
1618                 } else {
1619                         wpa_printf(MSG_DEBUG, "NDIS: Driver re-initialized");
1620                         break;
1621                 }
1622         }
1623
1624         os_memset(&event, 0, sizeof(event));
1625         os_strlcpy(event.interface_status.ifname, drv->ifname,
1626                    sizeof(event.interface_status.ifname));
1627         event.interface_status.ievent = EVENT_INTERFACE_ADDED;
1628         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_STATUS, &event);
1629 }
1630
1631
1632 /* Called when an adapter is removed */
1633 void wpa_driver_ndis_event_adapter_removal(struct wpa_driver_ndis_data *drv)
1634 {
1635         union wpa_event_data event;
1636
1637         wpa_printf(MSG_DEBUG, "NDIS: Notify Adapter Removal");
1638         os_memset(&event, 0, sizeof(event));
1639         os_strlcpy(event.interface_status.ifname, drv->ifname,
1640                    sizeof(event.interface_status.ifname));
1641         event.interface_status.ievent = EVENT_INTERFACE_REMOVED;
1642         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_STATUS, &event);
1643 }
1644
1645
1646 static void
1647 wpa_driver_ndis_get_wpa_capability(struct wpa_driver_ndis_data *drv)
1648 {
1649         wpa_printf(MSG_DEBUG, "NDIS: verifying driver WPA capability");
1650
1651         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeWPA) == 0 &&
1652             ndis_get_auth_mode(drv) == Ndis802_11AuthModeWPA) {
1653                 wpa_printf(MSG_DEBUG, "NDIS: WPA key management supported");
1654                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA;
1655         }
1656
1657         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeWPAPSK) == 0 &&
1658             ndis_get_auth_mode(drv) == Ndis802_11AuthModeWPAPSK) {
1659                 wpa_printf(MSG_DEBUG, "NDIS: WPA-PSK key management "
1660                            "supported");
1661                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK;
1662         }
1663
1664         if (ndis_set_encr_status(drv, Ndis802_11Encryption3Enabled) == 0 &&
1665             ndis_get_encr_status(drv) == Ndis802_11Encryption3KeyAbsent) {
1666                 wpa_printf(MSG_DEBUG, "NDIS: CCMP encryption supported");
1667                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_CCMP;
1668         }
1669
1670         if (ndis_set_encr_status(drv, Ndis802_11Encryption2Enabled) == 0 &&
1671             ndis_get_encr_status(drv) == Ndis802_11Encryption2KeyAbsent) {
1672                 wpa_printf(MSG_DEBUG, "NDIS: TKIP encryption supported");
1673                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_TKIP;
1674         }
1675
1676         if (ndis_set_encr_status(drv, Ndis802_11Encryption1Enabled) == 0 &&
1677             ndis_get_encr_status(drv) == Ndis802_11Encryption1KeyAbsent) {
1678                 wpa_printf(MSG_DEBUG, "NDIS: WEP encryption supported");
1679                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP40 |
1680                         WPA_DRIVER_CAPA_ENC_WEP104;
1681         }
1682
1683         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeShared) == 0 &&
1684             ndis_get_auth_mode(drv) == Ndis802_11AuthModeShared) {
1685                 drv->capa.auth |= WPA_DRIVER_AUTH_SHARED;
1686         }
1687
1688         if (ndis_set_auth_mode(drv, Ndis802_11AuthModeOpen) == 0 &&
1689             ndis_get_auth_mode(drv) == Ndis802_11AuthModeOpen) {
1690                 drv->capa.auth |= WPA_DRIVER_AUTH_OPEN;
1691         }
1692
1693         ndis_set_encr_status(drv, Ndis802_11EncryptionDisabled);
1694
1695         /* Could also verify OID_802_11_ADD_KEY error reporting and
1696          * support for OID_802_11_ASSOCIATION_INFORMATION. */
1697
1698         if (drv->capa.key_mgmt & WPA_DRIVER_CAPA_KEY_MGMT_WPA &&
1699             drv->capa.enc & (WPA_DRIVER_CAPA_ENC_TKIP |
1700                              WPA_DRIVER_CAPA_ENC_CCMP)) {
1701                 wpa_printf(MSG_DEBUG, "NDIS: driver supports WPA");
1702                 drv->has_capability = 1;
1703         } else {
1704                 wpa_printf(MSG_DEBUG, "NDIS: no WPA support found");
1705         }
1706
1707         wpa_printf(MSG_DEBUG, "NDIS: driver capabilities: key_mgmt 0x%x "
1708                    "enc 0x%x auth 0x%x",
1709                    drv->capa.key_mgmt, drv->capa.enc, drv->capa.auth);
1710 }
1711
1712
1713 static void wpa_driver_ndis_get_capability(struct wpa_driver_ndis_data *drv)
1714 {
1715         char buf[512];
1716         int len;
1717         size_t i;
1718         NDIS_802_11_CAPABILITY *c;
1719
1720         drv->capa.flags = WPA_DRIVER_FLAGS_DRIVER_IE;
1721
1722         len = ndis_get_oid(drv, OID_802_11_CAPABILITY, buf, sizeof(buf));
1723         if (len < 0) {
1724                 wpa_driver_ndis_get_wpa_capability(drv);
1725                 return;
1726         }
1727
1728         wpa_hexdump(MSG_MSGDUMP, "OID_802_11_CAPABILITY", (u8 *) buf, len);
1729         c = (NDIS_802_11_CAPABILITY *) buf;
1730         if (len < sizeof(*c) || c->Version != 2) {
1731                 wpa_printf(MSG_DEBUG, "NDIS: unsupported "
1732                            "OID_802_11_CAPABILITY data");
1733                 return;
1734         }
1735         wpa_printf(MSG_DEBUG, "NDIS: Driver supports OID_802_11_CAPABILITY - "
1736                    "NoOfPMKIDs %d NoOfAuthEncrPairs %d",
1737                    (int) c->NoOfPMKIDs,
1738                    (int) c->NoOfAuthEncryptPairsSupported);
1739         drv->has_capability = 1;
1740         drv->no_of_pmkid = c->NoOfPMKIDs;
1741         for (i = 0; i < c->NoOfAuthEncryptPairsSupported; i++) {
1742                 NDIS_802_11_AUTHENTICATION_ENCRYPTION *ae;
1743                 ae = &c->AuthenticationEncryptionSupported[i];
1744                 if ((char *) (ae + 1) > buf + len) {
1745                         wpa_printf(MSG_DEBUG, "NDIS: auth/encr pair list "
1746                                    "overflow");
1747                         break;
1748                 }
1749                 wpa_printf(MSG_MSGDUMP, "NDIS: %d - auth %d encr %d",
1750                            i, (int) ae->AuthModeSupported,
1751                            (int) ae->EncryptStatusSupported);
1752                 switch (ae->AuthModeSupported) {
1753                 case Ndis802_11AuthModeOpen:
1754                         drv->capa.auth |= WPA_DRIVER_AUTH_OPEN;
1755                         break;
1756                 case Ndis802_11AuthModeShared:
1757                         drv->capa.auth |= WPA_DRIVER_AUTH_SHARED;
1758                         break;
1759                 case Ndis802_11AuthModeWPA:
1760                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA;
1761                         break;
1762                 case Ndis802_11AuthModeWPAPSK:
1763                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK;
1764                         break;
1765                 case Ndis802_11AuthModeWPA2:
1766                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA2;
1767                         break;
1768                 case Ndis802_11AuthModeWPA2PSK:
1769                         drv->capa.key_mgmt |=
1770                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
1771                         break;
1772                 case Ndis802_11AuthModeWPANone:
1773                         drv->capa.key_mgmt |=
1774                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_NONE;
1775                         break;
1776                 default:
1777                         break;
1778                 }
1779                 switch (ae->EncryptStatusSupported) {
1780                 case Ndis802_11Encryption1Enabled:
1781                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP40;
1782                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP104;
1783                         break;
1784                 case Ndis802_11Encryption2Enabled:
1785                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_TKIP;
1786                         break;
1787                 case Ndis802_11Encryption3Enabled:
1788                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_CCMP;
1789                         break;
1790                 default:
1791                         break;
1792                 }
1793         }
1794
1795         wpa_printf(MSG_DEBUG, "NDIS: driver capabilities: key_mgmt 0x%x "
1796                    "enc 0x%x auth 0x%x",
1797                    drv->capa.key_mgmt, drv->capa.enc, drv->capa.auth);
1798 }
1799
1800
1801 static int wpa_driver_ndis_get_capa(void *priv, struct wpa_driver_capa *capa)
1802 {
1803         struct wpa_driver_ndis_data *drv = priv;
1804         if (!drv->has_capability)
1805                 return -1;
1806         os_memcpy(capa, &drv->capa, sizeof(*capa));
1807         return 0;
1808 }
1809
1810
1811 static const char * wpa_driver_ndis_get_ifname(void *priv)
1812 {
1813         struct wpa_driver_ndis_data *drv = priv;
1814         return drv->ifname;
1815 }
1816
1817
1818 static const u8 * wpa_driver_ndis_get_mac_addr(void *priv)
1819 {
1820         struct wpa_driver_ndis_data *drv = priv;
1821         return drv->own_addr;
1822 }
1823
1824
1825 #ifdef _WIN32_WCE
1826
1827 #define NDISUIO_MSG_SIZE (sizeof(NDISUIO_DEVICE_NOTIFICATION) + 512)
1828
1829 static void ndisuio_notification_receive(void *eloop_data, void *user_ctx)
1830 {
1831         struct wpa_driver_ndis_data *drv = eloop_data;
1832         NDISUIO_DEVICE_NOTIFICATION *hdr;
1833         u8 buf[NDISUIO_MSG_SIZE];
1834         DWORD len, flags;
1835
1836         if (!ReadMsgQueue(drv->event_queue, buf, NDISUIO_MSG_SIZE, &len, 0,
1837                           &flags)) {
1838                 wpa_printf(MSG_DEBUG, "ndisuio_notification_receive: "
1839                            "ReadMsgQueue failed: %d", (int) GetLastError());
1840                 return;
1841         }
1842
1843         if (len < sizeof(NDISUIO_DEVICE_NOTIFICATION)) {
1844                 wpa_printf(MSG_DEBUG, "ndisuio_notification_receive: "
1845                            "Too short message (len=%d)", (int) len);
1846                 return;
1847         }
1848
1849         hdr = (NDISUIO_DEVICE_NOTIFICATION *) buf;
1850         wpa_printf(MSG_DEBUG, "NDIS: Notification received: len=%d type=0x%x",
1851                    (int) len, hdr->dwNotificationType);
1852
1853         switch (hdr->dwNotificationType) {
1854 #ifdef NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL
1855         case NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL:
1856                 wpa_printf(MSG_DEBUG, "NDIS: ADAPTER_ARRIVAL");
1857                 wpa_driver_ndis_event_adapter_arrival(drv);
1858                 break;
1859 #endif
1860 #ifdef NDISUIO_NOTIFICATION_ADAPTER_REMOVAL
1861         case NDISUIO_NOTIFICATION_ADAPTER_REMOVAL:
1862                 wpa_printf(MSG_DEBUG, "NDIS: ADAPTER_REMOVAL");
1863                 wpa_driver_ndis_event_adapter_removal(drv);
1864                 break;
1865 #endif
1866         case NDISUIO_NOTIFICATION_MEDIA_CONNECT:
1867                 wpa_printf(MSG_DEBUG, "NDIS: MEDIA_CONNECT");
1868                 SetEvent(drv->connected_event);
1869                 wpa_driver_ndis_event_connect(drv);
1870                 break;
1871         case NDISUIO_NOTIFICATION_MEDIA_DISCONNECT:
1872                 ResetEvent(drv->connected_event);
1873                 wpa_printf(MSG_DEBUG, "NDIS: MEDIA_DISCONNECT");
1874                 wpa_driver_ndis_event_disconnect(drv);
1875                 break;
1876         case NDISUIO_NOTIFICATION_MEDIA_SPECIFIC_NOTIFICATION:
1877                 wpa_printf(MSG_DEBUG, "NDIS: MEDIA_SPECIFIC_NOTIFICATION");
1878 #if _WIN32_WCE == 420 || _WIN32_WCE == 0x420
1879                 wpa_driver_ndis_event_media_specific(
1880                         drv, hdr->pvStatusBuffer, hdr->uiStatusBufferSize);
1881 #else
1882                 wpa_driver_ndis_event_media_specific(
1883                         drv, ((const u8 *) hdr) + hdr->uiOffsetToStatusBuffer,
1884                         (size_t) hdr->uiStatusBufferSize);
1885 #endif
1886                 break;
1887         default:
1888                 wpa_printf(MSG_DEBUG, "NDIS: Unknown notification type 0x%x",
1889                            hdr->dwNotificationType);
1890                 break;
1891         }
1892 }
1893
1894
1895 static void ndisuio_notification_deinit(struct wpa_driver_ndis_data *drv)
1896 {
1897         NDISUIO_REQUEST_NOTIFICATION req;
1898
1899         memset(&req, 0, sizeof(req));
1900         req.hMsgQueue = drv->event_queue;
1901         req.dwNotificationTypes = 0;
1902
1903         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_REQUEST_NOTIFICATION,
1904                              &req, sizeof(req), NULL, 0, NULL, NULL)) {
1905                 wpa_printf(MSG_INFO, "ndisuio_notification_deinit: "
1906                            "IOCTL_NDISUIO_REQUEST_NOTIFICATION failed: %d",
1907                            (int) GetLastError());
1908         }
1909
1910         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_CANCEL_NOTIFICATION,
1911                              NULL, 0, NULL, 0, NULL, NULL)) {
1912                 wpa_printf(MSG_INFO, "ndisuio_notification_deinit: "
1913                            "IOCTL_NDISUIO_CANCEL_NOTIFICATION failed: %d",
1914                            (int) GetLastError());
1915         }
1916
1917         if (drv->event_queue) {
1918                 eloop_unregister_event(drv->event_queue,
1919                                        sizeof(drv->event_queue));
1920                 CloseHandle(drv->event_queue);
1921                 drv->event_queue = NULL;
1922         }
1923
1924         if (drv->connected_event) {
1925                 CloseHandle(drv->connected_event);
1926                 drv->connected_event = NULL;
1927         }
1928 }
1929
1930
1931 static int ndisuio_notification_init(struct wpa_driver_ndis_data *drv)
1932 {
1933         MSGQUEUEOPTIONS opt;
1934         NDISUIO_REQUEST_NOTIFICATION req;
1935
1936         drv->connected_event =
1937                 CreateEvent(NULL, TRUE, FALSE, TEXT("WpaSupplicantConnected"));
1938         if (drv->connected_event == NULL) {
1939                 wpa_printf(MSG_INFO, "ndisuio_notification_init: "
1940                            "CreateEvent failed: %d",
1941                            (int) GetLastError());
1942                 return -1;
1943         }
1944
1945         memset(&opt, 0, sizeof(opt));
1946         opt.dwSize = sizeof(opt);
1947         opt.dwMaxMessages = 5;
1948         opt.cbMaxMessage = NDISUIO_MSG_SIZE;
1949         opt.bReadAccess = TRUE;
1950
1951         drv->event_queue = CreateMsgQueue(NULL, &opt);
1952         if (drv->event_queue == NULL) {
1953                 wpa_printf(MSG_INFO, "ndisuio_notification_init: "
1954                            "CreateMsgQueue failed: %d",
1955                            (int) GetLastError());
1956                 ndisuio_notification_deinit(drv);
1957                 return -1;
1958         }
1959
1960         memset(&req, 0, sizeof(req));
1961         req.hMsgQueue = drv->event_queue;
1962         req.dwNotificationTypes =
1963 #ifdef NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL
1964                 NDISUIO_NOTIFICATION_ADAPTER_ARRIVAL |
1965 #endif
1966 #ifdef NDISUIO_NOTIFICATION_ADAPTER_REMOVAL
1967                 NDISUIO_NOTIFICATION_ADAPTER_REMOVAL |
1968 #endif
1969                 NDISUIO_NOTIFICATION_MEDIA_CONNECT |
1970                 NDISUIO_NOTIFICATION_MEDIA_DISCONNECT |
1971                 NDISUIO_NOTIFICATION_MEDIA_SPECIFIC_NOTIFICATION;
1972
1973         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_REQUEST_NOTIFICATION,
1974                              &req, sizeof(req), NULL, 0, NULL, NULL)) {
1975                 wpa_printf(MSG_INFO, "ndisuio_notification_init: "
1976                            "IOCTL_NDISUIO_REQUEST_NOTIFICATION failed: %d",
1977                            (int) GetLastError());
1978                 ndisuio_notification_deinit(drv);
1979                 return -1;
1980         }
1981
1982         eloop_register_event(drv->event_queue, sizeof(drv->event_queue),
1983                              ndisuio_notification_receive, drv, NULL);
1984
1985         return 0;
1986 }
1987 #endif /* _WIN32_WCE */
1988
1989
1990 static int wpa_driver_ndis_get_names(struct wpa_driver_ndis_data *drv)
1991 {
1992 #ifdef CONFIG_USE_NDISUIO
1993         NDISUIO_QUERY_BINDING *b;
1994         size_t blen = sizeof(*b) + 1024;
1995         int i, error, found = 0;
1996         DWORD written;
1997         char name[256], desc[256], *dpos;
1998         WCHAR *pos;
1999         size_t j, len, dlen;
2000
2001         b = os_malloc(blen);
2002         if (b == NULL)
2003                 return -1;
2004
2005         for (i = 0; ; i++) {
2006                 os_memset(b, 0, blen);
2007                 b->BindingIndex = i;
2008                 if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_QUERY_BINDING,
2009                                      b, sizeof(NDISUIO_QUERY_BINDING), b, blen,
2010                                      &written, NULL)) {
2011                         error = (int) GetLastError();
2012                         if (error == ERROR_NO_MORE_ITEMS)
2013                                 break;
2014                         wpa_printf(MSG_DEBUG, "IOCTL_NDISUIO_QUERY_BINDING "
2015                                    "failed: %d", error);
2016                         break;
2017                 }
2018
2019                 pos = (WCHAR *) ((char *) b + b->DeviceNameOffset);
2020                 len = b->DeviceNameLength;
2021                 if (len >= sizeof(name))
2022                         len = sizeof(name) - 1;
2023                 for (j = 0; j < len; j++)
2024                         name[j] = (char) pos[j];
2025                 name[len] = '\0';
2026
2027                 pos = (WCHAR *) ((char *) b + b->DeviceDescrOffset);
2028                 len = b->DeviceDescrLength;
2029                 if (len >= sizeof(desc))
2030                         len = sizeof(desc) - 1;
2031                 for (j = 0; j < len; j++)
2032                         desc[j] = (char) pos[j];
2033                 desc[len] = '\0';
2034
2035                 wpa_printf(MSG_DEBUG, "NDIS: %d - %s - %s", i, name, desc);
2036
2037                 if (os_strstr(name, drv->ifname)) {
2038                         wpa_printf(MSG_DEBUG, "NDIS: Interface name match");
2039                         found = 1;
2040                         break;
2041                 }
2042
2043                 if (os_strncmp(desc, drv->ifname, os_strlen(drv->ifname)) == 0)
2044                 {
2045                         wpa_printf(MSG_DEBUG, "NDIS: Interface description "
2046                                    "match");
2047                         found = 1;
2048                         break;
2049                 }
2050         }
2051
2052         if (!found) {
2053                 wpa_printf(MSG_DEBUG, "NDIS: Could not find interface '%s'",
2054                            drv->ifname);
2055                 os_free(b);
2056                 return -1;
2057         }
2058
2059         os_strlcpy(drv->ifname,
2060                    os_strncmp(name, "\\DEVICE\\", 8) == 0 ? name + 8 : name,
2061                    sizeof(drv->ifname));
2062 #ifdef _WIN32_WCE
2063         drv->adapter_name = wpa_strdup_tchar(drv->ifname);
2064         if (drv->adapter_name == NULL) {
2065                 wpa_printf(MSG_ERROR, "NDIS: Failed to allocate memory for "
2066                            "adapter name");
2067                 os_free(b);
2068                 return -1;
2069         }
2070 #endif /* _WIN32_WCE */
2071
2072         dpos = os_strstr(desc, " - ");
2073         if (dpos)
2074                 dlen = dpos - desc;
2075         else
2076                 dlen = os_strlen(desc);
2077         drv->adapter_desc = os_malloc(dlen + 1);
2078         if (drv->adapter_desc) {
2079                 os_memcpy(drv->adapter_desc, desc, dlen);
2080                 drv->adapter_desc[dlen] = '\0';
2081         }
2082
2083         os_free(b);
2084
2085         if (drv->adapter_desc == NULL)
2086                 return -1;
2087
2088         wpa_printf(MSG_DEBUG, "NDIS: Adapter description prefix '%s'",
2089                    drv->adapter_desc);
2090
2091         return 0;
2092 #else /* CONFIG_USE_NDISUIO */
2093         PTSTR _names;
2094         char *names, *pos, *pos2;
2095         ULONG len;
2096         BOOLEAN res;
2097 #define MAX_ADAPTERS 32
2098         char *name[MAX_ADAPTERS];
2099         char *desc[MAX_ADAPTERS];
2100         int num_name, num_desc, i, found_name, found_desc;
2101         size_t dlen;
2102
2103         wpa_printf(MSG_DEBUG, "NDIS: Packet.dll version: %s",
2104                    PacketGetVersion());
2105
2106         len = 8192;
2107         _names = os_zalloc(len);
2108         if (_names == NULL)
2109                 return -1;
2110
2111         res = PacketGetAdapterNames(_names, &len);
2112         if (!res && len > 8192) {
2113                 os_free(_names);
2114                 _names = os_zalloc(len);
2115                 if (_names == NULL)
2116                         return -1;
2117                 res = PacketGetAdapterNames(_names, &len);
2118         }
2119
2120         if (!res) {
2121                 wpa_printf(MSG_ERROR, "NDIS: Failed to get adapter list "
2122                            "(PacketGetAdapterNames)");
2123                 os_free(_names);
2124                 return -1;
2125         }
2126
2127         names = (char *) _names;
2128         if (names[0] && names[1] == '\0' && names[2] && names[3] == '\0') {
2129                 wpa_printf(MSG_DEBUG, "NDIS: Looks like adapter names are in "
2130                            "UNICODE");
2131                 /* Convert to ASCII */
2132                 pos2 = pos = names;
2133                 while (pos2 < names + len) {
2134                         if (pos2[0] == '\0' && pos2[1] == '\0' &&
2135                             pos2[2] == '\0' && pos2[3] == '\0') {
2136                                 pos2 += 4;
2137                                 break;
2138                         }
2139                         *pos++ = pos2[0];
2140                         pos2 += 2;
2141                 }
2142                 os_memcpy(pos + 2, names, pos - names);
2143                 pos += 2;
2144         } else
2145                 pos = names;
2146
2147         num_name = 0;
2148         while (pos < names + len) {
2149                 name[num_name] = pos;
2150                 while (*pos && pos < names + len)
2151                         pos++;
2152                 if (pos + 1 >= names + len) {
2153                         os_free(names);
2154                         return -1;
2155                 }
2156                 pos++;
2157                 num_name++;
2158                 if (num_name >= MAX_ADAPTERS) {
2159                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapters");
2160                         os_free(names);
2161                         return -1;
2162                 }
2163                 if (*pos == '\0') {
2164                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter names found",
2165                                    num_name);
2166                         pos++;
2167                         break;
2168                 }
2169         }
2170
2171         num_desc = 0;
2172         while (pos < names + len) {
2173                 desc[num_desc] = pos;
2174                 while (*pos && pos < names + len)
2175                         pos++;
2176                 if (pos + 1 >= names + len) {
2177                         os_free(names);
2178                         return -1;
2179                 }
2180                 pos++;
2181                 num_desc++;
2182                 if (num_desc >= MAX_ADAPTERS) {
2183                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapter "
2184                                    "descriptions");
2185                         os_free(names);
2186                         return -1;
2187                 }
2188                 if (*pos == '\0') {
2189                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter descriptions "
2190                                    "found", num_name);
2191                         pos++;
2192                         break;
2193                 }
2194         }
2195
2196         /*
2197          * Windows 98 with Packet.dll 3.0 alpha3 does not include adapter
2198          * descriptions. Fill in dummy descriptors to work around this.
2199          */
2200         while (num_desc < num_name)
2201                 desc[num_desc++] = "dummy description";
2202
2203         if (num_name != num_desc) {
2204                 wpa_printf(MSG_DEBUG, "NDIS: mismatch in adapter name and "
2205                            "description counts (%d != %d)",
2206                            num_name, num_desc);
2207                 os_free(names);
2208                 return -1;
2209         }
2210
2211         found_name = found_desc = -1;
2212         for (i = 0; i < num_name; i++) {
2213                 wpa_printf(MSG_DEBUG, "NDIS: %d - %s - %s",
2214                            i, name[i], desc[i]);
2215                 if (found_name == -1 && os_strstr(name[i], drv->ifname))
2216                         found_name = i;
2217                 if (found_desc == -1 &&
2218                     os_strncmp(desc[i], drv->ifname, os_strlen(drv->ifname)) ==
2219                     0)
2220                         found_desc = i;
2221         }
2222
2223         if (found_name < 0 && found_desc >= 0) {
2224                 wpa_printf(MSG_DEBUG, "NDIS: Matched interface '%s' based on "
2225                            "description '%s'",
2226                            name[found_desc], desc[found_desc]);
2227                 found_name = found_desc;
2228                 os_strlcpy(drv->ifname,
2229                            os_strncmp(name[found_desc], "\\Device\\NPF_", 12)
2230                            == 0 ? name[found_desc] + 12 : name[found_desc],
2231                            sizeof(drv->ifname));
2232         }
2233
2234         if (found_name < 0) {
2235                 wpa_printf(MSG_DEBUG, "NDIS: Could not find interface '%s'",
2236                            drv->ifname);
2237                 os_free(names);
2238                 return -1;
2239         }
2240
2241         i = found_name;
2242         pos = os_strrchr(desc[i], '(');
2243         if (pos) {
2244                 dlen = pos - desc[i];
2245                 pos--;
2246                 if (pos > desc[i] && *pos == ' ')
2247                         dlen--;
2248         } else {
2249                 dlen = os_strlen(desc[i]);
2250         }
2251         drv->adapter_desc = os_malloc(dlen + 1);
2252         if (drv->adapter_desc) {
2253                 os_memcpy(drv->adapter_desc, desc[i], dlen);
2254                 drv->adapter_desc[dlen] = '\0';
2255         }
2256
2257         os_free(names);
2258
2259         if (drv->adapter_desc == NULL)
2260                 return -1;
2261
2262         wpa_printf(MSG_DEBUG, "NDIS: Adapter description prefix '%s'",
2263                    drv->adapter_desc);
2264
2265         return 0;
2266 #endif /* CONFIG_USE_NDISUIO */
2267 }
2268
2269
2270 #if defined(CONFIG_NATIVE_WINDOWS) || defined(__CYGWIN__)
2271 #ifndef _WIN32_WCE
2272 /*
2273  * These structures are undocumented for WinXP; only WinCE version is
2274  * documented. These would be included wzcsapi.h if it were available. Some
2275  * changes here have been needed to make the structures match with WinXP SP2.
2276  * It is unclear whether these work with any other version.
2277  */
2278
2279 typedef struct {
2280         LPWSTR wszGuid;
2281 } INTF_KEY_ENTRY, *PINTF_KEY_ENTRY;
2282
2283 typedef struct {
2284         DWORD dwNumIntfs;
2285         PINTF_KEY_ENTRY pIntfs;
2286 } INTFS_KEY_TABLE, *PINTFS_KEY_TABLE;
2287
2288 typedef struct {
2289         DWORD dwDataLen;
2290         LPBYTE pData;
2291 } RAW_DATA, *PRAW_DATA;
2292
2293 typedef struct {
2294         LPWSTR wszGuid;
2295         LPWSTR wszDescr;
2296         ULONG ulMediaState;
2297         ULONG ulMediaType;
2298         ULONG ulPhysicalMediaType;
2299         INT nInfraMode;
2300         INT nAuthMode;
2301         INT nWepStatus;
2302 #ifndef _WIN32_WCE
2303         u8 pad[2]; /* why is this needed? */
2304 #endif /* _WIN32_WCE */
2305         DWORD dwCtlFlags;
2306         DWORD dwCapabilities; /* something added for WinXP SP2(?) */
2307         RAW_DATA rdSSID;
2308         RAW_DATA rdBSSID;
2309         RAW_DATA rdBSSIDList;
2310         RAW_DATA rdStSSIDList;
2311         RAW_DATA rdCtrlData;
2312 #ifdef UNDER_CE
2313         BOOL bInitialized;
2314 #endif
2315         DWORD nWPAMCastCipher;
2316         /* add some extra buffer for later additions since this interface is
2317          * far from stable */
2318         u8 later_additions[100];
2319 } INTF_ENTRY, *PINTF_ENTRY;
2320
2321 #define INTF_ALL 0xffffffff
2322 #define INTF_ALL_FLAGS 0x0000ffff
2323 #define INTF_CTLFLAGS 0x00000010
2324 #define INTFCTL_ENABLED 0x8000
2325 #endif /* _WIN32_WCE */
2326
2327
2328 #ifdef _WIN32_WCE
2329 static int wpa_driver_ndis_rebind_adapter(struct wpa_driver_ndis_data *drv)
2330 {
2331         HANDLE ndis;
2332         TCHAR multi[100];
2333         int len;
2334
2335         len = _tcslen(drv->adapter_name);
2336         if (len > 80)
2337                 return -1;
2338
2339         ndis = CreateFile(DD_NDIS_DEVICE_NAME, GENERIC_READ | GENERIC_WRITE,
2340                           0, NULL, OPEN_EXISTING, 0, NULL);
2341         if (ndis == INVALID_HANDLE_VALUE) {
2342                 wpa_printf(MSG_DEBUG, "NDIS: Failed to open file to NDIS "
2343                            "device: %d", (int) GetLastError());
2344                 return -1;
2345         }
2346
2347         len++;
2348         memcpy(multi, drv->adapter_name, len * sizeof(TCHAR));
2349         memcpy(&multi[len], TEXT("NDISUIO\0"), 9 * sizeof(TCHAR));
2350         len += 9;
2351
2352         if (!DeviceIoControl(ndis, IOCTL_NDIS_REBIND_ADAPTER,
2353                              multi, len * sizeof(TCHAR), NULL, 0, NULL, NULL))
2354         {
2355                 wpa_printf(MSG_DEBUG, "NDIS: IOCTL_NDIS_REBIND_ADAPTER "
2356                            "failed: 0x%x", (int) GetLastError());
2357                 wpa_hexdump_ascii(MSG_DEBUG, "NDIS: rebind multi_sz",
2358                                   (u8 *) multi, len * sizeof(TCHAR));
2359                 CloseHandle(ndis);
2360                 return -1;
2361         }
2362
2363         CloseHandle(ndis);
2364
2365         wpa_printf(MSG_DEBUG, "NDIS: Requested NDIS rebind of NDISUIO "
2366                    "protocol");
2367
2368         return 0;
2369 }
2370 #endif /* _WIN32_WCE */
2371
2372
2373 static int wpa_driver_ndis_set_wzc(struct wpa_driver_ndis_data *drv,
2374                                    int enable)
2375 {
2376 #ifdef _WIN32_WCE
2377         HKEY hk, hk2;
2378         LONG ret;
2379         DWORD i, hnd, len;
2380         TCHAR keyname[256], devname[256];
2381
2382 #define WZC_DRIVER TEXT("Drivers\\BuiltIn\\ZeroConfig")
2383
2384         if (enable) {
2385                 HANDLE h;
2386                 h = ActivateDeviceEx(WZC_DRIVER, NULL, 0, NULL);
2387                 if (h == INVALID_HANDLE_VALUE || h == 0) {
2388                         wpa_printf(MSG_DEBUG, "NDIS: Failed to re-enable WZC "
2389                                    "- ActivateDeviceEx failed: %d",
2390                                    (int) GetLastError());
2391                         return -1;
2392                 }
2393
2394                 wpa_printf(MSG_DEBUG, "NDIS: WZC re-enabled");
2395                 return wpa_driver_ndis_rebind_adapter(drv);
2396         }
2397
2398         /*
2399          * Unfortunately, just disabling the WZC for an interface is not enough
2400          * to free NDISUIO for us, so need to disable and unload WZC completely
2401          * for now when using WinCE with NDISUIO. In addition, must request
2402          * NDISUIO protocol to be rebound to the adapter in order to free the
2403          * NDISUIO binding that WZC hold before us.
2404          */
2405
2406         /* Enumerate HKLM\Drivers\Active\* to find a handle to WZC. */
2407         ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, DEVLOAD_ACTIVE_KEY, 0, 0, &hk);
2408         if (ret != ERROR_SUCCESS) {
2409                 wpa_printf(MSG_DEBUG, "NDIS: RegOpenKeyEx(DEVLOAD_ACTIVE_KEY) "
2410                            "failed: %d %d", (int) ret, (int) GetLastError());
2411                 return -1;
2412         }
2413
2414         for (i = 0; ; i++) {
2415                 len = sizeof(keyname);
2416                 ret = RegEnumKeyEx(hk, i, keyname, &len, NULL, NULL, NULL,
2417                                    NULL);
2418                 if (ret != ERROR_SUCCESS) {
2419                         wpa_printf(MSG_DEBUG, "NDIS: Could not find active "
2420                                    "WZC - assuming it is not running.");
2421                         RegCloseKey(hk);
2422                         return -1;
2423                 }
2424
2425                 ret = RegOpenKeyEx(hk, keyname, 0, 0, &hk2);
2426                 if (ret != ERROR_SUCCESS) {
2427                         wpa_printf(MSG_DEBUG, "NDIS: RegOpenKeyEx(active dev) "
2428                                    "failed: %d %d",
2429                                    (int) ret, (int) GetLastError());
2430                         continue;
2431                 }
2432
2433                 len = sizeof(devname);
2434                 ret = RegQueryValueEx(hk2, DEVLOAD_DEVKEY_VALNAME, NULL, NULL,
2435                                       (LPBYTE) devname, &len);
2436                 if (ret != ERROR_SUCCESS) {
2437                         wpa_printf(MSG_DEBUG, "NDIS: RegQueryValueEx("
2438                                    "DEVKEY_VALNAME) failed: %d %d",
2439                                    (int) ret, (int) GetLastError());
2440                         RegCloseKey(hk2);
2441                         continue;
2442                 }
2443
2444                 if (_tcscmp(devname, WZC_DRIVER) == 0)
2445                         break;
2446
2447                 RegCloseKey(hk2);
2448         }
2449
2450         RegCloseKey(hk);
2451
2452         /* Found WZC - get handle to it. */
2453         len = sizeof(hnd);
2454         ret = RegQueryValueEx(hk2, DEVLOAD_HANDLE_VALNAME, NULL, NULL,
2455                               (PUCHAR) &hnd, &len);
2456         if (ret != ERROR_SUCCESS) {
2457                 wpa_printf(MSG_DEBUG, "NDIS: RegQueryValueEx(HANDLE_VALNAME) "
2458                            "failed: %d %d", (int) ret, (int) GetLastError());
2459                 RegCloseKey(hk2);
2460                 return -1;
2461         }
2462
2463         RegCloseKey(hk2);
2464
2465         /* Deactivate WZC */
2466         if (!DeactivateDevice((HANDLE) hnd)) {
2467                 wpa_printf(MSG_DEBUG, "NDIS: DeactivateDevice failed: %d",
2468                            (int) GetLastError());
2469                 return -1;
2470         }
2471
2472         wpa_printf(MSG_DEBUG, "NDIS: Disabled WZC temporarily");
2473         drv->wzc_disabled = 1;
2474         return wpa_driver_ndis_rebind_adapter(drv);
2475
2476 #else /* _WIN32_WCE */
2477
2478         HMODULE hm;
2479         DWORD (WINAPI *wzc_enum_interf)(LPWSTR pSrvAddr,
2480                                         PINTFS_KEY_TABLE pIntfs);
2481         DWORD (WINAPI *wzc_query_interf)(LPWSTR pSrvAddr, DWORD dwInFlags,
2482                                          PINTF_ENTRY pIntf,
2483                                          LPDWORD pdwOutFlags);
2484         DWORD (WINAPI *wzc_set_interf)(LPWSTR pSrvAddr, DWORD dwInFlags,
2485                                        PINTF_ENTRY pIntf, LPDWORD pdwOutFlags);
2486         int ret = -1, j;
2487         DWORD res;
2488         INTFS_KEY_TABLE guids;
2489         INTF_ENTRY intf;
2490         char guid[128];
2491         WCHAR *pos;
2492         DWORD flags, i;
2493
2494         hm = LoadLibrary(TEXT("wzcsapi.dll"));
2495         if (hm == NULL) {
2496                 wpa_printf(MSG_DEBUG, "NDIS: Failed to load wzcsapi.dll (%u) "
2497                            "- WZC probably not running",
2498                            (unsigned int) GetLastError());
2499                 return -1;
2500         }
2501
2502 #ifdef _WIN32_WCE
2503         wzc_enum_interf = (void *) GetProcAddressA(hm, "WZCEnumInterfaces");
2504         wzc_query_interf = (void *) GetProcAddressA(hm, "WZCQueryInterface");
2505         wzc_set_interf = (void *) GetProcAddressA(hm, "WZCSetInterface");
2506 #else /* _WIN32_WCE */
2507         wzc_enum_interf = (void *) GetProcAddress(hm, "WZCEnumInterfaces");
2508         wzc_query_interf = (void *) GetProcAddress(hm, "WZCQueryInterface");
2509         wzc_set_interf = (void *) GetProcAddress(hm, "WZCSetInterface");
2510 #endif /* _WIN32_WCE */
2511
2512         if (wzc_enum_interf == NULL || wzc_query_interf == NULL ||
2513             wzc_set_interf == NULL) {
2514                 wpa_printf(MSG_DEBUG, "NDIS: WZCEnumInterfaces, "
2515                            "WZCQueryInterface, or WZCSetInterface not found "
2516                            "in wzcsapi.dll");
2517                 goto fail;
2518         }
2519
2520         os_memset(&guids, 0, sizeof(guids));
2521         res = wzc_enum_interf(NULL, &guids);
2522         if (res != 0) {
2523                 wpa_printf(MSG_DEBUG, "NDIS: WZCEnumInterfaces failed: %d; "
2524                            "WZC service is apparently not running",
2525                            (int) res);
2526                 goto fail;
2527         }
2528
2529         wpa_printf(MSG_DEBUG, "NDIS: WZCEnumInterfaces: %d interfaces",
2530                    (int) guids.dwNumIntfs);
2531
2532         for (i = 0; i < guids.dwNumIntfs; i++) {
2533                 pos = guids.pIntfs[i].wszGuid;
2534                 for (j = 0; j < sizeof(guid); j++) {
2535                         guid[j] = (char) *pos;
2536                         if (*pos == 0)
2537                                 break;
2538                         pos++;
2539                 }
2540                 guid[sizeof(guid) - 1] = '\0';
2541                 wpa_printf(MSG_DEBUG, "NDIS: intfs %d GUID '%s'",
2542                            (int) i, guid);
2543                 if (os_strstr(drv->ifname, guid) == NULL)
2544                         continue;
2545
2546                 wpa_printf(MSG_DEBUG, "NDIS: Current interface found from "
2547                            "WZC");
2548                 break;
2549         }
2550
2551         if (i >= guids.dwNumIntfs) {
2552                 wpa_printf(MSG_DEBUG, "NDIS: Current interface not found from "
2553                            "WZC");
2554                 goto fail;
2555         }
2556
2557         os_memset(&intf, 0, sizeof(intf));
2558         intf.wszGuid = guids.pIntfs[i].wszGuid;
2559         /* Set flags to verify that the structure has not changed. */
2560         intf.dwCtlFlags = -1;
2561         flags = 0;
2562         res = wzc_query_interf(NULL, INTFCTL_ENABLED, &intf, &flags);
2563         if (res != 0) {
2564                 wpa_printf(MSG_DEBUG, "NDIS: Could not query flags for the "
2565                            "WZC interface: %d (0x%x)",
2566                            (int) res, (int) res);
2567                 wpa_printf(MSG_DEBUG, "NDIS: GetLastError: %u",
2568                            (unsigned int) GetLastError());
2569                 goto fail;
2570         }
2571
2572         wpa_printf(MSG_DEBUG, "NDIS: WZC interface flags 0x%x dwCtlFlags 0x%x",
2573                    (int) flags, (int) intf.dwCtlFlags);
2574
2575         if (intf.dwCtlFlags == -1) {
2576                 wpa_printf(MSG_DEBUG, "NDIS: Looks like wzcsapi has changed "
2577                            "again - could not disable WZC");
2578                 wpa_hexdump(MSG_MSGDUMP, "NDIS: intf",
2579                             (u8 *) &intf, sizeof(intf));
2580                 goto fail;
2581         }
2582
2583         if (enable) {
2584                 if (!(intf.dwCtlFlags & INTFCTL_ENABLED)) {
2585                         wpa_printf(MSG_DEBUG, "NDIS: Enabling WZC for this "
2586                                    "interface");
2587                         intf.dwCtlFlags |= INTFCTL_ENABLED;
2588                         res = wzc_set_interf(NULL, INTFCTL_ENABLED, &intf,
2589                                              &flags);
2590                         if (res != 0) {
2591                                 wpa_printf(MSG_DEBUG, "NDIS: Failed to enable "
2592                                            "WZC: %d (0x%x)",
2593                                            (int) res, (int) res);
2594                                 wpa_printf(MSG_DEBUG, "NDIS: GetLastError: %u",
2595                                            (unsigned int) GetLastError());
2596                                 goto fail;
2597                         }
2598                         wpa_printf(MSG_DEBUG, "NDIS: Re-enabled WZC for this "
2599                                    "interface");
2600                         drv->wzc_disabled = 0;
2601                 }
2602         } else {
2603                 if (intf.dwCtlFlags & INTFCTL_ENABLED) {
2604                         wpa_printf(MSG_DEBUG, "NDIS: Disabling WZC for this "
2605                                    "interface");
2606                         intf.dwCtlFlags &= ~INTFCTL_ENABLED;
2607                         res = wzc_set_interf(NULL, INTFCTL_ENABLED, &intf,
2608                                              &flags);
2609                         if (res != 0) {
2610                                 wpa_printf(MSG_DEBUG, "NDIS: Failed to "
2611                                            "disable WZC: %d (0x%x)",
2612                                            (int) res, (int) res);
2613                                 wpa_printf(MSG_DEBUG, "NDIS: GetLastError: %u",
2614                                            (unsigned int) GetLastError());
2615                                 goto fail;
2616                         }
2617                         wpa_printf(MSG_DEBUG, "NDIS: Disabled WZC temporarily "
2618                                    "for this interface");
2619                         drv->wzc_disabled = 1;
2620                 } else {
2621                         wpa_printf(MSG_DEBUG, "NDIS: WZC was not enabled for "
2622                                    "this interface");
2623                 }
2624         }
2625
2626         ret = 0;
2627
2628 fail:
2629         FreeLibrary(hm);
2630
2631         return ret;
2632 #endif /* _WIN32_WCE */
2633 }
2634
2635 #else /* CONFIG_NATIVE_WINDOWS || __CYGWIN__ */
2636
2637 static int wpa_driver_ndis_set_wzc(struct wpa_driver_ndis_data *drv,
2638                                    int enable)
2639 {
2640         return 0;
2641 }
2642
2643 #endif /* CONFIG_NATIVE_WINDOWS || __CYGWIN__ */
2644
2645
2646 #ifdef CONFIG_USE_NDISUIO
2647 /*
2648  * l2_packet_ndis.c is sharing the same handle to NDISUIO, so we must be able
2649  * to export this handle. This is somewhat ugly, but there is no better
2650  * mechanism available to pass data from driver interface to l2_packet wrapper.
2651  */
2652 static HANDLE driver_ndis_ndisuio_handle = INVALID_HANDLE_VALUE;
2653
2654 HANDLE driver_ndis_get_ndisuio_handle(void)
2655 {
2656         return driver_ndis_ndisuio_handle;
2657 }
2658 #endif /* CONFIG_USE_NDISUIO */
2659
2660
2661 static int wpa_driver_ndis_adapter_init(struct wpa_driver_ndis_data *drv)
2662 {
2663 #ifdef CONFIG_USE_NDISUIO
2664 #ifndef _WIN32_WCE
2665 #define NDISUIO_DEVICE_NAME TEXT("\\\\.\\\\Ndisuio")
2666         DWORD written;
2667 #endif /* _WIN32_WCE */
2668         drv->ndisuio = CreateFile(NDISUIO_DEVICE_NAME,
2669                                   GENERIC_READ | GENERIC_WRITE, 0, NULL,
2670                                   OPEN_EXISTING,
2671                                   FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED,
2672                                   INVALID_HANDLE_VALUE);
2673         if (drv->ndisuio == INVALID_HANDLE_VALUE) {
2674                 wpa_printf(MSG_ERROR, "NDIS: Failed to open connection to "
2675                            "NDISUIO: %d", (int) GetLastError());
2676                 return -1;
2677         }
2678         driver_ndis_ndisuio_handle = drv->ndisuio;
2679
2680 #ifndef _WIN32_WCE
2681         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_BIND_WAIT, NULL, 0,
2682                              NULL, 0, &written, NULL)) {
2683                 wpa_printf(MSG_ERROR, "NDIS: IOCTL_NDISUIO_BIND_WAIT failed: "
2684                            "%d", (int) GetLastError());
2685                 CloseHandle(drv->ndisuio);
2686                 drv->ndisuio = INVALID_HANDLE_VALUE;
2687                 return -1;
2688         }
2689 #endif /* _WIN32_WCE */
2690
2691         return 0;
2692 #else /* CONFIG_USE_NDISUIO */
2693         return 0;
2694 #endif /* CONFIG_USE_NDISUIO */
2695 }
2696
2697
2698 static int wpa_driver_ndis_adapter_open(struct wpa_driver_ndis_data *drv)
2699 {
2700 #ifdef CONFIG_USE_NDISUIO
2701         DWORD written;
2702 #define MAX_NDIS_DEVICE_NAME_LEN 256
2703         WCHAR ifname[MAX_NDIS_DEVICE_NAME_LEN];
2704         size_t len, i, pos;
2705         const char *prefix = "\\DEVICE\\";
2706
2707 #ifdef _WIN32_WCE
2708         pos = 0;
2709 #else /* _WIN32_WCE */
2710         pos = 8;
2711 #endif /* _WIN32_WCE */
2712         len = pos + os_strlen(drv->ifname);
2713         if (len >= MAX_NDIS_DEVICE_NAME_LEN)
2714                 return -1;
2715         for (i = 0; i < pos; i++)
2716                 ifname[i] = (WCHAR) prefix[i];
2717         for (i = pos; i < len; i++)
2718                 ifname[i] = (WCHAR) drv->ifname[i - pos];
2719         ifname[i] = L'\0';
2720
2721         if (!DeviceIoControl(drv->ndisuio, IOCTL_NDISUIO_OPEN_DEVICE,
2722                              ifname, len * sizeof(WCHAR), NULL, 0, &written,
2723                              NULL)) {
2724                 wpa_printf(MSG_ERROR, "NDIS: IOCTL_NDISUIO_OPEN_DEVICE "
2725                            "failed: %d", (int) GetLastError());
2726                 wpa_hexdump_ascii(MSG_DEBUG, "NDIS: ifname",
2727                                   (const u8 *) ifname, len * sizeof(WCHAR));
2728                 CloseHandle(drv->ndisuio);
2729                 drv->ndisuio = INVALID_HANDLE_VALUE;
2730                 return -1;
2731         }
2732
2733         wpa_printf(MSG_DEBUG, "NDIS: Opened NDISUIO device successfully");
2734
2735         return 0;
2736 #else /* CONFIG_USE_NDISUIO */
2737         char ifname[128];
2738         os_snprintf(ifname, sizeof(ifname), "\\Device\\NPF_%s", drv->ifname);
2739         drv->adapter = PacketOpenAdapter(ifname);
2740         if (drv->adapter == NULL) {
2741                 wpa_printf(MSG_DEBUG, "NDIS: PacketOpenAdapter failed for "
2742                            "'%s'", ifname);
2743                 return -1;
2744         }
2745         return 0;
2746 #endif /* CONFIG_USE_NDISUIO */
2747 }
2748
2749
2750 static void wpa_driver_ndis_adapter_close(struct wpa_driver_ndis_data *drv)
2751 {
2752 #ifdef CONFIG_USE_NDISUIO
2753         driver_ndis_ndisuio_handle = INVALID_HANDLE_VALUE;
2754         if (drv->ndisuio != INVALID_HANDLE_VALUE)
2755                 CloseHandle(drv->ndisuio);
2756 #else /* CONFIG_USE_NDISUIO */
2757         if (drv->adapter)
2758                 PacketCloseAdapter(drv->adapter);
2759 #endif /* CONFIG_USE_NDISUIO */
2760 }
2761
2762
2763 static int ndis_add_multicast(struct wpa_driver_ndis_data *drv)
2764 {
2765         if (ndis_set_oid(drv, OID_802_3_MULTICAST_LIST,
2766                          (const char *) pae_group_addr, ETH_ALEN) < 0) {
2767                 wpa_printf(MSG_DEBUG, "NDIS: Failed to add PAE group address "
2768                            "to the multicast list");
2769                 return -1;
2770         }
2771
2772         return 0;
2773 }
2774
2775
2776 static void * wpa_driver_ndis_init(void *ctx, const char *ifname)
2777 {
2778         struct wpa_driver_ndis_data *drv;
2779         u32 mode;
2780
2781         drv = os_zalloc(sizeof(*drv));
2782         if (drv == NULL)
2783                 return NULL;
2784         drv->ctx = ctx;
2785         /*
2786          * Compatibility code to strip possible prefix from the GUID. Previous
2787          * versions include \Device\NPF_ prefix for all names, but the internal
2788          * interface name is now only the GUI. Both Packet32 and NDISUIO
2789          * prefixes are supported.
2790          */
2791         if (os_strncmp(ifname, "\\Device\\NPF_", 12) == 0)
2792                 ifname += 12;
2793         else if (os_strncmp(ifname, "\\DEVICE\\", 8) == 0)
2794                 ifname += 8;
2795         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
2796
2797         if (wpa_driver_ndis_adapter_init(drv) < 0) {
2798                 os_free(drv);
2799                 return NULL;
2800         }
2801
2802         if (wpa_driver_ndis_get_names(drv) < 0) {
2803                 wpa_driver_ndis_adapter_close(drv);
2804                 os_free(drv);
2805                 return NULL;
2806         }
2807
2808         wpa_driver_ndis_set_wzc(drv, 0);
2809
2810         if (wpa_driver_ndis_adapter_open(drv) < 0) {
2811                 wpa_driver_ndis_adapter_close(drv);
2812                 os_free(drv);
2813                 return NULL;
2814         }
2815
2816         if (ndis_get_oid(drv, OID_802_3_CURRENT_ADDRESS,
2817                          (char *) drv->own_addr, ETH_ALEN) < 0) {
2818                 wpa_printf(MSG_DEBUG, "NDIS: Get OID_802_3_CURRENT_ADDRESS "
2819                            "failed");
2820                 wpa_driver_ndis_adapter_close(drv);
2821                 os_free(drv);
2822                 return NULL;
2823         }
2824         wpa_driver_ndis_get_capability(drv);
2825
2826         /* Make sure that the driver does not have any obsolete PMKID entries.
2827          */
2828         wpa_driver_ndis_flush_pmkid(drv);
2829
2830         /*
2831          * Disconnect to make sure that driver re-associates if it was
2832          * connected.
2833          */
2834         wpa_driver_ndis_disconnect(drv);
2835
2836         eloop_register_timeout(1, 0, wpa_driver_ndis_poll_timeout, drv, NULL);
2837
2838 #ifdef CONFIG_NDIS_EVENTS_INTEGRATED
2839         drv->events = ndis_events_init(&drv->events_pipe, &drv->event_avail,
2840                                        drv->ifname, drv->adapter_desc);
2841         if (drv->events == NULL) {
2842                 wpa_driver_ndis_deinit(drv);
2843                 return NULL;
2844         }
2845         eloop_register_event(drv->event_avail, sizeof(drv->event_avail),
2846                              wpa_driver_ndis_event_pipe_cb, drv, NULL);
2847 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
2848
2849 #ifdef _WIN32_WCE
2850         if (ndisuio_notification_init(drv) < 0) {
2851                 wpa_driver_ndis_deinit(drv);
2852                 return NULL;
2853         }
2854 #endif /* _WIN32_WCE */
2855
2856         /* Set mode here in case card was configured for ad-hoc mode
2857          * previously. */
2858         mode = Ndis802_11Infrastructure;
2859         if (ndis_set_oid(drv, OID_802_11_INFRASTRUCTURE_MODE,
2860                          (char *) &mode, sizeof(mode)) < 0) {
2861                 char buf[8];
2862                 int res;
2863                 wpa_printf(MSG_DEBUG, "NDIS: Failed to set "
2864                            "OID_802_11_INFRASTRUCTURE_MODE (%d)",
2865                            (int) mode);
2866                 /* Try to continue anyway */
2867
2868                 res = ndis_get_oid(drv, OID_DOT11_CURRENT_OPERATION_MODE, buf,
2869                                    sizeof(buf));
2870                 if (res > 0) {
2871                         wpa_printf(MSG_INFO, "NDIS: The driver seems to use "
2872                                    "Native 802.11 OIDs. These are not yet "
2873                                    "fully supported.");
2874                         drv->native80211 = 1;
2875                 } else if (!drv->has_capability || drv->capa.enc == 0) {
2876                         /*
2877                          * Note: This will also happen with NDIS 6 drivers with
2878                          * Vista.
2879                          */
2880                         wpa_printf(MSG_DEBUG, "NDIS: Driver did not provide "
2881                                    "any wireless capabilities - assume it is "
2882                                    "a wired interface");
2883                         drv->wired = 1;
2884                         drv->capa.flags |= WPA_DRIVER_FLAGS_WIRED;
2885                         drv->has_capability = 1;
2886                         ndis_add_multicast(drv);
2887                 }
2888         }
2889
2890         return drv;
2891 }
2892
2893
2894 static void wpa_driver_ndis_deinit(void *priv)
2895 {
2896         struct wpa_driver_ndis_data *drv = priv;
2897
2898 #ifdef CONFIG_NDIS_EVENTS_INTEGRATED
2899         if (drv->events) {
2900                 eloop_unregister_event(drv->event_avail,
2901                                        sizeof(drv->event_avail));
2902                 ndis_events_deinit(drv->events);
2903         }
2904 #endif /* CONFIG_NDIS_EVENTS_INTEGRATED */
2905
2906 #ifdef _WIN32_WCE
2907         ndisuio_notification_deinit(drv);
2908 #endif /* _WIN32_WCE */
2909
2910         eloop_cancel_timeout(wpa_driver_ndis_scan_timeout, drv, drv->ctx);
2911         eloop_cancel_timeout(wpa_driver_ndis_poll_timeout, drv, NULL);
2912         wpa_driver_ndis_flush_pmkid(drv);
2913         wpa_driver_ndis_disconnect(drv);
2914         if (wpa_driver_ndis_radio_off(drv) < 0) {
2915                 wpa_printf(MSG_DEBUG, "NDIS: failed to disassociate and turn "
2916                            "radio off");
2917         }
2918
2919         wpa_driver_ndis_adapter_close(drv);
2920
2921         if (drv->wzc_disabled)
2922                 wpa_driver_ndis_set_wzc(drv, 1);
2923
2924 #ifdef _WIN32_WCE
2925         os_free(drv->adapter_name);
2926 #endif /* _WIN32_WCE */
2927         os_free(drv->adapter_desc);
2928         os_free(drv);
2929 }
2930
2931
2932 static struct wpa_interface_info *
2933 wpa_driver_ndis_get_interfaces(void *global_priv)
2934 {
2935         struct wpa_interface_info *iface = NULL, *niface;
2936
2937 #ifdef CONFIG_USE_NDISUIO
2938         NDISUIO_QUERY_BINDING *b;
2939         size_t blen = sizeof(*b) + 1024;
2940         int i, error;
2941         DWORD written;
2942         char name[256], desc[256];
2943         WCHAR *pos;
2944         size_t j, len;
2945         HANDLE ndisuio;
2946
2947         ndisuio = CreateFile(NDISUIO_DEVICE_NAME,
2948                              GENERIC_READ | GENERIC_WRITE, 0, NULL,
2949                              OPEN_EXISTING,
2950                              FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED,
2951                              INVALID_HANDLE_VALUE);
2952         if (ndisuio == INVALID_HANDLE_VALUE) {
2953                 wpa_printf(MSG_ERROR, "NDIS: Failed to open connection to "
2954                            "NDISUIO: %d", (int) GetLastError());
2955                 return NULL;
2956         }
2957
2958 #ifndef _WIN32_WCE
2959         if (!DeviceIoControl(ndisuio, IOCTL_NDISUIO_BIND_WAIT, NULL, 0,
2960                              NULL, 0, &written, NULL)) {
2961                 wpa_printf(MSG_ERROR, "NDIS: IOCTL_NDISUIO_BIND_WAIT failed: "
2962                            "%d", (int) GetLastError());
2963                 CloseHandle(ndisuio);
2964                 return NULL;
2965         }
2966 #endif /* _WIN32_WCE */
2967
2968         b = os_malloc(blen);
2969         if (b == NULL) {
2970                 CloseHandle(ndisuio);
2971                 return NULL;
2972         }
2973
2974         for (i = 0; ; i++) {
2975                 os_memset(b, 0, blen);
2976                 b->BindingIndex = i;
2977                 if (!DeviceIoControl(ndisuio, IOCTL_NDISUIO_QUERY_BINDING,
2978                                      b, sizeof(NDISUIO_QUERY_BINDING), b, blen,
2979                                      &written, NULL)) {
2980                         error = (int) GetLastError();
2981                         if (error == ERROR_NO_MORE_ITEMS)
2982                                 break;
2983                         wpa_printf(MSG_DEBUG, "IOCTL_NDISUIO_QUERY_BINDING "
2984                                    "failed: %d", error);
2985                         break;
2986                 }
2987
2988                 pos = (WCHAR *) ((char *) b + b->DeviceNameOffset);
2989                 len = b->DeviceNameLength;
2990                 if (len >= sizeof(name))
2991                         len = sizeof(name) - 1;
2992                 for (j = 0; j < len; j++)
2993                         name[j] = (char) pos[j];
2994                 name[len] = '\0';
2995
2996                 pos = (WCHAR *) ((char *) b + b->DeviceDescrOffset);
2997                 len = b->DeviceDescrLength;
2998                 if (len >= sizeof(desc))
2999                         len = sizeof(desc) - 1;
3000                 for (j = 0; j < len; j++)
3001                         desc[j] = (char) pos[j];
3002                 desc[len] = '\0';
3003
3004                 wpa_printf(MSG_DEBUG, "NDIS: %d - %s - %s", i, name, desc);
3005
3006                 niface = os_zalloc(sizeof(*niface));
3007                 if (niface == NULL)
3008                         break;
3009                 niface->drv_name = "ndis";
3010                 if (os_strncmp(name, "\\DEVICE\\", 8) == 0)
3011                         niface->ifname = os_strdup(name + 8);
3012                 else
3013                         niface->ifname = os_strdup(name);
3014                 if (niface->ifname == NULL) {
3015                         os_free(niface);
3016                         break;
3017                 }
3018                 niface->desc = os_strdup(desc);
3019                 niface->next = iface;
3020                 iface = niface;
3021         }
3022
3023         os_free(b);
3024         CloseHandle(ndisuio);
3025 #else /* CONFIG_USE_NDISUIO */
3026         PTSTR _names;
3027         char *names, *pos, *pos2;
3028         ULONG len;
3029         BOOLEAN res;
3030         char *name[MAX_ADAPTERS];
3031         char *desc[MAX_ADAPTERS];
3032         int num_name, num_desc, i;
3033
3034         wpa_printf(MSG_DEBUG, "NDIS: Packet.dll version: %s",
3035                    PacketGetVersion());
3036
3037         len = 8192;
3038         _names = os_zalloc(len);
3039         if (_names == NULL)
3040                 return NULL;
3041
3042         res = PacketGetAdapterNames(_names, &len);
3043         if (!res && len > 8192) {
3044                 os_free(_names);
3045                 _names = os_zalloc(len);
3046                 if (_names == NULL)
3047                         return NULL;
3048                 res = PacketGetAdapterNames(_names, &len);
3049         }
3050
3051         if (!res) {
3052                 wpa_printf(MSG_ERROR, "NDIS: Failed to get adapter list "
3053                            "(PacketGetAdapterNames)");
3054                 os_free(_names);
3055                 return NULL;
3056         }
3057
3058         names = (char *) _names;
3059         if (names[0] && names[1] == '\0' && names[2] && names[3] == '\0') {
3060                 wpa_printf(MSG_DEBUG, "NDIS: Looks like adapter names are in "
3061                            "UNICODE");
3062                 /* Convert to ASCII */
3063                 pos2 = pos = names;
3064                 while (pos2 < names + len) {
3065                         if (pos2[0] == '\0' && pos2[1] == '\0' &&
3066                             pos2[2] == '\0' && pos2[3] == '\0') {
3067                                 pos2 += 4;
3068                                 break;
3069                         }
3070                         *pos++ = pos2[0];
3071                         pos2 += 2;
3072                 }
3073                 os_memcpy(pos + 2, names, pos - names);
3074                 pos += 2;
3075         } else
3076                 pos = names;
3077
3078         num_name = 0;
3079         while (pos < names + len) {
3080                 name[num_name] = pos;
3081                 while (*pos && pos < names + len)
3082                         pos++;
3083                 if (pos + 1 >= names + len) {
3084                         os_free(names);
3085                         return NULL;
3086                 }
3087                 pos++;
3088                 num_name++;
3089                 if (num_name >= MAX_ADAPTERS) {
3090                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapters");
3091                         os_free(names);
3092                         return NULL;
3093                 }
3094                 if (*pos == '\0') {
3095                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter names found",
3096                                    num_name);
3097                         pos++;
3098                         break;
3099                 }
3100         }
3101
3102         num_desc = 0;
3103         while (pos < names + len) {
3104                 desc[num_desc] = pos;
3105                 while (*pos && pos < names + len)
3106                         pos++;
3107                 if (pos + 1 >= names + len) {
3108                         os_free(names);
3109                         return NULL;
3110                 }
3111                 pos++;
3112                 num_desc++;
3113                 if (num_desc >= MAX_ADAPTERS) {
3114                         wpa_printf(MSG_DEBUG, "NDIS: Too many adapter "
3115                                    "descriptions");
3116                         os_free(names);
3117                         return NULL;
3118                 }
3119                 if (*pos == '\0') {
3120                         wpa_printf(MSG_DEBUG, "NDIS: %d adapter descriptions "
3121                                    "found", num_name);
3122                         pos++;
3123                         break;
3124                 }
3125         }
3126
3127         /*
3128          * Windows 98 with Packet.dll 3.0 alpha3 does not include adapter
3129          * descriptions. Fill in dummy descriptors to work around this.
3130          */
3131         while (num_desc < num_name)
3132                 desc[num_desc++] = "dummy description";
3133
3134         if (num_name != num_desc) {
3135                 wpa_printf(MSG_DEBUG, "NDIS: mismatch in adapter name and "
3136                            "description counts (%d != %d)",
3137                            num_name, num_desc);
3138                 os_free(names);
3139                 return NULL;
3140         }
3141
3142         for (i = 0; i < num_name; i++) {
3143                 niface = os_zalloc(sizeof(*niface));
3144                 if (niface == NULL)
3145                         break;
3146                 niface->drv_name = "ndis";
3147                 if (os_strncmp(name[i], "\\Device\\NPF_", 12) == 0)
3148                         niface->ifname = os_strdup(name[i] + 12);
3149                 else
3150                         niface->ifname = os_strdup(name[i]);
3151                 if (niface->ifname == NULL) {
3152                         os_free(niface);
3153                         break;
3154                 }
3155                 niface->desc = os_strdup(desc[i]);
3156                 niface->next = iface;
3157                 iface = niface;
3158         }
3159
3160 #endif /* CONFIG_USE_NDISUIO */
3161
3162         return iface;
3163 }
3164
3165
3166 const struct wpa_driver_ops wpa_driver_ndis_ops = {
3167         "ndis",
3168         "Windows NDIS driver",
3169         wpa_driver_ndis_get_bssid,
3170         wpa_driver_ndis_get_ssid,
3171         wpa_driver_ndis_set_key,
3172         wpa_driver_ndis_init,
3173         wpa_driver_ndis_deinit,
3174         NULL /* set_param */,
3175         NULL /* set_countermeasures */,
3176         wpa_driver_ndis_scan,
3177         NULL /* get_scan_results */,
3178         wpa_driver_ndis_deauthenticate,
3179         wpa_driver_ndis_disassociate,
3180         wpa_driver_ndis_associate,
3181         wpa_driver_ndis_add_pmkid,
3182         wpa_driver_ndis_remove_pmkid,
3183         wpa_driver_ndis_flush_pmkid,
3184         wpa_driver_ndis_get_capa,
3185         wpa_driver_ndis_poll,
3186         wpa_driver_ndis_get_ifname,
3187         wpa_driver_ndis_get_mac_addr,
3188         NULL /* send_eapol */,
3189         NULL /* set_operstate */,
3190         NULL /* mlme_setprotection */,
3191         NULL /* get_hw_feature_data */,
3192         NULL /* set_channel */,
3193         NULL /* set_ssid */,
3194         NULL /* set_bssid */,
3195         NULL /* send_mlme */,
3196         NULL /* mlme_add_sta */,
3197         NULL /* mlme_remove_sta */,
3198         NULL /* update_ft_ies */,
3199         NULL /* send_ft_action */,
3200         wpa_driver_ndis_get_scan_results,
3201         NULL /* set_probe_req_ie */,
3202         NULL /* set_country */,
3203         NULL /* global_init */,
3204         NULL /* global_deinit */,
3205         NULL /* init2 */,
3206         wpa_driver_ndis_get_interfaces,
3207         NULL /* scan2 */,
3208         NULL /* authenticate */,
3209         NULL /* set_beacon */,
3210         NULL /* hapd_init */,
3211         NULL /* hapd_deinit */,
3212         NULL /* set_ieee8021x */,
3213         NULL /* set_privacy */,
3214         NULL /* get_seqnum */,
3215         NULL /* get_seqnum_igtk */,
3216         NULL /* flush */,
3217         NULL /* set_generic_elem */,
3218         NULL /* read_sta_data */,
3219         NULL /* hapd_send_eapol */,
3220         NULL /* sta_deauth */,
3221         NULL /* sta_disassoc */,
3222         NULL /* sta_remove */,
3223         NULL /* hapd_get_ssid */,
3224         NULL /* hapd_set_ssid */,
3225         NULL /* hapd_set_countermeasures */,
3226         NULL /* sta_add */,
3227         NULL /* get_inact_sec */,
3228         NULL /* sta_clear_stats */,
3229         NULL /* set_freq */,
3230         NULL /* set_rts */,
3231         NULL /* set_frag */,
3232         NULL /* sta_set_flags */,
3233         NULL /* set_rate_sets */,
3234         NULL /* set_internal_bridge */,
3235         NULL /* set_cts_protect */,
3236         NULL /* set_preamble */,
3237         NULL /* set_short_slot_time */,
3238         NULL /* set_tx_queue_params */,
3239         NULL /* bss_add */,
3240         NULL /* bss_remove */,
3241         NULL /* valid_bss_mask */,
3242         NULL /* if_add */,
3243         NULL /* if_update */,
3244         NULL /* if_remove */,
3245         NULL /* set_sta_vlan */,
3246         NULL /* commit */,
3247         NULL /* send_ether */,
3248         NULL /* set_radius_acl_auth */,
3249         NULL /* set_radius_acl_expire */,
3250         NULL /* set_ht_params */,
3251         NULL /* set_wps_beacon_ie */,
3252         NULL /* set_wps_probe_resp_ie */,
3253         NULL /* set_supp_port */
3254 };