3 * GTK Tap implementation of stats_tree
4 * 2005, Luis E. G. Ontanon
8 * Wireshark - Network traffic analyzer
9 * By Gerald Combs <gerald@wireshark.org>
10 * Copyright 1998 Gerald Combs
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License
14 * as published by the Free Software Foundation; either version 2
15 * of the License, or (at your option) any later version.
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU General Public License for more details.
22 * You should have received a copy of the GNU General Public License
23 * along with this program; if not, write to the Free Software
24 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
34 #include <epan/stats_tree_priv.h>
35 #include <epan/report_err.h>
37 #include "../simple_dialog.h"
38 #include "../globals.h"
39 #include "../stat_menu.h"
41 #include "gtk/gui_utils.h"
42 #include "gtk/dlg_utils.h"
43 #include "gtk/tap_dfilter_dlg.h"
46 struct _st_node_pres {
50 struct _tree_cfg_pres {
51 tap_dfilter_dlg* stat_dlg;
61 /* the columns of the tree pane */
62 enum _stat_tree_columns {
70 /* used for converting numbers */
71 #define NUM_BUF_SIZE 32
73 /* creates the gtk representation for a stat_node
76 static void setup_gtk_node_pr(stat_node* node) {
77 GtkTreeIter* parent = NULL;
79 node->pr = g_malloc(sizeof(st_node_pres));
81 if (node->st->pr->store) {
82 node->pr->iter = g_malloc0(sizeof(GtkTreeIter));
84 if ( node->parent && node->parent->pr ) {
85 parent = node->parent->pr->iter;
87 gtk_tree_store_append (node->st->pr->store, node->pr->iter, parent);
88 gtk_tree_store_set(node->st->pr->store, node->pr->iter, TITLE_COLUMN, node->name, RATE_COLUMN, "", COUNT_COLUMN, "", -1);
93 static void draw_gtk_node(stat_node* node) {
94 static gchar value[NUM_BUF_SIZE];
95 static gchar rate[NUM_BUF_SIZE];
96 static gchar percent[NUM_BUF_SIZE];
99 stats_tree_get_strs_from_node(node, value, rate,
102 if (node->st->pr->store && node->pr->iter) {
103 gtk_tree_store_set(node->st->pr->store, node->pr->iter,
106 PERCENT_COLUMN, percent,
110 if (node->children) {
111 for (child = node->children; child; child = child->next )
112 draw_gtk_node(child);
116 static void draw_gtk_tree( void *psp ) {
117 stats_tree *st = psp;
120 for (child = st->root.children; child; child = child->next ) {
121 draw_gtk_node(child);
123 if (child->pr->iter && st->pr->store) {
124 gtk_tree_view_expand_row(GTK_TREE_VIEW(st->pr->tree),
125 gtk_tree_model_get_path(GTK_TREE_MODEL(st->pr->store),
133 void protect_thread_critical_region(void);
134 void unprotect_thread_critical_region(void);
136 static void free_gtk_tree(GtkWindow *win _U_, stats_tree *st)
139 protect_thread_critical_region();
140 remove_tap_listener(st);
141 unprotect_thread_critical_region();
144 st->root.pr->iter = NULL;
146 st->cfg->in_use = FALSE;
151 static void clear_node_pr(stat_node* n) {
153 for (c = n->children; c; c = c->next) {
158 gtk_tree_store_remove(n->st->pr->store, n->pr->iter);
163 static void reset_tap(void* p) {
166 for (c = st->root.children; c; c = c->next) {
173 /* initializes the stats_tree window */
174 static void init_gtk_tree(const char* optarg, void *userdata _U_) {
175 guint8* abbr = stats_tree_get_abbr(optarg);
176 stats_tree* st = NULL;
177 stats_tree_cfg* cfg = NULL;
178 tree_pres* pr = g_malloc(sizeof(tree_pres));
180 gchar* window_name = NULL;
181 GString* error_string;
184 GtkWidget *main_vb, *bbox, *bt_close;
185 GtkTreeViewColumn* column;
186 GtkCellRenderer* renderer;
189 cfg = stats_tree_get_cfg_by_abbr(abbr);
191 if (cfg && cfg->in_use) {
193 report_failure("cannot open more than one tree of the same type at once");
198 init_strlen = strlen(cfg->pr->stat_dlg->init_string);
200 if (strncmp (optarg, cfg->pr->stat_dlg->init_string, init_strlen) == 0){
201 if (init_strlen == strlen(optarg)) {
202 st = stats_tree_new(cfg,pr,NULL);
204 st = stats_tree_new(cfg,pr,(char*)optarg+init_strlen+1);
208 st = stats_tree_new(cfg,pr,NULL);
211 report_failure("no such stats_tree (%s) in stats_tree registry",abbr);
218 report_failure("could not obtain stats_tree abbr from optarg");
225 window_name = g_strdup_printf("%s Stats Tree", cfg->name);
227 st->pr->win = window_new_with_geom(GTK_WINDOW_TOPLEVEL,window_name,window_name);
228 gtk_window_set_default_size(GTK_WINDOW(st->pr->win), 400, 400);
232 title=g_strdup_printf("%s with filter: %s",cfg->name,st->filter);
235 title=g_strdup_printf("%s", cfg->name);
238 gtk_window_set_title(GTK_WINDOW(st->pr->win), title);
241 main_vb = gtk_vbox_new(FALSE, 3);
242 gtk_container_border_width(GTK_CONTAINER(main_vb), 12);
243 gtk_container_add(GTK_CONTAINER(st->pr->win), main_vb);
245 scr_win = scrolled_window_new(NULL, NULL);
247 st->pr->store = gtk_tree_store_new (N_COLUMNS, G_TYPE_STRING, G_TYPE_STRING,
248 G_TYPE_STRING, G_TYPE_STRING);
250 st->pr->tree = gtk_tree_view_new_with_model (GTK_TREE_MODEL (st->pr->store));
251 g_object_unref(G_OBJECT(st->pr->store));
253 gtk_container_add( GTK_CONTAINER(scr_win), st->pr->tree);
256 renderer = gtk_cell_renderer_text_new ();
257 column = gtk_tree_view_column_new_with_attributes ("Topic / Item", renderer,
258 "text", TITLE_COLUMN,
260 gtk_tree_view_column_set_resizable (column,TRUE);
261 gtk_tree_view_column_set_sizing(column,GTK_TREE_VIEW_COLUMN_AUTOSIZE);
262 gtk_tree_view_append_column (GTK_TREE_VIEW (st->pr->tree), column);
264 renderer = gtk_cell_renderer_text_new ();
265 column = gtk_tree_view_column_new_with_attributes ("Count", renderer,
266 "text", COUNT_COLUMN,
269 gtk_tree_view_column_set_resizable (column,TRUE);
270 gtk_tree_view_column_set_sizing(column,GTK_TREE_VIEW_COLUMN_AUTOSIZE);
271 gtk_tree_view_append_column (GTK_TREE_VIEW (st->pr->tree), column);
273 renderer = gtk_cell_renderer_text_new ();
274 column = gtk_tree_view_column_new_with_attributes ("Rate", renderer,
277 gtk_tree_view_column_set_resizable (column,TRUE);
278 gtk_tree_view_column_set_sizing(column,GTK_TREE_VIEW_COLUMN_AUTOSIZE);
279 gtk_tree_view_append_column (GTK_TREE_VIEW (st->pr->tree), column);
281 renderer = gtk_cell_renderer_text_new ();
282 column = gtk_tree_view_column_new_with_attributes ("Percent", renderer,
283 "text", PERCENT_COLUMN,
285 gtk_tree_view_column_set_resizable(column,TRUE);
286 gtk_tree_view_column_set_sizing(column,GTK_TREE_VIEW_COLUMN_AUTOSIZE);
287 gtk_tree_view_append_column (GTK_TREE_VIEW (st->pr->tree), column);
289 gtk_container_add( GTK_CONTAINER(main_vb), scr_win);
291 error_string = register_tap_listener( cfg->tapname,
299 /* error, we failed to attach to the tap. clean up */
300 simple_dialog( ESD_TYPE_ERROR, ESD_BTN_OK, error_string->str );
301 /* destroy_stat_tree_window(st); */
302 report_failure("stats_tree for: %s failed to attach to the tap: %s",cfg->name,error_string->str);
303 g_string_free(error_string, TRUE);
307 bbox = dlg_button_row_new(GTK_STOCK_CLOSE, NULL);
308 gtk_box_pack_start(GTK_BOX(main_vb), bbox, FALSE, FALSE, 0);
310 bt_close = g_object_get_data(G_OBJECT(bbox), GTK_STOCK_CLOSE);
311 window_set_cancel_button(st->pr->win, bt_close, window_cancel_button_cb);
313 g_signal_connect(GTK_WINDOW(st->pr->win), "delete_event", G_CALLBACK(window_delete_event_cb), NULL);
314 g_signal_connect(GTK_WINDOW(st->pr->win), "destroy", G_CALLBACK(free_gtk_tree), st);
316 gtk_widget_show_all(st->pr->win);
317 window_present(st->pr->win);
319 cf_retap_packets(&cfile, FALSE);
323 static void register_gtk_stats_tree_tap (gpointer k _U_, gpointer v, gpointer p _U_) {
324 stats_tree_cfg* cfg = v;
326 cfg->pr = g_malloc(sizeof(tree_pres));
328 cfg->pr->stat_dlg = g_malloc(sizeof(tap_dfilter_dlg));
330 cfg->pr->stat_dlg->win_title = g_strdup_printf("%s Stats Tree",cfg->name);
331 cfg->pr->stat_dlg->init_string = g_strdup_printf("%s,tree",cfg->abbr);
332 cfg->pr->stat_dlg->tap_init_cb = init_gtk_tree;
333 cfg->pr->stat_dlg->index = -1;
335 register_dfilter_stat(cfg->pr->stat_dlg, cfg->name,
336 REGISTER_STAT_GROUP_UNSORTED);
339 static void free_tree_presentation(stats_tree* st) {
344 register_tap_listener_stats_tree_stat(void)
347 stats_tree_presentation(register_gtk_stats_tree_tap,
353 free_tree_presentation,