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
  36 /* actually unused */
  37 struct _st_node_pres {
  38         void* dummy;
  39 };
  40
  41 struct _tree_pres {
  42         void* dummy;
  43 };
  44
  45 struct _tree_cfg_pres {
  46         guint8* init_string;
  47 };
  48
  49 static void draw_stats_tree(void *psp) {
  50         stats_tree *st = psp;
  51         GString* s;
  52         gchar* fmt;
  53         stat_node* child;
  54
  55         s = g_string_new("\n===================================================================\n");
  56         fmt = g_strdup_printf(" %%s%%-%us%%12s\t%%12s\t%%12s\n",stats_tree_branch_max_namelen(&st->root,0));
  57         g_string_sprintfa(s,fmt,"",st->cfg->name,"value","rate","percent");
  58         g_free(fmt);
  59         g_string_sprintfa(s,"-------------------------------------------------------------------\n");
  60
  61         for (child = st->root.children; child; child = child->next ) {
  62                 stats_tree_branch_to_str(child,s,0);
  63         }
  64
  65         s = g_string_append(s,"\n===================================================================\n");
  66
  67         printf("%s",s->str);
  68
  69 }
  70
  71 static void  init_stats_tree(const char *optarg) {
  72         guint8* abbr = stats_tree_get_abbr(optarg);
  73         GString *error_string;
  74         stats_tree_cfg *cfg = NULL;
  75         stats_tree* st = NULL;
  76
  77         if (abbr) {
  78                 cfg = stats_tree_get_cfg_by_abbr(abbr);
  79
  80                 if (cfg != NULL) {
  81                         if (strncmp (optarg, cfg->pr->init_string, strlen(cfg->pr->init_string)) == 0){
  82                                 st = stats_tree_new(cfg,NULL,((guint8*)optarg)+strlen(cfg->pr->init_string));
  83                         } else {
  84                                 st->filter=NULL;
  85                         }
  86                 } else {
  87                         g_error("no such stats_tree (%s) found in stats_tree registry",abbr);
  88                 }
  89
  90                 g_free(abbr);
  91
  92         } else {
  93                 g_error("could not obtain stats_tree abbr (%s) from optarg '%s'",abbr,optarg);
  94         }
  95
  96         error_string = register_tap_listener( st->cfg->tapname,
  97                                                                                   st,
  98                                                                                   st->filter,
  99                                                                                   stats_tree_reset,
 100                                                                                   stats_tree_packet,
 101                                                                                   draw_stats_tree);
 102
 103         if (error_string) {
 104                 g_error("stats_tree for: %s failed to attach to the tap: %s",cfg->name,error_string->str);
 105         }
 106
 107         if (cfg->init) cfg->init(st);
 108
 109 }
 110
 111 void register_stats_tree_tap (gpointer k _U_, gpointer v, gpointer p _U_) {
 112         stats_tree_cfg* cfg = v;
 113
 114         cfg->pr = g_malloc(sizeof(tree_cfg_pres));
 115         cfg->pr->init_string = g_strdup_printf("%s,tree",cfg->abbr);
 116
 117         register_stat_cmd_arg(cfg->pr->init_string, init_stats_tree);
 118
 119 }
 120
 121 static void free_tree_presentation(stats_tree* st) {
 122         g_free(st->pr);
 123 }
 124
 125
 126 void
 127 register_tap_listener_stats_tree_stat(void)
 128 {
 129         stats_tree_presentation(register_stats_tree_tap,
 130                                                         NULL, NULL, NULL, NULL, NULL,
 131                                                         free_tree_presentation, NULL, NULL, NULL);
 132 }