| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // This file contains the code to apply a Courgette patch. | 5 // This file contains the code to apply a Courgette patch. |
| 6 | 6 |
| 7 #include "courgette/ensemble.h" | 7 #include "courgette/ensemble.h" |
| 8 | 8 |
| 9 #include "base/basictypes.h" | 9 #include "base/basictypes.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| (...skipping 118 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 129 SourceStream* transformation_parameters) { | 129 SourceStream* transformation_parameters) { |
| 130 uint32 number_of_transformations = 0; | 130 uint32 number_of_transformations = 0; |
| 131 if (!transformation_parameters->ReadVarint32(&number_of_transformations)) | 131 if (!transformation_parameters->ReadVarint32(&number_of_transformations)) |
| 132 return C_BAD_ENSEMBLE_HEADER; | 132 return C_BAD_ENSEMBLE_HEADER; |
| 133 | 133 |
| 134 for (size_t i = 0; i < number_of_transformations; ++i) { | 134 for (size_t i = 0; i < number_of_transformations; ++i) { |
| 135 uint32 kind; | 135 uint32 kind; |
| 136 if (!transformation_parameters->ReadVarint32(&kind)) | 136 if (!transformation_parameters->ReadVarint32(&kind)) |
| 137 return C_BAD_ENSEMBLE_HEADER; | 137 return C_BAD_ENSEMBLE_HEADER; |
| 138 | 138 |
| 139 if (kind == CourgettePatchFile::T_COURGETTE_WIN32_X86) { | 139 TransformationPatcher* patcher = NULL; |
| 140 TransformationPatcher* patcher = | 140 |
| 141 new CourgetteWin32X86Patcher(base_region_); | 141 switch (kind) |
| 142 { |
| 143 case CourgettePatchFile::T_COURGETTE_WIN32_X86: |
| 144 patcher = new CourgetteWin32X86Patcher(base_region_); |
| 145 break; |
| 146 } |
| 147 |
| 148 if (patcher) |
| 142 patchers_.push_back(patcher); | 149 patchers_.push_back(patcher); |
| 143 } else { | 150 else |
| 144 return C_BAD_ENSEMBLE_HEADER; | 151 return C_BAD_ENSEMBLE_HEADER; |
| 145 } | |
| 146 } | 152 } |
| 147 | 153 |
| 148 for (size_t i = 0; i < patchers_.size(); ++i) { | 154 for (size_t i = 0; i < patchers_.size(); ++i) { |
| 149 Status status = patchers_[i]->Init(transformation_parameters); | 155 Status status = patchers_[i]->Init(transformation_parameters); |
| 150 if (status != C_OK) | 156 if (status != C_OK) |
| 151 return status; | 157 return status; |
| 152 } | 158 } |
| 153 | 159 |
| 154 // All transformation_parameters should have been consumed by the above loop. | 160 // All transformation_parameters should have been consumed by the above loop. |
| 155 if (!transformation_parameters->Empty()) | 161 if (!transformation_parameters->Empty()) |
| (...skipping 252 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 408 static_cast<int>(new_sink_stream.Length())); | 414 static_cast<int>(new_sink_stream.Length())); |
| 409 if (written == -1) | 415 if (written == -1) |
| 410 return C_WRITE_OPEN_ERROR; | 416 return C_WRITE_OPEN_ERROR; |
| 411 if (static_cast<size_t>(written) != new_sink_stream.Length()) | 417 if (static_cast<size_t>(written) != new_sink_stream.Length()) |
| 412 return C_WRITE_ERROR; | 418 return C_WRITE_ERROR; |
| 413 | 419 |
| 414 return C_OK; | 420 return C_OK; |
| 415 } | 421 } |
| 416 | 422 |
| 417 } // namespace | 423 } // namespace |
| OLD | NEW |