diff options
| author |   Max Gurtovoy <mgurtovoy@nvidia.com> | 2021-02-01 16:28:25 +0000 |
|---|---|---|
| committer |   Alex Williamson <alex.williamson@redhat.com> | 2021-02-02 09:06:02 -0700 |
| commit | 7e31d6dc2c78b2a0ba0039ca97ca98a581e8db82 (patch) | |
| tree | 4b83cdfe921969ac1ee674d4a1810f31639dce83 /drivers/vfio/pci | |
| parent | vfio-pci/zdev: remove unused vdev argument (diff) | |
| download | linux-dev-7e31d6dc2c78b2a0ba0039ca97ca98a581e8db82.tar.xz linux-dev-7e31d6dc2c78b2a0ba0039ca97ca98a581e8db82.zip | |
vfio-pci/zdev: fix possible segmentation fault issue
In case allocation fails, we must behave correctly and exit with error.
Fixes: e6b817d4b821 ("vfio-pci/zdev: Add zPCI capabilities to VFIO_DEVICE_GET_INFO")
Signed-off-by: Max Gurtovoy <mgurtovoy@nvidia.com>
Reviewed-by: Cornelia Huck <cohuck@redhat.com>
Reviewed-by: Matthew Rosato <mjrosato@linux.ibm.com>
Signed-off-by: Alex Williamson <alex.williamson@redhat.com>
Diffstat (limited to 'drivers/vfio/pci')
| -rw-r--r-- | drivers/vfio/pci/vfio_pci_zdev.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/drivers/vfio/pci/vfio_pci_zdev.c b/drivers/vfio/pci/vfio_pci_zdev.c index 53084521cc80..7b011b62c766 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; |