blob: aadb6bcf41e713fd5b7aedcf831b3d20d9752972 [file] [log] [blame]
Junxiao Shia60d9362014-11-12 09:38:21 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Alexander Afanasyevaf99f462015-01-19 21:43:09 -08003 * Copyright (c) 2013-2015 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
27namespace ndn {
28namespace util {
29
30const DummyClientFace::Options DummyClientFace::DEFAULT_OPTIONS { true, false };
31
32class DummyClientFace::Transport : public ndn::Transport
33{
34public:
35 void
Eric Newberry83872fd2015-08-06 17:01:24 -070036 receive(Block block)
Junxiao Shia60d9362014-11-12 09:38:21 -070037 {
Eric Newberry83872fd2015-08-06 17:01:24 -070038 block.encode();
39 if (static_cast<bool>(m_receiveCallback)) {
Junxiao Shia60d9362014-11-12 09:38:21 -070040 m_receiveCallback(block);
Eric Newberry83872fd2015-08-06 17:01:24 -070041 }
Junxiao Shia60d9362014-11-12 09:38:21 -070042 }
43
44 virtual void
45 close()
46 {
47 }
48
49 virtual void
50 pause()
51 {
52 }
53
54 virtual void
55 resume()
56 {
57 }
58
59 virtual void
60 send(const Block& wire)
61 {
Junxiao Shi27913b42014-12-23 14:49:38 -070062 onSendBlock(wire);
Junxiao Shia60d9362014-11-12 09:38:21 -070063 }
64
65 virtual void
66 send(const Block& header, const Block& payload)
67 {
Alexander Afanasyevea719672015-02-10 20:25:23 -080068 EncodingBuffer encoder(header.size() + payload.size(), header.size() + payload.size());
69 encoder.appendByteArray(header.wire(), header.size());
70 encoder.appendByteArray(payload.wire(), payload.size());
71
72 this->send(encoder.block());
Junxiao Shia60d9362014-11-12 09:38:21 -070073 }
74
75 boost::asio::io_service&
76 getIoService()
77 {
78 return *m_ioService;
79 }
80
Junxiao Shi27913b42014-12-23 14:49:38 -070081public:
82 Signal<Transport, Block> onSendBlock;
Junxiao Shia60d9362014-11-12 09:38:21 -070083};
84
85DummyClientFace::DummyClientFace(const Options& options, shared_ptr<Transport> transport)
86 : Face(transport)
87 , m_transport(transport)
88{
89 this->construct(options);
90}
91
92DummyClientFace::DummyClientFace(const Options& options, shared_ptr<Transport> transport,
93 boost::asio::io_service& ioService)
94 : Face(transport, ioService)
95 , m_transport(transport)
96{
97 this->construct(options);
98}
99
100void
101DummyClientFace::construct(const Options& options)
102{
Alexander Afanasyevea719672015-02-10 20:25:23 -0800103 m_transport->onSendBlock.connect([this] (const Block& blockFromDaemon) {
Eric Newberrye3e25052015-11-28 20:37:08 -0700104 Block packet(blockFromDaemon);
105 packet.encode();
106 lp::Packet lpPacket(packet);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800107
Eric Newberrye3e25052015-11-28 20:37:08 -0700108 Buffer::const_iterator begin, end;
109 std::tie(begin, end) = lpPacket.get<lp::FragmentField>();
110 Block block(&*begin, std::distance(begin, end));
Alexander Afanasyevea719672015-02-10 20:25:23 -0800111
Eric Newberrye3e25052015-11-28 20:37:08 -0700112 if (block.type() == tlv::Interest) {
113 shared_ptr<Interest> interest = make_shared<Interest>(block);
114 if (lpPacket.has<lp::NackField>()) {
115 shared_ptr<lp::Nack> nack = make_shared<lp::Nack>(std::move(*interest));
116 nack->setHeader(lpPacket.get<lp::NackField>());
117 if (lpPacket.has<lp::NextHopFaceIdField>()) {
Junxiao Shi4b469982015-12-03 18:20:19 +0000118 nack->setTag(make_shared<lp::NextHopFaceIdTag>(lpPacket.get<lp::NextHopFaceIdField>()));
Eric Newberry83872fd2015-08-06 17:01:24 -0700119 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700120 onSendNack(*nack);
Eric Newberry83872fd2015-08-06 17:01:24 -0700121 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700122 else {
123 if (lpPacket.has<lp::NextHopFaceIdField>()) {
Junxiao Shi4b469982015-12-03 18:20:19 +0000124 interest->setTag(make_shared<lp::NextHopFaceIdTag>(lpPacket.get<lp::NextHopFaceIdField>()));
Eric Newberry83872fd2015-08-06 17:01:24 -0700125 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700126 onSendInterest(*interest);
Eric Newberry83872fd2015-08-06 17:01:24 -0700127 }
Junxiao Shi27913b42014-12-23 14:49:38 -0700128 }
Eric Newberrye3e25052015-11-28 20:37:08 -0700129 else if (block.type() == tlv::Data) {
130 shared_ptr<Data> data = make_shared<Data>(block);
131
132 if (lpPacket.has<lp::CachePolicyField>()) {
Junxiao Shi4b469982015-12-03 18:20:19 +0000133 data->setTag(make_shared<lp::CachePolicyTag>(lpPacket.get<lp::CachePolicyField>()));
Eric Newberrye3e25052015-11-28 20:37:08 -0700134 }
135
136 onSendData(*data);
Junxiao Shi27913b42014-12-23 14:49:38 -0700137 }
138 });
139
Junxiao Shia60d9362014-11-12 09:38:21 -0700140 if (options.enablePacketLogging)
141 this->enablePacketLogging();
142
143 if (options.enableRegistrationReply)
144 this->enableRegistrationReply();
145}
146
147void
148DummyClientFace::enablePacketLogging()
149{
Junxiao Shi27913b42014-12-23 14:49:38 -0700150 onSendInterest.connect([this] (const Interest& interest) {
151 this->sentInterests.push_back(interest);
152 });
153 onSendData.connect([this] (const Data& data) {
154 this->sentDatas.push_back(data);
155 });
Eric Newberry83872fd2015-08-06 17:01:24 -0700156 onSendNack.connect([this] (const lp::Nack& nack) {
157 this->sentNacks.push_back(nack);
158 });
Junxiao Shia60d9362014-11-12 09:38:21 -0700159}
160
161void
162DummyClientFace::enableRegistrationReply()
163{
Junxiao Shi27913b42014-12-23 14:49:38 -0700164 onSendInterest.connect([this] (const Interest& interest) {
Junxiao Shia60d9362014-11-12 09:38:21 -0700165 static const Name localhostRegistration("/localhost/nfd/rib");
166 if (!localhostRegistration.isPrefixOf(interest.getName()))
167 return;
168
169 nfd::ControlParameters params(interest.getName().get(-5).blockFromValue());
170 params.setFaceId(1);
171 params.setOrigin(0);
172 if (interest.getName().get(3) == name::Component("register")) {
173 params.setCost(0);
174 }
175
176 nfd::ControlResponse resp;
177 resp.setCode(200);
178 resp.setBody(params.wireEncode());
179
180 shared_ptr<Data> data = make_shared<Data>(interest.getName());
181 data->setContent(resp.wireEncode());
182
183 KeyChain keyChain;
Yingdi Yu1b0311c2015-06-10 14:58:47 -0700184 keyChain.sign(*data, security::SigningInfo(security::SigningInfo::SIGNER_TYPE_SHA256));
Junxiao Shia60d9362014-11-12 09:38:21 -0700185
186 this->getIoService().post([this, data] { this->receive(*data); });
Junxiao Shi27913b42014-12-23 14:49:38 -0700187 });
Junxiao Shia60d9362014-11-12 09:38:21 -0700188}
189
190template<typename Packet>
191void
192DummyClientFace::receive(const Packet& packet)
193{
Eric Newberry83872fd2015-08-06 17:01:24 -0700194 lp::Packet lpPacket(packet.wireEncode());
Alexander Afanasyevea719672015-02-10 20:25:23 -0800195
Junxiao Shi4b469982015-12-03 18:20:19 +0000196 shared_ptr<lp::IncomingFaceIdTag> incomingFaceIdTag =
197 static_cast<const TagHost&>(packet).getTag<lp::IncomingFaceIdTag>();
198 if (incomingFaceIdTag != nullptr) {
199 lpPacket.add<lp::IncomingFaceIdField>(*incomingFaceIdTag);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800200 }
Eric Newberry83872fd2015-08-06 17:01:24 -0700201
Junxiao Shi4b469982015-12-03 18:20:19 +0000202 shared_ptr<lp::NextHopFaceIdTag> nextHopFaceIdTag =
203 static_cast<const TagHost&>(packet).getTag<lp::NextHopFaceIdTag>();
204 if (nextHopFaceIdTag != nullptr) {
205 lpPacket.add<lp::NextHopFaceIdField>(*nextHopFaceIdTag);
Alexander Afanasyevea719672015-02-10 20:25:23 -0800206 }
Eric Newberry83872fd2015-08-06 17:01:24 -0700207 m_transport->receive(lpPacket.wireEncode());
Junxiao Shia60d9362014-11-12 09:38:21 -0700208}
209
210template void
211DummyClientFace::receive<Interest>(const Interest& packet);
212
213template void
214DummyClientFace::receive<Data>(const Data& packet);
215
Eric Newberry83872fd2015-08-06 17:01:24 -0700216template<>
217void
218DummyClientFace::receive<lp::Nack>(const lp::Nack& nack)
219{
220 lp::Packet lpPacket;
221 lpPacket.add<lp::NackField>(nack.getHeader());
222 Block interest = nack.getInterest().wireEncode();
223 lpPacket.add<lp::FragmentField>(make_pair(interest.begin(), interest.end()));
224
Junxiao Shi4b469982015-12-03 18:20:19 +0000225 shared_ptr<lp::IncomingFaceIdTag> incomingFaceIdTag = nack.getTag<lp::IncomingFaceIdTag>();
226 if (incomingFaceIdTag != nullptr) {
227 lpPacket.add<lp::IncomingFaceIdField>(*incomingFaceIdTag);
Eric Newberry83872fd2015-08-06 17:01:24 -0700228 }
229
230 m_transport->receive(lpPacket.wireEncode());
231}
Junxiao Shia60d9362014-11-12 09:38:21 -0700232
233shared_ptr<DummyClientFace>
234makeDummyClientFace(const DummyClientFace::Options& options)
235{
Junxiao Shia1ea5062014-12-27 22:33:39 -0700236 // cannot use make_shared<DummyClientFace> because DummyClientFace constructor is private
Junxiao Shia60d9362014-11-12 09:38:21 -0700237 return shared_ptr<DummyClientFace>(
238 new DummyClientFace(options, make_shared<DummyClientFace::Transport>()));
239}
240
241shared_ptr<DummyClientFace>
242makeDummyClientFace(boost::asio::io_service& ioService,
243 const DummyClientFace::Options& options)
244{
Junxiao Shia1ea5062014-12-27 22:33:39 -0700245 // cannot use make_shared<DummyClientFace> because DummyClientFace constructor is private
Junxiao Shia60d9362014-11-12 09:38:21 -0700246 return shared_ptr<DummyClientFace>(
247 new DummyClientFace(options, make_shared<DummyClientFace::Transport>(),
248 ref(ioService)));
249}
250
251} // namespace util
252} // namespace ndn