Add app_mem_usage.h.
[metze/wireshark/wip.git] / rawshark.c
1 /* rawshark.c
2  *
3  * $Id$
4  *
5  * Wireshark - Network traffic analyzer
6  * By Gerald Combs <gerald@wireshark.org>
7  * Copyright 1998 Gerald Combs
8  *
9  * Rawshark - Raw field extractor by Gerald Combs <gerald@wireshark.org>
10  * and Loris Degioanni <loris.degioanni@cacetech.com>
11  * Based on TShark, by Gilbert Ramirez <gram@alumni.rice.edu> and Guy Harris
12  * <guy@alum.mit.edu>.
13  *
14  * This program is free software; you can redistribute it and/or
15  * modify it under the terms of the GNU General Public License
16  * as published by the Free Software Foundation; either version 2
17  * of the License, or (at your option) any later version.
18  *
19  * This program is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22  * GNU General Public License for more details.
23  *
24  * You should have received a copy of the GNU General Public License
25  * along with this program; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
27  */
28
29 /*
30  * Rawshark does the following:
31  * - Opens a specified file or named pipe
32  * - Applies a specfied DLT or "decode as" encapsulation
33  * - Reads frames prepended with a libpcap packet header.
34  * - Prints a status line, followed by fields from a specified list.
35  */
36
37 #include "config.h"
38
39 #include <stdlib.h>
40 #include <stdio.h>
41 #include <string.h>
42 #include <ctype.h>
43 #include <locale.h>
44 #include <limits.h>
45
46 #ifdef HAVE_UNISTD_H
47 #include <unistd.h>
48 #endif
49
50 #include <errno.h>
51
52 #ifdef HAVE_FCNTL_H
53 #include <fcntl.h>
54 #endif
55
56 #include <signal.h>
57
58 #ifdef HAVE_SYS_STAT_H
59 # include <sys/stat.h>
60 #endif
61
62 #ifndef HAVE_GETOPT
63 #include "wsutil/wsgetopt.h"
64 #endif
65
66 #include <glib.h>
67 #include <epan/epan-int.h>
68 #include <epan/epan.h>
69 #include <epan/filesystem.h>
70 #include <wsutil/crash_info.h>
71 #include <wsutil/privileges.h>
72 #include <wsutil/file_util.h>
73
74 #include "globals.h"
75 #include <epan/packet.h>
76 #include "file.h"
77 #include "frame_tvbuff.h"
78 #include <epan/disabled_protos.h>
79 #include <epan/prefs.h>
80 #include <epan/column.h>
81 #include <epan/print.h>
82 #include <epan/addr_resolv.h>
83 #include "ui/util.h"
84 #include "clopts_common.h"
85 #include "cmdarg_err.h"
86 #include "version_info.h"
87 #include <epan/plugins.h>
88 #include "register.h"
89 #include "conditions.h"
90 #include "capture_stop_conditions.h"
91 #include "capture_ui_utils.h"
92 #include <epan/epan_dissect.h>
93 #include <epan/stat_cmd_args.h>
94 #include <epan/timestamp.h>
95 #include <wsutil/unicode-utils.h>
96 #include "epan/column-utils.h"
97 #include "epan/proto.h"
98 #include <epan/tap.h>
99
100 #include <wiretap/wtap.h>
101 #include <wiretap/libpcap.h>
102 #include <wiretap/pcap-encap.h>
103
104 #ifdef HAVE_LIBPCAP
105 #include <setjmp.h>
106 #include "capture-pcap-util.h"
107 #ifdef _WIN32
108 #include "capture-wpcap.h"
109 #endif /* _WIN32 */
110 #endif /* HAVE_LIBPCAP */
111 #include "log.h"
112
113 #ifdef _WIN32
114 #include <wsutil/unicode-utils.h>
115 #endif /* _WIN32 */
116
117 /*
118  * This is the template for the decode as option; it is shared between the
119  * various functions that output the usage for this parameter.
120  */
121 static const gchar decode_as_arg_template[] = "<layer_type>==<selector>,<decode_as_protocol>";
122
123 static guint32 cum_bytes;
124 static const frame_data *ref;
125 static frame_data ref_frame;
126 static frame_data *prev_dis;
127 static frame_data prev_dis_frame;
128 static frame_data *prev_cap;
129 static frame_data prev_cap_frame;
130
131 /*
132  * The way the packet decode is to be written.
133  */
134 typedef enum {
135     WRITE_TEXT, /* summary or detail text */
136     WRITE_XML   /* PDML or PSML */
137     /* Add CSV and the like here */
138 } output_action_e;
139
140 static gboolean line_buffered;
141 static print_format_e print_format = PR_FMT_TEXT;
142
143 static gboolean want_pcap_pkthdr;
144
145 cf_status_t raw_cf_open(capture_file *cf, const char *fname);
146 static int load_cap_file(capture_file *cf);
147 static gboolean process_packet(capture_file *cf, gint64 offset,
148                                struct wtap_pkthdr *whdr, const guchar *pd);
149 static void show_print_file_io_error(int err);
150
151 static void open_failure_message(const char *filename, int err,
152                                  gboolean for_writing);
153 static void failure_message(const char *msg_format, va_list ap);
154 static void read_failure_message(const char *filename, int err);
155 static void write_failure_message(const char *filename, int err);
156 static void protocolinfo_init(char *field);
157 static gboolean parse_field_string_format(char *format);
158
159 typedef enum {
160     SF_NONE,    /* No format (placeholder) */
161     SF_NAME,    /* %D Field name / description */
162     SF_NUMVAL,  /* %N Numeric value */
163     SF_STRVAL   /* %S String value */
164 } string_fmt_e;
165
166 typedef struct string_fmt_s {
167     gchar *plain;
168     string_fmt_e format;    /* Valid if plain is NULL */
169 } string_fmt_t;
170
171 capture_file cfile;
172 int n_rfilters;
173 int n_rfcodes;
174 dfilter_t *rfcodes[64];
175 int n_rfieldfilters;
176 dfilter_t *rfieldfcodes[64];
177 int fd;
178 int encap;
179 GPtrArray *string_fmts;
180
181 static void
182 print_usage(gboolean print_ver)
183 {
184     FILE *output;
185
186     if (print_ver) {
187         output = stdout;
188         fprintf(output,
189                 "Rawshark " VERSION "%s\n"
190                 "Dump and analyze network traffic.\n"
191                 "See http://www.wireshark.org for more information.\n"
192                 "\n"
193                 "%s",
194                 wireshark_svnversion, get_copyright_info());
195     } else {
196         output = stderr;
197     }
198     fprintf(output, "\n");
199     fprintf(output, "Usage: rawshark [options] ...\n");
200     fprintf(output, "\n");
201
202     fprintf(output, "Input file:\n");
203     fprintf(output, "  -r <infile>              set the pipe or file name to read from\n");
204
205     fprintf(output, "\n");
206     fprintf(output, "Processing:\n");
207     fprintf(output, "  -d <encap:dlt>|<proto:protoname>\n");
208     fprintf(output, "                           packet encapsulation or protocol\n");
209     fprintf(output, "  -F <field>               field to display\n");
210     fprintf(output, "  -n                       disable all name resolution (def: all enabled)\n");
211     fprintf(output, "  -N <name resolve flags>  enable specific name resolution(s): \"mntC\"\n");
212     fprintf(output, "  -p                       use the system's packet header format\n");
213     fprintf(output, "                           (which may have 64-bit timestamps)\n");
214     fprintf(output, "  -R <read filter>         packet filter in Wireshark display filter syntax\n");
215     fprintf(output, "  -s                       skip PCAP header on input\n");
216
217     fprintf(output, "\n");
218     fprintf(output, "Output:\n");
219     fprintf(output, "  -l                       flush output after each packet\n");
220     fprintf(output, "  -S                       format string for fields\n");
221     fprintf(output, "                           (%%D - name, %%S - stringval, %%N numval)\n");
222     fprintf(output, "  -t ad|a|r|d|dd|e         output format of time stamps (def: r: rel. to first)\n");
223
224     fprintf(output, "\n");
225     fprintf(output, "Miscellaneous:\n");
226     fprintf(output, "  -h                       display this help and exit\n");
227     fprintf(output, "  -o <name>:<value> ...    override preference setting\n");
228     fprintf(output, "  -v                       display version info and exit\n");
229 }
230
231 static void
232 log_func_ignore (const gchar *log_domain _U_, GLogLevelFlags log_level _U_,
233                  const gchar *message _U_, gpointer user_data _U_)
234 {
235 }
236
237 /**
238  * Open a pipe for raw input.  This is a stripped-down version of
239  * pcap_loop.c:cap_pipe_open_live().
240  * We check if "pipe_name" is "-" (stdin) or a FIFO, and open it.
241  * @param pipe_name The name of the pipe or FIFO.
242  * @return A POSIX file descriptor on success, or -1 on failure.
243  */
244 static int
245 raw_pipe_open(const char *pipe_name)
246 {
247 #ifndef _WIN32
248     ws_statb64 pipe_stat;
249 #else
250     char *pncopy, *pos = NULL;
251     DWORD err;
252     wchar_t *err_str;
253     HANDLE hPipe = NULL;
254 #endif
255     int          rfd;
256
257     g_log(LOG_DOMAIN_CAPTURE_CHILD, G_LOG_LEVEL_DEBUG, "open_raw_pipe: %s", pipe_name);
258
259     /*
260      * XXX Rawshark blocks until we return
261      */
262     if (strcmp(pipe_name, "-") == 0) {
263         rfd = 0; /* read from stdin */
264 #ifdef _WIN32
265         /*
266          * This is needed to set the stdin pipe into binary mode, otherwise
267          * CR/LF are mangled...
268          */
269         _setmode(0, _O_BINARY);
270 #endif  /* _WIN32 */
271     } else {
272 #ifndef _WIN32
273         if (ws_stat64(pipe_name, &pipe_stat) < 0) {
274             fprintf(stderr, "rawshark: The pipe %s could not be checked: %s\n",
275                     pipe_name, g_strerror(errno));
276             return -1;
277         }
278         if (! S_ISFIFO(pipe_stat.st_mode)) {
279             if (S_ISCHR(pipe_stat.st_mode)) {
280                 /*
281                  * Assume the user specified an interface on a system where
282                  * interfaces are in /dev.  Pretend we haven't seen it.
283                  */
284             } else
285             {
286                 fprintf(stderr, "rawshark: \"%s\" is neither an interface nor a pipe\n",
287                         pipe_name);
288             }
289             return -1;
290         }
291         rfd = ws_open(pipe_name, O_RDONLY | O_NONBLOCK, 0000 /* no creation so don't matter */);
292         if (rfd == -1) {
293             fprintf(stderr, "rawshark: \"%s\" could not be opened: %s\n",
294                     pipe_name, g_strerror(errno));
295             return -1;
296         }
297 #else /* _WIN32 */
298 #define PIPE_STR "\\pipe\\"
299         /* Under Windows, named pipes _must_ have the form
300          * "\\<server>\pipe\<pipe_name>".  <server> may be "." for localhost.
301          */
302         pncopy = g_strdup(pipe_name);
303         if (strstr(pncopy, "\\\\") == pncopy) {
304             pos = strchr(pncopy + 3, '\\');
305             if (pos && g_ascii_strncasecmp(pos, PIPE_STR, strlen(PIPE_STR)) != 0)
306                 pos = NULL;
307         }
308
309         g_free(pncopy);
310
311         if (!pos) {
312             fprintf(stderr, "rawshark: \"%s\" is neither an interface nor a pipe\n",
313                     pipe_name);
314             return -1;
315         }
316
317         /* Wait for the pipe to appear */
318         while (1) {
319             hPipe = CreateFile(utf_8to16(pipe_name), GENERIC_READ, 0, NULL,
320                                OPEN_EXISTING, 0, NULL);
321
322             if (hPipe != INVALID_HANDLE_VALUE)
323                 break;
324
325             err = GetLastError();
326             if (err != ERROR_PIPE_BUSY) {
327                 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS,
328                               NULL, err, 0, (LPTSTR) &err_str, 0, NULL);
329                 fprintf(stderr, "rawshark: \"%s\" could not be opened: %s (error %d)\n",
330                         pipe_name, utf_16to8(err_str), err);
331                 LocalFree(err_str);
332                 return -1;
333             }
334
335             if (!WaitNamedPipe(utf_8to16(pipe_name), 30 * 1000)) {
336                 err = GetLastError();
337                 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS,
338                               NULL, err, 0, (LPTSTR) &err_str, 0, NULL);
339                 fprintf(stderr, "rawshark: \"%s\" could not be waited for: %s (error %d)\n",
340                         pipe_name, utf_16to8(err_str), err);
341                 LocalFree(err_str);
342                 return -1;
343             }
344         }
345
346         rfd = _open_osfhandle((long) hPipe, _O_RDONLY);
347         if (rfd == -1) {
348             fprintf(stderr, "rawshark: \"%s\" could not be opened: %s\n",
349                     pipe_name, g_strerror(errno));
350             return -1;
351         }
352 #endif /* _WIN32 */
353     }
354
355     return rfd;
356 }
357
358 /**
359  * Parse a link-type argument of the form "encap:<pcap dlt>" or
360  * "proto:<proto name>".  "Pcap dlt" must be a name conforming to
361  * pcap_datalink_name_to_val() or an integer.  "Proto name" must be
362  * a protocol name, e.g. "http".
363  */
364 static gboolean
365 set_link_type(const char *lt_arg) {
366     char *spec_ptr = strchr(lt_arg, ':');
367     char *p;
368     int dlt_val;
369     long val;
370     dissector_handle_t dhandle;
371     GString *pref_str;
372
373     if (!spec_ptr)
374         return FALSE;
375
376     spec_ptr++;
377
378     if (strncmp(lt_arg, "encap:", strlen("encap:")) == 0) {
379         dlt_val = linktype_name_to_val(spec_ptr);
380         if (dlt_val == -1) {
381             errno = 0;
382             val = strtol(spec_ptr, &p, 10);
383             if (p == spec_ptr || *p != '\0' || errno != 0 || val > INT_MAX) {
384                 return FALSE;
385             }
386             dlt_val = (int)val;
387         }
388         encap = wtap_pcap_encap_to_wtap_encap(dlt_val);
389         if (encap == WTAP_ENCAP_UNKNOWN) {
390             return FALSE;
391         }
392         return TRUE;
393     } else if (strncmp(lt_arg, "proto:", strlen("proto:")) == 0) {
394         dhandle = find_dissector(spec_ptr);
395         if (dhandle) {
396             encap = WTAP_ENCAP_USER0;
397             pref_str = g_string_new("uat:user_dlts:");
398             /* This must match the format used in the user_dlts file */
399             g_string_append_printf(pref_str,
400                                    "\"User 0 (DLT=147)\",\"%s\",\"0\",\"\",\"0\",\"\"",
401                                    spec_ptr);
402             if (prefs_set_pref(pref_str->str) != PREFS_SET_OK) {
403                 g_string_free(pref_str, TRUE);
404                 return FALSE;
405             }
406             g_string_free(pref_str, TRUE);
407             return TRUE;
408         }
409     }
410     return FALSE;
411 }
412
413 static void
414 show_version(GString *comp_info_str, GString *runtime_info_str)
415 {
416     printf("Rawshark " VERSION "%s\n"
417            "\n"
418            "%s"
419            "\n"
420            "%s"
421            "\n"
422            "%s",
423            wireshark_svnversion, get_copyright_info(), comp_info_str->str,
424            runtime_info_str->str);
425 }
426
427 int
428 main(int argc, char *argv[])
429 {
430     GString             *comp_info_str;
431     GString             *runtime_info_str;
432     char                *init_progfile_dir_error;
433     int                  opt, i;
434     gboolean             arg_error = FALSE;
435
436 #ifdef _WIN32
437     WSADATA              wsaData;
438 #endif  /* _WIN32 */
439
440     char                *gpf_path, *pf_path;
441     char                *gdp_path, *dp_path;
442     int                  gpf_open_errno, gpf_read_errno;
443     int                  pf_open_errno, pf_read_errno;
444     int                  gdp_open_errno, gdp_read_errno;
445     int                  dp_open_errno, dp_read_errno;
446     int                  err;
447     gchar               *pipe_name = NULL;
448     gchar               *rfilters[64];
449     e_prefs             *prefs_p;
450     char                 badopt;
451     int                  log_flags;
452     GPtrArray           *disp_fields = g_ptr_array_new();
453     guint                fc;
454     gboolean             skip_pcap_header = FALSE;
455
456 #define OPTSTRING_INIT "d:F:hlnN:o:pr:R:sS:t:v"
457
458     static const char    optstring[] = OPTSTRING_INIT;
459
460     /* Assemble the compile-time version information string */
461     comp_info_str = g_string_new("Compiled ");
462     get_compiled_version_info(comp_info_str, NULL, epan_get_compiled_version_info);
463
464     /* Assemble the run-time version information string */
465     runtime_info_str = g_string_new("Running ");
466     get_runtime_version_info(runtime_info_str, NULL);
467
468     /* Add it to the information to be reported on a crash. */
469     ws_add_crash_info("Rawshark " VERSION "%s\n"
470            "\n"
471            "%s"
472            "\n"
473            "%s",
474         wireshark_svnversion, comp_info_str->str, runtime_info_str->str);
475
476 #ifdef _WIN32
477     arg_list_utf_16to8(argc, argv);
478     create_app_running_mutex();
479 #endif /* _WIN32 */
480
481     /*
482      * Get credential information for later use.
483      */
484     init_process_policies();
485
486     /*
487      * Clear the filters arrays
488      */
489     memset(rfilters, 0, sizeof(rfilters));
490     memset(rfcodes, 0, sizeof(rfcodes));
491     n_rfilters = 0;
492     n_rfcodes = 0;
493
494     /*
495      * Initialize our string format
496      */
497     string_fmts = g_ptr_array_new();
498
499     /*
500      * Attempt to get the pathname of the executable file.
501      */
502     init_progfile_dir_error = init_progfile_dir(argv[0], main);
503     if (init_progfile_dir_error != NULL) {
504         fprintf(stderr, "rawshark: Can't get pathname of rawshark program: %s.\n",
505                 init_progfile_dir_error);
506     }
507
508     /*
509      * Get credential information for later use.
510      */
511     init_process_policies();
512
513     /* nothing more than the standard GLib handler, but without a warning */
514     log_flags =
515         G_LOG_LEVEL_WARNING |
516         G_LOG_LEVEL_MESSAGE |
517         G_LOG_LEVEL_INFO |
518         G_LOG_LEVEL_DEBUG;
519
520     g_log_set_handler(NULL,
521                       (GLogLevelFlags)log_flags,
522                       log_func_ignore, NULL /* user_data */);
523     g_log_set_handler(LOG_DOMAIN_CAPTURE_CHILD,
524                       (GLogLevelFlags)log_flags,
525                       log_func_ignore, NULL /* user_data */);
526
527     timestamp_set_type(TS_RELATIVE);
528     timestamp_set_precision(TS_PREC_AUTO);
529     timestamp_set_seconds_type(TS_SECONDS_DEFAULT);
530
531     /* Register all dissectors; we must do this before checking for the
532        "-G" flag, as the "-G" flag dumps information registered by the
533        dissectors, and we must do it before we read the preferences, in
534        case any dissectors register preferences. */
535     epan_init(register_all_protocols, register_all_protocol_handoffs, NULL, NULL,
536               failure_message, open_failure_message, read_failure_message,
537               write_failure_message);
538
539     /* Set the C-language locale to the native environment. */
540     setlocale(LC_ALL, "");
541
542     prefs_p = read_prefs(&gpf_open_errno, &gpf_read_errno, &gpf_path,
543                          &pf_open_errno, &pf_read_errno, &pf_path);
544     if (gpf_path != NULL) {
545         if (gpf_open_errno != 0) {
546             cmdarg_err("Can't open global preferences file \"%s\": %s.",
547                        pf_path, g_strerror(gpf_open_errno));
548         }
549         if (gpf_read_errno != 0) {
550             cmdarg_err("I/O error reading global preferences file \"%s\": %s.",
551                        pf_path, g_strerror(gpf_read_errno));
552         }
553     }
554     if (pf_path != NULL) {
555         if (pf_open_errno != 0) {
556             cmdarg_err("Can't open your preferences file \"%s\": %s.", pf_path,
557                        g_strerror(pf_open_errno));
558         }
559         if (pf_read_errno != 0) {
560             cmdarg_err("I/O error reading your preferences file \"%s\": %s.",
561                        pf_path, g_strerror(pf_read_errno));
562         }
563         g_free(pf_path);
564         pf_path = NULL;
565     }
566
567     /* Read the disabled protocols file. */
568     read_disabled_protos_list(&gdp_path, &gdp_open_errno, &gdp_read_errno,
569                               &dp_path, &dp_open_errno, &dp_read_errno);
570     if (gdp_path != NULL) {
571         if (gdp_open_errno != 0) {
572             cmdarg_err("Could not open global disabled protocols file\n\"%s\": %s.",
573                        gdp_path, g_strerror(gdp_open_errno));
574         }
575         if (gdp_read_errno != 0) {
576             cmdarg_err("I/O error reading global disabled protocols file\n\"%s\": %s.",
577                        gdp_path, g_strerror(gdp_read_errno));
578         }
579         g_free(gdp_path);
580     }
581     if (dp_path != NULL) {
582         if (dp_open_errno != 0) {
583             cmdarg_err(
584                 "Could not open your disabled protocols file\n\"%s\": %s.", dp_path,
585                 g_strerror(dp_open_errno));
586         }
587         if (dp_read_errno != 0) {
588             cmdarg_err(
589                 "I/O error reading your disabled protocols file\n\"%s\": %s.", dp_path,
590                 g_strerror(dp_read_errno));
591         }
592         g_free(dp_path);
593     }
594
595 #ifdef _WIN32
596     /* Load Wpcap, if possible */
597     load_wpcap();
598 #endif
599
600     cap_file_init(&cfile);
601
602     /* Print format defaults to this. */
603     print_format = PR_FMT_TEXT;
604
605     /* Initialize our encapsulation type */
606     encap = WTAP_ENCAP_UNKNOWN;
607
608     /* Now get our args */
609     /* XXX - We should probably have an option to dump libpcap link types */
610     while ((opt = getopt(argc, argv, optstring)) != -1) {
611         switch (opt) {
612             case 'd':        /* Payload type */
613                 if (!set_link_type(optarg)) {
614                     cmdarg_err("Invalid link type or protocol \"%s\"", optarg);
615                     exit(1);
616                 }
617                 break;
618             case 'F':        /* Read field to display */
619                 g_ptr_array_add(disp_fields, g_strdup(optarg));
620                 break;
621             case 'h':        /* Print help and exit */
622                 print_usage(TRUE);
623                 exit(0);
624                 break;
625             case 'l':        /* "Line-buffer" standard output */
626                 /* This isn't line-buffering, strictly speaking, it's just
627                    flushing the standard output after the information for
628                    each packet is printed; however, that should be good
629                    enough for all the purposes to which "-l" is put (and
630                    is probably actually better for "-V", as it does fewer
631                    writes).
632
633                    See the comment in "process_packet()" for an explanation of
634                    why we do that, and why we don't just use "setvbuf()" to
635                    make the standard output line-buffered (short version: in
636                    Windows, "line-buffered" is the same as "fully-buffered",
637                    and the output buffer is only flushed when it fills up). */
638                 line_buffered = TRUE;
639                 break;
640             case 'n':        /* No name resolution */
641                 gbl_resolv_flags.mac_name = FALSE;
642                 gbl_resolv_flags.network_name = FALSE;
643                 gbl_resolv_flags.transport_name = FALSE;
644                 gbl_resolv_flags.concurrent_dns = FALSE;
645                 break;
646             case 'N':        /* Select what types of addresses/port #s to resolve */
647                 badopt = string_to_name_resolve(optarg, &gbl_resolv_flags);
648                 if (badopt != '\0') {
649                     cmdarg_err("-N specifies unknown resolving option '%c'; valid options are 'm', 'n', and 't'",
650                                badopt);
651                     exit(1);
652                 }
653                 break;
654             case 'o':        /* Override preference from command line */
655                 switch (prefs_set_pref(optarg)) {
656
657                     case PREFS_SET_OK:
658                         break;
659
660                     case PREFS_SET_SYNTAX_ERR:
661                         cmdarg_err("Invalid -o flag \"%s\"", optarg);
662                         exit(1);
663                         break;
664
665                     case PREFS_SET_NO_SUCH_PREF:
666                     case PREFS_SET_OBSOLETE:
667                         cmdarg_err("-o flag \"%s\" specifies unknown preference", optarg);
668                         exit(1);
669                         break;
670                 }
671                 break;
672             case 'p':        /* Expect pcap_pkthdr packet headers, which may have 64-bit timestamps */
673                 want_pcap_pkthdr = TRUE;
674                 break;
675             case 'r':        /* Read capture file xxx */
676                 pipe_name = g_strdup(optarg);
677                 break;
678             case 'R':        /* Read file filter */
679                 if(n_rfilters < (int) sizeof(rfilters) / (int) sizeof(rfilters[0])) {
680                     rfilters[n_rfilters++] = optarg;
681                 }
682                 else {
683                     cmdarg_err("Too many display filters");
684                     exit(1);
685                 }
686                 break;
687             case 's':        /* Skip PCAP header */
688                 skip_pcap_header = TRUE;
689                 break;
690             case 'S':        /* Print string representations */
691                 if (!parse_field_string_format(optarg)) {
692                     cmdarg_err("Invalid field string format");
693                     exit(1);
694                 }
695                 break;
696             case 't':        /* Time stamp type */
697                 if (strcmp(optarg, "r") == 0)
698                     timestamp_set_type(TS_RELATIVE);
699                 else if (strcmp(optarg, "a") == 0)
700                     timestamp_set_type(TS_ABSOLUTE);
701                 else if (strcmp(optarg, "ad") == 0)
702                     timestamp_set_type(TS_ABSOLUTE_WITH_DATE);
703                 else if (strcmp(optarg, "d") == 0)
704                     timestamp_set_type(TS_DELTA);
705                 else if (strcmp(optarg, "dd") == 0)
706                     timestamp_set_type(TS_DELTA_DIS);
707                 else if (strcmp(optarg, "e") == 0)
708                     timestamp_set_type(TS_EPOCH);
709                 else if (strcmp(optarg, "u") == 0)
710                     timestamp_set_type(TS_UTC);
711                 else if (strcmp(optarg, "ud") == 0)
712                     timestamp_set_type(TS_UTC_WITH_DATE);
713                 else {
714                     cmdarg_err("Invalid time stamp type \"%s\"",
715                                optarg);
716                     cmdarg_err_cont("It must be \"r\" for relative, \"a\" for absolute,");
717                     cmdarg_err_cont("\"ad\" for absolute with date, or \"d\" for delta.");
718                     exit(1);
719                 }
720                 break;
721             case 'v':        /* Show version and exit */
722             {
723                 show_version(comp_info_str, runtime_info_str);
724                 g_string_free(comp_info_str, TRUE);
725                 g_string_free(runtime_info_str, TRUE);
726                 exit(0);
727                 break;
728             }
729             default:
730             case '?':        /* Bad flag - print usage message */
731                 print_usage(TRUE);
732             exit(1);
733             break;
734         }
735     }
736
737     /* Notify all registered modules that have had any of their preferences
738        changed either from one of the preferences file or from the command
739        line that their preferences have changed.
740        Initialize preferences before display filters, otherwise modules
741        like MATE won't work. */
742     prefs_apply_all();
743
744     /* Initialize our display fields */
745     for (fc = 0; fc < disp_fields->len; fc++) {
746         protocolinfo_init((char *)g_ptr_array_index(disp_fields, fc));
747     }
748     g_ptr_array_free(disp_fields, TRUE);
749     printf("\n");
750     fflush(stdout);
751
752     /* If no capture filter or read filter has been specified, and there are
753        still command-line arguments, treat them as the tokens of a capture
754        filter (if no "-r" flag was specified) or a read filter (if a "-r"
755        flag was specified. */
756     if (optind < argc) {
757         if (pipe_name != NULL) {
758             if (n_rfilters != 0) {
759                 cmdarg_err("Read filters were specified both with \"-R\" "
760                            "and with additional command-line arguments");
761                 exit(1);
762             }
763             rfilters[n_rfilters] = get_args_as_string(argc, argv, optind);
764         }
765     }
766
767     /* Make sure we got a dissector handle for our payload. */
768     if (encap == WTAP_ENCAP_UNKNOWN) {
769         cmdarg_err("No valid payload dissector specified.");
770         exit(1);
771     }
772
773     if (arg_error) {
774         print_usage(FALSE);
775         exit(1);
776     }
777
778
779 #ifdef _WIN32
780     /* Start windows sockets */
781     WSAStartup( MAKEWORD( 1, 1 ), &wsaData );
782 #endif /* _WIN32 */
783
784     /* At this point MATE will have registered its field array so we can
785        have a tap filter with one of MATE's late-registered fields as part
786        of the filter.  We can now process all the "-z" arguments. */
787     start_requested_stats();
788
789     /* disabled protocols as per configuration file */
790     if (gdp_path == NULL && dp_path == NULL) {
791         set_disabled_protos_list();
792     }
793
794     /* Build the column format array */
795     build_column_format_array(&cfile.cinfo, prefs_p->num_cols, TRUE);
796
797     if (n_rfilters != 0) {
798         for (i = 0; i < n_rfilters; i++) {
799             if (!dfilter_compile(rfilters[i], &rfcodes[n_rfcodes])) {
800                 cmdarg_err("%s", dfilter_error_msg);
801                 epan_cleanup();
802                 exit(2);
803             }
804             n_rfcodes++;
805         }
806     }
807
808     if (pipe_name) {
809         /*
810          * We're reading a pipe (or capture file).
811          */
812
813         /*
814          * Immediately relinquish any special privileges we have; we must not
815          * be allowed to read any capture files the user running Rawshark
816          * can't open.
817          */
818         relinquish_special_privs_perm();
819
820         if (raw_cf_open(&cfile, pipe_name) != CF_OK) {
821             epan_cleanup();
822             exit(2);
823         }
824
825         /* Do we need to PCAP header and magic? */
826         if (skip_pcap_header) {
827             size_t bytes_left = sizeof(struct pcap_hdr) + sizeof(guint32);
828             gchar buf[sizeof(struct pcap_hdr) + sizeof(guint32)];
829             while (bytes_left != 0) {
830                 ssize_t bytes = read(fd, buf, (int)bytes_left);
831                 if (bytes <= 0) {
832                     cmdarg_err("Not enough bytes for pcap header.");
833                     exit(2);
834                 }
835                 bytes_left -= bytes;
836             }
837         }
838
839         /* Set timestamp precision; there should arguably be a command-line
840            option to let the user set this. */
841 #if 0
842         switch(wtap_file_tsprecision(cfile.wth)) {
843             case(WTAP_FILE_TSPREC_SEC):
844                 timestamp_set_precision(TS_PREC_AUTO_SEC);
845                 break;
846             case(WTAP_FILE_TSPREC_DSEC):
847                 timestamp_set_precision(TS_PREC_AUTO_DSEC);
848                 break;
849             case(WTAP_FILE_TSPREC_CSEC):
850                 timestamp_set_precision(TS_PREC_AUTO_CSEC);
851                 break;
852             case(WTAP_FILE_TSPREC_MSEC):
853                 timestamp_set_precision(TS_PREC_AUTO_MSEC);
854                 break;
855             case(WTAP_FILE_TSPREC_USEC):
856                 timestamp_set_precision(TS_PREC_AUTO_USEC);
857                 break;
858             case(WTAP_FILE_TSPREC_NSEC):
859                 timestamp_set_precision(TS_PREC_AUTO_NSEC);
860                 break;
861             default:
862                 g_assert_not_reached();
863         }
864 #else
865         timestamp_set_precision(TS_PREC_AUTO_USEC);
866 #endif
867
868         /* Process the packets in the file */
869         err = load_cap_file(&cfile);
870
871         if (err != 0) {
872             epan_cleanup();
873             exit(2);
874         }
875     } else {
876         /* If you want to capture live packets, use TShark. */
877         cmdarg_err("Input file or pipe name not specified.");
878         exit(2);
879     }
880
881     epan_cleanup();
882
883     return 0;
884 }
885
886 /**
887  * Read data from a raw pipe.  The "raw" data consists of a libpcap
888  * packet header followed by the payload.
889  * @param fd [IN] A POSIX file descriptor.  Because that's _exactly_ the sort
890  *           of thing you want to use in Windows.
891  * @param phdr [OUT] Packet header information.
892  * @param err [OUT] Error indicator.  Uses wiretap values.
893  * @param err_info [OUT] Error message.
894  * @param data_offset [OUT] data offset in the pipe.
895  * @return TRUE on success, FALSE on failure.
896  */
897 static gboolean
898 raw_pipe_read(struct wtap_pkthdr *phdr, guchar * pd, int *err, const gchar **err_info, gint64 *data_offset) {
899     struct pcap_pkthdr mem_hdr;
900     struct pcaprec_hdr disk_hdr;
901     ssize_t bytes_read = 0;
902     size_t bytes_needed = sizeof(disk_hdr);
903     guchar *ptr = (guchar*) &disk_hdr;
904     static gchar err_str[100];
905
906     if (want_pcap_pkthdr) {
907         bytes_needed = sizeof(mem_hdr);
908         ptr = (guchar*) &mem_hdr;
909     }
910
911     /* Copied from capture_loop.c */
912     while (bytes_needed > 0) {
913         bytes_read = read(fd, ptr, (int)bytes_needed);
914         if (bytes_read == 0) {
915             *err = 0;
916             return FALSE;
917         } else if (bytes_read < 0) {
918             *err = WTAP_ERR_CANT_READ;
919             *err_info = "Error reading header from pipe";
920             return FALSE;
921         }
922         bytes_needed -= bytes_read;
923         *data_offset += bytes_read;
924         ptr += bytes_read;
925     }
926
927     if (want_pcap_pkthdr) {
928         phdr->ts.secs = mem_hdr.ts.tv_sec;
929         phdr->ts.nsecs = mem_hdr.ts.tv_usec * 1000;
930         phdr->caplen = mem_hdr.caplen;
931         phdr->len = mem_hdr.len;
932     } else {
933         phdr->ts.secs = disk_hdr.ts_sec;
934         phdr->ts.nsecs = disk_hdr.ts_usec * 1000;
935         phdr->caplen = disk_hdr.incl_len;
936         phdr->len = disk_hdr.orig_len;
937     }
938     bytes_needed = phdr->caplen;
939
940     phdr->pkt_encap = encap;
941
942 #if 0
943     printf("tv_sec: %d (%04x)\n", hdr.ts.tv_sec, hdr.ts.tv_sec);
944     printf("tv_usec: %d (%04x)\n", hdr.ts.tv_usec, hdr.ts.tv_usec);
945     printf("caplen: %d (%04x)\n", hdr.caplen, hdr.caplen);
946     printf("len: %d (%04x)\n", hdr.len, hdr.len);
947 #endif
948     if (bytes_needed > WTAP_MAX_PACKET_SIZE) {
949         *err = WTAP_ERR_BAD_FILE;
950         g_snprintf(err_str, 100, "Bad packet length: %lu\n",
951                    (unsigned long) bytes_needed);
952         *err_info = err_str;
953         return FALSE;
954     }
955
956     ptr = pd;
957     while (bytes_needed > 0) {
958         bytes_read = read(fd, ptr, (int)bytes_needed);
959         if (bytes_read == 0) {
960             *err = WTAP_ERR_SHORT_READ;
961             *err_info = "Got zero bytes reading data from pipe";
962             return FALSE;
963         } else if (bytes_read < 0) {
964             *err = WTAP_ERR_CANT_READ;
965             *err_info = "Error reading data from pipe";
966             return FALSE;
967         }
968         bytes_needed -= bytes_read;
969         *data_offset += bytes_read;
970         ptr += bytes_read;
971     }
972     return TRUE;
973 }
974
975 static int
976 load_cap_file(capture_file *cf)
977 {
978     int          err;
979     const gchar  *err_info;
980     gint64       data_offset = 0;
981     struct wtap_pkthdr  phdr;
982     guchar       pd[WTAP_MAX_PACKET_SIZE];
983
984     while (raw_pipe_read(&phdr, pd, &err, &err_info, &data_offset)) {
985         process_packet(cf, data_offset, &phdr, pd);
986     }
987
988     if (err != 0) {
989         /* Print a message noting that the read failed somewhere along the line. */
990         switch (err) {
991
992             case WTAP_ERR_UNSUPPORTED_ENCAP:
993                 cmdarg_err("The file \"%s\" has a packet with a network type that Rawshark doesn't support.\n(%s)",
994                            cf->filename, err_info);
995                 break;
996
997             case WTAP_ERR_CANT_READ:
998                 cmdarg_err("An attempt to read from the file \"%s\" failed for some unknown reason.",
999                            cf->filename);
1000                 break;
1001
1002             case WTAP_ERR_SHORT_READ:
1003                 cmdarg_err("The file \"%s\" appears to have been cut short in the middle of a packet.",
1004                            cf->filename);
1005                 break;
1006
1007             case WTAP_ERR_BAD_FILE:
1008                 cmdarg_err("The file \"%s\" appears to be damaged or corrupt.\n(%s)",
1009                            cf->filename, err_info);
1010                 break;
1011
1012             case WTAP_ERR_DECOMPRESS:
1013                 cmdarg_err("The compressed file \"%s\" appears to be damaged or corrupt.\n(%s)",
1014                            cf->filename, err_info);
1015                 break;
1016
1017             default:
1018                 cmdarg_err("An error occurred while reading the file \"%s\": %s.",
1019                            cf->filename, wtap_strerror(err));
1020                 break;
1021         }
1022     }
1023
1024     return err;
1025 }
1026
1027 static gboolean
1028 process_packet(capture_file *cf, gint64 offset, struct wtap_pkthdr *whdr,
1029                const guchar *pd)
1030 {
1031     frame_data fdata;
1032     gboolean create_proto_tree;
1033     epan_dissect_t edt;
1034     gboolean passed;
1035     int i;
1036
1037     if(whdr->len == 0)
1038     {
1039         /* The user sends an empty packet when he wants to get output from us even if we don't currently have
1040            packets to process. We spit out a line with the timestamp and the text "void"
1041         */
1042         printf("%lu %lu %lu void -\n", (unsigned long int)cf->count,
1043                (unsigned long int)whdr->ts.secs,
1044                (unsigned long int)whdr->ts.nsecs);
1045
1046         fflush(stdout);
1047
1048         return FALSE;
1049     }
1050
1051     /* Count this packet. */
1052     cf->count++;
1053
1054     /* If we're going to print packet information, or we're going to
1055        run a read filter, or we're going to process taps, set up to
1056        do a dissection and do so. */
1057     frame_data_init(&fdata, cf->count, whdr, offset, cum_bytes);
1058
1059     passed = TRUE;
1060     create_proto_tree = TRUE;
1061
1062     /* The protocol tree will be "visible", i.e., printed, only if we're
1063        printing packet details, which is true if we're in verbose mode ("verbose"
1064        is true). */
1065     epan_dissect_init(&edt, cf->epan, create_proto_tree, FALSE);
1066
1067     /* If we're running a read filter, prime the epan_dissect_t with that
1068        filter. */
1069     if (n_rfilters > 0) {
1070         for(i = 0; i < n_rfcodes; i++) {
1071             epan_dissect_prime_dfilter(&edt, rfcodes[i]);
1072         }
1073     }
1074
1075     printf("%lu", (unsigned long int) cf->count);
1076
1077     frame_data_set_before_dissect(&fdata, &cf->elapsed_time,
1078                                   &ref, prev_dis);
1079
1080     if (ref == &fdata) {
1081        ref_frame = fdata;
1082        ref = &ref_frame;
1083     }
1084
1085     /* We only need the columns if we're printing packet info but we're
1086      *not* verbose; in verbose mode, we print the protocol tree, not
1087      the protocol summary. */
1088     epan_dissect_run_with_taps(&edt, whdr, frame_tvbuff_new(&fdata, pd), &fdata, &cf->cinfo);
1089
1090     frame_data_set_after_dissect(&fdata, &cum_bytes);
1091     prev_dis_frame = fdata;
1092     prev_dis = &prev_dis_frame;
1093
1094     prev_cap_frame = fdata;
1095     prev_cap = &prev_cap_frame;
1096
1097     for(i = 0; i < n_rfilters; i++) {
1098         /* Run the read filter if we have one. */
1099         if (rfcodes[i])
1100             passed = dfilter_apply_edt(rfcodes[i], &edt);
1101         else
1102             passed = TRUE;
1103
1104         /* Print a one-line summary */
1105         printf(" %u", passed ? 1 : 0);
1106     }
1107
1108     printf(" -\n");
1109
1110     /* The ANSI C standard does not appear to *require* that a line-buffered
1111        stream be flushed to the host environment whenever a newline is
1112        written, it just says that, on such a stream, characters "are
1113        intended to be transmitted to or from the host environment as a
1114        block when a new-line character is encountered".
1115
1116        The Visual C++ 6.0 C implementation doesn't do what is intended;
1117        even if you set a stream to be line-buffered, it still doesn't
1118        flush the buffer at the end of every line.
1119
1120        So, if the "-l" flag was specified, we flush the standard output
1121        at the end of a packet.  This will do the right thing if we're
1122        printing packet summary lines, and, as we print the entire protocol
1123        tree for a single packet without waiting for anything to happen,
1124        it should be as good as line-buffered mode if we're printing
1125        protocol trees.  (The whole reason for the "-l" flag in either
1126        tcpdump or Rawshark is to allow the output of a live capture to
1127        be piped to a program or script and to have that script see the
1128        information for the packet as soon as it's printed, rather than
1129        having to wait until a standard I/O buffer fills up. */
1130     if (line_buffered)
1131         fflush(stdout);
1132
1133     if (ferror(stdout)) {
1134         show_print_file_io_error(errno);
1135         exit(2);
1136     }
1137
1138     epan_dissect_cleanup(&edt);
1139     frame_data_destroy(&fdata);
1140
1141     return passed;
1142 }
1143
1144 /****************************************************************************************
1145  * FIELD EXTRACTION ROUTINES
1146  ****************************************************************************************/
1147 typedef struct _pci_t {
1148     char *filter;
1149     int hf_index;
1150     int cmd_line_index;
1151 } pci_t;
1152
1153 static const char* ftenum_to_string(header_field_info *hfi)
1154 {
1155     if (!hfi) {
1156         return "n.a.";
1157     }
1158
1159     if (string_fmts->len > 0 && hfi->strings) {
1160         return "FT_STRING";
1161     }
1162
1163     switch(hfi->type) {
1164         case FT_NONE:
1165             return "FT_NONE";
1166         case FT_PROTOCOL:
1167             return "FT_PROTOCOL";
1168         case FT_BOOLEAN:
1169             return "FT_BOOLEAN";
1170         case FT_UINT8:
1171             return "FT_UINT8";
1172         case FT_UINT16:
1173             return "FT_UINT16";
1174         case FT_UINT24:
1175             return "FT_UINT24";
1176         case FT_UINT32:
1177             return "FT_UINT32";
1178         case FT_UINT64:
1179             return "FT_UINT64";
1180         case FT_INT8:
1181             return "FT_INT8";
1182         case FT_INT16:
1183             return "FT_INT16";
1184         case FT_INT24:
1185             return "FT_INT24";
1186         case FT_INT32:
1187             return "FT_INT32";
1188         case FT_INT64:
1189             return "FT_INT64";
1190         case FT_FLOAT:
1191             return "FT_FLOAT";
1192         case FT_DOUBLE:
1193             return "FT_DOUBLE";
1194         case FT_ABSOLUTE_TIME:
1195             return "FT_ABSOLUTE_TIME";
1196         case FT_RELATIVE_TIME:
1197             return "FT_RELATIVE_TIME";
1198         case FT_STRING:
1199             return "FT_STRING";
1200         case FT_STRINGZ:
1201             return "FT_STRINGZ";
1202         case FT_UINT_STRING:
1203             return "FT_UINT_STRING";
1204         case FT_ETHER:
1205             return "FT_ETHER";
1206         case FT_BYTES:
1207             return "FT_BYTES";
1208         case FT_UINT_BYTES:
1209             return "FT_UINT_BYTES";
1210         case FT_IPv4:
1211             return "FT_IPv4";
1212         case FT_IPv6:
1213             return "FT_IPv6";
1214         case FT_IPXNET:
1215             return "FT_IPXNET";
1216         case FT_FRAMENUM:
1217             return "FT_FRAMENUM";
1218         case FT_PCRE:
1219             return "FT_PCRE";
1220         case FT_GUID:
1221             return "FT_GUID";
1222         case FT_OID:
1223             return "FT_OID";
1224         case FT_NUM_TYPES:
1225             return "FT_NUM_TYPES";
1226         default:
1227             return "n.a.";
1228     };
1229 }
1230
1231 static const char* absolute_time_display_e_to_string(absolute_time_display_e atd)
1232 {
1233     switch(atd) {
1234         case ABSOLUTE_TIME_LOCAL:
1235             return "ABSOLUTE_TIME_LOCAL";
1236         case ABSOLUTE_TIME_UTC:
1237             return "ABSOLUTE_TIME_UTC";
1238         default:
1239             return "n.a.";
1240     }
1241 }
1242
1243 static const char* base_display_e_to_string(base_display_e bd)
1244 {
1245     switch(bd) {
1246         case BASE_NONE:
1247             return "BASE_NONE";
1248         case BASE_DEC:
1249             return "BASE_DEC";
1250         case BASE_HEX:
1251             return "BASE_HEX";
1252         case BASE_OCT:
1253             return "BASE_OCT";
1254         case BASE_DEC_HEX:
1255             return "BASE_DEC_HEX";
1256         case BASE_HEX_DEC:
1257             return "BASE_HEX_DEC";
1258         default:
1259             return "n.a.";
1260     }
1261 }
1262
1263 /*
1264  * Copied from various parts of proto.c
1265  */
1266 #define FIELD_STR_INIT_LEN 256
1267 #define cVALS(x) (const value_string*)(x)
1268 static gboolean print_field_value(field_info *finfo, int cmd_line_index)
1269 {
1270     header_field_info   *hfinfo;
1271     static char         *fs_buf = NULL;
1272     char                *fs_ptr = fs_buf;
1273     static GString     *label_s = NULL;
1274     int                 fs_buf_len = FIELD_STR_INIT_LEN, fs_len;
1275     guint              i;
1276     string_fmt_t       *sf;
1277     guint32            uvalue;
1278     gint32             svalue;
1279     guint64            uvalue64;
1280     gint64             svalue64;
1281     const true_false_string *tfstring = &tfs_true_false;
1282
1283     hfinfo = finfo->hfinfo;
1284
1285     if (!fs_buf) {
1286         fs_buf = (char *)g_malloc(fs_buf_len + 1);
1287         fs_ptr = fs_buf;
1288     }
1289
1290     if (!label_s) {
1291         label_s = g_string_new("");
1292     }
1293
1294     if(finfo->value.ftype->val_to_string_repr)
1295     {
1296         /*
1297          * this field has an associated value,
1298          * e.g: ip.hdr_len
1299          */
1300         fs_len = fvalue_string_repr_len(&finfo->value, FTREPR_DFILTER);
1301         while (fs_buf_len < fs_len) {
1302             fs_buf_len *= 2;
1303             fs_buf = (char *)g_realloc(fs_buf, fs_buf_len + 1);
1304             fs_ptr = fs_buf;
1305         }
1306         fvalue_to_string_repr(&finfo->value,
1307                               FTREPR_DFILTER,
1308                               fs_buf);
1309
1310         /* String types are quoted. Remove them. */
1311         if ((finfo->value.ftype->ftype == FT_STRING || finfo->value.ftype->ftype == FT_STRINGZ) && fs_len > 2) {
1312             fs_buf[fs_len - 1] = '\0';
1313             fs_ptr++;
1314         }
1315     }
1316
1317     if (string_fmts->len > 0 && finfo->hfinfo->strings) {
1318         g_string_truncate(label_s, 0);
1319         for (i = 0; i < string_fmts->len; i++) {
1320             sf = (string_fmt_t *)g_ptr_array_index(string_fmts, i);
1321             if (sf->plain) {
1322                 g_string_append(label_s, sf->plain);
1323             } else {
1324                 switch (sf->format) {
1325                     case SF_NAME:
1326                         g_string_append(label_s, hfinfo->name);
1327                         break;
1328                     case SF_NUMVAL:
1329                         g_string_append(label_s, fs_ptr);
1330                         break;
1331                     case SF_STRVAL:
1332                         switch(hfinfo->type) {
1333                             case FT_BOOLEAN:
1334                                 uvalue = fvalue_get_uinteger(&finfo->value);
1335                                 tfstring = (const struct true_false_string*) hfinfo->strings;
1336                                 g_string_append(label_s, uvalue ? tfstring->true_string : tfstring->false_string);
1337                                 break;
1338                             case FT_INT8:
1339                             case FT_INT16:
1340                             case FT_INT24:
1341                             case FT_INT32:
1342                                 DISSECTOR_ASSERT(!hfinfo->bitmask);
1343                                 svalue = fvalue_get_sinteger(&finfo->value);
1344                                 if (hfinfo->display & BASE_RANGE_STRING) {
1345                                     g_string_append(label_s, rval_to_str(svalue, RVALS(hfinfo->strings), "Unknown"));
1346                                 } else if (hfinfo->display & BASE_EXT_STRING) {
1347                                     g_string_append(label_s, val_to_str_ext(svalue, (const value_string_ext *) hfinfo->strings, "Unknown"));
1348                                 } else {
1349                                     g_string_append(label_s, val_to_str(svalue, cVALS(hfinfo->strings), "Unknown"));
1350                                 }
1351                                 break;
1352                             case FT_INT64:
1353                                 DISSECTOR_ASSERT(!hfinfo->bitmask);
1354                                 svalue64 = (gint64)fvalue_get_integer64(&finfo->value);
1355                                 if (hfinfo->display & BASE_VAL64_STRING) {
1356                                     g_string_append(label_s, val64_to_str(svalue64, (const val64_string *)(hfinfo->strings), "Unknown"));
1357                                 }
1358                                 break;
1359                             case FT_UINT8:
1360                             case FT_UINT16:
1361                             case FT_UINT24:
1362                             case FT_UINT32:
1363                                 uvalue = fvalue_get_uinteger(&finfo->value);
1364                                 if (!hfinfo->bitmask && hfinfo->display & BASE_RANGE_STRING) {
1365                                     g_string_append(label_s, rval_to_str(uvalue, RVALS(hfinfo->strings), "Unknown"));
1366                                 } else if (hfinfo->display & BASE_EXT_STRING) {
1367                                     g_string_append(label_s, val_to_str_ext(uvalue, (const value_string_ext *) hfinfo->strings, "Unknown"));
1368                                 } else {
1369                                     g_string_append(label_s, val_to_str(uvalue, cVALS(hfinfo->strings), "Unknown"));
1370                                 }
1371                                 break;
1372                             case FT_UINT64:
1373                                 DISSECTOR_ASSERT(!hfinfo->bitmask);
1374                                 uvalue64 = fvalue_get_integer64(&finfo->value);
1375                                 if (hfinfo->display & BASE_VAL64_STRING) {
1376                                     g_string_append(label_s, val64_to_str(uvalue64, (const val64_string *)(hfinfo->strings), "Unknown"));
1377                                 }
1378                                 break;
1379                             default:
1380                                 break;
1381                         }
1382                         break;
1383                     default:
1384                         break;
1385                 }
1386             }
1387         }
1388         printf(" %u=\"%s\"", cmd_line_index, label_s->str);
1389         return TRUE;
1390     }
1391
1392     if(finfo->value.ftype->val_to_string_repr)
1393     {
1394         printf(" %u=\"%s\"", cmd_line_index, fs_ptr);
1395         return TRUE;
1396     }
1397
1398     /*
1399      * This field doesn't have an associated value,
1400      * e.g. http
1401      * We return n.a.
1402      */
1403     printf(" %u=\"n.a.\"", cmd_line_index);
1404     return TRUE;
1405 }
1406
1407 static int
1408 protocolinfo_packet(void *prs, packet_info *pinfo _U_, epan_dissect_t *edt, const void *dummy _U_)
1409 {
1410     pci_t *rs=(pci_t *)prs;
1411     GPtrArray *gp;
1412     guint i;
1413
1414     gp=proto_get_finfo_ptr_array(edt->tree, rs->hf_index);
1415     if(!gp){
1416         printf(" n.a.");
1417         return 0;
1418     }
1419
1420     /*
1421      * Print each occurrence of the field
1422      */
1423     for (i = 0; i < gp->len; i++) {
1424         print_field_value((field_info *)gp->pdata[i], rs->cmd_line_index);
1425     }
1426
1427     return 0;
1428 }
1429
1430 int g_cmd_line_index = 0;
1431
1432 /*
1433  * field must be persistent - we don't g_strdup() it below
1434  */
1435 static void
1436 protocolinfo_init(char *field)
1437 {
1438     pci_t *rs;
1439     header_field_info *hfi;
1440     GString *error_string;
1441
1442     hfi=proto_registrar_get_byname(field);
1443     if(!hfi){
1444         fprintf(stderr, "rawshark: Field \"%s\" doesn't exist.\n", field);
1445         exit(1);
1446     }
1447
1448     switch (hfi->type) {
1449
1450         case FT_ABSOLUTE_TIME:
1451             printf("%u %s %s - ",
1452                    g_cmd_line_index,
1453                    ftenum_to_string(hfi),
1454                    absolute_time_display_e_to_string((absolute_time_display_e)hfi->display));
1455             break;
1456
1457         default:
1458             printf("%u %s %s - ",
1459                    g_cmd_line_index,
1460                    ftenum_to_string(hfi),
1461                    base_display_e_to_string((base_display_e)hfi->display));
1462             break;
1463     }
1464
1465     rs=(pci_t *)g_malloc(sizeof(pci_t));
1466     rs->hf_index=hfi->id;
1467     rs->filter=field;
1468     rs->cmd_line_index = g_cmd_line_index++;
1469
1470     error_string=register_tap_listener("frame", rs, rs->filter, TL_REQUIRES_PROTO_TREE, NULL, protocolinfo_packet, NULL);
1471     if(error_string){
1472         /* error, we failed to attach to the tap. complain and clean up */
1473         fprintf(stderr, "rawshark: Couldn't register field extraction tap: %s\n",
1474                 error_string->str);
1475         g_string_free(error_string, TRUE);
1476         if(rs->filter){
1477             g_free(rs->filter);
1478         }
1479         g_free(rs);
1480
1481         exit(1);
1482     }
1483 }
1484
1485 /*
1486  * Given a format string, split it into a GPtrArray of string_fmt_t structs
1487  * and fill in string_fmt_parts.
1488  */
1489
1490 static void
1491 add_string_fmt(string_fmt_e format, gchar *plain) {
1492     string_fmt_t *sf = (string_fmt_t *)g_malloc(sizeof(string_fmt_t));
1493
1494     sf->format = format;
1495     sf->plain = g_strdup(plain);
1496
1497     g_ptr_array_add(string_fmts, sf);
1498 }
1499
1500 static gboolean
1501 parse_field_string_format(gchar *format) {
1502     GString *plain_s = g_string_new("");
1503     size_t len;
1504     size_t pos = 0;
1505
1506     if (!format) {
1507         return FALSE;
1508     }
1509
1510     len = strlen(format);
1511     g_ptr_array_set_size(string_fmts, 0);
1512
1513     while (pos < len) {
1514         if (format[pos] == '%') {
1515             if (pos >= len) { /* There should always be a following character */
1516                 return FALSE;
1517             }
1518             pos++;
1519             if (plain_s->len > 0) {
1520                 add_string_fmt(SF_NONE, plain_s->str);
1521                 g_string_truncate(plain_s, 0);
1522             }
1523             switch (format[pos]) {
1524                 case 'D':
1525                     add_string_fmt(SF_NAME, NULL);
1526                     break;
1527                 case 'N':
1528                     add_string_fmt(SF_NUMVAL, NULL);
1529                     break;
1530                 case 'S':
1531                     add_string_fmt(SF_STRVAL, NULL);
1532                     break;
1533                 case '%':
1534                     g_string_append_c(plain_s, '%');
1535                     break;
1536                 default: /* Invalid format */
1537                     return FALSE;
1538             }
1539         } else {
1540             g_string_append_c(plain_s, format[pos]);
1541         }
1542         pos++;
1543     }
1544
1545     if (plain_s->len > 0) {
1546         add_string_fmt(SF_NONE, plain_s->str);
1547     }
1548     g_string_free(plain_s, TRUE);
1549
1550     return TRUE;
1551 }
1552 /****************************************************************************************
1553  * END OF FIELD EXTRACTION ROUTINES
1554  ****************************************************************************************/
1555
1556 static void
1557 show_print_file_io_error(int err)
1558 {
1559     switch (err) {
1560
1561         case ENOSPC:
1562             cmdarg_err("Not all the packets could be printed because there is "
1563                        "no space left on the file system.");
1564             break;
1565
1566 #ifdef EDQUOT
1567         case EDQUOT:
1568             cmdarg_err("Not all the packets could be printed because you are "
1569                        "too close to, or over your disk quota.");
1570             break;
1571 #endif
1572
1573         default:
1574             cmdarg_err("An error occurred while printing packets: %s.",
1575                        g_strerror(err));
1576             break;
1577     }
1578 }
1579
1580 /*
1581  * Open/create errors are reported with an console message in Rawshark.
1582  */
1583 static void
1584 open_failure_message(const char *filename, int err, gboolean for_writing)
1585 {
1586     fprintf(stderr, "rawshark: ");
1587     fprintf(stderr, file_open_error_message(err, for_writing), filename);
1588     fprintf(stderr, "\n");
1589 }
1590
1591 static const nstime_t *
1592 raw_get_frame_ts(void *data _U_, guint32 frame_num)
1593 {
1594     if (ref && ref->num == frame_num)
1595         return &ref->abs_ts;
1596
1597     if (prev_dis && prev_dis->num == frame_num)
1598         return &prev_dis->abs_ts;
1599
1600     if (prev_cap && prev_cap->num == frame_num)
1601         return &prev_cap->abs_ts;
1602
1603     return NULL;
1604 }
1605
1606 static epan_t *
1607 raw_epan_new(capture_file *cf)
1608 {
1609     epan_t *epan = epan_new();
1610
1611     epan->data = cf;
1612     epan->get_frame_ts = raw_get_frame_ts;
1613     epan->get_interface_name = cap_file_get_interface_name;
1614
1615     return epan;
1616 }
1617
1618 cf_status_t
1619 raw_cf_open(capture_file *cf, const char *fname)
1620 {
1621     if ((fd = raw_pipe_open(fname)) < 0)
1622         return CF_ERROR;
1623
1624     /* The open succeeded.  Fill in the information for this file. */
1625
1626     /* Create new epan session for dissection. */
1627     epan_free(cf->epan);
1628     cf->epan = raw_epan_new(cf);
1629
1630     cf->wth = NULL;
1631     cf->f_datalen = 0; /* not used, but set it anyway */
1632
1633     /* Set the file name because we need it to set the follow stream filter.
1634        XXX - is that still true?  We need it for other reasons, though,
1635        in any case. */
1636     cf->filename = g_strdup(fname);
1637
1638     /* Indicate whether it's a permanent or temporary file. */
1639     cf->is_tempfile = FALSE;
1640
1641     /* No user changes yet. */
1642     cf->unsaved_changes = FALSE;
1643
1644     cf->cd_t      = WTAP_FILE_UNKNOWN;
1645     cf->count     = 0;
1646     cf->drops_known = FALSE;
1647     cf->drops     = 0;
1648     cf->has_snap = FALSE;
1649     cf->snap = WTAP_MAX_PACKET_SIZE;
1650     nstime_set_zero(&cf->elapsed_time);
1651     ref = NULL;
1652     prev_dis = NULL;
1653     prev_cap = NULL;
1654
1655     return CF_OK;
1656 }
1657
1658
1659 /*
1660  * General errors are reported with an console message in Rawshark.
1661  */
1662 static void
1663 failure_message(const char *msg_format, va_list ap)
1664 {
1665     fprintf(stderr, "rawshark: ");
1666     vfprintf(stderr, msg_format, ap);
1667     fprintf(stderr, "\n");
1668 }
1669
1670 /*
1671  * Read errors are reported with an console message in Rawshark.
1672  */
1673 static void
1674 read_failure_message(const char *filename, int err)
1675 {
1676     cmdarg_err("An error occurred while reading from the file \"%s\": %s.",
1677                filename, g_strerror(err));
1678 }
1679
1680 /*
1681  * Write errors are reported with an console message in Rawshark.
1682  */
1683 static void
1684 write_failure_message(const char *filename, int err)
1685 {
1686     cmdarg_err("An error occurred while writing to the file \"%s\": %s.",
1687                filename, g_strerror(err));
1688 }
1689
1690 /*
1691  * Report an error in command-line arguments.
1692  */
1693 void
1694 cmdarg_err(const char *fmt, ...)
1695 {
1696     va_list ap;
1697
1698     va_start(ap, fmt);
1699     fprintf(stderr, "rawshark: ");
1700     vfprintf(stderr, fmt, ap);
1701     fprintf(stderr, "\n");
1702     va_end(ap);
1703 }
1704
1705 /*
1706  * Report additional information for an error in command-line arguments.
1707  */
1708 void
1709 cmdarg_err_cont(const char *fmt, ...)
1710 {
1711     va_list ap;
1712
1713     va_start(ap, fmt);
1714     vfprintf(stderr, fmt, ap);
1715     fprintf(stderr, "\n");
1716     va_end(ap);
1717 }
1718
1719
1720 /*
1721  * Editor modelines
1722  *
1723  * Local Variables:
1724  * c-basic-offset: 4
1725  * tab-width: 8
1726  * indent-tabs-mode: nil
1727  * End:
1728  *
1729  * ex: set shiftwidth=4 tabstop=8 expandtab:
1730  * :indentSize=4:tabSize=8:noTabs=true:
1731  */