git.meshlink.io Git - catta/blob - avahi-core/avahi-test.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of avahi.
   5
   6   avahi is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as
   8   published by the Free Software Foundation; either version 2.1 of the
   9   License, or (at your option) any later version.
  10
  11   avahi is distributed in the hope that it will be useful, but WITHOUT
  12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  14   Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with avahi; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <sys/socket.h>
  27 #include <netinet/in.h>
  28 #include <arpa/inet.h>
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <assert.h>
  32
  33 #include <avahi-common/malloc.h>
  34 #include <avahi-common/simple-watch.h>
  35 #include <avahi-common/alternative.h>
  36 #include <avahi-core/core.h>
  37 #include <avahi-core/log.h>
  38
  39 static AvahiSEntryGroup *group = NULL;
  40 static AvahiServer *server = NULL;
  41 static char *service_name = NULL;
  42
  43 static const AvahiPoll *poll_api;
  44
  45 static void quit_timeout_callback(AvahiTimeout *timeout, void* userdata) {
  46     AvahiSimplePoll *simple_poll = userdata;
  47
  48     avahi_simple_poll_quit(simple_poll);
  49 }
  50
  51 static void dump_line(const char *text, void* userdata) {
  52     printf("%s\n", text);
  53 }
  54
  55 static void dump_timeout_callback(AvahiTimeout *timeout, void* userdata) {
  56     struct timeval tv;
  57
  58     AvahiServer *avahi = userdata;
  59     avahi_server_dump(avahi, dump_line, NULL);
  60
  61     avahi_elapse_time(&tv, 5000, 0);
  62     poll_api->timeout_update(timeout, &tv);
  63 }
  64
  65 static void record_browser_callback(AvahiSRecordBrowser *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, AvahiRecord *record, void* userdata) {
  66     char *t;
  67
  68     assert(r);
  69     assert(record);
  70     assert(interface > 0);
  71     assert(protocol != AVAHI_PROTO_UNSPEC);
  72
  73     avahi_log_debug("SUBSCRIPTION: record [%s] on %i.%i is %s", t = avahi_record_to_string(record), interface, protocol,
  74               event == AVAHI_BROWSER_NEW ? "new" : "remove");
  75
  76     avahi_free(t);
  77 }
  78
  79 static void remove_entries(void);
  80 static void create_entries(int new_name);
  81
  82 static void entry_group_callback(AvahiServer *s, AvahiSEntryGroup *g, AvahiEntryGroupState state, void* userdata) {
  83     avahi_log_debug("entry group state: %i", state);
  84
  85     if (state == AVAHI_ENTRY_GROUP_COLLISION) {
  86         remove_entries();
  87         create_entries(1);
  88         avahi_log_debug("Service name conflict, retrying with <%s>", service_name);
  89     } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
  90         avahi_log_debug("Service established under name <%s>", service_name);
  91     }
  92 }
  93
  94 static void server_callback(AvahiServer *s, AvahiServerState state, void* userdata) {
  95
  96      avahi_log_debug("server state: %i", state);
  97
  98     if (state == AVAHI_SERVER_RUNNING) {
  99         avahi_log_debug("Server startup complete.  Host name is <%s>", avahi_server_get_host_name_fqdn(s));
 100         create_entries(0);
 101     } else if (state == AVAHI_SERVER_COLLISION) {
 102         char *n;
 103         remove_entries();
 104
 105         n = avahi_alternative_host_name(avahi_server_get_host_name(s));
 106
 107         avahi_log_debug("Host name conflict, retrying with <%s>", n);
 108         avahi_server_set_host_name(s, n);
 109         avahi_free(n);
 110     }
 111 }
 112
 113 static void remove_entries(void) {
 114     if (group)
 115         avahi_s_entry_group_reset(group);
 116 }
 117
 118 static void create_entries(int new_name) {
 119     AvahiAddress a;
 120
 121     remove_entries();
 122
 123     if (!group)
 124         group = avahi_s_entry_group_new(server, entry_group_callback, NULL);
 125
 126     assert(avahi_s_entry_group_is_empty(group));
 127
 128     if (!service_name)
 129         service_name = avahi_strdup("Test Service");
 130     else if (new_name) {
 131         char *n = avahi_alternative_service_name(service_name);
 132         avahi_free(service_name);
 133         service_name = n;
 134     }
 135
 136     if (avahi_server_add_service(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, service_name, "_http._tcp", NULL, NULL, 80, "foo", NULL) < 0) {
 137         avahi_log_error("Failed to add HTTP service");
 138         goto fail;
 139     }
 140
 141     if (avahi_server_add_service(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, service_name, "_ftp._tcp", NULL, NULL, 21, "foo", NULL) < 0) {
 142         avahi_log_error("Failed to add FTP service");
 143         goto fail;
 144     }
 145
 146     if (avahi_server_add_service(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, service_name, "_webdav._tcp", NULL, NULL, 80, "foo", NULL) < 0) {
 147         avahi_log_error("Failed to add WEBDAV service");
 148         goto fail;
 149     }
 150
 151     if (avahi_server_add_dns_server_address(server, group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, NULL, AVAHI_DNS_SERVER_RESOLVE, avahi_address_parse("192.168.50.1", AVAHI_PROTO_UNSPEC, &a), 53) < 0) {
 152         avahi_log_error("Failed to add new DNS Server address");
 153         goto fail;
 154     }
 155
 156     avahi_s_entry_group_commit(group);
 157     return;
 158
 159 fail:
 160     if (group)
 161         avahi_s_entry_group_free(group);
 162
 163     group = NULL;
 164 }
 165
 166 static void hnr_callback(AvahiSHostNameResolver *r, AvahiIfIndex iface, AvahiProtocol protocol, AvahiResolverEvent event, const char *hostname, const AvahiAddress *a, void* userdata) {
 167     char t[64];
 168
 169     if (a)
 170         avahi_address_snprint(t, sizeof(t), a);
 171
 172     avahi_log_debug("HNR: (%i.%i) <%s> -> %s [%s]", iface, protocol, hostname, a ? t : "n/a", event == AVAHI_RESOLVER_FOUND ? "found" : "timeout");
 173 }
 174
 175 static void ar_callback(AvahiSAddressResolver *r, AvahiIfIndex iface, AvahiProtocol protocol, AvahiResolverEvent event, const AvahiAddress *a, const char *hostname, void* userdata) {
 176     char t[64];
 177
 178     avahi_address_snprint(t, sizeof(t), a);
 179
 180     avahi_log_debug("AR: (%i.%i) %s -> <%s> [%s]", iface, protocol, t, hostname ? hostname : "n/a", event == AVAHI_RESOLVER_FOUND ? "found" : "timeout");
 181 }
 182
 183 static void db_callback(AvahiSDomainBrowser *b, AvahiIfIndex iface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *domain, void* userdata) {
 184
 185     avahi_log_debug("DB: (%i.%i) <%s> [%s]", iface, protocol, domain, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 186 }
 187
 188 static void stb_callback(AvahiSServiceTypeBrowser *b, AvahiIfIndex iface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *service_type, const char *domain, void* userdata) {
 189
 190     avahi_log_debug("STB: (%i.%i) %s in <%s> [%s]", iface, protocol, service_type, domain, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 191 }
 192
 193 static void sb_callback(AvahiSServiceBrowser *b, AvahiIfIndex iface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *service_type, const char *domain, void* userdata) {
 194    avahi_log_debug("SB: (%i.%i) <%s> as %s in <%s> [%s]", iface, protocol, name, service_type, domain, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 195 }
 196
 197 static void sr_callback(AvahiSServiceResolver *r, AvahiIfIndex iface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char*service_type, const char*domain_name, const char*hostname, const AvahiAddress *a, uint16_t port, AvahiStringList *txt, void* userdata) {
 198
 199     if (event == AVAHI_RESOLVER_TIMEOUT)
 200         avahi_log_debug("SR: (%i.%i) <%s> as %s in <%s> [timeout]", iface, protocol, name, service_type, domain_name);
 201     else {
 202         char t[64], *s;
 203
 204         avahi_address_snprint(t, sizeof(t), a);
 205
 206         s = avahi_string_list_to_string(txt);
 207         avahi_log_debug("SR: (%i.%i) <%s> as %s in <%s>: %s/%s:%i (%s) [found]", iface, protocol, name, service_type, domain_name, hostname, t, port, s);
 208         avahi_free(s);
 209     }
 210 }
 211
 212 static void dsb_callback(AvahiSDNSServerBrowser *b, AvahiIfIndex iface, AvahiProtocol protocol, AvahiBrowserEvent event, const char*hostname, const AvahiAddress *a, uint16_t port, void* userdata) {
 213     char t[64];
 214     avahi_address_snprint(t, sizeof(t), a);
 215     avahi_log_debug("DSB: (%i.%i): %s/%s:%i [%s]", iface, protocol, hostname, t, port, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 216 }
 217
 218 int main(int argc, char *argv[]) {
 219     AvahiSRecordBrowser *r;
 220     AvahiSHostNameResolver *hnr;
 221     AvahiSAddressResolver *ar;
 222     AvahiKey *k;
 223     AvahiServerConfig config;
 224     AvahiAddress a;
 225     AvahiSDomainBrowser *db;
 226     AvahiSServiceTypeBrowser *stb;
 227     AvahiSServiceBrowser *sb;
 228     AvahiSServiceResolver *sr;
 229     AvahiSDNSServerBrowser *dsb;
 230     AvahiSimplePoll *simple_poll;
 231     int error;
 232     struct timeval tv;
 233
 234     simple_poll = avahi_simple_poll_new();
 235     poll_api = avahi_simple_poll_get(simple_poll);
 236
 237     avahi_server_config_init(&config);
 238 /*     config.host_name = g_strdup("test"); */
 239     server = avahi_server_new(poll_api, &config, server_callback, NULL, &error);
 240     avahi_server_config_free(&config);
 241
 242     k = avahi_key_new("_http._tcp.local", AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR);
 243     r = avahi_s_record_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, k, record_browser_callback, NULL);
 244     avahi_key_unref(k);
 245
 246     hnr = avahi_s_host_name_resolver_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "codes-CompUTER.local", AVAHI_PROTO_UNSPEC, hnr_callback, NULL);
 247
 248     ar = avahi_s_address_resolver_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, avahi_address_parse("192.168.50.15", AVAHI_PROTO_INET, &a), ar_callback, NULL);
 249
 250     db = avahi_s_domain_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, NULL, AVAHI_DOMAIN_BROWSER_BROWSE, db_callback, NULL);
 251
 252     stb = avahi_s_service_type_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, NULL, stb_callback, NULL);
 253
 254     sb = avahi_s_service_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "_http._tcp", NULL, sb_callback, NULL);
 255
 256     sr = avahi_s_service_resolver_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "Ecstasy HTTP", "_http._tcp", "local", AVAHI_PROTO_UNSPEC, sr_callback, NULL);
 257
 258     dsb = avahi_s_dns_server_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "local", AVAHI_DNS_SERVER_RESOLVE, AVAHI_PROTO_UNSPEC, dsb_callback, NULL);
 259
 260     avahi_elapse_time(&tv, 1000*5, 0);
 261     poll_api->timeout_new(poll_api, &tv, dump_timeout_callback, server);
 262
 263     avahi_elapse_time(&tv, 1000*60, 0);
 264     poll_api->timeout_new(poll_api, &tv, quit_timeout_callback, server);
 265
 266     for (;;)
 267         if (avahi_simple_poll_iterate(simple_poll, -1) != 0)
 268             break;
 269
 270     avahi_s_record_browser_free(r);
 271     avahi_s_host_name_resolver_free(hnr);
 272     avahi_s_address_resolver_free(ar);
 273     avahi_s_service_type_browser_free(stb);
 274     avahi_s_service_browser_free(sb);
 275     avahi_s_service_resolver_free(sr);
 276     avahi_s_dns_server_browser_free(dsb);
 277
 278     if (group)
 279         avahi_s_entry_group_free(group);
 280
 281     if (server)
 282         avahi_server_free(server);
 283
 284     if (simple_poll)
 285         avahi_simple_poll_free(simple_poll);
 286
 287     avahi_free(service_name);
 288
 289     return 0;
 290 }