1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #ifndef COURGETTE_INSTRUCTION_UTILS_H_
6 #define COURGETTE_INSTRUCTION_UTILS_H_
10 #include "base/callback.h"
11 #include "base/macros.h"
12 #include "courgette/image_utils.h"
13 #include "courgette/memory_allocator.h"
17 // An interface to receive emitted instructions parsed from an executable.
18 class InstructionReceptor {
20 InstructionReceptor() = default;
21 virtual ~InstructionReceptor() = default;
23 // Generates an entire base relocation table.
24 virtual CheckBool EmitPeRelocs() = 0;
26 // Generates an ELF style relocation table for X86.
27 virtual CheckBool EmitElfRelocation() = 0;
29 // Generates an ELF style relocation table for ARM.
30 virtual CheckBool EmitElfARMRelocation() = 0;
32 // Following instruction will be assembled at address 'rva'.
33 virtual CheckBool EmitOrigin(RVA rva) = 0;
35 // Generates a single byte of data or machine instruction.
36 virtual CheckBool EmitSingleByte(uint8_t byte) = 0;
38 // Generates multiple bytes of data or machine instructions.
39 virtual CheckBool EmitMultipleBytes(const uint8_t* bytes, size_t len) = 0;
41 // Generates a 4-byte relative reference to address of 'label'.
42 virtual CheckBool EmitRel32(Label* label) = 0;
44 // Generates a 4-byte relative reference to address of 'label' for ARM.
45 virtual CheckBool EmitRel32ARM(uint16_t op,
47 const uint8_t* arm_op,
48 uint16_t op_size) = 0;
50 // Generates a 4-byte absolute reference to address of 'label'.
51 virtual CheckBool EmitAbs32(Label* label) = 0;
53 // Generates an 8-byte absolute reference to address of 'label'.
54 virtual CheckBool EmitAbs64(Label* label) = 0;
57 DISALLOW_COPY_AND_ASSIGN(InstructionReceptor);
60 // A rerunable callback that emit instructions to a provided receptor. Returns
61 // true on success, and false otherwise.
62 using InstructionGenerator =
63 base::RepeatingCallback<CheckBool(InstructionReceptor*)>;
65 // A counter that increments via .push_back(), so it can be passed via template
66 // to substitute std::vector<T>, to count elements instead of storing them.
68 class CountingVector {
72 void push_back(const T& /* unused */) { ++size_; }
73 size_t size() const { return size_; }
79 } // namespace courgette
81 #endif // COURGETTE_INSTRUCTION_UTILS_H_