service.h

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