404 lines
9.4 KiB
C++
404 lines
9.4 KiB
C++
// MemStatConfigDlg.cpp : 구현 파일
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "ServiceInstaller.h"
|
|
#include ".\ServiceInstallerdlg.h"
|
|
#include "ServiceScriptInfo.h"
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#endif
|
|
|
|
// 응용 프로그램 정보에 사용되는 CAboutDlg 대화 상자입니다.
|
|
|
|
#define MAX_ERRORMSG_SIZE 128
|
|
|
|
class CAboutDlg : public CDialog
|
|
{
|
|
public:
|
|
CAboutDlg();
|
|
|
|
// 대화 상자 데이터
|
|
enum { IDD = IDD_ABOUTBOX };
|
|
|
|
protected:
|
|
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 지원
|
|
|
|
// 구현
|
|
protected:
|
|
DECLARE_MESSAGE_MAP()
|
|
};
|
|
|
|
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
|
|
{
|
|
}
|
|
|
|
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
}
|
|
|
|
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
|
|
END_MESSAGE_MAP()
|
|
|
|
|
|
// CServiceInstallerDlg 대화 상자
|
|
|
|
|
|
|
|
CServiceInstallerDlg::CServiceInstallerDlg(CWnd* pParent /*=NULL*/)
|
|
: CDialog(CServiceInstallerDlg::IDD, pParent)
|
|
, m_strEditExplaination(_T(""))
|
|
, m_strNowStatus(_T(""))
|
|
, m_strServiceName(_T(""))
|
|
, m_nServiceType(0)
|
|
{
|
|
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
|
|
|
|
m_nComboIdx = 0;
|
|
m_nServiceType = 0;
|
|
}
|
|
|
|
void CServiceInstallerDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
DDX_Text(pDX, IDC_EDIT_EXPLANATION, m_strEditExplaination);
|
|
DDX_Text(pDX, IDC_STATIC_NOW_STATUS, m_strNowStatus);
|
|
DDX_Text(pDX, IDC_STATIC_NAME, m_strServiceName);
|
|
DDX_Control(pDX, IDC_COMBO_SERVICE, m_comboService);
|
|
DDX_Control(pDX, IDC_COMBO_SERVICE_TYPE, m_comboServiceType);
|
|
}
|
|
|
|
BEGIN_MESSAGE_MAP(CServiceInstallerDlg, CDialog)
|
|
ON_WM_SYSCOMMAND()
|
|
ON_WM_PAINT()
|
|
ON_WM_QUERYDRAGICON()
|
|
//}}AFX_MSG_MAP
|
|
ON_BN_CLICKED(IDC_BTN_INSTALL, OnBnClickedBtnInstall)
|
|
ON_BN_CLICKED(IDC_BTN_UNINSTALL, OnBnClickedBtnUninstall)
|
|
ON_BN_CLICKED(IDOK, OnBnClickedOk)
|
|
ON_CBN_SELCHANGE(IDC_COMBO_SERVICE, OnCbnSelchangeComboService)
|
|
ON_CBN_SELCHANGE(IDC_COMBO_SERVICE_TYPE, OnCbnSelchangeComboServiceType)
|
|
END_MESSAGE_MAP()
|
|
|
|
|
|
// CServiceInstallerDlg 메시지 처리기
|
|
|
|
BOOL CServiceInstallerDlg::OnInitDialog()
|
|
{
|
|
CDialog::OnInitDialog();
|
|
|
|
// 시스템 메뉴에 "정보..." 메뉴 항목을 추가합니다.
|
|
|
|
// IDM_ABOUTBOX는 시스템 명령 범위에 있어야 합니다.
|
|
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
|
|
ASSERT(IDM_ABOUTBOX < 0xF000);
|
|
|
|
CMenu* pSysMenu = GetSystemMenu(FALSE);
|
|
if (pSysMenu != NULL)
|
|
{
|
|
CString strAboutMenu;
|
|
strAboutMenu.LoadString(IDS_ABOUTBOX);
|
|
if (!strAboutMenu.IsEmpty())
|
|
{
|
|
pSysMenu->AppendMenu(MF_SEPARATOR);
|
|
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
|
|
}
|
|
}
|
|
|
|
// 이 대화 상자의 아이콘을 설정합니다. 응용 프로그램의 주 창이 대화 상자가 아닐 경우에는
|
|
// 프레임워크가 이 작업을 자동으로 수행합니다.
|
|
SetIcon(m_hIcon, TRUE); // 큰 아이콘을 설정합니다.
|
|
SetIcon(m_hIcon, FALSE); // 작은 아이콘을 설정합니다.
|
|
|
|
// TODO: 여기에 추가 초기화 작업을 추가합니다.
|
|
if( !m_info.LoadData() )
|
|
AfxMessageBox( "서비스 데이터 읽기 에러" );
|
|
|
|
// 서비스 콤보박스 초기화
|
|
InitServiceCombo();
|
|
m_comboService.SetCurSel( m_nComboIdx );
|
|
m_comboServiceType.SetCurSel( m_nServiceType );
|
|
|
|
DispCurStatus();
|
|
|
|
SetWindowText( "ServiceInstaller" );
|
|
|
|
return TRUE; // 컨트롤에 대한 포커스를 설정하지 않을 경우 TRUE를 반환합니다.
|
|
}
|
|
|
|
void CServiceInstallerDlg::OnSysCommand(UINT nID, LPARAM lParam)
|
|
{
|
|
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
|
|
{
|
|
CAboutDlg dlgAbout;
|
|
dlgAbout.DoModal();
|
|
}
|
|
else
|
|
{
|
|
CDialog::OnSysCommand(nID, lParam);
|
|
}
|
|
}
|
|
|
|
// 대화 상자에 최소화 단추를 추가할 경우 아이콘을 그리려면
|
|
// 아래 코드가 필요합니다. 문서/뷰 모델을 사용하는 MFC 응용 프로그램의 경우에는
|
|
// 프레임워크에서 이 작업을 자동으로 수행합니다.
|
|
|
|
void CServiceInstallerDlg::OnPaint()
|
|
{
|
|
if (IsIconic())
|
|
{
|
|
CPaintDC dc(this); // 그리기를 위한 디바이스 컨텍스트
|
|
|
|
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
|
|
|
|
// 클라이언트 사각형에서 아이콘을 가운데에 맞춥니다.
|
|
int cxIcon = GetSystemMetrics(SM_CXICON);
|
|
int cyIcon = GetSystemMetrics(SM_CYICON);
|
|
CRect rect;
|
|
GetClientRect(&rect);
|
|
int x = (rect.Width() - cxIcon + 1) / 2;
|
|
int y = (rect.Height() - cyIcon + 1) / 2;
|
|
|
|
// 아이콘을 그립니다.
|
|
dc.DrawIcon(x, y, m_hIcon);
|
|
}
|
|
else
|
|
{
|
|
CDialog::OnPaint();
|
|
}
|
|
}
|
|
|
|
// 사용자가 최소화된 창을 끄는 동안에 커서가 표시되도록 시스템에서
|
|
// 이 함수를 호출합니다.
|
|
HCURSOR CServiceInstallerDlg::OnQueryDragIcon()
|
|
{
|
|
return static_cast<HCURSOR>(m_hIcon);
|
|
}
|
|
|
|
void CServiceInstallerDlg::OnBnClickedBtnInstall()
|
|
{
|
|
// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
|
|
if( m_nComboIdx == 0 )
|
|
{
|
|
// 전 서비스를 등록한다.
|
|
// 현제 등록되어 있는 써비스는 제외한다.
|
|
for( int i = 1; i < m_info.GetServiceCount(); i++ )
|
|
InstallService( i, m_nServiceType );
|
|
}
|
|
else
|
|
{
|
|
InstallService( m_nComboIdx, m_nServiceType );
|
|
}
|
|
}
|
|
|
|
|
|
void CServiceInstallerDlg::OnBnClickedBtnUninstall()
|
|
{
|
|
// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
|
|
if( m_nComboIdx == 0 )
|
|
{
|
|
// 전 서비스를 등록한다.
|
|
// 현제 등록되어 있는 써비스는 제외한다.
|
|
for( int i = 1; i < m_info.GetServiceCount(); i++ )
|
|
UnInstallService( i );
|
|
}
|
|
else
|
|
{
|
|
UnInstallService( m_nComboIdx );
|
|
}
|
|
|
|
}
|
|
|
|
|
|
void CServiceInstallerDlg::OnBnClickedOk()
|
|
{
|
|
// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
|
|
OnOK();
|
|
}
|
|
|
|
void CServiceInstallerDlg::DispCurStatus()
|
|
{
|
|
|
|
if( m_nComboIdx == 0 )
|
|
{
|
|
m_strServiceName.Format( "서비스 이름 : 전체 " );
|
|
m_strNowStatus = "";
|
|
}
|
|
else if( m_nComboIdx > 0 )
|
|
{
|
|
TCHAR szServiceName[MAX_PATH];
|
|
m_info.GetServiceName( szServiceName, MAX_PATH, m_nComboIdx );
|
|
|
|
if( !szServiceName )
|
|
{
|
|
MessageBox( NULL, "해당 서비스 이름를 가지고 올수 없습니다." );
|
|
return;
|
|
}
|
|
|
|
SC_HANDLE hScm, hSrv;
|
|
hScm = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
|
|
if( NULL == hScm )
|
|
{
|
|
MessageBox( "SCM를 열수 없습니다.", "알림", MB_OK );
|
|
return;
|
|
}
|
|
|
|
m_strServiceName.Format( "서비스 이름 : %s ", szServiceName );
|
|
|
|
// 서비스의 핸들을 구한다.
|
|
hSrv = OpenService( hScm, szServiceName, SERVICE_ALL_ACCESS );
|
|
if( NULL == hSrv )
|
|
m_strNowStatus = "서비스가 설치되어 있지 않습니다.";
|
|
else
|
|
m_strNowStatus = "서비스가 설치되어 있습니다.";
|
|
|
|
|
|
CloseServiceHandle( hScm );
|
|
}
|
|
else
|
|
return;
|
|
|
|
UpdateData( FALSE );
|
|
}
|
|
|
|
void CServiceInstallerDlg::OnCbnSelchangeComboService()
|
|
{
|
|
// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
|
|
m_nComboIdx = m_comboService.GetCurSel();
|
|
DispCurStatus();
|
|
}
|
|
|
|
void CServiceInstallerDlg::InitServiceCombo(void)
|
|
{
|
|
// 서비스 어플리케이션 콤보 설정
|
|
m_comboService.ResetContent();
|
|
m_comboService.AddString("전체");
|
|
|
|
for( int i = 1; i <= m_info.GetServiceCount(); i++ )
|
|
{
|
|
TCHAR szName[MAX_PATH];
|
|
m_info.GetAppName( szName, MAX_PATH, i );
|
|
m_comboService.AddString( szName );
|
|
}
|
|
|
|
// 서비스 설치 종류 콤보 설정
|
|
m_comboServiceType.ResetContent();
|
|
m_comboServiceType.AddString("수동시작");
|
|
m_comboServiceType.AddString("자동시작");
|
|
}
|
|
|
|
//서비스를 설치한다.
|
|
BOOL CServiceInstallerDlg::InstallService( int nAppNum, int nServiceType )
|
|
{
|
|
SC_HANDLE hScm, hSrv;
|
|
SERVICE_DESCRIPTION lpDes;
|
|
|
|
//SCM을 연다.
|
|
hScm = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
|
|
if( hScm == NULL )
|
|
{
|
|
MessageBox( "SCM을 열 수 없습니다.", "알림", MB_OK );
|
|
return FALSE;
|
|
}
|
|
|
|
//서비스를 등록한다.
|
|
TCHAR szServiceName[MAX_PATH], szDispName[MAX_PATH], szBinPath[MAX_PATH];
|
|
m_info.GetServiceName( szServiceName, MAX_PATH, nAppNum );
|
|
m_info.GetDisplayName( szDispName, MAX_PATH, nAppNum );
|
|
m_info.GetBinPath( szBinPath, MAX_PATH, nAppNum );
|
|
|
|
int nStartType = 0;
|
|
if( nServiceType == 0 )
|
|
nStartType = SERVICE_DEMAND_START;
|
|
else if( nServiceType == 1)
|
|
nStartType = SERVICE_AUTO_START;
|
|
|
|
hSrv = CreateService( hScm, szServiceName, szDispName, SERVICE_PAUSE_CONTINUE | SERVICE_CHANGE_CONFIG,
|
|
SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS, nStartType, SERVICE_ERROR_IGNORE, szBinPath,
|
|
NULL, NULL, NULL, NULL, NULL );
|
|
if( NULL == hSrv )
|
|
{
|
|
DWORD dwError = GetLastError();
|
|
char szErr[MAX_ERRORMSG_SIZE];
|
|
wsprintf( szErr, " 설치하지 못했습니다. 에러코드 = %d ", dwError );
|
|
MessageBox( szErr, "알림", MB_OK );
|
|
}
|
|
else
|
|
{
|
|
UpdateData( TRUE );
|
|
//설명을 등록한다.
|
|
lpDes.lpDescription = (LPSTR)m_strEditExplaination.GetString();
|
|
ChangeServiceConfig2( hSrv, SERVICE_CONFIG_DESCRIPTION, &lpDes );
|
|
m_strNowStatus = "설치되어 있습니다.";
|
|
CloseServiceHandle( hSrv );
|
|
UpdateData( FALSE );
|
|
}
|
|
|
|
CloseServiceHandle( hScm );
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
//서비스를 제거한다.
|
|
BOOL CServiceInstallerDlg::UnInstallService( int nAppNum )
|
|
{
|
|
SC_HANDLE hScm, hSrv;
|
|
SERVICE_STATUS ss;
|
|
|
|
TCHAR szServiceName[MAX_PATH];
|
|
m_info.GetServiceName( szServiceName, MAX_PATH, nAppNum );
|
|
|
|
// SCM를 연다.
|
|
hScm = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
|
|
if( NULL == hScm )
|
|
{
|
|
MessageBox( "SCM를 열수 없습니다.", "알림", MB_OK );
|
|
return FALSE;
|
|
}
|
|
|
|
// 서비스의 핸들을 구한다.
|
|
hSrv = OpenService( hScm, szServiceName, SERVICE_ALL_ACCESS );
|
|
if( NULL == hSrv )
|
|
{
|
|
CloseServiceHandle( hScm );
|
|
MessageBox( "서비스가 설치되어 있지 않습니다", "알림", MB_OK );
|
|
return FALSE;
|
|
}
|
|
|
|
// 실행중이면 중지시킨다.
|
|
QueryServiceStatus( hSrv, &ss );
|
|
if( ss.dwCurrentState != SERVICE_STOPPED )
|
|
{
|
|
ControlService( hSrv, SERVICE_CONTROL_STOP, &ss );
|
|
Sleep(2000);
|
|
}
|
|
|
|
// 서비스 제거
|
|
if( DeleteService( hSrv ) )
|
|
{
|
|
m_strNowStatus = "설치되지 않았습니다.";
|
|
}
|
|
else
|
|
{
|
|
DWORD dwError = GetLastError();
|
|
char szErr[MAX_ERRORMSG_SIZE];
|
|
wsprintf( szErr, " 서비스를 제거하지 못했습니다. 에러코드 = %d ", dwError );
|
|
MessageBox( szErr, "알림", MB_OK );
|
|
}
|
|
|
|
CloseServiceHandle( hSrv );
|
|
CloseServiceHandle( hScm );
|
|
|
|
UpdateData( FALSE );
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
void CServiceInstallerDlg::OnCbnSelchangeComboServiceType()
|
|
{
|
|
// TODO: 여기에 컨트롤 알림 처리기 코드를 추가합니다.
|
|
m_nServiceType = m_comboServiceType.GetCurSel();
|
|
}
|