Make Document::isSecureContext() work for OOPIFs

third_party/WebKit/Source/core/dom/DocumentTest.cpp

 /*
  * Copyright (c) 2014, Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 19 matching lines @@
 
 #include "core/dom/Document.h"
 
 #include "core/dom/DocumentVisibilityObserver.h"
 #include "core/frame/FrameView.h"
 #include "core/html/HTMLHeadElement.h"
 #include "core/html/HTMLLinkElement.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/heap/Handle.h"
 #include "platform/weborigin/ReferrerPolicy.h"
+#include "platform/weborigin/SchemeRegistry.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 class DocumentTest : public ::testing::Test {
 protected:
     void SetUp() override;
 
@@ skipping 296 matching lines @@
     element->setAttribute(blink::HTMLNames::classAttr, "a");
     EXPECT_NE(previousStyleVersion, document().styleVersion());
 
     document().view()->updateAllLifecyclePhases();
 
     previousStyleVersion = document().styleVersion();
     element->setAttribute(blink::HTMLNames::classAttr, "a b");
     EXPECT_NE(previousStyleVersion, document().styleVersion());
 }
 
+TEST_F(DocumentTest, EnforceSandboxFlags)
+{
+    RefPtr<SecurityOrigin> origin = SecurityOrigin::createFromString("http://example.test");
+    document().setSecurityOrigin(origin);
+    SandboxFlags mask = SandboxNavigation;
+    document().enforceSandboxFlags(mask);
+    EXPECT_EQ(origin, document().getSecurityOrigin());
+    EXPECT_FALSE(document().getSecurityOrigin()->isPotentiallyTrustworthy());
+
+    mask |= SandboxOrigin;
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().getSecurityOrigin()->isUnique());
+    EXPECT_FALSE(document().getSecurityOrigin()->isPotentiallyTrustworthy());
+
+    // A unique origin does not bypass secure context checks unless it
+    // is also potentially trustworthy.
+    SchemeRegistry::registerURLSchemeBypassingSecureContextCheck("very-special-scheme");
+    origin = SecurityOrigin::createFromString("very-special-scheme://example.test");
+    document().setSecurityOrigin(origin);
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().getSecurityOrigin()->isUnique());
+    EXPECT_FALSE(document().getSecurityOrigin()->isPotentiallyTrustworthy());
+
+    SchemeRegistry::registerURLSchemeAsSecure("very-special-scheme");
+    document().setSecurityOrigin(origin);
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().getSecurityOrigin()->isUnique());
+    EXPECT_TRUE(document().getSecurityOrigin()->isPotentiallyTrustworthy());
+
+    origin = SecurityOrigin::createFromString("https://example.test");
+    document().setSecurityOrigin(origin);
+    document().enforceSandboxFlags(mask);
+    EXPECT_TRUE(document().getSecurityOrigin()->isUnique());
+    EXPECT_TRUE(document().getSecurityOrigin()->isPotentiallyTrustworthy());
+}
+
 } // namespace blink