The kernel loading code is now isolated in the DynamicLoaderDarwinKernel;
remove the duplicates of this code in ProcessGDBRemote and ProcessKDP.
These two Process plugins will hardcode their DynamicLoader name to be
the DynamicLoaderDarwinKernel so the correct DynamicLoader is picked,
and return the kernel load address as the ImageInfosAddress.
<rdar://problem/12417038> 


git-svn-id: https://llvm.org/svn/llvm-project/lldb/trunk@165080 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/source/Plugins/DynamicLoader/Darwin-Kernel/DynamicLoaderDarwinKernel.cpp b/source/Plugins/DynamicLoader/Darwin-Kernel/DynamicLoaderDarwinKernel.cpp
index 1231e88..67a3b16 100644
--- a/source/Plugins/DynamicLoader/Darwin-Kernel/DynamicLoaderDarwinKernel.cpp
+++ b/source/Plugins/DynamicLoader/Darwin-Kernel/DynamicLoaderDarwinKernel.cpp
@@ -20,7 +20,6 @@
 #include "lldb/Core/State.h"
 #include "lldb/Host/Symbols.h"
 #include "lldb/Symbol/ObjectFile.h"
-#include "lldb/Target/ObjCLanguageRuntime.h"
 #include "lldb/Target/RegisterContext.h"
 #include "lldb/Target/StackFrame.h"
 #include "lldb/Target/Target.h"
@@ -41,11 +40,6 @@
 using namespace lldb;
 using namespace lldb_private;
 
-/// FIXME - The ObjC Runtime trampoline handler doesn't really belong here.
-/// I am putting it here so I can invoke it in the Trampoline code here, but
-/// it should be moved to the ObjC Runtime support when it is set up.
-
-
 //----------------------------------------------------------------------
 // Create an instance of this class. This function is filled into
 // the plugin info class that gets handed out by the plugin factory and
diff --git a/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.cpp b/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.cpp
index 4e85689..88b93f1 100644
--- a/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.cpp
+++ b/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.cpp
@@ -31,6 +31,7 @@
 #include "ProcessKDP.h"
 #include "ProcessKDPLog.h"
 #include "ThreadKDP.h"
+#include "Plugins/DynamicLoader/Darwin-Kernel/DynamicLoaderDarwinKernel.h"
 
 using namespace lldb;
 using namespace lldb_private;
@@ -105,7 +106,9 @@
     m_comm("lldb.process.kdp-remote.communication"),
     m_async_broadcaster (NULL, "lldb.process.kdp-remote.async-broadcaster"),
     m_async_thread (LLDB_INVALID_HOST_THREAD),
-    m_destroy_in_process (false)
+    m_destroy_in_process (false),
+    m_dyld_plugin_name (),
+    m_kernel_load_addr (LLDB_INVALID_ADDRESS)
 {
     m_async_broadcaster.SetEventName (eBroadcastBitAsyncThreadShouldExit,   "async thread should exit");
     m_async_broadcaster.SetEventName (eBroadcastBitAsyncContinue,           "async thread continue");
@@ -222,85 +225,13 @@
                         
                     UUID kernel_uuid = m_comm.GetUUID ();
                     addr_t kernel_load_addr = m_comm.GetLoadAddress ();
-                    if (strm)
-                    {
-                        char uuidbuf[64];
-                        strm->Printf ("Kernel UUID: %s\n", kernel_uuid.GetAsCString (uuidbuf, sizeof (uuidbuf)));
-                        strm->Printf ("Load Address: 0x%llx\n", kernel_load_addr);
-                        strm->Flush ();
-                    }
 
-                    /* Set the kernel's LoadAddress based on the information from kdp.
-                       This would normally be handled by the DynamicLoaderDarwinKernel plugin but there's no easy
-                       way to communicate the UUID / load addr from kdp back up to that plugin so we'll set it here. */
-                    ModuleSP exe_module_sp = m_target.GetExecutableModule ();
-                    bool find_and_load_kernel = true;
-                    if (exe_module_sp.get ())
+                    if (kernel_load_addr != LLDB_INVALID_ADDRESS)
                     {
-                        ObjectFile *exe_objfile = exe_module_sp->GetObjectFile();
-                        if (exe_objfile->GetType() == ObjectFile::eTypeExecutable && 
-                            exe_objfile->GetStrata() == ObjectFile::eStrataKernel)
-                        {
-                            UUID exe_objfile_uuid;
-                            if (exe_objfile->GetUUID (&exe_objfile_uuid) && kernel_uuid == exe_objfile_uuid
-                                && exe_objfile->GetHeaderAddress().IsValid())
-                            {
-                                find_and_load_kernel = false;
-                                addr_t slide = kernel_load_addr - exe_objfile->GetHeaderAddress().GetFileAddress();
-                                if (slide != 0)
-                                {
-                                    bool changed = false;
-                                    exe_module_sp->SetLoadAddress (m_target, slide, changed);
-                                    if (changed)
-                                    {
-                                        ModuleList modlist;
-                                        modlist.Append (exe_module_sp);
-                                        m_target.ModulesDidLoad (modlist);
-                                    }
-                                }
-                            }
-                        }
+                        m_kernel_load_addr = kernel_load_addr;
+                        m_dyld_plugin_name = DynamicLoaderDarwinKernel::GetPluginNameStatic();
                     }
 
-                    // If the executable binary is not the same as the kernel being run on the remote host,
-                    // see if Symbols::DownloadObjectAndSymbolFile can find us a symbol file based on the UUID
-                    // and if so, load it at the correct address.
-                    if (find_and_load_kernel && kernel_load_addr != LLDB_INVALID_ADDRESS && kernel_uuid.IsValid())
-                    {
-                        ModuleSpec sym_spec;
-                        sym_spec.GetUUID() = kernel_uuid;
-                        if (Symbols::DownloadObjectAndSymbolFile (sym_spec) 
-                            && sym_spec.GetArchitecture().IsValid() 
-                            && sym_spec.GetSymbolFileSpec().Exists())
-                        {
-                            ModuleSP kernel_sp = m_target.GetSharedModule (sym_spec);
-                            if (kernel_sp.get())
-                            {
-                                m_target.SetExecutableModule(kernel_sp, false);
-                                if (kernel_sp->GetObjectFile() && kernel_sp->GetObjectFile()->GetHeaderAddress().IsValid())
-                                {
-                                    addr_t slide = kernel_load_addr - kernel_sp->GetObjectFile()->GetHeaderAddress().GetFileAddress();
-                                    bool changed = false;
-                                    kernel_sp->SetLoadAddress (m_target, slide, changed);
-                                    if (changed)
-                                    {
-                                        ModuleList modlist;
-                                        modlist.Append (kernel_sp);
-                                        m_target.ModulesDidLoad (modlist);
-                                    }
-                                    if (strm)
-                                    {
-                                        strm->Printf ("Loaded kernel file %s/%s\n", 
-                                                      kernel_sp->GetFileSpec().GetDirectory().AsCString(),
-                                                      kernel_sp->GetFileSpec().GetFilename().AsCString());
-                                        strm->Flush ();
-                                    }
-                                }
-                            }
-                        }
-                    }
-
-
                     // Set the thread ID
                     UpdateThreadListIfNeeded ();
                     SetID (1);
@@ -400,6 +331,20 @@
     }
 }
 
+addr_t
+ProcessKDP::GetImageInfoAddress()
+{
+    return m_kernel_load_addr;
+}
+
+lldb_private::DynamicLoader *
+ProcessKDP::GetDynamicLoader ()
+{
+    if (m_dyld_ap.get() == NULL)
+        m_dyld_ap.reset (DynamicLoader::FindPlugin(this, m_dyld_plugin_name.empty() ? NULL : m_dyld_plugin_name.c_str()));
+    return m_dyld_ap.get();
+}
+
 Error
 ProcessKDP::WillResume ()
 {
diff --git a/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.h b/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.h
index 4aebcdd..5fb1027 100644
--- a/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.h
+++ b/source/Plugins/Process/MacOSX-Kernel/ProcessKDP.h
@@ -104,6 +104,12 @@
     virtual void
     DidAttach ();
     
+    lldb::addr_t
+    GetImageInfoAddress();
+
+    lldb_private::DynamicLoader *
+    GetDynamicLoader ();
+
     //------------------------------------------------------------------
     // PluginInterface protocol
     //------------------------------------------------------------------
@@ -252,6 +258,8 @@
     lldb_private::Broadcaster m_async_broadcaster;
     lldb::thread_t m_async_thread;
     bool m_destroy_in_process;
+    std::string m_dyld_plugin_name;
+    lldb::addr_t m_kernel_load_addr;
 
     bool
     StartAsyncThread ();
diff --git a/source/Plugins/Process/gdb-remote/ProcessGDBRemote.cpp b/source/Plugins/Process/gdb-remote/ProcessGDBRemote.cpp
index d39d1a5..7e20b0a 100644
--- a/source/Plugins/Process/gdb-remote/ProcessGDBRemote.cpp
+++ b/source/Plugins/Process/gdb-remote/ProcessGDBRemote.cpp
@@ -58,6 +58,8 @@
 #include "ProcessGDBRemoteLog.h"
 #include "ThreadGDBRemote.h"
 
+#include "Plugins/DynamicLoader/Darwin-Kernel/DynamicLoaderDarwinKernel.h"
+
 namespace lldb
 {
     // Provide a function that can easily dump the packet history if we know a
@@ -192,7 +194,9 @@
     m_addr_to_mmap_size (),
     m_thread_create_bp_sp (),
     m_waiting_for_attach (false),
-    m_destroy_tried_resuming (false)
+    m_destroy_tried_resuming (false),
+    m_dyld_plugin_name(),
+    m_kernel_load_addr (LLDB_INVALID_ADDRESS)
 {
     m_async_broadcaster.SetEventName (eBroadcastBitAsyncThreadShouldExit,   "async thread should exit");
     m_async_broadcaster.SetEventName (eBroadcastBitAsyncContinue,           "async thread continue");
@@ -428,7 +432,7 @@
         return error;
     StartAsyncThread ();
 
-    RelocateOrLoadKernel (strm);
+    CheckForKernel (strm);
 
     lldb::pid_t pid = m_gdb_comm.GetCurrentProcessID ();
     if (pid == LLDB_INVALID_PROCESS_ID)
@@ -471,28 +475,14 @@
 // or the executable is a kernel, we may be looking at a KASLR situation (where the kernel has been
 // slid in memory.)
 //
-// This function does several things:
+// This function tries to locate the kernel in memory if this is possibly a kernel debug session.
 //
-// 1. If a non-kernel executable is provided, do nothing.  If the executable provided is a kernel and
-//    it loaded at a non-slid address (FileAddress == LoadAddress), do nothing.
-//
-// 2. When in debug mode the kernel will record its actual load address at a fixed address in memory.
-//    Check those addresses, see if there is a kernel binary at them.
-//
-// 3. If we find a kernel in memory and it matches the executable provided, adjust to the slide.
-//
-// 4. If lldb was given no executable at startup, or the one we find in memory does not match the one
-//    provided, try to locate a copy of the correct kernel on the host system.  Else read it out of memory.
-//
-// gdb would take an additional series of steps where it would scan through memory looking for a kernel
-// to find a slid kernel that was not booted in debug mode (these were also needed back when the kernel
-// didn't record its load address anywhere).  With luck we won't need to pull those in to lldb.
-//
-// The obvious location for all of this code would be in the DynamicLoaderDarwinKernel -- but if we're started
-// without any executable binary provided, we won't know to use that plugin.
+// If a kernel is found, return the address of the kernel in GetImageInfoAddress() -- the 
+// DynamicLoaderDarwinKernel plugin uses this address as the kernel load address and will load the
+// binary, if needed, along with all the kexts.
 
 void
-ProcessGDBRemote::RelocateOrLoadKernel (Stream *strm)
+ProcessGDBRemote::CheckForKernel (Stream *strm)
 {
     // early return if this isn't an "unknown" system (kernel debugging doesn't have a system type)
     const ArchSpec &gdb_remote_arch = m_gdb_comm.GetHostArchitecture();
@@ -519,8 +509,8 @@
             && memory_module_sp->GetUUID().IsValid() 
             && memory_module_sp->GetUUID() == exe_module->GetUUID())
         {
-            bool changed = false;
-            exe_module->SetLoadAddress (GetTarget(), 0, changed);
+            m_kernel_load_addr = exe_objfile->GetHeaderAddress().GetFileAddress();
+            m_dyld_plugin_name = DynamicLoaderDarwinKernel::GetPluginNameStatic();
             return;
         }
     }
@@ -573,86 +563,13 @@
         }
     }
 
-    if (memory_module_sp.get())
+    if (memory_module_sp.get() 
+        && memory_module_sp->GetArchitecture().IsValid() 
+        && memory_module_sp->GetArchitecture().GetTriple().getVendor() == llvm::Triple::Apple)
     {
-        if (strm)
-        {
-            char uuidbuf[64];
-            strm->Printf ("Kernel UUID: %s\n", memory_module_sp->GetUUID().GetAsCString (uuidbuf, sizeof (uuidbuf)));
-            strm->Printf ("Load Address: 0x%llx\n", kernel_addr);
-            strm->Flush ();
-        }
-
-        // Did the user already give us the correct binary?  Don't re-load it if so, just set the load addr.
-        if (exe_module && exe_objfile && exe_module->GetUUID() == memory_module_sp->GetUUID())
-        {
-            bool changed = false;
-            addr_t slide = kernel_addr - exe_objfile->GetHeaderAddress().GetFileAddress();
-            exe_module->SetLoadAddress (GetTarget(), slide, changed);
-            if (changed)
-            {
-                    ModuleList modlist;
-                    modlist.Append (GetTarget().GetExecutableModule());
-                    GetTarget().ModulesDidLoad (modlist);
-            }
-            return;
-        }
-
-        // OK try to find a kernel on the local system, or get it from memory
-        LoadKernel (strm, memory_module_sp->GetUUID(), kernel_addr);
-    }
-}
-
-void
-ProcessGDBRemote::LoadKernel (Stream *strm, UUID kernel_uuid, addr_t kernel_load_addr)
-{
-
-    // First try to find the kernel binary by calling Symbols::DownloadObjectAndSymbolFile
-    ModuleSpec sym_spec;
-    sym_spec.GetUUID() = kernel_uuid;
-    if (Symbols::DownloadObjectAndSymbolFile (sym_spec) 
-        && sym_spec.GetArchitecture().IsValid() 
-        && sym_spec.GetSymbolFileSpec().Exists())
-    {
-        ModuleSP kernel_sp = GetTarget().GetSharedModule (sym_spec);
-        if (kernel_sp.get())
-        {
-            GetTarget().SetExecutableModule(kernel_sp, false);
-            if (kernel_sp->GetObjectFile() && kernel_sp->GetObjectFile()->GetHeaderAddress().IsValid())
-            {
-                addr_t slide = kernel_load_addr - kernel_sp->GetObjectFile()->GetHeaderAddress().GetFileAddress();
-                bool changed = false;
-                kernel_sp->SetLoadAddress (GetTarget(), slide, changed);
-                if (changed)
-                {
-                    ModuleList modlist;
-                    modlist.Append (kernel_sp);
-                    GetTarget().ModulesDidLoad (modlist);
-                }
-                if (strm)
-                {
-                    strm->Printf ("Loaded kernel file %s/%s\n", 
-                                  kernel_sp->GetFileSpec().GetDirectory().AsCString(),
-                                  kernel_sp->GetFileSpec().GetFilename().AsCString());
-                    strm->Flush ();
-                }
-                return;
-            }
-        }
-    }
-
-    // If nothing better, load the kernel binary out of memory - this is likely slow and may not get us symbols.
-    ModuleSP memory_module_sp = ReadModuleFromMemory (FileSpec("mach_kernel", false), kernel_load_addr, true, true);
-    if (memory_module_sp.get()
-        && memory_module_sp->GetUUID().IsValid()
-        && memory_module_sp->GetObjectFile()
-        && memory_module_sp->GetObjectFile()->GetType() == ObjectFile::eTypeExecutable
-        && memory_module_sp->GetObjectFile()->GetStrata() == ObjectFile::eStrataKernel)
-    {
-        bool changed;
-        uint64_t slide = kernel_load_addr - memory_module_sp->GetObjectFile()->GetHeaderAddress().GetFileAddress();
-        memory_module_sp->SetLoadAddress (GetTarget(), slide, changed);
-        GetTarget().SetExecutableModule(memory_module_sp, false);
+        m_kernel_load_addr = kernel_addr;
+        m_dyld_plugin_name = DynamicLoaderDarwinKernel::GetPluginNameStatic();
+        return;
     }
 }
 
@@ -2067,7 +1984,10 @@
 addr_t
 ProcessGDBRemote::GetImageInfoAddress()
 {
-    return m_gdb_comm.GetShlibInfoAddr();
+    if (m_kernel_load_addr != LLDB_INVALID_ADDRESS)
+        return m_kernel_load_addr;
+    else
+        return m_gdb_comm.GetShlibInfoAddr();
 }
 
 //------------------------------------------------------------------
@@ -3080,4 +3000,11 @@
     return true;
 }
     
+lldb_private::DynamicLoader *
+ProcessGDBRemote::GetDynamicLoader ()
+{
+    if (m_dyld_ap.get() == NULL)
+        m_dyld_ap.reset (DynamicLoader::FindPlugin(this, m_dyld_plugin_name.empty() ? NULL : m_dyld_plugin_name.c_str()));
+    return m_dyld_ap.get();
+}
 
diff --git a/source/Plugins/Process/gdb-remote/ProcessGDBRemote.h b/source/Plugins/Process/gdb-remote/ProcessGDBRemote.h
index 4732fc3..82f840d 100644
--- a/source/Plugins/Process/gdb-remote/ProcessGDBRemote.h
+++ b/source/Plugins/Process/gdb-remote/ProcessGDBRemote.h
@@ -295,10 +295,7 @@
     }
 
     void
-    RelocateOrLoadKernel (lldb_private::Stream *strm);
-
-    void
-    LoadKernel (lldb_private::Stream *strm, lldb_private::UUID kernel_uuid, lldb::addr_t kernel_load_addr);
+    CheckForKernel (lldb_private::Stream *strm);
 
     //------------------------------------------------------------------
     /// Broadcaster event bits definitions.
@@ -332,6 +329,8 @@
     lldb::BreakpointSP m_thread_create_bp_sp;
     bool m_waiting_for_attach;
     bool m_destroy_tried_resuming;
+    std::string m_dyld_plugin_name;
+    lldb::addr_t m_kernel_load_addr;
     
     bool
     StartAsyncThread ();
@@ -380,6 +379,9 @@
                         bool catch_stop_event, 
                         lldb::EventSP &stop_event_sp);
 
+    lldb_private::DynamicLoader *
+    GetDynamicLoader ();
+
 private:
     //------------------------------------------------------------------
     // For ProcessGDBRemote only