fix: error handling for domainToASCII (#1608)

MichaelSun90 · web-flow · commit af211974892f · 2024-08-07T14:09:45.000-07:00
diff --git a/src/connector.ts b/src/connector.ts
@@ -164,7 +164,8 @@ export async function lookupAllAddresses(host: string, lookup: LookupFunction, s
 
       signal.addEventListener('abort', onAbort);
 
-      lookup(url.domainToASCII(host), { all: true }, (err, addresses) => {
+      const domainInASCII = url.domainToASCII(host);
+      lookup(domainInASCII === '' ? host : domainInASCII, { all: true }, (err, addresses) => {
         signal.removeEventListener('abort', onAbort);
 
         err ? reject(err) : resolve(addresses);
diff --git a/src/sender.ts b/src/sender.ts
@@ -80,9 +80,8 @@ export async function sendMessage(host: string, port: number, lookup: LookupFunc
         reject(new AbortError());
       };
 
-      signal.addEventListener('abort', onAbort);
-
-      lookup(url.domainToASCII(host), { all: true }, (err, addresses) => {
+      const domainInASCII = url.domainToASCII(host);
+      lookup(domainInASCII === '' ? host : domainInASCII, { all: true }, (err, addresses) => {
         signal.removeEventListener('abort', onAbort);
 
         err ? reject(err) : resolve(addresses);
diff --git a/test/unit/connector-test.js b/test/unit/connector-test.js
@@ -1,6 +1,6 @@
 const Mitm = require('mitm');
 const sinon = require('sinon');
-const url = require('node:url');
+const dns = require('dns');
 const assert = require('chai').assert;
 
 const {
@@ -25,7 +25,7 @@ describe('lookupAllAddresses', function() {
     }
 
     assert.isOk(lookup.called, 'Failed to call `lookup` function for hostname');
-    assert.isOk(lookup.calledWithMatch(url.domainToASCII(server)), 'Unexpected hostname passed to `lookup`');
+    assert.isOk(lookup.calledWithMatch('xn--tiq21tzznxb.ad'), 'Unexpected hostname passed to `lookup`');
   });
 
   it('test ASCII Server name', async function() {
@@ -45,6 +45,21 @@ describe('lookupAllAddresses', function() {
     assert.isOk(lookup.called, 'Failed to call `lookup` function for hostname');
     assert.isOk(lookup.calledWithMatch(server), 'Unexpected hostname passed to `lookup`');
   });
+
+  it('test invalid ASCII Server name', async function() {
+    const server = 'http:wrong';
+    const controller = new AbortController();
+
+    let actualError;
+    try {
+      await lookupAllAddresses(server, dns.lookup, controller.signal);
+    } catch (err) {
+      actualError = err;
+    }
+
+    assert.instanceOf(actualError, Error);
+    assert.strictEqual(actualError.message, 'getaddrinfo ENOTFOUND http:wrong');
+  });
 });
 
 describe('connectInSequence', function() {
