1 //===- ASanStackFrameLayoutTest.cpp - Tests for ComputeASanStackFrameLayout===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 #include "llvm/Transforms/Utils/ASanStackFrameLayout.h" 10 #include "llvm/ADT/ArrayRef.h" 11 #include "gtest/gtest.h" 12 #include <sstream> 13 14 using namespace llvm; 15 16 static std::string 17 ShadowBytesToString(ArrayRef<uint8_t> ShadowBytes) { 18 std::ostringstream os; 19 for (size_t i = 0, n = ShadowBytes.size(); i < n; i++) { 20 switch (ShadowBytes[i]) { 21 case kAsanStackLeftRedzoneMagic: os << "L"; break; 22 case kAsanStackRightRedzoneMagic: os << "R"; break; 23 case kAsanStackMidRedzoneMagic: os << "M"; break; 24 case kAsanStackUseAfterScopeMagic: 25 os << "S"; 26 break; 27 default: os << (unsigned)ShadowBytes[i]; 28 } 29 } 30 return os.str(); 31 } 32 33 // Use macro to preserve line information in EXPECT_EQ output. 34 #define TEST_LAYOUT(V, Granularity, MinHeaderSize, ExpectedDescr, \ 35 ExpectedShadow, ExpectedShadowAfterScope) \ 36 { \ 37 SmallVector<ASanStackVariableDescription, 10> Vars = V; \ 38 ASanStackFrameLayout L = \ 39 ComputeASanStackFrameLayout(Vars, Granularity, MinHeaderSize); \ 40 EXPECT_STREQ(ExpectedDescr, \ 41 ComputeASanStackFrameDescription(Vars).c_str()); \ 42 EXPECT_EQ(ExpectedShadow, ShadowBytesToString(GetShadowBytes(Vars, L))); \ 43 EXPECT_EQ(ExpectedShadowAfterScope, \ 44 ShadowBytesToString(GetShadowBytesAfterScope(Vars, L))); \ 45 } 46 47 TEST(ASanStackFrameLayout, Test) { 48 #define VAR(name, size, lifetime, alignment, line) \ 49 ASanStackVariableDescription name##size##_##alignment = { \ 50 #name #size "_" #alignment, \ 51 size, \ 52 lifetime, \ 53 alignment, \ 54 0, \ 55 0, \ 56 line, \ 57 } 58 59 VAR(a, 1, 0, 1, 0); 60 VAR(p, 1, 0, 32, 15); 61 VAR(p, 1, 0, 256, 2700); 62 VAR(a, 2, 0, 1, 0); 63 VAR(a, 3, 0, 1, 0); 64 VAR(a, 4, 0, 1, 0); 65 VAR(a, 7, 0, 1, 0); 66 VAR(a, 8, 8, 1, 0); 67 VAR(a, 9, 0, 1, 0); 68 VAR(a, 16, 16, 1, 0); 69 VAR(a, 41, 9, 1, 7); 70 VAR(a, 105, 103, 1, 0); 71 VAR(a, 200, 97, 1, 0); 72 73 TEST_LAYOUT({a1_1}, 8, 16, "1 16 1 4 a1_1", "LL1R", "LL1R"); 74 TEST_LAYOUT({a1_1}, 16, 16, "1 16 1 4 a1_1", "L1R", "L1R"); 75 TEST_LAYOUT({a1_1}, 32, 32, "1 32 1 4 a1_1", "L1R", "L1R"); 76 TEST_LAYOUT({a1_1}, 64, 64, "1 64 1 4 a1_1", "L1R", "L1R"); 77 TEST_LAYOUT({p1_32}, 8, 32, "1 32 1 8 p1_32:15", "LLLL1RRR", "LLLL1RRR"); 78 TEST_LAYOUT({p1_32}, 8, 64, "1 64 1 8 p1_32:15", "LLLLLLLL1RRRRRRR", 79 "LLLLLLLL1RRRRRRR"); 80 81 TEST_LAYOUT({a1_1}, 8, 32, "1 32 1 4 a1_1", "LLLL1RRR", "LLLL1RRR"); 82 TEST_LAYOUT({a2_1}, 8, 32, "1 32 2 4 a2_1", "LLLL2RRR", "LLLL2RRR"); 83 TEST_LAYOUT({a3_1}, 8, 32, "1 32 3 4 a3_1", "LLLL3RRR", "LLLL3RRR"); 84 TEST_LAYOUT({a4_1}, 8, 32, "1 32 4 4 a4_1", "LLLL4RRR", "LLLL4RRR"); 85 TEST_LAYOUT({a7_1}, 8, 32, "1 32 7 4 a7_1", "LLLL7RRR", "LLLL7RRR"); 86 TEST_LAYOUT({a8_1}, 8, 32, "1 32 8 4 a8_1", "LLLL0RRR", "LLLLSRRR"); 87 TEST_LAYOUT({a9_1}, 8, 32, "1 32 9 4 a9_1", "LLLL01RR", "LLLL01RR"); 88 TEST_LAYOUT({a16_1}, 8, 32, "1 32 16 5 a16_1", "LLLL00RR", "LLLLSSRR"); 89 TEST_LAYOUT({p1_256}, 8, 32, "1 256 1 11 p1_256:2700", 90 "LLLLLLLLLLLLLLLLLLLLLLLLLLLLLLLL1RRR", 91 "LLLLLLLLLLLLLLLLLLLLLLLLLLLLLLLL1RRR"); 92 TEST_LAYOUT({a41_1}, 8, 32, "1 32 41 7 a41_1:7", "LLLL000001RRRRRR", 93 "LLLLSS0001RRRRRR"); 94 TEST_LAYOUT({a105_1}, 8, 32, "1 32 105 6 a105_1", "LLLL00000000000001RRRRRR", 95 "LLLLSSSSSSSSSSSSS1RRRRRR"); 96 97 { 98 SmallVector<ASanStackVariableDescription, 10> t = {a1_1, p1_256}; 99 TEST_LAYOUT(t, 8, 32, "2 256 1 11 p1_256:2700 272 1 4 a1_1", 100 "LLLLLLLLLLLLLLLLLLLLLLLLLLLLLLLL1M1R", 101 "LLLLLLLLLLLLLLLLLLLLLLLLLLLLLLLL1M1R"); 102 } 103 104 { 105 SmallVector<ASanStackVariableDescription, 10> t = {a1_1, a16_1, a41_1}; 106 TEST_LAYOUT(t, 8, 32, "3 32 1 4 a1_1 48 16 5 a16_1 80 41 7 a41_1:7", 107 "LLLL1M00MM000001RRRR", "LLLL1MSSMMSS0001RRRR"); 108 } 109 110 TEST_LAYOUT({a2_1}, 32, 32, "1 32 2 4 a2_1", "L2R", "L2R"); 111 TEST_LAYOUT({a9_1}, 32, 32, "1 32 9 4 a9_1", "L9R", "L9R"); 112 TEST_LAYOUT({a16_1}, 32, 32, "1 32 16 5 a16_1", "L16R", "LSR"); 113 TEST_LAYOUT({p1_256}, 32, 32, "1 256 1 11 p1_256:2700", 114 "LLLLLLLL1R", "LLLLLLLL1R"); 115 TEST_LAYOUT({a41_1}, 32, 32, "1 32 41 7 a41_1:7", "L09R", 116 "LS9R"); 117 TEST_LAYOUT({a105_1}, 32, 32, "1 32 105 6 a105_1", "L0009R", 118 "LSSSSR"); 119 TEST_LAYOUT({a200_1}, 32, 32, "1 32 200 6 a200_1", "L0000008RR", 120 "LSSSS008RR"); 121 122 { 123 SmallVector<ASanStackVariableDescription, 10> t = {a1_1, p1_256}; 124 TEST_LAYOUT(t, 32, 32, "2 256 1 11 p1_256:2700 320 1 4 a1_1", 125 "LLLLLLLL1M1R", "LLLLLLLL1M1R"); 126 } 127 128 { 129 SmallVector<ASanStackVariableDescription, 10> t = {a1_1, a16_1, a41_1}; 130 TEST_LAYOUT(t, 32, 32, "3 32 1 4 a1_1 96 16 5 a16_1 160 41 7 a41_1:7", 131 "L1M16M09R", "L1MSMS9R"); 132 } 133 #undef VAR 134 #undef TEST_LAYOUT 135 } 136