certificate-cache.hpp

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

#include <map>
#include "../certificate/identity-certificate.hpp"

namespace ndn {

class CertificateCache {
public:
  void
  insertCertificate(const IdentityCertificate& certificate)
  {
    Name certName = certificate.getName().getPrefix(-1);
    cache_[certName.toUri()] = certificate.wireEncode();
  }

  void
  deleteCertificate(const Name& certificateName)
  {
    std::map<std::string, Blob>::iterator entry
      (cache_.find(certificateName.toUri()));
    if (entry != cache_.end())
      cache_.erase(entry);
  }

  ptr_lib::shared_ptr<IdentityCertificate>
  getCertificate(const Name& certificateName) const
  {
    std::map<std::string, Blob>::const_iterator entry
      (cache_.find(certificateName.toUri()));
    if (entry == cache_.end())
      return ptr_lib::shared_ptr<IdentityCertificate>();

    ptr_lib::shared_ptr<IdentityCertificate> cert(new IdentityCertificate());
    Blob certData = entry->second;
    cert->wireDecode(certData);
    return cert;
  }

  void
  reset()
  {
    cache_.clear();
  }

private:
  // The key is the certificate name URI. The value is the wire encoding.
  std::map<std::string, Blob> cache_;
};

}

#endif