Use val_to_str_const().
[obnox/wireshark/wip.git] / capinfos.c
index aae98e71b80366d28125d5f64045cbc059e24b8e..831f48120e165e4e2173ea54b786a4d3f105dcd8 100644 (file)
@@ -32,7 +32,7 @@
  * Continue processing additional files after
  * a wiretap open failure.  The new -C option
  * reverts to capinfos' original behavior which
- * is to cancels any further file processing at
+ * is to cancel any further file processing at
  * first file open failure.
  *
  * Change the behavior of how the default display
  * into a tab delimited text file, or to a comma
  * separated variables file (*.csv) instead of the
  * original "long" format.
+ *
+ * 2011-04-05: wmeier
+ * behaviour changed: Upon exit capinfos will return
+ *  an error status if an error occurred at any
+ *  point during "continuous" file processing.
+ *  (Previously a success status was always
+ *   returned if the -C option was not used).
+ *
+
  */
 
 
 #include <wsutil/file_util.h>
 #endif
 
-#ifdef HAVE_GETOPT_H
-#include <getopt.h>
-#else
+#ifndef HAVE_GETOPT
 #include "wsutil/wsgetopt.h"
 #endif
 
 #ifdef _WIN32
-#include <shellapi.h>
+#include <wsutil/unicode-utils.h>
 #endif /* _WIN32 */
 
 #include "svnversion.h"
@@ -187,6 +194,8 @@ typedef struct _capture_info {
   double        packet_size;
   double        data_rate;              /* in bytes */
   gboolean      in_order;
+
+  int          *encap_counts;           /* array of per_packet encap counts; array has one entry per wtap_encap type */
 } capture_info;
 
 static void
@@ -315,6 +324,13 @@ print_stats(const gchar *filename, capture_info *cf_info)
   if (filename)           printf     ("File name:           %s\n", filename);
   if (cap_file_type)      printf     ("File type:           %s\n", file_type_string);
   if (cap_file_encap)     printf     ("File encapsulation:  %s\n", file_encap_string);
+  if (cap_file_encap && (cf_info->file_encap == WTAP_ENCAP_PER_PACKET)) {
+    int i;
+    for (i=0; i<WTAP_NUM_ENCAP_TYPES; i++) {
+      if (cf_info->encap_counts[i] > 0)
+        printf("                       %s\n", wtap_encap_string(i));
+    }
+  }
   if (cap_snaplen && cf_info->snap_set)
                           printf     ("Packet size limit:   file hdr: %u bytes\n", cf_info->snaplen);
   else if(cap_snaplen && !cf_info->snap_set)
@@ -342,8 +358,8 @@ print_stats(const gchar *filename, capture_info *cf_info)
                           printf     ("RIPEMD160:           %s\n", file_rmd160);
                           printf     ("MD5:                 %s\n", file_md5);
   }
-  if (cap_in_order)       printf     ("Strict time order:   %s\n", (cf_info->in_order) ? "True" : "False");
 #endif /* HAVE_LIBGCRYPT */
+  if (cap_in_order)       printf     ("Strict time order:   %s\n", (cf_info->in_order) ? "True" : "False");
 }
 
 static void
@@ -427,6 +443,11 @@ print_stats_table(const gchar *filename, capture_info *cf_info)
     putquote();
   }
 
+  /* ToDo: If WTAP_ENCAP_PER_PACKET, show the list of encapsulations encountered;
+   *       Output a line for each different encap with all fields repeated except
+   *        the encapsulation field which has "Per Packet: ..." for each
+   *        encapsulation type seen ?
+   */
   if (cap_file_encap) {
     putsep();
     putquote();
@@ -583,6 +604,8 @@ process_cap_file(wtap *wth, const char *filename)
   double               prev_time = 0;
   gboolean             in_order = TRUE;
 
+  cf_info.encap_counts = g_malloc0(WTAP_NUM_ENCAP_TYPES * sizeof(int));
+
   /* Tally up data that we need to parse through the file to find */
   while (wtap_read(wth, &err, &err_info, &data_offset))  {
     phdr = wtap_phdr(wth);
@@ -617,7 +640,16 @@ process_cap_file(wtap *wth, const char *filename)
         snaplen_max_inferred = phdr->caplen;
     }
 
-  }
+    /* Per-packet encapsulation */
+    if (wtap_file_encap(wth) == WTAP_ENCAP_PER_PACKET) {
+        if ((phdr->pkt_encap > 0) && (phdr->pkt_encap < WTAP_NUM_ENCAP_TYPES)) {
+            cf_info.encap_counts[phdr->pkt_encap] += 1;
+        } else {
+            fprintf(stderr, "capinfos: Unknown per-packet encapsulation: %d [frame number: %d]\n", phdr->pkt_encap, packet);
+        }
+    }
+
+  } /* while */
 
   if (err != 0) {
     fprintf(stderr,
@@ -628,10 +660,12 @@ process_cap_file(wtap *wth, const char *filename)
     case WTAP_ERR_UNSUPPORTED:
     case WTAP_ERR_UNSUPPORTED_ENCAP:
     case WTAP_ERR_BAD_RECORD:
+    case WTAP_ERR_DECOMPRESS:
       fprintf(stderr, "(%s)\n", err_info);
       g_free(err_info);
       break;
     }
+    g_free(cf_info.encap_counts);
     return 1;
   }
 
@@ -640,7 +674,8 @@ process_cap_file(wtap *wth, const char *filename)
   if (size == -1) {
     fprintf(stderr,
             "capinfos: Can't get size of \"%s\": %s.\n",
-            filename, strerror(err));
+            filename, g_strerror(err));
+    g_free(cf_info.encap_counts);
     return 1;
   }
 
@@ -692,6 +727,8 @@ process_cap_file(wtap *wth, const char *filename)
     print_stats_table(filename, &cf_info);
   }
 
+  g_free(cf_info.encap_counts);
+
   return 0;
 }
 
@@ -794,7 +831,7 @@ hash_to_str(const unsigned char *hash, size_t length, char *str) {
   int i;
 
   for (i = 0; i < (int) length; i++) {
-    sprintf(str+(i*2), "%02x", hash[i]);
+    g_snprintf(str+(i*2), 3, "%02x", hash[i]);
   }
 }
 #endif /* HAVE_LIBGCRYPT */
@@ -802,35 +839,25 @@ hash_to_str(const unsigned char *hash, size_t length, char *str) {
 int
 main(int argc, char *argv[])
 {
-  wtap *wth;
-  int err;
+  wtap  *wth;
+  int    err;
   gchar *err_info;
-  int opt;
-
-#ifdef _WIN32
-  LPWSTR              *wc_argv;
-  int                  wc_argc, i;
-#endif  /* _WIN32 */
+  int    opt;
+  int    overall_error_status;
 
   int status = 0;
 #ifdef HAVE_PLUGINS
-  charinit_progfile_dir_error;
+  char  *init_progfile_dir_error;
 #endif
 #ifdef HAVE_LIBGCRYPT
-  FILE *fh;
-  char *hash_buf = NULL;
+  FILE  *fh;
+  char  *hash_buf = NULL;
   gcry_md_hd_t hd = NULL;
   size_t hash_bytes;
 #endif
 
 #ifdef _WIN32
-  /* Convert our arg list to UTF-8. */
-  wc_argv = CommandLineToArgvW(GetCommandLineW(), &wc_argc);
-  if (wc_argv && wc_argc == argc) {
-    for (i = 0; i < argc; i++) {
-      argv[i] = g_utf16_to_utf8(wc_argv[i], -1, NULL, NULL, NULL);
-    }
-  } /* XXX else bail because something is horribly, horribly wrong? */
+  arg_list_utf_16to8(argc, argv);
 #endif /* _WIN32 */
 
   /*
@@ -1018,12 +1045,14 @@ main(int argc, char *argv[])
   }
 #endif
 
+  overall_error_status = 0;
+
   for (opt = optind; opt < argc; opt++) {
 
 #ifdef HAVE_LIBGCRYPT
-    strcpy(file_sha1, "<unknown>");
-    strcpy(file_rmd160, "<unknown>");
-    strcpy(file_md5, "<unknown>");
+    g_strlcpy(file_sha1, "<unknown>", HASH_STR_SIZE);
+    g_strlcpy(file_rmd160, "<unknown>", HASH_STR_SIZE);
+    g_strlcpy(file_md5, "<unknown>", HASH_STR_SIZE);
 
     if (cap_file_hashes) {
       fh = ws_fopen(argv[opt], "rb");
@@ -1055,8 +1084,9 @@ main(int argc, char *argv[])
         g_free(err_info);
         break;
       }
+      overall_error_status = 1; /* remember that an error has occurred */
       if(!continue_after_wtap_open_offline_failure)
-        exit(1);
+        exit(1); /* error status */
     }
 
     if(wth) {
@@ -1069,6 +1099,5 @@ main(int argc, char *argv[])
         exit(status);
     }
   }
-  return 0;
+  return overall_error_status;
 }
-