1 //===-- tsan_sync_test.cpp ------------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file is a part of ThreadSanitizer (TSan), a race detector. 10 // 11 //===----------------------------------------------------------------------===// 12 #include "tsan_sync.h" 13 #include "tsan_rtl.h" 14 #include "gtest/gtest.h" 15 16 namespace __tsan { 17 18 TEST(MetaMap, Basic) { 19 ThreadState *thr = cur_thread(); 20 MetaMap *m = &ctx->metamap; 21 u64 block[1] = {}; // fake malloc block 22 m->AllocBlock(thr, 0, (uptr)&block[0], 1 * sizeof(u64)); 23 MBlock *mb = m->GetBlock((uptr)&block[0]); 24 EXPECT_NE(mb, (MBlock*)0); 25 EXPECT_EQ(mb->siz, 1 * sizeof(u64)); 26 EXPECT_EQ(mb->tid, thr->tid); 27 uptr sz = m->FreeBlock(thr->proc(), (uptr)&block[0]); 28 EXPECT_EQ(sz, 1 * sizeof(u64)); 29 mb = m->GetBlock((uptr)&block[0]); 30 EXPECT_EQ(mb, (MBlock*)0); 31 } 32 33 TEST(MetaMap, FreeRange) { 34 ThreadState *thr = cur_thread(); 35 MetaMap *m = &ctx->metamap; 36 u64 block[4] = {}; // fake malloc block 37 m->AllocBlock(thr, 0, (uptr)&block[0], 1 * sizeof(u64)); 38 m->AllocBlock(thr, 0, (uptr)&block[1], 3 * sizeof(u64)); 39 MBlock *mb1 = m->GetBlock((uptr)&block[0]); 40 EXPECT_EQ(mb1->siz, 1 * sizeof(u64)); 41 MBlock *mb2 = m->GetBlock((uptr)&block[1]); 42 EXPECT_EQ(mb2->siz, 3 * sizeof(u64)); 43 m->FreeRange(thr->proc(), (uptr)&block[0], 4 * sizeof(u64)); 44 mb1 = m->GetBlock((uptr)&block[0]); 45 EXPECT_EQ(mb1, (MBlock*)0); 46 mb2 = m->GetBlock((uptr)&block[1]); 47 EXPECT_EQ(mb2, (MBlock*)0); 48 } 49 50 TEST(MetaMap, Sync) { 51 // EXPECT can call memset/etc. Disable interceptors to prevent 52 // them from detecting that we exit runtime with mutexes held. 53 ScopedIgnoreInterceptors ignore; 54 ThreadState *thr = cur_thread(); 55 MetaMap *m = &ctx->metamap; 56 u64 block[4] = {}; // fake malloc block 57 m->AllocBlock(thr, 0, (uptr)&block[0], 4 * sizeof(u64)); 58 SyncVar *s1 = m->GetSyncIfExists((uptr)&block[0]); 59 EXPECT_EQ(s1, (SyncVar*)0); 60 s1 = m->GetSyncOrCreate(thr, 0, (uptr)&block[0], false); 61 EXPECT_NE(s1, (SyncVar*)0); 62 EXPECT_EQ(s1->addr, (uptr)&block[0]); 63 SyncVar *s2 = m->GetSyncOrCreate(thr, 0, (uptr)&block[1], false); 64 EXPECT_NE(s2, (SyncVar*)0); 65 EXPECT_EQ(s2->addr, (uptr)&block[1]); 66 m->FreeBlock(thr->proc(), (uptr)&block[0]); 67 s1 = m->GetSyncIfExists((uptr)&block[0]); 68 EXPECT_EQ(s1, (SyncVar*)0); 69 s2 = m->GetSyncIfExists((uptr)&block[1]); 70 EXPECT_EQ(s2, (SyncVar*)0); 71 m->OnProcIdle(thr->proc()); 72 } 73 74 TEST(MetaMap, MoveMemory) { 75 ScopedIgnoreInterceptors ignore; 76 ThreadState *thr = cur_thread(); 77 MetaMap *m = &ctx->metamap; 78 u64 block1[4] = {}; // fake malloc block 79 u64 block2[4] = {}; // fake malloc block 80 m->AllocBlock(thr, 0, (uptr)&block1[0], 3 * sizeof(u64)); 81 m->AllocBlock(thr, 0, (uptr)&block1[3], 1 * sizeof(u64)); 82 SyncVar *s1 = m->GetSyncOrCreate(thr, 0, (uptr)&block1[0], false); 83 SyncVar *s2 = m->GetSyncOrCreate(thr, 0, (uptr)&block1[1], false); 84 m->MoveMemory((uptr)&block1[0], (uptr)&block2[0], 4 * sizeof(u64)); 85 MBlock *mb1 = m->GetBlock((uptr)&block1[0]); 86 EXPECT_EQ(mb1, (MBlock*)0); 87 MBlock *mb2 = m->GetBlock((uptr)&block1[3]); 88 EXPECT_EQ(mb2, (MBlock*)0); 89 mb1 = m->GetBlock((uptr)&block2[0]); 90 EXPECT_NE(mb1, (MBlock*)0); 91 EXPECT_EQ(mb1->siz, 3 * sizeof(u64)); 92 mb2 = m->GetBlock((uptr)&block2[3]); 93 EXPECT_NE(mb2, (MBlock*)0); 94 EXPECT_EQ(mb2->siz, 1 * sizeof(u64)); 95 s1 = m->GetSyncIfExists((uptr)&block1[0]); 96 EXPECT_EQ(s1, (SyncVar*)0); 97 s2 = m->GetSyncIfExists((uptr)&block1[1]); 98 EXPECT_EQ(s2, (SyncVar*)0); 99 s1 = m->GetSyncIfExists((uptr)&block2[0]); 100 EXPECT_NE(s1, (SyncVar*)0); 101 EXPECT_EQ(s1->addr, (uptr)&block2[0]); 102 s2 = m->GetSyncIfExists((uptr)&block2[1]); 103 EXPECT_NE(s2, (SyncVar*)0); 104 EXPECT_EQ(s2->addr, (uptr)&block2[1]); 105 m->FreeRange(thr->proc(), (uptr)&block2[0], 4 * sizeof(u64)); 106 } 107 108 TEST(MetaMap, ResetSync) { 109 ScopedIgnoreInterceptors ignore; 110 ThreadState *thr = cur_thread(); 111 MetaMap *m = &ctx->metamap; 112 u64 block[1] = {}; // fake malloc block 113 m->AllocBlock(thr, 0, (uptr)&block[0], 1 * sizeof(u64)); 114 SyncVar *s = m->GetSyncOrCreate(thr, 0, (uptr)&block[0], false); 115 s->Reset(thr->proc()); 116 uptr sz = m->FreeBlock(thr->proc(), (uptr)&block[0]); 117 EXPECT_EQ(sz, 1 * sizeof(u64)); 118 } 119 120 } // namespace __tsan 121