From 32e2880ed54fe58eb250f32b83ff854939d8b793 Mon Sep 17 00:00:00 2001
From: Ding Wei <leo.ding@rock-chips.com>
Date: Tue, 2 Feb 2021 11:06:00 +0800
Subject: [PATCH] video: rockchip: mpp: issue when rcb size is invalid

when rcb size is 0, return error directly.

Change-Id: I232e210cc88d0ee382970e6c5548eee5c9b975c4
Signed-off-by: Ding Wei <leo.ding@rock-chips.com>
---
 drivers/video/rockchip/mpp/mpp_rkvdec2.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/drivers/video/rockchip/mpp/mpp_rkvdec2.c b/drivers/video/rockchip/mpp/mpp_rkvdec2.c
index 120ea0f45545..dc4b6d922410 100644
--- a/drivers/video/rockchip/mpp/mpp_rkvdec2.c
+++ b/drivers/video/rockchip/mpp/mpp_rkvdec2.c
@@ -859,6 +859,10 @@ static int rkvdec_alloc_rcbbuf(struct platform_device *pdev, struct rkvdec_dev *
 	}
 	iova = PAGE_ALIGN(vals[0]);
 	rcb_size = PAGE_ALIGN(vals[1]);
+	if (!rcb_size) {
+		dev_err(dev, "rcb_size invalid.\n");
+		return -EINVAL;
+	}
 	/* alloc reserve iova for rcb */
 	ret = iommu_dma_reserve_iova(dev, iova, rcb_size);
 	if (ret) {