1 /****************************************************************************** 2 * 3 * Copyright (C) 2014 Google, Inc. 4 * 5 * Licensed under the Apache License, Version 2.0 (the "License"); 6 * you may not use this file except in compliance with the License. 7 * You may obtain a copy of the License at: 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 * 17 ******************************************************************************/ 18 #include <assert.h> 19 #include <stdlib.h> 20 #include <string.h> 21 22 #include "osi/include/allocator.h" 23 #include "osi/include/allocation_tracker.h" 24 25 static const allocator_id_t alloc_allocator_id = 42; 26 osi_strdup(const char * str)27 char *osi_strdup(const char *str) { 28 size_t size = strlen(str) + 1; // + 1 for the null terminator 29 size_t real_size = allocation_tracker_resize_for_canary(size); 30 void *ptr = malloc(real_size); 31 assert(ptr); 32 33 char *new_string = allocation_tracker_notify_alloc( 34 alloc_allocator_id, 35 ptr, 36 size); 37 if (!new_string) 38 return NULL; 39 40 memcpy(new_string, str, size); 41 return new_string; 42 } 43 osi_strndup(const char * str,size_t len)44 char *osi_strndup(const char *str, size_t len) { 45 size_t size = strlen(str); 46 if (len < size) 47 size = len; 48 49 size_t real_size = allocation_tracker_resize_for_canary(size + 1); 50 void *ptr = malloc(real_size); 51 assert(ptr); 52 53 char *new_string = allocation_tracker_notify_alloc( 54 alloc_allocator_id, 55 ptr, 56 size + 1); 57 if (!new_string) 58 return NULL; 59 60 memcpy(new_string, str, size); 61 new_string[size] = '\0'; 62 return new_string; 63 } 64 osi_malloc(size_t size)65 void *osi_malloc(size_t size) { 66 size_t real_size = allocation_tracker_resize_for_canary(size); 67 void *ptr = malloc(real_size); 68 assert(ptr); 69 return allocation_tracker_notify_alloc(alloc_allocator_id, ptr, size); 70 } 71 osi_calloc(size_t size)72 void *osi_calloc(size_t size) { 73 size_t real_size = allocation_tracker_resize_for_canary(size); 74 void *ptr = calloc(1, real_size); 75 assert(ptr); 76 return allocation_tracker_notify_alloc(alloc_allocator_id, ptr, size); 77 } 78 osi_free(void * ptr)79 void osi_free(void *ptr) { 80 free(allocation_tracker_notify_free(alloc_allocator_id, ptr)); 81 } 82 osi_free_and_reset(void ** p_ptr)83 void osi_free_and_reset(void **p_ptr) 84 { 85 assert(p_ptr != NULL); 86 osi_free(*p_ptr); 87 *p_ptr = NULL; 88 } 89 90 const allocator_t allocator_calloc = { 91 osi_calloc, 92 osi_free 93 }; 94 95 const allocator_t allocator_malloc = { 96 osi_malloc, 97 osi_free 98 }; 99