osmith
/
wireshark
forked from osmocom/wireshark
1
0
Fork 0
Code Pull Requests Releases Wiki Activity
wireshark/ui/gtk/lbm_uimflow_dlg.c

435 lines
15 KiB
C
Raw Blame History

/* lbm_uimflow_dlg.c
* Routines for LBMC UIM flow graph
*
* Copyright (c) 2005-2014 Informatica Corporation. All Rights Reserved.
*
* 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 <epan/to_str.h>
#include <epan/packet.h>
#include <epan/tap.h>
#include <gtk/gtk.h>
#include "ui/gtk/gui_utils.h"
#include "ui/gtk/dlg_utils.h"
#include <globals.h>
#include "ui/gtk/graph_analysis.h"
#include <epan/dissectors/packet-lbm.h>
#include "lbm_uimflow_dlg.h"
typedef enum
{
select_all_packets,
select_displayed_packets
} lbmc_uim_flow_select_t;
typedef struct
{
gboolean have_tap_listener;
int tap_identifier;
seq_analysis_info_t * graph_analysis;
graph_analysis_data_t * graph_analysis_data;
GtkWidget * flow_graph_dialog;
lbmc_uim_flow_select_t packet_select_type;
GtkWidget * select_all_radio_button;
GtkWidget * select_displayed_radio_button;
} lbm_uimflow_dialog_t;
static lbm_uimflow_dialog_t dialog_data = { FALSE, -1, NULL, NULL, NULL, select_displayed_packets, NULL, NULL };
static void lbmc_uim_flow_graph_data_init(void)
{
dialog_data.graph_analysis = sequence_analysis_info_new();
dialog_data.graph_analysis->type = SEQ_ANALYSIS_ANY;
dialog_data.graph_analysis->all_packets = TRUE;
dialog_data.graph_analysis->any_addr = TRUE;
}
static void lbmc_uim_flow_toggle_select_all_cb(GtkWidget * widget _U_, gpointer user_data _U_)
{
/* is the button now active? */
if (gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(dialog_data.select_all_radio_button)))
{
dialog_data.packet_select_type = select_all_packets;
}
}
static void lbmc_uim_flow_toggle_select_displayed_cb(GtkWidget * widget _U_, gpointer user_data _U_)
{
/* is the button now active? */
if (gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(dialog_data.select_displayed_radio_button)))
{
dialog_data.packet_select_type = select_displayed_packets;
}
}
static void lbmc_uim_flow_tap_reset(void * tap_data _U_)
{
if (dialog_data.graph_analysis != NULL)
{
/* free the graph data items */
sequence_analysis_list_free(dialog_data.graph_analysis);
}
}
static int lbmc_uim_flow_graph_add_to_graph(packet_info * pinfo, const lbm_uim_stream_info_t * stream_info)
{
lbm_uim_stream_endpoint_t epa;
lbm_uim_stream_endpoint_t epb;
seq_analysis_item_t * item;
gchar * ctxinst1;
gchar * ctxinst2;
gboolean swap_endpoints = FALSE;
int rc;
if (stream_info->endpoint_a.type != stream_info->endpoint_b.type)
{
return (1);
}
if (stream_info->endpoint_a.type == lbm_uim_instance_stream)
{
rc = memcmp((void *)stream_info->endpoint_a.stream_info.ctxinst.ctxinst,
(void *)stream_info->endpoint_b.stream_info.ctxinst.ctxinst,
LBM_CONTEXT_INSTANCE_BLOCK_SZ);
if (rc <= 0)
{
swap_endpoints = FALSE;
}
else
{
swap_endpoints = TRUE;
}
}
else
{
if (stream_info->endpoint_a.stream_info.dest.domain < stream_info->endpoint_b.stream_info.dest.domain)
{
swap_endpoints = FALSE;
}
else if (stream_info->endpoint_a.stream_info.dest.domain > stream_info->endpoint_b.stream_info.dest.domain)
{
swap_endpoints = TRUE;
}
else
{
int compare;
compare = cmp_address(&(stream_info->endpoint_a.stream_info.dest.addr), &(stream_info->endpoint_b.stream_info.dest.addr));
if (compare < 0)
{
swap_endpoints = FALSE;
}
else if (compare > 0)
{
swap_endpoints = TRUE;
}
else
{
if (stream_info->endpoint_a.stream_info.dest.port <= stream_info->endpoint_b.stream_info.dest.port)
{
swap_endpoints = FALSE;
}
else
{
swap_endpoints = TRUE;
}
}
}
}
if (swap_endpoints == FALSE)
{
epa = stream_info->endpoint_a;
epb = stream_info->endpoint_b;
}
else
{
epb = stream_info->endpoint_a;
epa = stream_info->endpoint_b;
}
item = (seq_analysis_item_t *)g_malloc0(sizeof(seq_analysis_item_t));
copy_address(&(item->src_addr), &(pinfo->src));
copy_address(&(item->dst_addr), &(pinfo->dst));
item->frame_number = pinfo->num;
item->port_src = pinfo->srcport;
item->port_dst = pinfo->destport;
item->protocol = g_strdup(port_type_to_str(pinfo->ptype));
if (stream_info->description == NULL)
{
item->frame_label = g_strdup_printf("(%" G_GUINT32_FORMAT ")", stream_info->sqn);
}
else
{
item->frame_label = g_strdup_printf("%s (%" G_GUINT32_FORMAT ")", stream_info->description, stream_info->sqn);
}
if (epa.type == lbm_uim_instance_stream)
{
ctxinst1 = bytes_to_str(pinfo->pool, epa.stream_info.ctxinst.ctxinst, sizeof(epa.stream_info.ctxinst.ctxinst));
ctxinst2 = bytes_to_str(pinfo->pool, epb.stream_info.ctxinst.ctxinst, sizeof(epb.stream_info.ctxinst.ctxinst));
item->comment = g_strdup_printf("%s <-> %s [%" G_GUINT64_FORMAT "]",
ctxinst1,
ctxinst2,
stream_info->channel);
}
else
{
item->comment = g_strdup_printf("%" G_GUINT32_FORMAT ":%s:%" G_GUINT16_FORMAT " <-> %" G_GUINT32_FORMAT ":%s:%" G_GUINT16_FORMAT " [%" G_GUINT64_FORMAT "]",
epa.stream_info.dest.domain,
address_to_str(pinfo->pool, &(epa.stream_info.dest.addr)),
epa.stream_info.dest.port,
epb.stream_info.dest.domain,
address_to_str(pinfo->pool, &(epb.stream_info.dest.addr)),
epb.stream_info.dest.port,
stream_info->channel);
}
item->conv_num = (guint16)LBM_CHANNEL_ID(stream_info->channel);
item->display = TRUE;
item->line_style = 1;
g_queue_push_tail(dialog_data.graph_analysis->items, item);
return (1);
}
static gboolean lbmc_uim_flow_tap_packet(void * tap_data _U_, packet_info * pinfo, epan_dissect_t * edt _U_, const void * stream_info)
{
const lbm_uim_stream_info_t * info = (const lbm_uim_stream_info_t *)stream_info;
if ((dialog_data.packet_select_type == select_all_packets) || (pinfo->fd->flags.passed_dfilter == 1))
{
lbmc_uim_flow_graph_add_to_graph(pinfo, info);
return (TRUE);
}
return (FALSE);
}
static void lbmc_uim_flow_tap_draw(void * tap_data _U_)
{
return;
}
static void lbmc_uim_flow_remove_tap_listener(void)
{
remove_tap_listener(&(dialog_data.tap_identifier));
dialog_data.have_tap_listener = FALSE;
}
static void lbmc_uim_flow_graph_on_ok_cb(GtkButton * button _U_, gpointer user_data)
{
GList * list = NULL;
gchar time_str[COL_MAX_LEN];
if (dialog_data.have_tap_listener == TRUE)
{
/* remove_tap_listeners */
lbmc_uim_flow_remove_tap_listener();
}
/* Scan for displayed packets (retap all packets) */
if (dialog_data.have_tap_listener == FALSE)
{
gchar * err_msg;
err_msg = register_tap_listener("lbm_uim",
&(dialog_data.tap_identifier),
NULL,
TL_REQUIRES_COLUMNS,
lbmc_uim_flow_tap_reset,
lbmc_uim_flow_tap_packet,
lbmc_uim_flow_tap_draw);
if (err_msg != NULL)
{
fprintf(stderr, "register_tap_listener: %s\n", err_msg);
wmem_free(NULL, err_msg);
}
dialog_data.have_tap_listener = TRUE;
}
cf_retap_packets(&cfile);
/* Fill in the timestamps. */
list = g_queue_peek_nth_link(dialog_data.graph_analysis->items, 0);
while (list != NULL)
{
seq_analysis_item_t * seq_item = (seq_analysis_item_t *)list->data;
set_fd_time(cfile.epan, frame_data_sequence_find(cfile.frames, seq_item->frame_number), time_str);
seq_item->time_str = g_strdup(time_str);
list = g_list_next(list);
}
if (dialog_data.graph_analysis_data->dlg.window != NULL)
{
graph_analysis_update(dialog_data.graph_analysis_data);
}
else
{
dialog_data.graph_analysis_data->dlg.parent_w = (GtkWidget *)user_data;
graph_analysis_create(dialog_data.graph_analysis_data);
}
}
static void lbmc_uim_flow_graph_on_destroy_cb(GtkWidget * widget _U_, gpointer user_data _U_)
{
/* remove_tap_listeners */
lbmc_uim_flow_remove_tap_listener();
/* Clean up memory used by tap */
lbmc_uim_flow_tap_reset(NULL);
g_assert(dialog_data.graph_analysis != NULL);
g_assert(dialog_data.graph_analysis_data != NULL);
sequence_analysis_info_free(dialog_data.graph_analysis);
dialog_data.graph_analysis = NULL;
g_free(dialog_data.graph_analysis_data);
dialog_data.graph_analysis_data = NULL;
/* Note that we no longer have a "Flow Graph" dialog box. */
dialog_data.flow_graph_dialog = NULL;
}
static void lbmc_uim_flow_graph_dlg_create(void)
{
GtkWidget * flow_graph_dlg_w = NULL;
GtkWidget * main_vb = NULL;
GtkWidget * hbuttonbox = NULL;
GtkWidget * bt_cancel = NULL;
GtkWidget * bt_ok = NULL;
GtkWidget * range_fr = NULL;
GtkWidget * range_grid = NULL;
flow_graph_dlg_w = dlg_window_new("Wireshark: UIM Flow Graph");
gtk_window_set_destroy_with_parent(GTK_WINDOW(flow_graph_dlg_w), TRUE);
gtk_window_set_default_size(GTK_WINDOW(flow_graph_dlg_w), 250, 150);
main_vb = ws_gtk_box_new(GTK_ORIENTATION_VERTICAL, 0, FALSE);
gtk_container_add(GTK_CONTAINER(flow_graph_dlg_w), main_vb);
gtk_container_set_border_width(GTK_CONTAINER(main_vb), 7);
gtk_widget_show(flow_graph_dlg_w);
/*** Packet range frame ***/
range_fr = gtk_frame_new("Choose packets");
gtk_box_pack_start(GTK_BOX(main_vb), range_fr, FALSE, FALSE, 5);
range_grid = ws_gtk_grid_new();
gtk_container_set_border_width(GTK_CONTAINER(range_grid), 5);
gtk_container_add(GTK_CONTAINER(range_fr), range_grid);
/* Process all packets */
dialog_data.select_all_radio_button = gtk_radio_button_new_with_mnemonic_from_widget(NULL, "_All packets");
gtk_widget_set_tooltip_text(dialog_data.select_all_radio_button, ("Process all packets"));
g_signal_connect(dialog_data.select_all_radio_button, "toggled", G_CALLBACK(lbmc_uim_flow_toggle_select_all_cb), NULL);
ws_gtk_grid_attach_extended(GTK_GRID(range_grid), dialog_data.select_all_radio_button, 0, 0, 1, 1,
(GtkAttachOptions)(GTK_FILL), (GtkAttachOptions)(0), 0, 0);
if (dialog_data.packet_select_type == select_all_packets)
{
gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(dialog_data.select_all_radio_button), TRUE);
}
gtk_widget_show(dialog_data.select_all_radio_button);
/* Process displayed packets */
dialog_data.select_displayed_radio_button =
gtk_radio_button_new_with_mnemonic_from_widget(GTK_RADIO_BUTTON(dialog_data.select_all_radio_button), "_Displayed packets");
gtk_widget_set_tooltip_text(dialog_data.select_displayed_radio_button, ("Process displayed packets"));
g_signal_connect(dialog_data.select_displayed_radio_button, "toggled", G_CALLBACK(lbmc_uim_flow_toggle_select_displayed_cb), NULL);
ws_gtk_grid_attach_extended(GTK_GRID(range_grid), dialog_data.select_displayed_radio_button, 0, 1, 1, 1,
(GtkAttachOptions)(GTK_FILL), (GtkAttachOptions)(0), 0, 0);
if (dialog_data.packet_select_type == select_displayed_packets)
{
gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(dialog_data.select_displayed_radio_button), TRUE);
}
gtk_widget_show(dialog_data.select_displayed_radio_button);
gtk_widget_show(range_grid);
gtk_widget_show(range_fr);
/* button row */
hbuttonbox = gtk_button_box_new(GTK_ORIENTATION_HORIZONTAL);
gtk_box_pack_start(GTK_BOX(main_vb), hbuttonbox, FALSE, FALSE, 5);
gtk_button_box_set_layout(GTK_BUTTON_BOX(hbuttonbox), GTK_BUTTONBOX_SPREAD);
gtk_box_set_spacing(GTK_BOX(hbuttonbox), 30);
bt_ok = gtk_button_new_from_stock(GTK_STOCK_OK);
gtk_container_add(GTK_CONTAINER(hbuttonbox), bt_ok);
gtk_widget_set_tooltip_text(bt_ok, "Show the flow graph");
g_signal_connect(bt_ok, "clicked", G_CALLBACK(lbmc_uim_flow_graph_on_ok_cb), flow_graph_dlg_w);
gtk_widget_show(bt_ok);
bt_cancel = gtk_button_new_from_stock(GTK_STOCK_CANCEL);
gtk_container_add(GTK_CONTAINER(hbuttonbox), bt_cancel);
#if GTK_CHECK_VERSION(2,18,0)
gtk_widget_set_can_default(bt_cancel, TRUE);
#else
GTK_WIDGET_SET_FLAGS(bt_cancel, GTK_CAN_DEFAULT);
#endif
gtk_widget_set_tooltip_text(bt_cancel, "Cancel this dialog");
window_set_cancel_button(flow_graph_dlg_w, bt_cancel, window_cancel_button_cb);
g_signal_connect(flow_graph_dlg_w, "delete_event", G_CALLBACK(window_delete_event_cb), NULL);
g_signal_connect(flow_graph_dlg_w, "destroy", G_CALLBACK(lbmc_uim_flow_graph_on_destroy_cb), NULL);
gtk_widget_show_all(flow_graph_dlg_w);
window_present(flow_graph_dlg_w);
dialog_data.flow_graph_dialog = flow_graph_dlg_w;
}
static void lbmc_uim_flow_graph_init_tap(const char * dummy _U_, void * user_data _U_)
{
/*
The storage allocated by lbmc_uim_flow_graph_data_init() and lbmc_uim_graph_analysis_init()
will be considered to be "associated with" the dialog_data.flow_graph_dialog dialog box. It will be freed
when the dialog_data.flow_graph_dialog dialog box is destroyed.
*/
if (dialog_data.flow_graph_dialog != NULL)
{
g_assert(dialog_data.graph_analysis != NULL);
g_assert(dialog_data.graph_analysis_data != NULL);
/* There's already a dialog box; reactivate it. */
reactivate_window(dialog_data.flow_graph_dialog);
}
else
{
g_assert(dialog_data.graph_analysis == NULL);
g_assert(dialog_data.graph_analysis_data == NULL);
/* initialize graph items store */
lbmc_uim_flow_graph_data_init();
/* init the Graph Analysis */
dialog_data.graph_analysis_data = graph_analysis_init(dialog_data.graph_analysis);
lbmc_uim_flow_graph_dlg_create();
}
}
void lbmc_uim_flow_menu_cb(gpointer arg _U_)
{
lbmc_uim_flow_graph_init_tap("", NULL);
}
/*
* Editor modelines - http://www.wireshark.org/tools/modelines.html
*
* Local variables:
* c-basic-offset: 4
* tab-width: 8
* indent-tabs-mode: nil
* End:
*
* vi: set shiftwidth=4 tabstop=8 expandtab:
* :indentSize=4:tabSize=8:noTabs=true:
*/
View Git Blame Copy Permalink