nvmmctl.c revision 1.1.2.2 1 1.1.2.2 martin /* $NetBSD: nvmmctl.c,v 1.1.2.2 2019/11/10 12:58:29 martin Exp $ */
2 1.1.2.2 martin
3 1.1.2.2 martin /*
4 1.1.2.2 martin * Copyright (c) 2019 The NetBSD Foundation, Inc.
5 1.1.2.2 martin * All rights reserved.
6 1.1.2.2 martin *
7 1.1.2.2 martin * This code is derived from software contributed to The NetBSD Foundation
8 1.1.2.2 martin * by Maxime Villard.
9 1.1.2.2 martin *
10 1.1.2.2 martin * Redistribution and use in source and binary forms, with or without
11 1.1.2.2 martin * modification, are permitted provided that the following conditions
12 1.1.2.2 martin * are met:
13 1.1.2.2 martin * 1. Redistributions of source code must retain the above copyright
14 1.1.2.2 martin * notice, this list of conditions and the following disclaimer.
15 1.1.2.2 martin * 2. Redistributions in binary form must reproduce the above copyright
16 1.1.2.2 martin * notice, this list of conditions and the following disclaimer in the
17 1.1.2.2 martin * documentation and/or other materials provided with the distribution.
18 1.1.2.2 martin *
19 1.1.2.2 martin * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 1.1.2.2 martin * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 1.1.2.2 martin * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 1.1.2.2 martin * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 1.1.2.2 martin * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 1.1.2.2 martin * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 1.1.2.2 martin * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 1.1.2.2 martin * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 1.1.2.2 martin * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 1.1.2.2 martin * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 1.1.2.2 martin * POSSIBILITY OF SUCH DAMAGE.
30 1.1.2.2 martin */
31 1.1.2.2 martin
32 1.1.2.2 martin #include <sys/cdefs.h>
33 1.1.2.2 martin #ifndef lint
34 1.1.2.2 martin __RCSID("$NetBSD: nvmmctl.c,v 1.1.2.2 2019/11/10 12:58:29 martin Exp $");
35 1.1.2.2 martin #endif /* not lint */
36 1.1.2.2 martin
37 1.1.2.2 martin #include <sys/param.h>
38 1.1.2.2 martin
39 1.1.2.2 martin #include <err.h>
40 1.1.2.2 martin #include <errno.h>
41 1.1.2.2 martin #include <fcntl.h>
42 1.1.2.2 martin #include <stdio.h>
43 1.1.2.2 martin #include <stdlib.h>
44 1.1.2.2 martin #include <stdarg.h>
45 1.1.2.2 martin #include <stdbool.h>
46 1.1.2.2 martin #include <string.h>
47 1.1.2.2 martin #include <unistd.h>
48 1.1.2.2 martin #include <time.h>
49 1.1.2.2 martin #include <util.h>
50 1.1.2.2 martin #include <nvmm.h>
51 1.1.2.2 martin
52 1.1.2.2 martin #include <x86/specialreg.h>
53 1.1.2.2 martin
54 1.1.2.2 martin __dead static void usage(void);
55 1.1.2.2 martin static void nvmm_identify(char **);
56 1.1.2.2 martin static void nvmm_list(char **);
57 1.1.2.2 martin
58 1.1.2.2 martin static struct cmdtab {
59 1.1.2.2 martin const char *label;
60 1.1.2.2 martin bool takesargs;
61 1.1.2.2 martin bool argsoptional;
62 1.1.2.2 martin void (*func)(char **);
63 1.1.2.2 martin } const nvmm_cmdtab[] = {
64 1.1.2.2 martin { "identify", false, false, nvmm_identify },
65 1.1.2.2 martin { "list", false, false, nvmm_list },
66 1.1.2.2 martin { NULL, false, false, NULL },
67 1.1.2.2 martin };
68 1.1.2.2 martin
69 1.1.2.2 martin static struct nvmm_capability cap;
70 1.1.2.2 martin
71 1.1.2.2 martin int
72 1.1.2.2 martin main(int argc, char **argv)
73 1.1.2.2 martin {
74 1.1.2.2 martin const struct cmdtab *ct;
75 1.1.2.2 martin
76 1.1.2.2 martin argc -= 1;
77 1.1.2.2 martin argv += 1;
78 1.1.2.2 martin if (argc < 1)
79 1.1.2.2 martin usage();
80 1.1.2.2 martin
81 1.1.2.2 martin for (ct = nvmm_cmdtab; ct->label != NULL; ct++) {
82 1.1.2.2 martin if (strcmp(argv[0], ct->label) == 0) {
83 1.1.2.2 martin if (!ct->argsoptional &&
84 1.1.2.2 martin ((ct->takesargs == 0) ^ (argv[1] == NULL)))
85 1.1.2.2 martin {
86 1.1.2.2 martin usage();
87 1.1.2.2 martin }
88 1.1.2.2 martin (*ct->func)(argv + 1);
89 1.1.2.2 martin break;
90 1.1.2.2 martin }
91 1.1.2.2 martin }
92 1.1.2.2 martin
93 1.1.2.2 martin if (ct->label == NULL)
94 1.1.2.2 martin errx(EXIT_FAILURE, "unknown command ``%s''", argv[0]);
95 1.1.2.2 martin
96 1.1.2.2 martin exit(EXIT_SUCCESS);
97 1.1.2.2 martin /* NOTREACHED */
98 1.1.2.2 martin }
99 1.1.2.2 martin
100 1.1.2.2 martin static void
101 1.1.2.2 martin usage(void)
102 1.1.2.2 martin {
103 1.1.2.2 martin const char *progname = getprogname();
104 1.1.2.2 martin
105 1.1.2.2 martin fprintf(stderr, "usage: %s identify\n", progname);
106 1.1.2.2 martin fprintf(stderr, " %s list\n", progname);
107 1.1.2.2 martin exit(EXIT_FAILURE);
108 1.1.2.2 martin /* NOTREACHED */
109 1.1.2.2 martin }
110 1.1.2.2 martin
111 1.1.2.2 martin #define MACH_CONF_FLAGS "\20"
112 1.1.2.2 martin #define VCPU_CONF_FLAGS "\20" "\1" "CPUID" "\2" "TPR"
113 1.1.2.2 martin
114 1.1.2.2 martin static void
115 1.1.2.2 martin nvmm_identify(char **argv)
116 1.1.2.2 martin {
117 1.1.2.2 martin char buf[256], ram[4+1];
118 1.1.2.2 martin
119 1.1.2.2 martin if (nvmm_init() == -1)
120 1.1.2.2 martin err(EXIT_FAILURE, "nvmm_init failed");
121 1.1.2.2 martin if (nvmm_capability(&cap) == -1)
122 1.1.2.2 martin err(EXIT_FAILURE, "nvmm_capability failed");
123 1.1.2.2 martin
124 1.1.2.2 martin printf("nvmm: Kernel API version %u\n", cap.version);
125 1.1.2.2 martin printf("nvmm: State size %u\n", cap.state_size);
126 1.1.2.2 martin printf("nvmm: Max machines %u\n", cap.max_machines);
127 1.1.2.2 martin printf("nvmm: Max VCPUs per machine %u\n", cap.max_vcpus);
128 1.1.2.2 martin
129 1.1.2.2 martin if (humanize_number(ram, sizeof(ram), cap.max_ram, NULL, HN_AUTOSCALE,
130 1.1.2.2 martin (HN_DECIMAL | HN_B | HN_NOSPACE)) == -1)
131 1.1.2.2 martin err(EXIT_FAILURE, "humanize_number");
132 1.1.2.2 martin printf("nvmm: Max RAM per machine %s\n", ram);
133 1.1.2.2 martin
134 1.1.2.2 martin snprintb(buf, sizeof(buf), MACH_CONF_FLAGS, cap.arch.mach_conf_support);
135 1.1.2.2 martin printf("nvmm: Arch Mach conf %s\n", buf);
136 1.1.2.2 martin
137 1.1.2.2 martin snprintb(buf, sizeof(buf), VCPU_CONF_FLAGS, cap.arch.vcpu_conf_support);
138 1.1.2.2 martin printf("nvmm: Arch VCPU conf %s\n", buf);
139 1.1.2.2 martin
140 1.1.2.2 martin snprintb(buf, sizeof(buf), XCR0_FLAGS1, cap.arch.xcr0_mask);
141 1.1.2.2 martin printf("nvmm: Guest FPU states %s\n", buf);
142 1.1.2.2 martin }
143 1.1.2.2 martin
144 1.1.2.2 martin static void
145 1.1.2.2 martin nvmm_list(char **argv)
146 1.1.2.2 martin {
147 1.1.2.2 martin struct nvmm_ctl_mach_info machinfo;
148 1.1.2.2 martin char ram[4+1], *ts;
149 1.1.2.2 martin size_t i;
150 1.1.2.2 martin int ret;
151 1.1.2.2 martin
152 1.1.2.2 martin if (nvmm_root_init() == -1)
153 1.1.2.2 martin err(EXIT_FAILURE, "nvmm_root_init failed");
154 1.1.2.2 martin if (nvmm_capability(&cap) == -1)
155 1.1.2.2 martin err(EXIT_FAILURE, "nvmm_capability failed");
156 1.1.2.2 martin
157 1.1.2.2 martin printf(
158 1.1.2.2 martin "Machine ID VCPUs RAM Owner PID Creation Time \n"
159 1.1.2.2 martin "---------- ----- ---- --------- ------------------------\n");
160 1.1.2.2 martin
161 1.1.2.2 martin for (i = 0; i < cap.max_machines; i++) {
162 1.1.2.2 martin machinfo.machid = i;
163 1.1.2.2 martin ret = nvmm_ctl(NVMM_CTL_MACH_INFO, &machinfo, sizeof(machinfo));
164 1.1.2.2 martin if (ret == -1) {
165 1.1.2.2 martin if (errno == ENOENT)
166 1.1.2.2 martin continue;
167 1.1.2.2 martin err(EXIT_FAILURE, "nvmm_ctl failed");
168 1.1.2.2 martin }
169 1.1.2.2 martin
170 1.1.2.2 martin ts = asctime(localtime(&machinfo.time));
171 1.1.2.2 martin ts[strlen(ts) - 1] = '\0';
172 1.1.2.2 martin
173 1.1.2.2 martin if (humanize_number(ram, sizeof(ram), machinfo.nram, NULL,
174 1.1.2.2 martin HN_AUTOSCALE, (HN_DECIMAL | HN_B | HN_NOSPACE)) == -1)
175 1.1.2.2 martin err(EXIT_FAILURE, "humanize_number");
176 1.1.2.2 martin
177 1.1.2.2 martin printf("%-10zu %-5u %-4s %-9d %s\n", i, machinfo.nvcpus, ram,
178 1.1.2.2 martin machinfo.pid, ts);
179 1.1.2.2 martin }
180 1.1.2.2 martin }
181