1/*
2 * wpa_supplicant ctrl_iface helpers
3 * Copyright (c) 2010-2011, Atheros Communications, Inc.
4 * Copyright (c) 2011-2012, Qualcomm Atheros, Inc.
5 *
6 * This software may be distributed under the terms of the BSD license.
7 * See README for more details.
8 */
9
10#include "includes.h"
11#include <time.h>
12
13#include "common.h"
14#include "wpa_ctrl.h"
15#include "wpa_helpers.h"
16
17
18char *wpas_ctrl_path = "/var/run/wpa_supplicant/";
19static int default_timeout = 60;
20
21
22static struct wpa_ctrl * wpa_open_ctrl(const char *ifname)
23{
24	char buf[128];
25	struct wpa_ctrl *ctrl;
26
27	os_snprintf(buf, sizeof(buf), "%s%s", wpas_ctrl_path, ifname);
28	ctrl = wpa_ctrl_open(buf);
29	if (ctrl == NULL)
30		printf("wpa_command: wpa_ctrl_open(%s) failed\n", buf);
31	return ctrl;
32}
33
34
35int wpa_command(const char *ifname, const char *cmd)
36{
37	struct wpa_ctrl *ctrl;
38	char buf[128];
39	size_t len;
40
41	printf("wpa_command(ifname='%s', cmd='%s')\n", ifname, cmd);
42	ctrl = wpa_open_ctrl(ifname);
43	if (ctrl == NULL)
44		return -1;
45	len = sizeof(buf);
46	if (wpa_ctrl_request(ctrl, cmd, strlen(cmd), buf, &len, NULL) < 0) {
47		printf("wpa_command: wpa_ctrl_request failed\n");
48		wpa_ctrl_close(ctrl);
49		return -1;
50	}
51	wpa_ctrl_close(ctrl);
52	buf[len] = '\0';
53	if (strncmp(buf, "FAIL", 4) == 0) {
54		printf("wpa_command: Command failed (FAIL received)\n");
55		return -1;
56	}
57	return 0;
58}
59
60
61int wpa_command_resp(const char *ifname, const char *cmd,
62		     char *resp, size_t resp_size)
63{
64	struct wpa_ctrl *ctrl;
65	size_t len;
66
67	printf("wpa_command(ifname='%s', cmd='%s')\n", ifname, cmd);
68	ctrl = wpa_open_ctrl(ifname);
69	if (ctrl == NULL)
70		return -1;
71	len = resp_size;
72	if (wpa_ctrl_request(ctrl, cmd, strlen(cmd), resp, &len, NULL) < 0) {
73		printf("wpa_command: wpa_ctrl_request failed\n");
74		wpa_ctrl_close(ctrl);
75		return -1;
76	}
77	wpa_ctrl_close(ctrl);
78	resp[len] = '\0';
79	return 0;
80}
81
82
83struct wpa_ctrl * open_wpa_mon(const char *ifname)
84{
85	struct wpa_ctrl *ctrl;
86
87	ctrl = wpa_open_ctrl(ifname);
88	if (ctrl == NULL)
89		return NULL;
90	if (wpa_ctrl_attach(ctrl) < 0) {
91		wpa_ctrl_close(ctrl);
92		return NULL;
93	}
94
95	return ctrl;
96}
97
98
99int get_wpa_cli_event2(struct wpa_ctrl *mon,
100		       const char *event, const char *event2,
101		       char *buf, size_t buf_size)
102{
103	int fd, ret;
104	fd_set rfd;
105	char *pos;
106	struct timeval tv;
107	time_t start, now;
108
109	printf("Waiting for wpa_cli event %s\n", event);
110	fd = wpa_ctrl_get_fd(mon);
111	if (fd < 0)
112		return -1;
113
114	time(&start);
115	while (1) {
116		size_t len;
117
118		FD_ZERO(&rfd);
119		FD_SET(fd, &rfd);
120		tv.tv_sec = default_timeout;
121		tv.tv_usec = 0;
122		ret = select(fd + 1, &rfd, NULL, NULL, &tv);
123		if (ret == 0) {
124			printf("Timeout on waiting for event %s\n", event);
125			return -1;
126		}
127		if (ret < 0) {
128			printf("select: %s\n", strerror(errno));
129			return -1;
130		}
131		len = buf_size;
132		if (wpa_ctrl_recv(mon, buf, &len) < 0) {
133			printf("Failure while waiting for event %s\n", event);
134			return -1;
135		}
136		if (len == buf_size)
137			len--;
138		buf[len] = '\0';
139
140		pos = strchr(buf, '>');
141		if (pos &&
142		    (strncmp(pos + 1, event, strlen(event)) == 0 ||
143		     (event2 &&
144		      strncmp(pos + 1, event2, strlen(event2)) == 0)))
145			return 0; /* Event found */
146
147		time(&now);
148		if ((int) (now - start) > default_timeout) {
149			printf("Timeout on waiting for event %s\n", event);
150			return -1;
151		}
152	}
153}
154
155
156int get_wpa_cli_event(struct wpa_ctrl *mon,
157		      const char *event, char *buf, size_t buf_size)
158{
159	return get_wpa_cli_event2(mon, event, NULL, buf, buf_size);
160}
161
162
163int get_wpa_status(const char *ifname, const char *field, char *obuf,
164		   size_t obuf_size)
165{
166	struct wpa_ctrl *ctrl;
167	char buf[4096];
168	char *pos, *end;
169	size_t len, flen;
170
171	ctrl = wpa_open_ctrl(ifname);
172	if (ctrl == NULL)
173		return -1;
174	len = sizeof(buf);
175	if (wpa_ctrl_request(ctrl, "STATUS-NO_EVENTS", 16, buf, &len,
176			     NULL) < 0) {
177		wpa_ctrl_close(ctrl);
178		return -1;
179	}
180	wpa_ctrl_close(ctrl);
181	buf[len] = '\0';
182
183	flen = strlen(field);
184	pos = buf;
185	while (pos + flen < buf + len) {
186		if (pos > buf) {
187			if (*pos != '\n') {
188				pos++;
189				continue;
190			}
191			pos++;
192		}
193		if (strncmp(pos, field, flen) != 0 || pos[flen] != '=') {
194			pos++;
195			continue;
196		}
197		pos += flen + 1;
198		end = strchr(pos, '\n');
199		if (end == NULL)
200			return -1;
201		*end++ = '\0';
202		if (end - pos > (int) obuf_size)
203			return -1;
204		memcpy(obuf, pos, end - pos);
205		return 0;
206	}
207
208	return -1;
209}
210
211
212int wait_ip_addr(const char *ifname, int timeout)
213{
214	char ip[30];
215	int count = timeout;
216	struct wpa_ctrl *ctrl;
217
218	while (count > 0) {
219		printf("%s: ifname='%s' - %d seconds remaining\n",
220		       __func__, ifname, count);
221		count--;
222		if (get_wpa_status(ifname, "ip_address", ip, sizeof(ip)) == 0
223		    && strlen(ip) > 0) {
224			printf("IP address found: '%s'\n", ip);
225			return 0;
226		}
227		ctrl = wpa_open_ctrl(ifname);
228		if (ctrl == NULL)
229			return -1;
230		wpa_ctrl_close(ctrl);
231		sleep(1);
232	}
233	printf("%s: Could not get IP address for ifname='%s'", __func__,
234	       ifname);
235	return -1;
236}
237
238
239int add_network(const char *ifname)
240{
241	char res[30];
242
243	if (wpa_command_resp(ifname, "ADD_NETWORK", res, sizeof(res)) < 0)
244		return -1;
245	return atoi(res);
246}
247
248
249int set_network(const char *ifname, int id, const char *field,
250		const char *value)
251{
252	char buf[200];
253	snprintf(buf, sizeof(buf), "SET_NETWORK %d %s %s", id, field, value);
254	return wpa_command(ifname, buf);
255}
256
257
258int set_network_quoted(const char *ifname, int id, const char *field,
259		       const char *value)
260{
261	char buf[200];
262	snprintf(buf, sizeof(buf), "SET_NETWORK %d %s \"%s\"",
263		 id, field, value);
264	return wpa_command(ifname, buf);
265}
266
267
268int add_cred(const char *ifname)
269{
270	char res[30];
271
272	if (wpa_command_resp(ifname, "ADD_CRED", res, sizeof(res)) < 0)
273		return -1;
274	return atoi(res);
275}
276
277
278int set_cred(const char *ifname, int id, const char *field, const char *value)
279{
280	char buf[200];
281	snprintf(buf, sizeof(buf), "SET_CRED %d %s %s", id, field, value);
282	return wpa_command(ifname, buf);
283}
284
285
286int set_cred_quoted(const char *ifname, int id, const char *field,
287		    const char *value)
288{
289	char buf[200];
290	snprintf(buf, sizeof(buf), "SET_CRED %d %s \"%s\"",
291		 id, field, value);
292	return wpa_command(ifname, buf);
293}
294