blob: bf0be8ecee92bbdffb55303c2a119749577e4c6a [file] [log] [blame]
Eugene Zelenko149178d2017-10-10 22:33:29 +00001//===- AsmPrinter.cpp - Common AsmPrinter code ----------------------------===//
Chris Lattner6a8e0f52004-08-16 23:15:22 +00002//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Chris Lattner6a8e0f52004-08-16 23:15:22 +00006//
7//===----------------------------------------------------------------------===//
8//
9// This file implements the AsmPrinter class.
10//
11//===----------------------------------------------------------------------===//
12
Chandler Carruth6bda14b2017-06-06 11:49:48 +000013#include "llvm/CodeGen/AsmPrinter.h"
Mehdi Aminib550cb12016-04-18 09:17:29 +000014#include "CodeViewDebug.h"
Chris Lattnerf0d6bd32010-04-05 05:11:15 +000015#include "DwarfDebug.h"
16#include "DwarfException.h"
Heejin Ahn24faf852018-10-25 23:55:10 +000017#include "WasmException.h"
Adrian McCarthydb2736d2018-01-09 23:49:30 +000018#include "WinCFGuard.h"
Reid Kleckner60b640b2015-05-28 22:47:01 +000019#include "WinException.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000020#include "llvm/ADT/APFloat.h"
21#include "llvm/ADT/APInt.h"
22#include "llvm/ADT/DenseMap.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +000023#include "llvm/ADT/STLExtras.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000024#include "llvm/ADT/SmallPtrSet.h"
25#include "llvm/ADT/SmallString.h"
26#include "llvm/ADT/SmallVector.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000027#include "llvm/ADT/Statistic.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000028#include "llvm/ADT/StringRef.h"
29#include "llvm/ADT/Triple.h"
30#include "llvm/ADT/Twine.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000031#include "llvm/Analysis/ConstantFolding.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000032#include "llvm/Analysis/EHPersonalities.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000033#include "llvm/Analysis/OptimizationRemarkEmitter.h"
Martin Storsjo489993d2018-08-29 17:28:34 +000034#include "llvm/BinaryFormat/COFF.h"
Zachary Turner264b5d92017-06-07 03:48:56 +000035#include "llvm/BinaryFormat/Dwarf.h"
36#include "llvm/BinaryFormat/ELF.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000037#include "llvm/CodeGen/GCMetadata.h"
Gordon Henriksend930f912008-08-17 18:44:35 +000038#include "llvm/CodeGen/GCMetadataPrinter.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000039#include "llvm/CodeGen/GCStrategy.h"
40#include "llvm/CodeGen/MachineBasicBlock.h"
Chris Lattnerf2991ce2005-11-21 08:25:09 +000041#include "llvm/CodeGen/MachineConstantPool.h"
Michael Zolotukhin8d052a02018-04-09 00:54:47 +000042#include "llvm/CodeGen/MachineDominators.h"
David Greene659c1a92009-11-13 21:34:57 +000043#include "llvm/CodeGen/MachineFrameInfo.h"
David Greene60eb4732009-08-18 19:22:55 +000044#include "llvm/CodeGen/MachineFunction.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000045#include "llvm/CodeGen/MachineFunctionPass.h"
46#include "llvm/CodeGen/MachineInstr.h"
Jakob Stoklund Olesen1995b9f2014-01-12 19:24:08 +000047#include "llvm/CodeGen/MachineInstrBundle.h"
Nate Begeman4ca2ea52006-04-22 18:53:45 +000048#include "llvm/CodeGen/MachineJumpTableInfo.h"
David Greenef354b6e2009-08-10 16:38:07 +000049#include "llvm/CodeGen/MachineLoopInfo.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000050#include "llvm/CodeGen/MachineMemOperand.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000051#include "llvm/CodeGen/MachineModuleInfo.h"
Rafael Espindolab91455b2015-04-07 13:42:44 +000052#include "llvm/CodeGen/MachineModuleInfoImpls.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000053#include "llvm/CodeGen/MachineOperand.h"
Justin Bogner369ba752017-02-24 00:19:22 +000054#include "llvm/CodeGen/MachineOptimizationRemarkEmitter.h"
Than McIntosh30c804b2018-11-26 18:43:48 +000055#include "llvm/CodeGen/StackMaps.h"
David Blaikie3f833ed2017-11-08 01:01:31 +000056#include "llvm/CodeGen/TargetFrameLowering.h"
57#include "llvm/CodeGen/TargetInstrInfo.h"
David Blaikieb3bde2e2017-11-17 01:07:10 +000058#include "llvm/CodeGen/TargetLowering.h"
David Blaikieb3bde2e2017-11-17 01:07:10 +000059#include "llvm/CodeGen/TargetOpcodes.h"
60#include "llvm/CodeGen/TargetRegisterInfo.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000061#include "llvm/IR/BasicBlock.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000062#include "llvm/IR/Comdat.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000063#include "llvm/IR/Constant.h"
64#include "llvm/IR/Constants.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000065#include "llvm/IR/DataLayout.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000066#include "llvm/IR/DebugInfoMetadata.h"
67#include "llvm/IR/DerivedTypes.h"
68#include "llvm/IR/Function.h"
69#include "llvm/IR/GlobalAlias.h"
70#include "llvm/IR/GlobalIFunc.h"
71#include "llvm/IR/GlobalIndirectSymbol.h"
72#include "llvm/IR/GlobalObject.h"
73#include "llvm/IR/GlobalValue.h"
74#include "llvm/IR/GlobalVariable.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000075#include "llvm/IR/Instruction.h"
Rafael Espindola894843c2014-01-07 21:19:40 +000076#include "llvm/IR/Mangler.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000077#include "llvm/IR/Metadata.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000078#include "llvm/IR/Module.h"
79#include "llvm/IR/Operator.h"
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +000080#include "llvm/IR/RemarkStreamer.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000081#include "llvm/IR/Type.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000082#include "llvm/IR/Value.h"
Chris Lattner1e158692010-04-04 18:34:07 +000083#include "llvm/MC/MCAsmInfo.h"
Omer Paparo Bivas2251c792017-10-24 06:16:03 +000084#include "llvm/MC/MCCodePadder.h"
Chris Lattner9d0e7622009-07-27 21:28:04 +000085#include "llvm/MC/MCContext.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000086#include "llvm/MC/MCDirectives.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000087#include "llvm/MC/MCDwarf.h"
Chris Lattnerd1acffc2010-01-23 06:17:14 +000088#include "llvm/MC/MCExpr.h"
David Greenede9cd442009-07-16 22:24:20 +000089#include "llvm/MC/MCInst.h"
Chris Lattner4d2c0f92009-07-31 18:48:30 +000090#include "llvm/MC/MCSection.h"
Martin Storsjod2662c32018-07-25 18:35:31 +000091#include "llvm/MC/MCSectionCOFF.h"
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +000092#include "llvm/MC/MCSectionELF.h"
93#include "llvm/MC/MCSectionMachO.h"
Chris Lattner4d2c0f92009-07-31 18:48:30 +000094#include "llvm/MC/MCStreamer.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000095#include "llvm/MC/MCSubtargetInfo.h"
96#include "llvm/MC/MCSymbol.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +000097#include "llvm/MC/MCSymbolELF.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +000098#include "llvm/MC/MCTargetOptions.h"
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +000099#include "llvm/MC/MCValue.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000100#include "llvm/MC/SectionKind.h"
101#include "llvm/Pass.h"
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +0000102#include "llvm/Remarks/Remark.h"
Francis Visoiu Mistrihcc909812019-07-16 15:24:59 +0000103#include "llvm/Remarks/RemarkFormat.h"
Francis Visoiu Mistrih6ada11f2019-05-30 21:45:59 +0000104#include "llvm/Remarks/RemarkStringTable.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000105#include "llvm/Support/Casting.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +0000106#include "llvm/Support/CommandLine.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000107#include "llvm/Support/Compiler.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +0000108#include "llvm/Support/ErrorHandling.h"
109#include "llvm/Support/Format.h"
110#include "llvm/Support/MathExtras.h"
Taewook Ohf5040b92017-08-10 18:17:11 +0000111#include "llvm/Support/Path.h"
Eric Christopher78a3f6c2015-02-20 06:59:48 +0000112#include "llvm/Support/TargetRegistry.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +0000113#include "llvm/Support/Timer.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +0000114#include "llvm/Support/raw_ostream.h"
David Blaikie6054e652018-03-23 23:58:19 +0000115#include "llvm/Target/TargetLoweringObjectFile.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000116#include "llvm/Target/TargetMachine.h"
Eugene Zelenko149178d2017-10-10 22:33:29 +0000117#include "llvm/Target/TargetOptions.h"
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000118#include <algorithm>
119#include <cassert>
120#include <cinttypes>
121#include <cstdint>
Eugene Zelenko149178d2017-10-10 22:33:29 +0000122#include <iterator>
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000123#include <limits>
124#include <memory>
125#include <string>
126#include <utility>
127#include <vector>
128
Chris Lattner6a8e0f52004-08-16 23:15:22 +0000129using namespace llvm;
130
Chandler Carruth1b9dde02014-04-22 02:02:50 +0000131#define DEBUG_TYPE "asm-printer"
132
Matthias Braun9f15a792016-11-18 19:43:18 +0000133static const char *const DWARFGroupName = "dwarf";
134static const char *const DWARFGroupDescription = "DWARF Emission";
135static const char *const DbgTimerName = "emit";
136static const char *const DbgTimerDescription = "Debug Info Emission";
137static const char *const EHTimerName = "write_exception";
138static const char *const EHTimerDescription = "DWARF Exception Writer";
Adrian McCarthydb2736d2018-01-09 23:49:30 +0000139static const char *const CFGuardName = "Control Flow Guard";
140static const char *const CFGuardDescription = "Control Flow Guard Tables";
Matthias Braun9f15a792016-11-18 19:43:18 +0000141static const char *const CodeViewLineTablesGroupName = "linetables";
142static const char *const CodeViewLineTablesGroupDescription =
143 "CodeView Line Tables";
Torok Edwinf8dba242010-04-07 10:44:46 +0000144
Chris Lattner94a946c2010-01-28 01:02:27 +0000145STATISTIC(EmittedInsts, "Number of machine instrs printed");
146
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +0000147static cl::opt<bool> EnableRemarksSection(
148 "remarks-section",
149 cl::desc("Emit a section containing remark diagnostics metadata"),
150 cl::init(false));
151
Devang Patel8c78a0b2007-05-03 01:11:54 +0000152char AsmPrinter::ID = 0;
Chris Lattnere468f882010-03-13 20:55:24 +0000153
Eugene Zelenko149178d2017-10-10 22:33:29 +0000154using gcp_map_type = DenseMap<GCStrategy *, std::unique_ptr<GCMetadataPrinter>>;
155
Chris Lattner2cf5f9e2010-04-04 17:57:56 +0000156static gcp_map_type &getGCMap(void *&P) {
Craig Topper353eda42014-04-24 06:44:33 +0000157 if (!P)
Chris Lattner2cf5f9e2010-04-04 17:57:56 +0000158 P = new gcp_map_type();
159 return *(gcp_map_type*)P;
160}
161
Chris Lattner08e9e722010-04-28 19:58:07 +0000162/// getGVAlignmentLog2 - Return the alignment to use for the specified global
163/// value in log2 form. This rounds up to the preferred alignment if possible
164/// and legal.
Xing Xueef039a32019-08-25 15:17:25 +0000165unsigned AsmPrinter::getGVAlignmentLog2(const GlobalValue *GV,
166 const DataLayout &DL,
167 unsigned InBits) {
Chris Lattner08e9e722010-04-28 19:58:07 +0000168 unsigned NumBits = 0;
169 if (const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GV))
Mehdi Aminia28d91d2015-03-10 02:37:25 +0000170 NumBits = DL.getPreferredAlignmentLog(GVar);
Jim Grosbach1900c732011-03-29 23:20:22 +0000171
Chris Lattner08e9e722010-04-28 19:58:07 +0000172 // If InBits is specified, round it to it.
173 if (InBits > NumBits)
174 NumBits = InBits;
Jim Grosbach1900c732011-03-29 23:20:22 +0000175
Chris Lattner08e9e722010-04-28 19:58:07 +0000176 // If the GV has a specified alignment, take it into account.
177 if (GV->getAlignment() == 0)
178 return NumBits;
Jim Grosbach1900c732011-03-29 23:20:22 +0000179
Chris Lattner08e9e722010-04-28 19:58:07 +0000180 unsigned GVAlign = Log2_32(GV->getAlignment());
Jim Grosbach1900c732011-03-29 23:20:22 +0000181
Chris Lattner08e9e722010-04-28 19:58:07 +0000182 // If the GVAlign is larger than NumBits, or if we are required to obey
183 // NumBits because the GV has an assigned section, obey it.
184 if (GVAlign > NumBits || GV->hasSection())
185 NumBits = GVAlign;
186 return NumBits;
187}
188
David Blaikie94598322015-01-18 20:29:04 +0000189AsmPrinter::AsmPrinter(TargetMachine &tm, std::unique_ptr<MCStreamer> Streamer)
Eric Christopherd9134482014-08-04 21:25:23 +0000190 : MachineFunctionPass(ID), TM(tm), MAI(tm.getMCAsmInfo()),
Eugene Zelenkod96089b2017-02-14 00:33:36 +0000191 OutContext(Streamer->getContext()), OutStreamer(std::move(Streamer)) {
Lang Hames9ff69c82015-04-24 19:11:51 +0000192 VerboseAsm = OutStreamer->isVerboseAsm();
Evan Cheng5e5a63c2009-03-25 01:47:28 +0000193}
Chris Lattnerf0e9aef2005-12-13 06:32:10 +0000194
Gordon Henriksendbe06d32008-08-17 12:08:44 +0000195AsmPrinter::~AsmPrinter() {
Craig Toppere73658d2014-04-28 04:05:08 +0000196 assert(!DD && Handlers.empty() && "Debug/EH info didn't get finalized");
Jim Grosbach1900c732011-03-29 23:20:22 +0000197
Craig Topper353eda42014-04-24 06:44:33 +0000198 if (GCMetadataPrinters) {
Chris Lattner2cf5f9e2010-04-04 17:57:56 +0000199 gcp_map_type &GCMap = getGCMap(GCMetadataPrinters);
Jim Grosbach1900c732011-03-29 23:20:22 +0000200
Chris Lattner2cf5f9e2010-04-04 17:57:56 +0000201 delete &GCMap;
Craig Topper353eda42014-04-24 06:44:33 +0000202 GCMetadataPrinters = nullptr;
Chris Lattner2cf5f9e2010-04-04 17:57:56 +0000203 }
Gordon Henriksendbe06d32008-08-17 12:08:44 +0000204}
Chris Lattnerf0e9aef2005-12-13 06:32:10 +0000205
Rafael Espindola0db11db2016-06-27 14:19:45 +0000206bool AsmPrinter::isPositionIndependent() const {
Rafael Espindolab1556c42016-06-28 20:13:36 +0000207 return TM.isPositionIndependent();
Rafael Espindola0db11db2016-06-27 14:19:45 +0000208}
209
Chris Lattner67159522010-01-26 04:35:26 +0000210/// getFunctionNumber - Return a unique ID for the current function.
Chris Lattner67159522010-01-26 04:35:26 +0000211unsigned AsmPrinter::getFunctionNumber() const {
212 return MF->getFunctionNumber();
213}
214
Dan Gohman53d4a082010-04-17 16:44:48 +0000215const TargetLoweringObjectFile &AsmPrinter::getObjFileLowering() const {
Eric Christopher36fe0282015-02-03 07:22:52 +0000216 return *TM.getObjFileLowering();
Chris Lattner5e693ed2009-07-28 03:13:23 +0000217}
218
Micah Villmowcdfe20b2012-10-08 16:38:25 +0000219const DataLayout &AsmPrinter::getDataLayout() const {
Mehdi Amini5c0fa582015-07-16 06:04:17 +0000220 return MMI->getModule()->getDataLayout();
Chris Lattner3a383cb2010-04-05 00:13:49 +0000221}
222
Mehdi Amini26d48132015-07-24 16:04:22 +0000223// Do not use the cached DataLayout because some client use it without a Module
Jonas Devliegherea6ef1ab2018-03-18 11:38:41 +0000224// (dsymutil, llvm-dwarfdump).
Matt Arsenault41e5ac42018-03-14 00:36:23 +0000225unsigned AsmPrinter::getPointerSize() const {
226 return TM.getPointerSize(0); // FIXME: Default address space
227}
Mehdi Amini1660cab2015-07-16 05:59:25 +0000228
David Woodhousee6c13e42014-01-28 23:12:42 +0000229const MCSubtargetInfo &AsmPrinter::getSubtargetInfo() const {
Eric Christophera7249ec2015-02-20 07:16:19 +0000230 assert(MF && "getSubtargetInfo requires a valid MachineFunction!");
231 return MF->getSubtarget<MCSubtargetInfo>();
David Woodhousee6c13e42014-01-28 23:12:42 +0000232}
233
234void AsmPrinter::EmitToStreamer(MCStreamer &S, const MCInst &Inst) {
235 S.EmitInstruction(Inst, getSubtargetInfo());
236}
237
Alexey Bataev9d5974a2019-01-22 17:24:16 +0000238void AsmPrinter::emitInitialRawDwarfLocDirective(const MachineFunction &MF) {
239 assert(DD && "Dwarf debug file is not defined.");
240 assert(OutStreamer->hasRawTextSupport() && "Expected assembly output mode.");
241 (void)DD->emitInitialLocDirective(MF, /*CUID=*/0);
242}
243
Chris Lattner0adae252009-08-18 06:15:16 +0000244/// getCurrentSection() - Return the current section we are emitting to.
245const MCSection *AsmPrinter::getCurrentSection() const {
Eric Christopher445c9522016-10-14 05:47:37 +0000246 return OutStreamer->getCurrentSectionOnly();
Chris Lattner0adae252009-08-18 06:15:16 +0000247}
248
Gordon Henriksen5180e852008-01-07 01:30:38 +0000249void AsmPrinter::getAnalysisUsage(AnalysisUsage &AU) const {
Dan Gohman04023152009-07-31 23:37:33 +0000250 AU.setPreservesAll();
Gordon Henriksen5180e852008-01-07 01:30:38 +0000251 MachineFunctionPass::getAnalysisUsage(AU);
Chris Lattnere468f882010-03-13 20:55:24 +0000252 AU.addRequired<MachineModuleInfo>();
Justin Bogner369ba752017-02-24 00:19:22 +0000253 AU.addRequired<MachineOptimizationRemarkEmitterPass>();
Gordon Henriksend930f912008-08-17 18:44:35 +0000254 AU.addRequired<GCModuleInfo>();
Gordon Henriksen5180e852008-01-07 01:30:38 +0000255}
256
Chris Lattner6a8e0f52004-08-16 23:15:22 +0000257bool AsmPrinter::doInitialization(Module &M) {
Chris Lattner0124fe32010-04-04 05:09:10 +0000258 MMI = getAnalysisIfAvailable<MachineModuleInfo>();
Chris Lattnere468f882010-03-13 20:55:24 +0000259
Chris Lattner51d5b432009-07-31 17:42:42 +0000260 // Initialize TargetLoweringObjectFile.
261 const_cast<TargetLoweringObjectFile&>(getObjFileLowering())
262 .Initialize(OutContext, TM);
Jim Grosbach1900c732011-03-29 23:20:22 +0000263
Shiva Chen7cc03bd2019-04-11 04:59:13 +0000264 const_cast<TargetLoweringObjectFile &>(getObjFileLowering())
265 .getModuleMetadata(M);
266
Lang Hames9ff69c82015-04-24 19:11:51 +0000267 OutStreamer->InitSections(false);
Nico Rieck316c3742013-07-04 21:37:26 +0000268
Simi Pallipurathad96d9a2018-03-06 14:35:23 +0000269 // Emit the version-min deployment target directive if needed.
Jim Grosbacheaf60362014-03-18 22:09:08 +0000270 //
271 // FIXME: If we end up with a collection of these sorts of Darwin-specific
272 // or ELF-specific things, it may make sense to have a platform helper class
273 // that will work with the target helper class. For now keep it here, as the
274 // alternative is duplicated code in each of the target asm printers that
275 // use the directive, where it would need the same conditionalization
276 // anyway.
Matthias Braun5dd72ad2017-12-14 03:59:24 +0000277 const Triple &Target = TM.getTargetTriple();
Alex Lorenzafa75d72018-12-14 01:14:10 +0000278 OutStreamer->EmitVersionForTarget(Target, M.getSDKVersion());
Jim Grosbacheaf60362014-03-18 22:09:08 +0000279
Bob Wilsonb633d7a2009-09-30 22:06:26 +0000280 // Allow the target to emit any magic that it wants at the start of the file.
281 EmitStartOfAsmFile(M);
Rafael Espindolacda011b2008-12-03 11:01:37 +0000282
Chris Lattner601ef332010-01-25 18:58:59 +0000283 // Very minimal debug info. It is ignored if we emit actual debug info. If we
284 // don't, this at least helps the user find where a global came from.
Chris Lattnere9a75a62009-08-22 21:43:10 +0000285 if (MAI->hasSingleParameterDotFile()) {
Chris Lattner601ef332010-01-25 18:58:59 +0000286 // .file "foo.c"
Taewook Ohf5040b92017-08-10 18:17:11 +0000287 OutStreamer->EmitFileDirective(
288 llvm::sys::path::filename(M.getSourceFileName()));
Rafael Espindolacda011b2008-12-03 11:01:37 +0000289 }
290
Bob Wilsonb633d7a2009-09-30 22:06:26 +0000291 GCModuleInfo *MI = getAnalysisIfAvailable<GCModuleInfo>();
292 assert(MI && "AsmPrinter didn't require GCModuleInfo?");
Jim Grosbach6623e7f2014-04-16 22:38:02 +0000293 for (auto &I : *MI)
294 if (GCMetadataPrinter *MP = GetOrCreateGCPrinter(*I))
Philip Reames1e308972014-12-11 01:47:23 +0000295 MP->beginAssembly(M, *MI, *this);
Chris Lattner8a87fb72010-04-03 21:35:55 +0000296
297 // Emit module-level inline asm if it exists.
Chris Lattnerfed39fa2010-04-03 21:13:18 +0000298 if (!M.getModuleInlineAsm().empty()) {
Akira Hatanaka322ffce2015-03-16 18:02:16 +0000299 // We're at the module level. Construct MCSubtarget from the default CPU
300 // and target triple.
301 std::unique_ptr<MCSubtargetInfo> STI(TM.getTarget().createMCSubtargetInfo(
Daniel Sanders335487a2015-06-16 13:15:50 +0000302 TM.getTargetTriple().str(), TM.getTargetCPU(),
303 TM.getTargetFeatureString()));
Lang Hames9ff69c82015-04-24 19:11:51 +0000304 OutStreamer->AddComment("Start of file scope inline assembly");
305 OutStreamer->AddBlankLine();
Akira Hatanakab11ef082015-11-14 06:35:56 +0000306 EmitInlineAsm(M.getModuleInlineAsm()+"\n",
307 OutContext.getSubtargetCopy(*STI), TM.Options.MCOptions);
Lang Hames9ff69c82015-04-24 19:11:51 +0000308 OutStreamer->AddComment("End of file scope inline assembly");
309 OutStreamer->AddBlankLine();
Chris Lattnerfed39fa2010-04-03 21:13:18 +0000310 }
Chris Lattnere3a79262006-01-23 23:47:53 +0000311
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000312 if (MAI->doesSupportDebugInformation()) {
Reid Kleckner12d2c122015-08-05 22:26:20 +0000313 bool EmitCodeView = MMI->getModule()->getCodeViewFlag();
Martin Storsjo966d3452018-05-08 20:56:04 +0000314 if (EmitCodeView && TM.getTargetTriple().isOSWindows()) {
Jonas Devlieghere0eaee542019-08-15 15:54:37 +0000315 Handlers.emplace_back(std::make_unique<CodeViewDebug>(this),
Fangrui Song7bce25c2019-04-30 09:14:02 +0000316 DbgTimerName, DbgTimerDescription,
317 CodeViewLineTablesGroupName,
318 CodeViewLineTablesGroupDescription);
Timur Iskhodzhanovb6fa52f2014-12-26 17:00:51 +0000319 }
Reid Kleckner12d2c122015-08-05 22:26:20 +0000320 if (!EmitCodeView || MMI->getModule()->getDwarfVersion()) {
Timur Iskhodzhanovb6fa52f2014-12-26 17:00:51 +0000321 DD = new DwarfDebug(this, &M);
Peter Collingbourne5973bc82016-05-20 19:35:35 +0000322 DD->beginModule();
Fangrui Song7bce25c2019-04-30 09:14:02 +0000323 Handlers.emplace_back(std::unique_ptr<DwarfDebug>(DD), DbgTimerName,
324 DbgTimerDescription, DWARFGroupName,
325 DWARFGroupDescription);
Timur Iskhodzhanovb6fa52f2014-12-26 17:00:51 +0000326 }
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000327 }
Anton Korobeynikov61d167e2011-01-14 21:57:45 +0000328
Joerg Sonnenberger83963992017-01-05 20:55:28 +0000329 switch (MAI->getExceptionHandlingType()) {
330 case ExceptionHandling::SjLj:
331 case ExceptionHandling::DwarfCFI:
332 case ExceptionHandling::ARM:
333 isCFIMoveForDebugging = true;
334 if (MAI->getExceptionHandlingType() != ExceptionHandling::DwarfCFI)
335 break;
336 for (auto &F: M.getFunctionList()) {
337 // If the module contains any function with unwind data,
338 // .eh_frame has to be emitted.
339 // Ignore functions that won't get emitted.
340 if (!F.isDeclarationForLinker() && F.needsUnwindTableEntry()) {
341 isCFIMoveForDebugging = false;
342 break;
343 }
344 }
345 break;
346 default:
347 isCFIMoveForDebugging = false;
348 break;
349 }
350
Saleem Abdulrasool8076cab2014-06-11 01:19:03 +0000351 EHStreamer *ES = nullptr;
Rafael Espindola240c96e2011-05-05 19:48:34 +0000352 switch (MAI->getExceptionHandlingType()) {
353 case ExceptionHandling::None:
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000354 break;
Rafael Espindola240c96e2011-05-05 19:48:34 +0000355 case ExceptionHandling::SjLj:
Rafael Espindola240c96e2011-05-05 19:48:34 +0000356 case ExceptionHandling::DwarfCFI:
Saleem Abdulrasool8076cab2014-06-11 01:19:03 +0000357 ES = new DwarfCFIException(this);
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000358 break;
Rafael Espindola240c96e2011-05-05 19:48:34 +0000359 case ExceptionHandling::ARM:
Saleem Abdulrasool8076cab2014-06-11 01:19:03 +0000360 ES = new ARMException(this);
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000361 break;
Reid Kleckner5cc15692015-01-23 18:49:01 +0000362 case ExceptionHandling::WinEH:
Saleem Abdulrasoold1a4ed62014-09-01 23:48:39 +0000363 switch (MAI->getWinEHEncodingType()) {
364 default: llvm_unreachable("unsupported unwinding information encoding");
Reid Kleckner0738a9c2015-05-05 17:44:16 +0000365 case WinEH::EncodingType::Invalid:
366 break;
Reid Kleckner1d3d4ad2015-05-29 17:00:57 +0000367 case WinEH::EncodingType::X86:
Saleem Abdulrasoold1a4ed62014-09-01 23:48:39 +0000368 case WinEH::EncodingType::Itanium:
Reid Kleckner60b640b2015-05-28 22:47:01 +0000369 ES = new WinException(this);
Saleem Abdulrasoold1a4ed62014-09-01 23:48:39 +0000370 break;
371 }
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000372 break;
Heejin Ahn9386bde2018-02-24 00:40:50 +0000373 case ExceptionHandling::Wasm:
Heejin Ahn24faf852018-10-25 23:55:10 +0000374 ES = new WasmException(this);
Heejin Ahn9386bde2018-02-24 00:40:50 +0000375 break;
Rafael Espindola240c96e2011-05-05 19:48:34 +0000376 }
Saleem Abdulrasool8076cab2014-06-11 01:19:03 +0000377 if (ES)
Fangrui Song7bce25c2019-04-30 09:14:02 +0000378 Handlers.emplace_back(std::unique_ptr<EHStreamer>(ES), EHTimerName,
379 EHTimerDescription, DWARFGroupName,
380 DWARFGroupDescription);
Adrian McCarthydb2736d2018-01-09 23:49:30 +0000381
382 if (mdconst::extract_or_null<ConstantInt>(
Hans Wennborgd4090be2018-08-10 09:48:53 +0000383 MMI->getModule()->getModuleFlag("cfguardtable")))
Jonas Devlieghere0eaee542019-08-15 15:54:37 +0000384 Handlers.emplace_back(std::make_unique<WinCFGuard>(this), CFGuardName,
Fangrui Song7bce25c2019-04-30 09:14:02 +0000385 CFGuardDescription, DWARFGroupName,
386 DWARFGroupDescription);
Adrian McCarthydb2736d2018-01-09 23:49:30 +0000387
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000388 return false;
Chris Lattner6a8e0f52004-08-16 23:15:22 +0000389}
390
Rafael Espindolaa0053422014-02-07 16:07:11 +0000391static bool canBeHidden(const GlobalValue *GV, const MCAsmInfo &MAI) {
Rafael Espindolaa0053422014-02-07 16:07:11 +0000392 if (!MAI.hasWeakDefCanBeHiddenDirective())
393 return false;
394
David Blaikie88209292018-03-21 19:23:45 +0000395 return GV->canBeOmittedFromSymbolTable();
Rafael Espindolaa0053422014-02-07 16:07:11 +0000396}
397
Rafael Espindola6f1b2852013-10-30 22:08:11 +0000398void AsmPrinter::EmitLinkage(const GlobalValue *GV, MCSymbol *GVSym) const {
399 GlobalValue::LinkageTypes Linkage = GV->getLinkage();
Rafael Espindolab0287742013-10-23 21:24:34 +0000400 switch (Linkage) {
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000401 case GlobalValue::CommonLinkage:
402 case GlobalValue::LinkOnceAnyLinkage:
403 case GlobalValue::LinkOnceODRLinkage:
404 case GlobalValue::WeakAnyLinkage:
405 case GlobalValue::WeakODRLinkage:
Rafael Espindola04867ce2013-12-02 23:04:51 +0000406 if (MAI->hasWeakDefDirective()) {
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000407 // .globl _foo
Lang Hames9ff69c82015-04-24 19:11:51 +0000408 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_Global);
Bill Wendling578ee402010-08-20 22:05:50 +0000409
Rafael Espindolaa0053422014-02-07 16:07:11 +0000410 if (!canBeHidden(GV, *MAI))
Bill Wendling578ee402010-08-20 22:05:50 +0000411 // .weak_definition _foo
Lang Hames9ff69c82015-04-24 19:11:51 +0000412 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_WeakDefinition);
Bill Wendling578ee402010-08-20 22:05:50 +0000413 else
Lang Hames9ff69c82015-04-24 19:11:51 +0000414 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_WeakDefAutoPrivate);
Rafael Espindola04867ce2013-12-02 23:04:51 +0000415 } else if (MAI->hasLinkOnceDirective()) {
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000416 // .globl _foo
Lang Hames9ff69c82015-04-24 19:11:51 +0000417 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_Global);
Duncan Sands2576db72010-05-12 07:11:33 +0000418 //NOTE: linkonce is handled by the section the symbol was assigned to.
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000419 } else {
420 // .weak _foo
Lang Hames9ff69c82015-04-24 19:11:51 +0000421 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_Weak);
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000422 }
Rafael Espindolab0287742013-10-23 21:24:34 +0000423 return;
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000424 case GlobalValue::ExternalLinkage:
Rafael Espindolae64619c2016-05-16 21:14:24 +0000425 // If external, declare as a global symbol: .globl _foo
Lang Hames9ff69c82015-04-24 19:11:51 +0000426 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_Global);
Rafael Espindolab0287742013-10-23 21:24:34 +0000427 return;
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000428 case GlobalValue::PrivateLinkage:
429 case GlobalValue::InternalLinkage:
Rafael Espindolab0287742013-10-23 21:24:34 +0000430 return;
Rafael Espindolae64619c2016-05-16 21:14:24 +0000431 case GlobalValue::AppendingLinkage:
Rafael Espindolab0287742013-10-23 21:24:34 +0000432 case GlobalValue::AvailableExternallyLinkage:
Rafael Espindolab0287742013-10-23 21:24:34 +0000433 case GlobalValue::ExternalWeakLinkage:
Rafael Espindola610a4e92016-05-11 14:41:30 +0000434 llvm_unreachable("Should never emit this");
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000435 }
Rafael Espindolab0287742013-10-23 21:24:34 +0000436 llvm_unreachable("Unknown linkage type!");
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000437}
438
Rafael Espindoladaeafb42014-02-19 17:23:20 +0000439void AsmPrinter::getNameWithPrefix(SmallVectorImpl<char> &Name,
440 const GlobalValue *GV) const {
Eric Christopher4367c7f2016-09-16 07:33:15 +0000441 TM.getNameWithPrefix(Name, GV, getObjFileLowering().getMangler());
Rafael Espindoladaeafb42014-02-19 17:23:20 +0000442}
443
Rafael Espindola79858aa2013-10-29 17:07:16 +0000444MCSymbol *AsmPrinter::getSymbol(const GlobalValue *GV) const {
Tim Northoverb64fb452016-11-22 16:17:20 +0000445 return TM.getSymbol(GV);
Rafael Espindola79858aa2013-10-29 17:07:16 +0000446}
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000447
Chris Lattner6a160512010-01-19 04:39:15 +0000448/// EmitGlobalVariable - Emit the specified global variable to the .s file.
449void AsmPrinter::EmitGlobalVariable(const GlobalVariable *GV) {
Chih-Hung Hsieh9f9e4682018-02-28 17:48:55 +0000450 bool IsEmuTLSVar = TM.useEmulatedTLS() && GV->isThreadLocal();
Chih-Hung Hsieh1e859582015-07-28 16:24:05 +0000451 assert(!(IsEmuTLSVar && GV->hasCommonLinkage()) &&
452 "No emulated TLS variables in the common section");
453
Chih-Hung Hsieh57886402016-01-13 23:56:37 +0000454 // Never emit TLS variable xyz in emulated TLS model.
455 // The initialization value is in __emutls_t.xyz instead of xyz.
456 if (IsEmuTLSVar)
457 return;
458
Rafael Espindola7dd4d6e2011-04-05 15:51:32 +0000459 if (GV->hasInitializer()) {
460 // Check to see if this is a special global used by LLVM, if so, emit it.
461 if (EmitSpecialLLVMGlobal(GV))
462 return;
Jim Grosbach1900c732011-03-29 23:20:22 +0000463
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +0000464 // Skip the emission of global equivalents. The symbol can be emitted later
465 // on by emitGlobalGOTEquivs in case it turns out to be needed.
466 if (GlobalGOTEquivs.count(getSymbol(GV)))
467 return;
468
Chih-Hung Hsieh57886402016-01-13 23:56:37 +0000469 if (isVerbose()) {
Chih-Hung Hsieh1e859582015-07-28 16:24:05 +0000470 // When printing the control variable __emutls_v.*,
471 // we don't need to print the original TLS variable name.
Lang Hames9ff69c82015-04-24 19:11:51 +0000472 GV->printAsOperand(OutStreamer->GetCommentOS(),
Rafael Espindola7dd4d6e2011-04-05 15:51:32 +0000473 /*PrintType=*/false, GV->getParent());
Lang Hames9ff69c82015-04-24 19:11:51 +0000474 OutStreamer->GetCommentOS() << '\n';
Rafael Espindola7dd4d6e2011-04-05 15:51:32 +0000475 }
Eric Christopherf3925432010-05-25 21:49:43 +0000476 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000477
Rafael Espindola79858aa2013-10-29 17:07:16 +0000478 MCSymbol *GVSym = getSymbol(GV);
Chih-Hung Hsieh57886402016-01-13 23:56:37 +0000479 MCSymbol *EmittedSym = GVSym;
Chih-Hung Hsieh1e859582015-07-28 16:24:05 +0000480
Eric Christopher0b6537e2016-07-01 06:07:31 +0000481 // getOrCreateEmuTLSControlSym only creates the symbol with name and default
482 // attributes.
483 // GV's or GVSym's attributes will be used for the EmittedSym.
Chih-Hung Hsieh1e859582015-07-28 16:24:05 +0000484 EmitVisibility(EmittedSym, GV->getVisibility(), !GV->isDeclaration());
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000485
Rafael Espindola7dd4d6e2011-04-05 15:51:32 +0000486 if (!GV->hasInitializer()) // External globals require no extra code.
487 return;
488
David Majnemer8e92dfe2014-12-24 22:44:29 +0000489 GVSym->redefineIfPossible();
David Majnemer2913eca2014-12-24 23:06:55 +0000490 if (GVSym->isDefined() || GVSym->isVariable())
491 report_fatal_error("symbol '" + Twine(GVSym->getName()) +
492 "' is already defined");
David Majnemer8e92dfe2014-12-24 22:44:29 +0000493
Chris Lattnerbc696442010-01-25 18:33:40 +0000494 if (MAI->hasDotTypeDotSizeDirective())
Chih-Hung Hsieh1e859582015-07-28 16:24:05 +0000495 OutStreamer->EmitSymbolAttribute(EmittedSym, MCSA_ELF_TypeObject);
Jim Grosbach1900c732011-03-29 23:20:22 +0000496
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000497 SectionKind GVKind = TargetLoweringObjectFile::getKindForGlobal(GV, TM);
498
Mehdi Amini5c0fa582015-07-16 06:04:17 +0000499 const DataLayout &DL = GV->getParent()->getDataLayout();
Tim Northover67828ed2019-07-11 13:13:02 +0000500 uint64_t Size = DL.getTypeAllocSize(GV->getValueType());
Jim Grosbach1900c732011-03-29 23:20:22 +0000501
Chris Lattnerf740a8c2010-04-26 18:46:46 +0000502 // If the alignment is specified, we *must* obey it. Overaligning a global
503 // with a specified alignment is a prompt way to break globals emitted to
504 // sections and expected to be contiguous (e.g. ObjC metadata).
Mehdi Amini5c0fa582015-07-16 06:04:17 +0000505 unsigned AlignLog = getGVAlignmentLog2(GV, DL);
Jim Grosbach1900c732011-03-29 23:20:22 +0000506
Jim Grosbach6623e7f2014-04-16 22:38:02 +0000507 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +0000508 NamedRegionTimer T(HI.TimerName, HI.TimerDescription,
509 HI.TimerGroupName, HI.TimerGroupDescription,
510 TimePassesIsEnabled);
Jim Grosbach6623e7f2014-04-16 22:38:02 +0000511 HI.Handler->setSymbolSize(GVSym, Size);
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +0000512 }
Richard Mitton089ed892013-09-23 17:56:20 +0000513
Nirav Davef8d00d52016-06-14 15:09:30 +0000514 // Handle common symbols
515 if (GVKind.isCommon()) {
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000516 if (Size == 0) Size = 1; // .comm Foo, 0 is undefined, avoid it.
Benjamin Kramer63970512011-09-01 23:04:27 +0000517 unsigned Align = 1 << AlignLog;
Nirav Davef8d00d52016-06-14 15:09:30 +0000518 if (!getObjFileLowering().getCommDirectiveSupportsAlignment())
519 Align = 0;
Jim Grosbach1900c732011-03-29 23:20:22 +0000520
Nirav Davef8d00d52016-06-14 15:09:30 +0000521 // .comm _foo, 42, 4
522 OutStreamer->EmitCommonSymbol(GVSym, Size, Align);
523 return;
524 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000525
Nirav Davef8d00d52016-06-14 15:09:30 +0000526 // Determine to which section this global should be emitted.
Eric Christopher4367c7f2016-09-16 07:33:15 +0000527 MCSection *TheSection = getObjFileLowering().SectionForGlobal(GV, GVKind, TM);
Jim Grosbach1900c732011-03-29 23:20:22 +0000528
Nirav Davef8d00d52016-06-14 15:09:30 +0000529 // If we have a bss global going to a section that supports the
530 // zerofill directive, do so here.
531 if (GVKind.isBSS() && MAI->hasMachoZeroFillDirective() &&
532 TheSection->isVirtualSection()) {
533 if (Size == 0)
534 Size = 1; // zerofill of 0 bytes is undefined.
535 unsigned Align = 1 << AlignLog;
536 EmitLinkage(GV, GVSym);
537 // .zerofill __DATA, __bss, _foo, 400, 5
538 OutStreamer->EmitZerofill(TheSection, GVSym, Size, Align);
539 return;
540 }
541
542 // If this is a BSS local symbol and we are emitting in the BSS
543 // section use .lcomm/.comm directive.
544 if (GVKind.isBSSLocal() &&
545 getObjFileLowering().getBSSSection() == TheSection) {
546 if (Size == 0)
547 Size = 1; // .comm Foo, 0 is undefined, avoid it.
548 unsigned Align = 1 << AlignLog;
Jim Grosbach1900c732011-03-29 23:20:22 +0000549
Ulrich Weigande5f94052012-11-27 16:11:16 +0000550 // Use .lcomm only if it supports user-specified alignment.
551 // Otherwise, while it would still be correct to use .lcomm in some
552 // cases (e.g. when Align == 1), the external assembler might enfore
553 // some -unknown- default alignment behavior, which could cause
554 // spurious differences between external and integrated assembler.
555 // Prefer to simply fall back to .local / .comm in this case.
556 if (MAI->getLCOMMDirectiveAlignmentType() != LCOMM::NoAlignment) {
Chris Lattnercd2915e2010-01-19 06:25:51 +0000557 // .lcomm _foo, 42
Lang Hames9ff69c82015-04-24 19:11:51 +0000558 OutStreamer->EmitLocalCommonSymbol(GVSym, Size, Align);
Chris Lattnercd2915e2010-01-19 06:25:51 +0000559 return;
560 }
Chris Lattner9f06f912010-09-27 06:44:54 +0000561
Chris Lattner9f06f912010-09-27 06:44:54 +0000562 if (!getObjFileLowering().getCommDirectiveSupportsAlignment())
563 Align = 0;
Jim Grosbach1900c732011-03-29 23:20:22 +0000564
Chris Lattnercd2915e2010-01-19 06:25:51 +0000565 // .local _foo
Lang Hames9ff69c82015-04-24 19:11:51 +0000566 OutStreamer->EmitSymbolAttribute(GVSym, MCSA_Local);
Chris Lattnercd2915e2010-01-19 06:25:51 +0000567 // .comm _foo, 42, 4
Lang Hames9ff69c82015-04-24 19:11:51 +0000568 OutStreamer->EmitCommonSymbol(GVSym, Size, Align);
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000569 return;
570 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000571
Eric Christopher19a4b842010-05-25 21:28:50 +0000572 // Handle thread local data for mach-o which requires us to output an
573 // additional structure of data and mangle the original symbol so that we
574 // can reference it later.
Chris Lattner419d0aa2010-09-05 20:33:40 +0000575 //
576 // TODO: This should become an "emit thread local global" method on TLOF.
577 // All of this macho specific stuff should be sunk down into TLOFMachO and
578 // stuff like "TLSExtraDataSection" should no longer be part of the parent
579 // TLOF class. This will also make it more obvious that stuff like
580 // MCStreamer::EmitTBSSSymbol is macho specific and only called from macho
581 // specific code.
Chih-Hung Hsieh57886402016-01-13 23:56:37 +0000582 if (GVKind.isThreadLocal() && MAI->hasMachoTBSSDirective()) {
Eric Christopher6fdea1b2010-05-22 00:10:22 +0000583 // Emit the .tbss symbol
Jim Grosbach1900c732011-03-29 23:20:22 +0000584 MCSymbol *MangSym =
Eric Christopher4367c7f2016-09-16 07:33:15 +0000585 OutContext.getOrCreateSymbol(GVSym->getName() + Twine("$tlv$init"));
Jim Grosbach1900c732011-03-29 23:20:22 +0000586
Eric Christopher7af8baf2013-08-07 21:13:06 +0000587 if (GVKind.isThreadBSS()) {
588 TheSection = getObjFileLowering().getTLSBSSSection();
Lang Hames9ff69c82015-04-24 19:11:51 +0000589 OutStreamer->EmitTBSSSymbol(TheSection, MangSym, Size, 1 << AlignLog);
Eric Christopher7af8baf2013-08-07 21:13:06 +0000590 } else if (GVKind.isThreadData()) {
Lang Hames9ff69c82015-04-24 19:11:51 +0000591 OutStreamer->SwitchSection(TheSection);
Eric Christopher19a4b842010-05-25 21:28:50 +0000592
Jim Grosbach1900c732011-03-29 23:20:22 +0000593 EmitAlignment(AlignLog, GV);
Lang Hames9ff69c82015-04-24 19:11:51 +0000594 OutStreamer->EmitLabel(MangSym);
Jim Grosbach1900c732011-03-29 23:20:22 +0000595
Mehdi Aminibd7287e2015-07-16 06:11:10 +0000596 EmitGlobalConstant(GV->getParent()->getDataLayout(),
597 GV->getInitializer());
Eric Christopher19a4b842010-05-25 21:28:50 +0000598 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000599
Lang Hames9ff69c82015-04-24 19:11:51 +0000600 OutStreamer->AddBlankLine();
Jim Grosbach1900c732011-03-29 23:20:22 +0000601
Eric Christopher6fdea1b2010-05-22 00:10:22 +0000602 // Emit the variable struct for the runtime.
Rafael Espindola0709a7b2015-05-21 19:20:38 +0000603 MCSection *TLVSect = getObjFileLowering().getTLSExtraDataSection();
Jim Grosbach1900c732011-03-29 23:20:22 +0000604
Lang Hames9ff69c82015-04-24 19:11:51 +0000605 OutStreamer->SwitchSection(TLVSect);
Eric Christopher6fdea1b2010-05-22 00:10:22 +0000606 // Emit the linkage here.
Rafael Espindola6f1b2852013-10-30 22:08:11 +0000607 EmitLinkage(GV, GVSym);
Lang Hames9ff69c82015-04-24 19:11:51 +0000608 OutStreamer->EmitLabel(GVSym);
Jim Grosbach1900c732011-03-29 23:20:22 +0000609
Eric Christopher6fdea1b2010-05-22 00:10:22 +0000610 // Three pointers in size:
611 // - __tlv_bootstrap - used to make sure support exists
612 // - spare pointer, used when mapped by the runtime
613 // - pointer to mangled symbol above with initializer
Mehdi Amini5c0fa582015-07-16 06:04:17 +0000614 unsigned PtrSize = DL.getPointerTypeSize(GV->getType());
Lang Hames9ff69c82015-04-24 19:11:51 +0000615 OutStreamer->EmitSymbolValue(GetExternalSymbolSymbol("_tlv_bootstrap"),
Eric Christopher5fdd68e2013-06-24 23:20:02 +0000616 PtrSize);
Lang Hames9ff69c82015-04-24 19:11:51 +0000617 OutStreamer->EmitIntValue(0, PtrSize);
618 OutStreamer->EmitSymbolValue(MangSym, PtrSize);
Jim Grosbach1900c732011-03-29 23:20:22 +0000619
Lang Hames9ff69c82015-04-24 19:11:51 +0000620 OutStreamer->AddBlankLine();
Eric Christopher27e7ffc2010-05-20 00:49:07 +0000621 return;
622 }
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000623
Nirav Davef8d00d52016-06-14 15:09:30 +0000624 MCSymbol *EmittedInitSym = GVSym;
625
Lang Hames9ff69c82015-04-24 19:11:51 +0000626 OutStreamer->SwitchSection(TheSection);
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000627
Chih-Hung Hsieh57886402016-01-13 23:56:37 +0000628 EmitLinkage(GV, EmittedInitSym);
Chris Lattnere80442a2010-04-26 18:30:45 +0000629 EmitAlignment(AlignLog, GV);
Chris Lattnerdbf5a762010-01-26 23:47:12 +0000630
Chih-Hung Hsieh1e859582015-07-28 16:24:05 +0000631 OutStreamer->EmitLabel(EmittedInitSym);
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000632
Mehdi Aminibd7287e2015-07-16 06:11:10 +0000633 EmitGlobalConstant(GV->getParent()->getDataLayout(), GV->getInitializer());
Chris Lattnerc7a062d2010-01-19 05:38:33 +0000634
635 if (MAI->hasDotTypeDotSizeDirective())
Chris Lattner866a5392010-01-25 07:53:05 +0000636 // .size foo, 42
Dan Gohman734c59d2016-12-01 23:39:08 +0000637 OutStreamer->emitELFSize(EmittedInitSym,
Rafael Espindolaa8695762015-06-02 00:25:12 +0000638 MCConstantExpr::create(Size, OutContext));
Jim Grosbach1900c732011-03-29 23:20:22 +0000639
Lang Hames9ff69c82015-04-24 19:11:51 +0000640 OutStreamer->AddBlankLine();
Chris Lattner6a160512010-01-19 04:39:15 +0000641}
642
Petar Jovanovicdbb39352017-01-20 17:53:30 +0000643/// Emit the directive and value for debug thread local expression
644///
645/// \p Value - The value to emit.
646/// \p Size - The size of the integer (in bytes) to emit.
Simon Atanasyanaf860d42018-11-28 11:48:07 +0000647void AsmPrinter::EmitDebugValue(const MCExpr *Value, unsigned Size) const {
Petar Jovanovicdbb39352017-01-20 17:53:30 +0000648 OutStreamer->EmitValue(Value, Size);
649}
650
Chris Lattnerc7b91152010-01-26 23:18:44 +0000651/// EmitFunctionHeader - This method emits the header for the current
652/// function.
653void AsmPrinter::EmitFunctionHeader() {
Matthias Braunf1caa282017-12-15 22:22:58 +0000654 const Function &F = MF->getFunction();
Francis Visoiu Mistrih1c987012017-05-23 21:22:16 +0000655
656 if (isVerbose())
Reid Kleckner4a4d7262017-06-30 18:22:51 +0000657 OutStreamer->GetCommentOS()
658 << "-- Begin function "
Matthias Braunf1caa282017-12-15 22:22:58 +0000659 << GlobalValue::dropLLVMManglingEscape(F.getName()) << '\n';
Francis Visoiu Mistrih1c987012017-05-23 21:22:16 +0000660
Chris Lattnerc7b91152010-01-26 23:18:44 +0000661 // Print out constants referenced by the function
Chris Lattner2b796242010-01-28 00:19:24 +0000662 EmitConstantPool();
Jim Grosbach1900c732011-03-29 23:20:22 +0000663
Chris Lattnerc7b91152010-01-26 23:18:44 +0000664 // Print the 'header' of function.
Matthias Braunf1caa282017-12-15 22:22:58 +0000665 OutStreamer->SwitchSection(getObjFileLowering().SectionForGlobal(&F, TM));
666 EmitVisibility(CurrentFnSym, F.getVisibility());
Chris Lattnerc7b91152010-01-26 23:18:44 +0000667
Matthias Braunf1caa282017-12-15 22:22:58 +0000668 EmitLinkage(&F, CurrentFnSym);
Jingyue Wue8290f22015-03-12 01:50:30 +0000669 if (MAI->hasFunctionAlignment())
Guillaume Chateletaff45e42019-09-05 10:00:22 +0000670 EmitAlignment(MF->getLogAlignment(), &F);
Chris Lattner22fc38d2010-01-26 23:41:48 +0000671
Chris Lattnerc7b91152010-01-26 23:18:44 +0000672 if (MAI->hasDotTypeDotSizeDirective())
Lang Hames9ff69c82015-04-24 19:11:51 +0000673 OutStreamer->EmitSymbolAttribute(CurrentFnSym, MCSA_ELF_TypeFunction);
Chris Lattnerc7b91152010-01-26 23:18:44 +0000674
Vedant Kumar13ef84f2019-01-25 18:30:22 +0000675 if (F.hasFnAttribute(Attribute::Cold))
676 OutStreamer->EmitSymbolAttribute(CurrentFnSym, MCSA_Cold);
677
Chris Lattner7bde8c02010-04-04 18:52:31 +0000678 if (isVerbose()) {
Matthias Braunf1caa282017-12-15 22:22:58 +0000679 F.printAsOperand(OutStreamer->GetCommentOS(),
680 /*PrintType=*/false, F.getParent());
Lang Hames9ff69c82015-04-24 19:11:51 +0000681 OutStreamer->GetCommentOS() << '\n';
Chris Lattnerc7b91152010-01-26 23:18:44 +0000682 }
Chris Lattnerc7b91152010-01-26 23:18:44 +0000683
Peter Collingbourne51d2de72014-12-03 02:08:38 +0000684 // Emit the prefix data.
Matthias Braunf1caa282017-12-15 22:22:58 +0000685 if (F.hasPrefixData()) {
Peter Collingbourne7f6e2c92017-03-15 04:18:16 +0000686 if (MAI->hasSubsectionsViaSymbols()) {
687 // Preserving prefix data on platforms which use subsections-via-symbols
688 // is a bit tricky. Here we introduce a symbol for the prefix data
689 // and use the .alt_entry attribute to mark the function's real entry point
690 // as an alternative entry point to the prefix-data symbol.
691 MCSymbol *PrefixSym = OutContext.createLinkerPrivateTempSymbol();
692 OutStreamer->EmitLabel(PrefixSym);
693
Matthias Braunf1caa282017-12-15 22:22:58 +0000694 EmitGlobalConstant(F.getParent()->getDataLayout(), F.getPrefixData());
Peter Collingbourne7f6e2c92017-03-15 04:18:16 +0000695
696 // Emit an .alt_entry directive for the actual function symbol.
697 OutStreamer->EmitSymbolAttribute(CurrentFnSym, MCSA_AltEntry);
698 } else {
Matthias Braunf1caa282017-12-15 22:22:58 +0000699 EmitGlobalConstant(F.getParent()->getDataLayout(), F.getPrefixData());
Peter Collingbourne7f6e2c92017-03-15 04:18:16 +0000700 }
701 }
Peter Collingbourne51d2de72014-12-03 02:08:38 +0000702
Dan Gohman4a618822010-02-10 16:03:48 +0000703 // Emit the CurrentFnSym. This is a virtual function to allow targets to
Chris Lattnerbc1e6f02010-01-27 07:21:55 +0000704 // do their wild and crazy things as required.
705 EmitFunctionEntryLabel();
Jim Grosbach1900c732011-03-29 23:20:22 +0000706
Chris Lattner561334a2010-03-15 20:39:00 +0000707 // If the function had address-taken blocks that got deleted, then we have
708 // references to the dangling symbols. Emit them at the start of the function
709 // so that we don't get references to undefined symbols.
710 std::vector<MCSymbol*> DeadBlockSyms;
Matthias Braunf1caa282017-12-15 22:22:58 +0000711 MMI->takeDeletedSymbolsForFunction(&F, DeadBlockSyms);
Chris Lattner561334a2010-03-15 20:39:00 +0000712 for (unsigned i = 0, e = DeadBlockSyms.size(); i != e; ++i) {
Lang Hames9ff69c82015-04-24 19:11:51 +0000713 OutStreamer->AddComment("Address taken block that was later removed");
714 OutStreamer->EmitLabel(DeadBlockSyms[i]);
Chris Lattner561334a2010-03-15 20:39:00 +0000715 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000716
Rafael Espindola092b6192015-03-05 19:47:50 +0000717 if (CurrentFnBegin) {
Rafael Espindola629cdba2015-02-27 18:18:39 +0000718 if (MAI->useAssignmentForEHBegin()) {
Jim Grosbach6f482002015-05-18 18:43:14 +0000719 MCSymbol *CurPos = OutContext.createTempSymbol();
Lang Hames9ff69c82015-04-24 19:11:51 +0000720 OutStreamer->EmitLabel(CurPos);
721 OutStreamer->EmitAssignment(CurrentFnBegin,
Jim Grosbach13760bd2015-05-30 01:25:56 +0000722 MCSymbolRefExpr::create(CurPos, OutContext));
Rafael Espindola629cdba2015-02-27 18:18:39 +0000723 } else {
Lang Hames9ff69c82015-04-24 19:11:51 +0000724 OutStreamer->EmitLabel(CurrentFnBegin);
Rafael Espindola629cdba2015-02-27 18:18:39 +0000725 }
726 }
727
Chris Lattnerc7b91152010-01-26 23:18:44 +0000728 // Emit pre-function debug and/or EH information.
Jim Grosbach6623e7f2014-04-16 22:38:02 +0000729 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +0000730 NamedRegionTimer T(HI.TimerName, HI.TimerDescription, HI.TimerGroupName,
731 HI.TimerGroupDescription, TimePassesIsEnabled);
Jim Grosbach6623e7f2014-04-16 22:38:02 +0000732 HI.Handler->beginFunction(MF);
Torok Edwinf8dba242010-04-07 10:44:46 +0000733 }
Peter Collingbournee9f45e22013-10-20 02:16:21 +0000734
Peter Collingbourne51d2de72014-12-03 02:08:38 +0000735 // Emit the prologue data.
Matthias Braunf1caa282017-12-15 22:22:58 +0000736 if (F.hasPrologueData())
737 EmitGlobalConstant(F.getParent()->getDataLayout(), F.getPrologueData());
Chris Lattnerc7b91152010-01-26 23:18:44 +0000738}
739
Chris Lattnerbc1e6f02010-01-27 07:21:55 +0000740/// EmitFunctionEntryLabel - Emit the label that is the entrypoint for the
741/// function. This can be overridden by targets as required to do custom stuff.
742void AsmPrinter::EmitFunctionEntryLabel() {
David Majnemer58cb80c2014-12-24 10:27:50 +0000743 CurrentFnSym->redefineIfPossible();
744
Chris Lattner35096e82010-05-06 00:05:37 +0000745 // The function label could have already been emitted if two symbols end up
746 // conflicting due to asm renaming. Detect this and emit an error.
David Majnemer2913eca2014-12-24 23:06:55 +0000747 if (CurrentFnSym->isVariable())
748 report_fatal_error("'" + Twine(CurrentFnSym->getName()) +
749 "' is a protected alias");
750 if (CurrentFnSym->isDefined())
751 report_fatal_error("'" + Twine(CurrentFnSym->getName()) +
752 "' label emitted multiple times to assembly file");
Chris Lattner35096e82010-05-06 00:05:37 +0000753
Lang Hames9ff69c82015-04-24 19:11:51 +0000754 return OutStreamer->EmitLabel(CurrentFnSym);
Chris Lattnerbc1e6f02010-01-27 07:21:55 +0000755}
Chris Lattnerc7b91152010-01-26 23:18:44 +0000756
Chad Rosierb759ede2012-09-07 18:16:38 +0000757/// emitComments - Pretty-print comments for instructions.
Andrea Di Biagioedbf06a2019-02-04 12:51:26 +0000758static void emitComments(const MachineInstr &MI, raw_ostream &CommentOS) {
Justin Bognerfdf9bf42017-10-10 23:50:49 +0000759 const MachineFunction *MF = MI.getMF();
Eric Christopher97ea7622015-02-20 06:35:21 +0000760 const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
Jim Grosbach1900c732011-03-29 23:20:22 +0000761
Chris Lattner74e68522010-02-10 00:47:53 +0000762 // Check for spills and reloads
Wolfgang Piebfacd0522019-01-30 20:37:14 +0000763
Chris Lattner74e68522010-02-10 00:47:53 +0000764 // We assume a single instruction only has a spill or reload, not
765 // both.
Wolfgang Pieb90d856c2019-02-04 20:42:45 +0000766 Optional<unsigned> Size;
767 if ((Size = MI.getRestoreSize(TII))) {
768 CommentOS << *Size << "-byte Reload\n";
769 } else if ((Size = MI.getFoldedRestoreSize(TII))) {
770 if (*Size)
771 CommentOS << *Size << "-byte Folded Reload\n";
772 } else if ((Size = MI.getSpillSize(TII))) {
773 CommentOS << *Size << "-byte Spill\n";
774 } else if ((Size = MI.getFoldedSpillSize(TII))) {
775 if (*Size)
776 CommentOS << *Size << "-byte Folded Spill\n";
Chris Lattner74e68522010-02-10 00:47:53 +0000777 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000778
Chris Lattner74e68522010-02-10 00:47:53 +0000779 // Check for spill-induced copies
Andrea Di Biagioedbf06a2019-02-04 12:51:26 +0000780 if (MI.getAsmPrinterFlag(MachineInstr::ReloadReuse))
781 CommentOS << " Reload Reuse\n";
Chris Lattner74e68522010-02-10 00:47:53 +0000782}
783
Chad Rosierb759ede2012-09-07 18:16:38 +0000784/// emitImplicitDef - This method emits the specified machine instruction
Chris Lattner2b40a202010-04-04 18:58:53 +0000785/// that is an implicit def.
Justin Holewinski660597d2013-10-11 12:39:36 +0000786void AsmPrinter::emitImplicitDef(const MachineInstr *MI) const {
Daniel Sanders0c476112019-08-15 19:22:08 +0000787 Register RegNo = MI->getOperand(0).getReg();
Dan Gohman7aa4aba2015-11-17 16:01:28 +0000788
789 SmallString<128> Str;
790 raw_svector_ostream OS(Str);
791 OS << "implicit-def: "
Francis Visoiu Mistrih9d419d32017-11-28 12:42:37 +0000792 << printReg(RegNo, MF->getSubtarget().getRegisterInfo());
Dan Gohman7aa4aba2015-11-17 16:01:28 +0000793
794 OutStreamer->AddComment(OS.str());
Lang Hames9ff69c82015-04-24 19:11:51 +0000795 OutStreamer->AddBlankLine();
Chris Lattner2b40a202010-04-04 18:58:53 +0000796}
797
Chad Rosierb759ede2012-09-07 18:16:38 +0000798static void emitKill(const MachineInstr *MI, AsmPrinter &AP) {
Dan Gohman7aa4aba2015-11-17 16:01:28 +0000799 std::string Str;
800 raw_string_ostream OS(Str);
801 OS << "kill:";
Chris Lattner2b40a202010-04-04 18:58:53 +0000802 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
803 const MachineOperand &Op = MI->getOperand(i);
804 assert(Op.isReg() && "KILL instruction must have only register operands");
Francis Visoiu Mistriha8a83d12017-12-07 10:40:31 +0000805 OS << ' ' << (Op.isDef() ? "def " : "killed ")
806 << printReg(Op.getReg(), AP.MF->getSubtarget().getRegisterInfo());
Chris Lattner2b40a202010-04-04 18:58:53 +0000807 }
Matthias Braun07985802016-07-09 00:18:43 +0000808 AP.OutStreamer->AddComment(OS.str());
Lang Hames9ff69c82015-04-24 19:11:51 +0000809 AP.OutStreamer->AddBlankLine();
Chris Lattner2b40a202010-04-04 18:58:53 +0000810}
811
Chad Rosierb759ede2012-09-07 18:16:38 +0000812/// emitDebugValueComment - This method handles the target-independent form
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000813/// of DBG_VALUE, returning true if it was able to do so. A false return
814/// means the target will need to handle MI in EmitInstruction.
Chad Rosierb759ede2012-09-07 18:16:38 +0000815static bool emitDebugValueComment(const MachineInstr *MI, AsmPrinter &AP) {
Adrian Prantl87b7eb92014-10-01 18:55:02 +0000816 // This code handles only the 4-operand target-independent form.
817 if (MI->getNumOperands() != 4)
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000818 return false;
Chris Lattner74e68522010-02-10 00:47:53 +0000819
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000820 SmallString<128> Str;
821 raw_svector_ostream OS(Str);
Rafael Espindola0b694812014-01-16 16:28:37 +0000822 OS << "DEBUG_VALUE: ";
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000823
Duncan P. N. Exon Smitha9308c42015-04-29 16:38:44 +0000824 const DILocalVariable *V = MI->getDebugVariable();
825 if (auto *SP = dyn_cast<DISubprogram>(V->getScope())) {
Adrian Prantl9d2f0192017-04-26 23:59:52 +0000826 StringRef Name = SP->getName();
David Blaikie2dbebcf2013-06-15 00:33:47 +0000827 if (!Name.empty())
828 OS << Name << ":";
829 }
Duncan P. N. Exon Smith7348dda2015-04-14 02:22:36 +0000830 OS << V->getName();
Adrian Prantlb1416832014-08-01 22:11:58 +0000831 OS << " <- ";
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000832
Adrian Prantl418d1d12013-07-09 20:28:37 +0000833 // The second operand is only an offset if it's an immediate.
Adrian Prantl6825fb62017-04-18 01:21:53 +0000834 bool MemLoc = MI->getOperand(0).isReg() && MI->getOperand(1).isImm();
835 int64_t Offset = MemLoc ? MI->getOperand(1).getImm() : 0;
Adrian Prantlde1a8b42017-04-25 17:22:09 +0000836 const DIExpression *Expr = MI->getDebugExpression();
837 if (Expr->getNumElements()) {
838 OS << '[';
839 bool NeedSep = false;
840 for (auto Op : Expr->expr_ops()) {
841 if (NeedSep)
842 OS << ", ";
843 else
844 NeedSep = true;
845 OS << dwarf::OperationEncodingString(Op.getOp());
846 for (unsigned I = 0; I < Op.getNumArgs(); ++I)
847 OS << ' ' << Op.getArg(I);
Keno Fischer00cbf9a2015-12-19 02:02:44 +0000848 }
Adrian Prantlde1a8b42017-04-25 17:22:09 +0000849 OS << "] ";
Keno Fischer00cbf9a2015-12-19 02:02:44 +0000850 }
851
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000852 // Register or immediate value. Register 0 means undef.
853 if (MI->getOperand(0).isFPImm()) {
854 APFloat APF = APFloat(MI->getOperand(0).getFPImm()->getValueAPF());
855 if (MI->getOperand(0).getFPImm()->getType()->isFloatTy()) {
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000856 OS << (double)APF.convertToFloat();
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000857 } else if (MI->getOperand(0).getFPImm()->getType()->isDoubleTy()) {
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000858 OS << APF.convertToDouble();
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000859 } else {
860 // There is no good way to print long double. Convert a copy to
861 // double. Ah well, it's only a comment.
862 bool ignored;
Stephan Bergmann17c7f702016-12-14 11:57:17 +0000863 APF.convert(APFloat::IEEEdouble(), APFloat::rmNearestTiesToEven,
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000864 &ignored);
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000865 OS << "(long double) " << APF.convertToDouble();
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000866 }
867 } else if (MI->getOperand(0).isImm()) {
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000868 OS << MI->getOperand(0).getImm();
Devang Patelf071d722011-06-24 20:46:11 +0000869 } else if (MI->getOperand(0).isCImm()) {
870 MI->getOperand(0).getCImm()->getValue().print(OS, false /*isSigned*/);
Chris Lattnerb50e7952010-04-07 22:29:10 +0000871 } else {
David Blaikie0252265b2013-06-16 20:34:15 +0000872 unsigned Reg;
873 if (MI->getOperand(0).isReg()) {
874 Reg = MI->getOperand(0).getReg();
David Blaikie0252265b2013-06-16 20:34:15 +0000875 } else {
876 assert(MI->getOperand(0).isFI() && "Unknown operand type");
Eric Christophera7249ec2015-02-20 07:16:19 +0000877 const TargetFrameLowering *TFI = AP.MF->getSubtarget().getFrameLowering();
Eric Christopher5fdd68e2013-06-24 23:20:02 +0000878 Offset += TFI->getFrameIndexReference(*AP.MF,
879 MI->getOperand(0).getIndex(), Reg);
Adrian Prantlde1a8b42017-04-25 17:22:09 +0000880 MemLoc = true;
David Blaikie0252265b2013-06-16 20:34:15 +0000881 }
David Blaikiec9380db2013-06-15 15:52:58 +0000882 if (Reg == 0) {
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000883 // Suppress offset, it is not meaningful here.
Benjamin Kramer43c275f2010-04-07 09:26:51 +0000884 OS << "undef";
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000885 // NOTE: Want this comment at start of line, don't emit with AddComment.
Lang Hames9ff69c82015-04-24 19:11:51 +0000886 AP.OutStreamer->emitRawComment(OS.str());
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000887 return true;
888 }
Adrian Prantlde1a8b42017-04-25 17:22:09 +0000889 if (MemLoc)
David Blaikiec9380db2013-06-15 15:52:58 +0000890 OS << '[';
Francis Visoiu Mistrih9d419d32017-11-28 12:42:37 +0000891 OS << printReg(Reg, AP.MF->getSubtarget().getRegisterInfo());
Chris Lattnerb50e7952010-04-07 22:29:10 +0000892 }
Jim Grosbach1900c732011-03-29 23:20:22 +0000893
Adrian Prantlde1a8b42017-04-25 17:22:09 +0000894 if (MemLoc)
Adrian Prantl418d1d12013-07-09 20:28:37 +0000895 OS << '+' << Offset << ']';
896
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000897 // NOTE: Want this comment at start of line, don't emit with AddComment.
Lang Hames9ff69c82015-04-24 19:11:51 +0000898 AP.OutStreamer->emitRawComment(OS.str());
Dale Johannesen5d7f0a02010-04-07 01:15:14 +0000899 return true;
900}
Chris Lattner74e68522010-02-10 00:47:53 +0000901
Shiva Chencd070cd2018-05-09 02:41:08 +0000902/// This method handles the target-independent form of DBG_LABEL, returning
903/// true if it was able to do so. A false return means the target will need
904/// to handle MI in EmitInstruction.
905static bool emitDebugLabelComment(const MachineInstr *MI, AsmPrinter &AP) {
906 if (MI->getNumOperands() != 1)
907 return false;
908
909 SmallString<128> Str;
910 raw_svector_ostream OS(Str);
911 OS << "DEBUG_LABEL: ";
912
913 const DILabel *V = MI->getDebugLabel();
Taewook Ohdf702282019-08-14 17:58:45 +0000914 if (auto *SP = dyn_cast<DISubprogram>(
915 V->getScope()->getNonLexicalBlockFileScope())) {
Shiva Chencd070cd2018-05-09 02:41:08 +0000916 StringRef Name = SP->getName();
917 if (!Name.empty())
918 OS << Name << ":";
919 }
920 OS << V->getName();
921
922 // NOTE: Want this comment at start of line, don't emit with AddComment.
923 AP.OutStreamer->emitRawComment(OS.str());
924 return true;
925}
926
Francis Visoiu Mistrih9b97a312017-07-19 23:47:33 +0000927AsmPrinter::CFIMoveType AsmPrinter::needsCFIMoves() const {
Rafael Espindolafc9bae62011-05-25 03:44:17 +0000928 if (MAI->getExceptionHandlingType() == ExceptionHandling::DwarfCFI &&
Matthias Braunf1caa282017-12-15 22:22:58 +0000929 MF->getFunction().needsUnwindTableEntry())
Rafael Espindolafc9bae62011-05-25 03:44:17 +0000930 return CFI_M_EH;
Rafael Espindolab7a012a2011-04-29 14:14:06 +0000931
Rafael Espindolaa7160962011-05-06 14:56:22 +0000932 if (MMI->hasDebugInfo())
Rafael Espindolafdc3e6f2011-05-10 18:39:09 +0000933 return CFI_M_Debug;
Rafael Espindolab7a012a2011-04-29 14:14:06 +0000934
Rafael Espindolafdc3e6f2011-05-10 18:39:09 +0000935 return CFI_M_None;
Rafael Espindolab7a012a2011-04-29 14:14:06 +0000936}
937
Charles Davis5638b9f2011-05-28 04:21:04 +0000938bool AsmPrinter::needsSEHMoves() {
Matthias Braunf1caa282017-12-15 22:22:58 +0000939 return MAI->usesWindowsCFI() && MF->getFunction().needsUnwindTableEntry();
Charles Davis5638b9f2011-05-28 04:21:04 +0000940}
941
Rafael Espindolab1f25f12014-03-07 06:08:31 +0000942void AsmPrinter::emitCFIInstruction(const MachineInstr &MI) {
Saleem Abdulrasool67b54812014-06-29 21:43:47 +0000943 ExceptionHandling ExceptionHandlingType = MAI->getExceptionHandlingType();
Artyom Skrobovf6830f42014-02-14 17:19:07 +0000944 if (ExceptionHandlingType != ExceptionHandling::DwarfCFI &&
945 ExceptionHandlingType != ExceptionHandling::ARM)
Rafael Espindolaa01cdb02011-04-15 15:11:06 +0000946 return;
Rafael Espindolaa01cdb02011-04-15 15:11:06 +0000947
Rafael Espindolafdc3e6f2011-05-10 18:39:09 +0000948 if (needsCFIMoves() == CFI_M_None)
Rafael Espindolab7a012a2011-04-29 14:14:06 +0000949 return;
950
Adrian Prantl083e6a52017-04-24 18:45:59 +0000951 // If there is no "real" instruction following this CFI instruction, skip
952 // emitting it; it would be beyond the end of the function's FDE range.
953 auto *MBB = MI.getParent();
954 auto I = std::next(MI.getIterator());
955 while (I != MBB->end() && I->isTransient())
956 ++I;
957 if (I == MBB->instr_end() &&
958 MBB->getReverseIterator() == MBB->getParent()->rbegin())
959 return;
960
Matthias Braunf23ef432016-11-30 23:48:42 +0000961 const std::vector<MCCFIInstruction> &Instrs = MF->getFrameInstructions();
Rafael Espindolab1f25f12014-03-07 06:08:31 +0000962 unsigned CFIIndex = MI.getOperand(0).getCFIIndex();
963 const MCCFIInstruction &CFI = Instrs[CFIIndex];
964 emitCFIInstruction(CFI);
Rafael Espindolaa01cdb02011-04-15 15:11:06 +0000965}
966
Reid Klecknere9b89312015-01-13 00:48:10 +0000967void AsmPrinter::emitFrameAlloc(const MachineInstr &MI) {
968 // The operands are the MCSymbol and the frame offset of the allocation.
969 MCSymbol *FrameAllocSym = MI.getOperand(0).getMCSymbol();
970 int FrameOffset = MI.getOperand(1).getImm();
971
972 // Emit a symbol assignment.
Lang Hames9ff69c82015-04-24 19:11:51 +0000973 OutStreamer->EmitAssignment(FrameAllocSym,
Jim Grosbach13760bd2015-05-30 01:25:56 +0000974 MCConstantExpr::create(FrameOffset, OutContext));
Reid Klecknere9b89312015-01-13 00:48:10 +0000975}
976
Sean Evesona6bcd532017-11-30 13:05:14 +0000977void AsmPrinter::emitStackSizeSection(const MachineFunction &MF) {
978 if (!MF.getTarget().Options.EmitStackSizeSection)
979 return;
980
George Rimardcf59c52018-06-22 10:53:47 +0000981 MCSection *StackSizeSection =
982 getObjFileLowering().getStackSizesSection(*getCurrentSection());
Sean Evesona6bcd532017-11-30 13:05:14 +0000983 if (!StackSizeSection)
984 return;
985
986 const MachineFrameInfo &FrameInfo = MF.getFrameInfo();
987 // Don't emit functions with dynamic stack allocations.
988 if (FrameInfo.hasVarSizedObjects())
989 return;
990
991 OutStreamer->PushSection();
992 OutStreamer->SwitchSection(StackSizeSection);
993
Rafael Espindola78fdca32018-03-26 20:40:22 +0000994 const MCSymbol *FunctionSymbol = getFunctionBegin();
Sean Evesona6bcd532017-11-30 13:05:14 +0000995 uint64_t StackSize = FrameInfo.getStackSize();
Matt Arsenault41e5ac42018-03-14 00:36:23 +0000996 OutStreamer->EmitSymbolValue(FunctionSymbol, TM.getProgramPointerSize());
Sean Evesona6bcd532017-11-30 13:05:14 +0000997 OutStreamer->EmitULEB128IntValue(StackSize);
998
999 OutStreamer->PopSection();
1000}
1001
Reid Klecknerc2f1bbf2017-05-31 22:18:49 +00001002static bool needFuncLabelsForEHOrDebugInfo(const MachineFunction &MF,
1003 MachineModuleInfo *MMI) {
1004 if (!MF.getLandingPads().empty() || MF.hasEHFunclets() || MMI->hasDebugInfo())
1005 return true;
1006
Reid Kleckner57ac61e2017-05-31 22:21:20 +00001007 // We might emit an EH table that uses function begin and end labels even if
1008 // we don't have any landingpads.
Matthias Braunf1caa282017-12-15 22:22:58 +00001009 if (!MF.getFunction().hasPersonalityFn())
Reid Klecknerc2f1bbf2017-05-31 22:18:49 +00001010 return false;
Reid Kleckner57ac61e2017-05-31 22:21:20 +00001011 return !isNoOpWithoutInvoke(
Matthias Braunf1caa282017-12-15 22:22:58 +00001012 classifyEHPersonality(MF.getFunction().getPersonalityFn()));
Reid Klecknerc2f1bbf2017-05-31 22:18:49 +00001013}
1014
Chris Lattner94a946c2010-01-28 01:02:27 +00001015/// EmitFunctionBody - This method emits the body and trailer for a
1016/// function.
1017void AsmPrinter::EmitFunctionBody() {
Rafael Espindola8dc4e102015-03-17 14:38:30 +00001018 EmitFunctionHeader();
1019
Chris Lattnerc968f442010-01-28 01:58:58 +00001020 // Emit target-specific gunk before the function body.
1021 EmitFunctionBodyStart();
Jim Grosbach1900c732011-03-29 23:20:22 +00001022
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001023 bool ShouldPrintDebugScopes = MMI->hasDebugInfo();
Jim Grosbach1900c732011-03-29 23:20:22 +00001024
Michael Zolotukhin8d052a02018-04-09 00:54:47 +00001025 if (isVerbose()) {
1026 // Get MachineDominatorTree or compute it on the fly if it's unavailable
1027 MDT = getAnalysisIfAvailable<MachineDominatorTree>();
1028 if (!MDT) {
Jonas Devlieghere0eaee542019-08-15 15:54:37 +00001029 OwnedMDT = std::make_unique<MachineDominatorTree>();
Michael Zolotukhin8d052a02018-04-09 00:54:47 +00001030 OwnedMDT->getBase().recalculate(*MF);
1031 MDT = OwnedMDT.get();
1032 }
1033
1034 // Get MachineLoopInfo or compute it on the fly if it's unavailable
1035 MLI = getAnalysisIfAvailable<MachineLoopInfo>();
1036 if (!MLI) {
Jonas Devlieghere0eaee542019-08-15 15:54:37 +00001037 OwnedMLI = std::make_unique<MachineLoopInfo>();
Michael Zolotukhin8d052a02018-04-09 00:54:47 +00001038 OwnedMLI->getBase().analyze(MDT->getBase());
1039 MLI = OwnedMLI.get();
1040 }
1041 }
1042
Chris Lattner94a946c2010-01-28 01:02:27 +00001043 // Print out code for the function.
1044 bool HasAnyRealCode = false;
Justin Bogner369ba752017-02-24 00:19:22 +00001045 int NumInstsInFunction = 0;
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001046 for (auto &MBB : *MF) {
Chris Lattner94a946c2010-01-28 01:02:27 +00001047 // Print a label for the basic block.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001048 EmitBasicBlockStart(MBB);
1049 for (auto &MI : MBB) {
Chris Lattner94a946c2010-01-28 01:02:27 +00001050 // Print the assembly for the instruction.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001051 if (!MI.isPosition() && !MI.isImplicitDef() && !MI.isKill() &&
Shiva Chen801bf7e2018-05-09 02:42:00 +00001052 !MI.isDebugInstr()) {
Chris Lattner94a946c2010-01-28 01:02:27 +00001053 HasAnyRealCode = true;
Justin Bogner369ba752017-02-24 00:19:22 +00001054 ++NumInstsInFunction;
Evan Chengeb828b62010-04-27 19:38:45 +00001055 }
1056
Chandler Carruth75ca6be2018-08-16 23:11:05 +00001057 // If there is a pre-instruction symbol, emit a label for it here.
1058 if (MCSymbol *S = MI.getPreInstrSymbol())
1059 OutStreamer->EmitLabel(S);
1060
Torok Edwinf8dba242010-04-07 10:44:46 +00001061 if (ShouldPrintDebugScopes) {
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001062 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +00001063 NamedRegionTimer T(HI.TimerName, HI.TimerDescription,
1064 HI.TimerGroupName, HI.TimerGroupDescription,
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001065 TimePassesIsEnabled);
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001066 HI.Handler->beginInstruction(&MI);
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001067 }
Torok Edwinf8dba242010-04-07 10:44:46 +00001068 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001069
Andrea Di Biagioedbf06a2019-02-04 12:51:26 +00001070 if (isVerbose())
1071 emitComments(MI, OutStreamer->GetCommentOS());
Chris Lattner74e68522010-02-10 00:47:53 +00001072
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001073 switch (MI.getOpcode()) {
Rafael Espindolab1f25f12014-03-07 06:08:31 +00001074 case TargetOpcode::CFI_INSTRUCTION:
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001075 emitCFIInstruction(MI);
Rafael Espindolaa01cdb02011-04-15 15:11:06 +00001076 break;
Reid Kleckner60381792015-07-07 22:25:32 +00001077 case TargetOpcode::LOCAL_ESCAPE:
Reid Klecknere9b89312015-01-13 00:48:10 +00001078 emitFrameAlloc(MI);
1079 break;
Reid Kleckner48824902019-05-15 21:46:05 +00001080 case TargetOpcode::ANNOTATION_LABEL:
Chris Lattnerb06015a2010-02-09 19:54:29 +00001081 case TargetOpcode::EH_LABEL:
1082 case TargetOpcode::GC_LABEL:
Lang Hames9ff69c82015-04-24 19:11:51 +00001083 OutStreamer->EmitLabel(MI.getOperand(0).getMCSymbol());
Chris Lattner41ad1902010-02-03 01:00:52 +00001084 break;
Chris Lattnerb06015a2010-02-09 19:54:29 +00001085 case TargetOpcode::INLINEASM:
Craig Topper784929d2019-02-08 20:48:56 +00001086 case TargetOpcode::INLINEASM_BR:
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001087 EmitInlineAsm(&MI);
Chris Lattner41ad1902010-02-03 01:00:52 +00001088 break;
Dale Johannesen5d7f0a02010-04-07 01:15:14 +00001089 case TargetOpcode::DBG_VALUE:
1090 if (isVerbose()) {
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001091 if (!emitDebugValueComment(&MI, *this))
1092 EmitInstruction(&MI);
Dale Johannesen5d7f0a02010-04-07 01:15:14 +00001093 }
1094 break;
Shiva Chencd070cd2018-05-09 02:41:08 +00001095 case TargetOpcode::DBG_LABEL:
1096 if (isVerbose()) {
1097 if (!emitDebugLabelComment(&MI, *this))
1098 EmitInstruction(&MI);
1099 }
1100 break;
Chris Lattnerb06015a2010-02-09 19:54:29 +00001101 case TargetOpcode::IMPLICIT_DEF:
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001102 if (isVerbose()) emitImplicitDef(&MI);
Chris Lattner41ad1902010-02-03 01:00:52 +00001103 break;
Chris Lattnerb06015a2010-02-09 19:54:29 +00001104 case TargetOpcode::KILL:
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001105 if (isVerbose()) emitKill(&MI, *this);
Chris Lattner41ad1902010-02-03 01:00:52 +00001106 break;
1107 default:
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001108 EmitInstruction(&MI);
Chris Lattner41ad1902010-02-03 01:00:52 +00001109 break;
1110 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001111
Chandler Carruth75ca6be2018-08-16 23:11:05 +00001112 // If there is a post-instruction symbol, emit a label for it here.
1113 if (MCSymbol *S = MI.getPostInstrSymbol())
1114 OutStreamer->EmitLabel(S);
1115
Torok Edwinf8dba242010-04-07 10:44:46 +00001116 if (ShouldPrintDebugScopes) {
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001117 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +00001118 NamedRegionTimer T(HI.TimerName, HI.TimerDescription,
1119 HI.TimerGroupName, HI.TimerGroupDescription,
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001120 TimePassesIsEnabled);
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001121 HI.Handler->endInstruction();
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001122 }
Torok Edwinf8dba242010-04-07 10:44:46 +00001123 }
Chris Lattner94a946c2010-01-28 01:02:27 +00001124 }
Lang Hamesf49bc3f2014-07-24 20:40:55 +00001125
1126 EmitBasicBlockEnd(MBB);
Chris Lattner94a946c2010-01-28 01:02:27 +00001127 }
Bill Wendlingbf8370f2010-07-16 22:51:10 +00001128
Justin Bogner369ba752017-02-24 00:19:22 +00001129 EmittedInsts += NumInstsInFunction;
1130 MachineOptimizationRemarkAnalysis R(DEBUG_TYPE, "InstructionCount",
Matthias Braunf1caa282017-12-15 22:22:58 +00001131 MF->getFunction().getSubprogram(),
Justin Bogner369ba752017-02-24 00:19:22 +00001132 &MF->front());
1133 R << ore::NV("NumInstructions", NumInstsInFunction)
1134 << " instructions in function";
1135 ORE->emit(R);
1136
Chris Lattner94a946c2010-01-28 01:02:27 +00001137 // If the function is empty and the object file uses .subsections_via_symbols,
Chris Lattner73de5fb2010-01-28 01:28:58 +00001138 // then we need to emit *something* to the function body to prevent the
Chris Lattner6a5e7062010-04-26 23:37:21 +00001139 // labels from collapsing together. Just emit a noop.
Hans Wennborg9b9a5352017-04-21 21:48:41 +00001140 // Similarly, don't emit empty functions on Windows either. It can lead to
1141 // duplicate entries (two functions with the same RVA) in the Guard CF Table
1142 // after linking, causing the kernel not to load the binary:
1143 // https://developercommunity.visualstudio.com/content/problem/45366/vc-linker-creates-invalid-dll-with-clang-cl.html
1144 // FIXME: Hide this behind some API in e.g. MCAsmInfo or MCTargetStreamer.
1145 const Triple &TT = TM.getTargetTriple();
1146 if (!HasAnyRealCode && (MAI->hasSubsectionsViaSymbols() ||
1147 (TT.isOSWindows() && TT.isOSBinFormatCOFF()))) {
Chris Lattner6a5e7062010-04-26 23:37:21 +00001148 MCInst Noop;
Hans Wennborg9b9a5352017-04-21 21:48:41 +00001149 MF->getSubtarget().getInstrInfo()->getNoop(Noop);
Daniel Sanders11e78c22017-04-25 14:27:27 +00001150
1151 // Targets can opt-out of emitting the noop here by leaving the opcode
1152 // unspecified.
1153 if (Noop.getOpcode()) {
1154 OutStreamer->AddComment("avoids zero-length function");
1155 OutStreamer->EmitInstruction(Noop, getSubtargetInfo());
1156 }
Chris Lattner6a5e7062010-04-26 23:37:21 +00001157 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001158
Matthias Braunf1caa282017-12-15 22:22:58 +00001159 const Function &F = MF->getFunction();
1160 for (const auto &BB : F) {
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001161 if (!BB.hasAddressTaken())
Rafael Espindolaf11e7f12011-11-15 19:08:46 +00001162 continue;
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001163 MCSymbol *Sym = GetBlockAddressSymbol(&BB);
Rafael Espindolaf11e7f12011-11-15 19:08:46 +00001164 if (Sym->isDefined())
1165 continue;
Lang Hames9ff69c82015-04-24 19:11:51 +00001166 OutStreamer->AddComment("Address of block that was removed by CodeGen");
1167 OutStreamer->EmitLabel(Sym);
Rafael Espindolaf11e7f12011-11-15 19:08:46 +00001168 }
1169
Chris Lattnerc968f442010-01-28 01:58:58 +00001170 // Emit target-specific gunk after the function body.
1171 EmitFunctionBodyEnd();
Jim Grosbach1900c732011-03-29 23:20:22 +00001172
Reid Klecknerc2f1bbf2017-05-31 22:18:49 +00001173 if (needFuncLabelsForEHOrDebugInfo(*MF, MMI) ||
1174 MAI->hasDotTypeDotSizeDirective()) {
Rafael Espindola629cdba2015-02-27 18:18:39 +00001175 // Create a symbol for the end of function.
Rafael Espindola9ab09232015-03-17 20:07:06 +00001176 CurrentFnEnd = createTempSymbol("func_end");
Lang Hames9ff69c82015-04-24 19:11:51 +00001177 OutStreamer->EmitLabel(CurrentFnEnd);
Rafael Espindola629cdba2015-02-27 18:18:39 +00001178 }
1179
Chris Lattnerc6d67d42010-04-03 22:19:41 +00001180 // If the target wants a .size directive for the size of the function, emit
1181 // it.
1182 if (MAI->hasDotTypeDotSizeDirective()) {
Rafael Espindola310e4b52015-03-04 01:35:23 +00001183 // We can get the size as difference between the function label and the
1184 // temp label.
NAKAMURA Takumibf6ad022015-06-14 00:23:33 +00001185 const MCExpr *SizeExp = MCBinaryExpr::createSub(
1186 MCSymbolRefExpr::create(CurrentFnEnd, OutContext),
1187 MCSymbolRefExpr::create(CurrentFnSymForSize, OutContext), OutContext);
Dan Gohman734c59d2016-12-01 23:39:08 +00001188 OutStreamer->emitELFSize(CurrentFnSym, SizeExp);
Chris Lattnerc6d67d42010-04-03 22:19:41 +00001189 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001190
Rafael Espindolaa6001792015-03-09 18:29:12 +00001191 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +00001192 NamedRegionTimer T(HI.TimerName, HI.TimerDescription, HI.TimerGroupName,
1193 HI.TimerGroupDescription, TimePassesIsEnabled);
Rafael Espindolaa6001792015-03-09 18:29:12 +00001194 HI.Handler->markFunctionEnd();
1195 }
1196
1197 // Print out jump tables referenced by the function.
1198 EmitJumpTableInfo();
1199
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001200 // Emit post-function debug and/or EH information.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001201 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +00001202 NamedRegionTimer T(HI.TimerName, HI.TimerDescription, HI.TimerGroupName,
1203 HI.TimerGroupDescription, TimePassesIsEnabled);
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001204 HI.Handler->endFunction(MF);
Torok Edwinf8dba242010-04-07 10:44:46 +00001205 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001206
Sean Evesona6bcd532017-11-30 13:05:14 +00001207 // Emit section containing stack size metadata.
1208 emitStackSizeSection(*MF);
1209
Francis Visoiu Mistrih1c987012017-05-23 21:22:16 +00001210 if (isVerbose())
1211 OutStreamer->GetCommentOS() << "-- End function\n";
1212
Lang Hames9ff69c82015-04-24 19:11:51 +00001213 OutStreamer->AddBlankLine();
Chris Lattner94a946c2010-01-28 01:02:27 +00001214}
1215
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00001216/// Compute the number of Global Variables that uses a Constant.
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001217static unsigned getNumGlobalVariableUses(const Constant *C) {
1218 if (!C)
1219 return 0;
1220
1221 if (isa<GlobalVariable>(C))
1222 return 1;
1223
1224 unsigned NumUses = 0;
1225 for (auto *CU : C->users())
1226 NumUses += getNumGlobalVariableUses(dyn_cast<Constant>(CU));
1227
1228 return NumUses;
1229}
1230
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00001231/// Only consider global GOT equivalents if at least one user is a
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001232/// cstexpr inside an initializer of another global variables. Also, don't
1233/// handle cstexpr inside instructions. During global variable emission,
1234/// candidates are skipped and are emitted later in case at least one cstexpr
1235/// isn't replaced by a PC relative GOT entry access.
1236static bool isGOTEquivalentCandidate(const GlobalVariable *GV,
1237 unsigned &NumGOTEquivUsers) {
1238 // Global GOT equivalents are unnamed private globals with a constant
1239 // pointer initializer to another global symbol. They must point to a
1240 // GlobalVariable or Function, i.e., as GlobalValue.
Eric Christopher0b6537e2016-07-01 06:07:31 +00001241 if (!GV->hasGlobalUnnamedAddr() || !GV->hasInitializer() ||
1242 !GV->isConstant() || !GV->isDiscardableIfUnused() ||
Fangrui Song2c5c12c2019-04-05 16:16:23 +00001243 !isa<GlobalValue>(GV->getOperand(0)))
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001244 return false;
1245
1246 // To be a got equivalent, at least one of its users need to be a constant
1247 // expression used by another global variable.
1248 for (auto *U : GV->users())
Ahmed Bougacha821880a2015-03-27 01:40:54 +00001249 NumGOTEquivUsers += getNumGlobalVariableUses(dyn_cast<Constant>(U));
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001250
1251 return NumGOTEquivUsers > 0;
1252}
1253
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00001254/// Unnamed constant global variables solely contaning a pointer to
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001255/// another globals variable is equivalent to a GOT table entry; it contains the
1256/// the address of another symbol. Optimize it and replace accesses to these
1257/// "GOT equivalents" by using the GOT entry for the final global instead.
1258/// Compute GOT equivalent candidates among all global variables to avoid
1259/// emitting them if possible later on, after it use is replaced by a GOT entry
1260/// access.
1261void AsmPrinter::computeGlobalGOTEquivs(Module &M) {
1262 if (!getObjFileLowering().supportIndirectSymViaGOTPCRel())
1263 return;
1264
1265 for (const auto &G : M.globals()) {
1266 unsigned NumGOTEquivUsers = 0;
1267 if (!isGOTEquivalentCandidate(&G, NumGOTEquivUsers))
1268 continue;
1269
1270 const MCSymbol *GOTEquivSym = getSymbol(&G);
1271 GlobalGOTEquivs[GOTEquivSym] = std::make_pair(&G, NumGOTEquivUsers);
1272 }
1273}
1274
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00001275/// Constant expressions using GOT equivalent globals may not be eligible
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001276/// for PC relative GOT entry conversion, in such cases we need to emit such
1277/// globals we previously omitted in EmitGlobalVariable.
1278void AsmPrinter::emitGlobalGOTEquivs() {
1279 if (!getObjFileLowering().supportIndirectSymViaGOTPCRel())
1280 return;
1281
Bruno Cardoso Lopesb3a58b42015-03-10 20:05:23 +00001282 SmallVector<const GlobalVariable *, 8> FailedCandidates;
1283 for (auto &I : GlobalGOTEquivs) {
1284 const GlobalVariable *GV = I.second.first;
1285 unsigned Cnt = I.second.second;
1286 if (Cnt)
1287 FailedCandidates.push_back(GV);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001288 }
Bruno Cardoso Lopesb3a58b42015-03-10 20:05:23 +00001289 GlobalGOTEquivs.clear();
1290
1291 for (auto *GV : FailedCandidates)
1292 EmitGlobalVariable(GV);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001293}
1294
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00001295void AsmPrinter::emitGlobalIndirectSymbol(Module &M,
1296 const GlobalIndirectSymbol& GIS) {
1297 MCSymbol *Name = getSymbol(&GIS);
1298
1299 if (GIS.hasExternalLinkage() || !MAI->getWeakRefDirective())
1300 OutStreamer->EmitSymbolAttribute(Name, MCSA_Global);
1301 else if (GIS.hasWeakLinkage() || GIS.hasLinkOnceLinkage())
1302 OutStreamer->EmitSymbolAttribute(Name, MCSA_WeakReference);
1303 else
Dmitry Polukhina1feff72016-04-07 12:32:19 +00001304 assert(GIS.hasLocalLinkage() && "Invalid alias or ifunc linkage");
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00001305
Tim Northover67828ed2019-07-11 13:13:02 +00001306 bool IsFunction = GIS.getValueType()->isFunctionTy();
Sam Clegg313f9f52019-02-04 23:07:34 +00001307
1308 // Treat bitcasts of functions as functions also. This is important at least
1309 // on WebAssembly where object and function addresses can't alias each other.
1310 if (!IsFunction)
1311 if (auto *CE = dyn_cast<ConstantExpr>(GIS.getIndirectSymbol()))
1312 if (CE->getOpcode() == Instruction::BitCast)
1313 IsFunction =
1314 CE->getOperand(0)->getType()->getPointerElementType()->isFunctionTy();
1315
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00001316 // Set the symbol type to function if the alias has a function type.
1317 // This affects codegen when the aliasee is not a function.
Fangrui Song8caa0aa2019-08-14 10:30:27 +00001318 if (IsFunction)
1319 OutStreamer->EmitSymbolAttribute(Name, isa<GlobalIFunc>(GIS)
1320 ? MCSA_ELF_TypeIndFunction
1321 : MCSA_ELF_TypeFunction);
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00001322
1323 EmitVisibility(Name, GIS.getVisibility());
1324
1325 const MCExpr *Expr = lowerConstant(GIS.getIndirectSymbol());
1326
1327 if (isa<GlobalAlias>(&GIS) && MAI->hasAltEntry() && isa<MCBinaryExpr>(Expr))
1328 OutStreamer->EmitSymbolAttribute(Name, MCSA_AltEntry);
1329
1330 // Emit the directives as assignments aka .set:
1331 OutStreamer->EmitAssignment(Name, Expr);
1332
1333 if (auto *GA = dyn_cast<GlobalAlias>(&GIS)) {
1334 // If the aliasee does not correspond to a symbol in the output, i.e. the
1335 // alias is not of an object or the aliased object is private, then set the
1336 // size of the alias symbol from the type of the alias. We don't do this in
1337 // other situations as the alias and aliasee having differing types but same
1338 // size may be intentional.
1339 const GlobalObject *BaseObject = GA->getBaseObject();
1340 if (MAI->hasDotTypeDotSizeDirective() && GA->getValueType()->isSized() &&
1341 (!BaseObject || BaseObject->hasPrivateLinkage())) {
1342 const DataLayout &DL = M.getDataLayout();
1343 uint64_t Size = DL.getTypeAllocSize(GA->getValueType());
Dan Gohman734c59d2016-12-01 23:39:08 +00001344 OutStreamer->emitELFSize(Name, MCConstantExpr::create(Size, OutContext));
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00001345 }
1346 }
1347}
1348
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +00001349void AsmPrinter::emitRemarksSection(Module &M) {
1350 RemarkStreamer *RS = M.getContext().getRemarkStreamer();
1351 if (!RS)
1352 return;
Francis Visoiu Mistrih2d8fdca2019-07-26 01:33:30 +00001353 remarks::RemarkSerializer &RemarkSerializer = RS->getSerializer();
1354
1355 StringRef FilenameRef = RS->getFilename();
1356 SmallString<128> Filename = FilenameRef;
1357 sys::fs::make_absolute(Filename);
1358 assert(!Filename.empty() && "The filename can't be empty.");
1359
1360 std::string Buf;
1361 raw_string_ostream OS(Buf);
1362 std::unique_ptr<remarks::MetaSerializer> MetaSerializer =
1363 RemarkSerializer.metaSerializer(OS, StringRef(Filename));
1364 MetaSerializer->emit();
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +00001365
1366 // Switch to the right section: .remarks/__remarks.
1367 MCSection *RemarksSection =
1368 OutContext.getObjectFileInfo()->getRemarksSection();
1369 OutStreamer->SwitchSection(RemarksSection);
1370
Francis Visoiu Mistrih2d8fdca2019-07-26 01:33:30 +00001371 OutStreamer->EmitBinaryData(OS.str());
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +00001372}
1373
Chris Lattner6a8e0f52004-08-16 23:15:22 +00001374bool AsmPrinter::doFinalization(Module &M) {
Eric Christopherd43c5c72015-03-20 16:03:39 +00001375 // Set the MachineFunction to nullptr so that we can catch attempted
1376 // accesses to MF specific features at the module level and so that
1377 // we can conditionalize accesses based on whether or not it is nullptr.
1378 MF = nullptr;
1379
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001380 // Gather all GOT equivalent globals in the module. We really need two
1381 // passes over the globals: one to compute and another to avoid its emission
1382 // in EmitGlobalVariable, otherwise we would not be able to handle cases
1383 // where the got equivalent shows up before its use.
1384 computeGlobalGOTEquivs(M);
1385
Chris Lattner100865e2009-07-21 18:38:57 +00001386 // Emit global variables.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001387 for (const auto &G : M.globals())
1388 EmitGlobalVariable(&G);
Rafael Espindola6c17d542011-01-28 03:20:10 +00001389
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00001390 // Emit remaining GOT equivalent globals.
1391 emitGlobalGOTEquivs();
1392
Rafael Espindola6c17d542011-01-28 03:20:10 +00001393 // Emit visibility info for declarations
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001394 for (const Function &F : M) {
Rafael Espindola7068cbb2015-07-13 13:55:18 +00001395 if (!F.isDeclarationForLinker())
Rafael Espindola6c17d542011-01-28 03:20:10 +00001396 continue;
1397 GlobalValue::VisibilityTypes V = F.getVisibility();
1398 if (V == GlobalValue::DefaultVisibility)
1399 continue;
1400
Rafael Espindola79858aa2013-10-29 17:07:16 +00001401 MCSymbol *Name = getSymbol(&F);
Stuart Hastingsbf836592011-02-23 02:27:05 +00001402 EmitVisibility(Name, V, false);
Rafael Espindola6c17d542011-01-28 03:20:10 +00001403 }
1404
Francis Visoiu Mistrihee1a6e72019-03-27 01:13:59 +00001405 // Emit the remarks section contents.
1406 // FIXME: Figure out when is the safest time to emit this section. It should
1407 // not come after debug info.
1408 if (EnableRemarksSection)
1409 emitRemarksSection(M);
1410
Rafael Espindolab91455b2015-04-07 13:42:44 +00001411 const TargetLoweringObjectFile &TLOF = getObjFileLowering();
1412
Eric Christopheraadbabc2018-04-20 19:07:57 +00001413 TLOF.emitModuleMetadata(*OutStreamer, M);
Rafael Espindolab91455b2015-04-07 13:42:44 +00001414
Daniel Sanders335487a2015-06-16 13:15:50 +00001415 if (TM.getTargetTriple().isOSBinFormatELF()) {
Rafael Espindolab91455b2015-04-07 13:42:44 +00001416 MachineModuleInfoELF &MMIELF = MMI->getObjFileInfo<MachineModuleInfoELF>();
1417
1418 // Output stubs for external and common global variables.
1419 MachineModuleInfoELF::SymbolListTy Stubs = MMIELF.GetGVStubList();
1420 if (!Stubs.empty()) {
Rafael Espindola449711c2015-11-18 06:02:15 +00001421 OutStreamer->SwitchSection(TLOF.getDataSection());
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001422 const DataLayout &DL = M.getDataLayout();
Rafael Espindolab91455b2015-04-07 13:42:44 +00001423
Rafael Espindola8c587502018-04-02 23:20:30 +00001424 EmitAlignment(Log2_32(DL.getPointerSize()));
Rafael Espindolab91455b2015-04-07 13:42:44 +00001425 for (const auto &Stub : Stubs) {
Lang Hames9ff69c82015-04-24 19:11:51 +00001426 OutStreamer->EmitLabel(Stub.first);
1427 OutStreamer->EmitSymbolValue(Stub.second.getPointer(),
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001428 DL.getPointerSize());
Rafael Espindolab91455b2015-04-07 13:42:44 +00001429 }
1430 }
1431 }
Bill Wendling06df7722012-02-14 21:28:13 +00001432
Martin Storsjo489993d2018-08-29 17:28:34 +00001433 if (TM.getTargetTriple().isOSBinFormatCOFF()) {
1434 MachineModuleInfoCOFF &MMICOFF =
1435 MMI->getObjFileInfo<MachineModuleInfoCOFF>();
1436
1437 // Output stubs for external and common global variables.
1438 MachineModuleInfoCOFF::SymbolListTy Stubs = MMICOFF.GetGVStubList();
1439 if (!Stubs.empty()) {
1440 const DataLayout &DL = M.getDataLayout();
1441
1442 for (const auto &Stub : Stubs) {
1443 SmallString<256> SectionName = StringRef(".rdata$");
1444 SectionName += Stub.first->getName();
1445 OutStreamer->SwitchSection(OutContext.getCOFFSection(
1446 SectionName,
1447 COFF::IMAGE_SCN_CNT_INITIALIZED_DATA | COFF::IMAGE_SCN_MEM_READ |
1448 COFF::IMAGE_SCN_LNK_COMDAT,
1449 SectionKind::getReadOnly(), Stub.first->getName(),
1450 COFF::IMAGE_COMDAT_SELECT_ANY));
1451 EmitAlignment(Log2_32(DL.getPointerSize()));
1452 OutStreamer->EmitSymbolAttribute(Stub.first, MCSA_Global);
1453 OutStreamer->EmitLabel(Stub.first);
1454 OutStreamer->EmitSymbolValue(Stub.second.getPointer(),
1455 DL.getPointerSize());
1456 }
1457 }
1458 }
1459
Chris Lattnerf0d6bd32010-04-05 05:11:15 +00001460 // Finalize debug and EH information.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001461 for (const HandlerInfo &HI : Handlers) {
Matthias Braun9f15a792016-11-18 19:43:18 +00001462 NamedRegionTimer T(HI.TimerName, HI.TimerDescription, HI.TimerGroupName,
1463 HI.TimerGroupDescription, TimePassesIsEnabled);
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001464 HI.Handler->endModule();
Chris Lattnerf0d6bd32010-04-05 05:11:15 +00001465 }
Timur Iskhodzhanov1cd14442013-12-03 15:10:23 +00001466 Handlers.clear();
Craig Topper353eda42014-04-24 06:44:33 +00001467 DD = nullptr;
Jim Grosbach1900c732011-03-29 23:20:22 +00001468
Chris Lattner2981dc12009-06-24 18:52:01 +00001469 // If the target wants to know about weak references, print them all.
Chris Lattnere9a75a62009-08-22 21:43:10 +00001470 if (MAI->getWeakRefDirective()) {
Chris Lattner2981dc12009-06-24 18:52:01 +00001471 // FIXME: This is not lazy, it would be nice to only print weak references
1472 // to stuff that is actually used. Note that doing so would require targets
1473 // to notice uses in operands (due to constant exprs etc). This should
1474 // happen with the MC stuff eventually.
Rafael Espindolad7998d02006-12-18 03:37:18 +00001475
Peter Collingbourne6d88fde2016-06-22 20:29:42 +00001476 // Print out module-level global objects here.
1477 for (const auto &GO : M.global_objects()) {
1478 if (!GO.hasExternalWeakLinkage())
Jim Grosbach6623e7f2014-04-16 22:38:02 +00001479 continue;
Peter Collingbourne6d88fde2016-06-22 20:29:42 +00001480 OutStreamer->EmitSymbolAttribute(getSymbol(&GO), MCSA_WeakReference);
Chris Lattner2981dc12009-06-24 18:52:01 +00001481 }
Rafael Espindolad7998d02006-12-18 03:37:18 +00001482 }
1483
Lang Hames9ff69c82015-04-24 19:11:51 +00001484 OutStreamer->AddBlankLine();
Anton Korobeynikova07765b2007-09-06 17:21:48 +00001485
Tim Shen800ed432016-03-31 22:08:19 +00001486 // Print aliases in topological order, that is, for each alias a = b,
1487 // b must be printed before a.
1488 // This is because on some targets (e.g. PowerPC) linker expects aliases in
1489 // such an order to generate correct TOC information.
1490 SmallVector<const GlobalAlias *, 16> AliasStack;
1491 SmallPtrSet<const GlobalAlias *, 16> AliasVisited;
1492 for (const auto &Alias : M.aliases()) {
1493 for (const GlobalAlias *Cur = &Alias; Cur;
1494 Cur = dyn_cast<GlobalAlias>(Cur->getAliasee())) {
1495 if (!AliasVisited.insert(Cur).second)
1496 break;
1497 AliasStack.push_back(Cur);
1498 }
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001499 for (const GlobalAlias *AncestorAlias : llvm::reverse(AliasStack))
Dmitry Polukhina3d5b0b2016-04-05 08:47:51 +00001500 emitGlobalIndirectSymbol(M, *AncestorAlias);
Tim Shen800ed432016-03-31 22:08:19 +00001501 AliasStack.clear();
Anton Korobeynikova97b6942007-04-25 14:27:10 +00001502 }
Dmitry Polukhina1feff72016-04-07 12:32:19 +00001503 for (const auto &IFunc : M.ifuncs())
1504 emitGlobalIndirectSymbol(M, IFunc);
Anton Korobeynikova97b6942007-04-25 14:27:10 +00001505
Duncan Sands5a913d62009-01-28 13:14:17 +00001506 GCModuleInfo *MI = getAnalysisIfAvailable<GCModuleInfo>();
Gordon Henriksend930f912008-08-17 18:44:35 +00001507 assert(MI && "AsmPrinter didn't require GCModuleInfo?");
1508 for (GCModuleInfo::iterator I = MI->end(), E = MI->begin(); I != E; )
David Blaikie88368ba2014-04-15 05:34:49 +00001509 if (GCMetadataPrinter *MP = GetOrCreateGCPrinter(**--I))
Philip Reames1e308972014-12-11 01:47:23 +00001510 MP->finishAssembly(M, *MI, *this);
Gordon Henriksen5180e852008-01-07 01:30:38 +00001511
Rafael Espindola0018a592013-10-16 01:49:05 +00001512 // Emit llvm.ident metadata in an '.ident' directive.
1513 EmitModuleIdents(M);
1514
Scott Linderde6beb02018-12-14 15:38:15 +00001515 // Emit bytes for llvm.commandline metadata.
1516 EmitModuleCommandLines(M);
1517
Peter Collingbourne7ef497b2014-12-30 20:05:19 +00001518 // Emit __morestack address if needed for indirect calls.
1519 if (MMI->usesMorestackAddr()) {
David Majnemera3ea4072016-02-21 01:30:30 +00001520 unsigned Align = 1;
Mehdi Amini5c0fa582015-07-16 06:04:17 +00001521 MCSection *ReadOnlySection = getObjFileLowering().getSectionForConstant(
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001522 getDataLayout(), SectionKind::getReadOnly(),
David Majnemera3ea4072016-02-21 01:30:30 +00001523 /*C=*/nullptr, Align);
Lang Hames9ff69c82015-04-24 19:11:51 +00001524 OutStreamer->SwitchSection(ReadOnlySection);
Peter Collingbourne7ef497b2014-12-30 20:05:19 +00001525
1526 MCSymbol *AddrSymbol =
Jim Grosbach6f482002015-05-18 18:43:14 +00001527 OutContext.getOrCreateSymbol(StringRef("__morestack_addr"));
Lang Hames9ff69c82015-04-24 19:11:51 +00001528 OutStreamer->EmitLabel(AddrSymbol);
Peter Collingbourne7ef497b2014-12-30 20:05:19 +00001529
Konstantin Zhuravlyovdc77b2e2017-04-17 17:41:25 +00001530 unsigned PtrSize = MAI->getCodePointerSize();
Lang Hames9ff69c82015-04-24 19:11:51 +00001531 OutStreamer->EmitSymbolValue(GetExternalSymbolSymbol("__morestack"),
1532 PtrSize);
Peter Collingbourne7ef497b2014-12-30 20:05:19 +00001533 }
1534
Than McIntoshdee2cf62017-09-27 19:34:00 +00001535 // Emit .note.GNU-split-stack and .note.GNU-no-split-stack sections if
1536 // split-stack is used.
1537 if (TM.getTargetTriple().isOSBinFormatELF() && MMI->hasSplitStack()) {
1538 OutStreamer->SwitchSection(
1539 OutContext.getELFSection(".note.GNU-split-stack", ELF::SHT_PROGBITS, 0));
1540 if (MMI->hasNosplitStack())
1541 OutStreamer->SwitchSection(
1542 OutContext.getELFSection(".note.GNU-no-split-stack", ELF::SHT_PROGBITS, 0));
1543 }
1544
Dan Gohmanbcde1722008-05-05 00:28:39 +00001545 // If we don't have any trampolines, then we don't require stack memory
1546 // to be executable. Some targets have a directive to declare this.
Chris Lattner2981dc12009-06-24 18:52:01 +00001547 Function *InitTrampolineIntrinsic = M.getFunction("llvm.init.trampoline");
Dan Gohmanbcde1722008-05-05 00:28:39 +00001548 if (!InitTrampolineIntrinsic || InitTrampolineIntrinsic->use_empty())
Rafael Espindola0709a7b2015-05-21 19:20:38 +00001549 if (MCSection *S = MAI->getNonexecutableStackSection(OutContext))
Lang Hames9ff69c82015-04-24 19:11:51 +00001550 OutStreamer->SwitchSection(S);
Jim Grosbach1900c732011-03-29 23:20:22 +00001551
Reid Kleckner1aa90612018-01-17 23:55:23 +00001552 if (TM.getTargetTriple().isOSBinFormatCOFF()) {
1553 // Emit /EXPORT: flags for each exported global as necessary.
1554 const auto &TLOF = getObjFileLowering();
1555 std::string Flags;
Saleem Abdulrasool99f479a2018-01-20 00:28:02 +00001556
Reid Kleckner1aa90612018-01-17 23:55:23 +00001557 for (const GlobalValue &GV : M.global_values()) {
1558 raw_string_ostream OS(Flags);
1559 TLOF.emitLinkerFlagsForGlobal(OS, &GV);
1560 OS.flush();
1561 if (!Flags.empty()) {
1562 OutStreamer->SwitchSection(TLOF.getDrectveSection());
1563 OutStreamer->EmitBytes(Flags);
1564 }
1565 Flags.clear();
1566 }
Saleem Abdulrasool99f479a2018-01-20 00:28:02 +00001567
1568 // Emit /INCLUDE: flags for each used global as necessary.
1569 if (const auto *LU = M.getNamedGlobal("llvm.used")) {
1570 assert(LU->hasInitializer() &&
1571 "expected llvm.used to have an initializer");
1572 assert(isa<ArrayType>(LU->getValueType()) &&
1573 "expected llvm.used to be an array type");
1574 if (const auto *A = cast<ConstantArray>(LU->getInitializer())) {
1575 for (const Value *Op : A->operands()) {
Peter Collingbourne2452d702019-08-22 19:56:14 +00001576 const auto *GV = cast<GlobalValue>(Op->stripPointerCasts());
Shoaib Meenaid8fd16b2018-01-26 00:15:25 +00001577 // Global symbols with internal or private linkage are not visible to
1578 // the linker, and thus would cause an error when the linker tried to
1579 // preserve the symbol due to the `/include:` directive.
Shoaib Meenai63b6a0e2018-01-29 20:28:04 +00001580 if (GV->hasLocalLinkage())
Saleem Abdulrasool99f479a2018-01-20 00:28:02 +00001581 continue;
1582
1583 raw_string_ostream OS(Flags);
1584 TLOF.emitLinkerFlagsForUsed(OS, GV);
1585 OS.flush();
1586
1587 if (!Flags.empty()) {
1588 OutStreamer->SwitchSection(TLOF.getDrectveSection());
1589 OutStreamer->EmitBytes(Flags);
1590 }
1591 Flags.clear();
1592 }
1593 }
1594 }
Reid Kleckner1aa90612018-01-17 23:55:23 +00001595 }
1596
Peter Collingbournebd9d3132018-07-17 22:40:08 +00001597 if (TM.Options.EmitAddrsig) {
1598 // Emit address-significance attributes for all globals.
1599 OutStreamer->EmitAddrsig();
1600 for (const GlobalValue &GV : M.global_values())
Peter Collingbourne3f792232018-08-24 20:37:09 +00001601 if (!GV.use_empty() && !GV.isThreadLocal() &&
1602 !GV.hasDLLImportStorageClass() && !GV.getName().startswith("llvm.") &&
Peter Collingbournefc504982018-07-18 00:21:40 +00001603 !GV.hasAtLeastLocalUnnamedAddr())
Peter Collingbournebd9d3132018-07-17 22:40:08 +00001604 OutStreamer->EmitAddrsigSym(getSymbol(&GV));
1605 }
1606
Peter Collingbourne31fda092019-05-29 03:29:01 +00001607 // Emit symbol partition specifications (ELF only).
1608 if (TM.getTargetTriple().isOSBinFormatELF()) {
1609 unsigned UniqueID = 0;
1610 for (const GlobalValue &GV : M.global_values()) {
1611 if (!GV.hasPartition() || GV.isDeclarationForLinker() ||
1612 GV.getVisibility() != GlobalValue::DefaultVisibility)
1613 continue;
1614
1615 OutStreamer->SwitchSection(OutContext.getELFSection(
1616 ".llvm_sympart", ELF::SHT_LLVM_SYMPART, 0, 0, "", ++UniqueID));
1617 OutStreamer->EmitBytes(GV.getPartition());
1618 OutStreamer->EmitZeros(1);
1619 OutStreamer->EmitValue(
1620 MCSymbolRefExpr::create(getSymbol(&GV), OutContext),
1621 MAI->getCodePointerSize());
1622 }
1623 }
1624
Chris Lattner71a15b12009-09-18 20:17:03 +00001625 // Allow the target to emit any magic that it wants at the end of the file,
1626 // after everything else has gone out.
1627 EmitEndOfAsmFile(M);
Jim Grosbach1900c732011-03-29 23:20:22 +00001628
Craig Topper353eda42014-04-24 06:44:33 +00001629 MMI = nullptr;
Jim Grosbach1900c732011-03-29 23:20:22 +00001630
Lang Hames9ff69c82015-04-24 19:11:51 +00001631 OutStreamer->Finish();
1632 OutStreamer->reset();
Michael Zolotukhin8d052a02018-04-09 00:54:47 +00001633 OwnedMLI.reset();
1634 OwnedMDT.reset();
Pedro Artigas7212ee42012-12-12 22:59:46 +00001635
Chris Lattner6a8e0f52004-08-16 23:15:22 +00001636 return false;
1637}
1638
Rafael Espindoladc4263c2015-03-17 13:57:48 +00001639MCSymbol *AsmPrinter::getCurExceptionSym() {
1640 if (!CurExceptionSym)
Rafael Espindola9ab09232015-03-17 20:07:06 +00001641 CurExceptionSym = createTempSymbol("exception");
Rafael Espindoladc4263c2015-03-17 13:57:48 +00001642 return CurExceptionSym;
1643}
1644
Chris Lattnerbb644e32005-11-21 07:51:36 +00001645void AsmPrinter::SetupMachineFunction(MachineFunction &MF) {
Chris Lattner67159522010-01-26 04:35:26 +00001646 this->MF = &MF;
Chris Lattnera0911792010-01-16 01:24:10 +00001647 // Get the function symbol.
Matthias Braunf1caa282017-12-15 22:22:58 +00001648 CurrentFnSym = getSymbol(&MF.getFunction());
Hal Finkelad4d9f52012-02-22 21:11:47 +00001649 CurrentFnSymForSize = CurrentFnSym;
Rafael Espindola092b6192015-03-05 19:47:50 +00001650 CurrentFnBegin = nullptr;
Rafael Espindoladc4263c2015-03-17 13:57:48 +00001651 CurExceptionSym = nullptr;
Rafael Espindola092b6192015-03-05 19:47:50 +00001652 bool NeedsLocalForSize = MAI->needsLocalForSize();
Rafael Espindola78fdca32018-03-26 20:40:22 +00001653 if (needFuncLabelsForEHOrDebugInfo(MF, MMI) || NeedsLocalForSize ||
1654 MF.getTarget().Options.EmitStackSizeSection) {
Rafael Espindola9ab09232015-03-17 20:07:06 +00001655 CurrentFnBegin = createTempSymbol("func_begin");
Rafael Espindola092b6192015-03-05 19:47:50 +00001656 if (NeedsLocalForSize)
1657 CurrentFnSymForSize = CurrentFnBegin;
1658 }
David Greenef354b6e2009-08-10 16:38:07 +00001659
Justin Bogner369ba752017-02-24 00:19:22 +00001660 ORE = &getAnalysis<MachineOptimizationRemarkEmitterPass>().getORE();
Chris Lattner6a8e0f52004-08-16 23:15:22 +00001661}
1662
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001663namespace {
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001664
Rafael Espindola0709a7b2015-05-21 19:20:38 +00001665// Keep track the alignment, constpool entries per Section.
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001666 struct SectionCPs {
Rafael Espindola0709a7b2015-05-21 19:20:38 +00001667 MCSection *S;
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001668 unsigned Alignment;
1669 SmallVector<unsigned, 4> CPEs;
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001670
Rafael Espindola0709a7b2015-05-21 19:20:38 +00001671 SectionCPs(MCSection *s, unsigned a) : S(s), Alignment(a) {}
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001672 };
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001673
1674} // end anonymous namespace
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001675
Chris Lattnerf2991ce2005-11-21 08:25:09 +00001676/// EmitConstantPool - Print to the current output stream assembly
1677/// representations of the constants in the constant pool MCP. This is
1678/// used to print out constants which have been "spilled to memory" by
1679/// the code generator.
Chris Lattner2b796242010-01-28 00:19:24 +00001680void AsmPrinter::EmitConstantPool() {
1681 const MachineConstantPool *MCP = MF->getConstantPool();
Chris Lattnerba972642006-02-09 04:22:52 +00001682 const std::vector<MachineConstantPoolEntry> &CP = MCP->getConstants();
Chris Lattnerf2991ce2005-11-21 08:25:09 +00001683 if (CP.empty()) return;
Evan Chengec1d60b2006-06-29 00:26:09 +00001684
Anton Korobeynikov123afb82008-09-24 22:17:59 +00001685 // Calculate sections for constant pool entries. We collect entries to go into
1686 // the same section together to reduce amount of section switch statements.
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001687 SmallVector<SectionCPs, 4> CPSections;
Chris Lattnerf2991ce2005-11-21 08:25:09 +00001688 for (unsigned i = 0, e = CP.size(); i != e; ++i) {
Chris Lattnerb300a4f2009-07-22 00:28:43 +00001689 const MachineConstantPoolEntry &CPE = CP[i];
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001690 unsigned Align = CPE.getAlignment();
Jim Grosbach1900c732011-03-29 23:20:22 +00001691
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001692 SectionKind Kind = CPE.getSectionKind(&getDataLayout());
Chris Lattnerfb6867c2009-07-26 06:26:55 +00001693
David Majnemer8bce66b2014-07-14 22:57:27 +00001694 const Constant *C = nullptr;
1695 if (!CPE.isMachineConstantPoolEntry())
1696 C = CPE.Val.ConstVal;
1697
David Majnemera3ea4072016-02-21 01:30:30 +00001698 MCSection *S = getObjFileLowering().getSectionForConstant(getDataLayout(),
1699 Kind, C, Align);
Jim Grosbach1900c732011-03-29 23:20:22 +00001700
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001701 // The number of sections are small, just do a linear search from the
1702 // last section to the first.
1703 bool Found = false;
1704 unsigned SecIdx = CPSections.size();
1705 while (SecIdx != 0) {
1706 if (CPSections[--SecIdx].S == S) {
1707 Found = true;
1708 break;
1709 }
1710 }
1711 if (!Found) {
1712 SecIdx = CPSections.size();
1713 CPSections.push_back(SectionCPs(S, Align));
1714 }
1715
1716 if (Align > CPSections[SecIdx].Alignment)
1717 CPSections[SecIdx].Alignment = Align;
1718 CPSections[SecIdx].CPEs.push_back(i);
Evan Chengec1d60b2006-06-29 00:26:09 +00001719 }
1720
Anton Korobeynikov123afb82008-09-24 22:17:59 +00001721 // Now print stuff into the calculated sections.
David Majnemer8bce66b2014-07-14 22:57:27 +00001722 const MCSection *CurSection = nullptr;
1723 unsigned Offset = 0;
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001724 for (unsigned i = 0, e = CPSections.size(); i != e; ++i) {
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001725 for (unsigned j = 0, ee = CPSections[i].CPEs.size(); j != ee; ++j) {
1726 unsigned CPI = CPSections[i].CPEs[j];
David Majnemer8bce66b2014-07-14 22:57:27 +00001727 MCSymbol *Sym = GetCPISymbol(CPI);
1728 if (!Sym->isUndefined())
1729 continue;
1730
1731 if (CurSection != CPSections[i].S) {
Lang Hames9ff69c82015-04-24 19:11:51 +00001732 OutStreamer->SwitchSection(CPSections[i].S);
David Majnemer8bce66b2014-07-14 22:57:27 +00001733 EmitAlignment(Log2_32(CPSections[i].Alignment));
1734 CurSection = CPSections[i].S;
1735 Offset = 0;
1736 }
1737
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001738 MachineConstantPoolEntry CPE = CP[CPI];
Anton Korobeynikov123afb82008-09-24 22:17:59 +00001739
Chris Lattnerf6190822006-02-09 04:46:04 +00001740 // Emit inter-object padding for alignment.
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001741 unsigned AlignMask = CPE.getAlignment() - 1;
1742 unsigned NewOffset = (Offset + AlignMask) & ~AlignMask;
Lang Hames9ff69c82015-04-24 19:11:51 +00001743 OutStreamer->EmitZeros(NewOffset - Offset);
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001744
Chris Lattner229907c2011-07-18 04:54:35 +00001745 Type *Ty = CPE.getType();
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001746 Offset = NewOffset + getDataLayout().getTypeAllocSize(Ty);
Chris Lattner4225a7b2010-01-23 05:19:23 +00001747
Lang Hames9ff69c82015-04-24 19:11:51 +00001748 OutStreamer->EmitLabel(Sym);
Evan Cheng1fb8aed2009-03-13 07:51:59 +00001749 if (CPE.isMachineConstantPoolEntry())
1750 EmitMachineConstantPoolValue(CPE.Val.MachineCPVal);
1751 else
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001752 EmitGlobalConstant(getDataLayout(), CPE.Val.ConstVal);
Chris Lattnerf6190822006-02-09 04:46:04 +00001753 }
Chris Lattnerf2991ce2005-11-21 08:25:09 +00001754 }
1755}
1756
Nate Begeman4ca2ea52006-04-22 18:53:45 +00001757/// EmitJumpTableInfo - Print assembly representations of the jump tables used
Jim Grosbach1900c732011-03-29 23:20:22 +00001758/// by the current function to the current output stream.
Chris Lattner94a946c2010-01-28 01:02:27 +00001759void AsmPrinter::EmitJumpTableInfo() {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001760 const DataLayout &DL = MF->getDataLayout();
Chris Lattner94a946c2010-01-28 01:02:27 +00001761 const MachineJumpTableInfo *MJTI = MF->getJumpTableInfo();
Craig Topper353eda42014-04-24 06:44:33 +00001762 if (!MJTI) return;
Richard Osborne6d3e92d2010-03-11 14:58:16 +00001763 if (MJTI->getEntryKind() == MachineJumpTableInfo::EK_Inline) return;
Nate Begeman4ca2ea52006-04-22 18:53:45 +00001764 const std::vector<MachineJumpTableEntry> &JT = MJTI->getJumpTables();
1765 if (JT.empty()) return;
Anton Korobeynikov2c638782007-11-14 09:18:41 +00001766
Jim Grosbach1900c732011-03-29 23:20:22 +00001767 // Pick the directive to use to print the jump table entries, and switch to
Nate Begeman78756502006-07-27 01:13:04 +00001768 // the appropriate section.
Matthias Braunf1caa282017-12-15 22:22:58 +00001769 const Function &F = MF->getFunction();
Rafael Espindola29786d42015-02-12 17:16:46 +00001770 const TargetLoweringObjectFile &TLOF = getObjFileLowering();
Rafael Espindoladf195192015-02-17 23:34:51 +00001771 bool JTInDiffSection = !TLOF.shouldPutJumpTableInFunctionSection(
1772 MJTI->getEntryKind() == MachineJumpTableInfo::EK_LabelDifference32,
Matthias Braunf1caa282017-12-15 22:22:58 +00001773 F);
Rafael Espindolaa6001792015-03-09 18:29:12 +00001774 if (JTInDiffSection) {
1775 // Drop it in the readonly section.
Matthias Braunf1caa282017-12-15 22:22:58 +00001776 MCSection *ReadOnlySection = TLOF.getSectionForJumpTable(F, TM);
Lang Hames9ff69c82015-04-24 19:11:51 +00001777 OutStreamer->SwitchSection(ReadOnlySection);
Nate Begeman78756502006-07-27 01:13:04 +00001778 }
Chris Lattnerb6db2c62010-01-25 23:26:13 +00001779
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001780 EmitAlignment(Log2_32(MJTI->getEntryAlignment(DL)));
Jim Grosbach1900c732011-03-29 23:20:22 +00001781
Jim Grosbach361ca342012-09-24 23:06:27 +00001782 // Jump tables in code sections are marked with a data_region directive
1783 // where that's supported.
1784 if (!JTInDiffSection)
Lang Hames9ff69c82015-04-24 19:11:51 +00001785 OutStreamer->EmitDataRegion(MCDR_DataRegionJT32);
Jim Grosbach361ca342012-09-24 23:06:27 +00001786
Chris Lattner4db603f2010-01-26 06:42:44 +00001787 for (unsigned JTI = 0, e = JT.size(); JTI != e; ++JTI) {
1788 const std::vector<MachineBasicBlock*> &JTBBs = JT[JTI].MBBs;
Jim Grosbach1900c732011-03-29 23:20:22 +00001789
1790 // If this jump table was deleted, ignore it.
Chris Lattner28bfe382006-10-28 18:10:06 +00001791 if (JTBBs.empty()) continue;
Nate Begeman984c1a42006-08-12 21:29:52 +00001792
Rafael Espindolac606bfe2014-10-21 01:17:30 +00001793 // For the EK_LabelDifference32 entry, if using .set avoids a relocation,
1794 /// emit a .set directive for each unique entry.
Chris Lattnercc0f7342010-01-26 05:15:20 +00001795 if (MJTI->getEntryKind() == MachineJumpTableInfo::EK_LabelDifference32 &&
Joerg Sonnenberger22982032016-06-18 23:25:37 +00001796 MAI->doesSetDirectiveSuppressReloc()) {
Chris Lattner4db603f2010-01-26 06:42:44 +00001797 SmallPtrSet<const MachineBasicBlock*, 16> EmittedSets;
Eric Christopher504f3882015-02-19 18:46:23 +00001798 const TargetLowering *TLI = MF->getSubtarget().getTargetLowering();
Chris Lattner94a946c2010-01-28 01:02:27 +00001799 const MCExpr *Base = TLI->getPICJumpTableRelocBaseExpr(MF,JTI,OutContext);
Chris Lattner4db603f2010-01-26 06:42:44 +00001800 for (unsigned ii = 0, ee = JTBBs.size(); ii != ee; ++ii) {
1801 const MachineBasicBlock *MBB = JTBBs[ii];
David Blaikie70573dc2014-11-19 07:49:26 +00001802 if (!EmittedSets.insert(MBB).second)
1803 continue;
Jim Grosbach1900c732011-03-29 23:20:22 +00001804
Chris Lattnerb657c4c2010-01-26 21:53:08 +00001805 // .set LJTSet, LBB32-base
1806 const MCExpr *LHS =
Jim Grosbach13760bd2015-05-30 01:25:56 +00001807 MCSymbolRefExpr::create(MBB->getSymbol(), OutContext);
Lang Hames9ff69c82015-04-24 19:11:51 +00001808 OutStreamer->EmitAssignment(GetJTSetSymbol(JTI, MBB->getNumber()),
Jim Grosbach13760bd2015-05-30 01:25:56 +00001809 MCBinaryExpr::createSub(LHS, Base,
Lang Hames9ff69c82015-04-24 19:11:51 +00001810 OutContext));
Chris Lattner4db603f2010-01-26 06:42:44 +00001811 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001812 }
1813
Duncan Sands75b5d272011-02-15 09:23:02 +00001814 // On some targets (e.g. Darwin) we want to emit two consecutive labels
Chris Lattner0ee2d462007-01-18 01:12:56 +00001815 // before each jump table. The first label is never referenced, but tells
1816 // the assembler and linker the extents of the jump table object. The
1817 // second label is actually referenced by the code.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001818 if (JTInDiffSection && DL.hasLinkerPrivateGlobalPrefix())
Chris Lattner273735b2010-01-26 05:58:28 +00001819 // FIXME: This doesn't have to have any specific name, just any randomly
1820 // named and numbered 'l' label would work. Simplify GetJTISymbol.
Lang Hames9ff69c82015-04-24 19:11:51 +00001821 OutStreamer->EmitLabel(GetJTISymbol(JTI, true));
Chris Lattner4225a7b2010-01-23 05:19:23 +00001822
Lang Hames9ff69c82015-04-24 19:11:51 +00001823 OutStreamer->EmitLabel(GetJTISymbol(JTI));
Chris Lattner4225a7b2010-01-23 05:19:23 +00001824
Chris Lattner279de3e2010-01-26 05:10:10 +00001825 for (unsigned ii = 0, ee = JTBBs.size(); ii != ee; ++ii)
Chris Lattner4db603f2010-01-26 06:42:44 +00001826 EmitJumpTableEntry(MJTI, JTBBs[ii], JTI);
Nate Begeman4ca2ea52006-04-22 18:53:45 +00001827 }
Jim Grosbach361ca342012-09-24 23:06:27 +00001828 if (!JTInDiffSection)
Lang Hames9ff69c82015-04-24 19:11:51 +00001829 OutStreamer->EmitDataRegion(MCDR_DataRegionEnd);
Nate Begeman4ca2ea52006-04-22 18:53:45 +00001830}
1831
Chris Lattner279de3e2010-01-26 05:10:10 +00001832/// EmitJumpTableEntry - Emit a jump table entry for the specified MBB to the
1833/// current stream.
1834void AsmPrinter::EmitJumpTableEntry(const MachineJumpTableInfo *MJTI,
1835 const MachineBasicBlock *MBB,
1836 unsigned UID) const {
Jakob Stoklund Olesen6d4d8582011-02-09 21:52:06 +00001837 assert(MBB && MBB->getNumber() >= 0 && "Invalid basic block");
Craig Topper353eda42014-04-24 06:44:33 +00001838 const MCExpr *Value = nullptr;
Chris Lattnera7c11602010-01-26 03:43:22 +00001839 switch (MJTI->getEntryKind()) {
Richard Osborne6d3e92d2010-03-11 14:58:16 +00001840 case MachineJumpTableInfo::EK_Inline:
David Blaikie46a9f012012-01-20 21:51:11 +00001841 llvm_unreachable("Cannot emit EK_Inline jump table entry");
Chris Lattner5fc41602010-01-26 04:05:28 +00001842 case MachineJumpTableInfo::EK_Custom32:
Eric Christopher504f3882015-02-19 18:46:23 +00001843 Value = MF->getSubtarget().getTargetLowering()->LowerCustomJumpTableEntry(
1844 MJTI, MBB, UID, OutContext);
Chris Lattner5fc41602010-01-26 04:05:28 +00001845 break;
Chris Lattnera7c11602010-01-26 03:43:22 +00001846 case MachineJumpTableInfo::EK_BlockAddress:
1847 // EK_BlockAddress - Each entry is a plain address of block, e.g.:
1848 // .word LBB123
Jim Grosbach13760bd2015-05-30 01:25:56 +00001849 Value = MCSymbolRefExpr::create(MBB->getSymbol(), OutContext);
Chris Lattnera7c11602010-01-26 03:43:22 +00001850 break;
Chris Lattnera7c11602010-01-26 03:43:22 +00001851 case MachineJumpTableInfo::EK_GPRel32BlockAddress: {
1852 // EK_GPRel32BlockAddress - Each entry is an address of block, encoded
1853 // with a relocation as gp-relative, e.g.:
1854 // .gprel32 LBB123
Chris Lattner29bdac42010-03-13 21:04:28 +00001855 MCSymbol *MBBSym = MBB->getSymbol();
Jim Grosbach13760bd2015-05-30 01:25:56 +00001856 OutStreamer->EmitGPRel32Value(MCSymbolRefExpr::create(MBBSym, OutContext));
Chris Lattner19bd0392010-01-25 21:10:10 +00001857 return;
Anton Korobeynikov2c638782007-11-14 09:18:41 +00001858 }
Chris Lattner5fc41602010-01-26 04:05:28 +00001859
Akira Hatanakaf0b08442012-02-03 04:33:00 +00001860 case MachineJumpTableInfo::EK_GPRel64BlockAddress: {
1861 // EK_GPRel64BlockAddress - Each entry is an address of block, encoded
1862 // with a relocation as gp-relative, e.g.:
1863 // .gpdword LBB123
1864 MCSymbol *MBBSym = MBB->getSymbol();
Jim Grosbach13760bd2015-05-30 01:25:56 +00001865 OutStreamer->EmitGPRel64Value(MCSymbolRefExpr::create(MBBSym, OutContext));
Akira Hatanakaf0b08442012-02-03 04:33:00 +00001866 return;
1867 }
1868
Chris Lattnera7c11602010-01-26 03:43:22 +00001869 case MachineJumpTableInfo::EK_LabelDifference32: {
Rafael Espindolac606bfe2014-10-21 01:17:30 +00001870 // Each entry is the address of the block minus the address of the jump
1871 // table. This is used for PIC jump tables where gprel32 is not supported.
1872 // e.g.:
Chris Lattnera7c11602010-01-26 03:43:22 +00001873 // .word LBB123 - LJTI1_2
Rafael Espindolac606bfe2014-10-21 01:17:30 +00001874 // If the .set directive avoids relocations, this is emitted as:
Chris Lattnera7c11602010-01-26 03:43:22 +00001875 // .set L4_5_set_123, LBB123 - LJTI1_2
1876 // .word L4_5_set_123
Joerg Sonnenberger22982032016-06-18 23:25:37 +00001877 if (MAI->doesSetDirectiveSuppressReloc()) {
Jim Grosbach13760bd2015-05-30 01:25:56 +00001878 Value = MCSymbolRefExpr::create(GetJTSetSymbol(UID, MBB->getNumber()),
Chris Lattnera7c11602010-01-26 03:43:22 +00001879 OutContext);
1880 break;
1881 }
Jim Grosbach13760bd2015-05-30 01:25:56 +00001882 Value = MCSymbolRefExpr::create(MBB->getSymbol(), OutContext);
Eric Christophera7249ec2015-02-20 07:16:19 +00001883 const TargetLowering *TLI = MF->getSubtarget().getTargetLowering();
Rafael Espindola26cfbea2014-11-06 14:39:49 +00001884 const MCExpr *Base = TLI->getPICJumpTableRelocBaseExpr(MF, UID, OutContext);
Jim Grosbach13760bd2015-05-30 01:25:56 +00001885 Value = MCBinaryExpr::createSub(Value, Base, OutContext);
Chris Lattnera7c11602010-01-26 03:43:22 +00001886 break;
1887 }
Anton Korobeynikov2c638782007-11-14 09:18:41 +00001888 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001889
Chris Lattnera7c11602010-01-26 03:43:22 +00001890 assert(Value && "Unknown entry kind!");
Jim Grosbach1900c732011-03-29 23:20:22 +00001891
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001892 unsigned EntrySize = MJTI->getEntrySize(getDataLayout());
Lang Hames9ff69c82015-04-24 19:11:51 +00001893 OutStreamer->EmitValue(Value, EntrySize);
Anton Korobeynikov2c638782007-11-14 09:18:41 +00001894}
1895
Chris Lattnerf0e9aef2005-12-13 06:32:10 +00001896/// EmitSpecialLLVMGlobal - Check to see if the specified global is a
1897/// special global used by LLVM. If so, emit it and return true, otherwise
1898/// do nothing and return false.
1899bool AsmPrinter::EmitSpecialLLVMGlobal(const GlobalVariable *GV) {
Daniel Dunbare03eecb2009-07-25 23:55:21 +00001900 if (GV->getName() == "llvm.used") {
Chris Lattner1deb09c2010-01-23 05:51:36 +00001901 if (MAI->hasNoDeadStrip()) // No need to emit this at all.
Rafael Espindola74f2e462013-04-22 14:58:02 +00001902 EmitLLVMUsedList(cast<ConstantArray>(GV->getInitializer()));
Andrew Lenharthbeb80a92007-08-22 19:33:11 +00001903 return true;
1904 }
1905
Chris Lattner58f9bb22009-07-20 06:14:25 +00001906 // Ignore debug and non-emitted data. This handles llvm.compiler.used.
Rafael Espindola83658d62016-05-11 18:21:59 +00001907 if (GV->getSection() == "llvm.metadata" ||
Chris Lattner184f1be2009-04-13 05:44:34 +00001908 GV->hasAvailableExternallyLinkage())
1909 return true;
Jim Grosbach1900c732011-03-29 23:20:22 +00001910
Jim Laskey313570f2006-03-07 22:00:35 +00001911 if (!GV->hasAppendingLinkage()) return false;
1912
1913 assert(GV->hasInitializer() && "Not a special LLVM global!");
Jim Grosbach1900c732011-03-29 23:20:22 +00001914
Chris Lattner317293b2009-03-09 05:52:15 +00001915 if (GV->getName() == "llvm.global_ctors") {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001916 EmitXXStructorList(GV->getParent()->getDataLayout(), GV->getInitializer(),
1917 /* isCtor */ true);
Jim Grosbach1900c732011-03-29 23:20:22 +00001918
Chris Lattner126dab22009-03-09 08:18:48 +00001919 return true;
Jim Grosbach1900c732011-03-29 23:20:22 +00001920 }
1921
Chris Lattner317293b2009-03-09 05:52:15 +00001922 if (GV->getName() == "llvm.global_dtors") {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001923 EmitXXStructorList(GV->getParent()->getDataLayout(), GV->getInitializer(),
1924 /* isCtor */ false);
Chris Lattnere9d28b12010-01-19 04:34:02 +00001925
Chris Lattner126dab22009-03-09 08:18:48 +00001926 return true;
Chris Lattnerf0e9aef2005-12-13 06:32:10 +00001927 }
Jim Grosbach1900c732011-03-29 23:20:22 +00001928
Rafael Espindolae64619c2016-05-16 21:14:24 +00001929 report_fatal_error("unknown special variable");
Chris Lattnerf0e9aef2005-12-13 06:32:10 +00001930}
1931
Chris Lattnere9a75a62009-08-22 21:43:10 +00001932/// EmitLLVMUsedList - For targets that define a MAI::UsedDirective, mark each
Sam Clegg1e71b042019-02-07 00:11:43 +00001933/// global in the specified llvm.used list.
Rafael Espindola74f2e462013-04-22 14:58:02 +00001934void AsmPrinter::EmitLLVMUsedList(const ConstantArray *InitList) {
Dan Gohman4fe64de2009-06-14 23:30:43 +00001935 // Should be an array of 'i8*'.
Chris Lattner66af3902006-09-26 03:38:18 +00001936 for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) {
Chris Lattner1fcac9f2009-07-17 22:00:23 +00001937 const GlobalValue *GV =
1938 dyn_cast<GlobalValue>(InitList->getOperand(i)->stripPointerCasts());
Rafael Espindola3b30cb42014-03-06 22:47:08 +00001939 if (GV)
Lang Hames9ff69c82015-04-24 19:11:51 +00001940 OutStreamer->EmitSymbolAttribute(getSymbol(GV), MCSA_NoDeadStrip);
Chris Lattner66af3902006-09-26 03:38:18 +00001941 }
1942}
1943
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001944namespace {
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001945
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001946struct Structor {
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001947 int Priority = 0;
1948 Constant *Func = nullptr;
1949 GlobalValue *ComdatKey = nullptr;
1950
1951 Structor() = default;
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001952};
Eugene Zelenkod96089b2017-02-14 00:33:36 +00001953
Eugene Zelenko149178d2017-10-10 22:33:29 +00001954} // end anonymous namespace
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001955
Duncan Sands4d635422011-08-28 13:17:22 +00001956/// EmitXXStructorList - Emit the ctor or dtor list taking into account the init
1957/// priority.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001958void AsmPrinter::EmitXXStructorList(const DataLayout &DL, const Constant *List,
1959 bool isCtor) {
Fangrui Songf4dfd632019-05-15 02:35:32 +00001960 // Should be an array of '{ i32, void ()*, i8* }' structs. The first value is the
Duncan Sands4d635422011-08-28 13:17:22 +00001961 // init priority.
Chris Lattnerf0e9aef2005-12-13 06:32:10 +00001962 if (!isa<ConstantArray>(List)) return;
Chris Lattner434ffe42005-12-21 01:17:37 +00001963
Duncan Sands4d635422011-08-28 13:17:22 +00001964 // Sanity check the structors list.
1965 const ConstantArray *InitList = dyn_cast<ConstantArray>(List);
1966 if (!InitList) return; // Not an array!
1967 StructType *ETy = dyn_cast<StructType>(InitList->getType()->getElementType());
Fangrui Songf4dfd632019-05-15 02:35:32 +00001968 if (!ETy || ETy->getNumElements() != 3 ||
1969 !isa<IntegerType>(ETy->getTypeAtIndex(0U)) ||
1970 !isa<PointerType>(ETy->getTypeAtIndex(1U)) ||
1971 !isa<PointerType>(ETy->getTypeAtIndex(2U)))
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001972 return; // Not (int, ptr, ptr).
Duncan Sands4d635422011-08-28 13:17:22 +00001973
1974 // Gather the structors in a form that's convenient for sorting by priority.
1975 SmallVector<Structor, 8> Structors;
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001976 for (Value *O : InitList->operands()) {
1977 ConstantStruct *CS = dyn_cast<ConstantStruct>(O);
Duncan Sands4d635422011-08-28 13:17:22 +00001978 if (!CS) continue; // Malformed.
1979 if (CS->getOperand(1)->isNullValue())
1980 break; // Found a null terminator, skip the rest.
1981 ConstantInt *Priority = dyn_cast<ConstantInt>(CS->getOperand(0));
1982 if (!Priority) continue; // Malformed.
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001983 Structors.push_back(Structor());
1984 Structor &S = Structors.back();
1985 S.Priority = Priority->getLimitedValue(65535);
1986 S.Func = CS->getOperand(1);
Fangrui Songf4dfd632019-05-15 02:35:32 +00001987 if (!CS->getOperand(2)->isNullValue())
Eric Christopher0b6537e2016-07-01 06:07:31 +00001988 S.ComdatKey =
1989 dyn_cast<GlobalValue>(CS->getOperand(2)->stripPointerCasts());
Duncan Sands4d635422011-08-28 13:17:22 +00001990 }
1991
Anton Korobeynikov7722a2d2012-01-25 22:24:19 +00001992 // Emit the function pointers in the target-specific order
Mehdi Aminibd7287e2015-07-16 06:11:10 +00001993 unsigned Align = Log2_32(DL.getPointerPrefAlignment());
Fangrui Songefd94c52019-04-23 14:51:27 +00001994 llvm::stable_sort(Structors, [](const Structor &L, const Structor &R) {
1995 return L.Priority < R.Priority;
1996 });
Reid Klecknerfceb76f2014-05-16 20:39:27 +00001997 for (Structor &S : Structors) {
1998 const TargetLoweringObjectFile &Obj = getObjFileLowering();
1999 const MCSymbol *KeySym = nullptr;
Hans Wennborg8e873322014-06-04 21:04:54 +00002000 if (GlobalValue *GV = S.ComdatKey) {
Teresa Johnson2d384ac2017-01-18 16:58:43 +00002001 if (GV->isDeclarationForLinker())
2002 // If the associated variable is not defined in this module
2003 // (it might be available_externally, or have been an
2004 // available_externally definition that was dropped by the
2005 // EliminateAvailableExternally pass), some other TU
Hans Wennborg8e873322014-06-04 21:04:54 +00002006 // will provide its dynamic initializer.
2007 continue;
2008
2009 KeySym = getSymbol(GV);
Reid Klecknerfceb76f2014-05-16 20:39:27 +00002010 }
Rafael Espindola0709a7b2015-05-21 19:20:38 +00002011 MCSection *OutputSection =
Rafael Espindola0766ae02014-06-06 19:26:12 +00002012 (isCtor ? Obj.getStaticCtorSection(S.Priority, KeySym)
2013 : Obj.getStaticDtorSection(S.Priority, KeySym));
Lang Hames9ff69c82015-04-24 19:11:51 +00002014 OutStreamer->SwitchSection(OutputSection);
2015 if (OutStreamer->getCurrentSection() != OutStreamer->getPreviousSection())
Anton Korobeynikov7722a2d2012-01-25 22:24:19 +00002016 EmitAlignment(Align);
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002017 EmitXXStructor(DL, S.Func);
Anton Korobeynikov7722a2d2012-01-25 22:24:19 +00002018 }
Chris Lattnerf0e9aef2005-12-13 06:32:10 +00002019}
Chris Lattnerf2991ce2005-11-21 08:25:09 +00002020
Rafael Espindola0018a592013-10-16 01:49:05 +00002021void AsmPrinter::EmitModuleIdents(Module &M) {
2022 if (!MAI->hasIdentDirective())
2023 return;
2024
2025 if (const NamedMDNode *NMD = M.getNamedMetadata("llvm.ident")) {
2026 for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) {
Duncan P. N. Exon Smithde36e802014-11-11 21:30:22 +00002027 const MDNode *N = NMD->getOperand(i);
Adrian Prantlc4fd6b72014-02-11 21:23:02 +00002028 assert(N->getNumOperands() == 1 &&
Rafael Espindola0018a592013-10-16 01:49:05 +00002029 "llvm.ident metadata entry can have only one operand");
2030 const MDString *S = cast<MDString>(N->getOperand(0));
Lang Hames9ff69c82015-04-24 19:11:51 +00002031 OutStreamer->EmitIdent(S->getString());
Rafael Espindola0018a592013-10-16 01:49:05 +00002032 }
2033 }
2034}
2035
Scott Linderde6beb02018-12-14 15:38:15 +00002036void AsmPrinter::EmitModuleCommandLines(Module &M) {
2037 MCSection *CommandLine = getObjFileLowering().getSectionForCommandLines();
2038 if (!CommandLine)
2039 return;
2040
2041 const NamedMDNode *NMD = M.getNamedMetadata("llvm.commandline");
2042 if (!NMD || !NMD->getNumOperands())
2043 return;
2044
2045 OutStreamer->PushSection();
2046 OutStreamer->SwitchSection(CommandLine);
2047 OutStreamer->EmitZeros(1);
2048 for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) {
2049 const MDNode *N = NMD->getOperand(i);
2050 assert(N->getNumOperands() == 1 &&
2051 "llvm.commandline metadata entry can have only one operand");
2052 const MDString *S = cast<MDString>(N->getOperand(0));
2053 OutStreamer->EmitBytes(S->getString());
2054 OutStreamer->EmitZeros(1);
2055 }
2056 OutStreamer->PopSection();
2057}
2058
Jim Laskey1c055e82007-01-25 15:12:02 +00002059//===--------------------------------------------------------------------===//
2060// Emission and print routines
2061//
2062
Rafael Espindola4b4d85f2018-03-29 23:32:54 +00002063/// Emit a byte directive and value.
Jim Laskey1c055e82007-01-25 15:12:02 +00002064///
Rafael Espindola4b4d85f2018-03-29 23:32:54 +00002065void AsmPrinter::emitInt8(int Value) const {
Lang Hames9ff69c82015-04-24 19:11:51 +00002066 OutStreamer->EmitIntValue(Value, 1);
Jim Laskey1c055e82007-01-25 15:12:02 +00002067}
2068
Rafael Espindola4b4d85f2018-03-29 23:32:54 +00002069/// Emit a short directive and value.
2070void AsmPrinter::emitInt16(int Value) const {
Lang Hames9ff69c82015-04-24 19:11:51 +00002071 OutStreamer->EmitIntValue(Value, 2);
Jim Laskey1c055e82007-01-25 15:12:02 +00002072}
2073
Rafael Espindola4b4d85f2018-03-29 23:32:54 +00002074/// Emit a long directive and value.
2075void AsmPrinter::emitInt32(int Value) const {
Lang Hames9ff69c82015-04-24 19:11:51 +00002076 OutStreamer->EmitIntValue(Value, 4);
Jim Laskey1c055e82007-01-25 15:12:02 +00002077}
2078
James Hendersona3acf992018-05-10 10:51:33 +00002079/// Emit a long long directive and value.
2080void AsmPrinter::emitInt64(uint64_t Value) const {
2081 OutStreamer->EmitIntValue(Value, 8);
2082}
2083
Rafael Espindolac606bfe2014-10-21 01:17:30 +00002084/// Emit something like ".long Hi-Lo" where the size in bytes of the directive
2085/// is specified by Size and Hi/Lo specify the labels. This implicitly uses
2086/// .set if it avoids relocations.
Chris Lattner9889c1e2010-03-08 23:58:37 +00002087void AsmPrinter::EmitLabelDifference(const MCSymbol *Hi, const MCSymbol *Lo,
2088 unsigned Size) const {
Rafael Espindola7c6e6e42015-06-11 18:58:08 +00002089 OutStreamer->emitAbsoluteSymbolDiff(Hi, Lo, Size);
Chris Lattner9889c1e2010-03-08 23:58:37 +00002090}
2091
Jim Grosbach1900c732011-03-29 23:20:22 +00002092/// EmitLabelPlusOffset - Emit something like ".long Label+Offset"
Devang Patelda3ef852010-09-02 16:43:44 +00002093/// where the size in bytes of the directive is specified by Size and Label
2094/// specifies the label. This implicitly uses .set if it is available.
2095void AsmPrinter::EmitLabelPlusOffset(const MCSymbol *Label, uint64_t Offset,
Eric Christopherf8da6aa2013-11-23 00:05:06 +00002096 unsigned Size,
2097 bool IsSectionRelative) const {
Matt Arsenault40dddd72013-10-03 19:50:01 +00002098 if (MAI->needsDwarfSectionOffsetDirective() && IsSectionRelative) {
Keno Fischerf7d84ee2017-01-02 03:00:19 +00002099 OutStreamer->EmitCOFFSecRel32(Label, Offset);
2100 if (Size > 4)
2101 OutStreamer->EmitZeros(Size - 4);
Carlo Kok4382da92013-08-02 16:14:15 +00002102 return;
2103 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002104
Nick Lewycky33da3362012-06-22 01:25:12 +00002105 // Emit Label+Offset (or just Label if Offset is zero)
Jim Grosbach13760bd2015-05-30 01:25:56 +00002106 const MCExpr *Expr = MCSymbolRefExpr::create(Label, OutContext);
Nick Lewycky33da3362012-06-22 01:25:12 +00002107 if (Offset)
Jim Grosbach13760bd2015-05-30 01:25:56 +00002108 Expr = MCBinaryExpr::createAdd(
2109 Expr, MCConstantExpr::create(Offset, OutContext), OutContext);
Jim Grosbach1900c732011-03-29 23:20:22 +00002110
Lang Hames9ff69c82015-04-24 19:11:51 +00002111 OutStreamer->EmitValue(Expr, Size);
Devang Patelda3ef852010-09-02 16:43:44 +00002112}
Jim Grosbach1900c732011-03-29 23:20:22 +00002113
Jim Laskey1c055e82007-01-25 15:12:02 +00002114//===----------------------------------------------------------------------===//
2115
Chris Lattner3e3ff302007-05-31 18:57:45 +00002116// EmitAlignment - Emit an alignment directive to the specified power of
2117// two boundary. For example, if you pass in 3 here, you will get an 8
2118// byte alignment. If a global value is specified, and if that global has
Chris Lattner08e9e722010-04-28 19:58:07 +00002119// an explicit alignment requested, it will override the alignment request
2120// if required for correctness.
Rafael Espindola99e05cf2014-05-13 18:45:48 +00002121void AsmPrinter::EmitAlignment(unsigned NumBits, const GlobalObject *GV) const {
Eric Christopherd9134482014-08-04 21:25:23 +00002122 if (GV)
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002123 NumBits = getGVAlignmentLog2(GV, GV->getParent()->getDataLayout(), NumBits);
Jim Grosbach1900c732011-03-29 23:20:22 +00002124
Chris Lattner08e9e722010-04-28 19:58:07 +00002125 if (NumBits == 0) return; // 1-byte aligned: no need to emit alignment.
Jim Grosbach1900c732011-03-29 23:20:22 +00002126
Aaron Ballman0d6a0102014-12-16 14:04:11 +00002127 assert(NumBits <
2128 static_cast<unsigned>(std::numeric_limits<unsigned>::digits) &&
Michael Ilsemanaddddc42014-12-15 18:48:43 +00002129 "undefined behavior");
Chris Lattner0adae252009-08-18 06:15:16 +00002130 if (getCurrentSection()->getKind().isText())
Lang Hames9ff69c82015-04-24 19:11:51 +00002131 OutStreamer->EmitCodeAlignment(1u << NumBits);
Chris Lattnere4f249b2010-02-23 18:46:22 +00002132 else
Lang Hames9ff69c82015-04-24 19:11:51 +00002133 OutStreamer->EmitValueToAlignment(1u << NumBits);
Chris Lattner1d35c162004-08-17 19:14:29 +00002134}
David Greenefdd25192009-08-05 21:00:52 +00002135
Chris Lattner51003672010-04-20 06:20:21 +00002136//===----------------------------------------------------------------------===//
2137// Constant emission.
2138//===----------------------------------------------------------------------===//
2139
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002140const MCExpr *AsmPrinter::lowerConstant(const Constant *CV) {
2141 MCContext &Ctx = OutContext;
Jim Grosbach1900c732011-03-29 23:20:22 +00002142
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002143 if (CV->isNullValue() || isa<UndefValue>(CV))
Jim Grosbach13760bd2015-05-30 01:25:56 +00002144 return MCConstantExpr::create(0, Ctx);
Chris Lattner7e0c1642010-01-13 04:29:19 +00002145
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002146 if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV))
Jim Grosbach13760bd2015-05-30 01:25:56 +00002147 return MCConstantExpr::create(CI->getZExtValue(), Ctx);
Jim Grosbach1900c732011-03-29 23:20:22 +00002148
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002149 if (const GlobalValue *GV = dyn_cast<GlobalValue>(CV))
Jim Grosbach13760bd2015-05-30 01:25:56 +00002150 return MCSymbolRefExpr::create(getSymbol(GV), Ctx);
Bill Wendling0d323ae2010-08-18 18:41:13 +00002151
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002152 if (const BlockAddress *BA = dyn_cast<BlockAddress>(CV))
Jim Grosbach13760bd2015-05-30 01:25:56 +00002153 return MCSymbolRefExpr::create(GetBlockAddressSymbol(BA), Ctx);
Jim Grosbach1900c732011-03-29 23:20:22 +00002154
Chris Lattner7e0c1642010-01-13 04:29:19 +00002155 const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV);
Craig Topper353eda42014-04-24 06:44:33 +00002156 if (!CE) {
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002157 llvm_unreachable("Unknown constant value to lower!");
Chris Lattner7e0c1642010-01-13 04:29:19 +00002158 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002159
Chris Lattner7e0c1642010-01-13 04:29:19 +00002160 switch (CE->getOpcode()) {
Dan Gohman4268d6a2010-02-08 22:02:38 +00002161 default:
2162 // If the code isn't optimized, there may be outstanding folding
Micah Villmowcdfe20b2012-10-08 16:38:25 +00002163 // opportunities. Attempt to fold the expression using DataLayout as a
Dan Gohman4268d6a2010-02-08 22:02:38 +00002164 // last resort before giving up.
David Majnemerd536f232016-07-29 03:27:26 +00002165 if (Constant *C = ConstantFoldConstant(CE, getDataLayout()))
Dan Gohmane51b57b2010-02-08 22:19:11 +00002166 if (C != CE)
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002167 return lowerConstant(C);
Dan Gohman23922872010-08-04 18:51:09 +00002168
2169 // Otherwise report the problem to the user.
2170 {
Alp Tokere69170a2014-06-26 22:52:05 +00002171 std::string S;
2172 raw_string_ostream OS(S);
Dan Gohman23922872010-08-04 18:51:09 +00002173 OS << "Unsupported expression in static initializer: ";
Chandler Carruthd48cdbf2014-01-09 02:29:41 +00002174 CE->printAsOperand(OS, /*PrintType=*/false,
Matthias Braunf1caa282017-12-15 22:22:58 +00002175 !MF ? nullptr : MF->getFunction().getParent());
Dan Gohman23922872010-08-04 18:51:09 +00002176 report_fatal_error(OS.str());
2177 }
Chris Lattner7e0c1642010-01-13 04:29:19 +00002178 case Instruction::GetElementPtr: {
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002179 // Generate a symbolic expression for the byte address
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002180 APInt OffsetAI(getDataLayout().getPointerTypeSizeInBits(CE->getType()), 0);
2181 cast<GEPOperator>(CE)->accumulateConstantOffset(getDataLayout(), OffsetAI);
Jim Grosbach1900c732011-03-29 23:20:22 +00002182
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002183 const MCExpr *Base = lowerConstant(CE->getOperand(0));
Nuno Lopesb6ad9822012-12-30 16:25:48 +00002184 if (!OffsetAI)
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002185 return Base;
Jim Grosbach1900c732011-03-29 23:20:22 +00002186
Nuno Lopesb6ad9822012-12-30 16:25:48 +00002187 int64_t Offset = OffsetAI.getSExtValue();
Jim Grosbach13760bd2015-05-30 01:25:56 +00002188 return MCBinaryExpr::createAdd(Base, MCConstantExpr::create(Offset, Ctx),
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002189 Ctx);
Chris Lattner7e0c1642010-01-13 04:29:19 +00002190 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002191
Chris Lattner7e0c1642010-01-13 04:29:19 +00002192 case Instruction::Trunc:
2193 // We emit the value and depend on the assembler to truncate the generated
2194 // expression properly. This is important for differences between
2195 // blockaddress labels. Since the two labels are in the same function, it
2196 // is reasonable to treat their delta as a 32-bit value.
Justin Bognercd1d5aa2016-08-17 20:30:52 +00002197 LLVM_FALLTHROUGH;
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002198 case Instruction::BitCast:
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002199 return lowerConstant(CE->getOperand(0));
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002200
2201 case Instruction::IntToPtr: {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002202 const DataLayout &DL = getDataLayout();
Eric Christopher8b770652015-01-26 19:03:15 +00002203
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002204 // Handle casts to pointers by changing them into casts to the appropriate
2205 // integer type. This promotes constant folding and simplifies this code.
2206 Constant *Op = CE->getOperand(0);
Matt Arsenaultb7689122013-10-21 20:03:54 +00002207 Op = ConstantExpr::getIntegerCast(Op, DL.getIntPtrType(CV->getType()),
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002208 false/*ZExt*/);
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002209 return lowerConstant(Op);
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002210 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002211
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002212 case Instruction::PtrToInt: {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002213 const DataLayout &DL = getDataLayout();
Eric Christopher8b770652015-01-26 19:03:15 +00002214
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002215 // Support only foldable casts to/from pointers that can be eliminated by
2216 // changing the pointer to the appropriately sized integer type.
2217 Constant *Op = CE->getOperand(0);
Chris Lattner229907c2011-07-18 04:54:35 +00002218 Type *Ty = CE->getType();
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002219
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002220 const MCExpr *OpExpr = lowerConstant(Op);
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002221
2222 // We can emit the pointer value into this slot if the slot is an
2223 // integer slot equal to the size of the pointer.
Shoaib Meenai87226a72019-05-23 16:29:09 +00002224 //
2225 // If the pointer is larger than the resultant integer, then
2226 // as with Trunc just depend on the assembler to truncate it.
2227 if (DL.getTypeAllocSize(Ty) <= DL.getTypeAllocSize(Op->getType()))
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002228 return OpExpr;
2229
2230 // Otherwise the pointer is smaller than the resultant integer, mask off
2231 // the high bits so we are sure to get a proper truncation if the input is
2232 // a constant expr.
Matt Arsenault40dddd72013-10-03 19:50:01 +00002233 unsigned InBits = DL.getTypeAllocSizeInBits(Op->getType());
Jim Grosbach13760bd2015-05-30 01:25:56 +00002234 const MCExpr *MaskExpr = MCConstantExpr::create(~0ULL >> (64-InBits), Ctx);
2235 return MCBinaryExpr::createAnd(OpExpr, MaskExpr, Ctx);
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002236 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002237
Peter Collingbourne265ebd72016-04-22 20:40:10 +00002238 case Instruction::Sub: {
2239 GlobalValue *LHSGV;
2240 APInt LHSOffset;
2241 if (IsConstantOffsetFromGlobal(CE->getOperand(0), LHSGV, LHSOffset,
2242 getDataLayout())) {
2243 GlobalValue *RHSGV;
2244 APInt RHSOffset;
2245 if (IsConstantOffsetFromGlobal(CE->getOperand(1), RHSGV, RHSOffset,
2246 getDataLayout())) {
Eric Christopher4367c7f2016-09-16 07:33:15 +00002247 const MCExpr *RelocExpr =
2248 getObjFileLowering().lowerRelativeReference(LHSGV, RHSGV, TM);
Peter Collingbourne265ebd72016-04-22 20:40:10 +00002249 if (!RelocExpr)
2250 RelocExpr = MCBinaryExpr::createSub(
2251 MCSymbolRefExpr::create(getSymbol(LHSGV), Ctx),
2252 MCSymbolRefExpr::create(getSymbol(RHSGV), Ctx), Ctx);
2253 int64_t Addend = (LHSOffset - RHSOffset).getSExtValue();
2254 if (Addend != 0)
2255 RelocExpr = MCBinaryExpr::createAdd(
2256 RelocExpr, MCConstantExpr::create(Addend, Ctx), Ctx);
2257 return RelocExpr;
2258 }
2259 }
2260 }
2261 // else fallthrough
Adrian Prantl0e6694d2017-12-19 22:05:25 +00002262 LLVM_FALLTHROUGH;
Peter Collingbourne265ebd72016-04-22 20:40:10 +00002263
Dan Gohman227077d2010-02-09 00:02:37 +00002264 // The MC library also has a right-shift operator, but it isn't consistently
2265 // signed or unsigned between different targets.
Chris Lattner7e0c1642010-01-13 04:29:19 +00002266 case Instruction::Add:
Dan Gohman227077d2010-02-09 00:02:37 +00002267 case Instruction::Mul:
2268 case Instruction::SDiv:
2269 case Instruction::SRem:
2270 case Instruction::Shl:
Chris Lattner7e0c1642010-01-13 04:29:19 +00002271 case Instruction::And:
2272 case Instruction::Or:
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002273 case Instruction::Xor: {
Matt Arsenaultdd3b77d2014-12-16 19:16:14 +00002274 const MCExpr *LHS = lowerConstant(CE->getOperand(0));
2275 const MCExpr *RHS = lowerConstant(CE->getOperand(1));
Chris Lattner7e0c1642010-01-13 04:29:19 +00002276 switch (CE->getOpcode()) {
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002277 default: llvm_unreachable("Unknown binary operator constant cast expr");
Jim Grosbach13760bd2015-05-30 01:25:56 +00002278 case Instruction::Add: return MCBinaryExpr::createAdd(LHS, RHS, Ctx);
2279 case Instruction::Sub: return MCBinaryExpr::createSub(LHS, RHS, Ctx);
2280 case Instruction::Mul: return MCBinaryExpr::createMul(LHS, RHS, Ctx);
2281 case Instruction::SDiv: return MCBinaryExpr::createDiv(LHS, RHS, Ctx);
2282 case Instruction::SRem: return MCBinaryExpr::createMod(LHS, RHS, Ctx);
2283 case Instruction::Shl: return MCBinaryExpr::createShl(LHS, RHS, Ctx);
2284 case Instruction::And: return MCBinaryExpr::createAnd(LHS, RHS, Ctx);
2285 case Instruction::Or: return MCBinaryExpr::createOr (LHS, RHS, Ctx);
2286 case Instruction::Xor: return MCBinaryExpr::createXor(LHS, RHS, Ctx);
Chris Lattner7e0c1642010-01-13 04:29:19 +00002287 }
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002288 }
Chris Lattner6a8e0f52004-08-16 23:15:22 +00002289 }
2290}
Chris Lattner8452a1f2004-08-17 06:36:49 +00002291
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002292static void emitGlobalConstantImpl(const DataLayout &DL, const Constant *C,
2293 AsmPrinter &AP,
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002294 const Constant *BaseCV = nullptr,
2295 uint64_t Offset = 0);
Chris Lattner51003672010-04-20 06:20:21 +00002296
Justin Bogner3135ba92015-12-08 02:37:48 +00002297static void emitGlobalConstantFP(const ConstantFP *CFP, AsmPrinter &AP);
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002298static void emitGlobalConstantFP(APFloat APF, Type *ET, AsmPrinter &AP);
Justin Bogner3135ba92015-12-08 02:37:48 +00002299
David Greenecdef71f2011-08-31 17:30:56 +00002300/// isRepeatedByteSequence - Determine whether the given value is
2301/// composed of a repeated sequence of identical bytes and return the
2302/// byte value. If it is not a repeated sequence, return -1.
Chris Lattner5d4497b2012-01-24 09:31:43 +00002303static int isRepeatedByteSequence(const ConstantDataSequential *V) {
2304 StringRef Data = V->getRawDataValues();
2305 assert(!Data.empty() && "Empty aggregates should be CAZ node");
2306 char C = Data[0];
2307 for (unsigned i = 1, e = Data.size(); i != e; ++i)
2308 if (Data[i] != C) return -1;
Chandler Carruth2c469ff2012-01-30 23:47:44 +00002309 return static_cast<uint8_t>(C); // Ensure 255 is not returned as -1.
Chris Lattner5d4497b2012-01-24 09:31:43 +00002310}
2311
Chris Lattner5d4497b2012-01-24 09:31:43 +00002312/// isRepeatedByteSequence - Determine whether the given value is
2313/// composed of a repeated sequence of identical bytes and return the
2314/// byte value. If it is not a repeated sequence, return -1.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002315static int isRepeatedByteSequence(const Value *V, const DataLayout &DL) {
David Greenecdef71f2011-08-31 17:30:56 +00002316 if (const ConstantInt *CI = dyn_cast<ConstantInt>(V)) {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002317 uint64_t Size = DL.getTypeAllocSizeInBits(V->getType());
Benjamin Kramerc6e8bfc2015-06-17 23:55:17 +00002318 assert(Size % 8 == 0);
David Greenecdef71f2011-08-31 17:30:56 +00002319
Benjamin Kramerc6e8bfc2015-06-17 23:55:17 +00002320 // Extend the element to take zero padding into account.
2321 APInt Value = CI->getValue().zextOrSelf(Size);
2322 if (!Value.isSplat(8))
2323 return -1;
David Greenecdef71f2011-08-31 17:30:56 +00002324
Benjamin Kramerc6e8bfc2015-06-17 23:55:17 +00002325 return Value.zextOrTrunc(8).getZExtValue();
David Greenecdef71f2011-08-31 17:30:56 +00002326 }
2327 if (const ConstantArray *CA = dyn_cast<ConstantArray>(V)) {
2328 // Make sure all array elements are sequences of the same repeated
2329 // byte.
Chris Lattner5d4497b2012-01-24 09:31:43 +00002330 assert(CA->getNumOperands() != 0 && "Should be a CAZ");
Benjamin Kramerc6e8bfc2015-06-17 23:55:17 +00002331 Constant *Op0 = CA->getOperand(0);
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002332 int Byte = isRepeatedByteSequence(Op0, DL);
Benjamin Kramerc6e8bfc2015-06-17 23:55:17 +00002333 if (Byte == -1)
2334 return -1;
David Greenecdef71f2011-08-31 17:30:56 +00002335
Benjamin Kramerc6e8bfc2015-06-17 23:55:17 +00002336 // All array elements must be equal.
2337 for (unsigned i = 1, e = CA->getNumOperands(); i != e; ++i)
2338 if (CA->getOperand(i) != Op0)
2339 return -1;
David Greenecdef71f2011-08-31 17:30:56 +00002340 return Byte;
2341 }
Eric Christopher7c718e42012-11-20 20:34:44 +00002342
Chris Lattner5d4497b2012-01-24 09:31:43 +00002343 if (const ConstantDataSequential *CDS = dyn_cast<ConstantDataSequential>(V))
2344 return isRepeatedByteSequence(CDS);
David Greenecdef71f2011-08-31 17:30:56 +00002345
2346 return -1;
2347}
2348
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002349static void emitGlobalConstantDataSequential(const DataLayout &DL,
2350 const ConstantDataSequential *CDS,
2351 AsmPrinter &AP) {
Chris Lattner5d4497b2012-01-24 09:31:43 +00002352 // See if we can aggregate this into a .fill, if so, emit it as such.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002353 int Value = isRepeatedByteSequence(CDS, DL);
Chris Lattner5d4497b2012-01-24 09:31:43 +00002354 if (Value != -1) {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002355 uint64_t Bytes = DL.getTypeAllocSize(CDS->getType());
Chris Lattner0d3785e2012-01-31 03:39:24 +00002356 // Don't emit a 1-byte object as a .fill.
2357 if (Bytes > 1)
Petr Hosekfaef3202016-06-01 01:59:58 +00002358 return AP.OutStreamer->emitFill(Bytes, Value);
Chris Lattner5d4497b2012-01-24 09:31:43 +00002359 }
Eric Christopher7c718e42012-11-20 20:34:44 +00002360
Chris Lattner5d4497b2012-01-24 09:31:43 +00002361 // If this can be emitted with .ascii/.asciz, emit it as such.
2362 if (CDS->isString())
Lang Hames9ff69c82015-04-24 19:11:51 +00002363 return AP.OutStreamer->EmitBytes(CDS->getAsString());
Chris Lattner5d4497b2012-01-24 09:31:43 +00002364
2365 // Otherwise, emit the values in successive locations.
2366 unsigned ElementByteSize = CDS->getElementByteSize();
2367 if (isa<IntegerType>(CDS->getElementType())) {
Chris Lattner9be59592012-01-25 01:27:20 +00002368 for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) {
Chris Lattner829400b2012-01-30 05:55:11 +00002369 if (AP.isVerbose())
Lang Hames9ff69c82015-04-24 19:11:51 +00002370 AP.OutStreamer->GetCommentOS() << format("0x%" PRIx64 "\n",
2371 CDS->getElementAsInteger(i));
2372 AP.OutStreamer->EmitIntValue(CDS->getElementAsInteger(i),
2373 ElementByteSize);
Chris Lattner5d4497b2012-01-24 09:31:43 +00002374 }
Chris Lattner51ebe142012-01-30 05:49:43 +00002375 } else {
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002376 Type *ET = CDS->getElementType();
Justin Bogner3135ba92015-12-08 02:37:48 +00002377 for (unsigned I = 0, E = CDS->getNumElements(); I != E; ++I)
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002378 emitGlobalConstantFP(CDS->getElementAsAPFloat(I), ET, AP);
Chris Lattner5d4497b2012-01-24 09:31:43 +00002379 }
2380
Matt Arsenault40dddd72013-10-03 19:50:01 +00002381 unsigned Size = DL.getTypeAllocSize(CDS->getType());
2382 unsigned EmittedSize = DL.getTypeAllocSize(CDS->getType()->getElementType()) *
Chris Lattner51ebe142012-01-30 05:49:43 +00002383 CDS->getNumElements();
Krzysztof Parzyszek5e3aba82018-05-02 17:20:22 +00002384 assert(EmittedSize <= Size && "Size cannot be less than EmittedSize!");
Chris Lattner51ebe142012-01-30 05:49:43 +00002385 if (unsigned Padding = Size - EmittedSize)
Lang Hames9ff69c82015-04-24 19:11:51 +00002386 AP.OutStreamer->EmitZeros(Padding);
Chris Lattner5d4497b2012-01-24 09:31:43 +00002387}
2388
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002389static void emitGlobalConstantArray(const DataLayout &DL,
2390 const ConstantArray *CA, AsmPrinter &AP,
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002391 const Constant *BaseCV, uint64_t Offset) {
Chris Lattnercf9e8f62012-02-05 02:29:43 +00002392 // See if we can aggregate some values. Make sure it can be
2393 // represented as a series of bytes of the constant value.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002394 int Value = isRepeatedByteSequence(CA, DL);
David Greenecdef71f2011-08-31 17:30:56 +00002395
Chris Lattnercf9e8f62012-02-05 02:29:43 +00002396 if (Value != -1) {
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002397 uint64_t Bytes = DL.getTypeAllocSize(CA->getType());
Petr Hosekfaef3202016-06-01 01:59:58 +00002398 AP.OutStreamer->emitFill(Bytes, Value);
Dan Gohman014caa42008-12-22 21:14:27 +00002399 }
Chris Lattnercf9e8f62012-02-05 02:29:43 +00002400 else {
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002401 for (unsigned i = 0, e = CA->getNumOperands(); i != e; ++i) {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002402 emitGlobalConstantImpl(DL, CA->getOperand(i), AP, BaseCV, Offset);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002403 Offset += DL.getTypeAllocSize(CA->getOperand(i)->getType());
2404 }
Chris Lattnercf9e8f62012-02-05 02:29:43 +00002405 }
Dan Gohman014caa42008-12-22 21:14:27 +00002406}
2407
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002408static void emitGlobalConstantVector(const DataLayout &DL,
2409 const ConstantVector *CV, AsmPrinter &AP) {
Chris Lattner71601e82010-01-20 07:41:15 +00002410 for (unsigned i = 0, e = CV->getType()->getNumElements(); i != e; ++i)
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002411 emitGlobalConstantImpl(DL, CV->getOperand(i), AP);
Nick Lewycky6208a2f2011-06-22 18:55:03 +00002412
Matt Arsenault40dddd72013-10-03 19:50:01 +00002413 unsigned Size = DL.getTypeAllocSize(CV->getType());
2414 unsigned EmittedSize = DL.getTypeAllocSize(CV->getType()->getElementType()) *
Nick Lewycky6208a2f2011-06-22 18:55:03 +00002415 CV->getType()->getNumElements();
2416 if (unsigned Padding = Size - EmittedSize)
Lang Hames9ff69c82015-04-24 19:11:51 +00002417 AP.OutStreamer->EmitZeros(Padding);
Dan Gohman014caa42008-12-22 21:14:27 +00002418}
2419
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002420static void emitGlobalConstantStruct(const DataLayout &DL,
2421 const ConstantStruct *CS, AsmPrinter &AP,
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002422 const Constant *BaseCV, uint64_t Offset) {
Dan Gohman014caa42008-12-22 21:14:27 +00002423 // Print the fields in successive locations. Pad to align if needed!
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002424 unsigned Size = DL.getTypeAllocSize(CS->getType());
2425 const StructLayout *Layout = DL.getStructLayout(CS->getType());
Chris Lattnerfefcb152010-01-19 19:10:44 +00002426 uint64_t SizeSoFar = 0;
2427 for (unsigned i = 0, e = CS->getNumOperands(); i != e; ++i) {
Chris Lattner71601e82010-01-20 07:41:15 +00002428 const Constant *Field = CS->getOperand(i);
Dan Gohman014caa42008-12-22 21:14:27 +00002429
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002430 // Print the actual field value.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002431 emitGlobalConstantImpl(DL, Field, AP, BaseCV, Offset + SizeSoFar);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002432
Dan Gohman014caa42008-12-22 21:14:27 +00002433 // Check if padding is needed and insert one or more 0s.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002434 uint64_t FieldSize = DL.getTypeAllocSize(Field->getType());
Chris Lattner03cb2a302010-01-20 07:11:32 +00002435 uint64_t PadSize = ((i == e-1 ? Size : Layout->getElementOffset(i+1))
2436 - Layout->getElementOffset(i)) - FieldSize;
2437 SizeSoFar += FieldSize + PadSize;
Dan Gohman014caa42008-12-22 21:14:27 +00002438
Dan Gohman014caa42008-12-22 21:14:27 +00002439 // Insert padding - this may include padding to increase the size of the
2440 // current field up to the ABI size (if the struct is not packed) as well
2441 // as padding to ensure that the next field starts at the right offset.
Lang Hames9ff69c82015-04-24 19:11:51 +00002442 AP.OutStreamer->EmitZeros(PadSize);
Dan Gohman014caa42008-12-22 21:14:27 +00002443 }
Chris Lattner03cb2a302010-01-20 07:11:32 +00002444 assert(SizeSoFar == Layout->getSizeInBytes() &&
Dan Gohman014caa42008-12-22 21:14:27 +00002445 "Layout of constant struct may be incorrect!");
2446}
2447
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002448static void emitGlobalConstantFP(APFloat APF, Type *ET, AsmPrinter &AP) {
2449 APInt API = APF.bitcastToAPInt();
Tim Northover3a51aab2013-01-11 10:36:13 +00002450
2451 // First print a comment with what we think the original floating-point value
2452 // should have been.
2453 if (AP.isVerbose()) {
2454 SmallString<8> StrVal;
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002455 APF.toString(StrVal);
Tim Northover3a51aab2013-01-11 10:36:13 +00002456
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002457 if (ET)
2458 ET->print(AP.OutStreamer->GetCommentOS());
Richard Trieuc1485222014-06-21 02:43:02 +00002459 else
Lang Hames9ff69c82015-04-24 19:11:51 +00002460 AP.OutStreamer->GetCommentOS() << "Printing <null> Type";
2461 AP.OutStreamer->GetCommentOS() << ' ' << StrVal << '\n';
Chris Lattnereaef5e12010-01-19 22:16:33 +00002462 }
Tim Northover3a51aab2013-01-11 10:36:13 +00002463
2464 // Now iterate through the APInt chunks, emitting them in endian-correct
2465 // order, possibly with a smaller chunk at beginning/end (e.g. for x87 80-bit
2466 // floats).
2467 unsigned NumBytes = API.getBitWidth() / 8;
2468 unsigned TrailingBytes = NumBytes % sizeof(uint64_t);
2469 const uint64_t *p = API.getRawData();
2470
2471 // PPC's long double has odd notions of endianness compared to how LLVM
2472 // handles it: p[0] goes first for *big* endian on PPC.
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002473 if (AP.getDataLayout().isBigEndian() && !ET->isPPC_FP128Ty()) {
Tim Northover3a51aab2013-01-11 10:36:13 +00002474 int Chunk = API.getNumWords() - 1;
2475
2476 if (TrailingBytes)
Lang Hames9ff69c82015-04-24 19:11:51 +00002477 AP.OutStreamer->EmitIntValue(p[Chunk--], TrailingBytes);
Tim Northover3a51aab2013-01-11 10:36:13 +00002478
2479 for (; Chunk >= 0; --Chunk)
Lang Hames9ff69c82015-04-24 19:11:51 +00002480 AP.OutStreamer->EmitIntValue(p[Chunk], sizeof(uint64_t));
Tim Northover3a51aab2013-01-11 10:36:13 +00002481 } else {
2482 unsigned Chunk;
2483 for (Chunk = 0; Chunk < NumBytes / sizeof(uint64_t); ++Chunk)
Lang Hames9ff69c82015-04-24 19:11:51 +00002484 AP.OutStreamer->EmitIntValue(p[Chunk], sizeof(uint64_t));
Tim Northover3a51aab2013-01-11 10:36:13 +00002485
2486 if (TrailingBytes)
Lang Hames9ff69c82015-04-24 19:11:51 +00002487 AP.OutStreamer->EmitIntValue(p[Chunk], TrailingBytes);
Tim Northover3a51aab2013-01-11 10:36:13 +00002488 }
2489
2490 // Emit the tail padding for the long double.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002491 const DataLayout &DL = AP.getDataLayout();
Alina Sbirlea08fa5942018-03-09 18:48:20 +00002492 AP.OutStreamer->EmitZeros(DL.getTypeAllocSize(ET) - DL.getTypeStoreSize(ET));
2493}
2494
2495static void emitGlobalConstantFP(const ConstantFP *CFP, AsmPrinter &AP) {
2496 emitGlobalConstantFP(CFP->getValueAPF(), CFP->getType(), AP);
Dan Gohman014caa42008-12-22 21:14:27 +00002497}
2498
Rafael Espindola64e1af82013-07-02 15:49:13 +00002499static void emitGlobalConstantLargeInt(const ConstantInt *CI, AsmPrinter &AP) {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002500 const DataLayout &DL = AP.getDataLayout();
Dan Gohman014caa42008-12-22 21:14:27 +00002501 unsigned BitWidth = CI->getBitWidth();
Quentin Colombetba366012013-06-07 18:36:03 +00002502
2503 // Copy the value as we may massage the layout for constants whose bit width
2504 // is not a multiple of 64-bits.
2505 APInt Realigned(CI->getValue());
2506 uint64_t ExtraBits = 0;
2507 unsigned ExtraBitsSize = BitWidth & 63;
2508
2509 if (ExtraBitsSize) {
2510 // The bit width of the data is not a multiple of 64-bits.
2511 // The extra bits are expected to be at the end of the chunk of the memory.
2512 // Little endian:
2513 // * Nothing to be done, just record the extra bits to emit.
2514 // Big endian:
2515 // * Record the extra bits to emit.
2516 // * Realign the raw data to emit the chunks of 64-bits.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002517 if (DL.isBigEndian()) {
Quentin Colombetba366012013-06-07 18:36:03 +00002518 // Basically the structure of the raw data is a chunk of 64-bits cells:
2519 // 0 1 BitWidth / 64
2520 // [chunk1][chunk2] ... [chunkN].
2521 // The most significant chunk is chunkN and it should be emitted first.
2522 // However, due to the alignment issue chunkN contains useless bits.
2523 // Realign the chunks so that they contain only useless information:
2524 // ExtraBits 0 1 (BitWidth / 64) - 1
2525 // chu[nk1 chu][nk2 chu] ... [nkN-1 chunkN]
2526 ExtraBits = Realigned.getRawData()[0] &
2527 (((uint64_t)-1) >> (64 - ExtraBitsSize));
Craig Topperfc947bc2017-04-18 17:14:21 +00002528 Realigned.lshrInPlace(ExtraBitsSize);
Quentin Colombetba366012013-06-07 18:36:03 +00002529 } else
2530 ExtraBits = Realigned.getRawData()[BitWidth / 64];
2531 }
Dan Gohman014caa42008-12-22 21:14:27 +00002532
2533 // We don't expect assemblers to support integer data directives
2534 // for more than 64 bits, so we emit the data in at most 64-bit
2535 // quantities at a time.
Quentin Colombetba366012013-06-07 18:36:03 +00002536 const uint64_t *RawData = Realigned.getRawData();
Dan Gohman014caa42008-12-22 21:14:27 +00002537 for (unsigned i = 0, e = BitWidth / 64; i != e; ++i) {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002538 uint64_t Val = DL.isBigEndian() ? RawData[e - i - 1] : RawData[i];
Lang Hames9ff69c82015-04-24 19:11:51 +00002539 AP.OutStreamer->EmitIntValue(Val, 8);
Dan Gohman014caa42008-12-22 21:14:27 +00002540 }
Quentin Colombetba366012013-06-07 18:36:03 +00002541
2542 if (ExtraBitsSize) {
2543 // Emit the extra bits after the 64-bits chunks.
2544
2545 // Emit a directive that fills the expected size.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002546 uint64_t Size = AP.getDataLayout().getTypeAllocSize(CI->getType());
Quentin Colombetba366012013-06-07 18:36:03 +00002547 Size -= (BitWidth / 64) * 8;
2548 assert(Size && Size * 8 >= ExtraBitsSize &&
2549 (ExtraBits & (((uint64_t)-1) >> (64 - ExtraBitsSize)))
2550 == ExtraBits && "Directive too small for extra bits.");
Lang Hames9ff69c82015-04-24 19:11:51 +00002551 AP.OutStreamer->EmitIntValue(ExtraBits, Size);
Quentin Colombetba366012013-06-07 18:36:03 +00002552 }
Dan Gohman014caa42008-12-22 21:14:27 +00002553}
2554
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00002555/// Transform a not absolute MCExpr containing a reference to a GOT
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002556/// equivalent global, by a target specific GOT pc relative access to the
2557/// final symbol.
2558static void handleIndirectSymViaGOTPCRel(AsmPrinter &AP, const MCExpr **ME,
2559 const Constant *BaseCst,
2560 uint64_t Offset) {
2561 // The global @foo below illustrates a global that uses a got equivalent.
2562 //
2563 // @bar = global i32 42
2564 // @gotequiv = private unnamed_addr constant i32* @bar
2565 // @foo = i32 trunc (i64 sub (i64 ptrtoint (i32** @gotequiv to i64),
2566 // i64 ptrtoint (i32* @foo to i64))
2567 // to i32)
2568 //
2569 // The cstexpr in @foo is converted into the MCExpr `ME`, where we actually
2570 // check whether @foo is suitable to use a GOTPCREL. `ME` is usually in the
2571 // form:
2572 //
2573 // foo = cstexpr, where
2574 // cstexpr := <gotequiv> - "." + <cst>
2575 // cstexpr := <gotequiv> - (<foo> - <offset from @foo base>) + <cst>
2576 //
Jim Grosbach13760bd2015-05-30 01:25:56 +00002577 // After canonicalization by evaluateAsRelocatable `ME` turns into:
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002578 //
2579 // cstexpr := <gotequiv> - <foo> + gotpcrelcst, where
2580 // gotpcrelcst := <offset from @foo base> + <cst>
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002581 MCValue MV;
Jim Grosbach13760bd2015-05-30 01:25:56 +00002582 if (!(*ME)->evaluateAsRelocatable(MV, nullptr, nullptr) || MV.isAbsolute())
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002583 return;
Bruno Cardoso Lopesedb876d2015-06-25 15:17:23 +00002584 const MCSymbolRefExpr *SymA = MV.getSymA();
2585 if (!SymA)
2586 return;
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002587
Bruno Cardoso Lopesedb876d2015-06-25 15:17:23 +00002588 // Check that GOT equivalent symbol is cached.
2589 const MCSymbol *GOTEquivSym = &SymA->getSymbol();
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002590 if (!AP.GlobalGOTEquivs.count(GOTEquivSym))
2591 return;
2592
Bruno Cardoso Lopese8640512015-07-21 21:45:42 +00002593 const GlobalValue *BaseGV = dyn_cast_or_null<GlobalValue>(BaseCst);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002594 if (!BaseGV)
2595 return;
2596
Bruno Cardoso Lopesedb876d2015-06-25 15:17:23 +00002597 // Check for a valid base symbol
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002598 const MCSymbol *BaseSym = AP.getSymbol(BaseGV);
Bruno Cardoso Lopesedb876d2015-06-25 15:17:23 +00002599 const MCSymbolRefExpr *SymB = MV.getSymB();
2600
2601 if (!SymB || BaseSym != &SymB->getSymbol())
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002602 return;
2603
2604 // Make sure to match:
2605 //
2606 // gotpcrelcst := <offset from @foo base> + <cst>
2607 //
Bruno Cardoso Lopes618c67a2015-03-06 13:49:05 +00002608 // If gotpcrelcst is positive it means that we can safely fold the pc rel
2609 // displacement into the GOTPCREL. We can also can have an extra offset <cst>
2610 // if the target knows how to encode it.
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002611 int64_t GOTPCRelCst = Offset + MV.getConstant();
2612 if (GOTPCRelCst < 0)
2613 return;
Bruno Cardoso Lopes52b13912015-03-06 13:48:45 +00002614 if (!AP.getObjFileLowering().supportGOTPCRelWithOffset() && GOTPCRelCst != 0)
2615 return;
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002616
2617 // Emit the GOT PC relative to replace the got equivalent global, i.e.:
2618 //
2619 // bar:
2620 // .long 42
2621 // gotequiv:
2622 // .quad bar
2623 // foo:
2624 // .long gotequiv - "." + <cst>
2625 //
2626 // is replaced by the target specific equivalent to:
2627 //
2628 // bar:
2629 // .long 42
2630 // foo:
2631 // .long bar@GOTPCREL+<gotpcrelcst>
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002632 AsmPrinter::GOTEquivUsePair Result = AP.GlobalGOTEquivs[GOTEquivSym];
2633 const GlobalVariable *GV = Result.first;
Bruno Cardoso Lopesb3a58b42015-03-10 20:05:23 +00002634 int NumUses = (int)Result.second;
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002635 const GlobalValue *FinalGV = dyn_cast<GlobalValue>(GV->getOperand(0));
2636 const MCSymbol *FinalSym = AP.getSymbol(FinalGV);
Bruno Cardoso Lopes618c67a2015-03-06 13:49:05 +00002637 *ME = AP.getObjFileLowering().getIndirectSymViaGOTPCRel(
Francis Visoiu Mistrih5b5ee612019-08-22 16:59:00 +00002638 FinalGV, FinalSym, MV, Offset, AP.MMI, *AP.OutStreamer);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002639
2640 // Update GOT equivalent usage information
2641 --NumUses;
Bruno Cardoso Lopesb3a58b42015-03-10 20:05:23 +00002642 if (NumUses >= 0)
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002643 AP.GlobalGOTEquivs[GOTEquivSym] = std::make_pair(GV, NumUses);
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002644}
2645
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002646static void emitGlobalConstantImpl(const DataLayout &DL, const Constant *CV,
2647 AsmPrinter &AP, const Constant *BaseCV,
2648 uint64_t Offset) {
2649 uint64_t Size = DL.getTypeAllocSize(CV->getType());
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002650
2651 // Globals with sub-elements such as combinations of arrays and structs
2652 // are handled recursively by emitGlobalConstantImpl. Keep track of the
2653 // constant symbol base and the current position with BaseCV and Offset.
2654 if (!BaseCV && CV->hasOneUse())
2655 BaseCV = dyn_cast<Constant>(CV->user_back());
2656
Bill Wendling7315c4b2012-03-20 08:56:43 +00002657 if (isa<ConstantAggregateZero>(CV) || isa<UndefValue>(CV))
Lang Hames9ff69c82015-04-24 19:11:51 +00002658 return AP.OutStreamer->EmitZeros(Size);
Chris Lattner03cb2a302010-01-20 07:11:32 +00002659
2660 if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
Chris Lattner03cb2a302010-01-20 07:11:32 +00002661 switch (Size) {
2662 case 1:
2663 case 2:
2664 case 4:
2665 case 8:
Chris Lattner51003672010-04-20 06:20:21 +00002666 if (AP.isVerbose())
Lang Hames9ff69c82015-04-24 19:11:51 +00002667 AP.OutStreamer->GetCommentOS() << format("0x%" PRIx64 "\n",
2668 CI->getZExtValue());
2669 AP.OutStreamer->EmitIntValue(CI->getZExtValue(), Size);
Chris Lattner03cb2a302010-01-20 07:11:32 +00002670 return;
2671 default:
Rafael Espindola64e1af82013-07-02 15:49:13 +00002672 emitGlobalConstantLargeInt(CI, AP);
Chris Lattner03cb2a302010-01-20 07:11:32 +00002673 return;
2674 }
2675 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002676
Chris Lattnerfefcb152010-01-19 19:10:44 +00002677 if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV))
Rafael Espindola64e1af82013-07-02 15:49:13 +00002678 return emitGlobalConstantFP(CFP, AP);
Chris Lattner8452a1f2004-08-17 06:36:49 +00002679
Chris Lattnerc47ece52010-01-20 17:57:50 +00002680 if (isa<ConstantPointerNull>(CV)) {
Lang Hames9ff69c82015-04-24 19:11:51 +00002681 AP.OutStreamer->EmitIntValue(0, Size);
Chris Lattnerc47ece52010-01-20 17:57:50 +00002682 return;
2683 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002684
Chris Lattner5d4497b2012-01-24 09:31:43 +00002685 if (const ConstantDataSequential *CDS = dyn_cast<ConstantDataSequential>(CV))
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002686 return emitGlobalConstantDataSequential(DL, CDS, AP);
Eric Christopher7c718e42012-11-20 20:34:44 +00002687
Chris Lattner5d4497b2012-01-24 09:31:43 +00002688 if (const ConstantArray *CVA = dyn_cast<ConstantArray>(CV))
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002689 return emitGlobalConstantArray(DL, CVA, AP, BaseCV, Offset);
Jim Grosbach1900c732011-03-29 23:20:22 +00002690
Chris Lattner5d4497b2012-01-24 09:31:43 +00002691 if (const ConstantStruct *CVS = dyn_cast<ConstantStruct>(CV))
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002692 return emitGlobalConstantStruct(DL, CVS, AP, BaseCV, Offset);
Chris Lattner5d4497b2012-01-24 09:31:43 +00002693
Bill Wendling7315c4b2012-03-20 08:56:43 +00002694 if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
2695 // Look through bitcasts, which might not be able to be MCExpr'ized (e.g. of
2696 // vectors).
Chris Lattner5d4497b2012-01-24 09:31:43 +00002697 if (CE->getOpcode() == Instruction::BitCast)
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002698 return emitGlobalConstantImpl(DL, CE->getOperand(0), AP);
Bill Wendling7315c4b2012-03-20 08:56:43 +00002699
2700 if (Size > 8) {
2701 // If the constant expression's size is greater than 64-bits, then we have
2702 // to emit the value in chunks. Try to constant fold the value and emit it
2703 // that way.
David Majnemerd536f232016-07-29 03:27:26 +00002704 Constant *New = ConstantFoldConstant(CE, DL);
Bill Wendling7315c4b2012-03-20 08:56:43 +00002705 if (New && New != CE)
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002706 return emitGlobalConstantImpl(DL, New, AP);
Bill Wendling7315c4b2012-03-20 08:56:43 +00002707 }
2708 }
Eric Christopher7c718e42012-11-20 20:34:44 +00002709
Chris Lattner5d4497b2012-01-24 09:31:43 +00002710 if (const ConstantVector *V = dyn_cast<ConstantVector>(CV))
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002711 return emitGlobalConstantVector(DL, V, AP);
Eric Christopher7c718e42012-11-20 20:34:44 +00002712
Chris Lattnerd1acffc2010-01-23 06:17:14 +00002713 // Otherwise, it must be a ConstantExpr. Lower it to an MCExpr, then emit it
2714 // thread the streamer with EmitValue.
Bruno Cardoso Lopes24492b02015-02-23 21:26:18 +00002715 const MCExpr *ME = AP.lowerConstant(CV);
2716
2717 // Since lowerConstant already folded and got rid of all IR pointer and
2718 // integer casts, detect GOT equivalent accesses by looking into the MCExpr
2719 // directly.
2720 if (AP.getObjFileLowering().supportIndirectSymViaGOTPCRel())
2721 handleIndirectSymViaGOTPCRel(AP, &ME, BaseCV, Offset);
2722
Lang Hames9ff69c82015-04-24 19:11:51 +00002723 AP.OutStreamer->EmitValue(ME, Size);
Chris Lattner51003672010-04-20 06:20:21 +00002724}
2725
2726/// EmitGlobalConstant - Print a general LLVM constant to the .s file.
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002727void AsmPrinter::EmitGlobalConstant(const DataLayout &DL, const Constant *CV) {
2728 uint64_t Size = DL.getTypeAllocSize(CV->getType());
Chris Lattner51003672010-04-20 06:20:21 +00002729 if (Size)
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002730 emitGlobalConstantImpl(DL, CV, *this);
Chris Lattner51003672010-04-20 06:20:21 +00002731 else if (MAI->hasSubsectionsViaSymbols()) {
2732 // If the global has zero size, emit a single byte so that two labels don't
2733 // look like they are at the same location.
Lang Hames9ff69c82015-04-24 19:11:51 +00002734 OutStreamer->EmitIntValue(0, 1);
Chris Lattner51003672010-04-20 06:20:21 +00002735 }
Chris Lattner8452a1f2004-08-17 06:36:49 +00002736}
Chris Lattner061d9e22006-01-27 02:10:10 +00002737
Chris Lattner17f71652008-08-17 07:19:36 +00002738void AsmPrinter::EmitMachineConstantPoolValue(MachineConstantPoolValue *MCPV) {
Evan Cheng45fe3bc2006-09-12 21:00:35 +00002739 // Target doesn't support this yet!
Torok Edwinfbcc6632009-07-14 16:55:14 +00002740 llvm_unreachable("Target does not support EmitMachineConstantPoolValue");
Evan Cheng45fe3bc2006-09-12 21:00:35 +00002741}
2742
Chris Lattnerc334d802010-04-04 18:16:38 +00002743void AsmPrinter::printOffset(int64_t Offset, raw_ostream &OS) const {
2744 if (Offset > 0)
2745 OS << '+' << Offset;
2746 else if (Offset < 0)
2747 OS << Offset;
2748}
2749
Chris Lattnera179b522010-04-04 19:25:43 +00002750//===----------------------------------------------------------------------===//
2751// Symbol Lowering Routines.
2752//===----------------------------------------------------------------------===//
2753
Rafael Espindola9ab09232015-03-17 20:07:06 +00002754MCSymbol *AsmPrinter::createTempSymbol(const Twine &Name) const {
2755 return OutContext.createTempSymbol(Name, true);
Rafael Espindola629cdba2015-02-27 18:18:39 +00002756}
Chris Lattneraa23fa92006-02-01 22:41:11 +00002757
Chris Lattnerebc970e2010-02-08 23:10:08 +00002758MCSymbol *AsmPrinter::GetBlockAddressSymbol(const BlockAddress *BA) const {
Chris Lattner9efbbcb2010-03-14 17:53:23 +00002759 return MMI->getAddrLabelSymbol(BA->getBasicBlock());
Dan Gohman6c938802009-10-30 01:27:03 +00002760}
2761
Chris Lattner9efbbcb2010-03-14 17:53:23 +00002762MCSymbol *AsmPrinter::GetBlockAddressSymbol(const BasicBlock *BB) const {
2763 return MMI->getAddrLabelSymbol(BB);
Dan Gohman6c938802009-10-30 01:27:03 +00002764}
2765
Chris Lattner4225a7b2010-01-23 05:19:23 +00002766/// GetCPISymbol - Return the symbol for the specified constant pool entry.
2767MCSymbol *AsmPrinter::GetCPISymbol(unsigned CPID) const {
Reid Kleckner2f07c2e2019-07-08 21:05:20 +00002768 if (getSubtargetInfo().getTargetTriple().isWindowsMSVCEnvironment()) {
Martin Storsjod2662c32018-07-25 18:35:31 +00002769 const MachineConstantPoolEntry &CPE =
2770 MF->getConstantPool()->getConstants()[CPID];
2771 if (!CPE.isMachineConstantPoolEntry()) {
2772 const DataLayout &DL = MF->getDataLayout();
2773 SectionKind Kind = CPE.getSectionKind(&DL);
2774 const Constant *C = CPE.Val.ConstVal;
2775 unsigned Align = CPE.Alignment;
2776 if (const MCSectionCOFF *S = dyn_cast<MCSectionCOFF>(
2777 getObjFileLowering().getSectionForConstant(DL, Kind, C, Align))) {
2778 if (MCSymbol *Sym = S->getCOMDATSymbol()) {
2779 if (Sym->isUndefined())
2780 OutStreamer->EmitSymbolAttribute(Sym, MCSA_Global);
2781 return Sym;
2782 }
2783 }
2784 }
2785 }
2786
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002787 const DataLayout &DL = getDataLayout();
2788 return OutContext.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) +
2789 "CPI" + Twine(getFunctionNumber()) + "_" +
2790 Twine(CPID));
Chris Lattner4225a7b2010-01-23 05:19:23 +00002791}
2792
2793/// GetJTISymbol - Return the symbol for the specified jump table entry.
Bill Wendling36321712010-06-29 22:34:52 +00002794MCSymbol *AsmPrinter::GetJTISymbol(unsigned JTID, bool isLinkerPrivate) const {
2795 return MF->getJTISymbol(JTID, OutContext, isLinkerPrivate);
Chris Lattnera602beb2009-09-12 23:02:08 +00002796}
2797
Chris Lattner00d188f2010-01-25 21:17:10 +00002798/// GetJTSetSymbol - Return the symbol for the specified jump table .set
2799/// FIXME: privatize to AsmPrinter.
2800MCSymbol *AsmPrinter::GetJTSetSymbol(unsigned UID, unsigned MBBID) const {
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002801 const DataLayout &DL = getDataLayout();
2802 return OutContext.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) +
2803 Twine(getFunctionNumber()) + "_" +
2804 Twine(UID) + "_set_" + Twine(MBBID));
Chris Lattner00d188f2010-01-25 21:17:10 +00002805}
2806
Rafael Espindolaf4e6b292013-12-02 16:25:47 +00002807MCSymbol *AsmPrinter::getSymbolWithGlobalValueBase(const GlobalValue *GV,
Rafael Espindola61b3d0c2013-11-28 19:35:07 +00002808 StringRef Suffix) const {
Eric Christopher4367c7f2016-09-16 07:33:15 +00002809 return getObjFileLowering().getSymbolWithGlobalValueBase(GV, Suffix, TM);
Chris Lattner0dca0a92010-01-15 23:25:11 +00002810}
2811
Rafael Espindolac233f742015-06-23 13:59:29 +00002812/// Return the MCSymbol for the specified ExternalSymbol.
Chris Lattner91687f62010-01-15 23:18:17 +00002813MCSymbol *AsmPrinter::GetExternalSymbolSymbol(StringRef Sym) const {
2814 SmallString<60> NameStr;
Mehdi Aminibd7287e2015-07-16 06:11:10 +00002815 Mangler::getNameWithPrefix(NameStr, Sym, getDataLayout());
Jim Grosbach6f482002015-05-18 18:43:14 +00002816 return OutContext.getOrCreateSymbol(NameStr);
Jim Grosbach1900c732011-03-29 23:20:22 +00002817}
Chris Lattner91687f62010-01-15 23:18:17 +00002818
Chris Lattner978e2f02010-01-22 21:50:41 +00002819/// PrintParentLoopComment - Print comments about parent loops of this one.
2820static void PrintParentLoopComment(raw_ostream &OS, const MachineLoop *Loop,
2821 unsigned FunctionNumber) {
Craig Topper353eda42014-04-24 06:44:33 +00002822 if (!Loop) return;
Chris Lattner978e2f02010-01-22 21:50:41 +00002823 PrintParentLoopComment(OS, Loop->getParentLoop(), FunctionNumber);
2824 OS.indent(Loop->getLoopDepth()*2)
2825 << "Parent Loop BB" << FunctionNumber << "_"
2826 << Loop->getHeader()->getNumber()
2827 << " Depth=" << Loop->getLoopDepth() << '\n';
2828}
2829
Chris Lattner978e2f02010-01-22 21:50:41 +00002830/// PrintChildLoopComment - Print comments about child loops within
2831/// the loop for this basic block, with nesting.
2832static void PrintChildLoopComment(raw_ostream &OS, const MachineLoop *Loop,
2833 unsigned FunctionNumber) {
2834 // Add child loop information
Jim Grosbach6623e7f2014-04-16 22:38:02 +00002835 for (const MachineLoop *CL : *Loop) {
2836 OS.indent(CL->getLoopDepth()*2)
Chris Lattner978e2f02010-01-22 21:50:41 +00002837 << "Child Loop BB" << FunctionNumber << "_"
Jim Grosbach6623e7f2014-04-16 22:38:02 +00002838 << CL->getHeader()->getNumber() << " Depth " << CL->getLoopDepth()
Chris Lattner978e2f02010-01-22 21:50:41 +00002839 << '\n';
Jim Grosbach6623e7f2014-04-16 22:38:02 +00002840 PrintChildLoopComment(OS, CL, FunctionNumber);
Chris Lattner978e2f02010-01-22 21:50:41 +00002841 }
2842}
2843
Chad Rosierb759ede2012-09-07 18:16:38 +00002844/// emitBasicBlockLoopComments - Pretty-print comments for basic blocks.
2845static void emitBasicBlockLoopComments(const MachineBasicBlock &MBB,
Chris Lattnerc334d802010-04-04 18:16:38 +00002846 const MachineLoopInfo *LI,
2847 const AsmPrinter &AP) {
Chris Lattner978e2f02010-01-22 21:50:41 +00002848 // Add loop depth information
2849 const MachineLoop *Loop = LI->getLoopFor(&MBB);
Craig Topper353eda42014-04-24 06:44:33 +00002850 if (!Loop) return;
Jim Grosbach1900c732011-03-29 23:20:22 +00002851
Chris Lattner978e2f02010-01-22 21:50:41 +00002852 MachineBasicBlock *Header = Loop->getHeader();
2853 assert(Header && "No header for loop");
Jim Grosbach1900c732011-03-29 23:20:22 +00002854
Chris Lattner978e2f02010-01-22 21:50:41 +00002855 // If this block is not a loop header, just print out what is the loop header
2856 // and return.
2857 if (Header != &MBB) {
Lang Hames9ff69c82015-04-24 19:11:51 +00002858 AP.OutStreamer->AddComment(" in Loop: Header=BB" +
2859 Twine(AP.getFunctionNumber())+"_" +
2860 Twine(Loop->getHeader()->getNumber())+
2861 " Depth="+Twine(Loop->getLoopDepth()));
Chris Lattner978e2f02010-01-22 21:50:41 +00002862 return;
2863 }
Jim Grosbach1900c732011-03-29 23:20:22 +00002864
Chris Lattner978e2f02010-01-22 21:50:41 +00002865 // Otherwise, it is a loop header. Print out information about child and
2866 // parent loops.
Lang Hames9ff69c82015-04-24 19:11:51 +00002867 raw_ostream &OS = AP.OutStreamer->GetCommentOS();
Jim Grosbach1900c732011-03-29 23:20:22 +00002868
2869 PrintParentLoopComment(OS, Loop->getParentLoop(), AP.getFunctionNumber());
2870
Chris Lattner978e2f02010-01-22 21:50:41 +00002871 OS << "=>";
2872 OS.indent(Loop->getLoopDepth()*2-2);
Jim Grosbach1900c732011-03-29 23:20:22 +00002873
Chris Lattner978e2f02010-01-22 21:50:41 +00002874 OS << "This ";
2875 if (Loop->empty())
2876 OS << "Inner ";
2877 OS << "Loop Header: Depth=" + Twine(Loop->getLoopDepth()) << '\n';
Jim Grosbach1900c732011-03-29 23:20:22 +00002878
Chris Lattner978e2f02010-01-22 21:50:41 +00002879 PrintChildLoopComment(OS, Loop, AP.getFunctionNumber());
2880}
2881
Omer Paparo Bivas2251c792017-10-24 06:16:03 +00002882void AsmPrinter::setupCodePaddingContext(const MachineBasicBlock &MBB,
2883 MCCodePaddingContext &Context) const {
2884 assert(MF != nullptr && "Machine function must be valid");
Omer Paparo Bivas2251c792017-10-24 06:16:03 +00002885 Context.IsPaddingActive = !MF->hasInlineAsm() &&
Evandro Menezes85bd3972019-04-04 22:40:06 +00002886 !MF->getFunction().hasOptSize() &&
Omer Paparo Bivas2251c792017-10-24 06:16:03 +00002887 TM.getOptLevel() != CodeGenOpt::None;
Omer Paparo Bivas2251c792017-10-24 06:16:03 +00002888 Context.IsBasicBlockReachableViaFallthrough =
2889 std::find(MBB.pred_begin(), MBB.pred_end(), MBB.getPrevNode()) !=
2890 MBB.pred_end();
2891 Context.IsBasicBlockReachableViaBranch =
2892 MBB.pred_size() > 0 && !isBlockOnlyReachableByFallthrough(&MBB);
2893}
2894
Chris Lattner1d04aa72009-09-13 18:25:37 +00002895/// EmitBasicBlockStart - This method prints the label for the specified
2896/// MachineBasicBlock, an alignment (if present) and a comment describing
2897/// it if appropriate.
Karl-Johan Karlsson40da6be2019-08-20 05:13:57 +00002898void AsmPrinter::EmitBasicBlockStart(const MachineBasicBlock &MBB) {
David Majnemera80c1512015-09-29 20:12:33 +00002899 // End the previous funclet and start a new one.
2900 if (MBB.isEHFuncletEntry()) {
2901 for (const HandlerInfo &HI : Handlers) {
2902 HI.Handler->endFunclet();
2903 HI.Handler->beginFunclet(MBB);
2904 }
2905 }
2906
Dan Gohmanc83dc4b2009-10-30 01:34:35 +00002907 // Emit an alignment directive for this block, if needed.
Guillaume Chateletaff45e42019-09-05 10:00:22 +00002908 if (unsigned LogAlign = MBB.getLogAlignment())
2909 EmitAlignment(LogAlign);
Omer Paparo Bivas2251c792017-10-24 06:16:03 +00002910 MCCodePaddingContext Context;
2911 setupCodePaddingContext(MBB, Context);
2912 OutStreamer->EmitCodePaddingBasicBlockStart(Context);
Evan Chengc7990652008-02-28 00:43:03 +00002913
Chris Lattnerdb035a02010-03-16 00:29:39 +00002914 // If the block has its address taken, emit any labels that were used to
2915 // reference the block. It is possible that there is more than one label
2916 // here, because multiple LLVM BB's may have been RAUW'd to this block after
2917 // the references were generated.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00002918 if (MBB.hasAddressTaken()) {
2919 const BasicBlock *BB = MBB.getBasicBlock();
Chris Lattner7bde8c02010-04-04 18:52:31 +00002920 if (isVerbose())
Lang Hames9ff69c82015-04-24 19:11:51 +00002921 OutStreamer->AddComment("Block address taken");
Jim Grosbach1900c732011-03-29 23:20:22 +00002922
Joseph Tremoulet3d0fbf12015-10-23 15:06:05 +00002923 // MBBs can have their address taken as part of CodeGen without having
2924 // their corresponding BB's address taken in IR
2925 if (BB->hasAddressTaken())
2926 for (MCSymbol *Sym : MMI->getAddrLabelSymbolToEmit(BB))
2927 OutStreamer->EmitLabel(Sym);
Dan Gohman6c938802009-10-30 01:27:03 +00002928 }
2929
Jakob Stoklund Olesen936656b2012-03-06 22:27:13 +00002930 // Print some verbose block comments.
2931 if (isVerbose()) {
David Majnemer11234ed2015-12-25 09:37:26 +00002932 if (const BasicBlock *BB = MBB.getBasicBlock()) {
2933 if (BB->hasName()) {
2934 BB->printAsOperand(OutStreamer->GetCommentOS(),
2935 /*PrintType=*/false, BB->getModule());
2936 OutStreamer->GetCommentOS() << '\n';
2937 }
2938 }
Michael Zolotukhin8d052a02018-04-09 00:54:47 +00002939
2940 assert(MLI != nullptr && "MachineLoopInfo should has been computed");
2941 emitBasicBlockLoopComments(MBB, MLI, *this);
Jakob Stoklund Olesen936656b2012-03-06 22:27:13 +00002942 }
2943
Dan Gohmanc83dc4b2009-10-30 01:34:35 +00002944 // Print the main label for the block.
Reid Kleckner0e288232015-08-27 23:27:47 +00002945 if (MBB.pred_empty() ||
Alex Bradburyda20f5c2019-04-01 14:42:56 +00002946 (isBlockOnlyReachableByFallthrough(&MBB) && !MBB.isEHFuncletEntry() &&
2947 !MBB.hasLabelMustBeEmitted())) {
Rafael Espindola499a7482014-01-31 23:14:01 +00002948 if (isVerbose()) {
Chris Lattner17167212010-04-03 22:12:35 +00002949 // NOTE: Want this comment at start of line, don't emit with AddComment.
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +00002950 OutStreamer->emitRawComment(" %bb." + Twine(MBB.getNumber()) + ":",
2951 false);
Chris Lattnera3eee3c2010-01-22 19:52:01 +00002952 }
Dan Gohman10d3dc52009-10-06 17:38:38 +00002953 } else {
Alex Bradburyda20f5c2019-04-01 14:42:56 +00002954 if (isVerbose() && MBB.hasLabelMustBeEmitted())
2955 OutStreamer->AddComment("Label of block must be emitted");
Lang Hames9ff69c82015-04-24 19:11:51 +00002956 OutStreamer->EmitLabel(MBB.getSymbol());
Dan Gohman10d3dc52009-10-06 17:38:38 +00002957 }
Nate Begeman4ca2ea52006-04-22 18:53:45 +00002958}
Nate Begeman984c1a42006-08-12 21:29:52 +00002959
Omer Paparo Bivas2251c792017-10-24 06:16:03 +00002960void AsmPrinter::EmitBasicBlockEnd(const MachineBasicBlock &MBB) {
2961 MCCodePaddingContext Context;
2962 setupCodePaddingContext(MBB, Context);
2963 OutStreamer->EmitCodePaddingBasicBlockEnd(Context);
2964}
2965
Stuart Hastingsbf836592011-02-23 02:27:05 +00002966void AsmPrinter::EmitVisibility(MCSymbol *Sym, unsigned Visibility,
2967 bool IsDefinition) const {
Chris Lattner0bfd2792010-01-23 06:53:23 +00002968 MCSymbolAttr Attr = MCSA_Invalid;
Jim Grosbach1900c732011-03-29 23:20:22 +00002969
Chris Lattner0bfd2792010-01-23 06:53:23 +00002970 switch (Visibility) {
2971 default: break;
2972 case GlobalValue::HiddenVisibility:
Stuart Hastingsbf836592011-02-23 02:27:05 +00002973 if (IsDefinition)
2974 Attr = MAI->getHiddenVisibilityAttr();
2975 else
2976 Attr = MAI->getHiddenDeclarationVisibilityAttr();
Chris Lattner0bfd2792010-01-23 06:53:23 +00002977 break;
2978 case GlobalValue::ProtectedVisibility:
2979 Attr = MAI->getProtectedVisibilityAttr();
2980 break;
Chris Lattner7fec1802010-01-15 23:38:51 +00002981 }
Chris Lattner0bfd2792010-01-23 06:53:23 +00002982
2983 if (Attr != MCSA_Invalid)
Lang Hames9ff69c82015-04-24 19:11:51 +00002984 OutStreamer->EmitSymbolAttribute(Sym, Attr);
Chris Lattner7fec1802010-01-15 23:38:51 +00002985}
2986
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00002987/// isBlockOnlyReachableByFallthough - Return true if the basic block has
2988/// exactly one predecessor and the control transfer mechanism between
2989/// the predecessor and this block is a fall-through.
Chris Lattner2cf5f9e2010-04-04 17:57:56 +00002990bool AsmPrinter::
2991isBlockOnlyReachableByFallthrough(const MachineBasicBlock *MBB) const {
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00002992 // If this is a landing pad, it isn't a fall through. If it has no preds,
2993 // then nothing falls through to it.
Reid Kleckner0e288232015-08-27 23:27:47 +00002994 if (MBB->isEHPad() || MBB->pred_empty())
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00002995 return false;
Jim Grosbach1900c732011-03-29 23:20:22 +00002996
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00002997 // If there isn't exactly one predecessor, it can't be a fall through.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00002998 if (MBB->pred_size() > 1)
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00002999 return false;
Jim Grosbach1900c732011-03-29 23:20:22 +00003000
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00003001 // The predecessor has to be immediately before this block.
Jim Grosbach6623e7f2014-04-16 22:38:02 +00003002 MachineBasicBlock *Pred = *MBB->pred_begin();
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00003003 if (!Pred->isLayoutSuccessor(MBB))
3004 return false;
Jim Grosbach1900c732011-03-29 23:20:22 +00003005
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00003006 // If the block is completely empty, then it definitely does fall through.
3007 if (Pred->empty())
3008 return true;
Jim Grosbach1900c732011-03-29 23:20:22 +00003009
Eli Friedman8a3264a2011-06-14 19:30:33 +00003010 // Check the terminators in the previous blocks
Jim Grosbach6623e7f2014-04-16 22:38:02 +00003011 for (const auto &MI : Pred->terminators()) {
Eli Friedman8a3264a2011-06-14 19:30:33 +00003012 // If it is not a simple branch, we are in a table somewhere.
Evan Cheng7f8e5632011-12-07 07:15:52 +00003013 if (!MI.isBranch() || MI.isIndirectBranch())
Eli Friedman8a3264a2011-06-14 19:30:33 +00003014 return false;
3015
Jakob Stoklund Olesen1995b9f2014-01-12 19:24:08 +00003016 // If we are the operands of one of the branches, this is not a fall
3017 // through. Note that targets with delay slots will usually bundle
3018 // terminators with the delay slot instruction.
Duncan P. N. Exon Smithf9ab4162016-02-27 17:05:33 +00003019 for (ConstMIBundleOperands OP(MI); OP.isValid(); ++OP) {
Jakob Stoklund Olesen1995b9f2014-01-12 19:24:08 +00003020 if (OP->isJTI())
Rafael Espindolaab205672011-06-15 21:00:28 +00003021 return false;
Jakob Stoklund Olesen1995b9f2014-01-12 19:24:08 +00003022 if (OP->isMBB() && OP->getMBB() == MBB)
Eli Friedman8a3264a2011-06-14 19:30:33 +00003023 return false;
3024 }
3025 }
3026
3027 return true;
Chris Lattner1fa9c2cc2010-02-17 18:52:56 +00003028}
3029
David Blaikie88368ba2014-04-15 05:34:49 +00003030GCMetadataPrinter *AsmPrinter::GetOrCreateGCPrinter(GCStrategy &S) {
3031 if (!S.usesMetadata())
Craig Topper353eda42014-04-24 06:44:33 +00003032 return nullptr;
Chris Lattner2cf5f9e2010-04-04 17:57:56 +00003033
3034 gcp_map_type &GCMap = getGCMap(GCMetadataPrinters);
David Blaikie88368ba2014-04-15 05:34:49 +00003035 gcp_map_type::iterator GCPI = GCMap.find(&S);
Chris Lattner2cf5f9e2010-04-04 17:57:56 +00003036 if (GCPI != GCMap.end())
David Blaikieec528ee2014-04-15 05:53:26 +00003037 return GCPI->second.get();
Jim Grosbach1900c732011-03-29 23:20:22 +00003038
Mehdi Amini74107172016-10-01 15:44:54 +00003039 auto Name = S.getName();
Jim Grosbach1900c732011-03-29 23:20:22 +00003040
Gordon Henriksendbe06d32008-08-17 12:08:44 +00003041 for (GCMetadataPrinterRegistry::iterator
3042 I = GCMetadataPrinterRegistry::begin(),
3043 E = GCMetadataPrinterRegistry::end(); I != E; ++I)
Mehdi Amini74107172016-10-01 15:44:54 +00003044 if (Name == I->getName()) {
David Blaikieec528ee2014-04-15 05:53:26 +00003045 std::unique_ptr<GCMetadataPrinter> GMP = I->instantiate();
David Blaikie88368ba2014-04-15 05:34:49 +00003046 GMP->S = &S;
David Blaikieec528ee2014-04-15 05:53:26 +00003047 auto IterBool = GCMap.insert(std::make_pair(&S, std::move(GMP)));
3048 return IterBool.first->second.get();
Gordon Henriksendbe06d32008-08-17 12:08:44 +00003049 }
Jim Grosbach1900c732011-03-29 23:20:22 +00003050
Chris Lattner2104b8d2010-04-07 22:58:41 +00003051 report_fatal_error("no GCMetadataPrinter registered for GC: " + Twine(Name));
Gordon Henriksendbe06d32008-08-17 12:08:44 +00003052}
Juergen Ributzkadb9ee002013-12-14 12:23:14 +00003053
Than McIntosh30c804b2018-11-26 18:43:48 +00003054void AsmPrinter::emitStackMaps(StackMaps &SM) {
3055 GCModuleInfo *MI = getAnalysisIfAvailable<GCModuleInfo>();
3056 assert(MI && "AsmPrinter didn't require GCModuleInfo?");
3057 bool NeedsDefault = false;
3058 if (MI->begin() == MI->end())
3059 // No GC strategy, use the default format.
3060 NeedsDefault = true;
3061 else
3062 for (auto &I : *MI) {
3063 if (GCMetadataPrinter *MP = GetOrCreateGCPrinter(*I))
3064 if (MP->emitStackMaps(SM, *this))
3065 continue;
3066 // The strategy doesn't have printer or doesn't emit custom stack maps.
3067 // Use the default format.
3068 NeedsDefault = true;
3069 }
3070
3071 if (NeedsDefault)
3072 SM.serializeToStackMapSection();
3073}
3074
Juergen Ributzkadb9ee002013-12-14 12:23:14 +00003075/// Pin vtable to this file.
Eugene Zelenkod96089b2017-02-14 00:33:36 +00003076AsmPrinterHandler::~AsmPrinterHandler() = default;
Rafael Espindolaa6001792015-03-09 18:29:12 +00003077
3078void AsmPrinterHandler::markFunctionEnd() {}
Dean Michael Berris464015442016-09-19 00:54:35 +00003079
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003080// In the binary's "xray_instr_map" section, an array of these function entries
3081// describes each instrumentation point. When XRay patches your code, the index
3082// into this table will be given to your handler as a patch point identifier.
3083void AsmPrinter::XRayFunctionEntry::emit(int Bytes, MCStreamer *Out,
3084 const MCSymbol *CurrentFnSym) const {
3085 Out->EmitSymbolValue(Sled, Bytes);
3086 Out->EmitSymbolValue(CurrentFnSym, Bytes);
3087 auto Kind8 = static_cast<uint8_t>(Kind);
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003088 Out->EmitBinaryData(StringRef(reinterpret_cast<const char *>(&Kind8), 1));
3089 Out->EmitBinaryData(
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003090 StringRef(reinterpret_cast<const char *>(&AlwaysInstrument), 1));
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003091 Out->EmitBinaryData(StringRef(reinterpret_cast<const char *>(&Version), 1));
3092 auto Padding = (4 * Bytes) - ((2 * Bytes) + 3);
3093 assert(Padding >= 0 && "Instrumentation map entry > 4 * Word Size");
3094 Out->EmitZeros(Padding);
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003095}
3096
3097void AsmPrinter::emitXRayTable() {
3098 if (Sleds.empty())
3099 return;
3100
3101 auto PrevSection = OutStreamer->getCurrentSectionOnly();
Matthias Braunf1caa282017-12-15 22:22:58 +00003102 const Function &F = MF->getFunction();
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003103 MCSection *InstMap = nullptr;
3104 MCSection *FnSledIndex = nullptr;
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003105 if (MF->getSubtarget().getTargetTriple().isOSBinFormatELF()) {
Dean Michael Berris01fd7c82017-09-14 07:08:23 +00003106 auto Associated = dyn_cast<MCSymbolELF>(CurrentFnSym);
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003107 assert(Associated != nullptr);
3108 auto Flags = ELF::SHF_WRITE | ELF::SHF_ALLOC | ELF::SHF_LINK_ORDER;
3109 std::string GroupName;
Matthias Braunf1caa282017-12-15 22:22:58 +00003110 if (F.hasComdat()) {
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003111 Flags |= ELF::SHF_GROUP;
Matthias Braunf1caa282017-12-15 22:22:58 +00003112 GroupName = F.getComdat()->getName();
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003113 }
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003114
3115 auto UniqueID = ++XRayFnUniqueID;
3116 InstMap =
3117 OutContext.getELFSection("xray_instr_map", ELF::SHT_PROGBITS, Flags, 0,
3118 GroupName, UniqueID, Associated);
3119 FnSledIndex =
3120 OutContext.getELFSection("xray_fn_idx", ELF::SHT_PROGBITS, Flags, 0,
3121 GroupName, UniqueID, Associated);
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003122 } else if (MF->getSubtarget().getTargetTriple().isOSBinFormatMachO()) {
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003123 InstMap = OutContext.getMachOSection("__DATA", "xray_instr_map", 0,
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003124 SectionKind::getReadOnlyWithRel());
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003125 FnSledIndex = OutContext.getMachOSection("__DATA", "xray_fn_idx", 0,
3126 SectionKind::getReadOnlyWithRel());
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003127 } else {
3128 llvm_unreachable("Unsupported target");
3129 }
3130
Konstantin Zhuravlyovdc77b2e2017-04-17 17:41:25 +00003131 auto WordSizeBytes = MAI->getCodePointerSize();
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003132
3133 // Now we switch to the instrumentation map section. Because this is done
3134 // per-function, we are able to create an index entry that will represent the
3135 // range of sleds associated with a function.
Dean Michael Berris28ecff52017-06-21 06:39:42 +00003136 MCSymbol *SledsStart = OutContext.createTempSymbol("xray_sleds_start", true);
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003137 OutStreamer->SwitchSection(InstMap);
3138 OutStreamer->EmitLabel(SledsStart);
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003139 for (const auto &Sled : Sleds)
3140 Sled.emit(WordSizeBytes, OutStreamer.get(), CurrentFnSym);
Dean Michael Berris28ecff52017-06-21 06:39:42 +00003141 MCSymbol *SledsEnd = OutContext.createTempSymbol("xray_sleds_end", true);
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003142 OutStreamer->EmitLabel(SledsEnd);
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003143
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003144 // We then emit a single entry in the index per function. We use the symbols
3145 // that bound the instrumentation map as the range for a specific function.
Dean Michael Berrisebdccd02017-05-04 04:55:46 +00003146 // Each entry here will be 2 * word size aligned, as we're writing down two
3147 // pointers. This should work for both 32-bit and 64-bit platforms.
Dean Michael Berrisbdfe9002017-05-04 03:37:57 +00003148 OutStreamer->SwitchSection(FnSledIndex);
Dean Michael Berrisebdccd02017-05-04 04:55:46 +00003149 OutStreamer->EmitCodeAlignment(2 * WordSizeBytes);
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003150 OutStreamer->EmitSymbolValue(SledsStart, WordSizeBytes, false);
3151 OutStreamer->EmitSymbolValue(SledsEnd, WordSizeBytes, false);
Dean Michael Berrisf7e7b932017-01-03 04:30:21 +00003152 OutStreamer->SwitchSection(PrevSection);
3153 Sleds.clear();
3154}
3155
Dean Michael Berris464015442016-09-19 00:54:35 +00003156void AsmPrinter::recordSled(MCSymbol *Sled, const MachineInstr &MI,
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003157 SledKind Kind, uint8_t Version) {
Matthias Braunf1caa282017-12-15 22:22:58 +00003158 const Function &F = MI.getMF()->getFunction();
3159 auto Attr = F.getFnAttribute("function-instrument");
3160 bool LogArgs = F.hasFnAttribute("xray-log-args");
Dean Michael Berris464015442016-09-19 00:54:35 +00003161 bool AlwaysInstrument =
3162 Attr.isStringAttribute() && Attr.getValueAsString() == "xray-always";
Dean Michael Berris7e8eea42017-03-06 06:48:56 +00003163 if (Kind == SledKind::FUNCTION_ENTER && LogArgs)
3164 Kind = SledKind::LOG_ARGS_ENTER;
Dean Michael Berrisebc16592017-09-04 05:34:58 +00003165 Sleds.emplace_back(XRayFunctionEntry{Sled, CurrentFnSym, Kind,
Matthias Braunf1caa282017-12-15 22:22:58 +00003166 AlwaysInstrument, &F, Version});
Dean Michael Berris464015442016-09-19 00:54:35 +00003167}
Greg Claytone6543972016-11-23 23:30:37 +00003168
3169uint16_t AsmPrinter::getDwarfVersion() const {
3170 return OutStreamer->getContext().getDwarfVersion();
3171}
3172
3173void AsmPrinter::setDwarfVersion(uint16_t Version) {
3174 OutStreamer->getContext().setDwarfVersion(Version);
3175}