lightnvm: free reverse device map

Free the reverse mapping table correctly on target tear down

Signed-off-by: Javier González <javier@cnexlabs.com>
Signed-off-by: Matias Bjørling <matias@cnexlabs.com>
Signed-off-by: Jens Axboe <axboe@fb.com>

1 files changed, 13 insertions, 1 deletions

diff --git a/drivers/lightnvm/core.c b/drivers/lightnvm/core.c
index 95105c47e082..a14c52c24e23 100644
--- a/drivers/lightnvm/core.c
+++ b/drivers/lightnvm/core.c
@@ -411,6 +411,18 @@ err_rmap:
         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;
@@ -992,7 +1004,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);