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   unsigned long envlen;\r
  48   TCHAR *env_extra;\r
  49   unsigned long env_extralen;\r
  50   TCHAR stdin_path[MAX_PATH];\r
  51   unsigned long stdin_sharing;\r
  52   unsigned long stdin_disposition;\r
  53   unsigned long stdin_flags;\r
  54   TCHAR stdout_path[MAX_PATH];\r
  55   unsigned long stdout_sharing;\r
  56   unsigned long stdout_disposition;\r
  57   unsigned long stdout_flags;\r
  58   TCHAR stderr_path[MAX_PATH];\r
  59   unsigned long stderr_sharing;\r
  60   unsigned long stderr_disposition;\r
  61   unsigned long stderr_flags;\r
  62   bool rotate_files;\r
  63   unsigned long rotate_seconds;\r
  64   unsigned long rotate_bytes_low;\r
  65   unsigned long rotate_bytes_high;\r
  66   unsigned long default_exit_action;\r
  67   unsigned long throttle_delay;\r
  68   unsigned long stop_method;\r
  69   unsigned long kill_console_delay;\r
  70   unsigned long kill_window_delay;\r
  71   unsigned long kill_threads_delay;\r
  72   SC_HANDLE handle;\r
  73   SERVICE_STATUS status;\r
  74   SERVICE_STATUS_HANDLE status_handle;\r
  75   HANDLE process_handle;\r
  76   unsigned long pid;\r
  77   HANDLE wait_handle;\r
  78   bool stopping;\r
  79   bool allow_restart;\r
  80   unsigned long throttle;\r
  81   CRITICAL_SECTION throttle_section;\r
  82   bool throttle_section_initialised;\r
  83   CONDITION_VARIABLE throttle_condition;\r
  84   HANDLE throttle_timer;\r
  85   LARGE_INTEGER throttle_duetime;\r
  86   FILETIME creation_time;\r
  87   FILETIME exit_time;\r
  88 } nssm_service_t;\r
  89 \r
  90 void WINAPI service_main(unsigned long, TCHAR **);\r
  91 TCHAR *service_control_text(unsigned long);\r
  92 void log_service_control(TCHAR *, unsigned long, bool);\r
  93 unsigned long WINAPI service_control_handler(unsigned long, unsigned long, void *, void *);\r
  94 \r
  95 nssm_service_t *alloc_nssm_service();\r
  96 void set_nssm_service_defaults(nssm_service_t *);\r
  97 void cleanup_nssm_service(nssm_service_t *);\r
  98 SC_HANDLE open_service_manager();\r
  99 QUERY_SERVICE_CONFIG *query_service_config(const TCHAR *, SC_HANDLE);\r
 100 int set_service_description(const TCHAR *, SC_HANDLE, TCHAR *);\r
 101 int get_service_description(const TCHAR *, SC_HANDLE, unsigned long, TCHAR *);\r
 102 int get_service_startup(const TCHAR *, SC_HANDLE, const QUERY_SERVICE_CONFIG *, unsigned long *);\r
 103 int get_service_username(const TCHAR *, const QUERY_SERVICE_CONFIG *, TCHAR **, size_t *);\r
 104 int grant_logon_as_service(const TCHAR *);\r
 105 int pre_install_service(int, TCHAR **);\r
 106 int pre_remove_service(int, TCHAR **);\r
 107 int pre_edit_service(int, TCHAR **);\r
 108 int install_service(nssm_service_t *);\r
 109 int remove_service(nssm_service_t *);\r
 110 int edit_service(nssm_service_t *, bool);\r
 111 int control_service(unsigned long, int, TCHAR **);\r
 112 void set_service_recovery(nssm_service_t *);\r
 113 int monitor_service(nssm_service_t *);\r
 114 int start_service(nssm_service_t *);\r
 115 int stop_service(nssm_service_t *, unsigned long, bool, bool);\r
 116 void CALLBACK end_service(void *, unsigned char);\r
 117 void throttle_restart(nssm_service_t *);\r
 118 int await_shutdown(nssm_service_t *, TCHAR *, unsigned long);\r
 119 \r
 120 #endif\r