mirror of
https://github.com/Icinga/icinga2.git
synced 2025-08-16 07:08:18 +02:00
506624442f
Adds a third coroutine that detects when a TLS-shutdown is initiated by the client. It also adds an optional `reason` parameter to `HttpServerConnection::Disconnect()` that adds the reason for the shutdown to the log message that is generated.
60 lines
1.5 KiB
C++
60 lines
1.5 KiB
C++
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
|
|
|
|
#ifndef HTTPSERVERCONNECTION_H
|
|
#define HTTPSERVERCONNECTION_H
|
|
|
|
#include "remote/apiuser.hpp"
|
|
#include "base/io-engine.hpp"
|
|
#include "base/string.hpp"
|
|
#include "base/tlsstream.hpp"
|
|
#include "base/wait-group.hpp"
|
|
#include <memory>
|
|
#include <boost/asio/deadline_timer.hpp>
|
|
#include <boost/asio/io_context.hpp>
|
|
#include <boost/asio/io_context_strand.hpp>
|
|
#include <boost/asio/spawn.hpp>
|
|
|
|
namespace icinga
|
|
{
|
|
|
|
/**
|
|
* An API client connection.
|
|
*
|
|
* @ingroup remote
|
|
*/
|
|
class HttpServerConnection final : public Object
|
|
{
|
|
public:
|
|
DECLARE_PTR_TYPEDEFS(HttpServerConnection);
|
|
|
|
HttpServerConnection(const WaitGroup::Ptr& waitGroup, const String& identity, bool authenticated,
|
|
const Shared<AsioTlsStream>::Ptr& stream);
|
|
|
|
void Start();
|
|
bool Disconnected();
|
|
|
|
private:
|
|
WaitGroup::Ptr m_WaitGroup;
|
|
ApiUser::Ptr m_ApiUser;
|
|
Shared<AsioTlsStream>::Ptr m_Stream;
|
|
double m_Seen;
|
|
AsioDualEvent m_CanRead;
|
|
String m_PeerAddress;
|
|
boost::asio::io_context::strand m_IoStrand;
|
|
bool m_ShuttingDown;
|
|
boost::asio::deadline_timer m_CheckLivenessTimer;
|
|
|
|
HttpServerConnection(const WaitGroup::Ptr& waitGroup, const String& identity, bool authenticated,
|
|
const Shared<AsioTlsStream>::Ptr& stream, boost::asio::io_context& io);
|
|
|
|
void Disconnect(boost::asio::yield_context yc, std::string_view reason = {});
|
|
|
|
void ProcessMessages(boost::asio::yield_context yc);
|
|
void CheckLiveness(boost::asio::yield_context yc);
|
|
void DetectClientShutdown(boost::asio::yield_context yc);
|
|
};
|
|
|
|
}
|
|
|
|
#endif /* HTTPSERVERCONNECTION_H */
|