Home | History | Annotate | Line # | Download | only in intrctl
intrctl.c revision 1.3
      1 /*	$NetBSD: intrctl.c,v 1.3 2016/09/19 18:46:39 ryo Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2015 Internet Initiative Japan Inc.
      5  * All rights reserved.
      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  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     26  * POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 
     29 #include <sys/cdefs.h>
     30 __RCSID("$NetBSD: intrctl.c,v 1.3 2016/09/19 18:46:39 ryo Exp $");
     31 
     32 #include <sys/param.h>
     33 #include <sys/sysctl.h>
     34 #include <sys/intrio.h>
     35 #include <sys/types.h>
     36 
     37 #include <err.h>
     38 #include <errno.h>
     39 #include <fcntl.h>
     40 #include <limits.h>
     41 #include <paths.h>
     42 #include <sched.h>
     43 #include <stdint.h>
     44 #include <stdio.h>
     45 #include <stdlib.h>
     46 #include <string.h>
     47 #include <unistd.h>
     48 
     49 #include "intrctl_io.h"
     50 
     51 __dead static void	usage(void);
     52 
     53 int		verbose;
     54 
     55 static void	intrctl_list(int, char **);
     56 static void	intrctl_affinity(int, char **);
     57 static void	intrctl_intr(int, char **);
     58 static void	intrctl_nointr(int, char **);
     59 
     60 static struct cmdtab {
     61 	const char	*label;
     62 	void	(*func)(int, char **);
     63 } const intrctl_cmdtab[] = {
     64 	{ "list", intrctl_list },
     65 	{ "affinity", intrctl_affinity },
     66 	{ "intr", intrctl_intr },
     67 	{ "nointr", intrctl_nointr },
     68 	{ NULL, NULL },
     69 };
     70 
     71 int
     72 main(int argc, char **argv)
     73 {
     74 	const struct cmdtab *ct;
     75 	char *cmdname;
     76 
     77 	if (argc < 2)
     78 		usage();
     79 
     80 	cmdname = argv[1];
     81 	argv += 1;
     82 	argc -= 1;
     83 
     84 	for (ct = intrctl_cmdtab; ct->label != NULL; ct++) {
     85 		if (strcmp(cmdname, ct->label) == 0) {
     86 			break;
     87 		}
     88 	}
     89 	if (ct->label == NULL)
     90 		errx(EXIT_FAILURE, "unknown command ``%s''", cmdname);
     91 
     92 	(*ct->func)(argc, argv);
     93 	exit(EXIT_SUCCESS);
     94 	/* NOTREACHED */
     95 }
     96 
     97 static void
     98 usage(void)
     99 {
    100 	const char *progname = getprogname();
    101 
    102 	fprintf(stderr, "usage: %s list\n", progname);
    103 	fprintf(stderr, "       %s affinity -i interrupt_name -c cpu_index\n", progname);
    104 	fprintf(stderr, "       %s intr -c cpu_index\n", progname);
    105 	fprintf(stderr, "       %s nointr -c cpu_index\n", progname);
    106 	exit(EXIT_FAILURE);
    107 	/* NOTREACHED */
    108 }
    109 
    110 static int intrctl_io_alloc_retry_count = 4;
    111 
    112 static void
    113 intrctl_list(int argc, char **argv)
    114 {
    115 	char buf[64];
    116 	struct intrio_list_line *illine;
    117 	int i, ncpus, *cpucol;
    118 	void *handle;
    119 	size_t intridlen;
    120 
    121 	handle = intrctl_io_alloc(intrctl_io_alloc_retry_count);
    122 	if (handle == NULL)
    123 		err(EXIT_FAILURE, "intrctl_io_alloc");
    124 
    125 	/* calc columns */
    126 	ncpus = intrctl_io_ncpus(handle);
    127 	intridlen = strlen("interrupt id");
    128 	for (illine = intrctl_io_firstline(handle); illine != NULL;
    129 	    illine = intrctl_io_nextline(handle, illine)) {
    130 		size_t len = strlen(illine->ill_intrid);
    131 		if (intridlen < len)
    132 			intridlen = len;
    133 	}
    134 
    135 	cpucol = malloc(sizeof(*cpucol) * (size_t)ncpus);
    136 	if (cpucol == NULL)
    137 		err(EXIT_FAILURE, "malloc");
    138 	for (i = 0; i < ncpus; i++) {
    139 		snprintf(buf, sizeof(buf), "CPU%u", i);
    140 		cpucol[i] = strlen(buf);
    141 	}
    142 	for (illine = intrctl_io_firstline(handle); illine != NULL;
    143 	    illine = intrctl_io_nextline(handle, illine)) {
    144 		for (i = 0; i < ncpus; i++) {
    145 			int len;
    146 			snprintf(buf, sizeof(buf), "%" PRIu64,
    147 			    illine->ill_cpu[i].illc_count);
    148 			len = (int)strlen(buf);
    149 			if (cpucol[i] < len)
    150 				cpucol[i] = len;
    151 		}
    152 	}
    153 
    154 	/* header */
    155 	printf("%-*s  ", (int)intridlen, "interrupt id");
    156 	for (i = 0; i < ncpus; i++) {
    157 		snprintf(buf, sizeof(buf), "CPU%u", i);
    158 		printf("%*s  ", cpucol[i], buf);
    159 	}
    160 	printf("device name(s)\n");
    161 
    162 	/* body */
    163 	for (illine = intrctl_io_firstline(handle); illine != NULL;
    164 	    illine = intrctl_io_nextline(handle, illine)) {
    165 		printf("%-*s  ", (int)intridlen, illine->ill_intrid);
    166 		for (i = 0; i < ncpus; i++) {
    167 			struct intrio_list_line_cpu *illc = &illine->ill_cpu[i];
    168 			printf("%*" PRIu64 "%c ", cpucol[i], illc->illc_count,
    169 			    illc->illc_assigned ? '*' : ' ');
    170 		}
    171 		printf("%s\n", illine->ill_xname);
    172 	}
    173 
    174 	free(cpucol);
    175 	intrctl_io_free(handle);
    176 }
    177 
    178 static void
    179 intrctl_affinity(int argc, char **argv)
    180 {
    181 	struct intrio_set iset;
    182 	cpuset_t *cpuset;
    183 	unsigned long index;
    184 	int ch, error;
    185 
    186 	index = ULONG_MAX;
    187 	memset(&iset.intrid, 0, sizeof(iset.intrid));
    188 
    189 	while ((ch = getopt(argc, argv, "c:i:")) != -1) {
    190 		switch (ch) {
    191 		case 'c':
    192 			index = strtoul(optarg, NULL, 10);
    193 			break;
    194 		case 'i':
    195 			if (strnlen(optarg, ARG_MAX) > INTRIDBUF)
    196 				usage();
    197 			strlcpy(iset.intrid, optarg, INTRIDBUF);
    198 			break;
    199 		default:
    200 			usage();
    201 		}
    202 	}
    203 
    204 	if (iset.intrid[0] == '\0' || index == ULONG_MAX)
    205 		usage();
    206 
    207 	if (index >= (u_long)sysconf(_SC_NPROCESSORS_CONF))
    208 		err(EXIT_FAILURE, "invalid cpu index");
    209 
    210 	cpuset = cpuset_create();
    211 	if (cpuset == NULL)
    212 		err(EXIT_FAILURE, "create_cpuset()");
    213 
    214 	cpuset_zero(cpuset);
    215 	cpuset_set(index, cpuset);
    216 	iset.cpuset = cpuset;
    217 	iset.cpuset_size = cpuset_size(cpuset);
    218 	error = sysctlbyname("kern.intr.affinity", NULL, NULL, &iset, sizeof(iset));
    219 	cpuset_destroy(cpuset);
    220 	if (error < 0)
    221 		err(EXIT_FAILURE, "sysctl kern.intr.affinity");
    222 }
    223 
    224 static void
    225 intrctl_intr(int argc, char **argv)
    226 {
    227 	struct intrio_set iset;
    228 	cpuset_t *cpuset;
    229 	unsigned long index;
    230 	int ch, error;
    231 
    232 	index = ULONG_MAX;
    233 
    234 	while ((ch = getopt(argc, argv, "c:")) != -1) {
    235 		switch (ch) {
    236 		case 'c':
    237 			index = strtoul(optarg, NULL, 10);
    238 			break;
    239 		default:
    240 			usage();
    241 		}
    242 	}
    243 
    244 	if (index == ULONG_MAX)
    245 		usage();
    246 
    247 	if (index >= (u_long)sysconf(_SC_NPROCESSORS_CONF))
    248 		err(EXIT_FAILURE, "invalid cpu index");
    249 
    250 	cpuset = cpuset_create();
    251 	if (cpuset == NULL)
    252 		err(EXIT_FAILURE, "create_cpuset()");
    253 
    254 	cpuset_zero(cpuset);
    255 	cpuset_set(index, cpuset);
    256 	iset.cpuset = cpuset;
    257 	iset.cpuset_size = cpuset_size(cpuset);
    258 	error = sysctlbyname("kern.intr.intr", NULL, NULL, &iset, sizeof(iset));
    259 	cpuset_destroy(cpuset);
    260 	if (error < 0)
    261 		err(EXIT_FAILURE, "sysctl kern.intr.intr");
    262 }
    263 
    264 static void
    265 intrctl_nointr(int argc, char **argv)
    266 {
    267 	struct intrio_set iset;
    268 	cpuset_t *cpuset;
    269 	unsigned long index;
    270 	int ch, error;
    271 
    272 	index = ULONG_MAX;
    273 
    274 	while ((ch = getopt(argc, argv, "c:")) != -1) {
    275 		switch (ch) {
    276 		case 'c':
    277 			index = strtoul(optarg, NULL, 10);
    278 			break;
    279 		default:
    280 			usage();
    281 		}
    282 	}
    283 
    284 	if (index == ULONG_MAX)
    285 		usage();
    286 
    287 	if (index >= (u_long)sysconf(_SC_NPROCESSORS_CONF))
    288 		err(EXIT_FAILURE, "invalid cpu index");
    289 
    290 	cpuset = cpuset_create();
    291 	if (cpuset == NULL)
    292 		err(EXIT_FAILURE, "create_cpuset()");
    293 
    294 	cpuset_zero(cpuset);
    295 	cpuset_set(index, cpuset);
    296 	iset.cpuset = cpuset;
    297 	iset.cpuset_size = cpuset_size(cpuset);
    298 	error = sysctlbyname("kern.intr.nointr", NULL, NULL, &iset, sizeof(iset));
    299 	cpuset_destroy(cpuset);
    300 	if (error < 0)
    301 		err(EXIT_FAILURE, "sysctl kern.intr.nointr");
    302 }
    303