X-Git-Url: http://git.samba.org/?p=metze%2Fwireshark%2Fwip.git;a=blobdiff_plain;f=reordercap.c;h=7f3463d89bd66f04ab2ebffb85e2c56f09a6d4e8;hp=ba5a9d87a44027052cca469e59ee086314a307f0;hb=8dbca7320d98606cedbf720ee38d1982ad9eee24;hpb=2d8ec49ffad14a1f1ea7a71c99e617bbfd0767a9 diff --git a/reordercap.c b/reordercap.c index ba5a9d87a4..7f3463d89b 100644 --- a/reordercap.c +++ b/reordercap.c @@ -5,20 +5,7 @@ * By Gerald Combs * 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., - * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. - * + * SPDX-License-Identifier: GPL-2.0-or-later */ #include @@ -28,30 +15,35 @@ #include #include -#ifdef HAVE_UNISTD_H -#include -#endif - #ifdef HAVE_GETOPT_H #include #endif -#ifdef HAVE_LIBZ -#include /* to get the libz version number */ -#endif - -#include "wtap.h" +#include -#ifndef HAVE_GETOPT +#ifndef HAVE_GETOPT_LONG #include "wsutil/wsgetopt.h" #endif +#include +#include #include -#include -#include -#include +#include +#include +#include +#include -#include "version_info.h" +#ifdef HAVE_PLUGINS +#include +#endif + +#include + +#include "ui/failure_message.h" + +#define INVALID_OPTION 1 +#define OPEN_ERROR 2 +#define OUTPUT_FILE_ERROR 1 /* Show command-line usage */ static void @@ -70,7 +62,7 @@ typedef struct FrameRecord_t { gint64 offset; guint num; - nstime_t time; + nstime_t frame_time; } FrameRecord_t; @@ -90,7 +82,8 @@ typedef struct FrameRecord_t { static void frame_write(FrameRecord_t *frame, wtap *wth, wtap_dumper *pdh, - struct wtap_pkthdr *phdr, Buffer *buf, const char *infile) + wtap_rec *rec, Buffer *buf, const char *infile, + const char *outfile) { int err; gchar *err_info; @@ -100,33 +93,27 @@ frame_write(FrameRecord_t *frame, wtap *wth, wtap_dumper *pdh, /* Re-read the frame from the stored location */ - if (!wtap_seek_read(wth, frame->offset, phdr, buf, &err, &err_info)) { + if (!wtap_seek_read(wth, frame->offset, rec, buf, &err, &err_info)) { if (err != 0) { /* Print a message noting that the read failed somewhere along the line. */ fprintf(stderr, - "reordercap: An error occurred while re-reading \"%s\": %s.\n", - infile, wtap_strerror(err)); - if (err_info != NULL) { - fprintf(stderr, "(%s)\n", err_info); - g_free(err_info); - } + "reordercap: An error occurred while re-reading \"%s\".\n", + infile); + cfile_read_failure_message("reordercap", infile, err, err_info); exit(1); } } /* Copy, and set length and timestamp from item. */ - /* TODO: remove when wtap_seek_read() fills in phdr, + /* TODO: remove when wtap_seek_read() fills in rec, including time stamps, for all file types */ - phdr->ts = frame->time; + rec->ts = frame->frame_time; /* Dump frame to outfile */ - if (!wtap_dump(pdh, phdr, ws_buffer_start_ptr(buf), &err, &err_info)) { - fprintf(stderr, "reordercap: Error (%s) writing frame to outfile\n", - wtap_strerror(err)); - if (err_info != NULL) { - fprintf(stderr, "(%s)\n", err_info); - g_free(err_info); - } + if (!wtap_dump(pdh, rec, ws_buffer_start_ptr(buf), &err, &err_info)) { + cfile_write_failure_message("reordercap", infile, outfile, err, + err_info, frame->num, + wtap_file_type_subtype(wth)); exit(1); } } @@ -142,36 +129,32 @@ frames_compare(gconstpointer a, gconstpointer b) const FrameRecord_t *frame1 = *(const FrameRecord_t *const *) a; const FrameRecord_t *frame2 = *(const FrameRecord_t *const *) b; - const nstime_t *time1 = &frame1->time; - const nstime_t *time2 = &frame2->time; + const nstime_t *time1 = &frame1->frame_time; + const nstime_t *time2 = &frame2->frame_time; return nstime_cmp(time1, time2); } +/* + * General errors and warnings are reported with an console message + * in reordercap. + */ static void -get_reordercap_compiled_info(GString *str) +failure_warning_message(const char *msg_format, va_list ap) { - /* LIBZ */ - g_string_append(str, ", "); -#ifdef HAVE_LIBZ - g_string_append(str, "with libz "); -#ifdef ZLIB_VERSION - g_string_append(str, ZLIB_VERSION); -#else /* ZLIB_VERSION */ - g_string_append(str, "(version unknown)"); -#endif /* ZLIB_VERSION */ -#else /* HAVE_LIBZ */ - g_string_append(str, "without libz"); -#endif /* HAVE_LIBZ */ + fprintf(stderr, "reordercap: "); + vfprintf(stderr, msg_format, ap); + fprintf(stderr, "\n"); } +/* + * Report additional information for an error in command-line arguments. + */ static void -get_reordercap_runtime_info(GString *str) +failure_message_cont(const char *msg_format, va_list ap) { - /* zlib */ -#if defined(HAVE_LIBZ) && !defined(_WIN32) - g_string_append_printf(str, ", with libz %s", zlibVersion()); -#endif + vfprintf(stderr, msg_format, ap); + fprintf(stderr, "\n"); } /********************************************************************/ @@ -180,50 +163,60 @@ get_reordercap_runtime_info(GString *str) int main(int argc, char *argv[]) { - GString *comp_info_str; - GString *runtime_info_str; + char *init_progfile_dir_error; wtap *wth = NULL; wtap_dumper *pdh = NULL; - struct wtap_pkthdr dump_phdr; + wtap_rec dump_rec; Buffer buf; int err; gchar *err_info; gint64 data_offset; - const struct wtap_pkthdr *phdr; + const wtap_rec *rec; guint wrong_order_count = 0; gboolean write_output_regardless = TRUE; guint i; - wtapng_section_t *shb_hdr; - wtapng_iface_descriptions_t *idb_inf; + wtap_dump_params params; + int ret = EXIT_SUCCESS; GPtrArray *frames; FrameRecord_t *prevFrame = NULL; int opt; static const struct option long_options[] = { - {(char *)"help", no_argument, NULL, 'h'}, - {(char *)"version", no_argument, NULL, 'v'}, + {"help", no_argument, NULL, 'h'}, + {"version", no_argument, NULL, 'v'}, {0, 0, 0, 0 } }; int file_count; char *infile; - char *outfile; - - /* Assemble the compile-time version information string */ - comp_info_str = g_string_new("Compiled "); - get_compiled_version_info(comp_info_str, NULL, get_reordercap_compiled_info); + const char *outfile; + + cmdarg_err_init(failure_warning_message, failure_message_cont); + + /* Initialize the version information. */ + ws_init_version_info("Reordercap (Wireshark)", NULL, NULL, NULL); + + /* + * Get credential information for later use. + */ + init_process_policies(); + + /* + * Attempt to get the pathname of the directory containing the + * executable file. + */ + init_progfile_dir_error = init_progfile_dir(argv[0]); + if (init_progfile_dir_error != NULL) { + fprintf(stderr, + "reordercap: Can't get pathname of directory containing the reordercap program: %s.\n", + init_progfile_dir_error); + g_free(init_progfile_dir_error); + } - /* Assemble the run-time version information string */ - runtime_info_str = g_string_new("Running "); - get_runtime_version_info(runtime_info_str, get_reordercap_runtime_info); + init_report_message(failure_warning_message, failure_warning_message, + NULL, NULL, NULL); - /* Add it to the information to be reported on a crash. */ - ws_add_crash_info("Reordercap (Wireshark) %s\n" - "\n" - "%s" - "\n" - "%s", - get_ws_vcs_version_info(), comp_info_str->str, runtime_info_str->str); + wtap_init(TRUE); /* Process the options first */ while ((opt = getopt_long(argc, argv, "hnv", long_options, NULL)) != -1) { @@ -232,20 +225,16 @@ main(int argc, char *argv[]) write_output_regardless = FALSE; break; case 'h': - printf("Reordercap (Wireshark) %s\n" - "Reorder timestamps of input file frames into output file.\n" - "See http://www.wireshark.org for more information.\n", - get_ws_vcs_version_info()); + show_help_header("Reorder timestamps of input file frames into output file."); print_usage(stdout); - exit(0); + goto clean_exit; case 'v': - show_version("Reordercap (Wireshark)", comp_info_str, runtime_info_str); - g_string_free(comp_info_str, TRUE); - g_string_free(runtime_info_str, TRUE); - exit(0); + show_version(); + goto clean_exit; case '?': print_usage(stderr); - exit(1); + ret = INVALID_OPTION; + goto clean_exit; } } @@ -257,7 +246,8 @@ main(int argc, char *argv[]) } else { print_usage(stderr); - exit(1); + ret = INVALID_OPTION; + goto clean_exit; } /* Open infile */ @@ -265,28 +255,29 @@ main(int argc, char *argv[]) open_routine reader to use, then the following needs to change. */ wth = wtap_open_offline(infile, WTAP_TYPE_AUTO, &err, &err_info, TRUE); if (wth == NULL) { - fprintf(stderr, "reordercap: Can't open %s: %s\n", infile, - wtap_strerror(err)); - if (err_info != NULL) { - fprintf(stderr, "(%s)\n", err_info); - g_free(err_info); - } - exit(1); + cfile_open_failure_message("reordercap", infile, err, err_info); + ret = OPEN_ERROR; + goto clean_exit; } - DEBUG_PRINT("file_type_subtype is %u\n", wtap_file_type_subtype(wth)); + DEBUG_PRINT("file_type_subtype is %d\n", wtap_file_type_subtype(wth)); - shb_hdr = wtap_file_get_shb_info(wth); - idb_inf = wtap_file_get_idb_info(wth); + wtap_dump_params_init(¶ms, wth); /* Open outfile (same filetype/encap as input file) */ - pdh = wtap_dump_open_ng(outfile, wtap_file_type_subtype(wth), wtap_file_encap(wth), - 65535, FALSE, shb_hdr, idb_inf, &err); - g_free(idb_inf); + if (strcmp(outfile, "-") == 0) { + pdh = wtap_dump_open_stdout(wtap_file_type_subtype(wth), WTAP_UNCOMPRESSED, ¶ms, &err); + } else { + pdh = wtap_dump_open(outfile, wtap_file_type_subtype(wth), WTAP_UNCOMPRESSED, ¶ms, &err); + } + g_free(params.idb_inf); + params.idb_inf = NULL; + if (pdh == NULL) { - fprintf(stderr, "reordercap: Failed to open output file: (%s) - error %s\n", - outfile, wtap_strerror(err)); - g_free(shb_hdr); - exit(1); + cfile_dump_open_failure_message("reordercap", outfile, err, + wtap_file_type_subtype(wth)); + wtap_dump_params_cleanup(¶ms); + ret = OUTPUT_FILE_ERROR; + goto clean_exit; } /* Allocate the array of frame pointers. */ @@ -296,15 +287,15 @@ main(int argc, char *argv[]) while (wtap_read(wth, &err, &err_info, &data_offset)) { FrameRecord_t *newFrameRecord; - phdr = wtap_phdr(wth); + rec = wtap_get_rec(wth); newFrameRecord = g_slice_new(FrameRecord_t); newFrameRecord->num = frames->len + 1; newFrameRecord->offset = data_offset; - if (phdr->presence_flags & WTAP_HAS_TS) { - newFrameRecord->time = phdr->ts; + if (rec->presence_flags & WTAP_HAS_TS) { + newFrameRecord->frame_time = rec->ts; } else { - nstime_set_unset(&newFrameRecord->time); + nstime_set_unset(&newFrameRecord->frame_time); } if (prevFrame && frames_compare(&newFrameRecord, &prevFrame) < 0) { @@ -316,13 +307,7 @@ main(int argc, char *argv[]) } if (err != 0) { /* Print a message noting that the read failed somewhere along the line. */ - fprintf(stderr, - "reordercap: An error occurred while reading \"%s\": %s.\n", - infile, wtap_strerror(err)); - if (err_info != NULL) { - fprintf(stderr, "(%s)\n", err_info); - g_free(err_info); - } + cfile_read_failure_message("reordercap", infile, err, err_info); } printf("%u frames, %u out of order\n", frames->len, wrong_order_count); @@ -333,22 +318,22 @@ main(int argc, char *argv[]) } /* Write out each sorted frame in turn */ - wtap_phdr_init(&dump_phdr); + wtap_rec_init(&dump_rec); ws_buffer_init(&buf, 1500); for (i = 0; i < frames->len; i++) { FrameRecord_t *frame = (FrameRecord_t *)frames->pdata[i]; /* Avoid writing if already sorted and configured to */ if (write_output_regardless || (wrong_order_count > 0)) { - frame_write(frame, wth, pdh, &dump_phdr, &buf, infile); + frame_write(frame, wth, pdh, &dump_rec, &buf, infile, outfile); } g_slice_free(FrameRecord_t, frame); } - wtap_phdr_cleanup(&dump_phdr); + wtap_rec_cleanup(&dump_rec); ws_buffer_free(&buf); if (!write_output_regardless && (wrong_order_count == 0)) { - printf("Not writing output file because input file is already in order!\n"); + printf("Not writing output file because input file is already in order.\n"); } /* Free the whole array */ @@ -356,17 +341,20 @@ main(int argc, char *argv[]) /* Close outfile */ if (!wtap_dump_close(pdh, &err)) { - fprintf(stderr, "reordercap: Error closing %s: %s\n", outfile, - wtap_strerror(err)); - g_free(shb_hdr); - exit(1); + cfile_close_failure_message(outfile, err); + wtap_dump_params_cleanup(¶ms); + ret = OUTPUT_FILE_ERROR; + goto clean_exit; } - g_free(shb_hdr); + wtap_dump_params_cleanup(¶ms); - /* Finally, close infile */ - wtap_fdclose(wth); + /* Finally, close infile and release resources. */ + wtap_close(wth); - return 0; +clean_exit: + wtap_cleanup(); + free_progdirs(); + return ret; } /*