remoteproc: Clone rproc_ops in rproc_alloc()

In order to allow rproc_alloc() to, in a future patch, update entries in
the "ops" struct we need to make a local copy of it.

Reviewed-By: Loic Pallardy <loic.pallardy@st.com>
Tested-By: Loic Pallardy <loic.pallardy@st.com>
Signed-off-by: Bjorn Andersson <bjorn.andersson@linaro.org>

1 files changed, 8 insertions, 1 deletions

diff --git a/drivers/remoteproc/remoteproc_core.c b/drivers/remoteproc/remoteproc_core.c
index 208ccf709cad..dbf685dbafcf 100644
--- a/drivers/remoteproc/remoteproc_core.c
+++ b/drivers/remoteproc/remoteproc_core.c
@@ -1342,6 +1342,7 @@ static void rproc_type_release(struct device *dev)
                 ida_simple_remove(&rproc_dev_index, rproc->index);
 
         kfree(rproc->firmware);
+        kfree(rproc->ops);
         kfree(rproc);
 }
 
@@ -1406,9 +1407,15 @@ struct rproc *rproc_alloc(struct device *dev, const char *name,
                 return NULL;
         }
 
+        rproc->ops = kmemdup(ops, sizeof(*ops), GFP_KERNEL);
+        if (!rproc->ops) {
+                kfree(p);
+                kfree(rproc);
+                return NULL;
+        }
+
         rproc->firmware = p;
         rproc->name = name;
-        rproc->ops = ops;
         rproc->priv = &rproc[1];
         rproc->auto_boot = true;