nerix
97 lines
2.5 KiB
C++
97 lines
2.5 KiB
C++
#include "common/websockets/detail/WebSocketPoolImpl.hpp"
|
|
|
|
#include "Application.hpp"
|
|
#include "common/QLogging.hpp"
|
|
#include "common/websockets/detail/WebSocketConnection.hpp"
|
|
#include "util/RenameThread.hpp"
|
|
|
|
#include <boost/certify/https_verification.hpp>
|
|
|
|
namespace chatterino::ws::detail {
|
|
|
|
WebSocketPoolImpl::WebSocketPoolImpl()
|
|
: ioc(1)
|
|
, ssl(boost::asio::ssl::context::tls_client)
|
|
, work(this->ioc.get_executor())
|
|
{
|
|
boost::system::error_code ec;
|
|
auto _ = this->ssl.set_options(
|
|
boost::asio::ssl::context::no_tlsv1 |
|
|
boost::asio::ssl::context::no_tlsv1_1 |
|
|
boost::asio::ssl::context::default_workarounds |
|
|
boost::asio::ssl::context::single_dh_use,
|
|
ec);
|
|
if (ec)
|
|
{
|
|
qCWarning(chatterinoWebsocket) << "Failed to set SSL context options"
|
|
<< QString::fromStdString(ec.message());
|
|
}
|
|
|
|
#ifdef CHATTERINO_WITH_TESTS
|
|
if (!getApp()->isTest())
|
|
#endif
|
|
{
|
|
this->ssl.set_verify_mode(
|
|
boost::asio::ssl::verify_peer |
|
|
boost::asio::ssl::verify_fail_if_no_peer_cert);
|
|
this->ssl.set_default_verify_paths();
|
|
|
|
boost::certify::enable_native_https_server_verification(this->ssl);
|
|
}
|
|
|
|
this->ioThread = std::make_unique<std::thread>([this] {
|
|
this->ioc.run();
|
|
this->shutdownFlag.set();
|
|
});
|
|
renameThread(*this->ioThread, "WebSocketPool");
|
|
}
|
|
|
|
WebSocketPoolImpl::~WebSocketPoolImpl()
|
|
{
|
|
assert(this->closing);
|
|
// After 10s, we stop and, through std::thread::~thread, std::terminate will
|
|
// be called.
|
|
this->tryShutdown(std::chrono::seconds{10});
|
|
}
|
|
|
|
bool WebSocketPoolImpl::tryShutdown(std::chrono::milliseconds timeout)
|
|
{
|
|
this->closing = true;
|
|
if (!this->ioThread || !this->ioThread->joinable())
|
|
{
|
|
return true;
|
|
}
|
|
|
|
this->work.reset();
|
|
{
|
|
std::lock_guard g(this->connectionMutex);
|
|
for (const auto &conn : this->connections)
|
|
{
|
|
conn->close();
|
|
}
|
|
}
|
|
|
|
if (!this->shutdownFlag.waitFor(timeout))
|
|
{
|
|
qCWarning(chatterinoWebsocket)
|
|
<< "Failed to gracefully close all connections in time";
|
|
return false;
|
|
}
|
|
|
|
if (this->ioThread->joinable())
|
|
{
|
|
this->ioThread->join();
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void WebSocketPoolImpl::removeConnection(WebSocketConnection *conn)
|
|
{
|
|
std::lock_guard g(this->connectionMutex);
|
|
std::erase_if(this->connections, [conn](const auto &v) {
|
|
return v.get() == conn;
|
|
});
|
|
}
|
|
|
|
} // namespace chatterino::ws::detail
|