/* smb_stat.c * smb_stat 2003 Ronnie Sahlberg * * $Id: smb_stat.c,v 1.9 2003/06/22 04:00:21 gerald Exp $ * * Ethereal - Network traffic analyzer * By Gerald Combs * 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ #ifdef HAVE_CONFIG_H # include "config.h" #endif #include #ifdef HAVE_SYS_TYPES_H # include #endif #include #include #include "menu.h" #include "../epan/packet_info.h" #include "../tap.h" #include "../epan/value_string.h" #include "../smb.h" #include "../register.h" #include "../timestats.h" #include "compat_macros.h" #include "../simple_dialog.h" #include "../file.h" #include "../globals.h" #include "service_response_time_table.h" /* used to keep track of the statistics for an entire program interface */ typedef struct _smbstat_t { GtkWidget *win; srt_stat_table smb_srt_table; srt_stat_table trans2_srt_table; srt_stat_table nt_trans_srt_table; } smbstat_t; static void smbstat_reset(void *pss) { smbstat_t *ss=(smbstat_t *)pss; reset_srt_table_data(&ss->smb_srt_table); reset_srt_table_data(&ss->trans2_srt_table); reset_srt_table_data(&ss->nt_trans_srt_table); } static int smbstat_packet(void *pss, packet_info *pinfo, epan_dissect_t *edt _U_, void *psi) { smbstat_t *ss=(smbstat_t *)pss; smb_info_t *si=psi; /* we are only interested in reply packets */ if(si->request){ return 0; } /* if we havnt seen the request, just ignore it */ if(!si->sip){ return 0; } add_srt_table_data(&ss->smb_srt_table, si->cmd, &si->sip->req_time, pinfo); if(si->cmd==0xA0){ smb_nt_transact_info_t *sti=(smb_nt_transact_info_t *)si->sip->extra_info; add_srt_table_data(&ss->nt_trans_srt_table, sti->subcmd, &si->sip->req_time, pinfo); } else if(si->cmd==0x32){ smb_transact2_info_t *st2i=(smb_transact2_info_t *)si->sip->extra_info; add_srt_table_data(&ss->trans2_srt_table, st2i->subcmd, &si->sip->req_time, pinfo); } return 1; } static void smbstat_draw(void *pss) { smbstat_t *ss=(smbstat_t *)pss; draw_srt_table_data(&ss->smb_srt_table); draw_srt_table_data(&ss->trans2_srt_table); draw_srt_table_data(&ss->nt_trans_srt_table); } void protect_thread_critical_region(void); void unprotect_thread_critical_region(void); static void win_destroy_cb(GtkWindow *win _U_, gpointer data) { smbstat_t *ss=(smbstat_t *)data; protect_thread_critical_region(); remove_tap_listener(ss); unprotect_thread_critical_region(); free_srt_table_data(&ss->smb_srt_table); free_srt_table_data(&ss->trans2_srt_table); free_srt_table_data(&ss->nt_trans_srt_table); g_free(ss); } static void gtk_smbstat_init(char *optarg) { smbstat_t *ss; char *filter=NULL; GtkWidget *label; char filter_string[256]; GString *error_string; int i; GtkWidget *vbox; if(!strncmp(optarg,"smb,srt,",8)){ filter=optarg+8; } else { filter=NULL; } ss=g_malloc(sizeof(smbstat_t)); ss->win=gtk_window_new(GTK_WINDOW_TOPLEVEL); gtk_window_set_default_size(GTK_WINDOW(ss->win), 550, 600); gtk_window_set_title(GTK_WINDOW(ss->win), "SMB Service Response Time statistics"); SIGNAL_CONNECT(ss->win, "destroy", win_destroy_cb, ss); vbox=gtk_vbox_new(FALSE, 0); gtk_container_add(GTK_CONTAINER(ss->win), vbox); gtk_container_set_border_width(GTK_CONTAINER(vbox), 10); gtk_widget_show(vbox); label=gtk_label_new("SMB Service Response Time statistics"); gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0); gtk_widget_show(label); snprintf(filter_string,255,"Filter:%s",filter?filter:""); label=gtk_label_new(filter_string); gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0); gtk_widget_show(label); label=gtk_label_new("SMB Commands"); gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0); gtk_widget_show(label); /* We must display TOP LEVEL Widget before calling init_srt_table() */ gtk_widget_show(ss->win); init_srt_table(&ss->smb_srt_table, 256, vbox); for(i=0;i<256;i++){ init_srt_table_row(&ss->smb_srt_table, i, val_to_str(i, smb_cmd_vals, "Unknown(0x%02x)")); } label=gtk_label_new("Transaction2 Sub-Commands"); gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0); gtk_widget_show(label); init_srt_table(&ss->trans2_srt_table, 256, vbox); for(i=0;i<256;i++){ init_srt_table_row(&ss->trans2_srt_table, i, val_to_str(i, trans2_cmd_vals, "Unknown(0x%02x)")); } label=gtk_label_new("NT Transaction Sub-Commands"); gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0); gtk_widget_show(label); init_srt_table(&ss->nt_trans_srt_table, 256, vbox); for(i=0;i<256;i++){ init_srt_table_row(&ss->nt_trans_srt_table, i, val_to_str(i, nt_cmd_vals, "Unknown(0x%02x)")); } error_string=register_tap_listener("smb", ss, filter, smbstat_reset, smbstat_packet, smbstat_draw); if(error_string){ simple_dialog(ESD_TYPE_WARN, NULL, error_string->str); g_string_free(error_string, TRUE); g_free(ss); return; } gtk_widget_show_all(ss->win); redissect_packets(&cfile); } static GtkWidget *dlg=NULL, *dlg_box; static GtkWidget *filter_box; static GtkWidget *filter_label, *filter_entry; static GtkWidget *start_button; static void dlg_destroy_cb(void) { dlg=NULL; } static void smbstat_start_button_clicked(GtkWidget *item _U_, gpointer data _U_) { char *filter; char str[256]; filter=(char *)gtk_entry_get_text(GTK_ENTRY(filter_entry)); if(filter[0]==0){ gtk_smbstat_init("smb,srt"); } else { sprintf(str,"smb,srt,%s", filter); gtk_smbstat_init(str); } } static void gtk_smbstat_cb(GtkWidget *w _U_, gpointer d _U_) { /* if the window is already open, bring it to front */ if(dlg){ gdk_window_raise(dlg->window); return; } dlg=gtk_window_new(GTK_WINDOW_TOPLEVEL); gtk_window_set_title(GTK_WINDOW(dlg), "SMB Service Response Time statistics"); SIGNAL_CONNECT(dlg, "destroy", dlg_destroy_cb, NULL); dlg_box=gtk_vbox_new(FALSE, 0); gtk_container_add(GTK_CONTAINER(dlg), dlg_box); gtk_widget_show(dlg_box); /* filter box */ filter_box=gtk_hbox_new(FALSE, 10); /* Filter label */ gtk_container_set_border_width(GTK_CONTAINER(filter_box), 10); filter_label=gtk_label_new("Filter:"); gtk_box_pack_start(GTK_BOX(filter_box), filter_label, FALSE, FALSE, 0); gtk_widget_show(filter_label); filter_entry=gtk_entry_new_with_max_length(250); gtk_box_pack_start(GTK_BOX(filter_box), filter_entry, FALSE, FALSE, 0); gtk_widget_show(filter_entry); gtk_box_pack_start(GTK_BOX(dlg_box), filter_box, TRUE, TRUE, 0); gtk_widget_show(filter_box); /* the start button */ start_button=gtk_button_new_with_label("Create Stat"); SIGNAL_CONNECT_OBJECT(start_button, "clicked", smbstat_start_button_clicked, NULL); gtk_box_pack_start(GTK_BOX(dlg_box), start_button, TRUE, TRUE, 0); gtk_widget_show(start_button); gtk_widget_show_all(dlg); } void register_tap_listener_gtksmbstat(void) { register_ethereal_tap("smb,srt", gtk_smbstat_init); } void register_tap_menu_gtksmbstat(void) { register_tap_menu_item("Service Response Time/SMB", gtk_smbstat_cb); }