void CTlsSocket::TrustCurrentCert(bool trusted)
{
if (m_tlsState != verifycert)
{
m_pOwner->LogMessage(Debug_Warning, _T("TrustCurrentCert called at wrong time."));
return;
}
if (trusted)
{
m_tlsState = conn;
if (m_lastWriteFailed)
m_lastWriteFailed = false;
CheckResumeFailedReadWrite();
if (m_tlsState == conn)
{
CSocketEvent *evt = new CSocketEvent(m_pEvtHandler, this, CSocketEvent::connection);
CSocketEventDispatcher::Get().SendEvent(evt);
}
TriggerEvents();
return;
}
m_pOwner->LogMessage(::Error, _("Remote certificate not trusted."));
Failure(0, ECONNABORTED);
}
void CTlsSocket::TrustCurrentCert(bool trusted)
{
if (m_tlsState != TlsState::verifycert) {
m_pOwner->LogMessage(MessageType::Debug_Warning, _T("TrustCurrentCert called at wrong time."));
return;
}
if (trusted) {
m_tlsState = TlsState::conn;
if (m_lastWriteFailed)
m_lastWriteFailed = false;
CheckResumeFailedReadWrite();
if (m_tlsState == TlsState::conn) {
m_pEvtHandler->send_event<CSocketEvent>(this, SocketEventType::connection, 0);
}
TriggerEvents();
return;
}
m_pOwner->LogMessage(MessageType::Error, _("Remote certificate not trusted."));
Failure(0, true);
}
void CTlsSocket::OnRead()
{
m_pOwner->LogMessage(MessageType::Debug_Debug, _T("CTlsSocket::OnRead()"));
m_canReadFromSocket = true;
if (!m_session)
return;
const int direction = gnutls_record_get_direction(m_session);
if (direction && !m_lastReadFailed) {
m_pOwner->LogMessage(MessageType::Debug_Debug, _T("CTlsSocket::Postponing read"));
return;
}
if (m_tlsState == TlsState::handshake)
ContinueHandshake();
if (m_tlsState == TlsState::closing)
ContinueShutdown();
else if (m_tlsState == TlsState::conn)
{
CheckResumeFailedReadWrite();
TriggerEvents();
}
}
void CTlsSocket::OnSend()
{
m_pOwner->LogMessage(Debug_Debug, _T("CTlsSocket::OnSend()"));
m_canWriteToSocket = true;
if (!m_session)
return;
const int direction = gnutls_record_get_direction(m_session);
if (!direction && !m_lastWriteFailed)
return;
if (m_tlsState == handshake)
ContinueHandshake();
else if (m_tlsState == closing)
ContinueShutdown();
else if (m_tlsState == conn)
{
CheckResumeFailedReadWrite();
TriggerEvents();
}
}