| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 1 | /* | 
 | 2 |  * Copyright (C) 2013 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 |  | 
| Christopher Ferris | ca09ce9 | 2015-03-31 17:28:22 -0700 | [diff] [blame] | 17 | #define _GNU_SOURCE 1 | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 18 | #include <dirent.h> | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 19 | #include <dlfcn.h> | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 20 | #include <errno.h> | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 21 | #include <fcntl.h> | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 22 | #include <inttypes.h> | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 23 | #include <pthread.h> | 
 | 24 | #include <signal.h> | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 25 | #include <stdint.h> | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 26 | #include <stdio.h> | 
 | 27 | #include <stdlib.h> | 
 | 28 | #include <string.h> | 
 | 29 | #include <sys/ptrace.h> | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 30 | #include <sys/stat.h> | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 31 | #include <sys/types.h> | 
 | 32 | #include <sys/wait.h> | 
 | 33 | #include <time.h> | 
 | 34 | #include <unistd.h> | 
 | 35 |  | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 36 | #include <algorithm> | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 37 | #include <list> | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 38 | #include <memory> | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 39 | #include <ostream> | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 40 | #include <string> | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 41 | #include <vector> | 
 | 42 |  | 
| Dan Albert | 23f750b | 2015-04-30 12:52:21 -0700 | [diff] [blame] | 43 | #include <backtrace/Backtrace.h> | 
 | 44 | #include <backtrace/BacktraceMap.h> | 
 | 45 |  | 
| Christopher Ferris | f5e568e | 2017-03-22 13:18:31 -0700 | [diff] [blame] | 46 | #include <android-base/macros.h> | 
| Elliott Hughes | 4f71319 | 2015-12-04 22:00:26 -0800 | [diff] [blame] | 47 | #include <android-base/stringprintf.h> | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 48 | #include <android-base/unique_fd.h> | 
| Dan Albert | 23f750b | 2015-04-30 12:52:21 -0700 | [diff] [blame] | 49 | #include <cutils/atomic.h> | 
 | 50 | #include <cutils/threads.h> | 
 | 51 |  | 
 | 52 | #include <gtest/gtest.h> | 
 | 53 |  | 
 | 54 | // For the THREAD_SIGNAL definition. | 
 | 55 | #include "BacktraceCurrent.h" | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 56 | #include "backtrace_testlib.h" | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 57 | #include "thread_utils.h" | 
 | 58 |  | 
 | 59 | // Number of microseconds per milliseconds. | 
 | 60 | #define US_PER_MSEC             1000 | 
 | 61 |  | 
 | 62 | // Number of nanoseconds in a second. | 
 | 63 | #define NS_PER_SEC              1000000000ULL | 
 | 64 |  | 
 | 65 | // Number of simultaneous dumping operations to perform. | 
| Christopher Ferris | 3cdbfdc | 2014-11-08 15:57:11 -0800 | [diff] [blame] | 66 | #define NUM_THREADS  40 | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 67 |  | 
 | 68 | // Number of simultaneous threads running in our forked process. | 
 | 69 | #define NUM_PTRACE_THREADS 5 | 
 | 70 |  | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 71 | struct thread_t { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 72 |   pid_t tid; | 
 | 73 |   int32_t state; | 
 | 74 |   pthread_t threadId; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 75 |   void* data; | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 76 | }; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 77 |  | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 78 | struct dump_thread_t { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 79 |   thread_t thread; | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 80 |   BacktraceMap* map; | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 81 |   Backtrace* backtrace; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 82 |   int32_t* now; | 
 | 83 |   int32_t done; | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 84 | }; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 85 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 86 | typedef Backtrace* (*create_func_t)(pid_t, pid_t, BacktraceMap*); | 
 | 87 | typedef BacktraceMap* (*map_create_func_t)(pid_t, bool); | 
 | 88 |  | 
 | 89 | static void VerifyLevelDump(Backtrace* backtrace, create_func_t create_func = nullptr, | 
 | 90 |                             map_create_func_t map_func = nullptr); | 
 | 91 | static void VerifyMaxDump(Backtrace* backtrace, create_func_t create_func = nullptr, | 
 | 92 |                           map_create_func_t map_func = nullptr); | 
 | 93 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 94 | static uint64_t NanoTime() { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 95 |   struct timespec t = { 0, 0 }; | 
 | 96 |   clock_gettime(CLOCK_MONOTONIC, &t); | 
 | 97 |   return static_cast<uint64_t>(t.tv_sec * NS_PER_SEC + t.tv_nsec); | 
 | 98 | } | 
 | 99 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 100 | static std::string DumpFrames(Backtrace* backtrace) { | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 101 |   if (backtrace->NumFrames() == 0) { | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 102 |     return "   No frames to dump.\n"; | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 103 |   } | 
 | 104 |  | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 105 |   std::string frame; | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 106 |   for (size_t i = 0; i < backtrace->NumFrames(); i++) { | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 107 |     frame += "   " + backtrace->FormatFrameData(i) + '\n'; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 108 |   } | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 109 |   return frame; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 110 | } | 
 | 111 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 112 | static void WaitForStop(pid_t pid) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 113 |   uint64_t start = NanoTime(); | 
 | 114 |  | 
 | 115 |   siginfo_t si; | 
 | 116 |   while (ptrace(PTRACE_GETSIGINFO, pid, 0, &si) < 0 && (errno == EINTR || errno == ESRCH)) { | 
 | 117 |     if ((NanoTime() - start) > NS_PER_SEC) { | 
 | 118 |       printf("The process did not get to a stopping point in 1 second.\n"); | 
 | 119 |       break; | 
 | 120 |     } | 
 | 121 |     usleep(US_PER_MSEC); | 
 | 122 |   } | 
 | 123 | } | 
 | 124 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 125 | static void CreateRemoteProcess(pid_t* pid) { | 
 | 126 |   if ((*pid = fork()) == 0) { | 
 | 127 |     while (true) | 
 | 128 |       ; | 
 | 129 |     _exit(0); | 
 | 130 |   } | 
 | 131 |   ASSERT_NE(-1, *pid); | 
 | 132 |  | 
 | 133 |   ASSERT_TRUE(ptrace(PTRACE_ATTACH, *pid, 0, 0) == 0); | 
 | 134 |  | 
 | 135 |   // Wait for the process to get to a stopping point. | 
 | 136 |   WaitForStop(*pid); | 
 | 137 | } | 
 | 138 |  | 
 | 139 | static void FinishRemoteProcess(pid_t pid) { | 
 | 140 |   ASSERT_TRUE(ptrace(PTRACE_DETACH, pid, 0, 0) == 0); | 
 | 141 |  | 
 | 142 |   kill(pid, SIGKILL); | 
 | 143 |   ASSERT_EQ(waitpid(pid, nullptr, 0), pid); | 
 | 144 | } | 
 | 145 |  | 
 | 146 | static bool ReadyLevelBacktrace(Backtrace* backtrace) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 147 |   // See if test_level_four is in the backtrace. | 
 | 148 |   bool found = false; | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 149 |   for (Backtrace::const_iterator it = backtrace->begin(); it != backtrace->end(); ++it) { | 
 | 150 |     if (it->func_name == "test_level_four") { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 151 |       found = true; | 
 | 152 |       break; | 
 | 153 |     } | 
 | 154 |   } | 
 | 155 |  | 
 | 156 |   return found; | 
 | 157 | } | 
 | 158 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 159 | static void VerifyLevelDump(Backtrace* backtrace, create_func_t, map_create_func_t) { | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 160 |   ASSERT_GT(backtrace->NumFrames(), static_cast<size_t>(0)) | 
 | 161 |     << DumpFrames(backtrace); | 
 | 162 |   ASSERT_LT(backtrace->NumFrames(), static_cast<size_t>(MAX_BACKTRACE_FRAMES)) | 
 | 163 |     << DumpFrames(backtrace); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 164 |  | 
 | 165 |   // Look through the frames starting at the highest to find the | 
 | 166 |   // frame we want. | 
 | 167 |   size_t frame_num = 0; | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 168 |   for (size_t i = backtrace->NumFrames()-1; i > 2; i--) { | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 169 |     if (backtrace->GetFrame(i)->func_name == "test_level_one") { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 170 |       frame_num = i; | 
 | 171 |       break; | 
 | 172 |     } | 
 | 173 |   } | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 174 |   ASSERT_LT(static_cast<size_t>(0), frame_num) << DumpFrames(backtrace); | 
 | 175 |   ASSERT_LE(static_cast<size_t>(3), frame_num) << DumpFrames(backtrace); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 176 |  | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 177 |   ASSERT_EQ(backtrace->GetFrame(frame_num)->func_name, "test_level_one") | 
 | 178 |     << DumpFrames(backtrace); | 
 | 179 |   ASSERT_EQ(backtrace->GetFrame(frame_num-1)->func_name, "test_level_two") | 
 | 180 |     << DumpFrames(backtrace); | 
 | 181 |   ASSERT_EQ(backtrace->GetFrame(frame_num-2)->func_name, "test_level_three") | 
 | 182 |     << DumpFrames(backtrace); | 
 | 183 |   ASSERT_EQ(backtrace->GetFrame(frame_num-3)->func_name, "test_level_four") | 
 | 184 |     << DumpFrames(backtrace); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 185 | } | 
 | 186 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 187 | static void VerifyLevelBacktrace(void*) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 188 |   std::unique_ptr<Backtrace> backtrace( | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 189 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 190 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 191 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 192 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 193 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 194 |   VerifyLevelDump(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 195 | } | 
 | 196 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 197 | static bool ReadyMaxBacktrace(Backtrace* backtrace) { | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 198 |   return (backtrace->NumFrames() == MAX_BACKTRACE_FRAMES); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 199 | } | 
 | 200 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 201 | static void VerifyMaxDump(Backtrace* backtrace, create_func_t, map_create_func_t) { | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 202 |   ASSERT_EQ(backtrace->NumFrames(), static_cast<size_t>(MAX_BACKTRACE_FRAMES)) | 
 | 203 |     << DumpFrames(backtrace); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 204 |   // Verify that the last frame is our recursive call. | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 205 |   ASSERT_EQ(backtrace->GetFrame(MAX_BACKTRACE_FRAMES-1)->func_name, "test_recursive_call") | 
 | 206 |     << DumpFrames(backtrace); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 207 | } | 
 | 208 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 209 | static void VerifyMaxBacktrace(void*) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 210 |   std::unique_ptr<Backtrace> backtrace( | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 211 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 212 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 213 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 214 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 215 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 216 |   VerifyMaxDump(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 217 | } | 
 | 218 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 219 | static void ThreadSetState(void* data) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 220 |   thread_t* thread = reinterpret_cast<thread_t*>(data); | 
 | 221 |   android_atomic_acquire_store(1, &thread->state); | 
 | 222 |   volatile int i = 0; | 
 | 223 |   while (thread->state) { | 
 | 224 |     i++; | 
 | 225 |   } | 
 | 226 | } | 
 | 227 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 228 | static bool WaitForNonZero(int32_t* value, uint64_t seconds) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 229 |   uint64_t start = NanoTime(); | 
 | 230 |   do { | 
 | 231 |     if (android_atomic_acquire_load(value)) { | 
 | 232 |       return true; | 
 | 233 |     } | 
 | 234 |   } while ((NanoTime() - start) < seconds * NS_PER_SEC); | 
 | 235 |   return false; | 
 | 236 | } | 
 | 237 |  | 
| Christopher Ferris | ca09ce9 | 2015-03-31 17:28:22 -0700 | [diff] [blame] | 238 | TEST(libbacktrace, local_no_unwind_frames) { | 
 | 239 |   // Verify that a local unwind does not include any frames within | 
 | 240 |   // libunwind or libbacktrace. | 
 | 241 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), getpid())); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 242 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | ca09ce9 | 2015-03-31 17:28:22 -0700 | [diff] [blame] | 243 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 244 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | ca09ce9 | 2015-03-31 17:28:22 -0700 | [diff] [blame] | 245 |  | 
 | 246 |   ASSERT_TRUE(backtrace->NumFrames() != 0); | 
 | 247 |   for (const auto& frame : *backtrace ) { | 
 | 248 |     if (BacktraceMap::IsValid(frame.map)) { | 
 | 249 |       const std::string name = basename(frame.map.name.c_str()); | 
 | 250 |       ASSERT_TRUE(name != "libunwind.so" && name != "libbacktrace.so") | 
 | 251 |         << DumpFrames(backtrace.get()); | 
 | 252 |     } | 
 | 253 |     break; | 
 | 254 |   } | 
 | 255 | } | 
 | 256 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 257 | TEST(libbacktrace, local_trace) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 258 |   ASSERT_NE(test_level_one(1, 2, 3, 4, VerifyLevelBacktrace, nullptr), 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 259 | } | 
 | 260 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 261 | static void VerifyIgnoreFrames(Backtrace* bt_all, Backtrace* bt_ign1, Backtrace* bt_ign2, | 
 | 262 |                                const char* cur_proc) { | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 263 |   ASSERT_EQ(bt_all->NumFrames(), bt_ign1->NumFrames() + 1) << "All backtrace:\n" | 
 | 264 |                                                            << DumpFrames(bt_all) | 
 | 265 |                                                            << "Ignore 1 backtrace:\n" | 
 | 266 |                                                            << DumpFrames(bt_ign1); | 
 | 267 |   ASSERT_EQ(bt_all->NumFrames(), bt_ign2->NumFrames() + 2) << "All backtrace:\n" | 
 | 268 |                                                            << DumpFrames(bt_all) | 
 | 269 |                                                            << "Ignore 2 backtrace:\n" | 
 | 270 |                                                            << DumpFrames(bt_ign2); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 271 |  | 
 | 272 |   // Check all of the frames are the same > the current frame. | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 273 |   bool check = (cur_proc == nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 274 |   for (size_t i = 0; i < bt_ign2->NumFrames(); i++) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 275 |     if (check) { | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 276 |       EXPECT_EQ(bt_ign2->GetFrame(i)->pc, bt_ign1->GetFrame(i+1)->pc); | 
 | 277 |       EXPECT_EQ(bt_ign2->GetFrame(i)->sp, bt_ign1->GetFrame(i+1)->sp); | 
 | 278 |       EXPECT_EQ(bt_ign2->GetFrame(i)->stack_size, bt_ign1->GetFrame(i+1)->stack_size); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 279 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 280 |       EXPECT_EQ(bt_ign2->GetFrame(i)->pc, bt_all->GetFrame(i+2)->pc); | 
 | 281 |       EXPECT_EQ(bt_ign2->GetFrame(i)->sp, bt_all->GetFrame(i+2)->sp); | 
 | 282 |       EXPECT_EQ(bt_ign2->GetFrame(i)->stack_size, bt_all->GetFrame(i+2)->stack_size); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 283 |     } | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 284 |     if (!check && bt_ign2->GetFrame(i)->func_name == cur_proc) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 285 |       check = true; | 
 | 286 |     } | 
 | 287 |   } | 
 | 288 | } | 
 | 289 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 290 | static void VerifyLevelIgnoreFrames(void*) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 291 |   std::unique_ptr<Backtrace> all( | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 292 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 293 |   ASSERT_TRUE(all.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 294 |   ASSERT_TRUE(all->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 295 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, all->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 296 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 297 |   std::unique_ptr<Backtrace> ign1( | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 298 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 299 |   ASSERT_TRUE(ign1.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 300 |   ASSERT_TRUE(ign1->Unwind(1)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 301 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, ign1->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 302 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 303 |   std::unique_ptr<Backtrace> ign2( | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 304 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 305 |   ASSERT_TRUE(ign2.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 306 |   ASSERT_TRUE(ign2->Unwind(2)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 307 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, ign2->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 308 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 309 |   VerifyIgnoreFrames(all.get(), ign1.get(), ign2.get(), "VerifyLevelIgnoreFrames"); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 310 | } | 
 | 311 |  | 
 | 312 | TEST(libbacktrace, local_trace_ignore_frames) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 313 |   ASSERT_NE(test_level_one(1, 2, 3, 4, VerifyLevelIgnoreFrames, nullptr), 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 314 | } | 
 | 315 |  | 
 | 316 | TEST(libbacktrace, local_max_trace) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 317 |   ASSERT_NE(test_recursive_call(MAX_BACKTRACE_FRAMES+10, VerifyMaxBacktrace, nullptr), 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 318 | } | 
 | 319 |  | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 320 | static void VerifyProcTest(pid_t pid, pid_t tid, bool (*ReadyFunc)(Backtrace*), | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 321 |                            void (*VerifyFunc)(Backtrace*, create_func_t, map_create_func_t), | 
 | 322 |                            create_func_t create_func, map_create_func_t map_create_func) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 323 |   pid_t ptrace_tid; | 
 | 324 |   if (tid < 0) { | 
 | 325 |     ptrace_tid = pid; | 
 | 326 |   } else { | 
 | 327 |     ptrace_tid = tid; | 
 | 328 |   } | 
 | 329 |   uint64_t start = NanoTime(); | 
 | 330 |   bool verified = false; | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 331 |   std::string last_dump; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 332 |   do { | 
 | 333 |     usleep(US_PER_MSEC); | 
 | 334 |     if (ptrace(PTRACE_ATTACH, ptrace_tid, 0, 0) == 0) { | 
 | 335 |       // Wait for the process to get to a stopping point. | 
 | 336 |       WaitForStop(ptrace_tid); | 
 | 337 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 338 |       std::unique_ptr<BacktraceMap> map; | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 339 |       map.reset(map_create_func(pid, false)); | 
 | 340 |       std::unique_ptr<Backtrace> backtrace(create_func(pid, tid, map.get())); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 341 |       ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 342 |       ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 343 |       ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 344 |       if (ReadyFunc(backtrace.get())) { | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 345 |         VerifyFunc(backtrace.get(), create_func, map_create_func); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 346 |         verified = true; | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 347 |       } else { | 
 | 348 |         last_dump = DumpFrames(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 349 |       } | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 350 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 351 |       ASSERT_TRUE(ptrace(PTRACE_DETACH, ptrace_tid, 0, 0) == 0); | 
 | 352 |     } | 
 | 353 |     // If 5 seconds have passed, then we are done. | 
 | 354 |   } while (!verified && (NanoTime() - start) <= 5 * NS_PER_SEC); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 355 |   ASSERT_TRUE(verified) << "Last backtrace:\n" << last_dump; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 356 | } | 
 | 357 |  | 
 | 358 | TEST(libbacktrace, ptrace_trace) { | 
 | 359 |   pid_t pid; | 
 | 360 |   if ((pid = fork()) == 0) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 361 |     ASSERT_NE(test_level_one(1, 2, 3, 4, nullptr, nullptr), 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 362 |     _exit(1); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 363 |   } | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 364 |   VerifyProcTest(pid, BACKTRACE_CURRENT_THREAD, ReadyLevelBacktrace, VerifyLevelDump, | 
 | 365 |                  Backtrace::Create, BacktraceMap::Create); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 366 |  | 
 | 367 |   kill(pid, SIGKILL); | 
 | 368 |   int status; | 
 | 369 |   ASSERT_EQ(waitpid(pid, &status, 0), pid); | 
 | 370 | } | 
 | 371 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 372 | TEST(libbacktrace, ptrace_max_trace) { | 
 | 373 |   pid_t pid; | 
 | 374 |   if ((pid = fork()) == 0) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 375 |     ASSERT_NE(test_recursive_call(MAX_BACKTRACE_FRAMES+10, nullptr, nullptr), 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 376 |     _exit(1); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 377 |   } | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 378 |   VerifyProcTest(pid, BACKTRACE_CURRENT_THREAD, ReadyMaxBacktrace, VerifyMaxDump, Backtrace::Create, | 
 | 379 |                  BacktraceMap::Create); | 
 | 380 |  | 
 | 381 |   kill(pid, SIGKILL); | 
 | 382 |   int status; | 
 | 383 |   ASSERT_EQ(waitpid(pid, &status, 0), pid); | 
 | 384 | } | 
 | 385 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 386 | static void VerifyProcessIgnoreFrames(Backtrace* bt_all, create_func_t create_func, | 
 | 387 |                                       map_create_func_t map_create_func) { | 
 | 388 |   std::unique_ptr<BacktraceMap> map(map_create_func(bt_all->Pid(), false)); | 
 | 389 |   std::unique_ptr<Backtrace> ign1(create_func(bt_all->Pid(), BACKTRACE_CURRENT_THREAD, map.get())); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 390 |   ASSERT_TRUE(ign1.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 391 |   ASSERT_TRUE(ign1->Unwind(1)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 392 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, ign1->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 393 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 394 |   std::unique_ptr<Backtrace> ign2(create_func(bt_all->Pid(), BACKTRACE_CURRENT_THREAD, map.get())); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 395 |   ASSERT_TRUE(ign2.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 396 |   ASSERT_TRUE(ign2->Unwind(2)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 397 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, ign2->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 398 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 399 |   VerifyIgnoreFrames(bt_all, ign1.get(), ign2.get(), nullptr); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 400 | } | 
 | 401 |  | 
 | 402 | TEST(libbacktrace, ptrace_ignore_frames) { | 
 | 403 |   pid_t pid; | 
 | 404 |   if ((pid = fork()) == 0) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 405 |     ASSERT_NE(test_level_one(1, 2, 3, 4, nullptr, nullptr), 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 406 |     _exit(1); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 407 |   } | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 408 |   VerifyProcTest(pid, BACKTRACE_CURRENT_THREAD, ReadyLevelBacktrace, VerifyProcessIgnoreFrames, | 
 | 409 |                  Backtrace::Create, BacktraceMap::Create); | 
 | 410 |  | 
 | 411 |   kill(pid, SIGKILL); | 
 | 412 |   int status; | 
 | 413 |   ASSERT_EQ(waitpid(pid, &status, 0), pid); | 
 | 414 | } | 
 | 415 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 416 | // Create a process with multiple threads and dump all of the threads. | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 417 | static void* PtraceThreadLevelRun(void*) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 418 |   EXPECT_NE(test_level_one(1, 2, 3, 4, nullptr, nullptr), 0); | 
 | 419 |   return nullptr; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 420 | } | 
 | 421 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 422 | static void GetThreads(pid_t pid, std::vector<pid_t>* threads) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 423 |   // Get the list of tasks. | 
 | 424 |   char task_path[128]; | 
 | 425 |   snprintf(task_path, sizeof(task_path), "/proc/%d/task", pid); | 
 | 426 |  | 
| James Hawkins | 588a2ca | 2016-02-18 14:52:46 -0800 | [diff] [blame] | 427 |   std::unique_ptr<DIR, decltype(&closedir)> tasks_dir(opendir(task_path), closedir); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 428 |   ASSERT_TRUE(tasks_dir != nullptr); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 429 |   struct dirent* entry; | 
| James Hawkins | 588a2ca | 2016-02-18 14:52:46 -0800 | [diff] [blame] | 430 |   while ((entry = readdir(tasks_dir.get())) != nullptr) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 431 |     char* end; | 
 | 432 |     pid_t tid = strtoul(entry->d_name, &end, 10); | 
 | 433 |     if (*end == '\0') { | 
 | 434 |       threads->push_back(tid); | 
 | 435 |     } | 
 | 436 |   } | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 437 | } | 
 | 438 |  | 
 | 439 | TEST(libbacktrace, ptrace_threads) { | 
 | 440 |   pid_t pid; | 
 | 441 |   if ((pid = fork()) == 0) { | 
 | 442 |     for (size_t i = 0; i < NUM_PTRACE_THREADS; i++) { | 
 | 443 |       pthread_attr_t attr; | 
 | 444 |       pthread_attr_init(&attr); | 
 | 445 |       pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | 
 | 446 |  | 
 | 447 |       pthread_t thread; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 448 |       ASSERT_TRUE(pthread_create(&thread, &attr, PtraceThreadLevelRun, nullptr) == 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 449 |     } | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 450 |     ASSERT_NE(test_level_one(1, 2, 3, 4, nullptr, nullptr), 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 451 |     _exit(1); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 452 |   } | 
 | 453 |  | 
 | 454 |   // Check to see that all of the threads are running before unwinding. | 
 | 455 |   std::vector<pid_t> threads; | 
 | 456 |   uint64_t start = NanoTime(); | 
 | 457 |   do { | 
 | 458 |     usleep(US_PER_MSEC); | 
 | 459 |     threads.clear(); | 
 | 460 |     GetThreads(pid, &threads); | 
 | 461 |   } while ((threads.size() != NUM_PTRACE_THREADS + 1) && | 
 | 462 |       ((NanoTime() - start) <= 5 * NS_PER_SEC)); | 
 | 463 |   ASSERT_EQ(threads.size(), static_cast<size_t>(NUM_PTRACE_THREADS + 1)); | 
 | 464 |  | 
 | 465 |   ASSERT_TRUE(ptrace(PTRACE_ATTACH, pid, 0, 0) == 0); | 
 | 466 |   WaitForStop(pid); | 
 | 467 |   for (std::vector<int>::const_iterator it = threads.begin(); it != threads.end(); ++it) { | 
 | 468 |     // Skip the current forked process, we only care about the threads. | 
 | 469 |     if (pid == *it) { | 
 | 470 |       continue; | 
 | 471 |     } | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 472 |     VerifyProcTest(pid, *it, ReadyLevelBacktrace, VerifyLevelDump, Backtrace::Create, | 
 | 473 |                    BacktraceMap::Create); | 
 | 474 |   } | 
 | 475 |  | 
 | 476 |   FinishRemoteProcess(pid); | 
 | 477 | } | 
 | 478 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 479 | void VerifyLevelThread(void*) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 480 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), gettid())); | 
 | 481 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 482 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 483 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 484 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 485 |   VerifyLevelDump(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 486 | } | 
 | 487 |  | 
 | 488 | TEST(libbacktrace, thread_current_level) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 489 |   ASSERT_NE(test_level_one(1, 2, 3, 4, VerifyLevelThread, nullptr), 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 490 | } | 
 | 491 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 492 | static void VerifyMaxThread(void*) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 493 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), gettid())); | 
 | 494 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 495 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 496 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 497 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 498 |   VerifyMaxDump(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 499 | } | 
 | 500 |  | 
 | 501 | TEST(libbacktrace, thread_current_max) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 502 |   ASSERT_NE(test_recursive_call(MAX_BACKTRACE_FRAMES+10, VerifyMaxThread, nullptr), 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 503 | } | 
 | 504 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 505 | static void* ThreadLevelRun(void* data) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 506 |   thread_t* thread = reinterpret_cast<thread_t*>(data); | 
 | 507 |  | 
 | 508 |   thread->tid = gettid(); | 
 | 509 |   EXPECT_NE(test_level_one(1, 2, 3, 4, ThreadSetState, data), 0); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 510 |   return nullptr; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 511 | } | 
 | 512 |  | 
 | 513 | TEST(libbacktrace, thread_level_trace) { | 
 | 514 |   pthread_attr_t attr; | 
 | 515 |   pthread_attr_init(&attr); | 
 | 516 |   pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | 
 | 517 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 518 |   thread_t thread_data = { 0, 0, 0, nullptr }; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 519 |   pthread_t thread; | 
 | 520 |   ASSERT_TRUE(pthread_create(&thread, &attr, ThreadLevelRun, &thread_data) == 0); | 
 | 521 |  | 
 | 522 |   // Wait up to 2 seconds for the tid to be set. | 
 | 523 |   ASSERT_TRUE(WaitForNonZero(&thread_data.state, 2)); | 
 | 524 |  | 
| Christopher Ferris | aa63d9f | 2014-04-29 09:35:30 -0700 | [diff] [blame] | 525 |   // Make sure that the thread signal used is not visible when compiled for | 
 | 526 |   // the target. | 
 | 527 | #if !defined(__GLIBC__) | 
 | 528 |   ASSERT_LT(THREAD_SIGNAL, SIGRTMIN); | 
 | 529 | #endif | 
 | 530 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 531 |   // Save the current signal action and make sure it is restored afterwards. | 
 | 532 |   struct sigaction cur_action; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 533 |   ASSERT_TRUE(sigaction(THREAD_SIGNAL, nullptr, &cur_action) == 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 534 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 535 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), thread_data.tid)); | 
 | 536 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 537 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 538 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 539 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 540 |   VerifyLevelDump(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 541 |  | 
 | 542 |   // Tell the thread to exit its infinite loop. | 
 | 543 |   android_atomic_acquire_store(0, &thread_data.state); | 
 | 544 |  | 
 | 545 |   // Verify that the old action was restored. | 
 | 546 |   struct sigaction new_action; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 547 |   ASSERT_TRUE(sigaction(THREAD_SIGNAL, nullptr, &new_action) == 0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 548 |   EXPECT_EQ(cur_action.sa_sigaction, new_action.sa_sigaction); | 
| Christopher Ferris | 3cdbfdc | 2014-11-08 15:57:11 -0800 | [diff] [blame] | 549 |   // The SA_RESTORER flag gets set behind our back, so a direct comparison | 
 | 550 |   // doesn't work unless we mask the value off. Mips doesn't have this | 
 | 551 |   // flag, so skip this on that platform. | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 552 | #if defined(SA_RESTORER) | 
| Christopher Ferris | 3cdbfdc | 2014-11-08 15:57:11 -0800 | [diff] [blame] | 553 |   cur_action.sa_flags &= ~SA_RESTORER; | 
 | 554 |   new_action.sa_flags &= ~SA_RESTORER; | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 555 | #elif defined(__GLIBC__) | 
 | 556 |   // Our host compiler doesn't appear to define this flag for some reason. | 
 | 557 |   cur_action.sa_flags &= ~0x04000000; | 
 | 558 |   new_action.sa_flags &= ~0x04000000; | 
| Christopher Ferris | 3cdbfdc | 2014-11-08 15:57:11 -0800 | [diff] [blame] | 559 | #endif | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 560 |   EXPECT_EQ(cur_action.sa_flags, new_action.sa_flags); | 
 | 561 | } | 
 | 562 |  | 
 | 563 | TEST(libbacktrace, thread_ignore_frames) { | 
 | 564 |   pthread_attr_t attr; | 
 | 565 |   pthread_attr_init(&attr); | 
 | 566 |   pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | 
 | 567 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 568 |   thread_t thread_data = { 0, 0, 0, nullptr }; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 569 |   pthread_t thread; | 
 | 570 |   ASSERT_TRUE(pthread_create(&thread, &attr, ThreadLevelRun, &thread_data) == 0); | 
 | 571 |  | 
 | 572 |   // Wait up to 2 seconds for the tid to be set. | 
 | 573 |   ASSERT_TRUE(WaitForNonZero(&thread_data.state, 2)); | 
 | 574 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 575 |   std::unique_ptr<Backtrace> all(Backtrace::Create(getpid(), thread_data.tid)); | 
 | 576 |   ASSERT_TRUE(all.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 577 |   ASSERT_TRUE(all->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 578 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, all->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 579 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 580 |   std::unique_ptr<Backtrace> ign1(Backtrace::Create(getpid(), thread_data.tid)); | 
 | 581 |   ASSERT_TRUE(ign1.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 582 |   ASSERT_TRUE(ign1->Unwind(1)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 583 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, ign1->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 584 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 585 |   std::unique_ptr<Backtrace> ign2(Backtrace::Create(getpid(), thread_data.tid)); | 
 | 586 |   ASSERT_TRUE(ign2.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 587 |   ASSERT_TRUE(ign2->Unwind(2)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 588 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, ign2->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 589 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 590 |   VerifyIgnoreFrames(all.get(), ign1.get(), ign2.get(), nullptr); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 591 |  | 
 | 592 |   // Tell the thread to exit its infinite loop. | 
 | 593 |   android_atomic_acquire_store(0, &thread_data.state); | 
 | 594 | } | 
 | 595 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 596 | static void* ThreadMaxRun(void* data) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 597 |   thread_t* thread = reinterpret_cast<thread_t*>(data); | 
 | 598 |  | 
 | 599 |   thread->tid = gettid(); | 
 | 600 |   EXPECT_NE(test_recursive_call(MAX_BACKTRACE_FRAMES+10, ThreadSetState, data), 0); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 601 |   return nullptr; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 602 | } | 
 | 603 |  | 
 | 604 | TEST(libbacktrace, thread_max_trace) { | 
 | 605 |   pthread_attr_t attr; | 
 | 606 |   pthread_attr_init(&attr); | 
 | 607 |   pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | 
 | 608 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 609 |   thread_t thread_data = { 0, 0, 0, nullptr }; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 610 |   pthread_t thread; | 
 | 611 |   ASSERT_TRUE(pthread_create(&thread, &attr, ThreadMaxRun, &thread_data) == 0); | 
 | 612 |  | 
 | 613 |   // Wait for the tid to be set. | 
 | 614 |   ASSERT_TRUE(WaitForNonZero(&thread_data.state, 2)); | 
 | 615 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 616 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), thread_data.tid)); | 
 | 617 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 618 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 619 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 620 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 621 |   VerifyMaxDump(backtrace.get()); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 622 |  | 
 | 623 |   // Tell the thread to exit its infinite loop. | 
 | 624 |   android_atomic_acquire_store(0, &thread_data.state); | 
 | 625 | } | 
 | 626 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 627 | static void* ThreadDump(void* data) { | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 628 |   dump_thread_t* dump = reinterpret_cast<dump_thread_t*>(data); | 
 | 629 |   while (true) { | 
 | 630 |     if (android_atomic_acquire_load(dump->now)) { | 
 | 631 |       break; | 
 | 632 |     } | 
 | 633 |   } | 
 | 634 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 635 |   // The status of the actual unwind will be checked elsewhere. | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 636 |   dump->backtrace = Backtrace::Create(getpid(), dump->thread.tid, dump->map); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 637 |   dump->backtrace->Unwind(0); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 638 |  | 
 | 639 |   android_atomic_acquire_store(1, &dump->done); | 
 | 640 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 641 |   return nullptr; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 642 | } | 
 | 643 |  | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 644 | static void MultipleThreadDumpTest(bool share_map) { | 
 | 645 |   // Dump NUM_THREADS simultaneously using the same map. | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 646 |   std::vector<thread_t> runners(NUM_THREADS); | 
 | 647 |   std::vector<dump_thread_t> dumpers(NUM_THREADS); | 
 | 648 |  | 
 | 649 |   pthread_attr_t attr; | 
 | 650 |   pthread_attr_init(&attr); | 
 | 651 |   pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | 
 | 652 |   for (size_t i = 0; i < NUM_THREADS; i++) { | 
 | 653 |     // Launch the runners, they will spin in hard loops doing nothing. | 
 | 654 |     runners[i].tid = 0; | 
 | 655 |     runners[i].state = 0; | 
 | 656 |     ASSERT_TRUE(pthread_create(&runners[i].threadId, &attr, ThreadMaxRun, &runners[i]) == 0); | 
 | 657 |   } | 
 | 658 |  | 
 | 659 |   // Wait for tids to be set. | 
 | 660 |   for (std::vector<thread_t>::iterator it = runners.begin(); it != runners.end(); ++it) { | 
| Christopher Ferris | 3cdbfdc | 2014-11-08 15:57:11 -0800 | [diff] [blame] | 661 |     ASSERT_TRUE(WaitForNonZero(&it->state, 30)); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 662 |   } | 
 | 663 |  | 
 | 664 |   // Start all of the dumpers at once, they will spin until they are signalled | 
 | 665 |   // to begin their dump run. | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 666 |   std::unique_ptr<BacktraceMap> map; | 
 | 667 |   if (share_map) { | 
 | 668 |     map.reset(BacktraceMap::Create(getpid())); | 
 | 669 |   } | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 670 |   int32_t dump_now = 0; | 
 | 671 |   for (size_t i = 0; i < NUM_THREADS; i++) { | 
 | 672 |     dumpers[i].thread.tid = runners[i].tid; | 
 | 673 |     dumpers[i].thread.state = 0; | 
 | 674 |     dumpers[i].done = 0; | 
 | 675 |     dumpers[i].now = &dump_now; | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 676 |     dumpers[i].map = map.get(); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 677 |  | 
 | 678 |     ASSERT_TRUE(pthread_create(&dumpers[i].thread.threadId, &attr, ThreadDump, &dumpers[i]) == 0); | 
 | 679 |   } | 
 | 680 |  | 
 | 681 |   // Start all of the dumpers going at once. | 
 | 682 |   android_atomic_acquire_store(1, &dump_now); | 
 | 683 |  | 
 | 684 |   for (size_t i = 0; i < NUM_THREADS; i++) { | 
| Christopher Ferris | 3cdbfdc | 2014-11-08 15:57:11 -0800 | [diff] [blame] | 685 |     ASSERT_TRUE(WaitForNonZero(&dumpers[i].done, 30)); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 686 |  | 
 | 687 |     // Tell the runner thread to exit its infinite loop. | 
 | 688 |     android_atomic_acquire_store(0, &runners[i].state); | 
 | 689 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 690 |     ASSERT_TRUE(dumpers[i].backtrace != nullptr); | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 691 |     VerifyMaxDump(dumpers[i].backtrace); | 
 | 692 |  | 
 | 693 |     delete dumpers[i].backtrace; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 694 |     dumpers[i].backtrace = nullptr; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 695 |   } | 
 | 696 | } | 
 | 697 |  | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 698 | TEST(libbacktrace, thread_multiple_dump) { | 
 | 699 |   MultipleThreadDumpTest(false); | 
 | 700 | } | 
| Christopher Ferris | a2efd3a | 2014-05-06 15:23:59 -0700 | [diff] [blame] | 701 |  | 
| Christopher Ferris | be788d8 | 2017-11-27 14:50:38 -0800 | [diff] [blame] | 702 | TEST(libbacktrace, thread_multiple_dump_same_map) { | 
 | 703 |   MultipleThreadDumpTest(true); | 
| Christopher Ferris | a2efd3a | 2014-05-06 15:23:59 -0700 | [diff] [blame] | 704 | } | 
 | 705 |  | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 706 | // This test is for UnwindMaps that should share the same map cursor when | 
 | 707 | // multiple maps are created for the current process at the same time. | 
 | 708 | TEST(libbacktrace, simultaneous_maps) { | 
 | 709 |   BacktraceMap* map1 = BacktraceMap::Create(getpid()); | 
 | 710 |   BacktraceMap* map2 = BacktraceMap::Create(getpid()); | 
 | 711 |   BacktraceMap* map3 = BacktraceMap::Create(getpid()); | 
 | 712 |  | 
 | 713 |   Backtrace* back1 = Backtrace::Create(getpid(), BACKTRACE_CURRENT_THREAD, map1); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 714 |   ASSERT_TRUE(back1 != nullptr); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 715 |   EXPECT_TRUE(back1->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 716 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, back1->GetError().error_code); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 717 |   delete back1; | 
 | 718 |   delete map1; | 
 | 719 |  | 
 | 720 |   Backtrace* back2 = Backtrace::Create(getpid(), BACKTRACE_CURRENT_THREAD, map2); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 721 |   ASSERT_TRUE(back2 != nullptr); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 722 |   EXPECT_TRUE(back2->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 723 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, back2->GetError().error_code); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 724 |   delete back2; | 
 | 725 |   delete map2; | 
 | 726 |  | 
 | 727 |   Backtrace* back3 = Backtrace::Create(getpid(), BACKTRACE_CURRENT_THREAD, map3); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 728 |   ASSERT_TRUE(back3 != nullptr); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 729 |   EXPECT_TRUE(back3->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 730 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, back3->GetError().error_code); | 
| Christopher Ferris | df29061 | 2014-01-22 19:21:07 -0800 | [diff] [blame] | 731 |   delete back3; | 
 | 732 |   delete map3; | 
 | 733 | } | 
 | 734 |  | 
| Christopher Ferris | 12385e3 | 2015-02-06 13:22:01 -0800 | [diff] [blame] | 735 | TEST(libbacktrace, fillin_erases) { | 
 | 736 |   BacktraceMap* back_map = BacktraceMap::Create(getpid()); | 
 | 737 |  | 
 | 738 |   backtrace_map_t map; | 
 | 739 |  | 
 | 740 |   map.start = 1; | 
 | 741 |   map.end = 3; | 
 | 742 |   map.flags = 1; | 
 | 743 |   map.name = "Initialized"; | 
 | 744 |   back_map->FillIn(0, &map); | 
 | 745 |   delete back_map; | 
 | 746 |  | 
 | 747 |   ASSERT_FALSE(BacktraceMap::IsValid(map)); | 
 | 748 |   ASSERT_EQ(static_cast<uintptr_t>(0), map.start); | 
 | 749 |   ASSERT_EQ(static_cast<uintptr_t>(0), map.end); | 
 | 750 |   ASSERT_EQ(0, map.flags); | 
 | 751 |   ASSERT_EQ("", map.name); | 
 | 752 | } | 
 | 753 |  | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 754 | TEST(libbacktrace, format_test) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 755 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), BACKTRACE_CURRENT_THREAD)); | 
 | 756 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 757 |  | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 758 |   backtrace_frame_data_t frame; | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 759 |   frame.num = 1; | 
 | 760 |   frame.pc = 2; | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 761 |   frame.rel_pc = 2; | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 762 |   frame.sp = 0; | 
 | 763 |   frame.stack_size = 0; | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 764 |   frame.func_offset = 0; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 765 |  | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 766 |   // Check no map set. | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 767 |   frame.num = 1; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 768 | #if defined(__LP64__) | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 769 |   EXPECT_EQ("#01 pc 0000000000000002  <unknown>", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 770 | #else | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 771 |   EXPECT_EQ("#01 pc 00000002  <unknown>", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 772 | #endif | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 773 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 774 |  | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 775 |   // Check map name empty, but exists. | 
| Christopher Ferris | da750a7 | 2015-11-30 13:36:08 -0800 | [diff] [blame] | 776 |   frame.pc = 0xb0020; | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 777 |   frame.rel_pc = 0x20; | 
| Christopher Ferris | da750a7 | 2015-11-30 13:36:08 -0800 | [diff] [blame] | 778 |   frame.map.start = 0xb0000; | 
 | 779 |   frame.map.end = 0xbffff; | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 780 |   frame.map.load_bias = 0; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 781 | #if defined(__LP64__) | 
| Christopher Ferris | da750a7 | 2015-11-30 13:36:08 -0800 | [diff] [blame] | 782 |   EXPECT_EQ("#01 pc 0000000000000020  <anonymous:00000000000b0000>", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 783 | #else | 
| Christopher Ferris | da750a7 | 2015-11-30 13:36:08 -0800 | [diff] [blame] | 784 |   EXPECT_EQ("#01 pc 00000020  <anonymous:000b0000>", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 785 | #endif | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 786 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 787 |  | 
| Christopher Ferris | da750a7 | 2015-11-30 13:36:08 -0800 | [diff] [blame] | 788 |   // Check map name begins with a [. | 
 | 789 |   frame.pc = 0xc0020; | 
 | 790 |   frame.map.start = 0xc0000; | 
 | 791 |   frame.map.end = 0xcffff; | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 792 |   frame.map.load_bias = 0; | 
| Christopher Ferris | da750a7 | 2015-11-30 13:36:08 -0800 | [diff] [blame] | 793 |   frame.map.name = "[anon:thread signal stack]"; | 
 | 794 | #if defined(__LP64__) | 
 | 795 |   EXPECT_EQ("#01 pc 0000000000000020  [anon:thread signal stack:00000000000c0000]", | 
 | 796 | #else | 
 | 797 |   EXPECT_EQ("#01 pc 00000020  [anon:thread signal stack:000c0000]", | 
 | 798 | #endif | 
 | 799 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 800 |  | 
 | 801 |   // Check relative pc is set and map name is set. | 
 | 802 |   frame.pc = 0x12345679; | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 803 |   frame.rel_pc = 0x12345678; | 
| Christopher Ferris | 12385e3 | 2015-02-06 13:22:01 -0800 | [diff] [blame] | 804 |   frame.map.name = "MapFake"; | 
 | 805 |   frame.map.start =  1; | 
 | 806 |   frame.map.end =  1; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 807 | #if defined(__LP64__) | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 808 |   EXPECT_EQ("#01 pc 0000000012345678  MapFake", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 809 | #else | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 810 |   EXPECT_EQ("#01 pc 12345678  MapFake", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 811 | #endif | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 812 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 813 |  | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 814 |   // Check func_name is set, but no func offset. | 
 | 815 |   frame.func_name = "ProcFake"; | 
 | 816 | #if defined(__LP64__) | 
 | 817 |   EXPECT_EQ("#01 pc 0000000012345678  MapFake (ProcFake)", | 
 | 818 | #else | 
 | 819 |   EXPECT_EQ("#01 pc 12345678  MapFake (ProcFake)", | 
 | 820 | #endif | 
 | 821 |             backtrace->FormatFrameData(&frame)); | 
 | 822 |  | 
 | 823 |   // Check func_name is set, and func offset is non-zero. | 
| Christopher Ferris | 20303f8 | 2014-01-10 16:33:16 -0800 | [diff] [blame] | 824 |   frame.func_offset = 645; | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 825 | #if defined(__LP64__) | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 826 |   EXPECT_EQ("#01 pc 0000000012345678  MapFake (ProcFake+645)", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 827 | #else | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 828 |   EXPECT_EQ("#01 pc 12345678  MapFake (ProcFake+645)", | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 829 | #endif | 
| Christopher Ferris | 4675682 | 2014-01-14 20:16:30 -0800 | [diff] [blame] | 830 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | 2106f4b | 2015-05-01 15:02:03 -0700 | [diff] [blame] | 831 |  | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 832 |   // Check func_name is set, func offset is non-zero, and load_bias is non-zero. | 
 | 833 |   frame.rel_pc = 0x123456dc; | 
| Christopher Ferris | 2106f4b | 2015-05-01 15:02:03 -0700 | [diff] [blame] | 834 |   frame.func_offset = 645; | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 835 |   frame.map.load_bias = 100; | 
| Christopher Ferris | 2106f4b | 2015-05-01 15:02:03 -0700 | [diff] [blame] | 836 | #if defined(__LP64__) | 
 | 837 |   EXPECT_EQ("#01 pc 00000000123456dc  MapFake (ProcFake+645)", | 
 | 838 | #else | 
 | 839 |   EXPECT_EQ("#01 pc 123456dc  MapFake (ProcFake+645)", | 
 | 840 | #endif | 
 | 841 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | e0ab232 | 2015-08-20 11:16:54 -0700 | [diff] [blame] | 842 |  | 
 | 843 |   // Check a non-zero map offset. | 
 | 844 |   frame.map.offset = 0x1000; | 
 | 845 | #if defined(__LP64__) | 
 | 846 |   EXPECT_EQ("#01 pc 00000000123456dc  MapFake (offset 0x1000) (ProcFake+645)", | 
 | 847 | #else | 
 | 848 |   EXPECT_EQ("#01 pc 123456dc  MapFake (offset 0x1000) (ProcFake+645)", | 
 | 849 | #endif | 
 | 850 |             backtrace->FormatFrameData(&frame)); | 
| Christopher Ferris | 17e91d4 | 2013-10-21 13:30:52 -0700 | [diff] [blame] | 851 | } | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 852 |  | 
 | 853 | struct map_test_t { | 
 | 854 |   uintptr_t start; | 
 | 855 |   uintptr_t end; | 
 | 856 | }; | 
 | 857 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 858 | static bool map_sort(map_test_t i, map_test_t j) { return i.start < j.start; } | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 859 |  | 
| Christopher Ferris | b7de5f5 | 2017-12-01 21:37:37 -0800 | [diff] [blame] | 860 | static std::string GetTestMapsAsString(const std::vector<map_test_t>& maps) { | 
 | 861 |   if (maps.size() == 0) { | 
 | 862 |     return "No test map entries\n"; | 
 | 863 |   } | 
 | 864 |   std::string map_txt; | 
 | 865 |   for (auto map : maps) { | 
 | 866 |     map_txt += android::base::StringPrintf("%" PRIxPTR "-%" PRIxPTR "\n", map.start, map.end); | 
 | 867 |   } | 
 | 868 |   return map_txt; | 
 | 869 | } | 
 | 870 |  | 
 | 871 | static std::string GetMapsAsString(BacktraceMap* maps) { | 
 | 872 |   if (maps->size() == 0) { | 
 | 873 |     return "No map entries\n"; | 
 | 874 |   } | 
 | 875 |   std::string map_txt; | 
 | 876 |   for (const backtrace_map_t* map : *maps) { | 
 | 877 |     map_txt += android::base::StringPrintf( | 
 | 878 |         "%" PRIxPTR "-%" PRIxPTR " flags: 0x%x offset: 0x%" PRIxPTR " load_bias: 0x%" PRIxPTR, | 
 | 879 |         map->start, map->end, map->flags, map->offset, map->load_bias); | 
 | 880 |     if (!map->name.empty()) { | 
 | 881 |       map_txt += ' ' + map->name; | 
 | 882 |     } | 
 | 883 |     map_txt += '\n'; | 
 | 884 |   } | 
 | 885 |   return map_txt; | 
 | 886 | } | 
 | 887 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 888 | static void VerifyMap(pid_t pid) { | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 889 |   char buffer[4096]; | 
 | 890 |   snprintf(buffer, sizeof(buffer), "/proc/%d/maps", pid); | 
 | 891 |  | 
 | 892 |   FILE* map_file = fopen(buffer, "r"); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 893 |   ASSERT_TRUE(map_file != nullptr); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 894 |   std::vector<map_test_t> test_maps; | 
 | 895 |   while (fgets(buffer, sizeof(buffer), map_file)) { | 
 | 896 |     map_test_t map; | 
 | 897 |     ASSERT_EQ(2, sscanf(buffer, "%" SCNxPTR "-%" SCNxPTR " ", &map.start, &map.end)); | 
 | 898 |     test_maps.push_back(map); | 
 | 899 |   } | 
 | 900 |   fclose(map_file); | 
 | 901 |   std::sort(test_maps.begin(), test_maps.end(), map_sort); | 
 | 902 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 903 |   std::unique_ptr<BacktraceMap> map(BacktraceMap::Create(pid)); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 904 |  | 
 | 905 |   // Basic test that verifies that the map is in the expected order. | 
| Christopher Ferris | b7de5f5 | 2017-12-01 21:37:37 -0800 | [diff] [blame] | 906 |   auto test_it = test_maps.begin(); | 
 | 907 |   for (auto it = map->begin(); it != map->end(); ++it) { | 
 | 908 |     ASSERT_TRUE(test_it != test_maps.end()) << "Mismatch in number of maps, expected test maps:\n" | 
 | 909 |                                             << GetTestMapsAsString(test_maps) << "Actual maps:\n" | 
 | 910 |                                             << GetMapsAsString(map.get()); | 
 | 911 |     ASSERT_EQ(test_it->start, (*it)->start) << "Mismatch in map data, expected test maps:\n" | 
 | 912 |                                             << GetTestMapsAsString(test_maps) << "Actual maps:\n" | 
 | 913 |                                             << GetMapsAsString(map.get()); | 
 | 914 |     ASSERT_EQ(test_it->end, (*it)->end) << "Mismatch maps in map data, expected test maps:\n" | 
 | 915 |                                         << GetTestMapsAsString(test_maps) << "Actual maps:\n" | 
 | 916 |                                         << GetMapsAsString(map.get()); | 
 | 917 |     // Make sure the load bias get set to a value. | 
 | 918 |     ASSERT_NE(static_cast<uint64_t>(-1), (*it)->load_bias) << "Found uninitialized load_bias\n" | 
 | 919 |                                                            << GetMapsAsString(map.get()); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 920 |     ++test_it; | 
 | 921 |   } | 
 | 922 |   ASSERT_TRUE(test_it == test_maps.end()); | 
 | 923 | } | 
 | 924 |  | 
 | 925 | TEST(libbacktrace, verify_map_remote) { | 
 | 926 |   pid_t pid; | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 927 |   CreateRemoteProcess(&pid); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 928 |  | 
 | 929 |   // The maps should match exactly since the forked process has been paused. | 
 | 930 |   VerifyMap(pid); | 
 | 931 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 932 |   FinishRemoteProcess(pid); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 933 | } | 
 | 934 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 935 | static void InitMemory(uint8_t* memory, size_t bytes) { | 
| Christopher Ferris | 944f417 | 2015-05-06 16:36:34 -0700 | [diff] [blame] | 936 |   for (size_t i = 0; i < bytes; i++) { | 
 | 937 |     memory[i] = i; | 
 | 938 |     if (memory[i] == '\0') { | 
 | 939 |       // Don't use '\0' in our data so we can verify that an overread doesn't | 
 | 940 |       // occur by using a '\0' as the character after the read data. | 
 | 941 |       memory[i] = 23; | 
 | 942 |     } | 
 | 943 |   } | 
 | 944 | } | 
 | 945 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 946 | static void* ThreadReadTest(void* data) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 947 |   thread_t* thread_data = reinterpret_cast<thread_t*>(data); | 
 | 948 |  | 
 | 949 |   thread_data->tid = gettid(); | 
 | 950 |  | 
 | 951 |   // Create two map pages. | 
 | 952 |   // Mark the second page as not-readable. | 
 | 953 |   size_t pagesize = static_cast<size_t>(sysconf(_SC_PAGE_SIZE)); | 
 | 954 |   uint8_t* memory; | 
 | 955 |   if (posix_memalign(reinterpret_cast<void**>(&memory), pagesize, 2 * pagesize) != 0) { | 
 | 956 |     return reinterpret_cast<void*>(-1); | 
 | 957 |   } | 
 | 958 |  | 
 | 959 |   if (mprotect(&memory[pagesize], pagesize, PROT_NONE) != 0) { | 
 | 960 |     return reinterpret_cast<void*>(-1); | 
 | 961 |   } | 
 | 962 |  | 
 | 963 |   // Set up a simple pattern in memory. | 
| Christopher Ferris | 944f417 | 2015-05-06 16:36:34 -0700 | [diff] [blame] | 964 |   InitMemory(memory, pagesize); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 965 |  | 
 | 966 |   thread_data->data = memory; | 
 | 967 |  | 
 | 968 |   // Tell the caller it's okay to start reading memory. | 
 | 969 |   android_atomic_acquire_store(1, &thread_data->state); | 
 | 970 |  | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 971 |   // Loop waiting for the caller to finish reading the memory. | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 972 |   while (thread_data->state) { | 
 | 973 |   } | 
 | 974 |  | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 975 |   // Re-enable read-write on the page so that we don't crash if we try | 
 | 976 |   // and access data on this page when freeing the memory. | 
 | 977 |   if (mprotect(&memory[pagesize], pagesize, PROT_READ | PROT_WRITE) != 0) { | 
 | 978 |     return reinterpret_cast<void*>(-1); | 
 | 979 |   } | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 980 |   free(memory); | 
 | 981 |  | 
 | 982 |   android_atomic_acquire_store(1, &thread_data->state); | 
 | 983 |  | 
 | 984 |   return nullptr; | 
 | 985 | } | 
 | 986 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 987 | static void RunReadTest(Backtrace* backtrace, uintptr_t read_addr) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 988 |   size_t pagesize = static_cast<size_t>(sysconf(_SC_PAGE_SIZE)); | 
 | 989 |  | 
 | 990 |   // Create a page of data to use to do quick compares. | 
 | 991 |   uint8_t* expected = new uint8_t[pagesize]; | 
| Christopher Ferris | 944f417 | 2015-05-06 16:36:34 -0700 | [diff] [blame] | 992 |   InitMemory(expected, pagesize); | 
 | 993 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 994 |   uint8_t* data = new uint8_t[2*pagesize]; | 
 | 995 |   // Verify that we can only read one page worth of data. | 
 | 996 |   size_t bytes_read = backtrace->Read(read_addr, data, 2 * pagesize); | 
 | 997 |   ASSERT_EQ(pagesize, bytes_read); | 
 | 998 |   ASSERT_TRUE(memcmp(data, expected, pagesize) == 0); | 
 | 999 |  | 
 | 1000 |   // Verify unaligned reads. | 
 | 1001 |   for (size_t i = 1; i < sizeof(word_t); i++) { | 
 | 1002 |     bytes_read = backtrace->Read(read_addr + i, data, 2 * sizeof(word_t)); | 
 | 1003 |     ASSERT_EQ(2 * sizeof(word_t), bytes_read); | 
 | 1004 |     ASSERT_TRUE(memcmp(data, &expected[i], 2 * sizeof(word_t)) == 0) | 
 | 1005 |         << "Offset at " << i << " failed"; | 
 | 1006 |   } | 
| Christopher Ferris | 944f417 | 2015-05-06 16:36:34 -0700 | [diff] [blame] | 1007 |  | 
 | 1008 |   // Verify small unaligned reads. | 
 | 1009 |   for (size_t i = 1; i < sizeof(word_t); i++) { | 
 | 1010 |     for (size_t j = 1; j < sizeof(word_t); j++) { | 
 | 1011 |       // Set one byte past what we expect to read, to guarantee we don't overread. | 
 | 1012 |       data[j] = '\0'; | 
 | 1013 |       bytes_read = backtrace->Read(read_addr + i, data, j); | 
 | 1014 |       ASSERT_EQ(j, bytes_read); | 
 | 1015 |       ASSERT_TRUE(memcmp(data, &expected[i], j) == 0) | 
 | 1016 |           << "Offset at " << i << " length " << j << " miscompared"; | 
 | 1017 |       ASSERT_EQ('\0', data[j]) | 
 | 1018 |           << "Offset at " << i << " length " << j << " wrote too much data"; | 
 | 1019 |     } | 
 | 1020 |   } | 
| Pirama Arumuga Nainar | 837eff2 | 2015-07-09 10:50:04 -0700 | [diff] [blame] | 1021 |   delete[] data; | 
 | 1022 |   delete[] expected; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1023 | } | 
 | 1024 |  | 
 | 1025 | TEST(libbacktrace, thread_read) { | 
 | 1026 |   pthread_attr_t attr; | 
 | 1027 |   pthread_attr_init(&attr); | 
 | 1028 |   pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED); | 
 | 1029 |   pthread_t thread; | 
 | 1030 |   thread_t thread_data = { 0, 0, 0, nullptr }; | 
 | 1031 |   ASSERT_TRUE(pthread_create(&thread, &attr, ThreadReadTest, &thread_data) == 0); | 
 | 1032 |  | 
 | 1033 |   ASSERT_TRUE(WaitForNonZero(&thread_data.state, 10)); | 
 | 1034 |  | 
 | 1035 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(getpid(), thread_data.tid)); | 
 | 1036 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1037 |  | 
 | 1038 |   RunReadTest(backtrace.get(), reinterpret_cast<uintptr_t>(thread_data.data)); | 
 | 1039 |  | 
 | 1040 |   android_atomic_acquire_store(0, &thread_data.state); | 
 | 1041 |  | 
 | 1042 |   ASSERT_TRUE(WaitForNonZero(&thread_data.state, 10)); | 
 | 1043 | } | 
 | 1044 |  | 
 | 1045 | volatile uintptr_t g_ready = 0; | 
 | 1046 | volatile uintptr_t g_addr = 0; | 
 | 1047 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1048 | static void ForkedReadTest() { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1049 |   // Create two map pages. | 
 | 1050 |   size_t pagesize = static_cast<size_t>(sysconf(_SC_PAGE_SIZE)); | 
 | 1051 |   uint8_t* memory; | 
 | 1052 |   if (posix_memalign(reinterpret_cast<void**>(&memory), pagesize, 2 * pagesize) != 0) { | 
 | 1053 |     perror("Failed to allocate memory\n"); | 
 | 1054 |     exit(1); | 
 | 1055 |   } | 
 | 1056 |  | 
 | 1057 |   // Mark the second page as not-readable. | 
 | 1058 |   if (mprotect(&memory[pagesize], pagesize, PROT_NONE) != 0) { | 
 | 1059 |     perror("Failed to mprotect memory\n"); | 
 | 1060 |     exit(1); | 
 | 1061 |   } | 
 | 1062 |  | 
 | 1063 |   // Set up a simple pattern in memory. | 
| Christopher Ferris | 944f417 | 2015-05-06 16:36:34 -0700 | [diff] [blame] | 1064 |   InitMemory(memory, pagesize); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1065 |  | 
 | 1066 |   g_addr = reinterpret_cast<uintptr_t>(memory); | 
 | 1067 |   g_ready = 1; | 
 | 1068 |  | 
 | 1069 |   while (1) { | 
 | 1070 |     usleep(US_PER_MSEC); | 
 | 1071 |   } | 
 | 1072 | } | 
 | 1073 |  | 
 | 1074 | TEST(libbacktrace, process_read) { | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1075 |   g_ready = 0; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1076 |   pid_t pid; | 
 | 1077 |   if ((pid = fork()) == 0) { | 
 | 1078 |     ForkedReadTest(); | 
 | 1079 |     exit(0); | 
 | 1080 |   } | 
 | 1081 |   ASSERT_NE(-1, pid); | 
 | 1082 |  | 
 | 1083 |   bool test_executed = false; | 
 | 1084 |   uint64_t start = NanoTime(); | 
 | 1085 |   while (1) { | 
 | 1086 |     if (ptrace(PTRACE_ATTACH, pid, 0, 0) == 0) { | 
 | 1087 |       WaitForStop(pid); | 
 | 1088 |  | 
 | 1089 |       std::unique_ptr<Backtrace> backtrace(Backtrace::Create(pid, pid)); | 
| Christopher Ferris | 97e00bb | 2015-04-02 14:22:31 -0700 | [diff] [blame] | 1090 |       ASSERT_TRUE(backtrace.get() != nullptr); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1091 |  | 
 | 1092 |       uintptr_t read_addr; | 
 | 1093 |       size_t bytes_read = backtrace->Read(reinterpret_cast<uintptr_t>(&g_ready), | 
 | 1094 |                                           reinterpret_cast<uint8_t*>(&read_addr), | 
 | 1095 |                                           sizeof(uintptr_t)); | 
 | 1096 |       ASSERT_EQ(sizeof(uintptr_t), bytes_read); | 
 | 1097 |       if (read_addr) { | 
 | 1098 |         // The forked process is ready to be read. | 
 | 1099 |         bytes_read = backtrace->Read(reinterpret_cast<uintptr_t>(&g_addr), | 
 | 1100 |                                      reinterpret_cast<uint8_t*>(&read_addr), | 
 | 1101 |                                      sizeof(uintptr_t)); | 
 | 1102 |         ASSERT_EQ(sizeof(uintptr_t), bytes_read); | 
 | 1103 |  | 
 | 1104 |         RunReadTest(backtrace.get(), read_addr); | 
 | 1105 |  | 
 | 1106 |         test_executed = true; | 
 | 1107 |         break; | 
 | 1108 |       } | 
 | 1109 |       ASSERT_TRUE(ptrace(PTRACE_DETACH, pid, 0, 0) == 0); | 
 | 1110 |     } | 
 | 1111 |     if ((NanoTime() - start) > 5 * NS_PER_SEC) { | 
 | 1112 |       break; | 
 | 1113 |     } | 
 | 1114 |     usleep(US_PER_MSEC); | 
 | 1115 |   } | 
 | 1116 |   kill(pid, SIGKILL); | 
 | 1117 |   ASSERT_EQ(waitpid(pid, nullptr, 0), pid); | 
 | 1118 |  | 
 | 1119 |   ASSERT_TRUE(test_executed); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1120 | } | 
 | 1121 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1122 | static void VerifyFunctionsFound(const std::vector<std::string>& found_functions) { | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1123 |   // We expect to find these functions in libbacktrace_test. If we don't | 
 | 1124 |   // find them, that's a bug in the memory read handling code in libunwind. | 
 | 1125 |   std::list<std::string> expected_functions; | 
 | 1126 |   expected_functions.push_back("test_recursive_call"); | 
 | 1127 |   expected_functions.push_back("test_level_one"); | 
 | 1128 |   expected_functions.push_back("test_level_two"); | 
 | 1129 |   expected_functions.push_back("test_level_three"); | 
 | 1130 |   expected_functions.push_back("test_level_four"); | 
 | 1131 |   for (const auto& found_function : found_functions) { | 
 | 1132 |     for (const auto& expected_function : expected_functions) { | 
 | 1133 |       if (found_function == expected_function) { | 
 | 1134 |         expected_functions.remove(found_function); | 
 | 1135 |         break; | 
 | 1136 |       } | 
 | 1137 |     } | 
 | 1138 |   } | 
 | 1139 |   ASSERT_TRUE(expected_functions.empty()) << "Not all functions found in shared library."; | 
 | 1140 | } | 
 | 1141 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1142 | static const char* CopySharedLibrary() { | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1143 | #if defined(__LP64__) | 
 | 1144 |   const char* lib_name = "lib64"; | 
 | 1145 | #else | 
 | 1146 |   const char* lib_name = "lib"; | 
 | 1147 | #endif | 
 | 1148 |  | 
 | 1149 | #if defined(__BIONIC__) | 
 | 1150 |   const char* tmp_so_name = "/data/local/tmp/libbacktrace_test.so"; | 
 | 1151 |   std::string cp_cmd = android::base::StringPrintf("cp /system/%s/libbacktrace_test.so %s", | 
 | 1152 |                                                    lib_name, tmp_so_name); | 
 | 1153 | #else | 
 | 1154 |   const char* tmp_so_name = "/tmp/libbacktrace_test.so"; | 
 | 1155 |   if (getenv("ANDROID_HOST_OUT") == NULL) { | 
 | 1156 |     fprintf(stderr, "ANDROID_HOST_OUT not set, make sure you run lunch."); | 
 | 1157 |     return nullptr; | 
 | 1158 |   } | 
 | 1159 |   std::string cp_cmd = android::base::StringPrintf("cp %s/%s/libbacktrace_test.so %s", | 
 | 1160 |                                                    getenv("ANDROID_HOST_OUT"), lib_name, | 
 | 1161 |                                                    tmp_so_name); | 
 | 1162 | #endif | 
 | 1163 |  | 
 | 1164 |   // Copy the shared so to a tempory directory. | 
 | 1165 |   system(cp_cmd.c_str()); | 
 | 1166 |  | 
 | 1167 |   return tmp_so_name; | 
 | 1168 | } | 
 | 1169 |  | 
 | 1170 | TEST(libbacktrace, check_unreadable_elf_local) { | 
 | 1171 |   const char* tmp_so_name = CopySharedLibrary(); | 
 | 1172 |   ASSERT_TRUE(tmp_so_name != nullptr); | 
 | 1173 |  | 
 | 1174 |   struct stat buf; | 
 | 1175 |   ASSERT_TRUE(stat(tmp_so_name, &buf) != -1); | 
 | 1176 |   uintptr_t map_size = buf.st_size; | 
 | 1177 |  | 
 | 1178 |   int fd = open(tmp_so_name, O_RDONLY); | 
 | 1179 |   ASSERT_TRUE(fd != -1); | 
 | 1180 |  | 
| Christopher Ferris | 61c48ac | 2016-01-15 16:08:58 -0800 | [diff] [blame] | 1181 |   void* map = mmap(NULL, map_size, PROT_READ | PROT_EXEC, MAP_PRIVATE, fd, 0); | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1182 |   ASSERT_TRUE(map != MAP_FAILED); | 
 | 1183 |   close(fd); | 
 | 1184 |   ASSERT_TRUE(unlink(tmp_so_name) != -1); | 
 | 1185 |  | 
 | 1186 |   std::vector<std::string> found_functions; | 
 | 1187 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(BACKTRACE_CURRENT_PROCESS, | 
 | 1188 |                                                          BACKTRACE_CURRENT_THREAD)); | 
 | 1189 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1190 |  | 
 | 1191 |   // Needed before GetFunctionName will work. | 
 | 1192 |   backtrace->Unwind(0); | 
 | 1193 |  | 
 | 1194 |   // Loop through the entire map, and get every function we can find. | 
 | 1195 |   map_size += reinterpret_cast<uintptr_t>(map); | 
 | 1196 |   std::string last_func; | 
 | 1197 |   for (uintptr_t read_addr = reinterpret_cast<uintptr_t>(map); | 
 | 1198 |        read_addr < map_size; read_addr += 4) { | 
 | 1199 |     uintptr_t offset; | 
 | 1200 |     std::string func_name = backtrace->GetFunctionName(read_addr, &offset); | 
 | 1201 |     if (!func_name.empty() && last_func != func_name) { | 
 | 1202 |       found_functions.push_back(func_name); | 
 | 1203 |     } | 
 | 1204 |     last_func = func_name; | 
 | 1205 |   } | 
 | 1206 |  | 
 | 1207 |   ASSERT_TRUE(munmap(map, map_size - reinterpret_cast<uintptr_t>(map)) == 0); | 
 | 1208 |  | 
 | 1209 |   VerifyFunctionsFound(found_functions); | 
 | 1210 | } | 
 | 1211 |  | 
 | 1212 | TEST(libbacktrace, check_unreadable_elf_remote) { | 
 | 1213 |   const char* tmp_so_name = CopySharedLibrary(); | 
 | 1214 |   ASSERT_TRUE(tmp_so_name != nullptr); | 
 | 1215 |  | 
 | 1216 |   g_ready = 0; | 
 | 1217 |  | 
 | 1218 |   struct stat buf; | 
 | 1219 |   ASSERT_TRUE(stat(tmp_so_name, &buf) != -1); | 
 | 1220 |   uintptr_t map_size = buf.st_size; | 
 | 1221 |  | 
 | 1222 |   pid_t pid; | 
 | 1223 |   if ((pid = fork()) == 0) { | 
 | 1224 |     int fd = open(tmp_so_name, O_RDONLY); | 
 | 1225 |     if (fd == -1) { | 
 | 1226 |       fprintf(stderr, "Failed to open file %s: %s\n", tmp_so_name, strerror(errno)); | 
 | 1227 |       unlink(tmp_so_name); | 
 | 1228 |       exit(0); | 
 | 1229 |     } | 
 | 1230 |  | 
| Christopher Ferris | 61c48ac | 2016-01-15 16:08:58 -0800 | [diff] [blame] | 1231 |     void* map = mmap(NULL, map_size, PROT_READ | PROT_EXEC, MAP_PRIVATE, fd, 0); | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1232 |     if (map == MAP_FAILED) { | 
 | 1233 |       fprintf(stderr, "Failed to map in memory: %s\n", strerror(errno)); | 
 | 1234 |       unlink(tmp_so_name); | 
 | 1235 |       exit(0); | 
 | 1236 |     } | 
 | 1237 |     close(fd); | 
 | 1238 |     if (unlink(tmp_so_name) == -1) { | 
 | 1239 |       fprintf(stderr, "Failed to unlink: %s\n", strerror(errno)); | 
 | 1240 |       exit(0); | 
 | 1241 |     } | 
 | 1242 |  | 
 | 1243 |     g_addr = reinterpret_cast<uintptr_t>(map); | 
 | 1244 |     g_ready = 1; | 
 | 1245 |     while (true) { | 
 | 1246 |       usleep(US_PER_MSEC); | 
 | 1247 |     } | 
 | 1248 |     exit(0); | 
 | 1249 |   } | 
 | 1250 |   ASSERT_TRUE(pid > 0); | 
 | 1251 |  | 
 | 1252 |   std::vector<std::string> found_functions; | 
 | 1253 |   uint64_t start = NanoTime(); | 
 | 1254 |   while (true) { | 
 | 1255 |     ASSERT_TRUE(ptrace(PTRACE_ATTACH, pid, 0, 0) == 0); | 
 | 1256 |  | 
 | 1257 |     // Wait for the process to get to a stopping point. | 
 | 1258 |     WaitForStop(pid); | 
 | 1259 |  | 
 | 1260 |     std::unique_ptr<Backtrace> backtrace(Backtrace::Create(pid, BACKTRACE_CURRENT_THREAD)); | 
 | 1261 |     ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1262 |  | 
 | 1263 |     uintptr_t read_addr; | 
 | 1264 |     ASSERT_EQ(sizeof(uintptr_t), backtrace->Read(reinterpret_cast<uintptr_t>(&g_ready), reinterpret_cast<uint8_t*>(&read_addr), sizeof(uintptr_t))); | 
 | 1265 |     if (read_addr) { | 
 | 1266 |       ASSERT_EQ(sizeof(uintptr_t), backtrace->Read(reinterpret_cast<uintptr_t>(&g_addr), reinterpret_cast<uint8_t*>(&read_addr), sizeof(uintptr_t))); | 
 | 1267 |  | 
 | 1268 |       // Needed before GetFunctionName will work. | 
 | 1269 |       backtrace->Unwind(0); | 
 | 1270 |  | 
 | 1271 |       // Loop through the entire map, and get every function we can find. | 
 | 1272 |       map_size += read_addr; | 
 | 1273 |       std::string last_func; | 
 | 1274 |       for (; read_addr < map_size; read_addr += 4) { | 
 | 1275 |         uintptr_t offset; | 
 | 1276 |         std::string func_name = backtrace->GetFunctionName(read_addr, &offset); | 
 | 1277 |         if (!func_name.empty() && last_func != func_name) { | 
 | 1278 |           found_functions.push_back(func_name); | 
 | 1279 |         } | 
 | 1280 |         last_func = func_name; | 
 | 1281 |       } | 
 | 1282 |       break; | 
 | 1283 |     } | 
 | 1284 |     ASSERT_TRUE(ptrace(PTRACE_DETACH, pid, 0, 0) == 0); | 
 | 1285 |  | 
 | 1286 |     if ((NanoTime() - start) > 5 * NS_PER_SEC) { | 
 | 1287 |       break; | 
 | 1288 |     } | 
 | 1289 |     usleep(US_PER_MSEC); | 
 | 1290 |   } | 
 | 1291 |  | 
 | 1292 |   kill(pid, SIGKILL); | 
 | 1293 |   ASSERT_EQ(waitpid(pid, nullptr, 0), pid); | 
 | 1294 |  | 
 | 1295 |   VerifyFunctionsFound(found_functions); | 
 | 1296 | } | 
 | 1297 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1298 | static bool FindFuncFrameInBacktrace(Backtrace* backtrace, uintptr_t test_func, size_t* frame_num) { | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1299 |   backtrace_map_t map; | 
 | 1300 |   backtrace->FillInMap(test_func, &map); | 
 | 1301 |   if (!BacktraceMap::IsValid(map)) { | 
 | 1302 |     return false; | 
 | 1303 |   } | 
 | 1304 |  | 
 | 1305 |   // Loop through the frames, and find the one that is in the map. | 
 | 1306 |   *frame_num = 0; | 
 | 1307 |   for (Backtrace::const_iterator it = backtrace->begin(); it != backtrace->end(); ++it) { | 
 | 1308 |     if (BacktraceMap::IsValid(it->map) && map.start == it->map.start && | 
 | 1309 |         it->pc >= test_func) { | 
 | 1310 |       *frame_num = it->num; | 
 | 1311 |       return true; | 
 | 1312 |     } | 
 | 1313 |   } | 
 | 1314 |   return false; | 
 | 1315 | } | 
 | 1316 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1317 | static void VerifyUnreadableElfFrame(Backtrace* backtrace, uintptr_t test_func, size_t frame_num) { | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1318 |   ASSERT_LT(backtrace->NumFrames(), static_cast<size_t>(MAX_BACKTRACE_FRAMES)) | 
 | 1319 |     << DumpFrames(backtrace); | 
 | 1320 |  | 
 | 1321 |   ASSERT_TRUE(frame_num != 0) << DumpFrames(backtrace); | 
 | 1322 |   // Make sure that there is at least one more frame above the test func call. | 
 | 1323 |   ASSERT_LT(frame_num, backtrace->NumFrames()) << DumpFrames(backtrace); | 
 | 1324 |  | 
 | 1325 |   uintptr_t diff = backtrace->GetFrame(frame_num)->pc - test_func; | 
 | 1326 |   ASSERT_LT(diff, 200U) << DumpFrames(backtrace); | 
 | 1327 | } | 
 | 1328 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1329 | static void VerifyUnreadableElfBacktrace(uintptr_t test_func) { | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1330 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(BACKTRACE_CURRENT_PROCESS, | 
 | 1331 |                                                          BACKTRACE_CURRENT_THREAD)); | 
 | 1332 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1333 |   ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 1334 |   ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1335 |  | 
 | 1336 |   size_t frame_num; | 
 | 1337 |   ASSERT_TRUE(FindFuncFrameInBacktrace(backtrace.get(), test_func, &frame_num)); | 
 | 1338 |  | 
 | 1339 |   VerifyUnreadableElfFrame(backtrace.get(), test_func, frame_num); | 
 | 1340 | } | 
 | 1341 |  | 
 | 1342 | typedef int (*test_func_t)(int, int, int, int, void (*)(uintptr_t), uintptr_t); | 
 | 1343 |  | 
 | 1344 | TEST(libbacktrace, unwind_through_unreadable_elf_local) { | 
 | 1345 |   const char* tmp_so_name = CopySharedLibrary(); | 
 | 1346 |   ASSERT_TRUE(tmp_so_name != nullptr); | 
 | 1347 |   void* lib_handle = dlopen(tmp_so_name, RTLD_NOW); | 
 | 1348 |   ASSERT_TRUE(lib_handle != nullptr); | 
 | 1349 |   ASSERT_TRUE(unlink(tmp_so_name) != -1); | 
 | 1350 |  | 
 | 1351 |   test_func_t test_func; | 
 | 1352 |   test_func = reinterpret_cast<test_func_t>(dlsym(lib_handle, "test_level_one")); | 
 | 1353 |   ASSERT_TRUE(test_func != nullptr); | 
 | 1354 |  | 
 | 1355 |   ASSERT_NE(test_func(1, 2, 3, 4, VerifyUnreadableElfBacktrace, | 
 | 1356 |                       reinterpret_cast<uintptr_t>(test_func)), 0); | 
 | 1357 |  | 
 | 1358 |   ASSERT_TRUE(dlclose(lib_handle) == 0); | 
 | 1359 | } | 
 | 1360 |  | 
 | 1361 | TEST(libbacktrace, unwind_through_unreadable_elf_remote) { | 
 | 1362 |   const char* tmp_so_name = CopySharedLibrary(); | 
 | 1363 |   ASSERT_TRUE(tmp_so_name != nullptr); | 
 | 1364 |   void* lib_handle = dlopen(tmp_so_name, RTLD_NOW); | 
 | 1365 |   ASSERT_TRUE(lib_handle != nullptr); | 
 | 1366 |   ASSERT_TRUE(unlink(tmp_so_name) != -1); | 
 | 1367 |  | 
 | 1368 |   test_func_t test_func; | 
 | 1369 |   test_func = reinterpret_cast<test_func_t>(dlsym(lib_handle, "test_level_one")); | 
 | 1370 |   ASSERT_TRUE(test_func != nullptr); | 
 | 1371 |  | 
 | 1372 |   pid_t pid; | 
 | 1373 |   if ((pid = fork()) == 0) { | 
 | 1374 |     test_func(1, 2, 3, 4, 0, 0); | 
 | 1375 |     exit(0); | 
 | 1376 |   } | 
 | 1377 |   ASSERT_TRUE(pid > 0); | 
 | 1378 |   ASSERT_TRUE(dlclose(lib_handle) == 0); | 
 | 1379 |  | 
 | 1380 |   uint64_t start = NanoTime(); | 
 | 1381 |   bool done = false; | 
 | 1382 |   while (!done) { | 
 | 1383 |     ASSERT_TRUE(ptrace(PTRACE_ATTACH, pid, 0, 0) == 0); | 
 | 1384 |  | 
 | 1385 |     // Wait for the process to get to a stopping point. | 
 | 1386 |     WaitForStop(pid); | 
 | 1387 |  | 
 | 1388 |     std::unique_ptr<Backtrace> backtrace(Backtrace::Create(pid, BACKTRACE_CURRENT_THREAD)); | 
 | 1389 |     ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1390 |     ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 1391 |     ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | 67aba68 | 2015-05-08 15:44:46 -0700 | [diff] [blame] | 1392 |  | 
 | 1393 |     size_t frame_num; | 
 | 1394 |     if (FindFuncFrameInBacktrace(backtrace.get(), | 
 | 1395 |                                  reinterpret_cast<uintptr_t>(test_func), &frame_num)) { | 
 | 1396 |  | 
 | 1397 |       VerifyUnreadableElfFrame(backtrace.get(), reinterpret_cast<uintptr_t>(test_func), frame_num); | 
 | 1398 |       done = true; | 
 | 1399 |     } | 
 | 1400 |  | 
 | 1401 |     ASSERT_TRUE(ptrace(PTRACE_DETACH, pid, 0, 0) == 0); | 
 | 1402 |  | 
 | 1403 |     if ((NanoTime() - start) > 5 * NS_PER_SEC) { | 
 | 1404 |       break; | 
 | 1405 |     } | 
 | 1406 |     usleep(US_PER_MSEC); | 
 | 1407 |   } | 
 | 1408 |  | 
 | 1409 |   kill(pid, SIGKILL); | 
 | 1410 |   ASSERT_EQ(waitpid(pid, nullptr, 0), pid); | 
 | 1411 |  | 
 | 1412 |   ASSERT_TRUE(done) << "Test function never found in unwind."; | 
 | 1413 | } | 
 | 1414 |  | 
| Christopher Ferris | 206a3b9 | 2016-03-09 14:35:54 -0800 | [diff] [blame] | 1415 | TEST(libbacktrace, unwind_thread_doesnt_exist) { | 
 | 1416 |   std::unique_ptr<Backtrace> backtrace( | 
 | 1417 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, 99999999)); | 
 | 1418 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1419 |   ASSERT_FALSE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 1420 |   ASSERT_EQ(BACKTRACE_UNWIND_ERROR_THREAD_DOESNT_EXIST, backtrace->GetError().error_code); | 
| Christopher Ferris | 206a3b9 | 2016-03-09 14:35:54 -0800 | [diff] [blame] | 1421 | } | 
 | 1422 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1423 | TEST(libbacktrace, local_get_function_name_before_unwind) { | 
 | 1424 |   std::unique_ptr<Backtrace> backtrace( | 
 | 1425 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
 | 1426 |   ASSERT_TRUE(backtrace.get() != nullptr); | 
 | 1427 |  | 
 | 1428 |   // Verify that trying to get a function name before doing an unwind works. | 
 | 1429 |   uintptr_t cur_func_offset = reinterpret_cast<uintptr_t>(&test_level_one) + 1; | 
 | 1430 |   size_t offset; | 
 | 1431 |   ASSERT_NE(std::string(""), backtrace->GetFunctionName(cur_func_offset, &offset)); | 
 | 1432 | } | 
 | 1433 |  | 
 | 1434 | TEST(libbacktrace, remote_get_function_name_before_unwind) { | 
 | 1435 |   pid_t pid; | 
 | 1436 |   CreateRemoteProcess(&pid); | 
 | 1437 |  | 
 | 1438 |   // Now create an unwind object. | 
 | 1439 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(pid, pid)); | 
 | 1440 |  | 
 | 1441 |   // Verify that trying to get a function name before doing an unwind works. | 
 | 1442 |   uintptr_t cur_func_offset = reinterpret_cast<uintptr_t>(&test_level_one) + 1; | 
 | 1443 |   size_t offset; | 
 | 1444 |   ASSERT_NE(std::string(""), backtrace->GetFunctionName(cur_func_offset, &offset)); | 
 | 1445 |  | 
 | 1446 |   FinishRemoteProcess(pid); | 
 | 1447 | } | 
 | 1448 |  | 
| Christopher Ferris | f5e568e | 2017-03-22 13:18:31 -0700 | [diff] [blame] | 1449 | static void SetUcontextSp(uintptr_t sp, ucontext_t* ucontext) { | 
 | 1450 | #if defined(__arm__) | 
 | 1451 |   ucontext->uc_mcontext.arm_sp = sp; | 
 | 1452 | #elif defined(__aarch64__) | 
 | 1453 |   ucontext->uc_mcontext.sp = sp; | 
 | 1454 | #elif defined(__i386__) | 
 | 1455 |   ucontext->uc_mcontext.gregs[REG_ESP] = sp; | 
 | 1456 | #elif defined(__x86_64__) | 
 | 1457 |   ucontext->uc_mcontext.gregs[REG_RSP] = sp; | 
 | 1458 | #else | 
 | 1459 |   UNUSED(sp); | 
 | 1460 |   UNUSED(ucontext); | 
 | 1461 |   ASSERT_TRUE(false) << "Unsupported architecture"; | 
 | 1462 | #endif | 
 | 1463 | } | 
 | 1464 |  | 
 | 1465 | static void SetUcontextPc(uintptr_t pc, ucontext_t* ucontext) { | 
 | 1466 | #if defined(__arm__) | 
 | 1467 |   ucontext->uc_mcontext.arm_pc = pc; | 
 | 1468 | #elif defined(__aarch64__) | 
 | 1469 |   ucontext->uc_mcontext.pc = pc; | 
 | 1470 | #elif defined(__i386__) | 
 | 1471 |   ucontext->uc_mcontext.gregs[REG_EIP] = pc; | 
 | 1472 | #elif defined(__x86_64__) | 
 | 1473 |   ucontext->uc_mcontext.gregs[REG_RIP] = pc; | 
 | 1474 | #else | 
 | 1475 |   UNUSED(pc); | 
 | 1476 |   UNUSED(ucontext); | 
 | 1477 |   ASSERT_TRUE(false) << "Unsupported architecture"; | 
 | 1478 | #endif | 
 | 1479 | } | 
 | 1480 |  | 
 | 1481 | static void SetUcontextLr(uintptr_t lr, ucontext_t* ucontext) { | 
 | 1482 | #if defined(__arm__) | 
 | 1483 |   ucontext->uc_mcontext.arm_lr = lr; | 
 | 1484 | #elif defined(__aarch64__) | 
 | 1485 |   ucontext->uc_mcontext.regs[30] = lr; | 
 | 1486 | #elif defined(__i386__) | 
 | 1487 |   // The lr is on the stack. | 
 | 1488 |   ASSERT_TRUE(lr != 0); | 
 | 1489 |   ASSERT_TRUE(ucontext != nullptr); | 
 | 1490 | #elif defined(__x86_64__) | 
 | 1491 |   // The lr is on the stack. | 
 | 1492 |   ASSERT_TRUE(lr != 0); | 
 | 1493 |   ASSERT_TRUE(ucontext != nullptr); | 
 | 1494 | #else | 
 | 1495 |   UNUSED(lr); | 
 | 1496 |   UNUSED(ucontext); | 
 | 1497 |   ASSERT_TRUE(false) << "Unsupported architecture"; | 
 | 1498 | #endif | 
 | 1499 | } | 
 | 1500 |  | 
 | 1501 | static constexpr size_t DEVICE_MAP_SIZE = 1024; | 
 | 1502 |  | 
 | 1503 | static void SetupDeviceMap(void** device_map) { | 
 | 1504 |   // Make sure that anything in a device map will result in fails | 
 | 1505 |   // to read. | 
 | 1506 |   android::base::unique_fd device_fd(open("/dev/zero", O_RDONLY | O_CLOEXEC)); | 
 | 1507 |  | 
 | 1508 |   *device_map = mmap(nullptr, 1024, PROT_READ, MAP_PRIVATE, device_fd, 0); | 
 | 1509 |   ASSERT_TRUE(*device_map != MAP_FAILED); | 
 | 1510 |  | 
 | 1511 |   // Make sure the map is readable. | 
 | 1512 |   ASSERT_EQ(0, reinterpret_cast<int*>(*device_map)[0]); | 
 | 1513 | } | 
 | 1514 |  | 
 | 1515 | static void UnwindFromDevice(Backtrace* backtrace, void* device_map) { | 
 | 1516 |   uintptr_t device_map_uint = reinterpret_cast<uintptr_t>(device_map); | 
 | 1517 |  | 
 | 1518 |   backtrace_map_t map; | 
 | 1519 |   backtrace->FillInMap(device_map_uint, &map); | 
 | 1520 |   // Verify the flag is set. | 
 | 1521 |   ASSERT_EQ(PROT_DEVICE_MAP, map.flags & PROT_DEVICE_MAP); | 
 | 1522 |  | 
 | 1523 |   // Quick sanity checks. | 
 | 1524 |   size_t offset; | 
 | 1525 |   ASSERT_EQ(std::string(""), backtrace->GetFunctionName(device_map_uint, &offset)); | 
 | 1526 |   ASSERT_EQ(std::string(""), backtrace->GetFunctionName(device_map_uint, &offset, &map)); | 
 | 1527 |   ASSERT_EQ(std::string(""), backtrace->GetFunctionName(0, &offset)); | 
 | 1528 |  | 
 | 1529 |   uintptr_t cur_func_offset = reinterpret_cast<uintptr_t>(&test_level_one) + 1; | 
 | 1530 |   // Now verify the device map flag actually causes the function name to be empty. | 
 | 1531 |   backtrace->FillInMap(cur_func_offset, &map); | 
 | 1532 |   ASSERT_TRUE((map.flags & PROT_DEVICE_MAP) == 0); | 
 | 1533 |   ASSERT_NE(std::string(""), backtrace->GetFunctionName(cur_func_offset, &offset, &map)); | 
 | 1534 |   map.flags |= PROT_DEVICE_MAP; | 
 | 1535 |   ASSERT_EQ(std::string(""), backtrace->GetFunctionName(cur_func_offset, &offset, &map)); | 
 | 1536 |  | 
 | 1537 |   ucontext_t ucontext; | 
 | 1538 |  | 
 | 1539 |   // Create a context that has the pc in the device map, but the sp | 
 | 1540 |   // in a non-device map. | 
 | 1541 |   memset(&ucontext, 0, sizeof(ucontext)); | 
 | 1542 |   SetUcontextSp(reinterpret_cast<uintptr_t>(&ucontext), &ucontext); | 
 | 1543 |   SetUcontextPc(device_map_uint, &ucontext); | 
 | 1544 |   SetUcontextLr(cur_func_offset, &ucontext); | 
 | 1545 |  | 
 | 1546 |   ASSERT_TRUE(backtrace->Unwind(0, &ucontext)); | 
 | 1547 |  | 
 | 1548 |   // The buffer should only be a single element. | 
 | 1549 |   ASSERT_EQ(1U, backtrace->NumFrames()); | 
 | 1550 |   const backtrace_frame_data_t* frame = backtrace->GetFrame(0); | 
 | 1551 |   ASSERT_EQ(device_map_uint, frame->pc); | 
 | 1552 |   ASSERT_EQ(reinterpret_cast<uintptr_t>(&ucontext), frame->sp); | 
 | 1553 |  | 
 | 1554 |   // Check what happens when skipping the first frame. | 
 | 1555 |   ASSERT_TRUE(backtrace->Unwind(1, &ucontext)); | 
 | 1556 |   ASSERT_EQ(0U, backtrace->NumFrames()); | 
 | 1557 |  | 
 | 1558 |   // Create a context that has the sp in the device map, but the pc | 
 | 1559 |   // in a non-device map. | 
 | 1560 |   memset(&ucontext, 0, sizeof(ucontext)); | 
 | 1561 |   SetUcontextSp(device_map_uint, &ucontext); | 
 | 1562 |   SetUcontextPc(cur_func_offset, &ucontext); | 
 | 1563 |   SetUcontextLr(cur_func_offset, &ucontext); | 
 | 1564 |  | 
 | 1565 |   ASSERT_TRUE(backtrace->Unwind(0, &ucontext)); | 
 | 1566 |  | 
 | 1567 |   // The buffer should only be a single element. | 
 | 1568 |   ASSERT_EQ(1U, backtrace->NumFrames()); | 
 | 1569 |   frame = backtrace->GetFrame(0); | 
 | 1570 |   ASSERT_EQ(cur_func_offset, frame->pc); | 
 | 1571 |   ASSERT_EQ(device_map_uint, frame->sp); | 
 | 1572 |  | 
 | 1573 |   // Check what happens when skipping the first frame. | 
 | 1574 |   ASSERT_TRUE(backtrace->Unwind(1, &ucontext)); | 
 | 1575 |   ASSERT_EQ(0U, backtrace->NumFrames()); | 
 | 1576 | } | 
 | 1577 |  | 
 | 1578 | TEST(libbacktrace, unwind_disallow_device_map_local) { | 
 | 1579 |   void* device_map; | 
 | 1580 |   SetupDeviceMap(&device_map); | 
 | 1581 |  | 
 | 1582 |   // Now create an unwind object. | 
 | 1583 |   std::unique_ptr<Backtrace> backtrace( | 
 | 1584 |       Backtrace::Create(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD)); | 
 | 1585 |   ASSERT_TRUE(backtrace); | 
 | 1586 |  | 
 | 1587 |   UnwindFromDevice(backtrace.get(), device_map); | 
 | 1588 |  | 
 | 1589 |   munmap(device_map, DEVICE_MAP_SIZE); | 
 | 1590 | } | 
 | 1591 |  | 
| Christopher Ferris | 086baf9 | 2017-10-17 14:12:52 -0700 | [diff] [blame] | 1592 | TEST(libbacktrace, unwind_disallow_device_map_remote) { | 
| Christopher Ferris | f5e568e | 2017-03-22 13:18:31 -0700 | [diff] [blame] | 1593 |   void* device_map; | 
 | 1594 |   SetupDeviceMap(&device_map); | 
 | 1595 |  | 
 | 1596 |   // Fork a process to do a remote backtrace. | 
 | 1597 |   pid_t pid; | 
 | 1598 |   CreateRemoteProcess(&pid); | 
 | 1599 |  | 
 | 1600 |   // Now create an unwind object. | 
| Christopher Ferris | 086baf9 | 2017-10-17 14:12:52 -0700 | [diff] [blame] | 1601 |   std::unique_ptr<Backtrace> backtrace(Backtrace::Create(pid, pid)); | 
| Christopher Ferris | f5e568e | 2017-03-22 13:18:31 -0700 | [diff] [blame] | 1602 |  | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 1603 |   UnwindFromDevice(backtrace.get(), device_map); | 
| Christopher Ferris | f5e568e | 2017-03-22 13:18:31 -0700 | [diff] [blame] | 1604 |  | 
 | 1605 |   FinishRemoteProcess(pid); | 
 | 1606 |  | 
 | 1607 |   munmap(device_map, DEVICE_MAP_SIZE); | 
 | 1608 | } | 
 | 1609 |  | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 1610 | class ScopedSignalHandler { | 
 | 1611 |  public: | 
 | 1612 |   ScopedSignalHandler(int signal_number, void (*handler)(int)) : signal_number_(signal_number) { | 
 | 1613 |     memset(&action_, 0, sizeof(action_)); | 
 | 1614 |     action_.sa_handler = handler; | 
 | 1615 |     sigaction(signal_number_, &action_, &old_action_); | 
 | 1616 |   } | 
 | 1617 |  | 
 | 1618 |   ScopedSignalHandler(int signal_number, void (*action)(int, siginfo_t*, void*)) | 
 | 1619 |       : signal_number_(signal_number) { | 
 | 1620 |     memset(&action_, 0, sizeof(action_)); | 
 | 1621 |     action_.sa_flags = SA_SIGINFO; | 
 | 1622 |     action_.sa_sigaction = action; | 
 | 1623 |     sigaction(signal_number_, &action_, &old_action_); | 
 | 1624 |   } | 
 | 1625 |  | 
 | 1626 |   ~ScopedSignalHandler() { sigaction(signal_number_, &old_action_, nullptr); } | 
 | 1627 |  | 
 | 1628 |  private: | 
 | 1629 |   struct sigaction action_; | 
 | 1630 |   struct sigaction old_action_; | 
 | 1631 |   const int signal_number_; | 
 | 1632 | }; | 
 | 1633 |  | 
 | 1634 | static void SetValueAndLoop(void* data) { | 
 | 1635 |   volatile int* value = reinterpret_cast<volatile int*>(data); | 
 | 1636 |  | 
 | 1637 |   *value = 1; | 
 | 1638 |   for (volatile int i = 0;; i++) | 
 | 1639 |     ; | 
 | 1640 | } | 
 | 1641 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 1642 | static void UnwindThroughSignal(bool use_action, create_func_t create_func, | 
 | 1643 |                                 map_create_func_t map_create_func) { | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 1644 |   volatile int value = 0; | 
 | 1645 |   pid_t pid; | 
 | 1646 |   if ((pid = fork()) == 0) { | 
 | 1647 |     if (use_action) { | 
 | 1648 |       ScopedSignalHandler ssh(SIGUSR1, test_signal_action); | 
 | 1649 |  | 
 | 1650 |       test_level_one(1, 2, 3, 4, SetValueAndLoop, const_cast<int*>(&value)); | 
 | 1651 |     } else { | 
 | 1652 |       ScopedSignalHandler ssh(SIGUSR1, test_signal_handler); | 
 | 1653 |  | 
 | 1654 |       test_level_one(1, 2, 3, 4, SetValueAndLoop, const_cast<int*>(&value)); | 
 | 1655 |     } | 
 | 1656 |   } | 
 | 1657 |   ASSERT_NE(-1, pid); | 
 | 1658 |  | 
 | 1659 |   int read_value = 0; | 
 | 1660 |   uint64_t start = NanoTime(); | 
 | 1661 |   while (read_value == 0) { | 
 | 1662 |     usleep(1000); | 
 | 1663 |  | 
 | 1664 |     // Loop until the remote function gets into the final function. | 
 | 1665 |     ASSERT_TRUE(ptrace(PTRACE_ATTACH, pid, 0, 0) == 0); | 
 | 1666 |  | 
 | 1667 |     WaitForStop(pid); | 
 | 1668 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 1669 |     std::unique_ptr<BacktraceMap> map(map_create_func(pid, false)); | 
 | 1670 |     std::unique_ptr<Backtrace> backtrace(create_func(pid, pid, map.get())); | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 1671 |  | 
 | 1672 |     size_t bytes_read = backtrace->Read(reinterpret_cast<uintptr_t>(const_cast<int*>(&value)), | 
 | 1673 |                                         reinterpret_cast<uint8_t*>(&read_value), sizeof(read_value)); | 
 | 1674 |     ASSERT_EQ(sizeof(read_value), bytes_read); | 
 | 1675 |  | 
 | 1676 |     ASSERT_TRUE(ptrace(PTRACE_DETACH, pid, 0, 0) == 0); | 
 | 1677 |  | 
 | 1678 |     ASSERT_TRUE(NanoTime() - start < 5 * NS_PER_SEC) | 
 | 1679 |         << "Remote process did not execute far enough in 5 seconds."; | 
 | 1680 |   } | 
 | 1681 |  | 
 | 1682 |   // Now need to send a signal to the remote process. | 
 | 1683 |   kill(pid, SIGUSR1); | 
 | 1684 |  | 
 | 1685 |   // Wait for the process to get to the signal handler loop. | 
 | 1686 |   Backtrace::const_iterator frame_iter; | 
 | 1687 |   start = NanoTime(); | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 1688 |   std::unique_ptr<BacktraceMap> map; | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 1689 |   std::unique_ptr<Backtrace> backtrace; | 
 | 1690 |   while (true) { | 
 | 1691 |     usleep(1000); | 
 | 1692 |  | 
 | 1693 |     ASSERT_TRUE(ptrace(PTRACE_ATTACH, pid, 0, 0) == 0); | 
 | 1694 |  | 
 | 1695 |     WaitForStop(pid); | 
 | 1696 |  | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 1697 |     map.reset(map_create_func(pid, false)); | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 1698 |     ASSERT_TRUE(map.get() != nullptr); | 
| Christopher Ferris | b9de87f | 2017-09-20 13:37:24 -0700 | [diff] [blame] | 1699 |     backtrace.reset(create_func(pid, pid, map.get())); | 
| Christopher Ferris | 5ea2c1f | 2017-03-23 14:55:01 -0700 | [diff] [blame] | 1700 |     ASSERT_TRUE(backtrace->Unwind(0)); | 
 | 1701 |     bool found = false; | 
 | 1702 |     for (frame_iter = backtrace->begin(); frame_iter != backtrace->end(); ++frame_iter) { | 
 | 1703 |       if (frame_iter->func_name == "test_loop_forever") { | 
 | 1704 |         ++frame_iter; | 
 | 1705 |         found = true; | 
 | 1706 |         break; | 
 | 1707 |       } | 
 | 1708 |     } | 
 | 1709 |     if (found) { | 
 | 1710 |       break; | 
 | 1711 |     } | 
 | 1712 |  | 
 | 1713 |     ASSERT_TRUE(ptrace(PTRACE_DETACH, pid, 0, 0) == 0); | 
 | 1714 |  | 
 | 1715 |     ASSERT_TRUE(NanoTime() - start < 5 * NS_PER_SEC) | 
 | 1716 |         << "Remote process did not get in signal handler in 5 seconds." << std::endl | 
 | 1717 |         << DumpFrames(backtrace.get()); | 
 | 1718 |   } | 
 | 1719 |  | 
 | 1720 |   std::vector<std::string> names; | 
 | 1721 |   // Loop through the frames, and save the function names. | 
 | 1722 |   size_t frame = 0; | 
 | 1723 |   for (; frame_iter != backtrace->end(); ++frame_iter) { | 
 | 1724 |     if (frame_iter->func_name == "test_level_four") { | 
 | 1725 |       frame = names.size() + 1; | 
 | 1726 |     } | 
 | 1727 |     names.push_back(frame_iter->func_name); | 
 | 1728 |   } | 
 | 1729 |   ASSERT_NE(0U, frame) << "Unable to find test_level_four in backtrace" << std::endl | 
 | 1730 |                        << DumpFrames(backtrace.get()); | 
 | 1731 |  | 
 | 1732 |   // The expected order of the frames: | 
 | 1733 |   //   test_loop_forever | 
 | 1734 |   //   test_signal_handler|test_signal_action | 
 | 1735 |   //   <OPTIONAL_FRAME> May or may not exist. | 
 | 1736 |   //   SetValueAndLoop (but the function name might be empty) | 
 | 1737 |   //   test_level_four | 
 | 1738 |   //   test_level_three | 
 | 1739 |   //   test_level_two | 
 | 1740 |   //   test_level_one | 
 | 1741 |   ASSERT_LE(frame + 2, names.size()) << DumpFrames(backtrace.get()); | 
 | 1742 |   ASSERT_LE(2U, frame) << DumpFrames(backtrace.get()); | 
 | 1743 |   if (use_action) { | 
 | 1744 |     ASSERT_EQ("test_signal_action", names[0]) << DumpFrames(backtrace.get()); | 
 | 1745 |   } else { | 
 | 1746 |     ASSERT_EQ("test_signal_handler", names[0]) << DumpFrames(backtrace.get()); | 
 | 1747 |   } | 
 | 1748 |   ASSERT_EQ("test_level_three", names[frame]) << DumpFrames(backtrace.get()); | 
 | 1749 |   ASSERT_EQ("test_level_two", names[frame + 1]) << DumpFrames(backtrace.get()); | 
 | 1750 |   ASSERT_EQ("test_level_one", names[frame + 2]) << DumpFrames(backtrace.get()); | 
 | 1751 |  | 
 | 1752 |   FinishRemoteProcess(pid); | 
 | 1753 | } | 
 | 1754 |  | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 1755 | TEST(libbacktrace, unwind_remote_through_signal_using_handler) { | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 1756 |   UnwindThroughSignal(false, Backtrace::Create, BacktraceMap::Create); | 
 | 1757 | } | 
 | 1758 |  | 
| Christopher Ferris | 96722b0 | 2017-07-19 14:20:46 -0700 | [diff] [blame] | 1759 | TEST(libbacktrace, unwind_remote_through_signal_using_action) { | 
| Christopher Ferris | 458cc66 | 2017-08-28 16:31:18 -0700 | [diff] [blame] | 1760 |   UnwindThroughSignal(true, Backtrace::Create, BacktraceMap::Create); | 
 | 1761 | } | 
 | 1762 |  | 
| Josh Gao | cd546c1 | 2017-10-25 15:21:45 -0700 | [diff] [blame] | 1763 | static void TestFrameSkipNumbering(create_func_t create_func, map_create_func_t map_create_func) { | 
 | 1764 |   std::unique_ptr<BacktraceMap> map(map_create_func(getpid(), false)); | 
 | 1765 |   std::unique_ptr<Backtrace> backtrace(create_func(getpid(), gettid(), map.get())); | 
 | 1766 |   backtrace->Unwind(1); | 
 | 1767 |   ASSERT_NE(0U, backtrace->NumFrames()); | 
 | 1768 |   ASSERT_EQ(0U, backtrace->GetFrame(0)->num); | 
 | 1769 | } | 
 | 1770 |  | 
 | 1771 | TEST(libbacktrace, unwind_frame_skip_numbering) { | 
 | 1772 |   TestFrameSkipNumbering(Backtrace::Create, BacktraceMap::Create); | 
 | 1773 | } | 
 | 1774 |  | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1775 | #if defined(ENABLE_PSS_TESTS) | 
 | 1776 | #include "GetPss.h" | 
 | 1777 |  | 
| Chih-Hung Hsieh | 67867db | 2016-05-18 15:53:15 -0700 | [diff] [blame] | 1778 | #define MAX_LEAK_BYTES (32*1024UL) | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1779 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1780 | static void CheckForLeak(pid_t pid, pid_t tid) { | 
| Christopher Ferris | 086baf9 | 2017-10-17 14:12:52 -0700 | [diff] [blame] | 1781 |   std::unique_ptr<BacktraceMap> map(BacktraceMap::Create(pid)); | 
 | 1782 |  | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1783 |   // Do a few runs to get the PSS stable. | 
 | 1784 |   for (size_t i = 0; i < 100; i++) { | 
| Christopher Ferris | 086baf9 | 2017-10-17 14:12:52 -0700 | [diff] [blame] | 1785 |     Backtrace* backtrace = Backtrace::Create(pid, tid, map.get()); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1786 |     ASSERT_TRUE(backtrace != nullptr); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1787 |     ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 1788 |     ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1789 |     delete backtrace; | 
 | 1790 |   } | 
 | 1791 |   size_t stable_pss = GetPssBytes(); | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 1792 |   ASSERT_TRUE(stable_pss != 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1793 |  | 
 | 1794 |   // Loop enough that even a small leak should be detectable. | 
 | 1795 |   for (size_t i = 0; i < 4096; i++) { | 
| Christopher Ferris | 086baf9 | 2017-10-17 14:12:52 -0700 | [diff] [blame] | 1796 |     Backtrace* backtrace = Backtrace::Create(pid, tid, map.get()); | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1797 |     ASSERT_TRUE(backtrace != nullptr); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1798 |     ASSERT_TRUE(backtrace->Unwind(0)); | 
| Yabin Cui | f880828 | 2017-12-12 18:04:10 -0800 | [diff] [blame] | 1799 |     ASSERT_EQ(BACKTRACE_UNWIND_NO_ERROR, backtrace->GetError().error_code); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1800 |     delete backtrace; | 
 | 1801 |   } | 
 | 1802 |   size_t new_pss = GetPssBytes(); | 
| Christopher Ferris | 2c43cff | 2015-03-26 19:18:36 -0700 | [diff] [blame] | 1803 |   ASSERT_TRUE(new_pss != 0); | 
| Christopher Ferris | 5ccdfa6 | 2016-03-07 19:18:31 -0800 | [diff] [blame] | 1804 |   if (new_pss > stable_pss) { | 
 | 1805 |     ASSERT_LE(new_pss - stable_pss, MAX_LEAK_BYTES); | 
 | 1806 |   } | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1807 | } | 
 | 1808 |  | 
 | 1809 | TEST(libbacktrace, check_for_leak_local) { | 
 | 1810 |   CheckForLeak(BACKTRACE_CURRENT_PROCESS, BACKTRACE_CURRENT_THREAD); | 
 | 1811 | } | 
 | 1812 |  | 
 | 1813 | TEST(libbacktrace, check_for_leak_local_thread) { | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1814 |   thread_t thread_data = { 0, 0, 0, nullptr }; | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1815 |   pthread_t thread; | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1816 |   ASSERT_TRUE(pthread_create(&thread, nullptr, ThreadLevelRun, &thread_data) == 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1817 |  | 
 | 1818 |   // Wait up to 2 seconds for the tid to be set. | 
 | 1819 |   ASSERT_TRUE(WaitForNonZero(&thread_data.state, 2)); | 
 | 1820 |  | 
 | 1821 |   CheckForLeak(BACKTRACE_CURRENT_PROCESS, thread_data.tid); | 
 | 1822 |  | 
 | 1823 |   // Tell the thread to exit its infinite loop. | 
 | 1824 |   android_atomic_acquire_store(0, &thread_data.state); | 
 | 1825 |  | 
| Christopher Ferris | 2b4a63f | 2015-03-17 14:42:03 -0700 | [diff] [blame] | 1826 |   ASSERT_TRUE(pthread_join(thread, nullptr) == 0); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1827 | } | 
 | 1828 |  | 
 | 1829 | TEST(libbacktrace, check_for_leak_remote) { | 
 | 1830 |   pid_t pid; | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1831 |   CreateRemoteProcess(&pid); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1832 |  | 
 | 1833 |   CheckForLeak(pid, BACKTRACE_CURRENT_THREAD); | 
 | 1834 |  | 
| Christopher Ferris | 82f3bbd | 2017-03-14 15:22:26 -0700 | [diff] [blame] | 1835 |   FinishRemoteProcess(pid); | 
| Christopher Ferris | e296091 | 2014-03-07 19:42:19 -0800 | [diff] [blame] | 1836 | } | 
 | 1837 | #endif |