| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 #include "chrome/installer/test/resource_updater.h" | 5 #include "chrome/installer/test/resource_updater.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 | 8 |
| 9 #include "base/file_util.h" | |
| 10 #include "base/files/file_path.h" | 9 #include "base/files/file_path.h" |
| 10 #include "base/files/memory_mapped_file.h" |
| 11 #include "base/logging.h" | 11 #include "base/logging.h" |
| 12 | 12 |
| 13 namespace upgrade_test { | 13 namespace upgrade_test { |
| 14 | 14 |
| 15 ResourceUpdater::ResourceUpdater() : handle_(NULL) { | 15 ResourceUpdater::ResourceUpdater() : handle_(NULL) { |
| 16 } | 16 } |
| 17 | 17 |
| 18 ResourceUpdater::~ResourceUpdater() { | 18 ResourceUpdater::~ResourceUpdater() { |
| 19 if (handle_ != NULL) { | 19 if (handle_ != NULL) { |
| 20 // An update wasn't committed, so discard it. | 20 // An update wasn't committed, so discard it. |
| (...skipping 11 matching lines...) Expand all Loading... |
| 32 return false; | 32 return false; |
| 33 } | 33 } |
| 34 return true; | 34 return true; |
| 35 } | 35 } |
| 36 | 36 |
| 37 bool ResourceUpdater::Update(const std::wstring& name, | 37 bool ResourceUpdater::Update(const std::wstring& name, |
| 38 const std::wstring& type, | 38 const std::wstring& type, |
| 39 WORD language_id, | 39 WORD language_id, |
| 40 const base::FilePath& input_file) { | 40 const base::FilePath& input_file) { |
| 41 DCHECK(handle_ != NULL); | 41 DCHECK(handle_ != NULL); |
| 42 file_util::MemoryMappedFile input; | 42 base::MemoryMappedFile input; |
| 43 | 43 |
| 44 if (input.Initialize(input_file)) { | 44 if (input.Initialize(input_file)) { |
| 45 if (UpdateResource(handle_, type.c_str(), name.c_str(), language_id, | 45 if (UpdateResource(handle_, type.c_str(), name.c_str(), language_id, |
| 46 const_cast<uint8*>(input.data()), input.length()) | 46 const_cast<uint8*>(input.data()), input.length()) |
| 47 != FALSE) { | 47 != FALSE) { |
| 48 return true; | 48 return true; |
| 49 } | 49 } |
| 50 PLOG(DFATAL) << "UpdateResource failed for resource \"" << name << "\""; | 50 PLOG(DFATAL) << "UpdateResource failed for resource \"" << name << "\""; |
| 51 } else { | 51 } else { |
| 52 PLOG(DFATAL) << "Failed mapping \"" << input_file.value() << "\""; | 52 PLOG(DFATAL) << "Failed mapping \"" << input_file.value() << "\""; |
| 53 } | 53 } |
| 54 return false; | 54 return false; |
| 55 } | 55 } |
| 56 | 56 |
| 57 bool ResourceUpdater::Commit() { | 57 bool ResourceUpdater::Commit() { |
| 58 DCHECK(handle_ != NULL); | 58 DCHECK(handle_ != NULL); |
| 59 bool result = true; | 59 bool result = true; |
| 60 if (EndUpdateResource(handle_, FALSE) == FALSE) { | 60 if (EndUpdateResource(handle_, FALSE) == FALSE) { |
| 61 PLOG(DFATAL) << "EndUpdateResource failed"; | 61 PLOG(DFATAL) << "EndUpdateResource failed"; |
| 62 result = false; | 62 result = false; |
| 63 } | 63 } |
| 64 handle_ = NULL; | 64 handle_ = NULL; |
| 65 return result; | 65 return result; |
| 66 } | 66 } |
| 67 | 67 |
| 68 } // namespace upgrade_test | 68 } // namespace upgrade_test |
| OLD | NEW |