1//===-- memprof_interceptors.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.cpp
12//===----------------------------------------------------------------------===//
13#ifndef MEMPROF_INTERCEPTORS_H
14#define MEMPROF_INTERCEPTORS_H
15
16#include "interception/interception.h"
17#include "memprof_interceptors_memintrinsics.h"
18#include "memprof_internal.h"
19#include "sanitizer_common/sanitizer_platform_interceptors.h"
20
21namespace __memprof {
22
23void InitializeMemprofInterceptors();
24void InitializePlatformInterceptors();
25
26#define ENSURE_MEMPROF_INITED() \
27 do { \
28 CHECK(!memprof_init_is_running); \
29 if (UNLIKELY(!memprof_inited)) { \
30 MemprofInitFromRtl(); \
31 } \
32 } while (0)
33
34} // namespace __memprof
35
36DECLARE_REAL(int, memcmp, const void *a1, const void *a2, SIZE_T size)
37DECLARE_REAL(char *, strchr, const char *str, int c)
38DECLARE_REAL(SIZE_T, strlen, const char *s)
39DECLARE_REAL(char *, strncpy, char *to, const char *from, SIZE_T size)
40DECLARE_REAL(SIZE_T, strnlen, const char *s, SIZE_T maxlen)
41DECLARE_REAL(char *, strstr, const char *s1, const char *s2)
42
43#define MEMPROF_INTERCEPT_FUNC(name) \
44 do { \
45 if (!INTERCEPT_FUNCTION(name)) \
46 VReport(1, "MemProfiler: failed to intercept '%s'\n'", #name); \
47 } while (0)
48#define MEMPROF_INTERCEPT_FUNC_VER(name, ver) \
49 do { \
50 if (!INTERCEPT_FUNCTION_VER(name, ver)) \
51 VReport(1, "MemProfiler: failed to intercept '%s@@%s'\n", #name, ver); \
52 } while (0)
53#define MEMPROF_INTERCEPT_FUNC_VER_UNVERSIONED_FALLBACK(name, ver) \
54 do { \
55 if (!INTERCEPT_FUNCTION_VER(name, ver) && !INTERCEPT_FUNCTION(name)) \
56 VReport(1, "MemProfiler: failed to intercept '%s@@%s' or '%s'\n", #name, \
57 ver, #name); \
58 } while (0)
59
60#define MEMPROF_INTERCEPTOR_ENTER(ctx, func) \
61 ctx = 0; \
62 (void)ctx;
63
64#define COMMON_INTERCEPT_FUNCTION(name) MEMPROF_INTERCEPT_FUNC(name)
65
66#endif // MEMPROF_INTERCEPTORS_H
67