1 files changed, 293 insertions, 0 deletions
diff --git a/ui/cli/tap-wspstat.c b/ui/cli/tap-wspstat.c
new file mode 100644
index 00000000..89daed31
--- /dev/null
+++ b/ui/cli/tap-wspstat.c
@@ -0,0 +1,293 @@
+/* tap-wspstat.c
+ * wspstat   2003 Jean-Michel FAYARD
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+/* This module provides WSP  statistics to tshark.
+ * It is only used by tshark and not wireshark
+ *
+ */
+
+#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-wsp.h>
+
+#include <wsutil/cmdarg_err.h>
+
+void register_tap_listener_wspstat(void);
+
+/* used to keep track of the stats for a specific PDU type*/
+typedef struct _wsp_pdu_t {
+	const gchar 	*type;
+	guint32		 packets;
+} wsp_pdu_t;
+/* used to keep track of SRT statistics */
+typedef struct _wsp_status_code_t {
+	const gchar	*name;
+	guint32		 packets;
+} wsp_status_code_t;
+/* used to keep track of the statictics for an entire program interface */
+typedef struct _wsp_stats_t {
+	char 		*filter;
+	wsp_pdu_t 	*pdu_stats;
+	guint32	num_pdus;
+	GHashTable	*hash;
+} wspstat_t;
+
+static void
+wsp_reset_hash(gchar *key _U_ , wsp_status_code_t *data, gpointer ptr _U_)
+{
+	data->packets = 0;
+}
+static void
+wsp_print_statuscode(gpointer key, wsp_status_code_t *data, char *format)
+{
+	if (data && (data->packets != 0))
+		printf(format, GPOINTER_TO_INT(key), data->packets , data->name);
+}
+static void
+wsp_free_hash_table( gpointer key, gpointer value, gpointer user_data _U_ )
+{
+	g_free(key);
+	g_free(value);
+}
+static void
+wspstat_reset(void *psp)
+{
+	wspstat_t *sp = (wspstat_t *)psp;
+	guint32 i;
+
+	for (i=1; i<=sp->num_pdus; i++)
+	{
+		sp->pdu_stats[i].packets = 0;
+	}
+	g_hash_table_foreach( sp->hash, (GHFunc)wsp_reset_hash, NULL);
+}
+
+
+/* This callback is invoked whenever the tap system has seen a packet
+ * we might be interested in.
+ * The function is to be used to only update internal state information
+ * in the *tapdata structure, and if there were state changes which requires
+ * the window to be redrawn, return 1 and (*draw) will be called sometime
+ * later.
+ *
+ * We didn't apply a filter when we registered so we will be called for
+ * ALL packets and not just the ones we are collecting stats for.
+ *
+ */
+static gint
+pdut2index(gint pdut)
+{
+	if (pdut <= 0x09)
+		return pdut;
+	if (pdut >= 0x40) {
+		if (pdut <= 0x44) {
+			return pdut - 54;
+		} else if (pdut == 0x60 || pdut == 0x61) {
+			return pdut - 81;
+		}
+	}
+	return 0;
+}
+static gint
+index2pdut(gint pdut)
+{
+	if (pdut <= 0x09)
+		return pdut;
+	if (pdut <= 14)
+		return pdut + 54;
+	if (pdut <= 16)
+		return pdut + 81;
+	return 0;
+}
+static tap_packet_status
+wspstat_packet(void *psp, packet_info *pinfo _U_, epan_dissect_t *edt _U_, const void *pri, tap_flags_t flags _U_)
+{
+	wspstat_t *sp = (wspstat_t *)psp;
+	const wsp_info_value_t *value = (const wsp_info_value_t *)pri;
+	gint idx = pdut2index(value->pdut);
+	tap_packet_status retour = TAP_PACKET_DONT_REDRAW;
+
+	if (value->status_code != 0) {
+		wsp_status_code_t *sc;
+		sc = (wsp_status_code_t *)g_hash_table_lookup(
+				sp->hash,
+				GINT_TO_POINTER(value->status_code));
+		if (!sc) {
+			sc = g_new(wsp_status_code_t, 1);
+			sc -> packets = 1;
+			sc -> name = NULL;
+			g_hash_table_insert(
+				sp->hash,
+				GINT_TO_POINTER(value->status_code),
+				sc);
+		} else {
+			sc->packets++;
+		}
+		retour = TAP_PACKET_REDRAW;
+	}
+
+
+
+	if (idx != 0) {
+		sp->pdu_stats[idx].packets++;
+		retour = TAP_PACKET_REDRAW;
+	}
+	return retour;
+}
+
+
+/* This callback is used when tshark wants us to draw/update our
+ * data to the output device. Since this is tshark only output is
+ * stdout.
+ * TShark will only call this callback once, which is when tshark has
+ * finished reading all packets and exists.
+ * If used with wireshark this may be called any time, perhaps once every 3
+ * seconds or so.
+ * This function may even be called in parallell with (*reset) or (*draw)
+ * so make sure there are no races. The data in the rpcstat_t can thus change
+ * beneath us. Beware.
+ */
+static void
+wspstat_draw(void *psp)
+{
+	wspstat_t *sp = (wspstat_t *)psp;
+	guint32 i;
+
+	printf("\n");
+	printf("===================================================================\n");
+	printf("WSP Statistics:\n");
+	printf("%-23s %9s || %-23s %9s\n", "PDU Type", "Packets", "PDU Type", "Packets");
+	for (i=1; i <= ((sp->num_pdus+1)/2); i++)
+	{
+		guint32 ii = i+sp->num_pdus/2;
+		printf("%-23s %9u", sp->pdu_stats[i ].type, sp->pdu_stats[i ].packets);
+		printf(" || ");
+		if (ii< (sp->num_pdus) )
+			printf("%-23s %9u\n", sp->pdu_stats[ii].type, sp->pdu_stats[ii].packets);
+		else
+			printf("\n");
+		}
+	printf("\nStatus code in reply packets\n");
+	printf(		"Status Code    Packets  Description\n");
+	g_hash_table_foreach( sp->hash, (GHFunc) wsp_print_statuscode,
+			(gpointer)"       0x%02X  %9d  %s\n" ) ;
+	printf("===================================================================\n");
+}
+
+/* When called, this function will create a new instance of wspstat.
+ * program and version are whick onc-rpc program/version we want to
+ * collect statistics for.
+ * This function is called from tshark when it parses the -z wsp, arguments
+ * and it creates a new instance to store statistics in and registers this
+ * new instance for the wsp tap.
+ */
+static void
+wspstat_init(const char *opt_arg, void *userdata _U_)
+{
+	wspstat_t          *sp;
+	const char         *filter = NULL;
+	guint32             i;
+	GString	           *error_string;
+	wsp_status_code_t  *sc;
+	const value_string *wsp_vals_status_p;
+
+	if (!strncmp (opt_arg, "wsp,stat,", 9)) {
+		filter = opt_arg+9;
+	} else {
+		filter = NULL;
+	}
+
+
+	sp = g_new(wspstat_t, 1);
+	sp->hash = g_hash_table_new(g_direct_hash, g_direct_equal);
+	wsp_vals_status_p = VALUE_STRING_EXT_VS_P(&wsp_vals_status_ext);
+	for (i=0; wsp_vals_status_p[i].strptr; i++ )
+	{
+		gint *key;
+		sc = g_new(wsp_status_code_t, 1);
+		key = g_new(gint, 1);
+		sc->packets = 0;
+		sc->name = wsp_vals_status_p[i].strptr;
+		*key = wsp_vals_status_p[i].value;
+		g_hash_table_insert(
+				sp->hash,
+				key,
+				sc);
+	}
+	sp->num_pdus = 16;
+	sp->pdu_stats = g_new(wsp_pdu_t, (sp->num_pdus+1));
+	sp->filter = g_strdup(filter);
+
+	for (i=0; i<sp->num_pdus; i++)
+	{
+		sp->pdu_stats[i].packets = 0;
+		sp->pdu_stats[i].type = try_val_to_str_ext( index2pdut( i ), &wsp_vals_pdu_type_ext) ;
+	}
+
+	error_string = register_tap_listener(
+			"wsp",
+			sp,
+			filter,
+			0,
+			wspstat_reset,
+			wspstat_packet,
+			wspstat_draw,
+			NULL);
+	if (error_string) {
+		/* error, we failed to attach to the tap. clean up */
+		g_free(sp->pdu_stats);
+		g_free(sp->filter);
+		g_hash_table_foreach( sp->hash, (GHFunc) wsp_free_hash_table, NULL ) ;
+		g_hash_table_destroy( sp->hash );
+		g_free(sp);
+		cmdarg_err("Couldn't register wsp,stat tap: %s",
+				error_string->str);
+		g_string_free(error_string, TRUE);
+		exit(1);
+	}
+}
+
+static stat_tap_ui wspstat_ui = {
+	REGISTER_STAT_GROUP_GENERIC,
+	NULL,
+	"wsp,stat",
+	wspstat_init,
+	0,
+	NULL
+};
+
+void
+register_tap_listener_wspstat(void)
+{
+	register_stat_tap_ui(&wspstat_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:
+ */