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