| /*** |
| This file is part of avahi. |
| |
| avahi is free software; you can redistribute it and/or modify it |
| under the terms of the GNU Lesser General Public License as |
| published by the Free Software Foundation; either version 2.1 of the |
| License, or (at your option) any later version. |
| |
| avahi is distributed in the hope that it will be useful, but WITHOUT |
| ANY WARRANTY; without even the implied warranty of MERCHANTABILITY |
| or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General |
| Public License for more details. |
| |
| You should have received a copy of the GNU Lesser General Public |
| License along with avahi; if not, write to the Free Software |
| Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 |
| USA. |
| ***/ |
| |
| #ifdef HAVE_CONFIG_H |
| #include <config.h> |
| #endif |
| |
| #include <string.h> |
| |
| #include <avahi-common/domain.h> |
| #include "avahi-common/avahi-malloc.h" |
| #include <avahi-common/error.h> |
| |
| #include "browse.h" |
| #include "log.h" |
| #include "rr.h" |
| |
| typedef struct AvahiDNSServerInfo AvahiDNSServerInfo; |
| |
| struct AvahiDNSServerInfo { |
| AvahiSDNSServerBrowser *browser; |
| |
| AvahiIfIndex interface; |
| AvahiProtocol protocol; |
| AvahiRecord *srv_record; |
| AvahiSHostNameResolver *host_name_resolver; |
| AvahiAddress address; |
| AvahiLookupResultFlags flags; |
| |
| AVAHI_LLIST_FIELDS(AvahiDNSServerInfo, info); |
| }; |
| |
| struct AvahiSDNSServerBrowser { |
| AvahiServer *server; |
| |
| AvahiSRecordBrowser *record_browser; |
| AvahiSDNSServerBrowserCallback callback; |
| void* userdata; |
| AvahiProtocol aprotocol; |
| AvahiLookupFlags user_flags; |
| |
| unsigned n_info; |
| |
| AVAHI_LLIST_FIELDS(AvahiSDNSServerBrowser, browser); |
| AVAHI_LLIST_HEAD(AvahiDNSServerInfo, info); |
| }; |
| |
| static AvahiDNSServerInfo* get_server_info(AvahiSDNSServerBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiRecord *r) { |
| AvahiDNSServerInfo *i; |
| |
| assert(b); |
| assert(r); |
| |
| for (i = b->info; i; i = i->info_next) |
| if (i->interface == interface && |
| i->protocol == protocol && |
| avahi_record_equal_no_ttl(r, i->srv_record)) |
| return i; |
| |
| return NULL; |
| } |
| |
| static void server_info_free(AvahiSDNSServerBrowser *b, AvahiDNSServerInfo *i) { |
| assert(b); |
| assert(i); |
| |
| avahi_record_unref(i->srv_record); |
| if (i->host_name_resolver) |
| avahi_s_host_name_resolver_free(i->host_name_resolver); |
| |
| AVAHI_LLIST_REMOVE(AvahiDNSServerInfo, info, b->info, i); |
| |
| assert(b->n_info >= 1); |
| b->n_info--; |
| |
| avahi_free(i); |
| } |
| |
| static void host_name_resolver_callback( |
| AvahiSHostNameResolver *r, |
| AVAHI_GCC_UNUSED AvahiIfIndex interface, |
| AVAHI_GCC_UNUSED AvahiProtocol protocol, |
| AvahiResolverEvent event, |
| const char *host_name, |
| const AvahiAddress *a, |
| AvahiLookupResultFlags flags, |
| void* userdata) { |
| |
| AvahiDNSServerInfo *i = userdata; |
| |
| assert(r); |
| assert(host_name); |
| assert(i); |
| |
| switch (event) { |
| case AVAHI_RESOLVER_FOUND: { |
| i->address = *a; |
| |
| i->browser->callback( |
| i->browser, |
| i->interface, |
| i->protocol, |
| AVAHI_BROWSER_NEW, |
| i->srv_record->data.srv.name, |
| &i->address, |
| i->srv_record->data.srv.port, |
| i->flags | flags, |
| i->browser->userdata); |
| |
| break; |
| } |
| |
| case AVAHI_RESOLVER_FAILURE: |
| /* Ignore */ |
| break; |
| } |
| |
| avahi_s_host_name_resolver_free(i->host_name_resolver); |
| i->host_name_resolver = NULL; |
| } |
| |
| static void record_browser_callback( |
| AvahiSRecordBrowser*rr, |
| AvahiIfIndex interface, |
| AvahiProtocol protocol, |
| AvahiBrowserEvent event, |
| AvahiRecord *record, |
| AvahiLookupResultFlags flags, |
| void* userdata) { |
| |
| AvahiSDNSServerBrowser *b = userdata; |
| |
| assert(rr); |
| assert(b); |
| |
| /* Filter flags */ |
| flags &= AVAHI_LOOKUP_RESULT_CACHED | AVAHI_LOOKUP_RESULT_MULTICAST | AVAHI_LOOKUP_RESULT_WIDE_AREA; |
| |
| switch (event) { |
| case AVAHI_BROWSER_NEW: { |
| AvahiDNSServerInfo *i; |
| |
| assert(record); |
| assert(record->key->type == AVAHI_DNS_TYPE_SRV); |
| |
| if (get_server_info(b, interface, protocol, record)) |
| return; |
| |
| if (b->n_info >= 10) |
| return; |
| |
| if (!(i = avahi_new(AvahiDNSServerInfo, 1))) |
| return; /* OOM */ |
| |
| i->browser = b; |
| i->interface = interface; |
| i->protocol = protocol; |
| i->srv_record = avahi_record_ref(record); |
| i->host_name_resolver = avahi_s_host_name_resolver_new( |
| b->server, |
| interface, protocol, |
| record->data.srv.name, |
| b->aprotocol, |
| b->user_flags, |
| host_name_resolver_callback, i); |
| i->flags = flags; |
| |
| AVAHI_LLIST_PREPEND(AvahiDNSServerInfo, info, b->info, i); |
| |
| b->n_info++; |
| break; |
| } |
| |
| case AVAHI_BROWSER_REMOVE: { |
| AvahiDNSServerInfo *i; |
| |
| assert(record); |
| assert(record->key->type == AVAHI_DNS_TYPE_SRV); |
| |
| if (!(i = get_server_info(b, interface, protocol, record))) |
| return; |
| |
| if (!i->host_name_resolver) |
| b->callback( |
| b, |
| interface, |
| protocol, |
| event, |
| i->srv_record->data.srv.name, |
| &i->address, |
| i->srv_record->data.srv.port, |
| i->flags | flags, |
| b->userdata); |
| |
| server_info_free(b, i); |
| break; |
| } |
| |
| case AVAHI_BROWSER_FAILURE: |
| case AVAHI_BROWSER_ALL_FOR_NOW: |
| case AVAHI_BROWSER_CACHE_EXHAUSTED: |
| |
| b->callback( |
| b, |
| interface, |
| protocol, |
| event, |
| NULL, |
| NULL, |
| 0, |
| flags, |
| b->userdata); |
| |
| break; |
| } |
| } |
| |
| AvahiSDNSServerBrowser *avahi_s_dns_server_browser_new( |
| AvahiServer *server, |
| AvahiIfIndex interface, |
| AvahiProtocol protocol, |
| const char *domain, |
| AvahiDNSServerType type, |
| AvahiProtocol aprotocol, |
| AvahiLookupFlags flags, |
| AvahiSDNSServerBrowserCallback callback, |
| void* userdata) { |
| |
| static const char * const type_table[AVAHI_DNS_SERVER_MAX] = { |
| "_domain._udp", |
| "_dns-update._udp" |
| }; |
| |
| AvahiSDNSServerBrowser *b; |
| AvahiKey *k = NULL; |
| char n[AVAHI_DOMAIN_NAME_MAX]; |
| int r; |
| |
| assert(server); |
| assert(callback); |
| |
| AVAHI_CHECK_VALIDITY_RETURN_NULL(server, AVAHI_IF_VALID(interface), AVAHI_ERR_INVALID_INTERFACE); |
| AVAHI_CHECK_VALIDITY_RETURN_NULL(server, AVAHI_PROTO_VALID(protocol), AVAHI_ERR_INVALID_PROTOCOL); |
| AVAHI_CHECK_VALIDITY_RETURN_NULL(server, AVAHI_PROTO_VALID(aprotocol), AVAHI_ERR_INVALID_PROTOCOL); |
| AVAHI_CHECK_VALIDITY_RETURN_NULL(server, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME); |
| AVAHI_CHECK_VALIDITY_RETURN_NULL(server, AVAHI_FLAGS_VALID(flags, AVAHI_LOOKUP_USE_WIDE_AREA|AVAHI_LOOKUP_USE_MULTICAST), AVAHI_ERR_INVALID_FLAGS); |
| AVAHI_CHECK_VALIDITY_RETURN_NULL(server, type < AVAHI_DNS_SERVER_MAX, AVAHI_ERR_INVALID_FLAGS); |
| |
| if (!domain) |
| domain = server->domain_name; |
| |
| if ((r = avahi_service_name_join(n, sizeof(n), NULL, type_table[type], domain)) < 0) { |
| avahi_server_set_errno(server, r); |
| return NULL; |
| } |
| |
| if (!(b = avahi_new(AvahiSDNSServerBrowser, 1))) { |
| avahi_server_set_errno(server, AVAHI_ERR_NO_MEMORY); |
| return NULL; |
| } |
| |
| b->server = server; |
| b->callback = callback; |
| b->userdata = userdata; |
| b->aprotocol = aprotocol; |
| b->n_info = 0; |
| b->user_flags = flags; |
| |
| AVAHI_LLIST_HEAD_INIT(AvahiDNSServerInfo, b->info); |
| AVAHI_LLIST_PREPEND(AvahiSDNSServerBrowser, browser, server->dns_server_browsers, b); |
| |
| if (!(k = avahi_key_new(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_SRV))) { |
| avahi_server_set_errno(server, AVAHI_ERR_NO_MEMORY); |
| goto fail; |
| } |
| |
| if (!(b->record_browser = avahi_s_record_browser_new(server, interface, protocol, k, flags, record_browser_callback, b))) |
| goto fail; |
| |
| avahi_key_unref(k); |
| |
| return b; |
| |
| fail: |
| |
| if (k) |
| avahi_key_unref(k); |
| |
| avahi_s_dns_server_browser_free(b); |
| return NULL; |
| } |
| |
| void avahi_s_dns_server_browser_free(AvahiSDNSServerBrowser *b) { |
| assert(b); |
| |
| while (b->info) |
| server_info_free(b, b->info); |
| |
| AVAHI_LLIST_REMOVE(AvahiSDNSServerBrowser, browser, b->server->dns_server_browsers, b); |
| |
| if (b->record_browser) |
| avahi_s_record_browser_free(b->record_browser); |
| |
| avahi_free(b); |
| } |
| |