Index: chrome/installer/test/resource_updater.cc |
=================================================================== |
--- chrome/installer/test/resource_updater.cc (revision 0) |
+++ chrome/installer/test/resource_updater.cc (revision 0) |
@@ -0,0 +1,67 @@ |
+// Copyright (c) 2010 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. |
+ |
+#include "chrome/installer/test/resource_updater.h" |
+ |
+#include <windows.h> |
+ |
+#include "base/file_path.h" |
+#include "base/file_util.h" |
+#include "base/logging.h" |
+ |
+namespace upgrade_test { |
+ |
+ResourceUpdater::ResourceUpdater() : handle_(NULL) { |
+} |
+ |
+ResourceUpdater::~ResourceUpdater() { |
+ if (handle_ != NULL) { |
+ // An update wasn't committed, so discard it. |
+ BOOL result = EndUpdateResource(handle_, TRUE); |
+ DPCHECK(result != FALSE) << "EndUpdateResource failed"; |
+ } |
+} |
+ |
+bool ResourceUpdater::Initialize(const FilePath& pe_image_path) { |
+ DCHECK(handle_ == NULL); |
+ handle_ = BeginUpdateResource(pe_image_path.value().c_str(), FALSE); |
+ if (handle_ == NULL) { |
+ PLOG(DFATAL) |
+ << "BeginUpdateResource failed on \"" << pe_image_path.value() << "\""; |
+ return false; |
+ } |
+ return true; |
+} |
+ |
+bool ResourceUpdater::Update(const std::wstring& name, |
+ const std::wstring& type, |
+ WORD language_id, const FilePath& input_file) { |
+ DCHECK(handle_ != NULL); |
+ file_util::MemoryMappedFile input; |
+ |
+ if (input.Initialize(input_file)) { |
+ if (UpdateResource(handle_, type.c_str(), name.c_str(), language_id, |
+ const_cast<uint8*>(input.data()), input.length()) |
+ != FALSE) { |
+ return true; |
+ } |
+ PLOG(DFATAL) << "UpdateResource failed for resource \"" << name << "\""; |
+ } else { |
+ PLOG(DFATAL) << "Failed mapping \"" << input_file.value() << "\""; |
+ } |
+ return false; |
+} |
+ |
+bool ResourceUpdater::Commit() { |
+ DCHECK(handle_ != NULL); |
+ bool result = true; |
+ if (EndUpdateResource(handle_, FALSE) == FALSE) { |
+ PLOG(DFATAL) << "EndUpdateResource failed"; |
+ result = false; |
+ } |
+ handle_ = NULL; |
+ return result; |
+} |
+ |
+} // namespace upgrade_test |
Property changes on: chrome\installer\test\resource_updater.cc |
___________________________________________________________________ |
Added: svn:eol-style |
+ LF |