1 //===-- SystemZAsmPrinter.cpp - SystemZ LLVM assembly writer ---------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file contains a printer that converts from our internal representation
11 // of machine-dependent LLVM code to the SystemZ assembly language.
13 //===----------------------------------------------------------------------===//
15 #define DEBUG_TYPE "asm-printer"
17 #include "SystemZInstrInfo.h"
18 #include "SystemZTargetMachine.h"
19 #include "llvm/Constants.h"
20 #include "llvm/DerivedTypes.h"
21 #include "llvm/Module.h"
22 #include "llvm/CodeGen/AsmPrinter.h"
23 #include "llvm/CodeGen/DwarfWriter.h"
24 #include "llvm/CodeGen/MachineModuleInfo.h"
25 #include "llvm/CodeGen/MachineFunctionPass.h"
26 #include "llvm/CodeGen/MachineConstantPool.h"
27 #include "llvm/CodeGen/MachineInstr.h"
28 #include "llvm/Target/TargetAsmInfo.h"
29 #include "llvm/Target/TargetData.h"
30 #include "llvm/Target/TargetRegistry.h"
31 #include "llvm/ADT/Statistic.h"
32 #include "llvm/Support/Compiler.h"
33 #include "llvm/Support/FormattedStream.h"
34 #include "llvm/Support/Mangler.h"
38 STATISTIC(EmittedInsts, "Number of machine instrs printed");
41 class VISIBILITY_HIDDEN SystemZAsmPrinter : public AsmPrinter {
43 SystemZAsmPrinter(formatted_raw_ostream &O, TargetMachine &TM,
44 const TargetAsmInfo *TAI, bool V)
45 : AsmPrinter(O, TM, TAI, V) {}
47 virtual const char *getPassName() const {
48 return "SystemZ Assembly Printer";
51 void printOperand(const MachineInstr *MI, int OpNum,
52 const char* Modifier = 0);
53 void printPCRelImmOperand(const MachineInstr *MI, int OpNum);
54 void printRIAddrOperand(const MachineInstr *MI, int OpNum,
55 const char* Modifier = 0);
56 void printRRIAddrOperand(const MachineInstr *MI, int OpNum,
57 const char* Modifier = 0);
58 void printS16ImmOperand(const MachineInstr *MI, int OpNum) {
59 O << (int16_t)MI->getOperand(OpNum).getImm();
61 void printS32ImmOperand(const MachineInstr *MI, int OpNum) {
62 O << (int32_t)MI->getOperand(OpNum).getImm();
65 bool printInstruction(const MachineInstr *MI); // autogenerated.
66 void printMachineInstruction(const MachineInstr * MI);
68 void emitFunctionHeader(const MachineFunction &MF);
69 bool runOnMachineFunction(MachineFunction &F);
70 bool doInitialization(Module &M);
71 bool doFinalization(Module &M);
72 void printModuleLevelGV(const GlobalVariable* GVar);
74 void getAnalysisUsage(AnalysisUsage &AU) const {
75 AsmPrinter::getAnalysisUsage(AU);
79 } // end of anonymous namespace
81 #include "SystemZGenAsmWriter.inc"
83 /// createSystemZCodePrinterPass - Returns a pass that prints the SystemZ
84 /// assembly code for a MachineFunction to the given output stream,
85 /// using the given target machine description. This should work
86 /// regardless of whether the function is in SSA form.
88 FunctionPass *llvm::createSystemZCodePrinterPass(formatted_raw_ostream &o,
91 return new SystemZAsmPrinter(o, tm, tm.getTargetAsmInfo(), verbose);
94 bool SystemZAsmPrinter::doInitialization(Module &M) {
95 Mang = new Mangler(M, "", TAI->getPrivateGlobalPrefix());
96 return false; // success
100 bool SystemZAsmPrinter::doFinalization(Module &M) {
101 // Print out module-level global variables here.
102 for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
104 printModuleLevelGV(I);
106 return AsmPrinter::doFinalization(M);
109 void SystemZAsmPrinter::emitFunctionHeader(const MachineFunction &MF) {
110 unsigned FnAlign = MF.getAlignment();
111 const Function *F = MF.getFunction();
113 SwitchToSection(TAI->SectionForGlobal(F));
115 EmitAlignment(FnAlign, F);
117 switch (F->getLinkage()) {
118 default: assert(0 && "Unknown linkage type!");
119 case Function::InternalLinkage: // Symbols default to internal.
120 case Function::PrivateLinkage:
122 case Function::ExternalLinkage:
123 O << "\t.globl\t" << CurrentFnName << '\n';
125 case Function::LinkOnceAnyLinkage:
126 case Function::LinkOnceODRLinkage:
127 case Function::WeakAnyLinkage:
128 case Function::WeakODRLinkage:
129 O << "\t.weak\t" << CurrentFnName << '\n';
133 printVisibility(CurrentFnName, F->getVisibility());
135 O << "\t.type\t" << CurrentFnName << ",@function\n"
136 << CurrentFnName << ":\n";
139 bool SystemZAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
140 SetupMachineFunction(MF);
143 // Print out constants referenced by the function
144 EmitConstantPool(MF.getConstantPool());
146 // Print the 'header' of function
147 emitFunctionHeader(MF);
149 // Print out code for the function.
150 for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
152 // Print a label for the basic block.
153 if (!VerboseAsm && (I->pred_empty() || I->isOnlyReachableByFallthrough())) {
154 // This is an entry block or a block that's only reachable via a
155 // fallthrough edge. In non-VerboseAsm mode, don't print the label.
157 printBasicBlockLabel(I, true, true, VerboseAsm);
161 for (MachineBasicBlock::const_iterator II = I->begin(), E = I->end();
163 // Print the assembly for the instruction.
164 printMachineInstruction(II);
167 if (TAI->hasDotTypeDotSizeDirective())
168 O << "\t.size\t" << CurrentFnName << ", .-" << CurrentFnName << '\n';
170 // Print out jump tables referenced by the function.
171 EmitJumpTableInfo(MF.getJumpTableInfo(), MF);
175 // We didn't modify anything
179 void SystemZAsmPrinter::printMachineInstruction(const MachineInstr *MI) {
182 // Call the autogenerated instruction printer routines.
183 if (printInstruction(MI))
186 assert(0 && "Should not happen");
189 void SystemZAsmPrinter::printPCRelImmOperand(const MachineInstr *MI, int OpNum) {
190 const MachineOperand &MO = MI->getOperand(OpNum);
191 switch (MO.getType()) {
192 case MachineOperand::MO_Immediate:
195 case MachineOperand::MO_MachineBasicBlock:
196 printBasicBlockLabel(MO.getMBB(), false, false, VerboseAsm);
198 case MachineOperand::MO_GlobalAddress: {
199 const GlobalValue *GV = MO.getGlobal();
200 std::string Name = Mang->getMangledName(GV);
204 // Assemble calls via PLT for externally visible symbols if PIC.
205 if (TM.getRelocationModel() == Reloc::PIC_ &&
206 !GV->hasHiddenVisibility() && !GV->hasProtectedVisibility() &&
207 !GV->hasLocalLinkage())
210 printOffset(MO.getOffset());
213 case MachineOperand::MO_ExternalSymbol: {
214 std::string Name(TAI->getGlobalPrefix());
215 Name += MO.getSymbolName();
218 if (TM.getRelocationModel() == Reloc::PIC_)
224 assert(0 && "Not implemented yet!");
229 void SystemZAsmPrinter::printOperand(const MachineInstr *MI, int OpNum,
230 const char* Modifier) {
231 const MachineOperand &MO = MI->getOperand(OpNum);
232 switch (MO.getType()) {
233 case MachineOperand::MO_Register: {
234 assert (TargetRegisterInfo::isPhysicalRegister(MO.getReg()) &&
235 "Virtual registers should be already mapped!");
236 unsigned Reg = MO.getReg();
237 if (Modifier && strncmp(Modifier, "subreg", 6) == 0) {
238 if (strncmp(Modifier + 7, "even", 4) == 0)
239 Reg = TRI->getSubReg(Reg, SystemZ::SUBREG_EVEN);
240 else if (strncmp(Modifier + 7, "odd", 3) == 0)
241 Reg = TRI->getSubReg(Reg, SystemZ::SUBREG_ODD);
243 assert(0 && "Invalid subreg modifier");
246 O << '%' << TRI->getAsmName(Reg);
249 case MachineOperand::MO_Immediate:
252 case MachineOperand::MO_MachineBasicBlock:
253 printBasicBlockLabel(MO.getMBB());
255 case MachineOperand::MO_JumpTableIndex:
256 O << TAI->getPrivateGlobalPrefix() << "JTI" << getFunctionNumber() << '_'
260 case MachineOperand::MO_ConstantPoolIndex:
261 O << TAI->getPrivateGlobalPrefix() << "CPI" << getFunctionNumber() << '_'
264 printOffset(MO.getOffset());
266 case MachineOperand::MO_GlobalAddress: {
267 const GlobalValue *GV = MO.getGlobal();
268 std::string Name = Mang->getMangledName(GV);
273 case MachineOperand::MO_ExternalSymbol: {
274 std::string Name(TAI->getGlobalPrefix());
275 Name += MO.getSymbolName();
280 assert(0 && "Not implemented yet!");
283 switch (MO.getTargetFlags()) {
285 assert(0 && "Unknown target flag on GV operand");
286 case SystemZII::MO_NO_FLAG:
288 case SystemZII::MO_GOTENT: O << "@GOTENT"; break;
289 case SystemZII::MO_PLT: O << "@PLT"; break;
292 printOffset(MO.getOffset());
295 void SystemZAsmPrinter::printRIAddrOperand(const MachineInstr *MI, int OpNum,
296 const char* Modifier) {
297 const MachineOperand &Base = MI->getOperand(OpNum);
299 // Print displacement operand.
300 printOperand(MI, OpNum+1);
302 // Print base operand (if any)
305 printOperand(MI, OpNum);
310 void SystemZAsmPrinter::printRRIAddrOperand(const MachineInstr *MI, int OpNum,
311 const char* Modifier) {
312 const MachineOperand &Base = MI->getOperand(OpNum);
313 const MachineOperand &Index = MI->getOperand(OpNum+2);
315 // Print displacement operand.
316 printOperand(MI, OpNum+1);
318 // Print base operand (if any)
321 printOperand(MI, OpNum);
322 if (Index.getReg()) {
324 printOperand(MI, OpNum+2);
328 assert(!Index.getReg() && "Should allocate base register first!");
331 /// PrintUnmangledNameSafely - Print out the printable characters in the name.
332 /// Don't print things like \\n or \\0.
333 static void PrintUnmangledNameSafely(const Value *V, formatted_raw_ostream &OS) {
334 for (const char *Name = V->getNameStart(), *E = Name+V->getNameLen();
340 void SystemZAsmPrinter::printModuleLevelGV(const GlobalVariable* GVar) {
341 const TargetData *TD = TM.getTargetData();
343 if (!GVar->hasInitializer())
344 return; // External global require no code
346 // Check to see if this is a special global used by LLVM, if so, emit it.
347 if (EmitSpecialLLVMGlobal(GVar))
350 std::string name = Mang->getMangledName(GVar);
351 Constant *C = GVar->getInitializer();
352 unsigned Size = TD->getTypeAllocSize(C->getType());
353 unsigned Align = std::max(1U, TD->getPreferredAlignmentLog(GVar));
355 printVisibility(name, GVar->getVisibility());
357 O << "\t.type\t" << name << ",@object\n";
359 SwitchToSection(TAI->SectionForGlobal(GVar));
361 if (C->isNullValue() && !GVar->hasSection() &&
362 !GVar->isThreadLocal() &&
363 (GVar->hasLocalLinkage() || GVar->isWeakForLinker())) {
365 if (Size == 0) Size = 1; // .comm Foo, 0 is undefined, avoid it.
367 if (GVar->hasLocalLinkage())
368 O << "\t.local\t" << name << '\n';
370 O << TAI->getCOMMDirective() << name << ',' << Size;
371 if (TAI->getCOMMDirectiveTakesAlignment())
372 O << ',' << (TAI->getAlignmentIsInBytes() ? (1 << Align) : Align);
375 O << "\t\t" << TAI->getCommentString() << ' ';
376 PrintUnmangledNameSafely(GVar, O);
382 switch (GVar->getLinkage()) {
383 case GlobalValue::CommonLinkage:
384 case GlobalValue::LinkOnceAnyLinkage:
385 case GlobalValue::LinkOnceODRLinkage:
386 case GlobalValue::WeakAnyLinkage:
387 case GlobalValue::WeakODRLinkage:
388 O << "\t.weak\t" << name << '\n';
390 case GlobalValue::DLLExportLinkage:
391 case GlobalValue::AppendingLinkage:
392 // FIXME: appending linkage variables should go into a section of
393 // their name or something. For now, just emit them as external.
394 case GlobalValue::ExternalLinkage:
395 // If external or appending, declare as a global symbol
396 O << "\t.globl " << name << '\n';
398 case GlobalValue::PrivateLinkage:
399 case GlobalValue::InternalLinkage:
402 assert(0 && "Unknown linkage type!");
405 // Use 16-bit alignment by default to simplify bunch of stuff
406 EmitAlignment(Align, GVar, 1);
409 O << "\t\t\t\t" << TAI->getCommentString() << ' ';
410 PrintUnmangledNameSafely(GVar, O);
413 if (TAI->hasDotTypeDotSizeDirective())
414 O << "\t.size\t" << name << ", " << Size << '\n';
416 EmitGlobalConstant(C);
419 // Force static initialization.
420 extern "C" void LLVMInitializeSystemZAsmPrinter() {
421 extern Target TheSystemZTarget;
422 TargetRegistry::RegisterAsmPrinter(TheSystemZTarget,
423 createSystemZCodePrinterPass);