Home | History | Annotate | Download | only in tests 
      1 /*
      2  * Copyright (C) 2016 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #include <fcntl.h>
     18 #include <stdlib.h>
     19 #include <sys/prctl.h>
     20 #include <unistd.h>
     21 
     22 #include <gtest/gtest.h>
     23 
     24 #include <memunreachable/memunreachable.h>
     25 
     26 namespace android {
     27 
     28 class HiddenPointer {
     29  public:
     30   explicit HiddenPointer(size_t size = 256) { Set(malloc(size)); }
     31   ~HiddenPointer() { Free(); }
     32   void* Get() { return reinterpret_cast<void*>(~ptr_); }
     33   void Free() {
     34     free(Get());
     35     Set(nullptr);
     36   }
     37 
     38  private:
     39   void Set(void* ptr) { ptr_ = ~reinterpret_cast<uintptr_t>(ptr); }
     40   volatile uintptr_t ptr_;
     41 };
     42 
     43 // Trick the compiler into thinking a value on the stack is still referenced.
     44 static void Ref(void** ptr) {
     45   write(0, ptr, 0);
     46 }
     47 
     48 TEST(MemunreachableTest, clean) {
     49   UnreachableMemoryInfo info;
     50 
     51   ASSERT_TRUE(LogUnreachableMemory(true, 100));
     52 
     53   ASSERT_TRUE(GetUnreachableMemory(info));
     54   ASSERT_EQ(0U, info.leaks.size());
     55 }
     56 
     57 TEST(MemunreachableTest, stack) {
     58   HiddenPointer hidden_ptr;
     59 
     60   {
     61     void* ptr = hidden_ptr.Get();
     62     Ref(&ptr);
     63 
     64     UnreachableMemoryInfo info;
     65 
     66     ASSERT_TRUE(GetUnreachableMemory(info));
     67     ASSERT_EQ(0U, info.leaks.size());
     68 
     69     ptr = nullptr;
     70   }
     71 
     72   {
     73     UnreachableMemoryInfo info;
     74 
     75     ASSERT_TRUE(GetUnreachableMemory(info));
     76     ASSERT_EQ(1U, info.leaks.size());
     77   }
     78 
     79   hidden_ptr.Free();
     80 
     81   {
     82     UnreachableMemoryInfo info;
     83 
     84     ASSERT_TRUE(GetUnreachableMemory(info));
     85     ASSERT_EQ(0U, info.leaks.size());
     86   }
     87 }
     88 
     89 void* g_ptr;
     90 
     91 TEST(MemunreachableTest, global) {
     92   HiddenPointer hidden_ptr;
     93 
     94   g_ptr = hidden_ptr.Get();
     95 
     96   {
     97     UnreachableMemoryInfo info;
     98 
     99     ASSERT_TRUE(GetUnreachableMemory(info));
    100     ASSERT_EQ(0U, info.leaks.size());
    101   }
    102 
    103   g_ptr = nullptr;
    104 
    105   {
    106     UnreachableMemoryInfo info;
    107 
    108     ASSERT_TRUE(GetUnreachableMemory(info));
    109     ASSERT_EQ(1U, info.leaks.size());
    110   }
    111 
    112   hidden_ptr.Free();
    113 
    114   {
    115     UnreachableMemoryInfo info;
    116 
    117     ASSERT_TRUE(GetUnreachableMemory(info));
    118     ASSERT_EQ(0U, info.leaks.size());
    119   }
    120 }
    121 
    122 TEST(MemunreachableTest, tls) {
    123   HiddenPointer hidden_ptr;
    124   pthread_key_t key;
    125   pthread_key_create(&key, nullptr);
    126 
    127   pthread_setspecific(key, hidden_ptr.Get());
    128 
    129   {
    130     UnreachableMemoryInfo info;
    131 
    132     ASSERT_TRUE(GetUnreachableMemory(info));
    133     ASSERT_EQ(0U, info.leaks.size());
    134   }
    135 
    136   pthread_setspecific(key, nullptr);
    137 
    138   {
    139     UnreachableMemoryInfo info;
    140 
    141     ASSERT_TRUE(GetUnreachableMemory(info));
    142     ASSERT_EQ(1U, info.leaks.size());
    143   }
    144 
    145   hidden_ptr.Free();
    146 
    147   {
    148     UnreachableMemoryInfo info;
    149 
    150     ASSERT_TRUE(GetUnreachableMemory(info));
    151     ASSERT_EQ(0U, info.leaks.size());
    152   }
    153 
    154   pthread_key_delete(key);
    155 }
    156 
    157 TEST(MemunreachableTest, twice) {
    158   HiddenPointer hidden_ptr;
    159 
    160   {
    161     UnreachableMemoryInfo info;
    162 
    163     ASSERT_TRUE(GetUnreachableMemory(info));
    164     ASSERT_EQ(1U, info.leaks.size());
    165   }
    166 
    167   {
    168     UnreachableMemoryInfo info;
    169 
    170     ASSERT_TRUE(GetUnreachableMemory(info));
    171     ASSERT_EQ(1U, info.leaks.size());
    172   }
    173 
    174   hidden_ptr.Free();
    175 
    176   {
    177     UnreachableMemoryInfo info;
    178 
    179     ASSERT_TRUE(GetUnreachableMemory(info));
    180     ASSERT_EQ(0U, info.leaks.size());
    181   }
    182 }
    183 
    184 TEST(MemunreachableTest, log) {
    185   HiddenPointer hidden_ptr;
    186 
    187   ASSERT_TRUE(LogUnreachableMemory(true, 100));
    188 
    189   hidden_ptr.Free();
    190 
    191   {
    192     UnreachableMemoryInfo info;
    193 
    194     ASSERT_TRUE(GetUnreachableMemory(info));
    195     ASSERT_EQ(0U, info.leaks.size());
    196   }
    197 }
    198 
    199 TEST(MemunreachableTest, notdumpable) {
    200   ASSERT_EQ(0, prctl(PR_SET_DUMPABLE, 0));
    201 
    202   HiddenPointer hidden_ptr;
    203 
    204   ASSERT_TRUE(LogUnreachableMemory(true, 100));
    205 
    206   ASSERT_EQ(0, prctl(PR_SET_DUMPABLE, 1));
    207 }
    208 
    209 TEST(MemunreachableTest, leak_lots) {
    210   std::vector<HiddenPointer> hidden_ptrs;
    211   hidden_ptrs.resize(1024);
    212 
    213   ASSERT_TRUE(LogUnreachableMemory(true, 100));
    214 }
    215 
    216 }  // namespace android
    217