80 lines
4.1 KiB
C++
80 lines
4.1 KiB
C++
//===-- memprof_interceptors_memintrinsics.h -------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===---------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of MemProfiler, a memory profiler.
|
|
//
|
|
// MemProf-private header for memprof_interceptors_memintrinsics.cpp
|
|
//===---------------------------------------------------------------------===//
|
|
#ifndef MEMPROF_MEMINTRIN_H
|
|
#define MEMPROF_MEMINTRIN_H
|
|
|
|
#include "interception/interception.h"
|
|
#include "memprof_interface_internal.h"
|
|
#include "memprof_internal.h"
|
|
#include "memprof_mapping.h"
|
|
|
|
DECLARE_REAL(void *, memcpy, void *to, const void *from, uptr size)
|
|
DECLARE_REAL(void *, memset, void *block, int c, uptr size)
|
|
|
|
namespace __memprof {
|
|
|
|
// We implement ACCESS_MEMORY_RANGE, MEMPROF_READ_RANGE,
|
|
// and MEMPROF_WRITE_RANGE as macro instead of function so
|
|
// that no extra frames are created, and stack trace contains
|
|
// relevant information only.
|
|
#define ACCESS_MEMORY_RANGE(offset, size) \
|
|
do { \
|
|
__memprof_record_access_range(offset, size); \
|
|
} while (0)
|
|
|
|
// memcpy is called during __memprof_init() from the internals of printf(...).
|
|
// We do not treat memcpy with to==from as a bug.
|
|
// See http://llvm.org/bugs/show_bug.cgi?id=11763.
|
|
#define MEMPROF_MEMCPY_IMPL(to, from, size) \
|
|
do { \
|
|
if (UNLIKELY(!memprof_inited)) \
|
|
return internal_memcpy(to, from, size); \
|
|
if (memprof_init_is_running) { \
|
|
return REAL(memcpy)(to, from, size); \
|
|
} \
|
|
ENSURE_MEMPROF_INITED(); \
|
|
MEMPROF_READ_RANGE(from, size); \
|
|
MEMPROF_WRITE_RANGE(to, size); \
|
|
return REAL(memcpy)(to, from, size); \
|
|
} while (0)
|
|
|
|
// memset is called inside Printf.
|
|
#define MEMPROF_MEMSET_IMPL(block, c, size) \
|
|
do { \
|
|
if (UNLIKELY(!memprof_inited)) \
|
|
return internal_memset(block, c, size); \
|
|
if (memprof_init_is_running) { \
|
|
return REAL(memset)(block, c, size); \
|
|
} \
|
|
ENSURE_MEMPROF_INITED(); \
|
|
MEMPROF_WRITE_RANGE(block, size); \
|
|
return REAL(memset)(block, c, size); \
|
|
} while (0)
|
|
|
|
#define MEMPROF_MEMMOVE_IMPL(to, from, size) \
|
|
do { \
|
|
if (UNLIKELY(!memprof_inited)) \
|
|
return internal_memmove(to, from, size); \
|
|
ENSURE_MEMPROF_INITED(); \
|
|
MEMPROF_READ_RANGE(from, size); \
|
|
MEMPROF_WRITE_RANGE(to, size); \
|
|
return internal_memmove(to, from, size); \
|
|
} while (0)
|
|
|
|
#define MEMPROF_READ_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size)
|
|
#define MEMPROF_WRITE_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size)
|
|
|
|
} // namespace __memprof
|
|
|
|
#endif // MEMPROF_MEMINTRIN_H
|