Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| 2 | /** |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 3 | * Copyright (c) 2014-2017, Regents of the University of California, |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [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 | #include "face-system.hpp" |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 27 | #include "protocol-factory.hpp" |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 28 | #include "fw/face-table.hpp" |
| 29 | |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 30 | namespace nfd { |
| 31 | namespace face { |
| 32 | |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 33 | NFD_LOG_INIT("FaceSystem"); |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 34 | |
| 35 | FaceSystem::FaceSystem(FaceTable& faceTable) |
| 36 | : m_faceTable(faceTable) |
| 37 | { |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 38 | for (const std::string& id : ProtocolFactory::listRegistered()) { |
| 39 | NFD_LOG_TRACE("creating factory " << id); |
| 40 | m_factories[id] = ProtocolFactory::create(id); |
| 41 | } |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 42 | } |
| 43 | |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 44 | FaceSystem::~FaceSystem() = default; |
| 45 | |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 46 | std::set<const ProtocolFactory*> |
| 47 | FaceSystem::listProtocolFactories() const |
| 48 | { |
| 49 | std::set<const ProtocolFactory*> factories; |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 50 | for (const auto& p : m_factories) { |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 51 | factories.insert(p.second.get()); |
| 52 | } |
| 53 | return factories; |
| 54 | } |
| 55 | |
| 56 | ProtocolFactory* |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 57 | FaceSystem::getFactoryById(const std::string& id) |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 58 | { |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 59 | auto found = m_factories.find(id); |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 60 | return found == m_factories.end() ? nullptr : found->second.get(); |
| 61 | } |
| 62 | |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 63 | ProtocolFactory* |
| 64 | FaceSystem::getFactoryByScheme(const std::string& scheme) |
| 65 | { |
| 66 | auto found = m_factoryByScheme.find(scheme); |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 67 | return found == m_factoryByScheme.end() ? nullptr : found->second; |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 68 | } |
| 69 | |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 70 | void |
| 71 | FaceSystem::setConfigFile(ConfigFile& configFile) |
| 72 | { |
| 73 | configFile.addSectionHandler("face_system", bind(&FaceSystem::processConfig, this, _1, _2, _3)); |
| 74 | } |
| 75 | |
| 76 | void |
| 77 | FaceSystem::processConfig(const ConfigSection& configSection, bool isDryRun, const std::string& filename) |
| 78 | { |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 79 | ConfigContext context; |
| 80 | context.isDryRun = isDryRun; |
| 81 | context.addFace = bind(&FaceTable::add, &m_faceTable, _1); |
Junxiao Shi | 7003c60 | 2017-01-10 13:35:28 +0000 | [diff] [blame] | 82 | context.m_netifs = listNetworkInterfaces(); |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 83 | |
| 84 | // process sections in protocol factories |
| 85 | for (const auto& pair : m_factories) { |
| 86 | const std::string& sectionName = pair.first; |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 87 | ProtocolFactory* factory = pair.second.get(); |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 88 | |
| 89 | std::set<std::string> oldProvidedSchemes = factory->getProvidedSchemes(); |
| 90 | factory->processConfig(configSection.get_child_optional(sectionName), context); |
| 91 | |
| 92 | if (!isDryRun) { |
| 93 | for (const std::string& scheme : factory->getProvidedSchemes()) { |
| 94 | m_factoryByScheme[scheme] = factory; |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 95 | if (oldProvidedSchemes.erase(scheme) == 0) { |
| 96 | NFD_LOG_TRACE("factory " << sectionName << |
| 97 | " provides " << scheme << " FaceUri scheme"); |
| 98 | } |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 99 | } |
| 100 | for (const std::string& scheme : oldProvidedSchemes) { |
| 101 | m_factoryByScheme.erase(scheme); |
Junxiao Shi | b47247d | 2017-01-24 15:09:16 +0000 | [diff] [blame] | 102 | NFD_LOG_TRACE("factory " << sectionName << |
| 103 | " no longer provides " << scheme << " FaceUri scheme"); |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 104 | } |
| 105 | } |
| 106 | } |
| 107 | |
| 108 | // process other sections |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 109 | std::set<std::string> seenSections; |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 110 | for (const auto& pair : configSection) { |
| 111 | const std::string& sectionName = pair.first; |
Junxiao Shi | 64d99f2 | 2017-01-21 23:06:36 +0000 | [diff] [blame] | 112 | // const ConfigSection& subSection = pair.second; |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 113 | |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 114 | if (!seenSections.insert(sectionName).second) { |
| 115 | BOOST_THROW_EXCEPTION(ConfigFile::Error("Duplicate section face_system." + sectionName)); |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 116 | } |
| 117 | |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 118 | if (m_factories.count(sectionName) > 0) { |
| 119 | continue; |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 120 | } |
Junxiao Shi | 38b24c7 | 2017-01-05 02:59:31 +0000 | [diff] [blame] | 121 | |
| 122 | ///\todo #3521 nicfaces |
| 123 | |
Junxiao Shi | 64d99f2 | 2017-01-21 23:06:36 +0000 | [diff] [blame] | 124 | BOOST_THROW_EXCEPTION(ConfigFile::Error("Unrecognized option face_system." + sectionName)); |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 125 | } |
| 126 | } |
| 127 | |
Junxiao Shi | b859031 | 2016-12-29 21:22:25 +0000 | [diff] [blame] | 128 | } // namespace face |
| 129 | } // namespace nfd |