1 files changed, 6 insertions, 3 deletions

diff --git a/drivers/uio/uio_dmem_genirq.c b/drivers/uio/uio_dmem_genirq.c
index 4d4dd008c8b9..d8bbe0783cdc 100644
--- a/drivers/uio/uio_dmem_genirq.c
+++ b/drivers/uio/uio_dmem_genirq.c
@@ -37,6 +37,7 @@ struct uio_dmem_genirq_platdata {
         struct platform_device *pdev;
         unsigned int dmem_region_start;
         unsigned int num_dmem_regions;
+        void *dmem_region_vaddr[MAX_UIO_MAPS];
         struct mutex alloc_lock;
         unsigned int refcnt;
 };
@@ -46,6 +47,7 @@ static int uio_dmem_genirq_open(struct uio_info *info, struct inode *inode)
         struct uio_dmem_genirq_platdata *priv = info->priv;
         struct uio_mem *uiomem;
         int ret = 0;
+        int dmem_region = priv->dmem_region_start;
 
         uiomem = &priv->uioinfo->mem[priv->dmem_region_start];
 
@@ -61,8 +63,7 @@ static int uio_dmem_genirq_open(struct uio_info *info, struct inode *inode)
                         ret = -ENOMEM;
                         break;
                 }
-
-                uiomem->internal_addr = addr;
+                priv->dmem_region_vaddr[dmem_region++] = addr;
                 ++uiomem;
         }
         priv->refcnt++;
@@ -77,6 +78,7 @@ static int uio_dmem_genirq_release(struct uio_info *info, struct inode *inode)
 {
         struct uio_dmem_genirq_platdata *priv = info->priv;
         struct uio_mem *uiomem;
+        int dmem_region = priv->dmem_region_start;
 
         /* Tell the Runtime PM code that the device has become idle */
         pm_runtime_put_sync(&priv->pdev->dev);
@@ -91,7 +93,8 @@ static int uio_dmem_genirq_release(struct uio_info *info, struct inode *inode)
                         break;
 
                 dma_free_coherent(&priv->pdev->dev, uiomem->size,
-                                uiomem->internal_addr, uiomem->addr);
+                                priv->dmem_region_vaddr[dmem_region++],
+                                uiomem->addr);
                 uiomem->addr = DMA_ERROR_CODE;
                 ++uiomem;
         }