Free the reverse mapping table correctly on target tear down

Signed-off-by: Javier González <jav...@cnexlabs.com>
---
 drivers/lightnvm/core.c | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/drivers/lightnvm/core.c b/drivers/lightnvm/core.c
index ca48792..f1cb485 100644
--- a/drivers/lightnvm/core.c
+++ b/drivers/lightnvm/core.c
@@ -413,6 +413,18 @@ static int nvm_register_map(struct nvm_dev *dev)
        return -ENOMEM;
 }
 
+static void nvm_unregister_map(struct nvm_dev *dev)
+{
+       struct nvm_dev_map *rmap = dev->rmap;
+       int i;
+
+       for (i = 0; i < dev->geo.nr_chnls; i++)
+               kfree(rmap->chnls[i].lun_offs);
+
+       kfree(rmap->chnls);
+       kfree(rmap);
+}
+
 static void nvm_map_to_dev(struct nvm_tgt_dev *tgt_dev, struct ppa_addr *p)
 {
        struct nvm_dev_map *dev_map = tgt_dev->map;
@@ -994,7 +1006,7 @@ void nvm_free(struct nvm_dev *dev)
        if (dev->dma_pool)
                dev->ops->destroy_dma_pool(dev->dma_pool);
 
-       kfree(dev->rmap);
+       nvm_unregister_map(dev);
        kfree(dev->lptbl);
        kfree(dev->lun_map);
        kfree(dev);
-- 
2.7.4

Reply via email to