2 This file is part of catta.
4 catta is free software; you can redistribute it and/or modify it
5 under the terms of the GNU Lesser General Public License as
6 published by the Free Software Foundation; either version 2.1 of the
7 License, or (at your option) any later version.
9 catta is distributed in the hope that it will be useful, but WITHOUT
10 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
12 Public License for more details.
14 You should have received a copy of the GNU Lesser General Public
15 License along with catta; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
28 #include <sys/types.h>
29 #include <sys/socket.h>
30 #include <netinet/in.h>
32 #include <catta/error.h>
33 #include <catta/malloc.h>
34 #include <catta/domain.h>
35 #include <catta/log.h>
42 #include "multicast-lookup.h"
45 void catta_interface_address_update_rrs(CattaInterfaceAddress *a, int remove_rrs) {
46 CattaInterfaceMonitor *m;
51 if (m->list_complete &&
52 catta_interface_address_is_relevant(a) &&
53 catta_interface_is_relevant(a->iface) &&
55 m->server->config.publish_addresses &&
56 (m->server->state == CATTA_SERVER_RUNNING ||
57 m->server->state == CATTA_SERVER_REGISTERING)) {
59 /* Fill the entry group */
61 a->entry_group = catta_s_entry_group_new(m->server, catta_host_rr_entry_group_callback, NULL);
63 if (!a->entry_group) /* OOM */
66 if (catta_s_entry_group_is_empty(a->entry_group)) {
67 char t[CATTA_ADDRESS_STR_MAX];
70 p = (a->iface->protocol == CATTA_PROTO_INET && m->server->config.publish_a_on_ipv6) ||
71 (a->iface->protocol == CATTA_PROTO_INET6 && m->server->config.publish_aaaa_on_ipv4) ? CATTA_PROTO_UNSPEC : a->iface->protocol;
73 catta_address_snprint(t, sizeof(t), &a->address);
74 catta_log_info("Registering new address record for %s on %s.%s.", t, a->iface->hardware->name, p == CATTA_PROTO_UNSPEC ? "*" : catta_proto_to_string(p));
76 if (catta_server_add_address(m->server, a->entry_group, a->iface->hardware->index, p, m->server->config.publish_no_reverse ? CATTA_PUBLISH_NO_REVERSE : 0, NULL, &a->address) < 0) {
77 catta_log_warn(__FILE__": catta_server_add_address() failed: %s", catta_strerror(m->server->error));
78 catta_s_entry_group_free(a->entry_group);
79 a->entry_group = NULL;
83 catta_s_entry_group_commit(a->entry_group);
87 /* Clear the entry group */
89 if (a->entry_group && !catta_s_entry_group_is_empty(a->entry_group)) {
90 char t[CATTA_ADDRESS_STR_MAX];
91 catta_address_snprint(t, sizeof(t), &a->address);
93 catta_log_info("Withdrawing address record for %s on %s.", t, a->iface->hardware->name);
95 if (catta_s_entry_group_get_state(a->entry_group) == CATTA_ENTRY_GROUP_REGISTERING &&
96 m->server->state == CATTA_SERVER_REGISTERING)
97 catta_server_decrease_host_rr_pending(m->server);
99 catta_s_entry_group_reset(a->entry_group);
104 void catta_interface_update_rrs(CattaInterface *i, int remove_rrs) {
105 CattaInterfaceAddress *a;
109 for (a = i->addresses; a; a = a->address_next)
110 catta_interface_address_update_rrs(a, remove_rrs);
113 void catta_hw_interface_update_rrs(CattaHwInterface *hw, int remove_rrs) {
115 CattaInterfaceMonitor *m;
120 for (i = hw->interfaces; i; i = i->by_hardware_next)
121 catta_interface_update_rrs(i, remove_rrs);
123 if (m->list_complete &&
125 m->server->config.publish_workstation &&
126 (m->server->state == CATTA_SERVER_RUNNING)) {
128 if (!hw->entry_group)
129 hw->entry_group = catta_s_entry_group_new(m->server, catta_host_rr_entry_group_callback, NULL);
131 if (!hw->entry_group)
134 if (catta_s_entry_group_is_empty(hw->entry_group)) {
135 char name[CATTA_LABEL_MAX], unescaped[CATTA_LABEL_MAX], mac[256];
136 const char *p = m->server->host_name;
138 catta_unescape_label(&p, unescaped, sizeof(unescaped));
139 catta_format_mac_address(mac, sizeof(mac), hw->mac_address, hw->mac_address_size);
140 snprintf(name, sizeof(name), "%s [%s]", unescaped, mac);
142 if (catta_server_add_service(m->server, hw->entry_group, hw->index, CATTA_PROTO_UNSPEC, 0, name, "_workstation._tcp", NULL, NULL, 9, NULL) < 0) {
143 catta_log_warn(__FILE__": catta_server_add_service() failed: %s", catta_strerror(m->server->error));
144 catta_s_entry_group_free(hw->entry_group);
145 hw->entry_group = NULL;
147 catta_s_entry_group_commit(hw->entry_group);
152 if (hw->entry_group && !catta_s_entry_group_is_empty(hw->entry_group)) {
154 catta_log_info("Withdrawing workstation service for %s.", hw->name);
156 if (catta_s_entry_group_get_state(hw->entry_group) == CATTA_ENTRY_GROUP_REGISTERING &&
157 m->server->state == CATTA_SERVER_REGISTERING)
158 catta_server_decrease_host_rr_pending(m->server);
160 catta_s_entry_group_reset(hw->entry_group);
165 void catta_interface_monitor_update_rrs(CattaInterfaceMonitor *m, int remove_rrs) {
166 CattaHwInterface *hw;
170 for (hw = m->hw_interfaces; hw; hw = hw->hardware_next)
171 catta_hw_interface_update_rrs(hw, remove_rrs);
174 static int interface_mdns_mcast_join(CattaInterface *i, int join) {
175 char at[CATTA_ADDRESS_STR_MAX];
179 if (!!join == !!i->mcast_joined)
182 if ((i->protocol == CATTA_PROTO_INET6 && i->monitor->server->fd_ipv6 < 0) ||
183 (i->protocol == CATTA_PROTO_INET && i->monitor->server->fd_ipv4 < 0))
187 CattaInterfaceAddress *a;
189 /* Look if there's an address with global scope */
190 for (a = i->addresses; a; a = a->address_next)
194 /* No address with a global scope has been found, so let's use
199 /* Hmm, there is no address available. */
203 i->local_mcast_address = a->address;
206 catta_log_info("%s mDNS multicast group on interface %s.%s with address %s.",
207 join ? "Joining" : "Leaving",
209 catta_proto_to_string(i->protocol),
210 catta_address_snprint(at, sizeof(at), &i->local_mcast_address));
212 if (i->protocol == CATTA_PROTO_INET6)
213 r = catta_mdns_mcast_join_ipv6(i->monitor->server->fd_ipv6, &i->local_mcast_address.data.ipv6, i->hardware->index, join);
215 assert(i->protocol == CATTA_PROTO_INET);
217 r = catta_mdns_mcast_join_ipv4(i->monitor->server->fd_ipv4, &i->local_mcast_address.data.ipv4, i->hardware->index, join);
223 i->mcast_joined = join;
228 static int interface_mdns_mcast_rejoin(CattaInterface *i) {
229 CattaInterfaceAddress *a, *usable = NULL, *found = NULL;
232 if (!i->mcast_joined)
235 /* Check whether old address we joined with is still available. If
236 * not, rejoin using an other address. */
238 for (a = i->addresses; a; a = a->address_next) {
239 if (a->global_scope && !usable)
242 if (catta_address_cmp(&a->address, &i->local_mcast_address) == 0) {
245 /* No action necessary: the address still exists and
246 * has global scope. */
253 if (found && !usable)
254 /* No action necessary: the address still exists and no better one has been found */
257 interface_mdns_mcast_join(i, 0);
258 return interface_mdns_mcast_join(i, 1);
261 void catta_interface_address_free(CattaInterfaceAddress *a) {
265 catta_interface_address_update_rrs(a, 1);
266 CATTA_LLIST_REMOVE(CattaInterfaceAddress, address, a->iface->addresses, a);
269 catta_s_entry_group_free(a->entry_group);
271 interface_mdns_mcast_rejoin(a->iface);
276 void catta_interface_free(CattaInterface *i, int send_goodbye) {
279 /* Handle goodbyes and remove announcers */
280 catta_goodbye_interface(i->monitor->server, i, send_goodbye, 1);
281 catta_response_scheduler_force(i->response_scheduler);
282 assert(!i->announcers);
285 interface_mdns_mcast_join(i, 0);
287 /* Remove queriers */
288 catta_querier_free_all(i);
289 catta_hashmap_free(i->queriers_by_key);
291 /* Remove local RRs */
292 catta_interface_update_rrs(i, 1);
295 catta_interface_address_free(i->addresses);
297 catta_response_scheduler_free(i->response_scheduler);
298 catta_query_scheduler_free(i->query_scheduler);
299 catta_probe_scheduler_free(i->probe_scheduler);
300 catta_cache_free(i->cache);
302 CATTA_LLIST_REMOVE(CattaInterface, iface, i->monitor->interfaces, i);
303 CATTA_LLIST_REMOVE(CattaInterface, by_hardware, i->hardware->interfaces, i);
308 void catta_hw_interface_free(CattaHwInterface *hw, int send_goodbye) {
311 catta_hw_interface_update_rrs(hw, 1);
313 while (hw->interfaces)
314 catta_interface_free(hw->interfaces, send_goodbye);
317 catta_s_entry_group_free(hw->entry_group);
319 CATTA_LLIST_REMOVE(CattaHwInterface, hardware, hw->monitor->hw_interfaces, hw);
320 catta_hashmap_remove(hw->monitor->hashmap, &hw->index);
322 catta_free(hw->name);
326 CattaInterface* catta_interface_new(CattaInterfaceMonitor *m, CattaHwInterface *hw, CattaProtocol protocol) {
331 assert(CATTA_PROTO_VALID(protocol));
333 if (!(i = catta_new(CattaInterface, 1)))
338 i->protocol = protocol;
342 CATTA_LLIST_HEAD_INIT(CattaInterfaceAddress, i->addresses);
343 CATTA_LLIST_HEAD_INIT(CattaAnnouncer, i->announcers);
345 CATTA_LLIST_HEAD_INIT(CattaQuerier, i->queriers);
346 i->queriers_by_key = catta_hashmap_new((CattaHashFunc) catta_key_hash, (CattaEqualFunc) catta_key_equal, NULL, NULL);
348 i->cache = catta_cache_new(m->server, i);
349 i->response_scheduler = catta_response_scheduler_new(i);
350 i->query_scheduler = catta_query_scheduler_new(i);
351 i->probe_scheduler = catta_probe_scheduler_new(i);
353 if (!i->cache || !i->response_scheduler || !i->query_scheduler || !i->probe_scheduler)
356 CATTA_LLIST_PREPEND(CattaInterface, by_hardware, hw->interfaces, i);
357 CATTA_LLIST_PREPEND(CattaInterface, iface, m->interfaces, i);
365 catta_cache_free(i->cache);
366 if (i->response_scheduler)
367 catta_response_scheduler_free(i->response_scheduler);
368 if (i->query_scheduler)
369 catta_query_scheduler_free(i->query_scheduler);
370 if (i->probe_scheduler)
371 catta_probe_scheduler_free(i->probe_scheduler);
377 CattaHwInterface *catta_hw_interface_new(CattaInterfaceMonitor *m, CattaIfIndex idx) {
378 CattaHwInterface *hw;
381 assert(CATTA_IF_VALID(idx));
383 if (!(hw = catta_new(CattaHwInterface, 1)))
391 hw->mac_address_size = 0;
392 hw->entry_group = NULL;
393 hw->ratelimit_begin.tv_sec = 0;
394 hw->ratelimit_begin.tv_usec = 0;
395 hw->ratelimit_counter = 0;
397 CATTA_LLIST_HEAD_INIT(CattaInterface, hw->interfaces);
398 CATTA_LLIST_PREPEND(CattaHwInterface, hardware, m->hw_interfaces, hw);
400 catta_hashmap_insert(m->hashmap, &hw->index, hw);
402 if (m->server->fd_ipv4 >= 0 || m->server->config.publish_a_on_ipv6)
403 catta_interface_new(m, hw, CATTA_PROTO_INET);
404 if (m->server->fd_ipv6 >= 0 || m->server->config.publish_aaaa_on_ipv4)
405 catta_interface_new(m, hw, CATTA_PROTO_INET6);
410 CattaInterfaceAddress *catta_interface_address_new(CattaInterfaceMonitor *m, CattaInterface *i, const CattaAddress *addr, unsigned prefix_len) {
411 CattaInterfaceAddress *a;
416 if (!(a = catta_new(CattaInterfaceAddress, 1)))
422 a->prefix_len = prefix_len;
425 a->entry_group = NULL;
427 CATTA_LLIST_PREPEND(CattaInterfaceAddress, address, i->addresses, a);
432 void catta_interface_check_relevant(CattaInterface *i) {
434 CattaInterfaceMonitor *m;
439 b = catta_interface_is_relevant(i);
441 if (m->list_complete && b && !i->announcing) {
442 interface_mdns_mcast_join(i, 1);
444 if (i->mcast_joined) {
445 catta_log_info("New relevant interface %s.%s for mDNS.", i->hardware->name, catta_proto_to_string(i->protocol));
448 catta_announce_interface(m->server, i);
449 catta_multicast_lookup_engine_new_interface(m->server->multicast_lookup_engine, i);
452 } else if (!b && i->announcing) {
453 catta_log_info("Interface %s.%s no longer relevant for mDNS.", i->hardware->name, catta_proto_to_string(i->protocol));
455 interface_mdns_mcast_join(i, 0);
457 catta_goodbye_interface(m->server, i, 0, 1);
458 catta_querier_free_all(i);
460 catta_response_scheduler_clear(i->response_scheduler);
461 catta_query_scheduler_clear(i->query_scheduler);
462 catta_probe_scheduler_clear(i->probe_scheduler);
463 catta_cache_flush(i->cache);
468 interface_mdns_mcast_rejoin(i);
471 void catta_hw_interface_check_relevant(CattaHwInterface *hw) {
476 for (i = hw->interfaces; i; i = i->by_hardware_next)
477 catta_interface_check_relevant(i);
480 void catta_interface_monitor_check_relevant(CattaInterfaceMonitor *m) {
485 for (i = m->interfaces; i; i = i->iface_next)
486 catta_interface_check_relevant(i);
489 CattaInterfaceMonitor *catta_interface_monitor_new(CattaServer *s) {
490 CattaInterfaceMonitor *m = NULL;
492 if (!(m = catta_new0(CattaInterfaceMonitor, 1)))
493 return NULL; /* OOM */
496 m->list_complete = 0;
497 m->hashmap = catta_hashmap_new(catta_int_hash, catta_int_equal, NULL, NULL);
499 CATTA_LLIST_HEAD_INIT(CattaInterface, m->interfaces);
500 CATTA_LLIST_HEAD_INIT(CattaHwInterface, m->hw_interfaces);
502 if (catta_interface_monitor_init_osdep(m) < 0)
508 catta_interface_monitor_free(m);
512 void catta_interface_monitor_free(CattaInterfaceMonitor *m) {
515 while (m->hw_interfaces)
516 catta_hw_interface_free(m->hw_interfaces, 1);
518 assert(!m->interfaces);
520 catta_interface_monitor_free_osdep(m);
523 catta_hashmap_free(m->hashmap);
529 CattaInterface* catta_interface_monitor_get_interface(CattaInterfaceMonitor *m, CattaIfIndex idx, CattaProtocol protocol) {
530 CattaHwInterface *hw;
535 assert(protocol != CATTA_PROTO_UNSPEC);
537 if (!(hw = catta_interface_monitor_get_hw_interface(m, idx)))
540 for (i = hw->interfaces; i; i = i->by_hardware_next)
541 if (i->protocol == protocol)
547 CattaHwInterface* catta_interface_monitor_get_hw_interface(CattaInterfaceMonitor *m, CattaIfIndex idx) {
551 return catta_hashmap_lookup(m->hashmap, &idx);
554 CattaInterfaceAddress* catta_interface_monitor_get_address(CattaInterfaceMonitor *m, CattaInterface *i, const CattaAddress *raddr) {
555 CattaInterfaceAddress *ia;
561 for (ia = i->addresses; ia; ia = ia->address_next)
562 if (catta_address_cmp(&ia->address, raddr) == 0)
568 void catta_interface_send_packet_unicast(CattaInterface *i, CattaDnsPacket *p, const CattaAddress *a, uint16_t port) {
575 assert(!a || a->proto == i->protocol);
577 if (i->monitor->server->config.ratelimit_interval > 0) {
578 struct timeval now, end;
580 gettimeofday(&now, NULL);
582 end = i->hardware->ratelimit_begin;
583 catta_timeval_add(&end, i->monitor->server->config.ratelimit_interval);
585 if (i->hardware->ratelimit_begin.tv_sec <= 0 ||
586 catta_timeval_compare(&end, &now) < 0) {
588 i->hardware->ratelimit_begin = now;
589 i->hardware->ratelimit_counter = 0;
592 if (i->hardware->ratelimit_counter > i->monitor->server->config.ratelimit_burst)
595 i->hardware->ratelimit_counter++;
598 if (i->protocol == CATTA_PROTO_INET && i->monitor->server->fd_ipv4 >= 0)
599 catta_send_dns_packet_ipv4(i->monitor->server->fd_ipv4, i->hardware->index, p, i->mcast_joined ? &i->local_mcast_address.data.ipv4 : NULL, a ? &a->data.ipv4 : NULL, port);
600 else if (i->protocol == CATTA_PROTO_INET6 && i->monitor->server->fd_ipv6 >= 0)
601 catta_send_dns_packet_ipv6(i->monitor->server->fd_ipv6, i->hardware->index, p, i->mcast_joined ? &i->local_mcast_address.data.ipv6 : NULL, a ? &a->data.ipv6 : NULL, port);
604 void catta_interface_send_packet(CattaInterface *i, CattaDnsPacket *p) {
608 catta_interface_send_packet_unicast(i, p, NULL, 0);
611 int catta_interface_post_query(CattaInterface *i, CattaKey *key, int immediately, unsigned *ret_id) {
618 return catta_query_scheduler_post(i->query_scheduler, key, immediately, ret_id);
621 int catta_interface_withraw_query(CattaInterface *i, unsigned id) {
623 return catta_query_scheduler_withdraw_by_id(i->query_scheduler, id);
626 int catta_interface_post_response(CattaInterface *i, CattaRecord *record, int flush_cache, const CattaAddress *querier, int immediately) {
633 return catta_response_scheduler_post(i->response_scheduler, record, flush_cache, querier, immediately);
636 int catta_interface_post_probe(CattaInterface *i, CattaRecord *record, int immediately) {
643 return catta_probe_scheduler_post(i->probe_scheduler, record, immediately);
646 int catta_dump_caches(CattaInterfaceMonitor *m, CattaDumpCallback callback, void* userdata) {
650 for (i = m->interfaces; i; i = i->iface_next) {
651 if (catta_interface_is_relevant(i)) {
653 snprintf(ln, sizeof(ln), ";;; INTERFACE %s.%s ;;;", i->hardware->name, catta_proto_to_string(i->protocol));
654 callback(ln, userdata);
655 if (catta_cache_dump(i->cache, callback, userdata) < 0)
663 static int catta_interface_is_relevant_internal(CattaInterface *i) {
664 CattaInterfaceAddress *a;
668 if (!i->hardware->flags_ok)
671 for (a = i->addresses; a; a = a->address_next)
672 if (catta_interface_address_is_relevant(a))
678 int catta_interface_is_relevant(CattaInterface *i) {
682 for (l = i->monitor->server->config.deny_interfaces; l; l = l->next)
683 if (strcasecmp((char*) l->text, i->hardware->name) == 0)
686 if (i->monitor->server->config.allow_interfaces) {
688 for (l = i->monitor->server->config.allow_interfaces; l; l = l->next)
689 if (strcasecmp((char*) l->text, i->hardware->name) == 0)
696 return catta_interface_is_relevant_internal(i);
699 int catta_interface_address_is_relevant(CattaInterfaceAddress *a) {
700 CattaInterfaceAddress *b;
703 /* Publish public and non-deprecated IP addresses */
704 if (a->global_scope && !a->deprecated)
707 /* Publish link-local and deprecated IP addresses only if they are
708 * the only ones on the link */
709 for (b = a->iface->addresses; b; b = b->address_next) {
713 if (b->global_scope && !b->deprecated)
720 int catta_interface_match(CattaInterface *i, CattaIfIndex idx, CattaProtocol protocol) {
723 if (idx != CATTA_IF_UNSPEC && idx != i->hardware->index)
726 if (protocol != CATTA_PROTO_UNSPEC && protocol != i->protocol)
732 void catta_interface_monitor_walk(CattaInterfaceMonitor *m, CattaIfIndex iface, CattaProtocol protocol, CattaInterfaceMonitorWalkCallback callback, void* userdata) {
736 if (iface != CATTA_IF_UNSPEC) {
737 if (protocol != CATTA_PROTO_UNSPEC) {
740 if ((i = catta_interface_monitor_get_interface(m, iface, protocol)))
741 callback(m, i, userdata);
744 CattaHwInterface *hw;
747 if ((hw = catta_interface_monitor_get_hw_interface(m, iface)))
748 for (i = hw->interfaces; i; i = i->by_hardware_next)
749 if (catta_interface_match(i, iface, protocol))
750 callback(m, i, userdata);
756 for (i = m->interfaces; i; i = i->iface_next)
757 if (catta_interface_match(i, iface, protocol))
758 callback(m, i, userdata);
763 int catta_address_is_local(CattaInterfaceMonitor *m, const CattaAddress *a) {
765 CattaInterfaceAddress *ia;
769 for (i = m->interfaces; i; i = i->iface_next)
770 for (ia = i->addresses; ia; ia = ia->address_next)
771 if (catta_address_cmp(a, &ia->address) == 0)
777 int catta_interface_address_on_link(CattaInterface *i, const CattaAddress *a) {
778 CattaInterfaceAddress *ia;
783 if (a->proto != i->protocol)
786 for (ia = i->addresses; ia; ia = ia->address_next) {
788 if (a->proto == CATTA_PROTO_INET) {
791 m = ~(((uint32_t) -1) >> ia->prefix_len);
793 if ((ntohl(a->data.ipv4.address) & m) == (ntohl(ia->address.data.ipv4.address) & m))
798 assert(a->proto == CATTA_PROTO_INET6);
802 for (j = 0; j < 16; j++) {
816 if ((a->data.ipv6.address[j] & m) != (ia->address.data.ipv6.address[j] & m))
825 int catta_interface_has_address(CattaInterfaceMonitor *m, CattaIfIndex iface, const CattaAddress *a) {
827 CattaInterfaceAddress *j;
830 assert(iface != CATTA_IF_UNSPEC);
833 if (!(i = catta_interface_monitor_get_interface(m, iface, a->proto)))
836 for (j = i->addresses; j; j = j->address_next)
837 if (catta_address_cmp(a, &j->address) == 0)
843 CattaIfIndex catta_find_interface_for_address(CattaInterfaceMonitor *m, const CattaAddress *a) {
847 /* Some stupid OS don't support passing the interface index when a
848 * packet is received. We have to work around that limitation by
849 * looking for an interface that has the incoming address
850 * attached. This is sometimes ambiguous, but we have to live with
853 for (i = m->interfaces; i; i = i->iface_next) {
854 CattaInterfaceAddress *ai;
856 if (i->protocol != a->proto)
859 for (ai = i->addresses; ai; ai = ai->address_next)
860 if (catta_address_cmp(a, &ai->address) == 0)
861 return i->hardware->index;
864 return CATTA_IF_UNSPEC;