![Guy Harris](/assets/img/avatar_default.png)
the ui directory. (Perhaps some other files that would be used by all flavors of Wireshark, for any GUI toolkit or for someting such as ncurses, and not for any command-line tool such as TShark, should be moved there as well.) Shuffle some #includes to put the "ui/XXX.h" includes together. svn path=/trunk/; revision=40529
203 lines
5.5 KiB
C
203 lines
5.5 KiB
C
/* rtp_stream.c
|
|
* RTP streams summary addition for Wireshark
|
|
*
|
|
* $Id$
|
|
*
|
|
* Copyright 2003, Alcatel Business Systems
|
|
* By Lars Ruoff <lars.ruoff@gmx.net>
|
|
*
|
|
* 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#ifdef HAVE_FCNTL_H
|
|
#include <fcntl.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
# include <sys/types.h>
|
|
#endif
|
|
|
|
|
|
#include <epan/epan.h>
|
|
#include <epan/packet.h>
|
|
#include <epan/tap.h>
|
|
#include <epan/dissectors/packet-rtp.h>
|
|
#include <epan/addr_resolv.h>
|
|
|
|
#include "../globals.h"
|
|
#include "ui/alert_box.h"
|
|
#include "ui/simple_dialog.h"
|
|
#include "../tap-rtp-common.h"
|
|
#include <wsutil/file_util.h>
|
|
|
|
#include "ui/gtk/rtp_stream.h"
|
|
#include "ui/gtk/rtp_stream_dlg.h"
|
|
#include "ui/gtk/main.h"
|
|
|
|
/* The one and only global rtpstream_tapinfo_t structure for tshark and wireshark.
|
|
*/
|
|
static rtpstream_tapinfo_t the_tapinfo_struct =
|
|
{0, NULL, 0, TAP_ANALYSE, NULL, NULL, NULL, 0, FALSE};
|
|
|
|
|
|
/****************************************************************************/
|
|
/* redraw the output */
|
|
static void rtpstream_draw(void *arg _U_)
|
|
{
|
|
/* XXX: see rtpstream_on_update in rtp_streams_dlg.c for comments
|
|
g_signal_emit_by_name(top_level, "signal_rtpstream_update");
|
|
*/
|
|
rtpstream_dlg_update(the_tapinfo_struct.strinfo_list);
|
|
return;
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
/* scan for RTP streams */
|
|
void rtpstream_scan(void)
|
|
{
|
|
gboolean was_registered = the_tapinfo_struct.is_registered;
|
|
if (!the_tapinfo_struct.is_registered)
|
|
register_tap_listener_rtp_stream();
|
|
|
|
the_tapinfo_struct.mode = TAP_ANALYSE;
|
|
cf_retap_packets(&cfile);
|
|
|
|
if (!was_registered)
|
|
remove_tap_listener_rtp_stream();
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
/* save rtp dump of stream_fwd */
|
|
gboolean rtpstream_save(rtp_stream_info_t* stream, const gchar *filename)
|
|
{
|
|
gboolean was_registered = the_tapinfo_struct.is_registered;
|
|
/* open file for saving */
|
|
the_tapinfo_struct.save_file = ws_fopen(filename, "wb");
|
|
if (the_tapinfo_struct.save_file==NULL) {
|
|
open_failure_alert_box(filename, errno, TRUE);
|
|
return FALSE;
|
|
}
|
|
|
|
rtp_write_header(stream, the_tapinfo_struct.save_file);
|
|
if (ferror(the_tapinfo_struct.save_file)) {
|
|
write_failure_alert_box(filename, errno);
|
|
fclose(the_tapinfo_struct.save_file);
|
|
return FALSE;
|
|
}
|
|
|
|
if (!the_tapinfo_struct.is_registered)
|
|
register_tap_listener_rtp_stream();
|
|
|
|
the_tapinfo_struct.mode = TAP_SAVE;
|
|
the_tapinfo_struct.filter_stream_fwd = stream;
|
|
cf_retap_packets(&cfile);
|
|
the_tapinfo_struct.mode = TAP_ANALYSE;
|
|
|
|
if (!was_registered)
|
|
remove_tap_listener_rtp_stream();
|
|
|
|
if (ferror(the_tapinfo_struct.save_file)) {
|
|
write_failure_alert_box(filename, errno);
|
|
fclose(the_tapinfo_struct.save_file);
|
|
return FALSE;
|
|
}
|
|
|
|
if (fclose(the_tapinfo_struct.save_file) == EOF) {
|
|
write_failure_alert_box(filename, errno);
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
/* mark packets in stream_fwd or stream_rev */
|
|
void rtpstream_mark(rtp_stream_info_t* stream_fwd, rtp_stream_info_t* stream_rev)
|
|
{
|
|
gboolean was_registered = the_tapinfo_struct.is_registered;
|
|
if (!the_tapinfo_struct.is_registered)
|
|
register_tap_listener_rtp_stream();
|
|
|
|
the_tapinfo_struct.mode = TAP_MARK;
|
|
the_tapinfo_struct.filter_stream_fwd = stream_fwd;
|
|
the_tapinfo_struct.filter_stream_rev = stream_rev;
|
|
cf_retap_packets(&cfile);
|
|
the_tapinfo_struct.mode = TAP_ANALYSE;
|
|
|
|
if (!was_registered)
|
|
remove_tap_listener_rtp_stream();
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
const rtpstream_tapinfo_t* rtpstream_get_info(void)
|
|
{
|
|
return &the_tapinfo_struct;
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
/* TAP INTERFACE */
|
|
/****************************************************************************/
|
|
|
|
/****************************************************************************/
|
|
void
|
|
remove_tap_listener_rtp_stream(void)
|
|
{
|
|
if (the_tapinfo_struct.is_registered) {
|
|
protect_thread_critical_region();
|
|
remove_tap_listener(&the_tapinfo_struct);
|
|
unprotect_thread_critical_region();
|
|
|
|
the_tapinfo_struct.is_registered = FALSE;
|
|
}
|
|
}
|
|
|
|
|
|
/****************************************************************************/
|
|
void
|
|
register_tap_listener_rtp_stream(void)
|
|
{
|
|
GString *error_string;
|
|
|
|
if (!the_tapinfo_struct.is_registered) {
|
|
error_string = register_tap_listener("rtp", &the_tapinfo_struct,
|
|
NULL, 0, rtpstream_reset_cb, rtpstream_packet,
|
|
rtpstream_draw);
|
|
|
|
if (error_string != NULL) {
|
|
simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
|
|
"%s", error_string->str);
|
|
g_string_free(error_string, TRUE);
|
|
exit(1);
|
|
}
|
|
|
|
the_tapinfo_struct.is_registered = TRUE;
|
|
}
|
|
}
|