add an userdata argument to register_stat_cmd_arg() and its callback to use the...
[obnox/wireshark/wip.git] / tap-stats_tree.c
1 /* tap-stats_tree.c
2  * tethereal's tap implememntation of stats_tree
3  * 2005, Luis E. G. Ontanon
4  *
5  * $Id$
6  *
7  * Ethereal - Network traffic analyzer
8  * By Gerald Combs <gerald@ethereal.com>
9  * Copyright 1998 Gerald Combs
10  * 
11  * This program is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU General Public License
13  * as published by the Free Software Foundation; either version 2
14  * of the License, or (at your option) any later version.
15  * 
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  * 
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
24  */
25
26 #ifdef HAVE_CONFIG_H
27 #include "config.h"
28 #endif
29
30 #include <string.h>
31 #include <stdio.h>
32 #include <glib.h>
33 #include <epan/stats_tree_priv.h>
34 #include <epan/stat_cmd_args.h>
35 #include <epan/report_err.h>
36
37 /* actually unused */
38 struct _st_node_pres {
39         void* dummy;
40 };
41
42 struct _tree_pres {
43         void* dummy;
44 };
45
46 struct _tree_cfg_pres {
47         guint8* init_string;    
48 };
49
50 static void draw_stats_tree(void *psp) {
51         stats_tree *st = psp;
52         GString* s;
53         gchar* fmt;
54         stat_node* child;
55         
56         s = g_string_new("\n===================================================================\n");
57         fmt = g_strdup_printf(" %%s%%-%us%%12s\t%%12s\t%%12s\n",stats_tree_branch_max_namelen(&st->root,0));
58         g_string_sprintfa(s,fmt,"",st->cfg->name,"value","rate","percent");
59         g_free(fmt);
60         g_string_sprintfa(s,"-------------------------------------------------------------------\n");
61         
62         for (child = st->root.children; child; child = child->next ) {
63                 stats_tree_branch_to_str(child,s,0);
64         }
65         
66         s = g_string_append(s,"\n===================================================================\n");
67         
68         printf("%s",s->str);
69         
70 }
71
72 static void  init_stats_tree(const char *optarg,void* userdata _U_) {
73         guint8* abbr = stats_tree_get_abbr(optarg);
74         GString *error_string;
75         stats_tree_cfg *cfg = NULL;
76         stats_tree* st = NULL;
77         
78         if (abbr) {
79                 cfg = stats_tree_get_cfg_by_abbr(abbr);
80
81                 if (cfg != NULL) {
82                         if (strncmp (optarg, cfg->pr->init_string, strlen(cfg->pr->init_string)) == 0){
83                                 st = stats_tree_new(cfg,NULL,((guint8*)optarg)+strlen(cfg->pr->init_string));
84                         } else {
85                                 st->filter=NULL;
86                         }
87                 } else {
88                         report_failure("no such stats_tree (%s) found in stats_tree registry",abbr);
89                         return;
90                 }
91                 
92                 g_free(abbr);
93                 
94         } else {
95                 report_failure("could not obtain stats_tree abbr (%s) from grg '%s'",abbr,optarg);              
96                 return;
97         }
98         
99         error_string = register_tap_listener( st->cfg->tapname,
100                                                                                   st,
101                                                                                   st->filter,
102                                                                                   stats_tree_reset,
103                                                                                   stats_tree_packet,
104                                                                                   draw_stats_tree);
105         
106         if (error_string) {
107                 report_failure("stats_tree for: %s failed to attach to the tap: %s",cfg->name,error_string->str);
108                 return;
109         }
110
111         if (cfg->init) cfg->init(st);
112
113 }
114
115 void register_stats_tree_tap (gpointer k _U_, gpointer v, gpointer p _U_) {
116         stats_tree_cfg* cfg = v;
117         
118         cfg->pr = g_malloc(sizeof(tree_cfg_pres));
119         cfg->pr->init_string = g_strdup_printf("%s,tree",cfg->abbr);
120
121         register_stat_cmd_arg(cfg->pr->init_string, init_stats_tree, NULL);
122         
123 }
124
125 static void free_tree_presentation(stats_tree* st) {
126         g_free(st->pr);
127 }
128
129
130 void
131 register_tap_listener_stats_tree_stat(void)
132 {
133         stats_tree_presentation(register_stats_tree_tap,
134                                                         NULL, NULL, NULL, NULL, NULL,
135                                                         free_tree_presentation, NULL, NULL, NULL);
136 }