1944b828aSGreg Clayton //===-- DynamicLoaderDarwinKernel.cpp -----------------------------*- C++ -*-===//
2d4bfbc9aSGreg Clayton //
3d4bfbc9aSGreg Clayton //                     The LLVM Compiler Infrastructure
4d4bfbc9aSGreg Clayton //
5d4bfbc9aSGreg Clayton // This file is distributed under the University of Illinois Open Source
6d4bfbc9aSGreg Clayton // License. See LICENSE.TXT for details.
7d4bfbc9aSGreg Clayton //
8d4bfbc9aSGreg Clayton //===----------------------------------------------------------------------===//
9d4bfbc9aSGreg Clayton 
1093a64300SDaniel Malea #include "lldb/lldb-python.h"
1193a64300SDaniel Malea 
12d4bfbc9aSGreg Clayton #include "lldb/Breakpoint/StoppointCallbackContext.h"
13d4bfbc9aSGreg Clayton #include "lldb/Core/DataBuffer.h"
14d4bfbc9aSGreg Clayton #include "lldb/Core/DataBufferHeap.h"
15d4bfbc9aSGreg Clayton #include "lldb/Core/Debugger.h"
16d4bfbc9aSGreg Clayton #include "lldb/Core/Log.h"
17d4bfbc9aSGreg Clayton #include "lldb/Core/Module.h"
181f746071SGreg Clayton #include "lldb/Core/ModuleSpec.h"
19d4bfbc9aSGreg Clayton #include "lldb/Core/PluginManager.h"
201f746071SGreg Clayton #include "lldb/Core/Section.h"
21d4bfbc9aSGreg Clayton #include "lldb/Core/State.h"
2268b3607fSJason Molenda #include "lldb/Host/Symbols.h"
23d4bfbc9aSGreg Clayton #include "lldb/Symbol/ObjectFile.h"
24d4bfbc9aSGreg Clayton #include "lldb/Target/RegisterContext.h"
2568b3607fSJason Molenda #include "lldb/Target/StackFrame.h"
26d4bfbc9aSGreg Clayton #include "lldb/Target/Target.h"
27d4bfbc9aSGreg Clayton #include "lldb/Target/Thread.h"
28d4bfbc9aSGreg Clayton #include "lldb/Target/ThreadPlanRunToAddress.h"
2968b3607fSJason Molenda 
30d4bfbc9aSGreg Clayton 
31944b828aSGreg Clayton #include "DynamicLoaderDarwinKernel.h"
32d4bfbc9aSGreg Clayton 
33d4bfbc9aSGreg Clayton //#define ENABLE_DEBUG_PRINTF // COMMENT THIS LINE OUT PRIOR TO CHECKIN
34d4bfbc9aSGreg Clayton #ifdef ENABLE_DEBUG_PRINTF
35d4bfbc9aSGreg Clayton #include <stdio.h>
36d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...) printf(fmt, ## __VA_ARGS__)
37d4bfbc9aSGreg Clayton #else
38d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...)
39d4bfbc9aSGreg Clayton #endif
40d4bfbc9aSGreg Clayton 
41d4bfbc9aSGreg Clayton using namespace lldb;
42d4bfbc9aSGreg Clayton using namespace lldb_private;
43d4bfbc9aSGreg Clayton 
446ba6d3d1SJason Molenda // Progressively greater amounts of scanning we will allow
456ba6d3d1SJason Molenda // For some targets very early in startup, we can't do any random reads of memory or we can crash the device
466ba6d3d1SJason Molenda // so a setting is needed that can completely disable the KASLR scans.
476ba6d3d1SJason Molenda 
486ba6d3d1SJason Molenda enum KASLRScanType
496ba6d3d1SJason Molenda {
506ba6d3d1SJason Molenda     eKASLRScanNone = 0,         // No reading into the inferior at all
516ba6d3d1SJason Molenda     eKASLRScanLowgloAddresses,  // Check one word of memory for a possible kernel addr, then see if a kernel is there
5263969221SJason Molenda     eKASLRScanNearPC,           // Scan backwards from the current $pc looking for kernel; checking at 96 locations total
536ba6d3d1SJason Molenda     eKASLRScanExhaustiveScan    // Scan through the entire possible kernel address range looking for a kernel
546ba6d3d1SJason Molenda };
556ba6d3d1SJason Molenda 
566ba6d3d1SJason Molenda OptionEnumValueElement
576ba6d3d1SJason Molenda g_kaslr_kernel_scan_enum_values[] =
586ba6d3d1SJason Molenda {
596ba6d3d1SJason Molenda     { eKASLRScanNone,            "none",            "Do not read memory looking for a Darwin kernel when attaching." },
606ba6d3d1SJason Molenda     { eKASLRScanLowgloAddresses, "basic",           "Check for the Darwin kernel's load addr in the lowglo page (boot-args=debug) only." },
616ba6d3d1SJason Molenda     { eKASLRScanNearPC,          "fast-scan",       "Scan near the pc value on attach to find the Darwin kernel's load address."},
626ba6d3d1SJason Molenda     { eKASLRScanExhaustiveScan,  "exhaustive-scan", "Scan through the entire potential address range of Darwin kernel (only on 32-bit targets)."},
636ba6d3d1SJason Molenda     { 0, NULL, NULL }
646ba6d3d1SJason Molenda };
656ba6d3d1SJason Molenda 
66e8cd0c98SGreg Clayton static PropertyDefinition
67e8cd0c98SGreg Clayton g_properties[] =
68e8cd0c98SGreg Clayton {
6966763eedSGreg Clayton     { "load-kexts" , OptionValue::eTypeBoolean, true, true, NULL, NULL, "Automatically loads kext images when attaching to a kernel." },
706ba6d3d1SJason Molenda     { "scan-type",   OptionValue::eTypeEnum,    true, eKASLRScanNearPC, NULL, g_kaslr_kernel_scan_enum_values, "Control how many reads lldb will make while searching for a Darwin kernel on attach." },
71e8cd0c98SGreg Clayton     {  NULL        , OptionValue::eTypeInvalid, false, 0  , NULL, NULL, NULL  }
72e8cd0c98SGreg Clayton };
73e8cd0c98SGreg Clayton 
74e8cd0c98SGreg Clayton enum {
756ba6d3d1SJason Molenda     ePropertyLoadKexts,
766ba6d3d1SJason Molenda     ePropertyScanType
77e8cd0c98SGreg Clayton };
78e8cd0c98SGreg Clayton 
79e8cd0c98SGreg Clayton class DynamicLoaderDarwinKernelProperties : public Properties
80e8cd0c98SGreg Clayton {
81e8cd0c98SGreg Clayton public:
82e8cd0c98SGreg Clayton 
83e8cd0c98SGreg Clayton     static ConstString &
84e8cd0c98SGreg Clayton     GetSettingName ()
85e8cd0c98SGreg Clayton     {
86468ea4ebSGreg Clayton         static ConstString g_setting_name("darwin-kernel");
87e8cd0c98SGreg Clayton         return g_setting_name;
88e8cd0c98SGreg Clayton     }
89e8cd0c98SGreg Clayton 
90e8cd0c98SGreg Clayton     DynamicLoaderDarwinKernelProperties() :
91e8cd0c98SGreg Clayton         Properties ()
92e8cd0c98SGreg Clayton     {
93e8cd0c98SGreg Clayton         m_collection_sp.reset (new OptionValueProperties(GetSettingName()));
94e8cd0c98SGreg Clayton         m_collection_sp->Initialize(g_properties);
95e8cd0c98SGreg Clayton     }
96e8cd0c98SGreg Clayton 
97e8cd0c98SGreg Clayton     virtual
98e8cd0c98SGreg Clayton     ~DynamicLoaderDarwinKernelProperties()
99e8cd0c98SGreg Clayton     {
100e8cd0c98SGreg Clayton     }
101e8cd0c98SGreg Clayton 
102e8cd0c98SGreg Clayton     bool
10366763eedSGreg Clayton     GetLoadKexts() const
104e8cd0c98SGreg Clayton     {
10566763eedSGreg Clayton         const uint32_t idx = ePropertyLoadKexts;
106e8cd0c98SGreg Clayton         return m_collection_sp->GetPropertyAtIndexAsBoolean (NULL, idx, g_properties[idx].default_uint_value != 0);
107e8cd0c98SGreg Clayton     }
108e8cd0c98SGreg Clayton 
1096ba6d3d1SJason Molenda     KASLRScanType
1106ba6d3d1SJason Molenda     GetScanType() const
1116ba6d3d1SJason Molenda     {
1126ba6d3d1SJason Molenda         const uint32_t idx = ePropertyScanType;
11363969221SJason Molenda         return (KASLRScanType) m_collection_sp->GetPropertyAtIndexAsEnumeration (NULL, idx, g_properties[idx].default_uint_value);
1146ba6d3d1SJason Molenda     }
1156ba6d3d1SJason Molenda 
1166ba6d3d1SJason Molenda 
117e8cd0c98SGreg Clayton };
118e8cd0c98SGreg Clayton 
119e8cd0c98SGreg Clayton typedef STD_SHARED_PTR(DynamicLoaderDarwinKernelProperties) DynamicLoaderDarwinKernelPropertiesSP;
120e8cd0c98SGreg Clayton 
121e8cd0c98SGreg Clayton static const DynamicLoaderDarwinKernelPropertiesSP &
122e8cd0c98SGreg Clayton GetGlobalProperties()
123e8cd0c98SGreg Clayton {
124e8cd0c98SGreg Clayton     static DynamicLoaderDarwinKernelPropertiesSP g_settings_sp;
125e8cd0c98SGreg Clayton     if (!g_settings_sp)
126e8cd0c98SGreg Clayton         g_settings_sp.reset (new DynamicLoaderDarwinKernelProperties ());
127e8cd0c98SGreg Clayton     return g_settings_sp;
128e8cd0c98SGreg Clayton }
129e8cd0c98SGreg Clayton 
130d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
131d4bfbc9aSGreg Clayton // Create an instance of this class. This function is filled into
132d4bfbc9aSGreg Clayton // the plugin info class that gets handed out by the plugin factory and
133d4bfbc9aSGreg Clayton // allows the lldb to instantiate an instance of this class.
134d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
135d4bfbc9aSGreg Clayton DynamicLoader *
136944b828aSGreg Clayton DynamicLoaderDarwinKernel::CreateInstance (Process* process, bool force)
137d4bfbc9aSGreg Clayton {
1386ba6d3d1SJason Molenda     if (!force)
139d4bfbc9aSGreg Clayton     {
1406ba6d3d1SJason Molenda         // If the user provided an executable binary and it is not a kernel,
1416ba6d3d1SJason Molenda         // this plugin should not create an instance.
142d4bfbc9aSGreg Clayton         Module* exe_module = process->GetTarget().GetExecutableModulePointer();
143d4bfbc9aSGreg Clayton         if (exe_module)
144d4bfbc9aSGreg Clayton         {
145d4bfbc9aSGreg Clayton             ObjectFile *object_file = exe_module->GetObjectFile();
146d4bfbc9aSGreg Clayton             if (object_file)
147d4bfbc9aSGreg Clayton             {
1486ba6d3d1SJason Molenda                 if (object_file->GetStrata() != ObjectFile::eStrataKernel)
1496ba6d3d1SJason Molenda                 {
1506ba6d3d1SJason Molenda                     return NULL;
1516ba6d3d1SJason Molenda                 }
152d4bfbc9aSGreg Clayton             }
153d4bfbc9aSGreg Clayton         }
154d4bfbc9aSGreg Clayton 
1556ba6d3d1SJason Molenda         // If the target's architecture does not look like an Apple environment,
1566ba6d3d1SJason Molenda         // this plugin should not create an instance.
157d4bfbc9aSGreg Clayton         const llvm::Triple &triple_ref = process->GetTarget().GetArchitecture().GetTriple();
15870512317SGreg Clayton         switch (triple_ref.getOS())
15970512317SGreg Clayton         {
16070512317SGreg Clayton             case llvm::Triple::Darwin:
16170512317SGreg Clayton             case llvm::Triple::MacOSX:
16270512317SGreg Clayton             case llvm::Triple::IOS:
1636ba6d3d1SJason Molenda                 if (triple_ref.getVendor() != llvm::Triple::Apple)
1646ba6d3d1SJason Molenda                 {
1656ba6d3d1SJason Molenda                    return NULL;
1666ba6d3d1SJason Molenda                 }
1676ba6d3d1SJason Molenda                 break;
1686ba6d3d1SJason Molenda             // If we have triple like armv7-unknown-unknown, we should try looking for a Darwin kernel.
1696ba6d3d1SJason Molenda             case llvm::Triple::UnknownOS:
17070512317SGreg Clayton                 break;
17170512317SGreg Clayton             default:
1726ba6d3d1SJason Molenda                 return NULL;
17370512317SGreg Clayton                 break;
17470512317SGreg Clayton         }
175d4bfbc9aSGreg Clayton     }
1766ba6d3d1SJason Molenda 
1776ba6d3d1SJason Molenda     // At this point if there is an ExecutableModule, it is a kernel and the Target is some variant of an Apple system.
1786ba6d3d1SJason Molenda     // If the Process hasn't provided the kernel load address, we need to look around in memory to find it.
1796ba6d3d1SJason Molenda 
1806ba6d3d1SJason Molenda     addr_t kernel_load_address = process->GetImageInfoAddress();
1816ba6d3d1SJason Molenda     if (kernel_load_address == LLDB_INVALID_ADDRESS)
1826ba6d3d1SJason Molenda     {
1836ba6d3d1SJason Molenda         kernel_load_address = SearchForKernelAtSameLoadAddr (process);
1846ba6d3d1SJason Molenda         if (kernel_load_address == LLDB_INVALID_ADDRESS)
1856ba6d3d1SJason Molenda         {
1866ba6d3d1SJason Molenda             kernel_load_address = SearchForKernelWithDebugHints (process);
1876ba6d3d1SJason Molenda             if (kernel_load_address == LLDB_INVALID_ADDRESS)
1886ba6d3d1SJason Molenda             {
1896ba6d3d1SJason Molenda                 kernel_load_address = SearchForKernelNearPC (process);
1906ba6d3d1SJason Molenda                 if (kernel_load_address == LLDB_INVALID_ADDRESS)
1916ba6d3d1SJason Molenda                 {
1926ba6d3d1SJason Molenda                     kernel_load_address = SearchForKernelViaExhaustiveSearch (process);
1936ba6d3d1SJason Molenda                 }
1946ba6d3d1SJason Molenda             }
1956ba6d3d1SJason Molenda         }
196d4bfbc9aSGreg Clayton     }
197d4bfbc9aSGreg Clayton 
1986ba6d3d1SJason Molenda     if (kernel_load_address != LLDB_INVALID_ADDRESS)
19990539456SSean Callanan     {
20090539456SSean Callanan         process->SetCanJIT(false);
2016ba6d3d1SJason Molenda         return new DynamicLoaderDarwinKernel (process, kernel_load_address);
20290539456SSean Callanan     }
203d4bfbc9aSGreg Clayton     return NULL;
204d4bfbc9aSGreg Clayton }
205d4bfbc9aSGreg Clayton 
206d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
2076ba6d3d1SJason Molenda // Check if the kernel binary is loaded in memory without a slide.
2086ba6d3d1SJason Molenda // First verify that the ExecutableModule is a kernel before we proceed.
2096ba6d3d1SJason Molenda // Returns the address of the kernel if one was found, else LLDB_INVALID_ADDRESS.
2106ba6d3d1SJason Molenda //----------------------------------------------------------------------
2116ba6d3d1SJason Molenda lldb::addr_t
2126ba6d3d1SJason Molenda DynamicLoaderDarwinKernel::SearchForKernelAtSameLoadAddr (Process *process)
2136ba6d3d1SJason Molenda {
2146ba6d3d1SJason Molenda     Module *exe_module = process->GetTarget().GetExecutableModulePointer();
2156ba6d3d1SJason Molenda     if (exe_module == NULL)
2166ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2176ba6d3d1SJason Molenda 
2186ba6d3d1SJason Molenda     ObjectFile *exe_objfile = exe_module->GetObjectFile();
2196ba6d3d1SJason Molenda     if (exe_objfile == NULL)
2206ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2216ba6d3d1SJason Molenda 
2226ba6d3d1SJason Molenda     if (exe_objfile->GetType() != ObjectFile::eTypeExecutable || exe_objfile->GetStrata() != ObjectFile::eStrataKernel)
2236ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2246ba6d3d1SJason Molenda 
2256ba6d3d1SJason Molenda     if (!exe_objfile->GetHeaderAddress().IsValid())
2266ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2276ba6d3d1SJason Molenda 
2286ba6d3d1SJason Molenda     if (CheckForKernelImageAtAddress (exe_objfile->GetHeaderAddress().GetFileAddress(), process) == exe_module->GetUUID())
2296ba6d3d1SJason Molenda         return exe_objfile->GetHeaderAddress().GetFileAddress();
2306ba6d3d1SJason Molenda 
2316ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2326ba6d3d1SJason Molenda }
2336ba6d3d1SJason Molenda 
2346ba6d3d1SJason Molenda //----------------------------------------------------------------------
2356ba6d3d1SJason Molenda // If the debug flag is included in the boot-args nvram setting, the kernel's load address
2366ba6d3d1SJason Molenda // will be noted in the lowglo page at a fixed address
2376ba6d3d1SJason Molenda // Returns the address of the kernel if one was found, else LLDB_INVALID_ADDRESS.
2386ba6d3d1SJason Molenda //----------------------------------------------------------------------
2396ba6d3d1SJason Molenda lldb::addr_t
2406ba6d3d1SJason Molenda DynamicLoaderDarwinKernel::SearchForKernelWithDebugHints (Process *process)
2416ba6d3d1SJason Molenda {
2426ba6d3d1SJason Molenda     if (GetGlobalProperties()->GetScanType() == eKASLRScanNone)
2436ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2446ba6d3d1SJason Molenda 
2456ba6d3d1SJason Molenda     Error read_err;
2466ba6d3d1SJason Molenda     addr_t addr = LLDB_INVALID_ADDRESS;
2476ba6d3d1SJason Molenda     if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
2486ba6d3d1SJason Molenda     {
2496ba6d3d1SJason Molenda         addr = process->ReadUnsignedIntegerFromMemory (0xffffff8000002010ULL, 8, LLDB_INVALID_ADDRESS, read_err);
2506ba6d3d1SJason Molenda     }
2516ba6d3d1SJason Molenda     else
2526ba6d3d1SJason Molenda     {
2536ba6d3d1SJason Molenda         addr = process->ReadUnsignedIntegerFromMemory (0xffff0110, 4, LLDB_INVALID_ADDRESS, read_err);
2546ba6d3d1SJason Molenda     }
2556ba6d3d1SJason Molenda 
2566ba6d3d1SJason Molenda     if (addr == 0)
2576ba6d3d1SJason Molenda         addr = LLDB_INVALID_ADDRESS;
2586ba6d3d1SJason Molenda 
2596ba6d3d1SJason Molenda     if (addr != LLDB_INVALID_ADDRESS)
2606ba6d3d1SJason Molenda     {
2616ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr, process).IsValid())
2626ba6d3d1SJason Molenda             return addr;
2636ba6d3d1SJason Molenda     }
2646ba6d3d1SJason Molenda 
2656ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
2666ba6d3d1SJason Molenda }
2676ba6d3d1SJason Molenda 
2686ba6d3d1SJason Molenda //----------------------------------------------------------------------
2696ba6d3d1SJason Molenda // If the kernel is currently executing when lldb attaches, and we don't have
2706ba6d3d1SJason Molenda // a better way of finding the kernel's load address, try searching backwards
2716ba6d3d1SJason Molenda // from the current pc value looking for the kernel's Mach header in memory.
2726ba6d3d1SJason Molenda // Returns the address of the kernel if one was found, else LLDB_INVALID_ADDRESS.
2736ba6d3d1SJason Molenda //----------------------------------------------------------------------
2746ba6d3d1SJason Molenda lldb::addr_t
2756ba6d3d1SJason Molenda DynamicLoaderDarwinKernel::SearchForKernelNearPC (Process *process)
2766ba6d3d1SJason Molenda {
2776ba6d3d1SJason Molenda     if (GetGlobalProperties()->GetScanType() == eKASLRScanNone
2786ba6d3d1SJason Molenda         || GetGlobalProperties()->GetScanType() == eKASLRScanLowgloAddresses)
2796ba6d3d1SJason Molenda     {
2806ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2816ba6d3d1SJason Molenda     }
2826ba6d3d1SJason Molenda 
2836ba6d3d1SJason Molenda     ThreadSP thread = process->GetThreadList().GetSelectedThread ();
2846ba6d3d1SJason Molenda     if (thread.get() == NULL)
2856ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2866ba6d3d1SJason Molenda     addr_t pc = thread->GetRegisterContext ()->GetPC(LLDB_INVALID_ADDRESS);
2876ba6d3d1SJason Molenda 
2886ba6d3d1SJason Molenda     if (pc == LLDB_INVALID_ADDRESS)
2896ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
2906ba6d3d1SJason Molenda 
2916ba6d3d1SJason Molenda     addr_t kernel_range_low, kernel_range_high;
2926ba6d3d1SJason Molenda     if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
2936ba6d3d1SJason Molenda     {
2946ba6d3d1SJason Molenda         kernel_range_low = 1ULL << 63;
2956ba6d3d1SJason Molenda         kernel_range_high = UINT64_MAX;
2966ba6d3d1SJason Molenda     }
2976ba6d3d1SJason Molenda     else
2986ba6d3d1SJason Molenda     {
2996ba6d3d1SJason Molenda         kernel_range_low = 1ULL << 31;
3006ba6d3d1SJason Molenda         kernel_range_high = UINT32_MAX;
3016ba6d3d1SJason Molenda     }
3026ba6d3d1SJason Molenda 
3036ba6d3d1SJason Molenda     // Outside the normal kernel address range, this is probably userland code running right now
3046ba6d3d1SJason Molenda     if (pc < kernel_range_low)
3056ba6d3d1SJason Molenda         LLDB_INVALID_ADDRESS;
3066ba6d3d1SJason Molenda 
3076ba6d3d1SJason Molenda     // The kernel will load at at one megabyte boundary (0x100000), or at that boundary plus
3086ba6d3d1SJason Molenda     // an offset of one page (0x1000) or two, depending on the device.
3096ba6d3d1SJason Molenda 
3106ba6d3d1SJason Molenda     // Round the current pc down to the nearest one megabyte boundary - the place where we will start searching.
3116ba6d3d1SJason Molenda     addr_t addr = pc & ~0xfffff;
3126ba6d3d1SJason Molenda 
3136ba6d3d1SJason Molenda     int i = 0;
3146ba6d3d1SJason Molenda     while (i < 32 && pc >= kernel_range_low)
3156ba6d3d1SJason Molenda     {
3166ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr, process).IsValid())
3176ba6d3d1SJason Molenda             return addr;
3186ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr + 0x1000, process).IsValid())
3196ba6d3d1SJason Molenda             return addr + 0x1000;
3206ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr + 0x2000, process).IsValid())
3216ba6d3d1SJason Molenda             return addr + 0x2000;
3226ba6d3d1SJason Molenda         i++;
3236ba6d3d1SJason Molenda         addr -= 0x100000;
3246ba6d3d1SJason Molenda     }
3256ba6d3d1SJason Molenda 
3266ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3276ba6d3d1SJason Molenda }
3286ba6d3d1SJason Molenda 
3296ba6d3d1SJason Molenda //----------------------------------------------------------------------
3306ba6d3d1SJason Molenda // Scan through the valid address range for a kernel binary.
3316ba6d3d1SJason Molenda // This is uselessly slow in 64-bit environments so we don't even try it.
3326ba6d3d1SJason Molenda // This scan is not enabled by default even for 32-bit targets.
3336ba6d3d1SJason Molenda // Returns the address of the kernel if one was found, else LLDB_INVALID_ADDRESS.
3346ba6d3d1SJason Molenda //----------------------------------------------------------------------
3356ba6d3d1SJason Molenda lldb::addr_t
3366ba6d3d1SJason Molenda DynamicLoaderDarwinKernel::SearchForKernelViaExhaustiveSearch (Process *process)
3376ba6d3d1SJason Molenda {
3386ba6d3d1SJason Molenda     if (GetGlobalProperties()->GetScanType() != eKASLRScanExhaustiveScan)
3396ba6d3d1SJason Molenda     {
3406ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
3416ba6d3d1SJason Molenda     }
3426ba6d3d1SJason Molenda 
3436ba6d3d1SJason Molenda     addr_t kernel_range_low, kernel_range_high;
3446ba6d3d1SJason Molenda     if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
3456ba6d3d1SJason Molenda     {
3466ba6d3d1SJason Molenda         kernel_range_low = 1ULL << 63;
3476ba6d3d1SJason Molenda         kernel_range_high = UINT64_MAX;
3486ba6d3d1SJason Molenda     }
3496ba6d3d1SJason Molenda     else
3506ba6d3d1SJason Molenda     {
3516ba6d3d1SJason Molenda         kernel_range_low = 1ULL << 31;
3526ba6d3d1SJason Molenda         kernel_range_high = UINT32_MAX;
3536ba6d3d1SJason Molenda     }
3546ba6d3d1SJason Molenda 
3556ba6d3d1SJason Molenda     // Stepping through memory at one-megabyte resolution looking for a kernel
3566ba6d3d1SJason Molenda     // rarely works (fast enough) with a 64-bit address space -- for now, let's
3576ba6d3d1SJason Molenda     // not even bother.  We may be attaching to something which *isn't* a kernel
3586ba6d3d1SJason Molenda     // and we don't want to spin for minutes on-end looking for a kernel.
3596ba6d3d1SJason Molenda     if (process->GetTarget().GetArchitecture().GetAddressByteSize() == 8)
3606ba6d3d1SJason Molenda         return LLDB_INVALID_ADDRESS;
3616ba6d3d1SJason Molenda 
3626ba6d3d1SJason Molenda     addr_t addr = kernel_range_low;
3636ba6d3d1SJason Molenda 
3646ba6d3d1SJason Molenda     while (addr >= kernel_range_low && addr < kernel_range_high)
3656ba6d3d1SJason Molenda     {
3666ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr, process).IsValid())
3676ba6d3d1SJason Molenda             return addr;
3686ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr + 0x1000, process).IsValid())
3696ba6d3d1SJason Molenda             return addr + 0x1000;
3706ba6d3d1SJason Molenda         if (CheckForKernelImageAtAddress (addr + 0x2000, process).IsValid())
3716ba6d3d1SJason Molenda             return addr + 0x2000;
3726ba6d3d1SJason Molenda         addr += 0x100000;
3736ba6d3d1SJason Molenda     }
3746ba6d3d1SJason Molenda     return LLDB_INVALID_ADDRESS;
3756ba6d3d1SJason Molenda }
3766ba6d3d1SJason Molenda 
3776ba6d3d1SJason Molenda //----------------------------------------------------------------------
3786ba6d3d1SJason Molenda // Given an address in memory, look to see if there is a kernel image at that
3796ba6d3d1SJason Molenda // address.
3806ba6d3d1SJason Molenda // Returns a UUID; if a kernel was not found at that address, UUID.IsValid() will be false.
3816ba6d3d1SJason Molenda //----------------------------------------------------------------------
3826ba6d3d1SJason Molenda lldb_private::UUID
3836ba6d3d1SJason Molenda DynamicLoaderDarwinKernel::CheckForKernelImageAtAddress (lldb::addr_t addr, Process *process)
3846ba6d3d1SJason Molenda {
3856ba6d3d1SJason Molenda     if (addr == LLDB_INVALID_ADDRESS)
3866ba6d3d1SJason Molenda         return UUID();
3876ba6d3d1SJason Molenda 
3886ba6d3d1SJason Molenda     // First try a quick test -- read the first 4 bytes and see if there is a valid Mach-O magic field there
3896ba6d3d1SJason Molenda     // (the first field of the mach_header/mach_header_64 struct).
3906ba6d3d1SJason Molenda 
3916ba6d3d1SJason Molenda     Error read_error;
3926ba6d3d1SJason Molenda     uint64_t result = process->ReadUnsignedIntegerFromMemory (addr, 4, LLDB_INVALID_ADDRESS, read_error);
3936ba6d3d1SJason Molenda     if (result != llvm::MachO::HeaderMagic64
3946ba6d3d1SJason Molenda         && result != llvm::MachO::HeaderMagic32
3956ba6d3d1SJason Molenda         && result != llvm::MachO::HeaderMagic32Swapped
3966ba6d3d1SJason Molenda         && result != llvm::MachO::HeaderMagic64Swapped)
3976ba6d3d1SJason Molenda     {
3986ba6d3d1SJason Molenda         return UUID();
3996ba6d3d1SJason Molenda     }
4006ba6d3d1SJason Molenda 
4016ba6d3d1SJason Molenda     // Read the mach header and see whether it looks like a kernel
4026ba6d3d1SJason Molenda     llvm::MachO::mach_header header;
4036ba6d3d1SJason Molenda     if (process->DoReadMemory (addr, &header, sizeof(header), read_error) != sizeof(header))
4046ba6d3d1SJason Molenda         return UUID();
4056ba6d3d1SJason Molenda 
4066ba6d3d1SJason Molenda     if (header.magic == llvm::MachO::HeaderMagic32Swapped ||
4076ba6d3d1SJason Molenda         header.magic == llvm::MachO::HeaderMagic64Swapped)
4086ba6d3d1SJason Molenda     {
4096ba6d3d1SJason Molenda         header.magic        = llvm::ByteSwap_32(header.magic);
4106ba6d3d1SJason Molenda         header.cputype      = llvm::ByteSwap_32(header.cputype);
4116ba6d3d1SJason Molenda         header.cpusubtype   = llvm::ByteSwap_32(header.cpusubtype);
4126ba6d3d1SJason Molenda         header.filetype     = llvm::ByteSwap_32(header.filetype);
4136ba6d3d1SJason Molenda         header.ncmds        = llvm::ByteSwap_32(header.ncmds);
4146ba6d3d1SJason Molenda         header.sizeofcmds   = llvm::ByteSwap_32(header.sizeofcmds);
4156ba6d3d1SJason Molenda         header.flags        = llvm::ByteSwap_32(header.flags);
4166ba6d3d1SJason Molenda     }
4176ba6d3d1SJason Molenda 
4186ba6d3d1SJason Molenda     // A kernel is an executable which does not have the dynamic link object flag set.
4196ba6d3d1SJason Molenda     if (header.filetype == llvm::MachO::HeaderFileTypeExecutable
4206ba6d3d1SJason Molenda         && (header.flags & llvm::MachO::HeaderFlagBitIsDynamicLinkObject) == 0)
4216ba6d3d1SJason Molenda     {
4226ba6d3d1SJason Molenda         // Create a full module to get the UUID
42339f7ee86SGreg Clayton         ModuleSP memory_module_sp = process->ReadModuleFromMemory (FileSpec ("temp_mach_kernel", false), addr);
4246ba6d3d1SJason Molenda         if (!memory_module_sp.get())
4256ba6d3d1SJason Molenda             return UUID();
4266ba6d3d1SJason Molenda 
4276ba6d3d1SJason Molenda         ObjectFile *exe_objfile = memory_module_sp->GetObjectFile();
4286ba6d3d1SJason Molenda         if (exe_objfile == NULL)
4296ba6d3d1SJason Molenda             return UUID();
4306ba6d3d1SJason Molenda 
4316ba6d3d1SJason Molenda         if (exe_objfile->GetType() == ObjectFile::eTypeExecutable && exe_objfile->GetStrata() == ObjectFile::eStrataKernel)
4326ba6d3d1SJason Molenda         {
4336ba6d3d1SJason Molenda             return memory_module_sp->GetUUID();
4346ba6d3d1SJason Molenda         }
4356ba6d3d1SJason Molenda     }
4366ba6d3d1SJason Molenda 
4376ba6d3d1SJason Molenda     return UUID();
4386ba6d3d1SJason Molenda }
4396ba6d3d1SJason Molenda 
4406ba6d3d1SJason Molenda //----------------------------------------------------------------------
441d4bfbc9aSGreg Clayton // Constructor
442d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
4436ba6d3d1SJason Molenda DynamicLoaderDarwinKernel::DynamicLoaderDarwinKernel (Process* process, lldb::addr_t kernel_addr) :
444d4bfbc9aSGreg Clayton     DynamicLoader(process),
4456ba6d3d1SJason Molenda     m_kernel_load_address (kernel_addr),
446d4bfbc9aSGreg Clayton     m_kernel(),
447d4bfbc9aSGreg Clayton     m_kext_summary_header_ptr_addr (),
448d4bfbc9aSGreg Clayton     m_kext_summary_header_addr (),
449d4bfbc9aSGreg Clayton     m_kext_summary_header (),
450*306bd0aaSJason Molenda     m_known_kexts (),
451a08823fdSDaniel Dunbar     m_mutex(Mutex::eMutexTypeRecursive),
452a08823fdSDaniel Dunbar     m_break_id (LLDB_INVALID_BREAK_ID)
453d4bfbc9aSGreg Clayton {
454d4bfbc9aSGreg Clayton }
455d4bfbc9aSGreg Clayton 
456d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
457d4bfbc9aSGreg Clayton // Destructor
458d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
459944b828aSGreg Clayton DynamicLoaderDarwinKernel::~DynamicLoaderDarwinKernel()
460d4bfbc9aSGreg Clayton {
461d4bfbc9aSGreg Clayton     Clear(true);
462d4bfbc9aSGreg Clayton }
463d4bfbc9aSGreg Clayton 
464d4bfbc9aSGreg Clayton void
465944b828aSGreg Clayton DynamicLoaderDarwinKernel::UpdateIfNeeded()
466d4bfbc9aSGreg Clayton {
467d4bfbc9aSGreg Clayton     LoadKernelModuleIfNeeded();
468d4bfbc9aSGreg Clayton     SetNotificationBreakpointIfNeeded ();
469d4bfbc9aSGreg Clayton }
470d4bfbc9aSGreg Clayton //------------------------------------------------------------------
471d4bfbc9aSGreg Clayton /// Called after attaching a process.
472d4bfbc9aSGreg Clayton ///
473d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
474d4bfbc9aSGreg Clayton /// attaching to a process.
475d4bfbc9aSGreg Clayton //------------------------------------------------------------------
476d4bfbc9aSGreg Clayton void
477944b828aSGreg Clayton DynamicLoaderDarwinKernel::DidAttach ()
478d4bfbc9aSGreg Clayton {
479d4bfbc9aSGreg Clayton     PrivateInitialize(m_process);
480d4bfbc9aSGreg Clayton     UpdateIfNeeded();
481d4bfbc9aSGreg Clayton }
482d4bfbc9aSGreg Clayton 
483d4bfbc9aSGreg Clayton //------------------------------------------------------------------
484d4bfbc9aSGreg Clayton /// Called after attaching a process.
485d4bfbc9aSGreg Clayton ///
486d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
487d4bfbc9aSGreg Clayton /// attaching to a process.
488d4bfbc9aSGreg Clayton //------------------------------------------------------------------
489d4bfbc9aSGreg Clayton void
490944b828aSGreg Clayton DynamicLoaderDarwinKernel::DidLaunch ()
491d4bfbc9aSGreg Clayton {
492d4bfbc9aSGreg Clayton     PrivateInitialize(m_process);
493d4bfbc9aSGreg Clayton     UpdateIfNeeded();
494d4bfbc9aSGreg Clayton }
495d4bfbc9aSGreg Clayton 
496d4bfbc9aSGreg Clayton 
497d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
498d4bfbc9aSGreg Clayton // Clear out the state of this class.
499d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
500d4bfbc9aSGreg Clayton void
501944b828aSGreg Clayton DynamicLoaderDarwinKernel::Clear (bool clear_process)
502d4bfbc9aSGreg Clayton {
503d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
504d4bfbc9aSGreg Clayton 
505d4bfbc9aSGreg Clayton     if (m_process->IsAlive() && LLDB_BREAK_ID_IS_VALID(m_break_id))
506d4bfbc9aSGreg Clayton         m_process->ClearBreakpointSiteByID(m_break_id);
507d4bfbc9aSGreg Clayton 
508d4bfbc9aSGreg Clayton     if (clear_process)
509d4bfbc9aSGreg Clayton         m_process = NULL;
510*306bd0aaSJason Molenda     m_kernel.Clear();
511*306bd0aaSJason Molenda     m_known_kexts.clear();
512d4bfbc9aSGreg Clayton     m_kext_summary_header_ptr_addr.Clear();
513d4bfbc9aSGreg Clayton     m_kext_summary_header_addr.Clear();
514d4bfbc9aSGreg Clayton     m_break_id = LLDB_INVALID_BREAK_ID;
515d4bfbc9aSGreg Clayton }
516d4bfbc9aSGreg Clayton 
517d4bfbc9aSGreg Clayton 
518c859e2d5SGreg Clayton bool
519*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::LoadImageAtFileAddress (Process *process)
5202af282a1SGreg Clayton {
5212af282a1SGreg Clayton     if (IsLoaded())
5222af282a1SGreg Clayton         return true;
5232af282a1SGreg Clayton 
524*306bd0aaSJason Molenda     if (m_module_sp)
5252af282a1SGreg Clayton     {
5262af282a1SGreg Clayton         bool changed = false;
527*306bd0aaSJason Molenda         if (m_module_sp->SetLoadAddress (process->GetTarget(), 0, changed))
528*306bd0aaSJason Molenda             m_load_process_stop_id = process->GetStopID();
5292af282a1SGreg Clayton     }
5302af282a1SGreg Clayton     return false;
5312af282a1SGreg Clayton }
5322af282a1SGreg Clayton 
533*306bd0aaSJason Molenda void
534*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetModule (ModuleSP module_sp)
535*306bd0aaSJason Molenda {
536*306bd0aaSJason Molenda     m_module_sp = module_sp;
537*306bd0aaSJason Molenda     if (module_sp.get() && module_sp->GetObjectFile())
538*306bd0aaSJason Molenda     {
539*306bd0aaSJason Molenda         if (module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeExecutable
540*306bd0aaSJason Molenda             && module_sp->GetObjectFile()->GetStrata() == ObjectFile::eStrataKernel)
541*306bd0aaSJason Molenda         {
542*306bd0aaSJason Molenda             m_kernel_image = true;
543*306bd0aaSJason Molenda         }
544*306bd0aaSJason Molenda         else
545*306bd0aaSJason Molenda         {
546*306bd0aaSJason Molenda             m_kernel_image = false;
547*306bd0aaSJason Molenda         }
548*306bd0aaSJason Molenda     }
549*306bd0aaSJason Molenda }
550*306bd0aaSJason Molenda 
551*306bd0aaSJason Molenda ModuleSP
552*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetModule ()
553*306bd0aaSJason Molenda {
554*306bd0aaSJason Molenda     return m_module_sp;
555*306bd0aaSJason Molenda }
556*306bd0aaSJason Molenda 
557*306bd0aaSJason Molenda void
558*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetLoadAddress (addr_t load_addr)
559*306bd0aaSJason Molenda {
560*306bd0aaSJason Molenda     m_load_address = load_addr;
561*306bd0aaSJason Molenda }
562*306bd0aaSJason Molenda 
563*306bd0aaSJason Molenda addr_t
564*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetLoadAddress () const
565*306bd0aaSJason Molenda {
566*306bd0aaSJason Molenda     return m_load_address;
567*306bd0aaSJason Molenda }
568*306bd0aaSJason Molenda 
569*306bd0aaSJason Molenda uint64_t
570*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetSize () const
571*306bd0aaSJason Molenda {
572*306bd0aaSJason Molenda     return m_size;
573*306bd0aaSJason Molenda }
574*306bd0aaSJason Molenda 
575*306bd0aaSJason Molenda void
576*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetSize (uint64_t size)
577*306bd0aaSJason Molenda {
578*306bd0aaSJason Molenda     m_size = size;
579*306bd0aaSJason Molenda }
580*306bd0aaSJason Molenda 
581*306bd0aaSJason Molenda uint32_t
582*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetProcessStopId () const
583*306bd0aaSJason Molenda {
584*306bd0aaSJason Molenda     return m_load_process_stop_id;
585*306bd0aaSJason Molenda }
586*306bd0aaSJason Molenda 
587*306bd0aaSJason Molenda void
588*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetProcessStopId (uint32_t stop_id)
589*306bd0aaSJason Molenda {
590*306bd0aaSJason Molenda     m_load_process_stop_id = stop_id;
591*306bd0aaSJason Molenda }
592*306bd0aaSJason Molenda 
5932af282a1SGreg Clayton bool
594*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::operator== (const KextImageInfo &rhs)
595*306bd0aaSJason Molenda {
596*306bd0aaSJason Molenda     if (m_uuid.IsValid() || rhs.GetUUID().IsValid())
597*306bd0aaSJason Molenda     {
598*306bd0aaSJason Molenda         if (m_uuid == rhs.GetUUID())
599*306bd0aaSJason Molenda         {
600*306bd0aaSJason Molenda             return true;
601*306bd0aaSJason Molenda         }
602*306bd0aaSJason Molenda         return false;
603*306bd0aaSJason Molenda     }
604*306bd0aaSJason Molenda 
605*306bd0aaSJason Molenda     if (m_name == rhs.GetName() && m_load_address == rhs.GetLoadAddress())
606*306bd0aaSJason Molenda         return true;
607*306bd0aaSJason Molenda 
608*306bd0aaSJason Molenda     return false;
609*306bd0aaSJason Molenda }
610*306bd0aaSJason Molenda 
611*306bd0aaSJason Molenda void
612*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetName (const char *name)
613*306bd0aaSJason Molenda {
614*306bd0aaSJason Molenda     m_name = name;
615*306bd0aaSJason Molenda }
616*306bd0aaSJason Molenda 
617*306bd0aaSJason Molenda std::string
618*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetName () const
619*306bd0aaSJason Molenda {
620*306bd0aaSJason Molenda     return m_name;
621*306bd0aaSJason Molenda }
622*306bd0aaSJason Molenda 
623*306bd0aaSJason Molenda void
624*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetUUID (const UUID &uuid)
625*306bd0aaSJason Molenda {
626*306bd0aaSJason Molenda     m_uuid = uuid;
627*306bd0aaSJason Molenda }
628*306bd0aaSJason Molenda 
629*306bd0aaSJason Molenda UUID
630*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetUUID () const
631*306bd0aaSJason Molenda {
632*306bd0aaSJason Molenda     return m_uuid;
633*306bd0aaSJason Molenda }
634*306bd0aaSJason Molenda 
635*306bd0aaSJason Molenda // Given the m_load_address from the kext summaries, and a UUID, try to create an in-memory
636*306bd0aaSJason Molenda // Module at that address.  Require that the MemoryModule have a matching UUID and detect
637*306bd0aaSJason Molenda // if this MemoryModule is a kernel or a kext.
638*306bd0aaSJason Molenda //
639*306bd0aaSJason Molenda // Returns true if m_memory_module_sp is now set to a valid Module.
640*306bd0aaSJason Molenda 
641*306bd0aaSJason Molenda bool
642*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::ReadMemoryModule (Process *process)
643*306bd0aaSJason Molenda {
644*306bd0aaSJason Molenda     if (m_memory_module_sp.get() != NULL)
645*306bd0aaSJason Molenda         return true;
646*306bd0aaSJason Molenda     if (m_load_address == LLDB_INVALID_ADDRESS)
647*306bd0aaSJason Molenda         return false;
648*306bd0aaSJason Molenda 
649*306bd0aaSJason Molenda     FileSpec file_spec;
650*306bd0aaSJason Molenda     file_spec.SetFile (m_name.c_str(), false);
651*306bd0aaSJason Molenda 
652*306bd0aaSJason Molenda     ModuleSP memory_module_sp = process->ReadModuleFromMemory (file_spec, m_load_address);
653*306bd0aaSJason Molenda 
654*306bd0aaSJason Molenda     if (memory_module_sp.get() == NULL)
655*306bd0aaSJason Molenda         return false;
656*306bd0aaSJason Molenda 
657*306bd0aaSJason Molenda     bool is_kernel = false;
658*306bd0aaSJason Molenda     if (memory_module_sp->GetObjectFile())
659*306bd0aaSJason Molenda     {
660*306bd0aaSJason Molenda         if (memory_module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeExecutable
661*306bd0aaSJason Molenda             && memory_module_sp->GetObjectFile()->GetStrata() == ObjectFile::eStrataKernel)
662*306bd0aaSJason Molenda         {
663*306bd0aaSJason Molenda             is_kernel = true;
664*306bd0aaSJason Molenda         }
665*306bd0aaSJason Molenda         else if (memory_module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeSharedLibrary)
666*306bd0aaSJason Molenda         {
667*306bd0aaSJason Molenda             is_kernel = false;
668*306bd0aaSJason Molenda         }
669*306bd0aaSJason Molenda     }
670*306bd0aaSJason Molenda 
671*306bd0aaSJason Molenda     // If the kernel specified what UUID we should find at this load address,
672*306bd0aaSJason Molenda     // require that the memory module have a matching UUID or something has gone
673*306bd0aaSJason Molenda     // wrong and we should discard it.
674*306bd0aaSJason Molenda     if (m_uuid.IsValid())
675*306bd0aaSJason Molenda     {
676*306bd0aaSJason Molenda         if (m_uuid != memory_module_sp->GetUUID())
677*306bd0aaSJason Molenda         {
678*306bd0aaSJason Molenda             return false;
679*306bd0aaSJason Molenda         }
680*306bd0aaSJason Molenda     }
681*306bd0aaSJason Molenda 
682*306bd0aaSJason Molenda     // If the in-memory Module has a UUID, let's use that.
683*306bd0aaSJason Molenda     if (!m_uuid.IsValid() && memory_module_sp->GetUUID().IsValid())
684*306bd0aaSJason Molenda     {
685*306bd0aaSJason Molenda         m_uuid = memory_module_sp->GetUUID();
686*306bd0aaSJason Molenda     }
687*306bd0aaSJason Molenda 
688*306bd0aaSJason Molenda     m_memory_module_sp = memory_module_sp;
689*306bd0aaSJason Molenda     m_kernel_image = is_kernel;
690*306bd0aaSJason Molenda     if (is_kernel)
691*306bd0aaSJason Molenda     {
692*306bd0aaSJason Molenda        if (memory_module_sp->GetArchitecture().IsValid())
693*306bd0aaSJason Molenda         {
694*306bd0aaSJason Molenda             process->GetTarget().SetArchitecture(memory_module_sp->GetArchitecture());
695*306bd0aaSJason Molenda         }
696*306bd0aaSJason Molenda     }
697*306bd0aaSJason Molenda 
698*306bd0aaSJason Molenda     return true;
699*306bd0aaSJason Molenda }
700*306bd0aaSJason Molenda 
701*306bd0aaSJason Molenda bool
702*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::IsKernel () const
703*306bd0aaSJason Molenda {
704*306bd0aaSJason Molenda     return m_kernel_image == true;
705*306bd0aaSJason Molenda }
706*306bd0aaSJason Molenda 
707*306bd0aaSJason Molenda void
708*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::SetIsKernel (bool is_kernel)
709*306bd0aaSJason Molenda {
710*306bd0aaSJason Molenda     m_kernel_image = is_kernel;
711*306bd0aaSJason Molenda }
712*306bd0aaSJason Molenda 
713*306bd0aaSJason Molenda bool
714*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::LoadImageUsingMemoryModule (Process *process)
715c859e2d5SGreg Clayton {
716c859e2d5SGreg Clayton     if (IsLoaded())
717c859e2d5SGreg Clayton         return true;
718c859e2d5SGreg Clayton 
719c859e2d5SGreg Clayton 
720c859e2d5SGreg Clayton     Target &target = process->GetTarget();
72187a04b24SJason Molenda 
722*306bd0aaSJason Molenda     // If we don't have / can't create a memory module for this kext, don't try to load it - we won't
723*306bd0aaSJason Molenda     // have the correct segment load addresses.
724*306bd0aaSJason Molenda     if (!ReadMemoryModule (process))
72587a04b24SJason Molenda     {
72687a04b24SJason Molenda         return false;
72787a04b24SJason Molenda     }
72887a04b24SJason Molenda 
729*306bd0aaSJason Molenda     // If this is a kext and the user asked us to ignore kexts, don't try to load it.
730*306bd0aaSJason Molenda     if (!IsKernel() && GetGlobalProperties()->GetLoadKexts() == false)
731c859e2d5SGreg Clayton     {
732*306bd0aaSJason Molenda         return false;
733c859e2d5SGreg Clayton     }
734c859e2d5SGreg Clayton 
735*306bd0aaSJason Molenda     bool uuid_is_valid = m_uuid.IsValid();
736*306bd0aaSJason Molenda 
737*306bd0aaSJason Molenda     if (!m_module_sp)
738c859e2d5SGreg Clayton     {
739*306bd0aaSJason Molenda         // See if the kext has already been loaded into the target, probably by the user doing target modules add.
7401759848bSEnrico Granata         const ModuleList &target_images = target.GetImages();
741*306bd0aaSJason Molenda         m_module_sp = target_images.FindModule(m_uuid);
742c859e2d5SGreg Clayton 
743*306bd0aaSJason Molenda         // Search for the kext on the local filesystem via the UUID
744*306bd0aaSJason Molenda         if (!m_module_sp && uuid_is_valid)
745b9a01b39SGreg Clayton         {
7462af282a1SGreg Clayton             ModuleSpec module_spec;
747*306bd0aaSJason Molenda             module_spec.GetUUID() = m_uuid;
74853667f5dSJason Molenda             module_spec.GetArchitecture() = target.GetArchitecture();
749bb860bd2SJason Molenda 
750*306bd0aaSJason Molenda             // For the kernel, we really do need an on-disk file copy of the binary.
751bb860bd2SJason Molenda             bool force_symbols_search = false;
752*306bd0aaSJason Molenda             if (IsKernel())
753bb860bd2SJason Molenda             {
754bb860bd2SJason Molenda                 force_symbols_search = true;
755bb860bd2SJason Molenda             }
756bb860bd2SJason Molenda 
757bb860bd2SJason Molenda             if (Symbols::DownloadObjectAndSymbolFile (module_spec, force_symbols_search))
758bb860bd2SJason Molenda             {
759bb860bd2SJason Molenda                 if (module_spec.GetFileSpec().Exists())
760bb860bd2SJason Molenda                 {
761*306bd0aaSJason Molenda                     m_module_sp.reset(new Module (module_spec.GetFileSpec(), target.GetArchitecture()));
762*306bd0aaSJason Molenda                     if (m_module_sp.get() && m_module_sp->MatchesModuleSpec (module_spec))
763bb860bd2SJason Molenda                     {
764bb860bd2SJason Molenda                         ModuleList loaded_module_list;
765*306bd0aaSJason Molenda                         loaded_module_list.Append (m_module_sp);
766bb860bd2SJason Molenda                         target.ModulesDidLoad (loaded_module_list);
767bb860bd2SJason Molenda                     }
768bb860bd2SJason Molenda                 }
769bb860bd2SJason Molenda             }
770bb860bd2SJason Molenda 
771*306bd0aaSJason Molenda             // Failing that, ask the Target to find this file on the local system, if possible.
772bb860bd2SJason Molenda             // This will search in the list of currently-loaded files, look in the
773bb860bd2SJason Molenda             // standard search paths on the system, and on a Mac it will try calling
774bb860bd2SJason Molenda             // the DebugSymbols framework with the UUID to find the binary via its
775bb860bd2SJason Molenda             // search methods.
776*306bd0aaSJason Molenda             if (!m_module_sp)
777bb860bd2SJason Molenda             {
778*306bd0aaSJason Molenda                 m_module_sp = target.GetSharedModule (module_spec);
779*306bd0aaSJason Molenda             }
780b9a01b39SGreg Clayton         }
78165d57a3dSJason Molenda 
782*306bd0aaSJason Molenda         // If we managed to find a module, append it to the target's list of images.
783*306bd0aaSJason Molenda         // If we also have a memory module, require that they have matching UUIDs
784*306bd0aaSJason Molenda         if (m_module_sp)
78565d57a3dSJason Molenda         {
786*306bd0aaSJason Molenda             bool uuid_match_ok = true;
787*306bd0aaSJason Molenda             if (m_memory_module_sp)
78865d57a3dSJason Molenda             {
789*306bd0aaSJason Molenda                 if (m_module_sp->GetUUID() != m_memory_module_sp->GetUUID())
790*306bd0aaSJason Molenda                 {
791*306bd0aaSJason Molenda                     uuid_match_ok = false;
79265d57a3dSJason Molenda                 }
79365d57a3dSJason Molenda             }
794*306bd0aaSJason Molenda             if (uuid_match_ok)
795*306bd0aaSJason Molenda             {
796*306bd0aaSJason Molenda                 target.GetImages().Append(m_module_sp);
797*306bd0aaSJason Molenda                 if (IsKernel() && target.GetExecutableModulePointer() != m_module_sp.get())
798*306bd0aaSJason Molenda                 {
799*306bd0aaSJason Molenda                     target.SetExecutableModule (m_module_sp, false);
800*306bd0aaSJason Molenda                 }
801c859e2d5SGreg Clayton             }
802c859e2d5SGreg Clayton         }
803bb860bd2SJason Molenda     }
804c859e2d5SGreg Clayton 
805c859e2d5SGreg Clayton 
80667408537SGreg Clayton     static ConstString g_section_name_LINKEDIT ("__LINKEDIT");
80767408537SGreg Clayton 
808*306bd0aaSJason Molenda     if (m_memory_module_sp && m_module_sp)
809c859e2d5SGreg Clayton     {
810*306bd0aaSJason Molenda         if (m_module_sp->GetUUID() == m_memory_module_sp->GetUUID())
811c859e2d5SGreg Clayton         {
812*306bd0aaSJason Molenda             ObjectFile *ondisk_object_file = m_module_sp->GetObjectFile();
813*306bd0aaSJason Molenda             ObjectFile *memory_object_file = m_memory_module_sp->GetObjectFile();
81467408537SGreg Clayton 
815c859e2d5SGreg Clayton             if (memory_object_file && ondisk_object_file)
816c859e2d5SGreg Clayton             {
817*306bd0aaSJason Molenda                 // The memory_module for kexts may have an invalid __LINKEDIT seg; skip it.
818*306bd0aaSJason Molenda                 const bool ignore_linkedit = !IsKernel ();
81967408537SGreg Clayton 
820c859e2d5SGreg Clayton                 SectionList *ondisk_section_list = ondisk_object_file->GetSectionList ();
821c859e2d5SGreg Clayton                 SectionList *memory_section_list = memory_object_file->GetSectionList ();
822c859e2d5SGreg Clayton                 if (memory_section_list && ondisk_section_list)
823c859e2d5SGreg Clayton                 {
824bae2f2f8SGreg Clayton                     const uint32_t num_ondisk_sections = ondisk_section_list->GetSize();
825c859e2d5SGreg Clayton                     // There may be CTF sections in the memory image so we can't
826c859e2d5SGreg Clayton                     // always just compare the number of sections (which are actually
827c859e2d5SGreg Clayton                     // segments in mach-o parlance)
828c859e2d5SGreg Clayton                     uint32_t sect_idx = 0;
829bae2f2f8SGreg Clayton 
83053667f5dSJason Molenda                     // Use the memory_module's addresses for each section to set the
83153667f5dSJason Molenda                     // file module's load address as appropriate.  We don't want to use
83253667f5dSJason Molenda                     // a single slide value for the entire kext - different segments may
83353667f5dSJason Molenda                     // be slid different amounts by the kext loader.
834bae2f2f8SGreg Clayton 
835bae2f2f8SGreg Clayton                     uint32_t num_sections_loaded = 0;
836bae2f2f8SGreg Clayton                     for (sect_idx=0; sect_idx<num_ondisk_sections; ++sect_idx)
837c859e2d5SGreg Clayton                     {
8387820bd1eSGreg Clayton                         SectionSP ondisk_section_sp(ondisk_section_list->GetSectionAtIndex(sect_idx));
8397820bd1eSGreg Clayton                         if (ondisk_section_sp)
840c859e2d5SGreg Clayton                         {
84167408537SGreg Clayton                             // Don't ever load __LINKEDIT as it may or may not be actually
84267408537SGreg Clayton                             // mapped into memory and there is no current way to tell.
84367408537SGreg Clayton                             // I filed rdar://problem/12851706 to track being able to tell
84467408537SGreg Clayton                             // if the __LINKEDIT is actually mapped, but until then, we need
84567408537SGreg Clayton                             // to not load the __LINKEDIT
84667408537SGreg Clayton                             if (ignore_linkedit && ondisk_section_sp->GetName() == g_section_name_LINKEDIT)
84767408537SGreg Clayton                                 continue;
84867408537SGreg Clayton 
8497820bd1eSGreg Clayton                             const Section *memory_section = memory_section_list->FindSectionByName(ondisk_section_sp->GetName()).get();
850bae2f2f8SGreg Clayton                             if (memory_section)
851c859e2d5SGreg Clayton                             {
8527820bd1eSGreg Clayton                                 target.GetSectionLoadList().SetSectionLoadAddress (ondisk_section_sp, memory_section->GetFileAddress());
853bae2f2f8SGreg Clayton                                 ++num_sections_loaded;
854c859e2d5SGreg Clayton                             }
855bae2f2f8SGreg Clayton                         }
856bae2f2f8SGreg Clayton                     }
857bae2f2f8SGreg Clayton                     if (num_sections_loaded > 0)
858*306bd0aaSJason Molenda                         m_load_process_stop_id = process->GetStopID();
859bae2f2f8SGreg Clayton                     else
860*306bd0aaSJason Molenda                         m_module_sp.reset(); // No sections were loaded
861c859e2d5SGreg Clayton                 }
862c859e2d5SGreg Clayton                 else
863*306bd0aaSJason Molenda                     m_module_sp.reset(); // One or both section lists
864c859e2d5SGreg Clayton             }
865c859e2d5SGreg Clayton             else
866*306bd0aaSJason Molenda                 m_module_sp.reset(); // One or both object files missing
867c859e2d5SGreg Clayton         }
868c859e2d5SGreg Clayton         else
869*306bd0aaSJason Molenda             m_module_sp.reset(); // UUID mismatch
870c859e2d5SGreg Clayton     }
871c859e2d5SGreg Clayton 
872c859e2d5SGreg Clayton     bool is_loaded = IsLoaded();
873c859e2d5SGreg Clayton 
874*306bd0aaSJason Molenda     if (is_loaded && m_module_sp && IsKernel())
87568b3607fSJason Molenda     {
87668b3607fSJason Molenda         Stream *s = &target.GetDebugger().GetOutputStream();
87768b3607fSJason Molenda         if (s)
87868b3607fSJason Molenda         {
87968b3607fSJason Molenda             char uuidbuf[64];
880*306bd0aaSJason Molenda             s->Printf ("Kernel UUID: %s\n", m_module_sp->GetUUID().GetAsCString(uuidbuf, sizeof (uuidbuf)));
881*306bd0aaSJason Molenda             s->Printf ("Load Address: 0x%" PRIx64 "\n", m_load_address);
882*306bd0aaSJason Molenda             if (m_module_sp->GetFileSpec().GetDirectory().IsEmpty())
883743e4396SJason Molenda             {
884*306bd0aaSJason Molenda                 s->Printf ("Loaded kernel file %s\n", m_module_sp->GetFileSpec().GetFilename().AsCString());
885743e4396SJason Molenda             }
886743e4396SJason Molenda             else
887743e4396SJason Molenda             {
88868b3607fSJason Molenda                 s->Printf ("Loaded kernel file %s/%s\n",
889*306bd0aaSJason Molenda                               m_module_sp->GetFileSpec().GetDirectory().AsCString(),
890*306bd0aaSJason Molenda                               m_module_sp->GetFileSpec().GetFilename().AsCString());
891743e4396SJason Molenda             }
89268b3607fSJason Molenda             s->Flush ();
89368b3607fSJason Molenda         }
89468b3607fSJason Molenda     }
895c859e2d5SGreg Clayton     return is_loaded;
896c859e2d5SGreg Clayton }
897c859e2d5SGreg Clayton 
8981f746071SGreg Clayton uint32_t
899*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetAddressByteSize ()
9001f746071SGreg Clayton {
901*306bd0aaSJason Molenda     if (m_memory_module_sp)
902*306bd0aaSJason Molenda         return m_memory_module_sp->GetArchitecture().GetAddressByteSize();
903*306bd0aaSJason Molenda     if (m_module_sp)
904*306bd0aaSJason Molenda         return m_module_sp->GetArchitecture().GetAddressByteSize();
9051f746071SGreg Clayton     return 0;
9061f746071SGreg Clayton }
9071f746071SGreg Clayton 
9081f746071SGreg Clayton lldb::ByteOrder
909*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetByteOrder()
9101f746071SGreg Clayton {
911*306bd0aaSJason Molenda     if (m_memory_module_sp)
912*306bd0aaSJason Molenda         return m_memory_module_sp->GetArchitecture().GetByteOrder();
913*306bd0aaSJason Molenda     if (m_module_sp)
914*306bd0aaSJason Molenda         return m_module_sp->GetArchitecture().GetByteOrder();
9151f746071SGreg Clayton     return lldb::endian::InlHostByteOrder();
9161f746071SGreg Clayton }
9171f746071SGreg Clayton 
9181f746071SGreg Clayton lldb_private::ArchSpec
919*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::GetArchitecture () const
9201f746071SGreg Clayton {
921*306bd0aaSJason Molenda     if (m_memory_module_sp)
922*306bd0aaSJason Molenda         return m_memory_module_sp->GetArchitecture();
923*306bd0aaSJason Molenda     if (m_module_sp)
924*306bd0aaSJason Molenda         return m_module_sp->GetArchitecture();
9251f746071SGreg Clayton     return lldb_private::ArchSpec ();
9261f746071SGreg Clayton }
9271f746071SGreg Clayton 
9281f746071SGreg Clayton 
929d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
930d4bfbc9aSGreg Clayton // Load the kernel module and initialize the "m_kernel" member. Return
931d4bfbc9aSGreg Clayton // true _only_ if the kernel is loaded the first time through (subsequent
932d4bfbc9aSGreg Clayton // calls to this function should return false after the kernel has been
933d4bfbc9aSGreg Clayton // already loaded).
934d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
935d4bfbc9aSGreg Clayton void
936944b828aSGreg Clayton DynamicLoaderDarwinKernel::LoadKernelModuleIfNeeded()
937d4bfbc9aSGreg Clayton {
938d4bfbc9aSGreg Clayton     if (!m_kext_summary_header_ptr_addr.IsValid())
939d4bfbc9aSGreg Clayton     {
940*306bd0aaSJason Molenda         m_kernel.Clear();
941*306bd0aaSJason Molenda         m_kernel.SetModule (m_process->GetTarget().GetExecutableModule());
942*306bd0aaSJason Molenda         m_kernel.SetIsKernel(true);
9434bd4e7e3SJason Molenda 
9444bd4e7e3SJason Molenda         ConstString kernel_name("mach_kernel");
945*306bd0aaSJason Molenda         if (m_kernel.GetModule().get()
946*306bd0aaSJason Molenda             && m_kernel.GetModule()->GetObjectFile()
947*306bd0aaSJason Molenda             && !m_kernel.GetModule()->GetObjectFile()->GetFileSpec().GetFilename().IsEmpty())
9484bd4e7e3SJason Molenda         {
949*306bd0aaSJason Molenda             kernel_name = m_kernel.GetModule()->GetObjectFile()->GetFileSpec().GetFilename();
9504bd4e7e3SJason Molenda         }
951*306bd0aaSJason Molenda         m_kernel.SetName (kernel_name.AsCString());
9524bd4e7e3SJason Molenda 
953*306bd0aaSJason Molenda         if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS)
954d4bfbc9aSGreg Clayton         {
955*306bd0aaSJason Molenda             m_kernel.SetLoadAddress(m_kernel_load_address);
956*306bd0aaSJason Molenda             if (m_kernel.GetLoadAddress() == LLDB_INVALID_ADDRESS && m_kernel.GetModule())
957c859e2d5SGreg Clayton             {
958c859e2d5SGreg Clayton                 // We didn't get a hint from the process, so we will
959c859e2d5SGreg Clayton                 // try the kernel at the address that it exists at in
960c859e2d5SGreg Clayton                 // the file if we have one
961*306bd0aaSJason Molenda                 ObjectFile *kernel_object_file = m_kernel.GetModule()->GetObjectFile();
962c859e2d5SGreg Clayton                 if (kernel_object_file)
9634bd4e7e3SJason Molenda                 {
9644bd4e7e3SJason Molenda                     addr_t load_address = kernel_object_file->GetHeaderAddress().GetLoadAddress(&m_process->GetTarget());
9654bd4e7e3SJason Molenda                     addr_t file_address = kernel_object_file->GetHeaderAddress().GetFileAddress();
9664bd4e7e3SJason Molenda                     if (load_address != LLDB_INVALID_ADDRESS && load_address != 0)
9674bd4e7e3SJason Molenda                     {
968*306bd0aaSJason Molenda                         m_kernel.SetLoadAddress (load_address);
9694bd4e7e3SJason Molenda                         if (load_address != file_address)
9704bd4e7e3SJason Molenda                         {
9714bd4e7e3SJason Molenda                             // Don't accidentally relocate the kernel to the File address --
9724bd4e7e3SJason Molenda                             // the Load address has already been set to its actual in-memory address.
9734bd4e7e3SJason Molenda                             // Mark it as IsLoaded.
974*306bd0aaSJason Molenda                             m_kernel.SetProcessStopId (m_process->GetStopID());
9754bd4e7e3SJason Molenda                         }
9764bd4e7e3SJason Molenda                     }
9774bd4e7e3SJason Molenda                     else
9784bd4e7e3SJason Molenda                     {
979*306bd0aaSJason Molenda                         m_kernel.SetLoadAddress(file_address);
9804bd4e7e3SJason Molenda                     }
9814bd4e7e3SJason Molenda                 }
982c859e2d5SGreg Clayton             }
983c859e2d5SGreg Clayton         }
984c859e2d5SGreg Clayton 
985*306bd0aaSJason Molenda         if (m_kernel.GetLoadAddress() != LLDB_INVALID_ADDRESS)
9862af282a1SGreg Clayton         {
9872af282a1SGreg Clayton             if (!m_kernel.LoadImageUsingMemoryModule (m_process))
9882af282a1SGreg Clayton             {
9892af282a1SGreg Clayton                 m_kernel.LoadImageAtFileAddress (m_process);
9902af282a1SGreg Clayton             }
9912af282a1SGreg Clayton         }
992c859e2d5SGreg Clayton 
993*306bd0aaSJason Molenda         if (m_kernel.IsLoaded() && m_kernel.GetModule())
994c859e2d5SGreg Clayton         {
995d4bfbc9aSGreg Clayton             static ConstString kext_summary_symbol ("gLoadedKextSummaries");
996*306bd0aaSJason Molenda             const Symbol *symbol = m_kernel.GetModule()->FindFirstSymbolWithNameAndType (kext_summary_symbol, eSymbolTypeData);
997d4bfbc9aSGreg Clayton             if (symbol)
998c859e2d5SGreg Clayton             {
999e7612134SGreg Clayton                 m_kext_summary_header_ptr_addr = symbol->GetAddress();
1000d4bfbc9aSGreg Clayton                 // Update all image infos
1001d4bfbc9aSGreg Clayton                 ReadAllKextSummaries ();
1002d4bfbc9aSGreg Clayton             }
1003d4bfbc9aSGreg Clayton         }
1004d4bfbc9aSGreg Clayton         else
1005d4bfbc9aSGreg Clayton         {
1006*306bd0aaSJason Molenda             m_kernel.Clear();
1007d4bfbc9aSGreg Clayton         }
1008d4bfbc9aSGreg Clayton     }
1009d4bfbc9aSGreg Clayton }
1010d4bfbc9aSGreg Clayton 
1011d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1012d4bfbc9aSGreg Clayton // Static callback function that gets called when our DYLD notification
1013d4bfbc9aSGreg Clayton // breakpoint gets hit. We update all of our image infos and then
1014d4bfbc9aSGreg Clayton // let our super class DynamicLoader class decide if we should stop
1015d4bfbc9aSGreg Clayton // or not (based on global preference).
1016d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1017d4bfbc9aSGreg Clayton bool
1018944b828aSGreg Clayton DynamicLoaderDarwinKernel::BreakpointHitCallback (void *baton,
1019d4bfbc9aSGreg Clayton                                                   StoppointCallbackContext *context,
1020d4bfbc9aSGreg Clayton                                                   user_id_t break_id,
1021d4bfbc9aSGreg Clayton                                                   user_id_t break_loc_id)
1022d4bfbc9aSGreg Clayton {
1023944b828aSGreg Clayton     return static_cast<DynamicLoaderDarwinKernel*>(baton)->BreakpointHit (context, break_id, break_loc_id);
1024d4bfbc9aSGreg Clayton }
1025d4bfbc9aSGreg Clayton 
1026d4bfbc9aSGreg Clayton bool
1027944b828aSGreg Clayton DynamicLoaderDarwinKernel::BreakpointHit (StoppointCallbackContext *context,
1028d4bfbc9aSGreg Clayton                                           user_id_t break_id,
1029d4bfbc9aSGreg Clayton                                           user_id_t break_loc_id)
1030d4bfbc9aSGreg Clayton {
1031d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAnyCategoriesSet (LIBLLDB_LOG_DYNAMIC_LOADER));
1032d4bfbc9aSGreg Clayton     if (log)
1033944b828aSGreg Clayton         log->Printf ("DynamicLoaderDarwinKernel::BreakpointHit (...)\n");
1034d4bfbc9aSGreg Clayton 
1035d4bfbc9aSGreg Clayton     ReadAllKextSummaries ();
1036d4bfbc9aSGreg Clayton 
1037d4bfbc9aSGreg Clayton     if (log)
1038d4bfbc9aSGreg Clayton         PutToLog(log.get());
1039d4bfbc9aSGreg Clayton 
1040d4bfbc9aSGreg Clayton     return GetStopWhenImagesChange();
1041d4bfbc9aSGreg Clayton }
1042d4bfbc9aSGreg Clayton 
1043d4bfbc9aSGreg Clayton 
1044d4bfbc9aSGreg Clayton bool
1045944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadKextSummaryHeader ()
1046d4bfbc9aSGreg Clayton {
1047d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
1048d4bfbc9aSGreg Clayton 
1049d4bfbc9aSGreg Clayton     // the all image infos is already valid for this process stop ID
1050d4bfbc9aSGreg Clayton 
1051d4bfbc9aSGreg Clayton     if (m_kext_summary_header_ptr_addr.IsValid())
1052d4bfbc9aSGreg Clayton     {
1053d4bfbc9aSGreg Clayton         const uint32_t addr_size = m_kernel.GetAddressByteSize ();
1054d4bfbc9aSGreg Clayton         const ByteOrder byte_order = m_kernel.GetByteOrder();
1055d4bfbc9aSGreg Clayton         Error error;
1056d4bfbc9aSGreg Clayton         // Read enough bytes for a "OSKextLoadedKextSummaryHeader" structure
1057d4bfbc9aSGreg Clayton         // which is currenty 4 uint32_t and a pointer.
1058d4bfbc9aSGreg Clayton         uint8_t buf[24];
1059d4bfbc9aSGreg Clayton         DataExtractor data (buf, sizeof(buf), byte_order, addr_size);
1060d4bfbc9aSGreg Clayton         const size_t count = 4 * sizeof(uint32_t) + addr_size;
1061d4bfbc9aSGreg Clayton         const bool prefer_file_cache = false;
1062d4bfbc9aSGreg Clayton         if (m_process->GetTarget().ReadPointerFromMemory (m_kext_summary_header_ptr_addr,
1063d4bfbc9aSGreg Clayton                                                           prefer_file_cache,
1064d4bfbc9aSGreg Clayton                                                           error,
1065d4bfbc9aSGreg Clayton                                                           m_kext_summary_header_addr))
1066d4bfbc9aSGreg Clayton         {
1067d4bfbc9aSGreg Clayton             // We got a valid address for our kext summary header and make sure it isn't NULL
1068d4bfbc9aSGreg Clayton             if (m_kext_summary_header_addr.IsValid() &&
1069d4bfbc9aSGreg Clayton                 m_kext_summary_header_addr.GetFileAddress() != 0)
1070d4bfbc9aSGreg Clayton             {
1071d4bfbc9aSGreg Clayton                 const size_t bytes_read = m_process->GetTarget().ReadMemory (m_kext_summary_header_addr, prefer_file_cache, buf, count, error);
1072d4bfbc9aSGreg Clayton                 if (bytes_read == count)
1073d4bfbc9aSGreg Clayton                 {
1074c7bece56SGreg Clayton                     lldb::offset_t offset = 0;
1075d4bfbc9aSGreg Clayton                     m_kext_summary_header.version = data.GetU32(&offset);
1076d4bfbc9aSGreg Clayton                     if (m_kext_summary_header.version >= 2)
1077d4bfbc9aSGreg Clayton                     {
1078d4bfbc9aSGreg Clayton                         m_kext_summary_header.entry_size = data.GetU32(&offset);
1079d4bfbc9aSGreg Clayton                     }
1080d4bfbc9aSGreg Clayton                     else
1081d4bfbc9aSGreg Clayton                     {
1082d4bfbc9aSGreg Clayton                         // Versions less than 2 didn't have an entry size, it was hard coded
1083d4bfbc9aSGreg Clayton                         m_kext_summary_header.entry_size = KERNEL_MODULE_ENTRY_SIZE_VERSION_1;
1084d4bfbc9aSGreg Clayton                     }
1085d4bfbc9aSGreg Clayton                     m_kext_summary_header.entry_count = data.GetU32(&offset);
1086d4bfbc9aSGreg Clayton                     return true;
1087d4bfbc9aSGreg Clayton                 }
1088d4bfbc9aSGreg Clayton             }
1089d4bfbc9aSGreg Clayton         }
1090d4bfbc9aSGreg Clayton     }
1091d4bfbc9aSGreg Clayton     m_kext_summary_header_addr.Clear();
1092d4bfbc9aSGreg Clayton     return false;
1093d4bfbc9aSGreg Clayton }
1094d4bfbc9aSGreg Clayton 
1095*306bd0aaSJason Molenda // We've either (a) just attached to a new kernel, or (b) the kexts-changed breakpoint was hit
1096*306bd0aaSJason Molenda // and we need to figure out what kexts have been added or removed.
1097*306bd0aaSJason Molenda // Read the kext summaries from the inferior kernel memory, compare them against the
1098*306bd0aaSJason Molenda // m_known_kexts vector and update the m_known_kexts vector as needed to keep in sync with the
1099*306bd0aaSJason Molenda // inferior.
1100d4bfbc9aSGreg Clayton 
1101d4bfbc9aSGreg Clayton bool
1102*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::ParseKextSummaries (const Address &kext_summary_addr, uint32_t count)
1103d4bfbc9aSGreg Clayton {
1104*306bd0aaSJason Molenda     KextImageInfo::collection kext_summaries;
1105d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAnyCategoriesSet (LIBLLDB_LOG_DYNAMIC_LOADER));
1106d4bfbc9aSGreg Clayton     if (log)
1107*306bd0aaSJason Molenda         log->Printf ("Kexts-changed breakpoint hit, there are %d kexts currently.\n", count);
1108d4bfbc9aSGreg Clayton 
1109d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
1110d4bfbc9aSGreg Clayton 
1111d4bfbc9aSGreg Clayton     if (!ReadKextSummaries (kext_summary_addr, count, kext_summaries))
1112d4bfbc9aSGreg Clayton         return false;
1113d4bfbc9aSGreg Clayton 
1114*306bd0aaSJason Molenda     // By default, all kexts we've loaded in the past are marked as "remove" and all of the kexts
1115*306bd0aaSJason Molenda     // we just found out about from ReadKextSummaries are marked as "add".
1116*306bd0aaSJason Molenda     std::vector<bool> to_be_removed(m_known_kexts.size(), true);
1117*306bd0aaSJason Molenda     std::vector<bool> to_be_added(count, true);
1118*306bd0aaSJason Molenda 
1119*306bd0aaSJason Molenda     int number_of_new_kexts_being_added = 0;
1120*306bd0aaSJason Molenda     int number_of_old_kexts_being_removed = m_known_kexts.size();
1121*306bd0aaSJason Molenda 
1122*306bd0aaSJason Molenda     const uint32_t new_kexts_size = kext_summaries.size();
1123*306bd0aaSJason Molenda     const uint32_t old_kexts_size = m_known_kexts.size();
1124*306bd0aaSJason Molenda 
1125*306bd0aaSJason Molenda     // The m_known_kexts vector may have entries that have been Cleared,
1126*306bd0aaSJason Molenda     // or are a kernel.
1127*306bd0aaSJason Molenda     for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++)
1128*306bd0aaSJason Molenda     {
1129*306bd0aaSJason Molenda         bool ignore = false;
1130*306bd0aaSJason Molenda         KextImageInfo &image_info = m_known_kexts[old_kext];
1131*306bd0aaSJason Molenda         if (image_info.IsKernel())
1132*306bd0aaSJason Molenda         {
1133*306bd0aaSJason Molenda             ignore = true;
1134*306bd0aaSJason Molenda         }
1135*306bd0aaSJason Molenda         else if (image_info.GetLoadAddress() == LLDB_INVALID_ADDRESS && !image_info.GetModule())
1136*306bd0aaSJason Molenda         {
1137*306bd0aaSJason Molenda             ignore = true;
1138*306bd0aaSJason Molenda         }
1139*306bd0aaSJason Molenda 
1140*306bd0aaSJason Molenda         if (ignore)
1141*306bd0aaSJason Molenda         {
1142*306bd0aaSJason Molenda             number_of_old_kexts_being_removed--;
1143*306bd0aaSJason Molenda             to_be_removed[old_kext] = false;
1144*306bd0aaSJason Molenda         }
1145*306bd0aaSJason Molenda     }
1146*306bd0aaSJason Molenda 
1147*306bd0aaSJason Molenda     // Scan over the list of kexts we just read from the kernel, note those that
1148*306bd0aaSJason Molenda     // need to be added and those already loaded.
1149*306bd0aaSJason Molenda     for (uint32_t new_kext = 0; new_kext < new_kexts_size; new_kext++)
1150*306bd0aaSJason Molenda     {
1151*306bd0aaSJason Molenda         bool add_this_one = true;
1152*306bd0aaSJason Molenda         for (uint32_t old_kext = 0; old_kext < old_kexts_size; old_kext++)
1153*306bd0aaSJason Molenda         {
1154*306bd0aaSJason Molenda             if (m_known_kexts[old_kext] == kext_summaries[new_kext])
1155*306bd0aaSJason Molenda             {
1156*306bd0aaSJason Molenda                 // We already have this kext, don't re-load it.
1157*306bd0aaSJason Molenda                 to_be_added[new_kext] = false;
1158*306bd0aaSJason Molenda                 // This kext is still present, do not remove it.
1159*306bd0aaSJason Molenda                 to_be_removed[old_kext] = false;
1160*306bd0aaSJason Molenda 
1161*306bd0aaSJason Molenda                 number_of_old_kexts_being_removed--;
1162*306bd0aaSJason Molenda                 add_this_one = false;
1163*306bd0aaSJason Molenda                 break;
1164*306bd0aaSJason Molenda             }
1165*306bd0aaSJason Molenda         }
1166*306bd0aaSJason Molenda         if (add_this_one)
1167*306bd0aaSJason Molenda         {
1168*306bd0aaSJason Molenda             number_of_new_kexts_being_added++;
1169*306bd0aaSJason Molenda         }
1170*306bd0aaSJason Molenda     }
1171*306bd0aaSJason Molenda 
1172*306bd0aaSJason Molenda     if (number_of_new_kexts_being_added == 0 && number_of_old_kexts_being_removed == 0)
1173*306bd0aaSJason Molenda         return true;
1174*306bd0aaSJason Molenda 
1175d4bfbc9aSGreg Clayton     Stream *s = &m_process->GetTarget().GetDebugger().GetOutputStream();
117668b3607fSJason Molenda     if (s)
1177d4bfbc9aSGreg Clayton     {
1178*306bd0aaSJason Molenda         if (number_of_new_kexts_being_added > 0 && number_of_old_kexts_being_removed > 0)
1179*306bd0aaSJason Molenda         {
1180*306bd0aaSJason Molenda             s->Printf ("Loading %d kext modules and unloading %d kext modules ", number_of_new_kexts_being_added, number_of_old_kexts_being_removed);
1181*306bd0aaSJason Molenda         }
1182*306bd0aaSJason Molenda         else if (number_of_new_kexts_being_added > 0)
1183*306bd0aaSJason Molenda         {
1184*306bd0aaSJason Molenda             s->Printf ("Loading %d kext modules ", number_of_new_kexts_being_added);
1185*306bd0aaSJason Molenda         }
1186*306bd0aaSJason Molenda         else if (number_of_old_kexts_being_removed > 0)
1187*306bd0aaSJason Molenda         {
1188*306bd0aaSJason Molenda             s->Printf ("Unloading %d kext modules ", number_of_old_kexts_being_removed);
1189*306bd0aaSJason Molenda         }
1190*306bd0aaSJason Molenda     }
1191*306bd0aaSJason Molenda 
1192*306bd0aaSJason Molenda     if (log)
1193*306bd0aaSJason Molenda         log->Printf ("DynamicLoaderDarwinKernel::ParseKextSummaries: %d kexts added, %d kexts removed", number_of_new_kexts_being_added, number_of_old_kexts_being_removed);
1194*306bd0aaSJason Molenda 
1195*306bd0aaSJason Molenda     if (number_of_new_kexts_being_added > 0)
1196*306bd0aaSJason Molenda     {
1197*306bd0aaSJason Molenda         ModuleList loaded_module_list;
1198*306bd0aaSJason Molenda 
1199*306bd0aaSJason Molenda         const uint32_t num_of_new_kexts = kext_summaries.size();
1200*306bd0aaSJason Molenda         for (uint32_t new_kext = 0; new_kext < num_of_new_kexts; new_kext++)
1201*306bd0aaSJason Molenda         {
1202*306bd0aaSJason Molenda             if (to_be_added[new_kext] == true)
1203*306bd0aaSJason Molenda             {
1204*306bd0aaSJason Molenda                 KextImageInfo &image_info = kext_summaries[new_kext];
1205*306bd0aaSJason Molenda                 if (!image_info.LoadImageUsingMemoryModule (m_process))
1206*306bd0aaSJason Molenda                     image_info.LoadImageAtFileAddress (m_process);
1207*306bd0aaSJason Molenda 
1208*306bd0aaSJason Molenda                 m_known_kexts.push_back(image_info);
1209*306bd0aaSJason Molenda 
1210*306bd0aaSJason Molenda                 if (image_info.GetModule() && m_process->GetStopID() == image_info.GetProcessStopId())
1211*306bd0aaSJason Molenda                     loaded_module_list.AppendIfNeeded (image_info.GetModule());
1212d4bfbc9aSGreg Clayton 
1213d4bfbc9aSGreg Clayton                 if (s)
121468b3607fSJason Molenda                     s->Printf (".");
1215d4bfbc9aSGreg Clayton 
1216d4bfbc9aSGreg Clayton                 if (log)
1217*306bd0aaSJason Molenda                     kext_summaries[new_kext].PutToLog (log.get());
1218d4bfbc9aSGreg Clayton             }
1219*306bd0aaSJason Molenda         }
1220*306bd0aaSJason Molenda         m_process->GetTarget().ModulesDidLoad (loaded_module_list);
1221*306bd0aaSJason Molenda     }
1222*306bd0aaSJason Molenda 
1223*306bd0aaSJason Molenda     if (number_of_old_kexts_being_removed > 0)
1224*306bd0aaSJason Molenda     {
1225*306bd0aaSJason Molenda         ModuleList loaded_module_list;
1226*306bd0aaSJason Molenda         const uint32_t num_of_old_kexts = m_known_kexts.size();
1227*306bd0aaSJason Molenda         for (uint32_t old_kext = 0; old_kext < num_of_old_kexts; old_kext++)
1228*306bd0aaSJason Molenda         {
1229*306bd0aaSJason Molenda             ModuleList unloaded_module_list;
1230*306bd0aaSJason Molenda             if (to_be_removed[old_kext])
1231*306bd0aaSJason Molenda             {
1232*306bd0aaSJason Molenda                 KextImageInfo &image_info = m_known_kexts[old_kext];
1233*306bd0aaSJason Molenda                 // You can't unload the kernel.
1234*306bd0aaSJason Molenda                 if (!image_info.IsKernel())
1235*306bd0aaSJason Molenda                 {
1236*306bd0aaSJason Molenda                     if (image_info.GetModule())
1237*306bd0aaSJason Molenda                     {
1238*306bd0aaSJason Molenda                         unloaded_module_list.AppendIfNeeded (image_info.GetModule());
1239*306bd0aaSJason Molenda                     }
1240*306bd0aaSJason Molenda                     if (s)
1241*306bd0aaSJason Molenda                         s->Printf (".");
1242*306bd0aaSJason Molenda                     image_info.Clear();
1243*306bd0aaSJason Molenda                     // should pull it out of the KextImageInfos vector but that would mutate the list and invalidate
1244*306bd0aaSJason Molenda                     // the to_be_removed bool vector; leaving it in place once Cleared() is relatively harmless.
1245*306bd0aaSJason Molenda                 }
1246*306bd0aaSJason Molenda             }
1247*306bd0aaSJason Molenda             m_process->GetTarget().ModulesDidUnload (unloaded_module_list);
1248*306bd0aaSJason Molenda         }
1249*306bd0aaSJason Molenda     }
1250*306bd0aaSJason Molenda 
125168b3607fSJason Molenda     if (s)
125268b3607fSJason Molenda     {
125368b3607fSJason Molenda         s->Printf (" done.\n");
125468b3607fSJason Molenda         s->Flush ();
125568b3607fSJason Molenda     }
125668b3607fSJason Molenda 
1257*306bd0aaSJason Molenda     return true;
1258d4bfbc9aSGreg Clayton }
1259d4bfbc9aSGreg Clayton 
1260*306bd0aaSJason Molenda // Adds the modules in image_infos to m_known_kexts.
1261d4bfbc9aSGreg Clayton 
1262d4bfbc9aSGreg Clayton bool
1263*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::AddModulesUsingImageInfos (KextImageInfo::collection &image_infos)
1264d4bfbc9aSGreg Clayton {
1265d4bfbc9aSGreg Clayton     // Now add these images to the main list.
1266d4bfbc9aSGreg Clayton     ModuleList loaded_module_list;
1267d4bfbc9aSGreg Clayton 
1268d4bfbc9aSGreg Clayton     for (uint32_t idx = 0; idx < image_infos.size(); ++idx)
1269d4bfbc9aSGreg Clayton     {
1270*306bd0aaSJason Molenda         KextImageInfo &image_info = image_infos[idx];
1271*306bd0aaSJason Molenda         m_known_kexts.push_back(image_info);
1272d4bfbc9aSGreg Clayton 
1273*306bd0aaSJason Molenda         if (image_info.GetModule() && m_process->GetStopID() == image_info.GetProcessStopId())
1274*306bd0aaSJason Molenda             loaded_module_list.AppendIfNeeded (image_infos[idx].GetModule());
1275d4bfbc9aSGreg Clayton     }
1276d4bfbc9aSGreg Clayton 
1277d4bfbc9aSGreg Clayton     m_process->GetTarget().ModulesDidLoad (loaded_module_list);
1278d4bfbc9aSGreg Clayton     return true;
1279d4bfbc9aSGreg Clayton }
1280d4bfbc9aSGreg Clayton 
1281d4bfbc9aSGreg Clayton 
1282d4bfbc9aSGreg Clayton uint32_t
1283944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadKextSummaries (const Address &kext_summary_addr,
1284d4bfbc9aSGreg Clayton                                               uint32_t image_infos_count,
1285*306bd0aaSJason Molenda                                               KextImageInfo::collection &image_infos)
1286d4bfbc9aSGreg Clayton {
1287d4bfbc9aSGreg Clayton     const ByteOrder endian = m_kernel.GetByteOrder();
1288d4bfbc9aSGreg Clayton     const uint32_t addr_size = m_kernel.GetAddressByteSize();
1289d4bfbc9aSGreg Clayton 
1290d4bfbc9aSGreg Clayton     image_infos.resize(image_infos_count);
1291d4bfbc9aSGreg Clayton     const size_t count = image_infos.size() * m_kext_summary_header.entry_size;
1292d4bfbc9aSGreg Clayton     DataBufferHeap data(count, 0);
1293d4bfbc9aSGreg Clayton     Error error;
1294d4bfbc9aSGreg Clayton 
1295d4bfbc9aSGreg Clayton     const bool prefer_file_cache = false;
1296d4bfbc9aSGreg Clayton     const size_t bytes_read = m_process->GetTarget().ReadMemory (kext_summary_addr,
1297d4bfbc9aSGreg Clayton                                                                  prefer_file_cache,
1298d4bfbc9aSGreg Clayton                                                                  data.GetBytes(),
1299d4bfbc9aSGreg Clayton                                                                  data.GetByteSize(),
1300d4bfbc9aSGreg Clayton                                                                  error);
1301d4bfbc9aSGreg Clayton     if (bytes_read == count)
1302d4bfbc9aSGreg Clayton     {
1303d4bfbc9aSGreg Clayton 
1304d4bfbc9aSGreg Clayton         DataExtractor extractor (data.GetBytes(), data.GetByteSize(), endian, addr_size);
1305d4bfbc9aSGreg Clayton         uint32_t i=0;
1306d4bfbc9aSGreg Clayton         for (uint32_t kext_summary_offset = 0;
1307d4bfbc9aSGreg Clayton              i < image_infos.size() && extractor.ValidOffsetForDataOfSize(kext_summary_offset, m_kext_summary_header.entry_size);
1308d4bfbc9aSGreg Clayton              ++i, kext_summary_offset += m_kext_summary_header.entry_size)
1309d4bfbc9aSGreg Clayton         {
1310c7bece56SGreg Clayton             lldb::offset_t offset = kext_summary_offset;
1311d4bfbc9aSGreg Clayton             const void *name_data = extractor.GetData(&offset, KERNEL_MODULE_MAX_NAME);
1312d4bfbc9aSGreg Clayton             if (name_data == NULL)
1313d4bfbc9aSGreg Clayton                 break;
1314*306bd0aaSJason Molenda             image_infos[i].SetName ((const char *) name_data);
1315*306bd0aaSJason Molenda             UUID uuid (extractor.GetData (&offset, 16), 16);
1316*306bd0aaSJason Molenda             image_infos[i].SetUUID (uuid);
1317*306bd0aaSJason Molenda             image_infos[i].SetLoadAddress (extractor.GetU64(&offset));
1318*306bd0aaSJason Molenda             image_infos[i].SetSize (extractor.GetU64(&offset));
1319d4bfbc9aSGreg Clayton         }
1320d4bfbc9aSGreg Clayton         if (i < image_infos.size())
1321d4bfbc9aSGreg Clayton             image_infos.resize(i);
1322d4bfbc9aSGreg Clayton     }
1323d4bfbc9aSGreg Clayton     else
1324d4bfbc9aSGreg Clayton     {
1325d4bfbc9aSGreg Clayton         image_infos.clear();
1326d4bfbc9aSGreg Clayton     }
1327d4bfbc9aSGreg Clayton     return image_infos.size();
1328d4bfbc9aSGreg Clayton }
1329d4bfbc9aSGreg Clayton 
1330d4bfbc9aSGreg Clayton bool
1331944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadAllKextSummaries ()
1332d4bfbc9aSGreg Clayton {
1333d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAnyCategoriesSet (LIBLLDB_LOG_DYNAMIC_LOADER));
1334d4bfbc9aSGreg Clayton 
1335d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
1336d4bfbc9aSGreg Clayton 
1337d4bfbc9aSGreg Clayton     if (ReadKextSummaryHeader ())
1338d4bfbc9aSGreg Clayton     {
1339d4bfbc9aSGreg Clayton         if (m_kext_summary_header.entry_count > 0 && m_kext_summary_header_addr.IsValid())
1340d4bfbc9aSGreg Clayton         {
1341d4bfbc9aSGreg Clayton             Address summary_addr (m_kext_summary_header_addr);
1342d4bfbc9aSGreg Clayton             summary_addr.Slide(m_kext_summary_header.GetSize());
1343d4bfbc9aSGreg Clayton             if (!ParseKextSummaries (summary_addr, m_kext_summary_header.entry_count))
1344d4bfbc9aSGreg Clayton             {
1345*306bd0aaSJason Molenda                 m_known_kexts.clear();
1346d4bfbc9aSGreg Clayton             }
1347d4bfbc9aSGreg Clayton             return true;
1348d4bfbc9aSGreg Clayton         }
1349d4bfbc9aSGreg Clayton     }
1350d4bfbc9aSGreg Clayton     return false;
1351d4bfbc9aSGreg Clayton }
1352d4bfbc9aSGreg Clayton 
1353d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1354d4bfbc9aSGreg Clayton // Dump an image info structure to the file handle provided.
1355d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1356d4bfbc9aSGreg Clayton void
1357*306bd0aaSJason Molenda DynamicLoaderDarwinKernel::KextImageInfo::PutToLog (Log *log) const
1358d4bfbc9aSGreg Clayton {
1359d4bfbc9aSGreg Clayton     if (log == NULL)
1360d4bfbc9aSGreg Clayton         return;
1361*306bd0aaSJason Molenda     const uint8_t *u = (uint8_t *) m_uuid.GetBytes();
1362d4bfbc9aSGreg Clayton 
1363*306bd0aaSJason Molenda     if (m_load_address == LLDB_INVALID_ADDRESS)
1364d4bfbc9aSGreg Clayton     {
1365d4bfbc9aSGreg Clayton         if (u)
1366d4bfbc9aSGreg Clayton         {
1367d4bfbc9aSGreg Clayton             log->Printf("\tuuid=%2.2X%2.2X%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X%2.2X%2.2X%2.2X%2.2X name=\"%s\" (UNLOADED)",
1368d4bfbc9aSGreg Clayton                         u[ 0], u[ 1], u[ 2], u[ 3],
1369d4bfbc9aSGreg Clayton                         u[ 4], u[ 5], u[ 6], u[ 7],
1370d4bfbc9aSGreg Clayton                         u[ 8], u[ 9], u[10], u[11],
1371d4bfbc9aSGreg Clayton                         u[12], u[13], u[14], u[15],
1372*306bd0aaSJason Molenda                         m_name.c_str());
1373d4bfbc9aSGreg Clayton         }
1374d4bfbc9aSGreg Clayton         else
1375*306bd0aaSJason Molenda             log->Printf("\tname=\"%s\" (UNLOADED)", m_name.c_str());
1376d4bfbc9aSGreg Clayton     }
1377d4bfbc9aSGreg Clayton     else
1378d4bfbc9aSGreg Clayton     {
1379d4bfbc9aSGreg Clayton         if (u)
1380d4bfbc9aSGreg Clayton         {
1381*306bd0aaSJason Molenda             log->Printf("\taddr=0x%16.16" PRIx64 " size=0x%16.16" PRIx64 " uuid=%2.2X%2.2X%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X-%2.2X%2.2X%2.2X%2.2X%2.2X%2.2X name=\"%s\"",
1382*306bd0aaSJason Molenda                         m_load_address, m_size,
1383d4bfbc9aSGreg Clayton                         u[ 0], u[ 1], u[ 2], u[ 3], u[ 4], u[ 5], u[ 6], u[ 7],
1384d4bfbc9aSGreg Clayton                         u[ 8], u[ 9], u[10], u[11], u[12], u[13], u[14], u[15],
1385*306bd0aaSJason Molenda                         m_name.c_str());
1386d4bfbc9aSGreg Clayton         }
1387d4bfbc9aSGreg Clayton         else
1388d4bfbc9aSGreg Clayton         {
1389*306bd0aaSJason Molenda             log->Printf("\t[0x%16.16" PRIx64 " - 0x%16.16" PRIx64 ") name=\"%s\"",
1390*306bd0aaSJason Molenda                         m_load_address, m_load_address+m_size, m_name.c_str());
1391d4bfbc9aSGreg Clayton         }
1392d4bfbc9aSGreg Clayton     }
1393d4bfbc9aSGreg Clayton }
1394d4bfbc9aSGreg Clayton 
1395d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1396d4bfbc9aSGreg Clayton // Dump the _dyld_all_image_infos members and all current image infos
1397d4bfbc9aSGreg Clayton // that we have parsed to the file handle provided.
1398d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1399d4bfbc9aSGreg Clayton void
1400944b828aSGreg Clayton DynamicLoaderDarwinKernel::PutToLog(Log *log) const
1401d4bfbc9aSGreg Clayton {
1402d4bfbc9aSGreg Clayton     if (log == NULL)
1403d4bfbc9aSGreg Clayton         return;
1404d4bfbc9aSGreg Clayton 
1405d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
1406d01b2953SDaniel Malea     log->Printf("gLoadedKextSummaries = 0x%16.16" PRIx64 " { version=%u, entry_size=%u, entry_count=%u }",
1407d4bfbc9aSGreg Clayton                 m_kext_summary_header_addr.GetFileAddress(),
1408d4bfbc9aSGreg Clayton                 m_kext_summary_header.version,
1409d4bfbc9aSGreg Clayton                 m_kext_summary_header.entry_size,
1410d4bfbc9aSGreg Clayton                 m_kext_summary_header.entry_count);
1411d4bfbc9aSGreg Clayton 
1412d4bfbc9aSGreg Clayton     size_t i;
1413*306bd0aaSJason Molenda     const size_t count = m_known_kexts.size();
1414d4bfbc9aSGreg Clayton     if (count > 0)
1415d4bfbc9aSGreg Clayton     {
1416d4bfbc9aSGreg Clayton         log->PutCString("Loaded:");
1417d4bfbc9aSGreg Clayton         for (i = 0; i<count; i++)
1418*306bd0aaSJason Molenda             m_known_kexts[i].PutToLog(log);
1419d4bfbc9aSGreg Clayton     }
1420d4bfbc9aSGreg Clayton }
1421d4bfbc9aSGreg Clayton 
1422d4bfbc9aSGreg Clayton void
1423944b828aSGreg Clayton DynamicLoaderDarwinKernel::PrivateInitialize(Process *process)
1424d4bfbc9aSGreg Clayton {
1425944b828aSGreg Clayton     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n", __FUNCTION__, StateAsCString(m_process->GetState()));
1426d4bfbc9aSGreg Clayton     Clear(true);
1427d4bfbc9aSGreg Clayton     m_process = process;
1428d4bfbc9aSGreg Clayton }
1429d4bfbc9aSGreg Clayton 
1430d4bfbc9aSGreg Clayton void
1431944b828aSGreg Clayton DynamicLoaderDarwinKernel::SetNotificationBreakpointIfNeeded ()
1432d4bfbc9aSGreg Clayton {
1433*306bd0aaSJason Molenda     if (m_break_id == LLDB_INVALID_BREAK_ID && m_kernel.GetModule())
1434d4bfbc9aSGreg Clayton     {
1435944b828aSGreg Clayton         DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n", __FUNCTION__, StateAsCString(m_process->GetState()));
1436d4bfbc9aSGreg Clayton 
1437d4bfbc9aSGreg Clayton 
1438a8558b62SJim Ingham         const bool internal_bp = true;
1439d4bfbc9aSGreg Clayton         const LazyBool skip_prologue = eLazyBoolNo;
1440969795f1SJim Ingham         FileSpecList module_spec_list;
1441*306bd0aaSJason Molenda         module_spec_list.Append (m_kernel.GetModule()->GetFileSpec());
1442969795f1SJim Ingham         Breakpoint *bp = m_process->GetTarget().CreateBreakpoint (&module_spec_list,
144387df91b8SJim Ingham                                                                   NULL,
1444d4bfbc9aSGreg Clayton                                                                   "OSKextLoadedKextSummariesUpdated",
1445d4bfbc9aSGreg Clayton                                                                   eFunctionNameTypeFull,
1446a8558b62SJim Ingham                                                                   skip_prologue,
1447a8558b62SJim Ingham                                                                   internal_bp).get();
1448d4bfbc9aSGreg Clayton 
1449944b828aSGreg Clayton         bp->SetCallback (DynamicLoaderDarwinKernel::BreakpointHitCallback, this, true);
1450d4bfbc9aSGreg Clayton         m_break_id = bp->GetID();
1451d4bfbc9aSGreg Clayton     }
1452d4bfbc9aSGreg Clayton }
1453d4bfbc9aSGreg Clayton 
1454d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1455d4bfbc9aSGreg Clayton // Member function that gets called when the process state changes.
1456d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
1457d4bfbc9aSGreg Clayton void
1458944b828aSGreg Clayton DynamicLoaderDarwinKernel::PrivateProcessStateChanged (Process *process, StateType state)
1459d4bfbc9aSGreg Clayton {
1460944b828aSGreg Clayton     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s(%s)\n", __FUNCTION__, StateAsCString(state));
1461d4bfbc9aSGreg Clayton     switch (state)
1462d4bfbc9aSGreg Clayton     {
1463d4bfbc9aSGreg Clayton     case eStateConnected:
1464d4bfbc9aSGreg Clayton     case eStateAttaching:
1465d4bfbc9aSGreg Clayton     case eStateLaunching:
1466d4bfbc9aSGreg Clayton     case eStateInvalid:
1467d4bfbc9aSGreg Clayton     case eStateUnloaded:
1468d4bfbc9aSGreg Clayton     case eStateExited:
1469d4bfbc9aSGreg Clayton     case eStateDetached:
1470d4bfbc9aSGreg Clayton         Clear(false);
1471d4bfbc9aSGreg Clayton         break;
1472d4bfbc9aSGreg Clayton 
1473d4bfbc9aSGreg Clayton     case eStateStopped:
1474d4bfbc9aSGreg Clayton         UpdateIfNeeded();
1475d4bfbc9aSGreg Clayton         break;
1476d4bfbc9aSGreg Clayton 
1477d4bfbc9aSGreg Clayton     case eStateRunning:
1478d4bfbc9aSGreg Clayton     case eStateStepping:
1479d4bfbc9aSGreg Clayton     case eStateCrashed:
1480d4bfbc9aSGreg Clayton     case eStateSuspended:
1481d4bfbc9aSGreg Clayton         break;
1482d4bfbc9aSGreg Clayton     }
1483d4bfbc9aSGreg Clayton }
1484d4bfbc9aSGreg Clayton 
1485d4bfbc9aSGreg Clayton ThreadPlanSP
1486944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetStepThroughTrampolinePlan (Thread &thread, bool stop_others)
1487d4bfbc9aSGreg Clayton {
1488d4bfbc9aSGreg Clayton     ThreadPlanSP thread_plan_sp;
1489d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAllCategoriesSet (LIBLLDB_LOG_STEP));
1490d4bfbc9aSGreg Clayton     if (log)
1491d4bfbc9aSGreg Clayton         log->Printf ("Could not find symbol for step through.");
1492d4bfbc9aSGreg Clayton     return thread_plan_sp;
1493d4bfbc9aSGreg Clayton }
1494d4bfbc9aSGreg Clayton 
1495d4bfbc9aSGreg Clayton Error
1496944b828aSGreg Clayton DynamicLoaderDarwinKernel::CanLoadImage ()
1497d4bfbc9aSGreg Clayton {
1498d4bfbc9aSGreg Clayton     Error error;
1499d4bfbc9aSGreg Clayton     error.SetErrorString("always unsafe to load or unload shared libraries in the darwin kernel");
1500d4bfbc9aSGreg Clayton     return error;
1501d4bfbc9aSGreg Clayton }
1502d4bfbc9aSGreg Clayton 
1503d4bfbc9aSGreg Clayton void
1504944b828aSGreg Clayton DynamicLoaderDarwinKernel::Initialize()
1505d4bfbc9aSGreg Clayton {
1506d4bfbc9aSGreg Clayton     PluginManager::RegisterPlugin (GetPluginNameStatic(),
1507d4bfbc9aSGreg Clayton                                    GetPluginDescriptionStatic(),
1508e8cd0c98SGreg Clayton                                    CreateInstance,
1509e8cd0c98SGreg Clayton                                    DebuggerInitialize);
1510d4bfbc9aSGreg Clayton }
1511d4bfbc9aSGreg Clayton 
1512d4bfbc9aSGreg Clayton void
1513944b828aSGreg Clayton DynamicLoaderDarwinKernel::Terminate()
1514d4bfbc9aSGreg Clayton {
1515d4bfbc9aSGreg Clayton     PluginManager::UnregisterPlugin (CreateInstance);
1516d4bfbc9aSGreg Clayton }
1517d4bfbc9aSGreg Clayton 
1518e8cd0c98SGreg Clayton void
1519e8cd0c98SGreg Clayton DynamicLoaderDarwinKernel::DebuggerInitialize (lldb_private::Debugger &debugger)
1520e8cd0c98SGreg Clayton {
1521e8cd0c98SGreg Clayton     if (!PluginManager::GetSettingForDynamicLoaderPlugin (debugger, DynamicLoaderDarwinKernelProperties::GetSettingName()))
1522e8cd0c98SGreg Clayton     {
1523e8cd0c98SGreg Clayton         const bool is_global_setting = true;
1524e8cd0c98SGreg Clayton         PluginManager::CreateSettingForDynamicLoaderPlugin (debugger,
1525e8cd0c98SGreg Clayton                                                             GetGlobalProperties()->GetValueProperties(),
1526e8cd0c98SGreg Clayton                                                             ConstString ("Properties for the DynamicLoaderDarwinKernel plug-in."),
1527e8cd0c98SGreg Clayton                                                             is_global_setting);
1528e8cd0c98SGreg Clayton     }
1529e8cd0c98SGreg Clayton }
1530d4bfbc9aSGreg Clayton 
1531d4bfbc9aSGreg Clayton const char *
1532944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginNameStatic()
1533d4bfbc9aSGreg Clayton {
1534468ea4ebSGreg Clayton     return "dynamic-loader.darwin-kernel";
1535d4bfbc9aSGreg Clayton }
1536d4bfbc9aSGreg Clayton 
1537d4bfbc9aSGreg Clayton const char *
1538944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginDescriptionStatic()
1539d4bfbc9aSGreg Clayton {
1540d4bfbc9aSGreg Clayton     return "Dynamic loader plug-in that watches for shared library loads/unloads in the MacOSX kernel.";
1541d4bfbc9aSGreg Clayton }
1542d4bfbc9aSGreg Clayton 
1543d4bfbc9aSGreg Clayton 
1544d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1545d4bfbc9aSGreg Clayton // PluginInterface protocol
1546d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1547d4bfbc9aSGreg Clayton const char *
1548944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginName()
1549d4bfbc9aSGreg Clayton {
1550944b828aSGreg Clayton     return "DynamicLoaderDarwinKernel";
1551d4bfbc9aSGreg Clayton }
1552d4bfbc9aSGreg Clayton 
1553d4bfbc9aSGreg Clayton const char *
1554944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetShortPluginName()
1555d4bfbc9aSGreg Clayton {
1556d4bfbc9aSGreg Clayton     return GetPluginNameStatic();
1557d4bfbc9aSGreg Clayton }
1558d4bfbc9aSGreg Clayton 
1559d4bfbc9aSGreg Clayton uint32_t
1560944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginVersion()
1561d4bfbc9aSGreg Clayton {
1562d4bfbc9aSGreg Clayton     return 1;
1563d4bfbc9aSGreg Clayton }
1564d4bfbc9aSGreg Clayton 
15651f746071SGreg Clayton lldb::ByteOrder
15661f746071SGreg Clayton DynamicLoaderDarwinKernel::GetByteOrderFromMagic (uint32_t magic)
15671f746071SGreg Clayton {
15681f746071SGreg Clayton     switch (magic)
15691f746071SGreg Clayton     {
15701f746071SGreg Clayton         case llvm::MachO::HeaderMagic32:
15711f746071SGreg Clayton         case llvm::MachO::HeaderMagic64:
15721f746071SGreg Clayton             return lldb::endian::InlHostByteOrder();
15731f746071SGreg Clayton 
15741f746071SGreg Clayton         case llvm::MachO::HeaderMagic32Swapped:
15751f746071SGreg Clayton         case llvm::MachO::HeaderMagic64Swapped:
15761f746071SGreg Clayton             if (lldb::endian::InlHostByteOrder() == lldb::eByteOrderBig)
15771f746071SGreg Clayton                 return lldb::eByteOrderLittle;
15781f746071SGreg Clayton             else
15791f746071SGreg Clayton                 return lldb::eByteOrderBig;
15801f746071SGreg Clayton 
15811f746071SGreg Clayton         default:
15821f746071SGreg Clayton             break;
15831f746071SGreg Clayton     }
15841f746071SGreg Clayton     return lldb::eByteOrderInvalid;
15851f746071SGreg Clayton }
15861f746071SGreg Clayton 
1587