staging: ozwpan: Increment reference counter.

Increment PD reference counter, on every timer event so that
we do not loose PD object by mistake.

Signed-off-by: Rupesh Gujare <rupesh.gujare@atmel.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/drivers/staging/ozwpan/ozproto.c b/drivers/staging/ozwpan/ozproto.c
index 8c2200b90d5965ce91c906300c45364809b18bde..d8ac823f1c977e33ed59e009e5257bdf6bd54b80 100644 (file)
--- a/drivers/staging/ozwpan/ozproto.c
+++ b/drivers/staging/ozwpan/ozproto.c
@@ -497,7 +497,7 @@ void oz_pd_heartbeat_handler(unsigned long data)
        spin_unlock_bh(&g_polling_lock);
        if (apps)
                oz_pd_heartbeat(pd, apps);
-
+       oz_pd_put(pd);
 }
 
 /*------------------------------------------------------------------------------
@@ -519,6 +519,7 @@ void oz_pd_timeout_handler(unsigned long data)
                oz_pd_stop(pd);
                break;
        }
+       oz_pd_put(pd);
 }
 
 /*------------------------------------------------------------------------------
@@ -531,6 +532,7 @@ enum hrtimer_restart oz_pd_heartbeat_event(struct hrtimer *timer)
        pd = container_of(timer, struct oz_pd, heartbeat);
        hrtimer_forward_now(timer, ktime_set(pd->pulse_period /
        MSEC_PER_SEC, (pd->pulse_period % MSEC_PER_SEC) * NSEC_PER_MSEC));
+       oz_pd_get(pd);
        tasklet_schedule(&pd->heartbeat_tasklet);
        return HRTIMER_RESTART;
 }
@@ -543,6 +545,7 @@ enum hrtimer_restart oz_pd_timeout_event(struct hrtimer *timer)
        struct oz_pd *pd;
 
        pd = container_of(timer, struct oz_pd, timeout);
+       oz_pd_get(pd);
        tasklet_schedule(&pd->timeout_tasklet);
        return HRTIMER_NORESTART;
 }