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