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, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  */
22 /*
23  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 /* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
28 /* All Rights Reserved */
29 
30 #pragma ident	"%Z%%M%	%I%	%E% SMI"
31 
32 #include	<stdio.h>
33 #include	<stdlib.h>
34 #include	<string.h>
35 #include	<unistd.h>
36 #include	<sys/types.h>
37 #include	<sys/priocntl.h>
38 #include	<sys/fxpriocntl.h>
39 #include	<sys/param.h>
40 #include	<errno.h>
41 #include	<sys/fx.h>
42 
43 #include	"dispadmin.h"
44 
45 /*
46  * This file contains the class specific code implementing
47  * the fixed-priority dispadmin sub-command.
48  */
49 
50 #define	BASENMSZ	16
51 
52 extern char	*basename();
53 
54 static void	get_fxdptbl(), set_fxdptbl();
55 
56 static char usage[] = "usage:	dispadmin -l\n"
57 			    "dispadmin -c FX -g [-r res]\n"
58 			    "dispadmin -c FX -s infile\n";
59 
60 static char	basenm[BASENMSZ];
61 static char	cmdpath[256];
62 
63 
64 int
main(int argc,char ** argv)65 main(int argc, char **argv)
66 {
67 	int		c;
68 	int		lflag, gflag, rflag, sflag;
69 	ulong_t		res;
70 	char		*infile;
71 	char 		*endp;
72 
73 	(void) strcpy(cmdpath, argv[0]);
74 	(void) strcpy(basenm, basename(argv[0]));
75 	lflag = gflag = rflag = sflag = 0;
76 	while ((c = getopt(argc, argv, "lc:gr:s:")) != -1) {
77 		switch (c) {
78 
79 		case 'l':
80 			lflag++;
81 			break;
82 
83 		case 'c':
84 			if (strcmp(optarg, "FX") != 0)
85 				fatalerr("error: %s executed for %s class, "
86 				    "%s is actually sub-command for FX class\n",
87 				    cmdpath, optarg, cmdpath);
88 			break;
89 
90 		case 'g':
91 			gflag++;
92 			break;
93 
94 		case 'r':
95 			rflag++;
96 			errno = 0;
97 			res = strtoul(optarg, &endp, 10);
98 			if (res == 0 || errno != 0 || *endp != '\0')
99 				fatalerr("%s: Can't convert to requested "
100 				    "resolution\n", basenm);
101 			break;
102 
103 		case 's':
104 			sflag++;
105 			infile = optarg;
106 			break;
107 
108 		case '?':
109 			fatalerr(usage);
110 
111 		default:
112 			break;
113 		}
114 	}
115 
116 
117 	if (lflag) {
118 		if (gflag || rflag || sflag)
119 			fatalerr(usage);
120 
121 		(void) printf("FX\t(Fixed Priority)\n");
122 		return (0);
123 
124 	} else if (gflag) {
125 		if (sflag)
126 			fatalerr(usage);
127 
128 		if (rflag == 0)
129 			res = 1000;
130 
131 		get_fxdptbl(res);
132 		return (0);
133 
134 	} else if (sflag) {
135 		if (rflag)
136 			fatalerr(usage);
137 
138 		set_fxdptbl(infile);
139 		return (0);
140 
141 	} else {
142 		fatalerr(usage);
143 	}
144 	return (1);
145 }
146 
147 
148 /*
149  * Retrieve the current fx_dptbl from memory, convert the time quantum
150  * values to the resolution specified by res and write the table to stdout.
151  */
152 static void
get_fxdptbl(ulong_t res)153 get_fxdptbl(ulong_t res)
154 {
155 	int		i;
156 	int		fxdpsz;
157 	pcinfo_t	pcinfo;
158 	pcadmin_t	pcadmin;
159 	fxadmin_t	fxadmin;
160 	fxdpent_t	*fx_dptbl;
161 	hrtimer_t	hrtime;
162 
163 	(void) strcpy(pcinfo.pc_clname, "FX");
164 	if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
165 		fatalerr("%s: Can't get FX class ID, priocntl system call "
166 		    "failed with errno %d\n", basenm, errno);
167 
168 	pcadmin.pc_cid = pcinfo.pc_cid;
169 	pcadmin.pc_cladmin = (char *)&fxadmin;
170 	fxadmin.fx_cmd = FX_GETDPSIZE;
171 
172 	if (priocntl(0, 0, PC_ADMIN, (caddr_t)&pcadmin) == -1)
173 		fatalerr("%s: Can't get fx_dptbl size, priocntl system call "
174 		    "failed with errno %d\n", basenm, errno);
175 
176 	fxdpsz = fxadmin.fx_ndpents * sizeof (fxdpent_t);
177 	if ((fx_dptbl = (fxdpent_t *)malloc(fxdpsz)) == NULL)
178 		fatalerr("%s: Can't allocate memory for fx_dptbl\n", basenm);
179 
180 	fxadmin.fx_dpents = fx_dptbl;
181 
182 	fxadmin.fx_cmd = FX_GETDPTBL;
183 	if (priocntl(0, 0, PC_ADMIN, (caddr_t)&pcadmin) == -1)
184 		fatalerr("%s: Can't get fx_dptbl, priocntl system call "
185 		    "failed with errno %d\n", basenm, errno);
186 
187 	(void) printf("# Fixed Priority Dispatcher Configuration\n");
188 	(void) printf("RES=%ld\n\n", res);
189 	(void) printf("# TIME QUANTUM                    PRIORITY\n");
190 	(void) printf("# (fx_quantum)                      LEVEL\n");
191 
192 	for (i = 0; i < fxadmin.fx_ndpents; i++) {
193 		if (res != HZ && fx_dptbl[i].fx_quantum != FX_TQINF) {
194 			hrtime.hrt_secs = 0;
195 			hrtime.hrt_rem = fx_dptbl[i].fx_quantum;
196 			hrtime.hrt_res = HZ;
197 			if (_hrtnewres(&hrtime, res, HRT_RNDUP) == -1)
198 				fatalerr("%s: Can't convert to requested "
199 				    "resolution\n", basenm);
200 			if ((fx_dptbl[i].fx_quantum = hrtconvert(&hrtime))
201 									== -1)
202 				fatalerr("%s: Can't express time quantum in "
203 				    "requested resolution,\n"
204 				    "try coarser resolution\n", basenm);
205 		}
206 		(void) printf("%10d                    #      %3d\n",
207 		    fx_dptbl[i].fx_quantum, i);
208 	}
209 }
210 
211 
212 /*
213  * Read the fx_dptbl values from infile, convert the time quantum values
214  * to HZ resolution, do a little sanity checking and overwrite the table
215  * in memory with the values from the file.
216  */
217 static void
set_fxdptbl(char * infile)218 set_fxdptbl(char *infile)
219 {
220 	int		i;
221 	int		nfxdpents;
222 	char		*tokp;
223 	pcinfo_t	pcinfo;
224 	pcadmin_t	pcadmin;
225 	fxadmin_t	fxadmin;
226 	fxdpent_t	*fx_dptbl;
227 	int		linenum;
228 	ulong_t		res;
229 	hrtimer_t	hrtime;
230 	FILE		*fp;
231 	char		buf[512];
232 	int		wslength;
233 	char 		*endp;
234 	char		name[512], value[512];
235 	int		len;
236 
237 	(void) strcpy(pcinfo.pc_clname, "FX");
238 	if (priocntl(0, 0, PC_GETCID, (caddr_t)&pcinfo) == -1)
239 		fatalerr("%s: Can't get FX class ID, priocntl system call "
240 		    "failed with errno %d\n", basenm, errno);
241 
242 	pcadmin.pc_cid = pcinfo.pc_cid;
243 	pcadmin.pc_cladmin = (char *)&fxadmin;
244 	fxadmin.fx_cmd = FX_GETDPSIZE;
245 
246 	if (priocntl(0, 0, PC_ADMIN, (caddr_t)&pcadmin) == -1)
247 		fatalerr("%s: Can't get fx_dptbl size, priocntl system call "
248 		    "failed with errno %d\n", basenm, errno);
249 
250 	nfxdpents = fxadmin.fx_ndpents;
251 	if ((fx_dptbl =
252 	    (fxdpent_t *)malloc(nfxdpents * sizeof (fxdpent_t))) == NULL)
253 		fatalerr("%s: Can't allocate memory for fx_dptbl\n", basenm);
254 
255 	if ((fp = fopen(infile, "r")) == NULL)
256 		fatalerr("%s: Can't open %s for input\n", basenm, infile);
257 
258 	linenum = 0;
259 
260 	/*
261 	 * Find the first non-blank, non-comment line.  A comment line
262 	 * is any line with '#' as the first non-white-space character.
263 	 */
264 	do {
265 		if (fgets(buf, sizeof (buf), fp) == NULL)
266 			fatalerr("%s: Too few lines in input table\n", basenm);
267 		linenum++;
268 	} while (buf[0] == '#' || buf[0] == '\0' ||
269 	    (wslength = strspn(buf, " \t\n")) == strlen(buf) ||
270 	    strchr(buf, '#') == buf + wslength);
271 
272 	/* LINTED - unbounded string specifier */
273 	if (sscanf(buf, " %[^=]=%s \n%n", name, value, &len) == 2 &&
274 	    name[0] != '\0' && value[0] != '\0' && len == strlen(buf)) {
275 
276 		if (strcmp(name, "RES") == 0) {
277 			errno = 0;
278 			i = (int)strtol(value, &endp, 10);
279 			if (errno != 0 || endp == value ||
280 			    i < 0 || *endp != '\0')
281 				fatalerr("%s: Bad RES specification, "
282 				    "line %d of input file\n", basenm, linenum);
283 			else
284 				res = i;
285 		} else {
286 			fatalerr("%s: Bad RES specification, "
287 			    "line %d of input file\n", basenm, linenum);
288 		}
289 	}
290 
291 	/*
292 	 * The remainder of the input file should contain exactly enough
293 	 * non-blank, non-comment lines to fill the table (fx_ndpents lines).
294 	 * We assume that any non-blank, non-comment line is data for the
295 	 * table and fail if we find more or less than we need.
296 	 */
297 	for (i = 0; i < fxadmin.fx_ndpents; i++) {
298 
299 		/*
300 		 * Get the next non-blank, non-comment line.
301 		 */
302 		do {
303 			if (fgets(buf, sizeof (buf), fp) == NULL)
304 				fatalerr("%s: Too few lines in input table\n",
305 				    basenm);
306 			linenum++;
307 		} while (buf[0] == '#' || buf[0] == '\0' ||
308 		    (wslength = strspn(buf, " \t\n")) == strlen(buf) ||
309 		    strchr(buf, '#') == buf + wslength);
310 
311 		if ((tokp = strtok(buf, " \t")) == NULL)
312 			fatalerr("%s: Too few values, line %d of input file\n",
313 			    basenm, linenum);
314 
315 		fx_dptbl[i].fx_quantum = atol(tokp);
316 		if (fx_dptbl[i].fx_quantum <= 0) {
317 				fatalerr("%s: fx_quantum value out of "
318 				    "valid range; line %d of input,\n"
319 				    "table not overwritten\n", basenm, linenum);
320 		} else if (res != HZ) {
321 			hrtime.hrt_secs = 0;
322 			hrtime.hrt_rem = fx_dptbl[i].fx_quantum;
323 			hrtime.hrt_res = res;
324 			if (_hrtnewres(&hrtime, HZ, HRT_RNDUP) == -1)
325 				fatalerr("%s: Can't convert specified "
326 				    "resolution to ticks\n", basenm);
327 			if ((fx_dptbl[i].fx_quantum =
328 			    hrtconvert(&hrtime)) == -1)
329 				fatalerr("%s: fx_quantum value out of "
330 				    "valid range; line %d of input,\n"
331 				    "table not overwritten\n", basenm, linenum);
332 		}
333 
334 		if ((tokp = strtok(NULL, " \t")) != NULL && tokp[0] != '#')
335 			fatalerr("%s: Too many values, line %d of input file\n",
336 			    basenm, linenum);
337 	}
338 
339 	/*
340 	 * We've read enough lines to fill the table.  We fail
341 	 * if the input file contains any more.
342 	 */
343 	while (fgets(buf, sizeof (buf), fp) != NULL) {
344 		if (buf[0] != '#' && buf[0] != '\0' &&
345 		    (wslength = strspn(buf, " \t\n")) != strlen(buf) &&
346 		    strchr(buf, '#') != buf + wslength)
347 			fatalerr("%s: Too many lines in input table\n",
348 				basenm);
349 	}
350 
351 	fxadmin.fx_dpents = fx_dptbl;
352 	fxadmin.fx_cmd = FX_SETDPTBL;
353 	if (priocntl(0, 0, PC_ADMIN, (caddr_t)&pcadmin) == -1)
354 		fatalerr("%s: Can't set fx_dptbl, priocntl system call failed "
355 		    "with errno %d\n", basenm, errno);
356 }
357