nvme: allow set-features to take input from stdin

Fixes: fed9e5ee04cd ("nvme: auto free nvme_dev resource")

diff --git a/nvme.c b/nvme.c
index 7ec03ece9822fd3e6f56194e45c2a2096a46b46f..06bd64cf40f6da5bbd550d382bfde0bb8663d66e 100644 (file)
--- a/nvme.c
+++ b/nvme.c
@@ -5875,16 +5875,18 @@ static int set_feature(int argc, char **argv, struct command *cmd, struct plugin
                if (cfg.feature_id == NVME_FEAT_FID_TIMESTAMP && cfg.value) {
                        memcpy(buf, &cfg.value, NVME_FEAT_TIMESTAMP_DATA_SIZE);
                } else {
-                       if (strlen(cfg.file)) {
+                       if (strlen(cfg.file))
                                ffd = open(cfg.file, O_RDONLY);
-                               if (ffd <= 0) {
-                                       nvme_show_error("Failed to open file %s: %s",
-                                                       cfg.file, strerror(errno));
-                                       return -EINVAL;
-                               }
+                       else
+                               ffd = dup(STDIN_FILENO);
+
+                       if (ffd < 0) {
+                               nvme_show_error("Failed to open file %s: %s",
+                                               cfg.file, strerror(errno));
+                               return -EINVAL;
                        }
 
-                       err = read(ffd, (void *)buf, cfg.data_len);
+                       err = read(ffd, buf, cfg.data_len);
                        if (err < 0) {
                                nvme_show_error("failed to read data buffer from input file: %s",
                                                strerror(errno));