1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Copyright (c) 2009 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.
#ifndef COURGETTE_ENCODED_PROGRAM_H_
#define COURGETTE_ENCODED_PROGRAM_H_
#include <vector>
#include "base/basictypes.h"
#include "courgette/image_info.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 {
public:
EncodedProgram();
~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.
void DefineRel32Label(int index, RVA address);
void DefineAbs32Label(int index, RVA address);
void EndLabels();
// (3) Add instructions in the order needed to generate bytes of file.
void AddOrigin(RVA rva);
void AddCopy(int count, const void* bytes);
void AddRel32(int label_index);
void AddAbs32(int label_index);
void AddMakeRelocs();
// (3) Serialize binary assembly language tables to a set of streams.
void WriteTo(SinkStreamSet *streams);
// 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.
bool AssembleTo(SinkStream *buffer);
private:
// 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.
OP_LAST
};
void DebuggingSummary();
void GenerateBaseRelocations(SinkStream *buffer);
void DefineLabelCommon(std::vector<RVA>*, int, RVA);
void FinishLabelsCommon(std::vector<RVA>* addresses);
// Binary assembly language tables.
uint64 image_base_;
std::vector<RVA> rel32_rva_;
std::vector<RVA> abs32_rva_;
std::vector<OP> ops_;
std::vector<RVA> origins_;
std::vector<int> copy_counts_;
std::vector<uint8> copy_bytes_;
std::vector<uint32> rel32_ix_;
std::vector<uint32> abs32_ix_;
// Table of the addresses containing abs32 relocations; computed during
// assembly, used to generate base relocation table.
std::vector<uint32> abs32_relocs_;
DISALLOW_COPY_AND_ASSIGN(EncodedProgram);
};
} // namespace courgette
#endif // COURGETTE_ENCODED_PROGRAM_H_
|
