blob: e5c6c47afe086b9da435c90a036696d76ff65fd5 [file] [log] [blame]
Ashlesh Gawande0b2897e2018-06-20 14:40:47 -05001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/*
3 * Copyright (c) 2014-2018, The University of Memphis
4 *
5 * This file is part of PSync.
6 * See AUTHORS.md for complete list of PSync authors and contributors.
7 *
8 * PSync is free software: you can redistribute it and/or modify it under the terms
9 * of the GNU General Public License as published by the Free Software Foundation,
10 * either version 3 of the License, or (at your option) any later version.
11 *
12 * PSync is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
13 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
14 * PURPOSE. See the GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along with
17 * PSync, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
18 **/
19
20#include "state.hpp"
21
22namespace psync {
23
24State::State(const ndn::Block& block)
25{
26 wireDecode(block);
27}
28
29void
30State::addContent(const ndn::Name& prefix)
31{
32 m_content.emplace_back(prefix);
33}
34
35const ndn::Block&
36State::wireEncode() const
37{
38 if (m_wire.hasWire()) {
39 return m_wire;
40 }
41
42 ndn::EncodingEstimator estimator;
43 size_t estimatedSize = wireEncode(estimator);
44
45 ndn::EncodingBuffer buffer(estimatedSize, 0);
46 wireEncode(buffer);
47
48 m_wire = buffer.block();
49 return m_wire;
50}
51
52template<ndn::encoding::Tag TAG>
53size_t
54State::wireEncode(ndn::EncodingImpl<TAG>& block) const
55{
56 size_t totalLength = 0;
57
58 for (std::vector<ndn::Name>::const_reverse_iterator it = m_content.rbegin();
59 it != m_content.rend(); ++it) {
60 totalLength += it->wireEncode(block);
61 }
62
63 totalLength += block.prependVarNumber(totalLength);
64 totalLength += block.prependVarNumber(tlv::PSyncContent);
65
66 return totalLength;
67}
68
69NDN_CXX_DEFINE_WIRE_ENCODE_INSTANTIATIONS(State);
70
71void
72State::wireDecode(const ndn::Block& wire)
73{
74 if (!wire.hasWire()) {
75 BOOST_THROW_EXCEPTION(ndn::tlv::Error("The supplied block does not contain wire format"));
76 }
77
78 wire.parse();
79 m_wire = wire;
80
81 auto it = m_wire.elements_begin();
82
83 if (it->type() != tlv::PSyncContent) {
84 BOOST_THROW_EXCEPTION(ndn::tlv::Error("Unexpected TLV type when decoding Content: " +
85 ndn::to_string(wire.type())));
86 }
87
88 it->parse();
89
90 for (auto val = it->elements_begin(); val != it->elements_end(); ++val) {
91 if (val->type() == ndn::tlv::Name) {
92 m_content.emplace_back(*val);
93 }
94 else {
95 BOOST_THROW_EXCEPTION(ndn::tlv::Error("Expected Name Block, but Block is of a different type: #" +
96 ndn::to_string(m_wire.type())));
97 }
98 }
99}
100
101std::ostream&
102operator<<(std::ostream& os, const State& state)
103{
104 std::vector<ndn::Name> content = state.getContent();
105
106 os << "[";
107 std::copy(content.begin(), content.end(), ndn::make_ostream_joiner(os, ", "));
108 os << "]";
109
110 return os;
111}
112
113} // namespace psync