blob: b1203535f25b1296cb45ab8679d78ee51c3cb782 [file] [log] [blame]
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <vector>
#include "base/basictypes.h"
#include "courgette/disassembler.h"
#include "courgette/memory_allocator.h"
namespace courgette {
class SinkStream;
class SinkStreamSet;
class SourceStreamSet;
// An EncodedProgram is a set of tables that contain a simple 'binary assembly
// language' that can be assembled to produce a sequence of bytes, for example,
// a Windows 32-bit executable.
class EncodedProgram {
// Generating an EncodedProgram:
// (1) The image base can be specified at any time.
void set_image_base(uint64 base) { image_base_ = base; }
// (2) Address tables and indexes defined first.
CheckBool DefineRel32Label(int index, RVA address) WARN_UNUSED_RESULT;
CheckBool DefineAbs32Label(int index, RVA address) WARN_UNUSED_RESULT;
void EndLabels();
// (3) Add instructions in the order needed to generate bytes of file.
// NOTE: If any of these methods ever fail, the EncodedProgram instance
// has failed and should be discarded.
CheckBool AddOrigin(RVA rva) WARN_UNUSED_RESULT;
CheckBool AddCopy(uint32 count, const void* bytes) WARN_UNUSED_RESULT;
CheckBool AddRel32(int label_index) WARN_UNUSED_RESULT;
CheckBool AddAbs32(int label_index) WARN_UNUSED_RESULT;
CheckBool AddMakeRelocs() WARN_UNUSED_RESULT;
// (3) Serialize binary assembly language tables to a set of streams.
CheckBool WriteTo(SinkStreamSet* streams) WARN_UNUSED_RESULT;
// Using an EncodedProgram to generate a byte stream:
// (4) Deserializes a fresh EncodedProgram from a set of streams.
bool ReadFrom(SourceStreamSet* streams);
// (5) Assembles the 'binary assembly language' into final file.
CheckBool AssembleTo(SinkStream* buffer) WARN_UNUSED_RESULT;
// Binary assembly language operations.
enum OP {
ORIGIN, // ORIGIN <rva> - set address for subsequent assembly.
COPY, // COPY <count> <bytes> - copy bytes to output.
COPY1, // COPY1 <byte> - same as COPY 1 <byte>.
REL32, // REL32 <index> - emit rel32 encoded reference to address at
// address table offset <index>
ABS32, // ABS32 <index> - emit abs32 encoded reference to address at
// address table offset <index>
MAKE_BASE_RELOCATION_TABLE, // Emit base relocation table blocks.
typedef NoThrowBuffer<RVA> RvaVector;
typedef NoThrowBuffer<uint32> UInt32Vector;
typedef NoThrowBuffer<uint8> UInt8Vector;
typedef NoThrowBuffer<OP> OPVector;
void DebuggingSummary();
CheckBool GenerateBaseRelocations(SinkStream *buffer) WARN_UNUSED_RESULT;
CheckBool DefineLabelCommon(RvaVector*, int, RVA) WARN_UNUSED_RESULT;
void FinishLabelsCommon(RvaVector* addresses);
// Binary assembly language tables.
uint64 image_base_;
RvaVector rel32_rva_;
RvaVector abs32_rva_;
OPVector ops_;
RvaVector origins_;
UInt32Vector copy_counts_;
UInt8Vector copy_bytes_;
UInt32Vector rel32_ix_;
UInt32Vector abs32_ix_;
// Table of the addresses containing abs32 relocations; computed during
// assembly, used to generate base relocation table.
UInt32Vector abs32_relocs_;
} // namespace courgette