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
  31 #include <avahi-core/core.h>
  32
  33 static AvahiEntryGroup *group = NULL;
  34 static AvahiServer *server = NULL;
  35 static gchar *service_name = NULL;
  36
  37 static gboolean quit_timeout(gpointer data) {
  38     g_main_loop_quit(data);
  39     return FALSE;
  40 }
  41
  42 static gboolean dump_timeout(gpointer data) {
  43     AvahiServer *Avahi = data;
  44     avahi_server_dump(Avahi, stdout);
  45     return TRUE;
  46 }
  47
  48 static void record_browser_callback(AvahiRecordBrowser *r, gint interface, guchar protocol, AvahiBrowserEvent event, AvahiRecord *record, gpointer userdata) {
  49     gchar *t;
  50
  51     g_assert(r);
  52     g_assert(record);
  53     g_assert(interface > 0);
  54     g_assert(protocol != AF_UNSPEC);
  55
  56     g_message("SUBSCRIPTION: record [%s] on %i.%i is %s", t = avahi_record_to_string(record), interface, protocol,
  57               event == AVAHI_BROWSER_NEW ? "new" : "remove");
  58
  59     g_free(t);
  60 }
  61
  62 static void remove_entries(void);
  63 static void create_entries(gboolean new_name);
  64
  65 static void entry_group_callback(AvahiServer *s, AvahiEntryGroup *g, AvahiEntryGroupState state, gpointer userdata) {
  66     g_message("entry group state: %i", state);
  67
  68     if (state == AVAHI_ENTRY_GROUP_COLLISION) {
  69         remove_entries();
  70         create_entries(TRUE);
  71         g_message("Service name conflict, retrying with <%s>", service_name);
  72     } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
  73         g_message("Service established under name <%s>", service_name);
  74     }
  75 }
  76
  77 static void server_callback(AvahiServer *s, AvahiServerState state, gpointer userdata) {
  78
  79      g_message("server state: %i", state);
  80
  81     if (state == AVAHI_SERVER_RUNNING) {
  82         g_message("Server startup complete.  Host name is <%s>", avahi_server_get_host_name_fqdn(s));
  83         create_entries(FALSE);
  84     } else if (state == AVAHI_SERVER_COLLISION) {
  85         gchar *n;
  86         remove_entries();
  87
  88         n = avahi_alternative_host_name(avahi_server_get_host_name(s));
  89
  90         g_message("Host name conflict, retrying with <%s>", n);
  91         avahi_server_set_host_name(s, n);
  92         g_free(n);
  93     }
  94 }
  95
  96 static void remove_entries(void) {
  97     if (group)
  98         avahi_entry_group_free(group);
  99
 100     group = NULL;
 101 }
 102
 103 static void create_entries(gboolean new_name) {
 104     remove_entries();
 105
 106     group = avahi_entry_group_new(server, entry_group_callback, NULL);
 107
 108     if (!service_name)
 109         service_name = g_strdup("Test Service");
 110     else if (new_name) {
 111         gchar *n = avahi_alternative_service_name(service_name);
 112         g_free(service_name);
 113         service_name = n;
 114     }
 115
 116     avahi_server_add_service(server, group, 0, AF_UNSPEC, "_http._tcp", service_name, NULL, NULL, 80, "foo", NULL);
 117     avahi_server_add_service(server, group, 0, AF_UNSPEC, "_ftp._tcp", service_name, NULL, NULL, 21, "foo", NULL);
 118     avahi_server_add_service(server, group, 0, AF_UNSPEC, "_webdav._tcp", service_name, NULL, NULL, 80, "foo", NULL);
 119
 120     avahi_entry_group_commit(group);
 121
 122 }
 123
 124 static void hnr_callback(AvahiHostNameResolver *r, gint iface, guchar protocol, AvahiBrowserEvent event, const gchar *hostname, const AvahiAddress *a, gpointer userdata) {
 125     gchar t[64];
 126
 127     if (a)
 128         avahi_address_snprint(t, sizeof(t), a);
 129
 130     g_message("HNR: (%i.%i) <%s> -> %s [%s]", iface, protocol, hostname, a ? t : "n/a", event == AVAHI_RESOLVER_FOUND ? "found" : "timeout");
 131 }
 132
 133 static void ar_callback(AvahiAddressResolver *r, gint iface, guchar protocol, AvahiBrowserEvent event, const AvahiAddress *a, const gchar *hostname, gpointer userdata) {
 134     gchar t[64];
 135
 136     avahi_address_snprint(t, sizeof(t), a);
 137
 138     g_message("AR: (%i.%i) %s -> <%s> [%s]", iface, protocol, t, hostname ? hostname : "n/a", event == AVAHI_RESOLVER_FOUND ? "found" : "timeout");
 139 }
 140
 141 static void db_callback(AvahiDomainBrowser *b, gint iface, guchar protocol, AvahiBrowserEvent event, const gchar *domain, gpointer userdata) {
 142
 143     g_message("DB: (%i.%i) <%s> [%s]", iface, protocol, domain, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 144 }
 145
 146 static void stb_callback(AvahiServiceTypeBrowser *b, gint iface, guchar protocol, AvahiBrowserEvent event, const gchar *service_type, const gchar *domain, gpointer userdata) {
 147
 148     g_message("STB: (%i.%i) %s in <%s> [%s]", iface, protocol, service_type, domain, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 149 }
 150
 151 static void sb_callback(AvahiServiceBrowser *b, gint iface, guchar protocol, AvahiBrowserEvent event, const gchar *name, const gchar *service_type, const gchar *domain, gpointer userdata) {
 152    g_message("SB: (%i.%i) <%s> as %s in <%s> [%s]", iface, protocol, name, service_type, domain, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 153 }
 154
 155
 156 static void sr_callback(AvahiServiceResolver *r, gint iface, guchar protocol, AvahiBrowserEvent event, const gchar *service_name, const gchar*service_type, const gchar*domain_name, const gchar*hostname, const AvahiAddress *a, guint16 port, AvahiStringList *txt, gpointer userdata) {
 157
 158     if (event == AVAHI_RESOLVER_TIMEOUT)
 159         g_message("SR: (%i.%i) <%s> as %s in <%s> [timeout]", iface, protocol, service_name, service_type, domain_name);
 160     else {
 161         gchar t[64], *s;
 162
 163         avahi_address_snprint(t, sizeof(t), a);
 164
 165         s = avahi_string_list_to_string(txt);
 166         g_message("SR: (%i.%i) <%s> as %s in <%s>: %s/%s:%i (%s) [found]", iface, protocol, service_name, service_type, domain_name, hostname, t, port, s);
 167         g_free(s);
 168     }
 169 }
 170
 171 int main(int argc, char *argv[]) {
 172     GMainLoop *loop = NULL;
 173     AvahiRecordBrowser *r;
 174     AvahiHostNameResolver *hnr;
 175     AvahiAddressResolver *ar;
 176     AvahiKey *k;
 177     AvahiServerConfig config;
 178     AvahiAddress a;
 179     AvahiDomainBrowser *db;
 180     AvahiServiceTypeBrowser *stb;
 181     AvahiServiceBrowser *sb;
 182     AvahiServiceResolver *sr;
 183
 184     avahi_server_config_init(&config);
 185 /*     config.host_name = g_strdup("test"); */
 186     server = avahi_server_new(NULL, &config, server_callback, NULL);
 187     avahi_server_config_free(&config);
 188
 189     k = avahi_key_new("_http._tcp.local", AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR);
 190     r = avahi_record_browser_new(server, -1, AF_UNSPEC, k, record_browser_callback, NULL);
 191     avahi_key_unref(k);
 192
 193     hnr = avahi_host_name_resolver_new(server, -1, AF_UNSPEC, "codes-CompUTER.local", AF_UNSPEC, hnr_callback, NULL);
 194
 195     ar = avahi_address_resolver_new(server, -1, AF_UNSPEC, avahi_address_parse("192.168.50.15", AF_INET, &a), ar_callback, NULL);
 196
 197     db = avahi_domain_browser_new(server, -1, AF_UNSPEC, NULL, AVAHI_DOMAIN_BROWSER_BROWSE, db_callback, NULL);
 198
 199     stb = avahi_service_type_browser_new(server, -1, AF_UNSPEC, NULL, stb_callback, NULL);
 200
 201     sb = avahi_service_browser_new(server, -1, AF_UNSPEC, "_http._tcp", NULL, sb_callback, NULL);
 202
 203     sr = avahi_service_resolver_new(server, -1, AF_UNSPEC, "Ecstasy HTTP", "_http._tcp", "local", AF_UNSPEC, sr_callback, NULL);
 204
 205     loop = g_main_loop_new(NULL, FALSE);
 206
 207     g_timeout_add(1000*5, dump_timeout, server);
 208     g_timeout_add(1000*60, quit_timeout, loop);
 209
 210     g_main_loop_run(loop);
 211     g_main_loop_unref(loop);
 212
 213     avahi_record_browser_free(r);
 214     avahi_host_name_resolver_free(hnr);
 215     avahi_address_resolver_free(ar);
 216     avahi_service_type_browser_free(stb);
 217     avahi_service_browser_free(sb);
 218     avahi_service_resolver_free(sr);
 219
 220     if (group)
 221         avahi_entry_group_free(group);
 222
 223     if (server)
 224         avahi_server_free(server);
 225
 226     g_free(service_name);
 227
 228     return 0;
 229 }