remoteproc: imx_rproc: Call of_node_put() on iteration error

Function of_phandle_iterator_next() calls of_node_put() on the last
device_node it iterated over, but when the loop exits prematurely it has
to be called explicitly.

Fixes: b29b4249f8f0 ("remoteproc: imx_rproc: add i.MX specific parse fw hook")
Cc: stable@vger.kernel.org
Signed-off-by: Mathieu Poirier <mathieu.poirier@linaro.org>
Reviewed-by: Peng Fan <peng.fan@nxp.com>
Link: https://lore.kernel.org/r/20230320221826.2728078-5-mathieu.poirier@linaro.org
Signed-off-by: Mathieu Poirier <mathieu.poirier@linaro.org>

diff --git a/drivers/remoteproc/imx_rproc.c b/drivers/remoteproc/imx_rproc.c
index 9fc978e0393cee92bae2bd4261bc26d8d292eb5d..0ab840dc7e97f7d4bac652f81f4af1fbf6f7d88e 100644 (file)
--- a/drivers/remoteproc/imx_rproc.c
+++ b/drivers/remoteproc/imx_rproc.c
@@ -541,6 +541,7 @@ static int imx_rproc_prepare(struct rproc *rproc)
 
                rmem = of_reserved_mem_lookup(it.node);
                if (!rmem) {
+                       of_node_put(it.node);
                        dev_err(priv->dev, "unable to acquire memory-region\n");
                        return -EINVAL;
                }
@@ -553,10 +554,12 @@ static int imx_rproc_prepare(struct rproc *rproc)
                                           imx_rproc_mem_alloc, imx_rproc_mem_release,
                                           it.node->name);
 
-               if (mem)
+               if (mem) {
                        rproc_coredump_add_segment(rproc, da, rmem->size);
-               else
+               } else {
+                       of_node_put(it.node);
                        return -ENOMEM;
+               }
 
                rproc_add_carveout(rproc, mem);
        }