/Users/maurits/Documents/studie/afstuderen/biosphere/daemon/options.c

Go to the documentation of this file.

/*
 * Author: MA Hartman
 * Date: feb 14, 2007
 * 
 * Function:
 * Command-line and config file option parsing.
 * 
 * License information:
 * 
 * Copyright (c) 2006 Maurits Hartman
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use,
 * copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following
 * conditions:
 * 
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 * 
 */

#include "options.h"
#include "node.h"
#include "settings.h"
#include <biosphere.h>
#include <list.h>

#include <apr_pools.h>
#include <apr_getopt.h>
#include <apr_strings.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>


/* Global daemon settings and options: */
unsigned        global_listen_port;
unsigned        global_max_connections;
char       *global_conffile =                   NULL;
bs_list    *global_module_names_list =  NULL;
char       *global_modules_dir =                NULL;
bs_list    *global_node_address_list =  NULL;


static void add_node_to_global_list(const char *optarg, apr_pool_t *mp)
{
        char *addr;
        bs_uint16 port = 2915;
        unsigned i;
        bs_status rv;
        
        /* Get the address and port: */
        addr = apr_pstrdup(mp, optarg);
        for (i = 0; i < strlen(optarg); i++) {
                if (optarg[i] == ':') {
                        addr = apr_pstrndup(mp, optarg, i);
                        if (optarg[i + 1] != '\0') {
                                const char *tmp = optarg + i + 1;
                                port = (bs_uint16) atoi(tmp);
                        }
                        break;
                }
        }
        
        /* Add this node: */
        rv = add_node(addr, port);
        if (rv != BS_OK)
                fprintf(stderr, "Unable to add node: %s:%d\n", addr, port);
}


static void init_defaults(apr_pool_t *mp)
{
        /* Numerical type options: */
        global_listen_port =            BIOSPHERED_PORT;
        global_max_connections =        BIOSPHERED_MAX_CONNECTIONS;
        
        /* String type options: */
        global_conffile =               apr_pstrdup(mp, BIOSPHERED_CONFFILE);
        global_modules_dir =    apr_pstrdup(mp, BIOSPHERED_MODULES_DIR);
        
        /* Default modules is an empty list: */
        global_module_names_list = new_list(mp);
        global_node_address_list = new_list(mp);
}


bs_status init_options(int argc, const char *argv[], apr_pool_t *mp)
{
        int                             optch;
    const char     *optarg;
        apr_status_t    rv;
        apr_getopt_t   *opt;
        
        static const apr_getopt_option_t opt_option[] = {
        /* long-option, short-option, has-arg flag, description */
        {"help",                 '?', FALSE, "print help information"},
        {"version",              'V', FALSE, "print version information"},
        {"max-clients",  'i', TRUE,  "maximum number of clients"},
        {"with-config",  'c', TRUE,  "path of config file to use"},
        {"with-module",  'm', TRUE,  "path to module to load"},
        {"node",         'n', TRUE,  "address of another node (address[:port]?)"},
        {"port",                 'p', TRUE,  "the port the daemon listens on"},
        {NULL, 0, 0, NULL} /* end (a.k.a. sentinel) */
    };
    
    init_defaults(mp);
    rv = apr_getopt_init(&opt, mp, argc, argv);
    if (rv != APR_SUCCESS) return BS_ERROR;
    
    /* parse the all options based on opt_option[] */
    while ((rv = apr_getopt_long(opt, opt_option, &optch, &optarg)) == APR_SUCCESS) {
        char *tmp;
        
        switch (optch) {
        case '?':
            printf("Here comes help information\n");
            return 1;
            break;
        case 'V':
            printf(BIOSPHERE_VERSION_STRING "\n");
            return 1;
            break;
        case 'i':
            global_max_connections = atoi(optarg);
            printf("Incoming connection limit: %d\n", global_max_connections);
            break;
        case 'c':
                global_conffile = apr_pstrdup(mp, optarg);
                printf("Reading config options from: %s\n", global_conffile);
            break;
        case 'm':
                tmp = apr_pstrdup(mp, optarg);
                list_append(global_module_names_list, tmp);
            printf("Using module: %s\n", tmp);
            break;
        case 'n':
                add_node_to_global_list(optarg, mp);
                printf("Adding node: %s\n", optarg);
                break;
        case 'p':
                global_listen_port = atoi(optarg);
                printf("Incoming connection port: %d\n", global_listen_port);
                break;
        default:
                fprintf(stderr, "Bad command-line option provided: -%c %s\n", optch, optarg);
                return BS_ERROR;
        }
    }
    if (rv != APR_EOF) {
        fprintf(stderr, "Bad command-line option provided\n");
        return BS_ERROR;
    }
      
    return BS_OK;
}

---