/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright (c) 2001, 2010, Oracle and/or its affiliates. All rights reserved.
 */

#include <sys/rctl_impl.h>
#include <sys/types.h>
#include <sys/stat.h>

#include <errno.h>
#include <libintl.h>
#include <locale.h>
#include <rctl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <syslog.h>
#include <unistd.h>
#include <fcntl.h>

#include "utils.h"

#define	ENABLE	1
#define	DISABLE	0

#define	CONFIGPATH	"/etc/rctladm.conf"
#define	CONFIGOWNER	0	/* uid 0 (root) */
#define	CONFIGGROUP	1	/* gid 1 (other) */
#define	CONFIGPERM	(S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH) /* 0644 */

/*
 *	Macros to produce a quoted string containing the value of a
 *	preprocessor macro. For example, if SIZE is defined to be 256,
 *	VAL2STR(SIZE) is "256". This is used to construct format
 *	strings for scanf-family functions below.
 */
#define	QUOTE(x)	#x
#define	VAL2STR(x)	QUOTE(x)

static const char USAGE[] =
	"Usage:\trctladm -l\n"
	"\trctladm -u\n"
	"\trctladm -e actions -d actions rctl_name\n";

static const char OPTS[] = "d:e:lu";
static int dflg, eflg, lflg, uflg;

static uint_t op_failures;

static void rctladm_enable(const char *, char *);

#define	BUFSIZE 256

static void
usage()
{
	(void) fprintf(stderr, gettext(USAGE));
	exit(E_USAGE);
}

#define	LOG_HIGHEST LOG_DEBUG
static const char *syslog_priorities[] = {
	"emerg",	/* LOG_EMERG	*/
	"alert",	/* LOG_ALERT	*/
	"crit",		/* LOG_CRIT	*/
	"err",		/* LOG_ERR	*/
	"warning",	/* LOG_WARNING	*/
	"notice",	/* LOG_NOTICE	*/
	"info",		/* LOG_INFO	*/
	"debug"		/* LOG_DEBUG	*/
};

static int
rctladm_syslog_prio(const char *priority)
{
	uint_t i;

	for (i = 0; i < LOG_HIGHEST + 1; i++) {
		if ((strcasecmp(priority, syslog_priorities[i]) == 0))
			return (i);
	}

	die(gettext("unknown syslog priority \"%s\"\n"), priority);

	/*NOTREACHED*/
}

/*ARGSUSED*/
static int
rctl_save_walk_cb(const char *rctl_name, void *file)
{
	FILE *fp = file;
	rctlblk_t *gblk;
	uint_t action;
	rctl_opaque_t *gopq;

	if ((gblk = malloc(rctlblk_size())) == NULL)
		die(gettext("unable to allocate control block"));


	if (rctlctl(rctl_name, gblk, RCTLCTL_GET) == -1) {
		warn(gettext("unable to obtain control block contents for %s"),
		    rctl_name);
	} else {
		action = rctlblk_get_global_action(gblk);
		gopq = (rctl_opaque_t *)gblk;

		(void) fprintf(fp, "%s=", rctl_name);
		if (action & RCTL_GLOBAL_SYSLOG)
			(void) fprintf(fp, "syslog=%s\n",
			    syslog_priorities[gopq->rcq_global_syslog_level]);
		else
			(void) fprintf(fp, "none\n");
	}

	free(gblk);

	return (0);
}

static void
rctladm_save_config()
{
	int fd;
	FILE *fp;

	/*
	 * Non-root users shouldn't update the configuration file.
	 */
	if (geteuid() != 0)
		return;

	if ((fd = open(CONFIGPATH, O_WRONLY|O_CREAT|O_TRUNC, CONFIGPERM)) == -1)
		die(gettext("failed to open %s"), CONFIGPATH);

	if ((fp = fdopen(fd, "w")) == NULL)
		die(gettext("failed to open stream for %s"), CONFIGPATH);

	(void) fputs(
	    "#\n"
	    "# rctladm.conf\n"
	    "#\n"
	    "# Parameters for resource controls configuration.\n"
	    "# Do NOT edit this file by hand -- use rctladm(8) instead.\n"
	    "#\n",
	    fp);

	(void) rctl_walk(rctl_save_walk_cb, fp);

	(void) fflush(fp);
	(void) fsync(fd);
	(void) fchmod(fd, CONFIGPERM);
	(void) fchown(fd, CONFIGOWNER, CONFIGGROUP);
	(void) fclose(fp);
}

static void
rctladm_setup_action(char *name, char *action, int line)
{
	if (action[0] == '\0') {
		warn(gettext("\"%s\", line %d, syntax error\n"), CONFIGPATH,
		    line);
		return;
	}
	rctladm_enable(name, action);
}

static void
rctladm_read_config()
{
	int fd;
	FILE *fp;
	char buf[BUFSIZE];
	char name[BUFSIZE+1], actions[BUFSIZE+1];
	char *action;
	int line, len, n;
	rctl_opaque_t *gblk;

	/*
	 * Non-root users shouldn't do this.
	 */
	if (geteuid() != 0)
		die(gettext("you must be root to use this option\n"));

	if ((fd = open(CONFIGPATH, O_RDONLY, CONFIGPERM)) == -1)
		die(gettext("failed to open %s"), CONFIGPATH);

	if ((fp = fdopen(fd, "r")) == NULL)
		die(gettext("failed to open stream for %s"), CONFIGPATH);

	if ((gblk = malloc(rctlblk_size())) == NULL)
		die(gettext("unable to allocate control block"));

	for (line = 1; fgets(buf, BUFSIZE, fp) != NULL; line++) {
		/*
		 * Skip comment lines and empty lines.
		 */
		if (buf[0] == '#' || buf[0] == '\n')
			continue;

		/*
		 * Look for "rctl_name=action;action;...;action, with
		 * optional whitespace on either side, terminated by a newline,
		 * and consuming the whole line.
		 */
		n = sscanf(buf,
		    " %" VAL2STR(BUFSIZE) "[^=]=%" VAL2STR(BUFSIZE) "s \n%n",
		    name, actions, &len);
		if (n >= 1 && name[0] != '\0' &&
		    (n == 1 || len == strlen(buf))) {
			if (n == 1) {
				warn(gettext("\"%s\", line %d, syntax error\n"),
				    CONFIGPATH, line);
				continue;
			}
			if (rctlctl(name, (rctlblk_t *)gblk,
			    RCTLCTL_GET) == -1) {
				warn(gettext("\"%s\", line %d, unknown resource"
				    " control: %s\n"), CONFIGPATH, line, name);
				continue;
			}
			if (actions[0] == ';') {
				warn(gettext("\"%s\", line %d, syntax error\n"),
				    CONFIGPATH, line);
				continue;
			}
			action = strtok(actions, ";");
			rctladm_setup_action(name, action, line);
			while (action = strtok(NULL, ";"))
				rctladm_setup_action(name, action, line);
		}
	}

	if (line == 1)
		die(gettext("failed to read rctl configuration from \"%s\""),
		    CONFIGPATH);
	free(gblk);
	(void) fclose(fp);
}

static void
rctladm_modify_action(const char *rctl_name, uint_t enable, uint_t action,
    int log_level)
{
	rctl_opaque_t *gblk;

	if ((gblk = malloc(rctlblk_size())) == NULL)
		die(gettext("unable to allocate control block"));

	if (rctlctl(rctl_name, (rctlblk_t *)gblk, RCTLCTL_GET) == -1)
		die(gettext("unable to obtain resource control block"));

	if ((gblk->rcq_global_flagaction & RCTL_GLOBAL_SYSLOG_NEVER) &&
	    (action == RCTL_GLOBAL_SYSLOG)) {
		warn(gettext("\"syslog\" action not valid for %s\n"),
		    rctl_name);
		op_failures++;
		free(gblk);
		return;
	}

	if (enable) {
		gblk->rcq_global_flagaction |= (action &
		    ~RCTL_GLOBAL_ACTION_MASK);
		gblk->rcq_global_syslog_level = log_level;
	} else {
		gblk->rcq_global_flagaction &= ~(action &
		    ~RCTL_GLOBAL_ACTION_MASK);
		gblk->rcq_global_syslog_level = LOG_NOTICE;
	}

	if (rctlctl(rctl_name, (rctlblk_t *)gblk, RCTLCTL_SET) == -1) {
		warn(gettext("unable to update control block contents"));
		op_failures++;
	}

	free(gblk);
}

static int
rctladm_get_log_level(char *action)
{
	char *log_lvl_str;

	/*
	 * Our syslog priority defaults to LOG_NOTICE.
	 */
	if (strcmp("syslog", action) == 0)
		return (LOG_NOTICE);

	if (strncmp("syslog=", action, strlen("syslog=")) != 0)
		die(gettext("unknown action \"%s\"\n"), action);

	log_lvl_str = action + strlen("syslog=");

	return (rctladm_syslog_prio(log_lvl_str));
}


static void
rctladm_enable(const char *rctl_name, char *action)
{
	/*
	 * Two valid values:  "none" and "syslog[=level]".
	 */
	if (strcmp("none", action) == 0) {
		rctladm_modify_action(rctl_name, DISABLE,
		    ~RCTL_GLOBAL_ACTION_MASK, 0);
		return;
	}

	rctladm_modify_action(rctl_name, ENABLE, RCTL_GLOBAL_SYSLOG,
	    rctladm_get_log_level(action));
}

static void
rctladm_disable(const char *rctl_name, char *action)
{
	/*
	 * Two valid values:  "all" and "syslog".
	 */
	if (strcmp("all", action) == 0) {
		rctladm_modify_action(rctl_name, DISABLE,
		    ~RCTL_GLOBAL_ACTION_MASK, 0);
		return;
	} else if (strcmp("syslog", action) == 0) {
		rctladm_modify_action(rctl_name, DISABLE, RCTL_GLOBAL_SYSLOG,
		    0);
		return;
	}

	die(gettext("unknown action \"%s\"\n"), action);
}

static void
rctlblk_display(FILE *f, rctlblk_t *gblk)
{
	uint_t action = rctlblk_get_global_action(gblk);
	uint_t flags = rctlblk_get_global_flags(gblk);
	rctl_opaque_t *gopq = (rctl_opaque_t *)gblk;

	if (flags & RCTL_GLOBAL_SYSLOG_NEVER)
		(void) fprintf(f, "syslog=n/a    ");
	else if (action & RCTL_GLOBAL_SYSLOG)
		(void) fprintf(f, "syslog=%-7s",
		    syslog_priorities[gopq->rcq_global_syslog_level]);
	else
		(void) fprintf(f, "syslog=off    ");

	if (flags & RCTL_GLOBAL_ACTION_MASK)
		(void) fprintf(f, " [");

	if (flags & RCTL_GLOBAL_NOBASIC)
		(void) fprintf(f, " no-basic");
	if (flags & RCTL_GLOBAL_LOWERABLE)
		(void) fprintf(f, " lowerable");
	if (flags & RCTL_GLOBAL_DENY_ALWAYS)
		(void) fprintf(f, " deny");
	if (flags & RCTL_GLOBAL_DENY_NEVER)
		(void) fprintf(f, " no-deny");
	if (flags & RCTL_GLOBAL_CPU_TIME)
		(void) fprintf(f, " cpu-time");
	if (flags & RCTL_GLOBAL_FILE_SIZE)
		(void) fprintf(f, " file-size");
	if (flags & RCTL_GLOBAL_SIGNAL_NEVER)
		(void) fprintf(f, " no-signal");
	if (flags & RCTL_GLOBAL_UNOBSERVABLE)
		(void) fprintf(f, " no-obs");
	if (flags & RCTL_GLOBAL_INFINITE)
		(void) fprintf(f, " inf");
	if (flags & RCTL_GLOBAL_SYSLOG_NEVER)
		(void) fprintf(f, " no-syslog");
	if (flags & RCTL_GLOBAL_SECONDS)
		(void) fprintf(f, " seconds");
	if (flags & RCTL_GLOBAL_BYTES)
		(void) fprintf(f, " bytes");
	if (flags & RCTL_GLOBAL_COUNT)
		(void) fprintf(f, " count");
	if (flags & RCTL_GLOBAL_ACTION_MASK)
		(void) fprintf(f, " ]");

	(void) fprintf(f, "\n");
}

/*ARGSUSED*/
static int
rctl_walk_cb(const char *rctl_name, void *pvt)
{
	rctlblk_t *gblk;

	if ((gblk = malloc(rctlblk_size())) == NULL)
		die(gettext("unable to allocate control block"));

	if (rctlctl(rctl_name, gblk, RCTLCTL_GET) == -1) {
		if (errno == ESRCH)
			warn(gettext("unknown resource control: %s\n"),
			    rctl_name);
		else
			warn(gettext("unable to obtain %s properties"),
			    rctl_name);
		op_failures++;
	} else {
		(void) printf("%-27s ", rctl_name);
		rctlblk_display(stdout, gblk);
	}

	free(gblk);

	return (0);
}

static void
rctladm_list_rctls(int optind, int argc, char *argv[])
{
	if (optind >= argc) {
		(void) rctl_walk(rctl_walk_cb, NULL);
		return;
	}

	for (; optind < argc; optind++)
		(void) rctl_walk_cb(argv[optind], NULL);
}

int
main(int argc, char *argv[])
{
	int c;			/* options character */
	char *action;
	char *rctl;

	(void) setlocale(LC_ALL, "");
	(void) textdomain(TEXT_DOMAIN);
	(void) setpname(argv[0]);

	while ((c = getopt(argc, argv, OPTS)) != EOF) {
		switch (c) {
			case 'd':
				dflg++;
				action = optarg;
				break;
			case 'e':
				eflg++;
				action = optarg;
				break;
			case 'l':
				lflg = 1;
				break;
			case 'u':
				uflg = 1;
				break;
			case '?':
			default:
				usage();
		}
	}

	if (uflg) {
		rctladm_read_config();
		return (E_SUCCESS);
	}

	if (lflg && (dflg || eflg)) {
		warn(gettext("-l, -d, and -e flags are exclusive\n"));
		usage();
	}

	if (dflg && eflg) {
		warn(gettext("-d and -e flags are exclusive\n"));
		usage();
	}

	if (dflg > 1 || eflg > 1) {
		warn(gettext("only one -d or -e flag per line\n"));
		usage();
	}

	if (lflg || !(dflg || eflg)) {
		rctladm_list_rctls(optind, argc, argv);
		rctladm_save_config();

		return (op_failures ? E_ERROR : E_SUCCESS);
	}

	if (optind >= argc) {
		warn(gettext("must specify one or more "
		    "resource control names\n"));
		usage();
	}

	for (; optind < argc; optind++) {
		rctl = argv[optind];

		if (eflg) {
			rctladm_enable(rctl, action);
			rctladm_save_config();
		} else if (dflg) {
			rctladm_disable(rctl, action);
			rctladm_save_config();
		} else {
			usage();
		}
	}

	return (op_failures ? E_ERROR : E_SUCCESS);
}