For AFP requests, add in the frame with the (last part of the) reply, if
[metze/wireshark/wip.git] / tap-afpstat.c
1 /* tap-afpstat.c
2  * Based on
3  * smbstat   2003 Ronnie Sahlberg
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 <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.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.secs=pinfo->fd->abs_secs;
68         t.nsecs=pinfo->fd->abs_usecs*1000;
69         get_timedelta(&deltat, &t, &request_val->req_time);
70
71         if(sp){
72                 time_stat_update(sp,&deltat, pinfo);
73         }
74
75         return 1;
76 }
77
78 static void
79 afpstat_draw(void *pss)
80 {
81         afpstat_t *ss=(afpstat_t *)pss;
82         guint32 i;
83 #ifdef G_HAVE_UINT64
84         guint64 td;
85 #else
86         guint32 td;
87 #endif
88         printf("\n");
89         printf("===================================================================\n");
90         printf("AFP RTT Statistics:\n");
91         printf("Filter: %s\n",ss->filter?ss->filter:"");
92         printf("Commands                   Calls   Min RTT   Max RTT   Avg RTT\n");
93         for(i=0;i<256;i++){
94                 /* nothing seen, nothing to do */
95                 if(ss->proc[i].num==0){
96                         continue;
97                 }
98
99                 /* scale it to units of 10us.*/
100                 /* for long captures with a large tot time, this can overflow on 32bit */
101                 td=(int)ss->proc[i].tot.secs;
102                 td=td*100000+(int)ss->proc[i].tot.nsecs/10000;
103                 if(ss->proc[i].num){
104                         td/=ss->proc[i].num;
105                 } else {
106                         td=0;
107                 }
108
109                 printf("%-25s %6d %3d.%05d %3d.%05d %3d.%05d\n",
110                         val_to_str(i, CommandCode_vals, "Unknown (%u)"),
111                         ss->proc[i].num,
112                         (int)ss->proc[i].min.secs,ss->proc[i].min.nsecs/10000,
113                         (int)ss->proc[i].max.secs,ss->proc[i].max.nsecs/10000,
114                         td/100000, td%100000
115                 );
116         }
117         printf("===================================================================\n");
118 }
119
120
121 static void
122 afpstat_init(const char *optarg)
123 {
124         afpstat_t *ss;
125         guint32 i;
126         const char *filter=NULL;
127         GString *error_string;
128
129         if(!strncmp(optarg,"afp,rtt,",8)){
130                 filter=optarg+8;
131         } else {
132                 filter=NULL;
133         }
134
135         ss=g_malloc(sizeof(afpstat_t));
136         if(filter){
137                 ss->filter=g_malloc(strlen(filter)+1);
138                 strcpy(ss->filter, filter);
139         } else {
140                 ss->filter=NULL;
141         }
142
143         for(i=0;i<256;i++){
144                 ss->proc[i].num=0;
145                 ss->proc[i].min_num=0;
146                 ss->proc[i].max_num=0;
147                 ss->proc[i].min.secs=0;
148                 ss->proc[i].min.nsecs=0;
149                 ss->proc[i].max.secs=0;
150                 ss->proc[i].max.nsecs=0;
151                 ss->proc[i].tot.secs=0;
152                 ss->proc[i].tot.nsecs=0;
153         }
154
155         error_string=register_tap_listener("afp", ss, filter, NULL, afpstat_packet, afpstat_draw);
156         if(error_string){
157                 /* error, we failed to attach to the tap. clean up */
158                 g_free(ss->filter);
159                 g_free(ss);
160
161                 fprintf(stderr, "tethereal: Couldn't register afp,rtt tap: %s\n",
162                     error_string->str);
163                 g_string_free(error_string, TRUE);
164                 exit(1);
165         }
166 }
167
168 void
169 register_tap_listener_afpstat(void)
170 {
171         register_stat_cmd_arg("afp,rtt", afpstat_init);
172 }