Only include <wsutil/os_version_info.h> if we need it.
[metze/wireshark/wip.git] / editcap.c
1 /* Edit capture files.  We can delete packets, adjust timestamps, or
2  * simply convert from one format to another format.
3  *
4  * Originally written by Richard Sharpe.
5  * Improved by Guy Harris.
6  * Further improved by Richard Sharpe.
7  *
8  * Copyright 2013, Richard Sharpe <realrichardsharpe[AT]gmail.com>
9  *
10  * Wireshark - Network traffic analyzer
11  * By Gerald Combs <gerald@wireshark.org>
12  * Copyright 1998 Gerald Combs
13  *
14  * This program is free software; you can redistribute it and/or modify
15  * it under the terms of the GNU General Public License as published by
16  * the Free Software Foundation; either version 2 of the License, or
17  * (at your option) any later version.
18  *
19  * This program is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22  * GNU General Public License for more details.
23  *
24  * You should have received a copy of the GNU General Public License along
25  * with this program; if not, write to the Free Software Foundation, Inc.,
26  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
27  */
28
29 #include <config.h>
30
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <stdarg.h>
35
36 /*
37  * Just make sure we include the prototype for strptime as well
38  * (needed for glibc 2.2) but make sure we do this only if not
39  * yet defined.
40  */
41
42 #ifndef __USE_XOPEN
43 #  define __USE_XOPEN
44 #endif
45
46 #include <time.h>
47 #include <glib.h>
48
49 #ifdef HAVE_UNISTD_H
50 #include <unistd.h>
51 #endif
52
53 #ifdef HAVE_GETOPT_H
54 #include <getopt.h>
55 #endif
56
57 #ifdef HAVE_SYS_TIME_H
58 #include <sys/time.h>
59 #endif
60
61 #ifdef HAVE_LIBZ
62 #include <zlib.h>     /* to get the libz version number */
63 #endif
64
65 #include "wtap.h"
66
67 #ifndef HAVE_GETOPT_LONG
68 #include "wsutil/wsgetopt.h"
69 #endif
70
71 #ifdef _WIN32
72 #include <wsutil/file_util.h>
73 #include <wsutil/unicode-utils.h>
74 #include <process.h>    /* getpid */
75 #ifdef HAVE_WINSOCK2_H
76 #include <winsock2.h>
77 #endif
78 #endif
79
80 #ifndef HAVE_STRPTIME
81 # include "wsutil/strptime.h"
82 #endif
83
84 #include <wsutil/privileges.h>
85 #include <wsutil/filesystem.h>
86 #include <wsutil/report_err.h>
87 #include <wsutil/strnatcmp.h>
88 #include <wsutil/md5.h>
89 #include <wsutil/plugins.h>
90 #include <wsutil/crash_info.h>
91 #include <wsutil/ws_version_info.h>
92
93 #include "version_info.h"
94
95 #include "ringbuffer.h" /* For RINGBUFFER_MAX_NUM_FILES */
96
97 /*
98  * Some globals so we can pass things to various routines
99  */
100
101 struct select_item {
102     int inclusive;
103     int first, second;
104 };
105
106 /*
107  * Duplicate frame detection
108  */
109 typedef struct _fd_hash_t {
110     md5_byte_t digest[16];
111     guint32    len;
112     nstime_t   time;
113 } fd_hash_t;
114
115 #define DEFAULT_DUP_DEPTH       5   /* Used with -d */
116 #define MAX_DUP_DEPTH     1000000   /* the maximum window (and actual size of fd_hash[]) for de-duplication */
117
118 static fd_hash_t fd_hash[MAX_DUP_DEPTH];
119 static int       dup_window    = DEFAULT_DUP_DEPTH;
120 static int       cur_dup_entry = 0;
121
122 static int       ignored_bytes  = 0;  /* Used with -I */
123
124 #define ONE_BILLION 1000000000
125
126 /* Weights of different errors we can introduce */
127 /* We should probably make these command-line arguments */
128 /* XXX - Should we add a bit-level error? */
129 #define ERR_WT_BIT      5   /* Flip a random bit */
130 #define ERR_WT_BYTE     5   /* Substitute a random byte */
131 #define ERR_WT_ALNUM    5   /* Substitute a random character in [A-Za-z0-9] */
132 #define ERR_WT_FMT      2   /* Substitute "%s" */
133 #define ERR_WT_AA       1   /* Fill the remainder of the buffer with 0xAA */
134 #define ERR_WT_TOTAL    (ERR_WT_BIT + ERR_WT_BYTE + ERR_WT_ALNUM + ERR_WT_FMT + ERR_WT_AA)
135
136 #define ALNUM_CHARS     "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
137 #define ALNUM_LEN       (sizeof(ALNUM_CHARS) - 1)
138
139 struct time_adjustment {
140     nstime_t tv;
141     int is_negative;
142 };
143
144 typedef struct _chop_t {
145     int len_begin;
146     int off_begin_pos;
147     int off_begin_neg;
148     int len_end;
149     int off_end_pos;
150     int off_end_neg;
151 } chop_t;
152
153 #define MAX_SELECTIONS 512
154 static struct select_item     selectfrm[MAX_SELECTIONS];
155 static int                    max_selected              = -1;
156 static int                    keep_em                   = 0;
157 #ifdef PCAP_NG_DEFAULT
158 static int                    out_file_type_subtype     = WTAP_FILE_TYPE_SUBTYPE_PCAPNG; /* default to pcapng   */
159 #else
160 static int                    out_file_type_subtype     = WTAP_FILE_TYPE_SUBTYPE_PCAP; /* default to pcap     */
161 #endif
162 static int                    out_frame_type            = -2; /* Leave frame type alone */
163 static int                    verbose                   = 0;  /* Not so verbose         */
164 static struct time_adjustment time_adj                  = {{0, 0}, 0}; /* no adjustment */
165 static nstime_t               relative_time_window      = {0, 0}; /* de-dup time window */
166 static double                 err_prob                  = 0.0;
167 static time_t                 starttime                 = 0;
168 static time_t                 stoptime                  = 0;
169 static gboolean               check_startstop           = FALSE;
170 static gboolean               dup_detect                = FALSE;
171 static gboolean               dup_detect_by_time        = FALSE;
172
173 static int                    do_strict_time_adjustment = FALSE;
174 static struct time_adjustment strict_time_adj           = {{0, 0}, 0}; /* strict time adjustment */
175 static nstime_t               previous_time             = {0, 0}; /* previous time */
176
177 static int find_dct2000_real_data(guint8 *buf);
178 static void handle_chopping(chop_t chop, struct wtap_pkthdr *out_phdr,
179                             const struct wtap_pkthdr *in_phdr, guint8 **buf,
180                             gboolean adjlen);
181
182 static gchar *
183 abs_time_to_str_with_sec_resolution(const nstime_t *abs_time)
184 {
185     struct tm *tmp;
186     gchar     *buf = (gchar *)g_malloc(16);
187
188     tmp = localtime(&abs_time->secs);
189
190     if (tmp) {
191         g_snprintf(buf, 16, "%d%02d%02d%02d%02d%02d",
192             tmp->tm_year + 1900,
193             tmp->tm_mon+1,
194             tmp->tm_mday,
195             tmp->tm_hour,
196             tmp->tm_min,
197             tmp->tm_sec);
198     } else {
199         buf[0] = '\0';
200     }
201
202     return buf;
203 }
204
205 static gchar *
206 fileset_get_filename_by_pattern(guint idx, const struct wtap_pkthdr *phdr,
207                                 gchar *fprefix, gchar *fsuffix)
208 {
209     gchar  filenum[5+1];
210     gchar *timestr;
211     gchar *abs_str;
212
213     g_snprintf(filenum, sizeof(filenum), "%05u", idx % RINGBUFFER_MAX_NUM_FILES);
214     if (phdr->presence_flags & WTAP_HAS_TS) {
215         timestr = abs_time_to_str_with_sec_resolution(&phdr->ts);
216         abs_str = g_strconcat(fprefix, "_", filenum, "_", timestr, fsuffix, NULL);
217         g_free(timestr);
218     } else
219         abs_str = g_strconcat(fprefix, "_", filenum, fsuffix, NULL);
220
221     return abs_str;
222 }
223
224 static gboolean
225 fileset_extract_prefix_suffix(const char *fname, gchar **fprefix, gchar **fsuffix)
226 {
227     char  *pfx, *last_pathsep;
228     gchar *save_file;
229
230     save_file = g_strdup(fname);
231     if (save_file == NULL) {
232         fprintf(stderr, "editcap: Out of memory\n");
233         return FALSE;
234     }
235
236     last_pathsep = strrchr(save_file, G_DIR_SEPARATOR);
237     pfx = strrchr(save_file,'.');
238     if (pfx != NULL && (last_pathsep == NULL || pfx > last_pathsep)) {
239         /* The pathname has a "." in it, and it's in the last component
240          * of the pathname (because there is either only one component,
241          * i.e. last_pathsep is null as there are no path separators,
242          * or the "." is after the path separator before the last
243          * component.
244
245          * Treat it as a separator between the rest of the file name and
246          * the file name suffix, and arrange that the names given to the
247          * ring buffer files have the specified suffix, i.e. put the
248          * changing part of the name *before* the suffix. */
249         pfx[0] = '\0';
250         *fprefix = g_strdup(save_file);
251         pfx[0] = '.'; /* restore capfile_name */
252         *fsuffix = g_strdup(pfx);
253     } else {
254         /* Either there's no "." in the pathname, or it's in a directory
255          * component, so the last component has no suffix. */
256         *fprefix = g_strdup(save_file);
257         *fsuffix = NULL;
258     }
259     g_free(save_file);
260     return TRUE;
261 }
262
263 /* Add a selection item, a simple parser for now */
264 static gboolean
265 add_selection(char *sel)
266 {
267     char *locn;
268     char *next;
269
270     if (++max_selected >= MAX_SELECTIONS) {
271         /* Let the user know we stopped selecting */
272         fprintf(stderr, "Out of room for packet selections!\n");
273         return(FALSE);
274     }
275
276     if (verbose)
277         fprintf(stderr, "Add_Selected: %s\n", sel);
278
279     if ((locn = strchr(sel, '-')) == NULL) { /* No dash, so a single number? */
280         if (verbose)
281             fprintf(stderr, "Not inclusive ...");
282
283         selectfrm[max_selected].inclusive = 0;
284         selectfrm[max_selected].first = atoi(sel);
285
286         if (verbose)
287             fprintf(stderr, " %i\n", selectfrm[max_selected].first);
288     } else {
289         if (verbose)
290             fprintf(stderr, "Inclusive ...");
291
292         next = locn + 1;
293         selectfrm[max_selected].inclusive = 1;
294         selectfrm[max_selected].first = atoi(sel);
295         selectfrm[max_selected].second = atoi(next);
296
297         if (verbose)
298             fprintf(stderr, " %i, %i\n", selectfrm[max_selected].first,
299                    selectfrm[max_selected].second);
300     }
301
302     return(TRUE);
303 }
304
305 /* Was the packet selected? */
306
307 static int
308 selected(int recno)
309 {
310     int i;
311
312     for (i = 0; i <= max_selected; i++) {
313         if (selectfrm[i].inclusive) {
314             if (selectfrm[i].first <= recno && selectfrm[i].second >= recno)
315                 return 1;
316         } else {
317             if (recno == selectfrm[i].first)
318                 return 1;
319         }
320     }
321
322   return 0;
323 }
324
325 static void
326 set_time_adjustment(char *optarg_str_p)
327 {
328     char   *frac, *end;
329     long    val;
330     size_t  frac_digits;
331
332     if (!optarg_str_p)
333         return;
334
335     /* skip leading whitespace */
336     while (*optarg_str_p == ' ' || *optarg_str_p == '\t')
337         optarg_str_p++;
338
339     /* check for a negative adjustment */
340     if (*optarg_str_p == '-') {
341         time_adj.is_negative = 1;
342         optarg_str_p++;
343     }
344
345     /* collect whole number of seconds, if any */
346     if (*optarg_str_p == '.') {         /* only fractional (i.e., .5 is ok) */
347         val  = 0;
348         frac = optarg_str_p;
349     } else {
350         val = strtol(optarg_str_p, &frac, 10);
351         if (frac == NULL || frac == optarg_str_p
352             || val == LONG_MIN || val == LONG_MAX) {
353             fprintf(stderr, "editcap: \"%s\" isn't a valid time adjustment\n",
354                     optarg_str_p);
355             exit(1);
356         }
357         if (val < 0) {            /* implies '--' since we caught '-' above  */
358             fprintf(stderr, "editcap: \"%s\" isn't a valid time adjustment\n",
359                     optarg_str_p);
360             exit(1);
361         }
362     }
363     time_adj.tv.secs = val;
364
365     /* now collect the partial seconds, if any */
366     if (*frac != '\0') {             /* chars left, so get fractional part */
367         val = strtol(&(frac[1]), &end, 10);
368         /* if more than 9 fractional digits truncate to 9 */
369         if ((end - &(frac[1])) > 9) {
370             frac[10] = 't'; /* 't' for truncate */
371             val = strtol(&(frac[1]), &end, 10);
372         }
373         if (*frac != '.' || end == NULL || end == frac || val < 0
374             || val > ONE_BILLION || val == LONG_MIN || val == LONG_MAX) {
375             fprintf(stderr, "editcap: \"%s\" isn't a valid time adjustment\n",
376                     optarg_str_p);
377             exit(1);
378         }
379     } else {
380         return;                     /* no fractional digits */
381     }
382
383     /* adjust fractional portion from fractional to numerator
384      * e.g., in "1.5" from 5 to 500000000 since .5*10^9 = 500000000 */
385     if (frac && end) {            /* both are valid */
386         frac_digits = end - frac - 1;   /* fractional digit count (remember '.') */
387         while(frac_digits < 9) {    /* this is frac of 10^9 */
388             val *= 10;
389             frac_digits++;
390         }
391     }
392     time_adj.tv.nsecs = (int)val;
393 }
394
395 static void
396 set_strict_time_adj(char *optarg_str_p)
397 {
398     char   *frac, *end;
399     long    val;
400     size_t  frac_digits;
401
402     if (!optarg_str_p)
403         return;
404
405     /* skip leading whitespace */
406     while (*optarg_str_p == ' ' || *optarg_str_p == '\t')
407         optarg_str_p++;
408
409     /*
410      * check for a negative adjustment
411      * A negative strict adjustment value is a flag
412      * to adjust all frames by the specifed delta time.
413      */
414     if (*optarg_str_p == '-') {
415         strict_time_adj.is_negative = 1;
416         optarg_str_p++;
417     }
418
419     /* collect whole number of seconds, if any */
420     if (*optarg_str_p == '.') {         /* only fractional (i.e., .5 is ok) */
421         val  = 0;
422         frac = optarg_str_p;
423     } else {
424         val = strtol(optarg_str_p, &frac, 10);
425         if (frac == NULL || frac == optarg_str_p
426             || val == LONG_MIN || val == LONG_MAX) {
427             fprintf(stderr, "editcap: \"%s\" isn't a valid time adjustment\n",
428                     optarg_str_p);
429             exit(1);
430         }
431         if (val < 0) {            /* implies '--' since we caught '-' above  */
432             fprintf(stderr, "editcap: \"%s\" isn't a valid time adjustment\n",
433                     optarg_str_p);
434             exit(1);
435         }
436     }
437     strict_time_adj.tv.secs = val;
438
439     /* now collect the partial seconds, if any */
440     if (*frac != '\0') {             /* chars left, so get fractional part */
441         val = strtol(&(frac[1]), &end, 10);
442         /* if more than 9 fractional digits truncate to 9 */
443         if ((end - &(frac[1])) > 9) {
444             frac[10] = 't'; /* 't' for truncate */
445             val = strtol(&(frac[1]), &end, 10);
446         }
447         if (*frac != '.' || end == NULL || end == frac || val < 0
448             || val > ONE_BILLION || val == LONG_MIN || val == LONG_MAX) {
449             fprintf(stderr, "editcap: \"%s\" isn't a valid time adjustment\n",
450                     optarg_str_p);
451             exit(1);
452         }
453     } else {
454         return;                     /* no fractional digits */
455     }
456
457     /* adjust fractional portion from fractional to numerator
458      * e.g., in "1.5" from 5 to 500000000 since .5*10^9 = 500000000 */
459     if (frac && end) {            /* both are valid */
460         frac_digits = end - frac - 1;   /* fractional digit count (remember '.') */
461         while(frac_digits < 9) {    /* this is frac of 10^9 */
462             val *= 10;
463             frac_digits++;
464         }
465     }
466     strict_time_adj.tv.nsecs = (int)val;
467 }
468
469 static void
470 set_rel_time(char *optarg_str_p)
471 {
472     char   *frac, *end;
473     long    val;
474     size_t  frac_digits;
475
476     if (!optarg_str_p)
477         return;
478
479     /* skip leading whitespace */
480     while (*optarg_str_p == ' ' || *optarg_str_p == '\t')
481         optarg_str_p++;
482
483     /* ignore negative adjustment  */
484     if (*optarg_str_p == '-')
485         optarg_str_p++;
486
487     /* collect whole number of seconds, if any */
488     if (*optarg_str_p == '.') {         /* only fractional (i.e., .5 is ok) */
489         val  = 0;
490         frac = optarg_str_p;
491     } else {
492         val = strtol(optarg_str_p, &frac, 10);
493         if (frac == NULL || frac == optarg_str_p
494             || val == LONG_MIN || val == LONG_MAX) {
495             fprintf(stderr, "1: editcap: \"%s\" isn't a valid rel time value\n",
496                     optarg_str_p);
497             exit(1);
498         }
499         if (val < 0) {            /* implies '--' since we caught '-' above  */
500             fprintf(stderr, "2: editcap: \"%s\" isn't a valid rel time value\n",
501                     optarg_str_p);
502             exit(1);
503         }
504     }
505     relative_time_window.secs = val;
506
507     /* now collect the partial seconds, if any */
508     if (*frac != '\0') {             /* chars left, so get fractional part */
509         val = strtol(&(frac[1]), &end, 10);
510         /* if more than 9 fractional digits truncate to 9 */
511         if ((end - &(frac[1])) > 9) {
512             frac[10] = 't'; /* 't' for truncate */
513             val = strtol(&(frac[1]), &end, 10);
514         }
515         if (*frac != '.' || end == NULL || end == frac || val < 0
516             || val > ONE_BILLION || val == LONG_MIN || val == LONG_MAX) {
517             fprintf(stderr, "3: editcap: \"%s\" isn't a valid rel time value\n",
518                     optarg_str_p);
519             exit(1);
520         }
521     } else {
522         return;                     /* no fractional digits */
523     }
524
525     /* adjust fractional portion from fractional to numerator
526      * e.g., in "1.5" from 5 to 500000000 since .5*10^9 = 500000000 */
527     if (frac && end) {            /* both are valid */
528         frac_digits = end - frac - 1;   /* fractional digit count (remember '.') */
529         while(frac_digits < 9) {    /* this is frac of 10^9 */
530             val *= 10;
531             frac_digits++;
532         }
533     }
534     relative_time_window.nsecs = (int)val;
535 }
536
537 static gboolean
538 is_duplicate(guint8* fd, guint32 len) {
539     int i;
540     md5_state_t ms;
541
542     /*Hint to ignore some bytes at the start of the frame for the digest calculation(-I option) */
543     guint32 new_len;
544     guint8 *new_fd;
545
546     new_fd  = &fd[ignored_bytes];
547     new_len = len - (ignored_bytes);
548
549     cur_dup_entry++;
550     if (cur_dup_entry >= dup_window)
551         cur_dup_entry = 0;
552
553     /* Calculate our digest */
554     md5_init(&ms);
555     md5_append(&ms, new_fd, new_len);
556     md5_finish(&ms, fd_hash[cur_dup_entry].digest);
557
558     fd_hash[cur_dup_entry].len = len;
559
560     /* Look for duplicates */
561     for (i = 0; i < dup_window; i++) {
562         if (i == cur_dup_entry)
563             continue;
564
565         if (fd_hash[i].len == fd_hash[cur_dup_entry].len
566             && memcmp(fd_hash[i].digest, fd_hash[cur_dup_entry].digest, 16) == 0) {
567             return TRUE;
568         }
569     }
570
571     return FALSE;
572 }
573
574 static gboolean
575 is_duplicate_rel_time(guint8* fd, guint32 len, const nstime_t *current) {
576     int i;
577     md5_state_t ms;
578
579     /*Hint to ignore some bytes at the start of the frame for the digest calculation(-I option) */
580     guint32 new_len;
581     guint8 *new_fd;
582
583     new_fd  = &fd[ignored_bytes];
584     new_len = len - (ignored_bytes);
585
586     cur_dup_entry++;
587     if (cur_dup_entry >= dup_window)
588         cur_dup_entry = 0;
589
590     /* Calculate our digest */
591     md5_init(&ms);
592     md5_append(&ms, new_fd, new_len);
593     md5_finish(&ms, fd_hash[cur_dup_entry].digest);
594
595     fd_hash[cur_dup_entry].len = len;
596     fd_hash[cur_dup_entry].time.secs = current->secs;
597     fd_hash[cur_dup_entry].time.nsecs = current->nsecs;
598
599     /*
600      * Look for relative time related duplicates.
601      * This is hopefully a reasonably efficient mechanism for
602      * finding duplicates by rel time in the fd_hash[] cache.
603      * We check starting from the most recently added hash
604      * entries and work backwards towards older packets.
605      * This approach allows the dup test to be terminated
606      * when the relative time of a cached entry is found to
607      * be beyond the dup time window.
608      *
609      * Of course this assumes that the input trace file is
610      * "well-formed" in the sense that the packet timestamps are
611      * in strict chronologically increasing order (which is NOT
612      * always the case!!).
613      *
614      * The fd_hash[] table was deliberately created large (1,000,000).
615      * Looking for time related duplicates in large trace files with
616      * non-fractional dup time window values can potentially take
617      * a long time to complete.
618      */
619
620     for (i = cur_dup_entry - 1;; i--) {
621         nstime_t delta;
622         int cmp;
623
624         if (i < 0)
625             i = dup_window - 1;
626
627         if (i == cur_dup_entry) {
628             /*
629              * We've decremented back to where we started.
630              * Check no more!
631              */
632             break;
633         }
634
635         if (nstime_is_unset(&(fd_hash[i].time))) {
636             /*
637              * We've decremented to an unused fd_hash[] entry.
638              * Check no more!
639              */
640             break;
641         }
642
643         nstime_delta(&delta, current, &fd_hash[i].time);
644
645         if (delta.secs < 0 || delta.nsecs < 0) {
646             /*
647              * A negative delta implies that the current packet
648              * has an absolute timestamp less than the cached packet
649              * that it is being compared to.  This is NOT a normal
650              * situation since trace files usually have packets in
651              * chronological order (oldest to newest).
652              *
653              * There are several possible ways to deal with this:
654              * 1. 'continue' dup checking with the next cached frame.
655              * 2. 'break' from looking for a duplicate of the current frame.
656              * 3. Take the absolute value of the delta and see if that
657              * falls within the specifed dup time window.
658              *
659              * Currently this code does option 1.  But it would pretty
660              * easy to add yet-another-editcap-option to select one of
661              * the other behaviors for dealing with out-of-sequence
662              * packets.
663              */
664             continue;
665         }
666
667         cmp = nstime_cmp(&delta, &relative_time_window);
668
669         if (cmp > 0) {
670             /*
671              * The delta time indicates that we are now looking at
672              * cached packets beyond the specified dup time window.
673              * Check no more!
674              */
675             break;
676         } else if (fd_hash[i].len == fd_hash[cur_dup_entry].len
677                    && memcmp(fd_hash[i].digest, fd_hash[cur_dup_entry].digest, 16) == 0) {
678             return TRUE;
679         }
680     }
681
682     return FALSE;
683 }
684
685 static void
686 print_usage(FILE *output)
687 {
688     fprintf(output, "\n");
689     fprintf(output, "Usage: editcap [options] ... <infile> <outfile> [ <packet#>[-<packet#>] ... ]\n");
690     fprintf(output, "\n");
691     fprintf(output, "<infile> and <outfile> must both be present.\n");
692     fprintf(output, "A single packet or a range of packets can be selected.\n");
693     fprintf(output, "\n");
694     fprintf(output, "Packet selection:\n");
695     fprintf(output, "  -r                     keep the selected packets; default is to delete them.\n");
696     fprintf(output, "  -A <start time>        only output packets whose timestamp is after (or equal\n");
697     fprintf(output, "                         to) the given time (format as YYYY-MM-DD hh:mm:ss).\n");
698     fprintf(output, "  -B <stop time>         only output packets whose timestamp is before the\n");
699     fprintf(output, "                         given time (format as YYYY-MM-DD hh:mm:ss).\n");
700     fprintf(output, "\n");
701     fprintf(output, "Duplicate packet removal:\n");
702     fprintf(output, "  -d                     remove packet if duplicate (window == %d).\n", DEFAULT_DUP_DEPTH);
703     fprintf(output, "  -D <dup window>        remove packet if duplicate; configurable <dup window>\n");
704     fprintf(output, "                         Valid <dup window> values are 0 to %d.\n", MAX_DUP_DEPTH);
705     fprintf(output, "                         NOTE: A <dup window> of 0 with -v (verbose option) is\n");
706     fprintf(output, "                         useful to print MD5 hashes.\n");
707     fprintf(output, "  -w <dup time window>   remove packet if duplicate packet is found EQUAL TO OR\n");
708     fprintf(output, "                         LESS THAN <dup time window> prior to current packet.\n");
709     fprintf(output, "                         A <dup time window> is specified in relative seconds\n");
710     fprintf(output, "                         (e.g. 0.000001).\n");
711     fprintf(output, "\n");
712     fprintf(output, "  -I <bytes to ignore>   ignore the specified bytes at the beginning of\n");
713     fprintf(output, "                         the frame during MD5 hash calculation\n");
714     fprintf(output, "                         Useful to remove duplicated packets taken on\n");
715     fprintf(output, "                         several routers(differents mac addresses for \n");
716     fprintf(output, "                         example)\n");
717     fprintf(output, "                         e.g. -I 26 in case of Ether/IP/ will ignore \n");
718     fprintf(output, "                         ether(14) and IP header(20 - 4(src ip) - 4(dst ip)).\n");
719     fprintf(output, "\n");
720     fprintf(output, "           NOTE: The use of the 'Duplicate packet removal' options with\n");
721     fprintf(output, "           other editcap options except -v may not always work as expected.\n");
722     fprintf(output, "           Specifically the -r, -t or -S options will very likely NOT have the\n");
723     fprintf(output, "           desired effect if combined with the -d, -D or -w.\n");
724     fprintf(output, "\n");
725     fprintf(output, "Packet manipulation:\n");
726     fprintf(output, "  -s <snaplen>           truncate each packet to max. <snaplen> bytes of data.\n");
727     fprintf(output, "  -C [offset:]<choplen>  chop each packet by <choplen> bytes. Positive values\n");
728     fprintf(output, "                         chop at the packet beginning, negative values at the\n");
729     fprintf(output, "                         packet end. If an optional offset precedes the length,\n");
730     fprintf(output, "                         then the bytes chopped will be offset from that value.\n");
731     fprintf(output, "                         Positive offsets are from the packet beginning,\n");
732     fprintf(output, "                         negative offsets are from the packet end. You can use\n");
733     fprintf(output, "                         this option more than once, allowing up to 2 chopping\n");
734     fprintf(output, "                         regions within a packet provided that at least 1\n");
735     fprintf(output, "                         choplen is positive and at least 1 is negative.\n");
736     fprintf(output, "  -L                     adjust the frame length when chopping and/or snapping\n");
737     fprintf(output, "  -t <time adjustment>   adjust the timestamp of each packet;\n");
738     fprintf(output, "                         <time adjustment> is in relative seconds (e.g. -0.5).\n");
739     fprintf(output, "  -S <strict adjustment> adjust timestamp of packets if necessary to insure\n");
740     fprintf(output, "                         strict chronological increasing order. The <strict\n");
741     fprintf(output, "                         adjustment> is specified in relative seconds with\n");
742     fprintf(output, "                         values of 0 or 0.000001 being the most reasonable.\n");
743     fprintf(output, "                         A negative adjustment value will modify timestamps so\n");
744     fprintf(output, "                         that each packet's delta time is the absolute value\n");
745     fprintf(output, "                         of the adjustment specified. A value of -0 will set\n");
746     fprintf(output, "                         all packets to the timestamp of the first packet.\n");
747     fprintf(output, "  -E <error probability> set the probability (between 0.0 and 1.0 incl.) that\n");
748     fprintf(output, "                         a particular packet byte will be randomly changed.\n");
749     fprintf(output, "\n");
750     fprintf(output, "Output File(s):\n");
751     fprintf(output, "  -c <packets per file>  split the packet output to different files based on\n");
752     fprintf(output, "                         uniform packet counts with a maximum of\n");
753     fprintf(output, "                         <packets per file> each.\n");
754     fprintf(output, "  -i <seconds per file>  split the packet output to different files based on\n");
755     fprintf(output, "                         uniform time intervals with a maximum of\n");
756     fprintf(output, "                         <seconds per file> each.\n");
757     fprintf(output, "  -F <capture type>      set the output file type; default is pcapng. An empty\n");
758     fprintf(output, "                         \"-F\" option will list the file types.\n");
759     fprintf(output, "  -T <encap type>        set the output file encapsulation type; default is the\n");
760     fprintf(output, "                         same as the input file. An empty \"-T\" option will\n");
761     fprintf(output, "                         list the encapsulation types.\n");
762     fprintf(output, "\n");
763     fprintf(output, "Miscellaneous:\n");
764     fprintf(output, "  -h                     display this help and exit.\n");
765     fprintf(output, "  -v                     verbose output.\n");
766     fprintf(output, "                         If -v is used with any of the 'Duplicate Packet\n");
767     fprintf(output, "                         Removal' options (-d, -D or -w) then Packet lengths\n");
768     fprintf(output, "                         and MD5 hashes are printed to standard-error.\n");
769     fprintf(output, "\n");
770 }
771
772 struct string_elem {
773     const char *sstr;   /* The short string */
774     const char *lstr;   /* The long string */
775 };
776
777 static gint
778 string_compare(gconstpointer a, gconstpointer b)
779 {
780     return strcmp(((const struct string_elem *)a)->sstr,
781         ((const struct string_elem *)b)->sstr);
782 }
783
784 static gint
785 string_nat_compare(gconstpointer a, gconstpointer b)
786 {
787     return ws_ascii_strnatcmp(((const struct string_elem *)a)->sstr,
788         ((const struct string_elem *)b)->sstr);
789 }
790
791 static void
792 string_elem_print(gpointer data, gpointer not_used _U_)
793 {
794     fprintf(stderr, "    %s - %s\n",
795         ((struct string_elem *)data)->sstr,
796         ((struct string_elem *)data)->lstr);
797 }
798
799 static void
800 list_capture_types(void) {
801     int i;
802     struct string_elem *captypes;
803     GSList *list = NULL;
804
805     captypes = g_new(struct string_elem,WTAP_NUM_FILE_TYPES_SUBTYPES);
806     fprintf(stderr, "editcap: The available capture file types for the \"-F\" flag are:\n");
807     for (i = 0; i < WTAP_NUM_FILE_TYPES_SUBTYPES; i++) {
808         if (wtap_dump_can_open(i)) {
809             captypes[i].sstr = wtap_file_type_subtype_short_string(i);
810             captypes[i].lstr = wtap_file_type_subtype_string(i);
811             list = g_slist_insert_sorted(list, &captypes[i], string_compare);
812         }
813     }
814     g_slist_foreach(list, string_elem_print, NULL);
815     g_slist_free(list);
816     g_free(captypes);
817 }
818
819 static void
820 list_encap_types(void) {
821     int i;
822     struct string_elem *encaps;
823     GSList *list = NULL;
824
825     encaps = (struct string_elem *)g_malloc(sizeof(struct string_elem) * WTAP_NUM_ENCAP_TYPES);
826     fprintf(stderr, "editcap: The available encapsulation types for the \"-T\" flag are:\n");
827     for (i = 0; i < WTAP_NUM_ENCAP_TYPES; i++) {
828         encaps[i].sstr = wtap_encap_short_string(i);
829         if (encaps[i].sstr != NULL) {
830             encaps[i].lstr = wtap_encap_string(i);
831             list = g_slist_insert_sorted(list, &encaps[i], string_nat_compare);
832         }
833     }
834     g_slist_foreach(list, string_elem_print, NULL);
835     g_slist_free(list);
836     g_free(encaps);
837 }
838
839 #ifdef HAVE_PLUGINS
840 /*
841  *  Don't report failures to load plugins because most (non-wiretap) plugins
842  *  *should* fail to load (because we're not linked against libwireshark and
843  *  dissector plugins need libwireshark).
844  */
845 static void
846 failure_message(const char *msg_format _U_, va_list ap _U_)
847 {
848 }
849 #endif
850
851 static void
852 get_editcap_compiled_info(GString *str)
853 {
854   /* LIBZ */
855   g_string_append(str, ", ");
856 #ifdef HAVE_LIBZ
857   g_string_append(str, "with libz ");
858 #ifdef ZLIB_VERSION
859   g_string_append(str, ZLIB_VERSION);
860 #else /* ZLIB_VERSION */
861   g_string_append(str, "(version unknown)");
862 #endif /* ZLIB_VERSION */
863 #else /* HAVE_LIBZ */
864   g_string_append(str, "without libz");
865 #endif /* HAVE_LIBZ */
866 }
867
868 static void
869 get_editcap_runtime_info(GString *str)
870 {
871   /* zlib */
872 #if defined(HAVE_LIBZ) && !defined(_WIN32)
873   g_string_append_printf(str, ", with libz %s", zlibVersion());
874 #endif
875 }
876
877 int
878 main(int argc, char *argv[])
879 {
880     GString      *comp_info_str;
881     GString      *runtime_info_str;
882     wtap         *wth;
883     int           i, j, err;
884     gchar        *err_info;
885     int           opt;
886     static const struct option long_options[] = {
887         {(char *)"help", no_argument, NULL, 'h'},
888         {(char *)"version", no_argument, NULL, 'V'},
889         {0, 0, 0, 0 }
890     };
891
892     char         *p;
893     guint32       snaplen            = 0; /* No limit               */
894     chop_t        chop               = {0, 0, 0, 0, 0, 0}; /* No chop */
895     gboolean      adjlen             = FALSE;
896     wtap_dumper  *pdh                = NULL;
897     unsigned int  count              = 1;
898     unsigned int  duplicate_count    = 0;
899     gint64        data_offset;
900     int           err_type;
901     guint8       *buf;
902     guint32       read_count         = 0;
903     int           split_packet_count = 0;
904     int           written_count      = 0;
905     char         *filename           = NULL;
906     gboolean      ts_okay;
907     int           secs_per_block     = 0;
908     int           block_cnt          = 0;
909     nstime_t      block_start;
910     gchar        *fprefix            = NULL;
911     gchar        *fsuffix            = NULL;
912
913     const struct wtap_pkthdr    *phdr;
914     struct wtap_pkthdr           snap_phdr;
915     wtapng_iface_descriptions_t *idb_inf;
916     wtapng_section_t            *shb_hdr;
917
918 #ifdef HAVE_PLUGINS
919     char* init_progfile_dir_error;
920 #endif
921
922 #ifdef _WIN32
923     arg_list_utf_16to8(argc, argv);
924     create_app_running_mutex();
925 #endif /* _WIN32 */
926
927     /* Assemble the compile-time version information string */
928     comp_info_str = g_string_new("Compiled ");
929     get_compiled_version_info(comp_info_str, NULL, get_editcap_compiled_info);
930
931     /* Assemble the run-time version information string */
932     runtime_info_str = g_string_new("Running ");
933     get_runtime_version_info(runtime_info_str, get_editcap_runtime_info);
934
935     /* Add it to the information to be reported on a crash. */
936     ws_add_crash_info("Editcap (Wireshark) %s\n"
937          "\n"
938          "%s"
939          "\n"
940          "%s",
941       get_ws_vcs_version_info(), comp_info_str->str, runtime_info_str->str);
942
943     /*
944      * Get credential information for later use.
945      */
946     init_process_policies();
947     init_open_routines();
948
949 #ifdef HAVE_PLUGINS
950     /* Register wiretap plugins */
951     if ((init_progfile_dir_error = init_progfile_dir(argv[0], main))) {
952         g_warning("editcap: init_progfile_dir(): %s", init_progfile_dir_error);
953         g_free(init_progfile_dir_error);
954     } else {
955         /* Register all the plugin types we have. */
956         wtap_register_plugin_types(); /* Types known to libwiretap */
957
958         init_report_err(failure_message,NULL,NULL,NULL);
959
960         /* Scan for plugins.  This does *not* call their registration routines;
961            that's done later. */
962         scan_plugins();
963
964         /* Register all libwiretap plugin modules. */
965         register_all_wiretap_modules();
966     }
967 #endif
968
969     /* Process the options */
970     while ((opt = getopt_long(argc, argv, "A:B:c:C:dD:E:F:hi:I:Lrs:S:t:T:vVw:", long_options, NULL)) != -1) {
971         switch (opt) {
972         case 'A':
973         {
974             struct tm starttm;
975
976             memset(&starttm,0,sizeof(struct tm));
977
978             if (!strptime(optarg,"%Y-%m-%d %T", &starttm)) {
979                 fprintf(stderr, "editcap: \"%s\" isn't a valid time format\n\n",
980                         optarg);
981                 exit(1);
982             }
983
984             check_startstop = TRUE;
985             starttm.tm_isdst = -1;
986
987             starttime = mktime(&starttm);
988             break;
989         }
990
991         case 'B':
992         {
993             struct tm stoptm;
994
995             memset(&stoptm,0,sizeof(struct tm));
996
997             if (!strptime(optarg,"%Y-%m-%d %T", &stoptm)) {
998                 fprintf(stderr, "editcap: \"%s\" isn't a valid time format\n\n",
999                         optarg);
1000                 exit(1);
1001             }
1002             check_startstop = TRUE;
1003             stoptm.tm_isdst = -1;
1004             stoptime = mktime(&stoptm);
1005             break;
1006         }
1007
1008         case 'c':
1009             split_packet_count = (int)strtol(optarg, &p, 10);
1010             if (p == optarg || *p != '\0') {
1011                 fprintf(stderr, "editcap: \"%s\" isn't a valid packet count\n",
1012                         optarg);
1013                 exit(1);
1014             }
1015             if (split_packet_count <= 0) {
1016                 fprintf(stderr, "editcap: \"%d\" packet count must be larger than zero\n",
1017                         split_packet_count);
1018                 exit(1);
1019             }
1020             break;
1021
1022         case 'C':
1023         {
1024             int choplen = 0, chopoff = 0;
1025
1026             switch (sscanf(optarg, "%d:%d", &chopoff, &choplen)) {
1027             case 1: /* only the chop length was specififed */
1028                 choplen = chopoff;
1029                 chopoff = 0;
1030                 break;
1031
1032             case 2: /* both an offset and chop length was specified */
1033                 break;
1034
1035             default:
1036                 fprintf(stderr, "editcap: \"%s\" isn't a valid chop length or offset:length\n",
1037                         optarg);
1038                 exit(1);
1039                 break;
1040             }
1041
1042             if (choplen > 0) {
1043                 chop.len_begin += choplen;
1044                 if (chopoff > 0)
1045                     chop.off_begin_pos += chopoff;
1046                 else
1047                     chop.off_begin_neg += chopoff;
1048             } else if (choplen < 0) {
1049                 chop.len_end += choplen;
1050                 if (chopoff > 0)
1051                     chop.off_end_pos += chopoff;
1052                 else
1053                     chop.off_end_neg += chopoff;
1054             }
1055             break;
1056         }
1057
1058         case 'd':
1059             dup_detect = TRUE;
1060             dup_detect_by_time = FALSE;
1061             dup_window = DEFAULT_DUP_DEPTH;
1062             break;
1063
1064         case 'D':
1065             dup_detect = TRUE;
1066             dup_detect_by_time = FALSE;
1067             dup_window = (int)strtol(optarg, &p, 10);
1068             if (p == optarg || *p != '\0') {
1069                 fprintf(stderr, "editcap: \"%s\" isn't a valid duplicate window value\n",
1070                         optarg);
1071                 exit(1);
1072             }
1073             if (dup_window < 0 || dup_window > MAX_DUP_DEPTH) {
1074                 fprintf(stderr, "editcap: \"%d\" duplicate window value must be between 0 and %d inclusive.\n",
1075                         dup_window, MAX_DUP_DEPTH);
1076                 exit(1);
1077             }
1078             break;
1079
1080         case 'E':
1081             err_prob = strtod(optarg, &p);
1082             if (p == optarg || err_prob < 0.0 || err_prob > 1.0) {
1083                 fprintf(stderr, "editcap: probability \"%s\" must be between 0.0 and 1.0\n",
1084                         optarg);
1085                 exit(1);
1086             }
1087             srand( (unsigned int) (time(NULL) + getpid()) );
1088             break;
1089
1090         case 'F':
1091             out_file_type_subtype = wtap_short_string_to_file_type_subtype(optarg);
1092             if (out_file_type_subtype < 0) {
1093                 fprintf(stderr, "editcap: \"%s\" isn't a valid capture file type\n\n",
1094                         optarg);
1095                 list_capture_types();
1096                 exit(1);
1097             }
1098             break;
1099
1100         case 'h':
1101             printf("Editcap (Wireshark) %s\n"
1102                    "Edit and/or translate the format of capture files.\n"
1103                    "See http://www.wireshark.org for more information.\n",
1104                get_ws_vcs_version_info());
1105             print_usage(stdout);
1106             exit(0);
1107             break;
1108
1109         case 'i': /* break capture file based on time interval */
1110             secs_per_block = atoi(optarg);
1111             if (secs_per_block <= 0) {
1112                 fprintf(stderr, "editcap: \"%s\" isn't a valid time interval\n\n",
1113                         optarg);
1114                 exit(1);
1115             }
1116             break;
1117
1118         case 'I': /* ignored_bytes at the beginning of the frame for duplications removal */
1119             ignored_bytes = atoi(optarg);
1120             if(ignored_bytes <= 0) {
1121                 fprintf(stderr, "editcap: \"%s\" isn't a valid number of bytes to ignore\n", optarg);
1122                 exit(1);
1123             }
1124             break;
1125
1126         case 'L':
1127             adjlen = TRUE;
1128             break;
1129
1130         case 'r':
1131             keep_em = !keep_em;  /* Just invert */
1132             break;
1133
1134         case 's':
1135             snaplen = (guint32)strtol(optarg, &p, 10);
1136             if (p == optarg || *p != '\0') {
1137                 fprintf(stderr, "editcap: \"%s\" isn't a valid snapshot length\n",
1138                         optarg);
1139                 exit(1);
1140             }
1141             break;
1142
1143         case 'S':
1144             set_strict_time_adj(optarg);
1145             do_strict_time_adjustment = TRUE;
1146             break;
1147
1148         case 't':
1149             set_time_adjustment(optarg);
1150             break;
1151
1152         case 'T':
1153             out_frame_type = wtap_short_string_to_encap(optarg);
1154             if (out_frame_type < 0) {
1155                 fprintf(stderr, "editcap: \"%s\" isn't a valid encapsulation type\n\n",
1156                         optarg);
1157                 list_encap_types();
1158                 exit(1);
1159             }
1160             break;
1161
1162         case 'v':
1163             verbose = !verbose;  /* Just invert */
1164             break;
1165
1166         case 'V':
1167             show_version("Editcap (Wireshark)", comp_info_str, runtime_info_str);
1168             g_string_free(comp_info_str, TRUE);
1169             g_string_free(runtime_info_str, TRUE);
1170             exit(0);
1171             break;
1172
1173         case 'w':
1174             dup_detect = FALSE;
1175             dup_detect_by_time = TRUE;
1176             dup_window = MAX_DUP_DEPTH;
1177             set_rel_time(optarg);
1178             break;
1179
1180         case '?':              /* Bad options if GNU getopt */
1181             switch(optopt) {
1182             case'F':
1183                 list_capture_types();
1184                 break;
1185             case'T':
1186                 list_encap_types();
1187                 break;
1188             default:
1189                 print_usage(stderr);
1190                 break;
1191             }
1192             exit(1);
1193             break;
1194         }
1195     }
1196
1197 #ifdef DEBUG
1198     fprintf(stderr, "Optind = %i, argc = %i\n", optind, argc);
1199 #endif
1200
1201     if ((argc - optind) < 1) {
1202         print_usage(stderr);
1203         exit(1);
1204     }
1205
1206     if (check_startstop && !stoptime) {
1207         struct tm stoptm;
1208
1209         /* XXX: will work until 2035 */
1210         memset(&stoptm,0,sizeof(struct tm));
1211         stoptm.tm_year = 135;
1212         stoptm.tm_mday = 31;
1213         stoptm.tm_mon = 11;
1214
1215         stoptime = mktime(&stoptm);
1216     }
1217
1218     nstime_set_unset(&block_start);
1219
1220     if (starttime > stoptime) {
1221         fprintf(stderr, "editcap: start time is after the stop time\n");
1222         exit(1);
1223     }
1224
1225     if (split_packet_count > 0 && secs_per_block > 0) {
1226         fprintf(stderr, "editcap: can't split on both packet count and time interval\n");
1227         fprintf(stderr, "editcap: at the same time\n");
1228         exit(1);
1229     }
1230
1231     wth = wtap_open_offline(argv[optind], WTAP_TYPE_AUTO, &err, &err_info, FALSE);
1232
1233     if (!wth) {
1234         fprintf(stderr, "editcap: Can't open %s: %s\n", argv[optind],
1235                 wtap_strerror(err));
1236         if (err_info != NULL) {
1237             fprintf(stderr, "(%s)\n", err_info);
1238             g_free(err_info);
1239         }
1240         exit(2);
1241     }
1242
1243     if (verbose) {
1244         fprintf(stderr, "File %s is a %s capture file.\n", argv[optind],
1245                 wtap_file_type_subtype_string(wtap_file_type_subtype(wth)));
1246     }
1247
1248     shb_hdr = wtap_file_get_shb_info(wth);
1249     idb_inf = wtap_file_get_idb_info(wth);
1250
1251     /*
1252      * Now, process the rest, if any ... we only write if there is an extra
1253      * argument or so ...
1254      */
1255
1256     if ((argc - optind) >= 2) {
1257         if (out_frame_type == -2)
1258             out_frame_type = wtap_file_encap(wth);
1259
1260         for (i = optind + 2; i < argc; i++)
1261             if (add_selection(argv[i]) == FALSE)
1262                 break;
1263
1264         if (dup_detect || dup_detect_by_time) {
1265             for (i = 0; i < dup_window; i++) {
1266                 memset(&fd_hash[i].digest, 0, 16);
1267                 fd_hash[i].len = 0;
1268                 nstime_set_unset(&fd_hash[i].time);
1269             }
1270         }
1271
1272         while (wtap_read(wth, &err, &err_info, &data_offset)) {
1273             read_count++;
1274
1275             phdr = wtap_phdr(wth);
1276
1277             if (read_count == 1) {  /* the first packet */
1278                 if (split_packet_count > 0 || secs_per_block > 0) {
1279                     if (!fileset_extract_prefix_suffix(argv[optind+1], &fprefix, &fsuffix))
1280                         exit(2);
1281
1282                     filename = fileset_get_filename_by_pattern(block_cnt++, phdr, fprefix, fsuffix);
1283                 } else {
1284                     filename = g_strdup(argv[optind+1]);
1285                 }
1286                 g_assert(filename);
1287
1288                 /* If we don't have an application name add Editcap */
1289                 if (shb_hdr->shb_user_appl == NULL) {
1290                     shb_hdr->shb_user_appl = g_strdup("Editcap " VERSION);
1291                 }
1292
1293                 pdh = wtap_dump_open_ng(filename, out_file_type_subtype, out_frame_type,
1294                                         snaplen ? MIN(snaplen, wtap_snapshot_length(wth)) : wtap_snapshot_length(wth),
1295                                         FALSE /* compressed */, shb_hdr, idb_inf, &err);
1296
1297                 if (pdh == NULL) {
1298                     fprintf(stderr, "editcap: Can't open or create %s: %s\n",
1299                             filename, wtap_strerror(err));
1300                     exit(2);
1301                 }
1302             }
1303
1304             buf = wtap_buf_ptr(wth);
1305
1306             /*
1307              * Not all packets have time stamps. Only process the time
1308              * stamp if we have one.
1309              */
1310             if (phdr->presence_flags & WTAP_HAS_TS) {
1311                 if (nstime_is_unset(&block_start)) {
1312                     block_start.secs = phdr->ts.secs;
1313                     block_start.nsecs = phdr->ts.nsecs;
1314                 }
1315
1316                 if (secs_per_block > 0) {
1317                     while ((phdr->ts.secs - block_start.secs >  secs_per_block)
1318                            || (phdr->ts.secs - block_start.secs == secs_per_block
1319                                && phdr->ts.nsecs >= block_start.nsecs )) { /* time for the next file */
1320
1321                         if (!wtap_dump_close(pdh, &err)) {
1322                             fprintf(stderr, "editcap: Error writing to %s: %s\n",
1323                                     filename, wtap_strerror(err));
1324                             exit(2);
1325                         }
1326                         block_start.secs = block_start.secs +  secs_per_block; /* reset for next interval */
1327                         g_free(filename);
1328                         filename = fileset_get_filename_by_pattern(block_cnt++, phdr, fprefix, fsuffix);
1329                         g_assert(filename);
1330
1331                         if (verbose)
1332                             fprintf(stderr, "Continuing writing in file %s\n", filename);
1333
1334                         pdh = wtap_dump_open_ng(filename, out_file_type_subtype, out_frame_type,
1335                                                 snaplen ? MIN(snaplen, wtap_snapshot_length(wth)) : wtap_snapshot_length(wth),
1336                                                 FALSE /* compressed */, shb_hdr, idb_inf, &err);
1337
1338                         if (pdh == NULL) {
1339                             fprintf(stderr, "editcap: Can't open or create %s: %s\n",
1340                                     filename, wtap_strerror(err));
1341                             exit(2);
1342                         }
1343                     }
1344                 }
1345             }
1346
1347             if (split_packet_count > 0) {
1348                 /* time for the next file? */
1349                 if (written_count > 0 && written_count % split_packet_count == 0) {
1350                     if (!wtap_dump_close(pdh, &err)) {
1351                         fprintf(stderr, "editcap: Error writing to %s: %s\n",
1352                                 filename, wtap_strerror(err));
1353                         exit(2);
1354                     }
1355
1356                     g_free(filename);
1357                     filename = fileset_get_filename_by_pattern(block_cnt++, phdr, fprefix, fsuffix);
1358                     g_assert(filename);
1359
1360                     if (verbose)
1361                         fprintf(stderr, "Continuing writing in file %s\n", filename);
1362
1363                     pdh = wtap_dump_open_ng(filename, out_file_type_subtype, out_frame_type,
1364                                             snaplen ? MIN(snaplen, wtap_snapshot_length(wth)) : wtap_snapshot_length(wth),
1365                                             FALSE /* compressed */, shb_hdr, idb_inf, &err);
1366                     if (pdh == NULL) {
1367                         fprintf(stderr, "editcap: Can't open or create %s: %s\n",
1368                                 filename, wtap_strerror(err));
1369                         exit(2);
1370                     }
1371                 }
1372             }
1373
1374             if (check_startstop) {
1375                 /*
1376                  * Is the packet in the selected timeframe?
1377                  * If the packet has no time stamp, the answer is "no".
1378                  */
1379                 if (phdr->presence_flags & WTAP_HAS_TS)
1380                     ts_okay = (phdr->ts.secs >= starttime) && (phdr->ts.secs < stoptime);
1381                 else
1382                     ts_okay = FALSE;
1383             } else {
1384                 /*
1385                  * No selected timeframe, so all packets are "in the
1386                  * selected timeframe".
1387                  */
1388                 ts_okay = TRUE;
1389             }
1390
1391             if (ts_okay && ((!selected(count) && !keep_em)
1392                             || (selected(count) && keep_em))) {
1393
1394                 if (verbose && !dup_detect && !dup_detect_by_time)
1395                     fprintf(stderr, "Packet: %u\n", count);
1396
1397                 /* We simply write it, perhaps after truncating it; we could
1398                  * do other things, like modify it. */
1399
1400                 phdr = wtap_phdr(wth);
1401
1402                 if (snaplen != 0) {
1403                     if (phdr->caplen > snaplen) {
1404                         snap_phdr = *phdr;
1405                         snap_phdr.caplen = snaplen;
1406                         phdr = &snap_phdr;
1407                     }
1408                     if (adjlen && phdr->len > snaplen) {
1409                         snap_phdr = *phdr;
1410                         snap_phdr.len = snaplen;
1411                         phdr = &snap_phdr;
1412                     }
1413                 }
1414
1415                 /* CHOP */
1416                 snap_phdr = *phdr;
1417                 handle_chopping(chop, &snap_phdr, phdr, &buf, adjlen);
1418                 phdr = &snap_phdr;
1419
1420                 if (phdr->presence_flags & WTAP_HAS_TS) {
1421                     /* Do we adjust timestamps to ensure strict chronological
1422                      * order? */
1423                     if (do_strict_time_adjustment) {
1424                         if (previous_time.secs || previous_time.nsecs) {
1425                             if (!strict_time_adj.is_negative) {
1426                                 nstime_t current;
1427                                 nstime_t delta;
1428
1429                                 current.secs = phdr->ts.secs;
1430                                 current.nsecs = phdr->ts.nsecs;
1431
1432                                 nstime_delta(&delta, &current, &previous_time);
1433
1434                                 if (delta.secs < 0 || delta.nsecs < 0) {
1435                                     /*
1436                                      * A negative delta indicates that the current packet
1437                                      * has an absolute timestamp less than the previous packet
1438                                      * that it is being compared to.  This is NOT a normal
1439                                      * situation since trace files usually have packets in
1440                                      * chronological order (oldest to newest).
1441                                      */
1442                                     /* fprintf(stderr, "++out of order, need to adjust this packet!\n"); */
1443                                     snap_phdr = *phdr;
1444                                     snap_phdr.ts.secs = previous_time.secs + strict_time_adj.tv.secs;
1445                                     snap_phdr.ts.nsecs = previous_time.nsecs;
1446                                     if (snap_phdr.ts.nsecs + strict_time_adj.tv.nsecs > ONE_BILLION) {
1447                                         /* carry */
1448                                         snap_phdr.ts.secs++;
1449                                         snap_phdr.ts.nsecs += strict_time_adj.tv.nsecs - ONE_BILLION;
1450                                     } else {
1451                                         snap_phdr.ts.nsecs += strict_time_adj.tv.nsecs;
1452                                     }
1453                                     phdr = &snap_phdr;
1454                                 }
1455                             } else {
1456                                 /*
1457                                  * A negative strict time adjustment is requested.
1458                                  * Unconditionally set each timestamp to previous
1459                                  * packet's timestamp plus delta.
1460                                  */
1461                                 snap_phdr = *phdr;
1462                                 snap_phdr.ts.secs = previous_time.secs + strict_time_adj.tv.secs;
1463                                 snap_phdr.ts.nsecs = previous_time.nsecs;
1464                                 if (snap_phdr.ts.nsecs + strict_time_adj.tv.nsecs > ONE_BILLION) {
1465                                     /* carry */
1466                                     snap_phdr.ts.secs++;
1467                                     snap_phdr.ts.nsecs += strict_time_adj.tv.nsecs - ONE_BILLION;
1468                                 } else {
1469                                     snap_phdr.ts.nsecs += strict_time_adj.tv.nsecs;
1470                                 }
1471                                 phdr = &snap_phdr;
1472                             }
1473                         }
1474                         previous_time.secs = phdr->ts.secs;
1475                         previous_time.nsecs = phdr->ts.nsecs;
1476                     }
1477
1478                     /* assume that if the frame's tv_sec is 0, then
1479                      * the timestamp isn't supported */
1480                     if (phdr->ts.secs > 0 && time_adj.tv.secs != 0) {
1481                         snap_phdr = *phdr;
1482                         if (time_adj.is_negative)
1483                             snap_phdr.ts.secs -= time_adj.tv.secs;
1484                         else
1485                             snap_phdr.ts.secs += time_adj.tv.secs;
1486                         phdr = &snap_phdr;
1487                     }
1488
1489                     /* assume that if the frame's tv_sec is 0, then
1490                      * the timestamp isn't supported */
1491                     if (phdr->ts.secs > 0 && time_adj.tv.nsecs != 0) {
1492                         snap_phdr = *phdr;
1493                         if (time_adj.is_negative) { /* subtract */
1494                             if (snap_phdr.ts.nsecs < time_adj.tv.nsecs) { /* borrow */
1495                                 snap_phdr.ts.secs--;
1496                                 snap_phdr.ts.nsecs += ONE_BILLION;
1497                             }
1498                             snap_phdr.ts.nsecs -= time_adj.tv.nsecs;
1499                         } else {                  /* add */
1500                             if (snap_phdr.ts.nsecs + time_adj.tv.nsecs > ONE_BILLION) {
1501                                 /* carry */
1502                                 snap_phdr.ts.secs++;
1503                                 snap_phdr.ts.nsecs += time_adj.tv.nsecs - ONE_BILLION;
1504                             } else {
1505                                 snap_phdr.ts.nsecs += time_adj.tv.nsecs;
1506                             }
1507                         }
1508                         phdr = &snap_phdr;
1509                     }
1510                 }
1511
1512                 /* suppress duplicates by packet window */
1513                 if (dup_detect) {
1514                     if (is_duplicate(buf, phdr->caplen)) {
1515                         if (verbose) {
1516                             fprintf(stderr, "Skipped: %u, Len: %u, MD5 Hash: ",
1517                                     count, phdr->caplen);
1518                             for (i = 0; i < 16; i++)
1519                                 fprintf(stderr, "%02x",
1520                                         (unsigned char)fd_hash[cur_dup_entry].digest[i]);
1521                             fprintf(stderr, "\n");
1522                         }
1523                         duplicate_count++;
1524                         count++;
1525                         continue;
1526                     } else {
1527                         if (verbose) {
1528                             fprintf(stderr, "Packet: %u, Len: %u, MD5 Hash: ",
1529                                     count, phdr->caplen);
1530                             for (i = 0; i < 16; i++)
1531                                 fprintf(stderr, "%02x",
1532                                         (unsigned char)fd_hash[cur_dup_entry].digest[i]);
1533                             fprintf(stderr, "\n");
1534                         }
1535                     }
1536                 }
1537
1538                 if (phdr->presence_flags & WTAP_HAS_TS) {
1539                     /* suppress duplicates by time window */
1540                     if (dup_detect_by_time) {
1541                         nstime_t current;
1542
1543                         current.secs  = phdr->ts.secs;
1544                         current.nsecs = phdr->ts.nsecs;
1545
1546                         if (is_duplicate_rel_time(buf, phdr->caplen, &current)) {
1547                             if (verbose) {
1548                                 fprintf(stderr, "Skipped: %u, Len: %u, MD5 Hash: ",
1549                                         count, phdr->caplen);
1550                                 for (i = 0; i < 16; i++)
1551                                     fprintf(stderr, "%02x",
1552                                             (unsigned char)fd_hash[cur_dup_entry].digest[i]);
1553                                 fprintf(stderr, "\n");
1554                             }
1555                             duplicate_count++;
1556                             count++;
1557                             continue;
1558                         } else {
1559                             if (verbose) {
1560                                 fprintf(stderr, "Packet: %u, Len: %u, MD5 Hash: ",
1561                                         count, phdr->caplen);
1562                                 for (i = 0; i < 16; i++)
1563                                     fprintf(stderr, "%02x",
1564                                             (unsigned char)fd_hash[cur_dup_entry].digest[i]);
1565                                 fprintf(stderr, "\n");
1566                             }
1567                         }
1568                     }
1569                 }
1570
1571                 /* Random error mutation */
1572                 if (err_prob > 0.0) {
1573                     int real_data_start = 0;
1574
1575                     /* Protect non-protocol data */
1576                     if (wtap_file_type_subtype(wth) == WTAP_FILE_TYPE_SUBTYPE_CATAPULT_DCT2000)
1577                         real_data_start = find_dct2000_real_data(buf);
1578
1579                     for (i = real_data_start; i < (int) phdr->caplen; i++) {
1580                         if (rand() <= err_prob * RAND_MAX) {
1581                             err_type = rand() / (RAND_MAX / ERR_WT_TOTAL + 1);
1582
1583                             if (err_type < ERR_WT_BIT) {
1584                                 buf[i] ^= 1 << (rand() / (RAND_MAX / 8 + 1));
1585                                 err_type = ERR_WT_TOTAL;
1586                             } else {
1587                                 err_type -= ERR_WT_BYTE;
1588                             }
1589
1590                             if (err_type < ERR_WT_BYTE) {
1591                                 buf[i] = rand() / (RAND_MAX / 255 + 1);
1592                                 err_type = ERR_WT_TOTAL;
1593                             } else {
1594                                 err_type -= ERR_WT_BYTE;
1595                             }
1596
1597                             if (err_type < ERR_WT_ALNUM) {
1598                                 buf[i] = ALNUM_CHARS[rand() / (RAND_MAX / ALNUM_LEN + 1)];
1599                                 err_type = ERR_WT_TOTAL;
1600                             } else {
1601                                 err_type -= ERR_WT_ALNUM;
1602                             }
1603
1604                             if (err_type < ERR_WT_FMT) {
1605                                 if ((unsigned int)i < phdr->caplen - 2)
1606                                     g_strlcpy((char*) &buf[i], "%s", 2);
1607                                 err_type = ERR_WT_TOTAL;
1608                             } else {
1609                                 err_type -= ERR_WT_FMT;
1610                             }
1611
1612                             if (err_type < ERR_WT_AA) {
1613                                 for (j = i; j < (int) phdr->caplen; j++)
1614                                     buf[j] = 0xAA;
1615                                 i = phdr->caplen;
1616                             }
1617                         }
1618                     }
1619                 }
1620
1621                 if (!wtap_dump(pdh, phdr, buf, &err, &err_info)) {
1622                     switch (err) {
1623                     case WTAP_ERR_UNWRITABLE_ENCAP:
1624                         /*
1625                          * This is a problem with the particular frame we're
1626                          * writing and the file type and subtype we're
1627                          * writing; note that, and report the frame number
1628                          * and file type/subtype.
1629                          */
1630                         fprintf(stderr,
1631                                 "editcap: Frame %u of \"%s\" has a network type that can't be saved in a \"%s\" file.\n",
1632                                 read_count, argv[optind],
1633                                 wtap_file_type_subtype_string(out_file_type_subtype));
1634                         break;
1635
1636                     case WTAP_ERR_PACKET_TOO_LARGE:
1637                         /*
1638                          * This is a problem with the particular frame we're
1639                          * writing and the file type and subtype we're
1640                          * writing; note that, and report the frame number
1641                          * and file type/subtype.
1642                          */
1643                         fprintf(stderr,
1644                                 "editcap: Frame %u of \"%s\" is too large for a \"%s\" file.\n",
1645                                 read_count, argv[optind],
1646                                 wtap_file_type_subtype_string(out_file_type_subtype));
1647                         break;
1648
1649                     case WTAP_ERR_UNWRITABLE_REC_TYPE:
1650                         /*
1651                          * This is a problem with the particular record we're
1652                          * writing and the file type and subtype we're
1653                          * writing; note that, and report the record number
1654                          * and file type/subtype.
1655                          */
1656                         fprintf(stderr,
1657                                 "editcap: Record %u of \"%s\" has a record type that can't be saved in a \"%s\" file.\n",
1658                                 read_count, argv[optind],
1659                                 wtap_file_type_subtype_string(out_file_type_subtype));
1660                         break;
1661
1662                     case WTAP_ERR_UNWRITABLE_REC_DATA:
1663                         /*
1664                          * This is a problem with the particular record we're
1665                          * writing and the file type and subtype we're
1666                          * writing; note that, and report the record number
1667                          * and file type/subtype.
1668                          */
1669                         fprintf(stderr,
1670                                 "editcap: Record %u of \"%s\" has data that can't be saved in a \"%s\" file.\n(%s)\n",
1671                                 read_count, argv[optind],
1672                                 wtap_file_type_subtype_string(out_file_type_subtype),
1673                                 err_info != NULL ? err_info : "no information supplied");
1674                         g_free(err_info);
1675                         break;
1676
1677                     default:
1678                         fprintf(stderr, "editcap: Error writing to %s: %s\n",
1679                                 filename, wtap_strerror(err));
1680                         break;
1681                     }
1682                     exit(2);
1683                 }
1684                 written_count++;
1685             }
1686             count++;
1687         }
1688
1689         g_free(fprefix);
1690         g_free(fsuffix);
1691
1692         if (err != 0) {
1693             /* Print a message noting that the read failed somewhere along the
1694              * line. */
1695             fprintf(stderr,
1696                     "editcap: An error occurred while reading \"%s\": %s.\n",
1697                     argv[optind], wtap_strerror(err));
1698             if (err_info != NULL) {
1699                 fprintf(stderr, "(%s)\n", err_info);
1700                 g_free(err_info);
1701             }
1702         }
1703
1704         if (!pdh) {
1705             /* No valid packages found, open the outfile so we can write an
1706              * empty header */
1707             g_free (filename);
1708             filename = g_strdup(argv[optind+1]);
1709
1710             pdh = wtap_dump_open_ng(filename, out_file_type_subtype, out_frame_type,
1711                                     snaplen ? MIN(snaplen, wtap_snapshot_length(wth)): wtap_snapshot_length(wth),
1712                                     FALSE /* compressed */, shb_hdr, idb_inf, &err);
1713             if (pdh == NULL) {
1714                 fprintf(stderr, "editcap: Can't open or create %s: %s\n",
1715                         filename, wtap_strerror(err));
1716                 exit(2);
1717             }
1718         }
1719
1720         g_free(idb_inf);
1721         idb_inf = NULL;
1722
1723         if (!wtap_dump_close(pdh, &err)) {
1724             fprintf(stderr, "editcap: Error writing to %s: %s\n", filename,
1725                     wtap_strerror(err));
1726             exit(2);
1727         }
1728         g_free(shb_hdr);
1729         g_free(filename);
1730     }
1731
1732     if (dup_detect) {
1733         fprintf(stderr, "%u packet%s seen, %u packet%s skipped with duplicate window of %i packets.\n",
1734                 count - 1, plurality(count - 1, "", "s"), duplicate_count,
1735                 plurality(duplicate_count, "", "s"), dup_window);
1736     } else if (dup_detect_by_time) {
1737         fprintf(stderr, "%u packet%s seen, %u packet%s skipped with duplicate time window equal to or less than %ld.%09ld seconds.\n",
1738                 count - 1, plurality(count - 1, "", "s"), duplicate_count,
1739                 plurality(duplicate_count, "", "s"),
1740                 (long)relative_time_window.secs,
1741                 (long int)relative_time_window.nsecs);
1742     }
1743
1744     return 0;
1745 }
1746
1747 /* Skip meta-information read from file to return offset of real
1748  * protocol data */
1749 static int
1750 find_dct2000_real_data(guint8 *buf)
1751 {
1752     int n = 0;
1753
1754     for (n = 0; buf[n] != '\0'; n++);   /* Context name */
1755     n++;
1756     n++;                                /* Context port number */
1757     for (; buf[n] != '\0'; n++);        /* Timestamp */
1758     n++;
1759     for (; buf[n] != '\0'; n++);        /* Protocol name */
1760     n++;
1761     for (; buf[n] != '\0'; n++);        /* Variant number (as string) */
1762     n++;
1763     for (; buf[n] != '\0'; n++);        /* Outhdr (as string) */
1764     n++;
1765     n += 2;                             /* Direction & encap */
1766
1767     return n;
1768 }
1769
1770 /*
1771  * We support up to 2 chopping regions in a single pass: one specified by the
1772  * positive chop length, and one by the negative chop length.
1773  */
1774 static void
1775 handle_chopping(chop_t chop, struct wtap_pkthdr *out_phdr,
1776                 const struct wtap_pkthdr *in_phdr, guint8 **buf,
1777                 gboolean adjlen)
1778 {
1779     /* Only packets can be chopped. */
1780     if (in_phdr->rec_type != REC_TYPE_PACKET)
1781         return;
1782
1783     /* If we're not chopping anything from one side, then the offset for that
1784      * side is meaningless. */
1785     if (chop.len_begin == 0)
1786         chop.off_begin_pos = chop.off_begin_neg = 0;
1787     if (chop.len_end == 0)
1788         chop.off_end_pos = chop.off_end_neg = 0;
1789
1790     if (chop.off_begin_neg < 0) {
1791         chop.off_begin_pos += in_phdr->caplen + chop.off_begin_neg;
1792         chop.off_begin_neg = 0;
1793     }
1794     if (chop.off_end_pos > 0) {
1795         chop.off_end_neg += chop.off_end_pos - in_phdr->caplen;
1796         chop.off_end_pos = 0;
1797     }
1798
1799     /* If we've crossed chopping regions, swap them */
1800     if (chop.len_begin && chop.len_end) {
1801         if (chop.off_begin_pos > ((int)in_phdr->caplen + chop.off_end_neg)) {
1802             int tmp_len, tmp_off;
1803
1804             tmp_off = in_phdr->caplen + chop.off_end_neg + chop.len_end;
1805             tmp_len = -chop.len_end;
1806
1807             chop.off_end_neg = chop.len_begin + chop.off_begin_pos - in_phdr->caplen;
1808             chop.len_end = -chop.len_begin;
1809
1810             chop.len_begin = tmp_len;
1811             chop.off_begin_pos = tmp_off;
1812         }
1813     }
1814
1815     /* Make sure we don't chop off more than we have available */
1816     if (in_phdr->caplen < (guint32)(chop.off_begin_pos - chop.off_end_neg)) {
1817         chop.len_begin = 0;
1818         chop.len_end = 0;
1819     }
1820     if ((guint32)(chop.len_begin - chop.len_end) >
1821         (in_phdr->caplen - (guint32)(chop.off_begin_pos - chop.off_end_neg))) {
1822         chop.len_begin = in_phdr->caplen - (chop.off_begin_pos - chop.off_end_neg);
1823         chop.len_end = 0;
1824     }
1825
1826     /* Handle chopping from the beginning.  Note that if a beginning offset
1827      * was specified, we need to keep that piece */
1828     if (chop.len_begin > 0) {
1829         *out_phdr = *in_phdr;
1830
1831         if (chop.off_begin_pos > 0) {
1832             memmove(*buf + chop.off_begin_pos,
1833                     *buf + chop.off_begin_pos + chop.len_begin,
1834                     out_phdr->caplen - chop.len_begin);
1835         } else {
1836             *buf += chop.len_begin;
1837         }
1838         out_phdr->caplen -= chop.len_begin;
1839
1840         if (adjlen) {
1841             if (in_phdr->len > (guint32)chop.len_begin)
1842                 out_phdr->len -= chop.len_begin;
1843             else
1844                 out_phdr->len = 0;
1845         }
1846         in_phdr = out_phdr;
1847     }
1848
1849     /* Handle chopping from the end.  Note that if an ending offset was
1850      * specified, we need to keep that piece */
1851     if (chop.len_end < 0) {
1852         *out_phdr = *in_phdr;
1853
1854         if (chop.off_end_neg < 0) {
1855             memmove(*buf + (gint)out_phdr->caplen + (chop.len_end + chop.off_end_neg),
1856                     *buf + (gint)out_phdr->caplen + chop.off_end_neg,
1857                     -chop.off_end_neg);
1858         }
1859         out_phdr->caplen += chop.len_end;
1860
1861         if (adjlen) {
1862             if (((signed int) in_phdr->len + chop.len_end) > 0)
1863                 out_phdr->len += chop.len_end;
1864             else
1865                 out_phdr->len = 0;
1866         }
1867         /*in_phdr = out_phdr;*/
1868     }
1869 }
1870
1871 /*
1872  * Editor modelines  -  http://www.wireshark.org/tools/modelines.html
1873  *
1874  * Local variables:
1875  * c-basic-offset: 4
1876  * tab-width: 8
1877  * indent-tabs-mode: nil
1878  * End:
1879  *
1880  * vi: set shiftwidth=4 tabstop=8 expandtab:
1881  * :indentSize=4:tabSize=8:noTabs=true:
1882  */
1883