summaryrefslogtreecommitdiff
path: root/usr/src/cmd/nvmeadm/nvmeadm_ofmt.c
blob: 825417eb76e9f6a29b741903aca350e38b73d930 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
/*
 * 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 2021 Oxide Computer Company
 */

/*
 * nvmeadm output formatting for ofmt based rendering
 */

#include <strings.h>

#include "nvmeadm.h"

typedef enum nvme_list_ofmt_field {
	NVME_LIST_MODEL,
	NVME_LIST_SERIAL,
	NVME_LIST_FWREV,
	NVME_LIST_VERSION,
	NVME_LIST_SIZE,
	NVME_LIST_CAPACITY,
	NVME_LIST_USED,
	NVME_LIST_INSTANCE,
	NVME_LIST_NAMESPACE,
	NVME_LIST_DISK
} nvme_list_ofmt_field_t;

static boolean_t
nvme_list_ofmt_cb(ofmt_arg_t *ofmt_arg, char *buf, uint_t buflen)
{
	const nvme_process_arg_t *npa = ofmt_arg->ofmt_cbarg;
	nvme_idns_lbaf_t *lbaf;
	uint_t blksize;
	uint64_t val;
	int nvmelen;
	size_t ret;

	lbaf = &npa->npa_idns->id_lbaf[npa->npa_idns->id_flbas.lba_format];
	blksize = 1 << lbaf->lbaf_lbads;

	switch (ofmt_arg->ofmt_id) {
	case NVME_LIST_MODEL:
		nvmelen = nvme_strlen(npa->npa_idctl->id_model,
		    sizeof (npa->npa_idctl->id_model));
		if (nvmelen <= 0 || nvmelen > buflen) {
			return (B_FALSE);
		}
		(void) memcpy(buf, npa->npa_idctl->id_model, nvmelen);
		buf[nvmelen] = '\0';
		ret = nvmelen;
		break;
	case NVME_LIST_SERIAL:
		nvmelen = nvme_strlen(npa->npa_idctl->id_serial,
		    sizeof (npa->npa_idctl->id_serial));
		if (nvmelen <= 0 || nvmelen >= buflen) {
			return (B_FALSE);
		}
		(void) memcpy(buf, npa->npa_idctl->id_serial, nvmelen);
		buf[nvmelen] = '\0';
		ret = nvmelen;
		break;
	case NVME_LIST_FWREV:
		nvmelen = nvme_strlen(npa->npa_idctl->id_fwrev,
		    sizeof (npa->npa_idctl->id_fwrev));
		if (nvmelen <= 0 || nvmelen >= buflen) {
			return (B_FALSE);
		}
		(void) memcpy(buf, npa->npa_idctl->id_fwrev, nvmelen);
		buf[nvmelen] = '\0';
		ret = nvmelen;
		break;
	case NVME_LIST_VERSION:
		ret = snprintf(buf, buflen, "%u.%u", npa->npa_version->v_major,
		    npa->npa_version->v_minor);
		break;
	case NVME_LIST_INSTANCE:
		ret = strlcat(buf, npa->npa_name, buflen);
		break;
	case NVME_LIST_NAMESPACE:
		ret = strlcat(buf, di_minor_name(npa->npa_minor), buflen);
		break;
	case NVME_LIST_DISK:
		if (npa->npa_dsk != NULL) {
			ret = strlcat(buf, npa->npa_dsk, buflen);
		} else {
			ret = strlcat(buf, "--", buflen);
		}
		break;
	case NVME_LIST_SIZE:
		val = npa->npa_idns->id_nsize * blksize;
		ret = snprintf(buf, buflen, "%" PRIu64, val);
		break;
	case NVME_LIST_CAPACITY:
		val = npa->npa_idns->id_ncap * blksize;
		ret = snprintf(buf, buflen, "%" PRIu64, val);
		break;
	case NVME_LIST_USED:
		val = npa->npa_idns->id_nuse * blksize;
		ret = snprintf(buf, buflen, "%" PRIu64, val);
		break;
	default:
		abort();
	}

	if (ret >= buflen) {
		return (B_FALSE);
	}
	return (B_TRUE);
}

const ofmt_field_t nvme_list_ofmt[] = {
	{ "MODEL", 30, NVME_LIST_MODEL, nvme_list_ofmt_cb },
	{ "SERIAL", 30, NVME_LIST_SERIAL, nvme_list_ofmt_cb },
	{ "FWREV", 10, NVME_LIST_FWREV, nvme_list_ofmt_cb },
	{ "VERSION", 10, NVME_LIST_VERSION, nvme_list_ofmt_cb },
	{ "SIZE", 15, NVME_LIST_SIZE, nvme_list_ofmt_cb },
	{ "CAPACITY", 15, NVME_LIST_CAPACITY, nvme_list_ofmt_cb },
	{ "USED", 15, NVME_LIST_USED, nvme_list_ofmt_cb },
	{ "INSTANCE", 10, NVME_LIST_INSTANCE, nvme_list_ofmt_cb },
	{ "NAMESPACE", 10, NVME_LIST_NAMESPACE, nvme_list_ofmt_cb },
	{ "DISK", 15, NVME_LIST_DISK, nvme_list_ofmt_cb },
	{ NULL, 0, 0, NULL }
};