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 
11b9c1b51eSKate Stone #include "Plugins/Platform/MacOSX/PlatformDarwinKernel.h"
12d4bfbc9aSGreg Clayton #include "lldb/Breakpoint/StoppointCallbackContext.h"
13d4bfbc9aSGreg Clayton #include "lldb/Core/Debugger.h"
14d4bfbc9aSGreg Clayton #include "lldb/Core/Module.h"
151f746071SGreg Clayton #include "lldb/Core/ModuleSpec.h"
16d4bfbc9aSGreg Clayton #include "lldb/Core/PluginManager.h"
171f746071SGreg Clayton #include "lldb/Core/Section.h"
1844d93782SGreg Clayton #include "lldb/Core/StreamFile.h"
1968b3607fSJason Molenda #include "lldb/Host/Symbols.h"
2041204d09SIlia K #include "lldb/Interpreter/OptionValueProperties.h"
21d4bfbc9aSGreg Clayton #include "lldb/Symbol/ObjectFile.h"
22519b0816SJim Ingham #include "lldb/Target/OperatingSystem.h"
23d4bfbc9aSGreg Clayton #include "lldb/Target/RegisterContext.h"
24b57e4a1bSJason Molenda #include "lldb/Target/StackFrame.h"
25d4bfbc9aSGreg Clayton #include "lldb/Target/Target.h"
26d4bfbc9aSGreg Clayton #include "lldb/Target/Thread.h"
27d4bfbc9aSGreg Clayton #include "lldb/Target/ThreadPlanRunToAddress.h"
28666cc0b2SZachary Turner #include "lldb/Utility/DataBuffer.h"
29666cc0b2SZachary Turner #include "lldb/Utility/DataBufferHeap.h"
306f9e6901SZachary Turner #include "lldb/Utility/Log.h"
31d821c997SPavel Labath #include "lldb/Utility/State.h"
32d4bfbc9aSGreg Clayton 
33944b828aSGreg Clayton #include "DynamicLoaderDarwinKernel.h"
34d4bfbc9aSGreg Clayton 
35d4bfbc9aSGreg Clayton //#define ENABLE_DEBUG_PRINTF // COMMENT THIS LINE OUT PRIOR TO CHECKIN
36d4bfbc9aSGreg Clayton #ifdef ENABLE_DEBUG_PRINTF
37d4bfbc9aSGreg Clayton #include <stdio.h>
38d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...) printf(fmt, ##__VA_ARGS__)
39d4bfbc9aSGreg Clayton #else
40d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...)
41d4bfbc9aSGreg Clayton #endif
42d4bfbc9aSGreg Clayton 
43d4bfbc9aSGreg Clayton using namespace lldb;
44d4bfbc9aSGreg Clayton using namespace lldb_private;
45d4bfbc9aSGreg Clayton 
4605097246SAdrian Prantl // Progressively greater amounts of scanning we will allow For some targets
4705097246SAdrian Prantl // very early in startup, we can't do any random reads of memory or we can
4805097246SAdrian Prantl // crash the device so a setting is needed that can completely disable the
4905097246SAdrian Prantl // KASLR scans.
506ba6d3d1SJason Molenda 
51b9c1b51eSKate Stone enum KASLRScanType {
526ba6d3d1SJason Molenda   eKASLRScanNone = 0,        // No reading into the inferior at all
53b9c1b51eSKate Stone   eKASLRScanLowgloAddresses, // Check one word of memory for a possible kernel
54b9c1b51eSKate Stone                              // addr, then see if a kernel is there
55b9c1b51eSKate Stone   eKASLRScanNearPC, // Scan backwards from the current $pc looking for kernel;
56b9c1b51eSKate Stone                     // checking at 96 locations total
57b9c1b51eSKate Stone   eKASLRScanExhaustiveScan // Scan through the entire possible kernel address
58b9c1b51eSKate Stone                            // range looking for a kernel
596ba6d3d1SJason Molenda };
606ba6d3d1SJason Molenda 
618fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_kaslr_kernel_scan_enum_values[] = {
62b9c1b51eSKate Stone     {eKASLRScanNone, "none",
63b9c1b51eSKate Stone      "Do not read memory looking for a Darwin kernel when attaching."},
64b9c1b51eSKate Stone     {eKASLRScanLowgloAddresses, "basic", "Check for the Darwin kernel's load "
65b9c1b51eSKate Stone                                          "addr in the lowglo page "
66b9c1b51eSKate Stone                                          "(boot-args=debug) only."},
67b9c1b51eSKate Stone     {eKASLRScanNearPC, "fast-scan", "Scan near the pc value on attach to find "
68b9c1b51eSKate Stone                                     "the Darwin kernel's load address."},
69b9c1b51eSKate Stone     {eKASLRScanExhaustiveScan, "exhaustive-scan",
70b9c1b51eSKate Stone      "Scan through the entire potential address range of Darwin kernel (only "
718fe53c49STatyana Krasnukha      "on 32-bit targets)."}};
726ba6d3d1SJason Molenda 
738fe53c49STatyana Krasnukha static constexpr PropertyDefinition g_properties[] = {
748fe53c49STatyana Krasnukha     {"load-kexts", OptionValue::eTypeBoolean, true, true, NULL, {},
75b9c1b51eSKate Stone      "Automatically loads kext images when attaching to a kernel."},
76b9c1b51eSKate Stone     {"scan-type", OptionValue::eTypeEnum, true, eKASLRScanNearPC, NULL,
778fe53c49STatyana Krasnukha      OptionEnumValues(g_kaslr_kernel_scan_enum_values),
788fe53c49STatyana Krasnukha      "Control how many reads lldb will make while searching for a Darwin "
79e40db05bSTatyana Krasnukha      "kernel on attach."}};
80e8cd0c98SGreg Clayton 
81b9c1b51eSKate Stone enum { ePropertyLoadKexts, ePropertyScanType };
82e8cd0c98SGreg Clayton 
83b9c1b51eSKate Stone class DynamicLoaderDarwinKernelProperties : public Properties {
84e8cd0c98SGreg Clayton public:
85b9c1b51eSKate Stone   static ConstString &GetSettingName() {
86468ea4ebSGreg Clayton     static ConstString g_setting_name("darwin-kernel");
87e8cd0c98SGreg Clayton     return g_setting_name;
88e8cd0c98SGreg Clayton   }
89e8cd0c98SGreg Clayton 
90b9c1b51eSKate Stone   DynamicLoaderDarwinKernelProperties() : Properties() {
91e8cd0c98SGreg Clayton     m_collection_sp.reset(new OptionValueProperties(GetSettingName()));
92e8cd0c98SGreg Clayton     m_collection_sp->Initialize(g_properties);
93e8cd0c98SGreg Clayton   }
94e8cd0c98SGreg Clayton 
95b9c1b51eSKate Stone   virtual ~DynamicLoaderDarwinKernelProperties() {}
96e8cd0c98SGreg Clayton 
97b9c1b51eSKate Stone   bool GetLoadKexts() const {
9866763eedSGreg Clayton     const uint32_t idx = ePropertyLoadKexts;
99b9c1b51eSKate Stone     return m_collection_sp->GetPropertyAtIndexAsBoolean(
100b9c1b51eSKate Stone         NULL, idx, g_properties[idx].default_uint_value != 0);
101e8cd0c98SGreg Clayton   }
102e8cd0c98SGreg Clayton 
103b9c1b51eSKate Stone   KASLRScanType GetScanType() const {
1046ba6d3d1SJason Molenda     const uint32_t idx = ePropertyScanType;
105b9c1b51eSKate Stone     return (KASLRScanType)m_collection_sp->GetPropertyAtIndexAsEnumeration(
106b9c1b51eSKate Stone         NULL, idx, g_properties[idx].default_uint_value);
1076ba6d3d1SJason Molenda   }
108e8cd0c98SGreg Clayton };
109e8cd0c98SGreg Clayton 
110b9c1b51eSKate Stone typedef std::shared_ptr<DynamicLoaderDarwinKernelProperties>
111b9c1b51eSKate Stone     DynamicLoaderDarwinKernelPropertiesSP;
112e8cd0c98SGreg Clayton 
113b9c1b51eSKate Stone static const DynamicLoaderDarwinKernelPropertiesSP &GetGlobalProperties() {
114e8cd0c98SGreg Clayton   static DynamicLoaderDarwinKernelPropertiesSP g_settings_sp;
115e8cd0c98SGreg Clayton   if (!g_settings_sp)
116e8cd0c98SGreg Clayton     g_settings_sp.reset(new DynamicLoaderDarwinKernelProperties());
117e8cd0c98SGreg Clayton   return g_settings_sp;
118e8cd0c98SGreg Clayton }
119e8cd0c98SGreg Clayton 
120d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
12105097246SAdrian Prantl // Create an instance of this class. This function is filled into the plugin
12205097246SAdrian Prantl // info class that gets handed out by the plugin factory and allows the lldb to
12305097246SAdrian Prantl // instantiate an instance of this class.
124d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
125b9c1b51eSKate Stone DynamicLoader *DynamicLoaderDarwinKernel::CreateInstance(Process *process,
126b9c1b51eSKate Stone                                                          bool force) {
127b9c1b51eSKate Stone   if (!force) {
12805097246SAdrian Prantl     // If the user provided an executable binary and it is not a kernel, this
12905097246SAdrian Prantl     // plugin should not create an instance.
130d4bfbc9aSGreg Clayton     Module *exe_module = process->GetTarget().GetExecutableModulePointer();
131b9c1b51eSKate Stone     if (exe_module) {
132d4bfbc9aSGreg Clayton       ObjectFile *object_file = exe_module->GetObjectFile();
133b9c1b51eSKate Stone       if (object_file) {
134b9c1b51eSKate Stone         if (object_file->GetStrata() != ObjectFile::eStrataKernel) {
1356ba6d3d1SJason Molenda           return NULL;
1366ba6d3d1SJason Molenda         }
137d4bfbc9aSGreg Clayton       }
138d4bfbc9aSGreg Clayton     }
139d4bfbc9aSGreg Clayton 
1406ba6d3d1SJason Molenda     // If the target's architecture does not look like an Apple environment,
1416ba6d3d1SJason Molenda     // this plugin should not create an instance.
142b9c1b51eSKate Stone     const llvm::Triple &triple_ref =
143b9c1b51eSKate Stone         process->GetTarget().GetArchitecture().GetTriple();
144b9c1b51eSKate Stone     switch (triple_ref.getOS()) {
14570512317SGreg Clayton     case llvm::Triple::Darwin:
14670512317SGreg Clayton     case llvm::Triple::MacOSX:
14770512317SGreg Clayton     case llvm::Triple::IOS:
148a814f704SJason Molenda     case llvm::Triple::TvOS:
149a814f704SJason Molenda     case llvm::Triple::WatchOS:
15032762fd2SJason Molenda     // NEED_BRIDGEOS_TRIPLE case llvm::Triple::BridgeOS:
151b9c1b51eSKate Stone       if (triple_ref.getVendor() != llvm::Triple::Apple) {
1526ba6d3d1SJason Molenda         return NULL;
1536ba6d3d1SJason Molenda       }
1546ba6d3d1SJason Molenda       break;
15505097246SAdrian Prantl     // If we have triple like armv7-unknown-unknown, we should try looking for
15605097246SAdrian Prantl     // a Darwin kernel.
1576ba6d3d1SJason Molenda     case llvm::Triple::UnknownOS:
15870512317SGreg Clayton       break;
15970512317SGreg Clayton     default:
1606ba6d3d1SJason Molenda       return NULL;
16170512317SGreg Clayton       break;
16270512317SGreg Clayton     }
163d4bfbc9aSGreg Clayton   }
1646ba6d3d1SJason Molenda 
165b9c1b51eSKate Stone   // At this point if there is an ExecutableModule, it is a kernel and the
16605097246SAdrian Prantl   // Target is some variant of an Apple system. If the Process hasn't provided
16705097246SAdrian Prantl   // the kernel load address, we need to look around in memory to find it.
1686ba6d3d1SJason Molenda 
169c44bcec6SGreg Clayton   const addr_t kernel_load_address = SearchForDarwinKernel(process);
170b9c1b51eSKate Stone   if (CheckForKernelImageAtAddress(kernel_load_address, process).IsValid()) {
17190ff7911SEwan Crawford     process->SetCanRunCode(false);
172503d0181SJason Molenda     return new DynamicLoaderDarwinKernel(process, kernel_load_address);
173503d0181SJason Molenda   }
174503d0181SJason Molenda   return NULL;
175503d0181SJason Molenda }
176503d0181SJason Molenda 
177503d0181SJason Molenda lldb::addr_t
178b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForDarwinKernel(Process *process) {
1796ba6d3d1SJason Molenda   addr_t kernel_load_address = process->GetImageInfoAddress();
180b9c1b51eSKate Stone   if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1816ba6d3d1SJason Molenda     kernel_load_address = SearchForKernelAtSameLoadAddr(process);
182b9c1b51eSKate Stone     if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1836ba6d3d1SJason Molenda       kernel_load_address = SearchForKernelWithDebugHints(process);
184b9c1b51eSKate Stone       if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1856ba6d3d1SJason Molenda         kernel_load_address = SearchForKernelNearPC(process);
186b9c1b51eSKate Stone         if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1876ba6d3d1SJason Molenda           kernel_load_address = SearchForKernelViaExhaustiveSearch(process);
1886ba6d3d1SJason Molenda         }
1896ba6d3d1SJason Molenda       }
1906ba6d3d1SJason Molenda     }
191d4bfbc9aSGreg Clayton   }
192503d0181SJason Molenda   return kernel_load_address;
193d4bfbc9aSGreg Clayton }
194d4bfbc9aSGreg Clayton 
195d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
19605097246SAdrian Prantl // Check if the kernel binary is loaded in memory without a slide. First verify
19705097246SAdrian Prantl // that the ExecutableModule is a kernel before we proceed. Returns the address
19805097246SAdrian Prantl // of the kernel if one was found, else LLDB_INVALID_ADDRESS.
1996ba6d3d1SJason Molenda //----------------------------------------------------------------------
2006ba6d3d1SJason Molenda lldb::addr_t
201b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelAtSameLoadAddr(Process *process) {
2026ba6d3d1SJason Molenda   Module *exe_module = process->GetTarget().GetExecutableModulePointer();
2036ba6d3d1SJason Molenda   if (exe_module == NULL)
2046ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2056ba6d3d1SJason Molenda 
2066ba6d3d1SJason Molenda   ObjectFile *exe_objfile = exe_module->GetObjectFile();
2076ba6d3d1SJason Molenda   if (exe_objfile == NULL)
2086ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2096ba6d3d1SJason Molenda 
210b9c1b51eSKate Stone   if (exe_objfile->GetType() != ObjectFile::eTypeExecutable ||
211b9c1b51eSKate Stone       exe_objfile->GetStrata() != ObjectFile::eStrataKernel)
2126ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2136ba6d3d1SJason Molenda 
2146ba6d3d1SJason Molenda   if (!exe_objfile->GetHeaderAddress().IsValid())
2156ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2166ba6d3d1SJason Molenda 
217b9c1b51eSKate Stone   if (CheckForKernelImageAtAddress(
218b9c1b51eSKate Stone           exe_objfile->GetHeaderAddress().GetFileAddress(), process) ==
219b9c1b51eSKate Stone       exe_module->GetUUID())
2206ba6d3d1SJason Molenda     return exe_objfile->GetHeaderAddress().GetFileAddress();
2216ba6d3d1SJason Molenda 
2226ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
2236ba6d3d1SJason Molenda }
2246ba6d3d1SJason Molenda 
2256ba6d3d1SJason Molenda //----------------------------------------------------------------------
226b9c1b51eSKate Stone // If the debug flag is included in the boot-args nvram setting, the kernel's
22705097246SAdrian Prantl // load address will be noted in the lowglo page at a fixed address Returns the
22805097246SAdrian Prantl // address of the kernel if one was found, else LLDB_INVALID_ADDRESS.
2296ba6d3d1SJason Molenda //----------------------------------------------------------------------
2306ba6d3d1SJason Molenda lldb::addr_t
231b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelWithDebugHints(Process *process) {
2326ba6d3d1SJason Molenda   if (GetGlobalProperties()->GetScanType() == eKASLRScanNone)
2336ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2346ba6d3d1SJason Molenda 
23597206d57SZachary Turner   Status read_err;
236b9c1b51eSKate Stone   addr_t kernel_addresses_64[] = {
237b9c1b51eSKate Stone       0xfffffff000004010ULL, // newest arm64 devices
23857086ed7SJason Molenda       0xffffff8000004010ULL, // 2014-2015-ish arm64 devices
23957086ed7SJason Molenda       0xffffff8000002010ULL, // oldest arm64 devices
240880988afSJason Molenda       LLDB_INVALID_ADDRESS};
24161a8e53bSJason Molenda   addr_t kernel_addresses_32[] = {0xffff0110, // 2016 and earlier armv7 devices
24261a8e53bSJason Molenda                                   0xffff1010,
24361a8e53bSJason Molenda                                   LLDB_INVALID_ADDRESS};
2442b0a7be9SJason Molenda 
2452b0a7be9SJason Molenda   uint8_t uval[8];
2462b0a7be9SJason Molenda   if (process->GetAddressByteSize() == 8) {
247b9c1b51eSKate Stone   for (size_t i = 0; kernel_addresses_64[i] != LLDB_INVALID_ADDRESS; i++) {
2482b0a7be9SJason Molenda       if (process->ReadMemoryFromInferior (kernel_addresses_64[i], uval, 8, read_err) == 8)
2492b0a7be9SJason Molenda       {
2502b0a7be9SJason Molenda           DataExtractor data (&uval, 8, process->GetByteOrder(), process->GetAddressByteSize());
2512b0a7be9SJason Molenda           offset_t offset = 0;
2522b0a7be9SJason Molenda           uint64_t addr = data.GetU64 (&offset);
253b9c1b51eSKate Stone           if (CheckForKernelImageAtAddress(addr, process).IsValid()) {
254ec504238SJason Molenda               return addr;
255ec504238SJason Molenda           }
2566ba6d3d1SJason Molenda       }
2572b0a7be9SJason Molenda   }
2582b0a7be9SJason Molenda   }
259880988afSJason Molenda 
2602b0a7be9SJason Molenda   if (process->GetAddressByteSize() == 4) {
261b9c1b51eSKate Stone   for (size_t i = 0; kernel_addresses_32[i] != LLDB_INVALID_ADDRESS; i++) {
2622b0a7be9SJason Molenda       if (process->ReadMemoryFromInferior (kernel_addresses_32[i], uval, 4, read_err) == 4)
2632b0a7be9SJason Molenda       {
2642b0a7be9SJason Molenda           DataExtractor data (&uval, 4, process->GetByteOrder(), process->GetAddressByteSize());
2652b0a7be9SJason Molenda           offset_t offset = 0;
2662b0a7be9SJason Molenda           uint32_t addr = data.GetU32 (&offset);
267b9c1b51eSKate Stone           if (CheckForKernelImageAtAddress(addr, process).IsValid()) {
2686ba6d3d1SJason Molenda               return addr;
2696ba6d3d1SJason Molenda           }
2707dd29390SJason Molenda       }
2712b0a7be9SJason Molenda   }
2722b0a7be9SJason Molenda   }
273880988afSJason Molenda 
2746ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
2756ba6d3d1SJason Molenda }
2766ba6d3d1SJason Molenda 
2776ba6d3d1SJason Molenda //----------------------------------------------------------------------
27805097246SAdrian Prantl // If the kernel is currently executing when lldb attaches, and we don't have a
27905097246SAdrian Prantl // better way of finding the kernel's load address, try searching backwards
2806ba6d3d1SJason Molenda // from the current pc value looking for the kernel's Mach header in memory.
281b9c1b51eSKate Stone // Returns the address of the kernel if one was found, else
282b9c1b51eSKate Stone // LLDB_INVALID_ADDRESS.
2836ba6d3d1SJason Molenda //----------------------------------------------------------------------
2846ba6d3d1SJason Molenda lldb::addr_t
285b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelNearPC(Process *process) {
286b9c1b51eSKate Stone   if (GetGlobalProperties()->GetScanType() == eKASLRScanNone ||
287b9c1b51eSKate Stone       GetGlobalProperties()->GetScanType() == eKASLRScanLowgloAddresses) {
2886ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2896ba6d3d1SJason Molenda   }
2906ba6d3d1SJason Molenda 
2916ba6d3d1SJason Molenda   ThreadSP thread = process->GetThreadList().GetSelectedThread();
2926ba6d3d1SJason Molenda   if (thread.get() == NULL)
2936ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2946ba6d3d1SJason Molenda   addr_t pc = thread->GetRegisterContext()->GetPC(LLDB_INVALID_ADDRESS);
2956ba6d3d1SJason Molenda 
2966ba6d3d1SJason Molenda   if (pc == LLDB_INVALID_ADDRESS)
2976ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2986ba6d3d1SJason Molenda 
299b9c1b51eSKate Stone   // The kernel will load at at one megabyte boundary (0x100000), or at that
30005097246SAdrian Prantl   // boundary plus an offset of one page (0x1000) or two, or four (0x4000),
30105097246SAdrian Prantl   // depending on the device.
3026ba6d3d1SJason Molenda 
303b9c1b51eSKate Stone   // Round the current pc down to the nearest one megabyte boundary - the place
304b9c1b51eSKate Stone   // where we will start searching.
3056ba6d3d1SJason Molenda   addr_t addr = pc & ~0xfffff;
3066ba6d3d1SJason Molenda 
307b9c1b51eSKate Stone   // Search backwards 32 megabytes, looking for the start of the kernel at each
308b9c1b51eSKate Stone   // one-megabyte boundary.
309b9c1b51eSKate Stone   for (int i = 0; i < 32; i++, addr -= 0x100000) {
3106ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr, process).IsValid())
3116ba6d3d1SJason Molenda       return addr;
3126ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x1000, process).IsValid())
3136ba6d3d1SJason Molenda       return addr + 0x1000;
3146ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x2000, process).IsValid())
3156ba6d3d1SJason Molenda       return addr + 0x2000;
316a02869deSJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x4000, process).IsValid())
317a02869deSJason Molenda       return addr + 0x4000;
3186ba6d3d1SJason Molenda   }
3196ba6d3d1SJason Molenda 
3206ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
3216ba6d3d1SJason Molenda }
3226ba6d3d1SJason Molenda 
3236ba6d3d1SJason Molenda //----------------------------------------------------------------------
32405097246SAdrian Prantl // Scan through the valid address range for a kernel binary. This is uselessly
32505097246SAdrian Prantl // slow in 64-bit environments so we don't even try it. This scan is not
32605097246SAdrian Prantl // enabled by default even for 32-bit targets. Returns the address of the
32705097246SAdrian Prantl // kernel if one was found, else LLDB_INVALID_ADDRESS.
3286ba6d3d1SJason Molenda //----------------------------------------------------------------------
329b9c1b51eSKate Stone lldb::addr_t DynamicLoaderDarwinKernel::SearchForKernelViaExhaustiveSearch(
330b9c1b51eSKate Stone     Process *process) {
331b9c1b51eSKate Stone   if (GetGlobalProperties()->GetScanType() != eKASLRScanExhaustiveScan) {
3326ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3336ba6d3d1SJason Molenda   }
3346ba6d3d1SJason Molenda 
3356ba6d3d1SJason Molenda   addr_t kernel_range_low, kernel_range_high;
336b9c1b51eSKate Stone   if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8) {
3376ba6d3d1SJason Molenda     kernel_range_low = 1ULL << 63;
3386ba6d3d1SJason Molenda     kernel_range_high = UINT64_MAX;
339b9c1b51eSKate Stone   } else {
3406ba6d3d1SJason Molenda     kernel_range_low = 1ULL << 31;
3416ba6d3d1SJason Molenda     kernel_range_high = UINT32_MAX;
3426ba6d3d1SJason Molenda   }
3436ba6d3d1SJason Molenda 
3446ba6d3d1SJason Molenda   // Stepping through memory at one-megabyte resolution looking for a kernel
3456ba6d3d1SJason Molenda   // rarely works (fast enough) with a 64-bit address space -- for now, let's
3466ba6d3d1SJason Molenda   // not even bother.  We may be attaching to something which *isn't* a kernel
3476ba6d3d1SJason Molenda   // and we don't want to spin for minutes on-end looking for a kernel.
3486ba6d3d1SJason Molenda   if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
3496ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3506ba6d3d1SJason Molenda 
3516ba6d3d1SJason Molenda   addr_t addr = kernel_range_low;
3526ba6d3d1SJason Molenda 
353b9c1b51eSKate Stone   while (addr >= kernel_range_low && addr < kernel_range_high) {
3546ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr, process).IsValid())
3556ba6d3d1SJason Molenda       return addr;
3566ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x1000, process).IsValid())
3576ba6d3d1SJason Molenda       return addr + 0x1000;
3586ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x2000, process).IsValid())
3596ba6d3d1SJason Molenda       return addr + 0x2000;
360a02869deSJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x4000, process).IsValid())
361a02869deSJason Molenda       return addr + 0x4000;
3626ba6d3d1SJason Molenda     addr += 0x100000;
3636ba6d3d1SJason Molenda   }
3646ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
3656ba6d3d1SJason Molenda }
3666ba6d3d1SJason Molenda 
3676ba6d3d1SJason Molenda //----------------------------------------------------------------------
36887f0f95eSJason Molenda // Read the mach_header struct out of memory and return it.
36987f0f95eSJason Molenda // Returns true if the mach_header was successfully read,
37087f0f95eSJason Molenda // Returns false if there was a problem reading the header, or it was not
37187f0f95eSJason Molenda // a Mach-O header.
37287f0f95eSJason Molenda //----------------------------------------------------------------------
37387f0f95eSJason Molenda 
37487f0f95eSJason Molenda bool
37587f0f95eSJason Molenda DynamicLoaderDarwinKernel::ReadMachHeader(addr_t addr, Process *process, llvm::MachO::mach_header &header) {
37687f0f95eSJason Molenda   Status read_error;
37787f0f95eSJason Molenda 
37887f0f95eSJason Molenda   // Read the mach header and see whether it looks like a kernel
37987f0f95eSJason Molenda   if (process->DoReadMemory (addr, &header, sizeof(header), read_error) !=
38087f0f95eSJason Molenda       sizeof(header))
38187f0f95eSJason Molenda     return false;
38287f0f95eSJason Molenda 
38387f0f95eSJason Molenda   const uint32_t magicks[] = { llvm::MachO::MH_MAGIC_64, llvm::MachO::MH_MAGIC, llvm::MachO::MH_CIGAM, llvm::MachO::MH_CIGAM_64};
38487f0f95eSJason Molenda 
38587f0f95eSJason Molenda   bool found_matching_pattern = false;
38687f0f95eSJason Molenda   for (size_t i = 0; i < llvm::array_lengthof (magicks); i++)
38787f0f95eSJason Molenda     if (::memcmp (&header.magic, &magicks[i], sizeof (uint32_t)) == 0)
38887f0f95eSJason Molenda         found_matching_pattern = true;
38987f0f95eSJason Molenda 
39087f0f95eSJason Molenda   if (found_matching_pattern == false)
39187f0f95eSJason Molenda       return false;
39287f0f95eSJason Molenda 
39387f0f95eSJason Molenda   if (header.magic == llvm::MachO::MH_CIGAM ||
39487f0f95eSJason Molenda       header.magic == llvm::MachO::MH_CIGAM_64) {
39587f0f95eSJason Molenda     header.magic = llvm::ByteSwap_32(header.magic);
39687f0f95eSJason Molenda     header.cputype = llvm::ByteSwap_32(header.cputype);
39787f0f95eSJason Molenda     header.cpusubtype = llvm::ByteSwap_32(header.cpusubtype);
39887f0f95eSJason Molenda     header.filetype = llvm::ByteSwap_32(header.filetype);
39987f0f95eSJason Molenda     header.ncmds = llvm::ByteSwap_32(header.ncmds);
40087f0f95eSJason Molenda     header.sizeofcmds = llvm::ByteSwap_32(header.sizeofcmds);
40187f0f95eSJason Molenda     header.flags = llvm::ByteSwap_32(header.flags);
40287f0f95eSJason Molenda   }
40387f0f95eSJason Molenda 
40487f0f95eSJason Molenda   return true;
40587f0f95eSJason Molenda }
40687f0f95eSJason Molenda 
40787f0f95eSJason Molenda //----------------------------------------------------------------------
4086ba6d3d1SJason Molenda // Given an address in memory, look to see if there is a kernel image at that
40987f0f95eSJason Molenda // address.
41087f0f95eSJason Molenda // Returns a UUID; if a kernel was not found at that address, UUID.IsValid()
41187f0f95eSJason Molenda // will be false.
4126ba6d3d1SJason Molenda //----------------------------------------------------------------------
4136ba6d3d1SJason Molenda lldb_private::UUID
414b9c1b51eSKate Stone DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress(lldb::addr_t addr,
415b9c1b51eSKate Stone                                                         Process *process) {
416ec504238SJason Molenda   Log *log(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
4176ba6d3d1SJason Molenda   if (addr == LLDB_INVALID_ADDRESS)
4186ba6d3d1SJason Molenda     return UUID();
4196ba6d3d1SJason Molenda 
420ec504238SJason Molenda   if (log)
421b9c1b51eSKate Stone     log->Printf("DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress: "
422b9c1b51eSKate Stone                 "looking for kernel binary at 0x%" PRIx64,
423b9c1b51eSKate Stone                 addr);
424ec504238SJason Molenda 
4256ba6d3d1SJason Molenda   llvm::MachO::mach_header header;
42687f0f95eSJason Molenda 
42787f0f95eSJason Molenda   if (ReadMachHeader (addr, process, header) == false)
4286ba6d3d1SJason Molenda     return UUID();
4296ba6d3d1SJason Molenda 
43087f0f95eSJason Molenda   // First try a quick test -- read the first 4 bytes and see if there is a
43187f0f95eSJason Molenda   // valid Mach-O magic field there
43287f0f95eSJason Molenda   // (the first field of the mach_header/mach_header_64 struct).
433b9c1b51eSKate Stone   // A kernel is an executable which does not have the dynamic link object flag
434b9c1b51eSKate Stone   // set.
435b9c1b51eSKate Stone   if (header.filetype == llvm::MachO::MH_EXECUTE &&
436b9c1b51eSKate Stone       (header.flags & llvm::MachO::MH_DYLDLINK) == 0) {
4376ba6d3d1SJason Molenda     // Create a full module to get the UUID
438b9c1b51eSKate Stone     ModuleSP memory_module_sp = process->ReadModuleFromMemory(
439b9c1b51eSKate Stone         FileSpec("temp_mach_kernel", false), addr);
4406ba6d3d1SJason Molenda     if (!memory_module_sp.get())
4416ba6d3d1SJason Molenda       return UUID();
4426ba6d3d1SJason Molenda 
4436ba6d3d1SJason Molenda     ObjectFile *exe_objfile = memory_module_sp->GetObjectFile();
444bc22c8d8SJason Molenda     if (exe_objfile == NULL) {
445bc22c8d8SJason Molenda       if (log)
446bc22c8d8SJason Molenda         log->Printf("DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress "
447bc22c8d8SJason Molenda                     "found a binary at 0x%" PRIx64
448bc22c8d8SJason Molenda                     " but could not create an object file from memory",
449bc22c8d8SJason Molenda                     addr);
4506ba6d3d1SJason Molenda       return UUID();
451bc22c8d8SJason Molenda     }
4526ba6d3d1SJason Molenda 
453b9c1b51eSKate Stone     if (exe_objfile->GetType() == ObjectFile::eTypeExecutable &&
454b9c1b51eSKate Stone         exe_objfile->GetStrata() == ObjectFile::eStrataKernel) {
455a4ce2538SJason Molenda       ArchSpec kernel_arch(eArchTypeMachO, header.cputype, header.cpusubtype);
456b9c1b51eSKate Stone       if (!process->GetTarget().GetArchitecture().IsCompatibleMatch(
457b9c1b51eSKate Stone               kernel_arch)) {
458a4ce2538SJason Molenda         process->GetTarget().SetArchitecture(kernel_arch);
459a4ce2538SJason Molenda       }
460bc22c8d8SJason Molenda       if (log) {
461bc22c8d8SJason Molenda         std::string uuid_str;
462bc22c8d8SJason Molenda         if (memory_module_sp->GetUUID().IsValid()) {
463bc22c8d8SJason Molenda           uuid_str = "with UUID ";
464bc22c8d8SJason Molenda           uuid_str += memory_module_sp->GetUUID().GetAsString();
465bc22c8d8SJason Molenda         } else {
466bc22c8d8SJason Molenda           uuid_str = "and no LC_UUID found in load commands ";
467bc22c8d8SJason Molenda         }
468bc22c8d8SJason Molenda         log->Printf(
469bc22c8d8SJason Molenda             "DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress: "
470bc22c8d8SJason Molenda             "kernel binary image found at 0x%" PRIx64 " with arch '%s' %s",
471bc22c8d8SJason Molenda             addr, kernel_arch.GetTriple().str().c_str(), uuid_str.c_str());
472bc22c8d8SJason Molenda       }
4736ba6d3d1SJason Molenda       return memory_module_sp->GetUUID();
4746ba6d3d1SJason Molenda     }
4756ba6d3d1SJason Molenda   }
4766ba6d3d1SJason Molenda 
4776ba6d3d1SJason Molenda   return UUID();
4786ba6d3d1SJason Molenda }
4796ba6d3d1SJason Molenda 
4806ba6d3d1SJason Molenda //----------------------------------------------------------------------
481d4bfbc9aSGreg Clayton // Constructor
482d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
483b9c1b51eSKate Stone DynamicLoaderDarwinKernel::DynamicLoaderDarwinKernel(Process *process,
484b9c1b51eSKate Stone                                                      lldb::addr_t kernel_addr)
485b9c1b51eSKate Stone     : DynamicLoader(process), m_kernel_load_address(kernel_addr), m_kernel(),
486b9c1b51eSKate Stone       m_kext_summary_header_ptr_addr(), m_kext_summary_header_addr(),
487b9c1b51eSKate Stone       m_kext_summary_header(), m_known_kexts(), m_mutex(),
488b9c1b51eSKate Stone       m_break_id(LLDB_INVALID_BREAK_ID) {
48997206d57SZachary Turner   Status error;
490b9c1b51eSKate Stone   PlatformSP platform_sp(
491b9c1b51eSKate Stone       Platform::Create(PlatformDarwinKernel::GetPluginNameStatic(), error));
4921c627543SJason Molenda   // Only select the darwin-kernel Platform if we've been asked to load kexts.
4931c627543SJason Molenda   // It can take some time to scan over all of the kext info.plists and that
4941c627543SJason Molenda   // shouldn't be done if kext loading is explicitly disabled.
495b9c1b51eSKate Stone   if (platform_sp.get() && GetGlobalProperties()->GetLoadKexts()) {
4961c627543SJason Molenda     process->GetTarget().SetPlatform(platform_sp);
4971c627543SJason Molenda   }
498d4bfbc9aSGreg Clayton }
499d4bfbc9aSGreg Clayton 
500d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
501d4bfbc9aSGreg Clayton // Destructor
502d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
503b9c1b51eSKate Stone DynamicLoaderDarwinKernel::~DynamicLoaderDarwinKernel() { Clear(true); }
504d4bfbc9aSGreg Clayton 
505b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::UpdateIfNeeded() {
506d4bfbc9aSGreg Clayton   LoadKernelModuleIfNeeded();
507d4bfbc9aSGreg Clayton   SetNotificationBreakpointIfNeeded();
508d4bfbc9aSGreg Clayton }
509d4bfbc9aSGreg Clayton //------------------------------------------------------------------
510d4bfbc9aSGreg Clayton /// Called after attaching a process.
511d4bfbc9aSGreg Clayton ///
512d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
513d4bfbc9aSGreg Clayton /// attaching to a process.
514d4bfbc9aSGreg Clayton //------------------------------------------------------------------
515b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DidAttach() {
516d4bfbc9aSGreg Clayton   PrivateInitialize(m_process);
517d4bfbc9aSGreg Clayton   UpdateIfNeeded();
518d4bfbc9aSGreg Clayton }
519d4bfbc9aSGreg Clayton 
520d4bfbc9aSGreg Clayton //------------------------------------------------------------------
521d4bfbc9aSGreg Clayton /// Called after attaching a process.
522d4bfbc9aSGreg Clayton ///
523d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
524d4bfbc9aSGreg Clayton /// attaching to a process.
525d4bfbc9aSGreg Clayton //------------------------------------------------------------------
526b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DidLaunch() {
527d4bfbc9aSGreg Clayton   PrivateInitialize(m_process);
528d4bfbc9aSGreg Clayton   UpdateIfNeeded();
529d4bfbc9aSGreg Clayton }
530d4bfbc9aSGreg Clayton 
531d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
532d4bfbc9aSGreg Clayton // Clear out the state of this class.
533d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
534b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Clear(bool clear_process) {
53516ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
536d4bfbc9aSGreg Clayton 
537d4bfbc9aSGreg Clayton   if (m_process->IsAlive() && LLDB_BREAK_ID_IS_VALID(m_break_id))
538d4bfbc9aSGreg Clayton     m_process->ClearBreakpointSiteByID(m_break_id);
539d4bfbc9aSGreg Clayton 
540d4bfbc9aSGreg Clayton   if (clear_process)
541d4bfbc9aSGreg Clayton     m_process = NULL;
542306bd0aaSJason Molenda   m_kernel.Clear();
543306bd0aaSJason Molenda   m_known_kexts.clear();
544d4bfbc9aSGreg Clayton   m_kext_summary_header_ptr_addr.Clear();
545d4bfbc9aSGreg Clayton   m_kext_summary_header_addr.Clear();
546d4bfbc9aSGreg Clayton   m_break_id = LLDB_INVALID_BREAK_ID;
547d4bfbc9aSGreg Clayton }
548d4bfbc9aSGreg Clayton 
549b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::LoadImageAtFileAddress(
550b9c1b51eSKate Stone     Process *process) {
5512af282a1SGreg Clayton   if (IsLoaded())
5522af282a1SGreg Clayton     return true;
5532af282a1SGreg Clayton 
554b9c1b51eSKate Stone   if (m_module_sp) {
5552af282a1SGreg Clayton     bool changed = false;
556751caf65SGreg Clayton     if (m_module_sp->SetLoadAddress(process->GetTarget(), 0, true, changed))
557306bd0aaSJason Molenda       m_load_process_stop_id = process->GetStopID();
5582af282a1SGreg Clayton   }
5592af282a1SGreg Clayton   return false;
5602af282a1SGreg Clayton }
5612af282a1SGreg Clayton 
562b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetModule(ModuleSP module_sp) {
563306bd0aaSJason Molenda   m_module_sp = module_sp;
564b9c1b51eSKate Stone   if (module_sp.get() && module_sp->GetObjectFile()) {
565b9c1b51eSKate Stone     if (module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeExecutable &&
566b9c1b51eSKate Stone         module_sp->GetObjectFile()->GetStrata() == ObjectFile::eStrataKernel) {
567306bd0aaSJason Molenda       m_kernel_image = true;
568b9c1b51eSKate Stone     } else {
569306bd0aaSJason Molenda       m_kernel_image = false;
570306bd0aaSJason Molenda     }
571306bd0aaSJason Molenda   }
572306bd0aaSJason Molenda }
573306bd0aaSJason Molenda 
574b9c1b51eSKate Stone ModuleSP DynamicLoaderDarwinKernel::KextImageInfo::GetModule() {
575306bd0aaSJason Molenda   return m_module_sp;
576306bd0aaSJason Molenda }
577306bd0aaSJason Molenda 
578b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetLoadAddress(
579b9c1b51eSKate Stone     addr_t load_addr) {
580306bd0aaSJason Molenda   m_load_address = load_addr;
581306bd0aaSJason Molenda }
582306bd0aaSJason Molenda 
583b9c1b51eSKate Stone addr_t DynamicLoaderDarwinKernel::KextImageInfo::GetLoadAddress() const {
584306bd0aaSJason Molenda   return m_load_address;
585306bd0aaSJason Molenda }
586306bd0aaSJason Molenda 
587b9c1b51eSKate Stone uint64_t DynamicLoaderDarwinKernel::KextImageInfo::GetSize() const {
588306bd0aaSJason Molenda   return m_size;
589306bd0aaSJason Molenda }
590306bd0aaSJason Molenda 
591b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetSize(uint64_t size) {
592306bd0aaSJason Molenda   m_size = size;
593306bd0aaSJason Molenda }
594306bd0aaSJason Molenda 
595b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::KextImageInfo::GetProcessStopId() const {
596306bd0aaSJason Molenda   return m_load_process_stop_id;
597306bd0aaSJason Molenda }
598306bd0aaSJason Molenda 
599b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetProcessStopId(
600b9c1b51eSKate Stone     uint32_t stop_id) {
601306bd0aaSJason Molenda   m_load_process_stop_id = stop_id;
602306bd0aaSJason Molenda }
603306bd0aaSJason Molenda 
604b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::
605b9c1b51eSKate Stone operator==(const KextImageInfo &rhs) {
606b9c1b51eSKate Stone   if (m_uuid.IsValid() || rhs.GetUUID().IsValid()) {
607b9c1b51eSKate Stone     if (m_uuid == rhs.GetUUID()) {
608306bd0aaSJason Molenda       return true;
609306bd0aaSJason Molenda     }
610306bd0aaSJason Molenda     return false;
611306bd0aaSJason Molenda   }
612306bd0aaSJason Molenda 
613306bd0aaSJason Molenda   if (m_name == rhs.GetName() && m_load_address == rhs.GetLoadAddress())
614306bd0aaSJason Molenda     return true;
615306bd0aaSJason Molenda 
616306bd0aaSJason Molenda   return false;
617306bd0aaSJason Molenda }
618306bd0aaSJason Molenda 
619b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetName(const char *name) {
620306bd0aaSJason Molenda   m_name = name;
621306bd0aaSJason Molenda }
622306bd0aaSJason Molenda 
623b9c1b51eSKate Stone std::string DynamicLoaderDarwinKernel::KextImageInfo::GetName() const {
624306bd0aaSJason Molenda   return m_name;
625306bd0aaSJason Molenda }
626306bd0aaSJason Molenda 
627b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetUUID(const UUID &uuid) {
628306bd0aaSJason Molenda   m_uuid = uuid;
629306bd0aaSJason Molenda }
630306bd0aaSJason Molenda 
631b9c1b51eSKate Stone UUID DynamicLoaderDarwinKernel::KextImageInfo::GetUUID() const {
632306bd0aaSJason Molenda   return m_uuid;
633306bd0aaSJason Molenda }
634306bd0aaSJason Molenda 
635b9c1b51eSKate Stone // Given the m_load_address from the kext summaries, and a UUID, try to create
63605097246SAdrian Prantl // an in-memory Module at that address.  Require that the MemoryModule have a
63705097246SAdrian Prantl // matching UUID and detect if this MemoryModule is a kernel or a kext.
638306bd0aaSJason Molenda //
639306bd0aaSJason Molenda // Returns true if m_memory_module_sp is now set to a valid Module.
640306bd0aaSJason Molenda 
641b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::ReadMemoryModule(
642b9c1b51eSKate Stone     Process *process) {
64308a3258bSJason Molenda   Log *log = lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_HOST);
644306bd0aaSJason Molenda   if (m_memory_module_sp.get() != NULL)
645306bd0aaSJason Molenda     return true;
646306bd0aaSJason Molenda   if (m_load_address == LLDB_INVALID_ADDRESS)
647306bd0aaSJason Molenda     return false;
648306bd0aaSJason Molenda 
649306bd0aaSJason Molenda   FileSpec file_spec;
650937348cdSJonas Devlieghere   file_spec.SetFile(m_name.c_str(), false, FileSpec::Style::native);
651306bd0aaSJason Molenda 
65287f0f95eSJason Molenda   llvm::MachO::mach_header mh;
65387f0f95eSJason Molenda   size_t size_to_read = 512;
65487f0f95eSJason Molenda   if (ReadMachHeader(m_load_address, process, mh)) {
65552cadf0dSDavid Bolvansky     if (mh.magic == llvm::MachO::MH_CIGAM || mh.magic == llvm::MachO::MH_MAGIC)
65687f0f95eSJason Molenda       size_to_read = sizeof(llvm::MachO::mach_header) + mh.sizeofcmds;
65752cadf0dSDavid Bolvansky     if (mh.magic == llvm::MachO::MH_CIGAM_64 ||
65852cadf0dSDavid Bolvansky         mh.magic == llvm::MachO::MH_MAGIC_64)
65987f0f95eSJason Molenda       size_to_read = sizeof(llvm::MachO::mach_header_64) + mh.sizeofcmds;
66087f0f95eSJason Molenda   }
66187f0f95eSJason Molenda 
662b9c1b51eSKate Stone   ModuleSP memory_module_sp =
66387f0f95eSJason Molenda       process->ReadModuleFromMemory(file_spec, m_load_address, size_to_read);
664306bd0aaSJason Molenda 
665306bd0aaSJason Molenda   if (memory_module_sp.get() == NULL)
666306bd0aaSJason Molenda     return false;
667306bd0aaSJason Molenda 
668306bd0aaSJason Molenda   bool is_kernel = false;
669b9c1b51eSKate Stone   if (memory_module_sp->GetObjectFile()) {
670b9c1b51eSKate Stone     if (memory_module_sp->GetObjectFile()->GetType() ==
671b9c1b51eSKate Stone             ObjectFile::eTypeExecutable &&
672b9c1b51eSKate Stone         memory_module_sp->GetObjectFile()->GetStrata() ==
673b9c1b51eSKate Stone             ObjectFile::eStrataKernel) {
674306bd0aaSJason Molenda       is_kernel = true;
675b9c1b51eSKate Stone     } else if (memory_module_sp->GetObjectFile()->GetType() ==
676b9c1b51eSKate Stone                ObjectFile::eTypeSharedLibrary) {
677306bd0aaSJason Molenda       is_kernel = false;
678306bd0aaSJason Molenda     }
679306bd0aaSJason Molenda   }
680306bd0aaSJason Molenda 
681b9c1b51eSKate Stone   // If this is a kext, and the kernel specified what UUID we should find at
68205097246SAdrian Prantl   // this load address, require that the memory module have a matching UUID or
68305097246SAdrian Prantl   // something has gone wrong and we should discard it.
684b9c1b51eSKate Stone   if (m_uuid.IsValid()) {
685b9c1b51eSKate Stone     if (m_uuid != memory_module_sp->GetUUID()) {
686b9c1b51eSKate Stone       if (log) {
687b9c1b51eSKate Stone         log->Printf("KextImageInfo::ReadMemoryModule the kernel said to find "
688b9c1b51eSKate Stone                     "uuid %s at 0x%" PRIx64
689b9c1b51eSKate Stone                     " but instead we found uuid %s, throwing it away",
690b9c1b51eSKate Stone                     m_uuid.GetAsString().c_str(), m_load_address,
691b9c1b51eSKate Stone                     memory_module_sp->GetUUID().GetAsString().c_str());
69208a3258bSJason Molenda       }
693306bd0aaSJason Molenda       return false;
694306bd0aaSJason Molenda     }
695306bd0aaSJason Molenda   }
696306bd0aaSJason Molenda 
697306bd0aaSJason Molenda   // If the in-memory Module has a UUID, let's use that.
698b9c1b51eSKate Stone   if (!m_uuid.IsValid() && memory_module_sp->GetUUID().IsValid()) {
699306bd0aaSJason Molenda     m_uuid = memory_module_sp->GetUUID();
700306bd0aaSJason Molenda   }
701306bd0aaSJason Molenda 
702306bd0aaSJason Molenda   m_memory_module_sp = memory_module_sp;
703306bd0aaSJason Molenda   m_kernel_image = is_kernel;
704b9c1b51eSKate Stone   if (is_kernel) {
705b9c1b51eSKate Stone     if (log) {
70608a3258bSJason Molenda       // This is unusual and probably not intended
707b9c1b51eSKate Stone       log->Printf("KextImageInfo::ReadMemoryModule read the kernel binary out "
708b9c1b51eSKate Stone                   "of memory");
70908a3258bSJason Molenda     }
710b9c1b51eSKate Stone     if (memory_module_sp->GetArchitecture().IsValid()) {
711306bd0aaSJason Molenda       process->GetTarget().SetArchitecture(memory_module_sp->GetArchitecture());
712306bd0aaSJason Molenda     }
713b9c1b51eSKate Stone     if (m_uuid.IsValid()) {
71448975833SJason Molenda       ModuleSP exe_module_sp = process->GetTarget().GetExecutableModule();
715b9c1b51eSKate Stone       if (exe_module_sp.get() && exe_module_sp->GetUUID().IsValid()) {
716b9c1b51eSKate Stone         if (m_uuid != exe_module_sp->GetUUID()) {
71748975833SJason Molenda           // The user specified a kernel binary that has a different UUID than
71848975833SJason Molenda           // the kernel actually running in memory.  This never ends well;
71948975833SJason Molenda           // clear the user specified kernel binary from the Target.
72048975833SJason Molenda 
72148975833SJason Molenda           m_module_sp.reset();
72248975833SJason Molenda 
72348975833SJason Molenda           ModuleList user_specified_kernel_list;
72448975833SJason Molenda           user_specified_kernel_list.Append(exe_module_sp);
72548975833SJason Molenda           process->GetTarget().GetImages().Remove(user_specified_kernel_list);
72638e70d11SJason Molenda         }
72738e70d11SJason Molenda       }
72838e70d11SJason Molenda     }
729306bd0aaSJason Molenda   }
730306bd0aaSJason Molenda 
731306bd0aaSJason Molenda   return true;
732306bd0aaSJason Molenda }
733306bd0aaSJason Molenda 
734b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::IsKernel() const {
735306bd0aaSJason Molenda   return m_kernel_image == true;
736306bd0aaSJason Molenda }
737306bd0aaSJason Molenda 
738b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetIsKernel(bool is_kernel) {
739306bd0aaSJason Molenda   m_kernel_image = is_kernel;
740306bd0aaSJason Molenda }
741306bd0aaSJason Molenda 
742b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::LoadImageUsingMemoryModule(
743b9c1b51eSKate Stone     Process *process) {
744c859e2d5SGreg Clayton   if (IsLoaded())
745c859e2d5SGreg Clayton     return true;
746c859e2d5SGreg Clayton 
747c859e2d5SGreg Clayton   Target &target = process->GetTarget();
74887a04b24SJason Molenda 
74905097246SAdrian Prantl   // If we don't have / can't create a memory module for this kext, don't try
75005097246SAdrian Prantl   // to load it - we won't have the correct segment load addresses.
751b9c1b51eSKate Stone   if (!ReadMemoryModule(process)) {
752a8cb765cSJason Molenda     Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
753a8cb765cSJason Molenda     if (log)
754b9c1b51eSKate Stone       log->Printf("Unable to read '%s' from memory at address 0x%" PRIx64
755b9c1b51eSKate Stone                   " to get the segment load addresses.",
756b9c1b51eSKate Stone                   m_name.c_str(), m_load_address);
75787a04b24SJason Molenda     return false;
75887a04b24SJason Molenda   }
75987a04b24SJason Molenda 
760306bd0aaSJason Molenda   bool uuid_is_valid = m_uuid.IsValid();
761306bd0aaSJason Molenda 
762b9c1b51eSKate Stone   if (IsKernel() && uuid_is_valid && m_memory_module_sp.get()) {
76344d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
764b9c1b51eSKate Stone     if (s) {
765b9c1b51eSKate Stone       s->Printf("Kernel UUID: %s\n",
766b9c1b51eSKate Stone                 m_memory_module_sp->GetUUID().GetAsString().c_str());
767e575e7bcSJason Molenda       s->Printf("Load Address: 0x%" PRIx64 "\n", m_load_address);
768e575e7bcSJason Molenda     }
769e575e7bcSJason Molenda   }
770e575e7bcSJason Molenda 
771b9c1b51eSKate Stone   if (!m_module_sp) {
772b9c1b51eSKate Stone     // See if the kext has already been loaded into the target, probably by the
773b9c1b51eSKate Stone     // user doing target modules add.
7741759848bSEnrico Granata     const ModuleList &target_images = target.GetImages();
775306bd0aaSJason Molenda     m_module_sp = target_images.FindModule(m_uuid);
776c859e2d5SGreg Clayton 
777306bd0aaSJason Molenda     // Search for the kext on the local filesystem via the UUID
778b9c1b51eSKate Stone     if (!m_module_sp && uuid_is_valid) {
7792af282a1SGreg Clayton       ModuleSpec module_spec;
780306bd0aaSJason Molenda       module_spec.GetUUID() = m_uuid;
78153667f5dSJason Molenda       module_spec.GetArchitecture() = target.GetArchitecture();
782bb860bd2SJason Molenda 
78305097246SAdrian Prantl       // For the kernel, we really do need an on-disk file copy of the binary
78405097246SAdrian Prantl       // to do anything useful. This will force a clal to
785b9c1b51eSKate Stone       if (IsKernel()) {
786b9c1b51eSKate Stone         if (Symbols::DownloadObjectAndSymbolFile(module_spec, true)) {
787*dbd7fabaSJonas Devlieghere           if (FileSystem::Instance().Exists(module_spec.GetFileSpec())) {
788b9c1b51eSKate Stone             m_module_sp.reset(new Module(module_spec.GetFileSpec(),
789b9c1b51eSKate Stone                                          target.GetArchitecture()));
790b9c1b51eSKate Stone             if (m_module_sp.get() &&
791b9c1b51eSKate Stone                 m_module_sp->MatchesModuleSpec(module_spec)) {
792bb860bd2SJason Molenda               ModuleList loaded_module_list;
793306bd0aaSJason Molenda               loaded_module_list.Append(m_module_sp);
794bb860bd2SJason Molenda               target.ModulesDidLoad(loaded_module_list);
795bb860bd2SJason Molenda             }
796bb860bd2SJason Molenda           }
797bb860bd2SJason Molenda         }
798d76fb6eaSJason Molenda       }
799bb860bd2SJason Molenda 
800b9c1b51eSKate Stone       // If the current platform is PlatformDarwinKernel, create a ModuleSpec
80105097246SAdrian Prantl       // with the filename set to be the bundle ID for this kext, e.g.
80205097246SAdrian Prantl       // "com.apple.filesystems.msdosfs", and ask the platform to find it.
8031c627543SJason Molenda       PlatformSP platform_sp(target.GetPlatform());
804b9c1b51eSKate Stone       if (!m_module_sp && platform_sp) {
80557abc5d6SGreg Clayton         ConstString platform_name(platform_sp->GetPluginName());
806b9c1b51eSKate Stone         static ConstString g_platform_name(
807b9c1b51eSKate Stone             PlatformDarwinKernel::GetPluginNameStatic());
808b9c1b51eSKate Stone         if (platform_name == g_platform_name) {
8091c627543SJason Molenda           ModuleSpec kext_bundle_module_spec(module_spec);
8101c627543SJason Molenda           FileSpec kext_filespec(m_name.c_str(), false);
8111c627543SJason Molenda           kext_bundle_module_spec.GetFileSpec() = kext_filespec;
812b9c1b51eSKate Stone           platform_sp->GetSharedModule(
813b9c1b51eSKate Stone               kext_bundle_module_spec, process, m_module_sp,
814b9c1b51eSKate Stone               &target.GetExecutableSearchPaths(), NULL, NULL);
8151c627543SJason Molenda         }
8161c627543SJason Molenda       }
8171c627543SJason Molenda 
818d76fb6eaSJason Molenda       // Ask the Target to find this file on the local system, if possible.
819bb860bd2SJason Molenda       // This will search in the list of currently-loaded files, look in the
820bb860bd2SJason Molenda       // standard search paths on the system, and on a Mac it will try calling
821bb860bd2SJason Molenda       // the DebugSymbols framework with the UUID to find the binary via its
822bb860bd2SJason Molenda       // search methods.
823b9c1b51eSKate Stone       if (!m_module_sp) {
824306bd0aaSJason Molenda         m_module_sp = target.GetSharedModule(module_spec);
825306bd0aaSJason Molenda       }
826e575e7bcSJason Molenda 
827b9c1b51eSKate Stone       if (IsKernel() && !m_module_sp) {
82844d93782SGreg Clayton         Stream *s = target.GetDebugger().GetOutputFile().get();
829b9c1b51eSKate Stone         if (s) {
830b9c1b51eSKate Stone           s->Printf("WARNING: Unable to locate kernel binary on the debugger "
831b9c1b51eSKate Stone                     "system.\n");
832e575e7bcSJason Molenda         }
833e575e7bcSJason Molenda       }
834b9a01b39SGreg Clayton     }
83565d57a3dSJason Molenda 
83605097246SAdrian Prantl     // If we managed to find a module, append it to the target's list of
83705097246SAdrian Prantl     // images. If we also have a memory module, require that they have matching
83805097246SAdrian Prantl     // UUIDs
839b9c1b51eSKate Stone     if (m_module_sp) {
840306bd0aaSJason Molenda       bool uuid_match_ok = true;
841b9c1b51eSKate Stone       if (m_memory_module_sp) {
842b9c1b51eSKate Stone         if (m_module_sp->GetUUID() != m_memory_module_sp->GetUUID()) {
843306bd0aaSJason Molenda           uuid_match_ok = false;
84465d57a3dSJason Molenda         }
84565d57a3dSJason Molenda       }
846b9c1b51eSKate Stone       if (uuid_match_ok) {
847a4d3e1d2SJason Molenda         target.GetImages().AppendIfNeeded(m_module_sp);
848b9c1b51eSKate Stone         if (IsKernel() &&
849b9c1b51eSKate Stone             target.GetExecutableModulePointer() != m_module_sp.get()) {
850f9a07e9fSJonas Devlieghere           target.SetExecutableModule(m_module_sp, eLoadDependentsNo);
851306bd0aaSJason Molenda         }
852c859e2d5SGreg Clayton       }
853c859e2d5SGreg Clayton     }
854bb860bd2SJason Molenda   }
855c859e2d5SGreg Clayton 
856b9c1b51eSKate Stone   if (!m_module_sp && !IsKernel() && m_uuid.IsValid() && !m_name.empty()) {
85744d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
858b9c1b51eSKate Stone     if (s) {
859b9c1b51eSKate Stone       s->Printf("warning: Can't find binary/dSYM for %s (%s)\n", m_name.c_str(),
860b9c1b51eSKate Stone                 m_uuid.GetAsString().c_str());
86156c23285SJason Molenda     }
86256c23285SJason Molenda   }
863c859e2d5SGreg Clayton 
86467408537SGreg Clayton   static ConstString g_section_name_LINKEDIT("__LINKEDIT");
86567408537SGreg Clayton 
866b9c1b51eSKate Stone   if (m_memory_module_sp && m_module_sp) {
867b9c1b51eSKate Stone     if (m_module_sp->GetUUID() == m_memory_module_sp->GetUUID()) {
868306bd0aaSJason Molenda       ObjectFile *ondisk_object_file = m_module_sp->GetObjectFile();
869306bd0aaSJason Molenda       ObjectFile *memory_object_file = m_memory_module_sp->GetObjectFile();
87067408537SGreg Clayton 
871b9c1b51eSKate Stone       if (memory_object_file && ondisk_object_file) {
872b9c1b51eSKate Stone         // The memory_module for kexts may have an invalid __LINKEDIT seg; skip
873b9c1b51eSKate Stone         // it.
874306bd0aaSJason Molenda         const bool ignore_linkedit = !IsKernel();
87567408537SGreg Clayton 
876c859e2d5SGreg Clayton         SectionList *ondisk_section_list = ondisk_object_file->GetSectionList();
877c859e2d5SGreg Clayton         SectionList *memory_section_list = memory_object_file->GetSectionList();
878b9c1b51eSKate Stone         if (memory_section_list && ondisk_section_list) {
879bae2f2f8SGreg Clayton           const uint32_t num_ondisk_sections = ondisk_section_list->GetSize();
88005097246SAdrian Prantl           // There may be CTF sections in the memory image so we can't always
88105097246SAdrian Prantl           // just compare the number of sections (which are actually segments
88205097246SAdrian Prantl           // in mach-o parlance)
883c859e2d5SGreg Clayton           uint32_t sect_idx = 0;
884bae2f2f8SGreg Clayton 
88505097246SAdrian Prantl           // Use the memory_module's addresses for each section to set the file
88605097246SAdrian Prantl           // module's load address as appropriate.  We don't want to use a
88705097246SAdrian Prantl           // single slide value for the entire kext - different segments may be
88805097246SAdrian Prantl           // slid different amounts by the kext loader.
889bae2f2f8SGreg Clayton 
890bae2f2f8SGreg Clayton           uint32_t num_sections_loaded = 0;
891b9c1b51eSKate Stone           for (sect_idx = 0; sect_idx < num_ondisk_sections; ++sect_idx) {
892b9c1b51eSKate Stone             SectionSP ondisk_section_sp(
893b9c1b51eSKate Stone                 ondisk_section_list->GetSectionAtIndex(sect_idx));
894b9c1b51eSKate Stone             if (ondisk_section_sp) {
89567408537SGreg Clayton               // Don't ever load __LINKEDIT as it may or may not be actually
89667408537SGreg Clayton               // mapped into memory and there is no current way to tell.
89767408537SGreg Clayton               // I filed rdar://problem/12851706 to track being able to tell
89867408537SGreg Clayton               // if the __LINKEDIT is actually mapped, but until then, we need
89967408537SGreg Clayton               // to not load the __LINKEDIT
900b9c1b51eSKate Stone               if (ignore_linkedit &&
901b9c1b51eSKate Stone                   ondisk_section_sp->GetName() == g_section_name_LINKEDIT)
90267408537SGreg Clayton                 continue;
90367408537SGreg Clayton 
904b9c1b51eSKate Stone               const Section *memory_section =
905b9c1b51eSKate Stone                   memory_section_list
906b9c1b51eSKate Stone                       ->FindSectionByName(ondisk_section_sp->GetName())
907b9c1b51eSKate Stone                       .get();
908b9c1b51eSKate Stone               if (memory_section) {
909b9c1b51eSKate Stone                 target.SetSectionLoadAddress(ondisk_section_sp,
910b9c1b51eSKate Stone                                              memory_section->GetFileAddress());
911bae2f2f8SGreg Clayton                 ++num_sections_loaded;
912c859e2d5SGreg Clayton               }
913bae2f2f8SGreg Clayton             }
914bae2f2f8SGreg Clayton           }
915bae2f2f8SGreg Clayton           if (num_sections_loaded > 0)
916306bd0aaSJason Molenda             m_load_process_stop_id = process->GetStopID();
917bae2f2f8SGreg Clayton           else
918306bd0aaSJason Molenda             m_module_sp.reset(); // No sections were loaded
919b9c1b51eSKate Stone         } else
920306bd0aaSJason Molenda           m_module_sp.reset(); // One or both section lists
921b9c1b51eSKate Stone       } else
922306bd0aaSJason Molenda         m_module_sp.reset(); // One or both object files missing
923b9c1b51eSKate Stone     } else
924306bd0aaSJason Molenda       m_module_sp.reset(); // UUID mismatch
925c859e2d5SGreg Clayton   }
926c859e2d5SGreg Clayton 
927c859e2d5SGreg Clayton   bool is_loaded = IsLoaded();
928c859e2d5SGreg Clayton 
929b9c1b51eSKate Stone   if (is_loaded && m_module_sp && IsKernel()) {
93044d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
931b9c1b51eSKate Stone     if (s) {
932732238cdSJason Molenda       ObjectFile *kernel_object_file = m_module_sp->GetObjectFile();
933b9c1b51eSKate Stone       if (kernel_object_file) {
934b9c1b51eSKate Stone         addr_t file_address =
935b9c1b51eSKate Stone             kernel_object_file->GetHeaderAddress().GetFileAddress();
936b9c1b51eSKate Stone         if (m_load_address != LLDB_INVALID_ADDRESS &&
937b9c1b51eSKate Stone             file_address != LLDB_INVALID_ADDRESS) {
938b9c1b51eSKate Stone           s->Printf("Kernel slid 0x%" PRIx64 " in memory.\n",
939b9c1b51eSKate Stone                     m_load_address - file_address);
940732238cdSJason Molenda         }
941732238cdSJason Molenda       }
942743e4396SJason Molenda       {
943b5ad4ec7SGreg Clayton         s->Printf("Loaded kernel file %s\n",
944b5ad4ec7SGreg Clayton                   m_module_sp->GetFileSpec().GetPath().c_str());
945743e4396SJason Molenda       }
94668b3607fSJason Molenda       s->Flush();
94768b3607fSJason Molenda     }
94868b3607fSJason Molenda   }
949c859e2d5SGreg Clayton   return is_loaded;
950c859e2d5SGreg Clayton }
951c859e2d5SGreg Clayton 
952b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::KextImageInfo::GetAddressByteSize() {
953306bd0aaSJason Molenda   if (m_memory_module_sp)
954306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture().GetAddressByteSize();
955306bd0aaSJason Molenda   if (m_module_sp)
956306bd0aaSJason Molenda     return m_module_sp->GetArchitecture().GetAddressByteSize();
9571f746071SGreg Clayton   return 0;
9581f746071SGreg Clayton }
9591f746071SGreg Clayton 
960b9c1b51eSKate Stone lldb::ByteOrder DynamicLoaderDarwinKernel::KextImageInfo::GetByteOrder() {
961306bd0aaSJason Molenda   if (m_memory_module_sp)
962306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture().GetByteOrder();
963306bd0aaSJason Molenda   if (m_module_sp)
964306bd0aaSJason Molenda     return m_module_sp->GetArchitecture().GetByteOrder();
9659ccb970fSBruce Mitchener   return endian::InlHostByteOrder();
9661f746071SGreg Clayton }
9671f746071SGreg Clayton 
9681f746071SGreg Clayton lldb_private::ArchSpec
969b9c1b51eSKate Stone DynamicLoaderDarwinKernel::KextImageInfo::GetArchitecture() const {
970306bd0aaSJason Molenda   if (m_memory_module_sp)
971306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture();
972306bd0aaSJason Molenda   if (m_module_sp)
973306bd0aaSJason Molenda     return m_module_sp->GetArchitecture();
9741f746071SGreg Clayton   return lldb_private::ArchSpec();
9751f746071SGreg Clayton }
9761f746071SGreg Clayton 
977d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
97805097246SAdrian Prantl // Load the kernel module and initialize the "m_kernel" member. Return true
97905097246SAdrian Prantl // _only_ if the kernel is loaded the first time through (subsequent calls to
98005097246SAdrian Prantl // this function should return false after the kernel has been already loaded).
981d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
982b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::LoadKernelModuleIfNeeded() {
983b9c1b51eSKate Stone   if (!m_kext_summary_header_ptr_addr.IsValid()) {
984306bd0aaSJason Molenda     m_kernel.Clear();
985306bd0aaSJason Molenda     m_kernel.SetModule(m_process->GetTarget().GetExecutableModule());
986306bd0aaSJason Molenda     m_kernel.SetIsKernel(true);
9874bd4e7e3SJason Molenda 
9884bd4e7e3SJason Molenda     ConstString kernel_name("mach_kernel");
989b9c1b51eSKate Stone     if (m_kernel.GetModule().get() && m_kernel.GetModule()->GetObjectFile() &&
990b9c1b51eSKate Stone         !m_kernel.GetModule()
991b9c1b51eSKate Stone              ->GetObjectFile()
992b9c1b51eSKate Stone              ->GetFileSpec()
993b9c1b51eSKate Stone              .GetFilename()
994b9c1b51eSKate Stone              .IsEmpty()) {
995b9c1b51eSKate Stone       kernel_name =
996b9c1b51eSKate Stone           m_kernel.GetModule()->GetObjectFile()->GetFileSpec().GetFilename();
9974bd4e7e3SJason Molenda     }
998306bd0aaSJason Molenda     m_kernel.SetName(kernel_name.AsCString());
9994bd4e7e3SJason Molenda 
1000b9c1b51eSKate Stone     if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS) {
1001306bd0aaSJason Molenda       m_kernel.SetLoadAddress(m_kernel_load_address);
1002b9c1b51eSKate Stone       if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS &&
1003b9c1b51eSKate Stone           m_kernel.GetModule()) {
100405097246SAdrian Prantl         // We didn't get a hint from the process, so we will try the kernel at
100505097246SAdrian Prantl         // the address that it exists at in the file if we have one
1006306bd0aaSJason Molenda         ObjectFile *kernel_object_file = m_kernel.GetModule()->GetObjectFile();
1007b9c1b51eSKate Stone         if (kernel_object_file) {
1008b9c1b51eSKate Stone           addr_t load_address =
1009b9c1b51eSKate Stone               kernel_object_file->GetHeaderAddress().GetLoadAddress(
1010b9c1b51eSKate Stone                   &m_process->GetTarget());
1011b9c1b51eSKate Stone           addr_t file_address =
1012b9c1b51eSKate Stone               kernel_object_file->GetHeaderAddress().GetFileAddress();
1013b9c1b51eSKate Stone           if (load_address != LLDB_INVALID_ADDRESS && load_address != 0) {
1014306bd0aaSJason Molenda             m_kernel.SetLoadAddress(load_address);
1015b9c1b51eSKate Stone             if (load_address != file_address) {
10164bd4e7e3SJason Molenda               // Don't accidentally relocate the kernel to the File address --
1017b9c1b51eSKate Stone               // the Load address has already been set to its actual in-memory
101805097246SAdrian Prantl               // address. Mark it as IsLoaded.
1019306bd0aaSJason Molenda               m_kernel.SetProcessStopId(m_process->GetStopID());
10204bd4e7e3SJason Molenda             }
1021b9c1b51eSKate Stone           } else {
1022306bd0aaSJason Molenda             m_kernel.SetLoadAddress(file_address);
10234bd4e7e3SJason Molenda           }
10244bd4e7e3SJason Molenda         }
1025c859e2d5SGreg Clayton       }
1026c859e2d5SGreg Clayton     }
1027c859e2d5SGreg Clayton 
1028b9c1b51eSKate Stone     if (m_kernel.GetLoadAddress() != LLDB_INVALID_ADDRESS) {
1029b9c1b51eSKate Stone       if (!m_kernel.LoadImageUsingMemoryModule(m_process)) {
10302af282a1SGreg Clayton         m_kernel.LoadImageAtFileAddress(m_process);
10312af282a1SGreg Clayton       }
10322af282a1SGreg Clayton     }
1033c859e2d5SGreg Clayton 
1034519b0816SJim Ingham     // The operating system plugin gets loaded and initialized in
103505097246SAdrian Prantl     // LoadImageUsingMemoryModule when we discover the kernel dSYM.  For a core
103605097246SAdrian Prantl     // file in particular, that's the wrong place to do this, since  we haven't
103705097246SAdrian Prantl     // fixed up the section addresses yet.  So let's redo it here.
1038519b0816SJim Ingham     LoadOperatingSystemPlugin(false);
1039519b0816SJim Ingham 
1040b9c1b51eSKate Stone     if (m_kernel.IsLoaded() && m_kernel.GetModule()) {
1041d4bfbc9aSGreg Clayton       static ConstString kext_summary_symbol("gLoadedKextSummaries");
1042b9c1b51eSKate Stone       const Symbol *symbol =
1043b9c1b51eSKate Stone           m_kernel.GetModule()->FindFirstSymbolWithNameAndType(
1044b9c1b51eSKate Stone               kext_summary_symbol, eSymbolTypeData);
1045b9c1b51eSKate Stone       if (symbol) {
1046e7612134SGreg Clayton         m_kext_summary_header_ptr_addr = symbol->GetAddress();
1047d4bfbc9aSGreg Clayton         // Update all image infos
1048d4bfbc9aSGreg Clayton         ReadAllKextSummaries();
1049d4bfbc9aSGreg Clayton       }
1050b9c1b51eSKate Stone     } else {
1051306bd0aaSJason Molenda       m_kernel.Clear();
1052d4bfbc9aSGreg Clayton     }
1053d4bfbc9aSGreg Clayton   }
1054d4bfbc9aSGreg Clayton }
1055d4bfbc9aSGreg Clayton 
1056d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1057d4bfbc9aSGreg Clayton // Static callback function that gets called when our DYLD notification
105805097246SAdrian Prantl // breakpoint gets hit. We update all of our image infos and then let our super
105905097246SAdrian Prantl // class DynamicLoader class decide if we should stop or not (based on global
106005097246SAdrian Prantl // preference).
1061d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1062b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::BreakpointHitCallback(
1063b9c1b51eSKate Stone     void *baton, StoppointCallbackContext *context, user_id_t break_id,
1064b9c1b51eSKate Stone     user_id_t break_loc_id) {
1065b9c1b51eSKate Stone   return static_cast<DynamicLoaderDarwinKernel *>(baton)->BreakpointHit(
1066b9c1b51eSKate Stone       context, break_id, break_loc_id);
1067d4bfbc9aSGreg Clayton }
1068d4bfbc9aSGreg Clayton 
1069b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::BreakpointHit(StoppointCallbackContext *context,
1070d4bfbc9aSGreg Clayton                                               user_id_t break_id,
1071b9c1b51eSKate Stone                                               user_id_t break_loc_id) {
10725160ce5cSGreg Clayton   Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
1073d4bfbc9aSGreg Clayton   if (log)
1074944b828aSGreg Clayton     log->Printf("DynamicLoaderDarwinKernel::BreakpointHit (...)\n");
1075d4bfbc9aSGreg Clayton 
1076d4bfbc9aSGreg Clayton   ReadAllKextSummaries();
1077d4bfbc9aSGreg Clayton 
1078d4bfbc9aSGreg Clayton   if (log)
10795160ce5cSGreg Clayton     PutToLog(log);
1080d4bfbc9aSGreg Clayton 
1081d4bfbc9aSGreg Clayton   return GetStopWhenImagesChange();
1082d4bfbc9aSGreg Clayton }
1083d4bfbc9aSGreg Clayton 
1084b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ReadKextSummaryHeader() {
108516ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1086d4bfbc9aSGreg Clayton 
1087d4bfbc9aSGreg Clayton   // the all image infos is already valid for this process stop ID
1088d4bfbc9aSGreg Clayton 
1089b9c1b51eSKate Stone   if (m_kext_summary_header_ptr_addr.IsValid()) {
1090d4bfbc9aSGreg Clayton     const uint32_t addr_size = m_kernel.GetAddressByteSize();
1091d4bfbc9aSGreg Clayton     const ByteOrder byte_order = m_kernel.GetByteOrder();
109297206d57SZachary Turner     Status error;
109305097246SAdrian Prantl     // Read enough bytes for a "OSKextLoadedKextSummaryHeader" structure which
109405097246SAdrian Prantl     // is currently 4 uint32_t and a pointer.
1095d4bfbc9aSGreg Clayton     uint8_t buf[24];
1096d4bfbc9aSGreg Clayton     DataExtractor data(buf, sizeof(buf), byte_order, addr_size);
1097d4bfbc9aSGreg Clayton     const size_t count = 4 * sizeof(uint32_t) + addr_size;
1098d4bfbc9aSGreg Clayton     const bool prefer_file_cache = false;
1099b9c1b51eSKate Stone     if (m_process->GetTarget().ReadPointerFromMemory(
1100b9c1b51eSKate Stone             m_kext_summary_header_ptr_addr, prefer_file_cache, error,
1101b9c1b51eSKate Stone             m_kext_summary_header_addr)) {
1102b9c1b51eSKate Stone       // We got a valid address for our kext summary header and make sure it
1103b9c1b51eSKate Stone       // isn't NULL
1104d4bfbc9aSGreg Clayton       if (m_kext_summary_header_addr.IsValid() &&
1105b9c1b51eSKate Stone           m_kext_summary_header_addr.GetFileAddress() != 0) {
1106b9c1b51eSKate Stone         const size_t bytes_read = m_process->GetTarget().ReadMemory(
1107b9c1b51eSKate Stone             m_kext_summary_header_addr, prefer_file_cache, buf, count, error);
1108b9c1b51eSKate Stone         if (bytes_read == count) {
1109c7bece56SGreg Clayton           lldb::offset_t offset = 0;
1110d4bfbc9aSGreg Clayton           m_kext_summary_header.version = data.GetU32(&offset);
1111b9c1b51eSKate Stone           if (m_kext_summary_header.version > 128) {
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 version number %u\n",
1116b9c1b51eSKate Stone                       m_kext_summary_header.version);
1117b9c1b51eSKate Stone             // If we get an improbably large version number, we're probably
1118b9c1b51eSKate Stone             // getting bad memory.
1119c6fa5db7SJason Molenda             m_kext_summary_header_addr.Clear();
1120c6fa5db7SJason Molenda             return false;
1121c6fa5db7SJason Molenda           }
1122b9c1b51eSKate Stone           if (m_kext_summary_header.version >= 2) {
1123d4bfbc9aSGreg Clayton             m_kext_summary_header.entry_size = data.GetU32(&offset);
1124b9c1b51eSKate Stone             if (m_kext_summary_header.entry_size > 4096) {
1125b9c1b51eSKate Stone               // If we get an improbably large entry_size, we're probably
1126b9c1b51eSKate Stone               // getting bad memory.
1127b9c1b51eSKate Stone               Stream *s =
1128b9c1b51eSKate Stone                   m_process->GetTarget().GetDebugger().GetOutputFile().get();
1129b9c1b51eSKate Stone               s->Printf("WARNING: Unable to read kext summary header, got "
1130b9c1b51eSKate Stone                         "improbable entry_size %u\n",
1131b9c1b51eSKate Stone                         m_kext_summary_header.entry_size);
1132c6fa5db7SJason Molenda               m_kext_summary_header_addr.Clear();
1133c6fa5db7SJason Molenda               return false;
1134c6fa5db7SJason Molenda             }
1135b9c1b51eSKate Stone           } else {
113605097246SAdrian Prantl             // Versions less than 2 didn't have an entry size, it was hard
113705097246SAdrian Prantl             // coded
1138b9c1b51eSKate Stone             m_kext_summary_header.entry_size =
1139b9c1b51eSKate Stone                 KERNEL_MODULE_ENTRY_SIZE_VERSION_1;
1140d4bfbc9aSGreg Clayton           }
1141d4bfbc9aSGreg Clayton           m_kext_summary_header.entry_count = data.GetU32(&offset);
1142b9c1b51eSKate Stone           if (m_kext_summary_header.entry_count > 10000) {
1143b9c1b51eSKate Stone             // If we get an improbably large number of kexts, we're probably
1144b9c1b51eSKate Stone             // getting bad memory.
1145b9c1b51eSKate Stone             Stream *s =
1146b9c1b51eSKate Stone                 m_process->GetTarget().GetDebugger().GetOutputFile().get();
1147b9c1b51eSKate Stone             s->Printf("WARNING: Unable to read kext summary header, got "
1148b9c1b51eSKate Stone                       "improbable number of kexts %u\n",
1149b9c1b51eSKate Stone                       m_kext_summary_header.entry_count);
1150c6fa5db7SJason Molenda             m_kext_summary_header_addr.Clear();
1151c6fa5db7SJason Molenda             return false;
1152c6fa5db7SJason Molenda           }
1153d4bfbc9aSGreg Clayton           return true;
1154d4bfbc9aSGreg Clayton         }
1155d4bfbc9aSGreg Clayton       }
1156d4bfbc9aSGreg Clayton     }
1157d4bfbc9aSGreg Clayton   }
1158d4bfbc9aSGreg Clayton   m_kext_summary_header_addr.Clear();
1159d4bfbc9aSGreg Clayton   return false;
1160d4bfbc9aSGreg Clayton }
1161d4bfbc9aSGreg Clayton 
1162b9c1b51eSKate Stone // We've either (a) just attached to a new kernel, or (b) the kexts-changed
116305097246SAdrian Prantl // breakpoint was hit and we need to figure out what kexts have been added or
116405097246SAdrian Prantl // removed. Read the kext summaries from the inferior kernel memory, compare
116505097246SAdrian Prantl // them against the m_known_kexts vector and update the m_known_kexts vector as
116605097246SAdrian Prantl // needed to keep in sync with the inferior.
1167d4bfbc9aSGreg Clayton 
1168b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ParseKextSummaries(
1169b9c1b51eSKate Stone     const Address &kext_summary_addr, uint32_t count) {
1170306bd0aaSJason Molenda   KextImageInfo::collection kext_summaries;
11715160ce5cSGreg Clayton   Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
1172d4bfbc9aSGreg Clayton   if (log)
1173b9c1b51eSKate Stone     log->Printf("Kexts-changed breakpoint hit, there are %d kexts currently.\n",
1174b9c1b51eSKate Stone                 count);
1175d4bfbc9aSGreg Clayton 
117616ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1177d4bfbc9aSGreg Clayton 
1178d4bfbc9aSGreg Clayton   if (!ReadKextSummaries(kext_summary_addr, count, kext_summaries))
1179d4bfbc9aSGreg Clayton     return false;
1180d4bfbc9aSGreg Clayton 
1181b9c1b51eSKate Stone   // read the plugin.dynamic-loader.darwin-kernel.load-kexts setting -- if the
118205097246SAdrian Prantl   // user requested no kext loading, don't print any messages about kexts &
118305097246SAdrian Prantl   // don't try to read them.
11844da2e32fSJason Molenda   const bool load_kexts = GetGlobalProperties()->GetLoadKexts();
11854da2e32fSJason Molenda 
1186b9c1b51eSKate Stone   // By default, all kexts we've loaded in the past are marked as "remove" and
118705097246SAdrian Prantl   // all of the kexts we just found out about from ReadKextSummaries are marked
118805097246SAdrian Prantl   // as "add".
1189306bd0aaSJason Molenda   std::vector<bool> to_be_removed(m_known_kexts.size(), true);
1190306bd0aaSJason Molenda   std::vector<bool> to_be_added(count, true);
1191306bd0aaSJason Molenda 
1192306bd0aaSJason Molenda   int number_of_new_kexts_being_added = 0;
1193306bd0aaSJason Molenda   int number_of_old_kexts_being_removed = m_known_kexts.size();
1194306bd0aaSJason Molenda 
1195306bd0aaSJason Molenda   const uint32_t new_kexts_size = kext_summaries.size();
1196306bd0aaSJason Molenda   const uint32_t old_kexts_size = m_known_kexts.size();
1197306bd0aaSJason Molenda 
119805097246SAdrian Prantl   // The m_known_kexts vector may have entries that have been Cleared, or are a
119905097246SAdrian Prantl   // kernel.
1200b9c1b51eSKate Stone   for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++) {
1201306bd0aaSJason Molenda     bool ignore = false;
1202306bd0aaSJason Molenda     KextImageInfo &image_info = m_known_kexts[old_kext];
1203b9c1b51eSKate Stone     if (image_info.IsKernel()) {
1204306bd0aaSJason Molenda       ignore = true;
1205b9c1b51eSKate Stone     } else if (image_info.GetLoadAddress() == LLDB_INVALID_ADDRESS &&
1206b9c1b51eSKate Stone                !image_info.GetModule()) {
1207306bd0aaSJason Molenda       ignore = true;
1208306bd0aaSJason Molenda     }
1209306bd0aaSJason Molenda 
1210b9c1b51eSKate Stone     if (ignore) {
1211306bd0aaSJason Molenda       number_of_old_kexts_being_removed--;
1212306bd0aaSJason Molenda       to_be_removed[old_kext] = false;
1213306bd0aaSJason Molenda     }
1214306bd0aaSJason Molenda   }
1215306bd0aaSJason Molenda 
1216306bd0aaSJason Molenda   // Scan over the list of kexts we just read from the kernel, note those that
1217306bd0aaSJason Molenda   // need to be added and those already loaded.
1218b9c1b51eSKate Stone   for (uint32_t new_kext = 0; new_kext < new_kexts_size; new_kext++) {
1219306bd0aaSJason Molenda     bool add_this_one = true;
1220b9c1b51eSKate Stone     for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++) {
1221b9c1b51eSKate Stone       if (m_known_kexts[old_kext] == kext_summaries[new_kext]) {
1222306bd0aaSJason Molenda         // We already have this kext, don't re-load it.
1223306bd0aaSJason Molenda         to_be_added[new_kext] = false;
1224306bd0aaSJason Molenda         // This kext is still present, do not remove it.
1225306bd0aaSJason Molenda         to_be_removed[old_kext] = false;
1226306bd0aaSJason Molenda 
1227306bd0aaSJason Molenda         number_of_old_kexts_being_removed--;
1228306bd0aaSJason Molenda         add_this_one = false;
1229306bd0aaSJason Molenda         break;
1230306bd0aaSJason Molenda       }
1231306bd0aaSJason Molenda     }
123205097246SAdrian Prantl     // If this "kext" entry is actually an alias for the kernel -- the kext was
123305097246SAdrian Prantl     // compiled into the kernel or something -- then we don't want to load the
123405097246SAdrian Prantl     // kernel's text section at a different address.  Ignore this kext entry.
1235c1871475SJason Molenda     if (kext_summaries[new_kext].GetUUID().IsValid()
1236c1871475SJason Molenda         && m_kernel.GetUUID().IsValid()
1237c1871475SJason Molenda         && kext_summaries[new_kext].GetUUID() == m_kernel.GetUUID()) {
1238c1871475SJason Molenda       to_be_added[new_kext] = false;
1239c1871475SJason Molenda       break;
1240c1871475SJason Molenda     }
1241b9c1b51eSKate Stone     if (add_this_one) {
1242306bd0aaSJason Molenda       number_of_new_kexts_being_added++;
1243306bd0aaSJason Molenda     }
1244306bd0aaSJason Molenda   }
1245306bd0aaSJason Molenda 
1246b9c1b51eSKate Stone   if (number_of_new_kexts_being_added == 0 &&
1247b9c1b51eSKate Stone       number_of_old_kexts_being_removed == 0)
1248306bd0aaSJason Molenda     return true;
1249306bd0aaSJason Molenda 
125044d93782SGreg Clayton   Stream *s = m_process->GetTarget().GetDebugger().GetOutputFile().get();
1251b9c1b51eSKate Stone   if (s && load_kexts) {
1252b9c1b51eSKate Stone     if (number_of_new_kexts_being_added > 0 &&
1253b9c1b51eSKate Stone         number_of_old_kexts_being_removed > 0) {
1254b9c1b51eSKate Stone       s->Printf("Loading %d kext modules and unloading %d kext modules ",
1255b9c1b51eSKate Stone                 number_of_new_kexts_being_added,
1256b9c1b51eSKate Stone                 number_of_old_kexts_being_removed);
1257b9c1b51eSKate Stone     } else if (number_of_new_kexts_being_added > 0) {
1258306bd0aaSJason Molenda       s->Printf("Loading %d kext modules ", number_of_new_kexts_being_added);
1259b9c1b51eSKate Stone     } else if (number_of_old_kexts_being_removed > 0) {
1260b9c1b51eSKate Stone       s->Printf("Unloading %d kext modules ",
1261b9c1b51eSKate Stone                 number_of_old_kexts_being_removed);
1262306bd0aaSJason Molenda     }
1263306bd0aaSJason Molenda   }
1264306bd0aaSJason Molenda 
1265b9c1b51eSKate Stone   if (log) {
1266b9c1b51eSKate Stone     if (load_kexts) {
1267b9c1b51eSKate Stone       log->Printf("DynamicLoaderDarwinKernel::ParseKextSummaries: %d kexts "
1268b9c1b51eSKate Stone                   "added, %d kexts removed",
1269b9c1b51eSKate Stone                   number_of_new_kexts_being_added,
1270b9c1b51eSKate Stone                   number_of_old_kexts_being_removed);
1271b9c1b51eSKate Stone     } else {
1272b9c1b51eSKate Stone       log->Printf(
1273b9c1b51eSKate Stone           "DynamicLoaderDarwinKernel::ParseKextSummaries kext loading is "
1274b9c1b51eSKate Stone           "disabled, else would have %d kexts added, %d kexts removed",
1275b9c1b51eSKate Stone           number_of_new_kexts_being_added, number_of_old_kexts_being_removed);
12764da2e32fSJason Molenda     }
12774da2e32fSJason Molenda   }
12784da2e32fSJason Molenda 
1279b9c1b51eSKate Stone   if (number_of_new_kexts_being_added > 0) {
1280306bd0aaSJason Molenda     ModuleList loaded_module_list;
1281306bd0aaSJason Molenda 
1282306bd0aaSJason Molenda     const uint32_t num_of_new_kexts = kext_summaries.size();
1283b9c1b51eSKate Stone     for (uint32_t new_kext = 0; new_kext < num_of_new_kexts; new_kext++) {
1284b9c1b51eSKate Stone       if (to_be_added[new_kext] == true) {
1285306bd0aaSJason Molenda         KextImageInfo &image_info = kext_summaries[new_kext];
1286b9c1b51eSKate Stone         if (load_kexts) {
1287b9c1b51eSKate Stone           if (!image_info.LoadImageUsingMemoryModule(m_process)) {
1288306bd0aaSJason Molenda             image_info.LoadImageAtFileAddress(m_process);
12894da2e32fSJason Molenda           }
12904da2e32fSJason Molenda         }
1291306bd0aaSJason Molenda 
1292306bd0aaSJason Molenda         m_known_kexts.push_back(image_info);
1293306bd0aaSJason Molenda 
1294b9c1b51eSKate Stone         if (image_info.GetModule() &&
1295b9c1b51eSKate Stone             m_process->GetStopID() == image_info.GetProcessStopId())
1296306bd0aaSJason Molenda           loaded_module_list.AppendIfNeeded(image_info.GetModule());
1297d4bfbc9aSGreg Clayton 
12984da2e32fSJason Molenda         if (s && load_kexts)
129968b3607fSJason Molenda           s->Printf(".");
1300d4bfbc9aSGreg Clayton 
1301d4bfbc9aSGreg Clayton         if (log)
13025160ce5cSGreg Clayton           kext_summaries[new_kext].PutToLog(log);
1303d4bfbc9aSGreg Clayton       }
1304306bd0aaSJason Molenda     }
1305306bd0aaSJason Molenda     m_process->GetTarget().ModulesDidLoad(loaded_module_list);
1306306bd0aaSJason Molenda   }
1307306bd0aaSJason Molenda 
1308b9c1b51eSKate Stone   if (number_of_old_kexts_being_removed > 0) {
1309306bd0aaSJason Molenda     ModuleList loaded_module_list;
1310306bd0aaSJason Molenda     const uint32_t num_of_old_kexts = m_known_kexts.size();
1311b9c1b51eSKate Stone     for (uint32_t old_kext = 0; old_kext < num_of_old_kexts; old_kext++) {
1312306bd0aaSJason Molenda       ModuleList unloaded_module_list;
1313b9c1b51eSKate Stone       if (to_be_removed[old_kext]) {
1314306bd0aaSJason Molenda         KextImageInfo &image_info = m_known_kexts[old_kext];
1315306bd0aaSJason Molenda         // You can't unload the kernel.
1316b9c1b51eSKate Stone         if (!image_info.IsKernel()) {
1317b9c1b51eSKate Stone           if (image_info.GetModule()) {
1318306bd0aaSJason Molenda             unloaded_module_list.AppendIfNeeded(image_info.GetModule());
1319306bd0aaSJason Molenda           }
1320306bd0aaSJason Molenda           if (s)
1321306bd0aaSJason Molenda             s->Printf(".");
1322306bd0aaSJason Molenda           image_info.Clear();
1323b9c1b51eSKate Stone           // should pull it out of the KextImageInfos vector but that would
132405097246SAdrian Prantl           // mutate the list and invalidate the to_be_removed bool vector;
132505097246SAdrian Prantl           // leaving it in place once Cleared() is relatively harmless.
1326306bd0aaSJason Molenda         }
1327306bd0aaSJason Molenda       }
1328095eeaa0SGreg Clayton       m_process->GetTarget().ModulesDidUnload(unloaded_module_list, false);
1329306bd0aaSJason Molenda     }
1330306bd0aaSJason Molenda   }
1331306bd0aaSJason Molenda 
1332b9c1b51eSKate Stone   if (s && load_kexts) {
133368b3607fSJason Molenda     s->Printf(" done.\n");
133468b3607fSJason Molenda     s->Flush();
133568b3607fSJason Molenda   }
133668b3607fSJason Molenda 
1337306bd0aaSJason Molenda   return true;
1338d4bfbc9aSGreg Clayton }
1339d4bfbc9aSGreg Clayton 
1340b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::ReadKextSummaries(
1341b9c1b51eSKate Stone     const Address &kext_summary_addr, uint32_t image_infos_count,
1342b9c1b51eSKate Stone     KextImageInfo::collection &image_infos) {
1343d4bfbc9aSGreg Clayton   const ByteOrder endian = m_kernel.GetByteOrder();
1344d4bfbc9aSGreg Clayton   const uint32_t addr_size = m_kernel.GetAddressByteSize();
1345d4bfbc9aSGreg Clayton 
1346d4bfbc9aSGreg Clayton   image_infos.resize(image_infos_count);
1347d4bfbc9aSGreg Clayton   const size_t count = image_infos.size() * m_kext_summary_header.entry_size;
1348d4bfbc9aSGreg Clayton   DataBufferHeap data(count, 0);
134997206d57SZachary Turner   Status error;
1350d4bfbc9aSGreg Clayton 
1351d4bfbc9aSGreg Clayton   const bool prefer_file_cache = false;
1352b9c1b51eSKate Stone   const size_t bytes_read = m_process->GetTarget().ReadMemory(
1353b9c1b51eSKate Stone       kext_summary_addr, prefer_file_cache, data.GetBytes(), data.GetByteSize(),
1354d4bfbc9aSGreg Clayton       error);
1355b9c1b51eSKate Stone   if (bytes_read == count) {
1356d4bfbc9aSGreg Clayton 
1357b9c1b51eSKate Stone     DataExtractor extractor(data.GetBytes(), data.GetByteSize(), endian,
1358b9c1b51eSKate Stone                             addr_size);
1359d4bfbc9aSGreg Clayton     uint32_t i = 0;
1360d4bfbc9aSGreg Clayton     for (uint32_t kext_summary_offset = 0;
1361b9c1b51eSKate Stone          i < image_infos.size() &&
1362b9c1b51eSKate Stone          extractor.ValidOffsetForDataOfSize(kext_summary_offset,
1363b9c1b51eSKate Stone                                             m_kext_summary_header.entry_size);
1364b9c1b51eSKate Stone          ++i, kext_summary_offset += m_kext_summary_header.entry_size) {
1365c7bece56SGreg Clayton       lldb::offset_t offset = kext_summary_offset;
1366b9c1b51eSKate Stone       const void *name_data =
1367b9c1b51eSKate Stone           extractor.GetData(&offset, KERNEL_MODULE_MAX_NAME);
1368d4bfbc9aSGreg Clayton       if (name_data == NULL)
1369d4bfbc9aSGreg Clayton         break;
1370306bd0aaSJason Molenda       image_infos[i].SetName((const char *)name_data);
13712f93fd1fSPavel Labath       UUID uuid = UUID::fromOptionalData(extractor.GetData(&offset, 16), 16);
1372306bd0aaSJason Molenda       image_infos[i].SetUUID(uuid);
1373306bd0aaSJason Molenda       image_infos[i].SetLoadAddress(extractor.GetU64(&offset));
1374306bd0aaSJason Molenda       image_infos[i].SetSize(extractor.GetU64(&offset));
1375d4bfbc9aSGreg Clayton     }
1376d4bfbc9aSGreg Clayton     if (i < image_infos.size())
1377d4bfbc9aSGreg Clayton       image_infos.resize(i);
1378b9c1b51eSKate Stone   } else {
1379d4bfbc9aSGreg Clayton     image_infos.clear();
1380d4bfbc9aSGreg Clayton   }
1381d4bfbc9aSGreg Clayton   return image_infos.size();
1382d4bfbc9aSGreg Clayton }
1383d4bfbc9aSGreg Clayton 
1384b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ReadAllKextSummaries() {
138516ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1386d4bfbc9aSGreg Clayton 
1387b9c1b51eSKate Stone   if (ReadKextSummaryHeader()) {
1388b9c1b51eSKate Stone     if (m_kext_summary_header.entry_count > 0 &&
1389b9c1b51eSKate Stone         m_kext_summary_header_addr.IsValid()) {
1390d4bfbc9aSGreg Clayton       Address summary_addr(m_kext_summary_header_addr);
1391d4bfbc9aSGreg Clayton       summary_addr.Slide(m_kext_summary_header.GetSize());
1392b9c1b51eSKate Stone       if (!ParseKextSummaries(summary_addr,
1393b9c1b51eSKate Stone                               m_kext_summary_header.entry_count)) {
1394306bd0aaSJason Molenda         m_known_kexts.clear();
1395d4bfbc9aSGreg Clayton       }
1396d4bfbc9aSGreg Clayton       return true;
1397d4bfbc9aSGreg Clayton     }
1398d4bfbc9aSGreg Clayton   }
1399d4bfbc9aSGreg Clayton   return false;
1400d4bfbc9aSGreg Clayton }
1401d4bfbc9aSGreg Clayton 
1402d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1403d4bfbc9aSGreg Clayton // Dump an image info structure to the file handle provided.
1404d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1405b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::PutToLog(Log *log) const {
1406b9c1b51eSKate Stone   if (m_load_address == LLDB_INVALID_ADDRESS) {
1407fbb14285SPavel Labath     LLDB_LOG(log, "uuid={0} name=\"{1}\" (UNLOADED)", m_uuid.GetAsString(),
1408fbb14285SPavel Labath              m_name);
1409b9c1b51eSKate Stone   } else {
1410fbb14285SPavel Labath     LLDB_LOG(log, "addr={0:x+16} size={1:x+16} uuid={2} name=\"{3}\"",
1411fbb14285SPavel Labath         m_load_address, m_size, m_uuid.GetAsString(), m_name);
1412d4bfbc9aSGreg Clayton   }
1413d4bfbc9aSGreg Clayton }
1414d4bfbc9aSGreg Clayton 
1415d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
141605097246SAdrian Prantl // Dump the _dyld_all_image_infos members and all current image infos that we
141705097246SAdrian Prantl // have parsed to the file handle provided.
1418d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1419b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PutToLog(Log *log) const {
1420d4bfbc9aSGreg Clayton   if (log == NULL)
1421d4bfbc9aSGreg Clayton     return;
1422d4bfbc9aSGreg Clayton 
142316ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1424b9c1b51eSKate Stone   log->Printf("gLoadedKextSummaries = 0x%16.16" PRIx64
1425b9c1b51eSKate Stone               " { version=%u, entry_size=%u, entry_count=%u }",
1426d4bfbc9aSGreg Clayton               m_kext_summary_header_addr.GetFileAddress(),
1427b9c1b51eSKate Stone               m_kext_summary_header.version, m_kext_summary_header.entry_size,
1428d4bfbc9aSGreg Clayton               m_kext_summary_header.entry_count);
1429d4bfbc9aSGreg Clayton 
1430d4bfbc9aSGreg Clayton   size_t i;
1431306bd0aaSJason Molenda   const size_t count = m_known_kexts.size();
1432b9c1b51eSKate Stone   if (count > 0) {
1433d4bfbc9aSGreg Clayton     log->PutCString("Loaded:");
1434d4bfbc9aSGreg Clayton     for (i = 0; i < count; i++)
1435306bd0aaSJason Molenda       m_known_kexts[i].PutToLog(log);
1436d4bfbc9aSGreg Clayton   }
1437d4bfbc9aSGreg Clayton }
1438d4bfbc9aSGreg Clayton 
1439b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PrivateInitialize(Process *process) {
1440b9c1b51eSKate Stone   DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n",
1441b9c1b51eSKate Stone                __FUNCTION__, StateAsCString(m_process->GetState()));
1442d4bfbc9aSGreg Clayton   Clear(true);
1443d4bfbc9aSGreg Clayton   m_process = process;
1444d4bfbc9aSGreg Clayton }
1445d4bfbc9aSGreg Clayton 
1446b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::SetNotificationBreakpointIfNeeded() {
1447b9c1b51eSKate Stone   if (m_break_id == LLDB_INVALID_BREAK_ID && m_kernel.GetModule()) {
1448b9c1b51eSKate Stone     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n",
1449b9c1b51eSKate Stone                  __FUNCTION__, StateAsCString(m_process->GetState()));
1450d4bfbc9aSGreg Clayton 
1451a8558b62SJim Ingham     const bool internal_bp = true;
1452eb023e75SGreg Clayton     const bool hardware = false;
1453d4bfbc9aSGreg Clayton     const LazyBool skip_prologue = eLazyBoolNo;
1454969795f1SJim Ingham     FileSpecList module_spec_list;
1455306bd0aaSJason Molenda     module_spec_list.Append(m_kernel.GetModule()->GetFileSpec());
1456b9c1b51eSKate Stone     Breakpoint *bp =
1457b9c1b51eSKate Stone         m_process->GetTarget()
1458b9c1b51eSKate Stone             .CreateBreakpoint(&module_spec_list, NULL,
1459d4bfbc9aSGreg Clayton                               "OSKextLoadedKextSummariesUpdated",
1460b9c1b51eSKate Stone                               eFunctionNameTypeFull, eLanguageTypeUnknown, 0,
1461b9c1b51eSKate Stone                               skip_prologue, internal_bp, hardware)
1462b9c1b51eSKate Stone             .get();
1463d4bfbc9aSGreg Clayton 
1464b9c1b51eSKate Stone     bp->SetCallback(DynamicLoaderDarwinKernel::BreakpointHitCallback, this,
1465b9c1b51eSKate Stone                     true);
1466d4bfbc9aSGreg Clayton     m_break_id = bp->GetID();
1467d4bfbc9aSGreg Clayton   }
1468d4bfbc9aSGreg Clayton }
1469d4bfbc9aSGreg Clayton 
1470d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1471d4bfbc9aSGreg Clayton // Member function that gets called when the process state changes.
1472d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1473b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PrivateProcessStateChanged(Process *process,
1474b9c1b51eSKate Stone                                                            StateType state) {
1475b9c1b51eSKate Stone   DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s(%s)\n", __FUNCTION__,
1476b9c1b51eSKate Stone                StateAsCString(state));
1477b9c1b51eSKate Stone   switch (state) {
1478d4bfbc9aSGreg Clayton   case eStateConnected:
1479d4bfbc9aSGreg Clayton   case eStateAttaching:
1480d4bfbc9aSGreg Clayton   case eStateLaunching:
1481d4bfbc9aSGreg Clayton   case eStateInvalid:
1482d4bfbc9aSGreg Clayton   case eStateUnloaded:
1483d4bfbc9aSGreg Clayton   case eStateExited:
1484d4bfbc9aSGreg Clayton   case eStateDetached:
1485d4bfbc9aSGreg Clayton     Clear(false);
1486d4bfbc9aSGreg Clayton     break;
1487d4bfbc9aSGreg Clayton 
1488d4bfbc9aSGreg Clayton   case eStateStopped:
1489d4bfbc9aSGreg Clayton     UpdateIfNeeded();
1490d4bfbc9aSGreg Clayton     break;
1491d4bfbc9aSGreg Clayton 
1492d4bfbc9aSGreg Clayton   case eStateRunning:
1493d4bfbc9aSGreg Clayton   case eStateStepping:
1494d4bfbc9aSGreg Clayton   case eStateCrashed:
1495d4bfbc9aSGreg Clayton   case eStateSuspended:
1496d4bfbc9aSGreg Clayton     break;
1497d4bfbc9aSGreg Clayton   }
1498d4bfbc9aSGreg Clayton }
1499d4bfbc9aSGreg Clayton 
1500d4bfbc9aSGreg Clayton ThreadPlanSP
1501b9c1b51eSKate Stone DynamicLoaderDarwinKernel::GetStepThroughTrampolinePlan(Thread &thread,
1502b9c1b51eSKate Stone                                                         bool stop_others) {
1503d4bfbc9aSGreg Clayton   ThreadPlanSP thread_plan_sp;
15045160ce5cSGreg Clayton   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_STEP));
1505d4bfbc9aSGreg Clayton   if (log)
1506d4bfbc9aSGreg Clayton     log->Printf("Could not find symbol for step through.");
1507d4bfbc9aSGreg Clayton   return thread_plan_sp;
1508d4bfbc9aSGreg Clayton }
1509d4bfbc9aSGreg Clayton 
151097206d57SZachary Turner Status DynamicLoaderDarwinKernel::CanLoadImage() {
151197206d57SZachary Turner   Status error;
1512b9c1b51eSKate Stone   error.SetErrorString(
1513b9c1b51eSKate Stone       "always unsafe to load or unload shared libraries in the darwin kernel");
1514d4bfbc9aSGreg Clayton   return error;
1515d4bfbc9aSGreg Clayton }
1516d4bfbc9aSGreg Clayton 
1517b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Initialize() {
1518d4bfbc9aSGreg Clayton   PluginManager::RegisterPlugin(GetPluginNameStatic(),
1519b9c1b51eSKate Stone                                 GetPluginDescriptionStatic(), CreateInstance,
1520e8cd0c98SGreg Clayton                                 DebuggerInitialize);
1521d4bfbc9aSGreg Clayton }
1522d4bfbc9aSGreg Clayton 
1523b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Terminate() {
1524d4bfbc9aSGreg Clayton   PluginManager::UnregisterPlugin(CreateInstance);
1525d4bfbc9aSGreg Clayton }
1526d4bfbc9aSGreg Clayton 
1527b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DebuggerInitialize(
1528b9c1b51eSKate Stone     lldb_private::Debugger &debugger) {
1529b9c1b51eSKate Stone   if (!PluginManager::GetSettingForDynamicLoaderPlugin(
1530b9c1b51eSKate Stone           debugger, DynamicLoaderDarwinKernelProperties::GetSettingName())) {
1531e8cd0c98SGreg Clayton     const bool is_global_setting = true;
1532b9c1b51eSKate Stone     PluginManager::CreateSettingForDynamicLoaderPlugin(
1533b9c1b51eSKate Stone         debugger, GetGlobalProperties()->GetValueProperties(),
1534e8cd0c98SGreg Clayton         ConstString("Properties for the DynamicLoaderDarwinKernel plug-in."),
1535e8cd0c98SGreg Clayton         is_global_setting);
1536e8cd0c98SGreg Clayton   }
1537e8cd0c98SGreg Clayton }
1538d4bfbc9aSGreg Clayton 
1539b9c1b51eSKate Stone lldb_private::ConstString DynamicLoaderDarwinKernel::GetPluginNameStatic() {
154057abc5d6SGreg Clayton   static ConstString g_name("darwin-kernel");
154157abc5d6SGreg Clayton   return g_name;
1542d4bfbc9aSGreg Clayton }
1543d4bfbc9aSGreg Clayton 
1544b9c1b51eSKate Stone const char *DynamicLoaderDarwinKernel::GetPluginDescriptionStatic() {
1545b9c1b51eSKate Stone   return "Dynamic loader plug-in that watches for shared library loads/unloads "
1546b9c1b51eSKate Stone          "in the MacOSX kernel.";
1547d4bfbc9aSGreg Clayton }
1548d4bfbc9aSGreg Clayton 
1549d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1550d4bfbc9aSGreg Clayton // PluginInterface protocol
1551d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1552b9c1b51eSKate Stone lldb_private::ConstString DynamicLoaderDarwinKernel::GetPluginName() {
1553d4bfbc9aSGreg Clayton   return GetPluginNameStatic();
1554d4bfbc9aSGreg Clayton }
1555d4bfbc9aSGreg Clayton 
1556b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::GetPluginVersion() { return 1; }
1557d4bfbc9aSGreg Clayton 
15581f746071SGreg Clayton lldb::ByteOrder
1559b9c1b51eSKate Stone DynamicLoaderDarwinKernel::GetByteOrderFromMagic(uint32_t magic) {
1560b9c1b51eSKate Stone   switch (magic) {
1561510938e5SCharles Davis   case llvm::MachO::MH_MAGIC:
1562510938e5SCharles Davis   case llvm::MachO::MH_MAGIC_64:
15639ccb970fSBruce Mitchener     return endian::InlHostByteOrder();
15641f746071SGreg Clayton 
1565510938e5SCharles Davis   case llvm::MachO::MH_CIGAM:
1566510938e5SCharles Davis   case llvm::MachO::MH_CIGAM_64:
15679ccb970fSBruce Mitchener     if (endian::InlHostByteOrder() == lldb::eByteOrderBig)
15681f746071SGreg Clayton       return lldb::eByteOrderLittle;
15691f746071SGreg Clayton     else
15701f746071SGreg Clayton       return lldb::eByteOrderBig;
15711f746071SGreg Clayton 
15721f746071SGreg Clayton   default:
15731f746071SGreg Clayton     break;
15741f746071SGreg Clayton   }
15751f746071SGreg Clayton   return lldb::eByteOrderInvalid;
15761f746071SGreg Clayton }
1577