Improve use of NeverDestroyed
[WebKit-https.git] / Source / WebCore / platform / SchemeRegistry.cpp
index 6076158..9d48421 100644 (file)
@@ -39,6 +39,8 @@
 
 namespace WebCore {
 
+// FIXME: URLSchemesMap is a peculiar type name given that it is a set.
+
 static const URLSchemesMap& builtinLocalURLSchemes();
 static const Vector<String>& builtinSecureSchemes();
 static const Vector<String>& builtinSchemesWithUniqueOrigins();
@@ -46,61 +48,72 @@ static const Vector<String>& builtinEmptyDocumentSchemes();
 static const Vector<String>& builtinCanDisplayOnlyIfCanRequestSchemes();
 static const Vector<String>& builtinCORSEnabledSchemes();
 
+using StringVectorFunction = const Vector<String>& (*)();
+
+static void add(URLSchemesMap& set, StringVectorFunction function)
+{
+    for (auto& scheme : function())
+        set.add(scheme);
+}
+
+static NeverDestroyed<URLSchemesMap> makeNeverDestroyedSchemeSet(const Vector<String>& (*function)())
+{
+    URLSchemesMap set;
+    add(set, function);
+    return set;
+}
+
 static const URLSchemesMap& allBuiltinSchemes()
 {
-    static NeverDestroyed<URLSchemesMap> schemes;
-    if (schemes.get().isEmpty()) {
-        for (const auto& scheme : builtinLocalURLSchemes())
-            schemes.get().add(scheme);
-        for (const auto& scheme : builtinSecureSchemes())
-            schemes.get().add(scheme);
-        for (const auto& scheme : builtinSchemesWithUniqueOrigins())
-            schemes.get().add(scheme);
-        for (const auto& scheme : builtinEmptyDocumentSchemes())
-            schemes.get().add(scheme);
-        for (const auto& scheme : builtinCanDisplayOnlyIfCanRequestSchemes())
-            schemes.get().add(scheme);
-        for (const auto& scheme : builtinCORSEnabledSchemes())
-            schemes.get().add(scheme);
+    static const auto schemes = makeNeverDestroyed([] {
+        static const StringVectorFunction functions[] {
+            builtinSecureSchemes,
+            builtinSchemesWithUniqueOrigins,
+            builtinEmptyDocumentSchemes,
+            builtinCanDisplayOnlyIfCanRequestSchemes,
+            builtinCORSEnabledSchemes,
+        };
 
         // Other misc schemes that the SchemeRegistry doesn't know about.
-        schemes.get().add("webkit-fake-url");
+        static const char* const otherSchemes[] = {
+            "webkit-fake-url",
 #if PLATFORM(MAC)
-        schemes.get().add("safari-extension");
+            "safari-extension",
 #endif
 #if USE(QUICK_LOOK)
-        schemes.get().add(QLPreviewProtocol());
+            QLPreviewProtocol(),
 #endif
 #if ENABLE(CONTENT_FILTERING)
-        schemes.get().add(ContentFilter::urlScheme());
+            ContentFilter::urlScheme(),
 #endif
-    }
-
+        };
+
+        URLSchemesMap set;
+        for (auto& scheme : builtinLocalURLSchemes())
+            set.add(scheme);
+        for (auto& function : functions)
+            add(set, function);
+        for (auto& scheme : otherSchemes)
+            set.add(scheme);
+        return set;
+    }());
     return schemes;
 }
 
-const URLSchemesMap& builtinLocalURLSchemes()
+static const URLSchemesMap& builtinLocalURLSchemes()
 {
-    static NeverDestroyed<URLSchemesMap> schemes;
-    if (schemes.get().isEmpty()) {
-        schemes.get().add("file");
+    static const auto schemes = makeNeverDestroyed(URLSchemesMap {
+        "file",
 #if PLATFORM(COCOA)
-        schemes.get().add("applewebdata");
+        "applewebdata",
 #endif
-    }
-
+    });
     return schemes;
 }
 
 static URLSchemesMap& localURLSchemes()
 {
-    static NeverDestroyed<URLSchemesMap> localSchemes;
-
-    if (localSchemes.get().isEmpty()) {
-        for (const auto& scheme : builtinLocalURLSchemes())
-            localSchemes.get().add(scheme);
-    }
-
+    static NeverDestroyed<URLSchemesMap> localSchemes = builtinLocalURLSchemes();
     return localSchemes;
 }
 
@@ -112,109 +125,69 @@ static URLSchemesMap& displayIsolatedURLSchemes()
 
 const Vector<String>& builtinSecureSchemes()
 {
-    static NeverDestroyed<Vector<String>> schemes;
-    if (schemes.get().isEmpty()) {
-        schemes.get().append("https");
-        schemes.get().append("about");
-        schemes.get().append("data");
-        schemes.get().append("wss");
+    static const auto schemes = makeNeverDestroyed(Vector<String> {
+        "https",
+        "about",
+        "data",
+        "wss",
 #if PLATFORM(GTK) || PLATFORM(WPE)
-        schemes.get().append("resource");
+        "resource",
 #endif
-        schemes.get().shrinkToFit();
-    }
-
+    });
     return schemes;
 }
 
 static URLSchemesMap& secureSchemes()
 {
-    static NeverDestroyed<URLSchemesMap> secureSchemes;
-
-    if (secureSchemes.get().isEmpty()) {
-        for (const auto& scheme : builtinSecureSchemes())
-            secureSchemes.get().add(scheme);
-    }
-
+    static auto secureSchemes = makeNeverDestroyedSchemeSet(builtinSecureSchemes);
     return secureSchemes;
 }
 
 const Vector<String>& builtinSchemesWithUniqueOrigins()
 {
-    static NeverDestroyed<Vector<String>> schemes;
-    if (schemes.get().isEmpty()) {
-        schemes.get().append("about");
-        schemes.get().append("javascript");
-        // This is a willful violation of HTML5.
+    static const auto schemes = makeNeverDestroyed(Vector<String> {
+        "about",
+        "javascript",
+        // This is an intentional difference from the behavior the HTML specification calls for.
         // See https://bugs.webkit.org/show_bug.cgi?id=11885
-        schemes.get().append("data");
-        schemes.get().shrinkToFit();
-    }
-
+        "data",
+    });
     return schemes;
 }
 
 static URLSchemesMap& schemesWithUniqueOrigins()
 {
-    static NeverDestroyed<URLSchemesMap> schemesWithUniqueOrigins;
-
-    if (schemesWithUniqueOrigins.get().isEmpty()) {
-        for (const auto& scheme : builtinSchemesWithUniqueOrigins())
-            schemesWithUniqueOrigins.get().add(scheme);
-    }
-
+    static auto schemesWithUniqueOrigins = makeNeverDestroyedSchemeSet(builtinSchemesWithUniqueOrigins);
     return schemesWithUniqueOrigins;
 }
 
 const Vector<String>& builtinEmptyDocumentSchemes()
 {
-    static NeverDestroyed<Vector<String>> schemes;
-    if (schemes.get().isEmpty()) {
-        schemes.get().append("about");
-        schemes.get().shrinkToFit();
-    }
-
+    static const auto schemes = makeNeverDestroyed(Vector<String> { "about" });
     return schemes;
 }
 
 static URLSchemesMap& emptyDocumentSchemes()
 {
-    static NeverDestroyed<URLSchemesMap> emptyDocumentSchemes;
-
-    if (emptyDocumentSchemes.get().isEmpty()) {
-        for (const auto& scheme : builtinEmptyDocumentSchemes())
-            emptyDocumentSchemes.get().add(scheme);
-    }
-
+    static auto emptyDocumentSchemes = makeNeverDestroyedSchemeSet(builtinEmptyDocumentSchemes);
     return emptyDocumentSchemes;
 }
 
-static HashSet<String, ASCIICaseInsensitiveHash>& schemesForbiddenFromDomainRelaxation()
+static URLSchemesMap& schemesForbiddenFromDomainRelaxation()
 {
-    static NeverDestroyed<HashSet<String, ASCIICaseInsensitiveHash>> schemes;
+    static NeverDestroyed<URLSchemesMap> schemes;
     return schemes;
 }
 
 const Vector<String>& builtinCanDisplayOnlyIfCanRequestSchemes()
 {
-    static NeverDestroyed<Vector<String>> schemes;
-    if (schemes.get().isEmpty()) {
-        schemes.get().append("blob");
-        schemes.get().shrinkToFit();
-    }
-
+    static const auto schemes = makeNeverDestroyed(Vector<String> { "blob" });
     return schemes;
 }
 
 static URLSchemesMap& canDisplayOnlyIfCanRequestSchemes()
 {
-    static NeverDestroyed<URLSchemesMap> canDisplayOnlyIfCanRequestSchemes;
-
-    if (canDisplayOnlyIfCanRequestSchemes.get().isEmpty()) {
-        for (const auto& scheme : builtinCanDisplayOnlyIfCanRequestSchemes())
-            canDisplayOnlyIfCanRequestSchemes.get().add(scheme);
-    }
-
+    static auto canDisplayOnlyIfCanRequestSchemes = makeNeverDestroyedSchemeSet(builtinCanDisplayOnlyIfCanRequestSchemes);
     return canDisplayOnlyIfCanRequestSchemes;
 }
 
@@ -256,27 +229,15 @@ static URLSchemesMap& schemesAllowingDatabaseAccessInPrivateBrowsing()
 
 const Vector<String>& builtinCORSEnabledSchemes()
 {
-    static NeverDestroyed<Vector<String>> schemes;
-    if (schemes.get().isEmpty()) {
-        schemes.get().append("http");
-        schemes.get().append("https");
-        schemes.get().shrinkToFit();
-    }
-
+    static const auto schemes = makeNeverDestroyed(Vector<String> { "http", "https" });
     return schemes;
 }
 
 static URLSchemesMap& CORSEnabledSchemes()
 {
     // FIXME: http://bugs.webkit.org/show_bug.cgi?id=77160
-    static NeverDestroyed<URLSchemesMap> CORSEnabledSchemes;
-
-    if (CORSEnabledSchemes.get().isEmpty()) {
-        for (const auto& scheme : builtinCORSEnabledSchemes())
-            CORSEnabledSchemes.get().add(scheme);
-    }
-
-    return CORSEnabledSchemes;
+    static auto schemes = makeNeverDestroyedSchemeSet(builtinCORSEnabledSchemes);
+    return schemes;
 }
 
 static URLSchemesMap& ContentSecurityPolicyBypassingSchemes()
@@ -299,62 +260,60 @@ static URLSchemesMap& alwaysRevalidatedSchemes()
 
 bool SchemeRegistry::shouldTreatURLSchemeAsLocal(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return localURLSchemes().contains(scheme);
+    return !scheme.isNull() && localURLSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsNoAccess(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     schemesWithUniqueOrigins().add(scheme);
 }
 
 bool SchemeRegistry::shouldTreatURLSchemeAsNoAccess(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return schemesWithUniqueOrigins().contains(scheme);
+    return !scheme.isNull() && schemesWithUniqueOrigins().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsDisplayIsolated(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     displayIsolatedURLSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldTreatURLSchemeAsDisplayIsolated(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return displayIsolatedURLSchemes().contains(scheme);
+    return !scheme.isNull() && displayIsolatedURLSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsSecure(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     secureSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldTreatURLSchemeAsSecure(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return secureSchemes().contains(scheme);
+    return !scheme.isNull() && secureSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsEmptyDocument(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     emptyDocumentSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldLoadURLSchemeAsEmptyDocument(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return emptyDocumentSchemes().contains(scheme);
+    return !scheme.isNull() && emptyDocumentSchemes().contains(scheme);
 }
 
 void SchemeRegistry::setDomainRelaxationForbiddenForURLScheme(bool forbidden, const String& scheme)
 {
-    if (scheme.isEmpty())
+    if (scheme.isNull())
         return;
 
     if (forbidden)
@@ -365,125 +324,126 @@ void SchemeRegistry::setDomainRelaxationForbiddenForURLScheme(bool forbidden, co
 
 bool SchemeRegistry::isDomainRelaxationForbiddenForURLScheme(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return schemesForbiddenFromDomainRelaxation().contains(scheme);
+    return !scheme.isNull() && schemesForbiddenFromDomainRelaxation().contains(scheme);
 }
 
 bool SchemeRegistry::canDisplayOnlyIfCanRequest(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return canDisplayOnlyIfCanRequestSchemes().contains(scheme);
+    return !scheme.isNull() && canDisplayOnlyIfCanRequestSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerAsCanDisplayOnlyIfCanRequest(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     canDisplayOnlyIfCanRequestSchemes().add(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsNotAllowingJavascriptURLs(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     notAllowingJavascriptURLsSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldTreatURLSchemeAsNotAllowingJavascriptURLs(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return notAllowingJavascriptURLsSchemes().contains(scheme);
+    return !scheme.isNull() && notAllowingJavascriptURLsSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsAllowingLocalStorageAccessInPrivateBrowsing(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     schemesAllowingLocalStorageAccessInPrivateBrowsing().add(scheme);
 }
 
 bool SchemeRegistry::allowsLocalStorageAccessInPrivateBrowsing(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return schemesAllowingLocalStorageAccessInPrivateBrowsing().contains(scheme);
+    return !scheme.isNull() && schemesAllowingLocalStorageAccessInPrivateBrowsing().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsAllowingDatabaseAccessInPrivateBrowsing(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     schemesAllowingDatabaseAccessInPrivateBrowsing().add(scheme);
 }
 
 bool SchemeRegistry::allowsDatabaseAccessInPrivateBrowsing(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return schemesAllowingDatabaseAccessInPrivateBrowsing().contains(scheme);
+    return !scheme.isNull() && schemesAllowingDatabaseAccessInPrivateBrowsing().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsCORSEnabled(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     CORSEnabledSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldTreatURLSchemeAsCORSEnabled(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return CORSEnabledSchemes().contains(scheme);
+    return !scheme.isNull() && CORSEnabledSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsBypassingContentSecurityPolicy(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     ContentSecurityPolicyBypassingSchemes().add(scheme);
 }
 
 void SchemeRegistry::removeURLSchemeRegisteredAsBypassingContentSecurityPolicy(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     ContentSecurityPolicyBypassingSchemes().remove(scheme);
 }
 
 bool SchemeRegistry::schemeShouldBypassContentSecurityPolicy(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return ContentSecurityPolicyBypassingSchemes().contains(scheme);
+    return !scheme.isNull() && ContentSecurityPolicyBypassingSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsAlwaysRevalidated(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     alwaysRevalidatedSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldAlwaysRevalidateURLScheme(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return alwaysRevalidatedSchemes().contains(scheme);
+    return !scheme.isNull() && alwaysRevalidatedSchemes().contains(scheme);
 }
 
 void SchemeRegistry::registerURLSchemeAsCachePartitioned(const String& scheme)
 {
+    if (scheme.isNull())
+        return;
     cachePartitioningSchemes().add(scheme);
 }
 
 bool SchemeRegistry::shouldPartitionCacheForURLScheme(const String& scheme)
 {
-    if (scheme.isEmpty())
-        return false;
-    return cachePartitioningSchemes().contains(scheme);
+    return !scheme.isNull() && cachePartitioningSchemes().contains(scheme);
 }
 
 bool SchemeRegistry::isUserExtensionScheme(const String& scheme)
 {
-    UNUSED_PARAM(scheme);
 #if PLATFORM(MAC)
     if (scheme == "safari-extension")
         return true;
+#else
+    UNUSED_PARAM(scheme);
 #endif
     return false;
 }
 
 bool SchemeRegistry::isBuiltinScheme(const String& scheme)
 {
-    return allBuiltinSchemes().contains(scheme) || URLParser::isSpecialScheme(scheme);
+    return !scheme.isNull() && (allBuiltinSchemes().contains(scheme) || URLParser::isSpecialScheme(scheme));
 }
 
 } // namespace WebCore