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