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
}
}
}

8
test/wire/messagestream.js
View file

@ -21,11 +21,15 @@ describe('MessageStream', function() {
var ds = new ms.DelimitedStream()
var spy = sinon.spy()
ds.on('data', spy)
ds.write(new Buffer([1]))
ds.write(new Buffer([3]))
expect(spy).to.not.have.been.called
ds.write(new Buffer([0x66]))
expect(spy).to.not.have.been.called
ds.write(new Buffer([0x65]))
expect(spy).to.not.have.been.called
ds.write(new Buffer([0x64]))
expect(spy).to.have.been.calledOnce
expect(spy.firstCall.args).to.eql([new Buffer([0x66])])
expect(spy.firstCall.args).to.eql([new Buffer([0x66, 0x65, 0x64])])
})
it('should read varint32 properly', function() {