xref: /netbsd-src/usr.bin/nice/nice.c (revision d48f14661dda8638fee055ba15d35bdfb29b9fa8)
1 /*	$NetBSD: nice.c,v 1.13 2005/10/05 21:20:46 christos Exp $	*/
2 
3 /*
4  * Copyright (c) 1989 The Regents of the University of California.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #include <sys/cdefs.h>
33 #ifndef lint
34 __COPYRIGHT(
35     "@(#) Copyright (c) 1989 The Regents of the University of California.\n\
36  All rights reserved.\n");
37 #endif /* not lint */
38 
39 #ifndef lint
40 #if 0
41 static char sccsid[] = "@(#)nice.c	5.4 (Berkeley) 6/1/90";
42 #endif
43 __RCSID("$NetBSD: nice.c,v 1.13 2005/10/05 21:20:46 christos Exp $");
44 #endif /* not lint */
45 
46 #include <sys/time.h>
47 #include <sys/resource.h>
48 
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <string.h>
52 #include <limits.h>
53 #include <locale.h>
54 #include <ctype.h>
55 #include <errno.h>
56 #include <err.h>
57 #include <unistd.h>
58 
59 #define	DEFNICE	10
60 
61 static void usage(void) __attribute__((__noreturn__));
62 
63 int
64 main(int argc, char **argv)
65 {
66 	char *ep;
67 	int niceness = DEFNICE;
68 	int c;
69 	long tmp;
70 
71 	setprogname(argv[0]);
72 	(void)setlocale(LC_ALL, "");
73 
74         /* handle obsolete -number syntax */
75         if (argc > 1 && argv[1][0] == '-' &&
76 	    isdigit((unsigned char)argv[1][1])) {
77 		niceness = atoi (argv[1] + 1);
78                 argc--; argv++;
79         }
80 
81 	while ((c = getopt (argc, argv, "n:")) != -1) {
82 		switch (c) {
83 		case 'n':
84 			errno = 0;
85 			tmp = strtol(optarg, &ep, 10);
86 			if (*ep != '\0' || tmp < INT_MIN || tmp > INT_MAX)
87 				errx(EXIT_FAILURE, "invalid argument: `%s'",
88 				    optarg);
89 			niceness = (int)tmp;
90 			break;
91 		default:
92 			usage();
93 			break;
94 		}
95 	}
96 	argc -= optind;
97 	argv += optind;
98 
99 	if (argc == 0)
100 		usage();
101 
102 	errno = 0;
103 	niceness += getpriority(PRIO_PROCESS, 0);
104 	if (errno) {
105 		err(EXIT_FAILURE, "getpriority");
106 		/* NOTREACHED */
107 	}
108 	if (setpriority(PRIO_PROCESS, 0, niceness) == -1) {
109 		warn("setpriority");
110 	}
111 
112 	(void)execvp(argv[0], &argv[0]);
113 	err((errno == ENOENT || errno == ENOTDIR) ? 127 : 126, "%s", argv[0]);
114 	/* NOTREACHED */
115 }
116 
117 static void
118 usage(void)
119 {
120 	(void)fprintf(stderr,
121 	    "Usage: %s [ -n increment ] utility [ argument ...]\n",
122 	    getprogname());
123 	exit(EXIT_FAILURE);
124 }
125