chrome/installer/util/app_commands.cc

   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/app_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 AppCommands::AppCommands() {
  17 }
  18
  19 AppCommands::~AppCommands() {
  20 }
  21
  22 bool AppCommands::Initialize(const base::win::RegKey& key) {
  23   if (!key.Valid()) {
  24     LOG(DFATAL) << "Cannot initialize AppCommands 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   AppCommand 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 it does not appear to hold a product command.";
  48     }
  49   }
  50
  51   return true;
  52 }
  53
  54 AppCommands& AppCommands::CopyFrom(const AppCommands& other) {
  55   commands_ = other.commands_;
  56
  57   return *this;
  58 }
  59
  60 void AppCommands::Clear() {
  61   commands_.clear();
  62 }
  63
  64 bool AppCommands::Get(const std::wstring& command_id,
  65                       AppCommand* 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 AppCommands::Set(const std::wstring& command_id,
  75                       const AppCommand& 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 AppCommands::Remove(const std::wstring& command_id) {
  84   return commands_.erase(command_id) != 0;
  85 }
  86
  87 AppCommands::CommandMapRange AppCommands::GetIterators() const {
  88   return std::make_pair(commands_.begin(), commands_.end());
  89 }
  90
  91 }  // namespace installer