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