forked from OSchip/llvm-project
86 lines
2.6 KiB
C++
86 lines
2.6 KiB
C++
//===- Writer.h -------------------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLD_COFF_WRITER_H
|
|
#define LLD_COFF_WRITER_H
|
|
|
|
#include "Chunks.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Object/COFF.h"
|
|
#include <chrono>
|
|
#include <cstdint>
|
|
#include <vector>
|
|
|
|
namespace lld {
|
|
namespace coff {
|
|
static const int PageSize = 4096;
|
|
|
|
void writeResult();
|
|
|
|
class PartialSection {
|
|
public:
|
|
PartialSection(StringRef N, uint32_t Chars)
|
|
: Name(N), Characteristics(Chars) {}
|
|
StringRef Name;
|
|
unsigned Characteristics;
|
|
std::vector<Chunk *> Chunks;
|
|
};
|
|
|
|
// OutputSection represents a section in an output file. It's a
|
|
// container of chunks. OutputSection and Chunk are 1:N relationship.
|
|
// Chunks cannot belong to more than one OutputSections. The writer
|
|
// creates multiple OutputSections and assign them unique,
|
|
// non-overlapping file offsets and RVAs.
|
|
class OutputSection {
|
|
public:
|
|
OutputSection(llvm::StringRef N, uint32_t Chars) : Name(N) {
|
|
Header.Characteristics = Chars;
|
|
}
|
|
void addChunk(Chunk *C);
|
|
void insertChunkAtStart(Chunk *C);
|
|
void merge(OutputSection *Other);
|
|
void setPermissions(uint32_t C);
|
|
uint64_t getRVA() { return Header.VirtualAddress; }
|
|
uint64_t getFileOff() { return Header.PointerToRawData; }
|
|
void writeHeaderTo(uint8_t *Buf);
|
|
void addContributingPartialSection(PartialSection *Sec);
|
|
|
|
// Returns the size of this section in an executable memory image.
|
|
// This may be smaller than the raw size (the raw size is multiple
|
|
// of disk sector size, so there may be padding at end), or may be
|
|
// larger (if that's the case, the loader reserves spaces after end
|
|
// of raw data).
|
|
uint64_t getVirtualSize() { return Header.VirtualSize; }
|
|
|
|
// Returns the size of the section in the output file.
|
|
uint64_t getRawSize() { return Header.SizeOfRawData; }
|
|
|
|
// Set offset into the string table storing this section name.
|
|
// Used only when the name is longer than 8 bytes.
|
|
void setStringTableOff(uint32_t V) { StringTableOff = V; }
|
|
|
|
// N.B. The section index is one based.
|
|
uint32_t SectionIndex = 0;
|
|
|
|
llvm::StringRef Name;
|
|
llvm::object::coff_section Header = {};
|
|
|
|
std::vector<Chunk *> Chunks;
|
|
std::vector<Chunk *> OrigChunks;
|
|
|
|
std::vector<PartialSection *> ContribSections;
|
|
|
|
private:
|
|
uint32_t StringTableOff = 0;
|
|
};
|
|
|
|
} // namespace coff
|
|
} // namespace lld
|
|
|
|
#endif
|