Skip to content

Commit

Permalink
net: make isIPv4 and isIPv6 more efficient
Browse files Browse the repository at this point in the history
`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: #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>
  • Loading branch information
vkurchatkin authored and rvagg committed Mar 16, 2016
1 parent 62926d8 commit b16f67a
Show file tree
Hide file tree
Showing 3 changed files with 54 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lib/net.js
Original file line number Diff line number Diff line change
Expand Up @@ -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);
};


Expand Down
23 changes: 23 additions & 0 deletions src/cares_wrap.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down Expand Up @@ -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);
Expand Down
29 changes: 29 additions & 0 deletions test/parallel/test-net-isip.js
Original file line number Diff line number Diff line change
Expand Up @@ -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);

0 comments on commit b16f67a

Please sign in to comment.