1//===- llvm/MC/MCWinCOFFObjectWriter.h - Win COFF Object Writer -*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef LLVM_MC_MCWINCOFFOBJECTWRITER_H
10#define LLVM_MC_MCWINCOFFOBJECTWRITER_H
11
12#include "llvm/MC/MCObjectWriter.h"
13#include <memory>
14
15namespace llvm {
16
17class MCAsmBackend;
18class MCContext;
19class MCFixup;
20class MCValue;
21class raw_pwrite_stream;
22
23class MCWinCOFFObjectTargetWriter : public MCObjectTargetWriter {
24 virtual void anchor();
25
26 const unsigned Machine;
27
28protected:
29 MCWinCOFFObjectTargetWriter(unsigned Machine_);
30
31public:
32 virtual ~MCWinCOFFObjectTargetWriter() = default;
33
34 Triple::ObjectFormatType getFormat() const override { return Triple::COFF; }
35 static bool classof(const MCObjectTargetWriter *W) {
36 return W->getFormat() == Triple::COFF;
37 }
38
39 unsigned getMachine() const { return Machine; }
40 virtual unsigned getRelocType(MCContext &Ctx, const MCValue &Target,
41 const MCFixup &Fixup, bool IsCrossSection,
42 const MCAsmBackend &MAB) const = 0;
43 virtual bool recordRelocation(const MCFixup &) const { return true; }
44};
45
46class WinCOFFWriter;
47
48class WinCOFFObjectWriter : public MCObjectWriter {
49 friend class WinCOFFWriter;
50
51 std::unique_ptr<MCWinCOFFObjectTargetWriter> TargetObjectWriter;
52 std::unique_ptr<WinCOFFWriter> ObjWriter, DwoWriter;
53 bool IncrementalLinkerCompatible = false;
54
55public:
56 WinCOFFObjectWriter(std::unique_ptr<MCWinCOFFObjectTargetWriter> MOTW,
57 raw_pwrite_stream &OS);
58 WinCOFFObjectWriter(std::unique_ptr<MCWinCOFFObjectTargetWriter> MOTW,
59 raw_pwrite_stream &OS, raw_pwrite_stream &DwoOS);
60
61 // MCObjectWriter interface implementation.
62 void reset() override;
63 void setIncrementalLinkerCompatible(bool Value) {
64 IncrementalLinkerCompatible = Value;
65 }
66 void executePostLayoutBinding(MCAssembler &Asm) override;
67 bool isSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm,
68 const MCSymbol &SymA,
69 const MCFragment &FB, bool InSet,
70 bool IsPCRel) const override;
71 void recordRelocation(MCAssembler &Asm, const MCFragment *Fragment,
72 const MCFixup &Fixup, MCValue Target,
73 uint64_t &FixedValue) override;
74 uint64_t writeObject(MCAssembler &Asm) override;
75};
76
77/// Construct a new Win COFF writer instance.
78///
79/// \param MOTW - The target specific WinCOFF writer subclass.
80/// \param OS - The stream to write to.
81/// \returns The constructed object writer.
82std::unique_ptr<MCObjectWriter>
83createWinCOFFObjectWriter(std::unique_ptr<MCWinCOFFObjectTargetWriter> MOTW,
84 raw_pwrite_stream &OS);
85
86std::unique_ptr<MCObjectWriter>
87createWinCOFFDwoObjectWriter(std::unique_ptr<MCWinCOFFObjectTargetWriter> MOTW,
88 raw_pwrite_stream &OS, raw_pwrite_stream &DwoOS);
89} // end namespace llvm
90
91#endif // LLVM_MC_MCWINCOFFOBJECTWRITER_H
92