Restricting redirects to chrome:

content/browser/child_process_security_policy_unittest.cc

 // Copyright (c) 2011 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 <string>
 
 #include "base/basictypes.h"
 #include "base/file_path.h"
 #include "base/platform_file.h"
 #include "content/browser/child_process_security_policy.h"
@@ skipping 38 matching lines @@
   EXPECT_TRUE(p->IsWebSafeScheme(chrome::kBlobScheme));
   EXPECT_TRUE(p->IsWebSafeScheme(chrome::kFileSystemScheme));
 
   EXPECT_FALSE(p->IsWebSafeScheme("registered-web-safe-scheme"));
   p->RegisterWebSafeScheme("registered-web-safe-scheme");
   EXPECT_TRUE(p->IsWebSafeScheme("registered-web-safe-scheme"));
 
   EXPECT_FALSE(p->IsWebSafeScheme(chrome::kChromeUIScheme));
 }
 
+TEST_F(ChildProcessSecurityPolicyTest, IsWebUISchemeTest) {
+  ChildProcessSecurityPolicy* p = ChildProcessSecurityPolicy::GetInstance();
+
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kHttpScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kHttpsScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kFtpScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kDataScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kExtensionScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kBlobScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kFileSystemScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kAboutScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kJavaScriptScheme));
+  EXPECT_FALSE(p->IsWebUIScheme(chrome::kViewSourceScheme));
+  EXPECT_TRUE(p->IsWebUIScheme(chrome::kChromeUIScheme));
+
+  EXPECT_FALSE(p->IsWebUIScheme("registered-webui-scheme"));
+  p->RegisterWebUIScheme("registered-webui-scheme");
+  EXPECT_TRUE(p->IsWebUIScheme("registered-webui-scheme"));
+}
+
 TEST_F(ChildProcessSecurityPolicyTest, IsPseudoSchemeTest) {
   ChildProcessSecurityPolicy* p = ChildProcessSecurityPolicy::GetInstance();
 
   EXPECT_TRUE(p->IsPseudoScheme(chrome::kAboutScheme));
   EXPECT_TRUE(p->IsPseudoScheme(chrome::kJavaScriptScheme));
   EXPECT_TRUE(p->IsPseudoScheme(chrome::kViewSourceScheme));
 
   EXPECT_FALSE(p->IsPseudoScheme("registered-pseudo-scheme"));
   p->RegisterPseudoScheme("registered-pseudo-scheme");
   EXPECT_TRUE(p->IsPseudoScheme("registered-pseudo-scheme"));
@@ skipping 165 matching lines @@
 
   p->GrantRequestURL(kRendererID, GURL("view-source:file:///etc/passwd"));
   // View source needs to be able to request the embedded scheme.
   EXPECT_TRUE(p->CanRequestURL(kRendererID,
                                GURL("view-source:file:///etc/passwd")));
   EXPECT_TRUE(p->CanRequestURL(kRendererID, GURL("file:///etc/passwd")));
 
   p->Remove(kRendererID);
 }
 
+TEST_F(ChildProcessSecurityPolicyTest, CanRedirectURL) {
+  ChildProcessSecurityPolicy* p = ChildProcessSecurityPolicy::GetInstance();
+
+  p->Add(kRendererID);
+
+  // Behaves the same as CanRequestURL, except for chrome: Urls
+  EXPECT_TRUE(p->CanRedirectURL(kRendererID, GURL("http://www.google.com")));
+  EXPECT_TRUE(p->CanRedirectURL(kRendererID, GURL("evil-scheme://path")));
+  EXPECT_FALSE(p->CanRedirectURL(kRendererID, GURL("chrome://settings")));
+  EXPECT_FALSE(p->CanRedirectURL(kRendererID, GURL("chrome://flags")));
+  EXPECT_FALSE(p->CanRedirectURL(kRendererID, GURL("about://flags")));
+
+  p->Remove(kRendererID);
+}
+
 TEST_F(ChildProcessSecurityPolicyTest, CanReadFiles) {
   ChildProcessSecurityPolicy* p = ChildProcessSecurityPolicy::GetInstance();
 
   p->Add(kRendererID);
 
   EXPECT_FALSE(p->CanReadFile(kRendererID,
       FilePath(FILE_PATH_LITERAL("/etc/passwd"))));
   p->GrantReadFile(kRendererID, FilePath(FILE_PATH_LITERAL("/etc/passwd")));
   EXPECT_TRUE(p->CanReadFile(kRendererID,
       FilePath(FILE_PATH_LITERAL("/etc/passwd"))));
@@ skipping 178 matching lines @@
 
   // Renderers are added and removed on the UI thread, but the policy can be
   // queried on the IO thread.  The ChildProcessSecurityPolicy needs to be
   // prepared to answer policy questions about renderers who no longer exist.
 
   // In this case, we default to secure behavior.
   EXPECT_FALSE(p->CanRequestURL(kRendererID, url));
   EXPECT_FALSE(p->CanReadFile(kRendererID, file));
   EXPECT_FALSE(p->HasWebUIBindings(kRendererID));
 }