BDatagram_unix.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835
  1. /**
  2. * @file BDatagram_unix.c
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * This file is part of BadVPN.
  8. *
  9. * BadVPN is free software: you can redistribute it and/or modify
  10. * it under the terms of the GNU General Public License version 2
  11. * as published by the Free Software Foundation.
  12. *
  13. * BadVPN is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. * GNU General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU General Public License along
  19. * with this program; if not, write to the Free Software Foundation, Inc.,
  20. * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21. */
  22. #ifndef _GNU_SOURCE
  23. #define _GNU_SOURCE
  24. #endif
  25. #include <stddef.h>
  26. #include <string.h>
  27. #include <unistd.h>
  28. #include <errno.h>
  29. #include <sys/types.h>
  30. #include <sys/socket.h>
  31. #ifdef BADVPN_LINUX
  32. # include <netpacket/packet.h>
  33. # include <net/ethernet.h>
  34. #endif
  35. #include <misc/nonblocking.h>
  36. #include <base/BLog.h>
  37. #include "BDatagram.h"
  38. #include <generated/blog_channel_BDatagram.h>
  39. struct sys_addr {
  40. socklen_t len;
  41. union {
  42. struct sockaddr generic;
  43. struct sockaddr_in ipv4;
  44. struct sockaddr_in6 ipv6;
  45. #ifdef BADVPN_LINUX
  46. struct sockaddr_ll packet;
  47. #endif
  48. } addr;
  49. };
  50. static int family_socket_to_sys (int family);
  51. static void addr_socket_to_sys (struct sys_addr *out, BAddr addr);
  52. static void addr_sys_to_socket (BAddr *out, struct sys_addr addr);
  53. static void set_pktinfo (int fd, int family);
  54. static void report_error (BDatagram *o);
  55. static void do_send (BDatagram *o);
  56. static void do_recv (BDatagram *o);
  57. static void fd_handler (BDatagram *o, int events);
  58. static void send_job_handler (BDatagram *o);
  59. static void recv_job_handler (BDatagram *o);
  60. static void send_if_handler_send (BDatagram *o, uint8_t *data, int data_len);
  61. static void recv_if_handler_recv (BDatagram *o, uint8_t *data);
  62. static int family_socket_to_sys (int family)
  63. {
  64. switch (family) {
  65. case BADDR_TYPE_IPV4:
  66. return AF_INET;
  67. case BADDR_TYPE_IPV6:
  68. return AF_INET6;
  69. #ifdef BADVPN_LINUX
  70. case BADDR_TYPE_PACKET:
  71. return AF_PACKET;
  72. #endif
  73. }
  74. ASSERT(0);
  75. return 0;
  76. }
  77. static void addr_socket_to_sys (struct sys_addr *out, BAddr addr)
  78. {
  79. switch (addr.type) {
  80. case BADDR_TYPE_IPV4: {
  81. out->len = sizeof(out->addr.ipv4);
  82. memset(&out->addr.ipv4, 0, sizeof(out->addr.ipv4));
  83. out->addr.ipv4.sin_family = AF_INET;
  84. out->addr.ipv4.sin_port = addr.ipv4.port;
  85. out->addr.ipv4.sin_addr.s_addr = addr.ipv4.ip;
  86. } break;
  87. case BADDR_TYPE_IPV6: {
  88. out->len = sizeof(out->addr.ipv6);
  89. memset(&out->addr.ipv6, 0, sizeof(out->addr.ipv6));
  90. out->addr.ipv6.sin6_family = AF_INET6;
  91. out->addr.ipv6.sin6_port = addr.ipv6.port;
  92. out->addr.ipv6.sin6_flowinfo = 0;
  93. memcpy(out->addr.ipv6.sin6_addr.s6_addr, addr.ipv6.ip, 16);
  94. out->addr.ipv6.sin6_scope_id = 0;
  95. } break;
  96. #ifdef BADVPN_LINUX
  97. case BADDR_TYPE_PACKET: {
  98. ASSERT(addr.packet.header_type == BADDR_PACKET_HEADER_TYPE_ETHERNET)
  99. memset(&out->addr.packet, 0, sizeof(out->addr.packet));
  100. out->len = sizeof(out->addr.packet);
  101. out->addr.packet.sll_family = AF_PACKET;
  102. out->addr.packet.sll_protocol = addr.packet.phys_proto;
  103. out->addr.packet.sll_ifindex = addr.packet.interface_index;
  104. out->addr.packet.sll_hatype = 1; // linux/if_arp.h: #define ARPHRD_ETHER 1
  105. switch (addr.packet.packet_type) {
  106. case BADDR_PACKET_PACKET_TYPE_HOST:
  107. out->addr.packet.sll_pkttype = PACKET_HOST;
  108. break;
  109. case BADDR_PACKET_PACKET_TYPE_BROADCAST:
  110. out->addr.packet.sll_pkttype = PACKET_BROADCAST;
  111. break;
  112. case BADDR_PACKET_PACKET_TYPE_MULTICAST:
  113. out->addr.packet.sll_pkttype = PACKET_MULTICAST;
  114. break;
  115. case BADDR_PACKET_PACKET_TYPE_OTHERHOST:
  116. out->addr.packet.sll_pkttype = PACKET_OTHERHOST;
  117. break;
  118. case BADDR_PACKET_PACKET_TYPE_OUTGOING:
  119. out->addr.packet.sll_pkttype = PACKET_OUTGOING;
  120. break;
  121. default:
  122. ASSERT(0);
  123. }
  124. out->addr.packet.sll_halen = 6;
  125. memcpy(out->addr.packet.sll_addr, addr.packet.phys_addr, 6);
  126. } break;
  127. #endif
  128. default: ASSERT(0);
  129. }
  130. }
  131. static void addr_sys_to_socket (BAddr *out, struct sys_addr addr)
  132. {
  133. switch (addr.addr.generic.sa_family) {
  134. case AF_INET: {
  135. ASSERT(addr.len == sizeof(struct sockaddr_in))
  136. BAddr_InitIPv4(out, addr.addr.ipv4.sin_addr.s_addr, addr.addr.ipv4.sin_port);
  137. } break;
  138. case AF_INET6: {
  139. ASSERT(addr.len == sizeof(struct sockaddr_in6))
  140. BAddr_InitIPv6(out, addr.addr.ipv6.sin6_addr.s6_addr, addr.addr.ipv6.sin6_port);
  141. } break;
  142. case AF_PACKET: {
  143. if (addr.len < offsetof(struct sockaddr_ll, sll_addr) + 6) {
  144. goto fail;
  145. }
  146. if (addr.addr.packet.sll_hatype != 1) { // linux/if_arp.h: #define ARPHRD_ETHER 1
  147. goto fail;
  148. }
  149. int packet_type;
  150. switch (addr.addr.packet.sll_pkttype) {
  151. case PACKET_HOST:
  152. packet_type = BADDR_PACKET_PACKET_TYPE_HOST;
  153. break;
  154. case PACKET_BROADCAST:
  155. packet_type = BADDR_PACKET_PACKET_TYPE_BROADCAST;
  156. break;
  157. case PACKET_MULTICAST:
  158. packet_type = BADDR_PACKET_PACKET_TYPE_MULTICAST;
  159. break;
  160. case PACKET_OTHERHOST:
  161. packet_type = BADDR_PACKET_PACKET_TYPE_OTHERHOST;
  162. break;
  163. case PACKET_OUTGOING:
  164. packet_type = BADDR_PACKET_PACKET_TYPE_OUTGOING;
  165. break;
  166. default:
  167. goto fail;
  168. }
  169. if (addr.addr.packet.sll_halen != 6) {
  170. goto fail;
  171. }
  172. uint8_t *mac = addr.addr.packet.sll_addr;
  173. BAddr_InitPacket(out, addr.addr.packet.sll_protocol, addr.addr.packet.sll_ifindex, BADDR_PACKET_HEADER_TYPE_ETHERNET, packet_type, addr.addr.packet.sll_addr);
  174. } break;
  175. fail:
  176. default: {
  177. BAddr_InitNone(out);
  178. } break;
  179. }
  180. }
  181. static void set_pktinfo (int fd, int family)
  182. {
  183. int opt = 1;
  184. switch (family) {
  185. case BADDR_TYPE_IPV4: {
  186. #ifdef BADVPN_FREEBSD
  187. if (setsockopt(fd, IPPROTO_IP, IP_RECVDSTADDR, &opt, sizeof(opt)) < 0) {
  188. BLog(BLOG_ERROR, "setsockopt(IP_RECVDSTADDR) failed");
  189. }
  190. #else
  191. if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &opt, sizeof(opt)) < 0) {
  192. BLog(BLOG_ERROR, "setsockopt(IP_PKTINFO) failed");
  193. }
  194. #endif
  195. } break;
  196. #ifdef IPV6_RECVPKTINFO
  197. case BADDR_TYPE_IPV6: {
  198. if (setsockopt(fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &opt, sizeof(opt)) < 0) {
  199. BLog(BLOG_ERROR, "setsockopt(IPV6_RECVPKTINFO) failed");
  200. }
  201. } break;
  202. #endif
  203. }
  204. }
  205. static void report_error (BDatagram *o)
  206. {
  207. DebugError_AssertNoError(&o->d_err);
  208. // report error
  209. DEBUGERROR(&o->d_err, o->handler(o->user, BDATAGRAM_EVENT_ERROR));
  210. return;
  211. }
  212. static void do_send (BDatagram *o)
  213. {
  214. DebugError_AssertNoError(&o->d_err);
  215. ASSERT(o->send.inited)
  216. ASSERT(o->send.busy)
  217. ASSERT(o->send.have_addrs)
  218. // limit
  219. if (!BReactorLimit_Increment(&o->send.limit)) {
  220. // wait for fd
  221. o->wait_events |= BREACTOR_WRITE;
  222. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  223. return;
  224. }
  225. // convert destination address
  226. struct sys_addr sysaddr;
  227. addr_socket_to_sys(&sysaddr, o->send.remote_addr);
  228. struct iovec iov;
  229. iov.iov_base = (uint8_t *)o->send.busy_data;
  230. iov.iov_len = o->send.busy_data_len;
  231. union {
  232. #ifdef BADVPN_FREEBSD
  233. char in[CMSG_SPACE(sizeof(struct in_addr))];
  234. #else
  235. char in[CMSG_SPACE(sizeof(struct in_pktinfo))];
  236. #endif
  237. char in6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
  238. } cdata;
  239. struct msghdr msg;
  240. memset(&msg, 0, sizeof(msg));
  241. msg.msg_name = &sysaddr.addr.generic;
  242. msg.msg_namelen = sysaddr.len;
  243. msg.msg_iov = &iov;
  244. msg.msg_iovlen = 1;
  245. msg.msg_control = &cdata;
  246. msg.msg_controllen = sizeof(cdata);
  247. struct cmsghdr *cmsg = CMSG_FIRSTHDR(&msg);
  248. size_t controllen = 0;
  249. switch (o->send.local_addr.type) {
  250. case BADDR_TYPE_IPV4: {
  251. #ifdef BADVPN_FREEBSD
  252. memset(cmsg, 0, CMSG_SPACE(sizeof(struct in_addr)));
  253. cmsg->cmsg_level = IPPROTO_IP;
  254. cmsg->cmsg_type = IP_SENDSRCADDR;
  255. cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
  256. struct in_addr *addrinfo = (struct in_addr *)CMSG_DATA(cmsg);
  257. addrinfo->s_addr = o->send.local_addr.ipv4;
  258. controllen += CMSG_SPACE(sizeof(struct in_addr));
  259. #else
  260. memset(cmsg, 0, CMSG_SPACE(sizeof(struct in_pktinfo)));
  261. cmsg->cmsg_level = IPPROTO_IP;
  262. cmsg->cmsg_type = IP_PKTINFO;
  263. cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
  264. struct in_pktinfo *pktinfo = (struct in_pktinfo *)CMSG_DATA(cmsg);
  265. pktinfo->ipi_spec_dst.s_addr = o->send.local_addr.ipv4;
  266. controllen += CMSG_SPACE(sizeof(struct in_pktinfo));
  267. #endif
  268. } break;
  269. case BADDR_TYPE_IPV6: {
  270. memset(cmsg, 0, CMSG_SPACE(sizeof(struct in6_pktinfo)));
  271. cmsg->cmsg_level = IPPROTO_IPV6;
  272. cmsg->cmsg_type = IPV6_PKTINFO;
  273. cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
  274. struct in6_pktinfo *pktinfo = (struct in6_pktinfo *)CMSG_DATA(cmsg);
  275. memcpy(pktinfo->ipi6_addr.s6_addr, o->send.local_addr.ipv6, 16);
  276. controllen += CMSG_SPACE(sizeof(struct in6_pktinfo));
  277. } break;
  278. }
  279. msg.msg_controllen = controllen;
  280. if (msg.msg_controllen == 0) {
  281. msg.msg_control = NULL;
  282. }
  283. // send
  284. int bytes = sendmsg(o->fd, &msg, 0);
  285. if (bytes < 0) {
  286. if (errno == EAGAIN || errno == EWOULDBLOCK) {
  287. // wait for fd
  288. o->wait_events |= BREACTOR_WRITE;
  289. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  290. return;
  291. }
  292. BLog(BLOG_ERROR, "send failed");
  293. report_error(o);
  294. return;
  295. }
  296. ASSERT(bytes >= 0)
  297. ASSERT(bytes <= o->send.busy_data_len)
  298. if (bytes < o->send.busy_data_len) {
  299. BLog(BLOG_ERROR, "send sent too little");
  300. }
  301. // if recv wasn't started yet, start it
  302. if (!o->recv.started) {
  303. // set recv started
  304. o->recv.started = 1;
  305. // continue receiving
  306. if (o->recv.inited && o->recv.busy) {
  307. BPending_Set(&o->recv.job);
  308. }
  309. }
  310. // set not busy
  311. o->send.busy = 0;
  312. // done
  313. PacketPassInterface_Done(&o->send.iface);
  314. }
  315. static void do_recv (BDatagram *o)
  316. {
  317. DebugError_AssertNoError(&o->d_err);
  318. ASSERT(o->recv.inited)
  319. ASSERT(o->recv.busy)
  320. ASSERT(o->recv.started)
  321. // limit
  322. if (!BReactorLimit_Increment(&o->recv.limit)) {
  323. // wait for fd
  324. o->wait_events |= BREACTOR_READ;
  325. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  326. return;
  327. }
  328. struct sys_addr sysaddr;
  329. struct iovec iov;
  330. iov.iov_base = o->recv.busy_data;
  331. iov.iov_len = o->recv.mtu;
  332. union {
  333. #ifdef BADVPN_FREEBSD
  334. char in[CMSG_SPACE(sizeof(struct in_addr))];
  335. #else
  336. char in[CMSG_SPACE(sizeof(struct in_pktinfo))];
  337. #endif
  338. char in6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
  339. } cdata;
  340. struct msghdr msg;
  341. memset(&msg, 0, sizeof(msg));
  342. msg.msg_name = &sysaddr.addr.generic;
  343. msg.msg_namelen = sizeof(sysaddr.addr);
  344. msg.msg_iov = &iov;
  345. msg.msg_iovlen = 1;
  346. msg.msg_control = &cdata;
  347. msg.msg_controllen = sizeof(cdata);
  348. // recv
  349. int bytes = recvmsg(o->fd, &msg, 0);
  350. if (bytes < 0) {
  351. if (errno == EAGAIN || errno == EWOULDBLOCK) {
  352. // wait for fd
  353. o->wait_events |= BREACTOR_READ;
  354. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  355. return;
  356. }
  357. BLog(BLOG_ERROR, "recv failed");
  358. report_error(o);
  359. return;
  360. }
  361. ASSERT(bytes >= 0)
  362. ASSERT(bytes <= o->recv.mtu)
  363. // read returned address
  364. sysaddr.len = msg.msg_namelen;
  365. addr_sys_to_socket(&o->recv.remote_addr, sysaddr);
  366. // read returned local address
  367. BIPAddr_InitInvalid(&o->recv.local_addr);
  368. for (struct cmsghdr *cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
  369. #ifdef BADVPN_FREEBSD
  370. if (cmsg->cmsg_level == IPPROTO_IP && cmsg->cmsg_type == IP_RECVDSTADDR) {
  371. struct in_addr *addrinfo = (struct in_addr *)CMSG_DATA(cmsg);
  372. BIPAddr_InitIPv4(&o->recv.local_addr, addrinfo->s_addr);
  373. }
  374. #else
  375. if (cmsg->cmsg_level == IPPROTO_IP && cmsg->cmsg_type == IP_PKTINFO) {
  376. struct in_pktinfo *pktinfo = (struct in_pktinfo *)CMSG_DATA(cmsg);
  377. BIPAddr_InitIPv4(&o->recv.local_addr, pktinfo->ipi_addr.s_addr);
  378. }
  379. #endif
  380. else if (cmsg->cmsg_level == IPPROTO_IPV6 && cmsg->cmsg_type == IPV6_PKTINFO) {
  381. struct in6_pktinfo *pktinfo = (struct in6_pktinfo *)CMSG_DATA(cmsg);
  382. BIPAddr_InitIPv6(&o->recv.local_addr, pktinfo->ipi6_addr.s6_addr);
  383. }
  384. }
  385. // set have addresses
  386. o->recv.have_addrs = 1;
  387. // set not busy
  388. o->recv.busy = 0;
  389. // done
  390. PacketRecvInterface_Done(&o->recv.iface, bytes);
  391. }
  392. static void fd_handler (BDatagram *o, int events)
  393. {
  394. DebugObject_Access(&o->d_obj);
  395. DebugError_AssertNoError(&o->d_err);
  396. // clear handled events
  397. o->wait_events &= ~events;
  398. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  399. int have_send = 0;
  400. int have_recv = 0;
  401. if ((events & BREACTOR_WRITE) || ((events & BREACTOR_ERROR) && o->send.inited && o->send.busy && o->send.have_addrs)) {
  402. ASSERT(o->send.inited)
  403. ASSERT(o->send.busy)
  404. ASSERT(o->send.have_addrs)
  405. have_send = 1;
  406. }
  407. if ((events & BREACTOR_READ) || ((events & BREACTOR_ERROR) && o->recv.inited && o->recv.busy && o->recv.started)) {
  408. ASSERT(o->recv.inited)
  409. ASSERT(o->recv.busy)
  410. ASSERT(o->recv.started)
  411. have_recv = 1;
  412. }
  413. if (have_send) {
  414. if (have_recv) {
  415. BPending_Set(&o->recv.job);
  416. }
  417. do_send(o);
  418. return;
  419. }
  420. if (have_recv) {
  421. do_recv(o);
  422. return;
  423. }
  424. BLog(BLOG_ERROR, "fd error event");
  425. report_error(o);
  426. return;
  427. }
  428. static void send_job_handler (BDatagram *o)
  429. {
  430. DebugObject_Access(&o->d_obj);
  431. DebugError_AssertNoError(&o->d_err);
  432. ASSERT(o->send.inited)
  433. ASSERT(o->send.busy)
  434. ASSERT(o->send.have_addrs)
  435. do_send(o);
  436. return;
  437. }
  438. static void recv_job_handler (BDatagram *o)
  439. {
  440. DebugObject_Access(&o->d_obj);
  441. DebugError_AssertNoError(&o->d_err);
  442. ASSERT(o->recv.inited)
  443. ASSERT(o->recv.busy)
  444. ASSERT(o->recv.started)
  445. do_recv(o);
  446. return;
  447. }
  448. static void send_if_handler_send (BDatagram *o, uint8_t *data, int data_len)
  449. {
  450. DebugObject_Access(&o->d_obj);
  451. DebugError_AssertNoError(&o->d_err);
  452. ASSERT(o->send.inited)
  453. ASSERT(!o->send.busy)
  454. ASSERT(data_len >= 0)
  455. ASSERT(data_len <= o->send.mtu)
  456. // remember data
  457. o->send.busy_data = data;
  458. o->send.busy_data_len = data_len;
  459. // set busy
  460. o->send.busy = 1;
  461. // if have no addresses, wait
  462. if (!o->send.have_addrs) {
  463. return;
  464. }
  465. // set job
  466. BPending_Set(&o->send.job);
  467. }
  468. static void recv_if_handler_recv (BDatagram *o, uint8_t *data)
  469. {
  470. DebugObject_Access(&o->d_obj);
  471. DebugError_AssertNoError(&o->d_err);
  472. ASSERT(o->recv.inited)
  473. ASSERT(!o->recv.busy)
  474. // remember data
  475. o->recv.busy_data = data;
  476. // set busy
  477. o->recv.busy = 1;
  478. // if recv not started yet, wait
  479. if (!o->recv.started) {
  480. return;
  481. }
  482. // set job
  483. BPending_Set(&o->recv.job);
  484. }
  485. int BDatagram_AddressFamilySupported (int family)
  486. {
  487. switch (family) {
  488. case BADDR_TYPE_IPV4:
  489. case BADDR_TYPE_IPV6:
  490. #ifdef BADVPN_LINUX
  491. case BADDR_TYPE_PACKET:
  492. #endif
  493. return 1;
  494. }
  495. return 0;
  496. }
  497. int BDatagram_Init (BDatagram *o, int family, BReactor *reactor, void *user,
  498. BDatagram_handler handler)
  499. {
  500. ASSERT(BDatagram_AddressFamilySupported(family))
  501. ASSERT(handler)
  502. BNetwork_Assert();
  503. // init arguments
  504. o->reactor = reactor;
  505. o->user = user;
  506. o->handler = handler;
  507. // init fd
  508. if ((o->fd = socket(family_socket_to_sys(family), SOCK_DGRAM, 0)) < 0) {
  509. BLog(BLOG_ERROR, "socket failed");
  510. goto fail0;
  511. }
  512. // set fd non-blocking
  513. if (!badvpn_set_nonblocking(o->fd)) {
  514. BLog(BLOG_ERROR, "badvpn_set_nonblocking failed");
  515. goto fail1;
  516. }
  517. // enable receiving pktinfo
  518. set_pktinfo(o->fd, family);
  519. // init BFileDescriptor
  520. BFileDescriptor_Init(&o->bfd, o->fd, (BFileDescriptor_handler)fd_handler, o);
  521. if (!BReactor_AddFileDescriptor(o->reactor, &o->bfd)) {
  522. BLog(BLOG_ERROR, "BReactor_AddFileDescriptor failed");
  523. goto fail1;
  524. }
  525. // set no wait events
  526. o->wait_events = 0;
  527. // init limits
  528. BReactorLimit_Init(&o->send.limit, o->reactor, BDATAGRAM_SEND_LIMIT);
  529. BReactorLimit_Init(&o->recv.limit, o->reactor, BDATAGRAM_RECV_LIMIT);
  530. // set have no send and recv addresses
  531. o->send.have_addrs = 0;
  532. o->recv.have_addrs = 0;
  533. // set recv not started
  534. o->recv.started = 0;
  535. // set send and recv not inited
  536. o->send.inited = 0;
  537. o->recv.inited = 0;
  538. DebugError_Init(&o->d_err, BReactor_PendingGroup(o->reactor));
  539. DebugObject_Init(&o->d_obj);
  540. return 1;
  541. fail1:
  542. if (close(o->fd) < 0) {
  543. BLog(BLOG_ERROR, "close failed");
  544. }
  545. fail0:
  546. return 0;
  547. }
  548. void BDatagram_Free (BDatagram *o)
  549. {
  550. DebugObject_Free(&o->d_obj);
  551. DebugError_Free(&o->d_err);
  552. ASSERT(!o->recv.inited)
  553. ASSERT(!o->send.inited)
  554. // free limits
  555. BReactorLimit_Free(&o->recv.limit);
  556. BReactorLimit_Free(&o->send.limit);
  557. // free BFileDescriptor
  558. BReactor_RemoveFileDescriptor(o->reactor, &o->bfd);
  559. // free fd
  560. if (close(o->fd) < 0) {
  561. BLog(BLOG_ERROR, "close failed");
  562. }
  563. }
  564. int BDatagram_Bind (BDatagram *o, BAddr addr)
  565. {
  566. DebugObject_Access(&o->d_obj);
  567. DebugError_AssertNoError(&o->d_err);
  568. ASSERT(BDatagram_AddressFamilySupported(addr.type))
  569. // translate address
  570. struct sys_addr sysaddr;
  571. addr_socket_to_sys(&sysaddr, addr);
  572. // bind
  573. if (bind(o->fd, &sysaddr.addr.generic, sysaddr.len) < 0) {
  574. BLog(BLOG_ERROR, "bind failed");
  575. return 0;
  576. }
  577. // if recv wasn't started yet, start it
  578. if (!o->recv.started) {
  579. // set recv started
  580. o->recv.started = 1;
  581. // continue receiving
  582. if (o->recv.inited && o->recv.busy) {
  583. BPending_Set(&o->recv.job);
  584. }
  585. }
  586. return 1;
  587. }
  588. void BDatagram_SetSendAddrs (BDatagram *o, BAddr remote_addr, BIPAddr local_addr)
  589. {
  590. DebugObject_Access(&o->d_obj);
  591. DebugError_AssertNoError(&o->d_err);
  592. ASSERT(BDatagram_AddressFamilySupported(remote_addr.type))
  593. ASSERT(local_addr.type == BADDR_TYPE_NONE || BDatagram_AddressFamilySupported(local_addr.type))
  594. // set addresses
  595. o->send.remote_addr = remote_addr;
  596. o->send.local_addr = local_addr;
  597. if (!o->send.have_addrs) {
  598. // set have addresses
  599. o->send.have_addrs = 1;
  600. // start sending
  601. if (o->send.inited && o->send.busy) {
  602. BPending_Set(&o->send.job);
  603. }
  604. }
  605. }
  606. int BDatagram_GetLastReceiveAddrs (BDatagram *o, BAddr *remote_addr, BIPAddr *local_addr)
  607. {
  608. DebugObject_Access(&o->d_obj);
  609. if (!o->recv.have_addrs) {
  610. return 0;
  611. }
  612. *remote_addr = o->recv.remote_addr;
  613. *local_addr = o->recv.local_addr;
  614. return 1;
  615. }
  616. int BDatagram_GetFd (BDatagram *o)
  617. {
  618. DebugObject_Access(&o->d_obj);
  619. return o->fd;
  620. }
  621. void BDatagram_SendAsync_Init (BDatagram *o, int mtu)
  622. {
  623. DebugObject_Access(&o->d_obj);
  624. DebugError_AssertNoError(&o->d_err);
  625. ASSERT(!o->send.inited)
  626. ASSERT(mtu >= 0)
  627. // init arguments
  628. o->send.mtu = mtu;
  629. // init interface
  630. PacketPassInterface_Init(&o->send.iface, o->send.mtu, (PacketPassInterface_handler_send)send_if_handler_send, o, BReactor_PendingGroup(o->reactor));
  631. // init job
  632. BPending_Init(&o->send.job, BReactor_PendingGroup(o->reactor), (BPending_handler)send_job_handler, o);
  633. // set not busy
  634. o->send.busy = 0;
  635. // set inited
  636. o->send.inited = 1;
  637. }
  638. void BDatagram_SendAsync_Free (BDatagram *o)
  639. {
  640. DebugObject_Access(&o->d_obj);
  641. ASSERT(o->send.inited)
  642. // update events
  643. o->wait_events &= ~BREACTOR_WRITE;
  644. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  645. // free job
  646. BPending_Free(&o->send.job);
  647. // free interface
  648. PacketPassInterface_Free(&o->send.iface);
  649. // set not inited
  650. o->send.inited = 0;
  651. }
  652. PacketPassInterface * BDatagram_SendAsync_GetIf (BDatagram *o)
  653. {
  654. DebugObject_Access(&o->d_obj);
  655. ASSERT(o->send.inited)
  656. return &o->send.iface;
  657. }
  658. void BDatagram_RecvAsync_Init (BDatagram *o, int mtu)
  659. {
  660. DebugObject_Access(&o->d_obj);
  661. DebugError_AssertNoError(&o->d_err);
  662. ASSERT(!o->recv.inited)
  663. ASSERT(mtu >= 0)
  664. // init arguments
  665. o->recv.mtu = mtu;
  666. // init interface
  667. PacketRecvInterface_Init(&o->recv.iface, o->recv.mtu, (PacketRecvInterface_handler_recv)recv_if_handler_recv, o, BReactor_PendingGroup(o->reactor));
  668. // init job
  669. BPending_Init(&o->recv.job, BReactor_PendingGroup(o->reactor), (BPending_handler)recv_job_handler, o);
  670. // set not busy
  671. o->recv.busy = 0;
  672. // set inited
  673. o->recv.inited = 1;
  674. }
  675. void BDatagram_RecvAsync_Free (BDatagram *o)
  676. {
  677. DebugObject_Access(&o->d_obj);
  678. ASSERT(o->recv.inited)
  679. // update events
  680. o->wait_events &= ~BREACTOR_READ;
  681. BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
  682. // free job
  683. BPending_Free(&o->recv.job);
  684. // free interface
  685. PacketRecvInterface_Free(&o->recv.iface);
  686. // set not inited
  687. o->recv.inited = 0;
  688. }
  689. PacketRecvInterface * BDatagram_RecvAsync_GetIf (BDatagram *o)
  690. {
  691. DebugObject_Access(&o->d_obj);
  692. ASSERT(o->recv.inited)
  693. return &o->recv.iface;
  694. }