VS2008 用 ATL 写 Windows 服务

来源：互联网发布：网络规划师真题编辑：程序博客网时间：2024/05/22 18:34

1. 新建 ATL 工程，比如 ClientItem

2. 选择服务程序，然后点“完成”

3. 在类 CClientItemModule （ClientItem.cpp）添加以下代码

//修改服务名称和描述
HRESULT RegisterAppId(bool bService = false) throw()
{
  HRESULT hr = S_OK;
  BOOL res = __super::RegisterAppId(bService);
  if( !bService )
   return hr;
  if( !IsInstalled() )
   return hr;

  SC_HANDLE hSCM = ::OpenSCManagerW(NULL, NULL, SERVICE_CHANGE_CONFIG);
  SC_HANDLE hService = NULL;
  if (hSCM == NULL)
  {
   hr = AtlHresultFromLastError();
   return hr;
  }

  hService = ::OpenService(hSCM, m_szServiceName, SERVICE_CHANGE_CONFIG);
  if (hService != NULL)
  {
   ::ChangeServiceConfig( hService,
         SERVICE_NO_CHANGE,
         SERVICE_AUTO_START, //修改服务为自动启动
         NULL, NULL, NULL, NULL, NULL, NULL, NULL,
         m_szServiceName); // 通过修改资源IDS_SERVICENAME 修改服务的显示名字

   SERVICE_DESCRIPTION Description;
   TCHAR szDescription[1024];
   ZeroMemory(szDescription, 1024);
   ZeroMemory(&Description, sizeof(SERVICE_DESCRIPTION));
   lstrcpy(szDescription, _T("服务描述"));
   Description.lpDescription = szDescription;
   ::ChangeServiceConfig2(hService, SERVICE_CONFIG_DESCRIPTION, &Description);
   ::CloseServiceHandle(hService);
  }
  else
  {
   hr = AtlHresultFromLastError();
  }

  ::CloseServiceHandle(hSCM);

  return hr;
}

//初始化代码
HRESULT PreMessageLoop(int nShowCmd) throw()
{
LogEvent( _T("进入 PreMessageLoop") );

  HRESULT hr = __super::PreMessageLoop(nShowCmd);
  if (SUCCEEDED(hr))
  {
   //启动主线程
   DWORD dwId = 0;
   m_hMainThread = CreateThread( NULL, 0, (LPTHREAD_START_ROUTINE)CMainServiceThread::_main_service_thread, &_mainServiceThread, 0, &dwId );
  }
  return hr;
}

//消息循环
void RunMessageLoop( ) throw()
{
LogEvent( _T("进入 RunMessageLoop") );

SetServiceStatus(SERVICE_RUNNING);
__super::RunMessageLoop();
}

//卸载服务
HRESULT PostMessageLoop() throw()
{
LogEvent( _T("进入 PostMessageLoop") );

  HRESULT hr = __super::PostMessageLoop();
  if (SUCCEEDED(hr))
  {
   //停止服务线程
   TerminateThread( m_hMainThread, 0 );
   CloseHandle(m_hMainThread);
  }

return hr;
}

private:
HANDLE m_hMainThread;

注意：CMainServiceThread 是一个自定义的类，里面就是一个线程运行体，根据自身需要编写代码。

4. 有一个特别要注意的，我也没去研究原因，不过记住就好了，在 stdafx.h 中，要加入 #define _ATL_NO_COM_SUPPORT，代码如下：

#pragma once

#ifndef STRICT
#define STRICT
#endif

#include "targetver.h"

#define _ATL_NO_COM_SUPPORT
#define _ATL_APARTMENT_THREADED
#define _ATL_NO_AUTOMATIC_NAMESPACE

#define _ATL_CSTRING_EXPLICIT_CONSTRUCTORS // 某些 CString 构造函数将是显式的

#include "resource.h"
#include <atlbase.h>
#include <atlcom.h>
#include <atlctl.h>

using namespace ATL;

如果缺少这代码，你的服务运行不了。