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 cpumask_t cpumask = 0; 56 int cpuid = -1; 57 58 while ((ch = getopt(ac, av, "d")) != -1) { 59 switch (ch) { 60 case 'd': 61 DebugOpt = 1; 62 break; 63 default: 64 usage(); 65 /* NOTREACHED */ 66 } 67 } 68 ac -= optind; 69 av += optind; 70 71 if (ac < 2) { 72 usage(); 73 /* NOTREACHED */ 74 } 75 sched_cpustr = strdup(av[0]); 76 sched = strsep(&sched_cpustr, ":"); 77 if (strcmp(sched, "default") == 0) 78 fprintf(stderr, "Ignoring scheduler == \"default\": not implemented\n"); 79 cpustr = strsep(&sched_cpustr, ""); 80 if (strlen(sched) == 0 && cpustr == NULL) { 81 usage(); 82 /* NOTREACHED */ 83 } 84 if (cpustr != NULL) 85 cpumask = strtoul(cpustr, NULL, 0); 86 87 if (strlen(sched) != 0) { 88 if (DebugOpt) 89 fprintf(stderr, "DEBUG: USCHED_SET_SCHEDULER: scheduler: %s\n", sched); 90 res = usched_set(getpid(), USCHED_SET_SCHEDULER, sched, strlen(sched)); 91 if (res != 0) { 92 perror("usched_set(,USCHED_SET_SCHEDULER,,)"); 93 exit(1); 94 } 95 } 96 if (cpumask != 0) { 97 while ((cpumask & 1) == 0) { 98 cpuid++; 99 cpumask >>= 1; 100 } 101 cpuid++; 102 cpumask >>= 1; 103 if (DebugOpt) 104 fprintf(stderr, "DEBUG: USCHED_SET_CPU: cpuid: %d\n", cpuid); 105 res = usched_set(getpid(), USCHED_SET_CPU, &cpuid, sizeof(int)); 106 if (res != 0) { 107 perror("usched_set(,USCHED_SET_CPU,,)"); 108 exit(1); 109 } 110 while (cpumask != 0) { 111 while ((cpumask & 1) == 0) { 112 cpuid++; 113 cpumask >>= 1; 114 } 115 cpuid++; 116 cpumask >>= 1; 117 if (DebugOpt) 118 fprintf(stderr, "DEBUG: USCHED_ADD_CPU: cpuid: %d\n", cpuid); 119 res = usched_set(getpid(), USCHED_ADD_CPU, &cpuid, sizeof(int)); 120 if (res != 0) { 121 perror("usched_set(,USCHED_ADD_CPU,,)"); 122 exit(1); 123 } 124 } 125 } 126 execvp(av[1], av + 1); 127 exit(1); 128 } 129 130 static 131 void 132 usage(void) 133 { 134 fprintf(stderr, "usage: usched [-d] {scheduler[:cpumask] | :cpumask} program [argument ...]\n"); 135 exit(1); 136 } 137