Adding upstream version 4.2.2.upstream/4.2.2

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

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 00000000..21e081eb
--- /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:
+ */