examples/streaming_parser.c - master - githubmirror/libcbor

examples/streaming_parser.c

72 строки2 KB

Pavel Kalvoda

Fix infinite loop in streaming_parser example on NEDATA; fix required doc (#414)

23 мар 2026, 02:18

23 мар 2026, 02:18e1f711a

/*
 * Copyright (c) 2014-2020 Pavel Kalvoda <me@pavelkalvoda.com>
 *
 * libcbor is free software; you can redistribute it and/or modify
 * it under the terms of the MIT license. See LICENSE for details.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "cbor.h"

void usage(void) {
  printf("Usage: streaming_parser [input file]\n");
  exit(1);
}

/*
 * Illustrates how one might skim through a map (which is assumed to have
 * string keys and values only), looking for the value of a specific key
 *
 * Use the examples/data/map.cbor input to test this.
 */

const char* key = "a secret key";
bool key_found = false;

void find_string(void* _ctx _CBOR_UNUSED, cbor_data buffer, uint64_t len) {
  if (key_found) {
    printf("Found the value: %.*s\n", (int)len, buffer);
    key_found = false;
  } else if (len == strlen(key)) {
    key_found = (memcmp(key, buffer, len) == 0);
  }
}

int main(int argc, char* argv[]) {
  if (argc != 2) usage();
  FILE* f = fopen(argv[1], "rb");
  if (f == NULL) usage();
  fseek(f, 0, SEEK_END);
  size_t length = (size_t)ftell(f);
  fseek(f, 0, SEEK_SET);
  unsigned char* buffer = malloc(length);
  if (fread(buffer, length, 1, f) != 1) {
    fprintf(stderr, "Failed to read input\n");
    exit(1);
  }

  struct cbor_callbacks callbacks = cbor_empty_callbacks;
  struct cbor_decoder_result decode_result;
  size_t bytes_read = 0;
  callbacks.string = find_string;
  while (bytes_read < length) {
    decode_result = cbor_stream_decode(buffer + bytes_read, length - bytes_read,
                                       &callbacks, NULL);
    if (decode_result.status == CBOR_DECODER_FINISHED) {
      bytes_read += decode_result.read;
    } else {
      // The input was fully loaded into memory, so NEDATA means truncated data.
      fprintf(stderr,
              decode_result.status == CBOR_DECODER_NEDATA
                  ? "Truncated data at byte %zu\n"
                  : "Error at byte %zu\n",
              bytes_read);
      break;
    }
  }

  free(buffer);
  fclose(f);
}