Parser: ensure no socket read if push() calls _read() during body finish

This fixes an edge case where the following happens:

 * a body stream had data pushed to it such that the highWaterMark was reached
 * when the body stream buffer dips below highWaterMark, _read() is called
 * _read() tries to read more data from the socket, which pushes the last part of the body
 * the last body part push() again calls _read() which in turn reads more data from the socket
 * at this point the parser state is not stable because _body._read and _body are not reset yet, this causes the parser to potentially try to start reading the beginning of a response in the middle of the data for another fetch result for example

Fixes #345
fork
Brian White 10 years ago
parent 42a486d2df
commit 8376f212e6

@ -89,25 +89,26 @@ Parser.prototype._parse = function(data) {
if (this._literallen > 0) {
if (this._body) {
var body = this._body;
if (datalen > this._literallen) {
if (datalen >= this._literallen) {
var litlen = this._literallen;
i = litlen;
this._literallen = 0;
this._body = undefined;
body._read = EMPTY_READCB;
if (datalen > litlen)
body.push(data.slice(0, litlen));
else
body.push(data);
body.push(null);
} else {
this._literallen -= datalen;
var r = body.push(data);
if (!r && this._literallen > 0) {
if (!r) {
body._read = this._cbReadable;
return;
}
i = datalen;
}
if (this._literallen === 0) {
this._body = undefined;
body._read = EMPTY_READCB;
body.push(null);
}
} else {
if (datalen > this._literallen)
this._literals.push(data.slice(0, this._literallen));

@ -0,0 +1,177 @@
// Test for _at least_ GH Issues #345, #379, #392, #411
var assert = require('assert'),
net = require('net'),
crypto = require('crypto'),
Imap = require('../lib/Connection');
var result = [],
body = [],
bodyInfo = [];
var CRLF = '\r\n';
// generate data larger than highWaterMark
var bytes = crypto.pseudoRandomBytes(10240).toString('hex');
var RESPONSES = [
['* CAPABILITY IMAP4rev1 UNSELECT IDLE NAMESPACE QUOTA CHILDREN',
'A0 OK Thats all she wrote!',
''
].join(CRLF),
['* CAPABILITY IMAP4rev1 UNSELECT IDLE NAMESPACE QUOTA CHILDREN UIDPLUS MOVE',
'A1 OK authenticated (Success)',
''
].join(CRLF),
['* NAMESPACE (("" "/")) NIL NIL',
'A2 OK Success',
''
].join(CRLF),
['* LIST (\\Noselect) "/" "/"',
'A3 OK Success',
''
].join(CRLF),
['* FLAGS (\\Answered \\Flagged \\Draft \\Deleted \\Seen)',
'* OK [PERMANENTFLAGS ()] Flags permitted.',
'* OK [UIDVALIDITY 2] UIDs valid.',
'* 685 EXISTS',
'* 0 RECENT',
'* OK [UIDNEXT 4422] Predicted next UID.',
'A4 OK [READ-ONLY] INBOX selected. (Success)',
''
].join(CRLF),
'* 1 FETCH (UID 1000 FLAGS (\\Seen) INTERNALDATE "05-Sep-2004 00:38:03 +0000" BODY[TEXT] {'
+ bytes.length
+ '}'
+ CRLF
+ bytes.substring(0, 20000),
['* BYE LOGOUT Requested',
'A6 OK good day (Success)',
''
].join(CRLF)
];
var EXPECTED = [
'A0 CAPABILITY',
'A1 LOGIN "foo" "bar"',
'A2 NAMESPACE',
'A3 LIST "" ""',
'A4 EXAMINE "INBOX"',
'A5 FETCH 1,2 (UID FLAGS INTERNALDATE BODY.PEEK[TEXT])',
'A6 LOGOUT'
];
var exp = -1,
res = -1;
var srv = net.createServer(function(sock) {
sock.write('* OK asdf\r\n');
var buf = '', lines;
sock.on('data', function(data) {
buf += data.toString('utf8');
if (buf.indexOf(CRLF) > -1) {
lines = buf.split(CRLF);
buf = lines.pop();
lines.forEach(function(l) {
assert(l === EXPECTED[++exp], 'Unexpected client request: ' + l);
assert(RESPONSES[++res], 'No response for client request: ' + l);
sock.write(RESPONSES[res]);
});
}
});
});
srv.listen(0, '127.0.0.1', function() {
var port = srv.address().port;
var imap = new Imap({
user: 'foo',
password: 'bar',
host: '127.0.0.1',
port: port,
keepalive: false,
debug: console.log
});
imap.on('ready', function() {
srv.close();
imap.openBox('INBOX', true, function() {
var f = imap.seq.fetch([1,2], { bodies: ['TEXT'] });
var nbody = -1;
f.on('message', function(m) {
m.on('body', function(stream, info) {
++nbody;
bodyInfo.push(info);
body[nbody] = '';
if (nbody === 0) {
// first allow body.push() to return false in parser.js
setTimeout(function() {
stream.on('data', function(chunk) {
body[nbody] += chunk.toString('binary');
});
setTimeout(function() {
var oldRead = stream._read,
readCalled = false;
stream._read = function(n) {
readCalled = true;
stream._read = oldRead;
imap._sock.push(bytes.substring(100, 200)
+ ')'
+ CRLF
+ 'A5 OK Success'
+ CRLF);
imap._parser._tryread();
};
imap._sock.push(bytes.substring(20000)
+ ')'
+ CRLF
+ '* 2 FETCH (UID 1001 FLAGS (\\Seen) INTERNALDATE "05-Sep-2004 00:38:13 +0000" BODY[TEXT] {200}'
+ CRLF
+ bytes.substring(0, 100));
// if we got this far, then we didn't get an exception and we
// are running with the bug fix in place
if (!readCalled) {
imap._sock.push(bytes.substring(100, 200)
+ ')'
+ CRLF
+ 'A5 OK Success'
+ CRLF);
}
}, 100);
}, 100);
} else {
stream.on('data', function(chunk) {
body[nbody] += chunk.toString('binary');
});
}
});
m.on('attributes', function(attrs) {
result.push(attrs);
});
});
f.on('end', function() {
imap.end();
});
});
});
imap.connect();
});
process.once('exit', function() {
assert.deepEqual(result, [{
uid: 1000,
date: new Date('05-Sep-2004 00:38:03 +0000'),
flags: [ '\\Seen' ]
}, {
uid: 1001,
date: new Date('05-Sep-2004 00:38:13 +0000'),
flags: [ '\\Seen' ]
}]);
assert.deepEqual(body, [bytes, bytes.substring(0, 200)]);
assert.deepEqual(bodyInfo, [{
seqno: 1,
which: 'TEXT',
size: bytes.length
}, {
seqno: 2,
which: 'TEXT',
size: 200
}]);
});
Loading…
Cancel
Save