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