rawshark.c

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