cmd/priocntl/sdcpriocntl.c

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

#include    <errno.h>
#include    <stdio.h>
#include    <stdlib.h>
#include    <string.h>
#include    <libgen.h>
#include    <sys/param.h>
#include    <sys/priocntl.h>
#include    <sys/types.h>

#include    "priocntl.h"

static char usage[] =
"usage: priocntl -l\n\
    priocntl -d [-i idtype] [idlist]\n";

/*
 * A whole lot of to-do for a scheduling class that can't actually be
 * configured or used by user processes.
 */
int
main(int argc, char *argv[])
{
    int dflag, eflag, iflag, lflag, sflag;
    int c;
    char    cmdpath[MAXPATHLEN];
    char    basenm[BASENMSZ];

    (void) strlcpy(cmdpath, argv[0], MAXPATHLEN);
    (void) strlcpy(basenm, basename(argv[0]), BASENMSZ);

    dflag = eflag = iflag = lflag = sflag = 0;
    while ((c = getopt(argc, argv, "c:dei:ls")) != -1) {
        switch (c) {

        case 'c':
            if (strcmp(optarg, "SDC") != 0)
                fatalerr("error: %s executed for %s class, "
                    "%s is actually sub-command for %s class\n",
                    cmdpath, optarg, cmdpath, "SDC");
            break;

        case 'd':
            dflag++;
            break;

        case 'e':
            eflag++;
            break;

        case 'i':
            iflag++;    /* optarg is parsed, but ignored */
            break;

        case 'l':
            lflag++;
            break;

        case 's':
            sflag++;
            break;

        case '?':
            fatalerr(usage);
            /*NOTREACHED*/

        default:
            break;
        }
    }

    if (sflag && eflag) {
        fatalerr(usage);
    }
    if (sflag || eflag) {
        fatalerr(
            "priocntl: \"-%c\" may not be used with the %s class\n",
            (sflag ? 's' : 'e'), "SDC");
    }

    if ((!dflag && !lflag) || (dflag && lflag)) {
        fatalerr(usage);
    }

    if (dflag) {
        pid_t *pidlist;
        size_t numread, i;

        /*
         * No scheduling-class-specific information to print,
         * but we read the pidlist to avoid generating a SIGPIPE
         * in the main priocntl process.  Once we've read it,
         * we might as well print it.
         */
        if ((pidlist = read_pidlist(&numread, stdin)) == NULL) {
            fatalerr("%s: Can't read pidlist.\n", basenm);
        } else if (numread == 0) {
            fatalerr("%s: No pids on input.\n", basenm);
        } else {
            (void) printf("SYSTEM DUTY-CYCLE PROCESSES:\n");
            (void) printf("%7s\n", "PID");
            for (i = 0; i < numread; i++) {
                (void) printf("%7ld\n", pidlist[i]);
            }
        }
        free_pidlist(pidlist);
    } else {
        (void) printf("SDC (System Duty-Cycle Class)\n");
    }

    return (0);
}