1//===- Writer.h -------------------------------------------------*- 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_TOOLS_OBJCOPY_COFF_WRITER_H 10#define LLVM_TOOLS_OBJCOPY_COFF_WRITER_H 11 12#include "Buffer.h" 13#include "llvm/MC/StringTableBuilder.h" 14#include "llvm/Support/Error.h" 15#include <cstddef> 16#include <utility> 17 18namespace llvm { 19namespace objcopy { 20namespace coff { 21 22struct Object; 23 24class COFFWriter { 25 Object &Obj; 26 Buffer &Buf; 27 28 size_t FileSize; 29 size_t FileAlignment; 30 size_t SizeOfInitializedData; 31 StringTableBuilder StrTabBuilder; 32 33 template <class SymbolTy> std::pair<size_t, size_t> finalizeSymbolTable(); 34 Error finalizeRelocTargets(); 35 Error finalizeSymbolContents(); 36 void layoutSections(); 37 size_t finalizeStringTable(); 38 39 Error finalize(bool IsBigObj); 40 41 void writeHeaders(bool IsBigObj); 42 void writeSections(); 43 template <class SymbolTy> void writeSymbolStringTables(); 44 45 Error write(bool IsBigObj); 46 47 Error patchDebugDirectory(); 48 49public: 50 virtual ~COFFWriter() {} 51 Error write(); 52 53 COFFWriter(Object &Obj, Buffer &Buf) 54 : Obj(Obj), Buf(Buf), StrTabBuilder(StringTableBuilder::WinCOFF) {} 55}; 56 57} // end namespace coff 58} // end namespace objcopy 59} // end namespace llvm 60 61#endif // LLVM_TOOLS_OBJCOPY_COFF_WRITER_H 62