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