// static member function (callback) to handle commands from the
// service control manager
void WINAPI CNTService::Handler(DWORD dwOpcode)
{
// Get a pointer to the object
CNTService* pService = m_pThis;
pService->DebugMsg("CNTService::Handler(%lu)", dwOpcode);
switch (dwOpcode) {
case SERVICE_CONTROL_STOP: // 1
pService->SetStatus(SERVICE_STOP_PENDING);
pService->OnStop();
pService->m_bIsRunning = FALSE;
pService->LogEvent(EVENTLOG_INFORMATION_TYPE, EVMSG_STOPPED);
pService->SetStatus(SERVICE_STOPPED);
break;
case SERVICE_CONTROL_PAUSE: // 2
pService->OnSignal();
pService->OnPause();
break;
case SERVICE_CONTROL_CONTINUE: // 3
pService->OnSignal();
pService->OnContinue();
break;
case SERVICE_CONTROL_INTERROGATE: // 4
pService->OnInterrogate();
pService->OnSignal();
break;
case SERVICE_CONTROL_SHUTDOWN: // 5
pService->DebugMsg("CNTService::Handler - Calling Shutdown");
pService->SetStatus(SERVICE_STOP_PENDING);
pService->OnShutdown();
pService->m_bIsRunning = FALSE;
pService->LogEvent(EVENTLOG_INFORMATION_TYPE, EVMSG_STOPPED);
pService->SetStatus(SERVICE_STOPPED);
pService->m_bIsRunning = FALSE;
break;
default:
pService->OnSignal();
if (dwOpcode >= SERVICE_CONTROL_USER) {
if (!pService->OnUserControl(dwOpcode)) {
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_BADREQUEST);
}
} else {
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_BADREQUEST);
}
break;
}
// Report current status
pService->DebugMsg("Updating status (%lu, %lu)",
pService->m_hServiceStatus,
pService->m_Status.dwCurrentState);
::SetServiceStatus(pService->m_hServiceStatus, &pService->m_Status);
}
void CNTService::ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv)
{
// Get a pointer to the C++ object
CNTService* pService = m_pThis;
pService->DebugMsg("Entering CNTService::ServiceMain()");
// Register the control request handler
pService->m_Status.dwCurrentState = SERVICE_START_PENDING;
pService->m_hServiceStatus = RegisterServiceCtrlHandlerA(pService->m_szServiceName,
Handler);
if (pService->m_hServiceStatus == NULL) {
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_CTRLHANDLERNOTINSTALLED);
return;
}
// Start the initialisation
if (pService->Initialize()) {
// Do the real work.
// When the Run function returns, the service has stopped.
pService->m_bIsRunning = TRUE;
pService->m_Status.dwWin32ExitCode = 0;
pService->m_Status.dwCheckPoint = 0;
pService->m_Status.dwWaitHint = 0;
pService->Run();
}
// Tell the service manager we are stopped
pService->SetStatus(SERVICE_STOPPED);
pService->DebugMsg("Leaving CNTService::ServiceMain()");
}
void CNTService::ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv)
{
// Get a pointer to the C++ object
CNTService* pService = m_pThis;
pService->DebugMsg(_("Entering CNTService::ServiceMain()"));
// Register the control request handler
pService->m_Status.dwCurrentState = SERVICE_START_PENDING;
pService->m_hServiceStatus = RegisterServiceCtrlHandler(pService->m_szServiceName,
Handler);
if ( pService->m_hServiceStatus == NULL ) {
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_CTRLHANDLERNOTINSTALLED);
return;
}
// Start the initialisation
if ( pService->Initialize() ) {
// Do the real work.
// When the Run function returns, the service has stopped.
pService->m_bIsRunning = true;
pService->m_Status.dwWin32ExitCode = 0;
pService->m_Status.dwCheckPoint = 0;
pService->m_Status.dwWaitHint = 0;
pService->Run();
}
// Now wait for threads to exit.
DWORD dwWaitRes;
if ( ( dwWaitRes = WaitForMultipleObjects( MAX_THREADS, ghThreads, TRUE, 3000 ) )
== WAIT_OBJECT_0 ) {
// This is OK! Nothing more to do.....
;
}
else if ( ( dwWaitRes == WAIT_FAILED ) || ( dwWaitRes == WAIT_ABANDONED ) ) {
// We failed to kill the threads we have to
// abort anyway but we tell the world that we do so....
pService->DebugMsg(_("Failed to terminate all threads in CNTService::ServiceMain()"));
}
// close the event handle and the thread handle
CloseHandle( ghStopEvent );
for (int i=0; i<MAX_THREADS; i++ ) {
if ( ghThreads[i] ) {
CloseHandle( ghThreads[i] );
}
}
// Tell the service manager we are stopped
pService->SetStatus( SERVICE_STOPPED );
pService->DebugMsg(_("Leaving CNTService::ServiceMain()"));
}
// static member function (callback)
void CNTService::ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv)
{
#ifdef _DEBUG
cout << "in NTService ServiceMain" << endl;
#endif
// Get a pointer to the C++ object
CNTService* pService = m_pThis;
// Register the control request handler
pService->m_Status.dwCurrentState = SERVICE_START_PENDING;
pService->m_hServiceStatus = RegisterServiceCtrlHandler(pService->m_szServiceName,
Handler);
if (pService->m_hServiceStatus == NULL) {
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_CTRLHANDLERNOTINSTALLED);
#ifdef _DEBUG
cout << "NTService:could not register control request handler" << endl;
#endif
return;
}
// Start the initialisation
if (pService->Initialize()) {
#ifdef _DEBUG
cout << "NTService: Initialization was successful - ready to go" << endl;
#endif
// Do the real work.
// When the Run function returns, the service has stopped.
pService->m_bIsRunning = TRUE;
pService->m_Status.dwWin32ExitCode = 0;
pService->m_Status.dwCheckPoint = 0;
pService->m_Status.dwWaitHint = 0;
pService->Run();
}
#ifdef _DEBUG
cout << "NTService: Initialize failed!" << endl;
#endif
// Tell the service manager we are stopped
pService->SetStatus(SERVICE_STOPPED);
}
// static member function (callback) to handle commands from the
// service control manager
void CNTService::Handler(DWORD dwOpcode)
{
#ifdef _DEBUG
cout << "NTService: handling service control manager command:";
#endif
// Get a pointer to the object
CNTService* pService = m_pThis;
switch (dwOpcode) {
case SERVICE_CONTROL_STOP: // 1
#ifdef _DEBUG
cout << "stop" << endl;
#endif
pService->SetStatus(SERVICE_STOP_PENDING);
pService->OnStop();
pService->m_bIsRunning = FALSE;
pService->LogEvent(EVENTLOG_INFORMATION_TYPE, EVMSG_STOPPED);
break;
case SERVICE_CONTROL_PAUSE: // 2
#ifdef _DEBUG
cout << "pause" << endl;
#endif
pService->OnPause();
break;
case SERVICE_CONTROL_CONTINUE: // 3
#ifdef _DEBUG
cout << "continue" << endl;
#endif
pService->OnContinue();
break;
case SERVICE_CONTROL_INTERROGATE: // 4
#ifdef _DEBUG
cout << "interrogate" << endl;
#endif
pService->OnInterrogate();
break;
case SERVICE_CONTROL_SHUTDOWN: // 5
#ifdef _DEBUG
cout << "shutdown" << endl;
#endif
pService->OnShutdown();
break;
default:
if (dwOpcode >= SERVICE_CONTROL_USER) {
#ifdef _DEBUG
cout << "user control" << endl;
#endif
if (!pService->OnUserControl(dwOpcode)) {
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_BADREQUEST);
}
} else {
#ifdef _DEBUG
cout << "unknown" << endl;
#endif
pService->LogEvent(EVENTLOG_ERROR_TYPE, EVMSG_BADREQUEST);
}
break;
}
// Report current status
::SetServiceStatus(pService->m_hServiceStatus, &pService->m_Status);
}