OpenSolaris_b135/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);
}