4 This file is part of avahi.
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.
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.
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
26 #include <sys/ioctl.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>
43 struct ServiceType *service_type;
47 AvahiIfIndex interface;
48 AvahiProtocol protocol;
50 GtkTreeRowReference *tree_ref;
55 AvahiSServiceBrowser *browser;
58 GtkTreeRowReference *tree_ref;
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;
71 static const char* getifname(int idx) {
73 static struct ifreq ifreq;
76 memset(&ifreq, 0, sizeof(ifreq));
78 if ((fd = socket(PF_INET, SOCK_DGRAM, 0)) < 0)
79 if ((fd = socket(PF_INET6, SOCK_DGRAM, 0)) < 0)
82 ifreq.ifr_ifindex = idx;
83 if (ioctl(fd, SIOCGIFNAME, &ifreq) < 0)
88 return ifreq.ifr_name;
95 snprintf(t, sizeof(t), "#%i", idx);
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;
103 if (!(st = g_hash_table_lookup(service_type_hash_table, service_type)))
106 for (l = st->services; l; l = l->next) {
107 struct Service *s = l->data;
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))
119 static void free_service(struct Service *s) {
123 if (current_service == s) {
124 current_service = NULL;
126 if (service_resolver) {
127 avahi_s_service_resolver_free(service_resolver);
128 service_resolver = NULL;
131 gtk_label_set_text(info_label, "<i>Service removed</i>");
134 s->service_type->services = g_list_remove(s->service_type->services, s);
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);
141 gtk_tree_store_remove(tree_store, &iter);
143 gtk_tree_row_reference_free(s->tree_ref);
145 g_free(s->service_name);
146 g_free(s->domain_name);
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,
161 if (event == AVAHI_BROWSER_NEW) {
163 GtkTreeIter iter, piter;
164 GtkTreePath *path, *ppath;
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);
175 s->service_type->services = g_list_prepend(s->service_type->services, s);
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);
180 snprintf(iface, sizeof(iface), "%s %s", getifname(interface), avahi_proto_to_string(protocol));
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);
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);
189 gtk_tree_view_expand_row(tree_view, ppath, FALSE);
190 gtk_tree_path_free(ppath);
193 } else if (event == AVAHI_BROWSER_REMOVE) {
196 if ((s = get_service(service_type, service_name, domain_name, interface, protocol)))
201 static void service_type_browser_callback(
202 AvahiSServiceTypeBrowser *b,
203 AvahiIfIndex interface,
204 AvahiProtocol protocol,
205 AvahiBrowserEvent event,
206 const char *service_type,
208 AvahiLookupResultFlags flags,
211 struct ServiceType *st;
215 if (event != AVAHI_BROWSER_NEW)
218 if (g_hash_table_lookup(service_type_hash_table, service_type))
221 st = g_new(struct ServiceType, 1);
222 st->service_type = g_strdup(service_type);
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);
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);
232 g_hash_table_insert(service_type_hash_table, st->service_type, st);
234 st->browser = avahi_s_service_browser_new(server, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, st->service_type, domain, 0, service_browser_callback, NULL);
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;
242 avahi_address_snprint(na, sizeof(na), a);
243 snprintf(address, sizeof(address), "%s/%s:%u", hostname, na, port);
246 txt_s = avahi_string_list_to_string(txt);
248 txt_s = g_strdup("<i>empty</i>");
250 snprintf(address, sizeof(address), "<i>n/a</i>");
251 txt_s = g_strdup("<i>n/a</i>");
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,
264 getifname(s->interface), avahi_proto_to_string(s->protocol),
268 gtk_label_set_markup(info_label, t);
273 static struct Service *get_service_on_cursor(void) {
278 gtk_tree_view_get_cursor(tree_view, &path, NULL);
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);
290 static void service_resolver_callback(
291 AvahiSServiceResolver *r,
292 AvahiIfIndex interface,
293 AvahiProtocol protocol,
294 AvahiResolverEvent event,
298 const char *host_name,
299 const AvahiAddress *a,
301 AvahiStringList *txt,
302 AvahiLookupResultFlags flags,
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;
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);
325 static void tree_view_on_cursor_changed(GtkTreeView *tv, gpointer userdata) {
328 if (!(s = get_service_on_cursor()))
331 if (service_resolver)
332 avahi_s_service_resolver_free(service_resolver);
334 update_label(s, NULL, NULL, 0, NULL);
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);
339 static gboolean main_window_on_delete_event(GtkWidget *widget, GdkEvent *event, gpointer user_data) {
344 int main(int argc, char *argv[]) {
346 AvahiServerConfig config;
347 GtkTreeViewColumn *c;
349 AvahiGLibPoll *poll_api;
351 gtk_init(&argc, &argv);
354 avahi_set_allocator(avahi_glib_allocator());
356 poll_api = avahi_glib_poll_new(NULL, G_PRIORITY_DEFAULT);
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);
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);
365 info_label = GTK_LABEL(glade_xml_get_widget(xml, "info_label"));
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);
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);
376 service_type_hash_table = g_hash_table_new((GHashFunc) avahi_domain_hash, (GEqualFunc) avahi_domain_equal);
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);
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);
389 avahi_server_free(server);
390 avahi_glib_poll_free(poll_api);