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