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