1/*
2 * Copyright (c) 1983, 1989, 1993
3 *	The Regents of the University of California.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 4. Neither the name of the University nor the names of its contributors
14 *    may be used to endorse or promote products derived from this software
15 *    without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30#ifndef lint
31static const char copyright[] =
32"@(#) Copyright (c) 1983, 1989, 1993\n\
33	The Regents of the University of California.  All rights reserved.\n";
34#endif /* not lint */
35
36#if 0
37#ifndef lint
38static char sccsid[] = "@(#)renice.c	8.1 (Berkeley) 6/9/93";
39#endif /* not lint */
40#endif
41
42#include <sys/cdefs.h>
43__FBSDID("$FreeBSD$");
44
45#include <sys/types.h>
46#include <sys/time.h>
47#include <sys/resource.h>
48
49#include <err.h>
50#include <errno.h>
51#include <limits.h>
52#include <pwd.h>
53#include <stdio.h>
54#include <stdlib.h>
55#include <string.h>
56
57static int	donice(int, int, int, int);
58static int	getnum(const char *, const char *, int *);
59static void	usage(void);
60
61/*
62 * Change the priority (nice) of processes
63 * or groups of processes which are already
64 * running.
65 */
66int
67main(int argc, char *argv[])
68{
69	struct passwd *pwd;
70	int errs, incr, prio, which, who;
71
72	errs = 0;
73	incr = 0;
74	which = PRIO_PROCESS;
75	who = 0;
76	argc--, argv++;
77	if (argc < 2)
78		usage();
79	if (strcmp(*argv, "-n") == 0) {
80		incr = 1;
81		argc--, argv++;
82		if (argc < 2)
83			usage();
84	}
85	if (getnum("priority", *argv, &prio))
86		return (1);
87	argc--, argv++;
88	for (; argc > 0; argc--, argv++) {
89		if (strcmp(*argv, "-g") == 0) {
90			which = PRIO_PGRP;
91			continue;
92		}
93		if (strcmp(*argv, "-u") == 0) {
94			which = PRIO_USER;
95			continue;
96		}
97		if (strcmp(*argv, "-p") == 0) {
98			which = PRIO_PROCESS;
99			continue;
100		}
101		if (which == PRIO_USER) {
102			if ((pwd = getpwnam(*argv)) != NULL)
103				who = pwd->pw_uid;
104			else if (getnum("uid", *argv, &who)) {
105				errs++;
106				continue;
107			} else if (who < 0) {
108				warnx("%s: bad value", *argv);
109				errs++;
110				continue;
111			}
112		} else {
113			if (getnum("pid", *argv, &who)) {
114				errs++;
115				continue;
116			}
117			if (who < 0) {
118				warnx("%s: bad value", *argv);
119				errs++;
120				continue;
121			}
122		}
123		errs += donice(which, who, prio, incr);
124	}
125	exit(errs != 0);
126}
127
128static int
129donice(int which, int who, int prio, int incr)
130{
131	int oldprio;
132
133	errno = 0;
134	oldprio = getpriority(which, who);
135	if (oldprio == -1 && errno) {
136		warn("%d: getpriority", who);
137		return (1);
138	}
139	if (incr)
140		prio = oldprio + prio;
141	if (prio > PRIO_MAX)
142		prio = PRIO_MAX;
143	if (prio < PRIO_MIN)
144		prio = PRIO_MIN;
145	if (setpriority(which, who, prio) < 0) {
146		warn("%d: setpriority", who);
147		return (1);
148	}
149	fprintf(stderr, "%d: old priority %d, new priority %d\n", who,
150	    oldprio, prio);
151	return (0);
152}
153
154static int
155getnum(const char *com, const char *str, int *val)
156{
157	long v;
158	char *ep;
159
160	errno = 0;
161	v = strtol(str, &ep, 10);
162	if (v < INT_MIN || v > INT_MAX || errno == ERANGE) {
163		warnx("%s argument %s is out of range.", com, str);
164		return (1);
165	}
166	if (ep == str || *ep != '\0' || errno != 0) {
167		warnx("Bad %s argument: %s.", com, str);
168		return (1);
169	}
170
171	*val = (int)v;
172	return (0);
173}
174
175static void
176usage(void)
177{
178	fprintf(stderr, "%s\n%s\n",
179"usage: renice priority [[-p] pid ...] [[-g] pgrp ...] [[-u] user ...]",
180"       renice -n increment [[-p] pid ...] [[-g] pgrp ...] [[-u] user ...]");
181	exit(1);
182}
183