00001
00002
00003
00004
00005
00006
00007
00008
00009 #ifndef zz_nt_service_h
00010 #define zz_nt_service_h
00011
00012 #include <winsock2.h>
00013 #include <zz_string.h>
00014 #include <zz_log.h>
00015 #include <zz_win_exception.h>
00016 #include <windows.h>
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034 class ZZ_API CzzNTService
00035 {
00036 public:
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089 CzzNTService( const char* appname,
00090 const char* svc,
00091 const char* args,
00092 const char* display,
00093 const char* dep,
00094 const char* descr,
00095 DWORD starttype );
00096
00097 CzzNTService( const char* appname,
00098 const char* svc,
00099 const char* args,
00100 const char* display,
00101 const char* dep,
00102 const char* descr
00103 );
00104
00105 CzzNTService( const char* appname,
00106 const char* svc,
00107 const char* args,
00108 const char* display,
00109 const char* dep
00110 );
00111
00112 virtual ~CzzNTService();
00113
00114
00115
00116
00117
00118
00119
00120
00121
00122
00123
00124 virtual void Install( void );
00125
00126
00127
00128
00129
00130
00131
00132
00133
00134
00135
00136 virtual void Remove( void );
00137
00138
00139
00140
00141
00142
00143
00144
00145
00146
00147
00148 virtual void Debug( int argc,
00149 char** argv
00150 );
00151
00152
00153
00154
00155
00156
00157
00158
00159
00160
00161
00162 virtual void Start( void );
00163
00164
00165 CzzString& GetServiceName( void ) { return m_ServiceName; };
00166
00167
00168 CzzString& GetAppName( void ) { return m_AppName; };
00169
00170
00171 SERVICE_STATUS& GetStatus( void ) { return m_Status; };
00172
00173
00174 CzzLog& GetLog( void ) { return m_Log; };
00175
00176
00177
00178
00179
00180
00181
00182
00183 virtual void OnStart( int argc,
00184 char** argv
00185 );
00186
00187
00188
00189
00190
00191
00192
00193
00194 virtual void OnStop( void );
00195
00196
00197 void ReportStatus( DWORD state, DWORD exitcode, DWORD waithint );
00198
00199
00200 void Register( void );
00201
00202 protected:
00203
00204 void Init( void );
00205 void Exit( void );
00206
00207 SERVICE_STATUS m_Status;
00208 SERVICE_STATUS_HANDLE m_Handle;
00209 CzzLog m_Log;
00210 CzzString m_AppName;
00211 CzzString m_ServiceName;
00212 CzzString m_Args;
00213 CzzString m_DisplayName;
00214 CzzString m_Description;
00215 CzzString m_Dependencies;
00216 bool m_Debug;
00217 DWORD m_Error;
00218 DWORD m_CheckPoint;
00219 DWORD m_StartType;
00220 HANDLE m_StopEvent;
00221
00222 private:
00223
00224 static CzzNTService* m_pThis;
00225
00226 static BOOL WINAPI ConsoleHandler( DWORD type );
00227 static VOID WINAPI ControlHandler( DWORD code );
00228 static void WINAPI Entrypoint( DWORD argc, LPSTR* argv );
00229 };
00230
00231 #endif // zz_nt_service_h
00232