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