tlv-decoder.hpp

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
#ifndef NDN_TLV_DECODER_HPP
#define NDN_TLV_DECODER_HPP

#include <stdexcept>
#include "../c/encoding/tlv/tlv-decoder.h"

using namespace std;

namespace ndn {

class TlvDecoder : public ndn_TlvDecoder {
public:
  TlvDecoder(const uint8_t *input, size_t inputLength)
  {
    ndn_TlvDecoder_initialize(this, input, inputLength);
  }

  size_t
  readNestedTlvsStart(unsigned int expectedType)
  {
    size_t endOffset;
    ndn_Error error;
    if ((error = ndn_TlvDecoder_readNestedTlvsStart
         (this, expectedType, &endOffset)))
      throw runtime_error(ndn_getErrorString(error));

    return endOffset;
  }

  void
  finishNestedTlvs(int endOffset)
  {
    ndn_Error error;
    if ((error = ndn_TlvDecoder_finishNestedTlvs(this, endOffset)))
      throw runtime_error(ndn_getErrorString(error));
  }

  void
  finishNestedTlvsSkipCritical(int endOffset)
  {
    ndn_Error error;
    if ((error = ndn_TlvDecoder_finishNestedTlvsSkipCritical(this, endOffset)))
      throw runtime_error(ndn_getErrorString(error));
  }

  bool
  peekType(unsigned int expectedType, size_t endOffset)
  {
    int gotExpectedType;
    ndn_Error error;
    if ((error = ndn_TlvDecoder_peekType
         (this, expectedType, endOffset, &gotExpectedType)))
      throw runtime_error(ndn_getErrorString(error));

    return gotExpectedType != 0 ? true : false;
  }

  uint64_t
  readNonNegativeIntegerTlv(unsigned int expectedType)
  {
    uint64_t value;
    ndn_Error error;
    if ((error = ndn_TlvDecoder_readNonNegativeIntegerTlv
         (this, expectedType, &value)))
      throw runtime_error(ndn_getErrorString(error));

    return value;
  }

  struct ndn_Blob
  readBlobTlv(unsigned int expectedType)
  {
    struct ndn_Blob value;
    ndn_Error error;
    if ((error = ndn_TlvDecoder_readBlobTlv(this, expectedType, &value)))
      throw runtime_error(ndn_getErrorString(error));

    return value;
  }

  bool
  readBooleanTlv(unsigned int expectedType, size_t endOffset)
  {
    int value;
    ndn_Error error;
    if ((error = ndn_TlvDecoder_readBooleanTlv
         (this, expectedType, endOffset, &value)))
      throw runtime_error(ndn_getErrorString(error));

    return value != 0;
  }

  void
  skipTlv(unsigned int expectedType)
  {
    ndn_Error error;
    if ((error = ndn_TlvDecoder_skipTlv(this, expectedType)))
      throw runtime_error(ndn_getErrorString(error));
  }

  void
  skipOptionalTlv(unsigned int expectedType, size_t endOffset)
  {
    ndn_Error error;
    if ((error = ndn_TlvDecoder_skipOptionalTlv
         (this, expectedType, endOffset)))
      throw runtime_error(ndn_getErrorString(error));
  }

  void
  seek(size_t offset) { ndn_TlvDecoder_seek(this, offset); }
};

}

#endif