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 #include <sys/types.h> 37 #include <sys/usched.h> 38 #include <stdio.h> 39 #include <stdlib.h> 40 #include <unistd.h> 41 #include <string.h> 42 43 static void usage(void); 44 45 int DebugOpt; 46 47 int 48 main(int ac, char **av) 49 { 50 int ch; 51 int res; 52 char *sched = NULL; 53 char *cpustr = NULL; 54 char *sched_cpustr = NULL; 55 char *p = NULL; 56 cpumask_t cpumask; 57 int cpuid; 58 pid_t pid = getpid(); /* See usched_set(2) - BUGS */ 59 60 CPUMASK_ASSZERO(cpumask); 61 62 while ((ch = getopt(ac, av, "d")) != -1) { 63 switch (ch) { 64 case 'd': 65 DebugOpt = 1; 66 break; 67 default: 68 usage(); 69 /* NOTREACHED */ 70 } 71 } 72 ac -= optind; 73 av += optind; 74 75 if (ac < 2) { 76 usage(); 77 /* NOTREACHED */ 78 } 79 sched_cpustr = strdup(av[0]); 80 sched = strsep(&sched_cpustr, ":"); 81 if (strcmp(sched, "default") == 0) 82 fprintf(stderr, "Ignoring scheduler == \"default\": not implemented\n"); 83 cpustr = strsep(&sched_cpustr, ""); 84 if (strlen(sched) == 0 && cpustr == NULL) { 85 usage(); 86 /* NOTREACHED */ 87 } 88 89 /* 90 * XXX needs expanded support for > 64 cpus 91 */ 92 if (cpustr != NULL) { 93 uint64_t v; 94 95 v = (uint64_t)strtoull(cpustr, NULL, 0); 96 for (cpuid = 0; cpuid < (int)sizeof(v) * 8; ++cpuid) { 97 if (v & (1LU << cpuid)) 98 CPUMASK_ORBIT(cpumask, cpuid); 99 } 100 } 101 102 if (strlen(sched) != 0) { 103 if (DebugOpt) 104 fprintf(stderr, "DEBUG: USCHED_SET_SCHEDULER: scheduler: %s\n", sched); 105 res = usched_set(pid, USCHED_SET_SCHEDULER, sched, strlen(sched)); 106 if (res != 0) { 107 asprintf(&p, "usched_set(%d, USCHED_SET_SCHEDULER, \"%s\", %d)", 108 pid, sched, (int)strlen(sched)); 109 perror(p); 110 exit(1); 111 } 112 } 113 if (CPUMASK_TESTNZERO(cpumask)) { 114 for (cpuid = 0; cpuid < (int)sizeof(cpumask) * 8; ++cpuid) { 115 if (CPUMASK_TESTBIT(cpumask, cpuid)) 116 break; 117 } 118 if (DebugOpt) { 119 fprintf(stderr, "DEBUG: USCHED_SET_CPU: cpuid: %d\n", 120 cpuid); 121 } 122 res = usched_set(pid, USCHED_SET_CPU, &cpuid, sizeof(int)); 123 if (res != 0) { 124 asprintf(&p, "usched_set(%d, USCHED_SET_CPU, &%d, %d)", 125 pid, cpuid, (int)sizeof(int)); 126 perror(p); 127 exit(1); 128 } 129 CPUMASK_NANDBIT(cpumask, cpuid); 130 while (CPUMASK_TESTNZERO(cpumask)) { 131 ++cpuid; 132 if (CPUMASK_TESTBIT(cpumask, cpuid) == 0) 133 continue; 134 CPUMASK_NANDBIT(cpumask, cpuid); 135 if (DebugOpt) { 136 fprintf(stderr, 137 "DEBUG: USCHED_ADD_CPU: cpuid: %d\n", 138 cpuid); 139 } 140 res = usched_set(pid, USCHED_ADD_CPU, &cpuid, sizeof(int)); 141 if (res != 0) { 142 asprintf(&p, "usched_set(%d, USCHED_ADD_CPU, &%d, %d)", 143 pid, cpuid, (int)sizeof(int)); 144 perror(p); 145 exit(1); 146 } 147 } 148 } 149 execvp(av[1], av + 1); 150 exit(1); 151 } 152 153 static 154 void 155 usage(void) 156 { 157 fprintf(stderr, 158 "usage: usched [-d] {scheduler[:cpumask] | :cpumask} " 159 "program [argument ...]\n"); 160 exit(1); 161 } 162