1*57718be8SEnji Cooper /* $NetBSD: t_nice.c,v 1.8 2012/03/18 07:00:51 jruoho Exp $ */ 2*57718be8SEnji Cooper 3*57718be8SEnji Cooper /*- 4*57718be8SEnji Cooper * Copyright (c) 2011 The NetBSD Foundation, Inc. 5*57718be8SEnji Cooper * All rights reserved. 6*57718be8SEnji Cooper * 7*57718be8SEnji Cooper * This code is derived from software contributed to The NetBSD Foundation 8*57718be8SEnji Cooper * by Jukka Ruohonen. 9*57718be8SEnji Cooper * 10*57718be8SEnji Cooper * Redistribution and use in source and binary forms, with or without 11*57718be8SEnji Cooper * modification, are permitted provided that the following conditions 12*57718be8SEnji Cooper * are met: 13*57718be8SEnji Cooper * 1. Redistributions of source code must retain the above copyright 14*57718be8SEnji Cooper * notice, this list of conditions and the following disclaimer. 15*57718be8SEnji Cooper * 2. Redistributions in binary form must reproduce the above copyright 16*57718be8SEnji Cooper * notice, this list of conditions and the following disclaimer in the 17*57718be8SEnji Cooper * documentation and/or other materials provided with the distribution. 18*57718be8SEnji Cooper * 19*57718be8SEnji Cooper * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20*57718be8SEnji Cooper * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21*57718be8SEnji Cooper * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22*57718be8SEnji Cooper * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23*57718be8SEnji Cooper * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24*57718be8SEnji Cooper * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25*57718be8SEnji Cooper * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26*57718be8SEnji Cooper * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27*57718be8SEnji Cooper * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28*57718be8SEnji Cooper * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29*57718be8SEnji Cooper * POSSIBILITY OF SUCH DAMAGE. 30*57718be8SEnji Cooper */ 31*57718be8SEnji Cooper #include <sys/cdefs.h> 32*57718be8SEnji Cooper __RCSID("$NetBSD: t_nice.c,v 1.8 2012/03/18 07:00:51 jruoho Exp $"); 33*57718be8SEnji Cooper 34*57718be8SEnji Cooper #include <sys/resource.h> 35*57718be8SEnji Cooper #include <sys/wait.h> 36*57718be8SEnji Cooper 37*57718be8SEnji Cooper #include <atf-c.h> 38*57718be8SEnji Cooper #include <errno.h> 39*57718be8SEnji Cooper #include <limits.h> 40*57718be8SEnji Cooper #include <pthread.h> 41*57718be8SEnji Cooper #include <stdlib.h> 42*57718be8SEnji Cooper #include <unistd.h> 43*57718be8SEnji Cooper 44*57718be8SEnji Cooper static void *threadfunc(void *); 45*57718be8SEnji Cooper 46*57718be8SEnji Cooper static void * 47*57718be8SEnji Cooper threadfunc(void *arg) 48*57718be8SEnji Cooper { 49*57718be8SEnji Cooper int pri, val; 50*57718be8SEnji Cooper 51*57718be8SEnji Cooper val = *(int *)arg; 52*57718be8SEnji Cooper 53*57718be8SEnji Cooper errno = 0; 54*57718be8SEnji Cooper pri = getpriority(PRIO_PROCESS, 0); 55*57718be8SEnji Cooper ATF_REQUIRE(errno == 0); 56*57718be8SEnji Cooper 57*57718be8SEnji Cooper if (pri != val) 58*57718be8SEnji Cooper atf_tc_fail("nice(3) value was not propagated to threads"); 59*57718be8SEnji Cooper 60*57718be8SEnji Cooper return NULL; 61*57718be8SEnji Cooper } 62*57718be8SEnji Cooper 63*57718be8SEnji Cooper ATF_TC(nice_err); 64*57718be8SEnji Cooper ATF_TC_HEAD(nice_err, tc) 65*57718be8SEnji Cooper { 66*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", 67*57718be8SEnji Cooper "Test nice(3) for invalid parameters (PR lib/42587)"); 68*57718be8SEnji Cooper atf_tc_set_md_var(tc, "require.user", "unprivileged"); 69*57718be8SEnji Cooper } 70*57718be8SEnji Cooper 71*57718be8SEnji Cooper ATF_TC_BODY(nice_err, tc) 72*57718be8SEnji Cooper { 73*57718be8SEnji Cooper int i; 74*57718be8SEnji Cooper 75*57718be8SEnji Cooper /* 76*57718be8SEnji Cooper * The call should fail with EPERM if the 77*57718be8SEnji Cooper * supplied parameter is negative and the 78*57718be8SEnji Cooper * caller does not have privileges. 79*57718be8SEnji Cooper */ 80*57718be8SEnji Cooper for (i = -20; i < 0; i++) { 81*57718be8SEnji Cooper 82*57718be8SEnji Cooper errno = 0; 83*57718be8SEnji Cooper 84*57718be8SEnji Cooper ATF_REQUIRE_ERRNO(EPERM, nice(i) == -1); 85*57718be8SEnji Cooper } 86*57718be8SEnji Cooper } 87*57718be8SEnji Cooper 88*57718be8SEnji Cooper ATF_TC(nice_priority); 89*57718be8SEnji Cooper ATF_TC_HEAD(nice_priority, tc) 90*57718be8SEnji Cooper { 91*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test nice(3) vs. getpriority(2)"); 92*57718be8SEnji Cooper } 93*57718be8SEnji Cooper 94*57718be8SEnji Cooper ATF_TC_BODY(nice_priority, tc) 95*57718be8SEnji Cooper { 96*57718be8SEnji Cooper int i, pri, nic; 97*57718be8SEnji Cooper pid_t pid; 98*57718be8SEnji Cooper int sta; 99*57718be8SEnji Cooper 100*57718be8SEnji Cooper for (i = 0; i <= 20; i++) { 101*57718be8SEnji Cooper 102*57718be8SEnji Cooper nic = nice(i); 103*57718be8SEnji Cooper ATF_REQUIRE(nic != -1); 104*57718be8SEnji Cooper 105*57718be8SEnji Cooper errno = 0; 106*57718be8SEnji Cooper pri = getpriority(PRIO_PROCESS, 0); 107*57718be8SEnji Cooper ATF_REQUIRE(errno == 0); 108*57718be8SEnji Cooper 109*57718be8SEnji Cooper if (nic != pri) 110*57718be8SEnji Cooper atf_tc_fail("nice(3) and getpriority(2) conflict"); 111*57718be8SEnji Cooper 112*57718be8SEnji Cooper /* 113*57718be8SEnji Cooper * Also verify that the nice(3) values 114*57718be8SEnji Cooper * are inherited by child processes. 115*57718be8SEnji Cooper */ 116*57718be8SEnji Cooper pid = fork(); 117*57718be8SEnji Cooper ATF_REQUIRE(pid >= 0); 118*57718be8SEnji Cooper 119*57718be8SEnji Cooper if (pid == 0) { 120*57718be8SEnji Cooper 121*57718be8SEnji Cooper errno = 0; 122*57718be8SEnji Cooper pri = getpriority(PRIO_PROCESS, 0); 123*57718be8SEnji Cooper ATF_REQUIRE(errno == 0); 124*57718be8SEnji Cooper 125*57718be8SEnji Cooper if (nic != pri) 126*57718be8SEnji Cooper _exit(EXIT_FAILURE); 127*57718be8SEnji Cooper 128*57718be8SEnji Cooper _exit(EXIT_SUCCESS); 129*57718be8SEnji Cooper } 130*57718be8SEnji Cooper 131*57718be8SEnji Cooper (void)wait(&sta); 132*57718be8SEnji Cooper 133*57718be8SEnji Cooper if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS) 134*57718be8SEnji Cooper atf_tc_fail("nice(3) value was not inherited"); 135*57718be8SEnji Cooper } 136*57718be8SEnji Cooper } 137*57718be8SEnji Cooper 138*57718be8SEnji Cooper ATF_TC(nice_root); 139*57718be8SEnji Cooper ATF_TC_HEAD(nice_root, tc) 140*57718be8SEnji Cooper { 141*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test that nice(3) works"); 142*57718be8SEnji Cooper atf_tc_set_md_var(tc, "require.user", "root"); 143*57718be8SEnji Cooper } 144*57718be8SEnji Cooper 145*57718be8SEnji Cooper ATF_TC_BODY(nice_root, tc) 146*57718be8SEnji Cooper { 147*57718be8SEnji Cooper int i; 148*57718be8SEnji Cooper 149*57718be8SEnji Cooper for (i = -20; i <= 20; i++) { 150*57718be8SEnji Cooper 151*57718be8SEnji Cooper ATF_REQUIRE(nice(i) != -1); 152*57718be8SEnji Cooper } 153*57718be8SEnji Cooper } 154*57718be8SEnji Cooper 155*57718be8SEnji Cooper ATF_TC(nice_thread); 156*57718be8SEnji Cooper ATF_TC_HEAD(nice_thread, tc) 157*57718be8SEnji Cooper { 158*57718be8SEnji Cooper atf_tc_set_md_var(tc, "descr", "Test nice(3) with threads"); 159*57718be8SEnji Cooper } 160*57718be8SEnji Cooper 161*57718be8SEnji Cooper ATF_TC_BODY(nice_thread, tc) 162*57718be8SEnji Cooper { 163*57718be8SEnji Cooper pthread_t tid[5]; 164*57718be8SEnji Cooper int rv, val; 165*57718be8SEnji Cooper size_t i; 166*57718be8SEnji Cooper 167*57718be8SEnji Cooper /* 168*57718be8SEnji Cooper * Test that the scheduling priority is 169*57718be8SEnji Cooper * propagated to all system scope threads. 170*57718be8SEnji Cooper */ 171*57718be8SEnji Cooper for (i = 0; i < __arraycount(tid); i++) { 172*57718be8SEnji Cooper 173*57718be8SEnji Cooper val = nice(i); 174*57718be8SEnji Cooper ATF_REQUIRE(val != -1); 175*57718be8SEnji Cooper 176*57718be8SEnji Cooper rv = pthread_create(&tid[i], NULL, threadfunc, &val); 177*57718be8SEnji Cooper ATF_REQUIRE(rv == 0); 178*57718be8SEnji Cooper 179*57718be8SEnji Cooper rv = pthread_join(tid[i], NULL); 180*57718be8SEnji Cooper ATF_REQUIRE(rv == 0); 181*57718be8SEnji Cooper } 182*57718be8SEnji Cooper } 183*57718be8SEnji Cooper 184*57718be8SEnji Cooper ATF_TP_ADD_TCS(tp) 185*57718be8SEnji Cooper { 186*57718be8SEnji Cooper 187*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, nice_err); 188*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, nice_priority); 189*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, nice_root); 190*57718be8SEnji Cooper ATF_TP_ADD_TC(tp, nice_thread); 191*57718be8SEnji Cooper 192*57718be8SEnji Cooper return atf_no_error(); 193*57718be8SEnji Cooper } 194