contrib/netbsd-tests/lib/libc/gen/t_nice.c

   1 /*      $NetBSD: t_nice.c,v 1.8 2012/03/18 07:00:51 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_nice.c,v 1.8 2012/03/18 07:00:51 jruoho Exp $");
  33
  34 #include <sys/resource.h>
  35 #include <sys/wait.h>
  36
  37 #include <atf-c.h>
  38 #include <errno.h>
  39 #include <limits.h>
  40 #include <pthread.h>
  41 #include <stdlib.h>
  42 #include <unistd.h>
  43
  44 static void     *threadfunc(void *);
  45
  46 static void *
  47 threadfunc(void *arg)
  48 {
  49         int pri, val;
  50
  51         val = *(int *)arg;
  52
  53         errno = 0;
  54         pri = getpriority(PRIO_PROCESS, 0);
  55         ATF_REQUIRE(errno == 0);
  56
  57         if (pri != val)
  58                 atf_tc_fail("nice(3) value was not propagated to threads");
  59
  60         return NULL;
  61 }
  62
  63 ATF_TC(nice_err);
  64 ATF_TC_HEAD(nice_err, tc)
  65 {
  66         atf_tc_set_md_var(tc, "descr",
  67             "Test nice(3) for invalid parameters (PR lib/42587)");
  68         atf_tc_set_md_var(tc, "require.user", "unprivileged");
  69 }
  70
  71 ATF_TC_BODY(nice_err, tc)
  72 {
  73         int i;
  74
  75         /*
  76          * The call should fail with EPERM if the
  77          * supplied parameter is negative and the
  78          * caller does not have privileges.
  79          */
  80         for (i = -20; i < 0; i++) {
  81
  82                 errno = 0;
  83
  84                 ATF_REQUIRE_ERRNO(EPERM, nice(i) == -1);
  85         }
  86 }
  87
  88 ATF_TC(nice_priority);
  89 ATF_TC_HEAD(nice_priority, tc)
  90 {
  91         atf_tc_set_md_var(tc, "descr", "Test nice(3) vs. getpriority(2)");
  92 }
  93
  94 ATF_TC_BODY(nice_priority, tc)
  95 {
  96 #ifdef __FreeBSD__
  97         int i, pri, pri2, nic;
  98 #else
  99         int i, pri, nic;
 100 #endif
 101         pid_t pid;
 102         int sta;
 103
 104         for (i = 0; i <= 20; i++) {
 105
 106                 nic = nice(i);
 107                 ATF_REQUIRE(nic != -1);
 108
 109                 errno = 0;
 110                 pri = getpriority(PRIO_PROCESS, 0);
 111                 ATF_REQUIRE(errno == 0);
 112
 113 #ifdef __NetBSD__
 114                 if (nic != pri)
 115                         atf_tc_fail("nice(3) and getpriority(2) conflict");
 116 #endif
 117
 118                 /*
 119                  * Also verify that the nice(3) values
 120                  * are inherited by child processes.
 121                  */
 122                 pid = fork();
 123                 ATF_REQUIRE(pid >= 0);
 124
 125                 if (pid == 0) {
 126
 127                         errno = 0;
 128 #ifdef __NetBSD__
 129                         pri = getpriority(PRIO_PROCESS, 0);
 130 #else
 131                         pri2 = getpriority(PRIO_PROCESS, 0);
 132 #endif
 133                         ATF_REQUIRE(errno == 0);
 134
 135 #ifdef __FreeBSD__
 136                         if (pri != pri2)
 137 #else
 138                         if (nic != pri)
 139 #endif
 140                                 _exit(EXIT_FAILURE);
 141
 142                         _exit(EXIT_SUCCESS);
 143                 }
 144
 145                 (void)wait(&sta);
 146
 147                 if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
 148                         atf_tc_fail("nice(3) value was not inherited");
 149         }
 150 }
 151
 152 ATF_TC(nice_root);
 153 ATF_TC_HEAD(nice_root, tc)
 154 {
 155         atf_tc_set_md_var(tc, "descr", "Test that nice(3) works");
 156         atf_tc_set_md_var(tc, "require.user", "root");
 157 }
 158
 159 ATF_TC_BODY(nice_root, tc)
 160 {
 161         int i;
 162
 163         for (i = -20; i <= 20; i++) {
 164
 165                 ATF_REQUIRE(nice(i) != -1);
 166         }
 167 }
 168
 169 ATF_TC(nice_thread);
 170 ATF_TC_HEAD(nice_thread, tc)
 171 {
 172         atf_tc_set_md_var(tc, "descr", "Test nice(3) with threads");
 173 }
 174
 175 ATF_TC_BODY(nice_thread, tc)
 176 {
 177         pthread_t tid[5];
 178 #ifdef __FreeBSD__
 179         int pri, rv, val;
 180 #else
 181         int rv, val;
 182 #endif
 183         size_t i;
 184
 185         /*
 186          * Test that the scheduling priority is
 187          * propagated to all system scope threads.
 188          */
 189         for (i = 0; i < __arraycount(tid); i++) {
 190
 191                 val = nice(i);
 192                 ATF_REQUIRE(val != -1);
 193
 194 #ifdef __FreeBSD__
 195                 pri = getpriority(PRIO_PROCESS, 0);
 196                 rv = pthread_create(&tid[i], NULL, threadfunc, &pri);
 197 #else
 198                 rv = pthread_create(&tid[i], NULL, threadfunc, &val);
 199 #endif
 200                 ATF_REQUIRE(rv == 0);
 201
 202                 rv = pthread_join(tid[i], NULL);
 203                 ATF_REQUIRE(rv == 0);
 204         }
 205 }
 206
 207 ATF_TP_ADD_TCS(tp)
 208 {
 209
 210         ATF_TP_ADD_TC(tp, nice_err);
 211         ATF_TP_ADD_TC(tp, nice_priority);
 212         ATF_TP_ADD_TC(tp, nice_root);
 213         ATF_TP_ADD_TC(tp, nice_thread);
 214
 215         return atf_no_error();
 216 }