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"

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>&)> VerifySegment;

  typedef func_lib::function<void(const Blob&)> 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);

private:
  SegmentFetcher
    (Face& face, const VerifySegment& verifySegment, const OnComplete& onComplete,
     const OnError& onError)
  : face_(face), 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
  onTimeout(const ptr_lib::shared_ptr<const Interest>& interest);

  static bool
  endsWithSegmentNumber(Name name)
  {
    return name.size() >= 1 &&
           name.get(-1).getValue().size() >= 1 &&
           name.get(-1).getValue().buf()[0] == 0;
  }

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

}

#endif