1272343Sngie/* $NetBSD: t_sched.c,v 1.5 2012/03/25 04:11:42 christos Exp $ */ 2272343Sngie 3272343Sngie/*- 4272343Sngie * Copyright (c) 2011 The NetBSD Foundation, Inc. 5272343Sngie * All rights reserved. 6272343Sngie * 7272343Sngie * This code is derived from software contributed to The NetBSD Foundation 8272343Sngie * by Jukka Ruohonen. 9272343Sngie * 10272343Sngie * Redistribution and use in source and binary forms, with or without 11272343Sngie * modification, are permitted provided that the following conditions 12272343Sngie * are met: 13272343Sngie * 1. Redistributions of source code must retain the above copyright 14272343Sngie * notice, this list of conditions and the following disclaimer. 15272343Sngie * 2. Redistributions in binary form must reproduce the above copyright 16272343Sngie * notice, this list of conditions and the following disclaimer in the 17272343Sngie * documentation and/or other materials provided with the distribution. 18272343Sngie * 19272343Sngie * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20272343Sngie * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21272343Sngie * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22272343Sngie * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23272343Sngie * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24272343Sngie * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25272343Sngie * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26272343Sngie * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27272343Sngie * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28272343Sngie * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29272343Sngie * POSSIBILITY OF SUCH DAMAGE. 30272343Sngie */ 31272343Sngie#include <sys/cdefs.h> 32272343Sngie__RCSID("$NetBSD: t_sched.c,v 1.5 2012/03/25 04:11:42 christos Exp $"); 33272343Sngie 34272343Sngie#include <sched.h> 35272343Sngie#include <limits.h> 36272343Sngie#include <unistd.h> 37272343Sngie 38272343Sngie#include <atf-c.h> 39272343Sngie 40272343Sngiestatic void sched_priority_set(int, int); 41272343Sngie 42272343SngieATF_TC(sched_getparam); 43272343SngieATF_TC_HEAD(sched_getparam, tc) 44272343Sngie{ 45272343Sngie atf_tc_set_md_var(tc, "descr", "A basic test of sched_getparam(3)"); 46272343Sngie} 47272343Sngie 48272343SngieATF_TC_BODY(sched_getparam, tc) 49272343Sngie{ 50272343Sngie struct sched_param s1, s2; 51272343Sngie pid_t p = getpid(); 52272343Sngie 53272343Sngie /* 54272343Sngie * IEEE Std 1003.1-2008: if the supplied pid is zero, 55272343Sngie * the parameters for the calling process are returned. 56272343Sngie */ 57272343Sngie ATF_REQUIRE(sched_getparam(0, &s1) == 0); 58272343Sngie ATF_REQUIRE(sched_getparam(p, &s2) == 0); 59272343Sngie 60272343Sngie ATF_CHECK_EQ(s1.sched_priority, s2.sched_priority); 61272343Sngie 62272343Sngie /* 63272343Sngie * The behavior is undefined but should error 64272343Sngie * out in case the supplied PID is negative. 65272343Sngie */ 66272343Sngie ATF_REQUIRE(sched_getparam(-1, &s1) != 0); 67272343Sngie} 68272343Sngie 69272343SngieATF_TC(sched_priority); 70272343SngieATF_TC_HEAD(sched_priority, tc) 71272343Sngie{ 72272343Sngie atf_tc_set_md_var(tc, "descr", "Test sched(3) priority ranges"); 73272343Sngie} 74272343Sngie 75272343SngieATF_TC_BODY(sched_priority, tc) 76272343Sngie{ 77272343Sngie static const int pol[3] = { SCHED_OTHER, SCHED_FIFO, SCHED_RR }; 78272343Sngie int pmax, pmin; 79272343Sngie size_t i; 80272343Sngie 81272343Sngie /* 82272343Sngie * Test that bogus values error out. 83272343Sngie */ 84272343Sngie if (INT_MAX > SCHED_RR) 85272343Sngie ATF_REQUIRE(sched_get_priority_max(INT_MAX) != 0); 86272343Sngie 87272343Sngie if (-INT_MAX < SCHED_OTHER) 88272343Sngie ATF_REQUIRE(sched_get_priority_max(-INT_MAX) != 0); 89272343Sngie 90272343Sngie /* 91272343Sngie * Test that we have a valid range. 92272343Sngie */ 93272343Sngie for (i = 0; i < __arraycount(pol); i++) { 94272343Sngie 95272343Sngie pmax = sched_get_priority_max(pol[i]); 96272343Sngie pmin = sched_get_priority_min(pol[i]); 97272343Sngie 98272343Sngie ATF_REQUIRE(pmax != -1); 99272343Sngie ATF_REQUIRE(pmin != -1); 100272343Sngie ATF_REQUIRE(pmax > pmin); 101272343Sngie } 102272343Sngie} 103272343Sngie 104272343Sngiestatic void 105272343Sngiesched_priority_set(int pri, int pol) 106272343Sngie{ 107272343Sngie struct sched_param sched; 108272343Sngie 109272343Sngie sched.sched_priority = pri; 110272343Sngie 111272343Sngie ATF_REQUIRE(pri >= 0); 112272343Sngie ATF_REQUIRE(sched_setscheduler(0, pol, &sched) == 0); 113272343Sngie 114272343Sngie /* 115272343Sngie * Test that the policy was changed. 116272343Sngie */ 117272343Sngie ATF_CHECK_EQ(sched_getscheduler(0), pol); 118272343Sngie 119272343Sngie /* 120272343Sngie * And that sched_getparam(3) returns the new priority. 121272343Sngie */ 122272343Sngie sched.sched_priority = -1; 123272343Sngie 124272343Sngie ATF_REQUIRE(sched_getparam(0, &sched) == 0); 125272343Sngie ATF_CHECK_EQ(sched.sched_priority, pri); 126272343Sngie} 127272343Sngie 128272343SngieATF_TC(sched_setscheduler_1); 129272343SngieATF_TC_HEAD(sched_setscheduler_1, tc) 130272343Sngie{ 131272343Sngie atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), max, RR"); 132272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 133272343Sngie} 134272343Sngie 135272343SngieATF_TC_BODY(sched_setscheduler_1, tc) 136272343Sngie{ 137272343Sngie int pri; 138272343Sngie 139272343Sngie pri = sched_get_priority_max(SCHED_RR); 140272343Sngie sched_priority_set(pri, SCHED_RR); 141272343Sngie} 142272343Sngie 143272343SngieATF_TC(sched_setscheduler_2); 144272343SngieATF_TC_HEAD(sched_setscheduler_2, tc) 145272343Sngie{ 146272343Sngie atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), min, RR"); 147272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 148272343Sngie} 149272343Sngie 150272343SngieATF_TC_BODY(sched_setscheduler_2, tc) 151272343Sngie{ 152272343Sngie int pri; 153272343Sngie 154272343Sngie pri = sched_get_priority_min(SCHED_RR); 155272343Sngie sched_priority_set(pri, SCHED_RR); 156272343Sngie} 157272343Sngie 158272343SngieATF_TC(sched_setscheduler_3); 159272343SngieATF_TC_HEAD(sched_setscheduler_3, tc) 160272343Sngie{ 161272343Sngie atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), max, FIFO"); 162272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 163272343Sngie} 164272343Sngie 165272343SngieATF_TC_BODY(sched_setscheduler_3, tc) 166272343Sngie{ 167272343Sngie int pri; 168272343Sngie 169272343Sngie pri = sched_get_priority_max(SCHED_FIFO); 170272343Sngie sched_priority_set(pri, SCHED_FIFO); 171272343Sngie} 172272343Sngie 173272343SngieATF_TC(sched_setscheduler_4); 174272343SngieATF_TC_HEAD(sched_setscheduler_4, tc) 175272343Sngie{ 176272343Sngie atf_tc_set_md_var(tc, "descr", "sched_setscheduler(3), min, FIFO"); 177272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 178272343Sngie} 179272343Sngie 180272343SngieATF_TC_BODY(sched_setscheduler_4, tc) 181272343Sngie{ 182272343Sngie int pri; 183272343Sngie 184272343Sngie pri = sched_get_priority_min(SCHED_FIFO); 185272343Sngie sched_priority_set(pri, SCHED_FIFO); 186272343Sngie} 187272343Sngie 188272343SngieATF_TC(sched_rr_get_interval_1); 189272343SngieATF_TC_HEAD(sched_rr_get_interval_1, tc) 190272343Sngie{ 191272343Sngie atf_tc_set_md_var(tc, "descr", "Test sched_rr_get_interval(3), #1" 192272343Sngie " (PR lib/44768)"); 193272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 194272343Sngie} 195272343Sngie 196272343SngieATF_TC_BODY(sched_rr_get_interval_1, tc) 197272343Sngie{ 198272343Sngie struct timespec tv; 199272343Sngie int pri; 200272343Sngie 201272343Sngie pri = sched_get_priority_min(SCHED_RR); 202272343Sngie sched_priority_set(pri, SCHED_RR); 203272343Sngie 204272343Sngie /* 205272343Sngie * This should fail with ESRCH for invalid PID. 206272343Sngie */ 207272343Sngie ATF_REQUIRE(sched_rr_get_interval(-INT_MAX, &tv) != 0); 208272343Sngie} 209272343Sngie 210272343SngieATF_TC(sched_rr_get_interval_2); 211272343SngieATF_TC_HEAD(sched_rr_get_interval_2, tc) 212272343Sngie{ 213272343Sngie atf_tc_set_md_var(tc, "descr", "Test sched_rr_get_interval(3), #2"); 214272343Sngie atf_tc_set_md_var(tc, "require.user", "root"); 215272343Sngie} 216272343Sngie 217272343SngieATF_TC_BODY(sched_rr_get_interval_2, tc) 218272343Sngie{ 219272343Sngie struct timespec tv1, tv2; 220272343Sngie int pri; 221272343Sngie 222272343Sngie pri = sched_get_priority_min(SCHED_RR); 223272343Sngie sched_priority_set(pri, SCHED_RR); 224272343Sngie 225272343Sngie tv1.tv_sec = tv2.tv_sec = -1; 226272343Sngie tv1.tv_nsec = tv2.tv_nsec = -1; 227272343Sngie 228272343Sngie ATF_REQUIRE(sched_rr_get_interval(0, &tv1) == 0); 229272343Sngie ATF_REQUIRE(sched_rr_get_interval(getpid(), &tv2) == 0); 230272343Sngie 231272343Sngie ATF_REQUIRE(tv1.tv_sec != -1); 232272343Sngie ATF_REQUIRE(tv2.tv_sec != -1); 233272343Sngie 234272343Sngie ATF_REQUIRE(tv1.tv_nsec != -1); 235272343Sngie ATF_REQUIRE(tv2.tv_nsec != -1); 236272343Sngie 237272343Sngie ATF_REQUIRE(tv1.tv_sec == tv2.tv_sec); 238272343Sngie ATF_REQUIRE(tv1.tv_nsec == tv2.tv_nsec); 239272343Sngie} 240272343Sngie 241272343SngieATF_TP_ADD_TCS(tp) 242272343Sngie{ 243272343Sngie 244272343Sngie ATF_TP_ADD_TC(tp, sched_getparam); 245272343Sngie ATF_TP_ADD_TC(tp, sched_priority); 246272343Sngie 247272343Sngie ATF_TP_ADD_TC(tp, sched_setscheduler_1); 248272343Sngie ATF_TP_ADD_TC(tp, sched_setscheduler_2); 249272343Sngie ATF_TP_ADD_TC(tp, sched_setscheduler_3); 250272343Sngie ATF_TP_ADD_TC(tp, sched_setscheduler_4); 251272343Sngie 252272343Sngie ATF_TP_ADD_TC(tp, sched_rr_get_interval_1); 253272343Sngie ATF_TP_ADD_TC(tp, sched_rr_get_interval_2); 254272343Sngie 255272343Sngie return atf_no_error(); 256272343Sngie} 257