ui/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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  26  */
  27
  28 #include "config.h"
  29
  30 #include <string.h>
  31
  32 #include <gtk/gtk.h>
  33
  34 #include <epan/packet_info.h>
  35 #include <epan/epan.h>
  36 #include <epan/tap.h>
  37 #include <epan/dissectors/packet-diameter.h>
  38
  39 #include "../file.h"
  40 #include "../stat_menu.h"
  41
  42 #include "ui/simple_dialog.h"
  43
  44 #include "ui/gtk/gui_utils.h"
  45 #include "ui/gtk/dlg_utils.h"
  46 #include "ui/gtk/service_response_time_table.h"
  47 #include "ui/gtk/tap_param_dlg.h"
  48 #include "ui/gtk/gtkglobals.h"
  49 #include "ui/gtk/main.h"
  50
  51 #include "ui/gtk/old-gtk-compat.h"
  52
  53 /* used to keep track of the statistics for an entire program interface */
  54 typedef struct _diameterstat_t {
  55         GtkWidget *win;
  56         srt_stat_table diameter_srt_table;
  57 } diameterstat_t;
  58
  59 static GHashTable* cmd_str_hash;
  60
  61 static void
  62 diameterstat_set_title(diameterstat_t *diameter)
  63 {
  64         set_window_title(diameter->win, "Diameter Service Response Time statistics");
  65 }
  66
  67 static void
  68 diameterstat_reset(void *pdiameter)
  69 {
  70         diameterstat_t *diameter=(diameterstat_t *)pdiameter;
  71
  72         reset_srt_table_data(&diameter->diameter_srt_table);
  73         diameterstat_set_title(diameter);
  74 }
  75
  76
  77 static int
  78 diameterstat_packet(void *pdiameter, packet_info *pinfo, epan_dissect_t *edt _U_, const void *pdi)
  79 {
  80         const diameter_req_ans_pair_t *diameter=(diameter_req_ans_pair_t *)pdi;
  81         diameterstat_t *fs=(diameterstat_t *)pdiameter;
  82         int* idx = NULL;
  83
  84         /* Process only answers where corresponding request is found.
  85          * Unpaired daimeter messages are currently not supported by statistics.
  86          * Return 0, since redraw is not needed. */
  87         if(!diameter || diameter->processing_request || !diameter->req_frame)
  88                 return 0;
  89
  90         idx = (int*) g_hash_table_lookup(cmd_str_hash, diameter->cmd_str);
  91         if (idx == NULL) {
  92                 idx = (int *)g_malloc(sizeof(int));
  93                 *idx = (int) g_hash_table_size(cmd_str_hash);
  94                 g_hash_table_insert(cmd_str_hash, (gchar*) diameter->cmd_str, idx);
  95                 init_srt_table_row(&fs->diameter_srt_table, *idx,  (const char*) diameter->cmd_str);
  96         }
  97
  98         add_srt_table_data(&fs->diameter_srt_table, *idx, &diameter->req_time, pinfo);
  99
 100         return 1;
 101 }
 102
 103
 104
 105 static void
 106 diameterstat_draw(void *pdiameter)
 107 {
 108         diameterstat_t *diameter=(diameterstat_t *)pdiameter;
 109
 110         draw_srt_table_data(&diameter->diameter_srt_table);
 111 }
 112
 113
 114 static void
 115 win_destroy_cb(GtkWindow *win _U_, gpointer data)
 116 {
 117         diameterstat_t *diameter=(diameterstat_t *)data;
 118
 119         remove_tap_listener(diameter);
 120
 121         free_srt_table_data(&diameter->diameter_srt_table);
 122         g_free(diameter);
 123         g_hash_table_destroy(cmd_str_hash);
 124 }
 125
 126
 127 static void
 128 gtk_diameterstat_init(const char *opt_arg, void *userdata _U_)
 129 {
 130         diameterstat_t *diameter;
 131         const char *filter=NULL;
 132         GtkWidget *label;
 133         char *filter_string;
 134         GString *error_string;
 135         GtkWidget *vbox;
 136         GtkWidget *bbox;
 137         GtkWidget *close_bt;
 138         int* idx;
 139
 140         if(!strncmp(opt_arg,"diameter,",9)){
 141                 filter=opt_arg+9;
 142         } else {
 143                 filter="diameter"; /*NULL doesn't work here like in LDAP. Too little time/lazy to find out why ?*/
 144         }
 145
 146         diameter=(diameterstat_t *)g_malloc(sizeof(diameterstat_t));
 147         idx = (int *)g_malloc(sizeof(int));
 148         *idx = 0;
 149         cmd_str_hash = g_hash_table_new(g_str_hash,g_str_equal);
 150         g_hash_table_insert(cmd_str_hash, (gchar *)"Unknown", idx);
 151
 152         diameter->win = dlg_window_new("diameter-stat");  /* transient_for top_level */
 153         gtk_window_set_destroy_with_parent (GTK_WINDOW(diameter->win), TRUE);
 154         gtk_window_set_default_size(GTK_WINDOW(diameter->win), 550, 400);
 155         diameterstat_set_title(diameter);
 156
 157         vbox=ws_gtk_box_new(GTK_ORIENTATION_VERTICAL, 3, FALSE);
 158         gtk_container_add(GTK_CONTAINER(diameter->win), vbox);
 159         gtk_container_set_border_width(GTK_CONTAINER(vbox), 12);
 160
 161         label=gtk_label_new("Diameter Service Response Time statistics");
 162         gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);
 163
 164         filter_string = g_strdup_printf("Filter: %s", filter ? filter : "");
 165         label=gtk_label_new(filter_string);
 166         g_free(filter_string);
 167         gtk_label_set_line_wrap(GTK_LABEL(label), TRUE);
 168         gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);
 169
 170         label=gtk_label_new("Diameter Requests");
 171         gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);
 172
 173         /* We must display TOP LEVEL Widget before calling init_srt_table() */
 174         gtk_widget_show_all(diameter->win);
 175
 176         /** @todo the filter to use in stead of NULL is "diameter.cmd.code"
 177          * to enable the filter popup in the service response time dalouge
 178          * Note to make it work the command code must be stored rather than the
 179          * index.
 180          */
 181         init_srt_table(&diameter->diameter_srt_table, 1, vbox, NULL);
 182         init_srt_table_row(&diameter->diameter_srt_table, 0, "Unknown");
 183
 184         error_string=register_tap_listener(
 185                 "diameter",
 186                 diameter, filter,
 187                 TL_REQUIRES_PROTO_TREE,
 188                 diameterstat_reset,
 189                 diameterstat_packet,
 190                 diameterstat_draw);
 191
 192         if(error_string){
 193                 simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK, "%s", error_string->str);
 194                 g_string_free(error_string, TRUE);
 195                 g_free(diameter);
 196                 return;
 197         }
 198
 199         /* Button row. */
 200         bbox = dlg_button_row_new(GTK_STOCK_CLOSE, NULL);
 201         gtk_box_pack_end(GTK_BOX(vbox), bbox, FALSE, FALSE, 0);
 202
 203         close_bt = (GtkWidget *)g_object_get_data(G_OBJECT(bbox), GTK_STOCK_CLOSE);
 204         window_set_cancel_button(diameter->win, close_bt, window_cancel_button_cb);
 205
 206         g_signal_connect(diameter->win, "delete_event", G_CALLBACK(window_delete_event_cb), NULL);
 207         g_signal_connect(diameter->win, "destroy", G_CALLBACK(win_destroy_cb), diameter);
 208
 209         gtk_widget_show_all(diameter->win);
 210         window_present(diameter->win);
 211
 212         cf_retap_packets(&cfile);
 213         gdk_window_raise(gtk_widget_get_window(diameter->win));
 214 }
 215
 216 static tap_param diameter_stat_params[] = {
 217         { PARAM_FILTER, "Filter", NULL }
 218 };
 219
 220 static tap_param_dlg diameter_stat_dlg = {
 221         "Diameter Service Response Time Statistics",
 222         "diameter",
 223         gtk_diameterstat_init,
 224         -1,
 225         G_N_ELEMENTS(diameter_stat_params),
 226         diameter_stat_params
 227 };
 228
 229 void
 230 register_tap_listener_gtkdiameterstat(void)
 231 {
 232         register_param_stat(&diameter_stat_dlg, "Diameter",
 233             REGISTER_STAT_GROUP_RESPONSE_TIME);
 234 }