1// Copyright 2013 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 COMPONENTS_POLICY_CORE_COMMON_REGISTRY_DICT_WIN_H_
6#define COMPONENTS_POLICY_CORE_COMMON_REGISTRY_DICT_WIN_H_
7
8#include <windows.h>
9
10#include <map>
11#include <string>
12
13#include "base/basictypes.h"
14#include "base/memory/scoped_ptr.h"
15#include "base/strings/string16.h"
16#include "components/policy/policy_export.h"
17
18namespace base {
19class Value;
20}
21
22namespace policy {
23
24class Schema;
25
26// A case-insensitive string comparison functor.
27struct POLICY_EXPORT CaseInsensitiveStringCompare {
28  bool operator()(const std::string& a, const std::string& b) const;
29};
30
31// In-memory representation of a registry subtree. Using a
32// base::DictionaryValue directly seems tempting, but that doesn't handle the
33// registry's case-insensitive-but-case-preserving semantics properly.
34class POLICY_EXPORT RegistryDict {
35 public:
36  typedef std::map<std::string, RegistryDict*,
37      CaseInsensitiveStringCompare> KeyMap;
38  typedef std::map<std::string, base::Value*,
39      CaseInsensitiveStringCompare> ValueMap;
40
41  RegistryDict();
42  ~RegistryDict();
43
44  // Returns a pointer to an existing key, NULL if not present.
45  RegistryDict* GetKey(const std::string& name);
46  const RegistryDict* GetKey(const std::string& name) const;
47  // Sets a key. If |dict| is NULL, clears that key.
48  void SetKey(const std::string& name, scoped_ptr<RegistryDict> dict);
49  // Removes a key. If the key doesn't exist, NULL is returned.
50  scoped_ptr<RegistryDict> RemoveKey(const std::string& name);
51  // Clears all keys.
52  void ClearKeys();
53
54  // Returns a pointer to a value, NULL if not present.
55  base::Value* GetValue(const std::string& name);
56  const base::Value* GetValue(const std::string& name) const;
57  // Sets a value. If |value| is NULL, removes the value.
58  void SetValue(const std::string& name, scoped_ptr<base::Value> value);
59  // Removes a value. If the value doesn't exist, NULL is returned.
60  scoped_ptr<base::Value> RemoveValue(const std::string& name);
61  // Clears all values.
62  void ClearValues();
63
64  // Merge keys and values from |other|, giving precedence to |other|.
65  void Merge(const RegistryDict& other);
66
67  // Swap with |other|.
68  void Swap(RegistryDict* other);
69
70  // Read a Windows registry subtree into this registry dictionary object.
71  void ReadRegistry(HKEY hive, const base::string16& root);
72
73  // Converts the dictionary to base::Value representation. For key/value name
74  // collisions, the key wins. |schema| is used to determine the expected type
75  // for each policy.
76  // The returned object is either a base::DictionaryValue or a base::ListValue.
77  scoped_ptr<base::Value> ConvertToJSON(const Schema& schema) const;
78
79  const KeyMap& keys() const { return keys_; }
80  const ValueMap& values() const { return values_; }
81
82 private:
83  KeyMap keys_;
84  ValueMap values_;
85
86  DISALLOW_COPY_AND_ASSIGN(RegistryDict);
87};
88
89}  // namespace policy
90
91#endif  // COMPONENTS_POLICY_CORE_COMMON_REGISTRY_DICT_WIN_H_
92