src/chrome/installer/test/resource_updater.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/installer/test/resource_updater.h"
   6
   7 #include <windows.h>
   8
   9 #include "base/files/file_path.h"
  10 #include "base/files/memory_mapped_file.h"
  11 #include "base/logging.h"
  12
  13 namespace upgrade_test {
  14
  15 ResourceUpdater::ResourceUpdater() : handle_(NULL) {
  16 }
  17
  18 ResourceUpdater::~ResourceUpdater() {
  19   if (handle_ != NULL) {
  20     // An update wasn't committed, so discard it.
  21     BOOL result = EndUpdateResource(handle_, TRUE);
  22     DPCHECK(result != FALSE) << "EndUpdateResource failed";
  23   }
  24 }
  25
  26 bool ResourceUpdater::Initialize(const base::FilePath& pe_image_path) {
  27   DCHECK(handle_ == NULL);
  28   handle_ = BeginUpdateResource(pe_image_path.value().c_str(), FALSE);
  29   if (handle_ == NULL) {
  30     PLOG(DFATAL)
  31       << "BeginUpdateResource failed on \"" << pe_image_path.value() << "\"";
  32     return false;
  33   }
  34   return true;
  35 }
  36
  37 bool ResourceUpdater::Update(const std::wstring& name,
  38                              const std::wstring& type,
  39                              WORD language_id,
  40                              const base::FilePath& input_file) {
  41   DCHECK(handle_ != NULL);
  42   base::MemoryMappedFile input;
  43
  44   if (input.Initialize(input_file)) {
  45     if (UpdateResource(handle_, type.c_str(), name.c_str(), language_id,
  46                        const_cast<uint8*>(input.data()), input.length())
  47         != FALSE) {
  48       return true;
  49     }
  50     PLOG(DFATAL) << "UpdateResource failed for resource \"" << name << "\"";
  51   } else {
  52     PLOG(DFATAL) << "Failed mapping \"" << input_file.value() << "\"";
  53   }
  54   return false;
  55 }
  56
  57 bool ResourceUpdater::Commit() {
  58   DCHECK(handle_ != NULL);
  59   bool result = true;
  60   if (EndUpdateResource(handle_, FALSE) == FALSE) {
  61     PLOG(DFATAL) << "EndUpdateResource failed";
  62     result = false;
  63   }
  64   handle_ = NULL;
  65   return result;
  66 }
  67
  68 }  // namespace upgrade_test