assembly_program.cc 3.7 KB

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