1 //===-- memprof_interceptors_memintrinsics.h -------------------*- C++ -*-===//
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 // This file is a part of MemProfiler, a memory profiler.
10 //
11 // MemProf-private header for memprof_interceptors_memintrinsics.cpp
12 //===---------------------------------------------------------------------===//
13 #ifndef MEMPROF_MEMINTRIN_H
14 #define MEMPROF_MEMINTRIN_H
15
16 #include "interception/interception.h"
17 #include "memprof_interface_internal.h"
18 #include "memprof_internal.h"
19 #include "memprof_mapping.h"
20
DECLARE_REAL(void *,memcpy,void * to,const void * from,uptr size)21 DECLARE_REAL(void *, memcpy, void *to, const void *from, uptr size)
22 DECLARE_REAL(void *, memset, void *block, int c, uptr size)
23
24 namespace __memprof {
25
26 // We implement ACCESS_MEMORY_RANGE, MEMPROF_READ_RANGE,
27 // and MEMPROF_WRITE_RANGE as macro instead of function so
28 // that no extra frames are created, and stack trace contains
29 // relevant information only.
30 #define ACCESS_MEMORY_RANGE(offset, size) \
31 do { \
32 __memprof_record_access_range(offset, size); \
33 } while (0)
34
35 // memcpy is called during __memprof_init() from the internals of printf(...).
36 // We do not treat memcpy with to==from as a bug.
37 // See http://llvm.org/bugs/show_bug.cgi?id=11763.
38 #define MEMPROF_MEMCPY_IMPL(to, from, size) \
39 do { \
40 if (UNLIKELY(!memprof_inited)) \
41 return internal_memcpy(to, from, size); \
42 if (memprof_init_is_running) { \
43 return REAL(memcpy)(to, from, size); \
44 } \
45 ENSURE_MEMPROF_INITED(); \
46 MEMPROF_READ_RANGE(from, size); \
47 MEMPROF_WRITE_RANGE(to, size); \
48 return REAL(memcpy)(to, from, size); \
49 } while (0)
50
51 // memset is called inside Printf.
52 #define MEMPROF_MEMSET_IMPL(block, c, size) \
53 do { \
54 if (UNLIKELY(!memprof_inited)) \
55 return internal_memset(block, c, size); \
56 if (memprof_init_is_running) { \
57 return REAL(memset)(block, c, size); \
58 } \
59 ENSURE_MEMPROF_INITED(); \
60 MEMPROF_WRITE_RANGE(block, size); \
61 return REAL(memset)(block, c, size); \
62 } while (0)
63
64 #define MEMPROF_MEMMOVE_IMPL(to, from, size) \
65 do { \
66 if (UNLIKELY(!memprof_inited)) \
67 return internal_memmove(to, from, size); \
68 ENSURE_MEMPROF_INITED(); \
69 MEMPROF_READ_RANGE(from, size); \
70 MEMPROF_WRITE_RANGE(to, size); \
71 return internal_memmove(to, from, size); \
72 } while (0)
73
74 #define MEMPROF_READ_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size)
75 #define MEMPROF_WRITE_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size)
76
77 } // namespace __memprof
78
79 #endif // MEMPROF_MEMINTRIN_H
80