wpa_ctrl.c 17 KB

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