#include <net/if.h>
#include <netinet/in.h>
#include <arpa/inet.h>
+#include <sys/types.h>
+#include <ifaddrs.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <libubox/uloop.h>
#include <libubox/avl-cmp.h>
#include <libubox/utils.h>
+#include "cache.h"
#include "interface.h"
#include "util.h"
#include "dns.h"
#include "announce.h"
+#include "service.h"
+
+static struct uloop_fd ufd[] = {
+ [SOCK_UC_IPV4] = { .fd = -1 },
+ [SOCK_UC_IPV6] = { .fd = -1 },
+ [SOCK_MC_IPV4] = { .fd = -1 },
+ [SOCK_MC_IPV6] = { .fd = -1 },
+};
static int
-interface_send_packet4(struct interface *iface, struct iovec *iov, int iov_len)
+interface_send_packet4(struct interface *iface, struct sockaddr_in *to, struct iovec *iov, int iov_len)
{
static size_t cmsg_data[( CMSG_SPACE(sizeof(struct in_pktinfo)) / sizeof(size_t)) + 1];
- static struct sockaddr_in a;
+ static struct sockaddr_in a = {};
static struct msghdr m = {
.msg_name = (struct sockaddr *) &a,
.msg_namelen = sizeof(a),
};
struct in_pktinfo *pkti;
struct cmsghdr *cmsg;
- int fd = iface->fd.fd;
+ int fd;
a.sin_family = AF_INET;
a.sin_port = htons(MCAST_PORT);
pkti = (struct in_pktinfo*) CMSG_DATA(cmsg);
pkti->ipi_ifindex = iface->ifindex;
- a.sin_addr.s_addr = inet_addr(MCAST_ADDR);
+ fd = ufd[iface->type].fd;
+ if (interface_multicast(iface)) {
+ a.sin_addr.s_addr = inet_addr(MCAST_ADDR);
+ if (to)
+ fprintf(stderr, "Ignoring IPv4 address for multicast interface\n");
+ } else {
+ a.sin_addr.s_addr = to->sin_addr.s_addr;
+ }
return sendmsg(fd, &m, 0);
}
static int
-interface_send_packet6(struct interface *iface, struct iovec *iov, int iov_len)
+interface_send_packet6(struct interface *iface, struct sockaddr_in6 *to, struct iovec *iov, int iov_len)
{
static size_t cmsg_data[( CMSG_SPACE(sizeof(struct in6_pktinfo)) / sizeof(size_t)) + 1];
- static struct sockaddr_in6 a;
+ static struct sockaddr_in6 a = {};
static struct msghdr m = {
.msg_name = (struct sockaddr *) &a,
.msg_namelen = sizeof(a),
};
struct in6_pktinfo *pkti;
struct cmsghdr *cmsg;
- int fd = iface->fd.fd;
+ int fd;
a.sin6_family = AF_INET6;
a.sin6_port = htons(MCAST_PORT);
+ a.sin6_scope_id = iface->ifindex;
m.msg_iov = iov;
m.msg_iovlen = iov_len;
pkti = (struct in6_pktinfo*) CMSG_DATA(cmsg);
pkti->ipi6_ifindex = iface->ifindex;
- inet_pton(AF_INET6, MCAST_ADDR6, &a.sin6_addr);
+ fd = ufd[iface->type].fd;
+ if (interface_multicast(iface)) {
+ inet_pton(AF_INET6, MCAST_ADDR6, &a.sin6_addr);
+ if (to)
+ fprintf(stderr, "Ignoring IPv6 address for multicast interface\n");
+ } else {
+ a.sin6_addr = to->sin6_addr;
+ }
return sendmsg(fd, &m, 0);
}
int
-interface_send_packet(struct interface *iface, struct iovec *iov, int iov_len)
+interface_send_packet(struct interface *iface, struct sockaddr *to, struct iovec *iov, int iov_len)
{
- if (iface->v6)
- return interface_send_packet6(iface, iov, iov_len);
+ if (!interface_multicast(iface) && !to) {
+ fprintf(stderr, "No IP address specified for unicast interface\n");
+ errno = EINVAL;
+ return -1;
+ }
+
+ if (debug > 1) {
+ fprintf(stderr, "TX ipv%d: %s\n", interface_ipv6(iface) ? 6 : 4, iface->name);
+ fprintf(stderr, " multicast: %d\n", interface_multicast(iface));
+ }
- return interface_send_packet4(iface, iov, iov_len);
+ if (interface_ipv6(iface))
+ return interface_send_packet6(iface, (struct sockaddr_in6 *)to, iov, iov_len);
+
+ return interface_send_packet4(iface, (struct sockaddr_in *)to, iov, iov_len);
}
-static void interface_close(struct interface *iface)
+static struct interface *interface_lookup(unsigned int ifindex, enum umdns_socket_type type)
{
- if (iface->fd.fd < 0)
- return;
+ struct interface *iface;
- announce_free(iface);
- uloop_fd_delete(&iface->fd);
- close(iface->fd.fd);
- iface->fd.fd = -1;
+ vlist_for_each_element(&interfaces, iface, node)
+ if (iface->ifindex == ifindex && iface->type == type)
+ return iface;
+
+ return NULL;
}
static void interface_free(struct interface *iface)
{
- interface_close(iface);
+ cache_cleanup(iface);
+ announce_free(iface);
+ free(iface->addrs.v4);
free(iface);
}
+static int
+interface_valid_src(void *ip1, void *mask, void *ip2, int len)
+{
+ uint8_t *i1 = ip1;
+ uint8_t *i2 = ip2;
+ uint8_t *m = mask;
+ int i;
+
+ if (cfg_no_subnet)
+ return 0;
+
+ for (i = 0; i < len; i++, i1++, i2++, m++) {
+ if ((*i1 & *m) != (*i2 & *m))
+ return -1;
+ }
+
+ return 0;
+}
+
+static void
+read_socket4(struct uloop_fd *u, unsigned int events)
+{
+ enum umdns_socket_type type = (enum umdns_socket_type)(u - ufd);
+ struct interface *iface;
+ static uint8_t buffer[8 * 1024];
+ struct iovec iov[1];
+ char cmsg[CMSG_SPACE(sizeof(struct in_pktinfo)) + CMSG_SPACE(sizeof(int)) + 1];
+ struct cmsghdr *cmsgptr;
+ struct msghdr msg;
+ socklen_t len;
+ struct sockaddr_in from;
+ int flags = 0;
+ uint8_t ttl = 0;
+ struct in_pktinfo *inp = NULL;
+ bool valid_src = false;
+
+ if (u->eof) {
+ uloop_end();
+ return;
+ }
+
+ iov[0].iov_base = buffer;
+ iov[0].iov_len = sizeof(buffer);
+
+ memset(&msg, 0, sizeof(msg));
+ msg.msg_name = (struct sockaddr *) &from;
+ msg.msg_namelen = sizeof(struct sockaddr_in);
+ msg.msg_iov = iov;
+ msg.msg_iovlen = 1;
+ msg.msg_control = &cmsg;
+ msg.msg_controllen = sizeof(cmsg);
+
+ len = recvmsg(u->fd, &msg, flags);
+ if (len == -1) {
+ perror("read failed");
+ return;
+ }
+ for (cmsgptr = CMSG_FIRSTHDR(&msg); cmsgptr != NULL; cmsgptr = CMSG_NXTHDR(&msg, cmsgptr)) {
+ void *c = CMSG_DATA(cmsgptr);
+
+ switch (cmsgptr->cmsg_type) {
+ case IP_PKTINFO:
+ inp = ((struct in_pktinfo *) c);
+ break;
+
+ case IP_TTL:
+ ttl = (uint8_t) *((int *) c);
+ break;
+
+ default:
+ fprintf(stderr, "unknown cmsg %x\n", cmsgptr->cmsg_type);
+ return;
+ }
+ }
+
+ if (!inp)
+ return;
+
+ iface = interface_lookup(inp->ipi_ifindex, type);
+ if (!iface)
+ return;
+
+ if (debug > 1) {
+ char buf[256];
+
+ fprintf(stderr, "RX ipv4: %s\n", iface->name);
+ fprintf(stderr, " multicast: %d\n", interface_multicast(iface));
+ inet_ntop(AF_INET, &from.sin_addr, buf, 256);
+ fprintf(stderr, " src %s:%d\n", buf, ntohs(from.sin_port));
+ inet_ntop(AF_INET, &inp->ipi_spec_dst, buf, 256);
+ fprintf(stderr, " dst %s\n", buf);
+ inet_ntop(AF_INET, &inp->ipi_addr, buf, 256);
+ fprintf(stderr, " real %s\n", buf);
+ fprintf(stderr, " ttl %u\n", ttl);
+ }
+
+ for (size_t i = 0; i < iface->addrs.n_addr; i++) {
+ if (!interface_valid_src((void *)&iface->addrs.v4[i].addr,
+ (void *)&iface->addrs.v4[i].mask,
+ (void *) &from.sin_addr, 4)) {
+ valid_src = true;
+ break;
+ }
+ }
+
+ if (!valid_src)
+ return;
+
+ dns_handle_packet(iface, (struct sockaddr *) &from, ntohs(from.sin_port), buffer, len);
+}
+
static void
-read_socket(struct uloop_fd *u, unsigned int events)
+read_socket6(struct uloop_fd *u, unsigned int events)
{
- struct interface *iface = container_of(u, struct interface, fd);
+ enum umdns_socket_type type = (enum umdns_socket_type)(u - ufd);
+ struct interface *iface;
static uint8_t buffer[8 * 1024];
- int len;
+ struct iovec iov[1];
+ char cmsg6[CMSG_SPACE(sizeof(struct in6_pktinfo)) + CMSG_SPACE(sizeof(int)) + 1];
+ struct cmsghdr *cmsgptr;
+ struct msghdr msg;
+ socklen_t len;
+ struct sockaddr_in6 from;
+ int flags = 0;
+ int ttl = 0;
+ struct in6_pktinfo *inp = NULL;
+ bool valid_src = false;
if (u->eof) {
- interface_close(iface);
- uloop_timeout_set(&iface->reconnect, 1000);
+ uloop_end();
+ return;
+ }
+
+ iov[0].iov_base = buffer;
+ iov[0].iov_len = sizeof(buffer);
+
+ memset(&msg, 0, sizeof(msg));
+ msg.msg_name = (struct sockaddr *) &from;
+ msg.msg_namelen = sizeof(struct sockaddr_in6);
+ msg.msg_iov = iov;
+ msg.msg_iovlen = 1;
+ msg.msg_control = &cmsg6;
+ msg.msg_controllen = sizeof(cmsg6);
+
+ len = recvmsg(u->fd, &msg, flags);
+ if (len == -1) {
+ perror("read failed");
return;
}
+ for (cmsgptr = CMSG_FIRSTHDR(&msg); cmsgptr != NULL; cmsgptr = CMSG_NXTHDR(&msg, cmsgptr)) {
+ void *c = CMSG_DATA(cmsgptr);
+
+ switch (cmsgptr->cmsg_type) {
+ case IPV6_PKTINFO:
+ inp = ((struct in6_pktinfo *) c);
+ break;
+
+ case IPV6_HOPLIMIT:
+ ttl = (uint8_t) *((int *) c);
+ break;
+
+ default:
+ fprintf(stderr, "unknown cmsg %x\n", cmsgptr->cmsg_type);
+ return;
+ }
+ }
+
+ if (!inp)
+ return;
- len = read(u->fd, buffer, sizeof(buffer));
- if (len < 1) {
- if (errno != EAGAIN)
- fprintf(stderr, "read failed: %s\n", strerror(errno));
+ iface = interface_lookup(inp->ipi6_ifindex, type);
+ if (!iface)
return;
+
+ if (debug > 1) {
+ char buf[256];
+
+ fprintf(stderr, "RX ipv6: %s\n", iface->name);
+ fprintf(stderr, " multicast: %d\n", interface_multicast(iface));
+ inet_ntop(AF_INET6, &from.sin6_addr, buf, 256);
+ fprintf(stderr, " src %s:%d\n", buf, ntohs(from.sin6_port));
+ inet_ntop(AF_INET6, &inp->ipi6_addr, buf, 256);
+ fprintf(stderr, " dst %s\n", buf);
+ fprintf(stderr, " ttl %u\n", ttl);
+ }
+
+ for (size_t i = 0; i < iface->addrs.n_addr; i++) {
+ if (!interface_valid_src((void *)&iface->addrs.v6[i].addr,
+ (void *)&iface->addrs.v6[i].mask,
+ (void *)&from.sin6_addr, 6)) {
+ valid_src = true;
+ break;
+ }
}
- dns_handle_packet(iface, buffer, len);
+ if (!valid_src)
+ return;
+
+ dns_handle_packet(iface, (struct sockaddr *) &from, ntohs(from.sin6_port), buffer, len);
}
static int
-interface_socket_setup4(struct interface *iface)
+interface_mcast_setup4(struct interface *iface)
{
struct ip_mreqn mreq;
- uint8_t ttl = 255;
- int yes = 1;
- int no = 0;
- struct sockaddr_in sa = { 0 };
- int fd = iface->fd.fd;
+ struct sockaddr_in sa = {};
+ int fd = ufd[SOCK_MC_IPV4].fd;
sa.sin_family = AF_INET;
sa.sin_port = htons(MCAST_PORT);
inet_pton(AF_INET, MCAST_ADDR, &sa.sin_addr);
memset(&mreq, 0, sizeof(mreq));
- mreq.imr_address.s_addr = iface->v4_addr.s_addr;
mreq.imr_multiaddr = sa.sin_addr;
mreq.imr_ifindex = iface->ifindex;
-
- if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl)) < 0)
- fprintf(stderr, "ioctl failed: IP_MULTICAST_TTL\n");
-
- if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0)
- fprintf(stderr, "ioctl failed: SO_REUSEADDR\n");
+ mreq.imr_address.s_addr = iface->addrs.v4[0].addr.s_addr;
/* Some network drivers have issues with dropping membership of
* mcast groups when the iface is down, but don't allow rejoining
* -- this was copied from avahi --
*/
setsockopt(fd, IPPROTO_IP, IP_DROP_MEMBERSHIP, &mreq, sizeof(mreq));
-
- if (setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) < 0) {
- fprintf(stderr, "failed to join multicast group: %s\n", strerror(errno));
- close(fd);
- fd = -1;
- return -1;
- }
-
- if (setsockopt(fd, IPPROTO_IP, IP_RECVTTL, &yes, sizeof(yes)) < 0)
- fprintf(stderr, "ioctl failed: IP_RECVTTL\n");
-
- if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &yes, sizeof(yes)) < 0)
- fprintf(stderr, "ioctl failed: IP_PKTINFO\n");
-
- if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP, &no, sizeof(no)) < 0)
- fprintf(stderr, "ioctl failed: IP_MULTICAST_LOOP\n");
+ setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq));
return 0;
}
static int
-interface_socket_setup6(struct interface *iface)
+interface_mcast_setup6(struct interface *iface)
{
struct ipv6_mreq mreq;
- int ttl = 255;
- int yes = 1;
- int no = 0;
- struct sockaddr_in6 sa = { 0 };
- int fd = iface->fd.fd;
+ struct sockaddr_in6 sa = {};
+ int fd = ufd[SOCK_MC_IPV6].fd;
sa.sin6_family = AF_INET6;
sa.sin6_port = htons(MCAST_PORT);
mreq.ipv6mr_multiaddr = sa.sin6_addr;
mreq.ipv6mr_interface = iface->ifindex;
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &ttl, sizeof(ttl)) < 0)
- fprintf(stderr, "ioctl failed: IPV6_MULTICAST_HOPS\n");
-
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &ttl, sizeof(ttl)) < 0)
- fprintf(stderr, "ioctl failed: IPV6_UNICAST_HOPS\n");
-
- if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0)
- fprintf(stderr, "ioctl failed: SO_REUSEADDR\n");
-
setsockopt(fd, IPPROTO_IPV6, IPV6_LEAVE_GROUP, &mreq, sizeof(mreq));
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) < 0) {
- fprintf(stderr, "failed to join multicast group: %s\n", strerror(errno));
- close(fd);
- fd = -1;
- return -1;
- }
-
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &yes, sizeof(yes)) < 0)
- fprintf(stderr, "ioctl failed: IPV6_RECVPKTINFO\n");
-
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_RECVHOPLIMIT, &yes, sizeof(yes)) < 0)
- fprintf(stderr, "ioctl failed: IPV6_RECVHOPLIMIT\n");
-
- if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, &no, sizeof(no)) < 0)
- fprintf(stderr, "ioctl failed: IPV6_MULTICAST_LOOP\n");
+ setsockopt(fd, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, &mreq, sizeof(mreq));
return 0;
}
-static void
-reconnect_socket(struct uloop_timeout *timeout)
+static void interface_start(struct interface *iface)
{
- struct interface *iface = container_of(timeout, struct interface, reconnect);
- char mcast_addr[16];
- int type = 0;
-
- if (iface->v6) {
- snprintf(mcast_addr, sizeof(mcast_addr), "%s%%%s", iface->mcast_addr, iface->name);
- type = USOCK_IPV6ONLY;
- } else {
- snprintf(mcast_addr, sizeof(mcast_addr), "%s", iface->mcast_addr);
- type = USOCK_IPV4ONLY;
- }
-
- iface->fd.fd = usock(USOCK_UDP | USOCK_SERVER | USOCK_NONBLOCK | type, mcast_addr, "5353");
- if (iface->fd.fd < 0) {
- fprintf(stderr, "failed to add listener %s: %s\n", mcast_addr, strerror(errno));
- goto retry;
- }
+ if (iface->type & SOCKTYPE_BIT_UNICAST)
+ return;
- if (!iface->v6 && interface_socket_setup4(iface)) {
- iface->fd.fd = -1;
- goto retry;
- }
+ if (iface->type & SOCKTYPE_BIT_IPV6)
+ interface_mcast_setup6(iface);
+ else
+ interface_mcast_setup4(iface);
- if (iface->v6 && interface_socket_setup6(iface)) {
- iface->fd.fd = -1;
- goto retry;
- }
-
- uloop_fd_add(&iface->fd, ULOOP_READ);
- dns_send_question(iface, "_services._dns-sd._udp.local", TYPE_PTR);
+ dns_send_question(iface, NULL, C_DNS_SD, TYPE_PTR, 0);
announce_init(iface);
- return;
-
-retry:
- uloop_timeout_set(timeout, 1000);
}
-
-static void interface_start(struct interface *iface)
+static bool
+iface_equal(struct interface *if_old, struct interface *if_new)
{
- iface->fd.cb = read_socket;
- iface->reconnect.cb = reconnect_socket;
- uloop_timeout_set(&iface->reconnect, 100);
+ size_t addr_size;
+
+ if (if_old->ifindex != if_new->ifindex ||
+ if_old->addrs.n_addr != if_new->addrs.n_addr)
+ return false;
+
+ if (if_old->type & SOCKTYPE_BIT_IPV6)
+ addr_size = sizeof(*if_old->addrs.v6);
+ else
+ addr_size = sizeof(*if_old->addrs.v4);
+ addr_size *= if_old->addrs.n_addr;
+ if (memcmp(if_old->addrs.v4, if_new->addrs.v4, addr_size) != 0)
+ return false;
+
+ return true;
}
static void
iface_update_cb(struct vlist_tree *tree, struct vlist_node *node_new,
struct vlist_node *node_old)
{
- struct interface *iface;
-
- if (node_old) {
- iface = container_of(node_old, struct interface, node);
- interface_free(iface);
+ struct interface *if_old = container_of_safe(node_old, struct interface, node);
+ struct interface *if_new = container_of_safe(node_new, struct interface, node);
+
+ if (if_old && if_new) {
+ if (!iface_equal(if_old, if_new))
+ cache_cleanup(if_old);
+ free(if_old->addrs.v4);
+ if_old->addrs = if_new->addrs;
+ if_old->ifindex = if_new->ifindex;
+ free(if_new);
+ return;
}
- if (node_new) {
- iface = container_of(node_new, struct interface, node);
- interface_start(iface);
- }
+ if (if_old)
+ interface_free(if_old);
+
+ if (if_new)
+ interface_start(if_new);
}
-static int
-get_iface_ipv4(struct interface *iface)
+static int interface_init_socket(enum umdns_socket_type type)
{
- struct sockaddr_in *sin;
- struct ifreq ir;
- int sock, ret = -1;
+ struct sockaddr_in6 local6 = {
+ .sin6_family = AF_INET6
+ };
+ struct sockaddr_in local = {
+ .sin_family = AF_INET
+ };
+ uint8_t ttl = 255;
+ int ittl = 255;
+ int yes = 1;
+ int no = 0;
+ int fd;
+ int af = (type & SOCKTYPE_BIT_IPV6) ? AF_INET6 : AF_INET;
+
+ if (ufd[type].fd >= 0)
+ return 0;
- sock = socket(AF_INET, SOCK_DGRAM, 0);
- if (sock < 0)
+ ufd[type].fd = fd = socket(af, SOCK_DGRAM, 0);
+ if (fd < 0)
return -1;
- memset(&ir, 0, sizeof(struct ifreq));
- strncpy(ir.ifr_name, iface->name, sizeof(ir.ifr_name));
+ setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes));
+#ifdef SO_REUSEPORT
+ setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &yes, sizeof(yes));
+#endif
+
+ switch (type) {
+ case SOCK_UC_IPV4:
+ case SOCK_UC_IPV6:
+ break;
+ case SOCK_MC_IPV4:
+ setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl));
+ setsockopt(fd, IPPROTO_IP, IP_TTL, &ittl, sizeof(ittl));
+ setsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP, &no, sizeof(no));
+ local.sin_port = htons(MCAST_PORT);
+ break;
+ case SOCK_MC_IPV6:
+ setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &ttl, sizeof(ttl));
+ setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &ttl, sizeof(ttl));
+ setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &yes, sizeof(yes));
+ setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, &no, sizeof(no));
+ local6.sin6_port = htons(MCAST_PORT);
+ break;
+ }
- ret = ioctl(sock, SIOCGIFADDR, &ir);
- if (ret < 0)
- goto out;
+ if (type & SOCKTYPE_BIT_IPV6) {
+ ufd[type].cb = read_socket6;
+ if (bind(fd, (struct sockaddr *)&local6, sizeof(local6)) < 0)
+ goto error;
- sin = (struct sockaddr_in *) &ir.ifr_addr;
- memcpy(&iface->v4_addr, &sin->sin_addr, sizeof(iface->v4_addr));
- iface->mcast_addr = MCAST_ADDR;
-out:
- close(sock);
- return ret;
-}
+ setsockopt(fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &yes, sizeof(yes));
+ setsockopt(fd, IPPROTO_IPV6, IPV6_RECVHOPLIMIT, &yes, sizeof(yes));
+ } else {
+ ufd[type].cb = read_socket4;
+ if (bind(fd, (struct sockaddr *)&local, sizeof(local)) < 0)
+ goto error;
-static int
-get_iface_ipv6(struct interface *iface)
-{
- struct sockaddr_in6 addr = {AF_INET6, 0, iface->ifindex, IN6ADDR_ANY_INIT, 0};
- socklen_t alen = sizeof(addr);
- int sock, ret = -1;
-
- addr.sin6_addr.s6_addr[0] = 0xff;
- addr.sin6_addr.s6_addr[1] = 0x02;
- addr.sin6_addr.s6_addr[15] = 0x01;
-
- sock = socket(AF_INET6, SOCK_RAW, IPPROTO_ICMPV6);
- connect(sock, (struct sockaddr*)&addr, sizeof(addr));
- ret = getsockname(sock, (struct sockaddr*)&addr, &alen);
- if (!ret) {
- memcpy(&iface->v6_addr, &addr.sin6_addr, sizeof(iface->v6_addr));
- iface->mcast_addr = MCAST_ADDR6;
- iface->v6 = 1;
+ setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &yes, sizeof(yes));
+ setsockopt(fd, IPPROTO_IP, IP_RECVTTL, &yes, sizeof(yes));
}
- close(sock);
- return ret;
+
+ uloop_fd_add(&ufd[type], ULOOP_READ);
+
+ return 0;
+
+error:
+ close(ufd[type].fd);
+ return -1;
}
-static int _interface_add(const char *name, int v6)
+static void
+__interface_add(const char *name, enum umdns_socket_type type,
+ struct interface_addr_list *list)
{
struct interface *iface;
- char *name_buf;
+ unsigned int ifindex;
char *id_buf;
- iface = calloc_a(sizeof(*iface),
- &name_buf, strlen(name) + 1,
- &id_buf, strlen(name) + 3);
-
- sprintf(id_buf, "%d_%s", v6, name);
- iface->name = strcpy(name_buf, name);
- iface->id = id_buf;
- iface->ifindex = if_nametoindex(name);
- iface->fd.fd = -1;
-
- if (iface->ifindex <= 0)
+ if (interface_init_socket(type))
goto error;
- if (!v6 && get_iface_ipv4(iface))
+ ifindex = if_nametoindex(name);
+ if (!ifindex)
goto error;
- if (v6 && get_iface_ipv6(iface))
- goto error;
+ iface = calloc_a(sizeof(*iface),
+ &id_buf, strlen(name) + 3);
- vlist_add(&interfaces, &iface->node, iface->id);
- return 0;
+ sprintf(id_buf, "%d_%s", type, name);
+ iface->name = id_buf + 2;
+ iface->ifindex = ifindex;
+ iface->type = type;
+ iface->addrs = *list;
+
+ vlist_add(&interfaces, &iface->node, id_buf);
+ return;
error:
- free(iface);
- return -1;
+ free(list->v4);
}
int interface_add(const char *name)
{
- int v4 = _interface_add(name, 0);
- int v6 = _interface_add(name, 1);
- return v4 && v6;
+ struct ifaddrs *ifap, *ifa;
+ struct interface_addr_list addr4 = {}, addr6 = {};
+
+ getifaddrs(&ifap);
+
+ for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
+ if (strcmp(ifa->ifa_name, name))
+ continue;
+ if (ifa->ifa_addr->sa_family == AF_INET) {
+ struct sockaddr_in *sin;
+
+ if (cfg_proto && (cfg_proto != 4))
+ continue;
+
+ addr4.v4 = realloc(addr4.v4, (addr4.n_addr + 1) * sizeof(*addr4.v4));
+ sin = (struct sockaddr_in *) ifa->ifa_addr;
+ addr4.v4[addr4.n_addr].addr = sin->sin_addr;
+ sin = (struct sockaddr_in *) ifa->ifa_netmask;
+ addr4.v4[addr4.n_addr++].mask = sin->sin_addr;
+ }
+
+ if (ifa->ifa_addr->sa_family == AF_INET6) {
+ uint8_t ll_prefix[] = {0xfe, 0x80 };
+ struct sockaddr_in6 *sin6;
+
+ if (cfg_proto && (cfg_proto != 6))
+ continue;
+
+ sin6 = (struct sockaddr_in6 *) ifa->ifa_addr;
+ if (memcmp(&sin6->sin6_addr, &ll_prefix, 2))
+ continue;
+
+ addr6.v6 = realloc(addr6.v6, (addr6.n_addr + 1) * sizeof(*addr6.v6));
+ sin6 = (struct sockaddr_in6 *) ifa->ifa_addr;
+ addr6.v6[addr6.n_addr].addr = sin6->sin6_addr;
+ sin6 = (struct sockaddr_in6 *) ifa->ifa_netmask;
+ addr6.v6[addr6.n_addr++].mask = sin6->sin6_addr;
+ }
+ }
+
+ freeifaddrs(ifap);
+
+ if (addr4.n_addr) {
+ size_t addr_size = addr4.n_addr * sizeof(*addr4.v4);
+ void *addr_dup = malloc(addr_size);
+
+ memcpy(addr_dup, addr4.v4, addr_size);
+ __interface_add(name, SOCK_UC_IPV4, &addr4);
+ addr4.v4 = addr_dup;
+ __interface_add(name, SOCK_MC_IPV4, &addr4);
+ }
+
+ if (addr6.n_addr) {
+ size_t addr_size = addr6.n_addr * sizeof(*addr6.v6);
+ void *addr_dup = malloc(addr_size);
+
+ memcpy(addr_dup, addr6.v6, addr_size);
+ __interface_add(name, SOCK_UC_IPV6, &addr6);
+ addr6.v6 = addr_dup;
+ __interface_add(name, SOCK_MC_IPV6, &addr6);
+ }
+
+ return !addr4.n_addr && !addr6.n_addr;
+}
+
+void interface_shutdown(void)
+{
+ struct interface *iface;
+
+ vlist_for_each_element(&interfaces, iface, node)
+ if (interface_multicast(iface)) {
+ dns_reply_a(iface, NULL, 0, NULL);
+ dns_reply_a_additional(iface, NULL, 0);
+ service_announce_services(iface, NULL, 0);
+ }
+
+ for (size_t i = 0; i < ARRAY_SIZE(ufd); i++) {
+ uloop_fd_delete(&ufd[i]);
+ close(ufd[i].fd);
+ ufd[i].fd = -1;
+ }
+}
+
+struct interface *interface_get(const char *name, enum umdns_socket_type type)
+{
+ char id_buf[32];
+ snprintf(id_buf, sizeof(id_buf), "%d_%s", type, name);
+ struct interface *iface = vlist_find(&interfaces, id_buf, iface, node);
+ return iface;
}
-VLIST_TREE(interfaces, avl_strcmp, iface_update_cb, false, false);
+VLIST_TREE(interfaces, avl_strcmp, iface_update_cb, true, false);
projects / project / mdnsd.git / blobdiff