Anonymous

static void resolve_callback(AvahiSServiceResolver *r,
			     AvahiIfIndex interface,
			     AvahiProtocol protocol,
			     AvahiResolverEvent event,
			     const char *name,
			     const char *type,
			     const char *domain,
			     const char *host_name,
			     const AvahiAddress *address,
			     uint16_t port,
			     AvahiStringList *txt,
			     AvahiLookupResultFlags flags,
			     void* userdata) 
{
	OSDAvahi* osd = userdata;
	assert(osd);
	assert(r);

	AvahiServer* server = osd->server;
	assert(server);

	/* Called whenever a service has been resolved successfully or timed out */
	char a[512], *t;
	switch (event) {
	case AVAHI_RESOLVER_FAILURE:
		fprintf(stderr, "(Resolver) Failed to resolve service '%s' of type '%s' in domain '%s': %s\n", name, type, domain, avahi_strerror(avahi_server_errno(server)));
		break;

	case AVAHI_RESOLVER_FOUND:
		fprintf(stderr, "(Resolver) Service '%s' of type '%s' in domain '%s':\n", name, type, domain);

		avahi_address_snprint(a, sizeof(a), address);
		t = avahi_string_list_to_string(txt);
		fprintf(stderr,
			"\t%s:%u (%s)\n"
			"\tTXT=%s\n"
			"\tcookie is %u\n"
			"\tis_local: %i\n"
			"\twide_area: %i\n"
			"\tmulticast: %i\n"
			"\tcached: %i\n",
			host_name, port, a,
			t,
			avahi_string_list_get_service_cookie(txt),
			!!(flags & AVAHI_LOOKUP_RESULT_LOCAL),
			!!(flags & AVAHI_LOOKUP_RESULT_WIDE_AREA),
			!!(flags & AVAHI_LOOKUP_RESULT_MULTICAST),
			!!(flags & AVAHI_LOOKUP_RESULT_CACHED));
		avahi_free(t);

	}

	avahi_s_service_resolver_free(r);
}

static void browse_callback(AvahiSServiceBrowser *b,
			    AvahiIfIndex interface,
			    AvahiProtocol protocol,
			    AvahiBrowserEvent event,
			    const char *name,
			    const char *type,
			    const char *domain,
			    AvahiLookupResultFlags flags,
			    void* userdata) 
{
	OSDAvahi* osd = userdata;
	assert(osd);

	AvahiServer* s = osd->server;
	assert(b);

	/* Called whenever a new services becomes available on the LAN or is removed from the LAN */
	switch (event) {
	case AVAHI_BROWSER_FAILURE:
		fprintf(stderr, "(Browser) %s\n", avahi_strerror(avahi_server_errno(s)));
		osd->fail_occurred = 1;
		return;

	case AVAHI_BROWSER_NEW:
		fprintf(stderr, "(Browser) NEW: service '%s' of type '%s' in domain '%s'\n", name, type, domain);

		/* We ignore the returned resolver object. In the callback
		   function we free it. If the server is terminated before
		   the callback function is called the server will free
		   the resolver for us. */
		if (!(avahi_s_service_resolver_new(s, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, 0, resolve_callback, osd)))
			fprintf(stderr, "Failed to resolve service '%s': %s\n", name, avahi_strerror(avahi_server_errno(s)));
		break;

	case AVAHI_BROWSER_REMOVE:
		fprintf(stderr, "(Browser) REMOVE: service '%s' of type '%s' in domain '%s'\n", name, type, domain);
		break;

	case AVAHI_BROWSER_ALL_FOR_NOW:
	case AVAHI_BROWSER_CACHE_EXHAUSTED:
		fprintf(stderr, "(Browser) %s\n", event == AVAHI_BROWSER_CACHE_EXHAUSTED ? "CACHE_EXHAUSTED" : "ALL_FOR_NOW");
		break;
	}
}


OSDAvahi* osd_avahi_initialize(void)
{
	AvahiServerConfig config;
	AvahiSServiceBrowser *sb = NULL;
	AvahiSimplePoll *simple_poll = NULL;
	AvahiServer *server = NULL;
	OSDAvahi* ret = malloc(sizeof(OSDAvahi));
	int error = 0;
	
	/* Allocate main loop object */
	if (!(simple_poll = avahi_simple_poll_new())) {
		fprintf(stderr, "Failed to create simple poll object.\n");
		goto fail;
	}

	/* Publish local records */
	avahi_server_config_init(&config);
	config.publish_hinfo = 1;
	config.publish_addresses = 1;
	config.publish_workstation = 1;
	config.publish_domain = 1;

#if 0
	/* Set a unicast DNS server for wide area DNS-SD */
	/* FIXME: We may use this functionality later to traverse
	 * separate LANs */
	avahi_address_parse("192.168.50.1", AVAHI_PROTO_UNSPEC, &config.wide_area_servers[0]);
	config.n_wide_area_servers = 1;
	config.enable_wide_area = 1;
#endif

	/* Allocate a new server */
	server = avahi_server_new(avahi_simple_poll_get(simple_poll), &config, NULL, NULL, &error);

	/* Free the configuration data */
	avahi_server_config_free(&config);

	/* Check whether creating the server object succeeded */
	if (!server) {
		fprintf(stderr, "Failed to create server: %s\n", avahi_strerror(error));
		goto fail;
	}

	/* Create the service browser */
	if (!(sb = avahi_s_service_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "_workstation._tcp", NULL, 0, browse_callback, ret))) {
		fprintf(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_server_errno(server)));
		goto fail;
	}

	ret->sb = sb;
	ret->simple_poll = simple_poll;
	ret->server = server;
	ret->fail_occurred = 0;

	return ret;

fail:
	if (sb)
		avahi_s_service_browser_free(sb);

	if (server)
		avahi_server_free(server);

	if (simple_poll)
		avahi_simple_poll_free(simple_poll);
	return NULL;
}

void osd_avahi_eventloop(OSDAvahi* server)
{
	avahi_simple_poll_iterate(server->simple_poll, 0 /* No blocking */);
}

void osd_avahi_free(OSDAvahi* obj)
{
	/* Cleanup things */
	if (obj->sb)
		avahi_s_service_browser_free(obj->sb);

	if (obj->server)
		avahi_server_free(obj->server);

	if (obj->simple_poll)
		avahi_simple_poll_free(obj->simple_poll);

	free(obj);
}