wpa_priv.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032
  1. /*
  2. * WPA Supplicant / privileged helper program
  3. * Copyright (c) 2007-2009, 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 __linux__
  10. #include <fcntl.h>
  11. #endif /* __linux__ */
  12. #include <sys/un.h>
  13. #include <sys/stat.h>
  14. #include "common.h"
  15. #include "eloop.h"
  16. #include "common/version.h"
  17. #include "drivers/driver.h"
  18. #include "l2_packet/l2_packet.h"
  19. #include "common/privsep_commands.h"
  20. #include "common/ieee802_11_defs.h"
  21. struct wpa_priv_interface {
  22. struct wpa_priv_interface *next;
  23. char *driver_name;
  24. char *ifname;
  25. char *sock_name;
  26. int fd;
  27. struct wpa_driver_ops *driver;
  28. void *drv_priv;
  29. struct sockaddr_un drv_addr;
  30. int wpas_registered;
  31. /* TODO: add support for multiple l2 connections */
  32. struct l2_packet_data *l2;
  33. struct sockaddr_un l2_addr;
  34. };
  35. static void wpa_priv_cmd_register(struct wpa_priv_interface *iface,
  36. struct sockaddr_un *from)
  37. {
  38. if (iface->drv_priv) {
  39. wpa_printf(MSG_DEBUG, "Cleaning up forgotten driver instance");
  40. if (iface->driver->deinit)
  41. iface->driver->deinit(iface->drv_priv);
  42. iface->drv_priv = NULL;
  43. iface->wpas_registered = 0;
  44. }
  45. if (iface->l2) {
  46. wpa_printf(MSG_DEBUG, "Cleaning up forgotten l2_packet "
  47. "instance");
  48. l2_packet_deinit(iface->l2);
  49. iface->l2 = NULL;
  50. }
  51. if (iface->driver->init == NULL)
  52. return;
  53. iface->drv_priv = iface->driver->init(iface, iface->ifname);
  54. if (iface->drv_priv == NULL) {
  55. wpa_printf(MSG_DEBUG, "Failed to initialize driver wrapper");
  56. return;
  57. }
  58. wpa_printf(MSG_DEBUG, "Driver wrapper '%s' initialized for interface "
  59. "'%s'", iface->driver_name, iface->ifname);
  60. os_memcpy(&iface->drv_addr, from, sizeof(iface->drv_addr));
  61. iface->wpas_registered = 1;
  62. if (iface->driver->set_param &&
  63. iface->driver->set_param(iface->drv_priv, NULL) < 0) {
  64. wpa_printf(MSG_ERROR, "Driver interface rejected param");
  65. }
  66. }
  67. static void wpa_priv_cmd_unregister(struct wpa_priv_interface *iface,
  68. struct sockaddr_un *from)
  69. {
  70. if (iface->drv_priv) {
  71. if (iface->driver->deinit)
  72. iface->driver->deinit(iface->drv_priv);
  73. iface->drv_priv = NULL;
  74. iface->wpas_registered = 0;
  75. }
  76. }
  77. static void wpa_priv_cmd_scan(struct wpa_priv_interface *iface,
  78. char *buf, size_t len)
  79. {
  80. struct wpa_driver_scan_params params;
  81. if (iface->drv_priv == NULL)
  82. return;
  83. os_memset(&params, 0, sizeof(params));
  84. if (len) {
  85. params.ssids[0].ssid = (u8 *) buf;
  86. params.ssids[0].ssid_len = len;
  87. params.num_ssids = 1;
  88. }
  89. if (iface->driver->scan2)
  90. iface->driver->scan2(iface->drv_priv, &params);
  91. }
  92. static void wpa_priv_get_scan_results2(struct wpa_priv_interface *iface,
  93. struct sockaddr_un *from)
  94. {
  95. struct wpa_scan_results *res;
  96. u8 *buf = NULL, *pos, *end;
  97. int val;
  98. size_t i;
  99. res = iface->driver->get_scan_results2(iface->drv_priv);
  100. if (res == NULL)
  101. goto fail;
  102. buf = os_malloc(60000);
  103. if (buf == NULL)
  104. goto fail;
  105. pos = buf;
  106. end = buf + 60000;
  107. val = res->num;
  108. os_memcpy(pos, &val, sizeof(int));
  109. pos += sizeof(int);
  110. for (i = 0; i < res->num; i++) {
  111. struct wpa_scan_res *r = res->res[i];
  112. val = sizeof(*r) + r->ie_len;
  113. if (end - pos < (int) sizeof(int) + val)
  114. break;
  115. os_memcpy(pos, &val, sizeof(int));
  116. pos += sizeof(int);
  117. os_memcpy(pos, r, val);
  118. pos += val;
  119. }
  120. sendto(iface->fd, buf, pos - buf, 0, (struct sockaddr *) from,
  121. sizeof(*from));
  122. os_free(buf);
  123. wpa_scan_results_free(res);
  124. return;
  125. fail:
  126. os_free(buf);
  127. wpa_scan_results_free(res);
  128. sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
  129. }
  130. static void wpa_priv_cmd_get_scan_results(struct wpa_priv_interface *iface,
  131. struct sockaddr_un *from)
  132. {
  133. if (iface->drv_priv == NULL)
  134. return;
  135. if (iface->driver->get_scan_results2)
  136. wpa_priv_get_scan_results2(iface, from);
  137. else
  138. sendto(iface->fd, "", 0, 0, (struct sockaddr *) from,
  139. sizeof(*from));
  140. }
  141. static void wpa_priv_cmd_associate(struct wpa_priv_interface *iface,
  142. void *buf, size_t len)
  143. {
  144. struct wpa_driver_associate_params params;
  145. struct privsep_cmd_associate *assoc;
  146. u8 *bssid;
  147. int res;
  148. if (iface->drv_priv == NULL || iface->driver->associate == NULL)
  149. return;
  150. if (len < sizeof(*assoc)) {
  151. wpa_printf(MSG_DEBUG, "Invalid association request");
  152. return;
  153. }
  154. assoc = buf;
  155. if (sizeof(*assoc) + assoc->wpa_ie_len > len) {
  156. wpa_printf(MSG_DEBUG, "Association request overflow");
  157. return;
  158. }
  159. os_memset(&params, 0, sizeof(params));
  160. bssid = assoc->bssid;
  161. if (bssid[0] | bssid[1] | bssid[2] | bssid[3] | bssid[4] | bssid[5])
  162. params.bssid = bssid;
  163. params.ssid = assoc->ssid;
  164. if (assoc->ssid_len > 32)
  165. return;
  166. params.ssid_len = assoc->ssid_len;
  167. params.freq = assoc->freq;
  168. if (assoc->wpa_ie_len) {
  169. params.wpa_ie = (u8 *) (assoc + 1);
  170. params.wpa_ie_len = assoc->wpa_ie_len;
  171. }
  172. params.pairwise_suite = assoc->pairwise_suite;
  173. params.group_suite = assoc->group_suite;
  174. params.key_mgmt_suite = assoc->key_mgmt_suite;
  175. params.auth_alg = assoc->auth_alg;
  176. params.mode = assoc->mode;
  177. res = iface->driver->associate(iface->drv_priv, &params);
  178. wpa_printf(MSG_DEBUG, "drv->associate: res=%d", res);
  179. }
  180. static void wpa_priv_cmd_get_bssid(struct wpa_priv_interface *iface,
  181. struct sockaddr_un *from)
  182. {
  183. u8 bssid[ETH_ALEN];
  184. if (iface->drv_priv == NULL)
  185. goto fail;
  186. if (iface->driver->get_bssid == NULL ||
  187. iface->driver->get_bssid(iface->drv_priv, bssid) < 0)
  188. goto fail;
  189. sendto(iface->fd, bssid, ETH_ALEN, 0, (struct sockaddr *) from,
  190. sizeof(*from));
  191. return;
  192. fail:
  193. sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
  194. }
  195. static void wpa_priv_cmd_get_ssid(struct wpa_priv_interface *iface,
  196. struct sockaddr_un *from)
  197. {
  198. u8 ssid[sizeof(int) + 32];
  199. int res;
  200. if (iface->drv_priv == NULL)
  201. goto fail;
  202. if (iface->driver->get_ssid == NULL)
  203. goto fail;
  204. res = iface->driver->get_ssid(iface->drv_priv, &ssid[sizeof(int)]);
  205. if (res < 0 || res > 32)
  206. goto fail;
  207. os_memcpy(ssid, &res, sizeof(int));
  208. sendto(iface->fd, ssid, sizeof(ssid), 0, (struct sockaddr *) from,
  209. sizeof(*from));
  210. return;
  211. fail:
  212. sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
  213. }
  214. static void wpa_priv_cmd_set_key(struct wpa_priv_interface *iface,
  215. void *buf, size_t len)
  216. {
  217. struct privsep_cmd_set_key *params;
  218. int res;
  219. if (iface->drv_priv == NULL || iface->driver->set_key == NULL)
  220. return;
  221. if (len != sizeof(*params)) {
  222. wpa_printf(MSG_DEBUG, "Invalid set_key request");
  223. return;
  224. }
  225. params = buf;
  226. res = iface->driver->set_key(iface->ifname, iface->drv_priv,
  227. params->alg,
  228. params->addr, params->key_idx,
  229. params->set_tx,
  230. params->seq_len ? params->seq : NULL,
  231. params->seq_len,
  232. params->key_len ? params->key : NULL,
  233. params->key_len);
  234. wpa_printf(MSG_DEBUG, "drv->set_key: res=%d", res);
  235. }
  236. static void wpa_priv_cmd_get_capa(struct wpa_priv_interface *iface,
  237. struct sockaddr_un *from)
  238. {
  239. struct wpa_driver_capa capa;
  240. if (iface->drv_priv == NULL)
  241. goto fail;
  242. if (iface->driver->get_capa == NULL ||
  243. iface->driver->get_capa(iface->drv_priv, &capa) < 0)
  244. goto fail;
  245. sendto(iface->fd, &capa, sizeof(capa), 0, (struct sockaddr *) from,
  246. sizeof(*from));
  247. return;
  248. fail:
  249. sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
  250. }
  251. static void wpa_priv_l2_rx(void *ctx, const u8 *src_addr, const u8 *buf,
  252. size_t len)
  253. {
  254. struct wpa_priv_interface *iface = ctx;
  255. struct msghdr msg;
  256. struct iovec io[2];
  257. io[0].iov_base = (u8 *) src_addr;
  258. io[0].iov_len = ETH_ALEN;
  259. io[1].iov_base = (u8 *) buf;
  260. io[1].iov_len = len;
  261. os_memset(&msg, 0, sizeof(msg));
  262. msg.msg_iov = io;
  263. msg.msg_iovlen = 2;
  264. msg.msg_name = &iface->l2_addr;
  265. msg.msg_namelen = sizeof(iface->l2_addr);
  266. if (sendmsg(iface->fd, &msg, 0) < 0) {
  267. perror("sendmsg(l2 rx)");
  268. }
  269. }
  270. static void wpa_priv_cmd_l2_register(struct wpa_priv_interface *iface,
  271. struct sockaddr_un *from,
  272. void *buf, size_t len)
  273. {
  274. int *reg_cmd = buf;
  275. u8 own_addr[ETH_ALEN];
  276. int res;
  277. u16 proto;
  278. if (len != 2 * sizeof(int)) {
  279. wpa_printf(MSG_DEBUG, "Invalid l2_register length %lu",
  280. (unsigned long) len);
  281. return;
  282. }
  283. proto = reg_cmd[0];
  284. if (proto != ETH_P_EAPOL && proto != ETH_P_RSN_PREAUTH) {
  285. wpa_printf(MSG_DEBUG, "Refused l2_packet connection for "
  286. "ethertype 0x%x", proto);
  287. return;
  288. }
  289. if (iface->l2) {
  290. wpa_printf(MSG_DEBUG, "Cleaning up forgotten l2_packet "
  291. "instance");
  292. l2_packet_deinit(iface->l2);
  293. iface->l2 = NULL;
  294. }
  295. os_memcpy(&iface->l2_addr, from, sizeof(iface->l2_addr));
  296. iface->l2 = l2_packet_init(iface->ifname, NULL, proto,
  297. wpa_priv_l2_rx, iface, reg_cmd[1]);
  298. if (iface->l2 == NULL) {
  299. wpa_printf(MSG_DEBUG, "Failed to initialize l2_packet "
  300. "instance for protocol %d", proto);
  301. return;
  302. }
  303. if (l2_packet_get_own_addr(iface->l2, own_addr) < 0) {
  304. wpa_printf(MSG_DEBUG, "Failed to get own address from "
  305. "l2_packet");
  306. l2_packet_deinit(iface->l2);
  307. iface->l2 = NULL;
  308. return;
  309. }
  310. res = sendto(iface->fd, own_addr, ETH_ALEN, 0,
  311. (struct sockaddr *) from, sizeof(*from));
  312. wpa_printf(MSG_DEBUG, "L2 registration: res=%d", res);
  313. }
  314. static void wpa_priv_cmd_l2_unregister(struct wpa_priv_interface *iface,
  315. struct sockaddr_un *from)
  316. {
  317. if (iface->l2) {
  318. l2_packet_deinit(iface->l2);
  319. iface->l2 = NULL;
  320. }
  321. }
  322. static void wpa_priv_cmd_l2_notify_auth_start(struct wpa_priv_interface *iface,
  323. struct sockaddr_un *from)
  324. {
  325. if (iface->l2)
  326. l2_packet_notify_auth_start(iface->l2);
  327. }
  328. static void wpa_priv_cmd_l2_send(struct wpa_priv_interface *iface,
  329. struct sockaddr_un *from,
  330. void *buf, size_t len)
  331. {
  332. u8 *dst_addr;
  333. u16 proto;
  334. int res;
  335. if (iface->l2 == NULL)
  336. return;
  337. if (len < ETH_ALEN + 2) {
  338. wpa_printf(MSG_DEBUG, "Too short L2 send packet (len=%lu)",
  339. (unsigned long) len);
  340. return;
  341. }
  342. dst_addr = buf;
  343. os_memcpy(&proto, buf + ETH_ALEN, 2);
  344. if (proto != ETH_P_EAPOL && proto != ETH_P_RSN_PREAUTH) {
  345. wpa_printf(MSG_DEBUG, "Refused l2_packet send for ethertype "
  346. "0x%x", proto);
  347. return;
  348. }
  349. res = l2_packet_send(iface->l2, dst_addr, proto, buf + ETH_ALEN + 2,
  350. len - ETH_ALEN - 2);
  351. wpa_printf(MSG_DEBUG, "L2 send: res=%d", res);
  352. }
  353. static void wpa_priv_cmd_set_country(struct wpa_priv_interface *iface,
  354. char *buf)
  355. {
  356. if (iface->drv_priv == NULL || iface->driver->set_country == NULL ||
  357. *buf == '\0')
  358. return;
  359. iface->driver->set_country(iface->drv_priv, buf);
  360. }
  361. static void wpa_priv_receive(int sock, void *eloop_ctx, void *sock_ctx)
  362. {
  363. struct wpa_priv_interface *iface = eloop_ctx;
  364. char buf[2000], *pos;
  365. void *cmd_buf;
  366. size_t cmd_len;
  367. int res, cmd;
  368. struct sockaddr_un from;
  369. socklen_t fromlen = sizeof(from);
  370. res = recvfrom(sock, buf, sizeof(buf), 0, (struct sockaddr *) &from,
  371. &fromlen);
  372. if (res < 0) {
  373. perror("recvfrom");
  374. return;
  375. }
  376. if (res < (int) sizeof(int)) {
  377. wpa_printf(MSG_DEBUG, "Too short command (len=%d)", res);
  378. return;
  379. }
  380. os_memcpy(&cmd, buf, sizeof(int));
  381. wpa_printf(MSG_DEBUG, "Command %d for interface %s",
  382. cmd, iface->ifname);
  383. cmd_buf = &buf[sizeof(int)];
  384. cmd_len = res - sizeof(int);
  385. switch (cmd) {
  386. case PRIVSEP_CMD_REGISTER:
  387. wpa_priv_cmd_register(iface, &from);
  388. break;
  389. case PRIVSEP_CMD_UNREGISTER:
  390. wpa_priv_cmd_unregister(iface, &from);
  391. break;
  392. case PRIVSEP_CMD_SCAN:
  393. wpa_priv_cmd_scan(iface, cmd_buf, cmd_len);
  394. break;
  395. case PRIVSEP_CMD_GET_SCAN_RESULTS:
  396. wpa_priv_cmd_get_scan_results(iface, &from);
  397. break;
  398. case PRIVSEP_CMD_ASSOCIATE:
  399. wpa_priv_cmd_associate(iface, cmd_buf, cmd_len);
  400. break;
  401. case PRIVSEP_CMD_GET_BSSID:
  402. wpa_priv_cmd_get_bssid(iface, &from);
  403. break;
  404. case PRIVSEP_CMD_GET_SSID:
  405. wpa_priv_cmd_get_ssid(iface, &from);
  406. break;
  407. case PRIVSEP_CMD_SET_KEY:
  408. wpa_priv_cmd_set_key(iface, cmd_buf, cmd_len);
  409. break;
  410. case PRIVSEP_CMD_GET_CAPA:
  411. wpa_priv_cmd_get_capa(iface, &from);
  412. break;
  413. case PRIVSEP_CMD_L2_REGISTER:
  414. wpa_priv_cmd_l2_register(iface, &from, cmd_buf, cmd_len);
  415. break;
  416. case PRIVSEP_CMD_L2_UNREGISTER:
  417. wpa_priv_cmd_l2_unregister(iface, &from);
  418. break;
  419. case PRIVSEP_CMD_L2_NOTIFY_AUTH_START:
  420. wpa_priv_cmd_l2_notify_auth_start(iface, &from);
  421. break;
  422. case PRIVSEP_CMD_L2_SEND:
  423. wpa_priv_cmd_l2_send(iface, &from, cmd_buf, cmd_len);
  424. break;
  425. case PRIVSEP_CMD_SET_COUNTRY:
  426. pos = cmd_buf;
  427. if (pos + cmd_len >= buf + sizeof(buf))
  428. break;
  429. pos[cmd_len] = '\0';
  430. wpa_priv_cmd_set_country(iface, pos);
  431. break;
  432. }
  433. }
  434. static void wpa_priv_interface_deinit(struct wpa_priv_interface *iface)
  435. {
  436. if (iface->drv_priv && iface->driver->deinit)
  437. iface->driver->deinit(iface->drv_priv);
  438. if (iface->fd >= 0) {
  439. eloop_unregister_read_sock(iface->fd);
  440. close(iface->fd);
  441. unlink(iface->sock_name);
  442. }
  443. if (iface->l2)
  444. l2_packet_deinit(iface->l2);
  445. os_free(iface->ifname);
  446. os_free(iface->driver_name);
  447. os_free(iface->sock_name);
  448. os_free(iface);
  449. }
  450. extern struct wpa_driver_ops *wpa_drivers[];
  451. static struct wpa_priv_interface *
  452. wpa_priv_interface_init(const char *dir, const char *params)
  453. {
  454. struct wpa_priv_interface *iface;
  455. char *pos;
  456. size_t len;
  457. struct sockaddr_un addr;
  458. int i;
  459. pos = os_strchr(params, ':');
  460. if (pos == NULL)
  461. return NULL;
  462. iface = os_zalloc(sizeof(*iface));
  463. if (iface == NULL)
  464. return NULL;
  465. iface->fd = -1;
  466. len = pos - params;
  467. iface->driver_name = dup_binstr(params, len);
  468. if (iface->driver_name == NULL) {
  469. wpa_priv_interface_deinit(iface);
  470. return NULL;
  471. }
  472. for (i = 0; wpa_drivers[i]; i++) {
  473. if (os_strcmp(iface->driver_name,
  474. wpa_drivers[i]->name) == 0) {
  475. iface->driver = wpa_drivers[i];
  476. break;
  477. }
  478. }
  479. if (iface->driver == NULL) {
  480. wpa_printf(MSG_ERROR, "Unsupported driver '%s'",
  481. iface->driver_name);
  482. wpa_priv_interface_deinit(iface);
  483. return NULL;
  484. }
  485. pos++;
  486. iface->ifname = os_strdup(pos);
  487. if (iface->ifname == NULL) {
  488. wpa_priv_interface_deinit(iface);
  489. return NULL;
  490. }
  491. len = os_strlen(dir) + 1 + os_strlen(iface->ifname);
  492. iface->sock_name = os_malloc(len + 1);
  493. if (iface->sock_name == NULL) {
  494. wpa_priv_interface_deinit(iface);
  495. return NULL;
  496. }
  497. os_snprintf(iface->sock_name, len + 1, "%s/%s", dir, iface->ifname);
  498. if (os_strlen(iface->sock_name) >= sizeof(addr.sun_path)) {
  499. wpa_priv_interface_deinit(iface);
  500. return NULL;
  501. }
  502. iface->fd = socket(PF_UNIX, SOCK_DGRAM, 0);
  503. if (iface->fd < 0) {
  504. perror("socket(PF_UNIX)");
  505. wpa_priv_interface_deinit(iface);
  506. return NULL;
  507. }
  508. os_memset(&addr, 0, sizeof(addr));
  509. addr.sun_family = AF_UNIX;
  510. os_strlcpy(addr.sun_path, iface->sock_name, sizeof(addr.sun_path));
  511. if (bind(iface->fd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
  512. wpa_printf(MSG_DEBUG, "bind(PF_UNIX) failed: %s",
  513. strerror(errno));
  514. if (connect(iface->fd, (struct sockaddr *) &addr,
  515. sizeof(addr)) < 0) {
  516. wpa_printf(MSG_DEBUG, "Socket exists, but does not "
  517. "allow connections - assuming it was "
  518. "leftover from forced program termination");
  519. if (unlink(iface->sock_name) < 0) {
  520. perror("unlink[ctrl_iface]");
  521. wpa_printf(MSG_ERROR, "Could not unlink "
  522. "existing ctrl_iface socket '%s'",
  523. iface->sock_name);
  524. goto fail;
  525. }
  526. if (bind(iface->fd, (struct sockaddr *) &addr,
  527. sizeof(addr)) < 0) {
  528. perror("wpa-priv-iface-init: bind(PF_UNIX)");
  529. goto fail;
  530. }
  531. wpa_printf(MSG_DEBUG, "Successfully replaced leftover "
  532. "socket '%s'", iface->sock_name);
  533. } else {
  534. wpa_printf(MSG_INFO, "Socket exists and seems to be "
  535. "in use - cannot override it");
  536. wpa_printf(MSG_INFO, "Delete '%s' manually if it is "
  537. "not used anymore", iface->sock_name);
  538. goto fail;
  539. }
  540. }
  541. if (chmod(iface->sock_name, S_IRWXU | S_IRWXG | S_IRWXO) < 0) {
  542. perror("chmod");
  543. goto fail;
  544. }
  545. eloop_register_read_sock(iface->fd, wpa_priv_receive, iface, NULL);
  546. return iface;
  547. fail:
  548. wpa_priv_interface_deinit(iface);
  549. return NULL;
  550. }
  551. static int wpa_priv_send_event(struct wpa_priv_interface *iface, int event,
  552. const void *data, size_t data_len)
  553. {
  554. struct msghdr msg;
  555. struct iovec io[2];
  556. io[0].iov_base = &event;
  557. io[0].iov_len = sizeof(event);
  558. io[1].iov_base = (u8 *) data;
  559. io[1].iov_len = data_len;
  560. os_memset(&msg, 0, sizeof(msg));
  561. msg.msg_iov = io;
  562. msg.msg_iovlen = data ? 2 : 1;
  563. msg.msg_name = &iface->drv_addr;
  564. msg.msg_namelen = sizeof(iface->drv_addr);
  565. if (sendmsg(iface->fd, &msg, 0) < 0) {
  566. perror("sendmsg(wpas_socket)");
  567. return -1;
  568. }
  569. return 0;
  570. }
  571. static void wpa_priv_send_assoc(struct wpa_priv_interface *iface, int event,
  572. union wpa_event_data *data)
  573. {
  574. size_t buflen = 3 * sizeof(int);
  575. u8 *buf, *pos;
  576. int len;
  577. if (data) {
  578. buflen += data->assoc_info.req_ies_len +
  579. data->assoc_info.resp_ies_len +
  580. data->assoc_info.beacon_ies_len;
  581. }
  582. buf = os_malloc(buflen);
  583. if (buf == NULL)
  584. return;
  585. pos = buf;
  586. if (data && data->assoc_info.req_ies) {
  587. len = data->assoc_info.req_ies_len;
  588. os_memcpy(pos, &len, sizeof(int));
  589. pos += sizeof(int);
  590. os_memcpy(pos, data->assoc_info.req_ies, len);
  591. pos += len;
  592. } else {
  593. len = 0;
  594. os_memcpy(pos, &len, sizeof(int));
  595. pos += sizeof(int);
  596. }
  597. if (data && data->assoc_info.resp_ies) {
  598. len = data->assoc_info.resp_ies_len;
  599. os_memcpy(pos, &len, sizeof(int));
  600. pos += sizeof(int);
  601. os_memcpy(pos, data->assoc_info.resp_ies, len);
  602. pos += len;
  603. } else {
  604. len = 0;
  605. os_memcpy(pos, &len, sizeof(int));
  606. pos += sizeof(int);
  607. }
  608. if (data && data->assoc_info.beacon_ies) {
  609. len = data->assoc_info.beacon_ies_len;
  610. os_memcpy(pos, &len, sizeof(int));
  611. pos += sizeof(int);
  612. os_memcpy(pos, data->assoc_info.beacon_ies, len);
  613. pos += len;
  614. } else {
  615. len = 0;
  616. os_memcpy(pos, &len, sizeof(int));
  617. pos += sizeof(int);
  618. }
  619. wpa_priv_send_event(iface, event, buf, buflen);
  620. os_free(buf);
  621. }
  622. static void wpa_priv_send_interface_status(struct wpa_priv_interface *iface,
  623. union wpa_event_data *data)
  624. {
  625. int ievent;
  626. size_t len, maxlen;
  627. u8 *buf;
  628. char *ifname;
  629. if (data == NULL)
  630. return;
  631. ievent = data->interface_status.ievent;
  632. maxlen = sizeof(data->interface_status.ifname);
  633. ifname = data->interface_status.ifname;
  634. for (len = 0; len < maxlen && ifname[len]; len++)
  635. ;
  636. buf = os_malloc(sizeof(int) + len);
  637. if (buf == NULL)
  638. return;
  639. os_memcpy(buf, &ievent, sizeof(int));
  640. os_memcpy(buf + sizeof(int), ifname, len);
  641. wpa_priv_send_event(iface, PRIVSEP_EVENT_INTERFACE_STATUS,
  642. buf, sizeof(int) + len);
  643. os_free(buf);
  644. }
  645. static void wpa_priv_send_ft_response(struct wpa_priv_interface *iface,
  646. union wpa_event_data *data)
  647. {
  648. size_t len;
  649. u8 *buf, *pos;
  650. if (data == NULL || data->ft_ies.ies == NULL)
  651. return;
  652. len = sizeof(int) + ETH_ALEN + data->ft_ies.ies_len;
  653. buf = os_malloc(len);
  654. if (buf == NULL)
  655. return;
  656. pos = buf;
  657. os_memcpy(pos, &data->ft_ies.ft_action, sizeof(int));
  658. pos += sizeof(int);
  659. os_memcpy(pos, data->ft_ies.target_ap, ETH_ALEN);
  660. pos += ETH_ALEN;
  661. os_memcpy(pos, data->ft_ies.ies, data->ft_ies.ies_len);
  662. wpa_priv_send_event(iface, PRIVSEP_EVENT_FT_RESPONSE, buf, len);
  663. os_free(buf);
  664. }
  665. void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
  666. union wpa_event_data *data)
  667. {
  668. struct wpa_priv_interface *iface = ctx;
  669. wpa_printf(MSG_DEBUG, "%s - event=%d", __func__, event);
  670. if (!iface->wpas_registered) {
  671. wpa_printf(MSG_DEBUG, "Driver event received, but "
  672. "wpa_supplicant not registered");
  673. return;
  674. }
  675. switch (event) {
  676. case EVENT_ASSOC:
  677. wpa_priv_send_assoc(iface, PRIVSEP_EVENT_ASSOC, data);
  678. break;
  679. case EVENT_DISASSOC:
  680. wpa_priv_send_event(iface, PRIVSEP_EVENT_DISASSOC, NULL, 0);
  681. break;
  682. case EVENT_ASSOCINFO:
  683. if (data == NULL)
  684. return;
  685. wpa_priv_send_assoc(iface, PRIVSEP_EVENT_ASSOCINFO, data);
  686. break;
  687. case EVENT_MICHAEL_MIC_FAILURE:
  688. if (data == NULL)
  689. return;
  690. wpa_priv_send_event(iface, PRIVSEP_EVENT_MICHAEL_MIC_FAILURE,
  691. &data->michael_mic_failure.unicast,
  692. sizeof(int));
  693. break;
  694. case EVENT_SCAN_RESULTS:
  695. wpa_priv_send_event(iface, PRIVSEP_EVENT_SCAN_RESULTS, NULL,
  696. 0);
  697. break;
  698. case EVENT_INTERFACE_STATUS:
  699. wpa_priv_send_interface_status(iface, data);
  700. break;
  701. case EVENT_PMKID_CANDIDATE:
  702. if (data == NULL)
  703. return;
  704. wpa_priv_send_event(iface, PRIVSEP_EVENT_PMKID_CANDIDATE,
  705. &data->pmkid_candidate,
  706. sizeof(struct pmkid_candidate));
  707. break;
  708. case EVENT_STKSTART:
  709. if (data == NULL)
  710. return;
  711. wpa_priv_send_event(iface, PRIVSEP_EVENT_STKSTART,
  712. &data->stkstart.peer, ETH_ALEN);
  713. break;
  714. case EVENT_FT_RESPONSE:
  715. wpa_priv_send_ft_response(iface, data);
  716. break;
  717. default:
  718. wpa_printf(MSG_DEBUG, "Unsupported driver event %d - TODO",
  719. event);
  720. break;
  721. }
  722. }
  723. void wpa_supplicant_rx_eapol(void *ctx, const u8 *src_addr,
  724. const u8 *buf, size_t len)
  725. {
  726. struct wpa_priv_interface *iface = ctx;
  727. struct msghdr msg;
  728. struct iovec io[3];
  729. int event = PRIVSEP_EVENT_RX_EAPOL;
  730. wpa_printf(MSG_DEBUG, "RX EAPOL from driver");
  731. io[0].iov_base = &event;
  732. io[0].iov_len = sizeof(event);
  733. io[1].iov_base = (u8 *) src_addr;
  734. io[1].iov_len = ETH_ALEN;
  735. io[2].iov_base = (u8 *) buf;
  736. io[2].iov_len = len;
  737. os_memset(&msg, 0, sizeof(msg));
  738. msg.msg_iov = io;
  739. msg.msg_iovlen = 3;
  740. msg.msg_name = &iface->drv_addr;
  741. msg.msg_namelen = sizeof(iface->drv_addr);
  742. if (sendmsg(iface->fd, &msg, 0) < 0)
  743. perror("sendmsg(wpas_socket)");
  744. }
  745. static void wpa_priv_terminate(int sig, void *signal_ctx)
  746. {
  747. wpa_printf(MSG_DEBUG, "wpa_priv termination requested");
  748. eloop_terminate();
  749. }
  750. static void wpa_priv_fd_workaround(void)
  751. {
  752. #ifdef __linux__
  753. int s, i;
  754. /* When started from pcmcia-cs scripts, wpa_supplicant might start with
  755. * fd 0, 1, and 2 closed. This will cause some issues because many
  756. * places in wpa_supplicant are still printing out to stdout. As a
  757. * workaround, make sure that fd's 0, 1, and 2 are not used for other
  758. * sockets. */
  759. for (i = 0; i < 3; i++) {
  760. s = open("/dev/null", O_RDWR);
  761. if (s > 2) {
  762. close(s);
  763. break;
  764. }
  765. }
  766. #endif /* __linux__ */
  767. }
  768. static void usage(void)
  769. {
  770. printf("wpa_priv v" VERSION_STR "\n"
  771. "Copyright (c) 2007-2009, Jouni Malinen <j@w1.fi> and "
  772. "contributors\n"
  773. "\n"
  774. "usage:\n"
  775. " wpa_priv [-Bdd] [-P<pid file>] <driver:ifname> "
  776. "[driver:ifname ...]\n");
  777. }
  778. extern int wpa_debug_level;
  779. int main(int argc, char *argv[])
  780. {
  781. int c, i;
  782. int ret = -1;
  783. char *pid_file = NULL;
  784. int daemonize = 0;
  785. char *ctrl_dir = "/var/run/wpa_priv";
  786. struct wpa_priv_interface *interfaces = NULL, *iface;
  787. if (os_program_init())
  788. return -1;
  789. wpa_priv_fd_workaround();
  790. for (;;) {
  791. c = getopt(argc, argv, "Bc:dP:");
  792. if (c < 0)
  793. break;
  794. switch (c) {
  795. case 'B':
  796. daemonize++;
  797. break;
  798. case 'c':
  799. ctrl_dir = optarg;
  800. break;
  801. case 'd':
  802. wpa_debug_level--;
  803. break;
  804. case 'P':
  805. pid_file = os_rel2abs_path(optarg);
  806. break;
  807. default:
  808. usage();
  809. goto out;
  810. }
  811. }
  812. if (optind >= argc) {
  813. usage();
  814. goto out;
  815. }
  816. wpa_printf(MSG_DEBUG, "wpa_priv control directory: '%s'", ctrl_dir);
  817. if (eloop_init()) {
  818. wpa_printf(MSG_ERROR, "Failed to initialize event loop");
  819. goto out;
  820. }
  821. for (i = optind; i < argc; i++) {
  822. wpa_printf(MSG_DEBUG, "Adding driver:interface %s", argv[i]);
  823. iface = wpa_priv_interface_init(ctrl_dir, argv[i]);
  824. if (iface == NULL)
  825. goto out;
  826. iface->next = interfaces;
  827. interfaces = iface;
  828. }
  829. if (daemonize && os_daemonize(pid_file))
  830. goto out;
  831. eloop_register_signal_terminate(wpa_priv_terminate, NULL);
  832. eloop_run();
  833. ret = 0;
  834. out:
  835. iface = interfaces;
  836. while (iface) {
  837. struct wpa_priv_interface *prev = iface;
  838. iface = iface->next;
  839. wpa_priv_interface_deinit(prev);
  840. }
  841. eloop_destroy();
  842. os_daemonize_terminate(pid_file);
  843. os_free(pid_file);
  844. os_program_deinit();
  845. return ret;
  846. }