4 This file is part of avahi.
6 avahi is free software; you can redistribute it and/or modify it
7 under the terms of the GNU Lesser General Public License as
8 published by the Free Software Foundation; either version 2.1 of the
9 License, or (at your option) any later version.
11 avahi is distributed in the hope that it will be useful, but WITHOUT
12 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13 or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
14 Public License for more details.
16 You should have received a copy of the GNU Lesser General Public
17 License along with avahi; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
27 #include <sys/socket.h>
28 #include <asm/types.h>
29 #include <linux/netlink.h>
30 #include <linux/rtnetlink.h>
35 #include <avahi-common/malloc.h>
45 static void update_address_rr(AvahiInterfaceMonitor *m, AvahiInterfaceAddress *a, int remove_rrs) {
49 if (avahi_interface_address_relevant(a) &&
51 m->server->config.publish_addresses &&
52 (m->server->state == AVAHI_SERVER_RUNNING ||
53 m->server->state == AVAHI_SERVER_REGISTERING)) {
55 /* Fill the entry group */
57 a->entry_group = avahi_s_entry_group_new(m->server, avahi_host_rr_entry_group_callback, NULL);
59 if (!a->entry_group) /* OOM */
62 if (avahi_s_entry_group_is_empty(a->entry_group)) {
64 if (avahi_server_add_address(m->server, a->entry_group, a->interface->hardware->index, a->interface->protocol, 0, NULL, &a->address) < 0) {
65 avahi_log_warn(__FILE__": avahi_server_add_address() failed: %s", avahi_strerror(m->server->error));
66 avahi_s_entry_group_free(a->entry_group);
67 a->entry_group = NULL;
71 avahi_s_entry_group_commit(a->entry_group);
75 /* Clear the entry group */
77 if (a->entry_group && !avahi_s_entry_group_is_empty(a->entry_group)) {
79 if (avahi_s_entry_group_get_state(a->entry_group) == AVAHI_ENTRY_GROUP_REGISTERING)
80 avahi_server_decrease_host_rr_pending(m->server);
82 avahi_s_entry_group_reset(a->entry_group);
87 static void update_interface_rr(AvahiInterfaceMonitor *m, AvahiInterface *i, int remove_rrs) {
88 AvahiInterfaceAddress *a;
93 for (a = i->addresses; a; a = a->address_next)
94 update_address_rr(m, a, remove_rrs);
97 static void update_hw_interface_rr(AvahiInterfaceMonitor *m, AvahiHwInterface *hw, int remove_rrs) {
103 for (i = hw->interfaces; i; i = i->by_hardware_next)
104 update_interface_rr(m, i, remove_rrs);
107 m->server->config.publish_workstation &&
108 (m->server->state == AVAHI_SERVER_RUNNING ||
109 m->server->state == AVAHI_SERVER_REGISTERING)) {
111 if (!hw->entry_group)
112 hw->entry_group = avahi_s_entry_group_new(m->server, avahi_host_rr_entry_group_callback, NULL);
114 if (!hw->entry_group)
117 if (avahi_s_entry_group_is_empty(hw->entry_group)) {
121 if (!(t = avahi_format_mac_address(hw->mac_address, hw->mac_address_size)))
124 name = avahi_strdup_printf("%s [%s]", m->server->host_name, t);
130 if (avahi_server_add_service(m->server, hw->entry_group, hw->index, AVAHI_PROTO_UNSPEC, name, "_workstation._tcp", NULL, NULL, 9, NULL) < 0) {
131 avahi_log_warn(__FILE__": avahi_server_add_service() failed.");
132 avahi_s_entry_group_free(hw->entry_group);
133 hw->entry_group = NULL;
135 avahi_s_entry_group_commit(hw->entry_group);
142 if (hw->entry_group && !avahi_s_entry_group_is_empty(hw->entry_group)) {
144 if (avahi_s_entry_group_get_state(hw->entry_group) == AVAHI_ENTRY_GROUP_REGISTERING)
145 avahi_server_decrease_host_rr_pending(m->server);
147 avahi_s_entry_group_reset(hw->entry_group);
152 static void free_address(AvahiInterfaceMonitor *m, AvahiInterfaceAddress *a) {
155 assert(a->interface);
157 update_address_rr(m, a, 1);
158 AVAHI_LLIST_REMOVE(AvahiInterfaceAddress, address, a->interface->addresses, a);
161 avahi_s_entry_group_free(a->entry_group);
166 static void free_interface(AvahiInterfaceMonitor *m, AvahiInterface *i, int send_goodbye) {
170 avahi_goodbye_interface(m->server, i, send_goodbye);
171 avahi_response_scheduler_force(i->response_scheduler);
173 assert(!i->announcements);
175 update_interface_rr(m, i, 1);
178 free_address(m, i->addresses);
180 avahi_response_scheduler_free(i->response_scheduler);
181 avahi_query_scheduler_free(i->query_scheduler);
182 avahi_probe_scheduler_free(i->probe_scheduler);
183 avahi_cache_free(i->cache);
185 AVAHI_LLIST_REMOVE(AvahiInterface, interface, m->interfaces, i);
186 AVAHI_LLIST_REMOVE(AvahiInterface, by_hardware, i->hardware->interfaces, i);
191 static void free_hw_interface(AvahiInterfaceMonitor *m, AvahiHwInterface *hw, int send_goodbye) {
195 update_hw_interface_rr(m, hw, 1);
197 while (hw->interfaces)
198 free_interface(m, hw->interfaces, send_goodbye);
201 avahi_s_entry_group_free(hw->entry_group);
203 AVAHI_LLIST_REMOVE(AvahiHwInterface, hardware, m->hw_interfaces, hw);
204 avahi_hashmap_remove(m->hashmap, &hw->index);
206 avahi_free(hw->name);
210 static AvahiInterfaceAddress* get_address(AvahiInterfaceMonitor *m, AvahiInterface *i, const AvahiAddress *raddr) {
211 AvahiInterfaceAddress *ia;
217 for (ia = i->addresses; ia; ia = ia->address_next)
218 if (avahi_address_cmp(&ia->address, raddr) == 0)
224 static int netlink_list_items(AvahiNetlink *nl, uint16_t type, unsigned *ret_seq) {
226 struct rtgenmsg *gen;
229 memset(&req, 0, sizeof(req));
230 n = (struct nlmsghdr*) req;
231 n->nlmsg_len = NLMSG_LENGTH(sizeof(struct rtgenmsg));
232 n->nlmsg_type = type;
233 n->nlmsg_flags = NLM_F_ROOT/*|NLM_F_MATCH*/|NLM_F_REQUEST;
237 memset(gen, 0, sizeof(struct rtgenmsg));
238 gen->rtgen_family = AF_UNSPEC;
240 return avahi_netlink_send(nl, n, ret_seq);
243 static void new_interface(AvahiInterfaceMonitor *m, AvahiHwInterface *hw, AvahiProtocol protocol) {
248 assert(protocol != AVAHI_PROTO_UNSPEC);
250 if (!(i = avahi_new(AvahiInterface, 1)))
255 i->protocol = protocol;
258 AVAHI_LLIST_HEAD_INIT(AvahiInterfaceAddress, i->addresses);
259 AVAHI_LLIST_HEAD_INIT(AvahiAnnouncement, i->announcements);
261 i->cache = avahi_cache_new(m->server, i);
262 i->response_scheduler = avahi_response_scheduler_new(i);
263 i->query_scheduler = avahi_query_scheduler_new(i);
264 i->probe_scheduler = avahi_probe_scheduler_new(i);
266 if (!i->cache || !i->response_scheduler || !i->query_scheduler || !i->probe_scheduler)
269 AVAHI_LLIST_PREPEND(AvahiInterface, by_hardware, hw->interfaces, i);
270 AVAHI_LLIST_PREPEND(AvahiInterface, interface, m->interfaces, i);
277 avahi_cache_free(i->cache);
278 if (i->response_scheduler)
279 avahi_response_scheduler_free(i->response_scheduler);
280 if (i->query_scheduler)
281 avahi_query_scheduler_free(i->query_scheduler);
282 if (i->probe_scheduler)
283 avahi_probe_scheduler_free(i->probe_scheduler);
288 static void check_interface_relevant(AvahiInterfaceMonitor *m, AvahiInterface *i) {
294 b = avahi_interface_relevant(i);
296 if (b && !i->announcing) {
297 avahi_log_debug("New relevant interface %s.%i (#%i)", i->hardware->name, i->protocol, i->hardware->index);
299 if (i->protocol == AVAHI_PROTO_INET)
300 avahi_mdns_mcast_join_ipv4(m->server->fd_ipv4, i->hardware->index);
301 if (i->protocol == AVAHI_PROTO_INET6)
302 avahi_mdns_mcast_join_ipv6(m->server->fd_ipv6, i->hardware->index);
305 avahi_announce_interface(m->server, i);
306 avahi_browser_new_interface(m->server, i);
307 } else if (!b && i->announcing) {
308 avahi_log_debug("Interface %s.%i no longer relevant", i->hardware->name, i->protocol);
310 if (i->protocol == AVAHI_PROTO_INET)
311 avahi_mdns_mcast_leave_ipv4(m->server->fd_ipv4, i->hardware->index);
312 if (i->protocol == AVAHI_PROTO_INET6)
313 avahi_mdns_mcast_leave_ipv6(m->server->fd_ipv6, i->hardware->index);
315 avahi_goodbye_interface(m->server, i, 0);
316 avahi_response_scheduler_clear(i->response_scheduler);
317 avahi_query_scheduler_clear(i->query_scheduler);
318 avahi_probe_scheduler_clear(i->probe_scheduler);
319 avahi_cache_flush(i->cache);
325 static void check_hw_interface_relevant(AvahiInterfaceMonitor *m, AvahiHwInterface *hw) {
331 for (i = hw->interfaces; i; i = i->by_hardware_next)
332 check_interface_relevant(m, i);
335 static void netlink_callback(AvahiNetlink *nl, struct nlmsghdr *n, void* userdata) {
336 AvahiInterfaceMonitor *m = userdata;
340 assert(m->netlink == nl);
342 if (n->nlmsg_type == RTM_NEWLINK) {
343 struct ifinfomsg *ifinfomsg = NLMSG_DATA(n);
344 AvahiHwInterface *hw;
345 struct rtattr *a = NULL;
348 if (ifinfomsg->ifi_family != AF_UNSPEC)
351 if (!(hw = avahi_hashmap_lookup(m->hashmap, &ifinfomsg->ifi_index))) {
353 if (!(hw = avahi_new(AvahiHwInterface, 1)))
360 hw->index = (AvahiIfIndex) ifinfomsg->ifi_index;
361 hw->mac_address_size = 0;
362 hw->entry_group = NULL;
364 AVAHI_LLIST_HEAD_INIT(AvahiInterface, hw->interfaces);
365 AVAHI_LLIST_PREPEND(AvahiHwInterface, hardware, m->hw_interfaces, hw);
367 avahi_hashmap_insert(m->hashmap, &hw->index, hw);
369 if (m->server->fd_ipv4 >= 0)
370 new_interface(m, hw, AVAHI_PROTO_INET);
371 if (m->server->fd_ipv6 >= 0)
372 new_interface(m, hw, AVAHI_PROTO_INET6);
375 hw->flags = ifinfomsg->ifi_flags;
377 l = NLMSG_PAYLOAD(n, sizeof(struct ifinfomsg));
378 a = IFLA_RTA(ifinfomsg);
380 while (RTA_OK(a, l)) {
381 switch(a->rta_type) {
383 avahi_free(hw->name);
384 hw->name = avahi_strndup(RTA_DATA(a), RTA_PAYLOAD(a));
388 assert(RTA_PAYLOAD(a) == sizeof(unsigned int));
389 hw->mtu = *((unsigned int*) RTA_DATA(a));
393 hw->mac_address_size = RTA_PAYLOAD(a);
394 if (hw->mac_address_size > AVAHI_MAX_MAC_ADDRESS)
395 hw->mac_address_size = AVAHI_MAX_MAC_ADDRESS;
397 memcpy(hw->mac_address, RTA_DATA(a), hw->mac_address_size);
408 update_hw_interface_rr(m, hw, 0);
409 check_hw_interface_relevant(m, hw);
411 } else if (n->nlmsg_type == RTM_DELLINK) {
412 struct ifinfomsg *ifinfomsg = NLMSG_DATA(n);
413 AvahiHwInterface *hw;
415 if (ifinfomsg->ifi_family != AF_UNSPEC)
418 if (!(hw = avahi_interface_monitor_get_hw_interface(m, (AvahiIfIndex) ifinfomsg->ifi_index)))
421 update_hw_interface_rr(m, hw, 1);
422 free_hw_interface(m, hw, 0);
424 } else if (n->nlmsg_type == RTM_NEWADDR || n->nlmsg_type == RTM_DELADDR) {
426 struct ifaddrmsg *ifaddrmsg = NLMSG_DATA(n);
428 struct rtattr *a = NULL;
433 if (ifaddrmsg->ifa_family != AVAHI_PROTO_INET && ifaddrmsg->ifa_family != AVAHI_PROTO_INET6)
436 if (!(i = (AvahiInterface*) avahi_interface_monitor_get_interface(m, (AvahiIfIndex) ifaddrmsg->ifa_index, (AvahiProtocol) ifaddrmsg->ifa_family)))
439 raddr.family = (AvahiProtocol) ifaddrmsg->ifa_family;
441 l = NLMSG_PAYLOAD(n, sizeof(struct ifaddrmsg));
442 a = IFA_RTA(ifaddrmsg);
444 while (RTA_OK(a, l)) {
446 switch(a->rta_type) {
448 if ((raddr.family == AVAHI_PROTO_INET6 && RTA_PAYLOAD(a) != 16) ||
449 (raddr.family == AVAHI_PROTO_INET && RTA_PAYLOAD(a) != 4))
452 memcpy(raddr.data.data, RTA_DATA(a), RTA_PAYLOAD(a));
467 if (n->nlmsg_type == RTM_NEWADDR) {
468 AvahiInterfaceAddress *addr;
470 if (!(addr = get_address(m, i, &raddr))) {
471 if (!(addr = avahi_new(AvahiInterfaceAddress, 1)))
475 addr->address = raddr;
477 addr->entry_group = NULL;
479 AVAHI_LLIST_PREPEND(AvahiInterfaceAddress, address, i->addresses, addr);
482 addr->flags = ifaddrmsg->ifa_flags;
483 addr->scope = ifaddrmsg->ifa_scope;
484 addr->prefix_len = ifaddrmsg->ifa_prefixlen;
486 update_address_rr(m, addr, 0);
488 AvahiInterfaceAddress *addr;
490 if (!(addr = get_address(m, i, &raddr)))
493 update_address_rr(m, addr, 1);
494 free_address(m, addr);
497 check_interface_relevant(m, i);
499 } else if (n->nlmsg_type == NLMSG_DONE) {
501 if (m->list == LIST_IFACE) {
504 if (netlink_list_items(m->netlink, RTM_GETADDR, &m->query_addr_seq) < 0)
505 avahi_log_warn("NETLINK: Failed to list addrs: %s", strerror(errno));
510 avahi_log_debug("Network interface enumeration completed");
513 } else if (n->nlmsg_type == NLMSG_ERROR && (n->nlmsg_seq == m->query_link_seq || n->nlmsg_seq == m->query_addr_seq)) {
514 struct nlmsgerr *e = NLMSG_DATA (n);
517 avahi_log_warn("NETLINK: Failed to browse: %s", strerror(-e->error));
521 AvahiInterfaceMonitor *avahi_interface_monitor_new(AvahiServer *s) {
522 AvahiInterfaceMonitor *m = NULL;
524 if (!(m = avahi_new0(AvahiInterfaceMonitor, 1)))
525 return NULL; /* OOM */
528 if (!(m->netlink = avahi_netlink_new(s->poll_api, RTMGRP_LINK|RTMGRP_IPV4_IFADDR|RTMGRP_IPV6_IFADDR, netlink_callback, m)))
531 m->hashmap = avahi_hashmap_new(avahi_int_hash, avahi_int_equal, NULL, NULL);
533 AVAHI_LLIST_HEAD_INIT(AvahiInterface, m->interfaces);
534 AVAHI_LLIST_HEAD_INIT(AvahiHwInterface, m->hw_interfaces);
536 if (netlink_list_items(m->netlink, RTM_GETLINK, &m->query_link_seq) < 0)
539 m->list = LIST_IFACE;
544 avahi_interface_monitor_free(m);
548 void avahi_interface_monitor_sync(AvahiInterfaceMonitor *m) {
551 while (m->list != LIST_DONE) {
552 if (!avahi_netlink_work(m->netlink, 1))
557 void avahi_interface_monitor_free(AvahiInterfaceMonitor *m) {
560 while (m->hw_interfaces)
561 free_hw_interface(m, m->hw_interfaces, 1);
563 assert(!m->interfaces);
567 avahi_netlink_free(m->netlink);
570 avahi_hashmap_free(m->hashmap);
576 AvahiInterface* avahi_interface_monitor_get_interface(AvahiInterfaceMonitor *m, AvahiIfIndex idx, AvahiProtocol protocol) {
577 AvahiHwInterface *hw;
582 assert(protocol != AVAHI_PROTO_UNSPEC);
584 if (!(hw = avahi_interface_monitor_get_hw_interface(m, idx)))
587 for (i = hw->interfaces; i; i = i->by_hardware_next)
588 if (i->protocol == protocol)
594 AvahiHwInterface* avahi_interface_monitor_get_hw_interface(AvahiInterfaceMonitor *m, AvahiIfIndex idx) {
598 return avahi_hashmap_lookup(m->hashmap, &idx);
601 void avahi_interface_send_packet_unicast(AvahiInterface *i, AvahiDnsPacket *p, const AvahiAddress *a, uint16_t port) {
606 if (!avahi_interface_relevant(i))
609 assert(!a || a->family == i->protocol);
612 /* avahi_log_debug("unicast sending on '%s.%i' to %s:%u", i->hardware->name, i->protocol, avahi_address_snprint(t, sizeof(t), a), port); */
614 /* avahi_log_debug("multicast sending on '%s.%i'", i->hardware->name, i->protocol); */
616 if (i->protocol == AVAHI_PROTO_INET && i->monitor->server->fd_ipv4 >= 0)
617 avahi_send_dns_packet_ipv4(i->monitor->server->fd_ipv4, i->hardware->index, p, a ? &a->data.ipv4 : NULL, port);
618 else if (i->protocol == AVAHI_PROTO_INET6 && i->monitor->server->fd_ipv6 >= 0)
619 avahi_send_dns_packet_ipv6(i->monitor->server->fd_ipv6, i->hardware->index, p, a ? &a->data.ipv6 : NULL, port);
622 void avahi_interface_send_packet(AvahiInterface *i, AvahiDnsPacket *p) {
626 avahi_interface_send_packet_unicast(i, p, NULL, 0);
629 int avahi_interface_post_query(AvahiInterface *i, AvahiKey *key, int immediately) {
633 if (avahi_interface_relevant(i))
634 return avahi_query_scheduler_post(i->query_scheduler, key, immediately);
639 int avahi_interface_post_response(AvahiInterface *i, AvahiRecord *record, int flush_cache, const AvahiAddress *querier, int immediately) {
643 if (avahi_interface_relevant(i))
644 return avahi_response_scheduler_post(i->response_scheduler, record, flush_cache, querier, immediately);
649 int avahi_interface_post_probe(AvahiInterface *i, AvahiRecord *record, int immediately) {
653 if (avahi_interface_relevant(i))
654 return avahi_probe_scheduler_post(i->probe_scheduler, record, immediately);
659 int avahi_dump_caches(AvahiInterfaceMonitor *m, AvahiDumpCallback callback, void* userdata) {
663 for (i = m->interfaces; i; i = i->interface_next) {
664 if (avahi_interface_relevant(i)) {
666 snprintf(ln, sizeof(ln), ";;; INTERFACE %s.%i ;;;", i->hardware->name, i->protocol);
667 callback(ln, userdata);
668 if (avahi_cache_dump(i->cache, callback, userdata) < 0)
676 int avahi_interface_relevant(AvahiInterface *i) {
677 AvahiInterfaceAddress *a;
678 int relevant_address;
682 relevant_address = 0;
684 for (a = i->addresses; a; a = a->address_next)
685 if (avahi_interface_address_relevant(a)) {
686 relevant_address = 1;
690 /* avahi_log_debug("%p. iface-relevant: %i %i %i %i %i %i", i, relevant_address, */
691 /* (i->hardware->flags & IFF_UP), */
692 /* (i->hardware->flags & IFF_RUNNING), */
693 /* !(i->hardware->flags & IFF_LOOPBACK), */
694 /* (i->hardware->flags & IFF_MULTICAST), */
695 /* !(i->hardware->flags & IFF_POINTOPOINT)); */
698 (i->hardware->flags & IFF_UP) &&
699 (!i->monitor->server->config.use_iff_running || (i->hardware->flags & IFF_RUNNING)) &&
700 !(i->hardware->flags & IFF_LOOPBACK) &&
701 (i->hardware->flags & IFF_MULTICAST) &&
702 !(i->hardware->flags & IFF_POINTOPOINT) &&
706 int avahi_interface_address_relevant(AvahiInterfaceAddress *a) {
709 return a->scope == RT_SCOPE_UNIVERSE;
713 int avahi_interface_match(AvahiInterface *i, AvahiIfIndex idx, AvahiProtocol protocol) {
716 if (idx > 0 && idx != i->hardware->index)
719 if (protocol != AVAHI_PROTO_UNSPEC && protocol != i->protocol)
725 void avahi_interface_monitor_walk(AvahiInterfaceMonitor *m, AvahiIfIndex interface, AvahiProtocol protocol, AvahiInterfaceMonitorWalkCallback callback, void* userdata) {
730 if (protocol != AVAHI_PROTO_UNSPEC) {
733 if ((i = avahi_interface_monitor_get_interface(m, interface, protocol)))
734 callback(m, i, userdata);
737 AvahiHwInterface *hw;
740 if ((hw = avahi_interface_monitor_get_hw_interface(m, interface)))
741 for (i = hw->interfaces; i; i = i->by_hardware_next)
742 if (avahi_interface_match(i, interface, protocol))
743 callback(m, i, userdata);
749 for (i = m->interfaces; i; i = i->interface_next)
750 if (avahi_interface_match(i, interface, protocol))
751 callback(m, i, userdata);
755 void avahi_update_host_rrs(AvahiInterfaceMonitor *m, int remove_rrs) {
756 AvahiHwInterface *hw;
760 for (hw = m->hw_interfaces; hw; hw = hw->hardware_next)
761 update_hw_interface_rr(m, hw, remove_rrs);
764 int avahi_address_is_local(AvahiInterfaceMonitor *m, const AvahiAddress *a) {
766 AvahiInterfaceAddress *ia;
770 for (i = m->interfaces; i; i = i->interface_next)
771 for (ia = i->addresses; ia; ia = ia->address_next)
772 if (avahi_address_cmp(a, &ia->address) == 0)
778 int avahi_interface_address_on_link(AvahiInterface *i, const AvahiAddress *a) {
779 AvahiInterfaceAddress *ia;
784 if (a->family != i->protocol)
787 for (ia = i->addresses; ia; ia = ia->address_next) {
789 if (a->family == AVAHI_PROTO_INET) {
792 m = ~(((uint32_t) -1) >> ia->prefix_len);
794 if ((ntohl(a->data.ipv4.address) & m) == (ntohl(ia->address.data.ipv4.address) & m))
799 assert(a->family == AVAHI_PROTO_INET6);
803 for (j = 0; j < 16; j++) {
817 if ((a->data.ipv6.address[j] & m) != (ia->address.data.ipv6.address[j] & m))