15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2010 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/test/resource_updater.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/memory_mapped_file.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace upgrade_test { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ResourceUpdater::ResourceUpdater() : handle_(NULL) { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ResourceUpdater::~ResourceUpdater() { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (handle_ != NULL) { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // An update wasn't committed, so discard it. 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BOOL result = EndUpdateResource(handle_, TRUE); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DPCHECK(result != FALSE) << "EndUpdateResource failed"; 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool ResourceUpdater::Initialize(const base::FilePath& pe_image_path) { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(handle_ == NULL); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handle_ = BeginUpdateResource(pe_image_path.value().c_str(), FALSE); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (handle_ == NULL) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(DFATAL) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "BeginUpdateResource failed on \"" << pe_image_path.value() << "\""; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ResourceUpdater::Update(const std::wstring& name, 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::wstring& type, 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) WORD language_id, 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& input_file) { 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(handle_ != NULL); 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::MemoryMappedFile input; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (input.Initialize(input_file)) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (UpdateResource(handle_, type.c_str(), name.c_str(), language_id, 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const_cast<uint8*>(input.data()), input.length()) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) != FALSE) { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(DFATAL) << "UpdateResource failed for resource \"" << name << "\""; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(DFATAL) << "Failed mapping \"" << input_file.value() << "\""; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ResourceUpdater::Commit() { 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(handle_ != NULL); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool result = true; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (EndUpdateResource(handle_, FALSE) == FALSE) { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(DFATAL) << "EndUpdateResource failed"; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = false; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handle_ = NULL; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace upgrade_test 69