face.hpp

Go to the documentation of this file.

#ifndef NDNPH_FACE_FACE_HPP
#define NDNPH_FACE_FACE_HPP
 
#include "../packet/lp.hpp"
#include "transport.hpp"
 
namespace ndnph {
 
class PacketHandler;
 
class Face : public WithRegion {
public:
  struct PacketInfo {
    uint64_t endpointId = 0;
    lp::PitToken pitToken;
  };
 
  explicit Face(Region& region, Transport& transport)
    : WithRegion(region)
    , m_transport(transport) {
    m_transport.setRxCallback(transportRx, this);
  }
 
  explicit Face(Transport& transport)
    : Face(*new OwnRegion(), transport) {
    m_ownRegion.reset(static_cast<OwnRegion*>(&region));
  }
 
  Transport& getTransport() const {
    return m_transport;
  }
 
  void setFragmenter(lp::Fragmenter& frag) {
    m_frag = &frag;
  }
 
  void setReassembler(lp::Reassembler& reass) {
    m_reass = &reass;
  }
 
  bool addHandler(PacketHandler& h, int8_t prio = 0);
 
  bool removeHandler(PacketHandler& h);
 
  void loop();
 
  const PacketInfo* getCurrentPacketInfo() const {
    return m_currentPacketInfo;
  }
 
  template<typename Packet>
  bool send(Region& region, const Packet& packet, PacketInfo pi);
 
private:
  class ScopedCurrentPacketInfo {
  public:
    explicit ScopedCurrentPacketInfo(Face& face, PacketInfo& pi)
      : m_face(face) {
      m_face.m_currentPacketInfo = &pi;
    }
 
    ~ScopedCurrentPacketInfo() {
      m_face.m_currentPacketInfo = nullptr;
    }
 
  private:
    Face& m_face;
  };
 
  static void transportRx(void* self, const uint8_t* pkt, size_t pktLen, uint64_t endpointId) {
    reinterpret_cast<Face*>(self)->transportRx(pkt, pktLen, endpointId);
  }
 
  void transportRx(const uint8_t* pkt, size_t pktLen, uint64_t endpointId);
 
  template<typename Packet, typename H = bool (PacketHandler::*)(Packet)>
  bool process(H processPacket, Packet packet);
 
private:
  using OwnRegion = StaticRegion<2048>;
  std::unique_ptr<OwnRegion> m_ownRegion;
  Transport& m_transport;
  lp::Fragmenter* m_frag = nullptr;
  lp::Reassembler* m_reass = nullptr;
  PacketHandler* m_handler = nullptr;
  const PacketInfo* m_currentPacketInfo = nullptr;
};
 
} // namespace ndnph
 
#define NDNPH_FACE_FACE_HPP_END
#include "face-impl.inc"
#include "packet-handler.hpp"
 
#endif // NDNPH_FACE_FACE_HPP