diff options
Diffstat (limited to 'view')
| -rw-r--r-- | view/elf/elfview.cpp | 10 | ||||
| -rw-r--r-- | view/pe/peview.cpp | 2 |
2 files changed, 10 insertions, 2 deletions
diff --git a/view/elf/elfview.cpp b/view/elf/elfview.cpp index f938e978..a01d043b 100644 --- a/view/elf/elfview.cpp +++ b/view/elf/elfview.cpp @@ -1539,9 +1539,15 @@ bool ElfView::Init() entry += imageBaseAdjustment; Ref<Architecture> entryArch = entryPointArch->GetAssociatedArchitectureByAddress(entry); if (entryArch != entryPointArch) - AddFunctionForAnalysis(platform->GetRelatedPlatform(entryArch), entry); + { + auto func = AddFunctionForAnalysis(platform->GetRelatedPlatform(entryArch), entry); + AddToEntryFunctions(func); + } else - AddFunctionForAnalysis(platform, entry); + { + auto func = AddFunctionForAnalysis(platform, entry); + AddToEntryFunctions(func); + } m_logger->LogDebug("Adding function start: %#" PRIx64 "\n", entry); // name functions in .init_array, .fini_array, .ctors and .dtors diff --git a/view/pe/peview.cpp b/view/pe/peview.cpp index 89740413..bdf8175d 100644 --- a/view/pe/peview.cpp +++ b/view/pe/peview.cpp @@ -1920,6 +1920,8 @@ bool PEView::Init() m_logger->LogInfo("Found TLS entrypoint %s: 0x%" PRIx64, name, address); Ref<Platform> assPlatform = platform->GetAssociatedPlatformByAddress(address); AddPESymbol(FunctionSymbol, "", name, address - m_imageBase); + auto func = AddFunctionForAnalysis(platform, address); + AddToEntryFunctions(func); } else m_logger->LogInfo("Found TLS entrypoint %s: 0x%" PRIx64 " however it is not backed by file!", |