tests/lib/librt/t_sched.c

   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);
  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_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);
  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_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);
 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             " (PR lib/44768)");
 193         atf_tc_set_md_var(tc, "require.user", "root");
 194 }
 195
 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);
 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 }