courgette/assembly_program.cc

   1 // Copyright 2011 The Chromium Authors
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <stddef.h>
   6
   7 #include "courgette/assembly_program.h"
   8
   9 #include "courgette/encoded_program.h"
  10 #include "courgette/instruction_utils.h"
  11
  12 namespace courgette {
  13
  14 namespace {
  15
  16 // An instruction receptor that adds each received abs32/rel32 Label* to the
  17 // matching VECTOR member variable. Template VECTOR allows code reuse for
  18 // counting (CountingVector) and storage (std::vector).
  19 template <template <typename T, typename... Args> class CONTAINER>
  20 class LabelReceptor : public InstructionReceptor {
  21  public:
  22   using VECTOR = CONTAINER<Label*>;
  23
  24   LabelReceptor() = default;
  25
  26   LabelReceptor(const LabelReceptor&) = delete;
  27   LabelReceptor& operator=(const LabelReceptor&) = delete;
  28
  29   ~LabelReceptor() override = default;
  30
  31   VECTOR* mutable_abs32_vector() { return &abs32_vector_; }
  32   VECTOR* mutable_rel32_vector() { return &rel32_vector_; }
  33
  34   // InstructionReceptor:
  35   CheckBool EmitPeRelocs() override { return true; }
  36   CheckBool EmitElfRelocation() override { return true; }
  37   CheckBool EmitOrigin(RVA rva) override { return true; }
  38   CheckBool EmitSingleByte(uint8_t byte) override { return true; }
  39   CheckBool EmitMultipleBytes(const uint8_t* bytes, size_t len) override {
  40     return true;
  41   }
  42   CheckBool EmitRel32(Label* label) override {
  43     rel32_vector_.push_back(label);
  44     return true;
  45   }
  46   CheckBool EmitAbs32(Label* label) override {
  47     abs32_vector_.push_back(label);
  48     return true;
  49   }
  50   CheckBool EmitAbs64(Label* label) override {
  51     abs32_vector_.push_back(label);
  52     return true;
  53   }
  54
  55  private:
  56   VECTOR abs32_vector_;
  57   VECTOR rel32_vector_;
  58 };
  59
  60 }  // namespace
  61
  62 AssemblyProgram::AssemblyProgram(ExecutableType kind, uint64_t image_base)
  63     : kind_(kind), image_base_(image_base) {}
  64
  65 AssemblyProgram::~AssemblyProgram() = default;
  66
  67 void AssemblyProgram::PrecomputeLabels(RvaVisitor* abs32_visitor,
  68                                        RvaVisitor* rel32_visitor) {
  69   abs32_label_manager_.Read(abs32_visitor);
  70   rel32_label_manager_.Read(rel32_visitor);
  71 }
  72
  73 void AssemblyProgram::UnassignIndexes() {
  74   abs32_label_manager_.UnassignIndexes();
  75   rel32_label_manager_.UnassignIndexes();
  76 }
  77
  78 void AssemblyProgram::DefaultAssignIndexes() {
  79   abs32_label_manager_.DefaultAssignIndexes();
  80   rel32_label_manager_.DefaultAssignIndexes();
  81 }
  82
  83 void AssemblyProgram::AssignRemainingIndexes() {
  84   abs32_label_manager_.AssignRemainingIndexes();
  85   rel32_label_manager_.AssignRemainingIndexes();
  86 }
  87
  88 Label* AssemblyProgram::FindAbs32Label(RVA rva) {
  89   return abs32_label_manager_.Find(rva);
  90 }
  91
  92 Label* AssemblyProgram::FindRel32Label(RVA rva) {
  93   return rel32_label_manager_.Find(rva);
  94 }
  95
  96 CheckBool AssemblyProgram::AnnotateLabels(const InstructionGenerator& gen) {
  97   // Pass 1: Compute required space.
  98   LabelReceptor<CountingVector> count_receptor;
  99   if (!gen.Run(&count_receptor))
 100     return false;
 101
 102   // Pass 2: Reserve and store annotations.
 103   LabelReceptor<std::vector> annotate_receptor;
 104   annotate_receptor.mutable_abs32_vector()->reserve(
 105       count_receptor.mutable_abs32_vector()->size());
 106   annotate_receptor.mutable_rel32_vector()->reserve(
 107       count_receptor.mutable_rel32_vector()->size());
 108   if (!gen.Run(&annotate_receptor))
 109     return false;
 110
 111   // Move results to |abs32_label_annotations_| and |re32_label_annotations_|.
 112   abs32_label_annotations_.swap(*annotate_receptor.mutable_abs32_vector());
 113   rel32_label_annotations_.swap(*annotate_receptor.mutable_rel32_vector());
 114   return true;
 115 }
 116
 117 bool AssemblyProgram::PrepareEncodedProgram(EncodedProgram* encoded) const {
 118   encoded->set_image_base(image_base_);
 119   return encoded->ImportLabels(abs32_label_manager_, rel32_label_manager_);
 120 }
 121
 122 }  // namespace courgette