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>
41 static void update_address_rr(AvahiInterfaceMonitor *m, AvahiInterfaceAddress *a, gboolean remove) {
45 if (avahi_interface_address_relevant(a) &&
47 m->server->config.register_addresses &&
48 (m->server->state == AVAHI_SERVER_RUNNING ||
49 m->server->state == AVAHI_SERVER_REGISTERING)) {
51 if (!a->entry_group) {
52 a->entry_group = avahi_entry_group_new(m->server, avahi_host_rr_entry_group_callback, NULL);
53 avahi_server_add_address(m->server, a->entry_group, a->interface->hardware->index, a->interface->protocol, 0, NULL, &a->address);
54 avahi_entry_group_commit(a->entry_group);
60 if (avahi_entry_group_get_state(a->entry_group) == AVAHI_ENTRY_GROUP_REGISTERING)
61 avahi_server_decrease_host_rr_pending(m->server);
63 avahi_entry_group_free(a->entry_group);
64 a->entry_group = NULL;
69 static void update_interface_rr(AvahiInterfaceMonitor *m, AvahiInterface *i, gboolean remove) {
70 AvahiInterfaceAddress *a;
75 for (a = i->addresses; a; a = a->address_next)
76 update_address_rr(m, a, remove);
79 static void update_hw_interface_rr(AvahiInterfaceMonitor *m, AvahiHwInterface *hw, gboolean remove) {
85 for (i = hw->interfaces; i; i = i->by_hardware_next)
86 update_interface_rr(m, i, remove);
89 m->server->config.register_workstation &&
90 (m->server->state == AVAHI_SERVER_RUNNING ||
91 m->server->state == AVAHI_SERVER_REGISTERING)) {
93 if (!hw->entry_group) {
95 gchar *t = avahi_format_mac_address(hw->mac_address, hw->mac_address_size);
96 name = g_strdup_printf("%s [%s]", m->server->host_name, t);
99 hw->entry_group = avahi_entry_group_new(m->server, avahi_host_rr_entry_group_callback, NULL);
100 avahi_server_add_service(m->server, hw->entry_group, hw->index, AF_UNSPEC, "_workstation._tcp", name, NULL, NULL, 9, NULL);
101 avahi_entry_group_commit(hw->entry_group);
108 if (hw->entry_group) {
110 if (avahi_entry_group_get_state(hw->entry_group) == AVAHI_ENTRY_GROUP_REGISTERING)
111 avahi_server_decrease_host_rr_pending(m->server);
113 avahi_entry_group_free(hw->entry_group);
114 hw->entry_group = NULL;
119 static void free_address(AvahiInterfaceMonitor *m, AvahiInterfaceAddress *a) {
122 g_assert(a->interface);
124 update_address_rr(m, a, TRUE);
125 AVAHI_LLIST_REMOVE(AvahiInterfaceAddress, address, a->interface->addresses, a);
130 static void free_interface(AvahiInterfaceMonitor *m, AvahiInterface *i, gboolean send_goodbye) {
134 avahi_goodbye_interface(m->server, i, send_goodbye);
135 avahi_response_scheduler_force(i->response_scheduler);
137 g_assert(!i->announcements);
139 update_interface_rr(m, i, TRUE);
142 free_address(m, i->addresses);
144 avahi_response_scheduler_free(i->response_scheduler);
145 avahi_query_scheduler_free(i->query_scheduler);
146 avahi_probe_scheduler_free(i->probe_scheduler);
147 avahi_cache_free(i->cache);
149 AVAHI_LLIST_REMOVE(AvahiInterface, interface, m->interfaces, i);
150 AVAHI_LLIST_REMOVE(AvahiInterface, by_hardware, i->hardware->interfaces, i);
155 static void free_hw_interface(AvahiInterfaceMonitor *m, AvahiHwInterface *hw, gboolean send_goodbye) {
159 update_hw_interface_rr(m, hw, TRUE);
161 while (hw->interfaces)
162 free_interface(m, hw->interfaces, send_goodbye);
164 AVAHI_LLIST_REMOVE(AvahiHwInterface, hardware, m->hw_interfaces, hw);
165 g_hash_table_remove(m->hash_table, &hw->index);
171 static AvahiInterfaceAddress* get_address(AvahiInterfaceMonitor *m, AvahiInterface *i, const AvahiAddress *raddr) {
172 AvahiInterfaceAddress *ia;
178 for (ia = i->addresses; ia; ia = ia->address_next)
179 if (avahi_address_cmp(&ia->address, raddr) == 0)
185 static int netlink_list_items(AvahiNetlink *nl, guint16 type, guint *ret_seq) {
187 struct rtgenmsg *gen;
190 memset(&req, 0, sizeof(req));
191 n = (struct nlmsghdr*) req;
192 n->nlmsg_len = NLMSG_LENGTH(sizeof(struct rtgenmsg));
193 n->nlmsg_type = type;
194 n->nlmsg_flags = NLM_F_ROOT/*|NLM_F_MATCH*/|NLM_F_REQUEST;
198 memset(gen, 0, sizeof(struct rtgenmsg));
199 gen->rtgen_family = AF_UNSPEC;
201 return avahi_netlink_send(nl, n, ret_seq);
204 static void new_interface(AvahiInterfaceMonitor *m, AvahiHwInterface *hw, guchar protocol) {
209 g_assert(protocol != AF_UNSPEC);
211 i = g_new(AvahiInterface, 1);
214 i->protocol = protocol;
215 i->announcing = FALSE;
217 AVAHI_LLIST_HEAD_INIT(AvahiInterfaceAddress, i->addresses);
218 AVAHI_LLIST_HEAD_INIT(AvahiAnnouncement, i->announcements);
220 i->cache = avahi_cache_new(m->server, i);
221 i->response_scheduler = avahi_response_scheduler_new(i);
222 i->query_scheduler = avahi_query_scheduler_new(i);
223 i->probe_scheduler = avahi_probe_scheduler_new(i);
225 AVAHI_LLIST_PREPEND(AvahiInterface, by_hardware, hw->interfaces, i);
226 AVAHI_LLIST_PREPEND(AvahiInterface, interface, m->interfaces, i);
229 static void check_interface_relevant(AvahiInterfaceMonitor *m, AvahiInterface *i) {
235 b = avahi_interface_relevant(i);
237 if (b && !i->announcing) {
238 g_message("New relevant interface %s.%i (#%i)", i->hardware->name, i->protocol, i->hardware->index);
240 if (i->protocol == AF_INET)
241 avahi_mdns_mcast_join_ipv4(i->hardware->index, m->server->fd_ipv4);
242 if (i->protocol == AF_INET6)
243 avahi_mdns_mcast_join_ipv6(i->hardware->index, m->server->fd_ipv6);
245 i->announcing = TRUE;
246 avahi_announce_interface(m->server, i);
247 } else if (!b && i->announcing) {
248 g_message("Interface %s.%i no longer relevant", i->hardware->name, i->protocol);
250 if (i->protocol == AF_INET)
251 avahi_mdns_mcast_leave_ipv4(i->hardware->index, m->server->fd_ipv4);
252 if (i->protocol == AF_INET6)
253 avahi_mdns_mcast_leave_ipv6(i->hardware->index, m->server->fd_ipv6);
255 avahi_goodbye_interface(m->server, i, FALSE);
256 avahi_response_scheduler_clear(i->response_scheduler);
257 avahi_query_scheduler_clear(i->query_scheduler);
258 avahi_probe_scheduler_clear(i->probe_scheduler);
259 avahi_cache_flush(i->cache);
261 i->announcing = FALSE;
265 static void check_hw_interface_relevant(AvahiInterfaceMonitor *m, AvahiHwInterface *hw) {
271 for (i = hw->interfaces; i; i = i->by_hardware_next)
272 check_interface_relevant(m, i);
275 static void callback(AvahiNetlink *nl, struct nlmsghdr *n, gpointer userdata) {
276 AvahiInterfaceMonitor *m = userdata;
280 g_assert(m->netlink == nl);
282 if (n->nlmsg_type == RTM_NEWLINK) {
283 struct ifinfomsg *ifinfomsg = NLMSG_DATA(n);
284 AvahiHwInterface *hw;
285 struct rtattr *a = NULL;
288 if (ifinfomsg->ifi_family != AF_UNSPEC)
291 if (!(hw = g_hash_table_lookup(m->hash_table, &ifinfomsg->ifi_index))) {
292 hw = g_new(AvahiHwInterface, 1);
297 hw->index = ifinfomsg->ifi_index;
298 hw->mac_address_size = 0;
299 hw->entry_group = NULL;
301 AVAHI_LLIST_HEAD_INIT(AvahiInterface, hw->interfaces);
302 AVAHI_LLIST_PREPEND(AvahiHwInterface, hardware, m->hw_interfaces, hw);
304 g_hash_table_insert(m->hash_table, &hw->index, hw);
306 if (m->server->fd_ipv4 >= 0)
307 new_interface(m, hw, AF_INET);
308 if (m->server->fd_ipv6 >= 0)
309 new_interface(m, hw, AF_INET6);
312 hw->flags = ifinfomsg->ifi_flags;
314 l = NLMSG_PAYLOAD(n, sizeof(struct ifinfomsg));
315 a = IFLA_RTA(ifinfomsg);
317 while (RTA_OK(a, l)) {
318 switch(a->rta_type) {
321 hw->name = g_strndup(RTA_DATA(a), RTA_PAYLOAD(a));
325 g_assert(RTA_PAYLOAD(a) == sizeof(unsigned int));
326 hw->mtu = *((unsigned int*) RTA_DATA(a));
330 hw->mac_address_size = RTA_PAYLOAD(a);
331 if (hw->mac_address_size > AVAHI_MAX_MAC_ADDRESS)
332 hw->mac_address_size = AVAHI_MAX_MAC_ADDRESS;
334 memcpy(hw->mac_address, RTA_DATA(a), hw->mac_address_size);
345 update_hw_interface_rr(m, hw, FALSE);
346 check_hw_interface_relevant(m, hw);
348 } else if (n->nlmsg_type == RTM_DELLINK) {
349 struct ifinfomsg *ifinfomsg = NLMSG_DATA(n);
350 AvahiHwInterface *hw;
352 if (ifinfomsg->ifi_family != AF_UNSPEC)
355 if (!(hw = avahi_interface_monitor_get_hw_interface(m, ifinfomsg->ifi_index)))
358 update_hw_interface_rr(m, hw, TRUE);
359 free_hw_interface(m, hw, FALSE);
361 } else if (n->nlmsg_type == RTM_NEWADDR || n->nlmsg_type == RTM_DELADDR) {
363 struct ifaddrmsg *ifaddrmsg = NLMSG_DATA(n);
365 struct rtattr *a = NULL;
368 gboolean raddr_valid = FALSE;
370 if (ifaddrmsg->ifa_family != AF_INET && ifaddrmsg->ifa_family != AF_INET6)
373 if (!(i = (AvahiInterface*) avahi_interface_monitor_get_interface(m, ifaddrmsg->ifa_index, ifaddrmsg->ifa_family)))
376 raddr.family = ifaddrmsg->ifa_family;
378 l = NLMSG_PAYLOAD(n, sizeof(struct ifaddrmsg));
379 a = IFA_RTA(ifaddrmsg);
381 while (RTA_OK(a, l)) {
383 switch(a->rta_type) {
385 if ((raddr.family == AF_INET6 && RTA_PAYLOAD(a) != 16) ||
386 (raddr.family == AF_INET && RTA_PAYLOAD(a) != 4))
389 memcpy(raddr.data.data, RTA_DATA(a), RTA_PAYLOAD(a));
404 if (n->nlmsg_type == RTM_NEWADDR) {
405 AvahiInterfaceAddress *addr;
407 if (!(addr = get_address(m, i, &raddr))) {
408 addr = g_new(AvahiInterfaceAddress, 1);
410 addr->address = raddr;
412 addr->entry_group = NULL;
414 AVAHI_LLIST_PREPEND(AvahiInterfaceAddress, address, i->addresses, addr);
417 addr->flags = ifaddrmsg->ifa_flags;
418 addr->scope = ifaddrmsg->ifa_scope;
420 update_address_rr(m, addr, FALSE);
421 check_interface_relevant(m, i);
423 AvahiInterfaceAddress *addr;
425 if (!(addr = get_address(m, i, &raddr)))
428 update_address_rr(m, addr, TRUE);
429 free_address(m, addr);
431 check_interface_relevant(m, i);
434 } else if (n->nlmsg_type == NLMSG_DONE) {
436 if (m->list == LIST_IFACE) {
439 if (netlink_list_items(m->netlink, RTM_GETADDR, &m->query_addr_seq) < 0)
440 g_warning("NETLINK: Failed to list addrs: %s", strerror(errno));
445 g_message("Enumeration complete");
448 } else if (n->nlmsg_type == NLMSG_ERROR && (n->nlmsg_seq == m->query_link_seq || n->nlmsg_seq == m->query_addr_seq)) {
449 struct nlmsgerr *e = NLMSG_DATA (n);
452 g_warning("NETLINK: Failed to browse: %s", strerror(-e->error));
456 AvahiInterfaceMonitor *avahi_interface_monitor_new(AvahiServer *s) {
457 AvahiInterfaceMonitor *m = NULL;
459 m = g_new0(AvahiInterfaceMonitor, 1);
461 if (!(m->netlink = avahi_netlink_new(s->context, G_PRIORITY_DEFAULT-10, RTMGRP_LINK|RTMGRP_IPV4_IFADDR|RTMGRP_IPV6_IFADDR, callback, m)))
464 m->hash_table = g_hash_table_new(g_int_hash, g_int_equal);
466 AVAHI_LLIST_HEAD_INIT(AvahiInterface, m->interfaces);
467 AVAHI_LLIST_HEAD_INIT(AvahiHwInterface, m->hw_interfaces);
469 if (netlink_list_items(m->netlink, RTM_GETLINK, &m->query_link_seq) < 0)
472 m->list = LIST_IFACE;
477 avahi_interface_monitor_free(m);
481 void avahi_interface_monitor_sync(AvahiInterfaceMonitor *m) {
484 while (m->list != LIST_DONE) {
485 if (!avahi_netlink_work(m->netlink, TRUE))
490 void avahi_interface_monitor_free(AvahiInterfaceMonitor *m) {
493 while (m->hw_interfaces)
494 free_hw_interface(m, m->hw_interfaces, TRUE);
496 g_assert(!m->interfaces);
500 avahi_netlink_free(m->netlink);
503 g_hash_table_destroy(m->hash_table);
509 AvahiInterface* avahi_interface_monitor_get_interface(AvahiInterfaceMonitor *m, gint index, guchar protocol) {
510 AvahiHwInterface *hw;
515 g_assert(protocol != AF_UNSPEC);
517 if (!(hw = avahi_interface_monitor_get_hw_interface(m, index)))
520 for (i = hw->interfaces; i; i = i->by_hardware_next)
521 if (i->protocol == protocol)
527 AvahiHwInterface* avahi_interface_monitor_get_hw_interface(AvahiInterfaceMonitor *m, gint index) {
531 return g_hash_table_lookup(m->hash_table, &index);
535 void avahi_interface_send_packet_unicast(AvahiInterface *i, AvahiDnsPacket *p, const AvahiAddress *a, guint16 port) {
540 if (!avahi_interface_relevant(i))
543 g_assert(!a || a->family == i->protocol);
546 /* g_message("unicast sending on '%s.%i' to %s:%u", i->hardware->name, i->protocol, avahi_address_snprint(t, sizeof(t), a), port); */
548 /* g_message("multicast sending on '%s.%i'", i->hardware->name, i->protocol); */
550 if (i->protocol == AF_INET && i->monitor->server->fd_ipv4 >= 0)
551 avahi_send_dns_packet_ipv4(i->monitor->server->fd_ipv4, i->hardware->index, p, a ? &a->data.ipv4 : NULL, port);
552 else if (i->protocol == AF_INET6 && i->monitor->server->fd_ipv6 >= 0)
553 avahi_send_dns_packet_ipv6(i->monitor->server->fd_ipv6, i->hardware->index, p, a ? &a->data.ipv6 : NULL, port);
556 void avahi_interface_send_packet(AvahiInterface *i, AvahiDnsPacket *p) {
560 avahi_interface_send_packet_unicast(i, p, NULL, 0);
563 gboolean avahi_interface_post_query(AvahiInterface *i, AvahiKey *key, gboolean immediately) {
567 if (avahi_interface_relevant(i))
568 return avahi_query_scheduler_post(i->query_scheduler, key, immediately);
573 gboolean avahi_interface_post_response(AvahiInterface *i, AvahiRecord *record, gboolean flush_cache, const AvahiAddress *querier, gboolean immediately) {
577 if (avahi_interface_relevant(i))
578 return avahi_response_scheduler_post(i->response_scheduler, record, flush_cache, querier, immediately);
583 gboolean avahi_interface_post_probe(AvahiInterface *i, AvahiRecord *record, gboolean immediately) {
587 if (avahi_interface_relevant(i))
588 return avahi_probe_scheduler_post(i->probe_scheduler, record, immediately);
593 void avahi_dump_caches(AvahiInterfaceMonitor *m, FILE *f) {
597 for (i = m->interfaces; i; i = i->interface_next) {
598 if (avahi_interface_relevant(i)) {
599 fprintf(f, "\n;;; INTERFACE %s.%i ;;;\n", i->hardware->name, i->protocol);
600 avahi_cache_dump(i->cache, f);
606 gboolean avahi_interface_relevant(AvahiInterface *i) {
607 AvahiInterfaceAddress *a;
608 gboolean relevant_address;
612 relevant_address = FALSE;
614 for (a = i->addresses; a; a = a->address_next)
615 if (avahi_interface_address_relevant(a)) {
616 relevant_address = TRUE;
620 /* g_message("%p. iface-relevant: %i %i %i %i %i %i", i, relevant_address, */
621 /* (i->hardware->flags & IFF_UP), */
622 /* (i->hardware->flags & IFF_RUNNING), */
623 /* !(i->hardware->flags & IFF_LOOPBACK), */
624 /* (i->hardware->flags & IFF_MULTICAST), */
625 /* !(i->hardware->flags & IFF_POINTOPOINT)); */
628 (i->hardware->flags & IFF_UP) &&
629 (!i->monitor->server->config.use_iff_running || (i->hardware->flags & IFF_RUNNING)) &&
630 !(i->hardware->flags & IFF_LOOPBACK) &&
631 (i->hardware->flags & IFF_MULTICAST) &&
632 !(i->hardware->flags & IFF_POINTOPOINT) &&
636 gboolean avahi_interface_address_relevant(AvahiInterfaceAddress *a) {
639 return a->scope == RT_SCOPE_UNIVERSE;
643 gboolean avahi_interface_match(AvahiInterface *i, gint index, guchar protocol) {
646 if (index > 0 && index != i->hardware->index)
649 if (protocol != AF_UNSPEC && protocol != i->protocol)
655 void avahi_interface_monitor_walk(AvahiInterfaceMonitor *m, gint interface, guchar protocol, AvahiInterfaceMonitorWalkCallback callback, gpointer userdata) {
660 if (protocol != AF_UNSPEC) {
663 if ((i = avahi_interface_monitor_get_interface(m, interface, protocol)))
664 callback(m, i, userdata);
667 AvahiHwInterface *hw;
670 if ((hw = avahi_interface_monitor_get_hw_interface(m, interface)))
671 for (i = hw->interfaces; i; i = i->by_hardware_next)
672 if (avahi_interface_match(i, interface, protocol))
673 callback(m, i, userdata);
679 for (i = m->interfaces; i; i = i->interface_next)
680 if (avahi_interface_match(i, interface, protocol))
681 callback(m, i, userdata);
685 void avahi_update_host_rrs(AvahiInterfaceMonitor *m, gboolean remove) {
686 AvahiHwInterface *hw;
690 for (hw = m->hw_interfaces; hw; hw = hw->hardware_next)
691 update_hw_interface_rr(m, hw, remove);
694 gboolean avahi_address_is_local(AvahiInterfaceMonitor *m, const AvahiAddress *a) {
696 AvahiInterfaceAddress *ia;
700 for (i = m->interfaces; i; i = i->interface_next)
701 for (ia = i->addresses; ia; ia = ia->address_next)
702 if (avahi_address_cmp(a, &ia->address) == 0)