Daniel Dunbar | badeace | 2009-06-23 23:39:15 +0000 | [diff] [blame] | 1 | //===- lib/MC/MCContext.cpp - Machine Code Context ------------------------===// |
Daniel Dunbar | ca29e4d | 2009-06-23 22:01:43 +0000 | [diff] [blame] | 2 | // |
Chandler Carruth | 2946cd7 | 2019-01-19 08:50:56 +0000 | [diff] [blame^] | 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
Daniel Dunbar | ca29e4d | 2009-06-23 22:01:43 +0000 | [diff] [blame] | 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | |
Chandler Carruth | 6bda14b | 2017-06-06 11:49:48 +0000 | [diff] [blame] | 9 | #include "llvm/MC/MCContext.h" |
Scott Linder | 16c7bda | 2018-02-23 23:01:06 +0000 | [diff] [blame] | 10 | #include "llvm/ADT/Optional.h" |
Chris Lattner | 86dfd73 | 2009-10-19 22:49:00 +0000 | [diff] [blame] | 11 | #include "llvm/ADT/SmallString.h" |
Eugene Zelenko | d96089b | 2017-02-14 00:33:36 +0000 | [diff] [blame] | 12 | #include "llvm/ADT/SmallVector.h" |
| 13 | #include "llvm/ADT/StringMap.h" |
| 14 | #include "llvm/ADT/StringRef.h" |
Chris Lattner | 86dfd73 | 2009-10-19 22:49:00 +0000 | [diff] [blame] | 15 | #include "llvm/ADT/Twine.h" |
Zachary Turner | 264b5d9 | 2017-06-07 03:48:56 +0000 | [diff] [blame] | 16 | #include "llvm/BinaryFormat/COFF.h" |
| 17 | #include "llvm/BinaryFormat/ELF.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 18 | #include "llvm/MC/MCAsmInfo.h" |
Reid Kleckner | 2214ed8 | 2016-01-29 00:49:42 +0000 | [diff] [blame] | 19 | #include "llvm/MC/MCCodeView.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 20 | #include "llvm/MC/MCDwarf.h" |
Eugene Zelenko | d96089b | 2017-02-14 00:33:36 +0000 | [diff] [blame] | 21 | #include "llvm/MC/MCExpr.h" |
| 22 | #include "llvm/MC/MCFragment.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 23 | #include "llvm/MC/MCLabel.h" |
| 24 | #include "llvm/MC/MCObjectFileInfo.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 25 | #include "llvm/MC/MCSectionCOFF.h" |
| 26 | #include "llvm/MC/MCSectionELF.h" |
| 27 | #include "llvm/MC/MCSectionMachO.h" |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 28 | #include "llvm/MC/MCSectionWasm.h" |
Pete Cooper | ef21bd4 | 2015-03-04 01:24:11 +0000 | [diff] [blame] | 29 | #include "llvm/MC/MCStreamer.h" |
Eugene Zelenko | d96089b | 2017-02-14 00:33:36 +0000 | [diff] [blame] | 30 | #include "llvm/MC/MCSymbol.h" |
Pete Cooper | ad9f9c3 | 2015-06-08 17:17:12 +0000 | [diff] [blame] | 31 | #include "llvm/MC/MCSymbolCOFF.h" |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 32 | #include "llvm/MC/MCSymbolELF.h" |
Pete Cooper | eb012fa | 2015-06-08 17:17:23 +0000 | [diff] [blame] | 33 | #include "llvm/MC/MCSymbolMachO.h" |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 34 | #include "llvm/MC/MCSymbolWasm.h" |
Eugene Zelenko | d96089b | 2017-02-14 00:33:36 +0000 | [diff] [blame] | 35 | #include "llvm/MC/SectionKind.h" |
Chandler Carruth | 6bda14b | 2017-06-06 11:49:48 +0000 | [diff] [blame] | 36 | #include "llvm/Support/Casting.h" |
Igor Laevsky | 7b99885 | 2016-06-17 15:19:41 +0000 | [diff] [blame] | 37 | #include "llvm/Support/CommandLine.h" |
Jim Grosbach | b18b409 | 2012-01-26 23:20:11 +0000 | [diff] [blame] | 38 | #include "llvm/Support/ErrorHandling.h" |
Eric Christopher | 906da23 | 2012-12-18 00:31:01 +0000 | [diff] [blame] | 39 | #include "llvm/Support/MemoryBuffer.h" |
Jim Grosbach | b18b409 | 2012-01-26 23:20:11 +0000 | [diff] [blame] | 40 | #include "llvm/Support/Signals.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 41 | #include "llvm/Support/SourceMgr.h" |
Chandler Carruth | 6bda14b | 2017-06-06 11:49:48 +0000 | [diff] [blame] | 42 | #include "llvm/Support/raw_ostream.h" |
Eugene Zelenko | d96089b | 2017-02-14 00:33:36 +0000 | [diff] [blame] | 43 | #include <cassert> |
| 44 | #include <cstdlib> |
| 45 | #include <tuple> |
| 46 | #include <utility> |
David Blaikie | 15b25df | 2013-10-22 23:41:52 +0000 | [diff] [blame] | 47 | |
Daniel Dunbar | ca29e4d | 2009-06-23 22:01:43 +0000 | [diff] [blame] | 48 | using namespace llvm; |
| 49 | |
Igor Laevsky | 7b99885 | 2016-06-17 15:19:41 +0000 | [diff] [blame] | 50 | static cl::opt<char*> |
| 51 | AsSecureLogFileName("as-secure-log-file-name", |
| 52 | cl::desc("As secure log file name (initialized from " |
| 53 | "AS_SECURE_LOG_FILE env variable)"), |
| 54 | cl::init(getenv("AS_SECURE_LOG_FILE")), cl::Hidden); |
| 55 | |
Bill Wendling | bc07a89 | 2013-06-18 07:20:20 +0000 | [diff] [blame] | 56 | MCContext::MCContext(const MCAsmInfo *mai, const MCRegisterInfo *mri, |
Pedro Artigas | e84b13f | 2012-12-06 22:12:44 +0000 | [diff] [blame] | 57 | const MCObjectFileInfo *mofi, const SourceMgr *mgr, |
David Blaikie | 7400a97 | 2014-03-27 20:45:58 +0000 | [diff] [blame] | 58 | bool DoAutoReset) |
Evgeniy Stepanov | 0338ce8 | 2017-02-24 21:44:52 +0000 | [diff] [blame] | 59 | : SrcMgr(mgr), InlineSrcMgr(nullptr), MAI(mai), MRI(mri), MOFI(mofi), |
| 60 | Symbols(Allocator), UsedNames(Allocator), |
| 61 | CurrentDwarfLoc(0, 0, 0, DWARF2_FLAG_IS_STMT, 0, 0), |
Eli Friedman | a6e3a82 | 2018-10-12 19:41:05 +0000 | [diff] [blame] | 62 | AutoReset(DoAutoReset) { |
Igor Laevsky | 7b99885 | 2016-06-17 15:19:41 +0000 | [diff] [blame] | 63 | SecureLogFile = AsSecureLogFileName; |
Eric Christopher | 906da23 | 2012-12-18 00:31:01 +0000 | [diff] [blame] | 64 | |
Alp Toker | a55b95b | 2014-07-06 10:33:31 +0000 | [diff] [blame] | 65 | if (SrcMgr && SrcMgr->getNumBuffers()) |
| 66 | MainFileName = |
| 67 | SrcMgr->getMemoryBuffer(SrcMgr->getMainFileID())->getBufferIdentifier(); |
Pedro Artigas | e84b13f | 2012-12-06 22:12:44 +0000 | [diff] [blame] | 68 | } |
| 69 | |
| 70 | MCContext::~MCContext() { |
Pedro Artigas | 7212ee4 | 2012-12-12 22:59:46 +0000 | [diff] [blame] | 71 | if (AutoReset) |
| 72 | reset(); |
Pedro Artigas | e84b13f | 2012-12-06 22:12:44 +0000 | [diff] [blame] | 73 | |
| 74 | // NOTE: The symbols are all allocated out of a bump pointer allocator, |
| 75 | // we don't need to free them here. |
Pedro Artigas | e84b13f | 2012-12-06 22:12:44 +0000 | [diff] [blame] | 76 | } |
| 77 | |
| 78 | //===----------------------------------------------------------------------===// |
| 79 | // Module Lifetime Management |
| 80 | //===----------------------------------------------------------------------===// |
| 81 | |
Pedro Artigas | 7212ee4 | 2012-12-12 22:59:46 +0000 | [diff] [blame] | 82 | void MCContext::reset() { |
Rafael Espindola | ed34d58 | 2015-05-26 01:52:19 +0000 | [diff] [blame] | 83 | // Call the destructors so the fragments are freed |
Rafael Espindola | 4264e2d | 2015-10-07 19:08:19 +0000 | [diff] [blame] | 84 | COFFAllocator.DestroyAll(); |
| 85 | ELFAllocator.DestroyAll(); |
| 86 | MachOAllocator.DestroyAll(); |
Rafael Espindola | ed34d58 | 2015-05-26 01:52:19 +0000 | [diff] [blame] | 87 | |
Akira Hatanaka | b11ef08 | 2015-11-14 06:35:56 +0000 | [diff] [blame] | 88 | MCSubtargetAllocator.DestroyAll(); |
Pedro Artigas | e84b13f | 2012-12-06 22:12:44 +0000 | [diff] [blame] | 89 | UsedNames.clear(); |
| 90 | Symbols.clear(); |
| 91 | Allocator.Reset(); |
| 92 | Instances.clear(); |
Yaron Keren | 559b47d | 2014-09-17 09:25:36 +0000 | [diff] [blame] | 93 | CompilationDir.clear(); |
| 94 | MainFileName.clear(); |
David Blaikie | d9012ba | 2014-03-13 21:59:51 +0000 | [diff] [blame] | 95 | MCDwarfLineTablesCUMap.clear(); |
Rafael Espindola | e074679 | 2015-05-21 16:52:32 +0000 | [diff] [blame] | 96 | SectionsForRanges.clear(); |
Pedro Artigas | e84b13f | 2012-12-06 22:12:44 +0000 | [diff] [blame] | 97 | MCGenDwarfLabelEntries.clear(); |
| 98 | DwarfDebugFlags = StringRef(); |
Pedro Artigas | 7ba2edc | 2013-02-20 00:10:29 +0000 | [diff] [blame] | 99 | DwarfCompileUnitID = 0; |
Jim Grosbach | 008359a | 2015-05-18 18:43:23 +0000 | [diff] [blame] | 100 | CurrentDwarfLoc = MCDwarfLoc(0, 0, 0, DWARF2_FLAG_IS_STMT, 0, 0); |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 101 | |
Reid Kleckner | 2214ed8 | 2016-01-29 00:49:42 +0000 | [diff] [blame] | 102 | CVContext.reset(); |
| 103 | |
David Blaikie | 9ec3212 | 2014-04-10 23:55:11 +0000 | [diff] [blame] | 104 | MachOUniquingMap.clear(); |
| 105 | ELFUniquingMap.clear(); |
| 106 | COFFUniquingMap.clear(); |
Sam Clegg | 105bdc2 | 2018-05-30 02:57:20 +0000 | [diff] [blame] | 107 | WasmUniquingMap.clear(); |
Pedro Artigas | 7212ee4 | 2012-12-12 22:59:46 +0000 | [diff] [blame] | 108 | |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 109 | NextID.clear(); |
Pedro Artigas | 7212ee4 | 2012-12-12 22:59:46 +0000 | [diff] [blame] | 110 | AllowTemporaryLabels = true; |
| 111 | DwarfLocSeen = false; |
| 112 | GenDwarfForAssembly = false; |
| 113 | GenDwarfFileNumber = 0; |
Oliver Stannard | 07b43d3 | 2015-11-17 09:58:07 +0000 | [diff] [blame] | 114 | |
| 115 | HadError = false; |
Daniel Dunbar | ca29e4d | 2009-06-23 22:01:43 +0000 | [diff] [blame] | 116 | } |
| 117 | |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 118 | //===----------------------------------------------------------------------===// |
| 119 | // Symbol Manipulation |
| 120 | //===----------------------------------------------------------------------===// |
| 121 | |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 122 | MCSymbol *MCContext::getOrCreateSymbol(const Twine &Name) { |
Yaron Keren | 1ee89fc | 2015-03-17 09:51:17 +0000 | [diff] [blame] | 123 | SmallString<128> NameSV; |
| 124 | StringRef NameRef = Name.toStringRef(NameSV); |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 125 | |
Yaron Keren | 1ee89fc | 2015-03-17 09:51:17 +0000 | [diff] [blame] | 126 | assert(!NameRef.empty() && "Normal symbols cannot be unnamed!"); |
Rafael Espindola | 5fe5f45 | 2010-12-01 20:46:11 +0000 | [diff] [blame] | 127 | |
Yaron Keren | 1ee89fc | 2015-03-17 09:51:17 +0000 | [diff] [blame] | 128 | MCSymbol *&Sym = Symbols[NameRef]; |
David Blaikie | 5106ce7 | 2014-11-19 05:49:42 +0000 | [diff] [blame] | 129 | if (!Sym) |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 130 | Sym = createSymbol(NameRef, false, false); |
Rafael Espindola | 5fe5f45 | 2010-12-01 20:46:11 +0000 | [diff] [blame] | 131 | |
Rafael Espindola | 5fe5f45 | 2010-12-01 20:46:11 +0000 | [diff] [blame] | 132 | return Sym; |
| 133 | } |
| 134 | |
Reid Kleckner | cfb9ce5 | 2015-03-05 18:26:34 +0000 | [diff] [blame] | 135 | MCSymbol *MCContext::getOrCreateFrameAllocSymbol(StringRef FuncName, |
| 136 | unsigned Idx) { |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 137 | return getOrCreateSymbol(Twine(MAI->getPrivateGlobalPrefix()) + FuncName + |
Reid Kleckner | cfb9ce5 | 2015-03-05 18:26:34 +0000 | [diff] [blame] | 138 | "$frame_escape_" + Twine(Idx)); |
Reid Kleckner | e9b8931 | 2015-01-13 00:48:10 +0000 | [diff] [blame] | 139 | } |
| 140 | |
David Majnemer | a225a19 | 2015-03-31 22:35:44 +0000 | [diff] [blame] | 141 | MCSymbol *MCContext::getOrCreateParentFrameOffsetSymbol(StringRef FuncName) { |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 142 | return getOrCreateSymbol(Twine(MAI->getPrivateGlobalPrefix()) + FuncName + |
David Majnemer | a225a19 | 2015-03-31 22:35:44 +0000 | [diff] [blame] | 143 | "$parent_frame_offset"); |
| 144 | } |
| 145 | |
Reid Kleckner | 2632f0d | 2015-05-20 23:08:04 +0000 | [diff] [blame] | 146 | MCSymbol *MCContext::getOrCreateLSDASymbol(StringRef FuncName) { |
| 147 | return getOrCreateSymbol(Twine(MAI->getPrivateGlobalPrefix()) + "__ehtable$" + |
| 148 | FuncName); |
| 149 | } |
| 150 | |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 151 | MCSymbol *MCContext::createSymbolImpl(const StringMapEntry<bool> *Name, |
| 152 | bool IsTemporary) { |
Pete Cooper | ad9f9c3 | 2015-06-08 17:17:12 +0000 | [diff] [blame] | 153 | if (MOFI) { |
Rafael Espindola | dbaf049 | 2015-08-14 15:48:41 +0000 | [diff] [blame] | 154 | switch (MOFI->getObjectFileType()) { |
| 155 | case MCObjectFileInfo::IsCOFF: |
Pete Cooper | 234b875 | 2015-06-09 18:36:13 +0000 | [diff] [blame] | 156 | return new (Name, *this) MCSymbolCOFF(Name, IsTemporary); |
Rafael Espindola | dbaf049 | 2015-08-14 15:48:41 +0000 | [diff] [blame] | 157 | case MCObjectFileInfo::IsELF: |
Pete Cooper | 234b875 | 2015-06-09 18:36:13 +0000 | [diff] [blame] | 158 | return new (Name, *this) MCSymbolELF(Name, IsTemporary); |
Rafael Espindola | dbaf049 | 2015-08-14 15:48:41 +0000 | [diff] [blame] | 159 | case MCObjectFileInfo::IsMachO: |
Pete Cooper | 234b875 | 2015-06-09 18:36:13 +0000 | [diff] [blame] | 160 | return new (Name, *this) MCSymbolMachO(Name, IsTemporary); |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 161 | case MCObjectFileInfo::IsWasm: |
| 162 | return new (Name, *this) MCSymbolWasm(Name, IsTemporary); |
Pete Cooper | ad9f9c3 | 2015-06-08 17:17:12 +0000 | [diff] [blame] | 163 | } |
| 164 | } |
Pete Cooper | 234b875 | 2015-06-09 18:36:13 +0000 | [diff] [blame] | 165 | return new (Name, *this) MCSymbol(MCSymbol::SymbolKindUnset, Name, |
| 166 | IsTemporary); |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 167 | } |
| 168 | |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 169 | MCSymbol *MCContext::createSymbol(StringRef Name, bool AlwaysAddSuffix, |
| 170 | bool CanBeUnnamed) { |
| 171 | if (CanBeUnnamed && !UseNamesOnTempLabels) |
| 172 | return createSymbolImpl(nullptr, true); |
| 173 | |
Eric Christopher | 8e94895 | 2016-09-29 02:03:44 +0000 | [diff] [blame] | 174 | // Determine whether this is a user written assembler temporary or normal |
Rafael Espindola | 3e9e72a | 2015-06-02 22:52:13 +0000 | [diff] [blame] | 175 | // label, if used. |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 176 | bool IsTemporary = CanBeUnnamed; |
| 177 | if (AllowTemporaryLabels && !IsTemporary) |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 178 | IsTemporary = Name.startswith(MAI->getPrivateGlobalPrefix()); |
Rafael Espindola | 5fe5f45 | 2010-12-01 20:46:11 +0000 | [diff] [blame] | 179 | |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 180 | SmallString<128> NewName = Name; |
| 181 | bool AddSuffix = AlwaysAddSuffix; |
| 182 | unsigned &NextUniqueID = NextID[Name]; |
Eugene Zelenko | d96089b | 2017-02-14 00:33:36 +0000 | [diff] [blame] | 183 | while (true) { |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 184 | if (AddSuffix) { |
Benjamin Kramer | 2b6c96b | 2011-04-09 11:26:27 +0000 | [diff] [blame] | 185 | NewName.resize(Name.size()); |
| 186 | raw_svector_ostream(NewName) << NextUniqueID++; |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 187 | } |
| 188 | auto NameEntry = UsedNames.insert(std::make_pair(NewName, true)); |
Evgeniy Stepanov | a023f79 | 2016-03-28 20:36:28 +0000 | [diff] [blame] | 189 | if (NameEntry.second || !NameEntry.first->second) { |
| 190 | // Ok, we found a name. |
| 191 | // Mark it as used for a non-section symbol. |
| 192 | NameEntry.first->second = true; |
| 193 | // Have the MCSymbol object itself refer to the copy of the string that is |
| 194 | // embedded in the UsedNames entry. |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 195 | return createSymbolImpl(&*NameEntry.first, IsTemporary); |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 196 | } |
| 197 | assert(IsTemporary && "Cannot rename non-temporary symbols"); |
| 198 | AddSuffix = true; |
Rafael Espindola | 5fe5f45 | 2010-12-01 20:46:11 +0000 | [diff] [blame] | 199 | } |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 200 | llvm_unreachable("Infinite loop"); |
Chris Lattner | 3f5738d | 2009-06-24 04:31:49 +0000 | [diff] [blame] | 201 | } |
| 202 | |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 203 | MCSymbol *MCContext::createTempSymbol(const Twine &Name, bool AlwaysAddSuffix, |
| 204 | bool CanBeUnnamed) { |
Rafael Espindola | 629cdba | 2015-02-27 18:18:39 +0000 | [diff] [blame] | 205 | SmallString<128> NameSV; |
| 206 | raw_svector_ostream(NameSV) << MAI->getPrivateGlobalPrefix() << Name; |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 207 | return createSymbol(NameSV, AlwaysAddSuffix, CanBeUnnamed); |
Rafael Espindola | 629cdba | 2015-02-27 18:18:39 +0000 | [diff] [blame] | 208 | } |
| 209 | |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 210 | MCSymbol *MCContext::createLinkerPrivateTempSymbol() { |
Alp Toker | e69170a | 2014-06-26 22:52:05 +0000 | [diff] [blame] | 211 | SmallString<128> NameSV; |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 212 | raw_svector_ostream(NameSV) << MAI->getLinkerPrivateGlobalPrefix() << "tmp"; |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 213 | return createSymbol(NameSV, true, false); |
Tim Northover | c3988b4 | 2014-03-29 07:05:06 +0000 | [diff] [blame] | 214 | } |
| 215 | |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 216 | MCSymbol *MCContext::createTempSymbol(bool CanBeUnnamed) { |
| 217 | return createTempSymbol("tmp", true, CanBeUnnamed); |
Chris Lattner | 073d817 | 2010-03-14 08:23:30 +0000 | [diff] [blame] | 218 | } |
| 219 | |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 220 | unsigned MCContext::NextInstance(unsigned LocalLabelVal) { |
Benjamin Kramer | ab7be75 | 2010-05-18 12:15:34 +0000 | [diff] [blame] | 221 | MCLabel *&Label = Instances[LocalLabelVal]; |
| 222 | if (!Label) |
| 223 | Label = new (*this) MCLabel(0); |
| 224 | return Label->incInstance(); |
Kevin Enderby | 0510b48 | 2010-05-17 23:08:19 +0000 | [diff] [blame] | 225 | } |
| 226 | |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 227 | unsigned MCContext::GetInstance(unsigned LocalLabelVal) { |
Benjamin Kramer | ab7be75 | 2010-05-18 12:15:34 +0000 | [diff] [blame] | 228 | MCLabel *&Label = Instances[LocalLabelVal]; |
| 229 | if (!Label) |
| 230 | Label = new (*this) MCLabel(0); |
| 231 | return Label->getInstance(); |
Kevin Enderby | 0510b48 | 2010-05-17 23:08:19 +0000 | [diff] [blame] | 232 | } |
| 233 | |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 234 | MCSymbol *MCContext::getOrCreateDirectionalLocalSymbol(unsigned LocalLabelVal, |
| 235 | unsigned Instance) { |
| 236 | MCSymbol *&Sym = LocalSymbols[std::make_pair(LocalLabelVal, Instance)]; |
| 237 | if (!Sym) |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 238 | Sym = createTempSymbol(false); |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 239 | return Sym; |
Kevin Enderby | 0510b48 | 2010-05-17 23:08:19 +0000 | [diff] [blame] | 240 | } |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 241 | |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 242 | MCSymbol *MCContext::createDirectionalLocalSymbol(unsigned LocalLabelVal) { |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 243 | unsigned Instance = NextInstance(LocalLabelVal); |
| 244 | return getOrCreateDirectionalLocalSymbol(LocalLabelVal, Instance); |
| 245 | } |
| 246 | |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 247 | MCSymbol *MCContext::getDirectionalLocalSymbol(unsigned LocalLabelVal, |
Rafael Espindola | 4269b9e | 2014-03-13 18:09:26 +0000 | [diff] [blame] | 248 | bool Before) { |
| 249 | unsigned Instance = GetInstance(LocalLabelVal); |
| 250 | if (!Before) |
| 251 | ++Instance; |
| 252 | return getOrCreateDirectionalLocalSymbol(LocalLabelVal, Instance); |
Kevin Enderby | 0510b48 | 2010-05-17 23:08:19 +0000 | [diff] [blame] | 253 | } |
| 254 | |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 255 | MCSymbol *MCContext::lookupSymbol(const Twine &Name) const { |
Roman Divacky | 0be3359 | 2012-09-18 17:10:37 +0000 | [diff] [blame] | 256 | SmallString<128> NameSV; |
Yaron Keren | d7c546c | 2015-03-17 18:55:30 +0000 | [diff] [blame] | 257 | StringRef NameRef = Name.toStringRef(NameSV); |
| 258 | return Symbols.lookup(NameRef); |
Roman Divacky | 0be3359 | 2012-09-18 17:10:37 +0000 | [diff] [blame] | 259 | } |
| 260 | |
Mandeep Singh Grang | 9a561aa | 2016-12-06 02:49:17 +0000 | [diff] [blame] | 261 | void MCContext::setSymbolValue(MCStreamer &Streamer, |
| 262 | StringRef Sym, |
| 263 | uint64_t Val) { |
| 264 | auto Symbol = getOrCreateSymbol(Sym); |
| 265 | Streamer.EmitAssignment(Symbol, MCConstantExpr::create(Val, *this)); |
Mandeep Singh Grang | 3236007 | 2016-12-01 18:42:04 +0000 | [diff] [blame] | 266 | } |
| 267 | |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 268 | //===----------------------------------------------------------------------===// |
| 269 | // Section Management |
| 270 | //===----------------------------------------------------------------------===// |
| 271 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 272 | MCSectionMachO *MCContext::getMachOSection(StringRef Segment, StringRef Section, |
| 273 | unsigned TypeAndAttributes, |
| 274 | unsigned Reserved2, SectionKind Kind, |
| 275 | const char *BeginSymName) { |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 276 | // We unique sections by their segment/section pair. The returned section |
| 277 | // may not have the same flags as the requested section, if so this should be |
| 278 | // diagnosed by the client as an error. |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 279 | |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 280 | // Form the name to look up. |
| 281 | SmallString<64> Name; |
| 282 | Name += Segment; |
| 283 | Name.push_back(','); |
| 284 | Name += Section; |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 285 | |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 286 | // Do the lookup, if we have a hit, return it. |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 287 | MCSectionMachO *&Entry = MachOUniquingMap[Name]; |
Rafael Espindola | 6ed58a2 | 2015-03-10 21:16:18 +0000 | [diff] [blame] | 288 | if (Entry) |
| 289 | return Entry; |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 290 | |
Rafael Espindola | 6b9998b | 2015-03-10 22:00:25 +0000 | [diff] [blame] | 291 | MCSymbol *Begin = nullptr; |
| 292 | if (BeginSymName) |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 293 | Begin = createTempSymbol(BeginSymName, false); |
Rafael Espindola | 6b9998b | 2015-03-10 22:00:25 +0000 | [diff] [blame] | 294 | |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 295 | // Otherwise, return a new section. |
Rafael Espindola | 4264e2d | 2015-10-07 19:08:19 +0000 | [diff] [blame] | 296 | return Entry = new (MachOAllocator.Allocate()) MCSectionMachO( |
| 297 | Segment, Section, TypeAndAttributes, Reserved2, Kind, Begin); |
Chris Lattner | 2073112 | 2010-04-08 20:30:37 +0000 | [diff] [blame] | 298 | } |
Chris Lattner | 5418dd5 | 2010-04-08 21:26:26 +0000 | [diff] [blame] | 299 | |
Richard Smith | b910e56 | 2016-05-25 00:14:12 +0000 | [diff] [blame] | 300 | void MCContext::renameELFSection(MCSectionELF *Section, StringRef Name) { |
| 301 | StringRef GroupName; |
| 302 | if (const MCSymbol *Group = Section->getGroup()) |
| 303 | GroupName = Group->getName(); |
| 304 | |
| 305 | unsigned UniqueID = Section->getUniqueID(); |
| 306 | ELFUniquingMap.erase( |
| 307 | ELFSectionKey{Section->getSectionName(), GroupName, UniqueID}); |
| 308 | auto I = ELFUniquingMap.insert(std::make_pair( |
| 309 | ELFSectionKey{Name, GroupName, UniqueID}, |
| 310 | Section)) |
| 311 | .first; |
| 312 | StringRef CachedName = I->first.SectionName; |
| 313 | const_cast<MCSectionELF *>(Section)->setSectionName(CachedName); |
| 314 | } |
| 315 | |
Rafael Espindola | 13a79bb | 2017-02-02 21:26:06 +0000 | [diff] [blame] | 316 | MCSectionELF *MCContext::createELFSectionImpl(StringRef Section, unsigned Type, |
| 317 | unsigned Flags, SectionKind K, |
| 318 | unsigned EntrySize, |
| 319 | const MCSymbolELF *Group, |
| 320 | unsigned UniqueID, |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 321 | const MCSymbolELF *Associated) { |
Rafael Espindola | 13a79bb | 2017-02-02 21:26:06 +0000 | [diff] [blame] | 322 | MCSymbolELF *R; |
| 323 | MCSymbol *&Sym = Symbols[Section]; |
Evgeniy Stepanov | 00400d3 | 2017-02-24 21:44:58 +0000 | [diff] [blame] | 324 | // A section symbol can not redefine regular symbols. There may be multiple |
| 325 | // sections with the same name, in which case the first such section wins. |
| 326 | if (Sym && Sym->isDefined() && |
| 327 | (!Sym->isInSection() || Sym->getSection().getBeginSymbol() != Sym)) |
| 328 | reportError(SMLoc(), "invalid symbol redefinition"); |
Rafael Espindola | 13a79bb | 2017-02-02 21:26:06 +0000 | [diff] [blame] | 329 | if (Sym && Sym->isUndefined()) { |
| 330 | R = cast<MCSymbolELF>(Sym); |
| 331 | } else { |
| 332 | auto NameIter = UsedNames.insert(std::make_pair(Section, false)).first; |
| 333 | R = new (&*NameIter, *this) MCSymbolELF(&*NameIter, /*isTemporary*/ false); |
| 334 | if (!Sym) |
| 335 | Sym = R; |
| 336 | } |
| 337 | R->setBinding(ELF::STB_LOCAL); |
| 338 | R->setType(ELF::STT_SECTION); |
Rafael Espindola | 13a79bb | 2017-02-02 21:26:06 +0000 | [diff] [blame] | 339 | |
| 340 | auto *Ret = new (ELFAllocator.Allocate()) MCSectionELF( |
| 341 | Section, Type, Flags, K, EntrySize, Group, UniqueID, R, Associated); |
| 342 | |
| 343 | auto *F = new MCDataFragment(); |
| 344 | Ret->getFragmentList().insert(Ret->begin(), F); |
| 345 | F->setParent(Ret); |
| 346 | R->setFragment(F); |
| 347 | |
| 348 | return Ret; |
| 349 | } |
| 350 | |
Eric Christopher | 36e601c | 2016-07-01 06:07:38 +0000 | [diff] [blame] | 351 | MCSectionELF *MCContext::createELFRelSection(const Twine &Name, unsigned Type, |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 352 | unsigned Flags, unsigned EntrySize, |
Rafael Espindola | 0ccf9b7 | 2015-06-02 21:30:13 +0000 | [diff] [blame] | 353 | const MCSymbolELF *Group, |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 354 | const MCSectionELF *RelInfoSection) { |
Rafael Espindola | c9d0692 | 2015-03-30 13:39:16 +0000 | [diff] [blame] | 355 | StringMap<bool>::iterator I; |
| 356 | bool Inserted; |
Eric Christopher | 36e601c | 2016-07-01 06:07:38 +0000 | [diff] [blame] | 357 | std::tie(I, Inserted) = |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 358 | RelSecNames.insert(std::make_pair(Name.str(), true)); |
Rafael Espindola | c9d0692 | 2015-03-30 13:39:16 +0000 | [diff] [blame] | 359 | |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 360 | return createELFSectionImpl( |
| 361 | I->getKey(), Type, Flags, SectionKind::getReadOnly(), EntrySize, Group, |
| 362 | true, cast<MCSymbolELF>(RelInfoSection->getBeginSymbol())); |
Rafael Espindola | c9d0692 | 2015-03-30 13:39:16 +0000 | [diff] [blame] | 363 | } |
| 364 | |
Eric Christopher | 36e601c | 2016-07-01 06:07:38 +0000 | [diff] [blame] | 365 | MCSectionELF *MCContext::getELFNamedSection(const Twine &Prefix, |
| 366 | const Twine &Suffix, unsigned Type, |
| 367 | unsigned Flags, |
| 368 | unsigned EntrySize) { |
| 369 | return getELFSection(Prefix + "." + Suffix, Type, Flags, EntrySize, Suffix); |
| 370 | } |
| 371 | |
| 372 | MCSectionELF *MCContext::getELFSection(const Twine &Section, unsigned Type, |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 373 | unsigned Flags, unsigned EntrySize, |
Rafael Espindola | dc1c301 | 2017-02-09 14:59:20 +0000 | [diff] [blame] | 374 | const Twine &Group, unsigned UniqueID, |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 375 | const MCSymbolELF *Associated) { |
Rafael Espindola | 0ccf9b7 | 2015-06-02 21:30:13 +0000 | [diff] [blame] | 376 | MCSymbolELF *GroupSym = nullptr; |
Eric Christopher | 36e601c | 2016-07-01 06:07:38 +0000 | [diff] [blame] | 377 | if (!Group.isTriviallyEmpty() && !Group.str().empty()) |
Rafael Espindola | 0ccf9b7 | 2015-06-02 21:30:13 +0000 | [diff] [blame] | 378 | GroupSym = cast<MCSymbolELF>(getOrCreateSymbol(Group)); |
Rafael Espindola | d3ac79b | 2015-03-30 13:59:06 +0000 | [diff] [blame] | 379 | |
Rafael Espindola | 61e8ce3 | 2015-04-06 04:25:18 +0000 | [diff] [blame] | 380 | return getELFSection(Section, Type, Flags, EntrySize, GroupSym, UniqueID, |
Rafael Espindola | dc1c301 | 2017-02-09 14:59:20 +0000 | [diff] [blame] | 381 | Associated); |
Rafael Espindola | 61e8ce3 | 2015-04-06 04:25:18 +0000 | [diff] [blame] | 382 | } |
| 383 | |
Eric Christopher | 36e601c | 2016-07-01 06:07:38 +0000 | [diff] [blame] | 384 | MCSectionELF *MCContext::getELFSection(const Twine &Section, unsigned Type, |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 385 | unsigned Flags, unsigned EntrySize, |
Rafael Espindola | 0ccf9b7 | 2015-06-02 21:30:13 +0000 | [diff] [blame] | 386 | const MCSymbolELF *GroupSym, |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 387 | unsigned UniqueID, |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 388 | const MCSymbolELF *Associated) { |
Rafael Espindola | 61e8ce3 | 2015-04-06 04:25:18 +0000 | [diff] [blame] | 389 | StringRef Group = ""; |
| 390 | if (GroupSym) |
| 391 | Group = GroupSym->getName(); |
Chris Lattner | 5418dd5 | 2010-04-08 21:26:26 +0000 | [diff] [blame] | 392 | // Do the lookup, if we have a hit, return it. |
David Blaikie | 9ec3212 | 2014-04-10 23:55:11 +0000 | [diff] [blame] | 393 | auto IterBool = ELFUniquingMap.insert( |
Eric Christopher | 36e601c | 2016-07-01 06:07:38 +0000 | [diff] [blame] | 394 | std::make_pair(ELFSectionKey{Section.str(), Group, UniqueID}, nullptr)); |
David Blaikie | 9ec3212 | 2014-04-10 23:55:11 +0000 | [diff] [blame] | 395 | auto &Entry = *IterBool.first; |
Rafael Espindola | 8ca44f0 | 2015-04-04 18:02:01 +0000 | [diff] [blame] | 396 | if (!IterBool.second) |
Rafael Espindola | 68fa249 | 2015-02-17 20:48:01 +0000 | [diff] [blame] | 397 | return Entry.second; |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 398 | |
Rafael Espindola | 44d5057 | 2015-03-27 21:34:24 +0000 | [diff] [blame] | 399 | StringRef CachedName = Entry.first.SectionName; |
Rafael Espindola | ba31e27 | 2015-01-29 17:33:21 +0000 | [diff] [blame] | 400 | |
| 401 | SectionKind Kind; |
Prakhar Bahuguna | 52a7dd7 | 2016-12-15 07:59:08 +0000 | [diff] [blame] | 402 | if (Flags & ELF::SHF_ARM_PURECODE) |
| 403 | Kind = SectionKind::getExecuteOnly(); |
| 404 | else if (Flags & ELF::SHF_EXECINSTR) |
Rafael Espindola | ba31e27 | 2015-01-29 17:33:21 +0000 | [diff] [blame] | 405 | Kind = SectionKind::getText(); |
| 406 | else |
| 407 | Kind = SectionKind::getReadOnly(); |
| 408 | |
George Rimar | 9fbecc9 | 2018-08-29 09:04:52 +0000 | [diff] [blame] | 409 | MCSectionELF *Result = createELFSectionImpl( |
| 410 | CachedName, Type, Flags, Kind, EntrySize, GroupSym, UniqueID, Associated); |
| 411 | Entry.second = Result; |
| 412 | return Result; |
Chris Lattner | 5418dd5 | 2010-04-08 21:26:26 +0000 | [diff] [blame] | 413 | } |
| 414 | |
Rafael Espindola | 0ccf9b7 | 2015-06-02 21:30:13 +0000 | [diff] [blame] | 415 | MCSectionELF *MCContext::createELFGroupSection(const MCSymbolELF *Group) { |
Rafael Espindola | 13a79bb | 2017-02-02 21:26:06 +0000 | [diff] [blame] | 416 | return createELFSectionImpl(".group", ELF::SHT_GROUP, 0, |
| 417 | SectionKind::getReadOnly(), 4, Group, ~0, |
| 418 | nullptr); |
Rafael Espindola | a3e9a22 | 2010-11-11 18:13:52 +0000 | [diff] [blame] | 419 | } |
| 420 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 421 | MCSectionCOFF *MCContext::getCOFFSection(StringRef Section, |
| 422 | unsigned Characteristics, |
| 423 | SectionKind Kind, |
| 424 | StringRef COMDATSymName, int Selection, |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 425 | unsigned UniqueID, |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 426 | const char *BeginSymName) { |
Rafael Espindola | d3ac79b | 2015-03-30 13:59:06 +0000 | [diff] [blame] | 427 | MCSymbol *COMDATSymbol = nullptr; |
| 428 | if (!COMDATSymName.empty()) { |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 429 | COMDATSymbol = getOrCreateSymbol(COMDATSymName); |
Rafael Espindola | d3ac79b | 2015-03-30 13:59:06 +0000 | [diff] [blame] | 430 | COMDATSymName = COMDATSymbol->getName(); |
| 431 | } |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 432 | |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 433 | |
Rafael Espindola | d3ac79b | 2015-03-30 13:59:06 +0000 | [diff] [blame] | 434 | // Do the lookup, if we have a hit, return it. |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 435 | COFFSectionKey T{Section, COMDATSymName, Selection, UniqueID}; |
David Majnemer | c57d038 | 2014-06-27 17:19:44 +0000 | [diff] [blame] | 436 | auto IterBool = COFFUniquingMap.insert(std::make_pair(T, nullptr)); |
David Blaikie | 9ec3212 | 2014-04-10 23:55:11 +0000 | [diff] [blame] | 437 | auto Iter = IterBool.first; |
| 438 | if (!IterBool.second) |
Rafael Espindola | 60ec383 | 2013-11-19 19:52:52 +0000 | [diff] [blame] | 439 | return Iter->second; |
Michael J. Spencer | f13f442 | 2010-11-26 04:16:08 +0000 | [diff] [blame] | 440 | |
Rafael Espindola | 6b9998b | 2015-03-10 22:00:25 +0000 | [diff] [blame] | 441 | MCSymbol *Begin = nullptr; |
| 442 | if (BeginSymName) |
Rafael Espindola | 9ab0923 | 2015-03-17 20:07:06 +0000 | [diff] [blame] | 443 | Begin = createTempSymbol(BeginSymName, false); |
Rafael Espindola | 6b9998b | 2015-03-10 22:00:25 +0000 | [diff] [blame] | 444 | |
Rafael Espindola | 44d5057 | 2015-03-27 21:34:24 +0000 | [diff] [blame] | 445 | StringRef CachedName = Iter->first.SectionName; |
Rafael Espindola | 4264e2d | 2015-10-07 19:08:19 +0000 | [diff] [blame] | 446 | MCSectionCOFF *Result = new (COFFAllocator.Allocate()) MCSectionCOFF( |
Rafael Espindola | 6b9998b | 2015-03-10 22:00:25 +0000 | [diff] [blame] | 447 | CachedName, Characteristics, COMDATSymbol, Selection, Kind, Begin); |
Rafael Espindola | 60ec383 | 2013-11-19 19:52:52 +0000 | [diff] [blame] | 448 | |
| 449 | Iter->second = Result; |
Chris Lattner | 87cffa9 | 2010-05-07 17:17:41 +0000 | [diff] [blame] | 450 | return Result; |
| 451 | } |
Kevin Enderby | e5930f1 | 2010-07-28 20:55:35 +0000 | [diff] [blame] | 452 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 453 | MCSectionCOFF *MCContext::getCOFFSection(StringRef Section, |
| 454 | unsigned Characteristics, |
| 455 | SectionKind Kind, |
| 456 | const char *BeginSymName) { |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 457 | return getCOFFSection(Section, Characteristics, Kind, "", 0, GenericSectionID, |
| 458 | BeginSymName); |
Rafael Espindola | 60ec383 | 2013-11-19 19:52:52 +0000 | [diff] [blame] | 459 | } |
| 460 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 461 | MCSectionCOFF *MCContext::getCOFFSection(StringRef Section) { |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 462 | COFFSectionKey T{Section, "", 0, GenericSectionID}; |
David Majnemer | c57d038 | 2014-06-27 17:19:44 +0000 | [diff] [blame] | 463 | auto Iter = COFFUniquingMap.find(T); |
David Blaikie | 9ec3212 | 2014-04-10 23:55:11 +0000 | [diff] [blame] | 464 | if (Iter == COFFUniquingMap.end()) |
Craig Topper | bb694de | 2014-04-13 04:57:38 +0000 | [diff] [blame] | 465 | return nullptr; |
Rafael Espindola | 60ec383 | 2013-11-19 19:52:52 +0000 | [diff] [blame] | 466 | return Iter->second; |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 467 | } |
| 468 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 469 | MCSectionCOFF *MCContext::getAssociativeCOFFSection(MCSectionCOFF *Sec, |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 470 | const MCSymbol *KeySym, |
| 471 | unsigned UniqueID) { |
| 472 | // Return the normal section if we don't have to be associative or unique. |
| 473 | if (!KeySym && UniqueID == GenericSectionID) |
Reid Kleckner | 7c4059e | 2014-09-04 17:42:03 +0000 | [diff] [blame] | 474 | return Sec; |
| 475 | |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 476 | // If we have a key symbol, make an associative section with the same name and |
| 477 | // kind as the normal section. |
| 478 | unsigned Characteristics = Sec->getCharacteristics(); |
| 479 | if (KeySym) { |
| 480 | Characteristics |= COFF::IMAGE_SCN_LNK_COMDAT; |
| 481 | return getCOFFSection(Sec->getSectionName(), Characteristics, |
| 482 | Sec->getKind(), KeySym->getName(), |
| 483 | COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE, UniqueID); |
| 484 | } |
| 485 | |
Reid Kleckner | 7c4059e | 2014-09-04 17:42:03 +0000 | [diff] [blame] | 486 | return getCOFFSection(Sec->getSectionName(), Characteristics, Sec->getKind(), |
Reid Kleckner | 97837b7 | 2016-05-02 23:22:18 +0000 | [diff] [blame] | 487 | "", 0, UniqueID); |
Reid Kleckner | 7c4059e | 2014-09-04 17:42:03 +0000 | [diff] [blame] | 488 | } |
| 489 | |
Sam Clegg | 12fd3da | 2017-10-20 21:28:38 +0000 | [diff] [blame] | 490 | MCSectionWasm *MCContext::getWasmSection(const Twine &Section, SectionKind K, |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 491 | const Twine &Group, unsigned UniqueID, |
| 492 | const char *BeginSymName) { |
| 493 | MCSymbolWasm *GroupSym = nullptr; |
Sam Clegg | d423f0d | 2018-01-11 20:35:17 +0000 | [diff] [blame] | 494 | if (!Group.isTriviallyEmpty() && !Group.str().empty()) { |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 495 | GroupSym = cast<MCSymbolWasm>(getOrCreateSymbol(Group)); |
Sam Clegg | d423f0d | 2018-01-11 20:35:17 +0000 | [diff] [blame] | 496 | GroupSym->setComdat(true); |
| 497 | } |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 498 | |
Sam Clegg | 12fd3da | 2017-10-20 21:28:38 +0000 | [diff] [blame] | 499 | return getWasmSection(Section, K, GroupSym, UniqueID, BeginSymName); |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 500 | } |
| 501 | |
Sam Clegg | 12fd3da | 2017-10-20 21:28:38 +0000 | [diff] [blame] | 502 | MCSectionWasm *MCContext::getWasmSection(const Twine &Section, SectionKind Kind, |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 503 | const MCSymbolWasm *GroupSym, |
| 504 | unsigned UniqueID, |
| 505 | const char *BeginSymName) { |
| 506 | StringRef Group = ""; |
| 507 | if (GroupSym) |
| 508 | Group = GroupSym->getName(); |
| 509 | // Do the lookup, if we have a hit, return it. |
| 510 | auto IterBool = WasmUniquingMap.insert( |
| 511 | std::make_pair(WasmSectionKey{Section.str(), Group, UniqueID}, nullptr)); |
| 512 | auto &Entry = *IterBool.first; |
| 513 | if (!IterBool.second) |
| 514 | return Entry.second; |
| 515 | |
| 516 | StringRef CachedName = Entry.first.SectionName; |
| 517 | |
Sam Clegg | b210c64 | 2018-05-10 17:38:35 +0000 | [diff] [blame] | 518 | MCSymbol *Begin = createSymbol(CachedName, false, false); |
| 519 | cast<MCSymbolWasm>(Begin)->setType(wasm::WASM_SYMBOL_TYPE_SECTION); |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 520 | |
| 521 | MCSectionWasm *Result = new (WasmAllocator.Allocate()) |
Sam Clegg | 12fd3da | 2017-10-20 21:28:38 +0000 | [diff] [blame] | 522 | MCSectionWasm(CachedName, Kind, GroupSym, UniqueID, Begin); |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 523 | Entry.second = Result; |
Sam Clegg | b210c64 | 2018-05-10 17:38:35 +0000 | [diff] [blame] | 524 | |
| 525 | auto *F = new MCDataFragment(); |
| 526 | Result->getFragmentList().insert(Result->begin(), F); |
| 527 | F->setParent(Result); |
| 528 | Begin->setFragment(F); |
| 529 | |
Dan Gohman | 18eafb6 | 2017-02-22 01:23:18 +0000 | [diff] [blame] | 530 | return Result; |
| 531 | } |
| 532 | |
Akira Hatanaka | b11ef08 | 2015-11-14 06:35:56 +0000 | [diff] [blame] | 533 | MCSubtargetInfo &MCContext::getSubtargetCopy(const MCSubtargetInfo &STI) { |
| 534 | return *new (MCSubtargetAllocator.Allocate()) MCSubtargetInfo(STI); |
| 535 | } |
| 536 | |
Paul Robinson | c17c8bf | 2018-07-10 14:41:54 +0000 | [diff] [blame] | 537 | void MCContext::addDebugPrefixMapEntry(const std::string &From, |
| 538 | const std::string &To) { |
| 539 | DebugPrefixMap.insert(std::make_pair(From, To)); |
| 540 | } |
| 541 | |
Jonas Devlieghere | 26ddf27 | 2018-07-11 12:30:35 +0000 | [diff] [blame] | 542 | void MCContext::RemapDebugPaths() { |
| 543 | const auto &DebugPrefixMap = this->DebugPrefixMap; |
| 544 | const auto RemapDebugPath = [&DebugPrefixMap](std::string &Path) { |
| 545 | for (const auto &Entry : DebugPrefixMap) |
| 546 | if (StringRef(Path).startswith(Entry.first)) { |
| 547 | std::string RemappedPath = |
| 548 | (Twine(Entry.second) + Path.substr(Entry.first.size())).str(); |
| 549 | Path.swap(RemappedPath); |
| 550 | } |
| 551 | }; |
Paul Robinson | c17c8bf | 2018-07-10 14:41:54 +0000 | [diff] [blame] | 552 | |
Jonas Devlieghere | 26ddf27 | 2018-07-11 12:30:35 +0000 | [diff] [blame] | 553 | // Remap compilation directory. |
Paul Robinson | c17c8bf | 2018-07-10 14:41:54 +0000 | [diff] [blame] | 554 | std::string CompDir = CompilationDir.str(); |
Jonas Devlieghere | 26ddf27 | 2018-07-11 12:30:35 +0000 | [diff] [blame] | 555 | RemapDebugPath(CompDir); |
Paul Robinson | c17c8bf | 2018-07-10 14:41:54 +0000 | [diff] [blame] | 556 | CompilationDir = CompDir; |
Jonas Devlieghere | 26ddf27 | 2018-07-11 12:30:35 +0000 | [diff] [blame] | 557 | |
| 558 | // Remap MCDwarfDirs in all compilation units. |
| 559 | for (auto &CUIDTablePair : MCDwarfLineTablesCUMap) |
| 560 | for (auto &Dir : CUIDTablePair.second.getMCDwarfDirs()) |
| 561 | RemapDebugPath(Dir); |
Paul Robinson | c17c8bf | 2018-07-10 14:41:54 +0000 | [diff] [blame] | 562 | } |
| 563 | |
Kevin Enderby | e5930f1 | 2010-07-28 20:55:35 +0000 | [diff] [blame] | 564 | //===----------------------------------------------------------------------===// |
| 565 | // Dwarf Management |
| 566 | //===----------------------------------------------------------------------===// |
| 567 | |
Scott Linder | 16c7bda | 2018-02-23 23:01:06 +0000 | [diff] [blame] | 568 | /// getDwarfFile - takes a file name and number to place in the dwarf file and |
Kevin Enderby | e5930f1 | 2010-07-28 20:55:35 +0000 | [diff] [blame] | 569 | /// directory tables. If the file number has already been allocated it is an |
| 570 | /// error and zero is returned and the client reports the error, else the |
| 571 | /// allocated file number is returned. The file numbers may be in any order. |
Paul Robinson | 70def12 | 2018-02-22 21:03:33 +0000 | [diff] [blame] | 572 | Expected<unsigned> MCContext::getDwarfFile(StringRef Directory, |
| 573 | StringRef FileName, |
| 574 | unsigned FileNumber, |
| 575 | MD5::MD5Result *Checksum, |
Scott Linder | 16c7bda | 2018-02-23 23:01:06 +0000 | [diff] [blame] | 576 | Optional<StringRef> Source, |
Paul Robinson | 70def12 | 2018-02-22 21:03:33 +0000 | [diff] [blame] | 577 | unsigned CUID) { |
David Blaikie | d9012ba | 2014-03-13 21:59:51 +0000 | [diff] [blame] | 578 | MCDwarfLineTable &Table = MCDwarfLineTablesCUMap[CUID]; |
Scott Linder | 16c7bda | 2018-02-23 23:01:06 +0000 | [diff] [blame] | 579 | return Table.tryGetFile(Directory, FileName, Checksum, Source, FileNumber); |
Kevin Enderby | e5930f1 | 2010-07-28 20:55:35 +0000 | [diff] [blame] | 580 | } |
Kevin Enderby | 1264b7c | 2010-08-24 20:32:42 +0000 | [diff] [blame] | 581 | |
Kevin Enderby | a68d004 | 2010-10-04 20:17:24 +0000 | [diff] [blame] | 582 | /// isValidDwarfFileNumber - takes a dwarf file number and returns true if it |
Kevin Enderby | 1264b7c | 2010-08-24 20:32:42 +0000 | [diff] [blame] | 583 | /// currently is assigned and false otherwise. |
Manman Ren | 1e42720 | 2013-03-07 01:42:00 +0000 | [diff] [blame] | 584 | bool MCContext::isValidDwarfFileNumber(unsigned FileNumber, unsigned CUID) { |
Paul Robinson | 11539b0 | 2018-06-22 14:16:11 +0000 | [diff] [blame] | 585 | const MCDwarfLineTable &LineTable = getMCDwarfLineTable(CUID); |
| 586 | if (FileNumber == 0) |
| 587 | return getDwarfVersion() >= 5 && LineTable.hasRootFile(); |
| 588 | if (FileNumber >= LineTable.getMCDwarfFiles().size()) |
Kevin Enderby | 1264b7c | 2010-08-24 20:32:42 +0000 | [diff] [blame] | 589 | return false; |
| 590 | |
Paul Robinson | 11539b0 | 2018-06-22 14:16:11 +0000 | [diff] [blame] | 591 | return !LineTable.getMCDwarfFiles()[FileNumber].Name.empty(); |
Kevin Enderby | 1264b7c | 2010-08-24 20:32:42 +0000 | [diff] [blame] | 592 | } |
Jim Grosbach | b18b409 | 2012-01-26 23:20:11 +0000 | [diff] [blame] | 593 | |
Fangrui Song | 58963e4 | 2018-09-08 02:04:20 +0000 | [diff] [blame] | 594 | /// Remove empty sections from SectionsForRanges, to avoid generating |
Oliver Stannard | 8b27308 | 2014-06-19 15:52:37 +0000 | [diff] [blame] | 595 | /// useless debug info for them. |
| 596 | void MCContext::finalizeDwarfSections(MCStreamer &MCOS) { |
Benjamin Kramer | 412c4db | 2015-05-31 18:49:28 +0000 | [diff] [blame] | 597 | SectionsForRanges.remove_if( |
| 598 | [&](MCSection *Sec) { return !MCOS.mayHaveInstructions(*Sec); }); |
Oliver Stannard | 8b27308 | 2014-06-19 15:52:37 +0000 | [diff] [blame] | 599 | } |
| 600 | |
Reid Kleckner | 2214ed8 | 2016-01-29 00:49:42 +0000 | [diff] [blame] | 601 | CodeViewContext &MCContext::getCVContext() { |
| 602 | if (!CVContext.get()) |
| 603 | CVContext.reset(new CodeViewContext); |
| 604 | return *CVContext.get(); |
| 605 | } |
| 606 | |
Oliver Stannard | 07b43d3 | 2015-11-17 09:58:07 +0000 | [diff] [blame] | 607 | //===----------------------------------------------------------------------===// |
| 608 | // Error Reporting |
| 609 | //===----------------------------------------------------------------------===// |
| 610 | |
| 611 | void MCContext::reportError(SMLoc Loc, const Twine &Msg) { |
| 612 | HadError = true; |
| 613 | |
Sanne Wouda | 2933875 | 2017-02-08 14:48:05 +0000 | [diff] [blame] | 614 | // If we have a source manager use it. Otherwise, try using the inline source |
| 615 | // manager. |
| 616 | // If that fails, use the generic report_fatal_error(). |
| 617 | if (SrcMgr) |
| 618 | SrcMgr->PrintMessage(Loc, SourceMgr::DK_Error, Msg); |
| 619 | else if (InlineSrcMgr) |
| 620 | InlineSrcMgr->PrintMessage(Loc, SourceMgr::DK_Error, Msg); |
| 621 | else |
Jim Grosbach | 9a3284f | 2014-03-14 22:41:58 +0000 | [diff] [blame] | 622 | report_fatal_error(Msg, false); |
Oliver Stannard | 07b43d3 | 2015-11-17 09:58:07 +0000 | [diff] [blame] | 623 | } |
| 624 | |
| 625 | void MCContext::reportFatalError(SMLoc Loc, const Twine &Msg) { |
| 626 | reportError(Loc, Msg); |
Jim Grosbach | b18b409 | 2012-01-26 23:20:11 +0000 | [diff] [blame] | 627 | |
| 628 | // If we reached here, we are failing ungracefully. Run the interrupt handlers |
| 629 | // to make sure any special cleanups get done, in particular that we remove |
| 630 | // files registered with RemoveFileOnSignal. |
| 631 | sys::RunInterruptHandlers(); |
| 632 | exit(1); |
| 633 | } |