1//===- AsmWriter.cpp - Printing LLVM as an assembly file ------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// 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.
12//
13// Note that these routines must be extremely tolerant of various errors in the
14// LLVM code, because it can be used for debugging transformations.
15//
16//===----------------------------------------------------------------------===//
17
18#include "LLVMContextImpl.h"
19#include "llvm/ADT/APFloat.h"
20#include "llvm/ADT/APInt.h"
21#include "llvm/ADT/ArrayRef.h"
22#include "llvm/ADT/DenseMap.h"
23#include "llvm/ADT/STLExtras.h"
24#include "llvm/ADT/STLFunctionalExtras.h"
25#include "llvm/ADT/SmallPtrSet.h"
26#include "llvm/ADT/SmallString.h"
27#include "llvm/ADT/SmallVector.h"
28#include "llvm/ADT/StringExtras.h"
29#include "llvm/ADT/StringRef.h"
30#include "llvm/ADT/iterator_range.h"
31#include "llvm/BinaryFormat/Dwarf.h"
32#include "llvm/Config/llvm-config.h"
33#include "llvm/IR/Argument.h"
34#include "llvm/IR/AssemblyAnnotationWriter.h"
35#include "llvm/IR/Attributes.h"
36#include "llvm/IR/BasicBlock.h"
37#include "llvm/IR/CFG.h"
38#include "llvm/IR/CallingConv.h"
39#include "llvm/IR/Comdat.h"
40#include "llvm/IR/Constant.h"
41#include "llvm/IR/Constants.h"
42#include "llvm/IR/DebugInfoMetadata.h"
43#include "llvm/IR/DebugProgramInstruction.h"
44#include "llvm/IR/DerivedTypes.h"
45#include "llvm/IR/Function.h"
46#include "llvm/IR/GlobalAlias.h"
47#include "llvm/IR/GlobalIFunc.h"
48#include "llvm/IR/GlobalObject.h"
49#include "llvm/IR/GlobalValue.h"
50#include "llvm/IR/GlobalVariable.h"
51#include "llvm/IR/IRPrintingPasses.h"
52#include "llvm/IR/InlineAsm.h"
53#include "llvm/IR/InstrTypes.h"
54#include "llvm/IR/Instruction.h"
55#include "llvm/IR/Instructions.h"
56#include "llvm/IR/IntrinsicInst.h"
57#include "llvm/IR/Intrinsics.h"
58#include "llvm/IR/LLVMContext.h"
59#include "llvm/IR/Metadata.h"
60#include "llvm/IR/Module.h"
61#include "llvm/IR/ModuleSlotTracker.h"
62#include "llvm/IR/ModuleSummaryIndex.h"
63#include "llvm/IR/Operator.h"
64#include "llvm/IR/Type.h"
65#include "llvm/IR/TypeFinder.h"
66#include "llvm/IR/TypedPointerType.h"
67#include "llvm/IR/Use.h"
68#include "llvm/IR/User.h"
69#include "llvm/IR/Value.h"
70#include "llvm/Support/AtomicOrdering.h"
71#include "llvm/Support/Casting.h"
72#include "llvm/Support/Compiler.h"
73#include "llvm/Support/Debug.h"
74#include "llvm/Support/ErrorHandling.h"
75#include "llvm/Support/FormattedStream.h"
76#include "llvm/Support/SaveAndRestore.h"
77#include "llvm/Support/raw_ostream.h"
78#include <cassert>
79#include <cctype>
80#include <cstddef>
81#include <cstdint>
82#include <iterator>
83#include <memory>
84#include <optional>
85#include <string>
86#include <tuple>
87#include <utility>
88#include <vector>
89
90using namespace llvm;
91
92// See https://llvm.org/docs/DebuggingLLVM.html for why these flags are useful.
93
94static cl::opt<bool>
95 PrintInstAddrs("print-inst-addrs", cl::Hidden,
96 cl::desc("Print addresses of instructions when dumping"));
97
98static cl::opt<bool> PrintInstDebugLocs(
99 "print-inst-debug-locs", cl::Hidden,
100 cl::desc("Pretty print debug locations of instructions when dumping"));
101
102static cl::opt<bool> PrintProfData(
103 "print-prof-data", cl::Hidden,
104 cl::desc("Pretty print perf data (branch weights, etc) when dumping"));
105
106static cl::opt<bool> PreserveAssemblyUseListOrder(
107 "preserve-ll-uselistorder", cl::Hidden, cl::init(Val: false),
108 cl::desc("Preserve use-list order when writing LLVM assembly."));
109
110static cl::opt<bool> PrintAddrspaceName("print-addrspace-name", cl::Hidden,
111 cl::init(Val: false),
112 cl::desc("Print address space names"));
113
114// Make virtual table appear in this compilation unit.
115AssemblyAnnotationWriter::~AssemblyAnnotationWriter() = default;
116
117//===----------------------------------------------------------------------===//
118// Helper Functions
119//===----------------------------------------------------------------------===//
120
121using OrderMap = MapVector<const Value *, unsigned>;
122
123using UseListOrderMap =
124 DenseMap<const Function *, MapVector<const Value *, std::vector<unsigned>>>;
125
126/// Look for a value that might be wrapped as metadata, e.g. a value in a
127/// metadata operand. Returns the input value as-is if it is not wrapped.
128static const Value *skipMetadataWrapper(const Value *V) {
129 if (const auto *MAV = dyn_cast<MetadataAsValue>(Val: V))
130 if (const auto *VAM = dyn_cast<ValueAsMetadata>(Val: MAV->getMetadata()))
131 return VAM->getValue();
132 return V;
133}
134
135static void orderValue(const Value *V, OrderMap &OM) {
136 if (OM.lookup(Key: V))
137 return;
138
139 if (const auto *C = dyn_cast<Constant>(Val: V)) {
140 if (isa<ConstantData>(Val: C))
141 return;
142
143 if (C->getNumOperands() && !isa<GlobalValue>(Val: C))
144 for (const Value *Op : C->operands())
145 if (!isa<BasicBlock>(Val: Op) && !isa<GlobalValue>(Val: Op))
146 orderValue(V: Op, OM);
147 }
148
149 // Note: we cannot cache this lookup above, since inserting into the map
150 // changes the map's size, and thus affects the other IDs.
151 unsigned ID = OM.size() + 1;
152 OM[V] = ID;
153}
154
155static OrderMap orderModule(const Module *M) {
156 OrderMap OM;
157
158 auto OrderConstantValue = [&OM](const Value *V) {
159 if (isa<Constant>(Val: V) || isa<InlineAsm>(Val: V))
160 orderValue(V, OM);
161 };
162
163 auto OrderConstantFromMetadata = [&](Metadata *MD) {
164 if (const auto *VAM = dyn_cast<ValueAsMetadata>(Val: MD)) {
165 OrderConstantValue(VAM->getValue());
166 } else if (const auto *AL = dyn_cast<DIArgList>(Val: MD)) {
167 for (const auto *VAM : AL->getArgs())
168 OrderConstantValue(VAM->getValue());
169 }
170 };
171
172 for (const GlobalVariable &G : M->globals()) {
173 if (G.hasInitializer())
174 if (!isa<GlobalValue>(Val: G.getInitializer()))
175 orderValue(V: G.getInitializer(), OM);
176 orderValue(V: &G, OM);
177 }
178 for (const GlobalAlias &A : M->aliases()) {
179 if (!isa<GlobalValue>(Val: A.getAliasee()))
180 orderValue(V: A.getAliasee(), OM);
181 orderValue(V: &A, OM);
182 }
183 for (const GlobalIFunc &I : M->ifuncs()) {
184 if (!isa<GlobalValue>(Val: I.getResolver()))
185 orderValue(V: I.getResolver(), OM);
186 orderValue(V: &I, OM);
187 }
188 for (const Function &F : *M) {
189 for (const Use &U : F.operands())
190 if (!isa<GlobalValue>(Val: U.get()))
191 orderValue(V: U.get(), OM);
192
193 orderValue(V: &F, OM);
194
195 if (F.isDeclaration())
196 continue;
197
198 for (const Argument &A : F.args())
199 orderValue(V: &A, OM);
200 for (const BasicBlock &BB : F) {
201 orderValue(V: &BB, OM);
202 for (const Instruction &I : BB) {
203 // Debug records can contain Value references, that can then contain
204 // Values disconnected from the rest of the Value hierachy, if wrapped
205 // in some kind of constant-expression. Find and order any Values that
206 // are wrapped in debug-info.
207 for (DbgVariableRecord &DVR : filterDbgVars(R: I.getDbgRecordRange())) {
208 OrderConstantFromMetadata(DVR.getRawLocation());
209 if (DVR.isDbgAssign())
210 OrderConstantFromMetadata(DVR.getRawAddress());
211 }
212
213 for (const Value *Op : I.operands()) {
214 Op = skipMetadataWrapper(V: Op);
215 if ((isa<Constant>(Val: *Op) && !isa<GlobalValue>(Val: *Op)) ||
216 isa<InlineAsm>(Val: *Op))
217 orderValue(V: Op, OM);
218 }
219 orderValue(V: &I, OM);
220 }
221 }
222 }
223 return OM;
224}
225
226static std::vector<unsigned>
227predictValueUseListOrder(const Value *V, unsigned ID, const OrderMap &OM) {
228 // Predict use-list order for this one.
229 using Entry = std::pair<const Use *, unsigned>;
230 SmallVector<Entry, 64> List;
231 for (const Use &U : V->uses())
232 // Check if this user will be serialized.
233 if (OM.lookup(Key: U.getUser()))
234 List.push_back(Elt: std::make_pair(x: &U, y: List.size()));
235
236 if (List.size() < 2)
237 // We may have lost some users.
238 return {};
239
240 // When referencing a value before its declaration, a temporary value is
241 // created, which will later be RAUWed with the actual value. This reverses
242 // the use list. This happens for all values apart from basic blocks.
243 bool GetsReversed = !isa<BasicBlock>(Val: V);
244 if (auto *BA = dyn_cast<BlockAddress>(Val: V))
245 ID = OM.lookup(Key: BA->getBasicBlock());
246 llvm::sort(C&: List, Comp: [&](const Entry &L, const Entry &R) {
247 const Use *LU = L.first;
248 const Use *RU = R.first;
249 if (LU == RU)
250 return false;
251
252 auto LID = OM.lookup(Key: LU->getUser());
253 auto RID = OM.lookup(Key: RU->getUser());
254
255 // If ID is 4, then expect: 7 6 5 1 2 3.
256 if (LID < RID) {
257 if (GetsReversed)
258 if (RID <= ID)
259 return true;
260 return false;
261 }
262 if (RID < LID) {
263 if (GetsReversed)
264 if (LID <= ID)
265 return false;
266 return true;
267 }
268
269 // LID and RID are equal, so we have different operands of the same user.
270 // Assume operands are added in order for all instructions.
271 if (GetsReversed)
272 if (LID <= ID)
273 return LU->getOperandNo() < RU->getOperandNo();
274 return LU->getOperandNo() > RU->getOperandNo();
275 });
276
277 if (llvm::is_sorted(Range&: List, C: llvm::less_second()))
278 // Order is already correct.
279 return {};
280
281 // Store the shuffle.
282 std::vector<unsigned> Shuffle(List.size());
283 for (size_t I = 0, E = List.size(); I != E; ++I)
284 Shuffle[I] = List[I].second;
285 return Shuffle;
286}
287
288static UseListOrderMap predictUseListOrder(const Module *M) {
289 OrderMap OM = orderModule(M);
290 UseListOrderMap ULOM;
291 for (const auto &Pair : OM) {
292 const Value *V = Pair.first;
293 if (V->use_empty() || std::next(x: V->use_begin()) == V->use_end())
294 continue;
295
296 std::vector<unsigned> Shuffle =
297 predictValueUseListOrder(V, ID: Pair.second, OM);
298 if (Shuffle.empty())
299 continue;
300
301 const Function *F = nullptr;
302 if (auto *I = dyn_cast<Instruction>(Val: V))
303 F = I->getFunction();
304 if (auto *A = dyn_cast<Argument>(Val: V))
305 F = A->getParent();
306 if (auto *BB = dyn_cast<BasicBlock>(Val: V))
307 F = BB->getParent();
308 ULOM[F][V] = std::move(Shuffle);
309 }
310 return ULOM;
311}
312
313static const Module *getModuleFromVal(const Value *V) {
314 if (const auto *MA = dyn_cast<Argument>(Val: V))
315 return MA->getParent() ? MA->getParent()->getParent() : nullptr;
316
317 if (const auto *BB = dyn_cast<BasicBlock>(Val: V))
318 return BB->getParent() ? BB->getParent()->getParent() : nullptr;
319
320 if (const auto *I = dyn_cast<Instruction>(Val: V)) {
321 const Function *M = I->getParent() ? I->getParent()->getParent() : nullptr;
322 return M ? M->getParent() : nullptr;
323 }
324
325 if (const auto *GV = dyn_cast<GlobalValue>(Val: V))
326 return GV->getParent();
327
328 if (const auto *MAV = dyn_cast<MetadataAsValue>(Val: V)) {
329 for (const User *U : MAV->users())
330 if (isa<Instruction>(Val: U))
331 if (const Module *M = getModuleFromVal(V: U))
332 return M;
333 return nullptr;
334 }
335
336 return nullptr;
337}
338
339static const Module *getModuleFromDPI(const DbgMarker *Marker) {
340 const Function *M =
341 Marker->getParent() ? Marker->getParent()->getParent() : nullptr;
342 return M ? M->getParent() : nullptr;
343}
344
345static const Module *getModuleFromDPI(const DbgRecord *DR) {
346 return DR->getMarker() ? getModuleFromDPI(Marker: DR->getMarker()) : nullptr;
347}
348
349static void printCallingConv(unsigned cc, raw_ostream &Out) {
350 switch (cc) {
351 default: Out << "cc" << cc; break;
352 case CallingConv::Fast: Out << "fastcc"; break;
353 case CallingConv::Cold: Out << "coldcc"; break;
354 case CallingConv::AnyReg: Out << "anyregcc"; break;
355 case CallingConv::PreserveMost: Out << "preserve_mostcc"; break;
356 case CallingConv::PreserveAll: Out << "preserve_allcc"; break;
357 case CallingConv::PreserveNone: Out << "preserve_nonecc"; break;
358 case CallingConv::CXX_FAST_TLS: Out << "cxx_fast_tlscc"; break;
359 case CallingConv::GHC: Out << "ghccc"; break;
360 case CallingConv::Tail: Out << "tailcc"; break;
361 case CallingConv::GRAAL: Out << "graalcc"; break;
362 case CallingConv::CFGuard_Check: Out << "cfguard_checkcc"; break;
363 case CallingConv::X86_StdCall: Out << "x86_stdcallcc"; break;
364 case CallingConv::X86_FastCall: Out << "x86_fastcallcc"; break;
365 case CallingConv::X86_ThisCall: Out << "x86_thiscallcc"; break;
366 case CallingConv::X86_RegCall: Out << "x86_regcallcc"; break;
367 case CallingConv::X86_VectorCall:Out << "x86_vectorcallcc"; break;
368 case CallingConv::Intel_OCL_BI: Out << "intel_ocl_bicc"; break;
369 case CallingConv::ARM_APCS: Out << "arm_apcscc"; break;
370 case CallingConv::ARM_AAPCS: Out << "arm_aapcscc"; break;
371 case CallingConv::ARM_AAPCS_VFP: Out << "arm_aapcs_vfpcc"; break;
372 case CallingConv::AArch64_VectorCall: Out << "aarch64_vector_pcs"; break;
373 case CallingConv::AArch64_SVE_VectorCall:
374 Out << "aarch64_sve_vector_pcs";
375 break;
376 case CallingConv::AArch64_SME_ABI_Support_Routines_PreserveMost_From_X0:
377 Out << "aarch64_sme_preservemost_from_x0";
378 break;
379 case CallingConv::AArch64_SME_ABI_Support_Routines_PreserveMost_From_X1:
380 Out << "aarch64_sme_preservemost_from_x1";
381 break;
382 case CallingConv::AArch64_SME_ABI_Support_Routines_PreserveMost_From_X2:
383 Out << "aarch64_sme_preservemost_from_x2";
384 break;
385 case CallingConv::MSP430_INTR: Out << "msp430_intrcc"; break;
386 case CallingConv::AVR_INTR: Out << "avr_intrcc "; break;
387 case CallingConv::AVR_SIGNAL: Out << "avr_signalcc "; break;
388 case CallingConv::PTX_Kernel: Out << "ptx_kernel"; break;
389 case CallingConv::PTX_Device: Out << "ptx_device"; break;
390 case CallingConv::X86_64_SysV: Out << "x86_64_sysvcc"; break;
391 case CallingConv::Win64: Out << "win64cc"; break;
392 case CallingConv::SPIR_FUNC: Out << "spir_func"; break;
393 case CallingConv::SPIR_KERNEL: Out << "spir_kernel"; break;
394 case CallingConv::Swift: Out << "swiftcc"; break;
395 case CallingConv::SwiftTail: Out << "swifttailcc"; break;
396 case CallingConv::X86_INTR: Out << "x86_intrcc"; break;
397 case CallingConv::DUMMY_HHVM:
398 Out << "hhvmcc";
399 break;
400 case CallingConv::DUMMY_HHVM_C:
401 Out << "hhvm_ccc";
402 break;
403 case CallingConv::AMDGPU_VS: Out << "amdgpu_vs"; break;
404 case CallingConv::AMDGPU_LS: Out << "amdgpu_ls"; break;
405 case CallingConv::AMDGPU_HS: Out << "amdgpu_hs"; break;
406 case CallingConv::AMDGPU_ES: Out << "amdgpu_es"; break;
407 case CallingConv::AMDGPU_GS: Out << "amdgpu_gs"; break;
408 case CallingConv::AMDGPU_PS: Out << "amdgpu_ps"; break;
409 case CallingConv::AMDGPU_CS: Out << "amdgpu_cs"; break;
410 case CallingConv::AMDGPU_CS_Chain:
411 Out << "amdgpu_cs_chain";
412 break;
413 case CallingConv::AMDGPU_CS_ChainPreserve:
414 Out << "amdgpu_cs_chain_preserve";
415 break;
416 case CallingConv::AMDGPU_KERNEL: Out << "amdgpu_kernel"; break;
417 case CallingConv::AMDGPU_Gfx: Out << "amdgpu_gfx"; break;
418 case CallingConv::AMDGPU_Gfx_WholeWave:
419 Out << "amdgpu_gfx_whole_wave";
420 break;
421 case CallingConv::M68k_RTD: Out << "m68k_rtdcc"; break;
422 case CallingConv::RISCV_VectorCall:
423 Out << "riscv_vector_cc";
424 break;
425#define CC_VLS_CASE(ABI_VLEN) \
426 case CallingConv::RISCV_VLSCall_##ABI_VLEN: \
427 Out << "riscv_vls_cc(" #ABI_VLEN ")"; \
428 break;
429 CC_VLS_CASE(32)
430 CC_VLS_CASE(64)
431 CC_VLS_CASE(128)
432 CC_VLS_CASE(256)
433 CC_VLS_CASE(512)
434 CC_VLS_CASE(1024)
435 CC_VLS_CASE(2048)
436 CC_VLS_CASE(4096)
437 CC_VLS_CASE(8192)
438 CC_VLS_CASE(16384)
439 CC_VLS_CASE(32768)
440 CC_VLS_CASE(65536)
441#undef CC_VLS_CASE
442 case CallingConv::CHERIoT_CompartmentCall:
443 Out << "cheriot_compartmentcallcc";
444 break;
445 case CallingConv::CHERIoT_CompartmentCallee:
446 Out << "cheriot_compartmentcalleecc";
447 break;
448 case CallingConv::CHERIoT_LibraryCall:
449 Out << "cheriot_librarycallcc";
450 break;
451 }
452}
453
454enum PrefixType {
455 GlobalPrefix,
456 ComdatPrefix,
457 LabelPrefix,
458 LocalPrefix,
459 NoPrefix
460};
461
462void llvm::printLLVMNameWithoutPrefix(raw_ostream &OS, StringRef Name) {
463 assert(!Name.empty() && "Cannot get empty name!");
464
465 // Scan the name to see if it needs quotes first.
466 bool NeedsQuotes = isdigit(static_cast<unsigned char>(Name[0]));
467 if (!NeedsQuotes) {
468 for (unsigned char C : Name) {
469 // By making this unsigned, the value passed in to isalnum will always be
470 // in the range 0-255. This is important when building with MSVC because
471 // its implementation will assert. This situation can arise when dealing
472 // with UTF-8 multibyte characters.
473 if (!isalnum(C) && C != '-' && C != '.' && C != '_') {
474 NeedsQuotes = true;
475 break;
476 }
477 }
478 }
479
480 // If we didn't need any quotes, just write out the name in one blast.
481 if (!NeedsQuotes) {
482 OS << Name;
483 return;
484 }
485
486 // Okay, we need quotes. Output the quotes and escape any scary characters as
487 // needed.
488 OS << '"';
489 printEscapedString(Name, Out&: OS);
490 OS << '"';
491}
492
493/// Turn the specified name into an 'LLVM name', which is either prefixed with %
494/// (if the string only contains simple characters) or is surrounded with ""'s
495/// (if it has special chars in it). Print it out.
496static void printLLVMName(raw_ostream &OS, StringRef Name, PrefixType Prefix) {
497 switch (Prefix) {
498 case NoPrefix:
499 break;
500 case GlobalPrefix:
501 OS << '@';
502 break;
503 case ComdatPrefix:
504 OS << '$';
505 break;
506 case LabelPrefix:
507 break;
508 case LocalPrefix:
509 OS << '%';
510 break;
511 }
512 printLLVMNameWithoutPrefix(OS, Name);
513}
514
515/// Turn the specified name into an 'LLVM name', which is either prefixed with %
516/// (if the string only contains simple characters) or is surrounded with ""'s
517/// (if it has special chars in it). Print it out.
518static void printLLVMName(raw_ostream &OS, const Value *V) {
519 printLLVMName(OS, Name: V->getName(),
520 Prefix: isa<GlobalValue>(Val: V) ? GlobalPrefix : LocalPrefix);
521}
522
523static void printShuffleMask(raw_ostream &Out, Type *Ty, ArrayRef<int> Mask) {
524 Out << ", <";
525 if (isa<ScalableVectorType>(Val: Ty))
526 Out << "vscale x ";
527 Out << Mask.size() << " x i32> ";
528 if (all_of(Range&: Mask, P: equal_to(Arg: 0))) {
529 Out << "zeroinitializer";
530 } else if (all_of(Range&: Mask, P: equal_to(Arg: PoisonMaskElem))) {
531 Out << "poison";
532 } else {
533 Out << "<";
534 ListSeparator LS;
535 for (int Elt : Mask) {
536 Out << LS << "i32 ";
537 if (Elt == PoisonMaskElem)
538 Out << "poison";
539 else
540 Out << Elt;
541 }
542 Out << ">";
543 }
544}
545
546namespace {
547
548class TypePrinting {
549public:
550 TypePrinting(const Module *M = nullptr)
551 : M(M), TypesIncorporated(M == nullptr) {}
552
553 TypePrinting(const TypePrinting &) = delete;
554 TypePrinting &operator=(const TypePrinting &) = delete;
555
556 /// The named types that are used by the current module.
557 TypeFinder &getNamedTypes();
558
559 /// The numbered types, number to type mapping.
560 std::vector<StructType *> &getNumberedTypes();
561
562 bool empty();
563
564 void print(Type *Ty, raw_ostream &OS);
565
566 void printStructBody(StructType *Ty, raw_ostream &OS);
567
568private:
569 void incorporateTypes();
570
571 /// A module to process lazily.
572 const Module *M;
573 bool TypesIncorporated;
574
575 TypeFinder NamedTypes;
576
577 // The numbered types, along with their value.
578 DenseMap<StructType *, unsigned> Type2Number;
579
580 std::vector<StructType *> NumberedTypes;
581};
582
583} // end anonymous namespace
584
585TypeFinder &TypePrinting::getNamedTypes() {
586 incorporateTypes();
587 return NamedTypes;
588}
589
590std::vector<StructType *> &TypePrinting::getNumberedTypes() {
591 incorporateTypes();
592
593 // We know all the numbers that each type is used and we know that it is a
594 // dense assignment. Convert the map to an index table, if it's not done
595 // already (judging from the sizes):
596 if (NumberedTypes.size() == Type2Number.size())
597 return NumberedTypes;
598
599 NumberedTypes.resize(new_size: Type2Number.size());
600 for (const auto &P : Type2Number) {
601 assert(P.second < NumberedTypes.size() && "Didn't get a dense numbering?");
602 assert(!NumberedTypes[P.second] && "Didn't get a unique numbering?");
603 NumberedTypes[P.second] = P.first;
604 }
605 return NumberedTypes;
606}
607
608bool TypePrinting::empty() {
609 incorporateTypes();
610 return NamedTypes.empty() && Type2Number.empty();
611}
612
613void TypePrinting::incorporateTypes() {
614 if (TypesIncorporated)
615 return;
616
617 NamedTypes.run(M: *M, onlyNamed: false);
618 TypesIncorporated = true;
619
620 // The list of struct types we got back includes all the struct types, split
621 // the unnamed ones out to a numbering and remove the anonymous structs.
622 unsigned NextNumber = 0;
623
624 std::vector<StructType *>::iterator NextToUse = NamedTypes.begin();
625 for (StructType *STy : NamedTypes) {
626 // Ignore anonymous types.
627 if (STy->isLiteral())
628 continue;
629
630 if (STy->getName().empty())
631 Type2Number[STy] = NextNumber++;
632 else
633 *NextToUse++ = STy;
634 }
635
636 NamedTypes.erase(I: NextToUse, E: NamedTypes.end());
637}
638
639static void printAddressSpace(const Module *M, unsigned AS, raw_ostream &OS,
640 StringRef Prefix = " ", StringRef Suffix = "",
641 bool ForcePrint = false) {
642 if (AS == 0 && !ForcePrint)
643 return;
644 OS << Prefix << "addrspace(";
645 StringRef ASName =
646 PrintAddrspaceName && M ? M->getDataLayout().getAddressSpaceName(AS) : "";
647 if (!ASName.empty())
648 OS << "\"" << ASName << "\"";
649 else
650 OS << AS;
651 OS << ")" << Suffix;
652}
653
654/// Write the specified type to the specified raw_ostream, making use of type
655/// names or up references to shorten the type name where possible.
656void TypePrinting::print(Type *Ty, raw_ostream &OS) {
657 switch (Ty->getTypeID()) {
658 case Type::VoidTyID: OS << "void"; return;
659 case Type::HalfTyID: OS << "half"; return;
660 case Type::BFloatTyID: OS << "bfloat"; return;
661 case Type::FloatTyID: OS << "float"; return;
662 case Type::DoubleTyID: OS << "double"; return;
663 case Type::X86_FP80TyID: OS << "x86_fp80"; return;
664 case Type::FP128TyID: OS << "fp128"; return;
665 case Type::PPC_FP128TyID: OS << "ppc_fp128"; return;
666 case Type::LabelTyID: OS << "label"; return;
667 case Type::MetadataTyID:
668 OS << "metadata";
669 return;
670 case Type::X86_AMXTyID: OS << "x86_amx"; return;
671 case Type::TokenTyID: OS << "token"; return;
672 case Type::ByteTyID:
673 OS << 'b' << Ty->getByteBitWidth();
674 return;
675 case Type::IntegerTyID:
676 OS << 'i' << cast<IntegerType>(Val: Ty)->getBitWidth();
677 return;
678
679 case Type::FunctionTyID: {
680 FunctionType *FTy = cast<FunctionType>(Val: Ty);
681 print(Ty: FTy->getReturnType(), OS);
682 OS << " (";
683 ListSeparator LS;
684 for (Type *Ty : FTy->params()) {
685 OS << LS;
686 print(Ty, OS);
687 }
688 if (FTy->isVarArg())
689 OS << LS << "...";
690 OS << ')';
691 return;
692 }
693 case Type::StructTyID: {
694 StructType *STy = cast<StructType>(Val: Ty);
695
696 if (STy->isLiteral())
697 return printStructBody(Ty: STy, OS);
698
699 if (!STy->getName().empty())
700 return printLLVMName(OS, Name: STy->getName(), Prefix: LocalPrefix);
701
702 incorporateTypes();
703 const auto I = Type2Number.find(Val: STy);
704 if (I != Type2Number.end())
705 OS << '%' << I->second;
706 else // Not enumerated, print the hex address.
707 OS << "%\"type " << STy << '\"';
708 return;
709 }
710 case Type::PointerTyID: {
711 PointerType *PTy = cast<PointerType>(Val: Ty);
712 OS << "ptr";
713 printAddressSpace(M, AS: PTy->getAddressSpace(), OS);
714 return;
715 }
716 case Type::ArrayTyID: {
717 ArrayType *ATy = cast<ArrayType>(Val: Ty);
718 OS << '[' << ATy->getNumElements() << " x ";
719 print(Ty: ATy->getElementType(), OS);
720 OS << ']';
721 return;
722 }
723 case Type::FixedVectorTyID:
724 case Type::ScalableVectorTyID: {
725 VectorType *PTy = cast<VectorType>(Val: Ty);
726 ElementCount EC = PTy->getElementCount();
727 OS << "<";
728 if (EC.isScalable())
729 OS << "vscale x ";
730 OS << EC.getKnownMinValue() << " x ";
731 print(Ty: PTy->getElementType(), OS);
732 OS << '>';
733 return;
734 }
735 case Type::TypedPointerTyID: {
736 TypedPointerType *TPTy = cast<TypedPointerType>(Val: Ty);
737 OS << "typedptr(" << *TPTy->getElementType() << ", "
738 << TPTy->getAddressSpace() << ")";
739 return;
740 }
741 case Type::TargetExtTyID:
742 TargetExtType *TETy = cast<TargetExtType>(Val: Ty);
743 OS << "target(\"";
744 printEscapedString(Name: Ty->getTargetExtName(), Out&: OS);
745 OS << "\"";
746 for (Type *Inner : TETy->type_params()) {
747 OS << ", ";
748 Inner->print(O&: OS, /*IsForDebug=*/false, /*NoDetails=*/true);
749 }
750 for (unsigned IntParam : TETy->int_params())
751 OS << ", " << IntParam;
752 OS << ")";
753 return;
754 }
755 llvm_unreachable("Invalid TypeID");
756}
757
758void TypePrinting::printStructBody(StructType *STy, raw_ostream &OS) {
759 if (STy->isOpaque()) {
760 OS << "opaque";
761 return;
762 }
763
764 if (STy->isPacked())
765 OS << '<';
766
767 if (STy->getNumElements() == 0) {
768 OS << "{}";
769 } else {
770 OS << "{ ";
771 ListSeparator LS;
772 for (Type *Ty : STy->elements()) {
773 OS << LS;
774 print(Ty, OS);
775 }
776
777 OS << " }";
778 }
779 if (STy->isPacked())
780 OS << '>';
781}
782
783AbstractSlotTrackerStorage::~AbstractSlotTrackerStorage() = default;
784
785//===----------------------------------------------------------------------===//
786// SlotTracker Class: Enumerate slot numbers for unnamed values
787//===----------------------------------------------------------------------===//
788/// This class provides computation of slot numbers for LLVM Assembly writing.
789///
790class llvm::SlotTracker : public AbstractSlotTrackerStorage {
791public:
792 /// ValueMap - A mapping of Values to slot numbers.
793 using ValueMap = DenseMap<const Value *, unsigned>;
794
795private:
796 /// TheModule - The module for which we are holding slot numbers.
797 const Module* TheModule;
798
799 /// TheFunction - The function for which we are holding slot numbers.
800 const Function* TheFunction = nullptr;
801 bool FunctionProcessed = false;
802 bool ShouldTrackMetadataDefinitions;
803
804 std::function<void(AbstractSlotTrackerStorage *, const Module *)>
805 ProcessModuleHookFn;
806 std::function<void(AbstractSlotTrackerStorage *, const Function *)>
807 ProcessFunctionHookFn;
808
809 /// The summary index for which we are holding slot numbers.
810 const ModuleSummaryIndex *TheIndex = nullptr;
811
812 /// mMap - The slot map for the module level data.
813 ValueMap mMap;
814 unsigned mNext = 0;
815
816 /// fMap - The slot map for the function level data.
817 ValueMap fMap;
818 unsigned fNext = 0;
819
820 /// mdnMap - Map for MDNodes.
821 DenseMap<const MDNode *, unsigned> mdnMap;
822 /// asMap - The slot map for attribute sets.
823 DenseMap<AttributeSet, unsigned> asMap;
824 unsigned asNext = 0;
825
826 /// ModulePathMap - The slot map for Module paths used in the summary index.
827 StringMap<unsigned> ModulePathMap;
828 unsigned ModulePathNext = 0;
829
830 /// GUIDMap - The slot map for GUIDs used in the summary index.
831 DenseMap<GlobalValue::GUID, unsigned> GUIDMap;
832 unsigned GUIDNext = 0;
833
834 /// TypeIdMap - The slot map for type ids used in the summary index.
835 StringMap<unsigned> TypeIdMap;
836 unsigned TypeIdNext = 0;
837
838 /// TypeIdCompatibleVtableMap - The slot map for type compatible vtable ids
839 /// used in the summary index.
840 StringMap<unsigned> TypeIdCompatibleVtableMap;
841 unsigned TypeIdCompatibleVtableNext = 0;
842
843public:
844 /// Construct from a module.
845 ///
846 explicit SlotTracker(const Module *M,
847 bool ShouldTrackMetadataDefinitions = false);
848
849 /// Construct from a function, starting out in incorp state.
850 ///
851 explicit SlotTracker(const Function *F);
852
853 /// Construct from a module summary index.
854 explicit SlotTracker(const ModuleSummaryIndex *Index);
855
856 SlotTracker(const SlotTracker &) = delete;
857 SlotTracker &operator=(const SlotTracker &) = delete;
858
859 ~SlotTracker() override = default;
860
861 void setProcessHook(
862 std::function<void(AbstractSlotTrackerStorage *, const Module *)>);
863 void setProcessHook(
864 std::function<void(AbstractSlotTrackerStorage *, const Function *)>);
865
866 void createMetadataSlot(const MDNode *N) override;
867
868 /// Return the slot number of the specified value in it's type
869 /// plane. If something is not in the SlotTracker, return -1.
870 int getLocalSlot(const Value *V);
871 int getGlobalSlot(const GlobalValue *V);
872 int getMetadataSlot(const MDNode *N) override;
873 int getAttributeGroupSlot(AttributeSet AS);
874 int getModulePathSlot(StringRef Path);
875 int getGUIDSlot(GlobalValue::GUID GUID);
876 int getTypeIdSlot(StringRef Id);
877 int getTypeIdCompatibleVtableSlot(StringRef Id);
878
879 /// If you'd like to deal with a function instead of just a module, use
880 /// this method to get its data into the SlotTracker.
881 void incorporateFunction(const Function *F) {
882 TheFunction = F;
883 FunctionProcessed = false;
884 }
885
886 const Function *getFunction() const { return TheFunction; }
887
888 /// After calling incorporateFunction, use this method to remove the
889 /// most recently incorporated function from the SlotTracker. This
890 /// will reset the state of the machine back to just the module contents.
891 void purgeFunction();
892
893 /// MDNode map iterators.
894 using mdn_iterator = DenseMap<const MDNode*, unsigned>::iterator;
895
896 mdn_iterator mdn_begin() { return mdnMap.begin(); }
897 mdn_iterator mdn_end() { return mdnMap.end(); }
898 unsigned mdn_size() const { return mdnMap.size(); }
899 bool mdn_empty() const { return mdnMap.empty(); }
900
901 /// AttributeSet map iterators.
902 using as_iterator = DenseMap<AttributeSet, unsigned>::iterator;
903
904 as_iterator as_begin() { return asMap.begin(); }
905 as_iterator as_end() { return asMap.end(); }
906 unsigned as_size() const { return asMap.size(); }
907 bool as_empty() const { return asMap.empty(); }
908
909 /// GUID map iterators.
910 using guid_iterator = DenseMap<GlobalValue::GUID, unsigned>::iterator;
911
912 /// These functions do the actual initialization.
913 inline void initializeIfNeeded();
914 int initializeIndexIfNeeded();
915
916 // Implementation Details
917private:
918 /// CreateModuleSlot - Insert the specified GlobalValue* into the slot table.
919 void CreateModuleSlot(const GlobalValue *V);
920
921 /// Record a metadata definition and the metadata nodes referenced by it.
922 void CreateMetadataSlot(const MDNode *N);
923
924 /// CreateFunctionSlot - Insert the specified Value* into the slot table.
925 void CreateFunctionSlot(const Value *V);
926
927 /// Insert the specified AttributeSet into the slot table.
928 void CreateAttributeSetSlot(AttributeSet AS);
929
930 inline void CreateModulePathSlot(StringRef Path);
931 void CreateGUIDSlot(GlobalValue::GUID GUID);
932 void CreateTypeIdSlot(StringRef Id);
933 void CreateTypeIdCompatibleVtableSlot(StringRef Id);
934
935 /// Add all of the module level global variables (and their initializers)
936 /// and function declarations, but not the contents of those functions.
937 void processModule();
938 // Returns number of allocated slots
939 int processIndex();
940
941 /// Add all of the functions arguments, basic blocks, and instructions.
942 void processFunction();
943};
944
945ModuleSlotTracker::ModuleSlotTracker(SlotTracker &Machine, const Module *M,
946 const Function *F)
947 : M(M), F(F), Machine(&Machine) {}
948
949ModuleSlotTracker::ModuleSlotTracker(const Module *M)
950 : ShouldCreateStorage(M), M(M) {}
951
952ModuleSlotTracker::~ModuleSlotTracker() = default;
953
954SlotTracker *ModuleSlotTracker::getMachine() {
955 if (!ShouldCreateStorage)
956 return Machine;
957
958 ShouldCreateStorage = false;
959 MachineStorage = std::make_unique<SlotTracker>(args&: M);
960 Machine = MachineStorage.get();
961 if (ProcessModuleHookFn)
962 Machine->setProcessHook(ProcessModuleHookFn);
963 if (ProcessFunctionHookFn)
964 Machine->setProcessHook(ProcessFunctionHookFn);
965 return Machine;
966}
967
968void ModuleSlotTracker::incorporateFunction(const Function &F) {
969 // Using getMachine() may lazily create the slot tracker.
970 if (!getMachine())
971 return;
972
973 // Nothing to do if this is the right function already.
974 if (this->F == &F)
975 return;
976 if (this->F)
977 Machine->purgeFunction();
978 Machine->incorporateFunction(F: &F);
979 this->F = &F;
980}
981
982int ModuleSlotTracker::getLocalSlot(const Value *V) {
983 assert(F && "No function incorporated");
984 return Machine->getLocalSlot(V);
985}
986
987void ModuleSlotTracker::setProcessHook(
988 std::function<void(AbstractSlotTrackerStorage *, const Module *)> Fn) {
989 ProcessModuleHookFn = std::move(Fn);
990}
991
992void ModuleSlotTracker::setProcessHook(
993 std::function<void(AbstractSlotTrackerStorage *, const Function *)> Fn) {
994 ProcessFunctionHookFn = std::move(Fn);
995}
996
997static SlotTracker *createSlotTracker(const Value *V) {
998 if (const auto *FA = dyn_cast<Argument>(Val: V))
999 return new SlotTracker(FA->getParent());
1000
1001 if (const auto *I = dyn_cast<Instruction>(Val: V))
1002 if (I->getParent())
1003 return new SlotTracker(I->getParent()->getParent());
1004
1005 if (const auto *BB = dyn_cast<BasicBlock>(Val: V))
1006 return new SlotTracker(BB->getParent());
1007
1008 if (const auto *GV = dyn_cast<GlobalVariable>(Val: V))
1009 return new SlotTracker(GV->getParent());
1010
1011 if (const auto *GA = dyn_cast<GlobalAlias>(Val: V))
1012 return new SlotTracker(GA->getParent());
1013
1014 if (const auto *GIF = dyn_cast<GlobalIFunc>(Val: V))
1015 return new SlotTracker(GIF->getParent());
1016
1017 if (const auto *Func = dyn_cast<Function>(Val: V))
1018 return new SlotTracker(Func);
1019
1020 return nullptr;
1021}
1022
1023#if 0
1024#define ST_DEBUG(X) dbgs() << X
1025#else
1026#define ST_DEBUG(X)
1027#endif
1028
1029// Module level constructor. Causes the contents of the Module (sans functions)
1030// to be added to the slot table.
1031SlotTracker::SlotTracker(const Module *M, bool ShouldTrackMetadataDefinitions)
1032 : TheModule(M),
1033 ShouldTrackMetadataDefinitions(ShouldTrackMetadataDefinitions) {}
1034
1035// Function level constructor. Causes the contents of the Module and the one
1036// function provided to be added to the slot table.
1037SlotTracker::SlotTracker(const Function *F)
1038 : TheModule(F ? F->getParent() : nullptr), TheFunction(F),
1039 ShouldTrackMetadataDefinitions(false) {}
1040
1041SlotTracker::SlotTracker(const ModuleSummaryIndex *Index)
1042 : TheModule(nullptr), ShouldTrackMetadataDefinitions(false),
1043 TheIndex(Index) {}
1044
1045inline void SlotTracker::initializeIfNeeded() {
1046 if (TheModule) {
1047 processModule();
1048 TheModule = nullptr; ///< Prevent re-processing next time we're called.
1049 }
1050
1051 if (TheFunction && !FunctionProcessed)
1052 processFunction();
1053}
1054
1055int SlotTracker::initializeIndexIfNeeded() {
1056 if (!TheIndex)
1057 return 0;
1058 int NumSlots = processIndex();
1059 TheIndex = nullptr; ///< Prevent re-processing next time we're called.
1060 return NumSlots;
1061}
1062
1063// Iterate through all the global variables, functions, and global
1064// variable initializers and create slots for them.
1065void SlotTracker::processModule() {
1066 ST_DEBUG("begin processModule!\n");
1067
1068 // Add all of the unnamed global variables to the value table.
1069 for (const GlobalVariable &Var : TheModule->globals()) {
1070 if (!Var.hasName())
1071 CreateModuleSlot(V: &Var);
1072 auto Attrs = Var.getAttributes();
1073 if (Attrs.hasAttributes())
1074 CreateAttributeSetSlot(AS: Attrs);
1075 }
1076
1077 for (const GlobalAlias &A : TheModule->aliases()) {
1078 if (!A.hasName())
1079 CreateModuleSlot(V: &A);
1080 }
1081
1082 for (const GlobalIFunc &I : TheModule->ifuncs()) {
1083 if (!I.hasName())
1084 CreateModuleSlot(V: &I);
1085 }
1086
1087 for (const Function &F : *TheModule) {
1088 if (!F.hasName())
1089 // Add all the unnamed functions to the table.
1090 CreateModuleSlot(V: &F);
1091
1092 // Add all the function attributes to the table.
1093 // FIXME: Add attributes of other objects?
1094 AttributeSet FnAttrs = F.getAttributes().getFnAttrs();
1095 if (FnAttrs.hasAttributes())
1096 CreateAttributeSetSlot(AS: FnAttrs);
1097 }
1098
1099 if (ProcessModuleHookFn)
1100 ProcessModuleHookFn(this, TheModule);
1101
1102 ST_DEBUG("end processModule!\n");
1103}
1104
1105// Process the arguments, basic blocks, and instructions of a function.
1106void SlotTracker::processFunction() {
1107 ST_DEBUG("begin processFunction!\n");
1108 fNext = 0;
1109
1110 // Add all the function arguments with no names.
1111 for(Function::const_arg_iterator AI = TheFunction->arg_begin(),
1112 AE = TheFunction->arg_end(); AI != AE; ++AI)
1113 if (!AI->hasName())
1114 CreateFunctionSlot(V: &*AI);
1115
1116 ST_DEBUG("Inserting Instructions:\n");
1117
1118 // Add all of the basic blocks and instructions with no names.
1119 for (auto &BB : *TheFunction) {
1120 if (!BB.hasName())
1121 CreateFunctionSlot(V: &BB);
1122
1123 for (auto &I : BB) {
1124 if (!I.getType()->isVoidTy() && !I.hasName())
1125 CreateFunctionSlot(V: &I);
1126
1127 // We allow direct calls to any llvm.foo function here, because the
1128 // target may not be linked into the optimizer.
1129 if (const auto *Call = dyn_cast<CallBase>(Val: &I)) {
1130 // Add all the call attributes to the table.
1131 AttributeSet Attrs = Call->getAttributes().getFnAttrs();
1132 if (Attrs.hasAttributes())
1133 CreateAttributeSetSlot(AS: Attrs);
1134 }
1135 }
1136 }
1137
1138 if (ProcessFunctionHookFn)
1139 ProcessFunctionHookFn(this, TheFunction);
1140
1141 FunctionProcessed = true;
1142
1143 ST_DEBUG("end processFunction!\n");
1144}
1145
1146// Iterate through all the GUID in the index and create slots for them.
1147int SlotTracker::processIndex() {
1148 ST_DEBUG("begin processIndex!\n");
1149 assert(TheIndex);
1150
1151 // The first block of slots are just the module ids, which start at 0 and are
1152 // assigned consecutively. Since the StringMap iteration order isn't
1153 // guaranteed, order by path string before assigning slots.
1154 std::vector<StringRef> ModulePaths;
1155 for (auto &[ModPath, _] : TheIndex->modulePaths())
1156 ModulePaths.push_back(x: ModPath);
1157 llvm::sort(C&: ModulePaths);
1158 for (auto &ModPath : ModulePaths)
1159 CreateModulePathSlot(Path: ModPath);
1160
1161 // Start numbering the GUIDs after the module ids.
1162 GUIDNext = ModulePathNext;
1163
1164 // Sort by GUID for deterministic slot assignment.
1165 for (const auto &GlobalList : TheIndex->sortedGlobalValueSummariesRange())
1166 CreateGUIDSlot(GUID: GlobalList.first);
1167
1168 // Start numbering the TypeIdCompatibleVtables after the GUIDs.
1169 TypeIdCompatibleVtableNext = GUIDNext;
1170 for (auto &TId : TheIndex->typeIdCompatibleVtableMap())
1171 CreateTypeIdCompatibleVtableSlot(Id: TId.first);
1172
1173 // Start numbering the TypeIds after the TypeIdCompatibleVtables.
1174 TypeIdNext = TypeIdCompatibleVtableNext;
1175 for (const auto &TID : TheIndex->typeIds())
1176 CreateTypeIdSlot(Id: TID.second.first);
1177
1178 ST_DEBUG("end processIndex!\n");
1179 return TypeIdNext;
1180}
1181
1182namespace {
1183class MetadataNodeVisitor {
1184 /// Visited MDNodes.
1185 SmallPtrSet<const MDNode *, 32> VisitedMDNodes;
1186 function_ref<void(const MDNode *)> Visit;
1187
1188 void visit(const MDNode *N) {
1189 if (isa<DIExpression>(Val: N) || !VisitedMDNodes.insert(Ptr: N).second)
1190 return;
1191
1192 Visit(N);
1193 for (const MDOperand &Op : N->operands())
1194 if (const auto *OpNode = dyn_cast_or_null<MDNode>(Val: Op.get()))
1195 visit(N: OpNode);
1196 }
1197
1198 void visitGlobalObjectMetadata(const GlobalObject &GO) {
1199 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
1200 GO.getAllMetadata(MDs);
1201 for (auto &MD : MDs)
1202 visit(N: MD.second);
1203 }
1204
1205 void visitDbgRecordMetadata(const DbgRecord &DR) {
1206 if (const auto *DVR = dyn_cast<const DbgVariableRecord>(Val: &DR)) {
1207 if (auto *Empty = dyn_cast_if_present<MDNode>(Val: DVR->getRawLocation()))
1208 visit(N: Empty);
1209 if (DVR->getRawVariable())
1210 visit(N: DVR->getRawVariable());
1211 if (DVR->isDbgAssign()) {
1212 if (auto *AssignID = DVR->getRawAssignID())
1213 visit(N: cast<MDNode>(Val: AssignID));
1214 if (auto *Empty = dyn_cast_if_present<MDNode>(Val: DVR->getRawAddress()))
1215 visit(N: Empty);
1216 }
1217 } else if (const auto *DLR = dyn_cast<const DbgLabelRecord>(Val: &DR)) {
1218 visit(N: DLR->getRawLabel());
1219 } else {
1220 llvm_unreachable("unsupported DbgRecord kind");
1221 }
1222 if (DR.getDebugLoc())
1223 visit(N: DR.getDebugLoc().getAsMDNode());
1224 }
1225
1226 void visitInstructionMetadata(const Instruction &I) {
1227 if (const auto *CI = dyn_cast<CallInst>(Val: &I))
1228 if (Function *F = CI->getCalledFunction())
1229 if (F->isIntrinsic())
1230 for (auto &Op : I.operands())
1231 if (auto *V = dyn_cast_or_null<MetadataAsValue>(Val: Op))
1232 if (auto *N = dyn_cast<MDNode>(Val: V->getMetadata()))
1233 visit(N);
1234
1235 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
1236 I.getAllMetadata(MDs);
1237 for (auto &MD : MDs)
1238 visit(N: MD.second);
1239 }
1240
1241 void visitFunctionMetadata(const Function &F) {
1242 visitGlobalObjectMetadata(GO: F);
1243 for (const BasicBlock &BB : F)
1244 for (const Instruction &I : BB) {
1245 for (const DbgRecord &DR : I.getDbgRecordRange())
1246 visitDbgRecordMetadata(DR);
1247 visitInstructionMetadata(I);
1248 }
1249 }
1250
1251public:
1252 MetadataNodeVisitor(function_ref<void(const MDNode *)> Visit)
1253 : Visit(Visit) {}
1254
1255 void visitModuleMetadata(const Module &M) {
1256 for (const GlobalVariable &Var : M.globals())
1257 visitGlobalObjectMetadata(GO: Var);
1258 for (const GlobalIFunc &I : M.ifuncs())
1259 visitGlobalObjectMetadata(GO: I);
1260 for (const NamedMDNode &NMD : M.named_metadata())
1261 for (const MDNode *N : NMD.operands())
1262 visit(N);
1263 for (const Function &F : M)
1264 visitFunctionMetadata(F);
1265 }
1266
1267 void visitMetadata(ArrayRef<const MDNode *> Metadata) {
1268 for (const MDNode *N : Metadata)
1269 visit(N);
1270 }
1271
1272 bool contains(const MDNode *N) const { return VisitedMDNodes.contains(Ptr: N); }
1273};
1274
1275class MetadataIDRenumberer {
1276 uint32_t NextID = 0;
1277
1278public:
1279 void run(const Module &M, ArrayRef<const MDNode *> AdditionalMetadata,
1280 ModuleSlotTracker::MachineMDNodeListType *AdditionalMetadataNodes =
1281 nullptr) {
1282 bool IsAdditionalMetadata = false;
1283 auto Renumber = [&](const MDNode *N) {
1284 N->getContext().pImpl->setMetadataPrintID(N: const_cast<MDNode *>(N),
1285 ID: NextID++);
1286 if (IsAdditionalMetadata && AdditionalMetadataNodes)
1287 AdditionalMetadataNodes->emplace_back(
1288 Args: N->getContext().pImpl->getMetadataPrintID(N), Args&: N);
1289 };
1290 MetadataNodeVisitor Visitor(Renumber);
1291
1292 Visitor.visitModuleMetadata(M);
1293
1294 IsAdditionalMetadata = true;
1295 Visitor.visitMetadata(Metadata: AdditionalMetadata);
1296
1297 // Keep IDs unique for nodes outside the canonical output.
1298 SmallVector<MDNode *, 32> RemainingNodes;
1299 M.getContext().pImpl->getAllMetadataNodes(Nodes&: RemainingNodes);
1300 llvm::erase_if(C&: RemainingNodes, P: [&](const MDNode *N) {
1301 return Visitor.contains(N) ||
1302 M.getContext().pImpl->getMetadataPrintID(N) >= NextID;
1303 });
1304 llvm::sort(C&: RemainingNodes, Comp: [&](const MDNode *LHS, const MDNode *RHS) {
1305 return M.getContext().pImpl->getMetadataPrintID(N: LHS) <
1306 M.getContext().pImpl->getMetadataPrintID(N: RHS);
1307 });
1308 for (MDNode *N : RemainingNodes)
1309 M.getContext().pImpl->setMetadataPrintID(
1310 N, ID: M.getContext().pImpl->allocateMetadataPrintID());
1311
1312 if (AdditionalMetadataNodes)
1313 llvm::sort(C&: *AdditionalMetadataNodes);
1314 }
1315};
1316} // namespace
1317
1318void Module::renumberMetadataForAssembly() {
1319 MetadataIDRenumberer().run(M: *this, AdditionalMetadata: {});
1320}
1321
1322void ModuleSlotTracker::renumberMetadataForAssembly(
1323 ArrayRef<const MDNode *> AdditionalMetadata,
1324 MachineMDNodeListType *AdditionalMetadataNodes) const {
1325 assert(M && "metadata renumbering requires a module");
1326 MetadataIDRenumberer().run(M: *M, AdditionalMetadata, AdditionalMetadataNodes);
1327}
1328
1329void ModuleSlotTracker::collectAdditionalMetadata(
1330 ArrayRef<const MDNode *> AdditionalMetadata,
1331 MachineMDNodeListType &AdditionalMetadataNodes) const {
1332 assert(M && "metadata collection requires a module");
1333 bool IsAdditionalMetadata = false;
1334 auto Collect = [&](const MDNode *N) {
1335 if (IsAdditionalMetadata)
1336 AdditionalMetadataNodes.emplace_back(
1337 Args: N->getContext().pImpl->getMetadataPrintID(N), Args&: N);
1338 };
1339 MetadataNodeVisitor Visitor(Collect);
1340 Visitor.visitModuleMetadata(M: *M);
1341 IsAdditionalMetadata = true;
1342 Visitor.visitMetadata(Metadata: AdditionalMetadata);
1343 llvm::sort(C&: AdditionalMetadataNodes);
1344}
1345
1346/// Clean up after incorporating a function. This is the only way to get out of
1347/// the function incorporation state that affects get*Slot/Create*Slot. Function
1348/// incorporation state is indicated by TheFunction != 0.
1349void SlotTracker::purgeFunction() {
1350 ST_DEBUG("begin purgeFunction!\n");
1351 fMap.clear(); // Simply discard the function level map
1352 TheFunction = nullptr;
1353 FunctionProcessed = false;
1354 ST_DEBUG("end purgeFunction!\n");
1355}
1356
1357/// getGlobalSlot - Get the slot number of a global value.
1358int SlotTracker::getGlobalSlot(const GlobalValue *V) {
1359 // Check for uninitialized state and do lazy initialization.
1360 initializeIfNeeded();
1361
1362 // Find the value in the module map
1363 ValueMap::iterator MI = mMap.find(Val: V);
1364 return MI == mMap.end() ? -1 : (int)MI->second;
1365}
1366
1367void SlotTracker::setProcessHook(
1368 std::function<void(AbstractSlotTrackerStorage *, const Module *)> Fn) {
1369 ProcessModuleHookFn = std::move(Fn);
1370}
1371
1372void SlotTracker::setProcessHook(
1373 std::function<void(AbstractSlotTrackerStorage *, const Function *)> Fn) {
1374 ProcessFunctionHookFn = std::move(Fn);
1375}
1376
1377/// getMetadataSlot - Get the slot number of a MDNode.
1378void SlotTracker::createMetadataSlot(const MDNode *N) { CreateMetadataSlot(N); }
1379
1380/// getMetadataSlot - Get the slot number of a MDNode.
1381int SlotTracker::getMetadataSlot(const MDNode *N) {
1382 // Check for uninitialized state and do lazy initialization.
1383 initializeIfNeeded();
1384
1385 if (isa<DIExpression>(Val: N))
1386 return -1;
1387 if (ShouldTrackMetadataDefinitions)
1388 CreateMetadataSlot(N);
1389 return N->getContext().pImpl->getMetadataPrintID(N);
1390}
1391
1392/// getLocalSlot - Get the slot number for a value that is local to a function.
1393int SlotTracker::getLocalSlot(const Value *V) {
1394 assert(!isa<Constant>(V) && "Can't get a constant or global slot with this!");
1395
1396 // Check for uninitialized state and do lazy initialization.
1397 initializeIfNeeded();
1398
1399 ValueMap::iterator FI = fMap.find(Val: V);
1400 return FI == fMap.end() ? -1 : (int)FI->second;
1401}
1402
1403int SlotTracker::getAttributeGroupSlot(AttributeSet AS) {
1404 // Check for uninitialized state and do lazy initialization.
1405 initializeIfNeeded();
1406
1407 // Find the AttributeSet in the module map.
1408 as_iterator AI = asMap.find(Val: AS);
1409 return AI == asMap.end() ? -1 : (int)AI->second;
1410}
1411
1412int SlotTracker::getModulePathSlot(StringRef Path) {
1413 // Check for uninitialized state and do lazy initialization.
1414 initializeIndexIfNeeded();
1415
1416 // Find the Module path in the map
1417 auto I = ModulePathMap.find(Key: Path);
1418 return I == ModulePathMap.end() ? -1 : (int)I->second;
1419}
1420
1421int SlotTracker::getGUIDSlot(GlobalValue::GUID GUID) {
1422 // Check for uninitialized state and do lazy initialization.
1423 initializeIndexIfNeeded();
1424
1425 // Find the GUID in the map
1426 guid_iterator I = GUIDMap.find(Val: GUID);
1427 return I == GUIDMap.end() ? -1 : (int)I->second;
1428}
1429
1430int SlotTracker::getTypeIdSlot(StringRef Id) {
1431 // Check for uninitialized state and do lazy initialization.
1432 initializeIndexIfNeeded();
1433
1434 // Find the TypeId string in the map
1435 auto I = TypeIdMap.find(Key: Id);
1436 return I == TypeIdMap.end() ? -1 : (int)I->second;
1437}
1438
1439int SlotTracker::getTypeIdCompatibleVtableSlot(StringRef Id) {
1440 // Check for uninitialized state and do lazy initialization.
1441 initializeIndexIfNeeded();
1442
1443 // Find the TypeIdCompatibleVtable string in the map
1444 auto I = TypeIdCompatibleVtableMap.find(Key: Id);
1445 return I == TypeIdCompatibleVtableMap.end() ? -1 : (int)I->second;
1446}
1447
1448/// CreateModuleSlot - Insert the specified GlobalValue* into the slot table.
1449void SlotTracker::CreateModuleSlot(const GlobalValue *V) {
1450 assert(V && "Can't insert a null Value into SlotTracker!");
1451 assert(!V->getType()->isVoidTy() && "Doesn't need a slot!");
1452 assert(!V->hasName() && "Doesn't need a slot!");
1453
1454 unsigned DestSlot = mNext++;
1455 mMap[V] = DestSlot;
1456
1457 ST_DEBUG(" Inserting value [" << V->getType() << "] = " << V << " slot=" <<
1458 DestSlot << " [");
1459 // G = Global, F = Function, A = Alias, I = IFunc, o = other
1460 ST_DEBUG((isa<GlobalVariable>(V) ? 'G' :
1461 (isa<Function>(V) ? 'F' :
1462 (isa<GlobalAlias>(V) ? 'A' :
1463 (isa<GlobalIFunc>(V) ? 'I' : 'o')))) << "]\n");
1464}
1465
1466/// CreateSlot - Create a new slot for the specified value if it has no name.
1467void SlotTracker::CreateFunctionSlot(const Value *V) {
1468 assert(!V->getType()->isVoidTy() && !V->hasName() && "Doesn't need a slot!");
1469
1470 unsigned DestSlot = fNext++;
1471 fMap[V] = DestSlot;
1472
1473 // G = Global, F = Function, o = other
1474 ST_DEBUG(" Inserting value [" << V->getType() << "] = " << V << " slot=" <<
1475 DestSlot << " [o]\n");
1476}
1477
1478/// CreateModuleSlot - Insert the specified MDNode* into the slot table.
1479void SlotTracker::CreateMetadataSlot(const MDNode *N) {
1480 assert(N && "Can't insert a null Value into SlotTracker!");
1481
1482 if (isa<DIExpression>(Val: N))
1483 return;
1484
1485 unsigned ID = N->getContext().pImpl->getMetadataPrintID(N);
1486 if (!mdnMap.try_emplace(Key: N, Args&: ID).second)
1487 return;
1488
1489 for (const MDOperand &Op : N->operands())
1490 if (const auto *OpNode = dyn_cast_or_null<MDNode>(Val: Op.get()))
1491 CreateMetadataSlot(N: OpNode);
1492}
1493
1494void SlotTracker::CreateAttributeSetSlot(AttributeSet AS) {
1495 assert(AS.hasAttributes() && "Doesn't need a slot!");
1496
1497 if (asMap.try_emplace(Key: AS, Args&: asNext).second)
1498 ++asNext;
1499}
1500
1501/// Create a new slot for the specified Module
1502void SlotTracker::CreateModulePathSlot(StringRef Path) {
1503 ModulePathMap[Path] = ModulePathNext++;
1504}
1505
1506/// Create a new slot for the specified GUID
1507void SlotTracker::CreateGUIDSlot(GlobalValue::GUID GUID) {
1508 GUIDMap[GUID] = GUIDNext++;
1509}
1510
1511/// Create a new slot for the specified Id
1512void SlotTracker::CreateTypeIdSlot(StringRef Id) {
1513 TypeIdMap[Id] = TypeIdNext++;
1514}
1515
1516/// Create a new slot for the specified Id
1517void SlotTracker::CreateTypeIdCompatibleVtableSlot(StringRef Id) {
1518 TypeIdCompatibleVtableMap[Id] = TypeIdCompatibleVtableNext++;
1519}
1520
1521namespace {
1522/// Common instances used by most of the printer functions.
1523struct AsmWriterContext {
1524 TypePrinting *TypePrinter = nullptr;
1525 SlotTracker *Machine = nullptr;
1526 const Module *Context = nullptr;
1527 const ModuleSlotTracker *MST = nullptr;
1528
1529 AsmWriterContext(TypePrinting *TP, SlotTracker *ST, const Module *M = nullptr,
1530 const ModuleSlotTracker *MST = nullptr)
1531 : TypePrinter(TP), Machine(ST), Context(M), MST(MST) {}
1532
1533 static AsmWriterContext &getEmpty() {
1534 static AsmWriterContext EmptyCtx(nullptr, nullptr);
1535 return EmptyCtx;
1536 }
1537
1538 /// A callback that will be triggered when the underlying printer
1539 /// prints a Metadata as operand.
1540 virtual void onWriteMetadataAsOperand(const Metadata *) {}
1541
1542 virtual ~AsmWriterContext() = default;
1543};
1544} // end anonymous namespace
1545
1546//===----------------------------------------------------------------------===//
1547// AsmWriter Implementation
1548//===----------------------------------------------------------------------===//
1549
1550static void writeAsOperandInternal(raw_ostream &Out, const Value *V,
1551 AsmWriterContext &WriterCtx,
1552 bool PrintType = false);
1553
1554static void writeAsOperandInternal(raw_ostream &Out, const Metadata *MD,
1555 AsmWriterContext &WriterCtx,
1556 bool FromValue = false);
1557
1558static void writeOptimizationInfo(raw_ostream &Out, const User *U) {
1559 if (const auto *FPO = dyn_cast<const FPMathOperator>(Val: U))
1560 Out << FPO->getFastMathFlags();
1561
1562 if (const auto *OBO = dyn_cast<OverflowingBinaryOperator>(Val: U)) {
1563 if (OBO->hasNoUnsignedWrap())
1564 Out << " nuw";
1565 if (OBO->hasNoSignedWrap())
1566 Out << " nsw";
1567 } else if (const auto *Div = dyn_cast<PossiblyExactOperator>(Val: U)) {
1568 if (Div->isExact())
1569 Out << " exact";
1570 } else if (const auto *PDI = dyn_cast<PossiblyDisjointInst>(Val: U)) {
1571 if (PDI->isDisjoint())
1572 Out << " disjoint";
1573 } else if (const auto *GEP = dyn_cast<GEPOperator>(Val: U)) {
1574 if (GEP->isInBounds())
1575 Out << " inbounds";
1576 else if (GEP->hasNoUnsignedSignedWrap())
1577 Out << " nusw";
1578 if (GEP->hasNoUnsignedWrap())
1579 Out << " nuw";
1580 if (auto InRange = GEP->getInRange()) {
1581 Out << " inrange(" << InRange->getLower() << ", " << InRange->getUpper()
1582 << ")";
1583 }
1584 } else if (const auto *NNI = dyn_cast<PossiblyNonNegInst>(Val: U)) {
1585 if (NNI->hasNonNeg())
1586 Out << " nneg";
1587 } else if (const auto *TI = dyn_cast<TruncInst>(Val: U)) {
1588 if (TI->hasNoUnsignedWrap())
1589 Out << " nuw";
1590 if (TI->hasNoSignedWrap())
1591 Out << " nsw";
1592 } else if (const auto *ICmp = dyn_cast<ICmpInst>(Val: U)) {
1593 if (ICmp->hasSameSign())
1594 Out << " samesign";
1595 } else if (const auto *ASC = dyn_cast<AddrSpaceCastInst>(Val: U)) {
1596 if (ASC->hasNonNull())
1597 Out << " nonnull";
1598 }
1599}
1600
1601static void WriteFullHexAPInt(raw_ostream &Out, const APInt &Val) {
1602 SmallVector<char, 32> Bits;
1603 Val.toStringUnsigned(Str&: Bits, Radix: 16);
1604 unsigned NumDigits = std::max(a: (Val.getBitWidth() + 3) / 4, b: 1U);
1605 Out << "0x";
1606 for (unsigned i = 0; i < NumDigits - Bits.size(); i++)
1607 Out << '0';
1608 Out << Bits;
1609}
1610
1611static void writeAPFloatInternal(raw_ostream &Out, const APFloat &APF) {
1612 bool ForceBitwiseOutput = false;
1613 if (&APF.getSemantics() == &APFloat::PPCDoubleDouble()) {
1614 // ppc_fp128 types are double-double. The special cases set the second
1615 // (high) double to +0.0, so if the high word is nonzero, force the use of
1616 // bitwise output.
1617 APInt HiWord = APF.bitcastToAPInt().lshr(shiftAmt: 64);
1618 ForceBitwiseOutput = !HiWord.isZero();
1619 }
1620
1621 if (!ForceBitwiseOutput) {
1622 // Check for special values in APFloat.
1623 if (APF.isInfinity()) {
1624 Out << (APF.isNegative() ? '-' : '+') << "inf";
1625 return;
1626 }
1627
1628 if (APF.isNaN()) {
1629 Out << (APF.isNegative() ? '-' : '+');
1630 APInt Payload = APF.getNaNPayload();
1631 // The quiet bit of a NaN is the highest bit of the payload, so the
1632 // preferred QNaN value happens to be the sign mask value.
1633 if (Payload.isSignMask()) {
1634 Out << "qnan";
1635 } else {
1636 if (APF.isSignaling())
1637 Out << 's';
1638 Out << "nan(";
1639 // Clear out the signaling/quiet bit of the payload for output.
1640 Payload.clearBit(BitPosition: Payload.getBitWidth() - 1);
1641 // Trim the string to exclude leading 0's.
1642 WriteFullHexAPInt(Out, Val: Payload.trunc(width: Payload.getActiveBits()));
1643 Out << ')';
1644 }
1645 return;
1646 }
1647 }
1648
1649 // Try for a decimal string output. If the value is convertible back to the
1650 // same APFloat value, then we know that it is safe to use it. Otherwise, fall
1651 // back onto the hexadecimal format.
1652 SmallString<128> StrVal;
1653 APF.toString(Str&: StrVal, FormatPrecision: 6, FormatMaxPadding: 0, TruncateZero: false);
1654 if (APFloat(APF.getSemantics(), StrVal) == APF) {
1655 Out << StrVal;
1656 return;
1657 }
1658
1659 // Fallback to the hexadecimal format representing the bit string exactly.
1660 Out << 'f';
1661 APInt API = APF.bitcastToAPInt();
1662 WriteFullHexAPInt(Out, Val: API);
1663}
1664
1665static void writeConstantInternal(raw_ostream &Out, const Constant *CV,
1666 AsmWriterContext &WriterCtx) {
1667 if (const auto *CI = dyn_cast<ConstantInt>(Val: CV)) {
1668 Type *Ty = CI->getType();
1669
1670 if (Ty->isVectorTy()) {
1671 Out << "splat (";
1672 WriterCtx.TypePrinter->print(Ty: Ty->getScalarType(), OS&: Out);
1673 Out << " ";
1674 }
1675
1676 if (Ty->getScalarType()->isIntegerTy(BitWidth: 1))
1677 Out << (CI->getZExtValue() ? "true" : "false");
1678 else
1679 Out << CI->getValue();
1680
1681 if (Ty->isVectorTy())
1682 Out << ")";
1683
1684 return;
1685 }
1686
1687 if (const auto *CB = dyn_cast<ConstantByte>(Val: CV)) {
1688 Type *Ty = CB->getType();
1689
1690 if (Ty->isVectorTy()) {
1691 Out << "splat (";
1692 WriterCtx.TypePrinter->print(Ty: Ty->getScalarType(), OS&: Out);
1693 Out << " ";
1694 }
1695
1696 Out << CB->getValue();
1697
1698 if (Ty->isVectorTy())
1699 Out << ")";
1700
1701 return;
1702 }
1703
1704 if (const auto *CFP = dyn_cast<ConstantFP>(Val: CV)) {
1705 Type *Ty = CFP->getType();
1706
1707 if (Ty->isVectorTy()) {
1708 if (CFP->getValue().bitcastToAPInt().isZero()) {
1709 Out << "zeroinitializer";
1710 return;
1711 }
1712
1713 Out << "splat (";
1714 WriterCtx.TypePrinter->print(Ty: Ty->getScalarType(), OS&: Out);
1715 Out << " ";
1716 }
1717
1718 writeAPFloatInternal(Out, APF: CFP->getValueAPF());
1719
1720 if (Ty->isVectorTy())
1721 Out << ")";
1722
1723 return;
1724 }
1725
1726 if (isa<ConstantAggregateZero>(Val: CV) || isa<ConstantTargetNone>(Val: CV)) {
1727 Out << "zeroinitializer";
1728 return;
1729 }
1730
1731 if (const auto *BA = dyn_cast<BlockAddress>(Val: CV)) {
1732 Out << "blockaddress(";
1733 writeAsOperandInternal(Out, V: BA->getFunction(), WriterCtx);
1734 Out << ", ";
1735 writeAsOperandInternal(Out, V: BA->getBasicBlock(), WriterCtx);
1736 Out << ")";
1737 return;
1738 }
1739
1740 if (const auto *Equiv = dyn_cast<DSOLocalEquivalent>(Val: CV)) {
1741 Out << "dso_local_equivalent ";
1742 writeAsOperandInternal(Out, V: Equiv->getGlobalValue(), WriterCtx);
1743 return;
1744 }
1745
1746 if (const auto *NC = dyn_cast<NoCFIValue>(Val: CV)) {
1747 Out << "no_cfi ";
1748 writeAsOperandInternal(Out, V: NC->getGlobalValue(), WriterCtx);
1749 return;
1750 }
1751
1752 if (const auto *CPA = dyn_cast<ConstantPtrAuth>(Val: CV)) {
1753 Out << "ptrauth (";
1754
1755 // ptrauth (ptr CST, i32 KEY[, i64 DISC[, ptr ADDRDISC[, ptr DS]?]?]?)
1756 unsigned NumOpsToWrite = 2;
1757 if (!CPA->getOperand(i_nocapture: 2)->isNullValue())
1758 NumOpsToWrite = 3;
1759 if (!isa<ConstantPointerNull>(Val: CPA->getOperand(i_nocapture: 3)))
1760 NumOpsToWrite = 4;
1761 if (!isa<ConstantPointerNull>(Val: CPA->getOperand(i_nocapture: 4)))
1762 NumOpsToWrite = 5;
1763
1764 ListSeparator LS;
1765 for (unsigned i = 0, e = NumOpsToWrite; i != e; ++i) {
1766 Out << LS;
1767 writeAsOperandInternal(Out, V: CPA->getOperand(i_nocapture: i), WriterCtx,
1768 /*PrintType=*/true);
1769 }
1770 Out << ')';
1771 return;
1772 }
1773
1774 if (const auto *CA = dyn_cast<ConstantArray>(Val: CV)) {
1775 Out << '[';
1776 ListSeparator LS;
1777 for (const Value *Op : CA->operands()) {
1778 Out << LS;
1779 writeAsOperandInternal(Out, V: Op, WriterCtx, /*PrintType=*/true);
1780 }
1781 Out << ']';
1782 return;
1783 }
1784
1785 if (const auto *CA = dyn_cast<ConstantDataArray>(Val: CV)) {
1786 // As a special case, print the array as a string if it is an array of
1787 // i8 with ConstantInt values.
1788 if (CA->isString()) {
1789 Out << "c\"";
1790 printEscapedString(Name: CA->getAsString(), Out);
1791 Out << '"';
1792 return;
1793 }
1794
1795 Out << '[';
1796 ListSeparator LS;
1797 for (uint64_t i = 0, e = CA->getNumElements(); i != e; ++i) {
1798 Out << LS;
1799 writeAsOperandInternal(Out, V: CA->getElementAsConstant(i), WriterCtx,
1800 /*PrintType=*/true);
1801 }
1802 Out << ']';
1803 return;
1804 }
1805
1806 if (const auto *CS = dyn_cast<ConstantStruct>(Val: CV)) {
1807 if (CS->getType()->isPacked())
1808 Out << '<';
1809 Out << '{';
1810 if (CS->getNumOperands() != 0) {
1811 Out << ' ';
1812 ListSeparator LS;
1813 for (const Value *Op : CS->operands()) {
1814 Out << LS;
1815 writeAsOperandInternal(Out, V: Op, WriterCtx, /*PrintType=*/true);
1816 }
1817 Out << ' ';
1818 }
1819 Out << '}';
1820 if (CS->getType()->isPacked())
1821 Out << '>';
1822 return;
1823 }
1824
1825 if (isa<ConstantVector>(Val: CV) || isa<ConstantDataVector>(Val: CV)) {
1826 auto *CVVTy = cast<FixedVectorType>(Val: CV->getType());
1827
1828 // Use the same shorthand for splat vector (i.e. "splat(Ty val)") as is
1829 // permitted on IR input to reduce the output changes when enabling
1830 // UseConstant{Int,FP}ForFixedLengthSplat.
1831 // TODO: Remove this block when the UseConstant{Int,FP}ForFixedLengthSplat
1832 // options are removed.
1833 if (auto *SplatVal = CV->getSplatValue()) {
1834 if (isa<ConstantInt>(Val: SplatVal) || isa<ConstantFP>(Val: SplatVal) ||
1835 isa<ConstantByte>(Val: SplatVal)) {
1836 Out << "splat (";
1837 writeAsOperandInternal(Out, V: SplatVal, WriterCtx, /*PrintType=*/true);
1838 Out << ')';
1839 return;
1840 }
1841 }
1842
1843 Out << '<';
1844 ListSeparator LS;
1845 for (unsigned i = 0, e = CVVTy->getNumElements(); i != e; ++i) {
1846 Out << LS;
1847 writeAsOperandInternal(Out, V: CV->getAggregateElement(Elt: i), WriterCtx,
1848 /*PrintType=*/true);
1849 }
1850 Out << '>';
1851 return;
1852 }
1853
1854 if (const auto *CPN = dyn_cast<ConstantPointerNull>(Val: CV)) {
1855 if (auto *VT = dyn_cast<VectorType>(Val: CPN->getType())) {
1856 Out << "splat (";
1857 writeAsOperandInternal(Out,
1858 V: ConstantPointerNull::get(T: VT->getElementType()),
1859 WriterCtx, /*PrintType=*/true);
1860 Out << ')';
1861 return;
1862 }
1863
1864 Out << "null";
1865 return;
1866 }
1867
1868 if (isa<ConstantTokenNone>(Val: CV)) {
1869 Out << "none";
1870 return;
1871 }
1872
1873 if (isa<PoisonValue>(Val: CV)) {
1874 Out << "poison";
1875 return;
1876 }
1877
1878 if (isa<UndefValue>(Val: CV)) {
1879 Out << "undef";
1880 return;
1881 }
1882
1883 if (const auto *CE = dyn_cast<ConstantExpr>(Val: CV)) {
1884 // Use the same shorthand for splat vector (i.e. "splat(Ty val)") as is
1885 // permitted on IR input to reduce the output changes when enabling
1886 // UseConstant{Int,FP}ForScalableSplat.
1887 // TODO: Remove this block when the UseConstant{Int,FP}ForScalableSplat
1888 // options are removed.
1889 if (CE->getOpcode() == Instruction::ShuffleVector) {
1890 if (auto *SplatVal = CE->getSplatValue()) {
1891 if (isa<ConstantInt>(Val: SplatVal) || isa<ConstantFP>(Val: SplatVal) ||
1892 isa<ConstantByte>(Val: SplatVal)) {
1893 Out << "splat (";
1894 writeAsOperandInternal(Out, V: SplatVal, WriterCtx, /*PrintType=*/true);
1895 Out << ')';
1896 return;
1897 }
1898 }
1899 }
1900
1901 Out << CE->getOpcodeName();
1902 writeOptimizationInfo(Out, U: CE);
1903 Out << " (";
1904
1905 if (const auto *GEP = dyn_cast<GEPOperator>(Val: CE)) {
1906 WriterCtx.TypePrinter->print(Ty: GEP->getSourceElementType(), OS&: Out);
1907 Out << ", ";
1908 }
1909
1910 ListSeparator LS;
1911 for (const Value *Op : CE->operands()) {
1912 Out << LS;
1913 writeAsOperandInternal(Out, V: Op, WriterCtx, /*PrintType=*/true);
1914 }
1915
1916 if (CE->isCast()) {
1917 Out << " to ";
1918 WriterCtx.TypePrinter->print(Ty: CE->getType(), OS&: Out);
1919 }
1920
1921 if (CE->getOpcode() == Instruction::ShuffleVector)
1922 printShuffleMask(Out, Ty: CE->getType(), Mask: CE->getShuffleMask());
1923
1924 Out << ')';
1925 return;
1926 }
1927
1928 Out << "<placeholder or erroneous Constant>";
1929}
1930
1931static void writeMDTuple(raw_ostream &Out, const MDTuple *Node,
1932 AsmWriterContext &WriterCtx) {
1933 Out << "!{";
1934 ListSeparator LS;
1935 for (const Metadata *MD : Node->operands()) {
1936 Out << LS;
1937 if (!MD) {
1938 Out << "null";
1939 } else if (auto *MDV = dyn_cast<ValueAsMetadata>(Val: MD)) {
1940 Value *V = MDV->getValue();
1941 writeAsOperandInternal(Out, V, WriterCtx, /*PrintType=*/true);
1942 } else {
1943 writeAsOperandInternal(Out, MD, WriterCtx);
1944 WriterCtx.onWriteMetadataAsOperand(MD);
1945 }
1946 }
1947
1948 Out << "}";
1949}
1950
1951namespace {
1952
1953struct MDFieldPrinter {
1954 raw_ostream &Out;
1955 ListSeparator FS;
1956 AsmWriterContext &WriterCtx;
1957
1958 explicit MDFieldPrinter(raw_ostream &Out)
1959 : Out(Out), WriterCtx(AsmWriterContext::getEmpty()) {}
1960 MDFieldPrinter(raw_ostream &Out, AsmWriterContext &Ctx)
1961 : Out(Out), WriterCtx(Ctx) {}
1962
1963 void printTag(const DINode *N);
1964 void printMacinfoType(const DIMacroNode *N);
1965 void printChecksum(const DIFile::ChecksumInfo<StringRef> &N);
1966 void printString(StringRef Name, StringRef Value,
1967 bool ShouldSkipEmpty = true);
1968 void printMetadata(StringRef Name, const Metadata *MD,
1969 bool ShouldSkipNull = true);
1970 void printMetadataOrInt(StringRef Name, const Metadata *MD, bool IsUnsigned,
1971 bool ShouldSkipZero = true);
1972 template <class IntTy>
1973 void printInt(StringRef Name, IntTy Int, bool ShouldSkipZero = true);
1974 void printAPInt(StringRef Name, const APInt &Int, bool IsUnsigned,
1975 bool ShouldSkipZero);
1976 void printBool(StringRef Name, bool Value,
1977 std::optional<bool> Default = std::nullopt);
1978 void printDIFlags(StringRef Name, DINode::DIFlags Flags);
1979 void printDISPFlags(StringRef Name, DISubprogram::DISPFlags Flags);
1980 template <class IntTy, class Stringifier>
1981 void printDwarfEnum(StringRef Name, IntTy Value, Stringifier toString,
1982 bool ShouldSkipZero = true);
1983 void printEmissionKind(StringRef Name, DICompileUnit::DebugEmissionKind EK);
1984 void printNameTableKind(StringRef Name,
1985 DICompileUnit::DebugNameTableKind NTK);
1986 void printFixedPointKind(StringRef Name, DIFixedPointType::FixedPointKind V);
1987};
1988
1989} // end anonymous namespace
1990
1991void MDFieldPrinter::printTag(const DINode *N) {
1992 Out << FS << "tag: ";
1993 auto Tag = dwarf::TagString(Tag: N->getTag());
1994 if (!Tag.empty())
1995 Out << Tag;
1996 else
1997 Out << N->getTag();
1998}
1999
2000void MDFieldPrinter::printMacinfoType(const DIMacroNode *N) {
2001 Out << FS << "type: ";
2002 auto Type = dwarf::MacinfoString(Encoding: N->getMacinfoType());
2003 if (!Type.empty())
2004 Out << Type;
2005 else
2006 Out << N->getMacinfoType();
2007}
2008
2009void MDFieldPrinter::printChecksum(
2010 const DIFile::ChecksumInfo<StringRef> &Checksum) {
2011 Out << FS << "checksumkind: " << Checksum.getKindAsString();
2012 printString(Name: "checksum", Value: Checksum.Value, /* ShouldSkipEmpty */ false);
2013}
2014
2015void MDFieldPrinter::printString(StringRef Name, StringRef Value,
2016 bool ShouldSkipEmpty) {
2017 if (ShouldSkipEmpty && Value.empty())
2018 return;
2019
2020 Out << FS << Name << ": \"";
2021 printEscapedString(Name: Value, Out);
2022 Out << "\"";
2023}
2024
2025static void writeMetadataAsOperand(raw_ostream &Out, const Metadata *MD,
2026 AsmWriterContext &WriterCtx) {
2027 if (!MD) {
2028 Out << "null";
2029 return;
2030 }
2031 writeAsOperandInternal(Out, MD, WriterCtx);
2032 WriterCtx.onWriteMetadataAsOperand(MD);
2033}
2034
2035void MDFieldPrinter::printMetadata(StringRef Name, const Metadata *MD,
2036 bool ShouldSkipNull) {
2037 if (ShouldSkipNull && !MD)
2038 return;
2039
2040 Out << FS << Name << ": ";
2041 writeMetadataAsOperand(Out, MD, WriterCtx);
2042}
2043
2044void MDFieldPrinter::printMetadataOrInt(StringRef Name, const Metadata *MD,
2045 bool IsUnsigned, bool ShouldSkipZero) {
2046 if (!MD)
2047 return;
2048
2049 if (auto *CI = dyn_cast<ConstantAsMetadata>(Val: MD)) {
2050 auto *CV = cast<ConstantInt>(Val: CI->getValue());
2051 if (IsUnsigned)
2052 printInt(Name, Int: CV->getZExtValue(), ShouldSkipZero);
2053 else
2054 printInt(Name, Int: CV->getSExtValue(), ShouldSkipZero);
2055 } else
2056 printMetadata(Name, MD);
2057}
2058
2059template <class IntTy>
2060void MDFieldPrinter::printInt(StringRef Name, IntTy Int, bool ShouldSkipZero) {
2061 if (ShouldSkipZero && !Int)
2062 return;
2063
2064 Out << FS << Name << ": " << Int;
2065}
2066
2067void MDFieldPrinter::printAPInt(StringRef Name, const APInt &Int,
2068 bool IsUnsigned, bool ShouldSkipZero) {
2069 if (ShouldSkipZero && Int.isZero())
2070 return;
2071
2072 Out << FS << Name << ": ";
2073 Int.print(OS&: Out, isSigned: !IsUnsigned);
2074}
2075
2076void MDFieldPrinter::printBool(StringRef Name, bool Value,
2077 std::optional<bool> Default) {
2078 if (Default && Value == *Default)
2079 return;
2080 Out << FS << Name << ": " << (Value ? "true" : "false");
2081}
2082
2083void MDFieldPrinter::printDIFlags(StringRef Name, DINode::DIFlags Flags) {
2084 if (!Flags)
2085 return;
2086
2087 Out << FS << Name << ": ";
2088
2089 SmallVector<DINode::DIFlags, 8> SplitFlags;
2090 auto Extra = DINode::splitFlags(Flags, SplitFlags);
2091
2092 ListSeparator FlagsFS(" | ");
2093 for (auto F : SplitFlags) {
2094 auto StringF = DINode::getFlagString(Flag: F);
2095 assert(!StringF.empty() && "Expected valid flag");
2096 Out << FlagsFS << StringF;
2097 }
2098 if (Extra || SplitFlags.empty())
2099 Out << FlagsFS << Extra;
2100}
2101
2102void MDFieldPrinter::printDISPFlags(StringRef Name,
2103 DISubprogram::DISPFlags Flags) {
2104 // Always print this field, because no flags in the IR at all will be
2105 // interpreted as old-style isDefinition: true.
2106 Out << FS << Name << ": ";
2107
2108 if (!Flags) {
2109 Out << 0;
2110 return;
2111 }
2112
2113 SmallVector<DISubprogram::DISPFlags, 8> SplitFlags;
2114 auto Extra = DISubprogram::splitFlags(Flags, SplitFlags);
2115
2116 ListSeparator FlagsFS(" | ");
2117 for (auto F : SplitFlags) {
2118 auto StringF = DISubprogram::getFlagString(Flag: F);
2119 assert(!StringF.empty() && "Expected valid flag");
2120 Out << FlagsFS << StringF;
2121 }
2122 if (Extra || SplitFlags.empty())
2123 Out << FlagsFS << Extra;
2124}
2125
2126void MDFieldPrinter::printEmissionKind(StringRef Name,
2127 DICompileUnit::DebugEmissionKind EK) {
2128 Out << FS << Name << ": " << DICompileUnit::emissionKindString(EK);
2129}
2130
2131void MDFieldPrinter::printNameTableKind(StringRef Name,
2132 DICompileUnit::DebugNameTableKind NTK) {
2133 if (NTK == DICompileUnit::DebugNameTableKind::Default)
2134 return;
2135 Out << FS << Name << ": " << DICompileUnit::nameTableKindString(PK: NTK);
2136}
2137
2138void MDFieldPrinter::printFixedPointKind(StringRef Name,
2139 DIFixedPointType::FixedPointKind V) {
2140 Out << FS << Name << ": " << DIFixedPointType::fixedPointKindString(V);
2141}
2142
2143template <class IntTy, class Stringifier>
2144void MDFieldPrinter::printDwarfEnum(StringRef Name, IntTy Value,
2145 Stringifier toString, bool ShouldSkipZero) {
2146 if (ShouldSkipZero && !Value)
2147 return;
2148
2149 Out << FS << Name << ": ";
2150 auto S = toString(Value);
2151 if (!S.empty())
2152 Out << S;
2153 else
2154 Out << Value;
2155}
2156
2157static void writeGenericDINode(raw_ostream &Out, const GenericDINode *N,
2158 AsmWriterContext &WriterCtx) {
2159 Out << "!GenericDINode(";
2160 MDFieldPrinter Printer(Out, WriterCtx);
2161 Printer.printTag(N);
2162 Printer.printString(Name: "header", Value: N->getHeader());
2163 if (N->getNumDwarfOperands()) {
2164 Out << Printer.FS << "operands: {";
2165 ListSeparator IFS;
2166 for (auto &I : N->dwarf_operands()) {
2167 Out << IFS;
2168 writeMetadataAsOperand(Out, MD: I, WriterCtx);
2169 }
2170 Out << "}";
2171 }
2172 Out << ")";
2173}
2174
2175static void writeDILocation(raw_ostream &Out, const DILocation *DL,
2176 AsmWriterContext &WriterCtx) {
2177 Out << "!DILocation(";
2178 MDFieldPrinter Printer(Out, WriterCtx);
2179 // Always output the line, since 0 is a relevant and important value for it.
2180 Printer.printInt(Name: "line", Int: DL->getLine(), /* ShouldSkipZero */ false);
2181 Printer.printInt(Name: "column", Int: DL->getColumn());
2182 Printer.printMetadata(Name: "scope", MD: DL->getRawScope(), /* ShouldSkipNull */ false);
2183 Printer.printMetadata(Name: "inlinedAt", MD: DL->getRawInlinedAt());
2184 Printer.printBool(Name: "isImplicitCode", Value: DL->isImplicitCode(),
2185 /* Default */ false);
2186 Printer.printInt(Name: "atomGroup", Int: DL->getAtomGroup());
2187 Printer.printInt<unsigned>(Name: "atomRank", Int: DL->getAtomRank());
2188 Out << ")";
2189}
2190
2191static void writeDIAssignID(raw_ostream &Out, const DIAssignID *DL,
2192 AsmWriterContext &WriterCtx) {
2193 Out << "!DIAssignID()";
2194 MDFieldPrinter Printer(Out, WriterCtx);
2195}
2196
2197static void writeDISubrange(raw_ostream &Out, const DISubrange *N,
2198 AsmWriterContext &WriterCtx) {
2199 Out << "!DISubrange(";
2200 MDFieldPrinter Printer(Out, WriterCtx);
2201
2202 Printer.printMetadataOrInt(Name: "count", MD: N->getRawCountNode(),
2203 /* IsUnsigned */ false,
2204 /* ShouldSkipZero */ false);
2205
2206 // A lowerBound of constant 0 should not be skipped, since it is different
2207 // from an unspecified lower bound (= nullptr).
2208 Printer.printMetadataOrInt(Name: "lowerBound", MD: N->getRawLowerBound(),
2209 /* IsUnsigned */ false,
2210 /* ShouldSkipZero */ false);
2211 Printer.printMetadataOrInt(Name: "upperBound", MD: N->getRawUpperBound(),
2212 /* IsUnsigned */ false,
2213 /* ShouldSkipZero */ false);
2214 Printer.printMetadataOrInt(Name: "stride", MD: N->getRawStride(),
2215 /* IsUnsigned */ false,
2216 /* ShouldSkipZero */ false);
2217
2218 Out << ")";
2219}
2220
2221static void writeDIGenericSubrange(raw_ostream &Out, const DIGenericSubrange *N,
2222 AsmWriterContext &WriterCtx) {
2223 Out << "!DIGenericSubrange(";
2224 MDFieldPrinter Printer(Out, WriterCtx);
2225
2226 auto GetConstant = [&](Metadata *Bound) -> std::optional<int64_t> {
2227 auto *BE = dyn_cast_or_null<DIExpression>(Val: Bound);
2228 if (!BE)
2229 return std::nullopt;
2230 if (BE->isConstant() &&
2231 DIExpression::SignedOrUnsignedConstant::SignedConstant ==
2232 *BE->isConstant()) {
2233 return static_cast<int64_t>(BE->getElement(I: 1));
2234 }
2235 return std::nullopt;
2236 };
2237
2238 auto *Count = N->getRawCountNode();
2239 if (auto ConstantCount = GetConstant(Count))
2240 Printer.printInt(Name: "count", Int: *ConstantCount,
2241 /* ShouldSkipZero */ false);
2242 else
2243 Printer.printMetadata(Name: "count", MD: Count, /*ShouldSkipNull */ true);
2244
2245 auto *LBound = N->getRawLowerBound();
2246 if (auto ConstantLBound = GetConstant(LBound))
2247 Printer.printInt(Name: "lowerBound", Int: *ConstantLBound,
2248 /* ShouldSkipZero */ false);
2249 else
2250 Printer.printMetadata(Name: "lowerBound", MD: LBound, /*ShouldSkipNull */ true);
2251
2252 auto *UBound = N->getRawUpperBound();
2253 if (auto ConstantUBound = GetConstant(UBound))
2254 Printer.printInt(Name: "upperBound", Int: *ConstantUBound,
2255 /* ShouldSkipZero */ false);
2256 else
2257 Printer.printMetadata(Name: "upperBound", MD: UBound, /*ShouldSkipNull */ true);
2258
2259 auto *Stride = N->getRawStride();
2260 if (auto ConstantStride = GetConstant(Stride))
2261 Printer.printInt(Name: "stride", Int: *ConstantStride,
2262 /* ShouldSkipZero */ false);
2263 else
2264 Printer.printMetadata(Name: "stride", MD: Stride, /*ShouldSkipNull */ true);
2265
2266 Out << ")";
2267}
2268
2269static void writeDIEnumerator(raw_ostream &Out, const DIEnumerator *N,
2270 AsmWriterContext &) {
2271 Out << "!DIEnumerator(";
2272 MDFieldPrinter Printer(Out);
2273 Printer.printString(Name: "name", Value: N->getName(), /* ShouldSkipEmpty */ false);
2274 Printer.printAPInt(Name: "value", Int: N->getValue(), IsUnsigned: N->isUnsigned(),
2275 /*ShouldSkipZero=*/false);
2276 if (N->isUnsigned())
2277 Printer.printBool(Name: "isUnsigned", Value: true);
2278 Out << ")";
2279}
2280
2281static void writeDIBasicType(raw_ostream &Out, const DIBasicType *N,
2282 AsmWriterContext &WriterCtx) {
2283 Out << "!DIBasicType(";
2284 MDFieldPrinter Printer(Out, WriterCtx);
2285 if (N->getTag() != dwarf::DW_TAG_base_type)
2286 Printer.printTag(N);
2287 Printer.printString(Name: "name", Value: N->getName());
2288 Printer.printMetadata(Name: "scope", MD: N->getRawScope());
2289 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2290 Printer.printInt(Name: "line", Int: N->getLine());
2291 Printer.printMetadataOrInt(Name: "size", MD: N->getRawSizeInBits(), IsUnsigned: true);
2292 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2293 Printer.printInt(Name: "dataSize", Int: N->getDataSizeInBits());
2294 Printer.printDwarfEnum(Name: "encoding", Value: N->getEncoding(),
2295 toString: dwarf::AttributeEncodingString);
2296 Printer.printInt(Name: "num_extra_inhabitants", Int: N->getNumExtraInhabitants());
2297 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2298 Out << ")";
2299}
2300
2301static void writeDIFixedPointType(raw_ostream &Out, const DIFixedPointType *N,
2302 AsmWriterContext &WriterCtx) {
2303 Out << "!DIFixedPointType(";
2304 MDFieldPrinter Printer(Out, WriterCtx);
2305 if (N->getTag() != dwarf::DW_TAG_base_type)
2306 Printer.printTag(N);
2307 Printer.printString(Name: "name", Value: N->getName());
2308 Printer.printMetadata(Name: "scope", MD: N->getRawScope());
2309 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2310 Printer.printInt(Name: "line", Int: N->getLine());
2311 Printer.printMetadataOrInt(Name: "size", MD: N->getRawSizeInBits(), IsUnsigned: true);
2312 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2313 Printer.printDwarfEnum(Name: "encoding", Value: N->getEncoding(),
2314 toString: dwarf::AttributeEncodingString);
2315 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2316 Printer.printFixedPointKind(Name: "kind", V: N->getKind());
2317 if (N->isRational()) {
2318 bool IsUnsigned = !N->isSigned();
2319 Printer.printAPInt(Name: "numerator", Int: N->getNumerator(), IsUnsigned, ShouldSkipZero: false);
2320 Printer.printAPInt(Name: "denominator", Int: N->getDenominator(), IsUnsigned, ShouldSkipZero: false);
2321 } else {
2322 Printer.printInt(Name: "factor", Int: N->getFactor());
2323 }
2324 Out << ")";
2325}
2326
2327static void writeDIStringType(raw_ostream &Out, const DIStringType *N,
2328 AsmWriterContext &WriterCtx) {
2329 Out << "!DIStringType(";
2330 MDFieldPrinter Printer(Out, WriterCtx);
2331 if (N->getTag() != dwarf::DW_TAG_string_type)
2332 Printer.printTag(N);
2333 Printer.printString(Name: "name", Value: N->getName());
2334 Printer.printMetadata(Name: "stringLength", MD: N->getRawStringLength());
2335 Printer.printMetadata(Name: "stringLengthExpression", MD: N->getRawStringLengthExp());
2336 Printer.printMetadata(Name: "stringLocationExpression",
2337 MD: N->getRawStringLocationExp());
2338 Printer.printMetadataOrInt(Name: "size", MD: N->getRawSizeInBits(), IsUnsigned: true);
2339 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2340 Printer.printDwarfEnum(Name: "encoding", Value: N->getEncoding(),
2341 toString: dwarf::AttributeEncodingString);
2342 Out << ")";
2343}
2344
2345static void writeDIDerivedType(raw_ostream &Out, const DIDerivedType *N,
2346 AsmWriterContext &WriterCtx) {
2347 Out << "!DIDerivedType(";
2348 MDFieldPrinter Printer(Out, WriterCtx);
2349 Printer.printTag(N);
2350 Printer.printString(Name: "name", Value: N->getName());
2351 Printer.printMetadata(Name: "scope", MD: N->getRawScope());
2352 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2353 Printer.printInt(Name: "line", Int: N->getLine());
2354 Printer.printMetadata(Name: "baseType", MD: N->getRawBaseType(),
2355 /* ShouldSkipNull */ false);
2356 Printer.printMetadataOrInt(Name: "size", MD: N->getRawSizeInBits(), IsUnsigned: true);
2357 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2358 Printer.printMetadataOrInt(Name: "offset", MD: N->getRawOffsetInBits(), IsUnsigned: true);
2359 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2360 Printer.printMetadata(Name: "extraData", MD: N->getRawExtraData());
2361 if (const auto &DWARFAddressSpace = N->getDWARFAddressSpace())
2362 Printer.printInt(Name: "dwarfAddressSpace", Int: *DWARFAddressSpace,
2363 /* ShouldSkipZero */ false);
2364 Printer.printMetadata(Name: "annotations", MD: N->getRawAnnotations());
2365 if (auto PtrAuthData = N->getPtrAuthData()) {
2366 Printer.printInt(Name: "ptrAuthKey", Int: PtrAuthData->key());
2367 Printer.printBool(Name: "ptrAuthIsAddressDiscriminated",
2368 Value: PtrAuthData->isAddressDiscriminated());
2369 Printer.printInt(Name: "ptrAuthExtraDiscriminator",
2370 Int: PtrAuthData->extraDiscriminator());
2371 Printer.printBool(Name: "ptrAuthIsaPointer", Value: PtrAuthData->isaPointer());
2372 Printer.printBool(Name: "ptrAuthAuthenticatesNullValues",
2373 Value: PtrAuthData->authenticatesNullValues());
2374 }
2375 Out << ")";
2376}
2377
2378static void writeDISubrangeType(raw_ostream &Out, const DISubrangeType *N,
2379 AsmWriterContext &WriterCtx) {
2380 Out << "!DISubrangeType(";
2381 MDFieldPrinter Printer(Out, WriterCtx);
2382 Printer.printString(Name: "name", Value: N->getName());
2383 Printer.printMetadata(Name: "scope", MD: N->getRawScope());
2384 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2385 Printer.printInt(Name: "line", Int: N->getLine());
2386 Printer.printMetadataOrInt(Name: "size", MD: N->getRawSizeInBits(), IsUnsigned: true);
2387 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2388 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2389 Printer.printMetadata(Name: "baseType", MD: N->getRawBaseType(),
2390 /* ShouldSkipNull */ false);
2391 Printer.printMetadata(Name: "lowerBound", MD: N->getRawLowerBound());
2392 Printer.printMetadata(Name: "upperBound", MD: N->getRawUpperBound());
2393 Printer.printMetadata(Name: "stride", MD: N->getRawStride());
2394 Printer.printMetadata(Name: "bias", MD: N->getRawBias());
2395 Out << ")";
2396}
2397
2398static void writeDICompositeType(raw_ostream &Out, const DICompositeType *N,
2399 AsmWriterContext &WriterCtx) {
2400 Out << "!DICompositeType(";
2401 MDFieldPrinter Printer(Out, WriterCtx);
2402 Printer.printTag(N);
2403 Printer.printString(Name: "name", Value: N->getName());
2404 Printer.printMetadata(Name: "scope", MD: N->getRawScope());
2405 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2406 Printer.printInt(Name: "line", Int: N->getLine());
2407 Printer.printMetadata(Name: "baseType", MD: N->getRawBaseType());
2408 Printer.printMetadataOrInt(Name: "size", MD: N->getRawSizeInBits(), IsUnsigned: true);
2409 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2410 Printer.printMetadataOrInt(Name: "offset", MD: N->getRawOffsetInBits(), IsUnsigned: true);
2411 Printer.printInt(Name: "num_extra_inhabitants", Int: N->getNumExtraInhabitants());
2412 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2413 Printer.printMetadata(Name: "elements", MD: N->getRawElements());
2414 Printer.printDwarfEnum(Name: "runtimeLang", Value: N->getRuntimeLang(),
2415 toString: dwarf::LanguageString);
2416 Printer.printMetadata(Name: "vtableHolder", MD: N->getRawVTableHolder());
2417 Printer.printMetadata(Name: "templateParams", MD: N->getRawTemplateParams());
2418 Printer.printString(Name: "identifier", Value: N->getIdentifier());
2419 Printer.printMetadata(Name: "discriminator", MD: N->getRawDiscriminator());
2420 Printer.printMetadata(Name: "dataLocation", MD: N->getRawDataLocation());
2421 Printer.printMetadata(Name: "associated", MD: N->getRawAssociated());
2422 Printer.printMetadata(Name: "allocated", MD: N->getRawAllocated());
2423 if (auto *RankConst = N->getRankConst())
2424 Printer.printInt(Name: "rank", Int: RankConst->getSExtValue(),
2425 /* ShouldSkipZero */ false);
2426 else
2427 Printer.printMetadata(Name: "rank", MD: N->getRawRank(), /*ShouldSkipNull */ true);
2428 Printer.printMetadata(Name: "annotations", MD: N->getRawAnnotations());
2429 if (auto *Specification = N->getRawSpecification())
2430 Printer.printMetadata(Name: "specification", MD: Specification);
2431
2432 if (auto EnumKind = N->getEnumKind())
2433 Printer.printDwarfEnum(Name: "enumKind", Value: *EnumKind, toString: dwarf::EnumKindString,
2434 /*ShouldSkipZero=*/false);
2435
2436 Printer.printMetadata(Name: "bitStride", MD: N->getRawBitStride());
2437 Out << ")";
2438}
2439
2440static void writeDISubroutineType(raw_ostream &Out, const DISubroutineType *N,
2441 AsmWriterContext &WriterCtx) {
2442 Out << "!DISubroutineType(";
2443 MDFieldPrinter Printer(Out, WriterCtx);
2444 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2445 Printer.printDwarfEnum(Name: "cc", Value: N->getCC(), toString: dwarf::ConventionString);
2446 Printer.printMetadata(Name: "types", MD: N->getRawTypeArray(),
2447 /* ShouldSkipNull */ false);
2448 Out << ")";
2449}
2450
2451static void writeDIFile(raw_ostream &Out, const DIFile *N, AsmWriterContext &) {
2452 Out << "!DIFile(";
2453 MDFieldPrinter Printer(Out);
2454 Printer.printString(Name: "filename", Value: N->getFilename(),
2455 /* ShouldSkipEmpty */ false);
2456 Printer.printString(Name: "directory", Value: N->getDirectory(),
2457 /* ShouldSkipEmpty */ false);
2458 // Print all values for checksum together, or not at all.
2459 if (N->getChecksum())
2460 Printer.printChecksum(Checksum: *N->getChecksum());
2461 if (N->getSource())
2462 Printer.printString(Name: "source", Value: *N->getSource(),
2463 /* ShouldSkipEmpty */ false);
2464 Out << ")";
2465}
2466
2467static void writeDICompileUnit(raw_ostream &Out, const DICompileUnit *N,
2468 AsmWriterContext &WriterCtx) {
2469 Out << "!DICompileUnit(";
2470 MDFieldPrinter Printer(Out, WriterCtx);
2471
2472 DISourceLanguageName Lang = N->getSourceLanguage();
2473
2474 if (Lang.hasVersionedName()) {
2475 Printer.printDwarfEnum(
2476 Name: "sourceLanguageName",
2477 Value: static_cast<llvm::dwarf::SourceLanguageName>(Lang.getName()),
2478 toString: dwarf::SourceLanguageNameString,
2479 /* ShouldSkipZero */ false);
2480
2481 Printer.printInt(Name: "sourceLanguageVersion", Int: Lang.getVersion(),
2482 /*ShouldSkipZero=*/true);
2483 } else {
2484 Printer.printDwarfEnum(Name: "language", Value: Lang.getName(), toString: dwarf::LanguageString,
2485 /* ShouldSkipZero */ false);
2486 }
2487
2488 Printer.printMetadata(Name: "file", MD: N->getRawFile(), /* ShouldSkipNull */ false);
2489 Printer.printString(Name: "producer", Value: N->getProducer());
2490 Printer.printBool(Name: "isOptimized", Value: N->isOptimized());
2491 Printer.printString(Name: "flags", Value: N->getFlags());
2492 Printer.printInt(Name: "runtimeVersion", Int: N->getRuntimeVersion(),
2493 /* ShouldSkipZero */ false);
2494 Printer.printString(Name: "splitDebugFilename", Value: N->getSplitDebugFilename());
2495 Printer.printEmissionKind(Name: "emissionKind", EK: N->getEmissionKind());
2496 Printer.printMetadata(Name: "enums", MD: N->getRawEnumTypes());
2497 Printer.printMetadata(Name: "retainedTypes", MD: N->getRawRetainedTypes());
2498 Printer.printMetadata(Name: "globals", MD: N->getRawGlobalVariables());
2499 Printer.printMetadata(Name: "imports", MD: N->getRawImportedEntities());
2500 Printer.printMetadata(Name: "macros", MD: N->getRawMacros());
2501 Printer.printInt(Name: "dwoId", Int: N->getDWOId());
2502 Printer.printBool(Name: "splitDebugInlining", Value: N->getSplitDebugInlining(), Default: true);
2503 Printer.printBool(Name: "debugInfoForProfiling", Value: N->getDebugInfoForProfiling(),
2504 Default: false);
2505 Printer.printNameTableKind(Name: "nameTableKind", NTK: N->getNameTableKind());
2506 Printer.printBool(Name: "rangesBaseAddress", Value: N->getRangesBaseAddress(), Default: false);
2507 Printer.printString(Name: "sysroot", Value: N->getSysRoot());
2508 Printer.printString(Name: "sdk", Value: N->getSDK());
2509 Printer.printDwarfEnum(Name: "dialect", Value: Lang.getDialect(),
2510 toString: dwarf::LanguageDialectString);
2511 Out << ")";
2512}
2513
2514static void writeDISubprogram(raw_ostream &Out, const DISubprogram *N,
2515 AsmWriterContext &WriterCtx) {
2516 Out << "!DISubprogram(";
2517 MDFieldPrinter Printer(Out, WriterCtx);
2518 Printer.printString(Name: "name", Value: N->getName());
2519 Printer.printString(Name: "linkageName", Value: N->getLinkageName());
2520 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2521 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2522 Printer.printInt(Name: "line", Int: N->getLine());
2523 Printer.printMetadata(Name: "type", MD: N->getRawType());
2524 Printer.printInt(Name: "scopeLine", Int: N->getScopeLine());
2525 Printer.printMetadata(Name: "containingType", MD: N->getRawContainingType());
2526 if (N->getVirtuality() != dwarf::DW_VIRTUALITY_none ||
2527 N->getVirtualIndex() != 0)
2528 Printer.printInt(Name: "virtualIndex", Int: N->getVirtualIndex(), ShouldSkipZero: false);
2529 Printer.printInt(Name: "thisAdjustment", Int: N->getThisAdjustment());
2530 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2531 Printer.printDISPFlags(Name: "spFlags", Flags: N->getSPFlags());
2532 Printer.printMetadata(Name: "unit", MD: N->getRawUnit());
2533 Printer.printMetadata(Name: "templateParams", MD: N->getRawTemplateParams());
2534 Printer.printMetadata(Name: "declaration", MD: N->getRawDeclaration());
2535 Printer.printMetadata(Name: "retainedNodes", MD: N->getRawRetainedNodes());
2536 Printer.printMetadata(Name: "thrownTypes", MD: N->getRawThrownTypes());
2537 Printer.printMetadata(Name: "annotations", MD: N->getRawAnnotations());
2538 Printer.printString(Name: "targetFuncName", Value: N->getTargetFuncName());
2539 Printer.printBool(Name: "keyInstructions", Value: N->getKeyInstructionsEnabled(), Default: false);
2540 Out << ")";
2541}
2542
2543static void writeDILexicalBlock(raw_ostream &Out, const DILexicalBlock *N,
2544 AsmWriterContext &WriterCtx) {
2545 Out << "!DILexicalBlock(";
2546 MDFieldPrinter Printer(Out, WriterCtx);
2547 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2548 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2549 Printer.printInt(Name: "line", Int: N->getLine());
2550 Printer.printInt(Name: "column", Int: N->getColumn());
2551 Out << ")";
2552}
2553
2554static void writeDILexicalBlockFile(raw_ostream &Out,
2555 const DILexicalBlockFile *N,
2556 AsmWriterContext &WriterCtx) {
2557 Out << "!DILexicalBlockFile(";
2558 MDFieldPrinter Printer(Out, WriterCtx);
2559 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2560 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2561 Printer.printInt(Name: "discriminator", Int: N->getDiscriminator(),
2562 /* ShouldSkipZero */ false);
2563 Out << ")";
2564}
2565
2566static void writeDINamespace(raw_ostream &Out, const DINamespace *N,
2567 AsmWriterContext &WriterCtx) {
2568 Out << "!DINamespace(";
2569 MDFieldPrinter Printer(Out, WriterCtx);
2570 Printer.printString(Name: "name", Value: N->getName());
2571 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2572 Printer.printBool(Name: "exportSymbols", Value: N->getExportSymbols(), Default: false);
2573 Out << ")";
2574}
2575
2576static void writeDICommonBlock(raw_ostream &Out, const DICommonBlock *N,
2577 AsmWriterContext &WriterCtx) {
2578 Out << "!DICommonBlock(";
2579 MDFieldPrinter Printer(Out, WriterCtx);
2580 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), ShouldSkipNull: false);
2581 Printer.printMetadata(Name: "declaration", MD: N->getRawDecl(), ShouldSkipNull: false);
2582 Printer.printString(Name: "name", Value: N->getName());
2583 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2584 Printer.printInt(Name: "line", Int: N->getLineNo());
2585 Out << ")";
2586}
2587
2588static void writeDIMacro(raw_ostream &Out, const DIMacro *N,
2589 AsmWriterContext &WriterCtx) {
2590 Out << "!DIMacro(";
2591 MDFieldPrinter Printer(Out, WriterCtx);
2592 Printer.printMacinfoType(N);
2593 Printer.printInt(Name: "line", Int: N->getLine());
2594 Printer.printString(Name: "name", Value: N->getName());
2595 Printer.printString(Name: "value", Value: N->getValue());
2596 Out << ")";
2597}
2598
2599static void writeDIMacroFile(raw_ostream &Out, const DIMacroFile *N,
2600 AsmWriterContext &WriterCtx) {
2601 Out << "!DIMacroFile(";
2602 MDFieldPrinter Printer(Out, WriterCtx);
2603 Printer.printInt(Name: "line", Int: N->getLine());
2604 Printer.printMetadata(Name: "file", MD: N->getRawFile(), /* ShouldSkipNull */ false);
2605 Printer.printMetadata(Name: "nodes", MD: N->getRawElements());
2606 Out << ")";
2607}
2608
2609static void writeDIModule(raw_ostream &Out, const DIModule *N,
2610 AsmWriterContext &WriterCtx) {
2611 Out << "!DIModule(";
2612 MDFieldPrinter Printer(Out, WriterCtx);
2613 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2614 Printer.printString(Name: "name", Value: N->getName());
2615 Printer.printString(Name: "configMacros", Value: N->getConfigurationMacros());
2616 Printer.printString(Name: "includePath", Value: N->getIncludePath());
2617 Printer.printString(Name: "apinotes", Value: N->getAPINotesFile());
2618 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2619 Printer.printInt(Name: "line", Int: N->getLineNo());
2620 Printer.printBool(Name: "isDecl", Value: N->getIsDecl(), /* Default */ false);
2621 Out << ")";
2622}
2623
2624static void writeDITemplateTypeParameter(raw_ostream &Out,
2625 const DITemplateTypeParameter *N,
2626 AsmWriterContext &WriterCtx) {
2627 Out << "!DITemplateTypeParameter(";
2628 MDFieldPrinter Printer(Out, WriterCtx);
2629 Printer.printString(Name: "name", Value: N->getName());
2630 Printer.printMetadata(Name: "type", MD: N->getRawType(), /* ShouldSkipNull */ false);
2631 Printer.printBool(Name: "defaulted", Value: N->isDefault(), /* Default= */ false);
2632 Out << ")";
2633}
2634
2635static void writeDITemplateValueParameter(raw_ostream &Out,
2636 const DITemplateValueParameter *N,
2637 AsmWriterContext &WriterCtx) {
2638 Out << "!DITemplateValueParameter(";
2639 MDFieldPrinter Printer(Out, WriterCtx);
2640 if (N->getTag() != dwarf::DW_TAG_template_value_parameter)
2641 Printer.printTag(N);
2642 Printer.printString(Name: "name", Value: N->getName());
2643 Printer.printMetadata(Name: "type", MD: N->getRawType());
2644 Printer.printBool(Name: "defaulted", Value: N->isDefault(), /* Default= */ false);
2645 Printer.printMetadata(Name: "value", MD: N->getValue(), /* ShouldSkipNull */ false);
2646 Out << ")";
2647}
2648
2649static void writeDIGlobalVariable(raw_ostream &Out, const DIGlobalVariable *N,
2650 AsmWriterContext &WriterCtx) {
2651 Out << "!DIGlobalVariable(";
2652 MDFieldPrinter Printer(Out, WriterCtx);
2653 Printer.printString(Name: "name", Value: N->getName());
2654 Printer.printString(Name: "linkageName", Value: N->getLinkageName());
2655 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2656 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2657 Printer.printInt(Name: "line", Int: N->getLine());
2658 Printer.printMetadata(Name: "type", MD: N->getRawType());
2659 Printer.printBool(Name: "isLocal", Value: N->isLocalToUnit());
2660 Printer.printBool(Name: "isDefinition", Value: N->isDefinition());
2661 Printer.printMetadata(Name: "declaration", MD: N->getRawStaticDataMemberDeclaration());
2662 Printer.printMetadata(Name: "templateParams", MD: N->getRawTemplateParams());
2663 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2664 Printer.printMetadata(Name: "annotations", MD: N->getRawAnnotations());
2665 Out << ")";
2666}
2667
2668static void writeDILocalVariable(raw_ostream &Out, const DILocalVariable *N,
2669 AsmWriterContext &WriterCtx) {
2670 Out << "!DILocalVariable(";
2671 MDFieldPrinter Printer(Out, WriterCtx);
2672 Printer.printString(Name: "name", Value: N->getName());
2673 Printer.printInt(Name: "arg", Int: N->getArg());
2674 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2675 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2676 Printer.printInt(Name: "line", Int: N->getLine());
2677 Printer.printMetadata(Name: "type", MD: N->getRawType());
2678 Printer.printDIFlags(Name: "flags", Flags: N->getFlags());
2679 Printer.printInt(Name: "align", Int: N->getAlignInBits());
2680 Printer.printMetadata(Name: "annotations", MD: N->getRawAnnotations());
2681 Out << ")";
2682}
2683
2684static void writeDILabel(raw_ostream &Out, const DILabel *N,
2685 AsmWriterContext &WriterCtx) {
2686 Out << "!DILabel(";
2687 MDFieldPrinter Printer(Out, WriterCtx);
2688 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2689 Printer.printString(Name: "name", Value: N->getName());
2690 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2691 Printer.printInt(Name: "line", Int: N->getLine(), /* ShouldSkipZero */ false);
2692 Printer.printInt(Name: "column", Int: N->getColumn());
2693 Printer.printBool(Name: "isArtificial", Value: N->isArtificial(), Default: false);
2694 if (N->getCoroSuspendIdx())
2695 Printer.printInt(Name: "coroSuspendIdx", Int: *N->getCoroSuspendIdx(),
2696 /* ShouldSkipZero */ false);
2697 Out << ")";
2698}
2699
2700static void writeDIExpression(raw_ostream &Out, const DIExpression *N,
2701 AsmWriterContext &WriterCtx) {
2702 Out << "!DIExpression(";
2703 ListSeparator FS;
2704 if (N->isValid()) {
2705 for (const DIExpression::ExprOperand &Op : N->expr_ops()) {
2706 auto OpStr = dwarf::OperationEncodingString(Encoding: Op.getOp());
2707 assert(!OpStr.empty() && "Expected valid opcode");
2708
2709 Out << FS << OpStr;
2710 if (auto Convert = dyn_cast<DIExpression::ConvertOp>(Val: Op)) {
2711 Out << FS << Convert.getBitSize();
2712 Out << FS << dwarf::AttributeEncodingString(Encoding: Convert.getEncoding());
2713 } else {
2714 for (unsigned A = 0, AE = Op.getNumArgs(); A != AE; ++A)
2715 Out << FS << Op.getArg(I: A);
2716 }
2717 }
2718 } else {
2719 for (const auto &I : N->getElements())
2720 Out << FS << I;
2721 }
2722 Out << ")";
2723}
2724
2725static void writeDIArgList(raw_ostream &Out, const DIArgList *N,
2726 AsmWriterContext &WriterCtx,
2727 bool FromValue = false) {
2728 assert(FromValue &&
2729 "Unexpected DIArgList metadata outside of value argument");
2730 Out << "!DIArgList(";
2731 ListSeparator FS;
2732 MDFieldPrinter Printer(Out, WriterCtx);
2733 for (const Metadata *Arg : N->getArgs()) {
2734 Out << FS;
2735 writeAsOperandInternal(Out, MD: Arg, WriterCtx, FromValue: true);
2736 }
2737 Out << ")";
2738}
2739
2740static void writeDIGlobalVariableExpression(raw_ostream &Out,
2741 const DIGlobalVariableExpression *N,
2742 AsmWriterContext &WriterCtx) {
2743 Out << "!DIGlobalVariableExpression(";
2744 MDFieldPrinter Printer(Out, WriterCtx);
2745 Printer.printMetadata(Name: "var", MD: N->getVariable());
2746 Printer.printMetadata(Name: "expr", MD: N->getExpression());
2747 Out << ")";
2748}
2749
2750static void writeDIObjCProperty(raw_ostream &Out, const DIObjCProperty *N,
2751 AsmWriterContext &WriterCtx) {
2752 Out << "!DIObjCProperty(";
2753 MDFieldPrinter Printer(Out, WriterCtx);
2754 Printer.printString(Name: "name", Value: N->getName());
2755 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2756 Printer.printInt(Name: "line", Int: N->getLine());
2757 Printer.printString(Name: "setter", Value: N->getSetterName());
2758 Printer.printString(Name: "getter", Value: N->getGetterName());
2759 Printer.printInt(Name: "attributes", Int: N->getAttributes());
2760 Printer.printMetadata(Name: "type", MD: N->getRawType());
2761 Out << ")";
2762}
2763
2764static void writeDIProperty(raw_ostream &Out, const DIProperty *N,
2765 AsmWriterContext &WriterCtx) {
2766 Out << "!DIProperty(";
2767 MDFieldPrinter Printer(Out, WriterCtx);
2768 Printer.printString(Name: "name", Value: N->getName());
2769 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2770 Printer.printInt(Name: "line", Int: N->getLine());
2771 Printer.printMetadata(Name: "type", MD: N->getRawType());
2772 Printer.printMetadata(Name: "backing_storage", MD: N->getRawBackingStorage());
2773 Out << ")";
2774}
2775
2776static void writeDIImportedEntity(raw_ostream &Out, const DIImportedEntity *N,
2777 AsmWriterContext &WriterCtx) {
2778 Out << "!DIImportedEntity(";
2779 MDFieldPrinter Printer(Out, WriterCtx);
2780 Printer.printTag(N);
2781 Printer.printString(Name: "name", Value: N->getName());
2782 Printer.printMetadata(Name: "scope", MD: N->getRawScope(), /* ShouldSkipNull */ false);
2783 Printer.printMetadata(Name: "entity", MD: N->getRawEntity());
2784 Printer.printMetadata(Name: "file", MD: N->getRawFile());
2785 Printer.printInt(Name: "line", Int: N->getLine());
2786 Printer.printMetadata(Name: "elements", MD: N->getRawElements());
2787 Out << ")";
2788}
2789
2790static void writeMDNodeBodyInternal(raw_ostream &Out, const MDNode *Node,
2791 AsmWriterContext &Ctx) {
2792 if (Node->isDistinct())
2793 Out << "distinct ";
2794 else if (Node->isTemporary())
2795 Out << "<temporary!> "; // Handle broken code.
2796
2797 switch (Node->getMetadataID()) {
2798 default:
2799 llvm_unreachable("Expected uniquable MDNode");
2800#define HANDLE_MDNODE_LEAF(CLASS) \
2801 case Metadata::CLASS##Kind: \
2802 write##CLASS(Out, cast<CLASS>(Node), Ctx); \
2803 break;
2804#include "llvm/IR/Metadata.def"
2805 }
2806}
2807
2808// Full implementation of printing a Value as an operand with support for
2809// TypePrinting, etc.
2810static void writeAsOperandInternal(raw_ostream &Out, const Value *V,
2811 AsmWriterContext &WriterCtx,
2812 bool PrintType) {
2813 if (PrintType) {
2814 WriterCtx.TypePrinter->print(Ty: V->getType(), OS&: Out);
2815 Out << ' ';
2816 }
2817
2818 if (V->hasName()) {
2819 printLLVMName(OS&: Out, V);
2820 return;
2821 }
2822
2823 const auto *CV = dyn_cast<Constant>(Val: V);
2824 if (CV && !isa<GlobalValue>(Val: CV)) {
2825 assert(WriterCtx.TypePrinter && "Constants require TypePrinting!");
2826 writeConstantInternal(Out, CV, WriterCtx);
2827 return;
2828 }
2829
2830 if (const auto *IA = dyn_cast<InlineAsm>(Val: V)) {
2831 Out << "asm ";
2832 if (IA->hasSideEffects())
2833 Out << "sideeffect ";
2834 if (IA->isAlignStack())
2835 Out << "alignstack ";
2836 // We don't emit the AD_ATT dialect as it's the assumed default.
2837 if (IA->getDialect() == InlineAsm::AD_Intel)
2838 Out << "inteldialect ";
2839 if (IA->canThrow())
2840 Out << "unwind ";
2841 Out << '"';
2842 printEscapedString(Name: IA->getAsmString(), Out);
2843 Out << "\", \"";
2844 printEscapedString(Name: IA->getConstraintString(), Out);
2845 Out << '"';
2846 return;
2847 }
2848
2849 if (auto *MD = dyn_cast<MetadataAsValue>(Val: V)) {
2850 writeAsOperandInternal(Out, MD: MD->getMetadata(), WriterCtx,
2851 /* FromValue */ true);
2852 return;
2853 }
2854
2855 char Prefix = '%';
2856 int Slot;
2857 auto *Machine = WriterCtx.Machine;
2858 // If we have a SlotTracker, use it.
2859 if (Machine) {
2860 if (const auto *GV = dyn_cast<GlobalValue>(Val: V)) {
2861 Slot = Machine->getGlobalSlot(V: GV);
2862 Prefix = '@';
2863 } else {
2864 Slot = Machine->getLocalSlot(V);
2865
2866 // If the local value didn't succeed, then we may be referring to a value
2867 // from a different function. Translate it, as this can happen when using
2868 // address of blocks.
2869 if (Slot == -1)
2870 if ((Machine = createSlotTracker(V))) {
2871 Slot = Machine->getLocalSlot(V);
2872 delete Machine;
2873 }
2874 }
2875 } else if ((Machine = createSlotTracker(V))) {
2876 // Otherwise, create one to get the # and then destroy it.
2877 if (const auto *GV = dyn_cast<GlobalValue>(Val: V)) {
2878 Slot = Machine->getGlobalSlot(V: GV);
2879 Prefix = '@';
2880 } else {
2881 Slot = Machine->getLocalSlot(V);
2882 }
2883 delete Machine;
2884 Machine = nullptr;
2885 } else {
2886 Slot = -1;
2887 }
2888
2889 if (Slot != -1)
2890 Out << Prefix << Slot;
2891 else
2892 Out << "<badref>";
2893}
2894
2895static void writeAsOperandInternal(raw_ostream &Out, const Metadata *MD,
2896 AsmWriterContext &WriterCtx,
2897 bool FromValue) {
2898 // Write DIExpressions and DIArgLists inline when used as a value. Improves
2899 // readability of debug info intrinsics.
2900 if (const auto *Expr = dyn_cast<DIExpression>(Val: MD)) {
2901 writeDIExpression(Out, N: Expr, WriterCtx);
2902 return;
2903 }
2904 if (const auto *ArgList = dyn_cast<DIArgList>(Val: MD)) {
2905 writeDIArgList(Out, N: ArgList, WriterCtx, FromValue);
2906 return;
2907 }
2908
2909 if (const auto *N = dyn_cast<MDNode>(Val: MD)) {
2910 if (const auto *Loc = dyn_cast<DILocation>(Val: N);
2911 Loc && WriterCtx.MST &&
2912 WriterCtx.MST->shouldPrintDebugLocationInline(Loc)) {
2913 writeDILocation(Out, DL: Loc, WriterCtx);
2914 return;
2915 }
2916
2917 std::unique_ptr<SlotTracker> MachineStorage;
2918 SaveAndRestore SARMachine(WriterCtx.Machine);
2919 if (!WriterCtx.Machine) {
2920 MachineStorage = std::make_unique<SlotTracker>(args&: WriterCtx.Context);
2921 WriterCtx.Machine = MachineStorage.get();
2922 }
2923 int Slot = WriterCtx.Machine->getMetadataSlot(N);
2924 if (Slot == -1) {
2925 if (const auto *Loc = dyn_cast<DILocation>(Val: N)) {
2926 writeDILocation(Out, DL: Loc, WriterCtx);
2927 return;
2928 }
2929 // Give the pointer value instead of "badref", since this comes up all
2930 // the time when debugging.
2931 Out << "<" << N << ">";
2932 } else
2933 Out << '!' << Slot;
2934 return;
2935 }
2936
2937 if (const auto *MDS = dyn_cast<MDString>(Val: MD)) {
2938 Out << "!\"";
2939 printEscapedString(Name: MDS->getString(), Out);
2940 Out << '"';
2941 return;
2942 }
2943
2944 auto *V = cast<ValueAsMetadata>(Val: MD);
2945 assert(WriterCtx.TypePrinter && "TypePrinter required for metadata values");
2946 assert((FromValue || !isa<LocalAsMetadata>(V)) &&
2947 "Unexpected function-local metadata outside of value argument");
2948
2949 writeAsOperandInternal(Out, V: V->getValue(), WriterCtx, /*PrintType=*/true);
2950}
2951
2952namespace {
2953
2954class AssemblyWriter {
2955 formatted_raw_ostream &Out;
2956 const Module *TheModule = nullptr;
2957 const ModuleSummaryIndex *TheIndex = nullptr;
2958 std::unique_ptr<SlotTracker> SlotTrackerStorage;
2959 SlotTracker &Machine;
2960 TypePrinting TypePrinter;
2961 AssemblyAnnotationWriter *AnnotationWriter = nullptr;
2962 bool IsForDebug;
2963 bool ShouldPreserveUseListOrder;
2964 UseListOrderMap UseListOrders;
2965 SmallVector<StringRef, 8> MDNames;
2966 /// Synchronization scope names registered with LLVMContext.
2967 SmallVector<StringRef, 8> SSNs;
2968 DenseMap<const GlobalValueSummary *, GlobalValue::GUID> SummaryToGUIDMap;
2969
2970public:
2971 /// Construct an AssemblyWriter with an external SlotTracker
2972 AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac, const Module *M,
2973 AssemblyAnnotationWriter *AAW, bool IsForDebug,
2974 bool ShouldPreserveUseListOrder = false);
2975
2976 AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
2977 const ModuleSummaryIndex *Index, bool IsForDebug);
2978
2979 AsmWriterContext getContext() {
2980 return AsmWriterContext(&TypePrinter, &Machine, TheModule);
2981 }
2982
2983 void printMDNodeBody(const MDNode *MD);
2984 void printNamedMDNode(const NamedMDNode *NMD);
2985
2986 void printModule(const Module *M);
2987
2988 void writeOperand(const Value *Op, bool PrintType);
2989 void writeParamOperand(const Value *Operand, AttributeSet Attrs);
2990 void writeOperandBundles(const CallBase *Call);
2991 void writeSyncScope(const LLVMContext &Context,
2992 SyncScope::ID SSID);
2993 void writeAtomic(const LLVMContext &Context,
2994 AtomicOrdering Ordering,
2995 SyncScope::ID SSID);
2996 void writeAtomicCmpXchg(const LLVMContext &Context,
2997 AtomicOrdering SuccessOrdering,
2998 AtomicOrdering FailureOrdering,
2999 SyncScope::ID SSID);
3000
3001 void writeAllMDNodes();
3002 void writeMDNode(unsigned Slot, const MDNode *Node);
3003 void writeAttribute(const Attribute &Attr, bool InAttrGroup = false);
3004 void writeAttributeSet(const AttributeSet &AttrSet, bool InAttrGroup = false);
3005 void writeAllAttributeGroups();
3006
3007 void printTypeIdentities();
3008 void printGlobal(const GlobalVariable *GV);
3009 void printAlias(const GlobalAlias *GA);
3010 void printIFunc(const GlobalIFunc *GI);
3011 void printComdat(const Comdat *C);
3012 void printFunction(const Function *F);
3013 void printArgument(const Argument *FA, AttributeSet Attrs);
3014 void printBasicBlock(const BasicBlock *BB);
3015 void printInstructionLine(const Instruction &I);
3016 void printInstruction(const Instruction &I);
3017 void printDbgMarker(const DbgMarker &DPI);
3018 void printDbgVariableRecord(const DbgVariableRecord &DVR);
3019 void printDbgLabelRecord(const DbgLabelRecord &DLR);
3020 void printDbgRecord(const DbgRecord &DR);
3021 void printDbgRecordLine(const DbgRecord &DR);
3022
3023 void printUseListOrder(const Value *V, ArrayRef<unsigned> Shuffle);
3024 void printUseLists(const Function *F);
3025
3026 void printModuleSummaryIndex();
3027 void printSummaryInfo(unsigned Slot, const ValueInfo &VI);
3028 void printSummary(const GlobalValueSummary &Summary);
3029 void printAliasSummary(const AliasSummary *AS);
3030 void printGlobalVarSummary(const GlobalVarSummary *GS);
3031 void printFunctionSummary(const FunctionSummary *FS);
3032 void printTypeIdSummary(const TypeIdSummary &TIS);
3033 void printTypeIdCompatibleVtableSummary(const TypeIdCompatibleVtableInfo &TI);
3034 void printTypeTestResolution(const TypeTestResolution &TTRes);
3035 void printArgs(ArrayRef<uint64_t> Args);
3036 void printWPDRes(const WholeProgramDevirtResolution &WPDRes);
3037 void printTypeIdInfo(const FunctionSummary::TypeIdInfo &TIDInfo);
3038 void printVFuncId(const FunctionSummary::VFuncId VFId);
3039 void printNonConstVCalls(ArrayRef<FunctionSummary::VFuncId> VCallList,
3040 const char *Tag);
3041 void printConstVCalls(ArrayRef<FunctionSummary::ConstVCall> VCallList,
3042 const char *Tag);
3043
3044private:
3045 /// Print out metadata attachments.
3046 void printMetadataAttachments(
3047 const SmallVectorImpl<std::pair<unsigned, MDNode *>> &MDs,
3048 StringRef Separator);
3049
3050 // printInfoComment - Print a little comment after the instruction indicating
3051 // which slot it occupies.
3052 void printInfoComment(const Value &V, bool isMaterializable = false);
3053
3054 // printGCRelocateComment - print comment after call to the gc.relocate
3055 // intrinsic indicating base and derived pointer names.
3056 void printGCRelocateComment(const GCRelocateInst &Relocate);
3057};
3058
3059} // end anonymous namespace
3060
3061AssemblyWriter::AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
3062 const Module *M, AssemblyAnnotationWriter *AAW,
3063 bool IsForDebug, bool ShouldPreserveUseListOrder)
3064 : Out(o), TheModule(M), Machine(Mac), TypePrinter(M), AnnotationWriter(AAW),
3065 IsForDebug(IsForDebug),
3066 ShouldPreserveUseListOrder(
3067 PreserveAssemblyUseListOrder.getNumOccurrences()
3068 ? PreserveAssemblyUseListOrder
3069 : ShouldPreserveUseListOrder) {}
3070
3071AssemblyWriter::AssemblyWriter(formatted_raw_ostream &o, SlotTracker &Mac,
3072 const ModuleSummaryIndex *Index, bool IsForDebug)
3073 : Out(o), TheIndex(Index), Machine(Mac), TypePrinter(/*Module=*/nullptr),
3074 IsForDebug(IsForDebug),
3075 ShouldPreserveUseListOrder(PreserveAssemblyUseListOrder) {}
3076
3077void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType) {
3078 if (!Operand) {
3079 Out << "<null operand!>";
3080 return;
3081 }
3082 auto WriteCtx = getContext();
3083 writeAsOperandInternal(Out, V: Operand, WriterCtx&: WriteCtx, PrintType);
3084}
3085
3086void AssemblyWriter::writeSyncScope(const LLVMContext &Context,
3087 SyncScope::ID SSID) {
3088 switch (SSID) {
3089 case SyncScope::System: {
3090 break;
3091 }
3092 default: {
3093 if (SSNs.empty())
3094 Context.getSyncScopeNames(SSNs);
3095
3096 Out << " syncscope(\"";
3097 printEscapedString(Name: SSNs[SSID], Out);
3098 Out << "\")";
3099 break;
3100 }
3101 }
3102}
3103
3104void AssemblyWriter::writeAtomic(const LLVMContext &Context,
3105 AtomicOrdering Ordering,
3106 SyncScope::ID SSID) {
3107 if (Ordering == AtomicOrdering::NotAtomic)
3108 return;
3109
3110 writeSyncScope(Context, SSID);
3111 Out << " " << toIRString(ao: Ordering);
3112}
3113
3114void AssemblyWriter::writeAtomicCmpXchg(const LLVMContext &Context,
3115 AtomicOrdering SuccessOrdering,
3116 AtomicOrdering FailureOrdering,
3117 SyncScope::ID SSID) {
3118 assert(SuccessOrdering != AtomicOrdering::NotAtomic &&
3119 FailureOrdering != AtomicOrdering::NotAtomic);
3120
3121 writeSyncScope(Context, SSID);
3122 Out << " " << toIRString(ao: SuccessOrdering);
3123 Out << " " << toIRString(ao: FailureOrdering);
3124}
3125
3126void AssemblyWriter::writeParamOperand(const Value *Operand,
3127 AttributeSet Attrs) {
3128 if (!Operand) {
3129 Out << "<null operand!>";
3130 return;
3131 }
3132
3133 // Print the type
3134 TypePrinter.print(Ty: Operand->getType(), OS&: Out);
3135 // Print parameter attributes list
3136 if (Attrs.hasAttributes()) {
3137 Out << ' ';
3138 writeAttributeSet(AttrSet: Attrs);
3139 }
3140 Out << ' ';
3141 // Print the operand
3142 auto WriterCtx = getContext();
3143 writeAsOperandInternal(Out, V: Operand, WriterCtx);
3144}
3145
3146void AssemblyWriter::writeOperandBundles(const CallBase *Call) {
3147 if (!Call->hasOperandBundles())
3148 return;
3149
3150 Out << " [ ";
3151
3152 ListSeparator LS;
3153 for (unsigned i = 0, e = Call->getNumOperandBundles(); i != e; ++i) {
3154 OperandBundleUse BU = Call->getOperandBundleAt(Index: i);
3155
3156 Out << LS << '"';
3157 printEscapedString(Name: BU.getTagName(), Out);
3158 Out << '"';
3159
3160 Out << '(';
3161
3162 ListSeparator InnerLS;
3163 auto WriterCtx = getContext();
3164 for (const auto &Input : BU.Inputs) {
3165 Out << InnerLS;
3166 if (Input == nullptr)
3167 Out << "<null operand bundle!>";
3168 else
3169 writeAsOperandInternal(Out, V: Input, WriterCtx, /*PrintType=*/true);
3170 }
3171
3172 Out << ')';
3173 }
3174
3175 Out << " ]";
3176}
3177
3178void AssemblyWriter::printModule(const Module *M) {
3179 Machine.initializeIfNeeded();
3180
3181 if (ShouldPreserveUseListOrder)
3182 UseListOrders = predictUseListOrder(M);
3183
3184 if (!M->getModuleIdentifier().empty() &&
3185 // Don't print the ID if it will start a new line (which would
3186 // require a comment char before it).
3187 M->getModuleIdentifier().find(c: '\n') == std::string::npos)
3188 Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n";
3189
3190 if (!M->getSourceFileName().empty()) {
3191 Out << "source_filename = \"";
3192 printEscapedString(Name: M->getSourceFileName(), Out);
3193 Out << "\"\n";
3194 }
3195
3196 const std::string &DL = M->getDataLayoutStr();
3197 if (!DL.empty())
3198 Out << "target datalayout = \"" << DL << "\"\n";
3199 if (!M->getTargetTriple().empty())
3200 Out << "target triple = \"" << M->getTargetTriple().str() << "\"\n";
3201
3202 if (M->hasModuleInlineAsm()) {
3203 Out << '\n';
3204
3205 for (const Module::GlobalAsmFragment &Frag : M->getModuleInlineAsm()) {
3206 Out << "module asm";
3207 SmallVector<std::pair<StringRef, StringRef>> Props =
3208 Frag.Props.getAsStrings();
3209 if (!Props.empty()) {
3210 ListSeparator LS;
3211 Out << "(";
3212 for (auto [Key, Value] : Props) {
3213 Out << LS;
3214 Out << Key << ": \"";
3215 printEscapedString(Name: Value, Out);
3216 Out << "\"";
3217 }
3218 Out << ")";
3219 }
3220 Out << "\n";
3221 // Split the string into lines, to make it easier to read the .ll file.
3222 StringRef Asm = Frag.Asm;
3223 do {
3224 StringRef Front;
3225 std::tie(args&: Front, args&: Asm) = Asm.split(Separator: '\n');
3226
3227 // We found a newline, print the portion of the asm string from the
3228 // last newline up to this newline.
3229 Out << " \"";
3230 printEscapedString(Name: Front, Out);
3231 Out << "\"\n";
3232 } while (!Asm.empty());
3233 }
3234 }
3235
3236 printTypeIdentities();
3237
3238 {
3239 // Output each used comdat at its first global object.
3240 SmallPtrSet<const Comdat *, 8> PrintedComdats;
3241 for (const GlobalObject &GO : M->global_objects()) {
3242 const Comdat *C = GO.getComdat();
3243 if (!C || !PrintedComdats.insert(Ptr: C).second)
3244 continue;
3245 Out << '\n';
3246 printComdat(C);
3247 }
3248 }
3249
3250 // Output all globals.
3251 if (!M->global_empty()) Out << '\n';
3252 for (const GlobalVariable &GV : M->globals()) {
3253 printGlobal(GV: &GV); Out << '\n';
3254 }
3255
3256 // Output all aliases.
3257 if (!M->alias_empty()) Out << "\n";
3258 for (const GlobalAlias &GA : M->aliases())
3259 printAlias(GA: &GA);
3260
3261 // Output all ifuncs.
3262 if (!M->ifunc_empty()) Out << "\n";
3263 for (const GlobalIFunc &GI : M->ifuncs())
3264 printIFunc(GI: &GI);
3265
3266 // Output all of the functions.
3267 for (const Function &F : *M) {
3268 Out << '\n';
3269 printFunction(F: &F);
3270 }
3271
3272 // Output global use-lists.
3273 printUseLists(F: nullptr);
3274
3275 // Output all attribute groups.
3276 if (!Machine.as_empty()) {
3277 Out << '\n';
3278 writeAllAttributeGroups();
3279 }
3280
3281 // Output named metadata.
3282 if (!M->named_metadata_empty()) Out << '\n';
3283
3284 for (const NamedMDNode &Node : M->named_metadata())
3285 printNamedMDNode(NMD: &Node);
3286
3287 // Output metadata.
3288 if (!Machine.mdn_empty()) {
3289 Out << '\n';
3290 writeAllMDNodes();
3291 }
3292}
3293
3294void AssemblyWriter::printModuleSummaryIndex() {
3295 assert(TheIndex);
3296 int NumSlots = Machine.initializeIndexIfNeeded();
3297
3298 Out << "\n";
3299
3300 // Print module path entries. To print in order, add paths to a vector
3301 // indexed by module slot.
3302 std::vector<std::pair<std::string, ModuleHash>> moduleVec;
3303 std::string RegularLTOModuleName =
3304 ModuleSummaryIndex::getRegularLTOModuleName();
3305 moduleVec.resize(new_size: TheIndex->modulePaths().size());
3306 for (auto &[ModPath, ModHash] : TheIndex->modulePaths())
3307 moduleVec[Machine.getModulePathSlot(Path: ModPath)] = std::make_pair(
3308 // An empty module path is a special entry for a regular LTO module
3309 // created during the thin link.
3310 x: ModPath.empty() ? RegularLTOModuleName : std::string(ModPath), y: ModHash);
3311
3312 unsigned i = 0;
3313 for (auto &ModPair : moduleVec) {
3314 Out << "^" << i++ << " = module: (";
3315 Out << "path: \"";
3316 printEscapedString(Name: ModPair.first, Out);
3317 Out << "\", hash: (";
3318 ListSeparator FS;
3319 for (auto Hash : ModPair.second)
3320 Out << FS << Hash;
3321 Out << "))\n";
3322 }
3323
3324 // FIXME: Change AliasSummary to hold a ValueInfo instead of summary pointer
3325 // for aliasee (then update BitcodeWriter.cpp and remove get/setAliaseeGUID).
3326 // Sort by GUID for deterministic output matching slot assignment order.
3327 auto SortedGVS = TheIndex->sortedGlobalValueSummariesRange();
3328
3329 for (const auto &GlobalList : SortedGVS) {
3330 auto GUID = GlobalList.first;
3331 for (auto &Summary : GlobalList.second.getSummaryList())
3332 SummaryToGUIDMap[Summary.get()] = GUID;
3333 }
3334
3335 // Print the global value summary entries.
3336 for (const auto &GlobalList : SortedGVS) {
3337 auto GUID = GlobalList.first;
3338 auto VI = TheIndex->getValueInfo(R: GlobalList);
3339 printSummaryInfo(Slot: Machine.getGUIDSlot(GUID), VI);
3340 }
3341
3342 // Print the TypeIdMap entries.
3343 for (const auto &TID : TheIndex->typeIds()) {
3344 Out << "^" << Machine.getTypeIdSlot(Id: TID.second.first)
3345 << " = typeid: (name: \"" << TID.second.first << "\"";
3346 printTypeIdSummary(TIS: TID.second.second);
3347 Out << ") ; guid = " << TID.first << "\n";
3348 }
3349
3350 // Print the TypeIdCompatibleVtableMap entries.
3351 for (auto &TId : TheIndex->typeIdCompatibleVtableMap()) {
3352 auto GUID = GlobalValue::getGUIDAssumingExternalLinkage(GlobalName: TId.first);
3353 Out << "^" << Machine.getTypeIdCompatibleVtableSlot(Id: TId.first)
3354 << " = typeidCompatibleVTable: (name: \"" << TId.first << "\"";
3355 printTypeIdCompatibleVtableSummary(TI: TId.second);
3356 Out << ") ; guid = " << GUID << "\n";
3357 }
3358
3359 // Don't emit flags when it's not really needed (value is zero by default).
3360 if (TheIndex->getFlags()) {
3361 Out << "^" << NumSlots << " = flags: " << TheIndex->getFlags() << "\n";
3362 ++NumSlots;
3363 }
3364
3365 Out << "^" << NumSlots << " = blockcount: " << TheIndex->getBlockCount()
3366 << "\n";
3367}
3368
3369static const char *
3370getWholeProgDevirtResKindName(WholeProgramDevirtResolution::Kind K) {
3371 switch (K) {
3372 case WholeProgramDevirtResolution::Indir:
3373 return "indir";
3374 case WholeProgramDevirtResolution::SingleImpl:
3375 return "singleImpl";
3376 case WholeProgramDevirtResolution::BranchFunnel:
3377 return "branchFunnel";
3378 }
3379 llvm_unreachable("invalid WholeProgramDevirtResolution kind");
3380}
3381
3382static const char *getWholeProgDevirtResByArgKindName(
3383 WholeProgramDevirtResolution::ByArg::Kind K) {
3384 switch (K) {
3385 case WholeProgramDevirtResolution::ByArg::Indir:
3386 return "indir";
3387 case WholeProgramDevirtResolution::ByArg::UniformRetVal:
3388 return "uniformRetVal";
3389 case WholeProgramDevirtResolution::ByArg::UniqueRetVal:
3390 return "uniqueRetVal";
3391 case WholeProgramDevirtResolution::ByArg::VirtualConstProp:
3392 return "virtualConstProp";
3393 }
3394 llvm_unreachable("invalid WholeProgramDevirtResolution::ByArg kind");
3395}
3396
3397static const char *getTTResKindName(TypeTestResolution::Kind K) {
3398 switch (K) {
3399 case TypeTestResolution::Unknown:
3400 return "unknown";
3401 case TypeTestResolution::Unsat:
3402 return "unsat";
3403 case TypeTestResolution::ByteArray:
3404 return "byteArray";
3405 case TypeTestResolution::Inline:
3406 return "inline";
3407 case TypeTestResolution::Single:
3408 return "single";
3409 case TypeTestResolution::AllOnes:
3410 return "allOnes";
3411 }
3412 llvm_unreachable("invalid TypeTestResolution kind");
3413}
3414
3415void AssemblyWriter::printTypeTestResolution(const TypeTestResolution &TTRes) {
3416 Out << "typeTestRes: (kind: " << getTTResKindName(K: TTRes.TheKind)
3417 << ", sizeM1BitWidth: " << TTRes.SizeM1BitWidth;
3418
3419 // The following fields are only used if the target does not support the use
3420 // of absolute symbols to store constants. Print only if non-zero.
3421 if (TTRes.AlignLog2)
3422 Out << ", alignLog2: " << TTRes.AlignLog2;
3423 if (TTRes.SizeM1)
3424 Out << ", sizeM1: " << TTRes.SizeM1;
3425 if (TTRes.BitMask)
3426 // BitMask is uint8_t which causes it to print the corresponding char.
3427 Out << ", bitMask: " << (unsigned)TTRes.BitMask;
3428 if (TTRes.InlineBits)
3429 Out << ", inlineBits: " << TTRes.InlineBits;
3430
3431 Out << ")";
3432}
3433
3434void AssemblyWriter::printTypeIdSummary(const TypeIdSummary &TIS) {
3435 Out << ", summary: (";
3436 printTypeTestResolution(TTRes: TIS.TTRes);
3437 if (!TIS.WPDRes.empty()) {
3438 Out << ", wpdResolutions: (";
3439 ListSeparator FS;
3440 for (auto &WPDRes : TIS.WPDRes) {
3441 Out << FS;
3442 Out << "(offset: " << WPDRes.first << ", ";
3443 printWPDRes(WPDRes: WPDRes.second);
3444 Out << ")";
3445 }
3446 Out << ")";
3447 }
3448 Out << ")";
3449}
3450
3451void AssemblyWriter::printTypeIdCompatibleVtableSummary(
3452 const TypeIdCompatibleVtableInfo &TI) {
3453 Out << ", summary: (";
3454 ListSeparator FS;
3455 for (auto &P : TI) {
3456 Out << FS;
3457 Out << "(offset: " << P.AddressPointOffset << ", ";
3458 Out << "^" << Machine.getGUIDSlot(GUID: P.VTableVI.getGUID());
3459 Out << ")";
3460 }
3461 Out << ")";
3462}
3463
3464void AssemblyWriter::printArgs(ArrayRef<uint64_t> Args) {
3465 Out << "args: (" << llvm::interleaved(R: Args) << ')';
3466}
3467
3468void AssemblyWriter::printWPDRes(const WholeProgramDevirtResolution &WPDRes) {
3469 Out << "wpdRes: (kind: ";
3470 Out << getWholeProgDevirtResKindName(K: WPDRes.TheKind);
3471
3472 if (WPDRes.TheKind == WholeProgramDevirtResolution::SingleImpl)
3473 Out << ", singleImplName: \"" << WPDRes.SingleImplName << "\"";
3474
3475 if (!WPDRes.ResByArg.empty()) {
3476 Out << ", resByArg: (";
3477 ListSeparator FS;
3478 for (auto &ResByArg : WPDRes.ResByArg) {
3479 Out << FS;
3480 printArgs(Args: ResByArg.first);
3481 Out << ", byArg: (kind: ";
3482 Out << getWholeProgDevirtResByArgKindName(K: ResByArg.second.TheKind);
3483 if (ResByArg.second.TheKind ==
3484 WholeProgramDevirtResolution::ByArg::UniformRetVal ||
3485 ResByArg.second.TheKind ==
3486 WholeProgramDevirtResolution::ByArg::UniqueRetVal)
3487 Out << ", info: " << ResByArg.second.Info;
3488
3489 // The following fields are only used if the target does not support the
3490 // use of absolute symbols to store constants. Print only if non-zero.
3491 if (ResByArg.second.Byte || ResByArg.second.Bit)
3492 Out << ", byte: " << ResByArg.second.Byte
3493 << ", bit: " << ResByArg.second.Bit;
3494
3495 Out << ")";
3496 }
3497 Out << ")";
3498 }
3499 Out << ")";
3500}
3501
3502static const char *getSummaryKindName(GlobalValueSummary::SummaryKind SK) {
3503 switch (SK) {
3504 case GlobalValueSummary::AliasKind:
3505 return "alias";
3506 case GlobalValueSummary::FunctionKind:
3507 return "function";
3508 case GlobalValueSummary::GlobalVarKind:
3509 return "variable";
3510 }
3511 llvm_unreachable("invalid summary kind");
3512}
3513
3514void AssemblyWriter::printAliasSummary(const AliasSummary *AS) {
3515 Out << ", aliasee: ";
3516 // The indexes emitted for distributed backends may not include the
3517 // aliasee summary (only if it is being imported directly). Handle
3518 // that case by just emitting "null" as the aliasee.
3519 if (AS->hasAliasee())
3520 Out << "^" << Machine.getGUIDSlot(GUID: SummaryToGUIDMap[&AS->getAliasee()]);
3521 else
3522 Out << "null";
3523}
3524
3525void AssemblyWriter::printGlobalVarSummary(const GlobalVarSummary *GS) {
3526 auto VTableFuncs = GS->vTableFuncs();
3527 Out << ", varFlags: (readonly: " << GS->VarFlags.MaybeReadOnly << ", "
3528 << "writeonly: " << GS->VarFlags.MaybeWriteOnly << ", "
3529 << "constant: " << GS->VarFlags.Constant;
3530 if (!VTableFuncs.empty())
3531 Out << ", "
3532 << "vcall_visibility: " << GS->VarFlags.VCallVisibility;
3533 Out << ")";
3534
3535 if (!VTableFuncs.empty()) {
3536 Out << ", vTableFuncs: (";
3537 ListSeparator FS;
3538 for (auto &P : VTableFuncs) {
3539 Out << FS;
3540 Out << "(virtFunc: ^" << Machine.getGUIDSlot(GUID: P.FuncVI.getGUID())
3541 << ", offset: " << P.VTableOffset;
3542 Out << ")";
3543 }
3544 Out << ")";
3545 }
3546}
3547
3548static std::string getLinkageName(GlobalValue::LinkageTypes LT) {
3549 switch (LT) {
3550 case GlobalValue::ExternalLinkage:
3551 return "external";
3552 case GlobalValue::PrivateLinkage:
3553 return "private";
3554 case GlobalValue::InternalLinkage:
3555 return "internal";
3556 case GlobalValue::LinkOnceAnyLinkage:
3557 return "linkonce";
3558 case GlobalValue::LinkOnceODRLinkage:
3559 return "linkonce_odr";
3560 case GlobalValue::WeakAnyLinkage:
3561 return "weak";
3562 case GlobalValue::WeakODRLinkage:
3563 return "weak_odr";
3564 case GlobalValue::CommonLinkage:
3565 return "common";
3566 case GlobalValue::AppendingLinkage:
3567 return "appending";
3568 case GlobalValue::ExternalWeakLinkage:
3569 return "extern_weak";
3570 case GlobalValue::AvailableExternallyLinkage:
3571 return "available_externally";
3572 }
3573 llvm_unreachable("invalid linkage");
3574}
3575
3576// When printing the linkage types in IR where the ExternalLinkage is
3577// not printed, and other linkage types are expected to be printed with
3578// a space after the name.
3579static std::string getLinkageNameWithSpace(GlobalValue::LinkageTypes LT) {
3580 if (LT == GlobalValue::ExternalLinkage)
3581 return "";
3582 return getLinkageName(LT) + " ";
3583}
3584
3585static const char *getVisibilityName(GlobalValue::VisibilityTypes Vis) {
3586 switch (Vis) {
3587 case GlobalValue::DefaultVisibility:
3588 return "default";
3589 case GlobalValue::HiddenVisibility:
3590 return "hidden";
3591 case GlobalValue::ProtectedVisibility:
3592 return "protected";
3593 }
3594 llvm_unreachable("invalid visibility");
3595}
3596
3597static const char *getImportTypeName(GlobalValueSummary::ImportKind IK) {
3598 switch (IK) {
3599 case GlobalValueSummary::Definition:
3600 return "definition";
3601 case GlobalValueSummary::Declaration:
3602 return "declaration";
3603 }
3604 llvm_unreachable("invalid import kind");
3605}
3606
3607void AssemblyWriter::printFunctionSummary(const FunctionSummary *FS) {
3608 Out << ", insts: " << FS->instCount();
3609 if (FS->fflags().anyFlagSet())
3610 Out << ", " << FS->fflags();
3611
3612 if (!FS->calls().empty()) {
3613 Out << ", calls: (";
3614 ListSeparator IFS;
3615 for (auto &Call : FS->calls()) {
3616 Out << IFS;
3617 Out << "(callee: ^" << Machine.getGUIDSlot(GUID: Call.first.getGUID());
3618 if (Call.second.getHotness() != CalleeInfo::HotnessType::Unknown)
3619 Out << ", hotness: " << getHotnessName(HT: Call.second.getHotness());
3620 // Follow the convention of emitting flags as a boolean value, but only
3621 // emit if true to avoid unnecessary verbosity and test churn.
3622 if (Call.second.HasTailCall)
3623 Out << ", tail: 1";
3624 Out << ")";
3625 }
3626 Out << ")";
3627 }
3628
3629 if (const auto *TIdInfo = FS->getTypeIdInfo())
3630 printTypeIdInfo(TIDInfo: *TIdInfo);
3631
3632 // The AllocationType identifiers capture the profiled context behavior
3633 // reaching a specific static allocation site (possibly cloned).
3634 auto AllocTypeName = [](uint8_t Type) -> const char * {
3635 switch (Type) {
3636 case (uint8_t)AllocationType::None:
3637 return "none";
3638 case (uint8_t)AllocationType::NotCold:
3639 return "notcold";
3640 case (uint8_t)AllocationType::Cold:
3641 return "cold";
3642 case (uint8_t)AllocationType::Hot:
3643 return "hot";
3644 }
3645 llvm_unreachable("Unexpected alloc type");
3646 };
3647
3648 if (!FS->allocs().empty()) {
3649 Out << ", allocs: (";
3650 ListSeparator AFS;
3651 for (auto &AI : FS->allocs()) {
3652 Out << AFS;
3653 Out << "(versions: (";
3654 ListSeparator VFS;
3655 for (auto V : AI.Versions) {
3656 Out << VFS;
3657 Out << AllocTypeName(V);
3658 }
3659 Out << "), memProf: (";
3660 ListSeparator MIBFS;
3661 for (auto &MIB : AI.MIBs) {
3662 Out << MIBFS;
3663 Out << "(type: " << AllocTypeName((uint8_t)MIB.AllocType);
3664 Out << ", stackIds: (";
3665 ListSeparator SIDFS;
3666 for (auto Id : MIB.StackIdIndices) {
3667 Out << SIDFS;
3668 Out << TheIndex->getStackIdAtIndex(Index: Id);
3669 }
3670 Out << "))";
3671 }
3672 Out << "))";
3673 }
3674 Out << ")";
3675 }
3676
3677 if (!FS->callsites().empty()) {
3678 Out << ", callsites: (";
3679 ListSeparator SNFS;
3680 for (auto &CI : FS->callsites()) {
3681 Out << SNFS;
3682 if (CI.Callee)
3683 Out << "(callee: ^" << Machine.getGUIDSlot(GUID: CI.Callee.getGUID());
3684 else
3685 Out << "(callee: null";
3686 Out << ", clones: (";
3687 ListSeparator VFS;
3688 for (auto V : CI.Clones) {
3689 Out << VFS;
3690 Out << V;
3691 }
3692 Out << "), stackIds: (";
3693 ListSeparator SIDFS;
3694 for (auto Id : CI.StackIdIndices) {
3695 Out << SIDFS;
3696 Out << TheIndex->getStackIdAtIndex(Index: Id);
3697 }
3698 Out << "))";
3699 }
3700 Out << ")";
3701 }
3702
3703 auto PrintRange = [&](const ConstantRange &Range) {
3704 Out << "[" << Range.getSignedMin() << ", " << Range.getSignedMax() << "]";
3705 };
3706
3707 if (!FS->paramAccesses().empty()) {
3708 Out << ", params: (";
3709 ListSeparator IFS;
3710 for (auto &PS : FS->paramAccesses()) {
3711 Out << IFS;
3712 Out << "(param: " << PS.ParamNo;
3713 Out << ", offset: ";
3714 PrintRange(PS.Use);
3715 if (!PS.Calls.empty()) {
3716 Out << ", calls: (";
3717 ListSeparator IFS;
3718 for (auto &Call : PS.Calls) {
3719 Out << IFS;
3720 Out << "(callee: ^" << Machine.getGUIDSlot(GUID: Call.Callee.getGUID());
3721 Out << ", param: " << Call.ParamNo;
3722 Out << ", offset: ";
3723 PrintRange(Call.Offsets);
3724 Out << ")";
3725 }
3726 Out << ")";
3727 }
3728 Out << ")";
3729 }
3730 Out << ")";
3731 }
3732}
3733
3734void AssemblyWriter::printTypeIdInfo(
3735 const FunctionSummary::TypeIdInfo &TIDInfo) {
3736 Out << ", typeIdInfo: (";
3737 ListSeparator TIDFS;
3738 if (!TIDInfo.TypeTests.empty()) {
3739 Out << TIDFS;
3740 Out << "typeTests: (";
3741 ListSeparator FS;
3742 for (auto &GUID : TIDInfo.TypeTests) {
3743 auto TidIter = TheIndex->typeIds().equal_range(x: GUID);
3744 if (TidIter.first == TidIter.second) {
3745 Out << FS;
3746 Out << GUID;
3747 continue;
3748 }
3749 // Print all type id that correspond to this GUID.
3750 for (const auto &[GUID, TypeIdPair] : make_range(p: TidIter)) {
3751 Out << FS;
3752 auto Slot = Machine.getTypeIdSlot(Id: TypeIdPair.first);
3753 assert(Slot != -1);
3754 Out << "^" << Slot;
3755 }
3756 }
3757 Out << ")";
3758 }
3759 if (!TIDInfo.TypeTestAssumeVCalls.empty()) {
3760 Out << TIDFS;
3761 printNonConstVCalls(VCallList: TIDInfo.TypeTestAssumeVCalls, Tag: "typeTestAssumeVCalls");
3762 }
3763 if (!TIDInfo.TypeCheckedLoadVCalls.empty()) {
3764 Out << TIDFS;
3765 printNonConstVCalls(VCallList: TIDInfo.TypeCheckedLoadVCalls, Tag: "typeCheckedLoadVCalls");
3766 }
3767 if (!TIDInfo.TypeTestAssumeConstVCalls.empty()) {
3768 Out << TIDFS;
3769 printConstVCalls(VCallList: TIDInfo.TypeTestAssumeConstVCalls,
3770 Tag: "typeTestAssumeConstVCalls");
3771 }
3772 if (!TIDInfo.TypeCheckedLoadConstVCalls.empty()) {
3773 Out << TIDFS;
3774 printConstVCalls(VCallList: TIDInfo.TypeCheckedLoadConstVCalls,
3775 Tag: "typeCheckedLoadConstVCalls");
3776 }
3777 Out << ")";
3778}
3779
3780void AssemblyWriter::printVFuncId(const FunctionSummary::VFuncId VFId) {
3781 auto TidIter = TheIndex->typeIds().equal_range(x: VFId.GUID);
3782 if (TidIter.first == TidIter.second) {
3783 Out << "vFuncId: (";
3784 Out << "guid: " << VFId.GUID;
3785 Out << ", offset: " << VFId.Offset;
3786 Out << ")";
3787 return;
3788 }
3789 // Print all type id that correspond to this GUID.
3790 ListSeparator FS;
3791 for (const auto &[GUID, TypeIdPair] : make_range(p: TidIter)) {
3792 Out << FS;
3793 Out << "vFuncId: (";
3794 auto Slot = Machine.getTypeIdSlot(Id: TypeIdPair.first);
3795 assert(Slot != -1);
3796 Out << "^" << Slot;
3797 Out << ", offset: " << VFId.Offset;
3798 Out << ")";
3799 }
3800}
3801
3802void AssemblyWriter::printNonConstVCalls(
3803 ArrayRef<FunctionSummary::VFuncId> VCallList, const char *Tag) {
3804 Out << Tag << ": (";
3805 ListSeparator FS;
3806 for (auto &VFuncId : VCallList) {
3807 Out << FS;
3808 printVFuncId(VFId: VFuncId);
3809 }
3810 Out << ")";
3811}
3812
3813void AssemblyWriter::printConstVCalls(
3814 ArrayRef<FunctionSummary::ConstVCall> VCallList, const char *Tag) {
3815 Out << Tag << ": (";
3816 ListSeparator FS;
3817 for (auto &ConstVCall : VCallList) {
3818 Out << FS;
3819 Out << "(";
3820 printVFuncId(VFId: ConstVCall.VFunc);
3821 if (!ConstVCall.Args.empty()) {
3822 Out << ", ";
3823 printArgs(Args: ConstVCall.Args);
3824 }
3825 Out << ")";
3826 }
3827 Out << ")";
3828}
3829
3830void AssemblyWriter::printSummary(const GlobalValueSummary &Summary) {
3831 GlobalValueSummary::GVFlags GVFlags = Summary.flags();
3832 GlobalValue::LinkageTypes LT = (GlobalValue::LinkageTypes)GVFlags.Linkage;
3833 Out << getSummaryKindName(SK: Summary.getSummaryKind()) << ": ";
3834 Out << "(module: ^" << Machine.getModulePathSlot(Path: Summary.modulePath())
3835 << ", flags: (";
3836 Out << "linkage: " << getLinkageName(LT);
3837 Out << ", visibility: "
3838 << getVisibilityName(Vis: (GlobalValue::VisibilityTypes)GVFlags.Visibility);
3839 Out << ", notEligibleToImport: " << GVFlags.NotEligibleToImport;
3840 Out << ", live: " << GVFlags.Live;
3841 Out << ", dsoLocal: " << GVFlags.DSOLocal;
3842 Out << ", canAutoHide: " << GVFlags.CanAutoHide;
3843 Out << ", importType: "
3844 << getImportTypeName(IK: GlobalValueSummary::ImportKind(GVFlags.ImportType));
3845 Out << ", noRenameOnPromotion: " << GVFlags.NoRenameOnPromotion;
3846 Out << ")";
3847
3848 if (Summary.getSummaryKind() == GlobalValueSummary::AliasKind)
3849 printAliasSummary(AS: cast<AliasSummary>(Val: &Summary));
3850 else if (Summary.getSummaryKind() == GlobalValueSummary::FunctionKind)
3851 printFunctionSummary(FS: cast<FunctionSummary>(Val: &Summary));
3852 else
3853 printGlobalVarSummary(GS: cast<GlobalVarSummary>(Val: &Summary));
3854
3855 auto RefList = Summary.refs();
3856 if (!RefList.empty()) {
3857 Out << ", refs: (";
3858 ListSeparator FS;
3859 for (auto &Ref : RefList) {
3860 Out << FS;
3861 if (Ref.isReadOnly())
3862 Out << "readonly ";
3863 else if (Ref.isWriteOnly())
3864 Out << "writeonly ";
3865 Out << "^" << Machine.getGUIDSlot(GUID: Ref.getGUID());
3866 }
3867 Out << ")";
3868 }
3869
3870 Out << ")";
3871}
3872
3873void AssemblyWriter::printSummaryInfo(unsigned Slot, const ValueInfo &VI) {
3874 Out << "^" << Slot << " = gv: (";
3875 if (VI.hasName() && !VI.name().empty())
3876 Out << "name: \"" << VI.name() << "\"";
3877 else
3878 Out << "guid: " << VI.getGUID();
3879 if (!VI.getSummaryList().empty()) {
3880 Out << ", summaries: (";
3881 ListSeparator FS;
3882 for (auto &Summary : VI.getSummaryList()) {
3883 Out << FS;
3884 printSummary(Summary: *Summary);
3885 }
3886 Out << ")";
3887 }
3888 Out << ")";
3889 if (VI.hasName() && !VI.name().empty())
3890 Out << " ; guid = " << VI.getGUID();
3891 Out << "\n";
3892}
3893
3894static void printMetadataIdentifier(StringRef Name,
3895 formatted_raw_ostream &Out) {
3896 if (Name.empty()) {
3897 Out << "<empty name> ";
3898 } else {
3899 unsigned char FirstC = static_cast<unsigned char>(Name[0]);
3900 if (isalpha(FirstC) || FirstC == '-' || FirstC == '$' || FirstC == '.' ||
3901 FirstC == '_')
3902 Out << FirstC;
3903 else
3904 Out << '\\' << hexdigit(X: FirstC >> 4) << hexdigit(X: FirstC & 0x0F);
3905 for (unsigned i = 1, e = Name.size(); i != e; ++i) {
3906 unsigned char C = Name[i];
3907 if (isalnum(C) || C == '-' || C == '$' || C == '.' || C == '_')
3908 Out << C;
3909 else
3910 Out << '\\' << hexdigit(X: C >> 4) << hexdigit(X: C & 0x0F);
3911 }
3912 }
3913}
3914
3915void AssemblyWriter::printNamedMDNode(const NamedMDNode *NMD) {
3916 Out << '!';
3917 printMetadataIdentifier(Name: NMD->getName(), Out);
3918 Out << " = !{";
3919 ListSeparator LS;
3920 for (const MDNode *Op : NMD->operands()) {
3921 Out << LS;
3922 // Write DIExpressions inline.
3923 // FIXME: Ban DIExpressions in NamedMDNodes, they will serve no purpose.
3924 if (auto *Expr = dyn_cast<DIExpression>(Val: Op)) {
3925 writeDIExpression(Out, N: Expr, WriterCtx&: AsmWriterContext::getEmpty());
3926 continue;
3927 }
3928
3929 int Slot = Machine.getMetadataSlot(N: Op);
3930 if (Slot == -1)
3931 Out << "<badref>";
3932 else
3933 Out << '!' << Slot;
3934 }
3935 Out << "}\n";
3936}
3937
3938static void printVisibility(GlobalValue::VisibilityTypes Vis,
3939 formatted_raw_ostream &Out) {
3940 switch (Vis) {
3941 case GlobalValue::DefaultVisibility: break;
3942 case GlobalValue::HiddenVisibility: Out << "hidden "; break;
3943 case GlobalValue::ProtectedVisibility: Out << "protected "; break;
3944 }
3945}
3946
3947static void printDSOLocation(const GlobalValue &GV,
3948 formatted_raw_ostream &Out) {
3949 if (GV.isDSOLocal() && !GV.isImplicitDSOLocal())
3950 Out << "dso_local ";
3951}
3952
3953static void printDLLStorageClass(GlobalValue::DLLStorageClassTypes SCT,
3954 formatted_raw_ostream &Out) {
3955 switch (SCT) {
3956 case GlobalValue::DefaultStorageClass: break;
3957 case GlobalValue::DLLImportStorageClass: Out << "dllimport "; break;
3958 case GlobalValue::DLLExportStorageClass: Out << "dllexport "; break;
3959 }
3960}
3961
3962static void printThreadLocalModel(GlobalVariable::ThreadLocalMode TLM,
3963 formatted_raw_ostream &Out) {
3964 switch (TLM) {
3965 case GlobalVariable::NotThreadLocal:
3966 break;
3967 case GlobalVariable::GeneralDynamicTLSModel:
3968 Out << "thread_local ";
3969 break;
3970 case GlobalVariable::LocalDynamicTLSModel:
3971 Out << "thread_local(localdynamic) ";
3972 break;
3973 case GlobalVariable::InitialExecTLSModel:
3974 Out << "thread_local(initialexec) ";
3975 break;
3976 case GlobalVariable::LocalExecTLSModel:
3977 Out << "thread_local(localexec) ";
3978 break;
3979 }
3980}
3981
3982static StringRef getUnnamedAddrEncoding(GlobalVariable::UnnamedAddr UA) {
3983 switch (UA) {
3984 case GlobalVariable::UnnamedAddr::None:
3985 return "";
3986 case GlobalVariable::UnnamedAddr::Local:
3987 return "local_unnamed_addr";
3988 case GlobalVariable::UnnamedAddr::Global:
3989 return "unnamed_addr";
3990 }
3991 llvm_unreachable("Unknown UnnamedAddr");
3992}
3993
3994static void maybePrintComdat(formatted_raw_ostream &Out,
3995 const GlobalObject &GO) {
3996 const Comdat *C = GO.getComdat();
3997 if (!C)
3998 return;
3999
4000 if (isa<GlobalVariable>(Val: GO))
4001 Out << ',';
4002 Out << " comdat";
4003
4004 if (GO.getName() == C->getName())
4005 return;
4006
4007 Out << '(';
4008 printLLVMName(OS&: Out, Name: C->getName(), Prefix: ComdatPrefix);
4009 Out << ')';
4010}
4011
4012void AssemblyWriter::printGlobal(const GlobalVariable *GV) {
4013 if (GV->isMaterializable())
4014 Out << "; Materializable\n";
4015
4016 AsmWriterContext WriterCtx(&TypePrinter, &Machine, GV->getParent());
4017 writeAsOperandInternal(Out, V: GV, WriterCtx);
4018 Out << " = ";
4019
4020 if (!GV->hasInitializer() && GV->hasExternalLinkage())
4021 Out << "external ";
4022
4023 Out << getLinkageNameWithSpace(LT: GV->getLinkage());
4024 printDSOLocation(GV: *GV, Out);
4025 printVisibility(Vis: GV->getVisibility(), Out);
4026 printDLLStorageClass(SCT: GV->getDLLStorageClass(), Out);
4027 printThreadLocalModel(TLM: GV->getThreadLocalMode(), Out);
4028 StringRef UA = getUnnamedAddrEncoding(UA: GV->getUnnamedAddr());
4029 if (!UA.empty())
4030 Out << UA << ' ';
4031
4032 printAddressSpace(M: GV->getParent(), AS: GV->getType()->getAddressSpace(), OS&: Out,
4033 /*Prefix=*/"", /*Suffix=*/" ");
4034 if (GV->isExternallyInitialized()) Out << "externally_initialized ";
4035 Out << (GV->isConstant() ? "constant " : "global ");
4036 TypePrinter.print(Ty: GV->getValueType(), OS&: Out);
4037
4038 if (GV->hasInitializer()) {
4039 Out << ' ';
4040 writeOperand(Operand: GV->getInitializer(), PrintType: false);
4041 }
4042
4043 if (GV->hasSection()) {
4044 Out << ", section \"";
4045 printEscapedString(Name: GV->getSection(), Out);
4046 Out << '"';
4047 }
4048 if (GV->hasPartition()) {
4049 Out << ", partition \"";
4050 printEscapedString(Name: GV->getPartition(), Out);
4051 Out << '"';
4052 }
4053 if (auto CM = GV->getCodeModel()) {
4054 Out << ", code_model \"";
4055 switch (*CM) {
4056 case CodeModel::Tiny:
4057 Out << "tiny";
4058 break;
4059 case CodeModel::Small:
4060 Out << "small";
4061 break;
4062 case CodeModel::Kernel:
4063 Out << "kernel";
4064 break;
4065 case CodeModel::Medium:
4066 Out << "medium";
4067 break;
4068 case CodeModel::Large:
4069 Out << "large";
4070 break;
4071 }
4072 Out << '"';
4073 }
4074
4075 using SanitizerMetadata = llvm::GlobalValue::SanitizerMetadata;
4076 if (GV->hasSanitizerMetadata()) {
4077 SanitizerMetadata MD = GV->getSanitizerMetadata();
4078 if (MD.NoAddress)
4079 Out << ", no_sanitize_address";
4080 if (MD.NoHWAddress)
4081 Out << ", no_sanitize_hwaddress";
4082 if (MD.Memtag)
4083 Out << ", sanitize_memtag";
4084 if (MD.IsDynInit)
4085 Out << ", sanitize_address_dyninit";
4086 }
4087
4088 maybePrintComdat(Out, GO: *GV);
4089 if (MaybeAlign A = GV->getAlign())
4090 Out << ", align " << A->value();
4091
4092 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
4093 GV->getAllMetadata(MDs);
4094 printMetadataAttachments(MDs, Separator: ", ");
4095
4096 auto Attrs = GV->getAttributes();
4097 if (Attrs.hasAttributes())
4098 Out << " #" << Machine.getAttributeGroupSlot(AS: Attrs);
4099
4100 printInfoComment(V: *GV, isMaterializable: GV->isMaterializable());
4101}
4102
4103void AssemblyWriter::printAlias(const GlobalAlias *GA) {
4104 if (GA->isMaterializable())
4105 Out << "; Materializable\n";
4106
4107 AsmWriterContext WriterCtx(&TypePrinter, &Machine, GA->getParent());
4108 writeAsOperandInternal(Out, V: GA, WriterCtx);
4109 Out << " = ";
4110
4111 Out << getLinkageNameWithSpace(LT: GA->getLinkage());
4112 printDSOLocation(GV: *GA, Out);
4113 printVisibility(Vis: GA->getVisibility(), Out);
4114 printDLLStorageClass(SCT: GA->getDLLStorageClass(), Out);
4115 printThreadLocalModel(TLM: GA->getThreadLocalMode(), Out);
4116 StringRef UA = getUnnamedAddrEncoding(UA: GA->getUnnamedAddr());
4117 if (!UA.empty())
4118 Out << UA << ' ';
4119
4120 Out << "alias ";
4121
4122 TypePrinter.print(Ty: GA->getValueType(), OS&: Out);
4123 Out << ", ";
4124
4125 if (const Constant *Aliasee = GA->getAliasee()) {
4126 writeOperand(Operand: Aliasee, PrintType: !isa<ConstantExpr>(Val: Aliasee));
4127 } else {
4128 TypePrinter.print(Ty: GA->getType(), OS&: Out);
4129 Out << " <<NULL ALIASEE>>";
4130 }
4131
4132 if (GA->hasPartition()) {
4133 Out << ", partition \"";
4134 printEscapedString(Name: GA->getPartition(), Out);
4135 Out << '"';
4136 }
4137
4138 printInfoComment(V: *GA, isMaterializable: GA->isMaterializable());
4139 Out << '\n';
4140}
4141
4142void AssemblyWriter::printIFunc(const GlobalIFunc *GI) {
4143 if (GI->isMaterializable())
4144 Out << "; Materializable\n";
4145
4146 AsmWriterContext WriterCtx(&TypePrinter, &Machine, GI->getParent());
4147 writeAsOperandInternal(Out, V: GI, WriterCtx);
4148 Out << " = ";
4149
4150 Out << getLinkageNameWithSpace(LT: GI->getLinkage());
4151 printDSOLocation(GV: *GI, Out);
4152 printVisibility(Vis: GI->getVisibility(), Out);
4153
4154 Out << "ifunc ";
4155
4156 TypePrinter.print(Ty: GI->getValueType(), OS&: Out);
4157 Out << ", ";
4158
4159 if (const Constant *Resolver = GI->getResolver()) {
4160 writeOperand(Operand: Resolver, PrintType: !isa<ConstantExpr>(Val: Resolver));
4161 } else {
4162 TypePrinter.print(Ty: GI->getType(), OS&: Out);
4163 Out << " <<NULL RESOLVER>>";
4164 }
4165
4166 if (GI->hasPartition()) {
4167 Out << ", partition \"";
4168 printEscapedString(Name: GI->getPartition(), Out);
4169 Out << '"';
4170 }
4171 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
4172 GI->getAllMetadata(MDs);
4173 if (!MDs.empty()) {
4174 printMetadataAttachments(MDs, Separator: ", ");
4175 }
4176
4177 printInfoComment(V: *GI, isMaterializable: GI->isMaterializable());
4178 Out << '\n';
4179}
4180
4181void AssemblyWriter::printComdat(const Comdat *C) {
4182 C->print(OS&: Out);
4183}
4184
4185void AssemblyWriter::printTypeIdentities() {
4186 if (TypePrinter.empty())
4187 return;
4188
4189 Out << '\n';
4190
4191 // Emit all numbered types.
4192 auto &NumberedTypes = TypePrinter.getNumberedTypes();
4193 for (unsigned I = 0, E = NumberedTypes.size(); I != E; ++I) {
4194 Out << '%' << I << " = type ";
4195
4196 // Make sure we print out at least one level of the type structure, so
4197 // that we do not get %2 = type %2
4198 TypePrinter.printStructBody(STy: NumberedTypes[I], OS&: Out);
4199 Out << '\n';
4200 }
4201
4202 auto &NamedTypes = TypePrinter.getNamedTypes();
4203 for (StructType *NamedType : NamedTypes) {
4204 printLLVMName(OS&: Out, Name: NamedType->getName(), Prefix: LocalPrefix);
4205 Out << " = type ";
4206
4207 // Make sure we print out at least one level of the type structure, so
4208 // that we do not get %FILE = type %FILE
4209 TypePrinter.printStructBody(STy: NamedType, OS&: Out);
4210 Out << '\n';
4211 }
4212}
4213
4214/// printFunction - Print all aspects of a function.
4215void AssemblyWriter::printFunction(const Function *F) {
4216 if (F->isMaterializable())
4217 Out << "; Materializable\n";
4218 else if (AnnotationWriter)
4219 AnnotationWriter->emitFunctionAnnot(F, Out);
4220
4221 const AttributeList &Attrs = F->getAttributes();
4222 if (Attrs.hasFnAttrs()) {
4223 AttributeSet AS = Attrs.getFnAttrs();
4224 std::string AttrStr;
4225
4226 for (const Attribute &Attr : AS) {
4227 if (!Attr.isStringAttribute()) {
4228 if (!AttrStr.empty()) AttrStr += ' ';
4229 AttrStr += Attr.getAsString();
4230 }
4231 }
4232
4233 if (!AttrStr.empty())
4234 Out << "; Function Attrs: " << AttrStr << '\n';
4235 }
4236
4237 if (F->isIntrinsic() && F->getIntrinsicID() == Intrinsic::not_intrinsic)
4238 Out << "; Unknown intrinsic\n";
4239
4240 Machine.incorporateFunction(F);
4241
4242 if (F->isDeclaration()) {
4243 Out << "declare";
4244 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
4245 F->getAllMetadata(MDs);
4246 printMetadataAttachments(MDs, Separator: " ");
4247 Out << ' ';
4248 } else
4249 Out << "define ";
4250
4251 Out << getLinkageNameWithSpace(LT: F->getLinkage());
4252 printDSOLocation(GV: *F, Out);
4253 printVisibility(Vis: F->getVisibility(), Out);
4254 printDLLStorageClass(SCT: F->getDLLStorageClass(), Out);
4255
4256 // Print the calling convention.
4257 if (F->getCallingConv() != CallingConv::C) {
4258 printCallingConv(cc: F->getCallingConv(), Out);
4259 Out << " ";
4260 }
4261
4262 FunctionType *FT = F->getFunctionType();
4263 if (Attrs.hasRetAttrs())
4264 Out << Attrs.getAsString(Index: AttributeList::ReturnIndex) << ' ';
4265 TypePrinter.print(Ty: F->getReturnType(), OS&: Out);
4266 AsmWriterContext WriterCtx(&TypePrinter, &Machine, F->getParent());
4267 Out << ' ';
4268 writeAsOperandInternal(Out, V: F, WriterCtx);
4269 Out << '(';
4270
4271 // Loop over the arguments, printing them...
4272 if (F->isDeclaration() && !IsForDebug) {
4273 // We're only interested in the type here - don't print argument names.
4274 ListSeparator LS;
4275 for (unsigned I = 0, E = FT->getNumParams(); I != E; ++I) {
4276 Out << LS;
4277 // Output type.
4278 TypePrinter.print(Ty: FT->getParamType(i: I), OS&: Out);
4279
4280 AttributeSet ArgAttrs = Attrs.getParamAttrs(ArgNo: I);
4281 if (ArgAttrs.hasAttributes()) {
4282 Out << ' ';
4283 writeAttributeSet(AttrSet: ArgAttrs);
4284 }
4285 }
4286 } else {
4287 // The arguments are meaningful here, print them in detail.
4288 ListSeparator LS;
4289 for (const Argument &Arg : F->args()) {
4290 Out << LS;
4291 printArgument(FA: &Arg, Attrs: Attrs.getParamAttrs(ArgNo: Arg.getArgNo()));
4292 }
4293 }
4294
4295 // Finish printing arguments...
4296 if (FT->isVarArg()) {
4297 if (FT->getNumParams()) Out << ", ";
4298 Out << "..."; // Output varargs portion of signature!
4299 }
4300 Out << ')';
4301 StringRef UA = getUnnamedAddrEncoding(UA: F->getUnnamedAddr());
4302 if (!UA.empty())
4303 Out << ' ' << UA;
4304 // We print the function address space if it is non-zero or if we are writing
4305 // a module with a non-zero program address space or if there is no valid
4306 // Module* so that the file can be parsed without the datalayout string.
4307 const Module *Mod = F->getParent();
4308 bool ForcePrintAddressSpace =
4309 !Mod || Mod->getDataLayout().getProgramAddressSpace() != 0;
4310 printAddressSpace(M: Mod, AS: F->getAddressSpace(), OS&: Out, /*Prefix=*/" ",
4311 /*Suffix=*/"", ForcePrint: ForcePrintAddressSpace);
4312 if (Attrs.hasFnAttrs())
4313 Out << " #" << Machine.getAttributeGroupSlot(AS: Attrs.getFnAttrs());
4314 if (F->hasSection()) {
4315 Out << " section \"";
4316 printEscapedString(Name: F->getSection(), Out);
4317 Out << '"';
4318 }
4319 if (F->hasPartition()) {
4320 Out << " partition \"";
4321 printEscapedString(Name: F->getPartition(), Out);
4322 Out << '"';
4323 }
4324 maybePrintComdat(Out, GO: *F);
4325 if (MaybeAlign A = F->getAlign())
4326 Out << " align " << A->value();
4327 if (MaybeAlign A = F->getPreferredAlignment())
4328 Out << " prefalign(" << A->value() << ')';
4329 if (F->hasGC())
4330 Out << " gc \"" << F->getGC() << '"';
4331 if (F->hasPrefixData()) {
4332 Out << " prefix ";
4333 writeOperand(Operand: F->getPrefixData(), PrintType: true);
4334 }
4335 if (F->hasPrologueData()) {
4336 Out << " prologue ";
4337 writeOperand(Operand: F->getPrologueData(), PrintType: true);
4338 }
4339 if (F->hasPersonalityFn()) {
4340 Out << " personality ";
4341 writeOperand(Operand: F->getPersonalityFn(), /*PrintType=*/true);
4342 }
4343
4344 if (PrintProfData) {
4345 if (auto *MDProf = F->getMetadata(KindID: LLVMContext::MD_prof)) {
4346 Out << " ";
4347 MDProf->print(OS&: Out, M: TheModule, /*IsForDebug=*/true);
4348 }
4349 }
4350
4351 if (F->isDeclaration()) {
4352 Out << '\n';
4353 } else {
4354 SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
4355 F->getAllMetadata(MDs);
4356 printMetadataAttachments(MDs, Separator: " ");
4357
4358 Out << " {";
4359 // Output all of the function's basic blocks.
4360 for (const BasicBlock &BB : *F)
4361 printBasicBlock(BB: &BB);
4362
4363 // Output the function's use-lists.
4364 printUseLists(F);
4365
4366 Out << "}\n";
4367 }
4368
4369 Machine.purgeFunction();
4370}
4371
4372/// printArgument - This member is called for every argument that is passed into
4373/// the function. Simply print it out
4374void AssemblyWriter::printArgument(const Argument *Arg, AttributeSet Attrs) {
4375 // Output type...
4376 TypePrinter.print(Ty: Arg->getType(), OS&: Out);
4377
4378 // Output parameter attributes list
4379 if (Attrs.hasAttributes()) {
4380 Out << ' ';
4381 writeAttributeSet(AttrSet: Attrs);
4382 }
4383
4384 // Output name, if available...
4385 if (Arg->hasName()) {
4386 Out << ' ';
4387 printLLVMName(OS&: Out, V: Arg);
4388 } else {
4389 int Slot = Machine.getLocalSlot(V: Arg);
4390 assert(Slot != -1 && "expect argument in function here");
4391 Out << " %" << Slot;
4392 }
4393}
4394
4395/// printBasicBlock - This member is called for each basic block in a method.
4396void AssemblyWriter::printBasicBlock(const BasicBlock *BB) {
4397 bool IsEntryBlock = BB->getParent() && BB->isEntryBlock();
4398 if (BB->hasName()) { // Print out the label if it exists...
4399 Out << "\n";
4400 printLLVMName(OS&: Out, Name: BB->getName(), Prefix: LabelPrefix);
4401 Out << ':';
4402 } else if (!IsEntryBlock) {
4403 Out << "\n";
4404 int Slot = Machine.getLocalSlot(V: BB);
4405 if (Slot != -1)
4406 Out << Slot << ":";
4407 else
4408 Out << "<badref>:";
4409 }
4410
4411 if (!IsEntryBlock) {
4412 // Output predecessors for the block.
4413 Out.PadToColumn(NewCol: 50);
4414 Out << ";";
4415 if (pred_empty(BB)) {
4416 Out << " No predecessors!";
4417 } else {
4418 Out << " preds = ";
4419 ListSeparator LS;
4420 for (const BasicBlock *Pred : predecessors(BB)) {
4421 Out << LS;
4422 writeOperand(Operand: Pred, PrintType: false);
4423 }
4424 }
4425 }
4426
4427 Out << "\n";
4428
4429 if (AnnotationWriter) AnnotationWriter->emitBasicBlockStartAnnot(BB, Out);
4430
4431 // Output all of the instructions in the basic block...
4432 for (const Instruction &I : *BB) {
4433 for (const DbgRecord &DR : I.getDbgRecordRange())
4434 printDbgRecordLine(DR);
4435 printInstructionLine(I);
4436 }
4437
4438 if (AnnotationWriter) AnnotationWriter->emitBasicBlockEndAnnot(BB, Out);
4439}
4440
4441/// printInstructionLine - Print an instruction and a newline character.
4442void AssemblyWriter::printInstructionLine(const Instruction &I) {
4443 printInstruction(I);
4444 Out << '\n';
4445}
4446
4447/// printGCRelocateComment - print comment after call to the gc.relocate
4448/// intrinsic indicating base and derived pointer names.
4449void AssemblyWriter::printGCRelocateComment(const GCRelocateInst &Relocate) {
4450 Out << " ; (";
4451 if (Value *BasePtr = Relocate.getBasePtr())
4452 writeOperand(Operand: BasePtr, PrintType: false);
4453 else
4454 Out << "invalid";
4455 Out << ", ";
4456 if (Value *DerivedPtr = Relocate.getDerivedPtr())
4457 writeOperand(Operand: DerivedPtr, PrintType: false);
4458 else
4459 Out << "invalid";
4460 Out << ")";
4461}
4462
4463/// printInfoComment - Print a little comment after the instruction indicating
4464/// which slot it occupies.
4465void AssemblyWriter::printInfoComment(const Value &V, bool isMaterializable) {
4466 if (const auto *Relocate = dyn_cast<GCRelocateInst>(Val: &V))
4467 printGCRelocateComment(Relocate: *Relocate);
4468
4469 if (AnnotationWriter && !isMaterializable)
4470 AnnotationWriter->printInfoComment(V, Out);
4471
4472 if (PrintInstDebugLocs) {
4473 if (auto *I = dyn_cast<Instruction>(Val: &V)) {
4474 if (I->getDebugLoc()) {
4475 Out << " ; ";
4476 I->getDebugLoc().print(OS&: Out);
4477 }
4478 }
4479 }
4480 if (PrintProfData) {
4481 if (auto *I = dyn_cast<Instruction>(Val: &V)) {
4482 if (auto *MD = I->getMetadata(KindID: LLVMContext::MD_prof)) {
4483 Out << " ; ";
4484 MD->print(OS&: Out, M: TheModule, /*IsForDebug=*/true);
4485 }
4486 }
4487 }
4488
4489 if (PrintInstAddrs)
4490 Out << " ; " << &V;
4491}
4492
4493static void maybePrintCallAddrSpace(const Value *Operand, const Instruction *I,
4494 raw_ostream &Out) {
4495 if (Operand == nullptr) {
4496 Out << " <cannot get addrspace!>";
4497 return;
4498 }
4499
4500 // We print the address space of the call if it is non-zero.
4501 // We also print it if it is zero but not equal to the program address space
4502 // or if we can't find a valid Module* to make it possible to parse
4503 // the resulting file even without a datalayout string.
4504 unsigned CallAddrSpace = Operand->getType()->getPointerAddressSpace();
4505 const Module *Mod = getModuleFromVal(V: I);
4506 bool ForcePrintAddrSpace =
4507 !Mod || Mod->getDataLayout().getProgramAddressSpace() != 0;
4508 printAddressSpace(M: Mod, AS: CallAddrSpace, OS&: Out, /*Prefix=*/" ", /*Suffix=*/"",
4509 ForcePrint: ForcePrintAddrSpace);
4510}
4511
4512// This member is called for each Instruction in a function..
4513void AssemblyWriter::printInstruction(const Instruction &I) {
4514 if (AnnotationWriter) AnnotationWriter->emitInstructionAnnot(&I, Out);
4515
4516 // Print out indentation for an instruction.
4517 Out << " ";
4518
4519 // Print out name if it exists...
4520 if (I.hasName()) {
4521 printLLVMName(OS&: Out, V: &I);
4522 Out << " = ";
4523 } else if (!I.getType()->isVoidTy()) {
4524 // Print out the def slot taken.
4525 int SlotNum = Machine.getLocalSlot(V: &I);
4526 if (SlotNum == -1)
4527 Out << "<badref> = ";
4528 else
4529 Out << '%' << SlotNum << " = ";
4530 }
4531
4532 if (const auto *CI = dyn_cast<CallInst>(Val: &I)) {
4533 if (CI->isMustTailCall())
4534 Out << "musttail ";
4535 else if (CI->isTailCall())
4536 Out << "tail ";
4537 else if (CI->isNoTailCall())
4538 Out << "notail ";
4539 }
4540
4541 // Print out the opcode...
4542 Out << I.getOpcodeName();
4543
4544 // If this is an atomic load or store, print out the atomic marker.
4545 if ((isa<LoadInst>(Val: I) && cast<LoadInst>(Val: I).isAtomic()) ||
4546 (isa<StoreInst>(Val: I) && cast<StoreInst>(Val: I).isAtomic()))
4547 Out << " atomic";
4548
4549 if (isa<AtomicCmpXchgInst>(Val: I) && cast<AtomicCmpXchgInst>(Val: I).isWeak())
4550 Out << " weak";
4551
4552 // If this is a volatile operation, print out the volatile marker.
4553 if ((isa<LoadInst>(Val: I) && cast<LoadInst>(Val: I).isVolatile()) ||
4554 (isa<StoreInst>(Val: I) && cast<StoreInst>(Val: I).isVolatile()) ||
4555 (isa<AtomicCmpXchgInst>(Val: I) && cast<AtomicCmpXchgInst>(Val: I).isVolatile()) ||
4556 (isa<AtomicRMWInst>(Val: I) && cast<AtomicRMWInst>(Val: I).isVolatile()))
4557 Out << " volatile";
4558
4559 // Print the elementwise marker for atomic loads and stores.
4560 if ((isa<LoadInst>(Val: I) && cast<LoadInst>(Val: I).isElementwise()) ||
4561 (isa<StoreInst>(Val: I) && cast<StoreInst>(Val: I).isElementwise()))
4562 Out << " elementwise";
4563
4564 // Print out optimization information.
4565 writeOptimizationInfo(Out, U: &I);
4566
4567 // Print out the compare instruction predicates
4568 if (const auto *CI = dyn_cast<CmpInst>(Val: &I))
4569 Out << ' ' << CI->getPredicate();
4570
4571 // Print out the atomicrmw operation
4572 if (const auto *RMWI = dyn_cast<AtomicRMWInst>(Val: &I)) {
4573 if (RMWI->isElementwise())
4574 Out << " elementwise";
4575 Out << ' ' << AtomicRMWInst::getOperationName(Op: RMWI->getOperation());
4576 }
4577
4578 // Print out the type of the operands...
4579 const Value *Operand = I.getNumOperands() ? I.getOperand(i: 0) : nullptr;
4580
4581 // Special case conditional branches to swizzle the condition out to the front
4582 if (const auto *BI = dyn_cast<CondBrInst>(Val: &I)) {
4583 Out << ' ';
4584 writeOperand(Operand: BI->getCondition(), PrintType: true);
4585 Out << ", ";
4586 writeOperand(Operand: BI->getSuccessor(i: 0), PrintType: true);
4587 Out << ", ";
4588 writeOperand(Operand: BI->getSuccessor(i: 1), PrintType: true);
4589 } else if (isa<SwitchInst>(Val: I)) {
4590 const SwitchInst& SI(cast<SwitchInst>(Val: I));
4591 // Special case switch instruction to get formatting nice and correct.
4592 Out << ' ';
4593 writeOperand(Operand: SI.getCondition(), PrintType: true);
4594 Out << ", ";
4595 writeOperand(Operand: SI.getDefaultDest(), PrintType: true);
4596 Out << " [";
4597 for (auto Case : SI.cases()) {
4598 Out << "\n ";
4599 writeOperand(Operand: Case.getCaseValue(), PrintType: true);
4600 Out << ", ";
4601 writeOperand(Operand: Case.getCaseSuccessor(), PrintType: true);
4602 }
4603 Out << "\n ]";
4604 } else if (isa<IndirectBrInst>(Val: I)) {
4605 // Special case indirectbr instruction to get formatting nice and correct.
4606 Out << ' ';
4607 writeOperand(Operand, PrintType: true);
4608 Out << ", [";
4609
4610 ListSeparator LS;
4611 for (unsigned i = 1, e = I.getNumOperands(); i != e; ++i) {
4612 Out << LS;
4613 writeOperand(Operand: I.getOperand(i), PrintType: true);
4614 }
4615 Out << ']';
4616 } else if (const auto *PN = dyn_cast<PHINode>(Val: &I)) {
4617 Out << ' ';
4618 TypePrinter.print(Ty: I.getType(), OS&: Out);
4619 Out << ' ';
4620
4621 ListSeparator LS;
4622 for (const auto &[V, Block] :
4623 zip_equal(t: PN->incoming_values(), u: PN->blocks())) {
4624 Out << LS << "[ ";
4625 writeOperand(Operand: V, PrintType: false);
4626 Out << ", ";
4627 writeOperand(Operand: Block, PrintType: false);
4628 Out << " ]";
4629 }
4630 } else if (const auto *EVI = dyn_cast<ExtractValueInst>(Val: &I)) {
4631 Out << ' ';
4632 writeOperand(Operand: I.getOperand(i: 0), PrintType: true);
4633 Out << ", ";
4634 Out << llvm::interleaved(R: EVI->indices());
4635 } else if (const auto *IVI = dyn_cast<InsertValueInst>(Val: &I)) {
4636 Out << ' ';
4637 writeOperand(Operand: I.getOperand(i: 0), PrintType: true); Out << ", ";
4638 writeOperand(Operand: I.getOperand(i: 1), PrintType: true);
4639 Out << ", ";
4640 Out << llvm::interleaved(R: IVI->indices());
4641 } else if (const auto *LPI = dyn_cast<LandingPadInst>(Val: &I)) {
4642 Out << ' ';
4643 TypePrinter.print(Ty: I.getType(), OS&: Out);
4644 if (LPI->isCleanup() || LPI->getNumClauses() != 0)
4645 Out << '\n';
4646
4647 if (LPI->isCleanup())
4648 Out << " cleanup";
4649
4650 for (unsigned i = 0, e = LPI->getNumClauses(); i != e; ++i) {
4651 if (i != 0 || LPI->isCleanup()) Out << "\n";
4652 if (LPI->isCatch(Idx: i))
4653 Out << " catch ";
4654 else
4655 Out << " filter ";
4656
4657 writeOperand(Operand: LPI->getClause(Idx: i), PrintType: true);
4658 }
4659 } else if (const auto *CatchSwitch = dyn_cast<CatchSwitchInst>(Val: &I)) {
4660 Out << " within ";
4661 writeOperand(Operand: CatchSwitch->getParentPad(), /*PrintType=*/false);
4662 Out << " [";
4663 ListSeparator LS;
4664 for (const BasicBlock *PadBB : CatchSwitch->handlers()) {
4665 Out << LS;
4666 writeOperand(Operand: PadBB, /*PrintType=*/true);
4667 }
4668 Out << "] unwind ";
4669 if (const BasicBlock *UnwindDest = CatchSwitch->getUnwindDest())
4670 writeOperand(Operand: UnwindDest, /*PrintType=*/true);
4671 else
4672 Out << "to caller";
4673 } else if (const auto *FPI = dyn_cast<FuncletPadInst>(Val: &I)) {
4674 Out << " within ";
4675 writeOperand(Operand: FPI->getParentPad(), /*PrintType=*/false);
4676 Out << " [";
4677 ListSeparator LS;
4678 for (const Value *Op : FPI->arg_operands()) {
4679 Out << LS;
4680 writeOperand(Operand: Op, /*PrintType=*/true);
4681 }
4682 Out << ']';
4683 } else if (isa<ReturnInst>(Val: I) && !Operand) {
4684 Out << " void";
4685 } else if (const auto *CRI = dyn_cast<CatchReturnInst>(Val: &I)) {
4686 Out << " from ";
4687 writeOperand(Operand: CRI->getOperand(i_nocapture: 0), /*PrintType=*/false);
4688
4689 Out << " to ";
4690 writeOperand(Operand: CRI->getOperand(i_nocapture: 1), /*PrintType=*/true);
4691 } else if (const auto *CRI = dyn_cast<CleanupReturnInst>(Val: &I)) {
4692 Out << " from ";
4693 writeOperand(Operand: CRI->getOperand(i_nocapture: 0), /*PrintType=*/false);
4694
4695 Out << " unwind ";
4696 if (CRI->hasUnwindDest())
4697 writeOperand(Operand: CRI->getOperand(i_nocapture: 1), /*PrintType=*/true);
4698 else
4699 Out << "to caller";
4700 } else if (const auto *CI = dyn_cast<CallInst>(Val: &I)) {
4701 // Print the calling convention being used.
4702 if (CI->getCallingConv() != CallingConv::C) {
4703 Out << " ";
4704 printCallingConv(cc: CI->getCallingConv(), Out);
4705 }
4706
4707 Operand = CI->getCalledOperand();
4708 FunctionType *FTy = CI->getFunctionType();
4709 Type *RetTy = FTy->getReturnType();
4710 const AttributeList &PAL = CI->getAttributes();
4711
4712 if (PAL.hasRetAttrs())
4713 Out << ' ' << PAL.getAsString(Index: AttributeList::ReturnIndex);
4714
4715 // Only print addrspace(N) if necessary:
4716 maybePrintCallAddrSpace(Operand, I: &I, Out);
4717
4718 // If possible, print out the short form of the call instruction. We can
4719 // only do this if the first argument is a pointer to a nonvararg function,
4720 // and if the return type is not a pointer to a function.
4721 Out << ' ';
4722 TypePrinter.print(Ty: FTy->isVarArg() ? FTy : RetTy, OS&: Out);
4723 Out << ' ';
4724 writeOperand(Operand, PrintType: false);
4725 Out << '(';
4726 bool HasPrettyPrintedArgs =
4727 isa<IntrinsicInst>(Val: CI) &&
4728 Intrinsic::hasPrettyPrintedArgs(id: CI->getIntrinsicID());
4729
4730 ListSeparator LS;
4731 Function *CalledFunc = CI->getCalledFunction();
4732 auto PrintArgComment = [&](unsigned ArgNo) {
4733 const auto *ConstArg = dyn_cast<Constant>(Val: CI->getArgOperand(i: ArgNo));
4734 if (!ConstArg || !CalledFunc)
4735 return;
4736 std::string ArgComment;
4737 raw_string_ostream ArgCommentStream(ArgComment);
4738 Intrinsic::ID IID = CalledFunc->getIntrinsicID();
4739 Intrinsic::printImmArg(IID, ArgIdx: ArgNo, OS&: ArgCommentStream, ImmArgVal: ConstArg);
4740 if (ArgComment.empty())
4741 return;
4742 Out << "/* " << ArgComment << " */ ";
4743 };
4744 if (HasPrettyPrintedArgs) {
4745 for (unsigned ArgNo = 0, NumArgs = CI->arg_size(); ArgNo < NumArgs;
4746 ++ArgNo) {
4747 Out << LS;
4748 PrintArgComment(ArgNo);
4749 writeParamOperand(Operand: CI->getArgOperand(i: ArgNo), Attrs: PAL.getParamAttrs(ArgNo));
4750 }
4751 } else {
4752 for (unsigned ArgNo = 0, NumArgs = CI->arg_size(); ArgNo < NumArgs;
4753 ++ArgNo) {
4754 Out << LS;
4755 writeParamOperand(Operand: CI->getArgOperand(i: ArgNo), Attrs: PAL.getParamAttrs(ArgNo));
4756 }
4757 }
4758 // Emit an ellipsis if this is a musttail call in a vararg function. This
4759 // is only to aid readability, musttail calls forward varargs by default.
4760 if (CI->isMustTailCall() && CI->getParent() &&
4761 CI->getParent()->getParent() &&
4762 CI->getParent()->getParent()->isVarArg()) {
4763 if (CI->arg_size() > 0)
4764 Out << ", ";
4765 Out << "...";
4766 }
4767
4768 Out << ')';
4769 if (PAL.hasFnAttrs())
4770 Out << " #" << Machine.getAttributeGroupSlot(AS: PAL.getFnAttrs());
4771
4772 writeOperandBundles(Call: CI);
4773 } else if (const auto *II = dyn_cast<InvokeInst>(Val: &I)) {
4774 Operand = II->getCalledOperand();
4775 FunctionType *FTy = II->getFunctionType();
4776 Type *RetTy = FTy->getReturnType();
4777 const AttributeList &PAL = II->getAttributes();
4778
4779 // Print the calling convention being used.
4780 if (II->getCallingConv() != CallingConv::C) {
4781 Out << " ";
4782 printCallingConv(cc: II->getCallingConv(), Out);
4783 }
4784
4785 if (PAL.hasRetAttrs())
4786 Out << ' ' << PAL.getAsString(Index: AttributeList::ReturnIndex);
4787
4788 // Only print addrspace(N) if necessary:
4789 maybePrintCallAddrSpace(Operand, I: &I, Out);
4790
4791 // If possible, print out the short form of the invoke instruction. We can
4792 // only do this if the first argument is a pointer to a nonvararg function,
4793 // and if the return type is not a pointer to a function.
4794 //
4795 Out << ' ';
4796 TypePrinter.print(Ty: FTy->isVarArg() ? FTy : RetTy, OS&: Out);
4797 Out << ' ';
4798 writeOperand(Operand, PrintType: false);
4799 Out << '(';
4800 ListSeparator LS;
4801 for (unsigned op = 0, Eop = II->arg_size(); op < Eop; ++op) {
4802 Out << LS;
4803 writeParamOperand(Operand: II->getArgOperand(i: op), Attrs: PAL.getParamAttrs(ArgNo: op));
4804 }
4805
4806 Out << ')';
4807 if (PAL.hasFnAttrs())
4808 Out << " #" << Machine.getAttributeGroupSlot(AS: PAL.getFnAttrs());
4809
4810 writeOperandBundles(Call: II);
4811
4812 Out << "\n to ";
4813 writeOperand(Operand: II->getNormalDest(), PrintType: true);
4814 Out << " unwind ";
4815 writeOperand(Operand: II->getUnwindDest(), PrintType: true);
4816 } else if (const auto *CBI = dyn_cast<CallBrInst>(Val: &I)) {
4817 Operand = CBI->getCalledOperand();
4818 FunctionType *FTy = CBI->getFunctionType();
4819 Type *RetTy = FTy->getReturnType();
4820 const AttributeList &PAL = CBI->getAttributes();
4821
4822 // Print the calling convention being used.
4823 if (CBI->getCallingConv() != CallingConv::C) {
4824 Out << " ";
4825 printCallingConv(cc: CBI->getCallingConv(), Out);
4826 }
4827
4828 if (PAL.hasRetAttrs())
4829 Out << ' ' << PAL.getAsString(Index: AttributeList::ReturnIndex);
4830
4831 // If possible, print out the short form of the callbr instruction. We can
4832 // only do this if the first argument is a pointer to a nonvararg function,
4833 // and if the return type is not a pointer to a function.
4834 //
4835 Out << ' ';
4836 TypePrinter.print(Ty: FTy->isVarArg() ? FTy : RetTy, OS&: Out);
4837 Out << ' ';
4838 writeOperand(Operand, PrintType: false);
4839 Out << '(';
4840 ListSeparator ArgLS;
4841 for (unsigned op = 0, Eop = CBI->arg_size(); op < Eop; ++op) {
4842 Out << ArgLS;
4843 writeParamOperand(Operand: CBI->getArgOperand(i: op), Attrs: PAL.getParamAttrs(ArgNo: op));
4844 }
4845
4846 Out << ')';
4847 if (PAL.hasFnAttrs())
4848 Out << " #" << Machine.getAttributeGroupSlot(AS: PAL.getFnAttrs());
4849
4850 writeOperandBundles(Call: CBI);
4851
4852 Out << "\n to ";
4853 writeOperand(Operand: CBI->getDefaultDest(), PrintType: true);
4854 Out << " [";
4855 ListSeparator DestLS;
4856 for (const BasicBlock *Dest : CBI->getIndirectDests()) {
4857 Out << DestLS;
4858 writeOperand(Operand: Dest, PrintType: true);
4859 }
4860 Out << ']';
4861 } else if (const auto *AI = dyn_cast<AllocaInst>(Val: &I)) {
4862 Out << ' ';
4863 if (AI->isUsedWithInAlloca())
4864 Out << "inalloca ";
4865 if (AI->isSwiftError())
4866 Out << "swifterror ";
4867 TypePrinter.print(Ty: AI->getAllocatedType(), OS&: Out);
4868
4869 // Explicitly write the array size if the code is broken, if it's an array
4870 // allocation, or if the type is not canonical for scalar allocations. The
4871 // latter case prevents the type from mutating when round-tripping through
4872 // assembly.
4873 if (!AI->getArraySize() || AI->isArrayAllocation() ||
4874 !AI->getArraySize()->getType()->isIntegerTy(BitWidth: 32)) {
4875 Out << ", ";
4876 writeOperand(Operand: AI->getArraySize(), PrintType: true);
4877 }
4878 if (MaybeAlign A = AI->getAlign()) {
4879 Out << ", align " << A->value();
4880 }
4881
4882 printAddressSpace(M: AI->getModule(), AS: AI->getAddressSpace(), OS&: Out,
4883 /*Prefix=*/", ");
4884 } else if (isa<CastInst>(Val: I)) {
4885 if (Operand) {
4886 Out << ' ';
4887 writeOperand(Operand, PrintType: true); // Work with broken code
4888 }
4889 Out << " to ";
4890 TypePrinter.print(Ty: I.getType(), OS&: Out);
4891 } else if (isa<VAArgInst>(Val: I)) {
4892 if (Operand) {
4893 Out << ' ';
4894 writeOperand(Operand, PrintType: true); // Work with broken code
4895 }
4896 Out << ", ";
4897 TypePrinter.print(Ty: I.getType(), OS&: Out);
4898 } else if (Operand) { // Print the normal way.
4899 if (const auto *GEP = dyn_cast<GetElementPtrInst>(Val: &I)) {
4900 Out << ' ';
4901 TypePrinter.print(Ty: GEP->getSourceElementType(), OS&: Out);
4902 Out << ',';
4903 } else if (const auto *LI = dyn_cast<LoadInst>(Val: &I)) {
4904 Out << ' ';
4905 TypePrinter.print(Ty: LI->getType(), OS&: Out);
4906 Out << ',';
4907 }
4908
4909 // PrintAllTypes - Instructions who have operands of all the same type
4910 // omit the type from all but the first operand. If the instruction has
4911 // different type operands (for example br), then they are all printed.
4912 bool PrintAllTypes = false;
4913 Type *TheType = Operand->getType();
4914
4915 // Select, Store, ShuffleVector, CmpXchg and AtomicRMW always print all
4916 // types.
4917 if (isa<SelectInst>(Val: I) || isa<StoreInst>(Val: I) || isa<ShuffleVectorInst>(Val: I) ||
4918 isa<ReturnInst>(Val: I) || isa<AtomicCmpXchgInst>(Val: I) ||
4919 isa<AtomicRMWInst>(Val: I)) {
4920 PrintAllTypes = true;
4921 } else {
4922 for (unsigned i = 1, E = I.getNumOperands(); i != E; ++i) {
4923 Operand = I.getOperand(i);
4924 // note that Operand shouldn't be null, but the test helps make dump()
4925 // more tolerant of malformed IR
4926 if (Operand && Operand->getType() != TheType) {
4927 PrintAllTypes = true; // We have differing types! Print them all!
4928 break;
4929 }
4930 }
4931 }
4932
4933 if (!PrintAllTypes) {
4934 Out << ' ';
4935 TypePrinter.print(Ty: TheType, OS&: Out);
4936 }
4937
4938 Out << ' ';
4939 ListSeparator LS;
4940 for (const Value *Op : I.operands()) {
4941 Out << LS;
4942 writeOperand(Operand: Op, PrintType: PrintAllTypes);
4943 }
4944 }
4945
4946 // Print atomic ordering/alignment for memory operations
4947 if (const auto *LI = dyn_cast<LoadInst>(Val: &I)) {
4948 if (LI->isAtomic())
4949 writeAtomic(Context: LI->getContext(), Ordering: LI->getOrdering(), SSID: LI->getSyncScopeID());
4950 if (MaybeAlign A = LI->getAlign())
4951 Out << ", align " << A->value();
4952 } else if (const auto *SI = dyn_cast<StoreInst>(Val: &I)) {
4953 if (SI->isAtomic())
4954 writeAtomic(Context: SI->getContext(), Ordering: SI->getOrdering(), SSID: SI->getSyncScopeID());
4955 if (MaybeAlign A = SI->getAlign())
4956 Out << ", align " << A->value();
4957 } else if (const auto *CXI = dyn_cast<AtomicCmpXchgInst>(Val: &I)) {
4958 writeAtomicCmpXchg(Context: CXI->getContext(), SuccessOrdering: CXI->getSuccessOrdering(),
4959 FailureOrdering: CXI->getFailureOrdering(), SSID: CXI->getSyncScopeID());
4960 Out << ", align " << CXI->getAlign().value();
4961 } else if (const auto *RMWI = dyn_cast<AtomicRMWInst>(Val: &I)) {
4962 writeAtomic(Context: RMWI->getContext(), Ordering: RMWI->getOrdering(),
4963 SSID: RMWI->getSyncScopeID());
4964 Out << ", align " << RMWI->getAlign().value();
4965 } else if (const auto *FI = dyn_cast<FenceInst>(Val: &I)) {
4966 writeAtomic(Context: FI->getContext(), Ordering: FI->getOrdering(), SSID: FI->getSyncScopeID());
4967 } else if (const auto *SVI = dyn_cast<ShuffleVectorInst>(Val: &I)) {
4968 printShuffleMask(Out, Ty: SVI->getType(), Mask: SVI->getShuffleMask());
4969 }
4970
4971 // Print Metadata info.
4972 SmallVector<std::pair<unsigned, MDNode *>, 4> InstMD;
4973 I.getAllMetadata(MDs&: InstMD);
4974 printMetadataAttachments(MDs: InstMD, Separator: ", ");
4975
4976 // Print a nice comment.
4977 printInfoComment(V: I);
4978}
4979
4980void AssemblyWriter::printDbgMarker(const DbgMarker &Marker) {
4981 // There's no formal representation of a DbgMarker -- print purely as a
4982 // debugging aid.
4983 for (const DbgRecord &DPR : Marker.StoredDbgRecords) {
4984 printDbgRecord(DR: DPR);
4985 Out << "\n";
4986 }
4987
4988 Out << " DbgMarker -> { ";
4989 printInstruction(I: *Marker.MarkedInstr);
4990 Out << " }";
4991}
4992
4993void AssemblyWriter::printDbgRecord(const DbgRecord &DR) {
4994 if (auto *DVR = dyn_cast<DbgVariableRecord>(Val: &DR))
4995 printDbgVariableRecord(DVR: *DVR);
4996 else if (auto *DLR = dyn_cast<DbgLabelRecord>(Val: &DR))
4997 printDbgLabelRecord(DLR: *DLR);
4998 else
4999 llvm_unreachable("Unexpected DbgRecord kind");
5000}
5001
5002void AssemblyWriter::printDbgVariableRecord(const DbgVariableRecord &DVR) {
5003 auto WriterCtx = getContext();
5004 Out << "#dbg_";
5005 switch (DVR.getType()) {
5006 case DbgVariableRecord::LocationType::Value:
5007 Out << "value";
5008 break;
5009 case DbgVariableRecord::LocationType::Declare:
5010 Out << "declare";
5011 break;
5012 case DbgVariableRecord::LocationType::DeclareValue:
5013 Out << "declare_value";
5014 break;
5015 case DbgVariableRecord::LocationType::Assign:
5016 Out << "assign";
5017 break;
5018 default:
5019 llvm_unreachable(
5020 "Tried to print a DbgVariableRecord with an invalid LocationType!");
5021 }
5022
5023 auto PrintOrNull = [&](Metadata *M) {
5024 if (!M)
5025 Out << "(null)";
5026 else
5027 writeAsOperandInternal(Out, MD: M, WriterCtx, FromValue: true);
5028 };
5029
5030 Out << "(";
5031 PrintOrNull(DVR.getRawLocation());
5032 Out << ", ";
5033 PrintOrNull(DVR.getRawVariable());
5034 Out << ", ";
5035 PrintOrNull(DVR.getRawExpression());
5036 Out << ", ";
5037 if (DVR.isDbgAssign()) {
5038 PrintOrNull(DVR.getRawAssignID());
5039 Out << ", ";
5040 PrintOrNull(DVR.getRawAddress());
5041 Out << ", ";
5042 PrintOrNull(DVR.getRawAddressExpression());
5043 Out << ", ";
5044 }
5045 PrintOrNull(DVR.getDebugLoc().getAsMDNode());
5046 Out << ")";
5047}
5048
5049/// printDbgRecordLine - Print a DbgRecord with indentation and a newline
5050/// character.
5051void AssemblyWriter::printDbgRecordLine(const DbgRecord &DR) {
5052 // Print lengthier indentation to bring out-of-line with instructions.
5053 Out << " ";
5054 printDbgRecord(DR);
5055 Out << '\n';
5056}
5057
5058void AssemblyWriter::printDbgLabelRecord(const DbgLabelRecord &Label) {
5059 auto WriterCtx = getContext();
5060 Out << "#dbg_label(";
5061 writeAsOperandInternal(Out, MD: Label.getRawLabel(), WriterCtx, FromValue: true);
5062 Out << ", ";
5063 writeAsOperandInternal(Out, MD: Label.getDebugLoc(), WriterCtx, FromValue: true);
5064 Out << ")";
5065}
5066
5067void AssemblyWriter::printMetadataAttachments(
5068 const SmallVectorImpl<std::pair<unsigned, MDNode *>> &MDs,
5069 StringRef Separator) {
5070 if (MDs.empty())
5071 return;
5072
5073 if (MDNames.empty())
5074 MDs[0].second->getContext().getMDKindNames(Result&: MDNames);
5075
5076 auto WriterCtx = getContext();
5077 for (const auto &I : MDs) {
5078 unsigned Kind = I.first;
5079 Out << Separator;
5080 if (Kind < MDNames.size()) {
5081 Out << "!";
5082 printMetadataIdentifier(Name: MDNames[Kind], Out);
5083 } else
5084 Out << "!<unknown kind #" << Kind << ">";
5085 Out << ' ';
5086 writeAsOperandInternal(Out, MD: I.second, WriterCtx);
5087 }
5088}
5089
5090void AssemblyWriter::writeMDNode(unsigned Slot, const MDNode *Node) {
5091 if (AnnotationWriter)
5092 AnnotationWriter->emitMDNodeAnnot(Node, Out);
5093
5094 Out << '!' << Slot << " = ";
5095 printMDNodeBody(MD: Node);
5096 Out << "\n";
5097}
5098
5099void AssemblyWriter::writeAllMDNodes() {
5100 SmallVector<std::pair<unsigned, const MDNode *>, 16> Nodes;
5101 Nodes.reserve(N: Machine.mdn_size());
5102 for (auto &I : llvm::make_range(x: Machine.mdn_begin(), y: Machine.mdn_end()))
5103 Nodes.emplace_back(Args&: I.second, Args: cast<MDNode>(Val: I.first));
5104 llvm::sort(C&: Nodes);
5105
5106 for (auto [Slot, Node] : Nodes)
5107 writeMDNode(Slot, Node);
5108}
5109
5110void AssemblyWriter::printMDNodeBody(const MDNode *Node) {
5111 auto WriterCtx = getContext();
5112 writeMDNodeBodyInternal(Out, Node, Ctx&: WriterCtx);
5113}
5114
5115void AssemblyWriter::writeAttribute(const Attribute &Attr, bool InAttrGroup) {
5116 if (!Attr.isTypeAttribute()) {
5117 Out << Attr.getAsString(InAttrGrp: InAttrGroup);
5118 return;
5119 }
5120
5121 Out << Attribute::getNameFromAttrKind(AttrKind: Attr.getKindAsEnum());
5122 if (Type *Ty = Attr.getValueAsType()) {
5123 Out << '(';
5124 TypePrinter.print(Ty, OS&: Out);
5125 Out << ')';
5126 }
5127}
5128
5129void AssemblyWriter::writeAttributeSet(const AttributeSet &AttrSet,
5130 bool InAttrGroup) {
5131 ListSeparator LS(" ");
5132 for (const auto &Attr : AttrSet) {
5133 Out << LS;
5134 writeAttribute(Attr, InAttrGroup);
5135 }
5136}
5137
5138void AssemblyWriter::writeAllAttributeGroups() {
5139 std::vector<std::pair<AttributeSet, unsigned>> asVec;
5140 asVec.resize(new_size: Machine.as_size());
5141
5142 for (auto &I : llvm::make_range(x: Machine.as_begin(), y: Machine.as_end()))
5143 asVec[I.second] = I;
5144
5145 for (const auto &I : asVec)
5146 Out << "attributes #" << I.second << " = { "
5147 << I.first.getAsString(InAttrGrp: true) << " }\n";
5148}
5149
5150void AssemblyWriter::printUseListOrder(const Value *V,
5151 ArrayRef<unsigned> Shuffle) {
5152 if (Machine.getFunction())
5153 Out << " ";
5154
5155 Out << "uselistorder ";
5156 writeOperand(Operand: V, PrintType: true);
5157
5158 assert(Shuffle.size() >= 2 && "Shuffle too small");
5159 Out << ", { " << llvm::interleaved(R: Shuffle) << " }\n";
5160}
5161
5162void AssemblyWriter::printUseLists(const Function *F) {
5163 auto It = UseListOrders.find(Val: F);
5164 if (It == UseListOrders.end())
5165 return;
5166
5167 Out << "\n; uselistorder directives\n";
5168 for (const auto &Pair : It->second)
5169 printUseListOrder(V: Pair.first, Shuffle: Pair.second);
5170}
5171
5172//===----------------------------------------------------------------------===//
5173// External Interface declarations
5174//===----------------------------------------------------------------------===//
5175
5176void Function::print(raw_ostream &ROS, AssemblyAnnotationWriter *AAW,
5177 bool ShouldPreserveUseListOrder, bool IsForDebug) const {
5178 SlotTracker SlotTable(this);
5179 formatted_raw_ostream OS(ROS);
5180 AssemblyWriter W(OS, SlotTable, this->getParent(), AAW, IsForDebug,
5181 ShouldPreserveUseListOrder);
5182 W.printFunction(F: this);
5183}
5184
5185void BasicBlock::print(raw_ostream &ROS, AssemblyAnnotationWriter *AAW,
5186 bool ShouldPreserveUseListOrder,
5187 bool IsForDebug) const {
5188 SlotTracker SlotTable(this->getParent());
5189 formatted_raw_ostream OS(ROS);
5190 AssemblyWriter W(OS, SlotTable, this->getModule(), AAW, IsForDebug,
5191 ShouldPreserveUseListOrder);
5192 W.printBasicBlock(BB: this);
5193}
5194
5195void Module::print(raw_ostream &ROS, AssemblyAnnotationWriter *AAW,
5196 bool ShouldPreserveUseListOrder, bool IsForDebug) const {
5197 SlotTracker SlotTable(this, /*ShouldTrackMetadataDefinitions=*/true);
5198 formatted_raw_ostream OS(ROS);
5199 AssemblyWriter W(OS, SlotTable, this, AAW, IsForDebug,
5200 ShouldPreserveUseListOrder);
5201 W.printModule(M: this);
5202}
5203
5204void NamedMDNode::print(raw_ostream &ROS, bool IsForDebug) const {
5205 SlotTracker SlotTable(getParent());
5206 formatted_raw_ostream OS(ROS);
5207 AssemblyWriter W(OS, SlotTable, getParent(), nullptr, IsForDebug);
5208 W.printNamedMDNode(NMD: this);
5209}
5210
5211void NamedMDNode::print(raw_ostream &ROS, ModuleSlotTracker &MST,
5212 bool IsForDebug) const {
5213 std::optional<SlotTracker> LocalST;
5214 SlotTracker *SlotTable;
5215 if (auto *ST = MST.getMachine())
5216 SlotTable = ST;
5217 else {
5218 LocalST.emplace(args: getParent());
5219 SlotTable = &*LocalST;
5220 }
5221
5222 formatted_raw_ostream OS(ROS);
5223 AssemblyWriter W(OS, *SlotTable, getParent(), nullptr, IsForDebug);
5224 W.printNamedMDNode(NMD: this);
5225}
5226
5227void Comdat::print(raw_ostream &ROS, bool /*IsForDebug*/) const {
5228 printLLVMName(OS&: ROS, Name: getName(), Prefix: ComdatPrefix);
5229 ROS << " = comdat ";
5230
5231 switch (getSelectionKind()) {
5232 case Comdat::Any:
5233 ROS << "any";
5234 break;
5235 case Comdat::ExactMatch:
5236 ROS << "exactmatch";
5237 break;
5238 case Comdat::Largest:
5239 ROS << "largest";
5240 break;
5241 case Comdat::NoDeduplicate:
5242 ROS << "nodeduplicate";
5243 break;
5244 case Comdat::SameSize:
5245 ROS << "samesize";
5246 break;
5247 }
5248
5249 ROS << '\n';
5250}
5251
5252void Type::print(raw_ostream &OS, bool /*IsForDebug*/, bool NoDetails) const {
5253 TypePrinting TP;
5254 TP.print(Ty: const_cast<Type*>(this), OS);
5255
5256 if (NoDetails)
5257 return;
5258
5259 // If the type is a named struct type, print the body as well.
5260 if (auto *STy = dyn_cast<StructType>(Val: const_cast<Type *>(this)))
5261 if (!STy->isLiteral()) {
5262 OS << " = type ";
5263 TP.printStructBody(STy, OS);
5264 }
5265}
5266
5267void DbgMarker::print(raw_ostream &ROS, bool IsForDebug) const {
5268
5269 ModuleSlotTracker MST(getModuleFromDPI(Marker: this));
5270 print(ROS, MST, IsForDebug);
5271}
5272
5273void DbgVariableRecord::print(raw_ostream &ROS, bool IsForDebug) const {
5274
5275 ModuleSlotTracker MST(getModuleFromDPI(DR: this));
5276 print(ROS, MST, IsForDebug);
5277}
5278
5279void DbgMarker::print(raw_ostream &ROS, ModuleSlotTracker &MST,
5280 bool IsForDebug) const {
5281 formatted_raw_ostream OS(ROS);
5282 SlotTracker EmptySlotTable(static_cast<const Module *>(nullptr));
5283 SlotTracker &SlotTable =
5284 MST.getMachine() ? *MST.getMachine() : EmptySlotTable;
5285 const Function *F = getParent() ? getParent()->getParent() : nullptr;
5286 if (F)
5287 MST.incorporateFunction(F: *F);
5288 AssemblyWriter W(OS, SlotTable, getModuleFromDPI(Marker: this), nullptr, IsForDebug);
5289 W.printDbgMarker(Marker: *this);
5290}
5291
5292void DbgLabelRecord::print(raw_ostream &ROS, bool IsForDebug) const {
5293
5294 ModuleSlotTracker MST(getModuleFromDPI(DR: this));
5295 print(ROS, MST, IsForDebug);
5296}
5297
5298void DbgVariableRecord::print(raw_ostream &ROS, ModuleSlotTracker &MST,
5299 bool IsForDebug) const {
5300 formatted_raw_ostream OS(ROS);
5301 SlotTracker EmptySlotTable(static_cast<const Module *>(nullptr));
5302 SlotTracker &SlotTable =
5303 MST.getMachine() ? *MST.getMachine() : EmptySlotTable;
5304 const Function *F = Marker && Marker->getParent()
5305 ? Marker->getParent()->getParent()
5306 : nullptr;
5307 if (F)
5308 MST.incorporateFunction(F: *F);
5309 AssemblyWriter W(OS, SlotTable, getModuleFromDPI(DR: this), nullptr, IsForDebug);
5310 W.printDbgVariableRecord(DVR: *this);
5311}
5312
5313void DbgLabelRecord::print(raw_ostream &ROS, ModuleSlotTracker &MST,
5314 bool IsForDebug) const {
5315 formatted_raw_ostream OS(ROS);
5316 SlotTracker EmptySlotTable(static_cast<const Module *>(nullptr));
5317 SlotTracker &SlotTable =
5318 MST.getMachine() ? *MST.getMachine() : EmptySlotTable;
5319 const Function *F =
5320 Marker->getParent() ? Marker->getParent()->getParent() : nullptr;
5321 if (F)
5322 MST.incorporateFunction(F: *F);
5323
5324 AssemblyWriter W(OS, SlotTable, getModuleFromDPI(DR: this), nullptr, IsForDebug);
5325 W.printDbgLabelRecord(Label: *this);
5326}
5327
5328void Value::print(raw_ostream &ROS, bool IsForDebug) const {
5329 if (const auto *F = dyn_cast<Function>(Val: this)) {
5330 F->print(ROS, AAW: nullptr, /*ShouldPreserveUseListOrder=*/false, IsForDebug);
5331 return;
5332 }
5333 if (const auto *BB = dyn_cast<BasicBlock>(Val: this)) {
5334 BB->print(ROS, AAW: nullptr, /*ShouldPreserveUseListOrder=*/false, IsForDebug);
5335 return;
5336 }
5337
5338 ModuleSlotTracker MST(getModuleFromVal(V: this));
5339 print(O&: ROS, MST, IsForDebug);
5340}
5341
5342void Value::print(raw_ostream &ROS, ModuleSlotTracker &MST,
5343 bool IsForDebug) const {
5344 formatted_raw_ostream OS(ROS);
5345 SlotTracker EmptySlotTable(static_cast<const Module *>(nullptr));
5346 SlotTracker &SlotTable =
5347 MST.getMachine() ? *MST.getMachine() : EmptySlotTable;
5348 auto IncorporateFunction = [&](const Function *F) {
5349 if (F)
5350 MST.incorporateFunction(F: *F);
5351 };
5352
5353 if (const auto *I = dyn_cast<Instruction>(Val: this)) {
5354 IncorporateFunction(I->getParent() ? I->getParent()->getParent() : nullptr);
5355 AssemblyWriter W(OS, SlotTable, getModuleFromVal(V: I), nullptr, IsForDebug);
5356 W.printInstruction(I: *I);
5357 } else if (const auto *BB = dyn_cast<BasicBlock>(Val: this)) {
5358 IncorporateFunction(BB->getParent());
5359 AssemblyWriter W(OS, SlotTable, getModuleFromVal(V: BB), nullptr, IsForDebug);
5360 W.printBasicBlock(BB);
5361 } else if (const auto *GV = dyn_cast<GlobalValue>(Val: this)) {
5362 AssemblyWriter W(OS, SlotTable, GV->getParent(), nullptr, IsForDebug);
5363 if (const auto *V = dyn_cast<GlobalVariable>(Val: GV))
5364 W.printGlobal(GV: V);
5365 else if (const auto *F = dyn_cast<Function>(Val: GV))
5366 W.printFunction(F);
5367 else if (const auto *A = dyn_cast<GlobalAlias>(Val: GV))
5368 W.printAlias(GA: A);
5369 else if (const auto *I = dyn_cast<GlobalIFunc>(Val: GV))
5370 W.printIFunc(GI: I);
5371 else
5372 llvm_unreachable("Unknown GlobalValue to print out!");
5373 } else if (const auto *V = dyn_cast<MetadataAsValue>(Val: this)) {
5374 V->getMetadata()->print(OS&: ROS, MST, M: getModuleFromVal(V));
5375 } else if (const auto *C = dyn_cast<Constant>(Val: this)) {
5376 TypePrinting TypePrinter;
5377 TypePrinter.print(Ty: C->getType(), OS);
5378 OS << ' ';
5379 AsmWriterContext WriterCtx(&TypePrinter, MST.getMachine());
5380 writeConstantInternal(Out&: OS, CV: C, WriterCtx);
5381 } else if (isa<InlineAsm>(Val: this) || isa<Argument>(Val: this)) {
5382 this->printAsOperand(O&: OS, /* PrintType */ true, MST);
5383 } else {
5384 llvm_unreachable("Unknown value to print out!");
5385 }
5386}
5387
5388/// Print without a type, skipping the TypePrinting object.
5389///
5390/// \return \c true iff printing was successful.
5391static bool printWithoutType(const Value &V, raw_ostream &O,
5392 SlotTracker *Machine, const Module *M) {
5393 if (V.hasName() || isa<GlobalValue>(Val: V) ||
5394 (!isa<Constant>(Val: V) && !isa<MetadataAsValue>(Val: V))) {
5395 AsmWriterContext WriterCtx(nullptr, Machine, M);
5396 writeAsOperandInternal(Out&: O, V: &V, WriterCtx);
5397 return true;
5398 }
5399 return false;
5400}
5401
5402static void printAsOperandImpl(const Value &V, raw_ostream &O, bool PrintType,
5403 ModuleSlotTracker &MST) {
5404 TypePrinting TypePrinter(MST.getModule());
5405 AsmWriterContext WriterCtx(&TypePrinter, MST.getMachine(), MST.getModule());
5406 writeAsOperandInternal(Out&: O, V: &V, WriterCtx, PrintType);
5407}
5408
5409void Value::printAsOperand(raw_ostream &O, bool PrintType,
5410 const Module *M) const {
5411 if (!M)
5412 M = getModuleFromVal(V: this);
5413
5414 if (!PrintType)
5415 if (printWithoutType(V: *this, O, Machine: nullptr, M))
5416 return;
5417
5418 SlotTracker Machine(M);
5419 ModuleSlotTracker MST(Machine, M);
5420 printAsOperandImpl(V: *this, O, PrintType, MST);
5421}
5422
5423void Value::printAsOperand(raw_ostream &O, bool PrintType,
5424 ModuleSlotTracker &MST) const {
5425 if (!PrintType)
5426 if (printWithoutType(V: *this, O, Machine: MST.getMachine(), M: MST.getModule()))
5427 return;
5428
5429 printAsOperandImpl(V: *this, O, PrintType, MST);
5430}
5431
5432/// Recursive version of printMetadataImpl.
5433static void printMetadataImplRec(raw_ostream &ROS, const Metadata &MD,
5434 AsmWriterContext &WriterCtx) {
5435 formatted_raw_ostream OS(ROS);
5436 writeAsOperandInternal(Out&: OS, MD: &MD, WriterCtx, /* FromValue */ true);
5437
5438 auto *N = dyn_cast<MDNode>(Val: &MD);
5439 if (!N || isa<DIExpression>(Val: MD))
5440 return;
5441
5442 OS << " = ";
5443 writeMDNodeBodyInternal(Out&: OS, Node: N, Ctx&: WriterCtx);
5444}
5445
5446namespace {
5447struct MDTreeAsmWriterContext : public AsmWriterContext {
5448 unsigned Level;
5449 // {Level, Printed string}
5450 using EntryTy = std::pair<unsigned, std::string>;
5451 SmallVector<EntryTy, 4> Buffer;
5452
5453 // Used to break the cycle in case there is any.
5454 SmallPtrSet<const Metadata *, 4> Visited;
5455
5456 raw_ostream &MainOS;
5457
5458 MDTreeAsmWriterContext(TypePrinting *TP, SlotTracker *ST, const Module *M,
5459 const ModuleSlotTracker *MST, raw_ostream &OS,
5460 const Metadata *InitMD)
5461 : AsmWriterContext(TP, ST, M, MST), Level(0U), Visited({InitMD}),
5462 MainOS(OS) {}
5463
5464 void onWriteMetadataAsOperand(const Metadata *MD) override {
5465 if (!Visited.insert(Ptr: MD).second)
5466 return;
5467
5468 std::string Str;
5469 raw_string_ostream SS(Str);
5470 ++Level;
5471 // A placeholder entry to memorize the correct
5472 // position in buffer.
5473 Buffer.emplace_back(Args: std::make_pair(x&: Level, y: ""));
5474 unsigned InsertIdx = Buffer.size() - 1;
5475
5476 printMetadataImplRec(ROS&: SS, MD: *MD, WriterCtx&: *this);
5477 Buffer[InsertIdx].second = std::move(SS.str());
5478 --Level;
5479 }
5480
5481 ~MDTreeAsmWriterContext() override {
5482 for (const auto &Entry : Buffer) {
5483 MainOS << "\n";
5484 unsigned NumIndent = Entry.first * 2U;
5485 MainOS.indent(NumSpaces: NumIndent) << Entry.second;
5486 }
5487 }
5488};
5489} // end anonymous namespace
5490
5491static void printMetadataImpl(raw_ostream &ROS, const Metadata &MD,
5492 ModuleSlotTracker &MST, const Module *M,
5493 bool OnlyAsOperand, bool PrintAsTree = false) {
5494 formatted_raw_ostream OS(ROS);
5495
5496 TypePrinting TypePrinter(M);
5497
5498 std::unique_ptr<AsmWriterContext> WriterCtx;
5499 if (PrintAsTree && !OnlyAsOperand)
5500 WriterCtx = std::make_unique<MDTreeAsmWriterContext>(
5501 args: &TypePrinter, args: MST.getMachine(), args&: M, args: &MST, args&: OS, args: &MD);
5502 else
5503 WriterCtx = std::make_unique<AsmWriterContext>(args: &TypePrinter,
5504 args: MST.getMachine(), args&: M, args: &MST);
5505
5506 writeAsOperandInternal(Out&: OS, MD: &MD, WriterCtx&: *WriterCtx, /* FromValue */ true);
5507
5508 auto *N = dyn_cast<MDNode>(Val: &MD);
5509 if (OnlyAsOperand || !N || isa<DIExpression>(Val: MD))
5510 return;
5511
5512 OS << " = ";
5513 writeMDNodeBodyInternal(Out&: OS, Node: N, Ctx&: *WriterCtx);
5514}
5515
5516void Metadata::printAsOperand(raw_ostream &OS, const Module *M) const {
5517 ModuleSlotTracker MST(M);
5518 printMetadataImpl(ROS&: OS, MD: *this, MST, M, /* OnlyAsOperand */ true);
5519}
5520
5521void Metadata::printAsOperand(raw_ostream &OS, ModuleSlotTracker &MST,
5522 const Module *M) const {
5523 printMetadataImpl(ROS&: OS, MD: *this, MST, M, /* OnlyAsOperand */ true);
5524}
5525
5526void Metadata::print(raw_ostream &OS, const Module *M,
5527 bool /*IsForDebug*/) const {
5528 ModuleSlotTracker MST(M);
5529 printMetadataImpl(ROS&: OS, MD: *this, MST, M, /* OnlyAsOperand */ false);
5530}
5531
5532void Metadata::print(raw_ostream &OS, ModuleSlotTracker &MST,
5533 const Module *M, bool /*IsForDebug*/) const {
5534 printMetadataImpl(ROS&: OS, MD: *this, MST, M, /* OnlyAsOperand */ false);
5535}
5536
5537void MDNode::printTree(raw_ostream &OS, const Module *M) const {
5538 ModuleSlotTracker MST(M);
5539 printMetadataImpl(ROS&: OS, MD: *this, MST, M, /* OnlyAsOperand */ false,
5540 /*PrintAsTree=*/true);
5541}
5542
5543void MDNode::printTree(raw_ostream &OS, ModuleSlotTracker &MST,
5544 const Module *M) const {
5545 printMetadataImpl(ROS&: OS, MD: *this, MST, M, /* OnlyAsOperand */ false,
5546 /*PrintAsTree=*/true);
5547}
5548
5549void ModuleSummaryIndex::print(raw_ostream &ROS, bool IsForDebug) const {
5550 SlotTracker SlotTable(this);
5551 formatted_raw_ostream OS(ROS);
5552 AssemblyWriter W(OS, SlotTable, this, IsForDebug);
5553 W.printModuleSummaryIndex();
5554}
5555
5556void ModuleSlotTracker::collectMDNodes(MachineMDNodeListType &L) const {
5557 SlotTracker *ST = MachineStorage.get();
5558 if (!ST)
5559 return;
5560
5561 for (auto &I : llvm::make_range(x: ST->mdn_begin(), y: ST->mdn_end()))
5562 L.push_back(Elt: std::make_pair(x&: I.second, y&: I.first));
5563}
5564
5565#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
5566// Value::dump - allow easy printing of Values from the debugger.
5567LLVM_DUMP_METHOD
5568void Value::dump() const { print(dbgs(), /*IsForDebug=*/true); dbgs() << '\n'; }
5569
5570// Value::dump - allow easy printing of Values from the debugger.
5571LLVM_DUMP_METHOD
5572void DbgMarker::dump() const {
5573 print(dbgs(), /*IsForDebug=*/true);
5574 dbgs() << '\n';
5575}
5576
5577// Value::dump - allow easy printing of Values from the debugger.
5578LLVM_DUMP_METHOD
5579void DbgRecord::dump() const { print(dbgs(), /*IsForDebug=*/true); dbgs() << '\n'; }
5580
5581// Type::dump - allow easy printing of Types from the debugger.
5582LLVM_DUMP_METHOD
5583void Type::dump() const { print(dbgs(), /*IsForDebug=*/true); dbgs() << '\n'; }
5584
5585// Module::dump() - Allow printing of Modules from the debugger.
5586LLVM_DUMP_METHOD
5587void Module::dump() const {
5588 print(dbgs(), nullptr,
5589 /*ShouldPreserveUseListOrder=*/false, /*IsForDebug=*/true);
5590}
5591
5592// Allow printing of Comdats from the debugger.
5593LLVM_DUMP_METHOD
5594void Comdat::dump() const { print(dbgs(), /*IsForDebug=*/true); }
5595
5596// NamedMDNode::dump() - Allow printing of NamedMDNodes from the debugger.
5597LLVM_DUMP_METHOD
5598void NamedMDNode::dump() const { print(dbgs(), /*IsForDebug=*/true); }
5599
5600LLVM_DUMP_METHOD
5601void Metadata::dump() const { dump(nullptr); }
5602
5603LLVM_DUMP_METHOD
5604void Metadata::dump(const Module *M) const {
5605 print(dbgs(), M, /*IsForDebug=*/true);
5606 dbgs() << '\n';
5607}
5608
5609LLVM_DUMP_METHOD
5610void MDNode::dumpTree() const { dumpTree(nullptr); }
5611
5612LLVM_DUMP_METHOD
5613void MDNode::dumpTree(const Module *M) const {
5614 printTree(dbgs(), M);
5615 dbgs() << '\n';
5616}
5617
5618// Allow printing of ModuleSummaryIndex from the debugger.
5619LLVM_DUMP_METHOD
5620void ModuleSummaryIndex::dump() const { print(dbgs(), /*IsForDebug=*/true); }
5621#endif
5622