diff options
Diffstat (limited to 'server')
-rw-r--r-- | server/docs/udm.conf.sample | 12 | ||||
-rw-r--r-- | server/include/config.h | 1 | ||||
-rw-r--r-- | server/modules/downloaders/curl/curl.cbp | 1 | ||||
-rw-r--r-- | server/src/main.cpp | 52 | ||||
-rw-r--r-- | server/src/modules_handler.cpp | 22 | ||||
-rw-r--r-- | server/src/server_session.cpp | 34 |
6 files changed, 104 insertions, 18 deletions
diff --git a/server/docs/udm.conf.sample b/server/docs/udm.conf.sample index 97b9cbb..e26b524 100644 --- a/server/docs/udm.conf.sample +++ b/server/docs/udm.conf.sample @@ -4,14 +4,14 @@ server ;section for core server settings { ;port 6613; server port ;password "client_passwod"; client password to connect - ;default_download_directory ~/udm_downloads ; default directory to store downloads + ;default_download_directory ~/udm/downloads ; default directory to store downloads ;default_metadata_module flat_files_metadata ; first loaded metadata module used by default if not set - ;default_data_storage_module <not set> ; first loaded data_storage module used by default if not set - ;daemon false ; for to background on startup (0 or 1) - ;verbosiry false; set server log level (higher is more noisy, also affects modules) - ;enable_encription 0; use tls encryption for client<-> server connection + ;default_data_storage_module <not set> ; first loaded data_storage module used by default if not set (not implemented) + ;daemon false ; for to background on startup (false or true) + ;verbosity 0; set server log level (higher is more noisy, also affects modules) + ;enable_encryption 0; use tls encryption for client<-> server connection ;ssl_certificate /path/to/cert/file.pem ; path to server certificate ;ssl_key /path/to/key/file.pem ; server certificate key + ;ssl_ca /path/to/ca/file.pem ; ca file to verify clients ;ssl_dh /path/to/dh/file ; dh parameters for ssl encryption - ;download_dir ~/.udm/downloads ; directory for downloads } diff --git a/server/include/config.h b/server/include/config.h index 2afbde3..74112ad 100644 --- a/server/include/config.h +++ b/server/include/config.h @@ -48,6 +48,7 @@ struct runtime_config_s { default_metadata_storage = nullptr; } + std::string config_file_path; }; diff --git a/server/modules/downloaders/curl/curl.cbp b/server/modules/downloaders/curl/curl.cbp index b0971b5..cec0d49 100644 --- a/server/modules/downloaders/curl/curl.cbp +++ b/server/modules/downloaders/curl/curl.cbp @@ -41,6 +41,7 @@ </Compiler> <Linker> <Add library="boost_system" /> + <Add library="curl" /> </Linker> <Unit filename="include/curl_download.h" /> <Unit filename="main.cpp" /> diff --git a/server/src/main.cpp b/server/src/main.cpp index 2786435..b856fd4 100644 --- a/server/src/main.cpp +++ b/server/src/main.cpp @@ -23,6 +23,8 @@ #include <iostream> #include <map> +#include <signal.h> + #include "api_module_metadata_storage.h" #include "modules_handler.h" #include "utilities.h" @@ -47,9 +49,15 @@ server *serv = nullptr; //TODO: "core" config section architecture, define base settings +void sigint_handler(int sig) +{ + bpt::write_info(runtime_config.config_file_path, runtime_config.config_file); //save config on sigint + exit(0); +} -int main(int argc, char *argv[]) +extern "C" int main(int argc, char *argv[]) { + signal(SIGINT, sigint_handler); bpo::options_description desc("Available commands and options"); desc.add_options() ("help,h", "this message") @@ -78,6 +86,48 @@ int main(int argc, char *argv[]) bpt::read_info(parsed_config_path, runtime_config.config_file); //TODO: finish this else std::cerr<<"failed to load config: \"" + config_path + "\", file does not exists or is not regular file\n"; + runtime_config.config_file_path = parsed_config_path; + } + { + auto server_pt = runtime_config.config_file.find("server"); + if(server_pt == runtime_config.config_file.not_found()) + { + runtime_config.config_file.put("server", ""); + server_pt = runtime_config.config_file.find("server"); + } + auto it = server_pt->second.find("port"); + if(it == server_pt->second.not_found()) + server_pt->second.put("port", 6613); + it = server_pt->second.find("password"); + if(it == server_pt->second.not_found()) + server_pt->second.put("password", ""); + it = server_pt->second.find("default_download_directory"); + if(it == server_pt->second.not_found()) + server_pt->second.put("default_download_directory", "~/udm/downloads"); + it = server_pt->second.find("default_metadata_module"); + if(it == server_pt->second.not_found()) + server_pt->second.put("default_metadata_module", "flat_files_metadata"); + it = server_pt->second.find("daemon"); + if(it == server_pt->second.not_found()) + server_pt->second.put("daemon", false); + it = server_pt->second.find("verbosity"); + if(it == server_pt->second.not_found()) + server_pt->second.put("verbosity", 0); + it = server_pt->second.find("enable_encryption"); + if(it == server_pt->second.not_found()) + server_pt->second.put("enable_encryption", false); + it = server_pt->second.find("ssl_certificate"); + if(it == server_pt->second.not_found()) + server_pt->second.put("ssl_certificate", "~/config/udm/cert.crt"); + it = server_pt->second.find("ssl_key"); + if(it == server_pt->second.not_found()) + server_pt->second.put("ssl_key", "~/config/udm/cert.key"); + it = server_pt->second.find("ssl_ca"); + if(it == server_pt->second.not_found()) + server_pt->second.put("ssl_ca", "~/config/udm/ca.crt"); + it = server_pt->second.find("ssl_dh"); + if(it == server_pt->second.not_found()) + server_pt->second.put("ssl_dh", "~/config/udm/dh.pem"); } //load all config variables here runtime_config.settings.verbosity = runtime_config.config_file.get<short>("server.verbosity", 0); diff --git a/server/src/modules_handler.cpp b/server/src/modules_handler.cpp index 78a6ad7..32c3b9f 100644 --- a/server/src/modules_handler.cpp +++ b/server/src/modules_handler.cpp @@ -123,21 +123,21 @@ std::string modules_handler::list_modules() std::string modules_handler::list_modules_single_type_internal(const std::list<module_base*> &modules) { std::string buf; - for(auto i = modules.begin(), end = modules.end(); i != end; ++i) + for(auto i : modules) { buf += "\tName: "; - buf += (*i)->get_module_info().name; - buf += "\n\tDescription: " +(*i)->get_module_info().description; - buf += "\n\tVersion: " + (*i)->get_module_info().version; + buf += i->get_module_info().name; + buf += "\n\tDescription: " + i->get_module_info().description; + buf += "\n\tVersion: " + i->get_module_info().version; if(runtime_config.settings.verbosity >= 1) { buf += "\n\tAvailable options:"; - for(auto i1 = (*i)->get_module_info().default_settings.begin(), end = (*i)->get_module_info().default_settings.end(); i1 != end; ++i1) + for(auto i1 : i->get_module_info().default_settings) { buf += "\n\t\t"; - buf += i1->first; + buf += i1.first; buf += " = "; - buf += i1->second; + buf += i1.second; } } buf += "\n\n"; @@ -147,12 +147,12 @@ std::string modules_handler::list_modules_single_type_internal(const std::list<m void modules_handler::sync_module_settings(module_base *m) { - for(auto i = m->get_module_info().default_settings.begin(), end = m->get_module_info().default_settings.end(); i != end; ++i) + for(auto i : m->get_module_info().default_settings) { - std::string setting = "modules." + m->get_module_info().name + "." + i->first; - std::string current_val = runtime_config.config_file.get<std::string>(setting, "not set"), default_val = i->second; + std::string setting = "modules." + m->get_module_info().name + "." + i.first; + std::string current_val = runtime_config.config_file.get<std::string>(setting, "not set"), default_val = i.second; if(current_val == "not set" || current_val != default_val) - runtime_config.config_file.put(setting, i->second); + runtime_config.config_file.put(setting, i.second); } } diff --git a/server/src/server_session.cpp b/server/src/server_session.cpp index 25e7820..64f2cfe 100644 --- a/server/src/server_session.cpp +++ b/server/src/server_session.cpp @@ -190,6 +190,40 @@ void server_session::handle_command(client_msg *msg) send_message(&m); } break; + case CLIENT_MSG_TYPE::CLIENT_MODULES_REQUEST: + { + //TODO: + } + break; + case CLIENT_MSG_TYPE::CLIENT_CORE_INFO_REQUEST: + { + server_msg msg; + msg.set_type(SERVER_MSG_TYPE::SERVER_CORE_INFO_REPLY); + msg.mutable_server_core_info_reply()->set_version(1); + try{ + for(auto it : runtime_config.config_file.get_child("server")) //load server node + { + std::string val = it.second.get_value<std::string>("empty_value"); //TODO: something better here. we need to avoid subtrees and empty vars + if(val == "" || val == "empty_value") + continue; + setting *i = msg.mutable_server_core_info_reply()->add_settings(); + i->set_name(it.first); + //TODO: is it possible to set something better than just list of core setting names ? + } + } + catch(...) + { + //TODO: + } + send_message(&msg); + + } + break; + case CLIENT_MSG_TYPE::CLIENT_DOWNLOADS_LIST_REQUEST: + { + //TODO: + } + break; default: break; } |