validator-fixture.hpp

#ifndef NDN_VALIDATOR_FIXTURE_HPP
#define NDN_VALIDATOR_FIXTURE_HPP

#include <ndn-cpp/security/v2/validator.hpp>
#include <ndn-cpp/security/v2/certificate-fetcher-from-network.hpp>
#include "identity-management-fixture.hpp"

class ValidatorFixture : public IdentityManagementFixture
{
public:
  ValidatorFixture(const ndn::ptr_lib::shared_ptr<ndn::ValidationPolicy>& policy);

  class TestFace : public ndn::Face {
  public:
    typedef ndn::func_lib::function<void
      (const ndn::Interest& interest, const ndn::OnData& onData,
       const ndn::OnTimeout& onTimeout, const ndn::OnNetworkNack& onNetworkNack)>
      ProcessInterest;

    TestFace()
    : Face("localhost")
    {}

    virtual uint64_t
    expressInterest
      (const ndn::Interest& interest, const ndn::OnData& onData,
       const ndn::OnTimeout& onTimeout, const ndn::OnNetworkNack& onNetworkNack,
       ndn::WireFormat& wireFormat = *ndn::WireFormat::getDefaultWireFormat());

    ProcessInterest processInterest_;
    std::vector<ndn::Interest> sentInterests_;
  };

  void
  processInterestFromCache
    (const ndn::Interest& interest, const ndn::OnData& onData,
     const ndn::OnTimeout& onTimeout, const ndn::OnNetworkNack& onNetworkNack);

  TestFace face_;
  ndn::Validator validator_;
  ndn::ptr_lib::shared_ptr<ndn::ValidationPolicy> policy_;
  ndn::CertificateCacheV2 cache_;
};

class HierarchicalValidatorFixture : public ValidatorFixture {
public:
  HierarchicalValidatorFixture
    (const ndn::ptr_lib::shared_ptr<ndn::ValidationPolicy>& policy)
  : ValidatorFixture(policy)
  {
    identity_ = addIdentity("/Security/V2/ValidatorFixture");
    subIdentity_ = addSubCertificate("/Security/V2/ValidatorFixture/Sub1", identity_);
    subSelfSignedIdentity_ = addIdentity("/Security/V2/ValidatorFixture/Sub1/Sub2");
    otherIdentity_ = addIdentity("/Security/V2/OtherIdentity");

    validator_.loadAnchor
      ("", ndn::CertificateV2(*identity_->getDefaultKey()->getDefaultCertificate()));

    cache_.insert(*identity_->getDefaultKey()->getDefaultCertificate());
    cache_.insert(*subIdentity_->getDefaultKey()->getDefaultCertificate());
    cache_.insert(*subSelfSignedIdentity_->getDefaultKey()->getDefaultCertificate());
    cache_.insert(*otherIdentity_->getDefaultKey()->getDefaultCertificate());
  }

  ndn::ptr_lib::shared_ptr<ndn::PibIdentity> identity_;
  ndn::ptr_lib::shared_ptr<ndn::PibIdentity> subIdentity_;
  ndn::ptr_lib::shared_ptr<ndn::PibIdentity> subSelfSignedIdentity_;
  ndn::ptr_lib::shared_ptr<ndn::PibIdentity> otherIdentity_;
};

#endif