|
|
|
#ifndef SERVICE_H
|
|
|
|
#define SERVICE_H
|
|
|
|
|
|
|
|
#include <ntsecapi.h>
|
|
|
|
|
|
|
|
/*
|
|
|
|
MSDN says the commandline in CreateProcess() is limited to 32768 characters
|
|
|
|
and the application name to MAX_PATH.
|
|
|
|
A registry key is limited to 255 characters.
|
|
|
|
A registry value is limited to 16383 characters.
|
|
|
|
Therefore we limit the service name to accommodate the path under HKLM.
|
|
|
|
*/
|
|
|
|
#define EXE_LENGTH MAX_PATH
|
|
|
|
#define CMD_LENGTH 32768
|
|
|
|
#define KEY_LENGTH 255
|
|
|
|
#define VALUE_LENGTH 16383
|
|
|
|
#define SERVICE_NAME_LENGTH KEY_LENGTH - 55
|
|
|
|
#define SERVICE_DISPLAYNAME_LENGTH 256
|
|
|
|
|
|
|
|
#define ACTION_LEN 16
|
|
|
|
|
|
|
|
#define NSSM_LOCALSYSTEM_ACCOUNT _T("LocalSystem")
|
|
|
|
#define NSSM_KERNEL_DRIVER _T("SERVICE_KERNEL_DRIVER")
|
|
|
|
#define NSSM_FILE_SYSTEM_DRIVER _T("SERVICE_FILE_SYSTEM_DRIVER")
|
|
|
|
#define NSSM_WIN32_OWN_PROCESS _T("SERVICE_WIN32_OWN_PROCESS")
|
|
|
|
#define NSSM_WIN32_SHARE_PROCESS _T("SERVICE_WIN32_SHARE_PROCESS")
|
|
|
|
#define NSSM_INTERACTIVE_PROCESS _T("SERVICE_INTERACTIVE_PROCESS")
|
|
|
|
#define NSSM_SHARE_INTERACTIVE_PROCESS NSSM_WIN32_SHARE_PROCESS _T("|") NSSM_INTERACTIVE_PROCESS
|
|
|
|
#define NSSM_UNKNOWN _T("?")
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
bool native;
|
|
|
|
TCHAR name[SERVICE_NAME_LENGTH];
|
|
|
|
TCHAR displayname[SERVICE_DISPLAYNAME_LENGTH];
|
|
|
|
TCHAR description[VALUE_LENGTH];
|
|
|
|
unsigned long startup;
|
|
|
|
TCHAR *username;
|
|
|
|
size_t usernamelen;
|
|
|
|
TCHAR *password;
|
|
|
|
size_t passwordlen;
|
|
|
|
unsigned long type;
|
|
|
|
TCHAR image[MAX_PATH];
|
|
|
|
TCHAR exe[EXE_LENGTH];
|
|
|
|
TCHAR flags[VALUE_LENGTH];
|
|
|
|
TCHAR dir[MAX_PATH];
|
|
|
|
TCHAR *env;
|
|
|
|
unsigned long envlen;
|
|
|
|
TCHAR *env_extra;
|
|
|
|
unsigned long env_extralen;
|
|
|
|
TCHAR stdin_path[MAX_PATH];
|
|
|
|
unsigned long stdin_sharing;
|
|
|
|
unsigned long stdin_disposition;
|
|
|
|
unsigned long stdin_flags;
|
|
|
|
TCHAR stdout_path[MAX_PATH];
|
|
|
|
unsigned long stdout_sharing;
|
|
|
|
unsigned long stdout_disposition;
|
|
|
|
unsigned long stdout_flags;
|
|
|
|
TCHAR stderr_path[MAX_PATH];
|
|
|
|
unsigned long stderr_sharing;
|
|
|
|
unsigned long stderr_disposition;
|
|
|
|
unsigned long stderr_flags;
|
|
|
|
bool rotate_files;
|
|
|
|
unsigned long rotate_seconds;
|
|
|
|
unsigned long rotate_bytes_low;
|
|
|
|
unsigned long rotate_bytes_high;
|
|
|
|
unsigned long default_exit_action;
|
|
|
|
unsigned long throttle_delay;
|
|
|
|
unsigned long stop_method;
|
|
|
|
unsigned long kill_console_delay;
|
|
|
|
unsigned long kill_window_delay;
|
|
|
|
unsigned long kill_threads_delay;
|
|
|
|
SC_HANDLE handle;
|
|
|
|
SERVICE_STATUS status;
|
|
|
|
SERVICE_STATUS_HANDLE status_handle;
|
|
|
|
HANDLE process_handle;
|
|
|
|
unsigned long pid;
|
|
|
|
HANDLE wait_handle;
|
|
|
|
bool stopping;
|
|
|
|
bool allow_restart;
|
|
|
|
unsigned long throttle;
|
|
|
|
CRITICAL_SECTION throttle_section;
|
|
|
|
bool throttle_section_initialised;
|
|
|
|
CONDITION_VARIABLE throttle_condition;
|
|
|
|
HANDLE throttle_timer;
|
|
|
|
LARGE_INTEGER throttle_duetime;
|
|
|
|
FILETIME creation_time;
|
|
|
|
FILETIME exit_time;
|
|
|
|
} nssm_service_t;
|
|
|
|
|
|
|
|
void WINAPI service_main(unsigned long, TCHAR **);
|
|
|
|
TCHAR *service_control_text(unsigned long);
|
|
|
|
void log_service_control(TCHAR *, unsigned long, bool);
|
|
|
|
unsigned long WINAPI service_control_handler(unsigned long, unsigned long, void *, void *);
|
|
|
|
|
|
|
|
nssm_service_t *alloc_nssm_service();
|
|
|
|
void set_nssm_service_defaults(nssm_service_t *);
|
|
|
|
void cleanup_nssm_service(nssm_service_t *);
|
|
|
|
SC_HANDLE open_service_manager();
|
|
|
|
SC_HANDLE open_service(SC_HANDLE, TCHAR *, TCHAR *, unsigned long);
|
|
|
|
QUERY_SERVICE_CONFIG *query_service_config(const TCHAR *, SC_HANDLE);
|
|
|
|
int set_service_description(const TCHAR *, SC_HANDLE, TCHAR *);
|
|
|
|
int get_service_description(const TCHAR *, SC_HANDLE, unsigned long, TCHAR *);
|
|
|
|
int get_service_startup(const TCHAR *, SC_HANDLE, const QUERY_SERVICE_CONFIG *, unsigned long *);
|
|
|
|
int get_service_username(const TCHAR *, const QUERY_SERVICE_CONFIG *, TCHAR **, size_t *);
|
|
|
|
int grant_logon_as_service(const TCHAR *);
|
|
|
|
int pre_install_service(int, TCHAR **);
|
|
|
|
int pre_remove_service(int, TCHAR **);
|
|
|
|
int pre_edit_service(int, TCHAR **);
|
|
|
|
int install_service(nssm_service_t *);
|
|
|
|
int remove_service(nssm_service_t *);
|
|
|
|
int edit_service(nssm_service_t *, bool);
|
|
|
|
int control_service(unsigned long, int, TCHAR **);
|
|
|
|
void set_service_recovery(nssm_service_t *);
|
|
|
|
int monitor_service(nssm_service_t *);
|
|
|
|
int start_service(nssm_service_t *);
|
|
|
|
int stop_service(nssm_service_t *, unsigned long, bool, bool);
|
|
|
|
void CALLBACK end_service(void *, unsigned char);
|
|
|
|
void throttle_restart(nssm_service_t *);
|
|
|
|
int await_shutdown(nssm_service_t *, TCHAR *, unsigned long);
|
|
|
|
|
|
|
|
#endif
|