blob: 2f000a45348526d7cfb12646fa1e2adfd5ab784f [file] [log] [blame]
/***
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.1 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 Lesser 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.
***/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <sys/ioctl.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <net/if.h>
#include <unistd.h>
#include <gtk/gtk.h>
#include <avahi-core/core.h>
#include <avahi-core/lookup.h>
#include <avahi-common/strlst.h>
#include <avahi-common/domain.h>
#include <avahi-common/error.h>
#include <avahi-glib/glib-watch.h>
#include <avahi-glib/glib-malloc.h>
struct ServiceType;
struct Service {
struct ServiceType *service_type;
gchar *service_name;
gchar *domain_name;
AvahiIfIndex interface;
AvahiProtocol protocol;
GtkTreeRowReference *tree_ref;
};
struct ServiceType {
gchar *service_type;
AvahiSServiceBrowser *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 AvahiSServiceTypeBrowser *service_type_browser = NULL;
static GHashTable *service_type_hash_table = NULL;
static AvahiSServiceResolver *service_resolver = NULL;
static struct Service *current_service = NULL;
static struct Service *get_service(const gchar *service_type, const gchar *service_name, const gchar*domain_name, AvahiIfIndex interface, AvahiProtocol 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_s_service_resolver_free(service_resolver);
service_resolver = NULL;
}
gtk_label_set_text(info_label, "<i>Service removed</i>");
}
s->service_type->services = g_list_remove(s->service_type->services, s);
if ((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(
AVAHI_GCC_UNUSED AvahiSServiceBrowser *b,
AvahiIfIndex interface,
AvahiProtocol protocol,
AvahiBrowserEvent event,
const char *service_name,
const char *service_type,
const char *domain_name,
AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
AVAHI_GCC_UNUSED void* userdata) {
if (event == AVAHI_BROWSER_NEW) {
struct Service *s;
GtkTreeIter iter, piter;
GtkTreePath *path, *ppath;
gchar iface[256];
char name[IF_NAMESIZE];
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), "%s %s", if_indextoname(interface, name), avahi_proto_to_string(protocol));
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(
AVAHI_GCC_UNUSED AvahiSServiceTypeBrowser *b,
AVAHI_GCC_UNUSED AvahiIfIndex interface,
AVAHI_GCC_UNUSED AvahiProtocol protocol,
AvahiBrowserEvent event,
const char *service_type,
const char *domain,
AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
AVAHI_GCC_UNUSED void * 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_s_service_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, st->service_type, domain, 0, 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;
char name[IF_NAMESIZE];
if (a && hostname) {
char na[AVAHI_ADDRESS_STR_MAX];
avahi_address_snprint(na, sizeof(na), a);
snprintf(address, sizeof(address), "%s/%s:%u", hostname, na, port);
if (txt)
txt_s = avahi_string_list_to_string(txt);
else
txt_s = g_strdup("<i>empty</i>");
} else {
snprintf(address, sizeof(address), "<i>n/a</i>");
txt_s = g_strdup("<i>n/a</i>");
}
snprintf(t, sizeof(t),
"<b>Service Type:</b> %s\n"
"<b>Service Name:</b> %s\n"
"<b>Domain Name:</b> %s\n"
"<b>Interface:</b> %s %s\n"
"<b>Address:</b> %s\n"
"<b>TXT Data:</b> %s",
s->service_type->service_type,
s->service_name,
s->domain_name,
if_indextoname(s->interface,name), avahi_proto_to_string(s->protocol),
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);
if (!path)
return 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(
AvahiSServiceResolver *r,
AVAHI_GCC_UNUSED AvahiIfIndex interface,
AVAHI_GCC_UNUSED AvahiProtocol protocol,
AvahiResolverEvent event,
AVAHI_GCC_UNUSED const char *name,
AVAHI_GCC_UNUSED const char *type,
AVAHI_GCC_UNUSED const char *domain,
const char *host_name,
const AvahiAddress *a,
uint16_t port,
AvahiStringList *txt,
AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
void* userdata) {
struct Service *s;
g_assert(r);
if (!(s = get_service_on_cursor()) || userdata != s) {
g_assert(r == service_resolver);
avahi_s_service_resolver_free(service_resolver);
service_resolver = NULL;
return;
}
if (event == AVAHI_RESOLVER_FAILURE) {
char t[256];
snprintf(t, sizeof(t), "<i>Failed to resolve: %s.</i>", avahi_strerror(avahi_server_errno(server)));
gtk_label_set_markup(info_label, t);
} else if (event == AVAHI_RESOLVER_FOUND)
update_label(s, host_name, a, port, txt);
}
static void tree_view_on_cursor_changed(AVAHI_GCC_UNUSED GtkTreeView *tv, AVAHI_GCC_UNUSED gpointer userdata) {
struct Service *s;
if (!(s = get_service_on_cursor()))
return;
if (service_resolver)
avahi_s_service_resolver_free(service_resolver);
update_label(s, NULL, NULL, 0, NULL);
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);
}
static gboolean main_window_on_delete_event(AVAHI_GCC_UNUSED GtkWidget *widget, AVAHI_GCC_UNUSED GdkEvent *event, AVAHI_GCC_UNUSED gpointer user_data) {
gtk_main_quit();
return FALSE;
}
int main(int argc, char *argv[]) {
GtkBuilder *ui;
AvahiServerConfig config;
GtkTreeViewColumn *c;
gint error;
AvahiGLibPoll *poll_api;
gtk_init(&argc, &argv);
avahi_set_allocator(avahi_glib_allocator());
poll_api = avahi_glib_poll_new(NULL, G_PRIORITY_DEFAULT);
ui = gtk_builder_new();
gtk_builder_add_from_file(ui, AVAHI_INTERFACES_DIR"avahi-discover.ui", NULL);
main_window = GTK_WIDGET(gtk_builder_get_object(ui, "main_window"));
g_signal_connect(main_window, "delete-event", (GCallback) main_window_on_delete_event, NULL);
tree_view = GTK_TREE_VIEW(gtk_builder_get_object(ui, "tree_view"));
g_signal_connect(GTK_WIDGET(tree_view), "cursor-changed", (GCallback) tree_view_on_cursor_changed, NULL);
info_label = GTK_LABEL(gtk_builder_get_object(ui, "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);
gtk_tree_view_column_set_resizable(c = gtk_tree_view_get_column(tree_view, 0), TRUE);
gtk_tree_view_column_set_sizing(c, GTK_TREE_VIEW_COLUMN_GROW_ONLY);
gtk_tree_view_column_set_expand(c, TRUE);
service_type_hash_table = g_hash_table_new((GHashFunc) avahi_domain_hash, (GEqualFunc) avahi_domain_equal);
avahi_server_config_init(&config);
config.publish_hinfo = config.publish_addresses = config.publish_domain = config.publish_workstation = FALSE;
server = avahi_server_new(avahi_glib_poll_get(poll_api), &config, NULL, NULL, &error);
avahi_server_config_free(&config);
g_assert(server);
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);
gtk_main();
avahi_server_free(server);
avahi_glib_poll_free(poll_api);
return 0;
}