generated_message_table_driven.cc 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  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. #include <google/protobuf/generated_message_table_driven.h>
  31. #include <type_traits>
  32. #include <google/protobuf/stubs/casts.h>
  33. #include <google/protobuf/generated_message_table_driven_lite.h>
  34. #include <google/protobuf/io/zero_copy_stream_impl_lite.h>
  35. #include <google/protobuf/metadata.h>
  36. #include <google/protobuf/repeated_field.h>
  37. #include <google/protobuf/wire_format.h>
  38. #include <google/protobuf/wire_format_lite.h>
  39. #include <google/protobuf/wire_format_lite_inl.h>
  40. namespace google {
  41. namespace protobuf {
  42. namespace internal {
  43. namespace {
  44. UnknownFieldSet* MutableUnknownFields(MessageLite* msg, int64 arena_offset) {
  45. return Raw<InternalMetadataWithArena>(msg, arena_offset)
  46. ->mutable_unknown_fields();
  47. }
  48. struct UnknownFieldHandler {
  49. static bool Skip(MessageLite* msg, const ParseTable& table,
  50. io::CodedInputStream* input,
  51. int tag) {
  52. GOOGLE_DCHECK(table.unknown_field_set);
  53. return WireFormat::SkipField(input, tag,
  54. MutableUnknownFields(msg, table.arena_offset));
  55. }
  56. static void Varint(MessageLite* msg, const ParseTable& table,
  57. int tag, int value) {
  58. GOOGLE_DCHECK(table.unknown_field_set);
  59. MutableUnknownFields(msg, table.arena_offset)->AddVarint(
  60. WireFormatLite::GetTagFieldNumber(tag), value);
  61. }
  62. static bool ParseExtension(
  63. MessageLite* msg, const ParseTable& table,
  64. io::CodedInputStream* input, int tag) {
  65. ExtensionSet* extensions = GetExtensionSet(msg, table.extension_offset);
  66. if (extensions == NULL) {
  67. return false;
  68. }
  69. const Message* prototype = down_cast<const Message*>(
  70. table.default_instance());
  71. GOOGLE_DCHECK(prototype != NULL);
  72. GOOGLE_DCHECK(table.unknown_field_set);
  73. UnknownFieldSet* unknown_fields =
  74. MutableUnknownFields(msg, table.arena_offset);
  75. return extensions->ParseField(tag, input, prototype, unknown_fields);
  76. }
  77. };
  78. } // namespace
  79. bool MergePartialFromCodedStream(
  80. MessageLite* msg, const ParseTable& table, io::CodedInputStream* input) {
  81. return MergePartialFromCodedStreamImpl<UnknownFieldHandler,
  82. InternalMetadataWithArena>(msg, table,
  83. input);
  84. }
  85. } // namespace internal
  86. } // namespace protobuf
  87. } // namespace google