Fix net.Socket.connect argument parsing

Fixes #1251.

2 files changed, 20 insertions, 9 deletions

diff --git a/lib/net_legacy.js b/lib/net_legacy.js
index 4254dc9a44..52f4e9405b 100644
--- a/lib/net_legacy.js
+++ b/lib/net_legacy.js
@@ -700,9 +700,14 @@ Socket.prototype.connect = function() {
   self._connecting = true; // set false in doConnect
   self.writable = true;
 
-  var lastArg = arguments[arguments.length - 1];
-  if (typeof lastArg == 'function') {
-    self.addListener('connect', lastArg);
+  var host;
+  if (typeof arguments[1] === 'function') {
+    self.on('connect', arguments[1]);
+  } else {
+    host = arguments[1];
+    if (typeof arguments[2] === 'function') {
+      self.on('connect', arguments[2]);
+    }
   }
 
   var port = toPort(arguments[0]);
@@ -715,7 +720,7 @@ Socket.prototype.connect = function() {
     doConnect(self, arguments[0]);
   } else {
     // TCP
-    require('dns').lookup(arguments[1], function(err, ip, addressType) {
+    require('dns').lookup(host, function(err, ip, addressType) {
       if (err) {
         self.emit('error', err);
       } else {
diff --git a/lib/net_uv.js b/lib/net_uv.js
index 63f0a15080..e9db1b2cba 100644
--- a/lib/net_uv.js
+++ b/lib/net_uv.js
@@ -24,9 +24,9 @@ exports.createServer = function() {
 };
 
 
-exports.connect = exports.createConnection = function(port, host /* [cb] */ ) {
+exports.connect = exports.createConnection = function(port /* [host], [cb] */) {
   var s = new Socket();
-  s.connect(port, host, arguments[2]);
+  s.connect(port, arguments[1], arguments[2]);
   return s;
 };
 
@@ -340,7 +340,7 @@ function connectip(self, port, ip) {
 }
 
 
-Socket.prototype.connect = function(port, host /* [cb] */) {
+Socket.prototype.connect = function(port /* [host], [cb] */) {
   var self = this;
 
   if (this.destroyed) {
@@ -348,8 +348,14 @@ Socket.prototype.connect = function(port, host /* [cb] */) {
     initSocketHandle(this);
   }
     
-  if (typeof arguments[2] === 'function') {
-    self.on('connect', arguments[2]);
+  var host;
+  if (typeof arguments[1] === 'function') {
+    self.on('connect', arguments[1]);
+  } else {
+    host = arguments[1];
+    if (typeof arguments[2] === 'function') {
+      self.on('connect', arguments[2]);
+    }
   }
 
   timers.active(this);