tap-rpcstat.c

   1 /* tap-rpcstat.c
   2  * rpcstat   2002 Ronnie Sahlberg
   3  *
   4  * $Id: tap-rpcstat.c,v 1.11 2003/11/27 22:15:05 guy Exp $
   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 rpc call/reply RTT statistics to tethereal.
  26  * It is only used by tethereal and not ethereal
  27  *
  28  * It serves as an example on how to use the tap api.
  29  */
  30
  31 #ifdef HAVE_CONFIG_H
  32 # include "config.h"
  33 #endif
  34
  35 #include <stdio.h>
  36
  37 #ifdef HAVE_SYS_TYPES_H
  38 # include <sys/types.h>
  39 #endif
  40
  41 #include <string.h>
  42 #include "epan/packet_info.h"
  43 #include "tap.h"
  44 #include "packet-rpc.h"
  45 #include "register.h"
  46
  47 /* used to keep track of statistics for a specific procedure */
  48 typedef struct _rpc_procedure_t {
  49         char *proc;
  50         int num;
  51         nstime_t min;
  52         nstime_t max;
  53         nstime_t tot;
  54 } rpc_procedure_t;
  55
  56 /* used to keep track of the statistics for an entire program interface */
  57 typedef struct _rpcstat_t {
  58         char *prog;
  59         char *filter;
  60         guint32 program;
  61         guint32 version;
  62         guint32 num_procedures;
  63         rpc_procedure_t *procedures;
  64 } rpcstat_t;
  65
  66
  67
  68 /* This callback is never used by tethereal but it is here for completeness.
  69  * When registering below, we could just have left this function as NULL.
  70  *
  71  * When used by ethereal, this function will be called whenever we would need
  72  * to reset all state. Such as when ethereal opens a new file, when it
  73  * starts a new capture, when it rescans the packetlist after some prefs have
  74  * changed etc.
  75  * So if your aplication has some state it needs to clean up in those
  76  * situations, here is a good place to put that code.
  77  */
  78 static void
  79 rpcstat_reset(void *prs)
  80 {
  81         rpcstat_t *rs=prs;
  82         guint32 i;
  83
  84         for(i=0;i<rs->num_procedures;i++){
  85                 rs->procedures[i].num=0;
  86                 rs->procedures[i].min.secs=0;
  87                 rs->procedures[i].min.nsecs=0;
  88                 rs->procedures[i].max.secs=0;
  89                 rs->procedures[i].max.nsecs=0;
  90                 rs->procedures[i].tot.secs=0;
  91                 rs->procedures[i].tot.nsecs=0;
  92         }
  93 }
  94
  95
  96 /* This callback is invoked whenever the tap system has seen a packet
  97  * we might be interested in.
  98  * The function is to be used to only update internal state information
  99  * in the *tapdata structure, and if there were state changes which requires
 100  * the window to be redrawn, return 1 and (*draw) will be called sometime
 101  * later.
 102  *
 103  * This function should be as lightweight as possible since it executes together
 104  * with the normal ethereal dissectors. Try to push as much processing as
 105  * possible into (*draw) instead since that function executes asynchronously
 106  * and does not affect the main threads performance.
 107  *
 108  * If it is possible, try to do all "filtering" explicitely as we do below in
 109  * this example since you will get MUCH better performance than applying
 110  * a similar display-filter in the register call.
 111  *
 112  * The third parameter is tap dependant. Since we register this one to the "rpc"
 113  * tap the third parameters type is rpc_call_info_value.
 114  *
 115  *
 116  * The filtering we do is just to check the rpc_call_info_value struct that
 117  * we were called for the proper program and version. We didnt apply a filter
 118  * when we registered so we will be called for ALL rpc packets and not just
 119  * the ones we are collecting stats for.
 120  *
 121  *
 122  * function returns :
 123  *  0: no updates, no need to call (*draw) later
 124  * !0: state has changed, call (*draw) sometime later
 125  */
 126 static int
 127 rpcstat_packet(void *prs, packet_info *pinfo, epan_dissect_t *edt _U_, void *pri)
 128 {
 129         rpcstat_t *rs=prs;
 130         rpc_call_info_value *ri=pri;
 131         nstime_t delta;
 132         rpc_procedure_t *rp;
 133
 134         if(ri->proc>=rs->num_procedures){
 135                 /* dont handle this since its outside of known table */
 136                 return 0;
 137         }
 138         /* we are only interested in reply packets */
 139         if(ri->request){
 140                 return 0;
 141         }
 142         /* we are only interested in certain program/versions */
 143         if( (ri->prog!=rs->program) || (ri->vers!=rs->version) ){
 144                 return 0;
 145         }
 146
 147         rp=&(rs->procedures[ri->proc]);
 148
 149         /* calculate time delta between request and reply */
 150         delta.secs=pinfo->fd->abs_secs-ri->req_time.secs;
 151         delta.nsecs=pinfo->fd->abs_usecs*1000-ri->req_time.nsecs;
 152         if(delta.nsecs<0){
 153                 delta.nsecs+=1000000000;
 154                 delta.secs--;
 155         }
 156
 157         if(rp->num==0){
 158                 rp->max.secs=delta.secs;
 159                 rp->max.nsecs=delta.nsecs;
 160         }
 161
 162         if(rp->num==0){
 163                 rp->min.secs=delta.secs;
 164                 rp->min.nsecs=delta.nsecs;
 165         }
 166
 167         if( (delta.secs<rp->min.secs)
 168         ||( (delta.secs==rp->min.secs)
 169           &&(delta.nsecs<rp->min.nsecs) ) ){
 170                 rp->min.secs=delta.secs;
 171                 rp->min.nsecs=delta.nsecs;
 172         }
 173
 174         if( (delta.secs>rp->max.secs)
 175         ||( (delta.secs==rp->max.secs)
 176           &&(delta.nsecs>rp->max.nsecs) ) ){
 177                 rp->max.secs=delta.secs;
 178                 rp->max.nsecs=delta.nsecs;
 179         }
 180
 181         rp->tot.secs += delta.secs;
 182         rp->tot.nsecs += delta.nsecs;
 183         if(rp->tot.nsecs>1000000000){
 184                 rp->tot.nsecs-=1000000000;
 185                 rp->tot.secs++;
 186         }
 187
 188         rp->num++;
 189
 190         return 1;
 191 }
 192
 193 /* This callback is used when tethereal wants us to draw/update our
 194  * data to the output device. Since this is tethereal only output is
 195  * stdout.
 196  * Tethereal will only call this callback once, which is when tethereal has
 197  * finished reading all packets and exists.
 198  * If used with ethereal this may be called any time, perhaps once every 3
 199  * seconds or so.
 200  * This function may even be called in parallell with (*reset) or (*draw)
 201  * so make sure there are no races. The data in the rpcstat_t can thus change
 202  * beneath us. Beware.
 203  */
 204 static void
 205 rpcstat_draw(void *prs)
 206 {
 207         rpcstat_t *rs=prs;
 208         guint32 i;
 209 #ifdef G_HAVE_UINT64
 210         guint64 td;
 211 #else
 212         guint32 td;
 213 #endif
 214         printf("\n");
 215         printf("===================================================================\n");
 216         printf("%s Version %d RTT Statistics:\n", rs->prog, rs->version);
 217         printf("Filter: %s\n",rs->filter?rs->filter:"");
 218         printf("Procedure        Calls   Min RTT   Max RTT   Avg RTT\n");
 219         for(i=0;i<rs->num_procedures;i++){
 220                 /* scale it to units of 10us.*/
 221                 /* for long captures with a large tot time, this can overflow on 32bit */
 222                 td=(int)rs->procedures[i].tot.secs;
 223                 td=td*100000+(int)rs->procedures[i].tot.nsecs/10000;
 224                 if(rs->procedures[i].num){
 225                         td/=rs->procedures[i].num;
 226                 } else {
 227                         td=0;
 228                 }
 229
 230                 printf("%-15s %6d %3d.%05d %3d.%05d %3d.%05d\n",
 231                         rs->procedures[i].proc,
 232                         rs->procedures[i].num,
 233                         (int)rs->procedures[i].min.secs,rs->procedures[i].min.nsecs/10000,
 234                         (int)rs->procedures[i].max.secs,rs->procedures[i].max.nsecs/10000,
 235                         td/100000, td%100000
 236                 );
 237         }
 238         printf("===================================================================\n");
 239 }
 240
 241 static guint32 rpc_program=0;
 242 static guint32 rpc_version=0;
 243 static gint32 rpc_min_proc=-1;
 244 static gint32 rpc_max_proc=-1;
 245
 246 static void *
 247 rpcstat_find_procs(gpointer *key, gpointer *value _U_, gpointer *user_data _U_)
 248 {
 249         rpc_proc_info_key *k=(rpc_proc_info_key *)key;
 250
 251         if(k->prog!=rpc_program){
 252                 return NULL;
 253         }
 254         if(k->vers!=rpc_version){
 255                 return NULL;
 256         }
 257         if(rpc_min_proc==-1){
 258                 rpc_min_proc=k->proc;
 259                 rpc_max_proc=k->proc;
 260         }
 261         if((gint32)k->proc<rpc_min_proc){
 262                 rpc_min_proc=k->proc;
 263         }
 264         if((gint32)k->proc>rpc_max_proc){
 265                 rpc_max_proc=k->proc;
 266         }
 267
 268         return NULL;
 269 }
 270
 271
 272 /* When called, this function will create a new instance of rpcstat.
 273  * program and version are whick onc-rpc program/version we want to
 274  * collect statistics for.
 275  * This function is called from tethereal when it parses the -z rpc, arguments
 276  * and it creates a new instance to store statistics in and registers this
 277  * new instance for the rpc tap.
 278  */
 279 static void
 280 rpcstat_init(char *optarg)
 281 {
 282         rpcstat_t *rs;
 283         guint32 i;
 284         int program, version;
 285         int pos=0;
 286         char *filter=NULL;
 287         GString *error_string;
 288
 289         if(sscanf(optarg,"rpc,rtt,%d,%d,%n",&program,&version,&pos)==2){
 290                 if(pos){
 291                         filter=optarg+pos;
 292                 } else {
 293                         filter=NULL;
 294                 }
 295         } else {
 296                 fprintf(stderr, "tethereal: invalid \"-z rpc,rtt,<program>,<version>[,<filter>]\" argument\n");
 297                 exit(1);
 298         }
 299
 300         rs=g_malloc(sizeof(rpcstat_t));
 301         rs->prog=rpc_prog_name(program);
 302         rs->program=program;
 303         rs->version=version;
 304         if(filter){
 305                 rs->filter=g_malloc(strlen(filter)+1);
 306                 strcpy(rs->filter, filter);
 307         } else {
 308                 rs->filter=NULL;
 309         }
 310         rpc_program=program;
 311         rpc_version=version;
 312         rpc_min_proc=-1;
 313         rpc_max_proc=-1;
 314         g_hash_table_foreach(rpc_procs, (GHFunc)rpcstat_find_procs, NULL);
 315         if(rpc_min_proc==-1){
 316                 fprintf(stderr,"tethereal: Invalid -z rpc,rrt,%d,%d\n",rpc_program,rpc_version);
 317                 fprintf(stderr,"   Program:%d version:%d is not supported by tethereal.\n", rpc_program, rpc_version);
 318                 exit(1);
 319         }
 320
 321
 322         rs->num_procedures=rpc_max_proc+1;
 323         rs->procedures=g_malloc(sizeof(rpc_procedure_t)*(rs->num_procedures+1));
 324         for(i=0;i<rs->num_procedures;i++){
 325                 rs->procedures[i].proc=rpc_proc_name(program, version, i);
 326                 rs->procedures[i].num=0;
 327                 rs->procedures[i].min.secs=0;
 328                 rs->procedures[i].min.nsecs=0;
 329                 rs->procedures[i].max.secs=0;
 330                 rs->procedures[i].max.nsecs=0;
 331                 rs->procedures[i].tot.secs=0;
 332                 rs->procedures[i].tot.nsecs=0;
 333         }
 334
 335 /* It is possible to create a filter and attach it to the callbacks. Then the
 336  * callbacks would only be invoked if the filter matched.
 337  * Evaluating filters is expensive and if we can avoid it and not use them
 338  * we gain performance.
 339  * In this case we do the filtering for protocol and version inside the
 340  * callback itself but use whatever filter the user provided.
 341  * (Perhaps the user only want the stats for nis+ traffic for certain objects?)
 342  *
 343  */
 344
 345         error_string=register_tap_listener("rpc", rs, filter, rpcstat_reset, rpcstat_packet, rpcstat_draw);
 346         if(error_string){
 347                 /* error, we failed to attach to the tap. clean up */
 348                 g_free(rs->procedures);
 349                 g_free(rs->filter);
 350                 g_free(rs);
 351
 352                 fprintf(stderr, "tethereal: Couldn't register rpc,rtt tap: %s\n",
 353                     error_string->str);
 354                 g_string_free(error_string, TRUE);
 355                 exit(1);
 356         }
 357 }
 358
 359
 360 void
 361 register_tap_listener_rpcstat(void)
 362 {
 363         register_ethereal_tap("rpc,rtt,", rpcstat_init);
 364 }
 365