编写 ServiceMain 函数
以下示例中的 SvcMain 函数是示例服务的 ServiceMain 函数。 SvcMain 可以按照控制台应用程序的 main 函数的方式访问服务的命令行参数。 第一个参数包含在第二个参数中传递给服务的参数数。 始终至少有一个参数。 第二个参数是指向字符串指针数组的指针。 数组中的第一项始终是服务名称。
SvcMain 函数首先调用 RegisterServiceCtrlHandler 函数,以将 SvcCtrlHandler 函数注册为服务的 Handler 函数并开始初始化。 RegisterServiceCtrlHandler 应为 ServiceMain 中的第一个非失败函数,以便服务可以使用此函数返回的状态句柄在发生错误时调用具有SERVICE_STOPPED状态的 SetServiceStatus 。
接下来,SvcMain 函数调用 ReportSvcStatus 函数以指示其初始状态为SERVICE_START_PENDING。 当服务处于此状态时,不会接受任何控件。 为了简化服务的逻辑,建议服务在执行初始化时不要接受任何控件。
最后,SvcMain 函数调用 SvcInit 函数来执行特定于服务的初始化,并开始服务要执行的工作。
示例初始化函数 SvcInit 是一个非常简单的示例;它不会执行更复杂的初始化任务,例如创建其他线程。 它创建一个事件,服务控制处理程序可以发出信号来指示服务应停止,然后调用 ReportSvcStatus 以指示服务已进入SERVICE_RUNNING状态。 此时,服务已完成其初始化,并已准备好接受控制。 为获得最佳系统性能,应用程序应在 25-100 毫秒内进入运行状态。
由于此示例服务未完成任何实际任务,SvcInit 只需通过调用 WaitForSingleObject 函数来等待服务停止事件发出信号,调用 ReportSvcStatus 以指示服务已进入SERVICE_STOPPED状态,并返回 。 (请注意,函数返回而不是调用 ExitThread 函数非常重要,因为返回 允许清理为 arguments.) 可以使用 RegisterWaitForSingleObject 函数而不是 WaitForSingleObject 函数来执行其他清理任务。 运行 ServiceMain 函数的线程终止,但服务本身继续运行。 当服务控制处理程序发出事件信号时,线程池中的线程将执行回调以执行其他清理,包括将状态设置为SERVICE_STOPPED。
请注意,此示例使用 SvcReportEvent 将错误事件写入事件日志。 有关 SvcReportEvent 的源代码,请参阅 Svc.cpp。 有关控件处理程序函数的示例,请参阅 编写控件处理程序函数。
此示例使用以下全局定义。
#define SVCNAME TEXT("SvcName")
SERVICE_STATUS gSvcStatus;
SERVICE_STATUS_HANDLE gSvcStatusHandle;
HANDLE ghSvcStopEvent = NULL;
以下示例片段取自完整的服务示例。
//
// Purpose:
// Entry point for the service
//
// Parameters:
// dwArgc - Number of arguments in the lpszArgv array
// lpszArgv - Array of strings. The first string is the name of
// the service and subsequent strings are passed by the process
// that called the StartService function to start the service.
//
// Return value:
// None.
//
VOID WINAPI SvcMain( DWORD dwArgc, LPTSTR *lpszArgv )
{
// Register the handler function for the service
gSvcStatusHandle = RegisterServiceCtrlHandler(
SVCNAME,
SvcCtrlHandler);
if( !gSvcStatusHandle )
{
SvcReportEvent(TEXT("RegisterServiceCtrlHandler"));
return;
}
// These SERVICE_STATUS members remain as set here
gSvcStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
gSvcStatus.dwServiceSpecificExitCode = 0;
// Report initial status to the SCM
ReportSvcStatus( SERVICE_START_PENDING, NO_ERROR, 3000 );
// Perform service-specific initialization and work.
SvcInit( dwArgc, lpszArgv );
}
//
// Purpose:
// The service code
//
// Parameters:
// dwArgc - Number of arguments in the lpszArgv array
// lpszArgv - Array of strings. The first string is the name of
// the service and subsequent strings are passed by the process
// that called the StartService function to start the service.
//
// Return value:
// None
//
VOID SvcInit( DWORD dwArgc, LPTSTR *lpszArgv)
{
// TO_DO: Declare and set any required variables.
// Be sure to periodically call ReportSvcStatus() with
// SERVICE_START_PENDING. If initialization fails, call
// ReportSvcStatus with SERVICE_STOPPED.
// Create an event. The control handler function, SvcCtrlHandler,
// signals this event when it receives the stop control code.
ghSvcStopEvent = CreateEvent(
NULL, // default security attributes
TRUE, // manual reset event
FALSE, // not signaled
NULL); // no name
if ( ghSvcStopEvent == NULL)
{
ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 );
return;
}
// Report running status when initialization is complete.
ReportSvcStatus( SERVICE_RUNNING, NO_ERROR, 0 );
// TO_DO: Perform work until service stops.
while(1)
{
// Check whether to stop the service.
WaitForSingleObject(ghSvcStopEvent, INFINITE);
ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 );
return;
}
}
//
// Purpose:
// Sets the current service status and reports it to the SCM.
//
// Parameters:
// dwCurrentState - The current state (see SERVICE_STATUS)
// dwWin32ExitCode - The system error code
// dwWaitHint - Estimated time for pending operation,
// in milliseconds
//
// Return value:
// None
//
VOID ReportSvcStatus( DWORD dwCurrentState,
DWORD dwWin32ExitCode,
DWORD dwWaitHint)
{
static DWORD dwCheckPoint = 1;
// Fill in the SERVICE_STATUS structure.
gSvcStatus.dwCurrentState = dwCurrentState;
gSvcStatus.dwWin32ExitCode = dwWin32ExitCode;
gSvcStatus.dwWaitHint = dwWaitHint;
if (dwCurrentState == SERVICE_START_PENDING)
gSvcStatus.dwControlsAccepted = 0;
else gSvcStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
if ( (dwCurrentState == SERVICE_RUNNING) ||
(dwCurrentState == SERVICE_STOPPED) )
gSvcStatus.dwCheckPoint = 0;
else gSvcStatus.dwCheckPoint = dwCheckPoint++;
// Report the status of the service to the SCM.
SetServiceStatus( gSvcStatusHandle, &gSvcStatus );
}
相关主题