Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
Junxiao Shi | f748a4e | 2017-07-05 23:41:48 +0000 | [diff] [blame] | 2 | /* |
| 3 | * Copyright (c) 2014-2017, Regents of the University of California, |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 4 | * 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 NFD (Named Data Networking Forwarding Daemon). |
| 12 | * See AUTHORS.md for complete list of NFD authors and contributors. |
| 13 | * |
| 14 | * NFD 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 | * NFD 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 | * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
| 24 | */ |
| 25 | |
| 26 | #ifndef NFD_TOOLS_NDN_AUTOCONFIG_MULTICAST_DISCOVERY_HPP |
| 27 | #define NFD_TOOLS_NDN_AUTOCONFIG_MULTICAST_DISCOVERY_HPP |
| 28 | |
Junxiao Shi | f748a4e | 2017-07-05 23:41:48 +0000 | [diff] [blame] | 29 | #include "stage.hpp" |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 30 | #include <ndn-cxx/face.hpp> |
| 31 | #include <ndn-cxx/mgmt/nfd/controller.hpp> |
| 32 | #include <ndn-cxx/mgmt/nfd/face-status.hpp> |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 33 | |
| 34 | namespace ndn { |
| 35 | namespace tools { |
| 36 | namespace autoconfig { |
| 37 | |
| 38 | /** |
| 39 | * @brief Multicast discovery stage |
| 40 | * |
| 41 | * - Request |
| 42 | * |
| 43 | * The end host sends an Interest over a multicast face. |
| 44 | * |
| 45 | * Interest Name is /localhop/ndn-autoconf/hub. |
| 46 | * |
| 47 | * - Response |
| 48 | * |
| 49 | * A producer app on the HUB answer this Interest with a Data packet that contains a |
| 50 | * TLV-encoded Uri block. The value of this block is the URI for the HUB, preferably a |
| 51 | * UDP tunnel. |
| 52 | */ |
Junxiao Shi | f748a4e | 2017-07-05 23:41:48 +0000 | [diff] [blame] | 53 | class MulticastDiscovery : public Stage |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 54 | { |
| 55 | public: |
| 56 | /** |
| 57 | * @brief Create multicast discovery stage |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 58 | */ |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 59 | MulticastDiscovery(Face& face, nfd::Controller& controller); |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 60 | |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 61 | const std::string& |
| 62 | getName() const override |
| 63 | { |
| 64 | static const std::string STAGE_NAME("multicast discovery"); |
| 65 | return STAGE_NAME; |
| 66 | } |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 67 | |
| 68 | private: |
| 69 | void |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 70 | doStart() override; |
| 71 | |
| 72 | void |
Junxiao Shi | a889111 | 2016-12-06 21:11:33 +0000 | [diff] [blame] | 73 | collectMulticastFaces(); |
| 74 | |
| 75 | void |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 76 | registerHubDiscoveryPrefix(const std::vector<nfd::FaceStatus>& dataset); |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 77 | |
| 78 | void |
| 79 | onRegisterSuccess(); |
| 80 | |
| 81 | void |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 82 | onRegisterFailure(const nfd::ControlResponse& response); |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 83 | |
| 84 | void |
| 85 | setStrategy(); |
| 86 | |
| 87 | void |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 88 | onSetStrategyFailure(const nfd::ControlResponse& response); |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 89 | |
| 90 | // Start to look for a hub (NDN hub discovery first stage) |
| 91 | void |
| 92 | requestHubData(); |
| 93 | |
| 94 | void |
Junxiao Shi | d0e276f | 2016-12-14 23:13:39 +0000 | [diff] [blame] | 95 | onSuccess(const Data& data); |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 96 | |
| 97 | private: |
Junxiao Shi | cb76686 | 2017-07-07 22:21:04 +0000 | [diff] [blame^] | 98 | Face& m_face; |
| 99 | nfd::Controller& m_controller; |
Junxiao Shi | 52fa45c | 2016-11-29 21:18:13 +0000 | [diff] [blame] | 100 | size_t m_nRequestedRegs; |
| 101 | size_t m_nFinishedRegs; |
Alexander Afanasyev | 2a655f7 | 2015-01-26 18:38:33 -0800 | [diff] [blame] | 102 | }; |
| 103 | |
| 104 | } // namespace autoconfig |
| 105 | } // namespace tools |
| 106 | } // namespace ndn |
| 107 | |
| 108 | #endif // NFD_TOOLS_NDN_AUTOCONFIG_MULTICAST_DISCOVERY_HPP |