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, guint8 *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 guint8 *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_option: 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_FILE;
 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_FILE;
 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.if_descr.opt_comment = g_strndup(option_content, sizeof(option_content));
 666                                 pcapng_debug1("pcapng_read_if_descr_block: opt_comment %s", wblock->data.if_descr.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         int fcslen;
 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_FILE;
 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_FILE;
 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
 856         wtap_encap = pcapng_get_encap(wblock->data.packet.interface_id, pn);
 857         pcapng_debug3("pcapng_read_packet_block: encapsulation = %d (%s), pseudo header size = %d.",
 858                        wtap_encap,
 859                        wtap_encap_string(wtap_encap),
 860                        pcap_get_phdr_size(wtap_encap, wblock->pseudo_header));
 861
 862         memset((void *)wblock->pseudo_header, 0, sizeof(union wtap_pseudo_header));
 863         pseudo_header_len = pcap_process_pseudo_header(fh,
 864                                                        WTAP_FILE_PCAPNG,
 865                                                        wtap_encap,
 866                                                        wblock->data.packet.cap_len,
 867                                                        TRUE,
 868                                                        wblock->packet_header,
 869                                                        (union wtap_pseudo_header *)wblock->pseudo_header,
 870                                                        err,
 871                                                        err_info);
 872         if (pseudo_header_len < 0) {
 873                 return 0;
 874         }
 875         wblock->data.packet.pseudo_header_len = (guint32)pseudo_header_len;
 876         block_read += pseudo_header_len;
 877         if (pseudo_header_len != pcap_get_phdr_size(wtap_encap, wblock->pseudo_header)) {
 878                 pcapng_debug1("pcapng_read_packet_block: Could only read %d bytes for pseudo header.",
 879                               pseudo_header_len);
 880         }
 881
 882         /* "(Enhanced) Packet Block" read capture data */
 883         errno = WTAP_ERR_CANT_READ;
 884         bytes_read = file_read((guint8 *) (wblock->frame_buffer), wblock->data.packet.cap_len - pseudo_header_len, fh);
 885         if (bytes_read != (int) (wblock->data.packet.cap_len - pseudo_header_len)) {
 886                 *err = file_error(fh, err_info);
 887                 pcapng_debug1("pcapng_read_packet_block: couldn't read %u bytes of captured data",
 888                               wblock->data.packet.cap_len - pseudo_header_len);
 889                 if (*err == 0)
 890                         *err = WTAP_ERR_SHORT_READ;
 891                 return 0;
 892         }
 893         block_read += bytes_read;
 894
 895         /* jump over potential padding bytes at end of the packet data */
 896         if( (wblock->data.packet.cap_len % 4) != 0) {
 897                 file_offset64 = file_seek(fh, 4 - (wblock->data.packet.cap_len % 4), SEEK_CUR, err);
 898                 if (file_offset64 <= 0) {
 899                         if (*err != 0)
 900                                 return -1;
 901                         return 0;
 902                 }
 903                 block_read += 4 - (wblock->data.packet.cap_len % 4);
 904         }
 905
 906         /* add padding bytes to "block total length" */
 907         /* (the "block total length" of some example files don't contain the packet data padding bytes!) */
 908         if (bh->block_total_length % 4) {
 909                 block_total_length = bh->block_total_length + 4 - (bh->block_total_length % 4);
 910         } else {
 911                 block_total_length = bh->block_total_length;
 912         }
 913
 914         /* Option defaults */
 915         wblock->data.packet.opt_comment = NULL;
 916         wblock->data.packet.drop_count  = -1;
 917         wblock->data.packet.pack_flags  = 0;    /* XXX - is 0 ok to signal "not used"? */
 918
 919         /* FCS length default */
 920         fcslen = pn->if_fcslen;
 921
 922         /* Options */
 923         errno = WTAP_ERR_CANT_READ;
 924         to_read = block_total_length
 925         - (int)sizeof(pcapng_block_header_t)
 926         - block_read    /* fixed and variable part, including padding */
 927         - (int)sizeof(bh->block_total_length);
 928         while(to_read > 0) {
 929                 /* read option */
 930                 bytes_read = pcapng_read_option(fh, pn, &oh, option_content, sizeof(option_content), err, err_info);
 931                 if (bytes_read <= 0) {
 932                         pcapng_debug0("pcapng_read_packet_block: failed to read option");
 933                         return bytes_read;
 934                 }
 935                 block_read += bytes_read;
 936                 to_read -= bytes_read;
 937
 938                 /* handle option content */
 939                 switch(oh.option_code) {
 940                     case(0): /* opt_endofopt */
 941                         if(to_read != 0) {
 942                                 pcapng_debug1("pcapng_read_packet_block: %u bytes after opt_endofopt", to_read);
 943                         }
 944                         /* padding should be ok here, just get out of this */
 945                         to_read = 0;
 946                         break;
 947                     case(1): /* opt_comment */
 948                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
 949                                 wblock->data.packet.opt_comment = g_strndup(option_content, sizeof(option_content));
 950                                 pcapng_debug1("pcapng_read_packet_block: opt_comment %s", wblock->data.packet.opt_comment);
 951                         } else {
 952                                 pcapng_debug1("pcapng_read_packet_block: opt_comment length %u seems strange", oh.option_length);
 953                         }
 954                         break;
 955                     case(2): /* pack_flags / epb_flags */
 956                         if(oh.option_length == 4) {
 957                                 /*  Don't cast a char[] into a guint32--the
 958                                  *  char[] may not be aligned correctly.
 959                                  */
 960                                 memcpy(&wblock->data.packet.pack_flags, option_content, sizeof(guint32));
 961                                 if(pn->byte_swapped)
 962                                         wblock->data.packet.pack_flags = BSWAP32(wblock->data.packet.pack_flags);
 963                                 if (wblock->data.packet.pack_flags & 0x000001E0) {
 964                                         /* The FCS length is present */
 965                                         fcslen = (wblock->data.packet.pack_flags & 0x000001E0) >> 5;
 966                                 }
 967                                 pcapng_debug1("pcapng_read_if_descr_block: pack_flags %u (ignored)", wblock->data.packet.pack_flags);
 968                         } else {
 969                                 pcapng_debug1("pcapng_read_if_descr_block: pack_flags length %u not 4 as expected", oh.option_length);
 970                         }
 971                         break;
 972                     default:
 973                         pcapng_debug2("pcapng_read_packet_block: unknown option %u - ignoring %u bytes",
 974                                       oh.option_code, oh.option_length);
 975                 }
 976         }
 977
 978         pcap_read_post_process(WTAP_FILE_PCAPNG, wtap_encap,
 979             (union wtap_pseudo_header *)wblock->pseudo_header,
 980             (guint8 *) (wblock->frame_buffer),
 981             (int) (wblock->data.packet.cap_len - pseudo_header_len),
 982             pn->byte_swapped, fcslen);
 983         return block_read;
 984 }
 985
 986
 987 static int
 988 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)
 989 {
 990         int bytes_read;
 991         int block_read;
 992         guint64 file_offset64;
 993         gint encap;
 994         int pseudo_header_len;
 995         pcapng_simple_packet_block_t spb;
 996
 997
 998         /* "Simple Packet Block" read fixed part */
 999         errno = WTAP_ERR_CANT_READ;
1000         bytes_read = file_read(&spb, sizeof spb, fh);
1001         if (bytes_read != sizeof spb) {
1002                 pcapng_debug0("pcapng_read_simple_packet_block: failed to read packet data");
1003                 *err = file_error(fh, err_info);
1004                 return 0;
1005         }
1006         block_read = bytes_read;
1007
1008         if (pn->byte_swapped) {
1009                 wblock->data.simple_packet.packet_len   = BSWAP32(spb.packet_len);
1010         } else {
1011                 wblock->data.simple_packet.packet_len   = spb.packet_len;
1012         }
1013
1014         wblock->data.simple_packet.cap_len = bh->block_total_length
1015                                              - (guint32)sizeof(pcapng_simple_packet_block_t)
1016                                              - (guint32)sizeof(bh->block_total_length);
1017
1018         if (wblock->data.simple_packet.cap_len > WTAP_MAX_PACKET_SIZE) {
1019                 *err = WTAP_ERR_BAD_FILE;
1020                 *err_info = g_strdup_printf("pcapng_read_simple_packet_block: cap_len %u is larger than WTAP_MAX_PACKET_SIZE %u.",
1021                     wblock->data.simple_packet.cap_len, WTAP_MAX_PACKET_SIZE);
1022                 return 0;
1023         }
1024         pcapng_debug1("pcapng_read_simple_packet_block: packet data: packet_len %u",
1025                        wblock->data.simple_packet.packet_len);
1026
1027         encap = pcapng_get_encap(0, pn);
1028         pcapng_debug1("pcapng_read_simple_packet_block: Need to read pseudo header of size %d",
1029                       pcap_get_phdr_size(encap, wblock->pseudo_header));
1030
1031         memset((void *)wblock->pseudo_header, 0, sizeof(union wtap_pseudo_header));
1032         pseudo_header_len = pcap_process_pseudo_header(fh,
1033                                                        WTAP_FILE_PCAPNG,
1034                                                        encap,
1035                                                        wblock->data.simple_packet.cap_len,
1036                                                        TRUE,
1037                                                        wblock->packet_header,
1038                                                        (union wtap_pseudo_header *)wblock->pseudo_header,
1039                                                        err,
1040                                                        err_info);
1041         if (pseudo_header_len < 0) {
1042                 return 0;
1043         }
1044         wblock->data.simple_packet.pseudo_header_len = (guint32)pseudo_header_len;
1045         block_read += pseudo_header_len;
1046         if (pseudo_header_len != pcap_get_phdr_size(encap, wblock->pseudo_header)) {
1047                 pcapng_debug1("pcapng_read_simple_packet_block: Could only read %d bytes for pseudo header.",
1048                               pseudo_header_len);
1049         }
1050
1051         memset((void *)wblock->pseudo_header, 0, sizeof(union wtap_pseudo_header));
1052
1053         /* "Simple Packet Block" read capture data */
1054         errno = WTAP_ERR_CANT_READ;
1055         bytes_read = file_read((guint8 *) (wblock->frame_buffer), wblock->data.simple_packet.cap_len, fh);
1056         if (bytes_read != (int) wblock->data.simple_packet.cap_len) {
1057                 *err = file_error(fh, err_info);
1058                 pcapng_debug1("pcapng_read_simple_packet_block: couldn't read %u bytes of captured data",
1059                               wblock->data.simple_packet.cap_len);
1060                 if (*err == 0)
1061                         *err = WTAP_ERR_SHORT_READ;
1062                 return 0;
1063         }
1064         block_read += bytes_read;
1065
1066         /* jump over potential padding bytes at end of the packet data */
1067         if ((wblock->data.simple_packet.cap_len % 4) != 0) {
1068                 file_offset64 = file_seek(fh, 4 - (wblock->data.simple_packet.cap_len % 4), SEEK_CUR, err);
1069                 if (file_offset64 <= 0) {
1070                         if (*err != 0)
1071                                 return -1;
1072                         return 0;
1073                 }
1074                 block_read += 4 - (wblock->data.simple_packet.cap_len % 4);
1075         }
1076
1077         pcap_read_post_process(WTAP_FILE_PCAPNG, encap,
1078             (union wtap_pseudo_header *)wblock->pseudo_header,
1079             (guint8 *) (wblock->frame_buffer),
1080             (int) wblock->data.simple_packet.cap_len,
1081             pn->byte_swapped, pn->if_fcslen);
1082         return block_read;
1083 }
1084
1085 #define NRES_ENDOFRECORD 0
1086 #define NRES_IP4RECORD 1
1087 #define NRES_IP6RECORD 2
1088 #define PADDING4(x) ((((x + 3) >> 2) << 2) - x)
1089 /* IPv6 + MAXNAMELEN */
1090 #define MAX_NRB_REC_SIZE (16 + 64)
1091 static int
1092 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)
1093 {
1094         int bytes_read = 0;
1095         int block_read = 0;
1096         int to_read;
1097         guint64 file_offset64;
1098         pcapng_name_resolution_block_t nrb;
1099         guint8 nrb_rec[MAX_NRB_REC_SIZE];
1100         guint32 v4_addr;
1101
1102         errno = WTAP_ERR_CANT_READ;
1103         to_read = bh->block_total_length
1104                 - sizeof(pcapng_block_header_t)
1105                 - sizeof(bh->block_total_length);
1106
1107         while (block_read < to_read) {
1108                 bytes_read = file_read(&nrb, sizeof nrb, fh);
1109                 if (bytes_read != sizeof nrb) {
1110                         pcapng_debug0("pcapng_read_name_resolution_block: failed to read record header");
1111                         *err = file_error(fh, err_info);
1112                         return 0;
1113                 }
1114                 block_read += bytes_read;
1115
1116                 if (pn->byte_swapped) {
1117                         nrb.record_type = BSWAP16(nrb.record_type);
1118                         nrb.record_len  = BSWAP16(nrb.record_len);
1119                 }
1120
1121                 switch(nrb.record_type) {
1122                         case NRES_ENDOFRECORD:
1123                                 /* There shouldn't be any more data */
1124                                 to_read = 0;
1125                                 break;
1126                         case NRES_IP4RECORD:
1127                                 if (nrb.record_len < 6 || nrb.record_len > MAX_NRB_REC_SIZE || to_read < nrb.record_len) {
1128                                         pcapng_debug0("pcapng_read_name_resolution_block: bad length or insufficient data for IPv4 record");
1129                                         return 0;
1130                                 }
1131                                 bytes_read = file_read(nrb_rec, nrb.record_len, fh);
1132                                 if (bytes_read != nrb.record_len) {
1133                                         pcapng_debug0("pcapng_read_name_resolution_block: failed to read IPv4 record data");
1134                                         *err = file_error(fh, err_info);
1135                                         return 0;
1136                                 }
1137                                 block_read += bytes_read;
1138
1139                                 if (pn->add_new_ipv4) {
1140                                         memcpy(&v4_addr, nrb_rec, 4);
1141                                         if (pn->byte_swapped)
1142                                                 v4_addr = BSWAP32(v4_addr);
1143                                         pn->add_new_ipv4(v4_addr, nrb_rec + 4);
1144                                 }
1145
1146                                 file_offset64 = file_seek(fh, PADDING4(nrb.record_len), SEEK_CUR, err);
1147                                 if (file_offset64 <= 0) {
1148                                         if (*err != 0)
1149                                                 return -1;
1150                                         return 0;
1151                                 }
1152                                 block_read += PADDING4(nrb.record_len);
1153                                 break;
1154                         case NRES_IP6RECORD:
1155                                 if (nrb.record_len < 18 || nrb.record_len > MAX_NRB_REC_SIZE || to_read < nrb.record_len) {
1156                                         pcapng_debug0("pcapng_read_name_resolution_block: bad length or insufficient data for IPv6 record");
1157                                         return 0;
1158                                 }
1159                                 bytes_read = file_read(nrb_rec, nrb.record_len, fh);
1160                                 if (bytes_read != nrb.record_len) {
1161                                         pcapng_debug0("pcapng_read_name_resolution_block: failed to read IPv6 record data");
1162                                         *err = file_error(fh, err_info);
1163                                         return 0;
1164                                 }
1165                                 block_read += bytes_read;
1166
1167                                 if (pn->add_new_ipv6) {
1168                                         pn->add_new_ipv6(nrb_rec, nrb_rec + 16);
1169                                 }
1170
1171                                 file_offset64 = file_seek(fh, PADDING4(nrb.record_len), SEEK_CUR, err);
1172                                 if (file_offset64 <= 0) {
1173                                         if (*err != 0)
1174                                                 return -1;
1175                                         return 0;
1176                                 }
1177                                 block_read += PADDING4(nrb.record_len);
1178                                 break;
1179                         default:
1180                                 pcapng_debug1("pcapng_read_name_resolution_block: unknown record type 0x%x", nrb.record_type);
1181                                 file_offset64 = file_seek(fh, nrb.record_len + PADDING4(nrb.record_len), SEEK_CUR, err);
1182                                 if (file_offset64 <= 0) {
1183                                         if (*err != 0)
1184                                                 return -1;
1185                                         return 0;
1186                                 }
1187                                 block_read += nrb.record_len + PADDING4(nrb.record_len);
1188                                 break;
1189                 }
1190         }
1191
1192         return block_read;
1193 }
1194
1195 static int
1196 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)
1197 {
1198         int bytes_read;
1199         int block_read;
1200         int to_read;
1201         pcapng_interface_statistics_block_t isb;
1202         pcapng_option_header_t oh;
1203         char option_content[100]; /* XXX - size might need to be increased, if we see longer options */
1204
1205
1206         /* "Interface Statistics Block" read fixed part */
1207         errno = WTAP_ERR_CANT_READ;
1208         bytes_read = file_read(&isb, sizeof isb, fh);
1209         if (bytes_read != sizeof isb) {
1210                 pcapng_debug0("pcapng_read_interface_statistics_block: failed to read packet data");
1211                 *err = file_error(fh, err_info);
1212                 return 0;
1213         }
1214         block_read = bytes_read;
1215
1216         if(pn->byte_swapped) {
1217                 wblock->data.if_stats.interface_id      = BSWAP64(isb.interface_id);
1218                 wblock->data.if_stats.ts_high           = BSWAP32(isb.timestamp_high);
1219                 wblock->data.if_stats.ts_low            = BSWAP32(isb.timestamp_low);
1220         } else {
1221                 wblock->data.if_stats.interface_id      = isb.interface_id;
1222                 wblock->data.if_stats.ts_high           = isb.timestamp_high;
1223                 wblock->data.if_stats.ts_low            = isb.timestamp_low;
1224         }
1225         pcapng_debug1("pcapng_read_interface_statistics_block: interface_id %" G_GINT64_MODIFIER "u", wblock->data.if_stats.interface_id);
1226
1227         /* Option defaults */
1228         wblock->data.if_stats.opt_comment = NULL;
1229         wblock->data.if_stats.isb_ifrecv  = -1;
1230         wblock->data.if_stats.isb_ifdrop  = -1;
1231
1232         /* Options */
1233         errno = WTAP_ERR_CANT_READ;
1234         to_read = bh->block_total_length
1235         - sizeof(pcapng_block_header_t)
1236         - block_read    /* fixed and variable part, including padding */
1237         - sizeof(bh->block_total_length);
1238         while(to_read > 0) {
1239                 /* read option */
1240                 bytes_read = pcapng_read_option(fh, pn, &oh, option_content, sizeof(option_content), err, err_info);
1241                 if (bytes_read <= 0) {
1242                         pcapng_debug0("pcapng_read_interface_statistics_block: failed to read option");
1243                         return bytes_read;
1244                 }
1245                 block_read += bytes_read;
1246                 to_read -= bytes_read;
1247
1248                 /* handle option content */
1249                 switch(oh.option_code) {
1250                     case(0): /* opt_endofopt */
1251                         if(to_read != 0) {
1252                                 pcapng_debug1("pcapng_read_interface_statistics_block: %u bytes after opt_endofopt", to_read);
1253                         }
1254                         /* padding should be ok here, just get out of this */
1255                         to_read = 0;
1256                         break;
1257                     case(1): /* opt_comment */
1258                         if(oh.option_length > 0 && oh.option_length < sizeof(option_content)) {
1259                                 wblock->data.if_stats.opt_comment = g_strndup(option_content, sizeof(option_content));
1260                                 pcapng_debug1("pcapng_read_interface_statistics_block: opt_comment %s", wblock->data.if_stats.opt_comment);
1261                         } else {
1262                                 pcapng_debug1("pcapng_read_interface_statistics_block: opt_comment length %u seems strange", oh.option_length);
1263                         }
1264                         break;
1265                     case(4): /* isb_ifrecv */
1266                         if(oh.option_length == 8) {
1267                                 /*  Don't cast a char[] into a guint32--the
1268                                  *  char[] may not be aligned correctly.
1269                                  */
1270                                 memcpy(&wblock->data.if_stats.isb_ifrecv, option_content, sizeof(guint64));
1271                                 if(pn->byte_swapped)
1272                                         wblock->data.if_stats.isb_ifrecv = BSWAP64(wblock->data.if_stats.isb_ifrecv);
1273                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifrecv %" G_GINT64_MODIFIER "u", wblock->data.if_stats.isb_ifrecv);
1274                         } else {
1275                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifrecv length %u not 8 as expected", oh.option_length);
1276                         }
1277                         break;
1278                     case(5): /* isb_ifdrop */
1279                         if(oh.option_length == 8) {
1280                                 /*  Don't cast a char[] into a guint32--the
1281                                  *  char[] may not be aligned correctly.
1282                                  */
1283                                 memcpy(&wblock->data.if_stats.isb_ifdrop, option_content, sizeof(guint64));
1284                                 if(pn->byte_swapped)
1285                                         wblock->data.if_stats.isb_ifdrop = BSWAP64(wblock->data.if_stats.isb_ifdrop);
1286                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifdrop %" G_GINT64_MODIFIER "u", wblock->data.if_stats.isb_ifdrop);
1287                         } else {
1288                                 pcapng_debug1("pcapng_read_interface_statistics_block: isb_ifdrop length %u not 8 as expected", oh.option_length);
1289                         }
1290                         break;
1291                     default:
1292                         pcapng_debug2("pcapng_read_interface_statistics_block: unknown option %u - ignoring %u bytes",
1293                                       oh.option_code, oh.option_length);
1294                 }
1295         }
1296
1297     return block_read;
1298 }
1299
1300
1301 static int
1302 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_)
1303 {
1304         int block_read;
1305         guint64 file_offset64;
1306         guint32 block_total_length;
1307
1308
1309         /* add padding bytes to "block total length" */
1310         /* (the "block total length" of some example files don't contain any padding bytes!) */
1311         if (bh->block_total_length % 4) {
1312                 block_total_length = bh->block_total_length + 4 - (bh->block_total_length % 4);
1313         } else {
1314                 block_total_length = bh->block_total_length;
1315         }
1316
1317         block_read = block_total_length - (guint32)sizeof(pcapng_block_header_t) - (guint32)sizeof(bh->block_total_length);
1318
1319         /* jump over this unknown block */
1320         file_offset64 = file_seek(fh, block_read, SEEK_CUR, err);
1321         if (file_offset64 <= 0) {
1322                 if (*err != 0)
1323                         return -1;
1324                 return 0;
1325         }
1326
1327         return block_read;
1328 }
1329
1330
1331 static int
1332 pcapng_read_block(FILE_T fh, gboolean first_block, pcapng_t *pn, wtapng_block_t *wblock, int *err, gchar **err_info)
1333 {
1334         int block_read;
1335         int bytes_read;
1336         pcapng_block_header_t bh;
1337         guint32 block_total_length;
1338
1339
1340         /* Try to read the (next) block header */
1341         errno = WTAP_ERR_CANT_READ;
1342         bytes_read = file_read(&bh, sizeof bh, fh);
1343         if (bytes_read != sizeof bh) {
1344                 *err = file_error(fh, err_info);
1345                 pcapng_debug3("pcapng_read_block: file_read() returned %d instead of %u, err = %d.", bytes_read, (unsigned int)sizeof bh, *err);
1346                 if (*err != 0)
1347                         return -1;
1348                 return 0;
1349         }
1350
1351         block_read = bytes_read;
1352         if (pn->byte_swapped) {
1353                 bh.block_type         = BSWAP32(bh.block_type);
1354                 bh.block_total_length = BSWAP32(bh.block_total_length);
1355         }
1356
1357         wblock->type = bh.block_type;
1358
1359         pcapng_debug1("pcapng_read_block: block_type 0x%x", bh.block_type);
1360
1361         if (first_block) {
1362                 /*
1363                  * This is being read in by pcapng_open(), so this block
1364                  * must be an SHB.  If it's not, this is not a pcap-ng
1365                  * file.
1366                  *
1367                  * XXX - check for various forms of Windows <-> UN*X
1368                  * mangling, and suggest that the file might be a
1369                  * pcap-ng file that was damaged in transit?
1370                  */
1371                 if (bh.block_type != BLOCK_TYPE_SHB)
1372                         return 0;       /* not a pcap-ng file */
1373         }
1374
1375         switch(bh.block_type) {
1376                 case(BLOCK_TYPE_SHB):
1377                         bytes_read = pcapng_read_section_header_block(fh, first_block, &bh, pn, wblock, err, err_info);
1378                         break;
1379                 case(BLOCK_TYPE_IDB):
1380                         bytes_read = pcapng_read_if_descr_block(fh, &bh, pn, wblock, err, err_info);
1381                         break;
1382                 case(BLOCK_TYPE_PB):
1383                         bytes_read = pcapng_read_packet_block(fh, &bh, pn, wblock, err, err_info, FALSE);
1384                         break;
1385                 case(BLOCK_TYPE_SPB):
1386                         bytes_read = pcapng_read_simple_packet_block(fh, &bh, pn, wblock, err, err_info);
1387                         break;
1388                 case(BLOCK_TYPE_EPB):
1389                         bytes_read = pcapng_read_packet_block(fh, &bh, pn, wblock, err, err_info, TRUE);
1390                         break;
1391                 case(BLOCK_TYPE_NRB):
1392                         bytes_read = pcapng_read_name_resolution_block(fh, &bh, pn, wblock, err, err_info);
1393                         break;
1394                 case(BLOCK_TYPE_ISB):
1395                         bytes_read = pcapng_read_interface_statistics_block(fh, &bh, pn, wblock, err, err_info);
1396                         break;
1397                 default:
1398                         pcapng_debug2("pcapng_read_block: Unknown block_type: 0x%x (block ignored), block total length %d", bh.block_type, bh.block_total_length);
1399                         bytes_read = pcapng_read_unknown_block(fh, &bh, pn, wblock, err, err_info);
1400         }
1401
1402         if (bytes_read <= 0) {
1403                 return bytes_read;
1404         }
1405         block_read += bytes_read;
1406
1407         /* sanity check: first and second block lengths must match */
1408         errno = WTAP_ERR_CANT_READ;
1409         bytes_read = file_read(&block_total_length, sizeof block_total_length, fh);
1410         if (bytes_read != sizeof block_total_length) {
1411                 pcapng_debug0("pcapng_read_block: couldn't read second block length");
1412                 *err = file_error(fh, err_info);
1413                 if (*err == 0)
1414                         *err = WTAP_ERR_SHORT_READ;
1415                 return -1;
1416         }
1417         block_read += bytes_read;
1418
1419         if (pn->byte_swapped)
1420                 block_total_length = BSWAP32(block_total_length);
1421
1422         if (!(block_total_length == bh.block_total_length)) {
1423                 *err = WTAP_ERR_BAD_FILE;
1424                 *err_info = g_strdup_printf("pcapng_read_block: total block lengths (first %u and second %u) don't match",
1425                               bh.block_total_length, block_total_length);
1426                 return -1;
1427         }
1428
1429         return block_read;
1430 }
1431
1432
1433 /* classic wtap: open capture file */
1434 int
1435 pcapng_open(wtap *wth, int *err, gchar **err_info)
1436 {
1437         int bytes_read;
1438         pcapng_t pn;
1439         wtapng_block_t wblock;
1440         pcapng_t *pcapng;
1441
1442         /* we don't know the byte swapping of the file yet */
1443         pn.byte_swapped = FALSE;
1444         pn.if_fcslen = -1;
1445         pn.version_major = -1;
1446         pn.version_minor = -1;
1447         pn.interface_data = NULL;
1448         pn.number_of_interfaces = 0;
1449
1450         /* we don't expect any packet blocks yet */
1451         wblock.frame_buffer = NULL;
1452         wblock.pseudo_header = NULL;
1453         wblock.packet_header = NULL;
1454         wblock.file_encap = &wth->file_encap;
1455
1456         pcapng_debug0("pcapng_open: opening file");
1457         /* read first block */
1458         bytes_read = pcapng_read_block(wth->fh, TRUE, &pn, &wblock, err, err_info);
1459         if (bytes_read <= 0) {
1460                 pcapng_debug0("pcapng_open: couldn't read first SHB");
1461                 *err = file_error(wth->fh, err_info);
1462                 if (*err != 0)
1463                         return -1;
1464                 return 0;
1465         }
1466         wth->data_offset += bytes_read;
1467
1468         /* first block must be a "Section Header Block" */
1469         if (wblock.type != BLOCK_TYPE_SHB) {
1470                 /*
1471                  * XXX - check for damage from transferring a file
1472                  * between Windows and UN*X as text rather than
1473                  * binary data?
1474                  */
1475                 pcapng_debug1("pcapng_open: first block type %u not SHB", wblock.type);
1476                 return 0;
1477         }
1478
1479         wth->file_encap = WTAP_ENCAP_UNKNOWN;
1480         wth->snapshot_length = 0;
1481         wth->tsprecision = WTAP_FILE_TSPREC_NSEC;
1482         pcapng = (pcapng_t *)g_malloc(sizeof(pcapng_t));
1483         wth->priv = (void *)pcapng;
1484         *pcapng = pn;
1485         wth->subtype_read = pcapng_read;
1486         wth->subtype_seek_read = pcapng_seek_read;
1487         wth->subtype_close = pcapng_close;
1488         wth->file_type = WTAP_FILE_PCAPNG;
1489
1490         return 1;
1491 }
1492
1493
1494 /* classic wtap: read packet */
1495 static gboolean
1496 pcapng_read(wtap *wth, int *err, gchar **err_info, gint64 *data_offset)
1497 {
1498         pcapng_t *pcapng = (pcapng_t *)wth->priv;
1499         int bytes_read;
1500         guint64 ts;
1501         wtapng_block_t wblock;
1502
1503         pcapng_debug1("pcapng_read: wth->data_offset is initially %" G_GINT64_MODIFIER "u", wth->data_offset);
1504         *data_offset = wth->data_offset;
1505         pcapng_debug1("pcapng_read: *data_offset is initially set to %" G_GINT64_MODIFIER "u", *data_offset);
1506
1507         /* XXX - This should be done in the packet block reading function and
1508          * should make use of the caplen of the packet.
1509          */
1510         if (wth->snapshot_length > 0) {
1511                 buffer_assure_space(wth->frame_buffer, wth->snapshot_length);
1512         } else {
1513                 buffer_assure_space(wth->frame_buffer, WTAP_MAX_PACKET_SIZE);
1514         }
1515
1516         wblock.frame_buffer  = buffer_start_ptr(wth->frame_buffer);
1517         wblock.pseudo_header = &wth->pseudo_header;
1518         wblock.packet_header = &wth->phdr;
1519         wblock.file_encap    = &wth->file_encap;
1520
1521         pcapng->add_new_ipv4 = wth->add_new_ipv4;
1522         pcapng->add_new_ipv6 = wth->add_new_ipv6;
1523
1524         /* read next block */
1525         while (1) {
1526                 bytes_read = pcapng_read_block(wth->fh, FALSE, pcapng, &wblock, err, err_info);
1527                 if (bytes_read <= 0) {
1528                         wth->data_offset = *data_offset;
1529                         pcapng_debug1("pcapng_read: wth->data_offset is finally %" G_GINT64_MODIFIER "u", wth->data_offset);
1530                         pcapng_debug0("pcapng_read: couldn't read packet block");
1531                         return FALSE;
1532                 }
1533
1534                 /* block must be a "Packet Block" or an "Enhanced Packet Block" -> otherwise continue */
1535                 if (wblock.type == BLOCK_TYPE_PB || wblock.type == BLOCK_TYPE_EPB) {
1536                         break;
1537                 }
1538
1539                 /* XXX - improve handling of "unknown" blocks */
1540                 pcapng_debug1("pcapng_read: block type 0x%x not PB/EPB", wblock.type);
1541                 *data_offset += bytes_read;
1542                 pcapng_debug1("pcapng_read: *data_offset is updated to %" G_GINT64_MODIFIER "u", *data_offset);
1543         }
1544
1545         /* Combine the two 32-bit pieces of the timestamp into one 64-bit value */
1546         ts = (((guint64)wblock.data.packet.ts_high) << 32) | ((guint64)wblock.data.packet.ts_low);
1547
1548         wth->phdr.caplen = wblock.data.packet.cap_len - wblock.data.packet.pseudo_header_len;
1549         wth->phdr.len = wblock.data.packet.packet_len - wblock.data.packet.pseudo_header_len;
1550         if (wblock.data.packet.interface_id < pcapng->number_of_interfaces) {
1551                 interface_data_t int_data;
1552                 guint64 time_units_per_second;
1553                 gint id;
1554
1555                 id = (gint)wblock.data.packet.interface_id;
1556                 int_data = g_array_index(pcapng->interface_data, interface_data_t, id);
1557                 time_units_per_second = int_data.time_units_per_second;
1558                 wth->phdr.pkt_encap = int_data.wtap_encap;
1559                 wth->phdr.ts.secs = (time_t)(ts / time_units_per_second);
1560                 wth->phdr.ts.nsecs = (int)(((ts % time_units_per_second) * 1000000000) / time_units_per_second);
1561         } else {
1562                 wth->phdr.pkt_encap = WTAP_ENCAP_UNKNOWN;
1563                 *err = WTAP_ERR_BAD_FILE;
1564                 *err_info = g_strdup_printf("pcapng: interface index %u is not less than interface count %u.",
1565                     wblock.data.packet.interface_id, pcapng->number_of_interfaces);
1566                 wth->data_offset = *data_offset + bytes_read;
1567                 pcapng_debug1("pcapng_read: wth->data_offset is finally %" G_GINT64_MODIFIER "u", wth->data_offset);
1568                 return FALSE;
1569         }
1570
1571         /*pcapng_debug2("Read length: %u Packet length: %u", bytes_read, wth->phdr.caplen);*/
1572         wth->data_offset = *data_offset + bytes_read;
1573         pcapng_debug1("pcapng_read: wth->data_offset is finally %" G_GINT64_MODIFIER "u", wth->data_offset);
1574
1575         return TRUE;
1576 }
1577
1578
1579 /* classic wtap: seek to file position and read packet */
1580 static gboolean
1581 pcapng_seek_read(wtap *wth, gint64 seek_off,
1582     union wtap_pseudo_header *pseudo_header, guint8 *pd, int length _U_,
1583     int *err, gchar **err_info)
1584 {
1585         pcapng_t *pcapng = (pcapng_t *)wth->priv;
1586         guint64 bytes_read64;
1587         int bytes_read;
1588         wtapng_block_t wblock;
1589
1590
1591         /* seek to the right file position */
1592         bytes_read64 = file_seek(wth->random_fh, seek_off, SEEK_SET, err);
1593         if (bytes_read64 <= 0) {
1594                 return FALSE;   /* Seek error */
1595         }
1596         pcapng_debug1("pcapng_seek_read: reading at offset %" G_GINT64_MODIFIER "u", seek_off);
1597
1598         wblock.frame_buffer = pd;
1599         wblock.pseudo_header = pseudo_header;
1600         wblock.packet_header = &wth->phdr;
1601         wblock.file_encap = &wth->file_encap;
1602
1603         /* read the block */
1604         bytes_read = pcapng_read_block(wth->random_fh, FALSE, pcapng, &wblock, err, err_info);
1605         if (bytes_read <= 0) {
1606                 *err = file_error(wth->random_fh, err_info);
1607                 pcapng_debug3("pcapng_seek_read: couldn't read packet block (err=%d, errno=%d, bytes_read=%d).",
1608                               *err, errno, bytes_read);
1609                 return FALSE;
1610         }
1611
1612         /* block must be a "Packet Block" or an "Enhanced Packet Block" */
1613         if (wblock.type != BLOCK_TYPE_PB && wblock.type != BLOCK_TYPE_EPB) {
1614                 pcapng_debug1("pcapng_seek_read: block type %u not PB/EPB", wblock.type);
1615                 return FALSE;
1616         }
1617
1618         return TRUE;
1619 }
1620
1621
1622 /* classic wtap: close capture file */
1623 static void
1624 pcapng_close(wtap *wth)
1625 {
1626         pcapng_t *pcapng = (pcapng_t *)wth->priv;
1627
1628         pcapng_debug0("pcapng_close: closing file");
1629         if (pcapng->interface_data != NULL) {
1630                 g_array_free(pcapng->interface_data, TRUE);
1631         }
1632 }
1633
1634
1635
1636 typedef struct {
1637         GArray *interface_data;
1638         guint number_of_interfaces;
1639         struct addrinfo *addrinfo_list_last;
1640 } pcapng_dump_t;
1641
1642 static gboolean
1643 pcapng_write_section_header_block(wtap_dumper *wdh, wtapng_block_t *wblock, int *err)
1644 {
1645         pcapng_block_header_t bh;
1646         pcapng_section_header_block_t shb;
1647
1648
1649         /* write block header */
1650         bh.block_type = wblock->type;
1651         bh.block_total_length = sizeof(bh) + sizeof(shb) /* + options */ + 4;
1652
1653         if (!wtap_dump_file_write(wdh, &bh, sizeof bh, err))
1654                 return FALSE;
1655         wdh->bytes_dumped += sizeof bh;
1656
1657         /* write block fixed content */
1658         /* XXX - get these values from wblock? */
1659         shb.magic = 0x1A2B3C4D;
1660         shb.version_major = 1;
1661         shb.version_minor = 0;
1662         shb.section_length = -1;
1663
1664         if (!wtap_dump_file_write(wdh, &shb, sizeof shb, err))
1665                 return FALSE;
1666         wdh->bytes_dumped += sizeof shb;
1667
1668         /* XXX - write (optional) block options */
1669
1670         /* write block footer */
1671         if (!wtap_dump_file_write(wdh, &bh.block_total_length,
1672             sizeof bh.block_total_length, err))
1673                 return FALSE;
1674         wdh->bytes_dumped += sizeof bh.block_total_length;
1675
1676         return TRUE;
1677 }
1678
1679
1680
1681 static gboolean
1682 pcapng_write_if_descr_block(wtap_dumper *wdh, wtapng_block_t *wblock, int *err)
1683 {
1684         pcapng_block_header_t bh;
1685         pcapng_interface_description_block_t idb;
1686
1687
1688         pcapng_debug3("pcapng_write_if_descr_block: encap = %d (%s), snaplen = %d",
1689                       wblock->data.if_descr.link_type,
1690                       wtap_encap_string(wtap_pcap_encap_to_wtap_encap(wblock->data.if_descr.link_type)),
1691                       wblock->data.if_descr.snap_len);
1692
1693         if (wblock->data.if_descr.link_type == (guint16)-1) {
1694                 *err = WTAP_ERR_UNSUPPORTED_ENCAP;
1695                 return FALSE;
1696         }
1697
1698         /* write block header */
1699         bh.block_type = wblock->type;
1700         bh.block_total_length = sizeof(bh) + sizeof(idb) /* + options */ + 4;
1701
1702         if (!wtap_dump_file_write(wdh, &bh, sizeof bh, err))
1703                 return FALSE;
1704         wdh->bytes_dumped += sizeof bh;
1705
1706         /* write block fixed content */
1707         idb.linktype    = wblock->data.if_descr.link_type;
1708         idb.reserved    = 0;
1709         idb.snaplen     = wblock->data.if_descr.snap_len;
1710
1711         if (!wtap_dump_file_write(wdh, &idb, sizeof idb, err))
1712                 return FALSE;
1713         wdh->bytes_dumped += sizeof idb;
1714
1715         /* XXX - write (optional) block options */
1716
1717         /* write block footer */
1718         if (!wtap_dump_file_write(wdh, &bh.block_total_length,
1719             sizeof bh.block_total_length, err))
1720                 return FALSE;
1721         wdh->bytes_dumped += sizeof bh.block_total_length;
1722
1723         return TRUE;
1724 }
1725
1726
1727 static gboolean
1728 pcapng_write_packet_block(wtap_dumper *wdh, wtapng_block_t *wblock, int *err)
1729 {
1730         pcapng_block_header_t bh;
1731         pcapng_enhanced_packet_block_t epb;
1732         const guint32 zero_pad = 0;
1733         guint32 pad_len;
1734         guint32 phdr_len;
1735
1736         phdr_len = (guint32)pcap_get_phdr_size(wblock->data.packet.wtap_encap, wblock->pseudo_header);
1737         if ((phdr_len + wblock->data.packet.cap_len) % 4) {
1738                 pad_len = 4 - ((phdr_len + wblock->data.packet.cap_len) % 4);
1739         } else {
1740                 pad_len = 0;
1741         }
1742
1743         /* write (enhanced) packet block header */
1744         bh.block_type = wblock->type;
1745         bh.block_total_length = (guint32)sizeof(bh) + (guint32)sizeof(epb) + phdr_len + wblock->data.packet.cap_len + pad_len /* + options */ + 4;
1746
1747         if (!wtap_dump_file_write(wdh, &bh, sizeof bh, err))
1748                 return FALSE;
1749         wdh->bytes_dumped += sizeof bh;
1750
1751         /* write block fixed content */
1752         epb.interface_id        = wblock->data.packet.interface_id;
1753         epb.timestamp_high      = wblock->data.packet.ts_high;
1754         epb.timestamp_low       = wblock->data.packet.ts_low;
1755         epb.captured_len        = wblock->data.packet.cap_len + phdr_len;
1756         epb.packet_len          = wblock->data.packet.packet_len + phdr_len;
1757
1758         if (!wtap_dump_file_write(wdh, &epb, sizeof epb, err))
1759                 return FALSE;
1760         wdh->bytes_dumped += sizeof epb;
1761
1762         /* write pseudo header */
1763         if (!pcap_write_phdr(wdh, wblock->data.packet.wtap_encap, wblock->pseudo_header, err)) {
1764                 return FALSE;
1765         }
1766         wdh->bytes_dumped += phdr_len;
1767
1768         /* write packet data */
1769         if (!wtap_dump_file_write(wdh, wblock->frame_buffer,
1770             wblock->data.packet.cap_len, err))
1771                 return FALSE;
1772         wdh->bytes_dumped += wblock->data.packet.cap_len;
1773
1774         /* write padding (if any) */
1775         if (pad_len != 0) {
1776                 if (!wtap_dump_file_write(wdh, &zero_pad, pad_len, err))
1777                         return FALSE;
1778                 wdh->bytes_dumped += pad_len;
1779         }
1780
1781         /* XXX - write (optional) block options */
1782
1783         /* write block footer */
1784         if (!wtap_dump_file_write(wdh, &bh.block_total_length,
1785             sizeof bh.block_total_length, err))
1786                 return FALSE;
1787         wdh->bytes_dumped += sizeof bh.block_total_length;
1788
1789         return TRUE;
1790 }
1791
1792 /* Arbitrary. */
1793 #define NRES_REC_MAX_SIZE ((WTAP_MAX_PACKET_SIZE * 4) + 16)
1794 static gboolean
1795 pcapng_write_name_resolution_block(wtap_dumper *wdh, pcapng_dump_t *pcapng, int *err)
1796 {
1797         pcapng_block_header_t bh;
1798         pcapng_name_resolution_block_t nrb;
1799         struct addrinfo *ai;
1800         struct sockaddr_in *sa4;
1801         struct sockaddr_in6 *sa6;
1802         guint8 *rec_data;
1803         gint rec_off, namelen, tot_rec_len;
1804
1805         if (! pcapng->addrinfo_list_last || ! pcapng->addrinfo_list_last->ai_next) {
1806                 return TRUE;
1807         }
1808
1809         rec_off = 8; /* block type + block total length */
1810         bh.block_type = BLOCK_TYPE_NRB;
1811         bh.block_total_length = rec_off + 8; /* end-of-record + block total length */
1812         rec_data = g_malloc(NRES_REC_MAX_SIZE);
1813
1814         for (; pcapng->addrinfo_list_last && pcapng->addrinfo_list_last->ai_next; pcapng->addrinfo_list_last = pcapng->addrinfo_list_last->ai_next ) {
1815                 ai = pcapng->addrinfo_list_last->ai_next; /* Skips over the first (dummy) entry */
1816                 namelen = (gint)strlen(ai->ai_canonname) + 1;
1817                 if (ai->ai_family == AF_INET) {
1818                         nrb.record_type = NRES_IP4RECORD;
1819                         nrb.record_len = 4 + namelen;
1820                         tot_rec_len = 4 + nrb.record_len + PADDING4(nrb.record_len);
1821                         bh.block_total_length += tot_rec_len;
1822
1823                         if (rec_off + tot_rec_len > NRES_REC_MAX_SIZE)
1824                                 break;
1825
1826                         /*
1827                          * The joys of BSD sockaddrs.  In practice, this
1828                          * cast is alignment-safe.
1829                          */
1830                         sa4 = (struct sockaddr_in *)(void *)ai->ai_addr;
1831                         memcpy(rec_data + rec_off, &nrb, sizeof(nrb));
1832                         rec_off += 4;
1833
1834                         memcpy(rec_data + rec_off, &(sa4->sin_addr.s_addr), 4);
1835                         rec_off += 4;
1836
1837                         memcpy(rec_data + rec_off, ai->ai_canonname, namelen);
1838                         rec_off += namelen;
1839
1840                         memset(rec_data + rec_off, 0, PADDING4(namelen));
1841                         rec_off += PADDING4(namelen);
1842                         pcapng_debug1("NRB: added IPv4 record for %s", ai->ai_canonname);
1843                 } else if (ai->ai_family == AF_INET6) {
1844                         nrb.record_type = NRES_IP6RECORD;
1845                         nrb.record_len = 16 + namelen;
1846                         tot_rec_len = 4 + nrb.record_len + PADDING4(nrb.record_len);
1847                         bh.block_total_length += tot_rec_len;
1848
1849                         if (rec_off + tot_rec_len > NRES_REC_MAX_SIZE)
1850                                 break;
1851
1852                         /*
1853                          * The joys of BSD sockaddrs.  In practice, this
1854                          * cast is alignment-safe.
1855                          */
1856                         sa6 = (struct sockaddr_in6 *)(void *)ai->ai_addr;
1857                         memcpy(rec_data + rec_off, &nrb, sizeof(nrb));
1858                         rec_off += 4;
1859
1860                         memcpy(rec_data + rec_off, sa6->sin6_addr.s6_addr, 16);
1861                         rec_off += 16;
1862
1863                         memcpy(rec_data + rec_off, ai->ai_canonname, namelen);
1864                         rec_off += namelen;
1865
1866                         memset(rec_data + rec_off, 0, PADDING4(namelen));
1867                         rec_off += PADDING4(namelen);
1868                         pcapng_debug1("NRB: added IPv6 record for %s", ai->ai_canonname);
1869                 }
1870         }
1871
1872         /* We know the total length now; copy the block header. */
1873         memcpy(rec_data, &bh, sizeof(bh));
1874
1875         /* End of record */
1876         memset(rec_data + rec_off, 0, 4);
1877         rec_off += 4;
1878
1879         memcpy(rec_data + rec_off, &bh.block_total_length, sizeof(bh.block_total_length));
1880
1881         if (!wtap_dump_file_write(wdh, rec_data, bh.block_total_length, err)) {
1882                 g_free(rec_data);
1883                 return FALSE;
1884         }
1885
1886         g_free(rec_data);
1887         wdh->bytes_dumped += bh.block_total_length;
1888         return TRUE;
1889 }
1890
1891
1892 static gboolean
1893 pcapng_write_block(wtap_dumper *wdh, /*pcapng_t *pn, */wtapng_block_t *wblock, int *err)
1894 {
1895         switch(wblock->type) {
1896             case(BLOCK_TYPE_SHB):
1897                 return pcapng_write_section_header_block(wdh, wblock, err);
1898             case(BLOCK_TYPE_IDB):
1899                 return pcapng_write_if_descr_block(wdh, wblock, err);
1900             case(BLOCK_TYPE_PB):
1901                 /* Packet Block is obsolete */
1902                 return FALSE;
1903             case(BLOCK_TYPE_EPB):
1904                 return pcapng_write_packet_block(wdh, wblock, err);
1905             default:
1906                 pcapng_debug1("Unknown block_type: 0x%x", wblock->type);
1907                 return FALSE;
1908         }
1909 }
1910
1911
1912 static guint32
1913 pcapng_lookup_interface_id_by_encap(int wtap_encap, wtap_dumper *wdh)
1914 {
1915         gint i;
1916         interface_data_t int_data;
1917         pcapng_dump_t *pcapng = (pcapng_dump_t *)wdh->priv;
1918
1919         for(i = 0; i < (gint)pcapng->number_of_interfaces; i++) {
1920                 int_data = g_array_index(pcapng->interface_data, interface_data_t, i);
1921                 if (wtap_encap == int_data.wtap_encap) {
1922                         return (guint32)i;
1923                 }
1924         }
1925         return G_MAXUINT32;
1926 }
1927
1928
1929 static gboolean pcapng_dump(wtap_dumper *wdh,
1930         const struct wtap_pkthdr *phdr,
1931         const union wtap_pseudo_header *pseudo_header,
1932         const guint8 *pd, int *err)
1933 {
1934         wtapng_block_t wblock;
1935         interface_data_t int_data;
1936         guint32 interface_id;
1937         guint64 ts;
1938         pcapng_dump_t *pcapng = (pcapng_dump_t *)wdh->priv;
1939         int pcap_encap;
1940
1941         pcapng_debug2("pcapng_dump: encap = %d (%s)",
1942                       phdr->pkt_encap,
1943                       wtap_encap_string(phdr->pkt_encap));
1944
1945         if (!pcapng->addrinfo_list_last)
1946                 pcapng->addrinfo_list_last = wdh->addrinfo_list;
1947
1948         interface_id = pcapng_lookup_interface_id_by_encap(phdr->pkt_encap, wdh);
1949         if (interface_id == G_MAXUINT32) {
1950                 /*
1951                  * We haven't yet written out an interface description
1952                  * block for an interface with this encapsulation.
1953                  *
1954                  * Is this encapsulation even supported in pcap-ng?
1955                  */
1956                 pcap_encap = wtap_wtap_encap_to_pcap_encap(phdr->pkt_encap);
1957                 if (pcap_encap == -1) {
1958                         /*
1959                          * No.  Fail.
1960                          */
1961                         *err = WTAP_ERR_UNSUPPORTED_ENCAP;
1962                         return FALSE;
1963                 }
1964
1965                 /* write the interface description block */
1966                 wblock.frame_buffer            = NULL;
1967                 wblock.pseudo_header           = NULL;
1968                 wblock.packet_header           = NULL;
1969                 wblock.file_encap              = NULL;
1970                 wblock.type                    = BLOCK_TYPE_IDB;
1971                 wblock.data.if_descr.link_type = pcap_encap;
1972                 wblock.data.if_descr.snap_len = (wdh->snaplen != 0) ? wdh->snaplen :
1973                                                                       WTAP_MAX_PACKET_SIZE; /* XXX */
1974
1975                 /* XXX - options unused */
1976                 wblock.data.if_descr.if_speed   = -1;
1977                 wblock.data.if_descr.if_tsresol = 6;    /* default: usec */
1978                 wblock.data.if_descr.if_os      = NULL;
1979                 wblock.data.if_descr.if_fcslen  = -1;
1980
1981                 if (!pcapng_write_block(wdh, &wblock, err)) {
1982                         return FALSE;
1983                 }
1984
1985                 interface_id = pcapng->number_of_interfaces;
1986                 int_data.wtap_encap = phdr->pkt_encap;
1987                 int_data.time_units_per_second = 0;
1988                 g_array_append_val(pcapng->interface_data, int_data);
1989                 pcapng->number_of_interfaces++;
1990
1991                 pcapng_debug3("pcapng_dump: added interface description block with index %u for encap = %d (%s).",
1992                               interface_id,
1993                               phdr->pkt_encap,
1994                               wtap_encap_string(phdr->pkt_encap));
1995         }
1996
1997         /* Flush any hostname resolution info we may have */
1998         while (pcapng->addrinfo_list_last && pcapng->addrinfo_list_last->ai_next) {
1999                 pcapng_write_name_resolution_block(wdh, pcapng, err);
2000         }
2001
2002         wblock.frame_buffer  = pd;
2003         wblock.pseudo_header = pseudo_header;
2004         wblock.packet_header = NULL;
2005         wblock.file_encap    = NULL;
2006
2007         /* write the (enhanced) packet block */
2008         wblock.type = BLOCK_TYPE_EPB;
2009
2010         /* default is to write out in microsecond resolution */
2011         ts = (((guint64)phdr->ts.secs) * 1000000) + (phdr->ts.nsecs / 1000);
2012
2013         /* Split the 64-bit timestamp into two 32-bit pieces */
2014         wblock.data.packet.ts_high      = (guint32)(ts >> 32);
2015         wblock.data.packet.ts_low       = (guint32)ts;
2016
2017         wblock.data.packet.cap_len      = phdr->caplen;
2018         wblock.data.packet.packet_len   = phdr->len;
2019         wblock.data.packet.interface_id = interface_id;
2020         wblock.data.packet.wtap_encap   = phdr->pkt_encap;
2021
2022         /* currently unused */
2023         wblock.data.packet.drop_count   = -1;
2024         wblock.data.packet.opt_comment  = NULL;
2025
2026         if (!pcapng_write_block(wdh, &wblock, err)) {
2027                 return FALSE;
2028         }
2029
2030         return TRUE;
2031 }
2032
2033
2034 /* Finish writing to a dump file.
2035    Returns TRUE on success, FALSE on failure. */
2036 static gboolean pcapng_dump_close(wtap_dumper *wdh, int *err _U_)
2037 {
2038         pcapng_dump_t *pcapng = (pcapng_dump_t *)wdh->priv;
2039
2040         pcapng_debug0("pcapng_dump_close");
2041         g_array_free(pcapng->interface_data, TRUE);
2042         pcapng->number_of_interfaces = 0;
2043         return TRUE;
2044 }
2045
2046
2047 /* Returns TRUE on success, FALSE on failure; sets "*err" to an error code on
2048    failure */
2049 gboolean
2050 pcapng_dump_open(wtap_dumper *wdh, int *err)
2051 {
2052         wtapng_block_t wblock;
2053         pcapng_dump_t *pcapng;
2054
2055         wblock.frame_buffer  = NULL;
2056         wblock.pseudo_header = NULL;
2057         wblock.packet_header = NULL;
2058         wblock.file_encap    = NULL;
2059
2060         pcapng_debug0("pcapng_dump_open");
2061         /* This is a pcapng file */
2062         wdh->subtype_write = pcapng_dump;
2063         wdh->subtype_close = pcapng_dump_close;
2064         pcapng = (pcapng_dump_t *)g_malloc0(sizeof(pcapng_dump_t));
2065         wdh->priv = (void *)pcapng;
2066         pcapng->interface_data = g_array_new(FALSE, FALSE, sizeof(interface_data_t));
2067
2068         /* write the section header block */
2069         wblock.type = BLOCK_TYPE_SHB;
2070         wblock.data.section.section_length = -1;
2071
2072         /* XXX - options unused */
2073         wblock.data.section.opt_comment   = NULL;
2074         wblock.data.section.shb_hardware  = NULL;
2075         wblock.data.section.shb_os        = NULL;
2076         wblock.data.section.shb_user_appl = NULL;
2077
2078         if (!pcapng_write_block(wdh, &wblock, err)) {
2079                 return FALSE;
2080         }
2081         pcapng_debug0("pcapng_dump_open: wrote section header block.");
2082
2083         return TRUE;
2084 }
2085
2086
2087 /* Returns 0 if we could write the specified encapsulation type,
2088    an error indication otherwise. */
2089 int pcapng_dump_can_write_encap(int wtap_encap)
2090 {
2091         pcapng_debug2("pcapng_dump_can_write_encap: encap = %d (%s)",
2092                       wtap_encap,
2093                       wtap_encap_string(wtap_encap));
2094
2095         /* Per-packet encapsulations is supported. */
2096         if (wtap_encap == WTAP_ENCAP_PER_PACKET)
2097                 return 0;
2098
2099         /* Make sure we can figure out this DLT type */
2100         if (wtap_wtap_encap_to_pcap_encap(wtap_encap) == -1)
2101                 return WTAP_ERR_UNSUPPORTED_ENCAP;
2102
2103         return 0;
2104 }