Home | History | Annotate | Download | only in unit 
      1 //===-- tsan_stack_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 (TSan), a race detector.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 #include "tsan_sync.h"
     14 #include "tsan_rtl.h"
     15 #include "gtest/gtest.h"
     16 #include <string.h>
     17 
     18 namespace __tsan {
     19 
     20 static void TestStackTrace(StackTrace *trace) {
     21   ThreadState thr(0, 0, 0, 0, 0, 0, 0, 0);
     22 
     23   trace->ObtainCurrent(&thr, 0);
     24   EXPECT_EQ(trace->Size(), (uptr)0);
     25 
     26   trace->ObtainCurrent(&thr, 42);
     27   EXPECT_EQ(trace->Size(), (uptr)1);
     28   EXPECT_EQ(trace->Get(0), (uptr)42);
     29 
     30   *thr.shadow_stack_pos++ = 100;
     31   *thr.shadow_stack_pos++ = 101;
     32   trace->ObtainCurrent(&thr, 0);
     33   EXPECT_EQ(trace->Size(), (uptr)2);
     34   EXPECT_EQ(trace->Get(0), (uptr)100);
     35   EXPECT_EQ(trace->Get(1), (uptr)101);
     36 
     37   trace->ObtainCurrent(&thr, 42);
     38   EXPECT_EQ(trace->Size(), (uptr)3);
     39   EXPECT_EQ(trace->Get(0), (uptr)100);
     40   EXPECT_EQ(trace->Get(1), (uptr)101);
     41   EXPECT_EQ(trace->Get(2), (uptr)42);
     42 }
     43 
     44 TEST(StackTrace, Basic) {
     45   ScopedInRtl in_rtl;
     46   StackTrace trace;
     47   TestStackTrace(&trace);
     48 }
     49 
     50 TEST(StackTrace, StaticBasic) {
     51   ScopedInRtl in_rtl;
     52   uptr buf[10];
     53   StackTrace trace1(buf, 10);
     54   TestStackTrace(&trace1);
     55   StackTrace trace2(buf, 3);
     56   TestStackTrace(&trace2);
     57 }
     58 
     59 TEST(StackTrace, StaticTrim) {
     60   ScopedInRtl in_rtl;
     61   uptr buf[2];
     62   StackTrace trace(buf, 2);
     63   ThreadState thr(0, 0, 0, 0, 0, 0, 0, 0);
     64 
     65   *thr.shadow_stack_pos++ = 100;
     66   *thr.shadow_stack_pos++ = 101;
     67   *thr.shadow_stack_pos++ = 102;
     68   trace.ObtainCurrent(&thr, 0);
     69   EXPECT_EQ(trace.Size(), (uptr)2);
     70   EXPECT_EQ(trace.Get(0), (uptr)101);
     71   EXPECT_EQ(trace.Get(1), (uptr)102);
     72 
     73   trace.ObtainCurrent(&thr, 42);
     74   EXPECT_EQ(trace.Size(), (uptr)2);
     75   EXPECT_EQ(trace.Get(0), (uptr)102);
     76   EXPECT_EQ(trace.Get(1), (uptr)42);
     77 }
     78 
     79 
     80 }  // namespace __tsan
     81