Try to fix some of the build problems, it still bumms out on:
[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 <epan/dissectors/packet-sctp.h>
43 #include <epan/to_str.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 static void
89 sctpstat_reset(void *phs)
90 {
91         sctpstat_t* sctp_stat = (sctpstat_t *)phs;
92         sctp_ep_t* list = (sctp_ep_t*)sctp_stat->ep_list;
93         sctp_ep_t* tmp = NULL;
94         guint16 chunk_type;
95
96         if(!list)
97                 return;
98
99         for(tmp = list; tmp ; tmp=tmp->next)
100                 for(chunk_type = 0; chunk_type < 256; chunk_type++)
101                         tmp->chunk_count[chunk_type] = 0;
102
103         sctp_stat->number_of_packets = 0;
104 }
105
106
107 static sctp_ep_t*
108 alloc_sctp_ep(const struct _sctp_info *si)
109 {
110         sctp_ep_t* ep;
111         guint16 chunk_type;
112
113         if(!si)
114                 return NULL;
115
116         if (!(ep = g_malloc(sizeof(sctp_ep_t))))
117                 return NULL;
118
119         COPY_ADDRESS(&ep->src,&si->ip_src);
120         COPY_ADDRESS(&ep->dst,&si->ip_dst);
121         ep->sport = si->sport;
122         ep->dport = si->dport;
123         ep->next = NULL;
124         for(chunk_type = 0; chunk_type < 256; chunk_type++)
125                 ep->chunk_count[chunk_type] = 0;
126         return ep;
127 }
128
129
130
131
132 static int
133 sctpstat_packet(void *phs, packet_info *pinfo _U_, epan_dissect_t *edt _U_, const void *phi)
134 {
135
136         sctpstat_t *hs=(sctpstat_t *)phs;
137         sctp_ep_t *tmp = NULL, *te = NULL;
138         const struct _sctp_info *si = (const struct _sctp_info *) phi;
139         guint32 tvb_number;
140         guint8 chunk_type;
141
142         if (!hs)
143                 return (0);
144
145         hs->number_of_packets++;
146
147         if(!hs->ep_list) {
148                 hs->ep_list = alloc_sctp_ep(si);
149                 te = hs->ep_list;
150         } else {
151                 for(tmp=hs->ep_list ; tmp ; tmp=tmp->next)
152                 {
153                         if((!CMP_ADDRESS(&tmp->src,&si->ip_src)) &&
154                            (!CMP_ADDRESS(&tmp->dst,&si->ip_dst)) &&
155                            (tmp->sport == si->sport) &&
156                            (tmp->dport == si->dport))
157                         {
158                                 te = tmp;
159                                 break;
160                         }
161                 }
162                 if(!te) {
163                         if ((te = alloc_sctp_ep(si))) {
164                                 te->next = hs->ep_list;
165                                 hs->ep_list = te;
166                         }
167                 }
168         }
169
170         if(!te)
171                 return (0);
172
173
174         if (si->number_of_tvbs > 0) {
175                 chunk_type = CHUNK_TYPE(si->tvb[0]);
176                 if ((chunk_type == SCTP_INIT_CHUNK_ID) ||
177                     (chunk_type == SCTP_INIT_ACK_CHUNK_ID)) {
178                         te->chunk_count[chunk_type]++;
179                 } else {
180                         for(tvb_number = 0; tvb_number < si->number_of_tvbs; tvb_number++)
181                                 te->chunk_count[CHUNK_TYPE(si->tvb[tvb_number])]++;
182                 }
183         }
184         return (1);
185 }
186
187
188 static void
189 sctpstat_draw(void *phs)
190 {
191         sctpstat_t *hs=(sctpstat_t *)phs;
192         sctp_ep_t* list = hs->ep_list, *tmp;
193
194         printf("-------------------------------------------- SCTP Statistics --------------------------------------------------------------------------\n");
195         printf("|  Total packets RX/TX %u\n", hs->number_of_packets);
196         printf("---------------------------------------------------------------------------------------------------------------------------------------\n");
197         printf("|   Source IP   |PortA|    Dest. IP   |PortB|  DATA  |  SACK  |  HBEAT |HBEATACK|  INIT  | INITACK| COOKIE |COOKIACK| ABORT  |  ERROR |\n");
198         printf("---------------------------------------------------------------------------------------------------------------------------------------\n");
199
200         for(tmp = list ; tmp ; tmp=tmp->next) {
201                 printf("|%15s|%5u|%15s|%5u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|\n",
202                        ep_address_to_str(&tmp->src),tmp->sport,
203                        ep_address_to_str(&tmp->dst),tmp->dport,
204                        tmp->chunk_count[SCTP_DATA_CHUNK_ID],
205                        tmp->chunk_count[SCTP_SACK_CHUNK_ID],
206                        tmp->chunk_count[SCTP_HEARTBEAT_CHUNK_ID],
207                        tmp->chunk_count[SCTP_HEARTBEAT_ACK_CHUNK_ID],
208                        tmp->chunk_count[SCTP_INIT_CHUNK_ID],
209                        tmp->chunk_count[SCTP_INIT_ACK_CHUNK_ID],
210                        tmp->chunk_count[SCTP_COOKIE_ECHO_CHUNK_ID],
211                        tmp->chunk_count[SCTP_COOKIE_ACK_CHUNK_ID],
212                        tmp->chunk_count[SCTP_ABORT_CHUNK_ID],
213                        tmp->chunk_count[SCTP_ERROR_CHUNK_ID]);
214         }
215         printf("---------------------------------------------------------------------------------------------------------------------------------------\n");
216 }
217
218
219 static void
220 sctpstat_init(const char *optarg, void* userdata _U_)
221 {
222         sctpstat_t *hs;
223         GString *error_string;
224
225         hs = (sctpstat_t *)g_malloc(sizeof(sctpstat_t));
226         if(!strncmp(optarg,"sctp,stat,",11)){
227                 hs->filter=g_strdup(optarg+11);
228         } else {
229                 hs->filter=NULL;
230         }
231         hs->ep_list = NULL;
232         hs->number_of_packets = 0;
233
234         sctpstat_reset(hs);
235
236         error_string=register_tap_listener("sctp", hs, hs->filter, 0, NULL, sctpstat_packet, sctpstat_draw);
237         if(error_string){
238                 /* error, we failed to attach to the tap. clean up */
239                 g_free(hs->filter);
240                 g_free(hs);
241
242                 fprintf(stderr, "tshark: Couldn't register sctp,stat tap: %s\n",
243                     error_string->str);
244                 g_string_free(error_string, TRUE);
245                 exit(1);
246         }
247 }
248
249
250 void
251 register_tap_listener_sctpstat(void)
252 {
253         register_stat_cmd_arg("sctp,stat", sctpstat_init,NULL);
254 }