feat: add err-recovery command

This is for Error Recovery feature to get and set.

Signed-off-by: Tokunori Ikegami <[email protected]>

Author: ikegami-t

nvme.c

@@ -187,6 +187,7 @@ static struct program nvme = {
 };
 
 const char *uuid_index = "UUID index";
+const char *namespace_id_desired = "identifier of desired namespace";
 
 static const char *app_tag = "app tag for end-to-end PI";
 static const char *app_tag_mask = "app tag mask for end-to-end PI";
@@ -214,7 +215,6 @@ static const char *lsp = "log specific field";
 static const char *mos = "management operation specific";
 static const char *mo = "management operation";
 static const char *namespace_desired = "desired namespace";
-static const char *namespace_id_desired = "identifier of desired namespace";
 static const char *namespace_id_optional = "optional namespace attached to controller";
 static const char *nssf = "NVMe Security Specific Field";
 static const char *only_char_dev = "Only character device is allowed";

nvme.h

@@ -120,6 +120,7 @@ static inline DEFINE_CLEANUP_FUNC(
 #define _cleanup_nvme_transport_handle_ __cleanup__(cleanup_nvme_transport_handle)
 
 extern const char *uuid_index;
+extern const char *namespace_id_desired;
 extern struct nvme_args nvme_args;
 
 int validate_output_format(const char *format, nvme_print_flags_t *flags);

plugins/feat/feat-nvme.c

@@ -41,6 +41,12 @@ struct arbitration_config {
 	__u8 sel;
 };
 
+struct err_recovery_config {
+	__u8 tler;
+	__u8 dulbe;
+	__u8 sel;
+};
+
 static const char *power_mgmt_feat = "power management feature";
 static const char *sel = "[0-3]: current/default/saved/supported";
 static const char *save = "Specifies that the controller shall save the attribute";
@@ -53,9 +59,11 @@ static const char *volatile_wc_feat = "volatile write cache feature";
 static const char *power_limit_feat = "power limit feature";
 static const char *power_thresh_feat = "power threshold feature";
 static const char *power_meas_feat = "power measurement feature";
+static const char *err_recovery_feat = "error recovery feature";
 
-static int feat_get(struct nvme_transport_handle *hdl, const __u8 fid,
-		    __u32 cdw11, __u8 sel, __u8 uidx, const char *feat)
+static int feat_get_nsid(struct nvme_transport_handle *hdl, __u32 nsid,
+			 const __u8 fid, __u32 cdw11, __u8 sel, __u8 uidx,
+			 const char *feat)
 {
 	__u64 result;
 	int err;
@@ -72,7 +80,7 @@ static int feat_get(struct nvme_transport_handle *hdl, const __u8 fid,
 			return -ENOMEM;
 	}
 
-	err = nvme_get_features(hdl, 0, fid, sel, cdw11, uidx, buf, len,
+	err = nvme_get_features(hdl, nsid, fid, sel, cdw11, uidx, buf, len,
 				&result);
 
 	nvme_show_init();
@@ -94,6 +102,12 @@ static int feat_get(struct nvme_transport_handle *hdl, const __u8 fid,
 	return err;
 }
 
+static int feat_get(struct nvme_transport_handle *hdl, const __u8 fid,
+		    __u32 cdw11, __u8 sel, __u8 uidx, const char *feat)
+{
+	return feat_get_nsid(hdl, 0, fid, cdw11, sel, uidx, feat);
+}
+
 static int power_mgmt_set(struct nvme_transport_handle *hdl, const __u8 fid,
 			  __u8 ps, __u8 wh, bool sv)
 {
@@ -831,3 +845,74 @@ static int feat_power_meas(int argc, char **argv, struct command *cmd,
 
 	return err;
 }
+
+static int err_recovery_set(struct nvme_transport_handle *hdl, const __u8 fid,
+			    __u32 nsid, __u16 tler, bool dulbe, bool sv)
+{
+	__u32 cdw11 = NVME_SET(tler, FEAT_ERROR_RECOVERY_TLER) |
+		      NVME_SET(dulbe, FEAT_ERROR_RECOVERY_DULBE);
+	__u64 result;
+	int err;
+
+	err = nvme_set_features(hdl, nsid, fid, sv, cdw11, 0, 0, 0, 0, NULL, 0,
+				&result);
+
+	nvme_show_init();
+
+	if (err > 0) {
+		nvme_show_status(err);
+	} else if (err < 0) {
+		nvme_show_perror("Set %s", err_recovery_feat);
+	} else {
+		nvme_show_result("Set %s: 0x%04x (%s)", err_recovery_feat,
+				 cdw11, sv ? "Save" : "Not save");
+		nvme_feature_show_fields(fid, cdw11, NULL);
+	}
+
+	nvme_show_finish();
+
+	return err;
+}
+
+static int feat_err_recovery(int argc, char **argv, struct command *acmd,
+			     struct plugin *plugin)
+{
+	_cleanup_nvme_transport_handle_ struct nvme_transport_handle *hdl =
+	    NULL;
+	_cleanup_nvme_global_ctx_ struct nvme_global_ctx *ctx = NULL;
+
+	const char *dulbe =
+	    "deallocated or unwritten logical block error enable";
+	const char *tler = "time limited error recovery";
+	const __u8 fid = NVME_FEAT_FID_ERR_RECOVERY;
+
+	int err;
+
+	struct config {
+		__u32 nsid;
+		__u16 tler;
+		bool dulbe;
+		__u8 sel;
+	};
+
+	struct config cfg = { 0 };
+
+	FEAT_ARGS(opts,
+		  OPT_UINT("nsid", 'n', &cfg.nsid, namespace_id_desired),
+		  OPT_SHRT("tler", 't', &cfg.tler, tler),
+		  OPT_FLAG("dulbe", 'd', &cfg.dulbe, dulbe));
+
+	err = parse_and_open(&ctx, &hdl, argc, argv, ERR_RECOVERY_DESC, opts);
+	if (err)
+		return err;
+
+	if (argconfig_parse_seen(opts, "tler") ||
+	    argconfig_parse_seen(opts, "dulbe"))
+		err = err_recovery_set(hdl, fid, cfg.nsid, cfg.tler, cfg.dulbe,
+				       argconfig_parse_seen(opts, "save"));
+	else
+		err = feat_get_nsid(hdl, cfg.nsid, fid, 0, cfg.sel, 0,
+				    err_recovery_feat);
+
+	return err;
+}

plugins/feat/feat-nvme.h

@@ -20,6 +20,7 @@
 #define POWER_LIMIT_DESC "Get and set power limit feature"
 #define POWER_THRESH_DESC "Get and set power threshold feature"
 #define POWER_MEAS_DESC "Get and set power measurement feature"
+#define ERR_RECOVERY_DESC "Get and set error recovery feature"
 
 #define FEAT_ARGS(n, ...)                                              \
 	NVME_ARGS(n, ##__VA_ARGS__, OPT_FLAG("save", 's', NULL, save), \
@@ -37,6 +38,7 @@ PLUGIN(NAME("feat", "NVMe feature extensions", FEAT_PLUGIN_VERSION),
 		ENTRY("power-limit", POWER_LIMIT_DESC, feat_power_limit)
 		ENTRY("power-thresh", POWER_THRESH_DESC, feat_power_thresh)
 		ENTRY("power-meas", POWER_MEAS_DESC, feat_power_meas)
+		ENTRY("err-recovery", ERR_RECOVERY_DESC, feat_err_recovery)
 	)
 );
 #endif /* !FEAT_NVME || CMD_HEADER_MULTI_READ */