diff options
Diffstat (limited to 'apps/consumers/icnet_iget.cc')
-rwxr-xr-x | apps/consumers/icnet_iget.cc | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/apps/consumers/icnet_iget.cc b/apps/consumers/icnet_iget.cc new file mode 100755 index 00000000..db5ef173 --- /dev/null +++ b/apps/consumers/icnet_iget.cc @@ -0,0 +1,159 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "icnet_socket_consumer.h" + +typedef std::chrono::time_point<std::chrono::system_clock> Time; +typedef std::chrono::milliseconds TimeDuration; + +Time t1 = std::chrono::system_clock::now(); + +#define DEFAULT_BETA 0.99 +#define DEFAULT_GAMMA 0.07 + +namespace icnet { + +class CallbackContainer { + public: + CallbackContainer() + : work_(new boost::asio::io_service::work(io_service_)), + handler_(std::async(std::launch::async, [this]() { io_service_.run(); })) { + seen_manifest_segments_ = 0; + seen_data_segments_ = 0; + byte_counter_ = 0; + } + + ~CallbackContainer() { + work_.reset(); + } + + void processPayload(ConsumerSocket &c, const uint8_t *buffer, size_t buffer_size) { + Name m_name; + c.getSocketOption(GeneralTransportOptions::NAME_PREFIX, m_name); + std::string filename = m_name.toString().substr(1 + m_name.toString().find_last_of("/")); + io_service_.dispatch([buffer, buffer_size, filename]() { + std::cout << "Saving to: " << filename << " " << buffer_size / 1024 << "kB" << std::endl; + Time t3 = std::chrono::system_clock::now();; + std::ofstream file(filename.c_str(), std::ofstream::binary); + file.write((char *) buffer, buffer_size); + file.close(); + Time t2 = std::chrono::system_clock::now();; + TimeDuration dt = std::chrono::duration_cast<std::chrono::milliseconds>(t2 - t1); + TimeDuration dt3 = std::chrono::duration_cast<std::chrono::milliseconds>(t3 - t1); + long msec = dt.count(); + long msec3 = dt3.count(); + std::cout << "Elapsed Time: " << msec / 1000.0 << " seconds -- " << buffer_size * 8 / msec / 1000.0 + << "[Mbps] -- " << buffer_size * 8 / msec3 / 1000.0 << "[Mbps]" << std::endl; + }); + } + + bool verifyData(ConsumerSocket &c, const ContentObject &content_object) { + if (content_object.getContentType() == PayloadType::DATA) { + std::cout << "VERIFY CONTENT" << std::endl; + } else if (content_object.getContentType() == PayloadType::MANIFEST) { + std::cout << "VERIFY MANIFEST" << std::endl; + } + + return true; + } + + void processLeavingInterest(ConsumerSocket &c, const Interest &interest) { + // std::cout << "OUTPUT: " << interest.getName() << std::endl; + } + + private: + int seen_manifest_segments_; + int seen_data_segments_; + int byte_counter_; + boost::asio::io_service io_service_; + std::shared_ptr<boost::asio::io_service::work> work_; + std::future<void> handler_; +}; + +/* + * The client signature verification is currently being reworked with the new API. + * The implementation is disabled for the moment. + */ + +class Verificator { + public: + Verificator() { + }; + + ~Verificator() { + // m_keyChain.deleteIdentity(Name(IDENTITY_NAME)); + } + + bool onPacket(ConsumerSocket &c, const ContentObject &contentObject) { + return true; + } +}; + +int main(int argc, char **argv) { + + std::string url = ""; + std::string locator = ""; + std::string path = ""; + std::string name = "ccnx:/locator/get/path"; + size_t found = 0; + size_t path_begin = 0; + + if (argv[optind] == 0) { + std::cerr << "Missing URL" << std::endl; + return 0; + } else { + url = argv[optind]; + std::cout << "Iget " << url << std::endl; + } + + found = url.find("//"); + path_begin = url.find('/', found + 2); + locator = url.substr(found + 2, path_begin - (found + 2)); + path = url.substr(path_begin, std::string::npos); + std::cout << "locator " << locator << std::endl; + std::cout << "path " << path << std::endl; + name = "ccnx:/" + locator + "/get" + path; + std::cout << "Iget ccnx name: " << name << std::endl; + + ConsumerSocket c(Name(name.c_str()), TransportProtocolAlgorithms::RAAQM); + CallbackContainer stubs; + Verificator verificator; + + c.setSocketOption(ConsumerCallbacksOptions::CONTENT_OBJECT_TO_VERIFY, + (ConsumerContentObjectVerificationCallback) std::bind(&Verificator::onPacket, + &verificator, + std::placeholders::_1, + std::placeholders::_2)); + c.setSocketOption(ConsumerCallbacksOptions::CONTENT_RETRIEVED, + (ConsumerContentCallback) std::bind(&CallbackContainer::processPayload, + &stubs, + std::placeholders::_1, + std::placeholders::_2, + std::placeholders::_3)); + c.setSocketOption(ConsumerCallbacksOptions::INTEREST_OUTPUT, + (ConsumerInterestCallback) std::bind(&CallbackContainer::processLeavingInterest, + &stubs, + std::placeholders::_1, + std::placeholders::_2)); + c.consume(Name()); + c.stop(); + return 0; +} + +} // end namespace icnet + +int main(int argc, char **argv) { + return icnet::main(argc, argv); +} |