![Evan Huus](/assets/img/avatar_default.png)
was done using textual search+replace, not anything syntax-aware, so presumably it got most comments as well (except where there were typos). Use a consistent coding style, and make proper use of the WS_DLL_* defines. Group the functions appropriately in the header. I ended up getting rid of most of the explanatory comments since many of them duplicated what was in the value_string.c file (and were out of sync with the recent updates I made to those in r48633). Presumably most of the comments should be in the .h file not the .c file, but there's enough churn ahead that it's not worth fixing yet. Part of https://bugs.wireshark.org/bugzilla/show_bug.cgi?id=8467 svn path=/trunk/; revision=48634
281 lines
7.2 KiB
C
281 lines
7.2 KiB
C
/* tap-rpcstat.c
|
|
* wspstat 2003 Jean-Michel FAYARD
|
|
*
|
|
* $Id$
|
|
*
|
|
* Wireshark - Network traffic analyzer
|
|
* By Gerald Combs <gerald@wireshark.org>
|
|
* Copyright 1998 Gerald Combs
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/* This module provides WSP statistics to tshark.
|
|
* It is only used by tshark and not wireshark
|
|
*
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <string.h>
|
|
#include "epan/packet_info.h"
|
|
#include <epan/tap.h>
|
|
#include <epan/stat_cmd_args.h>
|
|
#include "epan/value_string.h"
|
|
#include <epan/dissectors/packet-wsp.h>
|
|
|
|
/* 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(gint *key, wsp_status_code_t *data, char* format)
|
|
{
|
|
if (data && (data->packets!=0))
|
|
printf(format, *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 didnt 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 int
|
|
wspstat_packet(void *psp, packet_info *pinfo _U_, epan_dissect_t *edt _U_, const void *pri)
|
|
{
|
|
wspstat_t *sp=(wspstat_t *)psp;
|
|
const wsp_info_value_t *value=(const wsp_info_value_t *)pri;
|
|
gint idx = pdut2index(value->pdut);
|
|
int retour=0;
|
|
|
|
if (value->status_code != 0) {
|
|
gint *key=g_new(gint,1);
|
|
wsp_status_code_t *sc;
|
|
*key=value->status_code ;
|
|
sc = (wsp_status_code_t *)g_hash_table_lookup(
|
|
sp->hash,
|
|
key);
|
|
if (!sc) {
|
|
sc = g_new(wsp_status_code_t,1);
|
|
sc -> packets = 1;
|
|
sc -> name = NULL;
|
|
g_hash_table_insert(
|
|
sp->hash,
|
|
key,
|
|
sc);
|
|
} else {
|
|
sc->packets++;
|
|
}
|
|
retour=1;
|
|
}
|
|
|
|
|
|
|
|
if (idx!=0) {
|
|
sp->pdu_stats[ idx ].packets++;
|
|
retour = 1;
|
|
}
|
|
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 %9d", sp->pdu_stats[i ].type, sp->pdu_stats[i ].packets);
|
|
printf(" || ");
|
|
if (ii< (sp->num_pdus) )
|
|
printf("%-23s %9d\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 *optarg, 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 (optarg, "wsp,stat," , 9)){
|
|
filter=optarg+9;
|
|
} else {
|
|
filter=NULL;
|
|
}
|
|
|
|
|
|
sp = g_new(wspstat_t,1);
|
|
sp->hash = g_hash_table_new( g_int_hash, g_int_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));
|
|
if(filter){
|
|
sp->filter=g_strdup(filter);
|
|
} else {
|
|
sp->filter=NULL;
|
|
}
|
|
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);
|
|
if (error_string){
|
|
/* error, we failed to attach to the tap. clean up */
|
|
g_free(sp->pdu_stats);
|
|
g_free(sp->filter);
|
|
g_free(sp);
|
|
g_hash_table_foreach( sp->hash, (GHFunc) wsp_free_hash_table, NULL ) ;
|
|
g_hash_table_destroy( sp->hash );
|
|
fprintf(stderr, "tshark: Couldn't register wsp,stat tap: %s\n",
|
|
error_string->str);
|
|
g_string_free(error_string, TRUE);
|
|
exit(1);
|
|
}
|
|
}
|
|
void
|
|
register_tap_listener_wspstat(void)
|
|
{
|
|
register_stat_cmd_arg("wsp,stat,", wspstat_init,NULL);
|
|
}
|