segment-fetcher.hpp

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

#include "../face.hpp"
#include "../security/key-chain.hpp"

namespace ndn {

class SegmentFetcher : public ptr_lib::enable_shared_from_this<SegmentFetcher> {
public:
  enum ErrorCode {
    INTEREST_TIMEOUT = 1,
    DATA_HAS_NO_SEGMENT = 2,
    SEGMENT_VERIFICATION_FAILED = 3
  };

  typedef func_lib::function<bool(const ptr_lib::shared_ptr<Data>& data)> VerifySegment;

  typedef func_lib::function<void(const Blob& content)> OnComplete;

  typedef func_lib::function<void
    (ErrorCode errorCode, const std::string& message)> OnError;

  static bool
  DontVerifySegment(const ptr_lib::shared_ptr<Data>& data);

  static void
  fetch
    (Face& face, const Interest &baseInterest, const VerifySegment& verifySegment,
     const OnComplete& onComplete, const OnError& onError);

  static void
  fetch
    (Face& face, const Interest &baseInterest, KeyChain* validatorKeyChain,
     const OnComplete& onComplete, const OnError& onError);

private:
  SegmentFetcher
    (Face& face, KeyChain* validatorKeyChain, const VerifySegment& verifySegment,
     const OnComplete& onComplete, const OnError& onError)
  : face_(face), validatorKeyChain_(validatorKeyChain), verifySegment_(verifySegment),
    onComplete_(onComplete), onError_(onError)
  {
  }

  void
  fetchFirstSegment(const Interest& baseInterest);

  void
  fetchNextSegment
    (const Interest& originalInterest, const Name& dataName, uint64_t segment);

  void
  onSegmentReceived
    (const ptr_lib::shared_ptr<const Interest>& originalInterest,
     const ptr_lib::shared_ptr<Data>& data);

  void
  onVerified
    (const ptr_lib::shared_ptr<Data>& data,
     const ptr_lib::shared_ptr<const Interest>& originalInterest);

  void
  onValidationFailed
    (const ptr_lib::shared_ptr<Data>& data, const std::string& reason);

  void
  onTimeout(const ptr_lib::shared_ptr<const Interest>& interest);

  static bool
  endsWithSegmentNumber(const Name& name)
  {
    return name.size() >= 1 && name.get(-1).isSegment();
  }

  std::vector<Blob> contentParts_;
  Face& face_;
  KeyChain* validatorKeyChain_;
  VerifySegment verifySegment_;
  OnComplete onComplete_;
  OnError onError_;
};

}

#endif