/*-
 * Copyright (c) 2019 Klara Inc.
 * Copyright (c) 2015 Baptiste Daroussin <bapt@FreeBSD.org>
 * Copyright (c) 2015 Allan Jude <allanjude@FreeBSD.org>
 * Copyright (c) 2000 by Matthew Jacob
 * All rights reserved.
 *
 * Portions of this software were developed by Edward Tomasz Napierala
 * under sponsorship from Klara Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer
 *    in this position and unchanged.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#include <sys/endian.h>
#include <sys/param.h>
#include <sys/disk.h>
#include <sys/ioctl.h>
#include <sys/types.h>

#include <errno.h>
#include <fcntl.h>
#include <getopt.h>
#include <glob.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <libxo/xo.h>

#include <cam/scsi/scsi_enc.h>

#include "eltsub.h"

#define SESUTIL_XO_VERSION	"1"

#define TEMPERATURE_OFFSET	20

#define PRINT_STYLE_DASHED	0
#define PRINT_STYLE_DASHED_2	1
#define PRINT_STYLE_CSV 	2
#define PRINT_STYLE_CSV_2	3

static int encstatus(int argc, char **argv);
static int fault(int argc, char **argv);
static int locate(int argc, char **argv);
static int objmap(int argc, char **argv);
static int sesled(int argc, char **argv, bool fault);
static int show(int argc, char **argv);
static void sesutil_print(int *style, const char *fmt, ...) __printflike(2,3);

static struct command {
	const char *name;
	const char *param;
	const char *desc;
	int (*exec)(int argc, char **argv);
} cmds[] = {
	{ "fault",
	    "(<disk>|<sesid>|all) (on|off)",
	    "Change the state of the fault LED associated with a disk",
	    fault },
	{ "locate",
	    "(<disk>|<sesid>|all) (on|off)",
	    "Change the state of the locate LED associated with a disk",
	    locate },
	{ "map", "",
	    "Print a map of the devices managed by the enclosure", objmap } ,
	{ "show", "",
	    "Print a human-friendly summary of the enclosure", show } ,
	{ "status", "", "Print the status of the enclosure",
	    encstatus },
};

static const int nbcmds = nitems(cmds);
static const char *uflag;

static void
usage(const char *subcmd)
{
	int i;

	if (subcmd == NULL) {
		xo_error("usage: %s [-u /dev/ses<N>] <command> [options]\n",
		    getprogname());
		xo_error("Commands supported:\n");
	}
	for (i = 0; i < nbcmds; i++) {
		if (subcmd != NULL) {
			if (strcmp(subcmd, cmds[i].name) == 0) {
				xo_error("usage: %s %s [-u /dev/ses<N>] "
				    "%s\n\t%s\n", getprogname(), subcmd,
				    cmds[i].param, cmds[i].desc);
				break;
			}
			continue;
		}
		xo_error("    %-12s%s\n\t\t%s\n\n", cmds[i].name,
		    cmds[i].param, cmds[i].desc);
	}

	exit(EXIT_FAILURE);
}

static void
do_led(int fd, unsigned int idx, elm_type_t type, bool onoff, bool setfault)
{
	int state = onoff ? 1 : 0;
	encioc_elm_status_t o;
	struct ses_ctrl_dev_slot *slot;

	o.elm_idx = idx;
	if (ioctl(fd, ENCIOC_GETELMSTAT, (caddr_t) &o) < 0) {
		close(fd);
		xo_err(EXIT_FAILURE, "ENCIOC_GETELMSTAT");
	}
	ses_status_to_ctrl(type, &o.cstat[0]);
	switch (type) {
	case ELMTYP_DEVICE:
	case ELMTYP_ARRAY_DEV:
		slot = (struct ses_ctrl_dev_slot *) &o.cstat[0];
		ses_ctrl_common_set_select(&slot->common, 1);
		if (setfault)
			ses_ctrl_dev_slot_set_rqst_fault(slot, state);
		else
			ses_ctrl_dev_slot_set_rqst_ident(slot, state);
		break;
	default:
		return;
	}
	if (ioctl(fd, ENCIOC_SETELMSTAT, (caddr_t) &o) < 0) {
		close(fd);
		xo_err(EXIT_FAILURE, "ENCIOC_SETELMSTAT");
	}
}

static bool
disk_match(const char *devnames, const char *disk, size_t len)
{
	const char *dname;

	dname = devnames;
	while ((dname = strstr(dname, disk)) != NULL) {
		if (dname[len] == '\0' || dname[len] == ',') {
			return (true);
		}
		dname++;
	}

	return (false);
}

static int
sesled(int argc, char **argv, bool setfault)
{
	encioc_elm_devnames_t objdn;
	encioc_element_t *objp;
	glob_t g;
	char *disk, *endptr;
	size_t len, i, ndisks;
	int fd;
	unsigned int nobj, j, sesid;
	bool all, isses, onoff;

	isses = false;
	all = false;
	onoff = false;

	if (argc != 3) {
		usage(setfault ? "fault" : "locate");
	}

	disk = argv[1];

	sesid = strtoul(disk, &endptr, 10);
	if (*endptr == '\0') {
		endptr = strrchr(uflag, '*');
		if (endptr != NULL && *endptr == '*') {
			xo_warnx("Must specifying a SES device (-u) to use a SES "
			    "id# to identify a disk");
			usage(setfault ? "fault" : "locate");
		}
		isses = true;
	}

	if (strcmp(argv[2], "on") == 0) {
		onoff = true;
	} else if (strcmp(argv[2], "off") == 0) {
		onoff = false;
	} else {
		usage(setfault ? "fault" : "locate");
	}

	if (strcmp(disk, "all") == 0) {
		all = true;
	}
	len = strlen(disk);

	/* Get the list of ses devices */
	if (glob((uflag != NULL ? uflag : "/dev/ses[0-9]*"), 0, NULL, &g) ==
	    GLOB_NOMATCH) {
		globfree(&g);
		xo_errx(EXIT_FAILURE, "No SES devices found");
	}

	ndisks = 0;
	for (i = 0; i < g.gl_pathc; i++) {
		/* ensure we only got numbers after ses */
		if (strspn(g.gl_pathv[i] + 8, "0123456789") !=
		    strlen(g.gl_pathv[i] + 8)) {
			continue;
		}
		if ((fd = open(g.gl_pathv[i], O_RDWR)) < 0) {
			/*
			 * Don't treat non-access errors as critical if we are
			 * accessing all devices
			 */
			if (errno == EACCES && g.gl_pathc > 1) {
				xo_err(EXIT_FAILURE, "unable to access SES device");
			}
			xo_warn("unable to access SES device: %s", g.gl_pathv[i]);
			continue;
		}

		if (ioctl(fd, ENCIOC_GETNELM, (caddr_t) &nobj) < 0) {
			close(fd);
			xo_err(EXIT_FAILURE, "ENCIOC_GETNELM");
		}

		objp = calloc(nobj, sizeof(encioc_element_t));
		if (objp == NULL) {
			close(fd);
			xo_err(EXIT_FAILURE, "calloc()");
		}

		if (ioctl(fd, ENCIOC_GETELMMAP, (caddr_t) objp) < 0) {
			free(objp);
			close(fd);
			xo_err(EXIT_FAILURE, "ENCIOC_GETELMMAP");
		}

		if (isses) {
			if (sesid >= nobj) {
				free(objp);
				close(fd);
				xo_errx(EXIT_FAILURE,
				     "Requested SES ID does not exist");
			}
			do_led(fd, sesid, objp[sesid].elm_type, onoff, setfault);
			ndisks++;
			free(objp);
			close(fd);
			break;
		}
		for (j = 0; j < nobj; j++) {
			const int devnames_size = 128;
			char devnames[devnames_size];

			if (all) {
				encioc_elm_status_t es;
				memset(&es, 0, sizeof(es));
				es.elm_idx = objp[j].elm_idx;
				if (ioctl(fd, ENCIOC_GETELMSTAT, &es) < 0) {
					close(fd);
					xo_err(EXIT_FAILURE,
						"ENCIOC_GETELMSTAT");
				}
				if ((es.cstat[0] & 0xf) == SES_OBJSTAT_NOACCESS)
					continue;
				do_led(fd, objp[j].elm_idx, objp[j].elm_type,
				    onoff, setfault);
				continue;
			}
			memset(&objdn, 0, sizeof(objdn));
			memset(devnames, 0, devnames_size);
			objdn.elm_idx = objp[j].elm_idx;
			objdn.elm_names_size = devnames_size;
			objdn.elm_devnames = devnames;
			if (ioctl(fd, ENCIOC_GETELMDEVNAMES,
			    (caddr_t) &objdn) <0) {
				continue;
			}
			if (objdn.elm_names_len > 0) {
				if (disk_match(objdn.elm_devnames, disk, len)) {
					do_led(fd, objdn.elm_idx, objp[j].elm_type,
					    onoff, setfault);
					ndisks++;
					break;
				}
			}
		}
		free(objp);
		close(fd);
	}
	globfree(&g);
	if (ndisks == 0 && all == false) {
		xo_errx(EXIT_FAILURE, "Could not find the SES id of device '%s'",
		    disk);
	}

	return (EXIT_SUCCESS);
}

static int
locate(int argc, char **argv)
{

	return (sesled(argc, argv, false));
}

static int
fault(int argc, char **argv)
{

	return (sesled(argc, argv, true));
}

static void
sesutil_print(int *style, const char *fmt, ...)
{
	va_list args;

	if (*style == PRINT_STYLE_DASHED) {
		xo_open_container("extra_status");
		xo_emit("\t\tExtra status:\n");
		*style = PRINT_STYLE_DASHED_2;
	} else if (*style == PRINT_STYLE_CSV) {
		xo_open_container("extra_status");
		*style = PRINT_STYLE_CSV_2;
	}

	if (*style == PRINT_STYLE_DASHED_2)
		xo_emit("\t\t- ");
	else if (*style == PRINT_STYLE_CSV_2)
		xo_emit(", ");
	va_start(args, fmt);
	xo_emit_hv(NULL, fmt, args);
	va_end(args);
	if (*style == PRINT_STYLE_DASHED_2)
		xo_emit("\n");
}

static void
print_extra_status(int eletype, u_char *cstat, int style)
{

	if (cstat[0] & 0x40) {
		sesutil_print(&style, "{e:predicted_failure/true} Predicted Failure");
	}
	if (cstat[0] & 0x20) {
		sesutil_print(&style, "{e:disabled/true} Disabled");
	}
	if (cstat[0] & 0x10) {
		sesutil_print(&style, "{e:swapped/true} Swapped");
	}
	switch (eletype) {
	case ELMTYP_DEVICE:
	case ELMTYP_ARRAY_DEV:
		if (cstat[2] & 0x02) {
			sesutil_print(&style, "LED={q:led/locate}");
		}
		if (cstat[2] & 0x20) {
			sesutil_print(&style, "LED={q:led/fault}");
		}
		break;
	case ELMTYP_FAN:
		sesutil_print(&style, "Speed: {:speed/%d}{Uw:rpm}",
		    (((0x7 & cstat[1]) << 8) + cstat[2]) * 10);
		break;
	case ELMTYP_THERM:
		if (cstat[2]) {
			sesutil_print(&style, "Temperature: {:temperature/%d}{Uw:C}",
			    cstat[2] - TEMPERATURE_OFFSET);
		} else {
			sesutil_print(&style, "Temperature: -{q:temperature/reserved}");
		}
		break;
	case ELMTYP_VOM:
		sesutil_print(&style, "Voltage: {:voltage/%.2f}{Uw:V}",
		    be16dec(cstat + 2) / 100.0);
		break;
	}
	if (style) {
		xo_close_container("extra_status");
	}
}

static int
objmap(int argc, char **argv __unused)
{
	encioc_string_t stri;
	encioc_elm_devnames_t e_devname;
	encioc_elm_status_t e_status;
	encioc_elm_desc_t e_desc;
	encioc_element_t *e_ptr;
	glob_t g;
	int fd;
	unsigned int j, nobj;
	size_t i;
	char str[32];

	if (argc != 1) {
		usage("map");
	}

	memset(&e_desc, 0, sizeof(e_desc));
	/* SES4r02 allows element descriptors of up to 65536 characters */
	e_desc.elm_desc_str = calloc(UINT16_MAX, sizeof(char));
	if (e_desc.elm_desc_str == NULL)
		xo_err(EXIT_FAILURE, "calloc()");

	e_devname.elm_devnames = calloc(128, sizeof(char));
	if (e_devname.elm_devnames == NULL)
		xo_err(EXIT_FAILURE, "calloc()");
	e_devname.elm_names_size = 128;

	/* Get the list of ses devices */
	if (glob(uflag, 0, NULL, &g) == GLOB_NOMATCH) {
		globfree(&g);
		xo_errx(EXIT_FAILURE, "No SES devices found");
	}
	xo_set_version(SESUTIL_XO_VERSION);
	xo_open_container("sesutil");
	xo_open_list("enclosures");
	for (i = 0; i < g.gl_pathc; i++) {
		/* ensure we only got numbers after ses */
		if (strspn(g.gl_pathv[i] + 8, "0123456789") !=
		    strlen(g.gl_pathv[i] + 8)) {
			continue;
		}
		if ((fd = open(g.gl_pathv[i], O_RDWR)) < 0) {
			/*
			 * Don't treat non-access errors as critical if we are
			 * accessing all devices
			 */
			if (errno == EACCES && g.gl_pathc > 1) {
				xo_err(EXIT_FAILURE, "unable to access SES device");
			}
			xo_warn("unable to access SES device: %s", g.gl_pathv[i]);
			continue;
		}

		if (ioctl(fd, ENCIOC_GETNELM, (caddr_t) &nobj) < 0) {
			close(fd);
			xo_err(EXIT_FAILURE, "ENCIOC_GETNELM");
		}

		e_ptr = calloc(nobj, sizeof(encioc_element_t));
		if (e_ptr == NULL) {
			close(fd);
			xo_err(EXIT_FAILURE, "calloc()");
		}

		if (ioctl(fd, ENCIOC_GETELMMAP, (caddr_t) e_ptr) < 0) {
			close(fd);
			xo_err(EXIT_FAILURE, "ENCIOC_GETELMMAP");
		}

		xo_open_instance("enclosures");
		xo_emit("{t:enc/%s}:\n", g.gl_pathv[i] + 5);
		stri.bufsiz = sizeof(str);
		stri.buf = &str[0];
		if (ioctl(fd, ENCIOC_GETENCNAME, (caddr_t) &stri) == 0)
			xo_emit("\tEnclosure Name: {t:name/%s}\n", stri.buf);
		stri.bufsiz = sizeof(str);
		stri.buf = &str[0];
		if (ioctl(fd, ENCIOC_GETENCID, (caddr_t) &stri) == 0)
			xo_emit("\tEnclosure ID: {t:id/%s}\n", stri.buf);

		xo_open_list("elements");
		for (j = 0; j < nobj; j++) {
			/* Get the status of the element */
			memset(&e_status, 0, sizeof(e_status));
			e_status.elm_idx = e_ptr[j].elm_idx;
			if (ioctl(fd, ENCIOC_GETELMSTAT,
			    (caddr_t) &e_status) < 0) {
				close(fd);
				xo_err(EXIT_FAILURE, "ENCIOC_GETELMSTAT");
			}
			/* Get the description of the element */
			e_desc.elm_idx = e_ptr[j].elm_idx;
			e_desc.elm_desc_len = UINT16_MAX;
			if (ioctl(fd, ENCIOC_GETELMDESC,
			    (caddr_t) &e_desc) < 0) {
				close(fd);
				xo_err(EXIT_FAILURE, "ENCIOC_GETELMDESC");
			}
			e_desc.elm_desc_str[e_desc.elm_desc_len] = '\0';
			/* Get the device name(s) of the element */
			e_devname.elm_idx = e_ptr[j].elm_idx;
			if (ioctl(fd, ENCIOC_GETELMDEVNAMES,
			    (caddr_t) &e_devname) <0) {
				/* Continue even if we can't look up devnames */
				e_devname.elm_devnames[0] = '\0';
			}
			xo_open_instance("elements");
			xo_emit("\tElement {:id/%u}, Type: {:type/%s}\n", e_ptr[j].elm_idx,
			    geteltnm(e_ptr[j].elm_type));
			xo_emit("\t\tStatus: {:status/%s} ({q:status_code/0x%02x 0x%02x 0x%02x 0x%02x})\n",
			    scode2ascii(e_status.cstat[0]), e_status.cstat[0],
			    e_status.cstat[1], e_status.cstat[2],
			    e_status.cstat[3]);
			if (e_desc.elm_desc_len > 0) {
				xo_emit("\t\tDescription: {:description/%s}\n",
				    e_desc.elm_desc_str);
			}
			if (e_devname.elm_names_len > 0) {
				xo_emit("\t\tDevice Names: {:device_names/%s}\n",
				    e_devname.elm_devnames);
			}
			print_extra_status(e_ptr[j].elm_type, e_status.cstat, PRINT_STYLE_DASHED);
			xo_close_instance("elements");
		}
		xo_close_list("elements");
		free(e_ptr);
		close(fd);
	}
	globfree(&g);
	free(e_devname.elm_devnames);
	free(e_desc.elm_desc_str);
	xo_close_list("enclosures");
	xo_close_container("sesutil");
	if (xo_finish() < 0)
		xo_err(EXIT_FAILURE, "stdout");

	return (EXIT_SUCCESS);
}

/*
 * Get rid of the 'passN' devices, unless there's nothing else to show.
 */
static void
skip_pass_devices(char *devnames, size_t devnameslen)
{
	char *dev, devs[128], passes[128], *tmp;

	devs[0] = passes[0] = '\0';
	tmp = devnames;

	while ((dev = strsep(&tmp, ",")) != NULL) {
		if (strncmp(dev, "pass", 4) == 0) {
			if (passes[0] != '\0')
				strlcat(passes, ",", sizeof(passes));
			strlcat(passes, dev, sizeof(passes));
		} else {
			if (devs[0] != '\0')
				strlcat(devs, ",", sizeof(devs));
			strlcat(devs, dev, sizeof(devs));
		}
	}
	strlcpy(devnames, devs, devnameslen);
	if (devnames[0] == '\0')
		strlcpy(devnames, passes, devnameslen);
}

static void
fetch_device_details(char *devnames, char **model, char **serial, off_t *size)
{
	char ident[DISK_IDENT_SIZE];
	struct diocgattr_arg arg;
	char *tmp;
	off_t mediasize;
	int comma;
	int fd;

	comma = (int)strcspn(devnames, ",");
	asprintf(&tmp, "/dev/%.*s", comma, devnames);
	if (tmp == NULL)
		xo_err(EXIT_FAILURE, "asprintf");
	fd = open(tmp, O_RDONLY);
	free(tmp);
	if (fd < 0) {
		/*
		 * This can happen with a disk so broken it cannot
		 * be probed by GEOM.
		 */
		*model = strdup("?");
		*serial = strdup("?");
		*size = -1;
		close(fd);
		return;
	}

	strlcpy(arg.name, "GEOM::descr", sizeof(arg.name));
	arg.len = sizeof(arg.value.str);
	if (ioctl(fd, DIOCGATTR, &arg) == 0)
		*model = strdup(arg.value.str);
	else
		*model = NULL;

	if (ioctl(fd, DIOCGIDENT, ident) == 0)
		*serial = strdup(ident);
	else
		*serial = NULL;

	if (ioctl(fd, DIOCGMEDIASIZE, &mediasize) == 0)
		*size = mediasize;
	else
		*size = -1;
	close(fd);
}

static void
show_device(int fd, int elm_idx, encioc_elm_status_t e_status, encioc_elm_desc_t e_desc)
{
	encioc_elm_devnames_t e_devname;
	char *model = NULL, *serial = NULL;
	off_t size;

	/* Get the device name(s) of the element */
	memset(&e_devname, 0, sizeof(e_devname));
	e_devname.elm_idx = elm_idx;
	e_devname.elm_names_size = 128;
	e_devname.elm_devnames = calloc(128, sizeof(char));
	if (e_devname.elm_devnames == NULL) {
		close(fd);
		xo_err(EXIT_FAILURE, "calloc()");
	}

	if (ioctl(fd, ENCIOC_GETELMDEVNAMES,
	    (caddr_t) &e_devname) < 0) {
		/* We don't care if this fails */
		e_devname.elm_devnames[0] = '\0';
		size = -1;
	} else {
		skip_pass_devices(e_devname.elm_devnames, 128);
		fetch_device_details(e_devname.elm_devnames, &model, &serial, &size);
	}
	xo_open_instance("elements");
	xo_emit("{e:type/device_slot}");
	xo_emit("{d:description/%-15s} ", e_desc.elm_desc_len > 0 ? e_desc.elm_desc_str : "-");
	xo_emit("{e:description/%-15s}", e_desc.elm_desc_len > 0 ? e_desc.elm_desc_str : "");
	xo_emit("{d:device_names/%-7s} ", e_devname.elm_names_len > 0 ? e_devname.elm_devnames : "-");
	xo_emit("{e:device_names/%s}", e_devname.elm_names_len > 0 ? e_devname.elm_devnames : "");
	xo_emit("{d:model/%-25s} ", model ? model : "-");
	xo_emit("{e:model/%s}", model ? model : "");
	xo_emit("{d:serial/%-20s} ", serial != NULL ? serial : "-");
	xo_emit("{e:serial/%s}", serial != NULL ? serial : "");
	if ((e_status.cstat[0] & 0xf) == SES_OBJSTAT_OK && size >= 0) {
		xo_emit("{h,hn-1000:size/%ld}{e:status/%s}",
		    size, scode2ascii(e_status.cstat[0]));
	} else {
		xo_emit("{:status/%s}", scode2ascii(e_status.cstat[0]));
	}
	print_extra_status(ELMTYP_ARRAY_DEV, e_status.cstat, PRINT_STYLE_CSV);
	xo_emit("\n");
	xo_close_instance("elements");
	free(serial);
	free(model);
	free(e_devname.elm_devnames);
}

static void
show_therm(encioc_elm_status_t e_status, encioc_elm_desc_t e_desc)
{

	if (e_desc.elm_desc_len <= 0) {
		/* We don't have a label to display; might as well skip it. */
		return;
	}

	if (e_status.cstat[2] == 0) {
		/* No temperature to show. */
		return;
	}

	xo_open_instance("elements");
	xo_emit("{e:type/temperature_sensor}");
	xo_emit("{:description/%s}: {:temperature/%d}{Uw:C}",
	    e_desc.elm_desc_str, e_status.cstat[2] - TEMPERATURE_OFFSET);
	xo_close_instance("elements");
}

static void
show_vom(encioc_elm_status_t e_status, encioc_elm_desc_t e_desc)
{

	if (e_desc.elm_desc_len <= 0) {
		/* We don't have a label to display; might as well skip it. */
		return;
	}

	if (e_status.cstat[2] == 0) {
		/* No voltage to show. */
		return;
	}

	xo_open_instance("elements");
	xo_emit("{e:type/voltage_sensor}");
	xo_emit("{:description/%s}: {:voltage/%.2f}{Uw:V}",
	    e_desc.elm_desc_str, be16dec(e_status.cstat + 2) / 100.0);
	xo_close_instance("elements");
}

static int
show(int argc, char **argv __unused)
{
	encioc_string_t stri;
	encioc_elm_status_t e_status;
	encioc_elm_desc_t e_desc;
	encioc_element_t *e_ptr;
	glob_t g;
	elm_type_t prev_type;
	int fd;
	unsigned int j, nobj;
	size_t i;
	bool first_ses;
	char str[32];

	if (argc != 1) {
		usage("map");
	}

	first_ses = true;

	e_desc.elm_desc_str = calloc(UINT16_MAX, sizeof(char));
	if (e_desc.elm_desc_str == NULL)
		xo_err(EXIT_FAILURE, "calloc()");

	/* Get the list of ses devices */
	if (glob(uflag, 0, NULL, &g) == GLOB_NOMATCH) {
		globfree(&g);
		xo_errx(EXIT_FAILURE, "No SES devices found");
	}
	xo_set_version(SESUTIL_XO_VERSION);
	xo_open_container("sesutil");
	xo_open_list("enclosures");
	for (i = 0; i < g.gl_pathc; i++) {
		/* ensure we only got numbers after ses */
		if (strspn(g.gl_pathv[i] + 8, "0123456789") !=
		    strlen(g.gl_pathv[i] + 8)) {
			continue;
		}
		if ((fd = open(g.gl_pathv[i], O_RDWR)) < 0) {
			/*
			 * Don't treat non-access errors as critical if we are
			 * accessing all devices
			 */
			if (errno == EACCES && g.gl_pathc > 1) {
				xo_err(EXIT_FAILURE, "unable to access SES device");
			}
			xo_warn("unable to access SES device: %s", g.gl_pathv[i]);
			continue;
		}

		if (ioctl(fd, ENCIOC_GETNELM, (caddr_t) &nobj) < 0) {
			close(fd);
			xo_err(EXIT_FAILURE, "ENCIOC_GETNELM");
		}

		e_ptr = calloc(nobj, sizeof(encioc_element_t));
		if (e_ptr == NULL) {
			close(fd);
			xo_err(EXIT_FAILURE, "calloc()");
		}

		if (ioctl(fd, ENCIOC_GETELMMAP, (caddr_t) e_ptr) < 0) {
			close(fd);
			xo_err(EXIT_FAILURE, "ENCIOC_GETELMMAP");
		}

		xo_open_instance("enclosures");

		if (first_ses)
			first_ses = false;
		else
			xo_emit("\n");

		xo_emit("{t:enc/%s}: ", g.gl_pathv[i] + 5);
		stri.bufsiz = sizeof(str);
		stri.buf = &str[0];
		if (ioctl(fd, ENCIOC_GETENCNAME, (caddr_t) &stri) == 0)
			xo_emit("<{t:name/%s}>; ", stri.buf);
		stri.bufsiz = sizeof(str);
		stri.buf = &str[0];
		if (ioctl(fd, ENCIOC_GETENCID, (caddr_t) &stri) == 0)
			xo_emit("ID: {t:id/%s}", stri.buf);
		xo_emit("\n");

		xo_open_list("elements");
		prev_type = -1;
		for (j = 0; j < nobj; j++) {
			/* Get the status of the element */
			memset(&e_status, 0, sizeof(e_status));
			e_status.elm_idx = e_ptr[j].elm_idx;
			if (ioctl(fd, ENCIOC_GETELMSTAT,
			    (caddr_t) &e_status) < 0) {
				close(fd);
				xo_err(EXIT_FAILURE, "ENCIOC_GETELMSTAT");
			}

			/*
			 * Skip "Unsupported" elements; those usually precede
			 * the actual device entries and are not particularly
			 * interesting.
			 */
			if (e_status.cstat[0] == SES_OBJSTAT_UNSUPPORTED)
				continue;

			/* Get the description of the element */
			e_desc.elm_idx = e_ptr[j].elm_idx;
			e_desc.elm_desc_len = UINT16_MAX;
			if (ioctl(fd, ENCIOC_GETELMDESC,
			    (caddr_t) &e_desc) < 0) {
				close(fd);
				xo_err(EXIT_FAILURE, "ENCIOC_GETELMDESC");
			}
			e_desc.elm_desc_str[e_desc.elm_desc_len] = '\0';

			switch (e_ptr[j].elm_type) {
			case ELMTYP_DEVICE:
			case ELMTYP_ARRAY_DEV:
				if (e_ptr[j].elm_type != prev_type)
					xo_emit("Desc            Dev     Model                     Ident                Size/Status\n");

				show_device(fd, e_ptr[j].elm_idx, e_status, e_desc);
				prev_type = e_ptr[j].elm_type;
				break;
			case ELMTYP_THERM:
				if (e_ptr[j].elm_type != prev_type)
					xo_emit("\nTemperatures: ");
				else
					xo_emit(", ");
				prev_type = e_ptr[j].elm_type;
				show_therm(e_status, e_desc);
				break;
			case ELMTYP_VOM:
				if (e_ptr[j].elm_type != prev_type)
					xo_emit("\nVoltages: ");
				else
					xo_emit(", ");
				prev_type = e_ptr[j].elm_type;
				show_vom(e_status, e_desc);
				break;
			default:
				/*
				 * Ignore stuff not interesting to the user.
				 */
				break;
			}
		}
		if (prev_type != (elm_type_t)-1 &&
		    prev_type != ELMTYP_DEVICE && prev_type != ELMTYP_ARRAY_DEV)
			xo_emit("\n");
		xo_close_list("elements");
		free(e_ptr);
		close(fd);
	}
	globfree(&g);
	free(e_desc.elm_desc_str);
	xo_close_list("enclosures");
	xo_close_container("sesutil");
	if (xo_finish() < 0)
		xo_err(EXIT_FAILURE, "stdout");

	return (EXIT_SUCCESS);
}

static int
encstatus(int argc, char **argv __unused)
{
	glob_t g;
	int fd, status;
	size_t i, e;
	u_char estat;

	status = 0;
	if (argc != 1) {
		usage("status");
	}

	/* Get the list of ses devices */
	if (glob(uflag, 0, NULL, &g) == GLOB_NOMATCH) {
		globfree(&g);
		xo_errx(EXIT_FAILURE, "No SES devices found");
	}

	xo_set_version(SESUTIL_XO_VERSION);
	xo_open_container("sesutil");
	xo_open_list("enclosures");
	for (i = 0; i < g.gl_pathc; i++) {
		/* ensure we only got numbers after ses */
		if (strspn(g.gl_pathv[i] + 8, "0123456789") !=
		    strlen(g.gl_pathv[i] + 8)) {
			continue;
		}
		if ((fd = open(g.gl_pathv[i], O_RDWR)) < 0) {
			/*
			 * Don't treat non-access errors as critical if we are
			 * accessing all devices
			 */
			if (errno == EACCES && g.gl_pathc > 1) {
				xo_err(EXIT_FAILURE, "unable to access SES device");
			}
			xo_warn("unable to access SES device: %s", g.gl_pathv[i]);
			continue;
		}

		if (ioctl(fd, ENCIOC_GETENCSTAT, (caddr_t) &estat) < 0) {
			xo_err(EXIT_FAILURE, "ENCIOC_GETENCSTAT");
			close(fd);
		}

		xo_open_instance("enclosures");
		xo_emit("{:enc/%s}: ", g.gl_pathv[i] + 5);
		e = 0;
		if (estat == 0) {
			if (status == 0) {
				status = 1;
			}
			xo_emit("{q:status/OK}");
		} else {
			if (estat & SES_ENCSTAT_INFO) {
				xo_emit("{lq:status/INFO}");
				e++;
			}
			if (estat & SES_ENCSTAT_NONCRITICAL) {
				if (e)
					xo_emit(",");
				xo_emit("{lq:status/NONCRITICAL}");
				e++;
			}
			if (estat & SES_ENCSTAT_CRITICAL) {
				if (e)
					xo_emit(",");
				xo_emit("{lq:status/CRITICAL}");
				e++;
				status = -1;
			}
			if (estat & SES_ENCSTAT_UNRECOV) {
				if (e)
					xo_emit(",");
				xo_emit("{lq:status/UNRECOV}");
				e++;
				status = -1;
			}
		}
		xo_close_instance("enclosures");
		xo_emit("\n");
		close(fd);
	}
	globfree(&g);

	xo_close_list("enclosures");
	xo_close_container("sesutil");
	if (xo_finish() < 0)
		xo_err(EXIT_FAILURE, "stdout");

	if (status == 1) {
		return (EXIT_SUCCESS);
	} else {
		return (EXIT_FAILURE);
	}
}

int
main(int argc, char **argv)
{
	int i, ch;
	struct command *cmd = NULL;

	argc = xo_parse_args(argc, argv);
	if (argc < 0)
		exit(EXIT_FAILURE);

	uflag = "/dev/ses[0-9]*";
	while ((ch = getopt_long(argc, argv, "u:", NULL, NULL)) != -1) {
		switch (ch) {
		case 'u':
			uflag = optarg;
			break;
		case '?':
		default:
			usage(NULL);
		}
	}
	argc -= optind;
	argv += optind;

	if (argc < 1) {
		xo_warnx("Missing command");
		usage(NULL);
	}

	for (i = 0; i < nbcmds; i++) {
		if (strcmp(argv[0], cmds[i].name) == 0) {
			cmd = &cmds[i];
			break;
		}
	}

	if (cmd == NULL) {
		xo_warnx("unknown command %s", argv[0]);
		usage(NULL);
	}

	return (cmd->exec(argc, argv));
}
