blob: bdc8764339813bbc64f0e3f620fc97d3e26604b4 [file] [log] [blame]
Junxiao Shia60d9362014-11-12 09:38:21 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Davide Pesavento537dc3a2016-02-18 19:35:26 +01003 * Copyright (c) 2013-2016 Regents of the University of California.
Junxiao Shia60d9362014-11-12 09:38:21 -07004 *
5 * This file is part of ndn-cxx library (NDN C++ library with eXperimental eXtensions).
6 *
7 * ndn-cxx library is free software: you can redistribute it and/or modify it under the
8 * terms of the GNU Lesser General Public License as published by the Free Software
9 * Foundation, either version 3 of the License, or (at your option) any later version.
10 *
11 * ndn-cxx library is distributed in the hope that it will be useful, but WITHOUT ANY
12 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
13 * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
14 *
15 * You should have received copies of the GNU General Public License and GNU Lesser
16 * General Public License along with ndn-cxx, e.g., in COPYING.md file. If not, see
17 * <http://www.gnu.org/licenses/>.
18 *
19 * See AUTHORS.md for complete list of ndn-cxx authors and contributors.
20 */
21
22#include "dummy-client-face.hpp"
23#include "../transport/transport.hpp"
24#include "../management/nfd-controller.hpp"
25#include "../management/nfd-control-response.hpp"
26
Davide Pesavento537dc3a2016-02-18 19:35:26 +010027#include <boost/asio/io_service.hpp>
28
Junxiao Shia60d9362014-11-12 09:38:21 -070029namespace ndn {
30namespace util {
31
32const DummyClientFace::Options DummyClientFace::DEFAULT_OPTIONS { true, false };
33
34class DummyClientFace::Transport : public ndn::Transport
35{
36public:
37 void
Eric Newberry83872fd2015-08-06 17:01:24 -070038 receive(Block block)
Junxiao Shia60d9362014-11-12 09:38:21 -070039 {
Eric Newberry83872fd2015-08-06 17:01:24 -070040 block.encode();
41 if (static_cast<bool>(m_receiveCallback)) {
Junxiao Shia60d9362014-11-12 09:38:21 -070042 m_receiveCallback(block);
Eric Newberry83872fd2015-08-06 17:01:24 -070043 }
Junxiao Shia60d9362014-11-12 09:38:21 -070044 }
45
46 virtual void
47 close()
48 {
49 }
50
51 virtual void
52 pause()
53 {
54 }
55
56 virtual void
57 resume()
58 {
59 }
60
61 virtual void
62 send(const Block& wire)
63 {
Junxiao Shi27913b42014-12-23 14:49:38 -070064 onSendBlock(wire);
Junxiao Shia60d9362014-11-12 09:38:21 -070065 }
66
67 virtual void
68 send(const Block& header, const Block& payload)
69 {
Alexander Afanasyevea719672015-02-10 20:25:23 -080070 EncodingBuffer encoder(header.size() + payload.size(), header.size() + payload.size());
71 encoder.appendByteArray(header.wire(), header.size());
72 encoder.appendByteArray(payload.wire(), payload.size());
73
74 this->send(encoder.block());
Junxiao Shia60d9362014-11-12 09:38:21 -070075 }
76
77 boost::asio::io_service&
78 getIoService()
79 {
80 return *m_ioService;
81 }
82
Junxiao Shi27913b42014-12-23 14:49:38 -070083public:
84 Signal<Transport, Block> onSendBlock;
Junxiao Shia60d9362014-11-12 09:38:21 -070085};
86
Alexander Afanasyev3a6da362015-12-29 20:31:03 -080087DummyClientFace::DummyClientFace(const Options& options/* = DummyClientFace::DEFAULT_OPTIONS*/)
88 : Face(make_shared<DummyClientFace::Transport>())
Alexander Afanasyev8cf1c562016-06-23 16:01:55 -070089 , m_internalKeyChain(new KeyChain)
90 , m_keyChain(*m_internalKeyChain)
91{
92 this->construct(options);
93}
94
95DummyClientFace::DummyClientFace(KeyChain& keyChain,
96 const Options& options/* = DummyClientFace::DEFAULT_OPTIONS*/)
97 : Face(make_shared<DummyClientFace::Transport>(), keyChain)
Alexander Afanasyev8cf1c562016-06-23 16:01:55 -070098 , m_keyChain(keyChain)
Junxiao Shia60d9362014-11-12 09:38:21 -070099{
100 this->construct(options);
101}
102
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800103DummyClientFace::DummyClientFace(boost::asio::io_service& ioService,
104 const Options& options/* = DummyClientFace::DEFAULT_OPTIONS*/)
105 : Face(make_shared<DummyClientFace::Transport>(), ioService)
Alexander Afanasyev8cf1c562016-06-23 16:01:55 -0700106 , m_internalKeyChain(new KeyChain)
107 , m_keyChain(*m_internalKeyChain)
108{
109 this->construct(options);
110}
111
112DummyClientFace::DummyClientFace(boost::asio::io_service& ioService, KeyChain& keyChain,
113 const Options& options/* = DummyClientFace::DEFAULT_OPTIONS*/)
114 : Face(make_shared<DummyClientFace::Transport>(), ioService, keyChain)
Alexander Afanasyev8cf1c562016-06-23 16:01:55 -0700115 , m_keyChain(keyChain)
Junxiao Shia60d9362014-11-12 09:38:21 -0700116{
117 this->construct(options);
118}
119
120void
121DummyClientFace::construct(const Options& options)
122{
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800123 static_pointer_cast<Transport>(getTransport())->onSendBlock.connect([this] (const Block& blockFromDaemon) {
Eric Newberrye3e25052015-11-28 20:37:08 -0700124 Block packet(blockFromDaemon);
125 packet.encode();
126 lp::Packet lpPacket(packet);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800127
Eric Newberrye3e25052015-11-28 20:37:08 -0700128 Buffer::const_iterator begin, end;
129 std::tie(begin, end) = lpPacket.get<lp::FragmentField>();
130 Block block(&*begin, std::distance(begin, end));
Alexander Afanasyevea719672015-02-10 20:25:23 -0800131
Eric Newberrye3e25052015-11-28 20:37:08 -0700132 if (block.type() == tlv::Interest) {
133 shared_ptr<Interest> interest = make_shared<Interest>(block);
134 if (lpPacket.has<lp::NackField>()) {
135 shared_ptr<lp::Nack> nack = make_shared<lp::Nack>(std::move(*interest));
136 nack->setHeader(lpPacket.get<lp::NackField>());
137 if (lpPacket.has<lp::NextHopFaceIdField>()) {
Junxiao Shi4b469982015-12-03 18:20:19 +0000138 nack->setTag(make_shared<lp::NextHopFaceIdTag>(lpPacket.get<lp::NextHopFaceIdField>()));
Eric Newberry83872fd2015-08-06 17:01:24 -0700139 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700140 onSendNack(*nack);
Eric Newberry83872fd2015-08-06 17:01:24 -0700141 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700142 else {
143 if (lpPacket.has<lp::NextHopFaceIdField>()) {
Junxiao Shi4b469982015-12-03 18:20:19 +0000144 interest->setTag(make_shared<lp::NextHopFaceIdTag>(lpPacket.get<lp::NextHopFaceIdField>()));
Eric Newberry83872fd2015-08-06 17:01:24 -0700145 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700146 onSendInterest(*interest);
Eric Newberry83872fd2015-08-06 17:01:24 -0700147 }
Junxiao Shi27913b42014-12-23 14:49:38 -0700148 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700149 else if (block.type() == tlv::Data) {
150 shared_ptr<Data> data = make_shared<Data>(block);
151
152 if (lpPacket.has<lp::CachePolicyField>()) {
Junxiao Shi4b469982015-12-03 18:20:19 +0000153 data->setTag(make_shared<lp::CachePolicyTag>(lpPacket.get<lp::CachePolicyField>()));
Eric Newberrye3e25052015-11-28 20:37:08 -0700154 }
155
156 onSendData(*data);
Junxiao Shi27913b42014-12-23 14:49:38 -0700157 }
158 });
159
Junxiao Shia60d9362014-11-12 09:38:21 -0700160 if (options.enablePacketLogging)
161 this->enablePacketLogging();
162
163 if (options.enableRegistrationReply)
164 this->enableRegistrationReply();
165}
166
167void
168DummyClientFace::enablePacketLogging()
169{
Junxiao Shi27913b42014-12-23 14:49:38 -0700170 onSendInterest.connect([this] (const Interest& interest) {
171 this->sentInterests.push_back(interest);
172 });
173 onSendData.connect([this] (const Data& data) {
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800174 this->sentData.push_back(data);
Junxiao Shi27913b42014-12-23 14:49:38 -0700175 });
Eric Newberry83872fd2015-08-06 17:01:24 -0700176 onSendNack.connect([this] (const lp::Nack& nack) {
177 this->sentNacks.push_back(nack);
178 });
Junxiao Shia60d9362014-11-12 09:38:21 -0700179}
180
181void
182DummyClientFace::enableRegistrationReply()
183{
Junxiao Shi27913b42014-12-23 14:49:38 -0700184 onSendInterest.connect([this] (const Interest& interest) {
Junxiao Shia60d9362014-11-12 09:38:21 -0700185 static const Name localhostRegistration("/localhost/nfd/rib");
186 if (!localhostRegistration.isPrefixOf(interest.getName()))
187 return;
188
189 nfd::ControlParameters params(interest.getName().get(-5).blockFromValue());
190 params.setFaceId(1);
191 params.setOrigin(0);
192 if (interest.getName().get(3) == name::Component("register")) {
193 params.setCost(0);
194 }
195
196 nfd::ControlResponse resp;
197 resp.setCode(200);
198 resp.setBody(params.wireEncode());
199
200 shared_ptr<Data> data = make_shared<Data>(interest.getName());
201 data->setContent(resp.wireEncode());
202
Alexander Afanasyev8cf1c562016-06-23 16:01:55 -0700203 m_keyChain.sign(*data, security::SigningInfo(security::SigningInfo::SIGNER_TYPE_SHA256));
Junxiao Shia60d9362014-11-12 09:38:21 -0700204
205 this->getIoService().post([this, data] { this->receive(*data); });
Junxiao Shi27913b42014-12-23 14:49:38 -0700206 });
Junxiao Shia60d9362014-11-12 09:38:21 -0700207}
208
209template<typename Packet>
210void
211DummyClientFace::receive(const Packet& packet)
212{
Eric Newberry83872fd2015-08-06 17:01:24 -0700213 lp::Packet lpPacket(packet.wireEncode());
Alexander Afanasyevea719672015-02-10 20:25:23 -0800214
Junxiao Shi4b469982015-12-03 18:20:19 +0000215 shared_ptr<lp::IncomingFaceIdTag> incomingFaceIdTag =
216 static_cast<const TagHost&>(packet).getTag<lp::IncomingFaceIdTag>();
217 if (incomingFaceIdTag != nullptr) {
218 lpPacket.add<lp::IncomingFaceIdField>(*incomingFaceIdTag);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800219 }
Eric Newberry83872fd2015-08-06 17:01:24 -0700220
Junxiao Shi4b469982015-12-03 18:20:19 +0000221 shared_ptr<lp::NextHopFaceIdTag> nextHopFaceIdTag =
222 static_cast<const TagHost&>(packet).getTag<lp::NextHopFaceIdTag>();
223 if (nextHopFaceIdTag != nullptr) {
224 lpPacket.add<lp::NextHopFaceIdField>(*nextHopFaceIdTag);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800225 }
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800226 static_pointer_cast<Transport>(getTransport())->receive(lpPacket.wireEncode());
Junxiao Shia60d9362014-11-12 09:38:21 -0700227}
228
229template void
230DummyClientFace::receive<Interest>(const Interest& packet);
231
232template void
233DummyClientFace::receive<Data>(const Data& packet);
234
Eric Newberry83872fd2015-08-06 17:01:24 -0700235template<>
236void
237DummyClientFace::receive<lp::Nack>(const lp::Nack& nack)
238{
239 lp::Packet lpPacket;
240 lpPacket.add<lp::NackField>(nack.getHeader());
241 Block interest = nack.getInterest().wireEncode();
242 lpPacket.add<lp::FragmentField>(make_pair(interest.begin(), interest.end()));
243
Junxiao Shi4b469982015-12-03 18:20:19 +0000244 shared_ptr<lp::IncomingFaceIdTag> incomingFaceIdTag = nack.getTag<lp::IncomingFaceIdTag>();
245 if (incomingFaceIdTag != nullptr) {
246 lpPacket.add<lp::IncomingFaceIdField>(*incomingFaceIdTag);
Eric Newberry83872fd2015-08-06 17:01:24 -0700247 }
248
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800249 static_pointer_cast<Transport>(getTransport())->receive(lpPacket.wireEncode());
Eric Newberry83872fd2015-08-06 17:01:24 -0700250}
Junxiao Shia60d9362014-11-12 09:38:21 -0700251
Junxiao Shia60d9362014-11-12 09:38:21 -0700252} // namespace util
253} // namespace ndn