andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[#1878] Fix leak of ByteBuf when masked payload is used

Browse Source
This commit is contained in:
Norman Maurer 2013-10-01 07:18:16 +02:00
parent be126395e3
commit f70ceeab58
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
codec-http/src/main/java/io/netty/handler/codec/http/websocketx/WebSocket08FrameDecoder.java
View File

@ -91,7 +91,7 @@ public class WebSocket08FrameDecoder extends ReplayingDecoder<WebSocket08FrameDe
private long framePayloadLength;
private ByteBuf framePayload;
private int framePayloadBytesRead;
private ByteBuf maskingKey;
private byte[] maskingKey;
private ByteBuf payloadBuffer;
private final boolean allowExtensions;
@ -244,7 +244,10 @@ public class WebSocket08FrameDecoder extends ReplayingDecoder<WebSocket08FrameDe
checkpoint(State.MASKING_KEY);
case MASKING_KEY:
if (maskedPayload) {
maskingKey = in.readBytes(4);
if (maskingKey == null) {
maskingKey = new byte[4];
}
in.readBytes(maskingKey);
}
checkpoint(State.PAYLOAD);
case PAYLOAD:
@ -405,7 +408,7 @@ public class WebSocket08FrameDecoder extends ReplayingDecoder<WebSocket08FrameDe
private void unmask(ByteBuf frame) {
for (int i = frame.readerIndex(); i < frame.writerIndex(); i++) {
frame.setByte(i, frame.getByte(i) ^ maskingKey.getByte(i % 4));
frame.setByte(i, frame.getByte(i) ^ maskingKey[i % 4]);
}
}