mgmt refactoring: FibManager
Change-Id: Ied235d025cbe5059545c6482bfed0f38f987fdeb
Refs: #2107
diff --git a/daemon/mgmt/fib-manager.cpp b/daemon/mgmt/fib-manager.cpp
new file mode 100644
index 0000000..12374f1
--- /dev/null
+++ b/daemon/mgmt/fib-manager.cpp
@@ -0,0 +1,149 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+/**
+ * Copyright (c) 2014-2015, Regents of the University of California,
+ * Arizona Board of Regents,
+ * Colorado State University,
+ * University Pierre & Marie Curie, Sorbonne University,
+ * Washington University in St. Louis,
+ * Beijing Institute of Technology,
+ * The University of Memphis.
+ *
+ * This file is part of NFD (Named Data Networking Forwarding Daemon).
+ * See AUTHORS.md for complete list of NFD authors and contributors.
+ *
+ * NFD is free software: you can redistribute it and/or modify it under the terms
+ * of the GNU General Public License as published by the Free Software Foundation,
+ * either version 3 of the License, or (at your option) any later version.
+ *
+ * NFD 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "fib-manager.hpp"
+#include <ndn-cxx/management/nfd-fib-entry.hpp>
+
+namespace nfd {
+
+NFD_LOG_INIT("FibManager");
+
+FibManager::FibManager(Fib& fib,
+ function<shared_ptr<Face>(FaceId)> getFace,
+ Dispatcher& dispatcher,
+ CommandValidator& validator)
+ : ManagerBase(dispatcher, validator, "fib")
+ , m_fib(fib)
+ , m_getFace(getFace)
+{
+ registerCommandHandler<ndn::nfd::FibAddNextHopCommand>("add-nexthop",
+ bind(&FibManager::addNextHop, this, _2, _3, _4, _5));
+ registerCommandHandler<ndn::nfd::FibRemoveNextHopCommand>("remove-nexthop",
+ bind(&FibManager::removeNextHop, this, _2, _3, _4, _5));
+
+ registerStatusDatasetHandler("list", bind(&FibManager::listEntries, this, _1, _2, _3));
+}
+
+void
+FibManager::addNextHop(const Name& topPrefix, const Interest& interest,
+ ControlParameters parameters,
+ const ndn::mgmt::CommandContinuation& done)
+{
+ setFaceForSelfRegistration(interest, parameters);
+
+ const Name& prefix = parameters.getName();
+ FaceId faceId = parameters.getFaceId();
+ uint64_t cost = parameters.getCost();
+
+ NFD_LOG_TRACE("add-nexthop prefix: " << prefix
+ << " faceid: " << faceId
+ << " cost: " << cost);
+
+ auto face = m_getFace(faceId);
+ if (static_cast<bool>(face)) {
+ auto entry = m_fib.insert(prefix).first;
+
+ entry->addNextHop(face, cost);
+
+ NFD_LOG_DEBUG("add-nexthop result: OK"
+ << " prefix:" << prefix
+ << " faceid: " << faceId
+ << " cost: " << cost);
+
+ return done(ControlResponse(200, "Success").setBody(parameters.wireEncode()));
+ }
+ else {
+ NFD_LOG_INFO("add-nexthop result: FAIL reason: unknown-faceid: " << faceId);
+ return done(ControlResponse(410, "Face not found"));
+ }
+}
+
+void
+FibManager::removeNextHop(const Name& topPrefix, const Interest& interest,
+ ControlParameters parameters,
+ const ndn::mgmt::CommandContinuation& done)
+{
+ setFaceForSelfRegistration(interest, parameters);
+
+ NFD_LOG_TRACE("remove-nexthop prefix: " << parameters.getName()
+ << " faceid: " << parameters.getFaceId());
+
+ auto face = m_getFace(parameters.getFaceId());
+ if (static_cast<bool>(face)) {
+ auto entry = m_fib.findExactMatch(parameters.getName());
+ if (static_cast<bool>(entry)) {
+ entry->removeNextHop(face);
+ NFD_LOG_DEBUG("remove-nexthop result: OK prefix: " << parameters.getName()
+ << " faceid: " << parameters.getFaceId());
+
+ if (!entry->hasNextHops()) {
+ m_fib.erase(*entry);
+ }
+ }
+ else {
+ NFD_LOG_DEBUG("remove-nexthop result: OK");
+ }
+ }
+ else {
+ NFD_LOG_DEBUG("remove-nexthop result: OK");
+ }
+
+ done(ControlResponse(200, "Success").setBody(parameters.wireEncode()));
+}
+
+void
+FibManager::listEntries(const Name& topPrefix, const Interest& interest,
+ ndn::mgmt::StatusDatasetContext& context)
+{
+ for (auto&& entry : m_fib) {
+ auto prefix = entry.getPrefix();
+ ndn::nfd::FibEntry record;
+ const auto& nextHops = entry.getNextHops();
+
+ for (auto&& next : nextHops) {
+ ndn::nfd::NextHopRecord nextHopRecord;
+ nextHopRecord.setFaceId(next.getFace()->getId());
+ nextHopRecord.setCost(next.getCost());
+
+ record.addNextHopRecord(nextHopRecord);
+ }
+
+ record.setPrefix(prefix);
+ context.append(record.wireEncode());
+ }
+
+ context.end();
+}
+
+void
+FibManager::setFaceForSelfRegistration(const Interest& request, ControlParameters& parameters)
+{
+ bool isSelfRegistration = (parameters.getFaceId() == 0);
+ if (isSelfRegistration) {
+ parameters.setFaceId(request.getIncomingFaceId());
+ }
+}
+
+} // namespace
diff --git a/daemon/mgmt/fib-manager.hpp b/daemon/mgmt/fib-manager.hpp
new file mode 100644
index 0000000..7c66138
--- /dev/null
+++ b/daemon/mgmt/fib-manager.hpp
@@ -0,0 +1,82 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+/**
+ * Copyright (c) 2014-2015, Regents of the University of California,
+ * Arizona Board of Regents,
+ * Colorado State University,
+ * University Pierre & Marie Curie, Sorbonne University,
+ * Washington University in St. Louis,
+ * Beijing Institute of Technology,
+ * The University of Memphis.
+ *
+ * This file is part of NFD (Named Data Networking Forwarding Daemon).
+ * See AUTHORS.md for complete list of NFD authors and contributors.
+ *
+ * NFD is free software: you can redistribute it and/or modify it under the terms
+ * of the GNU General Public License as published by the Free Software Foundation,
+ * either version 3 of the License, or (at your option) any later version.
+ *
+ * NFD 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef NFD_DAEMON_MGMT_FIB_MANAGER_HPP
+#define NFD_DAEMON_MGMT_FIB_MANAGER_HPP
+
+#include "manager-base.hpp"
+#include "core/logger.hpp"
+#include "table/fib.hpp"
+#include "fw/forwarder.hpp"
+
+namespace nfd {
+
+/**
+ * @brief implement the FIB Management of NFD Management Protocol.
+ * @sa http://redmine.named-data.net/projects/nfd/wiki/FibMgmt
+ */
+class FibManager : public ManagerBase
+{
+public:
+ /**
+ * @brief construct a FibManger
+ *
+ * @param fib the managed FIB
+ * @param getFace a function used to retrive a face by FaceId from the face table
+ * @param dispatcher the management dispatcher
+ * @param validator the command validator
+ */
+ FibManager(Fib& fib,
+ function<shared_ptr<Face>(FaceId)> getFace,
+ Dispatcher& dispatcher,
+ CommandValidator& validator);
+
+private:
+ void
+ addNextHop(const Name& topPrefix, const Interest& interest,
+ ControlParameters parameters,
+ const ndn::mgmt::CommandContinuation& done);
+
+ void
+ removeNextHop(const Name& topPrefix, const Interest& interest,
+ ControlParameters parameters,
+ const ndn::mgmt::CommandContinuation& done);
+
+ void
+ listEntries(const Name& topPrefix, const Interest& interest,
+ ndn::mgmt::StatusDatasetContext& context);
+
+private:
+ void
+ setFaceForSelfRegistration(const Interest& request, ControlParameters& parameters);
+
+private:
+ Fib& m_fib;
+ function<shared_ptr<Face>(FaceId)> m_getFace;
+};
+
+} // namespace nfd
+
+#endif // NFD_DAEMON_MGMT_FIB_MANAGER_HPP