Implement ContentSecurityPolicy on the browser-side.

content/common/content_security_policy/csp_source_list_unittest.cc

+// Copyright 2017 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 "content/common/content_security_policy/csp_context.h"
+#include "content/common/content_security_policy/csp_source_list.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+namespace {
+class CSPContextTest : public CSPContext {
+ public:
+  const std::string& LastConsoleMessage() { return console_message_; }
+
+ private:
+  void LogToConsole(const std::string& message) override {
+    console_message_ = message;
+  }
+  std::string console_message_;
+};
+}
+
+TEST(CSPSourceListTest, MultipleSource) {
+  CSPContextTest context;
+  context.SetSelf(url::Origin(GURL("http://example.com")));
+  CSPSourceList source_list(
+      false,  // allow_self
+      false,  // allow_star:
+      {CSPSource("", "a.com", false, url::PORT_UNSPECIFIED, false, ""),
+       CSPSource("", "b.com", false, url::PORT_UNSPECIFIED, false, "")});
+  EXPECT_TRUE(source_list.Allow(&context, GURL("http://a.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("http://b.com")));
+  EXPECT_FALSE(source_list.Allow(&context, GURL("http://c.com")));
+}
+
+TEST(CSPSourceList, ReportInvalidSourceExpression) {
+  const char console_message[] =
+      "The source list for Content Security Policy directive 'default-src' "
+      "contains an invalid source: ''invalid''. It will be ignored.";
+  CSPContextTest context;
+  CSPSourceList::Parse(&context, "default-src", "'invalid'");
+  EXPECT_EQ(console_message, context.LastConsoleMessage());
+}
+
+TEST(CSPSourceList, ReportInvalidSourceExpressionNone) {
+  const char console_message[] =
+      "The source list for Content Security Policy directive 'default-src' "
+      "contains an invalid source: ''none''. It will be ignored. Note that "
+      "'none' has no effect unless it is the only expression in the source "
+      "list.";
+  {
+    CSPContextTest context;
+    CSPSourceList::Parse(&context, "default-src", "'none' 'none'");
+    EXPECT_EQ(console_message, context.LastConsoleMessage());
+  }
+  {
+    CSPContextTest context;
+    CSPSourceList::Parse(&context, "default-src", "'none' 'self'");
+    EXPECT_EQ(console_message, context.LastConsoleMessage());
+  }
+}
+
+TEST(CSPSourceList, AllowStar) {
+  CSPContextTest context;
+  context.SetSelf(url::Origin(GURL("http://example.com")));
+  CSPSourceList source_list(false,                      // allow_self
+                            true,                       // allow_star:
+                            std::vector<CSPSource>());  // source_list
+  EXPECT_TRUE(source_list.Allow(&context, GURL("http://not-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("https://no-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("http-so://no-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("https-so://no-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("ws://no-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("wss://no-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("ftp://no-example.com")));
+
+  EXPECT_FALSE(source_list.Allow(&context, GURL("file://no-example.com")));
+  context.SetSelf(url::Origin(GURL("file://example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("file://no-example.com")));
+}
+
+TEST(CSPSourceList, AllowSelf) {
+  CSPContextTest context;
+  context.SetSelf(url::Origin(GURL("http://example.com")));
+  CSPSourceList source_list(true,                       // allow_self
+                            false,                      // allow_star:
+                            std::vector<CSPSource>());  // source_list
+  EXPECT_TRUE(source_list.Allow(&context, GURL("http://example.com")));
+  EXPECT_FALSE(source_list.Allow(&context, GURL("http://not-example.com")));
+  EXPECT_TRUE(source_list.Allow(&context, GURL("https://example.com")));
+  EXPECT_FALSE(source_list.Allow(&context, GURL("ws://example.com")));
+}
+
+}  // namespace content