bugfix to a bug reported by Ian Schorr:
[obnox/wireshark/wip.git] / file.c
diff --git a/file.c b/file.c
index 2877876f9275661d7a66fdb377ea2ee1a3f995ce..24f46b810061462e73c2090c37bdfa3c3e5bd84f 100644 (file)
--- a/file.c
+++ b/file.c
@@ -1,7 +1,7 @@
 /* file.c
  * File I/O routines
  *
- * $Id: file.c,v 1.366 2004/02/22 22:33:59 guy Exp $
+ * $Id: file.c,v 1.389 2004/07/08 11:07:29 guy Exp $
  *
  * Ethereal - Network traffic analyzer
  * By Gerald Combs <gerald@ethereal.com>
  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  */
 
+/* With MSVC and a libethereal.dll this file needs to import some variables 
+   in a special way. Therefore _NEED_VAR_IMPORT_ is defined. */
+#define _NEED_VAR_IMPORT_
+
 #ifdef HAVE_CONFIG_H
 # include "config.h"
 #endif
 #include "tap_dfilter_dlg.h"
 #include "packet-data.h"
 
+/* Win32 needs the O_BINARY flag for open() */
+#ifndef O_BINARY
+#define O_BINARY       0
+#endif
+
 #ifdef HAVE_LIBPCAP
 gboolean auto_scroll_live;
 #endif
@@ -131,15 +140,6 @@ static   gboolean copy_binary_file(char *from_filename, char *to_filename);
    XXX - is this the right number? */
 #define        FRAME_DATA_CHUNK_SIZE   1024
 
-typedef struct {
-       int             level;
-       FILE            *fh;
-       GSList          *src_list;
-       gboolean        print_all_levels;
-       gboolean        print_hex_for_data;
-       char_enc        encoding;
-       gint            format;         /* text or PostScript */
-} print_data;
 
 int
 cf_open(char *fname, gboolean is_tempfile, capture_file *cf)
@@ -288,6 +288,8 @@ cf_close(capture_file *cf)
   set_menus_for_capture_in_progress(FALSE);
   set_menus_for_selected_tree_row(cf);
 
+  reset_tap_listeners();
+
   /* We have no file open. */
   cf->state = FILE_CLOSED;
 }
@@ -381,10 +383,6 @@ cf_read(capture_file *cf)
      bump that value by this amount. */
   progbar_quantum = cf->f_len/N_PROGBAR_UPDATES;
 
-#ifndef O_BINARY
-#define O_BINARY       0
-#endif
-
   packet_list_freeze();
 
   stop_flag = FALSE;
@@ -490,7 +488,10 @@ cf_read(capture_file *cf)
     switch (err) {
 
     case WTAP_ERR_UNSUPPORTED_ENCAP:
-      errmsg = "The capture file is for a network type that Ethereal doesn't support.";
+      snprintf(errmsg_errno, sizeof(errmsg_errno),
+               "The capture file has a packet with a network type that Ethereal doesn't support.\n(%s)",
+               err_info);
+      errmsg = errmsg_errno;
       break;
 
     case WTAP_ERR_CANT_READ:
@@ -529,6 +530,7 @@ int
 cf_start_tail(char *fname, gboolean is_tempfile, capture_file *cf)
 {
   int     err;
+  gchar *capture_msg;
 
   err = cf_open(fname, is_tempfile, cf);
   if (err == 0) {
@@ -540,7 +542,11 @@ cf_start_tail(char *fname, gboolean is_tempfile, capture_file *cf)
        packets (yes, I know, we don't have any *yet*). */
     set_menus_for_captured_packets(TRUE);
 
-    statusbar_push_file_msg(" <live capture in progress>");
+    capture_msg = g_strdup_printf(" %s: <live capture in progress>", cf->iface);
+
+    statusbar_push_file_msg(capture_msg);
+
+    g_free(capture_msg);
   }
   return err;
 }
@@ -960,17 +966,16 @@ read_packet(capture_file *cf, long offset)
   }
 }
 
-int
-filter_packets(capture_file *cf, gchar *dftext)
+gboolean
+filter_packets(capture_file *cf, gchar *dftext, gboolean force)
 {
   dfilter_t *dfcode;
   char      *filter_new = dftext ? dftext : "";
   char      *filter_old = cf->dfilter ? cf->dfilter : "";
 
-
-  /* if new filter equals old one, do nothing */
-  if (strcmp(filter_new, filter_old) == 0) {
-    return 1;
+  /* if new filter equals old one, do nothing unless told to do so */
+  if (!force && strcmp(filter_new, filter_old) == 0) {
+    return TRUE;
   }
 
   if (dftext == NULL) {
@@ -984,15 +989,20 @@ filter_packets(capture_file *cf, gchar *dftext)
     dftext = g_strdup(dftext);
     if (!dfilter_compile(dftext, &dfcode)) {
       /* The attempt failed; report an error. */
+      gchar *safe_dftext = simple_dialog_format_message(dftext);
+      gchar *safe_dfilter_error_msg = simple_dialog_format_message(
+         dfilter_error_msg);
       simple_dialog(ESD_TYPE_ERROR, ESD_BTN_OK, 
           "%s%s%s\n"
           "\n"
-          "The display filter \"%s\" is not a valid display filter.\n"
+          "The following display filter is not a valid display filter:\n%s\n"
           "See the help for a description of the display filter syntax.",
-          simple_dialog_primary_start(), dfilter_error_msg,
-          simple_dialog_primary_end(), dftext);
+          simple_dialog_primary_start(), safe_dfilter_error_msg,
+          simple_dialog_primary_end(), safe_dftext);
+      g_free(safe_dfilter_error_msg);
+      g_free(safe_dftext);
       g_free(dftext);
-      return 0;
+      return FALSE;
     }
 
     /* Was it empty? */
@@ -1018,7 +1028,7 @@ filter_packets(capture_file *cf, gchar *dftext)
   } else {
     rescan_packets(cf, "Filtering", dftext, TRUE, FALSE);
   }
-  return 1;
+  return TRUE;
 }
 
 void
@@ -1415,6 +1425,7 @@ process_specified_packets(capture_file *cf, packet_range_t *range,
       ret = PSP_FAILED;
       break;
     }
+    /* Process the packet */
     if (!callback(cf, fdata, &pseudo_header, pd, callback_args)) {
       /* Callback failed.  We assume it reported the error appropriately. */
       ret = PSP_FAILED;
@@ -1484,6 +1495,10 @@ retap_packets(capture_file *cf)
 typedef struct {
   print_args_t *print_args;
   FILE         *print_fh;
+  gboolean      print_header_line;
+  char         *header_line_buf;
+  int           header_line_buf_len;
+  gboolean      print_formfeed;
   gboolean      print_separator;
   char         *line_buf;
   int           line_buf_len;
@@ -1502,13 +1517,36 @@ print_packet(capture_file *cf, frame_data *fdata,
   int             line_len;
   int             column_len;
   int             cp_off;
+  gboolean        proto_tree_needed;
+
+  /* Create the protocol tree, and make it visible, if we're printing
+     the dissection or the hex data.
+     XXX - do we need it if we're just printing the hex data? */
+  proto_tree_needed = 
+      args->print_args->print_dissections != print_dissections_none || args->print_args->print_hex;
+  edt = epan_dissect_new(proto_tree_needed, proto_tree_needed);
 
+  /* Fill in the column information if we're printing the summary
+     information. */
   if (args->print_args->print_summary) {
-    /* Fill in the column information, but don't bother creating
-       the logical protocol tree. */
-    edt = epan_dissect_new(FALSE, FALSE);
     epan_dissect_run(edt, pseudo_header, pd, fdata, &cf->cinfo);
     epan_dissect_fill_in_columns(edt);
+  } else
+    epan_dissect_run(edt, pseudo_header, pd, fdata, NULL);
+
+  if (args->print_formfeed) {
+    print_formfeed(args->print_fh, args->print_args->format);
+  } else {
+      if (args->print_separator)
+        print_line(args->print_fh, 0, args->print_args->format, "");
+  }
+
+  if (args->print_args->print_summary) {
+    if (args->print_header_line) {
+      print_line(args->print_fh, 0, args->print_args->format,
+                 args->header_line_buf);
+      args->print_header_line = FALSE; /* we might not need to print any more */
+    }
     cp = &args->line_buf[0];
     line_len = 0;
     for (i = 0; i < cf->cinfo.num_cols; i++) {
@@ -1537,30 +1575,46 @@ print_packet(capture_file *cf, frame_data *fdata,
         *cp++ = ' ';
     }
     *cp = '\0';
+
+    print_packet_header(args->print_fh, args->print_args->format, fdata->num, args->line_buf);
+
     print_line(args->print_fh, 0, args->print_args->format, args->line_buf);
-  } else {
-    if (args->print_separator)
+  } /* if (print_summary) */
+  
+  if (args->print_args->print_dissections != print_dissections_none) {
+    if (args->print_args->print_summary) {
+      /* Separate the summary line from the tree with a blank line. */
       print_line(args->print_fh, 0, args->print_args->format, "");
-
-    /* Create the logical protocol tree, complete with the display
-       representation of the items; we don't need the columns here,
-       however. */
-    edt = epan_dissect_new(TRUE, TRUE);
-    epan_dissect_run(edt, pseudo_header, pd, fdata, NULL);
+    }
 
     /* Print the information in that tree. */
     proto_tree_print(args->print_args, edt, args->print_fh);
 
-    if (args->print_args->print_hex) {
-      /* Print the full packet data as hex. */
-      print_hex_data(args->print_fh, args->print_args->format, edt);
-    }
+    /* Print a blank line if we print anything after this (aka more than one packet). */
+    args->print_separator = TRUE;
+
+    /* Print a header line if we print any more packet summaries */
+    args->print_header_line = TRUE;
+  }
+
+  if (args->print_args->print_hex) {
+    /* Print the full packet data as hex. */
+    print_hex_data(args->print_fh, args->print_args->format, edt);
 
-    /* Print a blank line if we print anything after this. */
+    /* Print a blank line if we print anything after this (aka more than one packet). */
     args->print_separator = TRUE;
+
+    /* Print a header line if we print any more packet summaries */
+    args->print_header_line = TRUE;
   } /* if (print_summary) */
+
   epan_dissect_free(edt);
 
+  /* do we want to have a formfeed between each packet from now on? */
+  if(args->print_args->print_formfeed) {
+    args->print_formfeed = TRUE;
+  }
+
   return !ferror(args->print_fh);
 }
 
@@ -1576,32 +1630,41 @@ print_packets(capture_file *cf, print_args_t *print_args)
   int         line_len;
   psp_return_t ret;
 
-  callback_args.print_fh = open_print_dest(print_args->to_file,
-                                           print_args->dest);
+  if(print_args->to_file) {
+      callback_args.print_fh = open_print_dest(print_args->to_file,
+                                               print_args->file);
+  } else {
+      callback_args.print_fh = open_print_dest(print_args->to_file,
+                                               print_args->cmd);
+  }
   if (callback_args.print_fh == NULL)
     return PP_OPEN_ERROR;      /* attempt to open destination failed */
 
-  print_preamble(callback_args.print_fh, print_args->format);
+  print_preamble(callback_args.print_fh, print_args->format, cf->filename);
   if (ferror(callback_args.print_fh)) {
     close_print_dest(print_args->to_file, callback_args.print_fh);
     return PP_WRITE_ERROR;
   }
 
   callback_args.print_args = print_args;
+  callback_args.print_header_line = TRUE;
+  callback_args.header_line_buf = NULL;
+  callback_args.header_line_buf_len = 256;
+  callback_args.print_formfeed = FALSE;
   callback_args.print_separator = FALSE;
   callback_args.line_buf = NULL;
   callback_args.line_buf_len = 256;
   callback_args.col_widths = NULL;
   if (print_args->print_summary) {
-    /* We're printing packet summaries.  Allocate the line buffer at
-       its initial length. */
-    callback_args.line_buf = g_malloc(callback_args.line_buf_len + 1);
+    /* We're printing packet summaries.  Allocate the header line buffer
+       and get the column widths. */
+    callback_args.header_line_buf = g_malloc(callback_args.header_line_buf_len + 1);
 
     /* Find the widths for each of the columns - maximum of the
-       width of the title and the width of the data - and print
-       the column titles. */
+       width of the title and the width of the data - and construct
+       a buffer with a line containing the column titles. */
     callback_args.col_widths = (gint *) g_malloc(sizeof(gint) * cf->cinfo.num_cols);
-    cp = &callback_args.line_buf[0];
+    cp = &callback_args.header_line_buf[0];
     line_len = 0;
     for (i = 0; i < cf->cinfo.num_cols; i++) {
       /* Don't pad the last column. */
@@ -1622,26 +1685,29 @@ print_packets(capture_file *cf, print_args_t *print_args)
       /* Make sure there's room in the line buffer for the column; if not,
          double its length. */
       line_len += column_len + 1;      /* "+1" for space */
-      if (line_len > callback_args.line_buf_len) {
-        cp_off = cp - callback_args.line_buf;
-        callback_args.line_buf_len = 2 * line_len;
-        callback_args.line_buf = g_realloc(callback_args.line_buf,
-                                           callback_args.line_buf_len + 1);
-        cp = callback_args.line_buf + cp_off;
+      if (line_len > callback_args.header_line_buf_len) {
+        cp_off = cp - callback_args.header_line_buf;
+        callback_args.header_line_buf_len = 2 * line_len;
+        callback_args.header_line_buf = g_realloc(callback_args.header_line_buf,
+                                                  callback_args.header_line_buf_len + 1);
+        cp = callback_args.header_line_buf + cp_off;
       }
 
       /* Right-justify the packet number column. */
-      if (cf->cinfo.col_fmt[i] == COL_NUMBER)
+/*      if (cf->cinfo.col_fmt[i] == COL_NUMBER)
         sprintf(cp, "%*s", callback_args.col_widths[i], cf->cinfo.col_title[i]);
-      else
+      else*/
         sprintf(cp, "%-*s", callback_args.col_widths[i], cf->cinfo.col_title[i]);
       cp += column_len;
       if (i != cf->cinfo.num_cols - 1)
         *cp++ = ' ';
     }
     *cp = '\0';
-    print_line(callback_args.print_fh, 0, print_args->format,
-               callback_args.line_buf);
+
+    /* Now start out the main line buffer with the same length as the
+       header line buffer. */
+    callback_args.line_buf_len = callback_args.header_line_buf_len;
+    callback_args.line_buf = g_malloc(callback_args.line_buf_len + 1);
   } /* if (print_summary) */
 
   /* Iterate through the list of packets, printing the packets we were
@@ -1650,10 +1716,12 @@ print_packets(capture_file *cf, print_args_t *print_args)
                                   "selected packets", print_packet,
                                   &callback_args);
 
-  if (callback_args.col_widths != NULL)
-    g_free(callback_args.col_widths);
+  if (callback_args.header_line_buf != NULL)
+    g_free(callback_args.header_line_buf);
   if (callback_args.line_buf != NULL)
     g_free(callback_args.line_buf);
+  if (callback_args.col_widths != NULL)
+    g_free(callback_args.col_widths);
 
   switch (ret) {
 
@@ -1687,11 +1755,152 @@ print_packets(capture_file *cf, print_args_t *print_args)
     return PP_WRITE_ERROR;
   }
 
+  /* XXX - check for an error */
   close_print_dest(print_args->to_file, callback_args.print_fh);
 
   return PP_OK;
 }
 
+static gboolean
+write_pdml_packet(capture_file *cf _U_, frame_data *fdata,
+                  union wtap_pseudo_header *pseudo_header, const guint8 *pd,
+                 void *argsp)
+{
+  FILE *fh = argsp;
+  epan_dissect_t *edt;
+
+  /* Create the protocol tree, but don't fill in the column information. */
+  edt = epan_dissect_new(TRUE, TRUE);
+  epan_dissect_run(edt, pseudo_header, pd, fdata, NULL);
+
+  /* Write out the information in that tree. */
+  proto_tree_write_pdml(edt, fh);
+
+  epan_dissect_free(edt);
+
+  return !ferror(fh);
+}
+
+pp_return_t
+write_pdml_packets(capture_file *cf, print_args_t *print_args)
+{
+  FILE        *fh;
+  psp_return_t ret;
+
+  fh = fopen(print_args->file, "w");
+  if (fh == NULL)
+    return PP_OPEN_ERROR;      /* attempt to open destination failed */
+
+  write_pdml_preamble(fh);
+  if (ferror(fh)) {
+    fclose(fh);
+    return PP_WRITE_ERROR;
+  }
+
+  /* Iterate through the list of packets, printing the packets we were
+     told to print. */
+  ret = process_specified_packets(cf, &print_args->range, "Writing PDML",
+                                  "selected packets", write_pdml_packet,
+                                  fh);
+
+  switch (ret) {
+
+  case PSP_FINISHED:
+    /* Completed successfully. */
+    break;
+
+  case PSP_STOPPED:
+    /* Well, the user decided to abort the printing. */
+    break;
+
+  case PSP_FAILED:
+    /* Error while printing. */
+    fclose(fh);
+    return PP_WRITE_ERROR;
+  }
+
+  write_pdml_finale(fh);
+  if (ferror(fh)) {
+    fclose(fh);
+    return PP_WRITE_ERROR;
+  }
+
+  /* XXX - check for an error */
+  fclose(fh);
+
+  return PP_OK;
+}
+
+static gboolean
+write_psml_packet(capture_file *cf, frame_data *fdata,
+                  union wtap_pseudo_header *pseudo_header, const guint8 *pd,
+                 void *argsp)
+{
+  FILE *fh = argsp;
+  epan_dissect_t *edt;
+
+  /* Fill in the column information, but don't create the protocol tree. */
+  edt = epan_dissect_new(FALSE, FALSE);
+  epan_dissect_run(edt, pseudo_header, pd, fdata, &cf->cinfo);
+
+  /* Write out the information in that tree. */
+  proto_tree_write_psml(edt, fh);
+
+  epan_dissect_free(edt);
+
+  return !ferror(fh);
+}
+
+pp_return_t
+write_psml_packets(capture_file *cf, print_args_t *print_args)
+{
+  FILE        *fh;
+  psp_return_t ret;
+
+  fh = fopen(print_args->file, "w");
+  if (fh == NULL)
+    return PP_OPEN_ERROR;      /* attempt to open destination failed */
+
+  write_psml_preamble(fh);
+  if (ferror(fh)) {
+    fclose(fh);
+    return PP_WRITE_ERROR;
+  }
+
+  /* Iterate through the list of packets, printing the packets we were
+     told to print. */
+  ret = process_specified_packets(cf, &print_args->range, "Writing PSML",
+                                  "selected packets", write_psml_packet,
+                                  fh);
+
+  switch (ret) {
+
+  case PSP_FINISHED:
+    /* Completed successfully. */
+    break;
+
+  case PSP_STOPPED:
+    /* Well, the user decided to abort the printing. */
+    break;
+
+  case PSP_FAILED:
+    /* Error while printing. */
+    fclose(fh);
+    return PP_WRITE_ERROR;
+  }
+
+  write_psml_finale(fh);
+  if (ferror(fh)) {
+    fclose(fh);
+    return PP_WRITE_ERROR;
+  }
+
+  /* XXX - check for an error */
+  fclose(fh);
+
+  return PP_OK;
+}
+
 /* Scan through the packet list and change all columns that use the
    "command-line-specified" time stamp format to use the current
    value of that format. */
@@ -1911,7 +2120,7 @@ match_subtree_text(proto_node *node, gpointer data)
   }
 
   /* Don't match invisible entries. */
-  if (!fi->visible)
+  if (PROTO_ITEM_IS_HIDDEN(node))
     return;
 
   /* was a free format label produced? */
@@ -2015,7 +2224,7 @@ find_packet_data(capture_file *cf, const guint8 *string, size_t string_size)
   info.data_len = string_size;
 
   /* String or hex search? */
-  if (cf->ascii) {
+  if (cf->string) {
     /* String search - what type of string? */
     switch (cf->scs_type) {
 
@@ -2260,18 +2469,20 @@ find_packet(capture_file *cf,
            * we need an API for popping up alert boxes with
            * {Verb} and "Cancel".
            */
-          simple_dialog(ESD_TYPE_CONFIRMATION, ESD_BTN_OK,
-                        "You have searched up to the beginning of the capture.\n"
-                        "Do you want to continue the search from the end of the capture?");
+          simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
+                        "%sBeginning of capture exceeded!%s\n\n"
+                        "Search is continued from the end of the capture.",
+                        simple_dialog_primary_start(), simple_dialog_primary_end());
           fdata = cf->plist_end;       /* wrap around */
         }
       } else {
         /* Go on to the next frame. */
         fdata = fdata->next;
         if (fdata == NULL) {
-          simple_dialog(ESD_TYPE_CONFIRMATION, ESD_BTN_OK,
-                        "You have searched up to the end of the capture.\n"
-                        "Do you want to continue the search from the beginning of the capture?");
+          simple_dialog(ESD_TYPE_INFO, ESD_BTN_OK,
+                        "%sEnd of capture exceeded!%s\n\n"
+                        "Search is continued from the start of the capture.",
+                        simple_dialog_primary_start(), simple_dialog_primary_end());
           fdata = cf->plist;   /* wrap around */
         }
       }
@@ -2920,8 +3131,12 @@ cf_read_error_message(int err, gchar *err_info)
 
   switch (err) {
 
-  case WTAP_ERR_UNSUPPORTED:
   case WTAP_ERR_UNSUPPORTED_ENCAP:
+      snprintf(errmsg_errno, sizeof(errmsg_errno),
+               "The file \"%%s\" has a packet with a network type that Ethereal doesn't support.\n(%s)",
+               err_info);
+      break;
+
   case WTAP_ERR_BAD_RECORD:
     snprintf(errmsg_errno, sizeof(errmsg_errno),
             "An error occurred while reading from the file \"%%s\": %s.\n(%s)",