xref: /dragonfly/sbin/usched/usched.c (revision 25a2db75)
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