1 /* $NetBSD: t_sched.c,v 1.5 2012/03/25 04:11:42 christos 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_sched.c,v 1.5 2012/03/25 04:11:42 christos Exp $");
33
34 #include <sched.h>
35 #include <limits.h>
36 #include <unistd.h>
37
38 #include <atf-c.h>
39
40 static void sched_priority_set(int, int);
41
42 ATF_TC(sched_getparam);
ATF_TC_HEAD(sched_getparam,tc)43 ATF_TC_HEAD(sched_getparam, tc)
44 {
45 atf_tc_set_md_var(tc, "descr", "A basic test of sched_getparam(3)");
46 }
47
ATF_TC_BODY(sched_getparam,tc)48 ATF_TC_BODY(sched_getparam, tc)
49 {
50 struct sched_param s1, s2;
51 pid_t p = getpid();
52
53 /*
54 * IEEE Std 1003.1-2008: if the supplied pid is zero,
55 * the parameters for the calling process are returned.
56 */
57 ATF_REQUIRE(sched_getparam(0, &s1) == 0);
58 ATF_REQUIRE(sched_getparam(p, &s2) == 0);
59
60 ATF_CHECK_EQ(s1.sched_priority, s2.sched_priority);
61
62 /*
63 * The behavior is undefined but should error
64 * out in case the supplied PID is negative.
65 */
66 ATF_REQUIRE(sched_getparam(-1, &s1) != 0);
67 }
68
69 ATF_TC(sched_priority);
ATF_TC_HEAD(sched_priority,tc)70 ATF_TC_HEAD(sched_priority, tc)
71 {
72 atf_tc_set_md_var(tc, "descr", "Test sched(3) priority ranges");
73 }
74
ATF_TC_BODY(sched_priority,tc)75 ATF_TC_BODY(sched_priority, tc)
76 {
77 static const int pol[3] = { SCHED_OTHER, SCHED_FIFO, SCHED_RR };
78 int pmax, pmin;
79 size_t i;
80
81 /*
82 * Test that bogus values error out.
83 */
84 if (INT_MAX > SCHED_RR)
85 ATF_REQUIRE(sched_get_priority_max(INT_MAX) != 0);
86
87 if (-INT_MAX < SCHED_OTHER)
88 ATF_REQUIRE(sched_get_priority_max(-INT_MAX) != 0);
89
90 /*
91 * Test that we have a valid range.
92 */
93 for (i = 0; i < __arraycount(pol); i++) {
94
95 pmax = sched_get_priority_max(pol[i]);
96 pmin = sched_get_priority_min(pol[i]);
97
98 ATF_REQUIRE(pmax != -1);
99 ATF_REQUIRE(pmin != -1);
100 ATF_REQUIRE(pmax > pmin);
101 }
102 }
103
104 static void
sched_priority_set(int pri,int pol)105 sched_priority_set(int pri, int pol)
106 {
107 struct sched_param sched;
108
109 sched.sched_priority = pri;
110
111 ATF_REQUIRE(pri >= 0);
112 ATF_REQUIRE(sched_setscheduler(0, pol, &sched) == 0);
113
114 /*
115 * Test that the policy was changed.
116 */
117 ATF_CHECK_EQ(sched_getscheduler(0), pol);
118
119 /*
120 * And that sched_getparam(3) returns the new priority.
121 */
122 sched.sched_priority = -1;
123
124 ATF_REQUIRE(sched_getparam(0, &sched) == 0);
125 ATF_CHECK_EQ(sched.sched_priority, pri);
126 }
127
128 ATF_TC(sched_setscheduler_1);
ATF_TC_HEAD(sched_setscheduler_1,tc)129 ATF_TC_HEAD(sched_setscheduler_1, tc)
130 {
131 atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), max, RR");
132 atf_tc_set_md_var(tc, "require.user", "root");
133 }
134
ATF_TC_BODY(sched_setscheduler_1,tc)135 ATF_TC_BODY(sched_setscheduler_1, tc)
136 {
137 int pri;
138
139 pri = sched_get_priority_max(SCHED_RR);
140 sched_priority_set(pri, SCHED_RR);
141 }
142
143 ATF_TC(sched_setscheduler_2);
ATF_TC_HEAD(sched_setscheduler_2,tc)144 ATF_TC_HEAD(sched_setscheduler_2, tc)
145 {
146 atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), min, RR");
147 atf_tc_set_md_var(tc, "require.user", "root");
148 }
149
ATF_TC_BODY(sched_setscheduler_2,tc)150 ATF_TC_BODY(sched_setscheduler_2, tc)
151 {
152 int pri;
153
154 pri = sched_get_priority_min(SCHED_RR);
155 sched_priority_set(pri, SCHED_RR);
156 }
157
158 ATF_TC(sched_setscheduler_3);
ATF_TC_HEAD(sched_setscheduler_3,tc)159 ATF_TC_HEAD(sched_setscheduler_3, tc)
160 {
161 atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), max, FIFO");
162 atf_tc_set_md_var(tc, "require.user", "root");
163 }
164
ATF_TC_BODY(sched_setscheduler_3,tc)165 ATF_TC_BODY(sched_setscheduler_3, tc)
166 {
167 int pri;
168
169 pri = sched_get_priority_max(SCHED_FIFO);
170 sched_priority_set(pri, SCHED_FIFO);
171 }
172
173 ATF_TC(sched_setscheduler_4);
ATF_TC_HEAD(sched_setscheduler_4,tc)174 ATF_TC_HEAD(sched_setscheduler_4, tc)
175 {
176 atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), min, FIFO");
177 atf_tc_set_md_var(tc, "require.user", "root");
178 }
179
ATF_TC_BODY(sched_setscheduler_4,tc)180 ATF_TC_BODY(sched_setscheduler_4, tc)
181 {
182 int pri;
183
184 pri = sched_get_priority_min(SCHED_FIFO);
185 sched_priority_set(pri, SCHED_FIFO);
186 }
187
188 ATF_TC(sched_rr_get_interval_1);
ATF_TC_HEAD(sched_rr_get_interval_1,tc)189 ATF_TC_HEAD(sched_rr_get_interval_1, tc)
190 {
191 atf_tc_set_md_var(tc, "descr", "Test sched_rr_get_interval(3), #1"
192 " (PR lib/44768)");
193 atf_tc_set_md_var(tc, "require.user", "root");
194 }
195
ATF_TC_BODY(sched_rr_get_interval_1,tc)196 ATF_TC_BODY(sched_rr_get_interval_1, tc)
197 {
198 struct timespec tv;
199 int pri;
200
201 pri = sched_get_priority_min(SCHED_RR);
202 sched_priority_set(pri, SCHED_RR);
203
204 /*
205 * This should fail with ESRCH for invalid PID.
206 */
207 ATF_REQUIRE(sched_rr_get_interval(-INT_MAX, &tv) != 0);
208 }
209
210 ATF_TC(sched_rr_get_interval_2);
ATF_TC_HEAD(sched_rr_get_interval_2,tc)211 ATF_TC_HEAD(sched_rr_get_interval_2, tc)
212 {
213 atf_tc_set_md_var(tc, "descr", "Test sched_rr_get_interval(3), #2");
214 atf_tc_set_md_var(tc, "require.user", "root");
215 }
216
ATF_TC_BODY(sched_rr_get_interval_2,tc)217 ATF_TC_BODY(sched_rr_get_interval_2, tc)
218 {
219 struct timespec tv1, tv2;
220 int pri;
221
222 pri = sched_get_priority_min(SCHED_RR);
223 sched_priority_set(pri, SCHED_RR);
224
225 tv1.tv_sec = tv2.tv_sec = -1;
226 tv1.tv_nsec = tv2.tv_nsec = -1;
227
228 ATF_REQUIRE(sched_rr_get_interval(0, &tv1) == 0);
229 ATF_REQUIRE(sched_rr_get_interval(getpid(), &tv2) == 0);
230
231 ATF_REQUIRE(tv1.tv_sec != -1);
232 ATF_REQUIRE(tv2.tv_sec != -1);
233
234 ATF_REQUIRE(tv1.tv_nsec != -1);
235 ATF_REQUIRE(tv2.tv_nsec != -1);
236
237 ATF_REQUIRE(tv1.tv_sec == tv2.tv_sec);
238 ATF_REQUIRE(tv1.tv_nsec == tv2.tv_nsec);
239 }
240
ATF_TP_ADD_TCS(tp)241 ATF_TP_ADD_TCS(tp)
242 {
243
244 ATF_TP_ADD_TC(tp, sched_getparam);
245 ATF_TP_ADD_TC(tp, sched_priority);
246
247 ATF_TP_ADD_TC(tp, sched_setscheduler_1);
248 ATF_TP_ADD_TC(tp, sched_setscheduler_2);
249 ATF_TP_ADD_TC(tp, sched_setscheduler_3);
250 ATF_TP_ADD_TC(tp, sched_setscheduler_4);
251
252 ATF_TP_ADD_TC(tp, sched_rr_get_interval_1);
253 ATF_TP_ADD_TC(tp, sched_rr_get_interval_2);
254
255 return atf_no_error();
256 }
257