key-params.hpp

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

#include <stdint.h>
#include "../name.hpp"
#include "security-common.hpp"

namespace ndn {

class KeyParams {
public:
  virtual
  ~KeyParams()
  {
  }

  KeyType
  getKeyType() const { return keyType_; }

  KeyIdType
  getKeyIdType() const { return keyIdType_; }

  void
  setKeyId(const Name::Component& keyId) { keyId_ = keyId; }

  const Name::Component&
  getKeyId() const { return keyId_; }

protected:
  KeyParams(KeyType keyType, KeyIdType keyIdType);

  KeyParams(KeyType keyType, const Name::Component& keyId);

private:
  KeyType keyType_;
  KeyIdType keyIdType_;
  Name::Component keyId_;
};

class RsaKeyParams : public KeyParams {
public:
  RsaKeyParams
    (const Name::Component& keyId,
     uint32_t size = RsaKeyParams::getDefaultSize())
  : KeyParams(RsaKeyParams::getType(), keyId),
    size_(size)
  {
  }

  RsaKeyParams
    (uint32_t size = RsaKeyParams::getDefaultSize(),
     KeyIdType keyIdType = KEY_ID_TYPE_RANDOM)
  : KeyParams(RsaKeyParams::getType(), keyIdType),
    size_(size)
  {
  }

  uint32_t
  getKeySize() const
  {
    return size_;
  }

  static uint32_t
  getDefaultSize() { return 2048; }

  static KeyType
  getType() { return KEY_TYPE_RSA; }

private:
  uint32_t size_;
};

class EcKeyParams : public KeyParams {
public:
  EcKeyParams
    (const Name::Component& keyId,
     uint32_t size = EcKeyParams::getDefaultSize())
  : KeyParams(EcKeyParams::getType(), keyId),
    size_(size)
  {
  }

  EcKeyParams
    (uint32_t size = EcKeyParams::getDefaultSize(),
     KeyIdType keyIdType = KEY_ID_TYPE_RANDOM)
  : KeyParams(EcKeyParams::getType(), keyIdType),
    size_(size)
  {
  }

  uint32_t
  getKeySize() const
  {
    return size_;
  }

  static uint32_t
  getDefaultSize() { return 256; }

  static KeyType
  getType() { return KEY_TYPE_EC; }

private:
  uint32_t size_;
};


class EcdsaKeyParams : public EcKeyParams {
public:
  EcdsaKeyParams
    (const Name::Component& keyId,
     uint32_t size = EcKeyParams::getDefaultSize())
  : EcKeyParams(keyId, size)
  {
  }

  EcdsaKeyParams
    (uint32_t size = EcKeyParams::getDefaultSize(),
     KeyIdType keyIdType = KEY_ID_TYPE_RANDOM)
  : EcKeyParams(size, keyIdType)
  {
  }
};

class AesKeyParams : public KeyParams {
public:
  AesKeyParams
    (const Name::Component& keyId,
     uint32_t size = AesKeyParams::getDefaultSize())
  : KeyParams(AesKeyParams::getType(), keyId),
    size_(size)
  {
  }

  AesKeyParams
    (uint32_t size = AesKeyParams::getDefaultSize(),
     KeyIdType keyIdType = KEY_ID_TYPE_RANDOM)
  : KeyParams(AesKeyParams::getType(), keyIdType),
    size_(size)
  {
  }

  uint32_t
  getKeySize() const
  {
    return size_;
  }

  static uint32_t
  getDefaultSize() { return 64; }

  static KeyType
  getType() { return KEY_TYPE_AES; }

private:
  uint32_t size_;
};

}

#endif