port/gen/setpriority.c

	setpriority.c revision 7c478bd95313f5f23a4c958a745db2134aa03244
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*  Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/*    All Rights Reserved   */

/*
 * University Copyright- Copyright (c) 1982, 1986, 1988
 * The Regents of the University of California
 * All Rights Reserved
 *
 * University Acknowledgment- Portions of this document are derived from
 * software developed by the University of California, Berkeley, and its
 * contributors.
 */

#pragma ident   "%Z%%M% %I% %E% SMI"

#include "synonyms.h"

#include <string.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/procset.h>
#include <sys/priocntl.h>
#include <errno.h>

static idtype_t
prio_to_idtype(int which)
{
    switch (which) {

    case PRIO_PROCESS:
        return (P_PID);

    case PRIO_PGRP:
        return (P_PGID);

    case PRIO_USER:
        return (P_UID);

    case PRIO_GROUP:
        return (P_GID);

    case PRIO_SESSION:
        return (P_SID);

    case PRIO_LWP:
        return (P_LWPID);

    case PRIO_TASK:
        return (P_TASKID);

    case PRIO_PROJECT:
        return (P_PROJID);

    case PRIO_ZONE:
        return (P_ZONEID);

    case PRIO_CONTRACT:
        return (P_CTID);

    default:
        return (-1);
    }
}

static int
old_idtype(int which)
{
    switch (which) {
    case PRIO_PROCESS:
    case PRIO_PGRP:
    case PRIO_USER:
        return (1);
    default:
        return (0);
    }
}

int
getpriority(int which, id_t who)
{
    id_t id;
    idtype_t idtype;
    pcnice_t pcnice;

    if ((idtype = prio_to_idtype(which)) == -1) {
        errno = EINVAL;
        return (-1);
    }

    if (who < 0) {
        if (old_idtype(which)) {
            errno = EINVAL;
            return (-1);
        } else if (who != P_MYID) {
            errno = EINVAL;
            return (-1);
        }
    }

    /*
     * The POSIX standard requires that a 0 value for the who argument
     * should specify the current process, process group, or user.
     * For all other id types we can treat zero as normal id value.
     */
    if (who == 0 && old_idtype(which))
        id = P_MYID;
    else
        id = who;

    pcnice.pc_val = 0;
    pcnice.pc_op = PC_GETNICE;

    if (priocntl(idtype, id, PC_DONICE, (caddr_t)&pcnice) == -1)
        return (-1);
    else
        return (pcnice.pc_val);
}

int
setpriority(int which, id_t who, int prio)
{
    id_t id;
    idtype_t idtype;
    pcnice_t pcnice;

    if ((idtype = prio_to_idtype(which)) == -1) {
        errno = EINVAL;
        return (-1);
    }

    if (who < 0) {
        if (old_idtype(which)) {
            errno = EINVAL;
            return (-1);
        } else if (who != P_MYID) {
            errno = EINVAL;
            return (-1);
        }
    }

    if (who == 0 && old_idtype(which))
        id = P_MYID;
    else
        id = who;

    if (prio > 19)
        prio = 19;
    else if (prio < -20)
        prio = -20;

    pcnice.pc_val = prio;
    pcnice.pc_op = PC_SETNICE;

    return (priocntl(idtype, id, PC_DONICE, (caddr_t)&pcnice));
}