From 1698689412b82a99eb400bc1709b7ba498e874a9 Mon Sep 17 00:00:00 2001 From: Daniel Verkamp Date: Tue, 15 Mar 2016 14:36:52 -0700 Subject: [PATCH] nvme_manage: handle fgets() failure fgets() will return NULL if it encounters EOF. Change-Id: I35b1744514f2a90d6e95156c3916a18c1a4ecce2 Signed-off-by: Daniel Verkamp --- examples/nvme/nvme_manage/nvme_manage.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/nvme/nvme_manage/nvme_manage.c b/examples/nvme/nvme_manage/nvme_manage.c index 1ebb1f4a0..70de88d22 100644 --- a/examples/nvme/nvme_manage/nvme_manage.c +++ b/examples/nvme/nvme_manage/nvme_manage.c @@ -240,6 +240,9 @@ get_controller(void) while ((ch = getchar()) != '\n' && ch != EOF); p = fgets(address, 64, stdin); + if (p == NULL) { + return NULL; + } while (isspace(*p)) { p++;