validation-policy-command-interest.hpp

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

#include <map>
#include "validation-policy.hpp"

namespace ndn {

class ValidationPolicyCommandInterest : public ValidationPolicy
{
public:
  class Options {
  public:
    Options
      (Milliseconds gracePeriod = 2 * 60 * 1000.0, int maxRecords = 1000,
       Milliseconds recordLifetime = 3600 * 1000.0)
    : gracePeriod_(gracePeriod),
      maxRecords_(maxRecords),
      recordLifetime_(recordLifetime)
    {
    }

    Milliseconds gracePeriod_;

    int maxRecords_;

    Milliseconds recordLifetime_;
  };

  ValidationPolicyCommandInterest
    (const ptr_lib::shared_ptr<ValidationPolicy>& innerPolicy,
     const Options& options = Options());

  virtual void
  checkPolicy
    (const Data& data, const ptr_lib::shared_ptr<ValidationState>& state,
     const ValidationContinuation& continueValidation);

  virtual void
  checkPolicy
    (const Interest& interest, const ptr_lib::shared_ptr<ValidationState>& state,
     const ValidationContinuation& continueValidation);

  void
  setNowOffsetMilliseconds_(Milliseconds nowOffsetMilliseconds)
  {
    nowOffsetMilliseconds_ = nowOffsetMilliseconds;
  }

private:
  class LastTimestampRecord
  {
  public:
    LastTimestampRecord
      (const Name& keyName, MillisecondsSince1970 timestamp,
       MillisecondsSince1970 lastRefreshed)
    : keyName_(keyName), timestamp_(timestamp), lastRefreshed_(lastRefreshed)
    {}

    Name keyName_;
    MillisecondsSince1970 timestamp_;
    MillisecondsSince1970 lastRefreshed_;
  };

  void
  cleanUp();

  static bool
  parseCommandInterest
    (const Interest& interest, const ptr_lib::shared_ptr<ValidationState>& state,
     Name& keyLocatorName, MillisecondsSince1970& timestamp);

  bool
  checkTimestamp
    (const ptr_lib::shared_ptr<ValidationState>& state, const Name& keyName,
     MillisecondsSince1970 timestamp);

  void
  insertNewRecord
    (const Interest& interest, const Name& keyName,
     MillisecondsSince1970 timestamp);

  int
  findByKeyName(const Name& keyName);

  Options options_;
  std::vector<ptr_lib::shared_ptr<LastTimestampRecord> > container_;
  Milliseconds nowOffsetMilliseconds_;
};

}

#endif