/*
* Copyright (C) 2012 Yee Young Han (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
*/
#ifdef WIN32
#include "SipPlatformDefine.h"
#include "ServerServicePrivate.h"
#include
#include
#include "MemoryDebug.h"
/**
* @ingroup ServerPlatform
* @brief 현재 프로그램이 서비스에 등록되어 있는지 확인한다.
* @returns 현재 프로그램이 서비스에 등록되어 있으면 true 를 리턴한다.
* 그렇지 않으면 false 를 리턴한다.
*/
bool IsExistService()
{
SC_HANDLE hScm, hSrv;
bool fisResult = false;
hScm = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
if( hScm )
{
hSrv = OpenService( hScm, gclsService.m_strName.c_str(), SERVICE_ALL_ACCESS );
if( hSrv )
{
CloseServiceHandle( hSrv );
fisResult = true;
}
CloseServiceHandle(hScm);
}
return fisResult;
}
/**
* @ingroup ServerPlatform
* @brief 현재 프로그램을 서비스로 등록한다.
* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.
*/
bool InstallService( )
{
if( IsExistService() )
{
printf( "%s is already installed.\n", gclsService.m_strName.c_str() );
return false;
}
SC_HANDLE hScm, hSrv;
char SrvPath[MAX_PATH];
SERVICE_DESCRIPTION lpDes;
// SCM을 연다
hScm=OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
if (hScm==NULL)
{
printf( "Can not open SCM. Start dos with administrator authority\n" );
return false;
}
// 등록할 서비스 파일이 있는지 조사해 보고 경로를 구한다.
::GetModuleFileName( NULL, SrvPath, sizeof(SrvPath) );
// 서비스를 등록한다.
hSrv=CreateService( hScm
, gclsService.m_strName.c_str()
, gclsService.m_strDisplayName.c_str()
, SERVICE_PAUSE_CONTINUE | SERVICE_CHANGE_CONFIG
, SERVICE_WIN32_OWN_PROCESS
, SERVICE_AUTO_START
, SERVICE_ERROR_IGNORE
, SrvPath
, NULL,NULL,NULL,NULL,NULL);
if( hSrv == NULL )
{
printf( "Install is failed.\n" );
}
else
{
// 설명을 등록한다.
lpDes.lpDescription = (LPSTR)gclsService.m_strDescription.c_str();
ChangeServiceConfig2( hSrv, SERVICE_CONFIG_DESCRIPTION, &lpDes );
printf( "Installed.\n" );
CloseServiceHandle(hSrv);
}
CloseServiceHandle(hScm);
return true;
}
/**
* @ingroup ServerPlatform
* @brief 현재 프로그램을 서비스에서 등록 해제한다.
* @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.
*/
bool UninstallService( )
{
SC_HANDLE hScm, hSrv;
SERVICE_STATUS ss;
// SCM을 연다
hScm = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
if( hScm == NULL )
{
printf( "Can not open SCM.\n" );
return false;
}
// 서비스의 핸들을 구한다.
hSrv = OpenService( hScm, gclsService.m_strName.c_str(), SERVICE_ALL_ACCESS );
if( hSrv == NULL )
{
CloseServiceHandle(hScm);
printf( "%s is not installed.\n", gclsService.m_strName.c_str() );
return false;
}
// 실행중이면 중지시킨다.
QueryServiceStatus( hSrv, &ss );
if( ss.dwCurrentState != SERVICE_STOPPED )
{
ControlService( hSrv, SERVICE_CONTROL_STOP, &ss );
Sleep( 2000 );
}
// 서비스 제거
if( DeleteService( hSrv ) )
{
printf( "Service is deleted.\n" );
}
else
{
printf( "Deleting service is failed.\n" );
}
CloseServiceHandle( hSrv );
CloseServiceHandle( hScm );
return true;
}
#endif