tinc/src/process.c

608 lines
14 KiB
C
Raw Normal View History

/*
process.c -- process management functions
2005-05-04 18:09:30 +00:00
Copyright (C) 1999-2005 Ivo Timmermans <ivo@tinc-vpn.org>,
2000-2005 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., 675 Mass Ave, Cambridge, MA 02139, USA.
$Id$
*/
#include "system.h"
#include "conf.h"
#include "connection.h"
#include "device.h"
#include "edge.h"
#include "logger.h"
#include "node.h"
#include "pidfile.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 sighup = false;
bool sigalrm = false;
extern char *identname;
extern char *pidfilename;
extern char **g_argv;
2003-07-22 20:55:21 +00:00
extern bool use_logfile;
extern volatile bool running;
sigset_t emptysigset;
static int saved_debug_level = -1;
static void memory_full(int size)
{
logger(LOG_ERR, _("Memory exhausted (couldn't allocate %d bytes), exitting."), size);
2002-09-09 21:25:28 +00:00
cp_trace();
exit(1);
}
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;
bool install_service(void) {
char command[4096] = "\"";
char **argp;
2003-08-08 17:17:13 +00:00
bool space;
SERVICE_DESCRIPTION description = {"Virtual Private Network daemon"};
manager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if(!manager) {
logger(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
for(argp = g_argv + 1; *argp; argp++) {
space = strchr(*argp, ' ');
2005-01-04 22:18:58 +00:00
strncat(command, " ", sizeof command - strlen(command));
2003-08-08 19:56:11 +00:00
if(space)
2005-01-04 22:18:58 +00:00
strncat(command, "\"", sizeof command - strlen(command));
2003-08-08 19:56:11 +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);
if(!service) {
logger(LOG_ERR, _("Could not create %s service: %s"), identname, winerror(GetLastError()));
return false;
}
ChangeServiceConfig2(service, SERVICE_CONFIG_DESCRIPTION, &description);
logger(LOG_INFO, _("%s service installed"), identname);
if(!StartService(service, 0, NULL))
logger(LOG_WARNING, _("Could not start %s service: %s"), identname, winerror(GetLastError()));
else
logger(LOG_INFO, _("%s service started"), identname);
return true;
}
bool remove_service(void) {
manager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if(!manager) {
logger(LOG_ERR, _("Could not open service manager: %s"), winerror(GetLastError()));
return false;
}
service = OpenService(manager, identname, SERVICE_ALL_ACCESS);
if(!service) {
logger(LOG_ERR, _("Could not open %s service: %s"), identname, winerror(GetLastError()));
return false;
}
if(!ControlService(service, SERVICE_CONTROL_STOP, &status))
logger(LOG_ERR, _("Could not stop %s service: %s"), identname, winerror(GetLastError()));
else
logger(LOG_INFO, _("%s service stopped"), identname);
if(!DeleteService(service)) {
logger(LOG_ERR, _("Could not remove %s service: %s"), identname, winerror(GetLastError()));
return false;
}
logger(LOG_INFO, _("%s service removed"), identname);
return true;
}
DWORD WINAPI controlhandler(DWORD request, DWORD type, LPVOID boe, LPVOID bah) {
switch(request) {
case SERVICE_CONTROL_STOP:
logger(LOG_NOTICE, _("Got %s request"), "SERVICE_CONTROL_STOP");
break;
case SERVICE_CONTROL_SHUTDOWN:
logger(LOG_NOTICE, _("Got %s request"), "SERVICE_CONTROL_SHUTDOWN");
break;
default:
logger(LOG_WARNING, _("Got unexpected request %d"), request);
return ERROR_CALL_NOT_IMPLEMENTED;
}
if(running) {
running = false;
2003-08-08 12:55:05 +00:00
status.dwWaitHint = 30000;
2003-08-03 12:38:18 +00:00
status.dwCurrentState = SERVICE_STOP_PENDING;
SetServiceStatus(statushandle, &status);
return NO_ERROR;
} else {
status.dwWaitHint = 0;
status.dwCurrentState = SERVICE_STOPPED;
SetServiceStatus(statushandle, &status);
exit(1);
2003-08-03 12:38:18 +00:00
}
}
VOID WINAPI run_service(DWORD argc, LPTSTR* argv)
{
int err = 1;
extern int main2(int argc, char **argv);
status.dwServiceType = SERVICE_WIN32;
status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
status.dwWin32ExitCode = 0;
status.dwServiceSpecificExitCode = 0;
status.dwCheckPoint = 0;
statushandle = RegisterServiceCtrlHandlerEx(identname, controlhandler, NULL);
2002-09-09 21:25:28 +00:00
if (!statushandle) {
logger(LOG_ERR, _("System call `%s' failed: %s"), "RegisterServiceCtrlHandlerEx", winerror(GetLastError()));
err = 1;
} else {
2003-08-08 12:55:05 +00:00
status.dwWaitHint = 30000;
2003-08-03 12:38:18 +00:00
status.dwCurrentState = SERVICE_START_PENDING;
SetServiceStatus(statushandle, &status);
2003-08-08 12:55:05 +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;
status.dwCurrentState = SERVICE_STOPPED;
2003-08-03 12:38:18 +00:00
//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(LOG_ERR, _("System call `%s' failed: %s"), "StartServiceCtrlDispatcher", winerror(GetLastError()));
}
return true;
}
#endif
2003-07-28 22:06:09 +00:00
#ifndef HAVE_MINGW
/*
check for an existing tinc for this net, and write pid to pidfile
*/
2003-07-22 20:55:21 +00:00
static bool write_pidfile(void)
{
pid_t pid;
2002-09-09 21:25:28 +00:00
cp();
pid = check_pid(pidfilename);
if(pid) {
if(netname)
fprintf(stderr, _("A tincd is already running for net `%s' with pid %ld.\n"),
netname, (long)pid);
2002-09-09 21:25:28 +00:00
else
fprintf(stderr, _("A tincd is already running with pid %ld.\n"), (long)pid);
2003-07-22 20:55:21 +00:00
return false;
2002-09-09 21:25:28 +00:00
}
/* if it's locked, write-protected, or whatever */
if(!write_pid(pidfilename)) {
fprintf(stderr, _("Could write pid file %s: %s\n"), pidfilename, strerror(errno));
2003-07-22 20:55:21 +00:00
return false;
}
2002-09-09 21:25:28 +00:00
2003-07-22 20:55:21 +00:00
return true;
}
2003-07-28 22:06:09 +00:00
#endif
/*
kill older tincd for this net
*/
2003-07-22 20:55:21 +00:00
bool kill_other(int signal)
{
2003-07-28 22:06:09 +00:00
#ifndef HAVE_MINGW
pid_t pid;
2002-09-09 21:25:28 +00:00
cp();
pid = read_pid(pidfilename);
if(!pid) {
if(netname)
fprintf(stderr, _("No other tincd is running for net `%s'.\n"),
netname);
else
fprintf(stderr, _("No other tincd is running.\n"));
2003-07-22 20:55:21 +00:00
return false;
2002-09-09 21:25:28 +00:00
}
errno = 0; /* No error, sometimes errno is only changed on error */
/* ESRCH is returned when no process with that pid is found */
if(kill(pid, signal) && errno == ESRCH) {
if(netname)
fprintf(stderr, _("The tincd for net `%s' is no longer running. "),
netname);
else
fprintf(stderr, _("The tincd is no longer running. "));
fprintf(stderr, _("Removing stale lock file.\n"));
remove_pid(pidfilename);
}
2003-07-22 20:55:21 +00:00
return true;
#else
return remove_service();
#endif
}
/*
Detach from current terminal, write pidfile, kill parent
*/
2003-07-22 20:55:21 +00:00
bool detach(void)
{
2002-09-09 21:25:28 +00:00
cp();
2002-09-09 21:25:28 +00:00
setup_signals();
2002-09-09 21:25:28 +00:00
/* First check if we can open a fresh new pidfile */
2003-07-28 22:06:09 +00:00
#ifndef HAVE_MINGW
2003-07-22 20:55:21 +00:00
if(!write_pidfile())
return false;
2002-09-09 21:25:28 +00:00
/* If we succeeded in doing that, detach */
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)) {
2002-09-09 21:25:28 +00:00
fprintf(stderr, _("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
}
/* Now UPDATE the pid in the pidfile, because we changed it... */
if(!write_pid(pidfilename)) {
2003-12-07 14:29:02 +00:00
fprintf(stderr, _("Could not write pid file %s: %s\n"), pidfilename, strerror(errno));
2003-07-22 20:55:21 +00:00
return false;
}
#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(LOG_NOTICE, _("tincd %s (%s %s) starting, debug level %d"),
VERSION, __DATE__, __TIME__, debug_level);
2002-09-09 21:25:28 +00:00
xalloc_fail_func = memory_full;
2003-07-22 20:55:21 +00:00
return true;
}
2003-07-22 20:55:21 +00:00
bool execute_script(const char *name, char **envp)
{
#ifdef HAVE_SYSTEM
2003-08-16 12:11:11 +00:00
int status, len;
2002-09-09 21:25:28 +00:00
struct stat s;
char *scriptname, *p;
int i;
2002-09-09 21:25:28 +00:00
cp();
#ifndef HAVE_MINGW
2003-08-16 12:11:11 +00:00
len = asprintf(&scriptname, "\"%s/%s\"", confbase, name);
#else
len = asprintf(&scriptname, "\"%s/%s.bat\"", confbase, name);
#endif
if(len < 0)
return false;
scriptname[len - 1] = '\0';
2002-09-09 21:25:28 +00:00
/* First check if there is a script */
2003-08-16 12:11:11 +00:00
if(stat(scriptname + 1, &s))
2003-07-22 20:55:21 +00:00
return true;
2002-09-09 21:25:28 +00:00
ifdebug(STATUS) logger(LOG_INFO, _("Executing script %s"), name);
2003-08-08 19:56:11 +00:00
#ifdef HAVE_PUTENV
/* Set environment */
for(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
2003-08-16 12:11:11 +00:00
scriptname[len - 1] = '\"';
status = system(scriptname);
free(scriptname);
/* Unset environment */
for(i = 0; envp[i]; i++) {
char *e = strchr(envp[i], '=');
if(e) {
p = alloca(e - envp[i] + 1);
strncpy(p, envp[i], e - envp[i]);
p[e - envp[i]] = '\0';
putenv(p);
}
}
#ifdef WEXITSTATUS
if(status != -1) {
if(WIFEXITED(status)) { /* Child exited by itself */
if(WEXITSTATUS(status)) {
logger(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
}
} else if(WIFSIGNALED(status)) { /* Child was killed by a signal */
logger(LOG_ERR, _("Script %s was killed by signal %d (%s)"),
name, WTERMSIG(status), strsignal(WTERMSIG(status)));
return false;
} else { /* Something strange happened */
logger(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(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;
}
/*
Signal handlers.
*/
2003-07-28 22:06:09 +00:00
#ifndef HAVE_MINGW
static RETSIGTYPE sigterm_handler(int a)
{
2003-08-03 21:45:13 +00:00
logger(LOG_NOTICE, _("Got %s signal"), "TERM");
2003-12-07 14:29:02 +00:00
if(running)
running = false;
else
exit(1);
}
static RETSIGTYPE sigquit_handler(int a)
{
2003-08-03 21:45:13 +00:00
logger(LOG_NOTICE, _("Got %s signal"), "QUIT");
2003-12-07 14:29:02 +00:00
if(running)
running = false;
else
exit(1);
}
static RETSIGTYPE fatal_signal_square(int a)
{
logger(LOG_ERR, _("Got another fatal signal %d (%s): not restarting."), a,
2002-09-09 21:25:28 +00:00
strsignal(a));
cp_trace();
exit(1);
}
static RETSIGTYPE fatal_signal_handler(int a)
{
2002-09-09 21:25:28 +00:00
struct sigaction act;
logger(LOG_ERR, _("Got fatal signal %d (%s)"), a, strsignal(a));
2002-09-09 21:25:28 +00:00
cp_trace();
if(do_detach) {
logger(LOG_NOTICE, _("Trying to re-execute in 5 seconds..."));
2002-09-09 21:25:28 +00:00
act.sa_handler = fatal_signal_square;
act.sa_mask = emptysigset;
act.sa_flags = 0;
sigaction(SIGSEGV, &act, NULL);
close_network_connections();
sleep(5);
remove_pid(pidfilename);
execvp(g_argv[0], g_argv);
} else {
logger(LOG_NOTICE, _("Not restarting."));
2002-09-09 21:25:28 +00:00
exit(1);
}
}
static RETSIGTYPE sighup_handler(int a)
{
2003-08-03 21:45:13 +00:00
logger(LOG_NOTICE, _("Got %s signal"), "HUP");
2003-07-22 20:55:21 +00:00
sighup = true;
}
static RETSIGTYPE sigint_handler(int a)
{
2003-08-03 21:45:13 +00:00
logger(LOG_NOTICE, _("Got %s signal"), "INT");
if(saved_debug_level != -1) {
logger(LOG_NOTICE, _("Reverting to old debug level (%d)"),
saved_debug_level);
debug_level = saved_debug_level;
saved_debug_level = -1;
2002-09-09 21:25:28 +00:00
} else {
logger(LOG_NOTICE,
_("Temporarily setting debug level to 5. Kill me with SIGINT again to go back to level %d."),
debug_level);
saved_debug_level = debug_level;
debug_level = 5;
2002-09-09 21:25:28 +00:00
}
}
static RETSIGTYPE sigalrm_handler(int a)
2002-02-10 21:57:54 +00:00
{
2003-08-03 21:45:13 +00:00
logger(LOG_NOTICE, _("Got %s signal"), "ALRM");
2003-07-22 20:55:21 +00:00
sigalrm = true;
2002-02-10 21:57:54 +00:00
}
static RETSIGTYPE sigusr1_handler(int a)
{
2002-09-09 21:25:28 +00:00
dump_connections();
}
static RETSIGTYPE sigusr2_handler(int a)
{
2002-09-09 21:25:28 +00:00
dump_device_stats();
dump_nodes();
dump_edges();
dump_subnets();
}
static RETSIGTYPE sigwinch_handler(int a)
2002-02-10 21:57:54 +00:00
{
2003-07-22 20:55:21 +00:00
do_purge = true;
2002-02-10 21:57:54 +00:00
}
static RETSIGTYPE unexpected_signal_handler(int a)
{
logger(LOG_WARNING, _("Got unexpected signal %d (%s)"), a, strsignal(a));
2002-09-09 21:25:28 +00:00
cp_trace();
}
static RETSIGTYPE ignore_signal_handler(int a)
{
ifdebug(SCARY_THINGS) logger(LOG_DEBUG, _("Ignored signal %d (%s)"), a, strsignal(a));
}
static struct {
2002-09-09 21:25:28 +00:00
int signal;
2002-09-09 22:33:31 +00:00
void (*handler)(int);
} sighandlers[] = {
2003-07-22 20:55:21 +00:00
{SIGHUP, sighup_handler},
{SIGTERM, sigterm_handler},
{SIGQUIT, sigquit_handler},
{SIGSEGV, fatal_signal_handler},
{SIGBUS, fatal_signal_handler},
{SIGILL, fatal_signal_handler},
{SIGPIPE, ignore_signal_handler},
{SIGINT, sigint_handler},
{SIGUSR1, sigusr1_handler},
{SIGUSR2, sigusr2_handler},
{SIGCHLD, ignore_signal_handler},
{SIGALRM, sigalrm_handler},
{SIGWINCH, sigwinch_handler},
{0, NULL}
};
2003-07-28 22:06:09 +00:00
#endif
2002-09-09 21:25:28 +00:00
void setup_signals(void)
{
2003-07-28 22:06:09 +00:00
#ifndef HAVE_MINGW
2002-09-09 21:25:28 +00:00
int i;
struct sigaction act;
sigemptyset(&emptysigset);
act.sa_handler = NULL;
act.sa_mask = emptysigset;
act.sa_flags = 0;
/* Set a default signal handler for every signal, errors will be
ignored. */
for(i = 0; i < NSIG; i++) {
if(!do_detach)
act.sa_handler = SIG_DFL;
else
act.sa_handler = unexpected_signal_handler;
sigaction(i, &act, NULL);
}
/* If we didn't detach, allow coredumps */
if(!do_detach)
sighandlers[3].handler = SIG_DFL;
/* Then, for each known signal that we want to catch, assign a
handler to the signal, with error checking this time. */
for(i = 0; sighandlers[i].signal; i++) {
act.sa_handler = sighandlers[i].handler;
if(sigaction(sighandlers[i].signal, &act, NULL) < 0)
fprintf(stderr, _("Installing signal handler for signal %d (%s) failed: %s\n"),
2002-09-09 21:25:28 +00:00
sighandlers[i].signal, strsignal(sighandlers[i].signal),
strerror(errno));
}
2003-07-28 22:06:09 +00:00
#endif
}