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