Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1960)

Unified Diff: chrome/installer/test/resource_updater.cc

Issue 5236002: Add infrastructure enabling tests of installer upgrade scenarios.... (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src/
Patch Set: '' Created 10 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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

Powered by Google App Engine
This is Rietveld 408576698