mirror of
https://gitlab.torproject.org/tpo/core/tor.git
synced 2024-11-23 20:03:31 +01:00
r14940@catbus: nickm | 2007-09-06 13:22:26 -0400
Move NT services code into its own C file. Probably will not build happily on win32 yet; more hacking needed. svn:r11388
This commit is contained in:
Nick Mathewson
parent
c5b4f779ec
commit
f17a38ab04
@ -32,6 +32,7 @@ Changes in version 0.2.0.7-alpha - 2007-??-??
|
|||||||
- Revamp file-writing logic so we don't need to have the entire contents
|
- Revamp file-writing logic so we don't need to have the entire contents
|
||||||
of a file in memory at once before we write to disk. Tor, meet stdio.
|
of a file in memory at once before we write to disk. Tor, meet stdio.
|
||||||
- Turn "descriptor store" into a full-fledged type.
|
- Turn "descriptor store" into a full-fledged type.
|
||||||
|
- Move all NT services code into a separate source file.
|
||||||
|
|
||||||
|
|
||||||
Changes in version 0.1.2.17 - 2007-08-30
|
Changes in version 0.1.2.17 - 2007-08-30
|
||||||
|
|||||||
@ -126,6 +126,7 @@ fi
|
|||||||
if test $bwin32 = true; then
|
if test $bwin32 = true; then
|
||||||
AC_DEFINE(MS_WINDOWS, 1, [Define to 1 if we are building for Windows.])
|
AC_DEFINE(MS_WINDOWS, 1, [Define to 1 if we are building for Windows.])
|
||||||
fi
|
fi
|
||||||
|
AM_CONDITIONAL(BUILD_NT_SERVICES, test x$bwin32 = xtrue)
|
||||||
|
|
||||||
dnl Enable C99 when compiling with MIPSpro
|
dnl Enable C99 when compiling with MIPSpro
|
||||||
AC_MSG_CHECKING([for MIPSpro compiler])
|
AC_MSG_CHECKING([for MIPSpro compiler])
|
||||||
|
|||||||
@ -4,11 +4,19 @@ noinst_PROGRAMS = test
|
|||||||
|
|
||||||
bin_PROGRAMS = tor
|
bin_PROGRAMS = tor
|
||||||
|
|
||||||
|
if BUILD_NT_SERVICES
|
||||||
|
PLATFORM_SOURCES=ntmain.c
|
||||||
|
else
|
||||||
|
PLATFORM_SOURCES=
|
||||||
|
endif
|
||||||
|
|
||||||
|
EXTRA_DIST=ntmain.c
|
||||||
|
|
||||||
tor_SOURCES = buffers.c circuitbuild.c circuitlist.c \
|
tor_SOURCES = buffers.c circuitbuild.c circuitlist.c \
|
||||||
circuituse.c command.c config.c \
|
circuituse.c command.c config.c \
|
||||||
connection.c connection_edge.c connection_or.c control.c \
|
connection.c connection_edge.c connection_or.c control.c \
|
||||||
cpuworker.c directory.c dirserv.c dirvote.c \
|
cpuworker.c directory.c dirserv.c dirvote.c \
|
||||||
dns.c dnsserv.c hibernate.c main.c \
|
dns.c dnsserv.c hibernate.c main.c $(PLATFORM_SOURCES) \
|
||||||
onion.c policies.c relay.c rendcommon.c rendclient.c rendmid.c \
|
onion.c policies.c relay.c rendcommon.c rendclient.c rendmid.c \
|
||||||
rendservice.c rephist.c router.c routerlist.c routerparse.c \
|
rendservice.c rephist.c router.c routerlist.c routerparse.c \
|
||||||
eventdns.c \
|
eventdns.c \
|
||||||
@ -25,7 +33,7 @@ test_SOURCES = buffers.c circuitbuild.c circuitlist.c \
|
|||||||
circuituse.c command.c config.c \
|
circuituse.c command.c config.c \
|
||||||
connection.c connection_edge.c connection_or.c control.c \
|
connection.c connection_edge.c connection_or.c control.c \
|
||||||
cpuworker.c directory.c dirserv.c dirvote.c \
|
cpuworker.c directory.c dirserv.c dirvote.c \
|
||||||
dns.c dnsserv.c hibernate.c main.c \
|
dns.c dnsserv.c hibernate.c main.c $(PLATFORM_SOURCES) \
|
||||||
onion.c policies.c relay.c rendcommon.c rendclient.c rendmid.c \
|
onion.c policies.c relay.c rendcommon.c rendclient.c rendmid.c \
|
||||||
rendservice.c rephist.c router.c routerlist.c routerparse.c \
|
rendservice.c rephist.c router.c routerlist.c routerparse.c \
|
||||||
eventdns.c \
|
eventdns.c \
|
||||||
|
|||||||
@ -4610,6 +4610,9 @@ extern const char dirserv_c_id[];
|
|||||||
extern const char dns_c_id[];
|
extern const char dns_c_id[];
|
||||||
extern const char hibernate_c_id[];
|
extern const char hibernate_c_id[];
|
||||||
extern const char main_c_id[];
|
extern const char main_c_id[];
|
||||||
|
#ifdef NT_SERVICE
|
||||||
|
extern const char ntmain_c_id[];
|
||||||
|
#endif
|
||||||
extern const char onion_c_id[];
|
extern const char onion_c_id[];
|
||||||
extern const char policies_c_id[];
|
extern const char policies_c_id[];
|
||||||
extern const char relay_c_id[];
|
extern const char relay_c_id[];
|
||||||
@ -4663,6 +4666,9 @@ print_svn_version(void)
|
|||||||
puts(dns_c_id);
|
puts(dns_c_id);
|
||||||
puts(hibernate_c_id);
|
puts(hibernate_c_id);
|
||||||
puts(main_c_id);
|
puts(main_c_id);
|
||||||
|
#ifdef NT_SERVICE
|
||||||
|
puts(ntmain_c_id);
|
||||||
|
#endif
|
||||||
puts(onion_c_id);
|
puts(onion_c_id);
|
||||||
puts(policies_c_id);
|
puts(policies_c_id);
|
||||||
puts(relay_c_id);
|
puts(relay_c_id);
|
||||||
|
|||||||
755
src/or/main.c
755
src/or/main.c
@ -86,35 +86,6 @@ static int called_loop_once = 0;
|
|||||||
* entry to inform the user that Tor is working. */
|
* entry to inform the user that Tor is working. */
|
||||||
int has_completed_circuit=0;
|
int has_completed_circuit=0;
|
||||||
|
|
||||||
#ifdef MS_WINDOWS
|
|
||||||
#define MS_WINDOWS_SERVICE
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef MS_WINDOWS_SERVICE
|
|
||||||
#include <tchar.h>
|
|
||||||
#define GENSRV_SERVICENAME TEXT("tor")
|
|
||||||
#define GENSRV_DISPLAYNAME TEXT("Tor Win32 Service")
|
|
||||||
#define GENSRV_DESCRIPTION \
|
|
||||||
TEXT("Provides an anonymous Internet communication system")
|
|
||||||
#define GENSRV_USERACCT TEXT("NT AUTHORITY\\LocalService")
|
|
||||||
|
|
||||||
// Cheating: using the pre-defined error codes, tricks Windows into displaying
|
|
||||||
// a semi-related human-readable error message if startup fails as
|
|
||||||
// opposed to simply scaring people with Error: 0xffffffff
|
|
||||||
#define NT_SERVICE_ERROR_TORINIT_FAILED ERROR_EXCEPTION_IN_SERVICE
|
|
||||||
|
|
||||||
SERVICE_STATUS service_status;
|
|
||||||
SERVICE_STATUS_HANDLE hStatus;
|
|
||||||
/* XXXX020 This 'backup argv' and 'backup argc' business is an ugly hack. This
|
|
||||||
* is a job for arguments, not globals. */
|
|
||||||
static char **backup_argv;
|
|
||||||
static int backup_argc;
|
|
||||||
static int nt_service_is_stopping(void);
|
|
||||||
static char* nt_strerror(uint32_t errnum);
|
|
||||||
#else
|
|
||||||
#define nt_service_is_stopping() (0)
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/** If our router descriptor ever goes this long without being regenerated
|
/** If our router descriptor ever goes this long without being regenerated
|
||||||
* because something changed, we force an immediate regenerate-and-upload. */
|
* because something changed, we force an immediate regenerate-and-upload. */
|
||||||
#define FORCE_REGENERATE_DESCRIPTOR_INTERVAL (18*60*60)
|
#define FORCE_REGENERATE_DESCRIPTOR_INTERVAL (18*60*60)
|
||||||
@ -1909,685 +1880,6 @@ do_hash_password(void)
|
|||||||
printf("16:%s\n",output);
|
printf("16:%s\n",output);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef MS_WINDOWS_SERVICE
|
|
||||||
|
|
||||||
/* XXXX can some/all these functions become static? without breaking NT
|
|
||||||
* services? -NM */
|
|
||||||
/* XXXX I'd also like to move much of the NT service stuff into its own
|
|
||||||
* file. -RD */
|
|
||||||
void nt_service_control(DWORD request);
|
|
||||||
void nt_service_body(int argc, char **argv);
|
|
||||||
void nt_service_main(void);
|
|
||||||
SC_HANDLE nt_service_open_scm(void);
|
|
||||||
SC_HANDLE nt_service_open(SC_HANDLE hSCManager);
|
|
||||||
int nt_service_start(SC_HANDLE hService);
|
|
||||||
int nt_service_stop(SC_HANDLE hService);
|
|
||||||
int nt_service_install(int argc, char **argv);
|
|
||||||
int nt_service_remove(void);
|
|
||||||
int nt_service_cmd_start(void);
|
|
||||||
int nt_service_cmd_stop(void);
|
|
||||||
|
|
||||||
struct service_fns {
|
|
||||||
int loaded;
|
|
||||||
|
|
||||||
BOOL (WINAPI *ChangeServiceConfig2A_fn)(
|
|
||||||
SC_HANDLE hService,
|
|
||||||
DWORD dwInfoLevel,
|
|
||||||
LPVOID lpInfo);
|
|
||||||
|
|
||||||
BOOL (WINAPI *CloseServiceHandle_fn)(
|
|
||||||
SC_HANDLE hSCObject);
|
|
||||||
|
|
||||||
BOOL (WINAPI *ControlService_fn)(
|
|
||||||
SC_HANDLE hService,
|
|
||||||
DWORD dwControl,
|
|
||||||
LPSERVICE_STATUS lpServiceStatus);
|
|
||||||
|
|
||||||
SC_HANDLE (WINAPI *CreateServiceA_fn)(
|
|
||||||
SC_HANDLE hSCManager,
|
|
||||||
LPCTSTR lpServiceName,
|
|
||||||
LPCTSTR lpDisplayName,
|
|
||||||
DWORD dwDesiredAccess,
|
|
||||||
DWORD dwServiceType,
|
|
||||||
DWORD dwStartType,
|
|
||||||
DWORD dwErrorControl,
|
|
||||||
LPCTSTR lpBinaryPathName,
|
|
||||||
LPCTSTR lpLoadOrderGroup,
|
|
||||||
LPDWORD lpdwTagId,
|
|
||||||
LPCTSTR lpDependencies,
|
|
||||||
LPCTSTR lpServiceStartName,
|
|
||||||
LPCTSTR lpPassword);
|
|
||||||
|
|
||||||
BOOL (WINAPI *DeleteService_fn)(
|
|
||||||
SC_HANDLE hService);
|
|
||||||
|
|
||||||
SC_HANDLE (WINAPI *OpenSCManagerA_fn)(
|
|
||||||
LPCTSTR lpMachineName,
|
|
||||||
LPCTSTR lpDatabaseName,
|
|
||||||
DWORD dwDesiredAccess);
|
|
||||||
|
|
||||||
SC_HANDLE (WINAPI *OpenServiceA_fn)(
|
|
||||||
SC_HANDLE hSCManager,
|
|
||||||
LPCTSTR lpServiceName,
|
|
||||||
DWORD dwDesiredAccess);
|
|
||||||
|
|
||||||
BOOL (WINAPI *QueryServiceStatus_fn)(
|
|
||||||
SC_HANDLE hService,
|
|
||||||
LPSERVICE_STATUS lpServiceStatus);
|
|
||||||
|
|
||||||
SERVICE_STATUS_HANDLE (WINAPI *RegisterServiceCtrlHandlerA_fn)(
|
|
||||||
LPCTSTR lpServiceName,
|
|
||||||
LPHANDLER_FUNCTION lpHandlerProc);
|
|
||||||
|
|
||||||
BOOL (WINAPI *SetServiceStatus_fn)(SERVICE_STATUS_HANDLE,
|
|
||||||
LPSERVICE_STATUS);
|
|
||||||
|
|
||||||
BOOL (WINAPI *StartServiceCtrlDispatcherA_fn)(
|
|
||||||
const SERVICE_TABLE_ENTRY* lpServiceTable);
|
|
||||||
|
|
||||||
BOOL (WINAPI *StartServiceA_fn)(
|
|
||||||
SC_HANDLE hService,
|
|
||||||
DWORD dwNumServiceArgs,
|
|
||||||
LPCTSTR* lpServiceArgVectors);
|
|
||||||
|
|
||||||
BOOL (WINAPI *LookupAccountNameA_fn)(
|
|
||||||
LPCTSTR lpSystemName,
|
|
||||||
LPCTSTR lpAccountName,
|
|
||||||
PSID Sid,
|
|
||||||
LPDWORD cbSid,
|
|
||||||
LPTSTR ReferencedDomainName,
|
|
||||||
LPDWORD cchReferencedDomainName,
|
|
||||||
PSID_NAME_USE peUse);
|
|
||||||
} service_fns = { 0,
|
|
||||||
NULL, NULL, NULL, NULL, NULL, NULL,
|
|
||||||
NULL, NULL, NULL, NULL, NULL, NULL,
|
|
||||||
NULL};
|
|
||||||
|
|
||||||
/** Loads functions used by NT services. Returns on success, or prints a
|
|
||||||
* complaint to stdout and exits on error. */
|
|
||||||
static void
|
|
||||||
nt_service_loadlibrary(void)
|
|
||||||
{
|
|
||||||
HMODULE library = 0;
|
|
||||||
void *fn;
|
|
||||||
|
|
||||||
if (service_fns.loaded)
|
|
||||||
return;
|
|
||||||
|
|
||||||
/* XXXX Possibly, we should hardcode the location of this DLL. */
|
|
||||||
if (!(library = LoadLibrary("advapi32.dll"))) {
|
|
||||||
log_err(LD_GENERAL, "Couldn't open advapi32.dll. Are you trying to use "
|
|
||||||
"NT services on Windows 98? That doesn't work.");
|
|
||||||
goto err;
|
|
||||||
}
|
|
||||||
|
|
||||||
#define LOAD(f) STMT_BEGIN \
|
|
||||||
if (!(fn = GetProcAddress(library, #f))) { \
|
|
||||||
log_err(LD_BUG, \
|
|
||||||
"Couldn't find %s in advapi32.dll! We probably got the " \
|
|
||||||
"name wrong.", #f); \
|
|
||||||
goto err; \
|
|
||||||
} else { \
|
|
||||||
service_fns.f ## _fn = fn; \
|
|
||||||
} \
|
|
||||||
STMT_END
|
|
||||||
|
|
||||||
LOAD(ChangeServiceConfig2A);
|
|
||||||
LOAD(CloseServiceHandle);
|
|
||||||
LOAD(ControlService);
|
|
||||||
LOAD(CreateServiceA);
|
|
||||||
LOAD(DeleteService);
|
|
||||||
LOAD(OpenSCManagerA);
|
|
||||||
LOAD(OpenServiceA);
|
|
||||||
LOAD(QueryServiceStatus);
|
|
||||||
LOAD(RegisterServiceCtrlHandlerA);
|
|
||||||
LOAD(SetServiceStatus);
|
|
||||||
LOAD(StartServiceCtrlDispatcherA);
|
|
||||||
LOAD(StartServiceA);
|
|
||||||
LOAD(LookupAccountNameA);
|
|
||||||
|
|
||||||
service_fns.loaded = 1;
|
|
||||||
|
|
||||||
return;
|
|
||||||
err:
|
|
||||||
printf("Unable to load library support for NT services: exiting.\n");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
/** If we're compiled to run as an NT service, and the service wants to
|
|
||||||
* shut down, then change our current status and return 1. Else
|
|
||||||
* return 0.
|
|
||||||
*/
|
|
||||||
static int
|
|
||||||
nt_service_is_stopping(void)
|
|
||||||
/* XXXX this function would probably _love_ to be inline, in 0.2.0. */
|
|
||||||
{
|
|
||||||
/* If we haven't loaded the function pointers, we can't possibly be an NT
|
|
||||||
* service trying to shut down. */
|
|
||||||
if (!service_fns.loaded)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
if (service_status.dwCurrentState == SERVICE_STOP_PENDING) {
|
|
||||||
service_status.dwWin32ExitCode = 0;
|
|
||||||
service_status.dwCurrentState = SERVICE_STOPPED;
|
|
||||||
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
|
||||||
return 1;
|
|
||||||
} else if (service_status.dwCurrentState == SERVICE_STOPPED) {
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Handles service control requests, such as stopping or starting the
|
|
||||||
* Tor service. */
|
|
||||||
void
|
|
||||||
nt_service_control(DWORD request)
|
|
||||||
{
|
|
||||||
static struct timeval exit_now;
|
|
||||||
exit_now.tv_sec = 0;
|
|
||||||
exit_now.tv_usec = 0;
|
|
||||||
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
|
|
||||||
switch (request) {
|
|
||||||
case SERVICE_CONTROL_STOP:
|
|
||||||
case SERVICE_CONTROL_SHUTDOWN:
|
|
||||||
log_notice(LD_GENERAL,
|
|
||||||
"Got stop/shutdown request; shutting down cleanly.");
|
|
||||||
service_status.dwCurrentState = SERVICE_STOP_PENDING;
|
|
||||||
event_loopexit(&exit_now);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Called when the service is started via the system's service control
|
|
||||||
* manager. This calls tor_init() and starts the main event loop. If
|
|
||||||
* tor_init() fails, the service will be stopped and exit code set to
|
|
||||||
* NT_SERVICE_ERROR_TORINIT_FAILED. */
|
|
||||||
void
|
|
||||||
nt_service_body(int argc, char **argv)
|
|
||||||
{
|
|
||||||
int r;
|
|
||||||
(void) argc; /* unused */
|
|
||||||
(void) argv; /* unused */
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
|
||||||
service_status.dwCurrentState = SERVICE_START_PENDING;
|
|
||||||
service_status.dwControlsAccepted =
|
|
||||||
SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
|
|
||||||
service_status.dwWin32ExitCode = 0;
|
|
||||||
service_status.dwServiceSpecificExitCode = 0;
|
|
||||||
service_status.dwCheckPoint = 0;
|
|
||||||
service_status.dwWaitHint = 1000;
|
|
||||||
hStatus = service_fns.RegisterServiceCtrlHandlerA_fn(GENSRV_SERVICENAME,
|
|
||||||
(LPHANDLER_FUNCTION) nt_service_control);
|
|
||||||
|
|
||||||
if (hStatus == 0) {
|
|
||||||
/* Failed to register the service control handler function */
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
r = tor_init(backup_argc, backup_argv);
|
|
||||||
if (r) {
|
|
||||||
/* Failed to start the Tor service */
|
|
||||||
r = NT_SERVICE_ERROR_TORINIT_FAILED;
|
|
||||||
service_status.dwCurrentState = SERVICE_STOPPED;
|
|
||||||
service_status.dwWin32ExitCode = r;
|
|
||||||
service_status.dwServiceSpecificExitCode = r;
|
|
||||||
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Set the service's status to SERVICE_RUNNING and start the main
|
|
||||||
* event loop */
|
|
||||||
service_status.dwCurrentState = SERVICE_RUNNING;
|
|
||||||
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
|
||||||
do_main_loop();
|
|
||||||
tor_cleanup();
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Main service entry point. Starts the service control dispatcher and waits
|
|
||||||
* until the service status is set to SERVICE_STOPPED. */
|
|
||||||
void
|
|
||||||
nt_service_main(void)
|
|
||||||
{
|
|
||||||
SERVICE_TABLE_ENTRY table[2];
|
|
||||||
DWORD result = 0;
|
|
||||||
char *errmsg;
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
table[0].lpServiceName = (char*)GENSRV_SERVICENAME;
|
|
||||||
table[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)nt_service_body;
|
|
||||||
table[1].lpServiceName = NULL;
|
|
||||||
table[1].lpServiceProc = NULL;
|
|
||||||
|
|
||||||
if (!service_fns.StartServiceCtrlDispatcherA_fn(table)) {
|
|
||||||
result = GetLastError();
|
|
||||||
errmsg = nt_strerror(result);
|
|
||||||
printf("Service error %d : %s\n", (int) result, errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
if (result == ERROR_FAILED_SERVICE_CONTROLLER_CONNECT) {
|
|
||||||
if (tor_init(backup_argc, backup_argv) < 0)
|
|
||||||
return;
|
|
||||||
switch (get_options()->command) {
|
|
||||||
case CMD_RUN_TOR:
|
|
||||||
do_main_loop();
|
|
||||||
break;
|
|
||||||
case CMD_LIST_FINGERPRINT:
|
|
||||||
do_list_fingerprint();
|
|
||||||
break;
|
|
||||||
case CMD_HASH_PASSWORD:
|
|
||||||
do_hash_password();
|
|
||||||
break;
|
|
||||||
case CMD_VERIFY_CONFIG:
|
|
||||||
printf("Configuration was valid\n");
|
|
||||||
break;
|
|
||||||
case CMD_RUN_UNITTESTS:
|
|
||||||
default:
|
|
||||||
log_err(LD_CONFIG, "Illegal command number %d: internal error.",
|
|
||||||
get_options()->command);
|
|
||||||
}
|
|
||||||
tor_cleanup();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Return a handle to the service control manager on success, or NULL on
|
|
||||||
* failure. */
|
|
||||||
SC_HANDLE
|
|
||||||
nt_service_open_scm(void)
|
|
||||||
{
|
|
||||||
SC_HANDLE hSCManager;
|
|
||||||
char *errmsg = NULL;
|
|
||||||
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
if ((hSCManager = service_fns.OpenSCManagerA_fn(
|
|
||||||
NULL, NULL, SC_MANAGER_CREATE_SERVICE)) == NULL) {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("OpenSCManager() failed : %s\n", errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
}
|
|
||||||
return hSCManager;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Open a handle to the Tor service using <b>hSCManager</b>. Return NULL
|
|
||||||
* on failure. */
|
|
||||||
SC_HANDLE
|
|
||||||
nt_service_open(SC_HANDLE hSCManager)
|
|
||||||
{
|
|
||||||
SC_HANDLE hService;
|
|
||||||
char *errmsg = NULL;
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
if ((hService = service_fns.OpenServiceA_fn(hSCManager, GENSRV_SERVICENAME,
|
|
||||||
SERVICE_ALL_ACCESS)) == NULL) {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("OpenService() failed : %s\n", errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
}
|
|
||||||
return hService;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Start the Tor service. Return 0 if the service is started or was
|
|
||||||
* previously running. Return -1 on error. */
|
|
||||||
int
|
|
||||||
nt_service_start(SC_HANDLE hService)
|
|
||||||
{
|
|
||||||
char *errmsg = NULL;
|
|
||||||
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
|
|
||||||
service_fns.QueryServiceStatus_fn(hService, &service_status);
|
|
||||||
if (service_status.dwCurrentState == SERVICE_RUNNING) {
|
|
||||||
printf("Service is already running\n");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (service_fns.StartServiceA_fn(hService, 0, NULL)) {
|
|
||||||
/* Loop until the service has finished attempting to start */
|
|
||||||
while (service_fns.QueryServiceStatus_fn(hService, &service_status) &&
|
|
||||||
(service_status.dwCurrentState == SERVICE_START_PENDING)) {
|
|
||||||
Sleep(500);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Check if it started successfully or not */
|
|
||||||
if (service_status.dwCurrentState == SERVICE_RUNNING) {
|
|
||||||
printf("Service started successfully\n");
|
|
||||||
return 0;
|
|
||||||
} else {
|
|
||||||
errmsg = nt_strerror(service_status.dwWin32ExitCode);
|
|
||||||
printf("Service failed to start : %s\n", errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("StartService() failed : %s\n", errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
}
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Stop the Tor service. Return 0 if the service is stopped or was not
|
|
||||||
* previously running. Return -1 on error. */
|
|
||||||
int
|
|
||||||
nt_service_stop(SC_HANDLE hService)
|
|
||||||
{
|
|
||||||
/** Wait at most 10 seconds for the service to stop. */
|
|
||||||
#define MAX_SERVICE_WAIT_TIME 10
|
|
||||||
int wait_time;
|
|
||||||
char *errmsg = NULL;
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
|
|
||||||
service_fns.QueryServiceStatus_fn(hService, &service_status);
|
|
||||||
if (service_status.dwCurrentState == SERVICE_STOPPED) {
|
|
||||||
printf("Service is already stopped\n");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (service_fns.ControlService_fn(hService, SERVICE_CONTROL_STOP,
|
|
||||||
&service_status)) {
|
|
||||||
wait_time = 0;
|
|
||||||
while (service_fns.QueryServiceStatus_fn(hService, &service_status) &&
|
|
||||||
(service_status.dwCurrentState != SERVICE_STOPPED) &&
|
|
||||||
(wait_time < MAX_SERVICE_WAIT_TIME)) {
|
|
||||||
Sleep(1000);
|
|
||||||
wait_time++;
|
|
||||||
}
|
|
||||||
if (service_status.dwCurrentState == SERVICE_STOPPED) {
|
|
||||||
printf("Service stopped successfully\n");
|
|
||||||
return 0;
|
|
||||||
} else if (wait_time == MAX_SERVICE_WAIT_TIME) {
|
|
||||||
printf("Service did not stop within %d seconds.\n", wait_time);
|
|
||||||
} else {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("QueryServiceStatus() failed : %s\n",errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("ControlService() failed : %s\n", errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
}
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Build a formatted command line used for the NT service. Return a
|
|
||||||
* pointer to the formatted string on success, or NULL on failure. Set
|
|
||||||
* *<b>using_default_torrc</b> to true if we're going to use the default
|
|
||||||
* location to torrc, or 1 if an option was specified on the command line.
|
|
||||||
*/
|
|
||||||
static char *
|
|
||||||
nt_service_command_line(int *using_default_torrc)
|
|
||||||
{
|
|
||||||
TCHAR tor_exe[MAX_PATH+1];
|
|
||||||
char *command, *options=NULL;
|
|
||||||
smartlist_t *sl;
|
|
||||||
int i, cmdlen;
|
|
||||||
*using_default_torrc = 1;
|
|
||||||
|
|
||||||
/* Get the location of tor.exe */
|
|
||||||
if (0 == GetModuleFileName(NULL, tor_exe, MAX_PATH))
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
/* Get the service arguments */
|
|
||||||
sl = smartlist_create();
|
|
||||||
for (i = 1; i < backup_argc; ++i) {
|
|
||||||
if (!strcmp(backup_argv[i], "--options") ||
|
|
||||||
!strcmp(backup_argv[i], "-options")) {
|
|
||||||
while (++i < backup_argc) {
|
|
||||||
if (!strcmp(backup_argv[i], "-f"))
|
|
||||||
*using_default_torrc = 0;
|
|
||||||
smartlist_add(sl, backup_argv[i]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (smartlist_len(sl))
|
|
||||||
options = smartlist_join_strings(sl,"\" \"",0,NULL);
|
|
||||||
smartlist_free(sl);
|
|
||||||
|
|
||||||
/* Allocate a string for the NT service command line */
|
|
||||||
cmdlen = strlen(tor_exe) + (options?strlen(options):0) + 32;
|
|
||||||
command = tor_malloc(cmdlen);
|
|
||||||
|
|
||||||
/* Format the service command */
|
|
||||||
if (options) {
|
|
||||||
if (tor_snprintf(command, cmdlen, "\"%s\" --nt-service \"%s\"",
|
|
||||||
tor_exe, options)<0) {
|
|
||||||
tor_free(command); /* sets command to NULL. */
|
|
||||||
}
|
|
||||||
} else { /* ! options */
|
|
||||||
if (tor_snprintf(command, cmdlen, "\"%s\" --nt-service", tor_exe)<0) {
|
|
||||||
tor_free(command); /* sets command to NULL. */
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tor_free(options);
|
|
||||||
return command;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Creates a Tor NT service, set to start on boot. The service will be
|
|
||||||
* started if installation succeeds. Returns 0 on success, or -1 on
|
|
||||||
* failure. */
|
|
||||||
int
|
|
||||||
nt_service_install(int argc, char **argv)
|
|
||||||
{
|
|
||||||
/* Notes about developing NT services:
|
|
||||||
*
|
|
||||||
* 1. Don't count on your CWD. If an absolute path is not given, the
|
|
||||||
* fopen() function goes wrong.
|
|
||||||
* 2. The parameters given to the nt_service_body() function differ
|
|
||||||
* from those given to main() function.
|
|
||||||
*/
|
|
||||||
|
|
||||||
SC_HANDLE hSCManager = NULL;
|
|
||||||
SC_HANDLE hService = NULL;
|
|
||||||
SERVICE_DESCRIPTION sdBuff;
|
|
||||||
char *command;
|
|
||||||
char *errmsg;
|
|
||||||
const char *user_acct = GENSRV_USERACCT;
|
|
||||||
const char *password = "";
|
|
||||||
int i;
|
|
||||||
OSVERSIONINFOEX info;
|
|
||||||
SID_NAME_USE sidUse;
|
|
||||||
DWORD sidLen = 0, domainLen = 0;
|
|
||||||
int is_win2k_or_worse = 0;
|
|
||||||
int using_default_torrc = 0;
|
|
||||||
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
|
|
||||||
/* Open the service control manager so we can create a new service */
|
|
||||||
if ((hSCManager = nt_service_open_scm()) == NULL)
|
|
||||||
return -1;
|
|
||||||
/* Build the command line used for the service */
|
|
||||||
if ((command = nt_service_command_line(&using_default_torrc)) == NULL) {
|
|
||||||
printf("Unable to build service command line.\n");
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (i=1; i < argc; ++i) {
|
|
||||||
if (!strcmp(argv[i], "--user") && i+1<argc) {
|
|
||||||
user_acct = argv[i+1];
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
if (!strcmp(argv[i], "--password") && i+1<argc) {
|
|
||||||
password = argv[i+1];
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Compute our version and see whether we're running win2k or earlier. */
|
|
||||||
memset(&info, 0, sizeof(info));
|
|
||||||
info.dwOSVersionInfoSize = sizeof(info);
|
|
||||||
if (! GetVersionEx((LPOSVERSIONINFO)&info)) {
|
|
||||||
printf("Call to GetVersionEx failed.\n");
|
|
||||||
is_win2k_or_worse = 1;
|
|
||||||
} else {
|
|
||||||
if (info.dwMajorVersion < 5 ||
|
|
||||||
(info.dwMajorVersion == 5 && info.dwMinorVersion == 0))
|
|
||||||
is_win2k_or_worse = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (user_acct == GENSRV_USERACCT) {
|
|
||||||
if (is_win2k_or_worse) {
|
|
||||||
/* On Win2k, there is no LocalService account, so we actually need to
|
|
||||||
* fall back on NULL (the system account). */
|
|
||||||
printf("Running on Win2K or earlier, so the LocalService account "
|
|
||||||
"doesn't exist. Falling back to SYSTEM account.\n");
|
|
||||||
user_acct = NULL;
|
|
||||||
} else {
|
|
||||||
/* Genericity is apparently _so_ last year in Redmond, where some
|
|
||||||
* accounts are accounts that you can look up, and some accounts
|
|
||||||
* are magic and undetectable via the security subsystem. See
|
|
||||||
* http://msdn2.microsoft.com/en-us/library/ms684188.aspx
|
|
||||||
*/
|
|
||||||
printf("Running on a Post-Win2K OS, so we'll assume that the "
|
|
||||||
"LocalService account exists.\n");
|
|
||||||
}
|
|
||||||
} else if (0 && service_fns.LookupAccountNameA_fn(NULL, // On this system
|
|
||||||
user_acct,
|
|
||||||
NULL, &sidLen, // Don't care about the SID
|
|
||||||
NULL, &domainLen, // Don't care about the domain
|
|
||||||
&sidUse) == 0) {
|
|
||||||
/* XXXX For some reason, the above test segfaults. Fix that. */
|
|
||||||
printf("User \"%s\" doesn't seem to exist.\n", user_acct);
|
|
||||||
return -1;
|
|
||||||
} else {
|
|
||||||
printf("Will try to install service as user \"%s\".\n", user_acct);
|
|
||||||
}
|
|
||||||
/* XXXX This warning could be better about explaining how to resolve the
|
|
||||||
* situation. */
|
|
||||||
if (using_default_torrc)
|
|
||||||
printf("IMPORTANT NOTE:\n"
|
|
||||||
" The Tor service will run under the account \"%s\". This means\n"
|
|
||||||
" that Tor will look for its configuration file under that\n"
|
|
||||||
" account's Application Data directory, which is probably not\n"
|
|
||||||
" the same as yours.\n", user_acct?user_acct:"<local system>");
|
|
||||||
|
|
||||||
/* Create the Tor service, set to auto-start on boot */
|
|
||||||
if ((hService = service_fns.CreateServiceA_fn(hSCManager, GENSRV_SERVICENAME,
|
|
||||||
GENSRV_DISPLAYNAME,
|
|
||||||
SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
|
|
||||||
SERVICE_AUTO_START, SERVICE_ERROR_IGNORE,
|
|
||||||
command, NULL, NULL, NULL,
|
|
||||||
user_acct, password)) == NULL) {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("CreateService() failed : %s\n", errmsg);
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
tor_free(command);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
printf("Done with CreateService.\n");
|
|
||||||
|
|
||||||
/* Set the service's description */
|
|
||||||
sdBuff.lpDescription = (char*)GENSRV_DESCRIPTION;
|
|
||||||
service_fns.ChangeServiceConfig2A_fn(hService, SERVICE_CONFIG_DESCRIPTION,
|
|
||||||
&sdBuff);
|
|
||||||
printf("Service installed successfully\n");
|
|
||||||
|
|
||||||
/* Start the service initially */
|
|
||||||
nt_service_start(hService);
|
|
||||||
|
|
||||||
service_fns.CloseServiceHandle_fn(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
tor_free(command);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Removes the Tor NT service. Returns 0 if the service was successfully
|
|
||||||
* removed, or -1 on error. */
|
|
||||||
int
|
|
||||||
nt_service_remove(void)
|
|
||||||
{
|
|
||||||
SC_HANDLE hSCManager = NULL;
|
|
||||||
SC_HANDLE hService = NULL;
|
|
||||||
char *errmsg;
|
|
||||||
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
if ((hSCManager = nt_service_open_scm()) == NULL)
|
|
||||||
return -1;
|
|
||||||
if ((hService = nt_service_open(hSCManager)) == NULL) {
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
nt_service_stop(hService);
|
|
||||||
if (service_fns.DeleteService_fn(hService) == FALSE) {
|
|
||||||
errmsg = nt_strerror(GetLastError());
|
|
||||||
printf("DeleteService() failed : %s\n", errmsg);
|
|
||||||
LocalFree(errmsg);
|
|
||||||
service_fns.CloseServiceHandle_fn(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
service_fns.CloseServiceHandle_fn(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
printf("Service removed successfully\n");
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Starts the Tor service. Returns 0 on success, or -1 on error. */
|
|
||||||
int
|
|
||||||
nt_service_cmd_start(void)
|
|
||||||
{
|
|
||||||
SC_HANDLE hSCManager;
|
|
||||||
SC_HANDLE hService;
|
|
||||||
int start;
|
|
||||||
|
|
||||||
if ((hSCManager = nt_service_open_scm()) == NULL)
|
|
||||||
return -1;
|
|
||||||
if ((hService = nt_service_open(hSCManager)) == NULL) {
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
start = nt_service_start(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
|
|
||||||
return start;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Stops the Tor service. Returns 0 on success, or -1 on error. */
|
|
||||||
int
|
|
||||||
nt_service_cmd_stop(void)
|
|
||||||
{
|
|
||||||
SC_HANDLE hSCManager;
|
|
||||||
SC_HANDLE hService;
|
|
||||||
int stop;
|
|
||||||
|
|
||||||
if ((hSCManager = nt_service_open_scm()) == NULL)
|
|
||||||
return -1;
|
|
||||||
if ((hService = nt_service_open(hSCManager)) == NULL) {
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
stop = nt_service_stop(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hService);
|
|
||||||
service_fns.CloseServiceHandle_fn(hSCManager);
|
|
||||||
|
|
||||||
return stop;
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Given a Win32 error code, this attempts to make Windows
|
|
||||||
* return a human-readable error message. The char* returned
|
|
||||||
* is allocated by Windows, but should be freed with LocalFree()
|
|
||||||
* when finished with it. */
|
|
||||||
static char*
|
|
||||||
nt_strerror(uint32_t errnum)
|
|
||||||
{
|
|
||||||
char *msgbuf;
|
|
||||||
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
|
|
||||||
NULL, errnum, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
|
||||||
(LPSTR)&msgbuf, 0, NULL);
|
|
||||||
return msgbuf;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef USE_DMALLOC
|
#ifdef USE_DMALLOC
|
||||||
#include <openssl/crypto.h>
|
#include <openssl/crypto.h>
|
||||||
static void
|
static void
|
||||||
@ -2609,53 +1901,16 @@ tor_main(int argc, char *argv[])
|
|||||||
_tor_dmalloc_free);
|
_tor_dmalloc_free);
|
||||||
log_notice(LD_CONFIG, "Set up dmalloc; returned %d", r);
|
log_notice(LD_CONFIG, "Set up dmalloc; returned %d", r);
|
||||||
#endif
|
#endif
|
||||||
#ifdef MS_WINDOWS_SERVICE
|
#ifdef NT_SERVICE
|
||||||
backup_argv = argv;
|
if ((result = nt_service_parse_options(argv, argc)))
|
||||||
backup_argc = argc;
|
return result;
|
||||||
if ((argc >= 3) &&
|
|
||||||
(!strcmp(argv[1], "-service") || !strcmp(argv[1], "--service"))) {
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
if (!strcmp(argv[2], "install"))
|
|
||||||
return nt_service_install(argc, argv);
|
|
||||||
if (!strcmp(argv[2], "remove"))
|
|
||||||
return nt_service_remove();
|
|
||||||
if (!strcmp(argv[2], "start"))
|
|
||||||
return nt_service_cmd_start();
|
|
||||||
if (!strcmp(argv[2], "stop"))
|
|
||||||
return nt_service_cmd_stop();
|
|
||||||
printf("Unrecognized service command '%s'\n", argv[2]);
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (argc >= 2) {
|
|
||||||
if (!strcmp(argv[1], "-nt-service") || !strcmp(argv[1], "--nt-service")) {
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
nt_service_main();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
// These values have been deprecated since 0.1.1.2-alpha; we've warned
|
|
||||||
// about them since 0.1.2.7-alpha.
|
|
||||||
if (!strcmp(argv[1], "-install") || !strcmp(argv[1], "--install")) {
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
fprintf(stderr,
|
|
||||||
"The %s option is deprecated; use \"--service install\" instead.",
|
|
||||||
argv[1]);
|
|
||||||
return nt_service_install(argc, argv);
|
|
||||||
}
|
|
||||||
if (!strcmp(argv[1], "-remove") || !strcmp(argv[1], "--remove")) {
|
|
||||||
nt_service_loadlibrary();
|
|
||||||
fprintf(stderr,
|
|
||||||
"The %s option is deprecated; use \"--service remove\" instead.",
|
|
||||||
argv[1]);
|
|
||||||
return nt_service_remove();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif
|
#endif
|
||||||
if (tor_init(argc, argv)<0)
|
if (tor_init(argc, argv)<0)
|
||||||
return -1;
|
return -1;
|
||||||
switch (get_options()->command) {
|
switch (get_options()->command) {
|
||||||
case CMD_RUN_TOR:
|
case CMD_RUN_TOR:
|
||||||
#ifdef MS_WINDOWS_SERVICE
|
#ifdef NT_SERVICE
|
||||||
service_status.dwCurrentState = SERVICE_RUNNING;
|
nt_service_set_state(SERVICE_RUNNING);
|
||||||
#endif
|
#endif
|
||||||
result = do_main_loop();
|
result = do_main_loop();
|
||||||
break;
|
break;
|
||||||
|
|||||||
757
src/or/ntmain.c
Normal file
757
src/or/ntmain.c
Normal file
@ -0,0 +1,757 @@
|
|||||||
|
/* Copyright (c) 2001 Matej Pfajfar.
|
||||||
|
* copyright (c) 2001-2004, Roger Dingledine.
|
||||||
|
* Copyright (c) 2004-2007, Roger Dingledine, Nick Mathewson. */
|
||||||
|
/* See LICENSE for licensing information */
|
||||||
|
/* $Id$ */
|
||||||
|
|
||||||
|
#include "or.h"
|
||||||
|
|
||||||
|
const char ntmain_c_id[] =
|
||||||
|
"$Id$";
|
||||||
|
|
||||||
|
#include <tchar.h>
|
||||||
|
#define GENSRV_SERVICENAME TEXT("tor")
|
||||||
|
#define GENSRV_DISPLAYNAME TEXT("Tor Win32 Service")
|
||||||
|
#define GENSRV_DESCRIPTION \
|
||||||
|
TEXT("Provides an anonymous Internet communication system")
|
||||||
|
#define GENSRV_USERACCT TEXT("NT AUTHORITY\\LocalService")
|
||||||
|
|
||||||
|
// Cheating: using the pre-defined error codes, tricks Windows into displaying
|
||||||
|
// a semi-related human-readable error message if startup fails as
|
||||||
|
// opposed to simply scaring people with Error: 0xffffffff
|
||||||
|
#define NT_SERVICE_ERROR_TORINIT_FAILED ERROR_EXCEPTION_IN_SERVICE
|
||||||
|
|
||||||
|
static SERVICE_STATUS service_status;
|
||||||
|
static SERVICE_STATUS_HANDLE hStatus;
|
||||||
|
|
||||||
|
/* XXXX020 This 'backup argv' and 'backup argc' business is an ugly hack. This
|
||||||
|
* is a job for arguments, not globals. */
|
||||||
|
static char **backup_argv;
|
||||||
|
static int backup_argc;
|
||||||
|
static char* nt_strerror(uint32_t errnum);
|
||||||
|
|
||||||
|
static void nt_service_control(DWORD request);
|
||||||
|
static void nt_service_body(int argc, char **argv);
|
||||||
|
static void nt_service_main(void);
|
||||||
|
static SC_HANDLE nt_service_open_scm(void);
|
||||||
|
static SC_HANDLE nt_service_open(SC_HANDLE hSCManager);
|
||||||
|
static int nt_service_start(SC_HANDLE hService);
|
||||||
|
static int nt_service_stop(SC_HANDLE hService);
|
||||||
|
static int nt_service_install(int argc, char **argv);
|
||||||
|
static int nt_service_remove(void);
|
||||||
|
static int nt_service_cmd_start(void);
|
||||||
|
static int nt_service_cmd_stop(void);
|
||||||
|
|
||||||
|
struct service_fns {
|
||||||
|
int loaded;
|
||||||
|
|
||||||
|
BOOL (WINAPI *ChangeServiceConfig2A_fn)(
|
||||||
|
SC_HANDLE hService,
|
||||||
|
DWORD dwInfoLevel,
|
||||||
|
LPVOID lpInfo);
|
||||||
|
|
||||||
|
BOOL (WINAPI *CloseServiceHandle_fn)(
|
||||||
|
SC_HANDLE hSCObject);
|
||||||
|
|
||||||
|
BOOL (WINAPI *ControlService_fn)(
|
||||||
|
SC_HANDLE hService,
|
||||||
|
DWORD dwControl,
|
||||||
|
LPSERVICE_STATUS lpServiceStatus);
|
||||||
|
|
||||||
|
SC_HANDLE (WINAPI *CreateServiceA_fn)(
|
||||||
|
SC_HANDLE hSCManager,
|
||||||
|
LPCTSTR lpServiceName,
|
||||||
|
LPCTSTR lpDisplayName,
|
||||||
|
DWORD dwDesiredAccess,
|
||||||
|
DWORD dwServiceType,
|
||||||
|
DWORD dwStartType,
|
||||||
|
DWORD dwErrorControl,
|
||||||
|
LPCTSTR lpBinaryPathName,
|
||||||
|
LPCTSTR lpLoadOrderGroup,
|
||||||
|
LPDWORD lpdwTagId,
|
||||||
|
LPCTSTR lpDependencies,
|
||||||
|
LPCTSTR lpServiceStartName,
|
||||||
|
LPCTSTR lpPassword);
|
||||||
|
|
||||||
|
BOOL (WINAPI *DeleteService_fn)(
|
||||||
|
SC_HANDLE hService);
|
||||||
|
|
||||||
|
SC_HANDLE (WINAPI *OpenSCManagerA_fn)(
|
||||||
|
LPCTSTR lpMachineName,
|
||||||
|
LPCTSTR lpDatabaseName,
|
||||||
|
DWORD dwDesiredAccess);
|
||||||
|
|
||||||
|
SC_HANDLE (WINAPI *OpenServiceA_fn)(
|
||||||
|
SC_HANDLE hSCManager,
|
||||||
|
LPCTSTR lpServiceName,
|
||||||
|
DWORD dwDesiredAccess);
|
||||||
|
|
||||||
|
BOOL (WINAPI *QueryServiceStatus_fn)(
|
||||||
|
SC_HANDLE hService,
|
||||||
|
LPSERVICE_STATUS lpServiceStatus);
|
||||||
|
|
||||||
|
SERVICE_STATUS_HANDLE (WINAPI *RegisterServiceCtrlHandlerA_fn)(
|
||||||
|
LPCTSTR lpServiceName,
|
||||||
|
LPHANDLER_FUNCTION lpHandlerProc);
|
||||||
|
|
||||||
|
BOOL (WINAPI *SetServiceStatus_fn)(SERVICE_STATUS_HANDLE,
|
||||||
|
LPSERVICE_STATUS);
|
||||||
|
|
||||||
|
BOOL (WINAPI *StartServiceCtrlDispatcherA_fn)(
|
||||||
|
const SERVICE_TABLE_ENTRY* lpServiceTable);
|
||||||
|
|
||||||
|
BOOL (WINAPI *StartServiceA_fn)(
|
||||||
|
SC_HANDLE hService,
|
||||||
|
DWORD dwNumServiceArgs,
|
||||||
|
LPCTSTR* lpServiceArgVectors);
|
||||||
|
|
||||||
|
BOOL (WINAPI *LookupAccountNameA_fn)(
|
||||||
|
LPCTSTR lpSystemName,
|
||||||
|
LPCTSTR lpAccountName,
|
||||||
|
PSID Sid,
|
||||||
|
LPDWORD cbSid,
|
||||||
|
LPTSTR ReferencedDomainName,
|
||||||
|
LPDWORD cchReferencedDomainName,
|
||||||
|
PSID_NAME_USE peUse);
|
||||||
|
} service_fns = { 0,
|
||||||
|
NULL, NULL, NULL, NULL, NULL, NULL,
|
||||||
|
NULL, NULL, NULL, NULL, NULL, NULL,
|
||||||
|
NULL};
|
||||||
|
|
||||||
|
/** Loads functions used by NT services. Returns on success, or prints a
|
||||||
|
* complaint to stdout and exits on error. */
|
||||||
|
static void
|
||||||
|
nt_service_loadlibrary(void)
|
||||||
|
{
|
||||||
|
HMODULE library = 0;
|
||||||
|
void *fn;
|
||||||
|
|
||||||
|
if (service_fns.loaded)
|
||||||
|
return;
|
||||||
|
|
||||||
|
/* XXXX Possibly, we should hardcode the location of this DLL. */
|
||||||
|
if (!(library = LoadLibrary("advapi32.dll"))) {
|
||||||
|
log_err(LD_GENERAL, "Couldn't open advapi32.dll. Are you trying to use "
|
||||||
|
"NT services on Windows 98? That doesn't work.");
|
||||||
|
goto err;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define LOAD(f) STMT_BEGIN \
|
||||||
|
if (!(fn = GetProcAddress(library, #f))) { \
|
||||||
|
log_err(LD_BUG, \
|
||||||
|
"Couldn't find %s in advapi32.dll! We probably got the " \
|
||||||
|
"name wrong.", #f); \
|
||||||
|
goto err; \
|
||||||
|
} else { \
|
||||||
|
service_fns.f ## _fn = fn; \
|
||||||
|
} \
|
||||||
|
STMT_END
|
||||||
|
|
||||||
|
LOAD(ChangeServiceConfig2A);
|
||||||
|
LOAD(CloseServiceHandle);
|
||||||
|
LOAD(ControlService);
|
||||||
|
LOAD(CreateServiceA);
|
||||||
|
LOAD(DeleteService);
|
||||||
|
LOAD(OpenSCManagerA);
|
||||||
|
LOAD(OpenServiceA);
|
||||||
|
LOAD(QueryServiceStatus);
|
||||||
|
LOAD(RegisterServiceCtrlHandlerA);
|
||||||
|
LOAD(SetServiceStatus);
|
||||||
|
LOAD(StartServiceCtrlDispatcherA);
|
||||||
|
LOAD(StartServiceA);
|
||||||
|
LOAD(LookupAccountNameA);
|
||||||
|
|
||||||
|
service_fns.loaded = 1;
|
||||||
|
|
||||||
|
return;
|
||||||
|
err:
|
||||||
|
printf("Unable to load library support for NT services: exiting.\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** If we're compiled to run as an NT service, and the service wants to
|
||||||
|
* shut down, then change our current status and return 1. Else
|
||||||
|
* return 0.
|
||||||
|
*/
|
||||||
|
int
|
||||||
|
nt_service_is_stopping(void)
|
||||||
|
/* XXXX this function would probably _love_ to be inline, in 0.2.0. */
|
||||||
|
{
|
||||||
|
/* If we haven't loaded the function pointers, we can't possibly be an NT
|
||||||
|
* service trying to shut down. */
|
||||||
|
if (!service_fns.loaded)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
if (service_status.dwCurrentState == SERVICE_STOP_PENDING) {
|
||||||
|
service_status.dwWin32ExitCode = 0;
|
||||||
|
service_status.dwCurrentState = SERVICE_STOPPED;
|
||||||
|
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
||||||
|
return 1;
|
||||||
|
} else if (service_status.dwCurrentState == SERVICE_STOPPED) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** DOCDOC */
|
||||||
|
int
|
||||||
|
nt_service_set_state(DWORD state)
|
||||||
|
{
|
||||||
|
service_status.dwCurrentState = state;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Handles service control requests, such as stopping or starting the
|
||||||
|
* Tor service. */
|
||||||
|
static void
|
||||||
|
nt_service_control(DWORD request)
|
||||||
|
{
|
||||||
|
static struct timeval exit_now;
|
||||||
|
exit_now.tv_sec = 0;
|
||||||
|
exit_now.tv_usec = 0;
|
||||||
|
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
|
||||||
|
switch (request) {
|
||||||
|
case SERVICE_CONTROL_STOP:
|
||||||
|
case SERVICE_CONTROL_SHUTDOWN:
|
||||||
|
log_notice(LD_GENERAL,
|
||||||
|
"Got stop/shutdown request; shutting down cleanly.");
|
||||||
|
service_status.dwCurrentState = SERVICE_STOP_PENDING;
|
||||||
|
event_loopexit(&exit_now);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Called when the service is started via the system's service control
|
||||||
|
* manager. This calls tor_init() and starts the main event loop. If
|
||||||
|
* tor_init() fails, the service will be stopped and exit code set to
|
||||||
|
* NT_SERVICE_ERROR_TORINIT_FAILED. */
|
||||||
|
static void
|
||||||
|
nt_service_body(int argc, char **argv)
|
||||||
|
{
|
||||||
|
int r;
|
||||||
|
(void) argc; /* unused */
|
||||||
|
(void) argv; /* unused */
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
||||||
|
service_status.dwCurrentState = SERVICE_START_PENDING;
|
||||||
|
service_status.dwControlsAccepted =
|
||||||
|
SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
|
||||||
|
service_status.dwWin32ExitCode = 0;
|
||||||
|
service_status.dwServiceSpecificExitCode = 0;
|
||||||
|
service_status.dwCheckPoint = 0;
|
||||||
|
service_status.dwWaitHint = 1000;
|
||||||
|
hStatus = service_fns.RegisterServiceCtrlHandlerA_fn(GENSRV_SERVICENAME,
|
||||||
|
(LPHANDLER_FUNCTION) nt_service_control);
|
||||||
|
|
||||||
|
if (hStatus == 0) {
|
||||||
|
/* Failed to register the service control handler function */
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
r = tor_init(backup_argc, backup_argv);
|
||||||
|
if (r) {
|
||||||
|
/* Failed to start the Tor service */
|
||||||
|
r = NT_SERVICE_ERROR_TORINIT_FAILED;
|
||||||
|
service_status.dwCurrentState = SERVICE_STOPPED;
|
||||||
|
service_status.dwWin32ExitCode = r;
|
||||||
|
service_status.dwServiceSpecificExitCode = r;
|
||||||
|
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Set the service's status to SERVICE_RUNNING and start the main
|
||||||
|
* event loop */
|
||||||
|
service_status.dwCurrentState = SERVICE_RUNNING;
|
||||||
|
service_fns.SetServiceStatus_fn(hStatus, &service_status);
|
||||||
|
do_main_loop();
|
||||||
|
tor_cleanup();
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Main service entry point. Starts the service control dispatcher and waits
|
||||||
|
* until the service status is set to SERVICE_STOPPED. */
|
||||||
|
static void
|
||||||
|
nt_service_main(void)
|
||||||
|
{
|
||||||
|
SERVICE_TABLE_ENTRY table[2];
|
||||||
|
DWORD result = 0;
|
||||||
|
char *errmsg;
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
table[0].lpServiceName = (char*)GENSRV_SERVICENAME;
|
||||||
|
table[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)nt_service_body;
|
||||||
|
table[1].lpServiceName = NULL;
|
||||||
|
table[1].lpServiceProc = NULL;
|
||||||
|
|
||||||
|
if (!service_fns.StartServiceCtrlDispatcherA_fn(table)) {
|
||||||
|
result = GetLastError();
|
||||||
|
errmsg = nt_strerror(result);
|
||||||
|
printf("Service error %d : %s\n", (int) result, errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
if (result == ERROR_FAILED_SERVICE_CONTROLLER_CONNECT) {
|
||||||
|
if (tor_init(backup_argc, backup_argv) < 0)
|
||||||
|
return;
|
||||||
|
switch (get_options()->command) {
|
||||||
|
case CMD_RUN_TOR:
|
||||||
|
do_main_loop();
|
||||||
|
break;
|
||||||
|
case CMD_LIST_FINGERPRINT:
|
||||||
|
do_list_fingerprint();
|
||||||
|
break;
|
||||||
|
case CMD_HASH_PASSWORD:
|
||||||
|
do_hash_password();
|
||||||
|
break;
|
||||||
|
case CMD_VERIFY_CONFIG:
|
||||||
|
printf("Configuration was valid\n");
|
||||||
|
break;
|
||||||
|
case CMD_RUN_UNITTESTS:
|
||||||
|
default:
|
||||||
|
log_err(LD_CONFIG, "Illegal command number %d: internal error.",
|
||||||
|
get_options()->command);
|
||||||
|
}
|
||||||
|
tor_cleanup();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Return a handle to the service control manager on success, or NULL on
|
||||||
|
* failure. */
|
||||||
|
static SC_HANDLE
|
||||||
|
nt_service_open_scm(void)
|
||||||
|
{
|
||||||
|
SC_HANDLE hSCManager;
|
||||||
|
char *errmsg = NULL;
|
||||||
|
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
if ((hSCManager = service_fns.OpenSCManagerA_fn(
|
||||||
|
NULL, NULL, SC_MANAGER_CREATE_SERVICE)) == NULL) {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("OpenSCManager() failed : %s\n", errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
}
|
||||||
|
return hSCManager;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Open a handle to the Tor service using <b>hSCManager</b>. Return NULL
|
||||||
|
* on failure. */
|
||||||
|
static SC_HANDLE
|
||||||
|
nt_service_open(SC_HANDLE hSCManager)
|
||||||
|
{
|
||||||
|
SC_HANDLE hService;
|
||||||
|
char *errmsg = NULL;
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
if ((hService = service_fns.OpenServiceA_fn(hSCManager, GENSRV_SERVICENAME,
|
||||||
|
SERVICE_ALL_ACCESS)) == NULL) {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("OpenService() failed : %s\n", errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
}
|
||||||
|
return hService;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Start the Tor service. Return 0 if the service is started or was
|
||||||
|
* previously running. Return -1 on error. */
|
||||||
|
static int
|
||||||
|
nt_service_start(SC_HANDLE hService)
|
||||||
|
{
|
||||||
|
char *errmsg = NULL;
|
||||||
|
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
|
||||||
|
service_fns.QueryServiceStatus_fn(hService, &service_status);
|
||||||
|
if (service_status.dwCurrentState == SERVICE_RUNNING) {
|
||||||
|
printf("Service is already running\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (service_fns.StartServiceA_fn(hService, 0, NULL)) {
|
||||||
|
/* Loop until the service has finished attempting to start */
|
||||||
|
while (service_fns.QueryServiceStatus_fn(hService, &service_status) &&
|
||||||
|
(service_status.dwCurrentState == SERVICE_START_PENDING)) {
|
||||||
|
Sleep(500);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Check if it started successfully or not */
|
||||||
|
if (service_status.dwCurrentState == SERVICE_RUNNING) {
|
||||||
|
printf("Service started successfully\n");
|
||||||
|
return 0;
|
||||||
|
} else {
|
||||||
|
errmsg = nt_strerror(service_status.dwWin32ExitCode);
|
||||||
|
printf("Service failed to start : %s\n", errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("StartService() failed : %s\n", errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
}
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Stop the Tor service. Return 0 if the service is stopped or was not
|
||||||
|
* previously running. Return -1 on error. */
|
||||||
|
static int
|
||||||
|
nt_service_stop(SC_HANDLE hService)
|
||||||
|
{
|
||||||
|
/** Wait at most 10 seconds for the service to stop. */
|
||||||
|
#define MAX_SERVICE_WAIT_TIME 10
|
||||||
|
int wait_time;
|
||||||
|
char *errmsg = NULL;
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
|
||||||
|
service_fns.QueryServiceStatus_fn(hService, &service_status);
|
||||||
|
if (service_status.dwCurrentState == SERVICE_STOPPED) {
|
||||||
|
printf("Service is already stopped\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (service_fns.ControlService_fn(hService, SERVICE_CONTROL_STOP,
|
||||||
|
&service_status)) {
|
||||||
|
wait_time = 0;
|
||||||
|
while (service_fns.QueryServiceStatus_fn(hService, &service_status) &&
|
||||||
|
(service_status.dwCurrentState != SERVICE_STOPPED) &&
|
||||||
|
(wait_time < MAX_SERVICE_WAIT_TIME)) {
|
||||||
|
Sleep(1000);
|
||||||
|
wait_time++;
|
||||||
|
}
|
||||||
|
if (service_status.dwCurrentState == SERVICE_STOPPED) {
|
||||||
|
printf("Service stopped successfully\n");
|
||||||
|
return 0;
|
||||||
|
} else if (wait_time == MAX_SERVICE_WAIT_TIME) {
|
||||||
|
printf("Service did not stop within %d seconds.\n", wait_time);
|
||||||
|
} else {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("QueryServiceStatus() failed : %s\n",errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("ControlService() failed : %s\n", errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
}
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Build a formatted command line used for the NT service. Return a
|
||||||
|
* pointer to the formatted string on success, or NULL on failure. Set
|
||||||
|
* *<b>using_default_torrc</b> to true if we're going to use the default
|
||||||
|
* location to torrc, or 1 if an option was specified on the command line.
|
||||||
|
*/
|
||||||
|
static char *
|
||||||
|
nt_service_command_line(int *using_default_torrc)
|
||||||
|
{
|
||||||
|
TCHAR tor_exe[MAX_PATH+1];
|
||||||
|
char *command, *options=NULL;
|
||||||
|
smartlist_t *sl;
|
||||||
|
int i, cmdlen;
|
||||||
|
*using_default_torrc = 1;
|
||||||
|
|
||||||
|
/* Get the location of tor.exe */
|
||||||
|
if (0 == GetModuleFileName(NULL, tor_exe, MAX_PATH))
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
/* Get the service arguments */
|
||||||
|
sl = smartlist_create();
|
||||||
|
for (i = 1; i < backup_argc; ++i) {
|
||||||
|
if (!strcmp(backup_argv[i], "--options") ||
|
||||||
|
!strcmp(backup_argv[i], "-options")) {
|
||||||
|
while (++i < backup_argc) {
|
||||||
|
if (!strcmp(backup_argv[i], "-f"))
|
||||||
|
*using_default_torrc = 0;
|
||||||
|
smartlist_add(sl, backup_argv[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (smartlist_len(sl))
|
||||||
|
options = smartlist_join_strings(sl,"\" \"",0,NULL);
|
||||||
|
smartlist_free(sl);
|
||||||
|
|
||||||
|
/* Allocate a string for the NT service command line */
|
||||||
|
cmdlen = strlen(tor_exe) + (options?strlen(options):0) + 32;
|
||||||
|
command = tor_malloc(cmdlen);
|
||||||
|
|
||||||
|
/* Format the service command */
|
||||||
|
if (options) {
|
||||||
|
if (tor_snprintf(command, cmdlen, "\"%s\" --nt-service \"%s\"",
|
||||||
|
tor_exe, options)<0) {
|
||||||
|
tor_free(command); /* sets command to NULL. */
|
||||||
|
}
|
||||||
|
} else { /* ! options */
|
||||||
|
if (tor_snprintf(command, cmdlen, "\"%s\" --nt-service", tor_exe)<0) {
|
||||||
|
tor_free(command); /* sets command to NULL. */
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tor_free(options);
|
||||||
|
return command;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Creates a Tor NT service, set to start on boot. The service will be
|
||||||
|
* started if installation succeeds. Returns 0 on success, or -1 on
|
||||||
|
* failure. */
|
||||||
|
static int
|
||||||
|
nt_service_install(int argc, char **argv)
|
||||||
|
{
|
||||||
|
/* Notes about developing NT services:
|
||||||
|
*
|
||||||
|
* 1. Don't count on your CWD. If an absolute path is not given, the
|
||||||
|
* fopen() function goes wrong.
|
||||||
|
* 2. The parameters given to the nt_service_body() function differ
|
||||||
|
* from those given to main() function.
|
||||||
|
*/
|
||||||
|
|
||||||
|
SC_HANDLE hSCManager = NULL;
|
||||||
|
SC_HANDLE hService = NULL;
|
||||||
|
SERVICE_DESCRIPTION sdBuff;
|
||||||
|
char *command;
|
||||||
|
char *errmsg;
|
||||||
|
const char *user_acct = GENSRV_USERACCT;
|
||||||
|
const char *password = "";
|
||||||
|
int i;
|
||||||
|
OSVERSIONINFOEX info;
|
||||||
|
SID_NAME_USE sidUse;
|
||||||
|
DWORD sidLen = 0, domainLen = 0;
|
||||||
|
int is_win2k_or_worse = 0;
|
||||||
|
int using_default_torrc = 0;
|
||||||
|
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
|
||||||
|
/* Open the service control manager so we can create a new service */
|
||||||
|
if ((hSCManager = nt_service_open_scm()) == NULL)
|
||||||
|
return -1;
|
||||||
|
/* Build the command line used for the service */
|
||||||
|
if ((command = nt_service_command_line(&using_default_torrc)) == NULL) {
|
||||||
|
printf("Unable to build service command line.\n");
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i=1; i < argc; ++i) {
|
||||||
|
if (!strcmp(argv[i], "--user") && i+1<argc) {
|
||||||
|
user_acct = argv[i+1];
|
||||||
|
++i;
|
||||||
|
}
|
||||||
|
if (!strcmp(argv[i], "--password") && i+1<argc) {
|
||||||
|
password = argv[i+1];
|
||||||
|
++i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Compute our version and see whether we're running win2k or earlier. */
|
||||||
|
memset(&info, 0, sizeof(info));
|
||||||
|
info.dwOSVersionInfoSize = sizeof(info);
|
||||||
|
if (! GetVersionEx((LPOSVERSIONINFO)&info)) {
|
||||||
|
printf("Call to GetVersionEx failed.\n");
|
||||||
|
is_win2k_or_worse = 1;
|
||||||
|
} else {
|
||||||
|
if (info.dwMajorVersion < 5 ||
|
||||||
|
(info.dwMajorVersion == 5 && info.dwMinorVersion == 0))
|
||||||
|
is_win2k_or_worse = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (user_acct == GENSRV_USERACCT) {
|
||||||
|
if (is_win2k_or_worse) {
|
||||||
|
/* On Win2k, there is no LocalService account, so we actually need to
|
||||||
|
* fall back on NULL (the system account). */
|
||||||
|
printf("Running on Win2K or earlier, so the LocalService account "
|
||||||
|
"doesn't exist. Falling back to SYSTEM account.\n");
|
||||||
|
user_acct = NULL;
|
||||||
|
} else {
|
||||||
|
/* Genericity is apparently _so_ last year in Redmond, where some
|
||||||
|
* accounts are accounts that you can look up, and some accounts
|
||||||
|
* are magic and undetectable via the security subsystem. See
|
||||||
|
* http://msdn2.microsoft.com/en-us/library/ms684188.aspx
|
||||||
|
*/
|
||||||
|
printf("Running on a Post-Win2K OS, so we'll assume that the "
|
||||||
|
"LocalService account exists.\n");
|
||||||
|
}
|
||||||
|
} else if (0 && service_fns.LookupAccountNameA_fn(NULL, // On this system
|
||||||
|
user_acct,
|
||||||
|
NULL, &sidLen, // Don't care about the SID
|
||||||
|
NULL, &domainLen, // Don't care about the domain
|
||||||
|
&sidUse) == 0) {
|
||||||
|
/* XXXX For some reason, the above test segfaults. Fix that. */
|
||||||
|
printf("User \"%s\" doesn't seem to exist.\n", user_acct);
|
||||||
|
return -1;
|
||||||
|
} else {
|
||||||
|
printf("Will try to install service as user \"%s\".\n", user_acct);
|
||||||
|
}
|
||||||
|
/* XXXX This warning could be better about explaining how to resolve the
|
||||||
|
* situation. */
|
||||||
|
if (using_default_torrc)
|
||||||
|
printf("IMPORTANT NOTE:\n"
|
||||||
|
" The Tor service will run under the account \"%s\". This means\n"
|
||||||
|
" that Tor will look for its configuration file under that\n"
|
||||||
|
" account's Application Data directory, which is probably not\n"
|
||||||
|
" the same as yours.\n", user_acct?user_acct:"<local system>");
|
||||||
|
|
||||||
|
/* Create the Tor service, set to auto-start on boot */
|
||||||
|
if ((hService = service_fns.CreateServiceA_fn(hSCManager, GENSRV_SERVICENAME,
|
||||||
|
GENSRV_DISPLAYNAME,
|
||||||
|
SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
|
||||||
|
SERVICE_AUTO_START, SERVICE_ERROR_IGNORE,
|
||||||
|
command, NULL, NULL, NULL,
|
||||||
|
user_acct, password)) == NULL) {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("CreateService() failed : %s\n", errmsg);
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
tor_free(command);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
printf("Done with CreateService.\n");
|
||||||
|
|
||||||
|
/* Set the service's description */
|
||||||
|
sdBuff.lpDescription = (char*)GENSRV_DESCRIPTION;
|
||||||
|
service_fns.ChangeServiceConfig2A_fn(hService, SERVICE_CONFIG_DESCRIPTION,
|
||||||
|
&sdBuff);
|
||||||
|
printf("Service installed successfully\n");
|
||||||
|
|
||||||
|
/* Start the service initially */
|
||||||
|
nt_service_start(hService);
|
||||||
|
|
||||||
|
service_fns.CloseServiceHandle_fn(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
tor_free(command);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Removes the Tor NT service. Returns 0 if the service was successfully
|
||||||
|
* removed, or -1 on error. */
|
||||||
|
static int
|
||||||
|
nt_service_remove(void)
|
||||||
|
{
|
||||||
|
SC_HANDLE hSCManager = NULL;
|
||||||
|
SC_HANDLE hService = NULL;
|
||||||
|
char *errmsg;
|
||||||
|
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
if ((hSCManager = nt_service_open_scm()) == NULL)
|
||||||
|
return -1;
|
||||||
|
if ((hService = nt_service_open(hSCManager)) == NULL) {
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
nt_service_stop(hService);
|
||||||
|
if (service_fns.DeleteService_fn(hService) == FALSE) {
|
||||||
|
errmsg = nt_strerror(GetLastError());
|
||||||
|
printf("DeleteService() failed : %s\n", errmsg);
|
||||||
|
LocalFree(errmsg);
|
||||||
|
service_fns.CloseServiceHandle_fn(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
service_fns.CloseServiceHandle_fn(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
printf("Service removed successfully\n");
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Starts the Tor service. Returns 0 on success, or -1 on error. */
|
||||||
|
static int
|
||||||
|
nt_service_cmd_start(void)
|
||||||
|
{
|
||||||
|
SC_HANDLE hSCManager;
|
||||||
|
SC_HANDLE hService;
|
||||||
|
int start;
|
||||||
|
|
||||||
|
if ((hSCManager = nt_service_open_scm()) == NULL)
|
||||||
|
return -1;
|
||||||
|
if ((hService = nt_service_open(hSCManager)) == NULL) {
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
start = nt_service_start(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
|
||||||
|
return start;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Stops the Tor service. Returns 0 on success, or -1 on error. */
|
||||||
|
static int
|
||||||
|
nt_service_cmd_stop(void)
|
||||||
|
{
|
||||||
|
SC_HANDLE hSCManager;
|
||||||
|
SC_HANDLE hService;
|
||||||
|
int stop;
|
||||||
|
|
||||||
|
if ((hSCManager = nt_service_open_scm()) == NULL)
|
||||||
|
return -1;
|
||||||
|
if ((hService = nt_service_open(hSCManager)) == NULL) {
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
stop = nt_service_stop(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hService);
|
||||||
|
service_fns.CloseServiceHandle_fn(hSCManager);
|
||||||
|
|
||||||
|
return stop;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Given a Win32 error code, this attempts to make Windows
|
||||||
|
* return a human-readable error message. The char* returned
|
||||||
|
* is allocated by Windows, but should be freed with LocalFree()
|
||||||
|
* when finished with it. */
|
||||||
|
static char*
|
||||||
|
nt_strerror(uint32_t errnum)
|
||||||
|
{
|
||||||
|
char *msgbuf;
|
||||||
|
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
|
||||||
|
NULL, errnum, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
||||||
|
(LPSTR)&msgbuf, 0, NULL);
|
||||||
|
return msgbuf;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int
|
||||||
|
nt_service_parse_options(int argc, char **argv)
|
||||||
|
{
|
||||||
|
backup_argv = argv;
|
||||||
|
backup_argc = argc;
|
||||||
|
|
||||||
|
if ((argc >= 3) &&
|
||||||
|
(!strcmp(argv[1], "-service") || !strcmp(argv[1], "--service"))) {
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
if (!strcmp(argv[2], "install"))
|
||||||
|
return nt_service_install(argc, argv);
|
||||||
|
if (!strcmp(argv[2], "remove"))
|
||||||
|
return nt_service_remove();
|
||||||
|
if (!strcmp(argv[2], "start"))
|
||||||
|
return nt_service_cmd_start();
|
||||||
|
if (!strcmp(argv[2], "stop"))
|
||||||
|
return nt_service_cmd_stop();
|
||||||
|
printf("Unrecognized service command '%s'\n", argv[2]);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if (argc >= 2) {
|
||||||
|
if (!strcmp(argv[1], "-nt-service") || !strcmp(argv[1], "--nt-service")) {
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
nt_service_main();
|
||||||
|
return ;
|
||||||
|
}
|
||||||
|
// These values have been deprecated since 0.1.1.2-alpha; we've warned
|
||||||
|
// about them since 0.1.2.7-alpha.
|
||||||
|
if (!strcmp(argv[1], "-install") || !strcmp(argv[1], "--install")) {
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
fprintf(stderr,
|
||||||
|
"The %s option is deprecated; use \"--service install\" instead.",
|
||||||
|
argv[1]);
|
||||||
|
return nt_service_install(argc, argv);
|
||||||
|
}
|
||||||
|
if (!strcmp(argv[1], "-remove") || !strcmp(argv[1], "--remove")) {
|
||||||
|
nt_service_loadlibrary();
|
||||||
|
fprintf(stderr,
|
||||||
|
"The %s option is deprecated; use \"--service remove\" instead.",
|
||||||
|
argv[1]);
|
||||||
|
return nt_service_remove();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
13
src/or/or.h
13
src/or/or.h
@ -3018,6 +3018,19 @@ void tor_free_all(int postfork);
|
|||||||
|
|
||||||
int tor_main(int argc, char *argv[]);
|
int tor_main(int argc, char *argv[]);
|
||||||
|
|
||||||
|
/********************************* ntmain.c ***************************/
|
||||||
|
#ifdef MS_WINDOWS
|
||||||
|
#define NT_SERVICE
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef NT_SERVICE
|
||||||
|
int nt_service_parse_options(int argc, char **argv);
|
||||||
|
int nt_service_is_stopping(void);
|
||||||
|
void nt_service_set_state(DWORD state);
|
||||||
|
#else
|
||||||
|
#define nt_service_is_stopping() (0)
|
||||||
|
#endif
|
||||||
|
|
||||||
/********************************* onion.c ***************************/
|
/********************************* onion.c ***************************/
|
||||||
|
|
||||||
int onion_pending_add(or_circuit_t *circ);
|
int onion_pending_add(or_circuit_t *circ);
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user