/*
 * 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 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <assert.h>
#include <errno.h>
#include <libintl.h>
#include <sys/wait.h>
#include <sys/ctfs.h>
#include <sys/contract/process.h>
#include <libcontract.h>
#include <libcontract_priv.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "inetd_impl.h"


/* paths/filenames of contract related files */
#define	CONTRACT_ROOT_PATH	CTFS_ROOT "/process/"
#define	CONTRACT_TEMPLATE_PATH  CONTRACT_ROOT_PATH "template"

static int active_tmpl_fd = -1;

/*
 * Creates and configures the the contract template used for all inetd's
 * methods.
 * Returns -1 on error, else the fd of the created template.
 */
static int
create_contract_template(void)
{
	int		fd;
	int		err;

	if ((fd = open(CONTRACT_TEMPLATE_PATH, O_RDWR)) == -1) {
		error_msg(gettext("Failed to open contract file %s: %s"),
		    CONTRACT_TEMPLATE_PATH, strerror(errno));
		return (-1);
	}

	/*
	 * Make contract inheritable and make hardware errors fatal.
	 * We also limit the scope of fatal events to the process
	 * group.  In order of preference we would have contract-aware
	 * login services or a property indicating which services need
	 * such scoping, but for the time being we'll assume that most
	 * non login-style services run in a single process group.
	 */
	if (((err = ct_pr_tmpl_set_param(fd,
	    CT_PR_INHERIT|CT_PR_PGRPONLY)) != 0) ||
	    ((err = ct_pr_tmpl_set_fatal(fd, CT_PR_EV_HWERR)) != 0) ||
	    ((err = ct_tmpl_set_critical(fd, 0)) != 0) ||
	    ((err = ct_tmpl_set_informative(fd, 0)) != 0)) {
		error_msg(gettext(
		    "Failed to set parameter for contract template: %s"),
		    strerror(err));
		(void) close(fd);
		return (-1);
	}

	return (fd);
}

/* Returns -1 on error, else 0. */
int
contract_init(void)
{
	if ((active_tmpl_fd = create_contract_template()) == -1) {
		error_msg(gettext("Failed to create contract template"));
		return (-1);
	}
	return (0);
}

void
contract_fini(void)
{
	if (active_tmpl_fd != -1) {
		(void) close(active_tmpl_fd);
		active_tmpl_fd = -1;
	}
}

/*
 * To be called directly before a service method is forked, this function
 * results in the method process being in a new contract based on the active
 * contract template.
 */
int
contract_prefork(const char *fmri, int method)
{
	int err;

	if ((err = ct_pr_tmpl_set_svc_fmri(active_tmpl_fd, fmri)) != 0) {
		error_msg(gettext("Failed to set svc_fmri term: %s"),
		    strerror(err));
		return (-1);
	}
	if ((err = ct_pr_tmpl_set_svc_aux(active_tmpl_fd,
	    methods[method].name)) != 0) {
		error_msg(gettext("Failed to set svc_aux term: %s"),
		    strerror(err));
		return (-1);
	}

	if ((err = ct_tmpl_activate(active_tmpl_fd)) != 0) {
		error_msg(gettext("Failed to activate contract template: %s"),
		    strerror(err));
		return (-1);
	}
	return (0);
}

/*
 * To be called in both processes directly after a service method is forked,
 * this function results in switching off contract creation for any
 * forks done by either process, unless contract_prefork() is called beforehand.
 */
void
contract_postfork(void)
{
	int err;

	if ((err = ct_tmpl_clear(active_tmpl_fd)) != 0)
		error_msg("Failed to clear active contract template: %s",
		    strerror(err));
}

/*
 * Fetch the latest created contract id into the space referenced by 'cid'.
 * Returns -1 on error, else 0.
 */
int
get_latest_contract(ctid_t *cid)
{
	if ((errno = contract_latest(cid)) != 0) {
		error_msg(gettext("Failed to get new contract's id: %s"),
		    strerror(errno));
		return (-1);
	}

	return (0);
}

/* Returns -1 on error (with errno set), else fd. */
static int
open_contract_ctl_file(ctid_t cid)
{
	return (contract_open(cid, "process", "ctl", O_WRONLY));
}

/*
 * Adopt a contract.  Emits an error message and returns -1 on failure, else
 * 0.
 */
int
adopt_contract(ctid_t ctid, const char *fmri)
{
	int fd;
	int err;
	int ret = 0;

	if ((fd = open_contract_ctl_file(ctid)) == -1) {
		if (errno == EACCES || errno == ENOENT) {
			/*
			 * We must not have inherited this contract.  That can
			 * happen if we were disabled and restarted.
			 */
			debug_msg("Could not adopt contract %ld for %s "
			    "(could not open ctl file: permission denied).\n",
			    ctid, fmri);
			return (-1);
		}

		error_msg(gettext("Could not adopt contract id %ld registered "
		    "with %s (could not open ctl file: %s).  Events will be "
		    "ignored."), ctid, fmri, strerror(errno));
		return (-1);
	}

	if ((err = ct_ctl_adopt(fd)) != 0) {
		error_msg(gettext("Could not adopt contract id %ld registered "
		    "with %s (%s).  Events will be ignored."), ctid, fmri,
		    strerror(err));
		ret = -1;
	}

	err = close(fd);
	if (err != 0)
		error_msg(gettext("Could not close file descriptor %d."), fd);

	return (ret);
}

/* Returns -1 on error, else 0. */
int
abandon_contract(ctid_t ctid)
{
	int fd;
	int err;

	assert(ctid != -1);

	if ((fd = open_contract_ctl_file(ctid)) == -1) {
		error_msg(gettext("Failed to abandon contract %d: %s"), ctid,
		    strerror(errno));
		return (-1);
	}

	if ((err = ct_ctl_abandon(fd)) != 0) {
		(void) close(fd);
		error_msg(gettext("Failed to abandon contract %d: %s"), ctid,
		    strerror(err));
		return (-1);
	}

	(void) close(fd);

	return (0);
}