service.h

   1 #ifndef SERVICE_H\r
   2 #define SERVICE_H\r
   3 \r
   4 /*\r
   5   MSDN says the commandline in CreateProcess() is limited to 32768 characters\r
   6   and the application name to MAX_PATH.\r
   7   A service name and service display name are limited to 256 characters.\r
   8   A registry key is limited to 255 characters.\r
   9   A registry value is limited to 16383 characters.\r
  10   Therefore we limit the service name to accommodate the path under HKLM.\r
  11 */\r
  12 #define EXE_LENGTH PATH_LENGTH\r
  13 #define CMD_LENGTH 32768\r
  14 #define KEY_LENGTH 255\r
  15 #define VALUE_LENGTH 16383\r
  16 #define SERVICE_NAME_LENGTH 256\r
  17 \r
  18 #define ACTION_LEN 16\r
  19 \r
  20 #define NSSM_KERNEL_DRIVER _T("SERVICE_KERNEL_DRIVER")\r
  21 #define NSSM_FILE_SYSTEM_DRIVER _T("SERVICE_FILE_SYSTEM_DRIVER")\r
  22 #define NSSM_WIN32_OWN_PROCESS _T("SERVICE_WIN32_OWN_PROCESS")\r
  23 #define NSSM_WIN32_SHARE_PROCESS _T("SERVICE_WIN32_SHARE_PROCESS")\r
  24 #define NSSM_INTERACTIVE_PROCESS _T("SERVICE_INTERACTIVE_PROCESS")\r
  25 #define NSSM_SHARE_INTERACTIVE_PROCESS NSSM_WIN32_SHARE_PROCESS _T("|") NSSM_INTERACTIVE_PROCESS\r
  26 #define NSSM_UNKNOWN _T("?")\r
  27 \r
  28 #define NSSM_ROTATE_OFFLINE 0\r
  29 #define NSSM_ROTATE_ONLINE 1\r
  30 #define NSSM_ROTATE_ONLINE_ASAP 2\r
  31 \r
  32 typedef struct {\r
  33   bool native;\r
  34   TCHAR name[SERVICE_NAME_LENGTH];\r
  35   TCHAR displayname[SERVICE_NAME_LENGTH];\r
  36   TCHAR description[VALUE_LENGTH];\r
  37   unsigned long startup;\r
  38   TCHAR *username;\r
  39   size_t usernamelen;\r
  40   TCHAR *password;\r
  41   size_t passwordlen;\r
  42   unsigned long type;\r
  43   TCHAR image[PATH_LENGTH];\r
  44   TCHAR exe[EXE_LENGTH];\r
  45   TCHAR flags[VALUE_LENGTH];\r
  46   TCHAR dir[DIR_LENGTH];\r
  47   TCHAR *env;\r
  48   __int64 affinity;\r
  49   TCHAR *dependencies;\r
  50   unsigned long dependencieslen;\r
  51   unsigned long envlen;\r
  52   TCHAR *env_extra;\r
  53   unsigned long env_extralen;\r
  54   unsigned long priority;\r
  55   unsigned long no_console;\r
  56   TCHAR stdin_path[PATH_LENGTH];\r
  57   unsigned long stdin_sharing;\r
  58   unsigned long stdin_disposition;\r
  59   unsigned long stdin_flags;\r
  60   TCHAR stdout_path[PATH_LENGTH];\r
  61   unsigned long stdout_sharing;\r
  62   unsigned long stdout_disposition;\r
  63   unsigned long stdout_flags;\r
  64   bool use_stdout_pipe;\r
  65   HANDLE stdout_si;\r
  66   HANDLE stdout_pipe;\r
  67   HANDLE stdout_thread;\r
  68   unsigned long stdout_tid;\r
  69   TCHAR stderr_path[PATH_LENGTH];\r
  70   unsigned long stderr_sharing;\r
  71   unsigned long stderr_disposition;\r
  72   unsigned long stderr_flags;\r
  73   bool use_stderr_pipe;\r
  74   HANDLE stderr_si;\r
  75   HANDLE stderr_pipe;\r
  76   HANDLE stderr_thread;\r
  77   unsigned long stderr_tid;\r
  78   bool hook_share_output_handles;\r
  79   bool rotate_files;\r
  80   bool stdout_copy_and_truncate;\r
  81   bool stderr_copy_and_truncate;\r
  82   unsigned long rotate_stdout_online;\r
  83   unsigned long rotate_stderr_online;\r
  84   unsigned long rotate_seconds;\r
  85   unsigned long rotate_bytes_low;\r
  86   unsigned long rotate_bytes_high;\r
  87   unsigned long rotate_delay;\r
  88   unsigned long default_exit_action;\r
  89   unsigned long restart_delay;\r
  90   unsigned long throttle_delay;\r
  91   unsigned long stop_method;\r
  92   unsigned long kill_console_delay;\r
  93   unsigned long kill_window_delay;\r
  94   unsigned long kill_threads_delay;\r
  95   bool kill_process_tree;\r
  96   SC_HANDLE handle;\r
  97   SERVICE_STATUS status;\r
  98   SERVICE_STATUS_HANDLE status_handle;\r
  99   HANDLE process_handle;\r
 100   unsigned long pid;\r
 101   HANDLE wait_handle;\r
 102   unsigned long exitcode;\r
 103   bool stopping;\r
 104   bool allow_restart;\r
 105   unsigned long throttle;\r
 106   CRITICAL_SECTION throttle_section;\r
 107   bool throttle_section_initialised;\r
 108   CRITICAL_SECTION hook_section;\r
 109   bool hook_section_initialised;\r
 110   CONDITION_VARIABLE throttle_condition;\r
 111   HANDLE throttle_timer;\r
 112   LARGE_INTEGER throttle_duetime;\r
 113   FILETIME nssm_creation_time;\r
 114   FILETIME creation_time;\r
 115   FILETIME exit_time;\r
 116   TCHAR *initial_env;\r
 117   unsigned long last_control;\r
 118   unsigned long start_requested_count;\r
 119   unsigned long start_count;\r
 120   unsigned long exit_count;\r
 121 } nssm_service_t;\r
 122 \r
 123 void WINAPI service_main(unsigned long, TCHAR **);\r
 124 TCHAR *service_control_text(unsigned long);\r
 125 TCHAR *service_status_text(unsigned long);\r
 126 void log_service_control(TCHAR *, unsigned long, bool);\r
 127 unsigned long WINAPI service_control_handler(unsigned long, unsigned long, void *, void *);\r
 128 \r
 129 int affinity_mask_to_string(__int64, TCHAR **);\r
 130 int affinity_string_to_mask(TCHAR *, __int64 *);\r
 131 unsigned long priority_mask();\r
 132 int priority_constant_to_index(unsigned long);\r
 133 unsigned long priority_index_to_constant(int);\r
 134 \r
 135 nssm_service_t *alloc_nssm_service();\r
 136 void set_nssm_service_defaults(nssm_service_t *);\r
 137 void cleanup_nssm_service(nssm_service_t *);\r
 138 SC_HANDLE open_service_manager(unsigned long);\r
 139 SC_HANDLE open_service(SC_HANDLE, TCHAR *, unsigned long, TCHAR *, unsigned long);\r
 140 QUERY_SERVICE_CONFIG *query_service_config(const TCHAR *, SC_HANDLE);\r
 141 int append_to_dependencies(TCHAR *, unsigned long, TCHAR *, TCHAR **, unsigned long *, int);\r
 142 int remove_from_dependencies(TCHAR *, unsigned long, TCHAR *, TCHAR **, unsigned long *, int);\r
 143 int set_service_dependencies(const TCHAR *, SC_HANDLE, TCHAR *);\r
 144 int get_service_dependencies(const TCHAR *, SC_HANDLE, TCHAR **, unsigned long *, int);\r
 145 int get_service_dependencies(const TCHAR *, SC_HANDLE, TCHAR **, unsigned long *);\r
 146 int set_service_description(const TCHAR *, SC_HANDLE, TCHAR *);\r
 147 int get_service_description(const TCHAR *, SC_HANDLE, unsigned long, TCHAR *);\r
 148 int get_service_startup(const TCHAR *, SC_HANDLE, const QUERY_SERVICE_CONFIG *, unsigned long *);\r
 149 int get_service_username(const TCHAR *, const QUERY_SERVICE_CONFIG *, TCHAR **, size_t *);\r
 150 void set_service_environment(nssm_service_t *);\r
 151 void unset_service_environment(nssm_service_t *);\r
 152 int pre_install_service(int, TCHAR **);\r
 153 int pre_remove_service(int, TCHAR **);\r
 154 int pre_edit_service(int, TCHAR **);\r
 155 int install_service(nssm_service_t *);\r
 156 int remove_service(nssm_service_t *);\r
 157 int edit_service(nssm_service_t *, bool);\r
 158 int control_service(unsigned long, int, TCHAR **);\r
 159 void set_service_recovery(nssm_service_t *);\r
 160 int monitor_service(nssm_service_t *);\r
 161 int start_service(nssm_service_t *);\r
 162 int stop_service(nssm_service_t *, unsigned long, bool, bool);\r
 163 void CALLBACK end_service(void *, unsigned char);\r
 164 void throttle_restart(nssm_service_t *);\r
 165 int await_single_handle(SERVICE_STATUS_HANDLE, SERVICE_STATUS *, HANDLE, TCHAR *, TCHAR *, unsigned long);\r
 166 int list_nssm_services(int, TCHAR **);\r
 167 int service_process_tree(int, TCHAR **);\r
 168 \r
 169 #endif\r