/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2019 Joyent, Inc.
 */

#include <sys/cpuid_drv.h>

#include <libintl.h>
#include <string.h>
#include <locale.h>
#include <unistd.h>
#include <stdlib.h>
#include <fcntl.h>
#include <stdio.h>
#include <errno.h>
#include <err.h>

static const char dev_cpu_self_cpuid[] = "/dev/" CPUID_SELF_NAME;

int
main(int argc, char *argv[])
{
	int ret = EXIT_SUCCESS;
	int errflg = 0;
	int fd;
	int c;

	(void) setlocale(LC_ALL, "");
	(void) textdomain(TEXT_DOMAIN);

	while ((c = getopt(argc, argv, "")) != EOF) {
		switch (c) {
		case '?':
		default:
			errflg++;
			break;
		}
	}

	if (errflg != 0 || optind == argc) {
		fprintf(stderr, gettext("usage: rdmsr [0x<msrnr>...]\n"));
		return (EXIT_FAILURE);
	}

	if ((fd = open(dev_cpu_self_cpuid, O_RDONLY)) == -1) {
		err(EXIT_FAILURE, gettext("failed to open %s"),
		    dev_cpu_self_cpuid);
	}

	while (optind != argc) {
		struct cpuid_rdmsr crm;
		char *p;

		errno = 0;
		crm.cr_msr_nr = strtoull(argv[optind], &p, 0);

		if (errno != 0 || p == argv[optind] || *p != '\0') {
			fprintf(stderr,
			    gettext("rdmsr: invalid argument '%s'\n"),
			    argv[optind]);
			exit(EXIT_FAILURE);
		}

		if (ioctl(fd, CPUID_RDMSR, &crm) != 0) {
			warn(gettext("rdmsr of 0x%lx failed"), crm.cr_msr_nr);
			ret = EXIT_FAILURE;
		} else {
			printf("0x%lx: 0x%lx\n", crm.cr_msr_nr, crm.cr_msr_val);
		}

		optind++;
	}

	return (ret);
}