courgette/encoded_program.h

   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
   5 #ifndef COURGETTE_ENCODED_PROGRAM_H_
   6 #define COURGETTE_ENCODED_PROGRAM_H_
   7
   8 #include <vector>
   9
  10 #include "base/basictypes.h"
  11 #include "courgette/disassembler.h"
  12 #include "courgette/memory_allocator.h"
  13 #include "courgette/types_elf.h"
  14
  15 namespace courgette {
  16
  17 class SinkStream;
  18 class SinkStreamSet;
  19 class SourceStreamSet;
  20
  21 // An EncodedProgram is a set of tables that contain a simple 'binary assembly
  22 // language' that can be assembled to produce a sequence of bytes, for example,
  23 // a Windows 32-bit executable.
  24 //
  25 class EncodedProgram {
  26  public:
  27   EncodedProgram();
  28   ~EncodedProgram();
  29
  30   // Generating an EncodedProgram:
  31   //
  32   // (1) The image base can be specified at any time.
  33   void set_image_base(uint64 base) { image_base_ = base; }
  34
  35   // (2) Address tables and indexes defined first.
  36   CheckBool DefineRel32Label(int index, RVA address) WARN_UNUSED_RESULT;
  37   CheckBool DefineAbs32Label(int index, RVA address) WARN_UNUSED_RESULT;
  38   void EndLabels();
  39
  40   // (3) Add instructions in the order needed to generate bytes of file.
  41   // NOTE: If any of these methods ever fail, the EncodedProgram instance
  42   // has failed and should be discarded.
  43   CheckBool AddOrigin(RVA rva) WARN_UNUSED_RESULT;
  44   CheckBool AddCopy(size_t count, const void* bytes) WARN_UNUSED_RESULT;
  45   CheckBool AddRel32(int label_index) WARN_UNUSED_RESULT;
  46   CheckBool AddRel32ARM(uint16 op, int label_index) WARN_UNUSED_RESULT;
  47   CheckBool AddAbs32(int label_index) WARN_UNUSED_RESULT;
  48   CheckBool AddAbs64(int label_index) WARN_UNUSED_RESULT;
  49   CheckBool AddPeMakeRelocs(ExecutableType kind) WARN_UNUSED_RESULT;
  50   CheckBool AddElfMakeRelocs() WARN_UNUSED_RESULT;
  51   CheckBool AddElfARMMakeRelocs() WARN_UNUSED_RESULT;
  52
  53   // (3) Serialize binary assembly language tables to a set of streams.
  54   CheckBool WriteTo(SinkStreamSet* streams) WARN_UNUSED_RESULT;
  55
  56   // Using an EncodedProgram to generate a byte stream:
  57   //
  58   // (4) Deserializes a fresh EncodedProgram from a set of streams.
  59   bool ReadFrom(SourceStreamSet* streams);
  60
  61   // (5) Assembles the 'binary assembly language' into final file.
  62   CheckBool AssembleTo(SinkStream* buffer) WARN_UNUSED_RESULT;
  63
  64  private:
  65   // Binary assembly language operations.
  66   // These are part of the patch format. Reusing an existing value will
  67   // break backwards compatibility.
  68   enum OP {
  69     ORIGIN = 0,    // ORIGIN <rva> - set address for subsequent assembly.
  70     COPY = 1,      // COPY <count> <bytes> - copy bytes to output.
  71     COPY1 = 2,     // COPY1 <byte> - same as COPY 1 <byte>.
  72     REL32 = 3,     // REL32 <index> - emit rel32 encoded reference to address at
  73                    // address table offset <index>
  74     ABS32 = 4,     // ABS32 <index> - emit abs32 encoded reference to address at
  75                    // address table offset <index>
  76     MAKE_PE_RELOCATION_TABLE = 5,  // Emit PE base relocation table blocks.
  77     MAKE_ELF_RELOCATION_TABLE = 6, // Emit Elf relocation table for X86
  78     MAKE_ELF_ARM_RELOCATION_TABLE = 7, // Emit Elf relocation table for ARM
  79     MAKE_PE64_RELOCATION_TABLE = 8, // Emit PE64 base relocation table blocks.
  80     ABS64 = 9,     // ABS64 <index> - emit abs64 encoded reference to address at
  81                    // address table offset <index>
  82     // ARM reserves 0x1000-LAST_ARM, bits 13-16 define the opcode
  83     // subset, and 1-12 are the compressed ARM op.
  84     REL32ARM8   = 0x1000,
  85     REL32ARM11  = 0x2000,
  86     REL32ARM24  = 0x3000,
  87     REL32ARM25  = 0x4000,
  88     REL32ARM21  = 0x5000,
  89     LAST_ARM    = 0x5FFF,
  90   };
  91
  92   typedef NoThrowBuffer<RVA> RvaVector;
  93   typedef NoThrowBuffer<size_t> SizeTVector;
  94   typedef NoThrowBuffer<uint32> UInt32Vector;
  95   typedef NoThrowBuffer<uint8> UInt8Vector;
  96   typedef NoThrowBuffer<OP> OPVector;
  97
  98   void DebuggingSummary();
  99   CheckBool GeneratePeRelocations(SinkStream *buffer,
 100                                   uint8 type) WARN_UNUSED_RESULT;
 101   CheckBool GenerateElfRelocations(Elf32_Word pending_elf_relocation_table,
 102                                    SinkStream *buffer) WARN_UNUSED_RESULT;
 103   CheckBool DefineLabelCommon(RvaVector*, int, RVA) WARN_UNUSED_RESULT;
 104   void FinishLabelsCommon(RvaVector* addresses);
 105
 106   // Decodes and evaluates courgette ops for ARM rel32 addresses.
 107   CheckBool EvaluateRel32ARM(OP op, size_t& ix_rel32_ix, RVA& current_rva,
 108                              SinkStream* output);
 109
 110   // Binary assembly language tables.
 111   uint64 image_base_;
 112   RvaVector rel32_rva_;
 113   RvaVector abs32_rva_;
 114   OPVector ops_;
 115   RvaVector origins_;
 116   SizeTVector copy_counts_;
 117   UInt8Vector copy_bytes_;
 118   UInt32Vector rel32_ix_;
 119   UInt32Vector abs32_ix_;
 120
 121   // Table of the addresses containing abs32 relocations; computed during
 122   // assembly, used to generate base relocation table.
 123   UInt32Vector abs32_relocs_;
 124
 125   DISALLOW_COPY_AND_ASSIGN(EncodedProgram);
 126 };
 127
 128 }  // namespace courgette
 129 #endif  // COURGETTE_ENCODED_PROGRAM_H_