More getopt_long, to add --help and --version support.
[metze/wireshark/wip.git] / mergecap.c
index 3a84aeb74b8df34709a20aa8cde4837bc59a1b97..44c8c71b4308f8101d76851065cf676369902b11 100644 (file)
@@ -1,6 +1,4 @@
 /* Combine dump files, either by appending or by merging by timestamp
- *
- * $Id$
  *
  * Wireshark - Network traffic analyzer
  * By Gerald Combs <gerald@wireshark.org>
 #include <unistd.h>
 #endif
 
+#ifdef HAVE_GETOPT_H
+#include <getopt.h>
+#endif
+
 #ifdef HAVE_SYS_TIME_H
 #include <sys/time.h>
 #endif
@@ -52,7 +54,7 @@
 
 #include <wiretap/merge.h>
 
-#include "svnversion.h"
+#include "version.h"
 
 #ifdef HAVE_FCNTL_H
 #include <fcntl.h>
@@ -65,7 +67,7 @@
 static int
 get_natural_int(const char *string, const char *name)
 {
-  long number;
+  long  number;
   char *p;
 
   number = strtol(string, &p, 10);
@@ -101,36 +103,51 @@ get_positive_int(const char *string, const char *name)
   return number;
 }
 
+static void
+print_version(FILE *output)
+{
+  fprintf(output, "Mergecap %s"
+#ifdef GITVERSION
+      " (" GITVERSION " from " GITBRANCH ")"
+#endif
+      "\n", VERSION);
+}
+
 /*
  * Show the usage
  */
 static void
-usage(void)
+usage(gboolean is_error)
 {
-  fprintf(stderr, "Mergecap %s"
-#ifdef SVNVERSION
-          " (" SVNVERSION " from " SVNPATH ")"
-#endif
-          "\n", VERSION);
-  fprintf(stderr, "Merge two or more capture files into one.\n");
-  fprintf(stderr, "See http://www.wireshark.org for more information.\n");
-  fprintf(stderr, "\n");
-  fprintf(stderr, "Usage: mergecap [options] -w <outfile>|- <infile> [<infile> ...]\n");
-  fprintf(stderr, "\n");
-  fprintf(stderr, "Output:\n");
-  fprintf(stderr, "  -a                concatenate rather than merge files.\n");
-  fprintf(stderr, "                    default is to merge based on frame timestamps.\n");
-  fprintf(stderr, "  -s <snaplen>      truncate packets to <snaplen> bytes of data.\n");
-  fprintf(stderr, "  -w <outfile>|-    set the output filename to <outfile> or '-' for stdout.\n");
-  fprintf(stderr, "  -F <capture type> set the output file type; default is pcapng.\n");
-  fprintf(stderr, "                    an empty \"-F\" option will list the file types.\n");
-  fprintf(stderr, "  -T <encap type>   set the output file encapsulation type;\n");
-  fprintf(stderr, "                    default is the same as the first input file.\n");
-  fprintf(stderr, "                    an empty \"-T\" option will list the encapsulation types.\n");
-  fprintf(stderr, "\n");
-  fprintf(stderr, "Miscellaneous:\n");
-  fprintf(stderr, "  -h                display this help and exit.\n");
-  fprintf(stderr, "  -v                verbose output.\n");
+  FILE *output;
+
+  if (!is_error) {
+    output = stdout;
+  }
+  else {
+    output = stderr;
+  }
+
+  print_version(output);
+  fprintf(output, "Merge two or more capture files into one.\n");
+  fprintf(output, "See http://www.wireshark.org for more information.\n");
+  fprintf(output, "\n");
+  fprintf(output, "Usage: mergecap [options] -w <outfile>|- <infile> [<infile> ...]\n");
+  fprintf(output, "\n");
+  fprintf(output, "Output:\n");
+  fprintf(output, "  -a                concatenate rather than merge files.\n");
+  fprintf(output, "                    default is to merge based on frame timestamps.\n");
+  fprintf(output, "  -s <snaplen>      truncate packets to <snaplen> bytes of data.\n");
+  fprintf(output, "  -w <outfile>|-    set the output filename to <outfile> or '-' for stdout.\n");
+  fprintf(output, "  -F <capture type> set the output file type; default is pcapng.\n");
+  fprintf(output, "                    an empty \"-F\" option will list the file types.\n");
+  fprintf(output, "  -T <encap type>   set the output file encapsulation type;\n");
+  fprintf(output, "                    default is the same as the first input file.\n");
+  fprintf(output, "                    an empty \"-T\" option will list the encapsulation types.\n");
+  fprintf(output, "\n");
+  fprintf(output, "Miscellaneous:\n");
+  fprintf(output, "  -h                display this help and exit.\n");
+  fprintf(output, "  -v                verbose output.\n");
 }
 
 struct string_elem {
@@ -165,13 +182,13 @@ list_capture_types(void) {
   struct string_elem *captypes;
   GSList *list = NULL;
 
-  captypes = g_new(struct string_elem,WTAP_NUM_FILE_TYPES);
+  captypes = g_new(struct string_elem,WTAP_NUM_FILE_TYPES_SUBTYPES);
 
   fprintf(stderr, "mergecap: The available capture file types for the \"-F\" flag are:\n");
-  for (i = 0; i < WTAP_NUM_FILE_TYPES; i++) {
+  for (i = 0; i < WTAP_NUM_FILE_TYPES_SUBTYPES; i++) {
     if (wtap_dump_can_open(i)) {
-      captypes[i].sstr = wtap_file_type_short_string(i);
-      captypes[i].lstr = wtap_file_type_string(i);
+      captypes[i].sstr = wtap_file_type_subtype_short_string(i);
+      captypes[i].lstr = wtap_file_type_subtype_string(i);
       list = g_slist_insert_sorted(list, &captypes[i], string_compare);
     }
   }
@@ -203,28 +220,33 @@ list_encap_types(void) {
 int
 main(int argc, char *argv[])
 {
-  int          opt;
-  gboolean     do_append = FALSE;
-  gboolean     verbose = FALSE;
-  int          in_file_count = 0;
-  guint        snaplen = 0;
+  int                 opt;
+  static const struct option long_options[] = {
+      {(char *)"help", no_argument, NULL, 'h'},
+      {(char *)"version", no_argument, NULL, 'V'},
+      {0, 0, 0, 0 }
+  };
+  gboolean            do_append          = FALSE;
+  gboolean            verbose            = FALSE;
+  int                 in_file_count      = 0;
+  guint               snaplen            = 0;
 #ifdef PCAP_NG_DEFAULT
-  int          file_type = WTAP_FILE_PCAPNG;    /* default to pcap format */
+  int                 file_type          = WTAP_FILE_TYPE_SUBTYPE_PCAPNG; /* default to pcap format */
 #else
-  int          file_type = WTAP_FILE_PCAP;      /* default to pcapng format */
+  int                 file_type          = WTAP_FILE_TYPE_SUBTYPE_PCAP; /* default to pcapng format */
 #endif
-  int          frame_type = -2;
-  int          out_fd;
-  merge_in_file_t   *in_files = NULL, *in_file;
-  int          i;
+  int                 frame_type         = -2;
+  int                 out_fd;
+  merge_in_file_t    *in_files           = NULL, *in_file;
+  int                 i;
   struct wtap_pkthdr *phdr, snap_phdr;
-  wtap_dumper *pdh;
-  int          open_err, read_err = 0, write_err, close_err;
-  gchar       *err_info;
-  int          err_fileno;
-  char        *out_filename = NULL;
-  gboolean     got_read_error = FALSE, got_write_error = FALSE;
-  int          count;
+  wtap_dumper        *pdh;
+  int                 open_err, read_err = 0, write_err, close_err;
+  gchar              *err_info;
+  int                 err_fileno;
+  char               *out_filename       = NULL;
+  gboolean            got_read_error     = FALSE, got_write_error = FALSE;
+  int                 count;
 
 #ifdef _WIN32
   arg_list_utf_16to8(argc, argv);
@@ -232,7 +254,7 @@ main(int argc, char *argv[])
 #endif /* _WIN32 */
 
   /* Process the options first */
-  while ((opt = getopt(argc, argv, "aF:hs:T:vw:")) != -1) {
+  while ((opt = getopt_long(argc, argv, "aF:hs:T:vVw:", long_options, NULL)) != -1) {
 
     switch (opt) {
     case 'a':
@@ -240,7 +262,7 @@ main(int argc, char *argv[])
       break;
 
     case 'F':
-      file_type = wtap_short_string_to_file_type(optarg);
+      file_type = wtap_short_string_to_file_type_subtype(optarg);
       if (file_type < 0) {
         fprintf(stderr, "mergecap: \"%s\" isn't a valid capture file type\n",
                 optarg);
@@ -250,7 +272,7 @@ main(int argc, char *argv[])
       break;
 
     case 'h':
-      usage();
+      usage(FALSE);
       exit(0);
       break;
 
@@ -272,6 +294,11 @@ main(int argc, char *argv[])
       verbose = TRUE;
       break;
 
+    case 'V':
+      print_version(stdout);
+      exit(0);
+      break;
+
     case 'w':
       out_filename = optarg;
       break;
@@ -285,7 +312,7 @@ main(int argc, char *argv[])
         list_encap_types();
         break;
       default:
-        usage();
+        usage(TRUE);
       }
       exit(1);
       break;
@@ -326,7 +353,7 @@ main(int argc, char *argv[])
   if (verbose) {
     for (i = 0; i < in_file_count; i++)
       fprintf(stderr, "mergecap: %s is type %s.\n", argv[optind + i],
-              wtap_file_type_string(wtap_file_type(in_files[i].wth)));
+              wtap_file_type_subtype_string(wtap_file_type_subtype(in_files[i].wth)));
   }
 
   if (snaplen == 0) {
@@ -389,7 +416,7 @@ main(int argc, char *argv[])
   }
 
   /* prepare the outfile */
-  if(file_type == WTAP_FILE_PCAPNG ){
+  if(file_type == WTAP_FILE_TYPE_SUBTYPE_PCAPNG ){
     wtapng_section_t *shb_hdr;
     GString *comment_gstr;
 
@@ -459,11 +486,18 @@ main(int argc, char *argv[])
   }
 
   merge_close_in_files(in_file_count, in_files);
-  if (!got_read_error && !got_write_error) {
+  if (!got_write_error) {
     if (!wtap_dump_close(pdh, &write_err))
       got_write_error = TRUE;
-  } else
-    wtap_dump_close(pdh, &close_err);
+  } else {
+    /*
+     * We already got a write error; no need to report another
+     * write error on close.
+     *
+     * Don't overwrite the earlier write error.
+     */
+    (void)wtap_dump_close(pdh, &close_err);
+  }
 
   if (got_read_error) {
     /*
@@ -491,11 +525,24 @@ main(int argc, char *argv[])
 
     case WTAP_ERR_UNSUPPORTED_ENCAP:
       /*
-       * This is a problem with the particular frame we're writing;
-       * note that, and give the frame number.
+       * This is a problem with the particular frame we're writing and
+       * the file type and subtype we're wwriting; note that, and
+       * report the frame number and file type/subtype.
+       */
+      fprintf(stderr, "mergecap: Frame %u of \"%s\" has a network type that can't be saved in a \"%s\" file.\n",
+              in_file ? in_file->packet_num : 0, in_file ? in_file->filename : "UNKNOWN",
+              wtap_file_type_subtype_string(file_type));
+      break;
+
+    case WTAP_ERR_PACKET_TOO_LARGE:
+      /*
+       * This is a problem with the particular frame we're writing and
+       * the file type and subtype we're wwriting; note that, and
+       * report the frame number and file type/subtype.
        */
-      fprintf(stderr, "mergecap: Frame %u of \"%s\" has a network type that can't be saved in a file with that format\n.",
-              in_file->packet_num, in_file->filename);
+      fprintf(stderr, "mergecap: Frame %u of \"%s\" is too large for a \"%s\" file\n.",
+              in_file ? in_file->packet_num : 0, in_file ? in_file->filename : "UNKNOWN",
+              wtap_file_type_subtype_string(file_type));
       break;
 
     default:
@@ -515,11 +562,11 @@ main(int argc, char *argv[])
  *
  * Local variables:
  * c-basic-offset: 2
- * tab-width: 2
+ * tab-width: 8
  * indent-tabs-mode: nil
  * End:
  *
- * vi: set shiftwidth=2 tabstop=2 expandtab:
- * :indentSize=2:tabSize=2:noTabs=true:
+ * vi: set shiftwidth=2 tabstop=8 expandtab:
+ * :indentSize=2:tabSize=8:noTabs=true:
  */