sbin/usched/usched.c

   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 }