From Edgar Gladkich:
[obnox/wireshark/wip.git] / tap-afpstat.c
1 /* tap-afpstat.c
2  * Based on
3  * smbstat   2003 Ronnie Sahlberg
4  *
5  * $Id$
6  *
7  * Wireshark - Network traffic analyzer
8  * By Gerald Combs <gerald@wireshark.org>
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 <stdio.h>
31
32 #ifdef HAVE_SYS_TYPES_H
33 # include <sys/types.h>
34 #endif
35
36 #include <string.h>
37 #include <epan/packet_info.h>
38 #include <epan/tap.h>
39 #include <epan/stat_cmd_args.h>
40 #include <epan/value_string.h>
41 #include <epan/dissectors/packet-afp.h>
42 #include "register.h"
43 #include "timestats.h"
44
45 /* used to keep track of the statistics for an entire program interface */
46 typedef struct _afpstat_t {
47         char *filter;
48         timestat_t proc[256];
49 } afpstat_t;
50
51 static int
52 afpstat_packet(void *pss, packet_info *pinfo, epan_dissect_t *edt _U_, const void *prv)
53 {
54         afpstat_t *ss=(afpstat_t *)pss;
55         const afp_request_val *request_val=prv;
56         nstime_t t, deltat;
57         timestat_t *sp=NULL;
58
59         /* if we havnt seen the request, just ignore it */
60         if(!request_val){
61                 return 0;
62         }
63
64         sp=&(ss->proc[request_val->command]);
65
66         /* calculate time delta between request and reply */
67         t=pinfo->fd->abs_ts;
68         nstime_delta(&deltat, &t, &request_val->req_time);
69
70         if(sp){
71                 time_stat_update(sp,&deltat, pinfo);
72         }
73
74         return 1;
75 }
76
77 static void
78 afpstat_draw(void *pss)
79 {
80         afpstat_t *ss=(afpstat_t *)pss;
81         guint32 i;
82         guint64 td;
83         printf("\n");
84         printf("===================================================================\n");
85         printf("AFP RTT Statistics:\n");
86         printf("Filter: %s\n",ss->filter?ss->filter:"");
87         printf("Commands                   Calls   Min RTT   Max RTT   Avg RTT\n");
88         for(i=0;i<256;i++){
89                 /* nothing seen, nothing to do */
90                 if(ss->proc[i].num==0){
91                         continue;
92                 }
93
94                 /* scale it to units of 10us.*/
95                 td=ss->proc[i].tot.secs;
96                 td=td*100000+(int)ss->proc[i].tot.nsecs/10000;
97                 if(ss->proc[i].num){
98                         td/=ss->proc[i].num;
99                 } else {
100                         td=0;
101                 }
102
103                 printf("%-25s %6d %3d.%05d %3d.%05d %3" G_GINT64_MODIFIER "u.%05" G_GINT64_MODIFIER "u\n",
104                         val_to_str(i, CommandCode_vals, "Unknown (%u)"),
105                         ss->proc[i].num,
106                         (int)ss->proc[i].min.secs,ss->proc[i].min.nsecs/10000,
107                         (int)ss->proc[i].max.secs,ss->proc[i].max.nsecs/10000,
108                         td/100000, td%100000
109                 );
110         }
111         printf("===================================================================\n");
112 }
113
114
115 static void
116 afpstat_init(const char *optarg, void* userdata _U_)
117 {
118         afpstat_t *ss;
119         guint32 i;
120         const char *filter=NULL;
121         GString *error_string;
122
123         if(!strncmp(optarg,"afp,rtt,",8)){
124                 filter=optarg+8;
125         } else {
126                 filter=NULL;
127         }
128
129         ss=g_malloc(sizeof(afpstat_t));
130         if(filter){
131                 ss->filter=g_strdup(filter);
132         } else {
133                 ss->filter=NULL;
134         }
135
136         for(i=0;i<256;i++){
137                 ss->proc[i].num=0;
138                 ss->proc[i].min_num=0;
139                 ss->proc[i].max_num=0;
140                 ss->proc[i].min.secs=0;
141                 ss->proc[i].min.nsecs=0;
142                 ss->proc[i].max.secs=0;
143                 ss->proc[i].max.nsecs=0;
144                 ss->proc[i].tot.secs=0;
145                 ss->proc[i].tot.nsecs=0;
146         }
147
148         error_string=register_tap_listener("afp", ss, filter, 0, NULL, afpstat_packet, afpstat_draw);
149         if(error_string){
150                 /* error, we failed to attach to the tap. clean up */
151                 g_free(ss->filter);
152                 g_free(ss);
153
154                 fprintf(stderr, "tshark: Couldn't register afp,rtt tap: %s\n",
155                     error_string->str);
156                 g_string_free(error_string, TRUE);
157                 exit(1);
158         }
159 }
160
161 void
162 register_tap_listener_afpstat(void)
163 {
164         register_stat_cmd_arg("afp,rtt", afpstat_init,NULL);
165 }