83b8985a149447627f4382fd0478f43db35abb50
[obnox/wireshark/wip.git] / tap-sctpchunkstat.c
1 /* tap_sctpchunkstat.c
2  * SCTP chunk counter for ethereal
3  * Copyright 2005 Oleg Terletsky <oleg.terletsky@comverse.com>
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/addr_resolv.h"
39 #include <epan/tap.h>
40 #include "epan/value_string.h"
41 #include "register.h"
42 #include <epan/dissectors/packet-sctp.h>
43
44 typedef struct sctp_ep {
45         struct sctp_ep* next;
46         address src;
47         address dst;
48         guint16 sport;
49         guint16 dport;
50         guint32 chunk_count[256];
51 } sctp_ep_t;
52
53
54 /* used to keep track of the statistics for an entire program interface */
55 typedef struct _sctpstat_t {
56         char*      filter;
57         guint32    number_of_packets;
58         sctp_ep_t* ep_list;
59 } sctpstat_t;
60
61
62 #define SCTP_DATA_CHUNK_ID               0
63 #define SCTP_INIT_CHUNK_ID               1
64 #define SCTP_INIT_ACK_CHUNK_ID           2
65 #define SCTP_SACK_CHUNK_ID               3
66 #define SCTP_HEARTBEAT_CHUNK_ID          4
67 #define SCTP_HEARTBEAT_ACK_CHUNK_ID      5
68 #define SCTP_ABORT_CHUNK_ID              6
69 #define SCTP_SHUTDOWN_CHUNK_ID           7
70 #define SCTP_SHUTDOWN_ACK_CHUNK_ID       8
71 #define SCTP_ERROR_CHUNK_ID              9
72 #define SCTP_COOKIE_ECHO_CHUNK_ID       10
73 #define SCTP_COOKIE_ACK_CHUNK_ID        11
74 #define SCTP_ECNE_CHUNK_ID              12
75 #define SCTP_CWR_CHUNK_ID               13
76 #define SCTP_SHUTDOWN_COMPLETE_CHUNK_ID 14
77 #define SCTP_AUTH_CHUNK_ID            0x16
78 #define SCTP_ASCONF_ACK_CHUNK_ID      0x80
79 #define SCTP_PKTDROP_CHUNK_ID         0x81
80 #define SCTP_FORWARD_TSN_CHUNK_ID     0xC0
81 #define SCTP_ASCONF_CHUNK_ID          0xC1
82 #define SCTP_IETF_EXT                 0xFF
83
84 #define CHUNK_TYPE_OFFSET 0
85 #define CHUNK_TYPE(x)(tvb_get_guint8((x), CHUNK_TYPE_OFFSET))
86
87
88 extern gchar* address_to_str(const address *);
89
90
91 static void
92 sctpstat_reset(void *phs)
93 {
94         sctpstat_t* sctp_stat = (sctpstat_t *)phs;
95         sctp_ep_t* list = (sctp_ep_t*)sctp_stat->ep_list;
96         sctp_ep_t* tmp = NULL;
97         guint16 chunk_type;
98         
99         if(!list)
100                 return;
101
102         for(tmp = list; tmp ; tmp=tmp->next)
103                 for(chunk_type = 0; chunk_type < 256; chunk_type++)
104                         tmp->chunk_count[chunk_type] = 0;
105
106         sctp_stat->number_of_packets = 0;
107 }
108
109
110 sctp_ep_t* alloc_sctp_ep(struct _sctp_info *si)
111 {
112         sctp_ep_t* ep;
113         guint16 chunk_type;
114
115         if(!si)
116                 return NULL;
117
118         if (!(ep = g_malloc(sizeof(sctp_ep_t))))
119                 return NULL;
120         
121         COPY_ADDRESS(&ep->src,&si->ip_src);
122         COPY_ADDRESS(&ep->dst,&si->ip_dst);
123         ep->sport = si->sport;
124         ep->dport = si->dport;
125         ep->next = NULL;
126         for(chunk_type = 0; chunk_type < 256; chunk_type++)
127                 ep->chunk_count[chunk_type] = 0;
128         return ep;
129 }
130
131
132         
133
134 static int
135 sctpstat_packet(void *phs, packet_info *pinfo _U_, epan_dissect_t *edt _U_, const void *phi)
136 {
137
138         sctpstat_t *hs=(sctpstat_t *)phs;
139         sctp_ep_t *tmp = NULL, *te = NULL;
140         struct _sctp_info *si = (struct _sctp_info *) phi;
141         guint32 tvb_number;
142         guint8 chunk_type;
143         
144         if (!hs)
145                 return (0);
146                 
147         hs->number_of_packets++;
148         
149         if(!hs->ep_list) {
150                 hs->ep_list = alloc_sctp_ep(si);
151                 te = hs->ep_list;
152         } else {
153                 for(tmp=hs->ep_list ; tmp ; tmp=tmp->next)
154                 {
155                         if((!CMP_ADDRESS(&tmp->src,&si->ip_src)) &&
156                            (!CMP_ADDRESS(&tmp->dst,&si->ip_dst)) &&
157                            (tmp->sport == si->sport) &&
158                            (tmp->dport == si->dport))
159                         {
160                                 te = tmp;
161                                 break;
162                         }
163                 }
164                 if(!te) {
165                         if ((te = alloc_sctp_ep(si))) {
166                                 te->next = hs->ep_list;
167                                 hs->ep_list = te;
168                         }
169                 }
170         }
171
172         if(!te)
173                 return (0);
174
175         
176         if (si->number_of_tvbs > 0) {
177                 chunk_type = CHUNK_TYPE(si->tvb[0]);
178                 if ((chunk_type == SCTP_INIT_CHUNK_ID) ||
179                     (chunk_type == SCTP_INIT_ACK_CHUNK_ID)) {
180                         te->chunk_count[chunk_type]++;
181                 } else {
182                         for(tvb_number = 0; tvb_number < si->number_of_tvbs; tvb_number++)
183                                 te->chunk_count[CHUNK_TYPE(si->tvb[tvb_number])]++;
184                 }
185         }
186         return (1);
187 }
188
189
190 static void
191 sctpstat_draw(void *phs)
192 {
193         sctpstat_t *hs=(sctpstat_t *)phs;
194         sctp_ep_t* list = hs->ep_list, *tmp;
195
196         printf("-------------------------------------------- SCTP Statistics --------------------------------------------------------------------------\n");
197         printf("|  Total packets RX/TX %u\n", hs->number_of_packets);
198         printf("---------------------------------------------------------------------------------------------------------------------------------------\n");
199         printf("|   Source IP   |PortA|    Dest. IP   |PortB|  DATA  |  SACK  |  HBEAT |HBEATACK|  INIT  | INITACK| COOKIE |COOKIACK| ABORT  |  ERROR |\n");
200         printf("---------------------------------------------------------------------------------------------------------------------------------------\n");
201         
202         for(tmp = list ; tmp ; tmp=tmp->next) {
203                 printf("|%15s|%5u|%15s|%5u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|\n",
204                        address_to_str(&tmp->src),tmp->sport,
205                        address_to_str(&tmp->dst),tmp->dport,
206                        tmp->chunk_count[SCTP_DATA_CHUNK_ID],
207                        tmp->chunk_count[SCTP_SACK_CHUNK_ID],
208                        tmp->chunk_count[SCTP_HEARTBEAT_CHUNK_ID],
209                        tmp->chunk_count[SCTP_HEARTBEAT_ACK_CHUNK_ID],
210                        tmp->chunk_count[SCTP_INIT_CHUNK_ID],
211                        tmp->chunk_count[SCTP_INIT_ACK_CHUNK_ID],
212                        tmp->chunk_count[SCTP_COOKIE_ECHO_CHUNK_ID],
213                        tmp->chunk_count[SCTP_COOKIE_ACK_CHUNK_ID],
214                        tmp->chunk_count[SCTP_ABORT_CHUNK_ID],
215                        tmp->chunk_count[SCTP_ERROR_CHUNK_ID]);
216         }
217         printf("---------------------------------------------------------------------------------------------------------------------------------------\n");
218 }
219
220
221 static void
222 sctpstat_init(char *optarg)
223 {
224         sctpstat_t *hs;
225         char *filter=NULL;
226         GString *error_string;
227
228         if(!strncmp(optarg,"sctp,stat,",11)){
229                 filter=optarg+11;
230         } else {
231                 filter=g_malloc(1);
232                 *filter='\0';
233         }
234
235         hs = g_malloc(sizeof(sctpstat_t));
236         hs->filter=g_malloc(strlen(filter)+1);
237         hs->ep_list = NULL;
238         hs->number_of_packets = 0;
239         strcpy(hs->filter, filter);
240
241         sctpstat_reset(hs);
242
243         error_string=register_tap_listener("sctp", hs, filter, NULL, sctpstat_packet, sctpstat_draw);
244         if(error_string){
245                 /* error, we failed to attach to the tap. clean up */
246                 g_free(hs->filter);
247                 g_free(hs);
248
249                 fprintf(stderr, "tethereal: Couldn't register sctp,stat tap: %s\n",
250                     error_string->str);
251                 g_string_free(error_string, TRUE);
252                 exit(1);
253         }
254 }
255
256
257 void
258 register_tap_listener_sctpstat(void)
259 {
260         register_tap_listener_cmd_arg("sctp,stat", sctpstat_init);
261 }