tyler.durden
/
icinga2
mirror of https://github.com/Icinga/icinga2.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve performance for TlsStream::OnEvent 

refs #11014
This commit is contained in:
Gunnar Beutner 2016-02-02 13:51:17 +01:00
parent 3b4fc69fab
commit a2cbe1a0d5
1 changed files with 52 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
lib/base/tlsstream.cpp
View File

@ -126,7 +126,7 @@ boost::shared_ptr<X509> TlsStream::GetPeerCertificate(void) const
void TlsStream::OnEvent(int revents)
{
int rc, err;
int rc;
size_t count;
boost::mutex::scoped_lock lock(m_Mutex);
@ -147,6 +147,8 @@ void TlsStream::OnEvent(int revents)
}
}
bool success = false;
switch (m_CurrentAction) {
case TlsActionRead:
do {
@ -154,9 +156,12 @@ void TlsStream::OnEvent(int revents)
if (rc > 0) {
m_RecvQ->Write(buffer, rc);
m_CV.notify_all();
success = true;
}
} while (SSL_pending(m_SSL.get()));
} while (rc > 0);
if (success)
m_CV.notify_all();
break;
case TlsActionWrite:
@ -164,14 +169,17 @@ void TlsStream::OnEvent(int revents)
rc = SSL_write(m_SSL.get(), buffer, count);
if (rc > 0)
if (rc > 0) {
m_SendQ->Read(NULL, rc, true);
success = true;
}
break;
case TlsActionHandshake:
rc = SSL_do_handshake(m_SSL.get());
if (rc > 0) {
success = true;
m_HandshakeOK = true;
m_CV.notify_all();
}
@ -181,7 +189,46 @@ void TlsStream::OnEvent(int revents)
VERIFY(!"Invalid TlsAction");
}
if (rc > 0) {
if (rc < 0) {
int err = SSL_get_error(m_SSL.get(), rc);
switch (err) {
case SSL_ERROR_WANT_READ:
m_Retry = true;
ChangeEvents(POLLIN);
break;
case SSL_ERROR_WANT_WRITE:
m_Retry = true;
ChangeEvents(POLLOUT);
break;
case SSL_ERROR_ZERO_RETURN:
lock.unlock();
Close();
break;
default:
m_ErrorCode = ERR_peek_error();
m_ErrorOccurred = true;
if (m_ErrorCode != 0) {
Log(LogWarning, "TlsStream")
<< "OpenSSL error: " << ERR_error_string(m_ErrorCode, NULL);
} else {
Log(LogWarning, "TlsStream", "TLS stream was disconnected.");
}
lock.unlock();
Close();
break;
}
}
if (success) {
m_CurrentAction = TlsActionNone;
if (!m_Eof) {
@ -198,51 +245,6 @@ void TlsStream::OnEvent(int revents)
if (m_Shutdown && !m_SendQ->IsDataAvailable())
Close();
return;
}
err = SSL_get_error(m_SSL.get(), rc);
switch (err) {
case SSL_ERROR_WANT_READ:
m_Retry = true;
ChangeEvents(POLLIN);
break;
case SSL_ERROR_WANT_WRITE:
m_Retry = true;
ChangeEvents(POLLOUT);
break;
case SSL_ERROR_ZERO_RETURN:
lock.unlock();
if (IsHandlingEvents())
SignalDataAvailable();
Close();
break;
default:
m_ErrorCode = ERR_peek_error();
m_ErrorOccurred = true;
if (m_ErrorCode != 0) {
Log(LogWarning, "TlsStream")
<< "OpenSSL error: " << ERR_error_string(m_ErrorCode, NULL);
} else {
Log(LogWarning, "TlsStream", "TLS stream was disconnected.");
}
lock.unlock();
if (IsHandlingEvents())
SignalDataAvailable();
Close();
break;
}
}