git.meshlink.io Git - catta/blob - avahi-discover-standalone/main.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 <string.h>
  27
  28 #include <gtk/gtk.h>
  29 #include <glade/glade.h>
  30 #include <avahi-core/core.h>
  31 #include <avahi-common/strlst.h>
  32 #include <avahi-common/domain.h>
  33 #include <avahi-glib/glib-watch.h>
  34 #include <avahi-glib/glib-malloc.h>
  35
  36 struct ServiceType;
  37
  38 struct Service {
  39     struct ServiceType *service_type;
  40     gchar *service_name;
  41     gchar *domain_name;
  42
  43     AvahiIfIndex interface;
  44     AvahiProtocol protocol;
  45
  46     GtkTreeRowReference *tree_ref;
  47 };
  48
  49 struct ServiceType {
  50     gchar *service_type;
  51     AvahiSServiceBrowser *browser;
  52
  53     GList *services;
  54     GtkTreeRowReference *tree_ref;
  55 };
  56
  57 static GtkWidget *main_window = NULL;
  58 static GtkTreeView *tree_view = NULL;
  59 static GtkTreeStore *tree_store = NULL;
  60 static GtkLabel *info_label = NULL;
  61 static AvahiServer *server = NULL;
  62 static AvahiSServiceTypeBrowser *service_type_browser = NULL;
  63 static GHashTable *service_type_hash_table = NULL;
  64 static AvahiSServiceResolver *service_resolver = NULL;
  65 static struct Service *current_service = NULL;
  66
  67 static struct Service *get_service(const gchar *service_type, const gchar *service_name, const gchar*domain_name, AvahiIfIndex interface, AvahiProtocol protocol) {
  68     struct ServiceType *st;
  69     GList *l;
  70
  71     if (!(st = g_hash_table_lookup(service_type_hash_table, service_type)))
  72         return NULL;
  73
  74     for (l = st->services; l; l = l->next) {
  75         struct Service *s = l->data;
  76
  77         if (s->interface == interface &&
  78             s->protocol == protocol &&
  79             avahi_domain_equal(s->service_name, service_name) &&
  80             avahi_domain_equal(s->domain_name, domain_name))
  81             return s;
  82     }
  83
  84     return NULL;
  85 }
  86
  87 static void free_service(struct Service *s) {
  88     GtkTreeIter iter;
  89     GtkTreePath *path;
  90
  91     if (current_service == s) {
  92         current_service = NULL;
  93
  94         if (service_resolver) {
  95             avahi_s_service_resolver_free(service_resolver);
  96             service_resolver = NULL;
  97         }
  98
  99         gtk_label_set_text(info_label, "<i>Service removed</i>");
 100     }
 101
 102     s->service_type->services = g_list_remove(s->service_type->services, s);
 103
 104     path = gtk_tree_row_reference_get_path(s->tree_ref);
 105     gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &iter, path);
 106     gtk_tree_path_free(path);
 107
 108     gtk_tree_store_remove(tree_store, &iter);
 109
 110     gtk_tree_row_reference_free(s->tree_ref);
 111
 112     g_free(s->service_name);
 113     g_free(s->domain_name);
 114     g_free(s);
 115 }
 116
 117 static void service_browser_callback(AvahiSServiceBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const gchar *service_name, const gchar *service_type, const gchar *domain_name, gpointer userdata) {
 118
 119     if (event == AVAHI_BROWSER_NEW) {
 120         struct Service *s;
 121         GtkTreeIter iter, piter;
 122         GtkTreePath *path, *ppath;
 123         gchar iface[256];
 124
 125         s = g_new(struct Service, 1);
 126         s->service_name = g_strdup(service_name);
 127         s->domain_name = g_strdup(domain_name);
 128         s->interface = interface;
 129         s->protocol = protocol;
 130         s->service_type = g_hash_table_lookup(service_type_hash_table, service_type);
 131         g_assert(s->service_type);
 132
 133         s->service_type->services = g_list_prepend(s->service_type->services, s);
 134
 135         ppath = gtk_tree_row_reference_get_path(s->service_type->tree_ref);
 136         gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &piter, ppath);
 137
 138         snprintf(iface, sizeof(iface), "#%i %s", interface, protocol == AF_INET ? "IPv4" : "IPv6");
 139
 140         gtk_tree_store_append(tree_store, &iter, &piter);
 141         gtk_tree_store_set(tree_store, &iter, 0, s->service_name, 1, iface, 2, s, -1);
 142
 143         path = gtk_tree_model_get_path(GTK_TREE_MODEL(tree_store), &iter);
 144         s->tree_ref = gtk_tree_row_reference_new(GTK_TREE_MODEL(tree_store), path);
 145         gtk_tree_path_free(path);
 146
 147         gtk_tree_view_expand_row(tree_view, ppath, FALSE);
 148         gtk_tree_path_free(ppath);
 149
 150
 151     } else if (event == AVAHI_BROWSER_REMOVE) {
 152         struct Service* s;
 153
 154         if ((s = get_service(service_type, service_name, domain_name, interface, protocol)))
 155             free_service(s);
 156     }
 157 }
 158
 159 static void service_type_browser_callback(AvahiSServiceTypeBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const gchar *service_type, const gchar *domain, gpointer userdata) {
 160     struct ServiceType *st;
 161     GtkTreePath *path;
 162     GtkTreeIter iter;
 163
 164     if (event != AVAHI_BROWSER_NEW)
 165         return;
 166
 167     if (g_hash_table_lookup(service_type_hash_table, service_type))
 168         return;
 169
 170     st = g_new(struct ServiceType, 1);
 171     st->service_type = g_strdup(service_type);
 172     st->services = NULL;
 173
 174     gtk_tree_store_append(tree_store, &iter, NULL);
 175     gtk_tree_store_set(tree_store, &iter, 0, st->service_type, 1, "", 2, NULL, -1);
 176
 177     path = gtk_tree_model_get_path(GTK_TREE_MODEL(tree_store), &iter);
 178     st->tree_ref = gtk_tree_row_reference_new(GTK_TREE_MODEL(tree_store), path);
 179     gtk_tree_path_free(path);
 180
 181     g_hash_table_insert(service_type_hash_table, st->service_type, st);
 182
 183     st->browser = avahi_s_service_browser_new(server, -1, AF_UNSPEC, st->service_type, domain, service_browser_callback, NULL);
 184 }
 185
 186 static void update_label(struct Service *s, const gchar *hostname, const AvahiAddress *a, guint16 port, AvahiStringList *txt) {
 187     gchar t[512], address[64], *txt_s;
 188
 189     if (a && hostname) {
 190         char na[256];
 191         avahi_address_snprint(na, sizeof(na), a);
 192         snprintf(address, sizeof(address), "%s/%s:%u", hostname, na, port);
 193         txt_s = avahi_string_list_to_string(txt);
 194     } else {
 195         snprintf(address, sizeof(address), "<i>n/a</i>");
 196         txt_s = g_strdup("<i>n/a</i>");
 197     }
 198
 199     snprintf(t, sizeof(t),
 200              "<b>Service Type:</b> %s\n"
 201              "<b>Service Name:</b> %s\n"
 202              "<b>Domain Name:</b> %s\n"
 203              "<b>Interface:</b> %i %s\n"
 204              "<b>Address:</b> %s\n"
 205              "<b>TXT Data:</b> %s",
 206              s->service_type->service_type,
 207              s->service_name,
 208              s->domain_name,
 209              s->interface,
 210              s->protocol == AF_INET ? "IPv4" : "IPv6",
 211              address,
 212              txt_s);
 213
 214     gtk_label_set_markup(info_label, t);
 215
 216     g_free(txt_s);
 217 }
 218
 219 static struct Service *get_service_on_cursor(void) {
 220     GtkTreePath *path;
 221     struct Service *s;
 222     GtkTreeIter iter;
 223
 224     gtk_tree_view_get_cursor(tree_view, &path, NULL);
 225     gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &iter, path);
 226     gtk_tree_model_get(GTK_TREE_MODEL(tree_store), &iter, 2, &s, -1);
 227     gtk_tree_path_free(path);
 228
 229     return s;
 230 }
 231
 232
 233 static void service_resolver_callback(AvahiSServiceResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const gchar *name, const gchar *type, const gchar *domain, const gchar *host_name, const AvahiAddress *a, guint16 port, AvahiStringList *txt, gpointer userdata) {
 234     struct Service *s;
 235     g_assert(r);
 236
 237     if (!(s = get_service_on_cursor()) || userdata != s) {
 238         g_assert(r == service_resolver);
 239         avahi_s_service_resolver_free(service_resolver);
 240         service_resolver = NULL;
 241     }
 242
 243     if (event == AVAHI_RESOLVER_TIMEOUT)
 244         gtk_label_set_markup(info_label, "<i>Failed to resolve.</i>");
 245     else
 246         update_label(s, host_name, a, port, txt);
 247 }
 248
 249 static void tree_view_on_cursor_changed(GtkTreeView *tv, gpointer userdata) {
 250     struct Service *s;
 251
 252     if (!(s = get_service_on_cursor()))
 253         return;
 254
 255     if (service_resolver)
 256         avahi_s_service_resolver_free(service_resolver);
 257
 258     update_label(s, NULL, NULL, 0, NULL);
 259
 260     service_resolver = avahi_s_service_resolver_new(server, s->interface, s->protocol, s->service_name, s->service_type->service_type, s->domain_name, AF_UNSPEC, service_resolver_callback, s);
 261 }
 262
 263 static gboolean main_window_on_delete_event(GtkWidget *widget, GdkEvent *event, gpointer user_data) {
 264     gtk_main_quit();
 265     return FALSE;
 266 }
 267
 268 int main(int argc, char *argv[]) {
 269     GladeXML *xml;
 270     AvahiServerConfig config;
 271     GtkTreeViewColumn *c;
 272     gint error;
 273     AvahiGLibPoll *poll_api;
 274
 275     gtk_init(&argc, &argv);
 276     glade_init();
 277
 278     avahi_set_allocator(avahi_glib_allocator());
 279
 280     poll_api = avahi_glib_poll_new(NULL, G_PRIORITY_DEFAULT);
 281
 282     xml = glade_xml_new(AVAHI_INTERFACES_DIR"avahi-discover.glade", NULL, NULL);
 283     main_window = glade_xml_get_widget(xml, "main_window");
 284     g_signal_connect(main_window, "delete-event", (GCallback) main_window_on_delete_event, NULL);
 285
 286     tree_view = GTK_TREE_VIEW(glade_xml_get_widget(xml, "tree_view"));
 287     g_signal_connect(GTK_WIDGET(tree_view), "cursor-changed", (GCallback) tree_view_on_cursor_changed, NULL);
 288
 289     info_label = GTK_LABEL(glade_xml_get_widget(xml, "info_label"));
 290
 291     tree_store = gtk_tree_store_new(3, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_POINTER);
 292     gtk_tree_view_set_model(tree_view, GTK_TREE_MODEL(tree_store));
 293     gtk_tree_view_insert_column_with_attributes(tree_view, -1, "Name", gtk_cell_renderer_text_new(), "text", 0, NULL);
 294     gtk_tree_view_insert_column_with_attributes(tree_view, -1, "Interface", gtk_cell_renderer_text_new(), "text", 1, NULL);
 295
 296     gtk_tree_view_column_set_resizable(c = gtk_tree_view_get_column(tree_view, 0), TRUE);
 297     gtk_tree_view_column_set_sizing(c, GTK_TREE_VIEW_COLUMN_GROW_ONLY);
 298     gtk_tree_view_column_set_expand(c, TRUE);
 299
 300     service_type_hash_table = g_hash_table_new((GHashFunc) avahi_domain_hash, (GEqualFunc) avahi_domain_equal);
 301
 302     avahi_server_config_init(&config);
 303     config.publish_hinfo = config.publish_addresses = config.publish_domain = config.publish_workstation = FALSE;
 304     server = avahi_server_new(avahi_glib_poll_get(poll_api), &config, NULL, NULL, &error);
 305     avahi_server_config_free(&config);
 306
 307     g_assert(server);
 308
 309     service_type_browser = avahi_s_service_type_browser_new(server, -1, AF_UNSPEC, argc >= 2 ? argv[1] : NULL, service_type_browser_callback, NULL);
 310
 311     gtk_main();
 312
 313     avahi_server_free(server);
 314     avahi_glib_poll_free(poll_api);
 315
 316     return 0;
 317 }