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