msm: msm_fb: Add remove function for platform driver for clean unloading

This adds a  remove function to platform driver structure so that
resources are released when driver is unloaded.

Signed-off-by: Pramod Gurav <pramod.gurav@smartplayin.com>

CC: Jean-Christophe Plagniol-Villard <plagnioj@jcrosoft.com>
CC: Tomi Valkeinen <tomi.valkeinen@ti.com>
CC: Stephen Boyd <sboyd@codeaurora.org>
CC: Jingoo Han <jg1.han@samsung.com>
CC: Rob Clark <robdclark@gmail.com>
Signed-off-by: Tomi Valkeinen <tomi.valkeinen@ti.com>

1 files changed, 16 insertions, 0 deletions

diff --git a/drivers/video/fbdev/msm/msm_fb.c b/drivers/video/fbdev/msm/msm_fb.c
index 1374803fbcd9..4640188ffd6e 100644
--- a/drivers/video/fbdev/msm/msm_fb.c
+++ b/drivers/video/fbdev/msm/msm_fb.c
@@ -589,6 +589,8 @@ static int msmfb_probe(struct platform_device *pdev)
 
         msmfb->sleeping = WAKING;
 
+        platform_set_drvdata(pdev, msmfb);
+
         return 0;
 
 error_register_framebuffer:
@@ -598,9 +600,23 @@ error_setup_fbmem:
         return ret;
 }
 
+static int msmfb_remove(struct platform_device *pdev)
+{
+        struct msmfb_info *msmfb;
+
+        msmfb = platform_get_drvdata(pdev);
+
+        unregister_framebuffer(msmfb->fb);
+        iounmap(msmfb->fb->screen_base);
+        framebuffer_release(msmfb->fb);
+
+        return 0;
+}
+
 static struct platform_driver msm_panel_driver = {
         /* need to write remove */
         .probe = msmfb_probe,
+        .remove = msmfb_remove,
         .driver = {.name = "msm_panel"},
 };