15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 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/util/app_commands.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/registry.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/google_update_constants.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/work_item_list.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::win::RegKey; 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace installer { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AppCommands::AppCommands() { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AppCommands::~AppCommands() { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AppCommands::Initialize(const base::win::RegKey& key) { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!key.Valid()) { 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(DFATAL) << "Cannot initialize AppCommands from an invalid key."; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) using base::win::RegistryKeyIterator; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const wchar_t kEmptyString[] = L""; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) commands_.clear(); 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RegKey cmd_key; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LONG result; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AppCommand command; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (RegistryKeyIterator key_iterator(key.Handle(), kEmptyString); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) key_iterator.Valid(); ++key_iterator) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const wchar_t* name = key_iterator.Name(); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = cmd_key.Open(key.Handle(), name, KEY_QUERY_VALUE); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result != ERROR_SUCCESS) { 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Failed to open key \"" << name 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "\" with last-error code " << result; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (command.Initialize(cmd_key)) { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) commands_[name] = command; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VLOG(1) << "Skipping over key \"" << name 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "\" as it does not appear to hold a product command."; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AppCommands& AppCommands::CopyFrom(const AppCommands& other) { 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) commands_ = other.commands_; 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return *this; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void AppCommands::Clear() { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) commands_.clear(); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AppCommands::Get(const std::wstring& command_id, 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AppCommand* command) const { 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(command); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CommandMap::const_iterator it(commands_.find(command_id)); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (it == commands_.end()) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *command = it->second; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AppCommands::Set(const std::wstring& command_id, 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const AppCommand& command) { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::pair<CommandMap::iterator, bool> result( 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) commands_.insert(std::make_pair(command_id, command))); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!result.second) 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result.first->second = command; 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result.second; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool AppCommands::Remove(const std::wstring& command_id) { 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return commands_.erase(command_id) != 0; 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)AppCommands::CommandMapRange AppCommands::GetIterators() const { 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return std::make_pair(commands_.begin(), commands_.end()); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace installer 92