Commit 7e31d6dc authored by Max Gurtovoy's avatar Max Gurtovoy Committed by Alex Williamson
Browse files

vfio-pci/zdev: fix possible segmentation fault issue 



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

Fixes: e6b817d4 ("vfio-pci/zdev: Add zPCI capabilities to VFIO_DEVICE_GET_INFO")
Signed-off-by: default avatarMax Gurtovoy <mgurtovoy@nvidia.com>
Reviewed-by: default avatarCornelia Huck <cohuck@redhat.com>
Reviewed-by: default avatarMatthew Rosato <mjrosato@linux.ibm.com>
Signed-off-by: default avatarAlex Williamson <alex.williamson@redhat.com>
parent 46c47466

drivers/vfio/pci/vfio_pci_zdev.c

+4 −0
Original line number Diff line number Diff line
@@ -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;