1 tony  1.1 //%/////////////////////////////////////////////////////////////////////////////
  2           //
  3           // Copyright (c) 2000, 2001, 2002 BMC Software, Hewlett-Packard Company, IBM,
  4           // The Open Group, Tivoli Systems
  5           //
  6           // Permission is hereby granted, free of charge, to any person obtaining a copy
  7           // of this software and associated documentation files (the "Software"), to
  8           // deal in the Software without restriction, including without limitation the
  9           // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
 10           // sell copies of the Software, and to permit persons to whom the Software is
 11           // furnished to do so, subject to the following conditions:
 12           // 
 13           // THE ABOVE COPYRIGHT NOTICE AND THIS PERMISSION NOTICE SHALL BE INCLUDED IN
 14           // ALL COPIES OR SUBSTANTIAL PORTIONS OF THE SOFTWARE. THE SOFTWARE IS PROVIDED
 15           // "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
 16           // LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
 17           // PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 18           // HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 19           // ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 20           // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 21           //
 22 tony  1.1 //==============================================================================
 23           //
 24           // Author: Tony Fiorentino (fiorentino_tony@emc.com)
 25           //
 26           //%/////////////////////////////////////////////////////////////////////////////
 27           #ifndef SERVICE_H
 28           #define SERVICE_H
 29           
 30           typedef int (*SERVICE_MAIN_T)(int flag, int argc, char **argv);
 31           
 32           class Service
 33           {
 34           public:
 35             // default ctor
 36             Service(void);
 37           
 38             // ctor using service name
 39             Service(char *service_name);
 40           
 41             // ctor using service name and event source
 42             Service(char *service_name, char *event_source);
 43 tony  1.1 
 44             // dtor
 45             ~Service(void);
 46           
 47             // State
 48             enum State
 49               {
 50                 SERVICE_STATE_STOPPED            = SERVICE_STOPPED,
 51                 SERVICE_STATE_START_PENDING      = SERVICE_START_PENDING,
 52                 SERVICE_STATE_STOP_PENDING       = SERVICE_STOP_PENDING,
 53                 SERVICE_STATE_RUNNING            = SERVICE_RUNNING,
 54                 SERVICE_STATE_CONTINUE_PENDING   = SERVICE_CONTINUE_PENDING,
 55                 SERVICE_STATE_PAUSE_PENDING      = SERVICE_PAUSE_PENDING,
 56                 SERVICE_STATE_PAUSED             = SERVICE_PAUSED
 57               };
 58           
 59             // Return Codes
 60             enum ReturnCode
 61               {
 62                 SERVICE_RETURN_SUCCESS           = 0,
 63                 SERVICE_ERROR_MARKED_FOR_DELETE  = ERROR_SERVICE_MARKED_FOR_DELETE,
 64 tony  1.1       SERVICE_ERROR_DOES_NOT_EXIST     = ERROR_SERVICE_DOES_NOT_EXIST,
 65                 SERVICE_ERROR_REQUEST_TIMEOUT    = ERROR_SERVICE_REQUEST_TIMEOUT,
 66                 SERVICE_ERROR_NO_THREAD          = ERROR_SERVICE_NO_THREAD,
 67                 SERVICE_ERROR_DATABASE_LOCKED    = ERROR_SERVICE_DATABASE_LOCKED,
 68                 SERVICE_ERROR_ALREADY_RUNNING    = ERROR_SERVICE_ALREADY_RUNNING,
 69                 SERVICE_ERROR_DISABLED           = ERROR_SERVICE_DISABLED,
 70                 SERVICE_ERROR_CANNOT_ACCEPT_CTRL = ERROR_SERVICE_CANNOT_ACCEPT_CTRL,
 71                 SERVICE_ERROR_NOT_ACTIVE         = ERROR_SERVICE_NOT_ACTIVE,
 72                 SERVICE_ERROR_SPECIFIC_ERROR     = ERROR_SERVICE_SPECIFIC_ERROR,
 73                 SERVICE_ERROR_DEPENDENCY_FAIL    = ERROR_SERVICE_DEPENDENCY_FAIL,
 74                 SERVICE_ERROR_LOGON_FAILED       = ERROR_SERVICE_LOGON_FAILED,
 75                 SERVICE_ERROR_START_HANG         = ERROR_SERVICE_START_HANG,
 76                 SERVICE_ERROR_DOES_EXISTS        = ERROR_SERVICE_EXISTS,
 77                 SERVICE_ERROR_DEPENDENCY_DELETED = ERROR_SERVICE_DEPENDENCY_DELETED,
 78                 SERVICE_ERROR_NEVER_STARTED      = ERROR_SERVICE_NEVER_STARTED,
 79                 SERVICE_ERROR_NOT_FOUND          = ERROR_SERVICE_NOT_FOUND,
 80                 SERVICE_ERROR_CONTROLLER_CONNECT = ERROR_FAILED_SERVICE_CONTROLLER_CONNECT,
 81                 SERVICE_ERROR_INVALID_CONTROL    = ERROR_INVALID_SERVICE_CONTROL,
 82                 SERVICE_ERROR_INVALID_ACCOUNT    = ERROR_INVALID_SERVICE_ACCOUNT,
 83                 SERVICE_ERROR_INVALID_LOCK       = ERROR_INVALID_SERVICE_LOCK,
 84                 SERVICE_ERROR_DUPLICATE_NAME     = ERROR_DUPLICATE_SERVICE_NAME,
 85 tony  1.1       SERVICE_ERROR_DIFFERENT_ACCOUNT  = ERROR_DIFFERENT_SERVICE_ACCOUNT,
 86                 SERVICE_ERROR_UNKNOWN            = -999
 87               };
 88           
 89             // Flag
 90             enum Flag
 91               {
 92                 STARTUP_FLAG,
 93                 SHUTDOWN_FLAG
 94               };
 95           
 96             // Methods
 97             ReturnCode Install(char  *display_name, char  *description, char  *exe_name);
 98             ReturnCode Remove(void);
 99           
100             ReturnCode Start(int wait_time);
101             ReturnCode Stop(int wait_time);

102 s.hills 1.3   static bool report_status(DWORD current_state, DWORD exit_code, DWORD check_point, DWORD wait_hint);

103 tony    1.1 
104               static ReturnCode Run(SERVICE_MAIN_T service_main, DWORD flags = 0);
105               ReturnCode GetState(State *state);
106               static bool LogEvent(WORD event_type, DWORD event_id, const char *string);

107 tony    1.2   static void SetServiceName(char *service_name)
108                 {
109                   g_service_name = service_name;
110                 }
111               static char *GetServiceName(void)
112                 {
113                   return g_service_name;
114                 }

115 tony    1.1 
116             private:
117               static int                    g_argc;
118               static char                 **g_argv;
119               static char                  *g_service_name;
120               static char                  *g_event_source;
121               static DWORD                  g_flags;
122               static DWORD                  g_current_state;
123               static SERVICE_STATUS_HANDLE  g_service_status_handle;
124               static SERVICE_MAIN_T         g_service_main;
125             
126               static bool show_error(const char *action, const char *object, DWORD hr);
127               static void WINAPI service_control_handler(DWORD control);
128               static bool check_args_for_string(char *string);
129               static void __stdcall real_service_main(DWORD argc, LPTSTR *argv);
130               static void change_service_description(SC_HANDLE service, char *description);
131               State get_state(DWORD scm_state);
132               static ReturnCode get_error(DWORD error_status, const char action[] = "cimserver");
133             };
134             
135             #endif // SERVICE_H
136 tony    1.1