rtprio.c revision 3291 
1/*
2 * Copyright (c) 1994 David Greenman
3 * Copyright (c) 1994 Henrik Vestergaard Draboel (hvd@terry.ping.dk)
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 * 3. All advertising materials mentioning features or use of this software
15 *    must display the following acknowledgement:
16 *	This product includes software developed by Henrik Vestergaard Draboel.
17 *	This product includes software developed by David Greenman.
18 * 4. Neither the names of the authors nor the names of contributors
19 *    may be used to endorse or promote products derived from this software
20 *    without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 *
34 *	$Id$
35 */
36
37#include <sys/param.h>
38#include <sys/rtprio.h>
39#include <sys/errno.h>
40
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <ctype.h>
45#include <unistd.h>
46
47static void usage();
48
49int
50main(argc, argv)
51	int     argc;
52	char  **argv;
53{
54	char   *p;
55	int     proc = 0;
56	struct rtprio rtp;
57
58	/* find basename */
59	if ((p = rindex(argv[0], '/')) == NULL)
60		p = argv[0];
61	else
62		++p;
63
64	if (!strcmp(p, "rtprio"))
65		rtp.type = RTP_PRIO_REALTIME;
66	else if (!strcmp(p, "idprio"))
67		rtp.type = RTP_PRIO_IDLE;
68
69	switch (argc) {
70	case 2:
71		proc = abs(atoi(argv[1]));	/* Should check if numeric
72						 * arg! */
73		/* FALLTHROUGH */
74	case 1:
75		if (rtprio(RTP_LOOKUP, proc, &rtp) != 0) {
76			perror(argv[0]);
77			exit (1);
78		}
79		printf("%s: ", p);
80		switch (rtp.type) {
81		case RTP_PRIO_REALTIME:
82			printf("realtime priority %d\n", rtp.prio);
83			break;
84		case RTP_PRIO_NORMAL:
85			printf("normal priority\n");
86			break;
87		case RTP_PRIO_IDLE:
88			printf("idle priority %d\n", rtp.prio);
89			break;
90		default:
91			printf("invalid priority type %d\n", rtp.type);
92			break;
93		}
94		exit(0);
95	default:
96		if (argv[1][0] == '-' || isdigit(argv[1][0])) {
97			if (argv[1][0] == '-') {
98				if (strcmp(argv[1], "-t") == 0) {
99					rtp.type = RTP_PRIO_NORMAL;
100					rtp.prio = 0;
101				} else {
102					usage(p);
103					break;
104				}
105			} else {
106				rtp.prio = atoi(argv[1]);
107			}
108		} else {
109			usage(p);
110			break;
111		}
112
113		if (argv[2][0] == '-')
114			proc = -atoi(argv[2]);
115
116		if (rtprio(RTP_SET, proc, &rtp) != 0) {
117			perror(argv[0]);
118			exit (1);
119		}
120
121		if (proc == 0) {
122			execvp(argv[2], &argv[2]);
123			perror(argv[0]);
124			exit (1);
125		}
126	}
127	exit (1);
128}
129
130static void
131usage(basename)
132	char   *basename;
133{
134	(void) fprintf(stderr, "usage: %s\n", basename);
135	(void) fprintf(stderr, "usage: %s [-]pid\n", basename);
136	(void) fprintf(stderr, "usage: %s priority command [ args ] \n", basename);
137	(void) fprintf(stderr, "usage: %s priority -pid \n", basename);
138	(void) fprintf(stderr, "usage: %s -t command [ args ] \n", basename);
139	(void) fprintf(stderr, "usage: %s -t -pid \n", basename);
140
141	exit(-1);
142}
143