1/* $NetBSD: t_sched.c,v 1.1 2011/03/25 09:34:02 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_sched.c,v 1.1 2011/03/25 09:34:02 jruoho Exp $"); 33 34#include <sched.h> 35#include <limits.h> 36#include <unistd.h> 37 38#include <atf-c.h> 39 40static void sched_priority_set(int, int); 41 42ATF_TC(sched_getparam); 43ATF_TC_HEAD(sched_getparam, tc) 44{ 45 atf_tc_set_md_var(tc, "descr", "A basic test of sched_getparam(3)"); 46} 47 48ATF_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_REQUIRE(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 69ATF_TC(sched_priority); 70ATF_TC_HEAD(sched_priority, tc) 71{ 72 atf_tc_set_md_var(tc, "descr", "Test sched(3) priority ranges"); 73} 74 75ATF_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 104static void 105sched_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_REQUIRE(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_REQUIRE(sched.sched_priority == pri); 126} 127 128ATF_TC(sched_setscheduler_1); 129ATF_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 135ATF_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 143ATF_TC(sched_setscheduler_2); 144ATF_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 150ATF_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 158ATF_TC(sched_setscheduler_3); 159ATF_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 165ATF_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 173ATF_TC(sched_setscheduler_4); 174ATF_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 180ATF_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 188ATF_TC(sched_rr_get_interval_1); 189ATF_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 atf_tc_set_md_var(tc, "require.user", "root"); 193} 194 195ATF_TC_BODY(sched_rr_get_interval_1, tc) 196{ 197 struct timespec tv; 198 int pri; 199 200 pri = sched_get_priority_min(SCHED_RR); 201 sched_priority_set(pri, SCHED_RR); 202 203 /* 204 * This should fail with ESRCH for invalid PID. 205 */ 206 atf_tc_expect_fail("PR lib/44768"); 207 ATF_REQUIRE(sched_rr_get_interval(-INT_MAX, &tv) != 0); 208} 209 210ATF_TC(sched_rr_get_interval_2); 211ATF_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 217ATF_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 241ATF_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