| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878 |
- /**
- * @file BDatagram_unix.c
- * @author Ambroz Bizjak <ambrop7@gmail.com>
- *
- * @section LICENSE
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- * 3. Neither the name of the author nor the
- * names of its contributors may be used to endorse or promote products
- * derived from this software without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
- * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- #ifndef _GNU_SOURCE
- #define _GNU_SOURCE
- #endif
- #include <stddef.h>
- #include <string.h>
- #include <unistd.h>
- #include <errno.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #ifdef BADVPN_LINUX
- # include <netpacket/packet.h>
- # include <net/ethernet.h>
- #endif
- #include <misc/nonblocking.h>
- #include <base/BLog.h>
- #include "BDatagram.h"
- #include <generated/blog_channel_BDatagram.h>
- struct sys_addr {
- socklen_t len;
- union {
- struct sockaddr generic;
- struct sockaddr_in ipv4;
- struct sockaddr_in6 ipv6;
- #ifdef BADVPN_LINUX
- struct sockaddr_ll packet;
- #endif
- } addr;
- };
- static int family_socket_to_sys (int family);
- static void addr_socket_to_sys (struct sys_addr *out, BAddr addr);
- static void addr_sys_to_socket (BAddr *out, struct sys_addr addr);
- static void set_pktinfo (int fd, int family);
- static void report_error (BDatagram *o);
- static void do_send (BDatagram *o);
- static void do_recv (BDatagram *o);
- static void fd_handler (BDatagram *o, int events);
- static void send_job_handler (BDatagram *o);
- static void recv_job_handler (BDatagram *o);
- static void send_if_handler_send (BDatagram *o, uint8_t *data, int data_len);
- static void recv_if_handler_recv (BDatagram *o, uint8_t *data);
- static int family_socket_to_sys (int family)
- {
- switch (family) {
- case BADDR_TYPE_IPV4:
- return AF_INET;
- case BADDR_TYPE_IPV6:
- return AF_INET6;
- #ifdef BADVPN_LINUX
- case BADDR_TYPE_PACKET:
- return AF_PACKET;
- #endif
- }
-
- ASSERT(0);
- return 0;
- }
- static void addr_socket_to_sys (struct sys_addr *out, BAddr addr)
- {
- switch (addr.type) {
- case BADDR_TYPE_IPV4: {
- out->len = sizeof(out->addr.ipv4);
- memset(&out->addr.ipv4, 0, sizeof(out->addr.ipv4));
- out->addr.ipv4.sin_family = AF_INET;
- out->addr.ipv4.sin_port = addr.ipv4.port;
- out->addr.ipv4.sin_addr.s_addr = addr.ipv4.ip;
- } break;
-
- case BADDR_TYPE_IPV6: {
- out->len = sizeof(out->addr.ipv6);
- memset(&out->addr.ipv6, 0, sizeof(out->addr.ipv6));
- out->addr.ipv6.sin6_family = AF_INET6;
- out->addr.ipv6.sin6_port = addr.ipv6.port;
- out->addr.ipv6.sin6_flowinfo = 0;
- memcpy(out->addr.ipv6.sin6_addr.s6_addr, addr.ipv6.ip, 16);
- out->addr.ipv6.sin6_scope_id = 0;
- } break;
-
- #ifdef BADVPN_LINUX
- case BADDR_TYPE_PACKET: {
- ASSERT(addr.packet.header_type == BADDR_PACKET_HEADER_TYPE_ETHERNET)
- memset(&out->addr.packet, 0, sizeof(out->addr.packet));
- out->len = sizeof(out->addr.packet);
- out->addr.packet.sll_family = AF_PACKET;
- out->addr.packet.sll_protocol = addr.packet.phys_proto;
- out->addr.packet.sll_ifindex = addr.packet.interface_index;
- out->addr.packet.sll_hatype = 1; // linux/if_arp.h: #define ARPHRD_ETHER 1
- switch (addr.packet.packet_type) {
- case BADDR_PACKET_PACKET_TYPE_HOST:
- out->addr.packet.sll_pkttype = PACKET_HOST;
- break;
- case BADDR_PACKET_PACKET_TYPE_BROADCAST:
- out->addr.packet.sll_pkttype = PACKET_BROADCAST;
- break;
- case BADDR_PACKET_PACKET_TYPE_MULTICAST:
- out->addr.packet.sll_pkttype = PACKET_MULTICAST;
- break;
- case BADDR_PACKET_PACKET_TYPE_OTHERHOST:
- out->addr.packet.sll_pkttype = PACKET_OTHERHOST;
- break;
- case BADDR_PACKET_PACKET_TYPE_OUTGOING:
- out->addr.packet.sll_pkttype = PACKET_OUTGOING;
- break;
- default:
- ASSERT(0);
- }
- out->addr.packet.sll_halen = 6;
- memcpy(out->addr.packet.sll_addr, addr.packet.phys_addr, 6);
- } break;
- #endif
-
- default: ASSERT(0);
- }
- }
- static void addr_sys_to_socket (BAddr *out, struct sys_addr addr)
- {
- switch (addr.addr.generic.sa_family) {
- case AF_INET: {
- ASSERT(addr.len == sizeof(struct sockaddr_in))
- BAddr_InitIPv4(out, addr.addr.ipv4.sin_addr.s_addr, addr.addr.ipv4.sin_port);
- } break;
-
- case AF_INET6: {
- ASSERT(addr.len == sizeof(struct sockaddr_in6))
- BAddr_InitIPv6(out, addr.addr.ipv6.sin6_addr.s6_addr, addr.addr.ipv6.sin6_port);
- } break;
-
- #ifdef BADVPN_LINUX
- case AF_PACKET: {
- if (addr.len < offsetof(struct sockaddr_ll, sll_addr) + 6) {
- goto fail;
- }
- if (addr.addr.packet.sll_hatype != 1) { // linux/if_arp.h: #define ARPHRD_ETHER 1
- goto fail;
- }
- int packet_type;
- switch (addr.addr.packet.sll_pkttype) {
- case PACKET_HOST:
- packet_type = BADDR_PACKET_PACKET_TYPE_HOST;
- break;
- case PACKET_BROADCAST:
- packet_type = BADDR_PACKET_PACKET_TYPE_BROADCAST;
- break;
- case PACKET_MULTICAST:
- packet_type = BADDR_PACKET_PACKET_TYPE_MULTICAST;
- break;
- case PACKET_OTHERHOST:
- packet_type = BADDR_PACKET_PACKET_TYPE_OTHERHOST;
- break;
- case PACKET_OUTGOING:
- packet_type = BADDR_PACKET_PACKET_TYPE_OUTGOING;
- break;
- default:
- goto fail;
- }
- if (addr.addr.packet.sll_halen != 6) {
- goto fail;
- }
- 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);
- } break;
- #endif
-
- fail:
- default: {
- BAddr_InitNone(out);
- } break;
- }
- }
- static void set_pktinfo (int fd, int family)
- {
- int opt = 1;
-
- switch (family) {
- case BADDR_TYPE_IPV4: {
- #ifdef BADVPN_FREEBSD
- if (setsockopt(fd, IPPROTO_IP, IP_RECVDSTADDR, &opt, sizeof(opt)) < 0) {
- BLog(BLOG_ERROR, "setsockopt(IP_RECVDSTADDR) failed");
- }
- #else
- if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &opt, sizeof(opt)) < 0) {
- BLog(BLOG_ERROR, "setsockopt(IP_PKTINFO) failed");
- }
- #endif
- } break;
-
- #ifdef IPV6_RECVPKTINFO
- case BADDR_TYPE_IPV6: {
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &opt, sizeof(opt)) < 0) {
- BLog(BLOG_ERROR, "setsockopt(IPV6_RECVPKTINFO) failed");
- }
- } break;
- #endif
- }
- }
- static void report_error (BDatagram *o)
- {
- DebugError_AssertNoError(&o->d_err);
-
- // report error
- DEBUGERROR(&o->d_err, o->handler(o->user, BDATAGRAM_EVENT_ERROR));
- return;
- }
- static void do_send (BDatagram *o)
- {
- DebugError_AssertNoError(&o->d_err);
- ASSERT(o->send.inited)
- ASSERT(o->send.busy)
- ASSERT(o->send.have_addrs)
-
- // limit
- if (!BReactorLimit_Increment(&o->send.limit)) {
- // wait for fd
- o->wait_events |= BREACTOR_WRITE;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
- return;
- }
-
- // convert destination address
- struct sys_addr sysaddr;
- addr_socket_to_sys(&sysaddr, o->send.remote_addr);
-
- struct iovec iov;
- iov.iov_base = (uint8_t *)o->send.busy_data;
- iov.iov_len = o->send.busy_data_len;
-
- union {
- #ifdef BADVPN_FREEBSD
- char in[CMSG_SPACE(sizeof(struct in_addr))];
- #else
- char in[CMSG_SPACE(sizeof(struct in_pktinfo))];
- #endif
- char in6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
- } cdata;
-
- struct msghdr msg;
- memset(&msg, 0, sizeof(msg));
- msg.msg_name = &sysaddr.addr.generic;
- msg.msg_namelen = sysaddr.len;
- msg.msg_iov = &iov;
- msg.msg_iovlen = 1;
- msg.msg_control = &cdata;
- msg.msg_controllen = sizeof(cdata);
-
- struct cmsghdr *cmsg = CMSG_FIRSTHDR(&msg);
-
- size_t controllen = 0;
-
- switch (o->send.local_addr.type) {
- case BADDR_TYPE_IPV4: {
- #ifdef BADVPN_FREEBSD
- memset(cmsg, 0, CMSG_SPACE(sizeof(struct in_addr)));
- cmsg->cmsg_level = IPPROTO_IP;
- cmsg->cmsg_type = IP_SENDSRCADDR;
- cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
- struct in_addr *addrinfo = (struct in_addr *)CMSG_DATA(cmsg);
- addrinfo->s_addr = o->send.local_addr.ipv4;
- controllen += CMSG_SPACE(sizeof(struct in_addr));
- #else
- memset(cmsg, 0, CMSG_SPACE(sizeof(struct in_pktinfo)));
- cmsg->cmsg_level = IPPROTO_IP;
- cmsg->cmsg_type = IP_PKTINFO;
- cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
- struct in_pktinfo *pktinfo = (struct in_pktinfo *)CMSG_DATA(cmsg);
- pktinfo->ipi_spec_dst.s_addr = o->send.local_addr.ipv4;
- controllen += CMSG_SPACE(sizeof(struct in_pktinfo));
- #endif
- } break;
-
- case BADDR_TYPE_IPV6: {
- memset(cmsg, 0, CMSG_SPACE(sizeof(struct in6_pktinfo)));
- cmsg->cmsg_level = IPPROTO_IPV6;
- cmsg->cmsg_type = IPV6_PKTINFO;
- cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
- struct in6_pktinfo *pktinfo = (struct in6_pktinfo *)CMSG_DATA(cmsg);
- memcpy(pktinfo->ipi6_addr.s6_addr, o->send.local_addr.ipv6, 16);
- controllen += CMSG_SPACE(sizeof(struct in6_pktinfo));
- } break;
- }
-
- msg.msg_controllen = controllen;
-
- if (msg.msg_controllen == 0) {
- msg.msg_control = NULL;
- }
-
- // send
- int bytes = sendmsg(o->fd, &msg, 0);
- if (bytes < 0) {
- if (errno == EAGAIN || errno == EWOULDBLOCK) {
- // wait for fd
- o->wait_events |= BREACTOR_WRITE;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
- return;
- }
-
- report_error(o);
- return;
- }
-
- ASSERT(bytes >= 0)
- ASSERT(bytes <= o->send.busy_data_len)
-
- if (bytes < o->send.busy_data_len) {
- BLog(BLOG_ERROR, "send sent too little");
- }
-
- // if recv wasn't started yet, start it
- if (!o->recv.started) {
- // set recv started
- o->recv.started = 1;
-
- // continue receiving
- if (o->recv.inited && o->recv.busy) {
- BPending_Set(&o->recv.job);
- }
- }
-
- // set not busy
- o->send.busy = 0;
-
- // done
- PacketPassInterface_Done(&o->send.iface);
- }
- static void do_recv (BDatagram *o)
- {
- DebugError_AssertNoError(&o->d_err);
- ASSERT(o->recv.inited)
- ASSERT(o->recv.busy)
- ASSERT(o->recv.started)
-
- // limit
- if (!BReactorLimit_Increment(&o->recv.limit)) {
- // wait for fd
- o->wait_events |= BREACTOR_READ;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
- return;
- }
-
- struct sys_addr sysaddr;
-
- struct iovec iov;
- iov.iov_base = o->recv.busy_data;
- iov.iov_len = o->recv.mtu;
-
- union {
- #ifdef BADVPN_FREEBSD
- char in[CMSG_SPACE(sizeof(struct in_addr))];
- #else
- char in[CMSG_SPACE(sizeof(struct in_pktinfo))];
- #endif
- char in6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
- } cdata;
-
- struct msghdr msg;
- memset(&msg, 0, sizeof(msg));
- msg.msg_name = &sysaddr.addr.generic;
- msg.msg_namelen = sizeof(sysaddr.addr);
- msg.msg_iov = &iov;
- msg.msg_iovlen = 1;
- msg.msg_control = &cdata;
- msg.msg_controllen = sizeof(cdata);
-
- // recv
- int bytes = recvmsg(o->fd, &msg, 0);
- if (bytes < 0) {
- if (errno == EAGAIN || errno == EWOULDBLOCK) {
- // wait for fd
- o->wait_events |= BREACTOR_READ;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
- return;
- }
-
- BLog(BLOG_ERROR, "recv failed");
- report_error(o);
- return;
- }
-
- ASSERT(bytes >= 0)
- ASSERT(bytes <= o->recv.mtu)
-
- // read returned address
- sysaddr.len = msg.msg_namelen;
- addr_sys_to_socket(&o->recv.remote_addr, sysaddr);
-
- // read returned local address
- BIPAddr_InitInvalid(&o->recv.local_addr);
- for (struct cmsghdr *cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
- #ifdef BADVPN_FREEBSD
- if (cmsg->cmsg_level == IPPROTO_IP && cmsg->cmsg_type == IP_RECVDSTADDR) {
- struct in_addr *addrinfo = (struct in_addr *)CMSG_DATA(cmsg);
- BIPAddr_InitIPv4(&o->recv.local_addr, addrinfo->s_addr);
- }
- #else
- if (cmsg->cmsg_level == IPPROTO_IP && cmsg->cmsg_type == IP_PKTINFO) {
- struct in_pktinfo *pktinfo = (struct in_pktinfo *)CMSG_DATA(cmsg);
- BIPAddr_InitIPv4(&o->recv.local_addr, pktinfo->ipi_addr.s_addr);
- }
- #endif
- else if (cmsg->cmsg_level == IPPROTO_IPV6 && cmsg->cmsg_type == IPV6_PKTINFO) {
- struct in6_pktinfo *pktinfo = (struct in6_pktinfo *)CMSG_DATA(cmsg);
- BIPAddr_InitIPv6(&o->recv.local_addr, pktinfo->ipi6_addr.s6_addr);
- }
- }
-
- // set have addresses
- o->recv.have_addrs = 1;
-
- // set not busy
- o->recv.busy = 0;
-
- // done
- PacketRecvInterface_Done(&o->recv.iface, bytes);
- }
- static void fd_handler (BDatagram *o, int events)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
-
- // clear handled events
- o->wait_events &= ~events;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
-
- int have_send = 0;
- int have_recv = 0;
-
- if ((events & BREACTOR_WRITE) || ((events & (BREACTOR_ERROR|BREACTOR_HUP)) && o->send.inited && o->send.busy && o->send.have_addrs)) {
- ASSERT(o->send.inited)
- ASSERT(o->send.busy)
- ASSERT(o->send.have_addrs)
-
- have_send = 1;
- }
-
- if ((events & BREACTOR_READ) || ((events & (BREACTOR_ERROR|BREACTOR_HUP)) && o->recv.inited && o->recv.busy && o->recv.started)) {
- ASSERT(o->recv.inited)
- ASSERT(o->recv.busy)
- ASSERT(o->recv.started)
-
- have_recv = 1;
- }
-
- if (have_send) {
- if (have_recv) {
- BPending_Set(&o->recv.job);
- }
-
- do_send(o);
- return;
- }
-
- if (have_recv) {
- do_recv(o);
- return;
- }
-
- BLog(BLOG_ERROR, "fd error event");
- report_error(o);
- return;
- }
- static void send_job_handler (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(o->send.inited)
- ASSERT(o->send.busy)
- ASSERT(o->send.have_addrs)
-
- do_send(o);
- return;
- }
- static void recv_job_handler (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(o->recv.inited)
- ASSERT(o->recv.busy)
- ASSERT(o->recv.started)
-
- do_recv(o);
- return;
- }
- static void send_if_handler_send (BDatagram *o, uint8_t *data, int data_len)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(o->send.inited)
- ASSERT(!o->send.busy)
- ASSERT(data_len >= 0)
- ASSERT(data_len <= o->send.mtu)
-
- // remember data
- o->send.busy_data = data;
- o->send.busy_data_len = data_len;
-
- // set busy
- o->send.busy = 1;
-
- // if have no addresses, wait
- if (!o->send.have_addrs) {
- return;
- }
-
- // set job
- BPending_Set(&o->send.job);
- }
- static void recv_if_handler_recv (BDatagram *o, uint8_t *data)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(o->recv.inited)
- ASSERT(!o->recv.busy)
-
- // remember data
- o->recv.busy_data = data;
-
- // set busy
- o->recv.busy = 1;
-
- // if recv not started yet, wait
- if (!o->recv.started) {
- return;
- }
-
- // set job
- BPending_Set(&o->recv.job);
- }
- int BDatagram_AddressFamilySupported (int family)
- {
- switch (family) {
- case BADDR_TYPE_IPV4:
- case BADDR_TYPE_IPV6:
- #ifdef BADVPN_LINUX
- case BADDR_TYPE_PACKET:
- #endif
- return 1;
- }
-
- return 0;
- }
- int BDatagram_Init (BDatagram *o, int family, BReactor *reactor, void *user,
- BDatagram_handler handler)
- {
- ASSERT(BDatagram_AddressFamilySupported(family))
- ASSERT(handler)
- BNetwork_Assert();
-
- // init arguments
- o->reactor = reactor;
- o->user = user;
- o->handler = handler;
-
- // init fd
- if ((o->fd = socket(family_socket_to_sys(family), SOCK_DGRAM, 0)) < 0) {
- BLog(BLOG_ERROR, "socket failed");
- goto fail0;
- }
-
- // set fd non-blocking
- if (!badvpn_set_nonblocking(o->fd)) {
- BLog(BLOG_ERROR, "badvpn_set_nonblocking failed");
- goto fail1;
- }
-
- // enable receiving pktinfo
- set_pktinfo(o->fd, family);
-
- // init BFileDescriptor
- BFileDescriptor_Init(&o->bfd, o->fd, (BFileDescriptor_handler)fd_handler, o);
- if (!BReactor_AddFileDescriptor(o->reactor, &o->bfd)) {
- BLog(BLOG_ERROR, "BReactor_AddFileDescriptor failed");
- goto fail1;
- }
-
- // set no wait events
- o->wait_events = 0;
-
- // init limits
- BReactorLimit_Init(&o->send.limit, o->reactor, BDATAGRAM_SEND_LIMIT);
- BReactorLimit_Init(&o->recv.limit, o->reactor, BDATAGRAM_RECV_LIMIT);
-
- // set have no send and recv addresses
- o->send.have_addrs = 0;
- o->recv.have_addrs = 0;
-
- // set recv not started
- o->recv.started = 0;
-
- // set send and recv not inited
- o->send.inited = 0;
- o->recv.inited = 0;
-
- DebugError_Init(&o->d_err, BReactor_PendingGroup(o->reactor));
- DebugObject_Init(&o->d_obj);
- return 1;
-
- fail1:
- if (close(o->fd) < 0) {
- BLog(BLOG_ERROR, "close failed");
- }
- fail0:
- return 0;
- }
- void BDatagram_Free (BDatagram *o)
- {
- DebugObject_Free(&o->d_obj);
- DebugError_Free(&o->d_err);
- ASSERT(!o->recv.inited)
- ASSERT(!o->send.inited)
-
- // free limits
- BReactorLimit_Free(&o->recv.limit);
- BReactorLimit_Free(&o->send.limit);
-
- // free BFileDescriptor
- BReactor_RemoveFileDescriptor(o->reactor, &o->bfd);
-
- // free fd
- if (close(o->fd) < 0) {
- BLog(BLOG_ERROR, "close failed");
- }
- }
- int BDatagram_Bind (BDatagram *o, BAddr addr)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(BDatagram_AddressFamilySupported(addr.type))
-
- // translate address
- struct sys_addr sysaddr;
- addr_socket_to_sys(&sysaddr, addr);
-
- // bind
- if (bind(o->fd, &sysaddr.addr.generic, sysaddr.len) < 0) {
- BLog(BLOG_ERROR, "bind failed");
- return 0;
- }
-
- // if recv wasn't started yet, start it
- if (!o->recv.started) {
- // set recv started
- o->recv.started = 1;
-
- // continue receiving
- if (o->recv.inited && o->recv.busy) {
- BPending_Set(&o->recv.job);
- }
- }
-
- return 1;
- }
- void BDatagram_SetSendAddrs (BDatagram *o, BAddr remote_addr, BIPAddr local_addr)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(BDatagram_AddressFamilySupported(remote_addr.type))
- ASSERT(local_addr.type == BADDR_TYPE_NONE || BDatagram_AddressFamilySupported(local_addr.type))
-
- // set addresses
- o->send.remote_addr = remote_addr;
- o->send.local_addr = local_addr;
-
- if (!o->send.have_addrs) {
- // set have addresses
- o->send.have_addrs = 1;
-
- // start sending
- if (o->send.inited && o->send.busy) {
- BPending_Set(&o->send.job);
- }
- }
- }
- int BDatagram_GetLastReceiveAddrs (BDatagram *o, BAddr *remote_addr, BIPAddr *local_addr)
- {
- DebugObject_Access(&o->d_obj);
-
- if (!o->recv.have_addrs) {
- return 0;
- }
-
- *remote_addr = o->recv.remote_addr;
- *local_addr = o->recv.local_addr;
- return 1;
- }
- int BDatagram_GetLocalPort (BDatagram *o, uint16_t *local_port)
- {
- DebugObject_Access(&o->d_obj);
-
- struct sys_addr sysaddr;
- BAddr addr;
- sysaddr.len = sizeof(sysaddr.addr);
- if (getsockname(o->fd, &sysaddr.addr.generic, &sysaddr.len) != 0) {
- BLog(BLOG_ERROR, "getsockname failed");
- return 0;
- }
- addr_sys_to_socket(&addr, sysaddr);
- if (addr.type == BADDR_TYPE_IPV4) {
- *local_port = addr.ipv4.port;
- return 1;
- }
- if (addr.type == BADDR_TYPE_IPV6) {
- *local_port = addr.ipv6.port;
- return 1;
- }
- BLog(BLOG_ERROR, "Unknown address type from getsockname: %d", addr.type);
- return 0;
- }
- int BDatagram_GetFd (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
-
- return o->fd;
- }
- int BDatagram_SetReuseAddr (BDatagram *o, int reuse)
- {
- DebugObject_Access(&o->d_obj);
- ASSERT(reuse == 0 || reuse == 1)
-
- if (setsockopt(o->fd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0) {
- return 0;
- }
-
- return 1;
- }
- void BDatagram_SendAsync_Init (BDatagram *o, int mtu)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(!o->send.inited)
- ASSERT(mtu >= 0)
-
- // init arguments
- o->send.mtu = mtu;
-
- // init interface
- PacketPassInterface_Init(&o->send.iface, o->send.mtu, (PacketPassInterface_handler_send)send_if_handler_send, o, BReactor_PendingGroup(o->reactor));
-
- // init job
- BPending_Init(&o->send.job, BReactor_PendingGroup(o->reactor), (BPending_handler)send_job_handler, o);
-
- // set not busy
- o->send.busy = 0;
-
- // set inited
- o->send.inited = 1;
- }
- void BDatagram_SendAsync_Free (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
- ASSERT(o->send.inited)
-
- // update events
- o->wait_events &= ~BREACTOR_WRITE;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
-
- // free job
- BPending_Free(&o->send.job);
-
- // free interface
- PacketPassInterface_Free(&o->send.iface);
-
- // set not inited
- o->send.inited = 0;
- }
- PacketPassInterface * BDatagram_SendAsync_GetIf (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
- ASSERT(o->send.inited)
-
- return &o->send.iface;
- }
- void BDatagram_RecvAsync_Init (BDatagram *o, int mtu)
- {
- DebugObject_Access(&o->d_obj);
- DebugError_AssertNoError(&o->d_err);
- ASSERT(!o->recv.inited)
- ASSERT(mtu >= 0)
-
- // init arguments
- o->recv.mtu = mtu;
-
- // init interface
- PacketRecvInterface_Init(&o->recv.iface, o->recv.mtu, (PacketRecvInterface_handler_recv)recv_if_handler_recv, o, BReactor_PendingGroup(o->reactor));
-
- // init job
- BPending_Init(&o->recv.job, BReactor_PendingGroup(o->reactor), (BPending_handler)recv_job_handler, o);
-
- // set not busy
- o->recv.busy = 0;
-
- // set inited
- o->recv.inited = 1;
- }
- void BDatagram_RecvAsync_Free (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
- ASSERT(o->recv.inited)
-
- // update events
- o->wait_events &= ~BREACTOR_READ;
- BReactor_SetFileDescriptorEvents(o->reactor, &o->bfd, o->wait_events);
-
- // free job
- BPending_Free(&o->recv.job);
-
- // free interface
- PacketRecvInterface_Free(&o->recv.iface);
-
- // set not inited
- o->recv.inited = 0;
- }
- PacketRecvInterface * BDatagram_RecvAsync_GetIf (BDatagram *o)
- {
- DebugObject_Access(&o->d_obj);
- ASSERT(o->recv.inited)
-
- return &o->recv.iface;
- }
|