summaryrefslogtreecommitdiffstats
path: root/courgette/win32_x86_patcher.h
diff options
context:
space:
mode:
authorsra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-05-08 23:00:29 +0000
committersra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>2009-05-08 23:00:29 +0000
commit04ca1bc65afb76ea30698c25f24599d20119e3d2 (patch)
tree2bb65e974d478f4d607b83f6a84a56c01f501ac0 /courgette/win32_x86_patcher.h
parent9adf1dcf3281408560267787eddcc767566b425f (diff)
downloadchromium_src-04ca1bc65afb76ea30698c25f24599d20119e3d2.zip
chromium_src-04ca1bc65afb76ea30698c25f24599d20119e3d2.tar.gz
chromium_src-04ca1bc65afb76ea30698c25f24599d20119e3d2.tar.bz2
Move Courgette
from src\third_party\courgette to src\courgette and src\courgette\third_party Fixed #includes Added properties to ignore generated files: C:\c5\src>svn pg svn:ignore courgette courgette.xcodeproj courgette.sln courgette_fuzz.vcproj courgette_lib.vcproj courgette_minimal_tool.vcproj courgette_tool.vcproj courgette.vcproj courgette_unittests.vcproj SConstruct courgette_fuzz.scons courgette_lib.scons courgette_main.scons courgette_minimal_tool.scons courgette.scons courgette_tool.scons courgette_unittests.scons Review URL: http://codereview.chromium.org/115062 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@15692 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'courgette/win32_x86_patcher.h')
-rw-r--r--courgette/win32_x86_patcher.h95
1 files changed, 95 insertions, 0 deletions
diff --git a/courgette/win32_x86_patcher.h b/courgette/win32_x86_patcher.h
new file mode 100644
index 0000000..9ab53d9
--- /dev/null
+++ b/courgette/win32_x86_patcher.h
@@ -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 "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_