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 35 #include <sys/cdefs.h> 36 __FBSDID("$FreeBSD$"); 37 38 #include <sys/param.h> 39 #include <sys/rtprio.h> 40 41 #include <ctype.h> 42 #include <err.h> 43 #include <errno.h> 44 #include <stdio.h> 45 #include <stdlib.h> 46 #include <string.h> 47 #include <unistd.h> 48 49 static int parseint(const char *, const char *); 50 static void usage(void); 51 52 int 53 main(int argc, char *argv[]) 54 { 55 struct rtprio rtp; 56 const char *progname; 57 pid_t proc = 0; 58 59 progname = getprogname(); 60 61 if (strcmp(progname, "rtprio") == 0) 62 rtp.type = RTP_PRIO_REALTIME; 63 else if (strcmp(progname, "idprio") == 0) 64 rtp.type = RTP_PRIO_IDLE; 65 else 66 errx(1, "invalid progname"); 67 68 switch (argc) { 69 case 2: 70 proc = parseint(argv[1], "pid"); 71 proc = abs(proc); 72 /* FALLTHROUGH */ 73 case 1: 74 if (rtprio(RTP_LOOKUP, proc, &rtp) != 0) 75 err(1, "RTP_LOOKUP"); 76 switch (rtp.type) { 77 case RTP_PRIO_REALTIME: 78 case RTP_PRIO_FIFO: 79 warnx("realtime priority %d", rtp.prio); 80 break; 81 case RTP_PRIO_NORMAL: 82 warnx("normal priority"); 83 break; 84 case RTP_PRIO_IDLE: 85 warnx("idle priority %d", rtp.prio); 86 break; 87 default: 88 errx(1, "invalid priority type %d", rtp.type); 89 break; 90 } 91 exit(0); 92 default: 93 if (argv[1][0] == '-' || isdigit(argv[1][0])) { 94 if (argv[1][0] == '-') { 95 if (strcmp(argv[1], "-t") == 0) { 96 rtp.type = RTP_PRIO_NORMAL; 97 rtp.prio = 0; 98 } else { 99 usage(); 100 break; 101 } 102 } else 103 rtp.prio = parseint(argv[1], "priority"); 104 } else { 105 usage(); 106 break; 107 } 108 109 if (argv[2][0] == '-') { 110 proc = parseint(argv[2], "pid"); 111 proc = abs(proc); 112 } 113 114 if (rtprio(RTP_SET, proc, &rtp) != 0) 115 err(1, "RTP_SET"); 116 117 if (proc == 0) { 118 execvp(argv[2], &argv[2]); 119 err(1, "execvp: %s", argv[2]); 120 } 121 exit(0); 122 } 123 /* NOTREACHED */ 124 } 125 126 static int 127 parseint(const char *str, const char *errname) 128 { 129 char *endp; 130 long res; 131 132 errno = 0; 133 res = strtol(str, &endp, 10); 134 if (errno != 0 || endp == str || *endp != '\0') 135 errx(1, "%s must be a number", errname); 136 if (res >= INT_MAX) 137 errx(1, "Integer overflow parsing %s", errname); 138 return (res); 139 } 140 141 static void 142 usage(void) 143 { 144 145 (void) fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n", 146 "usage: [id|rt]prio", 147 " [id|rt]prio [-]pid", 148 " [id|rt]prio priority command [args]", 149 " [id|rt]prio priority -pid", 150 " [id|rt]prio -t command [args]", 151 " [id|rt]prio -t -pid"); 152 exit(1); 153 } 154