X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fencode_server_finder.cc;h=6cdd8ce3cc11adb05062c9fd8fdc8744588e33b5;hb=36e81aebf03d00c716b5e603deb0b5f8f6a99f53;hp=aa76e396bdea39c360dc71f28a112531f5894ce9;hpb=17df947ac256397311a11894062070f8069c7e75;p=dcpomatic.git diff --git a/src/lib/encode_server_finder.cc b/src/lib/encode_server_finder.cc index aa76e396b..6cdd8ce3c 100644 --- a/src/lib/encode_server_finder.cc +++ b/src/lib/encode_server_finder.cc @@ -1,19 +1,20 @@ /* - Copyright (C) 2013-2015 Carl Hetherington + Copyright (C) 2013-2018 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic 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 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic 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., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ @@ -24,7 +25,7 @@ #include "cross.h" #include "encode_server_description.h" #include "dcpomatic_socket.h" -#include "raw_convert.h" +#include #include #include #include @@ -38,12 +39,13 @@ using std::cout; using boost::shared_ptr; using boost::scoped_array; using boost::weak_ptr; +using boost::optional; +using dcp::raw_convert; EncodeServerFinder* EncodeServerFinder::_instance = 0; EncodeServerFinder::EncodeServerFinder () - : _disabled (false) - , _search_thread (0) + : _search_thread (0) , _listen_thread (0) , _stop (false) { @@ -55,23 +57,49 @@ EncodeServerFinder::start () { _search_thread = new boost::thread (boost::bind (&EncodeServerFinder::search_thread, this)); _listen_thread = new boost::thread (boost::bind (&EncodeServerFinder::listen_thread, this)); +#ifdef DCPOMATIC_LINUX + pthread_setname_np (_search_thread->native_handle(), "encode-server-search"); + pthread_setname_np (_listen_thread->native_handle(), "encode-server-listen"); +#endif } + EncodeServerFinder::~EncodeServerFinder () +{ + stop (); +} + +void +EncodeServerFinder::stop () { _stop = true; _search_condition.notify_all (); if (_search_thread) { - DCPOMATIC_ASSERT (_search_thread->joinable ()); - _search_thread->join (); + /* Ideally this would be a DCPOMATIC_ASSERT(_search_thread->joinable()) but we + can't throw exceptions from a destructor. + */ + if (_search_thread->joinable ()) { + _search_thread->join (); + } } + delete _search_thread; + _search_thread = 0; _listen_io_service.stop (); if (_listen_thread) { - DCPOMATIC_ASSERT (_listen_thread->joinable ()); - _listen_thread->join (); + /* Ideally this would be a DCPOMATIC_ASSERT(_listen_thread->joinable()) but we + can't throw exceptions from a destructor. + */ + if (_listen_thread->joinable ()) { + _listen_thread->join (); + } } + delete _listen_thread; + _listen_thread = 0; + + boost::mutex::scoped_lock lm (_servers_mutex); + _servers.clear (); } void @@ -90,12 +118,13 @@ try socket.set_option (boost::asio::socket_base::broadcast (true)); string const data = DCPOMATIC_HELLO; + int const interval = 10; while (!_stop) { if (Config::instance()->use_any_servers ()) { /* Broadcast to look for servers */ try { - boost::asio::ip::udp::endpoint end_point (boost::asio::ip::address_v4::broadcast(), Config::instance()->server_port_base() + 1); + boost::asio::ip::udp::endpoint end_point (boost::asio::ip::address_v4::broadcast(), HELLO_PORT); socket.send_to (boost::asio::buffer (data.c_str(), data.size() + 1), end_point); } catch (...) { @@ -111,7 +140,7 @@ try } try { boost::asio::ip::udp::resolver resolver (io_service); - boost::asio::ip::udp::resolver::query query (*i, raw_convert (Config::instance()->server_port_base() + 1)); + boost::asio::ip::udp::resolver::query query (*i, raw_convert (HELLO_PORT)); boost::asio::ip::udp::endpoint end_point (*resolver.resolve (query)); socket.send_to (boost::asio::buffer (data.c_str(), data.size() + 1), end_point); } catch (...) { @@ -119,8 +148,31 @@ try } } + /* Discard servers that we haven't seen for a while */ + bool removed = false; + { + boost::mutex::scoped_lock lm (_servers_mutex); + + list::iterator i = _servers.begin(); + while (i != _servers.end()) { + if (i->last_seen_seconds() > 2 * interval) { + list::iterator j = i; + ++j; + _servers.erase (i); + i = j; + removed = true; + } else { + ++i; + } + } + } + + if (removed) { + emit (boost::bind (boost::ref (ServersListChanged))); + } + boost::mutex::scoped_lock lm (_search_condition_mutex); - _search_condition.timed_wait (lm, boost::get_system_time() + boost::posix_time::seconds (10)); + _search_condition.timed_wait (lm, boost::get_system_time() + boost::posix_time::seconds (interval)); } } catch (...) @@ -134,7 +186,9 @@ try { using namespace boost::asio::ip; try { - _listen_acceptor.reset (new tcp::acceptor (_listen_io_service, tcp::endpoint (tcp::v4(), Config::instance()->server_port_base() + 1))); + _listen_acceptor.reset ( + new tcp::acceptor (_listen_io_service, tcp::endpoint (tcp::v4(), is_batch_converter ? BATCH_SERVER_PRESENCE_PORT : MAIN_SERVER_PRESENCE_PORT)) + ); } catch (...) { boost::throw_exception (NetworkError (_("Could not listen for remote encode servers. Perhaps another instance of DCP-o-matic is running."))); } @@ -177,8 +231,11 @@ EncodeServerFinder::handle_accept (boost::system::error_code ec, shared_ptrread_string (s); string const ip = socket->socket().remote_endpoint().address().to_string (); - if (!server_found (ip) && xml->optional_number_child("Version").get_value_or (0) == SERVER_LINK_VERSION) { - EncodeServerDescription sd (ip, xml->number_child ("Threads")); + optional::iterator> found = server_found (ip); + if (found) { + (*found)->set_seen (); + } else { + EncodeServerDescription sd (ip, xml->number_child("Threads"), xml->optional_number_child("Version").get_value_or(0)); { boost::mutex::scoped_lock lm (_servers_mutex); _servers.push_back (sd); @@ -189,16 +246,20 @@ EncodeServerFinder::handle_accept (boost::system::error_code ec, shared_ptr::iterator> +EncodeServerFinder::server_found (string ip) { boost::mutex::scoped_lock lm (_servers_mutex); - list::const_iterator i = _servers.begin(); + list::iterator i = _servers.begin(); while (i != _servers.end() && i->host_name() != ip) { ++i; } - return i != _servers.end (); + if (i != _servers.end()) { + return i; + } + + return optional::iterator>(); } EncodeServerFinder*