Home | History | Annotate | Download | only in MCTargetDesc
      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/BinaryFormat/COFF.h"
     13 #include "llvm/MC/MCContext.h"
     14 #include "llvm/MC/MCExpr.h"
     15 #include "llvm/MC/MCFixup.h"
     16 #include "llvm/MC/MCObjectWriter.h"
     17 #include "llvm/MC/MCValue.h"
     18 #include "llvm/MC/MCWinCOFFObjectWriter.h"
     19 #include "llvm/Support/ErrorHandling.h"
     20 
     21 using namespace llvm;
     22 
     23 namespace {
     24 
     25 class X86WinCOFFObjectWriter : public MCWinCOFFObjectTargetWriter {
     26 public:
     27   X86WinCOFFObjectWriter(bool Is64Bit);
     28   ~X86WinCOFFObjectWriter() override = default;
     29 
     30   unsigned getRelocType(MCContext &Ctx, const MCValue &Target,
     31                         const MCFixup &Fixup, bool IsCrossSection,
     32                         const MCAsmBackend &MAB) const override;
     33 };
     34 
     35 } // end anonymous namespace
     36 
     37 X86WinCOFFObjectWriter::X86WinCOFFObjectWriter(bool Is64Bit)
     38     : MCWinCOFFObjectTargetWriter(Is64Bit ? COFF::IMAGE_FILE_MACHINE_AMD64
     39                                           : COFF::IMAGE_FILE_MACHINE_I386) {}
     40 
     41 unsigned X86WinCOFFObjectWriter::getRelocType(MCContext &Ctx,
     42                                               const MCValue &Target,
     43                                               const MCFixup &Fixup,
     44                                               bool IsCrossSection,
     45                                               const MCAsmBackend &MAB) const {
     46   unsigned FixupKind = Fixup.getKind();
     47   if (IsCrossSection) {
     48     if (FixupKind != FK_Data_4 && FixupKind != llvm::X86::reloc_signed_4byte) {
     49       Ctx.reportError(Fixup.getLoc(), "Cannot represent this expression");
     50       return COFF::IMAGE_REL_AMD64_ADDR32;
     51     }
     52     FixupKind = FK_PCRel_4;
     53   }
     54 
     55   MCSymbolRefExpr::VariantKind Modifier = Target.isAbsolute() ?
     56     MCSymbolRefExpr::VK_None : Target.getSymA()->getKind();
     57 
     58   if (getMachine() == COFF::IMAGE_FILE_MACHINE_AMD64) {
     59     switch (FixupKind) {
     60     case FK_PCRel_4:
     61     case X86::reloc_riprel_4byte:
     62     case X86::reloc_riprel_4byte_movq_load:
     63     case X86::reloc_riprel_4byte_relax:
     64     case X86::reloc_riprel_4byte_relax_rex:
     65     case X86::reloc_branch_4byte_pcrel:
     66       return COFF::IMAGE_REL_AMD64_REL32;
     67     case FK_Data_4:
     68     case X86::reloc_signed_4byte:
     69     case X86::reloc_signed_4byte_relax:
     70       if (Modifier == MCSymbolRefExpr::VK_COFF_IMGREL32)
     71         return COFF::IMAGE_REL_AMD64_ADDR32NB;
     72       if (Modifier == MCSymbolRefExpr::VK_SECREL)
     73         return COFF::IMAGE_REL_AMD64_SECREL;
     74       return COFF::IMAGE_REL_AMD64_ADDR32;
     75     case FK_Data_8:
     76       return COFF::IMAGE_REL_AMD64_ADDR64;
     77     case FK_SecRel_2:
     78       return COFF::IMAGE_REL_AMD64_SECTION;
     79     case FK_SecRel_4:
     80       return COFF::IMAGE_REL_AMD64_SECREL;
     81     default:
     82       llvm_unreachable("unsupported relocation type");
     83     }
     84   } else if (getMachine() == COFF::IMAGE_FILE_MACHINE_I386) {
     85     switch (FixupKind) {
     86     case FK_PCRel_4:
     87     case X86::reloc_riprel_4byte:
     88     case X86::reloc_riprel_4byte_movq_load:
     89       return COFF::IMAGE_REL_I386_REL32;
     90     case FK_Data_4:
     91     case X86::reloc_signed_4byte:
     92     case X86::reloc_signed_4byte_relax:
     93       if (Modifier == MCSymbolRefExpr::VK_COFF_IMGREL32)
     94         return COFF::IMAGE_REL_I386_DIR32NB;
     95       if (Modifier == MCSymbolRefExpr::VK_SECREL)
     96         return COFF::IMAGE_REL_AMD64_SECREL;
     97       return COFF::IMAGE_REL_I386_DIR32;
     98     case FK_SecRel_2:
     99       return COFF::IMAGE_REL_I386_SECTION;
    100     case FK_SecRel_4:
    101       return COFF::IMAGE_REL_I386_SECREL;
    102     default:
    103       llvm_unreachable("unsupported relocation type");
    104     }
    105   } else
    106     llvm_unreachable("Unsupported COFF machine type.");
    107 }
    108 
    109 std::unique_ptr<MCObjectTargetWriter>
    110 llvm::createX86WinCOFFObjectWriter(bool Is64Bit) {
    111   return llvm::make_unique<X86WinCOFFObjectWriter>(Is64Bit);
    112 }
    113