src/chrome/installer/util/set_reg_value_work_item.h

   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 #ifndef CHROME_INSTALLER_UTIL_SET_REG_VALUE_WORK_ITEM_H__
   6 #define CHROME_INSTALLER_UTIL_SET_REG_VALUE_WORK_ITEM_H__
   7
   8 #include <windows.h>
   9
  10 #include <string>
  11 #include <vector>
  12
  13 #include "chrome/installer/util/work_item.h"
  14
  15 // A WorkItem subclass that sets a registry value with REG_SZ, REG_DWORD, or
  16 // REG_QWORD type at the specified path. The value is only set if the target key
  17 // exists.
  18 class SetRegValueWorkItem : public WorkItem {
  19  public:
  20   virtual ~SetRegValueWorkItem();
  21
  22   virtual bool Do();
  23
  24   virtual void Rollback();
  25
  26  private:
  27   friend class WorkItem;
  28
  29   enum SettingStatus {
  30     // The status before Do is called.
  31     SET_VALUE,
  32     // One possible outcome after Do(). A new value is created under the key.
  33     NEW_VALUE_CREATED,
  34     // One possible outcome after Do(). The previous value under the key has
  35     // been overwritten.
  36     VALUE_OVERWRITTEN,
  37     // One possible outcome after Do(). No change is applied, either
  38     // because we are not allowed to overwrite the previous value, or due to
  39     // some errors like the key does not exist.
  40     VALUE_UNCHANGED,
  41     // The status after Do and Rollback is called.
  42     VALUE_ROLL_BACK
  43   };
  44
  45   SetRegValueWorkItem(HKEY predefined_root,
  46                       const std::wstring& key_path,
  47                       const std::wstring& value_name,
  48                       const std::wstring& value_data,
  49                       bool overwrite);
  50
  51   SetRegValueWorkItem(HKEY predefined_root, const std::wstring& key_path,
  52                       const std::wstring& value_name, DWORD value_data,
  53                       bool overwrite);
  54
  55   SetRegValueWorkItem(HKEY predefined_root, const std::wstring& key_path,
  56                       const std::wstring& value_name, int64 value_data,
  57                       bool overwrite);
  58
  59   // Root key of the target key under which the value is set. The root key can
  60   // only be one of the predefined keys on Windows.
  61   HKEY predefined_root_;
  62
  63   // Path of the target key under which the value is set.
  64   std::wstring key_path_;
  65
  66   // Name of the value to be set.
  67   std::wstring value_name_;
  68
  69   // Whether to overwrite the existing value under the target key.
  70   bool overwrite_;
  71
  72   // Type of data to store
  73   DWORD type_;
  74   std::vector<uint8> value_;
  75   DWORD previous_type_;
  76   std::vector<uint8> previous_value_;
  77
  78   SettingStatus status_;
  79 };
  80
  81 #endif  // CHROME_INSTALLER_UTIL_SET_REG_VALUE_WORK_ITEM_H__