1 //===-- sanitizer_stackdepot_test.cc --------------------------------------===// 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 // 10 // This file is a part of ThreadSanitizer/AddressSanitizer runtime. 11 // 12 //===----------------------------------------------------------------------===// 13 #include "sanitizer_common/sanitizer_stackdepot.h" 14 #include "sanitizer_common/sanitizer_internal_defs.h" 15 #include "sanitizer_common/sanitizer_libc.h" 16 #include "gtest/gtest.h" 17 18 namespace __sanitizer { 19 20 TEST(SanitizerCommon, StackDepotBasic) { 21 uptr s1[] = {1, 2, 3, 4, 5}; 22 u32 i1 = StackDepotPut(s1, ARRAY_SIZE(s1)); 23 uptr sz1 = 0; 24 const uptr *sp1 = StackDepotGet(i1, &sz1); 25 EXPECT_NE(sp1, (uptr*)0); 26 EXPECT_EQ(sz1, ARRAY_SIZE(s1)); 27 EXPECT_EQ(internal_memcmp(sp1, s1, sizeof(s1)), 0); 28 } 29 30 TEST(SanitizerCommon, StackDepotAbsent) { 31 uptr sz1 = 0; 32 const uptr *sp1 = StackDepotGet((1 << 30) - 1, &sz1); 33 EXPECT_EQ(sp1, (uptr*)0); 34 } 35 36 TEST(SanitizerCommon, StackDepotEmptyStack) { 37 u32 i1 = StackDepotPut(0, 0); 38 uptr sz1 = 0; 39 const uptr *sp1 = StackDepotGet(i1, &sz1); 40 EXPECT_EQ(sp1, (uptr*)0); 41 } 42 43 TEST(SanitizerCommon, StackDepotZeroId) { 44 uptr sz1 = 0; 45 const uptr *sp1 = StackDepotGet(0, &sz1); 46 EXPECT_EQ(sp1, (uptr*)0); 47 } 48 49 TEST(SanitizerCommon, StackDepotSame) { 50 uptr s1[] = {1, 2, 3, 4, 6}; 51 u32 i1 = StackDepotPut(s1, ARRAY_SIZE(s1)); 52 u32 i2 = StackDepotPut(s1, ARRAY_SIZE(s1)); 53 EXPECT_EQ(i1, i2); 54 uptr sz1 = 0; 55 const uptr *sp1 = StackDepotGet(i1, &sz1); 56 EXPECT_NE(sp1, (uptr*)0); 57 EXPECT_EQ(sz1, ARRAY_SIZE(s1)); 58 EXPECT_EQ(internal_memcmp(sp1, s1, sizeof(s1)), 0); 59 } 60 61 TEST(SanitizerCommon, StackDepotSeveral) { 62 uptr s1[] = {1, 2, 3, 4, 7}; 63 u32 i1 = StackDepotPut(s1, ARRAY_SIZE(s1)); 64 uptr s2[] = {1, 2, 3, 4, 8, 9}; 65 u32 i2 = StackDepotPut(s2, ARRAY_SIZE(s2)); 66 EXPECT_NE(i1, i2); 67 } 68 69 } // namespace __sanitizer 70