Get ride of chrome.storage.* data when removing an application.

chrome/browser/extensions/data_deleter.cc

 // Copyright (c) 2012 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/extensions/data_deleter.h"
 
 #include "chrome/browser/extensions/api/storage/settings_frontend.h"
 #include "chrome/browser/extensions/extension_service.h"
+#include "chrome/browser/extensions/extension_special_storage_policy.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/common/extensions/manifest_handlers/app_isolation_info.h"
+#include "chrome/common/extensions/manifest_handlers/app_launch_info.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/site_instance.h"
 #include "content/public/browser/storage_partition.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension.h"
 #include "net/url_request/url_request_context_getter.h"
 
 using content::BrowserContext;
 using content::BrowserThread;
 using content::StoragePartition;
 
 namespace extensions {
 
 // static
-void DataDeleter::StartDeleting(Profile* profile,
-                                const std::string& extension_id,
-                                const GURL& storage_origin) {
+void DataDeleter::StartDeleting(Profile* profile, const Extension* extension) {
+  DCHECK(profile);
+  DCHECK(extension);
+
+  std::string extension_id = extension->id();

[not at google - send to devlin, 2014/01/08 18:01:36]

should be a const std::string&, however, IMO this variable doesn't improve the
code

[mlamouri (slow - plz ping), 2014/01/08 23:44:08]

Agreed. It's not that needed. That code changed a bit so maybe extension_id was
used more when I added it.

+
+  if (extensions::AppIsolationInfo::HasIsolatedStorage(extension)) {
+    BrowserContext::AsyncObliterateStoragePartition(
+        profile,
+        profile->GetExtensionService()->GetSiteForExtensionId(extension_id),
+        base::Bind(&ExtensionService::OnNeedsToGarbageCollectIsolatedStorage,
+                   profile->GetExtensionService()->AsWeakPtr()));
+  } else {
+    GURL launch_web_url_origin(
+      extensions::AppLaunchInfo::GetLaunchWebURL(extension).GetOrigin());
+
+    StoragePartition* partition = BrowserContext::GetStoragePartitionForSite(
+        profile,
+        Extension::GetBaseURLFromExtensionId(extension_id));
+
+    if (extension->is_hosted_app() &&
+        !profile->GetExtensionSpecialStoragePolicy()->
+            IsStorageProtected(launch_web_url_origin)) {
+      DeleteOrigin(profile, partition, launch_web_url_origin);
+    }
+    DeleteOrigin(profile, partition, extension->url());
+  }
+
+  // Begin removal of the settings for the current extension.
+  profile->GetExtensionService()->settings_frontend()->
+      DeleteStorageSoon(extension_id);
+}
+
+// static
+void DataDeleter::DeleteOrigin(Profile* profile,
+                               StoragePartition* partition,
+                               const GURL& origin) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   DCHECK(profile);
+  DCHECK(partition);
 
-  const GURL& site = Extension::GetBaseURLFromExtensionId(extension_id);
-
-  StoragePartition* partition =
-      BrowserContext::GetStoragePartitionForSite(profile, site);
-
-  if (storage_origin.SchemeIs(extensions::kExtensionScheme)) {
+  if (origin.SchemeIs(extensions::kExtensionScheme)) {
     // TODO(ajwong): Cookies are not properly isolated for
     // chrome-extension:// scheme.  (http://crbug.com/158386).
     //
     // However, no isolated apps actually can write to kExtensionScheme
     // origins. Thus, it is benign to delete from the
     // RequestContextForExtensions because there's nothing stored there. We
     // preserve this code path without checking for isolation because it's
     // simpler than special casing.  This code should go away once we merge
     // the various URLRequestContexts (http://crbug.com/159193).
     partition->ClearDataForOrigin(
         StoragePartition::REMOVE_DATA_MASK_ALL &
             (~StoragePartition::REMOVE_DATA_MASK_SHADER_CACHE),
         StoragePartition::QUOTA_MANAGED_STORAGE_MASK_ALL,
-        storage_origin,
+        origin,
         profile->GetRequestContextForExtensions());
   } else {
     // We don't need to worry about the media request context because that
     // shares the same cookie store as the main request context.
     partition->ClearDataForOrigin(
         StoragePartition::REMOVE_DATA_MASK_ALL &
             (~StoragePartition::REMOVE_DATA_MASK_SHADER_CACHE),
         StoragePartition::QUOTA_MANAGED_STORAGE_MASK_ALL,
-        storage_origin,
+        origin,
         partition->GetURLRequestContext());
   }
-
-  // Begin removal of the settings for the current extension.
-  profile->GetExtensionService()->settings_frontend()->
-      DeleteStorageSoon(extension_id);
 }
 
 }  // namespace extensions