Source: Desktop consolidation
Upstream promotion status: unknown

--- avahi-0.6.28/avahi-core/browse-service.c.orig	2010-08-26 01:51:38.983153000 +0100
+++ avahi-0.6.28/avahi-core/browse-service.c	2011-01-20 09:22:41.808471289 +0000
@@ -21,10 +21,14 @@
 #include <config.h>
 #endif
 
+#include <stdio.h>
 #include <string.h>
 
 #include <avahi-common/domain.h>
 #include <avahi-common/malloc.h>
+#ifdef HAVE_BONJOUR
+#include <avahi-common/timeval.h>
+#endif
 #include <avahi-common/error.h>
 
 #include "browse.h"
@@ -40,9 +44,131 @@
     AvahiSServiceBrowserCallback callback;
     void* userdata;
 
+#ifdef HAVE_BONJOUR
+    AvahiProtocol protocol;
+    AvahiIfIndex interface;
+    AvahiTimeEvent *browse_error_event;
+    AvahiTimeEvent *all_for_now_event;
+    AvahiLookupFlags flags;
+    AvahiWatch *watch;
+    DNSServiceRef client;
+#endif
+
     AVAHI_LLIST_FIELDS(AvahiSServiceBrowser, browser);
 };
 
+#ifdef HAVE_BONJOUR
+static void browse_reply(DNSServiceRef client, const DNSServiceFlags flags, uint32_t IfIndex, DNSServiceErrorType errorCode,
+        const char *name, const char *type, const char *domain, void *context) {
+    AvahiSServiceBrowser *b = context;
+    char *op = (flags &kDNSServiceFlagsAdd) ? "Add" : "Rmv";
+    AvahiBrowserEvent event;
+    AvahiIfIndex interface;
+    
+    interface = (IfIndex == kDNSServiceInterfaceIndexAny) ? AVAHI_IF_UNSPEC : IfIndex;
+
+    event = (flags &kDNSServiceFlagsAdd) ? AVAHI_BROWSER_NEW : AVAHI_BROWSER_REMOVE;
+
+    b->callback(b, interface, AVAHI_PROTO_UNSPEC, event, name, type, domain, 0, b->userdata);
+    
+}
+
+static void browse_socket_event(AvahiWatch *w, int fd, AvahiWatchEvent events, void *userdata) {
+    AvahiSServiceBrowser *b = userdata;
+    DNSServiceErrorType ret;
+
+    assert(w);
+    assert(fd >= 0);
+    assert(events & AVAHI_WATCH_IN);
+
+    assert (fd == DNSServiceRefSockFD(b->client));
+    ret = DNSServiceProcessResult(b->client);
+    if (ret != kDNSServiceErr_NoError) {
+        if (b->watch) {
+            b->server->poll_api->watch_free(b->watch);
+            b->watch = NULL;
+        }
+        DNSServiceRefDeallocate(b->client);
+        b->client = NULL;
+        avahi_server_set_errno(b->server, AVAHI_ERR_DISCONNECTED);
+        b->callback(b, 
+            b->interface, 
+            b->protocol, 
+            AVAHI_BROWSER_FAILURE,
+            NULL,
+            b->service_type, 
+            b->domain_name,
+            0,
+            b->userdata);
+    }
+}
+
+static void all_for_now_callback(AvahiTimeEvent *e, void* userdata) {
+    AvahiSServiceBrowser *b = userdata;
+
+    assert(e);
+    assert(b);
+
+    avahi_time_event_free(b->all_for_now_event);
+    b->all_for_now_event = NULL;
+
+    b->callback(b, 
+        b->interface, 
+        b->protocol,
+        AVAHI_BROWSER_ALL_FOR_NOW, 
+        NULL,
+        b->service_type,
+        NULL,
+        0,
+        b->userdata);
+}
+
+static void browse_error_callback(AvahiTimeEvent *e, void *userdata) {
+    AvahiSServiceBrowser *b = userdata;
+
+    if (b->browse_error_event) {
+        avahi_time_event_free(b->browse_error_event);
+        b->browse_error_event = NULL;
+    }
+    avahi_server_set_errno(b->server, AVAHI_ERR_FAILURE);
+    b->callback(b, 
+        b->interface, 
+        b->protocol, 
+        AVAHI_BROWSER_FAILURE,
+        NULL,
+        b->service_type, 
+        b->domain_name,
+        0,
+        b->userdata);
+}
+
+static void avahi_browse_service_start(AvahiSServiceBrowser *b) {
+    DNSServiceErrorType ret;
+    struct timeval tv;
+
+    ret = DNSServiceBrowse(&b->client,
+              0,
+              b->interface == AVAHI_IF_UNSPEC ?
+                  kDNSServiceInterfaceIndexAny :
+                  b->interface,
+              b->service_type,
+              b->domain_name,
+              browse_reply,
+              b);
+    if (ret != kDNSServiceErr_NoError || !b->client) {
+        b->browse_error_event = avahi_time_event_new(b->server->time_event_queue,
+NULL, browse_error_callback, b);
+    } else {
+        b->watch = b->server->poll_api->watch_new(b->server->poll_api, DNSServiceRefSockFD(b->client), AVAHI_WATCH_IN, browse_socket_event, b);
+
+        /* Add a second */
+        gettimeofday(&tv, NULL);
+        avahi_timeval_add(&tv, 1000000);
+        b->all_for_now_event = avahi_time_event_new(b->server->time_event_queue, &tv, all_for_now_callback, b);
+    }
+}
+#endif
+
 static void record_browser_callback(
     AvahiSRecordBrowser*rr,
     AvahiIfIndex interface,
@@ -102,7 +228,11 @@
     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, !domain || avahi_is_valid_domain_name(domain), AVAHI_ERR_INVALID_DOMAIN_NAME);
+#ifdef HAVE_BONJOUR
+    AVAHI_CHECK_VALIDITY_RETURN_NULL(server, AVAHI_FLAGS_VALID(flags, AVAHI_LOOKUP_USE_MULTICAST), AVAHI_ERR_INVALID_FLAGS);
+#else
     AVAHI_CHECK_VALIDITY_RETURN_NULL(server, AVAHI_FLAGS_VALID(flags, AVAHI_LOOKUP_USE_WIDE_AREA|AVAHI_LOOKUP_USE_MULTICAST), AVAHI_ERR_INVALID_FLAGS);
+#endif
     AVAHI_CHECK_VALIDITY_RETURN_NULL(server, avahi_is_valid_service_type_generic(service_type), AVAHI_ERR_INVALID_SERVICE_TYPE);
 
     if (!domain)
@@ -132,6 +262,16 @@
         goto fail;
     }
 
+#ifdef HAVE_BONJOUR
+    b->protocol = protocol;
+    b->interface = interface;
+    b->flags = flags;
+    b->browse_error_event = NULL;
+    b->all_for_now_event = NULL;
+    b->client = NULL;
+    b->watch = NULL;
+    avahi_browse_service_start(b);
+#else
     if (!(k = avahi_key_new(n, AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR))) {
         avahi_server_set_errno(server, AVAHI_ERR_NO_MEMORY);
         goto fail;
@@ -141,6 +281,7 @@
         goto fail;
 
     avahi_key_unref(k);
+#endif
 
     return b;
 
@@ -163,5 +304,21 @@
 
     avahi_free(b->domain_name);
     avahi_free(b->service_type);
+#ifdef HAVE_BONJOUR
+    if (b->browse_error_event) {
+        avahi_time_event_free(b->browse_error_event);
+        b->browse_error_event = NULL;
+    }
+    if (b->all_for_now_event) {
+        avahi_time_event_free(b->all_for_now_event);
+        b->all_for_now_event = NULL;
+    }
+
+    if (b->watch)
+        b->server->poll_api->watch_free(b->watch);
+
+    if (b->client)
+        DNSServiceRefDeallocate (b->client);
+#endif
     avahi_free(b);
 }