From: Minwoo Im <minwoo.im.dev@gmail.com>
Date: Wed, 17 Jan 2018 10:24:19 +0000 (+0900)
Subject: nvme-cli: add error handling for a failure of malloc
X-Git-Tag: v1.6~114^2
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=641b5a5a92370d503797267d0d8aec87cac776fc;p=users%2Fsagi%2Fnvme-cli.git

nvme-cli: add error handling for a failure of malloc

Make it return ENOMEM when malloc of metadat is failed with some error
message.

Signed-off-by: Minwoo Im <minwoo.im.dev@gmail.com>
---

diff --git a/nvme.c b/nvme.c
index a6f0639d..6563cfd2 100644
--- a/nvme.c
+++ b/nvme.c
@@ -3650,8 +3650,13 @@ static int passthru(int argc, char **argv, int ioctl_cmd, const char *desc, stru
 		}
 	}
 
-	if (cfg.metadata_len)
+	if (cfg.metadata_len) {
 		metadata = malloc(cfg.metadata_len);
+		if (!metadata) {
+			fprintf(stderr, "can not allocate metadata payload\n");
+			return ENOMEM;
+		}
+	}
 	if (cfg.data_len) {
 		if (posix_memalign(&data, getpagesize(), cfg.data_len)) {
 			fprintf(stderr, "can not allocate data payload\n");