- struct idb idb;
- struct option option;
- guint32 block_total_length;
- const guint32 padding = 0;
- gboolean have_options = FALSE;
-
- block_total_length = sizeof(struct idb) + sizeof(guint32);
- /* 01 - OPT_COMMENT */
- if ((comment != NULL) && (strlen(comment) > 0) && (strlen(comment) < G_MAXUINT16)) {
- block_total_length += sizeof(struct option) +
- (guint16)(ADD_PADDING(strlen(comment) + 1));
- have_options = TRUE;
- }
-
- /* 02 - IDB_NAME */
- if ((name != NULL) && (strlen(name) > 0) && (strlen(name) < G_MAXUINT16)) {
- block_total_length += sizeof(struct option) +
- (guint16)(ADD_PADDING(strlen(name) + 1));
- have_options = TRUE;
- }
-
- /* 03 - IDB_DESCRIPTION */
- if ((descr != NULL) && (strlen(descr) > 0) && (strlen(descr) < G_MAXUINT16)) {
- block_total_length += sizeof(struct option) +
- (guint16)(ADD_PADDING(strlen(descr) + 1));
- have_options = TRUE;
- }
-
- /* 08 - IDB_IF_SPEED */
- if (if_speed != 0) {
- block_total_length += sizeof(struct option) + sizeof(guint64);
- have_options = TRUE;
- }
-
- /* 09 - IDB_TSRESOL */
- if (tsresol != 0) {
- block_total_length += sizeof(struct option) + sizeof(struct option);
- have_options = TRUE;
- }
-
- /* 11 - IDB_FILTER */
- if ((filter != NULL) && (strlen(filter) > 0) && (strlen(filter) < G_MAXUINT16)) {
- block_total_length += sizeof(struct option) +
- (guint16)(ADD_PADDING(strlen(filter) + 1 + 1));
- have_options = TRUE;
- }
-
- /* 12 - IDB_OS */
- if ((os != NULL) && (strlen(os) > 0) && (strlen(os) < G_MAXUINT16)) {
- block_total_length += sizeof(struct option) +
- (guint16)(ADD_PADDING(strlen(os) + 1));
- have_options = TRUE;
- }
-
- /* If we have options add size of end-of-options */
- if (have_options) {
- block_total_length += sizeof(struct option);
- }
-
- /* write block header */
- idb.block_type = INTERFACE_DESCRIPTION_BLOCK_TYPE;
- idb.block_total_length = block_total_length;
- idb.link_type = link_type;
- idb.reserved = 0;
- idb.snap_len = snap_len;
- WRITE_DATA(fp, &idb, sizeof(struct idb), *bytes_written, err);
-
- /* 01 - OPT_COMMENT - write comment string if applicable */
- if ((comment != NULL) && (strlen(comment) > 0) && (strlen(comment) < G_MAXUINT16)) {
- option.type = OPT_COMMENT;
- option.value_length = (guint16)(strlen(comment) + 1);
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- WRITE_DATA(fp, comment, strlen(comment) + 1, *bytes_written, err);
- if ((strlen(comment) + 1) % 4) {
- WRITE_DATA(fp, &padding, 4 - (strlen(comment) + 1) % 4 , *bytes_written, err);
- }
- }
-
- /* 02 - IDB_NAME - write interface name string if applicable */
- if ((name != NULL) && (strlen(name) > 0) && (strlen(name) < G_MAXUINT16)) {
- option.type = IDB_NAME;
- option.value_length = (guint16)(strlen(name) + 1);
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- WRITE_DATA(fp, name, strlen(name) + 1, *bytes_written, err);
- if ((strlen(name) + 1) % 4) {
- WRITE_DATA(fp, &padding, 4 - (strlen(name) + 1) % 4 , *bytes_written, err);
- }
- }
-
- /* 03 - IDB_DESCRIPTION */
- /* write interface description string if applicable */
- if ((descr != NULL) && (strlen(descr) > 0) && (strlen(descr) < G_MAXUINT16)) {
- option.type = IDB_DESCRIPTION;
- option.value_length = (guint16)(strlen(descr) + 1);
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- WRITE_DATA(fp, name, strlen(descr) + 1, *bytes_written, err);
- if ((strlen(descr) + 1) % 4) {
- WRITE_DATA(fp, &padding, 4 - (strlen(descr) + 1) % 4 , *bytes_written, err);
- }
- }
-
- /* 08 - IDB_IF_SPEED */
- if (if_speed != 0) {
- option.type = IDB_IF_SPEED;
- option.value_length = sizeof(guint64);
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- WRITE_DATA(fp, &if_speed, sizeof(guint64), *bytes_written, err);
- }
-
- /* 09 - IDB_TSRESOL */
- if (tsresol != 0) {
- option.type = IDB_TSRESOL;
- option.value_length = sizeof(guint8);
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- WRITE_DATA(fp, &tsresol, sizeof(guint8), *bytes_written, err);
- WRITE_DATA(fp, &padding, 3 , *bytes_written, err);
- }
-
- /* 11 - IDB_FILTER - write filter string if applicable
- * We only write version 1 of the filter, libpcap string
- */
- if ((filter != NULL) && (strlen(filter) > 0) && (strlen(filter) < G_MAXUINT16)) {
- option.type = IDB_FILTER;
- option.value_length = (guint16)(strlen(filter) + 1 + 1 );
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
-
- /* The first byte of the Option Data keeps a code of the filter used, 0 = lipbpcap filter string */
- WRITE_DATA(fp, &padding, 1, *bytes_written, err);
- WRITE_DATA(fp, filter, strlen(filter) + 1, *bytes_written, err);
- if ((strlen(filter) + 2) % 4) {
- WRITE_DATA(fp, &padding, 4 - (strlen(filter) + 2) % 4 , *bytes_written, err);
- }
- }
-
- /* 12 - IDB_OS - write os string if applicable */
- if ((os != NULL) && (strlen(os) > 0) && (strlen(os) < G_MAXUINT16)) {
- option.type = IDB_OS;
- option.value_length = (guint16)(strlen(os) + 1);
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- WRITE_DATA(fp, os, strlen(os) + 1, *bytes_written, err);
- if ((strlen(os) + 1) % 4) {
- WRITE_DATA(fp, &padding, 4 - (strlen(os) + 1) % 4 , *bytes_written, err);
- }
- }
-
- if (have_options) {
- /* write end of options */
- option.type = OPT_ENDOFOPT;
- option.value_length = 0;
- WRITE_DATA(fp, &option, sizeof(struct option), *bytes_written, err);
- }
+ struct shb shb;
+ struct option option;
+ guint32 block_total_length;
+ guint32 options_length;
+
+ /* Size of base header */
+ block_total_length = sizeof(struct shb) +
+ sizeof(guint32);
+ options_length = 0;
+ options_length += pcapng_count_string_option(comment);
+ options_length += pcapng_count_string_option(hw);
+ options_length += pcapng_count_string_option(os);
+ options_length += pcapng_count_string_option(appname);
+ /* If we have options add size of end-of-options */
+ if (options_length != 0) {
+ options_length += (guint32)sizeof(struct option);
+ }
+ block_total_length += options_length;
+
+ /* write shb header */
+ shb.block_type = SECTION_HEADER_BLOCK_TYPE;
+ shb.block_total_length = block_total_length;
+ shb.byte_order_magic = PCAPNG_MAGIC;
+ shb.major_version = PCAPNG_MAJOR_VERSION;
+ shb.minor_version = PCAPNG_MINOR_VERSION;
+ shb.section_length = section_length;
+
+ if (!write_to_file(pfile, (const guint8*)&shb, sizeof(struct shb), bytes_written, err))
+ return FALSE;
+
+ if (!pcapng_write_string_option(pfile, OPT_COMMENT, comment,
+ bytes_written, err))
+ return FALSE;
+ if (!pcapng_write_string_option(pfile, SHB_HARDWARE, hw,
+ bytes_written, err))
+ return FALSE;
+ if (!pcapng_write_string_option(pfile, SHB_OS, os,
+ bytes_written, err))
+ return FALSE;
+ if (!pcapng_write_string_option(pfile, SHB_USERAPPL, appname,
+ bytes_written, err))
+ return FALSE;
+ if (options_length != 0) {
+ /* write end of options */
+ option.type = OPT_ENDOFOPT;
+ option.value_length = 0;
+ if (!write_to_file(pfile, (const guint8*)&option, sizeof(struct option), bytes_written, err))
+ return FALSE;
+ }
+
+ /* write the trailing block total length */
+ return write_to_file(pfile, (const guint8*)&block_total_length, sizeof(guint32), bytes_written, err);
+}