From 21090f1e2889190450ec0b833dffd0a3af372e2a Mon Sep 17 00:00:00 2001
From: Bharat Bhushan <Bharat.Bhushan@freescale.com>
Date: Mon, 8 Jun 2015 16:31:05 +0530
Subject: VFIO: Check for null pointer after malloc

Add the check whether malloc allocated memory successfully or not

Signed-off-by: Bharat Bhushan <Bharat.Bhushan@freescale.com>
Change-Id: If306002aa8541cf76286b7b78d0027c3395672a6
Reviewed-on: http://git.am.freescale.net:8181/37576
Tested-by: Review Code-CDREVIEW <CDREVIEW@freescale.com>
Reviewed-by: Stuart Yoder <stuart.yoder@freescale.com>
Reviewed-by: Honghua Yin <Hong-Hua.Yin@freescale.com>

diff --git a/drivers/vfio/vfio_iommu_fsl_pamu.c b/drivers/vfio/vfio_iommu_fsl_pamu.c
index d049edc..4d300da 100644
--- a/drivers/vfio/vfio_iommu_fsl_pamu.c
+++ b/drivers/vfio/vfio_iommu_fsl_pamu.c
@@ -200,6 +200,9 @@ static int vaddr_get_pfn(unsigned long vaddr, int prot, unsigned long *pfn,
 	int i;
 
 	pages = kzalloc(sizeof(*pages) * nr_pages, GFP_KERNEL);
+	if (!pages)
+		return -ENOMEM;
+
 	ret = get_user_pages_fast(vaddr, nr_pages, !!(prot & IOMMU_WRITE), pages);
 	if (ret != nr_pages)
 		goto error;
-- 
cgit v0.10.2