lib/Target/X86/MCTargetDesc/X86WinCOFFObjectWriter.cpp

   1 //===-- X86WinCOFFObjectWriter.cpp - X86 Win COFF Writer ------------------===//
   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 #include "MCTargetDesc/X86FixupKinds.h"
  11 #include "MCTargetDesc/X86MCTargetDesc.h"
  12 #include "llvm/MC/MCExpr.h"
  13 #include "llvm/MC/MCValue.h"
  14 #include "llvm/MC/MCWinCOFFObjectWriter.h"
  15 #include "llvm/Support/COFF.h"
  16 #include "llvm/Support/ErrorHandling.h"
  17
  18 using namespace llvm;
  19
  20 namespace llvm {
  21   class MCObjectWriter;
  22 }
  23
  24 namespace {
  25   class X86WinCOFFObjectWriter : public MCWinCOFFObjectTargetWriter {
  26     const bool Is64Bit;
  27
  28   public:
  29     X86WinCOFFObjectWriter(bool Is64Bit_);
  30     virtual ~X86WinCOFFObjectWriter();
  31
  32     unsigned getRelocType(const MCValue &Target, const MCFixup &Fixup,
  33                           bool IsCrossSection) const override;
  34   };
  35 }
  36
  37 X86WinCOFFObjectWriter::X86WinCOFFObjectWriter(bool Is64Bit_)
  38   : MCWinCOFFObjectTargetWriter(Is64Bit_ ? COFF::IMAGE_FILE_MACHINE_AMD64 :
  39                                 COFF::IMAGE_FILE_MACHINE_I386),
  40     Is64Bit(Is64Bit_) {}
  41
  42 X86WinCOFFObjectWriter::~X86WinCOFFObjectWriter() {}
  43
  44 unsigned X86WinCOFFObjectWriter::getRelocType(const MCValue &Target,
  45                                               const MCFixup &Fixup,
  46                                               bool IsCrossSection) const {
  47   unsigned FixupKind = IsCrossSection ? FK_PCRel_4 : Fixup.getKind();
  48
  49   MCSymbolRefExpr::VariantKind Modifier = Target.isAbsolute() ?
  50     MCSymbolRefExpr::VK_None : Target.getSymA()->getKind();
  51
  52   switch (FixupKind) {
  53   case FK_PCRel_4:
  54   case X86::reloc_riprel_4byte:
  55   case X86::reloc_riprel_4byte_movq_load:
  56     return Is64Bit ? COFF::IMAGE_REL_AMD64_REL32 : COFF::IMAGE_REL_I386_REL32;
  57   case FK_Data_4:
  58   case X86::reloc_signed_4byte:
  59     if (Modifier == MCSymbolRefExpr::VK_COFF_IMGREL32)
  60       return Is64Bit ? COFF::IMAGE_REL_AMD64_ADDR32NB :
  61                        COFF::IMAGE_REL_I386_DIR32NB;
  62     return Is64Bit ? COFF::IMAGE_REL_AMD64_ADDR32 : COFF::IMAGE_REL_I386_DIR32;
  63   case FK_Data_8:
  64     if (Is64Bit)
  65       return COFF::IMAGE_REL_AMD64_ADDR64;
  66     llvm_unreachable("unsupported relocation type");
  67   case FK_SecRel_2:
  68     return Is64Bit ? COFF::IMAGE_REL_AMD64_SECTION
  69                    : COFF::IMAGE_REL_I386_SECTION;
  70   case FK_SecRel_4:
  71     return Is64Bit ? COFF::IMAGE_REL_AMD64_SECREL : COFF::IMAGE_REL_I386_SECREL;
  72   default:
  73     llvm_unreachable("unsupported relocation type");
  74   }
  75 }
  76
  77 MCObjectWriter *llvm::createX86WinCOFFObjectWriter(raw_ostream &OS,
  78                                                    bool Is64Bit) {
  79   MCWinCOFFObjectTargetWriter *MOTW = new X86WinCOFFObjectWriter(Is64Bit);
  80   return createWinCOFFObjectWriter(MOTW, OS);
  81 }