Home | History | Annotate | Line # | Download | only in librt
t_sched.c revision 1.2
      1 /* $NetBSD: t_sched.c,v 1.2 2011/03/25 09:39:19 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.2 2011/03/25 09:39:19 jruoho 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);
     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 
     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_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 
     69 ATF_TC(sched_priority);
     70 ATF_TC_HEAD(sched_priority, tc)
     71 {
     72 	atf_tc_set_md_var(tc, "descr", "Test sched(3) priority ranges");
     73 }
     74 
     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
    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_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 
    128 ATF_TC(sched_setscheduler_1);
    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 
    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);
    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 
    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);
    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 
    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);
    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 
    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);
    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 	atf_tc_set_md_var(tc, "require.user", "root");
    193 }
    194 
    195 ATF_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 
    210 ATF_TC(sched_rr_get_interval_2);
    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 
    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 
    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