/* tap-mgcpstat.c
 * mgcpstat   2003 Lars Roland
 *
 * 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.
 */

#include "config.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.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-mgcp.h"
#include "epan/timestats.h"

#define NUM_TIMESTATS 11

void register_tap_listener_mgcpstat(void);

/* used to keep track of the statistics for an entire program interface */
typedef struct _mgcpstat_t {
	char *filter;
	timestat_t rtd[NUM_TIMESTATS];
	guint32 open_req_num;
	guint32 disc_rsp_num;
	guint32 req_dup_num;
	guint32 rsp_dup_num;
} mgcpstat_t;

static const value_string mgcp_mesage_type[] = {
	{  0, "Overall"},
	{  1, "EPCF   "},
	{  2, "CRCX   "},
	{  3, "MDCX   "},
	{  4, "DLCX   "},
	{  5, "RQNT   "},
	{  6, "NTFY   "},
	{  7, "AUEP   "},
	{  8, "AUCX   "},
	{  9, "RSIP   "},
	{  0, NULL}
};

static int
mgcpstat_packet(void *pms, packet_info *pinfo, epan_dissect_t *edt _U_, const void *pmi)
{
	mgcpstat_t *ms = (mgcpstat_t *)pms;
	const mgcp_info_t *mi = (const mgcp_info_t *)pmi;
	nstime_t delta;
	int ret = 0;

	switch (mi->mgcp_type) {

	case MGCP_REQUEST:
		if (mi->is_duplicate) {
			/* Duplicate is ignored */
			ms->req_dup_num++;
		}
		else {
			ms->open_req_num++;
		}
		break;

	case MGCP_RESPONSE:
		if (mi->is_duplicate) {
			/* Duplicate is ignored */
			ms->rsp_dup_num++;
		}
		else if (!mi->request_available) {
			/* no request was seen */
			ms->disc_rsp_num++;
		}
		else {
			ms->open_req_num--;
			/* calculate time delta between request and response */
			nstime_delta(&delta, &pinfo->fd->abs_ts, &mi->req_time);

			time_stat_update(&(ms->rtd[0]), &delta, pinfo);

			if (g_ascii_strncasecmp(mi->code, "EPCF", 4) == 0 ) {
				time_stat_update(&(ms->rtd[1]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "CRCX", 4) == 0 ) {
				time_stat_update(&(ms->rtd[2]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "MDCX", 4) == 0 ) {
				time_stat_update(&(ms->rtd[3]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "DLCX", 4) == 0 ) {
				time_stat_update(&(ms->rtd[4]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "RQNT", 4) == 0 ) {
				time_stat_update(&(ms->rtd[5]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "NTFY", 4) == 0 ) {
				time_stat_update(&(ms->rtd[6]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "AUEP", 4) == 0 ) {
				time_stat_update(&(ms->rtd[7]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "AUCX", 4) == 0 ) {
				time_stat_update(&(ms->rtd[8]), &delta, pinfo);
			}
			else if (g_ascii_strncasecmp(mi->code, "RSIP", 4) == 0 ) {
				time_stat_update(&(ms->rtd[9]), &delta, pinfo);
			}
			else {
				time_stat_update(&(ms->rtd[10]), &delta, pinfo);
			}

			ret = 1;
		}
		break;

	default:
		break;
	}

	return ret;
}

static void
mgcpstat_draw(void *pms)
{
	mgcpstat_t *ms = (mgcpstat_t *)pms;
	gchar* tmp_str;
	int i;

	/* printing results */
	printf("\n");
	printf("=====================================================================================================\n");
	printf("MGCP Response Time Delay (RTD) Statistics:\n");
	printf("Filter for statistics: %s\n", ms->filter ? ms->filter : "");
	printf("Duplicate requests: %u\n", ms->req_dup_num);
	printf("Duplicate responses: %u\n", ms->rsp_dup_num);
	printf("Open requests: %u\n", ms->open_req_num);
	printf("Discarded responses: %u\n", ms->disc_rsp_num);
	printf("Type    | Messages   |    Min RTD    |    Max RTD    |    Avg RTD    | Min in Frame | Max in Frame |\n");
	for (i=0; i<NUM_TIMESTATS; i++) {
		if (ms->rtd[i].num) {
			tmp_str = val_to_str_wmem(NULL, i, mgcp_mesage_type, "Other (%d)");
			printf("%s | %7u    | %8.2f msec | %8.2f msec | %8.2f msec |  %10u  |  %10u  |\n",
			       tmp_str, ms->rtd[i].num,
			       nstime_to_msec(&(ms->rtd[i].min)), nstime_to_msec(&(ms->rtd[i].max)),
			       get_average(&(ms->rtd[i].tot), ms->rtd[i].num),
			       ms->rtd[i].min_num, ms->rtd[i].max_num
			);
			wmem_free(NULL, tmp_str);
		}
	}
	printf("=====================================================================================================\n");
}


static void
mgcpstat_init(const char *opt_arg, void *userdata _U_)
{
	mgcpstat_t *ms;
	int i;
	GString *error_string;

	ms = g_new(mgcpstat_t, 1);
	if (!strncmp(opt_arg, "mgcp,rtd,", 9)) {
		ms->filter = g_strdup(opt_arg+9);
	} else {
		ms->filter = NULL;
	}

	for (i=0; i<NUM_TIMESTATS; i++) {
		ms->rtd[i].num	     = 0;
		ms->rtd[i].min_num   = 0;
		ms->rtd[i].max_num   = 0;
		ms->rtd[i].min.secs  = 0;
		ms->rtd[i].min.nsecs = 0;
		ms->rtd[i].max.secs  = 0;
		ms->rtd[i].max.nsecs = 0;
		ms->rtd[i].tot.secs  = 0;
		ms->rtd[i].tot.nsecs = 0;
	}

	ms->open_req_num = 0;
	ms->disc_rsp_num = 0;
	ms->req_dup_num	 = 0;
	ms->rsp_dup_num	 = 0;

	error_string = register_tap_listener("mgcp", ms, ms->filter, 0, NULL, mgcpstat_packet, mgcpstat_draw);
	if (error_string) {
		/* error, we failed to attach to the tap. clean up */
		g_free(ms->filter);
		g_free(ms);

		fprintf(stderr, "tshark: Couldn't register mgcp,rtd tap: %s\n",
			error_string->str);
		g_string_free(error_string, TRUE);
		exit(1);
	}
}

static stat_tap_ui mgcpstat_ui = {
	REGISTER_STAT_GROUP_GENERIC,
	NULL,
	"mgcp,rtd",
	mgcpstat_init,
	0,
	NULL
};

void
register_tap_listener_mgcpstat(void)
{
	/* We don't register this tap, if we don't have the mgcp plugin loaded.*/
	if (find_tap_id("mgcp")) {
		register_stat_tap_ui(&mgcpstat_ui, NULL);
	}
}

/*
 * Editor modelines  -  http://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:
 */