root/config.cc

/*  This file is part of Alue, the multiprotocol Internet discussion daemon

    Copyright © 2009, 2010 Antti-Juhani Kaijanaho

    Alue 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.

    Alue 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 Alue.  If not, see <http://www.gnu.org/licenses/>.

 */

#include "config.hh"

#include <cerrno>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <sys/resource.h>
#include <sys/utsname.h>
#include <unistd.h>


boost::program_options::variables_map config;

void configure(int argc, char *argv[], rlimit &rlim)
{
        namespace po = boost::program_options;

        if (getrlimit(RLIMIT_AS, &rlim) == -1)
        {
                std::cerr << "getrlimit failed: " << strerror(errno) 
                          << std::endl;
                std::exit(1);
        }

        struct utsname uts;
        if (uname(&uts) == -1)
        {
                std::cerr << "uname failed: " << strerror(errno) << std::endl;
                std::exit(1);
        }


        po::options_description opts;
        opts.add_options()
                ( "help", "show this help message")
                ( "executable",
                  po::value<std::string>()->default_value(argv[0]),
                  "the location of a debuggable executable (for backtraces)" )
                ( "daemon",
                  po::value<bool>()->default_value(true),
                  "create a daemon process and exit immediately afterward")
                ( "user", po::value<std::string>(),
                  "the user to run as (if started as root)")
                ( "canonical-name", 
                  po::value<std::string>()->default_value(uts.nodename),
                  "the canonical name of this host" )
                ("nntp-port",
                 po::value<std::string>()->default_value("119"),
                 "the port to listen on for NNTP clients")
                ("nntps-port",
                 po::value<std::string>()->default_value("563"),
                 "the port to listen on for NNTP/SSL clients")
                ("https-port",
                 po::value<std::string>()->default_value("443"),
                 "the port to listen on for HTTPS clients")
                ("unix-socket",
                 po::value<std::string>()->default_value("alue-socket"),
                 "the Unix socket (in the home directory) to listen for"
                 " email injections")
                ("key-file",
                 po::value<std::string>()->default_value("key.pem"),
                 "the file containing the SSL/TLS private key")
                ("cert-file",
                 po::value<std::string>()->default_value("cert.pem"),
                 "the file containing the SSL/TLS certificate")
                ("log-smtp",
                 po::value<bool>()->default_value(false),
                 "log SMTP extensively")
                ("log-http",
                 po::value<bool>()->default_value(false),
                 "log HTTP extensively (warning - user privacy invasive)")
                ("log-nntp",
                 po::value<bool>()->default_value(false),
                 "log HTTP extensively (warning - user privacy invasive)")
                ("logfile", 
                 po::value<std::string>()->default_value("alue.log"),
                 "the name of the logfile")
                ("vm-limit",
                 po::value<rlim_t>(&rlim.rlim_cur)->default_value(rlim.rlim_cur),
                 "the AS (virtual memory) resource limit for this process")
                ("include-dir", 
                 po::value<std::string>()->default_value("templates/"),
                 "include directory name")
                ("files-dir", 
                 po::value<std::string>()->default_value("templates/"),
                 "verbatim HTTPS file directory name")
                ("config-file",
                 po::value<std::string>()->default_value(CONFFILE),
                 "name of the the configuration file")
                ("db-file",
                 po::value<std::string>()->default_value("alue.db2"),
                 "name of the database file")
                ("pw-file",
                 po::value<std::string>()->default_value("passwds"),
                 "name of the passwords file")
                ("home",
                 po::value<std::string>()->default_value("/"),
                 "name of the daemon home directory")
                ("alue-name",
                 po::value<std::string>()->default_value("Alue"),
                 "the 'real name' Alue should use of itself")
                ("operator-email", 
                 po::value<std::string>(),
                 "a human operator's email address");
        po::store(po::parse_command_line(argc, argv, opts), config);
        {
                std::string config_file_name =
                        config["config-file"].as<std::string>();
                std::ifstream config_file(config_file_name.c_str());
                if (config_file)
                        po::store(po::parse_config_file(config_file, opts),
                                  config);
        }
        po::notify(config);

        if (config.count("help") > 0)
        {
                std::cout << opts << std::endl;
                std::exit(0);
        }
}