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 #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 }