wpa_priv.c 24 KB

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