Commit eab9b5d4 authored by jasonh@google.com's avatar jasonh@google.com

Fix issue 256: compute the correct size for negative enum values, which need to be sign extended.

parent 17d57dbb
...@@ -751,7 +751,7 @@ public final class CodedOutputStream { ...@@ -751,7 +751,7 @@ public final class CodedOutputStream {
* Caller is responsible for converting the enum value to its numeric value. * Caller is responsible for converting the enum value to its numeric value.
*/ */
public static int computeEnumSizeNoTag(final int value) { public static int computeEnumSizeNoTag(final int value) {
return computeRawVarint32Size(value); return computeInt32SizeNoTag(value);
} }
/** /**
......
...@@ -300,4 +300,19 @@ public class CodedOutputStreamTest extends TestCase { ...@@ -300,4 +300,19 @@ public class CodedOutputStreamTest extends TestCase {
assertEqualBytes(TestUtil.getGoldenPackedFieldsMessage().toByteArray(), assertEqualBytes(TestUtil.getGoldenPackedFieldsMessage().toByteArray(),
rawBytes); rawBytes);
} }
/** Test writing a message containing a negative enum value. This used to
* fail because the size was not properly computed as a sign-extended varint. */
public void testWriteMessageWithNegativeEnumValue() throws Exception {
protobuf_unittest.UnittestProto.SparseEnumMessage message =
protobuf_unittest.UnittestProto.SparseEnumMessage.newBuilder()
.setSparseEnum(protobuf_unittest.UnittestProto.TestSparseEnum.SPARSE_E)
.build();
assertTrue(message.getSparseEnum().getNumber() < 0);
byte[] rawBytes = message.toByteArray();
protobuf_unittest.UnittestProto.SparseEnumMessage message2 =
protobuf_unittest.UnittestProto.SparseEnumMessage.parseFrom(rawBytes);
assertEquals(protobuf_unittest.UnittestProto.TestSparseEnum.SPARSE_E,
message2.getSparseEnum());
}
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment