[openssh.git] / openbsd-compat / port-solaris.c

/* $Id$ */

/*
 * Copyright (c) 2006 Chad Mynhier.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include "config.h"
#include "includes.h"

#ifdef USE_SOLARIS_PROCESS_CONTRACTS

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

#include <errno.h>
#ifdef HAVE_FCNTL_H
# include <fcntl.h>
#endif
#include <stdarg.h>
#include <string.h>
#include <unistd.h>

#include <libcontract.h>
#include <sys/contract/process.h>
#include <sys/ctfs.h>

#include "log.h"

#define CT_TEMPLATE	CTFS_ROOT "/process/template"
#define CT_LATEST	CTFS_ROOT "/process/latest"

static int tmpl_fd = -1;

/* Lookup the latest process contract */
static ctid_t
get_active_process_contract_id(void)
{
	int stat_fd;
	ctid_t ctid = -1;
	ct_stathdl_t stathdl;

	if ((stat_fd = open64(CT_LATEST, O_RDONLY)) == -1) {
		error("%s: Error opening 'latest' process "
		    "contract: %s", __func__, strerror(errno));
		return -1;
	}
	if (ct_status_read(stat_fd, CTD_COMMON, &stathdl) != 0) {
		error("%s: Error reading process contract "
		    "status: %s", __func__, strerror(errno));
		goto out;
	}
	if ((ctid = ct_status_get_id(stathdl)) < 0) {
		error("%s: Error getting process contract id: %s",
		    __func__, strerror(errno));
		goto out;
	}

	ct_status_free(stathdl);
 out:
	close(stat_fd);
	return ctid;
}

void
solaris_contract_pre_fork(void)
{
	if ((tmpl_fd = open64(CT_TEMPLATE, O_RDWR)) == -1) {
		error("%s: open %s: %s", __func__,
		    CT_TEMPLATE, strerror(errno));
		return;
	}

	debug2("%s: setting up process contract template on fd %d",
	    __func__, tmpl_fd);

	/* First we set the template parameters and event sets. */
	if (ct_pr_tmpl_set_param(tmpl_fd, CT_PR_PGRPONLY) != 0) {
		error("%s: Error setting process contract parameter set "
		    "(pgrponly): %s", __func__, strerror(errno));
		goto fail;
	}
	if (ct_pr_tmpl_set_fatal(tmpl_fd, CT_PR_EV_HWERR) != 0) {
		error("%s: Error setting process contract template "
		    "fatal events: %s", __func__, strerror(errno));
		goto fail;
	}
	if (ct_tmpl_set_critical(tmpl_fd, 0) != 0) {
		error("%s: Error setting process contract template "
		    "critical events: %s", __func__, strerror(errno));
		goto fail;
	}
	if (ct_tmpl_set_informative(tmpl_fd, CT_PR_EV_HWERR) != 0) {
		error("%s: Error setting process contract template "
		    "informative events: %s", __func__, strerror(errno));
		goto fail;
	}

	/* Now make this the active template for this process. */
	if (ct_tmpl_activate(tmpl_fd) != 0) {
		error("%s: Error activating process contract "
		    "template: %s", __func__, strerror(errno));
		goto fail;
	}
	return;

 fail:
	if (tmpl_fd != -1) {
		close(tmpl_fd);
		tmpl_fd = -1;
	}
}

void
solaris_contract_post_fork_child()
{
	debug2("%s: clearing process contract template on fd %d",
	    __func__, tmpl_fd);

	/* Clear the active template. */
	if (ct_tmpl_clear(tmpl_fd) != 0)
		error("%s: Error clearing active process contract "
		    "template: %s", __func__, strerror(errno));

	close(tmpl_fd);
	tmpl_fd = -1;
}

void
solaris_contract_post_fork_parent(pid_t pid)
{
	ctid_t ctid;
	char ctl_path[256];
	int r, ctl_fd = -1, stat_fd = -1;

	debug2("%s: clearing template (fd %d)", __func__, tmpl_fd);

	if (tmpl_fd == -1)
		return;

	/* First clear the active template. */
	if ((r = ct_tmpl_clear(tmpl_fd)) != 0)
		error("%s: Error clearing active process contract "
		    "template: %s", __func__, strerror(errno));

	close(tmpl_fd);
	tmpl_fd = -1;

	/*
	 * If either the fork didn't succeed (pid < 0), or clearing
	 * th active contract failed (r != 0), then we have nothing
	 * more do.
	 */
	if (r != 0 || pid <= 0)
		return;

	/* Now lookup and abandon the contract we've created. */
	ctid = get_active_process_contract_id();

	debug2("%s: abandoning contract id %ld", __func__, ctid);

	snprintf(ctl_path, sizeof(ctl_path),
	    CTFS_ROOT "/process/%ld/ctl", ctid);
	if ((ctl_fd = open64(ctl_path, O_WRONLY)) < 0) {
		error("%s: Error opening process contract "
		    "ctl file: %s", __func__, strerror(errno));
		goto fail;
	}
	if (ct_ctl_abandon(ctl_fd) < 0) {
		error("%s: Error abandoning process contract: %s",
		    __func__, strerror(errno));
		goto fail;
	}
	close(ctl_fd);
	return;

 fail:
	if (tmpl_fd != -1) {
		close(tmpl_fd);
		tmpl_fd = -1;
	}
	if (stat_fd != -1)
		close(stat_fd);
	if (ctl_fd != -1)
		close(ctl_fd);
}
#endif
Commit	Line	Data
5b84789f	1	/* $Id$ */
	2
	3	/*
	4	* Copyright (c) 2006 Chad Mynhier.
	5	*
	6	* Permission to use, copy, modify, and distribute this software for any
	7	* purpose with or without fee is hereby granted, provided that the above
	8	* copyright notice and this permission notice appear in all copies.
	9	*
	10	* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
	11	* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
	12	* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
	13	* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
	14	* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
	15	* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
	16	* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
	17	*/
	18
	19	#include "config.h"
	20	#include "includes.h"
	21
	22	#ifdef USE_SOLARIS_PROCESS_CONTRACTS
	23
	24	#include <sys/types.h>
	25	#include <sys/stat.h>
	26	#include <sys/param.h>
	27
	28	#include <errno.h>
	29	#ifdef HAVE_FCNTL_H
	30	# include <fcntl.h>
	31	#endif
24436b92	32	#include <stdarg.h>
5b84789f	33	#include <string.h>
	34	#include <unistd.h>
	35
	36	#include <libcontract.h>
	37	#include <sys/contract/process.h>
	38	#include <sys/ctfs.h>
	39
	40	#include "log.h"
	41
	42	#define CT_TEMPLATE CTFS_ROOT "/process/template"
	43	#define CT_LATEST CTFS_ROOT "/process/latest"
	44
	45	static int tmpl_fd = -1;
	46
	47	/* Lookup the latest process contract */
	48	static ctid_t
	49	get_active_process_contract_id(void)
	50	{
	51	int stat_fd;
	52	ctid_t ctid = -1;
	53	ct_stathdl_t stathdl;
	54
	55	if ((stat_fd = open64(CT_LATEST, O_RDONLY)) == -1) {
	56	error("%s: Error opening 'latest' process "
	57	"contract: %s", __func__, strerror(errno));
	58	return -1;
	59	}
	60	if (ct_status_read(stat_fd, CTD_COMMON, &stathdl) != 0) {
	61	error("%s: Error reading process contract "
	62	"status: %s", __func__, strerror(errno));
	63	goto out;
	64	}
	65	if ((ctid = ct_status_get_id(stathdl)) < 0) {
	66	error("%s: Error getting process contract id: %s",
	67	__func__, strerror(errno));
	68	goto out;
	69	}
	70
	71	ct_status_free(stathdl);
	72	out:
	73	close(stat_fd);
	74	return ctid;
	75	}
	76
	77	void
	78	solaris_contract_pre_fork(void)
	79	{
	80	if ((tmpl_fd = open64(CT_TEMPLATE, O_RDWR)) == -1) {
	81	error("%s: open %s: %s", __func__,
	82	CT_TEMPLATE, strerror(errno));
	83	return;
	84	}
	85
	86	debug2("%s: setting up process contract template on fd %d",
	87	__func__, tmpl_fd);
	88
8852e1d4	89	/* First we set the template parameters and event sets. */
	90	if (ct_pr_tmpl_set_param(tmpl_fd, CT_PR_PGRPONLY) != 0) {
	91	error("%s: Error setting process contract parameter set "
	92	"(pgrponly): %s", __func__, strerror(errno));
	93	goto fail;
	94	}
	95	if (ct_pr_tmpl_set_fatal(tmpl_fd, CT_PR_EV_HWERR) != 0) {
5b84789f	96	error("%s: Error setting process contract template "
	97	"fatal events: %s", __func__, strerror(errno));
	98	goto fail;
	99	}
8852e1d4	100	if (ct_tmpl_set_critical(tmpl_fd, 0) != 0) {
5b84789f	101	error("%s: Error setting process contract template "
	102	"critical events: %s", __func__, strerror(errno));
	103	goto fail;
	104	}
8852e1d4	105	if (ct_tmpl_set_informative(tmpl_fd, CT_PR_EV_HWERR) != 0) {
	106	error("%s: Error setting process contract template "
	107	"informative events: %s", __func__, strerror(errno));
	108	goto fail;
	109	}
5b84789f	110
	111	/* Now make this the active template for this process. */
	112	if (ct_tmpl_activate(tmpl_fd) != 0) {
	113	error("%s: Error activating process contract "
	114	"template: %s", __func__, strerror(errno));
	115	goto fail;
	116	}
	117	return;
	118
	119	fail:
	120	if (tmpl_fd != -1) {
	121	close(tmpl_fd);
	122	tmpl_fd = -1;
	123	}
	124	}
	125
	126	void
	127	solaris_contract_post_fork_child()
	128	{
	129	debug2("%s: clearing process contract template on fd %d",
	130	__func__, tmpl_fd);
	131
	132	/* Clear the active template. */
	133	if (ct_tmpl_clear(tmpl_fd) != 0)
	134	error("%s: Error clearing active process contract "
	135	"template: %s", __func__, strerror(errno));
	136
	137	close(tmpl_fd);
	138	tmpl_fd = -1;
	139	}
	140
	141	void
	142	solaris_contract_post_fork_parent(pid_t pid)
	143	{
	144	ctid_t ctid;
	145	char ctl_path[256];
	146	int r, ctl_fd = -1, stat_fd = -1;
	147
	148	debug2("%s: clearing template (fd %d)", __func__, tmpl_fd);
	149
	150	if (tmpl_fd == -1)
	151	return;
	152
	153	/* First clear the active template. */
	154	if ((r = ct_tmpl_clear(tmpl_fd)) != 0)
	155	error("%s: Error clearing active process contract "
	156	"template: %s", __func__, strerror(errno));
	157
	158	close(tmpl_fd);
	159	tmpl_fd = -1;
	160
	161	/*
	162	* If either the fork didn't succeed (pid < 0), or clearing
	163	* th active contract failed (r != 0), then we have nothing
	164	* more do.
	165	*/
	166	if (r != 0 \|\| pid <= 0)
	167	return;
	168
	169	/* Now lookup and abandon the contract we've created. */
	170	ctid = get_active_process_contract_id();
	171
	172	debug2("%s: abandoning contract id %ld", __func__, ctid);
	173
174	snprintf(ctl_path, sizeof(ctl_path),
175	CTFS_ROOT "/process/%ld/ctl", ctid);
176	if ((ctl_fd = open64(ctl_path, O_WRONLY)) < 0) {
177	error("%s: Error opening process contract "
178	"ctl file: %s", __func__, strerror(errno));
179	goto fail;
180	}
181	if (ct_ctl_abandon(ctl_fd) < 0) {
182	error("%s: Error abandoning process contract: %s",
183	__func__, strerror(errno));
184	goto fail;
185	}
186	close(ctl_fd);
187	return;
188
189	fail:
190	if (tmpl_fd != -1) {
191	close(tmpl_fd);
192	tmpl_fd = -1;
193	}
194	if (stat_fd != -1)
195	close(stat_fd);
196	if (ctl_fd != -1)
197	close(ctl_fd);
198	}
199	#endif