1 //===-- ProcessElfCore.cpp ------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include <stdlib.h>
10 
11 #include <memory>
12 #include <mutex>
13 
14 #include "lldb/Core/Module.h"
15 #include "lldb/Core/ModuleSpec.h"
16 #include "lldb/Core/PluginManager.h"
17 #include "lldb/Core/Section.h"
18 #include "lldb/Target/DynamicLoader.h"
19 #include "lldb/Target/MemoryRegionInfo.h"
20 #include "lldb/Target/Target.h"
21 #include "lldb/Target/UnixSignals.h"
22 #include "lldb/Utility/DataBufferHeap.h"
23 #include "lldb/Utility/Log.h"
24 #include "lldb/Utility/State.h"
25 
26 #include "llvm/BinaryFormat/ELF.h"
27 #include "llvm/Support/Threading.h"
28 
29 #include "Plugins/DynamicLoader/POSIX-DYLD/DynamicLoaderPOSIXDYLD.h"
30 #include "Plugins/ObjectFile/ELF/ObjectFileELF.h"
31 #include "Plugins/Process/elf-core/RegisterUtilities.h"
32 #include "ProcessElfCore.h"
33 #include "ThreadElfCore.h"
34 
35 using namespace lldb_private;
36 namespace ELF = llvm::ELF;
37 
38 LLDB_PLUGIN_DEFINE(ProcessElfCore)
39 
40 ConstString ProcessElfCore::GetPluginNameStatic() {
41   static ConstString g_name("elf-core");
42   return g_name;
43 }
44 
45 const char *ProcessElfCore::GetPluginDescriptionStatic() {
46   return "ELF core dump plug-in.";
47 }
48 
49 void ProcessElfCore::Terminate() {
50   PluginManager::UnregisterPlugin(ProcessElfCore::CreateInstance);
51 }
52 
53 lldb::ProcessSP ProcessElfCore::CreateInstance(lldb::TargetSP target_sp,
54                                                lldb::ListenerSP listener_sp,
55                                                const FileSpec *crash_file,
56                                                bool can_connect) {
57   lldb::ProcessSP process_sp;
58   if (crash_file && !can_connect) {
59     // Read enough data for a ELF32 header or ELF64 header Note: Here we care
60     // about e_type field only, so it is safe to ignore possible presence of
61     // the header extension.
62     const size_t header_size = sizeof(llvm::ELF::Elf64_Ehdr);
63 
64     auto data_sp = FileSystem::Instance().CreateDataBuffer(
65         crash_file->GetPath(), header_size, 0);
66     if (data_sp && data_sp->GetByteSize() == header_size &&
67         elf::ELFHeader::MagicBytesMatch(data_sp->GetBytes())) {
68       elf::ELFHeader elf_header;
69       DataExtractor data(data_sp, lldb::eByteOrderLittle, 4);
70       lldb::offset_t data_offset = 0;
71       if (elf_header.Parse(data, &data_offset)) {
72         if (elf_header.e_type == llvm::ELF::ET_CORE)
73           process_sp = std::make_shared<ProcessElfCore>(target_sp, listener_sp,
74                                                         *crash_file);
75       }
76     }
77   }
78   return process_sp;
79 }
80 
81 bool ProcessElfCore::CanDebug(lldb::TargetSP target_sp,
82                               bool plugin_specified_by_name) {
83   // For now we are just making sure the file exists for a given module
84   if (!m_core_module_sp && FileSystem::Instance().Exists(m_core_file)) {
85     ModuleSpec core_module_spec(m_core_file, target_sp->GetArchitecture());
86     Status error(ModuleList::GetSharedModule(core_module_spec, m_core_module_sp,
87                                              nullptr, nullptr, nullptr));
88     if (m_core_module_sp) {
89       ObjectFile *core_objfile = m_core_module_sp->GetObjectFile();
90       if (core_objfile && core_objfile->GetType() == ObjectFile::eTypeCoreFile)
91         return true;
92     }
93   }
94   return false;
95 }
96 
97 // ProcessElfCore constructor
98 ProcessElfCore::ProcessElfCore(lldb::TargetSP target_sp,
99                                lldb::ListenerSP listener_sp,
100                                const FileSpec &core_file)
101     : PostMortemProcess(target_sp, listener_sp), m_core_file(core_file) {}
102 
103 // Destructor
104 ProcessElfCore::~ProcessElfCore() {
105   Clear();
106   // We need to call finalize on the process before destroying ourselves to
107   // make sure all of the broadcaster cleanup goes as planned. If we destruct
108   // this class, then Process::~Process() might have problems trying to fully
109   // destroy the broadcaster.
110   Finalize();
111 }
112 
113 // PluginInterface
114 ConstString ProcessElfCore::GetPluginName() { return GetPluginNameStatic(); }
115 
116 uint32_t ProcessElfCore::GetPluginVersion() { return 1; }
117 
118 lldb::addr_t ProcessElfCore::AddAddressRangeFromLoadSegment(
119     const elf::ELFProgramHeader &header) {
120   const lldb::addr_t addr = header.p_vaddr;
121   FileRange file_range(header.p_offset, header.p_filesz);
122   VMRangeToFileOffset::Entry range_entry(addr, header.p_memsz, file_range);
123 
124   // Only add to m_core_aranges if the file size is non zero. Some core files
125   // have PT_LOAD segments for all address ranges, but set f_filesz to zero for
126   // the .text sections since they can be retrieved from the object files.
127   if (header.p_filesz > 0) {
128     VMRangeToFileOffset::Entry *last_entry = m_core_aranges.Back();
129     if (last_entry && last_entry->GetRangeEnd() == range_entry.GetRangeBase() &&
130         last_entry->data.GetRangeEnd() == range_entry.data.GetRangeBase() &&
131         last_entry->GetByteSize() == last_entry->data.GetByteSize()) {
132       last_entry->SetRangeEnd(range_entry.GetRangeEnd());
133       last_entry->data.SetRangeEnd(range_entry.data.GetRangeEnd());
134     } else {
135       m_core_aranges.Append(range_entry);
136     }
137   }
138   // Keep a separate map of permissions that that isn't coalesced so all ranges
139   // are maintained.
140   const uint32_t permissions =
141       ((header.p_flags & llvm::ELF::PF_R) ? lldb::ePermissionsReadable : 0u) |
142       ((header.p_flags & llvm::ELF::PF_W) ? lldb::ePermissionsWritable : 0u) |
143       ((header.p_flags & llvm::ELF::PF_X) ? lldb::ePermissionsExecutable : 0u);
144 
145   m_core_range_infos.Append(
146       VMRangeToPermissions::Entry(addr, header.p_memsz, permissions));
147 
148   return addr;
149 }
150 
151 // Process Control
152 Status ProcessElfCore::DoLoadCore() {
153   Status error;
154   if (!m_core_module_sp) {
155     error.SetErrorString("invalid core module");
156     return error;
157   }
158 
159   ObjectFileELF *core = (ObjectFileELF *)(m_core_module_sp->GetObjectFile());
160   if (core == nullptr) {
161     error.SetErrorString("invalid core object file");
162     return error;
163   }
164 
165   llvm::ArrayRef<elf::ELFProgramHeader> segments = core->ProgramHeaders();
166   if (segments.size() == 0) {
167     error.SetErrorString("core file has no segments");
168     return error;
169   }
170 
171   SetCanJIT(false);
172 
173   m_thread_data_valid = true;
174 
175   bool ranges_are_sorted = true;
176   lldb::addr_t vm_addr = 0;
177   /// Walk through segments and Thread and Address Map information.
178   /// PT_NOTE - Contains Thread and Register information
179   /// PT_LOAD - Contains a contiguous range of Process Address Space
180   for (const elf::ELFProgramHeader &H : segments) {
181     DataExtractor data = core->GetSegmentData(H);
182 
183     // Parse thread contexts and auxv structure
184     if (H.p_type == llvm::ELF::PT_NOTE) {
185       if (llvm::Error error = ParseThreadContextsFromNoteSegment(H, data))
186         return Status(std::move(error));
187     }
188     // PT_LOAD segments contains address map
189     if (H.p_type == llvm::ELF::PT_LOAD) {
190       lldb::addr_t last_addr = AddAddressRangeFromLoadSegment(H);
191       if (vm_addr > last_addr)
192         ranges_are_sorted = false;
193       vm_addr = last_addr;
194     }
195   }
196 
197   if (!ranges_are_sorted) {
198     m_core_aranges.Sort();
199     m_core_range_infos.Sort();
200   }
201 
202   // Even if the architecture is set in the target, we need to override it to
203   // match the core file which is always single arch.
204   ArchSpec arch(m_core_module_sp->GetArchitecture());
205 
206   ArchSpec target_arch = GetTarget().GetArchitecture();
207   ArchSpec core_arch(m_core_module_sp->GetArchitecture());
208   target_arch.MergeFrom(core_arch);
209   GetTarget().SetArchitecture(target_arch);
210 
211   SetUnixSignals(UnixSignals::Create(GetArchitecture()));
212 
213   // Ensure we found at least one thread that was stopped on a signal.
214   bool siginfo_signal_found = false;
215   bool prstatus_signal_found = false;
216   // Check we found a signal in a SIGINFO note.
217   for (const auto &thread_data : m_thread_data) {
218     if (thread_data.signo != 0)
219       siginfo_signal_found = true;
220     if (thread_data.prstatus_sig != 0)
221       prstatus_signal_found = true;
222   }
223   if (!siginfo_signal_found) {
224     // If we don't have signal from SIGINFO use the signal from each threads
225     // PRSTATUS note.
226     if (prstatus_signal_found) {
227       for (auto &thread_data : m_thread_data)
228         thread_data.signo = thread_data.prstatus_sig;
229     } else if (m_thread_data.size() > 0) {
230       // If all else fails force the first thread to be SIGSTOP
231       m_thread_data.begin()->signo =
232           GetUnixSignals()->GetSignalNumberFromName("SIGSTOP");
233     }
234   }
235 
236   // Core files are useless without the main executable. See if we can locate
237   // the main executable using data we found in the core file notes.
238   lldb::ModuleSP exe_module_sp = GetTarget().GetExecutableModule();
239   if (!exe_module_sp) {
240     // The first entry in the NT_FILE might be our executable
241     if (!m_nt_file_entries.empty()) {
242       ModuleSpec exe_module_spec;
243       exe_module_spec.GetArchitecture() = arch;
244       exe_module_spec.GetFileSpec().SetFile(
245           m_nt_file_entries[0].path.GetCString(), FileSpec::Style::native);
246       if (exe_module_spec.GetFileSpec()) {
247         exe_module_sp = GetTarget().GetOrCreateModule(exe_module_spec,
248                                                       true /* notify */);
249         if (exe_module_sp)
250           GetTarget().SetExecutableModule(exe_module_sp, eLoadDependentsNo);
251       }
252     }
253   }
254   return error;
255 }
256 
257 lldb_private::DynamicLoader *ProcessElfCore::GetDynamicLoader() {
258   if (m_dyld_up.get() == nullptr)
259     m_dyld_up.reset(DynamicLoader::FindPlugin(
260         this, DynamicLoaderPOSIXDYLD::GetPluginNameStatic().GetCString()));
261   return m_dyld_up.get();
262 }
263 
264 bool ProcessElfCore::DoUpdateThreadList(ThreadList &old_thread_list,
265                                         ThreadList &new_thread_list) {
266   const uint32_t num_threads = GetNumThreadContexts();
267   if (!m_thread_data_valid)
268     return false;
269 
270   for (lldb::tid_t tid = 0; tid < num_threads; ++tid) {
271     const ThreadData &td = m_thread_data[tid];
272     lldb::ThreadSP thread_sp(new ThreadElfCore(*this, td));
273     new_thread_list.AddThread(thread_sp);
274   }
275   return new_thread_list.GetSize(false) > 0;
276 }
277 
278 void ProcessElfCore::RefreshStateAfterStop() {}
279 
280 Status ProcessElfCore::DoDestroy() { return Status(); }
281 
282 // Process Queries
283 
284 bool ProcessElfCore::IsAlive() { return true; }
285 
286 // Process Memory
287 size_t ProcessElfCore::ReadMemory(lldb::addr_t addr, void *buf, size_t size,
288                                   Status &error) {
289   // Don't allow the caching that lldb_private::Process::ReadMemory does since
290   // in core files we have it all cached our our core file anyway.
291   return DoReadMemory(addr, buf, size, error);
292 }
293 
294 Status ProcessElfCore::GetMemoryRegionInfo(lldb::addr_t load_addr,
295                                            MemoryRegionInfo &region_info) {
296   region_info.Clear();
297   const VMRangeToPermissions::Entry *permission_entry =
298       m_core_range_infos.FindEntryThatContainsOrFollows(load_addr);
299   if (permission_entry) {
300     if (permission_entry->Contains(load_addr)) {
301       region_info.GetRange().SetRangeBase(permission_entry->GetRangeBase());
302       region_info.GetRange().SetRangeEnd(permission_entry->GetRangeEnd());
303       const Flags permissions(permission_entry->data);
304       region_info.SetReadable(permissions.Test(lldb::ePermissionsReadable)
305                                   ? MemoryRegionInfo::eYes
306                                   : MemoryRegionInfo::eNo);
307       region_info.SetWritable(permissions.Test(lldb::ePermissionsWritable)
308                                   ? MemoryRegionInfo::eYes
309                                   : MemoryRegionInfo::eNo);
310       region_info.SetExecutable(permissions.Test(lldb::ePermissionsExecutable)
311                                     ? MemoryRegionInfo::eYes
312                                     : MemoryRegionInfo::eNo);
313       region_info.SetMapped(MemoryRegionInfo::eYes);
314     } else if (load_addr < permission_entry->GetRangeBase()) {
315       region_info.GetRange().SetRangeBase(load_addr);
316       region_info.GetRange().SetRangeEnd(permission_entry->GetRangeBase());
317       region_info.SetReadable(MemoryRegionInfo::eNo);
318       region_info.SetWritable(MemoryRegionInfo::eNo);
319       region_info.SetExecutable(MemoryRegionInfo::eNo);
320       region_info.SetMapped(MemoryRegionInfo::eNo);
321     }
322     return Status();
323   }
324 
325   region_info.GetRange().SetRangeBase(load_addr);
326   region_info.GetRange().SetRangeEnd(LLDB_INVALID_ADDRESS);
327   region_info.SetReadable(MemoryRegionInfo::eNo);
328   region_info.SetWritable(MemoryRegionInfo::eNo);
329   region_info.SetExecutable(MemoryRegionInfo::eNo);
330   region_info.SetMapped(MemoryRegionInfo::eNo);
331   return Status();
332 }
333 
334 size_t ProcessElfCore::DoReadMemory(lldb::addr_t addr, void *buf, size_t size,
335                                     Status &error) {
336   ObjectFile *core_objfile = m_core_module_sp->GetObjectFile();
337 
338   if (core_objfile == nullptr)
339     return 0;
340 
341   // Get the address range
342   const VMRangeToFileOffset::Entry *address_range =
343       m_core_aranges.FindEntryThatContains(addr);
344   if (address_range == nullptr || address_range->GetRangeEnd() < addr) {
345     error.SetErrorStringWithFormat("core file does not contain 0x%" PRIx64,
346                                    addr);
347     return 0;
348   }
349 
350   // Convert the address into core file offset
351   const lldb::addr_t offset = addr - address_range->GetRangeBase();
352   const lldb::addr_t file_start = address_range->data.GetRangeBase();
353   const lldb::addr_t file_end = address_range->data.GetRangeEnd();
354   size_t bytes_to_read = size; // Number of bytes to read from the core file
355   size_t bytes_copied = 0;   // Number of bytes actually read from the core file
356   lldb::addr_t bytes_left =
357       0; // Number of bytes available in the core file from the given address
358 
359   // Don't proceed if core file doesn't contain the actual data for this
360   // address range.
361   if (file_start == file_end)
362     return 0;
363 
364   // Figure out how many on-disk bytes remain in this segment starting at the
365   // given offset
366   if (file_end > file_start + offset)
367     bytes_left = file_end - (file_start + offset);
368 
369   if (bytes_to_read > bytes_left)
370     bytes_to_read = bytes_left;
371 
372   // If there is data available on the core file read it
373   if (bytes_to_read)
374     bytes_copied =
375         core_objfile->CopyData(offset + file_start, bytes_to_read, buf);
376 
377   return bytes_copied;
378 }
379 
380 void ProcessElfCore::Clear() {
381   m_thread_list.Clear();
382 
383   SetUnixSignals(std::make_shared<UnixSignals>());
384 }
385 
386 void ProcessElfCore::Initialize() {
387   static llvm::once_flag g_once_flag;
388 
389   llvm::call_once(g_once_flag, []() {
390     PluginManager::RegisterPlugin(GetPluginNameStatic(),
391                                   GetPluginDescriptionStatic(), CreateInstance);
392   });
393 }
394 
395 lldb::addr_t ProcessElfCore::GetImageInfoAddress() {
396   ObjectFile *obj_file = GetTarget().GetExecutableModule()->GetObjectFile();
397   Address addr = obj_file->GetImageInfoAddress(&GetTarget());
398 
399   if (addr.IsValid())
400     return addr.GetLoadAddress(&GetTarget());
401   return LLDB_INVALID_ADDRESS;
402 }
403 
404 // Parse a FreeBSD NT_PRSTATUS note - see FreeBSD sys/procfs.h for details.
405 static void ParseFreeBSDPrStatus(ThreadData &thread_data,
406                                  const DataExtractor &data,
407                                  const ArchSpec &arch) {
408   lldb::offset_t offset = 0;
409   bool lp64 = (arch.GetMachine() == llvm::Triple::aarch64 ||
410                arch.GetMachine() == llvm::Triple::mips64 ||
411                arch.GetMachine() == llvm::Triple::ppc64 ||
412                arch.GetMachine() == llvm::Triple::x86_64);
413   int pr_version = data.GetU32(&offset);
414 
415   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_PROCESS));
416   if (log) {
417     if (pr_version > 1)
418       LLDB_LOGF(log, "FreeBSD PRSTATUS unexpected version %d", pr_version);
419   }
420 
421   // Skip padding, pr_statussz, pr_gregsetsz, pr_fpregsetsz, pr_osreldate
422   if (lp64)
423     offset += 32;
424   else
425     offset += 16;
426 
427   thread_data.signo = data.GetU32(&offset); // pr_cursig
428   thread_data.tid = data.GetU32(&offset);   // pr_pid
429   if (lp64)
430     offset += 4;
431 
432   size_t len = data.GetByteSize() - offset;
433   thread_data.gpregset = DataExtractor(data, offset, len);
434 }
435 
436 static llvm::Error ParseNetBSDProcInfo(const DataExtractor &data,
437                                        uint32_t &cpi_nlwps,
438                                        uint32_t &cpi_signo,
439                                        uint32_t &cpi_siglwp,
440                                        uint32_t &cpi_pid) {
441   lldb::offset_t offset = 0;
442 
443   uint32_t version = data.GetU32(&offset);
444   if (version != 1)
445     return llvm::make_error<llvm::StringError>(
446         "Error parsing NetBSD core(5) notes: Unsupported procinfo version",
447         llvm::inconvertibleErrorCode());
448 
449   uint32_t cpisize = data.GetU32(&offset);
450   if (cpisize != NETBSD::NT_PROCINFO_SIZE)
451     return llvm::make_error<llvm::StringError>(
452         "Error parsing NetBSD core(5) notes: Unsupported procinfo size",
453         llvm::inconvertibleErrorCode());
454 
455   cpi_signo = data.GetU32(&offset); /* killing signal */
456 
457   offset += NETBSD::NT_PROCINFO_CPI_SIGCODE_SIZE;
458   offset += NETBSD::NT_PROCINFO_CPI_SIGPEND_SIZE;
459   offset += NETBSD::NT_PROCINFO_CPI_SIGMASK_SIZE;
460   offset += NETBSD::NT_PROCINFO_CPI_SIGIGNORE_SIZE;
461   offset += NETBSD::NT_PROCINFO_CPI_SIGCATCH_SIZE;
462   cpi_pid = data.GetU32(&offset);
463   offset += NETBSD::NT_PROCINFO_CPI_PPID_SIZE;
464   offset += NETBSD::NT_PROCINFO_CPI_PGRP_SIZE;
465   offset += NETBSD::NT_PROCINFO_CPI_SID_SIZE;
466   offset += NETBSD::NT_PROCINFO_CPI_RUID_SIZE;
467   offset += NETBSD::NT_PROCINFO_CPI_EUID_SIZE;
468   offset += NETBSD::NT_PROCINFO_CPI_SVUID_SIZE;
469   offset += NETBSD::NT_PROCINFO_CPI_RGID_SIZE;
470   offset += NETBSD::NT_PROCINFO_CPI_EGID_SIZE;
471   offset += NETBSD::NT_PROCINFO_CPI_SVGID_SIZE;
472   cpi_nlwps = data.GetU32(&offset); /* number of LWPs */
473 
474   offset += NETBSD::NT_PROCINFO_CPI_NAME_SIZE;
475   cpi_siglwp = data.GetU32(&offset); /* LWP target of killing signal */
476 
477   return llvm::Error::success();
478 }
479 
480 static void ParseOpenBSDProcInfo(ThreadData &thread_data,
481                                  const DataExtractor &data) {
482   lldb::offset_t offset = 0;
483 
484   int version = data.GetU32(&offset);
485   if (version != 1)
486     return;
487 
488   offset += 4;
489   thread_data.signo = data.GetU32(&offset);
490 }
491 
492 llvm::Expected<std::vector<CoreNote>>
493 ProcessElfCore::parseSegment(const DataExtractor &segment) {
494   lldb::offset_t offset = 0;
495   std::vector<CoreNote> result;
496 
497   while (offset < segment.GetByteSize()) {
498     ELFNote note = ELFNote();
499     if (!note.Parse(segment, &offset))
500       return llvm::make_error<llvm::StringError>(
501           "Unable to parse note segment", llvm::inconvertibleErrorCode());
502 
503     size_t note_start = offset;
504     size_t note_size = llvm::alignTo(note.n_descsz, 4);
505     DataExtractor note_data(segment, note_start, note_size);
506 
507     result.push_back({note, note_data});
508     offset += note_size;
509   }
510 
511   return std::move(result);
512 }
513 
514 llvm::Error ProcessElfCore::parseFreeBSDNotes(llvm::ArrayRef<CoreNote> notes) {
515   bool have_prstatus = false;
516   bool have_prpsinfo = false;
517   ThreadData thread_data;
518   for (const auto &note : notes) {
519     if (note.info.n_name != "FreeBSD")
520       continue;
521 
522     if ((note.info.n_type == ELF::NT_PRSTATUS && have_prstatus) ||
523         (note.info.n_type == ELF::NT_PRPSINFO && have_prpsinfo)) {
524       assert(thread_data.gpregset.GetByteSize() > 0);
525       // Add the new thread to thread list
526       m_thread_data.push_back(thread_data);
527       thread_data = ThreadData();
528       have_prstatus = false;
529       have_prpsinfo = false;
530     }
531 
532     switch (note.info.n_type) {
533     case ELF::NT_PRSTATUS:
534       have_prstatus = true;
535       ParseFreeBSDPrStatus(thread_data, note.data, GetArchitecture());
536       break;
537     case ELF::NT_PRPSINFO:
538       have_prpsinfo = true;
539       break;
540     case ELF::NT_FREEBSD_THRMISC: {
541       lldb::offset_t offset = 0;
542       thread_data.name = note.data.GetCStr(&offset, 20);
543       break;
544     }
545     case ELF::NT_FREEBSD_PROCSTAT_AUXV:
546       // FIXME: FreeBSD sticks an int at the beginning of the note
547       m_auxv = DataExtractor(note.data, 4, note.data.GetByteSize() - 4);
548       break;
549     default:
550       thread_data.notes.push_back(note);
551       break;
552     }
553   }
554   if (!have_prstatus) {
555     return llvm::make_error<llvm::StringError>(
556         "Could not find NT_PRSTATUS note in core file.",
557         llvm::inconvertibleErrorCode());
558   }
559   m_thread_data.push_back(thread_data);
560   return llvm::Error::success();
561 }
562 
563 /// NetBSD specific Thread context from PT_NOTE segment
564 ///
565 /// NetBSD ELF core files use notes to provide information about
566 /// the process's state.  The note name is "NetBSD-CORE" for
567 /// information that is global to the process, and "NetBSD-CORE@nn",
568 /// where "nn" is the lwpid of the LWP that the information belongs
569 /// to (such as register state).
570 ///
571 /// NetBSD uses the following note identifiers:
572 ///
573 ///      ELF_NOTE_NETBSD_CORE_PROCINFO (value 1)
574 ///             Note is a "netbsd_elfcore_procinfo" structure.
575 ///      ELF_NOTE_NETBSD_CORE_AUXV     (value 2; since NetBSD 8.0)
576 ///             Note is an array of AuxInfo structures.
577 ///
578 /// NetBSD also uses ptrace(2) request numbers (the ones that exist in
579 /// machine-dependent space) to identify register info notes.  The
580 /// info in such notes is in the same format that ptrace(2) would
581 /// export that information.
582 ///
583 /// For more information see /usr/include/sys/exec_elf.h
584 ///
585 llvm::Error ProcessElfCore::parseNetBSDNotes(llvm::ArrayRef<CoreNote> notes) {
586   ThreadData thread_data;
587   bool had_nt_regs = false;
588 
589   // To be extracted from struct netbsd_elfcore_procinfo
590   // Used to sanity check of the LWPs of the process
591   uint32_t nlwps = 0;
592   uint32_t signo;  // killing signal
593   uint32_t siglwp; // LWP target of killing signal
594   uint32_t pr_pid;
595 
596   for (const auto &note : notes) {
597     llvm::StringRef name = note.info.n_name;
598 
599     if (name == "NetBSD-CORE") {
600       if (note.info.n_type == NETBSD::NT_PROCINFO) {
601         llvm::Error error = ParseNetBSDProcInfo(note.data, nlwps, signo,
602                                                 siglwp, pr_pid);
603         if (error)
604           return error;
605         SetID(pr_pid);
606       } else if (note.info.n_type == NETBSD::NT_AUXV) {
607         m_auxv = note.data;
608       }
609     } else if (name.consume_front("NetBSD-CORE@")) {
610       lldb::tid_t tid;
611       if (name.getAsInteger(10, tid))
612         return llvm::make_error<llvm::StringError>(
613             "Error parsing NetBSD core(5) notes: Cannot convert LWP ID "
614             "to integer",
615             llvm::inconvertibleErrorCode());
616 
617       switch (GetArchitecture().GetMachine()) {
618       case llvm::Triple::aarch64: {
619         // Assume order PT_GETREGS, PT_GETFPREGS
620         if (note.info.n_type == NETBSD::AARCH64::NT_REGS) {
621           // If this is the next thread, push the previous one first.
622           if (had_nt_regs) {
623             m_thread_data.push_back(thread_data);
624             thread_data = ThreadData();
625             had_nt_regs = false;
626           }
627 
628           thread_data.gpregset = note.data;
629           thread_data.tid = tid;
630           if (thread_data.gpregset.GetByteSize() == 0)
631             return llvm::make_error<llvm::StringError>(
632                 "Could not find general purpose registers note in core file.",
633                 llvm::inconvertibleErrorCode());
634           had_nt_regs = true;
635         } else if (note.info.n_type == NETBSD::AARCH64::NT_FPREGS) {
636           if (!had_nt_regs || tid != thread_data.tid)
637             return llvm::make_error<llvm::StringError>(
638                 "Error parsing NetBSD core(5) notes: Unexpected order "
639                 "of NOTEs PT_GETFPREG before PT_GETREG",
640                 llvm::inconvertibleErrorCode());
641           thread_data.notes.push_back(note);
642         }
643       } break;
644       case llvm::Triple::x86_64: {
645         // Assume order PT_GETREGS, PT_GETFPREGS
646         if (note.info.n_type == NETBSD::AMD64::NT_REGS) {
647           // If this is the next thread, push the previous one first.
648           if (had_nt_regs) {
649             m_thread_data.push_back(thread_data);
650             thread_data = ThreadData();
651             had_nt_regs = false;
652           }
653 
654           thread_data.gpregset = note.data;
655           thread_data.tid = tid;
656           if (thread_data.gpregset.GetByteSize() == 0)
657             return llvm::make_error<llvm::StringError>(
658                 "Could not find general purpose registers note in core file.",
659                 llvm::inconvertibleErrorCode());
660           had_nt_regs = true;
661         } else if (note.info.n_type == NETBSD::AMD64::NT_FPREGS) {
662           if (!had_nt_regs || tid != thread_data.tid)
663             return llvm::make_error<llvm::StringError>(
664                 "Error parsing NetBSD core(5) notes: Unexpected order "
665                 "of NOTEs PT_GETFPREG before PT_GETREG",
666                 llvm::inconvertibleErrorCode());
667           thread_data.notes.push_back(note);
668         }
669       } break;
670       default:
671         break;
672       }
673     }
674   }
675 
676   // Push the last thread.
677   if (had_nt_regs)
678     m_thread_data.push_back(thread_data);
679 
680   if (m_thread_data.empty())
681     return llvm::make_error<llvm::StringError>(
682         "Error parsing NetBSD core(5) notes: No threads information "
683         "specified in notes",
684         llvm::inconvertibleErrorCode());
685 
686   if (m_thread_data.size() != nlwps)
687     return llvm::make_error<llvm::StringError>(
688         "Error parsing NetBSD core(5) notes: Mismatch between the number "
689         "of LWPs in netbsd_elfcore_procinfo and the number of LWPs specified "
690         "by MD notes",
691         llvm::inconvertibleErrorCode());
692 
693   // Signal targeted at the whole process.
694   if (siglwp == 0) {
695     for (auto &data : m_thread_data)
696       data.signo = signo;
697   }
698   // Signal destined for a particular LWP.
699   else {
700     bool passed = false;
701 
702     for (auto &data : m_thread_data) {
703       if (data.tid == siglwp) {
704         data.signo = signo;
705         passed = true;
706         break;
707       }
708     }
709 
710     if (!passed)
711       return llvm::make_error<llvm::StringError>(
712           "Error parsing NetBSD core(5) notes: Signal passed to unknown LWP",
713           llvm::inconvertibleErrorCode());
714   }
715 
716   return llvm::Error::success();
717 }
718 
719 llvm::Error ProcessElfCore::parseOpenBSDNotes(llvm::ArrayRef<CoreNote> notes) {
720   ThreadData thread_data;
721   for (const auto &note : notes) {
722     // OpenBSD per-thread information is stored in notes named "OpenBSD@nnn" so
723     // match on the initial part of the string.
724     if (!llvm::StringRef(note.info.n_name).startswith("OpenBSD"))
725       continue;
726 
727     switch (note.info.n_type) {
728     case OPENBSD::NT_PROCINFO:
729       ParseOpenBSDProcInfo(thread_data, note.data);
730       break;
731     case OPENBSD::NT_AUXV:
732       m_auxv = note.data;
733       break;
734     case OPENBSD::NT_REGS:
735       thread_data.gpregset = note.data;
736       break;
737     default:
738       thread_data.notes.push_back(note);
739       break;
740     }
741   }
742   if (thread_data.gpregset.GetByteSize() == 0) {
743     return llvm::make_error<llvm::StringError>(
744         "Could not find general purpose registers note in core file.",
745         llvm::inconvertibleErrorCode());
746   }
747   m_thread_data.push_back(thread_data);
748   return llvm::Error::success();
749 }
750 
751 /// A description of a linux process usually contains the following NOTE
752 /// entries:
753 /// - NT_PRPSINFO - General process information like pid, uid, name, ...
754 /// - NT_SIGINFO - Information about the signal that terminated the process
755 /// - NT_AUXV - Process auxiliary vector
756 /// - NT_FILE - Files mapped into memory
757 ///
758 /// Additionally, for each thread in the process the core file will contain at
759 /// least the NT_PRSTATUS note, containing the thread id and general purpose
760 /// registers. It may include additional notes for other register sets (floating
761 /// point and vector registers, ...). The tricky part here is that some of these
762 /// notes have "CORE" in their owner fields, while other set it to "LINUX".
763 llvm::Error ProcessElfCore::parseLinuxNotes(llvm::ArrayRef<CoreNote> notes) {
764   const ArchSpec &arch = GetArchitecture();
765   bool have_prstatus = false;
766   bool have_prpsinfo = false;
767   ThreadData thread_data;
768   for (const auto &note : notes) {
769     if (note.info.n_name != "CORE" && note.info.n_name != "LINUX")
770       continue;
771 
772     if ((note.info.n_type == ELF::NT_PRSTATUS && have_prstatus) ||
773         (note.info.n_type == ELF::NT_PRPSINFO && have_prpsinfo)) {
774       assert(thread_data.gpregset.GetByteSize() > 0);
775       // Add the new thread to thread list
776       m_thread_data.push_back(thread_data);
777       thread_data = ThreadData();
778       have_prstatus = false;
779       have_prpsinfo = false;
780     }
781 
782     switch (note.info.n_type) {
783     case ELF::NT_PRSTATUS: {
784       have_prstatus = true;
785       ELFLinuxPrStatus prstatus;
786       Status status = prstatus.Parse(note.data, arch);
787       if (status.Fail())
788         return status.ToError();
789       thread_data.prstatus_sig = prstatus.pr_cursig;
790       thread_data.tid = prstatus.pr_pid;
791       uint32_t header_size = ELFLinuxPrStatus::GetSize(arch);
792       size_t len = note.data.GetByteSize() - header_size;
793       thread_data.gpregset = DataExtractor(note.data, header_size, len);
794       break;
795     }
796     case ELF::NT_PRPSINFO: {
797       have_prpsinfo = true;
798       ELFLinuxPrPsInfo prpsinfo;
799       Status status = prpsinfo.Parse(note.data, arch);
800       if (status.Fail())
801         return status.ToError();
802       thread_data.name.assign (prpsinfo.pr_fname, strnlen (prpsinfo.pr_fname, sizeof (prpsinfo.pr_fname)));
803       SetID(prpsinfo.pr_pid);
804       break;
805     }
806     case ELF::NT_SIGINFO: {
807       ELFLinuxSigInfo siginfo;
808       Status status = siginfo.Parse(note.data, arch);
809       if (status.Fail())
810         return status.ToError();
811       thread_data.signo = siginfo.si_signo;
812       break;
813     }
814     case ELF::NT_FILE: {
815       m_nt_file_entries.clear();
816       lldb::offset_t offset = 0;
817       const uint64_t count = note.data.GetAddress(&offset);
818       note.data.GetAddress(&offset); // Skip page size
819       for (uint64_t i = 0; i < count; ++i) {
820         NT_FILE_Entry entry;
821         entry.start = note.data.GetAddress(&offset);
822         entry.end = note.data.GetAddress(&offset);
823         entry.file_ofs = note.data.GetAddress(&offset);
824         m_nt_file_entries.push_back(entry);
825       }
826       for (uint64_t i = 0; i < count; ++i) {
827         const char *path = note.data.GetCStr(&offset);
828         if (path && path[0])
829           m_nt_file_entries[i].path.SetCString(path);
830       }
831       break;
832     }
833     case ELF::NT_AUXV:
834       m_auxv = note.data;
835       break;
836     default:
837       thread_data.notes.push_back(note);
838       break;
839     }
840   }
841   // Add last entry in the note section
842   if (have_prstatus)
843     m_thread_data.push_back(thread_data);
844   return llvm::Error::success();
845 }
846 
847 /// Parse Thread context from PT_NOTE segment and store it in the thread list
848 /// A note segment consists of one or more NOTE entries, but their types and
849 /// meaning differ depending on the OS.
850 llvm::Error ProcessElfCore::ParseThreadContextsFromNoteSegment(
851     const elf::ELFProgramHeader &segment_header, DataExtractor segment_data) {
852   assert(segment_header.p_type == llvm::ELF::PT_NOTE);
853 
854   auto notes_or_error = parseSegment(segment_data);
855   if(!notes_or_error)
856     return notes_or_error.takeError();
857   switch (GetArchitecture().GetTriple().getOS()) {
858   case llvm::Triple::FreeBSD:
859     return parseFreeBSDNotes(*notes_or_error);
860   case llvm::Triple::Linux:
861     return parseLinuxNotes(*notes_or_error);
862   case llvm::Triple::NetBSD:
863     return parseNetBSDNotes(*notes_or_error);
864   case llvm::Triple::OpenBSD:
865     return parseOpenBSDNotes(*notes_or_error);
866   default:
867     return llvm::make_error<llvm::StringError>(
868         "Don't know how to parse core file. Unsupported OS.",
869         llvm::inconvertibleErrorCode());
870   }
871 }
872 
873 uint32_t ProcessElfCore::GetNumThreadContexts() {
874   if (!m_thread_data_valid)
875     DoLoadCore();
876   return m_thread_data.size();
877 }
878 
879 ArchSpec ProcessElfCore::GetArchitecture() {
880   ArchSpec arch = m_core_module_sp->GetObjectFile()->GetArchitecture();
881 
882   ArchSpec target_arch = GetTarget().GetArchitecture();
883   arch.MergeFrom(target_arch);
884 
885   // On MIPS there is no way to differentiate betwenn 32bit and 64bit core
886   // files and this information can't be merged in from the target arch so we
887   // fail back to unconditionally returning the target arch in this config.
888   if (target_arch.IsMIPS()) {
889     return target_arch;
890   }
891 
892   return arch;
893 }
894 
895 DataExtractor ProcessElfCore::GetAuxvData() {
896   const uint8_t *start = m_auxv.GetDataStart();
897   size_t len = m_auxv.GetByteSize();
898   lldb::DataBufferSP buffer(new lldb_private::DataBufferHeap(start, len));
899   return DataExtractor(buffer, GetByteOrder(), GetAddressByteSize());
900 }
901 
902 bool ProcessElfCore::GetProcessInfo(ProcessInstanceInfo &info) {
903   info.Clear();
904   info.SetProcessID(GetID());
905   info.SetArchitecture(GetArchitecture());
906   lldb::ModuleSP module_sp = GetTarget().GetExecutableModule();
907   if (module_sp) {
908     const bool add_exe_file_as_first_arg = false;
909     info.SetExecutableFile(GetTarget().GetExecutableModule()->GetFileSpec(),
910                            add_exe_file_as_first_arg);
911   }
912   return true;
913 }
914