[PATCH 6/9] vfio-pci/zdev: fix possible segmentation fault issue

[Max Gurtovoy]

In case allocation fails, we must behave correctly and exit with error.

Signed-off-by: Max Gurtovoy <mgurto...@nvidia.com>
---
 drivers/vfio/pci/vfio_pci_zdev.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/drivers/vfio/pci/vfio_pci_zdev.c b/drivers/vfio/pci/vfio_pci_zdev.c
index 175096fcd902..e9ef4239ef7a 100644
--- a/drivers/vfio/pci/vfio_pci_zdev.c
+++ b/drivers/vfio/pci/vfio_pci_zdev.c
@@ -71,6 +71,8 @@ static int zpci_util_cap(struct zpci_dev *zdev, struct 
vfio_info_cap *caps)
        int ret;
 
        cap = kmalloc(cap_size, GFP_KERNEL);
+       if (!cap)
+               return -ENOMEM;
 
        cap->header.id = VFIO_DEVICE_INFO_CAP_ZPCI_UTIL;
        cap->header.version = 1;
@@ -94,6 +96,8 @@ static int zpci_pfip_cap(struct zpci_dev *zdev, struct 
vfio_info_cap *caps)
        int ret;
 
        cap = kmalloc(cap_size, GFP_KERNEL);
+       if (!cap)
+               return -ENOMEM;
 
        cap->header.id = VFIO_DEVICE_INFO_CAP_ZPCI_PFIP;
        cap->header.version = 1;
-- 
2.25.4