diff options
Diffstat (limited to 'ui/cli/tap-rtspstat.c')
| -rw-r--r-- | ui/cli/tap-rtspstat.c | 289 |
1 files changed, 289 insertions, 0 deletions
diff --git a/ui/cli/tap-rtspstat.c b/ui/cli/tap-rtspstat.c new file mode 100644 index 00000000..a42be606 --- /dev/null +++ b/ui/cli/tap-rtspstat.c @@ -0,0 +1,289 @@ +/* tap-rtspstat.c + * tap-rtspstat March 2011 + * + * Stephane GORSE (Orange Labs / France Telecom) + * Copied from Jean-Michel FAYARD's works (HTTP) + * + * 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/value_string.h> +#include <epan/tap.h> +#include <epan/stat_tap_ui.h> +#include <epan/dissectors/packet-rtsp.h> + +#include <wsutil/wslog.h> + +#include <wsutil/cmdarg_err.h> + +void register_tap_listener_rtspstat(void); + +/* used to keep track of the statictics for an entire program interface */ +typedef struct _rtsp_stats_t { + char *filter; + GHashTable *hash_responses; + GHashTable *hash_requests; +} rtspstat_t; + +/* used to keep track of the stats for a specific response code + * for example it can be { 3, 404, "Not Found" ,...} + * which means we captured 3 reply rtsp/1.1 404 Not Found */ +typedef struct _rtsp_response_code_t { + guint32 packets; /* 3 */ + guint response_code; /* 404 */ + const gchar *name; /* Not Found */ + rtspstat_t *sp; +} rtsp_response_code_t; + +/* used to keep track of the stats for a specific request string */ +typedef struct _rtsp_request_methode_t { + gchar *response; /* eg. : SETUP */ + guint32 packets; + rtspstat_t *sp; +} rtsp_request_methode_t; + + +/* insert some entries */ +static void +rtsp_init_hash( rtspstat_t *sp) +{ + int i; + + sp->hash_responses = g_hash_table_new(g_direct_hash, g_direct_equal); + + for (i=0 ; rtsp_status_code_vals[i].strptr ; i++ ) + { + rtsp_response_code_t *sc = g_new (rtsp_response_code_t, 1); + sc->packets = 0; + sc->response_code = rtsp_status_code_vals[i].value; + sc->name = rtsp_status_code_vals[i].strptr; + sc->sp = sp; + g_hash_table_insert( sc->sp->hash_responses, GINT_TO_POINTER(rtsp_status_code_vals[i].value), sc); + } + sp->hash_requests = g_hash_table_new( g_str_hash, g_str_equal); +} +static void +rtsp_draw_hash_requests( gchar *key _U_ , rtsp_request_methode_t *data, gchar * format) +{ + if (data->packets == 0) + return; + printf( format, data->response, data->packets); +} + +static void +rtsp_draw_hash_responses( gpointer* key _U_ , rtsp_response_code_t *data, char * format) +{ + if (data == NULL) { + ws_warning("No data available, key=%d\n", GPOINTER_TO_INT(key)); + exit(EXIT_FAILURE); + } + if (data->packets == 0) + return; + /* " RTSP %3d %-35s %9d packets", */ + printf(format, data->response_code, data->name, data->packets ); +} + + + +/* NOT USED at this moment */ +/* +static void +rtsp_free_hash( gpointer key, gpointer value, gpointer user_data _U_ ) +{ + g_free(key); + g_free(value); +} +*/ +static void +rtsp_reset_hash_responses(gchar *key _U_ , rtsp_response_code_t *data, gpointer ptr _U_ ) +{ + data->packets = 0; +} +static void +rtsp_reset_hash_requests(gchar *key _U_ , rtsp_request_methode_t *data, gpointer ptr _U_ ) +{ + data->packets = 0; +} + +static void +rtspstat_reset(void *psp ) +{ + rtspstat_t *sp = (rtspstat_t *)psp; + + g_hash_table_foreach( sp->hash_responses, (GHFunc)rtsp_reset_hash_responses, NULL); + g_hash_table_foreach( sp->hash_requests, (GHFunc)rtsp_reset_hash_requests, NULL); + +} + +static tap_packet_status +rtspstat_packet(void *psp , packet_info *pinfo _U_, epan_dissect_t *edt _U_, const void *pri, tap_flags_t flags _U_) +{ + const rtsp_info_value_t *value = (const rtsp_info_value_t *)pri; + rtspstat_t *sp = (rtspstat_t *) psp; + + /* We are only interested in reply packets with a status code */ + /* Request or reply packets ? */ + if (value->response_code != 0) { + rtsp_response_code_t *sc; + + sc = (rtsp_response_code_t *)g_hash_table_lookup( + sp->hash_responses, + GINT_TO_POINTER(value->response_code)); + if (sc == NULL) { + gint key; + /* non standard status code ; we classify it as others + * in the relevant category (Informational,Success,Redirection,Client Error,Server Error) + */ + int i = value->response_code; + if ((i < 100) || (i >= 600)) { + return TAP_PACKET_DONT_REDRAW; + } + else if (i < 200) { + key = 199; /* Hopefully, this status code will never be used */ + } + else if (i < 300) { + key = 299; + } + else if (i < 400) { + key = 399; + } + else if (i < 500) { + key = 499; + } + else { + key = 599; + } + sc = (rtsp_response_code_t *)g_hash_table_lookup( + sp->hash_responses, + GINT_TO_POINTER(key)); + if (sc == NULL) + return TAP_PACKET_DONT_REDRAW; + } + sc->packets++; + } + else if (value->request_method) { + rtsp_request_methode_t *sc; + + sc = (rtsp_request_methode_t *)g_hash_table_lookup( + sp->hash_requests, + value->request_method); + if (sc == NULL) { + sc = g_new(rtsp_request_methode_t, 1); + sc->response = g_strdup( value->request_method ); + sc->packets = 1; + sc->sp = sp; + g_hash_table_insert( sp->hash_requests, sc->response, sc); + } else { + sc->packets++; + } + } else { + return TAP_PACKET_DONT_REDRAW; + } + return TAP_PACKET_REDRAW; +} + + +static void +rtspstat_draw(void *psp ) +{ + rtspstat_t *sp = (rtspstat_t *)psp; + printf("\n"); + printf("===================================================================\n"); + if (!sp->filter || !sp->filter[0]) + printf("RTSP Statistics\n"); + else + printf("RTSP Statistics with filter %s\n", sp->filter); + + printf("* RTSP Response Status Codes Packets\n"); + g_hash_table_foreach( sp->hash_responses, (GHFunc)rtsp_draw_hash_responses, + (gpointer)" %3d %-35s %9d\n"); + printf("* RTSP Request Methods Packets\n"); + g_hash_table_foreach( sp->hash_requests, (GHFunc)rtsp_draw_hash_requests, + (gpointer)" %-39s %9d\n"); + printf("===================================================================\n"); +} + + + +/* When called, this function will create a new instance of rtspstat. + */ +static void +rtspstat_init(const char *opt_arg, void *userdata _U_) +{ + rtspstat_t *sp; + const char *filter = NULL; + GString *error_string; + + if (!strncmp (opt_arg, "rtsp,stat,", 10)) { + filter = opt_arg+10; + } else { + filter = NULL; + } + + sp = g_new(rtspstat_t, 1); + sp->filter = g_strdup(filter); + /*g_hash_table_foreach( rtsp_status, (GHFunc)rtsp_reset_hash_responses, NULL);*/ + + + error_string = register_tap_listener( + "rtsp", + sp, + filter, + 0, + rtspstat_reset, + rtspstat_packet, + rtspstat_draw, + NULL); + if (error_string) { + /* error, we failed to attach to the tap. clean up */ + g_free(sp->filter); + g_free(sp); + cmdarg_err("Couldn't register rtsp,stat tap: %s", + error_string->str); + g_string_free(error_string, TRUE); + exit(1); + } + + rtsp_init_hash(sp); +} + +static stat_tap_ui rtspstat_ui = { + REGISTER_STAT_GROUP_GENERIC, + NULL, + "rtsp,stat", + rtspstat_init, + 0, + NULL +}; + +void +register_tap_listener_rtspstat(void) +{ + register_stat_tap_ui(&rtspstat_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: + */ |