diff options
author | Denys Vlasenko <vda.linux@googlemail.com> | 2017-04-12 14:16:29 +0200 |
---|---|---|
committer | Denys Vlasenko <vda.linux@googlemail.com> | 2017-04-12 14:16:29 +0200 |
commit | 0cecbe7d5de237a6c699c67ae53ae2e2481eff43 (patch) | |
tree | a775f19e4b5739d688c56f633584728718e2ae40 /util-linux | |
parent | 783d57af7bb2b851c16cf87df848e0365e5052da (diff) | |
download | busybox-w32-0cecbe7d5de237a6c699c67ae53ae2e2481eff43.tar.gz busybox-w32-0cecbe7d5de237a6c699c67ae53ae2e2481eff43.tar.bz2 busybox-w32-0cecbe7d5de237a6c699c67ae53ae2e2481eff43.zip |
Sort more misplaced applets into coreutils or util-linux
No code changes
Surprisingly, nice and renice are coming from different packages :)
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
Diffstat (limited to 'util-linux')
-rw-r--r-- | util-linux/cal.c | 390 | ||||
-rw-r--r-- | util-linux/mesg.c | 76 | ||||
-rw-r--r-- | util-linux/renice.c | 148 |
3 files changed, 614 insertions, 0 deletions
diff --git a/util-linux/cal.c b/util-linux/cal.c new file mode 100644 index 000000000..af02608f0 --- /dev/null +++ b/util-linux/cal.c | |||
@@ -0,0 +1,390 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * Calendar implementation for busybox | ||
4 | * | ||
5 | * See original copyright at the end of this file | ||
6 | * | ||
7 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. | ||
8 | */ | ||
9 | /* Mar 16, 2003 Manuel Novoa III (mjn3@codepoet.org) | ||
10 | * | ||
11 | * Major size reduction... over 50% (>1.5k) on i386. | ||
12 | */ | ||
13 | //config:config CAL | ||
14 | //config: bool "cal" | ||
15 | //config: default y | ||
16 | //config: help | ||
17 | //config: cal is used to display a monthly calendar. | ||
18 | |||
19 | //applet:IF_CAL(APPLET(cal, BB_DIR_USR_BIN, BB_SUID_DROP)) | ||
20 | |||
21 | //kbuild:lib-$(CONFIG_CAL) += cal.o | ||
22 | |||
23 | /* BB_AUDIT SUSv3 compliant with -j and -y extensions (from util-linux). */ | ||
24 | /* BB_AUDIT BUG: The output of 'cal -j 1752' is incorrect. The upstream | ||
25 | * BB_AUDIT BUG: version in util-linux seems to be broken as well. */ | ||
26 | /* http://www.opengroup.org/onlinepubs/007904975/utilities/cal.html */ | ||
27 | |||
28 | //usage:#define cal_trivial_usage | ||
29 | //usage: "[-jy] [[MONTH] YEAR]" | ||
30 | //usage:#define cal_full_usage "\n\n" | ||
31 | //usage: "Display a calendar\n" | ||
32 | //usage: "\n -j Use julian dates" | ||
33 | //usage: "\n -y Display the entire year" | ||
34 | |||
35 | #include "libbb.h" | ||
36 | #include "unicode.h" | ||
37 | |||
38 | /* We often use "unsigned" intead of "int", it's easier to div on most CPUs */ | ||
39 | |||
40 | #define THURSDAY 4 /* for reformation */ | ||
41 | #define SATURDAY 6 /* 1 Jan 1 was a Saturday */ | ||
42 | |||
43 | #define FIRST_MISSING_DAY 639787 /* 3 Sep 1752 */ | ||
44 | #define NUMBER_MISSING_DAYS 11 /* 11 day correction */ | ||
45 | |||
46 | #define MAXDAYS 42 /* max slots in a month array */ | ||
47 | #define SPACE -1 /* used in day array */ | ||
48 | |||
49 | static const unsigned char days_in_month[] ALIGN1 = { | ||
50 | 0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 | ||
51 | }; | ||
52 | |||
53 | static const unsigned char sep1752[] ALIGN1 = { | ||
54 | 1, 2, 14, 15, 16, | ||
55 | 17, 18, 19, 20, 21, 22, 23, | ||
56 | 24, 25, 26, 27, 28, 29, 30 | ||
57 | }; | ||
58 | |||
59 | /* Set to 0 or 1 in main */ | ||
60 | #define julian ((unsigned)option_mask32) | ||
61 | |||
62 | /* leap year -- account for Gregorian reformation in 1752 */ | ||
63 | static int leap_year(unsigned yr) | ||
64 | { | ||
65 | if (yr <= 1752) | ||
66 | return !(yr % 4); | ||
67 | return (!(yr % 4) && (yr % 100)) || !(yr % 400); | ||
68 | } | ||
69 | |||
70 | /* number of centuries since 1700, not inclusive */ | ||
71 | #define centuries_since_1700(yr) \ | ||
72 | ((yr) > 1700 ? (yr) / 100 - 17 : 0) | ||
73 | |||
74 | /* number of centuries since 1700 whose modulo of 400 is 0 */ | ||
75 | #define quad_centuries_since_1700(yr) \ | ||
76 | ((yr) > 1600 ? ((yr) - 1600) / 400 : 0) | ||
77 | |||
78 | /* number of leap years between year 1 and this year, not inclusive */ | ||
79 | #define leap_years_since_year_1(yr) \ | ||
80 | ((yr) / 4 - centuries_since_1700(yr) + quad_centuries_since_1700(yr)) | ||
81 | |||
82 | static void center(char *, unsigned, unsigned); | ||
83 | static void day_array(unsigned, unsigned, unsigned *); | ||
84 | static void trim_trailing_spaces_and_print(char *); | ||
85 | |||
86 | static void blank_string(char *buf, size_t buflen); | ||
87 | static char *build_row(char *p, unsigned *dp); | ||
88 | |||
89 | #define DAY_LEN 3 /* 3 spaces per day */ | ||
90 | #define J_DAY_LEN (DAY_LEN + 1) | ||
91 | #define WEEK_LEN 20 /* 7 * 3 - one space at the end */ | ||
92 | #define J_WEEK_LEN (WEEK_LEN + 7) | ||
93 | #define HEAD_SEP 2 /* spaces between day headings */ | ||
94 | |||
95 | int cal_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
96 | int cal_main(int argc UNUSED_PARAM, char **argv) | ||
97 | { | ||
98 | struct tm zero_tm; | ||
99 | time_t now; | ||
100 | unsigned month, year, flags, i; | ||
101 | char *month_names[12]; | ||
102 | /* normal heading: */ | ||
103 | /* "Su Mo Tu We Th Fr Sa" */ | ||
104 | /* -j heading: */ | ||
105 | /* " Su Mo Tu We Th Fr Sa" */ | ||
106 | char day_headings[ENABLE_UNICODE_SUPPORT ? 28 * 6 : 28]; | ||
107 | IF_UNICODE_SUPPORT(char *hp = day_headings;) | ||
108 | char buf[40]; | ||
109 | |||
110 | init_unicode(); | ||
111 | |||
112 | flags = getopt32(argv, "jy"); | ||
113 | /* This sets julian = flags & 1: */ | ||
114 | option_mask32 &= 1; | ||
115 | month = 0; | ||
116 | argv += optind; | ||
117 | |||
118 | if (!argv[0]) { | ||
119 | struct tm *ptm; | ||
120 | |||
121 | time(&now); | ||
122 | ptm = localtime(&now); | ||
123 | year = ptm->tm_year + 1900; | ||
124 | if (!(flags & 2)) { /* no -y */ | ||
125 | month = ptm->tm_mon + 1; | ||
126 | } | ||
127 | } else { | ||
128 | if (argv[1]) { | ||
129 | if (argv[2]) { | ||
130 | bb_show_usage(); | ||
131 | } | ||
132 | if (!(flags & 2)) { /* no -y */ | ||
133 | month = xatou_range(*argv, 1, 12); | ||
134 | } | ||
135 | argv++; | ||
136 | } | ||
137 | year = xatou_range(*argv, 1, 9999); | ||
138 | } | ||
139 | |||
140 | blank_string(day_headings, sizeof(day_headings) - 7 + 7*julian); | ||
141 | |||
142 | i = 0; | ||
143 | do { | ||
144 | zero_tm.tm_mon = i; | ||
145 | /* full month name according to locale */ | ||
146 | strftime(buf, sizeof(buf), "%B", &zero_tm); | ||
147 | month_names[i] = xstrdup(buf); | ||
148 | |||
149 | if (i < 7) { | ||
150 | zero_tm.tm_wday = i; | ||
151 | /* abbreviated weekday name according to locale */ | ||
152 | strftime(buf, sizeof(buf), "%a", &zero_tm); | ||
153 | #if ENABLE_UNICODE_SUPPORT | ||
154 | if (julian) | ||
155 | *hp++ = ' '; | ||
156 | { | ||
157 | char *two_wchars = unicode_conv_to_printable_fixedwidth(/*NULL,*/ buf, 2); | ||
158 | strcpy(hp, two_wchars); | ||
159 | free(two_wchars); | ||
160 | } | ||
161 | hp += strlen(hp); | ||
162 | *hp++ = ' '; | ||
163 | #else | ||
164 | strncpy(day_headings + i * (3+julian) + julian, buf, 2); | ||
165 | #endif | ||
166 | } | ||
167 | } while (++i < 12); | ||
168 | IF_UNICODE_SUPPORT(hp[-1] = '\0';) | ||
169 | |||
170 | if (month) { | ||
171 | unsigned row, len, days[MAXDAYS]; | ||
172 | unsigned *dp = days; | ||
173 | char lineout[30]; | ||
174 | |||
175 | day_array(month, year, dp); | ||
176 | len = sprintf(lineout, "%s %u", month_names[month - 1], year); | ||
177 | printf("%*s%s\n%s\n", | ||
178 | ((7*julian + WEEK_LEN) - len) / 2, "", | ||
179 | lineout, day_headings); | ||
180 | for (row = 0; row < 6; row++) { | ||
181 | build_row(lineout, dp)[0] = '\0'; | ||
182 | dp += 7; | ||
183 | trim_trailing_spaces_and_print(lineout); | ||
184 | } | ||
185 | } else { | ||
186 | unsigned row, which_cal, week_len, days[12][MAXDAYS]; | ||
187 | unsigned *dp; | ||
188 | char lineout[80]; | ||
189 | |||
190 | sprintf(lineout, "%u", year); | ||
191 | center(lineout, | ||
192 | (WEEK_LEN * 3 + HEAD_SEP * 2) | ||
193 | + julian * (J_WEEK_LEN * 2 + HEAD_SEP | ||
194 | - (WEEK_LEN * 3 + HEAD_SEP * 2)), | ||
195 | 0 | ||
196 | ); | ||
197 | puts("\n"); /* two \n's */ | ||
198 | for (i = 0; i < 12; i++) { | ||
199 | day_array(i + 1, year, days[i]); | ||
200 | } | ||
201 | blank_string(lineout, sizeof(lineout)); | ||
202 | week_len = WEEK_LEN + julian * (J_WEEK_LEN - WEEK_LEN); | ||
203 | for (month = 0; month < 12; month += 3-julian) { | ||
204 | center(month_names[month], week_len, HEAD_SEP); | ||
205 | if (!julian) { | ||
206 | center(month_names[month + 1], week_len, HEAD_SEP); | ||
207 | } | ||
208 | center(month_names[month + 2 - julian], week_len, 0); | ||
209 | printf("\n%s%*s%s", day_headings, HEAD_SEP, "", day_headings); | ||
210 | if (!julian) { | ||
211 | printf("%*s%s", HEAD_SEP, "", day_headings); | ||
212 | } | ||
213 | bb_putchar('\n'); | ||
214 | for (row = 0; row < (6*7); row += 7) { | ||
215 | for (which_cal = 0; which_cal < 3-julian; which_cal++) { | ||
216 | dp = days[month + which_cal] + row; | ||
217 | build_row(lineout + which_cal * (week_len + 2), dp); | ||
218 | } | ||
219 | /* blank_string took care of nul termination. */ | ||
220 | trim_trailing_spaces_and_print(lineout); | ||
221 | } | ||
222 | } | ||
223 | } | ||
224 | |||
225 | fflush_stdout_and_exit(EXIT_SUCCESS); | ||
226 | } | ||
227 | |||
228 | /* | ||
229 | * day_array -- | ||
230 | * Fill in an array of 42 integers with a calendar. Assume for a moment | ||
231 | * that you took the (maximum) 6 rows in a calendar and stretched them | ||
232 | * out end to end. You would have 42 numbers or spaces. This routine | ||
233 | * builds that array for any month from Jan. 1 through Dec. 9999. | ||
234 | */ | ||
235 | static void day_array(unsigned month, unsigned year, unsigned *days) | ||
236 | { | ||
237 | unsigned long temp; | ||
238 | unsigned i; | ||
239 | unsigned day, dw, dm; | ||
240 | |||
241 | memset(days, SPACE, MAXDAYS * sizeof(int)); | ||
242 | |||
243 | if ((month == 9) && (year == 1752)) { | ||
244 | /* Assumes the Gregorian reformation eliminates | ||
245 | * 3 Sep. 1752 through 13 Sep. 1752. | ||
246 | */ | ||
247 | unsigned j_offset = julian * 244; | ||
248 | size_t oday = 0; | ||
249 | |||
250 | do { | ||
251 | days[oday+2] = sep1752[oday] + j_offset; | ||
252 | } while (++oday < sizeof(sep1752)); | ||
253 | |||
254 | return; | ||
255 | } | ||
256 | |||
257 | /* day_in_year | ||
258 | * return the 1 based day number within the year | ||
259 | */ | ||
260 | day = 1; | ||
261 | if ((month > 2) && leap_year(year)) { | ||
262 | ++day; | ||
263 | } | ||
264 | |||
265 | i = month; | ||
266 | while (i) { | ||
267 | day += days_in_month[--i]; | ||
268 | } | ||
269 | |||
270 | /* day_in_week | ||
271 | * return the 0 based day number for any date from 1 Jan. 1 to | ||
272 | * 31 Dec. 9999. Assumes the Gregorian reformation eliminates | ||
273 | * 3 Sep. 1752 through 13 Sep. 1752. Returns Thursday for all | ||
274 | * missing days. | ||
275 | */ | ||
276 | temp = (long)(year - 1) * 365 + leap_years_since_year_1(year - 1) + day; | ||
277 | if (temp < FIRST_MISSING_DAY) { | ||
278 | dw = ((temp - 1 + SATURDAY) % 7); | ||
279 | } else { | ||
280 | dw = (((temp - 1 + SATURDAY) - NUMBER_MISSING_DAYS) % 7); | ||
281 | } | ||
282 | |||
283 | if (!julian) { | ||
284 | day = 1; | ||
285 | } | ||
286 | |||
287 | dm = days_in_month[month]; | ||
288 | if ((month == 2) && leap_year(year)) { | ||
289 | ++dm; | ||
290 | } | ||
291 | |||
292 | do { | ||
293 | days[dw++] = day++; | ||
294 | } while (--dm); | ||
295 | } | ||
296 | |||
297 | static void trim_trailing_spaces_and_print(char *s) | ||
298 | { | ||
299 | char *p = s; | ||
300 | |||
301 | while (*p) { | ||
302 | ++p; | ||
303 | } | ||
304 | while (p != s) { | ||
305 | --p; | ||
306 | if (!isspace(*p)) { | ||
307 | p[1] = '\0'; | ||
308 | break; | ||
309 | } | ||
310 | } | ||
311 | |||
312 | puts(s); | ||
313 | } | ||
314 | |||
315 | static void center(char *str, unsigned len, unsigned separate) | ||
316 | { | ||
317 | unsigned n = strlen(str); | ||
318 | len -= n; | ||
319 | printf("%*s%*s", (len/2) + n, str, (len/2) + (len % 2) + separate, ""); | ||
320 | } | ||
321 | |||
322 | static void blank_string(char *buf, size_t buflen) | ||
323 | { | ||
324 | memset(buf, ' ', buflen); | ||
325 | buf[buflen-1] = '\0'; | ||
326 | } | ||
327 | |||
328 | static char *build_row(char *p, unsigned *dp) | ||
329 | { | ||
330 | unsigned col, val, day; | ||
331 | |||
332 | memset(p, ' ', (julian + DAY_LEN) * 7); | ||
333 | |||
334 | col = 0; | ||
335 | do { | ||
336 | day = *dp++; | ||
337 | if (day != SPACE) { | ||
338 | if (julian) { | ||
339 | ++p; | ||
340 | if (day >= 100) { | ||
341 | *p = '0'; | ||
342 | p[-1] = (day / 100) + '0'; | ||
343 | day %= 100; | ||
344 | } | ||
345 | } | ||
346 | val = day / 10; | ||
347 | if (val > 0) { | ||
348 | *p = val + '0'; | ||
349 | } | ||
350 | *++p = day % 10 + '0'; | ||
351 | p += 2; | ||
352 | } else { | ||
353 | p += DAY_LEN + julian; | ||
354 | } | ||
355 | } while (++col < 7); | ||
356 | |||
357 | return p; | ||
358 | } | ||
359 | |||
360 | /* | ||
361 | * Copyright (c) 1989, 1993, 1994 | ||
362 | * The Regents of the University of California. All rights reserved. | ||
363 | * | ||
364 | * This code is derived from software contributed to Berkeley by | ||
365 | * Kim Letkeman. | ||
366 | * | ||
367 | * Redistribution and use in source and binary forms, with or without | ||
368 | * modification, are permitted provided that the following conditions | ||
369 | * are met: | ||
370 | * 1. Redistributions of source code must retain the above copyright | ||
371 | * notice, this list of conditions and the following disclaimer. | ||
372 | * 2. Redistributions in binary form must reproduce the above copyright | ||
373 | * notice, this list of conditions and the following disclaimer in the | ||
374 | * documentation and/or other materials provided with the distribution. | ||
375 | * 3. Neither the name of the University nor the names of its contributors | ||
376 | * may be used to endorse or promote products derived from this software | ||
377 | * without specific prior written permission. | ||
378 | * | ||
379 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND | ||
380 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | ||
381 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | ||
382 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE | ||
383 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | ||
384 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | ||
385 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | ||
386 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | ||
387 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | ||
388 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | ||
389 | * SUCH DAMAGE. | ||
390 | */ | ||
diff --git a/util-linux/mesg.c b/util-linux/mesg.c new file mode 100644 index 000000000..45c13b8e0 --- /dev/null +++ b/util-linux/mesg.c | |||
@@ -0,0 +1,76 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * mesg implementation for busybox | ||
4 | * | ||
5 | * Copyright (c) 2002 Manuel Novoa III <mjn3@codepoet.org> | ||
6 | * | ||
7 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. | ||
8 | */ | ||
9 | |||
10 | //config:config MESG | ||
11 | //config: bool "mesg" | ||
12 | //config: default y | ||
13 | //config: help | ||
14 | //config: Mesg controls access to your terminal by others. It is typically | ||
15 | //config: used to allow or disallow other users to write to your terminal | ||
16 | //config: | ||
17 | //config:config FEATURE_MESG_ENABLE_ONLY_GROUP | ||
18 | //config: bool "Enable writing to tty only by group, not by everybody" | ||
19 | //config: default y | ||
20 | //config: depends on MESG | ||
21 | //config: help | ||
22 | //config: Usually, ttys are owned by group "tty", and "write" tool is | ||
23 | //config: setgid to this group. This way, "mesg y" only needs to enable | ||
24 | //config: "write by owning group" bit in tty mode. | ||
25 | //config: | ||
26 | //config: If you set this option to N, "mesg y" will enable writing | ||
27 | //config: by anybody at all. This is not recommended. | ||
28 | |||
29 | //applet:IF_MESG(APPLET(mesg, BB_DIR_USR_BIN, BB_SUID_DROP)) | ||
30 | |||
31 | //kbuild:lib-$(CONFIG_MESG) += mesg.o | ||
32 | |||
33 | //usage:#define mesg_trivial_usage | ||
34 | //usage: "[y|n]" | ||
35 | //usage:#define mesg_full_usage "\n\n" | ||
36 | //usage: "Control write access to your terminal\n" | ||
37 | //usage: " y Allow write access to your terminal\n" | ||
38 | //usage: " n Disallow write access to your terminal" | ||
39 | |||
40 | #include "libbb.h" | ||
41 | |||
42 | #if ENABLE_FEATURE_MESG_ENABLE_ONLY_GROUP | ||
43 | #define S_IWGRP_OR_S_IWOTH S_IWGRP | ||
44 | #else | ||
45 | #define S_IWGRP_OR_S_IWOTH (S_IWGRP | S_IWOTH) | ||
46 | #endif | ||
47 | |||
48 | int mesg_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
49 | int mesg_main(int argc UNUSED_PARAM, char **argv) | ||
50 | { | ||
51 | struct stat sb; | ||
52 | mode_t m; | ||
53 | char c = 0; | ||
54 | |||
55 | argv++; | ||
56 | |||
57 | if (argv[0] | ||
58 | && (argv[1] || ((c = argv[0][0]) != 'y' && c != 'n')) | ||
59 | ) { | ||
60 | bb_show_usage(); | ||
61 | } | ||
62 | |||
63 | if (!isatty(STDIN_FILENO)) | ||
64 | bb_error_msg_and_die("not a tty"); | ||
65 | |||
66 | xfstat(STDIN_FILENO, &sb, "stderr"); | ||
67 | if (c == 0) { | ||
68 | puts((sb.st_mode & (S_IWGRP|S_IWOTH)) ? "is y" : "is n"); | ||
69 | return EXIT_SUCCESS; | ||
70 | } | ||
71 | m = (c == 'y') ? sb.st_mode | S_IWGRP_OR_S_IWOTH | ||
72 | : sb.st_mode & ~(S_IWGRP|S_IWOTH); | ||
73 | if (fchmod(STDIN_FILENO, m) != 0) | ||
74 | bb_perror_nomsg_and_die(); | ||
75 | return EXIT_SUCCESS; | ||
76 | } | ||
diff --git a/util-linux/renice.c b/util-linux/renice.c new file mode 100644 index 000000000..64213c680 --- /dev/null +++ b/util-linux/renice.c | |||
@@ -0,0 +1,148 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * renice implementation for busybox | ||
4 | * | ||
5 | * Copyright (C) 2005 Manuel Novoa III <mjn3@codepoet.org> | ||
6 | * | ||
7 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. | ||
8 | */ | ||
9 | |||
10 | /* Notes: | ||
11 | * Setting an absolute priority was obsoleted in SUSv2 and removed | ||
12 | * in SUSv3. However, the common linux version of renice does | ||
13 | * absolute and not relative. So we'll continue supporting absolute, | ||
14 | * although the stdout logging has been removed since both SUSv2 and | ||
15 | * SUSv3 specify that stdout isn't used. | ||
16 | * | ||
17 | * This version is lenient in that it doesn't require any IDs. The | ||
18 | * options -p, -g, and -u are treated as mode switches for the | ||
19 | * following IDs (if any). Multiple switches are allowed. | ||
20 | */ | ||
21 | //config:config RENICE | ||
22 | //config: bool "renice" | ||
23 | //config: default y | ||
24 | //config: help | ||
25 | //config: Renice alters the scheduling priority of one or more running | ||
26 | //config: processes. | ||
27 | |||
28 | //applet:IF_RENICE(APPLET(renice, BB_DIR_USR_BIN, BB_SUID_DROP)) | ||
29 | |||
30 | //kbuild:lib-$(CONFIG_RENICE) += renice.o | ||
31 | |||
32 | //usage:#define renice_trivial_usage | ||
33 | //usage: "[-n] PRIORITY [[-p | -g | -u] ID...]..." | ||
34 | //usage:#define renice_full_usage "\n\n" | ||
35 | //usage: "Change scheduling priority of a running process\n" | ||
36 | //usage: "\n -n Add PRIORITY to current nice value" | ||
37 | //usage: "\n Without -n, nice value is set to PRIORITY" | ||
38 | //usage: "\n -p Process ids (default)" | ||
39 | //usage: "\n -g Process group ids" | ||
40 | //usage: "\n -u Process user names" | ||
41 | |||
42 | #include "libbb.h" | ||
43 | #include <sys/resource.h> | ||
44 | |||
45 | void BUG_bad_PRIO_PROCESS(void); | ||
46 | void BUG_bad_PRIO_PGRP(void); | ||
47 | void BUG_bad_PRIO_USER(void); | ||
48 | |||
49 | int renice_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
50 | int renice_main(int argc UNUSED_PARAM, char **argv) | ||
51 | { | ||
52 | static const char Xetpriority_msg[] ALIGN1 = "%cetpriority"; | ||
53 | |||
54 | int retval = EXIT_SUCCESS; | ||
55 | int which = PRIO_PROCESS; /* Default 'which' value. */ | ||
56 | int use_relative = 0; | ||
57 | int adjustment, new_priority; | ||
58 | unsigned who; | ||
59 | char *arg; | ||
60 | |||
61 | /* Yes, they are not #defines in glibc 2.4! #if won't work */ | ||
62 | if (PRIO_PROCESS < CHAR_MIN || PRIO_PROCESS > CHAR_MAX) | ||
63 | BUG_bad_PRIO_PROCESS(); | ||
64 | if (PRIO_PGRP < CHAR_MIN || PRIO_PGRP > CHAR_MAX) | ||
65 | BUG_bad_PRIO_PGRP(); | ||
66 | if (PRIO_USER < CHAR_MIN || PRIO_USER > CHAR_MAX) | ||
67 | BUG_bad_PRIO_USER(); | ||
68 | |||
69 | arg = *++argv; | ||
70 | |||
71 | /* Check if we are using a relative adjustment. */ | ||
72 | if (arg && arg[0] == '-' && arg[1] == 'n') { | ||
73 | use_relative = 1; | ||
74 | if (!arg[2]) | ||
75 | arg = *++argv; | ||
76 | else | ||
77 | arg += 2; | ||
78 | } | ||
79 | |||
80 | if (!arg) { /* No args? Then show usage. */ | ||
81 | bb_show_usage(); | ||
82 | } | ||
83 | |||
84 | /* Get the priority adjustment (absolute or relative). */ | ||
85 | adjustment = xatoi_range(arg, INT_MIN/2, INT_MAX/2); | ||
86 | |||
87 | while ((arg = *++argv) != NULL) { | ||
88 | /* Check for a mode switch. */ | ||
89 | if (arg[0] == '-' && arg[1]) { | ||
90 | static const char opts[] ALIGN1 = { | ||
91 | 'p', 'g', 'u', 0, PRIO_PROCESS, PRIO_PGRP, PRIO_USER | ||
92 | }; | ||
93 | const char *p = strchr(opts, arg[1]); | ||
94 | if (p) { | ||
95 | which = p[4]; | ||
96 | if (!arg[2]) | ||
97 | continue; | ||
98 | arg += 2; | ||
99 | } | ||
100 | } | ||
101 | |||
102 | /* Process an ID arg. */ | ||
103 | if (which == PRIO_USER) { | ||
104 | struct passwd *p; | ||
105 | p = getpwnam(arg); | ||
106 | if (!p) { | ||
107 | bb_error_msg("unknown user %s", arg); | ||
108 | goto HAD_ERROR; | ||
109 | } | ||
110 | who = p->pw_uid; | ||
111 | } else { | ||
112 | who = bb_strtou(arg, NULL, 10); | ||
113 | if (errno) { | ||
114 | bb_error_msg("invalid number '%s'", arg); | ||
115 | goto HAD_ERROR; | ||
116 | } | ||
117 | } | ||
118 | |||
119 | /* Get priority to use, and set it. */ | ||
120 | if (use_relative) { | ||
121 | int old_priority; | ||
122 | |||
123 | errno = 0; /* Needed for getpriority error detection. */ | ||
124 | old_priority = getpriority(which, who); | ||
125 | if (errno) { | ||
126 | bb_perror_msg(Xetpriority_msg, 'g'); | ||
127 | goto HAD_ERROR; | ||
128 | } | ||
129 | |||
130 | new_priority = old_priority + adjustment; | ||
131 | } else { | ||
132 | new_priority = adjustment; | ||
133 | } | ||
134 | |||
135 | if (setpriority(which, who, new_priority) == 0) { | ||
136 | continue; | ||
137 | } | ||
138 | |||
139 | bb_perror_msg(Xetpriority_msg, 's'); | ||
140 | HAD_ERROR: | ||
141 | retval = EXIT_FAILURE; | ||
142 | } | ||
143 | |||
144 | /* No need to check for errors outputing to stderr since, if it | ||
145 | * was used, the HAD_ERROR label was reached and retval was set. */ | ||
146 | |||
147 | return retval; | ||
148 | } | ||