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