Received-SPF: pass (google.com: best guess record for domain of linux-kernel-owner@vger.kernel.org designates 209.132.180.67 as permitted sender) client-ip=209.132.180.67;
From:   Yi Wang <wang.yi59@zte.com.cn>
To:     mst@redhat.com
Cc:     jasowang@redhat.com, davem@davemloft.net,
        virtualization@lists.linux-foundation.org,
        linux-kernel@vger.kernel.org, xue.zhihong@zte.com.cn,
        wang.yi59@zte.com.cn, "huang.zijiang" <huang.zijiang@zte.com.cn>
Subject: [PATCH] virtio:linux:kernel:NULL check after kmalloc is  needed
Date:   Tue, 8 Jan 2019 13:40:33 +0800
Message-Id: <1546926033-31684-1-git-send-email-wang.yi59@zte.com.cn>
Sender: linux-kernel-owner@vger.kernel.org
Precedence: bulk

From: "huang.zijiang" <huang.zijiang@zte.com.cn>

NULL check is needed because kmalloc maybe return NULL.

Signed-off-by: huang.zijiang <huang.zijiang@zte.com.cn>
---
 tools/virtio/linux/kernel.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/virtio/linux/kernel.h b/tools/virtio/linux/kernel.h
index 7ef45a4..2afcad8 100644
--- a/tools/virtio/linux/kernel.h
+++ b/tools/virtio/linux/kernel.h
@@ -65,6 +65,8 @@ static inline void *kzalloc(size_t s, gfp_t gfp)
 {
 	void *p = kmalloc(s, gfp);
 
+	if (!p)
+		return -ENOMEM;
 	memset(p, 0, s);
 	return p;
 }
-- 
1.8.3.1