Support blacklist with port

2 files changed, 9 insertions, 2 deletions

diff --git a/src/shared/blacklists.js b/src/shared/blacklists.js
index 19ed3f1..61720c3 100644
--- a/src/shared/blacklists.js
+++ b/src/shared/blacklists.js
@@ -4,9 +4,9 @@ const includes = (blacklist, url) => {
   let u = new URL(url);
   return blacklist.some((item) => {
     if (!item.includes('/')) {
-      return re.fromWildcard(item).test(u.hostname);
+      return re.fromWildcard(item).test(u.host);
     }
-    return re.fromWildcard(item).test(u.hostname + u.pathname);
+    return re.fromWildcard(item).test(u.host + u.pathname);
   });
 };
 
diff --git a/test/shared/blacklists.test.js b/test/shared/blacklists.test.js
index 87e89c5..289ea0f 100644
--- a/test/shared/blacklists.test.js
+++ b/test/shared/blacklists.test.js
@@ -39,4 +39,11 @@ describe("shared/blacklist", () => {
     expect(includes(blacklist, 'https://github.com/abcdef')).to.be.true;
     expect(includes(blacklist, 'https://gist.github.com/abc')).to.be.false;
   })
+
+  it('matches address and port', () => {
+    let blacklist = ['127.0.0.1:8888'];
+
+    expect(includes(blacklist, 'http://127.0.0.1:8888/')).to.be.true;
+    expect(includes(blacklist, 'http://127.0.0.1:8888/hello')).to.be.true;
+  })
 });