blob: ca9475a391e63a9ec16cebd54b54c0e5e04ea982 [file] [log] [blame]
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
* Copyright (c) 2013-2014 Regents of the University of California.
*
* This file is part of ndn-cxx library (NDN C++ library with eXperimental eXtensions).
*
* ndn-cxx library is free software: you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any later version.
*
* ndn-cxx library is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
* PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
*
* You should have received copies of the GNU General Public License and GNU Lesser
* General Public License along with ndn-cxx, e.g., in COPYING.md file. If not, see
* <http://www.gnu.org/licenses/>.
*
* See AUTHORS.md for complete list of ndn-cxx authors and contributors.
*
* @author Wentao Shang <http://irl.cs.ucla.edu/~wentao/>
*/
#include "face.hpp"
#include "security/key-chain.hpp"
namespace ndn {
const size_t MAX_SEG_SIZE = 4096;
class Producer
{
public:
Producer(const char* name)
: m_name(name)
, m_isVerbose(false)
{
int segnum = 0;
char* buf = new char[MAX_SEG_SIZE];
do
{
std::cin.read(buf, MAX_SEG_SIZE);
int got = std::cin.gcount();
if (got > 0)
{
shared_ptr<Data> data =
make_shared<Data>(Name(m_name).appendSegment(segnum));
data->setFreshnessPeriod(time::milliseconds(10000)); // 10 sec
data->setContent(reinterpret_cast<const uint8_t*>(buf), got);
m_keychain.sign(*data);
m_store.push_back(data);
segnum++;
}
}
while (static_cast<bool>(std::cin));
if (m_isVerbose)
std::cerr << "Created " << segnum << " chunks for prefix [" << m_name << "]" << std::endl;
}
void
onInterest(const Interest& interest)
{
if (m_isVerbose)
std::cerr << "<< I: " << interest << std::endl;
size_t segnum = static_cast<size_t>(interest.getName().rbegin()->toSegment());
if (segnum < m_store.size())
{
m_face.put(*m_store[segnum]);
}
}
void
onRegisterFailed(const Name& prefix, const std::string& reason)
{
std::cerr << "ERROR: Failed to register prefix '"
<< prefix << "' in local hub's daemon (" << reason << ")"
<< std::endl;
m_face.shutdown();
}
void
run()
{
if (m_store.empty())
{
std::cerr << "Nothing to serve. Exiting." << std::endl;
return;
}
m_face.setInterestFilter(m_name,
bind(&Producer::onInterest, this, _2),
RegisterPrefixSuccessCallback(),
bind(&Producer::onRegisterFailed, this, _1, _2));
m_face.processEvents();
}
private:
Name m_name;
Face m_face;
KeyChain m_keychain;
std::vector< shared_ptr<Data> > m_store;
bool m_isVerbose;
};
int
main(int argc, char** argv)
{
if (argc < 2)
{
std::cerr << "Usage: ./ndnputchunks [data_prefix]\n";
return -1;
}
try
{
time::steady_clock::TimePoint startTime = time::steady_clock::now();
std::cerr << "Preparing the input..." << std::endl;
Producer producer(argv[1]);
std::cerr << "Ready... (took " << (time::steady_clock::now() - startTime) << std::endl;
while (true)
{
try
{
// this will exit when daemon dies... so try to connect again if possible
producer.run();
}
catch (std::exception& e)
{
std::cerr << "ERROR: " << e.what() << std::endl;
// and keep going
sleep(1);
}
}
}
catch (std::exception& e)
{
std::cerr << "ERROR: " << e.what() << std::endl;
}
return 0;
}
} // namespace ndn
int
main(int argc, char** argv)
{
return ndn::main(argc, argv);
}