Index: third_party/courgette/win32_x86_patcher.h |
=================================================================== |
--- third_party/courgette/win32_x86_patcher.h (revision 0) |
+++ third_party/courgette/win32_x86_patcher.h (revision 0) |
@@ -0,0 +1,95 @@ |
+// Copyright (c) 2009 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. |
+ |
+// This is the transformation for Windows X86 executables. |
+ |
+#ifndef COURGETTE_WIN32_X86_PATCHER_H_ |
+#define COURGETTE_WIN32_X86_PATCHER_H_ |
+ |
+#include "third_party/courgette/ensemble.h" |
+ |
+namespace courgette { |
+ |
+// CourgetteWin32X86Patcher is a TransformationPatcher for Windows 32-bit |
+// executables. |
+// |
+class CourgetteWin32X86Patcher : public TransformationPatcher { |
+ public: |
+ explicit CourgetteWin32X86Patcher(const Region& region) |
+ : ensemble_region_(region) { |
+ } |
+ |
+ Status Init(SourceStream* parameter_stream) { |
+ if (!parameter_stream->ReadVarint32(&base_offset_)) |
+ return C_BAD_TRANSFORM; |
+ if (!parameter_stream->ReadVarint32(&base_length_)) |
+ return C_BAD_TRANSFORM; |
+ |
+ if (base_offset_ > ensemble_region_.length()) |
+ return C_BAD_TRANSFORM; |
+ if (base_length_ > ensemble_region_.length() - base_offset_) |
+ return C_BAD_TRANSFORM; |
+ |
+ return C_OK; |
+ } |
+ |
+ Status PredictTransformParameters(SinkStreamSet* predicted_parameters) { |
+ // No code needed to write an 'empty' predicted parameter set. |
+ return C_OK; |
+ } |
+ |
+ Status Transform(SourceStreamSet* corrected_parameters, |
+ SinkStreamSet* transformed_element) { |
+ Status status; |
+ if (!corrected_parameters->Empty()) |
+ return C_GENERAL_ERROR; // Don't expect any corrected parameters. |
+ |
+ AssemblyProgram* program = NULL; |
+ status = ParseWin32X86PE(ensemble_region_.start() + base_offset_, |
+ base_length_, |
+ &program); |
+ if (status != C_OK) |
+ return status; |
+ |
+ EncodedProgram* encoded = NULL; |
+ status = Encode(program, &encoded); |
+ DeleteAssemblyProgram(program); |
+ if (status != C_OK) |
+ return status; |
+ |
+ status = WriteEncodedProgram(encoded, transformed_element); |
+ DeleteEncodedProgram(encoded); |
+ if (status != C_OK) |
+ return status; |
+ |
+ return status; |
+ } |
+ |
+ Status Reform(SourceStreamSet* transformed_element, |
+ SinkStream* reformed_element) { |
+ Status status; |
+ EncodedProgram* encoded_program = NULL; |
+ status = ReadEncodedProgram(transformed_element, &encoded_program); |
+ if (status != C_OK) |
+ return status; |
+ |
+ status = Assemble(encoded_program, reformed_element); |
+ DeleteEncodedProgram(encoded_program); |
+ if (status != C_OK) |
+ return status; |
+ |
+ return C_OK; |
+ } |
+ |
+ private: |
+ Region ensemble_region_; |
+ |
+ uint32 base_offset_; |
+ uint32 base_length_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(CourgetteWin32X86Patcher); |
+}; |
+ |
+} // namespace |
+#endif // COURGETTE_WIN32_X86_PATCHER_H_ |