1 /******************************************************************************
2 *
3 * Copyright 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 <base/logging.h>
19 #include <stdlib.h>
20 #include <string.h>
21
22 #include "osi/include/allocation_tracker.h"
23 #include "osi/include/allocator.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 CHECK(ptr);
32
33 char* new_string = static_cast<char*>(
34 allocation_tracker_notify_alloc(alloc_allocator_id, ptr, size));
35 if (!new_string) return NULL;
36
37 memcpy(new_string, str, size);
38 return new_string;
39 }
40
osi_strndup(const char * str,size_t len)41 char* osi_strndup(const char* str, size_t len) {
42 size_t size = strlen(str);
43 if (len < size) size = len;
44
45 size_t real_size = allocation_tracker_resize_for_canary(size + 1);
46 void* ptr = malloc(real_size);
47 CHECK(ptr);
48
49 char* new_string = static_cast<char*>(
50 allocation_tracker_notify_alloc(alloc_allocator_id, ptr, size + 1));
51 if (!new_string) return NULL;
52
53 memcpy(new_string, str, size);
54 new_string[size] = '\0';
55 return new_string;
56 }
57
osi_malloc(size_t size)58 void* osi_malloc(size_t size) {
59 CHECK(static_cast<ssize_t>(size) >= 0);
60 size_t real_size = allocation_tracker_resize_for_canary(size);
61 void* ptr = malloc(real_size);
62 CHECK(ptr);
63 return allocation_tracker_notify_alloc(alloc_allocator_id, ptr, size);
64 }
65
osi_calloc(size_t size)66 void* osi_calloc(size_t size) {
67 CHECK(static_cast<ssize_t>(size) >= 0);
68 size_t real_size = allocation_tracker_resize_for_canary(size);
69 void* ptr = calloc(1, real_size);
70 CHECK(ptr);
71 return allocation_tracker_notify_alloc(alloc_allocator_id, ptr, size);
72 }
73
osi_free(void * ptr)74 void osi_free(void* ptr) {
75 free(allocation_tracker_notify_free(alloc_allocator_id, ptr));
76 }
77
osi_free_and_reset(void ** p_ptr)78 void osi_free_and_reset(void** p_ptr) {
79 CHECK(p_ptr != NULL);
80 osi_free(*p_ptr);
81 *p_ptr = NULL;
82 }
83
84 const allocator_t allocator_calloc = {osi_calloc, osi_free};
85
86 const allocator_t allocator_malloc = {osi_malloc, osi_free};
87