github-starred/ultimatepp
Watch
2
0
Fork
You've already forked ultimatepp
 0
mirror of https://github.com/ultimatepp/ultimatepp.git synced 2026-10-06 00:30:10 -06:00
Code Issues 33 Projects Packages Wiki Activity Actions

.Core/SSL

Browse source 
git-svn-id: svn://ultimatepp.org/upp/trunk@4767 f0d560ea-af0d-0410-9eb7-867de7ffcac7
This commit is contained in:
cxl 2012-04-09 18:39:22 +00:00
commit 121acd6991
1 changed files with 16 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

29
uppsrc/Core/SSL/Socket.cpp
View file

@ -18,7 +18,8 @@ struct TcpSocket::SSLImp : TcpSocket::SSL {
void SetSSLResError(const char *context, int res);
bool IsAgain(int res) const;
SSLImp(TcpSocket& socket) : socket(socket) {}
SSLImp(TcpSocket& socket) : socket(socket) { ssl = NULL; }
~SSLImp();
};
TcpSocket::SSL *TcpSocket::CreateSSLImp(TcpSocket& socket)
@ -35,6 +36,12 @@ INITBLOCK {
InitCreateSSL();
}
TcpSocket::SSLImp::~SSLImp()
{
if(ssl)
SSL_free(ssl);
}
void TcpSocket::SSLImp::SetSSLError(const char *context)
{
int code;
@ -118,28 +125,23 @@ bool TcpSocket::SSLImp::Wait(dword flags)
int TcpSocket::SSLImp::Send(const void *buffer, int maxlen)
{
int res = SSL_write(ssl, (const char *)buffer, maxlen);
if(IsAgain(res))
return 0;
if(res <= 0) {
if(res > 0)
return res;
if(!IsAgain(res))
SetSSLResError("SSL_write", res);
return 0;
}
return res;
return 0;
}
int TcpSocket::SSLImp::Recv(void *buffer, int maxlen)
{
int res = SSL_read(ssl, (char *)buffer, maxlen);
if(IsAgain(res))
return 0;
if(res == 0) {
if(res > 0)
return res;
if(!IsAgain(res)) {
socket.is_eof = true;
if(SSL_get_shutdown(ssl) & SSL_RECEIVED_SHUTDOWN)
return 0;
}
if(res <= 0) {
SetSSLResError("SSL_read", res);
return 0;
}
return res;
}
@ -149,6 +151,7 @@ void TcpSocket::SSLImp::Close()
SSL_shutdown(ssl);
socket.RawClose();
SSL_free(ssl);
ssl = NULL;
}