tap-wspstat.c

   1 /* tap-rpcstat.c
   2  * wspstat   2003 Jean-Michel FAYARD
   3  *
   4  * $Id$
   5  *
   6  * Ethereal - Network traffic analyzer
   7  * By Gerald Combs <gerald@ethereal.com>
   8  * Copyright 1998 Gerald Combs
   9  *
  10  * This program is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU General Public License
  12  * as published by the Free Software Foundation; either version 2
  13  * of the License, or (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  23  */
  24
  25 /* This module provides WSP  statistics to tethereal.
  26  * It is only used by tethereal and not ethereal
  27  *
  28  */
  29
  30 /* With MSVC and a libethereal.dll this file needs to import some variables
  31    in a special way. Therefore _NEED_VAR_IMPORT_ is defined. */
  32 #define _NEED_VAR_IMPORT_
  33
  34 #ifdef HAVE_CONFIG_H
  35 # include "config.h"
  36 #endif
  37
  38 #include <stdio.h>
  39
  40 #ifdef HAVE_SYS_TYPES_H
  41 # include <sys/types.h>
  42 #endif
  43
  44 #include <string.h>
  45 #include "epan/packet_info.h"
  46 #include <epan/tap.h>
  47 #include "register.h"
  48 #include "epan/value_string.h"
  49 #include <epan/dissectors/packet-wsp.h>
  50
  51 /* used to keep track of the stats for a specific PDU type*/
  52 typedef struct _wsp_pdu_t {
  53         gchar           *type;
  54         guint32          packets;
  55 } wsp_pdu_t;
  56 /* used to keep track of RTT statistics */
  57 typedef struct _wsp_status_code_t {
  58         gchar           *name;
  59         guint32          packets;
  60 } wsp_status_code_t;
  61 /* used to keep track of the statictics for an entire program interface */
  62 typedef struct _wsp_stats_t {
  63         char            *filter;
  64         wsp_pdu_t       *pdu_stats;
  65         guint32 num_pdus;
  66         GHashTable      *hash;
  67 } wspstat_t;
  68
  69 static void
  70 wsp_reset_hash(gchar *key _U_ , wsp_status_code_t *data, gpointer ptr _U_ )
  71 {
  72         data->packets = 0;
  73 }
  74 static void
  75 wsp_print_statuscode(gint *key, wsp_status_code_t *data, char* format)
  76 {
  77         if (data && (data->packets!=0))
  78                 printf(format, *key, data->packets ,data->name);
  79 }
  80 static void
  81 wsp_free_hash_table( gpointer key, gpointer value, gpointer user_data _U_ )
  82 {
  83         g_free(key);
  84         g_free(value);
  85 }
  86 static void
  87 wspstat_reset(void *psp)
  88 {
  89         wspstat_t *sp=psp;
  90         guint32 i;
  91
  92         for(i=1;i<=sp->num_pdus;i++)
  93         {
  94                 sp->pdu_stats[i].packets=0;
  95         }
  96         g_hash_table_foreach( sp->hash, (GHFunc)wsp_reset_hash, NULL);
  97 }
  98
  99
 100 /* This callback is invoked whenever the tap system has seen a packet
 101  * we might be interested in.
 102  * The function is to be used to only update internal state information
 103  * in the *tapdata structure, and if there were state changes which requires
 104  * the window to be redrawn, return 1 and (*draw) will be called sometime
 105  * later.
 106  *
 107  * We didnt apply a filter when we registered so we will be called for
 108  * ALL packets and not just the ones we are collecting stats for.
 109  *
 110  */
 111 static gint
 112 pdut2index(gint pdut)
 113 {
 114         if (pdut<=0x09)         return pdut;
 115         if (pdut>=0x40){
 116                 if (pdut <= 0x44){
 117                         return pdut-54;
 118                 } else if (pdut==0x60||pdut==0x61){
 119                         return pdut-81;
 120                 }
 121         }
 122         return 0;
 123 }
 124 static gint
 125 index2pdut(gint pdut)
 126 {
 127         if (pdut<=0x09)
 128                 return pdut;
 129         if (pdut<=14)
 130                 return pdut+54;
 131         if (pdut<=16)
 132                 return pdut+81;
 133         return 0;
 134 }
 135 static int
 136 wspstat_packet(void *psp, packet_info *pinfo _U_, epan_dissect_t *edt _U_, void *pri)
 137 {
 138         wspstat_t *sp=psp;
 139         wsp_info_value_t *value=pri;
 140         gint index = pdut2index(value->pdut);
 141         int retour=0;
 142
 143         if (value->status_code != 0) {
 144                 gint *key=g_malloc( sizeof(gint) );
 145                 wsp_status_code_t *sc;
 146                 *key=value->status_code ;
 147                 sc = g_hash_table_lookup(
 148                                 sp->hash,
 149                                 key);
 150                 if (!sc) {
 151                         sc = g_malloc( sizeof(wsp_status_code_t) );
 152                         sc -> packets = 1;
 153                         sc -> name = NULL;
 154                         g_hash_table_insert(
 155                                 sp->hash,
 156                                 key,
 157                                 sc);
 158                 } else {
 159                         sc->packets++;
 160                 }
 161                 retour=1;
 162         }
 163
 164
 165
 166         if (index!=0) {
 167                 sp->pdu_stats[ index ].packets++;
 168                 retour = 1;
 169         }
 170         return retour;
 171 }
 172
 173
 174 /* This callback is used when tethereal wants us to draw/update our
 175  * data to the output device. Since this is tethereal only output is
 176  * stdout.
 177  * Tethereal will only call this callback once, which is when tethereal has
 178  * finished reading all packets and exists.
 179  * If used with ethereal this may be called any time, perhaps once every 3
 180  * seconds or so.
 181  * This function may even be called in parallell with (*reset) or (*draw)
 182  * so make sure there are no races. The data in the rpcstat_t can thus change
 183  * beneath us. Beware.
 184  */
 185 static void
 186 wspstat_draw(void *psp)
 187 {
 188         wspstat_t *sp=psp;
 189         guint32 i;
 190
 191         printf("\n");
 192         printf("===================================================================\n");
 193         printf("WSP Statistics:\n");
 194         printf("%-23s %9s || %-23s %9s\n","PDU Type", "Packets", "PDU Type", "Packets");
 195         for(i=1; i<= ((sp->num_pdus+1)/2) ; i++)
 196         {
 197                 guint32 ii=i+sp->num_pdus/2;
 198                 printf("%-23s %9d", sp->pdu_stats[i ].type, sp->pdu_stats[i ].packets);
 199                 printf(" || ");
 200                 if (ii< (sp->num_pdus) )
 201                         printf("%-23s %9d\n", sp->pdu_stats[ii].type, sp->pdu_stats[ii].packets);
 202                 else
 203                         printf("\n");
 204                 }
 205         printf("\nStatus code in reply packets\n");
 206         printf(         "Status Code    Packets  Description\n");
 207         g_hash_table_foreach( sp->hash, (GHFunc) wsp_print_statuscode,
 208                         "       0x%02X  %9d  %s\n" ) ;
 209         printf("===================================================================\n");
 210 }
 211
 212 /* When called, this function will create a new instance of wspstat.
 213  * program and version are whick onc-rpc program/version we want to
 214  * collect statistics for.
 215  * This function is called from tethereal when it parses the -z wsp, arguments
 216  * and it creates a new instance to store statistics in and registers this
 217  * new instance for the wsp tap.
 218  */
 219 static void
 220 wspstat_init(char *optarg)
 221 {
 222         wspstat_t *sp;
 223         char *filter=NULL;
 224         guint32 i;
 225         GString *error_string;
 226         wsp_status_code_t *sc;
 227
 228         if (!strncmp (optarg, "wsp,stat," , 9)){
 229                 filter=optarg+9;
 230         } else {
 231                 filter=NULL;
 232         }
 233
 234
 235         sp = g_malloc( sizeof(wspstat_t) );
 236         sp->hash = g_hash_table_new( g_int_hash, g_int_equal);
 237         for (i=0 ; vals_status[i].strptr ; i++ )
 238         {
 239                 gint *key;
 240                 sc=g_malloc( sizeof(wsp_status_code_t) );
 241                 key=g_malloc( sizeof(gint) );
 242                 sc->packets=0;
 243                 sc->name=vals_status[i].strptr;
 244                 *key=vals_status[i].value;
 245                 g_hash_table_insert(
 246                                 sp->hash,
 247                                 key,
 248                                 sc);
 249         }
 250         sp->num_pdus = 16;
 251         sp->pdu_stats=g_malloc( (sp->num_pdus+1) * sizeof( wsp_pdu_t) );
 252         if(filter){
 253                 sp->filter=g_malloc(strlen(filter)+1);
 254                 strcpy(sp->filter,filter);
 255         } else {
 256                 sp->filter=NULL;
 257         }
 258         for (i=0;i<sp->num_pdus; i++)
 259         {
 260                 sp->pdu_stats[i].packets=0;
 261                 sp->pdu_stats[i].type = match_strval( index2pdut( i ), vals_pdu_type) ;
 262         }
 263
 264         error_string = register_tap_listener(
 265                         "wsp",
 266                         sp,
 267                         filter,
 268                         wspstat_reset,
 269                         wspstat_packet,
 270                         wspstat_draw);
 271         if (error_string){
 272                 /* error, we failed to attach to the tap. clean up */
 273                 g_free(sp->pdu_stats);
 274                 g_free(sp->filter);
 275                 g_free(sp);
 276                 g_hash_table_foreach( sp->hash, (GHFunc) wsp_free_hash_table, NULL ) ;
 277                 g_hash_table_destroy( sp->hash );
 278                 fprintf(stderr, "tethereal: Couldn't register wsp,stat tap: %s\n",
 279                                 error_string->str);
 280                 g_string_free(error_string, TRUE);
 281                 exit(1);
 282         }
 283 }
 284 void
 285 register_tap_listener_wspstat(void)
 286 {
 287         register_ethereal_tap("wsp,stat,", wspstat_init);
 288 }