ui/gtk/rtp_stream.c

   1 /* rtp_stream.c
   2  * RTP streams summary addition for Wireshark
   3  *
   4  * $Id$
   5  *
   6  * Copyright 2003, Alcatel Business Systems
   7  * By Lars Ruoff <lars.ruoff@gmx.net>
   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 <stdio.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33
  34 #ifdef HAVE_FCNTL_H
  35 #include <fcntl.h>
  36 #endif
  37
  38 #include <epan/epan.h>
  39 #include <epan/packet.h>
  40 #include <epan/tap.h>
  41 #include <epan/dissectors/packet-rtp.h>
  42 #include <epan/addr_resolv.h>
  43
  44 #include "../globals.h"
  45 #include "ui/alert_box.h"
  46 #include "ui/simple_dialog.h"
  47 #include "ui/rtp_stream.h"
  48 #include "ui/tap-rtp-common.h"
  49 #include <wsutil/file_util.h>
  50
  51 #include "ui/rtp_stream.h"
  52 #include "ui/gtk/rtp_stream_dlg.h"
  53 #include "ui/gtk/main.h"
  54
  55 /* The one and only global rtpstream_tapinfo_t structure for tshark and wireshark.
  56  */
  57 static rtpstream_tapinfo_t the_tapinfo_struct =
  58         {0, NULL, 0, TAP_ANALYSE, NULL, NULL, NULL, 0, FALSE};
  59
  60
  61 /****************************************************************************/
  62 /* redraw the output */
  63 static void rtpstream_draw(void *arg _U_)
  64 {
  65 /* XXX: see rtpstream_on_update in rtp_streams_dlg.c for comments
  66         g_signal_emit_by_name(top_level, "signal_rtpstream_update");
  67 */
  68         rtpstream_dlg_update(the_tapinfo_struct.strinfo_list);
  69         return;
  70 }
  71
  72
  73 /****************************************************************************/
  74 /* scan for RTP streams */
  75 void rtpstream_scan(void)
  76 {
  77         gboolean was_registered = the_tapinfo_struct.is_registered;
  78         if (!the_tapinfo_struct.is_registered)
  79                 register_tap_listener_rtp_stream();
  80
  81         the_tapinfo_struct.mode = TAP_ANALYSE;
  82         cf_retap_packets(&cfile);
  83
  84         if (!was_registered)
  85                 remove_tap_listener_rtp_stream();
  86 }
  87
  88
  89 /****************************************************************************/
  90 /* save rtp dump of stream_fwd */
  91 gboolean rtpstream_save(rtp_stream_info_t* stream, const gchar *filename)
  92 {
  93         gboolean was_registered = the_tapinfo_struct.is_registered;
  94         /* open file for saving */
  95         the_tapinfo_struct.save_file = ws_fopen(filename, "wb");
  96         if (the_tapinfo_struct.save_file==NULL) {
  97                 open_failure_alert_box(filename, errno, TRUE);
  98                 return FALSE;
  99         }
 100
 101         rtp_write_header(stream, the_tapinfo_struct.save_file);
 102         if (ferror(the_tapinfo_struct.save_file)) {
 103                 write_failure_alert_box(filename, errno);
 104                 fclose(the_tapinfo_struct.save_file);
 105                 return FALSE;
 106         }
 107
 108         if (!the_tapinfo_struct.is_registered)
 109                 register_tap_listener_rtp_stream();
 110
 111         the_tapinfo_struct.mode = TAP_SAVE;
 112         the_tapinfo_struct.filter_stream_fwd = stream;
 113         cf_retap_packets(&cfile);
 114         the_tapinfo_struct.mode = TAP_ANALYSE;
 115
 116         if (!was_registered)
 117                 remove_tap_listener_rtp_stream();
 118
 119         if (ferror(the_tapinfo_struct.save_file)) {
 120                 write_failure_alert_box(filename, errno);
 121                 fclose(the_tapinfo_struct.save_file);
 122                 return FALSE;
 123         }
 124
 125         if (fclose(the_tapinfo_struct.save_file) == EOF) {
 126                 write_failure_alert_box(filename, errno);
 127                 return FALSE;
 128         }
 129         return TRUE;
 130 }
 131
 132
 133 /****************************************************************************/
 134 /* mark packets in stream_fwd or stream_rev */
 135 void rtpstream_mark(rtp_stream_info_t* stream_fwd, rtp_stream_info_t* stream_rev)
 136 {
 137         gboolean was_registered = the_tapinfo_struct.is_registered;
 138         if (!the_tapinfo_struct.is_registered)
 139                 register_tap_listener_rtp_stream();
 140
 141         the_tapinfo_struct.mode = TAP_MARK;
 142         the_tapinfo_struct.filter_stream_fwd = stream_fwd;
 143         the_tapinfo_struct.filter_stream_rev = stream_rev;
 144         cf_retap_packets(&cfile);
 145         the_tapinfo_struct.mode = TAP_ANALYSE;
 146
 147         if (!was_registered)
 148                 remove_tap_listener_rtp_stream();
 149 }
 150
 151
 152 /****************************************************************************/
 153 const rtpstream_tapinfo_t* rtpstream_get_info(void)
 154 {
 155         return &the_tapinfo_struct;
 156 }
 157
 158
 159 /****************************************************************************/
 160 /* TAP INTERFACE */
 161 /****************************************************************************/
 162
 163 /****************************************************************************/
 164 void
 165 remove_tap_listener_rtp_stream(void)
 166 {
 167         if (the_tapinfo_struct.is_registered) {
 168                 remove_tap_listener(&the_tapinfo_struct);
 169
 170                 the_tapinfo_struct.is_registered = FALSE;
 171         }
 172 }
 173
 174
 175 /****************************************************************************/
 176 void
 177 register_tap_listener_rtp_stream(void)
 178 {
 179         GString *error_string;
 180
 181         if (!the_tapinfo_struct.is_registered) {
 182                 error_string = register_tap_listener("rtp", &the_tapinfo_struct,
 183                         NULL, 0, rtpstream_reset_cb, rtpstream_packet,
 184                         rtpstream_draw);
 185
 186                 if (error_string != NULL) {
 187                         simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK,
 188                                       "%s", error_string->str);
 189                         g_string_free(error_string, TRUE);
 190                         exit(1);
 191                 }
 192
 193                 the_tapinfo_struct.is_registered = TRUE;
 194         }
 195 }