1 //===-- Simple malloc and free for use with integration tests -------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "src/__support/common.h"
10 #include "src/__support/macros/config.h"
11 #include <stddef.h>
12 #include <stdint.h>
13
14 #ifdef LIBC_TARGET_ARCH_IS_AARCH64
15 #include "src/sys/auxv/getauxval.h"
16 #endif
17
18 // Integration tests rely on the following memory functions. This is because the
19 // compiler code generation can emit calls to them. We want to map the external
20 // entrypoint to the internal implementation of the function used for testing.
21 // This is done manually as not all targets support aliases.
22
23 namespace LIBC_NAMESPACE_DECL {
24
25 int bcmp(const void *lhs, const void *rhs, size_t count);
26 void bzero(void *ptr, size_t count);
27 int memcmp(const void *lhs, const void *rhs, size_t count);
28 void *memcpy(void *__restrict, const void *__restrict, size_t);
29 void *memmove(void *dst, const void *src, size_t count);
30 void *memset(void *ptr, int value, size_t count);
31 int atexit(void (*func)(void));
32
33 } // namespace LIBC_NAMESPACE_DECL
34
35 extern "C" {
36
bcmp(const void * lhs,const void * rhs,size_t count)37 int bcmp(const void *lhs, const void *rhs, size_t count) {
38 return LIBC_NAMESPACE::bcmp(lhs, rhs, count);
39 }
bzero(void * ptr,size_t count)40 void bzero(void *ptr, size_t count) { LIBC_NAMESPACE::bzero(ptr, count); }
memcmp(const void * lhs,const void * rhs,size_t count)41 int memcmp(const void *lhs, const void *rhs, size_t count) {
42 return LIBC_NAMESPACE::memcmp(lhs, rhs, count);
43 }
memcpy(void * __restrict dst,const void * __restrict src,size_t count)44 void *memcpy(void *__restrict dst, const void *__restrict src, size_t count) {
45 return LIBC_NAMESPACE::memcpy(dst, src, count);
46 }
memmove(void * dst,const void * src,size_t count)47 void *memmove(void *dst, const void *src, size_t count) {
48 return LIBC_NAMESPACE::memmove(dst, src, count);
49 }
memset(void * ptr,int value,size_t count)50 void *memset(void *ptr, int value, size_t count) {
51 return LIBC_NAMESPACE::memset(ptr, value, count);
52 }
53
54 // This is needed if the test was compiled with '-fno-use-cxa-atexit'.
atexit(void (* func)(void))55 int atexit(void (*func)(void)) { return LIBC_NAMESPACE::atexit(func); }
56
57 } // extern "C"
58
59 // Integration tests cannot use the SCUDO standalone allocator as SCUDO pulls
60 // various other parts of the libc. Since SCUDO development does not use
61 // LLVM libc build rules, it is very hard to keep track or pull all that SCUDO
62 // requires. Hence, as a work around for this problem, we use a simple allocator
63 // which just hands out continuous blocks from a statically allocated chunk of
64 // memory.
65
66 static constexpr uint64_t MEMORY_SIZE = 16384;
67 static uint8_t memory[MEMORY_SIZE];
68 static uint8_t *ptr = memory;
69
70 extern "C" {
71
malloc(size_t s)72 void *malloc(size_t s) {
73 void *mem = ptr;
74 ptr += s;
75 return static_cast<uint64_t>(ptr - memory) >= MEMORY_SIZE ? nullptr : mem;
76 }
77
free(void *)78 void free(void *) {}
79
realloc(void * ptr,size_t s)80 void *realloc(void *ptr, size_t s) {
81 free(ptr);
82 return malloc(s);
83 }
84
85 // Integration tests are linked with -nostdlib. BFD linker expects
86 // __dso_handle when -nostdlib is used.
87 void *__dso_handle = nullptr;
88
89 #ifdef LIBC_TARGET_ARCH_IS_AARCH64
90 // Due to historical reasons, libgcc on aarch64 may expect __getauxval to be
91 // defined. See also https://gcc.gnu.org/pipermail/gcc-cvs/2020-June/300635.html
__getauxval(unsigned long id)92 unsigned long __getauxval(unsigned long id) {
93 return LIBC_NAMESPACE::getauxval(id);
94 }
95 #endif
96 } // extern "C"
97