/////////////////////////////////////////////////////////////////////////////// // Copyright (c) Winnov L.P., 1996. All rights reserved. // service.cpp: Implementation of CService class /////////////////////////////////////////////////////////////////////////////// // This class is a generic NT service wrapper. #include #include #include "wnverr.h" #include "debug.h" #include "wnv1632.h" #include "wnvevent.h" #include "service.h" #include "irq.h" #include "resource.h" #include "wnvhad.h" #include "log.h" #define TIMEOUT_TIME 3000 /* 3 seconds */ /////////////////////////////////////////////////////////////////////////////// // Method: CService // // Purpose: Constructor // // Parameter descriptions: // // pReport: Callback for reporting status to the service manager. // // return: none // /////////////////////////////////////////////////////////////////////////////// CService::CService (BOOL (* pReport)(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint)) { UINT nBoard; for (nBoard = 0; (nBoard < NBOARDS); nBoard++) { pIrq [nBoard] = NULL; // CIrq class is not allocated yet } pReportStatusToSCMgr = pReport; hEventStop = NULL; // stop the service hProcess = NULL; m_dwBoards = 0; } /////////////////////////////////////////////////////////////////////////////// // Method: ~CService // // Purpose: Destructor // // Parameter descriptions: None // // return: none // /////////////////////////////////////////////////////////////////////////////// CService::~CService () { } /////////////////////////////////////////////////////////////////////////////// // Method: Stop // // Purpose: Service is stopping, inform the driver. // // Parameter descriptions: None // // return: none // /////////////////////////////////////////////////////////////////////////////// VOID CService::Stop (void) { TraceString ("\nServiceStop: Stop requested"); // tell sc mgr that stop is pending pReportStatusToSCMgr (SERVICE_STOP_PENDING, 0, TIMEOUT_TIME); // Signal the main thread to exit SetEvent (hEventStop); TraceString ("\nCService::Stop: Stop event signalled."); } /////////////////////////////////////////////////////////////////////////////// void CService::Close (DWORD dwResult) { UINT nBoard; TraceString ("\nCService::Close: dwResult="); TraceLong (dwResult); // begin stopping pReportStatusToSCMgr( SERVICE_STOP_PENDING, // service state dwResult, // exit code TIMEOUT_TIME); // wait hint if (m_dwBoards) for (nBoard = 0; (nBoard < m_dwBoards); nBoard++) { if (pIrq [nBoard]) { pIrq [nBoard]->Close (); delete pIrq [nBoard]; pIrq [nBoard] = NULL; } } if (hEventStop) { CloseHandle (hEventStop); hEventStop = NULL; } if (hProcess) { CloseHandle (hProcess); hProcess = NULL; } // tell sc mgr that stop has been done pReportStatusToSCMgr( SERVICE_STOPPED, // service state dwResult, // exit code TIMEOUT_TIME); // wait hint } /////////////////////////////////////////////////////////////////////////////// // Method: Start // // Purpose: Start the service. // // Parameter descriptions: Same as "main" // // return: 0 = pass, else error code // /////////////////////////////////////////////////////////////////////////////// DWORD CService::Start (int nArgc, LPTSTR *lpszArgv) { DWORD dwResult; UINT nBoard; HANDLE hEvent [3]; WNVSTATUS status = STATUS_SUCCESS; DWORD dwBoards = 0; TraceString ("\nServiceStart: Begin"); // report the status to the service control manager. // if (!pReportStatusToSCMgr( SERVICE_START_PENDING, // service state NO_ERROR, // exit code 10000)) // wait hint { LogEvent (WNVERR_WNVIRQ_REPORT_FAIL, L"\n********** CService::Start: ReportStatusToSCMgr failed 1."); return WNVERR_WNVIRQ_REPORT_FAIL; } // create the STOP event (manual reset) hEventStop = CreateEvent (NULL, TRUE, FALSE, NULL); if (!hEventStop) { DWORD dw = GetLastError (); Close (GetLastError ()); LogEvent (dw, L"\n********** CService::Start: CreateEvent failed."); return dw; } dwResult = HadGetNumDevs (&m_dwBoards); if (dwResult) { LogEvent (dwResult, L"\n********** CService::Start: HadGetNumBoards failed, dwResult="); TraceLong (dwResult); Close (dwResult); return dwResult; } if (!m_dwBoards) { dwResult = 1; LogEvent (dwResult, L"\n********** CService::Start: No Boards."); TraceLong (dwResult); Close (dwResult); return dwResult; } // up the priority of this process if (!SetPriorityClass (GetCurrentProcess (), REALTIME_PRIORITY_CLASS)) { // TraceString ("\n********** CService::Start: SetPriorityClass failed (ignored)"); } for (nBoard = 0; (nBoard < m_dwBoards); nBoard++) { pIrq [nBoard] = new CIrq; // instantiate the Irq dwResult = pIrq[nBoard]->Open (nBoard, hEventStop); if (dwResult) { // Begin failed, stop service LogEvent (dwResult, L"\n********** CService::Start: CIrq::Open failed."); delete pIrq [nBoard]; pIrq [nBoard] = NULL; Close (dwResult); } } hProcess = OpenProcess (SYNCHRONIZE, FALSE, GetCurrentProcessId ()); if (!hProcess) { dwResult = GetLastError (); LogEvent (dwResult, L"\n********** CService::Start: OpenProcess failed."); Close (dwResult); return dwResult; } // report the status to the service control manager. // if (!pReportStatusToSCMgr( SERVICE_RUNNING, // service state NO_ERROR, // exit code 0)) // wait hint { LogEvent (WNVERR_WNVIRQ_REPORT_FAIL, L"\n********** CService::Start: ReportStatusToSCMgr failed 2"); Close (WNVERR_WNVIRQ_REPORT_FAIL); return WNVERR_WNVIRQ_REPORT_FAIL; } hEvent [0] = hEventStop; hEvent [1] = hProcess; //hEvent [2] = pIrq [0]->GetThread (); dwResult = WaitForMultipleObjects (2, hEvent, FALSE, INFINITE); Close (dwResult); TraceString ("\nCService::Start: Normal termination, status="); TraceLong (dwResult); return dwResult; }