path: root/config.cpp

#include "config.h"

#include <boost/algorithm/string/predicate.hpp>
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/xml_parser.hpp>

#include <algorithm>
#include <exception>
#include <iostream>

namespace pt = boost::property_tree;
using namespace std::string_literals;

namespace {

 void RemovePortFromHostname(std::string& hostname)
 {
  // besides hostnames and IPv4 addresses, consider IPv6 addresses: [xx::yy]:8080
  // so only remove ":N" if after "]"
  size_t pos = hostname.find_last_of(":]");
  if (pos != hostname.npos && hostname[pos] == ':') {
   hostname = hostname.substr(0, pos);
  }
 }

} // anonymous namespace

void Config::readConfigfile(std::string filename)
{
 if (filename == "") {
  filename = default_filename;
 }

 pt::ptree tree;

 pt::read_xml(filename, tree, pt::xml_parser::no_comments | pt::xml_parser::trim_whitespace);

 // mandatory
 m_user = tree.get<std::string>("webserver.user");
 
 m_group = tree.get<std::string>("webserver.group");

 m_threads = tree.get<int>("webserver.threads");

 // optional entries
 auto elements = tree.get_child_optional("webserver");
 if (elements) {
  for (const auto& element: *elements) {
   if (element.first == "plugin-directory"s) {
    m_plugin_directories.push_back(element.second.data());
   } else if (element.first == "sites"s) {
    for (const auto& site: element.second) {
     if (site.first != "site"s)
      throw std::runtime_error("<site> expected in <sites>");
     Site site_struct;
     for (const auto& x: site.second) {
      if (x.first == "name"s) {
       site_struct.name = x.second.data();
      } else if (x.first == "host"s) {
       site_struct.hosts.insert(x.second.data());
      } else if (x.first == "path"s) {
       Path path;
       auto attrs = x.second.get_child("<xmlattr>");
       path.requested = attrs.get<std::string>("requested");
       for (const auto& param: x.second) { // get all sub-elements of <path>
        if (param.first.size() > 0 && param.first[0] != '<') // exclude meta-elements like <xmlattr>
         path.params[param.first.data()] = param.second.data();
       }
       site_struct.paths.push_back(path);
      } else if (x.first == "certpath"s) {
       site_struct.cert_path = x.second.data();
      } else if (x.first == "keypath"s) {
       site_struct.key_path = x.second.data();
      } else
       throw std::runtime_error("Unknown element: "s + x.first);
     }
     m_sites.push_back(site_struct);
    }
   } else if (element.first == "sockets"s) {
    for (const auto& socket: element.second) {
     if (socket.first != "socket"s)
      throw std::runtime_error("<socket> expected in <sockets>");
     Socket socket_struct;
     for (const auto& x: socket.second) {
      if (x.first == "address"s) {
       socket_struct.address = x.second.data();
      } else if (x.first == "port"s) {
       socket_struct.port = x.second.data();
      } else if (x.first == "protocol"s) {
       if (x.second.data() == "http"s)
        socket_struct.protocol = SocketProtocol::HTTP;
       else if (x.second.data() == "https"s)
        socket_struct.protocol = SocketProtocol::HTTPS;
       else
        throw std::runtime_error("Unknown protocol: "s + x.second.data());
      } else
       throw std::runtime_error("Unknown element: "s + x.first);
     }
     if (geteuid() != 0 && stoi(socket_struct.port) < 1024)
      std::cout << "Warning: Skipping privileged port " << socket_struct.port << std::endl;
     else
      m_sockets.push_back(socket_struct);
    }
   }
  }
 }

 // expand socket sites
 for (auto& socket: m_sockets) {
  if (socket.serve_sites.empty()) {
   for (const auto& site: m_sites) {
    socket.serve_sites.push_back(site.name);
   }
  }
 }
}

Config::Config(const std::string& filename)
{
 readConfigfile(filename);
 dump();
}

std::string Config::User() const
{
 return m_user;
}

std::string Config::Group() const
{
 return m_group;
}

int Config::Threads() const
{
 return m_threads;
}

const std::vector<std::string>& Config::PluginDirectories() const
{
 return m_plugin_directories;
}

const std::vector<Site>& Config::Sites() const
{
 return m_sites;
}

const std::vector<Socket>& Config::Sockets() const
{
 return m_sockets;
}

void Config::dump() const
{
 std::cout << "=== Configuration ===========================" << std::endl;
 std::cout << "User: " << m_user << std::endl;
 std::cout << "Group: " << m_user << std::endl;
 
 std::cout << "Threads: " << m_threads << std::endl;

 std::cout << "Plugin Directories:";
 for (const auto& dir: m_plugin_directories)
  std::cout << " " << dir;
 std::cout << std::endl;

 for (const auto& site: m_sites) {
  std::cout << "Site: " << site.name << ":";
  for (const auto& host: site.hosts)
   std::cout << " " << host;
  std::cout << std::endl;
  if (site.paths.size() == 0)
   std::cout << "  Warning: No paths configured." << std::endl;
  for (const auto& path: site.paths) {
   std::cout << "  Path: " << path.requested << std::endl;
   for (const auto& param: path.params) {
    std::cout << "    " << param.first << ": " << param.second << std::endl;
   }
  }
  if (site.key_path != ""s) {
   std::cout << "  Key: " << site.key_path.generic_string() << std::endl;
   std::cout << "  Cert: " << site.cert_path.generic_string() << std::endl;
  }
 }

 for (const auto& socket: m_sockets) {
  std::cout << "Socket: " << socket.address << ":" << socket.port << " (" << (socket.protocol == SocketProtocol::HTTP ? "HTTP" : "HTTPS") << ")" << std::endl;
  std::cout << "  Serving:";
  for (const auto& site: socket.serve_sites) {
   std::cout << " " << site;
  }
  std::cout << std::endl;
 }
 std::cout << "=============================================" << std::endl;
}

std::string Config::DocRoot(const Socket& socket, const std::string& requested_host, const std::string& requested_path) const
{
 // TODO: speed this up
 std::string host{requested_host};
 std::string result;
 size_t path_len{0}; // find longest matching prefix

 RemovePortFromHostname(host);

 for (const auto& site: m_sites) {
  if (std::find(socket.serve_sites.begin(), socket.serve_sites.end(), site.name) != socket.serve_sites.end()) {
   for (const auto& m_host: site.hosts) {
    if (m_host == host) {
     for (const auto& path: site.paths) {
      if (boost::starts_with(requested_path, path.requested) && path.requested.size() > path_len) {
       path_len = path.requested.size();
       try {
        result = path.params.at("target");
       } catch (const std::out_of_range& ex) {
        std::cout << "Out of range at Config::DocRoot(): target" << std::endl;
        std::rethrow_exception(std::current_exception());
       }
      }
     }
    }
   }
  }
 }

 return result;
}

std::string Config::GetPlugin(const Socket& socket, const std::string& requested_host, const std::string& requested_path) const
{
 // TODO: speed this up
 std::string host{requested_host};
 std::string result;
 size_t path_len{0};

 RemovePortFromHostname(host);

 for (const auto& site: m_sites) {
  if (std::find(socket.serve_sites.begin(), socket.serve_sites.end(), site.name) != socket.serve_sites.end()) {
   for (const auto& m_host: site.hosts) {
    if (m_host == host) {
     for (const auto& path: site.paths) {
      if (boost::starts_with(requested_path, path.requested) && path.requested.size() > path_len) {
       path_len = path.requested.size();
       try {
        result = path.params.at("plugin");
       } catch (const std::out_of_range& ex) {
        std::cout << "Out of range at Config::DocRoot(): target" << std::endl;
        std::rethrow_exception(std::current_exception());
       }
      }
     }
    }
   }
  }
 }

 return result;
}

std::string Config::GetPluginPath(const Socket& socket, const std::string& requested_host, const std::string& requested_path) const
{
 // TODO: speed this up
 std::string host{requested_host};
 std::string result;
 size_t path_len{0};

 RemovePortFromHostname(host);

 for (const auto& site: m_sites) {
  if (std::find(socket.serve_sites.begin(), socket.serve_sites.end(), site.name) != socket.serve_sites.end()) {
   for (const auto& m_host: site.hosts) {
    if (m_host == host) {
     for (const auto& path: site.paths) {
      if (boost::starts_with(requested_path, path.requested) && path.requested.size() > path_len) {
       path_len = path.requested.size();
       result = path.requested;
      }
     }
    }
   }
  }
 }

 return result;
}

std::string Config::GetRelativePath(const Socket& socket, const std::string& requested_host, const std::string& requested_path) const
{
 // TODO: speed this up
 std::string host{requested_host};
 std::string result;
 size_t path_len{0};

 RemovePortFromHostname(host);

 for (const auto& site: m_sites) {
  if (std::find(socket.serve_sites.begin(), socket.serve_sites.end(), site.name) != socket.serve_sites.end()) {
   for (const auto& m_host: site.hosts) {
    if (m_host == host) {
     for (const auto& path: site.paths) {
      if (boost::starts_with(requested_path, path.requested) && path.requested.size() > path_len) {
       path_len = path.requested.size();
       result = requested_path.substr(path_len);
      }
     }
    }
   }
  }
 }

 if (result.size() > 0 && result[0] == '/')
  return result.substr(1);
 return result;
}

bool Config::PluginIsConfigured(const std::string& name) const
{
 for (const auto& site: m_sites) {
  for (const auto& path: site.paths) {
   if (path.params.find("plugin") != path.params.end())
    return true;
  }
 }
 return false;
}