Add null checking code after add_node

Change-Id: Ia8a18630551d988e01a90e3158ba4d42b183b4f9
Signed-off-by: lokilee73 <changjoo.lee@samsung.com>

diff --git a/plugins/iot/display/core.c b/plugins/iot/display/core.c
index b748d4a..e01f281 100644 (file)
--- a/plugins/iot/display/core.c
+++ b/plugins/iot/display/core.c
@@ -1014,9 +1014,13 @@ static void proc_condition_lock(PMMsg *data)
        holdkey_block = GET_COND_FLAG(data->cond) & PM_FLAG_BLOCK_HOLDKEY;
 
        tmp = find_node(state, pid);
-       if (!tmp)
-               add_node(state, pid, cond_timeout_id, holdkey_block);
-       else {
+       if (!tmp) {
+               tmp = add_node(state, pid, cond_timeout_id, holdkey_block);
+               if (!tmp) {
+                       _E("Failed to acquire lock, state: %d, pid: %d.", state, pid);
+                       return;
+               }
+       } else {
                update_lock_timer(data, tmp, cond_timeout_id);
                tmp->holdkey_block = holdkey_block;
        }

diff --git a/plugins/mobile/display/core.c b/plugins/mobile/display/core.c
index 98caba4..1c708b0 100644 (file)
--- a/plugins/mobile/display/core.c
+++ b/plugins/mobile/display/core.c
@@ -1026,9 +1026,13 @@ static void proc_condition_lock(PMMsg *data)
        holdkey_block = GET_COND_FLAG(data->cond) & PM_FLAG_BLOCK_HOLDKEY;
 
        tmp = find_node(state, pid);
-       if (!tmp)
-               add_node(state, pid, cond_timeout_id, holdkey_block);
-       else {
+       if (!tmp) {
+               tmp = add_node(state, pid, cond_timeout_id, holdkey_block);
+               if (!tmp) {
+                       _E("Failed to acquire lock, state: %d, pid: %d.", state, pid);
+                       return;
+               }
+       } else {
                update_lock_timer(data, tmp, cond_timeout_id);
                tmp->holdkey_block = holdkey_block;
        }

diff --git a/plugins/tv/display/core.c b/plugins/tv/display/core.c
index 9a3b157..aaa62db 100644 (file)
--- a/plugins/tv/display/core.c
+++ b/plugins/tv/display/core.c
@@ -1015,9 +1015,13 @@ static void proc_condition_lock(PMMsg *data)
        holdkey_block = GET_COND_FLAG(data->cond) & PM_FLAG_BLOCK_HOLDKEY;
 
        tmp = find_node(state, pid);
-       if (!tmp)
-               add_node(state, pid, cond_timeout_id, holdkey_block);
-       else {
+       if (!tmp) {
+               tmp = add_node(state, pid, cond_timeout_id, holdkey_block);
+               if (!tmp) {
+                       _E("Failed to acquire lock, state: %d, pid: %d.", state, pid);
+                       return;
+               }
+       } else {
                update_lock_timer(data, tmp, cond_timeout_id);
                tmp->holdkey_block = holdkey_block;
        }