diff options
| author | Robert Godfrey <rgodfrey@apache.org> | 2014-10-27 13:51:59 +0000 |
|---|---|---|
| committer | Robert Godfrey <rgodfrey@apache.org> | 2014-10-27 13:51:59 +0000 |
| commit | a1d58191d8b393f4eb7a2a1691572ef1b5aa9e8f (patch) | |
| tree | 452c0ef6fba936b9910ed54c034e4ecfccfc1a01 /qpid/java/common/src/test | |
| parent | 5db8a7db8abcb70c75647642cd9d19331fb6839c (diff) | |
| download | qpid-python-a1d58191d8b393f4eb7a2a1691572ef1b5aa9e8f.tar.gz | |
QPID-6189 : [Java Common] fix stack overflow bug when frame is broken into thousands of reads/writes
git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk@1634539 13f79535-47bb-0310-9956-ffa450edef68
Diffstat (limited to 'qpid/java/common/src/test')
| -rw-r--r-- | qpid/java/common/src/test/java/org/apache/qpid/codec/AMQDecoderTest.java | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/qpid/java/common/src/test/java/org/apache/qpid/codec/AMQDecoderTest.java b/qpid/java/common/src/test/java/org/apache/qpid/codec/AMQDecoderTest.java index 51f3ce1113..e597192766 100644 --- a/qpid/java/common/src/test/java/org/apache/qpid/codec/AMQDecoderTest.java +++ b/qpid/java/common/src/test/java/org/apache/qpid/codec/AMQDecoderTest.java @@ -25,17 +25,22 @@ import java.io.ByteArrayOutputStream; import java.io.DataOutputStream; import java.io.IOException; import java.nio.ByteBuffer; +import java.util.Arrays; import java.util.List; +import java.util.Random; import junit.framework.TestCase; +import org.apache.qpid.framing.AMQBody; import org.apache.qpid.framing.AMQDataBlock; import org.apache.qpid.framing.AMQFrame; import org.apache.qpid.framing.AMQFrameDecodingException; import org.apache.qpid.framing.AMQProtocolVersionException; +import org.apache.qpid.framing.ContentBody; import org.apache.qpid.framing.FrameCreatingMethodProcessor; import org.apache.qpid.framing.HeartbeatBody; import org.apache.qpid.framing.ProtocolVersion; +import org.apache.qpid.util.BytesDataOutput; public class AMQDecoderTest extends TestCase { @@ -72,6 +77,35 @@ public class AMQDecoderTest extends TestCase fail("decode was not a frame"); } } + + + public void testDecodeWithManyBuffers() throws AMQProtocolVersionException, AMQFrameDecodingException, IOException + { + Random random = new Random(); + final byte[] payload = new byte[2048]; + random.nextBytes(payload); + final AMQBody body = new ContentBody(payload); + AMQFrame frame = new AMQFrame(1, body); + byte[] outputBuf = new byte[4096]; + BytesDataOutput dataOutput = new BytesDataOutput(outputBuf); + frame.writePayload(dataOutput); + for(int i = 0 ; i < dataOutput.length(); i++) + { + _decoder.decodeBuffer(ByteBuffer.wrap(outputBuf, i, 1)); + + } + List<AMQDataBlock> frames = _methodProcessor.getProcessedMethods(); + if (frames.get(0) instanceof AMQFrame) + { + assertEquals(ContentBody.TYPE, ((AMQFrame) frames.get(0)).getBodyFrame().getFrameType()); + ContentBody decodedBody = (ContentBody) ((AMQFrame) frames.get(0)).getBodyFrame(); + assertTrue("Body was corrupted", Arrays.equals(payload, decodedBody.getPayload())); + } + else + { + fail("decode was not a frame"); + } + } public void testPartialFrameDecode() throws AMQProtocolVersionException, AMQFrameDecodingException, IOException { |
