1/***
2  This file is part of avahi.
3
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.
8
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.
13
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
17  USA.
18***/
19
20#ifdef HAVE_CONFIG_H
21#include <config.h>
22#endif
23
24#include <stdlib.h>
25#include <stdio.h>
26#include <assert.h>
27
28#include <sys/types.h>
29#include <sys/socket.h>
30#include <netinet/in.h>
31#include <arpa/inet.h>
32
33#include "avahi-common/avahi-malloc.h"
34#include <avahi-common/simple-watch.h>
35#include <avahi-common/alternative.h>
36#include <avahi-common/timeval.h>
37
38#include <avahi-core/core.h>
39#include <avahi-core/log.h>
40#include <avahi-core/publish.h>
41#include <avahi-core/lookup.h>
42#include <avahi-core/dns-srv-rr.h>
43
44static AvahiSEntryGroup *group = NULL;
45static AvahiServer *server = NULL;
46static char *service_name = NULL;
47
48static const AvahiPoll *poll_api;
49
50static void quit_timeout_callback(AVAHI_GCC_UNUSED AvahiTimeout *timeout, void* userdata) {
51    AvahiSimplePoll *simple_poll = userdata;
52
53    avahi_simple_poll_quit(simple_poll);
54}
55
56static void dump_line(const char *text, AVAHI_GCC_UNUSED void* userdata) {
57    printf("%s\n", text);
58}
59
60static void dump_timeout_callback(AvahiTimeout *timeout, void* userdata) {
61    struct timeval tv;
62
63    AvahiServer *avahi = userdata;
64    avahi_server_dump(avahi, dump_line, NULL);
65
66    avahi_elapse_time(&tv, 5000, 0);
67    poll_api->timeout_update(timeout, &tv);
68}
69
70static const char *browser_event_to_string(AvahiBrowserEvent event) {
71    switch (event) {
72        case AVAHI_BROWSER_NEW : return "NEW";
73        case AVAHI_BROWSER_REMOVE : return "REMOVE";
74        case AVAHI_BROWSER_CACHE_EXHAUSTED : return "CACHE_EXHAUSTED";
75        case AVAHI_BROWSER_ALL_FOR_NOW : return "ALL_FOR_NOW";
76        case AVAHI_BROWSER_FAILURE : return "FAILURE";
77    }
78
79    abort();
80}
81
82static const char *resolver_event_to_string(AvahiResolverEvent event) {
83    switch (event) {
84        case AVAHI_RESOLVER_FOUND: return "FOUND";
85        case AVAHI_RESOLVER_FAILURE: return "FAILURE";
86    }
87    abort();
88}
89
90static void record_browser_callback(
91    AvahiSRecordBrowser *r,
92    AvahiIfIndex interface,
93    AvahiProtocol protocol,
94    AvahiBrowserEvent event,
95    AvahiRecord *record,
96    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
97    AVAHI_GCC_UNUSED void* userdata) {
98    char *t;
99
100    assert(r);
101
102    if (record) {
103        avahi_log_debug("RB: record [%s] on %i.%i is %s", t = avahi_record_to_string(record), interface, protocol, browser_event_to_string(event));
104        avahi_free(t);
105    } else
106        avahi_log_debug("RB: [%s]", browser_event_to_string(event));
107
108}
109
110static void remove_entries(void);
111static void create_entries(int new_name);
112
113static void entry_group_callback(AVAHI_GCC_UNUSED AvahiServer *s, AVAHI_GCC_UNUSED AvahiSEntryGroup *g, AvahiEntryGroupState state, AVAHI_GCC_UNUSED void* userdata) {
114    avahi_log_debug("entry group state: %i", state);
115
116    if (state == AVAHI_ENTRY_GROUP_COLLISION) {
117        remove_entries();
118        create_entries(1);
119        avahi_log_debug("Service name conflict, retrying with <%s>", service_name);
120    } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
121        avahi_log_debug("Service established under name <%s>", service_name);
122    }
123}
124
125static void server_callback(AvahiServer *s, AvahiServerState state, AVAHI_GCC_UNUSED void* userdata) {
126
127    server = s;
128    avahi_log_debug("server state: %i", state);
129
130    if (state == AVAHI_SERVER_RUNNING) {
131        avahi_log_debug("Server startup complete. Host name is <%s>. Service cookie is %u", avahi_server_get_host_name_fqdn(s), avahi_server_get_local_service_cookie(s));
132        create_entries(0);
133    } else if (state == AVAHI_SERVER_COLLISION) {
134        char *n;
135        remove_entries();
136
137        n = avahi_alternative_host_name(avahi_server_get_host_name(s));
138
139        avahi_log_debug("Host name conflict, retrying with <%s>", n);
140        avahi_server_set_host_name(s, n);
141        avahi_free(n);
142    }
143}
144
145static void remove_entries(void) {
146    if (group)
147        avahi_s_entry_group_reset(group);
148}
149
150static void create_entries(int new_name) {
151    AvahiAddress a;
152    AvahiRecord *r;
153
154    remove_entries();
155
156    if (!group)
157        group = avahi_s_entry_group_new(server, entry_group_callback, NULL);
158
159    assert(avahi_s_entry_group_is_empty(group));
160
161    if (!service_name)
162        service_name = avahi_strdup("Test Service");
163    else if (new_name) {
164        char *n = avahi_alternative_service_name(service_name);
165        avahi_free(service_name);
166        service_name = n;
167    }
168
169    if (avahi_server_add_service(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, service_name, "_http._tcp", NULL, NULL, 80, "foo", NULL) < 0) {
170        avahi_log_error("Failed to add HTTP service");
171        goto fail;
172    }
173
174    if (avahi_server_add_service(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, service_name, "_ftp._tcp", NULL, NULL, 21, "foo", NULL) < 0) {
175        avahi_log_error("Failed to add FTP service");
176        goto fail;
177    }
178
179    if (avahi_server_add_service(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0,service_name, "_webdav._tcp", NULL, NULL, 80, "foo", NULL) < 0) {
180        avahi_log_error("Failed to add WEBDAV service");
181        goto fail;
182    }
183
184    if (avahi_server_add_dns_server_address(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, NULL, AVAHI_DNS_SERVER_RESOLVE, avahi_address_parse("192.168.50.1", AVAHI_PROTO_UNSPEC, &a), 53) < 0) {
185        avahi_log_error("Failed to add new DNS Server address");
186        goto fail;
187    }
188
189    r = avahi_record_new_full("cname.local", AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_CNAME, AVAHI_DEFAULT_TTL);
190    r->data.cname.name = avahi_strdup("cocaine.local");
191
192    if (avahi_server_add(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, 0, r) < 0) {
193        avahi_record_unref(r);
194        avahi_log_error("Failed to add CNAME record");
195        goto fail;
196    }
197    avahi_record_unref(r);
198
199    avahi_s_entry_group_commit(group);
200    return;
201
202fail:
203    if (group)
204        avahi_s_entry_group_free(group);
205
206    group = NULL;
207}
208
209static void hnr_callback(
210    AVAHI_GCC_UNUSED AvahiSHostNameResolver *r,
211    AvahiIfIndex iface,
212    AvahiProtocol protocol,
213    AvahiResolverEvent event,
214    const char *hostname,
215    const AvahiAddress *a,
216    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
217    AVAHI_GCC_UNUSED void* userdata) {
218    char t[AVAHI_ADDRESS_STR_MAX];
219
220    if (a)
221        avahi_address_snprint(t, sizeof(t), a);
222
223    avahi_log_debug("HNR: (%i.%i) <%s> -> %s [%s]", iface, protocol, hostname, a ? t : "n/a", resolver_event_to_string(event));
224}
225
226static void ar_callback(
227    AVAHI_GCC_UNUSED AvahiSAddressResolver *r,
228    AvahiIfIndex iface,
229    AvahiProtocol protocol,
230    AvahiResolverEvent event,
231    const AvahiAddress *a,
232    const char *hostname,
233    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
234    AVAHI_GCC_UNUSED void* userdata) {
235    char t[AVAHI_ADDRESS_STR_MAX];
236
237    avahi_address_snprint(t, sizeof(t), a);
238
239    avahi_log_debug("AR: (%i.%i) %s -> <%s> [%s]", iface, protocol, t, hostname ? hostname : "n/a", resolver_event_to_string(event));
240}
241
242static void db_callback(
243    AVAHI_GCC_UNUSED AvahiSDomainBrowser *b,
244    AvahiIfIndex iface,
245    AvahiProtocol protocol,
246    AvahiBrowserEvent event,
247    const char *domain,
248    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
249    AVAHI_GCC_UNUSED void* userdata) {
250
251    avahi_log_debug("DB: (%i.%i) <%s> [%s]", iface, protocol, domain ? domain : "NULL", browser_event_to_string(event));
252}
253
254static void stb_callback(
255    AVAHI_GCC_UNUSED AvahiSServiceTypeBrowser *b,
256    AvahiIfIndex iface,
257    AvahiProtocol protocol,
258    AvahiBrowserEvent event,
259    const char *service_type,
260    const char *domain,
261    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
262    AVAHI_GCC_UNUSED void* userdata) {
263
264    avahi_log_debug("STB: (%i.%i) %s in <%s> [%s]", iface, protocol, service_type ? service_type : "NULL", domain ? domain : "NULL", browser_event_to_string(event));
265}
266
267static void sb_callback(
268    AVAHI_GCC_UNUSED AvahiSServiceBrowser *b,
269    AvahiIfIndex iface,
270    AvahiProtocol protocol,
271    AvahiBrowserEvent event,
272    const char *name,
273    const char *service_type,
274    const char *domain,
275    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
276    AVAHI_GCC_UNUSED void* userdata) {
277    avahi_log_debug("SB: (%i.%i) <%s> as %s in <%s> [%s]", iface, protocol, name ? name : "NULL", service_type ? service_type : "NULL", domain ? domain : "NULL", browser_event_to_string(event));
278}
279
280static void sr_callback(
281    AVAHI_GCC_UNUSED AvahiSServiceResolver *r,
282    AvahiIfIndex iface,
283    AvahiProtocol protocol,
284    AvahiResolverEvent event,
285    const char *name,
286    const char*service_type,
287    const char*domain_name,
288    const char*hostname,
289    const AvahiAddress *a,
290    uint16_t port,
291    AvahiStringList *txt,
292    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
293    AVAHI_GCC_UNUSED void* userdata) {
294
295    if (event != AVAHI_RESOLVER_FOUND)
296        avahi_log_debug("SR: (%i.%i) <%s> as %s in <%s> [%s]", iface, protocol, name, service_type, domain_name, resolver_event_to_string(event));
297    else {
298        char t[AVAHI_ADDRESS_STR_MAX], *s;
299
300        avahi_address_snprint(t, sizeof(t), a);
301
302        s = avahi_string_list_to_string(txt);
303        avahi_log_debug("SR: (%i.%i) <%s> as %s in <%s>: %s/%s:%i (%s) [%s]", iface, protocol, name, service_type, domain_name, hostname, t, port, s, resolver_event_to_string(event));
304        avahi_free(s);
305    }
306}
307
308static void dsb_callback(
309    AVAHI_GCC_UNUSED AvahiSDNSServerBrowser *b,
310    AvahiIfIndex iface,
311    AvahiProtocol protocol,
312    AvahiBrowserEvent event,
313    const char*hostname,
314    const AvahiAddress *a,
315    uint16_t port,
316    AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
317    AVAHI_GCC_UNUSED void* userdata) {
318
319    char t[AVAHI_ADDRESS_STR_MAX] = "n/a";
320
321    if (a)
322        avahi_address_snprint(t, sizeof(t), a);
323
324    avahi_log_debug("DSB: (%i.%i): %s/%s:%i [%s]", iface, protocol, hostname ? hostname : "NULL", t, port, browser_event_to_string(event));
325}
326
327int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char *argv[]) {
328    AvahiSRecordBrowser *r;
329    AvahiSHostNameResolver *hnr;
330    AvahiSAddressResolver *ar;
331    AvahiKey *k;
332    AvahiServerConfig config;
333    AvahiAddress a;
334    AvahiSDomainBrowser *db;
335    AvahiSServiceTypeBrowser *stb;
336    AvahiSServiceBrowser *sb;
337    AvahiSServiceResolver *sr;
338    AvahiSDNSServerBrowser *dsb;
339    AvahiSimplePoll *simple_poll;
340    int error;
341    struct timeval tv;
342
343    simple_poll = avahi_simple_poll_new();
344    poll_api = avahi_simple_poll_get(simple_poll);
345
346    avahi_server_config_init(&config);
347
348    avahi_address_parse("192.168.50.1", AVAHI_PROTO_UNSPEC, &config.wide_area_servers[0]);
349    config.n_wide_area_servers = 1;
350    config.enable_wide_area = 1;
351
352    server = avahi_server_new(poll_api, &config, server_callback, NULL, &error);
353    avahi_server_config_free(&config);
354
355    k = avahi_key_new("_http._tcp.0pointer.de", AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR);
356    r = avahi_s_record_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, k, 0, record_browser_callback, NULL);
357    avahi_key_unref(k);
358
359    hnr = avahi_s_host_name_resolver_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "cname.local", AVAHI_PROTO_UNSPEC, 0, hnr_callback, NULL);
360
361    ar = avahi_s_address_resolver_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, avahi_address_parse("192.168.50.1", AVAHI_PROTO_INET, &a), 0, ar_callback, NULL);
362
363    db = avahi_s_domain_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, NULL, AVAHI_DOMAIN_BROWSER_BROWSE, 0, db_callback, NULL);
364
365    stb = avahi_s_service_type_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, NULL, 0, stb_callback, NULL);
366
367    sb = avahi_s_service_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "_http._tcp", NULL, 0, sb_callback, NULL);
368
369    sr = avahi_s_service_resolver_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "Ecstasy HTTP", "_http._tcp", "local", AVAHI_PROTO_UNSPEC, 0, sr_callback, NULL);
370
371    dsb = avahi_s_dns_server_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "local", AVAHI_DNS_SERVER_RESOLVE, AVAHI_PROTO_UNSPEC, 0, dsb_callback, NULL);
372
373    avahi_elapse_time(&tv, 1000*5, 0);
374    poll_api->timeout_new(poll_api, &tv, dump_timeout_callback, server);
375
376    avahi_elapse_time(&tv, 1000*60, 0);
377    poll_api->timeout_new(poll_api, &tv, quit_timeout_callback, simple_poll);
378
379    avahi_simple_poll_loop(simple_poll);
380
381    avahi_s_record_browser_free(r);
382    avahi_s_host_name_resolver_free(hnr);
383    avahi_s_address_resolver_free(ar);
384    avahi_s_domain_browser_free(db);
385    avahi_s_service_type_browser_free(stb);
386    avahi_s_service_browser_free(sb);
387    avahi_s_service_resolver_free(sr);
388    avahi_s_dns_server_browser_free(dsb);
389
390    if (group)
391        avahi_s_entry_group_free(group);
392
393    if (server)
394        avahi_server_free(server);
395
396    if (simple_poll)
397        avahi_simple_poll_free(simple_poll);
398
399    avahi_free(service_name);
400
401    return 0;
402}
403