added test case and patch for marshalling of long values which result in negative int / short values being unmarshalled

git-svn-id: https://svn.apache.org/repos/asf/incubator/activemq/trunk@382395 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
James Strachan 2006-03-02 14:33:45 +00:00
parent c717221cf8
commit f1e4459644
2 changed files with 150 additions and 2 deletions

View File

@ -90,17 +90,29 @@ abstract public class BaseDataStreamMarshaller implements DataStreamMarshaller {
if( bs.readBoolean() ) { if( bs.readBoolean() ) {
return dataIn.readLong(); return dataIn.readLong();
} else { } else {
return dataIn.readInt(); return toLong(dataIn.readInt());
} }
} else { } else {
if( bs.readBoolean() ) { if( bs.readBoolean() ) {
return dataIn.readShort(); return toLong(dataIn.readShort());
} else { } else {
return 0; return 0;
} }
} }
} }
protected long toLong(short value) {
// lets handle negative values
long answer = value;
return answer & 0xffffL;
}
protected long toLong(int value) {
// lets handle negative values
long answer = value;
return answer & 0xffffffffL;
}
protected DataStructure tightUnmarsalNestedObject(OpenWireFormat wireFormat, DataInputStream dataIn, BooleanStream bs) throws IOException { protected DataStructure tightUnmarsalNestedObject(OpenWireFormat wireFormat, DataInputStream dataIn, BooleanStream bs) throws IOException {
return wireFormat.tightUnmarshalNestedObject(dataIn, bs); return wireFormat.tightUnmarshalNestedObject(dataIn, bs);
} }

View File

@ -0,0 +1,136 @@
/**
*
* Copyright 2005-2006 The Apache Software Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.activemq.openwire;
import org.apache.activemq.command.SessionId;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import junit.framework.TestCase;
/**
*
* @version $Revision$
*/
public class NumberRangesWhileMarshallingTest extends TestCase {
protected String connectionId = "Cheese";
protected ByteArrayOutputStream buffer = new ByteArrayOutputStream();
protected DataOutputStream ds = new DataOutputStream(buffer);
protected OpenWireFormat openWireformat;
protected int endOfStreamMarker = 0x12345678;
public void testLongNumberRanges() throws Exception {
long[] numberValues = {
// bytes
0, 1, 0x7e, 0x7f, 0x80, 0x81, 0xf0, 0xff,
// shorts
0x7eff, 0x7fffL, 0x8001L, 0x8000L, 0xe000L, 0xe0001L, 0xff00L, 0xffffL,
// ints
0x10000L,
0x700000L,
0x12345678L,
0x72345678L,
0x7fffffffL,
0x80000000L,
0x80000001L,
0xE0000001L,
0xFFFFFFFFL,
// 3 byte longs
0x123456781L,
0x1234567812L,
0x12345678123L,
0x123456781234L,
0x1234567812345L,
0x12345678123456L,
0x7e345678123456L,
0x7fffffffffffffL,
0x80000000000000L,
0x80000000000001L,
0xe0000000000001L,
0xffffffffffffffL,
// 4 byte longs
0x1234567812345678L,
0x7fffffffffffffffL,
0x8000000000000000L,
0x8000000000000001L,
0xe000000000000001L,
0xffffffffffffffffL,
1 };
for (int i = 0; i < numberValues.length; i++) {
long value = numberValues[i];
SessionId object = new SessionId();
object.setConnectionId(connectionId);
object.setValue(value);
writeObject(object);
}
ds.writeInt(endOfStreamMarker);
// now lets read from the stream
ds.close();
ByteArrayInputStream in = new ByteArrayInputStream(buffer.toByteArray());
DataInputStream dis = new DataInputStream(in);
for (int i = 0; i < numberValues.length; i++) {
long value = numberValues[i];
String expected = Long.toHexString(value);
System.out.println("Unmarshaling value: " + i + " = " + expected);
SessionId command = (SessionId) openWireformat.unmarshal(dis);
assertEquals("connection ID in object: "+ i, connectionId, command.getConnectionId());
String actual = Long.toHexString(command.getValue());
assertEquals("value of object: "+ i + " was: " + actual, expected, actual);
}
int marker = dis.readInt();
assertEquals("Marker int", Integer.toHexString(endOfStreamMarker), Integer.toHexString(marker));
// lets try read and we should get an exception
try {
byte value = dis.readByte();
fail("Should have reached the end of the stream");
}
catch (IOException e) {
// worked!
}
}
protected void setUp() throws Exception {
super.setUp();
openWireformat = createOpenWireFormat();
}
protected OpenWireFormat createOpenWireFormat() {
OpenWireFormat wf = new OpenWireFormat();
wf.setCacheEnabled(true);
wf.setStackTraceEnabled(false);
wf.setVersion(1);
return wf;
}
private void writeObject(Object object) throws IOException {
openWireformat.marshal(object, ds);
}
}