diff options
Diffstat (limited to 'server/src/server.cpp')
| -rw-r--r-- | server/src/server.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/server/src/server.cpp b/server/src/server.cpp index 3c769d3..ef60cf4 100644 --- a/server/src/server.cpp +++ b/server/src/server.cpp @@ -28,7 +28,7 @@ #include "../../protocol/udm.pb.h" #include "socket_wraper.h" -extern std::list<server_session*> sessions; +extern lockable_container<std::list<server_session*> > sessions; @@ -51,7 +51,9 @@ server::server(boost::asio::io_service& io_service,runtime_config_s &config, std void server::start_accept() { server_session* new_session = new server_session(io_service_, runtime_config, clients, downloads); + sessions.lock(); sessions.push_back(new_session); + sessions.unlock(); if(runtime_config.config_file.get<bool>("server.enable_encryption", false)) acceptor_.async_accept(new_session->socket()->get_ssl_socket().lowest_layer(), boost::bind(&server::handle_accept, this, new_session, boost::asio::placeholders::error)); else @@ -67,9 +69,11 @@ void server::handle_accept(server_session* new_session, const boost::system::err else { //TODO: implement thread safety ? + sessions.lock(); auto s = std::find(sessions.begin(), sessions.end(), new_session); if(s != sessions.end()) sessions.erase(s); + sessions.unlock(); delete new_session; } start_accept(); |