1*f14fb602SLionel Sambuc /* $NetBSD: sched.c,v 1.4 2012/03/18 02:04:39 christos Exp $ */
22fe8fb19SBen Gras
32fe8fb19SBen Gras /*
42fe8fb19SBen Gras * Copyright (c) 2008, Mindaugas Rasiukevicius <rmind at NetBSD org>
52fe8fb19SBen Gras * All rights reserved.
62fe8fb19SBen Gras *
72fe8fb19SBen Gras * Redistribution and use in source and binary forms, with or without
82fe8fb19SBen Gras * modification, are permitted provided that the following conditions
92fe8fb19SBen Gras * are met:
102fe8fb19SBen Gras * 1. Redistributions of source code must retain the above copyright
112fe8fb19SBen Gras * notice, this list of conditions and the following disclaimer.
122fe8fb19SBen Gras * 2. Redistributions in binary form must reproduce the above copyright
132fe8fb19SBen Gras * notice, this list of conditions and the following disclaimer in the
142fe8fb19SBen Gras * documentation and/or other materials provided with the distribution.
152fe8fb19SBen Gras *
162fe8fb19SBen Gras * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
172fe8fb19SBen Gras * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
182fe8fb19SBen Gras * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
192fe8fb19SBen Gras * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
202fe8fb19SBen Gras * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
212fe8fb19SBen Gras * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
222fe8fb19SBen Gras * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
232fe8fb19SBen Gras * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
242fe8fb19SBen Gras * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
252fe8fb19SBen Gras * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
262fe8fb19SBen Gras * SUCH DAMAGE.
272fe8fb19SBen Gras */
282fe8fb19SBen Gras
292fe8fb19SBen Gras #include <sys/cdefs.h>
30*f14fb602SLionel Sambuc __RCSID("$NetBSD: sched.c,v 1.4 2012/03/18 02:04:39 christos Exp $");
312fe8fb19SBen Gras
322fe8fb19SBen Gras #include <string.h>
332fe8fb19SBen Gras #include <unistd.h>
342fe8fb19SBen Gras #include <errno.h>
352fe8fb19SBen Gras #include <sched.h>
36*f14fb602SLionel Sambuc #include <signal.h>
372fe8fb19SBen Gras #include <sys/param.h>
382fe8fb19SBen Gras #include <sys/types.h>
392fe8fb19SBen Gras
402fe8fb19SBen Gras /* All LWPs in the process */
412fe8fb19SBen Gras #define P_ALL_LWPS 0
422fe8fb19SBen Gras
432fe8fb19SBen Gras /*
442fe8fb19SBen Gras * Scheduling parameters.
452fe8fb19SBen Gras */
462fe8fb19SBen Gras
472fe8fb19SBen Gras int
sched_setparam(pid_t pid,const struct sched_param * param)482fe8fb19SBen Gras sched_setparam(pid_t pid, const struct sched_param *param)
492fe8fb19SBen Gras {
502fe8fb19SBen Gras struct sched_param sp;
512fe8fb19SBen Gras
522fe8fb19SBen Gras memset(&sp, 0, sizeof(struct sched_param));
532fe8fb19SBen Gras sp.sched_priority = param->sched_priority;
542fe8fb19SBen Gras return _sched_setparam(pid, P_ALL_LWPS, SCHED_NONE, &sp);
552fe8fb19SBen Gras }
562fe8fb19SBen Gras
572fe8fb19SBen Gras int
sched_getparam(pid_t pid,struct sched_param * param)582fe8fb19SBen Gras sched_getparam(pid_t pid, struct sched_param *param)
592fe8fb19SBen Gras {
602fe8fb19SBen Gras
612fe8fb19SBen Gras return _sched_getparam(pid, P_ALL_LWPS, NULL, param);
622fe8fb19SBen Gras }
632fe8fb19SBen Gras
642fe8fb19SBen Gras int
sched_setscheduler(pid_t pid,int policy,const struct sched_param * param)652fe8fb19SBen Gras sched_setscheduler(pid_t pid, int policy, const struct sched_param *param)
662fe8fb19SBen Gras {
672fe8fb19SBen Gras struct sched_param sp;
682fe8fb19SBen Gras int ret, old_policy;
692fe8fb19SBen Gras
702fe8fb19SBen Gras ret = _sched_getparam(pid, P_ALL_LWPS, &old_policy, &sp);
712fe8fb19SBen Gras if (ret < 0)
722fe8fb19SBen Gras return ret;
732fe8fb19SBen Gras
742fe8fb19SBen Gras memset(&sp, 0, sizeof(struct sched_param));
752fe8fb19SBen Gras sp.sched_priority = param->sched_priority;
762fe8fb19SBen Gras ret = _sched_setparam(pid, P_ALL_LWPS, policy, &sp);
772fe8fb19SBen Gras if (ret < 0)
782fe8fb19SBen Gras return ret;
792fe8fb19SBen Gras
802fe8fb19SBen Gras return old_policy;
812fe8fb19SBen Gras }
822fe8fb19SBen Gras
832fe8fb19SBen Gras int
sched_getscheduler(pid_t pid)842fe8fb19SBen Gras sched_getscheduler(pid_t pid)
852fe8fb19SBen Gras {
862fe8fb19SBen Gras struct sched_param sp;
872fe8fb19SBen Gras int ret, policy;
882fe8fb19SBen Gras
892fe8fb19SBen Gras ret = _sched_getparam(pid, P_ALL_LWPS, &policy, &sp);
902fe8fb19SBen Gras if (ret < 0)
912fe8fb19SBen Gras return ret;
922fe8fb19SBen Gras
932fe8fb19SBen Gras return policy;
942fe8fb19SBen Gras }
952fe8fb19SBen Gras
962fe8fb19SBen Gras /*
972fe8fb19SBen Gras * Scheduling priorities.
982fe8fb19SBen Gras */
992fe8fb19SBen Gras
1002fe8fb19SBen Gras int
sched_get_priority_max(int policy)1012fe8fb19SBen Gras sched_get_priority_max(int policy)
1022fe8fb19SBen Gras {
1032fe8fb19SBen Gras
1042fe8fb19SBen Gras if (policy < SCHED_OTHER || policy > SCHED_RR) {
1052fe8fb19SBen Gras errno = EINVAL;
1062fe8fb19SBen Gras return -1;
1072fe8fb19SBen Gras }
108*f14fb602SLionel Sambuc return (int)sysconf(_SC_SCHED_PRI_MAX);
1092fe8fb19SBen Gras }
1102fe8fb19SBen Gras
1112fe8fb19SBen Gras int
sched_get_priority_min(int policy)1122fe8fb19SBen Gras sched_get_priority_min(int policy)
1132fe8fb19SBen Gras {
1142fe8fb19SBen Gras
1152fe8fb19SBen Gras if (policy < SCHED_OTHER || policy > SCHED_RR) {
1162fe8fb19SBen Gras errno = EINVAL;
1172fe8fb19SBen Gras return -1;
1182fe8fb19SBen Gras }
119*f14fb602SLionel Sambuc return (int)sysconf(_SC_SCHED_PRI_MIN);
1202fe8fb19SBen Gras }
1212fe8fb19SBen Gras
1222fe8fb19SBen Gras int
1232fe8fb19SBen Gras /*ARGSUSED*/
sched_rr_get_interval(pid_t pid,struct timespec * interval)1242fe8fb19SBen Gras sched_rr_get_interval(pid_t pid, struct timespec *interval)
1252fe8fb19SBen Gras {
1262fe8fb19SBen Gras
127*f14fb602SLionel Sambuc if (pid && kill(pid, 0) == -1)
128*f14fb602SLionel Sambuc return -1;
1292fe8fb19SBen Gras interval->tv_sec = 0;
1302fe8fb19SBen Gras interval->tv_nsec = sysconf(_SC_SCHED_RT_TS) * 1000;
1312fe8fb19SBen Gras return 0;
1322fe8fb19SBen Gras }
1332fe8fb19SBen Gras
1342fe8fb19SBen Gras /*
1352fe8fb19SBen Gras * Process affinity.
1362fe8fb19SBen Gras */
1372fe8fb19SBen Gras
1382fe8fb19SBen Gras int
sched_getaffinity_np(pid_t pid,size_t size,cpuset_t * cpuset)1392fe8fb19SBen Gras sched_getaffinity_np(pid_t pid, size_t size, cpuset_t *cpuset)
1402fe8fb19SBen Gras {
1412fe8fb19SBen Gras
1422fe8fb19SBen Gras return _sched_getaffinity(pid, P_ALL_LWPS, size, cpuset);
1432fe8fb19SBen Gras }
1442fe8fb19SBen Gras
1452fe8fb19SBen Gras int
sched_setaffinity_np(pid_t pid,size_t size,cpuset_t * cpuset)1462fe8fb19SBen Gras sched_setaffinity_np(pid_t pid, size_t size, cpuset_t *cpuset)
1472fe8fb19SBen Gras {
1482fe8fb19SBen Gras
1492fe8fb19SBen Gras return _sched_setaffinity(pid, P_ALL_LWPS, size, cpuset);
1502fe8fb19SBen Gras }
151