1*7c478bd9Sstevel@tonic-gate /*
2*7c478bd9Sstevel@tonic-gate * CDDL HEADER START
3*7c478bd9Sstevel@tonic-gate *
4*7c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the
5*7c478bd9Sstevel@tonic-gate * Common Development and Distribution License, Version 1.0 only
6*7c478bd9Sstevel@tonic-gate * (the "License"). You may not use this file except in compliance
7*7c478bd9Sstevel@tonic-gate * with the License.
8*7c478bd9Sstevel@tonic-gate *
9*7c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*7c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing.
11*7c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions
12*7c478bd9Sstevel@tonic-gate * and limitations under the License.
13*7c478bd9Sstevel@tonic-gate *
14*7c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each
15*7c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*7c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the
17*7c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying
18*7c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner]
19*7c478bd9Sstevel@tonic-gate *
20*7c478bd9Sstevel@tonic-gate * CDDL HEADER END
21*7c478bd9Sstevel@tonic-gate */
22*7c478bd9Sstevel@tonic-gate /*
23*7c478bd9Sstevel@tonic-gate * Copyright 2001-2002 Sun Microsystems, Inc. All rights reserved.
24*7c478bd9Sstevel@tonic-gate * Use is subject to license terms.
25*7c478bd9Sstevel@tonic-gate */
26*7c478bd9Sstevel@tonic-gate
27*7c478bd9Sstevel@tonic-gate /* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
28*7c478bd9Sstevel@tonic-gate /* All Rights Reserved */
29*7c478bd9Sstevel@tonic-gate
30*7c478bd9Sstevel@tonic-gate #include <stdio.h>
31*7c478bd9Sstevel@tonic-gate #include <string.h>
32*7c478bd9Sstevel@tonic-gate #include <libgen.h>
33*7c478bd9Sstevel@tonic-gate #include <unistd.h>
34*7c478bd9Sstevel@tonic-gate #include <sys/types.h>
35*7c478bd9Sstevel@tonic-gate #include <sys/procset.h>
36*7c478bd9Sstevel@tonic-gate #include <sys/priocntl.h>
37*7c478bd9Sstevel@tonic-gate #include <sys/fxpriocntl.h>
38*7c478bd9Sstevel@tonic-gate #include <limits.h>
39*7c478bd9Sstevel@tonic-gate #include <errno.h>
40*7c478bd9Sstevel@tonic-gate
41*7c478bd9Sstevel@tonic-gate #include "priocntl.h"
42*7c478bd9Sstevel@tonic-gate
43*7c478bd9Sstevel@tonic-gate /*
44*7c478bd9Sstevel@tonic-gate * This file contains the class specific code implementing
45*7c478bd9Sstevel@tonic-gate * the fixed-priority priocntl sub-command.
46*7c478bd9Sstevel@tonic-gate */
47*7c478bd9Sstevel@tonic-gate
48*7c478bd9Sstevel@tonic-gate #define ADDKEYVAL(p, k, v) { (p[0]) = (k); (p[1]) = (v); p += 2; }
49*7c478bd9Sstevel@tonic-gate
50*7c478bd9Sstevel@tonic-gate #define FX_KEYCNT 4 /* maximal number of (key, value) pairs */
51*7c478bd9Sstevel@tonic-gate
52*7c478bd9Sstevel@tonic-gate /*
53*7c478bd9Sstevel@tonic-gate * control flags
54*7c478bd9Sstevel@tonic-gate */
55*7c478bd9Sstevel@tonic-gate #define FX_DOUPRILIM 0x01 /* user priority limit */
56*7c478bd9Sstevel@tonic-gate #define FX_DOUPRI 0x02 /* user priority */
57*7c478bd9Sstevel@tonic-gate #define FX_DOTQ 0x04 /* time quantum */
58*7c478bd9Sstevel@tonic-gate
59*7c478bd9Sstevel@tonic-gate
60*7c478bd9Sstevel@tonic-gate
61*7c478bd9Sstevel@tonic-gate static void print_fxinfo();
62*7c478bd9Sstevel@tonic-gate static int print_fxprocs();
63*7c478bd9Sstevel@tonic-gate static int set_fxprocs(idtype_t, int, char **, uint_t, pri_t, pri_t, long,
64*7c478bd9Sstevel@tonic-gate long);
65*7c478bd9Sstevel@tonic-gate static void exec_fxcmd(char **, uint_t, pri_t, pri_t, long, long);
66*7c478bd9Sstevel@tonic-gate static int fx_priocntl(idtype_t, id_t, int, char *, uintptr_t *);
67*7c478bd9Sstevel@tonic-gate
68*7c478bd9Sstevel@tonic-gate static char usage[] =
69*7c478bd9Sstevel@tonic-gate "usage: priocntl -l\n\
70*7c478bd9Sstevel@tonic-gate priocntl -d [-d idtype] [idlist]\n\
71*7c478bd9Sstevel@tonic-gate priocntl -s [-c FX] [-m fxuprilim] [-p fxupri] [-t tqntm [-r res]] \
72*7c478bd9Sstevel@tonic-gate [-i idtype] [idlist]\n\
73*7c478bd9Sstevel@tonic-gate priocntl -e [-c FX] [-m fxuprilim] [-p fxupri] [-t tqntm [-r res]] \
74*7c478bd9Sstevel@tonic-gate command [argument(s)]\n";
75*7c478bd9Sstevel@tonic-gate
76*7c478bd9Sstevel@tonic-gate static char cmdpath[MAXPATHLEN];
77*7c478bd9Sstevel@tonic-gate static char basenm[BASENMSZ];
78*7c478bd9Sstevel@tonic-gate
79*7c478bd9Sstevel@tonic-gate int
main(int argc,char * argv[])80*7c478bd9Sstevel@tonic-gate main(int argc, char *argv[])
81*7c478bd9Sstevel@tonic-gate {
82*7c478bd9Sstevel@tonic-gate extern char *optarg;
83*7c478bd9Sstevel@tonic-gate extern int optind;
84*7c478bd9Sstevel@tonic-gate
85*7c478bd9Sstevel@tonic-gate int c;
86*7c478bd9Sstevel@tonic-gate int lflag, dflag, sflag, mflag, pflag, eflag, iflag, tflag;
87*7c478bd9Sstevel@tonic-gate int rflag;
88*7c478bd9Sstevel@tonic-gate pri_t fxuprilim;
89*7c478bd9Sstevel@tonic-gate pri_t fxupri;
90*7c478bd9Sstevel@tonic-gate char *idtypnm;
91*7c478bd9Sstevel@tonic-gate idtype_t idtype;
92*7c478bd9Sstevel@tonic-gate int idargc;
93*7c478bd9Sstevel@tonic-gate long res;
94*7c478bd9Sstevel@tonic-gate long tqntm;
95*7c478bd9Sstevel@tonic-gate uint_t cflags;
96*7c478bd9Sstevel@tonic-gate
97*7c478bd9Sstevel@tonic-gate (void) strlcpy(cmdpath, argv[0], MAXPATHLEN);
98*7c478bd9Sstevel@tonic-gate (void) strlcpy(basenm, basename(argv[0]), BASENMSZ);
99*7c478bd9Sstevel@tonic-gate lflag = dflag = sflag = mflag = pflag = eflag = iflag = tflag = 0;
100*7c478bd9Sstevel@tonic-gate rflag = 0;
101*7c478bd9Sstevel@tonic-gate while ((c = getopt(argc, argv, "ldsm:p:ec:i:t:r:")) != -1) {
102*7c478bd9Sstevel@tonic-gate switch (c) {
103*7c478bd9Sstevel@tonic-gate
104*7c478bd9Sstevel@tonic-gate case 'l':
105*7c478bd9Sstevel@tonic-gate lflag++;
106*7c478bd9Sstevel@tonic-gate break;
107*7c478bd9Sstevel@tonic-gate
108*7c478bd9Sstevel@tonic-gate case 'd':
109*7c478bd9Sstevel@tonic-gate dflag++;
110*7c478bd9Sstevel@tonic-gate break;
111*7c478bd9Sstevel@tonic-gate
112*7c478bd9Sstevel@tonic-gate case 's':
113*7c478bd9Sstevel@tonic-gate sflag++;
114*7c478bd9Sstevel@tonic-gate break;
115*7c478bd9Sstevel@tonic-gate
116*7c478bd9Sstevel@tonic-gate case 'm':
117*7c478bd9Sstevel@tonic-gate mflag++;
118*7c478bd9Sstevel@tonic-gate fxuprilim = (pri_t)str2num(optarg, SHRT_MIN, SHRT_MAX);
119*7c478bd9Sstevel@tonic-gate if (errno)
120*7c478bd9Sstevel@tonic-gate fatalerr("%s: Specified user priority limit %s"
121*7c478bd9Sstevel@tonic-gate " out of configured range\n",
122*7c478bd9Sstevel@tonic-gate basenm, optarg);
123*7c478bd9Sstevel@tonic-gate break;
124*7c478bd9Sstevel@tonic-gate
125*7c478bd9Sstevel@tonic-gate case 'p':
126*7c478bd9Sstevel@tonic-gate pflag++;
127*7c478bd9Sstevel@tonic-gate fxupri = (pri_t)str2num(optarg, SHRT_MIN, SHRT_MAX);
128*7c478bd9Sstevel@tonic-gate if (errno)
129*7c478bd9Sstevel@tonic-gate fatalerr("%s: Specified user priority %s out of"
130*7c478bd9Sstevel@tonic-gate " configured range\n", basenm, optarg);
131*7c478bd9Sstevel@tonic-gate break;
132*7c478bd9Sstevel@tonic-gate
133*7c478bd9Sstevel@tonic-gate case 'e':
134*7c478bd9Sstevel@tonic-gate eflag++;
135*7c478bd9Sstevel@tonic-gate break;
136*7c478bd9Sstevel@tonic-gate
137*7c478bd9Sstevel@tonic-gate case 'c':
138*7c478bd9Sstevel@tonic-gate if (strcmp(optarg, "FX") != 0)
139*7c478bd9Sstevel@tonic-gate fatalerr("error: %s executed for %s class, %s"
140*7c478bd9Sstevel@tonic-gate " is actually sub-command for FX class\n",
141*7c478bd9Sstevel@tonic-gate cmdpath, optarg, cmdpath);
142*7c478bd9Sstevel@tonic-gate break;
143*7c478bd9Sstevel@tonic-gate
144*7c478bd9Sstevel@tonic-gate case 'i':
145*7c478bd9Sstevel@tonic-gate iflag++;
146*7c478bd9Sstevel@tonic-gate idtypnm = optarg;
147*7c478bd9Sstevel@tonic-gate break;
148*7c478bd9Sstevel@tonic-gate case 't':
149*7c478bd9Sstevel@tonic-gate tflag++;
150*7c478bd9Sstevel@tonic-gate tqntm = str2num(optarg, 1, INT_MAX);
151*7c478bd9Sstevel@tonic-gate if (errno)
152*7c478bd9Sstevel@tonic-gate fatalerr("%s: Invalid time quantum specified;"
153*7c478bd9Sstevel@tonic-gate " time quantum must be positive\n", basenm);
154*7c478bd9Sstevel@tonic-gate break;
155*7c478bd9Sstevel@tonic-gate
156*7c478bd9Sstevel@tonic-gate case 'r':
157*7c478bd9Sstevel@tonic-gate rflag++;
158*7c478bd9Sstevel@tonic-gate res = str2num(optarg, 1, 1000000000);
159*7c478bd9Sstevel@tonic-gate if (errno)
160*7c478bd9Sstevel@tonic-gate fatalerr("%s: Invalid resolution specified;"
161*7c478bd9Sstevel@tonic-gate " resolution must be between"
162*7c478bd9Sstevel@tonic-gate " 1 and 1,000,000,000\n", basenm);
163*7c478bd9Sstevel@tonic-gate
164*7c478bd9Sstevel@tonic-gate break;
165*7c478bd9Sstevel@tonic-gate
166*7c478bd9Sstevel@tonic-gate case '?':
167*7c478bd9Sstevel@tonic-gate fatalerr(usage);
168*7c478bd9Sstevel@tonic-gate
169*7c478bd9Sstevel@tonic-gate default:
170*7c478bd9Sstevel@tonic-gate break;
171*7c478bd9Sstevel@tonic-gate }
172*7c478bd9Sstevel@tonic-gate }
173*7c478bd9Sstevel@tonic-gate
174*7c478bd9Sstevel@tonic-gate if (lflag) {
175*7c478bd9Sstevel@tonic-gate if (dflag || sflag || mflag || pflag || eflag || iflag ||
176*7c478bd9Sstevel@tonic-gate tflag || rflag)
177*7c478bd9Sstevel@tonic-gate fatalerr(usage);
178*7c478bd9Sstevel@tonic-gate
179*7c478bd9Sstevel@tonic-gate print_fxinfo();
180*7c478bd9Sstevel@tonic-gate
181*7c478bd9Sstevel@tonic-gate } else if (dflag) {
182*7c478bd9Sstevel@tonic-gate if (sflag || mflag || pflag || eflag)
183*7c478bd9Sstevel@tonic-gate fatalerr(usage);
184*7c478bd9Sstevel@tonic-gate
185*7c478bd9Sstevel@tonic-gate return (print_fxprocs());
186*7c478bd9Sstevel@tonic-gate
187*7c478bd9Sstevel@tonic-gate } else if (sflag) {
188*7c478bd9Sstevel@tonic-gate if (eflag)
189*7c478bd9Sstevel@tonic-gate fatalerr(usage);
190*7c478bd9Sstevel@tonic-gate
191*7c478bd9Sstevel@tonic-gate if (iflag) {
192*7c478bd9Sstevel@tonic-gate if (str2idtyp(idtypnm, &idtype) == -1)
193*7c478bd9Sstevel@tonic-gate fatalerr("%s: Bad idtype %s\n", basenm,
194*7c478bd9Sstevel@tonic-gate idtypnm);
195*7c478bd9Sstevel@tonic-gate } else
196*7c478bd9Sstevel@tonic-gate idtype = P_PID;
197*7c478bd9Sstevel@tonic-gate
198*7c478bd9Sstevel@tonic-gate cflags = (pflag ? FX_DOUPRI : 0);
199*7c478bd9Sstevel@tonic-gate
200*7c478bd9Sstevel@tonic-gate if (mflag)
201*7c478bd9Sstevel@tonic-gate cflags |= FX_DOUPRILIM;
202*7c478bd9Sstevel@tonic-gate
203*7c478bd9Sstevel@tonic-gate if (tflag)
204*7c478bd9Sstevel@tonic-gate cflags |= FX_DOTQ;
205*7c478bd9Sstevel@tonic-gate
206*7c478bd9Sstevel@tonic-gate if (rflag == 0)
207*7c478bd9Sstevel@tonic-gate res = 1000;
208*7c478bd9Sstevel@tonic-gate
209*7c478bd9Sstevel@tonic-gate if (optind < argc)
210*7c478bd9Sstevel@tonic-gate idargc = argc - optind;
211*7c478bd9Sstevel@tonic-gate else
212*7c478bd9Sstevel@tonic-gate idargc = 0;
213*7c478bd9Sstevel@tonic-gate
214*7c478bd9Sstevel@tonic-gate return (set_fxprocs(idtype, idargc, &argv[optind], cflags,
215*7c478bd9Sstevel@tonic-gate fxuprilim, fxupri, tqntm, res));
216*7c478bd9Sstevel@tonic-gate
217*7c478bd9Sstevel@tonic-gate } else if (eflag) {
218*7c478bd9Sstevel@tonic-gate if (iflag)
219*7c478bd9Sstevel@tonic-gate fatalerr(usage);
220*7c478bd9Sstevel@tonic-gate
221*7c478bd9Sstevel@tonic-gate cflags = (pflag ? FX_DOUPRI : 0);
222*7c478bd9Sstevel@tonic-gate
223*7c478bd9Sstevel@tonic-gate if (mflag)
224*7c478bd9Sstevel@tonic-gate cflags |= FX_DOUPRILIM;
225*7c478bd9Sstevel@tonic-gate
226*7c478bd9Sstevel@tonic-gate if (tflag)
227*7c478bd9Sstevel@tonic-gate cflags |= FX_DOTQ;
228*7c478bd9Sstevel@tonic-gate
229*7c478bd9Sstevel@tonic-gate if (rflag == 0)
230*7c478bd9Sstevel@tonic-gate res = 1000;
231*7c478bd9Sstevel@tonic-gate
232*7c478bd9Sstevel@tonic-gate exec_fxcmd(&argv[optind], cflags, fxuprilim, fxupri, tqntm,
233*7c478bd9Sstevel@tonic-gate res);
234*7c478bd9Sstevel@tonic-gate
235*7c478bd9Sstevel@tonic-gate } else {
236*7c478bd9Sstevel@tonic-gate fatalerr(usage);
237*7c478bd9Sstevel@tonic-gate }
238*7c478bd9Sstevel@tonic-gate return (0);
239*7c478bd9Sstevel@tonic-gate }
240*7c478bd9Sstevel@tonic-gate
241*7c478bd9Sstevel@tonic-gate
242*7c478bd9Sstevel@tonic-gate /*
243*7c478bd9Sstevel@tonic-gate * Print our class name and the configured user priority range.
244*7c478bd9Sstevel@tonic-gate */
245*7c478bd9Sstevel@tonic-gate static void
print_fxinfo()246*7c478bd9Sstevel@tonic-gate print_fxinfo()
247*7c478bd9Sstevel@tonic-gate {
248*7c478bd9Sstevel@tonic-gate pcinfo_t pcinfo;
249*7c478bd9Sstevel@tonic-gate
250*7c478bd9Sstevel@tonic-gate (void) strcpy(pcinfo.pc_clname, "FX");
251*7c478bd9Sstevel@tonic-gate
252*7c478bd9Sstevel@tonic-gate (void) printf("FX (Fixed priority)\n");
253*7c478bd9Sstevel@tonic-gate
254*7c478bd9Sstevel@tonic-gate if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
255*7c478bd9Sstevel@tonic-gate fatalerr("\tCan't get configured FX user priority range\n");
256*7c478bd9Sstevel@tonic-gate
257*7c478bd9Sstevel@tonic-gate (void) printf("\tConfigured FX User Priority Range: 0 through %d\n",
258*7c478bd9Sstevel@tonic-gate ((fxinfo_t *)pcinfo.pc_clinfo)->fx_maxupri);
259*7c478bd9Sstevel@tonic-gate }
260*7c478bd9Sstevel@tonic-gate
261*7c478bd9Sstevel@tonic-gate
262*7c478bd9Sstevel@tonic-gate /*
263*7c478bd9Sstevel@tonic-gate * Read a list of pids from stdin and print the user priority and user
264*7c478bd9Sstevel@tonic-gate * priority limit for each of the corresponding processes.
265*7c478bd9Sstevel@tonic-gate */
266*7c478bd9Sstevel@tonic-gate static int
print_fxprocs()267*7c478bd9Sstevel@tonic-gate print_fxprocs()
268*7c478bd9Sstevel@tonic-gate {
269*7c478bd9Sstevel@tonic-gate pid_t *pidlist;
270*7c478bd9Sstevel@tonic-gate size_t numread;
271*7c478bd9Sstevel@tonic-gate int i;
272*7c478bd9Sstevel@tonic-gate char clname[PC_CLNMSZ];
273*7c478bd9Sstevel@tonic-gate uint_t fx_tqsecs;
274*7c478bd9Sstevel@tonic-gate int fx_tqnsecs;
275*7c478bd9Sstevel@tonic-gate pri_t fx_uprilim;
276*7c478bd9Sstevel@tonic-gate pri_t fx_upri;
277*7c478bd9Sstevel@tonic-gate int error = 0;
278*7c478bd9Sstevel@tonic-gate
279*7c478bd9Sstevel@tonic-gate
280*7c478bd9Sstevel@tonic-gate /*
281*7c478bd9Sstevel@tonic-gate * Read a list of pids from stdin.
282*7c478bd9Sstevel@tonic-gate */
283*7c478bd9Sstevel@tonic-gate if ((pidlist = read_pidlist(&numread, stdin)) == NULL)
284*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't read pidlist.\n", basenm);
285*7c478bd9Sstevel@tonic-gate
286*7c478bd9Sstevel@tonic-gate (void) printf("FIXED PRIORITY PROCESSES:\n PID FXUPRILIM "
287*7c478bd9Sstevel@tonic-gate "FXUPRI FXTQNTM\n");
288*7c478bd9Sstevel@tonic-gate
289*7c478bd9Sstevel@tonic-gate if (numread == 0)
290*7c478bd9Sstevel@tonic-gate fatalerr("%s: No pids on input\n", basenm);
291*7c478bd9Sstevel@tonic-gate
292*7c478bd9Sstevel@tonic-gate
293*7c478bd9Sstevel@tonic-gate for (i = 0; i < numread; i++) {
294*7c478bd9Sstevel@tonic-gate (void) printf("%7ld", pidlist[i]);
295*7c478bd9Sstevel@tonic-gate if (priocntl(P_PID, pidlist[i], PC_GETXPARMS, "FX",
296*7c478bd9Sstevel@tonic-gate FX_KY_UPRI, &fx_upri, FX_KY_UPRILIM, &fx_uprilim,
297*7c478bd9Sstevel@tonic-gate FX_KY_TQSECS, &fx_tqsecs, FX_KY_TQNSECS,
298*7c478bd9Sstevel@tonic-gate &fx_tqnsecs, 0) != -1) {
299*7c478bd9Sstevel@tonic-gate (void) printf(" %5d %5d", fx_uprilim, fx_upri);
300*7c478bd9Sstevel@tonic-gate
301*7c478bd9Sstevel@tonic-gate if (fx_tqnsecs == FX_TQINF)
302*7c478bd9Sstevel@tonic-gate (void) printf(" FX_TQINF\n");
303*7c478bd9Sstevel@tonic-gate else
304*7c478bd9Sstevel@tonic-gate (void) printf(" %11lld\n",
305*7c478bd9Sstevel@tonic-gate (longlong_t)fx_tqsecs * 1000 +
306*7c478bd9Sstevel@tonic-gate fx_tqnsecs / 1000000);
307*7c478bd9Sstevel@tonic-gate
308*7c478bd9Sstevel@tonic-gate } else {
309*7c478bd9Sstevel@tonic-gate error = 1;
310*7c478bd9Sstevel@tonic-gate
311*7c478bd9Sstevel@tonic-gate if (priocntl(P_PID, pidlist[i], PC_GETXPARMS, NULL,
312*7c478bd9Sstevel@tonic-gate PC_KY_CLNAME, clname, 0) != -1 &&
313*7c478bd9Sstevel@tonic-gate strcmp(clname, "FX")) {
314*7c478bd9Sstevel@tonic-gate /*
315*7c478bd9Sstevel@tonic-gate * Process from some class other than fixed priority.
316*7c478bd9Sstevel@tonic-gate * It has probably changed class while priocntl
317*7c478bd9Sstevel@tonic-gate * command was executing (otherwise we wouldn't
318*7c478bd9Sstevel@tonic-gate * have been passed its pid). Print the little
319*7c478bd9Sstevel@tonic-gate * we know about it.
320*7c478bd9Sstevel@tonic-gate */
321*7c478bd9Sstevel@tonic-gate (void) printf("\tChanged to class %s while priocntl"
322*7c478bd9Sstevel@tonic-gate " command executing\n", clname);
323*7c478bd9Sstevel@tonic-gate } else {
324*7c478bd9Sstevel@tonic-gate (void) printf("\tCan't get FX user priority\n");
325*7c478bd9Sstevel@tonic-gate }
326*7c478bd9Sstevel@tonic-gate }
327*7c478bd9Sstevel@tonic-gate }
328*7c478bd9Sstevel@tonic-gate
329*7c478bd9Sstevel@tonic-gate free_pidlist(pidlist);
330*7c478bd9Sstevel@tonic-gate return (error);
331*7c478bd9Sstevel@tonic-gate }
332*7c478bd9Sstevel@tonic-gate
333*7c478bd9Sstevel@tonic-gate /*
334*7c478bd9Sstevel@tonic-gate * Call priocntl() with command codes PC_SETXPARMS or PC_GETXPARMS.
335*7c478bd9Sstevel@tonic-gate * The first parameter behind the command code is always the class name.
336*7c478bd9Sstevel@tonic-gate * Each parameter is headed by a key, which determines the meaning of the
337*7c478bd9Sstevel@tonic-gate * following value. There are maximal FX_KEYCNT = 4 (key, value) pairs.
338*7c478bd9Sstevel@tonic-gate */
339*7c478bd9Sstevel@tonic-gate static int
fx_priocntl(idtype_t idtype,id_t id,int cmd,char * clname,uintptr_t * argsp)340*7c478bd9Sstevel@tonic-gate fx_priocntl(idtype_t idtype, id_t id, int cmd, char *clname, uintptr_t *argsp)
341*7c478bd9Sstevel@tonic-gate {
342*7c478bd9Sstevel@tonic-gate return (priocntl(idtype, id, cmd, clname, argsp[0], argsp[1],
343*7c478bd9Sstevel@tonic-gate argsp[2], argsp[3], argsp[4], argsp[5], argsp[6], argsp[7], 0));
344*7c478bd9Sstevel@tonic-gate }
345*7c478bd9Sstevel@tonic-gate
346*7c478bd9Sstevel@tonic-gate /*
347*7c478bd9Sstevel@tonic-gate * Set all processes in the set specified by idtype/idargv to fixed-priority
348*7c478bd9Sstevel@tonic-gate * (if they aren't already fixed-priority) and set their user priority limit
349*7c478bd9Sstevel@tonic-gate * and user priority to those specified by fxuprilim and fxupri.
350*7c478bd9Sstevel@tonic-gate */
351*7c478bd9Sstevel@tonic-gate static int
set_fxprocs(idtype_t idtype,int idargc,char ** idargv,uint_t cflags,pri_t fxuprilim,pri_t fxupri,long tqntm,long res)352*7c478bd9Sstevel@tonic-gate set_fxprocs(idtype_t idtype, int idargc, char **idargv, uint_t cflags,
353*7c478bd9Sstevel@tonic-gate pri_t fxuprilim, pri_t fxupri, long tqntm, long res)
354*7c478bd9Sstevel@tonic-gate {
355*7c478bd9Sstevel@tonic-gate pcinfo_t pcinfo;
356*7c478bd9Sstevel@tonic-gate uintptr_t args[2*FX_KEYCNT+1];
357*7c478bd9Sstevel@tonic-gate uintptr_t *argsp = &args[0];
358*7c478bd9Sstevel@tonic-gate pri_t maxupri;
359*7c478bd9Sstevel@tonic-gate char idtypnm[PC_IDTYPNMSZ];
360*7c478bd9Sstevel@tonic-gate int i;
361*7c478bd9Sstevel@tonic-gate id_t id;
362*7c478bd9Sstevel@tonic-gate hrtimer_t hrtime;
363*7c478bd9Sstevel@tonic-gate int error = 0;
364*7c478bd9Sstevel@tonic-gate
365*7c478bd9Sstevel@tonic-gate /*
366*7c478bd9Sstevel@tonic-gate * Get the fixed priority class ID and max configured user priority.
367*7c478bd9Sstevel@tonic-gate */
368*7c478bd9Sstevel@tonic-gate (void) strcpy(pcinfo.pc_clname, "FX");
369*7c478bd9Sstevel@tonic-gate if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
370*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't get FX class ID, priocntl system call"
371*7c478bd9Sstevel@tonic-gate " failed with errno %d\n", basenm, errno);
372*7c478bd9Sstevel@tonic-gate maxupri = ((fxinfo_t *)pcinfo.pc_clinfo)->fx_maxupri;
373*7c478bd9Sstevel@tonic-gate
374*7c478bd9Sstevel@tonic-gate /*
375*7c478bd9Sstevel@tonic-gate * Validate the fxuprilim and fxupri arguments.
376*7c478bd9Sstevel@tonic-gate */
377*7c478bd9Sstevel@tonic-gate if ((cflags & FX_DOUPRILIM) != 0) {
378*7c478bd9Sstevel@tonic-gate if (fxuprilim > maxupri || fxuprilim < 0) {
379*7c478bd9Sstevel@tonic-gate fatalerr("%s: Specified user priority limit %d out of"
380*7c478bd9Sstevel@tonic-gate " configured range\n", basenm, fxuprilim);
381*7c478bd9Sstevel@tonic-gate }
382*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_UPRILIM, fxuprilim);
383*7c478bd9Sstevel@tonic-gate }
384*7c478bd9Sstevel@tonic-gate
385*7c478bd9Sstevel@tonic-gate if ((cflags & FX_DOUPRI) != 0) {
386*7c478bd9Sstevel@tonic-gate if (fxupri > maxupri || fxupri < 0)
387*7c478bd9Sstevel@tonic-gate fatalerr("%s: Specified user priority %d out of "
388*7c478bd9Sstevel@tonic-gate "configured range\n", basenm, fxupri);
389*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_UPRI, fxupri);
390*7c478bd9Sstevel@tonic-gate
391*7c478bd9Sstevel@tonic-gate }
392*7c478bd9Sstevel@tonic-gate
393*7c478bd9Sstevel@tonic-gate if (cflags & FX_DOTQ) {
394*7c478bd9Sstevel@tonic-gate hrtime.hrt_secs = 0;
395*7c478bd9Sstevel@tonic-gate hrtime.hrt_rem = tqntm;
396*7c478bd9Sstevel@tonic-gate hrtime.hrt_res = res;
397*7c478bd9Sstevel@tonic-gate if (_hrtnewres(&hrtime, NANOSEC, HRT_RNDUP) == -1)
398*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't convert resolution.\n", basenm);
399*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_TQSECS, hrtime.hrt_secs);
400*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_TQNSECS, hrtime.hrt_rem);
401*7c478bd9Sstevel@tonic-gate }
402*7c478bd9Sstevel@tonic-gate
403*7c478bd9Sstevel@tonic-gate *argsp = 0;
404*7c478bd9Sstevel@tonic-gate
405*7c478bd9Sstevel@tonic-gate if (idtype == P_ALL) {
406*7c478bd9Sstevel@tonic-gate if (fx_priocntl(P_ALL, 0, PC_SETXPARMS, "FX", args) == -1) {
407*7c478bd9Sstevel@tonic-gate if (errno == EPERM) {
408*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
409*7c478bd9Sstevel@tonic-gate "Permissions error encountered "
410*7c478bd9Sstevel@tonic-gate "on one or more processes.\n");
411*7c478bd9Sstevel@tonic-gate error = 1;
412*7c478bd9Sstevel@tonic-gate } else {
413*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't reset fixed priority"
414*7c478bd9Sstevel@tonic-gate " parameters\npriocntl system call failed "
415*7c478bd9Sstevel@tonic-gate " with errno %d\n", basenm, errno);
416*7c478bd9Sstevel@tonic-gate }
417*7c478bd9Sstevel@tonic-gate } else if ((cflags & (FX_DOUPRILIM|FX_DOUPRI)) == FX_DOUPRI) {
418*7c478bd9Sstevel@tonic-gate (void) verifyupri(idtype, 0, "FX", FX_KY_UPRILIM,
419*7c478bd9Sstevel@tonic-gate fxupri, basenm);
420*7c478bd9Sstevel@tonic-gate }
421*7c478bd9Sstevel@tonic-gate } else if (idargc == 0) {
422*7c478bd9Sstevel@tonic-gate if (fx_priocntl(idtype, P_MYID, PC_SETXPARMS, "FX",
423*7c478bd9Sstevel@tonic-gate args) == -1) {
424*7c478bd9Sstevel@tonic-gate if (errno == EPERM) {
425*7c478bd9Sstevel@tonic-gate (void) idtyp2str(idtype, idtypnm);
426*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr, "Permissions error"
427*7c478bd9Sstevel@tonic-gate " encountered on current %s.\n", idtypnm);
428*7c478bd9Sstevel@tonic-gate error = 1;
429*7c478bd9Sstevel@tonic-gate } else {
430*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't reset fixed priority"
431*7c478bd9Sstevel@tonic-gate " parameters\npriocntl system call failed"
432*7c478bd9Sstevel@tonic-gate " with errno %d\n", basenm, errno);
433*7c478bd9Sstevel@tonic-gate }
434*7c478bd9Sstevel@tonic-gate } else if ((cflags & (FX_DOUPRILIM|FX_DOUPRI)) == FX_DOUPRI &&
435*7c478bd9Sstevel@tonic-gate getmyid(idtype, &id) != -1) {
436*7c478bd9Sstevel@tonic-gate (void) verifyupri(idtype, id, "FX", FX_KY_UPRILIM,
437*7c478bd9Sstevel@tonic-gate fxupri, basenm);
438*7c478bd9Sstevel@tonic-gate }
439*7c478bd9Sstevel@tonic-gate } else {
440*7c478bd9Sstevel@tonic-gate (void) idtyp2str(idtype, idtypnm);
441*7c478bd9Sstevel@tonic-gate for (i = 0; i < idargc; i++) {
442*7c478bd9Sstevel@tonic-gate if (idtype == P_CID) {
443*7c478bd9Sstevel@tonic-gate (void) strcpy(pcinfo.pc_clname, idargv[i]);
444*7c478bd9Sstevel@tonic-gate if (priocntl(0, 0, PC_GETCID,
445*7c478bd9Sstevel@tonic-gate (caddr_t)&pcinfo) == -1)
446*7c478bd9Sstevel@tonic-gate fatalerr("%s: Invalid or unconfigured"
447*7c478bd9Sstevel@tonic-gate " class %s, priocntl system call"
448*7c478bd9Sstevel@tonic-gate " failed with errno %d\n",
449*7c478bd9Sstevel@tonic-gate basenm, pcinfo.pc_clname, errno);
450*7c478bd9Sstevel@tonic-gate id = pcinfo.pc_cid;
451*7c478bd9Sstevel@tonic-gate } else {
452*7c478bd9Sstevel@tonic-gate id = (id_t)str2num(idargv[i], INT_MIN, INT_MAX);
453*7c478bd9Sstevel@tonic-gate }
454*7c478bd9Sstevel@tonic-gate
455*7c478bd9Sstevel@tonic-gate if (fx_priocntl(idtype, id, PC_SETXPARMS, "FX", args)
456*7c478bd9Sstevel@tonic-gate == -1) {
457*7c478bd9Sstevel@tonic-gate if (errno == EPERM) {
458*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
459*7c478bd9Sstevel@tonic-gate "Permissions error encountered on"
460*7c478bd9Sstevel@tonic-gate " %s %s.\n", idtypnm, idargv[i]);
461*7c478bd9Sstevel@tonic-gate error = 1;
462*7c478bd9Sstevel@tonic-gate } else {
463*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't reset fixed "
464*7c478bd9Sstevel@tonic-gate "priority"
465*7c478bd9Sstevel@tonic-gate " parameters\npriocntl system call"
466*7c478bd9Sstevel@tonic-gate " failed with errno %d\n",
467*7c478bd9Sstevel@tonic-gate basenm, errno);
468*7c478bd9Sstevel@tonic-gate }
469*7c478bd9Sstevel@tonic-gate } else if ((cflags & (FX_DOUPRILIM|FX_DOUPRI)) ==
470*7c478bd9Sstevel@tonic-gate FX_DOUPRI) {
471*7c478bd9Sstevel@tonic-gate (void) verifyupri(idtype, id, "FX",
472*7c478bd9Sstevel@tonic-gate FX_KY_UPRILIM, fxupri, basenm);
473*7c478bd9Sstevel@tonic-gate }
474*7c478bd9Sstevel@tonic-gate }
475*7c478bd9Sstevel@tonic-gate }
476*7c478bd9Sstevel@tonic-gate return (error);
477*7c478bd9Sstevel@tonic-gate
478*7c478bd9Sstevel@tonic-gate }
479*7c478bd9Sstevel@tonic-gate
480*7c478bd9Sstevel@tonic-gate
481*7c478bd9Sstevel@tonic-gate /*
482*7c478bd9Sstevel@tonic-gate * Execute the command pointed to by cmdargv as a fixed-priority process
483*7c478bd9Sstevel@tonic-gate * with the user priority limit given by fxuprilim and user priority fxupri.
484*7c478bd9Sstevel@tonic-gate */
485*7c478bd9Sstevel@tonic-gate static void
exec_fxcmd(char ** cmdargv,uint_t cflags,pri_t fxuprilim,pri_t fxupri,long tqntm,long res)486*7c478bd9Sstevel@tonic-gate exec_fxcmd(char **cmdargv, uint_t cflags, pri_t fxuprilim, pri_t fxupri,
487*7c478bd9Sstevel@tonic-gate long tqntm, long res)
488*7c478bd9Sstevel@tonic-gate {
489*7c478bd9Sstevel@tonic-gate pcinfo_t pcinfo;
490*7c478bd9Sstevel@tonic-gate uintptr_t args[2*FX_KEYCNT+1];
491*7c478bd9Sstevel@tonic-gate uintptr_t *argsp = &args[0];
492*7c478bd9Sstevel@tonic-gate
493*7c478bd9Sstevel@tonic-gate pri_t maxupri;
494*7c478bd9Sstevel@tonic-gate pri_t uprilim;
495*7c478bd9Sstevel@tonic-gate hrtimer_t hrtime;
496*7c478bd9Sstevel@tonic-gate
497*7c478bd9Sstevel@tonic-gate /*
498*7c478bd9Sstevel@tonic-gate * Get the fixed priority class ID and max configured user priority.
499*7c478bd9Sstevel@tonic-gate */
500*7c478bd9Sstevel@tonic-gate (void) strcpy(pcinfo.pc_clname, "FX");
501*7c478bd9Sstevel@tonic-gate if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
502*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't get FX class ID, priocntl system call"
503*7c478bd9Sstevel@tonic-gate " failed with errno %d\n", basenm, errno);
504*7c478bd9Sstevel@tonic-gate maxupri = ((fxinfo_t *)pcinfo.pc_clinfo)->fx_maxupri;
505*7c478bd9Sstevel@tonic-gate
506*7c478bd9Sstevel@tonic-gate if ((cflags & FX_DOUPRILIM) != 0) {
507*7c478bd9Sstevel@tonic-gate if (fxuprilim > maxupri || fxuprilim < 0)
508*7c478bd9Sstevel@tonic-gate fatalerr("%s: Specified user priority limit %d out of"
509*7c478bd9Sstevel@tonic-gate " configured range\n", basenm, fxuprilim);
510*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_UPRILIM, fxuprilim);
511*7c478bd9Sstevel@tonic-gate }
512*7c478bd9Sstevel@tonic-gate
513*7c478bd9Sstevel@tonic-gate if ((cflags & FX_DOUPRI) != 0) {
514*7c478bd9Sstevel@tonic-gate if (fxupri > maxupri || fxupri < 0)
515*7c478bd9Sstevel@tonic-gate fatalerr("%s: Specified user priority %d out of"
516*7c478bd9Sstevel@tonic-gate " configured range\n", basenm, fxupri);
517*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_UPRI, fxupri);
518*7c478bd9Sstevel@tonic-gate }
519*7c478bd9Sstevel@tonic-gate
520*7c478bd9Sstevel@tonic-gate if ((cflags & FX_DOTQ) != 0) {
521*7c478bd9Sstevel@tonic-gate hrtime.hrt_secs = 0;
522*7c478bd9Sstevel@tonic-gate hrtime.hrt_rem = tqntm;
523*7c478bd9Sstevel@tonic-gate hrtime.hrt_res = res;
524*7c478bd9Sstevel@tonic-gate if (_hrtnewres(&hrtime, NANOSEC, HRT_RNDUP) == -1)
525*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't convert resolution.\n", basenm);
526*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_TQSECS, hrtime.hrt_secs);
527*7c478bd9Sstevel@tonic-gate ADDKEYVAL(argsp, FX_KY_TQNSECS, hrtime.hrt_rem);
528*7c478bd9Sstevel@tonic-gate
529*7c478bd9Sstevel@tonic-gate }
530*7c478bd9Sstevel@tonic-gate *argsp = 0;
531*7c478bd9Sstevel@tonic-gate if (fx_priocntl(P_PID, P_MYID, PC_SETXPARMS, "FX", args) == -1)
532*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't reset fixed priority parameters\n"
533*7c478bd9Sstevel@tonic-gate " priocntl system call failed with errno %d\n",
534*7c478bd9Sstevel@tonic-gate basenm, errno);
535*7c478bd9Sstevel@tonic-gate
536*7c478bd9Sstevel@tonic-gate if ((cflags & (FX_DOUPRILIM|FX_DOUPRI)) == FX_DOUPRI) {
537*7c478bd9Sstevel@tonic-gate if (priocntl(P_PID, P_MYID, PC_GETXPARMS, "FX",
538*7c478bd9Sstevel@tonic-gate FX_KY_UPRILIM, &uprilim, 0) != -1 && fxupri > uprilim)
539*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
540*7c478bd9Sstevel@tonic-gate "%s: Specified user priority %d exceeds"
541*7c478bd9Sstevel@tonic-gate " limit %d; set to %d (pid %d)\n",
542*7c478bd9Sstevel@tonic-gate basenm, fxupri, uprilim, uprilim, (int)getpid());
543*7c478bd9Sstevel@tonic-gate }
544*7c478bd9Sstevel@tonic-gate
545*7c478bd9Sstevel@tonic-gate (void) execvp(cmdargv[0], cmdargv);
546*7c478bd9Sstevel@tonic-gate fatalerr("%s: Can't execute %s, exec failed with errno %d\n",
547*7c478bd9Sstevel@tonic-gate basenm, cmdargv[0], errno);
548*7c478bd9Sstevel@tonic-gate }
549