diff options
Diffstat (limited to 'src/bitz/config.cpp')
| -rw-r--r-- | src/bitz/config.cpp | 190 |
1 files changed, 190 insertions, 0 deletions
diff --git a/src/bitz/config.cpp b/src/bitz/config.cpp new file mode 100644 index 0000000..a9ca856 --- /dev/null +++ b/src/bitz/config.cpp @@ -0,0 +1,190 @@ +/* + * bitz-server, An ICAP server implementation in C++ + * Copyright (C) 2012 Uditha Atukorala + * + * This program 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. + * + * This program 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 this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#include "config.h" +#include "logger.h" + +#include <iostream> +#include <cstdlib> + + +namespace bitz { + + Config::Config() { + + // initialise config_t values + _config.port = 1344; + _config.pid_file = "/dev/null"; + _config.log_file = "/dev/null"; + _config.log_category = "bitz"; + _config.req_handlers_count = 0; + _config.req_handlers = NULL; + + _config.max_workers = 0; + _config.max_worker_requests = 0; + + + // defaults + _lconfig = NULL; + + } + + Config::~Config() { + + // cleanup + delete _lconfig; + + for (int i = 0; i < _config.req_handlers_count ; i++ ) { + if ( _config.req_handlers[i].modules_count > 0 ) { + delete [] _config.req_handlers[i].modules; + } + } + + if ( _config.req_handlers != NULL ) { + delete [] _config.req_handlers; + } + + } + + + const config_t &Config::initialise( const std::string &config_file ) { + + libconfig::Config * config; + config = new libconfig::Config; + + try { + config->readFile( config_file.c_str() ); + } catch( const libconfig::FileIOException &ex ) { + std::cerr << "[config] failed to read config file: " << config_file + << ", exception: " << ex.what() << std::endl; + exit( EXIT_FAILURE ); + } catch( const libconfig::ParseException &pex ) { + std::cerr << "[config] parse error at " << pex.getFile() + << ":" << pex.getLine() << " - " << pex.getError() << std::endl; + exit( EXIT_FAILURE ); + } + + try { + + // read core configs + config->lookupValue( "port", _config.port ); + config->lookupValue( "pid_file", _config.pid_file ); + config->lookupValue( "log_file", _config.log_file ); + config->lookupValue( "log_category", _config.log_category ); + + config->lookupValue( "max_workers", _config.max_workers ); + config->lookupValue( "max_worker_requests", _config.max_worker_requests ); + + } catch( const libconfig::SettingNotFoundException &e ) { + std::cerr << "[config] failed to load core configs, " + << e.getPath() << " : " << e.what() << std::endl; + } + + // cache configs + _lconfig = config; + + // read other configs + read_req_handler_configs(); + + return _config; + + } + + + const config_t &Config::configs() { + return _config; + } + + + const std::string Config::module_config( const std::string &module, const std::string &config ) throw() { + + std::string config_value = ""; + + if ( _lconfig->exists( "modules" ) ) { + + try { + libconfig::Setting &setting = _lconfig->lookup( std::string( "modules." ).append( module ) ); + setting.lookupValue( config, config_value ); + } catch( const libconfig::SettingNotFoundException &e ) { + // TODO: log errors ?? + std::cerr << "[config] " << e.getPath() << " : " << e.what() << std::endl; + } + + } else { + std::cout << "[config] 'modules' configs not found" << std::endl; + } + + return config_value; + + } + + + void Config::read_req_handler_configs() throw() { + + int i, j; + std::string s; + + std::cout << "[config] looking for req_handlers... "; + if ( _lconfig->exists( "req_handlers" ) ) { + + std::cout << "found "; + + libconfig::Setting &req_handlers = _lconfig->lookup( "req_handlers" ); + _config.req_handlers_count = req_handlers.getLength(); + _config.req_handlers = new req_handlers_config_t[_config.req_handlers_count]; + + std::cout << "(" << _config.req_handlers_count << ")" << std::endl; + + try { + + // read request handler configs + for ( i = 0; i < _config.req_handlers_count ; i++ ) { + _config.req_handlers[i].name = (const char *) req_handlers[i]["handler"]; + _config.req_handlers[i].class_name = (const char *) req_handlers[i]["class"]; + + // read request handler modules config + std::cout << "[config] looking for " << _config.req_handlers[i].name << " modules... "; + if ( req_handlers[i].exists( "modules" ) ) { + std::cout << "found "; + + _config.req_handlers[i].modules_count = req_handlers[i]["modules"].getLength(); + _config.req_handlers[i].modules = new modules_config_t[_config.req_handlers[i].modules_count]; + + std::cout << "(" << _config.req_handlers[i].modules_count << ")" << std::endl; + for ( j = 0; j < _config.req_handlers[i].modules_count; j++ ) { + _config.req_handlers[i].modules[j].name = (const char *) req_handlers[i]["modules"][j]["name"]; + _config.req_handlers[i].modules[j].module = (const char *) req_handlers[i]["modules"][j]["module"]; + } + } else { + std::cout << "not found" << std::endl; + } + } + + } catch ( const libconfig::SettingNotFoundException &ex ) { + std::cerr << "[config] Error: " << ex.getPath() << ex.what() << std::endl; + } + + } else { + std::cout << "not found" << std::endl; + } + + } + +} /* end of namespace bitz */ + |