1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*
22 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23 * Use is subject to license terms.
24 */
25
26#include	<errno.h>
27#include	<stdio.h>
28#include	<stdlib.h>
29#include	<string.h>
30#include	<libgen.h>
31#include	<sys/param.h>
32#include	<sys/priocntl.h>
33#include	<sys/types.h>
34
35#include	"priocntl.h"
36
37static char usage[] =
38"usage:	priocntl -l\n\
39	priocntl -d [-i idtype] [idlist]\n";
40
41/*
42 * A whole lot of to-do for a scheduling class that can't actually be
43 * configured or used by user processes.
44 */
45int
46main(int argc, char *argv[])
47{
48	int	dflag, eflag, iflag, lflag, sflag;
49	int	c;
50	char	cmdpath[MAXPATHLEN];
51	char	basenm[BASENMSZ];
52
53	(void) strlcpy(cmdpath, argv[0], MAXPATHLEN);
54	(void) strlcpy(basenm, basename(argv[0]), BASENMSZ);
55
56	dflag = eflag = iflag = lflag = sflag = 0;
57	while ((c = getopt(argc, argv, "c:dei:ls")) != -1) {
58		switch (c) {
59
60		case 'c':
61			if (strcmp(optarg, "SDC") != 0)
62				fatalerr("error: %s executed for %s class, "
63				    "%s is actually sub-command for %s class\n",
64				    cmdpath, optarg, cmdpath, "SDC");
65			break;
66
67		case 'd':
68			dflag++;
69			break;
70
71		case 'e':
72			eflag++;
73			break;
74
75		case 'i':
76			iflag++;	/* optarg is parsed, but ignored */
77			break;
78
79		case 'l':
80			lflag++;
81			break;
82
83		case 's':
84			sflag++;
85			break;
86
87		case '?':
88			fatalerr(usage);
89			/*NOTREACHED*/
90
91		default:
92			break;
93		}
94	}
95
96	if (sflag && eflag) {
97		fatalerr(usage);
98	}
99	if (sflag || eflag) {
100		fatalerr(
101		    "priocntl: \"-%c\" may not be used with the %s class\n",
102		    (sflag ? 's' : 'e'), "SDC");
103	}
104
105	if ((!dflag && !lflag) || (dflag && lflag)) {
106		fatalerr(usage);
107	}
108
109	if (dflag) {
110		pid_t *pidlist;
111		size_t numread, i;
112
113		/*
114		 * No scheduling-class-specific information to print,
115		 * but we read the pidlist to avoid generating a SIGPIPE
116		 * in the main priocntl process.  Once we've read it,
117		 * we might as well print it.
118		 */
119		if ((pidlist = read_pidlist(&numread, stdin)) == NULL) {
120			fatalerr("%s: Can't read pidlist.\n", basenm);
121		} else if (numread == 0) {
122			fatalerr("%s: No pids on input.\n", basenm);
123		} else {
124			(void) printf("SYSTEM DUTY-CYCLE PROCESSES:\n");
125			(void) printf("%7s\n", "PID");
126			for (i = 0; i < numread; i++) {
127				(void) printf("%7ld\n", pidlist[i]);
128			}
129		}
130		free_pidlist(pidlist);
131	} else {
132		(void) printf("SDC (System Duty-Cycle Class)\n");
133	}
134
135	return (0);
136}
137