| OLD | NEW |
|---|
| (Empty) | |
| 1 // Copyright 2017 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 |
| 5 #include "content/common/content_security_policy/csp_context.h" |
| 6 #include "content/common/content_security_policy_header.h" |
| 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 |
| 9 namespace content { |
| 10 |
| 11 namespace { |
| 12 |
| 13 class CSPContextTest : public CSPContext { |
| 14 public: |
| 15 const std::string& LastConsoleMessage() { return console_message_; } |
| 16 |
| 17 void AddSchemeToBypassCSP(const std::string& scheme) { |
| 18 scheme_to_bypass_.push_back(scheme); |
| 19 } |
| 20 |
| 21 bool SchemeShouldBypassCSP(const base::StringPiece& scheme) override { |
| 22 return std::find(scheme_to_bypass_.begin(), scheme_to_bypass_.end(), |
| 23 scheme) != scheme_to_bypass_.end(); |
| 24 } |
| 25 |
| 26 private: |
| 27 void LogToConsole(const std::string& message) override { |
| 28 console_message_ = message; |
| 29 } |
| 30 std::string console_message_; |
| 31 std::vector<std::string> scheme_to_bypass_; |
| 32 }; |
| 33 |
| 34 // Build a new policy made of only one directive and no report endpoints. |
| 35 CSPPolicy BuildPolicy(CSPDirective::Name directive_name, |
| 36 std::vector<CSPSource> sources) { |
| 37 return CSPPolicy( |
| 38 blink::WebContentSecurityPolicyTypeEnforce, |
| 39 blink::WebContentSecurityPolicySourceHTTP, |
| 40 {CSPDirective(directive_name, CSPSourceList(false, false, sources))}, |
| 41 std::vector<std::string>()); // report_end_points |
| 42 } |
| 43 |
| 44 } // namespace; |
| 45 |
| 46 TEST(CSPContextTest, SchemeShouldBypassCSP) { |
| 47 CSPContextTest context; |
| 48 CSPSource source("", "example.com", false, url::PORT_UNSPECIFIED, false, ""); |
| 49 CSPPolicy policy = BuildPolicy(CSPDirective::DefaultSrc, {source}); |
| 50 EXPECT_FALSE(context.Allow({policy}, CSPDirective::FrameSrc, |
| 51 GURL("data:text/html,<html></html>"))); |
| 52 context.AddSchemeToBypassCSP("data"); |
| 53 EXPECT_TRUE(context.Allow({policy}, CSPDirective::FrameSrc, |
| 54 GURL("data:text/html,<html></html>"))); |
| 55 } |
| 56 |
| 57 TEST(CSPContextTest, MultiplePolicies) { |
| 58 CSPContextTest context; |
| 59 context.SetSelf(url::Origin(GURL("http://example.com"))); |
| 60 |
| 61 CSPSource source_a("", "a.com", false, url::PORT_UNSPECIFIED, false, ""); |
| 62 CSPSource source_b("", "b.com", false, url::PORT_UNSPECIFIED, false, ""); |
| 63 CSPSource source_c("", "c.com", false, url::PORT_UNSPECIFIED, false, ""); |
| 64 |
| 65 CSPPolicy policy1 = BuildPolicy(CSPDirective::FrameSrc, {source_a, source_b}); |
| 66 CSPPolicy policy2 = BuildPolicy(CSPDirective::FrameSrc, {source_a, source_c}); |
| 67 |
| 68 std::vector<CSPPolicy> policies = {policy1, policy2}; |
| 69 |
| 70 EXPECT_TRUE( |
| 71 context.Allow(policies, CSPDirective::FrameSrc, GURL("http://a.com"))); |
| 72 EXPECT_FALSE( |
| 73 context.Allow(policies, CSPDirective::FrameSrc, GURL("http://b.com"))); |
| 74 EXPECT_FALSE( |
| 75 context.Allow(policies, CSPDirective::FrameSrc, GURL("http://c.com"))); |
| 76 EXPECT_FALSE( |
| 77 context.Allow(policies, CSPDirective::FrameSrc, GURL("http://d.com"))); |
| 78 } |
| 79 |
| 80 } // namespace content |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2612793002:
Implement ContentSecurityPolicy on the browser-side. (Closed)
