plugins/sed/sed.c - master - githubmirror/nvme-cli

plugins/sed/sed.c
202 строки5 KB
Daniel Wagner
nvme: do not print status message on default
07 июл 2026, 11:05
07 июл 2026, 11:0559c9414
// SPDX-License-Identifier: GPL-2.0-or-later
#include <errno.h>
#include <fcntl.h>
#include <inttypes.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include <linux/fs.h>
#include <linux/sed-opal.h>

#include <sys/stat.h>

#include <libnvme.h>

#include "common.h"
#include "nvme.h"
#include "nvme-print.h"
#include "sedopal_cmd.h"

#define CREATE_CMD
#include "sed.h"

OPT_ARGS(no_opts) = {
	OPT_END()
};

OPT_ARGS(init_opts) = {
	OPT_FLAG("read-only", 'r', &sedopal_lock_ro,
		 "Set locking range to read-only"),
	OPT_END()
};

OPT_ARGS(key_opts) = {
	OPT_FLAG("ask-key", 'k', &sedopal_ask_key,
			"prompt for SED authentication key"),
	OPT_END()
};

OPT_ARGS(revert_opts) = {
	OPT_FLAG("destructive", 'e', &sedopal_destructive_revert,
			"destructive revert"),
	OPT_FLAG("psid", 'p', &sedopal_psid_revert, "PSID revert"),
	OPT_END()
};

OPT_ARGS(lock_opts) = {
	OPT_FLAG("read-only", 'r', &sedopal_lock_ro,
		 "Set locking range to read-only"),
	OPT_FLAG("ask-key", 'k', &sedopal_ask_key,
			"prompt for SED authentication key"),
	OPT_END()
};

OPT_ARGS(discovery_opts) = {
	OPT_FLAG("verbose", 'V', &sedopal_discovery_verbose,
		"Print extended discovery information"),
	OPT_FLAG("udev", 'u', &sedopal_discovery_udev,
		"Print locking information in form suitable for udev rules"),
	OPT_END()
};

/*
 * Open the NVMe device specified on the command line. It must be the
 * NVMe namespace device (e.g. /dev/nvme0n1).
 */
static int sed_opal_open_device(struct libnvme_global_ctx **ctx, struct libnvme_transport_handle **hdl, int argc, char **argv,
		const char *desc, struct argconfig_commandline_options *opts)
{
	int err;

	err = parse_and_open(ctx, hdl, argc, argv, desc, opts);
	if (err)
		return err;

	if (!libnvme_transport_handle_is_ns(*hdl)) {
		nvme_show_error(
			"ERROR : The NVMe namespace device (e.g. /dev/nvme0n1) "
			"must be specified\n");
		err = -EINVAL;
	}

	return err;
}

static int sed_opal_discover(int argc, char **argv, struct command *acmd,
		struct plugin *plugin)
{
	const char *desc = "Query SED device and display locking features";
	__cleanup_nvme_global_ctx struct libnvme_global_ctx *ctx = NULL;
	__cleanup_nvme_transport_handle struct libnvme_transport_handle *hdl = NULL;
	int err;

	err = sed_opal_open_device(&ctx, &hdl, argc, argv, desc, discovery_opts);
	if (err)
		return err;

	err = sedopal_cmd_discover(libnvme_transport_handle_get_fd(hdl));

	return err;
}

static int sed_opal_initialize(int argc, char **argv, struct command *acmd,
		struct plugin *plugin)
{
	const char *desc = "Initialize a SED device for locking";
	__cleanup_nvme_global_ctx struct libnvme_global_ctx *ctx = NULL;
	__cleanup_nvme_transport_handle struct libnvme_transport_handle *hdl = NULL;
	int err;

	err = sed_opal_open_device(&ctx, &hdl, argc, argv, desc, init_opts);
	if (err)
		return err;

	err = sedopal_cmd_initialize(libnvme_transport_handle_get_fd(hdl));
	if ((err != 0) && (err != -EOPNOTSUPP))
		nvme_show_error("initialize: SED error -  %s",
				sedopal_error_to_text(err));

	return err;
}

static int sed_opal_revert(int argc, char **argv, struct command *acmd,
		struct plugin *plugin)
{
	const char *desc = "Revert a SED device from locking state";
	__cleanup_nvme_global_ctx struct libnvme_global_ctx *ctx = NULL;
	__cleanup_nvme_transport_handle struct libnvme_transport_handle *hdl = NULL;
	int err;

	err = sed_opal_open_device(&ctx, &hdl, argc, argv, desc, revert_opts);
	if (err)
		return err;

	err = sedopal_cmd_revert(libnvme_transport_handle_get_fd(hdl));
	if ((err != 0) && (err != -EOPNOTSUPP) && (err != EPERM))
		nvme_show_error("revert: SED error -  %s",
				sedopal_error_to_text(err));

	return err;
}

static int sed_opal_lock(int argc, char **argv, struct command *acmd,
		struct plugin *plugin)
{
	const char *desc = "Lock a SED device";
	__cleanup_nvme_global_ctx struct libnvme_global_ctx *ctx = NULL;
	__cleanup_nvme_transport_handle struct libnvme_transport_handle *hdl = NULL;
	int err;

	err = sed_opal_open_device(&ctx, &hdl, argc, argv, desc, lock_opts);
	if (err)
		return err;

	err = sedopal_cmd_lock(libnvme_transport_handle_get_fd(hdl));
	if ((err != 0) && (err != -EOPNOTSUPP))
		nvme_show_error("lock: SED error -  %s",
				sedopal_error_to_text(err));

	return err;
}

static int sed_opal_unlock(int argc, char **argv, struct command *acmd,
		struct plugin *plugin)
{
	const char *desc = "Unlock a SED device";
	__cleanup_nvme_global_ctx struct libnvme_global_ctx *ctx = NULL;
	__cleanup_nvme_transport_handle struct libnvme_transport_handle *hdl = NULL;
	int err;

	err = sed_opal_open_device(&ctx, &hdl, argc, argv, desc, lock_opts);
	if (err)
		return err;

	err = sedopal_cmd_unlock(libnvme_transport_handle_get_fd(hdl));
	if ((err != 0) && (err != -EOPNOTSUPP))
		nvme_show_error("unlock: SED error -  %s",
				sedopal_error_to_text(err));

	return err;
}

static int sed_opal_password(int argc, char **argv, struct command *acmd,
		struct plugin *plugin)
{
	int err;
	const char *desc = "Change the locking password of a SED device";
	__cleanup_nvme_global_ctx struct libnvme_global_ctx *ctx = NULL;
	__cleanup_nvme_transport_handle struct libnvme_transport_handle *hdl = NULL;

	err = sed_opal_open_device(&ctx, &hdl, argc, argv, desc, no_opts);
	if (err)
		return err;

	err = sedopal_cmd_password(libnvme_transport_handle_get_fd(hdl));
	if ((err != 0) && (err != EPERM))
		nvme_show_error("password: SED error -  %s",
				sedopal_error_to_text(err));

	return err;
}