Fix checkHost

fixed webpack#1604
3846masa-tmp · Dec 22, 2018 · 1c56116 · 1c56116
1 parent ff2874f
commit 1c56116
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/Server.js b/lib/Server.js
@@ -646,7 +646,12 @@ Server.prototype.checkHost = function (headers, headerToCheck) {
   }
 
   // use the node url-parser to retrieve the hostname from the host-header.
-  const hostname = url.parse(`//${hostHeader}`, false, true).hostname;
+  const hostname = url.parse(
+    // if hostHeader doesn't have scheme, add // for parsing.
+    /^(.+:)?\/\//.test(hostHeader) ? hostHeader : `//${hostHeader}`,
+    false,
+    true,
+  ).hostname;
   // always allow requests with explicit IPv4 or IPv6-address.
   // A note on IPv6 addresses:
   // hostHeader will always contain the brackets denoting