diff options
author | Brian White <mscdex@mscdex.net> | 2016-02-14 12:25:40 -0500 |
---|---|---|
committer | Brian White <mscdex@mscdex.net> | 2016-02-17 14:15:47 -0500 |
commit | 92212014ffc95a8bc5e56fcc2c076bac63c20c23 (patch) | |
tree | bbad5eaaea43383a56be4d1a8511aa66833dc32c | |
parent | 7885b1d7aa11434ae5692241ea2fdd4baccd10e5 (diff) | |
download | android-node-v8-92212014ffc95a8bc5e56fcc2c076bac63c20c23.tar.gz android-node-v8-92212014ffc95a8bc5e56fcc2c076bac63c20c23.tar.bz2 android-node-v8-92212014ffc95a8bc5e56fcc2c076bac63c20c23.zip |
stream: fix no data on partial decode
Before this commit, it was possible to push a partial character
to a readable stream where it was decoded as an empty string and
then added to the internal buffer. This caused the stream to not
emit any data, even when the rest of the character bytes were pushed
separately, because of a non-zero length check of the first chunk in
the internal buffer.
Fixes: https://github.com/nodejs/node/issues/5223
PR-URL: https://github.com/nodejs/node/pull/5226
Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
-rw-r--r-- | lib/_stream_readable.js | 37 | ||||
-rw-r--r-- | test/parallel/test-stream2-decode-partial.js | 23 |
2 files changed, 45 insertions, 15 deletions
diff --git a/lib/_stream_readable.js b/lib/_stream_readable.js index c64333c58a..1ad85dc81c 100644 --- a/lib/_stream_readable.js +++ b/lib/_stream_readable.js @@ -136,26 +136,33 @@ function readableAddChunk(stream, state, chunk, encoding, addToFront) { const e = new Error('stream.unshift() after end event'); stream.emit('error', e); } else { - if (state.decoder && !addToFront && !encoding) + var skipAdd; + if (state.decoder && !addToFront && !encoding) { chunk = state.decoder.write(chunk); + skipAdd = (!state.objectMode && chunk.length === 0); + } if (!addToFront) state.reading = false; - // if we want the data now, just emit it. - if (state.flowing && state.length === 0 && !state.sync) { - stream.emit('data', chunk); - stream.read(0); - } else { - // update the buffer info. - state.length += state.objectMode ? 1 : chunk.length; - if (addToFront) - state.buffer.unshift(chunk); - else - state.buffer.push(chunk); - - if (state.needReadable) - emitReadable(stream); + // Don't add to the buffer if we've decoded to an empty string chunk and + // we're not in object mode + if (!skipAdd) { + // if we want the data now, just emit it. + if (state.flowing && state.length === 0 && !state.sync) { + stream.emit('data', chunk); + stream.read(0); + } else { + // update the buffer info. + state.length += state.objectMode ? 1 : chunk.length; + if (addToFront) + state.buffer.unshift(chunk); + else + state.buffer.push(chunk); + + if (state.needReadable) + emitReadable(stream); + } } maybeReadMore(stream, state); diff --git a/test/parallel/test-stream2-decode-partial.js b/test/parallel/test-stream2-decode-partial.js new file mode 100644 index 0000000000..b58e192b9c --- /dev/null +++ b/test/parallel/test-stream2-decode-partial.js @@ -0,0 +1,23 @@ +'use strict'; +require('../common'); +const Readable = require('_stream_readable'); +const assert = require('assert'); + +var buf = ''; +const euro = new Buffer([0xE2, 0x82, 0xAC]); +const cent = new Buffer([0xC2, 0xA2]); +const source = Buffer.concat([euro, cent]); + +const readable = Readable({ encoding: 'utf8' }); +readable.push(source.slice(0, 2)); +readable.push(source.slice(2, 4)); +readable.push(source.slice(4, 6)); +readable.push(null); + +readable.on('data', function(data) { + buf += data; +}); + +process.on('exit', function() { + assert.strictEqual(buf, '€¢'); +}); |