/*

* Copyright (C) 2012 Yee Young Han <websearch@naver.com> (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

*/

/**

* @ingroup SipUserAgent

* @brief SIP 통화 요청 메시지를 전송한다.

* @param pszFrom 발신자 아이디

* @param pszTo 수신자 아이디

* @param pclsRtp local RTP 정보 저장 객체

* @param pclsRoute SIP 메시지 목적지 주소 저장 객체

* @param strCallId 생성된 INVITE 메시지의 Call-ID 가 저장될 변수

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

*/

bool CSipUserAgent::StartCall( const char * pszFrom, const char * pszTo, CSipCallRtp * pclsRtp, CSipCallRoute * pclsRoute, std::string & strCallId )

{

if( pszFrom == NULL || pszTo == NULL ) return false;

if( pclsRtp == NULL || pclsRoute == NULL ) return false;

if( strlen( pszFrom ) == 0 || strlen( pszTo ) == 0 ) return false;

if( pclsRoute->m_strDestIp.empty() ) return false;

if( pclsRoute->m_iDestPort <= 0 || pclsRoute->m_iDestPort > 65535 ) return false;

#ifndef USE_TLS

if( pclsRoute->m_eTransport == E_SIP_TLS ) return false;

#endif

CSipDialog clsDialog( &m_clsSipStack );

clsDialog.m_strFromId = pszFrom;

clsDialog.m_strToId = pszTo;

clsDialog.SetLocalRtp( pclsRtp );

clsDialog.m_strContactIp = pclsRoute->m_strDestIp;

clsDialog.m_iContactPort = pclsRoute->m_iDestPort;

clsDialog.m_eTransport = pclsRoute->m_eTransport;

clsDialog.m_b100rel = pclsRoute->m_b100rel;

if( SendInvite( clsDialog ) == false ) return false;

strCallId = clsDialog.m_strCallId;

return true;

}

/**

* @ingroup SipUserAgent

* @brief 통화를 종료한다.

* 통화 요청을 보내고 연결되지 않으면 통화 취소 메시지를 전송한다.

* 통화 연결되었으면 통화 종료 메시지를 전송한다.

* 통화 수락인 경우 통화 거절 응답 메시지를 전송한다.

* @param pszCallId SIP Call-ID

* @param iSipCode 통화 요청을 거절할 때에 전송할 SIP status code

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

*/

bool CSipUserAgent::StopCall( const char * pszCallId, int iSipCode )

{

SIP_DIALOG_MAP::iterator itMap;

bool bRes = false;

CSipMessage * pclsMessage = NULL;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

if( itMap->second.m_sttStartTime.tv_sec != 0 )

{

if( itMap->second.m_sttEndTime.tv_sec == 0 )

{

pclsMessage = itMap->second.CreateBye();

//gettimeofday( &itMap->second.m_sttEndTime, NULL );

Delete( itMap );

}

}

else

{

if( itMap->second.m_pclsInvite )

{

if( iSipCode )

{

pclsMessage = itMap->second.m_pclsInvite->CreateResponse( iSipCode );

}

else

{

pclsMessage = itMap->second.m_pclsInvite->CreateResponse( SIP_DECLINE );

}

//gettimeofday( &itMap->second.m_sttEndTime, NULL );

Delete( itMap );

}

else if( itMap->second.m_sttCancelTime.tv_sec == 0 && itMap->second.m_sttEndTime.tv_sec == 0 )

{

pclsMessage = itMap->second.CreateCancel();

gettimeofday( &itMap->second.m_sttCancelTime, NULL );

}

}

bRes = true;

}

m_clsDialogMutex.release();

if( pclsMessage )

{

m_clsSipStack.SendSipMessage( pclsMessage );

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief 착신 전환한다.

* @param pszCallId SIP Call-ID

* @param pszForward 착신 전환 전화번호

* @returns 착신 전환에 성공하면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.

*/

bool CSipUserAgent::StopCall( const char * pszCallId, const char * pszForward )

{

SIP_DIALOG_MAP::iterator itMap;

bool bRes = false;

CSipMessage * pclsMessage = NULL;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

if( itMap->second.m_sttStartTime.tv_sec == 0 && itMap->second.m_pclsInvite )

{

pclsMessage = itMap->second.m_pclsInvite->CreateResponse( SIP_MOVED_TEMPORARILY );

Delete( itMap );

}

}

m_clsDialogMutex.release();

if( pclsMessage )

{

CSipFrom clsContact = pclsMessage->m_clsFrom;

clsContact.m_clsUri.m_strUser = pszForward;

pclsMessage->m_clsContactList.push_back( clsContact );

m_clsSipStack.SendSipMessage( pclsMessage );

bRes = true;

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief 183 응답 메시지를 전송한다. SIP 클라이언트에서 183 응답 메시지를 전송할 때에 사용된다.

* @param pszCallId SIP Call-ID

* @param pclsRtp local RTP 정보 저장 객체

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

*/

bool CSipUserAgent::RingCall( const char * pszCallId, CSipCallRtp * pclsRtp )

{

SIP_DIALOG_MAP::iterator itMap;

bool bRes = false;

CSipMessage * pclsMessage = NULL;

if( pclsRtp == NULL ) return false;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

if( itMap->second.m_sttStartTime.tv_sec == 0 && itMap->second.m_pclsInvite )

{

itMap->second.SetLocalRtp( pclsRtp );

pclsMessage = itMap->second.m_pclsInvite->CreateResponse( SIP_SESSION_PROGRESS );

if( pclsMessage )

{

if( itMap->second.m_pclsInvite->Is100rel() )

{

int iRSeq = itMap->second.m_pclsInvite->m_clsCSeq.m_iDigit + RandomGet() % 10000;

char szRSeq[21];

snprintf( szRSeq, sizeof(szRSeq), "%d", iRSeq );

pclsMessage->AddHeader( "RSeq", szRSeq );

}

itMap->second.AddSdp( pclsMessage );

bRes = true;

}

}

}

m_clsDialogMutex.release();

if( pclsMessage )

{

m_clsSipStack.SendSipMessage( pclsMessage );

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief SIP 통화 요청에 대한 Ring / Session Progress 응답 메시지를 전송한다. IP-PBX 에서 Ring / Session Progress 메시지를 전달할 때에 사용된다.

* @param pszCallId SIP Call-ID

* @param iSipStatus SIP 응답 코드

* @param pclsRtp local RTP 정보 저장 객체

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

*/

bool CSipUserAgent::RingCall( const char * pszCallId, int iSipStatus, CSipCallRtp * pclsRtp )

{

SIP_DIALOG_MAP::iterator itMap;

CSipMessage * pclsMessage = NULL;

bool bRes = false;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

if( itMap->second.m_sttStartTime.tv_sec == 0 && itMap->second.m_pclsInvite )

{

pclsMessage = itMap->second.m_pclsInvite->CreateResponse( iSipStatus );

if( pclsRtp )

{

itMap->second.SetLocalRtp( pclsRtp );

itMap->second.AddSdp( pclsMessage );

}

if( itMap->second.m_iRSeq != -1 )

{

pclsMessage->AddHeader( "Allow", "PRACK, INVITE, ACK, BYE, CANCEL, REFER, NOTIFY, MESSAGE" );

pclsMessage->AddHeader( "Require", "100rel" );

char szRSeq[21];

snprintf( szRSeq, sizeof(szRSeq), "%d", itMap->second.m_iRSeq );

pclsMessage->AddHeader( "RSeq", szRSeq );

}

bRes = true;

}

}

m_clsDialogMutex.release();

if( pclsMessage )

{

m_clsSipStack.SendSipMessage( pclsMessage );

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief 수신된 통화를 수락한다.

* @param pszCallId SIP Call-ID

* @param pclsRtp local RTP 정보 저장 객체

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

*/

bool CSipUserAgent::AcceptCall( const char * pszCallId, CSipCallRtp * pclsRtp )

{

SIP_DIALOG_MAP::iterator itMap;

bool bRes = false;

CSipMessage * pclsMessage = NULL;

if( pclsRtp == NULL ) return false;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

if( itMap->second.m_sttStartTime.tv_sec == 0 && itMap->second.m_pclsInvite )

{

itMap->second.SetLocalRtp( pclsRtp );

pclsMessage = itMap->second.m_pclsInvite->CreateResponse( SIP_OK );

if( pclsMessage )

{

gettimeofday( &itMap->second.m_sttStartTime, NULL );

itMap->second.AddSdp( pclsMessage );

delete itMap->second.m_pclsInvite;

itMap->second.m_pclsInvite = NULL;

bRes = true;

}

}

}

m_clsDialogMutex.release();

if( pclsMessage )

{

return m_clsSipStack.SendSipMessage( pclsMessage );

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief 통화 hold 요청 메시지를 전송한다.

* @param pszCallId SIP Call-ID

* @param eDirection 미디어 direction

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

*/

bool CSipUserAgent::HoldCall( const char * pszCallId, ERtpDirection eDirection )

{

SIP_DIALOG_MAP::iterator itMap;

CSipMessage * pclsRequest = NULL;

bool bRes = false;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

itMap->second.m_eLocalDirection = eDirection;

pclsRequest = itMap->second.CreateInvite();

bRes = true;

}

m_clsDialogMutex.release();

if( pclsRequest )

{

m_clsSipStack.SendSipMessage( pclsRequest );

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief 통화 resume 요청 메시지를 전송한다.

* @param pszCallId SIP Call-ID

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

*/

bool CSipUserAgent::ResumeCall( const char * pszCallId )

{

SIP_DIALOG_MAP::iterator itMap;

CSipMessage * pclsRequest = NULL;

bool bRes = false;

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

itMap->second.m_eLocalDirection = E_RTP_SEND_RECV;

pclsRequest = itMap->second.CreateInvite();

bRes = true;

}

m_clsDialogMutex.release();

if( pclsRequest )

{

m_clsSipStack.SendSipMessage( pclsRequest );

}

return bRes;

}

/**

* @ingroup SipUserAgent

* @brief 통화 개수를 리턴한다.

* @returns 통화 개수를 리턴한다.

*/

int CSipUserAgent::GetCallCount( )

{

int iCallCount;

m_clsDialogMutex.acquire();

iCallCount = (int)m_clsDialogMap.size();

m_clsDialogMutex.release();

return iCallCount;

}

/**

* @ingroup SipUserAgent

* @brief 모든 통화의 SIP Call-ID 를 리스트에 저장한다.

* @param clsList SIP Call-ID 를 저장할 변수

*/

void CSipUserAgent::GetCallIdList( SIP_CALL_ID_LIST & clsList )

{

SIP_DIALOG_MAP::iterator itMap;

clsList.clear();

m_clsDialogMutex.acquire();

for( itMap = m_clsDialogMap.begin(); itMap != m_clsDialogMap.end(); ++itMap )

{

clsList.push_back( itMap->first );

}

m_clsDialogMutex.release();

}

/**

* @ingroup SipUserAgent

* @brief 모든 통화를 종료시킨다.

*/

void CSipUserAgent::StopCallAll( )

{

SIP_CALL_ID_LIST clsList;

SIP_CALL_ID_LIST::iterator itList;

GetCallIdList( clsList );

for( itList = clsList.begin(); itList != clsList.end(); ++itList )

{

StopCall( itList->c_str() );

}

}

/**

* @ingroup SipUserAgent

* @brief SIP INVITE 메시지를 생성하고 Dialog 에 저장한다. 본 메소드는 SIP INVITE 메시지를 전송하지는 않는다.

* 본 메소드로 생성된 SIP INVITE 메시지를 전송하려면 StartCall( const char * pszCallId, CSipMessage * pclsInvite ) 메소드를 실행해야 한다.

* 본 메소드가 개발된 이유는 StartCall( const char * pszFrom, const char * pszTo, CSipCallRtp * pclsRtp, CSipCallRoute * pclsRoute, std::string & strCallId ) 을 호출한 후,

* 응용 프로그램에서 생성된 SIP Call-ID 를 자료구조에 저장하기 전에 SIP INVITE 응답 메시지를 수신하면 응용 프로그램에서 해당 통화 정보가

* 자료구조에 없어서 정상적으로 처리하지 못 하기 때문에 이를 방지하기 위해서 SIP INVITE 메시지와 SIP Dialog 를 생성하여서 SIP Call-ID 를

* 응용 프로그램에 전달하여 응용 프로그램 자료구조에 저장한 후, SIP INVITE 메시지를 전송하기 위함이다.

* @param pszFrom 발신자 아이디

* @param pszTo 수신자 아이디

* @param pclsRtp local RTP 정보 저장 객체

* @param pclsRoute SIP 메시지 목적지 주소 저장 객체

* @param strCallId 생성된 INVITE 메시지의 Call-ID 가 저장될 변수

* @param ppclsInvite 생성된 SIP INVITE 메시지가 저장될 변수

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

*/

bool CSipUserAgent::CreateCall( const char * pszFrom, const char * pszTo, CSipCallRtp * pclsRtp, CSipCallRoute * pclsRoute, std::string & strCallId, CSipMessage ** ppclsInvite )

{

if( pszFrom == NULL || pszTo == NULL ) return false;

if( pclsRtp == NULL || pclsRoute == NULL ) return false;

if( strlen( pszFrom ) == 0 || strlen( pszTo ) == 0 ) return false;

if( pclsRoute->m_strDestIp.empty() ) return false;

if( pclsRoute->m_iDestPort <= 0 || pclsRoute->m_iDestPort > 65535 ) return false;

if( ppclsInvite == NULL ) return false;

#ifndef USE_TLS

if( pclsRoute->m_eTransport == E_SIP_TLS ) return false;

#endif

CSipDialog clsDialog( &m_clsSipStack );

clsDialog.m_strFromId = pszFrom;

clsDialog.m_strToId = pszTo;

clsDialog.SetLocalRtp( pclsRtp );

clsDialog.m_strContactIp = pclsRoute->m_strDestIp;

clsDialog.m_iContactPort = pclsRoute->m_iDestPort;

clsDialog.m_eTransport = pclsRoute->m_eTransport;

clsDialog.m_b100rel = pclsRoute->m_b100rel;

SIP_DIALOG_MAP::iterator itMap;

char szTag[SIP_TAG_MAX_SIZE], szBranch[SIP_BRANCH_MAX_SIZE], szCallIdName[SIP_CALL_ID_NAME_MAX_SIZE];

bool bInsert = false;

CSipMessage * pclsMessage = NULL;

SipMakeTag( szTag, sizeof(szTag) );

SipMakeBranch( szBranch, sizeof(szBranch) );

clsDialog.m_strFromTag = szTag;

clsDialog.m_strViaBranch = szBranch;

gettimeofday( &clsDialog.m_sttInviteTime, NULL );

while( 1 )

{

SipMakeCallIdName( szCallIdName, sizeof(szCallIdName) );

clsDialog.m_strCallId = szCallIdName;

clsDialog.m_strCallId.append( "@" );

clsDialog.m_strCallId.append( m_clsSipStack.m_clsSetup.m_strLocalIp );

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( clsDialog.m_strCallId );

if( itMap == m_clsDialogMap.end() )

{

pclsMessage = clsDialog.CreateInvite();

if( pclsMessage )

{

m_clsDialogMap.insert( SIP_DIALOG_MAP::value_type( clsDialog.m_strCallId, clsDialog ) );

bInsert = true;

}

}

m_clsDialogMutex.release();

if( bInsert ) break;

}

strCallId = clsDialog.m_strCallId;

*ppclsInvite = pclsMessage;

return true;

}

/**

* @ingroup SipUserAgent

* @brief CreateCall 메소드로 생성된 INVITE 메시지를 전송한다.

* @param pszCallId SIP Call-ID

* @param pclsInvite SIP INVITE 메시지

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

*/

bool CSipUserAgent::StartCall( const char * pszCallId, CSipMessage * pclsInvite )

{

if( pszCallId == NULL || pclsInvite == NULL ) return false;

if( m_clsSipStack.SendSipMessage( pclsInvite ) == false )

{

Delete( pszCallId );

return false;

}

return true;

}

/**

* @ingroup SipUserAgent

* @brief Blind transfer 를 실행한다.

* @param pszCallId SIP Call-ID

* @param pszTo 통화 전달을 받을 아이디

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

*/

bool CSipUserAgent::TransferCallBlind( const char * pszCallId, const char * pszTo )

{

if( pszCallId == NULL || pszTo == NULL ) return false;

SIP_DIALOG_MAP::iterator itMap;

CSipMessage * pclsMessage = NULL;

char szReferTo[1024], szReferBy[512];

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

pclsMessage = itMap->second.CreateRefer();

snprintf( szReferTo, sizeof(szReferTo), "<sip:%s@%s:%d>"

, pszTo, itMap->second.m_strContactIp.c_str(), itMap->second.m_iContactPort );

snprintf( szReferBy, sizeof(szReferBy), "<sip:%s@%s:%d>"

, itMap->second.m_strFromId.c_str(), m_clsSipStack.m_clsSetup.m_strLocalIp.c_str(), m_clsSipStack.m_clsSetup.m_iLocalUdpPort );

}

m_clsDialogMutex.release();

if( pclsMessage == NULL ) return false;

pclsMessage->AddHeader( "Refer-To", szReferTo );

pclsMessage->AddHeader( "Referred-By", szReferBy );

return m_clsSipStack.SendSipMessage( pclsMessage );

}

/**

* @ingroup SipUserAgent

* @brief Screened / Unscreened transfer 를 실행한다.

* @param pszCallId SIP Call-ID

* @param pszToCallId 통화 전달을 받을 SIP Call-ID

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

*/

bool CSipUserAgent::TransferCall( const char * pszCallId, const char * pszToCallId )

{

if( pszCallId == NULL || pszToCallId == NULL ) return false;

SIP_DIALOG_MAP::iterator itMap;

CSipMessage * pclsMessage = NULL;

std::string strReplaces, strToId, strFromId;

char szReferTo[1024], szReferBy[512];

m_clsDialogMutex.acquire();

itMap = m_clsDialogMap.find( pszToCallId );

if( itMap != m_clsDialogMap.end() )

{

strFromId = itMap->second.m_strFromId;

strToId = itMap->second.m_strToId;

strReplaces = pszToCallId;

ReplaceString( strReplaces, "@", "%40" );

strReplaces.append( "%3B" );

strReplaces.append( "to-tag%3D" );

strReplaces.append( itMap->second.m_strToTag );

strReplaces.append( "%3B" );

strReplaces.append( "from-tag%3D" );

strReplaces.append( itMap->second.m_strFromTag );

snprintf( szReferTo, sizeof(szReferTo), "<sip:%s@%s:%d?Replaces=%s>"

, strToId.c_str(), itMap->second.m_strContactIp.c_str(), itMap->second.m_iContactPort, strReplaces.c_str() );

snprintf( szReferBy, sizeof(szReferBy), "<sip:%s@%s:%d>"

, strFromId.c_str(), m_clsSipStack.m_clsSetup.m_strLocalIp.c_str(), m_clsSipStack.m_clsSetup.m_iLocalUdpPort );

itMap = m_clsDialogMap.find( pszCallId );

if( itMap != m_clsDialogMap.end() )

{

pclsMessage = itMap->second.CreateRefer();

}

}

m_clsDialogMutex.release();

if( pclsMessage == NULL ) return false;

pclsMessage->AddHeader( "Refer-To", szReferTo );

pclsMessage->AddHeader( "Referred-By", szReferBy );

return m_clsSipStack.SendSipMessage( pclsMessage );

}