[dragonfly.git] / usr.sbin / dschedctl / dschedctl.c

/*
 * Copyright (c) 2010 The DragonFly Project.  All rights reserved.
 *
 * This code is derived from software contributed to The DragonFly Project
 * by Alex Hornung <ahornung@gmail.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/cdefs.h>
#include <sys/syslimits.h>
#include <sys/ioctl.h>
#include <sys/device.h>
#include <sys/queue.h>
#include <sys/stat.h>
#include <sys/dsched.h>

#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <grp.h>
#include <libgen.h>
#include <pwd.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define DSCHED_FOREACH_DISK(dev_fd, diocp)		\
	for ((diocp)->num_elem = 0;			\
	     (ioctl((dev_fd),				\
		 DSCHED_LIST_DISKS, (diocp)) != -1);	\
	     ++(diocp)->num_elem)

#define DSCHED_FOREACH_POLICY(dev_fd, diocp)		\
	for ((diocp)->num_elem = 0;			\
	     (ioctl((dev_fd),				\
		 DSCHED_LIST_POLICIES, (diocp)) != -1);	\
	     ++(diocp)->num_elem)

static int dev_fd;
static int verbose = 0;

static void
usage(void)
{
	fprintf(stderr,
	    "Usage: dschedctl <commands>\n"
	    "Valid commands are:\n"
	    " -l [disk]\n"
	    "\t Lists all disks and their policies, or just for [disk]\n"
	    " -p\n"
	    "\t Lists all available I/O scheduling policies\n"
	    " -s <policy> [disk]\n"
	    "\t Switches the policy of [disk] or otherwise all disks to "
	    "<policy>\n");

	exit(1);
}

static int
dsched_ioctl(unsigned long cmd, struct dsched_ioctl *pdioc)
{
	if (ioctl(dev_fd, cmd, pdioc) == -1)
		err(1, "ioctl");

	return 0;
}

static int dsched_set_disk_policy(char *disk_name, char *policy)
{
	struct dsched_ioctl	dioc;
	int error;

	bzero(&dioc, sizeof(dioc));
	strncpy(dioc.dev_name, disk_name, DSCHED_NAME_LENGTH);
	strncpy(dioc.pol_name, policy, DSCHED_NAME_LENGTH);
	error = dsched_ioctl(DSCHED_SET_DEVICE_POLICY, &dioc);

	if ((!error) && verbose) {
		printf("Switched scheduler policy of %s successfully to %s\n",
		    disk_name, policy);
	}

	return error;
}

int main(int argc, char *argv[])
{
	struct dsched_ioctl	dioc;
	char	*disk_name = NULL;
	char	*policy = NULL;
	int	dflag = 0, lflag = 0, pflag = 0, sflag = 0;
	int	ch, error = 0;

	while ((ch = getopt(argc, argv, "hlps:v")) != -1) {
		switch (ch) {
		case 'l':
			lflag = 1;
			break;
		case 'p':
			pflag = 1;
			break;
		case 's':
			sflag = 1;
			policy = optarg;
			break;
		case 'v':
			verbose = 1;
			break;
		case 'h':
		case '?':
		default:
			usage();
			/* NOT REACHED */
		}
	}

	argc -= optind;
	argv += optind;

	if (argc == 1) {
		dflag = 1;
		disk_name = basename(argv[0]);
		if (!disk_name)
			err(1, "basename");
	} else if (argc > 1) {
		usage();
		/* NOT REACHED */
	}

	/*
	 * Check arguments:
	 * - need to use at least one mode
	 */
	if (!(lflag || pflag || sflag)) {
		usage();
		/* NOT REACHED */
	}

	dev_fd = open("/dev/dsched", O_RDWR);
	if (dev_fd == -1)
		err(1, "open(/dev/dsched)");

	if (lflag) {
		if (dflag) {
			strncpy(dioc.dev_name, disk_name, DSCHED_NAME_LENGTH);
			error = dsched_ioctl(DSCHED_LIST_DISK, &dioc);
			if (!error) {
				printf("%s\t=>\t%s\n",
				    disk_name, dioc.pol_name);
			}
		} else {
			DSCHED_FOREACH_DISK(dev_fd, &dioc) {
				printf("%s\t=>\t%s\n",
				    dioc.dev_name, dioc.pol_name);
			}
		}
	}

	if (pflag) {
		DSCHED_FOREACH_POLICY(dev_fd, &dioc) {
			printf("%s\n", dioc.pol_name);
		}
	}

	if (sflag) {
		if (dflag) {
			error = dsched_set_disk_policy(disk_name, policy);
		} else {
			DSCHED_FOREACH_DISK(dev_fd, &dioc) {
				error = dsched_set_disk_policy(dioc.dev_name,
				    policy);
			}
		}
	}

	close(dev_fd);

	return (error);
}
Commit	Line	Data
b80a9543 AH	1	/*
	2	* Copyright (c) 2010 The DragonFly Project. All rights reserved.
	3	*
	4	* This code is derived from software contributed to The DragonFly Project
	5	* by Alex Hornung <ahornung@gmail.com>
	6	*
	7	* Redistribution and use in source and binary forms, with or without
	8	* modification, are permitted provided that the following conditions
	9	* are met:
	10	*
	11	* 1. Redistributions of source code must retain the above copyright
	12	* notice, this list of conditions and the following disclaimer.
	13	* 2. Redistributions in binary form must reproduce the above copyright
	14	* notice, this list of conditions and the following disclaimer in
	15	* the documentation and/or other materials provided with the
	16	* distribution.
	17	* 3. Neither the name of The DragonFly Project nor the names of its
	18	* contributors may be used to endorse or promote products derived
	19	* from this software without specific, prior written permission.
	20	*
	21	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
	22	* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
	23	* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
	24	* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
	25	* COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
	26	* INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
	27	* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
	28	* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
	29	* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
	30	* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
	31	* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
	32	* SUCH DAMAGE.
	33	*/
	34
	35	#include <sys/types.h>
	36	#include <sys/cdefs.h>
	37	#include <sys/syslimits.h>
	38	#include <sys/ioctl.h>
	39	#include <sys/device.h>
	40	#include <sys/queue.h>
	41	#include <sys/stat.h>
	42	#include <sys/dsched.h>
	43
	44	#include <ctype.h>
	45	#include <err.h>
	46	#include <errno.h>
	47	#include <fcntl.h>
	48	#include <grp.h>
	49	#include <libgen.h>
	50	#include <pwd.h>
	51	#include <stdarg.h>
	52	#include <stdio.h>
	53	#include <stdlib.h>
	54	#include <string.h>
	55	#include <unistd.h>
	56
9e551a87 SK	57	#define DSCHED_FOREACH_DISK(dev_fd, diocp) \
	58	for ((diocp)->num_elem = 0; \
	59	(ioctl((dev_fd), \
	60	DSCHED_LIST_DISKS, (diocp)) != -1); \
	61	++(diocp)->num_elem)
	62
	63	#define DSCHED_FOREACH_POLICY(dev_fd, diocp) \
	64	for ((diocp)->num_elem = 0; \
	65	(ioctl((dev_fd), \
	66	DSCHED_LIST_POLICIES, (diocp)) != -1); \
	67	++(diocp)->num_elem)
b80a9543	68
0e03113f AH	69	static int dev_fd;
0e03113f AH	70	static int verbose = 0;
b80a9543 AH	71
	72	static void
	73	usage(void)
	74	{
	75	fprintf(stderr,
9e551a87 SK	76	"Usage: dschedctl <commands>\n"
	77	"Valid commands are:\n"
	78	" -l [disk]\n"
	79	"\t Lists all disks and their policies, or just for [disk]\n"
	80	" -p\n"
	81	"\t Lists all available I/O scheduling policies\n"
	82	" -s <policy> [disk]\n"
	83	"\t Switches the policy of [disk] or otherwise all disks to "
	84	"<policy>\n");
b80a9543 AH	85
	86	exit(1);
	87	}
	88
b80a9543 AH	89	static int
	90	dsched_ioctl(unsigned long cmd, struct dsched_ioctl *pdioc)
	91	{
	92	if (ioctl(dev_fd, cmd, pdioc) == -1)
	93	err(1, "ioctl");
	94
	95	return 0;
	96	}
	97
0e03113f AH	98	static int dsched_set_disk_policy(char disk_name, char policy)
	99	{
	100	struct dsched_ioctl dioc;
	101	int error;
	102
	103	bzero(&dioc, sizeof(dioc));
	104	strncpy(dioc.dev_name, disk_name, DSCHED_NAME_LENGTH);
	105	strncpy(dioc.pol_name, policy, DSCHED_NAME_LENGTH);
	106	error = dsched_ioctl(DSCHED_SET_DEVICE_POLICY, &dioc);
	107
	108	if ((!error) && verbose) {
	109	printf("Switched scheduler policy of %s successfully to %s\n",
	110	disk_name, policy);
	111	}
	112
	113	return error;
	114	}
b80a9543 AH	115
	116	int main(int argc, char *argv[])
	117	{
	118	struct dsched_ioctl dioc;
	119	char *disk_name = NULL;
	120	char *policy = NULL;
	121	int dflag = 0, lflag = 0, pflag = 0, sflag = 0;
0e03113f	122	int ch, error = 0;
b80a9543	123
0e03113f	124	while ((ch = getopt(argc, argv, "hlps:v")) != -1) {
b80a9543	125	switch (ch) {
b80a9543 AH	126	case 'l':
	127	lflag = 1;
	128	break;
	129	case 'p':
	130	pflag = 1;
	131	break;
	132	case 's':
	133	sflag = 1;
	134	policy = optarg;
	135	break;
0e03113f AH	136	case 'v':
	137	verbose = 1;
	138	break;
b80a9543 AH	139	case 'h':
	140	case '?':
	141	default:
	142	usage();
	143	/* NOT REACHED */
	144	}
	145	}
	146
	147	argc -= optind;
	148	argv += optind;
	149
0e03113f AH	150	if (argc == 1) {
0e03113f AH	151	dflag = 1;
f10a41a5 SK	152	disk_name = basename(argv[0]);
	153	if (!disk_name)
	154	err(1, "basename");
0e03113f AH	155	} else if (argc > 1) {
	156	usage();
	157	/* NOT REACHED */
	158	}
	159
b80a9543 AH	160	/*
	161	* Check arguments:
	162	* - need to use at least one mode
b80a9543	163	*/
0e03113f	164	if (!(lflag \|\| pflag \|\| sflag)) {
b80a9543 AH	165	usage();
	166	/* NOT REACHED */
	167	}
	168
	169	dev_fd = open("/dev/dsched", O_RDWR);
	170	if (dev_fd == -1)
	171	err(1, "open(/dev/dsched)");
	172
	173	if (lflag) {
	174	if (dflag) {
	175	strncpy(dioc.dev_name, disk_name, DSCHED_NAME_LENGTH);
	176	error = dsched_ioctl(DSCHED_LIST_DISK, &dioc);
	177	if (!error) {
9e551a87 SK	178	printf("%s\t=>\t%s\n",
9e551a87 SK	179	disk_name, dioc.pol_name);
b80a9543 AH	180	}
b80a9543 AH	181	} else {
0e03113f	182	DSCHED_FOREACH_DISK(dev_fd, &dioc) {
9e551a87 SK	183	printf("%s\t=>\t%s\n",
9e551a87 SK	184	dioc.dev_name, dioc.pol_name);
b80a9543 AH	185	}
	186	}
	187	}
	188
	189	if (pflag) {
0e03113f AH	190	DSCHED_FOREACH_POLICY(dev_fd, &dioc) {
0e03113f AH	191	printf("%s\n", dioc.pol_name);
b80a9543 AH	192	}
	193	}
	194
	195	if (sflag) {
0e03113f AH	196	if (dflag) {
	197	error = dsched_set_disk_policy(disk_name, policy);
	198	} else {
	199	DSCHED_FOREACH_DISK(dev_fd, &dioc) {
9e551a87 SK	200	error = dsched_set_disk_policy(dioc.dev_name,
9e551a87 SK	201	policy);
0e03113f	202	}
b80a9543 AH	203	}
	204	}
	205
	206	close(dev_fd);
	207
0e03113f	208	return (error);
b80a9543	209	}