From baa8c15abfcc51ba0191e14ee2a665e069ed5998 Mon Sep 17 00:00:00 2001
From: Ouyang Yadong <oyydoibh@gmail.com>
Date: Tue, 6 Apr 2021 00:31:36 +0800
Subject: [PATCH] dns: allow NODE_DNS_VERBATIM to change default verbatim

Allow the NODE_DNS_VERBATIM environment variable to change the default
value of `verbatim` in `dns.lookup()`.
---
 doc/api/cli.md                             | 10 +++++
 lib/dns.js                                 |  3 +-
 lib/internal/dns/promises.js               |  4 +-
 lib/internal/dns/utils.js                  | 12 ++++++
 lib/internal/main/print_help.js            |  2 +
 test/parallel/test-dns-default-verbatim.js | 50 ++++++++++++++++++++++
 6 files changed, 78 insertions(+), 3 deletions(-)
 create mode 100644 test/parallel/test-dns-default-verbatim.js

diff --git a/doc/api/cli.md b/doc/api/cli.md
index 055eac22f6f0a2..8c31b800365989 100644
--- a/doc/api/cli.md
+++ b/doc/api/cli.md
@@ -1295,6 +1295,15 @@ added: v0.3.0
 
 When set, colors will not be used in the REPL.
 
+### `NODE_DNS_VERBATIM=[0, 1]`
+<!-- YAML
+added: REPLACEME
+-->
+
+Set the default value of `verbatim` in [`dns.lookup()`][]. The value may be:
+* `0` to indicate `false`
+* `1` to indicate `true`
+
 ### `NODE_EXTRA_CA_CERTS=file`
 <!-- YAML
 added: v7.3.0
@@ -1728,6 +1737,7 @@ $ node --max-old-space-size=1536 index.js
 [`NODE_OPTIONS`]: #cli_node_options_options
 [`NO_COLOR`]: https://no-color.org
 [`SlowBuffer`]: buffer.md#buffer_class_slowbuffer
+[`dns.lookup()`]: dns.html#dns_dns_lookup_hostname_options_callback
 [`process.setUncaughtExceptionCaptureCallback()`]: process.md#process_process_setuncaughtexceptioncapturecallback_fn
 [`tls.DEFAULT_MAX_VERSION`]: tls.md#tls_tls_default_max_version
 [`tls.DEFAULT_MIN_VERSION`]: tls.md#tls_tls_default_min_version
diff --git a/lib/dns.js b/lib/dns.js
index b75ca1ca7da082..b22a09e5c41afc 100644
--- a/lib/dns.js
+++ b/lib/dns.js
@@ -41,6 +41,7 @@ const {
   Resolver,
   validateHints,
   emitInvalidHostnameWarning,
+  getDefaultVerbatim,
 } = require('internal/dns/utils');
 const {
   ERR_INVALID_ARG_TYPE,
@@ -96,7 +97,7 @@ function lookup(hostname, options, callback) {
   let hints = 0;
   let family = -1;
   let all = false;
-  let verbatim = false;
+  let verbatim = getDefaultVerbatim();
 
   // Parse arguments
   if (hostname) {
diff --git a/lib/internal/dns/promises.js b/lib/internal/dns/promises.js
index e0158eef81307e..0f59c800552d23 100644
--- a/lib/internal/dns/promises.js
+++ b/lib/internal/dns/promises.js
@@ -1,5 +1,4 @@
 'use strict';
-
 const {
   ArrayPrototypeMap,
   ObjectCreate,
@@ -14,6 +13,7 @@ const {
   validateHints,
   validateTimeout,
   emitInvalidHostnameWarning,
+  getDefaultVerbatim,
 } = require('internal/dns/utils');
 const { codes, dnsException } = require('internal/errors');
 const { toASCII } = require('internal/idna');
@@ -103,7 +103,7 @@ function lookup(hostname, options) {
   var hints = 0;
   var family = -1;
   var all = false;
-  var verbatim = false;
+  var verbatim = getDefaultVerbatim();
 
   // Parse arguments
   if (hostname && typeof hostname !== 'string') {
diff --git a/lib/internal/dns/utils.js b/lib/internal/dns/utils.js
index f228ef38e270ad..70e95a2481bb04 100644
--- a/lib/internal/dns/utils.js
+++ b/lib/internal/dns/utils.js
@@ -184,6 +184,17 @@ function emitInvalidHostnameWarning(hostname) {
   );
 }
 
+function getDefaultVerbatim() {
+  const { NODE_DNS_VERBATIM } = process.env;
+  switch (NODE_DNS_VERBATIM) {
+    case '1':
+      return true;
+    case '0':
+    default:
+      return false;
+  }
+}
+
 module.exports = {
   bindDefaultResolver,
   getDefaultResolver,
@@ -192,4 +203,5 @@ module.exports = {
   validateTimeout,
   Resolver,
   emitInvalidHostnameWarning,
+  getDefaultVerbatim,
 };
diff --git a/lib/internal/main/print_help.js b/lib/internal/main/print_help.js
index 029701307980b1..42ccb6cadae2ae 100644
--- a/lib/internal/main/print_help.js
+++ b/lib/internal/main/print_help.js
@@ -38,6 +38,8 @@ const envVars = new SafeMap(ArrayPrototypeConcat([
     'categories that should print debug output' }],
   ['NODE_DISABLE_COLORS', { helpText: 'set to 1 to disable colors in ' +
     'the REPL' }],
+  ['NODE_DNS_VERBATIM', { helpText: 'set to 1 or 0 to enable or disable ' +
+    'verbatim in dns.lookup by default' }],
   ['NODE_EXTRA_CA_CERTS', { helpText: 'path to additional CA certificates ' +
     'file. Only read once during process startup.' }],
   ['NODE_NO_WARNINGS', { helpText: 'set to 1 to silence process warnings' }],
diff --git a/test/parallel/test-dns-default-verbatim.js b/test/parallel/test-dns-default-verbatim.js
new file mode 100644
index 00000000000000..f385240904d74d
--- /dev/null
+++ b/test/parallel/test-dns-default-verbatim.js
@@ -0,0 +1,50 @@
+// Flags: --expose-internals
+'use strict';
+const common = require('../common');
+const assert = require('assert');
+const { internalBinding } = require('internal/test/binding');
+const cares = internalBinding('cares_wrap');
+const { promisify } = require('util')
+
+// Test that NODE_DNS_VERBATIM works as expected.
+
+const originalGetaddrinfo = cares.getaddrinfo;
+const calls = [];
+cares.getaddrinfo = (...args) => {
+  calls.push(args);
+  originalGetaddrinfo(...args);
+};
+
+const dns = require('dns');
+const dnsPromises = dns.promises;
+
+(async () => {
+  {
+    process.env.NODE_DNS_VERBATIM = '1';
+    let verbatim;
+
+    await promisify(dns.lookup)('example.org');
+
+    assert.strictEqual(calls.length, 1);
+    verbatim = calls[0][4];
+    assert.strictEqual(verbatim, true);
+
+    await dnsPromises.lookup('example.org');
+    assert.strictEqual(calls.length, 2);
+    verbatim = calls[1][4];
+    assert.strictEqual(verbatim, true);
+
+    process.env.NODE_DNS_VERBATIM = '0';
+
+    await promisify(dns.lookup)('example.org');
+
+    assert.strictEqual(calls.length, 3);
+    verbatim = calls[2][4];
+    assert.strictEqual(verbatim, false);
+
+    await dnsPromises.lookup('example.org');
+    assert.strictEqual(calls.length, 4);
+    verbatim = calls[3][4];
+    assert.strictEqual(verbatim, false);
+  }
+})();