yetangitu/OpenSTF
1
0
Fork 0
mirror of https://github.com/openstf/stf synced 2025-10-04 02:09:32 +02:00
Code Issues Releases Wiki Activity

Fix major oversight in MessageStream that was causing multi-chunk messages to fail.

Browse source
This commit is contained in:
Simo Kinnunen 2014-04-01 17:54:31 +09:00
parent 09eb8c539d
commit 2bf8f2352b
2 changed files with 29 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

48
lib/wire/messagestream.js
View file

@ -14,35 +14,33 @@ util.inherits(DelimitedStream, stream.Transform)
DelimitedStream.prototype._transform = function(chunk, encoding, done) {
this._buffer = Buffer.concat([this._buffer, chunk])
var lo = 0
var hi = this._buffer.length
while (lo < hi) {
while (this._buffer.length) {
if (this._readingLength) {
while (lo < hi) {
var byte = this._buffer[lo++]
if (byte & (1 << 7)) {
this._length += (byte & 0x7f) << (7 * this._lengthIndex)
this._lengthIndex += 1
this._readingLength = true
}
else {
this._length += (byte & 0x7f) << (7 * this._lengthIndex)
this._lengthIndex = 0
this._readingLength = false
break
}
var byte = this._buffer[0]
if (byte & (1 << 7)) {
this._length += (byte & 0x7f) << (7 * this._lengthIndex)
this._lengthIndex += 1
this._readingLength = true
}
else {
this._length += (byte & 0x7f) << (7 * this._lengthIndex)
this._lengthIndex = 0
this._readingLength = false
}
this._buffer = this._buffer.slice(1)
}
if (!this._readingLength && lo + this._length <= hi) {
this.push(chunk.slice(lo, lo + this._length))
lo += this._length
this._length = 0
this._readingLength = true
}
else {
break
if (!this._readingLength) {
if (this._length <= this._buffer.length) {
this.push(this._buffer.slice(0, this._length))
this._buffer = this._buffer.slice(this._length)
this._length = 0
this._readingLength = true
}
else {
// Wait for more chunks
break
}
}
}