forwarder-general-module.hpp

Go to the documentation of this file.

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

#include "module.hpp"
#include <ndn-cxx/security/validator.hpp>

namespace nfd {
namespace tools {
namespace nfdc {

using ndn::nfd::ForwarderStatus;

class NfdIdCollector;

class ForwarderGeneralModule : public Module, noncopyable
{
public:
  ForwarderGeneralModule();

  virtual void
  fetchStatus(Controller& controller,
              const function<void()>& onSuccess,
              const Controller::DatasetFailCallback& onFailure,
              const CommandOptions& options) override;

  void
  setNfdIdCollector(const NfdIdCollector& nfdIdCollector)
  {
    m_nfdIdCollector = &nfdIdCollector;
  }

  virtual void
  formatStatusXml(std::ostream& os) const override;

  void
  formatItemXml(std::ostream& os, const ForwarderStatus& item, const Name& nfdId) const;

  virtual void
  formatStatusText(std::ostream& os) const override;

  void
  formatItemText(std::ostream& os, const ForwarderStatus& item, const Name& nfdId) const;

private:
  const Name&
  getNfdId() const;

private:
  const NfdIdCollector* m_nfdIdCollector;
  ForwarderStatus m_status;
};


class NfdIdCollector : public ndn::Validator
{
public:
  explicit
  NfdIdCollector(unique_ptr<ndn::Validator> inner);

  bool
  hasNfdId() const
  {
    return m_hasNfdId;
  }

  const Name&
  getNfdId() const;

protected:
  virtual void
  checkPolicy(const Interest& interest, int nSteps,
              const ndn::OnInterestValidated& accept,
              const ndn::OnInterestValidationFailed& reject,
              std::vector<shared_ptr<ndn::ValidationRequest>>& nextSteps) override
  {
    BOOST_ASSERT(nSteps == 0);
    m_inner->validate(interest, accept, reject);
  }

  virtual void
  checkPolicy(const Data& data, int nSteps,
              const ndn::OnDataValidated& accept,
              const ndn::OnDataValidationFailed& reject,
              std::vector<shared_ptr<ndn::ValidationRequest>>& nextSteps) override;

private:
  unique_ptr<ndn::Validator> m_inner;
  bool m_hasNfdId;
  Name m_nfdId;
};

} // namespace nfdc
} // namespace tools
} // namespace nfd

#endif // NFD_TOOLS_NFDC_FORWARDER_GENERAL_MODULE_HPP