Skip to content

Commit

Permalink
net: fix race write() before and after connect()
Browse files Browse the repository at this point in the history
  • Loading branch information
koichik committed Feb 28, 2012
1 parent 44daa98 commit cd5d247
Show file tree
Hide file tree
Showing 2 changed files with 50 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 @@ -647,8 +647,6 @@ function afterConnect(status, handle, req, readable, writable) {
handle.readStart();
}

self.emit('connect');

if (self._connectQueue) {
debug('Drain the connect queue');
for (var i = 0; i < self._connectQueue.length; i++) {
Expand All @@ -657,6 +655,8 @@ function afterConnect(status, handle, req, readable, writable) {
self._connectQueueCleanUp();
}

self.emit('connect');

if (self._flags & FLAG_SHUTDOWNQUED) {
// end called before connected - call end now with no data
self._flags &= ~FLAG_SHUTDOWNQUED;
Expand Down
48 changes: 48 additions & 0 deletions test/simple/test-net-write-connect-write.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
// 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 common = require('../common');
var assert = require('assert');
var net = require('net');

var received = '';

var server = net.createServer(function(socket) {
socket.pipe(socket);
}).listen(common.PORT, function() {
var conn = net.connect(common.PORT);
conn.setEncoding('utf8');
conn.write('before');
conn.on('connect', function() {
conn.write('after');
});
conn.on('data', function(buf) {
received += buf;
conn.end();
});
conn.on('end', function() {
server.close();
});
});

process.on('exit', function() {
assert.equal(received, 'before' + 'after');
});

0 comments on commit cd5d247

Please sign in to comment.