lib/CodeGen/MachOWriter.cpp

   1 //===-- MachOWriter.cpp - Target-independent Mach-O Writer code -----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the target-independent Mach-O writer.  This file writes
  11 // out the Mach-O file in the following order:
  12 //
  13 //  #1 FatHeader (universal-only)
  14 //  #2 FatArch (universal-only, 1 per universal arch)
  15 //  Per arch:
  16 //    #3 Header
  17 //    #4 Load Commands
  18 //    #5 Sections
  19 //    #6 Relocations
  20 //    #7 Symbols
  21 //    #8 Strings
  22 //
  23 //===----------------------------------------------------------------------===//
  24
  25 #include "MachOWriter.h"
  26 #include "llvm/Function.h"
  27 #include "llvm/CodeGen/FileWriters.h"
  28 #include "llvm/CodeGen/MachineFunction.h"
  29 #include "llvm/MC/MCAsmInfo.h"
  30 #include "llvm/MC/MCContext.h"
  31 #include "llvm/MC/MCCodeEmitter.h"
  32 #include "llvm/MC/MCInst.h"
  33 #include "llvm/MC/MCStreamer.h"
  34 #include "llvm/Support/ErrorHandling.h"
  35 #include "llvm/Support/FormattedStream.h"
  36 #include "llvm/Support/Mangler.h"
  37 #include "llvm/Support/raw_ostream.h"
  38 #include "llvm/Target/TargetData.h"
  39 #include "llvm/Target/TargetLowering.h"
  40 #include "llvm/Target/TargetLoweringObjectFile.h"
  41 using namespace llvm;
  42
  43 namespace llvm {
  44 MachineFunctionPass *createMachOWriter(formatted_raw_ostream &O,
  45                                        TargetMachine &TM,
  46                                        const MCAsmInfo *T,
  47                                        MCCodeEmitter *MCE) {
  48   return new MachOWriter(O, TM, T, MCE);
  49 }
  50 }
  51
  52 //===----------------------------------------------------------------------===//
  53 //                          MachOWriter Implementation
  54 //===----------------------------------------------------------------------===//
  55
  56 char MachOWriter::ID = 0;
  57
  58 MachOWriter::MachOWriter(formatted_raw_ostream &o, TargetMachine &tm,
  59                          const MCAsmInfo *T, MCCodeEmitter *MCE)
  60   : MachineFunctionPass(&ID), O(o), TM(tm), MAI(T), MCCE(MCE),
  61     OutContext(*new MCContext()),
  62     OutStreamer(*createMachOStreamer(OutContext, O, MCCE)) {
  63 }
  64
  65 MachOWriter::~MachOWriter() {
  66   delete &OutStreamer;
  67   delete &OutContext;
  68   delete MCCE;
  69 }
  70
  71 bool MachOWriter::doInitialization(Module &M) {
  72   Mang = new Mangler(M, MAI->getGlobalPrefix(), MAI->getPrivateGlobalPrefix(),
  73                      MAI->getLinkerPrivateGlobalPrefix());
  74
  75   if (MAI->doesAllowQuotesInName())
  76     Mang->setUseQuotes(true);
  77
  78   if (MAI->doesAllowNameToStartWithDigit())
  79     Mang->setSymbolsCanStartWithDigit(true);
  80
  81   // Initialize TargetLoweringObjectFile.
  82   TM.getTargetLowering()->getObjFileLowering().Initialize(OutContext, TM);
  83
  84   return false;
  85 }
  86
  87 /// doFinalization - Now that the module has been completely processed, emit
  88 /// the Mach-O file to 'O'.
  89 bool MachOWriter::doFinalization(Module &M) {
  90   // Release the name mangler object.
  91   delete Mang; Mang = 0;
  92
  93   OutStreamer.Finish();
  94   return false;
  95 }
  96
  97 bool MachOWriter::runOnMachineFunction(MachineFunction &MF) {
  98   const Function *F = MF.getFunction();
  99   TargetLoweringObjectFile &TLOF = TM.getTargetLowering()->getObjFileLowering();
 100   const MCSection *S = TLOF.SectionForGlobal(F, Mang, TM);
 101   OutStreamer.SwitchSection(S);
 102
 103   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
 104        I != E; ++I) {
 105     // Print a label for the basic block.
 106     for (MachineBasicBlock::const_iterator II = I->begin(), IE = I->end();
 107          II != IE; ++II) {
 108       const MachineInstr *MI = II;
 109       MCInst OutMI;
 110       OutMI.setOpcode(MI->getOpcode());
 111
 112       for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
 113         const MachineOperand &MO = MI->getOperand(i);
 114         MCOperand MCOp;
 115
 116         switch (MO.getType()) {
 117           default:
 118             MI->dump();
 119             llvm_unreachable("unknown operand type");
 120           case MachineOperand::MO_Register:
 121             // Ignore all implicit register operands.
 122             if (MO.isImplicit()) continue;
 123             MCOp = MCOperand::CreateReg(MO.getReg());
 124             break;
 125           case MachineOperand::MO_Immediate:
 126             MCOp = MCOperand::CreateImm(MO.getImm());
 127             break;
 128         }
 129         OutMI.addOperand(MCOp);
 130       }
 131
 132       OutStreamer.EmitInstruction(OutMI);
 133     }
 134   }
 135
 136   return false;
 137 }