123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122 |
- // 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 <stddef.h>
- #include "courgette/assembly_program.h"
- #include "courgette/encoded_program.h"
- #include "courgette/instruction_utils.h"
- namespace courgette {
- namespace {
- // An instruction receptor that adds each received abs32/rel32 Label* to the
- // matching VECTOR member variable. Template VECTOR allows code reuse for
- // counting (CountingVector) and storage (std::vector).
- template <template <typename T, typename... Args> class CONTAINER>
- class LabelReceptor : public InstructionReceptor {
- public:
- using VECTOR = CONTAINER<Label*>;
- LabelReceptor() = default;
- LabelReceptor(const LabelReceptor&) = delete;
- LabelReceptor& operator=(const LabelReceptor&) = delete;
- ~LabelReceptor() override = default;
- VECTOR* mutable_abs32_vector() { return &abs32_vector_; }
- VECTOR* mutable_rel32_vector() { return &rel32_vector_; }
- // InstructionReceptor:
- CheckBool EmitPeRelocs() override { return true; }
- CheckBool EmitElfRelocation() override { return true; }
- CheckBool EmitOrigin(RVA rva) override { return true; }
- CheckBool EmitSingleByte(uint8_t byte) override { return true; }
- CheckBool EmitMultipleBytes(const uint8_t* bytes, size_t len) override {
- return true;
- }
- CheckBool EmitRel32(Label* label) override {
- rel32_vector_.push_back(label);
- return true;
- }
- CheckBool EmitAbs32(Label* label) override {
- abs32_vector_.push_back(label);
- return true;
- }
- CheckBool EmitAbs64(Label* label) override {
- abs32_vector_.push_back(label);
- return true;
- }
- private:
- VECTOR abs32_vector_;
- VECTOR rel32_vector_;
- };
- } // namespace
- AssemblyProgram::AssemblyProgram(ExecutableType kind, uint64_t image_base)
- : kind_(kind), image_base_(image_base) {}
- AssemblyProgram::~AssemblyProgram() = default;
- void AssemblyProgram::PrecomputeLabels(RvaVisitor* abs32_visitor,
- RvaVisitor* rel32_visitor) {
- abs32_label_manager_.Read(abs32_visitor);
- rel32_label_manager_.Read(rel32_visitor);
- }
- void AssemblyProgram::UnassignIndexes() {
- abs32_label_manager_.UnassignIndexes();
- rel32_label_manager_.UnassignIndexes();
- }
- void AssemblyProgram::DefaultAssignIndexes() {
- abs32_label_manager_.DefaultAssignIndexes();
- rel32_label_manager_.DefaultAssignIndexes();
- }
- void AssemblyProgram::AssignRemainingIndexes() {
- abs32_label_manager_.AssignRemainingIndexes();
- rel32_label_manager_.AssignRemainingIndexes();
- }
- Label* AssemblyProgram::FindAbs32Label(RVA rva) {
- return abs32_label_manager_.Find(rva);
- }
- Label* AssemblyProgram::FindRel32Label(RVA rva) {
- return rel32_label_manager_.Find(rva);
- }
- CheckBool AssemblyProgram::AnnotateLabels(const InstructionGenerator& gen) {
- // Pass 1: Compute required space.
- LabelReceptor<CountingVector> count_receptor;
- if (!gen.Run(&count_receptor))
- return false;
- // Pass 2: Reserve and store annotations.
- LabelReceptor<std::vector> annotate_receptor;
- annotate_receptor.mutable_abs32_vector()->reserve(
- count_receptor.mutable_abs32_vector()->size());
- annotate_receptor.mutable_rel32_vector()->reserve(
- count_receptor.mutable_rel32_vector()->size());
- if (!gen.Run(&annotate_receptor))
- return false;
- // Move results to |abs32_label_annotations_| and |re32_label_annotations_|.
- abs32_label_annotations_.swap(*annotate_receptor.mutable_abs32_vector());
- rel32_label_annotations_.swap(*annotate_receptor.mutable_rel32_vector());
- return true;
- }
- bool AssemblyProgram::PrepareEncodedProgram(EncodedProgram* encoded) const {
- encoded->set_image_base(image_base_);
- return encoded->ImportLabels(abs32_label_manager_, rel32_label_manager_);
- }
- } // namespace courgette
|