mirror of
https://github.com/nodejs/node.git
synced 2024-12-01 16:10:02 +01:00
c2b4f4809b
Currently when a server receives a new connection the underlying socket handle begins reading data immediately. This causes problems when sockets are passed between processes, as data can be read by the first process and thus never read by the second process. This commit allows sockets that are constructed with a handle to be paused initially. PR-URL: https://github.com/joyent/node/pull/8576 Fixes: https://github.com/joyent/node/issues/7905 Fixes: https://github.com/joyent/node/issues/7784 Reviewed-by: Trevor Norris <trev.norris@gmail.com>
1460 lines
36 KiB
JavaScript
1460 lines
36 KiB
JavaScript
// Copyright Joyent, Inc. and other Node contributors.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the
|
|
// "Software"), to deal in the Software without restriction, including
|
|
// without limitation the rights to use, copy, modify, merge, publish,
|
|
// distribute, sublicense, and/or sell copies of the Software, and to permit
|
|
// persons to whom the Software is furnished to do so, subject to the
|
|
// following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included
|
|
// in all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
|
|
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
|
|
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
|
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
|
// USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
var events = require('events');
|
|
var stream = require('stream');
|
|
var timers = require('timers');
|
|
var util = require('util');
|
|
var assert = require('assert');
|
|
var cares = process.binding('cares_wrap');
|
|
var uv = process.binding('uv');
|
|
var Pipe = process.binding('pipe_wrap').Pipe;
|
|
|
|
|
|
var cluster;
|
|
var errnoException = util._errnoException;
|
|
|
|
function noop() {}
|
|
|
|
// constructor for lazy loading
|
|
function createPipe() {
|
|
return new Pipe();
|
|
}
|
|
|
|
// constructor for lazy loading
|
|
function createTCP() {
|
|
var TCP = process.binding('tcp_wrap').TCP;
|
|
return new TCP();
|
|
}
|
|
|
|
|
|
function createHandle(fd) {
|
|
var tty = process.binding('tty_wrap');
|
|
var type = tty.guessHandleType(fd);
|
|
if (type === 'PIPE') return createPipe();
|
|
if (type === 'TCP') return createTCP();
|
|
throw new TypeError('Unsupported fd type: ' + type);
|
|
}
|
|
|
|
|
|
var debug = util.debuglog('net');
|
|
|
|
function isPipeName(s) {
|
|
return util.isString(s) && toNumber(s) === false;
|
|
}
|
|
|
|
|
|
exports.createServer = function() {
|
|
return new Server(arguments[0], arguments[1]);
|
|
};
|
|
|
|
|
|
// Target API:
|
|
//
|
|
// var s = net.connect({port: 80, host: 'google.com'}, function() {
|
|
// ...
|
|
// });
|
|
//
|
|
// There are various forms:
|
|
//
|
|
// connect(options, [cb])
|
|
// connect(port, [host], [cb])
|
|
// connect(path, [cb]);
|
|
//
|
|
exports.connect = exports.createConnection = function() {
|
|
var args = normalizeConnectArgs(arguments);
|
|
debug('createConnection', args);
|
|
var s = new Socket(args[0]);
|
|
return Socket.prototype.connect.apply(s, args);
|
|
};
|
|
|
|
// Returns an array [options] or [options, cb]
|
|
// It is the same as the argument of Socket.prototype.connect().
|
|
function normalizeConnectArgs(args) {
|
|
var options = {};
|
|
|
|
if (util.isObject(args[0])) {
|
|
// connect(options, [cb])
|
|
options = args[0];
|
|
} else if (isPipeName(args[0])) {
|
|
// connect(path, [cb]);
|
|
options.path = args[0];
|
|
} else {
|
|
// connect(port, [host], [cb])
|
|
options.port = args[0];
|
|
if (util.isString(args[1])) {
|
|
options.host = args[1];
|
|
}
|
|
}
|
|
|
|
var cb = args[args.length - 1];
|
|
return util.isFunction(cb) ? [options, cb] : [options];
|
|
}
|
|
exports._normalizeConnectArgs = normalizeConnectArgs;
|
|
|
|
|
|
// called when creating new Socket, or when re-using a closed Socket
|
|
function initSocketHandle(self) {
|
|
self.destroyed = false;
|
|
self.bytesRead = 0;
|
|
self._bytesDispatched = 0;
|
|
|
|
// Handle creation may be deferred to bind() or connect() time.
|
|
if (self._handle) {
|
|
self._handle.owner = self;
|
|
self._handle.onread = onread;
|
|
|
|
// If handle doesn't support writev - neither do we
|
|
if (!self._handle.writev)
|
|
self._writev = null;
|
|
}
|
|
}
|
|
|
|
function Socket(options) {
|
|
if (!(this instanceof Socket)) return new Socket(options);
|
|
|
|
this._connecting = false;
|
|
this._hadError = false;
|
|
this._handle = null;
|
|
this._host = null;
|
|
|
|
if (util.isNumber(options))
|
|
options = { fd: options }; // Legacy interface.
|
|
else if (util.isUndefined(options))
|
|
options = {};
|
|
|
|
stream.Duplex.call(this, options);
|
|
|
|
if (options.handle) {
|
|
this._handle = options.handle; // private
|
|
} else if (!util.isUndefined(options.fd)) {
|
|
this._handle = createHandle(options.fd);
|
|
this._handle.open(options.fd);
|
|
if ((options.fd == 1 || options.fd == 2) &&
|
|
(this._handle instanceof Pipe) &&
|
|
process.platform === 'win32') {
|
|
// Make stdout and stderr blocking on Windows
|
|
var err = this._handle.setBlocking(true);
|
|
if (err)
|
|
throw errnoException(err, 'setBlocking');
|
|
}
|
|
this.readable = options.readable !== false;
|
|
this.writable = options.writable !== false;
|
|
} else {
|
|
// these will be set once there is a connection
|
|
this.readable = this.writable = false;
|
|
}
|
|
|
|
// shut down the socket when we're finished with it.
|
|
this.on('finish', onSocketFinish);
|
|
this.on('_socketEnd', onSocketEnd);
|
|
|
|
initSocketHandle(this);
|
|
|
|
this._pendingData = null;
|
|
this._pendingEncoding = '';
|
|
|
|
// handle strings directly
|
|
this._writableState.decodeStrings = false;
|
|
|
|
// default to *not* allowing half open sockets
|
|
this.allowHalfOpen = options && options.allowHalfOpen || false;
|
|
|
|
// if we have a handle, then start the flow of data into the
|
|
// buffer. if not, then this will happen when we connect
|
|
if (this._handle && options.readable !== false) {
|
|
if (options.pauseOnCreate) {
|
|
// stop the handle from reading and pause the stream
|
|
this._handle.reading = false;
|
|
this._handle.readStop();
|
|
this._readableState.flowing = false;
|
|
} else {
|
|
this.read(0);
|
|
}
|
|
}
|
|
}
|
|
util.inherits(Socket, stream.Duplex);
|
|
|
|
// the user has called .end(), and all the bytes have been
|
|
// sent out to the other side.
|
|
// If allowHalfOpen is false, or if the readable side has
|
|
// ended already, then destroy.
|
|
// If allowHalfOpen is true, then we need to do a shutdown,
|
|
// so that only the writable side will be cleaned up.
|
|
function onSocketFinish() {
|
|
// If still connecting - defer handling 'finish' until 'connect' will happen
|
|
if (this._connecting) {
|
|
debug('osF: not yet connected');
|
|
return this.once('connect', onSocketFinish);
|
|
}
|
|
|
|
debug('onSocketFinish');
|
|
if (!this.readable || this._readableState.ended) {
|
|
debug('oSF: ended, destroy', this._readableState);
|
|
return this.destroy();
|
|
}
|
|
|
|
debug('oSF: not ended, call shutdown()');
|
|
|
|
// otherwise, just shutdown, or destroy() if not possible
|
|
if (!this._handle || !this._handle.shutdown)
|
|
return this.destroy();
|
|
|
|
var req = { oncomplete: afterShutdown };
|
|
var err = this._handle.shutdown(req);
|
|
|
|
if (err)
|
|
return this._destroy(errnoException(err, 'shutdown'));
|
|
}
|
|
|
|
|
|
function afterShutdown(status, handle, req) {
|
|
var self = handle.owner;
|
|
|
|
debug('afterShutdown destroyed=%j', self.destroyed,
|
|
self._readableState);
|
|
|
|
// callback may come after call to destroy.
|
|
if (self.destroyed)
|
|
return;
|
|
|
|
if (self._readableState.ended) {
|
|
debug('readableState ended, destroying');
|
|
self.destroy();
|
|
} else {
|
|
self.once('_socketEnd', self.destroy);
|
|
}
|
|
}
|
|
|
|
// the EOF has been received, and no more bytes are coming.
|
|
// if the writable side has ended already, then clean everything
|
|
// up.
|
|
function onSocketEnd() {
|
|
// XXX Should not have to do as much crap in this function.
|
|
// ended should already be true, since this is called *after*
|
|
// the EOF errno and onread has eof'ed
|
|
debug('onSocketEnd', this._readableState);
|
|
this._readableState.ended = true;
|
|
if (this._readableState.endEmitted) {
|
|
this.readable = false;
|
|
maybeDestroy(this);
|
|
} else {
|
|
this.once('end', function() {
|
|
this.readable = false;
|
|
maybeDestroy(this);
|
|
});
|
|
this.read(0);
|
|
}
|
|
|
|
if (!this.allowHalfOpen) {
|
|
this.write = writeAfterFIN;
|
|
this.destroySoon();
|
|
}
|
|
}
|
|
|
|
// Provide a better error message when we call end() as a result
|
|
// of the other side sending a FIN. The standard 'write after end'
|
|
// is overly vague, and makes it seem like the user's code is to blame.
|
|
function writeAfterFIN(chunk, encoding, cb) {
|
|
if (util.isFunction(encoding)) {
|
|
cb = encoding;
|
|
encoding = null;
|
|
}
|
|
|
|
var er = new Error('This socket has been ended by the other party');
|
|
er.code = 'EPIPE';
|
|
var self = this;
|
|
// TODO: defer error events consistently everywhere, not just the cb
|
|
self.emit('error', er);
|
|
if (util.isFunction(cb)) {
|
|
process.nextTick(function() {
|
|
cb(er);
|
|
});
|
|
}
|
|
}
|
|
|
|
exports.Socket = Socket;
|
|
exports.Stream = Socket; // Legacy naming.
|
|
|
|
Socket.prototype.read = function(n) {
|
|
if (n === 0)
|
|
return stream.Readable.prototype.read.call(this, n);
|
|
|
|
this.read = stream.Readable.prototype.read;
|
|
this._consuming = true;
|
|
return this.read(n);
|
|
};
|
|
|
|
|
|
Socket.prototype.listen = function() {
|
|
debug('socket.listen');
|
|
var self = this;
|
|
self.on('connection', arguments[0]);
|
|
listen(self, null, null, null);
|
|
};
|
|
|
|
|
|
Socket.prototype.setTimeout = function(msecs, callback) {
|
|
if (msecs > 0 && isFinite(msecs)) {
|
|
timers.enroll(this, msecs);
|
|
timers._unrefActive(this);
|
|
if (callback) {
|
|
this.once('timeout', callback);
|
|
}
|
|
} else if (msecs === 0) {
|
|
timers.unenroll(this);
|
|
if (callback) {
|
|
this.removeListener('timeout', callback);
|
|
}
|
|
}
|
|
};
|
|
|
|
|
|
Socket.prototype._onTimeout = function() {
|
|
debug('_onTimeout');
|
|
this.emit('timeout');
|
|
};
|
|
|
|
|
|
Socket.prototype.setNoDelay = function(enable) {
|
|
// backwards compatibility: assume true when `enable` is omitted
|
|
if (this._handle && this._handle.setNoDelay)
|
|
this._handle.setNoDelay(util.isUndefined(enable) ? true : !!enable);
|
|
};
|
|
|
|
|
|
Socket.prototype.setKeepAlive = function(setting, msecs) {
|
|
if (this._handle && this._handle.setKeepAlive)
|
|
this._handle.setKeepAlive(setting, ~~(msecs / 1000));
|
|
};
|
|
|
|
|
|
Socket.prototype.address = function() {
|
|
return this._getsockname();
|
|
};
|
|
|
|
|
|
Object.defineProperty(Socket.prototype, 'readyState', {
|
|
get: function() {
|
|
if (this._connecting) {
|
|
return 'opening';
|
|
} else if (this.readable && this.writable) {
|
|
return 'open';
|
|
} else if (this.readable && !this.writable) {
|
|
return 'readOnly';
|
|
} else if (!this.readable && this.writable) {
|
|
return 'writeOnly';
|
|
} else {
|
|
return 'closed';
|
|
}
|
|
}
|
|
});
|
|
|
|
|
|
Object.defineProperty(Socket.prototype, 'bufferSize', {
|
|
get: function() {
|
|
if (this._handle) {
|
|
return this._handle.writeQueueSize + this._writableState.length;
|
|
}
|
|
}
|
|
});
|
|
|
|
|
|
// Just call handle.readStart until we have enough in the buffer
|
|
Socket.prototype._read = function(n) {
|
|
debug('_read');
|
|
|
|
if (this._connecting || !this._handle) {
|
|
debug('_read wait for connection');
|
|
this.once('connect', this._read.bind(this, n));
|
|
} else if (!this._handle.reading) {
|
|
// not already reading, start the flow
|
|
debug('Socket._read readStart');
|
|
this._handle.reading = true;
|
|
var err = this._handle.readStart();
|
|
if (err)
|
|
this._destroy(errnoException(err, 'read'));
|
|
}
|
|
};
|
|
|
|
|
|
Socket.prototype.end = function(data, encoding) {
|
|
stream.Duplex.prototype.end.call(this, data, encoding);
|
|
this.writable = false;
|
|
DTRACE_NET_STREAM_END(this);
|
|
|
|
// just in case we're waiting for an EOF.
|
|
if (this.readable && !this._readableState.endEmitted)
|
|
this.read(0);
|
|
else
|
|
maybeDestroy(this);
|
|
};
|
|
|
|
|
|
// Call whenever we set writable=false or readable=false
|
|
function maybeDestroy(socket) {
|
|
if (!socket.readable &&
|
|
!socket.writable &&
|
|
!socket.destroyed &&
|
|
!socket._connecting &&
|
|
!socket._writableState.length) {
|
|
socket.destroy();
|
|
}
|
|
}
|
|
|
|
|
|
Socket.prototype.destroySoon = function() {
|
|
if (this.writable)
|
|
this.end();
|
|
|
|
if (this._writableState.finished)
|
|
this.destroy();
|
|
else
|
|
this.once('finish', this.destroy);
|
|
};
|
|
|
|
|
|
Socket.prototype._destroy = function(exception, cb) {
|
|
debug('destroy');
|
|
|
|
var self = this;
|
|
|
|
function fireErrorCallbacks() {
|
|
if (cb) cb(exception);
|
|
if (exception && !self._writableState.errorEmitted) {
|
|
process.nextTick(function() {
|
|
self.emit('error', exception);
|
|
});
|
|
self._writableState.errorEmitted = true;
|
|
}
|
|
};
|
|
|
|
if (this.destroyed) {
|
|
debug('already destroyed, fire error callbacks');
|
|
fireErrorCallbacks();
|
|
return;
|
|
}
|
|
|
|
self._connecting = false;
|
|
|
|
this.readable = this.writable = false;
|
|
|
|
timers.unenroll(this);
|
|
|
|
debug('close');
|
|
if (this._handle) {
|
|
if (this !== process.stderr)
|
|
debug('close handle');
|
|
var isException = exception ? true : false;
|
|
this._handle.close(function() {
|
|
debug('emit close');
|
|
self.emit('close', isException);
|
|
});
|
|
this._handle.onread = noop;
|
|
this._handle = null;
|
|
}
|
|
|
|
// we set destroyed to true before firing error callbacks in order
|
|
// to make it re-entrance safe in case Socket.prototype.destroy()
|
|
// is called within callbacks
|
|
this.destroyed = true;
|
|
fireErrorCallbacks();
|
|
|
|
if (this.server) {
|
|
COUNTER_NET_SERVER_CONNECTION_CLOSE(this);
|
|
debug('has server');
|
|
this.server._connections--;
|
|
if (this.server._emitCloseIfDrained) {
|
|
this.server._emitCloseIfDrained();
|
|
}
|
|
}
|
|
};
|
|
|
|
|
|
Socket.prototype.destroy = function(exception) {
|
|
debug('destroy', exception);
|
|
this._destroy(exception);
|
|
};
|
|
|
|
|
|
// This function is called whenever the handle gets a
|
|
// buffer, or when there's an error reading.
|
|
function onread(nread, buffer) {
|
|
var handle = this;
|
|
var self = handle.owner;
|
|
assert(handle === self._handle, 'handle != self._handle');
|
|
|
|
timers._unrefActive(self);
|
|
|
|
debug('onread', nread);
|
|
|
|
if (nread > 0) {
|
|
debug('got data');
|
|
|
|
// read success.
|
|
// In theory (and in practice) calling readStop right now
|
|
// will prevent this from being called again until _read() gets
|
|
// called again.
|
|
|
|
// if it's not enough data, we'll just call handle.readStart()
|
|
// again right away.
|
|
self.bytesRead += nread;
|
|
|
|
// Optimization: emit the original buffer with end points
|
|
var ret = self.push(buffer);
|
|
|
|
if (handle.reading && !ret) {
|
|
handle.reading = false;
|
|
debug('readStop');
|
|
var err = handle.readStop();
|
|
if (err)
|
|
self._destroy(errnoException(err, 'read'));
|
|
}
|
|
return;
|
|
}
|
|
|
|
// if we didn't get any bytes, that doesn't necessarily mean EOF.
|
|
// wait for the next one.
|
|
if (nread === 0) {
|
|
debug('not any data, keep waiting');
|
|
return;
|
|
}
|
|
|
|
// Error, possibly EOF.
|
|
if (nread !== uv.UV_EOF) {
|
|
return self._destroy(errnoException(nread, 'read'));
|
|
}
|
|
|
|
debug('EOF');
|
|
|
|
if (self._readableState.length === 0) {
|
|
self.readable = false;
|
|
maybeDestroy(self);
|
|
}
|
|
|
|
// push a null to signal the end of data.
|
|
self.push(null);
|
|
|
|
// internal end event so that we know that the actual socket
|
|
// is no longer readable, and we can start the shutdown
|
|
// procedure. No need to wait for all the data to be consumed.
|
|
self.emit('_socketEnd');
|
|
}
|
|
|
|
|
|
Socket.prototype._getpeername = function() {
|
|
if (!this._handle || !this._handle.getpeername) {
|
|
return {};
|
|
}
|
|
if (!this._peername) {
|
|
var out = {};
|
|
var err = this._handle.getpeername(out);
|
|
if (err) return {}; // FIXME(bnoordhuis) Throw?
|
|
this._peername = out;
|
|
}
|
|
return this._peername;
|
|
};
|
|
|
|
|
|
Socket.prototype.__defineGetter__('remoteAddress', function() {
|
|
return this._getpeername().address;
|
|
});
|
|
|
|
Socket.prototype.__defineGetter__('remoteFamily', function() {
|
|
return this._getpeername().family;
|
|
});
|
|
|
|
Socket.prototype.__defineGetter__('remotePort', function() {
|
|
return this._getpeername().port;
|
|
});
|
|
|
|
|
|
Socket.prototype._getsockname = function() {
|
|
if (!this._handle || !this._handle.getsockname) {
|
|
return {};
|
|
}
|
|
if (!this._sockname) {
|
|
var out = {};
|
|
var err = this._handle.getsockname(out);
|
|
if (err) return {}; // FIXME(bnoordhuis) Throw?
|
|
this._sockname = out;
|
|
}
|
|
return this._sockname;
|
|
};
|
|
|
|
|
|
Socket.prototype.__defineGetter__('localAddress', function() {
|
|
return this._getsockname().address;
|
|
});
|
|
|
|
|
|
Socket.prototype.__defineGetter__('localPort', function() {
|
|
return this._getsockname().port;
|
|
});
|
|
|
|
|
|
Socket.prototype.write = function(chunk, encoding, cb) {
|
|
'use strict';
|
|
if (!util.isString(chunk) && !util.isBuffer(chunk))
|
|
throw new TypeError('invalid data');
|
|
return stream.Duplex.prototype.write.apply(this, arguments);
|
|
};
|
|
|
|
|
|
Socket.prototype._writeGeneric = function(writev, data, encoding, cb) {
|
|
// If we are still connecting, then buffer this for later.
|
|
// The Writable logic will buffer up any more writes while
|
|
// waiting for this one to be done.
|
|
if (this._connecting) {
|
|
this._pendingData = data;
|
|
this._pendingEncoding = encoding;
|
|
this.once('connect', function() {
|
|
this._writeGeneric(writev, data, encoding, cb);
|
|
});
|
|
return;
|
|
}
|
|
this._pendingData = null;
|
|
this._pendingEncoding = '';
|
|
|
|
timers._unrefActive(this);
|
|
|
|
if (!this._handle) {
|
|
this._destroy(new Error('This socket is closed.'), cb);
|
|
return false;
|
|
}
|
|
|
|
var req = { oncomplete: afterWrite, async: false };
|
|
var err;
|
|
|
|
if (writev) {
|
|
var chunks = new Array(data.length << 1);
|
|
for (var i = 0; i < data.length; i++) {
|
|
var entry = data[i];
|
|
var chunk = entry.chunk;
|
|
var enc = entry.encoding;
|
|
chunks[i * 2] = chunk;
|
|
chunks[i * 2 + 1] = enc;
|
|
}
|
|
err = this._handle.writev(req, chunks);
|
|
|
|
// Retain chunks
|
|
if (err === 0) req._chunks = chunks;
|
|
} else {
|
|
var enc;
|
|
if (util.isBuffer(data)) {
|
|
req.buffer = data; // Keep reference alive.
|
|
enc = 'buffer';
|
|
} else {
|
|
enc = encoding;
|
|
}
|
|
err = createWriteReq(req, this._handle, data, enc);
|
|
}
|
|
|
|
if (err)
|
|
return this._destroy(errnoException(err, 'write', req.error), cb);
|
|
|
|
this._bytesDispatched += req.bytes;
|
|
|
|
// If it was entirely flushed, we can write some more right now.
|
|
// However, if more is left in the queue, then wait until that clears.
|
|
if (req.async && this._handle.writeQueueSize != 0)
|
|
req.cb = cb;
|
|
else
|
|
cb();
|
|
};
|
|
|
|
|
|
Socket.prototype._writev = function(chunks, cb) {
|
|
this._writeGeneric(true, chunks, '', cb);
|
|
};
|
|
|
|
|
|
Socket.prototype._write = function(data, encoding, cb) {
|
|
this._writeGeneric(false, data, encoding, cb);
|
|
};
|
|
|
|
function createWriteReq(req, handle, data, encoding) {
|
|
switch (encoding) {
|
|
case 'binary':
|
|
return handle.writeBinaryString(req, data);
|
|
|
|
case 'buffer':
|
|
return handle.writeBuffer(req, data);
|
|
|
|
case 'utf8':
|
|
case 'utf-8':
|
|
return handle.writeUtf8String(req, data);
|
|
|
|
case 'ascii':
|
|
return handle.writeAsciiString(req, data);
|
|
|
|
case 'ucs2':
|
|
case 'ucs-2':
|
|
case 'utf16le':
|
|
case 'utf-16le':
|
|
return handle.writeUcs2String(req, data);
|
|
|
|
default:
|
|
return handle.writeBuffer(req, new Buffer(data, encoding));
|
|
}
|
|
}
|
|
|
|
|
|
Socket.prototype.__defineGetter__('bytesWritten', function() {
|
|
var bytes = this._bytesDispatched,
|
|
state = this._writableState,
|
|
data = this._pendingData,
|
|
encoding = this._pendingEncoding;
|
|
|
|
state.buffer.forEach(function(el) {
|
|
if (util.isBuffer(el.chunk))
|
|
bytes += el.chunk.length;
|
|
else
|
|
bytes += Buffer.byteLength(el.chunk, el.encoding);
|
|
});
|
|
|
|
if (data) {
|
|
if (util.isBuffer(data))
|
|
bytes += data.length;
|
|
else
|
|
bytes += Buffer.byteLength(data, encoding);
|
|
}
|
|
|
|
return bytes;
|
|
});
|
|
|
|
|
|
function afterWrite(status, handle, req, err) {
|
|
var self = handle.owner;
|
|
if (self !== process.stderr && self !== process.stdout)
|
|
debug('afterWrite', status);
|
|
|
|
// callback may come after call to destroy.
|
|
if (self.destroyed) {
|
|
debug('afterWrite destroyed');
|
|
return;
|
|
}
|
|
|
|
if (status < 0) {
|
|
var ex = errnoException(status, 'write', err);
|
|
debug('write failure', ex);
|
|
self._destroy(ex, req.cb);
|
|
return;
|
|
}
|
|
|
|
timers._unrefActive(self);
|
|
|
|
if (self !== process.stderr && self !== process.stdout)
|
|
debug('afterWrite call cb');
|
|
|
|
if (req.cb)
|
|
req.cb.call(self);
|
|
}
|
|
|
|
|
|
function connect(self, address, port, addressType, localAddress, localPort) {
|
|
// TODO return promise from Socket.prototype.connect which
|
|
// wraps _connectReq.
|
|
|
|
assert.ok(self._connecting);
|
|
|
|
var err;
|
|
if (localAddress || localPort) {
|
|
if (localAddress && !exports.isIP(localAddress))
|
|
err = new TypeError(
|
|
'localAddress should be a valid IP: ' + localAddress);
|
|
|
|
if (localPort && !util.isNumber(localPort))
|
|
err = new TypeError('localPort should be a number: ' + localPort);
|
|
|
|
var bind;
|
|
|
|
switch (addressType) {
|
|
case 4:
|
|
if (!localAddress)
|
|
localAddress = '0.0.0.0';
|
|
bind = self._handle.bind;
|
|
break;
|
|
case 6:
|
|
if (!localAddress)
|
|
localAddress = '::';
|
|
bind = self._handle.bind6;
|
|
break;
|
|
default:
|
|
err = new TypeError('Invalid addressType: ' + addressType);
|
|
break;
|
|
}
|
|
|
|
if (err) {
|
|
self._destroy(err);
|
|
return;
|
|
}
|
|
|
|
debug('binding to localAddress: %s and localPort: %d',
|
|
localAddress,
|
|
localPort);
|
|
|
|
bind = bind.bind(self._handle);
|
|
err = bind(localAddress, localPort);
|
|
|
|
if (err) {
|
|
self._destroy(errnoException(err, 'bind'));
|
|
return;
|
|
}
|
|
}
|
|
|
|
var req = { oncomplete: afterConnect };
|
|
if (addressType === 6 || addressType === 4) {
|
|
port = port | 0;
|
|
if (port <= 0 || port > 65535)
|
|
throw new RangeError('Port should be > 0 and < 65536');
|
|
|
|
if (addressType === 6) {
|
|
err = self._handle.connect6(req, address, port);
|
|
} else if (addressType === 4) {
|
|
err = self._handle.connect(req, address, port);
|
|
}
|
|
} else {
|
|
err = self._handle.connect(req, address, afterConnect);
|
|
}
|
|
|
|
if (err) {
|
|
self._destroy(errnoException(err, 'connect'));
|
|
}
|
|
}
|
|
|
|
|
|
Socket.prototype.connect = function(options, cb) {
|
|
if (this.write !== Socket.prototype.write)
|
|
this.write = Socket.prototype.write;
|
|
|
|
if (!util.isObject(options)) {
|
|
// Old API:
|
|
// connect(port, [host], [cb])
|
|
// connect(path, [cb]);
|
|
var args = normalizeConnectArgs(arguments);
|
|
return Socket.prototype.connect.apply(this, args);
|
|
}
|
|
|
|
if (this.destroyed) {
|
|
this._readableState.reading = false;
|
|
this._readableState.ended = false;
|
|
this._readableState.endEmitted = false;
|
|
this._writableState.ended = false;
|
|
this._writableState.ending = false;
|
|
this._writableState.finished = false;
|
|
this._writableState.errorEmitted = false;
|
|
this.destroyed = false;
|
|
this._handle = null;
|
|
}
|
|
|
|
var self = this;
|
|
var pipe = !!options.path;
|
|
debug('pipe', pipe, options.path);
|
|
|
|
if (!this._handle) {
|
|
this._handle = pipe ? createPipe() : createTCP();
|
|
initSocketHandle(this);
|
|
}
|
|
|
|
if (util.isFunction(cb)) {
|
|
self.once('connect', cb);
|
|
}
|
|
|
|
timers._unrefActive(this);
|
|
|
|
self._connecting = true;
|
|
self.writable = true;
|
|
|
|
if (pipe) {
|
|
connect(self, options.path);
|
|
|
|
} else if (!options.host) {
|
|
debug('connect: missing host');
|
|
self._host = '127.0.0.1';
|
|
connect(self, self._host, options.port, 4);
|
|
|
|
} else {
|
|
var dns = require('dns');
|
|
var host = options.host;
|
|
var dnsopts = {
|
|
family: options.family,
|
|
hints: 0
|
|
};
|
|
|
|
if (dnsopts.family !== 4 && dnsopts.family !== 6)
|
|
dnsopts.hints = dns.ADDRCONFIG | dns.V4MAPPED;
|
|
|
|
debug('connect: find host ' + host);
|
|
debug('connect: dns options ' + dnsopts);
|
|
self._host = host;
|
|
dns.lookup(host, dnsopts, function(err, ip, addressType) {
|
|
self.emit('lookup', err, ip, addressType);
|
|
|
|
// It's possible we were destroyed while looking this up.
|
|
// XXX it would be great if we could cancel the promise returned by
|
|
// the look up.
|
|
if (!self._connecting) return;
|
|
|
|
if (err) {
|
|
// net.createConnection() creates a net.Socket object and
|
|
// immediately calls net.Socket.connect() on it (that's us).
|
|
// There are no event listeners registered yet so defer the
|
|
// error event to the next tick.
|
|
process.nextTick(function() {
|
|
self.emit('error', err);
|
|
self._destroy();
|
|
});
|
|
} else {
|
|
timers._unrefActive(self);
|
|
|
|
addressType = addressType || 4;
|
|
|
|
// node_net.cc handles null host names graciously but user land
|
|
// expects remoteAddress to have a meaningful value
|
|
ip = ip || (addressType === 4 ? '127.0.0.1' : '0:0:0:0:0:0:0:1');
|
|
|
|
connect(self,
|
|
ip,
|
|
options.port,
|
|
addressType,
|
|
options.localAddress,
|
|
options.localPort);
|
|
}
|
|
});
|
|
}
|
|
return self;
|
|
};
|
|
|
|
|
|
Socket.prototype.ref = function() {
|
|
if (this._handle)
|
|
this._handle.ref();
|
|
};
|
|
|
|
|
|
Socket.prototype.unref = function() {
|
|
if (this._handle)
|
|
this._handle.unref();
|
|
};
|
|
|
|
|
|
|
|
function afterConnect(status, handle, req, readable, writable) {
|
|
var self = handle.owner;
|
|
|
|
// callback may come after call to destroy
|
|
if (self.destroyed) {
|
|
return;
|
|
}
|
|
|
|
assert(handle === self._handle, 'handle != self._handle');
|
|
|
|
debug('afterConnect');
|
|
|
|
assert.ok(self._connecting);
|
|
self._connecting = false;
|
|
|
|
if (status == 0) {
|
|
self.readable = readable;
|
|
self.writable = writable;
|
|
timers._unrefActive(self);
|
|
|
|
self.emit('connect');
|
|
|
|
// start the first read, or get an immediate EOF.
|
|
// this doesn't actually consume any bytes, because len=0.
|
|
if (readable && !self.isPaused())
|
|
self.read(0);
|
|
|
|
} else {
|
|
self._connecting = false;
|
|
self._destroy(errnoException(status, 'connect'));
|
|
}
|
|
}
|
|
|
|
|
|
function Server(/* [ options, ] listener */) {
|
|
if (!(this instanceof Server)) return new Server(arguments[0], arguments[1]);
|
|
events.EventEmitter.call(this);
|
|
|
|
var self = this;
|
|
|
|
var options;
|
|
|
|
if (util.isFunction(arguments[0])) {
|
|
options = {};
|
|
self.on('connection', arguments[0]);
|
|
} else {
|
|
options = arguments[0] || {};
|
|
|
|
if (util.isFunction(arguments[1])) {
|
|
self.on('connection', arguments[1]);
|
|
}
|
|
}
|
|
|
|
this._connections = 0;
|
|
|
|
Object.defineProperty(this, 'connections', {
|
|
get: util.deprecate(function() {
|
|
|
|
if (self._usingSlaves) {
|
|
return null;
|
|
}
|
|
return self._connections;
|
|
}, 'connections property is deprecated. Use getConnections() method'),
|
|
set: util.deprecate(function(val) {
|
|
return (self._connections = val);
|
|
}, 'connections property is deprecated. Use getConnections() method'),
|
|
configurable: true, enumerable: false
|
|
});
|
|
|
|
this._handle = null;
|
|
this._usingSlaves = false;
|
|
this._slaves = [];
|
|
|
|
this.allowHalfOpen = options.allowHalfOpen || false;
|
|
this.pauseOnConnect = !!options.pauseOnConnect;
|
|
}
|
|
util.inherits(Server, events.EventEmitter);
|
|
exports.Server = Server;
|
|
|
|
|
|
function toNumber(x) { return (x = Number(x)) >= 0 ? x : false; }
|
|
|
|
function _listen(handle, backlog) {
|
|
// Use a backlog of 512 entries. We pass 511 to the listen() call because
|
|
// the kernel does: backlogsize = roundup_pow_of_two(backlogsize + 1);
|
|
// which will thus give us a backlog of 512 entries.
|
|
return handle.listen(backlog || 511);
|
|
}
|
|
|
|
var createServerHandle = exports._createServerHandle =
|
|
function(address, port, addressType, fd) {
|
|
var err = 0;
|
|
// assign handle in listen, and clean up if bind or listen fails
|
|
var handle;
|
|
|
|
var isTCP = false;
|
|
if (util.isNumber(fd) && fd >= 0) {
|
|
try {
|
|
handle = createHandle(fd);
|
|
}
|
|
catch (e) {
|
|
// Not a fd we can listen on. This will trigger an error.
|
|
debug('listen invalid fd=' + fd + ': ' + e.message);
|
|
return uv.UV_EINVAL;
|
|
}
|
|
handle.open(fd);
|
|
handle.readable = true;
|
|
handle.writable = true;
|
|
assert(!address && !port);
|
|
} else if (port === -1 && addressType === -1) {
|
|
handle = createPipe();
|
|
if (process.platform === 'win32') {
|
|
var instances = parseInt(process.env.NODE_PENDING_PIPE_INSTANCES);
|
|
if (!isNaN(instances)) {
|
|
handle.setPendingInstances(instances);
|
|
}
|
|
}
|
|
} else {
|
|
handle = createTCP();
|
|
isTCP = true;
|
|
}
|
|
|
|
if (address || port || isTCP) {
|
|
debug('bind to ' + (address || 'anycast'));
|
|
if (!address) {
|
|
// Try binding to ipv6 first
|
|
err = handle.bind6('::', port);
|
|
if (err) {
|
|
handle.close();
|
|
// Fallback to ipv4
|
|
return createServerHandle('0.0.0.0', port);
|
|
}
|
|
} else if (addressType === 6) {
|
|
err = handle.bind6(address, port);
|
|
} else {
|
|
err = handle.bind(address, port);
|
|
}
|
|
}
|
|
|
|
if (err) {
|
|
handle.close();
|
|
return err;
|
|
}
|
|
|
|
return handle;
|
|
};
|
|
|
|
|
|
Server.prototype._listen2 = function(address, port, addressType, backlog, fd) {
|
|
debug('listen2', address, port, addressType, backlog);
|
|
var self = this;
|
|
|
|
// If there is not yet a handle, we need to create one and bind.
|
|
// In the case of a server sent via IPC, we don't need to do this.
|
|
if (!self._handle) {
|
|
debug('_listen2: create a handle');
|
|
var rval = createServerHandle(address, port, addressType, fd);
|
|
if (util.isNumber(rval)) {
|
|
var error = errnoException(rval, 'listen');
|
|
process.nextTick(function() {
|
|
self.emit('error', error);
|
|
});
|
|
return;
|
|
}
|
|
self._handle = rval;
|
|
} else {
|
|
debug('_listen2: have a handle already');
|
|
}
|
|
|
|
self._handle.onconnection = onconnection;
|
|
self._handle.owner = self;
|
|
|
|
var err = _listen(self._handle, backlog);
|
|
|
|
if (err) {
|
|
var ex = errnoException(err, 'listen');
|
|
self._handle.close();
|
|
self._handle = null;
|
|
process.nextTick(function() {
|
|
self.emit('error', ex);
|
|
});
|
|
return;
|
|
}
|
|
|
|
// generate connection key, this should be unique to the connection
|
|
this._connectionKey = addressType + ':' + address + ':' + port;
|
|
|
|
process.nextTick(function() {
|
|
// ensure handle hasn't closed
|
|
if (self._handle)
|
|
self.emit('listening');
|
|
});
|
|
};
|
|
|
|
|
|
function listen(self, address, port, addressType, backlog, fd, exclusive) {
|
|
exclusive = !!exclusive;
|
|
|
|
if (!cluster) cluster = require('cluster');
|
|
|
|
if (cluster.isMaster || exclusive) {
|
|
self._listen2(address, port, addressType, backlog, fd);
|
|
return;
|
|
}
|
|
|
|
cluster._getServer(self, address, port, addressType, fd, cb);
|
|
|
|
function cb(err, handle) {
|
|
// EADDRINUSE may not be reported until we call listen(). To complicate
|
|
// matters, a failed bind() followed by listen() will implicitly bind to
|
|
// a random port. Ergo, check that the socket is bound to the expected
|
|
// port before calling listen().
|
|
//
|
|
// FIXME(bnoordhuis) Doesn't work for pipe handles, they don't have a
|
|
// getsockname() method. Non-issue for now, the cluster module doesn't
|
|
// really support pipes anyway.
|
|
if (err === 0 && port > 0 && handle.getsockname) {
|
|
var out = {};
|
|
err = handle.getsockname(out);
|
|
if (err === 0 && port !== out.port)
|
|
err = uv.UV_EADDRINUSE;
|
|
}
|
|
|
|
if (err)
|
|
return self.emit('error', errnoException(err, 'bind'));
|
|
|
|
self._handle = handle;
|
|
self._listen2(address, port, addressType, backlog, fd);
|
|
}
|
|
}
|
|
|
|
|
|
Server.prototype.listen = function() {
|
|
var self = this;
|
|
|
|
var lastArg = arguments[arguments.length - 1];
|
|
if (util.isFunction(lastArg)) {
|
|
self.once('listening', lastArg);
|
|
}
|
|
|
|
var port = toNumber(arguments[0]);
|
|
|
|
// The third optional argument is the backlog size.
|
|
// When the ip is omitted it can be the second argument.
|
|
var backlog = toNumber(arguments[1]) || toNumber(arguments[2]);
|
|
|
|
var TCP = process.binding('tcp_wrap').TCP;
|
|
|
|
if (arguments.length === 0 || util.isFunction(arguments[0])) {
|
|
// Bind to a random port.
|
|
listen(self, null, 0, null, backlog);
|
|
} else if (util.isObject(arguments[0])) {
|
|
var h = arguments[0];
|
|
h = h._handle || h.handle || h;
|
|
|
|
if (h instanceof TCP) {
|
|
self._handle = h;
|
|
listen(self, null, -1, -1, backlog);
|
|
} else if (util.isNumber(h.fd) && h.fd >= 0) {
|
|
listen(self, null, null, null, backlog, h.fd);
|
|
} else {
|
|
// The first argument is a configuration object
|
|
if (h.backlog)
|
|
backlog = h.backlog;
|
|
|
|
if (util.isNumber(h.port)) {
|
|
if (h.host)
|
|
listenAfterLookup(h.port, h.host, backlog, h.exclusive);
|
|
else
|
|
listen(self, null, h.port, 4, backlog, undefined, h.exclusive);
|
|
} else if (h.path && isPipeName(h.path)) {
|
|
var pipeName = self._pipeName = h.path;
|
|
listen(self, pipeName, -1, -1, backlog, undefined, h.exclusive);
|
|
} else {
|
|
throw new Error('Invalid listen argument: ' + h);
|
|
}
|
|
}
|
|
} else if (isPipeName(arguments[0])) {
|
|
// UNIX socket or Windows pipe.
|
|
var pipeName = self._pipeName = arguments[0];
|
|
listen(self, pipeName, -1, -1, backlog);
|
|
|
|
} else if (util.isUndefined(arguments[1]) ||
|
|
util.isFunction(arguments[1]) ||
|
|
util.isNumber(arguments[1])) {
|
|
// The first argument is the port, no IP given.
|
|
listen(self, null, port, 4, backlog);
|
|
|
|
} else {
|
|
// The first argument is the port, the second an IP.
|
|
listenAfterLookup(port, arguments[1], backlog);
|
|
}
|
|
|
|
function listenAfterLookup(port, address, backlog, exclusive) {
|
|
require('dns').lookup(address, function(err, ip, addressType) {
|
|
if (err) {
|
|
self.emit('error', err);
|
|
} else {
|
|
addressType = ip ? addressType : 4;
|
|
listen(self, ip, port, addressType, backlog, undefined, exclusive);
|
|
}
|
|
});
|
|
}
|
|
|
|
return self;
|
|
};
|
|
|
|
Server.prototype.address = function() {
|
|
if (this._handle && this._handle.getsockname) {
|
|
var out = {};
|
|
var err = this._handle.getsockname(out);
|
|
// TODO(bnoordhuis) Check err and throw?
|
|
return out;
|
|
} else if (this._pipeName) {
|
|
return this._pipeName;
|
|
} else {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
function onconnection(err, clientHandle) {
|
|
var handle = this;
|
|
var self = handle.owner;
|
|
|
|
debug('onconnection');
|
|
|
|
if (err) {
|
|
self.emit('error', errnoException(err, 'accept'));
|
|
return;
|
|
}
|
|
|
|
if (self.maxConnections && self._connections >= self.maxConnections) {
|
|
clientHandle.close();
|
|
return;
|
|
}
|
|
|
|
var socket = new Socket({
|
|
handle: clientHandle,
|
|
allowHalfOpen: self.allowHalfOpen,
|
|
pauseOnCreate: self.pauseOnConnect
|
|
});
|
|
socket.readable = socket.writable = true;
|
|
|
|
|
|
self._connections++;
|
|
socket.server = self;
|
|
|
|
DTRACE_NET_SERVER_CONNECTION(socket);
|
|
COUNTER_NET_SERVER_CONNECTION(socket);
|
|
self.emit('connection', socket);
|
|
}
|
|
|
|
|
|
Server.prototype.getConnections = function(cb) {
|
|
function end(err, connections) {
|
|
process.nextTick(function() {
|
|
cb(err, connections);
|
|
});
|
|
}
|
|
|
|
if (!this._usingSlaves) {
|
|
return end(null, this._connections);
|
|
}
|
|
|
|
// Poll slaves
|
|
var left = this._slaves.length,
|
|
total = this._connections;
|
|
|
|
function oncount(err, count) {
|
|
if (err) {
|
|
left = -1;
|
|
return end(err);
|
|
}
|
|
|
|
total += count;
|
|
if (--left === 0) return end(null, total);
|
|
}
|
|
|
|
this._slaves.forEach(function(slave) {
|
|
slave.getConnections(oncount);
|
|
});
|
|
};
|
|
|
|
|
|
Server.prototype.close = function(cb) {
|
|
function onSlaveClose() {
|
|
if (--left !== 0) return;
|
|
|
|
self._connections = 0;
|
|
self._emitCloseIfDrained();
|
|
}
|
|
|
|
if (cb) {
|
|
if (!this._handle) {
|
|
this.once('close', function() {
|
|
cb(new Error('Not running'));
|
|
});
|
|
} else {
|
|
this.once('close', cb);
|
|
}
|
|
}
|
|
|
|
if (this._handle) {
|
|
this._handle.close();
|
|
this._handle = null;
|
|
}
|
|
|
|
if (this._usingSlaves) {
|
|
var self = this,
|
|
left = this._slaves.length;
|
|
|
|
// Increment connections to be sure that, even if all sockets will be closed
|
|
// during polling of slaves, `close` event will be emitted only once.
|
|
this._connections++;
|
|
|
|
// Poll slaves
|
|
this._slaves.forEach(function(slave) {
|
|
slave.close(onSlaveClose);
|
|
});
|
|
} else {
|
|
this._emitCloseIfDrained();
|
|
}
|
|
|
|
return this;
|
|
};
|
|
|
|
Server.prototype._emitCloseIfDrained = function() {
|
|
debug('SERVER _emitCloseIfDrained');
|
|
var self = this;
|
|
|
|
if (self._handle || self._connections) {
|
|
debug('SERVER handle? %j connections? %d',
|
|
!!self._handle, self._connections);
|
|
return;
|
|
}
|
|
|
|
process.nextTick(function() {
|
|
debug('SERVER: emit close');
|
|
self.emit('close');
|
|
});
|
|
};
|
|
|
|
|
|
Server.prototype.listenFD = util.deprecate(function(fd, type) {
|
|
return this.listen({ fd: fd });
|
|
}, 'listenFD is deprecated. Use listen({fd: <number>}).');
|
|
|
|
Server.prototype._setupSlave = function(socketList) {
|
|
this._usingSlaves = true;
|
|
this._slaves.push(socketList);
|
|
};
|
|
|
|
Server.prototype.ref = function() {
|
|
if (this._handle)
|
|
this._handle.ref();
|
|
};
|
|
|
|
Server.prototype.unref = function() {
|
|
if (this._handle)
|
|
this._handle.unref();
|
|
};
|
|
|
|
|
|
// TODO: isIP should be moved to the DNS code. Putting it here now because
|
|
// this is what the legacy system did.
|
|
exports.isIP = cares.isIP;
|
|
|
|
|
|
exports.isIPv4 = function(input) {
|
|
return exports.isIP(input) === 4;
|
|
};
|
|
|
|
|
|
exports.isIPv6 = function(input) {
|
|
return exports.isIP(input) === 6;
|
|
};
|
|
|
|
|
|
if (process.platform === 'win32') {
|
|
var simultaneousAccepts;
|
|
|
|
exports._setSimultaneousAccepts = function(handle) {
|
|
if (util.isUndefined(handle)) {
|
|
return;
|
|
}
|
|
|
|
if (util.isUndefined(simultaneousAccepts)) {
|
|
simultaneousAccepts = (process.env.NODE_MANY_ACCEPTS &&
|
|
process.env.NODE_MANY_ACCEPTS !== '0');
|
|
}
|
|
|
|
if (handle._simultaneousAccepts !== simultaneousAccepts) {
|
|
handle.setSimultaneousAccepts(simultaneousAccepts);
|
|
handle._simultaneousAccepts = simultaneousAccepts;
|
|
}
|
|
};
|
|
} else {
|
|
exports._setSimultaneousAccepts = function(handle) {};
|
|
}
|