123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- #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;
- __int64 affinity;
- unsigned long envlen;
- TCHAR *env_extra;
- unsigned long env_extralen;
- unsigned long priority;
- 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 *);
- int affinity_mask_to_string(__int64, TCHAR **);
- int affinity_string_to_mask(TCHAR *, __int64 *);
- unsigned long priority_mask();
- int priority_constant_to_index(unsigned long);
- unsigned long priority_index_to_constant(int);
- 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
|