2 This file is part of avahi.
4 avahi 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 avahi 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 avahi; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
31 #include <netinet/in.h>
34 #include <sys/types.h>
35 #include <sys/socket.h>
37 #include <avahi-common/simple-watch.h>
38 #include <avahi-common/malloc.h>
39 #include <avahi-common/error.h>
40 #include <avahi-common/domain.h>
41 #include <avahi-common/alternative.h>
43 #include <avahi-client/client.h>
44 #include <avahi-client/publish.h>
45 #include <avahi-client/lookup.h>
53 COMMAND_POLL_DONE = 'P',
54 COMMAND_POLL_FAILED = 'F'
59 AvahiStringList *subtypes;
63 struct _DNSServiceRef_t {
66 AvahiSimplePoll *simple_poll;
68 int thread_fd, main_fd;
73 pthread_mutex_t mutex;
76 DNSServiceBrowseReply service_browser_callback;
77 DNSServiceResolveReply service_resolver_callback;
78 DNSServiceDomainEnumReply domain_browser_callback;
79 DNSServiceRegisterReply service_register_callback;
80 DNSServiceQueryRecordReply query_resolver_callback;
83 AvahiServiceBrowser *service_browser;
84 AvahiServiceResolver *service_resolver;
85 AvahiDomainBrowser *domain_browser;
86 AvahiRecordBrowser *record_browser;
88 struct type_info type_info;
89 char *service_name, *service_name_chosen, *service_domain, *service_host;
90 uint16_t service_port;
91 AvahiIfIndex service_interface;
92 AvahiStringList *service_txt;
94 AvahiEntryGroup *entry_group;
97 #define ASSERT_SUCCESS(r) { int __ret = (r); assert(__ret == 0); }
99 static DNSServiceErrorType map_error(int error) {
102 return kDNSServiceErr_NoError;
104 case AVAHI_ERR_BAD_STATE :
105 return kDNSServiceErr_BadState;
107 case AVAHI_ERR_INVALID_HOST_NAME:
108 case AVAHI_ERR_INVALID_DOMAIN_NAME:
109 case AVAHI_ERR_INVALID_TTL:
110 case AVAHI_ERR_IS_PATTERN:
111 case AVAHI_ERR_INVALID_RECORD:
112 case AVAHI_ERR_INVALID_SERVICE_NAME:
113 case AVAHI_ERR_INVALID_SERVICE_TYPE:
114 case AVAHI_ERR_INVALID_PORT:
115 case AVAHI_ERR_INVALID_KEY:
116 case AVAHI_ERR_INVALID_ADDRESS:
117 case AVAHI_ERR_INVALID_SERVICE_SUBTYPE:
118 return kDNSServiceErr_BadParam;
121 case AVAHI_ERR_COLLISION:
122 return kDNSServiceErr_NameConflict;
124 case AVAHI_ERR_TOO_MANY_CLIENTS:
125 case AVAHI_ERR_TOO_MANY_OBJECTS:
126 case AVAHI_ERR_TOO_MANY_ENTRIES:
127 case AVAHI_ERR_ACCESS_DENIED:
128 return kDNSServiceErr_Refused;
130 case AVAHI_ERR_INVALID_OPERATION:
131 case AVAHI_ERR_INVALID_OBJECT:
132 return kDNSServiceErr_Invalid;
134 case AVAHI_ERR_NO_MEMORY:
135 return kDNSServiceErr_NoMemory;
137 case AVAHI_ERR_INVALID_INTERFACE:
138 case AVAHI_ERR_INVALID_PROTOCOL:
139 return kDNSServiceErr_BadInterfaceIndex;
141 case AVAHI_ERR_INVALID_FLAGS:
142 return kDNSServiceErr_BadFlags;
144 case AVAHI_ERR_NOT_FOUND:
145 return kDNSServiceErr_NoSuchName;
147 case AVAHI_ERR_VERSION_MISMATCH:
148 return kDNSServiceErr_Incompatible;
150 case AVAHI_ERR_NO_NETWORK:
152 case AVAHI_ERR_INVALID_CONFIG:
153 case AVAHI_ERR_TIMEOUT:
154 case AVAHI_ERR_DBUS_ERROR:
155 case AVAHI_ERR_DISCONNECTED:
156 case AVAHI_ERR_NO_DAEMON:
161 return kDNSServiceErr_Unknown;
164 static void type_info_init(struct type_info *i) {
171 static void type_info_free(struct type_info *i) {
175 avahi_string_list_free(i->subtypes);
180 static int type_info_parse(struct type_info *i, const char *t) {
199 token = avahi_strndup(t, l);
205 /* This is the first token, hence the main type */
207 if (!avahi_is_valid_service_type_strict(token))
215 /* This is not the first token, hence a subtype */
217 if (!(fst = avahi_strdup_printf("%s._sub.%s", token, i->type)))
220 if (!avahi_is_valid_service_subtype(fst)) {
225 i->subtypes = avahi_string_list_add(i->subtypes, fst);
249 static const char *add_trailing_dot(const char *s, char *buf, size_t buf_len) {
256 if (s[strlen(s)-1] == '.')
259 snprintf(buf, buf_len, "%s.", s);
263 static int read_command(int fd) {
269 if ((r = read(fd, &command, 1)) != 1) {
270 fprintf(stderr, __FILE__": read() failed: %s\n", r < 0 ? strerror(errno) : "EOF");
277 static int write_command(int fd, char reply) {
280 if (write(fd, &reply, 1) != 1) {
281 fprintf(stderr, __FILE__": write() failed: %s\n", strerror(errno));
288 static int poll_func(struct pollfd *ufds, unsigned int nfds, int timeout, void *userdata) {
289 DNSServiceRef sdref = userdata;
294 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
296 /* fprintf(stderr, "pre-syscall\n"); */
297 ret = poll(ufds, nfds, timeout);
298 /* fprintf(stderr, "post-syscall\n"); */
300 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
305 static void * thread_func(void *data) {
306 DNSServiceRef sdref = data;
310 pthread_sigmask(SIG_BLOCK, &mask, NULL);
312 sdref->thread = pthread_self();
313 sdref->thread_running = 1;
318 if ((command = read_command(sdref->thread_fd)) < 0)
321 /* fprintf(stderr, "Command: %c\n", command); */
328 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
333 if ((ret = avahi_simple_poll_run(sdref->simple_poll)) < 0) {
338 fprintf(stderr, __FILE__": avahi_simple_poll_run() failed: %s\n", strerror(errno));
344 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
346 if (write_command(sdref->thread_fd, ret < 0 ? COMMAND_POLL_FAILED : COMMAND_POLL_DONE) < 0)
361 static DNSServiceRef sdref_new(void) {
362 int fd[2] = { -1, -1 };
363 DNSServiceRef sdref = NULL;
364 pthread_mutexattr_t mutex_attr;
366 if (socketpair(AF_UNIX, SOCK_STREAM, 0, fd) < 0)
369 if (!(sdref = avahi_new(struct _DNSServiceRef_t, 1)))
373 sdref->thread_fd = fd[0];
374 sdref->main_fd = fd[1];
376 sdref->client = NULL;
377 sdref->service_browser = NULL;
378 sdref->service_resolver = NULL;
379 sdref->domain_browser = NULL;
380 sdref->entry_group = NULL;
382 sdref->service_name = sdref->service_name_chosen = sdref->service_domain = sdref->service_host = NULL;
383 sdref->service_txt = NULL;
385 type_info_init(&sdref->type_info);
387 ASSERT_SUCCESS(pthread_mutexattr_init(&mutex_attr));
388 pthread_mutexattr_settype(&mutex_attr, PTHREAD_MUTEX_RECURSIVE);
389 ASSERT_SUCCESS(pthread_mutex_init(&sdref->mutex, &mutex_attr));
391 sdref->thread_running = 0;
393 if (!(sdref->simple_poll = avahi_simple_poll_new()))
396 avahi_simple_poll_set_func(sdref->simple_poll, poll_func, sdref);
398 /* Start simple poll */
399 if (avahi_simple_poll_prepare(sdref->simple_poll, -1) < 0)
402 /* Queue an initial POLL command for the thread */
403 if (write_command(sdref->main_fd, COMMAND_POLL) < 0)
406 if (pthread_create(&sdref->thread, NULL, thread_func, sdref) != 0)
409 sdref->thread_running = 1;
416 DNSServiceRefDeallocate(sdref);
421 static void sdref_free(DNSServiceRef sdref) {
424 if (sdref->thread_running) {
425 ASSERT_SUCCESS(write_command(sdref->main_fd, COMMAND_QUIT));
426 avahi_simple_poll_wakeup(sdref->simple_poll);
427 ASSERT_SUCCESS(pthread_join(sdref->thread, NULL));
431 avahi_client_free(sdref->client);
433 if (sdref->simple_poll)
434 avahi_simple_poll_free(sdref->simple_poll);
436 if (sdref->thread_fd >= 0)
437 close(sdref->thread_fd);
439 if (sdref->main_fd >= 0)
440 close(sdref->main_fd);
442 ASSERT_SUCCESS(pthread_mutex_destroy(&sdref->mutex));
444 avahi_free(sdref->service_name);
445 avahi_free(sdref->service_name_chosen);
446 avahi_free(sdref->service_domain);
447 avahi_free(sdref->service_host);
449 type_info_free(&sdref->type_info);
451 avahi_string_list_free(sdref->service_txt);
456 static void sdref_ref(DNSServiceRef sdref) {
458 assert(sdref->n_ref >= 1);
463 static void sdref_unref(DNSServiceRef sdref) {
465 assert(sdref->n_ref >= 1);
467 if (--(sdref->n_ref) <= 0)
471 int DNSSD_API DNSServiceRefSockFD(DNSServiceRef sdref) {
475 if (!sdref || sdref->n_ref <= 0)
478 return sdref->main_fd;
481 DNSServiceErrorType DNSSD_API DNSServiceProcessResult(DNSServiceRef sdref) {
482 DNSServiceErrorType ret = kDNSServiceErr_Unknown;
486 if (!sdref || sdref->n_ref <= 0)
487 return kDNSServiceErr_BadParam;
491 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
493 /* Cleanup notification socket */
494 if (read_command(sdref->main_fd) != COMMAND_POLL_DONE)
497 if (avahi_simple_poll_dispatch(sdref->simple_poll) < 0)
500 if (sdref->n_ref > 1) /* Perhaps we should die */
502 /* Dispatch events */
503 if (avahi_simple_poll_prepare(sdref->simple_poll, -1) < 0)
506 if (sdref->n_ref > 1)
508 /* Request the poll */
509 if (write_command(sdref->main_fd, COMMAND_POLL) < 0)
512 ret = kDNSServiceErr_NoError;
516 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
523 void DNSSD_API DNSServiceRefDeallocate(DNSServiceRef sdref) {
530 static void service_browser_callback(
531 AvahiServiceBrowser *b,
532 AvahiIfIndex interface,
533 AVAHI_GCC_UNUSED AvahiProtocol protocol,
534 AvahiBrowserEvent event,
538 AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
541 DNSServiceRef sdref = userdata;
542 char type_fixed[AVAHI_DOMAIN_NAME_MAX], domain_fixed[AVAHI_DOMAIN_NAME_MAX];
545 assert(sdref->n_ref >= 1);
547 type = add_trailing_dot(type, type_fixed, sizeof(type_fixed));
548 domain = add_trailing_dot(domain, domain_fixed, sizeof(domain_fixed));
551 case AVAHI_BROWSER_NEW:
552 sdref->service_browser_callback(sdref, kDNSServiceFlagsAdd, interface, kDNSServiceErr_NoError, name, type, domain, sdref->context);
555 case AVAHI_BROWSER_REMOVE:
556 sdref->service_browser_callback(sdref, 0, interface, kDNSServiceErr_NoError, name, type, domain, sdref->context);
559 case AVAHI_BROWSER_FAILURE:
560 sdref->service_browser_callback(sdref, 0, interface, map_error(avahi_client_errno(sdref->client)), NULL, NULL, NULL, sdref->context);
563 case AVAHI_BROWSER_CACHE_EXHAUSTED:
564 case AVAHI_BROWSER_ALL_FOR_NOW:
569 static void generic_client_callback(AvahiClient *s, AvahiClientState state, void* userdata) {
570 DNSServiceRef sdref = userdata;
571 int error = kDNSServiceErr_Unknown;
575 assert(sdref->n_ref >= 1);
579 case AVAHI_CLIENT_FAILURE:
581 if (sdref->service_browser_callback)
582 sdref->service_browser_callback(sdref, 0, 0, error, NULL, NULL, NULL, sdref->context);
583 else if (sdref->service_resolver_callback)
584 sdref->service_resolver_callback(sdref, 0, 0, error, NULL, NULL, 0, 0, NULL, sdref->context);
585 else if (sdref->domain_browser_callback)
586 sdref->domain_browser_callback(sdref, 0, 0, error, NULL, sdref->context);
587 else if (sdref->query_resolver_callback)
588 sdref->query_resolver_callback(sdref, 0, 0, error, NULL, 0, 0, 0, NULL, 0, sdref->context);
592 case AVAHI_CLIENT_S_RUNNING:
593 case AVAHI_CLIENT_S_COLLISION:
594 case AVAHI_CLIENT_S_REGISTERING:
595 case AVAHI_CLIENT_CONNECTING:
600 DNSServiceErrorType DNSSD_API DNSServiceBrowse(
601 DNSServiceRef *ret_sdref,
602 DNSServiceFlags flags,
606 DNSServiceBrowseReply callback,
609 DNSServiceErrorType ret = kDNSServiceErr_Unknown;
611 DNSServiceRef sdref = NULL;
612 AvahiIfIndex ifindex;
613 struct type_info type_info;
617 if (!ret_sdref || !regtype)
618 return kDNSServiceErr_BadParam;
621 if (interface == kDNSServiceInterfaceIndexLocalOnly || flags != 0) {
622 AVAHI_WARN_UNSUPPORTED;
623 return kDNSServiceErr_Unsupported;
626 type_info_init(&type_info);
628 if (type_info_parse(&type_info, regtype) < 0 || type_info.n_subtypes > 1) {
629 type_info_free(&type_info);
631 if (!avahi_is_valid_service_type_generic(regtype))
632 return kDNSServiceErr_Unsupported;
634 regtype = type_info.subtypes ? (char*) type_info.subtypes->text : type_info.type;
636 if (!(sdref = sdref_new())) {
637 type_info_free(&type_info);
638 return kDNSServiceErr_Unknown;
641 sdref->context = context;
642 sdref->service_browser_callback = callback;
644 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
646 if (!(sdref->client = avahi_client_new(avahi_simple_poll_get(sdref->simple_poll), 0, generic_client_callback, sdref, &error))) {
647 ret = map_error(error);
651 ifindex = interface == kDNSServiceInterfaceIndexAny ? AVAHI_IF_UNSPEC : (AvahiIfIndex) interface;
653 if (!(sdref->service_browser = avahi_service_browser_new(sdref->client, ifindex, AVAHI_PROTO_UNSPEC, regtype, domain, 0, service_browser_callback, sdref))) {
654 ret = map_error(avahi_client_errno(sdref->client));
658 ret = kDNSServiceErr_NoError;
663 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
665 if (ret != kDNSServiceErr_NoError)
666 DNSServiceRefDeallocate(sdref);
668 type_info_free(&type_info);
673 static void service_resolver_callback(
674 AvahiServiceResolver *r,
675 AvahiIfIndex interface,
676 AVAHI_GCC_UNUSED AvahiProtocol protocol,
677 AvahiResolverEvent event,
681 const char *host_name,
682 AVAHI_GCC_UNUSED const AvahiAddress *a,
684 AvahiStringList *txt,
685 AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
688 DNSServiceRef sdref = userdata;
692 assert(sdref->n_ref >= 1);
695 case AVAHI_RESOLVER_FOUND: {
697 char host_name_fixed[AVAHI_DOMAIN_NAME_MAX];
698 char full_name[AVAHI_DOMAIN_NAME_MAX];
703 host_name = add_trailing_dot(host_name, host_name_fixed, sizeof(host_name_fixed));
705 if ((p = avahi_new0(char, (l = avahi_string_list_serialize(txt, NULL, 0))+1)))
706 avahi_string_list_serialize(txt, p, l);
708 ret = avahi_service_name_join(full_name, sizeof(full_name), name, type, domain);
709 assert(ret == AVAHI_OK);
711 strcat(full_name, ".");
713 sdref->service_resolver_callback(sdref, 0, interface, kDNSServiceErr_NoError, full_name, host_name, htons(port), l, (unsigned char*) p, sdref->context);
719 case AVAHI_RESOLVER_FAILURE:
720 sdref->service_resolver_callback(sdref, 0, interface, map_error(avahi_client_errno(sdref->client)), NULL, NULL, 0, 0, NULL, sdref->context);
725 DNSServiceErrorType DNSSD_API DNSServiceResolve(
726 DNSServiceRef *ret_sdref,
727 DNSServiceFlags flags,
732 DNSServiceResolveReply callback,
735 DNSServiceErrorType ret = kDNSServiceErr_Unknown;
737 DNSServiceRef sdref = NULL;
738 AvahiIfIndex ifindex;
742 if (!ret_sdref || !name || !regtype || !domain || !callback)
743 return kDNSServiceErr_BadParam;
746 if (interface == kDNSServiceInterfaceIndexLocalOnly || flags != 0) {
747 AVAHI_WARN_UNSUPPORTED;
748 return kDNSServiceErr_Unsupported;
751 if (!(sdref = sdref_new()))
752 return kDNSServiceErr_Unknown;
754 sdref->context = context;
755 sdref->service_resolver_callback = callback;
757 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
759 if (!(sdref->client = avahi_client_new(avahi_simple_poll_get(sdref->simple_poll), 0, generic_client_callback, sdref, &error))) {
760 ret = map_error(error);
764 ifindex = interface == kDNSServiceInterfaceIndexAny ? AVAHI_IF_UNSPEC : (AvahiIfIndex) interface;
766 if (!(sdref->service_resolver = avahi_service_resolver_new(sdref->client, ifindex, AVAHI_PROTO_UNSPEC, name, regtype, domain, AVAHI_PROTO_UNSPEC, 0, service_resolver_callback, sdref))) {
767 ret = map_error(avahi_client_errno(sdref->client));
772 ret = kDNSServiceErr_NoError;
777 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
779 if (ret != kDNSServiceErr_NoError)
780 DNSServiceRefDeallocate(sdref);
785 int DNSSD_API DNSServiceConstructFullName (
789 const char *domain) {
793 if (!fullName || !regtype || !domain)
796 if (avahi_service_name_join(fullName, kDNSServiceMaxDomainName, service, regtype, domain) < 0)
802 static void domain_browser_callback(
803 AvahiDomainBrowser *b,
804 AvahiIfIndex interface,
805 AVAHI_GCC_UNUSED AvahiProtocol protocol,
806 AvahiBrowserEvent event,
808 AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
811 DNSServiceRef sdref = userdata;
812 static char domain_fixed[AVAHI_DOMAIN_NAME_MAX];
816 assert(sdref->n_ref >= 1);
818 domain = add_trailing_dot(domain, domain_fixed, sizeof(domain_fixed));
821 case AVAHI_BROWSER_NEW:
822 sdref->domain_browser_callback(sdref, kDNSServiceFlagsAdd, interface, kDNSServiceErr_NoError, domain, sdref->context);
825 case AVAHI_BROWSER_REMOVE:
826 sdref->domain_browser_callback(sdref, 0, interface, kDNSServiceErr_NoError, domain, sdref->context);
829 case AVAHI_BROWSER_FAILURE:
830 sdref->domain_browser_callback(sdref, 0, interface, map_error(avahi_client_errno(sdref->client)), domain, sdref->context);
833 case AVAHI_BROWSER_CACHE_EXHAUSTED:
834 case AVAHI_BROWSER_ALL_FOR_NOW:
839 DNSServiceErrorType DNSSD_API DNSServiceEnumerateDomains(
840 DNSServiceRef *ret_sdref,
841 DNSServiceFlags flags,
843 DNSServiceDomainEnumReply callback,
846 DNSServiceErrorType ret = kDNSServiceErr_Unknown;
848 DNSServiceRef sdref = NULL;
849 AvahiIfIndex ifindex;
853 if (!ret_sdref || !callback)
854 return kDNSServiceErr_BadParam;
857 if (interface == kDNSServiceInterfaceIndexLocalOnly ||
858 (flags != kDNSServiceFlagsBrowseDomains && flags != kDNSServiceFlagsRegistrationDomains)) {
859 AVAHI_WARN_UNSUPPORTED;
860 return kDNSServiceErr_Unsupported;
863 if (!(sdref = sdref_new()))
864 return kDNSServiceErr_Unknown;
866 sdref->context = context;
867 sdref->domain_browser_callback = callback;
869 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
871 if (!(sdref->client = avahi_client_new(avahi_simple_poll_get(sdref->simple_poll), 0, generic_client_callback, sdref, &error))) {
872 ret = map_error(error);
876 ifindex = interface == kDNSServiceInterfaceIndexAny ? AVAHI_IF_UNSPEC : (AvahiIfIndex) interface;
878 if (!(sdref->domain_browser = avahi_domain_browser_new(sdref->client, ifindex, AVAHI_PROTO_UNSPEC, "local",
879 flags == kDNSServiceFlagsRegistrationDomains ? AVAHI_DOMAIN_BROWSER_REGISTER : AVAHI_DOMAIN_BROWSER_BROWSE,
880 0, domain_browser_callback, sdref))) {
881 ret = map_error(avahi_client_errno(sdref->client));
885 ret = kDNSServiceErr_NoError;
890 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
892 if (ret != kDNSServiceErr_NoError)
893 DNSServiceRefDeallocate(sdref);
898 static void reg_report_error(DNSServiceRef sdref, DNSServiceErrorType error) {
899 char regtype_fixed[AVAHI_DOMAIN_NAME_MAX], domain_fixed[AVAHI_DOMAIN_NAME_MAX];
900 const char *regtype, *domain;
902 assert(sdref->n_ref >= 1);
904 if (!sdref->service_register_callback)
907 regtype = add_trailing_dot(sdref->type_info.type, regtype_fixed, sizeof(regtype_fixed));
908 domain = add_trailing_dot(sdref->service_domain, domain_fixed, sizeof(domain_fixed));
910 sdref->service_register_callback(
912 sdref->service_name_chosen ? sdref->service_name_chosen : sdref->service_name,
918 static int reg_create_service(DNSServiceRef sdref) {
923 assert(sdref->n_ref >= 1);
925 if ((ret = avahi_entry_group_add_service_strlst(
927 sdref->service_interface,
930 sdref->service_name_chosen,
931 sdref->type_info.type,
932 sdref->service_domain,
935 sdref->service_txt)) < 0)
938 for (l = sdref->type_info.subtypes; l; l = l->next) {
939 /* Create a subtype entry */
941 if (avahi_entry_group_add_service_subtype(
943 sdref->service_interface,
946 sdref->service_name_chosen,
947 sdref->type_info.type,
948 sdref->service_domain,
949 (const char*) l->text) < 0)
953 if ((ret = avahi_entry_group_commit(sdref->entry_group)) < 0)
959 static void reg_client_callback(AvahiClient *s, AvahiClientState state, void* userdata) {
960 DNSServiceRef sdref = userdata;
964 assert(sdref->n_ref >= 1);
966 /* We've not been setup completely */
967 if (!sdref->entry_group)
971 case AVAHI_CLIENT_FAILURE:
972 reg_report_error(sdref, kDNSServiceErr_Unknown);
975 case AVAHI_CLIENT_S_RUNNING: {
978 if (!sdref->service_name) {
980 /* If the service name is taken from the host name, copy that */
982 avahi_free(sdref->service_name_chosen);
983 sdref->service_name_chosen = NULL;
985 if (!(n = avahi_client_get_host_name(sdref->client))) {
986 reg_report_error(sdref, map_error(avahi_client_errno(sdref->client)));
990 if (!(sdref->service_name_chosen = avahi_strdup(n))) {
991 reg_report_error(sdref, kDNSServiceErr_NoMemory);
996 if (!sdref->service_name_chosen) {
998 assert(sdref->service_name);
1000 if (!(sdref->service_name_chosen = avahi_strdup(sdref->service_name))) {
1001 reg_report_error(sdref, kDNSServiceErr_NoMemory);
1006 /* Register the service */
1008 if ((ret = reg_create_service(sdref)) < 0) {
1009 reg_report_error(sdref, map_error(ret));
1016 case AVAHI_CLIENT_S_COLLISION:
1017 case AVAHI_CLIENT_S_REGISTERING:
1019 /* Remove our entry */
1020 avahi_entry_group_reset(sdref->entry_group);
1024 case AVAHI_CLIENT_CONNECTING:
1031 static void reg_entry_group_callback(AvahiEntryGroup *g, AvahiEntryGroupState state, void *userdata) {
1032 DNSServiceRef sdref = userdata;
1037 case AVAHI_ENTRY_GROUP_ESTABLISHED:
1039 /* Inform the user */
1040 reg_report_error(sdref, kDNSServiceErr_NoError);
1044 case AVAHI_ENTRY_GROUP_COLLISION: {
1048 /* Remove our entry */
1049 avahi_entry_group_reset(sdref->entry_group);
1051 assert(sdref->service_name_chosen);
1053 /* Pick a new name */
1054 if (!(n = avahi_alternative_service_name(sdref->service_name_chosen))) {
1055 reg_report_error(sdref, kDNSServiceErr_NoMemory);
1058 avahi_free(sdref->service_name_chosen);
1059 sdref->service_name_chosen = n;
1061 /* Register the service with that new name */
1062 if ((ret = reg_create_service(sdref)) < 0) {
1063 reg_report_error(sdref, map_error(ret));
1070 case AVAHI_ENTRY_GROUP_REGISTERING:
1071 case AVAHI_ENTRY_GROUP_UNCOMMITED:
1075 case AVAHI_ENTRY_GROUP_FAILURE:
1076 /* Inform the user */
1077 reg_report_error(sdref, map_error(avahi_client_errno(sdref->client)));
1083 DNSServiceErrorType DNSSD_API DNSServiceRegister (
1084 DNSServiceRef *ret_sdref,
1085 DNSServiceFlags flags,
1088 const char *regtype,
1093 const void *txtRecord,
1094 DNSServiceRegisterReply callback,
1097 DNSServiceErrorType ret = kDNSServiceErr_Unknown;
1099 DNSServiceRef sdref = NULL;
1100 AvahiStringList *txt = NULL;
1101 struct type_info type_info;
1105 if (!ret_sdref || !regtype)
1106 return kDNSServiceErr_BadParam;
1114 if (interface == kDNSServiceInterfaceIndexLocalOnly || flags) {
1115 AVAHI_WARN_UNSUPPORTED;
1116 return kDNSServiceErr_Unsupported;
1120 if (avahi_string_list_parse(txtRecord, txtLen, &txt) < 0)
1121 return kDNSServiceErr_Invalid;
1123 if (type_info_parse(&type_info, regtype) < 0) {
1124 avahi_string_list_free(txt);
1125 return kDNSServiceErr_Invalid;
1128 if (!(sdref = sdref_new())) {
1129 avahi_string_list_free(txt);
1130 type_info_free(&type_info);
1131 return kDNSServiceErr_Unknown;
1134 sdref->context = context;
1135 sdref->service_register_callback = callback;
1137 sdref->type_info = type_info;
1138 sdref->service_name = avahi_strdup(name);
1139 sdref->service_domain = domain ? avahi_normalize_name_strdup(domain) : NULL;
1140 sdref->service_host = host ? avahi_normalize_name_strdup(host) : NULL;
1141 sdref->service_interface = interface == kDNSServiceInterfaceIndexAny ? AVAHI_IF_UNSPEC : (AvahiIfIndex) interface;
1142 sdref->service_port = ntohs(port);
1143 sdref->service_txt = txt;
1145 /* Some OOM checking would be cool here */
1147 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
1149 if (!(sdref->client = avahi_client_new(avahi_simple_poll_get(sdref->simple_poll), 0, reg_client_callback, sdref, &error))) {
1150 ret = map_error(error);
1154 if (!sdref->service_domain) {
1157 if (!(d = avahi_client_get_domain_name(sdref->client))) {
1158 ret = map_error(avahi_client_errno(sdref->client));
1162 if (!(sdref->service_domain = avahi_strdup(d))) {
1163 ret = kDNSServiceErr_NoMemory;
1168 if (!(sdref->entry_group = avahi_entry_group_new(sdref->client, reg_entry_group_callback, sdref))) {
1169 ret = map_error(avahi_client_errno(sdref->client));
1173 if (avahi_client_get_state(sdref->client) == AVAHI_CLIENT_S_RUNNING) {
1176 if (sdref->service_name)
1177 n = sdref->service_name;
1179 if (!(n = avahi_client_get_host_name(sdref->client))) {
1180 ret = map_error(avahi_client_errno(sdref->client));
1185 if (!(sdref->service_name_chosen = avahi_strdup(n))) {
1186 ret = kDNSServiceErr_NoMemory;
1191 if ((error = reg_create_service(sdref)) < 0) {
1192 ret = map_error(error);
1197 ret = kDNSServiceErr_NoError;
1202 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
1204 if (ret != kDNSServiceErr_NoError)
1205 DNSServiceRefDeallocate(sdref);
1210 DNSServiceErrorType DNSSD_API DNSServiceUpdateRecord(
1211 DNSServiceRef sdref,
1213 DNSServiceFlags flags,
1216 AVAHI_GCC_UNUSED uint32_t ttl) {
1218 int ret = kDNSServiceErr_Unknown;
1219 AvahiStringList *txt = NULL;
1223 if (!sdref || sdref->n_ref <= 0)
1224 return kDNSServiceErr_BadParam;
1226 if (flags || rref) {
1227 AVAHI_WARN_UNSUPPORTED;
1228 return kDNSServiceErr_Unsupported;
1232 if (avahi_string_list_parse(rdata, rdlen, &txt) < 0)
1233 return kDNSServiceErr_Invalid;
1235 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
1237 if (!avahi_string_list_equal(txt, sdref->service_txt)) {
1239 avahi_string_list_free(sdref->service_txt);
1240 sdref->service_txt = txt;
1242 if (avahi_client_get_state(sdref->client) == AVAHI_CLIENT_S_RUNNING &&
1243 sdref->entry_group &&
1244 (avahi_entry_group_get_state(sdref->entry_group) == AVAHI_ENTRY_GROUP_ESTABLISHED ||
1245 avahi_entry_group_get_state(sdref->entry_group) == AVAHI_ENTRY_GROUP_REGISTERING))
1247 if (avahi_entry_group_update_service_txt_strlst(
1249 sdref->service_interface,
1252 sdref->service_name_chosen,
1253 sdref->type_info.type,
1254 sdref->service_domain,
1255 sdref->service_txt) < 0) {
1257 ret = map_error(avahi_client_errno(sdref->client));
1262 avahi_string_list_free(txt);
1264 ret = kDNSServiceErr_NoError;
1267 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
1272 static void query_resolver_callback(
1273 AvahiRecordBrowser *r,
1274 AvahiIfIndex interface,
1275 AVAHI_GCC_UNUSED AvahiProtocol protocol,
1276 AvahiBrowserEvent event,
1282 AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
1285 DNSServiceRef sdref = userdata;
1289 assert(sdref->n_ref >= 1);
1293 case AVAHI_BROWSER_NEW:
1294 case AVAHI_BROWSER_REMOVE: {
1296 DNSServiceFlags qflags = 0;
1297 if (event == AVAHI_BROWSER_NEW)
1298 qflags |= kDNSServiceFlagsAdd;
1300 sdref->query_resolver_callback(sdref, qflags, interface, kDNSServiceErr_NoError, name, type, clazz, size, rdata, 0, sdref->context);
1304 case AVAHI_BROWSER_ALL_FOR_NOW:
1305 case AVAHI_BROWSER_CACHE_EXHAUSTED:
1306 /* not implemented */
1309 case AVAHI_BROWSER_FAILURE:
1310 sdref->query_resolver_callback(sdref, 0, interface, map_error(avahi_client_errno(sdref->client)), NULL, 0, 0, 0, NULL, 0, sdref->context);
1315 DNSServiceErrorType DNSSD_API DNSServiceQueryRecord (
1316 DNSServiceRef *ret_sdref,
1317 DNSServiceFlags flags,
1319 const char *fullname,
1322 DNSServiceQueryRecordReply callback,
1325 DNSServiceErrorType ret = kDNSServiceErr_Unknown;
1327 DNSServiceRef sdref = NULL;
1328 AvahiIfIndex ifindex;
1332 if (!ret_sdref || !fullname)
1333 return kDNSServiceErr_BadParam;
1336 if (interface == kDNSServiceInterfaceIndexLocalOnly || flags != 0) {
1337 AVAHI_WARN_UNSUPPORTED;
1338 return kDNSServiceErr_Unsupported;
1341 if (!(sdref = sdref_new()))
1342 return kDNSServiceErr_Unknown;
1344 sdref->context = context;
1345 sdref->query_resolver_callback = callback;
1347 ASSERT_SUCCESS(pthread_mutex_lock(&sdref->mutex));
1349 if (!(sdref->client = avahi_client_new(avahi_simple_poll_get(sdref->simple_poll), 0, generic_client_callback, sdref, &error))) {
1350 ret = map_error(error);
1354 ifindex = interface == kDNSServiceInterfaceIndexAny ? AVAHI_IF_UNSPEC : (AvahiIfIndex) interface;
1356 if (!(sdref->record_browser = avahi_record_browser_new(sdref->client, ifindex, AVAHI_PROTO_UNSPEC, fullname, clazz, type, 0, query_resolver_callback, sdref))) {
1357 ret = map_error(avahi_client_errno(sdref->client));
1361 ret = kDNSServiceErr_NoError;
1366 ASSERT_SUCCESS(pthread_mutex_unlock(&sdref->mutex));
1368 if (ret != kDNSServiceErr_NoError)
1369 DNSServiceRefDeallocate(sdref);