Add null check when allocating memory

Change-Id: I949c7a314c2ac71efb6c34f64bcbe68b9a350715
Signed-off-by: Jiwoong Im <jiwoong.im@samsung.com>

diff --git a/src/message-port.c b/src/message-port.c
index fa96345..e25e5a2 100755 (executable)
--- a/src/message-port.c
+++ b/src/message-port.c
@@ -1172,6 +1172,7 @@ static bool __check_sender_validation(GVariant *parameters, const char *sender,
        char *local_appid = NULL;
        int pid = __get_sender_pid(conn, sender);
        int *watcher_id = (int *)calloc(1, sizeof(int));
+       retvm_if(!watcher_id, false, "Malloc failed");
 
        ret = aul_app_get_appid_bypid(pid, buffer, sizeof(buffer));
        if (ret != AUL_R_OK) {
@@ -2021,6 +2022,8 @@ int __messageport_watch_remote_port(int *watcher_id, const char *remote_app_id,
                __registered_callback_info_hash = g_hash_table_new_full(g_direct_hash,  g_direct_equal, NULL, __registered_callback_info_free);
 
        registered_callback_info_s *registered_cb_info = (registered_callback_info_s *)calloc(1, sizeof(registered_callback_info_s));
+       retvm_if(!registered_cb_info, MESSAGEPORT_ERROR_OUT_OF_MEMORY, "Malloc failed");
+
        registered_cb_info->registered_cb = registered_cb;
        registered_cb_info->unregistered_cb = unregistered_cb;
        registered_cb_info->user_data = user_data;