Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- ASanStackFrameLayout.h - ComputeASanStackFrameLayout -----*- 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 header defines ComputeASanStackFrameLayout and auxiliary data structs. |
||
10 | // |
||
11 | //===----------------------------------------------------------------------===// |
||
12 | #ifndef LLVM_TRANSFORMS_UTILS_ASANSTACKFRAMELAYOUT_H |
||
13 | #define LLVM_TRANSFORMS_UTILS_ASANSTACKFRAMELAYOUT_H |
||
14 | #include "llvm/ADT/SmallString.h" |
||
15 | #include "llvm/ADT/SmallVector.h" |
||
16 | |||
17 | namespace llvm { |
||
18 | |||
19 | class AllocaInst; |
||
20 | |||
21 | // These magic constants should be the same as in |
||
22 | // in asan_internal.h from ASan runtime in compiler-rt. |
||
23 | static const int kAsanStackLeftRedzoneMagic = 0xf1; |
||
24 | static const int kAsanStackMidRedzoneMagic = 0xf2; |
||
25 | static const int kAsanStackRightRedzoneMagic = 0xf3; |
||
26 | static const int kAsanStackUseAfterReturnMagic = 0xf5; |
||
27 | static const int kAsanStackUseAfterScopeMagic = 0xf8; |
||
28 | |||
29 | // Input/output data struct for ComputeASanStackFrameLayout. |
||
30 | struct ASanStackVariableDescription { |
||
31 | const char *Name; // Name of the variable that will be displayed by asan |
||
32 | // if a stack-related bug is reported. |
||
33 | uint64_t Size; // Size of the variable in bytes. |
||
34 | size_t LifetimeSize; // Size in bytes to use for lifetime analysis check. |
||
35 | // Will be rounded up to Granularity. |
||
36 | uint64_t Alignment; // Alignment of the variable (power of 2). |
||
37 | AllocaInst *AI; // The actual AllocaInst. |
||
38 | size_t Offset; // Offset from the beginning of the frame; |
||
39 | // set by ComputeASanStackFrameLayout. |
||
40 | unsigned Line; // Line number. |
||
41 | }; |
||
42 | |||
43 | // Output data struct for ComputeASanStackFrameLayout. |
||
44 | struct ASanStackFrameLayout { |
||
45 | uint64_t Granularity; // Shadow granularity. |
||
46 | uint64_t FrameAlignment; // Alignment for the entire frame. |
||
47 | uint64_t FrameSize; // Size of the frame in bytes. |
||
48 | }; |
||
49 | |||
50 | ASanStackFrameLayout ComputeASanStackFrameLayout( |
||
51 | // The array of stack variables. The elements may get reordered and changed. |
||
52 | SmallVectorImpl<ASanStackVariableDescription> &Vars, |
||
53 | // AddressSanitizer's shadow granularity. Usually 8, may also be 16, 32, 64. |
||
54 | uint64_t Granularity, |
||
55 | // The minimal size of the left-most redzone (header). |
||
56 | // At least 4 pointer sizes, power of 2, and >= Granularity. |
||
57 | // The resulting FrameSize should be multiple of MinHeaderSize. |
||
58 | uint64_t MinHeaderSize); |
||
59 | |||
60 | // Compute frame description, see DescribeAddressIfStack in ASan runtime. |
||
61 | SmallString<64> ComputeASanStackFrameDescription( |
||
62 | const SmallVectorImpl<ASanStackVariableDescription> &Vars); |
||
63 | |||
64 | // Returns shadow bytes with marked red zones. This shadow represents the state |
||
65 | // if the stack frame when all local variables are inside of the own scope. |
||
66 | SmallVector<uint8_t, 64> |
||
67 | GetShadowBytes(const SmallVectorImpl<ASanStackVariableDescription> &Vars, |
||
68 | const ASanStackFrameLayout &Layout); |
||
69 | |||
70 | // Returns shadow bytes with marked red zones and after scope. This shadow |
||
71 | // represents the state if the stack frame when all local variables are outside |
||
72 | // of the own scope. |
||
73 | SmallVector<uint8_t, 64> GetShadowBytesAfterScope( |
||
74 | // The array of stack variables. The elements may get reordered and changed. |
||
75 | const SmallVectorImpl<ASanStackVariableDescription> &Vars, |
||
76 | const ASanStackFrameLayout &Layout); |
||
77 | |||
78 | } // llvm namespace |
||
79 | |||
80 | #endif // LLVM_TRANSFORMS_UTILS_ASANSTACKFRAMELAYOUT_H |