From b16f67a0b9fa81a971fdeacb9abc6ce531e1b9ae Mon Sep 17 00:00:00 2001
From: Vladimir Kurchatkin <vladimir.kurchatkin@gmail.com>
Date: Sun, 28 Feb 2016 19:44:39 +0300
Subject: [PATCH] net: make `isIPv4` and `isIPv6` more efficient

`isIPv4` and `isIPv6` are implemented on top of `isIP`, which in turn
checks the sting for being both IPv4 and IPv6, which can be inefficient
in some scenarios. This commit makes them use `uv_inet_pton` directly
instead.

PR-URL: https://github.com/nodejs/node/pull/5478
Reviewed-By: Evan Lucas <evanlucas@me.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Colin Ihrig <cjihrig@gmail.com>
Reviewed-By: Roman Reiss <me@silverwind.io>
---
 lib/net.js                     |  4 ++--
 src/cares_wrap.cc              | 23 +++++++++++++++++++++++
 test/parallel/test-net-isip.js | 29 +++++++++++++++++++++++++++++
 3 files changed, 54 insertions(+), 2 deletions(-)

diff --git a/lib/net.js b/lib/net.js
index 0a5941f811d776..d0648e5d0d5b35 100644
--- a/lib/net.js
+++ b/lib/net.js
@@ -1568,12 +1568,12 @@ exports.isIP = cares.isIP;
 
 
 exports.isIPv4 = function(input) {
-  return exports.isIP(input) === 4;
+  return cares.isIPv4(input);
 };
 
 
 exports.isIPv6 = function(input) {
-  return exports.isIP(input) === 6;
+  return cares.isIPv6(input);
 };
 
 
diff --git a/src/cares_wrap.cc b/src/cares_wrap.cc
index ab76eb6c9a3d60..ade257963fe415 100644
--- a/src/cares_wrap.cc
+++ b/src/cares_wrap.cc
@@ -1044,6 +1044,27 @@ static void IsIP(const FunctionCallbackInfo<Value>& args) {
   args.GetReturnValue().Set(rc);
 }
 
+static void IsIPv4(const FunctionCallbackInfo<Value>& args) {
+  node::Utf8Value ip(args.GetIsolate(), args[0]);
+  char address_buffer[sizeof(struct in_addr)];
+
+  if (uv_inet_pton(AF_INET, *ip, &address_buffer) == 0) {
+    args.GetReturnValue().Set(true);
+  } else {
+    args.GetReturnValue().Set(false);
+  }
+}
+
+static void IsIPv6(const FunctionCallbackInfo<Value>& args) {
+  node::Utf8Value ip(args.GetIsolate(), args[0]);
+  char address_buffer[sizeof(struct in6_addr)];
+
+  if (uv_inet_pton(AF_INET6, *ip, &address_buffer) == 0) {
+    args.GetReturnValue().Set(true);
+  } else {
+    args.GetReturnValue().Set(false);
+  }
+}
 
 static void GetAddrInfo(const FunctionCallbackInfo<Value>& args) {
   Environment* env = Environment::GetCurrent(args);
@@ -1283,6 +1304,8 @@ static void Initialize(Local<Object> target,
   env->SetMethod(target, "getaddrinfo", GetAddrInfo);
   env->SetMethod(target, "getnameinfo", GetNameInfo);
   env->SetMethod(target, "isIP", IsIP);
+  env->SetMethod(target, "isIPv4", IsIPv4);
+  env->SetMethod(target, "isIPv6", IsIPv6);
 
   env->SetMethod(target, "strerror", StrError);
   env->SetMethod(target, "getServers", GetServers);
diff --git a/test/parallel/test-net-isip.js b/test/parallel/test-net-isip.js
index 76432ed3488240..6b159b59d8c8c0 100644
--- a/test/parallel/test-net-isip.js
+++ b/test/parallel/test-net-isip.js
@@ -31,11 +31,40 @@ assert.equal(net.isIP('0000:0000:0000:0000:0000:0000:12345:0000'), 0);
 assert.equal(net.isIP('0'), 0);
 assert.equal(net.isIP(), 0);
 assert.equal(net.isIP(''), 0);
+assert.equal(net.isIP(null), 0);
+assert.equal(net.isIP(123), 0);
+assert.equal(net.isIP(true), 0);
+assert.equal(net.isIP({}), 0);
+assert.equal(net.isIP({ toString: () => '::2001:252:1:255.255.255.255' }), 6);
+assert.equal(net.isIP({ toString: () => '127.0.0.1' }), 4);
+assert.equal(net.isIP({ toString: () => 'bla' }), 0);
 
 assert.equal(net.isIPv4('127.0.0.1'), true);
 assert.equal(net.isIPv4('example.com'), false);
 assert.equal(net.isIPv4('2001:252:0:1::2008:6'), false);
+assert.equal(net.isIPv4(), false);
+assert.equal(net.isIPv4(''), false);
+assert.equal(net.isIPv4(null), false);
+assert.equal(net.isIPv4(123), false);
+assert.equal(net.isIPv4(true), false);
+assert.equal(net.isIPv4({}), false);
+assert.equal(net.isIPv4({
+  toString: () => '::2001:252:1:255.255.255.255'
+}), false);
+assert.equal(net.isIPv4({ toString: () => '127.0.0.1' }), true);
+assert.equal(net.isIPv4({ toString: () => 'bla' }), false);
 
 assert.equal(net.isIPv6('127.0.0.1'), false);
 assert.equal(net.isIPv6('example.com'), false);
 assert.equal(net.isIPv6('2001:252:0:1::2008:6'), true);
+assert.equal(net.isIPv6(), false);
+assert.equal(net.isIPv6(''), false);
+assert.equal(net.isIPv6(null), false);
+assert.equal(net.isIPv6(123), false);
+assert.equal(net.isIPv6(true), false);
+assert.equal(net.isIPv6({}), false);
+assert.equal(net.isIPv6({
+  toString: () => '::2001:252:1:255.255.255.255'
+}), true);
+assert.equal(net.isIPv6({ toString: () => '127.0.0.1' }), false);
+assert.equal(net.isIPv6({ toString: () => 'bla' }), false);