blob: 376262b84620b341572303b1ff62da0d769599db [file] [log] [blame]
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001//===- AsmWriter.cpp - Printing LLVM as an assembly file ------------------===//
Misha Brukmanb1c93172005-04-21 23:48:37 +00002//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// 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
Misha Brukmanb1c93172005-04-21 23:48:37 +00006//
John Criswell482202a2003-10-20 19:43:21 +00007//===----------------------------------------------------------------------===//
Chris Lattner2f7c9632001-06-06 20:29:01 +00008//
Vlad Tsyrklevich3b59a8a2018-04-26 17:34:51 +00009// This library implements `print` family of functions in classes like
10// Module, Function, Value, etc. In-memory representation of those classes is
11// converted to IR strings.
Chris Lattner2f7c9632001-06-06 20:29:01 +000012//
Chris Lattner189088e2002-04-12 18:21:53 +000013// Note that these routines must be extremely tolerant of various errors in the
Chris Lattnerf70da102003-05-08 02:44:12 +000014// LLVM code, because it can be used for debugging transformations.
Chris Lattner189088e2002-04-12 18:21:53 +000015//
Chris Lattner2f7c9632001-06-06 20:29:01 +000016//===----------------------------------------------------------------------===//
17
Eugene Zelenkode6cce22017-06-19 22:05:08 +000018#include "llvm/ADT/APFloat.h"
19#include "llvm/ADT/APInt.h"
20#include "llvm/ADT/ArrayRef.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000021#include "llvm/ADT/DenseMap.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000022#include "llvm/ADT/None.h"
23#include "llvm/ADT/Optional.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000024#include "llvm/ADT/STLExtras.h"
Benjamin Kramerba05a782015-03-17 19:53:41 +000025#include "llvm/ADT/SetVector.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000026#include "llvm/ADT/SmallString.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000027#include "llvm/ADT/SmallVector.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000028#include "llvm/ADT/StringExtras.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000029#include "llvm/ADT/StringRef.h"
30#include "llvm/ADT/iterator_range.h"
Zachary Turner264b5d92017-06-07 03:48:56 +000031#include "llvm/BinaryFormat/Dwarf.h"
Nico Weber432a3882018-04-30 14:59:11 +000032#include "llvm/Config/llvm-config.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000033#include "llvm/IR/Argument.h"
Chandler Carruth9aca9182014-01-07 12:34:26 +000034#include "llvm/IR/AssemblyAnnotationWriter.h"
Reid Klecknereb9dd5b2017-04-10 23:31:05 +000035#include "llvm/IR/Attributes.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000036#include "llvm/IR/BasicBlock.h"
Chandler Carruth1305dc32014-03-04 11:45:46 +000037#include "llvm/IR/CFG.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000038#include "llvm/IR/CallingConv.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000039#include "llvm/IR/Comdat.h"
40#include "llvm/IR/Constant.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000041#include "llvm/IR/Constants.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000042#include "llvm/IR/DebugInfoMetadata.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000043#include "llvm/IR/DerivedTypes.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000044#include "llvm/IR/Function.h"
45#include "llvm/IR/GlobalAlias.h"
46#include "llvm/IR/GlobalIFunc.h"
47#include "llvm/IR/GlobalIndirectSymbol.h"
48#include "llvm/IR/GlobalObject.h"
49#include "llvm/IR/GlobalValue.h"
50#include "llvm/IR/GlobalVariable.h"
Chandler Carruthb8ddc702014-01-12 11:10:32 +000051#include "llvm/IR/IRPrintingPasses.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000052#include "llvm/IR/InlineAsm.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000053#include "llvm/IR/InstrTypes.h"
54#include "llvm/IR/Instruction.h"
55#include "llvm/IR/Instructions.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000056#include "llvm/IR/LLVMContext.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000057#include "llvm/IR/Metadata.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000058#include "llvm/IR/Module.h"
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +000059#include "llvm/IR/ModuleSlotTracker.h"
Teresa Johnson08d5b4e2018-05-26 02:34:13 +000060#include "llvm/IR/ModuleSummaryIndex.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000061#include "llvm/IR/Operator.h"
Igor Laevsky2aa8caf2015-05-05 13:20:42 +000062#include "llvm/IR/Statepoint.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000063#include "llvm/IR/Type.h"
Chandler Carruthdcb603f2013-01-07 15:43:51 +000064#include "llvm/IR/TypeFinder.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000065#include "llvm/IR/Use.h"
Benjamin Kramerba05a782015-03-17 19:53:41 +000066#include "llvm/IR/UseListOrder.h"
Eugene Zelenkode6cce22017-06-19 22:05:08 +000067#include "llvm/IR/User.h"
68#include "llvm/IR/Value.h"
69#include "llvm/Support/AtomicOrdering.h"
70#include "llvm/Support/Casting.h"
71#include "llvm/Support/Compiler.h"
David Greenec7f9b122010-01-05 01:29:26 +000072#include "llvm/Support/Debug.h"
Torok Edwin6dd27302009-07-08 18:01:40 +000073#include "llvm/Support/ErrorHandling.h"
Justin Bogner93289572015-12-04 02:14:34 +000074#include "llvm/Support/Format.h"
Dan Gohmane2745262009-08-12 17:23:50 +000075#include "llvm/Support/FormattedStream.h"
Benjamin Kramer799003b2015-03-23 19:32:43 +000076#include "llvm/Support/raw_ostream.h"
Chris Lattnerfee714f2001-09-07 16:36:04 +000077#include <algorithm>
Eugene Zelenkode6cce22017-06-19 22:05:08 +000078#include <cassert>
Reid Spencerbdf03b42007-05-22 19:27:35 +000079#include <cctype>
Eugene Zelenkode6cce22017-06-19 22:05:08 +000080#include <cstddef>
81#include <cstdint>
82#include <iterator>
83#include <memory>
84#include <string>
85#include <tuple>
86#include <utility>
87#include <vector>
88
Chris Lattner189d19f2003-11-21 20:23:48 +000089using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000090
Reid Spencer294715b2005-05-15 16:13:11 +000091// Make virtual table appear in this compilation unit.
Eugene Zelenkode6cce22017-06-19 22:05:08 +000092AssemblyAnnotationWriter::~AssemblyAnnotationWriter() = default;
Reid Spencer294715b2005-05-15 16:13:11 +000093
Chris Lattner3eee99c2008-08-19 04:36:02 +000094//===----------------------------------------------------------------------===//
95// Helper Functions
96//===----------------------------------------------------------------------===//
97
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +000098namespace {
Eugene Zelenkode6cce22017-06-19 22:05:08 +000099
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000100struct OrderMap {
101 DenseMap<const Value *, std::pair<unsigned, bool>> IDs;
102
103 unsigned size() const { return IDs.size(); }
104 std::pair<unsigned, bool> &operator[](const Value *V) { return IDs[V]; }
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000105
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000106 std::pair<unsigned, bool> lookup(const Value *V) const {
107 return IDs.lookup(V);
108 }
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000109
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000110 void index(const Value *V) {
111 // Explicitly sequence get-size and insert-value operations to avoid UB.
112 unsigned ID = IDs.size() + 1;
113 IDs[V].first = ID;
114 }
115};
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000116
117} // end anonymous namespace
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000118
119static void orderValue(const Value *V, OrderMap &OM) {
120 if (OM.lookup(V).first)
121 return;
122
123 if (const Constant *C = dyn_cast<Constant>(V))
124 if (C->getNumOperands() && !isa<GlobalValue>(C))
125 for (const Value *Op : C->operands())
126 if (!isa<BasicBlock>(Op) && !isa<GlobalValue>(Op))
127 orderValue(Op, OM);
128
129 // Note: we cannot cache this lookup above, since inserting into the map
130 // changes the map's size, and thus affects the other IDs.
131 OM.index(V);
132}
133
134static OrderMap orderModule(const Module *M) {
135 // This needs to match the order used by ValueEnumerator::ValueEnumerator()
136 // and ValueEnumerator::incorporateFunction().
137 OrderMap OM;
138
139 for (const GlobalVariable &G : M->globals()) {
140 if (G.hasInitializer())
141 if (!isa<GlobalValue>(G.getInitializer()))
142 orderValue(G.getInitializer(), OM);
143 orderValue(&G, OM);
144 }
145 for (const GlobalAlias &A : M->aliases()) {
146 if (!isa<GlobalValue>(A.getAliasee()))
147 orderValue(A.getAliasee(), OM);
148 orderValue(&A, OM);
149 }
Dmitry Polukhina1feff72016-04-07 12:32:19 +0000150 for (const GlobalIFunc &I : M->ifuncs()) {
151 if (!isa<GlobalValue>(I.getResolver()))
152 orderValue(I.getResolver(), OM);
153 orderValue(&I, OM);
154 }
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000155 for (const Function &F : *M) {
Vedant Kumar3a63fb32015-12-19 08:52:49 +0000156 for (const Use &U : F.operands())
157 if (!isa<GlobalValue>(U.get()))
158 orderValue(U.get(), OM);
David Majnemer7fddecc2015-06-17 20:52:32 +0000159
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000160 orderValue(&F, OM);
161
162 if (F.isDeclaration())
163 continue;
164
165 for (const Argument &A : F.args())
166 orderValue(&A, OM);
167 for (const BasicBlock &BB : F) {
168 orderValue(&BB, OM);
169 for (const Instruction &I : BB) {
170 for (const Value *Op : I.operands())
171 if ((isa<Constant>(*Op) && !isa<GlobalValue>(*Op)) ||
172 isa<InlineAsm>(*Op))
173 orderValue(Op, OM);
174 orderValue(&I, OM);
175 }
176 }
177 }
178 return OM;
179}
180
181static void predictValueUseListOrderImpl(const Value *V, const Function *F,
182 unsigned ID, const OrderMap &OM,
183 UseListOrderStack &Stack) {
184 // Predict use-list order for this one.
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000185 using Entry = std::pair<const Use *, unsigned>;
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000186 SmallVector<Entry, 64> List;
187 for (const Use &U : V->uses())
188 // Check if this user will be serialized.
189 if (OM.lookup(U.getUser()).first)
190 List.push_back(std::make_pair(&U, List.size()));
191
192 if (List.size() < 2)
193 // We may have lost some users.
194 return;
195
196 bool GetsReversed =
197 !isa<GlobalVariable>(V) && !isa<Function>(V) && !isa<BasicBlock>(V);
198 if (auto *BA = dyn_cast<BlockAddress>(V))
199 ID = OM.lookup(BA->getBasicBlock()).first;
Fangrui Song0cac7262018-09-27 02:13:45 +0000200 llvm::sort(List, [&](const Entry &L, const Entry &R) {
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000201 const Use *LU = L.first;
202 const Use *RU = R.first;
203 if (LU == RU)
204 return false;
205
206 auto LID = OM.lookup(LU->getUser()).first;
207 auto RID = OM.lookup(RU->getUser()).first;
208
209 // If ID is 4, then expect: 7 6 5 1 2 3.
210 if (LID < RID) {
211 if (GetsReversed)
212 if (RID <= ID)
213 return true;
214 return false;
215 }
216 if (RID < LID) {
217 if (GetsReversed)
218 if (LID <= ID)
219 return false;
220 return true;
221 }
222
223 // LID and RID are equal, so we have different operands of the same user.
224 // Assume operands are added in order for all instructions.
225 if (GetsReversed)
226 if (LID <= ID)
227 return LU->getOperandNo() < RU->getOperandNo();
228 return LU->getOperandNo() > RU->getOperandNo();
229 });
230
231 if (std::is_sorted(
232 List.begin(), List.end(),
233 [](const Entry &L, const Entry &R) { return L.second < R.second; }))
234 // Order is already correct.
235 return;
236
237 // Store the shuffle.
238 Stack.emplace_back(V, F, List.size());
239 assert(List.size() == Stack.back().Shuffle.size() && "Wrong size");
240 for (size_t I = 0, E = List.size(); I != E; ++I)
241 Stack.back().Shuffle[I] = List[I].second;
242}
243
244static void predictValueUseListOrder(const Value *V, const Function *F,
245 OrderMap &OM, UseListOrderStack &Stack) {
246 auto &IDPair = OM[V];
247 assert(IDPair.first && "Unmapped value");
248 if (IDPair.second)
249 // Already predicted.
250 return;
251
252 // Do the actual prediction.
253 IDPair.second = true;
254 if (!V->use_empty() && std::next(V->use_begin()) != V->use_end())
255 predictValueUseListOrderImpl(V, F, IDPair.first, OM, Stack);
256
257 // Recursive descent into constants.
258 if (const Constant *C = dyn_cast<Constant>(V))
259 if (C->getNumOperands()) // Visit GlobalValues.
260 for (const Value *Op : C->operands())
261 if (isa<Constant>(Op)) // Visit GlobalValues.
262 predictValueUseListOrder(Op, F, OM, Stack);
263}
264
265static UseListOrderStack predictUseListOrder(const Module *M) {
266 OrderMap OM = orderModule(M);
267
268 // Use-list orders need to be serialized after all the users have been added
269 // to a value, or else the shuffles will be incomplete. Store them per
270 // function in a stack.
271 //
272 // Aside from function order, the order of values doesn't matter much here.
273 UseListOrderStack Stack;
274
275 // We want to visit the functions backward now so we can list function-local
276 // constants in the last Function they're used in. Module-level constants
277 // have already been visited above.
Pete Cooper7679afd2015-07-24 21:13:43 +0000278 for (const Function &F : make_range(M->rbegin(), M->rend())) {
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000279 if (F.isDeclaration())
280 continue;
281 for (const BasicBlock &BB : F)
282 predictValueUseListOrder(&BB, &F, OM, Stack);
283 for (const Argument &A : F.args())
284 predictValueUseListOrder(&A, &F, OM, Stack);
285 for (const BasicBlock &BB : F)
286 for (const Instruction &I : BB)
287 for (const Value *Op : I.operands())
288 if (isa<Constant>(*Op) || isa<InlineAsm>(*Op)) // Visit GlobalValues.
289 predictValueUseListOrder(Op, &F, OM, Stack);
290 for (const BasicBlock &BB : F)
291 for (const Instruction &I : BB)
292 predictValueUseListOrder(&I, &F, OM, Stack);
293 }
294
295 // Visit globals last.
296 for (const GlobalVariable &G : M->globals())
297 predictValueUseListOrder(&G, nullptr, OM, Stack);
298 for (const Function &F : *M)
299 predictValueUseListOrder(&F, nullptr, OM, Stack);
300 for (const GlobalAlias &A : M->aliases())
301 predictValueUseListOrder(&A, nullptr, OM, Stack);
Dmitry Polukhina1feff72016-04-07 12:32:19 +0000302 for (const GlobalIFunc &I : M->ifuncs())
303 predictValueUseListOrder(&I, nullptr, OM, Stack);
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000304 for (const GlobalVariable &G : M->globals())
305 if (G.hasInitializer())
306 predictValueUseListOrder(G.getInitializer(), nullptr, OM, Stack);
307 for (const GlobalAlias &A : M->aliases())
308 predictValueUseListOrder(A.getAliasee(), nullptr, OM, Stack);
Dmitry Polukhina1feff72016-04-07 12:32:19 +0000309 for (const GlobalIFunc &I : M->ifuncs())
310 predictValueUseListOrder(I.getResolver(), nullptr, OM, Stack);
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000311 for (const Function &F : *M)
Vedant Kumar3a63fb32015-12-19 08:52:49 +0000312 for (const Use &U : F.operands())
313 predictValueUseListOrder(U.get(), nullptr, OM, Stack);
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000314
315 return Stack;
316}
317
Chris Lattner3eee99c2008-08-19 04:36:02 +0000318static const Module *getModuleFromVal(const Value *V) {
319 if (const Argument *MA = dyn_cast<Argument>(V))
Craig Topperc6207612014-04-09 06:08:46 +0000320 return MA->getParent() ? MA->getParent()->getParent() : nullptr;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000321
Chris Lattner3eee99c2008-08-19 04:36:02 +0000322 if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
Craig Topperc6207612014-04-09 06:08:46 +0000323 return BB->getParent() ? BB->getParent()->getParent() : nullptr;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000324
Chris Lattner3eee99c2008-08-19 04:36:02 +0000325 if (const Instruction *I = dyn_cast<Instruction>(V)) {
Craig Topperc6207612014-04-09 06:08:46 +0000326 const Function *M = I->getParent() ? I->getParent()->getParent() : nullptr;
327 return M ? M->getParent() : nullptr;
Chris Lattner3eee99c2008-08-19 04:36:02 +0000328 }
Andrew Trickec4b6e72011-09-30 19:48:58 +0000329
Chris Lattner3eee99c2008-08-19 04:36:02 +0000330 if (const GlobalValue *GV = dyn_cast<GlobalValue>(V))
331 return GV->getParent();
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000332
333 if (const auto *MAV = dyn_cast<MetadataAsValue>(V)) {
334 for (const User *U : MAV->users())
335 if (isa<Instruction>(U))
336 if (const Module *M = getModuleFromVal(U))
337 return M;
338 return nullptr;
339 }
340
Craig Topperc6207612014-04-09 06:08:46 +0000341 return nullptr;
Chris Lattner3eee99c2008-08-19 04:36:02 +0000342}
343
Bill Wendling90bc19c2013-02-20 07:21:42 +0000344static void PrintCallingConv(unsigned cc, raw_ostream &Out) {
Micah Villmow857186d2012-09-13 15:11:12 +0000345 switch (cc) {
Bill Wendling90bc19c2013-02-20 07:21:42 +0000346 default: Out << "cc" << cc; break;
347 case CallingConv::Fast: Out << "fastcc"; break;
348 case CallingConv::Cold: Out << "coldcc"; break;
Andrew Trick5ae6ed82013-11-11 22:40:22 +0000349 case CallingConv::WebKit_JS: Out << "webkit_jscc"; break;
350 case CallingConv::AnyReg: Out << "anyregcc"; break;
Juergen Ributzkae6250132014-01-17 19:47:03 +0000351 case CallingConv::PreserveMost: Out << "preserve_mostcc"; break;
352 case CallingConv::PreserveAll: Out << "preserve_allcc"; break;
Manman Ren19c7bbe2015-12-04 17:40:13 +0000353 case CallingConv::CXX_FAST_TLS: Out << "cxx_fast_tlscc"; break;
Reid Kleckner35fc3632014-12-01 21:04:44 +0000354 case CallingConv::GHC: Out << "ghccc"; break;
Bill Wendling90bc19c2013-02-20 07:21:42 +0000355 case CallingConv::X86_StdCall: Out << "x86_stdcallcc"; break;
356 case CallingConv::X86_FastCall: Out << "x86_fastcallcc"; break;
357 case CallingConv::X86_ThisCall: Out << "x86_thiscallcc"; break;
Oren Ben Simhon92ccbf22016-10-13 07:53:43 +0000358 case CallingConv::X86_RegCall: Out << "x86_regcallcc"; break;
Reid Kleckner9ccce992014-10-28 01:29:26 +0000359 case CallingConv::X86_VectorCall:Out << "x86_vectorcallcc"; break;
Bill Wendling90bc19c2013-02-20 07:21:42 +0000360 case CallingConv::Intel_OCL_BI: Out << "intel_ocl_bicc"; break;
361 case CallingConv::ARM_APCS: Out << "arm_apcscc"; break;
362 case CallingConv::ARM_AAPCS: Out << "arm_aapcscc"; break;
363 case CallingConv::ARM_AAPCS_VFP: Out << "arm_aapcs_vfpcc"; break;
Sander de Smalen4dbc5122018-09-12 08:54:06 +0000364 case CallingConv::AArch64_VectorCall: Out << "aarch64_vector_pcs"; break;
Bill Wendling90bc19c2013-02-20 07:21:42 +0000365 case CallingConv::MSP430_INTR: Out << "msp430_intrcc"; break;
Dylan McKay4fd0d4a2016-03-03 10:08:02 +0000366 case CallingConv::AVR_INTR: Out << "avr_intrcc "; break;
367 case CallingConv::AVR_SIGNAL: Out << "avr_signalcc "; break;
Bill Wendling90bc19c2013-02-20 07:21:42 +0000368 case CallingConv::PTX_Kernel: Out << "ptx_kernel"; break;
369 case CallingConv::PTX_Device: Out << "ptx_device"; break;
Charles Davise8f297c2013-07-12 06:02:35 +0000370 case CallingConv::X86_64_SysV: Out << "x86_64_sysvcc"; break;
Martin Storsjo2f24e932017-07-17 20:05:19 +0000371 case CallingConv::Win64: Out << "win64cc"; break;
Michael Kupersteine31b4862013-12-15 10:01:20 +0000372 case CallingConv::SPIR_FUNC: Out << "spir_func"; break;
373 case CallingConv::SPIR_KERNEL: Out << "spir_kernel"; break;
Manman Renf8bdd882016-04-05 22:41:47 +0000374 case CallingConv::Swift: Out << "swiftcc"; break;
Amjad Aboud60b5e1b2015-12-21 14:07:14 +0000375 case CallingConv::X86_INTR: Out << "x86_intrcc"; break;
Maksim Panchenkocce239c2015-09-29 22:09:16 +0000376 case CallingConv::HHVM: Out << "hhvmcc"; break;
377 case CallingConv::HHVM_C: Out << "hhvm_ccc"; break;
Nicolai Haehnledf3a20c2016-04-06 19:40:20 +0000378 case CallingConv::AMDGPU_VS: Out << "amdgpu_vs"; break;
Tim Renoufef1ae8f2017-09-29 09:51:22 +0000379 case CallingConv::AMDGPU_LS: Out << "amdgpu_ls"; break;
Marek Olsaka302a7362017-05-02 15:41:10 +0000380 case CallingConv::AMDGPU_HS: Out << "amdgpu_hs"; break;
Tim Renoufef1ae8f2017-09-29 09:51:22 +0000381 case CallingConv::AMDGPU_ES: Out << "amdgpu_es"; break;
Nicolai Haehnledf3a20c2016-04-06 19:40:20 +0000382 case CallingConv::AMDGPU_GS: Out << "amdgpu_gs"; break;
383 case CallingConv::AMDGPU_PS: Out << "amdgpu_ps"; break;
384 case CallingConv::AMDGPU_CS: Out << "amdgpu_cs"; break;
Nikolay Haustov1f7732a2016-05-06 09:07:29 +0000385 case CallingConv::AMDGPU_KERNEL: Out << "amdgpu_kernel"; break;
Micah Villmow857186d2012-09-13 15:11:12 +0000386 }
387}
Michael Ilseman26ee2b82012-11-15 22:34:00 +0000388
Chris Lattner3eee99c2008-08-19 04:36:02 +0000389enum PrefixType {
390 GlobalPrefix,
David Majnemerdad0a642014-06-27 18:19:56 +0000391 ComdatPrefix,
Chris Lattner3eee99c2008-08-19 04:36:02 +0000392 LabelPrefix,
Daniel Dunbar389529a2008-10-14 23:28:09 +0000393 LocalPrefix,
394 NoPrefix
Chris Lattner3eee99c2008-08-19 04:36:02 +0000395};
396
Alex Lorenz2b7f2652015-07-21 16:50:35 +0000397void llvm::printLLVMNameWithoutPrefix(raw_ostream &OS, StringRef Name) {
Jay Foadf7adb322011-04-24 14:30:00 +0000398 assert(!Name.empty() && "Cannot get empty name!");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000399
Chris Lattner3eee99c2008-08-19 04:36:02 +0000400 // Scan the name to see if it needs quotes first.
Guy Benyei83c74e92013-02-12 21:21:59 +0000401 bool NeedsQuotes = isdigit(static_cast<unsigned char>(Name[0]));
Chris Lattner3eee99c2008-08-19 04:36:02 +0000402 if (!NeedsQuotes) {
Daniel Dunbare03eecb2009-07-25 23:55:21 +0000403 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
Aaron Ballmaned9b0a92012-07-16 16:18:18 +0000404 // By making this unsigned, the value passed in to isalnum will always be
405 // in the range 0-255. This is important when building with MSVC because
406 // its implementation will assert. This situation can arise when dealing
407 // with UTF-8 multibyte characters.
408 unsigned char C = Name[i];
Guy Benyei83c74e92013-02-12 21:21:59 +0000409 if (!isalnum(static_cast<unsigned char>(C)) && C != '-' && C != '.' &&
410 C != '_') {
Chris Lattner3eee99c2008-08-19 04:36:02 +0000411 NeedsQuotes = true;
412 break;
413 }
414 }
415 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000416
Chris Lattner3eee99c2008-08-19 04:36:02 +0000417 // If we didn't need any quotes, just write out the name in one blast.
418 if (!NeedsQuotes) {
Daniel Dunbare03eecb2009-07-25 23:55:21 +0000419 OS << Name;
Chris Lattner3eee99c2008-08-19 04:36:02 +0000420 return;
421 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000422
Chris Lattner3eee99c2008-08-19 04:36:02 +0000423 // Okay, we need quotes. Output the quotes and escape any scary characters as
424 // needed.
425 OS << '"';
Jonas Devlieghere745918ff2018-05-31 17:01:42 +0000426 printEscapedString(Name, OS);
Chris Lattner3eee99c2008-08-19 04:36:02 +0000427 OS << '"';
428}
429
Alex Lorenz2b7f2652015-07-21 16:50:35 +0000430/// Turn the specified name into an 'LLVM name', which is either prefixed with %
431/// (if the string only contains simple characters) or is surrounded with ""'s
432/// (if it has special chars in it). Print it out.
433static void PrintLLVMName(raw_ostream &OS, StringRef Name, PrefixType Prefix) {
434 switch (Prefix) {
435 case NoPrefix:
436 break;
437 case GlobalPrefix:
438 OS << '@';
439 break;
440 case ComdatPrefix:
441 OS << '$';
442 break;
443 case LabelPrefix:
444 break;
445 case LocalPrefix:
446 OS << '%';
447 break;
448 }
449 printLLVMNameWithoutPrefix(OS, Name);
450}
451
452/// Turn the specified name into an 'LLVM name', which is either prefixed with %
453/// (if the string only contains simple characters) or is surrounded with ""'s
454/// (if it has special chars in it). Print it out.
Dan Gohman12dad632009-08-12 20:56:03 +0000455static void PrintLLVMName(raw_ostream &OS, const Value *V) {
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000456 PrintLLVMName(OS, V->getName(),
Chris Lattner3eee99c2008-08-19 04:36:02 +0000457 isa<GlobalValue>(V) ? GlobalPrefix : LocalPrefix);
458}
459
Benjamin Kramerba05a782015-03-17 19:53:41 +0000460namespace {
Benjamin Kramerba05a782015-03-17 19:53:41 +0000461
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000462class TypePrinting {
463public:
Roman Tereshind96de6f2018-03-22 21:29:07 +0000464 TypePrinting(const Module *M = nullptr) : DeferredM(M) {}
Benjamin Kramerba05a782015-03-17 19:53:41 +0000465
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000466 TypePrinting(const TypePrinting &) = delete;
467 TypePrinting &operator=(const TypePrinting &) = delete;
Benjamin Kramerba05a782015-03-17 19:53:41 +0000468
Roman Tereshind96de6f2018-03-22 21:29:07 +0000469 /// The named types that are used by the current module.
470 TypeFinder &getNamedTypes();
471
472 /// The numbered types, number to type mapping.
473 std::vector<StructType *> &getNumberedTypes();
474
475 bool empty();
Benjamin Kramerba05a782015-03-17 19:53:41 +0000476
477 void print(Type *Ty, raw_ostream &OS);
478
479 void printStructBody(StructType *Ty, raw_ostream &OS);
Roman Tereshind96de6f2018-03-22 21:29:07 +0000480
481private:
482 void incorporateTypes();
483
484 /// A module to process lazily when needed. Set to nullptr as soon as used.
485 const Module *DeferredM;
486
487 TypeFinder NamedTypes;
488
489 // The numbered types, along with their value.
490 DenseMap<StructType *, unsigned> Type2Number;
491
492 std::vector<StructType *> NumberedTypes;
Benjamin Kramerba05a782015-03-17 19:53:41 +0000493};
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000494
495} // end anonymous namespace
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000496
Roman Tereshind96de6f2018-03-22 21:29:07 +0000497TypeFinder &TypePrinting::getNamedTypes() {
498 incorporateTypes();
499 return NamedTypes;
500}
501
502std::vector<StructType *> &TypePrinting::getNumberedTypes() {
503 incorporateTypes();
504
505 // We know all the numbers that each type is used and we know that it is a
506 // dense assignment. Convert the map to an index table, if it's not done
507 // already (judging from the sizes):
508 if (NumberedTypes.size() == Type2Number.size())
509 return NumberedTypes;
510
511 NumberedTypes.resize(Type2Number.size());
512 for (const auto &P : Type2Number) {
513 assert(P.second < NumberedTypes.size() && "Didn't get a dense numbering?");
514 assert(!NumberedTypes[P.second] && "Didn't get a unique numbering?");
515 NumberedTypes[P.second] = P.first;
516 }
517 return NumberedTypes;
518}
519
520bool TypePrinting::empty() {
521 incorporateTypes();
522 return NamedTypes.empty() && Type2Number.empty();
523}
524
525void TypePrinting::incorporateTypes() {
526 if (!DeferredM)
527 return;
528
529 NamedTypes.run(*DeferredM, false);
530 DeferredM = nullptr;
Andrew Trickec4b6e72011-09-30 19:48:58 +0000531
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000532 // The list of struct types we got back includes all the struct types, split
533 // the unnamed ones out to a numbering and remove the anonymous structs.
534 unsigned NextNumber = 0;
Andrew Trickec4b6e72011-09-30 19:48:58 +0000535
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000536 std::vector<StructType*>::iterator NextToUse = NamedTypes.begin(), I, E;
537 for (I = NamedTypes.begin(), E = NamedTypes.end(); I != E; ++I) {
538 StructType *STy = *I;
Andrew Trickec4b6e72011-09-30 19:48:58 +0000539
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000540 // Ignore anonymous types.
Chris Lattner01beceb2011-08-12 18:07:07 +0000541 if (STy->isLiteral())
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000542 continue;
Andrew Trickec4b6e72011-09-30 19:48:58 +0000543
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000544 if (STy->getName().empty())
Roman Tereshind96de6f2018-03-22 21:29:07 +0000545 Type2Number[STy] = NextNumber++;
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000546 else
547 *NextToUse++ = STy;
548 }
Andrew Trickec4b6e72011-09-30 19:48:58 +0000549
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000550 NamedTypes.erase(NextToUse, NamedTypes.end());
551}
552
Roman Tereshind96de6f2018-03-22 21:29:07 +0000553/// Write the specified type to the specified raw_ostream, making use of type
554/// names or up references to shorten the type name where possible.
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000555void TypePrinting::print(Type *Ty, raw_ostream &OS) {
Chris Lattner0f578952009-02-28 20:25:14 +0000556 switch (Ty->getTypeID()) {
Rafael Espindolaba7df702013-12-07 02:27:52 +0000557 case Type::VoidTyID: OS << "void"; return;
558 case Type::HalfTyID: OS << "half"; return;
559 case Type::FloatTyID: OS << "float"; return;
560 case Type::DoubleTyID: OS << "double"; return;
561 case Type::X86_FP80TyID: OS << "x86_fp80"; return;
562 case Type::FP128TyID: OS << "fp128"; return;
563 case Type::PPC_FP128TyID: OS << "ppc_fp128"; return;
564 case Type::LabelTyID: OS << "label"; return;
565 case Type::MetadataTyID: OS << "metadata"; return;
566 case Type::X86_MMXTyID: OS << "x86_mmx"; return;
David Majnemerb611e3f2015-08-14 05:09:07 +0000567 case Type::TokenTyID: OS << "token"; return;
Chris Lattner68318b12009-02-28 21:18:43 +0000568 case Type::IntegerTyID:
Chris Lattner06a23212009-02-28 21:27:31 +0000569 OS << 'i' << cast<IntegerType>(Ty)->getBitWidth();
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000570 return;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000571
Chris Lattner07d88292009-02-28 20:35:42 +0000572 case Type::FunctionTyID: {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000573 FunctionType *FTy = cast<FunctionType>(Ty);
574 print(FTy->getReturnType(), OS);
Chris Lattner06a23212009-02-28 21:27:31 +0000575 OS << " (";
Chris Lattner07d88292009-02-28 20:35:42 +0000576 for (FunctionType::param_iterator I = FTy->param_begin(),
577 E = FTy->param_end(); I != E; ++I) {
578 if (I != FTy->param_begin())
Chris Lattner06a23212009-02-28 21:27:31 +0000579 OS << ", ";
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000580 print(*I, OS);
Chris Lattner0f578952009-02-28 20:25:14 +0000581 }
Chris Lattner07d88292009-02-28 20:35:42 +0000582 if (FTy->isVarArg()) {
Chris Lattner06a23212009-02-28 21:27:31 +0000583 if (FTy->getNumParams()) OS << ", ";
584 OS << "...";
Chris Lattner0f578952009-02-28 20:25:14 +0000585 }
Chris Lattner06a23212009-02-28 21:27:31 +0000586 OS << ')';
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000587 return;
Chris Lattner07d88292009-02-28 20:35:42 +0000588 }
589 case Type::StructTyID: {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000590 StructType *STy = cast<StructType>(Ty);
Andrew Trickec4b6e72011-09-30 19:48:58 +0000591
Chris Lattner01beceb2011-08-12 18:07:07 +0000592 if (STy->isLiteral())
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000593 return printStructBody(STy, OS);
594
595 if (!STy->getName().empty())
596 return PrintLLVMName(OS, STy->getName(), LocalPrefix);
Andrew Trickec4b6e72011-09-30 19:48:58 +0000597
Roman Tereshind96de6f2018-03-22 21:29:07 +0000598 incorporateTypes();
599 const auto I = Type2Number.find(STy);
600 if (I != Type2Number.end())
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000601 OS << '%' << I->second;
602 else // Not enumerated, print the hex address.
Benjamin Kramer4e8b4632011-11-02 17:24:36 +0000603 OS << "%\"type " << STy << '\"';
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000604 return;
Chris Lattner07d88292009-02-28 20:35:42 +0000605 }
606 case Type::PointerTyID: {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000607 PointerType *PTy = cast<PointerType>(Ty);
608 print(PTy->getElementType(), OS);
Chris Lattner07d88292009-02-28 20:35:42 +0000609 if (unsigned AddressSpace = PTy->getAddressSpace())
Chris Lattner06a23212009-02-28 21:27:31 +0000610 OS << " addrspace(" << AddressSpace << ')';
611 OS << '*';
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000612 return;
Chris Lattner07d88292009-02-28 20:35:42 +0000613 }
614 case Type::ArrayTyID: {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000615 ArrayType *ATy = cast<ArrayType>(Ty);
Chris Lattner06a23212009-02-28 21:27:31 +0000616 OS << '[' << ATy->getNumElements() << " x ";
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000617 print(ATy->getElementType(), OS);
Chris Lattner06a23212009-02-28 21:27:31 +0000618 OS << ']';
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000619 return;
Chris Lattner07d88292009-02-28 20:35:42 +0000620 }
621 case Type::VectorTyID: {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000622 VectorType *PTy = cast<VectorType>(Ty);
Chris Lattner06a23212009-02-28 21:27:31 +0000623 OS << "<" << PTy->getNumElements() << " x ";
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000624 print(PTy->getElementType(), OS);
Chris Lattner06a23212009-02-28 21:27:31 +0000625 OS << '>';
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000626 return;
Chris Lattner07d88292009-02-28 20:35:42 +0000627 }
Chris Lattner0f578952009-02-28 20:25:14 +0000628 }
Rafael Espindolaba7df702013-12-07 02:27:52 +0000629 llvm_unreachable("Invalid TypeID");
Chris Lattner0f578952009-02-28 20:25:14 +0000630}
631
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000632void TypePrinting::printStructBody(StructType *STy, raw_ostream &OS) {
633 if (STy->isOpaque()) {
634 OS << "opaque";
635 return;
Chris Lattner0f578952009-02-28 20:25:14 +0000636 }
Andrew Trickec4b6e72011-09-30 19:48:58 +0000637
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000638 if (STy->isPacked())
639 OS << '<';
Andrew Trickec4b6e72011-09-30 19:48:58 +0000640
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000641 if (STy->getNumElements() == 0) {
642 OS << "{}";
643 } else {
644 StructType::element_iterator I = STy->element_begin();
645 OS << "{ ";
646 print(*I++, OS);
647 for (StructType::element_iterator E = STy->element_end(); I != E; ++I) {
648 OS << ", ";
649 print(*I, OS);
Chris Lattner3243ea12009-03-01 00:03:38 +0000650 }
Andrew Trickec4b6e72011-09-30 19:48:58 +0000651
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000652 OS << " }";
Chris Lattner92c5c122009-02-28 23:20:19 +0000653 }
Chris Lattnerb1ed91f2011-07-09 17:41:24 +0000654 if (STy->isPacked())
655 OS << '>';
Chris Lattner92c5c122009-02-28 23:20:19 +0000656}
657
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000658namespace llvm {
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000659
Chris Lattner3eee99c2008-08-19 04:36:02 +0000660//===----------------------------------------------------------------------===//
661// SlotTracker Class: Enumerate slot numbers for unnamed values
662//===----------------------------------------------------------------------===//
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000663/// This class provides computation of slot numbers for LLVM Assembly writing.
Chris Lattner393b7cd2008-08-17 04:17:45 +0000664///
Chris Lattnere36fd8a2008-08-19 04:26:57 +0000665class SlotTracker {
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000666public:
Devang Patel983c6b12009-07-08 21:44:25 +0000667 /// ValueMap - A mapping of Values to slot numbers.
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000668 using ValueMap = DenseMap<const Value *, unsigned>;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000669
670private:
Devang Patel983c6b12009-07-08 21:44:25 +0000671 /// TheModule - The module for which we are holding slot numbers.
Chris Lattner393b7cd2008-08-17 04:17:45 +0000672 const Module* TheModule;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000673
Devang Patel983c6b12009-07-08 21:44:25 +0000674 /// TheFunction - The function for which we are holding slot numbers.
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000675 const Function* TheFunction = nullptr;
676 bool FunctionProcessed = false;
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000677 bool ShouldInitializeAllMetadata;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000678
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000679 /// The summary index for which we are holding slot numbers.
680 const ModuleSummaryIndex *TheIndex = nullptr;
681
Jay Foaddbb221d2011-07-11 07:28:49 +0000682 /// mMap - The slot map for the module level data.
Chris Lattner393b7cd2008-08-17 04:17:45 +0000683 ValueMap mMap;
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000684 unsigned mNext = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000685
Jay Foaddbb221d2011-07-11 07:28:49 +0000686 /// fMap - The slot map for the function level data.
Chris Lattner393b7cd2008-08-17 04:17:45 +0000687 ValueMap fMap;
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000688 unsigned fNext = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000689
Devang Patel983c6b12009-07-08 21:44:25 +0000690 /// mdnMap - Map for MDNodes.
Chris Lattnercf4a76e2009-12-31 02:20:11 +0000691 DenseMap<const MDNode*, unsigned> mdnMap;
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000692 unsigned mdnNext = 0;
Bill Wendling829b4782013-02-11 08:43:33 +0000693
694 /// asMap - The slot map for attribute sets.
Reid Klecknerc2cb5602017-04-12 00:38:00 +0000695 DenseMap<AttributeSet, unsigned> asMap;
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000696 unsigned asNext = 0;
697
Teresa Johnson8fc76662018-07-02 22:09:23 +0000698 /// ModulePathMap - The slot map for Module paths used in the summary index.
699 StringMap<unsigned> ModulePathMap;
700 unsigned ModulePathNext = 0;
701
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000702 /// GUIDMap - The slot map for GUIDs used in the summary index.
703 DenseMap<GlobalValue::GUID, unsigned> GUIDMap;
704 unsigned GUIDNext = 0;
705
Teresa Johnson7fb39df2018-09-25 20:14:40 +0000706 /// TypeIdMap - The slot map for type ids used in the summary index.
707 StringMap<unsigned> TypeIdMap;
708 unsigned TypeIdNext = 0;
709
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000710public:
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000711 /// Construct from a module.
712 ///
713 /// If \c ShouldInitializeAllMetadata, initializes all metadata in all
714 /// functions, giving correct numbering for metadata referenced only from
715 /// within a function (even if no functions have been initialized).
716 explicit SlotTracker(const Module *M,
717 bool ShouldInitializeAllMetadata = false);
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000718
Chris Lattner393b7cd2008-08-17 04:17:45 +0000719 /// Construct from a function, starting out in incorp state.
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000720 ///
721 /// If \c ShouldInitializeAllMetadata, initializes all metadata in all
722 /// functions, giving correct numbering for metadata referenced only from
723 /// within a function (even if no functions have been initialized).
724 explicit SlotTracker(const Function *F,
725 bool ShouldInitializeAllMetadata = false);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000726
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000727 /// Construct from a module summary index.
728 explicit SlotTracker(const ModuleSummaryIndex *Index);
729
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000730 SlotTracker(const SlotTracker &) = delete;
731 SlotTracker &operator=(const SlotTracker &) = delete;
732
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000733 /// Return the slot number of the specified value in it's type
Chris Lattnere36fd8a2008-08-19 04:26:57 +0000734 /// plane. If something is not in the SlotTracker, return -1.
Chris Lattner5e043322007-01-11 03:54:27 +0000735 int getLocalSlot(const Value *V);
736 int getGlobalSlot(const GlobalValue *V);
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +0000737 int getMetadataSlot(const MDNode *N);
Reid Klecknerc2cb5602017-04-12 00:38:00 +0000738 int getAttributeGroupSlot(AttributeSet AS);
Teresa Johnson8fc76662018-07-02 22:09:23 +0000739 int getModulePathSlot(StringRef Path);
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000740 int getGUIDSlot(GlobalValue::GUID GUID);
Teresa Johnson7fb39df2018-09-25 20:14:40 +0000741 int getTypeIdSlot(StringRef Id);
Reid Spencer8beac692004-06-09 15:26:53 +0000742
Misha Brukmanb1c93172005-04-21 23:48:37 +0000743 /// If you'd like to deal with a function instead of just a module, use
Chris Lattnere36fd8a2008-08-19 04:26:57 +0000744 /// this method to get its data into the SlotTracker.
Misha Brukmanb1c93172005-04-21 23:48:37 +0000745 void incorporateFunction(const Function *F) {
746 TheFunction = F;
Reid Spencerb0ac8c42004-08-16 07:46:33 +0000747 FunctionProcessed = false;
748 }
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000749
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +0000750 const Function *getFunction() const { return TheFunction; }
751
Misha Brukmanb1c93172005-04-21 23:48:37 +0000752 /// After calling incorporateFunction, use this method to remove the
Chris Lattnere36fd8a2008-08-19 04:26:57 +0000753 /// most recently incorporated function from the SlotTracker. This
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000754 /// will reset the state of the machine back to just the module contents.
755 void purgeFunction();
756
Devang Patel983c6b12009-07-08 21:44:25 +0000757 /// MDNode map iterators.
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000758 using mdn_iterator = DenseMap<const MDNode*, unsigned>::iterator;
759
Chris Lattnercf4a76e2009-12-31 02:20:11 +0000760 mdn_iterator mdn_begin() { return mdnMap.begin(); }
761 mdn_iterator mdn_end() { return mdnMap.end(); }
762 unsigned mdn_size() const { return mdnMap.size(); }
763 bool mdn_empty() const { return mdnMap.empty(); }
Devang Patel983c6b12009-07-08 21:44:25 +0000764
Reid Klecknerc2cb5602017-04-12 00:38:00 +0000765 /// AttributeSet map iterators.
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000766 using as_iterator = DenseMap<AttributeSet, unsigned>::iterator;
767
Bill Wendling829b4782013-02-11 08:43:33 +0000768 as_iterator as_begin() { return asMap.begin(); }
769 as_iterator as_end() { return asMap.end(); }
770 unsigned as_size() const { return asMap.size(); }
771 bool as_empty() const { return asMap.empty(); }
772
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000773 /// GUID map iterators.
774 using guid_iterator = DenseMap<GlobalValue::GUID, unsigned>::iterator;
775
776 /// These functions do the actual initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +0000777 inline void initializeIfNeeded();
778 void initializeIndexIfNeeded();
Reid Spencer56010e42004-05-26 21:56:09 +0000779
Devang Patel983c6b12009-07-08 21:44:25 +0000780 // Implementation Details
781private:
Chris Lattnerea862a32007-01-09 07:55:49 +0000782 /// CreateModuleSlot - Insert the specified GlobalValue* into the slot table.
783 void CreateModuleSlot(const GlobalValue *V);
Devang Patel983c6b12009-07-08 21:44:25 +0000784
785 /// CreateMetadataSlot - Insert the specified MDNode* into the slot table.
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +0000786 void CreateMetadataSlot(const MDNode *N);
Devang Patel983c6b12009-07-08 21:44:25 +0000787
Chris Lattnerea862a32007-01-09 07:55:49 +0000788 /// CreateFunctionSlot - Insert the specified Value* into the slot table.
789 void CreateFunctionSlot(const Value *V);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000790
Adrian Prantl5f8f34e42018-05-01 15:54:18 +0000791 /// Insert the specified AttributeSet into the slot table.
Reid Klecknerc2cb5602017-04-12 00:38:00 +0000792 void CreateAttributeSetSlot(AttributeSet AS);
Bill Wendling829b4782013-02-11 08:43:33 +0000793
Teresa Johnson8fc76662018-07-02 22:09:23 +0000794 inline void CreateModulePathSlot(StringRef Path);
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000795 void CreateGUIDSlot(GlobalValue::GUID GUID);
Teresa Johnson7fb39df2018-09-25 20:14:40 +0000796 void CreateTypeIdSlot(StringRef Id);
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000797
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000798 /// Add all of the module level global variables (and their initializers)
799 /// and function declarations, but not the contents of those functions.
800 void processModule();
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000801 void processIndex();
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000802
Devang Patel983c6b12009-07-08 21:44:25 +0000803 /// Add all of the functions arguments, basic blocks, and instructions.
Reid Spencer56010e42004-05-26 21:56:09 +0000804 void processFunction();
805
Peter Collingbournecceae7f2016-05-31 23:01:54 +0000806 /// Add the metadata directly attached to a GlobalObject.
807 void processGlobalObjectMetadata(const GlobalObject &GO);
808
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000809 /// Add all of the metadata from a function.
810 void processFunctionMetadata(const Function &F);
811
Duncan P. N. Exon Smith20b76ac2015-03-14 19:48:31 +0000812 /// Add all of the metadata from an instruction.
813 void processInstructionMetadata(const Instruction &I);
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000814};
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000815
816} // end namespace llvm
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000817
818ModuleSlotTracker::ModuleSlotTracker(SlotTracker &Machine, const Module *M,
819 const Function *F)
820 : M(M), F(F), Machine(&Machine) {}
821
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +0000822ModuleSlotTracker::ModuleSlotTracker(const Module *M,
823 bool ShouldInitializeAllMetadata)
Duncan P. N. Exon Smithaf0fdc22016-04-20 19:05:59 +0000824 : ShouldCreateStorage(M),
825 ShouldInitializeAllMetadata(ShouldInitializeAllMetadata), M(M) {}
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000826
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000827ModuleSlotTracker::~ModuleSlotTracker() = default;
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000828
Duncan P. N. Exon Smithaf0fdc22016-04-20 19:05:59 +0000829SlotTracker *ModuleSlotTracker::getMachine() {
830 if (!ShouldCreateStorage)
831 return Machine;
832
833 ShouldCreateStorage = false;
834 MachineStorage =
835 llvm::make_unique<SlotTracker>(M, ShouldInitializeAllMetadata);
836 Machine = MachineStorage.get();
837 return Machine;
838}
839
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000840void ModuleSlotTracker::incorporateFunction(const Function &F) {
Duncan P. N. Exon Smithaf0fdc22016-04-20 19:05:59 +0000841 // Using getMachine() may lazily create the slot tracker.
842 if (!getMachine())
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000843 return;
844
845 // Nothing to do if this is the right function already.
846 if (this->F == &F)
847 return;
848 if (this->F)
849 Machine->purgeFunction();
850 Machine->incorporateFunction(&F);
851 this->F = &F;
852}
Reid Spencer16f2f7f2004-05-26 07:18:52 +0000853
Alex Lorenz991a6242015-07-27 22:31:04 +0000854int ModuleSlotTracker::getLocalSlot(const Value *V) {
855 assert(F && "No function incorporated");
856 return Machine->getLocalSlot(V);
857}
858
Chris Lattner3eee99c2008-08-19 04:36:02 +0000859static SlotTracker *createSlotTracker(const Value *V) {
860 if (const Argument *FA = dyn_cast<Argument>(V))
861 return new SlotTracker(FA->getParent());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000862
Chris Lattner3eee99c2008-08-19 04:36:02 +0000863 if (const Instruction *I = dyn_cast<Instruction>(V))
Andrew Trick2f0cbf62011-09-30 19:50:40 +0000864 if (I->getParent())
865 return new SlotTracker(I->getParent()->getParent());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000866
Chris Lattner3eee99c2008-08-19 04:36:02 +0000867 if (const BasicBlock *BB = dyn_cast<BasicBlock>(V))
868 return new SlotTracker(BB->getParent());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000869
Chris Lattner3eee99c2008-08-19 04:36:02 +0000870 if (const GlobalVariable *GV = dyn_cast<GlobalVariable>(V))
871 return new SlotTracker(GV->getParent());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000872
Chris Lattner3eee99c2008-08-19 04:36:02 +0000873 if (const GlobalAlias *GA = dyn_cast<GlobalAlias>(V))
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000874 return new SlotTracker(GA->getParent());
875
Dmitry Polukhina1feff72016-04-07 12:32:19 +0000876 if (const GlobalIFunc *GIF = dyn_cast<GlobalIFunc>(V))
877 return new SlotTracker(GIF->getParent());
878
Chris Lattner3eee99c2008-08-19 04:36:02 +0000879 if (const Function *Func = dyn_cast<Function>(V))
880 return new SlotTracker(Func);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000881
Craig Topperc6207612014-04-09 06:08:46 +0000882 return nullptr;
Chris Lattner3eee99c2008-08-19 04:36:02 +0000883}
884
885#if 0
David Greenec7f9b122010-01-05 01:29:26 +0000886#define ST_DEBUG(X) dbgs() << X
Chris Lattner3eee99c2008-08-19 04:36:02 +0000887#else
Chris Lattner604e3512008-08-19 04:47:09 +0000888#define ST_DEBUG(X)
Chris Lattner3eee99c2008-08-19 04:36:02 +0000889#endif
890
891// Module level constructor. Causes the contents of the Module (sans functions)
892// to be added to the slot table.
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000893SlotTracker::SlotTracker(const Module *M, bool ShouldInitializeAllMetadata)
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000894 : TheModule(M), ShouldInitializeAllMetadata(ShouldInitializeAllMetadata) {}
Chris Lattner3eee99c2008-08-19 04:36:02 +0000895
896// Function level constructor. Causes the contents of the Module and the one
897// function provided to be added to the slot table.
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000898SlotTracker::SlotTracker(const Function *F, bool ShouldInitializeAllMetadata)
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +0000899 : TheModule(F ? F->getParent() : nullptr), TheFunction(F),
Eugene Zelenkode6cce22017-06-19 22:05:08 +0000900 ShouldInitializeAllMetadata(ShouldInitializeAllMetadata) {}
Chris Lattner3eee99c2008-08-19 04:36:02 +0000901
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000902SlotTracker::SlotTracker(const ModuleSummaryIndex *Index)
903 : TheModule(nullptr), ShouldInitializeAllMetadata(false), TheIndex(Index) {}
904
Teresa Johnsonfc801f52018-07-03 15:52:57 +0000905inline void SlotTracker::initializeIfNeeded() {
Chris Lattner3eee99c2008-08-19 04:36:02 +0000906 if (TheModule) {
907 processModule();
Craig Topperc6207612014-04-09 06:08:46 +0000908 TheModule = nullptr; ///< Prevent re-processing next time we're called.
Chris Lattner3eee99c2008-08-19 04:36:02 +0000909 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000910
Chris Lattner3eee99c2008-08-19 04:36:02 +0000911 if (TheFunction && !FunctionProcessed)
912 processFunction();
913}
914
Teresa Johnsonfc801f52018-07-03 15:52:57 +0000915void SlotTracker::initializeIndexIfNeeded() {
Teresa Johnson08d5b4e2018-05-26 02:34:13 +0000916 if (!TheIndex)
917 return;
918 processIndex();
919 TheIndex = nullptr; ///< Prevent re-processing next time we're called.
920}
921
Chris Lattner3eee99c2008-08-19 04:36:02 +0000922// Iterate through all the global variables, functions, and global
923// variable initializers and create slots for them.
924void SlotTracker::processModule() {
Chris Lattner604e3512008-08-19 04:47:09 +0000925 ST_DEBUG("begin processModule!\n");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000926
Chris Lattner3eee99c2008-08-19 04:36:02 +0000927 // Add all of the unnamed global variables to the value table.
Rafael Espindola5d8a1552015-06-17 17:33:37 +0000928 for (const GlobalVariable &Var : TheModule->globals()) {
929 if (!Var.hasName())
930 CreateModuleSlot(&Var);
Peter Collingbournecceae7f2016-05-31 23:01:54 +0000931 processGlobalObjectMetadata(Var);
Javed Absarf3d79042017-05-11 12:28:08 +0000932 auto Attrs = Var.getAttributes();
933 if (Attrs.hasAttributes())
934 CreateAttributeSetSlot(Attrs);
Devang Patel983c6b12009-07-08 21:44:25 +0000935 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000936
Rafael Espindola54fc2982015-06-17 17:53:31 +0000937 for (const GlobalAlias &A : TheModule->aliases()) {
938 if (!A.hasName())
939 CreateModuleSlot(&A);
940 }
941
Dmitry Polukhina1feff72016-04-07 12:32:19 +0000942 for (const GlobalIFunc &I : TheModule->ifuncs()) {
943 if (!I.hasName())
944 CreateModuleSlot(&I);
945 }
946
Devang Patel23e68302009-07-29 22:04:47 +0000947 // Add metadata used by named metadata.
Rafael Espindola5d8a1552015-06-17 17:33:37 +0000948 for (const NamedMDNode &NMD : TheModule->named_metadata()) {
949 for (unsigned i = 0, e = NMD.getNumOperands(); i != e; ++i)
950 CreateMetadataSlot(NMD.getOperand(i));
Devang Patel23e68302009-07-29 22:04:47 +0000951 }
952
Rafael Espindola5d8a1552015-06-17 17:33:37 +0000953 for (const Function &F : *TheModule) {
954 if (!F.hasName())
Bill Wendling829b4782013-02-11 08:43:33 +0000955 // Add all the unnamed functions to the table.
Rafael Espindola5d8a1552015-06-17 17:33:37 +0000956 CreateModuleSlot(&F);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000957
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000958 if (ShouldInitializeAllMetadata)
Rafael Espindola5d8a1552015-06-17 17:33:37 +0000959 processFunctionMetadata(F);
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +0000960
Bill Wendling829b4782013-02-11 08:43:33 +0000961 // Add all the function attributes to the table.
Bill Wendling90bc19c2013-02-20 07:21:42 +0000962 // FIXME: Add attributes of other objects?
Reid Klecknerc2cb5602017-04-12 00:38:00 +0000963 AttributeSet FnAttrs = F.getAttributes().getFnAttributes();
964 if (FnAttrs.hasAttributes())
Bill Wendling829b4782013-02-11 08:43:33 +0000965 CreateAttributeSetSlot(FnAttrs);
966 }
967
Chris Lattner604e3512008-08-19 04:47:09 +0000968 ST_DEBUG("end processModule!\n");
Chris Lattner3eee99c2008-08-19 04:36:02 +0000969}
970
Chris Lattner3eee99c2008-08-19 04:36:02 +0000971// Process the arguments, basic blocks, and instructions of a function.
972void SlotTracker::processFunction() {
Chris Lattner604e3512008-08-19 04:47:09 +0000973 ST_DEBUG("begin processFunction!\n");
Chris Lattner3eee99c2008-08-19 04:36:02 +0000974 fNext = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000975
Duncan P. N. Exon Smithadc54562015-09-11 01:34:59 +0000976 // Process function metadata if it wasn't hit at the module-level.
977 if (!ShouldInitializeAllMetadata)
978 processFunctionMetadata(*TheFunction);
979
Chris Lattner3eee99c2008-08-19 04:36:02 +0000980 // Add all the function arguments with no names.
981 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
982 AE = TheFunction->arg_end(); AI != AE; ++AI)
983 if (!AI->hasName())
Duncan P. N. Exon Smith52888a62015-10-08 23:49:46 +0000984 CreateFunctionSlot(&*AI);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000985
Chris Lattner604e3512008-08-19 04:47:09 +0000986 ST_DEBUG("Inserting Instructions:\n");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000987
Chris Lattner3eee99c2008-08-19 04:36:02 +0000988 // Add all of the basic blocks and instructions with no names.
Duncan P. N. Exon Smith27f33ee2015-03-14 19:44:01 +0000989 for (auto &BB : *TheFunction) {
990 if (!BB.hasName())
991 CreateFunctionSlot(&BB);
Andrew Trickec4b6e72011-09-30 19:48:58 +0000992
Duncan P. N. Exon Smith27f33ee2015-03-14 19:44:01 +0000993 for (auto &I : BB) {
994 if (!I.getType()->isVoidTy() && !I.hasName())
995 CreateFunctionSlot(&I);
Andrew Trickec4b6e72011-09-30 19:48:58 +0000996
Duncan P. N. Exon Smith20b76ac2015-03-14 19:48:31 +0000997 // We allow direct calls to any llvm.foo function here, because the
998 // target may not be linked into the optimizer.
Chandler Carruth90c09232019-01-07 07:31:49 +0000999 if (const auto *Call = dyn_cast<CallBase>(&I)) {
Bill Wendlinga0323742013-02-22 09:09:42 +00001000 // Add all the call attributes to the table.
Chandler Carruth90c09232019-01-07 07:31:49 +00001001 AttributeSet Attrs = Call->getAttributes().getFnAttributes();
Reid Klecknerc2cb5602017-04-12 00:38:00 +00001002 if (Attrs.hasAttributes())
Bill Wendlinga0323742013-02-22 09:09:42 +00001003 CreateAttributeSetSlot(Attrs);
Chris Lattner58aff8f2010-05-10 20:53:17 +00001004 }
Devang Patel983c6b12009-07-08 21:44:25 +00001005 }
Chris Lattner3eee99c2008-08-19 04:36:02 +00001006 }
Devang Pateldec23fd2009-09-16 20:21:17 +00001007
Chris Lattner3eee99c2008-08-19 04:36:02 +00001008 FunctionProcessed = true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001009
Chris Lattner604e3512008-08-19 04:47:09 +00001010 ST_DEBUG("end processFunction!\n");
Chris Lattner3eee99c2008-08-19 04:36:02 +00001011}
1012
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00001013// Iterate through all the GUID in the index and create slots for them.
1014void SlotTracker::processIndex() {
1015 ST_DEBUG("begin processIndex!\n");
1016 assert(TheIndex);
1017
1018 // The first block of slots are just the module ids, which start at 0 and are
Teresa Johnson8fc76662018-07-02 22:09:23 +00001019 // assigned consecutively. Since the StringMap iteration order isn't
1020 // guaranteed, use a std::map to order by module ID before assigning slots.
1021 std::map<uint64_t, StringRef> ModuleIdToPathMap;
1022 for (auto &ModPath : TheIndex->modulePaths())
1023 ModuleIdToPathMap[ModPath.second.first] = ModPath.first();
1024 for (auto &ModPair : ModuleIdToPathMap)
1025 CreateModulePathSlot(ModPair.second);
1026
1027 // Start numbering the GUIDs after the module ids.
1028 GUIDNext = ModulePathNext;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00001029
1030 for (auto &GlobalList : *TheIndex)
1031 CreateGUIDSlot(GlobalList.first);
1032
Teresa Johnson7fb39df2018-09-25 20:14:40 +00001033 // Start numbering the TypeIds after the GUIDs.
1034 TypeIdNext = GUIDNext;
1035
1036 for (auto TidIter = TheIndex->typeIds().begin();
1037 TidIter != TheIndex->typeIds().end(); TidIter++)
1038 CreateTypeIdSlot(TidIter->second.first);
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00001039
1040 ST_DEBUG("end processIndex!\n");
1041}
1042
Peter Collingbournecceae7f2016-05-31 23:01:54 +00001043void SlotTracker::processGlobalObjectMetadata(const GlobalObject &GO) {
Duncan P. N. Exon Smith3d4cd752015-04-24 22:04:41 +00001044 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
Peter Collingbournecceae7f2016-05-31 23:01:54 +00001045 GO.getAllMetadata(MDs);
Duncan P. N. Exon Smithadc54562015-09-11 01:34:59 +00001046 for (auto &MD : MDs)
1047 CreateMetadataSlot(MD.second);
Peter Collingbournecceae7f2016-05-31 23:01:54 +00001048}
Duncan P. N. Exon Smith3d4cd752015-04-24 22:04:41 +00001049
Peter Collingbournecceae7f2016-05-31 23:01:54 +00001050void SlotTracker::processFunctionMetadata(const Function &F) {
1051 processGlobalObjectMetadata(F);
Duncan P. N. Exon Smithadc54562015-09-11 01:34:59 +00001052 for (auto &BB : F) {
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00001053 for (auto &I : BB)
1054 processInstructionMetadata(I);
Duncan P. N. Exon Smith3d4cd752015-04-24 22:04:41 +00001055 }
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00001056}
1057
Duncan P. N. Exon Smith20b76ac2015-03-14 19:48:31 +00001058void SlotTracker::processInstructionMetadata(const Instruction &I) {
1059 // Process metadata used directly by intrinsics.
1060 if (const CallInst *CI = dyn_cast<CallInst>(&I))
1061 if (Function *F = CI->getCalledFunction())
Justin Lebar9cbc3012016-07-28 23:58:15 +00001062 if (F->isIntrinsic())
Duncan P. N. Exon Smith20b76ac2015-03-14 19:48:31 +00001063 for (auto &Op : I.operands())
1064 if (auto *V = dyn_cast_or_null<MetadataAsValue>(Op))
1065 if (MDNode *N = dyn_cast<MDNode>(V->getMetadata()))
1066 CreateMetadataSlot(N);
1067
1068 // Process metadata attached to this instruction.
1069 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
1070 I.getAllMetadata(MDs);
1071 for (auto &MD : MDs)
1072 CreateMetadataSlot(MD.second);
1073}
1074
Chris Lattner3eee99c2008-08-19 04:36:02 +00001075/// Clean up after incorporating a function. This is the only way to get out of
1076/// the function incorporation state that affects get*Slot/Create*Slot. Function
1077/// incorporation state is indicated by TheFunction != 0.
1078void SlotTracker::purgeFunction() {
Chris Lattner604e3512008-08-19 04:47:09 +00001079 ST_DEBUG("begin purgeFunction!\n");
Chris Lattner3eee99c2008-08-19 04:36:02 +00001080 fMap.clear(); // Simply discard the function level map
Craig Topperc6207612014-04-09 06:08:46 +00001081 TheFunction = nullptr;
Chris Lattner3eee99c2008-08-19 04:36:02 +00001082 FunctionProcessed = false;
Chris Lattner604e3512008-08-19 04:47:09 +00001083 ST_DEBUG("end purgeFunction!\n");
Chris Lattner3eee99c2008-08-19 04:36:02 +00001084}
1085
1086/// getGlobalSlot - Get the slot number of a global value.
1087int SlotTracker::getGlobalSlot(const GlobalValue *V) {
1088 // Check for uninitialized state and do lazy initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +00001089 initializeIfNeeded();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001090
Jay Foaddbb221d2011-07-11 07:28:49 +00001091 // Find the value in the module map
Chris Lattner3eee99c2008-08-19 04:36:02 +00001092 ValueMap::iterator MI = mMap.find(V);
Dan Gohman1dd27572008-10-01 19:58:59 +00001093 return MI == mMap.end() ? -1 : (int)MI->second;
Chris Lattner3eee99c2008-08-19 04:36:02 +00001094}
1095
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +00001096/// getMetadataSlot - Get the slot number of a MDNode.
1097int SlotTracker::getMetadataSlot(const MDNode *N) {
Devang Patel983c6b12009-07-08 21:44:25 +00001098 // Check for uninitialized state and do lazy initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +00001099 initializeIfNeeded();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001100
Jay Foaddbb221d2011-07-11 07:28:49 +00001101 // Find the MDNode in the module map
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +00001102 mdn_iterator MI = mdnMap.find(N);
Devang Patel983c6b12009-07-08 21:44:25 +00001103 return MI == mdnMap.end() ? -1 : (int)MI->second;
1104}
1105
Chris Lattner3eee99c2008-08-19 04:36:02 +00001106/// getLocalSlot - Get the slot number for a value that is local to a function.
1107int SlotTracker::getLocalSlot(const Value *V) {
1108 assert(!isa<Constant>(V) && "Can't get a constant or global slot with this!");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001109
Chris Lattner3eee99c2008-08-19 04:36:02 +00001110 // Check for uninitialized state and do lazy initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +00001111 initializeIfNeeded();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001112
Chris Lattner3eee99c2008-08-19 04:36:02 +00001113 ValueMap::iterator FI = fMap.find(V);
Dan Gohman1dd27572008-10-01 19:58:59 +00001114 return FI == fMap.end() ? -1 : (int)FI->second;
Chris Lattner3eee99c2008-08-19 04:36:02 +00001115}
1116
Reid Klecknerc2cb5602017-04-12 00:38:00 +00001117int SlotTracker::getAttributeGroupSlot(AttributeSet AS) {
Bill Wendling829b4782013-02-11 08:43:33 +00001118 // Check for uninitialized state and do lazy initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +00001119 initializeIfNeeded();
Bill Wendling829b4782013-02-11 08:43:33 +00001120
Reid Klecknerc2cb5602017-04-12 00:38:00 +00001121 // Find the AttributeSet in the module map.
Bill Wendling829b4782013-02-11 08:43:33 +00001122 as_iterator AI = asMap.find(AS);
1123 return AI == asMap.end() ? -1 : (int)AI->second;
1124}
Chris Lattner3eee99c2008-08-19 04:36:02 +00001125
Teresa Johnson8fc76662018-07-02 22:09:23 +00001126int SlotTracker::getModulePathSlot(StringRef Path) {
1127 // Check for uninitialized state and do lazy initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +00001128 initializeIndexIfNeeded();
Teresa Johnson8fc76662018-07-02 22:09:23 +00001129
1130 // Find the Module path in the map
1131 auto I = ModulePathMap.find(Path);
1132 return I == ModulePathMap.end() ? -1 : (int)I->second;
1133}
1134
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00001135int SlotTracker::getGUIDSlot(GlobalValue::GUID GUID) {
1136 // Check for uninitialized state and do lazy initialization.
Teresa Johnsonfc801f52018-07-03 15:52:57 +00001137 initializeIndexIfNeeded();
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00001138
1139 // Find the GUID in the map
1140 guid_iterator I = GUIDMap.find(GUID);
1141 return I == GUIDMap.end() ? -1 : (int)I->second;
1142}
1143
Teresa Johnson7fb39df2018-09-25 20:14:40 +00001144int SlotTracker::getTypeIdSlot(StringRef Id) {
1145 // Check for uninitialized state and do lazy initialization.
1146 initializeIndexIfNeeded();
1147
1148 // Find the TypeId string in the map
1149 auto I = TypeIdMap.find(Id);
1150 return I == TypeIdMap.end() ? -1 : (int)I->second;
1151}
1152
Chris Lattner3eee99c2008-08-19 04:36:02 +00001153/// CreateModuleSlot - Insert the specified GlobalValue* into the slot table.
1154void SlotTracker::CreateModuleSlot(const GlobalValue *V) {
1155 assert(V && "Can't insert a null Value into SlotTracker!");
Chris Lattner031560c2009-12-29 07:25:48 +00001156 assert(!V->getType()->isVoidTy() && "Doesn't need a slot!");
Chris Lattner3eee99c2008-08-19 04:36:02 +00001157 assert(!V->hasName() && "Doesn't need a slot!");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001158
Chris Lattner3eee99c2008-08-19 04:36:02 +00001159 unsigned DestSlot = mNext++;
1160 mMap[V] = DestSlot;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001161
Chris Lattner604e3512008-08-19 04:47:09 +00001162 ST_DEBUG(" Inserting value [" << V->getType() << "] = " << V << " slot=" <<
Chris Lattner3eee99c2008-08-19 04:36:02 +00001163 DestSlot << " [");
Dmitry Polukhina1feff72016-04-07 12:32:19 +00001164 // G = Global, F = Function, A = Alias, I = IFunc, o = other
Chris Lattner604e3512008-08-19 04:47:09 +00001165 ST_DEBUG((isa<GlobalVariable>(V) ? 'G' :
Chris Lattner3eee99c2008-08-19 04:36:02 +00001166 (isa<Function>(V) ? 'F' :
Dmitry Polukhina1feff72016-04-07 12:32:19 +00001167 (isa<GlobalAlias>(V) ? 'A' :
1168 (isa<GlobalIFunc>(V) ? 'I' : 'o')))) << "]\n");
Chris Lattner3eee99c2008-08-19 04:36:02 +00001169}
1170
Chris Lattner3eee99c2008-08-19 04:36:02 +00001171/// CreateSlot - Create a new slot for the specified value if it has no name.
1172void SlotTracker::CreateFunctionSlot(const Value *V) {
Chris Lattner031560c2009-12-29 07:25:48 +00001173 assert(!V->getType()->isVoidTy() && !V->hasName() && "Doesn't need a slot!");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001174
Chris Lattner3eee99c2008-08-19 04:36:02 +00001175 unsigned DestSlot = fNext++;
1176 fMap[V] = DestSlot;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001177
Chris Lattner3eee99c2008-08-19 04:36:02 +00001178 // G = Global, F = Function, o = other
Chris Lattner604e3512008-08-19 04:47:09 +00001179 ST_DEBUG(" Inserting value [" << V->getType() << "] = " << V << " slot=" <<
Chris Lattner3eee99c2008-08-19 04:36:02 +00001180 DestSlot << " [o]\n");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001181}
Chris Lattner3eee99c2008-08-19 04:36:02 +00001182
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +00001183/// CreateModuleSlot - Insert the specified MDNode* into the slot table.
1184void SlotTracker::CreateMetadataSlot(const MDNode *N) {
1185 assert(N && "Can't insert a null Value into SlotTracker!");
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001186
Reid Kleckner6d353342017-08-23 20:31:27 +00001187 // Don't make slots for DIExpressions. We just print them inline everywhere.
1188 if (isa<DIExpression>(N))
1189 return;
1190
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001191 unsigned DestSlot = mdnNext;
1192 if (!mdnMap.insert(std::make_pair(N, DestSlot)).second)
1193 return;
1194 ++mdnNext;
Devang Patel983c6b12009-07-08 21:44:25 +00001195
Chris Lattner0d50bdd2009-12-31 02:27:30 +00001196 // Recursively add any MDNodes referenced by operands.
1197 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i)
1198 if (const MDNode *Op = dyn_cast_or_null<MDNode>(N->getOperand(i)))
1199 CreateMetadataSlot(Op);
Devang Patel983c6b12009-07-08 21:44:25 +00001200}
Chris Lattner3eee99c2008-08-19 04:36:02 +00001201
Reid Klecknerc2cb5602017-04-12 00:38:00 +00001202void SlotTracker::CreateAttributeSetSlot(AttributeSet AS) {
1203 assert(AS.hasAttributes() && "Doesn't need a slot!");
Bill Wendling829b4782013-02-11 08:43:33 +00001204
1205 as_iterator I = asMap.find(AS);
1206 if (I != asMap.end())
1207 return;
1208
1209 unsigned DestSlot = asNext++;
1210 asMap[AS] = DestSlot;
1211}
1212
Teresa Johnson8fc76662018-07-02 22:09:23 +00001213/// Create a new slot for the specified Module
1214void SlotTracker::CreateModulePathSlot(StringRef Path) {
1215 ModulePathMap[Path] = ModulePathNext++;
1216}
1217
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00001218/// Create a new slot for the specified GUID
1219void SlotTracker::CreateGUIDSlot(GlobalValue::GUID GUID) {
1220 GUIDMap[GUID] = GUIDNext++;
1221}
1222
Teresa Johnson7fb39df2018-09-25 20:14:40 +00001223/// Create a new slot for the specified Id
1224void SlotTracker::CreateTypeIdSlot(StringRef Id) {
1225 TypeIdMap[Id] = TypeIdNext++;
1226}
1227
Chris Lattner3eee99c2008-08-19 04:36:02 +00001228//===----------------------------------------------------------------------===//
1229// AsmWriter Implementation
1230//===----------------------------------------------------------------------===//
Chris Lattner7f8845a2002-07-23 18:07:49 +00001231
Dan Gohman12dad632009-08-12 20:56:03 +00001232static void WriteAsOperandInternal(raw_ostream &Out, const Value *V,
Dan Gohman8061d9e2009-08-13 15:27:57 +00001233 TypePrinting *TypePrinter,
Dan Gohmana2489d12010-07-20 23:55:01 +00001234 SlotTracker *Machine,
1235 const Module *Context);
Reid Spencer58d30f22004-07-04 11:50:43 +00001236
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001237static void WriteAsOperandInternal(raw_ostream &Out, const Metadata *MD,
1238 TypePrinting *TypePrinter,
1239 SlotTracker *Machine, const Module *Context,
1240 bool FromValue = false);
1241
Dan Gohman12dad632009-08-12 20:56:03 +00001242static void WriteOptimizationInfo(raw_ostream &Out, const User *U) {
Michael Ilseman92053172012-11-27 00:42:44 +00001243 if (const FPMathOperator *FPO = dyn_cast<const FPMathOperator>(U)) {
Sanjay Patel629c4112017-11-06 16:27:15 +00001244 // 'Fast' is an abbreviation for all fast-math-flags.
1245 if (FPO->isFast())
Michael Ilseman92053172012-11-27 00:42:44 +00001246 Out << " fast";
1247 else {
Sanjay Patel629c4112017-11-06 16:27:15 +00001248 if (FPO->hasAllowReassoc())
1249 Out << " reassoc";
Michael Ilseman92053172012-11-27 00:42:44 +00001250 if (FPO->hasNoNaNs())
1251 Out << " nnan";
1252 if (FPO->hasNoInfs())
1253 Out << " ninf";
1254 if (FPO->hasNoSignedZeros())
1255 Out << " nsz";
1256 if (FPO->hasAllowReciprocal())
1257 Out << " arcp";
Adam Nemetcd847a82017-03-28 20:11:52 +00001258 if (FPO->hasAllowContract())
1259 Out << " contract";
Sanjay Patel629c4112017-11-06 16:27:15 +00001260 if (FPO->hasApproxFunc())
1261 Out << " afn";
Michael Ilseman92053172012-11-27 00:42:44 +00001262 }
1263 }
1264
Dan Gohman0ebd6962009-07-20 21:19:07 +00001265 if (const OverflowingBinaryOperator *OBO =
1266 dyn_cast<OverflowingBinaryOperator>(U)) {
Dan Gohman16f54152009-08-20 17:11:38 +00001267 if (OBO->hasNoUnsignedWrap())
Dan Gohman9c7f8082009-07-27 16:11:46 +00001268 Out << " nuw";
Dan Gohman16f54152009-08-20 17:11:38 +00001269 if (OBO->hasNoSignedWrap())
Dan Gohman9c7f8082009-07-27 16:11:46 +00001270 Out << " nsw";
Chris Lattner35315d02011-02-06 21:44:57 +00001271 } else if (const PossiblyExactOperator *Div =
1272 dyn_cast<PossiblyExactOperator>(U)) {
Dan Gohman0ebd6962009-07-20 21:19:07 +00001273 if (Div->isExact())
Dan Gohman9c7f8082009-07-27 16:11:46 +00001274 Out << " exact";
Dan Gohman1639c392009-07-27 21:53:46 +00001275 } else if (const GEPOperator *GEP = dyn_cast<GEPOperator>(U)) {
1276 if (GEP->isInBounds())
1277 Out << " inbounds";
Dan Gohman0ebd6962009-07-20 21:19:07 +00001278 }
1279}
1280
Dan Gohmanefb8dbb2010-07-14 20:57:55 +00001281static void WriteConstantInternal(raw_ostream &Out, const Constant *CV,
1282 TypePrinting &TypePrinter,
Dan Gohmana2489d12010-07-20 23:55:01 +00001283 SlotTracker *Machine,
1284 const Module *Context) {
Zhou Sheng75b871f2007-01-11 12:24:14 +00001285 if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
Duncan Sands9dff9be2010-02-15 16:12:20 +00001286 if (CI->getType()->isIntegerTy(1)) {
Reid Spencercddc9df2007-01-12 04:24:46 +00001287 Out << (CI->getZExtValue() ? "true" : "false");
Chris Lattner17f71652008-08-17 07:19:36 +00001288 return;
1289 }
1290 Out << CI->getValue();
1291 return;
1292 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001293
Chris Lattner17f71652008-08-17 07:19:36 +00001294 if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001295 const APFloat &APF = CFP->getValueAPF();
1296 if (&APF.getSemantics() == &APFloat::IEEEsingle() ||
1297 &APF.getSemantics() == &APFloat::IEEEdouble()) {
Dale Johannesen028084e2007-09-12 03:30:33 +00001298 // We would like to output the FP constant value in exponential notation,
1299 // but we cannot do this if doing so will lose precision. Check here to
1300 // make sure that we only output it in exponential format if we can parse
1301 // the value back and get the same value.
1302 //
Dale Johannesen1f864982009-01-21 20:32:55 +00001303 bool ignored;
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001304 bool isDouble = &APF.getSemantics() == &APFloat::IEEEdouble();
1305 bool isInf = APF.isInfinity();
1306 bool isNaN = APF.isNaN();
Justin Bognerb609b6b2015-12-04 01:14:24 +00001307 if (!isInf && !isNaN) {
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001308 double Val = isDouble ? APF.convertToDouble() : APF.convertToFloat();
Dan Gohman518cda42011-12-17 00:04:22 +00001309 SmallString<128> StrVal;
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001310 APF.toString(StrVal, 6, 0, false);
Dan Gohman518cda42011-12-17 00:04:22 +00001311 // Check to make sure that the stringized number is not some string like
1312 // "Inf" or NaN, that atof will accept, but the lexer will not. Check
1313 // that the string matches the "[-+]?[0-9]" regex.
1314 //
Serguei Katkovc9a752c2017-04-21 06:14:38 +00001315 assert(((StrVal[0] >= '0' && StrVal[0] <= '9') ||
1316 ((StrVal[0] == '-' || StrVal[0] == '+') &&
1317 (StrVal[1] >= '0' && StrVal[1] <= '9'))) &&
1318 "[-+]?[0-9] regex does not match!");
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001319 // Reparse stringized version!
1320 if (APFloat(APFloat::IEEEdouble(), StrVal).convertToDouble() == Val) {
1321 Out << StrVal;
1322 return;
Dale Johannesen028084e2007-09-12 03:30:33 +00001323 }
Chris Lattner1e194682002-04-18 18:53:13 +00001324 }
Dale Johannesen028084e2007-09-12 03:30:33 +00001325 // Otherwise we could not reparse it to exactly the same value, so we must
Dale Johannesen1f864982009-01-21 20:32:55 +00001326 // output the string in hexadecimal format! Note that loading and storing
1327 // floating point types changes the bits of NaNs on some hosts, notably
1328 // x86, so we must not use these types.
Benjamin Kramer86c77412014-03-15 18:47:07 +00001329 static_assert(sizeof(double) == sizeof(uint64_t),
1330 "assuming that double is 64 bits!");
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001331 APFloat apf = APF;
Justin Bognerb609b6b2015-12-04 01:14:24 +00001332 // Floats are represented in ASCII IR as double, convert.
Dale Johannesen1f864982009-01-21 20:32:55 +00001333 if (!isDouble)
Stephan Bergmann17c7f702016-12-14 11:57:17 +00001334 apf.convert(APFloat::IEEEdouble(), APFloat::rmNearestTiesToEven,
Dale Johannesen1f864982009-01-21 20:32:55 +00001335 &ignored);
Justin Bogner93289572015-12-04 02:14:34 +00001336 Out << format_hex(apf.bitcastToAPInt().getZExtValue(), 0, /*Upper=*/true);
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001337 return;
1338 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001339
Tobias Grosser6b31d172012-05-24 15:59:06 +00001340 // Either half, or some form of long double.
1341 // These appear as a magic letter identifying the type, then a
1342 // fixed number of hex digits.
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001343 Out << "0x";
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001344 APInt API = APF.bitcastToAPInt();
1345 if (&APF.getSemantics() == &APFloat::x87DoubleExtended()) {
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001346 Out << 'K';
Justin Bogner93289572015-12-04 02:14:34 +00001347 Out << format_hex_no_prefix(API.getHiBits(16).getZExtValue(), 4,
1348 /*Upper=*/true);
1349 Out << format_hex_no_prefix(API.getLoBits(64).getZExtValue(), 16,
1350 /*Upper=*/true);
Dale Johannesen93eefa02009-03-23 21:16:53 +00001351 return;
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001352 } else if (&APF.getSemantics() == &APFloat::IEEEquad()) {
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001353 Out << 'L';
Justin Bogner93289572015-12-04 02:14:34 +00001354 Out << format_hex_no_prefix(API.getLoBits(64).getZExtValue(), 16,
1355 /*Upper=*/true);
1356 Out << format_hex_no_prefix(API.getHiBits(64).getZExtValue(), 16,
1357 /*Upper=*/true);
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001358 } else if (&APF.getSemantics() == &APFloat::PPCDoubleDouble()) {
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001359 Out << 'M';
Justin Bogner93289572015-12-04 02:14:34 +00001360 Out << format_hex_no_prefix(API.getLoBits(64).getZExtValue(), 16,
1361 /*Upper=*/true);
1362 Out << format_hex_no_prefix(API.getHiBits(64).getZExtValue(), 16,
1363 /*Upper=*/true);
Serguei Katkov3a46eb42017-04-21 02:52:17 +00001364 } else if (&APF.getSemantics() == &APFloat::IEEEhalf()) {
Tobias Grosser6b31d172012-05-24 15:59:06 +00001365 Out << 'H';
Justin Bogner93289572015-12-04 02:14:34 +00001366 Out << format_hex_no_prefix(API.getZExtValue(), 4,
1367 /*Upper=*/true);
Tobias Grosser6b31d172012-05-24 15:59:06 +00001368 } else
Torok Edwinfbcc6632009-07-14 16:55:14 +00001369 llvm_unreachable("Unsupported floating point type");
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001370 return;
1371 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001372
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001373 if (isa<ConstantAggregateZero>(CV)) {
Chris Lattner76b2ff42004-02-15 05:55:15 +00001374 Out << "zeroinitializer";
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001375 return;
1376 }
Andrew Trickec4b6e72011-09-30 19:48:58 +00001377
Chris Lattner214cc702009-10-28 03:38:12 +00001378 if (const BlockAddress *BA = dyn_cast<BlockAddress>(CV)) {
1379 Out << "blockaddress(";
Dan Gohmana2489d12010-07-20 23:55:01 +00001380 WriteAsOperandInternal(Out, BA->getFunction(), &TypePrinter, Machine,
1381 Context);
Chris Lattner214cc702009-10-28 03:38:12 +00001382 Out << ", ";
Dan Gohmana2489d12010-07-20 23:55:01 +00001383 WriteAsOperandInternal(Out, BA->getBasicBlock(), &TypePrinter, Machine,
1384 Context);
Chris Lattner214cc702009-10-28 03:38:12 +00001385 Out << ")";
1386 return;
1387 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001388
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001389 if (const ConstantArray *CA = dyn_cast<ConstantArray>(CV)) {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00001390 Type *ETy = CA->getType()->getElementType();
Chris Lattnercf9e8f62012-02-05 02:29:43 +00001391 Out << '[';
1392 TypePrinter.print(ETy, Out);
1393 Out << ' ';
1394 WriteAsOperandInternal(Out, CA->getOperand(0),
1395 &TypePrinter, Machine,
1396 Context);
1397 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i) {
1398 Out << ", ";
Chris Lattner9c181f62012-01-31 03:15:40 +00001399 TypePrinter.print(ETy, Out);
1400 Out << ' ';
Chris Lattnercf9e8f62012-02-05 02:29:43 +00001401 WriteAsOperandInternal(Out, CA->getOperand(i), &TypePrinter, Machine,
Chris Lattner9c181f62012-01-31 03:15:40 +00001402 Context);
Chris Lattnerd84bb632002-04-16 21:36:08 +00001403 }
Chris Lattnercf9e8f62012-02-05 02:29:43 +00001404 Out << ']';
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001405 return;
1406 }
Michael Ilseman26ee2b82012-11-15 22:34:00 +00001407
Chris Lattnerfa775002012-01-26 02:32:04 +00001408 if (const ConstantDataArray *CA = dyn_cast<ConstantDataArray>(CV)) {
1409 // As a special case, print the array as a string if it is an array of
1410 // i8 with ConstantInt values.
1411 if (CA->isString()) {
1412 Out << "c\"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00001413 printEscapedString(CA->getAsString(), Out);
Chris Lattnerfa775002012-01-26 02:32:04 +00001414 Out << '"';
1415 return;
1416 }
1417
1418 Type *ETy = CA->getType()->getElementType();
1419 Out << '[';
Chris Lattner9c181f62012-01-31 03:15:40 +00001420 TypePrinter.print(ETy, Out);
1421 Out << ' ';
1422 WriteAsOperandInternal(Out, CA->getElementAsConstant(0),
1423 &TypePrinter, Machine,
1424 Context);
1425 for (unsigned i = 1, e = CA->getNumElements(); i != e; ++i) {
1426 Out << ", ";
Chris Lattnerfa775002012-01-26 02:32:04 +00001427 TypePrinter.print(ETy, Out);
1428 Out << ' ';
Chris Lattner9c181f62012-01-31 03:15:40 +00001429 WriteAsOperandInternal(Out, CA->getElementAsConstant(i), &TypePrinter,
1430 Machine, Context);
Chris Lattnerfa775002012-01-26 02:32:04 +00001431 }
Chris Lattner9c181f62012-01-31 03:15:40 +00001432 Out << ']';
Chris Lattnerfa775002012-01-26 02:32:04 +00001433 return;
1434 }
1435
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001436 if (const ConstantStruct *CS = dyn_cast<ConstantStruct>(CV)) {
Andrew Lenharth0d124b82007-01-08 18:21:30 +00001437 if (CS->getType()->isPacked())
1438 Out << '<';
Misha Brukman21bbdb92004-06-04 21:11:51 +00001439 Out << '{';
Jim Laskey3bb78742006-02-25 12:27:03 +00001440 unsigned N = CS->getNumOperands();
1441 if (N) {
Chris Lattner604e3512008-08-19 04:47:09 +00001442 Out << ' ';
Chris Lattnere101c442009-02-28 21:26:53 +00001443 TypePrinter.print(CS->getOperand(0)->getType(), Out);
Dan Gohman81313fd2008-09-14 17:21:12 +00001444 Out << ' ';
Chris Lattnerd84bb632002-04-16 21:36:08 +00001445
Dan Gohmana2489d12010-07-20 23:55:01 +00001446 WriteAsOperandInternal(Out, CS->getOperand(0), &TypePrinter, Machine,
1447 Context);
Chris Lattnerd84bb632002-04-16 21:36:08 +00001448
Jim Laskey3bb78742006-02-25 12:27:03 +00001449 for (unsigned i = 1; i < N; i++) {
Chris Lattnerd84bb632002-04-16 21:36:08 +00001450 Out << ", ";
Chris Lattnere101c442009-02-28 21:26:53 +00001451 TypePrinter.print(CS->getOperand(i)->getType(), Out);
Dan Gohman81313fd2008-09-14 17:21:12 +00001452 Out << ' ';
Chris Lattnerd84bb632002-04-16 21:36:08 +00001453
Dan Gohmana2489d12010-07-20 23:55:01 +00001454 WriteAsOperandInternal(Out, CS->getOperand(i), &TypePrinter, Machine,
1455 Context);
Chris Lattnerd84bb632002-04-16 21:36:08 +00001456 }
Dan Gohman81313fd2008-09-14 17:21:12 +00001457 Out << ' ';
Chris Lattnerd84bb632002-04-16 21:36:08 +00001458 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001459
Dan Gohman81313fd2008-09-14 17:21:12 +00001460 Out << '}';
Andrew Lenharth0d124b82007-01-08 18:21:30 +00001461 if (CS->getType()->isPacked())
1462 Out << '>';
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001463 return;
1464 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001465
Chris Lattnerfa775002012-01-26 02:32:04 +00001466 if (isa<ConstantVector>(CV) || isa<ConstantDataVector>(CV)) {
1467 Type *ETy = CV->getType()->getVectorElementType();
Dan Gohman1262a252009-02-11 00:25:25 +00001468 Out << '<';
Chris Lattnere101c442009-02-28 21:26:53 +00001469 TypePrinter.print(ETy, Out);
Dan Gohman81313fd2008-09-14 17:21:12 +00001470 Out << ' ';
Chris Lattnerfa775002012-01-26 02:32:04 +00001471 WriteAsOperandInternal(Out, CV->getAggregateElement(0U), &TypePrinter,
1472 Machine, Context);
1473 for (unsigned i = 1, e = CV->getType()->getVectorNumElements(); i != e;++i){
Chris Lattner585297e82008-08-19 05:26:17 +00001474 Out << ", ";
Chris Lattnere101c442009-02-28 21:26:53 +00001475 TypePrinter.print(ETy, Out);
Dan Gohman81313fd2008-09-14 17:21:12 +00001476 Out << ' ';
Chris Lattnerfa775002012-01-26 02:32:04 +00001477 WriteAsOperandInternal(Out, CV->getAggregateElement(i), &TypePrinter,
1478 Machine, Context);
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001479 }
Dan Gohman1262a252009-02-11 00:25:25 +00001480 Out << '>';
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001481 return;
1482 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001483
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001484 if (isa<ConstantPointerNull>(CV)) {
Chris Lattnerd84bb632002-04-16 21:36:08 +00001485 Out << "null";
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001486 return;
1487 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001488
David Majnemerf0f224d2015-11-11 21:57:16 +00001489 if (isa<ConstantTokenNone>(CV)) {
1490 Out << "none";
1491 return;
1492 }
1493
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001494 if (isa<UndefValue>(CV)) {
Chris Lattner5e0b9f22004-10-16 18:08:06 +00001495 Out << "undef";
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001496 return;
1497 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001498
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001499 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
Reid Spencer812a1be2006-12-04 05:19:18 +00001500 Out << CE->getOpcodeName();
Dan Gohman9c7f8082009-07-27 16:11:46 +00001501 WriteOptimizationInfo(Out, CE);
Reid Spencer812a1be2006-12-04 05:19:18 +00001502 if (CE->isCompare())
Tim Northoverde3aea0412016-08-17 20:25:25 +00001503 Out << ' ' << CmpInst::getPredicateName(
1504 static_cast<CmpInst::Predicate>(CE->getPredicate()));
Reid Spencer812a1be2006-12-04 05:19:18 +00001505 Out << " (";
Misha Brukmanb1c93172005-04-21 23:48:37 +00001506
Peter Collingbourned93620b2016-11-10 22:34:55 +00001507 Optional<unsigned> InRangeOp;
David Blaikief72d05b2015-03-13 18:20:45 +00001508 if (const GEPOperator *GEP = dyn_cast<GEPOperator>(CE)) {
David Blaikied06654f2015-07-29 20:26:23 +00001509 TypePrinter.print(GEP->getSourceElementType(), Out);
David Blaikief72d05b2015-03-13 18:20:45 +00001510 Out << ", ";
Peter Collingbourned93620b2016-11-10 22:34:55 +00001511 InRangeOp = GEP->getInRangeIndex();
1512 if (InRangeOp)
1513 ++*InRangeOp;
David Blaikief72d05b2015-03-13 18:20:45 +00001514 }
1515
Vikram S. Adveb952b542002-07-14 23:14:45 +00001516 for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) {
Zachary Turner804d5022016-11-11 23:58:11 +00001517 if (InRangeOp && unsigned(OI - CE->op_begin()) == *InRangeOp)
Peter Collingbourned93620b2016-11-10 22:34:55 +00001518 Out << "inrange ";
Chris Lattnere101c442009-02-28 21:26:53 +00001519 TypePrinter.print((*OI)->getType(), Out);
Dan Gohman81313fd2008-09-14 17:21:12 +00001520 Out << ' ';
Dan Gohmana2489d12010-07-20 23:55:01 +00001521 WriteAsOperandInternal(Out, *OI, &TypePrinter, Machine, Context);
Vikram S. Adveb952b542002-07-14 23:14:45 +00001522 if (OI+1 != CE->op_end())
Chris Lattner3cd8c562002-07-30 18:54:25 +00001523 Out << ", ";
Vikram S. Adveb952b542002-07-14 23:14:45 +00001524 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00001525
Dan Gohmana76f0f72008-05-31 19:12:39 +00001526 if (CE->hasIndices()) {
Jay Foad0091fe82011-04-13 15:22:40 +00001527 ArrayRef<unsigned> Indices = CE->getIndices();
Dan Gohmana76f0f72008-05-31 19:12:39 +00001528 for (unsigned i = 0, e = Indices.size(); i != e; ++i)
1529 Out << ", " << Indices[i];
1530 }
1531
Reid Spencer6c38f0b2006-11-27 01:05:10 +00001532 if (CE->isCast()) {
Chris Lattner83b396b2002-08-15 19:37:43 +00001533 Out << " to ";
Chris Lattnere101c442009-02-28 21:26:53 +00001534 TypePrinter.print(CE->getType(), Out);
Chris Lattner83b396b2002-08-15 19:37:43 +00001535 }
Reid Spencer6c38f0b2006-11-27 01:05:10 +00001536
Misha Brukman21bbdb92004-06-04 21:11:51 +00001537 Out << ')';
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001538 return;
Chris Lattnerd84bb632002-04-16 21:36:08 +00001539 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001540
Chris Lattner3a36d2c2008-08-19 05:06:27 +00001541 Out << "<placeholder or erroneous Constant>";
Chris Lattnerd84bb632002-04-16 21:36:08 +00001542}
1543
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00001544static void writeMDTuple(raw_ostream &Out, const MDTuple *Node,
1545 TypePrinting *TypePrinter, SlotTracker *Machine,
1546 const Module *Context) {
Chris Lattnercdec5812009-12-31 02:31:59 +00001547 Out << "!{";
1548 for (unsigned mi = 0, me = Node->getNumOperands(); mi != me; ++mi) {
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001549 const Metadata *MD = Node->getOperand(mi);
1550 if (!MD)
Chris Lattnercdec5812009-12-31 02:31:59 +00001551 Out << "null";
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001552 else if (auto *MDV = dyn_cast<ValueAsMetadata>(MD)) {
1553 Value *V = MDV->getValue();
Chris Lattnercdec5812009-12-31 02:31:59 +00001554 TypePrinter->print(V->getType(), Out);
1555 Out << ' ';
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001556 WriteAsOperandInternal(Out, V, TypePrinter, Machine, Context);
1557 } else {
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00001558 WriteAsOperandInternal(Out, MD, TypePrinter, Machine, Context);
Chris Lattnercdec5812009-12-31 02:31:59 +00001559 }
1560 if (mi + 1 != me)
1561 Out << ", ";
1562 }
Andrew Trickec4b6e72011-09-30 19:48:58 +00001563
Chris Lattnercdec5812009-12-31 02:31:59 +00001564 Out << "}";
1565}
1566
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001567namespace {
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001568
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001569struct FieldSeparator {
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001570 bool Skip = true;
Duncan P. N. Exon Smith70ab3d22015-02-21 01:02:18 +00001571 const char *Sep;
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001572
1573 FieldSeparator(const char *Sep = ", ") : Sep(Sep) {}
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001574};
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001575
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001576raw_ostream &operator<<(raw_ostream &OS, FieldSeparator &FS) {
1577 if (FS.Skip) {
1578 FS.Skip = false;
1579 return OS;
1580 }
Duncan P. N. Exon Smith70ab3d22015-02-21 01:02:18 +00001581 return OS << FS.Sep;
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001582}
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001583
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001584struct MDFieldPrinter {
1585 raw_ostream &Out;
1586 FieldSeparator FS;
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001587 TypePrinting *TypePrinter = nullptr;
1588 SlotTracker *Machine = nullptr;
1589 const Module *Context = nullptr;
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001590
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001591 explicit MDFieldPrinter(raw_ostream &Out) : Out(Out) {}
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001592 MDFieldPrinter(raw_ostream &Out, TypePrinting *TypePrinter,
1593 SlotTracker *Machine, const Module *Context)
1594 : Out(Out), TypePrinter(TypePrinter), Machine(Machine), Context(Context) {
1595 }
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001596
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001597 void printTag(const DINode *N);
Amjad Abouda9bcf162015-12-10 12:56:35 +00001598 void printMacinfoType(const DIMacroNode *N);
Scott Linder71603842018-02-12 19:45:54 +00001599 void printChecksum(const DIFile::ChecksumInfo<StringRef> &N);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001600 void printString(StringRef Name, StringRef Value,
1601 bool ShouldSkipEmpty = true);
1602 void printMetadata(StringRef Name, const Metadata *MD,
1603 bool ShouldSkipNull = true);
1604 template <class IntTy>
1605 void printInt(StringRef Name, IntTy Int, bool ShouldSkipZero = true);
David Blaikiea01f2952016-08-24 18:29:49 +00001606 void printBool(StringRef Name, bool Value, Optional<bool> Default = None);
Leny Kholodov5fcc4182016-09-06 10:46:28 +00001607 void printDIFlags(StringRef Name, DINode::DIFlags Flags);
Paul Robinsonadcdc1b2018-11-28 21:14:32 +00001608 void printDISPFlags(StringRef Name, DISubprogram::DISPFlags Flags);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001609 template <class IntTy, class Stringifier>
1610 void printDwarfEnum(StringRef Name, IntTy Value, Stringifier toString,
1611 bool ShouldSkipZero = true);
Adrian Prantlb939a252016-03-31 23:56:58 +00001612 void printEmissionKind(StringRef Name, DICompileUnit::DebugEmissionKind EK);
David Blaikie66cf14d2018-08-16 21:29:55 +00001613 void printNameTableKind(StringRef Name,
1614 DICompileUnit::DebugNameTableKind NTK);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001615};
Eugene Zelenkode6cce22017-06-19 22:05:08 +00001616
1617} // end anonymous namespace
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001618
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001619void MDFieldPrinter::printTag(const DINode *N) {
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001620 Out << FS << "tag: ";
Mehdi Amini149f6ea2016-10-05 05:59:29 +00001621 auto Tag = dwarf::TagString(N->getTag());
1622 if (!Tag.empty())
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001623 Out << Tag;
1624 else
1625 Out << N->getTag();
1626}
1627
Amjad Abouda9bcf162015-12-10 12:56:35 +00001628void MDFieldPrinter::printMacinfoType(const DIMacroNode *N) {
1629 Out << FS << "type: ";
Mehdi Amini149f6ea2016-10-05 05:59:29 +00001630 auto Type = dwarf::MacinfoString(N->getMacinfoType());
1631 if (!Type.empty())
Amjad Abouda9bcf162015-12-10 12:56:35 +00001632 Out << Type;
1633 else
1634 Out << N->getMacinfoType();
1635}
1636
Scott Linder71603842018-02-12 19:45:54 +00001637void MDFieldPrinter::printChecksum(
1638 const DIFile::ChecksumInfo<StringRef> &Checksum) {
1639 Out << FS << "checksumkind: " << Checksum.getKindAsString();
1640 printString("checksum", Checksum.Value, /* ShouldSkipEmpty */ false);
Amjad Aboud7faeecc2016-12-25 10:12:09 +00001641}
1642
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001643void MDFieldPrinter::printString(StringRef Name, StringRef Value,
1644 bool ShouldSkipEmpty) {
1645 if (ShouldSkipEmpty && Value.empty())
1646 return;
1647
1648 Out << FS << Name << ": \"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00001649 printEscapedString(Value, Out);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001650 Out << "\"";
1651}
1652
Duncan P. N. Exon Smith61a09332015-02-06 22:27:22 +00001653static void writeMetadataAsOperand(raw_ostream &Out, const Metadata *MD,
1654 TypePrinting *TypePrinter,
1655 SlotTracker *Machine,
1656 const Module *Context) {
1657 if (!MD) {
1658 Out << "null";
1659 return;
1660 }
1661 WriteAsOperandInternal(Out, MD, TypePrinter, Machine, Context);
1662}
1663
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001664void MDFieldPrinter::printMetadata(StringRef Name, const Metadata *MD,
1665 bool ShouldSkipNull) {
1666 if (ShouldSkipNull && !MD)
Duncan P. N. Exon Smith79cf9702015-02-28 22:16:56 +00001667 return;
1668
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001669 Out << FS << Name << ": ";
1670 writeMetadataAsOperand(Out, MD, TypePrinter, Machine, Context);
1671}
1672
1673template <class IntTy>
1674void MDFieldPrinter::printInt(StringRef Name, IntTy Int, bool ShouldSkipZero) {
1675 if (ShouldSkipZero && !Int)
1676 return;
1677
1678 Out << FS << Name << ": " << Int;
1679}
1680
David Blaikiea01f2952016-08-24 18:29:49 +00001681void MDFieldPrinter::printBool(StringRef Name, bool Value,
1682 Optional<bool> Default) {
1683 if (Default && Value == *Default)
1684 return;
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001685 Out << FS << Name << ": " << (Value ? "true" : "false");
1686}
1687
Leny Kholodov5fcc4182016-09-06 10:46:28 +00001688void MDFieldPrinter::printDIFlags(StringRef Name, DINode::DIFlags Flags) {
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001689 if (!Flags)
1690 return;
1691
1692 Out << FS << Name << ": ";
1693
Leny Kholodov5fcc4182016-09-06 10:46:28 +00001694 SmallVector<DINode::DIFlags, 8> SplitFlags;
1695 auto Extra = DINode::splitFlags(Flags, SplitFlags);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001696
1697 FieldSeparator FlagsFS(" | ");
Leny Kholodov5fcc4182016-09-06 10:46:28 +00001698 for (auto F : SplitFlags) {
Mehdi Aminif42ec792016-10-01 05:57:50 +00001699 auto StringF = DINode::getFlagString(F);
1700 assert(!StringF.empty() && "Expected valid flag");
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001701 Out << FlagsFS << StringF;
1702 }
1703 if (Extra || SplitFlags.empty())
1704 Out << FlagsFS << Extra;
1705}
1706
Paul Robinsonadcdc1b2018-11-28 21:14:32 +00001707void MDFieldPrinter::printDISPFlags(StringRef Name,
1708 DISubprogram::DISPFlags Flags) {
1709 // Always print this field, because no flags in the IR at all will be
1710 // interpreted as old-style isDefinition: true.
1711 Out << FS << Name << ": ";
1712
1713 if (!Flags) {
1714 Out << 0;
1715 return;
1716 }
1717
1718 SmallVector<DISubprogram::DISPFlags, 8> SplitFlags;
1719 auto Extra = DISubprogram::splitFlags(Flags, SplitFlags);
1720
1721 FieldSeparator FlagsFS(" | ");
1722 for (auto F : SplitFlags) {
1723 auto StringF = DISubprogram::getFlagString(F);
1724 assert(!StringF.empty() && "Expected valid flag");
1725 Out << FlagsFS << StringF;
1726 }
1727 if (Extra || SplitFlags.empty())
1728 Out << FlagsFS << Extra;
1729}
1730
Adrian Prantlb939a252016-03-31 23:56:58 +00001731void MDFieldPrinter::printEmissionKind(StringRef Name,
1732 DICompileUnit::DebugEmissionKind EK) {
Fangrui Song3c1b5db2018-07-06 19:26:00 +00001733 Out << FS << Name << ": " << DICompileUnit::emissionKindString(EK);
Adrian Prantlb939a252016-03-31 23:56:58 +00001734}
1735
David Blaikie66cf14d2018-08-16 21:29:55 +00001736void MDFieldPrinter::printNameTableKind(StringRef Name,
1737 DICompileUnit::DebugNameTableKind NTK) {
1738 if (NTK == DICompileUnit::DebugNameTableKind::Default)
1739 return;
1740 Out << FS << Name << ": " << DICompileUnit::nameTableKindString(NTK);
1741}
1742
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001743template <class IntTy, class Stringifier>
1744void MDFieldPrinter::printDwarfEnum(StringRef Name, IntTy Value,
1745 Stringifier toString, bool ShouldSkipZero) {
1746 if (!Value)
1747 return;
1748
1749 Out << FS << Name << ": ";
Mehdi Amini149f6ea2016-10-05 05:59:29 +00001750 auto S = toString(Value);
1751 if (!S.empty())
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001752 Out << S;
1753 else
1754 Out << Value;
Duncan P. N. Exon Smith79cf9702015-02-28 22:16:56 +00001755}
1756
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001757static void writeGenericDINode(raw_ostream &Out, const GenericDINode *N,
1758 TypePrinting *TypePrinter, SlotTracker *Machine,
1759 const Module *Context) {
1760 Out << "!GenericDINode(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001761 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1762 Printer.printTag(N);
1763 Printer.printString("header", N->getHeader());
Duncan P. N. Exon Smith4e4aa702015-02-03 21:54:14 +00001764 if (N->getNumDwarfOperands()) {
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001765 Out << Printer.FS << "operands: {";
Duncan P. N. Exon Smith4e4aa702015-02-03 21:54:14 +00001766 FieldSeparator IFS;
1767 for (auto &I : N->dwarf_operands()) {
1768 Out << IFS;
Duncan P. N. Exon Smith61a09332015-02-06 22:27:22 +00001769 writeMetadataAsOperand(Out, I, TypePrinter, Machine, Context);
Duncan P. N. Exon Smith4e4aa702015-02-03 21:54:14 +00001770 }
1771 Out << "}";
1772 }
1773 Out << ")";
Duncan P. N. Exon Smithfed199a2015-01-20 00:01:43 +00001774}
1775
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001776static void writeDILocation(raw_ostream &Out, const DILocation *DL,
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001777 TypePrinting *TypePrinter, SlotTracker *Machine,
1778 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001779 Out << "!DILocation(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001780 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
Duncan P. N. Exon Smith503cf3b2015-01-14 22:14:26 +00001781 // Always output the line, since 0 is a relevant and important value for it.
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001782 Printer.printInt("line", DL->getLine(), /* ShouldSkipZero */ false);
1783 Printer.printInt("column", DL->getColumn());
1784 Printer.printMetadata("scope", DL->getRawScope(), /* ShouldSkipNull */ false);
1785 Printer.printMetadata("inlinedAt", DL->getRawInlinedAt());
Calixte Denizeteb7f6022018-09-20 08:53:06 +00001786 Printer.printBool("isImplicitCode", DL->isImplicitCode(),
1787 /* Default */ false);
Duncan P. N. Exon Smithde03ff52015-01-13 20:44:56 +00001788 Out << ")";
1789}
1790
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001791static void writeDISubrange(raw_ostream &Out, const DISubrange *N,
Sander de Smalenfdf40912018-01-24 09:56:07 +00001792 TypePrinting *TypePrinter, SlotTracker *Machine,
1793 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001794 Out << "!DISubrange(";
Sander de Smalenfdf40912018-01-24 09:56:07 +00001795 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1796 if (auto *CE = N->getCount().dyn_cast<ConstantInt*>())
1797 Printer.printInt("count", CE->getSExtValue(), /* ShouldSkipZero */ false);
1798 else
1799 Printer.printMetadata("count", N->getCount().dyn_cast<DIVariable*>(),
1800 /*ShouldSkipNull */ false);
Duncan P. N. Exon Smith5dcf6212015-04-07 00:39:59 +00001801 Printer.printInt("lowerBound", N->getLowerBound());
Duncan P. N. Exon Smithc7363f12015-02-13 01:10:38 +00001802 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001803}
Duncan P. N. Exon Smithc7363f12015-02-13 01:10:38 +00001804
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001805static void writeDIEnumerator(raw_ostream &Out, const DIEnumerator *N,
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001806 TypePrinting *, SlotTracker *, const Module *) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001807 Out << "!DIEnumerator(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001808 MDFieldPrinter Printer(Out);
1809 Printer.printString("name", N->getName(), /* ShouldSkipEmpty */ false);
Momchil Velikov08dc66e2018-02-12 16:10:09 +00001810 if (N->isUnsigned()) {
1811 auto Value = static_cast<uint64_t>(N->getValue());
1812 Printer.printInt("value", Value, /* ShouldSkipZero */ false);
1813 Printer.printBool("isUnsigned", true);
1814 } else {
1815 Printer.printInt("value", N->getValue(), /* ShouldSkipZero */ false);
1816 }
Duncan P. N. Exon Smith87754762015-02-13 01:14:11 +00001817 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001818}
Duncan P. N. Exon Smith87754762015-02-13 01:14:11 +00001819
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001820static void writeDIBasicType(raw_ostream &Out, const DIBasicType *N,
Duncan P. N. Exon Smith09e03f32015-02-13 01:14:58 +00001821 TypePrinting *, SlotTracker *, const Module *) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001822 Out << "!DIBasicType(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001823 MDFieldPrinter Printer(Out);
Duncan P. N. Exon Smith16d182a2015-02-28 23:21:38 +00001824 if (N->getTag() != dwarf::DW_TAG_base_type)
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001825 Printer.printTag(N);
1826 Printer.printString("name", N->getName());
1827 Printer.printInt("size", N->getSizeInBits());
1828 Printer.printInt("align", N->getAlignInBits());
1829 Printer.printDwarfEnum("encoding", N->getEncoding(),
1830 dwarf::AttributeEncodingString);
Adrian Prantl55f42622018-08-14 19:35:34 +00001831 Printer.printDIFlags("flags", N->getFlags());
Duncan P. N. Exon Smith09e03f32015-02-13 01:14:58 +00001832 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001833}
Duncan P. N. Exon Smith09e03f32015-02-13 01:14:58 +00001834
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001835static void writeDIDerivedType(raw_ostream &Out, const DIDerivedType *N,
Duncan P. N. Exon Smith171d0772015-02-13 01:20:38 +00001836 TypePrinting *TypePrinter, SlotTracker *Machine,
1837 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001838 Out << "!DIDerivedType(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001839 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1840 Printer.printTag(N);
1841 Printer.printString("name", N->getName());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001842 Printer.printMetadata("scope", N->getRawScope());
1843 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001844 Printer.printInt("line", N->getLine());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001845 Printer.printMetadata("baseType", N->getRawBaseType(),
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001846 /* ShouldSkipNull */ false);
1847 Printer.printInt("size", N->getSizeInBits());
1848 Printer.printInt("align", N->getAlignInBits());
1849 Printer.printInt("offset", N->getOffsetInBits());
1850 Printer.printDIFlags("flags", N->getFlags());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001851 Printer.printMetadata("extraData", N->getRawExtraData());
Konstantin Zhuravlyovd5561e02017-03-08 23:55:44 +00001852 if (const auto &DWARFAddressSpace = N->getDWARFAddressSpace())
1853 Printer.printInt("dwarfAddressSpace", *DWARFAddressSpace,
1854 /* ShouldSkipZero */ false);
Duncan P. N. Exon Smith171d0772015-02-13 01:20:38 +00001855 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001856}
Duncan P. N. Exon Smith171d0772015-02-13 01:20:38 +00001857
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001858static void writeDICompositeType(raw_ostream &Out, const DICompositeType *N,
Duncan P. N. Exon Smith171d0772015-02-13 01:20:38 +00001859 TypePrinting *TypePrinter,
1860 SlotTracker *Machine, const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001861 Out << "!DICompositeType(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001862 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1863 Printer.printTag(N);
1864 Printer.printString("name", N->getName());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001865 Printer.printMetadata("scope", N->getRawScope());
1866 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001867 Printer.printInt("line", N->getLine());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001868 Printer.printMetadata("baseType", N->getRawBaseType());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001869 Printer.printInt("size", N->getSizeInBits());
1870 Printer.printInt("align", N->getAlignInBits());
1871 Printer.printInt("offset", N->getOffsetInBits());
1872 Printer.printDIFlags("flags", N->getFlags());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001873 Printer.printMetadata("elements", N->getRawElements());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001874 Printer.printDwarfEnum("runtimeLang", N->getRuntimeLang(),
1875 dwarf::LanguageString);
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001876 Printer.printMetadata("vtableHolder", N->getRawVTableHolder());
1877 Printer.printMetadata("templateParams", N->getRawTemplateParams());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001878 Printer.printString("identifier", N->getIdentifier());
Adrian Prantl8c599212018-02-06 23:45:59 +00001879 Printer.printMetadata("discriminator", N->getRawDiscriminator());
Duncan P. N. Exon Smith171d0772015-02-13 01:20:38 +00001880 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001881}
Duncan P. N. Exon Smith171d0772015-02-13 01:20:38 +00001882
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001883static void writeDISubroutineType(raw_ostream &Out, const DISubroutineType *N,
Duncan P. N. Exon Smith54e2bc62015-02-13 01:22:59 +00001884 TypePrinting *TypePrinter,
1885 SlotTracker *Machine, const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001886 Out << "!DISubroutineType(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001887 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1888 Printer.printDIFlags("flags", N->getFlags());
Reid Klecknerde3d8b52016-06-08 20:34:29 +00001889 Printer.printDwarfEnum("cc", N->getCC(), dwarf::ConventionString);
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001890 Printer.printMetadata("types", N->getRawTypeArray(),
1891 /* ShouldSkipNull */ false);
Duncan P. N. Exon Smith54e2bc62015-02-13 01:22:59 +00001892 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001893}
Duncan P. N. Exon Smithf14b9c72015-02-13 01:19:14 +00001894
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001895static void writeDIFile(raw_ostream &Out, const DIFile *N, TypePrinting *,
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001896 SlotTracker *, const Module *) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001897 Out << "!DIFile(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001898 MDFieldPrinter Printer(Out);
1899 Printer.printString("filename", N->getFilename(),
1900 /* ShouldSkipEmpty */ false);
1901 Printer.printString("directory", N->getDirectory(),
1902 /* ShouldSkipEmpty */ false);
Scott Linder71603842018-02-12 19:45:54 +00001903 // Print all values for checksum together, or not at all.
1904 if (N->getChecksum())
1905 Printer.printChecksum(*N->getChecksum());
Scott Linder16c7bda2018-02-23 23:01:06 +00001906 Printer.printString("source", N->getSource().getValueOr(StringRef()),
1907 /* ShouldSkipEmpty */ true);
Duncan P. N. Exon Smithf14b9c72015-02-13 01:19:14 +00001908 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001909}
Duncan P. N. Exon Smithf14b9c72015-02-13 01:19:14 +00001910
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001911static void writeDICompileUnit(raw_ostream &Out, const DICompileUnit *N,
Duncan P. N. Exon Smithc1f1acc2015-02-13 01:25:10 +00001912 TypePrinting *TypePrinter, SlotTracker *Machine,
1913 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001914 Out << "!DICompileUnit(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001915 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1916 Printer.printDwarfEnum("language", N->getSourceLanguage(),
1917 dwarf::LanguageString, /* ShouldSkipZero */ false);
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001918 Printer.printMetadata("file", N->getRawFile(), /* ShouldSkipNull */ false);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001919 Printer.printString("producer", N->getProducer());
1920 Printer.printBool("isOptimized", N->isOptimized());
1921 Printer.printString("flags", N->getFlags());
1922 Printer.printInt("runtimeVersion", N->getRuntimeVersion(),
1923 /* ShouldSkipZero */ false);
1924 Printer.printString("splitDebugFilename", N->getSplitDebugFilename());
Adrian Prantlb939a252016-03-31 23:56:58 +00001925 Printer.printEmissionKind("emissionKind", N->getEmissionKind());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001926 Printer.printMetadata("enums", N->getRawEnumTypes());
1927 Printer.printMetadata("retainedTypes", N->getRawRetainedTypes());
Duncan P. N. Exon Smith53855f02015-03-27 23:05:04 +00001928 Printer.printMetadata("globals", N->getRawGlobalVariables());
1929 Printer.printMetadata("imports", N->getRawImportedEntities());
Amjad Abouda9bcf162015-12-10 12:56:35 +00001930 Printer.printMetadata("macros", N->getRawMacros());
Adrian Prantl1f599f92015-05-21 20:37:30 +00001931 Printer.printInt("dwoId", N->getDWOId());
David Blaikiea01f2952016-08-24 18:29:49 +00001932 Printer.printBool("splitDebugInlining", N->getSplitDebugInlining(), true);
Dehao Chen0944a8c2017-02-01 22:45:09 +00001933 Printer.printBool("debugInfoForProfiling", N->getDebugInfoForProfiling(),
1934 false);
David Blaikie66cf14d2018-08-16 21:29:55 +00001935 Printer.printNameTableKind("nameTableKind", N->getNameTableKind());
David Blaikiebb279112018-11-13 20:08:10 +00001936 Printer.printBool("rangesBaseAddress", N->getRangesBaseAddress(), false);
Duncan P. N. Exon Smithc1f1acc2015-02-13 01:25:10 +00001937 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001938}
Duncan P. N. Exon Smithc1f1acc2015-02-13 01:25:10 +00001939
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001940static void writeDISubprogram(raw_ostream &Out, const DISubprogram *N,
Duncan P. N. Exon Smith19fc5ed2015-02-13 01:26:47 +00001941 TypePrinting *TypePrinter, SlotTracker *Machine,
1942 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001943 Out << "!DISubprogram(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001944 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1945 Printer.printString("name", N->getName());
1946 Printer.printString("linkageName", N->getLinkageName());
Duncan P. N. Exon Smith869db502015-03-30 16:19:15 +00001947 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
1948 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001949 Printer.printInt("line", N->getLine());
Duncan P. N. Exon Smith869db502015-03-30 16:19:15 +00001950 Printer.printMetadata("type", N->getRawType());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001951 Printer.printInt("scopeLine", N->getScopeLine());
Duncan P. N. Exon Smith869db502015-03-30 16:19:15 +00001952 Printer.printMetadata("containingType", N->getRawContainingType());
Peter Collingbournea1f86252016-03-17 23:58:03 +00001953 if (N->getVirtuality() != dwarf::DW_VIRTUALITY_none ||
1954 N->getVirtualIndex() != 0)
1955 Printer.printInt("virtualIndex", N->getVirtualIndex(), false);
Reid Klecknerb5af11d2016-07-01 02:41:21 +00001956 Printer.printInt("thisAdjustment", N->getThisAdjustment());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001957 Printer.printDIFlags("flags", N->getFlags());
Paul Robinsonadcdc1b2018-11-28 21:14:32 +00001958 Printer.printDISPFlags("spFlags", N->getSPFlags());
Adrian Prantl75819ae2016-04-15 15:57:41 +00001959 Printer.printMetadata("unit", N->getRawUnit());
Duncan P. N. Exon Smith869db502015-03-30 16:19:15 +00001960 Printer.printMetadata("templateParams", N->getRawTemplateParams());
1961 Printer.printMetadata("declaration", N->getRawDeclaration());
Shiva Chen2c864552018-05-09 02:40:45 +00001962 Printer.printMetadata("retainedNodes", N->getRawRetainedNodes());
Adrian Prantl1d12b882017-04-26 22:56:44 +00001963 Printer.printMetadata("thrownTypes", N->getRawThrownTypes());
Duncan P. N. Exon Smith19fc5ed2015-02-13 01:26:47 +00001964 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001965}
Duncan P. N. Exon Smith19fc5ed2015-02-13 01:26:47 +00001966
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001967static void writeDILexicalBlock(raw_ostream &Out, const DILexicalBlock *N,
1968 TypePrinting *TypePrinter, SlotTracker *Machine,
1969 const Module *Context) {
1970 Out << "!DILexicalBlock(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001971 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
Duncan P. N. Exon Smith0e202b92015-03-30 16:37:48 +00001972 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
1973 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001974 Printer.printInt("line", N->getLine());
1975 Printer.printInt("column", N->getColumn());
Duncan P. N. Exon Smitha96d4092015-02-13 01:29:28 +00001976 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001977}
Duncan P. N. Exon Smitha96d4092015-02-13 01:29:28 +00001978
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001979static void writeDILexicalBlockFile(raw_ostream &Out,
1980 const DILexicalBlockFile *N,
Duncan P. N. Exon Smith06a07022015-02-13 01:30:42 +00001981 TypePrinting *TypePrinter,
1982 SlotTracker *Machine,
1983 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001984 Out << "!DILexicalBlockFile(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001985 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
Duncan P. N. Exon Smith0e202b92015-03-30 16:37:48 +00001986 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
1987 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001988 Printer.printInt("discriminator", N->getDiscriminator(),
1989 /* ShouldSkipZero */ false);
Duncan P. N. Exon Smith06a07022015-02-13 01:30:42 +00001990 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00001991}
Duncan P. N. Exon Smith06a07022015-02-13 01:30:42 +00001992
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001993static void writeDINamespace(raw_ostream &Out, const DINamespace *N,
Duncan P. N. Exon Smithe1460002015-02-13 01:32:09 +00001994 TypePrinting *TypePrinter, SlotTracker *Machine,
1995 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00001996 Out << "!DINamespace(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00001997 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
1998 Printer.printString("name", N->getName());
Duncan P. N. Exon Smithf9b47752015-03-30 17:21:38 +00001999 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
Adrian Prantldbfda632016-11-03 19:42:02 +00002000 Printer.printBool("exportSymbols", N->getExportSymbols(), false);
Duncan P. N. Exon Smithe1460002015-02-13 01:32:09 +00002001 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002002}
Duncan P. N. Exon Smithe1460002015-02-13 01:32:09 +00002003
Adrian Prantl6ed57062019-04-08 19:13:55 +00002004static void writeDICommonBlock(raw_ostream &Out, const DICommonBlock *N,
2005 TypePrinting *TypePrinter, SlotTracker *Machine,
2006 const Module *Context) {
2007 Out << "!DICommonBlock(";
2008 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2009 Printer.printMetadata("scope", N->getRawScope(), false);
2010 Printer.printMetadata("declaration", N->getRawDecl(), false);
2011 Printer.printString("name", N->getName());
2012 Printer.printMetadata("file", N->getRawFile());
2013 Printer.printInt("line", N->getLineNo());
2014 Out << ")";
2015}
2016
Amjad Abouda9bcf162015-12-10 12:56:35 +00002017static void writeDIMacro(raw_ostream &Out, const DIMacro *N,
2018 TypePrinting *TypePrinter, SlotTracker *Machine,
2019 const Module *Context) {
2020 Out << "!DIMacro(";
2021 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2022 Printer.printMacinfoType(N);
2023 Printer.printInt("line", N->getLine());
2024 Printer.printString("name", N->getName());
2025 Printer.printString("value", N->getValue());
2026 Out << ")";
2027}
2028
2029static void writeDIMacroFile(raw_ostream &Out, const DIMacroFile *N,
2030 TypePrinting *TypePrinter, SlotTracker *Machine,
2031 const Module *Context) {
2032 Out << "!DIMacroFile(";
2033 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2034 Printer.printInt("line", N->getLine());
2035 Printer.printMetadata("file", N->getRawFile(), /* ShouldSkipNull */ false);
2036 Printer.printMetadata("nodes", N->getRawElements());
2037 Out << ")";
2038}
2039
Adrian Prantlab1243f2015-06-29 23:03:47 +00002040static void writeDIModule(raw_ostream &Out, const DIModule *N,
2041 TypePrinting *TypePrinter, SlotTracker *Machine,
2042 const Module *Context) {
2043 Out << "!DIModule(";
2044 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2045 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
2046 Printer.printString("name", N->getName());
2047 Printer.printString("configMacros", N->getConfigurationMacros());
2048 Printer.printString("includePath", N->getIncludePath());
2049 Printer.printString("isysroot", N->getISysRoot());
2050 Out << ")";
2051}
2052
2053
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002054static void writeDITemplateTypeParameter(raw_ostream &Out,
2055 const DITemplateTypeParameter *N,
Duncan P. N. Exon Smith2847f382015-02-13 01:34:32 +00002056 TypePrinting *TypePrinter,
2057 SlotTracker *Machine,
2058 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002059 Out << "!DITemplateTypeParameter(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002060 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2061 Printer.printString("name", N->getName());
Duncan P. N. Exon Smith3ec5fa62015-04-06 19:03:45 +00002062 Printer.printMetadata("type", N->getRawType(), /* ShouldSkipNull */ false);
Duncan P. N. Exon Smith2847f382015-02-13 01:34:32 +00002063 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002064}
Duncan P. N. Exon Smith2847f382015-02-13 01:34:32 +00002065
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002066static void writeDITemplateValueParameter(raw_ostream &Out,
2067 const DITemplateValueParameter *N,
Duncan P. N. Exon Smith2847f382015-02-13 01:34:32 +00002068 TypePrinting *TypePrinter,
2069 SlotTracker *Machine,
2070 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002071 Out << "!DITemplateValueParameter(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002072 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
Duncan P. N. Exon Smith16d182a2015-02-28 23:21:38 +00002073 if (N->getTag() != dwarf::DW_TAG_template_value_parameter)
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002074 Printer.printTag(N);
2075 Printer.printString("name", N->getName());
Duncan P. N. Exon Smith3ec5fa62015-04-06 19:03:45 +00002076 Printer.printMetadata("type", N->getRawType());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002077 Printer.printMetadata("value", N->getValue(), /* ShouldSkipNull */ false);
Duncan P. N. Exon Smith2847f382015-02-13 01:34:32 +00002078 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002079}
Duncan P. N. Exon Smith2847f382015-02-13 01:34:32 +00002080
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002081static void writeDIGlobalVariable(raw_ostream &Out, const DIGlobalVariable *N,
Duncan P. N. Exon Smithc8f810a2015-02-13 01:35:40 +00002082 TypePrinting *TypePrinter,
2083 SlotTracker *Machine, const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002084 Out << "!DIGlobalVariable(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002085 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2086 Printer.printString("name", N->getName());
2087 Printer.printString("linkageName", N->getLinkageName());
Duncan P. N. Exon Smith3d2afaa2015-03-27 17:29:58 +00002088 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
2089 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002090 Printer.printInt("line", N->getLine());
Duncan P. N. Exon Smith3d2afaa2015-03-27 17:29:58 +00002091 Printer.printMetadata("type", N->getRawType());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002092 Printer.printBool("isLocal", N->isLocalToUnit());
2093 Printer.printBool("isDefinition", N->isDefinition());
Duncan P. N. Exon Smith3d2afaa2015-03-27 17:29:58 +00002094 Printer.printMetadata("declaration", N->getRawStaticDataMemberDeclaration());
Matthew Vossf8ab35a2018-10-03 18:44:53 +00002095 Printer.printMetadata("templateParams", N->getRawTemplateParams());
Victor Leschuk2ede1262016-10-20 00:13:12 +00002096 Printer.printInt("align", N->getAlignInBits());
Duncan P. N. Exon Smithc8f810a2015-02-13 01:35:40 +00002097 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002098}
Duncan P. N. Exon Smithc8f810a2015-02-13 01:35:40 +00002099
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002100static void writeDILocalVariable(raw_ostream &Out, const DILocalVariable *N,
Duncan P. N. Exon Smith72fe2d02015-02-13 01:39:44 +00002101 TypePrinting *TypePrinter,
2102 SlotTracker *Machine, const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002103 Out << "!DILocalVariable(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002104 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002105 Printer.printString("name", N->getName());
Duncan P. N. Exon Smithed013cd2015-07-31 18:58:39 +00002106 Printer.printInt("arg", N->getArg());
Duncan P. N. Exon Smith3d2afaa2015-03-27 17:29:58 +00002107 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
2108 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002109 Printer.printInt("line", N->getLine());
Duncan P. N. Exon Smith3d2afaa2015-03-27 17:29:58 +00002110 Printer.printMetadata("type", N->getRawType());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002111 Printer.printDIFlags("flags", N->getFlags());
Victor Leschuk2ede1262016-10-20 00:13:12 +00002112 Printer.printInt("align", N->getAlignInBits());
Duncan P. N. Exon Smith72fe2d02015-02-13 01:39:44 +00002113 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002114}
Duncan P. N. Exon Smith72fe2d02015-02-13 01:39:44 +00002115
Shiva Chen2c864552018-05-09 02:40:45 +00002116static void writeDILabel(raw_ostream &Out, const DILabel *N,
2117 TypePrinting *TypePrinter,
2118 SlotTracker *Machine, const Module *Context) {
2119 Out << "!DILabel(";
2120 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2121 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
2122 Printer.printString("name", N->getName());
2123 Printer.printMetadata("file", N->getRawFile());
2124 Printer.printInt("line", N->getLine());
2125 Out << ")";
2126}
2127
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002128static void writeDIExpression(raw_ostream &Out, const DIExpression *N,
Duncan P. N. Exon Smith0c5c0122015-02-13 01:42:09 +00002129 TypePrinting *TypePrinter, SlotTracker *Machine,
2130 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002131 Out << "!DIExpression(";
Duncan P. N. Exon Smith0c5c0122015-02-13 01:42:09 +00002132 FieldSeparator FS;
2133 if (N->isValid()) {
2134 for (auto I = N->expr_op_begin(), E = N->expr_op_end(); I != E; ++I) {
Mehdi Amini149f6ea2016-10-05 05:59:29 +00002135 auto OpStr = dwarf::OperationEncodingString(I->getOp());
2136 assert(!OpStr.empty() && "Expected valid opcode");
Duncan P. N. Exon Smith0c5c0122015-02-13 01:42:09 +00002137
2138 Out << FS << OpStr;
Markus Lavinb86ce212019-03-19 13:16:28 +00002139 if (I->getOp() == dwarf::DW_OP_LLVM_convert) {
2140 Out << FS << I->getArg(0);
2141 Out << FS << dwarf::AttributeEncodingString(I->getArg(1));
2142 } else {
2143 for (unsigned A = 0, AE = I->getNumArgs(); A != AE; ++A)
2144 Out << FS << I->getArg(A);
2145 }
Duncan P. N. Exon Smith0c5c0122015-02-13 01:42:09 +00002146 }
2147 } else {
2148 for (const auto &I : N->getElements())
2149 Out << FS << I;
2150 }
2151 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002152}
Duncan P. N. Exon Smith0c5c0122015-02-13 01:42:09 +00002153
Adrian Prantlbceaaa92016-12-20 02:09:43 +00002154static void writeDIGlobalVariableExpression(raw_ostream &Out,
2155 const DIGlobalVariableExpression *N,
2156 TypePrinting *TypePrinter,
2157 SlotTracker *Machine,
2158 const Module *Context) {
2159 Out << "!DIGlobalVariableExpression(";
2160 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2161 Printer.printMetadata("var", N->getVariable());
2162 Printer.printMetadata("expr", N->getExpression());
2163 Out << ")";
2164}
2165
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002166static void writeDIObjCProperty(raw_ostream &Out, const DIObjCProperty *N,
Duncan P. N. Exon Smithd45ce962015-02-13 01:43:22 +00002167 TypePrinting *TypePrinter, SlotTracker *Machine,
2168 const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002169 Out << "!DIObjCProperty(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002170 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2171 Printer.printString("name", N->getName());
Duncan P. N. Exon Smithf9b47752015-03-30 17:21:38 +00002172 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002173 Printer.printInt("line", N->getLine());
2174 Printer.printString("setter", N->getSetterName());
2175 Printer.printString("getter", N->getGetterName());
2176 Printer.printInt("attributes", N->getAttributes());
Duncan P. N. Exon Smithf9b47752015-03-30 17:21:38 +00002177 Printer.printMetadata("type", N->getRawType());
Duncan P. N. Exon Smithd45ce962015-02-13 01:43:22 +00002178 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002179}
Duncan P. N. Exon Smithd45ce962015-02-13 01:43:22 +00002180
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002181static void writeDIImportedEntity(raw_ostream &Out, const DIImportedEntity *N,
Duncan P. N. Exon Smith1c931162015-02-13 01:46:02 +00002182 TypePrinting *TypePrinter,
2183 SlotTracker *Machine, const Module *Context) {
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +00002184 Out << "!DIImportedEntity(";
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002185 MDFieldPrinter Printer(Out, TypePrinter, Machine, Context);
2186 Printer.printTag(N);
2187 Printer.printString("name", N->getName());
Duncan P. N. Exon Smithf9b47752015-03-30 17:21:38 +00002188 Printer.printMetadata("scope", N->getRawScope(), /* ShouldSkipNull */ false);
2189 Printer.printMetadata("entity", N->getRawEntity());
Adrian Prantld63bfd22017-07-19 00:09:54 +00002190 Printer.printMetadata("file", N->getRawFile());
Duncan P. N. Exon Smith6d267f02015-03-27 00:17:42 +00002191 Printer.printInt("line", N->getLine());
Duncan P. N. Exon Smith1c931162015-02-13 01:46:02 +00002192 Out << ")";
Duncan P. N. Exon Smith01fc1762015-02-10 00:52:32 +00002193}
2194
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00002195static void WriteMDNodeBodyInternal(raw_ostream &Out, const MDNode *Node,
2196 TypePrinting *TypePrinter,
2197 SlotTracker *Machine,
2198 const Module *Context) {
Duncan P. N. Exon Smith2bc00f42015-01-19 23:13:14 +00002199 if (Node->isDistinct())
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00002200 Out << "distinct ";
Duncan P. N. Exon Smith3d510662015-03-16 21:21:10 +00002201 else if (Node->isTemporary())
2202 Out << "<temporary!> "; // Handle broken code.
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00002203
Duncan P. N. Exon Smith2bc00f42015-01-19 23:13:14 +00002204 switch (Node->getMetadataID()) {
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00002205 default:
2206 llvm_unreachable("Expected uniquable MDNode");
Duncan P. N. Exon Smith2bc00f42015-01-19 23:13:14 +00002207#define HANDLE_MDNODE_LEAF(CLASS) \
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00002208 case Metadata::CLASS##Kind: \
Duncan P. N. Exon Smith2bc00f42015-01-19 23:13:14 +00002209 write##CLASS(Out, cast<CLASS>(Node), TypePrinter, Machine, Context); \
Duncan P. N. Exon Smitha6de6a42015-01-12 23:45:31 +00002210 break;
2211#include "llvm/IR/Metadata.def"
2212 }
2213}
2214
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00002215// Full implementation of printing a Value as an operand with support for
2216// TypePrinting, etc.
Dan Gohman12dad632009-08-12 20:56:03 +00002217static void WriteAsOperandInternal(raw_ostream &Out, const Value *V,
Dan Gohman8061d9e2009-08-13 15:27:57 +00002218 TypePrinting *TypePrinter,
Dan Gohmana2489d12010-07-20 23:55:01 +00002219 SlotTracker *Machine,
2220 const Module *Context) {
Chris Lattner033935d2008-08-17 04:40:13 +00002221 if (V->hasName()) {
2222 PrintLLVMName(Out, V);
2223 return;
2224 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002225
Chris Lattner033935d2008-08-17 04:40:13 +00002226 const Constant *CV = dyn_cast<Constant>(V);
2227 if (CV && !isa<GlobalValue>(CV)) {
Dan Gohman8061d9e2009-08-13 15:27:57 +00002228 assert(TypePrinter && "Constants require TypePrinting!");
Dan Gohmana2489d12010-07-20 23:55:01 +00002229 WriteConstantInternal(Out, CV, *TypePrinter, Machine, Context);
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002230 return;
2231 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002232
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002233 if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) {
Chris Lattner033935d2008-08-17 04:40:13 +00002234 Out << "asm ";
2235 if (IA->hasSideEffects())
2236 Out << "sideeffect ";
Dale Johannesen1cfb9582009-10-21 23:28:00 +00002237 if (IA->isAlignStack())
2238 Out << "alignstack ";
Chad Rosierd8c76102012-09-05 19:00:49 +00002239 // We don't emit the AD_ATT dialect as it's the assumed default.
2240 if (IA->getDialect() == InlineAsm::AD_Intel)
2241 Out << "inteldialect ";
Chris Lattner033935d2008-08-17 04:40:13 +00002242 Out << '"';
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002243 printEscapedString(IA->getAsmString(), Out);
Chris Lattner033935d2008-08-17 04:40:13 +00002244 Out << "\", \"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002245 printEscapedString(IA->getConstraintString(), Out);
Chris Lattner033935d2008-08-17 04:40:13 +00002246 Out << '"';
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002247 return;
2248 }
Devang Patel7428d8a2009-07-22 17:43:22 +00002249
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002250 if (auto *MD = dyn_cast<MetadataAsValue>(V)) {
2251 WriteAsOperandInternal(Out, MD->getMetadata(), TypePrinter, Machine,
2252 Context, /* FromValue */ true);
Devang Patel7428d8a2009-07-22 17:43:22 +00002253 return;
2254 }
2255
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002256 char Prefix = '%';
2257 int Slot;
Chris Lattner5b82a0a2011-08-03 06:15:41 +00002258 // If we have a SlotTracker, use it.
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002259 if (Machine) {
2260 if (const GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
2261 Slot = Machine->getGlobalSlot(GV);
2262 Prefix = '@';
2263 } else {
2264 Slot = Machine->getLocalSlot(V);
Andrew Trickec4b6e72011-09-30 19:48:58 +00002265
Chris Lattner5b82a0a2011-08-03 06:15:41 +00002266 // If the local value didn't succeed, then we may be referring to a value
2267 // from a different function. Translate it, as this can happen when using
2268 // address of blocks.
2269 if (Slot == -1)
2270 if ((Machine = createSlotTracker(V))) {
2271 Slot = Machine->getLocalSlot(V);
2272 delete Machine;
2273 }
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002274 }
Chris Lattner5b82a0a2011-08-03 06:15:41 +00002275 } else if ((Machine = createSlotTracker(V))) {
2276 // Otherwise, create one to get the # and then destroy it.
2277 if (const GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
2278 Slot = Machine->getGlobalSlot(GV);
2279 Prefix = '@';
Chris Lattnera2d810d2006-01-25 22:26:05 +00002280 } else {
Chris Lattner5b82a0a2011-08-03 06:15:41 +00002281 Slot = Machine->getLocalSlot(V);
Chris Lattnerd84bb632002-04-16 21:36:08 +00002282 }
Chris Lattner5b82a0a2011-08-03 06:15:41 +00002283 delete Machine;
Craig Topperc6207612014-04-09 06:08:46 +00002284 Machine = nullptr;
Chris Lattner5b82a0a2011-08-03 06:15:41 +00002285 } else {
2286 Slot = -1;
Chris Lattnerd84bb632002-04-16 21:36:08 +00002287 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002288
Chris Lattner3a36d2c2008-08-19 05:06:27 +00002289 if (Slot != -1)
2290 Out << Prefix << Slot;
2291 else
2292 Out << "<badref>";
Chris Lattnerd84bb632002-04-16 21:36:08 +00002293}
2294
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002295static void WriteAsOperandInternal(raw_ostream &Out, const Metadata *MD,
2296 TypePrinting *TypePrinter,
2297 SlotTracker *Machine, const Module *Context,
2298 bool FromValue) {
Reid Kleckner6d353342017-08-23 20:31:27 +00002299 // Write DIExpressions inline when used as a value. Improves readability of
2300 // debug info intrinsics.
2301 if (const DIExpression *Expr = dyn_cast<DIExpression>(MD)) {
2302 writeDIExpression(Out, Expr, TypePrinter, Machine, Context);
2303 return;
2304 }
2305
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002306 if (const MDNode *N = dyn_cast<MDNode>(MD)) {
Duncan P. N. Exon Smithf8b3ad62015-06-27 00:15:32 +00002307 std::unique_ptr<SlotTracker> MachineStorage;
2308 if (!Machine) {
2309 MachineStorage = make_unique<SlotTracker>(Context);
2310 Machine = MachineStorage.get();
2311 }
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002312 int Slot = Machine->getMetadataSlot(N);
Daniel Sanders9f3cf552018-12-13 14:25:27 +00002313 if (Slot == -1) {
2314 if (const DILocation *Loc = dyn_cast<DILocation>(N)) {
2315 writeDILocation(Out, Loc, TypePrinter, Machine, Context);
2316 return;
2317 }
Duncan P. N. Exon Smithfee167f2014-12-16 07:09:37 +00002318 // Give the pointer value instead of "badref", since this comes up all
2319 // the time when debugging.
2320 Out << "<" << N << ">";
Daniel Sanders9f3cf552018-12-13 14:25:27 +00002321 } else
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002322 Out << '!' << Slot;
2323 return;
2324 }
2325
2326 if (const MDString *MDS = dyn_cast<MDString>(MD)) {
2327 Out << "!\"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002328 printEscapedString(MDS->getString(), Out);
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002329 Out << '"';
2330 return;
2331 }
2332
2333 auto *V = cast<ValueAsMetadata>(MD);
2334 assert(TypePrinter && "TypePrinter required for metadata values");
2335 assert((FromValue || !isa<LocalAsMetadata>(V)) &&
2336 "Unexpected function-local metadata outside of value argument");
2337
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002338 TypePrinter->print(V->getValue()->getType(), Out);
2339 Out << ' ';
2340 WriteAsOperandInternal(Out, V->getValue(), TypePrinter, Machine, Context);
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00002341}
2342
Benjamin Kramerba05a782015-03-17 19:53:41 +00002343namespace {
Eugene Zelenkode6cce22017-06-19 22:05:08 +00002344
Benjamin Kramerba05a782015-03-17 19:53:41 +00002345class AssemblyWriter {
2346 formatted_raw_ostream &Out;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002347 const Module *TheModule = nullptr;
2348 const ModuleSummaryIndex *TheIndex = nullptr;
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00002349 std::unique_ptr<SlotTracker> SlotTrackerStorage;
Benjamin Kramerba05a782015-03-17 19:53:41 +00002350 SlotTracker &Machine;
2351 TypePrinting TypePrinter;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002352 AssemblyAnnotationWriter *AnnotationWriter = nullptr;
Benjamin Kramerba05a782015-03-17 19:53:41 +00002353 SetVector<const Comdat *> Comdats;
Justin Bognerd7d1a722015-09-27 22:38:50 +00002354 bool IsForDebug;
Duncan P. N. Exon Smith89010d82015-04-15 01:36:30 +00002355 bool ShouldPreserveUseListOrder;
Benjamin Kramerba05a782015-03-17 19:53:41 +00002356 UseListOrderStack UseListOrders;
Duncan P. N. Exon Smithe30f10e2015-04-24 21:03:05 +00002357 SmallVector<StringRef, 8> MDNames;
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002358 /// Synchronization scope names registered with LLVMContext.
2359 SmallVector<StringRef, 8> SSNs;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002360 DenseMap<const GlobalValueSummary *, GlobalValue::GUID> SummaryToGUIDMap;
Benjamin Kramerba05a782015-03-17 19:53:41 +00002361
2362public:
2363 /// Construct an AssemblyWriter with an external SlotTracker
Duncan P. N. Exon Smith89010d82015-04-15 01:36:30 +00002364 AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac, const Module *M,
Justin Bognerd7d1a722015-09-27 22:38:50 +00002365 AssemblyAnnotationWriter *AAW, bool IsForDebug,
Duncan P. N. Exon Smith89010d82015-04-15 01:36:30 +00002366 bool ShouldPreserveUseListOrder = false);
Benjamin Kramerba05a782015-03-17 19:53:41 +00002367
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002368 AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
2369 const ModuleSummaryIndex *Index, bool IsForDebug);
2370
Benjamin Kramerba05a782015-03-17 19:53:41 +00002371 void printMDNodeBody(const MDNode *MD);
2372 void printNamedMDNode(const NamedMDNode *NMD);
2373
2374 void printModule(const Module *M);
2375
2376 void writeOperand(const Value *Op, bool PrintType);
Reid Kleckner6652a522017-04-28 18:37:16 +00002377 void writeParamOperand(const Value *Operand, AttributeSet Attrs);
Chandler Carruth90c09232019-01-07 07:31:49 +00002378 void writeOperandBundles(const CallBase *Call);
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002379 void writeSyncScope(const LLVMContext &Context,
2380 SyncScope::ID SSID);
2381 void writeAtomic(const LLVMContext &Context,
2382 AtomicOrdering Ordering,
2383 SyncScope::ID SSID);
2384 void writeAtomicCmpXchg(const LLVMContext &Context,
2385 AtomicOrdering SuccessOrdering,
Benjamin Kramerba05a782015-03-17 19:53:41 +00002386 AtomicOrdering FailureOrdering,
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002387 SyncScope::ID SSID);
Benjamin Kramerba05a782015-03-17 19:53:41 +00002388
2389 void writeAllMDNodes();
2390 void writeMDNode(unsigned Slot, const MDNode *Node);
2391 void writeAllAttributeGroups();
2392
2393 void printTypeIdentities();
2394 void printGlobal(const GlobalVariable *GV);
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00002395 void printIndirectSymbol(const GlobalIndirectSymbol *GIS);
Benjamin Kramerba05a782015-03-17 19:53:41 +00002396 void printComdat(const Comdat *C);
2397 void printFunction(const Function *F);
Reid Kleckner6652a522017-04-28 18:37:16 +00002398 void printArgument(const Argument *FA, AttributeSet Attrs);
Benjamin Kramerba05a782015-03-17 19:53:41 +00002399 void printBasicBlock(const BasicBlock *BB);
2400 void printInstructionLine(const Instruction &I);
2401 void printInstruction(const Instruction &I);
2402
2403 void printUseListOrder(const UseListOrder &Order);
2404 void printUseLists(const Function *F);
2405
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002406 void printModuleSummaryIndex();
2407 void printSummaryInfo(unsigned Slot, const ValueInfo &VI);
2408 void printSummary(const GlobalValueSummary &Summary);
2409 void printAliasSummary(const AliasSummary *AS);
2410 void printGlobalVarSummary(const GlobalVarSummary *GS);
2411 void printFunctionSummary(const FunctionSummary *FS);
2412 void printTypeIdSummary(const TypeIdSummary &TIS);
2413 void printTypeTestResolution(const TypeTestResolution &TTRes);
2414 void printArgs(const std::vector<uint64_t> &Args);
2415 void printWPDRes(const WholeProgramDevirtResolution &WPDRes);
2416 void printTypeIdInfo(const FunctionSummary::TypeIdInfo &TIDInfo);
2417 void printVFuncId(const FunctionSummary::VFuncId VFId);
2418 void
2419 printNonConstVCalls(const std::vector<FunctionSummary::VFuncId> VCallList,
2420 const char *Tag);
2421 void
2422 printConstVCalls(const std::vector<FunctionSummary::ConstVCall> VCallList,
2423 const char *Tag);
2424
Benjamin Kramerba05a782015-03-17 19:53:41 +00002425private:
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00002426 /// Print out metadata attachments.
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00002427 void printMetadataAttachments(
Duncan P. N. Exon Smithd3e4c2a2015-04-24 21:06:21 +00002428 const SmallVectorImpl<std::pair<unsigned, MDNode *>> &MDs,
2429 StringRef Separator);
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00002430
Benjamin Kramerba05a782015-03-17 19:53:41 +00002431 // printInfoComment - Print a little comment after the instruction indicating
2432 // which slot it occupies.
2433 void printInfoComment(const Value &V);
Igor Laevsky2aa8caf2015-05-05 13:20:42 +00002434
2435 // printGCRelocateComment - print comment after call to the gc.relocate
2436 // intrinsic indicating base and derived pointer names.
Manuel Jacob83eefa62016-01-05 04:03:00 +00002437 void printGCRelocateComment(const GCRelocateInst &Relocate);
Benjamin Kramerba05a782015-03-17 19:53:41 +00002438};
Eugene Zelenkode6cce22017-06-19 22:05:08 +00002439
2440} // end anonymous namespace
Benjamin Kramerba05a782015-03-17 19:53:41 +00002441
Justin Bogner0ffea9d2015-09-02 22:46:15 +00002442AssemblyWriter::AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
2443 const Module *M, AssemblyAnnotationWriter *AAW,
Justin Bognerd7d1a722015-09-27 22:38:50 +00002444 bool IsForDebug, bool ShouldPreserveUseListOrder)
Roman Tereshind96de6f2018-03-22 21:29:07 +00002445 : Out(o), TheModule(M), Machine(Mac), TypePrinter(M), AnnotationWriter(AAW),
Justin Bognerd7d1a722015-09-27 22:38:50 +00002446 IsForDebug(IsForDebug),
Justin Bogner0ffea9d2015-09-02 22:46:15 +00002447 ShouldPreserveUseListOrder(ShouldPreserveUseListOrder) {
David Majnemerdad0a642014-06-27 18:19:56 +00002448 if (!TheModule)
2449 return;
Peter Collingbourne6d88fde2016-06-22 20:29:42 +00002450 for (const GlobalObject &GO : TheModule->global_objects())
2451 if (const Comdat *C = GO.getComdat())
David Majnemerdad0a642014-06-27 18:19:56 +00002452 Comdats.insert(C);
Daniel Maleaded9f932013-05-08 20:38:31 +00002453}
Chris Lattner2e9fee42001-07-12 23:35:26 +00002454
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002455AssemblyWriter::AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
2456 const ModuleSummaryIndex *Index, bool IsForDebug)
2457 : Out(o), TheIndex(Index), Machine(Mac), TypePrinter(/*Module=*/nullptr),
2458 IsForDebug(IsForDebug), ShouldPreserveUseListOrder(false) {}
2459
Chris Lattner78e2e8b2006-12-06 06:24:27 +00002460void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType) {
Craig Topperc6207612014-04-09 06:08:46 +00002461 if (!Operand) {
Chris Lattner08f7d0c2005-02-24 16:58:29 +00002462 Out << "<null operand!>";
Chris Lattnerb419a1e2009-12-31 02:33:14 +00002463 return;
Chris Lattner08f7d0c2005-02-24 16:58:29 +00002464 }
Chris Lattnerb419a1e2009-12-31 02:33:14 +00002465 if (PrintType) {
2466 TypePrinter.print(Operand->getType(), Out);
2467 Out << ' ';
2468 }
Dan Gohmana2489d12010-07-20 23:55:01 +00002469 WriteAsOperandInternal(Out, Operand, &TypePrinter, &Machine, TheModule);
Chris Lattner2f7c9632001-06-06 20:29:01 +00002470}
2471
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002472void AssemblyWriter::writeSyncScope(const LLVMContext &Context,
2473 SyncScope::ID SSID) {
2474 switch (SSID) {
2475 case SyncScope::System: {
2476 break;
2477 }
2478 default: {
2479 if (SSNs.empty())
2480 Context.getSyncScopeNames(SSNs);
2481
2482 Out << " syncscope(\"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002483 printEscapedString(SSNs[SSID], Out);
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002484 Out << "\")";
2485 break;
2486 }
2487 }
2488}
2489
2490void AssemblyWriter::writeAtomic(const LLVMContext &Context,
2491 AtomicOrdering Ordering,
2492 SyncScope::ID SSID) {
JF Bastien800f87a2016-04-06 21:19:33 +00002493 if (Ordering == AtomicOrdering::NotAtomic)
Eli Friedmanfee02c62011-07-25 23:16:38 +00002494 return;
2495
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002496 writeSyncScope(Context, SSID);
JF Bastien800f87a2016-04-06 21:19:33 +00002497 Out << " " << toIRString(Ordering);
Eli Friedmanfee02c62011-07-25 23:16:38 +00002498}
2499
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002500void AssemblyWriter::writeAtomicCmpXchg(const LLVMContext &Context,
2501 AtomicOrdering SuccessOrdering,
Tim Northovere94a5182014-03-11 10:48:52 +00002502 AtomicOrdering FailureOrdering,
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002503 SyncScope::ID SSID) {
JF Bastien800f87a2016-04-06 21:19:33 +00002504 assert(SuccessOrdering != AtomicOrdering::NotAtomic &&
2505 FailureOrdering != AtomicOrdering::NotAtomic);
Tim Northovere94a5182014-03-11 10:48:52 +00002506
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00002507 writeSyncScope(Context, SSID);
JF Bastien800f87a2016-04-06 21:19:33 +00002508 Out << " " << toIRString(SuccessOrdering);
2509 Out << " " << toIRString(FailureOrdering);
Tim Northovere94a5182014-03-11 10:48:52 +00002510}
2511
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002512void AssemblyWriter::writeParamOperand(const Value *Operand,
Reid Kleckner6652a522017-04-28 18:37:16 +00002513 AttributeSet Attrs) {
Craig Topperc6207612014-04-09 06:08:46 +00002514 if (!Operand) {
Duncan Sandsad0ea2d2007-11-27 13:23:08 +00002515 Out << "<null operand!>";
Chris Lattnerb419a1e2009-12-31 02:33:14 +00002516 return;
Duncan Sandsad0ea2d2007-11-27 13:23:08 +00002517 }
Chris Lattnerb419a1e2009-12-31 02:33:14 +00002518
2519 // Print the type
2520 TypePrinter.print(Operand->getType(), Out);
2521 // Print parameter attributes list
Reid Kleckner6652a522017-04-28 18:37:16 +00002522 if (Attrs.hasAttributes())
2523 Out << ' ' << Attrs.getAsString();
Chris Lattnerb419a1e2009-12-31 02:33:14 +00002524 Out << ' ';
2525 // Print the operand
Dan Gohmana2489d12010-07-20 23:55:01 +00002526 WriteAsOperandInternal(Out, Operand, &TypePrinter, &Machine, TheModule);
Duncan Sandsad0ea2d2007-11-27 13:23:08 +00002527}
Chris Lattner2f7c9632001-06-06 20:29:01 +00002528
Chandler Carruth90c09232019-01-07 07:31:49 +00002529void AssemblyWriter::writeOperandBundles(const CallBase *Call) {
2530 if (!Call->hasOperandBundles())
Sanjoy Dasb513a9f2015-09-24 23:34:52 +00002531 return;
2532
2533 Out << " [ ";
2534
2535 bool FirstBundle = true;
Chandler Carruth90c09232019-01-07 07:31:49 +00002536 for (unsigned i = 0, e = Call->getNumOperandBundles(); i != e; ++i) {
2537 OperandBundleUse BU = Call->getOperandBundleAt(i);
Sanjoy Dasb513a9f2015-09-24 23:34:52 +00002538
2539 if (!FirstBundle)
2540 Out << ", ";
2541 FirstBundle = false;
2542
2543 Out << '"';
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002544 printEscapedString(BU.getTagName(), Out);
Sanjoy Dasb513a9f2015-09-24 23:34:52 +00002545 Out << '"';
2546
2547 Out << '(';
2548
2549 bool FirstInput = true;
2550 for (const auto &Input : BU.Inputs) {
2551 if (!FirstInput)
2552 Out << ", ";
2553 FirstInput = false;
2554
2555 TypePrinter.print(Input->getType(), Out);
2556 Out << " ";
2557 WriteAsOperandInternal(Out, Input, &TypePrinter, &Machine, TheModule);
2558 }
2559
2560 Out << ')';
2561 }
2562
2563 Out << " ]";
2564}
2565
Chris Lattner7bfee412001-10-29 16:05:51 +00002566void AssemblyWriter::printModule(const Module *M) {
Teresa Johnsonfc801f52018-07-03 15:52:57 +00002567 Machine.initializeIfNeeded();
Bill Wendling829b4782013-02-11 08:43:33 +00002568
Duncan P. N. Exon Smith89010d82015-04-15 01:36:30 +00002569 if (ShouldPreserveUseListOrder)
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +00002570 UseListOrders = predictUseListOrder(M);
2571
Chris Lattner4d8689e2005-03-02 23:12:40 +00002572 if (!M->getModuleIdentifier().empty() &&
Misha Brukmanb1c93172005-04-21 23:48:37 +00002573 // Don't print the ID if it will start a new line (which would
Chris Lattner4d8689e2005-03-02 23:12:40 +00002574 // require a comment char before it).
2575 M->getModuleIdentifier().find('\n') == std::string::npos)
2576 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
2577
Teresa Johnson83c517c2016-03-30 18:15:08 +00002578 if (!M->getSourceFileName().empty()) {
Teresa Johnsond8d94652016-03-30 22:17:28 +00002579 Out << "source_filename = \"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002580 printEscapedString(M->getSourceFileName(), Out);
Teresa Johnsond8d94652016-03-30 22:17:28 +00002581 Out << "\"\n";
Teresa Johnson83c517c2016-03-30 18:15:08 +00002582 }
2583
Rafael Espindolaf863ee22014-02-25 20:01:08 +00002584 const std::string &DL = M->getDataLayoutStr();
2585 if (!DL.empty())
2586 Out << "target datalayout = \"" << DL << "\"\n";
Reid Spencer48f98c82004-07-25 21:44:54 +00002587 if (!M->getTargetTriple().empty())
Reid Spencerffec7df2004-07-25 21:29:43 +00002588 Out << "target triple = \"" << M->getTargetTriple() << "\"\n";
Misha Brukmanb1c93172005-04-21 23:48:37 +00002589
Chris Lattnereef2fe72006-01-24 04:13:11 +00002590 if (!M->getModuleInlineAsm().empty()) {
Dan Gohman69273e62009-08-12 23:54:22 +00002591 Out << '\n';
Benjamin Kramerbbd05a22015-06-07 13:59:33 +00002592
2593 // Split the string into lines, to make it easier to read the .ll file.
Benjamin Kramerf1cfc422015-06-08 18:58:57 +00002594 StringRef Asm = M->getModuleInlineAsm();
Benjamin Kramerbbd05a22015-06-07 13:59:33 +00002595 do {
2596 StringRef Front;
2597 std::tie(Front, Asm) = Asm.split('\n');
2598
Chris Lattnerefaf35d2006-01-24 00:45:30 +00002599 // We found a newline, print the portion of the asm string from the
2600 // last newline up to this newline.
2601 Out << "module asm \"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00002602 printEscapedString(Front, Out);
Chris Lattnerefaf35d2006-01-24 00:45:30 +00002603 Out << "\"\n";
Benjamin Kramerbbd05a22015-06-07 13:59:33 +00002604 } while (!Asm.empty());
Chris Lattner6ed87bd2006-01-23 23:03:36 +00002605 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002606
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00002607 printTypeIdentities();
Misha Brukmanb1c93172005-04-21 23:48:37 +00002608
David Majnemerdad0a642014-06-27 18:19:56 +00002609 // Output all comdats.
2610 if (!Comdats.empty())
2611 Out << '\n';
2612 for (const Comdat *C : Comdats) {
2613 printComdat(C);
2614 if (C != Comdats.back())
2615 Out << '\n';
2616 }
2617
Dan Gohman69273e62009-08-12 23:54:22 +00002618 // Output all globals.
2619 if (!M->global_empty()) Out << '\n';
Yaron Kerenf3cf9d12015-06-13 17:50:47 +00002620 for (const GlobalVariable &GV : M->globals()) {
2621 printGlobal(&GV); Out << '\n';
Duncan Sands66fc0e62012-09-12 09:55:51 +00002622 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002623
Chris Lattnerd2747052007-04-26 02:24:10 +00002624 // Output all aliases.
2625 if (!M->alias_empty()) Out << "\n";
Yaron Kerenf3cf9d12015-06-13 17:50:47 +00002626 for (const GlobalAlias &GA : M->aliases())
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00002627 printIndirectSymbol(&GA);
Chris Lattnerfee714f2001-09-07 16:36:04 +00002628
Dmitry Polukhina1feff72016-04-07 12:32:19 +00002629 // Output all ifuncs.
2630 if (!M->ifunc_empty()) Out << "\n";
2631 for (const GlobalIFunc &GI : M->ifuncs())
2632 printIndirectSymbol(&GI);
2633
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +00002634 // Output global use-lists.
2635 printUseLists(nullptr);
2636
Chris Lattner2cdd49d2004-09-14 05:06:58 +00002637 // Output all of the functions.
Yaron Kerenf3cf9d12015-06-13 17:50:47 +00002638 for (const Function &F : *M)
2639 printFunction(&F);
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +00002640 assert(UseListOrders.empty() && "All use-lists should have been consumed");
Devang Patel983c6b12009-07-08 21:44:25 +00002641
Bill Wendling829b4782013-02-11 08:43:33 +00002642 // Output all attribute groups.
Bill Wendling04945972013-04-29 23:48:06 +00002643 if (!Machine.as_empty()) {
Bill Wendling829b4782013-02-11 08:43:33 +00002644 Out << '\n';
2645 writeAllAttributeGroups();
2646 }
2647
Devang Patel23e68302009-07-29 22:04:47 +00002648 // Output named metadata.
Dan Gohman69273e62009-08-12 23:54:22 +00002649 if (!M->named_metadata_empty()) Out << '\n';
Andrew Trickec4b6e72011-09-30 19:48:58 +00002650
Yaron Kerenf3cf9d12015-06-13 17:50:47 +00002651 for (const NamedMDNode &Node : M->named_metadata())
2652 printNamedMDNode(&Node);
Devang Patel23e68302009-07-29 22:04:47 +00002653
2654 // Output metadata.
Chris Lattnercf4a76e2009-12-31 02:20:11 +00002655 if (!Machine.mdn_empty()) {
Chris Lattnerbddea6a2009-12-31 02:13:35 +00002656 Out << '\n';
2657 writeAllMDNodes();
2658 }
Chris Lattnerfee714f2001-09-07 16:36:04 +00002659}
2660
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002661void AssemblyWriter::printModuleSummaryIndex() {
2662 assert(TheIndex);
Teresa Johnsonfc801f52018-07-03 15:52:57 +00002663 Machine.initializeIndexIfNeeded();
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002664
2665 Out << "\n";
2666
Teresa Johnson8fc76662018-07-02 22:09:23 +00002667 // Print module path entries. To print in order, add paths to a vector
2668 // indexed by module slot.
Teresa Johnson724e7a12018-05-26 02:53:52 +00002669 std::vector<std::pair<std::string, ModuleHash>> moduleVec;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002670 std::string RegularLTOModuleName = "[Regular LTO]";
2671 moduleVec.resize(TheIndex->modulePaths().size());
Teresa Johnson8fc76662018-07-02 22:09:23 +00002672 for (auto &ModPath : TheIndex->modulePaths())
2673 moduleVec[Machine.getModulePathSlot(ModPath.first())] = std::make_pair(
2674 // A module id of -1 is a special entry for a regular LTO module created
2675 // during the thin link.
2676 ModPath.second.first == -1u ? RegularLTOModuleName
2677 : (std::string)ModPath.first(),
2678 ModPath.second.second);
2679
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002680 unsigned i = 0;
2681 for (auto &ModPair : moduleVec) {
2682 Out << "^" << i++ << " = module: (";
Andrew Ngac305e12018-07-12 14:40:21 +00002683 Out << "path: \"";
2684 printEscapedString(ModPair.first, Out);
2685 Out << "\", hash: (";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002686 FieldSeparator FS;
2687 for (auto Hash : ModPair.second)
2688 Out << FS << Hash;
2689 Out << "))\n";
2690 }
2691
2692 // FIXME: Change AliasSummary to hold a ValueInfo instead of summary pointer
2693 // for aliasee (then update BitcodeWriter.cpp and remove get/setAliaseeGUID).
2694 for (auto &GlobalList : *TheIndex) {
2695 auto GUID = GlobalList.first;
2696 for (auto &Summary : GlobalList.second.SummaryList)
2697 SummaryToGUIDMap[Summary.get()] = GUID;
2698 }
2699
2700 // Print the global value summary entries.
2701 for (auto &GlobalList : *TheIndex) {
2702 auto GUID = GlobalList.first;
2703 auto VI = TheIndex->getValueInfo(GlobalList);
2704 printSummaryInfo(Machine.getGUIDSlot(GUID), VI);
2705 }
2706
2707 // Print the TypeIdMap entries.
Teresa Johnson7fb39df2018-09-25 20:14:40 +00002708 for (auto TidIter = TheIndex->typeIds().begin();
2709 TidIter != TheIndex->typeIds().end(); TidIter++) {
2710 Out << "^" << Machine.getTypeIdSlot(TidIter->second.first)
2711 << " = typeid: (name: \"" << TidIter->second.first << "\"";
2712 printTypeIdSummary(TidIter->second.second);
2713 Out << ") ; guid = " << TidIter->first << "\n";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002714 }
2715}
2716
2717static const char *
2718getWholeProgDevirtResKindName(WholeProgramDevirtResolution::Kind K) {
2719 switch (K) {
2720 case WholeProgramDevirtResolution::Indir:
2721 return "indir";
2722 case WholeProgramDevirtResolution::SingleImpl:
2723 return "singleImpl";
2724 case WholeProgramDevirtResolution::BranchFunnel:
2725 return "branchFunnel";
2726 }
2727 llvm_unreachable("invalid WholeProgramDevirtResolution kind");
2728}
2729
2730static const char *getWholeProgDevirtResByArgKindName(
2731 WholeProgramDevirtResolution::ByArg::Kind K) {
2732 switch (K) {
2733 case WholeProgramDevirtResolution::ByArg::Indir:
2734 return "indir";
2735 case WholeProgramDevirtResolution::ByArg::UniformRetVal:
2736 return "uniformRetVal";
2737 case WholeProgramDevirtResolution::ByArg::UniqueRetVal:
2738 return "uniqueRetVal";
2739 case WholeProgramDevirtResolution::ByArg::VirtualConstProp:
2740 return "virtualConstProp";
2741 }
2742 llvm_unreachable("invalid WholeProgramDevirtResolution::ByArg kind");
2743}
2744
2745static const char *getTTResKindName(TypeTestResolution::Kind K) {
2746 switch (K) {
2747 case TypeTestResolution::Unsat:
2748 return "unsat";
2749 case TypeTestResolution::ByteArray:
2750 return "byteArray";
2751 case TypeTestResolution::Inline:
2752 return "inline";
2753 case TypeTestResolution::Single:
2754 return "single";
2755 case TypeTestResolution::AllOnes:
2756 return "allOnes";
2757 }
2758 llvm_unreachable("invalid TypeTestResolution kind");
2759}
2760
2761void AssemblyWriter::printTypeTestResolution(const TypeTestResolution &TTRes) {
2762 Out << "typeTestRes: (kind: " << getTTResKindName(TTRes.TheKind)
2763 << ", sizeM1BitWidth: " << TTRes.SizeM1BitWidth;
2764
2765 // The following fields are only used if the target does not support the use
2766 // of absolute symbols to store constants. Print only if non-zero.
2767 if (TTRes.AlignLog2)
2768 Out << ", alignLog2: " << TTRes.AlignLog2;
2769 if (TTRes.SizeM1)
2770 Out << ", sizeM1: " << TTRes.SizeM1;
2771 if (TTRes.BitMask)
2772 // BitMask is uint8_t which causes it to print the corresponding char.
2773 Out << ", bitMask: " << (unsigned)TTRes.BitMask;
2774 if (TTRes.InlineBits)
2775 Out << ", inlineBits: " << TTRes.InlineBits;
2776
2777 Out << ")";
2778}
2779
2780void AssemblyWriter::printTypeIdSummary(const TypeIdSummary &TIS) {
2781 Out << ", summary: (";
2782 printTypeTestResolution(TIS.TTRes);
2783 if (!TIS.WPDRes.empty()) {
2784 Out << ", wpdResolutions: (";
2785 FieldSeparator FS;
2786 for (auto &WPDRes : TIS.WPDRes) {
2787 Out << FS;
2788 Out << "(offset: " << WPDRes.first << ", ";
2789 printWPDRes(WPDRes.second);
2790 Out << ")";
2791 }
2792 Out << ")";
2793 }
2794 Out << ")";
2795}
2796
2797void AssemblyWriter::printArgs(const std::vector<uint64_t> &Args) {
2798 Out << "args: (";
2799 FieldSeparator FS;
2800 for (auto arg : Args) {
2801 Out << FS;
2802 Out << arg;
2803 }
2804 Out << ")";
2805}
2806
2807void AssemblyWriter::printWPDRes(const WholeProgramDevirtResolution &WPDRes) {
2808 Out << "wpdRes: (kind: ";
2809 Out << getWholeProgDevirtResKindName(WPDRes.TheKind);
2810
2811 if (WPDRes.TheKind == WholeProgramDevirtResolution::SingleImpl)
2812 Out << ", singleImplName: \"" << WPDRes.SingleImplName << "\"";
2813
2814 if (!WPDRes.ResByArg.empty()) {
2815 Out << ", resByArg: (";
2816 FieldSeparator FS;
2817 for (auto &ResByArg : WPDRes.ResByArg) {
2818 Out << FS;
2819 printArgs(ResByArg.first);
2820 Out << ", byArg: (kind: ";
2821 Out << getWholeProgDevirtResByArgKindName(ResByArg.second.TheKind);
2822 if (ResByArg.second.TheKind ==
2823 WholeProgramDevirtResolution::ByArg::UniformRetVal ||
2824 ResByArg.second.TheKind ==
2825 WholeProgramDevirtResolution::ByArg::UniqueRetVal)
2826 Out << ", info: " << ResByArg.second.Info;
2827
2828 // The following fields are only used if the target does not support the
2829 // use of absolute symbols to store constants. Print only if non-zero.
2830 if (ResByArg.second.Byte || ResByArg.second.Bit)
2831 Out << ", byte: " << ResByArg.second.Byte
2832 << ", bit: " << ResByArg.second.Bit;
2833
2834 Out << ")";
2835 }
2836 Out << ")";
2837 }
2838 Out << ")";
2839}
2840
2841static const char *getSummaryKindName(GlobalValueSummary::SummaryKind SK) {
2842 switch (SK) {
2843 case GlobalValueSummary::AliasKind:
2844 return "alias";
2845 case GlobalValueSummary::FunctionKind:
2846 return "function";
2847 case GlobalValueSummary::GlobalVarKind:
2848 return "variable";
2849 }
2850 llvm_unreachable("invalid summary kind");
2851}
2852
2853void AssemblyWriter::printAliasSummary(const AliasSummary *AS) {
Teresa Johnsonf8182f12018-07-03 01:11:43 +00002854 Out << ", aliasee: ";
2855 // The indexes emitted for distributed backends may not include the
2856 // aliasee summary (only if it is being imported directly). Handle
2857 // that case by just emitting "null" as the aliasee.
2858 if (AS->hasAliasee())
2859 Out << "^" << Machine.getGUIDSlot(SummaryToGUIDMap[&AS->getAliasee()]);
2860 else
2861 Out << "null";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002862}
2863
2864void AssemblyWriter::printGlobalVarSummary(const GlobalVarSummary *GS) {
Eugene Leviant009d8332018-11-23 10:54:51 +00002865 Out << ", varFlags: (readonly: " << GS->VarFlags.ReadOnly << ")";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002866}
2867
Teresa Johnsonc80c8732018-05-25 15:15:39 +00002868static std::string getLinkageName(GlobalValue::LinkageTypes LT) {
2869 switch (LT) {
2870 case GlobalValue::ExternalLinkage:
2871 return "external";
2872 case GlobalValue::PrivateLinkage:
2873 return "private";
2874 case GlobalValue::InternalLinkage:
2875 return "internal";
2876 case GlobalValue::LinkOnceAnyLinkage:
2877 return "linkonce";
2878 case GlobalValue::LinkOnceODRLinkage:
2879 return "linkonce_odr";
2880 case GlobalValue::WeakAnyLinkage:
2881 return "weak";
2882 case GlobalValue::WeakODRLinkage:
2883 return "weak_odr";
2884 case GlobalValue::CommonLinkage:
2885 return "common";
2886 case GlobalValue::AppendingLinkage:
2887 return "appending";
2888 case GlobalValue::ExternalWeakLinkage:
2889 return "extern_weak";
2890 case GlobalValue::AvailableExternallyLinkage:
2891 return "available_externally";
2892 }
2893 llvm_unreachable("invalid linkage");
2894}
2895
2896// When printing the linkage types in IR where the ExternalLinkage is
2897// not printed, and other linkage types are expected to be printed with
2898// a space after the name.
2899static std::string getLinkageNameWithSpace(GlobalValue::LinkageTypes LT) {
2900 if (LT == GlobalValue::ExternalLinkage)
2901 return "";
2902 return getLinkageName(LT) + " ";
2903}
2904
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002905void AssemblyWriter::printFunctionSummary(const FunctionSummary *FS) {
2906 Out << ", insts: " << FS->instCount();
2907
2908 FunctionSummary::FFlags FFlags = FS->fflags();
2909 if (FFlags.ReadNone | FFlags.ReadOnly | FFlags.NoRecurse |
2910 FFlags.ReturnDoesNotAlias) {
2911 Out << ", funcFlags: (";
2912 Out << "readNone: " << FFlags.ReadNone;
2913 Out << ", readOnly: " << FFlags.ReadOnly;
2914 Out << ", noRecurse: " << FFlags.NoRecurse;
2915 Out << ", returnDoesNotAlias: " << FFlags.ReturnDoesNotAlias;
Teresa Johnsoncb397462018-11-06 19:41:35 +00002916 Out << ", noInline: " << FFlags.NoInline;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002917 Out << ")";
2918 }
2919 if (!FS->calls().empty()) {
2920 Out << ", calls: (";
2921 FieldSeparator IFS;
2922 for (auto &Call : FS->calls()) {
2923 Out << IFS;
2924 Out << "(callee: ^" << Machine.getGUIDSlot(Call.first.getGUID());
2925 if (Call.second.getHotness() != CalleeInfo::HotnessType::Unknown)
2926 Out << ", hotness: " << getHotnessName(Call.second.getHotness());
2927 else if (Call.second.RelBlockFreq)
2928 Out << ", relbf: " << Call.second.RelBlockFreq;
2929 Out << ")";
2930 }
2931 Out << ")";
2932 }
2933
2934 if (const auto *TIdInfo = FS->getTypeIdInfo())
2935 printTypeIdInfo(*TIdInfo);
2936}
2937
2938void AssemblyWriter::printTypeIdInfo(
2939 const FunctionSummary::TypeIdInfo &TIDInfo) {
2940 Out << ", typeIdInfo: (";
2941 FieldSeparator TIDFS;
2942 if (!TIDInfo.TypeTests.empty()) {
2943 Out << TIDFS;
2944 Out << "typeTests: (";
2945 FieldSeparator FS;
2946 for (auto &GUID : TIDInfo.TypeTests) {
Teresa Johnson7fb39df2018-09-25 20:14:40 +00002947 auto TidIter = TheIndex->typeIds().equal_range(GUID);
2948 if (TidIter.first == TidIter.second) {
2949 Out << FS;
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002950 Out << GUID;
Teresa Johnson7fb39df2018-09-25 20:14:40 +00002951 continue;
2952 }
2953 // Print all type id that correspond to this GUID.
2954 for (auto It = TidIter.first; It != TidIter.second; ++It) {
2955 Out << FS;
2956 auto Slot = Machine.getTypeIdSlot(It->second.first);
2957 assert(Slot != -1);
2958 Out << "^" << Slot;
2959 }
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002960 }
2961 Out << ")";
2962 }
2963 if (!TIDInfo.TypeTestAssumeVCalls.empty()) {
2964 Out << TIDFS;
2965 printNonConstVCalls(TIDInfo.TypeTestAssumeVCalls, "typeTestAssumeVCalls");
2966 }
2967 if (!TIDInfo.TypeCheckedLoadVCalls.empty()) {
2968 Out << TIDFS;
2969 printNonConstVCalls(TIDInfo.TypeCheckedLoadVCalls, "typeCheckedLoadVCalls");
2970 }
2971 if (!TIDInfo.TypeTestAssumeConstVCalls.empty()) {
2972 Out << TIDFS;
2973 printConstVCalls(TIDInfo.TypeTestAssumeConstVCalls,
2974 "typeTestAssumeConstVCalls");
2975 }
2976 if (!TIDInfo.TypeCheckedLoadConstVCalls.empty()) {
2977 Out << TIDFS;
2978 printConstVCalls(TIDInfo.TypeCheckedLoadConstVCalls,
2979 "typeCheckedLoadConstVCalls");
2980 }
2981 Out << ")";
2982}
2983
2984void AssemblyWriter::printVFuncId(const FunctionSummary::VFuncId VFId) {
Teresa Johnson7fb39df2018-09-25 20:14:40 +00002985 auto TidIter = TheIndex->typeIds().equal_range(VFId.GUID);
2986 if (TidIter.first == TidIter.second) {
2987 Out << "vFuncId: (";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00002988 Out << "guid: " << VFId.GUID;
Teresa Johnson7fb39df2018-09-25 20:14:40 +00002989 Out << ", offset: " << VFId.Offset;
2990 Out << ")";
2991 return;
2992 }
2993 // Print all type id that correspond to this GUID.
2994 FieldSeparator FS;
2995 for (auto It = TidIter.first; It != TidIter.second; ++It) {
2996 Out << FS;
2997 Out << "vFuncId: (";
2998 auto Slot = Machine.getTypeIdSlot(It->second.first);
2999 assert(Slot != -1);
3000 Out << "^" << Slot;
3001 Out << ", offset: " << VFId.Offset;
3002 Out << ")";
3003 }
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00003004}
3005
3006void AssemblyWriter::printNonConstVCalls(
3007 const std::vector<FunctionSummary::VFuncId> VCallList, const char *Tag) {
3008 Out << Tag << ": (";
3009 FieldSeparator FS;
3010 for (auto &VFuncId : VCallList) {
3011 Out << FS;
3012 printVFuncId(VFuncId);
3013 }
3014 Out << ")";
3015}
3016
3017void AssemblyWriter::printConstVCalls(
3018 const std::vector<FunctionSummary::ConstVCall> VCallList, const char *Tag) {
3019 Out << Tag << ": (";
3020 FieldSeparator FS;
3021 for (auto &ConstVCall : VCallList) {
3022 Out << FS;
Teresa Johnsonc7816802018-08-14 01:49:33 +00003023 Out << "(";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00003024 printVFuncId(ConstVCall.VFunc);
3025 if (!ConstVCall.Args.empty()) {
3026 Out << ", ";
3027 printArgs(ConstVCall.Args);
3028 }
Teresa Johnsonc7816802018-08-14 01:49:33 +00003029 Out << ")";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00003030 }
3031 Out << ")";
3032}
3033
3034void AssemblyWriter::printSummary(const GlobalValueSummary &Summary) {
3035 GlobalValueSummary::GVFlags GVFlags = Summary.flags();
3036 GlobalValue::LinkageTypes LT = (GlobalValue::LinkageTypes)GVFlags.Linkage;
3037 Out << getSummaryKindName(Summary.getSummaryKind()) << ": ";
Teresa Johnson8fc76662018-07-02 22:09:23 +00003038 Out << "(module: ^" << Machine.getModulePathSlot(Summary.modulePath())
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00003039 << ", flags: (";
3040 Out << "linkage: " << getLinkageName(LT);
3041 Out << ", notEligibleToImport: " << GVFlags.NotEligibleToImport;
3042 Out << ", live: " << GVFlags.Live;
3043 Out << ", dsoLocal: " << GVFlags.DSOLocal;
3044 Out << ")";
3045
3046 if (Summary.getSummaryKind() == GlobalValueSummary::AliasKind)
3047 printAliasSummary(cast<AliasSummary>(&Summary));
3048 else if (Summary.getSummaryKind() == GlobalValueSummary::FunctionKind)
3049 printFunctionSummary(cast<FunctionSummary>(&Summary));
3050 else
3051 printGlobalVarSummary(cast<GlobalVarSummary>(&Summary));
3052
3053 auto RefList = Summary.refs();
3054 if (!RefList.empty()) {
3055 Out << ", refs: (";
3056 FieldSeparator FS;
3057 for (auto &Ref : RefList) {
3058 Out << FS;
Eugene Leviant009d8332018-11-23 10:54:51 +00003059 if (Ref.isReadOnly())
3060 Out << "readonly ";
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00003061 Out << "^" << Machine.getGUIDSlot(Ref.getGUID());
3062 }
3063 Out << ")";
3064 }
3065
3066 Out << ")";
3067}
3068
3069void AssemblyWriter::printSummaryInfo(unsigned Slot, const ValueInfo &VI) {
3070 Out << "^" << Slot << " = gv: (";
3071 if (!VI.name().empty())
3072 Out << "name: \"" << VI.name() << "\"";
3073 else
3074 Out << "guid: " << VI.getGUID();
3075 if (!VI.getSummaryList().empty()) {
3076 Out << ", summaries: (";
3077 FieldSeparator FS;
3078 for (auto &Summary : VI.getSummaryList()) {
3079 Out << FS;
3080 printSummary(*Summary);
3081 }
3082 Out << ")";
3083 }
3084 Out << ")";
3085 if (!VI.name().empty())
3086 Out << " ; guid = " << VI.getGUID();
3087 Out << "\n";
3088}
3089
Filipe Cabecinhas62431b12015-06-02 21:25:08 +00003090static void printMetadataIdentifier(StringRef Name,
3091 formatted_raw_ostream &Out) {
Nick Lewycky5bcbd732011-06-15 06:37:58 +00003092 if (Name.empty()) {
3093 Out << "<empty name> ";
3094 } else {
Filipe Cabecinhas1ac0d2d2015-06-02 21:25:00 +00003095 if (isalpha(static_cast<unsigned char>(Name[0])) || Name[0] == '-' ||
3096 Name[0] == '$' || Name[0] == '.' || Name[0] == '_')
Nick Lewycky5bcbd732011-06-15 06:37:58 +00003097 Out << Name[0];
3098 else
3099 Out << '\\' << hexdigit(Name[0] >> 4) << hexdigit(Name[0] & 0x0F);
3100 for (unsigned i = 1, e = Name.size(); i != e; ++i) {
3101 unsigned char C = Name[i];
Guy Benyei83c74e92013-02-12 21:21:59 +00003102 if (isalnum(static_cast<unsigned char>(C)) || C == '-' || C == '$' ||
3103 C == '.' || C == '_')
Nick Lewycky5bcbd732011-06-15 06:37:58 +00003104 Out << C;
3105 else
3106 Out << '\\' << hexdigit(C >> 4) << hexdigit(C & 0x0F);
3107 }
3108 }
Filipe Cabecinhas62431b12015-06-02 21:25:08 +00003109}
3110
3111void AssemblyWriter::printNamedMDNode(const NamedMDNode *NMD) {
3112 Out << '!';
3113 printMetadataIdentifier(NMD->getName(), Out);
Nick Lewycky5bcbd732011-06-15 06:37:58 +00003114 Out << " = !{";
Chris Lattner2c48c642009-12-31 01:54:05 +00003115 for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) {
Filipe Cabecinhas1ac0d2d2015-06-02 21:25:00 +00003116 if (i)
3117 Out << ", ";
Reid Kleckner6d353342017-08-23 20:31:27 +00003118
3119 // Write DIExpressions inline.
3120 // FIXME: Ban DIExpressions in NamedMDNodes, they will serve no purpose.
3121 MDNode *Op = NMD->getOperand(i);
3122 if (auto *Expr = dyn_cast<DIExpression>(Op)) {
3123 writeDIExpression(Out, Expr, nullptr, nullptr, nullptr);
3124 continue;
3125 }
3126
3127 int Slot = Machine.getMetadataSlot(Op);
Dan Gohman0a54da22010-09-09 20:53:58 +00003128 if (Slot == -1)
3129 Out << "<badref>";
3130 else
3131 Out << '!' << Slot;
Chris Lattner2c48c642009-12-31 01:54:05 +00003132 }
3133 Out << "}\n";
3134}
3135
Chris Lattner3a36d2c2008-08-19 05:06:27 +00003136static void PrintVisibility(GlobalValue::VisibilityTypes Vis,
Dan Gohmane2745262009-08-12 17:23:50 +00003137 formatted_raw_ostream &Out) {
Chris Lattner3a36d2c2008-08-19 05:06:27 +00003138 switch (Vis) {
Chris Lattner3a36d2c2008-08-19 05:06:27 +00003139 case GlobalValue::DefaultVisibility: break;
3140 case GlobalValue::HiddenVisibility: Out << "hidden "; break;
3141 case GlobalValue::ProtectedVisibility: Out << "protected "; break;
3142 }
3143}
3144
Rafael Espindolae4b02312018-01-11 22:15:05 +00003145static void PrintDSOLocation(const GlobalValue &GV,
3146 formatted_raw_ostream &Out) {
Rafael Espindola9fbc0402018-01-18 02:08:23 +00003147 // GVs with local linkage or non default visibility are implicitly dso_local,
3148 // so we don't print it.
3149 bool Implicit = GV.hasLocalLinkage() ||
3150 (!GV.hasExternalWeakLinkage() && !GV.hasDefaultVisibility());
3151 if (GV.isDSOLocal() && !Implicit)
Sean Fertilec70d28b2017-10-26 15:00:26 +00003152 Out << "dso_local ";
3153}
3154
Nico Rieck7157bb72014-01-14 15:22:47 +00003155static void PrintDLLStorageClass(GlobalValue::DLLStorageClassTypes SCT,
3156 formatted_raw_ostream &Out) {
3157 switch (SCT) {
3158 case GlobalValue::DefaultStorageClass: break;
3159 case GlobalValue::DLLImportStorageClass: Out << "dllimport "; break;
3160 case GlobalValue::DLLExportStorageClass: Out << "dllexport "; break;
3161 }
3162}
3163
Hans Wennborgcbe34b42012-06-23 11:37:03 +00003164static void PrintThreadLocalModel(GlobalVariable::ThreadLocalMode TLM,
3165 formatted_raw_ostream &Out) {
3166 switch (TLM) {
3167 case GlobalVariable::NotThreadLocal:
3168 break;
3169 case GlobalVariable::GeneralDynamicTLSModel:
3170 Out << "thread_local ";
3171 break;
3172 case GlobalVariable::LocalDynamicTLSModel:
3173 Out << "thread_local(localdynamic) ";
3174 break;
3175 case GlobalVariable::InitialExecTLSModel:
3176 Out << "thread_local(initialexec) ";
3177 break;
3178 case GlobalVariable::LocalExecTLSModel:
3179 Out << "thread_local(localexec) ";
3180 break;
3181 }
3182}
3183
Peter Collingbourne96efdd62016-06-14 21:01:22 +00003184static StringRef getUnnamedAddrEncoding(GlobalVariable::UnnamedAddr UA) {
3185 switch (UA) {
3186 case GlobalVariable::UnnamedAddr::None:
3187 return "";
3188 case GlobalVariable::UnnamedAddr::Local:
3189 return "local_unnamed_addr";
3190 case GlobalVariable::UnnamedAddr::Global:
3191 return "unnamed_addr";
3192 }
Aaron Ballman2b9fa3f2016-06-15 15:27:53 +00003193 llvm_unreachable("Unknown UnnamedAddr");
Peter Collingbourne96efdd62016-06-14 21:01:22 +00003194}
3195
Rafael Espindola83a362c2015-01-06 22:55:16 +00003196static void maybePrintComdat(formatted_raw_ostream &Out,
3197 const GlobalObject &GO) {
3198 const Comdat *C = GO.getComdat();
3199 if (!C)
3200 return;
3201
3202 if (isa<GlobalVariable>(GO))
3203 Out << ',';
3204 Out << " comdat";
3205
3206 if (GO.getName() == C->getName())
3207 return;
3208
3209 Out << '(';
3210 PrintLLVMName(Out, C->getName(), ComdatPrefix);
3211 Out << ')';
3212}
3213
Chris Lattner7bfee412001-10-29 16:05:51 +00003214void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
Dan Gohman5ded1422010-01-29 23:12:36 +00003215 if (GV->isMaterializable())
3216 Out << "; Materializable\n";
3217
Dan Gohmana2489d12010-07-20 23:55:01 +00003218 WriteAsOperandInternal(Out, GV, &TypePrinter, &Machine, GV->getParent());
Dan Gohman466876b2009-08-12 23:32:33 +00003219 Out << " = ";
Chris Lattner37798642001-09-18 04:01:05 +00003220
Chris Lattner1508d3f2008-08-19 05:16:28 +00003221 if (!GV->hasInitializer() && GV->hasExternalLinkage())
3222 Out << "external ";
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003223
Teresa Johnsonc80c8732018-05-25 15:15:39 +00003224 Out << getLinkageNameWithSpace(GV->getLinkage());
Rafael Espindolae4b02312018-01-11 22:15:05 +00003225 PrintDSOLocation(*GV, Out);
Chris Lattner1508d3f2008-08-19 05:16:28 +00003226 PrintVisibility(GV->getVisibility(), Out);
Nico Rieck7157bb72014-01-14 15:22:47 +00003227 PrintDLLStorageClass(GV->getDLLStorageClass(), Out);
Hans Wennborgcbe34b42012-06-23 11:37:03 +00003228 PrintThreadLocalModel(GV->getThreadLocalMode(), Out);
Peter Collingbourne96efdd62016-06-14 21:01:22 +00003229 StringRef UA = getUnnamedAddrEncoding(GV->getUnnamedAddr());
3230 if (!UA.empty())
3231 Out << UA << ' ';
Lauro Ramos Venancio749e4662007-04-12 18:32:50 +00003232
Chris Lattnerac161bf2009-01-02 07:01:27 +00003233 if (unsigned AddressSpace = GV->getType()->getAddressSpace())
3234 Out << "addrspace(" << AddressSpace << ") ";
Michael Gottesman27e7ef32013-02-05 05:57:38 +00003235 if (GV->isExternallyInitialized()) Out << "externally_initialized ";
Misha Brukmana6619a92004-06-21 21:53:56 +00003236 Out << (GV->isConstant() ? "constant " : "global ");
Manuel Jacob5f6eaac2016-01-16 20:30:46 +00003237 TypePrinter.print(GV->getValueType(), Out);
Chris Lattner37798642001-09-18 04:01:05 +00003238
Dan Gohman81313fd2008-09-14 17:21:12 +00003239 if (GV->hasInitializer()) {
3240 Out << ' ';
Devang Patel983c6b12009-07-08 21:44:25 +00003241 writeOperand(GV->getInitializer(), false);
Dan Gohman81313fd2008-09-14 17:21:12 +00003242 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003243
Chris Lattner9380b812010-07-07 23:16:37 +00003244 if (GV->hasSection()) {
3245 Out << ", section \"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00003246 printEscapedString(GV->getSection(), Out);
Chris Lattner9380b812010-07-07 23:16:37 +00003247 Out << '"';
3248 }
Rafael Espindola83a362c2015-01-06 22:55:16 +00003249 maybePrintComdat(Out, *GV);
Chris Lattner4b96c542005-11-12 00:10:19 +00003250 if (GV->getAlignment())
Chris Lattnerf8a974d2005-11-06 06:48:53 +00003251 Out << ", align " << GV->getAlignment();
Anton Korobeynikova97b6942007-04-25 14:27:10 +00003252
Peter Collingbournecceae7f2016-05-31 23:01:54 +00003253 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
3254 GV->getAllMetadata(MDs);
3255 printMetadataAttachments(MDs, ", ");
3256
Javed Absarf3d79042017-05-11 12:28:08 +00003257 auto Attrs = GV->getAttributes();
3258 if (Attrs.hasAttributes())
3259 Out << " #" << Machine.getAttributeGroupSlot(Attrs);
3260
Chris Lattner113f4f42002-06-25 16:13:24 +00003261 printInfoComment(*GV);
Chris Lattnerda975502001-09-10 07:58:01 +00003262}
3263
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003264void AssemblyWriter::printIndirectSymbol(const GlobalIndirectSymbol *GIS) {
3265 if (GIS->isMaterializable())
Dan Gohman5ded1422010-01-29 23:12:36 +00003266 Out << "; Materializable\n";
3267
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003268 WriteAsOperandInternal(Out, GIS, &TypePrinter, &Machine, GIS->getParent());
Rafael Espindola54fc2982015-06-17 17:53:31 +00003269 Out << " = ";
3270
Teresa Johnsonc80c8732018-05-25 15:15:39 +00003271 Out << getLinkageNameWithSpace(GIS->getLinkage());
Rafael Espindolae4b02312018-01-11 22:15:05 +00003272 PrintDSOLocation(*GIS, Out);
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003273 PrintVisibility(GIS->getVisibility(), Out);
3274 PrintDLLStorageClass(GIS->getDLLStorageClass(), Out);
3275 PrintThreadLocalModel(GIS->getThreadLocalMode(), Out);
Peter Collingbourne96efdd62016-06-14 21:01:22 +00003276 StringRef UA = getUnnamedAddrEncoding(GIS->getUnnamedAddr());
3277 if (!UA.empty())
3278 Out << UA << ' ';
Anton Korobeynikova97b6942007-04-25 14:27:10 +00003279
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003280 if (isa<GlobalAlias>(GIS))
3281 Out << "alias ";
Dmitry Polukhina1feff72016-04-07 12:32:19 +00003282 else if (isa<GlobalIFunc>(GIS))
3283 Out << "ifunc ";
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003284 else
Dmitry Polukhina1feff72016-04-07 12:32:19 +00003285 llvm_unreachable("Not an alias or ifunc!");
Anton Korobeynikova97b6942007-04-25 14:27:10 +00003286
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003287 TypePrinter.print(GIS->getValueType(), Out);
David Blaikie2f408302015-09-11 03:22:04 +00003288
3289 Out << ", ";
3290
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003291 const Constant *IS = GIS->getIndirectSymbol();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003292
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003293 if (!IS) {
3294 TypePrinter.print(GIS->getType(), Out);
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003295 Out << " <<NULL ALIASEE>>";
Jay Foad92c19132011-08-01 12:48:54 +00003296 } else {
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003297 writeOperand(IS, !isa<ConstantExpr>(IS));
Jay Foad92c19132011-08-01 12:48:54 +00003298 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003299
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00003300 printInfoComment(*GIS);
Chris Lattner1508d3f2008-08-19 05:16:28 +00003301 Out << '\n';
Anton Korobeynikova97b6942007-04-25 14:27:10 +00003302}
3303
David Majnemerdad0a642014-06-27 18:19:56 +00003304void AssemblyWriter::printComdat(const Comdat *C) {
3305 C->print(Out);
3306}
3307
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003308void AssemblyWriter::printTypeIdentities() {
Roman Tereshind96de6f2018-03-22 21:29:07 +00003309 if (TypePrinter.empty())
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003310 return;
Andrew Trickec4b6e72011-09-30 19:48:58 +00003311
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003312 Out << '\n';
Andrew Trickec4b6e72011-09-30 19:48:58 +00003313
Chris Lattner3243ea12009-03-01 00:03:38 +00003314 // Emit all numbered types.
Roman Tereshind96de6f2018-03-22 21:29:07 +00003315 auto &NumberedTypes = TypePrinter.getNumberedTypes();
3316 for (unsigned I = 0, E = NumberedTypes.size(); I != E; ++I) {
3317 Out << '%' << I << " = type ";
Andrew Trickec4b6e72011-09-30 19:48:58 +00003318
Chris Lattner3243ea12009-03-01 00:03:38 +00003319 // Make sure we print out at least one level of the type structure, so
3320 // that we do not get %2 = type %2
Roman Tereshind96de6f2018-03-22 21:29:07 +00003321 TypePrinter.printStructBody(NumberedTypes[I], Out);
Dan Gohman69273e62009-08-12 23:54:22 +00003322 Out << '\n';
Chris Lattner3243ea12009-03-01 00:03:38 +00003323 }
Andrew Trickec4b6e72011-09-30 19:48:58 +00003324
Roman Tereshind96de6f2018-03-22 21:29:07 +00003325 auto &NamedTypes = TypePrinter.getNamedTypes();
3326 for (unsigned I = 0, E = NamedTypes.size(); I != E; ++I) {
3327 PrintLLVMName(Out, NamedTypes[I]->getName(), LocalPrefix);
Chris Lattner1508d3f2008-08-19 05:16:28 +00003328 Out << " = type ";
Reid Spencere7e96712004-05-25 08:53:40 +00003329
3330 // Make sure we print out at least one level of the type structure, so
3331 // that we do not get %FILE = type %FILE
Roman Tereshind96de6f2018-03-22 21:29:07 +00003332 TypePrinter.printStructBody(NamedTypes[I], Out);
Chris Lattner3a36d2c2008-08-19 05:06:27 +00003333 Out << '\n';
Reid Spencere7e96712004-05-25 08:53:40 +00003334 }
Reid Spencer32af9e82007-01-06 07:24:44 +00003335}
3336
Misha Brukmanc566ca362004-03-02 00:22:19 +00003337/// printFunction - Print all aspects of a function.
Chris Lattner113f4f42002-06-25 16:13:24 +00003338void AssemblyWriter::printFunction(const Function *F) {
Chris Lattner3a36d2c2008-08-19 05:06:27 +00003339 // Print out the return type and name.
3340 Out << '\n';
Chris Lattner379a8d22003-04-16 20:28:45 +00003341
Misha Brukmana6619a92004-06-21 21:53:56 +00003342 if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00003343
Dan Gohman5ded1422010-01-29 23:12:36 +00003344 if (F->isMaterializable())
3345 Out << "; Materializable\n";
3346
Reid Klecknerb5180542017-03-21 16:57:19 +00003347 const AttributeList &Attrs = F->getAttributes();
3348 if (Attrs.hasAttributes(AttributeList::FunctionIndex)) {
Reid Klecknerc2cb5602017-04-12 00:38:00 +00003349 AttributeSet AS = Attrs.getFnAttributes();
Rafael Espindolacbf5a7a2013-05-01 13:07:03 +00003350 std::string AttrStr;
3351
Reid Klecknerc2cb5602017-04-12 00:38:00 +00003352 for (const Attribute &Attr : AS) {
Rafael Espindolacbf5a7a2013-05-01 13:07:03 +00003353 if (!Attr.isStringAttribute()) {
3354 if (!AttrStr.empty()) AttrStr += ' ';
3355 AttrStr += Attr.getAsString();
3356 }
3357 }
3358
Bill Wendling847a5c32013-04-16 20:55:47 +00003359 if (!AttrStr.empty())
3360 Out << "; Function Attrs: " << AttrStr << '\n';
3361 }
3362
Peter Collingbourne21521892016-06-21 23:42:48 +00003363 Machine.incorporateFunction(F);
3364
3365 if (F->isDeclaration()) {
3366 Out << "declare";
3367 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
3368 F->getAllMetadata(MDs);
3369 printMetadataAttachments(MDs, " ");
3370 Out << ' ';
3371 } else
Reid Spencer7ce2d2a2006-12-29 20:29:48 +00003372 Out << "define ";
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003373
Teresa Johnsonc80c8732018-05-25 15:15:39 +00003374 Out << getLinkageNameWithSpace(F->getLinkage());
Rafael Espindolae4b02312018-01-11 22:15:05 +00003375 PrintDSOLocation(*F, Out);
Chris Lattner3a36d2c2008-08-19 05:06:27 +00003376 PrintVisibility(F->getVisibility(), Out);
Nico Rieck7157bb72014-01-14 15:22:47 +00003377 PrintDLLStorageClass(F->getDLLStorageClass(), Out);
Chris Lattner379a8d22003-04-16 20:28:45 +00003378
Chris Lattnerf7b6d312005-05-06 20:26:43 +00003379 // Print the calling convention.
Micah Villmow857186d2012-09-13 15:11:12 +00003380 if (F->getCallingConv() != CallingConv::C) {
3381 PrintCallingConv(F->getCallingConv(), Out);
3382 Out << " ";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00003383 }
3384
Chris Lattner229907c2011-07-18 04:54:35 +00003385 FunctionType *FT = F->getFunctionType();
Reid Klecknerb5180542017-03-21 16:57:19 +00003386 if (Attrs.hasAttributes(AttributeList::ReturnIndex))
3387 Out << Attrs.getAsString(AttributeList::ReturnIndex) << ' ';
Chris Lattnere101c442009-02-28 21:26:53 +00003388 TypePrinter.print(F->getReturnType(), Out);
Chris Lattner585297e82008-08-19 05:26:17 +00003389 Out << ' ';
Dan Gohmana2489d12010-07-20 23:55:01 +00003390 WriteAsOperandInternal(Out, F, &TypePrinter, &Machine, F->getParent());
Misha Brukmana6619a92004-06-21 21:53:56 +00003391 Out << '(';
Chris Lattnerfee714f2001-09-07 16:36:04 +00003392
Chris Lattner7bfee412001-10-29 16:05:51 +00003393 // Loop over the arguments, printing them...
Justin Bognerd7d1a722015-09-27 22:38:50 +00003394 if (F->isDeclaration() && !IsForDebug) {
Justin Bogner58e08232015-09-02 17:54:41 +00003395 // We're only interested in the type here - don't print argument names.
3396 for (unsigned I = 0, E = FT->getNumParams(); I != E; ++I) {
Chris Lattner82738fe2007-04-18 00:57:22 +00003397 // Insert commas as we go... the first arg doesn't get a comma
Justin Bogner58e08232015-09-02 17:54:41 +00003398 if (I)
3399 Out << ", ";
3400 // Output type...
3401 TypePrinter.print(FT->getParamType(I), Out);
3402
Reid Kleckner6652a522017-04-28 18:37:16 +00003403 AttributeSet ArgAttrs = Attrs.getParamAttributes(I);
3404 if (ArgAttrs.hasAttributes())
3405 Out << ' ' << ArgAttrs.getAsString();
Chris Lattner82738fe2007-04-18 00:57:22 +00003406 }
3407 } else {
Justin Bogner58e08232015-09-02 17:54:41 +00003408 // The arguments are meaningful here, print them in detail.
Justin Bogner58e08232015-09-02 17:54:41 +00003409 for (const Argument &Arg : F->args()) {
Chris Lattner82738fe2007-04-18 00:57:22 +00003410 // Insert commas as we go... the first arg doesn't get a comma
Reid Kleckner6652a522017-04-28 18:37:16 +00003411 if (Arg.getArgNo() != 0)
Justin Bogner58e08232015-09-02 17:54:41 +00003412 Out << ", ";
Reid Kleckner6652a522017-04-28 18:37:16 +00003413 printArgument(&Arg, Attrs.getParamAttributes(Arg.getArgNo()));
Chris Lattner82738fe2007-04-18 00:57:22 +00003414 }
Reid Spencer8c4914c2006-12-31 05:24:50 +00003415 }
Chris Lattnerfee714f2001-09-07 16:36:04 +00003416
3417 // Finish printing arguments...
Chris Lattner113f4f42002-06-25 16:13:24 +00003418 if (FT->isVarArg()) {
Misha Brukmana6619a92004-06-21 21:53:56 +00003419 if (FT->getNumParams()) Out << ", ";
3420 Out << "..."; // Output varargs portion of signature!
Chris Lattnerfee714f2001-09-07 16:36:04 +00003421 }
Misha Brukmana6619a92004-06-21 21:53:56 +00003422 Out << ')';
Peter Collingbourne96efdd62016-06-14 21:01:22 +00003423 StringRef UA = getUnnamedAddrEncoding(F->getUnnamedAddr());
3424 if (!UA.empty())
3425 Out << ' ' << UA;
Alexander Richardson6bcf2ba2018-08-23 09:25:17 +00003426 // We print the function address space if it is non-zero or if we are writing
3427 // a module with a non-zero program address space or if there is no valid
3428 // Module* so that the file can be parsed without the datalayout string.
3429 const Module *Mod = F->getParent();
3430 if (F->getAddressSpace() != 0 || !Mod ||
3431 Mod->getDataLayout().getProgramAddressSpace() != 0)
3432 Out << " addrspace(" << F->getAddressSpace() << ")";
Reid Klecknerb5180542017-03-21 16:57:19 +00003433 if (Attrs.hasAttributes(AttributeList::FunctionIndex))
Bill Wendling04945972013-04-29 23:48:06 +00003434 Out << " #" << Machine.getAttributeGroupSlot(Attrs.getFnAttributes());
Chris Lattner9380b812010-07-07 23:16:37 +00003435 if (F->hasSection()) {
3436 Out << " section \"";
Jonas Devlieghere745918ff2018-05-31 17:01:42 +00003437 printEscapedString(F->getSection(), Out);
Chris Lattner9380b812010-07-07 23:16:37 +00003438 Out << '"';
3439 }
Rafael Espindola83a362c2015-01-06 22:55:16 +00003440 maybePrintComdat(Out, *F);
Chris Lattnerf8a974d2005-11-06 06:48:53 +00003441 if (F->getAlignment())
3442 Out << " align " << F->getAlignment();
Gordon Henriksend930f912008-08-17 18:44:35 +00003443 if (F->hasGC())
3444 Out << " gc \"" << F->getGC() << '"';
Peter Collingbourne3fa50f92013-09-16 01:08:15 +00003445 if (F->hasPrefixData()) {
3446 Out << " prefix ";
3447 writeOperand(F->getPrefixData(), true);
3448 }
Peter Collingbourne51d2de72014-12-03 02:08:38 +00003449 if (F->hasPrologueData()) {
3450 Out << " prologue ";
3451 writeOperand(F->getPrologueData(), true);
3452 }
David Majnemer7fddecc2015-06-17 20:52:32 +00003453 if (F->hasPersonalityFn()) {
3454 Out << " personality ";
3455 writeOperand(F->getPersonalityFn(), /*PrintType=*/true);
3456 }
Peter Collingbourne51d2de72014-12-03 02:08:38 +00003457
Reid Spencer5301e7c2007-01-30 20:08:39 +00003458 if (F->isDeclaration()) {
Chris Lattnerfb483622010-09-02 22:52:10 +00003459 Out << '\n';
Chris Lattnerb2f02e52002-05-06 03:00:40 +00003460 } else {
Peter Collingbourne21521892016-06-21 23:42:48 +00003461 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
3462 F->getAllMetadata(MDs);
3463 printMetadataAttachments(MDs, " ");
3464
Chris Lattnerfb483622010-09-02 22:52:10 +00003465 Out << " {";
3466 // Output all of the function's basic blocks.
Benjamin Krameraf28e7d2016-06-26 14:10:56 +00003467 for (const BasicBlock &BB : *F)
3468 printBasicBlock(&BB);
Chris Lattnerfee714f2001-09-07 16:36:04 +00003469
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +00003470 // Output the function's use-lists.
3471 printUseLists(F);
3472
Misha Brukmana6619a92004-06-21 21:53:56 +00003473 Out << "}\n";
Chris Lattnerfee714f2001-09-07 16:36:04 +00003474 }
3475
Reid Spencer16f2f7f2004-05-26 07:18:52 +00003476 Machine.purgeFunction();
Chris Lattner2f7c9632001-06-06 20:29:01 +00003477}
3478
Misha Brukmanc566ca362004-03-02 00:22:19 +00003479/// printArgument - This member is called for every argument that is passed into
3480/// the function. Simply print it out
Reid Kleckner6652a522017-04-28 18:37:16 +00003481void AssemblyWriter::printArgument(const Argument *Arg, AttributeSet Attrs) {
Chris Lattner2f7c9632001-06-06 20:29:01 +00003482 // Output type...
Chris Lattnere101c442009-02-28 21:26:53 +00003483 TypePrinter.print(Arg->getType(), Out);
Misha Brukmanb1c93172005-04-21 23:48:37 +00003484
Duncan Sandsad0ea2d2007-11-27 13:23:08 +00003485 // Output parameter attributes list
Reid Kleckner6652a522017-04-28 18:37:16 +00003486 if (Attrs.hasAttributes())
3487 Out << ' ' << Attrs.getAsString();
Reid Spencer8c4914c2006-12-31 05:24:50 +00003488
Chris Lattner2f7c9632001-06-06 20:29:01 +00003489 // Output name, if available...
Chris Lattner033935d2008-08-17 04:40:13 +00003490 if (Arg->hasName()) {
3491 Out << ' ';
3492 PrintLLVMName(Out, Arg);
3493 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00003494}
3495
Misha Brukmanc566ca362004-03-02 00:22:19 +00003496/// printBasicBlock - This member is called for each basic block in a method.
Chris Lattner7bfee412001-10-29 16:05:51 +00003497void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
James Y Knightc0e6b8a2019-03-22 18:27:13 +00003498 bool IsEntryBlock = BB == &BB->getParent()->getEntryBlock();
Nick Lewycky4d43d3c2008-04-25 16:53:59 +00003499 if (BB->hasName()) { // Print out the label if it exists...
Chris Lattner033935d2008-08-17 04:40:13 +00003500 Out << "\n";
Daniel Dunbare03eecb2009-07-25 23:55:21 +00003501 PrintLLVMName(Out, BB->getName(), LabelPrefix);
Chris Lattner033935d2008-08-17 04:40:13 +00003502 Out << ':';
James Y Knightc0e6b8a2019-03-22 18:27:13 +00003503 } else if (!IsEntryBlock) {
3504 Out << "\n";
Chris Lattner5e043322007-01-11 03:54:27 +00003505 int Slot = Machine.getLocalSlot(BB);
Chris Lattner757ee0b2004-06-09 19:41:19 +00003506 if (Slot != -1)
Evgeniy Stepanove257f0f2016-01-27 21:53:08 +00003507 Out << Slot << ":";
Chris Lattner757ee0b2004-06-09 19:41:19 +00003508 else
James Y Knightc0e6b8a2019-03-22 18:27:13 +00003509 Out << "<badref>:";
Chris Lattner58185f22002-10-02 19:38:55 +00003510 }
Chris Lattner2447ef52003-11-20 00:09:43 +00003511
Craig Topperc6207612014-04-09 06:08:46 +00003512 if (!BB->getParent()) {
Chris Lattneree97b8b2009-08-17 15:48:08 +00003513 Out.PadToColumn(50);
Dan Gohmane2745262009-08-12 17:23:50 +00003514 Out << "; Error: Block without parent!";
James Y Knightc0e6b8a2019-03-22 18:27:13 +00003515 } else if (!IsEntryBlock) {
Chris Lattnerfb483622010-09-02 22:52:10 +00003516 // Output predecessors for the block.
Chris Lattneree97b8b2009-08-17 15:48:08 +00003517 Out.PadToColumn(50);
Dan Gohmane2745262009-08-12 17:23:50 +00003518 Out << ";";
Gabor Greif6c6b2fd2010-03-25 23:25:28 +00003519 const_pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003520
Chris Lattnerff834c02008-04-22 02:45:44 +00003521 if (PI == PE) {
3522 Out << " No predecessors!";
3523 } else {
Dan Gohman81313fd2008-09-14 17:21:12 +00003524 Out << " preds = ";
Chris Lattnerff834c02008-04-22 02:45:44 +00003525 writeOperand(*PI, false);
3526 for (++PI; PI != PE; ++PI) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003527 Out << ", ";
Chris Lattner78e2e8b2006-12-06 06:24:27 +00003528 writeOperand(*PI, false);
Chris Lattner00211f12003-11-16 22:59:57 +00003529 }
Chris Lattner58185f22002-10-02 19:38:55 +00003530 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00003531 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00003532
Chris Lattnerff834c02008-04-22 02:45:44 +00003533 Out << "\n";
Chris Lattner2f7c9632001-06-06 20:29:01 +00003534
Misha Brukmana6619a92004-06-21 21:53:56 +00003535 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00003536
Chris Lattnerfee714f2001-09-07 16:36:04 +00003537 // Output all of the instructions in the basic block...
Benjamin Krameraf28e7d2016-06-26 14:10:56 +00003538 for (const Instruction &I : *BB) {
3539 printInstructionLine(I);
Dan Gohmana4f709e2009-07-13 18:27:59 +00003540 }
Chris Lattner96cdd272004-03-08 18:51:45 +00003541
Misha Brukmana6619a92004-06-21 21:53:56 +00003542 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
Chris Lattner2f7c9632001-06-06 20:29:01 +00003543}
3544
Daniel Maleaded9f932013-05-08 20:38:31 +00003545/// printInstructionLine - Print an instruction and a newline character.
3546void AssemblyWriter::printInstructionLine(const Instruction &I) {
3547 printInstruction(I);
3548 Out << '\n';
3549}
3550
Igor Laevsky2aa8caf2015-05-05 13:20:42 +00003551/// printGCRelocateComment - print comment after call to the gc.relocate
3552/// intrinsic indicating base and derived pointer names.
Manuel Jacob83eefa62016-01-05 04:03:00 +00003553void AssemblyWriter::printGCRelocateComment(const GCRelocateInst &Relocate) {
Igor Laevsky2aa8caf2015-05-05 13:20:42 +00003554 Out << " ; (";
Manuel Jacob83eefa62016-01-05 04:03:00 +00003555 writeOperand(Relocate.getBasePtr(), false);
Igor Laevsky2aa8caf2015-05-05 13:20:42 +00003556 Out << ", ";
Manuel Jacob83eefa62016-01-05 04:03:00 +00003557 writeOperand(Relocate.getDerivedPtr(), false);
Igor Laevsky2aa8caf2015-05-05 13:20:42 +00003558 Out << ")";
3559}
3560
Misha Brukmanc566ca362004-03-02 00:22:19 +00003561/// printInfoComment - Print a little comment after the instruction indicating
3562/// which slot it occupies.
Chris Lattner113f4f42002-06-25 16:13:24 +00003563void AssemblyWriter::printInfoComment(const Value &V) {
Manuel Jacob83eefa62016-01-05 04:03:00 +00003564 if (const auto *Relocate = dyn_cast<GCRelocateInst>(&V))
3565 printGCRelocateComment(*Relocate);
Igor Laevsky2aa8caf2015-05-05 13:20:42 +00003566
Bill Wendling847a5c32013-04-16 20:55:47 +00003567 if (AnnotationWriter)
Dan Gohmane34890f2010-02-10 20:41:46 +00003568 AnnotationWriter->printInfoComment(V, Out);
Chris Lattner862e3382001-10-13 06:42:36 +00003569}
3570
Alexander Richardson6bcf2ba2018-08-23 09:25:17 +00003571static void maybePrintCallAddrSpace(const Value *Operand, const Instruction *I,
3572 raw_ostream &Out) {
3573 // We print the address space of the call if it is non-zero.
3574 unsigned CallAddrSpace = Operand->getType()->getPointerAddressSpace();
3575 bool PrintAddrSpace = CallAddrSpace != 0;
3576 if (!PrintAddrSpace) {
3577 const Module *Mod = getModuleFromVal(I);
3578 // We also print it if it is zero but not equal to the program address space
3579 // or if we can't find a valid Module* to make it possible to parse
3580 // the resulting file even without a datalayout string.
3581 if (!Mod || Mod->getDataLayout().getProgramAddressSpace() != 0)
3582 PrintAddrSpace = true;
3583 }
3584 if (PrintAddrSpace)
3585 Out << " addrspace(" << CallAddrSpace << ")";
3586}
3587
Reid Spencere7141c82006-08-28 01:02:49 +00003588// This member is called for each Instruction in a function..
Chris Lattner113f4f42002-06-25 16:13:24 +00003589void AssemblyWriter::printInstruction(const Instruction &I) {
Misha Brukmana6619a92004-06-21 21:53:56 +00003590 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
Chris Lattner8339f7d2003-10-30 23:41:03 +00003591
Dan Gohman466876b2009-08-12 23:32:33 +00003592 // Print out indentation for an instruction.
Dan Gohmanb4583b12009-08-13 01:41:52 +00003593 Out << " ";
Chris Lattner2f7c9632001-06-06 20:29:01 +00003594
3595 // Print out name if it exists...
Chris Lattner033935d2008-08-17 04:40:13 +00003596 if (I.hasName()) {
3597 PrintLLVMName(Out, &I);
3598 Out << " = ";
Chris Lattner031560c2009-12-29 07:25:48 +00003599 } else if (!I.getType()->isVoidTy()) {
Chris Lattnerb25e5ea2008-08-29 17:19:30 +00003600 // Print out the def slot taken.
3601 int SlotNum = Machine.getLocalSlot(&I);
3602 if (SlotNum == -1)
3603 Out << "<badref> = ";
3604 else
3605 Out << '%' << SlotNum << " = ";
Chris Lattner033935d2008-08-17 04:40:13 +00003606 }
Andrew Trickec4b6e72011-09-30 19:48:58 +00003607
Reid Kleckner5772b772014-04-24 20:14:34 +00003608 if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
3609 if (CI->isMustTailCall())
3610 Out << "musttail ";
3611 else if (CI->isTailCall())
3612 Out << "tail ";
Akira Hatanaka5cfcce122015-11-06 23:55:38 +00003613 else if (CI->isNoTailCall())
3614 Out << "notail ";
Reid Kleckner5772b772014-04-24 20:14:34 +00003615 }
Chris Lattner504f9242003-09-08 17:45:59 +00003616
Chris Lattner2f7c9632001-06-06 20:29:01 +00003617 // Print out the opcode...
Misha Brukmana6619a92004-06-21 21:53:56 +00003618 Out << I.getOpcodeName();
Chris Lattner2f7c9632001-06-06 20:29:01 +00003619
Eli Friedman02e737b2011-08-12 22:50:01 +00003620 // If this is an atomic load or store, print out the atomic marker.
3621 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isAtomic()) ||
3622 (isa<StoreInst>(I) && cast<StoreInst>(I).isAtomic()))
3623 Out << " atomic";
3624
Tim Northover420a2162014-06-13 14:24:07 +00003625 if (isa<AtomicCmpXchgInst>(I) && cast<AtomicCmpXchgInst>(I).isWeak())
3626 Out << " weak";
3627
Eli Friedman02e737b2011-08-12 22:50:01 +00003628 // If this is a volatile operation, print out the volatile marker.
3629 if ((isa<LoadInst>(I) && cast<LoadInst>(I).isVolatile()) ||
3630 (isa<StoreInst>(I) && cast<StoreInst>(I).isVolatile()) ||
3631 (isa<AtomicCmpXchgInst>(I) && cast<AtomicCmpXchgInst>(I).isVolatile()) ||
3632 (isa<AtomicRMWInst>(I) && cast<AtomicRMWInst>(I).isVolatile()))
3633 Out << " volatile";
3634
Dan Gohman9c7f8082009-07-27 16:11:46 +00003635 // Print out optimization information.
3636 WriteOptimizationInfo(Out, &I);
3637
Reid Spencer45e52392006-12-03 06:27:29 +00003638 // Print out the compare instruction predicates
Nate Begemand2195702008-05-12 19:01:56 +00003639 if (const CmpInst *CI = dyn_cast<CmpInst>(&I))
Tim Northoverde3aea0412016-08-17 20:25:25 +00003640 Out << ' ' << CmpInst::getPredicateName(CI->getPredicate());
Reid Spencer45e52392006-12-03 06:27:29 +00003641
Eli Friedmanc9a551e2011-07-28 21:48:00 +00003642 // Print out the atomicrmw operation
3643 if (const AtomicRMWInst *RMWI = dyn_cast<AtomicRMWInst>(&I))
Matt Arsenaultb02ba992018-10-02 23:44:11 +00003644 Out << ' ' << AtomicRMWInst::getOperationName(RMWI->getOperation());
Eli Friedmanc9a551e2011-07-28 21:48:00 +00003645
Chris Lattner2f7c9632001-06-06 20:29:01 +00003646 // Print out the type of the operands...
Craig Topperc6207612014-04-09 06:08:46 +00003647 const Value *Operand = I.getNumOperands() ? I.getOperand(0) : nullptr;
Chris Lattner2f7c9632001-06-06 20:29:01 +00003648
3649 // Special case conditional branches to swizzle the condition out to the front
Gabor Greifcab008f2009-02-09 15:45:06 +00003650 if (isa<BranchInst>(I) && cast<BranchInst>(I).isConditional()) {
David Blaikieb78e9e52013-02-11 01:16:51 +00003651 const BranchInst &BI(cast<BranchInst>(I));
Dan Gohman81313fd2008-09-14 17:21:12 +00003652 Out << ' ';
Gabor Greifcab008f2009-02-09 15:45:06 +00003653 writeOperand(BI.getCondition(), true);
Dan Gohman81313fd2008-09-14 17:21:12 +00003654 Out << ", ";
Gabor Greifcab008f2009-02-09 15:45:06 +00003655 writeOperand(BI.getSuccessor(0), true);
Dan Gohman81313fd2008-09-14 17:21:12 +00003656 Out << ", ";
Gabor Greifcab008f2009-02-09 15:45:06 +00003657 writeOperand(BI.getSuccessor(1), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00003658
Chris Lattner8d48df22002-04-13 18:34:38 +00003659 } else if (isa<SwitchInst>(I)) {
David Blaikieb78e9e52013-02-11 01:16:51 +00003660 const SwitchInst& SI(cast<SwitchInst>(I));
Chris Lattner3ed871f2009-10-27 19:13:16 +00003661 // Special case switch instruction to get formatting nice and correct.
Dan Gohman81313fd2008-09-14 17:21:12 +00003662 Out << ' ';
Eli Friedman95031ed2011-09-29 20:21:17 +00003663 writeOperand(SI.getCondition(), true);
Dan Gohman81313fd2008-09-14 17:21:12 +00003664 Out << ", ";
Eli Friedman95031ed2011-09-29 20:21:17 +00003665 writeOperand(SI.getDefaultDest(), true);
Chris Lattner82ff9232008-08-23 22:52:27 +00003666 Out << " [";
Chandler Carruth927d8e62017-04-12 07:27:28 +00003667 for (auto Case : SI.cases()) {
Dan Gohmanb4583b12009-08-13 01:41:52 +00003668 Out << "\n ";
Chandler Carruth927d8e62017-04-12 07:27:28 +00003669 writeOperand(Case.getCaseValue(), true);
Dan Gohman81313fd2008-09-14 17:21:12 +00003670 Out << ", ";
Chandler Carruth927d8e62017-04-12 07:27:28 +00003671 writeOperand(Case.getCaseSuccessor(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00003672 }
Dan Gohmanb4583b12009-08-13 01:41:52 +00003673 Out << "\n ]";
Chris Lattnerd04cb6d2009-10-28 00:19:10 +00003674 } else if (isa<IndirectBrInst>(I)) {
3675 // Special case indirectbr instruction to get formatting nice and correct.
Chris Lattner3ed871f2009-10-27 19:13:16 +00003676 Out << ' ';
3677 writeOperand(Operand, true);
Dan Gohman43c57402009-10-30 02:01:10 +00003678 Out << ", [";
Andrew Trickec4b6e72011-09-30 19:48:58 +00003679
Chris Lattner3ed871f2009-10-27 19:13:16 +00003680 for (unsigned i = 1, e = I.getNumOperands(); i != e; ++i) {
3681 if (i != 1)
3682 Out << ", ";
3683 writeOperand(I.getOperand(i), true);
3684 }
3685 Out << ']';
Jay Foad372ad642011-06-20 14:18:48 +00003686 } else if (const PHINode *PN = dyn_cast<PHINode>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00003687 Out << ' ';
Chris Lattnere101c442009-02-28 21:26:53 +00003688 TypePrinter.print(I.getType(), Out);
Misha Brukmana6619a92004-06-21 21:53:56 +00003689 Out << ' ';
Chris Lattner2f7c9632001-06-06 20:29:01 +00003690
Jay Foad372ad642011-06-20 14:18:48 +00003691 for (unsigned op = 0, Eop = PN->getNumIncomingValues(); op < Eop; ++op) {
Misha Brukmana6619a92004-06-21 21:53:56 +00003692 if (op) Out << ", ";
Dan Gohman81313fd2008-09-14 17:21:12 +00003693 Out << "[ ";
Jay Foad372ad642011-06-20 14:18:48 +00003694 writeOperand(PN->getIncomingValue(op), false); Out << ", ";
3695 writeOperand(PN->getIncomingBlock(op), false); Out << " ]";
Chris Lattner931ef3b2001-06-11 15:04:20 +00003696 }
Dan Gohmana76f0f72008-05-31 19:12:39 +00003697 } else if (const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(&I)) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003698 Out << ' ';
Dan Gohmana76f0f72008-05-31 19:12:39 +00003699 writeOperand(I.getOperand(0), true);
3700 for (const unsigned *i = EVI->idx_begin(), *e = EVI->idx_end(); i != e; ++i)
3701 Out << ", " << *i;
3702 } else if (const InsertValueInst *IVI = dyn_cast<InsertValueInst>(&I)) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003703 Out << ' ';
3704 writeOperand(I.getOperand(0), true); Out << ", ";
Dan Gohmana76f0f72008-05-31 19:12:39 +00003705 writeOperand(I.getOperand(1), true);
3706 for (const unsigned *i = IVI->idx_begin(), *e = IVI->idx_end(); i != e; ++i)
3707 Out << ", " << *i;
Bill Wendlingfae14752011-08-12 20:24:12 +00003708 } else if (const LandingPadInst *LPI = dyn_cast<LandingPadInst>(&I)) {
3709 Out << ' ';
3710 TypePrinter.print(I.getType(), Out);
David Majnemer7fddecc2015-06-17 20:52:32 +00003711 if (LPI->isCleanup() || LPI->getNumClauses() != 0)
3712 Out << '\n';
Bill Wendlingfae14752011-08-12 20:24:12 +00003713
3714 if (LPI->isCleanup())
3715 Out << " cleanup";
3716
3717 for (unsigned i = 0, e = LPI->getNumClauses(); i != e; ++i) {
3718 if (i != 0 || LPI->isCleanup()) Out << "\n";
3719 if (LPI->isCatch(i))
3720 Out << " catch ";
3721 else
3722 Out << " filter ";
3723
3724 writeOperand(LPI->getClause(i), true);
3725 }
David Majnemer8a1c45d2015-12-12 05:38:55 +00003726 } else if (const auto *CatchSwitch = dyn_cast<CatchSwitchInst>(&I)) {
3727 Out << " within ";
3728 writeOperand(CatchSwitch->getParentPad(), /*PrintType=*/false);
David Majnemer654e1302015-07-31 17:58:14 +00003729 Out << " [";
David Majnemer8a1c45d2015-12-12 05:38:55 +00003730 unsigned Op = 0;
3731 for (const BasicBlock *PadBB : CatchSwitch->handlers()) {
3732 if (Op > 0)
3733 Out << ", ";
3734 writeOperand(PadBB, /*PrintType=*/true);
3735 ++Op;
3736 }
3737 Out << "] unwind ";
3738 if (const BasicBlock *UnwindDest = CatchSwitch->getUnwindDest())
3739 writeOperand(UnwindDest, /*PrintType=*/true);
3740 else
3741 Out << "to caller";
3742 } else if (const auto *FPI = dyn_cast<FuncletPadInst>(&I)) {
3743 Out << " within ";
3744 writeOperand(FPI->getParentPad(), /*PrintType=*/false);
3745 Out << " [";
3746 for (unsigned Op = 0, NumOps = FPI->getNumArgOperands(); Op < NumOps;
David Majnemer654e1302015-07-31 17:58:14 +00003747 ++Op) {
3748 if (Op > 0)
3749 Out << ", ";
David Majnemer8a1c45d2015-12-12 05:38:55 +00003750 writeOperand(FPI->getArgOperand(Op), /*PrintType=*/true);
David Majnemer654e1302015-07-31 17:58:14 +00003751 }
David Majnemer8a1c45d2015-12-12 05:38:55 +00003752 Out << ']';
Devang Patel59643e52008-02-23 00:35:18 +00003753 } else if (isa<ReturnInst>(I) && !Operand) {
3754 Out << " void";
David Majnemer0bc0eef2015-08-15 02:46:08 +00003755 } else if (const auto *CRI = dyn_cast<CatchReturnInst>(&I)) {
David Majnemer8a1c45d2015-12-12 05:38:55 +00003756 Out << " from ";
3757 writeOperand(CRI->getOperand(0), /*PrintType=*/false);
David Majnemer0bc0eef2015-08-15 02:46:08 +00003758
3759 Out << " to ";
David Majnemer8a1c45d2015-12-12 05:38:55 +00003760 writeOperand(CRI->getOperand(1), /*PrintType=*/true);
David Majnemer654e1302015-07-31 17:58:14 +00003761 } else if (const auto *CRI = dyn_cast<CleanupReturnInst>(&I)) {
David Majnemer8a1c45d2015-12-12 05:38:55 +00003762 Out << " from ";
3763 writeOperand(CRI->getOperand(0), /*PrintType=*/false);
David Majnemer654e1302015-07-31 17:58:14 +00003764
3765 Out << " unwind ";
3766 if (CRI->hasUnwindDest())
David Majnemer8a1c45d2015-12-12 05:38:55 +00003767 writeOperand(CRI->getOperand(1), /*PrintType=*/true);
Joseph Tremoulet9ce71f72015-09-03 09:09:43 +00003768 else
3769 Out << "to caller";
Chris Lattnerf7b6d312005-05-06 20:26:43 +00003770 } else if (const CallInst *CI = dyn_cast<CallInst>(&I)) {
3771 // Print the calling convention being used.
Micah Villmow857186d2012-09-13 15:11:12 +00003772 if (CI->getCallingConv() != CallingConv::C) {
3773 Out << " ";
3774 PrintCallingConv(CI->getCallingConv(), Out);
Chris Lattnerf7b6d312005-05-06 20:26:43 +00003775 }
3776
Gabor Greifc9a92512010-06-23 13:09:06 +00003777 Operand = CI->getCalledValue();
Ahmed Bougachaaa9fe532016-12-21 23:26:13 +00003778 FunctionType *FTy = CI->getFunctionType();
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003779 Type *RetTy = FTy->getReturnType();
Reid Klecknerb5180542017-03-21 16:57:19 +00003780 const AttributeList &PAL = CI->getAttributes();
Chris Lattner2f2d9472001-11-06 21:28:12 +00003781
Reid Klecknerb5180542017-03-21 16:57:19 +00003782 if (PAL.hasAttributes(AttributeList::ReturnIndex))
3783 Out << ' ' << PAL.getAsString(AttributeList::ReturnIndex);
Devang Patel221fe422008-09-29 20:49:50 +00003784
Alexander Richardson6bcf2ba2018-08-23 09:25:17 +00003785 // Only print addrspace(N) if necessary:
3786 maybePrintCallAddrSpace(Operand, &I, Out);
3787
Chris Lattner463d6a52003-08-05 15:34:45 +00003788 // If possible, print out the short form of the call instruction. We can
Chris Lattner6915f8f2002-04-07 22:49:37 +00003789 // only do this if the first argument is a pointer to a nonvararg function,
Chris Lattner463d6a52003-08-05 15:34:45 +00003790 // and if the return type is not a pointer to a function.
Chris Lattner2f2d9472001-11-06 21:28:12 +00003791 //
Dan Gohman81313fd2008-09-14 17:21:12 +00003792 Out << ' ';
David Blaikie23af6482015-04-16 23:24:18 +00003793 TypePrinter.print(FTy->isVarArg() ? FTy : RetTy, Out);
3794 Out << ' ';
3795 writeOperand(Operand, false);
Misha Brukmana6619a92004-06-21 21:53:56 +00003796 Out << '(';
Gabor Greifc9a92512010-06-23 13:09:06 +00003797 for (unsigned op = 0, Eop = CI->getNumArgOperands(); op < Eop; ++op) {
3798 if (op > 0)
Dan Gohman81313fd2008-09-14 17:21:12 +00003799 Out << ", ";
Reid Kleckner6652a522017-04-28 18:37:16 +00003800 writeParamOperand(CI->getArgOperand(op), PAL.getParamAttributes(op));
Chris Lattner2f7c9632001-06-06 20:29:01 +00003801 }
Reid Kleckner83498642014-08-26 00:33:28 +00003802
3803 // Emit an ellipsis if this is a musttail call in a vararg function. This
3804 // is only to aid readability, musttail calls forward varargs by default.
3805 if (CI->isMustTailCall() && CI->getParent() &&
3806 CI->getParent()->getParent() &&
3807 CI->getParent()->getParent()->isVarArg())
3808 Out << ", ...";
3809
Dan Gohman81313fd2008-09-14 17:21:12 +00003810 Out << ')';
Reid Klecknerb5180542017-03-21 16:57:19 +00003811 if (PAL.hasAttributes(AttributeList::FunctionIndex))
Bill Wendlinga0323742013-02-22 09:09:42 +00003812 Out << " #" << Machine.getAttributeGroupSlot(PAL.getFnAttributes());
Sanjoy Dasb513a9f2015-09-24 23:34:52 +00003813
3814 writeOperandBundles(CI);
Chris Lattner113f4f42002-06-25 16:13:24 +00003815 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(&I)) {
Gabor Greifa2fbc0a2010-03-24 13:21:49 +00003816 Operand = II->getCalledValue();
Ahmed Bougachaaa9fe532016-12-21 23:26:13 +00003817 FunctionType *FTy = II->getFunctionType();
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003818 Type *RetTy = FTy->getReturnType();
Reid Klecknerb5180542017-03-21 16:57:19 +00003819 const AttributeList &PAL = II->getAttributes();
Chris Lattner463d6a52003-08-05 15:34:45 +00003820
Chris Lattnerf7b6d312005-05-06 20:26:43 +00003821 // Print the calling convention being used.
Micah Villmow857186d2012-09-13 15:11:12 +00003822 if (II->getCallingConv() != CallingConv::C) {
3823 Out << " ";
3824 PrintCallingConv(II->getCallingConv(), Out);
Chris Lattnerf7b6d312005-05-06 20:26:43 +00003825 }
3826
Reid Klecknerb5180542017-03-21 16:57:19 +00003827 if (PAL.hasAttributes(AttributeList::ReturnIndex))
3828 Out << ' ' << PAL.getAsString(AttributeList::ReturnIndex);
Devang Patel221fe422008-09-29 20:49:50 +00003829
Alexander Richardson6bcf2ba2018-08-23 09:25:17 +00003830 // Only print addrspace(N) if necessary:
3831 maybePrintCallAddrSpace(Operand, &I, Out);
3832
Chris Lattner463d6a52003-08-05 15:34:45 +00003833 // If possible, print out the short form of the invoke instruction. We can
3834 // only do this if the first argument is a pointer to a nonvararg function,
3835 // and if the return type is not a pointer to a function.
3836 //
Dan Gohmanc7e00ba2008-10-15 18:02:08 +00003837 Out << ' ';
David Blaikie445e3fb2015-04-24 19:32:54 +00003838 TypePrinter.print(FTy->isVarArg() ? FTy : RetTy, Out);
3839 Out << ' ';
3840 writeOperand(Operand, false);
Misha Brukmana6619a92004-06-21 21:53:56 +00003841 Out << '(';
Gabor Greifc9a92512010-06-23 13:09:06 +00003842 for (unsigned op = 0, Eop = II->getNumArgOperands(); op < Eop; ++op) {
Gabor Greifa2fbc0a2010-03-24 13:21:49 +00003843 if (op)
Dan Gohman81313fd2008-09-14 17:21:12 +00003844 Out << ", ";
Reid Kleckner6652a522017-04-28 18:37:16 +00003845 writeParamOperand(II->getArgOperand(op), PAL.getParamAttributes(op));
Chris Lattner862e3382001-10-13 06:42:36 +00003846 }
3847
Dan Gohman81313fd2008-09-14 17:21:12 +00003848 Out << ')';
Reid Klecknerb5180542017-03-21 16:57:19 +00003849 if (PAL.hasAttributes(AttributeList::FunctionIndex))
Bill Wendlinga0323742013-02-22 09:09:42 +00003850 Out << " #" << Machine.getAttributeGroupSlot(PAL.getFnAttributes());
Devang Patela05633e2008-09-26 22:53:05 +00003851
Sanjoy Dasb513a9f2015-09-24 23:34:52 +00003852 writeOperandBundles(II);
3853
Dan Gohmanb4583b12009-08-13 01:41:52 +00003854 Out << "\n to ";
Chris Lattner862e3382001-10-13 06:42:36 +00003855 writeOperand(II->getNormalDest(), true);
Dan Gohman81313fd2008-09-14 17:21:12 +00003856 Out << " unwind ";
Chris Lattnerfae8ab32004-02-08 21:44:31 +00003857 writeOperand(II->getUnwindDest(), true);
Craig Topper784929d2019-02-08 20:48:56 +00003858 } else if (const CallBrInst *CBI = dyn_cast<CallBrInst>(&I)) {
3859 Operand = CBI->getCalledValue();
3860 FunctionType *FTy = CBI->getFunctionType();
3861 Type *RetTy = FTy->getReturnType();
3862 const AttributeList &PAL = CBI->getAttributes();
3863
3864 // Print the calling convention being used.
3865 if (CBI->getCallingConv() != CallingConv::C) {
3866 Out << " ";
3867 PrintCallingConv(CBI->getCallingConv(), Out);
3868 }
3869
3870 if (PAL.hasAttributes(AttributeList::ReturnIndex))
3871 Out << ' ' << PAL.getAsString(AttributeList::ReturnIndex);
3872
3873 // If possible, print out the short form of the callbr instruction. We can
3874 // only do this if the first argument is a pointer to a nonvararg function,
3875 // and if the return type is not a pointer to a function.
3876 //
3877 Out << ' ';
3878 TypePrinter.print(FTy->isVarArg() ? FTy : RetTy, Out);
3879 Out << ' ';
3880 writeOperand(Operand, false);
3881 Out << '(';
3882 for (unsigned op = 0, Eop = CBI->getNumArgOperands(); op < Eop; ++op) {
3883 if (op)
3884 Out << ", ";
3885 writeParamOperand(CBI->getArgOperand(op), PAL.getParamAttributes(op));
3886 }
3887
3888 Out << ')';
3889 if (PAL.hasAttributes(AttributeList::FunctionIndex))
3890 Out << " #" << Machine.getAttributeGroupSlot(PAL.getFnAttributes());
3891
3892 writeOperandBundles(CBI);
3893
3894 Out << "\n to ";
3895 writeOperand(CBI->getDefaultDest(), true);
3896 Out << " [";
3897 for (unsigned i = 0, e = CBI->getNumIndirectDests(); i != e; ++i) {
3898 if (i != 0)
3899 Out << ", ";
3900 writeOperand(CBI->getIndirectDest(i), true);
3901 }
3902 Out << ']';
Victor Hernandez8acf2952009-10-23 21:09:37 +00003903 } else if (const AllocaInst *AI = dyn_cast<AllocaInst>(&I)) {
Misha Brukmana6619a92004-06-21 21:53:56 +00003904 Out << ' ';
Reid Kleckner20844032014-01-25 01:24:06 +00003905 if (AI->isUsedWithInAlloca())
David Majnemerc4ab61c2014-03-09 06:41:58 +00003906 Out << "inalloca ";
Manman Ren9bfd0d02016-04-01 21:41:15 +00003907 if (AI->isSwiftError())
3908 Out << "swifterror ";
David Majnemerc4ab61c2014-03-09 06:41:58 +00003909 TypePrinter.print(AI->getAllocatedType(), Out);
Duncan P. N. Exon Smith720762e2015-03-13 19:30:44 +00003910
3911 // Explicitly write the array size if the code is broken, if it's an array
3912 // allocation, or if the type is not canonical for scalar allocations. The
3913 // latter case prevents the type from mutating when round-tripping through
3914 // assembly.
3915 if (!AI->getArraySize() || AI->isArrayAllocation() ||
3916 !AI->getArraySize()->getType()->isIntegerTy(32)) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003917 Out << ", ";
Chris Lattner8d48df22002-04-13 18:34:38 +00003918 writeOperand(AI->getArraySize(), true);
Chris Lattner2f7c9632001-06-06 20:29:01 +00003919 }
Nate Begeman848622f2005-11-05 09:21:28 +00003920 if (AI->getAlignment()) {
Chris Lattner7aeee3a2005-11-05 21:20:34 +00003921 Out << ", align " << AI->getAlignment();
Nate Begeman848622f2005-11-05 09:21:28 +00003922 }
Matt Arsenault3c1fc762017-04-10 22:27:50 +00003923
3924 unsigned AddrSpace = AI->getType()->getAddressSpace();
3925 if (AddrSpace != 0) {
3926 Out << ", addrspace(" << AddrSpace << ')';
3927 }
Chris Lattner862e3382001-10-13 06:42:36 +00003928 } else if (isa<CastInst>(I)) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003929 if (Operand) {
3930 Out << ' ';
3931 writeOperand(Operand, true); // Work with broken code
3932 }
Misha Brukmana6619a92004-06-21 21:53:56 +00003933 Out << " to ";
Chris Lattnere101c442009-02-28 21:26:53 +00003934 TypePrinter.print(I.getType(), Out);
Chris Lattner5b337482003-10-18 05:57:43 +00003935 } else if (isa<VAArgInst>(I)) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003936 if (Operand) {
3937 Out << ' ';
3938 writeOperand(Operand, true); // Work with broken code
3939 }
Misha Brukmana6619a92004-06-21 21:53:56 +00003940 Out << ", ";
Chris Lattnere101c442009-02-28 21:26:53 +00003941 TypePrinter.print(I.getType(), Out);
3942 } else if (Operand) { // Print the normal way.
David Blaikiea79ac142015-02-27 21:17:42 +00003943 if (const auto *GEP = dyn_cast<GetElementPtrInst>(&I)) {
David Blaikie79e6c742015-02-27 19:29:02 +00003944 Out << ' ';
3945 TypePrinter.print(GEP->getSourceElementType(), Out);
3946 Out << ',';
David Blaikiea79ac142015-02-27 21:17:42 +00003947 } else if (const auto *LI = dyn_cast<LoadInst>(&I)) {
3948 Out << ' ';
3949 TypePrinter.print(LI->getType(), Out);
Benjamin Kramer257ed692015-03-02 15:24:41 +00003950 Out << ',';
David Blaikie79e6c742015-02-27 19:29:02 +00003951 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00003952
Misha Brukmanb1c93172005-04-21 23:48:37 +00003953 // PrintAllTypes - Instructions who have operands of all the same type
Chris Lattner2f7c9632001-06-06 20:29:01 +00003954 // omit the type from all but the first operand. If the instruction has
3955 // different type operands (for example br), then they are all printed.
3956 bool PrintAllTypes = false;
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00003957 Type *TheType = Operand->getType();
Chris Lattner2f7c9632001-06-06 20:29:01 +00003958
Reid Spencer0cdd04f2007-02-02 13:54:55 +00003959 // Select, Store and ShuffleVector always print all types.
Devang Patelce556d92008-03-04 22:05:14 +00003960 if (isa<SelectInst>(I) || isa<StoreInst>(I) || isa<ShuffleVectorInst>(I)
3961 || isa<ReturnInst>(I)) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00003962 PrintAllTypes = true;
3963 } else {
3964 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
3965 Operand = I.getOperand(i);
Nuno Lopes04fe2f02009-01-15 18:40:57 +00003966 // note that Operand shouldn't be null, but the test helps make dump()
3967 // more tolerant of malformed IR
Nuno Lopes0971e772009-01-14 17:51:41 +00003968 if (Operand && Operand->getType() != TheType) {
Chris Lattnerdeccfaf2003-04-16 20:20:02 +00003969 PrintAllTypes = true; // We have differing types! Print them all!
3970 break;
3971 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00003972 }
3973 }
Misha Brukmanb1c93172005-04-21 23:48:37 +00003974
Chris Lattner7bfee412001-10-29 16:05:51 +00003975 if (!PrintAllTypes) {
Misha Brukmana6619a92004-06-21 21:53:56 +00003976 Out << ' ';
Chris Lattnere101c442009-02-28 21:26:53 +00003977 TypePrinter.print(TheType, Out);
Chris Lattner7bfee412001-10-29 16:05:51 +00003978 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00003979
Dan Gohman81313fd2008-09-14 17:21:12 +00003980 Out << ' ';
Chris Lattner113f4f42002-06-25 16:13:24 +00003981 for (unsigned i = 0, E = I.getNumOperands(); i != E; ++i) {
Dan Gohman81313fd2008-09-14 17:21:12 +00003982 if (i) Out << ", ";
Chris Lattner113f4f42002-06-25 16:13:24 +00003983 writeOperand(I.getOperand(i), PrintAllTypes);
Chris Lattner2f7c9632001-06-06 20:29:01 +00003984 }
3985 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00003986
Eli Friedman59b66882011-08-09 23:02:53 +00003987 // Print atomic ordering/alignment for memory operations
3988 if (const LoadInst *LI = dyn_cast<LoadInst>(&I)) {
3989 if (LI->isAtomic())
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00003990 writeAtomic(LI->getContext(), LI->getOrdering(), LI->getSyncScopeID());
Eli Friedman59b66882011-08-09 23:02:53 +00003991 if (LI->getAlignment())
3992 Out << ", align " << LI->getAlignment();
3993 } else if (const StoreInst *SI = dyn_cast<StoreInst>(&I)) {
3994 if (SI->isAtomic())
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00003995 writeAtomic(SI->getContext(), SI->getOrdering(), SI->getSyncScopeID());
Eli Friedman59b66882011-08-09 23:02:53 +00003996 if (SI->getAlignment())
3997 Out << ", align " << SI->getAlignment();
Eli Friedmanc9a551e2011-07-28 21:48:00 +00003998 } else if (const AtomicCmpXchgInst *CXI = dyn_cast<AtomicCmpXchgInst>(&I)) {
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00003999 writeAtomicCmpXchg(CXI->getContext(), CXI->getSuccessOrdering(),
4000 CXI->getFailureOrdering(), CXI->getSyncScopeID());
Eli Friedmanc9a551e2011-07-28 21:48:00 +00004001 } else if (const AtomicRMWInst *RMWI = dyn_cast<AtomicRMWInst>(&I)) {
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00004002 writeAtomic(RMWI->getContext(), RMWI->getOrdering(),
4003 RMWI->getSyncScopeID());
Eli Friedmanfee02c62011-07-25 23:16:38 +00004004 } else if (const FenceInst *FI = dyn_cast<FenceInst>(&I)) {
Konstantin Zhuravlyovbb80d3e2017-07-11 22:23:00 +00004005 writeAtomic(FI->getContext(), FI->getOrdering(), FI->getSyncScopeID());
Christopher Lamb84485702007-04-22 19:24:39 +00004006 }
Chris Lattner2f7c9632001-06-06 20:29:01 +00004007
Chris Lattnerc9558df2009-12-28 20:10:43 +00004008 // Print Metadata info.
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +00004009 SmallVector<std::pair<unsigned, MDNode *>, 4> InstMD;
Nick Lewyckyba8ec9a2010-02-25 06:53:04 +00004010 I.getAllMetadata(InstMD);
Duncan P. N. Exon Smithd3e4c2a2015-04-24 21:06:21 +00004011 printMetadataAttachments(InstMD, ", ");
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00004012
4013 // Print a nice comment.
Chris Lattner862e3382001-10-13 06:42:36 +00004014 printInfoComment(I);
Chris Lattner2f7c9632001-06-06 20:29:01 +00004015}
4016
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00004017void AssemblyWriter::printMetadataAttachments(
Duncan P. N. Exon Smithd3e4c2a2015-04-24 21:06:21 +00004018 const SmallVectorImpl<std::pair<unsigned, MDNode *>> &MDs,
4019 StringRef Separator) {
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00004020 if (MDs.empty())
4021 return;
4022
Duncan P. N. Exon Smithe30f10e2015-04-24 21:03:05 +00004023 if (MDNames.empty())
Mehdi Aminib9b50aa2016-01-07 20:14:30 +00004024 MDs[0].second->getContext().getMDKindNames(MDNames);
Duncan P. N. Exon Smithe30f10e2015-04-24 21:03:05 +00004025
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00004026 for (const auto &I : MDs) {
4027 unsigned Kind = I.first;
Duncan P. N. Exon Smithd3e4c2a2015-04-24 21:06:21 +00004028 Out << Separator;
Filipe Cabecinhas62431b12015-06-02 21:25:08 +00004029 if (Kind < MDNames.size()) {
4030 Out << "!";
4031 printMetadataIdentifier(MDNames[Kind], Out);
4032 } else
Duncan P. N. Exon Smithd3e4c2a2015-04-24 21:06:21 +00004033 Out << "!<unknown kind #" << Kind << ">";
Duncan P. N. Exon Smith86979272015-04-24 20:59:52 +00004034 Out << ' ';
4035 WriteAsOperandInternal(Out, I.second, &TypePrinter, &Machine, TheModule);
4036 }
4037}
4038
Daniel Maleaded9f932013-05-08 20:38:31 +00004039void AssemblyWriter::writeMDNode(unsigned Slot, const MDNode *Node) {
Duncan P. N. Exon Smithbe7ea192014-12-15 19:07:53 +00004040 Out << '!' << Slot << " = ";
Daniel Maleaded9f932013-05-08 20:38:31 +00004041 printMDNodeBody(Node);
Duncan P. N. Exon Smith738889f2015-02-25 22:46:38 +00004042 Out << "\n";
Daniel Maleaded9f932013-05-08 20:38:31 +00004043}
4044
Chris Lattnerbddea6a2009-12-31 02:13:35 +00004045void AssemblyWriter::writeAllMDNodes() {
4046 SmallVector<const MDNode *, 16> Nodes;
Chris Lattnercf4a76e2009-12-31 02:20:11 +00004047 Nodes.resize(Machine.mdn_size());
4048 for (SlotTracker::mdn_iterator I = Machine.mdn_begin(), E = Machine.mdn_end();
4049 I != E; ++I)
Chris Lattnerbddea6a2009-12-31 02:13:35 +00004050 Nodes[I->second] = cast<MDNode>(I->first);
Andrew Trickec4b6e72011-09-30 19:48:58 +00004051
Chris Lattnerbddea6a2009-12-31 02:13:35 +00004052 for (unsigned i = 0, e = Nodes.size(); i != e; ++i) {
Daniel Maleaded9f932013-05-08 20:38:31 +00004053 writeMDNode(i, Nodes[i]);
Chris Lattnerbddea6a2009-12-31 02:13:35 +00004054 }
4055}
4056
4057void AssemblyWriter::printMDNodeBody(const MDNode *Node) {
Dan Gohmana2489d12010-07-20 23:55:01 +00004058 WriteMDNodeBodyInternal(Out, Node, &TypePrinter, &Machine, TheModule);
Chris Lattnerbddea6a2009-12-31 02:13:35 +00004059}
Chris Lattner2f7c9632001-06-06 20:29:01 +00004060
Bill Wendling829b4782013-02-11 08:43:33 +00004061void AssemblyWriter::writeAllAttributeGroups() {
Reid Klecknerc2cb5602017-04-12 00:38:00 +00004062 std::vector<std::pair<AttributeSet, unsigned>> asVec;
Bill Wendling829b4782013-02-11 08:43:33 +00004063 asVec.resize(Machine.as_size());
4064
4065 for (SlotTracker::as_iterator I = Machine.as_begin(), E = Machine.as_end();
4066 I != E; ++I)
4067 asVec[I->second] = *I;
4068
Benjamin Krameraf28e7d2016-06-26 14:10:56 +00004069 for (const auto &I : asVec)
4070 Out << "attributes #" << I.second << " = { "
Reid Klecknerc2cb5602017-04-12 00:38:00 +00004071 << I.first.getAsString(true) << " }\n";
Bill Wendling829b4782013-02-11 08:43:33 +00004072}
4073
Duncan P. N. Exon Smith0a448fb2014-08-19 21:30:15 +00004074void AssemblyWriter::printUseListOrder(const UseListOrder &Order) {
4075 bool IsInFunction = Machine.getFunction();
4076 if (IsInFunction)
4077 Out << " ";
4078
4079 Out << "uselistorder";
4080 if (const BasicBlock *BB =
4081 IsInFunction ? nullptr : dyn_cast<BasicBlock>(Order.V)) {
4082 Out << "_bb ";
4083 writeOperand(BB->getParent(), false);
4084 Out << ", ";
4085 writeOperand(BB, false);
4086 } else {
4087 Out << " ";
4088 writeOperand(Order.V, true);
4089 }
4090 Out << ", { ";
4091
4092 assert(Order.Shuffle.size() >= 2 && "Shuffle too small");
4093 Out << Order.Shuffle[0];
4094 for (unsigned I = 1, E = Order.Shuffle.size(); I != E; ++I)
4095 Out << ", " << Order.Shuffle[I];
4096 Out << " }\n";
4097}
4098
4099void AssemblyWriter::printUseLists(const Function *F) {
4100 auto hasMore =
4101 [&]() { return !UseListOrders.empty() && UseListOrders.back().F == F; };
4102 if (!hasMore())
4103 // Nothing to do.
4104 return;
4105
4106 Out << "\n; uselistorder directives\n";
4107 while (hasMore()) {
4108 printUseListOrder(UseListOrders.back());
4109 UseListOrders.pop_back();
4110 }
4111}
4112
Chris Lattner2f7c9632001-06-06 20:29:01 +00004113//===----------------------------------------------------------------------===//
4114// External Interface declarations
4115//===----------------------------------------------------------------------===//
4116
George Burgess IVe1100f52016-02-02 22:46:49 +00004117void Function::print(raw_ostream &ROS, AssemblyAnnotationWriter *AAW,
4118 bool ShouldPreserveUseListOrder,
4119 bool IsForDebug) const {
4120 SlotTracker SlotTable(this->getParent());
4121 formatted_raw_ostream OS(ROS);
4122 AssemblyWriter W(OS, SlotTable, this->getParent(), AAW,
4123 IsForDebug,
4124 ShouldPreserveUseListOrder);
4125 W.printFunction(this);
4126}
4127
Duncan P. N. Exon Smithc4f0a322015-04-15 02:12:41 +00004128void Module::print(raw_ostream &ROS, AssemblyAnnotationWriter *AAW,
Justin Bognerd7d1a722015-09-27 22:38:50 +00004129 bool ShouldPreserveUseListOrder, bool IsForDebug) const {
Chris Lattnere36fd8a2008-08-19 04:26:57 +00004130 SlotTracker SlotTable(this);
Dan Gohmane2745262009-08-12 17:23:50 +00004131 formatted_raw_ostream OS(ROS);
Justin Bognerd7d1a722015-09-27 22:38:50 +00004132 AssemblyWriter W(OS, SlotTable, this, AAW, IsForDebug,
4133 ShouldPreserveUseListOrder);
Chris Lattnerefb5e392009-12-31 02:23:35 +00004134 W.printModule(this);
Chris Lattner2f7c9632001-06-06 20:29:01 +00004135}
4136
Justin Bognerd7d1a722015-09-27 22:38:50 +00004137void NamedMDNode::print(raw_ostream &ROS, bool IsForDebug) const {
Dan Gohman2637cc12010-07-21 23:38:33 +00004138 SlotTracker SlotTable(getParent());
4139 formatted_raw_ostream OS(ROS);
Justin Bognerd7d1a722015-09-27 22:38:50 +00004140 AssemblyWriter W(OS, SlotTable, getParent(), nullptr, IsForDebug);
Dan Gohman2637cc12010-07-21 23:38:33 +00004141 W.printNamedMDNode(this);
4142}
4143
Duncan P. N. Exon Smith0ecff952016-04-20 17:27:44 +00004144void NamedMDNode::print(raw_ostream &ROS, ModuleSlotTracker &MST,
4145 bool IsForDebug) const {
4146 Optional<SlotTracker> LocalST;
4147 SlotTracker *SlotTable;
4148 if (auto *ST = MST.getMachine())
4149 SlotTable = ST;
4150 else {
4151 LocalST.emplace(getParent());
4152 SlotTable = &*LocalST;
4153 }
4154
4155 formatted_raw_ostream OS(ROS);
4156 AssemblyWriter W(OS, *SlotTable, getParent(), nullptr, IsForDebug);
4157 W.printNamedMDNode(this);
4158}
4159
Justin Bognerd7d1a722015-09-27 22:38:50 +00004160void Comdat::print(raw_ostream &ROS, bool /*IsForDebug*/) const {
David Majnemerdad0a642014-06-27 18:19:56 +00004161 PrintLLVMName(ROS, getName(), ComdatPrefix);
4162 ROS << " = comdat ";
4163
4164 switch (getSelectionKind()) {
4165 case Comdat::Any:
4166 ROS << "any";
4167 break;
4168 case Comdat::ExactMatch:
4169 ROS << "exactmatch";
4170 break;
4171 case Comdat::Largest:
4172 ROS << "largest";
4173 break;
4174 case Comdat::NoDuplicates:
4175 ROS << "noduplicates";
4176 break;
4177 case Comdat::SameSize:
4178 ROS << "samesize";
4179 break;
4180 }
4181
4182 ROS << '\n';
4183}
4184
Quentin Colombet774b1ef2016-03-07 22:32:42 +00004185void Type::print(raw_ostream &OS, bool /*IsForDebug*/, bool NoDetails) const {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00004186 TypePrinting TP;
4187 TP.print(const_cast<Type*>(this), OS);
Andrew Trickec4b6e72011-09-30 19:48:58 +00004188
Quentin Colombet774b1ef2016-03-07 22:32:42 +00004189 if (NoDetails)
4190 return;
4191
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00004192 // If the type is a named struct type, print the body as well.
4193 if (StructType *STy = dyn_cast<StructType>(const_cast<Type*>(this)))
Chris Lattner01beceb2011-08-12 18:07:07 +00004194 if (!STy->isLiteral()) {
Chris Lattnerb1ed91f2011-07-09 17:41:24 +00004195 OS << " = type ";
4196 TP.printStructBody(STy, OS);
4197 }
Chris Lattnerc0b4c7b2002-04-08 22:03:40 +00004198}
4199
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004200static bool isReferencingMDNode(const Instruction &I) {
4201 if (const auto *CI = dyn_cast<CallInst>(&I))
4202 if (Function *F = CI->getCalledFunction())
Justin Lebar9cbc3012016-07-28 23:58:15 +00004203 if (F->isIntrinsic())
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004204 for (auto &Op : I.operands())
4205 if (auto *V = dyn_cast_or_null<MetadataAsValue>(Op))
4206 if (isa<MDNode>(V->getMetadata()))
4207 return true;
4208 return false;
4209}
4210
Justin Bognerd7d1a722015-09-27 22:38:50 +00004211void Value::print(raw_ostream &ROS, bool IsForDebug) const {
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004212 bool ShouldInitializeAllMetadata = false;
4213 if (auto *I = dyn_cast<Instruction>(this))
4214 ShouldInitializeAllMetadata = isReferencingMDNode(*I);
4215 else if (isa<Function>(this) || isa<MetadataAsValue>(this))
4216 ShouldInitializeAllMetadata = true;
4217
4218 ModuleSlotTracker MST(getModuleFromVal(this), ShouldInitializeAllMetadata);
Justin Bognerd7d1a722015-09-27 22:38:50 +00004219 print(ROS, MST, IsForDebug);
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004220}
4221
Justin Bognerd7d1a722015-09-27 22:38:50 +00004222void Value::print(raw_ostream &ROS, ModuleSlotTracker &MST,
4223 bool IsForDebug) const {
Dan Gohman12dad632009-08-12 20:56:03 +00004224 formatted_raw_ostream OS(ROS);
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004225 SlotTracker EmptySlotTable(static_cast<const Module *>(nullptr));
4226 SlotTracker &SlotTable =
4227 MST.getMachine() ? *MST.getMachine() : EmptySlotTable;
4228 auto incorporateFunction = [&](const Function *F) {
4229 if (F)
4230 MST.incorporateFunction(*F);
4231 };
4232
Chris Lattner0c19df42008-08-23 22:23:09 +00004233 if (const Instruction *I = dyn_cast<Instruction>(this)) {
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004234 incorporateFunction(I->getParent() ? I->getParent()->getParent() : nullptr);
Justin Bognerd7d1a722015-09-27 22:38:50 +00004235 AssemblyWriter W(OS, SlotTable, getModuleFromVal(I), nullptr, IsForDebug);
Chris Lattnerefb5e392009-12-31 02:23:35 +00004236 W.printInstruction(*I);
Chris Lattner0c19df42008-08-23 22:23:09 +00004237 } else if (const BasicBlock *BB = dyn_cast<BasicBlock>(this)) {
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004238 incorporateFunction(BB->getParent());
Justin Bognerd7d1a722015-09-27 22:38:50 +00004239 AssemblyWriter W(OS, SlotTable, getModuleFromVal(BB), nullptr, IsForDebug);
Chris Lattnerefb5e392009-12-31 02:23:35 +00004240 W.printBasicBlock(BB);
Chris Lattner0c19df42008-08-23 22:23:09 +00004241 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(this)) {
Justin Bognerd7d1a722015-09-27 22:38:50 +00004242 AssemblyWriter W(OS, SlotTable, GV->getParent(), nullptr, IsForDebug);
Chris Lattnerefb5e392009-12-31 02:23:35 +00004243 if (const GlobalVariable *V = dyn_cast<GlobalVariable>(GV))
4244 W.printGlobal(V);
4245 else if (const Function *F = dyn_cast<Function>(GV))
4246 W.printFunction(F);
4247 else
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00004248 W.printIndirectSymbol(cast<GlobalIndirectSymbol>(GV));
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004249 } else if (const MetadataAsValue *V = dyn_cast<MetadataAsValue>(this)) {
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004250 V->getMetadata()->print(ROS, MST, getModuleFromVal(V));
Chris Lattner0c19df42008-08-23 22:23:09 +00004251 } else if (const Constant *C = dyn_cast<Constant>(this)) {
Chris Lattner92c5c122009-02-28 23:20:19 +00004252 TypePrinting TypePrinter;
Chris Lattnere101c442009-02-28 21:26:53 +00004253 TypePrinter.print(C->getType(), OS);
Chris Lattner2ee62d22009-02-28 21:11:05 +00004254 OS << ' ';
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004255 WriteConstantInternal(OS, C, TypePrinter, MST.getMachine(), nullptr);
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004256 } else if (isa<InlineAsm>(this) || isa<Argument>(this)) {
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004257 this->printAsOperand(OS, /* PrintType */ true, MST);
Chris Lattner0c19df42008-08-23 22:23:09 +00004258 } else {
Nick Lewyckyad1b3d12014-05-09 00:49:03 +00004259 llvm_unreachable("Unknown value to print out!");
Chris Lattner0c19df42008-08-23 22:23:09 +00004260 }
4261}
4262
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00004263/// Print without a type, skipping the TypePrinting object.
4264///
Benjamin Kramerdf005cb2015-08-08 18:27:36 +00004265/// \return \c true iff printing was successful.
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00004266static bool printWithoutType(const Value &V, raw_ostream &O,
4267 SlotTracker *Machine, const Module *M) {
4268 if (V.hasName() || isa<GlobalValue>(V) ||
4269 (!isa<Constant>(V) && !isa<MetadataAsValue>(V))) {
4270 WriteAsOperandInternal(O, &V, nullptr, Machine, M);
4271 return true;
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00004272 }
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00004273 return false;
4274}
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00004275
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00004276static void printAsOperandImpl(const Value &V, raw_ostream &O, bool PrintType,
4277 ModuleSlotTracker &MST) {
Roman Tereshind96de6f2018-03-22 21:29:07 +00004278 TypePrinting TypePrinter(MST.getModule());
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00004279 if (PrintType) {
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00004280 TypePrinter.print(V.getType(), O);
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00004281 O << ' ';
4282 }
4283
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +00004284 WriteAsOperandInternal(O, &V, &TypePrinter, MST.getMachine(),
4285 MST.getModule());
4286}
4287
4288void Value::printAsOperand(raw_ostream &O, bool PrintType,
4289 const Module *M) const {
4290 if (!M)
4291 M = getModuleFromVal(this);
4292
4293 if (!PrintType)
4294 if (printWithoutType(*this, O, nullptr, M))
4295 return;
4296
4297 SlotTracker Machine(
4298 M, /* ShouldInitializeAllMetadata */ isa<MetadataAsValue>(this));
4299 ModuleSlotTracker MST(Machine, M);
4300 printAsOperandImpl(*this, O, PrintType, MST);
4301}
4302
4303void Value::printAsOperand(raw_ostream &O, bool PrintType,
4304 ModuleSlotTracker &MST) const {
4305 if (!PrintType)
4306 if (printWithoutType(*this, O, MST.getMachine(), MST.getModule()))
4307 return;
4308
4309 printAsOperandImpl(*this, O, PrintType, MST);
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00004310}
4311
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004312static void printMetadataImpl(raw_ostream &ROS, const Metadata &MD,
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +00004313 ModuleSlotTracker &MST, const Module *M,
4314 bool OnlyAsOperand) {
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004315 formatted_raw_ostream OS(ROS);
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004316
Roman Tereshind96de6f2018-03-22 21:29:07 +00004317 TypePrinting TypePrinter(M);
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004318
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +00004319 WriteAsOperandInternal(OS, &MD, &TypePrinter, MST.getMachine(), M,
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004320 /* FromValue */ true);
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +00004321
4322 auto *N = dyn_cast<MDNode>(&MD);
Reid Klecknerb95b4272017-08-30 20:40:36 +00004323 if (OnlyAsOperand || !N || isa<DIExpression>(MD))
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004324 return;
4325
4326 OS << " = ";
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +00004327 WriteMDNodeBodyInternal(OS, N, &TypePrinter, MST.getMachine(), M);
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004328}
4329
4330void Metadata::printAsOperand(raw_ostream &OS, const Module *M) const {
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +00004331 ModuleSlotTracker MST(M, isa<MDNode>(this));
4332 printMetadataImpl(OS, *this, MST, M, /* OnlyAsOperand */ true);
4333}
4334
4335void Metadata::printAsOperand(raw_ostream &OS, ModuleSlotTracker &MST,
4336 const Module *M) const {
4337 printMetadataImpl(OS, *this, MST, M, /* OnlyAsOperand */ true);
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004338}
4339
Justin Bognerd7d1a722015-09-27 22:38:50 +00004340void Metadata::print(raw_ostream &OS, const Module *M,
4341 bool /*IsForDebug*/) const {
Duncan P. N. Exon Smith6529ed42015-06-26 22:28:47 +00004342 ModuleSlotTracker MST(M, isa<MDNode>(this));
4343 printMetadataImpl(OS, *this, MST, M, /* OnlyAsOperand */ false);
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004344}
4345
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004346void Metadata::print(raw_ostream &OS, ModuleSlotTracker &MST,
Justin Bognerd7d1a722015-09-27 22:38:50 +00004347 const Module *M, bool /*IsForDebug*/) const {
Duncan P. N. Exon Smith1f8a99a2015-06-27 00:38:26 +00004348 printMetadataImpl(OS, *this, MST, M, /* OnlyAsOperand */ false);
4349}
4350
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00004351void ModuleSummaryIndex::print(raw_ostream &ROS, bool IsForDebug) const {
4352 SlotTracker SlotTable(this);
4353 formatted_raw_ostream OS(ROS);
4354 AssemblyWriter W(OS, SlotTable, this, IsForDebug);
4355 W.printModuleSummaryIndex();
4356}
4357
Aaron Ballman615eb472017-10-15 14:32:27 +00004358#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
Chris Lattnerdab942552008-08-25 17:03:15 +00004359// Value::dump - allow easy printing of Values from the debugger.
Duncan P. N. Exon Smith89c1eaa2015-02-25 22:08:21 +00004360LLVM_DUMP_METHOD
Justin Bognerd7d1a722015-09-27 22:38:50 +00004361void Value::dump() const { print(dbgs(), /*IsForDebug=*/true); dbgs() << '\n'; }
Reid Spencer52641832004-05-25 18:14:38 +00004362
Chris Lattnerdab942552008-08-25 17:03:15 +00004363// Type::dump - allow easy printing of Types from the debugger.
Duncan P. N. Exon Smith89c1eaa2015-02-25 22:08:21 +00004364LLVM_DUMP_METHOD
Justin Bognerd7d1a722015-09-27 22:38:50 +00004365void Type::dump() const { print(dbgs(), /*IsForDebug=*/true); dbgs() << '\n'; }
Chris Lattner24025262009-02-28 21:05:51 +00004366
Chris Lattnerdab942552008-08-25 17:03:15 +00004367// Module::dump() - Allow printing of Modules from the debugger.
Duncan P. N. Exon Smith89c1eaa2015-02-25 22:08:21 +00004368LLVM_DUMP_METHOD
Justin Bognerd7d1a722015-09-27 22:38:50 +00004369void Module::dump() const {
4370 print(dbgs(), nullptr,
4371 /*ShouldPreserveUseListOrder=*/false, /*IsForDebug=*/true);
4372}
Bill Wendling3681d112011-12-09 23:18:34 +00004373
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00004374// Allow printing of Comdats from the debugger.
Duncan P. N. Exon Smith89c1eaa2015-02-25 22:08:21 +00004375LLVM_DUMP_METHOD
Justin Bognerd7d1a722015-09-27 22:38:50 +00004376void Comdat::dump() const { print(dbgs(), /*IsForDebug=*/true); }
David Majnemerdad0a642014-06-27 18:19:56 +00004377
Bill Wendling3681d112011-12-09 23:18:34 +00004378// NamedMDNode::dump() - Allow printing of NamedMDNodes from the debugger.
Duncan P. N. Exon Smith89c1eaa2015-02-25 22:08:21 +00004379LLVM_DUMP_METHOD
Justin Bognerd7d1a722015-09-27 22:38:50 +00004380void NamedMDNode::dump() const { print(dbgs(), /*IsForDebug=*/true); }
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004381
Duncan P. N. Exon Smith89c1eaa2015-02-25 22:08:21 +00004382LLVM_DUMP_METHOD
Duncan P. N. Exon Smitha66dc8f2015-03-15 06:53:32 +00004383void Metadata::dump() const { dump(nullptr); }
4384
4385LLVM_DUMP_METHOD
Duncan P. N. Exon Smithd6d70e72015-03-14 20:19:36 +00004386void Metadata::dump(const Module *M) const {
Justin Bognerd7d1a722015-09-27 22:38:50 +00004387 print(dbgs(), M, /*IsForDebug=*/true);
Duncan P. N. Exon Smith5bf8fef2014-12-09 18:38:53 +00004388 dbgs() << '\n';
4389}
Teresa Johnson08d5b4e2018-05-26 02:34:13 +00004390
4391// Allow printing of ModuleSummaryIndex from the debugger.
4392LLVM_DUMP_METHOD
4393void ModuleSummaryIndex::dump() const { print(dbgs(), /*IsForDebug=*/true); }
Matthias Braun8c209aa2017-01-28 02:02:38 +00004394#endif