| // Protocol Buffers - Google's data interchange format |
| // Copyright 2008 Google Inc. All rights reserved. |
| // http://code.google.com/p/protobuf/ |
| // |
| // 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. |
| |
| // Author: kenton@google.com (Kenton Varda) |
| // Based on original Protocol Buffers design by |
| // Sanjay Ghemawat, Jeff Dean, and others. |
| |
| #include <google/protobuf/wire_format_lite_inl.h> |
| |
| #include <stack> |
| #include <string> |
| #include <vector> |
| #include <google/protobuf/stubs/common.h> |
| #include <google/protobuf/io/coded_stream_inl.h> |
| #include <google/protobuf/io/zero_copy_stream.h> |
| #include <google/protobuf/io/zero_copy_stream_impl.h> |
| |
| namespace google { |
| namespace protobuf { |
| namespace internal { |
| |
| #ifndef _MSC_VER // MSVC doesn't like definitions of inline constants, GCC |
| // requires them. |
| const int WireFormatLite::kMessageSetItemStartTag; |
| const int WireFormatLite::kMessageSetItemEndTag; |
| const int WireFormatLite::kMessageSetTypeIdTag; |
| const int WireFormatLite::kMessageSetMessageTag; |
| |
| #endif |
| |
| const int WireFormatLite::kMessageSetItemTagsSize = |
| io::CodedOutputStream::VarintSize32(kMessageSetItemStartTag) + |
| io::CodedOutputStream::VarintSize32(kMessageSetItemEndTag) + |
| io::CodedOutputStream::VarintSize32(kMessageSetTypeIdTag) + |
| io::CodedOutputStream::VarintSize32(kMessageSetMessageTag); |
| |
| const WireFormatLite::CppType |
| WireFormatLite::kFieldTypeToCppTypeMap[MAX_FIELD_TYPE + 1] = { |
| static_cast<CppType>(0), // 0 is reserved for errors |
| |
| CPPTYPE_DOUBLE, // TYPE_DOUBLE |
| CPPTYPE_FLOAT, // TYPE_FLOAT |
| CPPTYPE_INT64, // TYPE_INT64 |
| CPPTYPE_UINT64, // TYPE_UINT64 |
| CPPTYPE_INT32, // TYPE_INT32 |
| CPPTYPE_UINT64, // TYPE_FIXED64 |
| CPPTYPE_UINT32, // TYPE_FIXED32 |
| CPPTYPE_BOOL, // TYPE_BOOL |
| CPPTYPE_STRING, // TYPE_STRING |
| CPPTYPE_MESSAGE, // TYPE_GROUP |
| CPPTYPE_MESSAGE, // TYPE_MESSAGE |
| CPPTYPE_STRING, // TYPE_BYTES |
| CPPTYPE_UINT32, // TYPE_UINT32 |
| CPPTYPE_ENUM, // TYPE_ENUM |
| CPPTYPE_INT32, // TYPE_SFIXED32 |
| CPPTYPE_INT64, // TYPE_SFIXED64 |
| CPPTYPE_INT32, // TYPE_SINT32 |
| CPPTYPE_INT64, // TYPE_SINT64 |
| }; |
| |
| const WireFormatLite::WireType |
| WireFormatLite::kWireTypeForFieldType[MAX_FIELD_TYPE + 1] = { |
| static_cast<WireFormatLite::WireType>(-1), // invalid |
| WireFormatLite::WIRETYPE_FIXED64, // TYPE_DOUBLE |
| WireFormatLite::WIRETYPE_FIXED32, // TYPE_FLOAT |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_INT64 |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_UINT64 |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_INT32 |
| WireFormatLite::WIRETYPE_FIXED64, // TYPE_FIXED64 |
| WireFormatLite::WIRETYPE_FIXED32, // TYPE_FIXED32 |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_BOOL |
| WireFormatLite::WIRETYPE_LENGTH_DELIMITED, // TYPE_STRING |
| WireFormatLite::WIRETYPE_START_GROUP, // TYPE_GROUP |
| WireFormatLite::WIRETYPE_LENGTH_DELIMITED, // TYPE_MESSAGE |
| WireFormatLite::WIRETYPE_LENGTH_DELIMITED, // TYPE_BYTES |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_UINT32 |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_ENUM |
| WireFormatLite::WIRETYPE_FIXED32, // TYPE_SFIXED32 |
| WireFormatLite::WIRETYPE_FIXED64, // TYPE_SFIXED64 |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_SINT32 |
| WireFormatLite::WIRETYPE_VARINT, // TYPE_SINT64 |
| }; |
| |
| bool WireFormatLite::SkipField( |
| io::CodedInputStream* input, uint32 tag) { |
| switch (WireFormatLite::GetTagWireType(tag)) { |
| case WireFormatLite::WIRETYPE_VARINT: { |
| uint64 value; |
| if (!input->ReadVarint64(&value)) return false; |
| return true; |
| } |
| case WireFormatLite::WIRETYPE_FIXED64: { |
| uint64 value; |
| if (!input->ReadLittleEndian64(&value)) return false; |
| return true; |
| } |
| case WireFormatLite::WIRETYPE_LENGTH_DELIMITED: { |
| uint32 length; |
| if (!input->ReadVarint32(&length)) return false; |
| if (!input->Skip(length)) return false; |
| return true; |
| } |
| case WireFormatLite::WIRETYPE_START_GROUP: { |
| if (!input->IncrementRecursionDepth()) return false; |
| if (!SkipMessage(input)) return false; |
| input->DecrementRecursionDepth(); |
| // Check that the ending tag matched the starting tag. |
| if (!input->LastTagWas(WireFormatLite::MakeTag( |
| WireFormatLite::GetTagFieldNumber(tag), |
| WireFormatLite::WIRETYPE_END_GROUP))) { |
| return false; |
| } |
| return true; |
| } |
| case WireFormatLite::WIRETYPE_END_GROUP: { |
| return false; |
| } |
| case WireFormatLite::WIRETYPE_FIXED32: { |
| uint32 value; |
| if (!input->ReadLittleEndian32(&value)) return false; |
| return true; |
| } |
| default: { |
| return false; |
| } |
| } |
| } |
| |
| bool WireFormatLite::SkipMessage(io::CodedInputStream* input) { |
| while(true) { |
| uint32 tag = input->ReadTag(); |
| if (tag == 0) { |
| // End of input. This is a valid place to end, so return true. |
| return true; |
| } |
| |
| WireFormatLite::WireType wire_type = WireFormatLite::GetTagWireType(tag); |
| |
| if (wire_type == WireFormatLite::WIRETYPE_END_GROUP) { |
| // Must be the end of the message. |
| return true; |
| } |
| |
| if (!SkipField(input, tag)) return false; |
| } |
| } |
| |
| bool FieldSkipper::SkipField( |
| io::CodedInputStream* input, uint32 tag) { |
| return WireFormatLite::SkipField(input, tag); |
| } |
| |
| bool FieldSkipper::SkipMessage(io::CodedInputStream* input) { |
| return WireFormatLite::SkipMessage(input); |
| } |
| |
| void FieldSkipper::SkipUnknownEnum( |
| int field_number, int value) { |
| // Nothing. |
| } |
| |
| bool WireFormatLite::ReadPackedEnumNoInline(io::CodedInputStream* input, |
| bool (*is_valid)(int), |
| RepeatedField<int>* values) { |
| uint32 length; |
| if (!input->ReadVarint32(&length)) return false; |
| io::CodedInputStream::Limit limit = input->PushLimit(length); |
| while (input->BytesUntilLimit() > 0) { |
| int value; |
| if (!google::protobuf::internal::WireFormatLite::ReadPrimitive< |
| int, WireFormatLite::TYPE_ENUM>(input, &value)) { |
| return false; |
| } |
| if (is_valid(value)) { |
| values->Add(value); |
| } |
| } |
| input->PopLimit(limit); |
| return true; |
| } |
| |
| void WireFormatLite::WriteInt32(int field_number, int32 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteInt32NoTag(value, output); |
| } |
| void WireFormatLite::WriteInt64(int field_number, int64 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteInt64NoTag(value, output); |
| } |
| void WireFormatLite::WriteUInt32(int field_number, uint32 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteUInt32NoTag(value, output); |
| } |
| void WireFormatLite::WriteUInt64(int field_number, uint64 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteUInt64NoTag(value, output); |
| } |
| void WireFormatLite::WriteSInt32(int field_number, int32 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteSInt32NoTag(value, output); |
| } |
| void WireFormatLite::WriteSInt64(int field_number, int64 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteSInt64NoTag(value, output); |
| } |
| void WireFormatLite::WriteFixed32(int field_number, uint32 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_FIXED32, output); |
| WriteFixed32NoTag(value, output); |
| } |
| void WireFormatLite::WriteFixed64(int field_number, uint64 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_FIXED64, output); |
| WriteFixed64NoTag(value, output); |
| } |
| void WireFormatLite::WriteSFixed32(int field_number, int32 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_FIXED32, output); |
| WriteSFixed32NoTag(value, output); |
| } |
| void WireFormatLite::WriteSFixed64(int field_number, int64 value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_FIXED64, output); |
| WriteSFixed64NoTag(value, output); |
| } |
| void WireFormatLite::WriteFloat(int field_number, float value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_FIXED32, output); |
| WriteFloatNoTag(value, output); |
| } |
| void WireFormatLite::WriteDouble(int field_number, double value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_FIXED64, output); |
| WriteDoubleNoTag(value, output); |
| } |
| void WireFormatLite::WriteBool(int field_number, bool value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteBoolNoTag(value, output); |
| } |
| void WireFormatLite::WriteEnum(int field_number, int value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_VARINT, output); |
| WriteEnumNoTag(value, output); |
| } |
| |
| void WireFormatLite::WriteString(int field_number, const string& value, |
| io::CodedOutputStream* output) { |
| // String is for UTF-8 text only |
| WriteTag(field_number, WIRETYPE_LENGTH_DELIMITED, output); |
| output->WriteVarint32(value.size()); |
| output->WriteString(value); |
| } |
| void WireFormatLite::WriteBytes(int field_number, const string& value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_LENGTH_DELIMITED, output); |
| output->WriteVarint32(value.size()); |
| output->WriteString(value); |
| } |
| |
| |
| void WireFormatLite::WriteGroup(int field_number, |
| const MessageLite& value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_START_GROUP, output); |
| value.SerializeWithCachedSizes(output); |
| WriteTag(field_number, WIRETYPE_END_GROUP, output); |
| } |
| |
| void WireFormatLite::WriteMessage(int field_number, |
| const MessageLite& value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_LENGTH_DELIMITED, output); |
| const int size = value.GetCachedSize(); |
| output->WriteVarint32(size); |
| value.SerializeWithCachedSizes(output); |
| } |
| |
| void WireFormatLite::WriteGroupMaybeToArray(int field_number, |
| const MessageLite& value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_START_GROUP, output); |
| const int size = value.GetCachedSize(); |
| uint8* target = output->GetDirectBufferForNBytesAndAdvance(size); |
| if (target != NULL) { |
| uint8* end = value.SerializeWithCachedSizesToArray(target); |
| GOOGLE_DCHECK_EQ(end - target, size); |
| } else { |
| value.SerializeWithCachedSizes(output); |
| } |
| WriteTag(field_number, WIRETYPE_END_GROUP, output); |
| } |
| |
| void WireFormatLite::WriteMessageMaybeToArray(int field_number, |
| const MessageLite& value, |
| io::CodedOutputStream* output) { |
| WriteTag(field_number, WIRETYPE_LENGTH_DELIMITED, output); |
| const int size = value.GetCachedSize(); |
| output->WriteVarint32(size); |
| uint8* target = output->GetDirectBufferForNBytesAndAdvance(size); |
| if (target != NULL) { |
| uint8* end = value.SerializeWithCachedSizesToArray(target); |
| GOOGLE_DCHECK_EQ(end - target, size); |
| } else { |
| value.SerializeWithCachedSizes(output); |
| } |
| } |
| |
| bool WireFormatLite::ReadString(io::CodedInputStream* input, |
| string* value) { |
| // String is for UTF-8 text only |
| uint32 length; |
| if (!input->ReadVarint32(&length)) return false; |
| if (!input->InternalReadStringInline(value, length)) return false; |
| return true; |
| } |
| bool WireFormatLite::ReadBytes(io::CodedInputStream* input, |
| string* value) { |
| uint32 length; |
| if (!input->ReadVarint32(&length)) return false; |
| return input->InternalReadStringInline(value, length); |
| } |
| |
| } // namespace internal |
| } // namespace protobuf |
| } // namespace google |