SVACE issue fix 36/250036/4
authorAbhimanyu Swami <abhimanyu1.s@samsung.com>
Fri, 18 Dec 2020 10:01:29 +0000 (15:31 +0530)
committerAbhimanyu Swami <abhimanyu1.s@samsung.com>
Tue, 22 Dec 2020 07:15:43 +0000 (12:45 +0530)
Change-Id: I16d7682a0072f9a0307538042c9509bb7547cc82
Signed-off-by: Abhimanyu Swami <abhimanyu1.s@samsung.com>
server/db/ctsvc_db_query.c

index 37d5614..69c38a6 100644 (file)
@@ -3020,6 +3020,13 @@ static int __ctsvc_db_search_records_exec(const char *view_uri,
 
                if (r_type == CTSVC_RECORD_PERSON) {
                        unsigned int *project = malloc(sizeof(unsigned int)*ids_count);
+                       if (project == NULL){
+                               ERR("Memory Allocation Failed");
+                               ctsvc_stmt_finalize(stmt);
+                               contacts_list_destroy(list, true);
+                               return CONTACTS_ERROR_OUT_OF_MEMORY;
+                       }
+
                        for (i = 0; i < ids_count; i++)
                                project[i] = properties[i].property_id;
 
@@ -4203,6 +4210,12 @@ static inline int __ctsvc_db_search_records_with_query_exec(ctsvc_query_s *s_que
                                ids_count = s_query->projection_count;
 
                        project = malloc(sizeof(unsigned int)*ids_count);
+                       if (project == NULL){
+                               ERR("Memory Allocation failed");
+                               ctsvc_stmt_finalize(stmt);
+                               contacts_list_destroy(list, true);
+                               return CONTACTS_ERROR_OUT_OF_MEMORY;
+                       }
 
                        for (i = 0; i < ids_count; i++) {
                                if (0 == s_query->projection_count)