01aeaa707734e91cd479caf9faebfbce773513ca
2 * Copyright (C) 2014 John Crispin <blogic@openwrt.org>
3 * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU Lesser General Public License version 2.1
7 * as published by the Free Software Foundation
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
15 #include <sys/socket.h>
16 #include <sys/ioctl.h>
17 #include <sys/types.h>
19 #include <sys/utsname.h>
21 #include <linux/sockios.h>
22 #include <arpa/inet.h>
30 #include <libubox/usock.h>
31 #include <libubox/uloop.h>
32 #include <libubox/avl-cmp.h>
33 #include <libubox/utils.h>
34 #include "interface.h"
40 interface_send_packet(struct interface
*iface
, struct iovec
*iov
, int iov_len
)
42 static size_t cmsg_data
[( CMSG_SPACE(sizeof(struct in_pktinfo
)) / sizeof(size_t)) + 1];
43 static struct sockaddr_in a
= {
44 .sin_family
= AF_INET
,
45 .sin_port
= htons(MCAST_PORT
),
47 static struct msghdr m
= {
48 .msg_name
= (struct sockaddr
*) &a
,
49 .msg_namelen
= sizeof(a
),
50 .msg_control
= cmsg_data
,
51 .msg_controllen
= CMSG_LEN(sizeof(struct in_pktinfo
)),
53 struct in_pktinfo
*pkti
;
55 int fd
= iface
->fd
.fd
;
58 m
.msg_iovlen
= iov_len
;
60 memset(cmsg_data
, 0, sizeof(cmsg_data
));
61 cmsg
= CMSG_FIRSTHDR(&m
);
62 cmsg
->cmsg_len
= m
.msg_controllen
;
63 cmsg
->cmsg_level
= IPPROTO_IP
;
64 cmsg
->cmsg_type
= IP_PKTINFO
;
66 pkti
= (struct in_pktinfo
*) CMSG_DATA(cmsg
);
67 pkti
->ipi_ifindex
= iface
->ifindex
;
69 a
.sin_addr
.s_addr
= inet_addr(MCAST_ADDR
);
71 return sendmsg(fd
, &m
, 0);
74 static void interface_close(struct interface
*iface
)
80 uloop_fd_delete(&iface
->fd
);
85 static void interface_free(struct interface
*iface
)
87 interface_close(iface
);
92 read_socket(struct uloop_fd
*u
, unsigned int events
)
94 struct interface
*iface
= container_of(u
, struct interface
, fd
);
95 static uint8_t buffer
[8 * 1024];
99 interface_close(iface
);
100 uloop_timeout_set(&iface
->reconnect
, 1000);
104 len
= read(u
->fd
, buffer
, sizeof(buffer
));
106 fprintf(stderr
, "read failed: %s\n", strerror(errno
));
110 dns_handle_packet(iface
, buffer
, len
);
114 interface_socket_setup(struct interface
*iface
)
116 struct ip_mreqn mreq
;
120 struct sockaddr_in sa
= { 0 };
122 int fd
= iface
->fd
.fd
;
124 inet_aton(iface
->ip
, &in
);
126 sa
.sin_family
= AF_INET
;
127 sa
.sin_port
= htons(MCAST_PORT
);
128 inet_pton(AF_INET
, MCAST_ADDR
, &sa
.sin_addr
);
130 memset(&mreq
, 0, sizeof(mreq
));
131 mreq
.imr_address
.s_addr
= in
.s_addr
;
132 mreq
.imr_multiaddr
= sa
.sin_addr
;
133 mreq
.imr_ifindex
= iface
->ifindex
;
135 if (setsockopt(fd
, IPPROTO_IP
, IP_MULTICAST_TTL
, &ttl
, sizeof(ttl
)) < 0)
136 fprintf(stderr
, "ioctl failed: IP_MULTICAST_TTL\n");
138 if (setsockopt(fd
, SOL_SOCKET
, SO_REUSEADDR
, &yes
, sizeof(yes
)) < 0)
139 fprintf(stderr
, "ioctl failed: SO_REUSEADDR\n");
141 /* Some network drivers have issues with dropping membership of
142 * mcast groups when the iface is down, but don't allow rejoining
143 * when it comes back up. This is an ugly workaround
144 * -- this was copied from avahi --
146 setsockopt(fd
, IPPROTO_IP
, IP_DROP_MEMBERSHIP
, &mreq
, sizeof(mreq
));
148 if (setsockopt(fd
, IPPROTO_IP
, IP_ADD_MEMBERSHIP
, &mreq
, sizeof(mreq
)) < 0) {
149 fprintf(stderr
, "failed to join multicast group: %s\n", strerror(errno
));
155 if (setsockopt(fd
, IPPROTO_IP
, IP_RECVTTL
, &yes
, sizeof(yes
)) < 0)
156 fprintf(stderr
, "ioctl failed: IP_RECVTTL\n");
158 if (setsockopt(fd
, IPPROTO_IP
, IP_PKTINFO
, &yes
, sizeof(yes
)) < 0)
159 fprintf(stderr
, "ioctl failed: IP_PKTINFO\n");
161 if (setsockopt(fd
, IPPROTO_IP
, IP_MULTICAST_LOOP
, &no
, sizeof(no
)) < 0)
162 fprintf(stderr
, "ioctl failed: IP_MULTICAST_LOOP\n");
168 reconnect_socket(struct uloop_timeout
*timeout
)
170 struct interface
*iface
= container_of(timeout
, struct interface
, reconnect
);
172 iface
->fd
.fd
= usock(USOCK_UDP
| USOCK_SERVER
| USOCK_NONBLOCK
, MCAST_ADDR
, "5353");
173 if (iface
->fd
.fd
< 0) {
174 fprintf(stderr
, "failed to add listener: %s\n", strerror(errno
));
178 if (interface_socket_setup(iface
)) {
183 uloop_fd_add(&iface
->fd
, ULOOP_READ
);
184 dns_send_question(iface
, "_services._dns-sd._udp.local", TYPE_PTR
);
185 announce_init(iface
);
189 uloop_timeout_set(timeout
, 1000);
193 static void interface_start(struct interface
*iface
)
195 iface
->fd
.cb
= read_socket
;
196 iface
->reconnect
.cb
= reconnect_socket
;
197 uloop_timeout_set(&iface
->reconnect
, 100);
201 iface_update_cb(struct vlist_tree
*tree
, struct vlist_node
*node_new
,
202 struct vlist_node
*node_old
)
204 struct interface
*iface
;
207 iface
= container_of(node_old
, struct interface
, node
);
208 interface_free(iface
);
212 iface
= container_of(node_new
, struct interface
, node
);
213 interface_start(iface
);
218 get_iface_ipv4(const char *ifname
)
220 static char buffer
[INET_ADDRSTRLEN
];
225 sock
= socket(AF_INET
, SOCK_DGRAM
, 0);
229 memset(&ir
, 0, sizeof(struct ifreq
));
230 strncpy(ir
.ifr_name
, ifname
, sizeof(ir
.ifr_name
));
232 if (ioctl(sock
, SIOCGIFADDR
, &ir
) < 0)
235 ret
= inet_ntop(AF_INET
, &((struct sockaddr_in
*) &ir
.ifr_addr
)->sin_addr
, buffer
, sizeof(buffer
));
241 int interface_add(const char *name
)
243 struct interface
*iface
;
245 char *name_buf
, *ip_buf
;
247 ip_str
= get_iface_ipv4(name
);
251 iface
= calloc_a(sizeof(*iface
),
252 &name_buf
, strlen(name
) + 1,
253 &ip_buf
, strlen(ip_str
) + 1);
255 iface
->name
= strcpy(name_buf
, name
);
256 iface
->ip
= strcpy(ip_buf
, ip_str
);
257 iface
->ifindex
= if_nametoindex(name
);
260 if (iface
->ifindex
<= 0)
263 vlist_add(&interfaces
, &iface
->node
, name
);
271 VLIST_TREE(interfaces
, avl_strcmp
, iface_update_cb
, false, false);