wiretap/pcapng.c

   1 /* pcapng.c
   2  *
   3  * $Id$
   4  *
   5  * Wiretap Library
   6  * Copyright (c) 1998 by Gilbert Ramirez <gram@alumni.rice.edu>
   7  *
   8  * File format support for pcap-ng file format
   9  * Copyright (c) 2007 by Ulf Lamping <ulf.lamping@web.de>
  10  *
  11  * This program is free software; you can redistribute it and/or
  12  * modify it under the terms of the GNU General Public License
  13  * as published by the Free Software Foundation; either version 2
  14  * of the License, or (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program; if not, write to the Free Software
  23  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  24  */
  25
  26 /* File format reference:
  27  *   http://www.winpcap.org/ntar/draft/PCAP-DumpFileFormat.html
  28  * Related Wiki page:
  29  *   http://wiki.wireshark.org/Development/PcapNg
  30  */
  31
  32 #ifdef HAVE_CONFIG_H
  33 #include "config.h"
  34 #endif
  35
  36 #include <stdlib.h>
  37 #include <string.h>
  38 #include <errno.h>
  39
  40 /* Needed for addrinfo */
  41 #ifdef HAVE_SYS_TYPES_H
  42 # include <sys/types.h>
  43 #endif
  44
  45 #ifdef HAVE_SYS_SOCKET_H
  46 #include <sys/socket.h>
  47 #endif
  48
  49 #ifdef HAVE_NETINET_IN_H
  50 # include <netinet/in.h>
  51 #endif
  52
  53 #ifdef HAVE_NETDB_H
  54 # include <netdb.h>
  55 #endif
  56
  57 #ifdef HAVE_WINSOCK2_H
  58 # include <winsock2.h>
  59 #endif
  60
  61 #if defined(_WIN32) && defined(INET6)
  62 # include <ws2tcpip.h>
  63 #endif
  64
  65 #include "wtap-int.h"
  66 #include "file_wrappers.h"
  67 #include "buffer.h"
  68 #include "libpcap.h"
  69 #include "pcap-common.h"
  70 #include "pcap-encap.h"
  71 #include "pcapng.h"
  72
  73 #if 0
  74 #define pcapng_debug0(str) g_warning(str)
  75 #define pcapng_debug1(str,p1) g_warning(str,p1)
  76 #define pcapng_debug2(str,p1,p2) g_warning(str,p1,p2)
  77 #define pcapng_debug3(str,p1,p2,p3) g_warning(str,p1,p2,p3)
  78 #else
  79 #define pcapng_debug0(str)
  80 #define pcapng_debug1(str,p1)
  81 #define pcapng_debug2(str,p1,p2)
  82 #define pcapng_debug3(str,p1,p2,p3)
  83 #endif
  84
  85 static gboolean
  86 pcapng_read(wtap *wth, int *err, gchar **err_info,
  87     gint64 *data_offset);
  88 static gboolean
  89 pcapng_seek_read(wtap *wth, gint64 seek_off,
  90     union wtap_pseudo_header *pseudo_header, guchar *pd, int length,
  91     int *err, gchar **err_info);
  92 static void
  93 pcapng_close(wtap *wth);
  94
  95
  96 /* pcapng: common block header for every block type */
  97 typedef struct pcapng_block_header_s {
  98         guint32 block_type;
  99         guint32 block_total_length;
 100         /* x bytes block_body */
 101         /* guint32 block_total_length */
 102 } pcapng_block_header_t;
 103
 104 /* pcapng: section header block */
 105 typedef struct pcapng_section_header_block_s {
 106         /* pcapng_block_header_t */
 107         guint32 magic;
 108         guint16 version_major;
 109         guint16 version_minor;
 110         guint64 section_length; /* might be -1 for unknown */
 111         /* ... Options ... */
 112 } pcapng_section_header_block_t;
 113
 114 /* pcapng: interface description block */
 115 typedef struct pcapng_interface_description_block_s {
 116         guint16 linktype;
 117         guint16 reserved;
 118         guint32 snaplen;
 119         /* ... Options ... */
 120 } pcapng_interface_description_block_t;
 121
 122 /* pcapng: packet block (obsolete) */
 123 typedef struct pcapng_packet_block_s {
 124         guint16 interface_id;
 125         guint16 drops_count;
 126         guint32 timestamp_high;
 127         guint32 timestamp_low;
 128         guint32 captured_len;
 129         guint32 packet_len;
 130         /* ... Packet Data ... */
 131         /* ... Padding ... */
 132         /* ... Options ... */
 133 } pcapng_packet_block_t;
 134
 135 /* pcapng: enhanced packet block */
 136 typedef struct pcapng_enhanced_packet_block_s {
 137         guint32 interface_id;
 138         guint32 timestamp_high;
 139         guint32 timestamp_low;
 140         guint32 captured_len;
 141         guint32 packet_len;
 142         /* ... Packet Data ... */
 143         /* ... Padding ... */
 144         /* ... Options ... */
 145 } pcapng_enhanced_packet_block_t;
 146
 147 /* pcapng: simple packet block */
 148 typedef struct pcapng_simple_packet_block_s {
 149         guint32 packet_len;
 150         /* ... Packet Data ... */
 151         /* ... Padding ... */
 152 } pcapng_simple_packet_block_t;
 153
 154 /* pcapng: simple packet block */
 155 typedef struct pcapng_name_resolution_block_s {
 156         guint16 record_type;
 157         guint16 record_len;
 158         /* ... Record ... */
 159 } pcapng_name_resolution_block_t;
 160
 161 /* pcapng: interface statistics block */
 162 typedef struct pcapng_interface_statistics_block_s {
 163         guint32 interface_id;
 164         guint32 timestamp_high;
 165         guint32 timestamp_low;
 166         /* ... Options ... */
 167 } pcapng_interface_statistics_block_t;
 168
 169 /* pcapng: common option header for every option type */
 170 typedef struct pcapng_option_header_s {
 171         guint16 option_code;
 172         guint16 option_length;
 173         /* ... x bytes Option Body ... */
 174     /* ... Padding ... */
 175 } pcapng_option_header_t;
 176
 177 /* Block types */
 178 #define BLOCK_TYPE_IDB 0x00000001 /* Interface Description Block */
 179 #define BLOCK_TYPE_PB  0x00000002 /* Packet Block (obsolete) */
 180 #define BLOCK_TYPE_SPB 0x00000003 /* Simple Packet Block */
 181 #define BLOCK_TYPE_NRB 0x00000004 /* Name Resolution Block */
 182 #define BLOCK_TYPE_ISB 0x00000005 /* Interface Statistics Block */
 183 #define BLOCK_TYPE_EPB 0x00000006 /* Enhanced Packet Block */
 184 #define BLOCK_TYPE_SHB 0x0A0D0D0A /* Section Header Block */
 185
 186
 187
 188 /* Capture section */
 189 typedef struct wtapng_section_s {
 190         /* mandatory */
 191         guint64                         section_length;
 192         /* options */
 193         gchar                           *opt_comment;   /* NULL if not available */
 194         gchar                           *shb_hardware;  /* NULL if not available */
 195         gchar                           *shb_os;        /* NULL if not available */
 196         gchar                           *shb_user_appl; /* NULL if not available */
 197 } wtapng_section_t;
 198
 199 /* Interface Description */
 200 typedef struct wtapng_if_descr_s {
 201         /* mandatory */
 202         guint16                         link_type;
 203         guint32                         snap_len;
 204         /* options */
 205         gchar                           *opt_comment;   /* NULL if not available */
 206         gchar                           *if_name;       /* NULL if not available */
 207         gchar                           *if_description;/* NULL if not available */
 208         /* XXX: if_IPv4addr */
 209         /* XXX: if_IPv6addr */
 210         /* XXX: if_MACaddr */
 211         /* XXX: if_EUIaddr */
 212         guint64                         if_speed;       /* 0xFFFFFFFF if unknown */
 213         guint8                          if_tsresol;     /* default is 6 for microsecond resolution */
 214         gchar                           *if_filter;     /* NULL if not available */
 215         gchar                           *if_os;         /* NULL if not available */
 216         gint8                           if_fcslen;      /* -1 if unknown or changes between packets */
 217         /* XXX: guint64 if_tsoffset; */
 218 } wtapng_if_descr_t;
 219
 220 /* Packets */
 221 typedef struct wtapng_packet_s {
 222         /* mandatory */
 223         guint32                         ts_high;        /* seconds since 1.1.1970 */
 224         guint32                         ts_low;         /* fraction of seconds, depends on if_tsresol */
 225         guint32                         cap_len;        /* data length in the file */
 226         guint32                         packet_len;     /* data length on the wire */
 227         guint32                         interface_id;   /* identifier of the interface. */
 228         guint16                         drops_count;    /* drops count, only valid for packet block */
 229                                                         /* 0xffff if information no available */
 230         /* options */
 231         gchar                           *opt_comment;   /* NULL if not available */
 232         guint64                         drop_count;
 233         guint32                         pack_flags;     /* XXX - 0 for now (any value for "we don't have it"?) */
 234         /* pack_hash */
 235
 236         guint32                         pseudo_header_len;
 237         int                             wtap_encap;
 238         /* XXX - put the packet data / pseudo_header here as well? */
 239 } wtapng_packet_t;
 240
 241 /* Simple Packets */
 242 typedef struct wtapng_simple_packet_s {
 243         /* mandatory */
 244         guint32                         cap_len;        /* data length in the file */
 245         guint32                         packet_len;     /* data length on the wire */
 246         guint32                         pseudo_header_len;
 247         int                             wtap_encap;
 248         /* XXX - put the packet data / pseudo_header here as well? */
 249 } wtapng_simple_packet_t;
 250
 251 /* Name Resolution */
 252 typedef struct wtapng_name_res_s {
 253         /* options */
 254         gchar                           *opt_comment;   /* NULL if not available */
 255         /* XXX */
 256 } wtapng_name_res_t;
 257
 258 /* Interface Statistics */
 259 typedef struct wtapng_if_stats_s {
 260         /* mandatory */
 261         guint64                         interface_id;
 262         guint32                         ts_high;
 263         guint32                         ts_low;
 264         /* options */
 265         gchar                           *opt_comment;   /* NULL if not available */
 266         /* XXX */
 267         /*guint32                               isb_starttime_high;*/
 268         /*guint32                               isb_starttime_low;*/
 269         /*guint32                               isb_endtime_high;*/
 270         /*guint32                               isb_endtime_low;*/
 271         guint64                         isb_ifrecv;
 272         guint64                         isb_ifdrop;
 273         /*guint64                               isb_filteraccept;*/
 274         /*guint64                               isb_osdrop;*/
 275         /*guint64                               isb_usrdeliv;*/
 276 } wtapng_if_stats_t;
 277
 278
 279 typedef struct wtapng_block_s {
 280         guint32                                 type;           /* block_type as defined by pcapng */
 281         union {
 282                 wtapng_section_t        section;
 283                 wtapng_if_descr_t       if_descr;
 284                 wtapng_packet_t         packet;
 285                 wtapng_simple_packet_t  simple_packet;
 286                 wtapng_name_res_t       name_res;
 287                 wtapng_if_stats_t       if_stats;
 288         } data;
 289
 290         /*
 291          * XXX - currently don't know how to handle these!
 292          *
 293          * For one thing, when we're reading a block, they must be
 294          * writable, i.e. not const, so that we can read into them,
 295          * but, when we're writing a block, they can be const, and,
 296          * in fact, they sometimes point to const values.
 297          */
 298         const union wtap_pseudo_header *pseudo_header;
 299         struct wtap_pkthdr *packet_header;
 300         const guchar *frame_buffer;
 301         int *file_encap;
 302 } wtapng_block_t;
 303
 304 typedef struct interface_data_s {
 305         int wtap_encap;
 306         guint64 time_units_per_second;
 307 } interface_data_t;
 308
 309
 310 typedef struct {
 311         gboolean byte_swapped;
 312         guint16 version_major;
 313         guint16 version_minor;
 314         gint8 if_fcslen;
 315         GArray *interface_data;
 316         guint number_of_interfaces;
 317         wtap_new_ipv4_callback_t add_new_ipv4;
 318         wtap_new_ipv6_callback_t add_new_ipv6;
 319 } pcapng_t;
 320
 321 static int
 322 pcapng_get_encap(gint id, pcapng_t *pn)
 323 {
 324         interface_data_t int_data;
 325
 326         if ((id >= 0) && ((guint)id < pn->number_of_interfaces)) {
 327                 int_data = g_array_index(pn->interface_data, interface_data_t, id);
 328                 return int_data.wtap_encap;
 329         } else {
 330                 return WTAP_ERR_UNSUPPORTED_ENCAP;
 331         }
 332 }
 333
 334
 335 static int
 336 pcapng_read_option(FILE_T fh, pcapng_t *pn, pcapng_option_header_t *oh,
 337                    char *content, int len, int *err, gchar **err_info)
 338 {
 339         int     bytes_read;
 340         int     block_read;
 341         guint64 file_offset64;
 342
 343
 344         /* read option header */
 345         errno = WTAP_ERR_CANT_READ;
 346         bytes_read = file_read(oh, sizeof (*oh), fh);
 347         if (bytes_read != sizeof (*oh)) {
 348             pcapng_debug0("pcapng_read_option: failed to read option");
 349             *err = file_error(fh, err_info);
 350             if (*err != 0)
 351                     return -1;
 352             return 0;
 353         }
 354         block_read = sizeof (*oh);
 355         if(pn->byte_swapped) {
 356                 oh->option_code      = BSWAP16(oh->option_code);
 357                 oh->option_length    = BSWAP16(oh->option_length);
 358         }
 359
 360         /* sanity check: option length */
 361         if (oh->option_length > len) {
 362                 pcapng_debug2("pcapng_read_option: option_length %u larger than buffer (%u)",
 363                               oh->option_length, len);
 364                 return 0;
 365         }
 366
 367         /* read option content */
 368         errno = WTAP_ERR_CANT_READ;
 369         bytes_read = file_read(content, oh->option_length, fh);
 370         if (bytes_read != oh->option_length) {
 371                 pcapng_debug1("pcapng_read_if_descr_block: failed to read content of option %u", oh->option_code);
 372                 *err = file_error(fh, err_info);
 373                 if (*err != 0)
 374                         return -1;
 375                 return 0;
 376         }
 377         block_read += oh->option_length;
 378
 379         /* jump over potential padding bytes at end of option */
 380         if( (oh->option_length % 4) != 0) {
 381                 file_offset64 = file_seek(fh, 4 - (oh->option_length % 4), SEEK_CUR, err);
 382                 if (file_offset64 <= 0) {
 383                         if (*err != 0)
 384                                 return -1;
 385                         return 0;
 386                 }
 387                 block_read += 4 - (oh->option_length % 4);
 388         }
 389
 390         return block_read;
 391 }
 392
 393
 394 static int
 395 pcapng_read_section_header_block(FILE_T fh, gboolean first_block,
 396                                  pcapng_block_header_t *bh, pcapng_t *pn,
 397                                  wtapng_block_t *wblock, int *err,
 398                                  gchar **err_info)
 399 {
 400         int     bytes_read;
 401         int     block_read;
 402         int to_read;
 403         pcapng_section_header_block_t shb;
 404         pcapng_option_header_t oh;
 405         char option_content[100]; /* XXX - size might need to be increased, if we see longer options */
 406
 407
 408         /* read block content */
 409         errno = WTAP_ERR_CANT_READ;
 410         bytes_read = file_read(&shb, sizeof shb, fh);
 411         if (bytes_read != sizeof shb) {
 412                 *err = file_error(fh, err_info);
 413                 if (*err == 0) {
 414                         if (first_block) {
 415                                 /*
 416                                  * We're reading this as part of an open,
 417                                  * and this block is too short to be
 418                                  * an SHB, so the file is too short
 419                                  * to be a pcap-ng file.
 420                                  */
 421                                 return 0;
 422                         }
 423
 424                         /*
 425                          * Otherwise, just report this as an error.
 426                          */
 427                         *err = WTAP_ERR_SHORT_READ;
 428                 }
 429                 return -1;
 430         }
 431         block_read = bytes_read;
 432
 433         /* is the magic number one we expect? */
 434         switch(shb.magic) {
 435             case(0x1A2B3C4D):
 436                 /* this seems pcapng with correct byte order */
 437                 pn->byte_swapped                = FALSE;
 438                 pn->version_major               = shb.version_major;
 439                 pn->version_minor               = shb.version_minor;
 440
 441                 pcapng_debug3("pcapng_read_section_header_block: SHB (little endian) V%u.%u, len %u",
 442                                 pn->version_major, pn->version_minor, bh->block_total_length);
 443                 break;
 444             case(0x4D3C2B1A):
 445                 /* this seems pcapng with swapped byte order */
 446                 pn->byte_swapped                = TRUE;
 447                 pn->version_major               = BSWAP16(shb.version_major);
 448                 pn->version_minor               = BSWAP16(shb.version_minor);
 449
 450                 /* tweak the block length to meet current swapping that we know now */
 451                 bh->block_total_length  = BSWAP32(bh->block_total_length);
 452
 453                 pcapng_debug3("pcapng_read_section_header_block: SHB (big endian) V%u.%u, len %u",
 454                                 pn->version_major, pn->version_minor, bh->block_total_length);
 455                 break;
 456             default:
 457                 /* Not a "pcapng" magic number we know about. */
 458                 if (first_block) {
 459                         /* Not a pcap-ng file. */
 460                         return 0;
 461                 }
 462
 463                 /* A bad block */
 464                 *err = WTAP_ERR_BAD_RECORD;
 465                 *err_info = g_strdup_printf("pcapng_read_section_header_block: unknown byte-order magic number 0x%08x", shb.magic);
 466                 return 0;
 467         }
 468
 469         /* OK, at this point we assume it's a pcap-ng file. */
 470
 471         /* we currently only understand SHB V1.0 */
 472         if (pn->version_major != 1 || pn->version_minor > 0) {
 473                 *err = WTAP_ERR_UNSUPPORTED;
 474                 *err_info = g_strdup_printf("pcapng_read_section_header_block: unknown SHB version %u.%u",
 475                               pn->version_major, pn->version_minor);
 476                 return -1;
 477         }
 478
 479         /* 64bit section_length (currently unused) */
 480         if (pn->byte_swapped) {
 481                 wblock->data.section.section_length = BSWAP64(shb.section_length);
 482         } else {
 483                 wblock->data.section.section_length = shb.section_length;
 484         }
 485
 486         /* Option defaults */
 487         wblock->data.section.opt_comment        = NULL;
 488         wblock->data.section.shb_hardware       = NULL;
 489         wblock->data.section.shb_os             = NULL;
 490         wblock->data.section.shb_user_appl      = NULL;
 491
 492         /* Options */
 493         errno = WTAP_ERR_CANT_READ;
 494         to_read = bh->block_total_length
 495         - (int)sizeof(pcapng_block_header_t)
 496         - (int)sizeof (pcapng_section_header_block_t)
 497         - (int)sizeof(bh->block_total_length);
 498         while(to_read > 0) {
 499                 /* read option */
 500                 bytes_read = pcapng_read_option(fh, pn, &oh, option_content, sizeof(option_content), err, err_info);
 501                 if (bytes_read <= 0) {
 502                         pcapng_debug0("pcapng_read_section_header_block: failed to read option");
 503                         return bytes_read;
 504                 }
 505                 block_read += bytes_read;
 506                 to_read -= bytes_read;
 507
 508                 /* handle option content */
 509                 switch(oh.option_code) {
 510                     case(0): /* opt_endofopt */
 511                         if(to_read != 0) {
 512                                 pcapng_debug1("pcapng_read_section_header_block: %u bytes after opt_endofopt", to_read);
 513                         }
 514                         /* padding should be ok here, just get out of this */
 515                         to_read = 0;
 516                         break;
 517                     case(1): /* opt_comment */
 518                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 519                                 wblock->data.section.opt_comment = g_strndup(option_content, sizeof(option_content));
 520                                 pcapng_debug1("pcapng_read_section_header_block: opt_comment %s", wblock->data.section.opt_comment);
 521                         } else {
 522                                 pcapng_debug1("pcapng_read_section_header_block: opt_comment length %u seems strange", oh.option_length);
 523                         }
 524                         break;
 525                     case(2): /* shb_hardware */
 526                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 527                                 wblock->data.section.shb_hardware = g_strndup(option_content, sizeof(option_content));
 528                                 pcapng_debug1("pcapng_read_section_header_block: shb_hardware %s", wblock->data.section.shb_hardware);
 529                         } else {
 530                                 pcapng_debug1("pcapng_read_section_header_block: shb_hardware length %u seems strange", oh.option_length);
 531                         }
 532                         break;
 533                     case(3): /* shb_os */
 534                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 535                                 wblock->data.section.shb_os = g_strndup(option_content, sizeof(option_content));
 536                                 pcapng_debug1("pcapng_read_section_header_block: shb_os %s", wblock->data.section.shb_os);
 537                         } else {
 538                                 pcapng_debug1("pcapng_read_section_header_block: shb_os length %u seems strange", oh.option_length);
 539                         }
 540                         break;
 541                     case(4): /* shb_userappl */
 542                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 543                                 wblock->data.section.shb_user_appl = g_strndup(option_content, sizeof(option_content));
 544                                 pcapng_debug1("pcapng_read_section_header_block: shb_userappl %s", wblock->data.section.shb_user_appl);
 545                         } else {
 546                                 pcapng_debug1("pcapng_read_section_header_block: shb_userappl length %u seems strange", oh.option_length);
 547                         }
 548                         break;
 549                     default:
 550                         pcapng_debug2("pcapng_read_section_header_block: unknown option %u - ignoring %u bytes",
 551                                       oh.option_code, oh.option_length);
 552                 }
 553         }
 554
 555         if (pn->interface_data != NULL) {
 556                 g_array_free(pn->interface_data, TRUE);
 557                 pn->interface_data = NULL;
 558                 *err = WTAP_ERR_BAD_RECORD;
 559                 *err_info = g_strdup_printf("pcapng: multiple section header blocks not supported.");
 560                 return 0;
 561         }
 562         pn->interface_data = g_array_new(FALSE, FALSE, sizeof(interface_data_t));
 563         pn->number_of_interfaces = 0;
 564
 565         return block_read;
 566 }
 567
 568
 569 /* "Interface Description Block" */
 570 static int
 571 pcapng_read_if_descr_block(FILE_T fh, pcapng_block_header_t *bh, pcapng_t *pn,
 572                            wtapng_block_t *wblock, int *err, gchar **err_info)
 573 {
 574         guint64 time_units_per_second;
 575         int     bytes_read;
 576         int     block_read;
 577         int to_read;
 578         pcapng_interface_description_block_t idb;
 579         pcapng_option_header_t oh;
 580         interface_data_t int_data;
 581         gint encap;
 582         char option_content[100]; /* XXX - size might need to be increased, if we see longer options */
 583
 584
 585         time_units_per_second = 1000000; /* default */
 586         /* read block content */
 587         errno = WTAP_ERR_CANT_READ;
 588         bytes_read = file_read(&idb, sizeof idb, fh);
 589         if (bytes_read != sizeof idb) {
 590                 pcapng_debug0("pcapng_read_if_descr_block: failed to read IDB");
 591                 *err = file_error(fh, err_info);
 592                 if (*err != 0)
 593                         return -1;
 594                 return 0;
 595         }
 596         block_read = bytes_read;
 597
 598         /* mandatory values */
 599         if (pn->byte_swapped) {
 600                 wblock->data.if_descr.link_type = BSWAP16(idb.linktype);
 601                 wblock->data.if_descr.snap_len  = BSWAP32(idb.snaplen);
 602         } else {
 603                 wblock->data.if_descr.link_type = idb.linktype;
 604                 wblock->data.if_descr.snap_len  = idb.snaplen;
 605         }
 606
 607         pcapng_debug3("pcapng_read_if_descr_block: IDB link_type %u (%s), snap %u",
 608                       wblock->data.if_descr.link_type,
 609                       wtap_encap_string(wtap_pcap_encap_to_wtap_encap(wblock->data.if_descr.link_type)),
 610                       wblock->data.if_descr.snap_len);
 611
 612         if (wblock->data.if_descr.snap_len > WTAP_MAX_PACKET_SIZE) {
 613                 /* This is unrealisitic, but text2pcap currently uses 102400.
 614                  * We do not use this value, maybe we should check the
 615                  * snap_len of the packets against it. For now, only warn.
 616                  */
 617                 pcapng_debug1("pcapng_read_if_descr_block: snapshot length %u unrealistic.",
 618                               wblock->data.if_descr.snap_len);
 619                 /*wblock->data.if_descr.snap_len = WTAP_MAX_PACKET_SIZE;*/
 620         }
 621
 622         /* Option defaults */
 623         wblock->data.if_descr.opt_comment       = NULL;
 624         wblock->data.if_descr.if_name           = NULL;
 625         wblock->data.if_descr.if_description    = NULL;
 626         /* XXX: if_IPv4addr */
 627         /* XXX: if_IPv6addr */
 628         /* XXX: if_MACaddr */
 629         /* XXX: if_EUIaddr */
 630         wblock->data.if_descr.if_speed          = 0xFFFFFFFF;   /* "unknown" */
 631         wblock->data.if_descr.if_tsresol        = 6;            /* default is 6 for microsecond resolution */
 632         wblock->data.if_descr.if_filter         = NULL;
 633         wblock->data.if_descr.if_os             = NULL;
 634         wblock->data.if_descr.if_fcslen         = -1;           /* unknown or changes between packets */
 635         /* XXX: guint64 if_tsoffset; */
 636
 637
 638         /* Options */
 639         errno = WTAP_ERR_CANT_READ;
 640         to_read = bh->block_total_length
 641         - (int)sizeof(pcapng_block_header_t)
 642         - (int)sizeof (pcapng_interface_description_block_t)
 643         - (int)sizeof(bh->block_total_length);
 644         while (to_read > 0) {
 645                 /* read option */
 646                 bytes_read = pcapng_read_option(fh, pn, &oh, option_content, sizeof(option_content), err, err_info);
 647                 if (bytes_read <= 0) {
 648                         pcapng_debug0("pcapng_read_if_descr_block: failed to read option");
 649                         return bytes_read;
 650                 }
 651                 block_read += bytes_read;
 652                 to_read -= bytes_read;
 653
 654                 /* handle option content */
 655                 switch(oh.option_code) {
 656                     case(0): /* opt_endofopt */
 657                         if(to_read != 0) {
 658                                 pcapng_debug1("pcapng_read_if_descr_block: %u bytes after opt_endofopt", to_read);
 659                         }
 660                         /* padding should be ok here, just get out of this */
 661                         to_read = 0;
 662                         break;
 663                     case(1): /* opt_comment */
 664                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 665                                 wblock->data.section.opt_comment = g_strndup(option_content, sizeof(option_content));
 666                                 pcapng_debug1("pcapng_read_if_descr_block: opt_comment %s", wblock->data.section.opt_comment);
 667                         } else {
 668                                 pcapng_debug1("pcapng_read_if_descr_block: opt_comment length %u seems strange", oh.option_length);
 669                         }
 670                         break;
 671                     case(2): /* if_name */
 672                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 673                                 wblock->data.if_descr.if_name = g_strndup(option_content, sizeof(option_content));
 674                                 pcapng_debug1("pcapng_read_if_descr_block: if_name %s", wblock->data.if_descr.if_name);
 675                         } else {
 676                                 pcapng_debug1("pcapng_read_if_descr_block: if_name length %u seems strange", oh.option_length);
 677                         }
 678                         break;
 679                     case(3): /* if_description */
 680                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 681                             wblock->data.if_descr.if_description = g_strndup(option_content, sizeof(option_content));
 682                                 pcapng_debug1("pcapng_read_if_descr_block: if_description %s", wblock->data.if_descr.if_description);
 683                         } else {
 684                                 pcapng_debug1("pcapng_read_if_descr_block: if_description length %u seems strange", oh.option_length);
 685                         }
 686                         break;
 687                     case(8): /* if_speed */
 688                         if(oh.option_length == 8) {
 689                                 /*  Don't cast a char[] into a guint64--the
 690                                  *  char[] may not be aligned correctly.
 691                                  */
 692                                 memcpy(&wblock->data.if_descr.if_speed, option_content, sizeof(guint64));
 693                                 if(pn->byte_swapped)
 694                                         wblock->data.if_descr.if_speed = BSWAP64(wblock->data.if_descr.if_speed);
 695                                 pcapng_debug1("pcapng_read_if_descr_block: if_speed %" G_GINT64_MODIFIER "u (bps)", wblock->data.if_descr.if_speed);
 696                         } else {
 697                                     pcapng_debug1("pcapng_read_if_descr_block: if_speed length %u not 8 as expected", oh.option_length);
 698                         }
 699                         break;
 700                     case(9): /* if_tsresol */
 701                         if (oh.option_length == 1) {
 702                                 guint64 base;
 703                                 guint64 result;
 704                                 guint8 i, exponent;
 705
 706                                 wblock->data.if_descr.if_tsresol = option_content[0];
 707                                 if (wblock->data.if_descr.if_tsresol & 0x80) {
 708                                         base = 2;
 709                                 } else {
 710                                         base = 10;
 711                                 }
 712                                 exponent = (guint8)(wblock->data.if_descr.if_tsresol & 0x7f);
 713                                 if (((base == 2) && (exponent < 64)) || ((base == 10) && (exponent < 20))) {
 714                                         result = 1;
 715                                         for (i = 0; i < exponent; i++) {
 716                                                 result *= base;
 717                                         }
 718                                         time_units_per_second = result;
 719                                 } else {
 720                                         time_units_per_second = G_MAXUINT64;
 721                                 }
 722                                 if (time_units_per_second > (((guint64)1) << 32)) {
 723                                         pcapng_debug0("pcapng_open: time conversion might be inaccurate");
 724                                 }
 725                                 pcapng_debug1("pcapng_read_if_descr_block: if_tsresol %u", wblock->data.if_descr.if_tsresol);
 726                         } else {
 727                                 pcapng_debug1("pcapng_read_if_descr_block: if_tsresol length %u not 1 as expected", oh.option_length);
 728                         }
 729                         break;
 730                     case(11): /* if_filter */
 731                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 732                                 wblock->data.if_descr.if_filter = g_strndup(option_content, sizeof(option_content));
 733                                 pcapng_debug1("pcapng_read_if_descr_block: if_filter %s", wblock->data.if_descr.if_filter);
 734                         } else {
 735                                 pcapng_debug1("pcapng_read_if_descr_block: if_filter length %u seems strange", oh.option_length);
 736                         }
 737                         break;
 738                     case(13): /* if_fcslen */
 739                         if(oh.option_length == 1) {
 740                                 wblock->data.if_descr.if_fcslen = option_content[0];
 741                                 pn->if_fcslen = wblock->data.if_descr.if_fcslen;
 742                                 pcapng_debug1("pcapng_read_if_descr_block: if_fcslen %u", wblock->data.if_descr.if_fcslen);
 743                                 /* XXX - add sanity check */
 744                         } else {
 745                                 pcapng_debug1("pcapng_read_if_descr_block: if_fcslen length %u not 1 as expected", oh.option_length);
 746                         }
 747                         break;
 748                     default:
 749                         pcapng_debug2("pcapng_read_if_descr_block: unknown option %u - ignoring %u bytes",
 750                                       oh.option_code, oh.option_length);
 751                 }
 752         }
 753
 754         encap = wtap_pcap_encap_to_wtap_encap(wblock->data.if_descr.link_type);
 755         if (*wblock->file_encap == WTAP_ENCAP_UNKNOWN) {
 756                 *wblock->file_encap = encap;
 757         } else {
 758                 if (*wblock->file_encap != encap) {
 759                         *wblock->file_encap = WTAP_ENCAP_PER_PACKET;
 760                 }
 761         }
 762
 763         int_data.wtap_encap = encap;
 764         int_data.time_units_per_second = time_units_per_second;
 765         g_array_append_val(pn->interface_data, int_data);
 766         pn->number_of_interfaces++;
 767         return block_read;
 768 }
 769
 770
 771 static int
 772 pcapng_read_packet_block(FILE_T fh, pcapng_block_header_t *bh, pcapng_t *pn, wtapng_block_t *wblock, int *err, gchar **err_info, gboolean enhanced)
 773 {
 774         int bytes_read;
 775         int block_read;
 776         int to_read;
 777         guint64 file_offset64;
 778         pcapng_enhanced_packet_block_t epb;
 779         pcapng_packet_block_t pb;
 780         guint32 block_total_length;
 781         pcapng_option_header_t oh;
 782         gint wtap_encap;
 783         int pseudo_header_len;
 784         char option_content[100]; /* XXX - size might need to be increased, if we see longer options */
 785
 786
 787         /* "(Enhanced) Packet Block" read fixed part */
 788         errno = WTAP_ERR_CANT_READ;
 789         if (enhanced) {
 790                 bytes_read = file_read(&epb, sizeof epb, fh);
 791                 if (bytes_read != sizeof epb) {
 792                         pcapng_debug0("pcapng_read_packet_block: failed to read packet data");
 793                         *err = file_error(fh, err_info);
 794                         return 0;
 795                 }
 796                 block_read = bytes_read;
 797
 798                 if (pn->byte_swapped) {
 799                         wblock->data.packet.interface_id        = BSWAP32(epb.interface_id);
 800                         wblock->data.packet.drops_count         = -1; /* invalid */
 801                         wblock->data.packet.ts_high             = BSWAP32(epb.timestamp_high);
 802                         wblock->data.packet.ts_low              = BSWAP32(epb.timestamp_low);
 803                         wblock->data.packet.cap_len             = BSWAP32(epb.captured_len);
 804                         wblock->data.packet.packet_len          = BSWAP32(epb.packet_len);
 805                 } else {
 806                         wblock->data.packet.interface_id        = epb.interface_id;
 807                         wblock->data.packet.drops_count         = -1; /* invalid */
 808                         wblock->data.packet.ts_high             = epb.timestamp_high;
 809                         wblock->data.packet.ts_low              = epb.timestamp_low;
 810                         wblock->data.packet.cap_len             = epb.captured_len;
 811                         wblock->data.packet.packet_len          = epb.packet_len;
 812                 }
 813         } else {
 814                 bytes_read = file_read(&pb, sizeof pb, fh);
 815                 if (bytes_read != sizeof pb) {
 816                         pcapng_debug0("pcapng_read_packet_block: failed to read packet data");
 817                         *err = file_error(fh, err_info);
 818                         return 0;
 819                 }
 820                 block_read = bytes_read;
 821
 822                 if (pn->byte_swapped) {
 823                         wblock->data.packet.interface_id        = BSWAP16(pb.interface_id);
 824                         wblock->data.packet.drops_count         = BSWAP16(pb.drops_count);
 825                         wblock->data.packet.ts_high             = BSWAP32(pb.timestamp_high);
 826                         wblock->data.packet.ts_low              = BSWAP32(pb.timestamp_low);
 827                         wblock->data.packet.cap_len             = BSWAP32(pb.captured_len);
 828                         wblock->data.packet.packet_len          = BSWAP32(pb.packet_len);
 829                 } else {
 830                         wblock->data.packet.interface_id        = pb.interface_id;
 831                         wblock->data.packet.drops_count         = pb.drops_count;
 832                         wblock->data.packet.ts_high             = pb.timestamp_high;
 833                         wblock->data.packet.ts_low              = pb.timestamp_low;
 834                         wblock->data.packet.cap_len             = pb.captured_len;
 835                         wblock->data.packet.packet_len          = pb.packet_len;
 836                 }
 837         }
 838
 839         if (wblock->data.packet.cap_len > wblock->data.packet.packet_len) {
 840                 *err = WTAP_ERR_BAD_RECORD;
 841                 *err_info = g_strdup_printf("pcapng_read_packet_block: cap_len %u is larger than packet_len %u.",
 842                     wblock->data.packet.cap_len, wblock->data.packet.packet_len);
 843                 return 0;
 844         }
 845         if (wblock->data.packet.cap_len > WTAP_MAX_PACKET_SIZE) {
 846                 *err = WTAP_ERR_BAD_RECORD;
 847                 *err_info = g_strdup_printf("pcapng_read_packet_block: cap_len %u is larger than WTAP_MAX_PACKET_SIZE %u.",
 848                     wblock->data.packet.cap_len, WTAP_MAX_PACKET_SIZE);
 849                 return 0;
 850         }
 851         pcapng_debug3("pcapng_read_packet_block: packet data: packet_len %u captured_len %u interface_id %u",
 852                       wblock->data.packet.packet_len,
 853                       wblock->data.packet.cap_len,
 854                       wblock->data.packet.interface_id);
 855         if (wblock->data.packet.packet_len > WTAP_MAX_PACKET_SIZE) {
 856                 *err = WTAP_ERR_BAD_RECORD;
 857                 *err_info = g_strdup_printf("pcapng_read_packet_block: packet_len %u is larger than WTAP_MAX_PACKET_SIZE %u.",
 858                     wblock->data.packet.packet_len, WTAP_MAX_PACKET_SIZE);
 859                 return 0;
 860         }
 861
 862         wtap_encap = pcapng_get_encap(wblock->data.packet.interface_id, pn);
 863         pcapng_debug3("pcapng_read_packet_block: encapsulation = %d (%s), pseudo header size = %d.",
 864                        wtap_encap,
 865                        wtap_encap_string(wtap_encap),
 866                        pcap_get_phdr_size(wtap_encap, wblock->pseudo_header));
 867
 868         memset((void *)wblock->pseudo_header, 0, sizeof(union wtap_pseudo_header));
 869         pseudo_header_len = pcap_process_pseudo_header(fh,
 870                                                        WTAP_FILE_PCAPNG,
 871                                                        wtap_encap,
 872                                                        wblock->data.packet.cap_len,
 873                                                        TRUE,
 874                                                        wblock->packet_header,
 875                                                        (union wtap_pseudo_header *)wblock->pseudo_header,
 876                                                        err,
 877                                                        err_info);
 878         if (pseudo_header_len < 0) {
 879                 return 0;
 880         }
 881         wblock->data.packet.pseudo_header_len = (guint32)pseudo_header_len;
 882         block_read += pseudo_header_len;
 883         if (pseudo_header_len != pcap_get_phdr_size(wtap_encap, wblock->pseudo_header)) {
 884                 pcapng_debug1("pcapng_read_packet_block: Could only read %d bytes for pseudo header.",
 885                               pseudo_header_len);
 886         }
 887
 888         /* "(Enhanced) Packet Block" read capture data */
 889         errno = WTAP_ERR_CANT_READ;
 890         bytes_read = file_read((guchar *) (wblock->frame_buffer), wblock->data.packet.cap_len - pseudo_header_len, fh);
 891         if (bytes_read != (int) (wblock->data.packet.cap_len - pseudo_header_len)) {
 892                 *err = file_error(fh, err_info);
 893                 pcapng_debug1("pcapng_read_packet_block: couldn't read %u bytes of captured data",
 894                               wblock->data.packet.cap_len - pseudo_header_len);
 895                 if (*err == 0)
 896                         *err = WTAP_ERR_SHORT_READ;
 897                 return 0;
 898         }
 899         block_read += bytes_read;
 900
 901         /* jump over potential padding bytes at end of the packet data */
 902         if( (wblock->data.packet.cap_len % 4) != 0) {
 903                 file_offset64 = file_seek(fh, 4 - (wblock->data.packet.cap_len % 4), SEEK_CUR, err);
 904                 if (file_offset64 <= 0) {
 905                         if (*err != 0)
 906                                 return -1;
 907                         return 0;
 908                 }
 909                 block_read += 4 - (wblock->data.packet.cap_len % 4);
 910         }
 911
 912         /* add padding bytes to "block total length" */
 913         /* (the "block total length" of some example files don't contain the packet data padding bytes!) */
 914         if (bh->block_total_length % 4) {
 915                 block_total_length = bh->block_total_length + 4 - (bh->block_total_length % 4);
 916         } else {
 917                 block_total_length = bh->block_total_length;
 918         }
 919
 920         /* Option defaults */
 921         wblock->data.packet.opt_comment = NULL;
 922         wblock->data.packet.drop_count  = -1;
 923         wblock->data.packet.pack_flags  = 0;    /* XXX - is 0 ok to signal "not used"? */
 924
 925         /* Options */
 926         errno = WTAP_ERR_CANT_READ;
 927         to_read = block_total_length
 928         - (int)sizeof(pcapng_block_header_t)
 929         - block_read    /* fixed and variable part, including padding */
 930         - (int)sizeof(bh->block_total_length);
 931         while(to_read > 0) {
 932                 /* read option */
 933                 bytes_read = pcapng_read_option(fh, pn, &oh, option_content, sizeof(option_content), err, err_info);
 934                 if (bytes_read <= 0) {
 935                         pcapng_debug0("pcapng_read_packet_block: failed to read option");
 936                         return bytes_read;
 937                 }
 938                 block_read += bytes_read;
 939                 to_read -= bytes_read;
 940
 941                 /* handle option content */
 942                 switch(oh.option_code) {
 943                     case(0): /* opt_endofopt */
 944                         if(to_read != 0) {
 945                                 pcapng_debug1("pcapng_read_packet_block: %u bytes after opt_endofopt", to_read);
 946                         }
 947                         /* padding should be ok here, just get out of this */
 948                         to_read = 0;
 949                         break;
 950                     case(1): /* opt_comment */
 951                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 952                                 wblock->data.section.opt_comment = g_strndup(option_content, sizeof(option_content));
 953                                 pcapng_debug1("pcapng_read_packet_block: opt_comment %s", wblock->data.section.opt_comment);
 954                         } else {
 955                                 pcapng_debug1("pcapng_read_packet_block: opt_comment length %u seems strange", oh.option_length);
 956                         }
 957                         break;
 958                     case(2): /* pack_flags / epb_flags */
 959                         if(oh.option_length == 4) {
 960                                 /*  Don't cast a char[] into a guint32--the
 961                                  *  char[] may not be aligned correctly.
 962                                  */
 963                                 memcpy(&wblock->data.packet.pack_flags, option_content, sizeof(guint32));
 964                                 if(pn->byte_swapped)
 965                                         wblock->data.packet.pack_flags = BSWAP32(wblock->data.packet.pack_flags);
 966                                 pcapng_debug1("pcapng_read_if_descr_block: pack_flags %u (ignored)", wblock->data.packet.pack_flags);
 967                         } else {
 968                                 pcapng_debug1("pcapng_read_if_descr_block: pack_flags length %u not 4 as expected", oh.option_length);
 969                         }
 970                         break;
 971                     default:
 972                         pcapng_debug2("pcapng_read_packet_block: unknown option %u - ignoring %u bytes",
 973                                       oh.option_code, oh.option_length);
 974                 }
 975         }
 976
 977         pcap_read_post_process(wtap_encap,
 978             (int) (wblock->data.packet.cap_len - pseudo_header_len),
 979             pn->byte_swapped, (guchar *) (wblock->frame_buffer));
 980         return block_read;
 981 }
 982
 983
 984 static int
 985 pcapng_read_simple_packet_block(FILE_T fh, pcapng_block_header_t *bh, pcapng_t *pn, wtapng_block_t *wblock, int *err, gchar **err_info)
 986 {
 987         int bytes_read;
 988         int block_read;
 989         guint64 file_offset64;
 990         gint encap;
 991         int pseudo_header_len;
 992         pcapng_simple_packet_block_t spb;
 993
 994
 995         /* "Simple Packet Block" read fixed part */
 996         errno = WTAP_ERR_CANT_READ;
 997         bytes_read = file_read(&spb, sizeof spb, fh);
 998         if (bytes_read != sizeof spb) {
 999                 pcapng_debug0("pcapng_read_simple_packet_block: failed to read packet data");
1000                 *err = file_error(fh, err_info);
1001                 return 0;
1002         }
1003         block_read = bytes_read;
1004
1005         if (pn->byte_swapped) {
1006                 wblock->data.simple_packet.packet_len   = BSWAP32(spb.packet_len);
1007         } else {
1008                 wblock->data.simple_packet.packet_len   = spb.packet_len;
1009         }
1010
1011         wblock->data.simple_packet.cap_len = bh->block_total_length
1012                                              - (guint32)sizeof(pcapng_simple_packet_block_t)
1013                                              - (guint32)sizeof(bh->block_total_length);
1014
1015         if (wblock->data.simple_packet.cap_len > WTAP_MAX_PACKET_SIZE) {
1016                 *err = WTAP_ERR_BAD_RECORD;
1017                 *err_info = g_strdup_printf("pcapng_read_simple_packet_block: cap_len %u is larger than WTAP_MAX_PACKET_SIZE %u.",
1018                     wblock->data.simple_packet.cap_len, WTAP_MAX_PACKET_SIZE);
1019                 return 0;
1020         }
1021         pcapng_debug1("pcapng_read_simple_packet_block: packet data: packet_len %u",
1022                        wblock->data.simple_packet.packet_len);
1023         if (wblock->data.simple_packet.packet_len > WTAP_MAX_PACKET_SIZE) {
1024                 *err = WTAP_ERR_BAD_RECORD;
1025                 *err_info = g_strdup_printf("pcapng_read_simple_packet_block: packet_len %u is larger than WTAP_MAX_PACKET_SIZE %u.",
1026                     wblock->data.simple_packet.packet_len, WTAP_MAX_PACKET_SIZE);
1027                 return 0;
1028         }
1029
1030         encap = pcapng_get_encap(0, pn);
1031         pcapng_debug1("pcapng_read_simple_packet_block: Need to read pseudo header of size %d",
1032                       pcap_get_phdr_size(encap, wblock->pseudo_header));
1033
1034         memset((void *)wblock->pseudo_header, 0, sizeof(union wtap_pseudo_header));
1035         pseudo_header_len = pcap_process_pseudo_header(fh,
1036                                                        WTAP_FILE_PCAPNG,
1037                                                        encap,
1038                                                        wblock->data.simple_packet.cap_len,
1039                                                        TRUE,
1040                                                        wblock->packet_header,
1041                                                        (union wtap_pseudo_header *)wblock->pseudo_header,
1042                                                        err,
1043                                                        err_info);
1044         if (pseudo_header_len < 0) {
1045                 return 0;
1046         }
1047         wblock->data.simple_packet.pseudo_header_len = (guint32)pseudo_header_len;
1048         block_read += pseudo_header_len;
1049         if (pseudo_header_len != pcap_get_phdr_size(encap, wblock->pseudo_header)) {
1050                 pcapng_debug1("pcapng_read_simple_packet_block: Could only read %d bytes for pseudo header.",
1051                               pseudo_header_len);
1052         }
1053
1054         /* XXX - implement other linktypes then Ethernet */
1055         /* (or even better share the code with libpcap.c) */
1056
1057         /* Ethernet FCS length, might be overwritten by "per packet" options */
1058         memset((void *)wblock->pseudo_header, 0, sizeof(union wtap_pseudo_header));
1059         ((union wtap_pseudo_header *) wblock->pseudo_header)->eth.fcs_len = pn->if_fcslen;
1060
1061         /* "Simple Packet Block" read capture data */
1062         errno = WTAP_ERR_CANT_READ;
1063         bytes_read = file_read((guchar *) (wblock->frame_buffer), wblock->data.simple_packet.cap_len, fh);
1064         if (bytes_read != (int) wblock->data.simple_packet.cap_len) {
1065                 *err = file_error(fh, err_info);
1066                 pcapng_debug1("pcapng_read_simple_packet_block: couldn't read %u bytes of captured data",
1067                               wblock->data.simple_packet.cap_len);
1068                 if (*err == 0)
1069                         *err = WTAP_ERR_SHORT_READ;
1070                 return 0;
1071         }
1072         block_read += bytes_read;
1073
1074         /* jump over potential padding bytes at end of the packet data */
1075         if ((wblock->data.simple_packet.cap_len % 4) != 0) {
1076                 file_offset64 = file_seek(fh, 4 - (wblock->data.simple_packet.cap_len % 4), SEEK_CUR, err);
1077                 if (file_offset64 <= 0) {
1078                         if (*err != 0)
1079                                 return -1;
1080                         return 0;
1081                 }
1082                 block_read += 4 - (wblock->data.simple_packet.cap_len % 4);
1083         }
1084
1085         pcap_read_post_process(encap, (int) wblock->data.simple_packet.cap_len,
1086             pn->byte_swapped, (guchar *) (wblock->frame_buffer));
1087         return block_read;
1088 }
1089
1090 #define NRES_ENDOFRECORD 0
1091 #define NRES_IP4RECORD 1
1092 #define NRES_IP6RECORD 2
1093 #define PADDING4(x) ((((x + 3) >> 2) << 2) - x)
1094 /* IPv6 + MAXNAMELEN */
1095 #define MAX_NRB_REC_SIZE (16 + 64)
1096 static int
1097 pcapng_read_name_resolution_block(FILE_T fh, pcapng_block_header_t *bh, pcapng_t *pn, wtapng_block_t *wblock _U_,int *err, gchar **err_info)
1098 {
1099         int bytes_read = 0;
1100         int block_read = 0;
1101         int to_read;
1102         guint64 file_offset64;
1103         pcapng_name_resolution_block_t nrb;
1104         guchar nrb_rec[MAX_NRB_REC_SIZE];
1105         guint32 v4_addr;
1106
1107         errno = WTAP_ERR_CANT_READ;
1108         to_read = bh->block_total_length
1109                 - sizeof(pcapng_block_header_t)
1110                 - sizeof(bh->block_total_length);
1111
1112         while (block_read < to_read) {
1113                 bytes_read = file_read(&nrb, sizeof nrb, fh);
1114                 if (bytes_read != sizeof nrb) {
1115                         pcapng_debug0("pcapng_read_name_resolution_block: failed to read record header");
1116                         *err = file_error(fh, err_info);
1117                         return 0;
1118                 }
1119                 block_read += bytes_read;
1120
1121                 if (pn->byte_swapped) {
1122                         nrb.record_type = BSWAP16(nrb.record_type);
1123                         nrb.record_len  = BSWAP16(nrb.record_len);
1124                 }
1125
1126                 switch(nrb.record_type) {
1127                         case NRES_ENDOFRECORD:
1128                                 /* There shouldn't be any more data */
1129                                 to_read = 0;
1130                                 break;
1131                         case NRES_IP4RECORD:
1132                                 if (nrb.record_len < 6 || nrb.record_len > MAX_NRB_REC_SIZE || to_read < nrb.record_len) {
1133                                         pcapng_debug0("pcapng_read_name_resolution_block: bad length or insufficient data for IPv4 record");
1134                                         return 0;
1135                                 }
1136                                 bytes_read = file_read(nrb_rec, nrb.record_len, fh);
1137                                 if (bytes_read != nrb.record_len) {
1138                                         pcapng_debug0("pcapng_read_name_resolution_block: failed to read IPv4 record data");
1139                                         *err = file_error(fh, err_info);
1140                                         return 0;
1141                                 }
1142                                 block_read += bytes_read;
1143
1144                                 if (pn->add_new_ipv4) {
1145                                         memcpy(&v4_addr, nrb_rec, 4);
1146                                         if (pn->byte_swapped)
1147                                                 v4_addr = BSWAP32(v4_addr);
1148                                         pn->add_new_ipv4(v4_addr, nrb_rec + 4);
1149                                 }
1150
1151                                 file_offset64 = file_seek(fh, PADDING4(nrb.record_len), SEEK_CUR, err);
1152                                 if (file_offset64 <= 0) {
1153                                         if (*err != 0)
1154                                                 return -1;
1155                                         return 0;
1156                                 }
1157                                 block_read += PADDING4(nrb.record_len);
1158                                 break;
1159                         case NRES_IP6RECORD:
1160                                 if (nrb.record_len < 18 || nrb.record_len > MAX_NRB_REC_SIZE || to_read < nrb.record_len) {
1161                                         pcapng_debug0("pcapng_read_name_resolution_block: bad length or insufficient data for IPv6 record");
1162                                         return 0;
1163                                 }
1164                                 bytes_read = file_read(nrb_rec, nrb.record_len, fh);
1165                                 if (bytes_read != nrb.record_len) {
1166                                         pcapng_debug0("pcapng_read_name_resolution_block: failed to read IPv6 record data");
1167                                         *err = file_error(fh, err_info);
1168                                         return 0;
1169                                 }
1170                                 block_read += bytes_read;
1171
1172                                 if (pn->add_new_ipv6) {
1173                                         pn->add_new_ipv6(nrb_rec, nrb_rec + 16);
1174                                 }
1175
1176                                 file_offset64 = file_seek(fh, PADDING4(nrb.record_len), SEEK_CUR, err);
1177                                 if (file_offset64 <= 0) {
1178                                         if (*err != 0)
1179                                                 return -1;
1180                                         return 0;
1181                                 }
1182                                 block_read += PADDING4(nrb.record_len);
1183                                 break;
1184                         default:
1185                                 pcapng_debug1("pcapng_read_name_resolution_block: unknown record type 0x%x", nrb.record_type);
1186                                 file_offset64 = file_seek(fh, nrb.record_len + PADDING4(nrb.record_len), SEEK_CUR, err);
1187                                 if (file_offset64 <= 0) {
1188                                         if (*err != 0)
1189                                                 return -1;
1190                                         return 0;
1191                                 }
1192                                 block_read += nrb.record_len + PADDING4(nrb.record_len);
1193                                 break;
1194                 }
1195         }
1196
1197         return block_read;
1198 }
1199
1200 static int
1201 pcapng_read_interface_statistics_block(FILE_T fh, pcapng_block_header_t *bh, pcapng_t *pn, wtapng_block_t *wblock,int *err, gchar **err_info)
1202 {
1203         int bytes_read;
1204         int block_read;
1205         int to_read;
1206         pcapng_interface_statistics_block_t isb;
1207         pcapng_option_header_t oh;
1208         char option_content[100]; /* XXX - size might need to be increased, if we see longer options */
1209
1210
1211         /* "Interface Statistics Block" read fixed part */
1212         errno = WTAP_ERR_CANT_READ;
1213         bytes_read = file_read(&isb, sizeof isb, fh);
1214         if (bytes_read != sizeof isb) {
1215                 pcapng_debug0("pcapng_read_interface_statistics_block: failed to read packet data");
1216                 *err = file_error(fh, err_info);
1217                 return 0;
1218         }
1219         block_read = bytes_read;
1220
1221         if(pn->byte_swapped) {
1222                 wblock->data.if_stats.interface_id      = BSWAP64(isb.interface_id);
1223                 wblock->data.if_stats.ts_high           = BSWAP32(isb.timestamp_high);
1224                 wblock->data.if_stats.ts_low            = BSWAP32(isb.timestamp_low);
1225         } else {
1226                 wblock->data.if_stats.interface_id      = isb.interface_id;
1227                 wblock->data.if_stats.ts_high           = isb.timestamp_high;
1228                 wblock->data.if_stats.ts_low            = isb.timestamp_low;
1229         }
1230         pcapng_debug1("pcapng_read_interface_statistics_block: interface_id %" G_GINT64_MODIFIER "u", wblock->data.if_stats.interface_id);
1231
1232         /* Option defaults */
1233         wblock->data.if_stats.opt_comment = NULL;
1234         wblock->data.if_stats.isb_ifrecv  = -1;
1235         wblock->data.if_stats.isb_ifdrop  = -1;
1236
1237         /* Options */
1238         errno = WTAP_ERR_CANT_READ;
1239         to_read = bh->block_total_length
1240         - sizeof(pcapng_block_header_t)
1241         - block_read    /* fixed and variable part, including padding */
1242         - sizeof(bh->block_total_length);
1243         while(to_read > 0) {
1244                 /* read option */
1245                 bytes_read = pcapng_read_option(fh, pn, &oh, option_content, sizeof(option_content), err, err_info);
1246                 if (bytes_read <= 0) {
1247                         pcapng_debug0("pcapng_read_interface_statistics_block: failed to read option");
1248                         return bytes_read;
1249                 }
1250                 block_read += bytes_read;
1251                 to_read -= bytes_read;
1252
1253                 /* handle option content */
1254                 switch(oh.option_code) {
1255                     case(0): /* opt_endofopt */
1256                         if(to_read != 0) {
1257                                 pcapng_debug1("pcapng_read_interface_statistics_block: %u bytes after opt_endofopt", to_read);
1258                         }
1259                         /* padding should be ok here, just get out of this */
1260                         to_read = 0;
1261                         break;
1262                     case(1): /* opt_comment */
1263                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
1264                                 wblock->data.section.opt_comment = g_strndup(option_content, sizeof(option_content));
1265                                 pcapng_debug1("pcapng_read_interface_statistics_block: opt_comment %s", wblock->data.section.opt_comment);
1266                         } else {
1267                                 pcapng_debug1("pcapng_read_interface_statistics_block: opt_comment length %u seems strange", oh.option_length);
1268                         }
1269                         break;
1270                     case(4): /* isb_ifrecv */
1271                         if(oh.option_length == 8) {
1272                                 /*  Don't cast a char[] into a guint32--the
1273                                  *  char[] may not be aligned correctly.
1274                                  */
1275                                 memcpy(&wblock->data.if_stats.isb_ifrecv, option_content, sizeof(guint64));
1276                                 if(pn->byte_swapped)
1277                                         wblock->data.if_stats.isb_ifrecv = BSWAP64(wblock->data.if_stats.isb_ifrecv);
1278                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifrecv %" G_GINT64_MODIFIER "u", wblock->data.if_stats.isb_ifrecv);
1279                         } else {
1280                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifrecv length %u not 8 as expected", oh.option_length);
1281                         }
1282                         break;
1283                     case(5): /* isb_ifdrop */
1284                         if(oh.option_length == 8) {
1285                                 /*  Don't cast a char[] into a guint32--the
1286                                  *  char[] may not be aligned correctly.
1287                                  */
1288                                 memcpy(&wblock->data.if_stats.isb_ifdrop, option_content, sizeof(guint64));
1289                                 if(pn->byte_swapped)
1290                                         wblock->data.if_stats.isb_ifdrop = BSWAP64(wblock->data.if_stats.isb_ifdrop);
1291                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifdrop %" G_GINT64_MODIFIER "u", wblock->data.if_stats.isb_ifdrop);
1292                         } else {
1293                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifdrop length %u not 8 as expected", oh.option_length);
1294                         }
1295                         break;
1296                     default:
1297                         pcapng_debug2("pcapng_read_interface_statistics_block: unknown option %u - ignoring %u bytes",
1298                                       oh.option_code, oh.option_length);
1299                 }
1300         }
1301
1302     return block_read;
1303 }
1304
1305
1306 static int
1307 pcapng_read_unknown_block(FILE_T fh, pcapng_block_header_t *bh, pcapng_t *pn _U_, wtapng_block_t *wblock _U_,int *err, gchar **err_info _U_)
1308 {
1309         int block_read;
1310         guint64 file_offset64;
1311         guint32 block_total_length;
1312
1313
1314         /* add padding bytes to "block total length" */
1315         /* (the "block total length" of some example files don't contain any padding bytes!) */
1316         if (bh->block_total_length % 4) {
1317                 block_total_length = bh->block_total_length + 4 - (bh->block_total_length % 4);
1318         } else {
1319                 block_total_length = bh->block_total_length;
1320         }
1321
1322         block_read = block_total_length - (guint32)sizeof(pcapng_block_header_t) - (guint32)sizeof(bh->block_total_length);
1323
1324         /* jump over this unknown block */
1325         file_offset64 = file_seek(fh, block_read, SEEK_CUR, err);
1326         if (file_offset64 <= 0) {
1327                 if (*err != 0)
1328                         return -1;
1329                 return 0;
1330         }
1331
1332         return block_read;
1333 }
1334
1335
1336 static int
1337 pcapng_read_block(FILE_T fh, gboolean first_block, pcapng_t *pn, wtapng_block_t *wblock, int *err, gchar **err_info)
1338 {
1339         int block_read;
1340         int bytes_read;
1341         pcapng_block_header_t bh;
1342         guint32 block_total_length;
1343
1344
1345         /* Try to read the (next) block header */
1346         errno = WTAP_ERR_CANT_READ;
1347         bytes_read = file_read(&bh, sizeof bh, fh);
1348         if (bytes_read != sizeof bh) {
1349                 *err = file_error(fh, err_info);
1350                 pcapng_debug3("pcapng_read_block: file_read() returned %d instead of %u, err = %d.", bytes_read, (unsigned int)sizeof bh, *err);
1351                 if (*err != 0)
1352                         return -1;
1353                 return 0;
1354         }
1355
1356         block_read = bytes_read;
1357         if (pn->byte_swapped) {
1358                 bh.block_type         = BSWAP32(bh.block_type);
1359                 bh.block_total_length = BSWAP32(bh.block_total_length);
1360         }
1361
1362         wblock->type = bh.block_type;
1363
1364         pcapng_debug1("pcapng_read_block: block_type 0x%x", bh.block_type);
1365
1366         if (first_block) {
1367                 /*
1368                  * This is being read in by pcapng_open(), so this block
1369                  * must be an SHB.  If it's not, this is not a pcap-ng
1370                  * file.
1371                  *
1372                  * XXX - check for various forms of Windows <-> UN*X
1373                  * mangling, and suggest that the file might be a
1374                  * pcap-ng file that was damaged in transit?
1375                  */
1376                 if (bh.block_type != BLOCK_TYPE_SHB)
1377                         return 0;       /* not a pcap-ng file */
1378         }
1379
1380         switch(bh.block_type) {
1381                 case(BLOCK_TYPE_SHB):
1382                         bytes_read = pcapng_read_section_header_block(fh, first_block, &bh, pn, wblock, err, err_info);
1383                         break;
1384                 case(BLOCK_TYPE_IDB):
1385                         bytes_read = pcapng_read_if_descr_block(fh, &bh, pn, wblock, err, err_info);
1386                         break;
1387                 case(BLOCK_TYPE_PB):
1388                         bytes_read = pcapng_read_packet_block(fh, &bh, pn, wblock, err, err_info, FALSE);
1389                         break;
1390                 case(BLOCK_TYPE_SPB):
1391                         bytes_read = pcapng_read_simple_packet_block(fh, &bh, pn, wblock, err, err_info);
1392                         break;
1393                 case(BLOCK_TYPE_EPB):
1394                         bytes_read = pcapng_read_packet_block(fh, &bh, pn, wblock, err, err_info, TRUE);
1395                         break;
1396                 case(BLOCK_TYPE_NRB):
1397                         bytes_read = pcapng_read_name_resolution_block(fh, &bh, pn, wblock, err, err_info);
1398                         break;
1399                 case(BLOCK_TYPE_ISB):
1400                         bytes_read = pcapng_read_interface_statistics_block(fh, &bh, pn, wblock, err, err_info);
1401                         break;
1402                 default:
1403                         pcapng_debug2("pcapng_read_block: Unknown block_type: 0x%x (block ignored), block total length %d", bh.block_type, bh.block_total_length);
1404                         bytes_read = pcapng_read_unknown_block(fh, &bh, pn, wblock, err, err_info);
1405         }
1406
1407         if (bytes_read <= 0) {
1408                 return bytes_read;
1409         }
1410         block_read += bytes_read;
1411
1412         /* sanity check: first and second block lengths must match */
1413         errno = WTAP_ERR_CANT_READ;
1414         bytes_read = file_read(&block_total_length, sizeof block_total_length, fh);
1415         if (bytes_read != sizeof block_total_length) {
1416                 pcapng_debug0("pcapng_read_block: couldn't read second block length");
1417                 *err = file_error(fh, err_info);
1418                 if (*err == 0)
1419                         *err = WTAP_ERR_SHORT_READ;
1420                 return -1;
1421         }
1422         block_read += bytes_read;
1423
1424         if (pn->byte_swapped)
1425                 block_total_length = BSWAP32(block_total_length);
1426
1427         if (!(block_total_length == bh.block_total_length)) {
1428                 *err = WTAP_ERR_BAD_RECORD;
1429                 *err_info = g_strdup_printf("pcapng_read_block: total block lengths (first %u and second %u) don't match",
1430                               bh.block_total_length, block_total_length);
1431                 return -1;
1432         }
1433
1434         return block_read;
1435 }
1436
1437
1438 /* classic wtap: open capture file */
1439 int
1440 pcapng_open(wtap *wth, int *err, gchar **err_info)
1441 {
1442         int bytes_read;
1443         pcapng_t pn;
1444         wtapng_block_t wblock;
1445         pcapng_t *pcapng;
1446
1447         /* we don't know the byte swapping of the file yet */
1448         pn.byte_swapped = FALSE;
1449         pn.if_fcslen = -1;
1450         pn.version_major = -1;
1451         pn.version_minor = -1;
1452         pn.interface_data = NULL;
1453         pn.number_of_interfaces = 0;
1454
1455         /* we don't expect any packet blocks yet */
1456         wblock.frame_buffer = NULL;
1457         wblock.pseudo_header = NULL;
1458         wblock.packet_header = NULL;
1459         wblock.file_encap = &wth->file_encap;
1460
1461         pcapng_debug0("pcapng_open: opening file");
1462         /* read first block */
1463         bytes_read = pcapng_read_block(wth->fh, TRUE, &pn, &wblock, err, err_info);
1464         if (bytes_read <= 0) {
1465                 pcapng_debug0("pcapng_open: couldn't read first SHB");
1466                 *err = file_error(wth->fh, err_info);
1467                 if (*err != 0)
1468                         return -1;
1469                 return 0;
1470         }
1471         wth->data_offset += bytes_read;
1472
1473         /* first block must be a "Section Header Block" */
1474         if (wblock.type != BLOCK_TYPE_SHB) {
1475                 /*
1476                  * XXX - check for damage from transferring a file
1477                  * between Windows and UN*X as text rather than
1478                  * binary data?
1479                  */
1480                 pcapng_debug1("pcapng_open: first block type %u not SHB", wblock.type);
1481                 return 0;
1482         }
1483
1484         wth->file_encap = WTAP_ENCAP_UNKNOWN;
1485         wth->snapshot_length = 0;
1486         wth->tsprecision = WTAP_FILE_TSPREC_NSEC;
1487         pcapng = (pcapng_t *)g_malloc(sizeof(pcapng_t));
1488         wth->priv = (void *)pcapng;
1489         *pcapng = pn;
1490         wth->subtype_read = pcapng_read;
1491         wth->subtype_seek_read = pcapng_seek_read;
1492         wth->subtype_close = pcapng_close;
1493         wth->file_type = WTAP_FILE_PCAPNG;
1494
1495         return 1;
1496 }
1497
1498
1499 /* classic wtap: read packet */
1500 static gboolean
1501 pcapng_read(wtap *wth, int *err, gchar **err_info, gint64 *data_offset)
1502 {
1503         pcapng_t *pcapng = (pcapng_t *)wth->priv;
1504         int bytes_read;
1505         guint64 ts;
1506         wtapng_block_t wblock;
1507
1508         pcapng_debug1("pcapng_read: wth->data_offset is initially %" G_GINT64_MODIFIER "u", wth->data_offset);
1509         *data_offset = wth->data_offset;
1510         pcapng_debug1("pcapng_read: *data_offset is initially set to %" G_GINT64_MODIFIER "u", *data_offset);
1511
1512         /* XXX - This should be done in the packet block reading function and
1513          * should make use of the caplen of the packet.
1514          */
1515         if (wth->snapshot_length > 0) {
1516                 buffer_assure_space(wth->frame_buffer, wth->snapshot_length);
1517         } else {
1518                 buffer_assure_space(wth->frame_buffer, WTAP_MAX_PACKET_SIZE);
1519         }
1520
1521         wblock.frame_buffer  = buffer_start_ptr(wth->frame_buffer);
1522         wblock.pseudo_header = &wth->pseudo_header;
1523         wblock.packet_header = &wth->phdr;
1524         wblock.file_encap    = &wth->file_encap;
1525
1526         pcapng->add_new_ipv4 = wth->add_new_ipv4;
1527         pcapng->add_new_ipv6 = wth->add_new_ipv6;
1528
1529         /* read next block */
1530         while (1) {
1531                 bytes_read = pcapng_read_block(wth->fh, FALSE, pcapng, &wblock, err, err_info);
1532                 if (bytes_read <= 0) {
1533                         wth->data_offset = *data_offset;
1534                         pcapng_debug1("pcapng_read: wth->data_offset is finally %" G_GINT64_MODIFIER "u", wth->data_offset);
1535                         pcapng_debug0("pcapng_read: couldn't read packet block");
1536                         return FALSE;
1537                 }
1538
1539                 /* block must be a "Packet Block" or an "Enhanced Packet Block" -> otherwise continue */
1540                 if (wblock.type == BLOCK_TYPE_PB || wblock.type == BLOCK_TYPE_EPB) {
1541                         break;
1542                 }
1543
1544                 /* XXX - improve handling of "unknown" blocks */
1545                 pcapng_debug1("pcapng_read: block type 0x%x not PB/EPB", wblock.type);
1546                 *data_offset += bytes_read;
1547                 pcapng_debug1("pcapng_read: *data_offset is updated to %" G_GINT64_MODIFIER "u", *data_offset);
1548         }
1549
1550         /* Combine the two 32-bit pieces of the timestamp into one 64-bit value */
1551         ts = (((guint64)wblock.data.packet.ts_high) << 32) | ((guint64)wblock.data.packet.ts_low);
1552
1553         wth->phdr.caplen = wblock.data.packet.cap_len - wblock.data.packet.pseudo_header_len;
1554         wth->phdr.len = wblock.data.packet.packet_len - wblock.data.packet.pseudo_header_len;
1555         if (wblock.data.packet.interface_id < pcapng->number_of_interfaces) {
1556                 interface_data_t int_data;
1557                 guint64 time_units_per_second;
1558                 gint id;
1559
1560                 id = (gint)wblock.data.packet.interface_id;
1561                 int_data = g_array_index(pcapng->interface_data, interface_data_t, id);
1562                 time_units_per_second = int_data.time_units_per_second;
1563                 wth->phdr.pkt_encap = int_data.wtap_encap;
1564                 wth->phdr.ts.secs = (time_t)(ts / time_units_per_second);
1565                 wth->phdr.ts.nsecs = (int)(((ts % time_units_per_second) * 1000000000) / time_units_per_second);
1566         } else {
1567                 wth->phdr.pkt_encap = WTAP_ENCAP_UNKNOWN;
1568                 *err = WTAP_ERR_BAD_RECORD;
1569                 *err_info = g_strdup_printf("pcapng: interface index %u is not less than interface count %u.",
1570                     wblock.data.packet.interface_id, pcapng->number_of_interfaces);
1571                 wth->data_offset = *data_offset + bytes_read;
1572                 pcapng_debug1("pcapng_read: wth->data_offset is finally %" G_GINT64_MODIFIER "u", wth->data_offset);
1573                 return FALSE;
1574         }
1575
1576         /*pcapng_debug2("Read length: %u Packet length: %u", bytes_read, wth->phdr.caplen);*/
1577         wth->data_offset = *data_offset + bytes_read;
1578         pcapng_debug1("pcapng_read: wth->data_offset is finally %" G_GINT64_MODIFIER "u", wth->data_offset);
1579
1580         return TRUE;
1581 }
1582
1583
1584 /* classic wtap: seek to file position and read packet */
1585 static gboolean
1586 pcapng_seek_read(wtap *wth, gint64 seek_off,
1587     union wtap_pseudo_header *pseudo_header, guchar *pd, int length _U_,
1588     int *err, gchar **err_info)
1589 {
1590         pcapng_t *pcapng = (pcapng_t *)wth->priv;
1591         guint64 bytes_read64;
1592         int bytes_read;
1593         wtapng_block_t wblock;
1594
1595
1596         /* seek to the right file position */
1597         bytes_read64 = file_seek(wth->random_fh, seek_off, SEEK_SET, err);
1598         if (bytes_read64 <= 0) {
1599                 return FALSE;   /* Seek error */
1600         }
1601         pcapng_debug1("pcapng_seek_read: reading at offset %" G_GINT64_MODIFIER "u", seek_off);
1602
1603         wblock.frame_buffer = pd;
1604         wblock.pseudo_header = pseudo_header;
1605         wblock.packet_header = &wth->phdr;
1606         wblock.file_encap = &wth->file_encap;
1607
1608         /* read the block */
1609         bytes_read = pcapng_read_block(wth->random_fh, FALSE, pcapng, &wblock, err, err_info);
1610         if (bytes_read <= 0) {
1611                 *err = file_error(wth->random_fh, err_info);
1612                 pcapng_debug3("pcapng_seek_read: couldn't read packet block (err=%d, errno=%d, bytes_read=%d).",
1613                               *err, errno, bytes_read);
1614                 return FALSE;
1615         }
1616
1617         /* block must be a "Packet Block" or an "Enhanced Packet Block" */
1618         if (wblock.type != BLOCK_TYPE_PB && wblock.type != BLOCK_TYPE_EPB) {
1619                 pcapng_debug1("pcapng_seek_read: block type %u not PB/EPB", wblock.type);
1620                 return FALSE;
1621         }
1622
1623         return TRUE;
1624 }
1625
1626
1627 /* classic wtap: close capture file */
1628 static void
1629 pcapng_close(wtap *wth)
1630 {
1631         pcapng_t *pcapng = (pcapng_t *)wth->priv;
1632
1633         pcapng_debug0("pcapng_close: closing file");
1634         if (pcapng->interface_data != NULL) {
1635                 g_array_free(pcapng->interface_data, TRUE);
1636         }
1637 }
1638
1639
1640
1641 typedef struct {
1642         GArray *interface_data;
1643         guint number_of_interfaces;
1644         struct addrinfo *addrinfo_list_last;
1645 } pcapng_dump_t;
1646
1647 static gboolean
1648 pcapng_write_section_header_block(wtap_dumper *wdh, wtapng_block_t *wblock, int *err)
1649 {
1650         pcapng_block_header_t bh;
1651         pcapng_section_header_block_t shb;
1652
1653
1654         /* write block header */
1655         bh.block_type = wblock->type;
1656         bh.block_total_length = sizeof(bh) + sizeof(shb) /* + options */ + 4;
1657
1658         if (!wtap_dump_file_write(wdh, &bh, sizeof bh, err))
1659                 return FALSE;
1660         wdh->bytes_dumped += sizeof bh;
1661
1662         /* write block fixed content */
1663         /* XXX - get these values from wblock? */
1664         shb.magic = 0x1A2B3C4D;
1665         shb.version_major = 1;
1666         shb.version_minor = 0;
1667         shb.section_length = -1;
1668
1669         if (!wtap_dump_file_write(wdh, &shb, sizeof shb, err))
1670                 return FALSE;
1671         wdh->bytes_dumped += sizeof shb;
1672
1673         /* XXX - write (optional) block options */
1674
1675         /* write block footer */
1676         if (!wtap_dump_file_write(wdh, &bh.block_total_length,
1677             sizeof bh.block_total_length, err))
1678                 return FALSE;
1679         wdh->bytes_dumped += sizeof bh.block_total_length;
1680
1681         return TRUE;
1682 }
1683
1684
1685
1686 static gboolean
1687 pcapng_write_if_descr_block(wtap_dumper *wdh, wtapng_block_t *wblock, int *err)
1688 {
1689         pcapng_block_header_t bh;
1690         pcapng_interface_description_block_t idb;
1691
1692
1693         pcapng_debug3("pcapng_write_if_descr_block: encap = %d (%s), snaplen = %d",
1694                       wblock->data.if_descr.link_type,
1695                       wtap_encap_string(wtap_pcap_encap_to_wtap_encap(wblock->data.if_descr.link_type)),
1696                       wblock->data.if_descr.snap_len);
1697
1698         if (wblock->data.if_descr.link_type == (guint16)-1) {
1699                 *err = WTAP_ERR_UNSUPPORTED_ENCAP;
1700                 return FALSE;
1701         }
1702
1703         /* write block header */
1704         bh.block_type = wblock->type;
1705         bh.block_total_length = sizeof(bh) + sizeof(idb) /* + options */ + 4;
1706
1707         if (!wtap_dump_file_write(wdh, &bh, sizeof bh, err))
1708                 return FALSE;
1709         wdh->bytes_dumped += sizeof bh;
1710
1711         /* write block fixed content */
1712         idb.linktype    = wblock->data.if_descr.link_type;
1713         idb.reserved    = 0;
1714         idb.snaplen     = wblock->data.if_descr.snap_len;
1715
1716         if (!wtap_dump_file_write(wdh, &idb, sizeof idb, err))
1717                 return FALSE;
1718         wdh->bytes_dumped += sizeof idb;
1719
1720         /* XXX - write (optional) block options */
1721
1722         /* write block footer */
1723         if (!wtap_dump_file_write(wdh, &bh.block_total_length,
1724             sizeof bh.block_total_length, err))
1725                 return FALSE;
1726         wdh->bytes_dumped += sizeof bh.block_total_length;
1727
1728         return TRUE;
1729 }
1730
1731
1732 static gboolean
1733 pcapng_write_packet_block(wtap_dumper *wdh, wtapng_block_t *wblock, int *err)
1734 {
1735         pcapng_block_header_t bh;
1736         pcapng_enhanced_packet_block_t epb;
1737         const guint32 zero_pad = 0;
1738         guint32 pad_len;
1739         guint32 phdr_len;
1740
1741         phdr_len = (guint32)pcap_get_phdr_size(wblock->data.packet.wtap_encap, wblock->pseudo_header);
1742         if ((phdr_len + wblock->data.packet.cap_len) % 4) {
1743                 pad_len = 4 - ((phdr_len + wblock->data.packet.cap_len) % 4);
1744         } else {
1745                 pad_len = 0;
1746         }
1747
1748         /* write (enhanced) packet block header */
1749         bh.block_type = wblock->type;
1750         bh.block_total_length = (guint32)sizeof(bh) + (guint32)sizeof(epb) + phdr_len + wblock->data.packet.cap_len + pad_len /* + options */ + 4;
1751
1752         if (!wtap_dump_file_write(wdh, &bh, sizeof bh, err))
1753                 return FALSE;
1754         wdh->bytes_dumped += sizeof bh;
1755
1756         /* write block fixed content */
1757         epb.interface_id        = wblock->data.packet.interface_id;
1758         epb.timestamp_high      = wblock->data.packet.ts_high;
1759         epb.timestamp_low       = wblock->data.packet.ts_low;
1760         epb.captured_len        = wblock->data.packet.cap_len + phdr_len;
1761         epb.packet_len          = wblock->data.packet.packet_len + phdr_len;
1762
1763         if (!wtap_dump_file_write(wdh, &epb, sizeof epb, err))
1764                 return FALSE;
1765         wdh->bytes_dumped += sizeof epb;
1766
1767         /* write pseudo header */
1768         if (!pcap_write_phdr(wdh, wblock->data.packet.wtap_encap, wblock->pseudo_header, err)) {
1769                 return FALSE;
1770         }
1771         wdh->bytes_dumped += phdr_len;
1772
1773         /* write packet data */
1774         if (!wtap_dump_file_write(wdh, wblock->frame_buffer,
1775             wblock->data.packet.cap_len, err))
1776                 return FALSE;
1777         wdh->bytes_dumped += wblock->data.packet.cap_len;
1778
1779         /* write padding (if any) */
1780         if (pad_len != 0) {
1781                 if (!wtap_dump_file_write(wdh, &zero_pad, pad_len, err))
1782                         return FALSE;
1783                 wdh->bytes_dumped += pad_len;
1784         }
1785
1786         /* XXX - write (optional) block options */
1787
1788         /* write block footer */
1789         if (!wtap_dump_file_write(wdh, &bh.block_total_length,
1790             sizeof bh.block_total_length, err))
1791                 return FALSE;
1792         wdh->bytes_dumped += sizeof bh.block_total_length;
1793
1794         return TRUE;
1795 }
1796
1797 /* Arbitrary. */
1798 #define NRES_REC_MAX_SIZE ((WTAP_MAX_PACKET_SIZE * 4) + 16)
1799 static gboolean
1800 pcapng_write_name_resolution_block(wtap_dumper *wdh, pcapng_dump_t *pcapng, int *err)
1801 {
1802         pcapng_block_header_t bh;
1803         pcapng_name_resolution_block_t nrb;
1804         struct addrinfo *ai;
1805         struct sockaddr_in *sa4;
1806         struct sockaddr_in6 *sa6;
1807         guchar *rec_data;
1808         gint rec_off, namelen, tot_rec_len;
1809
1810         if (! pcapng->addrinfo_list_last || ! pcapng->addrinfo_list_last->ai_next) {
1811                 return TRUE;
1812         }
1813
1814         rec_off = 8; /* block type + block total length */
1815         bh.block_type = BLOCK_TYPE_NRB;
1816         bh.block_total_length = rec_off + 8; /* end-of-record + block total length */
1817         rec_data = g_malloc(NRES_REC_MAX_SIZE);
1818
1819         for (; pcapng->addrinfo_list_last && pcapng->addrinfo_list_last->ai_next; pcapng->addrinfo_list_last = pcapng->addrinfo_list_last->ai_next ) {
1820                 ai = pcapng->addrinfo_list_last->ai_next; /* Skips over the first (dummy) entry */
1821                 namelen = (gint)strlen(ai->ai_canonname) + 1;
1822                 if (ai->ai_family == AF_INET) {
1823                         nrb.record_type = NRES_IP4RECORD;
1824                         nrb.record_len = 4 + namelen;
1825                         tot_rec_len = 4 + nrb.record_len + PADDING4(nrb.record_len);
1826                         bh.block_total_length += tot_rec_len;
1827
1828                         if (rec_off + tot_rec_len > NRES_REC_MAX_SIZE)
1829                                 break;
1830
1831                         /*
1832                          * The joys of BSD sockaddrs.  In practice, this
1833                          * cast is alignment-safe.
1834                          */
1835                         sa4 = (struct sockaddr_in *)(void *)ai->ai_addr;
1836                         memcpy(rec_data + rec_off, &nrb, sizeof(nrb));
1837                         rec_off += 4;
1838
1839                         memcpy(rec_data + rec_off, &(sa4->sin_addr.s_addr), 4);
1840                         rec_off += 4;
1841
1842                         memcpy(rec_data + rec_off, ai->ai_canonname, namelen);
1843                         rec_off += namelen;
1844
1845                         memset(rec_data + rec_off, 0, PADDING4(namelen));
1846                         rec_off += PADDING4(namelen);
1847                         pcapng_debug1("NRB: added IPv4 record for %s", ai->ai_canonname);
1848                 } else if (ai->ai_family == AF_INET6) {
1849                         nrb.record_type = NRES_IP6RECORD;
1850                         nrb.record_len = 16 + namelen;
1851                         tot_rec_len = 4 + nrb.record_len + PADDING4(nrb.record_len);
1852                         bh.block_total_length += tot_rec_len;
1853
1854                         if (rec_off + tot_rec_len > NRES_REC_MAX_SIZE)
1855                                 break;
1856
1857                         /*
1858                          * The joys of BSD sockaddrs.  In practice, this
1859                          * cast is alignment-safe.
1860                          */
1861                         sa6 = (struct sockaddr_in6 *)(void *)ai->ai_addr;
1862                         memcpy(rec_data + rec_off, &nrb, sizeof(nrb));
1863                         rec_off += 4;
1864
1865                         memcpy(rec_data + rec_off, sa6->sin6_addr.s6_addr, 16);
1866                         rec_off += 16;
1867
1868                         memcpy(rec_data + rec_off, ai->ai_canonname, namelen);
1869                         rec_off += namelen;
1870
1871                         memset(rec_data + rec_off, 0, PADDING4(namelen));
1872                         rec_off += PADDING4(namelen);
1873                         pcapng_debug1("NRB: added IPv6 record for %s", ai->ai_canonname);
1874                 }
1875         }
1876
1877         /* We know the total length now; copy the block header. */
1878         memcpy(rec_data, &bh, sizeof(bh));
1879
1880         /* End of record */
1881         memset(rec_data + rec_off, 0, 4);
1882         rec_off += 4;
1883
1884         memcpy(rec_data + rec_off, &bh.block_total_length, sizeof(bh.block_total_length));
1885
1886         if (!wtap_dump_file_write(wdh, rec_data, bh.block_total_length, err)) {
1887                 g_free(rec_data);
1888                 return FALSE;
1889         }
1890
1891         g_free(rec_data);
1892         wdh->bytes_dumped += bh.block_total_length;
1893         return TRUE;
1894 }
1895
1896
1897 static gboolean
1898 pcapng_write_block(wtap_dumper *wdh, /*pcapng_t *pn, */wtapng_block_t *wblock, int *err)
1899 {
1900         switch(wblock->type) {
1901             case(BLOCK_TYPE_SHB):
1902                 return pcapng_write_section_header_block(wdh, wblock, err);
1903             case(BLOCK_TYPE_IDB):
1904                 return pcapng_write_if_descr_block(wdh, wblock, err);
1905             case(BLOCK_TYPE_PB):
1906                 /* Packet Block is obsolete */
1907                 return FALSE;
1908             case(BLOCK_TYPE_EPB):
1909                 return pcapng_write_packet_block(wdh, wblock, err);
1910             default:
1911                 pcapng_debug1("Unknown block_type: 0x%x", wblock->type);
1912                 return FALSE;
1913         }
1914 }
1915
1916
1917 static guint32
1918 pcapng_lookup_interface_id_by_encap(int wtap_encap, wtap_dumper *wdh)
1919 {
1920         gint i;
1921         interface_data_t int_data;
1922         pcapng_dump_t *pcapng = (pcapng_dump_t *)wdh->priv;
1923
1924         for(i = 0; i < (gint)pcapng->number_of_interfaces; i++) {
1925                 int_data = g_array_index(pcapng->interface_data, interface_data_t, i);
1926                 if (wtap_encap == int_data.wtap_encap) {
1927                         return (guint32)i;
1928                 }
1929         }
1930         return G_MAXUINT32;
1931 }
1932
1933
1934 static gboolean pcapng_dump(wtap_dumper *wdh,
1935         const struct wtap_pkthdr *phdr,
1936         const union wtap_pseudo_header *pseudo_header,
1937         const guchar *pd, int *err)
1938 {
1939         wtapng_block_t wblock;
1940         interface_data_t int_data;
1941         guint32 interface_id;
1942         guint64 ts;
1943         pcapng_dump_t *pcapng = (pcapng_dump_t *)wdh->priv;
1944
1945         pcapng_debug2("pcapng_dump: encap = %d (%s)",
1946                       phdr->pkt_encap,
1947                       wtap_encap_string(phdr->pkt_encap));
1948
1949         if (!pcapng->addrinfo_list_last)
1950                 pcapng->addrinfo_list_last = wdh->addrinfo_list;
1951
1952         interface_id = pcapng_lookup_interface_id_by_encap(phdr->pkt_encap, wdh);
1953         if (interface_id == G_MAXUINT32) {
1954                 /* write the interface description block */
1955                 wblock.frame_buffer            = NULL;
1956                 wblock.pseudo_header           = NULL;
1957                 wblock.packet_header           = NULL;
1958                 wblock.file_encap              = NULL;
1959                 wblock.type                    = BLOCK_TYPE_IDB;
1960                 wblock.data.if_descr.link_type = wtap_wtap_encap_to_pcap_encap(phdr->pkt_encap);
1961                 wblock.data.if_descr.snap_len  = wdh->snaplen; /* XXX */
1962
1963                 /* XXX - options unused */
1964                 wblock.data.if_descr.if_speed   = -1;
1965                 wblock.data.if_descr.if_tsresol = 6;    /* default: usec */
1966                 wblock.data.if_descr.if_os      = NULL;
1967                 wblock.data.if_descr.if_fcslen  = -1;
1968
1969                 if (!pcapng_write_block(wdh, &wblock, err)) {
1970                         return FALSE;
1971                 }
1972
1973                 interface_id = pcapng->number_of_interfaces;
1974                 int_data.wtap_encap = phdr->pkt_encap;
1975                 int_data.time_units_per_second = 0;
1976                 g_array_append_val(pcapng->interface_data, int_data);
1977                 pcapng->number_of_interfaces++;
1978
1979                 pcapng_debug3("pcapng_dump: added interface description block with index %u for encap = %d (%s).",
1980                               interface_id,
1981                               phdr->pkt_encap,
1982                               wtap_encap_string(phdr->pkt_encap));
1983         }
1984
1985         /* Flush any hostname resolution info we may have */
1986         while (pcapng->addrinfo_list_last && pcapng->addrinfo_list_last->ai_next) {
1987                 pcapng_write_name_resolution_block(wdh, pcapng, err);
1988         }
1989
1990         wblock.frame_buffer  = pd;
1991         wblock.pseudo_header = pseudo_header;
1992         wblock.packet_header = NULL;
1993         wblock.file_encap    = NULL;
1994
1995         /* write the (enhanced) packet block */
1996         wblock.type = BLOCK_TYPE_EPB;
1997
1998         /* default is to write out in microsecond resolution */
1999         ts = (((guint64)phdr->ts.secs) * 1000000) + (phdr->ts.nsecs / 1000);
2000
2001         /* Split the 64-bit timestamp into two 32-bit pieces */
2002         wblock.data.packet.ts_high      = (guint32)(ts >> 32);
2003         wblock.data.packet.ts_low       = (guint32)ts;
2004
2005         wblock.data.packet.cap_len      = phdr->caplen;
2006         wblock.data.packet.packet_len   = phdr->len;
2007         wblock.data.packet.interface_id = interface_id;
2008         wblock.data.packet.wtap_encap   = phdr->pkt_encap;
2009
2010         /* currently unused */
2011         wblock.data.packet.drop_count   = -1;
2012         wblock.data.packet.opt_comment  = NULL;
2013
2014         if (!pcapng_write_block(wdh, &wblock, err)) {
2015                 return FALSE;
2016         }
2017
2018         return TRUE;
2019 }
2020
2021
2022 /* Finish writing to a dump file.
2023    Returns TRUE on success, FALSE on failure. */
2024 static gboolean pcapng_dump_close(wtap_dumper *wdh, int *err _U_)
2025 {
2026         pcapng_dump_t *pcapng = (pcapng_dump_t *)wdh->priv;
2027
2028         pcapng_debug0("pcapng_dump_close");
2029         g_array_free(pcapng->interface_data, TRUE);
2030         pcapng->number_of_interfaces = 0;
2031         return TRUE;
2032 }
2033
2034
2035 /* Returns TRUE on success, FALSE on failure; sets "*err" to an error code on
2036    failure */
2037 gboolean
2038 pcapng_dump_open(wtap_dumper *wdh, int *err)
2039 {
2040         wtapng_block_t wblock;
2041         pcapng_dump_t *pcapng;
2042
2043         wblock.frame_buffer  = NULL;
2044         wblock.pseudo_header = NULL;
2045         wblock.packet_header = NULL;
2046         wblock.file_encap    = NULL;
2047
2048         pcapng_debug0("pcapng_dump_open");
2049         /* This is a pcapng file */
2050         wdh->subtype_write = pcapng_dump;
2051         wdh->subtype_close = pcapng_dump_close;
2052         pcapng = (pcapng_dump_t *)g_malloc0(sizeof(pcapng_dump_t));
2053         wdh->priv = (void *)pcapng;
2054         pcapng->interface_data = g_array_new(FALSE, FALSE, sizeof(interface_data_t));
2055
2056         /* write the section header block */
2057         wblock.type = BLOCK_TYPE_SHB;
2058         wblock.data.section.section_length = -1;
2059
2060         /* XXX - options unused */
2061         wblock.data.section.opt_comment   = NULL;
2062         wblock.data.section.shb_hardware  = NULL;
2063         wblock.data.section.shb_os        = NULL;
2064         wblock.data.section.shb_user_appl = NULL;
2065
2066         if (!pcapng_write_block(wdh, &wblock, err)) {
2067                 return FALSE;
2068         }
2069         pcapng_debug0("pcapng_dump_open: wrote section header block.");
2070
2071         return TRUE;
2072 }
2073
2074
2075 /* Returns 0 if we could write the specified encapsulation type,
2076    an error indication otherwise. */
2077 int pcapng_dump_can_write_encap(int wtap_encap)
2078 {
2079         pcapng_debug2("pcapng_dump_can_write_encap: encap = %d (%s)",
2080                       wtap_encap,
2081                       wtap_encap_string(wtap_encap));
2082
2083         /* Per-packet encapsulations is supported. */
2084         if (wtap_encap == WTAP_ENCAP_PER_PACKET)
2085                 return 0;
2086
2087         /* Make sure we can figure out this DLT type */
2088         if (wtap_wtap_encap_to_pcap_encap(wtap_encap) == -1)
2089                 return WTAP_ERR_UNSUPPORTED_ENCAP;
2090
2091         return 0;
2092 }