blob: 4db5ebf5d13bc35d7a8028e71edd691b149a5eae [file] [log] [blame]
Eric Newberry2f041d22018-06-03 18:02:31 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/*
Junxiao Shi06d008c2019-02-04 08:26:59 +00003 * Copyright (c) 2014-2019, Regents of the University of California,
Eric Newberry2f041d22018-06-03 18:02:31 -07004 * Arizona Board of Regents,
5 * Colorado State University,
6 * University Pierre & Marie Curie, Sorbonne University,
7 * Washington University in St. Louis,
8 * Beijing Institute of Technology,
9 * The University of Memphis.
10 *
11 * This file is part of ndn-tools (Named Data Networking Essential Tools).
12 * See AUTHORS.md for complete list of ndn-tools authors and contributors.
13 *
14 * ndn-tools is free software: you can redistribute it and/or modify it under the terms
15 * of the GNU General Public License as published by the Free Software Foundation,
16 * either version 3 of the License, or (at your option) any later version.
17 *
18 * ndn-tools is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
19 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
20 * PURPOSE. See the GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License along with
23 * ndn-tools, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
24 *
25 * @author Jerald Paul Abraham <jeraldabraham@email.arizona.edu>
26 */
27
28#include "ndnpoke.hpp"
29
Davide Pesaventoe75861e2019-07-24 21:55:39 -040030#include <ndn-cxx/encoding/buffer-stream.hpp>
Eric Newberry2f041d22018-06-03 18:02:31 -070031
32namespace ndn {
33namespace peek {
34
Davide Pesaventoe75861e2019-07-24 21:55:39 -040035NdnPoke::NdnPoke(Face& face, KeyChain& keyChain, std::istream& input, const PokeOptions& options)
36 : m_options(options)
37 , m_face(face)
Eric Newberry2f041d22018-06-03 18:02:31 -070038 , m_keyChain(keyChain)
Davide Pesaventoe75861e2019-07-24 21:55:39 -040039 , m_input(input)
40 , m_scheduler(m_face.getIoService())
Eric Newberry2f041d22018-06-03 18:02:31 -070041{
42}
43
44void
45NdnPoke::start()
46{
Davide Pesaventoe75861e2019-07-24 21:55:39 -040047 auto data = createData();
48
Eric Newberry2f041d22018-06-03 18:02:31 -070049 if (m_options.wantForceData) {
Davide Pesaventob3ae6342019-07-25 21:07:16 -040050 return sendData(*data);
Eric Newberry2f041d22018-06-03 18:02:31 -070051 }
Davide Pesaventoe75861e2019-07-24 21:55:39 -040052
53 m_registeredPrefix = m_face.setInterestFilter(m_options.name,
Davide Pesaventob3ae6342019-07-25 21:07:16 -040054 [this, data] (auto&&, const auto& interest) { this->onInterest(interest, *data); },
55 [this] (auto&&) { this->onRegSuccess(); },
56 [this] (auto&&, const auto& reason) { this->onRegFailure(reason); });
Eric Newberry2f041d22018-06-03 18:02:31 -070057}
58
59shared_ptr<Data>
Davide Pesaventoe75861e2019-07-24 21:55:39 -040060NdnPoke::createData() const
Eric Newberry2f041d22018-06-03 18:02:31 -070061{
Davide Pesaventoe75861e2019-07-24 21:55:39 -040062 auto data = make_shared<Data>(m_options.name);
Eric Newberry2f041d22018-06-03 18:02:31 -070063 if (m_options.freshnessPeriod) {
Davide Pesaventoe75861e2019-07-24 21:55:39 -040064 data->setFreshnessPeriod(*m_options.freshnessPeriod);
65 }
66 if (m_options.wantFinalBlockId) {
67 data->setFinalBlock(m_options.name.at(-1));
Eric Newberry2f041d22018-06-03 18:02:31 -070068 }
69
Davide Pesaventoe75861e2019-07-24 21:55:39 -040070 OBufferStream os;
71 os << m_input.rdbuf();
72 data->setContent(os.buf());
Eric Newberry2f041d22018-06-03 18:02:31 -070073
Davide Pesaventoe75861e2019-07-24 21:55:39 -040074 m_keyChain.sign(*data, m_options.signingInfo);
Eric Newberry2f041d22018-06-03 18:02:31 -070075
Davide Pesaventoe75861e2019-07-24 21:55:39 -040076 return data;
Eric Newberry2f041d22018-06-03 18:02:31 -070077}
78
Davide Pesaventob3ae6342019-07-25 21:07:16 -040079void
80NdnPoke::sendData(const Data& data)
81{
82 m_face.put(data);
83 m_result = Result::DATA_SENT;
84
85 if (m_options.isVerbose) {
86 std::cerr << "DATA: " << data;
87 }
88}
89
90void
91NdnPoke::onInterest(const Interest& interest, const Data& data)
92{
93 if (m_options.isVerbose) {
94 std::cerr << "INTEREST: " << interest << std::endl;
95 }
96
97 m_timeoutEvent.cancel();
98 m_registeredPrefix.cancel();
99 sendData(data);
100}
101
102void
103NdnPoke::onRegSuccess()
104{
105 if (m_options.isVerbose) {
106 std::cerr << "Prefix registration successful" << std::endl;
107 }
108
109 if (m_options.timeout) {
110 m_timeoutEvent = m_scheduler.schedule(*m_options.timeout, [this] {
111 m_result = Result::TIMEOUT;
112 m_registeredPrefix.cancel();
113
114 if (m_options.isVerbose) {
115 std::cerr << "TIMEOUT" << std::endl;
116 }
117 });
118 }
119}
120
121void
122NdnPoke::onRegFailure(const std::string& reason)
123{
124 m_result = Result::PREFIX_REG_FAIL;
125 std::cerr << "Prefix registration failure (" << reason << ")" << std::endl;
126}
127
Eric Newberry2f041d22018-06-03 18:02:31 -0700128} // namespace peek
129} // namespace ndn