diff options
Diffstat (limited to 'ui/cli/tap-sctpchunkstat.c')
-rw-r--r-- | ui/cli/tap-sctpchunkstat.c | 243 |
1 files changed, 243 insertions, 0 deletions
diff --git a/ui/cli/tap-sctpchunkstat.c b/ui/cli/tap-sctpchunkstat.c new file mode 100644 index 0000000..21e081e --- /dev/null +++ b/ui/cli/tap-sctpchunkstat.c @@ -0,0 +1,243 @@ +/* tap_sctpchunkstat.c + * SCTP chunk counter for wireshark + * Copyright 2005 Oleg Terletsky <oleg.terletsky@comverse.com> + * + * Wireshark - Network traffic analyzer + * By Gerald Combs <gerald@wireshark.org> + * Copyright 1998 Gerald Combs + * + * SPDX-License-Identifier: GPL-2.0-or-later + */ + +#include "config.h" + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include <glib.h> + +#include <epan/packet_info.h> +#include <epan/tap.h> +#include <epan/stat_tap_ui.h> +#include <epan/value_string.h> +#include <epan/dissectors/packet-sctp.h> +#include <epan/to_str.h> + +#include <wsutil/cmdarg_err.h> + +void register_tap_listener_sctpstat(void); + +typedef struct sctp_ep { + struct sctp_ep *next; + address src; + address dst; + guint16 sport; + guint16 dport; + guint32 chunk_count[256]; +} sctp_ep_t; + + +/* used to keep track of the statistics for an entire program interface */ +typedef struct _sctpstat_t { + char *filter; + guint32 number_of_packets; + sctp_ep_t *ep_list; +} sctpstat_t; + +#define CHUNK_TYPE_OFFSET 0 +#define CHUNK_TYPE(x)(tvb_get_guint8((x), CHUNK_TYPE_OFFSET)) + +static void +sctpstat_reset(void *phs) +{ + sctpstat_t *sctp_stat = (sctpstat_t *)phs; + sctp_ep_t *list = (sctp_ep_t *)sctp_stat->ep_list; + sctp_ep_t *tmp = NULL; + guint16 chunk_type; + + if (!list) + return; + + for (tmp = list; tmp; tmp = tmp->next) + for (chunk_type = 0; chunk_type < 256; chunk_type++) + tmp->chunk_count[chunk_type] = 0; + + sctp_stat->number_of_packets = 0; +} + + +static sctp_ep_t * +alloc_sctp_ep(const struct _sctp_info *si) +{ + sctp_ep_t *ep; + guint16 chunk_type; + + if (!si) + return NULL; + + if (!(ep = g_new(sctp_ep_t, 1))) + return NULL; + + copy_address(&ep->src, &si->ip_src); + copy_address(&ep->dst, &si->ip_dst); + ep->sport = si->sport; + ep->dport = si->dport; + ep->next = NULL; + for (chunk_type = 0; chunk_type < 256; chunk_type++) + ep->chunk_count[chunk_type] = 0; + return ep; +} + + + + +static tap_packet_status +sctpstat_packet(void *phs, packet_info *pinfo _U_, epan_dissect_t *edt _U_, const void *phi, tap_flags_t flags _U_) +{ + + sctpstat_t *hs = (sctpstat_t *)phs; + sctp_ep_t *tmp = NULL, *te = NULL; + const struct _sctp_info *si = (const struct _sctp_info *)phi; + guint32 tvb_number; + guint8 chunk_type; + + if (!hs) + return (TAP_PACKET_DONT_REDRAW); + + hs->number_of_packets++; + + if (!hs->ep_list) { + hs->ep_list = alloc_sctp_ep(si); + te = hs->ep_list; + } else { + for (tmp = hs->ep_list; tmp; tmp = tmp->next) + { + if ((!cmp_address(&tmp->src, &si->ip_src)) && + (!cmp_address(&tmp->dst, &si->ip_dst)) && + (tmp->sport == si->sport) && + (tmp->dport == si->dport)) + { + te = tmp; + break; + } + } + if (!te) { + if ((te = alloc_sctp_ep(si))) { + te->next = hs->ep_list; + hs->ep_list = te; + } + } + } + + if (!te) + return (TAP_PACKET_DONT_REDRAW); + + + if (si->number_of_tvbs > 0) { + chunk_type = CHUNK_TYPE(si->tvb[0]); + if ((chunk_type == SCTP_INIT_CHUNK_ID) || + (chunk_type == SCTP_INIT_ACK_CHUNK_ID)) { + te->chunk_count[chunk_type]++; + } else { + for (tvb_number = 0; tvb_number < si->number_of_tvbs; tvb_number++) + te->chunk_count[CHUNK_TYPE(si->tvb[tvb_number])]++; + } + } + return (TAP_PACKET_REDRAW); +} + + +static void +sctpstat_draw(void *phs) +{ + sctpstat_t *hs = (sctpstat_t *)phs; + sctp_ep_t *list = hs->ep_list, *tmp; + char *src_addr, *dst_addr; + + printf("-------------------------------------------- SCTP Statistics --------------------------------------------------------------------------\n"); + printf("| Total packets RX/TX %u\n", hs->number_of_packets); + printf("---------------------------------------------------------------------------------------------------------------------------------------\n"); + printf("| Source IP |PortA| Dest. IP |PortB| DATA | SACK | HBEAT |HBEATACK| INIT | INITACK| COOKIE |COOKIACK| ABORT | ERROR |\n"); + printf("---------------------------------------------------------------------------------------------------------------------------------------\n"); + + for (tmp = list; tmp; tmp = tmp->next) { + src_addr = (char*)address_to_str(NULL, &tmp->src); + dst_addr = (char*)address_to_str(NULL, &tmp->dst); + printf("|%15s|%5u|%15s|%5u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|%8u|\n", + src_addr, tmp->sport, + dst_addr, tmp->dport, + tmp->chunk_count[SCTP_DATA_CHUNK_ID], + tmp->chunk_count[SCTP_SACK_CHUNK_ID], + tmp->chunk_count[SCTP_HEARTBEAT_CHUNK_ID], + tmp->chunk_count[SCTP_HEARTBEAT_ACK_CHUNK_ID], + tmp->chunk_count[SCTP_INIT_CHUNK_ID], + tmp->chunk_count[SCTP_INIT_ACK_CHUNK_ID], + tmp->chunk_count[SCTP_COOKIE_ECHO_CHUNK_ID], + tmp->chunk_count[SCTP_COOKIE_ACK_CHUNK_ID], + tmp->chunk_count[SCTP_ABORT_CHUNK_ID], + tmp->chunk_count[SCTP_ERROR_CHUNK_ID]); + wmem_free(NULL, src_addr); + wmem_free(NULL, dst_addr); + } + printf("---------------------------------------------------------------------------------------------------------------------------------------\n"); +} + + +static void +sctpstat_init(const char *opt_arg, void *userdata _U_) +{ + sctpstat_t *hs; + GString *error_string; + + hs = g_new(sctpstat_t, 1); + if (!strncmp(opt_arg, "sctp,stat,", 11)) { + hs->filter = g_strdup(opt_arg+11); + } else { + hs->filter = NULL; + } + hs->ep_list = NULL; + hs->number_of_packets = 0; + + sctpstat_reset(hs); + + error_string = register_tap_listener("sctp", hs, hs->filter, 0, NULL, sctpstat_packet, sctpstat_draw, NULL); + if (error_string) { + /* error, we failed to attach to the tap. clean up */ + g_free(hs->filter); + g_free(hs); + + cmdarg_err("Couldn't register sctp,stat tap: %s", + error_string->str); + g_string_free(error_string, TRUE); + exit(1); + } +} + +static stat_tap_ui sctpstat_ui = { + REGISTER_STAT_GROUP_GENERIC, + NULL, + "sctp,stat", + sctpstat_init, + 0, + NULL +}; + +void +register_tap_listener_sctpstat(void) +{ + register_stat_tap_ui(&sctpstat_ui, NULL); +} + +/* + * Editor modelines - https://www.wireshark.org/tools/modelines.html + * + * Local variables: + * c-basic-offset: 8 + * tab-width: 8 + * indent-tabs-mode: t + * End: + * + * vi: set shiftwidth=8 tabstop=8 noexpandtab: + * :indentSize=8:tabSize=8:noTabs=false: + */ |