OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2011 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/util/product_commands.h" |
| 6 |
| 7 #include "base/logging.h" |
| 8 #include "base/win/registry.h" |
| 9 #include "chrome/installer/util/google_update_constants.h" |
| 10 #include "chrome/installer/util/work_item_list.h" |
| 11 |
| 12 using base::win::RegKey; |
| 13 |
| 14 namespace installer { |
| 15 |
| 16 ProductCommands::ProductCommands() { |
| 17 } |
| 18 |
| 19 ProductCommands::~ProductCommands() { |
| 20 } |
| 21 |
| 22 bool ProductCommands::Initialize(const base::win::RegKey& key) { |
| 23 if (!key.Valid()) { |
| 24 LOG(DFATAL) << "Cannot initialize ProductCommands from an invalid key."; |
| 25 return false; |
| 26 } |
| 27 |
| 28 using base::win::RegistryKeyIterator; |
| 29 static const wchar_t kEmptyString[] = L""; |
| 30 |
| 31 commands_.clear(); |
| 32 |
| 33 RegKey cmd_key; |
| 34 LONG result; |
| 35 ProductCommand command; |
| 36 for (RegistryKeyIterator key_iterator(key.Handle(), kEmptyString); |
| 37 key_iterator.Valid(); ++key_iterator) { |
| 38 const wchar_t* name = key_iterator.Name(); |
| 39 result = cmd_key.Open(key.Handle(), name, KEY_QUERY_VALUE); |
| 40 if (result != ERROR_SUCCESS) { |
| 41 LOG(ERROR) << "Failed to open key \"" << name |
| 42 << "\" with last-error code " << result; |
| 43 } else if (command.Initialize(cmd_key)) { |
| 44 commands_[name] = command; |
| 45 } else { |
| 46 VLOG(1) << "Skipping over key \"" << name |
| 47 << "\" as is does not appear to hold a product command."; |
| 48 } |
| 49 } |
| 50 |
| 51 return true; |
| 52 } |
| 53 |
| 54 ProductCommands& ProductCommands::CopyFrom(const ProductCommands& other) { |
| 55 commands_ = other.commands_; |
| 56 |
| 57 return *this; |
| 58 } |
| 59 |
| 60 void ProductCommands::Clear() { |
| 61 commands_.clear(); |
| 62 } |
| 63 |
| 64 bool ProductCommands::Get(const std::wstring& command_id, |
| 65 ProductCommand* command) const { |
| 66 DCHECK(command); |
| 67 CommandMap::const_iterator it(commands_.find(command_id)); |
| 68 if (it == commands_.end()) |
| 69 return false; |
| 70 *command = it->second; |
| 71 return true; |
| 72 } |
| 73 |
| 74 bool ProductCommands::Set(const std::wstring& command_id, |
| 75 const ProductCommand& command) { |
| 76 std::pair<CommandMap::iterator, bool> result( |
| 77 commands_.insert(std::make_pair(command_id, command))); |
| 78 if (!result.second) |
| 79 result.first->second = command; |
| 80 return result.second; |
| 81 } |
| 82 |
| 83 bool ProductCommands::Remove(const std::wstring& command_id) { |
| 84 return commands_.erase(command_id) != 0; |
| 85 } |
| 86 |
| 87 ProductCommands::CommandMapRange ProductCommands::GetIterators() const { |
| 88 return std::make_pair(commands_.begin(), commands_.end()); |
| 89 } |
| 90 |
| 91 } // namespace installer |
OLD | NEW |