1/*
2 * WPA Supplicant / main() function for Win32 service
3 * Copyright (c) 2003-2006, 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 * The root of wpa_supplicant configuration in registry is
15 * HKEY_LOCAL_MACHINE\\SOFTWARE\\%wpa_supplicant. This level includes global
16 * parameters and a 'interfaces' subkey with all the interface configuration
17 * (adapter to confname mapping). Each such mapping is a subkey that has
18 * 'adapter' and 'config' values.
19 *
20 * This program can be run either as a normal command line application, e.g.,
21 * for debugging, with 'wpasvc.exe app' or as a Windows service. Service need
22 * to be registered with 'wpasvc.exe reg <full path to wpasvc.exe>'. After
23 * this, it can be started like any other Windows service (e.g., 'net start
24 * wpasvc') or it can be configured to start automatically through the Services
25 * tool in administrative tasks. The service can be unregistered with
26 * 'wpasvc.exe unreg'.
27 */
28
29#include "includes.h"
30#include <windows.h>
31
32#include "common.h"
33#include "wpa_supplicant_i.h"
34#include "eloop.h"
35
36#ifndef WPASVC_NAME
37#define WPASVC_NAME TEXT("wpasvc")
38#endif
39#ifndef WPASVC_DISPLAY_NAME
40#define WPASVC_DISPLAY_NAME TEXT("wpa_supplicant service")
41#endif
42#ifndef WPASVC_DESCRIPTION
43#define WPASVC_DESCRIPTION \
44TEXT("Provides IEEE 802.1X and WPA/WPA2 supplicant functionality")
45#endif
46
47static HANDLE kill_svc;
48
49static SERVICE_STATUS_HANDLE svc_status_handle;
50static SERVICE_STATUS svc_status;
51
52
53#ifndef WPA_KEY_ROOT
54#define WPA_KEY_ROOT HKEY_LOCAL_MACHINE
55#endif
56#ifndef WPA_KEY_PREFIX
57#define WPA_KEY_PREFIX TEXT("SOFTWARE\\wpa_supplicant")
58#endif
59
60#ifdef UNICODE
61#define TSTR "%S"
62#else /* UNICODE */
63#define TSTR "%s"
64#endif /* UNICODE */
65
66
67static int read_interface(struct wpa_global *global, HKEY _hk,
68			  const TCHAR *name)
69{
70	HKEY hk;
71#define TBUFLEN 255
72	TCHAR adapter[TBUFLEN], config[TBUFLEN], ctrl_interface[TBUFLEN];
73	DWORD buflen, val;
74	LONG ret;
75	struct wpa_interface iface;
76	int skip_on_error = 0;
77
78	ret = RegOpenKeyEx(_hk, name, 0, KEY_QUERY_VALUE, &hk);
79	if (ret != ERROR_SUCCESS) {
80		printf("Could not open wpa_supplicant interface key\n");
81		return -1;
82	}
83
84	os_memset(&iface, 0, sizeof(iface));
85	iface.driver = "ndis";
86
87	buflen = sizeof(ctrl_interface);
88	ret = RegQueryValueEx(hk, TEXT("ctrl_interface"), NULL, NULL,
89			      (LPBYTE) ctrl_interface, &buflen);
90	if (ret == ERROR_SUCCESS) {
91		ctrl_interface[TBUFLEN - 1] = TEXT('\0');
92		wpa_unicode2ascii_inplace(ctrl_interface);
93		printf("ctrl_interface[len=%d] '%s'\n",
94		       (int) buflen, (char *) ctrl_interface);
95		iface.ctrl_interface = (char *) ctrl_interface;
96	}
97
98	buflen = sizeof(adapter);
99	ret = RegQueryValueEx(hk, TEXT("adapter"), NULL, NULL,
100			      (LPBYTE) adapter, &buflen);
101	if (ret == ERROR_SUCCESS) {
102		adapter[TBUFLEN - 1] = TEXT('\0');
103		wpa_unicode2ascii_inplace(adapter);
104		printf("adapter[len=%d] '%s'\n",
105		       (int) buflen, (char *) adapter);
106		iface.ifname = (char *) adapter;
107	}
108
109	buflen = sizeof(config);
110	ret = RegQueryValueEx(hk, TEXT("config"), NULL, NULL,
111			      (LPBYTE) config, &buflen);
112	if (ret == ERROR_SUCCESS) {
113		config[sizeof(config) - 1] = '\0';
114		wpa_unicode2ascii_inplace(config);
115		printf("config[len=%d] '%s'\n",
116		       (int) buflen, (char *) config);
117		iface.confname = (char *) config;
118	}
119
120	buflen = sizeof(val);
121	ret = RegQueryValueEx(hk, TEXT("skip_on_error"), NULL, NULL,
122			      (LPBYTE) &val, &buflen);
123	if (ret == ERROR_SUCCESS && buflen == sizeof(val))
124		skip_on_error = val;
125
126	RegCloseKey(hk);
127
128	if (wpa_supplicant_add_iface(global, &iface) == NULL) {
129		if (skip_on_error)
130			wpa_printf(MSG_DEBUG, "Skipped interface '%s' due to "
131				   "initialization failure", iface.ifname);
132		else
133			return -1;
134	}
135
136	return 0;
137}
138
139
140static int wpa_supplicant_thread(void)
141{
142	int exitcode;
143	struct wpa_params params;
144	struct wpa_global *global;
145	HKEY hk, ihk;
146	DWORD val, buflen, i;
147	LONG ret;
148
149	if (os_program_init())
150		return -1;
151
152	os_memset(&params, 0, sizeof(params));
153	params.wpa_debug_level = MSG_INFO;
154
155	ret = RegOpenKeyEx(WPA_KEY_ROOT, WPA_KEY_PREFIX,
156			   0, KEY_QUERY_VALUE, &hk);
157	if (ret != ERROR_SUCCESS) {
158		printf("Could not open wpa_supplicant registry key\n");
159		return -1;
160	}
161
162	buflen = sizeof(val);
163	ret = RegQueryValueEx(hk, TEXT("debug_level"), NULL, NULL,
164			      (LPBYTE) &val, &buflen);
165	if (ret == ERROR_SUCCESS && buflen == sizeof(val)) {
166		params.wpa_debug_level = val;
167	}
168
169	buflen = sizeof(val);
170	ret = RegQueryValueEx(hk, TEXT("debug_show_keys"), NULL, NULL,
171			      (LPBYTE) &val, &buflen);
172	if (ret == ERROR_SUCCESS && buflen == sizeof(val)) {
173		params.wpa_debug_show_keys = val;
174	}
175
176	buflen = sizeof(val);
177	ret = RegQueryValueEx(hk, TEXT("debug_timestamp"), NULL, NULL,
178			      (LPBYTE) &val, &buflen);
179	if (ret == ERROR_SUCCESS && buflen == sizeof(val)) {
180		params.wpa_debug_timestamp = val;
181	}
182
183	buflen = sizeof(val);
184	ret = RegQueryValueEx(hk, TEXT("debug_use_file"), NULL, NULL,
185			      (LPBYTE) &val, &buflen);
186	if (ret == ERROR_SUCCESS && buflen == sizeof(val) && val) {
187		params.wpa_debug_file_path = "\\Temp\\wpa_supplicant-log.txt";
188	}
189
190	exitcode = 0;
191	global = wpa_supplicant_init(&params);
192	if (global == NULL) {
193		printf("Failed to initialize wpa_supplicant\n");
194		exitcode = -1;
195	}
196
197	ret = RegOpenKeyEx(hk, TEXT("interfaces"), 0, KEY_ENUMERATE_SUB_KEYS,
198			   &ihk);
199	RegCloseKey(hk);
200	if (ret != ERROR_SUCCESS) {
201		printf("Could not open wpa_supplicant interfaces registry "
202		       "key\n");
203		return -1;
204	}
205
206	for (i = 0; ; i++) {
207		TCHAR name[255];
208		DWORD namelen;
209
210		namelen = 255;
211		ret = RegEnumKeyEx(ihk, i, name, &namelen, NULL, NULL, NULL,
212				   NULL);
213
214		if (ret == ERROR_NO_MORE_ITEMS)
215			break;
216
217		if (ret != ERROR_SUCCESS) {
218			printf("RegEnumKeyEx failed: 0x%x\n",
219			       (unsigned int) ret);
220			break;
221		}
222
223		if (namelen >= 255)
224			namelen = 255 - 1;
225		name[namelen] = '\0';
226
227		wpa_printf(MSG_DEBUG, "interface %d: %s\n", (int) i, name);
228		if (read_interface(global, ihk, name) < 0)
229			exitcode = -1;
230	}
231
232	RegCloseKey(ihk);
233
234	if (exitcode == 0)
235		exitcode = wpa_supplicant_run(global);
236
237	wpa_supplicant_deinit(global);
238
239	os_program_deinit();
240
241	return exitcode;
242}
243
244
245static DWORD svc_thread(LPDWORD param)
246{
247	int ret = wpa_supplicant_thread();
248
249	svc_status.dwCurrentState = SERVICE_STOPPED;
250	svc_status.dwWaitHint = 0;
251	if (!SetServiceStatus(svc_status_handle, &svc_status)) {
252		printf("SetServiceStatus() failed: %d\n",
253		       (int) GetLastError());
254	}
255
256	return ret;
257}
258
259
260static int register_service(const TCHAR *exe)
261{
262	SC_HANDLE svc, scm;
263	SERVICE_DESCRIPTION sd;
264
265	printf("Registering service: " TSTR "\n", WPASVC_NAME);
266
267	scm = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE);
268	if (!scm) {
269		printf("OpenSCManager failed: %d\n", (int) GetLastError());
270		return -1;
271	}
272
273	svc = CreateService(scm, WPASVC_NAME, WPASVC_DISPLAY_NAME,
274			    SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
275			    SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
276			    exe, NULL, NULL, NULL, NULL, NULL);
277
278	if (!svc) {
279		printf("CreateService failed: %d\n\n", (int) GetLastError());
280		CloseServiceHandle(scm);
281		return -1;
282	}
283
284	os_memset(&sd, 0, sizeof(sd));
285	sd.lpDescription = WPASVC_DESCRIPTION;
286	if (!ChangeServiceConfig2(svc, SERVICE_CONFIG_DESCRIPTION, &sd)) {
287		printf("ChangeServiceConfig2 failed: %d\n",
288		       (int) GetLastError());
289		/* This is not a fatal error, so continue anyway. */
290	}
291
292	CloseServiceHandle(svc);
293	CloseServiceHandle(scm);
294
295	printf("Service registered successfully.\n");
296
297	return 0;
298}
299
300
301static int unregister_service(void)
302{
303	SC_HANDLE svc, scm;
304	SERVICE_STATUS status;
305
306	printf("Unregistering service: " TSTR "\n", WPASVC_NAME);
307
308	scm = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE);
309	if (!scm) {
310		printf("OpenSCManager failed: %d\n", (int) GetLastError());
311		return -1;
312	}
313
314	svc = OpenService(scm, WPASVC_NAME, SERVICE_ALL_ACCESS | DELETE);
315	if (!svc) {
316		printf("OpenService failed: %d\n\n", (int) GetLastError());
317		CloseServiceHandle(scm);
318		return -1;
319	}
320
321	if (QueryServiceStatus(svc, &status)) {
322		if (status.dwCurrentState != SERVICE_STOPPED) {
323			printf("Service currently active - stopping "
324			       "service...\n");
325			if (!ControlService(svc, SERVICE_CONTROL_STOP,
326					    &status)) {
327				printf("ControlService failed: %d\n",
328				       (int) GetLastError());
329			}
330			Sleep(500);
331		}
332	}
333
334	if (DeleteService(svc)) {
335		printf("Service unregistered successfully.\n");
336	} else {
337		printf("DeleteService failed: %d\n", (int) GetLastError());
338	}
339
340	CloseServiceHandle(svc);
341	CloseServiceHandle(scm);
342
343	return 0;
344}
345
346
347static void WINAPI service_ctrl_handler(DWORD control_code)
348{
349	switch (control_code) {
350	case SERVICE_CONTROL_INTERROGATE:
351		break;
352	case SERVICE_CONTROL_SHUTDOWN:
353	case SERVICE_CONTROL_STOP:
354		svc_status.dwCurrentState = SERVICE_STOP_PENDING;
355		svc_status.dwWaitHint = 2000;
356		eloop_terminate();
357		SetEvent(kill_svc);
358		break;
359	}
360
361	if (!SetServiceStatus(svc_status_handle, &svc_status)) {
362		printf("SetServiceStatus() failed: %d\n",
363		       (int) GetLastError());
364	}
365}
366
367
368static void WINAPI service_start(DWORD argc, LPTSTR *argv)
369{
370	DWORD id;
371
372	svc_status_handle = RegisterServiceCtrlHandler(WPASVC_NAME,
373						       service_ctrl_handler);
374	if (svc_status_handle == (SERVICE_STATUS_HANDLE) 0) {
375		printf("RegisterServiceCtrlHandler failed: %d\n",
376		       (int) GetLastError());
377		return;
378	}
379
380	os_memset(&svc_status, 0, sizeof(svc_status));
381	svc_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
382	svc_status.dwCurrentState = SERVICE_START_PENDING;
383	svc_status.dwWaitHint = 1000;
384
385	if (!SetServiceStatus(svc_status_handle, &svc_status)) {
386		printf("SetServiceStatus() failed: %d\n",
387		       (int) GetLastError());
388		return;
389	}
390
391	kill_svc = CreateEvent(0, TRUE, FALSE, 0);
392	if (!kill_svc) {
393		printf("CreateEvent failed: %d\n", (int) GetLastError());
394		return;
395	}
396
397	if (CreateThread(0, 0, (LPTHREAD_START_ROUTINE) svc_thread, 0, 0, &id)
398	    == 0) {
399		printf("CreateThread failed: %d\n", (int) GetLastError());
400		return;
401	}
402
403	if (svc_status.dwCurrentState == SERVICE_START_PENDING) {
404		svc_status.dwCurrentState = SERVICE_RUNNING;
405		svc_status.dwWaitHint = 0;
406		svc_status.dwControlsAccepted = SERVICE_ACCEPT_STOP |
407			SERVICE_ACCEPT_SHUTDOWN;
408	}
409
410	if (!SetServiceStatus(svc_status_handle, &svc_status)) {
411		printf("SetServiceStatus() failed: %d\n",
412		       (int) GetLastError());
413		return;
414	}
415
416	/* wait until service gets killed */
417	WaitForSingleObject(kill_svc, INFINITE);
418}
419
420
421int main(int argc, char *argv[])
422{
423	SERVICE_TABLE_ENTRY dt[] = {
424		{ WPASVC_NAME, service_start },
425		{ NULL, NULL }
426	};
427
428	if (argc > 1) {
429		if (os_strcmp(argv[1], "reg") == 0) {
430			TCHAR *path;
431			int ret;
432
433			if (argc < 3) {
434				path = os_malloc(MAX_PATH * sizeof(TCHAR));
435				if (path == NULL)
436					return -1;
437				if (!GetModuleFileName(NULL, path, MAX_PATH)) {
438					printf("GetModuleFileName failed: "
439					       "%d\n", (int) GetLastError());
440					os_free(path);
441					return -1;
442				}
443			} else {
444				path = wpa_strdup_tchar(argv[2]);
445				if (path == NULL)
446					return -1;
447			}
448			ret = register_service(path);
449			os_free(path);
450			return ret;
451		} else if (os_strcmp(argv[1], "unreg") == 0) {
452			return unregister_service();
453		} else if (os_strcmp(argv[1], "app") == 0) {
454			return wpa_supplicant_thread();
455		}
456	}
457
458	if (!StartServiceCtrlDispatcher(dt)) {
459		printf("StartServiceCtrlDispatcher failed: %d\n",
460		       (int) GetLastError());
461	}
462
463	return 0;
464}
465