Have (almost) all preferences use the generic preferences API (per https://bugs.wires...
[metze/wireshark/wip.git] / tshark.c
1 /* tshark.c
2  *
3  * Text-mode variant of Wireshark, along the lines of tcpdump and snoop,
4  * by Gilbert Ramirez <gram@alumni.rice.edu> and Guy Harris <guy@alum.mit.edu>.
5  *
6  * $Id$
7  *
8  * Wireshark - Network traffic analyzer
9  * By Gerald Combs <gerald@wireshark.org>
10  * Copyright 1998 Gerald Combs
11  *
12  * This program is free software; you can redistribute it and/or
13  * modify it under the terms of the GNU General Public License
14  * as published by the Free Software Foundation; either version 2
15  * of the License, or (at your option) any later version.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software
24  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
25  */
26
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #include <stdlib.h>
32 #include <stdio.h>
33 #include <string.h>
34 #include <ctype.h>
35 #include <locale.h>
36 #include <limits.h>
37
38 #ifdef HAVE_UNISTD_H
39 #include <unistd.h>
40 #endif
41
42 #include <errno.h>
43
44 #ifdef HAVE_FCNTL_H
45 #include <fcntl.h>
46 #endif
47
48 #include <signal.h>
49
50 #ifdef HAVE_SYS_STAT_H
51 # include <sys/stat.h>
52 #endif
53
54 #ifndef HAVE_GETOPT
55 #include "wsutil/wsgetopt.h"
56 #endif
57
58 #include <glib.h>
59 #include <epan/epan.h>
60 #include <epan/filesystem.h>
61 #include <wsutil/privileges.h>
62 #include <wsutil/file_util.h>
63
64 #include "globals.h"
65 #include <epan/timestamp.h>
66 #include <epan/packet.h>
67 #include "file.h"
68 #include "disabled_protos.h"
69 #include <epan/prefs.h>
70 #include <epan/column.h>
71 #include "print.h"
72 #include <epan/addr_resolv.h>
73 #include "ui/util.h"
74 #include "clopts_common.h"
75 #include "console_io.h"
76 #include "cmdarg_err.h"
77 #include "version_info.h"
78 #include <epan/plugins.h>
79 #include "register.h"
80 #include <epan/epan_dissect.h>
81 #include <epan/tap.h>
82 #include <epan/stat_cmd_args.h>
83 #include <epan/timestamp.h>
84 #include <epan/ex-opt.h>
85
86 #ifdef HAVE_LIBPCAP
87 #include "capture_ui_utils.h"
88 #include "capture_ifinfo.h"
89 #include "capture-pcap-util.h"
90 #ifdef _WIN32
91 #include "capture-wpcap.h"
92 #include <wsutil/unicode-utils.h>
93 #endif /* _WIN32 */
94 #include "capture_sync.h"
95 #endif /* HAVE_LIBPCAP */
96 #include "log.h"
97 #include <epan/funnel.h>
98 #include "capture_opts.h"
99
100 /*
101  * This is the template for the decode as option; it is shared between the
102  * various functions that output the usage for this parameter.
103  */
104 static const gchar decode_as_arg_template[] = "<layer_type>==<selector>,<decode_as_protocol>";
105
106 static guint32 cum_bytes;
107 static nstime_t first_ts;
108 static nstime_t prev_dis_ts;
109 static nstime_t prev_cap_ts;
110
111 static gboolean print_packet_info;      /* TRUE if we're to print packet information */
112
113 static gboolean perform_two_pass_analysis;
114
115 /*
116  * The way the packet decode is to be written.
117  */
118 typedef enum {
119   WRITE_TEXT,   /* summary or detail text */
120   WRITE_XML,    /* PDML or PSML */
121   WRITE_FIELDS  /* User defined list of fields */
122   /* Add CSV and the like here */
123 } output_action_e;
124
125 static output_action_e output_action;
126 static gboolean do_dissection;  /* TRUE if we have to dissect each packet */
127 static gboolean verbose;
128 static gboolean print_hex;
129 static gboolean line_buffered;
130
131 static print_format_e print_format = PR_FMT_TEXT;
132 static print_stream_t *print_stream;
133
134 static output_fields_t* output_fields  = NULL;
135
136 /* The line separator used between packets, changeable via the -S option */
137 static char *separator = "";
138
139 #ifdef HAVE_LIBPCAP
140 /*
141  * TRUE if we're to print packet counts to keep track of captured packets.
142  */
143 static gboolean print_packet_counts;
144
145 static capture_options global_capture_opts;
146
147 #ifdef SIGINFO
148 static gboolean infodelay;      /* if TRUE, don't print capture info in SIGINFO handler */
149 static gboolean infoprint;      /* if TRUE, print capture info after clearing infodelay */
150 #endif /* SIGINFO */
151
152 static gboolean capture(void);
153 static void report_counts(void);
154 #ifdef _WIN32
155 static BOOL WINAPI capture_cleanup(DWORD);
156 #else /* _WIN32 */
157 static void capture_cleanup(int);
158 #ifdef SIGINFO
159 static void report_counts_siginfo(int);
160 #endif /* SIGINFO */
161 #endif /* _WIN32 */
162 #endif /* HAVE_LIBPCAP */
163
164 static int load_cap_file(capture_file *, char *, int, gboolean, int, gint64);
165 static gboolean process_packet(capture_file *cf, gint64 offset,
166     const struct wtap_pkthdr *whdr, union wtap_pseudo_header *pseudo_header,
167     const guchar *pd, gboolean filtering_tap_listeners, guint tap_flags);
168 static void show_capture_file_io_error(const char *, int, gboolean);
169 static void show_print_file_io_error(int err);
170 static gboolean write_preamble(capture_file *cf);
171 static gboolean print_packet(capture_file *cf, epan_dissect_t *edt);
172 static gboolean write_finale(void);
173 static const char *cf_open_error_message(int err, gchar *err_info,
174     gboolean for_writing, int file_type);
175
176 static void open_failure_message(const char *filename, int err,
177     gboolean for_writing);
178 static void failure_message(const char *msg_format, va_list ap);
179 static void read_failure_message(const char *filename, int err);
180 static void write_failure_message(const char *filename, int err);
181
182 capture_file cfile;
183
184 struct string_elem {
185   const char *sstr;   /* The short string */
186   const char *lstr;   /* The long string */
187 };
188
189 static gint
190 string_compare(gconstpointer a, gconstpointer b)
191 {
192   return strcmp(((const struct string_elem *)a)->sstr,
193                 ((const struct string_elem *)b)->sstr);
194 }
195
196 static void
197 string_elem_print(gpointer data, gpointer not_used _U_)
198 {
199   fprintf(stderr, "    %s - %s\n",
200           ((struct string_elem *)data)->sstr,
201           ((struct string_elem *)data)->lstr);
202 }
203
204 static void
205 list_capture_types(void) {
206   int i;
207   struct string_elem *captypes;
208   GSList *list = NULL;
209
210   captypes = g_malloc(sizeof(struct string_elem) * WTAP_NUM_FILE_TYPES);
211
212   fprintf(stderr, "tshark: The available capture file types for the \"-F\" flag are:\n");
213   for (i = 0; i < WTAP_NUM_FILE_TYPES; i++) {
214     if (wtap_dump_can_open(i)) {
215       captypes[i].sstr = wtap_file_type_short_string(i);
216       captypes[i].lstr = wtap_file_type_string(i);
217       list = g_slist_insert_sorted(list, &captypes[i], string_compare);
218     }
219   }
220   g_slist_foreach(list, string_elem_print, NULL);
221   g_slist_free(list);
222   g_free(captypes);
223 }
224
225 static void
226 print_usage(gboolean print_ver)
227 {
228   FILE *output;
229
230   if (print_ver) {
231     output = stdout;
232     fprintf(output,
233         "TShark " VERSION "%s\n"
234         "Dump and analyze network traffic.\n"
235         "See http://www.wireshark.org for more information.\n"
236         "\n"
237         "%s",
238          wireshark_svnversion, get_copyright_info());
239   } else {
240     output = stderr;
241   }
242   fprintf(output, "\n");
243   fprintf(output, "Usage: tshark [options] ...\n");
244   fprintf(output, "\n");
245
246 #ifdef HAVE_LIBPCAP
247   fprintf(output, "Capture interface:\n");
248   fprintf(output, "  -i <interface>           name or idx of interface (def: first non-loopback)\n");
249   fprintf(output, "  -f <capture filter>      packet filter in libpcap filter syntax\n");
250   fprintf(output, "  -s <snaplen>             packet snapshot length (def: 65535)\n");
251   fprintf(output, "  -p                       don't capture in promiscuous mode\n");
252 #ifdef HAVE_PCAP_CREATE
253   fprintf(output, "  -I                       capture in monitor mode, if available\n");
254 #endif
255 #if defined(_WIN32) || defined(HAVE_PCAP_CREATE)
256   fprintf(output, "  -B <buffer size>         size of kernel buffer (def: 1MB)\n");
257 #endif
258   fprintf(output, "  -y <link type>           link layer type (def: first appropriate)\n");
259   fprintf(output, "  -D                       print list of interfaces and exit\n");
260   fprintf(output, "  -L                       print list of link-layer types of iface and exit\n");
261   fprintf(output, "\n");
262   fprintf(output, "Capture stop conditions:\n");
263   fprintf(output, "  -c <packet count>        stop after n packets (def: infinite)\n");
264   fprintf(output, "  -a <autostop cond.> ...  duration:NUM - stop after NUM seconds\n");
265   fprintf(output, "                           filesize:NUM - stop this file after NUM KB\n");
266   fprintf(output, "                              files:NUM - stop after NUM files\n");
267   /*fprintf(output, "\n");*/
268   fprintf(output, "Capture output:\n");
269   fprintf(output, "  -b <ringbuffer opt.> ... duration:NUM - switch to next file after NUM secs\n");
270   fprintf(output, "                           filesize:NUM - switch to next file after NUM KB\n");
271   fprintf(output, "                              files:NUM - ringbuffer: replace after NUM files\n");
272 #endif  /* HAVE_LIBPCAP */
273 #ifdef HAVE_PCAP_REMOTE
274   fprintf(output, "RPCAP options:\n");
275   fprintf(output, "  -A <user>:<password>     use RPCAP password authentication\n");
276 #endif
277   /*fprintf(output, "\n");*/
278   fprintf(output, "Input file:\n");
279   fprintf(output, "  -r <infile>              set the filename to read from (no pipes or stdin!)\n");
280
281   fprintf(output, "\n");
282   fprintf(output, "Processing:\n");
283   fprintf(output, "  -2                       perform a two-pass analysis\n");
284   fprintf(output, "  -R <read filter>         packet filter in Wireshark display filter syntax\n");
285   fprintf(output, "  -n                       disable all name resolutions (def: all enabled)\n");
286   fprintf(output, "  -N <name resolve flags>  enable specific name resolution(s): \"mntC\"\n");
287   fprintf(output, "  -d %s ...\n", decode_as_arg_template);
288   fprintf(output, "                           \"Decode As\", see the man page for details\n");
289   fprintf(output, "                           Example: tcp.port==8888,http\n");
290   fprintf(output, "  -H <hosts file>          read a list of entries from a hosts file, which will\n");
291   fprintf(output, "                           then be written to a capture file. (Implies -W n)\n");
292
293   /*fprintf(output, "\n");*/
294   fprintf(output, "Output:\n");
295   fprintf(output, "  -w <outfile|->           write packets to a pcap-format file named \"outfile\"\n");
296   fprintf(output, "                           (or to the standard output for \"-\")\n");
297   fprintf(output, "  -C <config profile>      start with specified configuration profile\n");
298   fprintf(output, "  -F <output file type>    set the output file type, default is pcapng\n");
299   fprintf(output, "                           an empty \"-F\" option will list the file types\n");
300   fprintf(output, "  -V                       add output of packet tree        (Packet Details)\n");
301   fprintf(output, "  -O <protocols>           Only show packet details of these protocols, comma\n");
302   fprintf(output, "                           separated\n");
303   fprintf(output, "  -P                       print packets even when writing to a file\n");
304   fprintf(output, "  -S <separator>           the line separator to print between packets\n");
305   fprintf(output, "  -x                       add output of hex and ASCII dump (Packet Bytes)\n");
306   fprintf(output, "  -T pdml|ps|psml|text|fields\n");
307   fprintf(output, "                           format of text output (def: text)\n");
308   fprintf(output, "  -e <field>               field to print if -Tfields selected (e.g. tcp.port);\n");
309   fprintf(output, "                           this option can be repeated to print multiple fields\n");
310   fprintf(output, "  -E<fieldsoption>=<value> set options for output when -Tfields selected:\n");
311   fprintf(output, "     header=y|n            switch headers on and off\n");
312   fprintf(output, "     separator=/t|/s|<char> select tab, space, printable character as separator\n");
313   fprintf(output, "     occurrence=f|l|a      print first, last or all occurrences of each field\n");
314   fprintf(output, "     aggregator=,|/s|<char> select comma, space, printable character as\n");
315   fprintf(output, "                           aggregator\n");
316   fprintf(output, "     quote=d|s|n           select double, single, no quotes for values\n");
317   fprintf(output, "  -t ad|a|r|d|dd|e         output format of time stamps (def: r: rel. to first)\n");
318   fprintf(output, "  -u s|hms                 output format of seconds (def: s: seconds)\n");
319   fprintf(output, "  -l                       flush standard output after each packet\n");
320   fprintf(output, "  -q                       be more quiet on stdout (e.g. when using statistics)\n");
321   fprintf(output, "  -W n                     Save extra information in the file, if supported.\n");
322   fprintf(output, "                           n = write network address resolution information\n");
323   fprintf(output, "  -X <key>:<value>         eXtension options, see the man page for details\n");
324   fprintf(output, "  -z <statistics>          various statistics, see the man page for details\n");
325
326   fprintf(output, "\n");
327   fprintf(output, "Miscellaneous:\n");
328   fprintf(output, "  -h                       display this help and exit\n");
329   fprintf(output, "  -v                       display version info and exit\n");
330   fprintf(output, "  -o <name>:<value> ...    override preference setting\n");
331   fprintf(output, "  -K <keytab>              keytab file to use for kerberos decryption\n");
332   fprintf(output, "  -G [report]              dump one of several available reports and exit\n");
333   fprintf(output, "                           default report=\"fields\"\n");
334   fprintf(output, "                           use \"-G ?\" for more help\n");
335 }
336
337 static void
338 glossary_option_help(void)
339 {
340   FILE *output;
341
342   output = stdout;
343
344   fprintf(output, "TShark " VERSION "%s\n", wireshark_svnversion);
345
346   fprintf(output, "\n");
347   fprintf(output, "Usage: tshark -G [report]\n");
348   fprintf(output, "\n");
349   fprintf(output, "Glossary table reports:\n");
350   fprintf(output, "  -G [fields]              dump glossary in original format and exit\n");
351   fprintf(output, "  -G fields2               dump glossary in format 2 and exit\n");
352   fprintf(output, "  -G fields3               dump glossary in format 3 and exit\n");
353   fprintf(output, "  -G protocols             dump protocols in registration database and exit\n");
354   fprintf(output, "  -G values                dump value, range, true/false strings and exit\n");
355   fprintf(output, "  -G ftypes                dump field type basic and descriptive names\n");
356   fprintf(output, "  -G decodes               dump \"layer type\"/\"decode as\" associations and exit\n");
357   fprintf(output, "  -G heuristic-decodes     dump heuristic dissector tables\n");
358   fprintf(output, "\n");
359   fprintf(output, "Preference reports:\n");
360   fprintf(output, "  -G defaultprefs          dump default preferences and exit\n");
361   fprintf(output, "  -G currentprefs          dump current preferences and exit\n");
362   fprintf(output, "\n");
363
364 }
365
366 /*
367  * For a dissector table, print on the stream described by output,
368  * its short name (which is what's used in the "-d" option) and its
369  * descriptive name.
370  */
371 static void
372 display_dissector_table_names(const char *table_name, const char *ui_name,
373                               gpointer output)
374 {
375   fprintf((FILE *)output, "\t%s (%s)\n", table_name, ui_name);
376 }
377
378 /*
379  * For a dissector handle, print on the stream described by output,
380  * the filter name (which is what's used in the "-d" option) and the full
381  * name for the protocol that corresponds to this handle.
382  */
383 static void
384 display_dissector_names(const gchar *table _U_, gpointer handle, gpointer output)
385 {
386   int                proto_id;
387   const gchar*       proto_filter_name;
388   const gchar*       proto_ui_name;
389
390   proto_id = dissector_handle_get_protocol_index((dissector_handle_t)handle);
391
392   if (proto_id != -1) {
393     proto_filter_name = proto_get_protocol_filter_name(proto_id);
394     proto_ui_name =  proto_get_protocol_name(proto_id);
395     g_assert(proto_filter_name != NULL);
396     g_assert(proto_ui_name != NULL);
397
398     fprintf((FILE *)output, "\t%s (%s)\n",
399             proto_filter_name,
400             proto_ui_name);
401   }
402 }
403
404 /*
405  * The protocol_name_search structure is used by find_protocol_name_func()
406  * to pass parameters and store results
407  */
408 struct protocol_name_search{
409   gchar              *searched_name;  /* Protocol filter name we are looking for */
410   dissector_handle_t  matched_handle; /* Handle for a dissector whose protocol has the specified filter name */
411   guint               nb_match;       /* How many dissectors matched searched_name */
412 };
413 typedef struct protocol_name_search *protocol_name_search_t;
414
415 /*
416  * This function parses all dissectors associated with a table to find the
417  * one whose protocol has the specified filter name.  It is called
418  * as a reference function in a call to dissector_table_foreach_handle.
419  * The name we are looking for, as well as the results, are stored in the
420  * protocol_name_search struct pointed to by user_data.
421  * If called using dissector_table_foreach_handle, we actually parse the
422  * whole list of dissectors.
423  */
424 static void
425 find_protocol_name_func(const gchar *table _U_, gpointer handle, gpointer user_data)
426
427 {
428   int                         proto_id;
429   const gchar                *protocol_filter_name;
430   protocol_name_search_t      search_info;
431
432   g_assert(handle);
433
434   search_info = (protocol_name_search_t)user_data;
435
436   proto_id = dissector_handle_get_protocol_index((dissector_handle_t)handle);
437   if (proto_id != -1) {
438     protocol_filter_name = proto_get_protocol_filter_name(proto_id);
439     g_assert(protocol_filter_name != NULL);
440     if (strcmp(protocol_filter_name, search_info->searched_name) == 0) {
441       /* Found a match */
442       if (search_info->nb_match == 0) {
443         /* Record this handle only if this is the first match */
444         search_info->matched_handle = (dissector_handle_t)handle; /* Record the handle for this matching dissector */
445       }
446       search_info->nb_match++;
447     }
448   }
449 }
450
451 /*
452  * Print all layer type names supported.
453  * We send the output to the stream described by the handle output.
454  */
455
456 static void
457 fprint_all_layer_types(FILE *output)
458
459 {
460   dissector_all_tables_foreach_table(display_dissector_table_names, (gpointer)output);
461 }
462
463 /*
464  * Print all protocol names supported for a specific layer type.
465  * table_name contains the layer type name in which the search is performed.
466  * We send the output to the stream described by the handle output.
467  */
468
469 static void
470 fprint_all_protocols_for_layer_types(FILE *output, gchar *table_name)
471
472 {
473   dissector_table_foreach_handle(table_name,
474                                  display_dissector_names,
475                                  (gpointer)output);
476 }
477
478 /*
479  * The function below parses the command-line parameters for the decode as
480  * feature (a string pointer by cl_param).
481  * It checks the format of the command-line, searches for a matching table
482  * and dissector.  If a table/dissector match is not found, we display a
483  * summary of the available tables/dissectors (on stderr) and return FALSE.
484  * If everything is fine, we get the "Decode as" preference activated,
485  * then we return TRUE.
486  */
487 static gboolean
488 add_decode_as(const gchar *cl_param)
489 {
490   gchar                        *table_name;
491   guint32                       selector;
492   gchar                        *decoded_param;
493   gchar                        *remaining_param;
494   gchar                        *selector_str;
495   gchar                        *dissector_str;
496   dissector_handle_t            dissector_matching;
497   dissector_table_t             table_matching;
498   ftenum_t                      dissector_table_selector_type;
499   struct protocol_name_search   user_protocol_name;
500
501   /* The following code will allocate and copy the command-line options in a string pointed by decoded_param */
502
503   g_assert(cl_param);
504   decoded_param = g_strdup(cl_param);
505   g_assert(decoded_param);
506
507
508   /* The lines below will parse this string (modifying it) to extract all
509     necessary information.  Note that decoded_param is still needed since
510     strings are not copied - we just save pointers. */
511
512   /* This section extracts a layer type (table_name) from decoded_param */
513   table_name = decoded_param; /* Layer type string starts from beginning */
514
515   remaining_param = strchr(table_name, '=');
516   if (remaining_param == NULL) {
517     cmdarg_err("Parameter \"%s\" doesn't follow the template \"%s\"", cl_param, decode_as_arg_template);
518     /* If the argument does not follow the template, carry on anyway to check
519        if the table name is at least correct.  If remaining_param is NULL,
520        we'll exit anyway further down */
521   }
522   else {
523     *remaining_param = '\0'; /* Terminate the layer type string (table_name) where '=' was detected */
524   }
525
526   /* Remove leading and trailing spaces from the table name */
527   while ( table_name[0] == ' ' )
528     table_name++;
529   while ( table_name[strlen(table_name) - 1] == ' ' )
530     table_name[strlen(table_name) - 1] = '\0'; /* Note: if empty string, while loop will eventually exit */
531
532 /* The following part searches a table matching with the layer type specified */
533   table_matching = NULL;
534
535 /* Look for the requested table */
536   if ( !(*(table_name)) ) { /* Is the table name empty, if so, don't even search for anything, display a message */
537     cmdarg_err("No layer type specified"); /* Note, we don't exit here, but table_matching will remain NULL, so we exit below */
538   }
539   else {
540     table_matching = find_dissector_table(table_name);
541     if (!table_matching) {
542       cmdarg_err("Unknown layer type -- %s", table_name); /* Note, we don't exit here, but table_matching will remain NULL, so we exit below */
543     }
544   }
545
546   if (!table_matching) {
547     /* Display a list of supported layer types to help the user, if the
548        specified layer type was not found */
549     cmdarg_err("Valid layer types are:");
550     fprint_all_layer_types(stderr);
551   }
552   if (remaining_param == NULL || !table_matching) {
553     /* Exit if the layer type was not found, or if no '=' separator was found
554        (see above) */
555     g_free(decoded_param);
556     return FALSE;
557   }
558
559   if (*(remaining_param + 1) != '=') { /* Check for "==" and not only '=' */
560     cmdarg_err("WARNING: -d requires \"==\" instead of \"=\". Option will be treated as \"%s==%s\"", table_name, remaining_param + 1);
561   }
562   else {
563     remaining_param++; /* Move to the second '=' */
564     *remaining_param = '\0'; /* Remove the second '=' */
565   }
566   remaining_param++; /* Position after the layer type string */
567
568   /* This section extracts a selector value (selector_str) from decoded_param */
569
570   selector_str = remaining_param; /* Next part starts with the selector number */
571
572   remaining_param = strchr(selector_str, ',');
573   if (remaining_param == NULL) {
574     cmdarg_err("Parameter \"%s\" doesn't follow the template \"%s\"", cl_param, decode_as_arg_template);
575     /* If the argument does not follow the template, carry on anyway to check
576        if the selector value is at least correct.  If remaining_param is NULL,
577        we'll exit anyway further down */
578   }
579   else {
580     *remaining_param = '\0'; /* Terminate the selector number string (selector_str) where ',' was detected */
581   }
582
583   dissector_table_selector_type = get_dissector_table_selector_type(table_name);
584
585   switch (dissector_table_selector_type) {
586
587   case FT_UINT8:
588   case FT_UINT16:
589   case FT_UINT24:
590   case FT_UINT32:
591     /* The selector for this table is an unsigned number.  Parse it as such.
592        There's no need to remove leading and trailing spaces from the
593        selector number string, because sscanf will do that for us. */
594     if ( sscanf(selector_str, "%u", &selector) != 1 ) {
595       cmdarg_err("Invalid selector number \"%s\"", selector_str);
596       g_free(decoded_param);
597       return FALSE;
598     }
599     break;
600
601   case FT_STRING:
602   case FT_STRINGZ:
603     /* The selector for this table is a string. */
604     break;
605
606   default:
607     /* There are currently no dissector tables with any types other
608        than the ones listed above. */
609     g_assert_not_reached();
610   }
611
612   if (remaining_param == NULL) {
613     /* Exit if no ',' separator was found (see above) */
614     cmdarg_err("Valid protocols for layer type \"%s\" are:", table_name);
615     fprint_all_protocols_for_layer_types(stderr, table_name);
616     g_free(decoded_param);
617     return FALSE;
618   }
619
620   remaining_param++; /* Position after the selector number string */
621
622   /* This section extracts a protocol filter name (dissector_str) from decoded_param */
623
624   dissector_str = remaining_param; /* All the rest of the string is the dissector (decode as protocol) name */
625
626   /* Remove leading and trailing spaces from the dissector name */
627   while ( dissector_str[0] == ' ' )
628     dissector_str++;
629   while ( dissector_str[strlen(dissector_str) - 1] == ' ' )
630     dissector_str[strlen(dissector_str) - 1] = '\0'; /* Note: if empty string, while loop will eventually exit */
631
632   dissector_matching = NULL;
633
634   /* We now have a pointer to the handle for the requested table inside the variable table_matching */
635   if ( ! (*dissector_str) ) { /* Is the dissector name empty, if so, don't even search for a matching dissector and display all dissectors found for the selected table */
636     cmdarg_err("No protocol name specified"); /* Note, we don't exit here, but dissector_matching will remain NULL, so we exit below */
637   }
638   else {
639     user_protocol_name.nb_match = 0;
640     user_protocol_name.searched_name = dissector_str;
641     user_protocol_name.matched_handle = NULL;
642
643     dissector_table_foreach_handle(table_name, find_protocol_name_func, &user_protocol_name); /* Go and perform the search for this dissector in the this table's dissectors' names and shortnames */
644
645     if (user_protocol_name.nb_match != 0) {
646       dissector_matching = user_protocol_name.matched_handle;
647       if (user_protocol_name.nb_match > 1) {
648         cmdarg_err("WARNING: Protocol \"%s\" matched %u dissectors, first one will be used", dissector_str, user_protocol_name.nb_match);
649       }
650     }
651     else {
652       /* OK, check whether the problem is that there isn't any such
653          protocol, or that there is but it's not specified as a protocol
654          that's valid for that dissector table.
655          Note, we don't exit here, but dissector_matching will remain NULL,
656          so we exit below */
657       if (proto_get_id_by_filter_name(dissector_str) == -1) {
658         /* No such protocol */
659         cmdarg_err("Unknown protocol -- \"%s\"", dissector_str);
660       } else {
661         cmdarg_err("Protocol \"%s\" isn't valid for layer type \"%s\"",
662                    dissector_str, table_name);
663       }
664     }
665   }
666
667   if (!dissector_matching) {
668     cmdarg_err("Valid protocols for layer type \"%s\" are:", table_name);
669     fprint_all_protocols_for_layer_types(stderr, table_name);
670     g_free(decoded_param);
671     return FALSE;
672   }
673
674 /* This is the end of the code that parses the command-line options.
675    All information is now stored in the variables:
676    table_name
677    selector
678    dissector_matching
679    The above variables that are strings are still pointing to areas within
680    decoded_parm.  decoded_parm thus still needs to be kept allocated in
681    until we stop needing these variables
682    decoded_param will be deallocated at each exit point of this function */
683
684
685   /* We now have a pointer to the handle for the requested dissector
686      (requested protocol) inside the variable dissector_matching */
687   switch (dissector_table_selector_type) {
688
689   case FT_UINT8:
690   case FT_UINT16:
691   case FT_UINT24:
692   case FT_UINT32:
693     /* The selector for this table is an unsigned number. */
694     dissector_change_uint(table_name, selector, dissector_matching);
695     break;
696
697   case FT_STRING:
698   case FT_STRINGZ:
699     /* The selector for this table is a string. */
700     dissector_change_string(table_name, selector_str, dissector_matching);
701     break;
702
703   default:
704     /* There are currently no dissector tables with any types other
705        than the ones listed above. */
706     g_assert_not_reached();
707   }
708   g_free(decoded_param); /* "Decode As" rule has been successfully added */
709   return TRUE;
710 }
711
712 static void
713 tshark_log_handler (const gchar *log_domain, GLogLevelFlags log_level,
714     const gchar *message, gpointer user_data)
715 {
716   /* ignore log message, if log_level isn't interesting based
717      upon the console log preferences.
718      If the preferences haven't been loaded loaded yet, display the
719      message anyway.
720
721      The default console_log_level preference value is such that only
722        ERROR, CRITICAL and WARNING level messages are processed;
723        MESSAGE, INFO and DEBUG level messages are ignored.
724
725      XXX: Aug 07, 2009: Prior tshark g_log code was hardwired to process only
726            ERROR and CRITICAL level messages so the current code is a behavioral
727            change.  The current behavior is the same as in Wireshark.
728   */
729   if((log_level & G_LOG_LEVEL_MASK & prefs.console_log_level) == 0 &&
730      prefs.console_log_level != 0) {
731     return;
732   }
733
734   g_log_default_handler(log_domain, log_level, message, user_data);
735
736 }
737
738 static char *
739 output_file_description(const char *fname)
740 {
741   char *save_file_string;
742
743   /* Get a string that describes what we're writing to */
744   if (strcmp(fname, "-") == 0) {
745     /* We're writing to the standard output */
746     save_file_string = g_strdup("standard output");
747   } else {
748     /* We're writing to a file with the name in save_file */
749     save_file_string = g_strdup_printf("file \"%s\"", fname);
750   }
751   return save_file_string;
752 }
753
754 static void
755 print_current_user(void) {
756   gchar *cur_user, *cur_group;
757   if (started_with_special_privs()) {
758     cur_user = get_cur_username();
759     cur_group = get_cur_groupname();
760     fprintf(stderr, "Running as user \"%s\" and group \"%s\".",
761       cur_user, cur_group);
762     g_free(cur_user);
763     g_free(cur_group);
764     if (running_with_special_privs()) {
765       fprintf(stderr, " This could be dangerous.");
766     }
767     fprintf(stderr, "\n");
768   }
769 }
770
771 static void
772 check_capture_privs(void) {
773 #ifdef _WIN32
774   load_wpcap();
775   /* Warn the user if npf.sys isn't loaded. */
776   if (!npf_sys_is_running() && get_os_major_version() >= 6) {
777     fprintf(stderr, "The NPF driver isn't running.  You may have trouble "
778       "capturing or\nlisting interfaces.\n");
779   }
780 #endif
781 }
782
783 static void
784 show_version(GString *comp_info_str, GString *runtime_info_str)
785 {
786   printf("TShark " VERSION "%s\n"
787          "\n"
788          "%s"
789          "\n"
790          "%s"
791          "\n"
792          "%s",
793          wireshark_svnversion, get_copyright_info(), comp_info_str->str,
794          runtime_info_str->str);
795 }
796
797 int
798 main(int argc, char *argv[])
799 {
800   char                *init_progfile_dir_error;
801   int                  opt;
802   gboolean             arg_error = FALSE;
803
804 #ifdef _WIN32
805   WSADATA              wsaData;
806 #endif  /* _WIN32 */
807
808   char                *gpf_path, *pf_path;
809   char                *gdp_path, *dp_path;
810   int                  gpf_open_errno, gpf_read_errno;
811   int                  pf_open_errno, pf_read_errno;
812   int                  gdp_open_errno, gdp_read_errno;
813   int                  dp_open_errno, dp_read_errno;
814   int                  err;
815   volatile int         exit_status = 0;
816 #ifdef HAVE_LIBPCAP
817   gboolean             list_link_layer_types = FALSE;
818   gboolean             start_capture = FALSE;
819   int                  status;
820   GList               *if_list;
821   gchar               *err_str;
822   guint                i;
823   interface_options    interface_opts;
824 #else
825   gboolean             capture_option_specified = FALSE;
826 #endif
827   gboolean             quiet = FALSE;
828 #ifdef PCAP_NG_DEFAULT
829   volatile int         out_file_type = WTAP_FILE_PCAPNG;
830 #else
831   volatile int         out_file_type = WTAP_FILE_PCAP;
832 #endif
833   volatile gboolean    out_file_name_res = FALSE;
834   gchar               *volatile cf_name = NULL;
835   gchar               *rfilter = NULL;
836 #ifdef HAVE_PCAP_OPEN_DEAD
837   struct bpf_program   fcode;
838 #endif
839   dfilter_t           *rfcode = NULL;
840   e_prefs             *prefs_p;
841   char                 badopt;
842   GLogLevelFlags       log_flags;
843   int                  optind_initial;
844   gchar               *output_only = NULL;
845
846 #ifdef HAVE_PCAP_REMOTE
847 #define OPTSTRING_A "A:"
848 #else
849 #define OPTSTRING_A ""
850 #endif
851 #ifdef HAVE_LIBPCAP
852 #if defined(_WIN32) || defined(HAVE_PCAP_CREATE)
853 #define OPTSTRING_B "B:"
854 #else
855 #define OPTSTRING_B ""
856 #endif  /* _WIN32 or HAVE_PCAP_CREATE */
857 #else /* HAVE_LIBPCAP */
858 #define OPTSTRING_B ""
859 #endif  /* HAVE_LIBPCAP */
860
861 #ifdef HAVE_PCAP_CREATE
862 #define OPTSTRING_I "I"
863 #else
864 #define OPTSTRING_I ""
865 #endif
866
867 #define OPTSTRING "2a:" OPTSTRING_A "b:" OPTSTRING_B "c:C:d:De:E:f:F:G:hH:i:" OPTSTRING_I "K:lLnN:o:O:pPqr:R:s:S:t:T:u:vVw:W:xX:y:z:"
868
869   static const char    optstring[] = OPTSTRING;
870
871 #ifdef _WIN32
872   arg_list_utf_16to8(argc, argv);
873 #if !GLIB_CHECK_VERSION(2,31,0)
874   g_thread_init(NULL);
875 #endif
876 #endif /* _WIN32 */
877
878   /*
879    * Get credential information for later use.
880    */
881   init_process_policies();
882
883   /*
884    * Attempt to get the pathname of the executable file.
885    */
886   init_progfile_dir_error = init_progfile_dir(argv[0], main);
887   if (init_progfile_dir_error != NULL) {
888     fprintf(stderr, "tshark: Can't get pathname of tshark program: %s.\n",
889             init_progfile_dir_error);
890   }
891
892   /*
893    * In order to have the -X opts assigned before the wslua machine starts
894    * we need to call getopts before epan_init() gets called.
895    */
896   opterr = 0;
897   optind_initial = optind;
898
899   while ((opt = getopt(argc, argv, optstring)) != -1) {
900     switch (opt) {
901     case 'C':        /* Configuration Profile */
902       if (profile_exists (optarg, FALSE)) {
903         set_profile_name (optarg);
904       } else {
905         cmdarg_err("Configuration Profile \"%s\" does not exist", optarg);
906         return 1;
907       }
908       break;
909     case 'X':
910       ex_opt_add(optarg);
911       break;
912     default:
913       break;
914     }
915   }
916
917   optind = optind_initial;
918   opterr = 1;
919
920
921
922 /** Send All g_log messages to our own handler **/
923
924   log_flags =
925                     G_LOG_LEVEL_ERROR|
926                     G_LOG_LEVEL_CRITICAL|
927                     G_LOG_LEVEL_WARNING|
928                     G_LOG_LEVEL_MESSAGE|
929                     G_LOG_LEVEL_INFO|
930                     G_LOG_LEVEL_DEBUG|
931                     G_LOG_FLAG_FATAL|G_LOG_FLAG_RECURSION;
932
933   g_log_set_handler(NULL,
934                     log_flags,
935                     tshark_log_handler, NULL /* user_data */);
936   g_log_set_handler(LOG_DOMAIN_MAIN,
937                     log_flags,
938                     tshark_log_handler, NULL /* user_data */);
939
940 #ifdef HAVE_LIBPCAP
941   g_log_set_handler(LOG_DOMAIN_CAPTURE,
942                     log_flags,
943                     tshark_log_handler, NULL /* user_data */);
944   g_log_set_handler(LOG_DOMAIN_CAPTURE_CHILD,
945                     log_flags,
946                     tshark_log_handler, NULL /* user_data */);
947 #endif
948
949   initialize_funnel_ops();
950
951 #ifdef HAVE_LIBPCAP
952   capture_opts_init(&global_capture_opts, &cfile);
953 #endif
954
955   timestamp_set_type(TS_RELATIVE);
956   timestamp_set_precision(TS_PREC_AUTO);
957   timestamp_set_seconds_type(TS_SECONDS_DEFAULT);
958
959   /* Register all dissectors; we must do this before checking for the
960      "-G" flag, as the "-G" flag dumps information registered by the
961      dissectors, and we must do it before we read the preferences, in
962      case any dissectors register preferences. */
963   epan_init(register_all_protocols, register_all_protocol_handoffs, NULL, NULL,
964             failure_message, open_failure_message, read_failure_message,
965             write_failure_message);
966
967   /* Register all tap listeners; we do this before we parse the arguments,
968      as the "-z" argument can specify a registered tap. */
969
970   /* we register the plugin taps before the other taps because
971      stats_tree taps plugins will be registered as tap listeners
972      by stats_tree_stat.c and need to registered before that */
973 #ifdef HAVE_PLUGINS
974   register_all_plugin_tap_listeners();
975 #endif
976   register_all_tap_listeners();
977
978   /* If invoked with the "-G" flag, we dump out information based on
979      the argument to the "-G" flag; if no argument is specified,
980      for backwards compatibility we dump out a glossary of display
981      filter symbols.
982
983      XXX - we do this here, for now, to support "-G" with no arguments.
984      If none of our build or other processes uses "-G" with no arguments,
985      we can just process it with the other arguments. */
986   if (argc >= 2 && strcmp(argv[1], "-G") == 0) {
987     proto_initialize_all_prefixes();
988
989     if (argc == 2)
990       proto_registrar_dump_fields(1);
991     else {
992       if (strcmp(argv[2], "fields") == 0)
993         proto_registrar_dump_fields(1);
994       else if (strcmp(argv[2], "fields2") == 0)
995         proto_registrar_dump_fields(2);
996       else if (strcmp(argv[2], "fields3") == 0)
997         proto_registrar_dump_fields(3);
998       else if (strcmp(argv[2], "protocols") == 0)
999         proto_registrar_dump_protocols();
1000       else if (strcmp(argv[2], "values") == 0)
1001         proto_registrar_dump_values();
1002       else if (strcmp(argv[2], "ftypes") == 0)
1003         proto_registrar_dump_ftypes();
1004       else if (strcmp(argv[2], "decodes") == 0)
1005         dissector_dump_decodes();
1006       else if (strcmp(argv[2], "heuristic-decodes") == 0)
1007         dissector_dump_heur_decodes();
1008       else if (strcmp(argv[2], "defaultprefs") == 0)
1009         write_prefs(NULL);
1010       else if (strcmp(argv[2], "plugins") == 0)
1011         plugins_dump_all();
1012       else if (strcmp(argv[2], "?") == 0)
1013         glossary_option_help();
1014       else if (strcmp(argv[2], "-?") == 0)
1015         glossary_option_help();
1016       else if (strcmp(argv[2], "currentprefs") == 0) {
1017         read_prefs(&gpf_open_errno, &gpf_read_errno, &gpf_path,
1018             &pf_open_errno, &pf_read_errno, &pf_path);
1019         write_prefs(NULL);
1020       } else {
1021         cmdarg_err("Invalid \"%s\" option for -G flag, enter -G ? for more help.", argv[2]);
1022         return 1;
1023       }
1024     }
1025     return 0;
1026   }
1027
1028   /* Set the C-language locale to the native environment. */
1029   setlocale(LC_ALL, "");
1030
1031   prefs_p = read_prefs(&gpf_open_errno, &gpf_read_errno, &gpf_path,
1032                      &pf_open_errno, &pf_read_errno, &pf_path);
1033   if (gpf_path != NULL) {
1034     if (gpf_open_errno != 0) {
1035       cmdarg_err("Can't open global preferences file \"%s\": %s.",
1036               pf_path, g_strerror(gpf_open_errno));
1037     }
1038     if (gpf_read_errno != 0) {
1039       cmdarg_err("I/O error reading global preferences file \"%s\": %s.",
1040               pf_path, g_strerror(gpf_read_errno));
1041     }
1042   }
1043   if (pf_path != NULL) {
1044     if (pf_open_errno != 0) {
1045       cmdarg_err("Can't open your preferences file \"%s\": %s.", pf_path,
1046               g_strerror(pf_open_errno));
1047     }
1048     if (pf_read_errno != 0) {
1049       cmdarg_err("I/O error reading your preferences file \"%s\": %s.",
1050               pf_path, g_strerror(pf_read_errno));
1051     }
1052     g_free(pf_path);
1053     pf_path = NULL;
1054   }
1055
1056   /* Read the disabled protocols file. */
1057   read_disabled_protos_list(&gdp_path, &gdp_open_errno, &gdp_read_errno,
1058                             &dp_path, &dp_open_errno, &dp_read_errno);
1059   if (gdp_path != NULL) {
1060     if (gdp_open_errno != 0) {
1061       cmdarg_err("Could not open global disabled protocols file\n\"%s\": %s.",
1062                  gdp_path, g_strerror(gdp_open_errno));
1063     }
1064     if (gdp_read_errno != 0) {
1065       cmdarg_err("I/O error reading global disabled protocols file\n\"%s\": %s.",
1066                  gdp_path, g_strerror(gdp_read_errno));
1067     }
1068     g_free(gdp_path);
1069   }
1070   if (dp_path != NULL) {
1071     if (dp_open_errno != 0) {
1072       cmdarg_err(
1073         "Could not open your disabled protocols file\n\"%s\": %s.", dp_path,
1074         g_strerror(dp_open_errno));
1075     }
1076     if (dp_read_errno != 0) {
1077       cmdarg_err(
1078         "I/O error reading your disabled protocols file\n\"%s\": %s.", dp_path,
1079         g_strerror(dp_read_errno));
1080     }
1081     g_free(dp_path);
1082   }
1083
1084   check_capture_privs();
1085
1086   cap_file_init(&cfile);
1087
1088   /* Print format defaults to this. */
1089   print_format = PR_FMT_TEXT;
1090
1091   output_fields = output_fields_new();
1092
1093   /* Now get our args */
1094   while ((opt = getopt(argc, argv, optstring)) != -1) {
1095     switch (opt) {
1096     case '2':        /* Perform two pass analysis */
1097       perform_two_pass_analysis = TRUE;
1098       break;
1099     case 'a':        /* autostop criteria */
1100     case 'b':        /* Ringbuffer option */
1101     case 'c':        /* Capture x packets */
1102     case 'f':        /* capture filter */
1103     case 'i':        /* Use interface x */
1104     case 'p':        /* Don't capture in promiscuous mode */
1105 #ifdef HAVE_PCAP_REMOTE
1106     case 'A':        /* Authentication */
1107 #endif
1108 #ifdef HAVE_PCAP_CREATE
1109     case 'I':        /* Capture in monitor mode, if available */
1110 #endif
1111     case 's':        /* Set the snapshot (capture) length */
1112     case 'w':        /* Write to capture file x */
1113     case 'y':        /* Set the pcap data link type */
1114 #if defined(_WIN32) || defined(HAVE_PCAP_CREATE)
1115     case 'B':        /* Buffer size */
1116 #endif /* _WIN32 or HAVE_PCAP_CREATE */
1117 #ifdef HAVE_LIBPCAP
1118       status = capture_opts_add_opt(&global_capture_opts, opt, optarg, &start_capture);
1119       if(status != 0) {
1120         return status;
1121       }
1122 #else
1123       capture_option_specified = TRUE;
1124       arg_error = TRUE;
1125 #endif
1126       break;
1127     case 'C':
1128       /* Configuration profile settings were already processed just ignore them this time*/
1129       break;
1130     case 'd':        /* Decode as rule */
1131       if (!add_decode_as(optarg))
1132         return 1;
1133       break;
1134 #if defined(HAVE_HEIMDAL_KERBEROS) || defined(HAVE_MIT_KERBEROS)
1135     case 'K':        /* Kerberos keytab file */
1136       read_keytab_file(optarg);
1137       break;
1138 #endif
1139     case 'D':        /* Print a list of capture devices and exit */
1140 #ifdef HAVE_LIBPCAP
1141       if_list = capture_interface_list(&err, &err_str);
1142       if (if_list == NULL) {
1143         switch (err) {
1144         case CANT_GET_INTERFACE_LIST:
1145         case DONT_HAVE_PCAP:
1146           cmdarg_err("%s", err_str);
1147           g_free(err_str);
1148           break;
1149
1150         case NO_INTERFACES_FOUND:
1151           cmdarg_err("There are no interfaces on which a capture can be done");
1152           break;
1153         }
1154         return 2;
1155       }
1156       capture_opts_print_interfaces(if_list);
1157       free_interface_list(if_list);
1158       return 0;
1159 #else
1160       capture_option_specified = TRUE;
1161       arg_error = TRUE;
1162 #endif
1163       break;
1164     case 'e':
1165       /* Field entry */
1166       output_fields_add(output_fields, optarg);
1167       break;
1168     case 'E':
1169       /* Field option */
1170       if(!output_fields_set_option(output_fields, optarg)) {
1171         cmdarg_err("\"%s\" is not a valid field output option=value pair.", optarg);
1172         output_fields_list_options(stderr);
1173         return 1;
1174       }
1175       break;
1176     case 'F':
1177       out_file_type = wtap_short_string_to_file_type(optarg);
1178       if (out_file_type < 0) {
1179         cmdarg_err("\"%s\" isn't a valid capture file type", optarg);
1180         list_capture_types();
1181         return 1;
1182       }
1183       break;
1184     case 'W':        /* Select extra information to save in our capture file */
1185       /* This is patterned after the -N flag which may not be the best idea. */
1186       if (strchr(optarg, 'n'))
1187         out_file_name_res = TRUE;
1188       break;
1189     case 'H':        /* Read address to name mappings from a hosts file */
1190       if (! read_hosts_file(optarg))
1191       {
1192         cmdarg_err("Can't read host entries from \"%s\"", optarg);
1193         return 1;
1194       }
1195       out_file_name_res = TRUE;
1196       break;
1197
1198     case 'h':        /* Print help and exit */
1199       print_usage(TRUE);
1200       return 0;
1201       break;
1202     case 'l':        /* "Line-buffer" standard output */
1203       /* This isn't line-buffering, strictly speaking, it's just
1204          flushing the standard output after the information for
1205          each packet is printed; however, that should be good
1206          enough for all the purposes to which "-l" is put (and
1207          is probably actually better for "-V", as it does fewer
1208          writes).
1209
1210          See the comment in "process_packet()" for an explanation of
1211          why we do that, and why we don't just use "setvbuf()" to
1212          make the standard output line-buffered (short version: in
1213          Windows, "line-buffered" is the same as "fully-buffered",
1214          and the output buffer is only flushed when it fills up). */
1215       line_buffered = TRUE;
1216       break;
1217     case 'L':        /* Print list of link-layer types and exit */
1218 #ifdef HAVE_LIBPCAP
1219       list_link_layer_types = TRUE;
1220 #else
1221       capture_option_specified = TRUE;
1222       arg_error = TRUE;
1223 #endif
1224       break;
1225     case 'n':        /* No name resolution */
1226       gbl_resolv_flags.mac_name = FALSE;
1227       gbl_resolv_flags.network_name = FALSE;
1228       gbl_resolv_flags.transport_name = FALSE;
1229       gbl_resolv_flags.concurrent_dns = FALSE;
1230       break;
1231     case 'N':        /* Select what types of addresses/port #s to resolve */
1232       badopt = string_to_name_resolve(optarg, &gbl_resolv_flags);
1233       if (badopt != '\0') {
1234         cmdarg_err("-N specifies unknown resolving option '%c';",
1235                    badopt);
1236         cmdarg_err_cont( "           Valid options are 'm', 'n', 't', and 'C'");
1237         return 1;
1238       }
1239       break;
1240     case 'o':        /* Override preference from command line */
1241       switch (prefs_set_pref(optarg)) {
1242
1243       case PREFS_SET_OK:
1244         break;
1245
1246       case PREFS_SET_SYNTAX_ERR:
1247         cmdarg_err("Invalid -o flag \"%s\"", optarg);
1248         return 1;
1249         break;
1250
1251       case PREFS_SET_NO_SUCH_PREF:
1252       case PREFS_SET_OBSOLETE:
1253         cmdarg_err("-o flag \"%s\" specifies unknown preference", optarg);
1254         return 1;
1255         break;
1256       }
1257       break;
1258     case 'q':        /* Quiet */
1259       quiet = TRUE;
1260       break;
1261     case 'r':        /* Read capture file x */
1262       cf_name = g_strdup(optarg);
1263       break;
1264     case 'R':        /* Read file filter */
1265       rfilter = optarg;
1266       break;
1267     case 'P':        /* Print packets even when writing to a file */
1268       print_packet_info = TRUE;
1269       break;
1270     case 'S':        /* Set the line Separator to be printed between packets */
1271       separator = strdup(optarg);
1272       break;
1273     case 't':        /* Time stamp type */
1274       if (strcmp(optarg, "r") == 0)
1275         timestamp_set_type(TS_RELATIVE);
1276       else if (strcmp(optarg, "a") == 0)
1277         timestamp_set_type(TS_ABSOLUTE);
1278       else if (strcmp(optarg, "ad") == 0)
1279         timestamp_set_type(TS_ABSOLUTE_WITH_DATE);
1280       else if (strcmp(optarg, "d") == 0)
1281         timestamp_set_type(TS_DELTA);
1282       else if (strcmp(optarg, "dd") == 0)
1283         timestamp_set_type(TS_DELTA_DIS);
1284       else if (strcmp(optarg, "e") == 0)
1285         timestamp_set_type(TS_EPOCH);
1286       else if (strcmp(optarg, "u") == 0)
1287         timestamp_set_type(TS_UTC);
1288       else if (strcmp(optarg, "ud") == 0)
1289         timestamp_set_type(TS_UTC_WITH_DATE);
1290       else {
1291         cmdarg_err("Invalid time stamp type \"%s\"",
1292                    optarg);
1293         cmdarg_err_cont("It must be \"r\" for relative, \"a\" for absolute,");
1294         cmdarg_err_cont("\"ad\" for absolute with date, or \"d\" for delta.");
1295         return 1;
1296       }
1297       break;
1298     case 'T':        /* printing Type */
1299       if (strcmp(optarg, "text") == 0) {
1300         output_action = WRITE_TEXT;
1301         print_format = PR_FMT_TEXT;
1302       } else if (strcmp(optarg, "ps") == 0) {
1303         output_action = WRITE_TEXT;
1304         print_format = PR_FMT_PS;
1305       } else if (strcmp(optarg, "pdml") == 0) {
1306         output_action = WRITE_XML;
1307         verbose = TRUE;
1308       } else if (strcmp(optarg, "psml") == 0) {
1309         output_action = WRITE_XML;
1310         verbose = FALSE;
1311       } else if(strcmp(optarg, "fields") == 0) {
1312         output_action = WRITE_FIELDS;
1313         verbose = TRUE; /* Need full tree info */
1314       } else {
1315         cmdarg_err("Invalid -T parameter.");
1316         cmdarg_err_cont("It must be \"ps\", \"text\", \"pdml\", \"psml\" or \"fields\".");
1317         return 1;
1318       }
1319       break;
1320     case 'u':        /* Seconds type */
1321       if (strcmp(optarg, "s") == 0)
1322         timestamp_set_seconds_type(TS_SECONDS_DEFAULT);
1323       else if (strcmp(optarg, "hms") == 0)
1324         timestamp_set_seconds_type(TS_SECONDS_HOUR_MIN_SEC);
1325       else {
1326         cmdarg_err("Invalid seconds type \"%s\"", optarg);
1327         cmdarg_err_cont("It must be \"s\" for seconds or \"hms\" for hours, minutes and seconds.");
1328         return 1;
1329       }
1330       break;
1331     case 'v':         /* Show version and exit */
1332     {
1333       GString             *comp_info_str;
1334       GString             *runtime_info_str;
1335       /* Assemble the compile-time version information string */
1336       comp_info_str = g_string_new("Compiled ");
1337       get_compiled_version_info(comp_info_str, NULL, epan_get_compiled_version_info);
1338
1339       /* Assemble the run-time version information string */
1340       runtime_info_str = g_string_new("Running ");
1341       get_runtime_version_info(runtime_info_str, NULL);
1342       show_version(comp_info_str, runtime_info_str);
1343       g_string_free(comp_info_str, TRUE);
1344       g_string_free(runtime_info_str, TRUE);
1345       return 0;
1346     }
1347     case 'O':        /* Only output these protocols */
1348       output_only = g_strdup(optarg);
1349       /* FALLTHROUGH */
1350     case 'V':        /* Verbose */
1351       verbose = TRUE;
1352       /*  The user asked for a verbose output, so let's ensure they get it,
1353        *  even if they're writing to a file.
1354        */
1355       print_packet_info = TRUE;
1356       break;
1357     case 'x':        /* Print packet data in hex (and ASCII) */
1358       print_hex = TRUE;
1359       /*  The user asked for hex output, so let's ensure they get it,
1360        *  even if they're writing to a file.
1361        */
1362       print_packet_info = TRUE;
1363       break;
1364     case 'X':
1365       break;
1366     case 'z':
1367       /* We won't call the init function for the stat this soon
1368          as it would disallow MATE's fields (which are registered
1369          by the preferences set callback) from being used as
1370          part of a tap filter.  Instead, we just add the argument
1371          to a list of stat arguments. */
1372       if (!process_stat_cmd_arg(optarg)) {
1373         cmdarg_err("invalid -z argument.");
1374         cmdarg_err_cont("  -z argument must be one of :");
1375         list_stat_cmd_args();
1376         return 1;
1377       }
1378       break;
1379     default:
1380     case '?':        /* Bad flag - print usage message */
1381       switch(optopt) {
1382       case 'F':
1383         list_capture_types();
1384         break;
1385       default:
1386         print_usage(TRUE);
1387       }
1388       return 1;
1389       break;
1390     }
1391   }
1392
1393   /* If we specified output fields, but not the output field type... */
1394   if(WRITE_FIELDS != output_action && 0 != output_fields_num_fields(output_fields)) {
1395         cmdarg_err("Output fields were specified with \"-e\", "
1396             "but \"-Tfields\" was not specified.");
1397         return 1;
1398   } else if(WRITE_FIELDS == output_action && 0 == output_fields_num_fields(output_fields)) {
1399         cmdarg_err("\"-Tfields\" was specified, but no fields were "
1400                     "specified with \"-e\".");
1401
1402         return 1;
1403   }
1404
1405   /* If no capture filter or read filter has been specified, and there are
1406      still command-line arguments, treat them as the tokens of a capture
1407      filter (if no "-r" flag was specified) or a read filter (if a "-r"
1408      flag was specified. */
1409   if (optind < argc) {
1410     if (cf_name != NULL) {
1411       if (rfilter != NULL) {
1412         cmdarg_err("Read filters were specified both with \"-R\" "
1413             "and with additional command-line arguments.");
1414         return 1;
1415       }
1416       rfilter = get_args_as_string(argc, argv, optind);
1417     } else {
1418 #ifdef HAVE_LIBPCAP
1419       if (global_capture_opts.default_options.cfilter) {
1420         cmdarg_err("A default capture filter was specified both with \"-f\""
1421             " and with additional command-line arguments.");
1422         return 1;
1423       }
1424       for (i = 0; i < global_capture_opts.ifaces->len; i++) {
1425         interface_opts = g_array_index(global_capture_opts.ifaces, interface_options, i);
1426         if (interface_opts.cfilter == NULL) {
1427           interface_opts.cfilter = get_args_as_string(argc, argv, optind);
1428           global_capture_opts.ifaces = g_array_remove_index(global_capture_opts.ifaces, i);
1429           g_array_insert_val(global_capture_opts.ifaces, i, interface_opts);
1430         } else {
1431           cmdarg_err("A capture filter was specified both with \"-f\""
1432               " and with additional command-line arguments.");
1433           return 1;
1434         }
1435       }
1436       global_capture_opts.default_options.cfilter = get_args_as_string(argc, argv, optind);
1437 #else
1438       capture_option_specified = TRUE;
1439 #endif
1440     }
1441   }
1442
1443 #ifdef HAVE_LIBPCAP
1444   if (!global_capture_opts.saving_to_file) {
1445     /* We're not saving the capture to a file; if "-q" wasn't specified,
1446        we should print packet information */
1447     if (!quiet)
1448       print_packet_info = TRUE;
1449   } else {
1450     /* We're saving to a file; if we're writing to the standard output.
1451        and we'll also be writing dissected packets to the standard
1452        output, reject the request.  At best, we could redirect that
1453        to the standard error; we *can't* write both to the standard
1454        output and have either of them be useful. */
1455     if (strcmp(global_capture_opts.save_file, "-") == 0 && print_packet_info) {
1456       cmdarg_err("You can't write both raw packet data and dissected packets"
1457           " to the standard output.");
1458       return 1;
1459     }
1460   }
1461 #else
1462   /* We're not saving the capture to a file; if "-q" wasn't specified,
1463      we should print packet information */
1464   if (!quiet)
1465     print_packet_info = TRUE;
1466 #endif
1467
1468 #ifndef HAVE_LIBPCAP
1469   if (capture_option_specified)
1470     cmdarg_err("This version of TShark was not built with support for capturing packets.");
1471 #endif
1472   if (arg_error) {
1473     print_usage(FALSE);
1474     return 1;
1475   }
1476
1477   /* We don't support capture filters when reading from a capture file
1478      (the BPF compiler doesn't support all link-layer types that we
1479      support in capture files we read). */
1480 #ifdef HAVE_LIBPCAP
1481   if (cf_name != NULL) {
1482     if (global_capture_opts.default_options.cfilter) {
1483       cmdarg_err("Only read filters, not capture filters, "
1484           "can be specified when reading a capture file.");
1485       return 1;
1486     }
1487   }
1488 #endif
1489
1490   if (print_hex) {
1491     if (output_action != WRITE_TEXT) {
1492       cmdarg_err("Raw packet hex data can only be printed as text or PostScript");
1493       return 1;
1494     }
1495   }
1496
1497   if (output_only != NULL) {
1498     char *ps;
1499
1500     if (!verbose) {
1501       cmdarg_err("-O requires -V");
1502       return 1;
1503     }
1504
1505     output_only_tables = g_hash_table_new (g_str_hash, g_str_equal);
1506     for (ps = strtok (output_only, ","); ps; ps = strtok (NULL, ",")) {
1507       g_hash_table_insert(output_only_tables, (gpointer)ps, (gpointer)ps);
1508     }
1509   }
1510
1511 #ifdef HAVE_LIBPCAP
1512   if (list_link_layer_types) {
1513     /* We're supposed to list the link-layer types for an interface;
1514        did the user also specify a capture file to be read? */
1515     if (cf_name) {
1516       /* Yes - that's bogus. */
1517       cmdarg_err("You can't specify -L and a capture file to be read.");
1518       return 1;
1519     }
1520     /* No - did they specify a ring buffer option? */
1521     if (global_capture_opts.multi_files_on) {
1522       cmdarg_err("Ring buffer requested, but a capture isn't being done.");
1523       return 1;
1524     }
1525   } else {
1526     if (cf_name) {
1527       /*
1528        * "-r" was specified, so we're reading a capture file.
1529        * Capture options don't apply here.
1530        */
1531       if (global_capture_opts.multi_files_on) {
1532         cmdarg_err("Multiple capture files requested, but "
1533                    "a capture isn't being done.");
1534         return 1;
1535       }
1536       if (global_capture_opts.has_file_duration) {
1537         cmdarg_err("Switching capture files after a time interval was specified, but "
1538                    "a capture isn't being done.");
1539         return 1;
1540       }
1541       if (global_capture_opts.has_ring_num_files) {
1542         cmdarg_err("A ring buffer of capture files was specified, but "
1543           "a capture isn't being done.");
1544         return 1;
1545       }
1546       if (global_capture_opts.has_autostop_files) {
1547         cmdarg_err("A maximum number of capture files was specified, but "
1548           "a capture isn't being done.");
1549         return 1;
1550       }
1551
1552       /* Note: TShark now allows the restriction of a _read_ file by packet count
1553        * and byte count as well as a write file. Other autostop options remain valid
1554        * only for a write file.
1555        */
1556       if (global_capture_opts.has_autostop_duration) {
1557         cmdarg_err("A maximum capture time was specified, but "
1558           "a capture isn't being done.");
1559         return 1;
1560       }
1561     } else {
1562       /*
1563        * "-r" wasn't specified, so we're doing a live capture.
1564        */
1565       if (global_capture_opts.saving_to_file) {
1566         /* They specified a "-w" flag, so we'll be saving to a capture file. */
1567
1568         /* When capturing, we only support writing pcap or pcap-ng format. */
1569         if (out_file_type != WTAP_FILE_PCAP && out_file_type != WTAP_FILE_PCAPNG) {
1570           cmdarg_err("Live captures can only be saved in libpcap format.");
1571           return 1;
1572         }
1573         if (global_capture_opts.multi_files_on) {
1574           /* Multiple-file mode doesn't work under certain conditions:
1575              a) it doesn't work if you're writing to the standard output;
1576              b) it doesn't work if you're writing to a pipe;
1577           */
1578           if (strcmp(global_capture_opts.save_file, "-") == 0) {
1579             cmdarg_err("Multiple capture files requested, but "
1580               "the capture is being written to the standard output.");
1581             return 1;
1582           }
1583           if (global_capture_opts.output_to_pipe) {
1584             cmdarg_err("Multiple capture files requested, but "
1585               "the capture file is a pipe.");
1586             return 1;
1587           }
1588           if (!global_capture_opts.has_autostop_filesize &&
1589               !global_capture_opts.has_file_duration) {
1590             cmdarg_err("Multiple capture files requested, but "
1591               "no maximum capture file size or duration was specified.");
1592             return 1;
1593           }
1594         }
1595         /* Currently, we don't support read filters when capturing
1596            and saving the packets. */
1597         if (rfilter != NULL) {
1598           cmdarg_err("Read filters aren't supported when capturing and saving the captured packets.");
1599           return 1;
1600         }
1601       } else {
1602         /* They didn't specify a "-w" flag, so we won't be saving to a
1603            capture file.  Check for options that only make sense if
1604            we're saving to a file. */
1605         if (global_capture_opts.has_autostop_filesize) {
1606           cmdarg_err("Maximum capture file size specified, but "
1607            "capture isn't being saved to a file.");
1608           return 1;
1609         }
1610         if (global_capture_opts.multi_files_on) {
1611           cmdarg_err("Multiple capture files requested, but "
1612             "the capture isn't being saved to a file.");
1613           return 1;
1614         }
1615       }
1616     }
1617   }
1618 #endif
1619
1620 #ifdef _WIN32
1621   /* Start windows sockets */
1622   WSAStartup( MAKEWORD( 1, 1 ), &wsaData );
1623 #endif /* _WIN32 */
1624
1625   /* Notify all registered modules that have had any of their preferences
1626      changed either from one of the preferences file or from the command
1627      line that their preferences have changed. */
1628   prefs_apply_all();
1629
1630   /* At this point MATE will have registered its field array so we can
1631      have a tap filter with one of MATE's late-registered fields as part
1632      of the filter.  We can now process all the "-z" arguments. */
1633   start_requested_stats();
1634
1635 #ifdef HAVE_LIBPCAP
1636   /* We currently don't support taps, or printing dissected packets,
1637      if we're writing to a pipe. */
1638   if (global_capture_opts.saving_to_file &&
1639       global_capture_opts.output_to_pipe) {
1640     if (tap_listeners_require_dissection()) {
1641       cmdarg_err("Taps aren't supported when saving to a pipe.");
1642       return 1;
1643     }
1644     if (print_packet_info) {
1645       cmdarg_err("Printing dissected packets isn't supported when saving to a pipe.");
1646       return 1;
1647     }
1648   }
1649 #endif
1650
1651   /* disabled protocols as per configuration file */
1652   if (gdp_path == NULL && dp_path == NULL) {
1653     set_disabled_protos_list();
1654   }
1655
1656   /* Build the column format array */
1657   build_column_format_array(&cfile.cinfo, prefs_p->num_cols, TRUE);
1658
1659 #ifdef HAVE_LIBPCAP
1660   capture_opts_trim_snaplen(&global_capture_opts, MIN_PACKET_SIZE);
1661   capture_opts_trim_ring_num_files(&global_capture_opts);
1662 #endif
1663
1664   if (rfilter != NULL) {
1665     if (!dfilter_compile(rfilter, &rfcode)) {
1666       cmdarg_err("%s", dfilter_error_msg);
1667       epan_cleanup();
1668 #ifdef HAVE_PCAP_OPEN_DEAD
1669       {
1670         pcap_t *pc;
1671
1672         pc = pcap_open_dead(DLT_EN10MB, MIN_PACKET_SIZE);
1673         if (pc != NULL) {
1674           if (pcap_compile(pc, &fcode, rfilter, 0, 0) != -1) {
1675             cmdarg_err_cont(
1676               "  Note: That display filter code looks like a valid capture filter;");
1677             cmdarg_err_cont(
1678               "        maybe you mixed them up?");
1679           }
1680           pcap_close(pc);
1681         }
1682       }
1683 #endif
1684       return 2;
1685     }
1686   }
1687   cfile.rfcode = rfcode;
1688
1689   if (print_packet_info) {
1690     /* If we're printing as text or PostScript, we have
1691        to create a print stream. */
1692     if (output_action == WRITE_TEXT) {
1693       switch (print_format) {
1694
1695       case PR_FMT_TEXT:
1696         print_stream = print_stream_text_stdio_new(stdout);
1697         break;
1698
1699       case PR_FMT_PS:
1700         print_stream = print_stream_ps_stdio_new(stdout);
1701         break;
1702
1703       default:
1704         g_assert_not_reached();
1705       }
1706     }
1707   }
1708
1709   /* We have to dissect each packet if:
1710
1711         we're printing information about each packet;
1712
1713         we're using a read filter on the packets;
1714
1715         we're using any taps that need dissection. */
1716   do_dissection = print_packet_info || rfcode || tap_listeners_require_dissection();
1717
1718   if (cf_name) {
1719     /*
1720      * We're reading a capture file.
1721      */
1722
1723     /*
1724      * Immediately relinquish any special privileges we have; we must not
1725      * be allowed to read any capture files the user running TShark
1726      * can't open.
1727      */
1728     relinquish_special_privs_perm();
1729     print_current_user();
1730
1731     if (cf_open(&cfile, cf_name, FALSE, &err) != CF_OK) {
1732       epan_cleanup();
1733       return 2;
1734     }
1735
1736     /* Set timestamp precision; there should arguably be a command-line
1737        option to let the user set this. */
1738     switch(wtap_file_tsprecision(cfile.wth)) {
1739     case(WTAP_FILE_TSPREC_SEC):
1740       timestamp_set_precision(TS_PREC_AUTO_SEC);
1741       break;
1742     case(WTAP_FILE_TSPREC_DSEC):
1743       timestamp_set_precision(TS_PREC_AUTO_DSEC);
1744       break;
1745     case(WTAP_FILE_TSPREC_CSEC):
1746       timestamp_set_precision(TS_PREC_AUTO_CSEC);
1747       break;
1748     case(WTAP_FILE_TSPREC_MSEC):
1749       timestamp_set_precision(TS_PREC_AUTO_MSEC);
1750       break;
1751     case(WTAP_FILE_TSPREC_USEC):
1752       timestamp_set_precision(TS_PREC_AUTO_USEC);
1753       break;
1754     case(WTAP_FILE_TSPREC_NSEC):
1755       timestamp_set_precision(TS_PREC_AUTO_NSEC);
1756       break;
1757     default:
1758       g_assert_not_reached();
1759     }
1760
1761     /* Process the packets in the file */
1762     TRY {
1763 #ifdef HAVE_LIBPCAP
1764       err = load_cap_file(&cfile, global_capture_opts.save_file, out_file_type, out_file_name_res,
1765           global_capture_opts.has_autostop_packets ? global_capture_opts.autostop_packets : 0,
1766           global_capture_opts.has_autostop_filesize ? global_capture_opts.autostop_filesize : 0);
1767 #else
1768       err = load_cap_file(&cfile, NULL, out_file_type, out_file_name_res, 0, 0);
1769 #endif
1770     }
1771     CATCH(OutOfMemoryError) {
1772       fprintf(stderr,
1773               "Out Of Memory!\n"
1774               "\n"
1775               "Sorry, but TShark has to terminate now!\n"
1776               "\n"
1777               "Some infos / workarounds can be found at:\n"
1778               "http://wiki.wireshark.org/KnownBugs/OutOfMemory\n");
1779       err = ENOMEM;
1780     }
1781     ENDTRY;
1782     if (err != 0) {
1783       /* We still dump out the results of taps, etc., as we might have
1784          read some packets; however, we exit with an error status. */
1785       exit_status = 2;
1786     }
1787   } else {
1788     /* No capture file specified, so we're supposed to do a live capture
1789        (or get a list of link-layer types for a live capture device);
1790        do we have support for live captures? */
1791 #ifdef HAVE_LIBPCAP
1792     /* trim the interface name and exit if that failed */
1793     if (!capture_opts_trim_iface(&global_capture_opts,
1794         ((prefs_p->capture_device) && (*prefs_p->capture_device != '\0')) ? get_if_name(prefs_p->capture_device) : NULL)) {
1795         return 2;
1796     }
1797
1798     /* if requested, list the link layer types and exit */
1799     if (list_link_layer_types) {
1800         guint i;
1801         interface_options interface_opts;
1802
1803         /* Get the list of link-layer types for the capture devices. */
1804         for (i = 0; i < global_capture_opts.ifaces->len; i++) {
1805             if_capabilities_t *caps;
1806
1807             interface_opts = g_array_index(global_capture_opts.ifaces, interface_options, i);
1808             caps = capture_get_if_capabilities(interface_opts.name, interface_opts.monitor_mode, &err_str);
1809             if (caps == NULL) {
1810                 cmdarg_err("%s", err_str);
1811                 g_free(err_str);
1812                 return 2;
1813             }
1814             if (caps->data_link_types == NULL) {
1815                 cmdarg_err("The capture device \"%s\" has no data link types.", interface_opts.name);
1816                 return 2;
1817             }
1818             capture_opts_print_if_capabilities(caps, interface_opts.name, interface_opts.monitor_mode);
1819             free_if_capabilities(caps);
1820         }
1821         return 0;
1822     }
1823
1824     if (print_packet_info) {
1825       if (!write_preamble(NULL)) {
1826         show_print_file_io_error(errno);
1827         return 2;
1828       }
1829     } else if (!quiet) {
1830       /*
1831        * We're not printing information for each packet, and the user
1832        * didn't ask us not to print a count of packets as they arrive,
1833        * so print that count so the user knows that packets are arriving.
1834        *
1835        * XXX - what if the user wants to do a live capture, doesn't want
1836        * to save it to a file, doesn't want information printed for each
1837        * packet, does want some "-z" statistic, and wants packet counts
1838        * so they know whether they're seeing any packets?
1839        */
1840       print_packet_counts = TRUE;
1841     }
1842
1843     /* For now, assume libpcap gives microsecond precision. */
1844     timestamp_set_precision(TS_PREC_AUTO_USEC);
1845
1846     /*
1847      * XXX - this returns FALSE if an error occurred, but it also
1848      * returns FALSE if the capture stops because a time limit
1849      * was reached (and possibly other limits), so we can't assume
1850      * it means an error.
1851      *
1852      * The capture code is a bit twisty, so it doesn't appear to
1853      * be an easy fix.  We just ignore the return value for now.
1854      * Instead, pass on the exit status from the capture child.
1855      */
1856     capture();
1857     exit_status = global_capture_opts.fork_child_status;
1858
1859     if (print_packet_info) {
1860       if (!write_finale()) {
1861         err = errno;
1862         show_print_file_io_error(err);
1863       }
1864     }
1865 #else
1866     /* No - complain. */
1867     cmdarg_err("This version of TShark was not built with support for capturing packets.");
1868     return 2;
1869 #endif
1870   }
1871
1872   g_free(cf_name);
1873
1874   if (cfile.frames != NULL) {
1875     free_frame_data_sequence(cfile.frames);
1876     cfile.frames = NULL;
1877   }
1878
1879   draw_tap_listeners(TRUE);
1880   funnel_dump_all_text_windows();
1881   epan_cleanup();
1882
1883   output_fields_free(output_fields);
1884   output_fields = NULL;
1885
1886   return exit_status;
1887 }
1888
1889 /*#define USE_BROKEN_G_MAIN_LOOP*/
1890
1891 #ifdef USE_BROKEN_G_MAIN_LOOP
1892   GMainLoop *loop;
1893 #else
1894   gboolean loop_running = FALSE;
1895 #endif
1896   guint32 packet_count = 0;
1897
1898
1899 /* XXX - move to the right position / file */
1900 /* read from a pipe (callback) */
1901 typedef gboolean (*pipe_input_cb_t) (gint source, gpointer user_data);
1902
1903 typedef struct pipe_input_tag {
1904   gint                source;
1905   gpointer            user_data;
1906   int                 *child_process;
1907   pipe_input_cb_t     input_cb;
1908   guint               pipe_input_id;
1909 #ifdef _WIN32
1910   GMutex              *callback_running;
1911 #endif
1912 } pipe_input_t;
1913
1914 static pipe_input_t pipe_input;
1915
1916 #ifdef _WIN32
1917 /* The timer has expired, see if there's stuff to read from the pipe,
1918    if so, do the callback */
1919 static gint
1920 pipe_timer_cb(gpointer data)
1921 {
1922   HANDLE handle;
1923   DWORD avail = 0;
1924   gboolean result, result1;
1925   DWORD childstatus;
1926   pipe_input_t *pipe_input_p = data;
1927   gint iterations = 0;
1928
1929   g_mutex_lock (pipe_input_p->callback_running);
1930
1931   /* try to read data from the pipe only 5 times, to avoid blocking */
1932   while(iterations < 5) {
1933     /*g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_timer_cb: new iteration");*/
1934
1935     /* Oddly enough although Named pipes don't work on win9x,
1936        PeekNamedPipe does !!! */
1937     handle = (HANDLE) _get_osfhandle (pipe_input_p->source);
1938     result = PeekNamedPipe(handle, NULL, 0, NULL, &avail, NULL);
1939
1940     /* Get the child process exit status */
1941     result1 = GetExitCodeProcess((HANDLE)*(pipe_input_p->child_process),
1942                                  &childstatus);
1943
1944     /* If the Peek returned an error, or there are bytes to be read
1945        or the childwatcher thread has terminated then call the normal
1946        callback */
1947     if (!result || avail > 0 || childstatus != STILL_ACTIVE) {
1948
1949       /*g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_timer_cb: data avail");*/
1950
1951       /* And call the real handler */
1952       if (!pipe_input_p->input_cb(pipe_input_p->source, pipe_input_p->user_data)) {
1953         g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_timer_cb: input pipe closed, iterations: %u", iterations);
1954         /* pipe closed, return false so that the timer is stopped */
1955         g_mutex_unlock (pipe_input_p->callback_running);
1956         return FALSE;
1957       }
1958     }
1959     else {
1960       /*g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_timer_cb: no data avail");*/
1961       /* No data, stop now */
1962       break;
1963     }
1964
1965     iterations++;
1966   }
1967
1968   /*g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_timer_cb: finished with iterations: %u, new timer", iterations);*/
1969
1970   g_mutex_unlock (pipe_input_p->callback_running);
1971
1972   /* we didn't stopped the timer, so let it run */
1973   return TRUE;
1974 }
1975 #endif
1976
1977
1978 void
1979 pipe_input_set_handler(gint source, gpointer user_data, int *child_process, pipe_input_cb_t input_cb)
1980 {
1981
1982   pipe_input.source         = source;
1983   pipe_input.child_process  = child_process;
1984   pipe_input.user_data      = user_data;
1985   pipe_input.input_cb       = input_cb;
1986
1987 #ifdef _WIN32
1988 #if GLIB_CHECK_VERSION(2,31,0)
1989   pipe_input.callback_running = g_malloc(sizeof(GMutex));
1990   g_mutex_init(pipe_input.callback_running);
1991 #else
1992   pipe_input.callback_running = g_mutex_new();
1993 #endif
1994   /* Tricky to use pipes in win9x, as no concept of wait.  NT can
1995      do this but that doesn't cover all win32 platforms.  GTK can do
1996      this but doesn't seem to work over processes.  Attempt to do
1997      something similar here, start a timer and check for data on every
1998      timeout. */
1999   /*g_log(NULL, G_LOG_LEVEL_DEBUG, "pipe_input_set_handler: new");*/
2000   pipe_input.pipe_input_id = g_timeout_add(200, pipe_timer_cb, &pipe_input);
2001 #endif
2002 }
2003
2004
2005 #ifdef HAVE_LIBPCAP
2006 static gboolean
2007 capture(void)
2008 {
2009   gboolean ret;
2010   guint i;
2011   GString *str = g_string_new("");
2012 #ifdef USE_TSHARK_SELECT
2013   fd_set readfds;
2014 #endif
2015 #ifndef _WIN32
2016   struct sigaction action, oldaction;
2017 #endif
2018
2019   /*
2020    * XXX - dropping privileges is still required, until code cleanup is done
2021    *
2022    * remove all dependencies to pcap specific code and using only dumpcap is almost done.
2023    * when it's done, we don't need special privileges to run tshark at all,
2024    * therefore we don't need to drop these privileges
2025    * The only thing we might want to keep is a warning if tshark is run as root,
2026    * as it's no longer necessary and potentially dangerous.
2027    *
2028    * THE FOLLOWING IS THE FORMER COMMENT WHICH IS NO LONGER REALLY VALID:
2029    * We've opened the capture device, so we shouldn't need any special
2030    * privileges any more; relinquish those privileges.
2031    *
2032    * XXX - if we have saved set-user-ID support, we should give up those
2033    * privileges immediately, and then reclaim them long enough to get
2034    * a list of network interfaces and to open one, and then give them
2035    * up again, so that stuff we do while processing the argument list,
2036    * reading the user's preferences, loading and starting plugins
2037    * (especially *user* plugins), etc. is done with the user's privileges,
2038    * not special privileges.
2039    */
2040   relinquish_special_privs_perm();
2041   print_current_user();
2042
2043   /* Cleanup all data structures used for dissection. */
2044   cleanup_dissection();
2045   /* Initialize all data structures used for dissection. */
2046   init_dissection();
2047
2048 #ifdef _WIN32
2049   /* Catch a CTRL+C event and, if we get it, clean up and exit. */
2050   SetConsoleCtrlHandler(capture_cleanup, TRUE);
2051 #else /* _WIN32 */
2052   /* Catch SIGINT and SIGTERM and, if we get either of them,
2053      clean up and exit.  If SIGHUP isn't being ignored, catch
2054      it too and, if we get it, clean up and exit.
2055
2056      We restart any read that was in progress, so that it doesn't
2057      disrupt reading from the sync pipe.  The signal handler tells
2058      the capture child to finish; it will report that it finished,
2059      or will exit abnormally, so  we'll stop reading from the sync
2060      pipe, pick up the exit status, and quit. */
2061   memset(&action, 0, sizeof(action));
2062   action.sa_handler = capture_cleanup;
2063   action.sa_flags = SA_RESTART;
2064   sigemptyset(&action.sa_mask);
2065   sigaction(SIGTERM, &action, NULL);
2066   sigaction(SIGINT, &action, NULL);
2067   sigaction(SIGHUP, NULL, &oldaction);
2068   if (oldaction.sa_handler == SIG_DFL)
2069     sigaction(SIGHUP, &action, NULL);
2070
2071 #ifdef SIGINFO
2072   /* Catch SIGINFO and, if we get it and we're capturing to a file in
2073      quiet mode, report the number of packets we've captured.
2074
2075      Again, restart any read that was in progress, so that it doesn't
2076      disrupt reading from the sync pipe. */
2077   action.sa_handler = report_counts_siginfo;
2078   action.sa_flags = SA_RESTART;
2079   sigemptyset(&action.sa_mask);
2080   sigaction(SIGINFO, &action, NULL);
2081 #endif /* SIGINFO */
2082 #endif /* _WIN32 */
2083
2084   global_capture_opts.state = CAPTURE_PREPARING;
2085
2086   /* Let the user know which interfaces were chosen. */
2087   for (i = 0; i < global_capture_opts.ifaces->len; i++) {
2088     interface_options interface_opts;
2089
2090     interface_opts = g_array_index(global_capture_opts.ifaces, interface_options, i);
2091     interface_opts.descr = get_interface_descriptive_name(interface_opts.name);
2092     global_capture_opts.ifaces = g_array_remove_index(global_capture_opts.ifaces, i);
2093     g_array_insert_val(global_capture_opts.ifaces, i, interface_opts);
2094   }
2095 #ifdef _WIN32
2096   if (global_capture_opts.ifaces->len < 2) {
2097 #else
2098   if (global_capture_opts.ifaces->len < 4) {
2099 #endif
2100     for (i = 0; i < global_capture_opts.ifaces->len; i++) {
2101       interface_options interface_opts;
2102
2103       interface_opts = g_array_index(global_capture_opts.ifaces, interface_options, i);
2104       if (i > 0) {
2105           if (global_capture_opts.ifaces->len > 2) {
2106               g_string_append_printf(str, ",");
2107           }
2108           g_string_append_printf(str, " ");
2109           if (i == global_capture_opts.ifaces->len - 1) {
2110               g_string_append_printf(str, "and ");
2111           }
2112       }
2113       g_string_append_printf(str, "%s", interface_opts.descr);
2114     }
2115   } else {
2116     g_string_append_printf(str, "%u interfaces", global_capture_opts.ifaces->len);
2117   }
2118   fprintf(stderr, "Capturing on %s\n", str->str);
2119   g_string_free(str, TRUE);
2120
2121   ret = sync_pipe_start(&global_capture_opts);
2122
2123   if (!ret)
2124     return FALSE;
2125
2126   /* the actual capture loop
2127    *
2128    * XXX - glib doesn't seem to provide any event based loop handling.
2129    *
2130    * XXX - for whatever reason,
2131    * calling g_main_loop_new() ends up in 100% cpu load.
2132    *
2133    * But that doesn't matter: in UNIX we can use select() to find an input
2134    * source with something to do.
2135    *
2136    * But that doesn't matter because we're in a CLI (that doesn't need to
2137    * update a GUI or something at the same time) so it's OK if we block
2138    * trying to read from the pipe.
2139    *
2140    * So all the stuff in USE_TSHARK_SELECT could be removed unless I'm
2141    * wrong (but I leave it there in case I am...).
2142    */
2143
2144 #ifdef USE_TSHARK_SELECT
2145   FD_ZERO(&readfds);
2146   FD_SET(pipe_input.source, &readfds);
2147 #endif
2148
2149   loop_running = TRUE;
2150
2151   TRY
2152   {
2153     while (loop_running)
2154     {
2155 #ifdef USE_TSHARK_SELECT
2156       ret = select(pipe_input.source+1, &readfds, NULL, NULL, NULL);
2157
2158       if (ret == -1)
2159       {
2160         perror("select()");
2161         return TRUE;
2162       } else if (ret == 1) {
2163 #endif
2164         /* Call the real handler */
2165         if (!pipe_input.input_cb(pipe_input.source, pipe_input.user_data)) {
2166           g_log(NULL, G_LOG_LEVEL_DEBUG, "input pipe closed");
2167           return FALSE;
2168         }
2169 #ifdef USE_TSHARK_SELECT
2170       }
2171 #endif
2172     }
2173   }
2174   CATCH(OutOfMemoryError) {
2175     fprintf(stderr,
2176             "Out Of Memory!\n"
2177             "\n"
2178             "Sorry, but TShark has to terminate now!\n"
2179             "\n"
2180             "Some infos / workarounds can be found at:\n"
2181             "http://wiki.wireshark.org/KnownBugs/OutOfMemory\n");
2182     exit(1);
2183   }
2184   ENDTRY;
2185   return TRUE;
2186 }
2187
2188
2189 /* XXX - move the call to main_window_update() out of capture_sync.c */
2190 /* dummy for capture_sync.c to make linker happy */
2191 void main_window_update(void)
2192 {
2193 }
2194
2195 /* capture child detected an error */
2196 void
2197 capture_input_error_message(capture_options *capture_opts _U_, char *error_msg, char *secondary_error_msg)
2198 {
2199   cmdarg_err("%s", error_msg);
2200   cmdarg_err_cont("%s", secondary_error_msg);
2201 }
2202
2203
2204 /* capture child detected an capture filter related error */
2205 void
2206 capture_input_cfilter_error_message(capture_options *capture_opts, guint i, char *error_message)
2207 {
2208   dfilter_t   *rfcode = NULL;
2209   interface_options interface_opts;
2210
2211   g_assert(i < capture_opts->ifaces->len);
2212   interface_opts = g_array_index(capture_opts->ifaces, interface_options, i);
2213
2214   if (dfilter_compile(interface_opts.cfilter, &rfcode) && rfcode != NULL) {
2215     cmdarg_err(
2216       "Invalid capture filter \"%s\" for interface %s!\n"
2217       "\n"
2218       "That string looks like a valid display filter; however, it isn't a valid\n"
2219       "capture filter (%s).\n"
2220       "\n"
2221       "Note that display filters and capture filters don't have the same syntax,\n"
2222       "so you can't use most display filter expressions as capture filters.\n"
2223       "\n"
2224       "See the User's Guide for a description of the capture filter syntax.",
2225       interface_opts.cfilter, interface_opts.descr, error_message);
2226     dfilter_free(rfcode);
2227   } else {
2228     cmdarg_err(
2229       "Invalid capture filter \"%s\" for interface %s!\n"
2230       "\n"
2231       "That string isn't a valid capture filter (%s).\n"
2232       "See the User's Guide for a description of the capture filter syntax.",
2233       interface_opts.cfilter, interface_opts.descr, error_message);
2234   }
2235 }
2236
2237
2238 /* capture child tells us we have a new (or the first) capture file */
2239 gboolean
2240 capture_input_new_file(capture_options *capture_opts, gchar *new_file)
2241 {
2242   gboolean is_tempfile;
2243   int  err;
2244
2245   if(capture_opts->state == CAPTURE_PREPARING) {
2246     g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "Capture started!");
2247   }
2248   g_log(LOG_DOMAIN_CAPTURE, G_LOG_LEVEL_MESSAGE, "File: \"%s\"", new_file);
2249
2250   g_assert(capture_opts->state == CAPTURE_PREPARING || capture_opts->state == CAPTURE_RUNNING);
2251
2252   /* free the old filename */
2253   if (capture_opts->save_file != NULL) {
2254
2255     /* we start a new capture file, close the old one (if we had one before) */
2256     if( ((capture_file *) capture_opts->cf)->state != FILE_CLOSED) {
2257       if ( ((capture_file *) capture_opts->cf)->wth != NULL) {
2258         wtap_close(((capture_file *) capture_opts->cf)->wth);
2259       }
2260       ((capture_file *) capture_opts->cf)->state = FILE_CLOSED;
2261     }
2262
2263     g_free(capture_opts->save_file);
2264     is_tempfile = FALSE;
2265   } else {
2266     /* we didn't had a save_file before, must be a tempfile */
2267     is_tempfile = TRUE;
2268   }
2269
2270   /* save the new filename */
2271   capture_opts->save_file = g_strdup(new_file);
2272
2273   /* if we are in real-time mode, open the new file now */
2274   if(do_dissection) {
2275     /* Attempt to open the capture file and set up to read from it. */
2276     switch(cf_open(capture_opts->cf, capture_opts->save_file, is_tempfile, &err)) {
2277     case CF_OK:
2278       break;
2279     case CF_ERROR:
2280       /* Don't unlink (delete) the save file - leave it around,
2281          for debugging purposes. */
2282       g_free(capture_opts->save_file);
2283       capture_opts->save_file = NULL;
2284       return FALSE;
2285     }
2286   }
2287
2288   capture_opts->state = CAPTURE_RUNNING;
2289
2290   return TRUE;
2291 }
2292
2293
2294 /* capture child tells us we have new packets to read */
2295 void
2296 capture_input_new_packets(capture_options *capture_opts, int to_read)
2297 {
2298   gboolean     ret;
2299   int          err;
2300   gchar        *err_info;
2301   gint64       data_offset;
2302   capture_file *cf = capture_opts->cf;
2303   gboolean filtering_tap_listeners;
2304   guint tap_flags;
2305
2306 #ifdef SIGINFO
2307   /*
2308    * Prevent a SIGINFO handler from writing to the standard error while
2309    * we're doing so or writing to the standard output; instead, have it
2310    * just set a flag telling us to print that information when we're done.
2311    */
2312   infodelay = TRUE;
2313 #endif /* SIGINFO */
2314
2315   /* Do we have any tap listeners with filters? */
2316   filtering_tap_listeners = have_filtering_tap_listeners();
2317
2318   /* Get the union of the flags for all tap listeners. */
2319   tap_flags = union_of_tap_listener_flags();
2320
2321   if(do_dissection) {
2322     while (to_read-- && cf->wth) {
2323       wtap_cleareof(cf->wth);
2324       ret = wtap_read(cf->wth, &err, &err_info, &data_offset);
2325       if(ret == FALSE) {
2326         /* read from file failed, tell the capture child to stop */
2327         sync_pipe_stop(capture_opts);
2328         wtap_close(cf->wth);
2329         cf->wth = NULL;
2330       } else {
2331         ret = process_packet(cf, data_offset, wtap_phdr(cf->wth),
2332                              wtap_pseudoheader(cf->wth), wtap_buf_ptr(cf->wth),
2333                              filtering_tap_listeners, tap_flags);
2334       }
2335       if (ret != FALSE) {
2336         /* packet successfully read and gone through the "Read Filter" */
2337         packet_count++;
2338       }
2339     }
2340   } else {
2341     /*
2342      * Dumpcap's doing all the work; we're not doing any dissection.
2343      * Count all the packets it wrote.
2344      */
2345     packet_count += to_read;
2346   }
2347
2348   if (print_packet_counts) {
2349       /* We're printing packet counts. */
2350       if (packet_count != 0) {
2351         fprintf(stderr, "\r%u ", packet_count);
2352         /* stderr could be line buffered */
2353         fflush(stderr);
2354       }
2355   }
2356
2357 #ifdef SIGINFO
2358   /*
2359    * Allow SIGINFO handlers to write.
2360    */
2361   infodelay = FALSE;
2362
2363   /*
2364    * If a SIGINFO handler asked us to write out capture counts, do so.
2365    */
2366   if (infoprint)
2367     report_counts();
2368 #endif /* SIGINFO */
2369 }
2370
2371 static void
2372 report_counts(void)
2373 {
2374   if (!print_packet_counts) {
2375     /* Report the count only if we aren't printing a packet count
2376        as packets arrive. */
2377     fprintf(stderr, "%u packet%s captured\n", packet_count,
2378             plurality(packet_count, "", "s"));
2379   }
2380 #ifdef SIGINFO
2381   infoprint = FALSE; /* we just reported it */
2382 #endif /* SIGINFO */
2383 }
2384
2385 #ifdef SIGINFO
2386 static void
2387 report_counts_siginfo(int signum _U_)
2388 {
2389   int sav_errno = errno;
2390   /* If we've been told to delay printing, just set a flag asking
2391      that we print counts (if we're supposed to), otherwise print
2392      the count of packets captured (if we're supposed to). */
2393   if (infodelay)
2394     infoprint = TRUE;
2395   else
2396     report_counts();
2397   errno = sav_errno;
2398 }
2399 #endif /* SIGINFO */
2400
2401
2402 /* capture child detected any packet drops? */
2403 void
2404 capture_input_drops(capture_options *capture_opts _U_, guint32 dropped)
2405 {
2406   if (print_packet_counts) {
2407     /* We're printing packet counts to stderr.
2408        Send a newline so that we move to the line after the packet count. */
2409     fprintf(stderr, "\n");
2410   }
2411
2412   if (dropped != 0) {
2413     /* We're printing packet counts to stderr.
2414        Send a newline so that we move to the line after the packet count. */
2415     fprintf(stderr, "%u packet%s dropped\n", dropped, plurality(dropped, "", "s"));
2416   }
2417 }
2418
2419
2420 /*
2421  * Capture child closed its side of the pipe, report any error and
2422  * do the required cleanup.
2423  */
2424 void
2425 capture_input_closed(capture_options *capture_opts, gchar *msg)
2426 {
2427   capture_file *cf = (capture_file *) capture_opts->cf;
2428
2429   if (msg != NULL)
2430     fprintf(stderr, "tshark: %s\n", msg);
2431
2432   report_counts();
2433
2434   if(cf != NULL && cf->wth != NULL) {
2435     wtap_close(cf->wth);
2436     if(cf->is_tempfile) {
2437       ws_unlink(cf->filename);
2438     }
2439   }
2440 #ifdef USE_BROKEN_G_MAIN_LOOP
2441   /*g_main_loop_quit(loop);*/
2442   g_main_quit(loop);
2443 #else
2444   loop_running = FALSE;
2445 #endif
2446 }
2447
2448
2449
2450
2451 #ifdef _WIN32
2452 static BOOL WINAPI
2453 capture_cleanup(DWORD ctrltype _U_)
2454 {
2455   /* CTRL_C_EVENT is sort of like SIGINT, CTRL_BREAK_EVENT is unique to
2456      Windows, CTRL_CLOSE_EVENT is sort of like SIGHUP, CTRL_LOGOFF_EVENT
2457      is also sort of like SIGHUP, and CTRL_SHUTDOWN_EVENT is sort of
2458      like SIGTERM at least when the machine's shutting down.
2459
2460      For now, we handle them all as indications that we should clean up
2461      and quit, just as we handle SIGINT, SIGHUP, and SIGTERM in that
2462      way on UNIX.
2463
2464      We must return TRUE so that no other handler - such as one that would
2465      terminate the process - gets called.
2466
2467      XXX - for some reason, typing ^C to TShark, if you run this in
2468      a Cygwin console window in at least some versions of Cygwin,
2469      causes TShark to terminate immediately; this routine gets
2470      called, but the main loop doesn't get a chance to run and
2471      exit cleanly, at least if this is compiled with Microsoft Visual
2472      C++ (i.e., it's a property of the Cygwin console window or Bash;
2473      it happens if TShark is not built with Cygwin - for all I know,
2474      building it with Cygwin may make the problem go away). */
2475
2476   /* tell the capture child to stop */
2477   sync_pipe_stop(&global_capture_opts);
2478
2479   /* don't stop our own loop already here, otherwise status messages and
2480    * cleanup wouldn't be done properly. The child will indicate the stop of
2481    * everything by calling capture_input_closed() later */
2482
2483   return TRUE;
2484 }
2485 #else
2486 static void
2487 capture_cleanup(int signum _U_)
2488 {
2489   /* tell the capture child to stop */
2490   sync_pipe_stop(&global_capture_opts);
2491
2492   /* don't stop our own loop already here, otherwise status messages and
2493    * cleanup wouldn't be done properly. The child will indicate the stop of
2494    * everything by calling capture_input_closed() later */
2495 }
2496 #endif /* _WIN32 */
2497 #endif /* HAVE_LIBPCAP */
2498
2499 static gboolean
2500 process_packet_first_pass(capture_file *cf,
2501                gint64 offset, const struct wtap_pkthdr *whdr,
2502                union wtap_pseudo_header *pseudo_header, const guchar *pd)
2503 {
2504   frame_data fdlocal;
2505   guint32 framenum;
2506   gboolean create_proto_tree = FALSE;
2507   epan_dissect_t edt;
2508   gboolean passed;
2509
2510   /* The frame number of this packet is one more than the count of
2511      frames in this packet. */
2512   framenum = cf->count + 1;
2513
2514   /* If we're not running a display filter and we're not printing any
2515      packet information, we don't need to do a dissection. This means
2516      that all packets can be marked as 'passed'. */
2517   passed = TRUE;
2518
2519   frame_data_init(&fdlocal, framenum, whdr, offset, cum_bytes);
2520
2521   /* If we're going to print packet information, or we're going to
2522      run a read filter, or we're going to process taps, set up to
2523      do a dissection and do so. */
2524   if (do_dissection) {
2525     if (gbl_resolv_flags.mac_name || gbl_resolv_flags.network_name || 
2526         gbl_resolv_flags.transport_name || gbl_resolv_flags.concurrent_dns)
2527       /* Grab any resolved addresses */
2528       host_name_lookup_process(NULL);
2529
2530       if (cf->rfcode)
2531         create_proto_tree = TRUE;
2532
2533     /* The protocol tree will be "visible", i.e., printed, only if we're
2534        printing packet details, which is true if we're printing stuff
2535        ("print_packet_info" is true) and we're in verbose mode ("verbose"
2536        is true). */
2537     epan_dissect_init(&edt, create_proto_tree, FALSE);
2538
2539     /* If we're running a read filter, prime the epan_dissect_t with that
2540        filter. */
2541     if (cf->rfcode)
2542       epan_dissect_prime_dfilter(&edt, cf->rfcode);
2543
2544     frame_data_set_before_dissect(&fdlocal, &cf->elapsed_time,
2545                                   &first_ts, &prev_dis_ts, &prev_cap_ts);
2546
2547     epan_dissect_run(&edt, pseudo_header, pd, &fdlocal, NULL);
2548
2549     /* Run the read filter if we have one. */
2550     if (cf->rfcode)
2551       passed = dfilter_apply_edt(cf->rfcode, &edt);
2552   }
2553
2554   if (passed) {
2555     frame_data_set_after_dissect(&fdlocal, &cum_bytes, &prev_dis_ts);
2556     frame_data_sequence_add(cf->frames, &fdlocal);
2557     cf->count++;
2558   }
2559
2560   if (do_dissection)
2561     epan_dissect_cleanup(&edt);
2562
2563   return passed;
2564 }
2565
2566 static gboolean
2567 process_packet_second_pass(capture_file *cf, frame_data *fdata,
2568                union wtap_pseudo_header *pseudo_header, const guchar *pd,
2569                gboolean filtering_tap_listeners, guint tap_flags)
2570 {
2571   gboolean create_proto_tree;
2572   column_info *cinfo;
2573   epan_dissect_t edt;
2574   gboolean passed;
2575
2576   /* If we're not running a display filter and we're not printing any
2577      packet information, we don't need to do a dissection. This means
2578      that all packets can be marked as 'passed'. */
2579   passed = TRUE;
2580
2581   /* If we're going to print packet information, or we're going to
2582      run a read filter, or we're going to process taps, set up to
2583      do a dissection and do so. */
2584   if (do_dissection) {
2585     if (gbl_resolv_flags.mac_name || gbl_resolv_flags.network_name || 
2586         gbl_resolv_flags.transport_name || gbl_resolv_flags.concurrent_dns)
2587       /* Grab any resolved addresses */
2588       host_name_lookup_process(NULL);
2589
2590     if (cf->rfcode || verbose || filtering_tap_listeners ||
2591         (tap_flags & TL_REQUIRES_PROTO_TREE) || have_custom_cols(&cf->cinfo))
2592       create_proto_tree = TRUE;
2593     else
2594       create_proto_tree = FALSE;
2595
2596     /* The protocol tree will be "visible", i.e., printed, only if we're
2597        printing packet details, which is true if we're printing stuff
2598        ("print_packet_info" is true) and we're in verbose mode ("verbose"
2599        is true). */
2600     epan_dissect_init(&edt, create_proto_tree, print_packet_info && verbose);
2601
2602     /* If we're running a read filter, prime the epan_dissect_t with that
2603        filter. */
2604     if (cf->rfcode)
2605       epan_dissect_prime_dfilter(&edt, cf->rfcode);
2606
2607     col_custom_prime_edt(&edt, &cf->cinfo);
2608
2609     tap_queue_init(&edt);
2610
2611     /* We only need the columns if either
2612
2613          1) some tap needs the columns
2614
2615        or
2616
2617          2) we're printing packet info but we're *not* verbose; in verbose
2618             mode, we print the protocol tree, not the protocol summary. */
2619     if ((tap_flags & TL_REQUIRES_COLUMNS) || (print_packet_info && !verbose))
2620       cinfo = &cf->cinfo;
2621     else
2622       cinfo = NULL;
2623
2624     epan_dissect_run(&edt, pseudo_header, pd, fdata, cinfo);
2625
2626     tap_push_tapped_queue(&edt);
2627
2628     /* Run the read filter if we have one. */
2629     if (cf->rfcode)
2630       passed = dfilter_apply_edt(cf->rfcode, &edt);
2631   }
2632
2633   if (passed) {
2634     /* Process this packet. */
2635     if (print_packet_info) {
2636       /* We're printing packet information; print the information for
2637          this packet. */
2638       if (do_dissection)
2639         print_packet(cf, &edt);
2640       else
2641         print_packet(cf, NULL);
2642
2643       /* The ANSI C standard does not appear to *require* that a line-buffered
2644          stream be flushed to the host environment whenever a newline is
2645          written, it just says that, on such a stream, characters "are
2646          intended to be transmitted to or from the host environment as a
2647          block when a new-line character is encountered".
2648
2649          The Visual C++ 6.0 C implementation doesn't do what is intended;
2650          even if you set a stream to be line-buffered, it still doesn't
2651          flush the buffer at the end of every line.
2652
2653          So, if the "-l" flag was specified, we flush the standard output
2654          at the end of a packet.  This will do the right thing if we're
2655          printing packet summary lines, and, as we print the entire protocol
2656          tree for a single packet without waiting for anything to happen,
2657          it should be as good as line-buffered mode if we're printing
2658          protocol trees.  (The whole reason for the "-l" flag in either
2659          tcpdump or TShark is to allow the output of a live capture to
2660          be piped to a program or script and to have that script see the
2661          information for the packet as soon as it's printed, rather than
2662          having to wait until a standard I/O buffer fills up. */
2663       if (line_buffered)
2664         fflush(stdout);
2665
2666       if (ferror(stdout)) {
2667         show_print_file_io_error(errno);
2668         exit(2);
2669       }
2670     }
2671   }
2672
2673   if (do_dissection) {
2674     epan_dissect_cleanup(&edt);
2675   }
2676   return passed;
2677 }
2678
2679 static int
2680 load_cap_file(capture_file *cf, char *save_file, int out_file_type,
2681     gboolean out_file_name_res, int max_packet_count, gint64 max_byte_count)
2682 {
2683   gint         linktype;
2684   int          snapshot_length;
2685   wtap_dumper *pdh;
2686   guint32      framenum;
2687   int          err;
2688   gchar        *err_info = NULL;
2689   gint64       data_offset;
2690   char         *save_file_string = NULL;
2691   gboolean     filtering_tap_listeners;
2692   guint        tap_flags;
2693   wtapng_section_t *shb_hdr;
2694   wtapng_iface_descriptions_t *idb_inf;
2695   char         appname[100];
2696
2697   shb_hdr = wtap_file_get_shb_info(cf->wth);
2698   idb_inf = wtap_file_get_idb_info(cf->wth);
2699 #ifdef PCAP_NG_DEFAULT
2700   if (idb_inf->number_of_interfaces > 0) {
2701     linktype = WTAP_ENCAP_PER_PACKET;
2702   } else {
2703     linktype = wtap_file_encap(cf->wth);
2704   }
2705 #else
2706   linktype = wtap_file_encap(cf->wth);
2707 #endif
2708   if (save_file != NULL) {
2709     /* Get a string that describes what we're writing to */
2710     save_file_string = output_file_description(save_file);
2711
2712     /* Set up to write to the capture file. */
2713     snapshot_length = wtap_snapshot_length(cf->wth);
2714     if (snapshot_length == 0) {
2715       /* Snapshot length of input file not known. */
2716       snapshot_length = WTAP_MAX_PACKET_SIZE;
2717     }
2718     /* If we don't have an application name add Tshark */
2719     if(shb_hdr->shb_user_appl == NULL) {
2720         g_snprintf(appname, sizeof(appname), "TShark " VERSION "%s", wireshark_svnversion);
2721         shb_hdr->shb_user_appl = appname;
2722     }
2723
2724     pdh = wtap_dump_open_ng(save_file, out_file_type, linktype, snapshot_length,
2725         FALSE /* compressed */, shb_hdr, idb_inf, &err);
2726
2727     g_free(idb_inf);
2728     idb_inf = NULL;
2729
2730     if (pdh == NULL) {
2731       /* We couldn't set up to write to the capture file. */
2732       switch (err) {
2733
2734       case WTAP_ERR_UNSUPPORTED_FILE_TYPE:
2735         cmdarg_err("Capture files can't be written in that format.");
2736         break;
2737
2738       case WTAP_ERR_UNSUPPORTED_ENCAP:
2739       case WTAP_ERR_ENCAP_PER_PACKET_UNSUPPORTED:
2740         cmdarg_err("The capture file being read can't be written as a "
2741           "\"%s\" file.", wtap_file_type_short_string(out_file_type));
2742         break;
2743
2744       case WTAP_ERR_CANT_OPEN:
2745         cmdarg_err("The %s couldn't be created for some "
2746           "unknown reason.", save_file_string);
2747         break;
2748
2749       case WTAP_ERR_SHORT_WRITE:
2750         cmdarg_err("A full header couldn't be written to the %s.",
2751                    save_file_string);
2752         break;
2753
2754       default:
2755         cmdarg_err("The %s could not be created: %s.", save_file_string,
2756                    wtap_strerror(err));
2757         break;
2758       }
2759       goto out;
2760     }
2761   } else {
2762     if (print_packet_info) {
2763       if (!write_preamble(cf)) {
2764         err = errno;
2765         show_print_file_io_error(err);
2766         goto out;
2767       }
2768     }
2769     pdh = NULL;
2770   }
2771
2772   if (pdh && out_file_name_res) {
2773     if (!wtap_dump_set_addrinfo_list(pdh, get_addrinfo_list())) {
2774       cmdarg_err("The file format \"%s\" doesn't support name resolution information.",
2775                  wtap_file_type_short_string(out_file_type));
2776     }
2777   }
2778
2779   /* Do we have any tap listeners with filters? */
2780   filtering_tap_listeners = have_filtering_tap_listeners();
2781
2782   /* Get the union of the flags for all tap listeners. */
2783   tap_flags = union_of_tap_listener_flags();
2784
2785   if (perform_two_pass_analysis) {
2786     frame_data *fdata;
2787     int old_max_packet_count = max_packet_count;
2788
2789     /* Allocate a frame_data_sequence for all the frames. */
2790     cf->frames = new_frame_data_sequence();
2791
2792     while (wtap_read(cf->wth, &err, &err_info, &data_offset)) {
2793       if (process_packet_first_pass(cf, data_offset, wtap_phdr(cf->wth),
2794                          wtap_pseudoheader(cf->wth), wtap_buf_ptr(cf->wth))) {
2795         /* Stop reading if we have the maximum number of packets;
2796          * When the -c option has not been used, max_packet_count
2797          * starts at 0, which practically means, never stop reading.
2798          * (unless we roll over max_packet_count ?)
2799          */
2800         if( (--max_packet_count == 0) || (max_byte_count != 0 && data_offset >= max_byte_count)) {
2801           err = 0; /* This is not an error */
2802           break;
2803         }
2804       }
2805     }
2806
2807     /* Close the sequential I/O side, to free up memory it requires. */
2808     wtap_sequential_close(cf->wth);
2809
2810     /* Allow the protocol dissectors to free up memory that they
2811      * don't need after the sequential run-through of the packets. */
2812     postseq_cleanup_all_protocols();
2813
2814     max_packet_count = old_max_packet_count;
2815
2816     for (framenum = 1; err == 0 && framenum <= cf->count; framenum++) {
2817       fdata = frame_data_sequence_find(cf->frames, framenum);
2818       if (wtap_seek_read(cf->wth, fdata->file_off, &cf->pseudo_header,
2819           cf->pd, fdata->cap_len, &err, &err_info)) {
2820         if (process_packet_second_pass(cf, fdata,
2821                            &cf->pseudo_header, cf->pd,
2822                            filtering_tap_listeners, tap_flags)) {
2823           /* Either there's no read filtering or this packet passed the
2824              filter, so, if we're writing to a capture file, write
2825              this packet out. */
2826           if (pdh != NULL) {
2827             if (!wtap_dump(pdh, wtap_phdr(cf->wth),
2828                            wtap_pseudoheader(cf->wth), wtap_buf_ptr(cf->wth),
2829                            &err)) {
2830               /* Error writing to a capture file */
2831               switch (err) {
2832
2833               case WTAP_ERR_UNSUPPORTED_ENCAP:
2834                 /*
2835                  * This is a problem with the particular frame we're writing;
2836                  * note that, and give the frame number.
2837                  *
2838                  * XXX - framenum is not necessarily the frame number in
2839                  * the input file if there was a read filter.
2840                  */
2841                 fprintf(stderr,
2842                         "Frame %u of \"%s\" has a network type that can't be saved in a \"%s\" file.\n",
2843                         framenum, cf->filename,
2844                         wtap_file_type_short_string(out_file_type));
2845                 break;
2846
2847               default:
2848                 show_capture_file_io_error(save_file, err, FALSE);
2849                 break;
2850               }
2851               wtap_dump_close(pdh, &err);
2852               g_free(shb_hdr);
2853               exit(2);
2854             }
2855           }
2856           /* Stop reading if we have the maximum number of packets;
2857            * When the -c option has not been used, max_packet_count
2858            * starts at 0, which practically means, never stop reading.
2859            * (unless we roll over max_packet_count ?)
2860            */
2861           if( (--max_packet_count == 0) || (max_byte_count != 0 && data_offset >= max_byte_count)) {
2862             err = 0; /* This is not an error */
2863             break;
2864           }
2865         }
2866       }
2867     }
2868   }
2869   else {
2870     framenum = 0;
2871     while (wtap_read(cf->wth, &err, &err_info, &data_offset)) {
2872       framenum++;
2873
2874       if (process_packet(cf, data_offset, wtap_phdr(cf->wth),
2875                          wtap_pseudoheader(cf->wth), wtap_buf_ptr(cf->wth),
2876                          filtering_tap_listeners, tap_flags)) {
2877         /* Either there's no read filtering or this packet passed the
2878            filter, so, if we're writing to a capture file, write
2879            this packet out. */
2880         if (pdh != NULL) {
2881           if (!wtap_dump(pdh, wtap_phdr(cf->wth),
2882                          wtap_pseudoheader(cf->wth), wtap_buf_ptr(cf->wth),
2883                          &err)) {
2884             /* Error writing to a capture file */
2885             switch (err) {
2886
2887             case WTAP_ERR_UNSUPPORTED_ENCAP:
2888               /*
2889                * This is a problem with the particular frame we're writing;
2890                * note that, and give the frame number.
2891                */
2892               fprintf(stderr,
2893                       "Frame %u of \"%s\" has a network type that can't be saved in a \"%s\" file.\n",
2894                       framenum, cf->filename,
2895                       wtap_file_type_short_string(out_file_type));
2896               break;
2897
2898             default:
2899               show_capture_file_io_error(save_file, err, FALSE);
2900               break;
2901             }
2902             wtap_dump_close(pdh, &err);
2903             g_free(shb_hdr);
2904             exit(2);
2905           }
2906         }
2907         /* Stop reading if we have the maximum number of packets;
2908          * When the -c option has not been used, max_packet_count
2909          * starts at 0, which practically means, never stop reading.
2910          * (unless we roll over max_packet_count ?)
2911          */
2912         if( (--max_packet_count == 0) || (max_byte_count != 0 && data_offset >= max_byte_count)) {
2913           err = 0; /* This is not an error */
2914           break;
2915         }
2916       }
2917     }
2918   }
2919
2920   if (err != 0) {
2921     /*
2922      * Print a message noting that the read failed somewhere along the line.
2923      *
2924      * If we're printing packet data, and the standard output and error are
2925      * going to the same place, flush the standard output, so everything
2926      * buffered up is written, and then print a newline to the standard error
2927      * before printing the error message, to separate it from the packet
2928      * data.  (Alas, that only works on UN*X; st_dev is meaningless, and
2929      * the _fstat() documentation at Microsoft doesn't indicate whether
2930      * st_ino is even supported.)
2931      */
2932 #ifndef _WIN32
2933     if (print_packet_info) {
2934       struct stat stat_stdout, stat_stderr;
2935
2936       if (fstat(1, &stat_stdout) == 0 && fstat(2, &stat_stderr) == 0) {
2937         if (stat_stdout.st_dev == stat_stderr.st_dev &&
2938             stat_stdout.st_ino == stat_stderr.st_ino) {
2939           fflush(stdout);
2940           fprintf(stderr, "\n");
2941         }
2942       }
2943     }
2944 #endif
2945     switch (err) {
2946
2947     case WTAP_ERR_UNSUPPORTED:
2948       cmdarg_err("The file \"%s\" contains record data that TShark doesn't support.\n(%s)",
2949                  cf->filename, err_info);
2950       g_free(err_info);
2951       break;
2952
2953     case WTAP_ERR_UNSUPPORTED_ENCAP:
2954       cmdarg_err("The file \"%s\" has a packet with a network type that TShark doesn't support.\n(%s)",
2955                  cf->filename, err_info);
2956       g_free(err_info);
2957       break;
2958
2959     case WTAP_ERR_CANT_READ:
2960       cmdarg_err("An attempt to read from the file \"%s\" failed for some unknown reason.",
2961                  cf->filename);
2962       break;
2963
2964     case WTAP_ERR_SHORT_READ:
2965       cmdarg_err("The file \"%s\" appears to have been cut short in the middle of a packet.",
2966                  cf->filename);
2967       break;
2968
2969     case WTAP_ERR_BAD_FILE:
2970       cmdarg_err("The file \"%s\" appears to be damaged or corrupt.\n(%s)",
2971                  cf->filename, err_info);
2972       g_free(err_info);
2973       break;
2974
2975     case WTAP_ERR_DECOMPRESS:
2976       cmdarg_err("The compressed file \"%s\" appears to be damaged or corrupt.\n"
2977                  "(%s)", cf->filename, err_info);
2978       break;
2979
2980     default:
2981       cmdarg_err("An error occurred while reading the file \"%s\": %s.",
2982                  cf->filename, wtap_strerror(err));
2983       break;
2984     }
2985     if (save_file != NULL) {
2986       /* Now close the capture file. */
2987       if (!wtap_dump_close(pdh, &err))
2988         show_capture_file_io_error(save_file, err, TRUE);
2989     }
2990   } else {
2991     if (save_file != NULL) {
2992       /* Now close the capture file. */
2993       if (!wtap_dump_close(pdh, &err))
2994         show_capture_file_io_error(save_file, err, TRUE);
2995     } else {
2996       if (print_packet_info) {
2997         if (!write_finale()) {
2998           err = errno;
2999           show_print_file_io_error(err);
3000         }
3001       }
3002     }
3003   }
3004
3005 out:
3006   wtap_close(cf->wth);
3007   cf->wth = NULL;
3008
3009   g_free(save_file_string);
3010   g_free(shb_hdr);
3011
3012   return err;
3013 }
3014
3015 static gboolean
3016 process_packet(capture_file *cf, gint64 offset, const struct wtap_pkthdr *whdr,
3017                union wtap_pseudo_header *pseudo_header, const guchar *pd,
3018                gboolean filtering_tap_listeners, guint tap_flags)
3019 {
3020   frame_data fdata;
3021   gboolean create_proto_tree;
3022   column_info *cinfo;
3023   epan_dissect_t edt;
3024   gboolean passed;
3025
3026   /* Count this packet. */
3027   cf->count++;
3028
3029   /* If we're not running a display filter and we're not printing any
3030      packet information, we don't need to do a dissection. This means
3031      that all packets can be marked as 'passed'. */
3032   passed = TRUE;
3033
3034   frame_data_init(&fdata, cf->count, whdr, offset, cum_bytes);
3035
3036   /* If we're going to print packet information, or we're going to
3037      run a read filter, or we're going to process taps, set up to
3038      do a dissection and do so. */
3039   if (do_dissection) {
3040     if (print_packet_info && (gbl_resolv_flags.mac_name || gbl_resolv_flags.network_name || 
3041         gbl_resolv_flags.transport_name || gbl_resolv_flags.concurrent_dns))
3042       /* Grab any resolved addresses */
3043       host_name_lookup_process(NULL);
3044
3045     if (cf->rfcode || verbose || filtering_tap_listeners ||
3046         (tap_flags & TL_REQUIRES_PROTO_TREE) || have_custom_cols(&cf->cinfo))
3047       create_proto_tree = TRUE;
3048     else
3049       create_proto_tree = FALSE;
3050
3051     /* The protocol tree will be "visible", i.e., printed, only if we're
3052        printing packet details, which is true if we're printing stuff
3053        ("print_packet_info" is true) and we're in verbose mode ("verbose"
3054        is true). */
3055     epan_dissect_init(&edt, create_proto_tree, print_packet_info && verbose);
3056
3057     /* If we're running a read filter, prime the epan_dissect_t with that
3058        filter. */
3059     if (cf->rfcode)
3060       epan_dissect_prime_dfilter(&edt, cf->rfcode);
3061
3062     col_custom_prime_edt(&edt, &cf->cinfo);
3063
3064     tap_queue_init(&edt);
3065
3066     /* We only need the columns if either
3067
3068          1) some tap needs the columns
3069
3070        or
3071
3072          2) we're printing packet info but we're *not* verbose; in verbose
3073             mode, we print the protocol tree, not the protocol summary. */
3074     if ((tap_flags & TL_REQUIRES_COLUMNS) || (print_packet_info && !verbose))
3075       cinfo = &cf->cinfo;
3076     else
3077       cinfo = NULL;
3078
3079     frame_data_set_before_dissect(&fdata, &cf->elapsed_time,
3080                                   &first_ts, &prev_dis_ts, &prev_cap_ts);
3081
3082     epan_dissect_run(&edt, pseudo_header, pd, &fdata, cinfo);
3083
3084     tap_push_tapped_queue(&edt);
3085
3086     /* Run the read filter if we have one. */
3087     if (cf->rfcode)
3088       passed = dfilter_apply_edt(cf->rfcode, &edt);
3089   }
3090
3091   if (passed) {
3092     frame_data_set_after_dissect(&fdata, &cum_bytes, &prev_dis_ts);
3093
3094     /* Process this packet. */
3095     if (print_packet_info) {
3096       /* We're printing packet information; print the information for
3097          this packet. */
3098       if (do_dissection)
3099         print_packet(cf, &edt);
3100       else
3101         print_packet(cf, NULL);
3102
3103       /* The ANSI C standard does not appear to *require* that a line-buffered
3104          stream be flushed to the host environment whenever a newline is
3105          written, it just says that, on such a stream, characters "are
3106          intended to be transmitted to or from the host environment as a
3107          block when a new-line character is encountered".
3108
3109          The Visual C++ 6.0 C implementation doesn't do what is intended;
3110          even if you set a stream to be line-buffered, it still doesn't
3111          flush the buffer at the end of every line.
3112
3113          So, if the "-l" flag was specified, we flush the standard output
3114          at the end of a packet.  This will do the right thing if we're
3115          printing packet summary lines, and, as we print the entire protocol
3116          tree for a single packet without waiting for anything to happen,
3117          it should be as good as line-buffered mode if we're printing
3118          protocol trees.  (The whole reason for the "-l" flag in either
3119          tcpdump or TShark is to allow the output of a live capture to
3120          be piped to a program or script and to have that script see the
3121          information for the packet as soon as it's printed, rather than
3122          having to wait until a standard I/O buffer fills up. */
3123       if (line_buffered)
3124         fflush(stdout);
3125
3126       if (ferror(stdout)) {
3127         show_print_file_io_error(errno);
3128         exit(2);
3129       }
3130     }
3131   }
3132
3133   if (do_dissection) {
3134     epan_dissect_cleanup(&edt);
3135     frame_data_cleanup(&fdata);
3136   }
3137   return passed;
3138 }
3139
3140 static gboolean
3141 write_preamble(capture_file *cf)
3142 {
3143   switch (output_action) {
3144
3145   case WRITE_TEXT:
3146     return print_preamble(print_stream, cf ? cf->filename : NULL);
3147
3148   case WRITE_XML:
3149     if (verbose)
3150       write_pdml_preamble(stdout, cf ? cf->filename : NULL);
3151     else
3152       write_psml_preamble(stdout);
3153     return !ferror(stdout);
3154
3155   case WRITE_FIELDS:
3156     write_fields_preamble(output_fields, stdout);
3157     return !ferror(stdout);
3158
3159   default:
3160     g_assert_not_reached();
3161     return FALSE;
3162   }
3163 }
3164
3165 static char *
3166 get_line_buf(size_t len)
3167 {
3168   static char *line_bufp = NULL;
3169   static size_t line_buf_len = 256;
3170   size_t new_line_buf_len;
3171
3172   for (new_line_buf_len = line_buf_len; len > new_line_buf_len;
3173        new_line_buf_len *= 2)
3174     ;
3175   if (line_bufp == NULL) {
3176     line_buf_len = new_line_buf_len;
3177     line_bufp = g_malloc(line_buf_len + 1);
3178   } else {
3179     if (new_line_buf_len > line_buf_len) {
3180       line_buf_len = new_line_buf_len;
3181       line_bufp = g_realloc(line_bufp, line_buf_len + 1);
3182     }
3183   }
3184   return line_bufp;
3185 }
3186
3187 static gboolean
3188 print_columns(capture_file *cf)
3189 {
3190   char *line_bufp;
3191   int i;
3192   size_t buf_offset;
3193   size_t column_len;
3194
3195   line_bufp = get_line_buf(256);
3196   buf_offset = 0;
3197   *line_bufp = '\0';
3198   for (i = 0; i < cf->cinfo.num_cols; i++) {
3199     /* Skip columns not marked as visible. */
3200     if (!get_column_visible(i))
3201       continue;
3202     switch (cf->cinfo.col_fmt[i]) {
3203     case COL_NUMBER:
3204 #ifdef HAVE_LIBPCAP
3205       /*
3206        * Don't print this if we're doing a live capture from a network
3207        * interface - if we're doing a live capture, you won't be
3208        * able to look at the capture in the future (it's not being
3209        * saved anywhere), so the frame numbers are unlikely to be
3210        * useful.
3211        *
3212        * (XXX - it might be nice to be able to save and print at
3213        * the same time, sort of like an "Update list of packets
3214        * in real time" capture in Wireshark.)
3215        */
3216       if (global_capture_opts.ifaces->len > 0)
3217         continue;
3218 #endif
3219       column_len = strlen(cf->cinfo.col_data[i]);
3220       if (column_len < 3)
3221         column_len = 3;
3222       line_bufp = get_line_buf(buf_offset + column_len);
3223       g_snprintf(line_bufp + buf_offset, (int)column_len + 1, "%3s", cf->cinfo.col_data[i]);
3224       break;
3225
3226     case COL_CLS_TIME:
3227     case COL_REL_TIME:
3228     case COL_ABS_TIME:
3229     case COL_ABS_DATE_TIME:
3230     case COL_UTC_TIME:
3231     case COL_UTC_DATE_TIME: /* XXX - wider */
3232       column_len = strlen(cf->cinfo.col_data[i]);
3233       if (column_len < 10)
3234         column_len = 10;
3235       line_bufp = get_line_buf(buf_offset + column_len);
3236       g_snprintf(line_bufp + buf_offset, (int)column_len + 1, "%10s", cf->cinfo.col_data[i]);
3237       break;
3238
3239     case COL_DEF_SRC:
3240     case COL_RES_SRC:
3241     case COL_UNRES_SRC:
3242     case COL_DEF_DL_SRC:
3243     case COL_RES_DL_SRC:
3244     case COL_UNRES_DL_SRC:
3245     case COL_DEF_NET_SRC:
3246     case COL_RES_NET_SRC:
3247     case COL_UNRES_NET_SRC:
3248       column_len = strlen(cf->cinfo.col_data[i]);
3249       if (column_len < 12)
3250         column_len = 12;
3251       line_bufp = get_line_buf(buf_offset + column_len);
3252       g_snprintf(line_bufp + buf_offset, (int)column_len + 1, "%12s", cf->cinfo.col_data[i]);
3253       break;
3254
3255     case COL_DEF_DST:
3256     case COL_RES_DST:
3257     case COL_UNRES_DST:
3258     case COL_DEF_DL_DST:
3259     case COL_RES_DL_DST:
3260     case COL_UNRES_DL_DST:
3261     case COL_DEF_NET_DST:
3262     case COL_RES_NET_DST:
3263     case COL_UNRES_NET_DST:
3264       column_len = strlen(cf->cinfo.col_data[i]);
3265       if (column_len < 12)
3266         column_len = 12;
3267       line_bufp = get_line_buf(buf_offset + column_len);
3268       g_snprintf(line_bufp + buf_offset, (int)column_len + 1, "%-12s", cf->cinfo.col_data[i]);
3269       break;
3270
3271     default:
3272       column_len = strlen(cf->cinfo.col_data[i]);
3273       line_bufp = get_line_buf(buf_offset + column_len);
3274       g_strlcat(line_bufp + buf_offset, cf->cinfo.col_data[i], column_len + 1);
3275       break;
3276     }
3277     buf_offset += column_len;
3278     if (i != cf->cinfo.num_cols - 1) {
3279       /*
3280        * This isn't the last column, so we need to print a
3281        * separator between this column and the next.
3282        *
3283        * If we printed a network source and are printing a
3284        * network destination of the same type next, separate
3285        * them with " -> "; if we printed a network destination
3286        * and are printing a network source of the same type
3287        * next, separate them with " <- "; otherwise separate them
3288        * with a space.
3289        *
3290        * We add enough space to the buffer for " <- " or " -> ",
3291        * even if we're only adding " ".
3292        */
3293       line_bufp = get_line_buf(buf_offset + 4);
3294       switch (cf->cinfo.col_fmt[i]) {
3295
3296       case COL_DEF_SRC:
3297       case COL_RES_SRC:
3298       case COL_UNRES_SRC:
3299         switch (cf->cinfo.col_fmt[i + 1]) {
3300
3301         case COL_DEF_DST:
3302         case COL_RES_DST:
3303         case COL_UNRES_DST:
3304           g_strlcat(line_bufp + buf_offset, " -> ", 5);
3305           buf_offset += 4;
3306           break;
3307
3308         default:
3309           g_strlcat(line_bufp + buf_offset, " ", 5);
3310           buf_offset += 1;
3311           break;
3312         }
3313         break;
3314
3315       case COL_DEF_DL_SRC:
3316       case COL_RES_DL_SRC:
3317       case COL_UNRES_DL_SRC:
3318         switch (cf->cinfo.col_fmt[i + 1]) {
3319
3320         case COL_DEF_DL_DST:
3321         case COL_RES_DL_DST:
3322         case COL_UNRES_DL_DST:
3323           g_strlcat(line_bufp + buf_offset, " -> ", 5);
3324           buf_offset += 4;
3325           break;
3326
3327         default:
3328           g_strlcat(line_bufp + buf_offset, " ", 5);
3329           buf_offset += 1;
3330           break;
3331         }
3332         break;
3333
3334       case COL_DEF_NET_SRC:
3335       case COL_RES_NET_SRC:
3336       case COL_UNRES_NET_SRC:
3337         switch (cf->cinfo.col_fmt[i + 1]) {
3338
3339         case COL_DEF_NET_DST:
3340         case COL_RES_NET_DST:
3341         case COL_UNRES_NET_DST:
3342           g_strlcat(line_bufp + buf_offset, " -> ", 5);
3343           buf_offset += 4;
3344           break;
3345
3346         default:
3347           g_strlcat(line_bufp + buf_offset, " ", 5);
3348           buf_offset += 1;
3349           break;
3350         }
3351         break;
3352
3353       case COL_DEF_DST:
3354       case COL_RES_DST:
3355       case COL_UNRES_DST:
3356         switch (cf->cinfo.col_fmt[i + 1]) {
3357
3358         case COL_DEF_SRC:
3359         case COL_RES_SRC:
3360         case COL_UNRES_SRC:
3361           g_strlcat(line_bufp + buf_offset, " <- ", 5);
3362           buf_offset += 4;
3363           break;
3364
3365         default:
3366           g_strlcat(line_bufp + buf_offset, " ", 5);
3367           buf_offset += 1;
3368           break;
3369         }
3370         break;
3371
3372       case COL_DEF_DL_DST:
3373       case COL_RES_DL_DST:
3374       case COL_UNRES_DL_DST:
3375         switch (cf->cinfo.col_fmt[i + 1]) {
3376
3377         case COL_DEF_DL_SRC:
3378         case COL_RES_DL_SRC:
3379         case COL_UNRES_DL_SRC:
3380           g_strlcat(line_bufp + buf_offset, " <- ", 5);
3381           buf_offset += 4;
3382           break;
3383
3384         default:
3385           g_strlcat(line_bufp + buf_offset, " ", 5);
3386           buf_offset += 1;
3387           break;
3388         }
3389         break;
3390
3391       case COL_DEF_NET_DST:
3392       case COL_RES_NET_DST:
3393       case COL_UNRES_NET_DST:
3394         switch (cf->cinfo.col_fmt[i + 1]) {
3395
3396         case COL_DEF_NET_SRC:
3397         case COL_RES_NET_SRC:
3398         case COL_UNRES_NET_SRC:
3399           g_strlcat(line_bufp + buf_offset, " <- ", 5);
3400           buf_offset += 4;
3401           break;
3402
3403         default:
3404           g_strlcat(line_bufp + buf_offset, " ", 5);
3405           buf_offset += 1;
3406           break;
3407         }
3408         break;
3409
3410       default:
3411         g_strlcat(line_bufp + buf_offset, " ", 5);
3412         buf_offset += 1;
3413         break;
3414       }
3415     }
3416   }
3417   return print_line(print_stream, 0, line_bufp);
3418 }
3419
3420 static gboolean
3421 print_packet(capture_file *cf, epan_dissect_t *edt)
3422 {
3423   print_args_t  print_args;
3424
3425   if (verbose) {
3426     /* Print the information in the protocol tree. */
3427     switch (output_action) {
3428
3429     case WRITE_TEXT:
3430       /* Only initialize the fields that are actually used in proto_tree_print.
3431        * This is particularly important for .range, as that's heap memory which
3432        * we would otherwise have to g_free().
3433       print_args.to_file = TRUE;
3434       print_args.format = print_format;
3435       print_args.print_summary = !verbose;
3436       print_args.print_formfeed = FALSE;
3437       packet_range_init(&print_args.range);
3438       */
3439       print_args.print_hex = verbose && print_hex;
3440       print_args.print_dissections = verbose ? print_dissections_expanded : print_dissections_none;
3441
3442       if (!proto_tree_print(&print_args, edt, print_stream))
3443         return FALSE;
3444       if (!print_hex) {
3445         /* "print_hex_data()" will put out a leading blank line, as well
3446          as a trailing one; print one here, to separate the packets,
3447          only if "print_hex_data()" won't be called. */
3448         if (!print_line(print_stream, 0, separator))
3449           return FALSE;
3450       }
3451       break;
3452
3453     case WRITE_XML:
3454       proto_tree_write_pdml(edt, stdout);
3455       printf("\n");
3456       return !ferror(stdout);
3457     case WRITE_FIELDS:
3458       proto_tree_write_fields(output_fields, edt, stdout);
3459       printf("\n");
3460       return !ferror(stdout);
3461     }
3462   } else {
3463     /* Just fill in the columns. */
3464     epan_dissect_fill_in_columns(edt, FALSE, TRUE);
3465
3466     /* Now print them. */
3467     switch (output_action) {
3468
3469     case WRITE_TEXT:
3470         if (!print_columns(cf))
3471           return FALSE;
3472         break;
3473
3474     case WRITE_XML:
3475         proto_tree_write_psml(edt, stdout);
3476         return !ferror(stdout);
3477     case WRITE_FIELDS: /*No non-verbose "fields" format */
3478         g_assert_not_reached();
3479         break;
3480     }
3481   }
3482   if (print_hex) {
3483     if (!print_hex_data(print_stream, edt))
3484       return FALSE;
3485     if (!print_line(print_stream, 0, ""))
3486       return FALSE;
3487   }
3488   return TRUE;
3489 }
3490
3491 static gboolean
3492 write_finale(void)
3493 {
3494   switch (output_action) {
3495
3496   case WRITE_TEXT:
3497     return print_finale(print_stream);
3498
3499   case WRITE_XML:
3500     if (verbose)
3501       write_pdml_finale(stdout);
3502     else
3503       write_psml_finale(stdout);
3504     return !ferror(stdout);
3505
3506   case WRITE_FIELDS:
3507     write_fields_finale(output_fields, stdout);
3508     return !ferror(stdout);
3509
3510   default:
3511     g_assert_not_reached();
3512     return FALSE;
3513   }
3514 }
3515
3516 cf_status_t
3517 cf_open(capture_file *cf, const char *fname, gboolean is_tempfile, int *err)
3518 {
3519   wtap       *wth;
3520   gchar       *err_info;
3521   char        err_msg[2048+1];
3522
3523   wth = wtap_open_offline(fname, err, &err_info, perform_two_pass_analysis);
3524   if (wth == NULL)
3525     goto fail;
3526
3527   /* The open succeeded.  Fill in the information for this file. */
3528
3529   /* Cleanup all data structures used for dissection. */
3530   cleanup_dissection();
3531   /* Initialize all data structures used for dissection. */
3532   init_dissection();
3533
3534   cf->wth = wth;
3535   cf->f_datalen = 0; /* not used, but set it anyway */
3536
3537   /* Set the file name because we need it to set the follow stream filter.
3538      XXX - is that still true?  We need it for other reasons, though,
3539      in any case. */
3540   cf->filename = g_strdup(fname);
3541
3542   /* Indicate whether it's a permanent or temporary file. */
3543   cf->is_tempfile = is_tempfile;
3544
3545   /* No user changes yet. */
3546   cf->unsaved_changes = FALSE;
3547
3548   cf->cd_t      = wtap_file_type(cf->wth);
3549   cf->count     = 0;
3550   cf->drops_known = FALSE;
3551   cf->drops     = 0;
3552   cf->snap      = wtap_snapshot_length(cf->wth);
3553   if (cf->snap == 0) {
3554     /* Snapshot length not known. */
3555     cf->has_snap = FALSE;
3556     cf->snap = WTAP_MAX_PACKET_SIZE;
3557   } else
3558     cf->has_snap = TRUE;
3559   nstime_set_zero(&cf->elapsed_time);
3560   nstime_set_unset(&first_ts);
3561   nstime_set_unset(&prev_dis_ts);
3562   nstime_set_unset(&prev_cap_ts);
3563
3564   cf->state = FILE_READ_IN_PROGRESS;
3565
3566   wtap_set_cb_new_ipv4(cf->wth, add_ipv4_name);
3567   wtap_set_cb_new_ipv6(cf->wth, (wtap_new_ipv6_callback_t) add_ipv6_name);
3568
3569   return CF_OK;
3570
3571 fail:
3572   g_snprintf(err_msg, sizeof err_msg,
3573              cf_open_error_message(*err, err_info, FALSE, cf->cd_t), fname);
3574   cmdarg_err("%s", err_msg);
3575   return CF_ERROR;
3576 }
3577
3578 static void
3579 show_capture_file_io_error(const char *fname, int err, gboolean is_close)
3580 {
3581   char *save_file_string;
3582
3583   save_file_string = output_file_description(fname);
3584
3585   switch (err) {
3586
3587   case ENOSPC:
3588     cmdarg_err("Not all the packets could be written to the %s because there is "
3589                "no space left on the file system.",
3590                save_file_string);
3591     break;
3592
3593 #ifdef EDQUOT
3594   case EDQUOT:
3595     cmdarg_err("Not all the packets could be written to the %s because you are "
3596                "too close to, or over your disk quota.",
3597                save_file_string);
3598   break;
3599 #endif
3600
3601   case WTAP_ERR_CANT_CLOSE:
3602     cmdarg_err("The %s couldn't be closed for some unknown reason.",
3603                save_file_string);
3604     break;
3605
3606   case WTAP_ERR_SHORT_WRITE:
3607     cmdarg_err("Not all the packets could be written to the %s.",
3608                save_file_string);
3609     break;
3610
3611   default:
3612     if (is_close) {
3613       cmdarg_err("The %s could not be closed: %s.", save_file_string,
3614                  wtap_strerror(err));
3615     } else {
3616       cmdarg_err("An error occurred while writing to the %s: %s.",
3617                  save_file_string, wtap_strerror(err));
3618     }
3619     break;
3620   }
3621   g_free(save_file_string);
3622 }
3623
3624 static void
3625 show_print_file_io_error(int err)
3626 {
3627   switch (err) {
3628
3629   case ENOSPC:
3630     cmdarg_err("Not all the packets could be printed because there is "
3631 "no space left on the file system.");
3632     break;
3633
3634 #ifdef EDQUOT
3635   case EDQUOT:
3636     cmdarg_err("Not all the packets could be printed because you are "
3637 "too close to, or over your disk quota.");
3638   break;
3639 #endif
3640
3641   default:
3642     cmdarg_err("An error occurred while printing packets: %s.",
3643       g_strerror(err));
3644     break;
3645   }
3646 }
3647
3648 static const char *
3649 cf_open_error_message(int err, gchar *err_info, gboolean for_writing,
3650                       int file_type)
3651 {
3652   const char *errmsg;
3653   static char errmsg_errno[1024+1];
3654
3655   if (err < 0) {
3656     /* Wiretap error. */
3657     switch (err) {
3658
3659     case WTAP_ERR_NOT_REGULAR_FILE:
3660       errmsg = "The file \"%s\" is a \"special file\" or socket or other non-regular file.";
3661       break;
3662
3663     case WTAP_ERR_RANDOM_OPEN_PIPE:
3664       /* Seen only when opening a capture file for reading. */
3665       errmsg = "The file \"%s\" is a pipe or FIFO; TShark can't read pipe or FIFO files in two-pass mode.";
3666       break;
3667
3668     case WTAP_ERR_FILE_UNKNOWN_FORMAT:
3669       /* Seen only when opening a capture file for reading. */
3670       errmsg = "The file \"%s\" isn't a capture file in a format TShark understands.";
3671       break;
3672
3673     case WTAP_ERR_UNSUPPORTED:
3674       /* Seen only when opening a capture file for reading. */
3675       g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3676                "The file \"%%s\" isn't a capture file in a format TShark understands.\n"
3677                "(%s)", err_info);
3678       g_free(err_info);
3679       errmsg = errmsg_errno;
3680       break;
3681
3682     case WTAP_ERR_CANT_WRITE_TO_PIPE:
3683       /* Seen only when opening a capture file for writing. */
3684       g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3685                  "The file \"%%s\" is a pipe, and \"%s\" capture files can't be "
3686                  "written to a pipe.", wtap_file_type_short_string(file_type));
3687       errmsg = errmsg_errno;
3688       break;
3689
3690     case WTAP_ERR_UNSUPPORTED_FILE_TYPE:
3691       /* Seen only when opening a capture file for writing. */
3692       errmsg = "TShark doesn't support writing capture files in that format.";
3693       break;
3694
3695     case WTAP_ERR_UNSUPPORTED_ENCAP:
3696       if (for_writing) {
3697         g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3698                    "TShark can't save this capture as a \"%s\" file.",
3699                    wtap_file_type_short_string(file_type));
3700       } else {
3701         g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3702                  "The file \"%%s\" is a capture for a network type that TShark doesn't support.\n"
3703                  "(%s)", err_info);
3704         g_free(err_info);
3705       }
3706       errmsg = errmsg_errno;
3707       break;
3708
3709     case WTAP_ERR_ENCAP_PER_PACKET_UNSUPPORTED:
3710       if (for_writing) {
3711         g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3712                    "TShark can't save this capture as a \"%s\" file.",
3713                    wtap_file_type_short_string(file_type));
3714         errmsg = errmsg_errno;
3715       } else
3716         errmsg = "The file \"%s\" is a capture for a network type that TShark doesn't support.";
3717       break;
3718
3719     case WTAP_ERR_BAD_FILE:
3720       /* Seen only when opening a capture file for reading. */
3721       g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3722                "The file \"%%s\" appears to be damaged or corrupt.\n"
3723                "(%s)", err_info);
3724       g_free(err_info);
3725       errmsg = errmsg_errno;
3726       break;
3727
3728     case WTAP_ERR_CANT_OPEN:
3729       if (for_writing)
3730         errmsg = "The file \"%s\" could not be created for some unknown reason.";
3731       else
3732         errmsg = "The file \"%s\" could not be opened for some unknown reason.";
3733       break;
3734
3735     case WTAP_ERR_SHORT_READ:
3736       errmsg = "The file \"%s\" appears to have been cut short"
3737                " in the middle of a packet or other data.";
3738       break;
3739
3740     case WTAP_ERR_SHORT_WRITE:
3741       errmsg = "A full header couldn't be written to the file \"%s\".";
3742       break;
3743
3744     case WTAP_ERR_COMPRESSION_NOT_SUPPORTED:
3745       errmsg = "This file type cannot be written as a compressed file.";
3746       break;
3747
3748     case WTAP_ERR_DECOMPRESS:
3749       /* Seen only when opening a capture file for reading. */
3750       g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3751                  "The compressed file \"%%s\" appears to be damaged or corrupt.\n"
3752                  "(%s)", err_info);
3753       g_free(err_info);
3754       errmsg = errmsg_errno;
3755       break;
3756
3757     default:
3758       g_snprintf(errmsg_errno, sizeof(errmsg_errno),
3759                  "The file \"%%s\" could not be %s: %s.",
3760                  for_writing ? "created" : "opened",
3761                  wtap_strerror(err));
3762       errmsg = errmsg_errno;
3763       break;
3764     }
3765   } else
3766     errmsg = file_open_error_message(err, for_writing);
3767   return errmsg;
3768 }
3769
3770 /*
3771  * Open/create errors are reported with an console message in TShark.
3772  */
3773 static void
3774 open_failure_message(const char *filename, int err, gboolean for_writing)
3775 {
3776   fprintf(stderr, "tshark: ");
3777   fprintf(stderr, file_open_error_message(err, for_writing), filename);
3778   fprintf(stderr, "\n");
3779 }
3780
3781
3782 /*
3783  * General errors are reported with an console message in TShark.
3784  */
3785 static void
3786 failure_message(const char *msg_format, va_list ap)
3787 {
3788   fprintf(stderr, "tshark: ");
3789   vfprintf(stderr, msg_format, ap);
3790   fprintf(stderr, "\n");
3791 }
3792
3793 /*
3794  * Read errors are reported with an console message in TShark.
3795  */
3796 static void
3797 read_failure_message(const char *filename, int err)
3798 {
3799   cmdarg_err("An error occurred while reading from the file \"%s\": %s.",
3800           filename, g_strerror(err));
3801 }
3802
3803 /*
3804  * Write errors are reported with an console message in TShark.
3805  */
3806 static void
3807 write_failure_message(const char *filename, int err)
3808 {
3809   cmdarg_err("An error occurred while writing to the file \"%s\": %s.",
3810           filename, g_strerror(err));
3811 }
3812
3813 /*
3814  * Print to the standard error.  This is a command-line tool, so there's
3815  * no need to pop up a console.
3816  */
3817 void
3818 vfprintf_stderr(const char *fmt, va_list ap)
3819 {
3820   vfprintf(stderr, fmt, ap);
3821 }
3822
3823 void
3824 fprintf_stderr(const char *fmt, ...)
3825 {
3826   va_list ap;
3827
3828   va_start(ap, fmt);
3829   vfprintf_stderr(fmt, ap);
3830   va_end(ap);
3831 }
3832
3833 /*
3834  * Report an error in command-line arguments.
3835  */
3836 void
3837 cmdarg_err(const char *fmt, ...)
3838 {
3839   va_list ap;
3840
3841   va_start(ap, fmt);
3842   failure_message(fmt, ap);
3843   va_end(ap);
3844 }
3845
3846 /*
3847  * Report additional information for an error in command-line arguments.
3848  */
3849 void
3850 cmdarg_err_cont(const char *fmt, ...)
3851 {
3852   va_list ap;
3853
3854   va_start(ap, fmt);
3855   vfprintf(stderr, fmt, ap);
3856   fprintf(stderr, "\n");
3857   va_end(ap);
3858 }