1*81ad6265SDimitry Andric //===- COFFWriter.h ---------------------------------------------*- C++ -*-===// 2*81ad6265SDimitry Andric // 3*81ad6265SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*81ad6265SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*81ad6265SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*81ad6265SDimitry Andric // 7*81ad6265SDimitry Andric //===----------------------------------------------------------------------===// 8*81ad6265SDimitry Andric 9*81ad6265SDimitry Andric #ifndef LLVM_LIB_OBJCOPY_COFF_COFFWRITER_H 10*81ad6265SDimitry Andric #define LLVM_LIB_OBJCOPY_COFF_COFFWRITER_H 11*81ad6265SDimitry Andric 12*81ad6265SDimitry Andric #include "llvm/MC/StringTableBuilder.h" 13*81ad6265SDimitry Andric #include "llvm/Support/Error.h" 14*81ad6265SDimitry Andric #include "llvm/Support/MemoryBuffer.h" 15*81ad6265SDimitry Andric #include <cstddef> 16*81ad6265SDimitry Andric #include <utility> 17*81ad6265SDimitry Andric 18*81ad6265SDimitry Andric namespace llvm { 19*81ad6265SDimitry Andric namespace objcopy { 20*81ad6265SDimitry Andric namespace coff { 21*81ad6265SDimitry Andric 22*81ad6265SDimitry Andric struct Object; 23*81ad6265SDimitry Andric 24*81ad6265SDimitry Andric class COFFWriter { 25*81ad6265SDimitry Andric Object &Obj; 26*81ad6265SDimitry Andric std::unique_ptr<WritableMemoryBuffer> Buf; 27*81ad6265SDimitry Andric raw_ostream &Out; 28*81ad6265SDimitry Andric 29*81ad6265SDimitry Andric size_t FileSize; 30*81ad6265SDimitry Andric size_t FileAlignment; 31*81ad6265SDimitry Andric size_t SizeOfInitializedData; 32*81ad6265SDimitry Andric StringTableBuilder StrTabBuilder; 33*81ad6265SDimitry Andric 34*81ad6265SDimitry Andric template <class SymbolTy> std::pair<size_t, size_t> finalizeSymbolTable(); 35*81ad6265SDimitry Andric Error finalizeRelocTargets(); 36*81ad6265SDimitry Andric Error finalizeSymbolContents(); 37*81ad6265SDimitry Andric void layoutSections(); 38*81ad6265SDimitry Andric Expected<size_t> finalizeStringTable(); 39*81ad6265SDimitry Andric 40*81ad6265SDimitry Andric Error finalize(bool IsBigObj); 41*81ad6265SDimitry Andric 42*81ad6265SDimitry Andric void writeHeaders(bool IsBigObj); 43*81ad6265SDimitry Andric void writeSections(); 44*81ad6265SDimitry Andric template <class SymbolTy> void writeSymbolStringTables(); 45*81ad6265SDimitry Andric 46*81ad6265SDimitry Andric Error write(bool IsBigObj); 47*81ad6265SDimitry Andric 48*81ad6265SDimitry Andric Error patchDebugDirectory(); 49*81ad6265SDimitry Andric Expected<uint32_t> virtualAddressToFileAddress(uint32_t RVA); 50*81ad6265SDimitry Andric 51*81ad6265SDimitry Andric public: ~COFFWriter()52*81ad6265SDimitry Andric virtual ~COFFWriter() {} 53*81ad6265SDimitry Andric Error write(); 54*81ad6265SDimitry Andric COFFWriter(Object & Obj,raw_ostream & Out)55*81ad6265SDimitry Andric COFFWriter(Object &Obj, raw_ostream &Out) 56*81ad6265SDimitry Andric : Obj(Obj), Out(Out), StrTabBuilder(StringTableBuilder::WinCOFF) {} 57*81ad6265SDimitry Andric }; 58*81ad6265SDimitry Andric 59*81ad6265SDimitry Andric } // end namespace coff 60*81ad6265SDimitry Andric } // end namespace objcopy 61*81ad6265SDimitry Andric } // end namespace llvm 62*81ad6265SDimitry Andric 63*81ad6265SDimitry Andric #endif // LLVM_LIB_OBJCOPY_COFF_COFFWRITER_H 64