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