| #region Copyright notice and license |
| // Protocol Buffers - Google's data interchange format |
| // Copyright 2008 Google Inc. All rights reserved. |
| // https://developers.google.com/protocol-buffers/ |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are |
| // met: |
| // |
| // * Redistributions of source code must retain the above copyright |
| // notice, this list of conditions and the following disclaimer. |
| // * Redistributions in binary form must reproduce the above |
| // copyright notice, this list of conditions and the following disclaimer |
| // in the documentation and/or other materials provided with the |
| // distribution. |
| // * Neither the name of Google Inc. nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| #endregion |
| |
| using Google.Protobuf.Collections; |
| using System; |
| using System.IO; |
| using System.Security; |
| using System.Text; |
| |
| namespace Google.Protobuf |
| { |
| /// <summary> |
| /// Encodes and writes protocol message fields. |
| /// </summary> |
| /// <remarks> |
| /// <para> |
| /// This class is generally used by generated code to write appropriate |
| /// primitives to the stream. It effectively encapsulates the lowest |
| /// levels of protocol buffer format. Unlike some other implementations, |
| /// this does not include combined "write tag and value" methods. Generated |
| /// code knows the exact byte representations of the tags they're going to write, |
| /// so there's no need to re-encode them each time. Manually-written code calling |
| /// this class should just call one of the <c>WriteTag</c> overloads before each value. |
| /// </para> |
| /// <para> |
| /// Repeated fields and map fields are not handled by this class; use <c>RepeatedField<T></c> |
| /// and <c>MapField<TKey, TValue></c> to serialize such fields. |
| /// </para> |
| /// </remarks> |
| [SecuritySafeCritical] |
| public sealed partial class CodedOutputStream : IDisposable |
| { |
| /// <summary> |
| /// The buffer size used by CreateInstance(Stream). |
| /// </summary> |
| public static readonly int DefaultBufferSize = 4096; |
| |
| private readonly bool leaveOpen; |
| private readonly byte[] buffer; |
| private WriterInternalState state; |
| |
| private readonly Stream output; |
| |
| #region Construction |
| /// <summary> |
| /// Creates a new CodedOutputStream that writes directly to the given |
| /// byte array. If more bytes are written than fit in the array, |
| /// OutOfSpaceException will be thrown. |
| /// </summary> |
| public CodedOutputStream(byte[] flatArray) : this(flatArray, 0, flatArray.Length) |
| { |
| } |
| |
| /// <summary> |
| /// Creates a new CodedOutputStream that writes directly to the given |
| /// byte array slice. If more bytes are written than fit in the array, |
| /// OutOfSpaceException will be thrown. |
| /// </summary> |
| private CodedOutputStream(byte[] buffer, int offset, int length) |
| { |
| this.output = null; |
| this.buffer = ProtoPreconditions.CheckNotNull(buffer, nameof(buffer)); |
| this.state.position = offset; |
| this.state.limit = offset + length; |
| WriteBufferHelper.Initialize(this, out this.state.writeBufferHelper); |
| leaveOpen = true; // Simple way of avoiding trying to dispose of a null reference |
| } |
| |
| private CodedOutputStream(Stream output, byte[] buffer, bool leaveOpen) |
| { |
| this.output = ProtoPreconditions.CheckNotNull(output, nameof(output)); |
| this.buffer = buffer; |
| this.state.position = 0; |
| this.state.limit = buffer.Length; |
| WriteBufferHelper.Initialize(this, out this.state.writeBufferHelper); |
| this.leaveOpen = leaveOpen; |
| } |
| |
| /// <summary> |
| /// Creates a new <see cref="CodedOutputStream" /> which write to the given stream, and disposes of that |
| /// stream when the returned <c>CodedOutputStream</c> is disposed. |
| /// </summary> |
| /// <param name="output">The stream to write to. It will be disposed when the returned <c>CodedOutputStream is disposed.</c></param> |
| public CodedOutputStream(Stream output) : this(output, DefaultBufferSize, false) |
| { |
| } |
| |
| /// <summary> |
| /// Creates a new CodedOutputStream which write to the given stream and uses |
| /// the specified buffer size. |
| /// </summary> |
| /// <param name="output">The stream to write to. It will be disposed when the returned <c>CodedOutputStream is disposed.</c></param> |
| /// <param name="bufferSize">The size of buffer to use internally.</param> |
| public CodedOutputStream(Stream output, int bufferSize) : this(output, new byte[bufferSize], false) |
| { |
| } |
| |
| /// <summary> |
| /// Creates a new CodedOutputStream which write to the given stream. |
| /// </summary> |
| /// <param name="output">The stream to write to.</param> |
| /// <param name="leaveOpen">If <c>true</c>, <paramref name="output"/> is left open when the returned <c>CodedOutputStream</c> is disposed; |
| /// if <c>false</c>, the provided stream is disposed as well.</param> |
| public CodedOutputStream(Stream output, bool leaveOpen) : this(output, DefaultBufferSize, leaveOpen) |
| { |
| } |
| |
| /// <summary> |
| /// Creates a new CodedOutputStream which write to the given stream and uses |
| /// the specified buffer size. |
| /// </summary> |
| /// <param name="output">The stream to write to.</param> |
| /// <param name="bufferSize">The size of buffer to use internally.</param> |
| /// <param name="leaveOpen">If <c>true</c>, <paramref name="output"/> is left open when the returned <c>CodedOutputStream</c> is disposed; |
| /// if <c>false</c>, the provided stream is disposed as well.</param> |
| public CodedOutputStream(Stream output, int bufferSize, bool leaveOpen) : this(output, new byte[bufferSize], leaveOpen) |
| { |
| } |
| #endregion |
| |
| /// <summary> |
| /// Returns the current position in the stream, or the position in the output buffer |
| /// </summary> |
| public long Position |
| { |
| get |
| { |
| if (output != null) |
| { |
| return output.Position + state.position; |
| } |
| return state.position; |
| } |
| } |
| |
| #region Writing of values (not including tags) |
| |
| /// <summary> |
| /// Writes a double field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteDouble(double value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteDouble(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a float field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteFloat(float value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteFloat(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a uint64 field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteUInt64(ulong value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteUInt64(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an int64 field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteInt64(long value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteInt64(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an int32 field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteInt32(int value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteInt32(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a fixed64 field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteFixed64(ulong value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteFixed64(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a fixed32 field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteFixed32(uint value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteFixed32(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a bool field value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteBool(bool value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteBool(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a string field value, without a tag, to the stream. |
| /// The data is length-prefixed. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteString(string value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteString(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a message, without a tag, to the stream. |
| /// The data is length-prefixed. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteMessage(IMessage value) |
| { |
| // TODO(jtattermusch): if the message doesn't implement IBufferMessage (and thus does not provide the InternalWriteTo method), |
| // what we're doing here works fine, but could be more efficient. |
| // For now, this inefficiency is fine, considering this is only a backward-compatibility scenario (and regenerating the code fixes it). |
| var span = new Span<byte>(buffer); |
| WriteContext.Initialize(ref span, ref state, out WriteContext ctx); |
| try |
| { |
| WritingPrimitivesMessages.WriteMessage(ref ctx, value); |
| } |
| finally |
| { |
| ctx.CopyStateTo(this); |
| } |
| } |
| |
| /// <summary> |
| /// Writes a message, without a tag, to the stream. |
| /// Only the message data is written, without a length-delimiter. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteRawMessage(IMessage value) |
| { |
| // TODO(jtattermusch): if the message doesn't implement IBufferMessage (and thus does not provide the InternalWriteTo method), |
| // what we're doing here works fine, but could be more efficient. |
| // For now, this inefficiency is fine, considering this is only a backward-compatibility scenario (and regenerating the code fixes it). |
| var span = new Span<byte>(buffer); |
| WriteContext.Initialize(ref span, ref state, out WriteContext ctx); |
| try |
| { |
| WritingPrimitivesMessages.WriteRawMessage(ref ctx, value); |
| } |
| finally |
| { |
| ctx.CopyStateTo(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) |
| { |
| var span = new Span<byte>(buffer); |
| WriteContext.Initialize(ref span, ref state, out WriteContext ctx); |
| try |
| { |
| WritingPrimitivesMessages.WriteGroup(ref ctx, value); |
| } |
| finally |
| { |
| ctx.CopyStateTo(this); |
| } |
| } |
| |
| /// <summary> |
| /// Write a byte string, without a tag, to the stream. |
| /// The data is length-prefixed. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteBytes(ByteString value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteBytes(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a uint32 value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteUInt32(uint value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteUInt32(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an enum value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteEnum(int value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteEnum(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an sfixed32 value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write.</param> |
| public void WriteSFixed32(int value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteSFixed32(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an sfixed64 value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteSFixed64(long value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteSFixed64(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an sint32 value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteSInt32(int value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteSInt32(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes an sint64 value, without a tag, to the stream. |
| /// </summary> |
| /// <param name="value">The value to write</param> |
| public void WriteSInt64(long value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteSInt64(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes a length (in bytes) for length-delimited data. |
| /// </summary> |
| /// <remarks> |
| /// This method simply writes a rawint, but exists for clarity in calling code. |
| /// </remarks> |
| /// <param name="length">Length value, in bytes.</param> |
| public void WriteLength(int length) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteLength(ref span, ref state, length); |
| } |
| |
| #endregion |
| |
| #region Raw tag writing |
| /// <summary> |
| /// Encodes and writes a tag. |
| /// </summary> |
| /// <param name="fieldNumber">The number of the field to write the tag for</param> |
| /// <param name="type">The wire format type of the tag to write</param> |
| public void WriteTag(int fieldNumber, WireFormat.WireType type) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteTag(ref span, ref state, fieldNumber, type); |
| } |
| |
| /// <summary> |
| /// Writes an already-encoded tag. |
| /// </summary> |
| /// <param name="tag">The encoded tag</param> |
| public void WriteTag(uint tag) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteTag(ref span, ref state, tag); |
| } |
| |
| /// <summary> |
| /// Writes the given single-byte tag directly to the stream. |
| /// </summary> |
| /// <param name="b1">The encoded tag</param> |
| public void WriteRawTag(byte b1) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawTag(ref span, ref state, b1); |
| } |
| |
| /// <summary> |
| /// Writes the given two-byte tag directly to the stream. |
| /// </summary> |
| /// <param name="b1">The first byte of the encoded tag</param> |
| /// <param name="b2">The second byte of the encoded tag</param> |
| public void WriteRawTag(byte b1, byte b2) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawTag(ref span, ref state, b1, b2); |
| } |
| |
| /// <summary> |
| /// Writes the given three-byte tag directly to the stream. |
| /// </summary> |
| /// <param name="b1">The first byte of the encoded tag</param> |
| /// <param name="b2">The second byte of the encoded tag</param> |
| /// <param name="b3">The third byte of the encoded tag</param> |
| public void WriteRawTag(byte b1, byte b2, byte b3) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawTag(ref span, ref state, b1, b2, b3); |
| } |
| |
| /// <summary> |
| /// Writes the given four-byte tag directly to the stream. |
| /// </summary> |
| /// <param name="b1">The first byte of the encoded tag</param> |
| /// <param name="b2">The second byte of the encoded tag</param> |
| /// <param name="b3">The third byte of the encoded tag</param> |
| /// <param name="b4">The fourth byte of the encoded tag</param> |
| public void WriteRawTag(byte b1, byte b2, byte b3, byte b4) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawTag(ref span, ref state, b1, b2, b3, b4); |
| } |
| |
| /// <summary> |
| /// Writes the given five-byte tag directly to the stream. |
| /// </summary> |
| /// <param name="b1">The first byte of the encoded tag</param> |
| /// <param name="b2">The second byte of the encoded tag</param> |
| /// <param name="b3">The third byte of the encoded tag</param> |
| /// <param name="b4">The fourth byte of the encoded tag</param> |
| /// <param name="b5">The fifth byte of the encoded tag</param> |
| public void WriteRawTag(byte b1, byte b2, byte b3, byte b4, byte b5) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawTag(ref span, ref state, b1, b2, b3, b4, b5); |
| } |
| #endregion |
| |
| #region Underlying writing primitives |
| |
| /// <summary> |
| /// Writes a 32 bit value as a varint. The fast route is taken when |
| /// there's enough buffer space left to whizz through without checking |
| /// for each byte; otherwise, we resort to calling WriteRawByte each time. |
| /// </summary> |
| internal void WriteRawVarint32(uint value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawVarint32(ref span, ref state, value); |
| } |
| |
| internal void WriteRawVarint64(ulong value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawVarint64(ref span, ref state, value); |
| } |
| |
| internal void WriteRawLittleEndian32(uint value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawLittleEndian32(ref span, ref state, value); |
| } |
| |
| internal void WriteRawLittleEndian64(ulong value) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawLittleEndian64(ref span, ref state, value); |
| } |
| |
| /// <summary> |
| /// Writes out an array of bytes. |
| /// </summary> |
| internal void WriteRawBytes(byte[] value) |
| { |
| WriteRawBytes(value, 0, value.Length); |
| } |
| |
| /// <summary> |
| /// Writes out part of an array of bytes. |
| /// </summary> |
| internal void WriteRawBytes(byte[] value, int offset, int length) |
| { |
| var span = new Span<byte>(buffer); |
| WritingPrimitives.WriteRawBytes(ref span, ref state, value, offset, length); |
| } |
| |
| #endregion |
| |
| /// <summary> |
| /// Indicates that a CodedOutputStream wrapping a flat byte array |
| /// ran out of space. |
| /// </summary> |
| public sealed class OutOfSpaceException : IOException |
| { |
| internal OutOfSpaceException() |
| : base("CodedOutputStream was writing to a flat byte array and ran out of space.") |
| { |
| } |
| } |
| |
| /// <summary> |
| /// Flushes any buffered data and optionally closes the underlying stream, if any. |
| /// </summary> |
| /// <remarks> |
| /// <para> |
| /// By default, any underlying stream is closed by this method. To configure this behaviour, |
| /// use a constructor overload with a <c>leaveOpen</c> parameter. If this instance does not |
| /// have an underlying stream, this method does nothing. |
| /// </para> |
| /// <para> |
| /// For the sake of efficiency, calling this method does not prevent future write calls - but |
| /// if a later write ends up writing to a stream which has been disposed, that is likely to |
| /// fail. It is recommend that you not call any other methods after this. |
| /// </para> |
| /// </remarks> |
| public void Dispose() |
| { |
| Flush(); |
| if (!leaveOpen) |
| { |
| output.Dispose(); |
| } |
| } |
| |
| /// <summary> |
| /// Flushes any buffered data to the underlying stream (if there is one). |
| /// </summary> |
| public void Flush() |
| { |
| var span = new Span<byte>(buffer); |
| WriteBufferHelper.Flush(ref span, ref state); |
| } |
| |
| /// <summary> |
| /// Verifies that SpaceLeft returns zero. It's common to create a byte array |
| /// that is exactly big enough to hold a message, then write to it with |
| /// a CodedOutputStream. Calling CheckNoSpaceLeft after writing verifies that |
| /// the message was actually as big as expected, which can help finding bugs. |
| /// </summary> |
| public void CheckNoSpaceLeft() |
| { |
| WriteBufferHelper.CheckNoSpaceLeft(ref state); |
| } |
| |
| /// <summary> |
| /// If writing to a flat array, returns the space left in the array. Otherwise, |
| /// throws an InvalidOperationException. |
| /// </summary> |
| public int SpaceLeft => WriteBufferHelper.GetSpaceLeft(ref state); |
| |
| internal byte[] InternalBuffer => buffer; |
| |
| internal Stream InternalOutputStream => output; |
| |
| internal ref WriterInternalState InternalState => ref state; |
| } |
| } |