1 /*
2 * Copyright (C) 2019 The Android Open Source Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * * Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in
12 * the documentation and/or other materials provided with the
13 * distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
18 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
19 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
21 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
22 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
23 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
24 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
25 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #pragma once
30
31 #include <stdatomic.h>
32 #include <stdio.h>
33
34 #include <async_safe/log.h>
35 #include <private/bionic_globals.h>
36 #include <private/bionic_malloc_dispatch.h>
37
38 #if __has_feature(hwaddress_sanitizer)
39
40 #include <sanitizer/hwasan_interface.h>
41
42 __BEGIN_DECLS
43
44 // FIXME: implement these in HWASan allocator.
45 int __sanitizer_malloc_iterate(uintptr_t base, size_t size,
46 void (*callback)(uintptr_t base, size_t size, void* arg),
47 void* arg);
48 void __sanitizer_malloc_disable();
49 void __sanitizer_malloc_enable();
50 int __sanitizer_malloc_info(int options, FILE* fp);
51
52 __END_DECLS
53
54 #define Malloc(function) __sanitizer_ ## function
55
56 #else // __has_feature(hwaddress_sanitizer)
57
58 #if defined(USE_SCUDO)
59
60 #include "scudo.h"
61 #define Malloc(function) scudo_ ## function
62
63 #elif defined(USE_SCUDO_SVELTE)
64
65 #include "scudo.h"
66 #define Malloc(function) scudo_svelte_ ## function
67
68 #else
69
70 #include "jemalloc.h"
71 #define Malloc(function) je_ ## function
72
73 #endif
74
75 #endif
76
77 const MallocDispatch* NativeAllocatorDispatch();
78
GetDispatchTable()79 static inline const MallocDispatch* GetDispatchTable() {
80 return atomic_load_explicit(&__libc_globals->current_dispatch_table, memory_order_acquire);
81 }
82
GetDefaultDispatchTable()83 static inline const MallocDispatch* GetDefaultDispatchTable() {
84 return atomic_load_explicit(&__libc_globals->default_dispatch_table, memory_order_acquire);
85 }
86
87 // =============================================================================
88 // Log functions
89 // =============================================================================
90 #define error_log(format, ...) \
91 async_safe_format_log(ANDROID_LOG_ERROR, "libc", (format), ##__VA_ARGS__ )
92 #define info_log(format, ...) \
93 async_safe_format_log(ANDROID_LOG_INFO, "libc", (format), ##__VA_ARGS__ )
94 #define warning_log(format, ...) \
95 async_safe_format_log(ANDROID_LOG_WARN, "libc", (format), ##__VA_ARGS__ )
96 // =============================================================================
97