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 
10d4bfbc9aSGreg Clayton #include "lldb/Breakpoint/StoppointCallbackContext.h"
11d4bfbc9aSGreg Clayton #include "lldb/Core/DataBuffer.h"
12d4bfbc9aSGreg Clayton #include "lldb/Core/DataBufferHeap.h"
13d4bfbc9aSGreg Clayton #include "lldb/Core/Debugger.h"
14d4bfbc9aSGreg Clayton #include "lldb/Core/Log.h"
15d4bfbc9aSGreg Clayton #include "lldb/Core/Module.h"
16d4bfbc9aSGreg Clayton #include "lldb/Core/PluginManager.h"
17d4bfbc9aSGreg Clayton #include "lldb/Core/State.h"
18d4bfbc9aSGreg Clayton #include "lldb/Symbol/ObjectFile.h"
19d4bfbc9aSGreg Clayton #include "lldb/Target/ObjCLanguageRuntime.h"
20d4bfbc9aSGreg Clayton #include "lldb/Target/RegisterContext.h"
21d4bfbc9aSGreg Clayton #include "lldb/Target/Target.h"
22d4bfbc9aSGreg Clayton #include "lldb/Target/Thread.h"
23d4bfbc9aSGreg Clayton #include "lldb/Target/ThreadPlanRunToAddress.h"
24d4bfbc9aSGreg Clayton #include "lldb/Target/StackFrame.h"
25d4bfbc9aSGreg Clayton 
26944b828aSGreg Clayton #include "DynamicLoaderDarwinKernel.h"
27d4bfbc9aSGreg Clayton 
28d4bfbc9aSGreg Clayton //#define ENABLE_DEBUG_PRINTF // COMMENT THIS LINE OUT PRIOR TO CHECKIN
29d4bfbc9aSGreg Clayton #ifdef ENABLE_DEBUG_PRINTF
30d4bfbc9aSGreg Clayton #include <stdio.h>
31d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...) printf(fmt, ## __VA_ARGS__)
32d4bfbc9aSGreg Clayton #else
33d4bfbc9aSGreg Clayton #define DEBUG_PRINTF(fmt, ...)
34d4bfbc9aSGreg Clayton #endif
35d4bfbc9aSGreg Clayton 
36d4bfbc9aSGreg Clayton using namespace lldb;
37d4bfbc9aSGreg Clayton using namespace lldb_private;
38d4bfbc9aSGreg Clayton 
39d4bfbc9aSGreg Clayton /// FIXME - The ObjC Runtime trampoline handler doesn't really belong here.
40d4bfbc9aSGreg Clayton /// I am putting it here so I can invoke it in the Trampoline code here, but
41d4bfbc9aSGreg Clayton /// it should be moved to the ObjC Runtime support when it is set up.
42d4bfbc9aSGreg Clayton 
43d4bfbc9aSGreg Clayton 
44d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
45d4bfbc9aSGreg Clayton // Create an instance of this class. This function is filled into
46d4bfbc9aSGreg Clayton // the plugin info class that gets handed out by the plugin factory and
47d4bfbc9aSGreg Clayton // allows the lldb to instantiate an instance of this class.
48d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
49d4bfbc9aSGreg Clayton DynamicLoader *
50944b828aSGreg Clayton DynamicLoaderDarwinKernel::CreateInstance (Process* process, bool force)
51d4bfbc9aSGreg Clayton {
52d4bfbc9aSGreg Clayton     bool create = force;
53d4bfbc9aSGreg Clayton     if (!create)
54d4bfbc9aSGreg Clayton     {
55d4bfbc9aSGreg Clayton         Module* exe_module = process->GetTarget().GetExecutableModulePointer();
56d4bfbc9aSGreg Clayton         if (exe_module)
57d4bfbc9aSGreg Clayton         {
58d4bfbc9aSGreg Clayton             ObjectFile *object_file = exe_module->GetObjectFile();
59d4bfbc9aSGreg Clayton             if (object_file)
60d4bfbc9aSGreg Clayton             {
61d4bfbc9aSGreg Clayton                 SectionList *section_list = object_file->GetSectionList();
62d4bfbc9aSGreg Clayton                 if (section_list)
63d4bfbc9aSGreg Clayton                 {
64d4bfbc9aSGreg Clayton                     static ConstString g_kld_section_name ("__KLD");
65d4bfbc9aSGreg Clayton                     if (section_list->FindSectionByName (g_kld_section_name))
66d4bfbc9aSGreg Clayton                     {
67d4bfbc9aSGreg Clayton                         create = true;
68d4bfbc9aSGreg Clayton                     }
69d4bfbc9aSGreg Clayton                 }
70d4bfbc9aSGreg Clayton             }
71d4bfbc9aSGreg Clayton         }
72d4bfbc9aSGreg Clayton 
73d4bfbc9aSGreg Clayton         if (create)
74d4bfbc9aSGreg Clayton         {
75d4bfbc9aSGreg Clayton             const llvm::Triple &triple_ref = process->GetTarget().GetArchitecture().GetTriple();
76d4bfbc9aSGreg Clayton             create = triple_ref.getOS() == llvm::Triple::Darwin && triple_ref.getVendor() == llvm::Triple::Apple;
77d4bfbc9aSGreg Clayton         }
78d4bfbc9aSGreg Clayton     }
79d4bfbc9aSGreg Clayton 
80d4bfbc9aSGreg Clayton     if (create)
81944b828aSGreg Clayton         return new DynamicLoaderDarwinKernel (process);
82d4bfbc9aSGreg Clayton     return NULL;
83d4bfbc9aSGreg Clayton }
84d4bfbc9aSGreg Clayton 
85d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
86d4bfbc9aSGreg Clayton // Constructor
87d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
88944b828aSGreg Clayton DynamicLoaderDarwinKernel::DynamicLoaderDarwinKernel (Process* process) :
89d4bfbc9aSGreg Clayton     DynamicLoader(process),
90d4bfbc9aSGreg Clayton     m_kernel(),
91d4bfbc9aSGreg Clayton     m_kext_summary_header_ptr_addr (),
92d4bfbc9aSGreg Clayton     m_kext_summary_header_addr (),
93d4bfbc9aSGreg Clayton     m_kext_summary_header (),
94d4bfbc9aSGreg Clayton     m_break_id (LLDB_INVALID_BREAK_ID),
95d4bfbc9aSGreg Clayton     m_kext_summaries(),
96d4bfbc9aSGreg Clayton     m_mutex(Mutex::eMutexTypeRecursive)
97d4bfbc9aSGreg Clayton {
98d4bfbc9aSGreg Clayton }
99d4bfbc9aSGreg Clayton 
100d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
101d4bfbc9aSGreg Clayton // Destructor
102d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
103944b828aSGreg Clayton DynamicLoaderDarwinKernel::~DynamicLoaderDarwinKernel()
104d4bfbc9aSGreg Clayton {
105d4bfbc9aSGreg Clayton     Clear(true);
106d4bfbc9aSGreg Clayton }
107d4bfbc9aSGreg Clayton 
108d4bfbc9aSGreg Clayton void
109944b828aSGreg Clayton DynamicLoaderDarwinKernel::UpdateIfNeeded()
110d4bfbc9aSGreg Clayton {
111d4bfbc9aSGreg Clayton     LoadKernelModuleIfNeeded();
112d4bfbc9aSGreg Clayton     SetNotificationBreakpointIfNeeded ();
113d4bfbc9aSGreg Clayton }
114d4bfbc9aSGreg Clayton //------------------------------------------------------------------
115d4bfbc9aSGreg Clayton /// Called after attaching a process.
116d4bfbc9aSGreg Clayton ///
117d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
118d4bfbc9aSGreg Clayton /// attaching to a process.
119d4bfbc9aSGreg Clayton //------------------------------------------------------------------
120d4bfbc9aSGreg Clayton void
121944b828aSGreg Clayton DynamicLoaderDarwinKernel::DidAttach ()
122d4bfbc9aSGreg Clayton {
123d4bfbc9aSGreg Clayton     PrivateInitialize(m_process);
124d4bfbc9aSGreg Clayton     UpdateIfNeeded();
125d4bfbc9aSGreg Clayton }
126d4bfbc9aSGreg Clayton 
127d4bfbc9aSGreg Clayton //------------------------------------------------------------------
128d4bfbc9aSGreg Clayton /// Called after attaching a process.
129d4bfbc9aSGreg Clayton ///
130d4bfbc9aSGreg Clayton /// Allow DynamicLoader plug-ins to execute some code after
131d4bfbc9aSGreg Clayton /// attaching to a process.
132d4bfbc9aSGreg Clayton //------------------------------------------------------------------
133d4bfbc9aSGreg Clayton void
134944b828aSGreg Clayton DynamicLoaderDarwinKernel::DidLaunch ()
135d4bfbc9aSGreg Clayton {
136d4bfbc9aSGreg Clayton     PrivateInitialize(m_process);
137d4bfbc9aSGreg Clayton     UpdateIfNeeded();
138d4bfbc9aSGreg Clayton }
139d4bfbc9aSGreg Clayton 
140d4bfbc9aSGreg Clayton 
141d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
142d4bfbc9aSGreg Clayton // Clear out the state of this class.
143d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
144d4bfbc9aSGreg Clayton void
145944b828aSGreg Clayton DynamicLoaderDarwinKernel::Clear (bool clear_process)
146d4bfbc9aSGreg Clayton {
147d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
148d4bfbc9aSGreg Clayton 
149d4bfbc9aSGreg Clayton     if (m_process->IsAlive() && LLDB_BREAK_ID_IS_VALID(m_break_id))
150d4bfbc9aSGreg Clayton         m_process->ClearBreakpointSiteByID(m_break_id);
151d4bfbc9aSGreg Clayton 
152d4bfbc9aSGreg Clayton     if (clear_process)
153d4bfbc9aSGreg Clayton         m_process = NULL;
154d4bfbc9aSGreg Clayton     m_kernel.Clear(false);
155d4bfbc9aSGreg Clayton     m_kext_summary_header_ptr_addr.Clear();
156d4bfbc9aSGreg Clayton     m_kext_summary_header_addr.Clear();
157d4bfbc9aSGreg Clayton     m_kext_summaries.clear();
158d4bfbc9aSGreg Clayton     m_break_id = LLDB_INVALID_BREAK_ID;
159d4bfbc9aSGreg Clayton }
160d4bfbc9aSGreg Clayton 
161d4bfbc9aSGreg Clayton 
162d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
163d4bfbc9aSGreg Clayton // Load the kernel module and initialize the "m_kernel" member. Return
164d4bfbc9aSGreg Clayton // true _only_ if the kernel is loaded the first time through (subsequent
165d4bfbc9aSGreg Clayton // calls to this function should return false after the kernel has been
166d4bfbc9aSGreg Clayton // already loaded).
167d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
168d4bfbc9aSGreg Clayton void
169944b828aSGreg Clayton DynamicLoaderDarwinKernel::LoadKernelModuleIfNeeded()
170d4bfbc9aSGreg Clayton {
171d4bfbc9aSGreg Clayton     if (!m_kext_summary_header_ptr_addr.IsValid())
172d4bfbc9aSGreg Clayton     {
173d4bfbc9aSGreg Clayton         m_kernel.Clear(false);
174d4bfbc9aSGreg Clayton         m_kernel.module_sp = m_process->GetTarget().GetExecutableModule();
175d4bfbc9aSGreg Clayton         if (m_kernel.module_sp)
176d4bfbc9aSGreg Clayton         {
177d4bfbc9aSGreg Clayton             static ConstString mach_header_name ("_mh_execute_header");
178d4bfbc9aSGreg Clayton             static ConstString kext_summary_symbol ("gLoadedKextSummaries");
179d4bfbc9aSGreg Clayton             const Symbol *symbol = NULL;
180d4bfbc9aSGreg Clayton             symbol = m_kernel.module_sp->FindFirstSymbolWithNameAndType (kext_summary_symbol, eSymbolTypeData);
181d4bfbc9aSGreg Clayton             if (symbol)
182d4bfbc9aSGreg Clayton                 m_kext_summary_header_ptr_addr = symbol->GetValue();
183d4bfbc9aSGreg Clayton 
184d4bfbc9aSGreg Clayton             symbol = m_kernel.module_sp->FindFirstSymbolWithNameAndType (mach_header_name, eSymbolTypeAbsolute);
185d4bfbc9aSGreg Clayton             if (symbol)
186d4bfbc9aSGreg Clayton             {
187d4bfbc9aSGreg Clayton                 // The "_mh_execute_header" symbol is absolute and not a section based
188d4bfbc9aSGreg Clayton                 // symbol that will have a valid address, so we need to resolve it...
189d4bfbc9aSGreg Clayton                 m_process->GetTarget().GetImages().ResolveFileAddress (symbol->GetValue().GetFileAddress(), m_kernel.so_address);
190d4bfbc9aSGreg Clayton                 DataExtractor data; // Load command data
191d4bfbc9aSGreg Clayton                 if (ReadMachHeader (m_kernel, &data))
192d4bfbc9aSGreg Clayton                 {
193d4bfbc9aSGreg Clayton                     if (m_kernel.header.filetype == llvm::MachO::HeaderFileTypeExecutable)
194d4bfbc9aSGreg Clayton                     {
195d4bfbc9aSGreg Clayton                         if (ParseLoadCommands (data, m_kernel))
196d4bfbc9aSGreg Clayton                             UpdateImageLoadAddress (m_kernel);
197d4bfbc9aSGreg Clayton 
198d4bfbc9aSGreg Clayton                         // Update all image infos
199d4bfbc9aSGreg Clayton                         ReadAllKextSummaries ();
200d4bfbc9aSGreg Clayton                     }
201d4bfbc9aSGreg Clayton                 }
202d4bfbc9aSGreg Clayton                 else
203d4bfbc9aSGreg Clayton                 {
204d4bfbc9aSGreg Clayton                     m_kernel.Clear(false);
205d4bfbc9aSGreg Clayton                 }
206d4bfbc9aSGreg Clayton             }
207d4bfbc9aSGreg Clayton         }
208d4bfbc9aSGreg Clayton     }
209d4bfbc9aSGreg Clayton }
210d4bfbc9aSGreg Clayton 
211d4bfbc9aSGreg Clayton bool
212944b828aSGreg Clayton DynamicLoaderDarwinKernel::FindTargetModule (OSKextLoadedKextSummary &image_info, bool can_create, bool *did_create_ptr)
213d4bfbc9aSGreg Clayton {
214d4bfbc9aSGreg Clayton     if (did_create_ptr)
215d4bfbc9aSGreg Clayton         *did_create_ptr = false;
216d4bfbc9aSGreg Clayton 
217d4bfbc9aSGreg Clayton     const bool image_info_uuid_is_valid = image_info.uuid.IsValid();
218d4bfbc9aSGreg Clayton 
219d4bfbc9aSGreg Clayton     if (image_info.module_sp)
220d4bfbc9aSGreg Clayton     {
221d4bfbc9aSGreg Clayton         if (image_info_uuid_is_valid)
222d4bfbc9aSGreg Clayton         {
223d4bfbc9aSGreg Clayton             if (image_info.module_sp->GetUUID() == image_info.uuid)
224d4bfbc9aSGreg Clayton                 return true;
225d4bfbc9aSGreg Clayton             else
226d4bfbc9aSGreg Clayton                 image_info.module_sp.reset();
227d4bfbc9aSGreg Clayton         }
228d4bfbc9aSGreg Clayton         else
229d4bfbc9aSGreg Clayton             return true;
230d4bfbc9aSGreg Clayton     }
231d4bfbc9aSGreg Clayton 
232d4bfbc9aSGreg Clayton     ModuleList &target_images = m_process->GetTarget().GetImages();
233d4bfbc9aSGreg Clayton     if (image_info_uuid_is_valid)
234d4bfbc9aSGreg Clayton         image_info.module_sp = target_images.FindModule(image_info.uuid);
235d4bfbc9aSGreg Clayton 
236d4bfbc9aSGreg Clayton     if (image_info.module_sp)
237d4bfbc9aSGreg Clayton         return true;
238d4bfbc9aSGreg Clayton 
239d4bfbc9aSGreg Clayton     ArchSpec arch (image_info.GetArchitecture ());
240d4bfbc9aSGreg Clayton     if (can_create)
241d4bfbc9aSGreg Clayton     {
242d4bfbc9aSGreg Clayton         if (image_info_uuid_is_valid)
243d4bfbc9aSGreg Clayton         {
244d4bfbc9aSGreg Clayton             image_info.module_sp = m_process->GetTarget().GetSharedModule (FileSpec(),
245d4bfbc9aSGreg Clayton                                                                            arch,
246d4bfbc9aSGreg Clayton                                                                            &image_info.uuid);
247d4bfbc9aSGreg Clayton             if (did_create_ptr)
248d4bfbc9aSGreg Clayton                 *did_create_ptr = image_info.module_sp;
249d4bfbc9aSGreg Clayton         }
250d4bfbc9aSGreg Clayton     }
251d4bfbc9aSGreg Clayton     return image_info.module_sp;
252d4bfbc9aSGreg Clayton }
253d4bfbc9aSGreg Clayton 
254d4bfbc9aSGreg Clayton bool
255944b828aSGreg Clayton DynamicLoaderDarwinKernel::UpdateCommPageLoadAddress(Module *module)
256d4bfbc9aSGreg Clayton {
257d4bfbc9aSGreg Clayton     bool changed = false;
258d4bfbc9aSGreg Clayton     if (module)
259d4bfbc9aSGreg Clayton     {
260d4bfbc9aSGreg Clayton         ObjectFile *image_object_file = module->GetObjectFile();
261d4bfbc9aSGreg Clayton         if (image_object_file)
262d4bfbc9aSGreg Clayton         {
263d4bfbc9aSGreg Clayton             SectionList *section_list = image_object_file->GetSectionList ();
264d4bfbc9aSGreg Clayton             if (section_list)
265d4bfbc9aSGreg Clayton             {
266d4bfbc9aSGreg Clayton                 uint32_t num_sections = section_list->GetSize();
267d4bfbc9aSGreg Clayton                 for (uint32_t i=0; i<num_sections; ++i)
268d4bfbc9aSGreg Clayton                 {
269d4bfbc9aSGreg Clayton                     Section* section = section_list->GetSectionAtIndex (i).get();
270d4bfbc9aSGreg Clayton                     if (section)
271d4bfbc9aSGreg Clayton                     {
272d4bfbc9aSGreg Clayton                         const addr_t new_section_load_addr = section->GetFileAddress ();
273d4bfbc9aSGreg Clayton                         const addr_t old_section_load_addr = m_process->GetTarget().GetSectionLoadList().GetSectionLoadAddress (section);
274d4bfbc9aSGreg Clayton                         if (old_section_load_addr == LLDB_INVALID_ADDRESS ||
275d4bfbc9aSGreg Clayton                             old_section_load_addr != new_section_load_addr)
276d4bfbc9aSGreg Clayton                         {
277d4bfbc9aSGreg Clayton                             if (m_process->GetTarget().GetSectionLoadList().SetSectionLoadAddress (section, section->GetFileAddress ()))
278d4bfbc9aSGreg Clayton                                 changed = true;
279d4bfbc9aSGreg Clayton                         }
280d4bfbc9aSGreg Clayton                     }
281d4bfbc9aSGreg Clayton                 }
282d4bfbc9aSGreg Clayton             }
283d4bfbc9aSGreg Clayton         }
284d4bfbc9aSGreg Clayton     }
285d4bfbc9aSGreg Clayton     return changed;
286d4bfbc9aSGreg Clayton }
287d4bfbc9aSGreg Clayton 
288d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
289d4bfbc9aSGreg Clayton // Update the load addresses for all segments in MODULE using the
290d4bfbc9aSGreg Clayton // updated INFO that is passed in.
291d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
292d4bfbc9aSGreg Clayton bool
293944b828aSGreg Clayton DynamicLoaderDarwinKernel::UpdateImageLoadAddress (OSKextLoadedKextSummary& info)
294d4bfbc9aSGreg Clayton {
295d4bfbc9aSGreg Clayton     Module *module = info.module_sp.get();
296d4bfbc9aSGreg Clayton     bool changed = false;
297d4bfbc9aSGreg Clayton     if (module)
298d4bfbc9aSGreg Clayton     {
299d4bfbc9aSGreg Clayton         ObjectFile *image_object_file = module->GetObjectFile();
300d4bfbc9aSGreg Clayton         if (image_object_file)
301d4bfbc9aSGreg Clayton         {
302d4bfbc9aSGreg Clayton             SectionList *section_list = image_object_file->GetSectionList ();
303d4bfbc9aSGreg Clayton             if (section_list)
304d4bfbc9aSGreg Clayton             {
305d4bfbc9aSGreg Clayton                 // We now know the slide amount, so go through all sections
306d4bfbc9aSGreg Clayton                 // and update the load addresses with the correct values.
307d4bfbc9aSGreg Clayton                 uint32_t num_segments = info.segments.size();
308d4bfbc9aSGreg Clayton                 for (uint32_t i=0; i<num_segments; ++i)
309d4bfbc9aSGreg Clayton                 {
310d4bfbc9aSGreg Clayton                     const addr_t new_section_load_addr = info.segments[i].vmaddr;
311d4bfbc9aSGreg Clayton                     if (section_list->FindSectionByName(info.segments[i].name))
312d4bfbc9aSGreg Clayton                     {
313d4bfbc9aSGreg Clayton                         SectionSP section_sp(section_list->FindSectionByName(info.segments[i].name));
314d4bfbc9aSGreg Clayton                         if (section_sp)
315d4bfbc9aSGreg Clayton                         {
316d4bfbc9aSGreg Clayton                             const addr_t old_section_load_addr = m_process->GetTarget().GetSectionLoadList().GetSectionLoadAddress (section_sp.get());
317d4bfbc9aSGreg Clayton                             if (old_section_load_addr == LLDB_INVALID_ADDRESS ||
318d4bfbc9aSGreg Clayton                                 old_section_load_addr != new_section_load_addr)
319d4bfbc9aSGreg Clayton                             {
320d4bfbc9aSGreg Clayton                                 if (m_process->GetTarget().GetSectionLoadList().SetSectionLoadAddress (section_sp.get(), new_section_load_addr))
321d4bfbc9aSGreg Clayton                                     changed = true;
322d4bfbc9aSGreg Clayton                             }
323d4bfbc9aSGreg Clayton                         }
324d4bfbc9aSGreg Clayton                         else
325d4bfbc9aSGreg Clayton                         {
326d4bfbc9aSGreg Clayton                             fprintf (stderr,
327d4bfbc9aSGreg Clayton                                      "warning: unable to find and load segment named '%s' at 0x%llx in '%s/%s' in macosx dynamic loader plug-in.\n",
328d4bfbc9aSGreg Clayton                                      info.segments[i].name.AsCString("<invalid>"),
329d4bfbc9aSGreg Clayton                                      (uint64_t)new_section_load_addr,
330d4bfbc9aSGreg Clayton                                      image_object_file->GetFileSpec().GetDirectory().AsCString(),
331d4bfbc9aSGreg Clayton                                      image_object_file->GetFileSpec().GetFilename().AsCString());
332d4bfbc9aSGreg Clayton                         }
333d4bfbc9aSGreg Clayton                     }
334d4bfbc9aSGreg Clayton                     else
335d4bfbc9aSGreg Clayton                     {
336d4bfbc9aSGreg Clayton                         // The segment name is empty which means this is a .o file.
337d4bfbc9aSGreg Clayton                         // Object files in LLDB end up getting reorganized so that
338d4bfbc9aSGreg Clayton                         // the segment name that is in the section is promoted into
339d4bfbc9aSGreg Clayton                         // an actual segment, so we just need to go through all sections
340d4bfbc9aSGreg Clayton                         // and slide them by a single amount.
341d4bfbc9aSGreg Clayton 
342d4bfbc9aSGreg Clayton                         uint32_t num_sections = section_list->GetSize();
343d4bfbc9aSGreg Clayton                         for (uint32_t i=0; i<num_sections; ++i)
344d4bfbc9aSGreg Clayton                         {
345d4bfbc9aSGreg Clayton                             Section* section = section_list->GetSectionAtIndex (i).get();
346d4bfbc9aSGreg Clayton                             if (section)
347d4bfbc9aSGreg Clayton                             {
348d4bfbc9aSGreg Clayton                                 if (m_process->GetTarget().GetSectionLoadList().SetSectionLoadAddress (section, section->GetFileAddress() + new_section_load_addr))
349d4bfbc9aSGreg Clayton                                     changed = true;
350d4bfbc9aSGreg Clayton                             }
351d4bfbc9aSGreg Clayton                         }
352d4bfbc9aSGreg Clayton                     }
353d4bfbc9aSGreg Clayton                 }
354d4bfbc9aSGreg Clayton             }
355d4bfbc9aSGreg Clayton         }
356d4bfbc9aSGreg Clayton     }
357d4bfbc9aSGreg Clayton     return changed;
358d4bfbc9aSGreg Clayton }
359d4bfbc9aSGreg Clayton 
360d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
361d4bfbc9aSGreg Clayton // Update the load addresses for all segments in MODULE using the
362d4bfbc9aSGreg Clayton // updated INFO that is passed in.
363d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
364d4bfbc9aSGreg Clayton bool
365944b828aSGreg Clayton DynamicLoaderDarwinKernel::UnloadImageLoadAddress (OSKextLoadedKextSummary& info)
366d4bfbc9aSGreg Clayton {
367d4bfbc9aSGreg Clayton     Module *module = info.module_sp.get();
368d4bfbc9aSGreg Clayton     bool changed = false;
369d4bfbc9aSGreg Clayton     if (module)
370d4bfbc9aSGreg Clayton     {
371d4bfbc9aSGreg Clayton         ObjectFile *image_object_file = module->GetObjectFile();
372d4bfbc9aSGreg Clayton         if (image_object_file)
373d4bfbc9aSGreg Clayton         {
374d4bfbc9aSGreg Clayton             SectionList *section_list = image_object_file->GetSectionList ();
375d4bfbc9aSGreg Clayton             if (section_list)
376d4bfbc9aSGreg Clayton             {
377d4bfbc9aSGreg Clayton                 uint32_t num_segments = info.segments.size();
378d4bfbc9aSGreg Clayton                 for (uint32_t i=0; i<num_segments; ++i)
379d4bfbc9aSGreg Clayton                 {
380d4bfbc9aSGreg Clayton                     SectionSP section_sp(section_list->FindSectionByName(info.segments[i].name));
381d4bfbc9aSGreg Clayton                     if (section_sp)
382d4bfbc9aSGreg Clayton                     {
383d4bfbc9aSGreg Clayton                         const addr_t old_section_load_addr = info.segments[i].vmaddr;
384d4bfbc9aSGreg Clayton                         if (m_process->GetTarget().GetSectionLoadList().SetSectionUnloaded (section_sp.get(), old_section_load_addr))
385d4bfbc9aSGreg Clayton                             changed = true;
386d4bfbc9aSGreg Clayton                     }
387d4bfbc9aSGreg Clayton                     else
388d4bfbc9aSGreg Clayton                     {
389d4bfbc9aSGreg Clayton                         fprintf (stderr,
390d4bfbc9aSGreg Clayton                                  "warning: unable to find and unload segment named '%s' in '%s/%s' in macosx dynamic loader plug-in.\n",
391d4bfbc9aSGreg Clayton                                  info.segments[i].name.AsCString("<invalid>"),
392d4bfbc9aSGreg Clayton                                  image_object_file->GetFileSpec().GetDirectory().AsCString(),
393d4bfbc9aSGreg Clayton                                  image_object_file->GetFileSpec().GetFilename().AsCString());
394d4bfbc9aSGreg Clayton                     }
395d4bfbc9aSGreg Clayton                 }
396d4bfbc9aSGreg Clayton             }
397d4bfbc9aSGreg Clayton         }
398d4bfbc9aSGreg Clayton     }
399d4bfbc9aSGreg Clayton     return changed;
400d4bfbc9aSGreg Clayton }
401d4bfbc9aSGreg Clayton 
402d4bfbc9aSGreg Clayton 
403d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
404d4bfbc9aSGreg Clayton // Static callback function that gets called when our DYLD notification
405d4bfbc9aSGreg Clayton // breakpoint gets hit. We update all of our image infos and then
406d4bfbc9aSGreg Clayton // let our super class DynamicLoader class decide if we should stop
407d4bfbc9aSGreg Clayton // or not (based on global preference).
408d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
409d4bfbc9aSGreg Clayton bool
410944b828aSGreg Clayton DynamicLoaderDarwinKernel::BreakpointHitCallback (void *baton,
411d4bfbc9aSGreg Clayton                                                   StoppointCallbackContext *context,
412d4bfbc9aSGreg Clayton                                                   user_id_t break_id,
413d4bfbc9aSGreg Clayton                                                   user_id_t break_loc_id)
414d4bfbc9aSGreg Clayton {
415944b828aSGreg Clayton     return static_cast<DynamicLoaderDarwinKernel*>(baton)->BreakpointHit (context, break_id, break_loc_id);
416d4bfbc9aSGreg Clayton }
417d4bfbc9aSGreg Clayton 
418d4bfbc9aSGreg Clayton bool
419944b828aSGreg Clayton DynamicLoaderDarwinKernel::BreakpointHit (StoppointCallbackContext *context,
420d4bfbc9aSGreg Clayton                                           user_id_t break_id,
421d4bfbc9aSGreg Clayton                                           user_id_t break_loc_id)
422d4bfbc9aSGreg Clayton {
423d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAnyCategoriesSet (LIBLLDB_LOG_DYNAMIC_LOADER));
424d4bfbc9aSGreg Clayton     if (log)
425944b828aSGreg Clayton         log->Printf ("DynamicLoaderDarwinKernel::BreakpointHit (...)\n");
426d4bfbc9aSGreg Clayton 
427d4bfbc9aSGreg Clayton     ReadAllKextSummaries ();
428d4bfbc9aSGreg Clayton 
429d4bfbc9aSGreg Clayton     if (log)
430d4bfbc9aSGreg Clayton         PutToLog(log.get());
431d4bfbc9aSGreg Clayton 
432d4bfbc9aSGreg Clayton     return GetStopWhenImagesChange();
433d4bfbc9aSGreg Clayton }
434d4bfbc9aSGreg Clayton 
435d4bfbc9aSGreg Clayton 
436d4bfbc9aSGreg Clayton bool
437944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadKextSummaryHeader ()
438d4bfbc9aSGreg Clayton {
439d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
440d4bfbc9aSGreg Clayton 
441d4bfbc9aSGreg Clayton     // the all image infos is already valid for this process stop ID
442d4bfbc9aSGreg Clayton 
443d4bfbc9aSGreg Clayton     m_kext_summaries.clear();
444d4bfbc9aSGreg Clayton     if (m_kext_summary_header_ptr_addr.IsValid())
445d4bfbc9aSGreg Clayton     {
446d4bfbc9aSGreg Clayton         const uint32_t addr_size = m_kernel.GetAddressByteSize ();
447d4bfbc9aSGreg Clayton         const ByteOrder byte_order = m_kernel.GetByteOrder();
448d4bfbc9aSGreg Clayton         Error error;
449d4bfbc9aSGreg Clayton         // Read enough bytes for a "OSKextLoadedKextSummaryHeader" structure
450d4bfbc9aSGreg Clayton         // which is currenty 4 uint32_t and a pointer.
451d4bfbc9aSGreg Clayton         uint8_t buf[24];
452d4bfbc9aSGreg Clayton         DataExtractor data (buf, sizeof(buf), byte_order, addr_size);
453d4bfbc9aSGreg Clayton         const size_t count = 4 * sizeof(uint32_t) + addr_size;
454d4bfbc9aSGreg Clayton         const bool prefer_file_cache = false;
455d4bfbc9aSGreg Clayton         if (m_process->GetTarget().ReadPointerFromMemory (m_kext_summary_header_ptr_addr,
456d4bfbc9aSGreg Clayton                                                           prefer_file_cache,
457d4bfbc9aSGreg Clayton                                                           error,
458d4bfbc9aSGreg Clayton                                                           m_kext_summary_header_addr))
459d4bfbc9aSGreg Clayton         {
460d4bfbc9aSGreg Clayton             // We got a valid address for our kext summary header and make sure it isn't NULL
461d4bfbc9aSGreg Clayton             if (m_kext_summary_header_addr.IsValid() &&
462d4bfbc9aSGreg Clayton                 m_kext_summary_header_addr.GetFileAddress() != 0)
463d4bfbc9aSGreg Clayton             {
464d4bfbc9aSGreg Clayton                 const size_t bytes_read = m_process->GetTarget().ReadMemory (m_kext_summary_header_addr, prefer_file_cache, buf, count, error);
465d4bfbc9aSGreg Clayton                 if (bytes_read == count)
466d4bfbc9aSGreg Clayton                 {
467d4bfbc9aSGreg Clayton                     uint32_t offset = 0;
468d4bfbc9aSGreg Clayton                     m_kext_summary_header.version = data.GetU32(&offset);
469d4bfbc9aSGreg Clayton                     if (m_kext_summary_header.version >= 2)
470d4bfbc9aSGreg Clayton                     {
471d4bfbc9aSGreg Clayton                         m_kext_summary_header.entry_size = data.GetU32(&offset);
472d4bfbc9aSGreg Clayton                     }
473d4bfbc9aSGreg Clayton                     else
474d4bfbc9aSGreg Clayton                     {
475d4bfbc9aSGreg Clayton                         // Versions less than 2 didn't have an entry size, it was hard coded
476d4bfbc9aSGreg Clayton                         m_kext_summary_header.entry_size = KERNEL_MODULE_ENTRY_SIZE_VERSION_1;
477d4bfbc9aSGreg Clayton                     }
478d4bfbc9aSGreg Clayton                     m_kext_summary_header.entry_count = data.GetU32(&offset);
479d4bfbc9aSGreg Clayton                     return true;
480d4bfbc9aSGreg Clayton                 }
481d4bfbc9aSGreg Clayton             }
482d4bfbc9aSGreg Clayton         }
483d4bfbc9aSGreg Clayton     }
484d4bfbc9aSGreg Clayton     m_kext_summary_header_addr.Clear();
485d4bfbc9aSGreg Clayton     return false;
486d4bfbc9aSGreg Clayton }
487d4bfbc9aSGreg Clayton 
488d4bfbc9aSGreg Clayton 
489d4bfbc9aSGreg Clayton bool
490944b828aSGreg Clayton DynamicLoaderDarwinKernel::ParseKextSummaries (const Address &kext_summary_addr,
491d4bfbc9aSGreg Clayton                                                uint32_t count)
492d4bfbc9aSGreg Clayton {
493d4bfbc9aSGreg Clayton     OSKextLoadedKextSummary::collection kext_summaries;
494d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAnyCategoriesSet (LIBLLDB_LOG_DYNAMIC_LOADER));
495d4bfbc9aSGreg Clayton     if (log)
496*fd54b368SJason Molenda         log->Printf ("Adding %d modules.\n", count);
497d4bfbc9aSGreg Clayton 
498d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
499d4bfbc9aSGreg Clayton 
500d4bfbc9aSGreg Clayton     if (!ReadKextSummaries (kext_summary_addr, count, kext_summaries))
501d4bfbc9aSGreg Clayton         return false;
502d4bfbc9aSGreg Clayton 
503d4bfbc9aSGreg Clayton     Stream *s = &m_process->GetTarget().GetDebugger().GetOutputStream();
504d4bfbc9aSGreg Clayton     for (uint32_t i = 0; i < count; i++)
505d4bfbc9aSGreg Clayton     {
506d4bfbc9aSGreg Clayton         if (s)
507d4bfbc9aSGreg Clayton         {
508d4bfbc9aSGreg Clayton             const uint8_t *u = (const uint8_t *)kext_summaries[i].uuid.GetBytes();
509d4bfbc9aSGreg Clayton             if (u)
510d4bfbc9aSGreg Clayton             {
511d4bfbc9aSGreg Clayton                 s->Printf("Loading kext: %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 0x%16.16llx \"%s\"...\n",
512d4bfbc9aSGreg Clayton                           u[ 0], u[ 1], u[ 2], u[ 3], u[ 4], u[ 5], u[ 6], u[ 7],
513d4bfbc9aSGreg Clayton                           u[ 8], u[ 9], u[10], u[11], u[12], u[13], u[14], u[15],
514d4bfbc9aSGreg Clayton                           kext_summaries[i].address, kext_summaries[i].name);
515d4bfbc9aSGreg Clayton             }
516d4bfbc9aSGreg Clayton             else
517d4bfbc9aSGreg Clayton             {
518d4bfbc9aSGreg Clayton                 s->Printf("0x%16.16llx \"%s\"...\n", kext_summaries[i].address, kext_summaries[i].name);
519d4bfbc9aSGreg Clayton             }
520d4bfbc9aSGreg Clayton         }
521d4bfbc9aSGreg Clayton 
522d4bfbc9aSGreg Clayton         DataExtractor data; // Load command data
523d4bfbc9aSGreg Clayton         if (ReadMachHeader (kext_summaries[i], &data))
524d4bfbc9aSGreg Clayton         {
525d4bfbc9aSGreg Clayton             ParseLoadCommands (data, kext_summaries[i]);
526d4bfbc9aSGreg Clayton         }
527d4bfbc9aSGreg Clayton 
528d4bfbc9aSGreg Clayton         if (s)
529d4bfbc9aSGreg Clayton         {
530d4bfbc9aSGreg Clayton             if (kext_summaries[i].module_sp)
531d4bfbc9aSGreg Clayton                 s->Printf("  found kext: %s/%s\n",
532d4bfbc9aSGreg Clayton                           kext_summaries[i].module_sp->GetFileSpec().GetDirectory().AsCString(),
533d4bfbc9aSGreg Clayton                           kext_summaries[i].module_sp->GetFileSpec().GetFilename().AsCString());
534d4bfbc9aSGreg Clayton         }
535d4bfbc9aSGreg Clayton 
536d4bfbc9aSGreg Clayton         if (log)
537d4bfbc9aSGreg Clayton             kext_summaries[i].PutToLog (log.get());
538d4bfbc9aSGreg Clayton     }
539d4bfbc9aSGreg Clayton     bool return_value = AddModulesUsingImageInfos (kext_summaries);
540d4bfbc9aSGreg Clayton     return return_value;
541d4bfbc9aSGreg Clayton }
542d4bfbc9aSGreg Clayton 
543d4bfbc9aSGreg Clayton // Adds the modules in image_infos to m_kext_summaries.
544d4bfbc9aSGreg Clayton // NB don't call this passing in m_kext_summaries.
545d4bfbc9aSGreg Clayton 
546d4bfbc9aSGreg Clayton bool
547944b828aSGreg Clayton DynamicLoaderDarwinKernel::AddModulesUsingImageInfos (OSKextLoadedKextSummary::collection &image_infos)
548d4bfbc9aSGreg Clayton {
549d4bfbc9aSGreg Clayton     // Now add these images to the main list.
550d4bfbc9aSGreg Clayton     ModuleList loaded_module_list;
551d4bfbc9aSGreg Clayton 
552d4bfbc9aSGreg Clayton     for (uint32_t idx = 0; idx < image_infos.size(); ++idx)
553d4bfbc9aSGreg Clayton     {
554d4bfbc9aSGreg Clayton         m_kext_summaries.push_back(image_infos[idx]);
555d4bfbc9aSGreg Clayton 
556d4bfbc9aSGreg Clayton         if (FindTargetModule (image_infos[idx], true, NULL))
557d4bfbc9aSGreg Clayton         {
558d4bfbc9aSGreg Clayton             // UpdateImageLoadAddress will return true if any segments
559d4bfbc9aSGreg Clayton             // change load address. We need to check this so we don't
560d4bfbc9aSGreg Clayton             // mention that all loaded shared libraries are newly loaded
561d4bfbc9aSGreg Clayton             // each time we hit out dyld breakpoint since dyld will list all
562d4bfbc9aSGreg Clayton             // shared libraries each time.
563d4bfbc9aSGreg Clayton             if (UpdateImageLoadAddress (image_infos[idx]))
564d4bfbc9aSGreg Clayton             {
565d4bfbc9aSGreg Clayton                 loaded_module_list.AppendIfNeeded (image_infos[idx].module_sp);
566d4bfbc9aSGreg Clayton             }
567d4bfbc9aSGreg Clayton         }
568d4bfbc9aSGreg Clayton     }
569d4bfbc9aSGreg Clayton 
570d4bfbc9aSGreg Clayton     if (loaded_module_list.GetSize() > 0)
571d4bfbc9aSGreg Clayton     {
572d4bfbc9aSGreg Clayton         // FIXME: This should really be in the Runtime handlers class, which should get
573d4bfbc9aSGreg Clayton         // called by the target's ModulesDidLoad, but we're doing it all locally for now
574d4bfbc9aSGreg Clayton         // to save time.
575d4bfbc9aSGreg Clayton         // Also, I'm assuming there can be only one libobjc dylib loaded...
576d4bfbc9aSGreg Clayton 
577d4bfbc9aSGreg Clayton         ObjCLanguageRuntime *objc_runtime = m_process->GetObjCLanguageRuntime();
578d4bfbc9aSGreg Clayton         if (objc_runtime != NULL && !objc_runtime->HasReadObjCLibrary())
579d4bfbc9aSGreg Clayton         {
580d4bfbc9aSGreg Clayton             size_t num_modules = loaded_module_list.GetSize();
581d4bfbc9aSGreg Clayton             for (int i = 0; i < num_modules; i++)
582d4bfbc9aSGreg Clayton             {
583d4bfbc9aSGreg Clayton                 if (objc_runtime->IsModuleObjCLibrary (loaded_module_list.GetModuleAtIndex (i)))
584d4bfbc9aSGreg Clayton                 {
585d4bfbc9aSGreg Clayton                     objc_runtime->ReadObjCLibrary (loaded_module_list.GetModuleAtIndex (i));
586d4bfbc9aSGreg Clayton                     break;
587d4bfbc9aSGreg Clayton                 }
588d4bfbc9aSGreg Clayton             }
589d4bfbc9aSGreg Clayton         }
590d4bfbc9aSGreg Clayton //        if (log)
591944b828aSGreg Clayton //            loaded_module_list.LogUUIDAndPaths (log, "DynamicLoaderDarwinKernel::ModulesDidLoad");
592d4bfbc9aSGreg Clayton         m_process->GetTarget().ModulesDidLoad (loaded_module_list);
593d4bfbc9aSGreg Clayton     }
594d4bfbc9aSGreg Clayton     return true;
595d4bfbc9aSGreg Clayton }
596d4bfbc9aSGreg Clayton 
597d4bfbc9aSGreg Clayton 
598d4bfbc9aSGreg Clayton uint32_t
599944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadKextSummaries (const Address &kext_summary_addr,
600d4bfbc9aSGreg Clayton                                               uint32_t image_infos_count,
601d4bfbc9aSGreg Clayton                                               OSKextLoadedKextSummary::collection &image_infos)
602d4bfbc9aSGreg Clayton {
603d4bfbc9aSGreg Clayton     const ByteOrder endian = m_kernel.GetByteOrder();
604d4bfbc9aSGreg Clayton     const uint32_t addr_size = m_kernel.GetAddressByteSize();
605d4bfbc9aSGreg Clayton 
606d4bfbc9aSGreg Clayton     image_infos.resize(image_infos_count);
607d4bfbc9aSGreg Clayton     const size_t count = image_infos.size() * m_kext_summary_header.entry_size;
608d4bfbc9aSGreg Clayton     DataBufferHeap data(count, 0);
609d4bfbc9aSGreg Clayton     Error error;
610d4bfbc9aSGreg Clayton 
611d4bfbc9aSGreg Clayton     Stream *s = &m_process->GetTarget().GetDebugger().GetOutputStream();
612d4bfbc9aSGreg Clayton 
613d4bfbc9aSGreg Clayton     if (s)
614d4bfbc9aSGreg Clayton         s->Printf ("Reading %u kext summaries...\n", image_infos_count);
615d4bfbc9aSGreg Clayton     const bool prefer_file_cache = false;
616d4bfbc9aSGreg Clayton     const size_t bytes_read = m_process->GetTarget().ReadMemory (kext_summary_addr,
617d4bfbc9aSGreg Clayton                                                                  prefer_file_cache,
618d4bfbc9aSGreg Clayton                                                                  data.GetBytes(),
619d4bfbc9aSGreg Clayton                                                                  data.GetByteSize(),
620d4bfbc9aSGreg Clayton                                                                  error);
621d4bfbc9aSGreg Clayton     if (bytes_read == count)
622d4bfbc9aSGreg Clayton     {
623d4bfbc9aSGreg Clayton 
624d4bfbc9aSGreg Clayton         DataExtractor extractor (data.GetBytes(), data.GetByteSize(), endian, addr_size);
625d4bfbc9aSGreg Clayton         uint32_t i=0;
626d4bfbc9aSGreg Clayton         for (uint32_t kext_summary_offset = 0;
627d4bfbc9aSGreg Clayton              i < image_infos.size() && extractor.ValidOffsetForDataOfSize(kext_summary_offset, m_kext_summary_header.entry_size);
628d4bfbc9aSGreg Clayton              ++i, kext_summary_offset += m_kext_summary_header.entry_size)
629d4bfbc9aSGreg Clayton         {
630d4bfbc9aSGreg Clayton             uint32_t offset = kext_summary_offset;
631d4bfbc9aSGreg Clayton             const void *name_data = extractor.GetData(&offset, KERNEL_MODULE_MAX_NAME);
632d4bfbc9aSGreg Clayton             if (name_data == NULL)
633d4bfbc9aSGreg Clayton                 break;
634d4bfbc9aSGreg Clayton             memcpy (image_infos[i].name, name_data, KERNEL_MODULE_MAX_NAME);
635d4bfbc9aSGreg Clayton             image_infos[i].uuid.SetBytes(extractor.GetData (&offset, 16));
636d4bfbc9aSGreg Clayton             image_infos[i].address          = extractor.GetU64(&offset);
637d4bfbc9aSGreg Clayton             if (!image_infos[i].so_address.SetLoadAddress (image_infos[i].address, &m_process->GetTarget()))
638d4bfbc9aSGreg Clayton                 m_process->GetTarget().GetImages().ResolveFileAddress (image_infos[i].address, image_infos[i].so_address);
639d4bfbc9aSGreg Clayton             image_infos[i].size             = extractor.GetU64(&offset);
640d4bfbc9aSGreg Clayton             image_infos[i].version          = extractor.GetU64(&offset);
641d4bfbc9aSGreg Clayton             image_infos[i].load_tag         = extractor.GetU32(&offset);
642d4bfbc9aSGreg Clayton             image_infos[i].flags            = extractor.GetU32(&offset);
643d4bfbc9aSGreg Clayton             if ((offset - kext_summary_offset) < m_kext_summary_header.entry_size)
644d4bfbc9aSGreg Clayton             {
645d4bfbc9aSGreg Clayton                 image_infos[i].reference_list = extractor.GetU64(&offset);
646d4bfbc9aSGreg Clayton             }
647d4bfbc9aSGreg Clayton             else
648d4bfbc9aSGreg Clayton             {
649d4bfbc9aSGreg Clayton                 image_infos[i].reference_list = 0;
650d4bfbc9aSGreg Clayton             }
651d4bfbc9aSGreg Clayton         }
652d4bfbc9aSGreg Clayton         if (i < image_infos.size())
653d4bfbc9aSGreg Clayton             image_infos.resize(i);
654d4bfbc9aSGreg Clayton     }
655d4bfbc9aSGreg Clayton     else
656d4bfbc9aSGreg Clayton     {
657d4bfbc9aSGreg Clayton         image_infos.clear();
658d4bfbc9aSGreg Clayton     }
659d4bfbc9aSGreg Clayton     return image_infos.size();
660d4bfbc9aSGreg Clayton }
661d4bfbc9aSGreg Clayton 
662d4bfbc9aSGreg Clayton bool
663944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadAllKextSummaries ()
664d4bfbc9aSGreg Clayton {
665d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAnyCategoriesSet (LIBLLDB_LOG_DYNAMIC_LOADER));
666d4bfbc9aSGreg Clayton 
667d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
668d4bfbc9aSGreg Clayton 
669d4bfbc9aSGreg Clayton     if (ReadKextSummaryHeader ())
670d4bfbc9aSGreg Clayton     {
671d4bfbc9aSGreg Clayton         if (m_kext_summary_header.entry_count > 0 && m_kext_summary_header_addr.IsValid())
672d4bfbc9aSGreg Clayton         {
673d4bfbc9aSGreg Clayton             Address summary_addr (m_kext_summary_header_addr);
674d4bfbc9aSGreg Clayton             summary_addr.Slide(m_kext_summary_header.GetSize());
675d4bfbc9aSGreg Clayton             if (!ParseKextSummaries (summary_addr, m_kext_summary_header.entry_count))
676d4bfbc9aSGreg Clayton             {
677d4bfbc9aSGreg Clayton                 m_kext_summaries.clear();
678d4bfbc9aSGreg Clayton             }
679d4bfbc9aSGreg Clayton             return true;
680d4bfbc9aSGreg Clayton         }
681d4bfbc9aSGreg Clayton     }
682d4bfbc9aSGreg Clayton     return false;
683d4bfbc9aSGreg Clayton }
684d4bfbc9aSGreg Clayton 
685d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
686d4bfbc9aSGreg Clayton // Read a mach_header at ADDR into HEADER, and also fill in the load
687d4bfbc9aSGreg Clayton // command data into LOAD_COMMAND_DATA if it is non-NULL.
688d4bfbc9aSGreg Clayton //
689d4bfbc9aSGreg Clayton // Returns true if we succeed, false if we fail for any reason.
690d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
691d4bfbc9aSGreg Clayton bool
692944b828aSGreg Clayton DynamicLoaderDarwinKernel::ReadMachHeader (OSKextLoadedKextSummary& kext_summary, DataExtractor *load_command_data)
693d4bfbc9aSGreg Clayton {
694d4bfbc9aSGreg Clayton     DataBufferHeap header_bytes(sizeof(llvm::MachO::mach_header), 0);
695d4bfbc9aSGreg Clayton     Error error;
696d4bfbc9aSGreg Clayton     const bool prefer_file_cache = false;
697d4bfbc9aSGreg Clayton     size_t bytes_read = m_process->GetTarget().ReadMemory (kext_summary.so_address,
698d4bfbc9aSGreg Clayton                                                            prefer_file_cache,
699d4bfbc9aSGreg Clayton                                                            header_bytes.GetBytes(),
700d4bfbc9aSGreg Clayton                                                            header_bytes.GetByteSize(),
701d4bfbc9aSGreg Clayton                                                            error);
702d4bfbc9aSGreg Clayton     if (bytes_read == sizeof(llvm::MachO::mach_header))
703d4bfbc9aSGreg Clayton     {
704d4bfbc9aSGreg Clayton         uint32_t offset = 0;
705d4bfbc9aSGreg Clayton         ::memset (&kext_summary.header, 0, sizeof(kext_summary.header));
706d4bfbc9aSGreg Clayton 
707d4bfbc9aSGreg Clayton         // Get the magic byte unswapped so we can figure out what we are dealing with
708d4bfbc9aSGreg Clayton         DataExtractor data(header_bytes.GetBytes(), header_bytes.GetByteSize(), endian::InlHostByteOrder(), 4);
709d4bfbc9aSGreg Clayton         kext_summary.header.magic = data.GetU32(&offset);
710d4bfbc9aSGreg Clayton         Address load_cmd_addr = kext_summary.so_address;
711944b828aSGreg Clayton         data.SetByteOrder(DynamicLoaderDarwinKernel::GetByteOrderFromMagic(kext_summary.header.magic));
712d4bfbc9aSGreg Clayton         switch (kext_summary.header.magic)
713d4bfbc9aSGreg Clayton         {
714d4bfbc9aSGreg Clayton         case llvm::MachO::HeaderMagic32:
715d4bfbc9aSGreg Clayton         case llvm::MachO::HeaderMagic32Swapped:
716d4bfbc9aSGreg Clayton             data.SetAddressByteSize(4);
717d4bfbc9aSGreg Clayton             load_cmd_addr.Slide (sizeof(llvm::MachO::mach_header));
718d4bfbc9aSGreg Clayton             break;
719d4bfbc9aSGreg Clayton 
720d4bfbc9aSGreg Clayton         case llvm::MachO::HeaderMagic64:
721d4bfbc9aSGreg Clayton         case llvm::MachO::HeaderMagic64Swapped:
722d4bfbc9aSGreg Clayton             data.SetAddressByteSize(8);
723d4bfbc9aSGreg Clayton             load_cmd_addr.Slide (sizeof(llvm::MachO::mach_header_64));
724d4bfbc9aSGreg Clayton             break;
725d4bfbc9aSGreg Clayton 
726d4bfbc9aSGreg Clayton         default:
727d4bfbc9aSGreg Clayton             return false;
728d4bfbc9aSGreg Clayton         }
729d4bfbc9aSGreg Clayton 
730d4bfbc9aSGreg Clayton         // Read the rest of dyld's mach header
731d4bfbc9aSGreg Clayton         if (data.GetU32(&offset, &kext_summary.header.cputype, (sizeof(llvm::MachO::mach_header)/sizeof(uint32_t)) - 1))
732d4bfbc9aSGreg Clayton         {
733d4bfbc9aSGreg Clayton             if (load_command_data == NULL)
734d4bfbc9aSGreg Clayton                 return true; // We were able to read the mach_header and weren't asked to read the load command bytes
735d4bfbc9aSGreg Clayton 
736d4bfbc9aSGreg Clayton             DataBufferSP load_cmd_data_sp(new DataBufferHeap(kext_summary.header.sizeofcmds, 0));
737d4bfbc9aSGreg Clayton 
738d4bfbc9aSGreg Clayton             size_t load_cmd_bytes_read = m_process->GetTarget().ReadMemory (load_cmd_addr,
739d4bfbc9aSGreg Clayton                                                                             prefer_file_cache,
740d4bfbc9aSGreg Clayton                                                                             load_cmd_data_sp->GetBytes(),
741d4bfbc9aSGreg Clayton                                                                             load_cmd_data_sp->GetByteSize(),
742d4bfbc9aSGreg Clayton                                                                             error);
743d4bfbc9aSGreg Clayton 
744d4bfbc9aSGreg Clayton             if (load_cmd_bytes_read == kext_summary.header.sizeofcmds)
745d4bfbc9aSGreg Clayton             {
746d4bfbc9aSGreg Clayton                 // Set the load command data and also set the correct endian
747d4bfbc9aSGreg Clayton                 // swap settings and the correct address size
748d4bfbc9aSGreg Clayton                 load_command_data->SetData(load_cmd_data_sp, 0, kext_summary.header.sizeofcmds);
749d4bfbc9aSGreg Clayton                 load_command_data->SetByteOrder(data.GetByteOrder());
750d4bfbc9aSGreg Clayton                 load_command_data->SetAddressByteSize(data.GetAddressByteSize());
751d4bfbc9aSGreg Clayton                 return true; // We successfully read the mach_header and the load command data
752d4bfbc9aSGreg Clayton             }
753d4bfbc9aSGreg Clayton 
754d4bfbc9aSGreg Clayton             return false; // We weren't able to read the load command data
755d4bfbc9aSGreg Clayton         }
756d4bfbc9aSGreg Clayton     }
757d4bfbc9aSGreg Clayton     return false; // We failed the read the mach_header
758d4bfbc9aSGreg Clayton }
759d4bfbc9aSGreg Clayton 
760d4bfbc9aSGreg Clayton 
761d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
762d4bfbc9aSGreg Clayton // Parse the load commands for an image
763d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
764d4bfbc9aSGreg Clayton uint32_t
765944b828aSGreg Clayton DynamicLoaderDarwinKernel::ParseLoadCommands (const DataExtractor& data, OSKextLoadedKextSummary& image_info)
766d4bfbc9aSGreg Clayton {
767d4bfbc9aSGreg Clayton     uint32_t offset = 0;
768d4bfbc9aSGreg Clayton     uint32_t cmd_idx;
769d4bfbc9aSGreg Clayton     Segment segment;
770d4bfbc9aSGreg Clayton     image_info.Clear (true);
771d4bfbc9aSGreg Clayton 
772d4bfbc9aSGreg Clayton     for (cmd_idx = 0; cmd_idx < image_info.header.ncmds; cmd_idx++)
773d4bfbc9aSGreg Clayton     {
774d4bfbc9aSGreg Clayton         // Clear out any load command specific data from image_info since
775d4bfbc9aSGreg Clayton         // we are about to read it.
776d4bfbc9aSGreg Clayton 
777d4bfbc9aSGreg Clayton         if (data.ValidOffsetForDataOfSize (offset, sizeof(llvm::MachO::load_command)))
778d4bfbc9aSGreg Clayton         {
779d4bfbc9aSGreg Clayton             llvm::MachO::load_command load_cmd;
780d4bfbc9aSGreg Clayton             uint32_t load_cmd_offset = offset;
781d4bfbc9aSGreg Clayton             load_cmd.cmd = data.GetU32 (&offset);
782d4bfbc9aSGreg Clayton             load_cmd.cmdsize = data.GetU32 (&offset);
783d4bfbc9aSGreg Clayton             switch (load_cmd.cmd)
784d4bfbc9aSGreg Clayton             {
785d4bfbc9aSGreg Clayton             case llvm::MachO::LoadCommandSegment32:
786d4bfbc9aSGreg Clayton                 {
787d4bfbc9aSGreg Clayton                     segment.name.SetTrimmedCStringWithLength ((const char *)data.GetData(&offset, 16), 16);
788d4bfbc9aSGreg Clayton                     // We are putting 4 uint32_t values 4 uint64_t values so
789d4bfbc9aSGreg Clayton                     // we have to use multiple 32 bit gets below.
790d4bfbc9aSGreg Clayton                     segment.vmaddr = data.GetU32 (&offset);
791d4bfbc9aSGreg Clayton                     segment.vmsize = data.GetU32 (&offset);
792d4bfbc9aSGreg Clayton                     segment.fileoff = data.GetU32 (&offset);
793d4bfbc9aSGreg Clayton                     segment.filesize = data.GetU32 (&offset);
794d4bfbc9aSGreg Clayton                     // Extract maxprot, initprot, nsects and flags all at once
795d4bfbc9aSGreg Clayton                     data.GetU32(&offset, &segment.maxprot, 4);
796d4bfbc9aSGreg Clayton                     image_info.segments.push_back (segment);
797d4bfbc9aSGreg Clayton                 }
798d4bfbc9aSGreg Clayton                 break;
799d4bfbc9aSGreg Clayton 
800d4bfbc9aSGreg Clayton             case llvm::MachO::LoadCommandSegment64:
801d4bfbc9aSGreg Clayton                 {
802d4bfbc9aSGreg Clayton                     segment.name.SetTrimmedCStringWithLength ((const char *)data.GetData(&offset, 16), 16);
803d4bfbc9aSGreg Clayton                     // Extract vmaddr, vmsize, fileoff, and filesize all at once
804d4bfbc9aSGreg Clayton                     data.GetU64(&offset, &segment.vmaddr, 4);
805d4bfbc9aSGreg Clayton                     // Extract maxprot, initprot, nsects and flags all at once
806d4bfbc9aSGreg Clayton                     data.GetU32(&offset, &segment.maxprot, 4);
807d4bfbc9aSGreg Clayton                     image_info.segments.push_back (segment);
808d4bfbc9aSGreg Clayton                 }
809d4bfbc9aSGreg Clayton                 break;
810d4bfbc9aSGreg Clayton 
811d4bfbc9aSGreg Clayton             case llvm::MachO::LoadCommandUUID:
812d4bfbc9aSGreg Clayton                 image_info.uuid.SetBytes(data.GetData (&offset, 16));
813d4bfbc9aSGreg Clayton                 break;
814d4bfbc9aSGreg Clayton 
815d4bfbc9aSGreg Clayton             default:
816d4bfbc9aSGreg Clayton                 break;
817d4bfbc9aSGreg Clayton             }
818d4bfbc9aSGreg Clayton             // Set offset to be the beginning of the next load command.
819d4bfbc9aSGreg Clayton             offset = load_cmd_offset + load_cmd.cmdsize;
820d4bfbc9aSGreg Clayton         }
821d4bfbc9aSGreg Clayton     }
822d4bfbc9aSGreg Clayton #if 0
823d4bfbc9aSGreg Clayton     // No slide in the kernel...
824d4bfbc9aSGreg Clayton 
825d4bfbc9aSGreg Clayton     // All sections listed in the dyld image info structure will all
826d4bfbc9aSGreg Clayton     // either be fixed up already, or they will all be off by a single
827d4bfbc9aSGreg Clayton     // slide amount that is determined by finding the first segment
828d4bfbc9aSGreg Clayton     // that is at file offset zero which also has bytes (a file size
829d4bfbc9aSGreg Clayton     // that is greater than zero) in the object file.
830d4bfbc9aSGreg Clayton 
831d4bfbc9aSGreg Clayton     // Determine the slide amount (if any)
832d4bfbc9aSGreg Clayton     const size_t num_sections = image_info.segments.size();
833d4bfbc9aSGreg Clayton     for (size_t i = 0; i < num_sections; ++i)
834d4bfbc9aSGreg Clayton     {
835d4bfbc9aSGreg Clayton         // Iterate through the object file sections to find the
836d4bfbc9aSGreg Clayton         // first section that starts of file offset zero and that
837d4bfbc9aSGreg Clayton         // has bytes in the file...
838d4bfbc9aSGreg Clayton         if (image_info.segments[i].fileoff == 0 && image_info.segments[i].filesize > 0)
839d4bfbc9aSGreg Clayton         {
840d4bfbc9aSGreg Clayton             image_info.slide = image_info.address - image_info.segments[i].vmaddr;
841d4bfbc9aSGreg Clayton             // We have found the slide amount, so we can exit
842d4bfbc9aSGreg Clayton             // this for loop.
843d4bfbc9aSGreg Clayton             break;
844d4bfbc9aSGreg Clayton         }
845d4bfbc9aSGreg Clayton     }
846d4bfbc9aSGreg Clayton #endif
847d4bfbc9aSGreg Clayton     if (image_info.uuid.IsValid())
848d4bfbc9aSGreg Clayton     {
849d4bfbc9aSGreg Clayton         bool did_create = false;
850d4bfbc9aSGreg Clayton         if (FindTargetModule(image_info, true, &did_create))
851d4bfbc9aSGreg Clayton         {
852d4bfbc9aSGreg Clayton             if (did_create)
853d4bfbc9aSGreg Clayton                 image_info.module_create_stop_id = m_process->GetStopID();
854d4bfbc9aSGreg Clayton         }
855d4bfbc9aSGreg Clayton     }
856d4bfbc9aSGreg Clayton     return cmd_idx;
857d4bfbc9aSGreg Clayton }
858d4bfbc9aSGreg Clayton 
859d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
860d4bfbc9aSGreg Clayton // Dump a Segment to the file handle provided.
861d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
862d4bfbc9aSGreg Clayton void
863944b828aSGreg Clayton DynamicLoaderDarwinKernel::Segment::PutToLog (Log *log, addr_t slide) const
864d4bfbc9aSGreg Clayton {
865d4bfbc9aSGreg Clayton     if (log)
866d4bfbc9aSGreg Clayton     {
867d4bfbc9aSGreg Clayton         if (slide == 0)
868d4bfbc9aSGreg Clayton             log->Printf ("\t\t%16s [0x%16.16llx - 0x%16.16llx)",
869d4bfbc9aSGreg Clayton                          name.AsCString(""),
870d4bfbc9aSGreg Clayton                          vmaddr + slide,
871d4bfbc9aSGreg Clayton                          vmaddr + slide + vmsize);
872d4bfbc9aSGreg Clayton         else
873d4bfbc9aSGreg Clayton             log->Printf ("\t\t%16s [0x%16.16llx - 0x%16.16llx) slide = 0x%llx",
874d4bfbc9aSGreg Clayton                          name.AsCString(""),
875d4bfbc9aSGreg Clayton                          vmaddr + slide,
876d4bfbc9aSGreg Clayton                          vmaddr + slide + vmsize,
877d4bfbc9aSGreg Clayton                          slide);
878d4bfbc9aSGreg Clayton     }
879d4bfbc9aSGreg Clayton }
880d4bfbc9aSGreg Clayton 
881944b828aSGreg Clayton const DynamicLoaderDarwinKernel::Segment *
882944b828aSGreg Clayton DynamicLoaderDarwinKernel::OSKextLoadedKextSummary::FindSegment (const ConstString &name) const
883d4bfbc9aSGreg Clayton {
884d4bfbc9aSGreg Clayton     const size_t num_segments = segments.size();
885d4bfbc9aSGreg Clayton     for (size_t i=0; i<num_segments; ++i)
886d4bfbc9aSGreg Clayton     {
887d4bfbc9aSGreg Clayton         if (segments[i].name == name)
888d4bfbc9aSGreg Clayton             return &segments[i];
889d4bfbc9aSGreg Clayton     }
890d4bfbc9aSGreg Clayton     return NULL;
891d4bfbc9aSGreg Clayton }
892d4bfbc9aSGreg Clayton 
893d4bfbc9aSGreg Clayton 
894d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
895d4bfbc9aSGreg Clayton // Dump an image info structure to the file handle provided.
896d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
897d4bfbc9aSGreg Clayton void
898944b828aSGreg Clayton DynamicLoaderDarwinKernel::OSKextLoadedKextSummary::PutToLog (Log *log) const
899d4bfbc9aSGreg Clayton {
900d4bfbc9aSGreg Clayton     if (log == NULL)
901d4bfbc9aSGreg Clayton         return;
902d4bfbc9aSGreg Clayton     const uint8_t *u = (uint8_t *)uuid.GetBytes();
903d4bfbc9aSGreg Clayton 
904d4bfbc9aSGreg Clayton     if (address == LLDB_INVALID_ADDRESS)
905d4bfbc9aSGreg Clayton     {
906d4bfbc9aSGreg Clayton         if (u)
907d4bfbc9aSGreg Clayton         {
908d4bfbc9aSGreg 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)",
909d4bfbc9aSGreg Clayton                         u[ 0], u[ 1], u[ 2], u[ 3],
910d4bfbc9aSGreg Clayton                         u[ 4], u[ 5], u[ 6], u[ 7],
911d4bfbc9aSGreg Clayton                         u[ 8], u[ 9], u[10], u[11],
912d4bfbc9aSGreg Clayton                         u[12], u[13], u[14], u[15],
913d4bfbc9aSGreg Clayton                         name);
914d4bfbc9aSGreg Clayton         }
915d4bfbc9aSGreg Clayton         else
916d4bfbc9aSGreg Clayton             log->Printf("\tname=\"%s\" (UNLOADED)", name);
917d4bfbc9aSGreg Clayton     }
918d4bfbc9aSGreg Clayton     else
919d4bfbc9aSGreg Clayton     {
920d4bfbc9aSGreg Clayton         if (u)
921d4bfbc9aSGreg Clayton         {
922d4bfbc9aSGreg Clayton             log->Printf("\taddr=0x%16.16llx size=0x%16.16llx version=0x%16.16llx load-tag=0x%8.8x flags=0x%8.8x ref-list=0x%16.16llx 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\"",
923d4bfbc9aSGreg Clayton                         address, size, version, load_tag, flags, reference_list,
924d4bfbc9aSGreg Clayton                         u[ 0], u[ 1], u[ 2], u[ 3], u[ 4], u[ 5], u[ 6], u[ 7],
925d4bfbc9aSGreg Clayton                         u[ 8], u[ 9], u[10], u[11], u[12], u[13], u[14], u[15],
926d4bfbc9aSGreg Clayton                         name);
927d4bfbc9aSGreg Clayton         }
928d4bfbc9aSGreg Clayton         else
929d4bfbc9aSGreg Clayton         {
930d4bfbc9aSGreg Clayton             log->Printf("\t[0x%16.16llx - 0x%16.16llx) version=0x%16.16llx load-tag=0x%8.8x flags=0x%8.8x ref-list=0x%16.16llx name=\"%s\"",
931d4bfbc9aSGreg Clayton                         address, address+size, version, load_tag, flags, reference_list,
932d4bfbc9aSGreg Clayton                         name);
933d4bfbc9aSGreg Clayton         }
934d4bfbc9aSGreg Clayton         for (uint32_t i=0; i<segments.size(); ++i)
935d4bfbc9aSGreg Clayton             segments[i].PutToLog(log, 0);
936d4bfbc9aSGreg Clayton     }
937d4bfbc9aSGreg Clayton }
938d4bfbc9aSGreg Clayton 
939d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
940d4bfbc9aSGreg Clayton // Dump the _dyld_all_image_infos members and all current image infos
941d4bfbc9aSGreg Clayton // that we have parsed to the file handle provided.
942d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
943d4bfbc9aSGreg Clayton void
944944b828aSGreg Clayton DynamicLoaderDarwinKernel::PutToLog(Log *log) const
945d4bfbc9aSGreg Clayton {
946d4bfbc9aSGreg Clayton     if (log == NULL)
947d4bfbc9aSGreg Clayton         return;
948d4bfbc9aSGreg Clayton 
949d4bfbc9aSGreg Clayton     Mutex::Locker locker(m_mutex);
950d4bfbc9aSGreg Clayton     log->Printf("gLoadedKextSummaries = 0x%16.16llx { version=%u, entry_size=%u, entry_count=%u }",
951d4bfbc9aSGreg Clayton                 m_kext_summary_header_addr.GetFileAddress(),
952d4bfbc9aSGreg Clayton                 m_kext_summary_header.version,
953d4bfbc9aSGreg Clayton                 m_kext_summary_header.entry_size,
954d4bfbc9aSGreg Clayton                 m_kext_summary_header.entry_count);
955d4bfbc9aSGreg Clayton 
956d4bfbc9aSGreg Clayton     size_t i;
957d4bfbc9aSGreg Clayton     const size_t count = m_kext_summaries.size();
958d4bfbc9aSGreg Clayton     if (count > 0)
959d4bfbc9aSGreg Clayton     {
960d4bfbc9aSGreg Clayton         log->PutCString("Loaded:");
961d4bfbc9aSGreg Clayton         for (i = 0; i<count; i++)
962d4bfbc9aSGreg Clayton             m_kext_summaries[i].PutToLog(log);
963d4bfbc9aSGreg Clayton     }
964d4bfbc9aSGreg Clayton }
965d4bfbc9aSGreg Clayton 
966d4bfbc9aSGreg Clayton void
967944b828aSGreg Clayton DynamicLoaderDarwinKernel::PrivateInitialize(Process *process)
968d4bfbc9aSGreg Clayton {
969944b828aSGreg Clayton     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n", __FUNCTION__, StateAsCString(m_process->GetState()));
970d4bfbc9aSGreg Clayton     Clear(true);
971d4bfbc9aSGreg Clayton     m_process = process;
972d4bfbc9aSGreg Clayton     m_process->GetTarget().GetSectionLoadList().Clear();
973d4bfbc9aSGreg Clayton }
974d4bfbc9aSGreg Clayton 
975d4bfbc9aSGreg Clayton void
976944b828aSGreg Clayton DynamicLoaderDarwinKernel::SetNotificationBreakpointIfNeeded ()
977d4bfbc9aSGreg Clayton {
978d4bfbc9aSGreg Clayton     if (m_break_id == LLDB_INVALID_BREAK_ID)
979d4bfbc9aSGreg Clayton     {
980944b828aSGreg Clayton         DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s() process state = %s\n", __FUNCTION__, StateAsCString(m_process->GetState()));
981d4bfbc9aSGreg Clayton 
982d4bfbc9aSGreg Clayton 
983d4bfbc9aSGreg Clayton         const bool internal_bp = false;
984d4bfbc9aSGreg Clayton         const LazyBool skip_prologue = eLazyBoolNo;
985d4bfbc9aSGreg Clayton         Breakpoint *bp = m_process->GetTarget().CreateBreakpoint (&m_kernel.module_sp->GetFileSpec(),
986d4bfbc9aSGreg Clayton                                                                   "OSKextLoadedKextSummariesUpdated",
987d4bfbc9aSGreg Clayton                                                                   eFunctionNameTypeFull,
988d4bfbc9aSGreg Clayton                                                                   internal_bp,
989d4bfbc9aSGreg Clayton                                                                   skip_prologue).get();
990d4bfbc9aSGreg Clayton 
991944b828aSGreg Clayton         bp->SetCallback (DynamicLoaderDarwinKernel::BreakpointHitCallback, this, true);
992d4bfbc9aSGreg Clayton         m_break_id = bp->GetID();
993d4bfbc9aSGreg Clayton     }
994d4bfbc9aSGreg Clayton }
995d4bfbc9aSGreg Clayton 
996d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
997d4bfbc9aSGreg Clayton // Member function that gets called when the process state changes.
998d4bfbc9aSGreg Clayton //----------------------------------------------------------------------
999d4bfbc9aSGreg Clayton void
1000944b828aSGreg Clayton DynamicLoaderDarwinKernel::PrivateProcessStateChanged (Process *process, StateType state)
1001d4bfbc9aSGreg Clayton {
1002944b828aSGreg Clayton     DEBUG_PRINTF("DynamicLoaderDarwinKernel::%s(%s)\n", __FUNCTION__, StateAsCString(state));
1003d4bfbc9aSGreg Clayton     switch (state)
1004d4bfbc9aSGreg Clayton     {
1005d4bfbc9aSGreg Clayton     case eStateConnected:
1006d4bfbc9aSGreg Clayton     case eStateAttaching:
1007d4bfbc9aSGreg Clayton     case eStateLaunching:
1008d4bfbc9aSGreg Clayton     case eStateInvalid:
1009d4bfbc9aSGreg Clayton     case eStateUnloaded:
1010d4bfbc9aSGreg Clayton     case eStateExited:
1011d4bfbc9aSGreg Clayton     case eStateDetached:
1012d4bfbc9aSGreg Clayton         Clear(false);
1013d4bfbc9aSGreg Clayton         break;
1014d4bfbc9aSGreg Clayton 
1015d4bfbc9aSGreg Clayton     case eStateStopped:
1016d4bfbc9aSGreg Clayton         UpdateIfNeeded();
1017d4bfbc9aSGreg Clayton         break;
1018d4bfbc9aSGreg Clayton 
1019d4bfbc9aSGreg Clayton     case eStateRunning:
1020d4bfbc9aSGreg Clayton     case eStateStepping:
1021d4bfbc9aSGreg Clayton     case eStateCrashed:
1022d4bfbc9aSGreg Clayton     case eStateSuspended:
1023d4bfbc9aSGreg Clayton         break;
1024d4bfbc9aSGreg Clayton 
1025d4bfbc9aSGreg Clayton     default:
1026d4bfbc9aSGreg Clayton         break;
1027d4bfbc9aSGreg Clayton     }
1028d4bfbc9aSGreg Clayton }
1029d4bfbc9aSGreg Clayton 
1030d4bfbc9aSGreg Clayton ThreadPlanSP
1031944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetStepThroughTrampolinePlan (Thread &thread, bool stop_others)
1032d4bfbc9aSGreg Clayton {
1033d4bfbc9aSGreg Clayton     ThreadPlanSP thread_plan_sp;
1034d4bfbc9aSGreg Clayton     LogSP log(GetLogIfAllCategoriesSet (LIBLLDB_LOG_STEP));
1035d4bfbc9aSGreg Clayton     if (log)
1036d4bfbc9aSGreg Clayton         log->Printf ("Could not find symbol for step through.");
1037d4bfbc9aSGreg Clayton     return thread_plan_sp;
1038d4bfbc9aSGreg Clayton }
1039d4bfbc9aSGreg Clayton 
1040d4bfbc9aSGreg Clayton Error
1041944b828aSGreg Clayton DynamicLoaderDarwinKernel::CanLoadImage ()
1042d4bfbc9aSGreg Clayton {
1043d4bfbc9aSGreg Clayton     Error error;
1044d4bfbc9aSGreg Clayton     error.SetErrorString("always unsafe to load or unload shared libraries in the darwin kernel");
1045d4bfbc9aSGreg Clayton     return error;
1046d4bfbc9aSGreg Clayton }
1047d4bfbc9aSGreg Clayton 
1048d4bfbc9aSGreg Clayton void
1049944b828aSGreg Clayton DynamicLoaderDarwinKernel::Initialize()
1050d4bfbc9aSGreg Clayton {
1051d4bfbc9aSGreg Clayton     PluginManager::RegisterPlugin (GetPluginNameStatic(),
1052d4bfbc9aSGreg Clayton                                    GetPluginDescriptionStatic(),
1053d4bfbc9aSGreg Clayton                                    CreateInstance);
1054d4bfbc9aSGreg Clayton }
1055d4bfbc9aSGreg Clayton 
1056d4bfbc9aSGreg Clayton void
1057944b828aSGreg Clayton DynamicLoaderDarwinKernel::Terminate()
1058d4bfbc9aSGreg Clayton {
1059d4bfbc9aSGreg Clayton     PluginManager::UnregisterPlugin (CreateInstance);
1060d4bfbc9aSGreg Clayton }
1061d4bfbc9aSGreg Clayton 
1062d4bfbc9aSGreg Clayton 
1063d4bfbc9aSGreg Clayton const char *
1064944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginNameStatic()
1065d4bfbc9aSGreg Clayton {
1066d4bfbc9aSGreg Clayton     return "dynamic-loader.macosx-kernel";
1067d4bfbc9aSGreg Clayton }
1068d4bfbc9aSGreg Clayton 
1069d4bfbc9aSGreg Clayton const char *
1070944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginDescriptionStatic()
1071d4bfbc9aSGreg Clayton {
1072d4bfbc9aSGreg Clayton     return "Dynamic loader plug-in that watches for shared library loads/unloads in the MacOSX kernel.";
1073d4bfbc9aSGreg Clayton }
1074d4bfbc9aSGreg Clayton 
1075d4bfbc9aSGreg Clayton 
1076d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1077d4bfbc9aSGreg Clayton // PluginInterface protocol
1078d4bfbc9aSGreg Clayton //------------------------------------------------------------------
1079d4bfbc9aSGreg Clayton const char *
1080944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginName()
1081d4bfbc9aSGreg Clayton {
1082944b828aSGreg Clayton     return "DynamicLoaderDarwinKernel";
1083d4bfbc9aSGreg Clayton }
1084d4bfbc9aSGreg Clayton 
1085d4bfbc9aSGreg Clayton const char *
1086944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetShortPluginName()
1087d4bfbc9aSGreg Clayton {
1088d4bfbc9aSGreg Clayton     return GetPluginNameStatic();
1089d4bfbc9aSGreg Clayton }
1090d4bfbc9aSGreg Clayton 
1091d4bfbc9aSGreg Clayton uint32_t
1092944b828aSGreg Clayton DynamicLoaderDarwinKernel::GetPluginVersion()
1093d4bfbc9aSGreg Clayton {
1094d4bfbc9aSGreg Clayton     return 1;
1095d4bfbc9aSGreg Clayton }
1096d4bfbc9aSGreg Clayton 
1097