From f1b541eb614861312f917dab1bf75ad8267e321a Mon Sep 17 00:00:00 2001 From: Lennart Poettering Date: Mon, 23 May 2005 16:39:01 +0000 Subject: [PATCH] * add new tool avahi-discover git-svn-id: file:///home/lennart/svn/public/avahi/trunk@91 941a03a8-eaeb-0310-b9a0-b1bbd8fe43fe --- Makefile.am | 2 +- avahi-core/util.c | 2 +- avahi-core/util.h | 2 +- avahi-discover/Makefile.am | 35 ++++ avahi-discover/avahi-discover.glade | 81 ++++++++ avahi-discover/avahi-discover.gladep | 8 + avahi-discover/main.c | 281 +++++++++++++++++++++++++++ configure.ac | 12 +- 8 files changed, 419 insertions(+), 4 deletions(-) create mode 100644 avahi-discover/Makefile.am create mode 100644 avahi-discover/avahi-discover.glade create mode 100644 avahi-discover/avahi-discover.gladep create mode 100644 avahi-discover/main.c diff --git a/Makefile.am b/Makefile.am index fce91f8..26124a8 100644 --- a/Makefile.am +++ b/Makefile.am @@ -18,7 +18,7 @@ # USA. EXTRA_DIST = bootstrap.sh LICENSE doxygen/Makefile.am doxygen/Makefile.in doxygen/doxygen.conf.in libtool.m4 -SUBDIRS = avahi-core +SUBDIRS = avahi-core avahi-discover pkgconfigdir = $(libdir)/pkgconfig pkgconfig_DATA = avahi-core.pc diff --git a/avahi-core/util.c b/avahi-core/util.c index 8dd9e64..b186817 100644 --- a/avahi-core/util.c +++ b/avahi-core/util.c @@ -381,7 +381,7 @@ void avahi_hexdump(gconstpointer p, guint size) { } } -gint avahi_domain_hash(const gchar *s) { +guint avahi_domain_hash(const gchar *s) { guint hash = 0; for (;;) { diff --git a/avahi-core/util.h b/avahi-core/util.h index 45b4ce7..ef87f5f 100644 --- a/avahi-core/util.h +++ b/avahi-core/util.h @@ -50,6 +50,6 @@ gchar *avahi_unescape_label(const gchar **name, gchar *dest, guint size); /* Escape the domain name in *src and write it to *ret_name */ gchar *avahi_escape_label(const guint8* src, guint src_length, gchar **ret_name, guint *ret_size); -gint avahi_domain_hash(const gchar *s); +guint avahi_domain_hash(const gchar *s); #endif diff --git a/avahi-discover/Makefile.am b/avahi-discover/Makefile.am new file mode 100644 index 0000000..9eca34d --- /dev/null +++ b/avahi-discover/Makefile.am @@ -0,0 +1,35 @@ +# $Id$ +# +# This file is part of avahi. +# +# avahi is free software; you can redistribute it and/or modify it +# under the terms of the GNU Lesser General Public License as +# published by the Free Software Foundation; either version 2 of the +# License, or (at your option) any later version. +# +# avahi is distributed in the hope that it will be useful, but WITHOUT +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY +# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public +# License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with avahi; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 +# USA. + +AM_CFLAGS=-I$(top_srcdir) -D_GNU_SOURCE + +AM_CFLAGS+=$(GLIB20_CFLAGS) $(GTK20_CFLAGS) $(GLADE20_CFLAGS) +AM_LDADD=$(GLIB20_LIBS) $(GTK20_LIBS) $(GLADE20_LIBS) + +# This cool debug trap works on i386/gcc only +AM_CFLAGS+='-DDEBUG_TRAP=__asm__("int $$3")' + +bin_PROGRAMS = \ + avahi-discover + +avahi_discover_SOURCES = \ + main.c + +avahi_discover_CFLAGS = $(AM_CFLAGS) +avahi_discover_LDADD = $(AM_LDADD) ../avahi-core/libavahi-core.la diff --git a/avahi-discover/avahi-discover.glade b/avahi-discover/avahi-discover.glade new file mode 100644 index 0000000..9465036 --- /dev/null +++ b/avahi-discover/avahi-discover.glade @@ -0,0 +1,81 @@ + + + + + + + True + Avahi Discovery + GTK_WINDOW_TOPLEVEL + GTK_WIN_POS_NONE + False + 500 + 400 + True + False + True + False + False + GDK_WINDOW_TYPE_HINT_NORMAL + GDK_GRAVITY_NORTH_WEST + + + + 12 + True + False + 12 + + + + True + True + GTK_POLICY_AUTOMATIC + GTK_POLICY_AUTOMATIC + GTK_SHADOW_IN + GTK_CORNER_TOP_LEFT + + + + True + True + True + False + True + True + + + + + 0 + True + True + + + + + + True + True + <i>No service currently selected.</i> + False + True + GTK_JUSTIFY_LEFT + False + True + 0 + 0.5 + 0 + 0 + + + 0 + False + False + + + + + + + diff --git a/avahi-discover/avahi-discover.gladep b/avahi-discover/avahi-discover.gladep new file mode 100644 index 0000000..a9feaa6 --- /dev/null +++ b/avahi-discover/avahi-discover.gladep @@ -0,0 +1,8 @@ + + + + + Avahi-discover + avahi-discover + FALSE + diff --git a/avahi-discover/main.c b/avahi-discover/main.c new file mode 100644 index 0000000..2f702a7 --- /dev/null +++ b/avahi-discover/main.c @@ -0,0 +1,281 @@ +#include + +#include +#include +#include +#include + +struct ServiceType; + +struct Service { + struct ServiceType *service_type; + gchar *service_name; + gchar *domain_name; + + gint interface; + guchar protocol; + + GtkTreeRowReference *tree_ref; +}; + +struct ServiceType { + gchar *service_type; + AvahiServiceBrowser *browser; + + GList *services; + GtkTreeRowReference *tree_ref; +}; + +static GtkWidget *main_window = NULL; +static GtkTreeView *tree_view = NULL; +static GtkTreeStore *tree_store = NULL; +static GtkLabel *info_label = NULL; +static AvahiServer *server = NULL; +static AvahiServiceTypeBrowser *service_type_browser = NULL; +static GHashTable *service_type_hash_table = NULL; +static AvahiServiceResolver *service_resolver = NULL; +static struct Service *current_service = NULL; + +/* very, very ugly: just import these two internal but useful functions from libavahi-core by hand */ +guint avahi_domain_hash(const gchar *s); +gboolean avahi_domain_equal(const gchar *a, const gchar *b); + +static struct Service *get_service(const gchar *service_type, const gchar *service_name, const gchar*domain_name, gint interface, guchar protocol) { + struct ServiceType *st; + GList *l; + + if (!(st = g_hash_table_lookup(service_type_hash_table, service_type))) + return NULL; + + for (l = st->services; l; l = l->next) { + struct Service *s = l->data; + + if (s->interface == interface && + s->protocol == protocol && + avahi_domain_equal(s->service_name, service_name) && + avahi_domain_equal(s->domain_name, domain_name)) + return s; + } + + return NULL; +} + +static void free_service(struct Service *s) { + GtkTreeIter iter; + GtkTreePath *path; + + if (current_service == s) { + current_service = NULL; + + if (service_resolver) { + avahi_service_resolver_free(service_resolver); + service_resolver = NULL; + } + + gtk_label_set_text(info_label, "Service removed"); + } + + s->service_type->services = g_list_remove(s->service_type->services, s); + + path = gtk_tree_row_reference_get_path(s->tree_ref); + gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &iter, path); + gtk_tree_path_free(path); + + gtk_tree_store_remove(tree_store, &iter); + + gtk_tree_row_reference_free(s->tree_ref); + + g_free(s->service_name); + g_free(s->domain_name); + g_free(s); +} + +static void service_browser_callback(AvahiServiceBrowser *b, gint interface, guchar protocol, AvahiBrowserEvent event, const gchar *service_name, const gchar *service_type, const gchar *domain_name, gpointer userdata) { + + if (event == AVAHI_BROWSER_NEW) { + struct Service *s; + GtkTreeIter iter, piter; + GtkTreePath *path, *ppath; + gchar iface[256]; + + s = g_new(struct Service, 1); + s->service_name = g_strdup(service_name); + s->domain_name = g_strdup(domain_name); + s->interface = interface; + s->protocol = protocol; + s->service_type = g_hash_table_lookup(service_type_hash_table, service_type); + g_assert(s->service_type); + + + + s->service_type->services = g_list_prepend(s->service_type->services, s); + + ppath = gtk_tree_row_reference_get_path(s->service_type->tree_ref); + gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &piter, ppath); + + snprintf(iface, sizeof(iface), "#%i %s", interface, protocol == AF_INET ? "IPv4" : "IPv6"); + + gtk_tree_store_append(tree_store, &iter, &piter); + gtk_tree_store_set(tree_store, &iter, 0, s->service_name, 1, iface, 2, s, -1); + + path = gtk_tree_model_get_path(GTK_TREE_MODEL(tree_store), &iter); + s->tree_ref = gtk_tree_row_reference_new(GTK_TREE_MODEL(tree_store), path); + gtk_tree_path_free(path); + + gtk_tree_view_expand_row(tree_view, ppath, FALSE); + gtk_tree_path_free(ppath); + + + } else if (event == AVAHI_BROWSER_REMOVE) { + struct Service* s; + + if ((s = get_service(service_type, service_name, domain_name, interface, protocol))) + free_service(s); + } +} + +static void service_type_browser_callback(AvahiServiceTypeBrowser *b, gint interface, guchar protocol, AvahiBrowserEvent event, const gchar *service_type, const gchar *domain, gpointer userdata) { + struct ServiceType *st; + GtkTreePath *path; + GtkTreeIter iter; + + if (event != AVAHI_BROWSER_NEW) + return; + + if (g_hash_table_lookup(service_type_hash_table, service_type)) + return; + + st = g_new(struct ServiceType, 1); + st->service_type = g_strdup(service_type); + st->services = NULL; + + gtk_tree_store_append(tree_store, &iter, NULL); + gtk_tree_store_set(tree_store, &iter, 0, st->service_type, 1, "", 2, NULL, -1); + + path = gtk_tree_model_get_path(GTK_TREE_MODEL(tree_store), &iter); + st->tree_ref = gtk_tree_row_reference_new(GTK_TREE_MODEL(tree_store), path); + gtk_tree_path_free(path); + + g_hash_table_insert(service_type_hash_table, st->service_type, st); + + st->browser = avahi_service_browser_new(server, -1, AF_UNSPEC, st->service_type, domain, service_browser_callback, NULL); +} + +static void update_label(struct Service *s, const gchar *hostname, const AvahiAddress *a, guint16 port, AvahiStringList *txt) { + gchar t[512], address[64], *txt_s; + + if (a && hostname) { + char na[256]; + avahi_address_snprint(na, sizeof(na), a); + snprintf(address, sizeof(address), "%s/%s:%u", hostname, na, port); + txt_s = avahi_string_list_to_string(txt); + } else { + snprintf(address, sizeof(address), "n/a"); + txt_s = g_strdup("n/a"); + } + + snprintf(t, sizeof(t), + "Service Type: %s\n" + "Service Name: %s\n" + "Domain Name: %s\n" + "Interface: %i %s\n" + "Address: %s\n" + "TXT Data: %s", + s->service_type->service_type, + s->service_name, + s->domain_name, + s->interface, + s->protocol == AF_INET ? "IPv4" : "IPv6", + address, + txt_s); + + gtk_label_set_markup(info_label, t); + + g_free(txt_s); +} + +static struct Service *get_service_on_cursor(void) { + GtkTreePath *path; + struct Service *s; + GtkTreeIter iter; + + gtk_tree_view_get_cursor(tree_view, &path, NULL); + gtk_tree_model_get_iter(GTK_TREE_MODEL(tree_store), &iter, path); + gtk_tree_model_get(GTK_TREE_MODEL(tree_store), &iter, 2, &s, -1); + gtk_tree_path_free(path); + + return s; +} + + +static void service_resolver_callback(AvahiServiceResolver *r, gint interface, guchar 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) { + struct Service *s; + g_assert(r); + + if (!(s = get_service_on_cursor()) || userdata != s) { + g_assert(r == service_resolver); + avahi_service_resolver_free(service_resolver); + service_resolver = NULL; + } + + if (event == AVAHI_RESOLVER_TIMEOUT) + gtk_label_set_text(info_label, "Failed to resolve."); + else + update_label(s, host_name, a, port, txt); +} + +static void tree_view_on_cursor_changed(GtkTreeView *tv, gpointer userdata) { + struct Service *s; + + if (!(s = get_service_on_cursor())) + return; + + if (service_resolver) + avahi_service_resolver_free(service_resolver); + + update_label(s, NULL, NULL, 0, NULL); + + service_resolver = avahi_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); +} + +gboolean main_window_on_delete_event(GtkWidget *widget, GdkEvent *event, gpointer user_data) { + gtk_main_quit(); + return FALSE; +} + +int main(int argc, char *argv[]) { + GladeXML *xml; + AvahiServerConfig config; + + gtk_init(&argc, &argv); + glade_init(); + + xml = glade_xml_new("avahi-discover.glade", NULL, NULL); + main_window = glade_xml_get_widget(xml, "main_window"); + g_signal_connect(main_window, "delete-event", (GCallback) main_window_on_delete_event, NULL); + + tree_view = GTK_TREE_VIEW(glade_xml_get_widget(xml, "tree_view")); + g_signal_connect(GTK_WIDGET(tree_view), "cursor-changed", (GCallback) tree_view_on_cursor_changed, NULL); + + info_label = GTK_LABEL(glade_xml_get_widget(xml, "info_label")); + + tree_store = gtk_tree_store_new(3, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_POINTER); + gtk_tree_view_set_model(tree_view, GTK_TREE_MODEL(tree_store)); + gtk_tree_view_insert_column_with_attributes(tree_view, -1, "Name", gtk_cell_renderer_text_new(), "text", 0, NULL); + gtk_tree_view_insert_column_with_attributes(tree_view, -1, "Interface", gtk_cell_renderer_text_new(), "text", 1, NULL); + + service_type_hash_table = g_hash_table_new((GHashFunc) avahi_domain_hash, (GEqualFunc) avahi_domain_equal); + + avahi_server_config_init(&config); + config.register_hinfo = config.register_addresses = config.announce_domain = FALSE; + server = avahi_server_new(NULL, &config, NULL, NULL); + avahi_server_config_free(&config); + + service_type_browser = avahi_service_type_browser_new(server, -1, AF_UNSPEC, argc >= 2 ? argv[1] : NULL, service_type_browser_callback, NULL); + + gtk_main(); + + avahi_server_free(server); + + return 0; +} diff --git a/configure.ac b/configure.ac index 4e58fb0..ca6a5e0 100644 --- a/configure.ac +++ b/configure.ac @@ -62,6 +62,16 @@ PKG_CHECK_MODULES(GLIB20, [ glib-2.0 >= 2.4.0 ]) AC_SUBST(GLIB20_CFLAGS) AC_SUBST(GLIB20_LIBS) +# Check for GTK 2.0 +PKG_CHECK_MODULES(GLIB20, [ gtk+-2.0 >= 2.4.0 ]) +AC_SUBST(GTK20_CFLAGS) +AC_SUBST(GTK20_LIBS) + +# Check for GLADE 2.0 +PKG_CHECK_MODULES(GLADE20, [ libglade-2.0 >= 2.4.0 ]) +AC_SUBST(GLADE20_CFLAGS) +AC_SUBST(GLADE20_LIBS) + # Check for DBUS PKG_CHECK_MODULES(DBUS, [ dbus-1 >= 0.23, dbus-glib-1 >= 0.23 ]) AC_SUBST(DBUS_CFLAGS) @@ -76,5 +86,5 @@ if test "x$GCC" = "xyes" ; then rm -f conftest.o fi -AC_CONFIG_FILES([Makefile avahi-core.pc doxygen/Makefile doxygen/doxygen.conf avahi-core/Makefile daemon/Makefile]) +AC_CONFIG_FILES([Makefile avahi-core.pc doxygen/Makefile doxygen/doxygen.conf avahi-core/Makefile daemon/Makefile avahi-discover/Makefile]) AC_OUTPUT -- 2.39.5