Home | History | Annotate | Download | only in common
      1 /*
      2  * wpa_supplicant/hostapd control interface library
      3  * Copyright (c) 2004-2007, Jouni Malinen <j (at) w1.fi>
      4  *
      5  * This software may be distributed under the terms of the BSD license.
      6  * See README for more details.
      7  */
      8 
      9 #include "includes.h"
     10 
     11 #ifdef CONFIG_CTRL_IFACE
     12 
     13 #ifdef CONFIG_CTRL_IFACE_UNIX
     14 #include <sys/un.h>
     15 #include <unistd.h>
     16 #include <fcntl.h>
     17 #endif /* CONFIG_CTRL_IFACE_UNIX */
     18 #ifdef CONFIG_CTRL_IFACE_UDP_REMOTE
     19 #include <netdb.h>
     20 #endif /* CONFIG_CTRL_IFACE_UDP_REMOTE */
     21 
     22 #ifdef ANDROID
     23 #include <dirent.h>
     24 #include <sys/stat.h>
     25 #include <cutils/sockets.h>
     26 #include "private/android_filesystem_config.h"
     27 #endif /* ANDROID */
     28 
     29 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
     30 #include <net/if.h>
     31 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
     32 
     33 #include "wpa_ctrl.h"
     34 #include "common.h"
     35 
     36 
     37 #if defined(CONFIG_CTRL_IFACE_UNIX) || defined(CONFIG_CTRL_IFACE_UDP)
     38 #define CTRL_IFACE_SOCKET
     39 #endif /* CONFIG_CTRL_IFACE_UNIX || CONFIG_CTRL_IFACE_UDP */
     40 
     41 
     42 /**
     43  * struct wpa_ctrl - Internal structure for control interface library
     44  *
     45  * This structure is used by the wpa_supplicant/hostapd control interface
     46  * library to store internal data. Programs using the library should not touch
     47  * this data directly. They can only use the pointer to the data structure as
     48  * an identifier for the control interface connection and use this as one of
     49  * the arguments for most of the control interface library functions.
     50  */
     51 struct wpa_ctrl {
     52 #ifdef CONFIG_CTRL_IFACE_UDP
     53 	int s;
     54 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
     55 	struct sockaddr_in6 local;
     56 	struct sockaddr_in6 dest;
     57 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
     58 	struct sockaddr_in local;
     59 	struct sockaddr_in dest;
     60 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
     61 	char *cookie;
     62 	char *remote_ifname;
     63 	char *remote_ip;
     64 #endif /* CONFIG_CTRL_IFACE_UDP */
     65 #ifdef CONFIG_CTRL_IFACE_UNIX
     66 	int s;
     67 	struct sockaddr_un local;
     68 	struct sockaddr_un dest;
     69 #endif /* CONFIG_CTRL_IFACE_UNIX */
     70 #ifdef CONFIG_CTRL_IFACE_NAMED_PIPE
     71 	HANDLE pipe;
     72 #endif /* CONFIG_CTRL_IFACE_NAMED_PIPE */
     73 };
     74 
     75 
     76 #ifdef CONFIG_CTRL_IFACE_UNIX
     77 
     78 #ifndef CONFIG_CTRL_IFACE_CLIENT_DIR
     79 #define CONFIG_CTRL_IFACE_CLIENT_DIR "/tmp"
     80 #endif /* CONFIG_CTRL_IFACE_CLIENT_DIR */
     81 #ifndef CONFIG_CTRL_IFACE_CLIENT_PREFIX
     82 #define CONFIG_CTRL_IFACE_CLIENT_PREFIX "wpa_ctrl_"
     83 #endif /* CONFIG_CTRL_IFACE_CLIENT_PREFIX */
     84 
     85 
     86 struct wpa_ctrl * wpa_ctrl_open(const char *ctrl_path)
     87 {
     88 	struct wpa_ctrl *ctrl;
     89 	static int counter = 0;
     90 	int ret;
     91 	size_t res;
     92 	int tries = 0;
     93 	int flags;
     94 
     95 	if (ctrl_path == NULL)
     96 		return NULL;
     97 
     98 	ctrl = os_zalloc(sizeof(*ctrl));
     99 	if (ctrl == NULL)
    100 		return NULL;
    101 
    102 	ctrl->s = socket(PF_UNIX, SOCK_DGRAM, 0);
    103 	if (ctrl->s < 0) {
    104 		os_free(ctrl);
    105 		return NULL;
    106 	}
    107 
    108 	ctrl->local.sun_family = AF_UNIX;
    109 	counter++;
    110 try_again:
    111 	ret = os_snprintf(ctrl->local.sun_path, sizeof(ctrl->local.sun_path),
    112 			  CONFIG_CTRL_IFACE_CLIENT_DIR "/"
    113 			  CONFIG_CTRL_IFACE_CLIENT_PREFIX "%d-%d",
    114 			  (int) getpid(), counter);
    115 	if (os_snprintf_error(sizeof(ctrl->local.sun_path), ret)) {
    116 		close(ctrl->s);
    117 		os_free(ctrl);
    118 		return NULL;
    119 	}
    120 	tries++;
    121 	if (bind(ctrl->s, (struct sockaddr *) &ctrl->local,
    122 		    sizeof(ctrl->local)) < 0) {
    123 		if (errno == EADDRINUSE && tries < 2) {
    124 			/*
    125 			 * getpid() returns unique identifier for this instance
    126 			 * of wpa_ctrl, so the existing socket file must have
    127 			 * been left by unclean termination of an earlier run.
    128 			 * Remove the file and try again.
    129 			 */
    130 			unlink(ctrl->local.sun_path);
    131 			goto try_again;
    132 		}
    133 		close(ctrl->s);
    134 		os_free(ctrl);
    135 		return NULL;
    136 	}
    137 
    138 #ifdef ANDROID
    139 	chmod(ctrl->local.sun_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP);
    140 	chown(ctrl->local.sun_path, AID_SYSTEM, AID_WIFI);
    141 
    142 	if (os_strncmp(ctrl_path, "@android:", 9) == 0) {
    143 		if (socket_local_client_connect(
    144 			    ctrl->s, ctrl_path + 9,
    145 			    ANDROID_SOCKET_NAMESPACE_RESERVED,
    146 			    SOCK_DGRAM) < 0) {
    147 			close(ctrl->s);
    148 			unlink(ctrl->local.sun_path);
    149 			os_free(ctrl);
    150 			return NULL;
    151 		}
    152 		return ctrl;
    153 	}
    154 
    155 	/*
    156 	 * If the ctrl_path isn't an absolute pathname, assume that
    157 	 * it's the name of a socket in the Android reserved namespace.
    158 	 * Otherwise, it's a normal UNIX domain socket appearing in the
    159 	 * filesystem.
    160 	 */
    161 	if (*ctrl_path != '/') {
    162 		char buf[21];
    163 		os_snprintf(buf, sizeof(buf), "wpa_%s", ctrl_path);
    164 		if (socket_local_client_connect(
    165 			    ctrl->s, buf,
    166 			    ANDROID_SOCKET_NAMESPACE_RESERVED,
    167 			    SOCK_DGRAM) < 0) {
    168 			close(ctrl->s);
    169 			unlink(ctrl->local.sun_path);
    170 			os_free(ctrl);
    171 			return NULL;
    172 		}
    173 		return ctrl;
    174 	}
    175 #endif /* ANDROID */
    176 
    177 	ctrl->dest.sun_family = AF_UNIX;
    178 	if (os_strncmp(ctrl_path, "@abstract:", 10) == 0) {
    179 		ctrl->dest.sun_path[0] = '\0';
    180 		os_strlcpy(ctrl->dest.sun_path + 1, ctrl_path + 10,
    181 			   sizeof(ctrl->dest.sun_path) - 1);
    182 	} else {
    183 		res = os_strlcpy(ctrl->dest.sun_path, ctrl_path,
    184 				 sizeof(ctrl->dest.sun_path));
    185 		if (res >= sizeof(ctrl->dest.sun_path)) {
    186 			close(ctrl->s);
    187 			os_free(ctrl);
    188 			return NULL;
    189 		}
    190 	}
    191 	if (connect(ctrl->s, (struct sockaddr *) &ctrl->dest,
    192 		    sizeof(ctrl->dest)) < 0) {
    193 		close(ctrl->s);
    194 		unlink(ctrl->local.sun_path);
    195 		os_free(ctrl);
    196 		return NULL;
    197 	}
    198 
    199 	/*
    200 	 * Make socket non-blocking so that we don't hang forever if
    201 	 * target dies unexpectedly.
    202 	 */
    203 	flags = fcntl(ctrl->s, F_GETFL);
    204 	if (flags >= 0) {
    205 		flags |= O_NONBLOCK;
    206 		if (fcntl(ctrl->s, F_SETFL, flags) < 0) {
    207 			perror("fcntl(ctrl->s, O_NONBLOCK)");
    208 			/* Not fatal, continue on.*/
    209 		}
    210 	}
    211 
    212 	return ctrl;
    213 }
    214 
    215 
    216 void wpa_ctrl_close(struct wpa_ctrl *ctrl)
    217 {
    218 	if (ctrl == NULL)
    219 		return;
    220 	unlink(ctrl->local.sun_path);
    221 	if (ctrl->s >= 0)
    222 		close(ctrl->s);
    223 	os_free(ctrl);
    224 }
    225 
    226 
    227 #ifdef ANDROID
    228 /**
    229  * wpa_ctrl_cleanup() - Delete any local UNIX domain socket files that
    230  * may be left over from clients that were previously connected to
    231  * wpa_supplicant. This keeps these files from being orphaned in the
    232  * event of crashes that prevented them from being removed as part
    233  * of the normal orderly shutdown.
    234  */
    235 void wpa_ctrl_cleanup(void)
    236 {
    237 	DIR *dir;
    238 	struct dirent entry;
    239 	struct dirent *result;
    240 	size_t dirnamelen;
    241 	size_t maxcopy;
    242 	char pathname[PATH_MAX];
    243 	char *namep;
    244 
    245 	if ((dir = opendir(CONFIG_CTRL_IFACE_CLIENT_DIR)) == NULL)
    246 		return;
    247 
    248 	dirnamelen = (size_t) os_snprintf(pathname, sizeof(pathname), "%s/",
    249 					  CONFIG_CTRL_IFACE_CLIENT_DIR);
    250 	if (dirnamelen >= sizeof(pathname)) {
    251 		closedir(dir);
    252 		return;
    253 	}
    254 	namep = pathname + dirnamelen;
    255 	maxcopy = PATH_MAX - dirnamelen;
    256 	while (readdir_r(dir, &entry, &result) == 0 && result != NULL) {
    257 		if (os_strlcpy(namep, entry.d_name, maxcopy) < maxcopy)
    258 			unlink(pathname);
    259 	}
    260 	closedir(dir);
    261 }
    262 #endif /* ANDROID */
    263 
    264 #else /* CONFIG_CTRL_IFACE_UNIX */
    265 
    266 #ifdef ANDROID
    267 void wpa_ctrl_cleanup(void)
    268 {
    269 }
    270 #endif /* ANDROID */
    271 
    272 #endif /* CONFIG_CTRL_IFACE_UNIX */
    273 
    274 
    275 #ifdef CONFIG_CTRL_IFACE_UDP
    276 
    277 struct wpa_ctrl * wpa_ctrl_open(const char *ctrl_path)
    278 {
    279 	struct wpa_ctrl *ctrl;
    280 	char buf[128];
    281 	size_t len;
    282 #ifdef CONFIG_CTRL_IFACE_UDP_REMOTE
    283 	struct hostent *h;
    284 #endif /* CONFIG_CTRL_IFACE_UDP_REMOTE */
    285 
    286 	ctrl = os_zalloc(sizeof(*ctrl));
    287 	if (ctrl == NULL)
    288 		return NULL;
    289 
    290 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    291 	ctrl->s = socket(PF_INET6, SOCK_DGRAM, 0);
    292 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    293 	ctrl->s = socket(PF_INET, SOCK_DGRAM, 0);
    294 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    295 	if (ctrl->s < 0) {
    296 		perror("socket");
    297 		os_free(ctrl);
    298 		return NULL;
    299 	}
    300 
    301 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    302 	ctrl->local.sin6_family = AF_INET6;
    303 #ifdef CONFIG_CTRL_IFACE_UDP_REMOTE
    304 	ctrl->local.sin6_addr = in6addr_any;
    305 #else /* CONFIG_CTRL_IFACE_UDP_REMOTE */
    306 	inet_pton(AF_INET6, "::1", &ctrl->local.sin6_addr);
    307 #endif /* CONFIG_CTRL_IFACE_UDP_REMOTE */
    308 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    309 	ctrl->local.sin_family = AF_INET;
    310 #ifdef CONFIG_CTRL_IFACE_UDP_REMOTE
    311 	ctrl->local.sin_addr.s_addr = INADDR_ANY;
    312 #else /* CONFIG_CTRL_IFACE_UDP_REMOTE */
    313 	ctrl->local.sin_addr.s_addr = htonl((127 << 24) | 1);
    314 #endif /* CONFIG_CTRL_IFACE_UDP_REMOTE */
    315 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    316 
    317 	if (bind(ctrl->s, (struct sockaddr *) &ctrl->local,
    318 		 sizeof(ctrl->local)) < 0) {
    319 		close(ctrl->s);
    320 		os_free(ctrl);
    321 		return NULL;
    322 	}
    323 
    324 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    325 	ctrl->dest.sin6_family = AF_INET6;
    326 	inet_pton(AF_INET6, "::1", &ctrl->dest.sin6_addr);
    327 	ctrl->dest.sin6_port = htons(WPA_CTRL_IFACE_PORT);
    328 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    329 	ctrl->dest.sin_family = AF_INET;
    330 	ctrl->dest.sin_addr.s_addr = htonl((127 << 24) | 1);
    331 	ctrl->dest.sin_port = htons(WPA_CTRL_IFACE_PORT);
    332 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    333 
    334 #ifdef CONFIG_CTRL_IFACE_UDP_REMOTE
    335 	if (ctrl_path) {
    336 		char *port, *name;
    337 		int port_id;
    338 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    339 		char *scope;
    340 		int scope_id = 0;
    341 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    342 
    343 		name = os_strdup(ctrl_path);
    344 		if (name == NULL) {
    345 			close(ctrl->s);
    346 			os_free(ctrl);
    347 			return NULL;
    348 		}
    349 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    350 		port = os_strchr(name, ',');
    351 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    352 		port = os_strchr(name, ':');
    353 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    354 
    355 		if (port) {
    356 			port_id = atoi(&port[1]);
    357 			port[0] = '\0';
    358 		} else
    359 			port_id = WPA_CTRL_IFACE_PORT;
    360 
    361 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    362 		scope = os_strchr(name, '%');
    363 		if (scope) {
    364 			scope_id = if_nametoindex(&scope[1]);
    365 			scope[0] = '\0';
    366 		}
    367 		h = gethostbyname2(name, AF_INET6);
    368 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    369 		h = gethostbyname(name);
    370 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    371 		ctrl->remote_ip = os_strdup(name);
    372 		os_free(name);
    373 		if (h == NULL) {
    374 			perror("gethostbyname");
    375 			close(ctrl->s);
    376 			os_free(ctrl->remote_ip);
    377 			os_free(ctrl);
    378 			return NULL;
    379 		}
    380 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    381 		ctrl->dest.sin6_scope_id = scope_id;
    382 		ctrl->dest.sin6_port = htons(port_id);
    383 		os_memcpy(&ctrl->dest.sin6_addr, h->h_addr, h->h_length);
    384 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    385 		ctrl->dest.sin_port = htons(port_id);
    386 		os_memcpy(&ctrl->dest.sin_addr.s_addr, h->h_addr, h->h_length);
    387 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    388 	} else
    389 		ctrl->remote_ip = os_strdup("localhost");
    390 #endif /* CONFIG_CTRL_IFACE_UDP_REMOTE */
    391 
    392 	if (connect(ctrl->s, (struct sockaddr *) &ctrl->dest,
    393 		    sizeof(ctrl->dest)) < 0) {
    394 #ifdef CONFIG_CTRL_IFACE_UDP_IPV6
    395 		char addr[INET6_ADDRSTRLEN];
    396 		wpa_printf(MSG_ERROR, "connect(%s:%d) failed: %s",
    397 			   inet_ntop(AF_INET6, &ctrl->dest.sin6_addr, addr,
    398 				     sizeof(ctrl->dest)),
    399 			   ntohs(ctrl->dest.sin6_port),
    400 			   strerror(errno));
    401 #else /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    402 		wpa_printf(MSG_ERROR, "connect(%s:%d) failed: %s",
    403 			   inet_ntoa(ctrl->dest.sin_addr),
    404 			   ntohs(ctrl->dest.sin_port),
    405 			   strerror(errno));
    406 #endif /* CONFIG_CTRL_IFACE_UDP_IPV6 */
    407 		close(ctrl->s);
    408 		os_free(ctrl->remote_ip);
    409 		os_free(ctrl);
    410 		return NULL;
    411 	}
    412 
    413 	len = sizeof(buf) - 1;
    414 	if (wpa_ctrl_request(ctrl, "GET_COOKIE", 10, buf, &len, NULL) == 0) {
    415 		buf[len] = '\0';
    416 		ctrl->cookie = os_strdup(buf);
    417 	}
    418 
    419 	if (wpa_ctrl_request(ctrl, "IFNAME", 6, buf, &len, NULL) == 0) {
    420 		buf[len] = '\0';
    421 		ctrl->remote_ifname = os_strdup(buf);
    422 	}
    423 
    424 	return ctrl;
    425 }
    426 
    427 
    428 char * wpa_ctrl_get_remote_ifname(struct wpa_ctrl *ctrl)
    429 {
    430 #define WPA_CTRL_MAX_PS_NAME 100
    431 	static char ps[WPA_CTRL_MAX_PS_NAME] = {};
    432 	os_snprintf(ps, WPA_CTRL_MAX_PS_NAME, "%s/%s",
    433 		    ctrl->remote_ip, ctrl->remote_ifname);
    434 	return ps;
    435 }
    436 
    437 
    438 void wpa_ctrl_close(struct wpa_ctrl *ctrl)
    439 {
    440 	close(ctrl->s);
    441 	os_free(ctrl->cookie);
    442 	os_free(ctrl->remote_ifname);
    443 	os_free(ctrl->remote_ip);
    444 	os_free(ctrl);
    445 }
    446 
    447 #endif /* CONFIG_CTRL_IFACE_UDP */
    448 
    449 
    450 #ifdef CTRL_IFACE_SOCKET
    451 int wpa_ctrl_request(struct wpa_ctrl *ctrl, const char *cmd, size_t cmd_len,
    452 		     char *reply, size_t *reply_len,
    453 		     void (*msg_cb)(char *msg, size_t len))
    454 {
    455 	struct timeval tv;
    456 	struct os_reltime started_at;
    457 	int res;
    458 	fd_set rfds;
    459 	const char *_cmd;
    460 	char *cmd_buf = NULL;
    461 	size_t _cmd_len;
    462 
    463 #ifdef CONFIG_CTRL_IFACE_UDP
    464 	if (ctrl->cookie) {
    465 		char *pos;
    466 		_cmd_len = os_strlen(ctrl->cookie) + 1 + cmd_len;
    467 		cmd_buf = os_malloc(_cmd_len);
    468 		if (cmd_buf == NULL)
    469 			return -1;
    470 		_cmd = cmd_buf;
    471 		pos = cmd_buf;
    472 		os_strlcpy(pos, ctrl->cookie, _cmd_len);
    473 		pos += os_strlen(ctrl->cookie);
    474 		*pos++ = ' ';
    475 		os_memcpy(pos, cmd, cmd_len);
    476 	} else
    477 #endif /* CONFIG_CTRL_IFACE_UDP */
    478 	{
    479 		_cmd = cmd;
    480 		_cmd_len = cmd_len;
    481 	}
    482 
    483 	errno = 0;
    484 	started_at.sec = 0;
    485 	started_at.usec = 0;
    486 retry_send:
    487 	if (send(ctrl->s, _cmd, _cmd_len, 0) < 0) {
    488 		if (errno == EAGAIN || errno == EBUSY || errno == EWOULDBLOCK)
    489 		{
    490 			/*
    491 			 * Must be a non-blocking socket... Try for a bit
    492 			 * longer before giving up.
    493 			 */
    494 			if (started_at.sec == 0)
    495 				os_get_reltime(&started_at);
    496 			else {
    497 				struct os_reltime n;
    498 				os_get_reltime(&n);
    499 				/* Try for a few seconds. */
    500 				if (os_reltime_expired(&n, &started_at, 5))
    501 					goto send_err;
    502 			}
    503 			os_sleep(1, 0);
    504 			goto retry_send;
    505 		}
    506 	send_err:
    507 		os_free(cmd_buf);
    508 		return -1;
    509 	}
    510 	os_free(cmd_buf);
    511 
    512 	for (;;) {
    513 		tv.tv_sec = 10;
    514 		tv.tv_usec = 0;
    515 		FD_ZERO(&rfds);
    516 		FD_SET(ctrl->s, &rfds);
    517 		res = select(ctrl->s + 1, &rfds, NULL, NULL, &tv);
    518 		if (res < 0)
    519 			return res;
    520 		if (FD_ISSET(ctrl->s, &rfds)) {
    521 			res = recv(ctrl->s, reply, *reply_len, 0);
    522 			if (res < 0)
    523 				return res;
    524 			if (res > 0 && reply[0] == '<') {
    525 				/* This is an unsolicited message from
    526 				 * wpa_supplicant, not the reply to the
    527 				 * request. Use msg_cb to report this to the
    528 				 * caller. */
    529 				if (msg_cb) {
    530 					/* Make sure the message is nul
    531 					 * terminated. */
    532 					if ((size_t) res == *reply_len)
    533 						res = (*reply_len) - 1;
    534 					reply[res] = '\0';
    535 					msg_cb(reply, res);
    536 				}
    537 				continue;
    538 			}
    539 			*reply_len = res;
    540 			break;
    541 		} else {
    542 			return -2;
    543 		}
    544 	}
    545 	return 0;
    546 }
    547 #endif /* CTRL_IFACE_SOCKET */
    548 
    549 
    550 static int wpa_ctrl_attach_helper(struct wpa_ctrl *ctrl, int attach)
    551 {
    552 	char buf[10];
    553 	int ret;
    554 	size_t len = 10;
    555 
    556 	ret = wpa_ctrl_request(ctrl, attach ? "ATTACH" : "DETACH", 6,
    557 			       buf, &len, NULL);
    558 	if (ret < 0)
    559 		return ret;
    560 	if (len == 3 && os_memcmp(buf, "OK\n", 3) == 0)
    561 		return 0;
    562 	return -1;
    563 }
    564 
    565 
    566 int wpa_ctrl_attach(struct wpa_ctrl *ctrl)
    567 {
    568 	return wpa_ctrl_attach_helper(ctrl, 1);
    569 }
    570 
    571 
    572 int wpa_ctrl_detach(struct wpa_ctrl *ctrl)
    573 {
    574 	return wpa_ctrl_attach_helper(ctrl, 0);
    575 }
    576 
    577 
    578 #ifdef CTRL_IFACE_SOCKET
    579 
    580 int wpa_ctrl_recv(struct wpa_ctrl *ctrl, char *reply, size_t *reply_len)
    581 {
    582 	int res;
    583 
    584 	res = recv(ctrl->s, reply, *reply_len, 0);
    585 	if (res < 0)
    586 		return res;
    587 	*reply_len = res;
    588 	return 0;
    589 }
    590 
    591 
    592 int wpa_ctrl_pending(struct wpa_ctrl *ctrl)
    593 {
    594 	struct timeval tv;
    595 	fd_set rfds;
    596 	tv.tv_sec = 0;
    597 	tv.tv_usec = 0;
    598 	FD_ZERO(&rfds);
    599 	FD_SET(ctrl->s, &rfds);
    600 	select(ctrl->s + 1, &rfds, NULL, NULL, &tv);
    601 	return FD_ISSET(ctrl->s, &rfds);
    602 }
    603 
    604 
    605 int wpa_ctrl_get_fd(struct wpa_ctrl *ctrl)
    606 {
    607 	return ctrl->s;
    608 }
    609 
    610 #endif /* CTRL_IFACE_SOCKET */
    611 
    612 
    613 #ifdef CONFIG_CTRL_IFACE_NAMED_PIPE
    614 
    615 #ifndef WPA_SUPPLICANT_NAMED_PIPE
    616 #define WPA_SUPPLICANT_NAMED_PIPE "WpaSupplicant"
    617 #endif
    618 #define NAMED_PIPE_PREFIX TEXT("\\\\.\\pipe\\") TEXT(WPA_SUPPLICANT_NAMED_PIPE)
    619 
    620 struct wpa_ctrl * wpa_ctrl_open(const char *ctrl_path)
    621 {
    622 	struct wpa_ctrl *ctrl;
    623 	DWORD mode;
    624 	TCHAR name[256];
    625 	int i, ret;
    626 
    627 	ctrl = os_malloc(sizeof(*ctrl));
    628 	if (ctrl == NULL)
    629 		return NULL;
    630 	os_memset(ctrl, 0, sizeof(*ctrl));
    631 
    632 #ifdef UNICODE
    633 	if (ctrl_path == NULL)
    634 		ret = _snwprintf(name, 256, NAMED_PIPE_PREFIX);
    635 	else
    636 		ret = _snwprintf(name, 256, NAMED_PIPE_PREFIX TEXT("-%S"),
    637 				 ctrl_path);
    638 #else /* UNICODE */
    639 	if (ctrl_path == NULL)
    640 		ret = os_snprintf(name, 256, NAMED_PIPE_PREFIX);
    641 	else
    642 		ret = os_snprintf(name, 256, NAMED_PIPE_PREFIX "-%s",
    643 				  ctrl_path);
    644 #endif /* UNICODE */
    645 	if (os_snprintf_error(256, ret)) {
    646 		os_free(ctrl);
    647 		return NULL;
    648 	}
    649 
    650 	for (i = 0; i < 10; i++) {
    651 		ctrl->pipe = CreateFile(name, GENERIC_READ | GENERIC_WRITE, 0,
    652 					NULL, OPEN_EXISTING, 0, NULL);
    653 		/*
    654 		 * Current named pipe server side in wpa_supplicant is
    655 		 * re-opening the pipe for new clients only after the previous
    656 		 * one is taken into use. This leaves a small window for race
    657 		 * conditions when two connections are being opened at almost
    658 		 * the same time. Retry if that was the case.
    659 		 */
    660 		if (ctrl->pipe != INVALID_HANDLE_VALUE ||
    661 		    GetLastError() != ERROR_PIPE_BUSY)
    662 			break;
    663 		WaitNamedPipe(name, 1000);
    664 	}
    665 	if (ctrl->pipe == INVALID_HANDLE_VALUE) {
    666 		os_free(ctrl);
    667 		return NULL;
    668 	}
    669 
    670 	mode = PIPE_READMODE_MESSAGE;
    671 	if (!SetNamedPipeHandleState(ctrl->pipe, &mode, NULL, NULL)) {
    672 		CloseHandle(ctrl->pipe);
    673 		os_free(ctrl);
    674 		return NULL;
    675 	}
    676 
    677 	return ctrl;
    678 }
    679 
    680 
    681 void wpa_ctrl_close(struct wpa_ctrl *ctrl)
    682 {
    683 	CloseHandle(ctrl->pipe);
    684 	os_free(ctrl);
    685 }
    686 
    687 
    688 int wpa_ctrl_request(struct wpa_ctrl *ctrl, const char *cmd, size_t cmd_len,
    689 		     char *reply, size_t *reply_len,
    690 		     void (*msg_cb)(char *msg, size_t len))
    691 {
    692 	DWORD written;
    693 	DWORD readlen = *reply_len;
    694 
    695 	if (!WriteFile(ctrl->pipe, cmd, cmd_len, &written, NULL))
    696 		return -1;
    697 
    698 	if (!ReadFile(ctrl->pipe, reply, *reply_len, &readlen, NULL))
    699 		return -1;
    700 	*reply_len = readlen;
    701 
    702 	return 0;
    703 }
    704 
    705 
    706 int wpa_ctrl_recv(struct wpa_ctrl *ctrl, char *reply, size_t *reply_len)
    707 {
    708 	DWORD len = *reply_len;
    709 	if (!ReadFile(ctrl->pipe, reply, *reply_len, &len, NULL))
    710 		return -1;
    711 	*reply_len = len;
    712 	return 0;
    713 }
    714 
    715 
    716 int wpa_ctrl_pending(struct wpa_ctrl *ctrl)
    717 {
    718 	DWORD left;
    719 
    720 	if (!PeekNamedPipe(ctrl->pipe, NULL, 0, NULL, &left, NULL))
    721 		return -1;
    722 	return left ? 1 : 0;
    723 }
    724 
    725 
    726 int wpa_ctrl_get_fd(struct wpa_ctrl *ctrl)
    727 {
    728 	return -1;
    729 }
    730 
    731 #endif /* CONFIG_CTRL_IFACE_NAMED_PIPE */
    732 
    733 #endif /* CONFIG_CTRL_IFACE */
    734