diff options
Diffstat (limited to 'apps/consumers/icnet_consumer_test.cc')
-rwxr-xr-x | apps/consumers/icnet_consumer_test.cc | 208 |
1 files changed, 208 insertions, 0 deletions
diff --git a/apps/consumers/icnet_consumer_test.cc b/apps/consumers/icnet_consumer_test.cc new file mode 100755 index 00000000..d5b57d6f --- /dev/null +++ b/apps/consumers/icnet_consumer_test.cc @@ -0,0 +1,208 @@ +/* + * 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" + +#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 bufferSize) { + std::cout << "Content retrieved!! Size: " << bufferSize << std::endl; + + io_service_.dispatch([buffer, bufferSize]() { + std::ofstream file("ciao.txt", std::ofstream::binary); + file.write((char *) buffer, bufferSize); + file.close(); + }); + } + + bool verifyData(ConsumerSocket &c, const ContentObject &contentObject) { + if (contentObject.getContentType() == PayloadType::DATA) { + std::cout << "VERIFY CONTENT" << std::endl; + } else if (contentObject.getContentType() == PayloadType::MANIFEST) { + std::cout << "VERIFY MANIFEST" << std::endl; + } + + return true; + } + + void processLeavingInterest(ConsumerSocket &c, const Interest &interest) { + // std::cout << "LEAVES " << interest.getName().toUri() << 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_; +}; + +class Verificator { + public: + Verificator() { + }; + + ~Verificator() { + // m_keyChain.deleteIdentity(Name(IDENTITY_NAME)); + } + + bool onPacket(ConsumerSocket &c, const ContentObject &contentObject) { + + return true; + } +}; + +void becomeDaemon() { + pid_t process_id = 0; + pid_t sid = 0; + + // Create child process + process_id = fork(); + + // Indication of fork() failure + if (process_id < 0) { + printf("fork failed!\n"); + // Return failure in exit status + exit(EXIT_FAILURE); + } + + // PARENT PROCESS. Need to kill it. + if (process_id > 0) { + printf("process_id of child process %d \n", process_id); + // return success in exit status + exit(EXIT_SUCCESS); + } + + //unmask the file mode + umask(0); + + //set new session + sid = setsid(); + if (sid < 0) { + // Return failure + exit(EXIT_FAILURE); + } + + // Change the current working directory to root. + chdir("/"); + + // Close stdin. stdout and stderr + close(STDIN_FILENO); + close(STDOUT_FILENO); + close(STDERR_FILENO); + + // Really start application +} + +int main(int argc, char *argv[]) { + double beta = DEFAULT_BETA; + double dropFactor = DEFAULT_GAMMA; + bool daemon = false; + bool rttStats = false; + + int opt; + while ((opt = getopt(argc, argv, "b:d:DR")) != -1) { + switch (opt) { + case 'b': + beta = std::stod(optarg); + break; + case 'd': + dropFactor = std::stod(optarg); + break; + case 'D': + daemon = true; + break; + case 'R': + rttStats = true; + break; + default: + exit(EXIT_FAILURE); + } + } + + std::string name = "ccnx:/ccnxtest"; + + if (argv[optind] == 0) { + std::cerr << "Using default name ccnx:/ccnxtest" << std::endl; + } else { + name = argv[optind]; + } + + if (daemon) { + becomeDaemon(); + } + + ConsumerSocket c(Name(name.c_str()), TransportProtocolAlgorithms::RAAQM); + + CallbackContainer stubs; + Verificator verificator; + + c.setSocketOption(GeneralTransportOptions::INTEREST_LIFETIME, 1001); + c.setSocketOption(RaaqmTransportOptions::BETA_VALUE, beta); + c.setSocketOption(RaaqmTransportOptions::DROP_FACTOR, dropFactor); + c.setSocketOption(GeneralTransportOptions::MAX_INTEREST_RETX, 200); + c.setSocketOption(OtherOptions::VIRTUAL_DOWNLOAD, true); + c.setSocketOption(RaaqmTransportOptions::RTT_STATS, rttStats); + + 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); +} |