patches/avahi-10-resolve-address.diff
author yippi
Mon, 27 Sep 2010 21:07:51 +0000
changeset 20108 51df67ca9307
parent 10044 7508208432e4
child 21365 ead32bf37db7
permissions -rw-r--r--
I had these modules listed as being owned by me, but they are really owned by wangke, correcting.

--- /usr/tmp/clean/avahi-0.6.13/avahi-core/resolve-address.c	2006-03-02 00:59:13.000000000 +0000
+++ avahi-0.6.13/avahi-core/resolve-address.c	2006-08-31 14:00:21.756949000 +0100
@@ -24,6 +24,7 @@
 #endif
 
 #include <stdlib.h>
+#include <stdio.h>
 
 #include <avahi-common/timeval.h>
 #include <avahi-common/malloc.h>
@@ -53,6 +54,13 @@
 
     AvahiTimeEvent *time_event;
 
+#ifdef HAVE_BONJOUR
+    AvahiTimeEvent *defer_time_event;
+    AvahiLookupFlags lookup_flags;
+    AvahiWatch *watch;
+    DNSServiceRef client;
+#endif
+    
     AVAHI_LLIST_FIELDS(AvahiSAddressResolver, resolver);
 };
 
@@ -76,6 +84,99 @@
     }
 }
 
+#ifdef HAVE_BONJOUR
+static void resolve_socket_event(AvahiWatch *w, int fd, AvahiWatchEvent events,
+void *userdata) {
+    AvahiSAddressResolver *r = userdata;
+    DNSServiceErrorType ret;
+
+    assert(w);
+    assert(fd >= 0);
+    assert(events & AVAHI_WATCH_IN);
+
+    assert (fd == DNSServiceRefSockFD(r->client));
+    
+    ret = DNSServiceProcessResult(r->client);
+    if (ret != kDNSServiceErr_NoError) {
+        if (r->watch) {
+            r->server->poll_api->watch_free(r->watch);
+            r->watch = NULL;
+        }
+        DNSServiceRefDeallocate(r->client);
+        r->client = NULL;
+        avahi_server_set_errno(r->server, AVAHI_ERR_DISCONNECTED);
+        finish(r, AVAHI_RESOLVER_FAILURE);
+    }
+}
+
+static void resolve_reply(DNSServiceRef client, DNSServiceFlags flags, uint32_t
+IfIndex, DNSServiceErrorType errorCode, const char *fullname, uint16_t rrtype, uint16_t rrclass, uint16_t rdlen, const void* rdata, uint32_t ttl, void *context) {
+    AvahiSAddressResolver *r = context;
+
+    assert(rrtype == kDNSServiceType_PTR);
+    if (r->interface > 0  && IfIndex != r->interface)
+        return;
+    if (r->interface <= 0)
+       r->interface = IfIndex;
+
+    /*
+     * Using Bonjour we cannot determine whether result was obtained from
+     * multicast ot unicast query
+     */
+    r->flags = 0;
+
+    if (!(r->ptr_record = avahi_record_new(r->key, ttl))) {
+        avahi_server_set_errno(r->server, AVAHI_ERR_NO_MEMORY);
+        finish(r, AVAHI_RESOLVER_FAILURE);
+        return;
+    }
+    if (avahi_rdata_parse(r->ptr_record, rdata, rdlen) != 0) {
+        avahi_server_set_errno(r->server, AVAHI_ERR_INVALID_PACKET);
+        finish(r, AVAHI_RESOLVER_FAILURE);
+        return;
+    }
+    finish(r, AVAHI_RESOLVER_FOUND);
+}
+
+static void resolve_error_callback(AvahiTimeEvent *e, void *userdata) {
+    AvahiSAddressResolver *r = userdata;
+
+    if (r->defer_time_event) {
+        avahi_time_event_free(r->defer_time_event);
+        r->defer_time_event = NULL;
+    }
+    avahi_server_set_errno(r->server, AVAHI_ERR_FAILURE);
+    finish(r, AVAHI_RESOLVER_FAILURE);
+}
+
+static void avahi_resolve_address_start(AvahiSAddressResolver *r)
+{
+    DNSServiceErrorType ret;
+    DNSServiceFlags flags;
+
+    if (r->flags != AVAHI_LOOKUP_USE_WIDE_AREA)
+        flags = kDNSServiceFlagsForceMulticast;
+    else
+        flags = 0;
+
+    ret = DNSServiceQueryRecord(&r->client,
+                                flags,
+                                r->interface == AVAHI_IF_UNSPEC ?
+                                    kDNSServiceInterfaceIndexAny :
+                                    r->interface,
+                                r->key->name,
+                                kDNSServiceType_PTR,
+                                kDNSServiceClass_IN,
+                                resolve_reply,
+                                r);
+    if (ret != kDNSServiceErr_NoError || !r->client) {
+        r->defer_time_event = avahi_time_event_new(r->server->time_event_queue, NULL, resolve_error_callback, r);
+    } else {
+        r->watch = r->server->poll_api->watch_new(r->server->poll_api, DNSServiceRefSockFD(r->client), AVAHI_WATCH_IN, resolve_socket_event, r);
+    }
+}
+#endif
+
 static void time_event_callback(AvahiTimeEvent *e, void *userdata) {
     AvahiSAddressResolver *r = userdata;
     
@@ -97,6 +198,7 @@
     r->time_event = avahi_time_event_new(r->server->time_event_queue, &tv, time_event_callback, r);
 }
 
+#ifndef HAVE_BONJOUR
 static void record_browser_callback(
     AvahiSRecordBrowser*rr,
     AvahiIfIndex interface,
@@ -175,6 +277,7 @@
             break;
     }
 }
+#endif
 
 AvahiSAddressResolver *avahi_s_address_resolver_new(
     AvahiServer *server,
@@ -227,6 +330,13 @@
 
     r->time_event = NULL;
 
+#ifdef HAVE_BONJOUR
+    r->defer_time_event = NULL;
+    r->watch = NULL;
+    r->client = NULL;
+    r->lookup_flags = flags;
+    avahi_resolve_address_start(r);
+#else
     if (!(flags & (AVAHI_LOOKUP_USE_MULTICAST|AVAHI_LOOKUP_USE_WIDE_AREA))) {
 
         if (!server->wide_area_lookup_engine || !avahi_wide_area_has_servers(server->wide_area_lookup_engine))
@@ -243,6 +353,7 @@
         avahi_s_address_resolver_free(r);
         return NULL;
     }
+#endif
 
     start_timeout(r);
     
@@ -266,5 +377,18 @@
     if (r->key)
         avahi_key_unref(r->key);
     
+#ifdef HAVE_BONJOUR
+    if (r->defer_time_event) {
+        avahi_time_event_free(r->defer_time_event);
+        r->defer_time_event = NULL;
+    }
+
+    if (r->watch)
+        r->server->poll_api->watch_free(r->watch);
+
+    if (r->client)
+        DNSServiceRefDeallocate(r->client);
+#endif
+
     avahi_free(r);
 }