From 32da7c25c64349bbc7fc330acb8922764e16e8c9 Mon Sep 17 00:00:00 2001 From: Minje Ahn Date: Fri, 14 Jul 2017 13:49:33 +0900 Subject: [SATIZENVUL-926] Check allocation Change-Id: I3ca13cf412738708ff05864312b375ff58106fc4 Signed-off-by: Minje Ahn --- server/thumb-server-internal.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/server/thumb-server-internal.c b/server/thumb-server-internal.c index 5974be3..1b8bac3 100755 --- a/server/thumb-server-internal.c +++ b/server/thumb-server-internal.c @@ -502,7 +502,12 @@ _media_thumb_process_raw(thumbMsg *req_msg, thumbMsg *res_msg) res_msg->thumb_height = thumb_h; res_msg->thumb_size = thumb_size; res_msg->thumb_data = malloc(thumb_size * sizeof(unsigned char)); - memcpy(res_msg->thumb_data, data, thumb_size); + if (res_msg->thumb_data != NULL) { + memcpy(res_msg->thumb_data, data, thumb_size); + } else { + thumb_err("Allocation failed"); + err = MS_MEDIA_ERR_OUT_OF_MEMORY; + } SAFE_FREE(data); -- cgit v1.2.3