X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fmdnsd.git;a=blobdiff_plain;f=interface.c;h=463335a89034692d2ca1048948dc11d0f528ab5b;hp=8df85bd388269838669ec6fe815ab92e8c0a301a;hb=c0c78c92d57c53276bb7f08468bcc716ba272497;hpb=db6febc526f9b235c4596e7d793c374f35f6edc1 diff --git a/interface.c b/interface.c index 8df85bd..463335a 100644 --- a/interface.c +++ b/interface.c @@ -21,7 +21,9 @@ #include #include #include +#include +#include #include #include #include @@ -36,6 +38,7 @@ #include "util.h" #include "dns.h" #include "announce.h" +#include "service.h" static int interface_send_packet4(struct interface *iface, struct iovec *iov, int iov_len) @@ -108,6 +111,11 @@ interface_send_packet6(struct interface *iface, struct iovec *iov, int iov_len) int interface_send_packet(struct interface *iface, struct iovec *iov, int iov_len) { + if (debug > 1) { + fprintf(stderr, "TX ipv%d: %s\n", iface->v6 * 2 + 4, iface->name); + fprintf(stderr, " multicast: %d\n", iface->multicast); + } + if (iface->v6) return interface_send_packet6(iface, iov, iov_len); @@ -131,12 +139,39 @@ static void interface_free(struct interface *iface) 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_socket(struct uloop_fd *u, unsigned int events) +read_socket4(struct uloop_fd *u, unsigned int events) { struct interface *iface = container_of(u, struct interface, fd); static uint8_t buffer[8 * 1024]; - int len; + 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, ifindex = -1; + uint8_t ttl = 0; + struct in_pktinfo *inp = NULL; if (u->eof) { interface_close(iface); @@ -144,22 +179,142 @@ read_socket(struct uloop_fd *u, unsigned int events) return; } - len = read(u->fd, buffer, sizeof(buffer)); - if (len < 1) { - if (errno != EAGAIN) - fprintf(stderr, "read failed: %s\n", strerror(errno)); + 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 (ttl != 255) + return; + + if (debug > 1) { + char buf[256]; + + fprintf(stderr, "RX ipv4: %s\n", iface->name); + fprintf(stderr, " multicast: %d\n", iface->multicast); + inet_ntop(AF_INET, &from.sin_addr, buf, 256); + fprintf(stderr, " src %s:%d\n", buf, 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); + } + + if (inp->ipi_ifindex != iface->ifindex) + fprintf(stderr, "invalid iface index %d != %d\n", ifindex, iface->ifindex); + else if (!interface_valid_src((void *) &iface->v4_addr, (void *) &iface->v4_netmask, (void *) &from.sin_addr, 4)) + dns_handle_packet(iface, (struct sockaddr *) &from, from.sin_port, buffer, len); +} + +static void +read_socket6(struct uloop_fd *u, unsigned int events) +{ + struct interface *iface = container_of(u, struct interface, fd); + static uint8_t buffer[8 * 1024]; + 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, ifindex = -1; + int ttl = 0; + struct in6_pktinfo *inp = NULL; + + if (u->eof) { + interface_close(iface); + uloop_timeout_set(&iface->reconnect, 1000); + 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 (ttl != 255) return; + + if (debug > 1) { + char buf[256]; + + fprintf(stderr, "RX ipv6: %s\n", iface->name); + fprintf(stderr, " multicast: %d\n", iface->multicast); + inet_ntop(AF_INET6, &from.sin6_addr, buf, 256); + fprintf(stderr, " src %s:%d\n", buf, from.sin6_port); + inet_ntop(AF_INET6, &inp->ipi6_addr, buf, 256); + fprintf(stderr, " dst %s\n", buf); } - dns_handle_packet(iface, buffer, len); + if (inp->ipi6_ifindex != iface->ifindex) + fprintf(stderr, "invalid iface index %d != %d\n", ifindex, iface->ifindex); + else if (!interface_valid_src((void *) &iface->v6_addr, (void *) &iface->v6_netmask, (void *) &from.sin6_addr, 16)) + dns_handle_packet(iface, (struct sockaddr *) &from, 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; @@ -176,9 +331,6 @@ interface_socket_setup4(struct interface *iface) 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"); - /* Some network drivers have issues with dropping membership of * mcast groups when the iface is down, but don't allow rejoining * when it comes back up. This is an ugly workaround @@ -193,12 +345,6 @@ interface_socket_setup4(struct interface *iface) 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"); @@ -210,7 +356,6 @@ interface_socket_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; @@ -226,12 +371,6 @@ interface_socket_setup6(struct interface *iface) 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)); @@ -240,12 +379,6 @@ interface_socket_setup6(struct interface *iface) 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"); @@ -253,39 +386,89 @@ interface_socket_setup6(struct interface *iface) } static void -reconnect_socket(struct uloop_timeout *timeout) +reconnect_socket4(struct uloop_timeout *timeout) { 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; - } + int yes = 1; - iface->fd.fd = usock(USOCK_UDP | USOCK_SERVER | USOCK_NONBLOCK | type, mcast_addr, "5353"); + iface->fd.fd = usock(USOCK_UDP | USOCK_SERVER | USOCK_NONBLOCK | USOCK_IPV4ONLY, + (iface->multicast) ? (iface->mcast_addr) : (iface->v4_addrs), "5353"); if (iface->fd.fd < 0) { - fprintf(stderr, "failed to add listener %s: %s\n", mcast_addr, strerror(errno)); + fprintf(stderr, "failed to add listener %s: %s\n", iface->mcast_addr, strerror(errno)); goto retry; } - if (!iface->v6 && interface_socket_setup4(iface)) { + if (setsockopt(iface->fd.fd, SOL_SOCKET, SO_BINDTODEVICE, iface->name, strlen(iface->name) < 0)) + fprintf(stderr, "ioctl failed: SO_BINDTODEVICE\n"); + + if (setsockopt(iface->fd.fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0) + fprintf(stderr, "ioctl failed: SO_REUSEADDR\n"); + + if (setsockopt(iface->fd.fd, IPPROTO_IP, IP_RECVTTL, &yes, sizeof(yes)) < 0) + fprintf(stderr, "ioctl failed: IP_RECVTTL\n"); + + if (setsockopt(iface->fd.fd, IPPROTO_IP, IP_PKTINFO, &yes, sizeof(yes)) < 0) + fprintf(stderr, "ioctl failed: IP_PKTINFO\n"); + + if (iface->multicast && interface_mcast_setup4(iface)) { iface->fd.fd = -1; goto retry; } - if (iface->v6 && interface_socket_setup6(iface)) { + uloop_fd_add(&iface->fd, ULOOP_READ); + if (iface->multicast) { + dns_send_question(iface, "_services._dns-sd._udp.local", TYPE_PTR, 0); + announce_init(iface); + } + + return; + +retry: + uloop_timeout_set(timeout, 1000); +} + +static void +reconnect_socket6(struct uloop_timeout *timeout) +{ + struct interface *iface = container_of(timeout, struct interface, reconnect); + char mcast_addr[128]; + int ttl = 255; + int yes = 1; + + snprintf(mcast_addr, sizeof(mcast_addr), "%s%%%s", (iface->multicast) ? (iface->mcast_addr) : (iface->v6_addrs), iface->name); + iface->fd.fd = usock(USOCK_UDP | USOCK_SERVER | USOCK_NONBLOCK | USOCK_IPV6ONLY, mcast_addr, "5353"); + if (iface->fd.fd < 0) { + fprintf(stderr, "failed to add listener %s: %s\n", mcast_addr, strerror(errno)); + goto retry; + } + + if (setsockopt(iface->fd.fd, SOL_SOCKET, SO_BINDTODEVICE, iface->name, strlen(iface->name) < 0)) + fprintf(stderr, "ioctl failed: SO_BINDTODEVICE\n"); + + if (setsockopt(iface->fd.fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &ttl, sizeof(ttl)) < 0) + fprintf(stderr, "ioctl failed: IPV6_UNICAST_HOPS\n"); + + if (setsockopt(iface->fd.fd, IPPROTO_IPV6, IPV6_RECVPKTINFO, &yes, sizeof(yes)) < 0) + fprintf(stderr, "ioctl failed: IPV6_RECVPKTINFO\n"); + + if (setsockopt(iface->fd.fd, IPPROTO_IPV6, IPV6_RECVHOPLIMIT, &yes, sizeof(yes)) < 0) + fprintf(stderr, "ioctl failed: IPV6_RECVHOPLIMIT\n"); + + if (setsockopt(iface->fd.fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0) + fprintf(stderr, "ioctl failed: SO_REUSEADDR\n"); + + if (iface->multicast && 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); - announce_init(iface); + + if (iface->multicast) { + dns_send_question(iface, "_services._dns-sd._udp.local", TYPE_PTR, 0); + announce_init(iface); + } + return; retry: @@ -295,8 +478,13 @@ retry: static void interface_start(struct interface *iface) { - iface->fd.cb = read_socket; - iface->reconnect.cb = reconnect_socket; + if (iface->v6) { + iface->fd.cb = read_socket6; + iface->reconnect.cb = reconnect_socket6; + } else { + iface->fd.cb = read_socket4; + iface->reconnect.cb = reconnect_socket4; + } uloop_timeout_set(&iface->reconnect, 100); } @@ -317,56 +505,7 @@ iface_update_cb(struct vlist_tree *tree, struct vlist_node *node_new, } } -static int -get_iface_ipv4(struct interface *iface) -{ - struct sockaddr_in *sin; - struct ifreq ir; - int sock, ret = -1; - - sock = socket(AF_INET, SOCK_DGRAM, 0); - if (sock < 0) - return -1; - - memset(&ir, 0, sizeof(struct ifreq)); - strncpy(ir.ifr_name, iface->name, sizeof(ir.ifr_name)); - - ret = ioctl(sock, SIOCGIFADDR, &ir); - if (ret < 0) - goto out; - - 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; -} - -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; - } - close(sock); - return ret; -} - -static int _interface_add(const char *name, int v6) +static struct interface* _interface_add(const char *name, int multicast, int v6) { struct interface *iface; char *name_buf; @@ -374,36 +513,127 @@ static int _interface_add(const char *name, int v6) iface = calloc_a(sizeof(*iface), &name_buf, strlen(name) + 1, - &id_buf, strlen(name) + 3); + &id_buf, strlen(name) + 5); - sprintf(id_buf, "%d_%s", v6, name); + sprintf(id_buf, "%d_%d_%s", multicast, v6, name); iface->name = strcpy(name_buf, name); iface->id = id_buf; iface->ifindex = if_nametoindex(name); iface->fd.fd = -1; + iface->multicast = multicast; + iface->v6 = v6; + if (v6) + iface->mcast_addr = MCAST_ADDR6; + else + iface->mcast_addr = MCAST_ADDR; if (iface->ifindex <= 0) goto error; - if (!v6 && get_iface_ipv4(iface)) - goto error; - - if (v6 && get_iface_ipv6(iface)) - goto error; - vlist_add(&interfaces, &iface->node, iface->id); - return 0; + return iface; error: free(iface); - return -1; + return NULL; } int interface_add(const char *name) { - int v4 = _interface_add(name, 0); - int v6 = _interface_add(name, 1); - return v4 && v6; + struct interface *v4 = NULL, *v6 = NULL, *unicast; + struct ifaddrs *ifap, *ifa; + + 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 && !v4) { + struct sockaddr_in *sa; + + if (cfg_proto && (cfg_proto != 4)) + continue; + + unicast = _interface_add(name, 0, 0); + if (!unicast) + continue; + v4 = _interface_add(name, 1, 0); + if (!v4) + continue; + + sa = (struct sockaddr_in *) ifa->ifa_addr; + memcpy(&v4->v4_addr, &sa->sin_addr, sizeof(v4->v4_addr)); + memcpy(&unicast->v4_addr, &sa->sin_addr, sizeof(unicast->v4_addr)); + + inet_ntop(AF_INET, &sa->sin_addr, v4->v4_addrs, sizeof(v4->v4_addrs)); + inet_ntop(AF_INET, &sa->sin_addr, unicast->v4_addrs, sizeof(unicast->v4_addrs)); + + sa = (struct sockaddr_in *) ifa->ifa_netmask; + memcpy(&unicast->v4_netmask, &sa->sin_addr, sizeof(unicast->v4_netmask)); + memcpy(&v4->v4_netmask, &sa->sin_addr, sizeof(v4->v4_netmask)); + + v4->peer = unicast; + unicast->peer = v4; + } + + if (ifa->ifa_addr->sa_family == AF_INET6 && !v6) { + uint8_t ll_prefix[] = {0xfe, 0x80 }; + struct sockaddr_in6 *sa6; + + if (cfg_proto && (cfg_proto != 6)) + continue; + + sa6 = (struct sockaddr_in6 *) ifa->ifa_addr; + if (memcmp(&sa6->sin6_addr, &ll_prefix, 2)) + continue; + + unicast = _interface_add(name, 0, 1); + if (!unicast) + continue; + v6 = _interface_add(name, 1, 1); + if (!v6) + continue; + + memcpy(&v6->v6_addr, &sa6->sin6_addr, sizeof(v6->v6_addr)); + memcpy(&unicast->v6_addr, &sa6->sin6_addr, sizeof(unicast->v6_addr)); + + inet_ntop(AF_INET6, &sa6->sin6_addr, v6->v6_addrs, sizeof(v6->v6_addrs)); + inet_ntop(AF_INET6, &sa6->sin6_addr, unicast->v6_addrs, sizeof(unicast->v6_addrs)); + + sa6 = (struct sockaddr_in6 *) ifa->ifa_netmask; + memcpy(&v6->v6_netmask, &sa6->sin6_addr, sizeof(v6->v6_netmask)); + memcpy(&unicast->v6_netmask, &sa6->sin6_addr, sizeof(unicast->v6_netmask)); + + v6->peer = unicast; + unicast->peer = v6; + } + } + + freeifaddrs(ifap); + + return !v4 && !v6; +} + +void interface_shutdown(void) +{ + struct interface *iface; + + vlist_for_each_element(&interfaces, iface, node) + if (iface->fd.fd > 0 && iface->multicast) { + service_announce(iface, 0); + dns_reply_a(iface, 0); + } + vlist_for_each_element(&interfaces, iface, node) + interface_close(iface); +} + +struct interface* +interface_get(const char *name, int v6, int multicast) +{ + char id_buf[32]; + snprintf(id_buf, sizeof(id_buf), "%d_%d_%s", multicast, v6, name); + struct interface *iface = vlist_find(&interfaces, id_buf, iface, node); + return iface; } VLIST_TREE(interfaces, avl_strcmp, iface_update_cb, false, false);