lib/Target/X86/X86AsmPrinter.h

   1 //===-- X86AsmPrinter.h - Convert X86 LLVM code to Intel assembly ---------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file the shared super class printer that converts from our internal
  11 // representation of machine-dependent LLVM code to Intel and AT&T format
  12 // assembly language.  This printer is the output mechanism used by `llc'.
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef X86ASMPRINTER_H
  17 #define X86ASMPRINTER_H
  18
  19 #include "X86.h"
  20 #include "X86MachineFunctionInfo.h"
  21 #include "X86TargetMachine.h"
  22 #include "llvm/CodeGen/AsmPrinter.h"
  23 #include "llvm/CodeGen/DwarfWriter.h"
  24 #include "llvm/CodeGen/MachineDebugInfo.h"
  25 #include "llvm/Support/Compiler.h"
  26 #include <set>
  27
  28
  29 namespace llvm {
  30
  31 // FIXME: Move this to CodeGen/AsmPrinter.h
  32 namespace PICStyle {
  33   enum X86AsmPICStyle {
  34     Stub, GOT
  35   };
  36 }
  37
  38 struct VISIBILITY_HIDDEN X86SharedAsmPrinter : public AsmPrinter {
  39   DwarfWriter DW;
  40
  41   X86SharedAsmPrinter(std::ostream &O, X86TargetMachine &TM,
  42                       const TargetAsmInfo *T)
  43     : AsmPrinter(O, TM, T), DW(O, this, T), X86PICStyle(PICStyle::GOT) {
  44     Subtarget = &TM.getSubtarget<X86Subtarget>();
  45   }
  46
  47   // We have to propagate some information about MachineFunction to
  48   // AsmPrinter. It's ok, when we're printing the function, since we have
  49   // access to MachineFunction and can get the appropriate MachineFunctionInfo.
  50   // Unfortunately, this is not possible when we're printing reference to
  51   // Function (e.g. calling it and so on). Even more, there is no way to get the
  52   // corresponding MachineFunctions: it can even be not created at all. That's
  53   // why we should use additional structure, when we're collecting all necessary
  54   // information.
  55   //
  56   // This structure is using e.g. for name decoration for stdcall & fastcall'ed
  57   // function, since we have to use arguments' size for decoration.
  58   typedef std::map<const Function*, X86FunctionInfo> FMFInfoMap;
  59   FMFInfoMap FunctionInfoMap;
  60
  61   void decorateName(std::string& Name, const GlobalValue* GV);
  62
  63   bool doInitialization(Module &M);
  64   bool doFinalization(Module &M);
  65
  66   void getAnalysisUsage(AnalysisUsage &AU) const {
  67     AU.setPreservesAll();
  68     if (Subtarget->isTargetDarwin() ||
  69         Subtarget->isTargetELF() ||
  70         Subtarget->isTargetCygMing()) {
  71       AU.addRequired<MachineDebugInfo>();
  72     }
  73     MachineFunctionPass::getAnalysisUsage(AU);
  74   }
  75
  76   PICStyle::X86AsmPICStyle X86PICStyle;
  77
  78   const X86Subtarget *Subtarget;
  79
  80   // Necessary for Darwin to print out the apprioriate types of linker stubs
  81   std::set<std::string> FnStubs, GVStubs, LinkOnceStubs;
  82
  83   // Necessary for dllexport support
  84   std::set<std::string> DLLExportedFns, DLLExportedGVs;
  85
  86   inline static bool isScale(const MachineOperand &MO) {
  87     return MO.isImmediate() &&
  88           (MO.getImmedValue() == 1 || MO.getImmedValue() == 2 ||
  89           MO.getImmedValue() == 4 || MO.getImmedValue() == 8);
  90   }
  91
  92   inline static bool isMem(const MachineInstr *MI, unsigned Op) {
  93     if (MI->getOperand(Op).isFrameIndex()) return true;
  94     return Op+4 <= MI->getNumOperands() &&
  95       MI->getOperand(Op  ).isRegister() && isScale(MI->getOperand(Op+1)) &&
  96       MI->getOperand(Op+2).isRegister() &&
  97       (MI->getOperand(Op+3).isImmediate() ||
  98        MI->getOperand(Op+3).isGlobalAddress() ||
  99        MI->getOperand(Op+3).isConstantPoolIndex() ||
 100        MI->getOperand(Op+3).isJumpTableIndex());
 101   }
 102 };
 103
 104 } // end namespace llvm
 105
 106 #endif