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

[Yi Wang]

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

NULL check is needed because kmalloc maybe return NULL.

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