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 reconnect_socket(struct uloop_timeout
*timeout
)
116 struct interface
*iface
= container_of(timeout
, struct interface
, reconnect
);
118 iface
->fd
.fd
= usock(USOCK_UDP
| USOCK_SERVER
| USOCK_NONBLOCK
, MCAST_ADDR
, "5353");
119 if (iface
->fd
.fd
< 0) {
120 fprintf(stderr
, "failed to add listener: %s\n", strerror(errno
));
124 if (interface_socket_setup(iface
)) {
129 uloop_fd_add(&iface
->fd
, ULOOP_READ
);
130 dns_send_question(iface
, "_services._dns-sd._udp.local", TYPE_PTR
);
131 announce_init(iface
);
135 uloop_timeout_set(timeout
, 1000);
139 static void interface_start(struct interface
*iface
)
141 iface
->fd
.cb
= read_socket
;
142 iface
->reconnect
.cb
= reconnect_socket
;
143 uloop_timeout_set(&iface
->reconnect
, 100);
147 iface_update_cb(struct vlist_tree
*tree
, struct vlist_node
*node_new
,
148 struct vlist_node
*node_old
)
150 struct interface
*iface
;
153 iface
= container_of(node_old
, struct interface
, node
);
154 interface_free(iface
);
158 iface
= container_of(node_new
, struct interface
, node
);
159 interface_start(iface
);
163 int interface_socket_setup(struct interface
*iface
)
165 struct ip_mreqn mreq
;
169 struct sockaddr_in sa
= { 0 };
171 int fd
= iface
->fd
.fd
;
173 inet_aton(iface
->ip
, &in
);
175 sa
.sin_family
= AF_INET
;
176 sa
.sin_port
= htons(MCAST_PORT
);
177 inet_pton(AF_INET
, MCAST_ADDR
, &sa
.sin_addr
);
179 memset(&mreq
, 0, sizeof(mreq
));
180 mreq
.imr_address
.s_addr
= in
.s_addr
;
181 mreq
.imr_multiaddr
= sa
.sin_addr
;
182 mreq
.imr_ifindex
= iface
->ifindex
;
184 if (setsockopt(fd
, IPPROTO_IP
, IP_MULTICAST_TTL
, &ttl
, sizeof(ttl
)) < 0)
185 fprintf(stderr
, "ioctl failed: IP_MULTICAST_TTL\n");
187 if (setsockopt(fd
, SOL_SOCKET
, SO_REUSEADDR
, &yes
, sizeof(yes
)) < 0)
188 fprintf(stderr
, "ioctl failed: SO_REUSEADDR\n");
190 /* Some network drivers have issues with dropping membership of
191 * mcast groups when the iface is down, but don't allow rejoining
192 * when it comes back up. This is an ugly workaround
193 * -- this was copied from avahi --
195 setsockopt(fd
, IPPROTO_IP
, IP_DROP_MEMBERSHIP
, &mreq
, sizeof(mreq
));
197 if (setsockopt(fd
, IPPROTO_IP
, IP_ADD_MEMBERSHIP
, &mreq
, sizeof(mreq
)) < 0) {
198 fprintf(stderr
, "failed to join multicast group: %s\n", strerror(errno
));
204 if (setsockopt(fd
, IPPROTO_IP
, IP_RECVTTL
, &yes
, sizeof(yes
)) < 0)
205 fprintf(stderr
, "ioctl failed: IP_RECVTTL\n");
207 if (setsockopt(fd
, IPPROTO_IP
, IP_PKTINFO
, &yes
, sizeof(yes
)) < 0)
208 fprintf(stderr
, "ioctl failed: IP_PKTINFO\n");
210 if (setsockopt(fd
, IPPROTO_IP
, IP_MULTICAST_LOOP
, &no
, sizeof(no
)) < 0)
211 fprintf(stderr
, "ioctl failed: IP_MULTICAST_LOOP\n");
217 get_iface_ipv4(const char *ifname
)
219 static char buffer
[INET_ADDRSTRLEN
];
224 sock
= socket(AF_INET
, SOCK_DGRAM
, 0);
228 memset(&ir
, 0, sizeof(struct ifreq
));
229 strncpy(ir
.ifr_name
, ifname
, sizeof(ir
.ifr_name
));
231 if (ioctl(sock
, SIOCGIFADDR
, &ir
) < 0)
234 ret
= inet_ntop(AF_INET
, &((struct sockaddr_in
*) &ir
.ifr_addr
)->sin_addr
, buffer
, sizeof(buffer
));
240 int interface_add(const char *name
)
242 struct interface
*iface
;
244 char *name_buf
, *ip_buf
;
246 ip_str
= get_iface_ipv4(name
);
250 iface
= calloc_a(sizeof(*iface
),
251 &name_buf
, strlen(name
) + 1,
252 &ip_buf
, strlen(ip_str
) + 1);
254 iface
->name
= strcpy(name_buf
, name
);
255 iface
->ip
= strcpy(ip_buf
, ip_str
);
256 iface
->ifindex
= if_nametoindex(name
);
259 if (iface
->ifindex
<= 0)
262 vlist_add(&interfaces
, &iface
->node
, name
);
270 VLIST_TREE(interfaces
, avl_strcmp
, iface_update_cb
, false, false);