/*
   rc-applets.c

   Handle multicall applets for use in our init scripts.
   Basically this makes us a lot faster for the most part, and removes
   any shell incompatabilities we might otherwise encounter.
   */

/*
 * Copyright 2007-2008 Roy Marples <roy@marples.name>
 * All rights reserved

 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#define SYSLOG_NAMES

#include <sys/types.h>
#include <sys/time.h>

#include <errno.h>
#include <ctype.h>
#include <inttypes.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include <syslog.h>
#include <time.h>
#include <unistd.h>

#include "builtins.h"
#include "einfo.h"
#include "rc-misc.h"

/* usecs to wait while we poll the file existance  */
#define WAIT_INTERVAL	20000000
#define ONE_SECOND      690000000

/* Applet is first parsed in rc.c - no point in doing it again */
extern const char *applet;

static int syslog_decode(char *name, CODE *codetab)
{
	CODE *c;

	if (isdigit((unsigned char)*name))
		return atoi(name);

	for (c = codetab; c->c_name; c++)
		if (! strcasecmp(name, c->c_name))
			return c->c_val;

	return -1;
}

static int do_e(int argc, char **argv)
{
	int retval = EXIT_SUCCESS;
	int i;
	size_t l = 0;
	char *message = NULL;
	char *p;
	int level = 0;
	struct timespec ts;
	struct timeval stop, now;
	int (*e) (const char *, ...) EINFO_PRINTF(1, 2) = NULL;
	int (*ee) (int, const char *, ...) EINFO_PRINTF(2, 3) = NULL;

	/* Punt applet */
	argc--;
	argv++;

	if (strcmp(applet, "eval_ecolors") == 0) {
		printf("GOOD='%s'\nWARN='%s'\nBAD='%s'\nHILITE='%s'\nBRACKET='%s'\nNORMAL='%s'\n",
			ecolor(ECOLOR_GOOD),
			ecolor(ECOLOR_WARN),
			ecolor(ECOLOR_BAD),
			ecolor(ECOLOR_HILITE),
			ecolor(ECOLOR_BRACKET),
			ecolor(ECOLOR_NORMAL));
		exit(EXIT_SUCCESS);
	}

	if (argc > 0) {
		if (strcmp(applet, "eend") == 0 ||
		    strcmp(applet, "ewend") == 0 ||
		    strcmp(applet, "veend") == 0 ||
		    strcmp(applet, "vweend") == 0 ||
		    strcmp(applet, "ewaitfile") == 0)
		{
			errno = 0;
			retval = (int)strtoimax(argv[0], &p, 0);
			if (!p || *p != '\0')
				errno = EINVAL;
			if (errno)
				retval = EXIT_FAILURE;
			else {
				argc--;
				argv++;
			}
		} else if (strcmp(applet, "esyslog") == 0 ||
			   strcmp(applet, "elog") == 0) {
			p = strchr(argv[0], '.');
			if (!p || (level = syslog_decode(p + 1, prioritynames)) == -1)
				eerrorx("%s: invalid log level `%s'", applet, argv[0]);

			if (argc < 3)
				eerrorx("%s: not enough arguments", applet);

			unsetenv("EINFO_LOG");
			setenv("EINFO_LOG", argv[1], 1);

			argc -= 2;
			argv += 2;
		}
	}

	if (strcmp(applet, "ewaitfile") == 0) {
		if (errno)
			eerrorx("%s: invalid timeout", applet);
		if (argc == 0)
			eerrorx("%s: not enough arguments", applet);

		gettimeofday(&stop, NULL);
		/* retval stores the timeout */
		stop.tv_sec += retval;
		ts.tv_sec = 0;
		ts.tv_nsec = WAIT_INTERVAL;
		for (i = 0; i < argc; i++) {
			ebeginv("Waiting for %s", argv[i]);
			for (;;){
				if (exists(argv[i]))
					break;
				if (nanosleep(&ts, NULL) == -1)
					return EXIT_FAILURE;
				gettimeofday(&now, NULL);
				if (retval <= 0)
					continue;
				if (timercmp(&now, &stop, <))
					continue;
				eendv(EXIT_FAILURE,
				      "timed out waiting for %s", argv[i]);
				return EXIT_FAILURE;
			}
			eendv(EXIT_SUCCESS, NULL);
		}
		return EXIT_SUCCESS;
	}

	if (argc > 0) {
		for (i = 0; i < argc; i++)
			l += strlen(argv[i]) + 1;

		message = xmalloc(l);
		p = message;

		for (i = 0; i < argc; i++) {
			if (i > 0)
				*p++ = ' ';
			l = strlen(argv[i]);
			memcpy(p, argv[i], l);
			p += l;
		}
		*p = 0;
	}

	if (strcmp(applet, "einfo") == 0)
		e = einfo;
	else if (strcmp(applet, "einfon") == 0)
		e = einfon;
	else if (strcmp(applet, "ewarn") == 0)
		e = ewarn;
	else if (strcmp(applet, "ewarnn") == 0)
		e = ewarnn;
	else if (strcmp(applet, "eerror") == 0) {
		e = eerror;
		retval = 1;
	} else if (strcmp(applet, "eerrorn") == 0) {
		e = eerrorn;
		retval = 1;
	} else if (strcmp(applet, "ebegin") == 0)
		e = ebegin;
	else if (strcmp(applet, "eend") == 0)
		ee = eend;
	else if (strcmp(applet, "ewend") == 0)
		ee = ewend;
	else if (strcmp(applet, "esyslog") == 0) {
		elog(retval, "%s", message);
		retval = 0;
	} else if (strcmp(applet, "veinfo") == 0)
		e = einfov;
	else if (strcmp(applet, "veinfon") == 0)
		e = einfovn;
	else if (strcmp(applet, "vewarn") == 0)
		e = ewarnv;
	else if (strcmp(applet, "vewarnn") == 0)
		e = ewarnvn;
	else if (strcmp(applet, "vebegin") == 0)
		e = ebeginv;
	else if (strcmp(applet, "veend") == 0)
		ee = eendv;
	else if (strcmp(applet, "vewend") == 0)
		ee = ewendv;
	else if (strcmp(applet, "eindent") == 0)
		eindent();
	else if (strcmp(applet, "eoutdent") == 0)
		eoutdent();
	else if (strcmp(applet, "veindent") == 0)
		eindentv();
	else if (strcmp(applet, "veoutdent") == 0)
		eoutdentv();
	else {
		eerror("%s: unknown applet", applet);
		retval = EXIT_FAILURE;
	}

	if (message) {
		if (e)
			e("%s", message);
		else if (ee)
			ee(retval, "%s", message);
	} else {
		if (e)
			e(NULL);
		else if (ee)
			ee(retval, NULL);
	}

	free(message);
	return retval;
}

static int do_service(int argc, char **argv)
{
	bool ok = false;
	char *service;
	char *exec;
	int idx = 0;

	if (argc > 1)
		service = argv[1];
	else
		service = getenv("RC_SVCNAME");

	if (! service || *service == '\0')
		eerrorx("%s: no service specified", applet);

	if (strcmp(applet, "service_started") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_STARTED);
	else if (strcmp(applet, "service_stopped") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_STOPPED);
	else if (strcmp(applet, "service_inactive") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_INACTIVE);
	else if (strcmp(applet, "service_starting") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_STARTING);
	else if (strcmp(applet, "service_stopping") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_STOPPING);
	else if (strcmp(applet, "service_hotplugged") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_HOTPLUGGED);
	else if (strcmp(applet, "service_wasinactive") == 0)
		ok = (rc_service_state(service) & RC_SERVICE_WASINACTIVE);
	else if (strcmp(applet, "service_started_daemon") == 0) {
		service = getenv("RC_SVCNAME");
		exec = argv[1];
		if (argc > 3) {
			service = argv[1];
			exec = argv[2];
			sscanf(argv[3], "%d", &idx);
		} else if (argc == 3) {
			if (sscanf(argv[2], "%d", &idx) != 1) {
				service = argv[1];
				exec = argv[2];
			}
		}
		ok = rc_service_started_daemon(service, exec, NULL, idx);
			
	} else
		eerrorx("%s: unknown applet", applet);

	return ok ? EXIT_SUCCESS : EXIT_FAILURE;
}

static int do_mark_service(int argc, char **argv)
{
	bool ok = false;
	char *svcname = getenv("RC_SVCNAME");
	char *service = NULL;
	char *runscript_pid;
	char *mtime;
	pid_t pid;
	size_t l;

	if (argc > 1)
		service = argv[1];
	else
		service = getenv("RC_SVCNAME");

	if (! service || *service == '\0')
		eerrorx("%s: no service specified", applet);

	if (strcmp(applet, "mark_service_started") == 0)
		ok = rc_service_mark(service, RC_SERVICE_STARTED);
	else if (strcmp(applet, "mark_service_stopped") == 0)
		ok = rc_service_mark(service, RC_SERVICE_STOPPED);
	else if (strcmp(applet, "mark_service_inactive") == 0)
		ok = rc_service_mark(service, RC_SERVICE_INACTIVE);
	else if (strcmp(applet, "mark_service_starting") == 0)
		ok = rc_service_mark(service, RC_SERVICE_STARTING);
	else if (strcmp(applet, "mark_service_stopping") == 0)
		ok = rc_service_mark(service, RC_SERVICE_STOPPING);
	else if (strcmp(applet, "mark_service_hotplugged") == 0)
		ok = rc_service_mark(service, RC_SERVICE_HOTPLUGGED);
	else if (strcmp(applet, "mark_service_failed") == 0)
		ok = rc_service_mark(service, RC_SERVICE_FAILED);
	else
		eerrorx("%s: unknown applet", applet);

	/* If we're marking ourselves then we need to inform our parent runscript
	   process so they do not mark us based on our exit code */
	if (ok && svcname && strcmp(svcname, service) == 0) {
		runscript_pid = getenv("RC_RUNSCRIPT_PID");
		if (runscript_pid && sscanf(runscript_pid, "%d", &pid) == 1)
			if (kill(pid, SIGHUP) != 0)
				eerror("%s: failed to signal parent %d: %s",
					applet, pid, strerror(errno));

		/* Remove the exclusive time test. This ensures that it's not
		   in control as well */
		l = strlen(RC_SVCDIR "/exclusive") + strlen(svcname) +
			strlen(runscript_pid) + 4;
		mtime = xmalloc(l);
		snprintf(mtime, l, RC_SVCDIR "/exclusive/%s.%s",
			 svcname, runscript_pid);
		if (exists(mtime) && unlink(mtime) != 0)
			eerror("%s: unlink: %s", applet, strerror(errno));
		free(mtime);
	}

	return ok ? EXIT_SUCCESS : EXIT_FAILURE;
}

static int do_value(int argc, char **argv)
{
	bool ok = false;
	char *service = getenv("RC_SVCNAME");
	char *option;

	if (! service)
		eerrorx("%s: no service specified", applet);

	if (argc < 2 || ! argv[1] || *argv[1] == '\0')
		eerrorx("%s: no option specified", applet);

	if (strcmp(applet, "service_get_value") == 0 ||
	    strcmp(applet, "get_options") == 0)
	{
		option = rc_service_value_get(service, argv[1]);
		if (option) {
			printf("%s", option);
			free(option);
			ok = true;
		}
	} else if (strcmp(applet, "service_set_value") == 0 ||
		   strcmp(applet, "save_options") == 0)
		ok = rc_service_value_set(service, argv[1], argv[2]);
	else
		eerrorx("%s: unknown applet", applet);

	return ok ? EXIT_SUCCESS : EXIT_FAILURE;
}

static int do_shell_var(int argc, char **argv)
{
	int i;
	char *p;
	int c;

	for (i = 1; i < argc; i++) {
		p = argv[i];

		if (i != 1)
			putchar(' ');

		while (*p) {
			c = (unsigned char)*p++;
			if (! isalnum(c))
				c = '_';
			putchar(c);
		}
	}
	putchar('\n');

	return EXIT_SUCCESS;
}

void run_applets(int argc, char **argv)
{
	int i = 2;
	char *p;
	pid_t pid = 0;

	/* These are designed to be applications in their own right */
	if (strcmp(applet, "fstabinfo") == 0)
		exit(fstabinfo(argc, argv));
	else if (strcmp(applet, "mountinfo") == 0)
		exit(mountinfo(argc, argv));
	else if (strcmp(applet, "rc-depend") == 0)
		exit(rc_depend(argc, argv));
	else if (strcmp(applet, "rc-service") == 0)
		exit(rc_service(argc, argv));
	else if (strcmp(applet, "rc-status") == 0)
		exit(rc_status(argc, argv));
	else if (strcmp(applet, "rc-update") == 0 ||
		 strcmp(applet, "update-rc") == 0)
		exit(rc_update(argc, argv));
	else if (strcmp(applet, "runscript") == 0)
		exit(runscript(argc, argv));
	else if (strcmp(applet, "start-stop-daemon") == 0)
		exit(start_stop_daemon(argc, argv));
	else if (strcmp (applet, "checkpath") == 0)
		exit(checkpath(argc, argv));

	/* These could also be applications in their own right */
	if (strcmp(applet, "shell_var") == 0)
		exit(do_shell_var(argc, argv));

	/* This test is perverted - historically the baselayout function
	 * returns 0 on *failure*, which is plain wrong */
	if (strcmp(applet, "is_older_than") == 0) {
		if (argc < 3)
			exit (EXIT_FAILURE);
		while (i < argc) {
			if (!rc_newer_than(argv[1], argv[i++]))
				exit(EXIT_SUCCESS);
		}
		exit(EXIT_FAILURE);
	};
	/* This tets is correct as it's not present in baselayout */
	if (strcmp(applet, "is_newer_than") == 0) {
		if (argc < 3)
			exit (EXIT_FAILURE);
		while (i < argc) {
			if (!rc_newer_than(argv[1], argv[i++]))
				exit(EXIT_FAILURE);
		}
		exit(EXIT_SUCCESS);
	}; 

	if (applet[0] == 'e' || (applet[0] == 'v' && applet[1] == 'e'))
		exit(do_e(argc, argv));

	/* These are purely for init scripts and do not make sense as
	 * anything else */
	if (strcmp(applet, "service_get_value") == 0 ||
	    strcmp(applet, "service_set_value") == 0 ||
	    strcmp(applet, "get_options") == 0 ||
	    strcmp(applet, "save_options") == 0)
		exit(do_value(argc, argv));

	if (strncmp(applet, "service_", strlen("service_")) == 0)
		exit(do_service(argc, argv));

	if (strncmp(applet, "mark_service_", strlen("mark_service_")) == 0)
		exit(do_mark_service(argc, argv));

	if (strcmp(applet, "is_runlevel_start") == 0)
		exit(rc_runlevel_starting() ? 0 : 1);
	else if (strcmp (applet, "is_runlevel_stop") == 0)
		exit(rc_runlevel_stopping() ? 0 : 1);

	if (strcmp(applet, "rc-abort") == 0) {
		p = getenv("RC_PID");
		if (p && sscanf(p, "%d", &pid) == 1) {
			if (kill(pid, SIGUSR1) != 0)
				eerrorx("rc-abort: failed to signal parent %d: %s",
					 pid, strerror(errno));
			exit(EXIT_SUCCESS);
		}
		exit(EXIT_FAILURE);
	}

	if (strcmp(applet, "rc") != 0)
		eerrorx("%s: unknown applet", applet);
}