/*

* 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

*/

#include "SipStackDefine.h"

#include "SipServerMap.h"

#include "UserMap.h"

#include "MemoryDebug.h"

CSipServerMap gclsSipServerMap;

CSipServerInfo::CSipServerInfo() : m_iPort(5060), m_bUse(true), m_bDelete(false)

{

}

/**

* @ingroup SipLoadBalancer

* @brief 입력된 IP 주소 및 포트 번호가 SIP 서버 정보와 일치하는지 검사한다.

* @param pszIp IP 주소

* @param iPort 포트 번호

* @returns 입력‰ 정보가 SIP 서버 정보와 일치하면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.

*/

bool CSipServerInfo::Equal( const char * pszIp, int iPort )

{

if( !strcmp( m_strIp.c_str(), pszIp ) && m_iPort == iPort ) return true;

return false;

}

CSipServerMap::CSipServerMap() : m_iIndex(0)

{

}

CSipServerMap::~CSipServerMap()

{

}

/**

* @ingroup SipLoadBalancer

* @brief SIP 서버 정보를 리스트에 추가한다.

* @param pszIp IP 주소

* @param iPort 포트 번호

* @param bUse 사용 유무

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

*/

bool CSipServerMap::Insert( const char * pszIp, int iPort, bool bUse )

{

if( pszIp == NULL || strlen(pszIp) == 0 ) return false;

if( iPort <= 0 || iPort > 65535 ) return false;

// 이미 리스트에 존재하면 추가하지 않는다.

if( Select( pszIp, iPort, bUse ) ) return true;

CSipServerInfo clsInfo;

clsInfo.m_strIp = pszIp;

clsInfo.m_iPort = iPort;

clsInfo.m_bUse = bUse;

m_clsMutex.acquire();

m_clsList.push_back( clsInfo );

m_clsMutex.release();

return true;

}

/**

* @ingroup SipLoadBalancer

* @brief 접속할 SIP 서버 정보를 가져온다.

* @param clsInfo SIP 서버 정보를 저장하는 변수

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

*/

bool CSipServerMap::SelectNext( CSipServerInfo & clsInfo )

{

bool bRes = false;

m_clsMutex.acquire();

int iCount = (int)m_clsList.size();

if( iCount > 0 )

{

if( m_iIndex >= iCount )

{

m_iIndex = 0;

}

clsInfo = m_clsList[m_iIndex];

++m_iIndex;

bRes = true;

}

m_clsMutex.release();

return bRes;

}

/**

* @ingroup SipLoadBalancer

* @brief 입력된 IP 주소, 포트번호와 일치하는 SIP 서버 정보가 존재하는지 검사한다.

* @param pszIp IP 주소

* @param iPort 포트 번호

* @returns SIP 서버가 검색되면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.

*/

bool CSipServerMap::Select( const char * pszIp, int iPort )

{

SIP_SERVER_LIST::iterator itList;

bool bFound = false;

m_clsMutex.acquire();

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

{

if( itList->Equal( pszIp, iPort ) )

{

bFound = true;

break;

}

}

m_clsMutex.release();

return bFound;

}

/**

* @ingroup SipLoadBalancer

* @brief 입력된 IP 주소, 포트번호와 일치하는 SIP 서버 정보가 존재하면 사용 유무 및 삭제 유무를 업데이트한다.

* @param pszIp IP 주소

* @param iPort 포트 번호

* @param bUse 사용 유무

* @returns SIP 서버가 검색되면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.

*/

bool CSipServerMap::Select( const char * pszIp, int iPort, bool bUse )

{

SIP_SERVER_LIST::iterator itList;

bool bFound = false;

m_clsMutex.acquire();

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

{

if( itList->Equal( pszIp, iPort ) )

{

itList->m_bUse = bUse;

itList->m_bDelete = false;

bFound = true;

break;

}

}

m_clsMutex.release();

return bFound;

}

/**

* @ingroup SipLoadBalancer

* @brief 모든 SIP 서버 정보에 삭제 표시를 한다.

*/

void CSipServerMap::SetDeleteAll( )

{

SIP_SERVER_LIST::iterator itList;

m_clsMutex.acquire();

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

{

itList->m_bDelete = true;

}

m_clsMutex.release();

}

/**

* @ingroup SipLoadBalancer

* @brief 삭제 표시가 된 SIP 서버를 삭제한다.

*/

void CSipServerMap::DeleteIfSet( )

{

SIP_SERVER_LIST::iterator itList, itNext;

m_clsMutex.acquire();

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

{

LOOP_START:

if( itList->m_bDelete == false )

{

if( itList->m_bUse == false )

{

gclsUserMap.DeleteSipServer( itList->m_strIp.c_str(), itList->m_iPort );

}

continue;

}

itNext = itList;

++itNext;

gclsUserMap.DeleteSipServer( itList->m_strIp.c_str(), itList->m_iPort );

m_clsList.erase( itList );

if( itNext == m_clsList.end() ) break;

itList = itNext;

goto LOOP_START;

}

m_clsMutex.release();

}

/**

* @ingroup SipLoadBalancer

* @brief 자료구조 모니터링용 문자열을 생성한다.

* @param strBuf 자료구조 모니터링용 문자열 변수

*/

void CSipServerMap::GetString( CMonitorString & strBuf )

{

SIP_SERVER_LIST::iterator itList;

strBuf.Clear();

m_clsMutex.acquire();

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

{

strBuf.AddCol( itList->m_strIp, itList->m_iPort );

strBuf.AddCol( itList->m_bUse ? "use" : "no use" );

strBuf.AddRow( itList->m_bDelete ? "del" : "" );

}

m_clsMutex.release();

}