summaryrefslogtreecommitdiffstats
path: root/compiler-rt/lib/asan/interception/interception_win.h
blob: 9d1586ecb1739b3f68775680b2c37fe85b6a1b4d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
//===-- interception_linux.h ------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file is a part of AddressSanitizer, an address sanity checker.
//
// Windows-specific interception methods.
//===----------------------------------------------------------------------===//

#ifdef _WIN32

#if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
# error "interception_win.h should be included from interception library only"
#endif

#ifndef INTERCEPTION_WIN_H
#define INTERCEPTION_WIN_H

namespace __interception {
// returns true if a function with the given name was found.
bool GetRealFunctionAddress(const char *func_name, void **func_addr);

// returns true if the old function existed, false on failure.
bool OverrideFunction(void *old_func, void *new_func, void **orig_old_func);
}  // namespace __interception

#if defined(_DLL)
# define INTERCEPT_FUNCTION_WIN(func) \
    ::__interception::GetRealFunctionAddress(#func, (void**)&REAL(func))
#else
# define INTERCEPT_FUNCTION_WIN(func) \
    ::__interception::OverrideFunction((void*)func, (void*)WRAP(func), \
                                       (void**)&REAL(func))
#endif

#endif  // INTERCEPTION_WIN_H
#endif  // _WIN32
OpenPOWER on IntegriCloud