Group library changes

csharp/src/Google.Protobuf/CodedInputStream.cs

@@ -108,6 +108,7 @@ public sealed class CodedInputStream : IDisposable
         /// The absolute position of the end of the current message.
         /// </summary> 
         private int currentLimit = int.MaxValue;
+        private uint tagLimit = uint.MaxValue;
 
         private int recursionDepth = 0;
 
@@ -373,7 +374,7 @@ public uint ReadTag()
                 if (IsAtEnd)
                 {
                     lastTag = 0;
-                    return 0; // This is the only case in which we return 0.
+                    return 0;
                 }
 
                 lastTag = ReadRawVarint32();
@@ -383,6 +384,10 @@ public uint ReadTag()
                 // If we actually read a tag with a field of 0, that's not a valid tag.
                 throw InvalidProtocolBufferException.InvalidTag();
             }
+            if (ReachedLimit)
+            {
+                return 0;
+            }
             return lastTag;
         }
 
@@ -591,6 +596,26 @@ public void ReadMessage(IMessage builder)
             PopLimit(oldLimit);
         }
 
+        /// <summary>
+        /// Reads an embedded group field from the stream.
+        /// </summary>
+        public void ReadGroup(IMessage builder)
+        {
+            if (recursionDepth >= recursionLimit)
+            {
+                throw InvalidProtocolBufferException.RecursionLimitExceeded();
+            }
+            uint oldTag = PushTag();
+            ++recursionDepth;
+            builder.MergeFrom(this);
+            if (!ReachedTagLimit)
+            {
+                throw InvalidProtocolBufferException.TruncatedMessage();
+            }
+            --recursionDepth;
+            PopTag(oldTag);
+        }
+
         /// <summary>
         /// Reads a bytes field value from the stream.
         /// </summary>   
@@ -953,6 +978,19 @@ internal int PushLimit(int byteLimit)
             return oldLimit;
         }
 
+        /// <summary>
+        /// Sets tagLimit to the limit calculated by the last tag.
+        /// This is called when descending into a group message. The previous limit is returns.
+        /// </summary>
+        /// <returns>The old limit</returns>
+        internal uint PushTag()
+        {
+            uint oldLimit = tagLimit;
+            uint newLimit = WireFormat.MakeTag(WireFormat.GetTagFieldNumber(LastTag), WireFormat.WireType.EndGroup);
+            tagLimit = newLimit;
+            return oldLimit;
+        }
+
         private void RecomputeBufferSizeAfterLimit()
         {
             bufferSize += bufferSizeAfterLimit;
@@ -978,6 +1016,14 @@ internal void PopLimit(int oldLimit)
             RecomputeBufferSizeAfterLimit();
         }
 
+        /// <summary>
+        /// Discards the current limit, returning the previous limit
+        /// </summary>
+        internal void PopTag(uint oldTag)
+        {
+            tagLimit = oldTag;
+        }
+
         /// <summary>
         /// Returns whether or not all the data before the limit has been read.
         /// </summary>
@@ -995,6 +1041,14 @@ internal bool ReachedLimit
             }
         }
 
+        internal bool ReachedTagLimit
+        {
+            get
+            {
+                return tagLimit == lastTag;
+            }
+        }
+
         /// <summary>
         /// Returns true if the stream has reached the end of the input. This is the
         /// case if either the end of the underlying input source has been reached or

csharp/src/Google.Protobuf/CodedOutputStream.cs

@@ -303,6 +303,15 @@ public void WriteMessage(IMessage value)
             value.WriteTo(this);
         }
 
+        /// <summary>
+        /// Writes a group, without a tag, to the stream.
+        /// </summary>
+        /// <param name="value">The value to write</param>
+        public void WriteGroup(IMessage value)
+        {
+            value.WriteTo(this);
+        }
+
         /// <summary>
         /// Write a byte string, without a tag, to the stream.
         /// The data is length-prefixed.

csharp/src/Google.Protobuf/FieldCodec.cs

@@ -225,6 +225,19 @@ public static FieldCodec<T> ForEnum<T>(uint tag, Func<T, int> toInt32, Func<int,
                 (output, value) => output.WriteMessage(value), message => CodedOutputStream.ComputeMessageSize(message), tag);
         }
 
+        /// <summary>
+        /// Retrieves a codec suitable for a group field with the given tag.
+        /// </summary>
+        /// <param name="startTag">The start group tag.</param>
+        /// <param name="endTag">The end group tag.</param>
+        /// <param name="parser">A parser to use for the group message type.</param>
+        /// <returns>A codec for given tag</returns>
+        public static FieldCodec<T> ForGroup<T>(uint startTag, uint endTag, MessageParser<T> parser) where T : IMessage<T>
+        {
+            return new FieldCodec<T>(input => { T message = parser.CreateTemplate(); input.ReadGroup(message); return message; },
+                (output, value) => output.WriteGroup(value), message => CodedOutputStream.ComputeGroupSize(message), startTag, endTag);
+        }
+
         /// <summary>
         /// Creates a codec for a wrapper type of a class - which must be string or ByteString.
         /// </summary>
@@ -235,7 +248,7 @@ public static FieldCodec<T> ForEnum<T>(uint tag, Func<T, int> toInt32, Func<int,
                 input => WrapperCodecs.Read<T>(input, nestedCodec),
                 (output, value) => WrapperCodecs.Write<T>(output, value, nestedCodec),
                 value => WrapperCodecs.CalculateSize<T>(value, nestedCodec),
-                tag,
+                tag, 0,
                 null); // Default value for the wrapper
         }
 
@@ -250,7 +263,7 @@ public static FieldCodec<T> ForEnum<T>(uint tag, Func<T, int> toInt32, Func<int,
                 input => WrapperCodecs.Read<T>(input, nestedCodec),
                 (output, value) => WrapperCodecs.Write<T>(output, value.Value, nestedCodec),
                 value => value == null ? 0 : WrapperCodecs.CalculateSize<T>(value.Value, nestedCodec),
-                tag,
+                tag, 0,
                 null); // Default value for the wrapper
         }
 
@@ -399,6 +412,14 @@ static FieldCodec()
         /// </value>
         internal uint Tag { get; }
 
+        /// <summary>
+        /// Gets the end tag of the codec or 0 if there is no end tag
+        /// </summary>
+        /// <value>
+        /// The end tag of the codec.
+        /// </value>
+        internal uint EndTag { get; }
+
         /// <summary>
         /// Default value for this codec. Usually the same for every instance of the same type, but
         /// for string/ByteString wrapper fields the codec's default value is null, whereas for
@@ -415,7 +436,7 @@ static FieldCodec()
                 Func<CodedInputStream, T> reader,
                 Action<CodedOutputStream, T> writer,
                 int fixedSize,
-                uint tag) : this(reader, writer, _ => fixedSize, tag)
+                uint tag) : this(reader, writer, _ => fixedSize, 0, tag)
         {
             FixedSize = fixedSize;
         }
@@ -424,7 +445,8 @@ static FieldCodec()
             Func<CodedInputStream, T> reader,
             Action<CodedOutputStream, T> writer,
             Func<T, int> sizeCalculator,
-            uint tag) : this(reader, writer, sizeCalculator, tag, DefaultDefault)
+            uint tag,
+            uint endTag = 0) : this(reader, writer, sizeCalculator, tag, endTag, DefaultDefault)
         {
         }
 
@@ -433,6 +455,7 @@ static FieldCodec()
             Action<CodedOutputStream, T> writer,
             Func<T, int> sizeCalculator,
             uint tag,
+            uint endTag,
             T defaultValue)
         {
             ValueReader = reader;
@@ -455,6 +478,10 @@ public void WriteTagAndValue(CodedOutputStream output, T value)
             {
                 output.WriteTag(Tag);
                 ValueWriter(output, value);
+                if (EndTag != 0)
+                {
+                    output.WriteTag(EndTag);
+                }
             }
         }
 

csharp/src/Google.Protobuf/UnknownField.cs

@@ -55,6 +55,7 @@ internal sealed class UnknownField
         private List<uint> fixed32List;
         private List<ulong> fixed64List;
         private List<ByteString> lengthDelimitedList;
+        private List<UnknownFieldSet> groupList;
 
         /// <summary>
         /// Creates a new UnknownField.
@@ -77,7 +78,8 @@ public override bool Equals(object other)
                    && Lists.Equals(varintList, otherField.varintList)
                    && Lists.Equals(fixed32List, otherField.fixed32List)
                    && Lists.Equals(fixed64List, otherField.fixed64List)
-                   && Lists.Equals(lengthDelimitedList, otherField.lengthDelimitedList);
+                   && Lists.Equals(lengthDelimitedList, otherField.lengthDelimitedList)
+                   && Lists.Equals(groupList, otherField.groupList);
         }
 
         /// <summary>
@@ -90,6 +92,7 @@ public override int GetHashCode()
             hash = hash * 47 + Lists.GetHashCode(fixed32List);
             hash = hash * 47 + Lists.GetHashCode(fixed64List);
             hash = hash * 47 + Lists.GetHashCode(lengthDelimitedList);
+            hash = hash * 47 + Lists.GetHashCode(groupList);
             return hash;
         }
 
@@ -133,6 +136,15 @@ internal void WriteTo(int fieldNumber, CodedOutputStream output)
                     output.WriteBytes(value);
                 }
             }
+            if (groupList != null)
+            {
+                foreach (UnknownFieldSet value in groupList)
+                {
+                    output.WriteTag(fieldNumber, WireFormat.WireType.StartGroup);
+                    value.WriteTo(output);
+                    output.WriteTag(fieldNumber, WireFormat.WireType.EndGroup);
+                }
+            }
         }
 
         /// <summary>
@@ -168,6 +180,14 @@ internal int GetSerializedSize(int fieldNumber)
                     result += CodedOutputStream.ComputeBytesSize(value);
                 }
             }
+            if (groupList != null)
+            {
+                result += CodedOutputStream.ComputeTagSize(fieldNumber) * 2 * groupList.Count;
+                foreach (UnknownFieldSet value in groupList)
+                {
+                    result += value.CalculateSize();
+                }
+            }
             return result;
         }
 
@@ -182,6 +202,7 @@ internal UnknownField MergeFrom(UnknownField other)
             fixed32List = AddAll(fixed32List, other.fixed32List);
             fixed64List = AddAll(fixed64List, other.fixed64List);
             lengthDelimitedList = AddAll(lengthDelimitedList, other.lengthDelimitedList);
+            groupList = AddAll(groupList, other.groupList);
             return this;
         }
 
@@ -245,6 +266,12 @@ internal UnknownField AddLengthDelimited(ByteString value)
             return this;
         }
 
+        internal UnknownField AddGroup(UnknownFieldSet value) 
+        {
+            groupList = Add(groupList, value);
+            return this;
+        }
+
         /// <summary>
         /// Adds <paramref name="value"/> to the <paramref name="list"/>, creating
         /// a new list if <paramref name="list"/> is null. The list is returned - either

csharp/src/Google.Protobuf/UnknownFieldSet.cs

@@ -215,7 +215,13 @@ private void MergeFieldFrom(CodedInputStream input)
                     }
                 case WireFormat.WireType.StartGroup:
                     {
-                        input.SkipGroup(tag);
+                        uint endTag = WireFormat.MakeTag(number, WireFormat.WireType.EndGroup);
+                        UnknownFieldSet set = new UnknownFieldSet();
+                        while (input.ReadTag() != endTag)
+                        {
+                            set.MergeFieldFrom(input);
+                        }
+                        GetOrAddField(number).AddGroup(set);
                         return;
                     }
                 case WireFormat.WireType.EndGroup: