blob: 43efff9a1c7353725bc941e8fe059e8e861ef209 [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"
Junxiao Shic828dfc2016-09-15 13:26:22 +000023#include "../lp/packet.hpp"
Junxiao Shi7357ef22016-09-07 02:39:37 +000024#include "../lp/tags.hpp"
25#include "../mgmt/nfd/controller.hpp"
26#include "../mgmt/nfd/control-response.hpp"
Junxiao Shia60d9362014-11-12 09:38:21 -070027#include "../transport/transport.hpp"
Junxiao Shia60d9362014-11-12 09:38:21 -070028
Davide Pesavento537dc3a2016-02-18 19:35:26 +010029#include <boost/asio/io_service.hpp>
30
Junxiao Shia60d9362014-11-12 09:38:21 -070031namespace ndn {
32namespace util {
33
Junxiao Shia60d9362014-11-12 09:38:21 -070034class DummyClientFace::Transport : public ndn::Transport
35{
36public:
37 void
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020038 receive(Block block) const
Junxiao Shia60d9362014-11-12 09:38:21 -070039 {
Eric Newberry83872fd2015-08-06 17:01:24 -070040 block.encode();
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020041 if (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
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020047 close() override
Junxiao Shia60d9362014-11-12 09:38:21 -070048 {
49 }
50
51 virtual void
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020052 pause() override
Junxiao Shia60d9362014-11-12 09:38:21 -070053 {
54 }
55
56 virtual void
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020057 resume() override
Junxiao Shia60d9362014-11-12 09:38:21 -070058 {
59 }
60
61 virtual void
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020062 send(const Block& wire) override
Junxiao Shia60d9362014-11-12 09:38:21 -070063 {
Junxiao Shi27913b42014-12-23 14:49:38 -070064 onSendBlock(wire);
Junxiao Shia60d9362014-11-12 09:38:21 -070065 }
66
67 virtual void
Davide Pesaventoaeeb3fc2016-08-14 03:40:02 +020068 send(const Block& header, const Block& payload) override
Junxiao Shia60d9362014-11-12 09:38:21 -070069 {
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();
Junxiao Shic828dfc2016-09-15 13:26:22 +0000165
166 m_processEventsOverride = options.processEventsOverride;
Junxiao Shia60d9362014-11-12 09:38:21 -0700167}
168
169void
170DummyClientFace::enablePacketLogging()
171{
Junxiao Shi27913b42014-12-23 14:49:38 -0700172 onSendInterest.connect([this] (const Interest& interest) {
173 this->sentInterests.push_back(interest);
174 });
175 onSendData.connect([this] (const Data& data) {
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800176 this->sentData.push_back(data);
Junxiao Shi27913b42014-12-23 14:49:38 -0700177 });
Eric Newberry83872fd2015-08-06 17:01:24 -0700178 onSendNack.connect([this] (const lp::Nack& nack) {
179 this->sentNacks.push_back(nack);
180 });
Junxiao Shia60d9362014-11-12 09:38:21 -0700181}
182
183void
184DummyClientFace::enableRegistrationReply()
185{
Junxiao Shi27913b42014-12-23 14:49:38 -0700186 onSendInterest.connect([this] (const Interest& interest) {
Junxiao Shia60d9362014-11-12 09:38:21 -0700187 static const Name localhostRegistration("/localhost/nfd/rib");
188 if (!localhostRegistration.isPrefixOf(interest.getName()))
189 return;
190
191 nfd::ControlParameters params(interest.getName().get(-5).blockFromValue());
192 params.setFaceId(1);
193 params.setOrigin(0);
194 if (interest.getName().get(3) == name::Component("register")) {
195 params.setCost(0);
196 }
197
198 nfd::ControlResponse resp;
199 resp.setCode(200);
200 resp.setBody(params.wireEncode());
201
202 shared_ptr<Data> data = make_shared<Data>(interest.getName());
203 data->setContent(resp.wireEncode());
204
Alexander Afanasyev8cf1c562016-06-23 16:01:55 -0700205 m_keyChain.sign(*data, security::SigningInfo(security::SigningInfo::SIGNER_TYPE_SHA256));
Junxiao Shia60d9362014-11-12 09:38:21 -0700206
207 this->getIoService().post([this, data] { this->receive(*data); });
Junxiao Shi27913b42014-12-23 14:49:38 -0700208 });
Junxiao Shia60d9362014-11-12 09:38:21 -0700209}
210
211template<typename Packet>
212void
213DummyClientFace::receive(const Packet& packet)
214{
Eric Newberry83872fd2015-08-06 17:01:24 -0700215 lp::Packet lpPacket(packet.wireEncode());
Alexander Afanasyevea719672015-02-10 20:25:23 -0800216
Junxiao Shi4b469982015-12-03 18:20:19 +0000217 shared_ptr<lp::IncomingFaceIdTag> incomingFaceIdTag =
218 static_cast<const TagHost&>(packet).getTag<lp::IncomingFaceIdTag>();
219 if (incomingFaceIdTag != nullptr) {
220 lpPacket.add<lp::IncomingFaceIdField>(*incomingFaceIdTag);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800221 }
Eric Newberry83872fd2015-08-06 17:01:24 -0700222
Junxiao Shi4b469982015-12-03 18:20:19 +0000223 shared_ptr<lp::NextHopFaceIdTag> nextHopFaceIdTag =
224 static_cast<const TagHost&>(packet).getTag<lp::NextHopFaceIdTag>();
225 if (nextHopFaceIdTag != nullptr) {
226 lpPacket.add<lp::NextHopFaceIdField>(*nextHopFaceIdTag);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800227 }
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800228 static_pointer_cast<Transport>(getTransport())->receive(lpPacket.wireEncode());
Junxiao Shia60d9362014-11-12 09:38:21 -0700229}
230
231template void
232DummyClientFace::receive<Interest>(const Interest& packet);
233
234template void
235DummyClientFace::receive<Data>(const Data& packet);
236
Eric Newberry83872fd2015-08-06 17:01:24 -0700237template<>
238void
239DummyClientFace::receive<lp::Nack>(const lp::Nack& nack)
240{
241 lp::Packet lpPacket;
242 lpPacket.add<lp::NackField>(nack.getHeader());
243 Block interest = nack.getInterest().wireEncode();
244 lpPacket.add<lp::FragmentField>(make_pair(interest.begin(), interest.end()));
245
Junxiao Shi4b469982015-12-03 18:20:19 +0000246 shared_ptr<lp::IncomingFaceIdTag> incomingFaceIdTag = nack.getTag<lp::IncomingFaceIdTag>();
247 if (incomingFaceIdTag != nullptr) {
248 lpPacket.add<lp::IncomingFaceIdField>(*incomingFaceIdTag);
Eric Newberry83872fd2015-08-06 17:01:24 -0700249 }
250
Alexander Afanasyev3a6da362015-12-29 20:31:03 -0800251 static_pointer_cast<Transport>(getTransport())->receive(lpPacket.wireEncode());
Eric Newberry83872fd2015-08-06 17:01:24 -0700252}
Junxiao Shia60d9362014-11-12 09:38:21 -0700253
Junxiao Shic828dfc2016-09-15 13:26:22 +0000254void
255DummyClientFace::doProcessEvents(const time::milliseconds& timeout, bool keepThread)
256{
257 if (m_processEventsOverride != nullptr) {
258 m_processEventsOverride(timeout);
259 }
260 else {
261 this->Face::doProcessEvents(timeout, keepThread);
262 }
263}
264
Junxiao Shia60d9362014-11-12 09:38:21 -0700265} // namespace util
266} // namespace ndn