/*

* Copyright (C) 2012 Yee Young Han <[email protected]> (http://blog.naver.com/websearch)

*

* This program is free software; you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation; either version 3 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

*/

#include "SipPlatformDefine.h"

#ifdef WIN32

#pragma comment( lib, "libeay32" )

#pragma comment( lib, "ssleay32" )

#endif

#include "TlsFunction.h"

#include "SipMutex.h"

#include "Log.h"

#include "FileUtility.h"

#include "MemoryDebug.h"

static SSL_CTX * gpsttServerCtx;

static SSL_CTX * gpsttClientCtx;

#if OPENSSL_VERSION_NUMBER >= 0x10000003L

static const SSL_METHOD * gpsttServerMeth;

static const SSL_METHOD * gpsttClientMeth;

#else

static SSL_METHOD * gpsttServerMeth;

static SSL_METHOD * gpsttClientMeth;

#endif

static bool gbStartSslServer = false;

static CSipMutex * garrMutex;

static CSipMutex gclsMutex;

/**

* @ingroup TcpStack

* @brief SSL 라이브러리를 multi-thread 에서 사용할 수 있기 위한 Lock/Unlock function

* @param mode CRYPTO_LOCK / CRYPTO_UNLOCK

* @param n 쓰레드 아이디

* @param file

* @param line

*/

static void SSLLockingFunction( int mode, int n, const char * file, int line )

{

if( mode & CRYPTO_LOCK )

{

garrMutex[n].acquire();

}

else

{

garrMutex[n].release();

}

}

/**

* @ingroup TcpStack

* @brief SSL 라이브러리를 multi-thread 에서 사용할 수 있기 위한 ID function

* @returns 현재 쓰레드 ID 를 리턴한다.

*/

static unsigned long SSLIdFunction( )

{

#ifdef WIN32

return GetCurrentThreadId();

#else

return (unsigned long)pthread_self();

#endif

}

/**

* @ingroup TcpStack

* @brief SSL 라이브러리를 multi-thread 기반으로 시작한다.

* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.

*/

static bool SSLStart( )

{

garrMutex = new CSipMutex[ CRYPTO_num_locks() ];

if( garrMutex == NULL )

{

CLog::Print( LOG_ERROR, "%s CMutex[] new error", __FUNCTION__ );

return false;

}

CRYPTO_set_id_callback( SSLIdFunction );

CRYPTO_set_locking_callback( SSLLockingFunction );

if( !SSL_library_init() )

{

CLog::Print( LOG_ERROR, "SSL_init_library error" );

return false;

}

return true;

}

/**

* @ingroup TcpStack

* @brief SSL 라이브러리를 중지시킨다.

* @returns true 를 리턴한다.

*/

static bool SSLStop( )

{

CRYPTO_set_id_callback(NULL);

CRYPTO_set_locking_callback(NULL);

if( garrMutex )

{

delete [] garrMutex;

garrMutex = NULL;

}

return true;

}

static void SSLPrintError( )

{

CLog::Print( ERR_print_errors_fp );

}

/**

* @ingroup TcpStack

* @brief SSL 서버 라이브러리를 시작한다.

* @param szCertFile 서버 인증서 및 개인키 파일

* @param szCaCertFile CA 인증서 파일

* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.

*/

bool SSLServerStart( const char * szCertFile )

{

if( szCertFile == NULL ) return false;

if( IsExistFile( szCertFile ) == false )

{

CLog::Print( LOG_ERROR, "cert file is not found" );

return false;

}

gclsMutex.acquire();

if( gbStartSslServer == false )

{

int n;

if( SSLStart() )

{

SSL_load_error_strings();

SSLeay_add_ssl_algorithms();

gpsttServerMeth = TLSv1_server_method();

if( (gpsttServerCtx = SSL_CTX_new( gpsttServerMeth )) == NULL )

{

CLog::Print( LOG_ERROR, "SSL_CTX_new error - server" );

}

else

{

gpsttClientMeth = TLSv1_client_method();

if( (gpsttClientCtx = SSL_CTX_new( gpsttClientMeth )) == NULL )

{

CLog::Print( LOG_ERROR, "SSL_CTX_new error - client" );

}

else if( SSL_CTX_use_certificate_file(gpsttServerCtx, szCertFile, SSL_FILETYPE_PEM) <= 0 )

{

CLog::Print( LOG_ERROR, "SSL_CTX_use_certificate_file error" );

SSLPrintError( );

}

else if( ( n = SSL_CTX_use_PrivateKey_file(gpsttServerCtx, szCertFile, SSL_FILETYPE_PEM)) <= 0 )

{

CLog::Print( LOG_ERROR, "SSL_CTX_use_PrivateKey_file error(%d)", n );

}

else if( !SSL_CTX_check_private_key( gpsttServerCtx ) )

{

CLog::Print( LOG_ERROR, "[SSL] Private key does not match the certificate public key");

}

else

{

gbStartSslServer = true;

}

}

}

}

gclsMutex.release();

return gbStartSslServer;

}

/**

* @ingroup TcpStack

* @brief SSL 서버 라이브러리를 종료한다.

* @returns true 를 리턴한다.

*/

bool SSLServerStop( )

{

gclsMutex.acquire();

if( gbStartSslServer )

{

SSLStop();

if( gpsttServerCtx )

{

SSL_CTX_free( gpsttServerCtx );

gpsttServerCtx = NULL;

}

if( gpsttClientCtx )

{

SSL_CTX_free( gpsttClientCtx );

gpsttClientCtx = NULL;

}

gbStartSslServer = false;

}

gclsMutex.release();

return true;

}

/**

* @ingroup TcpStack

* @brief SSL 클라이언트 라이브러리를 시작한다.

* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.

*/

bool SSLClientStart( )

{

gclsMutex.acquire();

if( gbStartSslServer == false )

{

if( SSLStart() )

{

SSL_load_error_strings();

SSLeay_add_ssl_algorithms();

gpsttClientMeth = TLSv1_client_method();

if( (gpsttClientCtx = SSL_CTX_new( gpsttClientMeth )) == NULL )

{

CLog::Print( LOG_ERROR, "SSL_CTX_new error - client" );

}

else

{

gbStartSslServer = true;

}

}

}

gclsMutex.release();

return gbStartSslServer;

}

/**

* @ingroup TcpStack

* @brief SSL 클라이언트 라이브러리를 종료한다.

* @returns true 를 리턴한다.

*/

bool SSLClientStop( )

{

gclsMutex.acquire();

if( gbStartSslServer )

{

SSLStop();

if( gpsttClientCtx )

{

SSL_CTX_free( gpsttClientCtx );

gpsttClientCtx = NULL;

}

gbStartSslServer = false;

}

gclsMutex.release();

return true;

}

/**

* @ingroup TcpStack

* @brief 프로세스가 종료될 때에 최종적으로 실행하여서 openssl 메모리 누수를 출력하지 않는다.

*/

void SSLFinal()

{

ERR_free_strings();

#ifdef USE_TLS_FREE

// http://clseto.mysinablog.com/index.php?op=ViewArticle&articleId=3304652

ERR_remove_state(0);

COMP_zlib_cleanup();

OBJ_NAME_cleanup(-1);

CRYPTO_cleanup_all_ex_data();

EVP_cleanup();

sk_SSL_COMP_free( SSL_COMP_get_compression_methods() );

#endif

}

/**

* @brief SSL 세션을 연결한다.

* @param iFd 클라이언트 TCP 소켓 핸들

* @param ppsttSsl SSL 구조체

* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.

*/

bool SSLConnect( Socket iFd, SSL ** ppsttSsl )

{

SSL * psttSsl;

SSLClientStart( );

if( (psttSsl = SSL_new(gpsttClientCtx)) == NULL )

{

return false;

}

try

{

SSL_set_fd( psttSsl, (int)iFd );

if( SSL_connect( psttSsl ) == -1 )

{

SSL_free( psttSsl );

return false;

}

}

catch( ... )

{

CLog::Print( LOG_ERROR, "[SSL] SSLConnect() undefined error" );

SSL_free( psttSsl );

return false;

}

*ppsttSsl = psttSsl;

return true;

}

/**

* @ingroup TcpStack

* @brief 클라이언트 SSL 접속 요청을 허용한다.

* @param iFd 클라이언트 TCP 소켓 핸들

* @param ppsttSsl SSL 구조체

* @param bCheckClientCert 클라이언트 인증서를 확인할 것인가?

* @param iVerifyDepth the maximum depth for the certificate chain verification that shall be allowed for ssl

* @param iAcceptTimeout SSL 접속 요청 처리 최대 시간 ( ms 단위 )

* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.

*/

bool SSLAccept( Socket iFd, SSL ** ppsttSsl, bool bCheckClientCert, int iVerifyDepth, int iAcceptTimeout )

{

SSL * psttSsl;

if( (psttSsl = SSL_new( gpsttServerCtx )) == NULL )

{

CLog::Print( LOG_ERROR, "SSL_new() error" );

return false;

}

SSL_set_fd( psttSsl, (int)iFd );

if( iAcceptTimeout > 0 )

{

#ifdef WIN32

int iTimeout = iAcceptTimeout;

setsockopt( iFd, SOL_SOCKET, SO_RCVTIMEO, (char *)&iTimeout, sizeof(iTimeout) );

#else

struct timeval sttTime;

sttTime.tv_sec = iAcceptTimeout / 1000;

sttTime.tv_usec = ( iAcceptTimeout % 1000 ) * 1000;

CLog::Print( LOG_DEBUG, "SO_RCVTIMEO(%d.%d)", sttTime.tv_sec, sttTime.tv_usec );

if( setsockopt( iFd, SOL_SOCKET, SO_RCVTIMEO, &sttTime, sizeof(sttTime) ) == -1 )

{

CLog::Print( LOG_ERROR, "setsockopt(SO_RCVTIMEO:%d.%d) error(%d)", sttTime.tv_sec, sttTime.tv_usec, GetError() );

}

#endif

}

// QQQ : SSL 프로토콜이 아닌 경우에 메모리 에러가 발생하므로 아래와 같이

// : 막아 놓았음. 더 좋은 방법을 모색하여야 함.

try

{

if( bCheckClientCert )

{

SSL_set_verify( psttSsl, SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT | SSL_VERIFY_CLIENT_ONCE, NULL );

SSL_set_verify_depth( psttSsl, iVerifyDepth );

}

if( SSL_accept( psttSsl ) == -1 )

{

CLog::Print( LOG_ERROR, "SSL_accept() error" );

SSLPrintError( );

SSL_free( psttSsl );

return false;

}

}

catch( ... )

{

CLog::Print( LOG_ERROR, "[SSL] SSL_accept() undefined error" );

SSL_free( psttSsl );

return false;

}

if( iAcceptTimeout > 0 )

{

#ifdef WIN32

int iTimeout = 0;

setsockopt( iFd, SOL_SOCKET, SO_RCVTIMEO, (char *)&iTimeout, sizeof(iTimeout) );

#else

struct timeval sttTime;

sttTime.tv_sec = 0;

sttTime.tv_usec = 0;

if( setsockopt( iFd, SOL_SOCKET, SO_RCVTIMEO, &sttTime, sizeof(sttTime) ) == -1 )

{

CLog::Print( LOG_ERROR, "setsockopt(SO_RCVTIMEO:%d.%d) error(%d)", sttTime.tv_sec, sttTime.tv_usec, GetError() );

}

#endif

}

*ppsttSsl = psttSsl;

return true;

}

/**

* @ingroup TcpStack

* @brief SSL 프로토콜로 패킷을 전송한다.

* @param ssl SSL 구조체

* @param szBuf 전송 패킷

* @param iBufLen 전송 패킷 크기

* @returns 전송 패킷 크기를 리턴한다.

*/

int SSLSend( SSL * ssl, const char * szBuf, int iBufLen )

{

int n;

int iSendLen = 0;

try

{

while( 1 )

{

n = SSL_write( ssl, szBuf + iSendLen, iBufLen - iSendLen );

if( n <= 0 ) return -1;

iSendLen += n;

if( iSendLen == iBufLen ) break;

}

}

catch( ... )

{

CLog::Print( LOG_ERROR, "[SSL] SSLSend() undefined error" );

}

return iBufLen;

}

/**

* @ingroup TcpStack

* @brief SSL 프로토콜로 수신된 패킷을 읽는다.

* @param ssl SSL 구조체

* @param szBuf 수신 패킷 저장 버퍼

* @param iBufLen 수신 패킷 저장 버퍼 크기

* @returns 성공하면 양수를 리턴하고 실패하면 0 또는 음수를 리턴한다.

*/

int SSLRecv( SSL * ssl, char * szBuf, int iBufLen )

{

return SSL_read( ssl, szBuf, iBufLen );

}

/**

* @ingroup TcpStack

* @brief SSL 세션을 종료한다.

* @param ssl SSL 구조체

* @returns true 를 리턴한다.

*/

bool SSLClose( SSL * ssl )

{

if( ssl )

{

SSL_free( ssl );

}

return true;

}

/**

* @ingroup TcpStack

* @brief SSL 서버에서 사용되는 cipher list 를 로그로 출력한다.

*/

void SSLPrintLogServerCipherList( )

{

if( gpsttServerCtx == NULL )

{

CLog::Print( LOG_ERROR, "gpsttServerCtx is null" );

return;

}

int iNum = sk_SSL_CIPHER_num( gpsttServerCtx->cipher_list );

for( int i = 0; i < iNum; ++i )

{

const SSL_CIPHER *c = sk_SSL_CIPHER_value( gpsttServerCtx->cipher_list, i );

CLog::Print( LOG_DEBUG, "[%s] [%s] [0x%04X] (%d)", SSL_CIPHER_get_version(c), SSL_CIPHER_get_name(c), c->id - 0x3000000, i );

}

}

/**

* @ingroup TcpStack

* @brief SSL 클라이언트에서 사용되는 cipher list 를 로그로 출력한다.

*/

void SSLPrintLogClientCipherList( )

{

if( gpsttClientCtx == NULL )

{

CLog::Print( LOG_ERROR, "gpsttServerCtx is null" );

return;

}

int iNum = sk_SSL_CIPHER_num( gpsttClientCtx->cipher_list );

for( int i = 0; i < iNum; ++i )

{

const SSL_CIPHER *c = sk_SSL_CIPHER_value( gpsttClientCtx->cipher_list, i );

CLog::Print( LOG_DEBUG, "[%s] [%s] [0x%04X] (%d)", SSL_CIPHER_get_version(c), SSL_CIPHER_get_name(c), c->id - 0x3000000, i );

}

}