[llvm] [llvm-profgen] Loading binary functions from .symtab when DWARF info is incomplete (PR #163654)

[via llvm-commits]

================
@@ -820,6 +830,63 @@ void ProfiledBinary::populateSymbolAddressList(const ObjectFile *Obj) {
   }
 }
 
+void ProfiledBinary::populateSymbolsFromBinary(const ObjectFile *Obj) {
+  // Load binary functions from symbol table when Debug info is incomplete.
+  // Strip the internal suffixes which are not reflected in the DWARF info.
+  const SmallVector<StringRef, 6> Suffixes(
+      {
+        // Internal suffixes from CoroSplit pass
+        ".cleanup", ".destroy", ".resume",
+        // Internal suffixes from Bolt
+        ".cold", ".warm",
+        // Compiler internal
+        ".llvm.",
+      });
+  StringRef FileName = Obj->getFileName();
+  for (const SymbolRef &Symbol : Obj->symbols()) {
+    const SymbolRef::Type Type = unwrapOrError(Symbol.getType(), FileName);
+    const uint64_t StartAddr = unwrapOrError(Symbol.getAddress(), FileName);
+    const StringRef Name = unwrapOrError(Symbol.getName(), FileName);
+    uint64_t Size = 0;
+    if (isa<ELFObjectFileBase>(Symbol.getObject())) {
+      ELFSymbolRef ElfSymbol(Symbol);
+      Size = ElfSymbol.getSize();
+    }
+
+    if (Size == 0 || Type != SymbolRef::ST_Function)
+      continue;
+
+    const StringRef SymName =
+        FunctionSamples::getCanonicalFnName(Name, Suffixes);
+
+    auto Ret = BinaryFunctions.emplace(SymName, BinaryFunction());
+    auto &Func = Ret.first->second;
+    if (Ret.second) {
+      Func.FuncName = Ret.first->first;
+      Func.FromSymtab = true;
+      HashBinaryFunctions[MD5Hash(StringRef(SymName))] = &Func;
+    }
+
+    if (auto Range = findFuncRange(StartAddr)) {
+      if (Ret.second && Range->getFuncName() != SymName && ShowDetailedWarning)
+        WithColor::warning()
+            << "Conflicting symbol " << Name << " already exists in DWARF as "
+            << Range->getFuncName() << " at address " << format("%8" PRIx64, StartAddr)
+            << ". The DWARF indicates a range from " << format("%8" PRIx64, Range->StartAddress) << " to "
+            << format("%8" PRIx64, Range->EndAddress) << "\n";
+    } else {
+      // Store/Update Function Range from SymTab
----------------
HighW4y2H3ll wrote:

Hmm, right.... "updating" is actually not an expected behavior. let me flag that in a warning too :)

https://github.com/llvm/llvm-project/pull/163654