Commit 5d7adf66 authored by Jon Skeet's avatar Jon Skeet

First unit test reading a complete message\!

parent 3f9a6f21
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
// limitations under the License. // limitations under the License.
using System; using System;
using System.IO; using System.IO;
using Google.ProtocolBuffers.TestProtos;
using NUnit.Framework; using NUnit.Framework;
namespace Google.ProtocolBuffers { namespace Google.ProtocolBuffers {
...@@ -183,23 +184,22 @@ namespace Google.ProtocolBuffers { ...@@ -183,23 +184,22 @@ namespace Google.ProtocolBuffers {
Assert.AreEqual(unchecked((long)0x8000000000000000L), CodedInputStream.DecodeZigZag64(0xFFFFFFFFFFFFFFFFL)); Assert.AreEqual(unchecked((long)0x8000000000000000L), CodedInputStream.DecodeZigZag64(0xFFFFFFFFFFFFFFFFL));
} }
/* TODO(jonskeet): Reinstate this when protoc is ready [Test]
public void testReadWholeMessage() throws Exception { public void ReadWholeMessage() {
TestAllTypes message = TestUtil.getAllSet(); TestAllTypes message = TestUtil.GetAllSet();
byte[] rawBytes = message.toByteArray(); byte[] rawBytes = message.ToByteArray();
assertEquals(rawBytes.length, message.getSerializedSize()); Assert.AreEqual(rawBytes.Length, message.SerializedSize);
TestAllTypes message2 = TestAllTypes.parseFrom(rawBytes); TestAllTypes message2 = TestAllTypes.ParseFrom(rawBytes);
TestUtil.assertAllFieldsSet(message2); TestUtil.AssertAllFieldsSet(message2);
// Try different block sizes. // Try different block sizes.
for (int blockSize = 1; blockSize < 256; blockSize *= 2) { for (int blockSize = 1; blockSize < 256; blockSize *= 2) {
message2 = TestAllTypes.parseFrom( message2 = TestAllTypes.ParseFrom(new SmallBlockInputStream(rawBytes, blockSize));
new SmallBlockInputStream(rawBytes, blockSize)); TestUtil.AssertAllFieldsSet(message2);
TestUtil.assertAllFieldsSet(message2);
} }
}*/ }
/* TODO(jonskeet): Reinstate this when protoc is ready /* TODO(jonskeet): Reinstate this when protoc is ready
public void testSkipWholeMessage() throws Exception { public void testSkipWholeMessage() throws Exception {
......
This diff is collapsed.
...@@ -104,16 +104,20 @@ namespace Google.ProtocolBuffers { ...@@ -104,16 +104,20 @@ namespace Google.ProtocolBuffers {
} }
IBuilder IBuilder.MergeFrom(CodedInputStream input) { IBuilder IBuilder.MergeFrom(CodedInputStream input) {
return ((IBuilder)this).MergeFrom(input, ExtensionRegistry.Empty); return MergeFromImpl(input, ExtensionRegistry.Empty);
} }
IBuilder IBuilder.MergeFrom(CodedInputStream input, ExtensionRegistry extensionRegistry) { protected virtual IBuilder MergeFromImpl(CodedInputStream input, ExtensionRegistry extensionRegistry) {
UnknownFieldSet.Builder unknownFields = UnknownFieldSet.CreateBuilder(UnknownFields); UnknownFieldSet.Builder unknownFields = UnknownFieldSet.CreateBuilder(UnknownFields);
FieldSet.MergeFrom(input, unknownFields, extensionRegistry, this); FieldSet.MergeFrom(input, unknownFields, extensionRegistry, this);
UnknownFields = unknownFields.Build(); UnknownFields = unknownFields.Build();
return this; return this;
} }
IBuilder IBuilder.MergeFrom(CodedInputStream input, ExtensionRegistry extensionRegistry) {
return MergeFromImpl(input, extensionRegistry);
}
IBuilder IBuilder.MergeUnknownFields(UnknownFieldSet unknownFields) { IBuilder IBuilder.MergeUnknownFields(UnknownFieldSet unknownFields) {
UnknownFields = UnknownFieldSet.CreateBuilder(UnknownFields) UnknownFields = UnknownFieldSet.CreateBuilder(UnknownFields)
.MergeFrom(unknownFields) .MergeFrom(unknownFields)
...@@ -121,8 +125,6 @@ namespace Google.ProtocolBuffers { ...@@ -121,8 +125,6 @@ namespace Google.ProtocolBuffers {
return this; return this;
} }
public UnknownFieldSet UnknownFields { get; set; }
IBuilder IBuilder.MergeFrom(ByteString data) { IBuilder IBuilder.MergeFrom(ByteString data) {
CodedInputStream input = data.CreateCodedInput(); CodedInputStream input = data.CreateCodedInput();
((IBuilder)this).MergeFrom(input); ((IBuilder)this).MergeFrom(input);
...@@ -160,9 +162,11 @@ namespace Google.ProtocolBuffers { ...@@ -160,9 +162,11 @@ namespace Google.ProtocolBuffers {
IBuilder IBuilder.MergeFrom(Stream input, ExtensionRegistry extensionRegistry) { IBuilder IBuilder.MergeFrom(Stream input, ExtensionRegistry extensionRegistry) {
CodedInputStream codedInput = CodedInputStream.CreateInstance(input); CodedInputStream codedInput = CodedInputStream.CreateInstance(input);
((IBuilder) this).MergeFrom(codedInput, extensionRegistry); ((IBuilder)this).MergeFrom(codedInput, extensionRegistry);
codedInput.CheckLastTagWas(0); codedInput.CheckLastTagWas(0);
return this; return this;
} }
public abstract UnknownFieldSet UnknownFields { get; set; }
} }
} }
...@@ -23,7 +23,7 @@ namespace Google.ProtocolBuffers { ...@@ -23,7 +23,7 @@ namespace Google.ProtocolBuffers {
/// Immutable array of bytes. /// Immutable array of bytes.
/// TODO(jonskeet): Implement the common collection interfaces? /// TODO(jonskeet): Implement the common collection interfaces?
/// </summary> /// </summary>
public sealed class ByteString : IEnumerable<byte> { public sealed class ByteString : IEnumerable<byte>, IEquatable<ByteString> {
private static readonly ByteString empty = new ByteString(new byte[0]); private static readonly ByteString empty = new ByteString(new byte[0]);
...@@ -126,6 +126,34 @@ namespace Google.ProtocolBuffers { ...@@ -126,6 +126,34 @@ namespace Google.ProtocolBuffers {
// TODO(jonskeet): CopyTo, Equals, GetHashCode if they turn out to be required // TODO(jonskeet): CopyTo, Equals, GetHashCode if they turn out to be required
public override bool Equals(object obj) {
ByteString other = obj as ByteString;
if (obj == null) {
return false;
}
return Equals(other);
}
public override int GetHashCode() {
int ret = 23;
foreach (byte b in bytes) {
ret = (ret << 8) | b;
}
return ret;
}
public bool Equals(ByteString other) {
if (other.bytes.Length != bytes.Length) {
return false;
}
for (int i = 0; i < bytes.Length; i++) {
if (other.bytes[i] != bytes[i]) {
return false;
}
}
return true;
}
/// <summary> /// <summary>
/// Builder for ByteStrings which allows them to be created without extra /// Builder for ByteStrings which allows them to be created without extra
/// copying being involved. This has to be a nested type in order to have access /// copying being involved. This has to be a nested type in order to have access
......
...@@ -569,8 +569,8 @@ namespace Google.ProtocolBuffers { ...@@ -569,8 +569,8 @@ namespace Google.ProtocolBuffers {
totalBytesRetired += bufferSize; totalBytesRetired += bufferSize;
bufferPos = 0; bufferPos = 0;
bufferSize = (input == null) ? -1 : input.Read(buffer, 0, buffer.Length); bufferSize = (input == null) ? 0 : input.Read(buffer, 0, buffer.Length);
if (bufferSize == -1) { if (bufferSize == 0) {
bufferSize = 0; bufferSize = 0;
if (mustSucceed) { if (mustSucceed) {
throw InvalidProtocolBufferException.TruncatedMessage(); throw InvalidProtocolBufferException.TruncatedMessage();
......
...@@ -206,6 +206,7 @@ namespace Google.ProtocolBuffers { ...@@ -206,6 +206,7 @@ namespace Google.ProtocolBuffers {
public void WriteBytes(int fieldNumber, ByteString value) { public void WriteBytes(int fieldNumber, ByteString value) {
// TODO(jonskeet): Optimise this! (No need to copy the bytes twice.) // TODO(jonskeet): Optimise this! (No need to copy the bytes twice.)
WriteTag(fieldNumber, WireFormat.WireType.LengthDelimited);
byte[] bytes = value.ToByteArray(); byte[] bytes = value.ToByteArray();
WriteRawVarint32((uint)bytes.Length); WriteRawVarint32((uint)bytes.Length);
WriteRawBytes(bytes); WriteRawBytes(bytes);
...@@ -223,12 +224,12 @@ namespace Google.ProtocolBuffers { ...@@ -223,12 +224,12 @@ namespace Google.ProtocolBuffers {
public void WriteSFixed32(int fieldNumber, int value) { public void WriteSFixed32(int fieldNumber, int value) {
WriteTag(fieldNumber, WireFormat.WireType.Fixed32); WriteTag(fieldNumber, WireFormat.WireType.Fixed32);
WriteRawVarint32((uint)value); WriteRawLittleEndian32((uint)value);
} }
public void WriteSFixed64(int fieldNumber, long value) { public void WriteSFixed64(int fieldNumber, long value) {
WriteTag(fieldNumber, WireFormat.WireType.Fixed64); WriteTag(fieldNumber, WireFormat.WireType.Fixed64);
WriteRawVarint64((ulong)value); WriteRawLittleEndian64((ulong)value);
} }
public void WriteSInt32(int fieldNumber, int value) { public void WriteSInt32(int fieldNumber, int value) {
......
...@@ -22,7 +22,7 @@ namespace Google.ProtocolBuffers.Descriptors { ...@@ -22,7 +22,7 @@ namespace Google.ProtocolBuffers.Descriptors {
if (proto.HasType) { if (proto.HasType) {
fieldType = GetFieldTypeFromProtoType(proto.Type); fieldType = GetFieldTypeFromProtoType(proto.Type);
//type = proto.Type; mappedType = FieldTypeToMappedTypeMap[fieldType];
} }
if (FieldNumber <= 0) { if (FieldNumber <= 0) {
...@@ -244,8 +244,7 @@ namespace Google.ProtocolBuffers.Descriptors { ...@@ -244,8 +244,7 @@ namespace Google.ProtocolBuffers.Descriptors {
/// Immutable mapping from field type to mapped type. Built using the attributes on /// Immutable mapping from field type to mapped type. Built using the attributes on
/// FieldType values. /// FieldType values.
/// </summary> /// </summary>
public static readonly IDictionary<FieldType, MappedType> FieldTypeToWireFormatMap = MapFieldTypes(); public static readonly IDictionary<FieldType, MappedType> FieldTypeToMappedTypeMap = MapFieldTypes();
private static IDictionary<FieldType, MappedType> MapFieldTypes() { private static IDictionary<FieldType, MappedType> MapFieldTypes() {
var map = new Dictionary<FieldType, MappedType>(); var map = new Dictionary<FieldType, MappedType>();
......
...@@ -19,16 +19,16 @@ namespace Google.ProtocolBuffers.Descriptors { ...@@ -19,16 +19,16 @@ namespace Google.ProtocolBuffers.Descriptors {
containingType = parent; containingType = parent;
nestedTypes = DescriptorUtil.ConvertAndMakeReadOnly(proto.NestedTypeList, nestedTypes = DescriptorUtil.ConvertAndMakeReadOnly(proto.NestedTypeList,
(type, index) => new MessageDescriptor(type, file, null, index)); (type, index) => new MessageDescriptor(type, file, this, index));
enumTypes = DescriptorUtil.ConvertAndMakeReadOnly(proto.EnumTypeList, enumTypes = DescriptorUtil.ConvertAndMakeReadOnly(proto.EnumTypeList,
(type, index) => new EnumDescriptor(type, file, null, index)); (type, index) => new EnumDescriptor(type, file, this, index));
fields = DescriptorUtil.ConvertAndMakeReadOnly(proto.FieldList, fields = DescriptorUtil.ConvertAndMakeReadOnly(proto.FieldList,
(field, index) => new FieldDescriptor(field, file, null, index, false)); (field, index) => new FieldDescriptor(field, file, this, index, false));
extensions = DescriptorUtil.ConvertAndMakeReadOnly(proto.ExtensionList, extensions = DescriptorUtil.ConvertAndMakeReadOnly(proto.ExtensionList,
(field, index) => new FieldDescriptor(field, file, null, index, true)); (field, index) => new FieldDescriptor(field, file, this, index, true));
file.DescriptorPool.AddSymbol(this); file.DescriptorPool.AddSymbol(this);
} }
......
...@@ -21,7 +21,7 @@ namespace Google.ProtocolBuffers.FieldAccess { ...@@ -21,7 +21,7 @@ namespace Google.ProtocolBuffers.FieldAccess {
internal RepeatedMessageAccessor(string name, Type messageType, Type builderType) internal RepeatedMessageAccessor(string name, Type messageType, Type builderType)
: base(name, messageType, builderType) { : base(name, messageType, builderType) {
createBuilderMethod = ClrType.GetMethod("CreateBuilder", BindingFlags.Public | BindingFlags.Static); createBuilderMethod = ClrType.GetMethod("CreateBuilder", new Type[0]);
if (createBuilderMethod == null) { if (createBuilderMethod == null) {
throw new ArgumentException("No public static CreateBuilder method declared in " + ClrType.Name); throw new ArgumentException("No public static CreateBuilder method declared in " + ClrType.Name);
} }
......
...@@ -11,7 +11,6 @@ namespace Google.ProtocolBuffers.FieldAccess { ...@@ -11,7 +11,6 @@ namespace Google.ProtocolBuffers.FieldAccess {
private readonly PropertyInfo messageProperty; private readonly PropertyInfo messageProperty;
private readonly PropertyInfo builderProperty; private readonly PropertyInfo builderProperty;
private readonly PropertyInfo hasProperty;
private readonly PropertyInfo countProperty; private readonly PropertyInfo countProperty;
private readonly MethodInfo clearMethod; private readonly MethodInfo clearMethod;
private readonly MethodInfo addMethod; private readonly MethodInfo addMethod;
...@@ -30,15 +29,13 @@ namespace Google.ProtocolBuffers.FieldAccess { ...@@ -30,15 +29,13 @@ namespace Google.ProtocolBuffers.FieldAccess {
internal RepeatedPrimitiveAccessor(string name, Type messageType, Type builderType) { internal RepeatedPrimitiveAccessor(string name, Type messageType, Type builderType) {
messageProperty = messageType.GetProperty(name + "List"); messageProperty = messageType.GetProperty(name + "List");
builderProperty = builderType.GetProperty(name + "List"); builderProperty = builderType.GetProperty(name + "List");
hasProperty = messageType.GetProperty("Has" + name);
countProperty = messageType.GetProperty(name + "Count"); countProperty = messageType.GetProperty(name + "Count");
clearMethod = builderType.GetMethod("Clear" + name); clearMethod = builderType.GetMethod("Clear" + name);
addMethod = builderType.GetMethod("Add" + name);
getElementMethod = messageType.GetMethod("Get" + name, new Type[] { typeof(int) }); getElementMethod = messageType.GetMethod("Get" + name, new Type[] { typeof(int) });
setElementMethod = builderType.GetMethod("Set" + name, new Type[] { typeof(int) }); addMethod = builderType.GetMethod("Add" + name, new Type[] { ClrType });
setElementMethod = builderType.GetMethod("Set" + name, new Type[] { typeof(int), ClrType });
if (messageProperty == null if (messageProperty == null
|| builderProperty == null || builderProperty == null
|| hasProperty == null
|| countProperty == null || countProperty == null
|| clearMethod == null || clearMethod == null
|| addMethod == null || addMethod == null
......
...@@ -19,7 +19,7 @@ namespace Google.ProtocolBuffers.FieldAccess { ...@@ -19,7 +19,7 @@ namespace Google.ProtocolBuffers.FieldAccess {
internal SingleMessageAccessor(string name, Type messageType, Type builderType) internal SingleMessageAccessor(string name, Type messageType, Type builderType)
: base(name, messageType, builderType) { : base(name, messageType, builderType) {
createBuilderMethod = ClrType.GetMethod("CreateBuilder", BindingFlags.Public | BindingFlags.Static); createBuilderMethod = ClrType.GetMethod("CreateBuilder", new Type[0]);//BindingFlags.Public | BindingFlags.Static | BindingFlags.DeclaredOnly);
if (createBuilderMethod == null) { if (createBuilderMethod == null) {
throw new ArgumentException("No public static CreateBuilder method declared in " + ClrType.Name); throw new ArgumentException("No public static CreateBuilder method declared in " + ClrType.Name);
} }
......
...@@ -91,11 +91,8 @@ namespace Google.ProtocolBuffers { ...@@ -91,11 +91,8 @@ namespace Google.ProtocolBuffers {
} }
// TODO(jonskeet): Should this be in UnknownFieldSet.Builder really? Or CodedInputStream? // TODO(jonskeet): Should this be in UnknownFieldSet.Builder really? Or CodedInputStream?
internal static void MergeFrom(CodedInputStream input, internal static void MergeFrom(CodedInputStream input, UnknownFieldSet.Builder unknownFields,
UnknownFieldSet.Builder unknownFields, ExtensionRegistry extensionRegistry, IBuilder builder) {
ExtensionRegistry extensionRegistry,
IBuilder builder) {
while (true) { while (true) {
uint tag = input.ReadTag(); uint tag = input.ReadTag();
if (tag == 0) { if (tag == 0) {
...@@ -120,15 +117,10 @@ namespace Google.ProtocolBuffers { ...@@ -120,15 +117,10 @@ namespace Google.ProtocolBuffers {
/// <param name="builder">Builder to merge field into, if it's a known field</param> /// <param name="builder">Builder to merge field into, if it's a known field</param>
/// <param name="tag">The tag, which should already have been read from the input</param> /// <param name="tag">The tag, which should already have been read from the input</param>
/// <returns>true unless the tag is an end-group tag</returns> /// <returns>true unless the tag is an end-group tag</returns>
internal static bool MergeFieldFrom(CodedInputStream input, internal static bool MergeFieldFrom(CodedInputStream input, UnknownFieldSet.Builder unknownFields,
UnknownFieldSet.Builder unknownFields, ExtensionRegistry extensionRegistry, IBuilder builder, uint tag) {
ExtensionRegistry extensionRegistry,
IBuilder builder,
uint tag) {
MessageDescriptor type = builder.DescriptorForType; MessageDescriptor type = builder.DescriptorForType;
if (type.Options.MessageSetWireFormat && tag == WireFormat.MessageSetTag.ItemStart) {
if (type.Options.MessageSetWireFormat
&& tag == WireFormat.MessageSetTag.ItemStart) {
MergeMessageSetExtensionFromCodedStream(input, unknownFields, extensionRegistry, builder); MergeMessageSetExtensionFromCodedStream(input, unknownFields, extensionRegistry, builder);
return true; return true;
} }
......
...@@ -199,6 +199,10 @@ namespace Google.ProtocolBuffers { ...@@ -199,6 +199,10 @@ namespace Google.ProtocolBuffers {
((IBuilder)this).MergeFrom(input, extensionRegistry); ((IBuilder)this).MergeFrom(input, extensionRegistry);
return this; return this;
} }
protected override IBuilder MergeFromImpl(CodedInputStream data, ExtensionRegistry extensionRegistry) {
return MergeFrom(data, extensionRegistry);
}
/// <summary> /// <summary>
/// Like Build(), but will wrap UninitializedMessageException in /// Like Build(), but will wrap UninitializedMessageException in
...@@ -222,6 +226,11 @@ namespace Google.ProtocolBuffers { ...@@ -222,6 +226,11 @@ namespace Google.ProtocolBuffers {
} }
return BuildPartial(); return BuildPartial();
} }
public override UnknownFieldSet UnknownFields {
get { return MessageBeingBuilt.UnknownFields; }
set { MessageBeingBuilt.SetUnknownFields(value); }
}
public abstract TMessage BuildPartial(); public abstract TMessage BuildPartial();
public abstract IBuilder<TMessage> Clone(); public abstract IBuilder<TMessage> Clone();
......
...@@ -187,15 +187,15 @@ namespace Google.ProtocolBuffers { ...@@ -187,15 +187,15 @@ namespace Google.ProtocolBuffers {
} }
internal static ulong ParseUInt64(string text) { internal static ulong ParseUInt64(string text) {
return (ulong) ParseInteger(text, true, false); return (ulong) ParseInteger(text, false, true);
} }
internal static long ParseInt64(string text) { internal static long ParseInt64(string text) {
return ParseInteger(text, true, false); return ParseInteger(text, true, true);
} }
internal static uint ParseUInt32(string text) { internal static uint ParseUInt32(string text) {
return (uint) ParseInteger(text, true, false); return (uint) ParseInteger(text, false, false);
} }
internal static int ParseInt32(string text) { internal static int ParseInt32(string text) {
...@@ -229,7 +229,7 @@ namespace Google.ProtocolBuffers { ...@@ -229,7 +229,7 @@ namespace Google.ProtocolBuffers {
ulong result = Convert.ToUInt64(text, radix); ulong result = Convert.ToUInt64(text, radix);
if (negative) { if (negative) {
ulong max = isLong ? 0x8000000UL : 0x8000L; ulong max = isLong ? 0x8000000000000000UL : 0x80000000L;
if (result > max) { if (result > max) {
throw new FormatException("Number of out range: " + original); throw new FormatException("Number of out range: " + original);
} }
......
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