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