tinc/src/process.c

293 lines
7.8 KiB
C
Raw Normal View History

/*
process.c -- process management functions
Copyright (C) 1999-2005 Ivo Timmermans,
2000-2011 Guus Sliepen <guus@tinc-vpn.org>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include "system.h"
#include "conf.h"
#include "connection.h"
#include "control.h"
#include "device.h"
#include "edge.h"
#include "logger.h"
#include "net.h"
#include "node.h"
#include "process.h"
#include "subnet.h"
#include "utils.h"
#include "xalloc.h"
/* If zero, don't detach from the terminal. */
2003-07-22 20:55:21 +00:00
bool do_detach = true;
bool sigalrm = false;
extern char *identname;
extern char **g_argv;
2003-07-22 20:55:21 +00:00
extern bool use_logfile;
2000-11-22 22:05:37 +00:00
/* Some functions the less gifted operating systems might lack... */
#ifdef HAVE_MINGW
extern char *identname;
extern char *program_name;
extern char **g_argv;
static SC_HANDLE manager = NULL;
static SC_HANDLE service = NULL;
static SERVICE_STATUS status = {0};
static SERVICE_STATUS_HANDLE statushandle = 0;
static bool install_service(void) {
char command[4096] = "\"";
SERVICE_DESCRIPTION description = {"Virtual Private Network daemon"};
manager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if(!manager) {
logger(DEBUG_ALWAYS, LOG_ERR, "Could not open service manager: %s", winerror(GetLastError()));
return false;
}
if(!strchr(program_name, '\\')) {
2005-01-04 22:18:58 +00:00
GetCurrentDirectory(sizeof command - 1, command + 1);
strncat(command, "\\", sizeof command - strlen(command));
}
2005-01-04 22:18:58 +00:00
strncat(command, program_name, sizeof command - strlen(command));
2003-08-16 12:11:11 +00:00
2005-01-04 22:18:58 +00:00
strncat(command, "\"", sizeof command - strlen(command));
2003-08-16 12:11:11 +00:00
2012-10-07 22:35:38 +00:00
for(char **argp = g_argv + 1; *argp; argp++) {
2012-10-14 14:39:16 +00:00
char *space = strchr(*argp, ' ');
2005-01-04 22:18:58 +00:00
strncat(command, " ", sizeof command - strlen(command));
2012-10-10 15:17:49 +00:00
2003-08-08 19:56:11 +00:00
if(space)
2005-01-04 22:18:58 +00:00
strncat(command, "\"", sizeof command - strlen(command));
2012-10-10 15:17:49 +00:00
2005-01-04 22:18:58 +00:00
strncat(command, *argp, sizeof command - strlen(command));
2003-08-08 19:56:11 +00:00
2003-08-08 17:17:13 +00:00
if(space)
2005-01-04 22:18:58 +00:00
strncat(command, "\"", sizeof command - strlen(command));
}
service = CreateService(manager, identname, identname,
SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, SERVICE_AUTO_START, SERVICE_ERROR_NORMAL,
command, NULL, NULL, NULL, NULL, NULL);
2012-10-10 15:17:49 +00:00
if(!service) {
DWORD lasterror = GetLastError();
logger(DEBUG_ALWAYS, LOG_ERR, "Could not create %s service: %s", identname, winerror(lasterror));
if(lasterror != ERROR_SERVICE_EXISTS)
return false;
}
if(service) {
ChangeServiceConfig2(service, SERVICE_CONFIG_DESCRIPTION, &description);
logger(DEBUG_ALWAYS, LOG_INFO, "%s service installed", identname);
} else {
service = OpenService(manager, identname, SERVICE_ALL_ACCESS);
}
if(!StartService(service, 0, NULL))
logger(DEBUG_ALWAYS, LOG_WARNING, "Could not start %s service: %s", identname, winerror(GetLastError()));
else
logger(DEBUG_ALWAYS, LOG_INFO, "%s service started", identname);
return true;
}
DWORD WINAPI controlhandler(DWORD request, DWORD type, LPVOID boe, LPVOID bah) {
switch(request) {
case SERVICE_CONTROL_INTERROGATE:
SetServiceStatus(statushandle, &status);
return NO_ERROR;
case SERVICE_CONTROL_STOP:
logger(DEBUG_ALWAYS, LOG_NOTICE, "Got %s request", "SERVICE_CONTROL_STOP");
break;
case SERVICE_CONTROL_SHUTDOWN:
logger(DEBUG_ALWAYS, LOG_NOTICE, "Got %s request", "SERVICE_CONTROL_SHUTDOWN");
break;
default:
logger(DEBUG_ALWAYS, LOG_WARNING, "Got unexpected request %d", request);
return ERROR_CALL_NOT_IMPLEMENTED;
}
event_loopexit(NULL);
2012-10-10 15:17:49 +00:00
status.dwWaitHint = 30000;
status.dwCurrentState = SERVICE_STOP_PENDING;
SetServiceStatus(statushandle, &status);
return NO_ERROR;
}
2007-05-18 10:00:00 +00:00
VOID WINAPI run_service(DWORD argc, LPTSTR* argv) {
int err = 1;
extern int main2(int argc, char **argv);
2012-10-10 15:17:49 +00:00
status.dwServiceType = SERVICE_WIN32;
status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
2012-10-10 15:17:49 +00:00
status.dwWin32ExitCode = 0;
status.dwServiceSpecificExitCode = 0;
status.dwCheckPoint = 0;
2012-10-10 15:17:49 +00:00
statushandle = RegisterServiceCtrlHandlerEx(identname, controlhandler, NULL);
2002-09-09 21:25:28 +00:00
if (!statushandle) {
logger(DEBUG_ALWAYS, LOG_ERR, "System call `%s' failed: %s", "RegisterServiceCtrlHandlerEx", winerror(GetLastError()));
err = 1;
} else {
2012-10-10 15:17:49 +00:00
status.dwWaitHint = 30000;
status.dwCurrentState = SERVICE_START_PENDING;
2003-08-03 12:38:18 +00:00
SetServiceStatus(statushandle, &status);
2012-10-10 15:17:49 +00:00
status.dwWaitHint = 0;
2003-08-03 12:38:18 +00:00
status.dwCurrentState = SERVICE_RUNNING;
SetServiceStatus(statushandle, &status);
err = main2(argc, argv);
2003-08-08 12:55:05 +00:00
status.dwWaitHint = 0;
2012-10-10 15:17:49 +00:00
status.dwCurrentState = SERVICE_STOPPED;
//status.dwWin32ExitCode = err;
SetServiceStatus(statushandle, &status);
}
return;
}
bool init_service(void) {
SERVICE_TABLE_ENTRY services[] = {
{identname, run_service},
{NULL, NULL}
};
if(!StartServiceCtrlDispatcher(services)) {
if(GetLastError() == ERROR_FAILED_SERVICE_CONTROLLER_CONNECT) {
return false;
}
else
logger(DEBUG_ALWAYS, LOG_ERR, "System call `%s' failed: %s", "StartServiceCtrlDispatcher", winerror(GetLastError()));
}
return true;
}
#endif
/*
Detach from current terminal
*/
2007-05-18 10:00:00 +00:00
bool detach(void) {
2003-07-28 22:06:09 +00:00
#ifndef HAVE_MINGW
signal(SIGPIPE, SIG_IGN);
signal(SIGUSR1, SIG_IGN);
signal(SIGUSR2, SIG_IGN);
signal(SIGWINCH, SIG_IGN);
2011-06-02 20:14:53 +00:00
2003-07-21 14:47:43 +00:00
closelogger();
#endif
2002-09-09 21:25:28 +00:00
if(do_detach) {
#ifndef HAVE_MINGW
2003-07-22 20:55:21 +00:00
if(daemon(0, 0)) {
2012-10-14 13:37:24 +00:00
logger(DEBUG_ALWAYS, LOG_ERR, "Couldn't detach from terminal: %s", strerror(errno));
2003-07-22 20:55:21 +00:00
return false;
2002-09-09 21:25:28 +00:00
}
#else
if(!statushandle)
exit(!install_service());
2003-07-28 22:06:09 +00:00
#endif
}
openlogger(identname, use_logfile?LOGMODE_FILE:(do_detach?LOGMODE_SYSLOG:LOGMODE_STDERR));
2002-09-09 21:25:28 +00:00
logger(DEBUG_ALWAYS, LOG_NOTICE, "tincd %s (%s %s) starting, debug level %d",
VERSION, __DATE__, __TIME__, debug_level);
2002-09-09 21:25:28 +00:00
2003-07-22 20:55:21 +00:00
return true;
}
2007-05-18 10:00:00 +00:00
bool execute_script(const char *name, char **envp) {
#ifdef HAVE_SYSTEM
2010-11-16 15:45:36 +00:00
char *scriptname;
2012-10-07 15:53:23 +00:00
char *command;
2002-09-09 21:25:28 +00:00
2012-10-07 15:53:23 +00:00
xasprintf(&scriptname, "%s" SLASH "%s%s", confbase, name, scriptextension);
2003-08-16 12:11:11 +00:00
2002-09-09 21:25:28 +00:00
/* First check if there is a script */
2012-10-07 15:53:23 +00:00
if(access(scriptname, F_OK)) {
free(scriptname);
2003-07-22 20:55:21 +00:00
return true;
}
2002-09-09 21:25:28 +00:00
logger(DEBUG_STATUS, LOG_INFO, "Executing script %s", name);
2003-08-08 19:56:11 +00:00
#ifdef HAVE_PUTENV
/* Set environment */
2012-10-10 15:17:49 +00:00
2012-10-07 15:53:23 +00:00
for(int i = 0; envp[i]; i++)
putenv(envp[i]);
2003-08-08 19:56:11 +00:00
#endif
2002-09-09 21:25:28 +00:00
2012-10-07 15:53:23 +00:00
if(scriptinterpreter)
xasprintf(&command, "%s \"%s\"", scriptinterpreter, scriptname);
else
xasprintf(&command, "\"%s\"", scriptname);
int status = system(command);
2012-10-07 15:53:23 +00:00
free(command);
free(scriptname);
/* Unset environment */
2012-10-07 15:53:23 +00:00
for(int i = 0; envp[i]; i++) {
char *e = strchr(envp[i], '=');
if(e) {
2010-11-16 15:45:36 +00:00
char p[e - envp[i] + 1];
strncpy(p, envp[i], e - envp[i]);
p[e - envp[i]] = '\0';
putenv(p);
}
}
#ifdef WEXITSTATUS
if(status != -1) {
2012-10-10 15:17:49 +00:00
if(WIFEXITED(status)) { /* Child exited by itself */
if(WEXITSTATUS(status)) {
logger(DEBUG_ALWAYS, LOG_ERR, "Script %s exited with non-zero status %d",
name, WEXITSTATUS(status));
2003-07-22 20:55:21 +00:00
return false;
2002-09-09 21:25:28 +00:00
}
2012-10-10 15:17:49 +00:00
} else if(WIFSIGNALED(status)) { /* Child was killed by a signal */
logger(DEBUG_ALWAYS, LOG_ERR, "Script %s was killed by signal %d (%s)",
name, WTERMSIG(status), strsignal(WTERMSIG(status)));
return false;
2012-10-10 15:17:49 +00:00
} else { /* Something strange happened */
logger(DEBUG_ALWAYS, LOG_ERR, "Script %s terminated abnormally", name);
2003-07-22 20:55:21 +00:00
return false;
2002-09-09 21:25:28 +00:00
}
} else {
logger(DEBUG_ALWAYS, LOG_ERR, "System call `%s' failed: %s", "system", strerror(errno));
return false;
2002-09-09 21:25:28 +00:00
}
#endif
2003-07-28 22:06:09 +00:00
#endif
return true;
}