mirror of
https://github.com/Icinga/icinga2.git
synced 2026-03-21 18:11:32 -04:00
91c7e60df8
I've used the following command to replace the original copyright header
lines in a C-style comment block:
```
$ find . \( -type d \( -name '\..*' -o -name third-party -o -name scripts -o -name prefix -o -name malloc -o -name server -o -name docker -o -name build -o -name doc \) -prune \) -o -type f -exec perl -pi -e 's{/\*[^*]*\(\s*c\s*\)\s*(\d{4})\s*Icinga\s+GmbH[^*]*\*/}{// SPDX-FileCopyrightText: \1 Icinga GmbH <https://icinga.com>\n// SPDX-License-Identifier: GPL-2.0-or-later}gi' {} +
```
For files that use shell-style comments (#) like CMakeLists.txt, I've
used this command:
```
$ find . \( -type d \( -name '\..*' -o -name third-party -o -name scripts -o -name prefix -o -name malloc -o -name server -o -name docker -o -name build -o -name doc \) -prune \) -o -type f -exec perl -pi -e 's{#.*\(\s*c\s*\)\s(\d{4})\sIcinga\s+GmbH.*}{# SPDX-FileCopyrightText: \1 Icinga GmbH <https://icinga.com>\n# SPDX-License-Identifier: GPL-2.0-or-later}gi' {} +
```
And for SQL files:
```
$ find . \( -type d \( -name '\..*' -o -name third-party -o -name scripts -o -name prefix -o -name malloc -o -name server -o -name docker -o -name build -o -name doc \) -prune \) -o -type f \( -name '*.sql' \) -exec perl -pi -e 's{--.*\(c\)\s(\d{4})\sIcinga\sGmbH.*}{-- SPDX-FileCopyrightText: \1 Icinga GmbH <https://icinga.com>\n-- SPDX-License-Identifier: GPL-2.0-or-later}gi' {} +
$ find . \( -type d \( -name '\..*' -o -name third-party -o -name scripts -o -name prefix -o -name malloc -o -name server -o -name docker -o -name build -o -name doc \) -prune \) -o -type f \( -name '*.sql' \) -exec perl -pi -e 's{-- Copyright \(c\)\s(\d{4})\sIcinga\s+Development\sTeam.*}{-- SPDX-FileCopyrightText: \1 Icinga GmbH <https://icinga.com>\n-- SPDX-License-Identifier: GPL-2.0-or-later}gi' {} +
```
132 lines
3.2 KiB
C++
132 lines
3.2 KiB
C++
// SPDX-FileCopyrightText: 2012 Icinga GmbH <https://icinga.com>
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#ifndef TLSSTREAM_H
|
|
#define TLSSTREAM_H
|
|
|
|
#include "base/i2-base.hpp"
|
|
#include "base/shared.hpp"
|
|
#include "base/socket.hpp"
|
|
#include "base/stream.hpp"
|
|
#include "base/tlsutility.hpp"
|
|
#include "base/fifo.hpp"
|
|
#include "base/utility.hpp"
|
|
#include <atomic>
|
|
#include <memory>
|
|
#include <utility>
|
|
#include <variant>
|
|
#include <boost/asio/buffered_stream.hpp>
|
|
#include <boost/asio/io_context.hpp>
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
#include <boost/asio/spawn.hpp>
|
|
#include <boost/asio/ssl/context.hpp>
|
|
#include <boost/asio/ssl/stream.hpp>
|
|
|
|
namespace icinga
|
|
{
|
|
|
|
template<class ARS>
|
|
class SeenStream : public ARS
|
|
{
|
|
public:
|
|
template<class... Args>
|
|
SeenStream(Args&&... args) : ARS(std::forward<Args>(args)...)
|
|
{
|
|
m_Seen.store(nullptr);
|
|
}
|
|
|
|
template<class... Args>
|
|
auto async_read_some(Args&&... args) -> decltype(((ARS*)nullptr)->async_read_some(std::forward<Args>(args)...))
|
|
{
|
|
{
|
|
auto seen (m_Seen.load());
|
|
|
|
if (seen) {
|
|
*seen = Utility::GetTime();
|
|
}
|
|
}
|
|
|
|
return ((ARS*)this)->async_read_some(std::forward<Args>(args)...);
|
|
}
|
|
|
|
inline void SetSeen(double* seen)
|
|
{
|
|
m_Seen.store(seen);
|
|
}
|
|
|
|
private:
|
|
std::atomic<double*> m_Seen;
|
|
};
|
|
|
|
struct UnbufferedAsioTlsStreamParams
|
|
{
|
|
boost::asio::io_context& IoContext;
|
|
boost::asio::ssl::context& SslContext;
|
|
const String& Hostname;
|
|
};
|
|
|
|
typedef SeenStream<boost::asio::ssl::stream<boost::asio::ip::tcp::socket>> AsioTcpTlsStream;
|
|
|
|
class UnbufferedAsioTlsStream : public AsioTcpTlsStream
|
|
{
|
|
public:
|
|
inline
|
|
UnbufferedAsioTlsStream(UnbufferedAsioTlsStreamParams& init)
|
|
: AsioTcpTlsStream(init.IoContext, init.SslContext), m_Hostname(init.Hostname)
|
|
{
|
|
}
|
|
|
|
bool IsVerifyOK();
|
|
String GetVerifyError();
|
|
std::shared_ptr<X509> GetPeerCertificate();
|
|
|
|
template<class... Args>
|
|
inline
|
|
auto async_handshake(handshake_type type, Args&&... args) -> decltype(((AsioTcpTlsStream*)nullptr)->async_handshake(type, std::forward<Args>(args)...))
|
|
{
|
|
BeforeHandshake(type);
|
|
|
|
return AsioTcpTlsStream::async_handshake(type, std::forward<Args>(args)...);
|
|
}
|
|
|
|
template<class... Args>
|
|
inline
|
|
auto handshake(handshake_type type, Args&&... args) -> decltype(((AsioTcpTlsStream*)nullptr)->handshake(type, std::forward<Args>(args)...))
|
|
{
|
|
BeforeHandshake(type);
|
|
|
|
return AsioTcpTlsStream::handshake(type, std::forward<Args>(args)...);
|
|
}
|
|
|
|
private:
|
|
String m_Hostname;
|
|
|
|
void BeforeHandshake(handshake_type type);
|
|
};
|
|
|
|
class AsioTlsStream : public boost::asio::buffered_stream<UnbufferedAsioTlsStream>
|
|
{
|
|
public:
|
|
inline
|
|
AsioTlsStream(boost::asio::io_context& ioContext, boost::asio::ssl::context& sslContext, const String& hostname = String())
|
|
: AsioTlsStream(UnbufferedAsioTlsStreamParams{ioContext, sslContext, hostname})
|
|
{
|
|
}
|
|
|
|
void ForceDisconnect();
|
|
void GracefulDisconnect(boost::asio::io_context::strand& strand, boost::asio::yield_context& yc);
|
|
|
|
private:
|
|
inline
|
|
AsioTlsStream(UnbufferedAsioTlsStreamParams init)
|
|
: buffered_stream(init)
|
|
{
|
|
}
|
|
};
|
|
|
|
using AsioTcpStream = boost::asio::buffered_stream<boost::asio::ip::tcp::socket>;
|
|
using OptionalTlsStream = std::pair<Shared<AsioTlsStream>::Ptr, Shared<AsioTcpStream>::Ptr>;
|
|
using AsioTlsOrTcpStream = std::variant<Shared<AsioTlsStream>::Ptr, Shared<AsioTcpStream>::Ptr>;
|
|
}
|
|
|
|
#endif /* TLSSTREAM_H */
|