ndn-ns3.cpp

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
#include "ndn-ns3.hpp"

#include <ndn-cxx/encoding/block.hpp>
#include <ndn-cxx/interest.hpp>
#include <ndn-cxx/data.hpp>

#include "ndn-header.hpp"
#include "../utils/ndn-ns3-packet-tag.hpp"

namespace ns3 {
namespace ndn {

template<class T>
std::shared_ptr<const T>
Convert::FromPacket(Ptr<Packet> packet)
{
  PacketHeader<T> header;
  packet->RemoveHeader(header);

  auto pkt = header.getPacket();
  pkt->setTag(make_shared<Ns3PacketTag>(packet));

  return pkt;
}

template std::shared_ptr<const Interest>
Convert::FromPacket<Interest>(Ptr<Packet> packet);

template std::shared_ptr<const Data>
Convert::FromPacket<Data>(Ptr<Packet> packet);

template<class T>
Ptr<Packet>
Convert::ToPacket(const T& pkt)
{
  PacketHeader<T> header(pkt);

  Ptr<Packet> packet;

  auto tag = pkt.template getTag<Ns3PacketTag>();
  if (tag != nullptr) {
    packet = tag->getPacket()->Copy();
  }
  else {
    packet = Create<Packet>();
  }

  packet->AddHeader(header);
  return packet;
}

template Ptr<Packet>
Convert::ToPacket<Interest>(const Interest& packet);

template Ptr<Packet>
Convert::ToPacket<Data>(const Data& packet);

uint32_t
Convert::getPacketType(Ptr<const Packet> packet)
{
  uint8_t type;
  uint32_t nRead = packet->CopyData(&type, 1);
  if (nRead != 1) {
    throw ::ndn::tlv::Error("Unknown header");
  }

  if (type == ::ndn::tlv::Interest || type == ::ndn::tlv::Data) {
    return type;
  }
  else {
    throw ::ndn::tlv::Error("Unknown header");
  }
}

} // namespace ndn
} // namespace ns3