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 
18 char *wpas_ctrl_path = "/var/run/wpa_supplicant/";
19 static int default_timeout = 60;
20 
21 
wpa_open_ctrl(const char * ifname)22 static 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 
wpa_command(const char * ifname,const char * cmd)35 int 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 
wpa_command_resp(const char * ifname,const char * cmd,char * resp,size_t resp_size)61 int 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 
open_wpa_mon(const char * ifname)83 struct 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 
get_wpa_cli_event2(struct wpa_ctrl * mon,const char * event,const char * event2,char * buf,size_t buf_size)99 int 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 
get_wpa_cli_event(struct wpa_ctrl * mon,const char * event,char * buf,size_t buf_size)156 int 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 
get_wpa_status(const char * ifname,const char * field,char * obuf,size_t obuf_size)163 int 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 
wait_ip_addr(const char * ifname,int timeout)212 int 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 			if (strncmp(ip, "169.254.", 8) != 0)
226 				return 0;
227 		}
228 		ctrl = wpa_open_ctrl(ifname);
229 		if (ctrl == NULL)
230 			return -1;
231 		wpa_ctrl_close(ctrl);
232 		sleep(1);
233 	}
234 	printf("%s: Could not get IP address for ifname='%s'", __func__,
235 	       ifname);
236 	return -1;
237 }
238 
239 
add_network(const char * ifname)240 int add_network(const char *ifname)
241 {
242 	char res[30];
243 
244 	if (wpa_command_resp(ifname, "ADD_NETWORK", res, sizeof(res)) < 0)
245 		return -1;
246 	return atoi(res);
247 }
248 
249 
set_network(const char * ifname,int id,const char * field,const char * value)250 int set_network(const char *ifname, int id, const char *field,
251 		const char *value)
252 {
253 	char buf[200];
254 	snprintf(buf, sizeof(buf), "SET_NETWORK %d %s %s", id, field, value);
255 	return wpa_command(ifname, buf);
256 }
257 
258 
set_network_quoted(const char * ifname,int id,const char * field,const char * value)259 int set_network_quoted(const char *ifname, int id, const char *field,
260 		       const char *value)
261 {
262 	char buf[200];
263 	snprintf(buf, sizeof(buf), "SET_NETWORK %d %s \"%s\"",
264 		 id, field, value);
265 	return wpa_command(ifname, buf);
266 }
267 
268 
add_cred(const char * ifname)269 int add_cred(const char *ifname)
270 {
271 	char res[30];
272 
273 	if (wpa_command_resp(ifname, "ADD_CRED", res, sizeof(res)) < 0)
274 		return -1;
275 	return atoi(res);
276 }
277 
278 
set_cred(const char * ifname,int id,const char * field,const char * value)279 int set_cred(const char *ifname, int id, const char *field, const char *value)
280 {
281 	char buf[200];
282 	snprintf(buf, sizeof(buf), "SET_CRED %d %s %s", id, field, value);
283 	return wpa_command(ifname, buf);
284 }
285 
286 
set_cred_quoted(const char * ifname,int id,const char * field,const char * value)287 int set_cred_quoted(const char *ifname, int id, const char *field,
288 		    const char *value)
289 {
290 	char buf[200];
291 	snprintf(buf, sizeof(buf), "SET_CRED %d %s \"%s\"",
292 		 id, field, value);
293 	return wpa_command(ifname, buf);
294 }
295