Index: head/sbin/nvmecontrol/resv.c =================================================================== --- head/sbin/nvmecontrol/resv.c (revision 350608) +++ head/sbin/nvmecontrol/resv.c (revision 350609) @@ -1,442 +1,442 @@ /*- * SPDX-License-Identifier: BSD-2-Clause-FreeBSD * * Copyright (C) 2019 Alexander Motin * * 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, * without modification, immediately at the beginning of the file. * 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 ``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 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 __FBSDID("$FreeBSD$"); #include #include #include #include #include #include #include #include #include #include #include "nvmecontrol.h" /* Tables for command line parsing */ static cmd_fn_t resv; static cmd_fn_t resvacquire; static cmd_fn_t resvregister; static cmd_fn_t resvrelease; static cmd_fn_t resvreport; #define NONE 0xffffffffu #define NONE64 0xffffffffffffffffull #define OPT(l, s, t, opt, addr, desc) { l, s, t, &opt.addr, desc } #define OPT_END { NULL, 0, arg_none, NULL, NULL } static struct cmd resv_cmd = { .name = "resv", .fn = resv, .descr = "Reservation commands", .ctx_size = 0, .opts = NULL, .args = NULL, }; CMD_COMMAND(resv_cmd); static struct acquire_options { uint64_t crkey; uint64_t prkey; uint8_t rtype; uint8_t racqa; const char *dev; } acquire_opt = { .crkey = 0, .prkey = 0, .rtype = 0, .racqa = 0, .dev = NULL, }; static const struct opts acquire_opts[] = { OPT("crkey", 'c', arg_uint64, acquire_opt, crkey, "Current Reservation Key"), OPT("prkey", 'p', arg_uint64, acquire_opt, prkey, "Preempt Reservation Key"), OPT("rtype", 't', arg_uint8, acquire_opt, rtype, "Reservation Type"), OPT("racqa", 'a', arg_uint8, acquire_opt, racqa, "Acquire Action (0=acq, 1=pre, 2=pre+ab)"), { NULL, 0, arg_none, NULL, NULL } }; static const struct args acquire_args[] = { { arg_string, &acquire_opt.dev, "namespace-id" }, { arg_none, NULL, NULL }, }; static struct cmd acquire_cmd = { .name = "acquire", .fn = resvacquire, .descr = "Acquire/preempt reservation", .ctx_size = sizeof(acquire_opt), .opts = acquire_opts, .args = acquire_args, }; CMD_SUBCOMMAND(resv_cmd, acquire_cmd); static struct register_options { uint64_t crkey; uint64_t nrkey; uint8_t rrega; bool iekey; uint8_t cptpl; const char *dev; } register_opt = { .crkey = 0, .nrkey = 0, .rrega = 0, .iekey = false, .cptpl = 0, .dev = NULL, }; static const struct opts register_opts[] = { OPT("crkey", 'c', arg_uint64, register_opt, crkey, "Current Reservation Key"), OPT("nrkey", 'k', arg_uint64, register_opt, nrkey, "New Reservation Key"), OPT("rrega", 'r', arg_uint8, register_opt, rrega, "Register Action (0=reg, 1=unreg, 2=replace)"), OPT("iekey", 'i', arg_none, register_opt, iekey, "Ignore Existing Key"), OPT("cptpl", 'p', arg_uint8, register_opt, cptpl, "Change Persist Through Power Loss State"), { NULL, 0, arg_none, NULL, NULL } }; static const struct args register_args[] = { { arg_string, ®ister_opt.dev, "namespace-id" }, { arg_none, NULL, NULL }, }; static struct cmd register_cmd = { .name = "register", .fn = resvregister, .descr = "Register/unregister reservation", .ctx_size = sizeof(register_opt), .opts = register_opts, .args = register_args, }; CMD_SUBCOMMAND(resv_cmd, register_cmd); static struct release_options { uint64_t crkey; uint8_t rtype; uint8_t rrela; const char *dev; } release_opt = { .crkey = 0, .rtype = 0, .rrela = 0, .dev = NULL, }; static const struct opts release_opts[] = { OPT("crkey", 'c', arg_uint64, release_opt, crkey, "Current Reservation Key"), OPT("rtype", 't', arg_uint8, release_opt, rtype, "Reservation Type"), OPT("rrela", 'a', arg_uint8, release_opt, rrela, "Release Action (0=release, 1=clear)"), { NULL, 0, arg_none, NULL, NULL } }; static const struct args release_args[] = { { arg_string, &release_opt.dev, "namespace-id" }, { arg_none, NULL, NULL }, }; static struct cmd release_cmd = { .name = "release", .fn = resvrelease, .descr = "Release/clear reservation", .ctx_size = sizeof(release_opt), .opts = release_opts, .args = release_args, }; CMD_SUBCOMMAND(resv_cmd, release_cmd); static struct report_options { bool hex; bool verbose; bool eds; const char *dev; } report_opt = { .hex = false, .verbose = false, .eds = false, .dev = NULL, }; static const struct opts report_opts[] = { OPT("hex", 'x', arg_none, report_opt, hex, "Print reservation status in hex"), OPT("verbose", 'v', arg_none, report_opt, verbose, "More verbosity"), OPT("eds", 'e', arg_none, report_opt, eds, "Extended Data Structure"), { NULL, 0, arg_none, NULL, NULL } }; static const struct args report_args[] = { { arg_string, &report_opt.dev, "namespace-id" }, { arg_none, NULL, NULL }, }; static struct cmd report_cmd = { .name = "report", .fn = resvreport, .descr = "Print reservation status", .ctx_size = sizeof(report_opt), .opts = report_opts, .args = report_args, }; CMD_SUBCOMMAND(resv_cmd, report_cmd); /* handles NVME_OPC_RESERVATION_* NVM commands */ static void resvacquire(const struct cmd *f, int argc, char *argv[]) { struct nvme_pt_command pt; uint64_t data[2]; int fd; uint32_t nsid; if (arg_parse(argc, argv, f)) return; open_dev(acquire_opt.dev, &fd, 1, 1); get_nsid(fd, NULL, &nsid); if (nsid == 0) { fprintf(stderr, "This command require namespace-id\n"); arg_help(argc, argv, f); } data[0] = htole64(acquire_opt.crkey); data[1] = htole64(acquire_opt.prkey); memset(&pt, 0, sizeof(pt)); pt.cmd.opc = NVME_OPC_RESERVATION_ACQUIRE; pt.cmd.cdw10 = htole32((acquire_opt.racqa & 7) | (acquire_opt.rtype << 8)); pt.buf = &data; pt.len = sizeof(data); pt.is_read = 0; if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0) err(1, "acquire request failed"); if (nvme_completion_is_error(&pt.cpl)) errx(1, "acquire request returned error"); close(fd); exit(0); } static void resvregister(const struct cmd *f, int argc, char *argv[]) { struct nvme_pt_command pt; uint64_t data[2]; int fd; uint32_t nsid; if (arg_parse(argc, argv, f)) return; open_dev(register_opt.dev, &fd, 1, 1); get_nsid(fd, NULL, &nsid); if (nsid == 0) { fprintf(stderr, "This command require namespace-id\n"); arg_help(argc, argv, f); } data[0] = htole64(register_opt.crkey); data[1] = htole64(register_opt.nrkey); memset(&pt, 0, sizeof(pt)); pt.cmd.opc = NVME_OPC_RESERVATION_REGISTER; pt.cmd.cdw10 = htole32((register_opt.rrega & 7) | (register_opt.iekey << 3) | (register_opt.cptpl << 30)); pt.buf = &data; pt.len = sizeof(data); pt.is_read = 0; if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0) err(1, "register request failed"); if (nvme_completion_is_error(&pt.cpl)) errx(1, "register request returned error"); close(fd); exit(0); } static void resvrelease(const struct cmd *f, int argc, char *argv[]) { struct nvme_pt_command pt; uint64_t data[1]; int fd; uint32_t nsid; if (arg_parse(argc, argv, f)) return; open_dev(release_opt.dev, &fd, 1, 1); get_nsid(fd, NULL, &nsid); if (nsid == 0) { fprintf(stderr, "This command require namespace-id\n"); arg_help(argc, argv, f); } data[0] = htole64(release_opt.crkey); memset(&pt, 0, sizeof(pt)); pt.cmd.opc = NVME_OPC_RESERVATION_RELEASE; pt.cmd.cdw10 = htole32((release_opt.rrela & 7) | (release_opt.rtype << 8)); pt.buf = &data; pt.len = sizeof(data); pt.is_read = 0; if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0) err(1, "release request failed"); if (nvme_completion_is_error(&pt.cpl)) errx(1, "release request returned error"); close(fd); exit(0); } static void resvreport(const struct cmd *f, int argc, char *argv[]) { struct nvme_pt_command pt; struct nvme_resv_status *s; struct nvme_resv_status_ext *e; - uint8_t data[4096]; + uint8_t data[4096] __aligned(4); int fd; u_int i, n; uint32_t nsid; if (arg_parse(argc, argv, f)) return; open_dev(report_opt.dev, &fd, 1, 1); get_nsid(fd, NULL, &nsid); if (nsid == 0) { fprintf(stderr, "This command require namespace-id\n"); arg_help(argc, argv, f); } bzero(data, sizeof(data)); memset(&pt, 0, sizeof(pt)); pt.cmd.opc = NVME_OPC_RESERVATION_REPORT; pt.cmd.cdw10 = htole32(sizeof(data) / 4 - 1); pt.cmd.cdw11 = htole32(report_opt.eds); /* EDS */ pt.buf = &data; pt.len = sizeof(data); pt.is_read = 1; if (ioctl(fd, NVME_PASSTHROUGH_CMD, &pt) < 0) err(1, "report request failed"); if (nvme_completion_is_error(&pt.cpl)) errx(1, "report request returned error"); close(fd); if (report_opt.eds) nvme_resv_status_ext_swapbytes((void *)data, sizeof(data)); else nvme_resv_status_swapbytes((void *)data, sizeof(data)); if (report_opt.hex) { i = sizeof(data); if (!report_opt.verbose) { for (; i > 64; i--) { if (data[i - 1] != 0) break; } } print_hex(&data, i); exit(0); } s = (struct nvme_resv_status *)data; n = (s->regctl[1] << 8) | s->regctl[0]; printf("Generation: %u\n", s->gen); printf("Reservation Type: %u\n", s->rtype); printf("Number of Registered Controllers: %u\n", n); printf("Persist Through Power Loss State: %u\n", s->ptpls); if (report_opt.eds) { e = (struct nvme_resv_status_ext *)data; n = MIN(n, (sizeof(data) - sizeof(e)) / sizeof(e->ctrlr[0])); for (i = 0; i < n; i++) { printf("Controller ID: 0x%04x\n", e->ctrlr[i].ctrlr_id); printf(" Reservation Status: %u\n", e->ctrlr[i].rcsts); printf(" Reservation Key: 0x%08jx\n", e->ctrlr[i].rkey); printf(" Host Identifier: 0x%08jx%08jx\n", e->ctrlr[i].hostid[0], e->ctrlr[i].hostid[1]); } } else { n = MIN(n, (sizeof(data) - sizeof(s)) / sizeof(s->ctrlr[0])); for (i = 0; i < n; i++) { printf("Controller ID: 0x%04x\n", s->ctrlr[i].ctrlr_id); printf(" Reservation Status: %u\n", s->ctrlr[i].rcsts); printf(" Host Identifier: 0x%08jx\n", s->ctrlr[i].hostid); printf(" Reservation Key: 0x%08jx\n", s->ctrlr[i].rkey); } } exit(0); } static void resv(const struct cmd *nf __unused, int argc, char *argv[]) { cmd_dispatch(argc, argv, &resv_cmd); }