Changes requested by keithbusch; no default for cfg.tl.

diff --git a/nvme.c b/nvme.c
index ad127b9b85d5ac3a7767e090222ee51608c9aaef..9b95cf660b586f62561f1bac1f813b9ac41e4ccf 100644 (file)
--- a/nvme.c
+++ b/nvme.c
@@ -3614,8 +3614,7 @@ static int sec_send(int argc, char **argv, struct command *cmd, struct plugin *p
        const char *nssf = "NVMe Security Specific Field";
        int err, fd, sec_fd = -1;
        void *sec_buf;
-       unsigned int sec_size, buf_size;
-       int nread = 0;
+       unsigned int sec_size;
 
        struct config {
                __u32 namespace_id;
@@ -3647,13 +3646,17 @@ static int sec_send(int argc, char **argv, struct command *cmd, struct plugin *p
        if (fd < 0)
                goto ret;
 
+       if (cfg.tl == 0) {
+               fprintf(stderr, "--tl unspecified or zero\n");
+               err = -EINVAL;
+               goto close_fd;
+       }
+       if ((cfg.tl & 3) != 0)
+               fprintf(stderr, "WARNING: --tl not dword aligned; unaligned bytes may be truncated\n");
+
        if (strlen(cfg.file) == 0) {
                sec_fd = STDIN_FILENO;
-
-               if (cfg.tl || ioctl(sec_fd, FIONREAD, &nread) < 0)
-                       sec_size = cfg.tl;
-               else
-                       sec_size = (unsigned int) nread;
+               sec_size = cfg.tl;
        } else {
                sec_fd = open(cfg.file, O_RDONLY);
                if (sec_fd < 0) {
@@ -3669,19 +3672,16 @@ static int sec_send(int argc, char **argv, struct command *cmd, struct plugin *p
                        goto close_sec_fd;
                }
 
-               sec_size = sb.st_size;
+               sec_size = cfg.tl > sb.st_size ? cfg.tl : sb.st_size;
        }
 
-       buf_size = cfg.tl ? cfg.tl : sec_size;
-       buf_size = (buf_size + 3) & ~3; // dword align to avoid truncation
-
-       if (posix_memalign(&sec_buf, getpagesize(), buf_size)) {
-               fprintf(stderr, "No memory for security size:%d\n", buf_size);
+       if (posix_memalign(&sec_buf, getpagesize(), cfg.tl)) {
+               fprintf(stderr, "No memory for security size:%d\n", cfg.tl);
                err = -ENOMEM;
                goto close_sec_fd;
        }
 
-       memset(sec_buf, 0, buf_size); // ensure zero fill if buf_size > sec_size
+       memset(sec_buf, 0, cfg.tl); // ensure zero fill if buf_size > sec_size
 
        err = read(sec_fd, sec_buf, sec_size);
        if (err < 0) {
@@ -3692,7 +3692,7 @@ static int sec_send(int argc, char **argv, struct command *cmd, struct plugin *p
        }
 
        err = nvme_sec_send(fd, cfg.namespace_id, cfg.nssf, cfg.spsp, cfg.secp,
-                       buf_size, buf_size, sec_buf);
+                       cfg.tl, cfg.tl, sec_buf);
        if (err < 0)
                perror("security-send");
        else if (err != 0)