1 /* $NetBSD: t_nice.c,v 1.8 2012/03/18 07:00:51 jruoho Exp $ */
2
3 /*-
4 * Copyright (c) 2011 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Jukka Ruohonen.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: t_nice.c,v 1.8 2012/03/18 07:00:51 jruoho Exp $");
33
34 #include <sys/resource.h>
35 #include <sys/wait.h>
36
37 #include <atf-c.h>
38 #include <errno.h>
39 #include <limits.h>
40 #include <pthread.h>
41 #include <stdlib.h>
42 #include <unistd.h>
43
44 static void *threadfunc(void *);
45
46 static void *
threadfunc(void * arg)47 threadfunc(void *arg)
48 {
49 int pri, val;
50
51 val = *(int *)arg;
52
53 errno = 0;
54 pri = getpriority(PRIO_PROCESS, 0);
55 ATF_REQUIRE(errno == 0);
56
57 if (pri != val)
58 atf_tc_fail("nice(3) value was not propagated to threads");
59
60 return NULL;
61 }
62
63 ATF_TC(nice_err);
ATF_TC_HEAD(nice_err,tc)64 ATF_TC_HEAD(nice_err, tc)
65 {
66 atf_tc_set_md_var(tc, "descr",
67 "Test nice(3) for invalid parameters (PR lib/42587)");
68 atf_tc_set_md_var(tc, "require.user", "unprivileged");
69 }
70
ATF_TC_BODY(nice_err,tc)71 ATF_TC_BODY(nice_err, tc)
72 {
73 int i;
74
75 /*
76 * The call should fail with EPERM if the
77 * supplied parameter is negative and the
78 * caller does not have privileges.
79 */
80 for (i = -20; i < 0; i++) {
81
82 errno = 0;
83
84 ATF_REQUIRE_ERRNO(EPERM, nice(i) == -1);
85 }
86 }
87
88 ATF_TC(nice_priority);
ATF_TC_HEAD(nice_priority,tc)89 ATF_TC_HEAD(nice_priority, tc)
90 {
91 atf_tc_set_md_var(tc, "descr", "Test nice(3) vs. getpriority(2)");
92 }
93
ATF_TC_BODY(nice_priority,tc)94 ATF_TC_BODY(nice_priority, tc)
95 {
96 #ifdef __FreeBSD__
97 int i, pri, pri2, nic;
98 #else
99 int i, pri, nic;
100 #endif
101 pid_t pid;
102 int sta;
103
104 for (i = 0; i <= 20; i++) {
105
106 nic = nice(i);
107 ATF_REQUIRE(nic != -1);
108
109 errno = 0;
110 pri = getpriority(PRIO_PROCESS, 0);
111 ATF_REQUIRE(errno == 0);
112
113 #ifdef __NetBSD__
114 if (nic != pri)
115 atf_tc_fail("nice(3) and getpriority(2) conflict");
116 #endif
117
118 /*
119 * Also verify that the nice(3) values
120 * are inherited by child processes.
121 */
122 pid = fork();
123 ATF_REQUIRE(pid >= 0);
124
125 if (pid == 0) {
126
127 errno = 0;
128 #ifdef __NetBSD__
129 pri = getpriority(PRIO_PROCESS, 0);
130 #else
131 pri2 = getpriority(PRIO_PROCESS, 0);
132 #endif
133 ATF_REQUIRE(errno == 0);
134
135 #ifdef __FreeBSD__
136 if (pri != pri2)
137 #else
138 if (nic != pri)
139 #endif
140 _exit(EXIT_FAILURE);
141
142 _exit(EXIT_SUCCESS);
143 }
144
145 (void)wait(&sta);
146
147 if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
148 atf_tc_fail("nice(3) value was not inherited");
149 }
150 }
151
152 ATF_TC(nice_root);
ATF_TC_HEAD(nice_root,tc)153 ATF_TC_HEAD(nice_root, tc)
154 {
155 atf_tc_set_md_var(tc, "descr", "Test that nice(3) works");
156 atf_tc_set_md_var(tc, "require.user", "root");
157 }
158
ATF_TC_BODY(nice_root,tc)159 ATF_TC_BODY(nice_root, tc)
160 {
161 int i;
162
163 for (i = -20; i <= 20; i++) {
164
165 ATF_REQUIRE(nice(i) != -1);
166 }
167 }
168
169 ATF_TC(nice_thread);
ATF_TC_HEAD(nice_thread,tc)170 ATF_TC_HEAD(nice_thread, tc)
171 {
172 atf_tc_set_md_var(tc, "descr", "Test nice(3) with threads");
173 }
174
ATF_TC_BODY(nice_thread,tc)175 ATF_TC_BODY(nice_thread, tc)
176 {
177 pthread_t tid[5];
178 #ifdef __FreeBSD__
179 int pri, rv, val;
180 #else
181 int rv, val;
182 #endif
183 size_t i;
184
185 /*
186 * Test that the scheduling priority is
187 * propagated to all system scope threads.
188 */
189 for (i = 0; i < __arraycount(tid); i++) {
190
191 val = nice(i);
192 ATF_REQUIRE(val != -1);
193
194 #ifdef __FreeBSD__
195 pri = getpriority(PRIO_PROCESS, 0);
196 rv = pthread_create(&tid[i], NULL, threadfunc, &pri);
197 #else
198 rv = pthread_create(&tid[i], NULL, threadfunc, &val);
199 #endif
200 ATF_REQUIRE(rv == 0);
201
202 rv = pthread_join(tid[i], NULL);
203 ATF_REQUIRE(rv == 0);
204 }
205 }
206
ATF_TP_ADD_TCS(tp)207 ATF_TP_ADD_TCS(tp)
208 {
209
210 ATF_TP_ADD_TC(tp, nice_err);
211 ATF_TP_ADD_TC(tp, nice_priority);
212 ATF_TP_ADD_TC(tp, nice_root);
213 ATF_TP_ADD_TC(tp, nice_thread);
214
215 return atf_no_error();
216 }
217