1 karl  1.6 //%2005////////////////////////////////////////////////////////////////////////

  2 tony  1.1 //

  3 karl  1.5 // Copyright (c) 2000, 2001, 2002 BMC Software; Hewlett-Packard Development
  4           // Company, L.P.; IBM Corp.; The Open Group; Tivoli Systems.
  5           // Copyright (c) 2003 BMC Software; Hewlett-Packard Development Company, L.P.;

  6 karl  1.4 // IBM Corp.; EMC Corporation, The Open Group.

  7 karl  1.5 // Copyright (c) 2004 BMC Software; Hewlett-Packard Development Company, L.P.;
  8           // IBM Corp.; EMC Corporation; VERITAS Software Corporation; The Open Group.

  9 karl  1.6 // Copyright (c) 2005 Hewlett-Packard Development Company, L.P.; IBM Corp.;
 10           // EMC Corporation; VERITAS Software Corporation; The Open Group.

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

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

109 tony    1.1 
110               static ReturnCode Run(SERVICE_MAIN_T service_main, DWORD flags = 0);
111               ReturnCode GetState(State *state);
112               static bool LogEvent(WORD event_type, DWORD event_id, const char *string);

113 tony    1.2   static void SetServiceName(char *service_name)
114                 {
115                   g_service_name = service_name;
116                 }

117 a.dunfey 1.7 
118                static void SetServiceArgs(int num_args, char **service_args)
119                  {
120                    g_argv = service_args;
121                    g_argc = num_args;
122                  }

123 tony     1.2   static char *GetServiceName(void)
124                  {
125                    return g_service_name;
126                  }

127 tony     1.1 
128              private:
129                static int                    g_argc;
130                static char                 **g_argv;
131                static char                  *g_service_name;
132                static char                  *g_event_source;
133                static DWORD                  g_flags;
134                static DWORD                  g_current_state;
135                static SERVICE_STATUS_HANDLE  g_service_status_handle;
136                static SERVICE_MAIN_T         g_service_main;
137              
138                static bool show_error(const char *action, const char *object, DWORD hr);
139                static void WINAPI service_control_handler(DWORD control);
140                static bool check_args_for_string(char *string);
141                static void __stdcall real_service_main(DWORD argc, LPTSTR *argv);
142                static void change_service_description(SC_HANDLE service, char *description);
143                State get_state(DWORD scm_state);
144                static ReturnCode get_error(DWORD error_status, const char action[] = "cimserver");
145              };
146              
147              #endif // SERVICE_H
148 tony     1.1