Allow AP mode to disconnect STAs based on low ACK condition
[mech_eap.git] / wpa_supplicant / config_winreg.c
1 /*
2  * WPA Supplicant / Configuration backend: Windows registry
3  * Copyright (c) 2003-2008, 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  * This file implements a configuration backend for Windows registry. All the
15  * configuration information is stored in the registry and the format for
16  * network configuration fields is same as described in the sample
17  * configuration file, wpa_supplicant.conf.
18  *
19  * Configuration data is in
20  * \a HKEY_LOCAL_MACHINE\\SOFTWARE\\%wpa_supplicant\\configs
21  * key. Each configuration profile has its own key under this. In terms of text
22  * files, each profile would map to a separate text file with possibly multiple
23  * networks. Under each profile, there is a networks key that lists all
24  * networks as a subkey. Each network has set of values in the same way as
25  * network block in the configuration file. In addition, blobs subkey has
26  * possible blobs as values.
27  *
28  * Example network configuration block:
29  * \verbatim
30 HKEY_LOCAL_MACHINE\SOFTWARE\wpa_supplicant\configs\test\networks\0000
31    ssid="example"
32    key_mgmt=WPA-PSK
33 \endverbatim
34  */
35
36 #include "includes.h"
37
38 #include "common.h"
39 #include "uuid.h"
40 #include "config.h"
41
42 #ifndef WPA_KEY_ROOT
43 #define WPA_KEY_ROOT HKEY_LOCAL_MACHINE
44 #endif
45 #ifndef WPA_KEY_PREFIX
46 #define WPA_KEY_PREFIX TEXT("SOFTWARE\\wpa_supplicant")
47 #endif
48
49 #ifdef UNICODE
50 #define TSTR "%S"
51 #else /* UNICODE */
52 #define TSTR "%s"
53 #endif /* UNICODE */
54
55
56 static int wpa_config_read_blobs(struct wpa_config *config, HKEY hk)
57 {
58         struct wpa_config_blob *blob;
59         int errors = 0;
60         HKEY bhk;
61         LONG ret;
62         DWORD i;
63
64         ret = RegOpenKeyEx(hk, TEXT("blobs"), 0, KEY_QUERY_VALUE, &bhk);
65         if (ret != ERROR_SUCCESS) {
66                 wpa_printf(MSG_DEBUG, "Could not open wpa_supplicant config "
67                            "blobs key");
68                 return 0; /* assume no blobs */
69         }
70
71         for (i = 0; ; i++) {
72 #define TNAMELEN 255
73                 TCHAR name[TNAMELEN];
74                 char data[4096];
75                 DWORD namelen, datalen, type;
76
77                 namelen = TNAMELEN;
78                 datalen = sizeof(data);
79                 ret = RegEnumValue(bhk, i, name, &namelen, NULL, &type,
80                                    (LPBYTE) data, &datalen);
81
82                 if (ret == ERROR_NO_MORE_ITEMS)
83                         break;
84
85                 if (ret != ERROR_SUCCESS) {
86                         wpa_printf(MSG_DEBUG, "RegEnumValue failed: 0x%x",
87                                    (unsigned int) ret);
88                         break;
89                 }
90
91                 if (namelen >= TNAMELEN)
92                         namelen = TNAMELEN - 1;
93                 name[namelen] = TEXT('\0');
94                 wpa_unicode2ascii_inplace(name);
95
96                 if (datalen >= sizeof(data))
97                         datalen = sizeof(data) - 1;
98
99                 wpa_printf(MSG_MSGDUMP, "blob %d: field='%s' len %d",
100                            (int) i, name, (int) datalen);
101
102                 blob = os_zalloc(sizeof(*blob));
103                 if (blob == NULL) {
104                         errors++;
105                         break;
106                 }
107                 blob->name = os_strdup((char *) name);
108                 blob->data = os_malloc(datalen);
109                 if (blob->name == NULL || blob->data == NULL) {
110                         wpa_config_free_blob(blob);
111                         errors++;
112                         break;
113                 }
114                 os_memcpy(blob->data, data, datalen);
115                 blob->len = datalen;
116
117                 wpa_config_set_blob(config, blob);
118         }
119
120         RegCloseKey(bhk);
121
122         return errors ? -1 : 0;
123 }
124
125
126 static int wpa_config_read_reg_dword(HKEY hk, const TCHAR *name, int *_val)
127 {
128         DWORD val, buflen;
129         LONG ret;
130
131         buflen = sizeof(val);
132         ret = RegQueryValueEx(hk, name, NULL, NULL, (LPBYTE) &val, &buflen);
133         if (ret == ERROR_SUCCESS && buflen == sizeof(val)) {
134                 wpa_printf(MSG_DEBUG, TSTR "=%d", name, (int) val);
135                 *_val = val;
136                 return 0;
137         }
138
139         return -1;
140 }
141
142
143 static char * wpa_config_read_reg_string(HKEY hk, const TCHAR *name)
144 {
145         DWORD buflen;
146         LONG ret;
147         TCHAR *val;
148
149         buflen = 0;
150         ret = RegQueryValueEx(hk, name, NULL, NULL, NULL, &buflen);
151         if (ret != ERROR_SUCCESS)
152                 return NULL;
153         val = os_malloc(buflen);
154         if (val == NULL)
155                 return NULL;
156
157         ret = RegQueryValueEx(hk, name, NULL, NULL, (LPBYTE) val, &buflen);
158         if (ret != ERROR_SUCCESS) {
159                 os_free(val);
160                 return NULL;
161         }
162
163         wpa_unicode2ascii_inplace(val);
164         wpa_printf(MSG_DEBUG, TSTR "=%s", name, (char *) val);
165         return (char *) val;
166 }
167
168
169 #ifdef CONFIG_WPS
170 static int wpa_config_read_global_uuid(struct wpa_config *config, HKEY hk)
171 {
172         char *str;
173         int ret = 0;
174
175         str = wpa_config_read_reg_string(hk, TEXT("uuid"));
176         if (str == NULL)
177                 return 0;
178
179         if (uuid_str2bin(str, config->uuid))
180                 ret = -1;
181
182         os_free(str);
183
184         return ret;
185 }
186
187
188 static int wpa_config_read_global_os_version(struct wpa_config *config,
189                                              HKEY hk)
190 {
191         char *str;
192         int ret = 0;
193
194         str = wpa_config_read_reg_string(hk, TEXT("os_version"));
195         if (str == NULL)
196                 return 0;
197
198         if (hexstr2bin(str, config->os_version, 4))
199                 ret = -1;
200
201         os_free(str);
202
203         return ret;
204 }
205 #endif /* CONFIG_WPS */
206
207
208 static int wpa_config_read_global(struct wpa_config *config, HKEY hk)
209 {
210         int errors = 0;
211
212         wpa_config_read_reg_dword(hk, TEXT("ap_scan"), &config->ap_scan);
213         wpa_config_read_reg_dword(hk, TEXT("fast_reauth"),
214                                   &config->fast_reauth);
215         wpa_config_read_reg_dword(hk, TEXT("dot11RSNAConfigPMKLifetime"),
216                                   (int *) &config->dot11RSNAConfigPMKLifetime);
217         wpa_config_read_reg_dword(hk,
218                                   TEXT("dot11RSNAConfigPMKReauthThreshold"),
219                                   (int *)
220                                   &config->dot11RSNAConfigPMKReauthThreshold);
221         wpa_config_read_reg_dword(hk, TEXT("dot11RSNAConfigSATimeout"),
222                                   (int *) &config->dot11RSNAConfigSATimeout);
223         wpa_config_read_reg_dword(hk, TEXT("update_config"),
224                                   &config->update_config);
225
226         if (wpa_config_read_reg_dword(hk, TEXT("eapol_version"),
227                                       &config->eapol_version) == 0) {
228                 if (config->eapol_version < 1 ||
229                     config->eapol_version > 2) {
230                         wpa_printf(MSG_ERROR, "Invalid EAPOL version (%d)",
231                                    config->eapol_version);
232                         errors++;
233                 }
234         }
235
236         config->ctrl_interface = wpa_config_read_reg_string(
237                 hk, TEXT("ctrl_interface"));
238
239 #ifdef CONFIG_WPS
240         if (wpa_config_read_global_uuid(config, hk))
241                 errors++;
242         config->device_name = wpa_config_read_reg_string(
243                 hk, TEXT("device_name"));
244         config->manufacturer = wpa_config_read_reg_string(
245                 hk, TEXT("manufacturer"));
246         config->model_name = wpa_config_read_reg_string(
247                 hk, TEXT("model_name"));
248         config->serial_number = wpa_config_read_reg_string(
249                 hk, TEXT("serial_number"));
250         config->device_type = wpa_config_read_reg_string(
251                 hk, TEXT("device_type"));
252         config->config_methods = wpa_config_read_reg_string(
253                 hk, TEXT("config_methods"));
254         if (wpa_config_read_global_os_version(config, hk))
255                 errors++;
256         wpa_config_read_reg_dword(hk, TEXT("wps_cred_processing"),
257                                   &config->wps_cred_processing);
258 #endif /* CONFIG_WPS */
259 #ifdef CONFIG_P2P
260         config->p2p_ssid_postfix = wpa_config_read_reg_string(
261                 hk, TEXT("p2p_ssid_postfix"));
262         wpa_config_read_reg_dword(hk, TEXT("p2p_group_idle"),
263                                   (int *) &config->p2p_group_idle);
264 #endif /* CONFIG_P2P */
265
266         wpa_config_read_reg_dword(hk, TEXT("bss_max_count"),
267                                   (int *) &config->bss_max_count);
268         wpa_config_read_reg_dword(hk, TEXT("filter_ssids"),
269                                   &config->filter_ssids);
270         wpa_config_read_reg_dword(hk, TEXT("max_num_sta"),
271                                   (int *) &config->max_num_sta);
272         wpa_config_read_reg_dword(hk, TEXT("disassoc_low_ack"),
273                                   (int *) &config->disassoc_low_ack);
274
275         return errors ? -1 : 0;
276 }
277
278
279 static struct wpa_ssid * wpa_config_read_network(HKEY hk, const TCHAR *netw,
280                                                  int id)
281 {
282         HKEY nhk;
283         LONG ret;
284         DWORD i;
285         struct wpa_ssid *ssid;
286         int errors = 0;
287
288         ret = RegOpenKeyEx(hk, netw, 0, KEY_QUERY_VALUE, &nhk);
289         if (ret != ERROR_SUCCESS) {
290                 wpa_printf(MSG_DEBUG, "Could not open wpa_supplicant config "
291                            "network '" TSTR "'", netw);
292                 return NULL;
293         }
294
295         wpa_printf(MSG_MSGDUMP, "Start of a new network '" TSTR "'", netw);
296         ssid = os_zalloc(sizeof(*ssid));
297         if (ssid == NULL) {
298                 RegCloseKey(nhk);
299                 return NULL;
300         }
301         ssid->id = id;
302
303         wpa_config_set_network_defaults(ssid);
304
305         for (i = 0; ; i++) {
306                 TCHAR name[255], data[1024];
307                 DWORD namelen, datalen, type;
308
309                 namelen = 255;
310                 datalen = sizeof(data);
311                 ret = RegEnumValue(nhk, i, name, &namelen, NULL, &type,
312                                    (LPBYTE) data, &datalen);
313
314                 if (ret == ERROR_NO_MORE_ITEMS)
315                         break;
316
317                 if (ret != ERROR_SUCCESS) {
318                         wpa_printf(MSG_ERROR, "RegEnumValue failed: 0x%x",
319                                    (unsigned int) ret);
320                         break;
321                 }
322
323                 if (namelen >= 255)
324                         namelen = 255 - 1;
325                 name[namelen] = TEXT('\0');
326
327                 if (datalen >= 1024)
328                         datalen = 1024 - 1;
329                 data[datalen] = TEXT('\0');
330
331                 wpa_unicode2ascii_inplace(name);
332                 wpa_unicode2ascii_inplace(data);
333                 if (wpa_config_set(ssid, (char *) name, (char *) data, 0) < 0)
334                         errors++;
335         }
336
337         RegCloseKey(nhk);
338
339         if (ssid->passphrase) {
340                 if (ssid->psk_set) {
341                         wpa_printf(MSG_ERROR, "Both PSK and passphrase "
342                                    "configured for network '" TSTR "'.", netw);
343                         errors++;
344                 }
345                 wpa_config_update_psk(ssid);
346         }
347
348         if ((ssid->key_mgmt & (WPA_KEY_MGMT_PSK | WPA_KEY_MGMT_FT_PSK |
349                                WPA_KEY_MGMT_PSK_SHA256)) &&
350             !ssid->psk_set) {
351                 wpa_printf(MSG_ERROR, "WPA-PSK accepted for key management, "
352                            "but no PSK configured for network '" TSTR "'.",
353                            netw);
354                 errors++;
355         }
356
357         if ((ssid->group_cipher & WPA_CIPHER_CCMP) &&
358             !(ssid->pairwise_cipher & WPA_CIPHER_CCMP) &&
359             !(ssid->pairwise_cipher & WPA_CIPHER_NONE)) {
360                 /* Group cipher cannot be stronger than the pairwise cipher. */
361                 wpa_printf(MSG_DEBUG, "Removed CCMP from group cipher "
362                            "list since it was not allowed for pairwise "
363                            "cipher for network '" TSTR "'.", netw);
364                 ssid->group_cipher &= ~WPA_CIPHER_CCMP;
365         }
366
367         if (errors) {
368                 wpa_config_free_ssid(ssid);
369                 ssid = NULL;
370         }
371
372         return ssid;
373 }
374
375
376 static int wpa_config_read_networks(struct wpa_config *config, HKEY hk)
377 {
378         HKEY nhk;
379         struct wpa_ssid *ssid, *tail = NULL, *head = NULL;
380         int errors = 0;
381         LONG ret;
382         DWORD i;
383
384         ret = RegOpenKeyEx(hk, TEXT("networks"), 0, KEY_ENUMERATE_SUB_KEYS,
385                            &nhk);
386         if (ret != ERROR_SUCCESS) {
387                 wpa_printf(MSG_ERROR, "Could not open wpa_supplicant networks "
388                            "registry key");
389                 return -1;
390         }
391
392         for (i = 0; ; i++) {
393                 TCHAR name[255];
394                 DWORD namelen;
395
396                 namelen = 255;
397                 ret = RegEnumKeyEx(nhk, i, name, &namelen, NULL, NULL, NULL,
398                                    NULL);
399
400                 if (ret == ERROR_NO_MORE_ITEMS)
401                         break;
402
403                 if (ret != ERROR_SUCCESS) {
404                         wpa_printf(MSG_DEBUG, "RegEnumKeyEx failed: 0x%x",
405                                    (unsigned int) ret);
406                         break;
407                 }
408
409                 if (namelen >= 255)
410                         namelen = 255 - 1;
411                 name[namelen] = '\0';
412
413                 ssid = wpa_config_read_network(nhk, name, i);
414                 if (ssid == NULL) {
415                         wpa_printf(MSG_ERROR, "Failed to parse network "
416                                    "profile '%s'.", name);
417                         errors++;
418                         continue;
419                 }
420                 if (head == NULL) {
421                         head = tail = ssid;
422                 } else {
423                         tail->next = ssid;
424                         tail = ssid;
425                 }
426                 if (wpa_config_add_prio_network(config, ssid)) {
427                         wpa_printf(MSG_ERROR, "Failed to add network profile "
428                                    "'%s' to priority list.", name);
429                         errors++;
430                         continue;
431                 }
432         }
433
434         RegCloseKey(nhk);
435
436         config->ssid = head;
437
438         return errors ? -1 : 0;
439 }
440
441
442 struct wpa_config * wpa_config_read(const char *name)
443 {
444         TCHAR buf[256];
445         int errors = 0;
446         struct wpa_config *config;
447         HKEY hk;
448         LONG ret;
449
450         config = wpa_config_alloc_empty(NULL, NULL);
451         if (config == NULL)
452                 return NULL;
453         wpa_printf(MSG_DEBUG, "Reading configuration profile '%s'", name);
454
455 #ifdef UNICODE
456         _snwprintf(buf, 256, WPA_KEY_PREFIX TEXT("\\configs\\%S"), name);
457 #else /* UNICODE */
458         os_snprintf(buf, 256, WPA_KEY_PREFIX TEXT("\\configs\\%s"), name);
459 #endif /* UNICODE */
460
461         ret = RegOpenKeyEx(WPA_KEY_ROOT, buf, 0, KEY_QUERY_VALUE, &hk);
462         if (ret != ERROR_SUCCESS) {
463                 wpa_printf(MSG_ERROR, "Could not open wpa_supplicant "
464                            "configuration registry HKLM\\" TSTR, buf);
465                 os_free(config);
466                 return NULL;
467         }
468
469         if (wpa_config_read_global(config, hk))
470                 errors++;
471
472         if (wpa_config_read_networks(config, hk))
473                 errors++;
474
475         if (wpa_config_read_blobs(config, hk))
476                 errors++;
477
478         wpa_config_debug_dump_networks(config);
479
480         RegCloseKey(hk);
481
482         if (errors) {
483                 wpa_config_free(config);
484                 config = NULL;
485         }
486
487         return config;
488 }
489
490
491 static int wpa_config_write_reg_dword(HKEY hk, const TCHAR *name, int val,
492                                       int def)
493 {
494         LONG ret;
495         DWORD _val = val;
496
497         if (val == def) {
498                 RegDeleteValue(hk, name);
499                 return 0;
500         }
501
502         ret = RegSetValueEx(hk, name, 0, REG_DWORD, (LPBYTE) &_val,
503                             sizeof(_val));
504         if (ret != ERROR_SUCCESS) {
505                 wpa_printf(MSG_ERROR, "WINREG: Failed to set %s=%d: error %d",
506                            name, val, (int) GetLastError());
507                 return -1;
508         }
509
510         return 0;
511 }
512
513
514 static int wpa_config_write_reg_string(HKEY hk, const char *name,
515                                        const char *val)
516 {
517         LONG ret;
518         TCHAR *_name, *_val;
519
520         _name = wpa_strdup_tchar(name);
521         if (_name == NULL)
522                 return -1;
523
524         if (val == NULL) {
525                 RegDeleteValue(hk, _name);
526                 os_free(_name);
527                 return 0;
528         }
529
530         _val = wpa_strdup_tchar(val);
531         if (_val == NULL) {
532                 os_free(_name);
533                 return -1;
534         }
535         ret = RegSetValueEx(hk, _name, 0, REG_SZ, (BYTE *) _val,
536                             (os_strlen(val) + 1) * sizeof(TCHAR));
537         if (ret != ERROR_SUCCESS) {
538                 wpa_printf(MSG_ERROR, "WINREG: Failed to set %s='%s': "
539                            "error %d", name, val, (int) GetLastError());
540                 os_free(_name);
541                 os_free(_val);
542                 return -1;
543         }
544
545         os_free(_name);
546         os_free(_val);
547         return 0;
548 }
549
550
551 static int wpa_config_write_global(struct wpa_config *config, HKEY hk)
552 {
553 #ifdef CONFIG_CTRL_IFACE
554         wpa_config_write_reg_string(hk, "ctrl_interface",
555                                     config->ctrl_interface);
556 #endif /* CONFIG_CTRL_IFACE */
557
558         wpa_config_write_reg_dword(hk, TEXT("eapol_version"),
559                                    config->eapol_version,
560                                    DEFAULT_EAPOL_VERSION);
561         wpa_config_write_reg_dword(hk, TEXT("ap_scan"), config->ap_scan,
562                                    DEFAULT_AP_SCAN);
563         wpa_config_write_reg_dword(hk, TEXT("fast_reauth"),
564                                    config->fast_reauth, DEFAULT_FAST_REAUTH);
565         wpa_config_write_reg_dword(hk, TEXT("dot11RSNAConfigPMKLifetime"),
566                                    config->dot11RSNAConfigPMKLifetime, 0);
567         wpa_config_write_reg_dword(hk,
568                                    TEXT("dot11RSNAConfigPMKReauthThreshold"),
569                                    config->dot11RSNAConfigPMKReauthThreshold,
570                                    0);
571         wpa_config_write_reg_dword(hk, TEXT("dot11RSNAConfigSATimeout"),
572                                    config->dot11RSNAConfigSATimeout, 0);
573         wpa_config_write_reg_dword(hk, TEXT("update_config"),
574                                    config->update_config,
575                                    0);
576 #ifdef CONFIG_WPS
577         if (!is_nil_uuid(config->uuid)) {
578                 char buf[40];
579                 uuid_bin2str(config->uuid, buf, sizeof(buf));
580                 wpa_config_write_reg_string(hk, "uuid", buf);
581         }
582         wpa_config_write_reg_string(hk, "device_name", config->device_name);
583         wpa_config_write_reg_string(hk, "manufacturer", config->manufacturer);
584         wpa_config_write_reg_string(hk, "model_name", config->model_name);
585         wpa_config_write_reg_string(hk, "model_number", config->model_number);
586         wpa_config_write_reg_string(hk, "serial_number",
587                                     config->serial_number);
588         wpa_config_write_reg_string(hk, "device_type", config->device_type);
589         wpa_config_write_reg_string(hk, "config_methods",
590                                     config->config_methods);
591         if (WPA_GET_BE32(config->os_version)) {
592                 char vbuf[10];
593                 os_snprintf(vbuf, sizeof(vbuf), "%08x",
594                             WPA_GET_BE32(config->os_version));
595                 wpa_config_write_reg_string(hk, "os_version", vbuf);
596         }
597         wpa_config_write_reg_dword(hk, TEXT("wps_cred_processing"),
598                                    config->wps_cred_processing, 0);
599 #endif /* CONFIG_WPS */
600 #ifdef CONFIG_P2P
601         wpa_config_write_reg_string(hk, "p2p_ssid_postfix",
602                                     config->p2p_ssid_postfix);
603         wpa_config_write_reg_dword(hk, TEXT("p2p_group_idle"),
604                                    config->p2p_group_idle, 0);
605 #endif /* CONFIG_P2P */
606
607         wpa_config_write_reg_dword(hk, TEXT("bss_max_count"),
608                                    config->bss_max_count,
609                                    DEFAULT_BSS_MAX_COUNT);
610         wpa_config_write_reg_dword(hk, TEXT("filter_ssids"),
611                                    config->filter_ssids, 0);
612         wpa_config_write_reg_dword(hk, TEXT("max_num_sta"),
613                                    config->max_num_sta, DEFAULT_MAX_NUM_STA);
614         wpa_config_write_reg_dword(hk, TEXT("disassoc_low_ack"),
615                                    config->disassoc_low_ack, 0);
616
617         return 0;
618 }
619
620
621 static int wpa_config_delete_subkeys(HKEY hk, const TCHAR *key)
622 {
623         HKEY nhk;
624         int i, errors = 0;
625         LONG ret;
626
627         ret = RegOpenKeyEx(hk, key, 0, KEY_ENUMERATE_SUB_KEYS | DELETE, &nhk);
628         if (ret != ERROR_SUCCESS) {
629                 wpa_printf(MSG_DEBUG, "WINREG: Could not open key '" TSTR
630                            "' for subkey deletion: error 0x%x (%d)", key,
631                            (unsigned int) ret, (int) GetLastError());
632                 return 0;
633         }
634
635         for (i = 0; ; i++) {
636                 TCHAR name[255];
637                 DWORD namelen;
638
639                 namelen = 255;
640                 ret = RegEnumKeyEx(nhk, i, name, &namelen, NULL, NULL, NULL,
641                                    NULL);
642
643                 if (ret == ERROR_NO_MORE_ITEMS)
644                         break;
645
646                 if (ret != ERROR_SUCCESS) {
647                         wpa_printf(MSG_DEBUG, "RegEnumKeyEx failed: 0x%x (%d)",
648                                    (unsigned int) ret, (int) GetLastError());
649                         break;
650                 }
651
652                 if (namelen >= 255)
653                         namelen = 255 - 1;
654                 name[namelen] = TEXT('\0');
655
656                 ret = RegDeleteKey(nhk, name);
657                 if (ret != ERROR_SUCCESS) {
658                         wpa_printf(MSG_DEBUG, "RegDeleteKey failed: 0x%x (%d)",
659                                    (unsigned int) ret, (int) GetLastError());
660                         errors++;
661                 }
662         }
663
664         RegCloseKey(nhk);
665
666         return errors ? -1 : 0;
667 }
668
669
670 static void write_str(HKEY hk, const char *field, struct wpa_ssid *ssid)
671 {
672         char *value = wpa_config_get(ssid, field);
673         if (value == NULL)
674                 return;
675         wpa_config_write_reg_string(hk, field, value);
676         os_free(value);
677 }
678
679
680 static void write_int(HKEY hk, const char *field, int value, int def)
681 {
682         char val[20];
683         if (value == def)
684                 return;
685         os_snprintf(val, sizeof(val), "%d", value);
686         wpa_config_write_reg_string(hk, field, val);
687 }
688
689
690 static void write_bssid(HKEY hk, struct wpa_ssid *ssid)
691 {
692         char *value = wpa_config_get(ssid, "bssid");
693         if (value == NULL)
694                 return;
695         wpa_config_write_reg_string(hk, "bssid", value);
696         os_free(value);
697 }
698
699
700 static void write_psk(HKEY hk, struct wpa_ssid *ssid)
701 {
702         char *value = wpa_config_get(ssid, "psk");
703         if (value == NULL)
704                 return;
705         wpa_config_write_reg_string(hk, "psk", value);
706         os_free(value);
707 }
708
709
710 static void write_proto(HKEY hk, struct wpa_ssid *ssid)
711 {
712         char *value;
713
714         if (ssid->proto == DEFAULT_PROTO)
715                 return;
716
717         value = wpa_config_get(ssid, "proto");
718         if (value == NULL)
719                 return;
720         if (value[0])
721                 wpa_config_write_reg_string(hk, "proto", value);
722         os_free(value);
723 }
724
725
726 static void write_key_mgmt(HKEY hk, struct wpa_ssid *ssid)
727 {
728         char *value;
729
730         if (ssid->key_mgmt == DEFAULT_KEY_MGMT)
731                 return;
732
733         value = wpa_config_get(ssid, "key_mgmt");
734         if (value == NULL)
735                 return;
736         if (value[0])
737                 wpa_config_write_reg_string(hk, "key_mgmt", value);
738         os_free(value);
739 }
740
741
742 static void write_pairwise(HKEY hk, struct wpa_ssid *ssid)
743 {
744         char *value;
745
746         if (ssid->pairwise_cipher == DEFAULT_PAIRWISE)
747                 return;
748
749         value = wpa_config_get(ssid, "pairwise");
750         if (value == NULL)
751                 return;
752         if (value[0])
753                 wpa_config_write_reg_string(hk, "pairwise", value);
754         os_free(value);
755 }
756
757
758 static void write_group(HKEY hk, struct wpa_ssid *ssid)
759 {
760         char *value;
761
762         if (ssid->group_cipher == DEFAULT_GROUP)
763                 return;
764
765         value = wpa_config_get(ssid, "group");
766         if (value == NULL)
767                 return;
768         if (value[0])
769                 wpa_config_write_reg_string(hk, "group", value);
770         os_free(value);
771 }
772
773
774 static void write_auth_alg(HKEY hk, struct wpa_ssid *ssid)
775 {
776         char *value;
777
778         if (ssid->auth_alg == 0)
779                 return;
780
781         value = wpa_config_get(ssid, "auth_alg");
782         if (value == NULL)
783                 return;
784         if (value[0])
785                 wpa_config_write_reg_string(hk, "auth_alg", value);
786         os_free(value);
787 }
788
789
790 #ifdef IEEE8021X_EAPOL
791 static void write_eap(HKEY hk, struct wpa_ssid *ssid)
792 {
793         char *value;
794
795         value = wpa_config_get(ssid, "eap");
796         if (value == NULL)
797                 return;
798
799         if (value[0])
800                 wpa_config_write_reg_string(hk, "eap", value);
801         os_free(value);
802 }
803 #endif /* IEEE8021X_EAPOL */
804
805
806 static void write_wep_key(HKEY hk, int idx, struct wpa_ssid *ssid)
807 {
808         char field[20], *value;
809
810         os_snprintf(field, sizeof(field), "wep_key%d", idx);
811         value = wpa_config_get(ssid, field);
812         if (value) {
813                 wpa_config_write_reg_string(hk, field, value);
814                 os_free(value);
815         }
816 }
817
818
819 static int wpa_config_write_network(HKEY hk, struct wpa_ssid *ssid, int id)
820 {
821         int i, errors = 0;
822         HKEY nhk, netw;
823         LONG ret;
824         TCHAR name[5];
825
826         ret = RegOpenKeyEx(hk, TEXT("networks"), 0, KEY_CREATE_SUB_KEY, &nhk);
827         if (ret != ERROR_SUCCESS) {
828                 wpa_printf(MSG_DEBUG, "WINREG: Could not open networks key "
829                            "for subkey addition: error 0x%x (%d)",
830                            (unsigned int) ret, (int) GetLastError());
831                 return 0;
832         }
833
834 #ifdef UNICODE
835         wsprintf(name, L"%04d", id);
836 #else /* UNICODE */
837         os_snprintf(name, sizeof(name), "%04d", id);
838 #endif /* UNICODE */
839         ret = RegCreateKeyEx(nhk, name, 0, NULL, 0, KEY_WRITE, NULL, &netw,
840                              NULL);
841         RegCloseKey(nhk);
842         if (ret != ERROR_SUCCESS) {
843                 wpa_printf(MSG_DEBUG, "WINREG: Could not add network key '%s':"
844                            " error 0x%x (%d)",
845                            name, (unsigned int) ret, (int) GetLastError());
846                 return -1;
847         }
848
849 #define STR(t) write_str(netw, #t, ssid)
850 #define INT(t) write_int(netw, #t, ssid->t, 0)
851 #define INTe(t) write_int(netw, #t, ssid->eap.t, 0)
852 #define INT_DEF(t, def) write_int(netw, #t, ssid->t, def)
853 #define INT_DEFe(t, def) write_int(netw, #t, ssid->eap.t, def)
854
855         STR(ssid);
856         INT(scan_ssid);
857         write_bssid(netw, ssid);
858         write_psk(netw, ssid);
859         write_proto(netw, ssid);
860         write_key_mgmt(netw, ssid);
861         write_pairwise(netw, ssid);
862         write_group(netw, ssid);
863         write_auth_alg(netw, ssid);
864 #ifdef IEEE8021X_EAPOL
865         write_eap(netw, ssid);
866         STR(identity);
867         STR(anonymous_identity);
868         STR(password);
869         STR(ca_cert);
870         STR(ca_path);
871         STR(client_cert);
872         STR(private_key);
873         STR(private_key_passwd);
874         STR(dh_file);
875         STR(subject_match);
876         STR(altsubject_match);
877         STR(ca_cert2);
878         STR(ca_path2);
879         STR(client_cert2);
880         STR(private_key2);
881         STR(private_key2_passwd);
882         STR(dh_file2);
883         STR(subject_match2);
884         STR(altsubject_match2);
885         STR(phase1);
886         STR(phase2);
887         STR(pcsc);
888         STR(pin);
889         STR(engine_id);
890         STR(key_id);
891         STR(cert_id);
892         STR(ca_cert_id);
893         STR(key2_id);
894         STR(pin2);
895         STR(engine2_id);
896         STR(cert2_id);
897         STR(ca_cert2_id);
898         INTe(engine);
899         INTe(engine2);
900         INT_DEF(eapol_flags, DEFAULT_EAPOL_FLAGS);
901 #endif /* IEEE8021X_EAPOL */
902         for (i = 0; i < 4; i++)
903                 write_wep_key(netw, i, ssid);
904         INT(wep_tx_keyidx);
905         INT(priority);
906 #ifdef IEEE8021X_EAPOL
907         INT_DEF(eap_workaround, DEFAULT_EAP_WORKAROUND);
908         STR(pac_file);
909         INT_DEFe(fragment_size, DEFAULT_FRAGMENT_SIZE);
910 #endif /* IEEE8021X_EAPOL */
911         INT(mode);
912         INT(proactive_key_caching);
913         INT(disabled);
914         INT(peerkey);
915 #ifdef CONFIG_IEEE80211W
916         INT(ieee80211w);
917 #endif /* CONFIG_IEEE80211W */
918         STR(id_str);
919
920 #undef STR
921 #undef INT
922 #undef INT_DEF
923
924         RegCloseKey(netw);
925
926         return errors ? -1 : 0;
927 }
928
929
930 static int wpa_config_write_blob(HKEY hk, struct wpa_config_blob *blob)
931 {
932         HKEY bhk;
933         LONG ret;
934         TCHAR *name;
935
936         ret = RegCreateKeyEx(hk, TEXT("blobs"), 0, NULL, 0, KEY_WRITE, NULL,
937                              &bhk, NULL);
938         if (ret != ERROR_SUCCESS) {
939                 wpa_printf(MSG_DEBUG, "WINREG: Could not add blobs key: "
940                            "error 0x%x (%d)",
941                            (unsigned int) ret, (int) GetLastError());
942                 return -1;
943         }
944
945         name = wpa_strdup_tchar(blob->name);
946         ret = RegSetValueEx(bhk, name, 0, REG_BINARY, blob->data,
947                             blob->len);
948         if (ret != ERROR_SUCCESS) {
949                 wpa_printf(MSG_ERROR, "WINREG: Failed to set blob %s': "
950                            "error 0x%x (%d)", blob->name, (unsigned int) ret,
951                            (int) GetLastError());
952                 RegCloseKey(bhk);
953                 os_free(name);
954                 return -1;
955         }
956         os_free(name);
957
958         RegCloseKey(bhk);
959
960         return 0;
961 }
962
963
964 int wpa_config_write(const char *name, struct wpa_config *config)
965 {
966         TCHAR buf[256];
967         HKEY hk;
968         LONG ret;
969         int errors = 0;
970         struct wpa_ssid *ssid;
971         struct wpa_config_blob *blob;
972         int id;
973
974         wpa_printf(MSG_DEBUG, "Writing configuration file '%s'", name);
975
976 #ifdef UNICODE
977         _snwprintf(buf, 256, WPA_KEY_PREFIX TEXT("\\configs\\%S"), name);
978 #else /* UNICODE */
979         os_snprintf(buf, 256, WPA_KEY_PREFIX TEXT("\\configs\\%s"), name);
980 #endif /* UNICODE */
981
982         ret = RegOpenKeyEx(WPA_KEY_ROOT, buf, 0, KEY_SET_VALUE | DELETE, &hk);
983         if (ret != ERROR_SUCCESS) {
984                 wpa_printf(MSG_ERROR, "Could not open wpa_supplicant "
985                            "configuration registry %s: error %d", buf,
986                            (int) GetLastError());
987                 return -1;
988         }
989
990         if (wpa_config_write_global(config, hk)) {
991                 wpa_printf(MSG_ERROR, "Failed to write global configuration "
992                            "data");
993                 errors++;
994         }
995
996         wpa_config_delete_subkeys(hk, TEXT("networks"));
997         for (ssid = config->ssid, id = 0; ssid; ssid = ssid->next, id++) {
998                 if (ssid->key_mgmt == WPA_KEY_MGMT_WPS)
999                         continue; /* do not save temporary WPS networks */
1000                 if (wpa_config_write_network(hk, ssid, id))
1001                         errors++;
1002         }
1003
1004         RegDeleteKey(hk, TEXT("blobs"));
1005         for (blob = config->blobs; blob; blob = blob->next) {
1006                 if (wpa_config_write_blob(hk, blob))
1007                         errors++;
1008         }
1009
1010         RegCloseKey(hk);
1011
1012         wpa_printf(MSG_DEBUG, "Configuration '%s' written %ssuccessfully",
1013                    name, errors ? "un" : "");
1014         return errors ? -1 : 0;
1015 }