blob: 00a519379f2fa8ee318228ef4b46438ed3ead885 [file] [log] [blame]
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/policy/registry_dict_win.h"
#include "base/json/json_reader.h"
#include "base/stl_util.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/sys_byteorder.h"
#include "base/values.h"
#include "base/win/registry.h"
#include "components/json_schema/json_schema_constants.h"
namespace schema = json_schema_constants;
using base::win::RegistryKeyIterator;
using base::win::RegistryValueIterator;
namespace policy {
namespace {
// Returns the entry with key |name| in |dictionary| (can be NULL), or NULL.
const base::DictionaryValue* GetEntry(const base::DictionaryValue* dictionary,
const std::string& name) {
if (!dictionary)
return NULL;
const base::DictionaryValue* entry = NULL;
dictionary->GetDictionaryWithoutPathExpansion(name, &entry);
return entry;
}
// Returns the Value type described in |schema|, or |default_type| if not found.
base::Value::Type GetValueTypeForSchema(const base::DictionaryValue* schema,
base::Value::Type default_type) {
// JSON-schema types to base::Value::Type mapping.
static const struct {
// JSON schema type.
const char* schema_type;
// Correspondent value type.
base::Value::Type value_type;
} kSchemaToValueTypeMap[] = {
{ schema::kArray, base::Value::TYPE_LIST },
{ schema::kBoolean, base::Value::TYPE_BOOLEAN },
{ schema::kInteger, base::Value::TYPE_INTEGER },
{ schema::kNull, base::Value::TYPE_NULL },
{ schema::kNumber, base::Value::TYPE_DOUBLE },
{ schema::kObject, base::Value::TYPE_DICTIONARY },
{ schema::kString, base::Value::TYPE_STRING },
};
if (!schema)
return default_type;
std::string type;
if (!schema->GetStringWithoutPathExpansion(schema::kType, &type))
return default_type;
for (size_t i = 0; i < arraysize(kSchemaToValueTypeMap); ++i) {
if (type == kSchemaToValueTypeMap[i].schema_type)
return kSchemaToValueTypeMap[i].value_type;
}
return default_type;
}
// Returns the schema for property |name| given the |schema| of an object.
// Returns the "additionalProperties" schema if no specific schema for
// |name| is present. Returns NULL if no schema is found.
const base::DictionaryValue* GetSchemaFor(const base::DictionaryValue* schema,
const std::string& name) {
const base::DictionaryValue* properties =
GetEntry(schema, schema::kProperties);
const base::DictionaryValue* sub_schema = GetEntry(properties, name);
if (sub_schema)
return sub_schema;
// "additionalProperties" can be a boolean, but that case is ignored.
return GetEntry(schema, schema::kAdditionalProperties);
}
// Converts a value (as read from the registry) to meet |schema|, converting
// types as necessary. Unconvertible types will show up as NULL values in the
// result.
scoped_ptr<base::Value> ConvertValue(const base::Value& value,
const base::DictionaryValue* schema) {
// Figure out the type to convert to from the schema.
const base::Value::Type result_type(
GetValueTypeForSchema(schema, value.GetType()));
// If the type is good already, go with it.
if (value.IsType(result_type)) {
// Recurse for complex types if there is a schema.
if (schema) {
const base::DictionaryValue* dict = NULL;
const base::ListValue* list = NULL;
if (value.GetAsDictionary(&dict)) {
scoped_ptr<base::DictionaryValue> result(new base::DictionaryValue());
for (base::DictionaryValue::Iterator entry(*dict); !entry.IsAtEnd();
entry.Advance()) {
scoped_ptr<base::Value> converted_value(
ConvertValue(entry.value(), GetSchemaFor(schema, entry.key())));
result->SetWithoutPathExpansion(entry.key(),
converted_value.release());
}
return result.Pass();
} else if (value.GetAsList(&list)) {
scoped_ptr<base::ListValue> result(new base::ListValue());
const base::DictionaryValue* item_schema =
GetEntry(schema, schema::kItems);
for (base::ListValue::const_iterator entry(list->begin());
entry != list->end(); ++entry) {
result->Append(ConvertValue(**entry, item_schema).release());
}
return result.Pass();
}
}
return make_scoped_ptr(value.DeepCopy());
}
// Else, do some conversions to map windows registry data types to JSON types.
std::string string_value;
int int_value = 0;
switch (result_type) {
case base::Value::TYPE_NULL: {
return make_scoped_ptr(base::Value::CreateNullValue());
}
case base::Value::TYPE_BOOLEAN: {
// Accept booleans encoded as either string or integer.
if (value.GetAsInteger(&int_value) ||
(value.GetAsString(&string_value) &&
base::StringToInt(string_value, &int_value))) {
return make_scoped_ptr(Value::CreateBooleanValue(int_value != 0));
}
break;
}
case base::Value::TYPE_INTEGER: {
// Integers may be string-encoded.
if (value.GetAsString(&string_value) &&
base::StringToInt(string_value, &int_value)) {
return make_scoped_ptr(base::Value::CreateIntegerValue(int_value));
}
break;
}
case base::Value::TYPE_DOUBLE: {
// Doubles may be string-encoded or integer-encoded.
double double_value = 0;
if (value.GetAsInteger(&int_value)) {
return make_scoped_ptr(base::Value::CreateDoubleValue(int_value));
} else if (value.GetAsString(&string_value) &&
base::StringToDouble(string_value, &double_value)) {
return make_scoped_ptr(base::Value::CreateDoubleValue(double_value));
}
break;
}
case base::Value::TYPE_LIST: {
// Lists are encoded as subkeys with numbered value in the registry.
const base::DictionaryValue* dict = NULL;
if (value.GetAsDictionary(&dict)) {
scoped_ptr<base::ListValue> result(new base::ListValue());
const base::DictionaryValue* item_schema =
GetEntry(schema, schema::kItems);
for (int i = 1; ; ++i) {
const base::Value* entry = NULL;
if (!dict->Get(base::IntToString(i), &entry))
break;
result->Append(ConvertValue(*entry, item_schema).release());
}
return result.Pass();
}
// Fall through in order to accept lists encoded as JSON strings.
}
case base::Value::TYPE_DICTIONARY: {
// Dictionaries may be encoded as JSON strings.
if (value.GetAsString(&string_value)) {
scoped_ptr<base::Value> result(base::JSONReader::Read(string_value));
if (result && result->IsType(result_type))
return result.Pass();
}
break;
}
case base::Value::TYPE_STRING:
case base::Value::TYPE_BINARY:
// No conversion possible.
break;
}
LOG(WARNING) << "Failed to convert " << value.GetType()
<< " to " << result_type;
return make_scoped_ptr(base::Value::CreateNullValue());
}
} // namespace
bool CaseInsensitiveStringCompare::operator()(const std::string& a,
const std::string& b) const {
return base::strcasecmp(a.c_str(), b.c_str()) < 0;
}
RegistryDict::RegistryDict() {}
RegistryDict::~RegistryDict() {
ClearKeys();
ClearValues();
}
RegistryDict* RegistryDict::GetKey(const std::string& name) {
KeyMap::iterator entry = keys_.find(name);
return entry != keys_.end() ? entry->second : NULL;
}
const RegistryDict* RegistryDict::GetKey(const std::string& name) const {
KeyMap::const_iterator entry = keys_.find(name);
return entry != keys_.end() ? entry->second : NULL;
}
void RegistryDict::SetKey(const std::string& name,
scoped_ptr<RegistryDict> dict) {
if (!dict) {
RemoveKey(name);
return;
}
RegistryDict*& entry = keys_[name];
delete entry;
entry = dict.release();
}
scoped_ptr<RegistryDict> RegistryDict::RemoveKey(const std::string& name) {
scoped_ptr<RegistryDict> result;
KeyMap::iterator entry = keys_.find(name);
if (entry != keys_.end()) {
result.reset(entry->second);
keys_.erase(entry);
}
return result.Pass();
}
void RegistryDict::ClearKeys() {
STLDeleteValues(&keys_);
}
base::Value* RegistryDict::GetValue(const std::string& name) {
ValueMap::iterator entry = values_.find(name);
return entry != values_.end() ? entry->second : NULL;
}
const base::Value* RegistryDict::GetValue(const std::string& name) const {
ValueMap::const_iterator entry = values_.find(name);
return entry != values_.end() ? entry->second : NULL;
}
void RegistryDict::SetValue(const std::string& name,
scoped_ptr<base::Value> dict) {
if (!dict) {
RemoveValue(name);
return;
}
Value*& entry = values_[name];
delete entry;
entry = dict.release();
}
scoped_ptr<base::Value> RegistryDict::RemoveValue(const std::string& name) {
scoped_ptr<base::Value> result;
ValueMap::iterator entry = values_.find(name);
if (entry != values_.end()) {
result.reset(entry->second);
values_.erase(entry);
}
return result.Pass();
}
void RegistryDict::ClearValues() {
STLDeleteValues(&values_);
}
void RegistryDict::Merge(const RegistryDict& other) {
for (KeyMap::const_iterator entry(other.keys_.begin());
entry != other.keys_.end(); ++entry) {
RegistryDict*& subdict = keys_[entry->first];
if (!subdict)
subdict = new RegistryDict();
subdict->Merge(*entry->second);
}
for (ValueMap::const_iterator entry(other.values_.begin());
entry != other.values_.end(); ++entry) {
SetValue(entry->first, make_scoped_ptr(entry->second->DeepCopy()));
}
}
void RegistryDict::Swap(RegistryDict* other) {
keys_.swap(other->keys_);
values_.swap(other->values_);
}
void RegistryDict::ReadRegistry(HKEY hive, const string16& root) {
ClearKeys();
ClearValues();
// First, read all the values of the key.
for (RegistryValueIterator it(hive, root.c_str()); it.Valid(); ++it) {
const std::string name = UTF16ToUTF8(it.Name());
switch (it.Type()) {
case REG_SZ:
case REG_EXPAND_SZ:
SetValue(
name,
make_scoped_ptr(new base::StringValue(UTF16ToUTF8(it.Value()))));
continue;
case REG_DWORD_LITTLE_ENDIAN:
case REG_DWORD_BIG_ENDIAN:
if (it.ValueSize() == sizeof(DWORD)) {
DWORD dword_value = *(reinterpret_cast<const DWORD*>(it.Value()));
if (it.Type() == REG_DWORD_BIG_ENDIAN)
dword_value = base::NetToHost32(dword_value);
else
dword_value = base::ByteSwapToLE32(dword_value);
SetValue(
name,
make_scoped_ptr(base::Value::CreateIntegerValue(dword_value)));
continue;
}
case REG_NONE:
case REG_LINK:
case REG_MULTI_SZ:
case REG_RESOURCE_LIST:
case REG_FULL_RESOURCE_DESCRIPTOR:
case REG_RESOURCE_REQUIREMENTS_LIST:
case REG_QWORD_LITTLE_ENDIAN:
// Unsupported type, message gets logged below.
break;
}
LOG(WARNING) << "Failed to read hive " << hive << " at "
<< root << "\\" << name
<< " type " << it.Type();
}
// Recurse for all subkeys.
for (RegistryKeyIterator it(hive, root.c_str()); it.Valid(); ++it) {
std::string name(UTF16ToUTF8(it.Name()));
scoped_ptr<RegistryDict> subdict(new RegistryDict());
subdict->ReadRegistry(hive, root + L"\\" + it.Name());
SetKey(name, subdict.Pass());
}
}
scoped_ptr<base::Value> RegistryDict::ConvertToJSON(
const base::DictionaryValue* schema) const {
base::Value::Type type =
GetValueTypeForSchema(schema, base::Value::TYPE_DICTIONARY);
switch (type) {
case base::Value::TYPE_DICTIONARY: {
scoped_ptr<base::DictionaryValue> result(new base::DictionaryValue());
for (RegistryDict::ValueMap::const_iterator entry(values_.begin());
entry != values_.end(); ++entry) {
result->SetWithoutPathExpansion(
entry->first,
ConvertValue(*entry->second,
GetSchemaFor(schema, entry->first)).release());
}
for (RegistryDict::KeyMap::const_iterator entry(keys_.begin());
entry != keys_.end(); ++entry) {
result->SetWithoutPathExpansion(
entry->first,
entry->second->ConvertToJSON(
GetSchemaFor(schema, entry->first)).release());
}
return result.Pass();
}
case base::Value::TYPE_LIST: {
scoped_ptr<base::ListValue> result(new base::ListValue());
const base::DictionaryValue* item_schema =
GetEntry(schema, schema::kItems);
for (int i = 1; ; ++i) {
const std::string name(base::IntToString(i));
const RegistryDict* key = GetKey(name);
if (key) {
result->Append(key->ConvertToJSON(item_schema).release());
continue;
}
const base::Value* value = GetValue(name);
if (value) {
result->Append(ConvertValue(*value, item_schema).release());
continue;
}
break;
}
return result.Pass();
}
default:
LOG(WARNING) << "Can't convert registry key to schema type " << type;
}
return make_scoped_ptr(base::Value::CreateNullValue());
}
} // namespace policy