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 "core.h"
  32 #include "alternative.h"
  33
  34 static AvahiEntryGroup *group = NULL;
  35 static AvahiServer *server = NULL;
  36 static gchar *service_name = NULL;
  37
  38 static gboolean quit_timeout(gpointer data) {
  39     g_main_loop_quit(data);
  40     return FALSE;
  41 }
  42
  43 static gboolean dump_timeout(gpointer data) {
  44     AvahiServer *Avahi = data;
  45     avahi_server_dump(Avahi, stdout);
  46     return TRUE;
  47 }
  48
  49 static void record_resolver_callback(AvahiRecordResolver *r, gint interface, guchar protocol, AvahiBrowserEvent event, AvahiRecord *record, gpointer userdata) {
  50     gchar *t;
  51
  52     g_assert(r);
  53     g_assert(record);
  54     g_assert(interface > 0);
  55     g_assert(protocol != AF_UNSPEC);
  56
  57     g_message("SUBSCRIPTION: record [%s] on %i.%i is %s", t = avahi_record_to_string(record), interface, protocol,
  58               event == AVAHI_BROWSER_NEW ? "new" : "removed");
  59
  60     g_free(t);
  61 }
  62
  63
  64 static void remove_entries(void);
  65 static void create_entries(gboolean new_name);
  66
  67 static void entry_group_callback(AvahiServer *s, AvahiEntryGroup *g, AvahiEntryGroupState state, gpointer userdata) {
  68     g_message("entry group state: %i", state);
  69
  70     if (state == AVAHI_ENTRY_GROUP_COLLISION) {
  71         remove_entries();
  72         create_entries(TRUE);
  73         g_message("Service name conflict, retrying with <%s>", service_name);
  74     } else if (state == AVAHI_ENTRY_GROUP_ESTABLISHED) {
  75         g_message("Service established under name <%s>", service_name);
  76     }
  77 }
  78
  79 static void server_callback(AvahiServer *s, AvahiServerState state, gpointer userdata) {
  80
  81      g_message("server state: %i", state);
  82
  83     if (state == AVAHI_SERVER_RUNNING) {
  84         g_message("Server startup complete.  Host name is <%s>", avahi_server_get_host_name_fqdn(s));
  85         create_entries(FALSE);
  86     } else if (state == AVAHI_SERVER_COLLISION) {
  87         gchar *n;
  88         remove_entries();
  89
  90         n = avahi_alternative_host_name(avahi_server_get_host_name(s));
  91
  92         g_message("Host name conflict, retrying with <%s>", n);
  93         avahi_server_set_host_name(s, n);
  94         g_free(n);
  95     }
  96 }
  97
  98 static void remove_entries(void) {
  99     if (group)
 100         avahi_entry_group_free(group);
 101
 102     group = NULL;
 103 }
 104
 105 static void create_entries(gboolean new_name) {
 106     remove_entries();
 107
 108     group = avahi_entry_group_new(server, entry_group_callback, NULL);
 109
 110     if (!service_name)
 111         service_name = g_strdup("Test Service");
 112     else if (new_name) {
 113         gchar *n = avahi_alternative_service_name(service_name);
 114         g_free(service_name);
 115         service_name = n;
 116     }
 117
 118     avahi_server_add_service(server, group, 0, AF_UNSPEC, "_http._tcp", service_name, NULL, NULL, 80, "foo", NULL);
 119     avahi_server_add_service(server, group, 0, AF_UNSPEC, "_ftp._tcp", service_name, NULL, NULL, 21, "foo", NULL);
 120     avahi_server_add_service(server, group, 0, AF_UNSPEC, "_webdav._tcp", service_name, NULL, NULL, 80, "foo", NULL);
 121
 122     avahi_entry_group_commit(group);
 123
 124 }
 125
 126 static void hnr_callback(AvahiHostNameResolver *r, gint iface, guchar protocol, AvahiBrowserEvent event, const gchar *hostname, const AvahiAddress *a, gpointer userdata) {
 127     gchar t[64];
 128
 129     avahi_address_snprint(t, sizeof(t), a);
 130
 131     g_message("HNR: (%i.%i) %s -> %s [%s]", iface, protocol, hostname, t, event == AVAHI_BROWSER_NEW ? "new" : "remove");
 132 }
 133
 134 int main(int argc, char *argv[]) {
 135     GMainLoop *loop = NULL;
 136     AvahiRecordResolver *r;
 137     AvahiHostNameResolver *hnr;
 138     AvahiKey *k;
 139     AvahiServerConfig config;
 140
 141     avahi_server_config_init(&config);
 142 /*     config.host_name = g_strdup("test"); */
 143     server = avahi_server_new(NULL, &config, server_callback, NULL);
 144     avahi_server_config_free(&config);
 145
 146     k = avahi_key_new("_http._tcp.local", AVAHI_DNS_CLASS_IN, AVAHI_DNS_TYPE_PTR);
 147     r = avahi_record_resolver_new(server, 0, AF_UNSPEC, k, record_resolver_callback, NULL);
 148     avahi_key_unref(k);
 149
 150     hnr = avahi_host_name_resolver_new(server, 0, AF_UNSPEC, "ecstasy.local", hnr_callback, NULL);
 151
 152     loop = g_main_loop_new(NULL, FALSE);
 153
 154 /*      g_timeout_add(1000*5, dump_timeout, server);   */
 155     g_timeout_add(1000*30, quit_timeout, loop);
 156
 157     g_main_loop_run(loop);
 158     g_main_loop_unref(loop);
 159
 160     avahi_record_resolver_free(r);
 161     avahi_host_name_resolver_free(hnr);
 162
 163     if (group)
 164         avahi_entry_group_free(group);
 165
 166     if (server)
 167         avahi_server_free(server);
 168
 169     g_free(service_name);
 170
 171     return 0;
 172 }