clk: mediatek: Check if clock ID is larger than clk_hw_onecell_data size

  The MediaTek clock driver library's simple-probe mechanism allocates
clk_hw_onecell_data based on how many clocks are defined. If there's a
mismatch between that and the actual number of clocks defined in the DT
binding, such that a clock ID exceeds the number, then an out-of-bounds
write will occur. This silently corrupts memory, maybe causing a crash
later on that seems unrelated. KASAN can detect this if turned on.

To avoid getting into said scenario, check if the to be registered
clock's ID will fit in the allocated clk_hw_onecell_data. If not, put
out a big warning and skip the clock.

One could argue that the proper fix is to let the drivers specify the
number of clocks (based on a DT binding macro) instead. However even
the DT binding macro could be wrong. And having code to catch errors
and give out warnings is better than having silent corruption.

Signed-off-by: Chen-Yu Tsai <wenst@chromium.org>
---
This one is less urgent.

Angelo, do you think we should add a field to struct mtk_clk_desc and
assign CLK_XXX_NR_CLK to it?

 drivers/clk/mediatek/clk-gate.c | 11 +++++++++
 drivers/clk/mediatek/clk-mtk.c  | 43 +++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
  

Il 19/07/23 10:20, Chen-Yu Tsai ha scritto:
> The MediaTek clock driver library's simple-probe mechanism allocates
> clk_hw_onecell_data based on how many clocks are defined. If there's a
> mismatch between that and the actual number of clocks defined in the DT
> binding, such that a clock ID exceeds the number, then an out-of-bounds
> write will occur. This silently corrupts memory, maybe causing a crash
> later on that seems unrelated. KASAN can detect this if turned on.
> 
> To avoid getting into said scenario, check if the to be registered
> clock's ID will fit in the allocated clk_hw_onecell_data. If not, put
> out a big warning and skip the clock.
> 
> One could argue that the proper fix is to let the drivers specify the
> number of clocks (based on a DT binding macro) instead. However even
> the DT binding macro could be wrong. And having code to catch errors
> and give out warnings is better than having silent corruption.
> 
> Signed-off-by: Chen-Yu Tsai <wenst@chromium.org>
> ---
> This one is less urgent.
> 
> Angelo, do you think we should add a field to struct mtk_clk_desc and
> assign CLK_XXX_NR_CLK to it?
> 

I get the point... but I don't know if it is a good idea to add checks for
*bad code* in the first place, as bad code shall not happen at all.
Validating whether a developer wrote the right thing is something that should
be done in code reviews, and such mistakes shouldn't be allowed to happen.

Besides, if you really want to go this route... In that case I disagree about
the `continue`, as I would be inflexible: if your code is bad, I will refuse
to register your clocks entirely.
That'll force the developer to actually fix it, as parts of the SoC and/or
platform will *not work at all* :-)

So, in that case...

	if (rc->id >= clk_data->num) {
		hw = PTR_ERR(-EINVAL);
		goto err;
	}

Thoughts?

Cheers!
Angelo

>   drivers/clk/mediatek/clk-gate.c | 11 +++++++++
>   drivers/clk/mediatek/clk-mtk.c  | 43 +++++++++++++++++++++++++++++++++
>   2 files changed, 54 insertions(+)
> 
> diff --git a/drivers/clk/mediatek/clk-gate.c b/drivers/clk/mediatek/clk-gate.c
> index 67d9e741c5e7..bb7c536ef60f 100644
> --- a/drivers/clk/mediatek/clk-gate.c
> +++ b/drivers/clk/mediatek/clk-gate.c
> @@ -222,6 +222,11 @@ int mtk_clk_register_gates(struct device *dev, struct device_node *node,
>   	for (i = 0; i < num; i++) {
>   		const struct mtk_gate *gate = &clks[i];
>   
> +		if (WARN(gate->id >= clk_data->num,
> +			 "%pOF: gateclock ID (%d)larger than expected (%d)\n",
> +			 node, gate->id, clk_data->num))
> +			continue;
> +
>   		if (!IS_ERR_OR_NULL(clk_data->hws[gate->id])) {
>   			pr_warn("%pOF: Trying to register duplicate clock ID: %d\n",
>   				node, gate->id);
> @@ -251,6 +256,9 @@ int mtk_clk_register_gates(struct device *dev, struct device_node *node,
>   	while (--i >= 0) {
>   		const struct mtk_gate *gate = &clks[i];
>   
> +		if (gate->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
>   			continue;
>   
> @@ -273,6 +281,9 @@ void mtk_clk_unregister_gates(const struct mtk_gate *clks, int num,
>   	for (i = num; i > 0; i--) {
>   		const struct mtk_gate *gate = &clks[i - 1];
>   
> +		if (gate->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
>   			continue;
>   
> diff --git a/drivers/clk/mediatek/clk-mtk.c b/drivers/clk/mediatek/clk-mtk.c
> index 2e55368dc4d8..09d50a15db77 100644
> --- a/drivers/clk/mediatek/clk-mtk.c
> +++ b/drivers/clk/mediatek/clk-mtk.c
> @@ -94,6 +94,10 @@ int mtk_clk_register_fixed_clks(const struct mtk_fixed_clk *clks, int num,
>   	for (i = 0; i < num; i++) {
>   		const struct mtk_fixed_clk *rc = &clks[i];
>   
> +		if (WARN(rc->id >= clk_data->num,
> +			 "Fixed clock ID (%d) larger than expected (%d)\n", rc->id, clk_data->num))
> +			continue;
> +
>   		if (!IS_ERR_OR_NULL(clk_data->hws[rc->id])) {
>   			pr_warn("Trying to register duplicate clock ID: %d\n", rc->id);
>   			continue;
> @@ -117,6 +121,9 @@ int mtk_clk_register_fixed_clks(const struct mtk_fixed_clk *clks, int num,
>   	while (--i >= 0) {
>   		const struct mtk_fixed_clk *rc = &clks[i];
>   
> +		if (rc->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[rc->id]))
>   			continue;
>   
> @@ -139,6 +146,9 @@ void mtk_clk_unregister_fixed_clks(const struct mtk_fixed_clk *clks, int num,
>   	for (i = num; i > 0; i--) {
>   		const struct mtk_fixed_clk *rc = &clks[i - 1];
>   
> +		if (rc->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[rc->id]))
>   			continue;
>   
> @@ -160,6 +170,11 @@ int mtk_clk_register_factors(const struct mtk_fixed_factor *clks, int num,
>   	for (i = 0; i < num; i++) {
>   		const struct mtk_fixed_factor *ff = &clks[i];
>   
> +		if (WARN(ff->id >= clk_data->num,
> +			 "Fixed factor clock ID (%d) larger than expected (%d)\n",
> +			 ff->id, clk_data->num))
> +			continue;
> +
>   		if (!IS_ERR_OR_NULL(clk_data->hws[ff->id])) {
>   			pr_warn("Trying to register duplicate clock ID: %d\n", ff->id);
>   			continue;
> @@ -183,6 +198,9 @@ int mtk_clk_register_factors(const struct mtk_fixed_factor *clks, int num,
>   	while (--i >= 0) {
>   		const struct mtk_fixed_factor *ff = &clks[i];
>   
> +		if (ff->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[ff->id]))
>   			continue;
>   
> @@ -205,6 +223,9 @@ void mtk_clk_unregister_factors(const struct mtk_fixed_factor *clks, int num,
>   	for (i = num; i > 0; i--) {
>   		const struct mtk_fixed_factor *ff = &clks[i - 1];
>   
> +		if (ff->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[ff->id]))
>   			continue;
>   
> @@ -339,6 +360,11 @@ int mtk_clk_register_composites(struct device *dev,
>   	for (i = 0; i < num; i++) {
>   		const struct mtk_composite *mc = &mcs[i];
>   
> +		if (WARN(mc->id >= clk_data->num,
> +			 "Composite clock ID (%d) larger than expected (%d)\n",
> +			 mc->id, clk_data->num))
> +			continue;
> +
>   		if (!IS_ERR_OR_NULL(clk_data->hws[mc->id])) {
>   			pr_warn("Trying to register duplicate clock ID: %d\n",
>   				mc->id);
> @@ -362,6 +388,9 @@ int mtk_clk_register_composites(struct device *dev,
>   	while (--i >= 0) {
>   		const struct mtk_composite *mc = &mcs[i];
>   
> +		if (mc->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[mcs->id]))
>   			continue;
>   
> @@ -384,6 +413,9 @@ void mtk_clk_unregister_composites(const struct mtk_composite *mcs, int num,
>   	for (i = num; i > 0; i--) {
>   		const struct mtk_composite *mc = &mcs[i - 1];
>   
> +		if (mc->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[mc->id]))
>   			continue;
>   
> @@ -407,6 +439,11 @@ int mtk_clk_register_dividers(struct device *dev,
>   	for (i = 0; i <  num; i++) {
>   		const struct mtk_clk_divider *mcd = &mcds[i];
>   
> +		if (WARN(mcd->id >= clk_data->num,
> +			 "Divider clock ID (%d) larger than expected (%d)\n",
> +			 mcd->id, clk_data->num))
> +			continue;
> +
>   		if (!IS_ERR_OR_NULL(clk_data->hws[mcd->id])) {
>   			pr_warn("Trying to register duplicate clock ID: %d\n",
>   				mcd->id);
> @@ -432,6 +469,9 @@ int mtk_clk_register_dividers(struct device *dev,
>   	while (--i >= 0) {
>   		const struct mtk_clk_divider *mcd = &mcds[i];
>   
> +		if (mcd->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[mcd->id]))
>   			continue;
>   
> @@ -454,6 +494,9 @@ void mtk_clk_unregister_dividers(const struct mtk_clk_divider *mcds, int num,
>   	for (i = num; i > 0; i--) {
>   		const struct mtk_clk_divider *mcd = &mcds[i - 1];
>   
> +		if (mcd->id >= clk_data->num)
> +			continue;
> +
>   		if (IS_ERR_OR_NULL(clk_data->hws[mcd->id]))
>   			continue;
>
  

On Wed, Jul 19, 2023 at 4:36 PM AngeloGioacchino Del Regno
<angelogioacchino.delregno@collabora.com> wrote:
>
> Il 19/07/23 10:20, Chen-Yu Tsai ha scritto:
> > The MediaTek clock driver library's simple-probe mechanism allocates
> > clk_hw_onecell_data based on how many clocks are defined. If there's a
> > mismatch between that and the actual number of clocks defined in the DT
> > binding, such that a clock ID exceeds the number, then an out-of-bounds
> > write will occur. This silently corrupts memory, maybe causing a crash
> > later on that seems unrelated. KASAN can detect this if turned on.
> >
> > To avoid getting into said scenario, check if the to be registered
> > clock's ID will fit in the allocated clk_hw_onecell_data. If not, put
> > out a big warning and skip the clock.
> >
> > One could argue that the proper fix is to let the drivers specify the
> > number of clocks (based on a DT binding macro) instead. However even
> > the DT binding macro could be wrong. And having code to catch errors
> > and give out warnings is better than having silent corruption.
> >
> > Signed-off-by: Chen-Yu Tsai <wenst@chromium.org>
> > ---
> > This one is less urgent.
> >
> > Angelo, do you think we should add a field to struct mtk_clk_desc and
> > assign CLK_XXX_NR_CLK to it?
> >
>
> I get the point... but I don't know if it is a good idea to add checks for
> *bad code* in the first place, as bad code shall not happen at all.
> Validating whether a developer wrote the right thing is something that should
> be done in code reviews, and such mistakes shouldn't be allowed to happen.

In theory, yeah. In practice, well, we already have such an error in tree. :p

> Besides, if you really want to go this route... In that case I disagree about
> the `continue`, as I would be inflexible: if your code is bad, I will refuse
> to register your clocks entirely.
> That'll force the developer to actually fix it, as parts of the SoC and/or
> platform will *not work at all* :-)
>
> So, in that case...
>
>         if (rc->id >= clk_data->num) {
>                 hw = PTR_ERR(-EINVAL);
>                 goto err;
>         }
>
> Thoughts?

That works for me as well. This was more my debug code cleaned up.
I assume we still want the warning / error message though?

The other choice would be adding .num_clks for CLK_XXX_NR_CLKS to
mtk_clk_desc. What are your thoughts on that?

ChenYu

> Cheers!
> Angelo
>
> >   drivers/clk/mediatek/clk-gate.c | 11 +++++++++
> >   drivers/clk/mediatek/clk-mtk.c  | 43 +++++++++++++++++++++++++++++++++
> >   2 files changed, 54 insertions(+)
> >
> > diff --git a/drivers/clk/mediatek/clk-gate.c b/drivers/clk/mediatek/clk-gate.c
> > index 67d9e741c5e7..bb7c536ef60f 100644
> > --- a/drivers/clk/mediatek/clk-gate.c
> > +++ b/drivers/clk/mediatek/clk-gate.c
> > @@ -222,6 +222,11 @@ int mtk_clk_register_gates(struct device *dev, struct device_node *node,
> >       for (i = 0; i < num; i++) {
> >               const struct mtk_gate *gate = &clks[i];
> >
> > +             if (WARN(gate->id >= clk_data->num,
> > +                      "%pOF: gateclock ID (%d)larger than expected (%d)\n",
> > +                      node, gate->id, clk_data->num))
> > +                     continue;
> > +
> >               if (!IS_ERR_OR_NULL(clk_data->hws[gate->id])) {
> >                       pr_warn("%pOF: Trying to register duplicate clock ID: %d\n",
> >                               node, gate->id);
> > @@ -251,6 +256,9 @@ int mtk_clk_register_gates(struct device *dev, struct device_node *node,
> >       while (--i >= 0) {
> >               const struct mtk_gate *gate = &clks[i];
> >
> > +             if (gate->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
> >                       continue;
> >
> > @@ -273,6 +281,9 @@ void mtk_clk_unregister_gates(const struct mtk_gate *clks, int num,
> >       for (i = num; i > 0; i--) {
> >               const struct mtk_gate *gate = &clks[i - 1];
> >
> > +             if (gate->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
> >                       continue;
> >
> > diff --git a/drivers/clk/mediatek/clk-mtk.c b/drivers/clk/mediatek/clk-mtk.c
> > index 2e55368dc4d8..09d50a15db77 100644
> > --- a/drivers/clk/mediatek/clk-mtk.c
> > +++ b/drivers/clk/mediatek/clk-mtk.c
> > @@ -94,6 +94,10 @@ int mtk_clk_register_fixed_clks(const struct mtk_fixed_clk *clks, int num,
> >       for (i = 0; i < num; i++) {
> >               const struct mtk_fixed_clk *rc = &clks[i];
> >
> > +             if (WARN(rc->id >= clk_data->num,
> > +                      "Fixed clock ID (%d) larger than expected (%d)\n", rc->id, clk_data->num))
> > +                     continue;
> > +
> >               if (!IS_ERR_OR_NULL(clk_data->hws[rc->id])) {
> >                       pr_warn("Trying to register duplicate clock ID: %d\n", rc->id);
> >                       continue;
> > @@ -117,6 +121,9 @@ int mtk_clk_register_fixed_clks(const struct mtk_fixed_clk *clks, int num,
> >       while (--i >= 0) {
> >               const struct mtk_fixed_clk *rc = &clks[i];
> >
> > +             if (rc->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[rc->id]))
> >                       continue;
> >
> > @@ -139,6 +146,9 @@ void mtk_clk_unregister_fixed_clks(const struct mtk_fixed_clk *clks, int num,
> >       for (i = num; i > 0; i--) {
> >               const struct mtk_fixed_clk *rc = &clks[i - 1];
> >
> > +             if (rc->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[rc->id]))
> >                       continue;
> >
> > @@ -160,6 +170,11 @@ int mtk_clk_register_factors(const struct mtk_fixed_factor *clks, int num,
> >       for (i = 0; i < num; i++) {
> >               const struct mtk_fixed_factor *ff = &clks[i];
> >
> > +             if (WARN(ff->id >= clk_data->num,
> > +                      "Fixed factor clock ID (%d) larger than expected (%d)\n",
> > +                      ff->id, clk_data->num))
> > +                     continue;
> > +
> >               if (!IS_ERR_OR_NULL(clk_data->hws[ff->id])) {
> >                       pr_warn("Trying to register duplicate clock ID: %d\n", ff->id);
> >                       continue;
> > @@ -183,6 +198,9 @@ int mtk_clk_register_factors(const struct mtk_fixed_factor *clks, int num,
> >       while (--i >= 0) {
> >               const struct mtk_fixed_factor *ff = &clks[i];
> >
> > +             if (ff->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[ff->id]))
> >                       continue;
> >
> > @@ -205,6 +223,9 @@ void mtk_clk_unregister_factors(const struct mtk_fixed_factor *clks, int num,
> >       for (i = num; i > 0; i--) {
> >               const struct mtk_fixed_factor *ff = &clks[i - 1];
> >
> > +             if (ff->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[ff->id]))
> >                       continue;
> >
> > @@ -339,6 +360,11 @@ int mtk_clk_register_composites(struct device *dev,
> >       for (i = 0; i < num; i++) {
> >               const struct mtk_composite *mc = &mcs[i];
> >
> > +             if (WARN(mc->id >= clk_data->num,
> > +                      "Composite clock ID (%d) larger than expected (%d)\n",
> > +                      mc->id, clk_data->num))
> > +                     continue;
> > +
> >               if (!IS_ERR_OR_NULL(clk_data->hws[mc->id])) {
> >                       pr_warn("Trying to register duplicate clock ID: %d\n",
> >                               mc->id);
> > @@ -362,6 +388,9 @@ int mtk_clk_register_composites(struct device *dev,
> >       while (--i >= 0) {
> >               const struct mtk_composite *mc = &mcs[i];
> >
> > +             if (mc->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[mcs->id]))
> >                       continue;
> >
> > @@ -384,6 +413,9 @@ void mtk_clk_unregister_composites(const struct mtk_composite *mcs, int num,
> >       for (i = num; i > 0; i--) {
> >               const struct mtk_composite *mc = &mcs[i - 1];
> >
> > +             if (mc->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[mc->id]))
> >                       continue;
> >
> > @@ -407,6 +439,11 @@ int mtk_clk_register_dividers(struct device *dev,
> >       for (i = 0; i <  num; i++) {
> >               const struct mtk_clk_divider *mcd = &mcds[i];
> >
> > +             if (WARN(mcd->id >= clk_data->num,
> > +                      "Divider clock ID (%d) larger than expected (%d)\n",
> > +                      mcd->id, clk_data->num))
> > +                     continue;
> > +
> >               if (!IS_ERR_OR_NULL(clk_data->hws[mcd->id])) {
> >                       pr_warn("Trying to register duplicate clock ID: %d\n",
> >                               mcd->id);
> > @@ -432,6 +469,9 @@ int mtk_clk_register_dividers(struct device *dev,
> >       while (--i >= 0) {
> >               const struct mtk_clk_divider *mcd = &mcds[i];
> >
> > +             if (mcd->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[mcd->id]))
> >                       continue;
> >
> > @@ -454,6 +494,9 @@ void mtk_clk_unregister_dividers(const struct mtk_clk_divider *mcds, int num,
> >       for (i = num; i > 0; i--) {
> >               const struct mtk_clk_divider *mcd = &mcds[i - 1];
> >
> > +             if (mcd->id >= clk_data->num)
> > +                     continue;
> > +
> >               if (IS_ERR_OR_NULL(clk_data->hws[mcd->id]))
> >                       continue;
> >
>
  

Il 19/07/23 10:52, Chen-Yu Tsai ha scritto:
> On Wed, Jul 19, 2023 at 4:36 PM AngeloGioacchino Del Regno
> <angelogioacchino.delregno@collabora.com> wrote:
>>
>> Il 19/07/23 10:20, Chen-Yu Tsai ha scritto:
>>> The MediaTek clock driver library's simple-probe mechanism allocates
>>> clk_hw_onecell_data based on how many clocks are defined. If there's a
>>> mismatch between that and the actual number of clocks defined in the DT
>>> binding, such that a clock ID exceeds the number, then an out-of-bounds
>>> write will occur. This silently corrupts memory, maybe causing a crash
>>> later on that seems unrelated. KASAN can detect this if turned on.
>>>
>>> To avoid getting into said scenario, check if the to be registered
>>> clock's ID will fit in the allocated clk_hw_onecell_data. If not, put
>>> out a big warning and skip the clock.
>>>
>>> One could argue that the proper fix is to let the drivers specify the
>>> number of clocks (based on a DT binding macro) instead. However even
>>> the DT binding macro could be wrong. And having code to catch errors
>>> and give out warnings is better than having silent corruption.
>>>
>>> Signed-off-by: Chen-Yu Tsai <wenst@chromium.org>
>>> ---
>>> This one is less urgent.
>>>
>>> Angelo, do you think we should add a field to struct mtk_clk_desc and
>>> assign CLK_XXX_NR_CLK to it?
>>>
>>
>> I get the point... but I don't know if it is a good idea to add checks for
>> *bad code* in the first place, as bad code shall not happen at all.
>> Validating whether a developer wrote the right thing is something that should
>> be done in code reviews, and such mistakes shouldn't be allowed to happen.
> 
> In theory, yeah. In practice, well, we already have such an error in tree. :p
> 

Yeah, I realize that reviews are done by humans, and humans naturally make
mistakes, which is fine.

I wonder if there is any way to solve this with a build time error... I didn't
try and didn't look though - so I'm just thinking out loud.

>> Besides, if you really want to go this route... In that case I disagree about
>> the `continue`, as I would be inflexible: if your code is bad, I will refuse
>> to register your clocks entirely.
>> That'll force the developer to actually fix it, as parts of the SoC and/or
>> platform will *not work at all* :-)
>>
>> So, in that case...
>>
>>          if (rc->id >= clk_data->num) {
>>                  hw = PTR_ERR(-EINVAL);
>>                  goto err;
>>          }
>>
>> Thoughts?
> 
> That works for me as well. This was more my debug code cleaned up.
> I assume we still want the warning / error message though?
> 

I don't have any strong opinions on this specific matter. My only concern is
that we'd increase the kernel size with those prints that should never happen
anyway...

...but again, I'll leave that choice to you - no strong opinions.

> The other choice would be adding .num_clks for CLK_XXX_NR_CLKS to
> mtk_clk_desc. What are your thoughts on that?
> 

A u16 wouldn't waste all that much memory, so that could cover the case in
which someone specifies *less* clocks than normal.
That'd be something necessary if we're really following that validation route,
as the actual validation would be *incomplete* without that, I think.

> ChenYu
> 
>> Cheers!
>> Angelo
>>
>>>    drivers/clk/mediatek/clk-gate.c | 11 +++++++++
>>>    drivers/clk/mediatek/clk-mtk.c  | 43 +++++++++++++++++++++++++++++++++
>>>    2 files changed, 54 insertions(+)
>>>
>>> diff --git a/drivers/clk/mediatek/clk-gate.c b/drivers/clk/mediatek/clk-gate.c
>>> index 67d9e741c5e7..bb7c536ef60f 100644
>>> --- a/drivers/clk/mediatek/clk-gate.c
>>> +++ b/drivers/clk/mediatek/clk-gate.c
>>> @@ -222,6 +222,11 @@ int mtk_clk_register_gates(struct device *dev, struct device_node *node,
>>>        for (i = 0; i < num; i++) {
>>>                const struct mtk_gate *gate = &clks[i];
>>>
>>> +             if (WARN(gate->id >= clk_data->num,
>>> +                      "%pOF: gateclock ID (%d)larger than expected (%d)\n",
>>> +                      node, gate->id, clk_data->num))
>>> +                     continue;
>>> +
>>>                if (!IS_ERR_OR_NULL(clk_data->hws[gate->id])) {
>>>                        pr_warn("%pOF: Trying to register duplicate clock ID: %d\n",
>>>                                node, gate->id);
>>> @@ -251,6 +256,9 @@ int mtk_clk_register_gates(struct device *dev, struct device_node *node,
>>>        while (--i >= 0) {
>>>                const struct mtk_gate *gate = &clks[i];
>>>
>>> +             if (gate->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
>>>                        continue;
>>>
>>> @@ -273,6 +281,9 @@ void mtk_clk_unregister_gates(const struct mtk_gate *clks, int num,
>>>        for (i = num; i > 0; i--) {
>>>                const struct mtk_gate *gate = &clks[i - 1];
>>>
>>> +             if (gate->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[gate->id]))
>>>                        continue;
>>>
>>> diff --git a/drivers/clk/mediatek/clk-mtk.c b/drivers/clk/mediatek/clk-mtk.c
>>> index 2e55368dc4d8..09d50a15db77 100644
>>> --- a/drivers/clk/mediatek/clk-mtk.c
>>> +++ b/drivers/clk/mediatek/clk-mtk.c
>>> @@ -94,6 +94,10 @@ int mtk_clk_register_fixed_clks(const struct mtk_fixed_clk *clks, int num,
>>>        for (i = 0; i < num; i++) {
>>>                const struct mtk_fixed_clk *rc = &clks[i];
>>>
>>> +             if (WARN(rc->id >= clk_data->num,
>>> +                      "Fixed clock ID (%d) larger than expected (%d)\n", rc->id, clk_data->num))
>>> +                     continue;
>>> +
>>>                if (!IS_ERR_OR_NULL(clk_data->hws[rc->id])) {
>>>                        pr_warn("Trying to register duplicate clock ID: %d\n", rc->id);
>>>                        continue;
>>> @@ -117,6 +121,9 @@ int mtk_clk_register_fixed_clks(const struct mtk_fixed_clk *clks, int num,
>>>        while (--i >= 0) {
>>>                const struct mtk_fixed_clk *rc = &clks[i];
>>>
>>> +             if (rc->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[rc->id]))
>>>                        continue;
>>>
>>> @@ -139,6 +146,9 @@ void mtk_clk_unregister_fixed_clks(const struct mtk_fixed_clk *clks, int num,
>>>        for (i = num; i > 0; i--) {
>>>                const struct mtk_fixed_clk *rc = &clks[i - 1];
>>>
>>> +             if (rc->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[rc->id]))
>>>                        continue;
>>>
>>> @@ -160,6 +170,11 @@ int mtk_clk_register_factors(const struct mtk_fixed_factor *clks, int num,
>>>        for (i = 0; i < num; i++) {
>>>                const struct mtk_fixed_factor *ff = &clks[i];
>>>
>>> +             if (WARN(ff->id >= clk_data->num,
>>> +                      "Fixed factor clock ID (%d) larger than expected (%d)\n",
>>> +                      ff->id, clk_data->num))
>>> +                     continue;
>>> +
>>>                if (!IS_ERR_OR_NULL(clk_data->hws[ff->id])) {
>>>                        pr_warn("Trying to register duplicate clock ID: %d\n", ff->id);
>>>                        continue;
>>> @@ -183,6 +198,9 @@ int mtk_clk_register_factors(const struct mtk_fixed_factor *clks, int num,
>>>        while (--i >= 0) {
>>>                const struct mtk_fixed_factor *ff = &clks[i];
>>>
>>> +             if (ff->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[ff->id]))
>>>                        continue;
>>>
>>> @@ -205,6 +223,9 @@ void mtk_clk_unregister_factors(const struct mtk_fixed_factor *clks, int num,
>>>        for (i = num; i > 0; i--) {
>>>                const struct mtk_fixed_factor *ff = &clks[i - 1];
>>>
>>> +             if (ff->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[ff->id]))
>>>                        continue;
>>>
>>> @@ -339,6 +360,11 @@ int mtk_clk_register_composites(struct device *dev,
>>>        for (i = 0; i < num; i++) {
>>>                const struct mtk_composite *mc = &mcs[i];
>>>
>>> +             if (WARN(mc->id >= clk_data->num,
>>> +                      "Composite clock ID (%d) larger than expected (%d)\n",
>>> +                      mc->id, clk_data->num))
>>> +                     continue;
>>> +
>>>                if (!IS_ERR_OR_NULL(clk_data->hws[mc->id])) {
>>>                        pr_warn("Trying to register duplicate clock ID: %d\n",
>>>                                mc->id);
>>> @@ -362,6 +388,9 @@ int mtk_clk_register_composites(struct device *dev,
>>>        while (--i >= 0) {
>>>                const struct mtk_composite *mc = &mcs[i];
>>>
>>> +             if (mc->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[mcs->id]))
>>>                        continue;
>>>
>>> @@ -384,6 +413,9 @@ void mtk_clk_unregister_composites(const struct mtk_composite *mcs, int num,
>>>        for (i = num; i > 0; i--) {
>>>                const struct mtk_composite *mc = &mcs[i - 1];
>>>
>>> +             if (mc->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[mc->id]))
>>>                        continue;
>>>
>>> @@ -407,6 +439,11 @@ int mtk_clk_register_dividers(struct device *dev,
>>>        for (i = 0; i <  num; i++) {
>>>                const struct mtk_clk_divider *mcd = &mcds[i];
>>>
>>> +             if (WARN(mcd->id >= clk_data->num,
>>> +                      "Divider clock ID (%d) larger than expected (%d)\n",
>>> +                      mcd->id, clk_data->num))
>>> +                     continue;
>>> +
>>>                if (!IS_ERR_OR_NULL(clk_data->hws[mcd->id])) {
>>>                        pr_warn("Trying to register duplicate clock ID: %d\n",
>>>                                mcd->id);
>>> @@ -432,6 +469,9 @@ int mtk_clk_register_dividers(struct device *dev,
>>>        while (--i >= 0) {
>>>                const struct mtk_clk_divider *mcd = &mcds[i];
>>>
>>> +             if (mcd->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[mcd->id]))
>>>                        continue;
>>>
>>> @@ -454,6 +494,9 @@ void mtk_clk_unregister_dividers(const struct mtk_clk_divider *mcds, int num,
>>>        for (i = num; i > 0; i--) {
>>>                const struct mtk_clk_divider *mcd = &mcds[i - 1];
>>>
>>> +             if (mcd->id >= clk_data->num)
>>> +                     continue;
>>> +
>>>                if (IS_ERR_OR_NULL(clk_data->hws[mcd->id]))
>>>                        continue;
>>>
>>