1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2008 Google Inc.  All rights reserved.
3 // https://developers.google.com/protocol-buffers/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 //     * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 //     * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 //     * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 
31 #include <google/protobuf/generated_message_table_driven.h>
32 
33 #include <type_traits>
34 
35 #include <google/protobuf/stubs/casts.h>
36 #include <google/protobuf/generated_message_table_driven_lite.h>
37 #include <google/protobuf/io/zero_copy_stream_impl_lite.h>
38 #include <google/protobuf/repeated_field.h>
39 #include <google/protobuf/wire_format.h>
40 #include <google/protobuf/wire_format_lite.h>
41 
42 namespace google {
43 namespace protobuf {
44 namespace internal {
45 
46 namespace {
47 
MutableUnknownFields(MessageLite * msg,int64 arena_offset)48 UnknownFieldSet* MutableUnknownFields(MessageLite* msg, int64 arena_offset) {
49   return Raw<InternalMetadata>(msg, arena_offset)
50       ->mutable_unknown_fields<UnknownFieldSet>();
51 }
52 
53 struct UnknownFieldHandler {
54   // TODO(mvels): consider renaming UnknownFieldHandler to (TableDrivenTraits?),
55   // and conflating InternalMetaData into it, simplifying the template.
IsLitegoogle::protobuf::internal::__anonfcbebdf80111::UnknownFieldHandler56   static constexpr bool IsLite() { return false; }
57 
Skipgoogle::protobuf::internal::__anonfcbebdf80111::UnknownFieldHandler58   static bool Skip(MessageLite* msg, const ParseTable& table,
59                    io::CodedInputStream* input, int tag) {
60     GOOGLE_DCHECK(table.unknown_field_set);
61 
62     return WireFormat::SkipField(input, tag,
63                                  MutableUnknownFields(msg, table.arena_offset));
64   }
65 
Varintgoogle::protobuf::internal::__anonfcbebdf80111::UnknownFieldHandler66   static void Varint(MessageLite* msg, const ParseTable& table, int tag,
67                      int value) {
68     GOOGLE_DCHECK(table.unknown_field_set);
69 
70     MutableUnknownFields(msg, table.arena_offset)
71         ->AddVarint(WireFormatLite::GetTagFieldNumber(tag), value);
72   }
73 
ParseExtensiongoogle::protobuf::internal::__anonfcbebdf80111::UnknownFieldHandler74   static bool ParseExtension(MessageLite* msg, const ParseTable& table,
75                              io::CodedInputStream* input, int tag) {
76     ExtensionSet* extensions = GetExtensionSet(msg, table.extension_offset);
77     if (extensions == NULL) {
78       return false;
79     }
80 
81     const Message* prototype =
82         down_cast<const Message*>(table.default_instance());
83 
84     GOOGLE_DCHECK(prototype != NULL);
85     GOOGLE_DCHECK(table.unknown_field_set);
86     UnknownFieldSet* unknown_fields =
87         MutableUnknownFields(msg, table.arena_offset);
88 
89     return extensions->ParseField(tag, input, prototype, unknown_fields);
90   }
91 };
92 
93 }  // namespace
94 
MergePartialFromCodedStream(MessageLite * msg,const ParseTable & table,io::CodedInputStream * input)95 bool MergePartialFromCodedStream(MessageLite* msg, const ParseTable& table,
96                                  io::CodedInputStream* input) {
97   return MergePartialFromCodedStreamImpl<UnknownFieldHandler>(msg, table,
98                                                               input);
99 }
100 
101 }  // namespace internal
102 }  // namespace protobuf
103 }  // namespace google
104