back-end-mem.cpp

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
#include "back-end-mem.hpp"
#include "key-handle-mem.hpp"
#include "../transform/private-key.hpp"
#include "../../encoding/buffer-stream.hpp"
#include <unordered_map>

namespace ndn {
namespace security {
namespace tpm {

using transform::PrivateKey;

class BackEndMem::Impl
{
public:
  std::unordered_map<Name, shared_ptr<PrivateKey>> keys;
};

BackEndMem::BackEndMem(const std::string&)
  : m_impl(new Impl)
{
}

BackEndMem::~BackEndMem() = default;

const std::string&
BackEndMem::getScheme()
{
  static std::string scheme = "tpm-memory";
  return scheme;
}

bool
BackEndMem::doHasKey(const Name& keyName) const
{
  return (m_impl->keys.count(keyName) > 0);
}

unique_ptr<KeyHandle>
BackEndMem::doGetKeyHandle(const Name& keyName) const
{
  auto it = m_impl->keys.find(keyName);
  if (it == m_impl->keys.end())
    return nullptr;
  return make_unique<KeyHandleMem>(it->second);
}

unique_ptr<KeyHandle>
BackEndMem::doCreateKey(const Name& identityName, const KeyParams& params)
{
  shared_ptr<PrivateKey> key(transform::generatePrivateKey(params).release());
  unique_ptr<KeyHandle> keyHandle = make_unique<KeyHandleMem>(key);

  setKeyName(*keyHandle, identityName, params);

  m_impl->keys[keyHandle->getKeyName()] = key;
  return keyHandle;
}

void
BackEndMem::doDeleteKey(const Name& keyName)
{
  m_impl->keys.erase(keyName);
}

ConstBufferPtr
BackEndMem::doExportKey(const Name& keyName, const char* pw, size_t pwLen)
{
  OBufferStream os;
  m_impl->keys[keyName]->savePkcs8(os, pw, pwLen);
  return os.buf();
}

void
BackEndMem::doImportKey(const Name& keyName, const uint8_t* buf, size_t size, const char* pw, size_t pwLen)
{
  try {
    auto key = make_shared<PrivateKey>();
    key->loadPkcs8(buf, size, pw, pwLen);
    m_impl->keys[keyName] = key;
  }
  catch (const PrivateKey::Error& e) {
    BOOST_THROW_EXCEPTION(Error(std::string("Cannot import private key: ") + e.what()));
  }
}

} // namespace tpm
} // namespace security
} // namespace ndn