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 <boost/timer/timer.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) {
    if (std::find(m_plugin_directories.begin(), m_plugin_directories.end(), element.second.data()) == m_plugin_directories.end())
     m_plugin_directories.push_back(element.second.data());
    else
     throw std::runtime_error("Found double plugin-directory element: "s + 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) {
       if (site_struct.name == "")
        site_struct.name = x.second.data();
       else
        throw std::runtime_error("Found double site name: "s + x.second.data());
      } else if (x.first == "host"s) {
       if (std::find(site_struct.hosts.begin(), site_struct.hosts.end(), x.second.data()) == site_struct.hosts.end())
        site_struct.hosts.insert(x.second.data());
       else
        throw std::runtime_error("Found double site host element: "s + 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>
         if (param.first == "auth") {
          try {
           std::string login{param.second.get<std::string>("<xmlattr>.login")};
           if (path.auth.find(login) == path.auth.end()) {
            std::string password{param.second.get<std::string>("<xmlattr>.password")};
            path.auth[login] = password;
           } else
            throw std::runtime_error("Found double auth (login): "s + login);
          } catch (const std::exception& ex) {
           std::cerr << "Warning: Can't read auth data from config: " << ex.what() << std::endl;
          }
         } else {
          if (path.params.find(param.first) == path.params.end())
           path.params[param.first] = param.second.data();
          else
           throw std::runtime_error("Found double path param: "s + param.first + ": " + param.second.data());
         }
        }
       }
       if (std::find_if(site_struct.paths.begin(), site_struct.paths.end(), [&](const Path& p){ return p.requested == path.requested;}) == site_struct.paths.end())
        site_struct.paths.push_back(path);
       else
        throw std::runtime_error("Found double path spec: "s + path.requested);
      } else if (x.first == "certpath"s) {
       if (site_struct.cert_path == "")
        site_struct.cert_path = x.second.data();
       else
        throw std::runtime_error("Found double certpath: "s + x.second.data());
      } else if (x.first == "keypath"s) {
       if (site_struct.key_path == "")
        site_struct.key_path = x.second.data();
       else
        throw std::runtime_error("Found double keypath: "s + x.second.data());
      } else
       throw std::runtime_error("Unknown element: "s + x.first);
     }
     if (std::find_if(m_sites.begin(), m_sites.end(), [&](const Site& site){return site.name == site_struct.name;}) == m_sites.end())
      m_sites.push_back(site_struct);
     else
      throw std::runtime_error("Found double site spec: "s + site_struct.name);
    }
   } 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) {
       if (socket_struct.address == "")
        socket_struct.address = x.second.data();
       else
        throw std::runtime_error("Found double address spec: "s + x.second.data());
      } else if (x.first == "port"s) {
       if (socket_struct.port == "")
        socket_struct.port = x.second.data();
       else
        throw std::runtime_error("Found double port spec: "s + 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;
}

// throws std::out_of_range if not found
const Path& Config::GetPath(const Socket& socket, const std::string& requested_host, const std::string& requested_path) const
{
 //boost::timer::auto_cpu_timer t;

 // TODO: speed this up
 std::string host{requested_host};
 const Path* result{nullptr};
 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) &&
          "/?\0"s.find(requested_path[path.requested.size()]) != std::string::npos &&
          path.requested.size() > path_len)
      {
       path_len = path.requested.size();
       result = &path;
      }
     }
    }
   }
  }
 }

 if (result == nullptr)
  throw std::out_of_range("Path not found for "s + requested_host + " " + requested_path);

 return *result;
}

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