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
18using namespace llvm;
19
20namespace llvm {
21  class MCObjectWriter;
22}
23
24namespace {
25  class X86WinCOFFObjectWriter : public MCWinCOFFObjectTargetWriter {
26  public:
27    X86WinCOFFObjectWriter(bool Is64Bit);
28    ~X86WinCOFFObjectWriter() override;
29
30    unsigned getRelocType(const MCValue &Target, const MCFixup &Fixup,
31                          bool IsCrossSection,
32                          const MCAsmBackend &MAB) const override;
33  };
34}
35
36X86WinCOFFObjectWriter::X86WinCOFFObjectWriter(bool Is64Bit)
37    : MCWinCOFFObjectTargetWriter(Is64Bit ? COFF::IMAGE_FILE_MACHINE_AMD64
38                                          : COFF::IMAGE_FILE_MACHINE_I386) {}
39
40X86WinCOFFObjectWriter::~X86WinCOFFObjectWriter() {}
41
42unsigned X86WinCOFFObjectWriter::getRelocType(const MCValue &Target,
43                                              const MCFixup &Fixup,
44                                              bool IsCrossSection,
45                                              const MCAsmBackend &MAB) const {
46  unsigned FixupKind = IsCrossSection ? FK_PCRel_4 : Fixup.getKind();
47
48  MCSymbolRefExpr::VariantKind Modifier = Target.isAbsolute() ?
49    MCSymbolRefExpr::VK_None : Target.getSymA()->getKind();
50
51  if (getMachine() == COFF::IMAGE_FILE_MACHINE_AMD64) {
52    switch (FixupKind) {
53    case FK_PCRel_4:
54    case X86::reloc_riprel_4byte:
55    case X86::reloc_riprel_4byte_movq_load:
56    case X86::reloc_riprel_4byte_relax:
57    case X86::reloc_riprel_4byte_relax_rex:
58      return COFF::IMAGE_REL_AMD64_REL32;
59    case FK_Data_4:
60    case X86::reloc_signed_4byte:
61    case X86::reloc_signed_4byte_relax:
62      if (Modifier == MCSymbolRefExpr::VK_COFF_IMGREL32)
63        return COFF::IMAGE_REL_AMD64_ADDR32NB;
64      if (Modifier == MCSymbolRefExpr::VK_SECREL)
65        return COFF::IMAGE_REL_AMD64_SECREL;
66      return COFF::IMAGE_REL_AMD64_ADDR32;
67    case FK_Data_8:
68      return COFF::IMAGE_REL_AMD64_ADDR64;
69    case FK_SecRel_2:
70      return COFF::IMAGE_REL_AMD64_SECTION;
71    case FK_SecRel_4:
72      return COFF::IMAGE_REL_AMD64_SECREL;
73    default:
74      llvm_unreachable("unsupported relocation type");
75    }
76  } else if (getMachine() == COFF::IMAGE_FILE_MACHINE_I386) {
77    switch (FixupKind) {
78    case FK_PCRel_4:
79    case X86::reloc_riprel_4byte:
80    case X86::reloc_riprel_4byte_movq_load:
81      return COFF::IMAGE_REL_I386_REL32;
82    case FK_Data_4:
83    case X86::reloc_signed_4byte:
84    case X86::reloc_signed_4byte_relax:
85      if (Modifier == MCSymbolRefExpr::VK_COFF_IMGREL32)
86        return COFF::IMAGE_REL_I386_DIR32NB;
87      if (Modifier == MCSymbolRefExpr::VK_SECREL)
88        return COFF::IMAGE_REL_AMD64_SECREL;
89      return COFF::IMAGE_REL_I386_DIR32;
90    case FK_SecRel_2:
91      return COFF::IMAGE_REL_I386_SECTION;
92    case FK_SecRel_4:
93      return COFF::IMAGE_REL_I386_SECREL;
94    default:
95      llvm_unreachable("unsupported relocation type");
96    }
97  } else
98    llvm_unreachable("Unsupported COFF machine type.");
99}
100
101MCObjectWriter *llvm::createX86WinCOFFObjectWriter(raw_pwrite_stream &OS,
102                                                   bool Is64Bit) {
103  MCWinCOFFObjectTargetWriter *MOTW = new X86WinCOFFObjectWriter(Is64Bit);
104  return createWinCOFFObjectWriter(MOTW, OS);
105}
106