Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
74 changes: 47 additions & 27 deletions src/responses/parser.js
Original file line number Diff line number Diff line change
Expand Up @@ -23,45 +23,65 @@ class Parser extends EventEmitter {
super();
this.buffer = null;
}


parseLoop(buffer) {
while(true) {
const ber = new BerReader(buffer);

try {
ber.readSequence();
} catch (e) {
this.emit('error', e);
return buffer;
}

if (ber.remain < ber.length || ber.length === 0) {
return buffer;
}

let nextMessages = null;
if (ber.remain > ber.length) {
nextMessages = buffer.slice(ber.offset + ber.length);
ber._size = ber.offset + ber.length;
assert.equal(ber.remain, ber.length);
}

try {
const message = getMessage(ber);
message.parse(ber);
this.emit('message', message);
} catch (e) {
this.emit('error', e);
}

if (nextMessages === null) {
return null;
} else {
buffer = nextMessages;
}
}
}

parse(data) {
assert.buffer(data, 'data');

this.buffer = this.buffer ? Buffer.concat([this.buffer, data]) : data;

const ber = new BerReader(this.buffer);

try {
ber.readSequence();
} catch (e) {
this.emit('error', e);
return;
}

if (ber.remain < ber.length) {

// If the chunk is not full, we continue accumulating the buffer.
if (ber.remain < ber.length || ber.length === 0) {
return;
}

let nextMessage = null;
if (ber.remain > ber.length) {
nextMessage = this.buffer.slice(ber.offset + ber.length);
ber._size = ber.offset + ber.length;
assert.equal(ber.remain, ber.length);
}

this.buffer = null;

try {
const message = getMessage(ber);
message.parse(ber);
this.emit('message', message);
} catch (e) {
this.emit('error', e);
}

if (nextMessage) {
this.parse(nextMessage);
}

this.buffer = this.parseLoop(this.buffer);
}
}

Expand Down