1b9c1b51eSKate Stone //===-- DynamicLoaderDarwinKernel.cpp -----------------------------*- C++
2b9c1b51eSKate Stone //-*-===//
3d4bfbc9aSGreg Clayton //
4d4bfbc9aSGreg Clayton //                     The LLVM Compiler Infrastructure
5d4bfbc9aSGreg Clayton //
6d4bfbc9aSGreg Clayton // This file is distributed under the University of Illinois Open Source
7d4bfbc9aSGreg Clayton // License. See LICENSE.TXT for details.
8d4bfbc9aSGreg Clayton //
9d4bfbc9aSGreg Clayton //===----------------------------------------------------------------------===//
10d4bfbc9aSGreg Clayton 
1146d005dbSJim Ingham #include "lldb/Utility/SafeMachO.h"
12eadcca97SGreg Clayton 
13b9c1b51eSKate Stone #include "Plugins/Platform/MacOSX/PlatformDarwinKernel.h"
14d4bfbc9aSGreg Clayton #include "lldb/Breakpoint/StoppointCallbackContext.h"
15d4bfbc9aSGreg Clayton #include "lldb/Core/Debugger.h"
16d4bfbc9aSGreg Clayton #include "lldb/Core/Module.h"
171f746071SGreg Clayton #include "lldb/Core/ModuleSpec.h"
18d4bfbc9aSGreg Clayton #include "lldb/Core/PluginManager.h"
191f746071SGreg Clayton #include "lldb/Core/Section.h"
20d4bfbc9aSGreg Clayton #include "lldb/Core/State.h"
2144d93782SGreg Clayton #include "lldb/Core/StreamFile.h"
2268b3607fSJason Molenda #include "lldb/Host/Symbols.h"
2341204d09SIlia K #include "lldb/Interpreter/OptionValueProperties.h"
24d4bfbc9aSGreg Clayton #include "lldb/Symbol/ObjectFile.h"
25519b0816SJim Ingham #include "lldb/Target/OperatingSystem.h"
26d4bfbc9aSGreg Clayton #include "lldb/Target/RegisterContext.h"
27b57e4a1bSJason Molenda #include "lldb/Target/StackFrame.h"
28d4bfbc9aSGreg Clayton #include "lldb/Target/Target.h"
29d4bfbc9aSGreg Clayton #include "lldb/Target/Thread.h"
30d4bfbc9aSGreg Clayton #include "lldb/Target/ThreadPlanRunToAddress.h"
31666cc0b2SZachary Turner #include "lldb/Utility/DataBuffer.h"
32666cc0b2SZachary Turner #include "lldb/Utility/DataBufferHeap.h"
336f9e6901SZachary Turner #include "lldb/Utility/Log.h"
34d4bfbc9aSGreg Clayton 
35944b828aSGreg Clayton #include "DynamicLoaderDarwinKernel.h"
36d4bfbc9aSGreg Clayton 
37d4bfbc9aSGreg Clayton //#define ENABLE_DEBUG_PRINTF // COMMENT THIS LINE OUT PRIOR TO CHECKIN
38d4bfbc9aSGreg Clayton #ifdef ENABLE_DEBUG_PRINTF
39d4bfbc9aSGreg Clayton #include <stdio.h>
40d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...) printf(fmt, ##__VA_ARGS__)
41d4bfbc9aSGreg Clayton #else
42d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...)
43d4bfbc9aSGreg Clayton #endif
44d4bfbc9aSGreg Clayton 
45d4bfbc9aSGreg Clayton using namespace lldb;
46d4bfbc9aSGreg Clayton using namespace lldb_private;
47d4bfbc9aSGreg Clayton 
4805097246SAdrian Prantl // Progressively greater amounts of scanning we will allow For some targets
4905097246SAdrian Prantl // very early in startup, we can't do any random reads of memory or we can
5005097246SAdrian Prantl // crash the device so a setting is needed that can completely disable the
5105097246SAdrian Prantl // KASLR scans.
526ba6d3d1SJason Molenda 
53b9c1b51eSKate Stone enum KASLRScanType {
546ba6d3d1SJason Molenda   eKASLRScanNone = 0,        // No reading into the inferior at all
55b9c1b51eSKate Stone   eKASLRScanLowgloAddresses, // Check one word of memory for a possible kernel
56b9c1b51eSKate Stone                              // addr, then see if a kernel is there
57b9c1b51eSKate Stone   eKASLRScanNearPC, // Scan backwards from the current $pc looking for kernel;
58b9c1b51eSKate Stone                     // checking at 96 locations total
59b9c1b51eSKate Stone   eKASLRScanExhaustiveScan // Scan through the entire possible kernel address
60b9c1b51eSKate Stone                            // range looking for a kernel
616ba6d3d1SJason Molenda };
626ba6d3d1SJason Molenda 
63b9c1b51eSKate Stone OptionEnumValueElement g_kaslr_kernel_scan_enum_values[] = {
64b9c1b51eSKate Stone     {eKASLRScanNone, "none",
65b9c1b51eSKate Stone      "Do not read memory looking for a Darwin kernel when attaching."},
66b9c1b51eSKate Stone     {eKASLRScanLowgloAddresses, "basic", "Check for the Darwin kernel's load "
67b9c1b51eSKate Stone                                          "addr in the lowglo page "
68b9c1b51eSKate Stone                                          "(boot-args=debug) only."},
69b9c1b51eSKate Stone     {eKASLRScanNearPC, "fast-scan", "Scan near the pc value on attach to find "
70b9c1b51eSKate Stone                                     "the Darwin kernel's load address."},
71b9c1b51eSKate Stone     {eKASLRScanExhaustiveScan, "exhaustive-scan",
72b9c1b51eSKate Stone      "Scan through the entire potential address range of Darwin kernel (only "
73b9c1b51eSKate Stone      "on 32-bit targets)."},
74b9c1b51eSKate Stone     {0, NULL, NULL}};
756ba6d3d1SJason Molenda 
76b9c1b51eSKate Stone static PropertyDefinition g_properties[] = {
77b9c1b51eSKate Stone     {"load-kexts", OptionValue::eTypeBoolean, true, true, NULL, NULL,
78b9c1b51eSKate Stone      "Automatically loads kext images when attaching to a kernel."},
79b9c1b51eSKate Stone     {"scan-type", OptionValue::eTypeEnum, true, eKASLRScanNearPC, NULL,
80b9c1b51eSKate Stone      g_kaslr_kernel_scan_enum_values, "Control how many reads lldb will make "
81b9c1b51eSKate Stone                                       "while searching for a Darwin kernel on "
82b9c1b51eSKate Stone                                       "attach."},
83b9c1b51eSKate Stone     {NULL, OptionValue::eTypeInvalid, false, 0, NULL, NULL, NULL}};
84e8cd0c98SGreg Clayton 
85b9c1b51eSKate Stone enum { ePropertyLoadKexts, ePropertyScanType };
86e8cd0c98SGreg Clayton 
87b9c1b51eSKate Stone class DynamicLoaderDarwinKernelProperties : public Properties {
88e8cd0c98SGreg Clayton public:
89b9c1b51eSKate Stone   static ConstString &GetSettingName() {
90468ea4ebSGreg Clayton     static ConstString g_setting_name("darwin-kernel");
91e8cd0c98SGreg Clayton     return g_setting_name;
92e8cd0c98SGreg Clayton   }
93e8cd0c98SGreg Clayton 
94b9c1b51eSKate Stone   DynamicLoaderDarwinKernelProperties() : Properties() {
95e8cd0c98SGreg Clayton     m_collection_sp.reset(new OptionValueProperties(GetSettingName()));
96e8cd0c98SGreg Clayton     m_collection_sp->Initialize(g_properties);
97e8cd0c98SGreg Clayton   }
98e8cd0c98SGreg Clayton 
99b9c1b51eSKate Stone   virtual ~DynamicLoaderDarwinKernelProperties() {}
100e8cd0c98SGreg Clayton 
101b9c1b51eSKate Stone   bool GetLoadKexts() const {
10266763eedSGreg Clayton     const uint32_t idx = ePropertyLoadKexts;
103b9c1b51eSKate Stone     return m_collection_sp->GetPropertyAtIndexAsBoolean(
104b9c1b51eSKate Stone         NULL, idx, g_properties[idx].default_uint_value != 0);
105e8cd0c98SGreg Clayton   }
106e8cd0c98SGreg Clayton 
107b9c1b51eSKate Stone   KASLRScanType GetScanType() const {
1086ba6d3d1SJason Molenda     const uint32_t idx = ePropertyScanType;
109b9c1b51eSKate Stone     return (KASLRScanType)m_collection_sp->GetPropertyAtIndexAsEnumeration(
110b9c1b51eSKate Stone         NULL, idx, g_properties[idx].default_uint_value);
1116ba6d3d1SJason Molenda   }
112e8cd0c98SGreg Clayton };
113e8cd0c98SGreg Clayton 
114b9c1b51eSKate Stone typedef std::shared_ptr<DynamicLoaderDarwinKernelProperties>
115b9c1b51eSKate Stone     DynamicLoaderDarwinKernelPropertiesSP;
116e8cd0c98SGreg Clayton 
117b9c1b51eSKate Stone static const DynamicLoaderDarwinKernelPropertiesSP &GetGlobalProperties() {
118e8cd0c98SGreg Clayton   static DynamicLoaderDarwinKernelPropertiesSP g_settings_sp;
119e8cd0c98SGreg Clayton   if (!g_settings_sp)
120e8cd0c98SGreg Clayton     g_settings_sp.reset(new DynamicLoaderDarwinKernelProperties());
121e8cd0c98SGreg Clayton   return g_settings_sp;
122e8cd0c98SGreg Clayton }
123e8cd0c98SGreg Clayton 
124d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
12505097246SAdrian Prantl // Create an instance of this class. This function is filled into the plugin
12605097246SAdrian Prantl // info class that gets handed out by the plugin factory and allows the lldb to
12705097246SAdrian Prantl // instantiate an instance of this class.
128d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
129b9c1b51eSKate Stone DynamicLoader *DynamicLoaderDarwinKernel::CreateInstance(Process *process,
130b9c1b51eSKate Stone                                                          bool force) {
131b9c1b51eSKate Stone   if (!force) {
13205097246SAdrian Prantl     // If the user provided an executable binary and it is not a kernel, this
13305097246SAdrian Prantl     // plugin should not create an instance.
134d4bfbc9aSGreg Clayton     Module *exe_module = process->GetTarget().GetExecutableModulePointer();
135b9c1b51eSKate Stone     if (exe_module) {
136d4bfbc9aSGreg Clayton       ObjectFile *object_file = exe_module->GetObjectFile();
137b9c1b51eSKate Stone       if (object_file) {
138b9c1b51eSKate Stone         if (object_file->GetStrata() != ObjectFile::eStrataKernel) {
1396ba6d3d1SJason Molenda           return NULL;
1406ba6d3d1SJason Molenda         }
141d4bfbc9aSGreg Clayton       }
142d4bfbc9aSGreg Clayton     }
143d4bfbc9aSGreg Clayton 
1446ba6d3d1SJason Molenda     // If the target's architecture does not look like an Apple environment,
1456ba6d3d1SJason Molenda     // this plugin should not create an instance.
146b9c1b51eSKate Stone     const llvm::Triple &triple_ref =
147b9c1b51eSKate Stone         process->GetTarget().GetArchitecture().GetTriple();
148b9c1b51eSKate Stone     switch (triple_ref.getOS()) {
14970512317SGreg Clayton     case llvm::Triple::Darwin:
15070512317SGreg Clayton     case llvm::Triple::MacOSX:
15170512317SGreg Clayton     case llvm::Triple::IOS:
152a814f704SJason Molenda     case llvm::Triple::TvOS:
153a814f704SJason Molenda     case llvm::Triple::WatchOS:
154b9c1b51eSKate Stone       if (triple_ref.getVendor() != llvm::Triple::Apple) {
1556ba6d3d1SJason Molenda         return NULL;
1566ba6d3d1SJason Molenda       }
1576ba6d3d1SJason Molenda       break;
15805097246SAdrian Prantl     // If we have triple like armv7-unknown-unknown, we should try looking for
15905097246SAdrian Prantl     // a Darwin kernel.
1606ba6d3d1SJason Molenda     case llvm::Triple::UnknownOS:
16170512317SGreg Clayton       break;
16270512317SGreg Clayton     default:
1636ba6d3d1SJason Molenda       return NULL;
16470512317SGreg Clayton       break;
16570512317SGreg Clayton     }
166d4bfbc9aSGreg Clayton   }
1676ba6d3d1SJason Molenda 
168b9c1b51eSKate Stone   // At this point if there is an ExecutableModule, it is a kernel and the
16905097246SAdrian Prantl   // Target is some variant of an Apple system. If the Process hasn't provided
17005097246SAdrian Prantl   // the kernel load address, we need to look around in memory to find it.
1716ba6d3d1SJason Molenda 
172c44bcec6SGreg Clayton   const addr_t kernel_load_address = SearchForDarwinKernel(process);
173b9c1b51eSKate Stone   if (CheckForKernelImageAtAddress(kernel_load_address, process).IsValid()) {
17490ff7911SEwan Crawford     process->SetCanRunCode(false);
175503d0181SJason Molenda     return new DynamicLoaderDarwinKernel(process, kernel_load_address);
176503d0181SJason Molenda   }
177503d0181SJason Molenda   return NULL;
178503d0181SJason Molenda }
179503d0181SJason Molenda 
180503d0181SJason Molenda lldb::addr_t
181b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForDarwinKernel(Process *process) {
1826ba6d3d1SJason Molenda   addr_t kernel_load_address = process->GetImageInfoAddress();
183b9c1b51eSKate Stone   if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1846ba6d3d1SJason Molenda     kernel_load_address = SearchForKernelAtSameLoadAddr(process);
185b9c1b51eSKate Stone     if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1866ba6d3d1SJason Molenda       kernel_load_address = SearchForKernelWithDebugHints(process);
187b9c1b51eSKate Stone       if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1886ba6d3d1SJason Molenda         kernel_load_address = SearchForKernelNearPC(process);
189b9c1b51eSKate Stone         if (kernel_load_address == LLDB_INVALID_ADDRESS) {
1906ba6d3d1SJason Molenda           kernel_load_address = SearchForKernelViaExhaustiveSearch(process);
1916ba6d3d1SJason Molenda         }
1926ba6d3d1SJason Molenda       }
1936ba6d3d1SJason Molenda     }
194d4bfbc9aSGreg Clayton   }
195503d0181SJason Molenda   return kernel_load_address;
196d4bfbc9aSGreg Clayton }
197d4bfbc9aSGreg Clayton 
198d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
19905097246SAdrian Prantl // Check if the kernel binary is loaded in memory without a slide. First verify
20005097246SAdrian Prantl // that the ExecutableModule is a kernel before we proceed. Returns the address
20105097246SAdrian Prantl // of the kernel if one was found, else LLDB_INVALID_ADDRESS.
2026ba6d3d1SJason Molenda //----------------------------------------------------------------------
2036ba6d3d1SJason Molenda lldb::addr_t
204b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelAtSameLoadAddr(Process *process) {
2056ba6d3d1SJason Molenda   Module *exe_module = process->GetTarget().GetExecutableModulePointer();
2066ba6d3d1SJason Molenda   if (exe_module == NULL)
2076ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2086ba6d3d1SJason Molenda 
2096ba6d3d1SJason Molenda   ObjectFile *exe_objfile = exe_module->GetObjectFile();
2106ba6d3d1SJason Molenda   if (exe_objfile == NULL)
2116ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2126ba6d3d1SJason Molenda 
213b9c1b51eSKate Stone   if (exe_objfile->GetType() != ObjectFile::eTypeExecutable ||
214b9c1b51eSKate Stone       exe_objfile->GetStrata() != ObjectFile::eStrataKernel)
2156ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2166ba6d3d1SJason Molenda 
2176ba6d3d1SJason Molenda   if (!exe_objfile->GetHeaderAddress().IsValid())
2186ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2196ba6d3d1SJason Molenda 
220b9c1b51eSKate Stone   if (CheckForKernelImageAtAddress(
221b9c1b51eSKate Stone           exe_objfile->GetHeaderAddress().GetFileAddress(), process) ==
222b9c1b51eSKate Stone       exe_module->GetUUID())
2236ba6d3d1SJason Molenda     return exe_objfile->GetHeaderAddress().GetFileAddress();
2246ba6d3d1SJason Molenda 
2256ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
2266ba6d3d1SJason Molenda }
2276ba6d3d1SJason Molenda 
2286ba6d3d1SJason Molenda //----------------------------------------------------------------------
229b9c1b51eSKate Stone // If the debug flag is included in the boot-args nvram setting, the kernel's
23005097246SAdrian Prantl // load address will be noted in the lowglo page at a fixed address Returns the
23105097246SAdrian Prantl // address of the kernel if one was found, else LLDB_INVALID_ADDRESS.
2326ba6d3d1SJason Molenda //----------------------------------------------------------------------
2336ba6d3d1SJason Molenda lldb::addr_t
234b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelWithDebugHints(Process *process) {
2356ba6d3d1SJason Molenda   if (GetGlobalProperties()->GetScanType() == eKASLRScanNone)
2366ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2376ba6d3d1SJason Molenda 
23897206d57SZachary Turner   Status read_err;
239b9c1b51eSKate Stone   addr_t kernel_addresses_64[] = {
240b9c1b51eSKate Stone       0xfffffff000004010ULL, // newest arm64 devices
24157086ed7SJason Molenda       0xffffff8000004010ULL, // 2014-2015-ish arm64 devices
24257086ed7SJason Molenda       0xffffff8000002010ULL, // oldest arm64 devices
243880988afSJason Molenda       LLDB_INVALID_ADDRESS};
24461a8e53bSJason Molenda   addr_t kernel_addresses_32[] = {0xffff0110, // 2016 and earlier armv7 devices
24561a8e53bSJason Molenda                                   0xffff1010,
24661a8e53bSJason Molenda                                   LLDB_INVALID_ADDRESS};
2472b0a7be9SJason Molenda 
2482b0a7be9SJason Molenda   uint8_t uval[8];
2492b0a7be9SJason Molenda   if (process->GetAddressByteSize() == 8) {
250b9c1b51eSKate Stone   for (size_t i = 0; kernel_addresses_64[i] != LLDB_INVALID_ADDRESS; i++) {
2512b0a7be9SJason Molenda       if (process->ReadMemoryFromInferior (kernel_addresses_64[i], uval, 8, read_err) == 8)
2522b0a7be9SJason Molenda       {
2532b0a7be9SJason Molenda           DataExtractor data (&uval, 8, process->GetByteOrder(), process->GetAddressByteSize());
2542b0a7be9SJason Molenda           offset_t offset = 0;
2552b0a7be9SJason Molenda           uint64_t addr = data.GetU64 (&offset);
256b9c1b51eSKate Stone           if (CheckForKernelImageAtAddress(addr, process).IsValid()) {
257ec504238SJason Molenda               return addr;
258ec504238SJason Molenda           }
2596ba6d3d1SJason Molenda       }
2602b0a7be9SJason Molenda   }
2612b0a7be9SJason Molenda   }
262880988afSJason Molenda 
2632b0a7be9SJason Molenda   if (process->GetAddressByteSize() == 4) {
264b9c1b51eSKate Stone   for (size_t i = 0; kernel_addresses_32[i] != LLDB_INVALID_ADDRESS; i++) {
2652b0a7be9SJason Molenda       if (process->ReadMemoryFromInferior (kernel_addresses_32[i], uval, 4, read_err) == 4)
2662b0a7be9SJason Molenda       {
2672b0a7be9SJason Molenda           DataExtractor data (&uval, 4, process->GetByteOrder(), process->GetAddressByteSize());
2682b0a7be9SJason Molenda           offset_t offset = 0;
2692b0a7be9SJason Molenda           uint32_t addr = data.GetU32 (&offset);
270b9c1b51eSKate Stone           if (CheckForKernelImageAtAddress(addr, process).IsValid()) {
2716ba6d3d1SJason Molenda               return addr;
2726ba6d3d1SJason Molenda           }
2737dd29390SJason Molenda       }
2742b0a7be9SJason Molenda   }
2752b0a7be9SJason Molenda   }
276880988afSJason Molenda 
2776ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
2786ba6d3d1SJason Molenda }
2796ba6d3d1SJason Molenda 
2806ba6d3d1SJason Molenda //----------------------------------------------------------------------
28105097246SAdrian Prantl // If the kernel is currently executing when lldb attaches, and we don't have a
28205097246SAdrian Prantl // better way of finding the kernel's load address, try searching backwards
2836ba6d3d1SJason Molenda // from the current pc value looking for the kernel's Mach header in memory.
284b9c1b51eSKate Stone // Returns the address of the kernel if one was found, else
285b9c1b51eSKate Stone // LLDB_INVALID_ADDRESS.
2866ba6d3d1SJason Molenda //----------------------------------------------------------------------
2876ba6d3d1SJason Molenda lldb::addr_t
288b9c1b51eSKate Stone DynamicLoaderDarwinKernel::SearchForKernelNearPC(Process *process) {
289b9c1b51eSKate Stone   if (GetGlobalProperties()->GetScanType() == eKASLRScanNone ||
290b9c1b51eSKate Stone       GetGlobalProperties()->GetScanType() == eKASLRScanLowgloAddresses) {
2916ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2926ba6d3d1SJason Molenda   }
2936ba6d3d1SJason Molenda 
2946ba6d3d1SJason Molenda   ThreadSP thread = process->GetThreadList().GetSelectedThread();
2956ba6d3d1SJason Molenda   if (thread.get() == NULL)
2966ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2976ba6d3d1SJason Molenda   addr_t pc = thread->GetRegisterContext()->GetPC(LLDB_INVALID_ADDRESS);
2986ba6d3d1SJason Molenda 
2996ba6d3d1SJason Molenda   if (pc == LLDB_INVALID_ADDRESS)
3006ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3016ba6d3d1SJason Molenda 
302b9c1b51eSKate Stone   // The kernel will load at at one megabyte boundary (0x100000), or at that
30305097246SAdrian Prantl   // boundary plus an offset of one page (0x1000) or two, or four (0x4000),
30405097246SAdrian Prantl   // depending on the device.
3056ba6d3d1SJason Molenda 
306b9c1b51eSKate Stone   // Round the current pc down to the nearest one megabyte boundary - the place
307b9c1b51eSKate Stone   // where we will start searching.
3086ba6d3d1SJason Molenda   addr_t addr = pc & ~0xfffff;
3096ba6d3d1SJason Molenda 
310b9c1b51eSKate Stone   // Search backwards 32 megabytes, looking for the start of the kernel at each
311b9c1b51eSKate Stone   // one-megabyte boundary.
312b9c1b51eSKate Stone   for (int i = 0; i < 32; i++, addr -= 0x100000) {
3136ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr, process).IsValid())
3146ba6d3d1SJason Molenda       return addr;
3156ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x1000, process).IsValid())
3166ba6d3d1SJason Molenda       return addr + 0x1000;
3176ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x2000, process).IsValid())
3186ba6d3d1SJason Molenda       return addr + 0x2000;
319a02869deSJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x4000, process).IsValid())
320a02869deSJason Molenda       return addr + 0x4000;
3216ba6d3d1SJason Molenda   }
3226ba6d3d1SJason Molenda 
3236ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
3246ba6d3d1SJason Molenda }
3256ba6d3d1SJason Molenda 
3266ba6d3d1SJason Molenda //----------------------------------------------------------------------
32705097246SAdrian Prantl // Scan through the valid address range for a kernel binary. This is uselessly
32805097246SAdrian Prantl // slow in 64-bit environments so we don't even try it. This scan is not
32905097246SAdrian Prantl // enabled by default even for 32-bit targets. Returns the address of the
33005097246SAdrian Prantl // kernel if one was found, else LLDB_INVALID_ADDRESS.
3316ba6d3d1SJason Molenda //----------------------------------------------------------------------
332b9c1b51eSKate Stone lldb::addr_t DynamicLoaderDarwinKernel::SearchForKernelViaExhaustiveSearch(
333b9c1b51eSKate Stone     Process *process) {
334b9c1b51eSKate Stone   if (GetGlobalProperties()->GetScanType() != eKASLRScanExhaustiveScan) {
3356ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3366ba6d3d1SJason Molenda   }
3376ba6d3d1SJason Molenda 
3386ba6d3d1SJason Molenda   addr_t kernel_range_low, kernel_range_high;
339b9c1b51eSKate Stone   if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8) {
3406ba6d3d1SJason Molenda     kernel_range_low = 1ULL << 63;
3416ba6d3d1SJason Molenda     kernel_range_high = UINT64_MAX;
342b9c1b51eSKate Stone   } else {
3436ba6d3d1SJason Molenda     kernel_range_low = 1ULL << 31;
3446ba6d3d1SJason Molenda     kernel_range_high = UINT32_MAX;
3456ba6d3d1SJason Molenda   }
3466ba6d3d1SJason Molenda 
3476ba6d3d1SJason Molenda   // Stepping through memory at one-megabyte resolution looking for a kernel
3486ba6d3d1SJason Molenda   // rarely works (fast enough) with a 64-bit address space -- for now, let's
3496ba6d3d1SJason Molenda   // not even bother.  We may be attaching to something which *isn't* a kernel
3506ba6d3d1SJason Molenda   // and we don't want to spin for minutes on-end looking for a kernel.
3516ba6d3d1SJason Molenda   if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
3526ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3536ba6d3d1SJason Molenda 
3546ba6d3d1SJason Molenda   addr_t addr = kernel_range_low;
3556ba6d3d1SJason Molenda 
356b9c1b51eSKate Stone   while (addr >= kernel_range_low && addr < kernel_range_high) {
3576ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr, process).IsValid())
3586ba6d3d1SJason Molenda       return addr;
3596ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x1000, process).IsValid())
3606ba6d3d1SJason Molenda       return addr + 0x1000;
3616ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x2000, process).IsValid())
3626ba6d3d1SJason Molenda       return addr + 0x2000;
363a02869deSJason Molenda     if (CheckForKernelImageAtAddress(addr + 0x4000, process).IsValid())
364a02869deSJason Molenda       return addr + 0x4000;
3656ba6d3d1SJason Molenda     addr += 0x100000;
3666ba6d3d1SJason Molenda   }
3676ba6d3d1SJason Molenda   return LLDB_INVALID_ADDRESS;
3686ba6d3d1SJason Molenda }
3696ba6d3d1SJason Molenda 
3706ba6d3d1SJason Molenda //----------------------------------------------------------------------
3716ba6d3d1SJason Molenda // Given an address in memory, look to see if there is a kernel image at that
37205097246SAdrian Prantl // address. Returns a UUID; if a kernel was not found at that address,
37305097246SAdrian Prantl // UUID.IsValid() will be false.
3746ba6d3d1SJason Molenda //----------------------------------------------------------------------
3756ba6d3d1SJason Molenda lldb_private::UUID
376b9c1b51eSKate Stone DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress(lldb::addr_t addr,
377b9c1b51eSKate Stone                                                         Process *process) {
378ec504238SJason Molenda   Log *log(lldb_private::GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
3796ba6d3d1SJason Molenda   if (addr == LLDB_INVALID_ADDRESS)
3806ba6d3d1SJason Molenda     return UUID();
3816ba6d3d1SJason Molenda 
382ec504238SJason Molenda   if (log)
383b9c1b51eSKate Stone     log->Printf("DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress: "
384b9c1b51eSKate Stone                 "looking for kernel binary at 0x%" PRIx64,
385b9c1b51eSKate Stone                 addr);
386ec504238SJason Molenda 
387b9c1b51eSKate Stone   // First try a quick test -- read the first 4 bytes and see if there is a
38805097246SAdrian Prantl   // valid Mach-O magic field there (the first field of the
38905097246SAdrian Prantl   // mach_header/mach_header_64 struct).
3906ba6d3d1SJason Molenda 
39197206d57SZachary Turner   Status read_error;
3922b0a7be9SJason Molenda   uint8_t magicbuf[4];
3932b0a7be9SJason Molenda   if (process->ReadMemoryFromInferior (addr, magicbuf, sizeof (magicbuf), read_error) != sizeof (magicbuf))
3946ba6d3d1SJason Molenda       return UUID();
3952b0a7be9SJason Molenda 
3962b0a7be9SJason Molenda   const uint32_t magicks[] = { llvm::MachO::MH_MAGIC_64, llvm::MachO::MH_MAGIC, llvm::MachO::MH_CIGAM, llvm::MachO::MH_CIGAM_64};
3972b0a7be9SJason Molenda 
3982b0a7be9SJason Molenda   bool found_matching_pattern = false;
3990ca1dde5STaras Tsugrii   for (size_t i = 0; i < llvm::array_lengthof (magicks); i++)
4002b0a7be9SJason Molenda     if (::memcmp (magicbuf, &magicks[i], sizeof (magicbuf)) == 0)
4012b0a7be9SJason Molenda         found_matching_pattern = true;
4022b0a7be9SJason Molenda 
4032b0a7be9SJason Molenda   if (found_matching_pattern == false)
4042b0a7be9SJason Molenda       return UUID();
4056ba6d3d1SJason Molenda 
4066ba6d3d1SJason Molenda   // Read the mach header and see whether it looks like a kernel
4076ba6d3d1SJason Molenda   llvm::MachO::mach_header header;
408b9c1b51eSKate Stone   if (process->DoReadMemory(addr, &header, sizeof(header), read_error) !=
409b9c1b51eSKate Stone       sizeof(header))
4106ba6d3d1SJason Molenda     return UUID();
4116ba6d3d1SJason Molenda 
412510938e5SCharles Davis   if (header.magic == llvm::MachO::MH_CIGAM ||
413b9c1b51eSKate Stone       header.magic == llvm::MachO::MH_CIGAM_64) {
4146ba6d3d1SJason Molenda     header.magic = llvm::ByteSwap_32(header.magic);
4156ba6d3d1SJason Molenda     header.cputype = llvm::ByteSwap_32(header.cputype);
4166ba6d3d1SJason Molenda     header.cpusubtype = llvm::ByteSwap_32(header.cpusubtype);
4176ba6d3d1SJason Molenda     header.filetype = llvm::ByteSwap_32(header.filetype);
4186ba6d3d1SJason Molenda     header.ncmds = llvm::ByteSwap_32(header.ncmds);
4196ba6d3d1SJason Molenda     header.sizeofcmds = llvm::ByteSwap_32(header.sizeofcmds);
4206ba6d3d1SJason Molenda     header.flags = llvm::ByteSwap_32(header.flags);
4216ba6d3d1SJason Molenda   }
4226ba6d3d1SJason Molenda 
423b9c1b51eSKate Stone   // A kernel is an executable which does not have the dynamic link object flag
424b9c1b51eSKate Stone   // set.
425b9c1b51eSKate Stone   if (header.filetype == llvm::MachO::MH_EXECUTE &&
426b9c1b51eSKate Stone       (header.flags & llvm::MachO::MH_DYLDLINK) == 0) {
4276ba6d3d1SJason Molenda     // Create a full module to get the UUID
428b9c1b51eSKate Stone     ModuleSP memory_module_sp = process->ReadModuleFromMemory(
429b9c1b51eSKate Stone         FileSpec("temp_mach_kernel", false), addr);
4306ba6d3d1SJason Molenda     if (!memory_module_sp.get())
4316ba6d3d1SJason Molenda       return UUID();
4326ba6d3d1SJason Molenda 
4336ba6d3d1SJason Molenda     ObjectFile *exe_objfile = memory_module_sp->GetObjectFile();
434bc22c8d8SJason Molenda     if (exe_objfile == NULL) {
435bc22c8d8SJason Molenda       if (log)
436bc22c8d8SJason Molenda         log->Printf("DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress "
437bc22c8d8SJason Molenda                     "found a binary at 0x%" PRIx64
438bc22c8d8SJason Molenda                     " but could not create an object file from memory",
439bc22c8d8SJason Molenda                     addr);
4406ba6d3d1SJason Molenda       return UUID();
441bc22c8d8SJason Molenda     }
4426ba6d3d1SJason Molenda 
443b9c1b51eSKate Stone     if (exe_objfile->GetType() == ObjectFile::eTypeExecutable &&
444b9c1b51eSKate Stone         exe_objfile->GetStrata() == ObjectFile::eStrataKernel) {
445a4ce2538SJason Molenda       ArchSpec kernel_arch(eArchTypeMachO, header.cputype, header.cpusubtype);
446b9c1b51eSKate Stone       if (!process->GetTarget().GetArchitecture().IsCompatibleMatch(
447b9c1b51eSKate Stone               kernel_arch)) {
448a4ce2538SJason Molenda         process->GetTarget().SetArchitecture(kernel_arch);
449a4ce2538SJason Molenda       }
450bc22c8d8SJason Molenda       if (log) {
451bc22c8d8SJason Molenda         std::string uuid_str;
452bc22c8d8SJason Molenda         if (memory_module_sp->GetUUID().IsValid()) {
453bc22c8d8SJason Molenda           uuid_str = "with UUID ";
454bc22c8d8SJason Molenda           uuid_str += memory_module_sp->GetUUID().GetAsString();
455bc22c8d8SJason Molenda         } else {
456bc22c8d8SJason Molenda           uuid_str = "and no LC_UUID found in load commands ";
457bc22c8d8SJason Molenda         }
458bc22c8d8SJason Molenda         log->Printf(
459bc22c8d8SJason Molenda             "DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress: "
460bc22c8d8SJason Molenda             "kernel binary image found at 0x%" PRIx64 " with arch '%s' %s",
461bc22c8d8SJason Molenda             addr, kernel_arch.GetTriple().str().c_str(), uuid_str.c_str());
462bc22c8d8SJason Molenda       }
4636ba6d3d1SJason Molenda       return memory_module_sp->GetUUID();
4646ba6d3d1SJason Molenda     }
4656ba6d3d1SJason Molenda   }
4666ba6d3d1SJason Molenda 
4676ba6d3d1SJason Molenda   return UUID();
4686ba6d3d1SJason Molenda }
4696ba6d3d1SJason Molenda 
4706ba6d3d1SJason Molenda //----------------------------------------------------------------------
471d4bfbc9aSGreg Clayton // Constructor
472d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
473b9c1b51eSKate Stone DynamicLoaderDarwinKernel::DynamicLoaderDarwinKernel(Process *process,
474b9c1b51eSKate Stone                                                      lldb::addr_t kernel_addr)
475b9c1b51eSKate Stone     : DynamicLoader(process), m_kernel_load_address(kernel_addr), m_kernel(),
476b9c1b51eSKate Stone       m_kext_summary_header_ptr_addr(), m_kext_summary_header_addr(),
477b9c1b51eSKate Stone       m_kext_summary_header(), m_known_kexts(), m_mutex(),
478b9c1b51eSKate Stone       m_break_id(LLDB_INVALID_BREAK_ID) {
47997206d57SZachary Turner   Status error;
480b9c1b51eSKate Stone   PlatformSP platform_sp(
481b9c1b51eSKate Stone       Platform::Create(PlatformDarwinKernel::GetPluginNameStatic(), error));
4821c627543SJason Molenda   // Only select the darwin-kernel Platform if we've been asked to load kexts.
4831c627543SJason Molenda   // It can take some time to scan over all of the kext info.plists and that
4841c627543SJason Molenda   // shouldn't be done if kext loading is explicitly disabled.
485b9c1b51eSKate Stone   if (platform_sp.get() && GetGlobalProperties()->GetLoadKexts()) {
4861c627543SJason Molenda     process->GetTarget().SetPlatform(platform_sp);
4871c627543SJason Molenda   }
488d4bfbc9aSGreg Clayton }
489d4bfbc9aSGreg Clayton 
490d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
491d4bfbc9aSGreg Clayton // Destructor
492d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
493b9c1b51eSKate Stone DynamicLoaderDarwinKernel::~DynamicLoaderDarwinKernel() { Clear(true); }
494d4bfbc9aSGreg Clayton 
495b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::UpdateIfNeeded() {
496d4bfbc9aSGreg Clayton   LoadKernelModuleIfNeeded();
497d4bfbc9aSGreg Clayton   SetNotificationBreakpointIfNeeded();
498d4bfbc9aSGreg Clayton }
499d4bfbc9aSGreg Clayton //------------------------------------------------------------------
500d4bfbc9aSGreg Clayton /// Called after attaching a process.
501d4bfbc9aSGreg Clayton ///
502d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
503d4bfbc9aSGreg Clayton /// attaching to a process.
504d4bfbc9aSGreg Clayton //------------------------------------------------------------------
505b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DidAttach() {
506d4bfbc9aSGreg Clayton   PrivateInitialize(m_process);
507d4bfbc9aSGreg Clayton   UpdateIfNeeded();
508d4bfbc9aSGreg Clayton }
509d4bfbc9aSGreg Clayton 
510d4bfbc9aSGreg Clayton //------------------------------------------------------------------
511d4bfbc9aSGreg Clayton /// Called after attaching a process.
512d4bfbc9aSGreg Clayton ///
513d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
514d4bfbc9aSGreg Clayton /// attaching to a process.
515d4bfbc9aSGreg Clayton //------------------------------------------------------------------
516b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DidLaunch() {
517d4bfbc9aSGreg Clayton   PrivateInitialize(m_process);
518d4bfbc9aSGreg Clayton   UpdateIfNeeded();
519d4bfbc9aSGreg Clayton }
520d4bfbc9aSGreg Clayton 
521d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
522d4bfbc9aSGreg Clayton // Clear out the state of this class.
523d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
524b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Clear(bool clear_process) {
52516ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
526d4bfbc9aSGreg Clayton 
527d4bfbc9aSGreg Clayton   if (m_process->IsAlive() && LLDB_BREAK_ID_IS_VALID(m_break_id))
528d4bfbc9aSGreg Clayton     m_process->ClearBreakpointSiteByID(m_break_id);
529d4bfbc9aSGreg Clayton 
530d4bfbc9aSGreg Clayton   if (clear_process)
531d4bfbc9aSGreg Clayton     m_process = NULL;
532306bd0aaSJason Molenda   m_kernel.Clear();
533306bd0aaSJason Molenda   m_known_kexts.clear();
534d4bfbc9aSGreg Clayton   m_kext_summary_header_ptr_addr.Clear();
535d4bfbc9aSGreg Clayton   m_kext_summary_header_addr.Clear();
536d4bfbc9aSGreg Clayton   m_break_id = LLDB_INVALID_BREAK_ID;
537d4bfbc9aSGreg Clayton }
538d4bfbc9aSGreg Clayton 
539b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::LoadImageAtFileAddress(
540b9c1b51eSKate Stone     Process *process) {
5412af282a1SGreg Clayton   if (IsLoaded())
5422af282a1SGreg Clayton     return true;
5432af282a1SGreg Clayton 
544b9c1b51eSKate Stone   if (m_module_sp) {
5452af282a1SGreg Clayton     bool changed = false;
546751caf65SGreg Clayton     if (m_module_sp->SetLoadAddress(process->GetTarget(), 0, true, changed))
547306bd0aaSJason Molenda       m_load_process_stop_id = process->GetStopID();
5482af282a1SGreg Clayton   }
5492af282a1SGreg Clayton   return false;
5502af282a1SGreg Clayton }
5512af282a1SGreg Clayton 
552b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetModule(ModuleSP module_sp) {
553306bd0aaSJason Molenda   m_module_sp = module_sp;
554b9c1b51eSKate Stone   if (module_sp.get() && module_sp->GetObjectFile()) {
555b9c1b51eSKate Stone     if (module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeExecutable &&
556b9c1b51eSKate Stone         module_sp->GetObjectFile()->GetStrata() == ObjectFile::eStrataKernel) {
557306bd0aaSJason Molenda       m_kernel_image = true;
558b9c1b51eSKate Stone     } else {
559306bd0aaSJason Molenda       m_kernel_image = false;
560306bd0aaSJason Molenda     }
561306bd0aaSJason Molenda   }
562306bd0aaSJason Molenda }
563306bd0aaSJason Molenda 
564b9c1b51eSKate Stone ModuleSP DynamicLoaderDarwinKernel::KextImageInfo::GetModule() {
565306bd0aaSJason Molenda   return m_module_sp;
566306bd0aaSJason Molenda }
567306bd0aaSJason Molenda 
568b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetLoadAddress(
569b9c1b51eSKate Stone     addr_t load_addr) {
570306bd0aaSJason Molenda   m_load_address = load_addr;
571306bd0aaSJason Molenda }
572306bd0aaSJason Molenda 
573b9c1b51eSKate Stone addr_t DynamicLoaderDarwinKernel::KextImageInfo::GetLoadAddress() const {
574306bd0aaSJason Molenda   return m_load_address;
575306bd0aaSJason Molenda }
576306bd0aaSJason Molenda 
577b9c1b51eSKate Stone uint64_t DynamicLoaderDarwinKernel::KextImageInfo::GetSize() const {
578306bd0aaSJason Molenda   return m_size;
579306bd0aaSJason Molenda }
580306bd0aaSJason Molenda 
581b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetSize(uint64_t size) {
582306bd0aaSJason Molenda   m_size = size;
583306bd0aaSJason Molenda }
584306bd0aaSJason Molenda 
585b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::KextImageInfo::GetProcessStopId() const {
586306bd0aaSJason Molenda   return m_load_process_stop_id;
587306bd0aaSJason Molenda }
588306bd0aaSJason Molenda 
589b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetProcessStopId(
590b9c1b51eSKate Stone     uint32_t stop_id) {
591306bd0aaSJason Molenda   m_load_process_stop_id = stop_id;
592306bd0aaSJason Molenda }
593306bd0aaSJason Molenda 
594b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::
595b9c1b51eSKate Stone operator==(const KextImageInfo &rhs) {
596b9c1b51eSKate Stone   if (m_uuid.IsValid() || rhs.GetUUID().IsValid()) {
597b9c1b51eSKate Stone     if (m_uuid == rhs.GetUUID()) {
598306bd0aaSJason Molenda       return true;
599306bd0aaSJason Molenda     }
600306bd0aaSJason Molenda     return false;
601306bd0aaSJason Molenda   }
602306bd0aaSJason Molenda 
603306bd0aaSJason Molenda   if (m_name == rhs.GetName() && m_load_address == rhs.GetLoadAddress())
604306bd0aaSJason Molenda     return true;
605306bd0aaSJason Molenda 
606306bd0aaSJason Molenda   return false;
607306bd0aaSJason Molenda }
608306bd0aaSJason Molenda 
609b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetName(const char *name) {
610306bd0aaSJason Molenda   m_name = name;
611306bd0aaSJason Molenda }
612306bd0aaSJason Molenda 
613b9c1b51eSKate Stone std::string DynamicLoaderDarwinKernel::KextImageInfo::GetName() const {
614306bd0aaSJason Molenda   return m_name;
615306bd0aaSJason Molenda }
616306bd0aaSJason Molenda 
617b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetUUID(const UUID &uuid) {
618306bd0aaSJason Molenda   m_uuid = uuid;
619306bd0aaSJason Molenda }
620306bd0aaSJason Molenda 
621b9c1b51eSKate Stone UUID DynamicLoaderDarwinKernel::KextImageInfo::GetUUID() const {
622306bd0aaSJason Molenda   return m_uuid;
623306bd0aaSJason Molenda }
624306bd0aaSJason Molenda 
625b9c1b51eSKate Stone // Given the m_load_address from the kext summaries, and a UUID, try to create
62605097246SAdrian Prantl // an in-memory Module at that address.  Require that the MemoryModule have a
62705097246SAdrian Prantl // matching UUID and detect if this MemoryModule is a kernel or a kext.
628306bd0aaSJason Molenda //
629306bd0aaSJason Molenda // Returns true if m_memory_module_sp is now set to a valid Module.
630306bd0aaSJason Molenda 
631b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::ReadMemoryModule(
632b9c1b51eSKate Stone     Process *process) {
63308a3258bSJason Molenda   Log *log = lldb_private::GetLogIfAllCategoriesSet(LIBLLDB_LOG_HOST);
634306bd0aaSJason Molenda   if (m_memory_module_sp.get() != NULL)
635306bd0aaSJason Molenda     return true;
636306bd0aaSJason Molenda   if (m_load_address == LLDB_INVALID_ADDRESS)
637306bd0aaSJason Molenda     return false;
638306bd0aaSJason Molenda 
639306bd0aaSJason Molenda   FileSpec file_spec;
640*937348cdSJonas Devlieghere   file_spec.SetFile(m_name.c_str(), false, FileSpec::Style::native);
641306bd0aaSJason Molenda 
642b9c1b51eSKate Stone   ModuleSP memory_module_sp =
643b9c1b51eSKate Stone       process->ReadModuleFromMemory(file_spec, m_load_address);
644306bd0aaSJason Molenda 
645306bd0aaSJason Molenda   if (memory_module_sp.get() == NULL)
646306bd0aaSJason Molenda     return false;
647306bd0aaSJason Molenda 
648306bd0aaSJason Molenda   bool is_kernel = false;
649b9c1b51eSKate Stone   if (memory_module_sp->GetObjectFile()) {
650b9c1b51eSKate Stone     if (memory_module_sp->GetObjectFile()->GetType() ==
651b9c1b51eSKate Stone             ObjectFile::eTypeExecutable &&
652b9c1b51eSKate Stone         memory_module_sp->GetObjectFile()->GetStrata() ==
653b9c1b51eSKate Stone             ObjectFile::eStrataKernel) {
654306bd0aaSJason Molenda       is_kernel = true;
655b9c1b51eSKate Stone     } else if (memory_module_sp->GetObjectFile()->GetType() ==
656b9c1b51eSKate Stone                ObjectFile::eTypeSharedLibrary) {
657306bd0aaSJason Molenda       is_kernel = false;
658306bd0aaSJason Molenda     }
659306bd0aaSJason Molenda   }
660306bd0aaSJason Molenda 
661b9c1b51eSKate Stone   // If this is a kext, and the kernel specified what UUID we should find at
66205097246SAdrian Prantl   // this load address, require that the memory module have a matching UUID or
66305097246SAdrian Prantl   // something has gone wrong and we should discard it.
664b9c1b51eSKate Stone   if (m_uuid.IsValid()) {
665b9c1b51eSKate Stone     if (m_uuid != memory_module_sp->GetUUID()) {
666b9c1b51eSKate Stone       if (log) {
667b9c1b51eSKate Stone         log->Printf("KextImageInfo::ReadMemoryModule the kernel said to find "
668b9c1b51eSKate Stone                     "uuid %s at 0x%" PRIx64
669b9c1b51eSKate Stone                     " but instead we found uuid %s, throwing it away",
670b9c1b51eSKate Stone                     m_uuid.GetAsString().c_str(), m_load_address,
671b9c1b51eSKate Stone                     memory_module_sp->GetUUID().GetAsString().c_str());
67208a3258bSJason Molenda       }
673306bd0aaSJason Molenda       return false;
674306bd0aaSJason Molenda     }
675306bd0aaSJason Molenda   }
676306bd0aaSJason Molenda 
677306bd0aaSJason Molenda   // If the in-memory Module has a UUID, let's use that.
678b9c1b51eSKate Stone   if (!m_uuid.IsValid() && memory_module_sp->GetUUID().IsValid()) {
679306bd0aaSJason Molenda     m_uuid = memory_module_sp->GetUUID();
680306bd0aaSJason Molenda   }
681306bd0aaSJason Molenda 
682306bd0aaSJason Molenda   m_memory_module_sp = memory_module_sp;
683306bd0aaSJason Molenda   m_kernel_image = is_kernel;
684b9c1b51eSKate Stone   if (is_kernel) {
685b9c1b51eSKate Stone     if (log) {
68608a3258bSJason Molenda       // This is unusual and probably not intended
687b9c1b51eSKate Stone       log->Printf("KextImageInfo::ReadMemoryModule read the kernel binary out "
688b9c1b51eSKate Stone                   "of memory");
68908a3258bSJason Molenda     }
690b9c1b51eSKate Stone     if (memory_module_sp->GetArchitecture().IsValid()) {
691306bd0aaSJason Molenda       process->GetTarget().SetArchitecture(memory_module_sp->GetArchitecture());
692306bd0aaSJason Molenda     }
693b9c1b51eSKate Stone     if (m_uuid.IsValid()) {
69448975833SJason Molenda       ModuleSP exe_module_sp = process->GetTarget().GetExecutableModule();
695b9c1b51eSKate Stone       if (exe_module_sp.get() && exe_module_sp->GetUUID().IsValid()) {
696b9c1b51eSKate Stone         if (m_uuid != exe_module_sp->GetUUID()) {
69748975833SJason Molenda           // The user specified a kernel binary that has a different UUID than
69848975833SJason Molenda           // the kernel actually running in memory.  This never ends well;
69948975833SJason Molenda           // clear the user specified kernel binary from the Target.
70048975833SJason Molenda 
70148975833SJason Molenda           m_module_sp.reset();
70248975833SJason Molenda 
70348975833SJason Molenda           ModuleList user_specified_kernel_list;
70448975833SJason Molenda           user_specified_kernel_list.Append(exe_module_sp);
70548975833SJason Molenda           process->GetTarget().GetImages().Remove(user_specified_kernel_list);
70638e70d11SJason Molenda         }
70738e70d11SJason Molenda       }
70838e70d11SJason Molenda     }
709306bd0aaSJason Molenda   }
710306bd0aaSJason Molenda 
711306bd0aaSJason Molenda   return true;
712306bd0aaSJason Molenda }
713306bd0aaSJason Molenda 
714b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::IsKernel() const {
715306bd0aaSJason Molenda   return m_kernel_image == true;
716306bd0aaSJason Molenda }
717306bd0aaSJason Molenda 
718b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::SetIsKernel(bool is_kernel) {
719306bd0aaSJason Molenda   m_kernel_image = is_kernel;
720306bd0aaSJason Molenda }
721306bd0aaSJason Molenda 
722b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::KextImageInfo::LoadImageUsingMemoryModule(
723b9c1b51eSKate Stone     Process *process) {
724c859e2d5SGreg Clayton   if (IsLoaded())
725c859e2d5SGreg Clayton     return true;
726c859e2d5SGreg Clayton 
727c859e2d5SGreg Clayton   Target &target = process->GetTarget();
72887a04b24SJason Molenda 
72905097246SAdrian Prantl   // If we don't have / can't create a memory module for this kext, don't try
73005097246SAdrian Prantl   // to load it - we won't have the correct segment load addresses.
731b9c1b51eSKate Stone   if (!ReadMemoryModule(process)) {
732a8cb765cSJason Molenda     Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
733a8cb765cSJason Molenda     if (log)
734b9c1b51eSKate Stone       log->Printf("Unable to read '%s' from memory at address 0x%" PRIx64
735b9c1b51eSKate Stone                   " to get the segment load addresses.",
736b9c1b51eSKate Stone                   m_name.c_str(), m_load_address);
73787a04b24SJason Molenda     return false;
73887a04b24SJason Molenda   }
73987a04b24SJason Molenda 
740306bd0aaSJason Molenda   bool uuid_is_valid = m_uuid.IsValid();
741306bd0aaSJason Molenda 
742b9c1b51eSKate Stone   if (IsKernel() && uuid_is_valid && m_memory_module_sp.get()) {
74344d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
744b9c1b51eSKate Stone     if (s) {
745b9c1b51eSKate Stone       s->Printf("Kernel UUID: %s\n",
746b9c1b51eSKate Stone                 m_memory_module_sp->GetUUID().GetAsString().c_str());
747e575e7bcSJason Molenda       s->Printf("Load Address: 0x%" PRIx64 "\n", m_load_address);
748e575e7bcSJason Molenda     }
749e575e7bcSJason Molenda   }
750e575e7bcSJason Molenda 
751b9c1b51eSKate Stone   if (!m_module_sp) {
752b9c1b51eSKate Stone     // See if the kext has already been loaded into the target, probably by the
753b9c1b51eSKate Stone     // user doing target modules add.
7541759848bSEnrico Granata     const ModuleList &target_images = target.GetImages();
755306bd0aaSJason Molenda     m_module_sp = target_images.FindModule(m_uuid);
756c859e2d5SGreg Clayton 
757306bd0aaSJason Molenda     // Search for the kext on the local filesystem via the UUID
758b9c1b51eSKate Stone     if (!m_module_sp && uuid_is_valid) {
7592af282a1SGreg Clayton       ModuleSpec module_spec;
760306bd0aaSJason Molenda       module_spec.GetUUID() = m_uuid;
76153667f5dSJason Molenda       module_spec.GetArchitecture() = target.GetArchitecture();
762bb860bd2SJason Molenda 
76305097246SAdrian Prantl       // For the kernel, we really do need an on-disk file copy of the binary
76405097246SAdrian Prantl       // to do anything useful. This will force a clal to
765b9c1b51eSKate Stone       if (IsKernel()) {
766b9c1b51eSKate Stone         if (Symbols::DownloadObjectAndSymbolFile(module_spec, true)) {
767b9c1b51eSKate Stone           if (module_spec.GetFileSpec().Exists()) {
768b9c1b51eSKate Stone             m_module_sp.reset(new Module(module_spec.GetFileSpec(),
769b9c1b51eSKate Stone                                          target.GetArchitecture()));
770b9c1b51eSKate Stone             if (m_module_sp.get() &&
771b9c1b51eSKate Stone                 m_module_sp->MatchesModuleSpec(module_spec)) {
772bb860bd2SJason Molenda               ModuleList loaded_module_list;
773306bd0aaSJason Molenda               loaded_module_list.Append(m_module_sp);
774bb860bd2SJason Molenda               target.ModulesDidLoad(loaded_module_list);
775bb860bd2SJason Molenda             }
776bb860bd2SJason Molenda           }
777bb860bd2SJason Molenda         }
778d76fb6eaSJason Molenda       }
779bb860bd2SJason Molenda 
780b9c1b51eSKate Stone       // If the current platform is PlatformDarwinKernel, create a ModuleSpec
78105097246SAdrian Prantl       // with the filename set to be the bundle ID for this kext, e.g.
78205097246SAdrian Prantl       // "com.apple.filesystems.msdosfs", and ask the platform to find it.
7831c627543SJason Molenda       PlatformSP platform_sp(target.GetPlatform());
784b9c1b51eSKate Stone       if (!m_module_sp && platform_sp) {
78557abc5d6SGreg Clayton         ConstString platform_name(platform_sp->GetPluginName());
786b9c1b51eSKate Stone         static ConstString g_platform_name(
787b9c1b51eSKate Stone             PlatformDarwinKernel::GetPluginNameStatic());
788b9c1b51eSKate Stone         if (platform_name == g_platform_name) {
7891c627543SJason Molenda           ModuleSpec kext_bundle_module_spec(module_spec);
7901c627543SJason Molenda           FileSpec kext_filespec(m_name.c_str(), false);
7911c627543SJason Molenda           kext_bundle_module_spec.GetFileSpec() = kext_filespec;
792b9c1b51eSKate Stone           platform_sp->GetSharedModule(
793b9c1b51eSKate Stone               kext_bundle_module_spec, process, m_module_sp,
794b9c1b51eSKate Stone               &target.GetExecutableSearchPaths(), NULL, NULL);
7951c627543SJason Molenda         }
7961c627543SJason Molenda       }
7971c627543SJason Molenda 
798d76fb6eaSJason Molenda       // Ask the Target to find this file on the local system, if possible.
799bb860bd2SJason Molenda       // This will search in the list of currently-loaded files, look in the
800bb860bd2SJason Molenda       // standard search paths on the system, and on a Mac it will try calling
801bb860bd2SJason Molenda       // the DebugSymbols framework with the UUID to find the binary via its
802bb860bd2SJason Molenda       // search methods.
803b9c1b51eSKate Stone       if (!m_module_sp) {
804306bd0aaSJason Molenda         m_module_sp = target.GetSharedModule(module_spec);
805306bd0aaSJason Molenda       }
806e575e7bcSJason Molenda 
807b9c1b51eSKate Stone       if (IsKernel() && !m_module_sp) {
80844d93782SGreg Clayton         Stream *s = target.GetDebugger().GetOutputFile().get();
809b9c1b51eSKate Stone         if (s) {
810b9c1b51eSKate Stone           s->Printf("WARNING: Unable to locate kernel binary on the debugger "
811b9c1b51eSKate Stone                     "system.\n");
812e575e7bcSJason Molenda         }
813e575e7bcSJason Molenda       }
814b9a01b39SGreg Clayton     }
81565d57a3dSJason Molenda 
81605097246SAdrian Prantl     // If we managed to find a module, append it to the target's list of
81705097246SAdrian Prantl     // images. If we also have a memory module, require that they have matching
81805097246SAdrian Prantl     // UUIDs
819b9c1b51eSKate Stone     if (m_module_sp) {
820306bd0aaSJason Molenda       bool uuid_match_ok = true;
821b9c1b51eSKate Stone       if (m_memory_module_sp) {
822b9c1b51eSKate Stone         if (m_module_sp->GetUUID() != m_memory_module_sp->GetUUID()) {
823306bd0aaSJason Molenda           uuid_match_ok = false;
82465d57a3dSJason Molenda         }
82565d57a3dSJason Molenda       }
826b9c1b51eSKate Stone       if (uuid_match_ok) {
827a4d3e1d2SJason Molenda         target.GetImages().AppendIfNeeded(m_module_sp);
828b9c1b51eSKate Stone         if (IsKernel() &&
829b9c1b51eSKate Stone             target.GetExecutableModulePointer() != m_module_sp.get()) {
830306bd0aaSJason Molenda           target.SetExecutableModule(m_module_sp, false);
831306bd0aaSJason Molenda         }
832c859e2d5SGreg Clayton       }
833c859e2d5SGreg Clayton     }
834bb860bd2SJason Molenda   }
835c859e2d5SGreg Clayton 
836b9c1b51eSKate Stone   if (!m_module_sp && !IsKernel() && m_uuid.IsValid() && !m_name.empty()) {
83744d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
838b9c1b51eSKate Stone     if (s) {
839b9c1b51eSKate Stone       s->Printf("warning: Can't find binary/dSYM for %s (%s)\n", m_name.c_str(),
840b9c1b51eSKate Stone                 m_uuid.GetAsString().c_str());
84156c23285SJason Molenda     }
84256c23285SJason Molenda   }
843c859e2d5SGreg Clayton 
84467408537SGreg Clayton   static ConstString g_section_name_LINKEDIT("__LINKEDIT");
84567408537SGreg Clayton 
846b9c1b51eSKate Stone   if (m_memory_module_sp && m_module_sp) {
847b9c1b51eSKate Stone     if (m_module_sp->GetUUID() == m_memory_module_sp->GetUUID()) {
848306bd0aaSJason Molenda       ObjectFile *ondisk_object_file = m_module_sp->GetObjectFile();
849306bd0aaSJason Molenda       ObjectFile *memory_object_file = m_memory_module_sp->GetObjectFile();
85067408537SGreg Clayton 
851b9c1b51eSKate Stone       if (memory_object_file && ondisk_object_file) {
852b9c1b51eSKate Stone         // The memory_module for kexts may have an invalid __LINKEDIT seg; skip
853b9c1b51eSKate Stone         // it.
854306bd0aaSJason Molenda         const bool ignore_linkedit = !IsKernel();
85567408537SGreg Clayton 
856c859e2d5SGreg Clayton         SectionList *ondisk_section_list = ondisk_object_file->GetSectionList();
857c859e2d5SGreg Clayton         SectionList *memory_section_list = memory_object_file->GetSectionList();
858b9c1b51eSKate Stone         if (memory_section_list && ondisk_section_list) {
859bae2f2f8SGreg Clayton           const uint32_t num_ondisk_sections = ondisk_section_list->GetSize();
86005097246SAdrian Prantl           // There may be CTF sections in the memory image so we can't always
86105097246SAdrian Prantl           // just compare the number of sections (which are actually segments
86205097246SAdrian Prantl           // in mach-o parlance)
863c859e2d5SGreg Clayton           uint32_t sect_idx = 0;
864bae2f2f8SGreg Clayton 
86505097246SAdrian Prantl           // Use the memory_module's addresses for each section to set the file
86605097246SAdrian Prantl           // module's load address as appropriate.  We don't want to use a
86705097246SAdrian Prantl           // single slide value for the entire kext - different segments may be
86805097246SAdrian Prantl           // slid different amounts by the kext loader.
869bae2f2f8SGreg Clayton 
870bae2f2f8SGreg Clayton           uint32_t num_sections_loaded = 0;
871b9c1b51eSKate Stone           for (sect_idx = 0; sect_idx < num_ondisk_sections; ++sect_idx) {
872b9c1b51eSKate Stone             SectionSP ondisk_section_sp(
873b9c1b51eSKate Stone                 ondisk_section_list->GetSectionAtIndex(sect_idx));
874b9c1b51eSKate Stone             if (ondisk_section_sp) {
87567408537SGreg Clayton               // Don't ever load __LINKEDIT as it may or may not be actually
87667408537SGreg Clayton               // mapped into memory and there is no current way to tell.
87767408537SGreg Clayton               // I filed rdar://problem/12851706 to track being able to tell
87867408537SGreg Clayton               // if the __LINKEDIT is actually mapped, but until then, we need
87967408537SGreg Clayton               // to not load the __LINKEDIT
880b9c1b51eSKate Stone               if (ignore_linkedit &&
881b9c1b51eSKate Stone                   ondisk_section_sp->GetName() == g_section_name_LINKEDIT)
88267408537SGreg Clayton                 continue;
88367408537SGreg Clayton 
884b9c1b51eSKate Stone               const Section *memory_section =
885b9c1b51eSKate Stone                   memory_section_list
886b9c1b51eSKate Stone                       ->FindSectionByName(ondisk_section_sp->GetName())
887b9c1b51eSKate Stone                       .get();
888b9c1b51eSKate Stone               if (memory_section) {
889b9c1b51eSKate Stone                 target.SetSectionLoadAddress(ondisk_section_sp,
890b9c1b51eSKate Stone                                              memory_section->GetFileAddress());
891bae2f2f8SGreg Clayton                 ++num_sections_loaded;
892c859e2d5SGreg Clayton               }
893bae2f2f8SGreg Clayton             }
894bae2f2f8SGreg Clayton           }
895bae2f2f8SGreg Clayton           if (num_sections_loaded > 0)
896306bd0aaSJason Molenda             m_load_process_stop_id = process->GetStopID();
897bae2f2f8SGreg Clayton           else
898306bd0aaSJason Molenda             m_module_sp.reset(); // No sections were loaded
899b9c1b51eSKate Stone         } else
900306bd0aaSJason Molenda           m_module_sp.reset(); // One or both section lists
901b9c1b51eSKate Stone       } else
902306bd0aaSJason Molenda         m_module_sp.reset(); // One or both object files missing
903b9c1b51eSKate Stone     } else
904306bd0aaSJason Molenda       m_module_sp.reset(); // UUID mismatch
905c859e2d5SGreg Clayton   }
906c859e2d5SGreg Clayton 
907c859e2d5SGreg Clayton   bool is_loaded = IsLoaded();
908c859e2d5SGreg Clayton 
909b9c1b51eSKate Stone   if (is_loaded && m_module_sp && IsKernel()) {
91044d93782SGreg Clayton     Stream *s = target.GetDebugger().GetOutputFile().get();
911b9c1b51eSKate Stone     if (s) {
912732238cdSJason Molenda       ObjectFile *kernel_object_file = m_module_sp->GetObjectFile();
913b9c1b51eSKate Stone       if (kernel_object_file) {
914b9c1b51eSKate Stone         addr_t file_address =
915b9c1b51eSKate Stone             kernel_object_file->GetHeaderAddress().GetFileAddress();
916b9c1b51eSKate Stone         if (m_load_address != LLDB_INVALID_ADDRESS &&
917b9c1b51eSKate Stone             file_address != LLDB_INVALID_ADDRESS) {
918b9c1b51eSKate Stone           s->Printf("Kernel slid 0x%" PRIx64 " in memory.\n",
919b9c1b51eSKate Stone                     m_load_address - file_address);
920732238cdSJason Molenda         }
921732238cdSJason Molenda       }
922743e4396SJason Molenda       {
923b5ad4ec7SGreg Clayton         s->Printf("Loaded kernel file %s\n",
924b5ad4ec7SGreg Clayton                   m_module_sp->GetFileSpec().GetPath().c_str());
925743e4396SJason Molenda       }
92668b3607fSJason Molenda       s->Flush();
92768b3607fSJason Molenda     }
92868b3607fSJason Molenda   }
929c859e2d5SGreg Clayton   return is_loaded;
930c859e2d5SGreg Clayton }
931c859e2d5SGreg Clayton 
932b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::KextImageInfo::GetAddressByteSize() {
933306bd0aaSJason Molenda   if (m_memory_module_sp)
934306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture().GetAddressByteSize();
935306bd0aaSJason Molenda   if (m_module_sp)
936306bd0aaSJason Molenda     return m_module_sp->GetArchitecture().GetAddressByteSize();
9371f746071SGreg Clayton   return 0;
9381f746071SGreg Clayton }
9391f746071SGreg Clayton 
940b9c1b51eSKate Stone lldb::ByteOrder DynamicLoaderDarwinKernel::KextImageInfo::GetByteOrder() {
941306bd0aaSJason Molenda   if (m_memory_module_sp)
942306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture().GetByteOrder();
943306bd0aaSJason Molenda   if (m_module_sp)
944306bd0aaSJason Molenda     return m_module_sp->GetArchitecture().GetByteOrder();
9459ccb970fSBruce Mitchener   return endian::InlHostByteOrder();
9461f746071SGreg Clayton }
9471f746071SGreg Clayton 
9481f746071SGreg Clayton lldb_private::ArchSpec
949b9c1b51eSKate Stone DynamicLoaderDarwinKernel::KextImageInfo::GetArchitecture() const {
950306bd0aaSJason Molenda   if (m_memory_module_sp)
951306bd0aaSJason Molenda     return m_memory_module_sp->GetArchitecture();
952306bd0aaSJason Molenda   if (m_module_sp)
953306bd0aaSJason Molenda     return m_module_sp->GetArchitecture();
9541f746071SGreg Clayton   return lldb_private::ArchSpec();
9551f746071SGreg Clayton }
9561f746071SGreg Clayton 
957d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
95805097246SAdrian Prantl // Load the kernel module and initialize the "m_kernel" member. Return true
95905097246SAdrian Prantl // _only_ if the kernel is loaded the first time through (subsequent calls to
96005097246SAdrian Prantl // this function should return false after the kernel has been already loaded).
961d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
962b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::LoadKernelModuleIfNeeded() {
963b9c1b51eSKate Stone   if (!m_kext_summary_header_ptr_addr.IsValid()) {
964306bd0aaSJason Molenda     m_kernel.Clear();
965306bd0aaSJason Molenda     m_kernel.SetModule(m_process->GetTarget().GetExecutableModule());
966306bd0aaSJason Molenda     m_kernel.SetIsKernel(true);
9674bd4e7e3SJason Molenda 
9684bd4e7e3SJason Molenda     ConstString kernel_name("mach_kernel");
969b9c1b51eSKate Stone     if (m_kernel.GetModule().get() && m_kernel.GetModule()->GetObjectFile() &&
970b9c1b51eSKate Stone         !m_kernel.GetModule()
971b9c1b51eSKate Stone              ->GetObjectFile()
972b9c1b51eSKate Stone              ->GetFileSpec()
973b9c1b51eSKate Stone              .GetFilename()
974b9c1b51eSKate Stone              .IsEmpty()) {
975b9c1b51eSKate Stone       kernel_name =
976b9c1b51eSKate Stone           m_kernel.GetModule()->GetObjectFile()->GetFileSpec().GetFilename();
9774bd4e7e3SJason Molenda     }
978306bd0aaSJason Molenda     m_kernel.SetName(kernel_name.AsCString());
9794bd4e7e3SJason Molenda 
980b9c1b51eSKate Stone     if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS) {
981306bd0aaSJason Molenda       m_kernel.SetLoadAddress(m_kernel_load_address);
982b9c1b51eSKate Stone       if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS &&
983b9c1b51eSKate Stone           m_kernel.GetModule()) {
98405097246SAdrian Prantl         // We didn't get a hint from the process, so we will try the kernel at
98505097246SAdrian Prantl         // the address that it exists at in the file if we have one
986306bd0aaSJason Molenda         ObjectFile *kernel_object_file = m_kernel.GetModule()->GetObjectFile();
987b9c1b51eSKate Stone         if (kernel_object_file) {
988b9c1b51eSKate Stone           addr_t load_address =
989b9c1b51eSKate Stone               kernel_object_file->GetHeaderAddress().GetLoadAddress(
990b9c1b51eSKate Stone                   &m_process->GetTarget());
991b9c1b51eSKate Stone           addr_t file_address =
992b9c1b51eSKate Stone               kernel_object_file->GetHeaderAddress().GetFileAddress();
993b9c1b51eSKate Stone           if (load_address != LLDB_INVALID_ADDRESS && load_address != 0) {
994306bd0aaSJason Molenda             m_kernel.SetLoadAddress(load_address);
995b9c1b51eSKate Stone             if (load_address != file_address) {
9964bd4e7e3SJason Molenda               // Don't accidentally relocate the kernel to the File address --
997b9c1b51eSKate Stone               // the Load address has already been set to its actual in-memory
99805097246SAdrian Prantl               // address. Mark it as IsLoaded.
999306bd0aaSJason Molenda               m_kernel.SetProcessStopId(m_process->GetStopID());
10004bd4e7e3SJason Molenda             }
1001b9c1b51eSKate Stone           } else {
1002306bd0aaSJason Molenda             m_kernel.SetLoadAddress(file_address);
10034bd4e7e3SJason Molenda           }
10044bd4e7e3SJason Molenda         }
1005c859e2d5SGreg Clayton       }
1006c859e2d5SGreg Clayton     }
1007c859e2d5SGreg Clayton 
1008b9c1b51eSKate Stone     if (m_kernel.GetLoadAddress() != LLDB_INVALID_ADDRESS) {
1009b9c1b51eSKate Stone       if (!m_kernel.LoadImageUsingMemoryModule(m_process)) {
10102af282a1SGreg Clayton         m_kernel.LoadImageAtFileAddress(m_process);
10112af282a1SGreg Clayton       }
10122af282a1SGreg Clayton     }
1013c859e2d5SGreg Clayton 
1014519b0816SJim Ingham     // The operating system plugin gets loaded and initialized in
101505097246SAdrian Prantl     // LoadImageUsingMemoryModule when we discover the kernel dSYM.  For a core
101605097246SAdrian Prantl     // file in particular, that's the wrong place to do this, since  we haven't
101705097246SAdrian Prantl     // fixed up the section addresses yet.  So let's redo it here.
1018519b0816SJim Ingham     LoadOperatingSystemPlugin(false);
1019519b0816SJim Ingham 
1020b9c1b51eSKate Stone     if (m_kernel.IsLoaded() && m_kernel.GetModule()) {
1021d4bfbc9aSGreg Clayton       static ConstString kext_summary_symbol("gLoadedKextSummaries");
1022b9c1b51eSKate Stone       const Symbol *symbol =
1023b9c1b51eSKate Stone           m_kernel.GetModule()->FindFirstSymbolWithNameAndType(
1024b9c1b51eSKate Stone               kext_summary_symbol, eSymbolTypeData);
1025b9c1b51eSKate Stone       if (symbol) {
1026e7612134SGreg Clayton         m_kext_summary_header_ptr_addr = symbol->GetAddress();
1027d4bfbc9aSGreg Clayton         // Update all image infos
1028d4bfbc9aSGreg Clayton         ReadAllKextSummaries();
1029d4bfbc9aSGreg Clayton       }
1030b9c1b51eSKate Stone     } else {
1031306bd0aaSJason Molenda       m_kernel.Clear();
1032d4bfbc9aSGreg Clayton     }
1033d4bfbc9aSGreg Clayton   }
1034d4bfbc9aSGreg Clayton }
1035d4bfbc9aSGreg Clayton 
1036d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1037d4bfbc9aSGreg Clayton // Static callback function that gets called when our DYLD notification
103805097246SAdrian Prantl // breakpoint gets hit. We update all of our image infos and then let our super
103905097246SAdrian Prantl // class DynamicLoader class decide if we should stop or not (based on global
104005097246SAdrian Prantl // preference).
1041d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1042b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::BreakpointHitCallback(
1043b9c1b51eSKate Stone     void *baton, StoppointCallbackContext *context, user_id_t break_id,
1044b9c1b51eSKate Stone     user_id_t break_loc_id) {
1045b9c1b51eSKate Stone   return static_cast<DynamicLoaderDarwinKernel *>(baton)->BreakpointHit(
1046b9c1b51eSKate Stone       context, break_id, break_loc_id);
1047d4bfbc9aSGreg Clayton }
1048d4bfbc9aSGreg Clayton 
1049b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::BreakpointHit(StoppointCallbackContext *context,
1050d4bfbc9aSGreg Clayton                                               user_id_t break_id,
1051b9c1b51eSKate Stone                                               user_id_t break_loc_id) {
10525160ce5cSGreg Clayton   Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
1053d4bfbc9aSGreg Clayton   if (log)
1054944b828aSGreg Clayton     log->Printf("DynamicLoaderDarwinKernel::BreakpointHit (...)\n");
1055d4bfbc9aSGreg Clayton 
1056d4bfbc9aSGreg Clayton   ReadAllKextSummaries();
1057d4bfbc9aSGreg Clayton 
1058d4bfbc9aSGreg Clayton   if (log)
10595160ce5cSGreg Clayton     PutToLog(log);
1060d4bfbc9aSGreg Clayton 
1061d4bfbc9aSGreg Clayton   return GetStopWhenImagesChange();
1062d4bfbc9aSGreg Clayton }
1063d4bfbc9aSGreg Clayton 
1064b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ReadKextSummaryHeader() {
106516ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1066d4bfbc9aSGreg Clayton 
1067d4bfbc9aSGreg Clayton   // the all image infos is already valid for this process stop ID
1068d4bfbc9aSGreg Clayton 
1069b9c1b51eSKate Stone   if (m_kext_summary_header_ptr_addr.IsValid()) {
1070d4bfbc9aSGreg Clayton     const uint32_t addr_size = m_kernel.GetAddressByteSize();
1071d4bfbc9aSGreg Clayton     const ByteOrder byte_order = m_kernel.GetByteOrder();
107297206d57SZachary Turner     Status error;
107305097246SAdrian Prantl     // Read enough bytes for a "OSKextLoadedKextSummaryHeader" structure which
107405097246SAdrian Prantl     // is currently 4 uint32_t and a pointer.
1075d4bfbc9aSGreg Clayton     uint8_t buf[24];
1076d4bfbc9aSGreg Clayton     DataExtractor data(buf, sizeof(buf), byte_order, addr_size);
1077d4bfbc9aSGreg Clayton     const size_t count = 4 * sizeof(uint32_t) + addr_size;
1078d4bfbc9aSGreg Clayton     const bool prefer_file_cache = false;
1079b9c1b51eSKate Stone     if (m_process->GetTarget().ReadPointerFromMemory(
1080b9c1b51eSKate Stone             m_kext_summary_header_ptr_addr, prefer_file_cache, error,
1081b9c1b51eSKate Stone             m_kext_summary_header_addr)) {
1082b9c1b51eSKate Stone       // We got a valid address for our kext summary header and make sure it
1083b9c1b51eSKate Stone       // isn't NULL
1084d4bfbc9aSGreg Clayton       if (m_kext_summary_header_addr.IsValid() &&
1085b9c1b51eSKate Stone           m_kext_summary_header_addr.GetFileAddress() != 0) {
1086b9c1b51eSKate Stone         const size_t bytes_read = m_process->GetTarget().ReadMemory(
1087b9c1b51eSKate Stone             m_kext_summary_header_addr, prefer_file_cache, buf, count, error);
1088b9c1b51eSKate Stone         if (bytes_read == count) {
1089c7bece56SGreg Clayton           lldb::offset_t offset = 0;
1090d4bfbc9aSGreg Clayton           m_kext_summary_header.version = data.GetU32(&offset);
1091b9c1b51eSKate Stone           if (m_kext_summary_header.version > 128) {
1092b9c1b51eSKate Stone             Stream *s =
1093b9c1b51eSKate Stone                 m_process->GetTarget().GetDebugger().GetOutputFile().get();
1094b9c1b51eSKate Stone             s->Printf("WARNING: Unable to read kext summary header, got "
1095b9c1b51eSKate Stone                       "improbable version number %u\n",
1096b9c1b51eSKate Stone                       m_kext_summary_header.version);
1097b9c1b51eSKate Stone             // If we get an improbably large version number, we're probably
1098b9c1b51eSKate Stone             // getting bad memory.
1099c6fa5db7SJason Molenda             m_kext_summary_header_addr.Clear();
1100c6fa5db7SJason Molenda             return false;
1101c6fa5db7SJason Molenda           }
1102b9c1b51eSKate Stone           if (m_kext_summary_header.version >= 2) {
1103d4bfbc9aSGreg Clayton             m_kext_summary_header.entry_size = data.GetU32(&offset);
1104b9c1b51eSKate Stone             if (m_kext_summary_header.entry_size > 4096) {
1105b9c1b51eSKate Stone               // If we get an improbably large entry_size, we're probably
1106b9c1b51eSKate Stone               // getting bad memory.
1107b9c1b51eSKate Stone               Stream *s =
1108b9c1b51eSKate Stone                   m_process->GetTarget().GetDebugger().GetOutputFile().get();
1109b9c1b51eSKate Stone               s->Printf("WARNING: Unable to read kext summary header, got "
1110b9c1b51eSKate Stone                         "improbable entry_size %u\n",
1111b9c1b51eSKate Stone                         m_kext_summary_header.entry_size);
1112c6fa5db7SJason Molenda               m_kext_summary_header_addr.Clear();
1113c6fa5db7SJason Molenda               return false;
1114c6fa5db7SJason Molenda             }
1115b9c1b51eSKate Stone           } else {
111605097246SAdrian Prantl             // Versions less than 2 didn't have an entry size, it was hard
111705097246SAdrian Prantl             // coded
1118b9c1b51eSKate Stone             m_kext_summary_header.entry_size =
1119b9c1b51eSKate Stone                 KERNEL_MODULE_ENTRY_SIZE_VERSION_1;
1120d4bfbc9aSGreg Clayton           }
1121d4bfbc9aSGreg Clayton           m_kext_summary_header.entry_count = data.GetU32(&offset);
1122b9c1b51eSKate Stone           if (m_kext_summary_header.entry_count > 10000) {
1123b9c1b51eSKate Stone             // If we get an improbably large number of kexts, we're probably
1124b9c1b51eSKate Stone             // getting bad memory.
1125b9c1b51eSKate Stone             Stream *s =
1126b9c1b51eSKate Stone                 m_process->GetTarget().GetDebugger().GetOutputFile().get();
1127b9c1b51eSKate Stone             s->Printf("WARNING: Unable to read kext summary header, got "
1128b9c1b51eSKate Stone                       "improbable number of kexts %u\n",
1129b9c1b51eSKate Stone                       m_kext_summary_header.entry_count);
1130c6fa5db7SJason Molenda             m_kext_summary_header_addr.Clear();
1131c6fa5db7SJason Molenda             return false;
1132c6fa5db7SJason Molenda           }
1133d4bfbc9aSGreg Clayton           return true;
1134d4bfbc9aSGreg Clayton         }
1135d4bfbc9aSGreg Clayton       }
1136d4bfbc9aSGreg Clayton     }
1137d4bfbc9aSGreg Clayton   }
1138d4bfbc9aSGreg Clayton   m_kext_summary_header_addr.Clear();
1139d4bfbc9aSGreg Clayton   return false;
1140d4bfbc9aSGreg Clayton }
1141d4bfbc9aSGreg Clayton 
1142b9c1b51eSKate Stone // We've either (a) just attached to a new kernel, or (b) the kexts-changed
114305097246SAdrian Prantl // breakpoint was hit and we need to figure out what kexts have been added or
114405097246SAdrian Prantl // removed. Read the kext summaries from the inferior kernel memory, compare
114505097246SAdrian Prantl // them against the m_known_kexts vector and update the m_known_kexts vector as
114605097246SAdrian Prantl // needed to keep in sync with the inferior.
1147d4bfbc9aSGreg Clayton 
1148b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ParseKextSummaries(
1149b9c1b51eSKate Stone     const Address &kext_summary_addr, uint32_t count) {
1150306bd0aaSJason Molenda   KextImageInfo::collection kext_summaries;
11515160ce5cSGreg Clayton   Log *log(GetLogIfAnyCategoriesSet(LIBLLDB_LOG_DYNAMIC_LOADER));
1152d4bfbc9aSGreg Clayton   if (log)
1153b9c1b51eSKate Stone     log->Printf("Kexts-changed breakpoint hit, there are %d kexts currently.\n",
1154b9c1b51eSKate Stone                 count);
1155d4bfbc9aSGreg Clayton 
115616ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1157d4bfbc9aSGreg Clayton 
1158d4bfbc9aSGreg Clayton   if (!ReadKextSummaries(kext_summary_addr, count, kext_summaries))
1159d4bfbc9aSGreg Clayton     return false;
1160d4bfbc9aSGreg Clayton 
1161b9c1b51eSKate Stone   // read the plugin.dynamic-loader.darwin-kernel.load-kexts setting -- if the
116205097246SAdrian Prantl   // user requested no kext loading, don't print any messages about kexts &
116305097246SAdrian Prantl   // don't try to read them.
11644da2e32fSJason Molenda   const bool load_kexts = GetGlobalProperties()->GetLoadKexts();
11654da2e32fSJason Molenda 
1166b9c1b51eSKate Stone   // By default, all kexts we've loaded in the past are marked as "remove" and
116705097246SAdrian Prantl   // all of the kexts we just found out about from ReadKextSummaries are marked
116805097246SAdrian Prantl   // as "add".
1169306bd0aaSJason Molenda   std::vector<bool> to_be_removed(m_known_kexts.size(), true);
1170306bd0aaSJason Molenda   std::vector<bool> to_be_added(count, true);
1171306bd0aaSJason Molenda 
1172306bd0aaSJason Molenda   int number_of_new_kexts_being_added = 0;
1173306bd0aaSJason Molenda   int number_of_old_kexts_being_removed = m_known_kexts.size();
1174306bd0aaSJason Molenda 
1175306bd0aaSJason Molenda   const uint32_t new_kexts_size = kext_summaries.size();
1176306bd0aaSJason Molenda   const uint32_t old_kexts_size = m_known_kexts.size();
1177306bd0aaSJason Molenda 
117805097246SAdrian Prantl   // The m_known_kexts vector may have entries that have been Cleared, or are a
117905097246SAdrian Prantl   // kernel.
1180b9c1b51eSKate Stone   for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++) {
1181306bd0aaSJason Molenda     bool ignore = false;
1182306bd0aaSJason Molenda     KextImageInfo &image_info = m_known_kexts[old_kext];
1183b9c1b51eSKate Stone     if (image_info.IsKernel()) {
1184306bd0aaSJason Molenda       ignore = true;
1185b9c1b51eSKate Stone     } else if (image_info.GetLoadAddress() == LLDB_INVALID_ADDRESS &&
1186b9c1b51eSKate Stone                !image_info.GetModule()) {
1187306bd0aaSJason Molenda       ignore = true;
1188306bd0aaSJason Molenda     }
1189306bd0aaSJason Molenda 
1190b9c1b51eSKate Stone     if (ignore) {
1191306bd0aaSJason Molenda       number_of_old_kexts_being_removed--;
1192306bd0aaSJason Molenda       to_be_removed[old_kext] = false;
1193306bd0aaSJason Molenda     }
1194306bd0aaSJason Molenda   }
1195306bd0aaSJason Molenda 
1196306bd0aaSJason Molenda   // Scan over the list of kexts we just read from the kernel, note those that
1197306bd0aaSJason Molenda   // need to be added and those already loaded.
1198b9c1b51eSKate Stone   for (uint32_t new_kext = 0; new_kext < new_kexts_size; new_kext++) {
1199306bd0aaSJason Molenda     bool add_this_one = true;
1200b9c1b51eSKate Stone     for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++) {
1201b9c1b51eSKate Stone       if (m_known_kexts[old_kext] == kext_summaries[new_kext]) {
1202306bd0aaSJason Molenda         // We already have this kext, don't re-load it.
1203306bd0aaSJason Molenda         to_be_added[new_kext] = false;
1204306bd0aaSJason Molenda         // This kext is still present, do not remove it.
1205306bd0aaSJason Molenda         to_be_removed[old_kext] = false;
1206306bd0aaSJason Molenda 
1207306bd0aaSJason Molenda         number_of_old_kexts_being_removed--;
1208306bd0aaSJason Molenda         add_this_one = false;
1209306bd0aaSJason Molenda         break;
1210306bd0aaSJason Molenda       }
1211306bd0aaSJason Molenda     }
121205097246SAdrian Prantl     // If this "kext" entry is actually an alias for the kernel -- the kext was
121305097246SAdrian Prantl     // compiled into the kernel or something -- then we don't want to load the
121405097246SAdrian Prantl     // kernel's text section at a different address.  Ignore this kext entry.
1215c1871475SJason Molenda     if (kext_summaries[new_kext].GetUUID().IsValid()
1216c1871475SJason Molenda         && m_kernel.GetUUID().IsValid()
1217c1871475SJason Molenda         && kext_summaries[new_kext].GetUUID() == m_kernel.GetUUID()) {
1218c1871475SJason Molenda       to_be_added[new_kext] = false;
1219c1871475SJason Molenda       break;
1220c1871475SJason Molenda     }
1221b9c1b51eSKate Stone     if (add_this_one) {
1222306bd0aaSJason Molenda       number_of_new_kexts_being_added++;
1223306bd0aaSJason Molenda     }
1224306bd0aaSJason Molenda   }
1225306bd0aaSJason Molenda 
1226b9c1b51eSKate Stone   if (number_of_new_kexts_being_added == 0 &&
1227b9c1b51eSKate Stone       number_of_old_kexts_being_removed == 0)
1228306bd0aaSJason Molenda     return true;
1229306bd0aaSJason Molenda 
123044d93782SGreg Clayton   Stream *s = m_process->GetTarget().GetDebugger().GetOutputFile().get();
1231b9c1b51eSKate Stone   if (s && load_kexts) {
1232b9c1b51eSKate Stone     if (number_of_new_kexts_being_added > 0 &&
1233b9c1b51eSKate Stone         number_of_old_kexts_being_removed > 0) {
1234b9c1b51eSKate Stone       s->Printf("Loading %d kext modules and unloading %d kext modules ",
1235b9c1b51eSKate Stone                 number_of_new_kexts_being_added,
1236b9c1b51eSKate Stone                 number_of_old_kexts_being_removed);
1237b9c1b51eSKate Stone     } else if (number_of_new_kexts_being_added > 0) {
1238306bd0aaSJason Molenda       s->Printf("Loading %d kext modules ", number_of_new_kexts_being_added);
1239b9c1b51eSKate Stone     } else if (number_of_old_kexts_being_removed > 0) {
1240b9c1b51eSKate Stone       s->Printf("Unloading %d kext modules ",
1241b9c1b51eSKate Stone                 number_of_old_kexts_being_removed);
1242306bd0aaSJason Molenda     }
1243306bd0aaSJason Molenda   }
1244306bd0aaSJason Molenda 
1245b9c1b51eSKate Stone   if (log) {
1246b9c1b51eSKate Stone     if (load_kexts) {
1247b9c1b51eSKate Stone       log->Printf("DynamicLoaderDarwinKernel::ParseKextSummaries: %d kexts "
1248b9c1b51eSKate Stone                   "added, %d kexts removed",
1249b9c1b51eSKate Stone                   number_of_new_kexts_being_added,
1250b9c1b51eSKate Stone                   number_of_old_kexts_being_removed);
1251b9c1b51eSKate Stone     } else {
1252b9c1b51eSKate Stone       log->Printf(
1253b9c1b51eSKate Stone           "DynamicLoaderDarwinKernel::ParseKextSummaries kext loading is "
1254b9c1b51eSKate Stone           "disabled, else would have %d kexts added, %d kexts removed",
1255b9c1b51eSKate Stone           number_of_new_kexts_being_added, number_of_old_kexts_being_removed);
12564da2e32fSJason Molenda     }
12574da2e32fSJason Molenda   }
12584da2e32fSJason Molenda 
1259b9c1b51eSKate Stone   if (number_of_new_kexts_being_added > 0) {
1260306bd0aaSJason Molenda     ModuleList loaded_module_list;
1261306bd0aaSJason Molenda 
1262306bd0aaSJason Molenda     const uint32_t num_of_new_kexts = kext_summaries.size();
1263b9c1b51eSKate Stone     for (uint32_t new_kext = 0; new_kext < num_of_new_kexts; new_kext++) {
1264b9c1b51eSKate Stone       if (to_be_added[new_kext] == true) {
1265306bd0aaSJason Molenda         KextImageInfo &image_info = kext_summaries[new_kext];
1266b9c1b51eSKate Stone         if (load_kexts) {
1267b9c1b51eSKate Stone           if (!image_info.LoadImageUsingMemoryModule(m_process)) {
1268306bd0aaSJason Molenda             image_info.LoadImageAtFileAddress(m_process);
12694da2e32fSJason Molenda           }
12704da2e32fSJason Molenda         }
1271306bd0aaSJason Molenda 
1272306bd0aaSJason Molenda         m_known_kexts.push_back(image_info);
1273306bd0aaSJason Molenda 
1274b9c1b51eSKate Stone         if (image_info.GetModule() &&
1275b9c1b51eSKate Stone             m_process->GetStopID() == image_info.GetProcessStopId())
1276306bd0aaSJason Molenda           loaded_module_list.AppendIfNeeded(image_info.GetModule());
1277d4bfbc9aSGreg Clayton 
12784da2e32fSJason Molenda         if (s && load_kexts)
127968b3607fSJason Molenda           s->Printf(".");
1280d4bfbc9aSGreg Clayton 
1281d4bfbc9aSGreg Clayton         if (log)
12825160ce5cSGreg Clayton           kext_summaries[new_kext].PutToLog(log);
1283d4bfbc9aSGreg Clayton       }
1284306bd0aaSJason Molenda     }
1285306bd0aaSJason Molenda     m_process->GetTarget().ModulesDidLoad(loaded_module_list);
1286306bd0aaSJason Molenda   }
1287306bd0aaSJason Molenda 
1288b9c1b51eSKate Stone   if (number_of_old_kexts_being_removed > 0) {
1289306bd0aaSJason Molenda     ModuleList loaded_module_list;
1290306bd0aaSJason Molenda     const uint32_t num_of_old_kexts = m_known_kexts.size();
1291b9c1b51eSKate Stone     for (uint32_t old_kext = 0; old_kext < num_of_old_kexts; old_kext++) {
1292306bd0aaSJason Molenda       ModuleList unloaded_module_list;
1293b9c1b51eSKate Stone       if (to_be_removed[old_kext]) {
1294306bd0aaSJason Molenda         KextImageInfo &image_info = m_known_kexts[old_kext];
1295306bd0aaSJason Molenda         // You can't unload the kernel.
1296b9c1b51eSKate Stone         if (!image_info.IsKernel()) {
1297b9c1b51eSKate Stone           if (image_info.GetModule()) {
1298306bd0aaSJason Molenda             unloaded_module_list.AppendIfNeeded(image_info.GetModule());
1299306bd0aaSJason Molenda           }
1300306bd0aaSJason Molenda           if (s)
1301306bd0aaSJason Molenda             s->Printf(".");
1302306bd0aaSJason Molenda           image_info.Clear();
1303b9c1b51eSKate Stone           // should pull it out of the KextImageInfos vector but that would
130405097246SAdrian Prantl           // mutate the list and invalidate the to_be_removed bool vector;
130505097246SAdrian Prantl           // leaving it in place once Cleared() is relatively harmless.
1306306bd0aaSJason Molenda         }
1307306bd0aaSJason Molenda       }
1308095eeaa0SGreg Clayton       m_process->GetTarget().ModulesDidUnload(unloaded_module_list, false);
1309306bd0aaSJason Molenda     }
1310306bd0aaSJason Molenda   }
1311306bd0aaSJason Molenda 
1312b9c1b51eSKate Stone   if (s && load_kexts) {
131368b3607fSJason Molenda     s->Printf(" done.\n");
131468b3607fSJason Molenda     s->Flush();
131568b3607fSJason Molenda   }
131668b3607fSJason Molenda 
1317306bd0aaSJason Molenda   return true;
1318d4bfbc9aSGreg Clayton }
1319d4bfbc9aSGreg Clayton 
1320b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::ReadKextSummaries(
1321b9c1b51eSKate Stone     const Address &kext_summary_addr, uint32_t image_infos_count,
1322b9c1b51eSKate Stone     KextImageInfo::collection &image_infos) {
1323d4bfbc9aSGreg Clayton   const ByteOrder endian = m_kernel.GetByteOrder();
1324d4bfbc9aSGreg Clayton   const uint32_t addr_size = m_kernel.GetAddressByteSize();
1325d4bfbc9aSGreg Clayton 
1326d4bfbc9aSGreg Clayton   image_infos.resize(image_infos_count);
1327d4bfbc9aSGreg Clayton   const size_t count = image_infos.size() * m_kext_summary_header.entry_size;
1328d4bfbc9aSGreg Clayton   DataBufferHeap data(count, 0);
132997206d57SZachary Turner   Status error;
1330d4bfbc9aSGreg Clayton 
1331d4bfbc9aSGreg Clayton   const bool prefer_file_cache = false;
1332b9c1b51eSKate Stone   const size_t bytes_read = m_process->GetTarget().ReadMemory(
1333b9c1b51eSKate Stone       kext_summary_addr, prefer_file_cache, data.GetBytes(), data.GetByteSize(),
1334d4bfbc9aSGreg Clayton       error);
1335b9c1b51eSKate Stone   if (bytes_read == count) {
1336d4bfbc9aSGreg Clayton 
1337b9c1b51eSKate Stone     DataExtractor extractor(data.GetBytes(), data.GetByteSize(), endian,
1338b9c1b51eSKate Stone                             addr_size);
1339d4bfbc9aSGreg Clayton     uint32_t i = 0;
1340d4bfbc9aSGreg Clayton     for (uint32_t kext_summary_offset = 0;
1341b9c1b51eSKate Stone          i < image_infos.size() &&
1342b9c1b51eSKate Stone          extractor.ValidOffsetForDataOfSize(kext_summary_offset,
1343b9c1b51eSKate Stone                                             m_kext_summary_header.entry_size);
1344b9c1b51eSKate Stone          ++i, kext_summary_offset += m_kext_summary_header.entry_size) {
1345c7bece56SGreg Clayton       lldb::offset_t offset = kext_summary_offset;
1346b9c1b51eSKate Stone       const void *name_data =
1347b9c1b51eSKate Stone           extractor.GetData(&offset, KERNEL_MODULE_MAX_NAME);
1348d4bfbc9aSGreg Clayton       if (name_data == NULL)
1349d4bfbc9aSGreg Clayton         break;
1350306bd0aaSJason Molenda       image_infos[i].SetName((const char *)name_data);
1351306bd0aaSJason Molenda       UUID uuid(extractor.GetData(&offset, 16), 16);
1352306bd0aaSJason Molenda       image_infos[i].SetUUID(uuid);
1353306bd0aaSJason Molenda       image_infos[i].SetLoadAddress(extractor.GetU64(&offset));
1354306bd0aaSJason Molenda       image_infos[i].SetSize(extractor.GetU64(&offset));
1355d4bfbc9aSGreg Clayton     }
1356d4bfbc9aSGreg Clayton     if (i < image_infos.size())
1357d4bfbc9aSGreg Clayton       image_infos.resize(i);
1358b9c1b51eSKate Stone   } else {
1359d4bfbc9aSGreg Clayton     image_infos.clear();
1360d4bfbc9aSGreg Clayton   }
1361d4bfbc9aSGreg Clayton   return image_infos.size();
1362d4bfbc9aSGreg Clayton }
1363d4bfbc9aSGreg Clayton 
1364b9c1b51eSKate Stone bool DynamicLoaderDarwinKernel::ReadAllKextSummaries() {
136516ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1366d4bfbc9aSGreg Clayton 
1367b9c1b51eSKate Stone   if (ReadKextSummaryHeader()) {
1368b9c1b51eSKate Stone     if (m_kext_summary_header.entry_count > 0 &&
1369b9c1b51eSKate Stone         m_kext_summary_header_addr.IsValid()) {
1370d4bfbc9aSGreg Clayton       Address summary_addr(m_kext_summary_header_addr);
1371d4bfbc9aSGreg Clayton       summary_addr.Slide(m_kext_summary_header.GetSize());
1372b9c1b51eSKate Stone       if (!ParseKextSummaries(summary_addr,
1373b9c1b51eSKate Stone                               m_kext_summary_header.entry_count)) {
1374306bd0aaSJason Molenda         m_known_kexts.clear();
1375d4bfbc9aSGreg Clayton       }
1376d4bfbc9aSGreg Clayton       return true;
1377d4bfbc9aSGreg Clayton     }
1378d4bfbc9aSGreg Clayton   }
1379d4bfbc9aSGreg Clayton   return false;
1380d4bfbc9aSGreg Clayton }
1381d4bfbc9aSGreg Clayton 
1382d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1383d4bfbc9aSGreg Clayton // Dump an image info structure to the file handle provided.
1384d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1385b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::KextImageInfo::PutToLog(Log *log) const {
1386d4bfbc9aSGreg Clayton   if (log == NULL)
1387d4bfbc9aSGreg Clayton     return;
13888465c38bSVedant Kumar   const uint8_t *u = static_cast<const uint8_t *>(m_uuid.GetBytes());
1389d4bfbc9aSGreg Clayton 
1390b9c1b51eSKate Stone   if (m_load_address == LLDB_INVALID_ADDRESS) {
1391b9c1b51eSKate Stone     if (u) {
1392b9c1b51eSKate Stone       log->Printf("\tuuid=%2.2X%2.2X%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2."
1393b9c1b51eSKate Stone                   "2X-%2.2X%2.2X%2.2X%2.2X%2.2X%2.2X name=\"%s\" (UNLOADED)",
1394b9c1b51eSKate Stone                   u[0], u[1], u[2], u[3], u[4], u[5], u[6], u[7], u[8], u[9],
1395b9c1b51eSKate Stone                   u[10], u[11], u[12], u[13], u[14], u[15], m_name.c_str());
1396b9c1b51eSKate Stone     } else
1397306bd0aaSJason Molenda       log->Printf("\tname=\"%s\" (UNLOADED)", m_name.c_str());
1398b9c1b51eSKate Stone   } else {
1399b9c1b51eSKate Stone     if (u) {
1400b9c1b51eSKate Stone       log->Printf("\taddr=0x%16.16" PRIx64 " size=0x%16.16" PRIx64
1401b9c1b51eSKate Stone                   " uuid=%2.2X%2.2X%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-"
1402b9c1b51eSKate Stone                   "%2.2X%2.2X%2.2X%2.2X%2.2X%2.2X name=\"%s\"",
1403b9c1b51eSKate Stone                   m_load_address, m_size, u[0], u[1], u[2], u[3], u[4], u[5],
1404b9c1b51eSKate Stone                   u[6], u[7], u[8], u[9], u[10], u[11], u[12], u[13], u[14],
1405b9c1b51eSKate Stone                   u[15], m_name.c_str());
1406b9c1b51eSKate Stone     } else {
1407306bd0aaSJason Molenda       log->Printf("\t[0x%16.16" PRIx64 " - 0x%16.16" PRIx64 ") name=\"%s\"",
1408306bd0aaSJason Molenda                   m_load_address, m_load_address + m_size, m_name.c_str());
1409d4bfbc9aSGreg Clayton     }
1410d4bfbc9aSGreg Clayton   }
1411d4bfbc9aSGreg Clayton }
1412d4bfbc9aSGreg Clayton 
1413d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
141405097246SAdrian Prantl // Dump the _dyld_all_image_infos members and all current image infos that we
141505097246SAdrian Prantl // have parsed to the file handle provided.
1416d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1417b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PutToLog(Log *log) const {
1418d4bfbc9aSGreg Clayton   if (log == NULL)
1419d4bfbc9aSGreg Clayton     return;
1420d4bfbc9aSGreg Clayton 
142116ff8604SSaleem Abdulrasool   std::lock_guard<std::recursive_mutex> guard(m_mutex);
1422b9c1b51eSKate Stone   log->Printf("gLoadedKextSummaries = 0x%16.16" PRIx64
1423b9c1b51eSKate Stone               " { version=%u, entry_size=%u, entry_count=%u }",
1424d4bfbc9aSGreg Clayton               m_kext_summary_header_addr.GetFileAddress(),
1425b9c1b51eSKate Stone               m_kext_summary_header.version, m_kext_summary_header.entry_size,
1426d4bfbc9aSGreg Clayton               m_kext_summary_header.entry_count);
1427d4bfbc9aSGreg Clayton 
1428d4bfbc9aSGreg Clayton   size_t i;
1429306bd0aaSJason Molenda   const size_t count = m_known_kexts.size();
1430b9c1b51eSKate Stone   if (count > 0) {
1431d4bfbc9aSGreg Clayton     log->PutCString("Loaded:");
1432d4bfbc9aSGreg Clayton     for (i = 0; i < count; i++)
1433306bd0aaSJason Molenda       m_known_kexts[i].PutToLog(log);
1434d4bfbc9aSGreg Clayton   }
1435d4bfbc9aSGreg Clayton }
1436d4bfbc9aSGreg Clayton 
1437b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PrivateInitialize(Process *process) {
1438b9c1b51eSKate Stone   DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n",
1439b9c1b51eSKate Stone                __FUNCTION__, StateAsCString(m_process->GetState()));
1440d4bfbc9aSGreg Clayton   Clear(true);
1441d4bfbc9aSGreg Clayton   m_process = process;
1442d4bfbc9aSGreg Clayton }
1443d4bfbc9aSGreg Clayton 
1444b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::SetNotificationBreakpointIfNeeded() {
1445b9c1b51eSKate Stone   if (m_break_id == LLDB_INVALID_BREAK_ID && m_kernel.GetModule()) {
1446b9c1b51eSKate Stone     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n",
1447b9c1b51eSKate Stone                  __FUNCTION__, StateAsCString(m_process->GetState()));
1448d4bfbc9aSGreg Clayton 
1449a8558b62SJim Ingham     const bool internal_bp = true;
1450eb023e75SGreg Clayton     const bool hardware = false;
1451d4bfbc9aSGreg Clayton     const LazyBool skip_prologue = eLazyBoolNo;
1452969795f1SJim Ingham     FileSpecList module_spec_list;
1453306bd0aaSJason Molenda     module_spec_list.Append(m_kernel.GetModule()->GetFileSpec());
1454b9c1b51eSKate Stone     Breakpoint *bp =
1455b9c1b51eSKate Stone         m_process->GetTarget()
1456b9c1b51eSKate Stone             .CreateBreakpoint(&module_spec_list, NULL,
1457d4bfbc9aSGreg Clayton                               "OSKextLoadedKextSummariesUpdated",
1458b9c1b51eSKate Stone                               eFunctionNameTypeFull, eLanguageTypeUnknown, 0,
1459b9c1b51eSKate Stone                               skip_prologue, internal_bp, hardware)
1460b9c1b51eSKate Stone             .get();
1461d4bfbc9aSGreg Clayton 
1462b9c1b51eSKate Stone     bp->SetCallback(DynamicLoaderDarwinKernel::BreakpointHitCallback, this,
1463b9c1b51eSKate Stone                     true);
1464d4bfbc9aSGreg Clayton     m_break_id = bp->GetID();
1465d4bfbc9aSGreg Clayton   }
1466d4bfbc9aSGreg Clayton }
1467d4bfbc9aSGreg Clayton 
1468d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1469d4bfbc9aSGreg Clayton // Member function that gets called when the process state changes.
1470d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1471b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::PrivateProcessStateChanged(Process *process,
1472b9c1b51eSKate Stone                                                            StateType state) {
1473b9c1b51eSKate Stone   DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s(%s)\n", __FUNCTION__,
1474b9c1b51eSKate Stone                StateAsCString(state));
1475b9c1b51eSKate Stone   switch (state) {
1476d4bfbc9aSGreg Clayton   case eStateConnected:
1477d4bfbc9aSGreg Clayton   case eStateAttaching:
1478d4bfbc9aSGreg Clayton   case eStateLaunching:
1479d4bfbc9aSGreg Clayton   case eStateInvalid:
1480d4bfbc9aSGreg Clayton   case eStateUnloaded:
1481d4bfbc9aSGreg Clayton   case eStateExited:
1482d4bfbc9aSGreg Clayton   case eStateDetached:
1483d4bfbc9aSGreg Clayton     Clear(false);
1484d4bfbc9aSGreg Clayton     break;
1485d4bfbc9aSGreg Clayton 
1486d4bfbc9aSGreg Clayton   case eStateStopped:
1487d4bfbc9aSGreg Clayton     UpdateIfNeeded();
1488d4bfbc9aSGreg Clayton     break;
1489d4bfbc9aSGreg Clayton 
1490d4bfbc9aSGreg Clayton   case eStateRunning:
1491d4bfbc9aSGreg Clayton   case eStateStepping:
1492d4bfbc9aSGreg Clayton   case eStateCrashed:
1493d4bfbc9aSGreg Clayton   case eStateSuspended:
1494d4bfbc9aSGreg Clayton     break;
1495d4bfbc9aSGreg Clayton   }
1496d4bfbc9aSGreg Clayton }
1497d4bfbc9aSGreg Clayton 
1498d4bfbc9aSGreg Clayton ThreadPlanSP
1499b9c1b51eSKate Stone DynamicLoaderDarwinKernel::GetStepThroughTrampolinePlan(Thread &thread,
1500b9c1b51eSKate Stone                                                         bool stop_others) {
1501d4bfbc9aSGreg Clayton   ThreadPlanSP thread_plan_sp;
15025160ce5cSGreg Clayton   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_STEP));
1503d4bfbc9aSGreg Clayton   if (log)
1504d4bfbc9aSGreg Clayton     log->Printf("Could not find symbol for step through.");
1505d4bfbc9aSGreg Clayton   return thread_plan_sp;
1506d4bfbc9aSGreg Clayton }
1507d4bfbc9aSGreg Clayton 
150897206d57SZachary Turner Status DynamicLoaderDarwinKernel::CanLoadImage() {
150997206d57SZachary Turner   Status error;
1510b9c1b51eSKate Stone   error.SetErrorString(
1511b9c1b51eSKate Stone       "always unsafe to load or unload shared libraries in the darwin kernel");
1512d4bfbc9aSGreg Clayton   return error;
1513d4bfbc9aSGreg Clayton }
1514d4bfbc9aSGreg Clayton 
1515b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Initialize() {
1516d4bfbc9aSGreg Clayton   PluginManager::RegisterPlugin(GetPluginNameStatic(),
1517b9c1b51eSKate Stone                                 GetPluginDescriptionStatic(), CreateInstance,
1518e8cd0c98SGreg Clayton                                 DebuggerInitialize);
1519d4bfbc9aSGreg Clayton }
1520d4bfbc9aSGreg Clayton 
1521b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::Terminate() {
1522d4bfbc9aSGreg Clayton   PluginManager::UnregisterPlugin(CreateInstance);
1523d4bfbc9aSGreg Clayton }
1524d4bfbc9aSGreg Clayton 
1525b9c1b51eSKate Stone void DynamicLoaderDarwinKernel::DebuggerInitialize(
1526b9c1b51eSKate Stone     lldb_private::Debugger &debugger) {
1527b9c1b51eSKate Stone   if (!PluginManager::GetSettingForDynamicLoaderPlugin(
1528b9c1b51eSKate Stone           debugger, DynamicLoaderDarwinKernelProperties::GetSettingName())) {
1529e8cd0c98SGreg Clayton     const bool is_global_setting = true;
1530b9c1b51eSKate Stone     PluginManager::CreateSettingForDynamicLoaderPlugin(
1531b9c1b51eSKate Stone         debugger, GetGlobalProperties()->GetValueProperties(),
1532e8cd0c98SGreg Clayton         ConstString("Properties for the DynamicLoaderDarwinKernel plug-in."),
1533e8cd0c98SGreg Clayton         is_global_setting);
1534e8cd0c98SGreg Clayton   }
1535e8cd0c98SGreg Clayton }
1536d4bfbc9aSGreg Clayton 
1537b9c1b51eSKate Stone lldb_private::ConstString DynamicLoaderDarwinKernel::GetPluginNameStatic() {
153857abc5d6SGreg Clayton   static ConstString g_name("darwin-kernel");
153957abc5d6SGreg Clayton   return g_name;
1540d4bfbc9aSGreg Clayton }
1541d4bfbc9aSGreg Clayton 
1542b9c1b51eSKate Stone const char *DynamicLoaderDarwinKernel::GetPluginDescriptionStatic() {
1543b9c1b51eSKate Stone   return "Dynamic loader plug-in that watches for shared library loads/unloads "
1544b9c1b51eSKate Stone          "in the MacOSX kernel.";
1545d4bfbc9aSGreg Clayton }
1546d4bfbc9aSGreg Clayton 
1547d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1548d4bfbc9aSGreg Clayton // PluginInterface protocol
1549d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1550b9c1b51eSKate Stone lldb_private::ConstString DynamicLoaderDarwinKernel::GetPluginName() {
1551d4bfbc9aSGreg Clayton   return GetPluginNameStatic();
1552d4bfbc9aSGreg Clayton }
1553d4bfbc9aSGreg Clayton 
1554b9c1b51eSKate Stone uint32_t DynamicLoaderDarwinKernel::GetPluginVersion() { return 1; }
1555d4bfbc9aSGreg Clayton 
15561f746071SGreg Clayton lldb::ByteOrder
1557b9c1b51eSKate Stone DynamicLoaderDarwinKernel::GetByteOrderFromMagic(uint32_t magic) {
1558b9c1b51eSKate Stone   switch (magic) {
1559510938e5SCharles Davis   case llvm::MachO::MH_MAGIC:
1560510938e5SCharles Davis   case llvm::MachO::MH_MAGIC_64:
15619ccb970fSBruce Mitchener     return endian::InlHostByteOrder();
15621f746071SGreg Clayton 
1563510938e5SCharles Davis   case llvm::MachO::MH_CIGAM:
1564510938e5SCharles Davis   case llvm::MachO::MH_CIGAM_64:
15659ccb970fSBruce Mitchener     if (endian::InlHostByteOrder() == lldb::eByteOrderBig)
15661f746071SGreg Clayton       return lldb::eByteOrderLittle;
15671f746071SGreg Clayton     else
15681f746071SGreg Clayton       return lldb::eByteOrderBig;
15691f746071SGreg Clayton 
15701f746071SGreg Clayton   default:
15711f746071SGreg Clayton     break;
15721f746071SGreg Clayton   }
15731f746071SGreg Clayton   return lldb::eByteOrderInvalid;
15741f746071SGreg Clayton }
1575