+#define __APPLE_USE_RFC_3542
#include "system.h"
#if defined(__APPLE__) || defined(__unix) && !defined(__linux)
+#include <net/if.h>
#include <net/route.h>
+#include <netinet/in.h>
#elif defined(__linux)
#include <asm/types.h>
#include <net/if.h>
#define MESHLINK_MDNS_NAME_KEY "name"
#define MESHLINK_MDNS_FINGERPRINT_KEY "fingerprint"
+#ifndef MSG_NOSIGNAL
+#define MSG_NOSIGNAL 0
+#endif
+
+#ifndef IPV6_ADD_MEMBERSHIP
+#define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP
+#endif
+
+#ifndef IPV6_DROP_MEMBERSHIP
+#define IPV6_DROP_MEMBERSHIP IPV6_LEAVE_GROUP
+#endif
+
static const sockaddr_t mdns_address_ipv4 = {
.in.sin_family = AF_INET,
.in.sin_addr.s_addr = 0xfb0000e0,
// Send the packet
ssize_t result = sendmsg(fd, &msg, MSG_DONTWAIT | MSG_NOSIGNAL);
#else
+ (void)index;
+ (void)src;
+
// Send the packet
ssize_t result = sendto(fd, data, len, MSG_DONTWAIT | MSG_NOSIGNAL, &dest->sa, SALEN(dest->sa));
#endif
// Send the packet
ssize_t result = sendmsg(fd, &msg, MSG_DONTWAIT | MSG_NOSIGNAL);
#else
+ (void)index;
+ (void)src;
+
// Send the packet
ssize_t result = sendto(fd, data, len, MSG_DONTWAIT | MSG_NOSIGNAL, &dest->sa, SALEN(dest->sa));
#endif
static void send_mdns_packet(meshlink_handle_t *mesh, const discovery_address_t *addr) {
// Configure the socket to send the packet to the right interface
int fd;
- uint8_t data[1024];
+ uint8_t request[1024], response[1024];
char *fingerprint = meshlink_get_fingerprint(mesh, (meshlink_node_t *)mesh->self);
const char *keys[] = {MESHLINK_MDNS_NAME_KEY, MESHLINK_MDNS_FINGERPRINT_KEY};
const char *values[] = {mesh->name, fingerprint};
- size_t size = prepare_packet(data, sizeof data, fingerprint, mesh->appname, "tcp", atoi(mesh->myport), 2, keys, values, false);
+ size_t request_size = prepare_request(request, sizeof request, mesh->appname, "tcp");
+ size_t response_size = prepare_response(response, sizeof response, fingerprint, mesh->appname, "tcp", atoi(mesh->myport), 2, keys, values);
free(fingerprint);
switch(addr->address.sa.sa_family) {
#endif
- send_mdns_packet_ipv4(mesh, fd, addr->index, &addr->address, &mdns_address_ipv4, data, size);
+ send_mdns_packet_ipv4(mesh, fd, addr->index, &addr->address, &mdns_address_ipv4, request, request_size);
+ send_mdns_packet_ipv4(mesh, fd, addr->index, &addr->address, &mdns_address_ipv4, response, response_size);
break;
case AF_INET6:
#endif
- send_mdns_packet_ipv6(mesh, fd, addr->index, &addr->address, &mdns_address_ipv6, data, size);
+ send_mdns_packet_ipv6(mesh, fd, addr->index, &addr->address, &mdns_address_ipv6, request, request_size);
+ send_mdns_packet_ipv6(mesh, fd, addr->index, &addr->address, &mdns_address_ipv6, response, response_size);
break;
default:
uint16_t port = 0;
const char *keys[2] = {MESHLINK_MDNS_NAME_KEY, MESHLINK_MDNS_FINGERPRINT_KEY};
char *values[2] = {NULL, NULL};
- bool response;
- if(parse_packet(buf, len, &name, mesh->appname, "tcp", &port, 2, keys, values, &response)) {
+ if(parse_response(buf, len, &name, mesh->appname, "tcp", &port, 2, keys, values)) {
node_t *n = (node_t *)meshlink_get_node(mesh, values[0]);
if(n) {
logger(mesh, MESHLINK_INFO, "Node %s discovered on the local network.\n", n->name);
}
- if(!response && n != mesh->self) {
- // Send a unicast response back
- char *fingerprint = meshlink_get_fingerprint(mesh, (meshlink_node_t *)mesh->self);
- const char *response_values[] = {mesh->name, fingerprint};
- size_t size = prepare_packet(buf, sizeof(buf), fingerprint, mesh->appname, "tcp", atoi(mesh->myport), 2, keys, response_values, true);
- sendto(io->fd, buf, size, MSG_DONTWAIT | MSG_NOSIGNAL, &sa.sa, sl);
- free(fingerprint);
- }
-
switch(sa.sa.sa_family) {
case AF_INET:
sa.in.sin_port = htons(port);
}
}
}
+ } else if(parse_request(buf, len, mesh->appname, "tcp")) {
+ // Send a unicast response back
+ char *fingerprint = meshlink_get_fingerprint(mesh, (meshlink_node_t *)mesh->self);
+ const char *response_values[] = {mesh->name, fingerprint};
+ size_t size = prepare_response(buf, sizeof(buf), fingerprint, mesh->appname, "tcp", atoi(mesh->myport), 2, keys, response_values);
+ sendto(io->fd, buf, size, MSG_DONTWAIT | MSG_NOSIGNAL, &sa.sa, sl);
+ free(fingerprint);
}
free(name);
if(rta->rta_len == 8) {
addr.address.sa.sa_family = AF_INET;
memcpy(&addr.address.in.sin_addr, ptr + 4, 4);
- addr.address.in.sin_port = 5353;
+ addr.address.in.sin_port = ntohs(5353);
} else if(rta->rta_len == 20) {
addr.address.sa.sa_family = AF_INET6;
memcpy(&addr.address.in6.sin6_addr, ptr + 4, 16);
- addr.address.in6.sin6_port = 5353;
+ addr.address.in6.sin6_port = ntohs(5353);
addr.address.in6.sin6_scope_id = ifa->ifa_index;
} else {
addr.address.sa.sa_family = AF_UNKNOWN;
}
}
#elif defined(RTM_NEWADDR)
+static void pfroute_parse_iface(meshlink_handle_t *mesh, const struct rt_msghdr *rtm) {
+ const struct if_msghdr *ifm = (const struct if_msghdr *)rtm;
+
+ if(ifm->ifm_flags & IFF_UP && ifm->ifm_flags & IFF_MULTICAST && !(ifm->ifm_flags & IFF_LOOPBACK)) {
+ iface_up(mesh, ifm->ifm_index);
+ } else {
+ iface_down(mesh, ifm->ifm_index);
+ }
+}
+
+static void pfroute_parse_addr(meshlink_handle_t *mesh, const struct rt_msghdr *rtm) {
+ const struct ifa_msghdr *ifam = (const struct ifa_msghdr *)rtm;
+ const char *p = (const char *)(ifam + 1);
+
+ for(unsigned int i = 1; i; i <<= 1) {
+ if(!(ifam->ifam_addrs & i)) {
+ continue;
+ }
+
+ const sockaddr_t *sa = (const sockaddr_t *)p;
+
+ if(i == RTA_IFA) {
+ discovery_address_t addr = {
+ .index = ifam->ifam_index,
+ };
+
+ if(sa->sa.sa_family == AF_INET) {
+ addr.address.in = sa->in;
+ addr.address.in.sin_port = ntohs(5353);
+ } else if(sa->sa.sa_family == AF_INET6) {
+ addr.address.in6 = sa->in6;
+ addr.address.in6.sin6_port = ntohs(5353);
+ } else {
+ addr.address.sa.sa_family = AF_UNKNOWN;
+ }
+
+ if(addr.address.sa.sa_family != AF_UNKNOWN) {
+ if(ifam->ifam_type == RTM_NEWADDR) {
+ addr_add(mesh, &addr);
+ } else {
+ addr_del(mesh, &addr);
+ }
+ }
+
+ break;
+ }
+
+ size_t len = (sa->sa.sa_len + 3) & ~3;
+ p += len;
+ }
+}
+
static void pfroute_io_handler(event_loop_t *loop, void *data, int flags) {
(void)flags;
(void)data;
- static time_t prev_update;
meshlink_handle_t *mesh = loop->data;
struct {
switch(msg.rtm.rtm_type) {
case RTM_IFINFO:
+ pfroute_parse_iface(mesh, &msg.rtm);
+ break;
+
case RTM_NEWADDR:
case RTM_DELADDR:
- if(loop->now.tv_sec > prev_update + 5) {
- prev_update = loop->now.tv_sec;
- handle_network_change(mesh, 1);
- }
-
+ pfroute_parse_addr(mesh, &msg.rtm);
break;
default:
// Set up multicast sockets for mDNS
static const int one = 1;
static const int ttl = 255;
+ static const uint8_t one8 = 1;
+ static const uint8_t ttl8 = 255;
int fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+
+ if(fd == -1) {
+ logger(mesh, MESHLINK_ERROR, "Error creating IPv4 socket: %s", strerror(errno));
+ }
+
sockaddr_t sa4 = {
.in.sin_family = AF_INET,
.in.sin_port = ntohs(5353),
};
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
- bind(fd, &sa4.sa, SALEN(sa4.sa));
- setsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP, &one, sizeof(one));
- setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl));
- io_add(&mesh->loop, &mesh->discovery.sockets[0], mdns_io_handler, &mesh->discovery.sockets[0], fd, IO_READ);
+ setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &one, sizeof(one));
+ setsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP, &one8, sizeof(one8));
+ setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl8, sizeof(ttl8));
+
+ if(bind(fd, &sa4.sa, SALEN(sa4.sa)) == -1) {
+ logger(mesh, MESHLINK_ERROR, "Error binding to IPv4 multicast socket: %s", strerror(errno));
+ } else {
+ io_add(&mesh->loop, &mesh->discovery.sockets[0], mdns_io_handler, &mesh->discovery.sockets[0], fd, IO_READ);
+ }
sockaddr_t sa6 = {
.in6.sin6_family = AF_INET6,
.in6.sin6_port = ntohs(5353),
};
fd = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
+
+ if(fd == -1) {
+ logger(mesh, MESHLINK_ERROR, "Error creating IPv6 socket: %s", strerror(errno));
+ }
+
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
+ setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &one, sizeof(one));
setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &one, sizeof(one));
- bind(fd, &sa6.sa, SALEN(sa6.sa));
setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, &one, sizeof(one));
setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &ttl, sizeof(ttl));
- io_add(&mesh->loop, &mesh->discovery.sockets[1], mdns_io_handler, &mesh->discovery.sockets[1], fd, IO_READ);
+ setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &ttl, sizeof(ttl));
+
+ if(bind(fd, &sa6.sa, SALEN(sa6.sa)) == -1) {
+ logger(mesh, MESHLINK_ERROR, "Error binding to IPv4 multicast socket: %s", strerror(errno));
+ } else {
+ io_add(&mesh->loop, &mesh->discovery.sockets[1], mdns_io_handler, &mesh->discovery.sockets[1], fd, IO_READ);
+ }
#if defined(__linux)
int sock = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE);
logger(mesh, MESHLINK_WARNING, "Could not open PF_ROUTE socket: %s", strerror(errno));
}
+ struct ifaddrs *ifa = NULL;
+
+ if(getifaddrs(&ifa) == -1) {
+ logger(mesh, MESHLINK_ERROR, "Could not get list of interface addresses: %s", strerror(errno));
+ return true;
+ }
+
+ for(struct ifaddrs *ifap = ifa; ifap; ifap = ifap->ifa_next) {
+ if(!ifap->ifa_name) {
+ continue;
+ }
+
+ int index = if_nametoindex(ifap->ifa_name);
+
+ if(ifap->ifa_flags & IFF_UP && ifap->ifa_flags & IFF_MULTICAST && !(ifap->ifa_flags & IFF_LOOPBACK)) {
+ iface_up(mesh, index);
+ }
+
+ if(!ifap->ifa_addr) {
+ continue;
+ }
+
+ discovery_address_t addr = {
+ .index = index,
+ };
+
+ sockaddr_t *sa = (sockaddr_t *)ifap->ifa_addr;
+
+ if(sa->sa.sa_family == AF_INET) {
+ memcpy(&addr.address.in, &sa->in, sizeof(sa->in));
+ addr.address.in.sin_port = ntohs(5353);
+ } else if(sa->sa.sa_family == AF_INET6) {
+ memcpy(&addr.address.in6, &sa->in6, sizeof(sa->in6));
+ addr.address.in6.sin6_port = ntohs(5353);
+ } else {
+ addr.address.sa.sa_family = AF_UNKNOWN;
+ }
+
+ if(addr.address.sa.sa_family != AF_UNKNOWN) {
+ addr_add(mesh, &addr);
+ }
+ }
+
+ freeifaddrs(ifa);
+
#endif
return true;