|
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
|
|
/**
|
|
* Copyright (c) 2014-2015, Regents of the University of California,
|
|
* Arizona Board of Regents,
|
|
* Colorado State University,
|
|
* University Pierre & Marie Curie, Sorbonne University,
|
|
* Washington University in St. Louis,
|
|
* Beijing Institute of Technology,
|
|
* The University of Memphis.
|
|
*
|
|
* This file is part of NFD (Named Data Networking Forwarding Daemon).
|
|
* See AUTHORS.md for complete list of NFD authors and contributors.
|
|
*
|
|
* NFD is free software: you can redistribute it and/or modify it under the terms
|
|
* of the GNU General Public License as published by the Free Software Foundation,
|
|
* either version 3 of the License, or (at your option) any later version.
|
|
*
|
|
* NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
|
|
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
* PURPOSE. See the GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "ndnlp-partial-message-store.hpp"
|
|
|
|
namespace nfd {
|
|
namespace ndnlp {
|
|
|
|
PartialMessage::PartialMessage()
|
|
: m_fragCount(0)
|
|
, m_received(0)
|
|
, m_totalLength(0)
|
|
{
|
|
}
|
|
|
|
bool
|
|
PartialMessage::add(uint16_t fragIndex, uint16_t fragCount, const Block& payload)
|
|
{
|
|
if (m_received == 0) { // first packet
|
|
m_fragCount = fragCount;
|
|
m_payloads.resize(fragCount);
|
|
}
|
|
|
|
if (m_fragCount != fragCount || fragIndex >= m_fragCount) {
|
|
return false;
|
|
}
|
|
|
|
if (!m_payloads[fragIndex].empty()) { // duplicate
|
|
return false;
|
|
}
|
|
|
|
m_payloads[fragIndex] = payload;
|
|
++m_received;
|
|
m_totalLength += payload.value_size();
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
PartialMessage::isComplete() const
|
|
{
|
|
return m_received == m_fragCount;
|
|
}
|
|
|
|
Block
|
|
PartialMessage::reassemble()
|
|
{
|
|
BOOST_ASSERT(this->isComplete());
|
|
|
|
ndn::BufferPtr buffer = make_shared<ndn::Buffer>(m_totalLength);
|
|
uint8_t* buf = buffer->get();
|
|
for (std::vector<Block>::const_iterator it = m_payloads.begin();
|
|
it != m_payloads.end(); ++it) {
|
|
const Block& payload = *it;
|
|
memcpy(buf, payload.value(), payload.value_size());
|
|
buf += payload.value_size();
|
|
}
|
|
|
|
return Block(buffer);
|
|
}
|
|
|
|
PartialMessageStore::PartialMessageStore(const time::nanoseconds& idleDuration)
|
|
: m_idleDuration(idleDuration)
|
|
{
|
|
}
|
|
|
|
PartialMessageStore::~PartialMessageStore()
|
|
{
|
|
}
|
|
|
|
void
|
|
PartialMessageStore::receiveNdnlpData(const Block& pkt)
|
|
{
|
|
NdnlpData parsed;
|
|
parsed.wireDecode(pkt);
|
|
if (parsed.m_fragCount == 1) { // single fragment
|
|
this->onReceive(parsed.m_payload.blockFromValue());
|
|
return;
|
|
}
|
|
|
|
uint64_t messageIdentifier = parsed.m_seq - parsed.m_fragIndex;
|
|
PartialMessage& pm = m_partialMessages[messageIdentifier];
|
|
this->scheduleCleanup(messageIdentifier, pm);
|
|
|
|
pm.add(parsed.m_fragIndex, parsed.m_fragCount, parsed.m_payload);
|
|
if (pm.isComplete()) {
|
|
this->onReceive(pm.reassemble());
|
|
this->cleanup(messageIdentifier);
|
|
}
|
|
}
|
|
|
|
void
|
|
PartialMessageStore::scheduleCleanup(uint64_t messageIdentifier,
|
|
PartialMessage& partialMessage)
|
|
{
|
|
partialMessage.expiry = scheduler::schedule(m_idleDuration,
|
|
bind(&PartialMessageStore::cleanup, this, messageIdentifier));
|
|
}
|
|
|
|
void
|
|
PartialMessageStore::cleanup(uint64_t messageIdentifier)
|
|
{
|
|
m_partialMessages.erase(messageIdentifier);
|
|
}
|
|
|
|
} // namespace ndnlp
|
|
} // namespace nfd
|