1 /* 2 * Copyright (c) 2012 The DragonFly Project. All rights reserved. 3 * 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> 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 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 17 * distribution. 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. 21 * 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 33 * SUCH DAMAGE. 34 */ 35 36 #define _KERNEL_STRUCTURES 37 #include <sys/types.h> 38 #include <sys/usched.h> 39 #include <stdio.h> 40 #include <stdlib.h> 41 #include <unistd.h> 42 #include <string.h> 43 44 static void usage(void); 45 46 int DebugOpt; 47 48 int 49 main(int ac, char **av) 50 { 51 int ch; 52 int res; 53 char *sched = NULL; 54 char *cpustr = NULL; 55 char *sched_cpustr = NULL; 56 cpumask_t cpumask; 57 int cpuid; 58 59 CPUMASK_ASSZERO(cpumask); 60 61 while ((ch = getopt(ac, av, "d")) != -1) { 62 switch (ch) { 63 case 'd': 64 DebugOpt = 1; 65 break; 66 default: 67 usage(); 68 /* NOTREACHED */ 69 } 70 } 71 ac -= optind; 72 av += optind; 73 74 if (ac < 2) { 75 usage(); 76 /* NOTREACHED */ 77 } 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) { 84 usage(); 85 /* NOTREACHED */ 86 } 87 88 /* 89 * XXX needs expanded support for > 64 cpus 90 */ 91 if (cpustr != NULL) { 92 unsigned long v; 93 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); 98 } 99 } 100 101 if (strlen(sched) != 0) { 102 if (DebugOpt) 103 fprintf(stderr, "DEBUG: USCHED_SET_SCHEDULER: scheduler: %s\n", sched); 104 res = usched_set(getpid(), USCHED_SET_SCHEDULER, sched, strlen(sched)); 105 if (res != 0) { 106 perror("usched_set(,USCHED_SET_SCHEDULER,,)"); 107 exit(1); 108 } 109 } 110 if (CPUMASK_TESTNZERO(cpumask)) { 111 for (cpuid = 0; cpuid < (int)sizeof(cpumask) * 8; ++cpuid) { 112 if (CPUMASK_TESTBIT(cpumask, cpuid)) 113 break; 114 } 115 if (DebugOpt) { 116 fprintf(stderr, "DEBUG: USCHED_SET_CPU: cpuid: %d\n", 117 cpuid); 118 } 119 res = usched_set(getpid(), USCHED_SET_CPU, 120 &cpuid, sizeof(int)); 121 if (res != 0) { 122 perror("usched_set(,USCHED_SET_CPU,,)"); 123 exit(1); 124 } 125 CPUMASK_NANDBIT(cpumask, cpuid); 126 while (CPUMASK_TESTNZERO(cpumask)) { 127 ++cpuid; 128 if (CPUMASK_TESTBIT(cpumask, cpuid) == 0) 129 continue; 130 CPUMASK_NANDBIT(cpumask, cpuid); 131 if (DebugOpt) { 132 fprintf(stderr, 133 "DEBUG: USCHED_ADD_CPU: cpuid: %d\n", 134 cpuid); 135 } 136 res = usched_set(getpid(), USCHED_ADD_CPU, 137 &cpuid, sizeof(int)); 138 if (res != 0) { 139 perror("usched_set(,USCHED_ADD_CPU,,)"); 140 exit(1); 141 } 142 } 143 } 144 execvp(av[1], av + 1); 145 exit(1); 146 } 147 148 static 149 void 150 usage(void) 151 { 152 fprintf(stderr, 153 "usage: usched [-d] {scheduler[:cpumask] | :cpumask} " 154 "program [argument ...]\n"); 155 exit(1); 156 } 157