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