usched - Add usched utility

* Currently must run as root

* usched {bsd4,dfly} program args...

diff --git a/sbin/Makefile b/sbin/Makefile
index 35ae39a..8d5bf47 100644 (file)
--- a/sbin/Makefile
+++ b/sbin/Makefile
@@ -96,6 +96,7 @@ SUBDIR=       adjkerntz \
        tunefs \
        udevd \
        umount \
+       usched \
        vinum \
        vquota
 

diff --git a/sbin/usched/Makefile b/sbin/usched/Makefile
new file mode 100644 (file)
index 0000000..c84f49e
--- /dev/null
+++ b/sbin/usched/Makefile
@@ -0,0 +1,8 @@
+#
+#
+PROG=  usched
+SRCS=  usched.c
+#MAN=  usched.8
+NOMAN=TRUE
+
+.include <bsd.prog.mk>

diff --git a/sbin/usched/usched.c b/sbin/usched/usched.c
new file mode 100644 (file)
index 0000000..ac0d843
--- /dev/null
+++ b/sbin/usched/usched.c
@@ -0,0 +1,92 @@
+/*
+ * Copyright (c) 2012 The DragonFly Project.  All rights reserved.
+ *
+ * This code is derived from software contributed to The DragonFly Project
+ * by Matthew Dillon <dillon@backplane.com>
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in
+ *    the documentation and/or other materials provided with the
+ *    distribution.
+ * 3. Neither the name of The DragonFly Project nor the names of its
+ *    contributors may be used to endorse or promote products derived
+ *    from this software without specific, prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
+ * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
+ * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <sys/types.h>
+#include <sys/usched.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+
+static void usage(void);
+
+int VerboseOpt;
+
+int
+main(int ac, char **av)
+{
+       int ch;
+       int res;
+       char *sched = NULL;
+       char *cpustr = NULL;
+       cpumask_t cpumask = 0;
+
+       while ((ch = getopt(ac, av, "v")) != -1) {
+               switch(ch) {
+               case 'v':
+                       VerboseOpt = 1;
+                       break;
+               default:
+                       usage();
+                       break;  /* NOT REACHED */
+               }
+       }
+       ac -= optind;
+       av += optind;
+
+       if (ac < 2)
+               usage();
+       sched = strtok(strdup(av[0]), ":");
+       cpustr = strtok(NULL, "");
+       if (cpustr == NULL)
+               cpumask = -1;
+       else
+               cpumask = strtoul(cpustr, NULL, 0);
+
+       res = usched_set(getpid(), USCHED_SET_SCHEDULER, sched, strlen(sched));
+       if (res != 0) {
+               perror("usched_set");
+               exit(1);
+       }
+       execvp(av[1], av + 1);
+       exit(1);
+}
+
+static
+void
+usage(void)
+{
+       fprintf(stderr, "usched scheduler[:cpumask] program args...\n");
+       exit(1);
+}