diff options
author | sra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2009-05-08 23:00:29 +0000 |
---|---|---|
committer | sra@chromium.org <sra@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2009-05-08 23:00:29 +0000 |
commit | 04ca1bc65afb76ea30698c25f24599d20119e3d2 (patch) | |
tree | 2bb65e974d478f4d607b83f6a84a56c01f501ac0 /courgette/win32_x86_patcher.h | |
parent | 9adf1dcf3281408560267787eddcc767566b425f (diff) | |
download | chromium_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.h | 95 |
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_ |