courgette/patcher_x86_32.h

   1 // Copyright (c) 2012 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 // This is the transformation for Windows X86 executables.
   6
   7 #ifndef COURGETTE_WIN32_X86_PATCHER_H_
   8 #define COURGETTE_WIN32_X86_PATCHER_H_
   9
  10 #include "courgette/ensemble.h"
  11
  12 namespace courgette {
  13
  14 // Courgette32X86Patcher is a TransformationPatcher for Windows 32-bit
  15 // executables.
  16 //
  17 class PatcherX86_32 : public TransformationPatcher {
  18  public:
  19   explicit PatcherX86_32(const Region& region)
  20       : ensemble_region_(region),
  21         base_offset_(0),
  22         base_length_(0) {
  23   }
  24
  25   Status Init(SourceStream* parameter_stream) {
  26     if (!parameter_stream->ReadVarint32(&base_offset_))
  27       return C_BAD_TRANSFORM;
  28     if (!parameter_stream->ReadVarint32(&base_length_))
  29       return C_BAD_TRANSFORM;
  30
  31     if (base_offset_ > ensemble_region_.length())
  32       return C_BAD_TRANSFORM;
  33     if (base_length_ > ensemble_region_.length() - base_offset_)
  34       return C_BAD_TRANSFORM;
  35
  36     return C_OK;
  37   }
  38
  39   Status PredictTransformParameters(SinkStreamSet* predicted_parameters) {
  40     // No code needed to write an 'empty' predicted parameter set.
  41     return C_OK;
  42   }
  43
  44   Status Transform(SourceStreamSet* corrected_parameters,
  45                    SinkStreamSet* transformed_element) {
  46     Status status;
  47     if (!corrected_parameters->Empty())
  48       return C_GENERAL_ERROR;   // Don't expect any corrected parameters.
  49
  50     AssemblyProgram* program = NULL;
  51     status = ParseDetectedExecutable(ensemble_region_.start() + base_offset_,
  52                                      base_length_,
  53                                      &program);
  54     if (status != C_OK)
  55       return status;
  56
  57     EncodedProgram* encoded = NULL;
  58     status = Encode(program, &encoded);
  59     DeleteAssemblyProgram(program);
  60     if (status != C_OK)
  61       return status;
  62
  63     status = WriteEncodedProgram(encoded, transformed_element);
  64     DeleteEncodedProgram(encoded);
  65
  66     return status;
  67   }
  68
  69   Status Reform(SourceStreamSet* transformed_element,
  70                 SinkStream* reformed_element) {
  71     Status status;
  72     EncodedProgram* encoded_program = NULL;
  73     status = ReadEncodedProgram(transformed_element, &encoded_program);
  74     if (status != C_OK)
  75       return status;
  76
  77     status = Assemble(encoded_program, reformed_element);
  78     DeleteEncodedProgram(encoded_program);
  79     if (status != C_OK)
  80       return status;
  81
  82     return C_OK;
  83   }
  84
  85  private:
  86   Region ensemble_region_;
  87
  88   uint32 base_offset_;
  89   uint32 base_length_;
  90
  91   DISALLOW_COPY_AND_ASSIGN(PatcherX86_32);
  92 };
  93
  94 }  // namespace
  95 #endif  // COURGETTE_WIN32_X86_PATCHER_H_