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 registry key is limited to 255 characters.\r
  10   A registry value is limited to 16383 characters.\r
  11   Therefore we limit the service name to accommodate the path under HKLM.\r
  12 */\r
  13 #define EXE_LENGTH MAX_PATH\r
  14 #define CMD_LENGTH 32768\r
  15 #define KEY_LENGTH 255\r
  16 #define VALUE_LENGTH 16383\r
  17 #define SERVICE_NAME_LENGTH KEY_LENGTH - 55\r
  18 #define SERVICE_DISPLAYNAME_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 typedef struct {\r
  32   bool native;\r
  33   TCHAR name[SERVICE_NAME_LENGTH];\r
  34   TCHAR displayname[SERVICE_DISPLAYNAME_LENGTH];\r
  35   TCHAR description[VALUE_LENGTH];\r
  36   unsigned long startup;\r
  37   TCHAR *username;\r
  38   size_t usernamelen;\r
  39   TCHAR *password;\r
  40   size_t passwordlen;\r
  41   unsigned long type;\r
  42   TCHAR image[MAX_PATH];\r
  43   TCHAR exe[EXE_LENGTH];\r
  44   TCHAR flags[VALUE_LENGTH];\r
  45   TCHAR dir[MAX_PATH];\r
  46   TCHAR *env;\r
  47   __int64 affinity;\r
  48   unsigned long envlen;\r
  49   TCHAR *env_extra;\r
  50   unsigned long env_extralen;\r
  51   unsigned long priority;\r
  52   TCHAR stdin_path[MAX_PATH];\r
  53   unsigned long stdin_sharing;\r
  54   unsigned long stdin_disposition;\r
  55   unsigned long stdin_flags;\r
  56   TCHAR stdout_path[MAX_PATH];\r
  57   unsigned long stdout_sharing;\r
  58   unsigned long stdout_disposition;\r
  59   unsigned long stdout_flags;\r
  60   TCHAR stderr_path[MAX_PATH];\r
  61   unsigned long stderr_sharing;\r
  62   unsigned long stderr_disposition;\r
  63   unsigned long stderr_flags;\r
  64   bool rotate_files;\r
  65   unsigned long rotate_seconds;\r
  66   unsigned long rotate_bytes_low;\r
  67   unsigned long rotate_bytes_high;\r
  68   unsigned long default_exit_action;\r
  69   unsigned long throttle_delay;\r
  70   unsigned long stop_method;\r
  71   unsigned long kill_console_delay;\r
  72   unsigned long kill_window_delay;\r
  73   unsigned long kill_threads_delay;\r
  74   SC_HANDLE handle;\r
  75   SERVICE_STATUS status;\r
  76   SERVICE_STATUS_HANDLE status_handle;\r
  77   HANDLE process_handle;\r
  78   unsigned long pid;\r
  79   HANDLE wait_handle;\r
  80   bool stopping;\r
  81   bool allow_restart;\r
  82   unsigned long throttle;\r
  83   CRITICAL_SECTION throttle_section;\r
  84   bool throttle_section_initialised;\r
  85   CONDITION_VARIABLE throttle_condition;\r
  86   HANDLE throttle_timer;\r
  87   LARGE_INTEGER throttle_duetime;\r
  88   FILETIME creation_time;\r
  89   FILETIME exit_time;\r
  90 } nssm_service_t;\r
  91 \r
  92 void WINAPI service_main(unsigned long, TCHAR **);\r
  93 TCHAR *service_control_text(unsigned long);\r
  94 void log_service_control(TCHAR *, unsigned long, bool);\r
  95 unsigned long WINAPI service_control_handler(unsigned long, unsigned long, void *, void *);\r
  96 \r
  97 int affinity_mask_to_string(__int64, TCHAR **);\r
  98 int affinity_string_to_mask(TCHAR *, __int64 *);\r
  99 unsigned long priority_mask();\r
 100 int priority_constant_to_index(unsigned long);\r
 101 unsigned long priority_index_to_constant(int);\r
 102 \r
 103 nssm_service_t *alloc_nssm_service();\r
 104 void set_nssm_service_defaults(nssm_service_t *);\r
 105 void cleanup_nssm_service(nssm_service_t *);\r
 106 SC_HANDLE open_service_manager();\r
 107 SC_HANDLE open_service(SC_HANDLE, TCHAR *, TCHAR *, unsigned long);\r
 108 QUERY_SERVICE_CONFIG *query_service_config(const TCHAR *, SC_HANDLE);\r
 109 int set_service_description(const TCHAR *, SC_HANDLE, TCHAR *);\r
 110 int get_service_description(const TCHAR *, SC_HANDLE, unsigned long, TCHAR *);\r
 111 int get_service_startup(const TCHAR *, SC_HANDLE, const QUERY_SERVICE_CONFIG *, unsigned long *);\r
 112 int get_service_username(const TCHAR *, const QUERY_SERVICE_CONFIG *, TCHAR **, size_t *);\r
 113 int grant_logon_as_service(const TCHAR *);\r
 114 int pre_install_service(int, TCHAR **);\r
 115 int pre_remove_service(int, TCHAR **);\r
 116 int pre_edit_service(int, TCHAR **);\r
 117 int install_service(nssm_service_t *);\r
 118 int remove_service(nssm_service_t *);\r
 119 int edit_service(nssm_service_t *, bool);\r
 120 int control_service(unsigned long, int, TCHAR **);\r
 121 void set_service_recovery(nssm_service_t *);\r
 122 int monitor_service(nssm_service_t *);\r
 123 int start_service(nssm_service_t *);\r
 124 int stop_service(nssm_service_t *, unsigned long, bool, bool);\r
 125 void CALLBACK end_service(void *, unsigned char);\r
 126 void throttle_restart(nssm_service_t *);\r
 127 int await_shutdown(nssm_service_t *, TCHAR *, unsigned long);\r
 128 \r
 129 #endif\r