Home | History | Annotate | Download | only in tests 
      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