[PATCH] virtio:linux:kernel:NULL check after kmalloc is needed

[Yi Wang]

From: "huang.zijiang" <huang.zijiang@xxxxxxxxxx>

NULL check is needed because kmalloc maybe return NULL.

Signed-off-by: huang.zijiang <huang.zijiang@xxxxxxxxxx>
---
 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