1 //===- SymbolTable.cpp ----------------------------------------------------===// 2 // 3 // The LLVM Linker 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // Symbol table is a bag of all known symbols. We put all symbols of 11 // all input files to the symbol table. The symbol table is basically 12 // a hash table with the logic to resolve symbol name conflicts using 13 // the symbol types. 14 // 15 //===----------------------------------------------------------------------===// 16 17 #include "SymbolTable.h" 18 #include "Config.h" 19 #include "Error.h" 20 #include "LinkerScript.h" 21 #include "Memory.h" 22 #include "SymbolListFile.h" 23 #include "Symbols.h" 24 25 using namespace llvm; 26 using namespace llvm::object; 27 using namespace llvm::ELF; 28 29 using namespace lld; 30 using namespace lld::elf; 31 32 // All input object files must be for the same architecture 33 // (e.g. it does not make sense to link x86 object files with 34 // MIPS object files.) This function checks for that error. 35 template <class ELFT> static bool isCompatible(InputFile *F) { 36 if (!isa<ELFFileBase<ELFT>>(F) && !isa<BitcodeFile>(F)) 37 return true; 38 if (F->EKind == Config->EKind && F->EMachine == Config->EMachine) { 39 if (Config->EMachine != EM_MIPS) 40 return true; 41 if (isMipsN32Abi(F) == Config->MipsN32Abi) 42 return true; 43 } 44 StringRef A = F->getName(); 45 StringRef B = Config->Emulation; 46 if (B.empty()) 47 B = Config->FirstElf->getName(); 48 error(A + " is incompatible with " + B); 49 return false; 50 } 51 52 // Add symbols in File to the symbol table. 53 template <class ELFT> void SymbolTable<ELFT>::addFile(InputFile *File) { 54 if (!isCompatible<ELFT>(File)) 55 return; 56 57 // Binary file 58 if (auto *F = dyn_cast<BinaryFile>(File)) { 59 BinaryFiles.push_back(F); 60 F->parse<ELFT>(); 61 return; 62 } 63 64 // .a file 65 if (auto *F = dyn_cast<ArchiveFile>(File)) { 66 F->parse<ELFT>(); 67 return; 68 } 69 70 // Lazy object file 71 if (auto *F = dyn_cast<LazyObjectFile>(File)) { 72 F->parse<ELFT>(); 73 return; 74 } 75 76 if (Config->Trace) 77 outs() << getFilename(File) << "\n"; 78 79 // .so file 80 if (auto *F = dyn_cast<SharedFile<ELFT>>(File)) { 81 // DSOs are uniquified not by filename but by soname. 82 F->parseSoName(); 83 if (HasError || !SoNames.insert(F->getSoName()).second) 84 return; 85 SharedFiles.push_back(F); 86 F->parseRest(); 87 return; 88 } 89 90 // LLVM bitcode file 91 if (auto *F = dyn_cast<BitcodeFile>(File)) { 92 BitcodeFiles.push_back(F); 93 F->parse<ELFT>(ComdatGroups); 94 return; 95 } 96 97 // Regular object file 98 auto *F = cast<ObjectFile<ELFT>>(File); 99 ObjectFiles.push_back(F); 100 F->parse(ComdatGroups); 101 } 102 103 // This function is where all the optimizations of link-time 104 // optimization happens. When LTO is in use, some input files are 105 // not in native object file format but in the LLVM bitcode format. 106 // This function compiles bitcode files into a few big native files 107 // using LLVM functions and replaces bitcode symbols with the results. 108 // Because all bitcode files that consist of a program are passed 109 // to the compiler at once, it can do whole-program optimization. 110 template <class ELFT> void SymbolTable<ELFT>::addCombinedLtoObject() { 111 if (BitcodeFiles.empty()) 112 return; 113 114 // Compile bitcode files and replace bitcode symbols. 115 Lto.reset(new BitcodeCompiler); 116 for (BitcodeFile *F : BitcodeFiles) 117 Lto->add(*F); 118 119 for (InputFile *File : Lto->compile()) { 120 ObjectFile<ELFT> *Obj = cast<ObjectFile<ELFT>>(File); 121 DenseSet<CachedHashStringRef> DummyGroups; 122 Obj->parse(DummyGroups); 123 ObjectFiles.push_back(Obj); 124 } 125 } 126 127 template <class ELFT> 128 DefinedRegular<ELFT> *SymbolTable<ELFT>::addAbsolute(StringRef Name, 129 uint8_t Visibility) { 130 Symbol *Sym = 131 addRegular(Name, Visibility, STT_NOTYPE, 0, 0, STB_GLOBAL, nullptr); 132 return cast<DefinedRegular<ELFT>>(Sym->body()); 133 } 134 135 // Add Name as an "ignored" symbol. An ignored symbol is a regular 136 // linker-synthesized defined symbol, but is only defined if needed. 137 template <class ELFT> 138 DefinedRegular<ELFT> *SymbolTable<ELFT>::addIgnored(StringRef Name, 139 uint8_t Visibility) { 140 if (!find(Name)) 141 return nullptr; 142 return addAbsolute(Name, Visibility); 143 } 144 145 // Set a flag for --trace-symbol so that we can print out a log message 146 // if a new symbol with the same name is inserted into the symbol table. 147 template <class ELFT> void SymbolTable<ELFT>::trace(StringRef Name) { 148 Symtab.insert({CachedHashStringRef(Name), {-1, true}}); 149 } 150 151 // Rename SYM as __wrap_SYM. The original symbol is preserved as __real_SYM. 152 // Used to implement --wrap. 153 template <class ELFT> void SymbolTable<ELFT>::wrap(StringRef Name) { 154 SymbolBody *B = find(Name); 155 if (!B) 156 return; 157 Symbol *Sym = B->symbol(); 158 Symbol *Real = addUndefined(Saver.save("__real_" + Name)); 159 Symbol *Wrap = addUndefined(Saver.save("__wrap_" + Name)); 160 161 // We rename symbols by replacing the old symbol's SymbolBody with the new 162 // symbol's SymbolBody. This causes all SymbolBody pointers referring to the 163 // old symbol to instead refer to the new symbol. 164 memcpy(Real->Body.buffer, Sym->Body.buffer, sizeof(Sym->Body)); 165 memcpy(Sym->Body.buffer, Wrap->Body.buffer, sizeof(Wrap->Body)); 166 } 167 168 static uint8_t getMinVisibility(uint8_t VA, uint8_t VB) { 169 if (VA == STV_DEFAULT) 170 return VB; 171 if (VB == STV_DEFAULT) 172 return VA; 173 return std::min(VA, VB); 174 } 175 176 // Parses a symbol in the form of <name>@<version> or <name>@@<version>. 177 static std::pair<StringRef, uint16_t> getSymbolVersion(StringRef S) { 178 if (Config->VersionDefinitions.empty()) 179 return {S, Config->DefaultSymbolVersion}; 180 181 size_t Pos = S.find('@'); 182 if (Pos == 0 || Pos == StringRef::npos) 183 return {S, Config->DefaultSymbolVersion}; 184 185 StringRef Name = S.substr(0, Pos); 186 StringRef Verstr = S.substr(Pos + 1); 187 if (Verstr.empty()) 188 return {S, Config->DefaultSymbolVersion}; 189 190 // '@@' in a symbol name means the default version. 191 // It is usually the most recent one. 192 bool IsDefault = (Verstr[0] == '@'); 193 if (IsDefault) 194 Verstr = Verstr.substr(1); 195 196 for (VersionDefinition &V : Config->VersionDefinitions) { 197 if (V.Name == Verstr) 198 return {Name, IsDefault ? V.Id : (V.Id | VERSYM_HIDDEN)}; 199 } 200 201 // It is an error if the specified version was not defined. 202 error("symbol " + S + " has undefined version " + Verstr); 203 return {S, Config->DefaultSymbolVersion}; 204 } 205 206 // Find an existing symbol or create and insert a new one. 207 template <class ELFT> 208 std::pair<Symbol *, bool> SymbolTable<ELFT>::insert(StringRef &Name) { 209 auto P = Symtab.insert( 210 {CachedHashStringRef(Name), SymIndex((int)SymVector.size(), false)}); 211 SymIndex &V = P.first->second; 212 bool IsNew = P.second; 213 214 if (V.Idx == -1) { 215 IsNew = true; 216 V = SymIndex((int)SymVector.size(), true); 217 } 218 219 Symbol *Sym; 220 if (IsNew) { 221 Sym = new (BAlloc) Symbol; 222 Sym->Binding = STB_WEAK; 223 Sym->Visibility = STV_DEFAULT; 224 Sym->IsUsedInRegularObj = false; 225 Sym->ExportDynamic = false; 226 Sym->Traced = V.Traced; 227 std::tie(Name, Sym->VersionId) = getSymbolVersion(Name); 228 SymVector.push_back(Sym); 229 } else { 230 Sym = SymVector[V.Idx]; 231 } 232 return {Sym, IsNew}; 233 } 234 235 // Construct a string in the form of "Sym in File1 and File2". 236 // Used to construct an error message. 237 static std::string conflictMsg(SymbolBody *Existing, InputFile *NewFile) { 238 return "'" + maybeDemangle(Existing->getName()) + "' in " + 239 getFilename(Existing->File) + " and " + getFilename(NewFile); 240 } 241 242 // Find an existing symbol or create and insert a new one, then apply the given 243 // attributes. 244 template <class ELFT> 245 std::pair<Symbol *, bool> 246 SymbolTable<ELFT>::insert(StringRef &Name, uint8_t Type, uint8_t Visibility, 247 bool CanOmitFromDynSym, InputFile *File) { 248 bool IsUsedInRegularObj = !File || File->kind() == InputFile::ObjectKind; 249 Symbol *S; 250 bool WasInserted; 251 std::tie(S, WasInserted) = insert(Name); 252 253 // Merge in the new symbol's visibility. 254 S->Visibility = getMinVisibility(S->Visibility, Visibility); 255 if (!CanOmitFromDynSym && (Config->Shared || Config->ExportDynamic)) 256 S->ExportDynamic = true; 257 if (IsUsedInRegularObj) 258 S->IsUsedInRegularObj = true; 259 if (!WasInserted && S->body()->Type != SymbolBody::UnknownType && 260 ((Type == STT_TLS) != S->body()->isTls())) 261 error("TLS attribute mismatch for symbol " + conflictMsg(S->body(), File)); 262 263 return {S, WasInserted}; 264 } 265 266 template <class ELFT> Symbol *SymbolTable<ELFT>::addUndefined(StringRef Name) { 267 return addUndefined(Name, STB_GLOBAL, STV_DEFAULT, /*Type*/ 0, 268 /*CanOmitFromDynSym*/ false, /*File*/ nullptr); 269 } 270 271 template <class ELFT> 272 Symbol *SymbolTable<ELFT>::addUndefined(StringRef Name, uint8_t Binding, 273 uint8_t StOther, uint8_t Type, 274 bool CanOmitFromDynSym, 275 InputFile *File) { 276 Symbol *S; 277 bool WasInserted; 278 std::tie(S, WasInserted) = 279 insert(Name, Type, StOther & 3, CanOmitFromDynSym, File); 280 if (WasInserted) { 281 S->Binding = Binding; 282 replaceBody<Undefined>(S, Name, StOther, Type, File); 283 return S; 284 } 285 if (Binding != STB_WEAK) { 286 if (S->body()->isShared() || S->body()->isLazy()) 287 S->Binding = Binding; 288 if (auto *SS = dyn_cast<SharedSymbol<ELFT>>(S->body())) 289 SS->file()->IsUsed = true; 290 } 291 if (auto *L = dyn_cast<Lazy>(S->body())) { 292 // An undefined weak will not fetch archive members, but we have to remember 293 // its type. See also comment in addLazyArchive. 294 if (S->isWeak()) 295 L->Type = Type; 296 else if (InputFile *F = L->fetch()) 297 addFile(F); 298 } 299 return S; 300 } 301 302 // We have a new defined symbol with the specified binding. Return 1 if the new 303 // symbol should win, -1 if the new symbol should lose, or 0 if both symbols are 304 // strong defined symbols. 305 static int compareDefined(Symbol *S, bool WasInserted, uint8_t Binding) { 306 if (WasInserted) 307 return 1; 308 SymbolBody *Body = S->body(); 309 if (Body->isLazy() || Body->isUndefined() || Body->isShared()) 310 return 1; 311 if (Binding == STB_WEAK) 312 return -1; 313 if (S->isWeak()) 314 return 1; 315 return 0; 316 } 317 318 // We have a new non-common defined symbol with the specified binding. Return 1 319 // if the new symbol should win, -1 if the new symbol should lose, or 0 if there 320 // is a conflict. If the new symbol wins, also update the binding. 321 static int compareDefinedNonCommon(Symbol *S, bool WasInserted, 322 uint8_t Binding) { 323 if (int Cmp = compareDefined(S, WasInserted, Binding)) { 324 if (Cmp > 0) 325 S->Binding = Binding; 326 return Cmp; 327 } 328 if (isa<DefinedCommon>(S->body())) { 329 // Non-common symbols take precedence over common symbols. 330 if (Config->WarnCommon) 331 warn("common " + S->body()->getName() + " is overridden"); 332 return 1; 333 } 334 return 0; 335 } 336 337 template <class ELFT> 338 Symbol *SymbolTable<ELFT>::addCommon(StringRef N, uint64_t Size, 339 uint64_t Alignment, uint8_t Binding, 340 uint8_t StOther, uint8_t Type, 341 InputFile *File) { 342 Symbol *S; 343 bool WasInserted; 344 std::tie(S, WasInserted) = 345 insert(N, Type, StOther & 3, /*CanOmitFromDynSym*/ false, File); 346 int Cmp = compareDefined(S, WasInserted, Binding); 347 if (Cmp > 0) { 348 S->Binding = Binding; 349 replaceBody<DefinedCommon>(S, N, Size, Alignment, StOther, Type, File); 350 } else if (Cmp == 0) { 351 auto *C = dyn_cast<DefinedCommon>(S->body()); 352 if (!C) { 353 // Non-common symbols take precedence over common symbols. 354 if (Config->WarnCommon) 355 warn("common " + S->body()->getName() + " is overridden"); 356 return S; 357 } 358 359 if (Config->WarnCommon) 360 warn("multiple common of " + S->body()->getName()); 361 362 Alignment = C->Alignment = std::max(C->Alignment, Alignment); 363 if (Size > C->Size) 364 replaceBody<DefinedCommon>(S, N, Size, Alignment, StOther, Type, File); 365 } 366 return S; 367 } 368 369 static void print(const Twine &Msg) { 370 if (Config->AllowMultipleDefinition) 371 warn(Msg); 372 else 373 error(Msg); 374 } 375 376 static void reportDuplicate(SymbolBody *Existing, InputFile *NewFile) { 377 print("duplicate symbol " + conflictMsg(Existing, NewFile)); 378 } 379 380 template <class ELFT> 381 static void reportDuplicate(SymbolBody *Existing, 382 InputSectionBase<ELFT> *ErrSec, 383 typename ELFT::uint ErrOffset) { 384 DefinedRegular<ELFT> *D = dyn_cast<DefinedRegular<ELFT>>(Existing); 385 if (!D || !D->Section || !ErrSec) { 386 reportDuplicate(Existing, ErrSec ? ErrSec->getFile() : nullptr); 387 return; 388 } 389 390 std::string OldLoc = getLocation(*D->Section, D->Value); 391 std::string NewLoc = getLocation(*ErrSec, ErrOffset); 392 393 print(NewLoc + ": duplicate symbol '" + maybeDemangle(Existing->getName()) + 394 "'"); 395 print(OldLoc + ": previous definition was here"); 396 } 397 398 template <typename ELFT> 399 Symbol *SymbolTable<ELFT>::addRegular(StringRef Name, const Elf_Sym &Sym, 400 InputSectionBase<ELFT> *Section) { 401 return addRegular(Name, Sym.st_other, Sym.getType(), Sym.st_value, 402 Sym.st_size, Sym.getBinding(), Section); 403 } 404 405 template <typename ELFT> 406 Symbol *SymbolTable<ELFT>::addRegular(StringRef Name, uint8_t StOther, 407 uint8_t Type, uintX_t Value, uintX_t Size, 408 uint8_t Binding, 409 InputSectionBase<ELFT> *Section) { 410 Symbol *S; 411 bool WasInserted; 412 std::tie(S, WasInserted) = insert(Name, Type, StOther & 3, 413 /*CanOmitFromDynSym*/ false, 414 Section ? Section->getFile() : nullptr); 415 int Cmp = compareDefinedNonCommon(S, WasInserted, Binding); 416 if (Cmp > 0) 417 replaceBody<DefinedRegular<ELFT>>(S, Name, StOther, Type, Value, Size, 418 Section); 419 else if (Cmp == 0) 420 reportDuplicate(S->body(), Section, Value); 421 return S; 422 } 423 424 template <typename ELFT> 425 Symbol *SymbolTable<ELFT>::addSynthetic(StringRef N, OutputSectionBase *Section, 426 uintX_t Value, uint8_t StOther) { 427 Symbol *S; 428 bool WasInserted; 429 std::tie(S, WasInserted) = insert(N, STT_NOTYPE, /*Visibility*/ StOther & 0x3, 430 /*CanOmitFromDynSym*/ false, nullptr); 431 int Cmp = compareDefinedNonCommon(S, WasInserted, STB_GLOBAL); 432 if (Cmp > 0) 433 replaceBody<DefinedSynthetic<ELFT>>(S, N, Value, Section); 434 else if (Cmp == 0) 435 reportDuplicate(S->body(), nullptr); 436 return S; 437 } 438 439 template <typename ELFT> 440 void SymbolTable<ELFT>::addShared(SharedFile<ELFT> *F, StringRef Name, 441 const Elf_Sym &Sym, 442 const typename ELFT::Verdef *Verdef) { 443 // DSO symbols do not affect visibility in the output, so we pass STV_DEFAULT 444 // as the visibility, which will leave the visibility in the symbol table 445 // unchanged. 446 Symbol *S; 447 bool WasInserted; 448 std::tie(S, WasInserted) = 449 insert(Name, Sym.getType(), STV_DEFAULT, /*CanOmitFromDynSym*/ true, F); 450 // Make sure we preempt DSO symbols with default visibility. 451 if (Sym.getVisibility() == STV_DEFAULT) 452 S->ExportDynamic = true; 453 if (WasInserted || isa<Undefined>(S->body())) { 454 replaceBody<SharedSymbol<ELFT>>(S, F, Name, Sym, Verdef); 455 if (!S->isWeak()) 456 F->IsUsed = true; 457 } 458 } 459 460 template <class ELFT> 461 Symbol *SymbolTable<ELFT>::addBitcode(StringRef Name, uint8_t Binding, 462 uint8_t StOther, uint8_t Type, 463 bool CanOmitFromDynSym, BitcodeFile *F) { 464 Symbol *S; 465 bool WasInserted; 466 std::tie(S, WasInserted) = 467 insert(Name, Type, StOther & 3, CanOmitFromDynSym, F); 468 int Cmp = compareDefinedNonCommon(S, WasInserted, Binding); 469 if (Cmp > 0) 470 replaceBody<DefinedRegular<ELFT>>(S, Name, StOther, Type, F); 471 else if (Cmp == 0) 472 reportDuplicate(S->body(), F); 473 return S; 474 } 475 476 template <class ELFT> SymbolBody *SymbolTable<ELFT>::find(StringRef Name) { 477 auto It = Symtab.find(CachedHashStringRef(Name)); 478 if (It == Symtab.end()) 479 return nullptr; 480 SymIndex V = It->second; 481 if (V.Idx == -1) 482 return nullptr; 483 return SymVector[V.Idx]->body(); 484 } 485 486 // Returns a list of defined symbols that match with a given regex. 487 template <class ELFT> 488 std::vector<SymbolBody *> SymbolTable<ELFT>::findAll(const StringMatcher &M) { 489 std::vector<SymbolBody *> Res; 490 for (Symbol *Sym : SymVector) { 491 SymbolBody *B = Sym->body(); 492 StringRef Name = B->getName(); 493 if (!B->isUndefined() && M.match(Name)) 494 Res.push_back(B); 495 } 496 return Res; 497 } 498 499 template <class ELFT> 500 void SymbolTable<ELFT>::addLazyArchive(ArchiveFile *F, 501 const object::Archive::Symbol Sym) { 502 Symbol *S; 503 bool WasInserted; 504 StringRef Name = Sym.getName(); 505 std::tie(S, WasInserted) = insert(Name); 506 if (WasInserted) { 507 replaceBody<LazyArchive>(S, *F, Sym, SymbolBody::UnknownType); 508 return; 509 } 510 if (!S->body()->isUndefined()) 511 return; 512 513 // Weak undefined symbols should not fetch members from archives. If we were 514 // to keep old symbol we would not know that an archive member was available 515 // if a strong undefined symbol shows up afterwards in the link. If a strong 516 // undefined symbol never shows up, this lazy symbol will get to the end of 517 // the link and must be treated as the weak undefined one. We already marked 518 // this symbol as used when we added it to the symbol table, but we also need 519 // to preserve its type. FIXME: Move the Type field to Symbol. 520 if (S->isWeak()) { 521 replaceBody<LazyArchive>(S, *F, Sym, S->body()->Type); 522 return; 523 } 524 std::pair<MemoryBufferRef, uint64_t> MBInfo = F->getMember(&Sym); 525 if (!MBInfo.first.getBuffer().empty()) 526 addFile(createObjectFile(MBInfo.first, F->getName(), MBInfo.second)); 527 } 528 529 template <class ELFT> 530 void SymbolTable<ELFT>::addLazyObject(StringRef Name, LazyObjectFile &Obj) { 531 Symbol *S; 532 bool WasInserted; 533 std::tie(S, WasInserted) = insert(Name); 534 if (WasInserted) { 535 replaceBody<LazyObject>(S, Name, Obj, SymbolBody::UnknownType); 536 return; 537 } 538 if (!S->body()->isUndefined()) 539 return; 540 541 // See comment for addLazyArchive above. 542 if (S->isWeak()) { 543 replaceBody<LazyObject>(S, Name, Obj, S->body()->Type); 544 } else { 545 MemoryBufferRef MBRef = Obj.getBuffer(); 546 if (!MBRef.getBuffer().empty()) 547 addFile(createObjectFile(MBRef)); 548 } 549 } 550 551 // Process undefined (-u) flags by loading lazy symbols named by those flags. 552 template <class ELFT> void SymbolTable<ELFT>::scanUndefinedFlags() { 553 for (StringRef S : Config->Undefined) 554 if (auto *L = dyn_cast_or_null<Lazy>(find(S))) 555 if (InputFile *File = L->fetch()) 556 addFile(File); 557 } 558 559 // This function takes care of the case in which shared libraries depend on 560 // the user program (not the other way, which is usual). Shared libraries 561 // may have undefined symbols, expecting that the user program provides 562 // the definitions for them. An example is BSD's __progname symbol. 563 // We need to put such symbols to the main program's .dynsym so that 564 // shared libraries can find them. 565 // Except this, we ignore undefined symbols in DSOs. 566 template <class ELFT> void SymbolTable<ELFT>::scanShlibUndefined() { 567 for (SharedFile<ELFT> *File : SharedFiles) 568 for (StringRef U : File->getUndefinedSymbols()) 569 if (SymbolBody *Sym = find(U)) 570 if (Sym->isDefined()) 571 Sym->symbol()->ExportDynamic = true; 572 } 573 574 // This function processes --export-dynamic-symbol and --dynamic-list. 575 template <class ELFT> void SymbolTable<ELFT>::scanDynamicList() { 576 for (StringRef S : Config->DynamicList) 577 if (SymbolBody *B = find(S)) 578 B->symbol()->ExportDynamic = true; 579 } 580 581 static void setVersionId(SymbolBody *Body, StringRef VersionName, 582 StringRef Name, uint16_t Version) { 583 if (!Body || Body->isUndefined()) { 584 if (Config->NoUndefinedVersion) 585 error("version script assignment of " + VersionName + " to symbol " + 586 Name + " failed: symbol not defined"); 587 return; 588 } 589 590 Symbol *Sym = Body->symbol(); 591 if (Sym->VersionId != Config->DefaultSymbolVersion) 592 warn("duplicate symbol " + Name + " in version script"); 593 Sym->VersionId = Version; 594 } 595 596 // Returns a map from demangled symbols to symbol objects. 597 // The relationship is 1:N instead of 1:1 because with the symbol 598 // versioning, more than one symbol may have the same name. 599 template <class ELFT> 600 std::map<std::string, std::vector<SymbolBody *>> 601 SymbolTable<ELFT>::getDemangledSyms() { 602 std::map<std::string, std::vector<SymbolBody *>> Result; 603 for (Symbol *Sym : SymVector) { 604 SymbolBody *B = Sym->body(); 605 Result[demangle(B->getName())].push_back(B); 606 } 607 return Result; 608 } 609 610 static bool hasExternCpp() { 611 for (VersionDefinition &V : Config->VersionDefinitions) 612 for (SymbolVersion Sym : V.Globals) 613 if (Sym.IsExternCpp) 614 return true; 615 return false; 616 } 617 618 static ArrayRef<SymbolBody *> 619 findDemangled(std::map<std::string, std::vector<SymbolBody *>> &D, 620 StringRef Name) { 621 auto I = D.find(Name); 622 if (I != D.end()) 623 return I->second; 624 return {}; 625 } 626 627 static std::vector<SymbolBody *> 628 findAllDemangled(const std::map<std::string, std::vector<SymbolBody *>> &D, 629 StringMatcher &M) { 630 std::vector<SymbolBody *> Res; 631 for (auto &P : D) { 632 if (M.match(P.first)) 633 for (SymbolBody *Body : P.second) 634 if (!Body->isUndefined()) 635 Res.push_back(Body); 636 } 637 return Res; 638 } 639 640 // If there's only one anonymous version definition in a version 641 // script file, the script does not actually define any symbol version, 642 // but just specifies symbols visibilities. We assume that the script was 643 // in the form of { global: foo; bar; local *; }. So, local is default. 644 // In this function, we make specified symbols global. 645 template <class ELFT> void SymbolTable<ELFT>::handleAnonymousVersion() { 646 std::vector<StringRef> Patterns; 647 for (SymbolVersion &Sym : Config->VersionScriptGlobals) { 648 if (hasWildcard(Sym.Name)) { 649 Patterns.push_back(Sym.Name); 650 continue; 651 } 652 if (SymbolBody *B = find(Sym.Name)) 653 B->symbol()->VersionId = VER_NDX_GLOBAL; 654 } 655 if (Patterns.empty()) 656 return; 657 StringMatcher M(Patterns); 658 std::vector<SymbolBody *> Syms = findAll(M); 659 for (SymbolBody *B : Syms) 660 B->symbol()->VersionId = VER_NDX_GLOBAL; 661 } 662 663 // This function processes version scripts by updating VersionId 664 // member of symbols. 665 template <class ELFT> void SymbolTable<ELFT>::scanVersionScript() { 666 // Handle edge cases first. 667 if (!Config->VersionScriptGlobals.empty()) { 668 handleAnonymousVersion(); 669 return; 670 } 671 672 if (Config->VersionDefinitions.empty()) 673 return; 674 675 // Now we have version definitions, so we need to set version ids to symbols. 676 // Each version definition has a glob pattern, and all symbols that match 677 // with the pattern get that version. 678 679 // Users can use "extern C++ {}" directive to match against demangled 680 // C++ symbols. For example, you can write a pattern such as 681 // "llvm::*::foo(int, ?)". Obviously, there's no way to handle this 682 // other than trying to match a regexp against all demangled symbols. 683 // So, if "extern C++" feature is used, we demangle all known symbols. 684 std::map<std::string, std::vector<SymbolBody *>> Demangled; 685 if (hasExternCpp()) 686 Demangled = getDemangledSyms(); 687 688 // First, we assign versions to exact matching symbols, 689 // i.e. version definitions not containing any glob meta-characters. 690 for (VersionDefinition &V : Config->VersionDefinitions) { 691 for (SymbolVersion Sym : V.Globals) { 692 if (Sym.HasWildcards) 693 continue; 694 695 StringRef N = Sym.Name; 696 if (Sym.IsExternCpp) { 697 for (SymbolBody *B : findDemangled(Demangled, N)) 698 setVersionId(B, V.Name, N, V.Id); 699 continue; 700 } 701 setVersionId(find(N), V.Name, N, V.Id); 702 } 703 for (SymbolVersion Sym : V.Locals) { 704 if (Sym.HasWildcards) 705 continue; 706 setVersionId(find(Sym.Name), V.Name, Sym.Name, VER_NDX_LOCAL); 707 } 708 } 709 710 // Next, we assign versions to fuzzy matching symbols, 711 // i.e. version definitions containing glob meta-characters. 712 // Note that because the last match takes precedence over previous matches, 713 // we iterate over the definitions in the reverse order. 714 auto assignFuzzyVersion = [&](SymbolVersion &Sym, size_t Version) { 715 if (!Sym.HasWildcards) 716 return; 717 StringMatcher M({Sym.Name}); 718 std::vector<SymbolBody *> Syms = 719 Sym.IsExternCpp ? findAllDemangled(Demangled, M) : findAll(M); 720 // Exact matching takes precendence over fuzzy matching, 721 // so we set a version to a symbol only if no version has been assigned 722 // to the symbol. This behavior is compatible with GNU. 723 for (SymbolBody *B : Syms) 724 if (B->symbol()->VersionId == Config->DefaultSymbolVersion) 725 B->symbol()->VersionId = Version; 726 }; 727 728 for (size_t I = Config->VersionDefinitions.size() - 1; I != (size_t)-1; --I) { 729 VersionDefinition &V = Config->VersionDefinitions[I]; 730 for (SymbolVersion &Sym : V.Locals) 731 assignFuzzyVersion(Sym, VER_NDX_LOCAL); 732 for (SymbolVersion &Sym : V.Globals) 733 assignFuzzyVersion(Sym, V.Id); 734 } 735 } 736 737 template class elf::SymbolTable<ELF32LE>; 738 template class elf::SymbolTable<ELF32BE>; 739 template class elf::SymbolTable<ELF64LE>; 740 template class elf::SymbolTable<ELF64BE>; 741