remoteproc plat: init shmem if required

memory map the shared memory if shared memory is provided.

Signed-off-by: Wendy Liang <jliang@xilinx.com>

diff --git a/lib/remoteproc/remoteproc_plat.c b/lib/remoteproc/remoteproc_plat.c
index f49cbe24c4835990f96083a27357a17dd9984aeb..decfab0dc25d292373a9e31bd58e2b12afe91f3f 100644 (file)
--- a/lib/remoteproc/remoteproc_plat.c
+++ b/lib/remoteproc/remoteproc_plat.c
@@ -127,6 +127,13 @@ int rproc_init_plat_data(void *pdata,
                        if (ret)
                                return ret;
                        proc->sh_buff.io = io;
+                       if (io->size != (size_t)(-1)) {
+                               metal_io_mem_map(
+                                       metal_io_virt_to_phys(io, io->virt),
+                                       io, io->size);
+                               proc->sh_buff.start_addr = io->virt;
+                               proc->sh_buff.size = io->size;
+                       }
                        pdata += sizeof(struct plat_shm);
                } else if (type == PLAT_RSC_RPMSG_CHANNEL) {
                        struct plat_rpmsg_chnl *pchl =
@@ -152,10 +159,14 @@ void rproc_close_plat(struct hil_proc *proc)
        for (i = 0; i < num_vrings; i++) {
                vring = &proc->vdev.vring_info[i];
                dev = vring->dev;
-               if (dev)
+               if (dev) {
                        metal_device_close(dev);
+                       vring->dev = NULL;
+               }
                dev = vring->intr_info.dev;
-               if (dev)
+               if (dev) {
                        metal_device_close(dev);
+                       vring->intr_info.dev = NULL;
+               }
        }
 }