| 1 |
// ntservice.h
|
| 2 |
//
|
| 3 |
// Definitions for CNTService
|
| 4 |
//
|
| 5 |
#pragma once
|
| 6 |
|
| 7 |
#ifndef _NTSERVICE_H_
|
| 8 |
#define _NTSERVICE_H_
|
| 9 |
|
| 10 |
#include "Winsvc.h"
|
| 11 |
#include ".\ntservmsg.h" // Event message ids
|
| 12 |
|
| 13 |
#define SERVICE_CONTROL_USER 128
|
| 14 |
|
| 15 |
class CNTService
|
| 16 |
{
|
| 17 |
public:
|
| 18 |
CNTService(const char* szServiceName);
|
| 19 |
virtual ~CNTService();
|
| 20 |
BOOL ParseStandardArgs(int argc, char* argv[]);
|
| 21 |
BOOL IsInstalled();
|
| 22 |
BOOL Install();
|
| 23 |
BOOL Uninstall();
|
| 24 |
void LogEvent(WORD wType, DWORD dwID,
|
| 25 |
const char* pszS1 = NULL,
|
| 26 |
const char* pszS2 = NULL,
|
| 27 |
const char* pszS3 = NULL);
|
| 28 |
BOOL StartService();
|
| 29 |
void SetStatus(DWORD dwState);
|
| 30 |
BOOL Initialize();
|
| 31 |
virtual void Run();
|
| 32 |
virtual BOOL OnInit();
|
| 33 |
virtual void OnStop();
|
| 34 |
virtual void OnInterrogate();
|
| 35 |
virtual void OnPause();
|
| 36 |
virtual void OnContinue();
|
| 37 |
virtual void OnShutdown();
|
| 38 |
virtual BOOL OnUserControl(DWORD dwOpcode);
|
| 39 |
void DebugMsg(const char* pszFormat, ...);
|
| 40 |
|
| 41 |
// static member functions
|
| 42 |
static void WINAPI ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv);
|
| 43 |
static void WINAPI Handler(DWORD dwOpcode);
|
| 44 |
|
| 45 |
// data members
|
| 46 |
char m_szServiceName[64];
|
| 47 |
int m_iMajorVersion;
|
| 48 |
int m_iMinorVersion;
|
| 49 |
SERVICE_STATUS_HANDLE m_hServiceStatus;
|
| 50 |
SERVICE_STATUS m_Status;
|
| 51 |
BOOL m_bIsRunning;
|
| 52 |
|
| 53 |
// static data
|
| 54 |
static CNTService* m_pThis; // nasty hack to get object ptr
|
| 55 |
|
| 56 |
private:
|
| 57 |
HANDLE m_hEventSource;
|
| 58 |
|
| 59 |
};
|
| 60 |
|
| 61 |
#endif // _NTSERVICE_H_
|
Parent Directory
| 
Revision Log