using namespace icinga;
+/**
+ * TCPClient
+ *
+ * Constructor for the TCPClient class.
+ *
+ * @param role The role of the TCP client socket.
+ */
TCPClient::TCPClient(TCPClientRole role)
{
m_Role = role;
m_RecvQueue = make_shared<FIFO>();
}
+/**
+ * GetRole
+ *
+ * Retrieves the role of the socket.
+ *
+ * @returns The role.
+ */
TCPClientRole TCPClient::GetRole(void) const
{
return m_Role;
}
+/**
+ * Start
+ *
+ * Registers the socket and starts processing events for it.
+ */
void TCPClient::Start(void)
{
TCPSocket::Start();
OnWritable += bind_weak(&TCPClient::WritableEventHandler, shared_from_this());
}
+/**
+ * Connect
+ *
+ * Creates a socket and connects to the specified node and service.
+ *
+ * @param node The node.
+ * @param service The service.
+ */
void TCPClient::Connect(const string& node, const string& service)
{
m_Role = RoleOutbound;
freeaddrinfo(result);
}
+/**
+ * GetSendQueue
+ *
+ * Retrieves the send queue for the socket.
+ *
+ * @returns The send queue.
+ */
FIFO::Ptr TCPClient::GetSendQueue(void)
{
return m_SendQueue;
}
+/**
+ * GetRecvQueue
+ *
+ * Retrieves the recv queue for the socket.
+ *
+ * @returns The recv queue.
+ */
FIFO::Ptr TCPClient::GetRecvQueue(void)
{
return m_RecvQueue;
}
+/**
+ * ReadableEventHandler
+ *
+ * Processes data that is available for this socket.
+ *
+ * @param ea Event arguments.
+ * @returns 0
+ */
int TCPClient::ReadableEventHandler(const EventArgs& ea)
{
int rc;
return 0;
}
+/**
+ * WritableEventHandler
+ *
+ * Processes data that can be written for this socket.
+ *
+ * @param ea Event arguments.
+ * @returns 0
+ */
int TCPClient::WritableEventHandler(const EventArgs& ea)
{
int rc;
return 0;
}
+/**
+ * WantsToRead
+ *
+ * Checks whether data should be read for this socket.
+ *
+ * @returns true
+ */
bool TCPClient::WantsToRead(void) const
{
return true;
}
+/**
+ * WantsToWrite
+ *
+ * Checks whether data should be written for this socket.
+ *
+ * @returns true if data should be written, false otherwise.
+ */
bool TCPClient::WantsToWrite(void) const
{
return (m_SendQueue->GetSize() > 0);
}
+/**
+ * TCPClientFactory
+ *
+ * Default factory function for TCP clients.
+ *
+ * @param role The role of the new client.
+ * @returns The new client.
+ */
TCPClient::Ptr icinga::TCPClientFactory(TCPClientRole role)
{
return make_shared<TCPClient>(role);
using namespace icinga;
+/**
+ * TCPServer
+ *
+ * Constructor for the TCPServer class.
+ */
TCPServer::TCPServer(void)
{
m_ClientFactory = bind(&TCPClientFactory, RoleInbound);
}
+/**
+ * SetClientFactory
+ *
+ * Sets the client factory.
+ *
+ * @param clientFactory The client factory function.
+ */
void TCPServer::SetClientFactory(function<TCPClient::Ptr()> clientFactory)
{
m_ClientFactory = clientFactory;
}
+/**
+ * GetFactoryFunction
+ *
+ * Retrieves the client factory.
+ *
+ * @returns The client factory function.
+ */
function<TCPClient::Ptr()> TCPServer::GetFactoryFunction(void) const
{
return m_ClientFactory;
}
+/**
+ * Start
+ *
+ * Registers the TCP server and starts processing events for it.
+ */
void TCPServer::Start(void)
{
TCPSocket::Start();
OnReadable += bind_weak(&TCPServer::ReadableEventHandler, shared_from_this());
}
+/**
+ * Listen
+ *
+ * Starts listening for incoming client connections.
+ */
void TCPServer::Listen(void)
{
int rc = listen(GetFD(), SOMAXCONN);
}
}
+/**
+ * ReadableEventHandler
+ *
+ * Accepts a new client and creates a new client object for it
+ * using the client factory function.
+ *
+ * @param ea Event arguments.
+ * @returns 0
+ */
int TCPServer::ReadableEventHandler(const EventArgs& ea)
{
int fd;
return 0;
}
+/**
+ * WantsToRead
+ *
+ * Checks whether the TCP server wants to read (i.e. accept new clients).
+ *
+ * @returns true
+ */
bool TCPServer::WantsToRead(void) const
{
return true;
int I2_EXPORT TLSClient::m_SSLIndex;
bool I2_EXPORT TLSClient::m_SSLIndexInitialized = false;
+/**
+ * TLSClient
+ *
+ * Constructor for the TLSClient class.
+ *
+ * @param role The role of the client.
+ * @param sslContext The SSL context for the client.
+ */
TLSClient::TLSClient(TCPClientRole role, shared_ptr<SSL_CTX> sslContext) : TCPClient(role)
{
m_SSLContext = sslContext;
m_BlockWrite = false;
}
+/**
+ * NullCertificateDeleter
+ *
+ * Takes a certificate as an argument. Does nothing.
+ *
+ * @param certificate An X509 certificate.
+ */
void TLSClient::NullCertificateDeleter(X509 *certificate)
{
/* Nothing to do here. */
}
+/**
+ * GetClientCertificate
+ *
+ * Retrieves the X509 certficate for this client.
+ *
+ * @returns The X509 certificate.
+ */
shared_ptr<X509> TLSClient::GetClientCertificate(void) const
{
return shared_ptr<X509>(SSL_get_certificate(m_SSL.get()), &TLSClient::NullCertificateDeleter);
}
+/**
+ * GetPeerCertificate
+ *
+ * Retrieves the X509 certficate for the peer.
+ *
+ * @returns The X509 certificate.
+ */
shared_ptr<X509> TLSClient::GetPeerCertificate(void) const
{
return shared_ptr<X509>(SSL_get_peer_certificate(m_SSL.get()), X509_free);
}
+/**
+ * Start
+ *
+ * Registers the TLS socket and starts processing events for it.
+ */
void TLSClient::Start(void)
{
TCPClient::Start();
SSL_do_handshake(m_SSL.get());
}
+/**
+ * ReadableEventHandler
+ *
+ * Processes data that is available for this socket.
+ *
+ * @param ea Event arguments.
+ * @returns 0
+ */
int TLSClient::ReadableEventHandler(const EventArgs& ea)
{
int rc;
return 0;
}
+/**
+ * WritableEventHandler
+ *
+ * Processes data that can be written for this socket.
+ *
+ * @param ea Event arguments.
+ * @returns 0
+ */
int TLSClient::WritableEventHandler(const EventArgs& ea)
{
int rc;
return 0;
}
+/**
+ * WantsToRead
+ *
+ * Checks whether data should be read for this socket.
+ *
+ * @returns true if data should be read, false otherwise.
+ */
bool TLSClient::WantsToRead(void) const
{
if (SSL_want_read(m_SSL.get()))
return TCPClient::WantsToRead();
}
+/**
+ * WantsToWrite
+ *
+ * Checks whether data should be written for this socket.
+ *
+ * @returns true if data should be written, false otherwise.
+ */
bool TLSClient::WantsToWrite(void) const
{
if (SSL_want_write(m_SSL.get()))
return TCPClient::WantsToWrite();
}
+/**
+ * CloseInternal
+ *
+ * Closes the socket.
+ *
+ * @param from_dtor Whether this method was invoked from the destructor.
+ */
void TLSClient::CloseInternal(bool from_dtor)
{
SSL_shutdown(m_SSL.get());
TCPClient::CloseInternal(from_dtor);
}
+/**
+ * HandleSSLError
+ *
+ * Handles an OpenSSL error.
+ */
void TLSClient::HandleSSLError(void)
{
int code = ERR_get_error();
return;
}
+/**
+ * TLSClientFactory
+ *
+ * Factory function for the TLSClient class.
+ *
+ * @param role The role of the TLS socket.
+ * @param sslContext The SSL context for the socket.
+ * @returns A new TLS socket.
+ */
TCPClient::Ptr icinga::TLSClientFactory(TCPClientRole role, shared_ptr<SSL_CTX> sslContext)
{
return make_shared<TLSClient>(role, sslContext);
}
+/**
+ * SSLVerifyCertificate
+ *
+ * Callback function that verifies SSL certificates.
+ *
+ * @param ok Whether pre-checks for the SSL certificates were successful.
+ * @param x509Context X509 context for the certificate.
+ * @returns 1 if the verification was successful, 0 otherwise.
+ */
int TLSClient::SSLVerifyCertificate(int ok, X509_STORE_CTX *x509Context)
{
SSL *ssl = (SSL *)X509_STORE_CTX_get_ex_data(x509Context, SSL_get_ex_data_X509_STORE_CTX_idx());
projects / icinga2 / commitdiff