server: add read configuration file function
Change-Id: I6a3fe8208219a18e0a67b791dedb6b93d5e9caab
diff --git a/server/repo.cpp b/server/repo.cpp
new file mode 100644
index 0000000..0c3165a
--- /dev/null
+++ b/server/repo.cpp
@@ -0,0 +1,157 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+/**
+ * Copyright (C) 2014 Regents of the University of California.
+ * See COPYING for copyright and distribution information.
+ */
+
+#include "repo.hpp"
+
+namespace repo {
+
+RepoConfig
+parseConfig(const std::string& configPath)
+{
+ if (configPath.empty()) {
+ std::cerr << "configuration file path is empty" << std::endl;
+ }
+
+ std::ifstream fin(configPath.c_str());
+ if (!fin.is_open())
+ throw Repo::Error("failed to open configuration file '"+ configPath +"'");
+
+ using namespace boost::property_tree;
+ ptree propertyTree;
+ try {
+ read_info(fin, propertyTree);
+ }
+ catch (ptree_error& e) {
+ throw Repo::Error("failed to read configuration file '"+ configPath +"'");
+ }
+
+ ptree repoConf = propertyTree.get_child("repo");
+
+ RepoConfig repoConfig;
+
+ ptree dataConf = repoConf.get_child("data");
+
+ for (ptree::const_iterator it = dataConf.begin();
+ it != dataConf.end();
+ ++it)
+ {
+ if (it->first == "prefix")
+ repoConfig.dataPrefixes.push_back(Name(it->second.get_value<std::string>()));
+ else
+ throw Repo::Error("Unrecognized '" + it->first + "' option in 'data' section in "
+ "configuration file '"+ configPath +"'");
+ }
+
+ ptree commandConf = repoConf.get_child("command");
+ for (ptree::const_iterator it = commandConf.begin();
+ it != commandConf.end();
+ ++it)
+ {
+ if (it->first == "prefix")
+ repoConfig.repoPrefixes.push_back(Name(it->second.get_value<std::string>()));
+ else
+ throw Repo::Error("Unrecognized '" + it->first + "' option in 'command' section in "
+ "configuration file '"+ configPath +"'");
+ }
+
+ ptree tcpBulkInsert = repoConf.get_child("tcp_bulk_insert");
+ bool isTcpBulkEnabled = false;
+ std::string host = "localhost";
+ std::string port = "7376";
+ for (ptree::const_iterator it = tcpBulkInsert.begin();
+ it != tcpBulkInsert.end();
+ ++it)
+ {
+ isTcpBulkEnabled = true;
+
+ // tcp_bulk_insert {
+ // host "localhost" ; IP address or hostname to listen on
+ // port 7635 ; Port number to listen on
+ // }
+ if (it->first == "host") {
+ host = it->second.get_value<std::string>();
+ }
+ else if (it->first == "port") {
+ port = it->second.get_value<std::string>();
+ }
+ else
+ throw Repo::Error("Unrecognized '" + it->first + "' option in 'tcp_bulk_insert' section in "
+ "configuration file '"+ configPath +"'");
+ }
+ if (isTcpBulkEnabled) {
+ repoConfig.tcpBulkInsertEndpoints.push_back(std::make_pair(host, port));
+ }
+
+ if (repoConf.get<std::string>("storage.method") != "sqlite")
+ throw Repo::Error("Only 'sqlite' storage method is supported");
+
+ repoConfig.dbPath = repoConf.get<std::string>("storage.path");
+
+ repoConfig.validatorNode = repoConf.get_child("validator");
+ return repoConfig;
+}
+
+inline static void
+NullDeleter(boost::asio::io_service* variable)
+{
+ // do nothing
+}
+
+Repo::Repo(boost::asio::io_service& ioService, const RepoConfig& config)
+ : m_config(config)
+ , m_scheduler(ioService)
+ , m_face(shared_ptr<boost::asio::io_service>(&ioService, &NullDeleter))
+ , m_storageHandle(openStorage(config))
+ , m_readHandle(m_face, *m_storageHandle, m_keyChain, m_scheduler)
+ , m_writeHandle(m_face, *m_storageHandle, m_keyChain, m_scheduler, m_validator)
+ , m_deleteHandle(m_face, *m_storageHandle, m_keyChain, m_scheduler, m_validator)
+ , m_tcpBulkInsertHandle(ioService, *m_storageHandle)
+
+{
+ //Trust model not implemented, this is just an empty validator
+ //@todo add a function to parse RepoConfig.validatorNode and define the trust model
+ m_validator.addInterestRule("^<>",
+ *m_keyChain.
+ getCertificate(m_keyChain.getDefaultCertificateName()));
+}
+
+shared_ptr<StorageHandle>
+Repo::openStorage(const RepoConfig& config)
+{
+ shared_ptr<StorageHandle> storageHandle = ndn::make_shared<SqliteHandle>(config.dbPath);
+ return storageHandle;
+}
+
+void
+Repo::enableListening()
+{
+ // Enable "listening" on Data prefixes
+ for (vector<ndn::Name>::iterator it = m_config.dataPrefixes.begin();
+ it != m_config.dataPrefixes.end();
+ ++it)
+ {
+ m_readHandle.listen(*it);
+ }
+
+ // Enable "listening" on control prefixes
+ for (vector<ndn::Name>::iterator it = m_config.repoPrefixes.begin();
+ it != m_config.repoPrefixes.end();
+ ++it)
+ {
+ m_writeHandle.listen(*it);
+ m_deleteHandle.listen(*it);
+ }
+
+ // Enable listening on TCP bulk insert addresses
+ for (vector<pair<string, string> >::iterator it = m_config.tcpBulkInsertEndpoints.begin();
+ it != m_config.tcpBulkInsertEndpoints.end();
+ ++it)
+ {
+ m_tcpBulkInsertHandle.listen(it->first, it->second);
+ }
+}
+
+} // namespace repo