From 81218366f19d6d290f861a3dcb565fb49c21b488 Mon Sep 17 00:00:00 2001 From: Shiraz Hashim Date: Mon, 4 Jul 2016 15:05:14 +0530 Subject: [PATCH] iommu/io-pgtable-fast: validate data before use Return error on data allocation failure to prevent use after free case. Change-Id: I9d173756e277c9006220c02c209309167f179609 Signed-off-by: Shiraz Hashim --- drivers/iommu/io-pgtable-fast.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/drivers/iommu/io-pgtable-fast.c b/drivers/iommu/io-pgtable-fast.c index 44307f67fd22..9a25ebfdc778 100644 --- a/drivers/iommu/io-pgtable-fast.c +++ b/drivers/iommu/io-pgtable-fast.c @@ -420,6 +420,9 @@ av8l_fast_alloc_pgtable(struct io_pgtable_cfg *cfg, void *cookie) struct av8l_fast_io_pgtable *data = av8l_fast_alloc_pgtable_data(cfg); + if (!data) + return NULL; + /* restrict according to the fast map requirements */ cfg->ias = 32; cfg->pgsize_bitmap = SZ_4K;