publisher: add LSDB dataset publishers
refs #2280
Change-Id: Ifb1920ef9b807610890b0d0a04f24141d39f011d
diff --git a/src/publisher/lsdb-dataset-interest-handler.cpp b/src/publisher/lsdb-dataset-interest-handler.cpp
new file mode 100644
index 0000000..5285d28
--- /dev/null
+++ b/src/publisher/lsdb-dataset-interest-handler.cpp
@@ -0,0 +1,103 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+/**
+ * Copyright (c) 2014-2015, The University of Memphis,
+ * Regents of the University of California,
+ * Arizona Board of Regents.
+ *
+ * This file is part of NLSR (Named-data Link State Routing).
+ * See AUTHORS.md for complete list of NLSR authors and contributors.
+ *
+ * NLSR 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.
+ *
+ * NLSR 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
+ * NLSR, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
+ **/
+
+#include "lsdb-dataset-interest-handler.hpp"
+
+#include "logger.hpp"
+
+#include <ndn-cxx/face.hpp>
+#include <ndn-cxx/management/nfd-control-response.hpp>
+#include <ndn-cxx/util/regex.hpp>
+
+namespace nlsr {
+
+INIT_LOGGER("LsdbDatasetInterestHandler");
+
+LsdbDatasetInterestHandler::LsdbDatasetInterestHandler(Lsdb& lsdb,
+ ndn::Face& face,
+ const ndn::Name& routerName,
+ ndn::KeyChain& keyChain)
+ : COMMAND_PREFIX(ndn::Name(routerName).append(Lsdb::NAME_COMPONENT))
+ , m_face(face)
+ , m_keyChain(keyChain)
+ , m_adjacencyLsaPublisher(lsdb, face, COMMAND_PREFIX, keyChain)
+ , m_coordinateLsaPublisher(lsdb, face, COMMAND_PREFIX, keyChain)
+ , m_nameLsaPublisher(lsdb, face, COMMAND_PREFIX, keyChain)
+ , m_lsdbStatusPublisher(lsdb, face, COMMAND_PREFIX, keyChain,
+ m_adjacencyLsaPublisher,
+ m_coordinateLsaPublisher,
+ m_nameLsaPublisher)
+
+{
+ _LOG_DEBUG("Setting interest filter for: " << COMMAND_PREFIX);
+ m_face.setInterestFilter(COMMAND_PREFIX,
+ std::bind(&LsdbDatasetInterestHandler::onInterest, this, _2));
+}
+
+void
+LsdbDatasetInterestHandler::onInterest(const ndn::Interest& interest)
+{
+ // Does interest match command prefix with one additional component?
+ if (interest.getName().size() != COMMAND_PREFIX.size() + 1 ||
+ !COMMAND_PREFIX.isPrefixOf(interest.getName()))
+ {
+ _LOG_DEBUG("Received malformed interest: " << interest.getName());
+
+ sendErrorResponse(interest.getName(), 400, "Malformed command");
+ return;
+ }
+
+ ndn::Name::Component command = interest.getName().get(COMMAND_PREFIX.size());
+ _LOG_TRACE("Received interest with command: " << command);
+
+ if (command.equals(AdjacencyLsaPublisher::DATASET_COMPONENT)) {
+ m_adjacencyLsaPublisher.publish();
+ }
+ else if (command.equals(CoordinateLsaPublisher::DATASET_COMPONENT)) {
+ m_coordinateLsaPublisher.publish();
+ }
+ else if (command.equals(NameLsaPublisher::DATASET_COMPONENT)) {
+ m_nameLsaPublisher.publish();
+ }
+ else if (command.equals(LsdbStatusPublisher::DATASET_COMPONENT)) {
+ m_lsdbStatusPublisher.publish();
+ }
+ else {
+ _LOG_DEBUG("Unsupported command: " << command);
+ sendErrorResponse(interest.getName(), 501, "Unsupported command");
+ }
+}
+
+void
+LsdbDatasetInterestHandler::sendErrorResponse(const ndn::Name& name,
+ uint32_t code,
+ const std::string& error)
+{
+ ndn::nfd::ControlResponse response(code, error);
+
+ std::shared_ptr<ndn::Data> data = std::make_shared<ndn::Data>(name);
+ data->setContent(response.wireEncode());
+
+ m_keyChain.sign(*data);
+ m_face.put(*data);
+}
+
+} // namespace nlsr