summaryrefslogtreecommitdiff
path: root/server/src/server_session.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'server/src/server_session.cpp')
-rw-r--r--server/src/server_session.cpp40
1 files changed, 38 insertions, 2 deletions
diff --git a/server/src/server_session.cpp b/server/src/server_session.cpp
index c9b8e91..d0ff26c 100644
--- a/server/src/server_session.cpp
+++ b/server/src/server_session.cpp
@@ -34,7 +34,7 @@ extern modules_handler *modules;
-server_session::server_session(boost::asio::io_service &s, runtime_config_s &config, std::map<std::string, client*> &clients_, boost::asio::ssl::context *c) : io_service_(s), context_(c), runtime_config(config), clients(clients_)
+server_session::server_session(boost::asio::io_service &s, runtime_config_s &config, std::map<std::string, client*> &clients_, std::map<int, download_internal_s> &downloads_, boost::asio::ssl::context *c) : io_service_(s), context_(c), runtime_config(config), clients(clients_), downloads(downloads_)
{
if(runtime_config.config_file.get<bool>("server.enable_encryption", false))
{
@@ -120,6 +120,12 @@ void server_session::handle_read(const boost::system::error_code& error, size_t
void server_session::handle_command(client_msg *msg)
{
+ if(msg->type() != CLIENT_MSG_TYPE::CLIENT_AUTH_REQUEST && clients.find(msg->auth_token()) == clients.end())
+ {
+ BOOST_LOG_TRIVIAL(error)<<"unauthorized client trying to execute command";
+ delete this; //unauthorized client, delete session
+ return;
+ }
switch(msg->type())
{
case CLIENT_MSG_TYPE::CLIENT_AUTH_REQUEST:
@@ -268,7 +274,37 @@ void server_session::handle_command(client_msg *msg)
break;
case CLIENT_MSG_TYPE::CLIENT_DOWNLOADS_LIST_REQUEST:
{
- //TODO:
+ //TODO: thread safety
+ //draft implementation, need a lot of optimizations
+ server_msg msg;
+ msg.set_type(SERVER_MSG_TYPE::SERVER_DOWNLOADS_LIST_REPLY);
+ for(auto i : downloads)
+ {
+ server_download_reply *r = msg.add_downloads();
+ download *d = r->mutable_download();
+ d->set_id(i.first); //set core_id for later access
+ d->set_module_name(i.second.module_name);
+ module_base *module = nullptr;
+ for(auto dm : modules->get_downloader_modules())
+ {
+ if(dm->get_module_info().name == i.second.module_name)
+ {
+ module = dm;
+ break;
+ }
+ }
+ if(!module)
+ {
+ //downloader module not found, this should be a fatal error
+ continue;
+ }
+ module_downloader *dm = static_cast<module_downloader*>(module);
+ download_s ds = dm->get_download(i.second.module_id);
+ d->set_downloaded(ds.downloaded);
+ d->set_size(ds.size);
+ d->set_name(ds.name);
+ }
+ send_message(&msg);
}
break;
default: