2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
4 * Copyright (C) 2012-2013 Intel Corporation
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 #include <sys/cdefs.h>
30 __FBSDID("$FreeBSD$");
32 #include <sys/param.h>
33 #include <sys/ioccom.h>
48 #include "nvmecontrol.h"
50 SET_DECLARE(top, struct nvme_function);
53 print_usage(const struct nvme_function *f)
57 bool need_prefix = true;
64 fputs(" nvmecontrol ", stderr);
69 need_prefix = (ch == '\n');
76 gen_usage_set(struct nvme_function **f, struct nvme_function **flimit)
79 fprintf(stderr, "usage:\n");
88 usage(const struct nvme_function *f)
91 fprintf(stderr, "usage:\n");
97 dispatch_set(int argc, char *argv[], struct nvme_function **tbl,
98 struct nvme_function **tbl_limit)
100 struct nvme_function **f = tbl;
102 if (argv[1] == NULL) {
103 gen_usage_set(tbl, tbl_limit);
107 while (f < tbl_limit) {
108 if (strcmp(argv[1], (*f)->name) == 0) {
109 (*f)->fn(*f, argc-1, &argv[1]);
115 fprintf(stderr, "Unknown command: %s\n", argv[1]);
116 gen_usage_set(tbl, tbl_limit);
120 print_bytes(void *data, uint32_t length)
125 end = (uint8_t *)data + length;
127 for (i = 0; i < length; i++) {
128 p = (uint8_t *)data + (i*16);
129 printf("%03x: ", i*16);
130 for (j = 0; j < 16 && p < end; j++)
131 printf("%02x ", *p++);
140 print_dwords(void *data, uint32_t length)
145 p = (uint32_t *)data;
146 length /= sizeof(uint32_t);
148 for (i = 0; i < length; i+=8) {
149 printf("%03x: ", i*4);
150 for (j = 0; j < 8; j++)
151 printf("%08x ", p[i+j]);
159 print_hex(void *data, uint32_t length)
161 if (length >= sizeof(uint32_t) || length % sizeof(uint32_t) == 0)
162 print_dwords(data, length);
164 print_bytes(data, length);
168 read_controller_data(int fd, struct nvme_controller_data *cdata)
170 struct nvme_pt_command pt;
172 memset(&pt, 0, sizeof(pt));
173 pt.cmd.opc = NVME_OPC_IDENTIFY;
174 pt.cmd.cdw10 = htole32(1);
176 pt.len = sizeof(*cdata);
179 if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0)
180 err(1, "identify request failed");
182 /* Convert data to host endian */
183 nvme_controller_data_swapbytes(cdata);
185 if (nvme_completion_is_error(&pt.cpl))
186 errx(1, "identify request returned error");
190 read_namespace_data(int fd, uint32_t nsid, struct nvme_namespace_data *nsdata)
192 struct nvme_pt_command pt;
194 memset(&pt, 0, sizeof(pt));
195 pt.cmd.opc = NVME_OPC_IDENTIFY;
196 pt.cmd.nsid = htole32(nsid);
198 pt.len = sizeof(*nsdata);
201 if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0)
202 err(1, "identify request failed");
204 /* Convert data to host endian */
205 nvme_namespace_data_swapbytes(nsdata);
207 if (nvme_completion_is_error(&pt.cpl))
208 errx(1, "identify request returned error");
212 open_dev(const char *str, int *fd, int show_error, int exit_on_error)
216 if (!strnstr(str, NVME_CTRLR_PREFIX, strlen(NVME_CTRLR_PREFIX))) {
218 warnx("controller/namespace ids must begin with '%s'",
226 snprintf(full_path, sizeof(full_path), _PATH_DEV"%s", str);
227 *fd = open(full_path, O_RDWR);
230 warn("could not open %s", full_path);
241 parse_ns_str(const char *ns_str, char *ctrlr_str, uint32_t *nsid)
246 * Pull the namespace id from the string. +2 skips past the "ns" part
247 * of the string. Don't search past 10 characters into the string,
248 * otherwise we know it is malformed.
250 nsloc = strnstr(ns_str, NVME_NS_PREFIX, 10);
252 *nsid = strtol(nsloc + 2, NULL, 10);
253 if (nsloc == NULL || (*nsid == 0 && errno != 0))
254 errx(1, "invalid namespace ID '%s'", ns_str);
257 * The controller string will include only the nvmX part of the
260 snprintf(ctrlr_str, nsloc - ns_str + 1, "%s", ns_str);
264 main(int argc, char *argv[])
268 gen_usage_set(SET_BEGIN(top), SET_LIMIT(top));
270 DISPATCH(argc, argv, top);