pr_getitimer.c

Sun Solaris 10 中的 DTrace 组件的源代码。请参看: http://www.sun.com/software/solaris/observability.jsp

/*
 * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only.
 * See the file usr/src/LICENSING.NOTICE in this distribution or
 * http://www.opensolaris.org/license/ for details.
 */

#pragma ident	"@(#)pr_getitimer.c	1.4	00/03/05 SMI"

#include <sys/isa_defs.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include "libproc.h"

/*
 * getitimer() system call -- executed by victim process.
 */
int
pr_getitimer(struct ps_prochandle *Pr, int which, struct itimerval *itv)
{
	sysret_t rval;		/* return value from getitimer() */
	argdes_t argd[2];	/* arg descriptors for getitimer() */
	argdes_t *adp;
	int error;
#ifdef _LP64
	int victim32 = (Pstatus(Pr)->pr_dmodel == PR_MODEL_ILP32);
	struct itimerval32 itimerval32;
#endif

	if (Pr == NULL)		/* no victim process */
		return (getitimer(which, itv));

	adp = &argd[0];		/* which argument */
	adp->arg_value = which;
	adp->arg_type = AT_BYVAL;
	adp->arg_inout = AI_INPUT;
	adp->arg_object = NULL;
	adp->arg_size = 0;

	adp++;			/* itv argument */
	adp->arg_value = 0;
	adp->arg_type = AT_BYREF;
	adp->arg_inout = AI_OUTPUT;
#ifdef _LP64
	if (victim32) {
		adp->arg_object = &itimerval32;
		adp->arg_size = sizeof (itimerval32);
	} else {
		adp->arg_object = itv;
		adp->arg_size = sizeof (*itv);
	}
#else	/* _LP64 */
	adp->arg_object = itv;
	adp->arg_size = sizeof (*itv);
#endif	/* _LP64 */

	error = Psyscall(Pr, &rval, SYS_getitimer, 2, &argd[0]);

	if (error) {
		errno = (error > 0)? error : ENOSYS;
		return (-1);
	}
#ifdef _LP64
	if (victim32) {
		ITIMERVAL32_TO_ITIMERVAL(itv, &itimerval32);
	}
#endif	/* _LP64 */
	return (rval.sys_rval1);
}

/*
 * setitimer() system call -- executed by victim process.
 */
int
pr_setitimer(struct ps_prochandle *Pr,
	int which, const struct itimerval *itv, struct itimerval *oitv)
{
	sysret_t rval;		/* return value from setitimer() */
	argdes_t argd[3];	/* arg descriptors for setitimer() */
	argdes_t *adp;
	int error;
#ifdef _LP64
	int victim32 = (Pstatus(Pr)->pr_dmodel == PR_MODEL_ILP32);
	struct itimerval32 itimerval32;
	struct itimerval32 oitimerval32;
#endif	/* _LP64 */

	if (Pr == NULL)		/* no victim process */
		return (setitimer(which, (struct itimerval *)itv, oitv));

	adp = &argd[0];		/* which argument */
	adp->arg_value = which;
	adp->arg_type = AT_BYVAL;
	adp->arg_inout = AI_INPUT;
	adp->arg_object = NULL;
	adp->arg_size = 0;

	adp++;			/* itv argument */
	adp->arg_value = 0;
	adp->arg_type = AT_BYREF;
	adp->arg_inout = AI_INPUT;
#ifdef _LP64
	if (victim32) {
		ITIMERVAL_TO_ITIMERVAL32(&itimerval32, itv);
		adp->arg_object = (void *)&itimerval32;
		adp->arg_size = sizeof (itimerval32);
	} else {
		adp->arg_object = (void *)itv;
		adp->arg_size = sizeof (*itv);
	}
#else	/* _LP64 */
	adp->arg_object = (void *)itv;
	adp->arg_size = sizeof (*itv);
#endif	/* _LP64 */

	adp++;			/* oitv argument */
	adp->arg_value = 0;
	if (oitv == NULL) {
		adp->arg_type = AT_BYVAL;
		adp->arg_inout = AI_INPUT;
		adp->arg_object = NULL;
		adp->arg_size = 0;
	} else {
		adp->arg_type = AT_BYREF;
		adp->arg_inout = AI_OUTPUT;
#ifdef _LP64
		if (victim32) {
			adp->arg_object = (void *)&oitimerval32;
			adp->arg_size = sizeof (oitimerval32);
		} else {
			adp->arg_object = oitv;
			adp->arg_size = sizeof (*oitv);
		}
#else	/* _LP64 */
		adp->arg_object = oitv;
		adp->arg_size = sizeof (*oitv);
#endif	/* _LP64 */
	}

	error = Psyscall(Pr, &rval, SYS_setitimer, 3, &argd[0]);

	if (error) {
		errno = (error > 0)? error : ENOSYS;
		return (-1);
	}
#ifdef _LP64
	if (victim32 && oitv != NULL) {
		ITIMERVAL32_TO_ITIMERVAL(oitv, &oitimerval32);
	}
#endif	/* _LP64 */
	return (rval.sys_rval1);
}