diff mbox series

[v11,01/19] media: mtk-vcodec: Get numbers of register bases from DT

Message ID 20211129034201.5767-2-yunfei.dong@mediatek.com (mailing list archive)
State New, archived
Headers show
Series Support multi hardware decode using of_platform_populate | expand

Commit Message

Yunfei Dong Nov. 29, 2021, 3:41 a.m. UTC 
Different platform may has different numbers of register bases. Gets the
numbers of register bases from DT (sizeof(u32) * 4 bytes for each).

Reviewed-by: Tzung-Bi Shih<tzungbi@google.com>
Signed-off-by: Yunfei Dong <yunfei.dong@mediatek.com>
---
 .../platform/mtk-vcodec/mtk_vcodec_dec_drv.c  | 37 ++++++++++++++-----
 1 file changed, 28 insertions(+), 9 deletions(-)

Comments

Benjamin Gaignard Nov. 30, 2021, 1:20 p.m. UTC | #1 
Le 29/11/2021 à 04:41, Yunfei Dong a écrit :
> Different platform may has different numbers of register bases. Gets the
> numbers of register bases from DT (sizeof(u32) * 4 bytes for each).
>
> Reviewed-by: Tzung-Bi Shih<tzungbi@google.com>
> Signed-off-by: Yunfei Dong <yunfei.dong@mediatek.com>
> ---
>   .../platform/mtk-vcodec/mtk_vcodec_dec_drv.c  | 37 ++++++++++++++-----
>   1 file changed, 28 insertions(+), 9 deletions(-)
>
> diff --git a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> index e6e6a8203eeb..59caf2163349 100644
> --- a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> +++ b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> @@ -78,6 +78,30 @@ static irqreturn_t mtk_vcodec_dec_irq_handler(int irq, void *priv)
>   	return IRQ_HANDLED;
>   }
>   
> +static int mtk_vcodec_get_reg_bases(struct mtk_vcodec_dev *dev)
> +{
> +	struct platform_device *pdev = dev->plat_dev;
> +	int reg_num, i;
> +
> +	/* Sizeof(u32) * 4 bytes for each register base. */
> +	reg_num = of_property_count_elems_of_size(pdev->dev.of_node, "reg",
> +		sizeof(u32) * 4);

It looks strange for me to have a "reg" size equal to sizeof(u32) * 4. Usually
we more see reg size = sizeof(u32).

> +	if (reg_num <= 0 || reg_num > NUM_MAX_VDEC_REG_BASE) {

If reg_num = NUM_MAX_VDEC_REG_BASE you will iterate out of bounds of dev->reg_base array.
That never happens because dev->reg_base size equal NUM_MAX_VCODEC_REG_BASE.
The question is what is the real needed size for dev->reg_base array ? NUM_MAX_VDEC_REG_BASE or
NUM_MAX_VCODEC_REG_BASE ?

Regards,
Benjamin

> +		dev_err(&pdev->dev, "Invalid register property size: %d\n", reg_num);
> +		return -EINVAL;
> +	}
> +
> +	for (i = 0; i < reg_num; i++) {
> +		dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
> +		if (IS_ERR(dev->reg_base[i]))
> +			return PTR_ERR(dev->reg_base[i]);
> +
> +		mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
> +	}
> +
> +	return 0;
> +}
> +
>   static int fops_vcodec_open(struct file *file)
>   {
>   	struct mtk_vcodec_dev *dev = video_drvdata(file);
> @@ -206,7 +230,7 @@ static int mtk_vcodec_probe(struct platform_device *pdev)
>   	struct resource *res;
>   	phandle rproc_phandle;
>   	enum mtk_vcodec_fw_type fw_type;
> -	int i, ret;
> +	int ret;
>   
>   	dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
>   	if (!dev)
> @@ -238,14 +262,9 @@ static int mtk_vcodec_probe(struct platform_device *pdev)
>   		goto err_dec_pm;
>   	}
>   
> -	for (i = 0; i < NUM_MAX_VDEC_REG_BASE; i++) {
> -		dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
> -		if (IS_ERR((__force void *)dev->reg_base[i])) {
> -			ret = PTR_ERR((__force void *)dev->reg_base[i]);
> -			goto err_res;
> -		}
> -		mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
> -	}
> +	ret = mtk_vcodec_get_reg_bases(dev);
> +	if (ret)
> +		goto err_res;
>   
>   	res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
>   	if (res == NULL) {
Steve Cho Dec. 1, 2021, 11:36 p.m. UTC | #2 
LGTM with few nits.

Thanks,
Steve

On Sun, Nov 28, 2021 at 7:44 PM Yunfei Dong <yunfei.dong@mediatek.com>
wrote:

> Different platform may has different numbers of register bases. Gets the
> numbers of register bases from DT (sizeof(u32) * 4 bytes for each)

Few nits.
s/platform/platforms/
s/has/have/

Btw, what is DT?

>
>
Reviewed-by: Tzung-Bi Shih<tzungbi@google.com>
> Signed-off-by: Yunfei Dong <yunfei.dong@mediatek.com>
> ---
>  .../platform/mtk-vcodec/mtk_vcodec_dec_drv.c  | 37 ++++++++++++++-----
>  1 file changed, 28 insertions(+), 9 deletions(-)
>
> diff --git a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> index e6e6a8203eeb..59caf2163349 100644
> --- a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> +++ b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> @@ -78,6 +78,30 @@ static irqreturn_t mtk_vcodec_dec_irq_handler(int irq,
> void *priv)
>         return IRQ_HANDLED;
>  }
>
> +static int mtk_vcodec_get_reg_bases(struct mtk_vcodec_dev *dev)
> +{
>
I see that dev is already checked before entering into this function, but
null check for dev would still be nice.


> +       struct platform_device *pdev = dev->plat_dev;
> +       int reg_num, i;
> +
> +       /* Sizeof(u32) * 4 bytes for each register base. */
> +       reg_num = of_property_count_elems_of_size(pdev->dev.of_node, "reg",
> +               sizeof(u32) * 4);
> +       if (reg_num <= 0 || reg_num > NUM_MAX_VDEC_REG_BASE) {
> +               dev_err(&pdev->dev, "Invalid register property size:
> %d\n", reg_num);
> +               return -EINVAL;
> +       }
> +
> +       for (i = 0; i < reg_num; i++) {
> +               dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
> +               if (IS_ERR(dev->reg_base[i]))
> +                       return PTR_ERR(dev->reg_base[i]);
> +
> +               mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
> +       }
> +
> +       return 0;
> +}
> +
>  static int fops_vcodec_open(struct file *file)
>  {
>         struct mtk_vcodec_dev *dev = video_drvdata(file);
> @@ -206,7 +230,7 @@ static int mtk_vcodec_probe(struct platform_device
> *pdev)
>         struct resource *res;
>         phandle rproc_phandle;
>         enum mtk_vcodec_fw_type fw_type;
> -       int i, ret;
> +       int ret;
>
>         dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
>         if (!dev)
> @@ -238,14 +262,9 @@ static int mtk_vcodec_probe(struct platform_device
> *pdev)
>                 goto err_dec_pm;
>         }
>
> -       for (i = 0; i < NUM_MAX_VDEC_REG_BASE; i++) {
> -               dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
> -               if (IS_ERR((__force void *)dev->reg_base[i])) {
> -                       ret = PTR_ERR((__force void *)dev->reg_base[i]);
> -                       goto err_res;
> -               }
> -               mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
> -       }
> +       ret = mtk_vcodec_get_reg_bases(dev);
> +       if (ret)
> +               goto err_res;
>
>         res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
>         if (res == NULL) {
> --
> 2.25.1
>
>
Yunfei Dong Dec. 2, 2021, 2:24 a.m. UTC | #3 
Hi Benjamin,

Thanks for your suggestion.
On Tue, 2021-11-30 at 14:20 +0100, Benjamin Gaignard wrote:
> Le 29/11/2021 à 04:41, Yunfei Dong a écrit :
> > Different platform may has different numbers of register bases.
> > Gets the
> > numbers of register bases from DT (sizeof(u32) * 4 bytes for each).
> > 
> > Reviewed-by: Tzung-Bi Shih<tzungbi@google.com>
> > Signed-off-by: Yunfei Dong <yunfei.dong@mediatek.com>
> > ---
> >   .../platform/mtk-vcodec/mtk_vcodec_dec_drv.c  | 37
> > ++++++++++++++-----
> >   1 file changed, 28 insertions(+), 9 deletions(-)
> > 
> > diff --git a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c 
> > b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> > index e6e6a8203eeb..59caf2163349 100644
> > --- a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> > +++ b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> > @@ -78,6 +78,30 @@ static irqreturn_t
> > mtk_vcodec_dec_irq_handler(int irq, void *priv)
> >   	return IRQ_HANDLED;
> >   }
> >   
> > +static int mtk_vcodec_get_reg_bases(struct mtk_vcodec_dev *dev)
> > +{
> > +	struct platform_device *pdev = dev->plat_dev;
> > +	int reg_num, i;
> > +
> > +	/* Sizeof(u32) * 4 bytes for each register base. */
> > +	reg_num = of_property_count_elems_of_size(pdev->dev.of_node,
> > "reg",
> > +		sizeof(u32) * 4);
> 
> It looks strange for me to have a "reg" size equal to sizeof(u32) *
> 4. Usually
> we more see reg size = sizeof(u32).
> 
For the definition of reg in dtsi is reg = <0 0x16000000 0 0x800>,
Need to using sizeof(u32) * 4, not sizeof(u32).
> > +	if (reg_num <= 0 || reg_num > NUM_MAX_VDEC_REG_BASE) {
> 
> If reg_num = NUM_MAX_VDEC_REG_BASE you will iterate out of bounds of
> dev->reg_base array.
> That never happens because dev->reg_base size equal
> NUM_MAX_VCODEC_REG_BASE.
> The question is what is the real needed size for dev->reg_base array
> ? NUM_MAX_VDEC_REG_BASE or
> NUM_MAX_VCODEC_REG_BASE ?
> 
> NUM_MAX_VDEC_REG_BASE is used for video decoder, and
> NUM_MAX_VCODEC_REG_BASE is used for encoder.

The max register number of video decoder is NUM_MAX_VDEC_REG_BASE, and
the ranges is 0 ~ NUM_MAX_VDEC_REG_BASE - 1.

> Regards,
> Benjamin
> 
Best Regards,
Yunfei Dong
> > +		dev_err(&pdev->dev, "Invalid register property size:
> > %d\n", reg_num);
> > +		return -EINVAL;
> > +	}
> > +
> > +	for (i = 0; i < reg_num; i++) {
> > +		dev->reg_base[i] = devm_platform_ioremap_resource(pdev,
> > i);
> > +		if (IS_ERR(dev->reg_base[i]))
> > +			return PTR_ERR(dev->reg_base[i]);
> > +
> > +		mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev-
> > >reg_base[i]);
> > +	}
> > +
> > +	return 0;
> > +}
> > +
> >   static int fops_vcodec_open(struct file *file)
> >   {
> >   	struct mtk_vcodec_dev *dev = video_drvdata(file);
> > @@ -206,7 +230,7 @@ static int mtk_vcodec_probe(struct
> > platform_device *pdev)
> >   	struct resource *res;
> >   	phandle rproc_phandle;
> >   	enum mtk_vcodec_fw_type fw_type;
> > -	int i, ret;
> > +	int ret;
> >   
> >   	dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
> >   	if (!dev)
> > @@ -238,14 +262,9 @@ static int mtk_vcodec_probe(struct
> > platform_device *pdev)
> >   		goto err_dec_pm;
> >   	}
> >   
> > -	for (i = 0; i < NUM_MAX_VDEC_REG_BASE; i++) {
> > -		dev->reg_base[i] = devm_platform_ioremap_resource(pdev,
> > i);
> > -		if (IS_ERR((__force void *)dev->reg_base[i])) {
> > -			ret = PTR_ERR((__force void *)dev-
> > >reg_base[i]);
> > -			goto err_res;
> > -		}
> > -		mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev-
> > >reg_base[i]);
> > -	}
> > +	ret = mtk_vcodec_get_reg_bases(dev);
> > +	if (ret)
> > +		goto err_res;
> >   
> >   	res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
> >   	if (res == NULL) {
Yunfei Dong Dec. 2, 2021, 2:38 a.m. UTC | #4 
Hi Steve,
Thanks for your suggestion.On Wed, 2021-12-01 at 15:36 -0800, Steve Cho
wrote:
> LGTM with few nits. 
> 
> Thanks,Steve
> 
> On Sun, Nov 28, 2021 at 7:44 PM Yunfei Dong <yunfei.dong@mediatek.com
> > wrote:
> > Different platform may has different numbers of register bases.
> > Gets the
> > 
> > numbers of register bases from DT (sizeof(u32) * 4 bytes for each)
> Few nits.
> s/platform/platforms/
> s/has/have/
> 
> Fix, DT is dts.
> Btw, what is DT?
> >  
> > 
> > Reviewed-by: Tzung-Bi Shih<tzungbi@google.com>
> > 
> > Signed-off-by: Yunfei Dong <yunfei.dong@mediatek.com>
> > 
> > ---
> > 
> >  .../platform/mtk-vcodec/mtk_vcodec_dec_drv.c  | 37 ++++++++++++++-
> > ----
> > 
> >  1 file changed, 28 insertions(+), 9 deletions(-)
> > 
> > 
> > 
> > diff --git a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c 
> > b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> > 
> > index e6e6a8203eeb..59caf2163349 100644
> > 
> > --- a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> > 
> > +++ b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> > 
> > @@ -78,6 +78,30 @@ static irqreturn_t
> > mtk_vcodec_dec_irq_handler(int irq, void *priv)
> > 
> >         return IRQ_HANDLED;
> > 
> >  }
> > 
> > 
> > 
> > +static int mtk_vcodec_get_reg_bases(struct mtk_vcodec_dev *dev)
> > 
> > +{
> I see that dev is already checked before entering into this function,
> but null check for dev would still be nice. 
>  
Dev is never null in this function, whether it looks not very
reasonable?
Best Regards,Yunfei Dong
> > +       struct platform_device *pdev = dev->plat_dev;
> > 
> > +       int reg_num, i;
> > 
> > +
> > 
> > +       /* Sizeof(u32) * 4 bytes for each register base. */
> > 
> > +       reg_num = of_property_count_elems_of_size(pdev-
> > >dev.of_node, "reg",
> > 
> > +               sizeof(u32) * 4);
> > 
> > +       if (reg_num <= 0 || reg_num > NUM_MAX_VDEC_REG_BASE) {
> > 
> > +               dev_err(&pdev->dev, "Invalid register property
> > size: %d\n", reg_num);
> > 
> > +               return -EINVAL;
> > 
> > +       }
> > 
> > +
> > 
> > +       for (i = 0; i < reg_num; i++) {
> > 
> > +               dev->reg_base[i] =
> > devm_platform_ioremap_resource(pdev, i);
> > 
> > +               if (IS_ERR(dev->reg_base[i]))
> > 
> > +                       return PTR_ERR(dev->reg_base[i]);
> > 
> > +
> > 
> > +               mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev-
> > >reg_base[i]);
> > 
> > +       }
> > 
> > +
> > 
> > +       return 0;
> > 
> > +}
> > 
> > +
> > 
> >  static int fops_vcodec_open(struct file *file)
> > 
> >  {
> > 
> >         struct mtk_vcodec_dev *dev = video_drvdata(file);
> > 
> > @@ -206,7 +230,7 @@ static int mtk_vcodec_probe(struct
> > platform_device *pdev)
> > 
> >         struct resource *res;
> > 
> >         phandle rproc_phandle;
> > 
> >         enum mtk_vcodec_fw_type fw_type;
> > 
> > -       int i, ret;
> > 
> > +       int ret;
> > 
> > 
> > 
> >         dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
> > 
> >         if (!dev)
> > 
> > @@ -238,14 +262,9 @@ static int mtk_vcodec_probe(struct
> > platform_device *pdev)
> > 
> >                 goto err_dec_pm;
> > 
> >         }
> > 
> > 
> > 
> > -       for (i = 0; i < NUM_MAX_VDEC_REG_BASE; i++) {
> > 
> > -               dev->reg_base[i] =
> > devm_platform_ioremap_resource(pdev, i);
> > 
> > -               if (IS_ERR((__force void *)dev->reg_base[i])) {
> > 
> > -                       ret = PTR_ERR((__force void *)dev-
> > >reg_base[i]);
> > 
> > -                       goto err_res;
> > 
> > -               }
> > 
> > -               mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev-
> > >reg_base[i]);
> > 
> > -       }
> > 
> > +       ret = mtk_vcodec_get_reg_bases(dev);
> > 
> > +       if (ret)
> > 
> > +               goto err_res;
> > 
> > 
> > 
> >         res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
> > 
> >         if (res == NULL) {
> >
Steve Cho Dec. 2, 2021, 4:38 p.m. UTC | #5 
Thanks,
Steve

On Wed, Dec 1, 2021 at 6:38 PM yunfei.dong@mediatek.com <
yunfei.dong@mediatek.com> wrote:

> Hi Steve,
>
> Thanks for your suggestion.
> On Wed, 2021-12-01 at 15:36 -0800, Steve Cho wrote:
>
> LGTM with few nits.
>
> Thanks,
> Steve
>
> On Sun, Nov 28, 2021 at 7:44 PM Yunfei Dong <yunfei.dong@mediatek.com>
> wrote:
>
> Different platform may has different numbers of register bases. Gets the
> numbers of register bases from DT (sizeof(u32) * 4 bytes for each)
>
> Few nits.
> s/platform/platforms/
> s/has/have/
>
> Fix, DT is dts.
> Btw, what is DT?
>
>
> Reviewed-by: Tzung-Bi Shih<tzungbi@google.com>
> Signed-off-by: Yunfei Dong <yunfei.dong@mediatek.com>
> ---
>  .../platform/mtk-vcodec/mtk_vcodec_dec_drv.c  | 37 ++++++++++++++-----
>  1 file changed, 28 insertions(+), 9 deletions(-)
>
> diff --git a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> index e6e6a8203eeb..59caf2163349 100644
> --- a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> +++ b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
> @@ -78,6 +78,30 @@ static irqreturn_t mtk_vcodec_dec_irq_handler(int irq,
> void *priv)
>         return IRQ_HANDLED;
>  }
>
> +static int mtk_vcodec_get_reg_bases(struct mtk_vcodec_dev *dev)
> +{
>
> I see that dev is already checked before entering into this function, but
> null check for dev would still be nice.
>
>
> Dev is never null in this function, whether it looks not very reasonable?
>
That is correct with current code. In Chromium world, reviewers still
suggest to null check in similar situations.
Up to you. It is fine with me as it is.


> Best Regards,
> Yunfei Dong
>
> +       struct platform_device *pdev = dev->plat_dev;
> +       int reg_num, i;
> +
> +       /* Sizeof(u32) * 4 bytes for each register base. */
> +       reg_num = of_property_count_elems_of_size(pdev->dev.of_node, "reg",
> +               sizeof(u32) * 4);
> +       if (reg_num <= 0 || reg_num > NUM_MAX_VDEC_REG_BASE) {
> +               dev_err(&pdev->dev, "Invalid register property size:
> %d\n", reg_num);
> +               return -EINVAL;
> +       }
> +
> +       for (i = 0; i < reg_num; i++) {
> +               dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
> +               if (IS_ERR(dev->reg_base[i]))
> +                       return PTR_ERR(dev->reg_base[i]);
> +
> +               mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
> +       }
> +
> +       return 0;
> +}
> +
>  static int fops_vcodec_open(struct file *file)
>  {
>         struct mtk_vcodec_dev *dev = video_drvdata(file);
> @@ -206,7 +230,7 @@ static int mtk_vcodec_probe(struct platform_device
> *pdev)
>         struct resource *res;
>         phandle rproc_phandle;
>         enum mtk_vcodec_fw_type fw_type;
> -       int i, ret;
> +       int ret;
>
>         dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
>         if (!dev)
> @@ -238,14 +262,9 @@ static int mtk_vcodec_probe(struct platform_device
> *pdev)
>                 goto err_dec_pm;
>         }
>
> -       for (i = 0; i < NUM_MAX_VDEC_REG_BASE; i++) {
> -               dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
> -               if (IS_ERR((__force void *)dev->reg_base[i])) {
> -                       ret = PTR_ERR((__force void *)dev->reg_base[i]);
> -                       goto err_res;
> -               }
> -               mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
> -       }
> +       ret = mtk_vcodec_get_reg_bases(dev);
> +       if (ret)
> +               goto err_res;
>
>         res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
>         if (res == NULL) {
>
>
diff mbox series

Patch

diff --git a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
index e6e6a8203eeb..59caf2163349 100644
--- a/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
+++ b/drivers/media/platform/mtk-vcodec/mtk_vcodec_dec_drv.c
@@ -78,6 +78,30 @@  static irqreturn_t mtk_vcodec_dec_irq_handler(int irq, void *priv)
 	return IRQ_HANDLED;
 }
 
+static int mtk_vcodec_get_reg_bases(struct mtk_vcodec_dev *dev)
+{
+	struct platform_device *pdev = dev->plat_dev;
+	int reg_num, i;
+
+	/* Sizeof(u32) * 4 bytes for each register base. */
+	reg_num = of_property_count_elems_of_size(pdev->dev.of_node, "reg",
+		sizeof(u32) * 4);
+	if (reg_num <= 0 || reg_num > NUM_MAX_VDEC_REG_BASE) {
+		dev_err(&pdev->dev, "Invalid register property size: %d\n", reg_num);
+		return -EINVAL;
+	}
+
+	for (i = 0; i < reg_num; i++) {
+		dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
+		if (IS_ERR(dev->reg_base[i]))
+			return PTR_ERR(dev->reg_base[i]);
+
+		mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
+	}
+
+	return 0;
+}
+
 static int fops_vcodec_open(struct file *file)
 {
 	struct mtk_vcodec_dev *dev = video_drvdata(file);
@@ -206,7 +230,7 @@  static int mtk_vcodec_probe(struct platform_device *pdev)
 	struct resource *res;
 	phandle rproc_phandle;
 	enum mtk_vcodec_fw_type fw_type;
-	int i, ret;
+	int ret;
 
 	dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
 	if (!dev)
@@ -238,14 +262,9 @@  static int mtk_vcodec_probe(struct platform_device *pdev)
 		goto err_dec_pm;
 	}
 
-	for (i = 0; i < NUM_MAX_VDEC_REG_BASE; i++) {
-		dev->reg_base[i] = devm_platform_ioremap_resource(pdev, i);
-		if (IS_ERR((__force void *)dev->reg_base[i])) {
-			ret = PTR_ERR((__force void *)dev->reg_base[i]);
-			goto err_res;
-		}
-		mtk_v4l2_debug(2, "reg[%d] base=%p", i, dev->reg_base[i]);
-	}
+	ret = mtk_vcodec_get_reg_bases(dev);
+	if (ret)
+		goto err_res;
 
 	res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
 	if (res == NULL) {