1b9c1b51eSKate Stone //===-- DynamicLoaderDarwinKernel.cpp -----------------------------*- C++
2b9c1b51eSKate Stone //-*-===//
3d4bfbc9aSGreg Clayton //
4d4bfbc9aSGreg Clayton //                     The LLVM Compiler Infrastructure
5d4bfbc9aSGreg Clayton //
6d4bfbc9aSGreg Clayton // This file is distributed under the University of Illinois Open Source
7d4bfbc9aSGreg Clayton // License. See LICENSE.TXT for details.
8d4bfbc9aSGreg Clayton //
9d4bfbc9aSGreg Clayton //===----------------------------------------------------------------------===//
10d4bfbc9aSGreg Clayton 
1146d005dbSJim Ingham #include "lldb/Utility/SafeMachO.h"
12eadcca97SGreg Clayton 
13b9c1b51eSKate Stone #include "Plugins/Platform/MacOSX/PlatformDarwinKernel.h"
14d4bfbc9aSGreg Clayton #include "lldb/Breakpoint/StoppointCallbackContext.h"
15d4bfbc9aSGreg Clayton #include "lldb/Core/DataBuffer.h"
16d4bfbc9aSGreg Clayton #include "lldb/Core/DataBufferHeap.h"
17d4bfbc9aSGreg Clayton #include "lldb/Core/Debugger.h"
18d4bfbc9aSGreg Clayton #include "lldb/Core/Log.h"
19d4bfbc9aSGreg Clayton #include "lldb/Core/Module.h"
201f746071SGreg Clayton #include "lldb/Core/ModuleSpec.h"
21d4bfbc9aSGreg Clayton #include "lldb/Core/PluginManager.h"
221f746071SGreg Clayton #include "lldb/Core/Section.h"
23d4bfbc9aSGreg Clayton #include "lldb/Core/State.h"
2444d93782SGreg Clayton #include "lldb/Core/StreamFile.h"
2568b3607fSJason Molenda #include "lldb/Host/Symbols.h"
2641204d09SIlia K #include "lldb/Interpreter/OptionValueProperties.h"
27d4bfbc9aSGreg Clayton #include "lldb/Symbol/ObjectFile.h"
28d4bfbc9aSGreg Clayton #include "lldb/Target/RegisterContext.h"
29b57e4a1bSJason Molenda #include "lldb/Target/StackFrame.h"
30d4bfbc9aSGreg Clayton #include "lldb/Target/Target.h"
31d4bfbc9aSGreg Clayton #include "lldb/Target/Thread.h"
32d4bfbc9aSGreg Clayton #include "lldb/Target/ThreadPlanRunToAddress.h"
33d4bfbc9aSGreg Clayton 
34944b828aSGreg Clayton #include "DynamicLoaderDarwinKernel.h"
35d4bfbc9aSGreg Clayton 
36d4bfbc9aSGreg Clayton //#define ENABLE_DEBUG_PRINTF // COMMENT THIS LINE OUT PRIOR TO CHECKIN
37d4bfbc9aSGreg Clayton #ifdef ENABLE_DEBUG_PRINTF
38d4bfbc9aSGreg Clayton #include <stdio.h>
39d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...) printf(fmt, ##__VA_ARGS__)
40d4bfbc9aSGreg Clayton #else
41d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...)
42d4bfbc9aSGreg Clayton #endif
43d4bfbc9aSGreg Clayton 
44d4bfbc9aSGreg Clayton using namespace lldb;
45d4bfbc9aSGreg Clayton using namespace lldb_private;
46d4bfbc9aSGreg Clayton 
476ba6d3d1SJason Molenda // Progressively greater amounts of scanning we will allow
48b9c1b51eSKate Stone // For some targets very early in startup, we can't do any random reads of
49b9c1b51eSKate Stone // memory or we can crash the device
506ba6d3d1SJason Molenda // so a setting is needed that can completely disable the KASLR scans.
516ba6d3d1SJason Molenda 
52b9c1b51eSKate Stone enum KASLRScanType {
536ba6d3d1SJason Molenda   eKASLRScanNone = 0,        // No reading into the inferior at all
54b9c1b51eSKate Stone   eKASLRScanLowgloAddresses, // Check one word of memory for a possible kernel
55b9c1b51eSKate Stone                              // addr, then see if a kernel is there
56b9c1b51eSKate Stone   eKASLRScanNearPC, // Scan backwards from the current $pc looking for kernel;
57b9c1b51eSKate Stone                     // checking at 96 locations total
58b9c1b51eSKate Stone   eKASLRScanExhaustiveScan // Scan through the entire possible kernel address
59b9c1b51eSKate Stone                            // range looking for a kernel
606ba6d3d1SJason Molenda };
616ba6d3d1SJason Molenda 
62b9c1b51eSKate Stone OptionEnumValueElement g_kaslr_kernel_scan_enum_values[] = {
63b9c1b51eSKate Stone     {eKASLRScanNone, "none",
64b9c1b51eSKate Stone      "Do not read memory looking for a Darwin kernel when attaching."},
65b9c1b51eSKate Stone     {eKASLRScanLowgloAddresses, "basic", "Check for the Darwin kernel's load "
66b9c1b51eSKate Stone                                          "addr in the lowglo page "
67b9c1b51eSKate Stone                                          "(boot-args=debug) only."},
68b9c1b51eSKate Stone     {eKASLRScanNearPC, "fast-scan", "Scan near the pc value on attach to find "
69b9c1b51eSKate Stone                                     "the Darwin kernel's load address."},
70b9c1b51eSKate Stone     {eKASLRScanExhaustiveScan, "exhaustive-scan",
71b9c1b51eSKate Stone      "Scan through the entire potential address range of Darwin kernel (only "
72b9c1b51eSKate Stone      "on 32-bit targets)."},
73b9c1b51eSKate Stone     {0, NULL, NULL}};
746ba6d3d1SJason Molenda 
75b9c1b51eSKate Stone static PropertyDefinition g_properties[] = {
76b9c1b51eSKate Stone     {"load-kexts", OptionValue::eTypeBoolean, true, true, NULL, NULL,
77b9c1b51eSKate Stone      "Automatically loads kext images when attaching to a kernel."},
78b9c1b51eSKate Stone     {"scan-type", OptionValue::eTypeEnum, true, eKASLRScanNearPC, NULL,
79b9c1b51eSKate Stone      g_kaslr_kernel_scan_enum_values, "Control how many reads lldb will make "
80b9c1b51eSKate Stone                                       "while searching for a Darwin kernel on "
81b9c1b51eSKate Stone                                       "attach."},
82b9c1b51eSKate Stone     {NULL, OptionValue::eTypeInvalid, false, 0, NULL, NULL, NULL}};
83e8cd0c98SGreg Clayton 
84b9c1b51eSKate Stone enum { ePropertyLoadKexts, ePropertyScanType };
85e8cd0c98SGreg Clayton 
86b9c1b51eSKate Stone class DynamicLoaderDarwinKernelProperties : public Properties {
87e8cd0c98SGreg Clayton public:
88b9c1b51eSKate Stone   static ConstString &GetSettingName() {
89468ea4ebSGreg Clayton     static ConstString g_setting_name("darwin-kernel");
90e8cd0c98SGreg Clayton     return g_setting_name;
91e8cd0c98SGreg Clayton   }
92e8cd0c98SGreg Clayton 
93b9c1b51eSKate Stone   DynamicLoaderDarwinKernelProperties() : Properties() {
94e8cd0c98SGreg Clayton     m_collection_sp.reset(new OptionValueProperties(GetSettingName()));
95e8cd0c98SGreg Clayton     m_collection_sp->Initialize(g_properties);
96e8cd0c98SGreg Clayton   }
97e8cd0c98SGreg Clayton 
98b9c1b51eSKate Stone   virtual ~DynamicLoaderDarwinKernelProperties() {}
99e8cd0c98SGreg Clayton 
100b9c1b51eSKate Stone   bool GetLoadKexts() const {
10166763eedSGreg Clayton     const uint32_t idx = ePropertyLoadKexts;
102b9c1b51eSKate Stone     return m_collection_sp->GetPropertyAtIndexAsBoolean(
103b9c1b51eSKate Stone         NULL, idx, g_properties[idx].default_uint_value != 0);
104e8cd0c98SGreg Clayton   }
105e8cd0c98SGreg Clayton 
106b9c1b51eSKate Stone   KASLRScanType GetScanType() const {
1076ba6d3d1SJason Molenda     const uint32_t idx = ePropertyScanType;
108b9c1b51eSKate Stone     return (KASLRScanType)m_collection_sp->GetPropertyAtIndexAsEnumeration(
109b9c1b51eSKate Stone         NULL, idx, g_properties[idx].default_uint_value);
1106ba6d3d1SJason Molenda   }
111e8cd0c98SGreg Clayton };
112e8cd0c98SGreg Clayton 
113b9c1b51eSKate Stone typedef std::shared_ptr<DynamicLoaderDarwinKernelProperties>
114b9c1b51eSKate Stone     DynamicLoaderDarwinKernelPropertiesSP;
115e8cd0c98SGreg Clayton 
116b9c1b51eSKate Stone static const DynamicLoaderDarwinKernelPropertiesSP &GetGlobalProperties() {
117e8cd0c98SGreg Clayton   static DynamicLoaderDarwinKernelPropertiesSP g_settings_sp;
118e8cd0c98SGreg Clayton   if (!g_settings_sp)
119e8cd0c98SGreg Clayton     g_settings_sp.reset(new DynamicLoaderDarwinKernelProperties());
120e8cd0c98SGreg Clayton   return g_settings_sp;
121e8cd0c98SGreg Clayton }
122e8cd0c98SGreg Clayton 
123d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
124d4bfbc9aSGreg Clayton // Create an instance of this class. This function is filled into
125d4bfbc9aSGreg Clayton // the plugin info class that gets handed out by the plugin factory and
126d4bfbc9aSGreg Clayton // allows the lldb to instantiate an instance of this class.
127d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
128b9c1b51eSKate Stone DynamicLoader *DynamicLoaderDarwinKernel::CreateInstance(Process *process,
129b9c1b51eSKate Stone                                                          bool force) {
130b9c1b51eSKate Stone   if (!force) {
1316ba6d3d1SJason Molenda     // If the user provided an executable binary and it is not a kernel,
1326ba6d3d1SJason Molenda     // this plugin should not create an instance.
133d4bfbc9aSGreg Clayton     Module *exe_module = process->GetTarget().GetExecutableModulePointer();
134b9c1b51eSKate Stone     if (exe_module) {
135d4bfbc9aSGreg Clayton       ObjectFile *object_file = exe_module->GetObjectFile();
136b9c1b51eSKate Stone       if (object_file) {
137b9c1b51eSKate Stone         if (object_file->GetStrata() != ObjectFile::eStrataKernel) {
1386ba6d3d1SJason Molenda           return NULL;
1396ba6d3d1SJason Molenda         }
140d4bfbc9aSGreg Clayton       }
141d4bfbc9aSGreg Clayton     }
142d4bfbc9aSGreg Clayton 
1436ba6d3d1SJason Molenda     // If the target's architecture does not look like an Apple environment,
1446ba6d3d1SJason Molenda     // this plugin should not create an instance.
145b9c1b51eSKate Stone     const llvm::Triple &triple_ref =
146b9c1b51eSKate Stone         process->GetTarget().GetArchitecture().GetTriple();
147b9c1b51eSKate Stone     switch (triple_ref.getOS()) {
14870512317SGreg Clayton     case llvm::Triple::Darwin:
14970512317SGreg Clayton     case llvm::Triple::MacOSX:
15070512317SGreg Clayton     case llvm::Triple::IOS:
151a814f704SJason Molenda     case llvm::Triple::TvOS:
152a814f704SJason Molenda     case llvm::Triple::WatchOS:
153b9c1b51eSKate Stone       if (triple_ref.getVendor() != llvm::Triple::Apple) {
1546ba6d3d1SJason Molenda         return NULL;
1556ba6d3d1SJason Molenda       }
1566ba6d3d1SJason Molenda       break;
157b9c1b51eSKate Stone     // If we have triple like armv7-unknown-unknown, we should try looking for a
158b9c1b51eSKate Stone     // Darwin kernel.
1596ba6d3d1SJason Molenda     case llvm::Triple::UnknownOS:
16070512317SGreg Clayton       break;
16170512317SGreg Clayton     default:
1626ba6d3d1SJason Molenda       return NULL;
16370512317SGreg Clayton       break;
16470512317SGreg Clayton     }
165d4bfbc9aSGreg Clayton   }
1666ba6d3d1SJason Molenda 
167b9c1b51eSKate Stone   // At this point if there is an ExecutableModule, it is a kernel and the
168b9c1b51eSKate Stone   // Target is some variant of an Apple system.
169b9c1b51eSKate Stone   // If the Process hasn't provided the kernel load address, we need to look
170b9c1b51eSKate Stone   // around in memory to find it.
1716ba6d3d1SJason Molenda 
172c44bcec6SGreg Clayton   const addr_t kernel_load_address = SearchForDarwinKernel(process);
173b9c1b51eSKate Stone   if (CheckForKernelImageAtAddress(kernel_load_address, process).IsValid()) {
17490ff7911SEwan Crawford     process->SetCanRunCode(false);
175503d0181SJason Molenda     return new DynamicLoaderDarwinKernel(process, kernel_load_address);
176503d0181SJason Molenda   }
177503d0181SJason Molenda   return NULL;
178503d0181SJason Molenda }
179503d0181SJason Molenda 
180503d0181SJason Molenda lldb::addr_t
181b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForDarwinKernel(Process *process) {
1826ba6d3d1SJason Molenda   addr_t kernel_load_address = process->GetImageInfoAddress();
183b9c1b51eSKate Stone   if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1846ba6d3d1SJason Molenda     kernel_load_address = SearchForKernelAtSameLoadAddr(process);
185b9c1b51eSKate Stone     if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1866ba6d3d1SJason Molenda       kernel_load_address = SearchForKernelWithDebugHints(process);
187b9c1b51eSKate Stone       if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1886ba6d3d1SJason Molenda         kernel_load_address = SearchForKernelNearPC(process);
189b9c1b51eSKate Stone         if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1906ba6d3d1SJason Molenda           kernel_load_address = SearchForKernelViaExhaustiveSearch(process);
1916ba6d3d1SJason Molenda         }
1926ba6d3d1SJason Molenda       }
1936ba6d3d1SJason Molenda     }
194d4bfbc9aSGreg Clayton   }
195503d0181SJason Molenda   return kernel_load_address;
196d4bfbc9aSGreg Clayton }
197d4bfbc9aSGreg Clayton 
198d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1996ba6d3d1SJason Molenda // Check if the kernel binary is loaded in memory without a slide.
2006ba6d3d1SJason Molenda // First verify that the ExecutableModule is a kernel before we proceed.
201b9c1b51eSKate Stone // Returns the address of the kernel if one was found, else
202b9c1b51eSKate Stone // LLDB_INVALID_ADDRESS.
2036ba6d3d1SJason Molenda //----------------------------------------------------------------------
2046ba6d3d1SJason Molenda lldb::addr_t
205b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelAtSameLoadAddr(Process *process) {
2066ba6d3d1SJason Molenda   Module *exe_module = process->GetTarget().GetExecutableModulePointer();
2076ba6d3d1SJason Molenda   if (exe_module == NULL)
2086ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2096ba6d3d1SJason Molenda 
2106ba6d3d1SJason Molenda   ObjectFile *exe_objfile = exe_module->GetObjectFile();
2116ba6d3d1SJason Molenda   if (exe_objfile == NULL)
2126ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2136ba6d3d1SJason Molenda 
214b9c1b51eSKate Stone   if (exe_objfile->GetType() != ObjectFile::eTypeExecutable ||
215b9c1b51eSKate Stone       exe_objfile->GetStrata() != ObjectFile::eStrataKernel)
2166ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2176ba6d3d1SJason Molenda 
2186ba6d3d1SJason Molenda   if (!exe_objfile->GetHeaderAddress().IsValid())
2196ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2206ba6d3d1SJason Molenda 
221b9c1b51eSKate Stone   if (CheckForKernelImageAtAddress(
222b9c1b51eSKate Stone           exe_objfile->GetHeaderAddress().GetFileAddress(), process) ==
223b9c1b51eSKate Stone       exe_module->GetUUID())
2246ba6d3d1SJason Molenda     return exe_objfile->GetHeaderAddress().GetFileAddress();
2256ba6d3d1SJason Molenda 
2266ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
2276ba6d3d1SJason Molenda }
2286ba6d3d1SJason Molenda 
2296ba6d3d1SJason Molenda //----------------------------------------------------------------------
230b9c1b51eSKate Stone // If the debug flag is included in the boot-args nvram setting, the kernel's
231b9c1b51eSKate Stone // load address
2326ba6d3d1SJason Molenda // will be noted in the lowglo page at a fixed address
233b9c1b51eSKate Stone // Returns the address of the kernel if one was found, else
234b9c1b51eSKate Stone // LLDB_INVALID_ADDRESS.
2356ba6d3d1SJason Molenda //----------------------------------------------------------------------
2366ba6d3d1SJason Molenda lldb::addr_t
237b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelWithDebugHints(Process *process) {
2386ba6d3d1SJason Molenda   if (GetGlobalProperties()->GetScanType() == eKASLRScanNone)
2396ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2406ba6d3d1SJason Molenda 
2416ba6d3d1SJason Molenda   Error read_err;
2426ba6d3d1SJason Molenda   addr_t addr = LLDB_INVALID_ADDRESS;
243b9c1b51eSKate Stone   addr_t kernel_addresses_64[] = {
244b9c1b51eSKate Stone       0xfffffff000004010ULL, // newest arm64 devices
24557086ed7SJason Molenda       0xffffff8000004010ULL, // 2014-2015-ish arm64 devices
24657086ed7SJason Molenda       0xffffff8000002010ULL, // oldest arm64 devices
247880988afSJason Molenda       LLDB_INVALID_ADDRESS};
248b9c1b51eSKate Stone   addr_t kernel_addresses_32[] = {0xffff0110, LLDB_INVALID_ADDRESS};
249b9c1b51eSKate Stone   for (size_t i = 0; kernel_addresses_64[i] != LLDB_INVALID_ADDRESS; i++) {
250b9c1b51eSKate Stone     addr = process->ReadUnsignedIntegerFromMemory(
251b9c1b51eSKate Stone         kernel_addresses_64[i], 8, LLDB_INVALID_ADDRESS, read_err);
252b9c1b51eSKate Stone     if (CheckForKernelImageAtAddress(addr, process).IsValid()) {
253ec504238SJason Molenda       return addr;
254ec504238SJason Molenda     }
2556ba6d3d1SJason Molenda   }
256880988afSJason Molenda 
257b9c1b51eSKate Stone   for (size_t i = 0; kernel_addresses_32[i] != LLDB_INVALID_ADDRESS; i++) {
258b9c1b51eSKate Stone     addr = process->ReadUnsignedIntegerFromMemory(
259b9c1b51eSKate Stone         kernel_addresses_32[i], 4, LLDB_INVALID_ADDRESS, read_err);
260b9c1b51eSKate Stone     if (CheckForKernelImageAtAddress(addr, process).IsValid()) {
2616ba6d3d1SJason Molenda       return addr;
2626ba6d3d1SJason Molenda     }
2637dd29390SJason Molenda   }
264880988afSJason Molenda 
2656ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
2666ba6d3d1SJason Molenda }
2676ba6d3d1SJason Molenda 
2686ba6d3d1SJason Molenda //----------------------------------------------------------------------
2696ba6d3d1SJason Molenda // If the kernel is currently executing when lldb attaches, and we don't have
2706ba6d3d1SJason Molenda // a better way of finding the kernel's load address, try searching backwards
2716ba6d3d1SJason Molenda // from the current pc value looking for the kernel's Mach header in memory.
272b9c1b51eSKate Stone // Returns the address of the kernel if one was found, else
273b9c1b51eSKate Stone // LLDB_INVALID_ADDRESS.
2746ba6d3d1SJason Molenda //----------------------------------------------------------------------
2756ba6d3d1SJason Molenda lldb::addr_t
276b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelNearPC(Process *process) {
277b9c1b51eSKate Stone   if (GetGlobalProperties()->GetScanType() == eKASLRScanNone ||
278b9c1b51eSKate Stone       GetGlobalProperties()->GetScanType() == eKASLRScanLowgloAddresses) {
2796ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2806ba6d3d1SJason Molenda   }
2816ba6d3d1SJason Molenda 
2826ba6d3d1SJason Molenda   ThreadSP thread = process->GetThreadList().GetSelectedThread();
2836ba6d3d1SJason Molenda   if (thread.get() == NULL)
2846ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2856ba6d3d1SJason Molenda   addr_t pc = thread->GetRegisterContext()->GetPC(LLDB_INVALID_ADDRESS);
2866ba6d3d1SJason Molenda 
2876ba6d3d1SJason Molenda   if (pc == LLDB_INVALID_ADDRESS)
2886ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2896ba6d3d1SJason Molenda 
290b9c1b51eSKate Stone   // The kernel will load at at one megabyte boundary (0x100000), or at that
291b9c1b51eSKate Stone   // boundary plus
292b9c1b51eSKate Stone   // an offset of one page (0x1000) or two, or four (0x4000), depending on the
293b9c1b51eSKate Stone   // device.
2946ba6d3d1SJason Molenda 
295b9c1b51eSKate Stone   // Round the current pc down to the nearest one megabyte boundary - the place
296b9c1b51eSKate Stone   // where we will start searching.
2976ba6d3d1SJason Molenda   addr_t addr = pc & ~0xfffff;
2986ba6d3d1SJason Molenda 
299b9c1b51eSKate Stone   // Search backwards 32 megabytes, looking for the start of the kernel at each
300b9c1b51eSKate Stone   // one-megabyte boundary.
301b9c1b51eSKate Stone   for (int i = 0; i < 32; i++, addr -= 0x100000) {
3026ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr, process).IsValid())
3036ba6d3d1SJason Molenda       return addr;
3046ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x1000, process).IsValid())
3056ba6d3d1SJason Molenda       return addr + 0x1000;
3066ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x2000, process).IsValid())
3076ba6d3d1SJason Molenda       return addr + 0x2000;
308a02869deSJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x4000, process).IsValid())
309a02869deSJason Molenda       return addr + 0x4000;
3106ba6d3d1SJason Molenda   }
3116ba6d3d1SJason Molenda 
3126ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
3136ba6d3d1SJason Molenda }
3146ba6d3d1SJason Molenda 
3156ba6d3d1SJason Molenda //----------------------------------------------------------------------
3166ba6d3d1SJason Molenda // Scan through the valid address range for a kernel binary.
3176ba6d3d1SJason Molenda // This is uselessly slow in 64-bit environments so we don't even try it.
3186ba6d3d1SJason Molenda // This scan is not enabled by default even for 32-bit targets.
319b9c1b51eSKate Stone // Returns the address of the kernel if one was found, else
320b9c1b51eSKate Stone // LLDB_INVALID_ADDRESS.
3216ba6d3d1SJason Molenda //----------------------------------------------------------------------
322b9c1b51eSKate Stone lldb::addr_t DynamicLoaderDarwinKernel::SearchForKernelViaExhaustiveSearch(
323b9c1b51eSKate Stone     Process *process) {
324b9c1b51eSKate Stone   if (GetGlobalProperties()->GetScanType() != eKASLRScanExhaustiveScan) {
3256ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3266ba6d3d1SJason Molenda   }
3276ba6d3d1SJason Molenda 
3286ba6d3d1SJason Molenda   addr_t kernel_range_low, kernel_range_high;
329b9c1b51eSKate Stone   if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8) {
3306ba6d3d1SJason Molenda     kernel_range_low = 1ULL << 63;
3316ba6d3d1SJason Molenda     kernel_range_high = UINT64_MAX;
332b9c1b51eSKate Stone   } else {
3336ba6d3d1SJason Molenda     kernel_range_low = 1ULL << 31;
3346ba6d3d1SJason Molenda     kernel_range_high = UINT32_MAX;
3356ba6d3d1SJason Molenda   }
3366ba6d3d1SJason Molenda 
3376ba6d3d1SJason Molenda   // Stepping through memory at one-megabyte resolution looking for a kernel
3386ba6d3d1SJason Molenda   // rarely works (fast enough) with a 64-bit address space -- for now, let's
3396ba6d3d1SJason Molenda   // not even bother.  We may be attaching to something which *isn't* a kernel
3406ba6d3d1SJason Molenda   // and we don't want to spin for minutes on-end looking for a kernel.
3416ba6d3d1SJason Molenda   if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
3426ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3436ba6d3d1SJason Molenda 
3446ba6d3d1SJason Molenda   addr_t addr = kernel_range_low;
3456ba6d3d1SJason Molenda 
346b9c1b51eSKate Stone   while (addr >= kernel_range_low && addr < kernel_range_high) {
3476ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr, process).IsValid())
3486ba6d3d1SJason Molenda       return addr;
3496ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x1000, process).IsValid())
3506ba6d3d1SJason Molenda       return addr + 0x1000;
3516ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x2000, process).IsValid())
3526ba6d3d1SJason Molenda       return addr + 0x2000;
353a02869deSJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x4000, process).IsValid())
354a02869deSJason Molenda       return addr + 0x4000;
3556ba6d3d1SJason Molenda     addr += 0x100000;
3566ba6d3d1SJason Molenda   }
3576ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
3586ba6d3d1SJason Molenda }
3596ba6d3d1SJason Molenda 
3606ba6d3d1SJason Molenda //----------------------------------------------------------------------
3616ba6d3d1SJason Molenda // Given an address in memory, look to see if there is a kernel image at that
3626ba6d3d1SJason Molenda // address.
363b9c1b51eSKate Stone // Returns a UUID; if a kernel was not found at that address, UUID.IsValid()
364b9c1b51eSKate Stone // will be false.
3656ba6d3d1SJason Molenda //----------------------------------------------------------------------
3666ba6d3d1SJason Molenda lldb_private::UUID
367b9c1b51eSKate Stone DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress(lldb::addr_t addr,
368b9c1b51eSKate Stone                                                         Process *process) {
369ec504238SJason Molenda   Log *log(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
3706ba6d3d1SJason Molenda   if (addr == LLDB_INVALID_ADDRESS)
3716ba6d3d1SJason Molenda     return UUID();
3726ba6d3d1SJason Molenda 
373ec504238SJason Molenda   if (log)
374b9c1b51eSKate Stone     log->Printf("DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress: "
375b9c1b51eSKate Stone                 "looking for kernel binary at 0x%" PRIx64,
376b9c1b51eSKate Stone                 addr);
377ec504238SJason Molenda 
378b9c1b51eSKate Stone   // First try a quick test -- read the first 4 bytes and see if there is a
379b9c1b51eSKate Stone   // valid Mach-O magic field there
3806ba6d3d1SJason Molenda   // (the first field of the mach_header/mach_header_64 struct).
3816ba6d3d1SJason Molenda 
3826ba6d3d1SJason Molenda   Error read_error;
383b9c1b51eSKate Stone   uint64_t result = process->ReadUnsignedIntegerFromMemory(
384b9c1b51eSKate Stone       addr, 4, LLDB_INVALID_ADDRESS, read_error);
385b9c1b51eSKate Stone   if (result != llvm::MachO::MH_MAGIC_64 && result != llvm::MachO::MH_MAGIC &&
386b9c1b51eSKate Stone       result != llvm::MachO::MH_CIGAM && result != llvm::MachO::MH_CIGAM_64) {
3876ba6d3d1SJason Molenda     return UUID();
3886ba6d3d1SJason Molenda   }
3896ba6d3d1SJason Molenda 
3906ba6d3d1SJason Molenda   // Read the mach header and see whether it looks like a kernel
3916ba6d3d1SJason Molenda   llvm::MachO::mach_header header;
392b9c1b51eSKate Stone   if (process->DoReadMemory(addr, &header, sizeof(header), read_error) !=
393b9c1b51eSKate Stone       sizeof(header))
3946ba6d3d1SJason Molenda     return UUID();
3956ba6d3d1SJason Molenda 
396510938e5SCharles Davis   if (header.magic == llvm::MachO::MH_CIGAM ||
397b9c1b51eSKate Stone       header.magic == llvm::MachO::MH_CIGAM_64) {
3986ba6d3d1SJason Molenda     header.magic = llvm::ByteSwap_32(header.magic);
3996ba6d3d1SJason Molenda     header.cputype = llvm::ByteSwap_32(header.cputype);
4006ba6d3d1SJason Molenda     header.cpusubtype = llvm::ByteSwap_32(header.cpusubtype);
4016ba6d3d1SJason Molenda     header.filetype = llvm::ByteSwap_32(header.filetype);
4026ba6d3d1SJason Molenda     header.ncmds = llvm::ByteSwap_32(header.ncmds);
4036ba6d3d1SJason Molenda     header.sizeofcmds = llvm::ByteSwap_32(header.sizeofcmds);
4046ba6d3d1SJason Molenda     header.flags = llvm::ByteSwap_32(header.flags);
4056ba6d3d1SJason Molenda   }
4066ba6d3d1SJason Molenda 
407b9c1b51eSKate Stone   // A kernel is an executable which does not have the dynamic link object flag
408b9c1b51eSKate Stone   // set.
409b9c1b51eSKate Stone   if (header.filetype == llvm::MachO::MH_EXECUTE &&
410b9c1b51eSKate Stone       (header.flags & llvm::MachO::MH_DYLDLINK) == 0) {
4116ba6d3d1SJason Molenda     // Create a full module to get the UUID
412b9c1b51eSKate Stone     ModuleSP memory_module_sp = process->ReadModuleFromMemory(
413b9c1b51eSKate Stone         FileSpec("temp_mach_kernel", false), addr);
4146ba6d3d1SJason Molenda     if (!memory_module_sp.get())
4156ba6d3d1SJason Molenda       return UUID();
4166ba6d3d1SJason Molenda 
4176ba6d3d1SJason Molenda     ObjectFile *exe_objfile = memory_module_sp->GetObjectFile();
418*bc22c8d8SJason Molenda     if (exe_objfile == NULL) {
419*bc22c8d8SJason Molenda       if (log)
420*bc22c8d8SJason Molenda         log->Printf("DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress "
421*bc22c8d8SJason Molenda                     "found a binary at 0x%" PRIx64
422*bc22c8d8SJason Molenda                     " but could not create an object file from memory",
423*bc22c8d8SJason Molenda                     addr);
4246ba6d3d1SJason Molenda       return UUID();
425*bc22c8d8SJason Molenda     }
4266ba6d3d1SJason Molenda 
427b9c1b51eSKate Stone     if (exe_objfile->GetType() == ObjectFile::eTypeExecutable &&
428b9c1b51eSKate Stone         exe_objfile->GetStrata() == ObjectFile::eStrataKernel) {
429a4ce2538SJason Molenda       ArchSpec kernel_arch(eArchTypeMachO, header.cputype, header.cpusubtype);
430b9c1b51eSKate Stone       if (!process->GetTarget().GetArchitecture().IsCompatibleMatch(
431b9c1b51eSKate Stone               kernel_arch)) {
432a4ce2538SJason Molenda         process->GetTarget().SetArchitecture(kernel_arch);
433a4ce2538SJason Molenda       }
434*bc22c8d8SJason Molenda       if (log) {
435*bc22c8d8SJason Molenda         std::string uuid_str;
436*bc22c8d8SJason Molenda         if (memory_module_sp->GetUUID().IsValid()) {
437*bc22c8d8SJason Molenda           uuid_str = "with UUID ";
438*bc22c8d8SJason Molenda           uuid_str += memory_module_sp->GetUUID().GetAsString();
439*bc22c8d8SJason Molenda         } else {
440*bc22c8d8SJason Molenda           uuid_str = "and no LC_UUID found in load commands ";
441*bc22c8d8SJason Molenda         }
442*bc22c8d8SJason Molenda         log->Printf(
443*bc22c8d8SJason Molenda             "DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress: "
444*bc22c8d8SJason Molenda             "kernel binary image found at 0x%" PRIx64 " with arch '%s' %s",
445*bc22c8d8SJason Molenda             addr, kernel_arch.GetTriple().str().c_str(), uuid_str.c_str());
446*bc22c8d8SJason Molenda       }
4476ba6d3d1SJason Molenda       return memory_module_sp->GetUUID();
4486ba6d3d1SJason Molenda     }
4496ba6d3d1SJason Molenda   }
4506ba6d3d1SJason Molenda 
4516ba6d3d1SJason Molenda   return UUID();
4526ba6d3d1SJason Molenda }
4536ba6d3d1SJason Molenda 
4546ba6d3d1SJason Molenda //----------------------------------------------------------------------
455d4bfbc9aSGreg Clayton // Constructor
456d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
457b9c1b51eSKate Stone DynamicLoaderDarwinKernel::DynamicLoaderDarwinKernel(Process *process,
458b9c1b51eSKate Stone                                                      lldb::addr_t kernel_addr)
459b9c1b51eSKate Stone     : DynamicLoader(process), m_kernel_load_address(kernel_addr), m_kernel(),
460b9c1b51eSKate Stone       m_kext_summary_header_ptr_addr(), m_kext_summary_header_addr(),
461b9c1b51eSKate Stone       m_kext_summary_header(), m_known_kexts(), m_mutex(),
462b9c1b51eSKate Stone       m_break_id(LLDB_INVALID_BREAK_ID) {
463615eb7e6SGreg Clayton   Error error;
464b9c1b51eSKate Stone   PlatformSP platform_sp(
465b9c1b51eSKate Stone       Platform::Create(PlatformDarwinKernel::GetPluginNameStatic(), error));
4661c627543SJason Molenda   // Only select the darwin-kernel Platform if we've been asked to load kexts.
4671c627543SJason Molenda   // It can take some time to scan over all of the kext info.plists and that
4681c627543SJason Molenda   // shouldn't be done if kext loading is explicitly disabled.
469b9c1b51eSKate Stone   if (platform_sp.get() && GetGlobalProperties()->GetLoadKexts()) {
4701c627543SJason Molenda     process->GetTarget().SetPlatform(platform_sp);
4711c627543SJason Molenda   }
472d4bfbc9aSGreg Clayton }
473d4bfbc9aSGreg Clayton 
474d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
475d4bfbc9aSGreg Clayton // Destructor
476d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
477b9c1b51eSKate Stone DynamicLoaderDarwinKernel::~DynamicLoaderDarwinKernel() { Clear(true); }
478d4bfbc9aSGreg Clayton 
479b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::UpdateIfNeeded() {
480d4bfbc9aSGreg Clayton   LoadKernelModuleIfNeeded();
481d4bfbc9aSGreg Clayton   SetNotificationBreakpointIfNeeded();
482d4bfbc9aSGreg Clayton }
483d4bfbc9aSGreg Clayton //------------------------------------------------------------------
484d4bfbc9aSGreg Clayton /// Called after attaching a process.
485d4bfbc9aSGreg Clayton ///
486d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
487d4bfbc9aSGreg Clayton /// attaching to a process.
488d4bfbc9aSGreg Clayton //------------------------------------------------------------------
489b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DidAttach() {
490d4bfbc9aSGreg Clayton   PrivateInitialize(m_process);
491d4bfbc9aSGreg Clayton   UpdateIfNeeded();
492d4bfbc9aSGreg Clayton }
493d4bfbc9aSGreg Clayton 
494d4bfbc9aSGreg Clayton //------------------------------------------------------------------
495d4bfbc9aSGreg Clayton /// Called after attaching a process.
496d4bfbc9aSGreg Clayton ///
497d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
498d4bfbc9aSGreg Clayton /// attaching to a process.
499d4bfbc9aSGreg Clayton //------------------------------------------------------------------
500b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DidLaunch() {
501d4bfbc9aSGreg Clayton   PrivateInitialize(m_process);
502d4bfbc9aSGreg Clayton   UpdateIfNeeded();
503d4bfbc9aSGreg Clayton }
504d4bfbc9aSGreg Clayton 
505d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
506d4bfbc9aSGreg Clayton // Clear out the state of this class.
507d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
508b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Clear(bool clear_process) {
50916ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
510d4bfbc9aSGreg Clayton 
511d4bfbc9aSGreg Clayton   if (m_process->IsAlive() && LLDB_BREAK_ID_IS_VALID(m_break_id))
512d4bfbc9aSGreg Clayton     m_process->ClearBreakpointSiteByID(m_break_id);
513d4bfbc9aSGreg Clayton 
514d4bfbc9aSGreg Clayton   if (clear_process)
515d4bfbc9aSGreg Clayton     m_process = NULL;
516306bd0aaSJason Molenda   m_kernel.Clear();
517306bd0aaSJason Molenda   m_known_kexts.clear();
518d4bfbc9aSGreg Clayton   m_kext_summary_header_ptr_addr.Clear();
519d4bfbc9aSGreg Clayton   m_kext_summary_header_addr.Clear();
520d4bfbc9aSGreg Clayton   m_break_id = LLDB_INVALID_BREAK_ID;
521d4bfbc9aSGreg Clayton }
522d4bfbc9aSGreg Clayton 
523b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::LoadImageAtFileAddress(
524b9c1b51eSKate Stone     Process *process) {
5252af282a1SGreg Clayton   if (IsLoaded())
5262af282a1SGreg Clayton     return true;
5272af282a1SGreg Clayton 
528b9c1b51eSKate Stone   if (m_module_sp) {
5292af282a1SGreg Clayton     bool changed = false;
530751caf65SGreg Clayton     if (m_module_sp->SetLoadAddress(process->GetTarget(), 0, true, changed))
531306bd0aaSJason Molenda       m_load_process_stop_id = process->GetStopID();
5322af282a1SGreg Clayton   }
5332af282a1SGreg Clayton   return false;
5342af282a1SGreg Clayton }
5352af282a1SGreg Clayton 
536b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetModule(ModuleSP module_sp) {
537306bd0aaSJason Molenda   m_module_sp = module_sp;
538b9c1b51eSKate Stone   if (module_sp.get() && module_sp->GetObjectFile()) {
539b9c1b51eSKate Stone     if (module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeExecutable &&
540b9c1b51eSKate Stone         module_sp->GetObjectFile()->GetStrata() == ObjectFile::eStrataKernel) {
541306bd0aaSJason Molenda       m_kernel_image = true;
542b9c1b51eSKate Stone     } else {
543306bd0aaSJason Molenda       m_kernel_image = false;
544306bd0aaSJason Molenda     }
545306bd0aaSJason Molenda   }
546306bd0aaSJason Molenda }
547306bd0aaSJason Molenda 
548b9c1b51eSKate Stone ModuleSP DynamicLoaderDarwinKernel::KextImageInfo::GetModule() {
549306bd0aaSJason Molenda   return m_module_sp;
550306bd0aaSJason Molenda }
551306bd0aaSJason Molenda 
552b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetLoadAddress(
553b9c1b51eSKate Stone     addr_t load_addr) {
554306bd0aaSJason Molenda   m_load_address = load_addr;
555306bd0aaSJason Molenda }
556306bd0aaSJason Molenda 
557b9c1b51eSKate Stone addr_t DynamicLoaderDarwinKernel::KextImageInfo::GetLoadAddress() const {
558306bd0aaSJason Molenda   return m_load_address;
559306bd0aaSJason Molenda }
560306bd0aaSJason Molenda 
561b9c1b51eSKate Stone uint64_t DynamicLoaderDarwinKernel::KextImageInfo::GetSize() const {
562306bd0aaSJason Molenda   return m_size;
563306bd0aaSJason Molenda }
564306bd0aaSJason Molenda 
565b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetSize(uint64_t size) {
566306bd0aaSJason Molenda   m_size = size;
567306bd0aaSJason Molenda }
568306bd0aaSJason Molenda 
569b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::KextImageInfo::GetProcessStopId() const {
570306bd0aaSJason Molenda   return m_load_process_stop_id;
571306bd0aaSJason Molenda }
572306bd0aaSJason Molenda 
573b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetProcessStopId(
574b9c1b51eSKate Stone     uint32_t stop_id) {
575306bd0aaSJason Molenda   m_load_process_stop_id = stop_id;
576306bd0aaSJason Molenda }
577306bd0aaSJason Molenda 
578b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::
579b9c1b51eSKate Stone operator==(const KextImageInfo &rhs) {
580b9c1b51eSKate Stone   if (m_uuid.IsValid() || rhs.GetUUID().IsValid()) {
581b9c1b51eSKate Stone     if (m_uuid == rhs.GetUUID()) {
582306bd0aaSJason Molenda       return true;
583306bd0aaSJason Molenda     }
584306bd0aaSJason Molenda     return false;
585306bd0aaSJason Molenda   }
586306bd0aaSJason Molenda 
587306bd0aaSJason Molenda   if (m_name == rhs.GetName() && m_load_address == rhs.GetLoadAddress())
588306bd0aaSJason Molenda     return true;
589306bd0aaSJason Molenda 
590306bd0aaSJason Molenda   return false;
591306bd0aaSJason Molenda }
592306bd0aaSJason Molenda 
593b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetName(const char *name) {
594306bd0aaSJason Molenda   m_name = name;
595306bd0aaSJason Molenda }
596306bd0aaSJason Molenda 
597b9c1b51eSKate Stone std::string DynamicLoaderDarwinKernel::KextImageInfo::GetName() const {
598306bd0aaSJason Molenda   return m_name;
599306bd0aaSJason Molenda }
600306bd0aaSJason Molenda 
601b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetUUID(const UUID &uuid) {
602306bd0aaSJason Molenda   m_uuid = uuid;
603306bd0aaSJason Molenda }
604306bd0aaSJason Molenda 
605b9c1b51eSKate Stone UUID DynamicLoaderDarwinKernel::KextImageInfo::GetUUID() const {
606306bd0aaSJason Molenda   return m_uuid;
607306bd0aaSJason Molenda }
608306bd0aaSJason Molenda 
609b9c1b51eSKate Stone // Given the m_load_address from the kext summaries, and a UUID, try to create
610b9c1b51eSKate Stone // an in-memory
611b9c1b51eSKate Stone // Module at that address.  Require that the MemoryModule have a matching UUID
612b9c1b51eSKate Stone // and detect
613306bd0aaSJason Molenda // if this MemoryModule is a kernel or a kext.
614306bd0aaSJason Molenda //
615306bd0aaSJason Molenda // Returns true if m_memory_module_sp is now set to a valid Module.
616306bd0aaSJason Molenda 
617b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::ReadMemoryModule(
618b9c1b51eSKate Stone     Process *process) {
61908a3258bSJason Molenda   Log *log = lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_HOST);
620306bd0aaSJason Molenda   if (m_memory_module_sp.get() != NULL)
621306bd0aaSJason Molenda     return true;
622306bd0aaSJason Molenda   if (m_load_address == LLDB_INVALID_ADDRESS)
623306bd0aaSJason Molenda     return false;
624306bd0aaSJason Molenda 
625306bd0aaSJason Molenda   FileSpec file_spec;
626306bd0aaSJason Molenda   file_spec.SetFile(m_name.c_str(), false);
627306bd0aaSJason Molenda 
628b9c1b51eSKate Stone   ModuleSP memory_module_sp =
629b9c1b51eSKate Stone       process->ReadModuleFromMemory(file_spec, m_load_address);
630306bd0aaSJason Molenda 
631306bd0aaSJason Molenda   if (memory_module_sp.get() == NULL)
632306bd0aaSJason Molenda     return false;
633306bd0aaSJason Molenda 
634306bd0aaSJason Molenda   bool is_kernel = false;
635b9c1b51eSKate Stone   if (memory_module_sp->GetObjectFile()) {
636b9c1b51eSKate Stone     if (memory_module_sp->GetObjectFile()->GetType() ==
637b9c1b51eSKate Stone             ObjectFile::eTypeExecutable &&
638b9c1b51eSKate Stone         memory_module_sp->GetObjectFile()->GetStrata() ==
639b9c1b51eSKate Stone             ObjectFile::eStrataKernel) {
640306bd0aaSJason Molenda       is_kernel = true;
641b9c1b51eSKate Stone     } else if (memory_module_sp->GetObjectFile()->GetType() ==
642b9c1b51eSKate Stone                ObjectFile::eTypeSharedLibrary) {
643306bd0aaSJason Molenda       is_kernel = false;
644306bd0aaSJason Molenda     }
645306bd0aaSJason Molenda   }
646306bd0aaSJason Molenda 
647b9c1b51eSKate Stone   // If this is a kext, and the kernel specified what UUID we should find at
648b9c1b51eSKate Stone   // this
649b9c1b51eSKate Stone   // load address, require that the memory module have a matching UUID or
650b9c1b51eSKate Stone   // something
65138e70d11SJason Molenda   // has gone wrong and we should discard it.
652b9c1b51eSKate Stone   if (m_uuid.IsValid()) {
653b9c1b51eSKate Stone     if (m_uuid != memory_module_sp->GetUUID()) {
654b9c1b51eSKate Stone       if (log) {
655b9c1b51eSKate Stone         log->Printf("KextImageInfo::ReadMemoryModule the kernel said to find "
656b9c1b51eSKate Stone                     "uuid %s at 0x%" PRIx64
657b9c1b51eSKate Stone                     " but instead we found uuid %s, throwing it away",
658b9c1b51eSKate Stone                     m_uuid.GetAsString().c_str(), m_load_address,
659b9c1b51eSKate Stone                     memory_module_sp->GetUUID().GetAsString().c_str());
66008a3258bSJason Molenda       }
661306bd0aaSJason Molenda       return false;
662306bd0aaSJason Molenda     }
663306bd0aaSJason Molenda   }
664306bd0aaSJason Molenda 
665306bd0aaSJason Molenda   // If the in-memory Module has a UUID, let's use that.
666b9c1b51eSKate Stone   if (!m_uuid.IsValid() && memory_module_sp->GetUUID().IsValid()) {
667306bd0aaSJason Molenda     m_uuid = memory_module_sp->GetUUID();
668306bd0aaSJason Molenda   }
669306bd0aaSJason Molenda 
670306bd0aaSJason Molenda   m_memory_module_sp = memory_module_sp;
671306bd0aaSJason Molenda   m_kernel_image = is_kernel;
672b9c1b51eSKate Stone   if (is_kernel) {
673b9c1b51eSKate Stone     if (log) {
67408a3258bSJason Molenda       // This is unusual and probably not intended
675b9c1b51eSKate Stone       log->Printf("KextImageInfo::ReadMemoryModule read the kernel binary out "
676b9c1b51eSKate Stone                   "of memory");
67708a3258bSJason Molenda     }
678b9c1b51eSKate Stone     if (memory_module_sp->GetArchitecture().IsValid()) {
679306bd0aaSJason Molenda       process->GetTarget().SetArchitecture(memory_module_sp->GetArchitecture());
680306bd0aaSJason Molenda     }
681b9c1b51eSKate Stone     if (m_uuid.IsValid()) {
68248975833SJason Molenda       ModuleSP exe_module_sp = process->GetTarget().GetExecutableModule();
683b9c1b51eSKate Stone       if (exe_module_sp.get() && exe_module_sp->GetUUID().IsValid()) {
684b9c1b51eSKate Stone         if (m_uuid != exe_module_sp->GetUUID()) {
68548975833SJason Molenda           // The user specified a kernel binary that has a different UUID than
68648975833SJason Molenda           // the kernel actually running in memory.  This never ends well;
68748975833SJason Molenda           // clear the user specified kernel binary from the Target.
68848975833SJason Molenda 
68948975833SJason Molenda           m_module_sp.reset();
69048975833SJason Molenda 
69148975833SJason Molenda           ModuleList user_specified_kernel_list;
69248975833SJason Molenda           user_specified_kernel_list.Append(exe_module_sp);
69348975833SJason Molenda           process->GetTarget().GetImages().Remove(user_specified_kernel_list);
69438e70d11SJason Molenda         }
69538e70d11SJason Molenda       }
69638e70d11SJason Molenda     }
697306bd0aaSJason Molenda   }
698306bd0aaSJason Molenda 
699306bd0aaSJason Molenda   return true;
700306bd0aaSJason Molenda }
701306bd0aaSJason Molenda 
702b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::IsKernel() const {
703306bd0aaSJason Molenda   return m_kernel_image == true;
704306bd0aaSJason Molenda }
705306bd0aaSJason Molenda 
706b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetIsKernel(bool is_kernel) {
707306bd0aaSJason Molenda   m_kernel_image = is_kernel;
708306bd0aaSJason Molenda }
709306bd0aaSJason Molenda 
710b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::LoadImageUsingMemoryModule(
711b9c1b51eSKate Stone     Process *process) {
712c859e2d5SGreg Clayton   if (IsLoaded())
713c859e2d5SGreg Clayton     return true;
714c859e2d5SGreg Clayton 
715c859e2d5SGreg Clayton   Target &target = process->GetTarget();
71687a04b24SJason Molenda 
717b9c1b51eSKate Stone   // If we don't have / can't create a memory module for this kext, don't try to
718b9c1b51eSKate Stone   // load it - we won't
719306bd0aaSJason Molenda   // have the correct segment load addresses.
720b9c1b51eSKate Stone   if (!ReadMemoryModule(process)) {
721a8cb765cSJason Molenda     Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
722a8cb765cSJason Molenda     if (log)
723b9c1b51eSKate Stone       log->Printf("Unable to read '%s' from memory at address 0x%" PRIx64
724b9c1b51eSKate Stone                   " to get the segment load addresses.",
725b9c1b51eSKate Stone                   m_name.c_str(), m_load_address);
72687a04b24SJason Molenda     return false;
72787a04b24SJason Molenda   }
72887a04b24SJason Molenda 
729306bd0aaSJason Molenda   bool uuid_is_valid = m_uuid.IsValid();
730306bd0aaSJason Molenda 
731b9c1b51eSKate Stone   if (IsKernel() && uuid_is_valid && m_memory_module_sp.get()) {
73244d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
733b9c1b51eSKate Stone     if (s) {
734b9c1b51eSKate Stone       s->Printf("Kernel UUID: %s\n",
735b9c1b51eSKate Stone                 m_memory_module_sp->GetUUID().GetAsString().c_str());
736e575e7bcSJason Molenda       s->Printf("Load Address: 0x%" PRIx64 "\n", m_load_address);
737e575e7bcSJason Molenda     }
738e575e7bcSJason Molenda   }
739e575e7bcSJason Molenda 
740b9c1b51eSKate Stone   if (!m_module_sp) {
741b9c1b51eSKate Stone     // See if the kext has already been loaded into the target, probably by the
742b9c1b51eSKate Stone     // user doing target modules add.
7431759848bSEnrico Granata     const ModuleList &target_images = target.GetImages();
744306bd0aaSJason Molenda     m_module_sp = target_images.FindModule(m_uuid);
745c859e2d5SGreg Clayton 
746306bd0aaSJason Molenda     // Search for the kext on the local filesystem via the UUID
747b9c1b51eSKate Stone     if (!m_module_sp && uuid_is_valid) {
7482af282a1SGreg Clayton       ModuleSpec module_spec;
749306bd0aaSJason Molenda       module_spec.GetUUID() = m_uuid;
75053667f5dSJason Molenda       module_spec.GetArchitecture() = target.GetArchitecture();
751bb860bd2SJason Molenda 
752b9c1b51eSKate Stone       // For the kernel, we really do need an on-disk file copy of the binary to
753b9c1b51eSKate Stone       // do anything useful.
754d76fb6eaSJason Molenda       // This will force a clal to
755b9c1b51eSKate Stone       if (IsKernel()) {
756b9c1b51eSKate Stone         if (Symbols::DownloadObjectAndSymbolFile(module_spec, true)) {
757b9c1b51eSKate Stone           if (module_spec.GetFileSpec().Exists()) {
758b9c1b51eSKate Stone             m_module_sp.reset(new Module(module_spec.GetFileSpec(),
759b9c1b51eSKate Stone                                          target.GetArchitecture()));
760b9c1b51eSKate Stone             if (m_module_sp.get() &&
761b9c1b51eSKate Stone                 m_module_sp->MatchesModuleSpec(module_spec)) {
762bb860bd2SJason Molenda               ModuleList loaded_module_list;
763306bd0aaSJason Molenda               loaded_module_list.Append(m_module_sp);
764bb860bd2SJason Molenda               target.ModulesDidLoad(loaded_module_list);
765bb860bd2SJason Molenda             }
766bb860bd2SJason Molenda           }
767bb860bd2SJason Molenda         }
768d76fb6eaSJason Molenda       }
769bb860bd2SJason Molenda 
770b9c1b51eSKate Stone       // If the current platform is PlatformDarwinKernel, create a ModuleSpec
771b9c1b51eSKate Stone       // with the filename set
772b9c1b51eSKate Stone       // to be the bundle ID for this kext, e.g.
773b9c1b51eSKate Stone       // "com.apple.filesystems.msdosfs", and ask the platform
7741c627543SJason Molenda       // to find it.
7751c627543SJason Molenda       PlatformSP platform_sp(target.GetPlatform());
776b9c1b51eSKate Stone       if (!m_module_sp && platform_sp) {
77757abc5d6SGreg Clayton         ConstString platform_name(platform_sp->GetPluginName());
778b9c1b51eSKate Stone         static ConstString g_platform_name(
779b9c1b51eSKate Stone             PlatformDarwinKernel::GetPluginNameStatic());
780b9c1b51eSKate Stone         if (platform_name == g_platform_name) {
7811c627543SJason Molenda           ModuleSpec kext_bundle_module_spec(module_spec);
7821c627543SJason Molenda           FileSpec kext_filespec(m_name.c_str(), false);
7831c627543SJason Molenda           kext_bundle_module_spec.GetFileSpec() = kext_filespec;
784b9c1b51eSKate Stone           platform_sp->GetSharedModule(
785b9c1b51eSKate Stone               kext_bundle_module_spec, process, m_module_sp,
786b9c1b51eSKate Stone               &target.GetExecutableSearchPaths(), NULL, NULL);
7871c627543SJason Molenda         }
7881c627543SJason Molenda       }
7891c627543SJason Molenda 
790d76fb6eaSJason Molenda       // Ask the Target to find this file on the local system, if possible.
791bb860bd2SJason Molenda       // This will search in the list of currently-loaded files, look in the
792bb860bd2SJason Molenda       // standard search paths on the system, and on a Mac it will try calling
793bb860bd2SJason Molenda       // the DebugSymbols framework with the UUID to find the binary via its
794bb860bd2SJason Molenda       // search methods.
795b9c1b51eSKate Stone       if (!m_module_sp) {
796306bd0aaSJason Molenda         m_module_sp = target.GetSharedModule(module_spec);
797306bd0aaSJason Molenda       }
798e575e7bcSJason Molenda 
799b9c1b51eSKate Stone       if (IsKernel() && !m_module_sp) {
80044d93782SGreg Clayton         Stream *s = target.GetDebugger().GetOutputFile().get();
801b9c1b51eSKate Stone         if (s) {
802b9c1b51eSKate Stone           s->Printf("WARNING: Unable to locate kernel binary on the debugger "
803b9c1b51eSKate Stone                     "system.\n");
804e575e7bcSJason Molenda         }
805e575e7bcSJason Molenda       }
806b9a01b39SGreg Clayton     }
80765d57a3dSJason Molenda 
808306bd0aaSJason Molenda     // If we managed to find a module, append it to the target's list of images.
809306bd0aaSJason Molenda     // If we also have a memory module, require that they have matching UUIDs
810b9c1b51eSKate Stone     if (m_module_sp) {
811306bd0aaSJason Molenda       bool uuid_match_ok = true;
812b9c1b51eSKate Stone       if (m_memory_module_sp) {
813b9c1b51eSKate Stone         if (m_module_sp->GetUUID() != m_memory_module_sp->GetUUID()) {
814306bd0aaSJason Molenda           uuid_match_ok = false;
81565d57a3dSJason Molenda         }
81665d57a3dSJason Molenda       }
817b9c1b51eSKate Stone       if (uuid_match_ok) {
818a4d3e1d2SJason Molenda         target.GetImages().AppendIfNeeded(m_module_sp);
819b9c1b51eSKate Stone         if (IsKernel() &&
820b9c1b51eSKate Stone             target.GetExecutableModulePointer() != m_module_sp.get()) {
821306bd0aaSJason Molenda           target.SetExecutableModule(m_module_sp, false);
822306bd0aaSJason Molenda         }
823c859e2d5SGreg Clayton       }
824c859e2d5SGreg Clayton     }
825bb860bd2SJason Molenda   }
826c859e2d5SGreg Clayton 
827b9c1b51eSKate Stone   if (!m_module_sp && !IsKernel() && m_uuid.IsValid() && !m_name.empty()) {
82844d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
829b9c1b51eSKate Stone     if (s) {
830b9c1b51eSKate Stone       s->Printf("warning: Can't find binary/dSYM for %s (%s)\n", m_name.c_str(),
831b9c1b51eSKate Stone                 m_uuid.GetAsString().c_str());
83256c23285SJason Molenda     }
83356c23285SJason Molenda   }
834c859e2d5SGreg Clayton 
83567408537SGreg Clayton   static ConstString g_section_name_LINKEDIT("__LINKEDIT");
83667408537SGreg Clayton 
837b9c1b51eSKate Stone   if (m_memory_module_sp && m_module_sp) {
838b9c1b51eSKate Stone     if (m_module_sp->GetUUID() == m_memory_module_sp->GetUUID()) {
839306bd0aaSJason Molenda       ObjectFile *ondisk_object_file = m_module_sp->GetObjectFile();
840306bd0aaSJason Molenda       ObjectFile *memory_object_file = m_memory_module_sp->GetObjectFile();
84167408537SGreg Clayton 
842b9c1b51eSKate Stone       if (memory_object_file && ondisk_object_file) {
843b9c1b51eSKate Stone         // The memory_module for kexts may have an invalid __LINKEDIT seg; skip
844b9c1b51eSKate Stone         // it.
845306bd0aaSJason Molenda         const bool ignore_linkedit = !IsKernel();
84667408537SGreg Clayton 
847c859e2d5SGreg Clayton         SectionList *ondisk_section_list = ondisk_object_file->GetSectionList();
848c859e2d5SGreg Clayton         SectionList *memory_section_list = memory_object_file->GetSectionList();
849b9c1b51eSKate Stone         if (memory_section_list && ondisk_section_list) {
850bae2f2f8SGreg Clayton           const uint32_t num_ondisk_sections = ondisk_section_list->GetSize();
851c859e2d5SGreg Clayton           // There may be CTF sections in the memory image so we can't
852c859e2d5SGreg Clayton           // always just compare the number of sections (which are actually
853c859e2d5SGreg Clayton           // segments in mach-o parlance)
854c859e2d5SGreg Clayton           uint32_t sect_idx = 0;
855bae2f2f8SGreg Clayton 
85653667f5dSJason Molenda           // Use the memory_module's addresses for each section to set the
85753667f5dSJason Molenda           // file module's load address as appropriate.  We don't want to use
85853667f5dSJason Molenda           // a single slide value for the entire kext - different segments may
85953667f5dSJason Molenda           // be slid different amounts by the kext loader.
860bae2f2f8SGreg Clayton 
861bae2f2f8SGreg Clayton           uint32_t num_sections_loaded = 0;
862b9c1b51eSKate Stone           for (sect_idx = 0; sect_idx < num_ondisk_sections; ++sect_idx) {
863b9c1b51eSKate Stone             SectionSP ondisk_section_sp(
864b9c1b51eSKate Stone                 ondisk_section_list->GetSectionAtIndex(sect_idx));
865b9c1b51eSKate Stone             if (ondisk_section_sp) {
86667408537SGreg Clayton               // Don't ever load __LINKEDIT as it may or may not be actually
86767408537SGreg Clayton               // mapped into memory and there is no current way to tell.
86867408537SGreg Clayton               // I filed rdar://problem/12851706 to track being able to tell
86967408537SGreg Clayton               // if the __LINKEDIT is actually mapped, but until then, we need
87067408537SGreg Clayton               // to not load the __LINKEDIT
871b9c1b51eSKate Stone               if (ignore_linkedit &&
872b9c1b51eSKate Stone                   ondisk_section_sp->GetName() == g_section_name_LINKEDIT)
87367408537SGreg Clayton                 continue;
87467408537SGreg Clayton 
875b9c1b51eSKate Stone               const Section *memory_section =
876b9c1b51eSKate Stone                   memory_section_list
877b9c1b51eSKate Stone                       ->FindSectionByName(ondisk_section_sp->GetName())
878b9c1b51eSKate Stone                       .get();
879b9c1b51eSKate Stone               if (memory_section) {
880b9c1b51eSKate Stone                 target.SetSectionLoadAddress(ondisk_section_sp,
881b9c1b51eSKate Stone                                              memory_section->GetFileAddress());
882bae2f2f8SGreg Clayton                 ++num_sections_loaded;
883c859e2d5SGreg Clayton               }
884bae2f2f8SGreg Clayton             }
885bae2f2f8SGreg Clayton           }
886bae2f2f8SGreg Clayton           if (num_sections_loaded > 0)
887306bd0aaSJason Molenda             m_load_process_stop_id = process->GetStopID();
888bae2f2f8SGreg Clayton           else
889306bd0aaSJason Molenda             m_module_sp.reset(); // No sections were loaded
890b9c1b51eSKate Stone         } else
891306bd0aaSJason Molenda           m_module_sp.reset(); // One or both section lists
892b9c1b51eSKate Stone       } else
893306bd0aaSJason Molenda         m_module_sp.reset(); // One or both object files missing
894b9c1b51eSKate Stone     } else
895306bd0aaSJason Molenda       m_module_sp.reset(); // UUID mismatch
896c859e2d5SGreg Clayton   }
897c859e2d5SGreg Clayton 
898c859e2d5SGreg Clayton   bool is_loaded = IsLoaded();
899c859e2d5SGreg Clayton 
900b9c1b51eSKate Stone   if (is_loaded && m_module_sp && IsKernel()) {
90144d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
902b9c1b51eSKate Stone     if (s) {
903732238cdSJason Molenda       ObjectFile *kernel_object_file = m_module_sp->GetObjectFile();
904b9c1b51eSKate Stone       if (kernel_object_file) {
905b9c1b51eSKate Stone         addr_t file_address =
906b9c1b51eSKate Stone             kernel_object_file->GetHeaderAddress().GetFileAddress();
907b9c1b51eSKate Stone         if (m_load_address != LLDB_INVALID_ADDRESS &&
908b9c1b51eSKate Stone             file_address != LLDB_INVALID_ADDRESS) {
909b9c1b51eSKate Stone           s->Printf("Kernel slid 0x%" PRIx64 " in memory.\n",
910b9c1b51eSKate Stone                     m_load_address - file_address);
911732238cdSJason Molenda         }
912732238cdSJason Molenda       }
913743e4396SJason Molenda       {
914b5ad4ec7SGreg Clayton         s->Printf("Loaded kernel file %s\n",
915b5ad4ec7SGreg Clayton                   m_module_sp->GetFileSpec().GetPath().c_str());
916743e4396SJason Molenda       }
91768b3607fSJason Molenda       s->Flush();
91868b3607fSJason Molenda     }
91968b3607fSJason Molenda   }
920c859e2d5SGreg Clayton   return is_loaded;
921c859e2d5SGreg Clayton }
922c859e2d5SGreg Clayton 
923b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::KextImageInfo::GetAddressByteSize() {
924306bd0aaSJason Molenda   if (m_memory_module_sp)
925306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture().GetAddressByteSize();
926306bd0aaSJason Molenda   if (m_module_sp)
927306bd0aaSJason Molenda     return m_module_sp->GetArchitecture().GetAddressByteSize();
9281f746071SGreg Clayton   return 0;
9291f746071SGreg Clayton }
9301f746071SGreg Clayton 
931b9c1b51eSKate Stone lldb::ByteOrder DynamicLoaderDarwinKernel::KextImageInfo::GetByteOrder() {
932306bd0aaSJason Molenda   if (m_memory_module_sp)
933306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture().GetByteOrder();
934306bd0aaSJason Molenda   if (m_module_sp)
935306bd0aaSJason Molenda     return m_module_sp->GetArchitecture().GetByteOrder();
9369ccb970fSBruce Mitchener   return endian::InlHostByteOrder();
9371f746071SGreg Clayton }
9381f746071SGreg Clayton 
9391f746071SGreg Clayton lldb_private::ArchSpec
940b9c1b51eSKate Stone DynamicLoaderDarwinKernel::KextImageInfo::GetArchitecture() const {
941306bd0aaSJason Molenda   if (m_memory_module_sp)
942306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture();
943306bd0aaSJason Molenda   if (m_module_sp)
944306bd0aaSJason Molenda     return m_module_sp->GetArchitecture();
9451f746071SGreg Clayton   return lldb_private::ArchSpec();
9461f746071SGreg Clayton }
9471f746071SGreg Clayton 
948d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
949d4bfbc9aSGreg Clayton // Load the kernel module and initialize the "m_kernel" member. Return
950d4bfbc9aSGreg Clayton // true _only_ if the kernel is loaded the first time through (subsequent
951d4bfbc9aSGreg Clayton // calls to this function should return false after the kernel has been
952d4bfbc9aSGreg Clayton // already loaded).
953d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
954b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::LoadKernelModuleIfNeeded() {
955b9c1b51eSKate Stone   if (!m_kext_summary_header_ptr_addr.IsValid()) {
956306bd0aaSJason Molenda     m_kernel.Clear();
957306bd0aaSJason Molenda     m_kernel.SetModule(m_process->GetTarget().GetExecutableModule());
958306bd0aaSJason Molenda     m_kernel.SetIsKernel(true);
9594bd4e7e3SJason Molenda 
9604bd4e7e3SJason Molenda     ConstString kernel_name("mach_kernel");
961b9c1b51eSKate Stone     if (m_kernel.GetModule().get() && m_kernel.GetModule()->GetObjectFile() &&
962b9c1b51eSKate Stone         !m_kernel.GetModule()
963b9c1b51eSKate Stone              ->GetObjectFile()
964b9c1b51eSKate Stone              ->GetFileSpec()
965b9c1b51eSKate Stone              .GetFilename()
966b9c1b51eSKate Stone              .IsEmpty()) {
967b9c1b51eSKate Stone       kernel_name =
968b9c1b51eSKate Stone           m_kernel.GetModule()->GetObjectFile()->GetFileSpec().GetFilename();
9694bd4e7e3SJason Molenda     }
970306bd0aaSJason Molenda     m_kernel.SetName(kernel_name.AsCString());
9714bd4e7e3SJason Molenda 
972b9c1b51eSKate Stone     if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS) {
973306bd0aaSJason Molenda       m_kernel.SetLoadAddress(m_kernel_load_address);
974b9c1b51eSKate Stone       if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS &&
975b9c1b51eSKate Stone           m_kernel.GetModule()) {
976c859e2d5SGreg Clayton         // We didn't get a hint from the process, so we will
977c859e2d5SGreg Clayton         // try the kernel at the address that it exists at in
978c859e2d5SGreg Clayton         // the file if we have one
979306bd0aaSJason Molenda         ObjectFile *kernel_object_file = m_kernel.GetModule()->GetObjectFile();
980b9c1b51eSKate Stone         if (kernel_object_file) {
981b9c1b51eSKate Stone           addr_t load_address =
982b9c1b51eSKate Stone               kernel_object_file->GetHeaderAddress().GetLoadAddress(
983b9c1b51eSKate Stone                   &m_process->GetTarget());
984b9c1b51eSKate Stone           addr_t file_address =
985b9c1b51eSKate Stone               kernel_object_file->GetHeaderAddress().GetFileAddress();
986b9c1b51eSKate Stone           if (load_address != LLDB_INVALID_ADDRESS && load_address != 0) {
987306bd0aaSJason Molenda             m_kernel.SetLoadAddress(load_address);
988b9c1b51eSKate Stone             if (load_address != file_address) {
9894bd4e7e3SJason Molenda               // Don't accidentally relocate the kernel to the File address --
990b9c1b51eSKate Stone               // the Load address has already been set to its actual in-memory
991b9c1b51eSKate Stone               // address.
9924bd4e7e3SJason Molenda               // Mark it as IsLoaded.
993306bd0aaSJason Molenda               m_kernel.SetProcessStopId(m_process->GetStopID());
9944bd4e7e3SJason Molenda             }
995b9c1b51eSKate Stone           } else {
996306bd0aaSJason Molenda             m_kernel.SetLoadAddress(file_address);
9974bd4e7e3SJason Molenda           }
9984bd4e7e3SJason Molenda         }
999c859e2d5SGreg Clayton       }
1000c859e2d5SGreg Clayton     }
1001c859e2d5SGreg Clayton 
1002b9c1b51eSKate Stone     if (m_kernel.GetLoadAddress() != LLDB_INVALID_ADDRESS) {
1003b9c1b51eSKate Stone       if (!m_kernel.LoadImageUsingMemoryModule(m_process)) {
10042af282a1SGreg Clayton         m_kernel.LoadImageAtFileAddress(m_process);
10052af282a1SGreg Clayton       }
10062af282a1SGreg Clayton     }
1007c859e2d5SGreg Clayton 
1008b9c1b51eSKate Stone     if (m_kernel.IsLoaded() && m_kernel.GetModule()) {
1009d4bfbc9aSGreg Clayton       static ConstString kext_summary_symbol("gLoadedKextSummaries");
1010b9c1b51eSKate Stone       const Symbol *symbol =
1011b9c1b51eSKate Stone           m_kernel.GetModule()->FindFirstSymbolWithNameAndType(
1012b9c1b51eSKate Stone               kext_summary_symbol, eSymbolTypeData);
1013b9c1b51eSKate Stone       if (symbol) {
1014e7612134SGreg Clayton         m_kext_summary_header_ptr_addr = symbol->GetAddress();
1015d4bfbc9aSGreg Clayton         // Update all image infos
1016d4bfbc9aSGreg Clayton         ReadAllKextSummaries();
1017d4bfbc9aSGreg Clayton       }
1018b9c1b51eSKate Stone     } else {
1019306bd0aaSJason Molenda       m_kernel.Clear();
1020d4bfbc9aSGreg Clayton     }
1021d4bfbc9aSGreg Clayton   }
1022d4bfbc9aSGreg Clayton }
1023d4bfbc9aSGreg Clayton 
1024d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1025d4bfbc9aSGreg Clayton // Static callback function that gets called when our DYLD notification
1026d4bfbc9aSGreg Clayton // breakpoint gets hit. We update all of our image infos and then
1027d4bfbc9aSGreg Clayton // let our super class DynamicLoader class decide if we should stop
1028d4bfbc9aSGreg Clayton // or not (based on global preference).
1029d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1030b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::BreakpointHitCallback(
1031b9c1b51eSKate Stone     void *baton, StoppointCallbackContext *context, user_id_t break_id,
1032b9c1b51eSKate Stone     user_id_t break_loc_id) {
1033b9c1b51eSKate Stone   return static_cast<DynamicLoaderDarwinKernel *>(baton)->BreakpointHit(
1034b9c1b51eSKate Stone       context, break_id, break_loc_id);
1035d4bfbc9aSGreg Clayton }
1036d4bfbc9aSGreg Clayton 
1037b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::BreakpointHit(StoppointCallbackContext *context,
1038d4bfbc9aSGreg Clayton                                               user_id_t break_id,
1039b9c1b51eSKate Stone                                               user_id_t break_loc_id) {
10405160ce5cSGreg Clayton   Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
1041d4bfbc9aSGreg Clayton   if (log)
1042944b828aSGreg Clayton     log->Printf("DynamicLoaderDarwinKernel::BreakpointHit (...)\n");
1043d4bfbc9aSGreg Clayton 
1044d4bfbc9aSGreg Clayton   ReadAllKextSummaries();
1045d4bfbc9aSGreg Clayton 
1046d4bfbc9aSGreg Clayton   if (log)
10475160ce5cSGreg Clayton     PutToLog(log);
1048d4bfbc9aSGreg Clayton 
1049d4bfbc9aSGreg Clayton   return GetStopWhenImagesChange();
1050d4bfbc9aSGreg Clayton }
1051d4bfbc9aSGreg Clayton 
1052b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ReadKextSummaryHeader() {
105316ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1054d4bfbc9aSGreg Clayton 
1055d4bfbc9aSGreg Clayton   // the all image infos is already valid for this process stop ID
1056d4bfbc9aSGreg Clayton 
1057b9c1b51eSKate Stone   if (m_kext_summary_header_ptr_addr.IsValid()) {
1058d4bfbc9aSGreg Clayton     const uint32_t addr_size = m_kernel.GetAddressByteSize();
1059d4bfbc9aSGreg Clayton     const ByteOrder byte_order = m_kernel.GetByteOrder();
1060d4bfbc9aSGreg Clayton     Error error;
1061d4bfbc9aSGreg Clayton     // Read enough bytes for a "OSKextLoadedKextSummaryHeader" structure
1062aaa0ba31SBruce Mitchener     // which is currently 4 uint32_t and a pointer.
1063d4bfbc9aSGreg Clayton     uint8_t buf[24];
1064d4bfbc9aSGreg Clayton     DataExtractor data(buf, sizeof(buf), byte_order, addr_size);
1065d4bfbc9aSGreg Clayton     const size_t count = 4 * sizeof(uint32_t) + addr_size;
1066d4bfbc9aSGreg Clayton     const bool prefer_file_cache = false;
1067b9c1b51eSKate Stone     if (m_process->GetTarget().ReadPointerFromMemory(
1068b9c1b51eSKate Stone             m_kext_summary_header_ptr_addr, prefer_file_cache, error,
1069b9c1b51eSKate Stone             m_kext_summary_header_addr)) {
1070b9c1b51eSKate Stone       // We got a valid address for our kext summary header and make sure it
1071b9c1b51eSKate Stone       // isn't NULL
1072d4bfbc9aSGreg Clayton       if (m_kext_summary_header_addr.IsValid() &&
1073b9c1b51eSKate Stone           m_kext_summary_header_addr.GetFileAddress() != 0) {
1074b9c1b51eSKate Stone         const size_t bytes_read = m_process->GetTarget().ReadMemory(
1075b9c1b51eSKate Stone             m_kext_summary_header_addr, prefer_file_cache, buf, count, error);
1076b9c1b51eSKate Stone         if (bytes_read == count) {
1077c7bece56SGreg Clayton           lldb::offset_t offset = 0;
1078d4bfbc9aSGreg Clayton           m_kext_summary_header.version = data.GetU32(&offset);
1079b9c1b51eSKate Stone           if (m_kext_summary_header.version > 128) {
1080b9c1b51eSKate Stone             Stream *s =
1081b9c1b51eSKate Stone                 m_process->GetTarget().GetDebugger().GetOutputFile().get();
1082b9c1b51eSKate Stone             s->Printf("WARNING: Unable to read kext summary header, got "
1083b9c1b51eSKate Stone                       "improbable version number %u\n",
1084b9c1b51eSKate Stone                       m_kext_summary_header.version);
1085b9c1b51eSKate Stone             // If we get an improbably large version number, we're probably
1086b9c1b51eSKate Stone             // getting bad memory.
1087c6fa5db7SJason Molenda             m_kext_summary_header_addr.Clear();
1088c6fa5db7SJason Molenda             return false;
1089c6fa5db7SJason Molenda           }
1090b9c1b51eSKate Stone           if (m_kext_summary_header.version >= 2) {
1091d4bfbc9aSGreg Clayton             m_kext_summary_header.entry_size = data.GetU32(&offset);
1092b9c1b51eSKate Stone             if (m_kext_summary_header.entry_size > 4096) {
1093b9c1b51eSKate Stone               // If we get an improbably large entry_size, we're probably
1094b9c1b51eSKate Stone               // getting bad memory.
1095b9c1b51eSKate Stone               Stream *s =
1096b9c1b51eSKate Stone                   m_process->GetTarget().GetDebugger().GetOutputFile().get();
1097b9c1b51eSKate Stone               s->Printf("WARNING: Unable to read kext summary header, got "
1098b9c1b51eSKate Stone                         "improbable entry_size %u\n",
1099b9c1b51eSKate Stone                         m_kext_summary_header.entry_size);
1100c6fa5db7SJason Molenda               m_kext_summary_header_addr.Clear();
1101c6fa5db7SJason Molenda               return false;
1102c6fa5db7SJason Molenda             }
1103b9c1b51eSKate Stone           } else {
1104d4bfbc9aSGreg Clayton             // Versions less than 2 didn't have an entry size, it was hard coded
1105b9c1b51eSKate Stone             m_kext_summary_header.entry_size =
1106b9c1b51eSKate Stone                 KERNEL_MODULE_ENTRY_SIZE_VERSION_1;
1107d4bfbc9aSGreg Clayton           }
1108d4bfbc9aSGreg Clayton           m_kext_summary_header.entry_count = data.GetU32(&offset);
1109b9c1b51eSKate Stone           if (m_kext_summary_header.entry_count > 10000) {
1110b9c1b51eSKate Stone             // If we get an improbably large number of kexts, we're probably
1111b9c1b51eSKate Stone             // getting bad memory.
1112b9c1b51eSKate Stone             Stream *s =
1113b9c1b51eSKate Stone                 m_process->GetTarget().GetDebugger().GetOutputFile().get();
1114b9c1b51eSKate Stone             s->Printf("WARNING: Unable to read kext summary header, got "
1115b9c1b51eSKate Stone                       "improbable number of kexts %u\n",
1116b9c1b51eSKate Stone                       m_kext_summary_header.entry_count);
1117c6fa5db7SJason Molenda             m_kext_summary_header_addr.Clear();
1118c6fa5db7SJason Molenda             return false;
1119c6fa5db7SJason Molenda           }
1120d4bfbc9aSGreg Clayton           return true;
1121d4bfbc9aSGreg Clayton         }
1122d4bfbc9aSGreg Clayton       }
1123d4bfbc9aSGreg Clayton     }
1124d4bfbc9aSGreg Clayton   }
1125d4bfbc9aSGreg Clayton   m_kext_summary_header_addr.Clear();
1126d4bfbc9aSGreg Clayton   return false;
1127d4bfbc9aSGreg Clayton }
1128d4bfbc9aSGreg Clayton 
1129b9c1b51eSKate Stone // We've either (a) just attached to a new kernel, or (b) the kexts-changed
1130b9c1b51eSKate Stone // breakpoint was hit
1131306bd0aaSJason Molenda // and we need to figure out what kexts have been added or removed.
1132b9c1b51eSKate Stone // Read the kext summaries from the inferior kernel memory, compare them against
1133b9c1b51eSKate Stone // the
1134b9c1b51eSKate Stone // m_known_kexts vector and update the m_known_kexts vector as needed to keep in
1135b9c1b51eSKate Stone // sync with the
1136306bd0aaSJason Molenda // inferior.
1137d4bfbc9aSGreg Clayton 
1138b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ParseKextSummaries(
1139b9c1b51eSKate Stone     const Address &kext_summary_addr, uint32_t count) {
1140306bd0aaSJason Molenda   KextImageInfo::collection kext_summaries;
11415160ce5cSGreg Clayton   Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
1142d4bfbc9aSGreg Clayton   if (log)
1143b9c1b51eSKate Stone     log->Printf("Kexts-changed breakpoint hit, there are %d kexts currently.\n",
1144b9c1b51eSKate Stone                 count);
1145d4bfbc9aSGreg Clayton 
114616ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1147d4bfbc9aSGreg Clayton 
1148d4bfbc9aSGreg Clayton   if (!ReadKextSummaries(kext_summary_addr, count, kext_summaries))
1149d4bfbc9aSGreg Clayton     return false;
1150d4bfbc9aSGreg Clayton 
1151b9c1b51eSKate Stone   // read the plugin.dynamic-loader.darwin-kernel.load-kexts setting -- if the
1152b9c1b51eSKate Stone   // user requested no
1153b9c1b51eSKate Stone   // kext loading, don't print any messages about kexts & don't try to read
1154b9c1b51eSKate Stone   // them.
11554da2e32fSJason Molenda   const bool load_kexts = GetGlobalProperties()->GetLoadKexts();
11564da2e32fSJason Molenda 
1157b9c1b51eSKate Stone   // By default, all kexts we've loaded in the past are marked as "remove" and
1158b9c1b51eSKate Stone   // all of the kexts
1159306bd0aaSJason Molenda   // we just found out about from ReadKextSummaries are marked as "add".
1160306bd0aaSJason Molenda   std::vector<bool> to_be_removed(m_known_kexts.size(), true);
1161306bd0aaSJason Molenda   std::vector<bool> to_be_added(count, true);
1162306bd0aaSJason Molenda 
1163306bd0aaSJason Molenda   int number_of_new_kexts_being_added = 0;
1164306bd0aaSJason Molenda   int number_of_old_kexts_being_removed = m_known_kexts.size();
1165306bd0aaSJason Molenda 
1166306bd0aaSJason Molenda   const uint32_t new_kexts_size = kext_summaries.size();
1167306bd0aaSJason Molenda   const uint32_t old_kexts_size = m_known_kexts.size();
1168306bd0aaSJason Molenda 
1169306bd0aaSJason Molenda   // The m_known_kexts vector may have entries that have been Cleared,
1170306bd0aaSJason Molenda   // or are a kernel.
1171b9c1b51eSKate Stone   for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++) {
1172306bd0aaSJason Molenda     bool ignore = false;
1173306bd0aaSJason Molenda     KextImageInfo &image_info = m_known_kexts[old_kext];
1174b9c1b51eSKate Stone     if (image_info.IsKernel()) {
1175306bd0aaSJason Molenda       ignore = true;
1176b9c1b51eSKate Stone     } else if (image_info.GetLoadAddress() == LLDB_INVALID_ADDRESS &&
1177b9c1b51eSKate Stone                !image_info.GetModule()) {
1178306bd0aaSJason Molenda       ignore = true;
1179306bd0aaSJason Molenda     }
1180306bd0aaSJason Molenda 
1181b9c1b51eSKate Stone     if (ignore) {
1182306bd0aaSJason Molenda       number_of_old_kexts_being_removed--;
1183306bd0aaSJason Molenda       to_be_removed[old_kext] = false;
1184306bd0aaSJason Molenda     }
1185306bd0aaSJason Molenda   }
1186306bd0aaSJason Molenda 
1187306bd0aaSJason Molenda   // Scan over the list of kexts we just read from the kernel, note those that
1188306bd0aaSJason Molenda   // need to be added and those already loaded.
1189b9c1b51eSKate Stone   for (uint32_t new_kext = 0; new_kext < new_kexts_size; new_kext++) {
1190306bd0aaSJason Molenda     bool add_this_one = true;
1191b9c1b51eSKate Stone     for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++) {
1192b9c1b51eSKate Stone       if (m_known_kexts[old_kext] == kext_summaries[new_kext]) {
1193306bd0aaSJason Molenda         // We already have this kext, don't re-load it.
1194306bd0aaSJason Molenda         to_be_added[new_kext] = false;
1195306bd0aaSJason Molenda         // This kext is still present, do not remove it.
1196306bd0aaSJason Molenda         to_be_removed[old_kext] = false;
1197306bd0aaSJason Molenda 
1198306bd0aaSJason Molenda         number_of_old_kexts_being_removed--;
1199306bd0aaSJason Molenda         add_this_one = false;
1200306bd0aaSJason Molenda         break;
1201306bd0aaSJason Molenda       }
1202306bd0aaSJason Molenda     }
1203b9c1b51eSKate Stone     if (add_this_one) {
1204306bd0aaSJason Molenda       number_of_new_kexts_being_added++;
1205306bd0aaSJason Molenda     }
1206306bd0aaSJason Molenda   }
1207306bd0aaSJason Molenda 
1208b9c1b51eSKate Stone   if (number_of_new_kexts_being_added == 0 &&
1209b9c1b51eSKate Stone       number_of_old_kexts_being_removed == 0)
1210306bd0aaSJason Molenda     return true;
1211306bd0aaSJason Molenda 
121244d93782SGreg Clayton   Stream *s = m_process->GetTarget().GetDebugger().GetOutputFile().get();
1213b9c1b51eSKate Stone   if (s && load_kexts) {
1214b9c1b51eSKate Stone     if (number_of_new_kexts_being_added > 0 &&
1215b9c1b51eSKate Stone         number_of_old_kexts_being_removed > 0) {
1216b9c1b51eSKate Stone       s->Printf("Loading %d kext modules and unloading %d kext modules ",
1217b9c1b51eSKate Stone                 number_of_new_kexts_being_added,
1218b9c1b51eSKate Stone                 number_of_old_kexts_being_removed);
1219b9c1b51eSKate Stone     } else if (number_of_new_kexts_being_added > 0) {
1220306bd0aaSJason Molenda       s->Printf("Loading %d kext modules ", number_of_new_kexts_being_added);
1221b9c1b51eSKate Stone     } else if (number_of_old_kexts_being_removed > 0) {
1222b9c1b51eSKate Stone       s->Printf("Unloading %d kext modules ",
1223b9c1b51eSKate Stone                 number_of_old_kexts_being_removed);
1224306bd0aaSJason Molenda     }
1225306bd0aaSJason Molenda   }
1226306bd0aaSJason Molenda 
1227b9c1b51eSKate Stone   if (log) {
1228b9c1b51eSKate Stone     if (load_kexts) {
1229b9c1b51eSKate Stone       log->Printf("DynamicLoaderDarwinKernel::ParseKextSummaries: %d kexts "
1230b9c1b51eSKate Stone                   "added, %d kexts removed",
1231b9c1b51eSKate Stone                   number_of_new_kexts_being_added,
1232b9c1b51eSKate Stone                   number_of_old_kexts_being_removed);
1233b9c1b51eSKate Stone     } else {
1234b9c1b51eSKate Stone       log->Printf(
1235b9c1b51eSKate Stone           "DynamicLoaderDarwinKernel::ParseKextSummaries kext loading is "
1236b9c1b51eSKate Stone           "disabled, else would have %d kexts added, %d kexts removed",
1237b9c1b51eSKate Stone           number_of_new_kexts_being_added, number_of_old_kexts_being_removed);
12384da2e32fSJason Molenda     }
12394da2e32fSJason Molenda   }
12404da2e32fSJason Molenda 
1241b9c1b51eSKate Stone   if (number_of_new_kexts_being_added > 0) {
1242306bd0aaSJason Molenda     ModuleList loaded_module_list;
1243306bd0aaSJason Molenda 
1244306bd0aaSJason Molenda     const uint32_t num_of_new_kexts = kext_summaries.size();
1245b9c1b51eSKate Stone     for (uint32_t new_kext = 0; new_kext < num_of_new_kexts; new_kext++) {
1246b9c1b51eSKate Stone       if (to_be_added[new_kext] == true) {
1247306bd0aaSJason Molenda         KextImageInfo &image_info = kext_summaries[new_kext];
1248b9c1b51eSKate Stone         if (load_kexts) {
1249b9c1b51eSKate Stone           if (!image_info.LoadImageUsingMemoryModule(m_process)) {
1250306bd0aaSJason Molenda             image_info.LoadImageAtFileAddress(m_process);
12514da2e32fSJason Molenda           }
12524da2e32fSJason Molenda         }
1253306bd0aaSJason Molenda 
1254306bd0aaSJason Molenda         m_known_kexts.push_back(image_info);
1255306bd0aaSJason Molenda 
1256b9c1b51eSKate Stone         if (image_info.GetModule() &&
1257b9c1b51eSKate Stone             m_process->GetStopID() == image_info.GetProcessStopId())
1258306bd0aaSJason Molenda           loaded_module_list.AppendIfNeeded(image_info.GetModule());
1259d4bfbc9aSGreg Clayton 
12604da2e32fSJason Molenda         if (s && load_kexts)
126168b3607fSJason Molenda           s->Printf(".");
1262d4bfbc9aSGreg Clayton 
1263d4bfbc9aSGreg Clayton         if (log)
12645160ce5cSGreg Clayton           kext_summaries[new_kext].PutToLog(log);
1265d4bfbc9aSGreg Clayton       }
1266306bd0aaSJason Molenda     }
1267306bd0aaSJason Molenda     m_process->GetTarget().ModulesDidLoad(loaded_module_list);
1268306bd0aaSJason Molenda   }
1269306bd0aaSJason Molenda 
1270b9c1b51eSKate Stone   if (number_of_old_kexts_being_removed > 0) {
1271306bd0aaSJason Molenda     ModuleList loaded_module_list;
1272306bd0aaSJason Molenda     const uint32_t num_of_old_kexts = m_known_kexts.size();
1273b9c1b51eSKate Stone     for (uint32_t old_kext = 0; old_kext < num_of_old_kexts; old_kext++) {
1274306bd0aaSJason Molenda       ModuleList unloaded_module_list;
1275b9c1b51eSKate Stone       if (to_be_removed[old_kext]) {
1276306bd0aaSJason Molenda         KextImageInfo &image_info = m_known_kexts[old_kext];
1277306bd0aaSJason Molenda         // You can't unload the kernel.
1278b9c1b51eSKate Stone         if (!image_info.IsKernel()) {
1279b9c1b51eSKate Stone           if (image_info.GetModule()) {
1280306bd0aaSJason Molenda             unloaded_module_list.AppendIfNeeded(image_info.GetModule());
1281306bd0aaSJason Molenda           }
1282306bd0aaSJason Molenda           if (s)
1283306bd0aaSJason Molenda             s->Printf(".");
1284306bd0aaSJason Molenda           image_info.Clear();
1285b9c1b51eSKate Stone           // should pull it out of the KextImageInfos vector but that would
1286b9c1b51eSKate Stone           // mutate the list and invalidate
1287b9c1b51eSKate Stone           // the to_be_removed bool vector; leaving it in place once Cleared()
1288b9c1b51eSKate Stone           // is relatively harmless.
1289306bd0aaSJason Molenda         }
1290306bd0aaSJason Molenda       }
1291095eeaa0SGreg Clayton       m_process->GetTarget().ModulesDidUnload(unloaded_module_list, false);
1292306bd0aaSJason Molenda     }
1293306bd0aaSJason Molenda   }
1294306bd0aaSJason Molenda 
1295b9c1b51eSKate Stone   if (s && load_kexts) {
129668b3607fSJason Molenda     s->Printf(" done.\n");
129768b3607fSJason Molenda     s->Flush();
129868b3607fSJason Molenda   }
129968b3607fSJason Molenda 
1300306bd0aaSJason Molenda   return true;
1301d4bfbc9aSGreg Clayton }
1302d4bfbc9aSGreg Clayton 
1303b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::ReadKextSummaries(
1304b9c1b51eSKate Stone     const Address &kext_summary_addr, uint32_t image_infos_count,
1305b9c1b51eSKate Stone     KextImageInfo::collection &image_infos) {
1306d4bfbc9aSGreg Clayton   const ByteOrder endian = m_kernel.GetByteOrder();
1307d4bfbc9aSGreg Clayton   const uint32_t addr_size = m_kernel.GetAddressByteSize();
1308d4bfbc9aSGreg Clayton 
1309d4bfbc9aSGreg Clayton   image_infos.resize(image_infos_count);
1310d4bfbc9aSGreg Clayton   const size_t count = image_infos.size() * m_kext_summary_header.entry_size;
1311d4bfbc9aSGreg Clayton   DataBufferHeap data(count, 0);
1312d4bfbc9aSGreg Clayton   Error error;
1313d4bfbc9aSGreg Clayton 
1314d4bfbc9aSGreg Clayton   const bool prefer_file_cache = false;
1315b9c1b51eSKate Stone   const size_t bytes_read = m_process->GetTarget().ReadMemory(
1316b9c1b51eSKate Stone       kext_summary_addr, prefer_file_cache, data.GetBytes(), data.GetByteSize(),
1317d4bfbc9aSGreg Clayton       error);
1318b9c1b51eSKate Stone   if (bytes_read == count) {
1319d4bfbc9aSGreg Clayton 
1320b9c1b51eSKate Stone     DataExtractor extractor(data.GetBytes(), data.GetByteSize(), endian,
1321b9c1b51eSKate Stone                             addr_size);
1322d4bfbc9aSGreg Clayton     uint32_t i = 0;
1323d4bfbc9aSGreg Clayton     for (uint32_t kext_summary_offset = 0;
1324b9c1b51eSKate Stone          i < image_infos.size() &&
1325b9c1b51eSKate Stone          extractor.ValidOffsetForDataOfSize(kext_summary_offset,
1326b9c1b51eSKate Stone                                             m_kext_summary_header.entry_size);
1327b9c1b51eSKate Stone          ++i, kext_summary_offset += m_kext_summary_header.entry_size) {
1328c7bece56SGreg Clayton       lldb::offset_t offset = kext_summary_offset;
1329b9c1b51eSKate Stone       const void *name_data =
1330b9c1b51eSKate Stone           extractor.GetData(&offset, KERNEL_MODULE_MAX_NAME);
1331d4bfbc9aSGreg Clayton       if (name_data == NULL)
1332d4bfbc9aSGreg Clayton         break;
1333306bd0aaSJason Molenda       image_infos[i].SetName((const char *)name_data);
1334306bd0aaSJason Molenda       UUID uuid(extractor.GetData(&offset, 16), 16);
1335306bd0aaSJason Molenda       image_infos[i].SetUUID(uuid);
1336306bd0aaSJason Molenda       image_infos[i].SetLoadAddress(extractor.GetU64(&offset));
1337306bd0aaSJason Molenda       image_infos[i].SetSize(extractor.GetU64(&offset));
1338d4bfbc9aSGreg Clayton     }
1339d4bfbc9aSGreg Clayton     if (i < image_infos.size())
1340d4bfbc9aSGreg Clayton       image_infos.resize(i);
1341b9c1b51eSKate Stone   } else {
1342d4bfbc9aSGreg Clayton     image_infos.clear();
1343d4bfbc9aSGreg Clayton   }
1344d4bfbc9aSGreg Clayton   return image_infos.size();
1345d4bfbc9aSGreg Clayton }
1346d4bfbc9aSGreg Clayton 
1347b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ReadAllKextSummaries() {
134816ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1349d4bfbc9aSGreg Clayton 
1350b9c1b51eSKate Stone   if (ReadKextSummaryHeader()) {
1351b9c1b51eSKate Stone     if (m_kext_summary_header.entry_count > 0 &&
1352b9c1b51eSKate Stone         m_kext_summary_header_addr.IsValid()) {
1353d4bfbc9aSGreg Clayton       Address summary_addr(m_kext_summary_header_addr);
1354d4bfbc9aSGreg Clayton       summary_addr.Slide(m_kext_summary_header.GetSize());
1355b9c1b51eSKate Stone       if (!ParseKextSummaries(summary_addr,
1356b9c1b51eSKate Stone                               m_kext_summary_header.entry_count)) {
1357306bd0aaSJason Molenda         m_known_kexts.clear();
1358d4bfbc9aSGreg Clayton       }
1359d4bfbc9aSGreg Clayton       return true;
1360d4bfbc9aSGreg Clayton     }
1361d4bfbc9aSGreg Clayton   }
1362d4bfbc9aSGreg Clayton   return false;
1363d4bfbc9aSGreg Clayton }
1364d4bfbc9aSGreg Clayton 
1365d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1366d4bfbc9aSGreg Clayton // Dump an image info structure to the file handle provided.
1367d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1368b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::PutToLog(Log *log) const {
1369d4bfbc9aSGreg Clayton   if (log == NULL)
1370d4bfbc9aSGreg Clayton     return;
1371306bd0aaSJason Molenda   const uint8_t *u = (uint8_t *)m_uuid.GetBytes();
1372d4bfbc9aSGreg Clayton 
1373b9c1b51eSKate Stone   if (m_load_address == LLDB_INVALID_ADDRESS) {
1374b9c1b51eSKate Stone     if (u) {
1375b9c1b51eSKate Stone       log->Printf("\tuuid=%2.2X%2.2X%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2."
1376b9c1b51eSKate Stone                   "2X-%2.2X%2.2X%2.2X%2.2X%2.2X%2.2X name=\"%s\" (UNLOADED)",
1377b9c1b51eSKate Stone                   u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7], u[8], u[9],
1378b9c1b51eSKate Stone                   u[10], u[11], u[12], u[13], u[14], u[15], m_name.c_str());
1379b9c1b51eSKate Stone     } else
1380306bd0aaSJason Molenda       log->Printf("\tname=\"%s\" (UNLOADED)", m_name.c_str());
1381b9c1b51eSKate Stone   } else {
1382b9c1b51eSKate Stone     if (u) {
1383b9c1b51eSKate Stone       log->Printf("\taddr=0x%16.16" PRIx64 " size=0x%16.16" PRIx64
1384b9c1b51eSKate Stone                   " uuid=%2.2X%2.2X%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-"
1385b9c1b51eSKate Stone                   "%2.2X%2.2X%2.2X%2.2X%2.2X%2.2X name=\"%s\"",
1386b9c1b51eSKate Stone                   m_load_address, m_size, u[0], u[1], u[2], u[3], u[4], u[5],
1387b9c1b51eSKate Stone                   u[6], u[7], u[8], u[9], u[10], u[11], u[12], u[13], u[14],
1388b9c1b51eSKate Stone                   u[15], m_name.c_str());
1389b9c1b51eSKate Stone     } else {
1390306bd0aaSJason Molenda       log->Printf("\t[0x%16.16" PRIx64 " - 0x%16.16" PRIx64 ") name=\"%s\"",
1391306bd0aaSJason Molenda                   m_load_address, m_load_address + m_size, m_name.c_str());
1392d4bfbc9aSGreg Clayton     }
1393d4bfbc9aSGreg Clayton   }
1394d4bfbc9aSGreg Clayton }
1395d4bfbc9aSGreg Clayton 
1396d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1397d4bfbc9aSGreg Clayton // Dump the _dyld_all_image_infos members and all current image infos
1398d4bfbc9aSGreg Clayton // that we have parsed to the file handle provided.
1399d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1400b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PutToLog(Log *log) const {
1401d4bfbc9aSGreg Clayton   if (log == NULL)
1402d4bfbc9aSGreg Clayton     return;
1403d4bfbc9aSGreg Clayton 
140416ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1405b9c1b51eSKate Stone   log->Printf("gLoadedKextSummaries = 0x%16.16" PRIx64
1406b9c1b51eSKate Stone               " { version=%u, entry_size=%u, entry_count=%u }",
1407d4bfbc9aSGreg Clayton               m_kext_summary_header_addr.GetFileAddress(),
1408b9c1b51eSKate Stone               m_kext_summary_header.version, m_kext_summary_header.entry_size,
1409d4bfbc9aSGreg Clayton               m_kext_summary_header.entry_count);
1410d4bfbc9aSGreg Clayton 
1411d4bfbc9aSGreg Clayton   size_t i;
1412306bd0aaSJason Molenda   const size_t count = m_known_kexts.size();
1413b9c1b51eSKate Stone   if (count > 0) {
1414d4bfbc9aSGreg Clayton     log->PutCString("Loaded:");
1415d4bfbc9aSGreg Clayton     for (i = 0; i < count; i++)
1416306bd0aaSJason Molenda       m_known_kexts[i].PutToLog(log);
1417d4bfbc9aSGreg Clayton   }
1418d4bfbc9aSGreg Clayton }
1419d4bfbc9aSGreg Clayton 
1420b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PrivateInitialize(Process *process) {
1421b9c1b51eSKate Stone   DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n",
1422b9c1b51eSKate Stone                __FUNCTION__, StateAsCString(m_process->GetState()));
1423d4bfbc9aSGreg Clayton   Clear(true);
1424d4bfbc9aSGreg Clayton   m_process = process;
1425d4bfbc9aSGreg Clayton }
1426d4bfbc9aSGreg Clayton 
1427b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::SetNotificationBreakpointIfNeeded() {
1428b9c1b51eSKate Stone   if (m_break_id == LLDB_INVALID_BREAK_ID && m_kernel.GetModule()) {
1429b9c1b51eSKate Stone     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n",
1430b9c1b51eSKate Stone                  __FUNCTION__, StateAsCString(m_process->GetState()));
1431d4bfbc9aSGreg Clayton 
1432a8558b62SJim Ingham     const bool internal_bp = true;
1433eb023e75SGreg Clayton     const bool hardware = false;
1434d4bfbc9aSGreg Clayton     const LazyBool skip_prologue = eLazyBoolNo;
1435969795f1SJim Ingham     FileSpecList module_spec_list;
1436306bd0aaSJason Molenda     module_spec_list.Append(m_kernel.GetModule()->GetFileSpec());
1437b9c1b51eSKate Stone     Breakpoint *bp =
1438b9c1b51eSKate Stone         m_process->GetTarget()
1439b9c1b51eSKate Stone             .CreateBreakpoint(&module_spec_list, NULL,
1440d4bfbc9aSGreg Clayton                               "OSKextLoadedKextSummariesUpdated",
1441b9c1b51eSKate Stone                               eFunctionNameTypeFull, eLanguageTypeUnknown, 0,
1442b9c1b51eSKate Stone                               skip_prologue, internal_bp, hardware)
1443b9c1b51eSKate Stone             .get();
1444d4bfbc9aSGreg Clayton 
1445b9c1b51eSKate Stone     bp->SetCallback(DynamicLoaderDarwinKernel::BreakpointHitCallback, this,
1446b9c1b51eSKate Stone                     true);
1447d4bfbc9aSGreg Clayton     m_break_id = bp->GetID();
1448d4bfbc9aSGreg Clayton   }
1449d4bfbc9aSGreg Clayton }
1450d4bfbc9aSGreg Clayton 
1451d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1452d4bfbc9aSGreg Clayton // Member function that gets called when the process state changes.
1453d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1454b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PrivateProcessStateChanged(Process *process,
1455b9c1b51eSKate Stone                                                            StateType state) {
1456b9c1b51eSKate Stone   DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s(%s)\n", __FUNCTION__,
1457b9c1b51eSKate Stone                StateAsCString(state));
1458b9c1b51eSKate Stone   switch (state) {
1459d4bfbc9aSGreg Clayton   case eStateConnected:
1460d4bfbc9aSGreg Clayton   case eStateAttaching:
1461d4bfbc9aSGreg Clayton   case eStateLaunching:
1462d4bfbc9aSGreg Clayton   case eStateInvalid:
1463d4bfbc9aSGreg Clayton   case eStateUnloaded:
1464d4bfbc9aSGreg Clayton   case eStateExited:
1465d4bfbc9aSGreg Clayton   case eStateDetached:
1466d4bfbc9aSGreg Clayton     Clear(false);
1467d4bfbc9aSGreg Clayton     break;
1468d4bfbc9aSGreg Clayton 
1469d4bfbc9aSGreg Clayton   case eStateStopped:
1470d4bfbc9aSGreg Clayton     UpdateIfNeeded();
1471d4bfbc9aSGreg Clayton     break;
1472d4bfbc9aSGreg Clayton 
1473d4bfbc9aSGreg Clayton   case eStateRunning:
1474d4bfbc9aSGreg Clayton   case eStateStepping:
1475d4bfbc9aSGreg Clayton   case eStateCrashed:
1476d4bfbc9aSGreg Clayton   case eStateSuspended:
1477d4bfbc9aSGreg Clayton     break;
1478d4bfbc9aSGreg Clayton   }
1479d4bfbc9aSGreg Clayton }
1480d4bfbc9aSGreg Clayton 
1481d4bfbc9aSGreg Clayton ThreadPlanSP
1482b9c1b51eSKate Stone DynamicLoaderDarwinKernel::GetStepThroughTrampolinePlan(Thread &thread,
1483b9c1b51eSKate Stone                                                         bool stop_others) {
1484d4bfbc9aSGreg Clayton   ThreadPlanSP thread_plan_sp;
14855160ce5cSGreg Clayton   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_STEP));
1486d4bfbc9aSGreg Clayton   if (log)
1487d4bfbc9aSGreg Clayton     log->Printf("Could not find symbol for step through.");
1488d4bfbc9aSGreg Clayton   return thread_plan_sp;
1489d4bfbc9aSGreg Clayton }
1490d4bfbc9aSGreg Clayton 
1491b9c1b51eSKate Stone Error DynamicLoaderDarwinKernel::CanLoadImage() {
1492d4bfbc9aSGreg Clayton   Error error;
1493b9c1b51eSKate Stone   error.SetErrorString(
1494b9c1b51eSKate Stone       "always unsafe to load or unload shared libraries in the darwin kernel");
1495d4bfbc9aSGreg Clayton   return error;
1496d4bfbc9aSGreg Clayton }
1497d4bfbc9aSGreg Clayton 
1498b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Initialize() {
1499d4bfbc9aSGreg Clayton   PluginManager::RegisterPlugin(GetPluginNameStatic(),
1500b9c1b51eSKate Stone                                 GetPluginDescriptionStatic(), CreateInstance,
1501e8cd0c98SGreg Clayton                                 DebuggerInitialize);
1502d4bfbc9aSGreg Clayton }
1503d4bfbc9aSGreg Clayton 
1504b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Terminate() {
1505d4bfbc9aSGreg Clayton   PluginManager::UnregisterPlugin(CreateInstance);
1506d4bfbc9aSGreg Clayton }
1507d4bfbc9aSGreg Clayton 
1508b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DebuggerInitialize(
1509b9c1b51eSKate Stone     lldb_private::Debugger &debugger) {
1510b9c1b51eSKate Stone   if (!PluginManager::GetSettingForDynamicLoaderPlugin(
1511b9c1b51eSKate Stone           debugger, DynamicLoaderDarwinKernelProperties::GetSettingName())) {
1512e8cd0c98SGreg Clayton     const bool is_global_setting = true;
1513b9c1b51eSKate Stone     PluginManager::CreateSettingForDynamicLoaderPlugin(
1514b9c1b51eSKate Stone         debugger, GetGlobalProperties()->GetValueProperties(),
1515e8cd0c98SGreg Clayton         ConstString("Properties for the DynamicLoaderDarwinKernel plug-in."),
1516e8cd0c98SGreg Clayton         is_global_setting);
1517e8cd0c98SGreg Clayton   }
1518e8cd0c98SGreg Clayton }
1519d4bfbc9aSGreg Clayton 
1520b9c1b51eSKate Stone lldb_private::ConstString DynamicLoaderDarwinKernel::GetPluginNameStatic() {
152157abc5d6SGreg Clayton   static ConstString g_name("darwin-kernel");
152257abc5d6SGreg Clayton   return g_name;
1523d4bfbc9aSGreg Clayton }
1524d4bfbc9aSGreg Clayton 
1525b9c1b51eSKate Stone const char *DynamicLoaderDarwinKernel::GetPluginDescriptionStatic() {
1526b9c1b51eSKate Stone   return "Dynamic loader plug-in that watches for shared library loads/unloads "
1527b9c1b51eSKate Stone          "in the MacOSX kernel.";
1528d4bfbc9aSGreg Clayton }
1529d4bfbc9aSGreg Clayton 
1530d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1531d4bfbc9aSGreg Clayton // PluginInterface protocol
1532d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1533b9c1b51eSKate Stone lldb_private::ConstString DynamicLoaderDarwinKernel::GetPluginName() {
1534d4bfbc9aSGreg Clayton   return GetPluginNameStatic();
1535d4bfbc9aSGreg Clayton }
1536d4bfbc9aSGreg Clayton 
1537b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::GetPluginVersion() { return 1; }
1538d4bfbc9aSGreg Clayton 
15391f746071SGreg Clayton lldb::ByteOrder
1540b9c1b51eSKate Stone DynamicLoaderDarwinKernel::GetByteOrderFromMagic(uint32_t magic) {
1541b9c1b51eSKate Stone   switch (magic) {
1542510938e5SCharles Davis   case llvm::MachO::MH_MAGIC:
1543510938e5SCharles Davis   case llvm::MachO::MH_MAGIC_64:
15449ccb970fSBruce Mitchener     return endian::InlHostByteOrder();
15451f746071SGreg Clayton 
1546510938e5SCharles Davis   case llvm::MachO::MH_CIGAM:
1547510938e5SCharles Davis   case llvm::MachO::MH_CIGAM_64:
15489ccb970fSBruce Mitchener     if (endian::InlHostByteOrder() == lldb::eByteOrderBig)
15491f746071SGreg Clayton       return lldb::eByteOrderLittle;
15501f746071SGreg Clayton     else
15511f746071SGreg Clayton       return lldb::eByteOrderBig;
15521f746071SGreg Clayton 
15531f746071SGreg Clayton   default:
15541f746071SGreg Clayton     break;
15551f746071SGreg Clayton   }
15561f746071SGreg Clayton   return lldb::eByteOrderInvalid;
15571f746071SGreg Clayton }
1558