gtk/diameter_stat.c

   1 /* diameter_stat.c
   2  *  Diameter Service Response Time Statistics
   3  * (c) 2008 Abhik Sarkar
   4  *
   5  * Based almost completely on gtp_stat by Kari Tiirikainen
   6  *
   7  * $Id$
   8  *
   9  * Wireshark - Network traffic analyzer
  10  * By Gerald Combs <gerald@wireshark.org>
  11  * Copyright 1998 Gerald Combs
  12  *
  13  * This program is free software; you can redistribute it and/or
  14  * modify it under the terms of the GNU General Public License
  15  * as published by the Free Software Foundation; either version 2
  16  * of the License, or (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License
  24  * along with this program; if not, write to the Free Software
  25  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  26  */
  27
  28 #ifdef HAVE_CONFIG_H
  29 # include "config.h"
  30 #endif
  31
  32 #ifdef HAVE_SYS_TYPES_H
  33 # include <sys/types.h>
  34 #endif
  35
  36 #include <string.h>
  37
  38 #include <gtk/gtk.h>
  39
  40 #include <epan/packet_info.h>
  41 #include <epan/epan.h>
  42 #include <epan/tap.h>
  43 #include <epan/dissectors/packet-diameter.h>
  44
  45 #include "../timestats.h"
  46 #include "../simple_dialog.h"
  47 #include "../file.h"
  48 #include "../stat_menu.h"
  49
  50 #include "gtk/gui_utils.h"
  51 #include "gtk/dlg_utils.h"
  52 #include "gtk/service_response_time_table.h"
  53 #include "gtk/tap_param_dlg.h"
  54 #include "gtk/gtkglobals.h"
  55 #include "gtk/main.h"
  56
  57 #include "gtk/old-gtk-compat.h"
  58
  59 /* used to keep track of the statistics for an entire program interface */
  60 typedef struct _diameterstat_t {
  61         GtkWidget *win;
  62         srt_stat_table diameter_srt_table;
  63 } diameterstat_t;
  64
  65 static GHashTable* cmd_str_hash;
  66
  67 static void
  68 diameterstat_set_title(diameterstat_t *diameter)
  69 {
  70         char            *title;
  71
  72         title = g_strdup_printf("Diameter Service Response Time statistics: %s",
  73             cf_get_display_name(&cfile));
  74         gtk_window_set_title(GTK_WINDOW(diameter->win), title);
  75         g_free(title);
  76 }
  77
  78 static void
  79 diameterstat_reset(void *pdiameter)
  80 {
  81         diameterstat_t *diameter=(diameterstat_t *)pdiameter;
  82
  83         reset_srt_table_data(&diameter->diameter_srt_table);
  84         diameterstat_set_title(diameter);
  85 }
  86
  87
  88 static int
  89 diameterstat_packet(void *pdiameter, packet_info *pinfo, epan_dissect_t *edt _U_, const void *pdi)
  90 {
  91         const diameter_req_ans_pair_t *diameter=pdi;
  92         diameterstat_t *fs=(diameterstat_t *)pdiameter;
  93         int* idx = NULL;
  94
  95         /* Process only answers where corresponding request is found.
  96          * Unpaired daimeter messages are currently not supported by statistics.
  97          * Return 0, since redraw is not needed. */
  98         if(!diameter || diameter->processing_request || !diameter->req_frame)
  99                 return 0;
 100
 101         idx = (int*) g_hash_table_lookup(cmd_str_hash, diameter->cmd_str);
 102         if (idx == NULL) {
 103                 idx = g_malloc(sizeof(int));
 104                 *idx = (int) g_hash_table_size(cmd_str_hash);
 105                 g_hash_table_insert(cmd_str_hash, (gchar*) diameter->cmd_str, idx);
 106                 init_srt_table_row(&fs->diameter_srt_table, *idx,  (const char*) diameter->cmd_str);
 107         }
 108
 109         add_srt_table_data(&fs->diameter_srt_table, *idx, &diameter->req_time, pinfo);
 110
 111         return 1;
 112 }
 113
 114
 115
 116 static void
 117 diameterstat_draw(void *pdiameter)
 118 {
 119         diameterstat_t *diameter=(diameterstat_t *)pdiameter;
 120
 121         draw_srt_table_data(&diameter->diameter_srt_table);
 122 }
 123
 124
 125 static void
 126 win_destroy_cb(GtkWindow *win _U_, gpointer data)
 127 {
 128         diameterstat_t *diameter=(diameterstat_t *)data;
 129
 130         protect_thread_critical_region();
 131         remove_tap_listener(diameter);
 132         unprotect_thread_critical_region();
 133
 134         free_srt_table_data(&diameter->diameter_srt_table);
 135         g_free(diameter);
 136         g_hash_table_destroy(cmd_str_hash);
 137 }
 138
 139
 140 static void
 141 gtk_diameterstat_init(const char *optarg, void *userdata _U_)
 142 {
 143         diameterstat_t *diameter;
 144         const char *filter=NULL;
 145         GtkWidget *label;
 146         char *filter_string;
 147         GString *error_string;
 148         GtkWidget *vbox;
 149         GtkWidget *bbox;
 150         GtkWidget *close_bt;
 151         int* idx;
 152
 153         if(!strncmp(optarg,"diameter,",9)){
 154                 filter=optarg+9;
 155         } else {
 156                 filter="diameter"; /*NULL doesn't work here like in LDAP. Too little time/lazy to find out why ?*/
 157         }
 158
 159         diameter=g_malloc(sizeof(diameterstat_t));
 160         idx = g_malloc(sizeof(int));
 161         *idx = 0;
 162         cmd_str_hash = g_hash_table_new(g_str_hash,g_str_equal);
 163         g_hash_table_insert(cmd_str_hash, (gchar *)"Unknown", idx);
 164
 165         diameter->win = dlg_window_new("diameter-stat");  /* transient_for top_level */
 166         gtk_window_set_destroy_with_parent (GTK_WINDOW(diameter->win), TRUE);
 167         gtk_window_set_default_size(GTK_WINDOW(diameter->win), 550, 400);
 168         diameterstat_set_title(diameter);
 169
 170         vbox=gtk_vbox_new(FALSE, 3);
 171         gtk_container_add(GTK_CONTAINER(diameter->win), vbox);
 172         gtk_container_set_border_width(GTK_CONTAINER(vbox), 12);
 173
 174         label=gtk_label_new("Diameter Service Response Time statistics");
 175         gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);
 176
 177         filter_string = g_strdup_printf("Filter: %s", filter ? filter : "");
 178         label=gtk_label_new(filter_string);
 179         g_free(filter_string);
 180         gtk_label_set_line_wrap(GTK_LABEL(label), TRUE);
 181         gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);
 182
 183         label=gtk_label_new("Diameter Requests");
 184         gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);
 185
 186         /* We must display TOP LEVEL Widget before calling init_srt_table() */
 187         gtk_widget_show_all(diameter->win);
 188
 189         /** @todo the filter to use in stead of NULL is "diameter.cmd.code"
 190          * to enable the filter popup in the service response time dalouge
 191          * Note to make it work the command code must be stored rather than the
 192          * index.
 193          */
 194         init_srt_table(&diameter->diameter_srt_table, 1, vbox, NULL);
 195         init_srt_table_row(&diameter->diameter_srt_table, 0, "Unknown");
 196
 197         error_string=register_tap_listener("diameter", diameter, filter, 0, diameterstat_reset, diameterstat_packet, diameterstat_draw);
 198         if(error_string){
 199                 simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK, "%s", error_string->str);
 200                 g_string_free(error_string, TRUE);
 201                 g_free(diameter);
 202                 return;
 203         }
 204
 205         /* Button row. */
 206         bbox = dlg_button_row_new(GTK_STOCK_CLOSE, NULL);
 207         gtk_box_pack_end(GTK_BOX(vbox), bbox, FALSE, FALSE, 0);
 208
 209         close_bt = g_object_get_data(G_OBJECT(bbox), GTK_STOCK_CLOSE);
 210         window_set_cancel_button(diameter->win, close_bt, window_cancel_button_cb);
 211
 212         g_signal_connect(diameter->win, "delete_event", G_CALLBACK(window_delete_event_cb), NULL);
 213         g_signal_connect(diameter->win, "destroy", G_CALLBACK(win_destroy_cb), diameter);
 214
 215         gtk_widget_show_all(diameter->win);
 216         window_present(diameter->win);
 217
 218         cf_retap_packets(&cfile);
 219         gdk_window_raise(gtk_widget_get_window(diameter->win));
 220 }
 221
 222 static tap_param diameter_stat_params[] = {
 223         { PARAM_FILTER, "Filter", NULL }
 224 };
 225
 226 static tap_param_dlg diameter_stat_dlg = {
 227         "Diameter Service Response Time Statistics",
 228         "diameter",
 229         gtk_diameterstat_init,
 230         -1,
 231         G_N_ELEMENTS(diameter_stat_params),
 232         diameter_stat_params
 233 };
 234
 235 void
 236 register_tap_listener_gtkdiameterstat(void)
 237 {
 238         register_dfilter_stat(&diameter_stat_dlg, "Diameter",
 239             REGISTER_STAT_GROUP_RESPONSE_TIME);
 240 }
 241
 242 #ifdef MAIN_MENU_USE_UIMANAGER
 243 void diameter_srt_cb(GtkAction *action, gpointer user_data _U_)
 244 {
 245         tap_param_dlg_cb(action, &diameter_stat_dlg);
 246 }
 247 #endif