StompDecoder handles partial headers correctly

Issue: SPR-13416
This commit is contained in:
Rossen Stoyanchev
2015-09-23 16:50:46 -04:00
parent 54d3d782a9
commit daa49c9bcd
2 changed files with 31 additions and 7 deletions

View File

@@ -208,10 +208,15 @@ public class StompDecoder {
private void readHeaders(ByteBuffer buffer, StompHeaderAccessor headerAccessor) {
while (true) {
ByteArrayOutputStream headerStream = new ByteArrayOutputStream(256);
while (buffer.remaining() > 0 && !tryConsumeEndOfLine(buffer)) {
boolean headerComplete = false;
while (buffer.hasRemaining()) {
if (tryConsumeEndOfLine(buffer)) {
headerComplete = true;
break;
}
headerStream.write(buffer.get());
}
if (headerStream.size() > 0) {
if (headerStream.size() > 0 && headerComplete) {
String header = new String(headerStream.toByteArray(), UTF8_CHARSET);
int colonIndex = header.indexOf(':');
if (colonIndex <= 0) {