Index: ELF/Relocations.h =================================================================== --- ELF/Relocations.h +++ ELF/Relocations.h @@ -133,14 +133,17 @@ uint32_t Pass = 0; - // Track Symbols that already have a Thunk - llvm::DenseMap ThunkedSymbols; + // Record all the available Thunks for a Symbol + llvm::DenseMap> ThunkedSymbols; // Find a Thunk from the Thunks symbol definition, we can use this to find // the Thunk from a relocation to the Thunks symbol definition. llvm::DenseMap Thunks; - // Track InputSections that have a ThunkSection placed in front + // Track InputSections that have an inline ThunkSection placed in front + // an inline ThunkSection may have control fall through to the section below + // so we need to make sure that there is only one of them. + // The Mips LA25 Thunk is an example of an inline ThunkSection. llvm::DenseMap ThunkedSections; // All the ThunkSections that we have created, organised by OutputSection Index: ELF/Relocations.cpp =================================================================== --- ELF/Relocations.cpp +++ ELF/Relocations.cpp @@ -1000,10 +1000,20 @@ template std::pair ThunkCreator::getThunk(SymbolBody &Body, uint32_t Type) { - auto res = ThunkedSymbols.insert({&Body, nullptr}); - if (res.second || !res.first->second->compatibleWith(Type)) - res.first->second = addThunk(Type, Body); - return std::make_pair(res.first->second, res.second); + Thunk *T = nullptr; + auto Res = ThunkedSymbols.insert({&Body, std::vector()}); + if (!Res.second) { + // Check existing Thunks for Body to see if they can be reused + for (Thunk *ET : Res.first->second) + if (ET->compatibleWith(Type)) { + T = ET; + return std::make_pair(T, Res.second); + } + } + // No existing compatible Thunk in range, create a new one + T = addThunk(Type, Body); + Res.first->second.push_back(T); + return std::make_pair(T, true); } // Make a new ThunkSection at offset Off and it to the persistent and per pass