[ELF] Optimize parseSymbolVersion

We can just scan objectFiles and sharedFiles that have versioned symbols
to skip scanning the global symtab. While we won't suggest __wrap_foo
for undefined __wrap_foo@v1 when --wrap=foo@v1 is specified
(internalFile isn't scanned), this edge case difference is acceptable.

Author: MaskRay

lld/ELF/InputFiles.cpp

@@ -1217,6 +1217,7 @@ template <class ELFT> void ObjFile<ELFT>::postParse() {
       Err(ctx) << "TLS attribute mismatch: " << &sym << "\n>>> in " << sym.file
                << "\n>>> in " << this;
 
+    hasVersionSyms |= sym.hasVersionSuffix;
     // Handle non-COMMON defined symbol below. !sym.file allows a symbol
     // assignment to redefine a symbol without an error.
     if (!sym.isDefined() || secIdx == SHN_UNDEF)
@@ -1432,6 +1433,9 @@ template <class ELFT> void SharedFile::parse() {
   const Elf_Shdr *verdefSec = nullptr;
   const Elf_Shdr *verneedSec = nullptr;
 
+  numSymbols = numELFSyms;
+  symbols = std::make_unique<Symbol *[]>(numSymbols);
+
   // Search for .dynsym, .dynamic, .symtab, .gnu.version and .gnu.version_d.
   for (const Elf_Shdr &sec : sections) {
     switch (sec.sh_type) {
@@ -1591,6 +1595,8 @@ template <class ELFT> void SharedFile::parse() {
     s->dsoDefined = true;
     if (s->file == this)
       s->versionId = idx;
+    symbols[firstGlobal + i] = s;
+    hasVersionSyms = true;
   }
 }
 

lld/ELF/InputFiles.h

@@ -242,6 +242,7 @@ class ELFFileBase : public InputFile {
   StringRef sourceFile;
   uint32_t andFeatures = 0;
   bool hasCommonSyms = false;
+  bool hasVersionSyms = false;
   ArrayRef<uint8_t> aarch64PauthAbiCoreInfo;
 };
 

lld/ELF/SymbolTable.cpp

@@ -353,9 +353,21 @@ void SymbolTable::scanVersionScript() {
   // Symbol themselves might know their versions because symbols
   // can contain versions in the form of <name>@<version>.
   // Let them parse and update their names to exclude version suffix.
-  for (Symbol *sym : symVector)
-    if (sym->hasVersionSuffix)
-      sym->parseSymbolVersion(ctx);
+  for (ELFFileBase *file : ctx.objectFiles) {
+    if (!file->hasVersionSyms)
+      continue;
+    for (Symbol *sym : file->getGlobalSymbols())
+      if (sym->hasVersionSuffix)
+        sym->parseSymbolVersion(ctx);
+  }
+  // Only used for undefined symbol suggestion.
+  for (ELFFileBase *file : ctx.sharedFiles) {
+    if (!file->hasVersionSyms)
+      continue;
+    for (Symbol *sym : file->getGlobalSymbols())
+      if (sym && sym->hasVersionSuffix)
+        sym->parseSymbolVersion(ctx);
+  }
 
   // isPreemptible is false at this point. To correctly compute the binding of a
   // Defined (which is used by includeInDynsym(ctx)), we need to know if it is

lld/test/ELF/symver.s

@@ -139,8 +139,7 @@
 
 # W3:      error: undefined symbol: __wrap_foo@v1
 # W3-NEXT: >>> referenced by {{.*}}ref1.o:(.text+0x1)
-# W3-NEXT: >>> did you mean: __wrap_foo{{$}}
-# W3-NEXT: >>> defined in: {{.*}}wrap.o
+# W3-NOT:  {{.}}
 
 ## foo@v1 is correctly wrapped.
 # RUN: ld.lld -shared --soname=t --version-script=%t/ver --wrap=foo@v1 %t/ref.o %t/ref1.o %t/def1.o %t/wrap1.o -o %t.w4