Merge pull request #1265 from filipe-norte-red/wpe-2.38-unrestricted-ports-for-uri-schemes2

Support unrestricted ports usage for custom URI schemes

Author: magomez

Source/WTF/wtf/URL.cpp

@@ -907,6 +907,32 @@ bool URL::protocolIsAbout() const
     return protocolIs("about"_s);
 }
 
+static bool protocolIsWhitelistedForAllPortsAcccess(StringView protocol)
+{
+    static Vector<String> s_protocolsWhitelisted;
+    static std::once_flag s_onceFlag;
+    std::call_once(s_onceFlag,
+        [] {
+            // The env var contains a comma separated list of protocols that need to have
+            // access to all ports.
+            // Example: WPE_WHITELIST_ALL_PORTS_FOR_PROTOCOLS="dvb,echo,custom"
+            String s(String::fromLatin1(std::getenv("WPE_WHITELIST_ALL_PORTS_FOR_PROTOCOLS")));
+            if (s.isEmpty())
+                return;
+
+            s_protocolsWhitelisted.appendVector(s.convertToASCIILowercase().split(','));
+
+            const Vector<String> excludeFromWhitelist( { "http"_s, "https"_s, "ws"_s, "wss"_s, "ftp"_s, "ftps"_s} );
+
+            // Ensure reserved protocols are not whitelisted
+            s_protocolsWhitelisted.removeAllMatching([&](const auto& protocol) {
+                return excludeFromWhitelist.contains(protocol);
+            });
+        });
+
+    return s_protocolsWhitelisted.contains(protocol.convertToASCIILowercase());
+}
+
 bool portAllowed(const URL& url)
 {
     std::optional<uint16_t> port = url.port();
@@ -915,6 +941,9 @@ bool portAllowed(const URL& url)
     if (!port)
         return true;
 
+    if (protocolIsWhitelistedForAllPortsAcccess(url.protocol()))
+        return true;
+
     // This blocked port list matches the port blocking that Mozilla implements.
     // See http://www.mozilla.org/projects/netlib/PortBanning.html for more information.
     static const uint16_t blockedPortList[] = {