git.meshlink.io Git - catta/blob - client-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 <avahi-client/client.h>
  27 #include <avahi-common/error.h>
  28 #include <avahi-common/simple-watch.h>
  29 #include <avahi-common/malloc.h>
  30 #include <stdio.h>
  31 #include <assert.h>
  32
  33 static const AvahiPoll *poll_api = NULL;
  34 static AvahiSimplePoll *simple_poll = NULL;
  35
  36
  37 static void
  38 avahi_client_callback (AvahiClient *c, AvahiClientState state, void *user_data)
  39 {
  40     printf ("CLIENT: Callback on %p, state -> %d, data -> %s\n", (void*) c, state, (char*)user_data);
  41 }
  42
  43 static void
  44 avahi_entry_group_callback (AvahiEntryGroup *g, AvahiEntryGroupState state, void *user_data)
  45 {
  46     printf ("ENTRY-GROUP: Callback on %p, state -> %d, data -> %s\n", (void*) g, state, (char*)user_data);
  47 }
  48
  49 static void
  50 avahi_domain_browser_callback (AvahiDomainBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *domain, void *user_data)
  51 {
  52     printf ("DOMAIN-BROWSER: Callback on %p, interface (%d), protocol (%d), event (%d), domain (%s), data (%s)\n", (void*) b, interface, protocol, event, domain, (char*)user_data);
  53 }
  54
  55 static void
  56 avahi_service_resolver_callback (AvahiServiceResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *host_name, const AvahiAddress *a, uint16_t port, AvahiStringList *txt, void *user_data)
  57 {
  58     char addr[64];
  59     char *txtr;
  60     if (event == AVAHI_RESOLVER_TIMEOUT)
  61     {
  62         printf ("SERVICE-RESOLVER: ServiceResolver %p timed out (%s %s)\n", (void*) r, name, type);
  63         return;
  64     }
  65     avahi_address_snprint (addr, sizeof (addr), a);
  66     txtr = avahi_string_list_to_string (txt);
  67     printf ("SERVICE-RESOLVER: Callback on ServiceResolver, interface (%d), protocol (%d), event (%d), name (%s), type (%s), domain (%s), host_name (%s), address (%s), port (%d), txtdata (%s), data(%s)\n", interface, protocol, event, name, type, domain, host_name, addr, port, txtr, (char*)user_data);
  68     avahi_free(txtr);
  69 }
  70
  71 static void
  72 avahi_service_browser_callback (AvahiServiceBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, void *user_data)
  73 {
  74     AvahiServiceResolver *sr;
  75
  76     printf ("SERVICE-BROWSER: Callback on %p, interface (%d), protocol (%d), event (%d), name (%s), type (%s), domain (%s), data (%s)\n", (void*) b, interface, protocol, event, name, type, domain, (char*)user_data);
  77
  78     if (b && name)
  79     {
  80         sr = avahi_service_resolver_new (avahi_service_browser_get_client (b), interface, protocol, name, type, domain, AF_UNSPEC, avahi_service_resolver_callback, "xxXXxx");
  81         printf("New service resolver %p\n", (void*) sr);
  82     }
  83 }
  84
  85 static void
  86 avahi_service_type_browser_callback (AvahiServiceTypeBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *type, const char *domain, void *user_data)
  87 {
  88     printf ("SERVICE-TYPE-BROWSER: Callback on %p, interface (%d), protocol (%d), event (%d), type (%s), domain (%s), data (%s)\n", (void*) b, interface, protocol, event, type, domain, (char*)user_data);
  89 }
  90
  91 static void
  92 avahi_address_resolver_callback (AvahiAddressResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, AvahiProtocol aprotocol, const AvahiAddress *address, const char *name, void *userdata)
  93 {
  94     char addr[64];
  95     if (event == AVAHI_RESOLVER_TIMEOUT)
  96     {
  97         printf ("ADDRESS-RESOLVER: Callback on AddressResolver, timed out.\n");
  98         return;
  99     }
 100     avahi_address_snprint (addr, sizeof (addr), address);
 101     printf ("ADDRESS-RESOLVER: Callback on AddressResolver, interface (%d), protocol (%d), even (%d), aprotocol (%d), address (%s), name (%s), data(%s)\n", interface, protocol, event, aprotocol, addr, name, (char*) userdata);
 102 }
 103
 104 static void
 105 avahi_host_name_resolver_callback (AvahiHostNameResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const AvahiAddress *a, void *user_data)
 106 {
 107     AvahiClient *client;
 108     AvahiAddressResolver *ar;
 109     char addr[64];
 110
 111     if (event == AVAHI_RESOLVER_TIMEOUT)
 112     {
 113         printf ("HOST-NAME-RESOLVER: Callback on HostNameResolver, timed out.\n");
 114         return;
 115     }
 116     client = avahi_host_name_resolver_get_client (r);
 117     ar = avahi_address_resolver_new_a (client, interface, protocol, a, avahi_address_resolver_callback, "omghai6u");
 118     if (ar)
 119     {
 120         printf ("Succesfully created address resolver object\n");
 121     } else {
 122         printf ("Failed to create AddressResolver\n");
 123     }
 124     avahi_address_snprint (addr, sizeof (addr), a);
 125     printf ("HOST-NAME-RESOLVER: Callback on HostNameResolver, interface (%d), protocol (%d), event (%d), name (%s), address (%s), data (%s)\n", interface, protocol, event, name, addr, (char*)user_data);
 126 }
 127 static void test_free_domain_browser(AvahiTimeout *timeout, void* userdata)
 128 {
 129     AvahiServiceBrowser *b = userdata;
 130     printf ("Freeing domain browser\n");
 131     avahi_service_browser_free (b);
 132 }
 133
 134 static void test_free_entry_group (AvahiTimeout *timeout, void* userdata)
 135 {
 136     AvahiEntryGroup *g = userdata;
 137     printf ("Freeing entry group\n");
 138     avahi_entry_group_free (g);
 139 }
 140
 141 static void test_entry_group_reset (AvahiTimeout *timeout, void* userdata)
 142 {
 143     AvahiEntryGroup *g = userdata;
 144
 145     printf ("Resetting entry group\n");
 146     avahi_entry_group_reset (g);
 147
 148     avahi_entry_group_add_service (g, AVAHI_IF_UNSPEC, AF_UNSPEC, "Lathiat's Site", "_http._tcp", NULL, NULL, 80, "foo=bar2", NULL);
 149
 150     avahi_entry_group_commit (g);
 151 }
 152
 153 static void terminate(AvahiTimeout *timeout, void *userdata) {
 154
 155     avahi_simple_poll_quit(simple_poll);
 156 }
 157
 158 int main (int argc, char *argv[]) {
 159     AvahiClient *avahi;
 160     AvahiEntryGroup *group;
 161     AvahiDomainBrowser *domain;
 162     AvahiServiceBrowser *sb;
 163     AvahiServiceTypeBrowser *st;
 164     AvahiHostNameResolver *hnr;
 165     const char *ret;
 166     int error;
 167     uint32_t cookie;
 168     struct timeval tv;
 169
 170     simple_poll = avahi_simple_poll_new();
 171     poll_api = avahi_simple_poll_get(simple_poll);
 172
 173     if (!(avahi = avahi_client_new(poll_api, avahi_client_callback, "omghai2u", &error))) {
 174         fprintf(stderr, "Client failed: %s\n", avahi_strerror(error));
 175         goto fail;
 176     }
 177
 178     printf("State: %i\n", avahi_client_get_state(avahi));
 179
 180     ret = avahi_client_get_version_string (avahi);
 181     printf("Avahi Server Version: %s (Error Return: %s)\n", ret, ret ? "OK" : avahi_strerror(avahi_client_errno(avahi)));
 182
 183     ret = avahi_client_get_host_name (avahi);
 184     printf("Host Name: %s (Error Return: %s)\n", ret, ret ? "OK" : avahi_strerror(avahi_client_errno(avahi)));
 185
 186     ret = avahi_client_get_domain_name (avahi);
 187     printf("Domain Name: %s (Error Return: %s)\n", ret, ret ? "OK" : avahi_strerror(avahi_client_errno(avahi)));
 188
 189     ret = avahi_client_get_host_name_fqdn (avahi);
 190     printf("FQDN: %s (Error Return: %s)\n", ret, ret ? "OK" : avahi_strerror(avahi_client_errno(avahi)));
 191
 192     cookie = avahi_client_get_local_service_cookie(avahi);
 193     printf("Local service cookie: %u (Error Return: %s)\n", cookie, cookie != AVAHI_SERVICE_COOKIE_INVALID ? "OK" : avahi_strerror(avahi_client_errno(avahi)));
 194
 195     group = avahi_entry_group_new (avahi, avahi_entry_group_callback, "omghai");
 196     printf("Creating entry group: %s\n", group ? "OK" : avahi_strerror(avahi_client_errno (avahi)));
 197
 198     assert(group);
 199
 200     printf("Sucessfully created entry group %p\n", (void*) group);
 201
 202     avahi_entry_group_add_service (group, AVAHI_IF_UNSPEC, AF_UNSPEC, "Lathiat's Site", "_http._tcp", NULL, NULL, 80, "foo=bar", NULL);
 203
 204     avahi_entry_group_commit (group);
 205
 206     domain = avahi_domain_browser_new (avahi, AVAHI_IF_UNSPEC, AF_UNSPEC, NULL, AVAHI_DOMAIN_BROWSER_BROWSE, avahi_domain_browser_callback, "omghai3u");
 207
 208     if (domain == NULL)
 209         printf ("Failed to create domain browser object\n");
 210     else
 211         printf ("Sucessfully created domain browser %p\n", (void*) domain);
 212
 213     st = avahi_service_type_browser_new (avahi, AVAHI_IF_UNSPEC, AF_UNSPEC, "", avahi_service_type_browser_callback, "omghai3u");
 214     if (st == NULL)
 215         printf ("Failed to create service type browser object\n");
 216     else
 217         printf ("Sucessfully created service type browser %p\n", (void*) st);
 218
 219     sb = avahi_service_browser_new (avahi, AVAHI_IF_UNSPEC, AF_UNSPEC, "_http._tcp", "", avahi_service_browser_callback, "omghai3u");
 220     if (sb == NULL)
 221         printf ("Failed to create service browser object\n");
 222     else
 223         printf ("Sucessfully created service browser %p\n", (void*) sb);
 224
 225     hnr = avahi_host_name_resolver_new (avahi, AVAHI_IF_UNSPEC, AF_UNSPEC, "ecstasy.local", AF_UNSPEC, avahi_host_name_resolver_callback, "omghai4u");
 226     if (hnr == NULL)
 227         printf ("Failed to create hostname resolver object\n");
 228     else
 229         printf ("Successfully created hostname resolver object\n");
 230
 231
 232     avahi_elapse_time(&tv, 8000, 0);
 233     poll_api->timeout_new(poll_api, &tv, test_entry_group_reset, group);
 234     avahi_elapse_time(&tv, 20000, 0);
 235     poll_api->timeout_new(poll_api, &tv, test_free_entry_group, group);
 236     avahi_elapse_time(&tv, 25000, 0);
 237     poll_api->timeout_new(poll_api, &tv, test_free_domain_browser, sb);
 238
 239     avahi_elapse_time(&tv, 30000, 0);
 240     poll_api->timeout_new(poll_api, &tv, terminate, NULL);
 241
 242     for (;;)
 243         if (avahi_simple_poll_iterate(simple_poll, -1) != 0)
 244             break;
 245
 246     printf("terminating...\n");
 247
 248 fail:
 249
 250     if (avahi)
 251         avahi_client_free (avahi);
 252
 253     if (simple_poll)
 254         avahi_simple_poll_free(simple_poll);
 255
 256     return 0;
 257 }