tracked_split_preference.cc 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. // Copyright 2014 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. #include "services/preferences/tracked/tracked_split_preference.h"
  5. #include <vector>
  6. #include "base/check.h"
  7. #include "base/notreached.h"
  8. #include "base/values.h"
  9. #include "services/preferences/public/mojom/tracked_preference_validation_delegate.mojom.h"
  10. #include "services/preferences/tracked/pref_hash_store_transaction.h"
  11. using ValueState =
  12. prefs::mojom::TrackedPreferenceValidationDelegate::ValueState;
  13. TrackedSplitPreference::TrackedSplitPreference(
  14. const std::string& pref_path,
  15. size_t reporting_id,
  16. size_t reporting_ids_count,
  17. prefs::mojom::TrackedPreferenceMetadata::EnforcementLevel enforcement_level,
  18. prefs::mojom::TrackedPreferenceMetadata::ValueType value_type,
  19. prefs::mojom::TrackedPreferenceValidationDelegate* delegate)
  20. : pref_path_(pref_path),
  21. helper_(pref_path,
  22. reporting_id,
  23. reporting_ids_count,
  24. enforcement_level,
  25. value_type),
  26. delegate_(delegate) {}
  27. TrackedPreferenceType TrackedSplitPreference::GetType() const {
  28. return TrackedPreferenceType::SPLIT;
  29. }
  30. void TrackedSplitPreference::OnNewValue(
  31. const base::Value* value,
  32. PrefHashStoreTransaction* transaction) const {
  33. const base::DictionaryValue* dict_value = NULL;
  34. if (value && !value->GetAsDictionary(&dict_value)) {
  35. NOTREACHED();
  36. return;
  37. }
  38. transaction->StoreSplitHash(pref_path_, dict_value);
  39. }
  40. bool TrackedSplitPreference::EnforceAndReport(
  41. base::DictionaryValue* pref_store_contents,
  42. PrefHashStoreTransaction* transaction,
  43. PrefHashStoreTransaction* external_validation_transaction) const {
  44. base::DictionaryValue* dict_value = nullptr;
  45. if (!pref_store_contents->GetDictionary(pref_path_, &dict_value) &&
  46. pref_store_contents->FindPath(pref_path_)) {
  47. // There should be a dictionary or nothing at |pref_path_|.
  48. NOTREACHED();
  49. return false;
  50. }
  51. std::vector<std::string> invalid_keys;
  52. ValueState value_state =
  53. transaction->CheckSplitValue(pref_path_, dict_value, &invalid_keys);
  54. helper_.ReportValidationResult(value_state, transaction->GetStoreUMASuffix());
  55. ValueState external_validation_value_state = ValueState::UNSUPPORTED;
  56. std::vector<std::string> external_validation_invalid_keys;
  57. if (external_validation_transaction) {
  58. external_validation_value_state =
  59. external_validation_transaction->CheckSplitValue(
  60. pref_path_, dict_value, &external_validation_invalid_keys);
  61. helper_.ReportValidationResult(
  62. external_validation_value_state,
  63. external_validation_transaction->GetStoreUMASuffix());
  64. }
  65. if (delegate_) {
  66. delegate_->OnSplitPreferenceValidation(
  67. pref_path_, invalid_keys, external_validation_invalid_keys, value_state,
  68. external_validation_value_state, helper_.IsPersonal());
  69. }
  70. TrackedPreferenceHelper::ResetAction reset_action =
  71. helper_.GetAction(value_state);
  72. helper_.ReportAction(reset_action);
  73. bool was_reset = false;
  74. if (reset_action == TrackedPreferenceHelper::DO_RESET) {
  75. if (value_state == ValueState::CHANGED) {
  76. DCHECK(!invalid_keys.empty());
  77. for (std::vector<std::string>::const_iterator it = invalid_keys.begin();
  78. it != invalid_keys.end(); ++it) {
  79. dict_value->RemoveKey(*it);
  80. }
  81. } else {
  82. pref_store_contents->RemovePath(pref_path_);
  83. }
  84. was_reset = true;
  85. }
  86. if (value_state != ValueState::UNCHANGED) {
  87. // Store the hash for the new value (whether it was reset or not).
  88. const base::DictionaryValue* new_dict_value = NULL;
  89. pref_store_contents->GetDictionary(pref_path_, &new_dict_value);
  90. transaction->StoreSplitHash(pref_path_, new_dict_value);
  91. }
  92. // Update MACs in the external store if there is one and there either was a
  93. // reset or external validation failed.
  94. if (external_validation_transaction &&
  95. (was_reset || external_validation_value_state != ValueState::UNCHANGED)) {
  96. const base::DictionaryValue* new_dict_value = nullptr;
  97. pref_store_contents->GetDictionary(pref_path_, &new_dict_value);
  98. external_validation_transaction->StoreSplitHash(pref_path_, new_dict_value);
  99. }
  100. return was_reset;
  101. }