forked from OSchip/llvm-project
94 lines
2.8 KiB
C++
94 lines
2.8 KiB
C++
//===-- sanitizer_stackdepot_test.cc --------------------------------------===//
|
|
//
|
|
// 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 ThreadSanitizer/AddressSanitizer runtime.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "sanitizer_common/sanitizer_stackdepot.h"
|
|
#include "sanitizer_common/sanitizer_internal_defs.h"
|
|
#include "sanitizer_common/sanitizer_libc.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace __sanitizer {
|
|
|
|
TEST(SanitizerCommon, StackDepotBasic) {
|
|
uptr array[] = {1, 2, 3, 4, 5};
|
|
StackTrace s1(array, ARRAY_SIZE(array));
|
|
u32 i1 = StackDepotPut(s1);
|
|
StackTrace stack = StackDepotGet(i1);
|
|
EXPECT_NE(stack.trace, (uptr*)0);
|
|
EXPECT_EQ(ARRAY_SIZE(array), stack.size);
|
|
EXPECT_EQ(0, internal_memcmp(stack.trace, array, sizeof(array)));
|
|
}
|
|
|
|
TEST(SanitizerCommon, StackDepotAbsent) {
|
|
StackTrace stack = StackDepotGet((1 << 30) - 1);
|
|
EXPECT_EQ((uptr*)0, stack.trace);
|
|
}
|
|
|
|
TEST(SanitizerCommon, StackDepotEmptyStack) {
|
|
u32 i1 = StackDepotPut(StackTrace());
|
|
StackTrace stack = StackDepotGet(i1);
|
|
EXPECT_EQ((uptr*)0, stack.trace);
|
|
}
|
|
|
|
TEST(SanitizerCommon, StackDepotZeroId) {
|
|
StackTrace stack = StackDepotGet(0);
|
|
EXPECT_EQ((uptr*)0, stack.trace);
|
|
}
|
|
|
|
TEST(SanitizerCommon, StackDepotSame) {
|
|
uptr array[] = {1, 2, 3, 4, 6};
|
|
StackTrace s1(array, ARRAY_SIZE(array));
|
|
u32 i1 = StackDepotPut(s1);
|
|
u32 i2 = StackDepotPut(s1);
|
|
EXPECT_EQ(i1, i2);
|
|
StackTrace stack = StackDepotGet(i1);
|
|
EXPECT_NE(stack.trace, (uptr*)0);
|
|
EXPECT_EQ(ARRAY_SIZE(array), stack.size);
|
|
EXPECT_EQ(0, internal_memcmp(stack.trace, array, sizeof(array)));
|
|
}
|
|
|
|
TEST(SanitizerCommon, StackDepotSeveral) {
|
|
uptr array1[] = {1, 2, 3, 4, 7};
|
|
StackTrace s1(array1, ARRAY_SIZE(array1));
|
|
u32 i1 = StackDepotPut(s1);
|
|
uptr array2[] = {1, 2, 3, 4, 8, 9};
|
|
StackTrace s2(array2, ARRAY_SIZE(array2));
|
|
u32 i2 = StackDepotPut(s2);
|
|
EXPECT_NE(i1, i2);
|
|
}
|
|
|
|
TEST(SanitizerCommon, StackDepotReverseMap) {
|
|
uptr array1[] = {1, 2, 3, 4, 5};
|
|
uptr array2[] = {7, 1, 3, 0};
|
|
uptr array3[] = {10, 2, 5, 3};
|
|
uptr array4[] = {1, 3, 2, 5};
|
|
u32 ids[4] = {0};
|
|
StackTrace s1(array1, ARRAY_SIZE(array1));
|
|
StackTrace s2(array2, ARRAY_SIZE(array2));
|
|
StackTrace s3(array3, ARRAY_SIZE(array3));
|
|
StackTrace s4(array4, ARRAY_SIZE(array4));
|
|
ids[0] = StackDepotPut(s1);
|
|
ids[1] = StackDepotPut(s2);
|
|
ids[2] = StackDepotPut(s3);
|
|
ids[3] = StackDepotPut(s4);
|
|
|
|
StackDepotReverseMap map;
|
|
|
|
for (uptr i = 0; i < 4; i++) {
|
|
StackTrace stack = StackDepotGet(ids[i]);
|
|
StackTrace from_map = map.Get(ids[i]);
|
|
EXPECT_EQ(stack.size, from_map.size);
|
|
EXPECT_EQ(stack.trace, from_map.trace);
|
|
}
|
|
}
|
|
|
|
} // namespace __sanitizer
|