| // |
| // ======================================================================== |
| // Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd. |
| // ------------------------------------------------------------------------ |
| // All rights reserved. This program and the accompanying materials |
| // are made available under the terms of the Eclipse Public License v1.0 |
| // and Apache License v2.0 which accompanies this distribution. |
| // |
| // The Eclipse Public License is available at |
| // http://www.eclipse.org/legal/epl-v10.html |
| // |
| // The Apache License v2.0 is available at |
| // http://www.opensource.org/licenses/apache2.0.php |
| // |
| // You may elect to redistribute this code under either of these licenses. |
| // ======================================================================== |
| // |
| |
| package org.eclipse.jetty.websocket.common.ab; |
| |
| import static org.hamcrest.Matchers.*; |
| |
| import java.nio.ByteBuffer; |
| |
| import org.eclipse.jetty.util.BufferUtil; |
| import org.eclipse.jetty.websocket.api.WebSocketBehavior; |
| import org.eclipse.jetty.websocket.api.WebSocketPolicy; |
| import org.eclipse.jetty.websocket.api.extensions.Frame; |
| import org.eclipse.jetty.websocket.common.OpCode; |
| import org.eclipse.jetty.websocket.common.Parser; |
| import org.eclipse.jetty.websocket.common.WebSocketFrame; |
| import org.eclipse.jetty.websocket.common.frames.BinaryFrame; |
| import org.eclipse.jetty.websocket.common.test.ByteBufferAssert; |
| import org.eclipse.jetty.websocket.common.test.IncomingFramesCapture; |
| import org.eclipse.jetty.websocket.common.test.UnitGenerator; |
| import org.eclipse.jetty.websocket.common.test.UnitParser; |
| import org.junit.Assert; |
| import org.junit.Test; |
| |
| /** |
| * Binary Message Spec testing the {@link Generator} and {@link Parser} |
| */ |
| public class TestABCase1_2 |
| { |
| private WebSocketPolicy policy = WebSocketPolicy.newClientPolicy(); |
| |
| @Test |
| public void testGenerate125ByteBinaryCase1_2_2() |
| { |
| int length = 125; |
| |
| ByteBuffer bb = ByteBuffer.allocate(length); |
| |
| for ( int i = 0 ; i < length ; ++i) |
| { |
| bb.put("*".getBytes()); |
| } |
| |
| bb.flip(); |
| |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(bb); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| |
| byte b = 0x00; // no masking |
| b |= length & 0x7F; |
| expected.put(b); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| } |
| |
| @Test |
| public void testGenerate126ByteBinaryCase1_2_3() |
| { |
| int length = 126; |
| |
| ByteBuffer bb = ByteBuffer.allocate(length); |
| |
| for ( int i = 0 ; i < length ; ++i) |
| { |
| bb.put("*".getBytes()); |
| } |
| |
| bb.flip(); |
| |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(bb); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| |
| byte b = 0x00; // no masking |
| b |= length & 0x7E; |
| expected.put(b); |
| |
| //expected.put((byte)((length>>8) & 0xFF)); |
| //expected.put((byte)(length & 0xFF)); |
| expected.putShort((short)length); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| } |
| |
| @Test |
| public void testGenerate127ByteBinaryCase1_2_4() |
| { |
| int length = 127; |
| |
| ByteBuffer bb = ByteBuffer.allocate(length); |
| |
| for ( int i = 0 ; i < length ; ++i) |
| { |
| bb.put("*".getBytes()); |
| |
| } |
| |
| bb.flip(); |
| |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(bb); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| |
| byte b = 0x00; // no masking |
| b |= length & 0x7E; |
| expected.put(b); |
| |
| //expected.put((byte)((length>>8) & 0xFF)); |
| //expected.put((byte)(length & 0xFF)); |
| expected.putShort((short)length); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| } |
| |
| @Test |
| public void testGenerate128ByteBinaryCase1_2_5() |
| { |
| int length = 128; |
| |
| ByteBuffer bb = ByteBuffer.allocate(length); |
| |
| for ( int i = 0 ; i < length ; ++i) |
| { |
| bb.put("*".getBytes()); |
| |
| } |
| |
| bb.flip(); |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(bb); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| |
| byte b = 0x00; // no masking |
| b |= 0x7E; |
| expected.put(b); |
| |
| expected.put((byte)(length>>8)); |
| expected.put((byte)(length & 0xFF)); |
| //expected.putShort((short)length); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| |
| } |
| |
| @Test |
| public void testGenerate65535ByteBinaryCase1_2_6() |
| { |
| int length = 65535; |
| |
| ByteBuffer bb = ByteBuffer.allocate(length); |
| |
| for ( int i = 0 ; i < length ; ++i) |
| { |
| bb.put("*".getBytes()); |
| |
| } |
| |
| bb.flip(); |
| |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(bb); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| |
| byte b = 0x00; // no masking |
| b |= 0x7E; |
| expected.put(b); |
| expected.put(new byte[]{ (byte)0xff, (byte)0xff}); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| } |
| |
| @Test |
| public void testGenerate65536ByteBinaryCase1_2_7() |
| { |
| int length = 65536; |
| |
| ByteBuffer bb = ByteBuffer.allocate(length); |
| |
| for ( int i = 0 ; i < length ; ++i) |
| { |
| bb.put("*".getBytes()); |
| |
| } |
| |
| bb.flip(); |
| |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(bb); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 11); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| |
| byte b = 0x00; // no masking |
| b |= 0x7F; |
| expected.put(b); |
| expected.put(new byte[]{0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00}); |
| |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| } |
| |
| @Test |
| public void testGenerateEmptyBinaryCase1_2_1() |
| { |
| WebSocketFrame binaryFrame = new BinaryFrame().setPayload(new byte[] {}); |
| |
| ByteBuffer actual = UnitGenerator.generate(binaryFrame); |
| |
| ByteBuffer expected = ByteBuffer.allocate(5); |
| |
| expected.put(new byte[] |
| { (byte)0x82, (byte)0x00 }); |
| |
| BufferUtil.flipToFlush(expected,0); |
| |
| ByteBufferAssert.assertEquals("buffers do not match",expected,actual); |
| } |
| |
| @Test |
| public void testParse125ByteBinaryCase1_2_2() |
| { |
| int length = 125; |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| byte b = 0x00; // no masking |
| b |= length & 0x7F; |
| expected.put(b); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| expected.flip(); |
| |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(length)); |
| // Assert.assertEquals("BinaryFrame.payload",length,pActual.getPayloadData().length); |
| } |
| |
| @Test |
| public void testParse126ByteBinaryCase1_2_3() |
| { |
| int length = 126; |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| byte b = 0x00; // no masking |
| b |= length & 0x7E; |
| expected.put(b); |
| expected.putShort((short)length); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| expected.flip(); |
| |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(length)); |
| // Assert.assertEquals("BinaryFrame.payload",length,pActual.getPayloadData().length); |
| } |
| |
| @Test |
| public void testParse127ByteBinaryCase1_2_4() |
| { |
| int length = 127; |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| byte b = 0x00; // no masking |
| b |= length & 0x7E; |
| expected.put(b); |
| expected.putShort((short)length); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| expected.flip(); |
| |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(length)); |
| // .assertEquals("BinaryFrame.payload",length,pActual.getPayloadData().length); |
| } |
| |
| @Test |
| public void testParse128ByteBinaryCase1_2_5() |
| { |
| int length = 128; |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| byte b = 0x00; // no masking |
| b |= 0x7E; |
| expected.put(b); |
| expected.putShort((short)length); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| expected.flip(); |
| |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(length)); |
| // Assert.assertEquals("BinaryFrame.payload",length,pActual.getPayloadData().length); |
| } |
| |
| @Test |
| public void testParse65535ByteBinaryCase1_2_6() |
| { |
| int length = 65535; |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 5); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| byte b = 0x00; // no masking |
| b |= 0x7E; |
| expected.put(b); |
| expected.put(new byte[]{ (byte)0xff, (byte)0xff}); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| expected.flip(); |
| WebSocketPolicy policy = new WebSocketPolicy(WebSocketBehavior.CLIENT); |
| policy.setMaxBinaryMessageSize(length); |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(length)); |
| // Assert.assertEquals("BinaryFrame.payload",length,pActual.getPayloadData().length); |
| } |
| |
| |
| @Test |
| public void testParse65536ByteBinaryCase1_2_7() |
| { |
| int length = 65536; |
| |
| ByteBuffer expected = ByteBuffer.allocate(length + 11); |
| |
| expected.put(new byte[] |
| { (byte)0x82 }); |
| byte b = 0x00; // no masking |
| b |= 0x7F; |
| expected.put(b); |
| expected.put(new byte[]{0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00}); |
| |
| for ( int i = 0 ; i < length ; ++i ) |
| { |
| expected.put("*".getBytes()); |
| } |
| |
| expected.flip(); |
| |
| WebSocketPolicy policy = new WebSocketPolicy(WebSocketBehavior.CLIENT); |
| policy.setMaxBinaryMessageSize(length); |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(length)); |
| // Assert.assertEquals("BinaryFrame.payload",length,pActual.getPayloadData().length); |
| } |
| |
| @Test |
| public void testParseEmptyBinaryCase1_2_1() |
| { |
| |
| ByteBuffer expected = ByteBuffer.allocate(5); |
| |
| expected.put(new byte[] |
| { (byte)0x82, (byte)0x00 }); |
| |
| expected.flip(); |
| |
| Parser parser = new UnitParser(policy); |
| IncomingFramesCapture capture = new IncomingFramesCapture(); |
| parser.setIncomingFramesHandler(capture); |
| parser.parse(expected); |
| |
| capture.assertNoErrors(); |
| capture.assertHasFrame(OpCode.BINARY,1); |
| |
| Frame pActual = capture.getFrames().poll(); |
| Assert.assertThat("BinaryFrame.payloadLength",pActual.getPayloadLength(),is(0)); |
| // Assert.assertNull("BinaryFrame.payload",pActual.getPayloadData()); |
| } |
| } |