/*
 * Soft:        Keepalived is a failover program for the LVS project
 *              <www.linuxvirtualserver.org>. It monitor & manipulate
 *              a loadbalanced server pool using multi-layer checks.
 *
 * Part:        Forked system call to launch an extra script.
 *
 * Author:      Alexandre Cassen, <acassen@linux-vs.org>
 *
 *              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.
 *
 *              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.
 *
 * Copyright (C) 2001-2017 Alexandre Cassen, <acassen@gmail.com>
 */

#include "config.h"

#ifdef HAVE_CLOSE_RANGE
#if !defined USE_CLOSE_RANGE_SYSCALL && !defined _GNU_SOURCE
#define _GNU_SOURCE         /* for close_range() in unistd.h */
#endif
#include <linux/close_range.h>
#endif
#include <unistd.h>
#include <stdlib.h>
#include <fcntl.h>
#include <errno.h>
#include <signal.h>
#include <grp.h>
#include <string.h>
#include <sys/stat.h>
#include <pwd.h>
#include <sys/resource.h>
#include <limits.h>
#include <sys/prctl.h>

#include "notify.h"
#include "signals.h"
#include "logger.h"
#include "utils.h"
#include "process.h"
#include "parser.h"
#include "keepalived_magic.h"
#include "scheduler.h"


/* Save our uid/gid */
uid_t our_uid;
gid_t our_gid;

/* Default user/group for script execution */
static uid_t default_script_uid;
static gid_t default_script_gid;

/* Have we got a default user OK? */
static bool default_script_uid_set = false;
static bool default_user_fail = false;

/* Script security enabled */
bool script_security = false;

/* Buffer length needed for getpwnam_r/getgrname_r */
static size_t getpwnam_buf_len;

static char *path;
static bool path_is_malloced;
static bool use_symlinks;


/* Buffer for expanding notify script commands */
static char cmd_str_buf[MAXBUF];

void
set_symlinks(bool state)
{
	use_symlinks = state;
}

static bool
set_script_env(uid_t uid, gid_t gid)
{
	if (gid != our_gid) {
		if (setgid(gid) < 0) {
			log_message(LOG_ALERT, "Couldn't setgid: %u (%m)", gid);
			return true;
		}
	}

	/* Clear any extra supplementary groups */
	if (setgroups(1, &gid) < 0) {
		log_message(LOG_ALERT, "Couldn't setgroups: %u (%m)", gid);
		return true;
	}

	if (uid != our_uid) {
		if (setuid(uid) < 0) {
			log_message(LOG_ALERT, "Couldn't setuid: %u (%m)", uid);
			return true;
		}
	}

	/* Prepare for invoking process/script */
	signal_handler_script();
	set_std_fd(false);

	return false;
}

const char *
cmd_str_r(const notify_script_t *script, char *buf, size_t len)
{
	char *str_p;
	int i;
	size_t str_len;

	str_p = buf;

	for (i = 0; i < script->num_args; i++) {
		/* Check there is enough room for the next word */
		str_len = strlen(script->args[i]);
		if (str_p + str_len + 2 + (i ? 1 : 0) >= buf + len)
			return NULL;

		if (i)
			*str_p++ = ' ';

		/* Allow special case of bash script which is redirection only to
		 * test for file existence. */
		if (i || (script->args[i][0] != '<' && script->args[i][0] != '>'))
			*str_p++ = '\'';

		strcpy(str_p, script->args[i]);
		str_p += str_len;

		/* Close opening ' if we added one */
		if (i || (script->args[i][0] != '<' && script->args[i][0] != '>'))
			*str_p++ = '\'';
	}
	*str_p = '\0';

	return buf;
}

const char *
cmd_str(const notify_script_t *script)
{
	size_t len;
	int i;

	for (i = 0, len = 0; i < script->num_args; i++)
		len += strlen(script->args[i]) + 3; /* Add two ', and trailing space (or null for last arg) */

	if (len > sizeof cmd_str_buf)
		return NULL;

	return cmd_str_r(script, cmd_str_buf, sizeof cmd_str_buf);
}

int
system_call_script(thread_master_t *m, thread_func_t func, void * arg, unsigned long timer, const notify_script_t* script)
{
	pid_t pid;
	const char *str;
	int retval;
	union non_const_args args;

	/* Daemonization to not degrade our scheduling timer */
#ifdef ENABLE_LOG_TO_FILE
	if (log_file_name)
		flush_log_file();
#endif

	pid = fork();

	if (pid < 0) {
		/* fork error */
		log_message(LOG_INFO, "Failed fork process");
		return -1;
	}

	if (pid) {
		/* parent process */
		if (func) {
			thread_add_child(m, func, arg, pid, timer);
#ifdef _SCRIPT_DEBUG_
			if (do_script_debug)
				log_message(LOG_INFO, "Running script %s with pid %d, timer %lu.%6.6lu", script->args[0], pid, timer / TIMER_HZ, timer % TIMER_HZ);
#endif
		}

		return 0;
	}

	/* Child process */
	reset_process_priorities();

#ifdef _MEM_CHECK_
	skip_mem_dump();
#endif

#ifdef HAVE_CLOSE_RANGE
	 /* We don't want anything past stderr here. This is belt
	  * and braces really, since all file desriptors should
	  * have FD_CLOEXEC set. CLOSE_RANGE_CLOEXEC uses fewer
	  * kernel resources that CLOSE_RANGE_UNSHARE. */
	close_range(STDERR_FILENO + 1, ~0U,
#if HAVE_DECL_CLOSE_RANGE_CLOEXEC
			    CLOSE_RANGE_CLOEXEC
#else
			    CLOSE_RANGE_UNSHARE
#endif
					       );
#endif

	if (set_script_env(script->uid, script->gid))
		exit(0);

	/* Move us into our own process group, so if the script needs to be killed
	 * all its child processes will also be killed. */
	setpgid(0, 0);

	if (script->flags & SC_EXECABLE) {
		/* If keepalived dies, we want the script to die */
		prctl(PR_SET_PDEATHSIG, SIGTERM);

		args.args = script->args;	/* Note: we are casting away constness, since execve parameter type is wrong */
		execve(script->path ? script->path : script->args[0], args.execve_args, environ);

		/* error */
		log_message(LOG_ALERT, "Error exec-ing command '%s', error %d: %m", script->path ? script->path : script->args[0], errno);
	} else {
		retval = system(str = cmd_str(script));

		if (retval == -1) {
			log_message(LOG_ALERT, "Error exec-ing command: %s", str);
			exit(0);
		}

		if (WIFEXITED(retval)) {
			if (WEXITSTATUS(retval) == 127) {
				/* couldn't find command */
				log_message(LOG_ALERT, "Couldn't find command: %s", str);
			}
			else if (WEXITSTATUS(retval) == 126) {
				/* couldn't find command */
				log_message(LOG_ALERT, "Couldn't execute command: %s", str);
			}
			else
				exit(WEXITSTATUS(retval));

			exit(0);
		}

		if (WIFSIGNALED(retval))
			kill(our_pid, WTERMSIG(retval));
	}

	exit(0); /* Script errors aren't server errors */
}

/* Execute external script/program */
int
notify_exec(const notify_script_t *script)
{
	return system_call_script(NULL, NULL, NULL, 0, script);
}

static void
fifo_open(notify_fifo_t* fifo, thread_func_t script_exit, const char *type)
{
	int ret;
	int sav_errno;

	if (fifo->name) {
		sav_errno = 0;

		if (!(ret = mkfifo(fifo->name, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH))) {
			fifo->created_fifo = true;

			if ((fifo->uid != our_uid || fifo->gid != our_gid) &&
			     chown(fifo->name, our_uid != fifo->uid ? fifo->uid : (uid_t)-1, our_gid != fifo->gid ? fifo->gid : (gid_t)-1))
				log_message(LOG_INFO, "Failed to set uid:gid for fifo %s", fifo->name);
		} else {
			sav_errno = errno;

			if (sav_errno != EEXIST)
				log_message(LOG_INFO, "Unable to create %snotify fifo %s", type, fifo->name);
		}

		if (!sav_errno || sav_errno == EEXIST) {
			/* Run the notify script if there is one */
			if (fifo->script)
				system_call_script(master, script_exit, fifo, TIMER_NEVER, fifo->script);

			/* Now open the fifo */
			if ((fifo->fd = open(fifo->name, O_RDWR | O_CLOEXEC | O_NONBLOCK | O_NOFOLLOW)) == -1) {
				log_message(LOG_INFO, "Unable to open %snotify fifo %s - errno %d", type, fifo->name, errno);
				if (fifo->created_fifo) {
					unlink(fifo->name);
					fifo->created_fifo = false;
				}
			}
		}

		if (fifo->fd == -1) {
			FREE_CONST(fifo->name);
			fifo->name = NULL;
		}
	}
}

void
notify_fifo_open(notify_fifo_t* global_fifo, notify_fifo_t* fifo, thread_func_t script_exit, const char *type)
{
	/* Open the global FIFO if specified */
	if (global_fifo->name)
		fifo_open(global_fifo, script_exit, "");

	/* Now the specific FIFO */
	if (fifo->name)
		fifo_open(fifo, script_exit, type);
}

static void
fifo_close(notify_fifo_t* fifo)
{
	/* We unlink the fifo first, so that a script that
	 * loops reopening the fifo if it is closed cannot
	 * reopen this fifo. */
	if (fifo->created_fifo)
		unlink(fifo->name);

	if (fifo->fd != -1) {
		close(fifo->fd);
		fifo->fd = -1;
	}
}

void
notify_fifo_close(notify_fifo_t* global_fifo, notify_fifo_t* fifo)
{
	if (global_fifo->fd != -1)
		fifo_close(global_fifo);

	fifo_close(fifo);
}

static void
child_killed_reload(thread_ref_t thread)
{
	/* If the child didn't die, then force it */
	if (thread->type == THREAD_CHILD_TIMEOUT)
		kill(-getpgid(thread->u.c.pid), SIGKILL);
}

void
child_killed_thread(thread_ref_t thread)
{
	thread_master_t *m = thread->master;

	/* If the child didn't die, then force it */
	if (thread->type == THREAD_CHILD_TIMEOUT)
		kill(-getpgid(thread->u.c.pid), SIGKILL);

	/* If all children have died, we can now complete the
	 * termination process */
	if (!m->child.rb_root.rb_node && !m->shutdown_timer_running)
		thread_add_terminate_event(m);
}

void
script_killall(thread_master_t *m, int signo, bool shutting_down)
{
	thread_t *thread;
	pid_t p_pgid, c_pgid;

	p_pgid = getpgid(0);

	rb_for_each_entry_cached(thread, &m->child, n) {
		c_pgid = getpgid(thread->u.c.pid);
		if (c_pgid != p_pgid)
			kill(-c_pgid, signo);
		else {
			log_message(LOG_INFO, "Child process %d in our process group %d", c_pgid, p_pgid);
			kill(thread->u.c.pid, signo);
		}
	}

	/* We want to timeout the killed children in 1 second */
	if (signo != SIGKILL)
		thread_children_reschedule(m, shutting_down ? child_killed_thread : child_killed_reload, TIMER_HZ);
}

static bool
is_executable(const struct stat *buf, uid_t uid, gid_t gid)
{
	return (uid == 0 && buf->st_mode & (S_IXUSR | S_IXGRP | S_IXOTH)) ||
	       (uid == buf->st_uid && buf->st_mode & S_IXUSR) ||
	       (uid != buf->st_uid &&
		((gid == buf->st_gid && buf->st_mode & S_IXGRP) ||
		 (gid != buf->st_gid && buf->st_mode & S_IXOTH)));
}

static void
replace_cmd_name(notify_script_t *script, const char *new_path)
{
	size_t len;
	const char * const *wp = &script->args[1];
	size_t num_words = 1;
	char **params;
	char **word_ptrs;
	char *words;
	/* Avoid gcc warning: to be safe all intermediate pointers in cast from ‘char **’
	 *   to ‘const char **’ must be ‘const’ qualified
	 * What we want to assign to is a pointer (therefore non const) to an array of pointers
	 *   which can be modified, therefore non-const, to const strings, i.e. const char **;
	 *   but if assign a char ** to a const char ** we get the above warning
	 */
	union {
		char **params;
		const char **cparams;
	} args;

	len = strlen(new_path) + 1;
	while (*wp)
		len += strlen(*wp++) + 1;
	num_words = (script->args[0] - PTR_CAST_CONST(char, &script->args[0])) - 1;

	params = word_ptrs = MALLOC((num_words + 1) * sizeof(char *) + len);
	words = PTR_CAST(char, &params[num_words + 1]);

	strcpy(words, new_path);
	*(word_ptrs++) = words;
	words += strlen(words) + 1;

	wp = &script->args[1];
	while (*wp) {
		strcpy(words, *wp);
		*(word_ptrs++) = words;
		words += strlen(*wp) + 1;
		wp++;
	}
	*word_ptrs = NULL;

	FREE_CONST(script->args);
	args.params = params;
	script->args = args.cparams;
}

/* The following function is essentially __execve_common() from glibc posix/execvpe.c */
static int
find_path(notify_script_t *script)
{
	size_t filename_len;
	size_t file_len;
	size_t path_len;
	const char *file = script->args[0];
	struct stat buf;
	int ret;
	int ret_val = ENOENT;
	int sgid_num;
	gid_t *sgid_list = NULL;
	const char *subp;
	bool got_eacces = false;
	const char *p;

	/* We check the simple case first. */
	if (*file == '\0')
		return ENOENT;

	filename_len = strlen(file);
	if (filename_len >= PATH_MAX)
		return ENAMETOOLONG;

	/* Don't search when it contains a slash. */
	if (strchr (file, '/') != NULL)
		return 0;

	/* Get the path if we haven't already done so, and if that doesn't
	 * exist, use CS_PATH */
	if (!path) {
		path = getenv ("PATH");

		if (!path) {
			size_t cs_path_len;
			path = MALLOC(cs_path_len = confstr(_CS_PATH, NULL, 0));
			confstr(_CS_PATH, path, cs_path_len);
			path_is_malloced = true;
		}
	}

	/* Although GLIBC does not enforce NAME_MAX, we set it as the maximum
	   size to avoid unbounded stack allocation.  Same applies for
	   PATH_MAX. */
	file_len = strnlen (file, NAME_MAX + 1);
	path_len = strnlen (path, PATH_MAX - 1) + 1;

	if (file_len > NAME_MAX)
		return ENAMETOOLONG;

	if (script->uid != our_uid || script->gid != our_gid) {
		/* Set file access to the relevant uid/gid */
		if (script->gid != our_gid) {
			if (setegid(script->gid)) {
				log_message(LOG_INFO, "Unable to set egid to %u (%m)", script->gid);
				return EACCES;
			}
		}

		if (script->uid != our_uid) {
			if (seteuid(script->uid)) {
				log_message(LOG_INFO, "Unable to set euid to %u (%m)", script->uid);
				ret_val = EACCES;
				goto exit;
			}
		}

		/* Get our supplementary groups */
		sgid_num = getgroups(0, NULL);
		if (sgid_num == -1) {
			log_message(LOG_INFO, "Unable to get number of supplementary gids (%m)");
			ret_val = EACCES;
			goto exit;
		}
		sgid_list = MALLOC(((size_t)sgid_num + 1) * sizeof(gid_t));
		sgid_num = getgroups(sgid_num, sgid_list);
		sgid_list[sgid_num++] = our_gid;

		/* Clear the supplementary group list */
		if (setgroups(1, &script->gid)) {
			log_message(LOG_INFO, "Unable to set supplementary gids (%m)");
			ret_val = EACCES;
			goto exit;
		}
	}

	for (p = path; ; p = subp)
	{
		subp = strchrnul (p, ':');

		/* PATH is larger than PATH_MAX and thus potentially larger than
		   the stack allocation. */
		if (subp >= p + path_len) {
			/* There are no more paths, bail out. */
			if (*subp == '\0') {
				ret_val = ENOENT;
				goto exit;
			}

			/* Otherwise skip to next one. */
			continue;
		}

		/* Use the current path entry, plus a '/' if nonempty, plus the file to execute. */
		char *buffer = MALLOC(path_len + file_len + 1);
		char *pend = mempcpy (buffer, p, (size_t)(subp - p));
		*pend = '/';
		memcpy (pend + (p < subp), file, file_len + 1);

		ret = stat (buffer, &buf);
		if (!ret) {
			if (!S_ISREG(buf.st_mode))
				errno = EACCES;
			else if (!is_executable(&buf, script->uid, script->gid)) {
				errno = EACCES;
			} else {
				/* Success */
				log_message(LOG_INFO, "WARNING - script `%s` resolved by path search to `%s`. Please specify full path.", script->args[0], buffer);

				/* Copy the found file name, and any parameters */
				replace_cmd_name(script, buffer);

				ret_val = 0;
				got_eacces = false;
				FREE(buffer);
				goto exit;
			}
		}
		FREE(buffer);

		switch (errno)
		{
		case ENOEXEC:
		case EACCES:
			/* Record that we got a 'Permission denied' error.  If we end
			   up finding no executable we can use, we want to diagnose
			   that we did find one but were denied access. */
			if (!ret)
				got_eacces = true;
		case ENOENT:
		case ESTALE:
		case ENOTDIR:
			/* Those errors indicate the file is missing or not executable
			   by us, in which case we want to just try the next path
			   directory. */
		case ENODEV:
		case ETIMEDOUT:
			/* Some strange filesystems like AFS return even
			   stranger error numbers.  They cannot reasonably mean
			   anything else so ignore those, too. */
			break;

		default:
			/* Some other error means we found an executable file, but
			   something went wrong accessing it; return the error to our
			   caller. */
			ret_val = -1;
			goto exit;
		}

		if (*subp++ == '\0')
			break;
	}

exit:
	/* Restore root euid/egid */
	if (script->gid != our_gid && setegid(our_gid))
		log_message(LOG_INFO, "Unable to restore egid after script search (%m)");
	if (script->uid != our_uid && seteuid(our_uid))
		log_message(LOG_INFO, "Unable to restore euid after script search (%m)");

	/* restore supplementary groups */
	if (sgid_list) {
		if (setgroups((size_t)sgid_num, sgid_list))
			log_message(LOG_INFO, "Unable to restore supplementary groups after script search (%m)");
		FREE(sgid_list);
	}

	/* We tried every element and none of them worked. */
	if (got_eacces) {
		/* At least one failure was due to permissions, so report that error. */
		return EACCES;
	}

	return ret_val;
}

static int
check_security(const char *filename, bool using_script_security)
{
	const char *next;
	char *slash;
	int ret;
	struct stat buf;
	int flags = 0;
	char *filename_copy;

	next = filename;
	while (next) {
		slash = strchrnul(next, '/');
		if (*slash)
			next = slash + 1;
		else {
			slash = NULL;
			next = NULL;
		}

		if (slash) {
			/* We want to check '/' for first time around */
			if (slash == filename)
				slash++;
			filename_copy = STRNDUP(filename, slash - filename);
		}

		ret = fstatat(0, slash ? filename_copy : filename, &buf, AT_SYMLINK_NOFOLLOW);

		/* Restore the full path name */
		if (slash)
			FREE(filename_copy);

		if (ret) {
			if (errno == EACCES || errno == ELOOP || errno == ENOENT || errno == ENOTDIR)
				log_message(LOG_INFO, "check_script_secure could not find script '%s' - disabling", filename);
			else
				log_message(LOG_INFO, "check_script_secure('%s') returned errno %d - %s - disabling", filename, errno, strerror(errno));
			return flags | SC_NOTFOUND;
		}

		/* If it is not the last item, it must be a directory. If it is the last item
		 * it must be a file or a symbolic link. */
		if ((slash && !S_ISDIR(buf.st_mode)) ||
		    (!slash &&
		     !S_ISREG(buf.st_mode) &&
		     !S_ISLNK(buf.st_mode))) {
			log_message(LOG_INFO, "Wrong file type found in script path '%s'.", filename);
			return flags | SC_INHIBIT;
		}

		if (buf.st_uid ||				/* Owner is not root */
		    (((S_ISDIR(buf.st_mode) &&			/* A directory without the sticky bit set */
		       !(buf.st_mode & S_ISVTX)) ||
		      S_ISREG(buf.st_mode)) &&			/* This is a file */
		     ((buf.st_gid && buf.st_mode & S_IWGRP) ||	/* Group is not root and group write permission */
		      buf.st_mode & S_IWOTH))) {		/* World has write permission */
			log_message(LOG_INFO, "Unsafe permissions found for script '%s'%s.", filename, using_script_security ? " - disabling" : "");
			flags |= SC_INSECURE;
			return flags | (using_script_security ? SC_INHIBIT : 0);
		}
	}

	return flags;
}

static char *
clean_path(const char *old_argv)
{
	char *new_argv = MALLOC(strlen(old_argv) + 1 + 1);	// We can add an initial '/'
	const char *slash, *next_slash;
	char *previous_slash;
	const char *ip = old_argv;
	char *op = new_argv;

	/* Remove leading ./ and ../ */
	if (!strncmp(ip, "./", 2))
		ip += 1;
	else if (!strncmp(ip, "../", 3))
		ip += 2;

	/* Remove any leading /../ and /./ recursively */
	while (!strncmp(ip, "/../", 4) || !strncmp(ip, "/./", 3) || !strncmp(ip, "//", 2))
		ip += ip[1] == '/' ? 1 : ip[2] == '/' ? 2 : 3;

	if (*ip != '/')
		*op++ = '/';

	slash = strchr(ip, '/');
	if (!slash)
		strcpy(op, ip);
	else if (slash > ip) {
		strncpy(op, ip, slash - ip);
		op += slash - ip;
	}

	for (; slash; slash = next_slash) {
		next_slash = strchr(slash + 1, '/');

		if (!next_slash) {
			strcpy(op, slash);
			break;
		}

		/* We are looking for '//', '/./' or '/../' */
		if (next_slash > slash + 3 ||
		    (slash[1] != '.' && slash[1] != '/') ||
		    (slash[1] == '.' &&
		     (slash[2] != '.' && slash[2] != '/'))) {
			strncpy(op, slash, next_slash - slash);
			op += next_slash - slash;
			continue;
		}

		/* We have found one of them */

		if (next_slash == slash + 1) {
			/* // */
			ip++;
			continue;
		}

		if (next_slash == slash + 2) {
			/* /./ */
			ip += 2;
			continue;
		}

		/* /../ - we must remove previous element */
		previous_slash = strrchr(new_argv, '/');
		if (!previous_slash)
			op = new_argv;
		else
			op = previous_slash;
		*op = '\0';
	}

	if (!strcmp(old_argv, new_argv)) {
		FREE(new_argv);
		return NULL;
	}

	return new_argv;
}

unsigned
check_script_secure(notify_script_t *script,
#ifndef _HAVE_LIBMAGIC_
					     __attribute__((unused))
#endif
								     magic_t magic)
{
	unsigned flags;
	int ret;
	struct stat file_buf;
	bool need_script_protection = false;
	const char *real_path;
	const char *new_argv_0;
	union {
		char *nc;	// needed for free()
		const char *c;
	} sav_path;
	int sav_errno;
	int sav_death_sig;

	if (!script)
		return 0;

	/* If the script starts "</" (possibly with white space between
	 * the '<' and '/'), it is checking for a file being openable,
	 * so it won't be executed */
	if (script->args[0][0] == '<' &&
	    script->args[0][strspn(script->args[0] + 1, " \t") + 1] == '/')
		return SC_SYSTEM;

	if (!strchr(script->args[0], '/')) {
		/* It is a bare file name, so do a path search */
		if ((ret = find_path(script))) {
			if (ret == EACCES)
				log_message(LOG_INFO, "Permissions failure for script %s in path - disabling", script->args[0]);
			else
				log_message(LOG_INFO, "Cannot find script %s in path - disabling", script->args[0]);
			return SC_NOTFOUND;
		}
	}

	/* Check script accessible by the user running it */
	if (script->gid != our_gid || script->uid != our_uid) {
		/* Save parent death signal */
		prctl(PR_GET_PDEATHSIG, &sav_death_sig);

		if ((script->gid != our_gid && setegid(script->gid)) ||
		    (script->uid != our_uid && seteuid(script->uid))) {
			log_message(LOG_INFO, "Unable to set uid:gid %u:%u for script %s - disabling", script->uid, script->gid, script->args[0]);

			if ((script->uid != our_uid && seteuid(our_uid)) ||
			    (script->gid != our_gid && setegid(our_gid)))
				log_message(LOG_INFO, "Unable to restore uid:gid from %u:%u %u:%u after script %s", our_uid, our_gid, script->uid, script->gid, script->args[0]);

			return SC_INHIBIT;
		}
	}

	real_path = realpath(script->args[0], NULL);
	sav_errno = errno;

	if (script->gid != our_gid || script->uid != our_uid) {
		if ((script->gid != our_gid && setegid(our_gid)) ||
		    (script->uid != our_uid && seteuid(our_uid)))
			log_message(LOG_INFO, "Unable to restore uid:gid %u:%u from %u:%u after checking script %s", our_uid, our_gid, script->uid, script->gid, script->args[0]);

		/* Restore parent death signal */
		prctl(PR_SET_PDEATHSIG, sav_death_sig);

		/* Check the parent didn't die in the window when PDEATHSIG was not set */
		if (!__test_bit(CONFIG_TEST_BIT, &debug) && main_pid != getppid())
			kill(our_pid, SIGTERM);
	}

	if (!real_path) {
		log_message(LOG_INFO, "Script %s cannot be accessed - %s", script->args[0], strerror(sav_errno));

		return SC_NOTFOUND;
	}

	/* It is much easier to ensure that real_path is part of
	 * keepalived's malloc handling. */
	sav_path.c = real_path;
	real_path = STRDUP(real_path);
	free(sav_path.nc);	/* malloc'd returned by realpath() */

	/* Get the permissions for the file itself */
	if (stat(real_path, &file_buf)) {
		log_message(LOG_INFO, "Unable to access script `%s` - disabling", script->args[0]);

		FREE_CONST(real_path);

		return SC_NOTFOUND;
	}

	if (use_symlinks || strcmp(real_path, script->args[0])) {
		/* Remove /./, /../, repeated /'s from argv[0] */
		new_argv_0 = clean_path(script->args[0]);
		if (new_argv_0) {
			/* The path name is different */
			replace_cmd_name(script, new_argv_0);
			FREE_CONST_ONLY(new_argv_0);
		}
	}

	if (strcmp(real_path, script->args[0]) && !use_symlinks) {
		script->path = real_path;
		real_path = NULL;
	}

	flags = SC_ISSCRIPT;

	/* We have the final file. Check if root is executing it, or it is set uid/gid root. */
	if (is_executable(&file_buf, script->uid, script->gid)) {
		flags |= SC_EXECUTABLE;
		if (script->uid == 0 || script->gid == 0 ||
		    (file_buf.st_uid == 0 && (file_buf.st_mode & S_IXUSR) && (file_buf.st_mode & S_ISUID)) ||
		    (file_buf.st_gid == 0 && (file_buf.st_mode & S_IXGRP) && (file_buf.st_mode & S_ISGID)))
			need_script_protection = true;
	} else
		log_message(LOG_INFO, "WARNING - script '%s' is not executable for uid:gid %u:%u - disabling.", script->args[0], script->uid, script->gid);

	/* Default to execable */
	script->flags |= SC_EXECABLE;
#ifdef _HAVE_LIBMAGIC_
	if (magic && flags & SC_EXECUTABLE) {
		const char *magic_desc = magic_file(magic, real_path ? real_path : script->path ? script->path : script->args[0]);
		if (!strstr(magic_desc, " executable") &&
		    !strstr(magic_desc, " shared object")) {
			log_message(LOG_INFO, "Please add a #! shebang to script %s", script->args[0]);
			script->flags &= ~SC_EXECABLE;
		}
	}
#endif

	if (real_path)
		FREE_CONST(real_path);

	if (need_script_protection) {
		/* Make sure that all parts of the path(s) are not non-root writable */
		flags |= check_security(script->args[0], script_security);

		if (script->path)
			flags |= check_security(script->path, script_security);
	}

	return flags;
}

unsigned
check_notify_script_secure(notify_script_t **script_p, magic_t magic)
{
	unsigned flags;
	notify_script_t *script = *script_p;

	if (!script)
		return 0;

	flags = check_script_secure(script, magic);

	/* Mark not to run if needs inhibiting */
	if ((flags & (SC_INHIBIT | SC_NOTFOUND)) ||
	    !(flags & SC_EXECUTABLE))
		free_notify_script(script_p);

	return flags;
}

static void
set_pwnam_buf_len(void)
{
	long buf_len;
	struct stat statbuf;

	/* Get buffer length needed for getpwnam_r/getgrnam_r */
	if ((buf_len = sysconf(_SC_GETPW_R_SIZE_MAX)) == -1)
		getpwnam_buf_len = 1024;	/* A safe default if no value is returned */
	else
		getpwnam_buf_len = (size_t)buf_len;
	if ((buf_len = sysconf(_SC_GETGR_R_SIZE_MAX)) != -1 &&
	    (size_t)buf_len > getpwnam_buf_len)
		getpwnam_buf_len = (size_t)buf_len;

	/* In case there is an excessive number of members of the group, which
	 * could cause the default buffer size to overflow, a reasonable upper
	 * bound for the buffer size is the size of the group file itself. */
	if (!stat("/etc/group", &statbuf) && (unsigned long)statbuf.st_size > getpwnam_buf_len)
		getpwnam_buf_len = statbuf.st_size;
}

static bool
set_uid_gid(const char *username, const char *groupname, uid_t *uid_p, gid_t *gid_p)
{
	uid_t uid;
	gid_t gid;
	struct passwd pwd;
	struct passwd *pwd_p;
	struct group grp;
	struct group *grp_p;
	int ret;
	char *buf;

	if (!getpwnam_buf_len)
		set_pwnam_buf_len();

	buf = MALLOC(getpwnam_buf_len);

	if ((ret = getpwnam_r(username, &pwd, buf, getpwnam_buf_len, &pwd_p))) {
		log_message(LOG_INFO, "Unable to resolve script username '%s' - ignoring", username);
		FREE(buf);
		return true;
	}
	if (!pwd_p) {
		log_message(LOG_INFO, "Script user '%s' does not exist", username);
		FREE(buf);
		return true;
	}

	uid = pwd.pw_uid;
	gid = pwd.pw_gid;

	if (groupname) {
		if ((ret = getgrnam_r(groupname, &grp, buf, getpwnam_buf_len, &grp_p))) {
			log_message(LOG_INFO, "Unable to resolve script group name '%s' - ignoring", groupname);
			FREE(buf);
			return true;
		}
		if (!grp_p) {
			log_message(LOG_INFO, "Script group '%s' does not exist", groupname);
			FREE(buf);
			return true;
		}
		gid = grp.gr_gid;
	}

	*uid_p = uid;
	*gid_p = gid;

	FREE(buf);

	return false;
}

/* The default script user/group is keepalived_script if it exists, or our uid/gid otherwise */
void
reset_default_script_user(void)
{
	default_script_uid_set = false;
	default_user_fail = false;
}

bool
set_default_script_user(const char *username, const char *groupname)
{
	/* Even if we fail to set it, there is no point in trying again */
	default_script_uid_set = true;

	if (set_uid_gid(username, groupname, &default_script_uid, &default_script_gid))
		default_user_fail = true;

	return default_user_fail;
}

bool
get_default_script_user(uid_t *uid, gid_t *gid)
{
	const char *default_user = "keepalived_script";

	if (default_user_fail)
		return true;

	if (!default_script_uid_set) {
		/* Even if we fail to set it, there is no point in trying again */
		default_script_uid_set = true;

		default_script_uid = our_uid;
		default_script_gid = our_gid;

		if (set_uid_gid(default_user, NULL, &default_script_uid, &default_script_gid) && script_security) {
			report_config_error(CONFIG_GENERAL_ERROR, "Unable to set default user %s for script", default_user);
			default_user_fail = true;
			return true;
		}
	}

	*uid = default_script_uid;
	*gid = default_script_gid;

	return false;
}

bool
set_script_uid_gid(const vector_t *strvec, unsigned keyword_offset, uid_t *uid_p, gid_t *gid_p)
{
	const char *username;
	const char *groupname;

	username = strvec_slot(strvec, keyword_offset);
	if (vector_size(strvec) > keyword_offset + 1)
		groupname = strvec_slot(strvec, keyword_offset + 1);
	else
		groupname = NULL;

	return set_uid_gid(username, groupname, uid_p, gid_p);
}

void
set_script_params_array(const vector_t *strvec, notify_script_t *script, unsigned extra_params)
{
	unsigned num_words = 0;
	size_t len = 0;
	char **word_ptrs;
	char *words;
	const vector_t *strvec_qe = NULL;
	unsigned i;
	union {		/* See replace_cmd_line for details of why this is necessary */
		char **params;
		const char **cparams;
	} args;

	/* Count the number of words, and total string length */
	if (vector_size(strvec) >= 2)
		strvec_qe = alloc_strvec_quoted_escaped(strvec_slot(strvec, 1));

	if (!strvec_qe)
		return;

	num_words = vector_size(strvec_qe);
	for (i = 0; i < num_words; i++)
		len += strlen(strvec_slot(strvec_qe, i)) + 1;

	/* Allocate memory for pointers to words and words themselves */
	word_ptrs = MALLOC((num_words + extra_params + 1) * sizeof(char *) + len);
	words = PTR_CAST(char, word_ptrs) + (num_words + extra_params + 1) * sizeof(char *);
	args.params = word_ptrs;
	script->args = args.cparams;

	/* Set up pointers to words, and copy the words */
	for (i = 0; i < num_words; i++) {
		strcpy(words, strvec_slot(strvec_qe, i));
		*(word_ptrs++) = words;
		words += strlen(words) + 1;
	}
	*word_ptrs = NULL;

	script->num_args = num_words;

	free_strvec(strvec_qe);
}

notify_script_t*
notify_script_init(int extra_params, const char *type)
{
	notify_script_t *script = MALLOC(sizeof(notify_script_t));
	const vector_t *strvec_qe;

	/* We need to reparse the command line, allowing for quoted and escaped strings */
	strvec_qe = alloc_strvec_quoted(NULL);

	if (!strvec_qe) {
		log_message(LOG_INFO, "Unable to parse notify script");
		FREE(script);
		return NULL;
	}

	set_script_params_array(strvec_qe, script, extra_params);
	if (!script->args) {
		log_message(LOG_INFO, "Unable to parse script '%s' - ignoring", strvec_slot(strvec_qe, 1));
		FREE(script);
		free_strvec(strvec_qe);
		return NULL;
	}

	script->flags = 0;

	if (vector_size(strvec_qe) > 2) {
		if (set_script_uid_gid(strvec_qe, 2, &script->uid, &script->gid)) {
			log_message(LOG_INFO, "Invalid user/group for %s script %s - ignoring", type, script->args[0]);
			FREE_CONST(script->args);
			FREE(script);
			free_strvec(strvec_qe);
			return NULL;
		}
	} else if (get_default_script_user(&script->uid, &script->gid)) {
		log_message(LOG_INFO, "Failed to set default user for %s script %s - ignoring", type, script->args[0]);
		FREE_CONST(script->args);
		FREE(script);
		free_strvec(strvec_qe);
		return NULL;
	}

	free_strvec(strvec_qe);

	return script;
}

void
add_script_param(notify_script_t *script, const char *param)
{
	/* We store the args as an array of pointers to the args, terminated
	 * by a NULL pointer, followed by the null terminated strings themselves
	 */

	if (script->args[script->num_args + 1]) {
		log_message(LOG_INFO, "notify_fifo_script %s no room to add parameter %s", script->args[0], param);
		return;
	}

	/* Add the extra parameter in the pre-reserved slot at the end */
	script->args[script->num_args++] = param;
}

void
notify_free_script(notify_script_t *script)
{
	FREE_CONST_PTR(script->path);
	FREE_CONST(script->args);
}

void
notify_resource_release(void)
{
	if (path_is_malloced) {
		FREE(path);
		path_is_malloced = false;
		path = NULL;
	}
}

bool __attribute__ ((pure))
notify_script_compare(const notify_script_t *a, const notify_script_t *b)
{
	int i;

	if (a->num_args != b->num_args)
		return false;
	for (i = 0; i < a->num_args; i++) {
		if (strcmp(a->args[i], b->args[i]))
			return false;
	}

	return true;
}

void
set_our_uid_gid(void)
{
	our_uid = geteuid();
	our_gid = getegid();
}

#ifdef THREAD_DUMP
void
register_notify_addresses(void)
{
	register_thread_address("child_killed_thread", child_killed_thread);
	register_thread_address("child_killed_reload", child_killed_reload);
}
#endif
