Store USB device permissions in website settings.

chrome/browser/permissions/chooser_permission_context_base.cc

+// Copyright 2015 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/permissions/chooser_permission_context_base.h"
+
+#include <string>
+
+#include "base/values.h"
+#include "chrome/browser/content_settings/host_content_settings_map_factory.h"
+#include "chrome/browser/permissions/permission_manager.h"
+#include "chrome/browser/permissions/permission_manager_factory.h"
+#include "components/content_settings/core/browser/host_content_settings_map.h"
+#include "content/public/common/permission_status.mojom.h"
+
+const char kObjectListKey[] = "chosen-objects";
+
+ChooserPermissionContextBase::ChooserPermissionContextBase(
+    Profile* profile,
+    const content::PermissionType permission_type,
+    const ContentSettingsType chooser_settings_type)
+    : profile_(profile),
+      permission_type_(permission_type),
+      chooser_settings_type_(chooser_settings_type) {}
+
+ChooserPermissionContextBase::~ChooserPermissionContextBase() {}
+
+ChooserPermissionContextBase::ObjectList
+ChooserPermissionContextBase::GetPreviouslyChosenObjects(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin) {
+  PermissionManager* permission_manager =
+      PermissionManagerFactory::GetForProfile(profile_);
+  content::PermissionStatus permission_status =
+      permission_manager->GetPermissionStatus(
+          permission_type_, requesting_origin, embedding_origin);
+  if (permission_status != content::PERMISSION_STATUS_GRANTED)
+    return ObjectList();
+
+  scoped_ptr<base::DictionaryValue> setting =
+      GetWebsiteSetting(requesting_origin, embedding_origin);
+  scoped_ptr<base::Value> objects;
+  if (!setting->Remove(kObjectListKey, &objects))
+    return ObjectList();
+
+  scoped_ptr<base::ListValue> object_list =
+      base::ListValue::From(objects.Pass());
+  if (!object_list)
+    return ObjectList();
+
+  ObjectList results;
+  for (base::ListValue::iterator it = object_list->begin();
+       it != object_list->end(); ++it) {
+    if ((*it)->GetType() != base::Value::TYPE_DICTIONARY)
+      continue;
+
+    scoped_ptr<base::DictionaryValue> object(
+        static_cast<base::DictionaryValue*>(*it));
+    *it = nullptr;
+
+    if (IsValidObject(*object))
+      results.push_back(object.Pass());
+  }
+  return results.Pass();
+}
+
+void ChooserPermissionContextBase::GrantObjectPermission(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    scoped_ptr<base::DictionaryValue> object) {
+  DCHECK(object && IsValidObject(*object));

[Bernhard Bauer, 2015/10/13 08:27:59]

Split this up into two different DCHECKs.

+  scoped_ptr<base::DictionaryValue> setting =
+      GetWebsiteSetting(requesting_origin, embedding_origin);
+  base::ListValue* object_list;
+  if (!setting->GetList(kObjectListKey, &object_list)) {
+    object_list = new base::ListValue();
+    setting->Set(kObjectListKey, object_list);
+  }
+  object_list->AppendIfNotPresent(object.release());
+  SetWebsiteSetting(requesting_origin, embedding_origin, setting.Pass());
+}
+
+void ChooserPermissionContextBase::RevokeObjectPermission(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    const base::DictionaryValue& object) {
+  DCHECK(IsValidObject(object));
+  scoped_ptr<base::DictionaryValue> setting =
+      GetWebsiteSetting(requesting_origin, embedding_origin);
+  base::ListValue* object_list;
+  if (!setting->GetList(kObjectListKey, &object_list))
+    return;
+  object_list->Remove(object, nullptr);
+  SetWebsiteSetting(requesting_origin, embedding_origin, setting.Pass());
+}
+
+scoped_ptr<base::DictionaryValue>
+ChooserPermissionContextBase::GetWebsiteSetting(const GURL& requesting_origin,
+                                                const GURL& embedding_origin) {
+  HostContentSettingsMap* settings =
+      HostContentSettingsMapFactory::GetForProfile(profile_);
+  DCHECK(settings);
+
+  scoped_ptr<base::DictionaryValue> value =
+      base::DictionaryValue::From(settings->GetWebsiteSetting(
+          requesting_origin, embedding_origin, chooser_settings_type_,
+          std::string(), nullptr));
+  if (!value)
+    return make_scoped_ptr(new base::DictionaryValue());
+
+  return value.Pass();
+}
+
+void ChooserPermissionContextBase::SetWebsiteSetting(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    scoped_ptr<base::Value> value) {
+  HostContentSettingsMap* settings =
+      HostContentSettingsMapFactory::GetForProfile(profile_);
+  DCHECK(settings);
+
+  ContentSettingsPattern primary_pattern(
+      ContentSettingsPattern::FromURLNoWildcard(requesting_origin));
+  ContentSettingsPattern secondary_pattern(
+      ContentSettingsPattern::FromURLNoWildcard(embedding_origin));
+  if (!primary_pattern.IsValid() || !secondary_pattern.IsValid())
+    return;
+
+  settings->SetWebsiteSetting(primary_pattern, secondary_pattern,
+                              chooser_settings_type_, std::string(),
+                              value.release());
+}