2 * Copyright (c) 2012 The DragonFly Project. All rights reserved.
4 * This code is derived from software contributed to The DragonFly Project
5 * by Matthew Dillon <dillon@backplane.com> and Thomas Nikolajsen
6 * <thomas.nikolajsen@mail.dk>
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in
16 * the documentation and/or other materials provided with the
18 * 3. Neither the name of The DragonFly Project nor the names of its
19 * contributors may be used to endorse or promote products derived
20 * from this software without specific, prior written permission.
22 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
25 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
26 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
27 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
28 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
29 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
30 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
31 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
32 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36 #define _KERNEL_STRUCTURES
37 #include <sys/types.h>
38 #include <sys/usched.h>
44 static void usage(void);
49 main(int ac, char **av)
55 char *sched_cpustr = NULL;
59 CPUMASK_ASSZERO(cpumask);
61 while ((ch = getopt(ac, av, "d")) != -1) {
78 sched_cpustr = strdup(av[0]);
79 sched = strsep(&sched_cpustr, ":");
80 if (strcmp(sched, "default") == 0)
81 fprintf(stderr, "Ignoring scheduler == \"default\": not implemented\n");
82 cpustr = strsep(&sched_cpustr, "");
83 if (strlen(sched) == 0 && cpustr == NULL) {
89 * XXX needs expanded support for > 64 cpus
94 v = strtoul(cpustr, NULL, 0);
95 for (cpuid = 0; cpuid < (int)sizeof(v) * 8; ++cpuid) {
96 if (v & (1LU << cpuid))
97 CPUMASK_ORBIT(cpumask, cpuid);
101 if (strlen(sched) != 0) {
103 fprintf(stderr, "DEBUG: USCHED_SET_SCHEDULER: scheduler: %s\n", sched);
104 res = usched_set(getpid(), USCHED_SET_SCHEDULER, sched, strlen(sched));
106 perror("usched_set(,USCHED_SET_SCHEDULER,,)");
110 if (CPUMASK_TESTNZERO(cpumask)) {
111 for (cpuid = 0; cpuid < (int)sizeof(cpumask) * 8; ++cpuid) {
112 if (CPUMASK_TESTBIT(cpumask, cpuid))
116 fprintf(stderr, "DEBUG: USCHED_SET_CPU: cpuid: %d\n",
119 res = usched_set(getpid(), USCHED_SET_CPU,
120 &cpuid, sizeof(int));
122 perror("usched_set(,USCHED_SET_CPU,,)");
125 CPUMASK_NANDBIT(cpumask, cpuid);
126 while (CPUMASK_TESTNZERO(cpumask)) {
128 if (CPUMASK_TESTBIT(cpumask, cpuid) == 0)
130 CPUMASK_NANDBIT(cpumask, cpuid);
133 "DEBUG: USCHED_ADD_CPU: cpuid: %d\n",
136 res = usched_set(getpid(), USCHED_ADD_CPU,
137 &cpuid, sizeof(int));
139 perror("usched_set(,USCHED_ADD_CPU,,)");
144 execvp(av[1], av + 1);
153 "usage: usched [-d] {scheduler[:cpumask] | :cpumask} "
154 "program [argument ...]\n");