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 // Author: kenton@google.com (Kenton Varda)
32 // Based on original Protocol Buffers design by
33 // Sanjay Ghemawat, Jeff Dean, and others.
34
35 #include <iostream>
36 #include <stack>
37 #include <unordered_map>
38
39 #include <google/protobuf/generated_message_reflection.h>
40 #include <google/protobuf/message.h>
41
42 #include <google/protobuf/stubs/casts.h>
43 #include <google/protobuf/stubs/logging.h>
44 #include <google/protobuf/stubs/common.h>
45 #include <google/protobuf/descriptor.pb.h>
46 #include <google/protobuf/parse_context.h>
47 #include <google/protobuf/reflection_internal.h>
48 #include <google/protobuf/io/coded_stream.h>
49 #include <google/protobuf/io/zero_copy_stream_impl.h>
50 #include <google/protobuf/descriptor.h>
51 #include <google/protobuf/generated_message_util.h>
52 #include <google/protobuf/map_field.h>
53 #include <google/protobuf/map_field_inl.h>
54 #include <google/protobuf/reflection_ops.h>
55 #include <google/protobuf/unknown_field_set.h>
56 #include <google/protobuf/wire_format.h>
57 #include <google/protobuf/wire_format_lite.h>
58 #include <google/protobuf/stubs/strutil.h>
59
60 #include <google/protobuf/stubs/map_util.h>
61 #include <google/protobuf/stubs/stl_util.h>
62 #include <google/protobuf/stubs/hash.h>
63
64 #include <google/protobuf/port_def.inc>
65
66 namespace google {
67 namespace protobuf {
68
69 namespace internal {
70
71 // TODO(gerbens) make this factorized better. This should not have to hop
72 // to reflection. Currently uses GeneratedMessageReflection and thus is
73 // defined in generated_message_reflection.cc
74 void RegisterFileLevelMetadata(const DescriptorTable* descriptor_table);
75
76 } // namespace internal
77
78 using internal::ReflectionOps;
79 using internal::WireFormat;
80 using internal::WireFormatLite;
81
MergeFrom(const Message & from)82 void Message::MergeFrom(const Message& from) {
83 const Descriptor* descriptor = GetDescriptor();
84 GOOGLE_CHECK_EQ(from.GetDescriptor(), descriptor)
85 << ": Tried to merge from a message with a different type. "
86 "to: "
87 << descriptor->full_name()
88 << ", "
89 "from: "
90 << from.GetDescriptor()->full_name();
91 ReflectionOps::Merge(from, this);
92 }
93
CheckTypeAndMergeFrom(const MessageLite & other)94 void Message::CheckTypeAndMergeFrom(const MessageLite& other) {
95 MergeFrom(*down_cast<const Message*>(&other));
96 }
97
CopyFrom(const Message & from)98 void Message::CopyFrom(const Message& from) {
99 const Descriptor* descriptor = GetDescriptor();
100 GOOGLE_CHECK_EQ(from.GetDescriptor(), descriptor)
101 << ": Tried to copy from a message with a different type. "
102 "to: "
103 << descriptor->full_name()
104 << ", "
105 "from: "
106 << from.GetDescriptor()->full_name();
107 ReflectionOps::Copy(from, this);
108 }
109
GetTypeName() const110 std::string Message::GetTypeName() const {
111 return GetDescriptor()->full_name();
112 }
113
Clear()114 void Message::Clear() { ReflectionOps::Clear(this); }
115
IsInitialized() const116 bool Message::IsInitialized() const {
117 return ReflectionOps::IsInitialized(*this);
118 }
119
FindInitializationErrors(std::vector<std::string> * errors) const120 void Message::FindInitializationErrors(std::vector<std::string>* errors) const {
121 return ReflectionOps::FindInitializationErrors(*this, "", errors);
122 }
123
InitializationErrorString() const124 std::string Message::InitializationErrorString() const {
125 std::vector<std::string> errors;
126 FindInitializationErrors(&errors);
127 return Join(errors, ", ");
128 }
129
CheckInitialized() const130 void Message::CheckInitialized() const {
131 GOOGLE_CHECK(IsInitialized()) << "Message of type \"" << GetDescriptor()->full_name()
132 << "\" is missing required fields: "
133 << InitializationErrorString();
134 }
135
DiscardUnknownFields()136 void Message::DiscardUnknownFields() {
137 return ReflectionOps::DiscardUnknownFields(this);
138 }
139
140 #if !GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER
MergePartialFromCodedStream(io::CodedInputStream * input)141 bool Message::MergePartialFromCodedStream(io::CodedInputStream* input) {
142 return WireFormat::ParseAndMergePartial(input, this);
143 }
144 #endif
145
146 #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER
147 namespace internal {
148
149 class ReflectionAccessor {
150 public:
GetOffset(void * msg,const google::protobuf::FieldDescriptor * f,const google::protobuf::Reflection * r)151 static void* GetOffset(void* msg, const google::protobuf::FieldDescriptor* f,
152 const google::protobuf::Reflection* r) {
153 return static_cast<char*>(msg) + r->schema_.GetFieldOffset(f);
154 }
155
GetRepeatedEnum(const Reflection * reflection,const FieldDescriptor * field,Message * msg)156 static void* GetRepeatedEnum(const Reflection* reflection,
157 const FieldDescriptor* field, Message* msg) {
158 return reflection->MutableRawRepeatedField(
159 msg, field, FieldDescriptor::CPPTYPE_ENUM, 0, nullptr);
160 }
161
MutableInternalMetadataWithArena(const Reflection * reflection,Message * msg)162 static InternalMetadataWithArena* MutableInternalMetadataWithArena(
163 const Reflection* reflection, Message* msg) {
164 return reflection->MutableInternalMetadataWithArena(msg);
165 }
166 };
167
168 } // namespace internal
169
SetField(uint64 val,const FieldDescriptor * field,Message * msg,const Reflection * reflection)170 void SetField(uint64 val, const FieldDescriptor* field, Message* msg,
171 const Reflection* reflection) {
172 #define STORE_TYPE(CPPTYPE_METHOD) \
173 do \
174 if (field->is_repeated()) { \
175 reflection->Add##CPPTYPE_METHOD(msg, field, value); \
176 } else { \
177 reflection->Set##CPPTYPE_METHOD(msg, field, value); \
178 } \
179 while (0)
180
181 switch (field->type()) {
182 #define HANDLE_TYPE(TYPE, CPPTYPE, CPPTYPE_METHOD) \
183 case FieldDescriptor::TYPE_##TYPE: { \
184 CPPTYPE value = val; \
185 STORE_TYPE(CPPTYPE_METHOD); \
186 break; \
187 }
188
189 // Varints
190 HANDLE_TYPE(INT32, int32, Int32)
191 HANDLE_TYPE(INT64, int64, Int64)
192 HANDLE_TYPE(UINT32, uint32, UInt32)
193 HANDLE_TYPE(UINT64, uint64, UInt64)
194 case FieldDescriptor::TYPE_SINT32: {
195 int32 value = WireFormatLite::ZigZagDecode32(val);
196 STORE_TYPE(Int32);
197 break;
198 }
199 case FieldDescriptor::TYPE_SINT64: {
200 int64 value = WireFormatLite::ZigZagDecode64(val);
201 STORE_TYPE(Int64);
202 break;
203 }
204 HANDLE_TYPE(BOOL, bool, Bool)
205
206 // Fixed
207 HANDLE_TYPE(FIXED32, uint32, UInt32)
208 HANDLE_TYPE(FIXED64, uint64, UInt64)
209 HANDLE_TYPE(SFIXED32, int32, Int32)
210 HANDLE_TYPE(SFIXED64, int64, Int64)
211
212 case FieldDescriptor::TYPE_FLOAT: {
213 float value;
214 uint32 bit_rep = val;
215 std::memcpy(&value, &bit_rep, sizeof(value));
216 STORE_TYPE(Float);
217 break;
218 }
219 case FieldDescriptor::TYPE_DOUBLE: {
220 double value;
221 uint64 bit_rep = val;
222 std::memcpy(&value, &bit_rep, sizeof(value));
223 STORE_TYPE(Double);
224 break;
225 }
226 case FieldDescriptor::TYPE_ENUM: {
227 int value = val;
228 if (field->is_repeated()) {
229 reflection->AddEnumValue(msg, field, value);
230 } else {
231 reflection->SetEnumValue(msg, field, value);
232 }
233 break;
234 }
235 default:
236 GOOGLE_LOG(FATAL) << "Error in descriptors, primitve field with field type "
237 << field->type();
238 }
239 #undef STORE_TYPE
240 #undef HANDLE_TYPE
241 }
242
ReflectiveValidator(const void * arg,int val)243 bool ReflectiveValidator(const void* arg, int val) {
244 auto d = static_cast<const EnumDescriptor*>(arg);
245 return d->FindValueByNumber(val) != nullptr;
246 }
247
ParsePackedField(const FieldDescriptor * field,Message * msg,const Reflection * reflection,const char * ptr,internal::ParseContext * ctx)248 const char* ParsePackedField(const FieldDescriptor* field, Message* msg,
249 const Reflection* reflection, const char* ptr,
250 internal::ParseContext* ctx) {
251 switch (field->type()) {
252 #define HANDLE_PACKED_TYPE(TYPE, CPPTYPE, METHOD_NAME) \
253 case FieldDescriptor::TYPE_##TYPE: \
254 return internal::Packed##METHOD_NAME##Parser( \
255 reflection->MutableRepeatedField<CPPTYPE>(msg, field), ptr, ctx)
256 HANDLE_PACKED_TYPE(INT32, int32, Int32);
257 HANDLE_PACKED_TYPE(INT64, int64, Int64);
258 HANDLE_PACKED_TYPE(SINT32, int32, SInt32);
259 HANDLE_PACKED_TYPE(SINT64, int64, SInt64);
260 HANDLE_PACKED_TYPE(UINT32, uint32, UInt32);
261 HANDLE_PACKED_TYPE(UINT64, uint64, UInt64);
262 HANDLE_PACKED_TYPE(BOOL, bool, Bool);
263 case FieldDescriptor::TYPE_ENUM: {
264 auto object =
265 internal::ReflectionAccessor::GetRepeatedEnum(reflection, field, msg);
266 if (field->file()->syntax() == FileDescriptor::SYNTAX_PROTO3) {
267 return internal::PackedEnumParser(object, ptr, ctx);
268 } else {
269 return internal::PackedEnumParserArg(
270 object, ptr, ctx, ReflectiveValidator, field->enum_type(),
271 internal::ReflectionAccessor::MutableInternalMetadataWithArena(
272 reflection, msg),
273 field->number());
274 }
275 }
276 HANDLE_PACKED_TYPE(FIXED32, uint32, Fixed32);
277 HANDLE_PACKED_TYPE(FIXED64, uint64, Fixed64);
278 HANDLE_PACKED_TYPE(SFIXED32, int32, SFixed32);
279 HANDLE_PACKED_TYPE(SFIXED64, int64, SFixed64);
280 HANDLE_PACKED_TYPE(FLOAT, float, Float);
281 HANDLE_PACKED_TYPE(DOUBLE, double, Double);
282 #undef HANDLE_PACKED_TYPE
283
284 default:
285 GOOGLE_LOG(FATAL) << "Type is not packable " << field->type();
286 return nullptr; // Make compiler happy
287 }
288 }
289
ParseLenDelim(int field_number,const FieldDescriptor * field,Message * msg,const Reflection * reflection,const char * ptr,internal::ParseContext * ctx)290 const char* ParseLenDelim(int field_number, const FieldDescriptor* field,
291 Message* msg, const Reflection* reflection,
292 const char* ptr, internal::ParseContext* ctx) {
293 if (WireFormat::WireTypeForFieldType(field->type()) !=
294 WireFormatLite::WIRETYPE_LENGTH_DELIMITED) {
295 GOOGLE_DCHECK(field->is_packable());
296 return ParsePackedField(field, msg, reflection, ptr, ctx);
297 }
298 enum { kNone = 0, kVerify, kStrict } utf8_level = kNone;
299 const char* field_name = nullptr;
300 auto parse_string = [ptr, ctx, &utf8_level, &field_name](std::string* s) {
301 switch (utf8_level) {
302 case kNone:
303 return internal::InlineGreedyStringParser(s, ptr, ctx);
304 case kVerify:
305 return internal::InlineGreedyStringParserUTF8Verify(s, ptr, ctx,
306 field_name);
307 case kStrict:
308 return internal::InlineGreedyStringParserUTF8(s, ptr, ctx, field_name);
309 }
310 };
311 switch (field->type()) {
312 case FieldDescriptor::TYPE_STRING: {
313 bool enforce_utf8 = true;
314 bool utf8_verification = true;
315 if (enforce_utf8 &&
316 field->file()->syntax() == FileDescriptor::SYNTAX_PROTO3) {
317 utf8_level = kStrict;
318 } else if (utf8_verification) {
319 utf8_level = kVerify;
320 }
321 field_name = field->full_name().c_str();
322 PROTOBUF_FALLTHROUGH_INTENDED;
323 }
324 case FieldDescriptor::TYPE_BYTES: {
325 if (field->is_repeated()) {
326 int index = reflection->FieldSize(*msg, field);
327 // Add new empty value.
328 reflection->AddString(msg, field, "");
329 if (field->options().ctype() == FieldOptions::STRING ||
330 field->is_extension()) {
331 auto object =
332 reflection->MutableRepeatedPtrField<std::string>(msg, field)
333 ->Mutable(index);
334 return parse_string(object);
335 } else {
336 auto object =
337 reflection->MutableRepeatedPtrField<std::string>(msg, field)
338 ->Mutable(index);
339 return parse_string(object);
340 }
341 } else {
342 // Clear value and make sure it's set.
343 reflection->SetString(msg, field, "");
344 if (field->options().ctype() == FieldOptions::STRING ||
345 field->is_extension()) {
346 // HACK around inability to get mutable_string in reflection
347 std::string* object = &const_cast<std::string&>(
348 reflection->GetStringReference(*msg, field, nullptr));
349 return parse_string(object);
350 } else {
351 // HACK around inability to get mutable_string in reflection
352 std::string* object = &const_cast<std::string&>(
353 reflection->GetStringReference(*msg, field, nullptr));
354 return parse_string(object);
355 }
356 }
357 GOOGLE_LOG(FATAL) << "No other type than string supported";
358 }
359 case FieldDescriptor::TYPE_MESSAGE: {
360 Message* object;
361 if (field->is_repeated()) {
362 object = reflection->AddMessage(msg, field, ctx->data().factory);
363 } else {
364 object = reflection->MutableMessage(msg, field, ctx->data().factory);
365 }
366 return ctx->ParseMessage(object, ptr);
367 }
368 default:
369 GOOGLE_LOG(FATAL) << "Wrong type for length delim " << field->type();
370 }
371 return nullptr; // Make compiler happy.
372 }
373
GetGroup(int field_number,const FieldDescriptor * field,Message * msg,const Reflection * reflection)374 Message* GetGroup(int field_number, const FieldDescriptor* field, Message* msg,
375 const Reflection* reflection) {
376 if (field->is_repeated()) {
377 return reflection->AddMessage(msg, field, nullptr);
378 } else {
379 return reflection->MutableMessage(msg, field, nullptr);
380 }
381 }
382
_InternalParse(const char * ptr,internal::ParseContext * ctx)383 const char* Message::_InternalParse(const char* ptr,
384 internal::ParseContext* ctx) {
385 class ReflectiveFieldParser {
386 public:
387 ReflectiveFieldParser(Message* msg, internal::ParseContext* ctx)
388 : ReflectiveFieldParser(msg, ctx, false) {}
389
390 void AddVarint(uint32 num, uint64 value) {
391 if (is_item_ && num == 2) {
392 if (!payload_.empty()) {
393 auto field = Field(value, 2);
394 if (field && field->message_type()) {
395 auto child = reflection_->MutableMessage(msg_, field);
396 // TODO(gerbens) signal error
397 child->ParsePartialFromString(payload_);
398 } else {
399 MutableUnknown()->AddLengthDelimited(value)->swap(payload_);
400 }
401 return;
402 }
403 type_id_ = value;
404 return;
405 }
406 auto field = Field(num, 0);
407 if (field) {
408 SetField(value, field, msg_, reflection_);
409 } else {
410 MutableUnknown()->AddVarint(num, value);
411 }
412 }
413 void AddFixed64(uint32 num, uint64 value) {
414 auto field = Field(num, 1);
415 if (field) {
416 SetField(value, field, msg_, reflection_);
417 } else {
418 MutableUnknown()->AddFixed64(num, value);
419 }
420 }
421 const char* ParseLengthDelimited(uint32 num, const char* ptr,
422 internal::ParseContext* ctx) {
423 if (is_item_ && num == 3) {
424 if (type_id_ == 0) {
425 return InlineGreedyStringParser(&payload_, ptr, ctx);
426 }
427 num = type_id_;
428 type_id_ = 0;
429 }
430 auto field = Field(num, 2);
431 if (field) {
432 return ParseLenDelim(num, field, msg_, reflection_, ptr, ctx);
433 } else {
434 return InlineGreedyStringParser(
435 MutableUnknown()->AddLengthDelimited(num), ptr, ctx);
436 }
437 }
438 const char* ParseGroup(uint32 num, const char* ptr,
439 internal::ParseContext* ctx) {
440 if (!is_item_ && descriptor_->options().message_set_wire_format() &&
441 num == 1) {
442 is_item_ = true;
443 ptr = ctx->ParseGroup(this, ptr, num * 8 + 3);
444 is_item_ = false;
445 type_id_ = 0;
446 return ptr;
447 }
448 auto field = Field(num, 3);
449 if (field) {
450 auto msg = GetGroup(num, field, msg_, reflection_);
451 return ctx->ParseGroup(msg, ptr, num * 8 + 3);
452 } else {
453 return UnknownFieldParse(num * 8 + 3, MutableUnknown(), ptr, ctx);
454 }
455 }
456 void AddFixed32(uint32 num, uint32 value) {
457 auto field = Field(num, 5);
458 if (field) {
459 SetField(value, field, msg_, reflection_);
460 } else {
461 MutableUnknown()->AddFixed32(num, value);
462 }
463 }
464
465 const char* _InternalParse(const char* ptr, internal::ParseContext* ctx) {
466 // We're parsing the a MessageSetItem
467 GOOGLE_DCHECK(is_item_);
468 return internal::WireFormatParser(*this, ptr, ctx);
469 }
470
471 private:
472 Message* msg_;
473 const Descriptor* descriptor_;
474 const Reflection* reflection_;
475 internal::ParseContext* ctx_;
476 UnknownFieldSet* unknown_ = nullptr;
477 bool is_item_ = false;
478 uint32 type_id_ = 0;
479 std::string payload_;
480
481 ReflectiveFieldParser(Message* msg, internal::ParseContext* ctx,
482 bool is_item)
483 : msg_(msg),
484 descriptor_(msg->GetDescriptor()),
485 reflection_(msg->GetReflection()),
486 ctx_(ctx),
487 is_item_(is_item) {
488 GOOGLE_CHECK(descriptor_) << msg->GetTypeName();
489 GOOGLE_CHECK(reflection_) << msg->GetTypeName();
490 }
491
492 const FieldDescriptor* Field(int num, int wire_type) {
493 auto field = descriptor_->FindFieldByNumber(num);
494
495 // If that failed, check if the field is an extension.
496 if (field == nullptr && descriptor_->IsExtensionNumber(num)) {
497 const DescriptorPool* pool = ctx_->data().pool;
498 if (pool == NULL) {
499 field = reflection_->FindKnownExtensionByNumber(num);
500 } else {
501 field = pool->FindExtensionByNumber(descriptor_, num);
502 }
503 }
504 if (field == nullptr) return nullptr;
505
506 if (internal::WireFormat::WireTypeForFieldType(field->type()) !=
507 wire_type) {
508 if (field->is_packable()) {
509 if (wire_type ==
510 internal::WireFormatLite::WIRETYPE_LENGTH_DELIMITED) {
511 return field;
512 }
513 }
514 return nullptr;
515 }
516 return field;
517 }
518
519 UnknownFieldSet* MutableUnknown() {
520 if (unknown_) return unknown_;
521 return unknown_ = reflection_->MutableUnknownFields(msg_);
522 }
523 };
524
525 ReflectiveFieldParser field_parser(this, ctx);
526 return internal::WireFormatParser(field_parser, ptr, ctx);
527 }
528 #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER
529
SerializeWithCachedSizes(io::CodedOutputStream * output) const530 void Message::SerializeWithCachedSizes(io::CodedOutputStream* output) const {
531 const internal::SerializationTable* table =
532 static_cast<const internal::SerializationTable*>(InternalGetTable());
533 if (table == 0) {
534 WireFormat::SerializeWithCachedSizes(*this, GetCachedSize(), output);
535 } else {
536 internal::TableSerialize(*this, table, output);
537 }
538 }
539
ByteSizeLong() const540 size_t Message::ByteSizeLong() const {
541 size_t size = WireFormat::ByteSize(*this);
542 SetCachedSize(internal::ToCachedSize(size));
543 return size;
544 }
545
SetCachedSize(int) const546 void Message::SetCachedSize(int /* size */) const {
547 GOOGLE_LOG(FATAL) << "Message class \"" << GetDescriptor()->full_name()
548 << "\" implements neither SetCachedSize() nor ByteSize(). "
549 "Must implement one or the other.";
550 }
551
SpaceUsedLong() const552 size_t Message::SpaceUsedLong() const {
553 return GetReflection()->SpaceUsedLong(*this);
554 }
555
556 // =============================================================================
557 // MessageFactory
558
~MessageFactory()559 MessageFactory::~MessageFactory() {}
560
561 namespace {
562
563 class GeneratedMessageFactory : public MessageFactory {
564 public:
565 static GeneratedMessageFactory* singleton();
566
567 void RegisterFile(const google::protobuf::internal::DescriptorTable* table);
568 void RegisterType(const Descriptor* descriptor, const Message* prototype);
569
570 // implements MessageFactory ---------------------------------------
571 const Message* GetPrototype(const Descriptor* type) override;
572
573 private:
574 // Only written at static init time, so does not require locking.
575 std::unordered_map<const char*, const google::protobuf::internal::DescriptorTable*,
576 hash<const char*>, streq>
577 file_map_;
578
579 internal::WrappedMutex mutex_;
580 // Initialized lazily, so requires locking.
581 std::unordered_map<const Descriptor*, const Message*> type_map_;
582 };
583
singleton()584 GeneratedMessageFactory* GeneratedMessageFactory::singleton() {
585 static auto instance =
586 internal::OnShutdownDelete(new GeneratedMessageFactory);
587 return instance;
588 }
589
RegisterFile(const google::protobuf::internal::DescriptorTable * table)590 void GeneratedMessageFactory::RegisterFile(
591 const google::protobuf::internal::DescriptorTable* table) {
592 if (!InsertIfNotPresent(&file_map_, table->filename, table)) {
593 GOOGLE_LOG(FATAL) << "File is already registered: " << table->filename;
594 }
595 }
596
RegisterType(const Descriptor * descriptor,const Message * prototype)597 void GeneratedMessageFactory::RegisterType(const Descriptor* descriptor,
598 const Message* prototype) {
599 GOOGLE_DCHECK_EQ(descriptor->file()->pool(), DescriptorPool::generated_pool())
600 << "Tried to register a non-generated type with the generated "
601 "type registry.";
602
603 // This should only be called as a result of calling a file registration
604 // function during GetPrototype(), in which case we already have locked
605 // the mutex.
606 mutex_.AssertHeld();
607 if (!InsertIfNotPresent(&type_map_, descriptor, prototype)) {
608 GOOGLE_LOG(DFATAL) << "Type is already registered: " << descriptor->full_name();
609 }
610 }
611
612
GetPrototype(const Descriptor * type)613 const Message* GeneratedMessageFactory::GetPrototype(const Descriptor* type) {
614 {
615 ReaderMutexLock lock(&mutex_);
616 const Message* result = FindPtrOrNull(type_map_, type);
617 if (result != NULL) return result;
618 }
619
620 // If the type is not in the generated pool, then we can't possibly handle
621 // it.
622 if (type->file()->pool() != DescriptorPool::generated_pool()) return NULL;
623
624 // Apparently the file hasn't been registered yet. Let's do that now.
625 const internal::DescriptorTable* registration_data =
626 FindPtrOrNull(file_map_, type->file()->name().c_str());
627 if (registration_data == NULL) {
628 GOOGLE_LOG(DFATAL) << "File appears to be in generated pool but wasn't "
629 "registered: "
630 << type->file()->name();
631 return NULL;
632 }
633
634 WriterMutexLock lock(&mutex_);
635
636 // Check if another thread preempted us.
637 const Message* result = FindPtrOrNull(type_map_, type);
638 if (result == NULL) {
639 // Nope. OK, register everything.
640 internal::RegisterFileLevelMetadata(registration_data);
641 // Should be here now.
642 result = FindPtrOrNull(type_map_, type);
643 }
644
645 if (result == NULL) {
646 GOOGLE_LOG(DFATAL) << "Type appears to be in generated pool but wasn't "
647 << "registered: " << type->full_name();
648 }
649
650 return result;
651 }
652
653 } // namespace
654
generated_factory()655 MessageFactory* MessageFactory::generated_factory() {
656 return GeneratedMessageFactory::singleton();
657 }
658
InternalRegisterGeneratedFile(const google::protobuf::internal::DescriptorTable * table)659 void MessageFactory::InternalRegisterGeneratedFile(
660 const google::protobuf::internal::DescriptorTable* table) {
661 GeneratedMessageFactory::singleton()->RegisterFile(table);
662 }
663
InternalRegisterGeneratedMessage(const Descriptor * descriptor,const Message * prototype)664 void MessageFactory::InternalRegisterGeneratedMessage(
665 const Descriptor* descriptor, const Message* prototype) {
666 GeneratedMessageFactory::singleton()->RegisterType(descriptor, prototype);
667 }
668
669
670 namespace {
671 template <typename T>
GetSingleton()672 T* GetSingleton() {
673 static T singleton;
674 return &singleton;
675 }
676 } // namespace
677
RepeatedFieldAccessor(const FieldDescriptor * field) const678 const internal::RepeatedFieldAccessor* Reflection::RepeatedFieldAccessor(
679 const FieldDescriptor* field) const {
680 GOOGLE_CHECK(field->is_repeated());
681 switch (field->cpp_type()) {
682 #define HANDLE_PRIMITIVE_TYPE(TYPE, type) \
683 case FieldDescriptor::CPPTYPE_##TYPE: \
684 return GetSingleton<internal::RepeatedFieldPrimitiveAccessor<type> >();
685 HANDLE_PRIMITIVE_TYPE(INT32, int32)
686 HANDLE_PRIMITIVE_TYPE(UINT32, uint32)
687 HANDLE_PRIMITIVE_TYPE(INT64, int64)
688 HANDLE_PRIMITIVE_TYPE(UINT64, uint64)
689 HANDLE_PRIMITIVE_TYPE(FLOAT, float)
690 HANDLE_PRIMITIVE_TYPE(DOUBLE, double)
691 HANDLE_PRIMITIVE_TYPE(BOOL, bool)
692 HANDLE_PRIMITIVE_TYPE(ENUM, int32)
693 #undef HANDLE_PRIMITIVE_TYPE
694 case FieldDescriptor::CPPTYPE_STRING:
695 switch (field->options().ctype()) {
696 default:
697 case FieldOptions::STRING:
698 return GetSingleton<internal::RepeatedPtrFieldStringAccessor>();
699 }
700 break;
701 case FieldDescriptor::CPPTYPE_MESSAGE:
702 if (field->is_map()) {
703 return GetSingleton<internal::MapFieldAccessor>();
704 } else {
705 return GetSingleton<internal::RepeatedPtrFieldMessageAccessor>();
706 }
707 }
708 GOOGLE_LOG(FATAL) << "Should not reach here.";
709 return NULL;
710 }
711
712 namespace internal {
713 template <>
714 #if defined(_MSC_VER) && (_MSC_VER >= 1800)
715 // Note: force noinline to workaround MSVC compiler bug with /Zc:inline, issue
716 // #240
717 PROTOBUF_NOINLINE
718 #endif
719 Message*
NewFromPrototype(const Message * prototype,Arena * arena)720 GenericTypeHandler<Message>::NewFromPrototype(const Message* prototype,
721 Arena* arena) {
722 return prototype->New(arena);
723 }
724 template <>
725 #if defined(_MSC_VER) && (_MSC_VER >= 1800)
726 // Note: force noinline to workaround MSVC compiler bug with /Zc:inline, issue
727 // #240
728 PROTOBUF_NOINLINE
729 #endif
730 Arena*
GetArena(Message * value)731 GenericTypeHandler<Message>::GetArena(Message* value) {
732 return value->GetArena();
733 }
734 template <>
735 #if defined(_MSC_VER) && (_MSC_VER >= 1800)
736 // Note: force noinline to workaround MSVC compiler bug with /Zc:inline, issue
737 // #240
738 PROTOBUF_NOINLINE
739 #endif
740 void*
GetMaybeArenaPointer(Message * value)741 GenericTypeHandler<Message>::GetMaybeArenaPointer(Message* value) {
742 return value->GetMaybeArenaPointer();
743 }
744 } // namespace internal
745
746 } // namespace protobuf
747 } // namespace google
748