Make the flash permission prompt work properly on file: URLs

chrome/browser/plugins/flash_permission_context.cc

Index: chrome/browser/plugins/flash_permission_context.cc
diff --git a/chrome/browser/plugins/flash_permission_context.cc b/chrome/browser/plugins/flash_permission_context.cc
index 633f159cd8ab239287ea3efb198991f72d1ef877..bb782a621a22d9bcc0f26811c8e46b7f3ff71df6 100644
--- a/chrome/browser/plugins/flash_permission_context.cc
+++ b/chrome/browser/plugins/flash_permission_context.cc
@@ -9,6 +9,7 @@
 #include "chrome/browser/plugins/plugin_utils.h"
 #include "chrome/browser/plugins/plugins_field_trial.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
+#include "components/content_settings/core/common/content_settings_pattern.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
@@ -46,6 +47,28 @@ void FlashPermissionContext::UpdateTabContext(const PermissionRequestID& id,
   web_contents->GetController().Reload(true /* check_for_repost */);
 }
 
+void FlashPermissionContext::UpdateContentSetting(
+    const GURL& requesting_origin,
+    const GURL& embedding_origin,
+    ContentSetting content_setting) {
+  DCHECK_EQ(requesting_origin, requesting_origin.GetOrigin());
+  DCHECK_EQ(embedding_origin, embedding_origin.GetOrigin());
+  DCHECK(content_setting == CONTENT_SETTING_ALLOW ||
+         content_setting == CONTENT_SETTING_BLOCK);
+
+  // If the request was for a file scheme, allow or deny all file:/// URLs.
+  ContentSettingsPattern pattern;
+  if (embedding_origin.SchemeIsFile())
+    pattern = ContentSettingsPattern::FromString("file:///*");
+  else
+    pattern = ContentSettingsPattern::FromURLNoWildcard(embedding_origin);
+
+  HostContentSettingsMapFactory::GetForProfile(profile())
+      ->SetContentSettingCustomScope(
+          pattern, ContentSettingsPattern::Wildcard(), content_settings_type(),
+          std::string(), content_setting);
+}
+
 bool FlashPermissionContext::IsRestrictedToSecureOrigins() const {
   return false;
 }