1fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Protocol Buffers - Google's data interchange format
2fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Copyright 2008 Google Inc.  All rights reserved.
3fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// http://code.google.com/p/protobuf/
4fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
5fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Redistribution and use in source and binary forms, with or without
6fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modification, are permitted provided that the following conditions are
7fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// met:
8fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
9fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Redistributions of source code must retain the above copyright
10fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// notice, this list of conditions and the following disclaimer.
11fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Redistributions in binary form must reproduce the above
12fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// copyright notice, this list of conditions and the following disclaimer
13fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// in the documentation and/or other materials provided with the
14fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// distribution.
15fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//     * Neither the name of Google Inc. nor the names of its
16fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contributors may be used to endorse or promote products derived from
17fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// this software without specific prior written permission.
18fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
19fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
31fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Author: kenton@google.com (Kenton Varda)
32fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//  Based on original Protocol Buffers design by
33fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//  Sanjay Ghemawat, Jeff Dean, and others.
34fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
35fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/hash.h>
36fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <map>
37fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <set>
38fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <vector>
39fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <algorithm>
40d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville#include <limits>
41fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
42fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/descriptor.h>
43fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/descriptor_database.h>
44fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/descriptor.pb.h>
45fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/text_format.h>
46fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/unknown_field_set.h>
47fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/wire_format.h>
48fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/coded_stream.h>
49fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/zero_copy_stream_impl.h>
50fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/common.h>
51fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/once.h>
52fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/strutil.h>
53fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/substitute.h>
54fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/map-util.h>
55fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/stl_util-inl.h>
56fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
57fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#undef PACKAGE  // autoheader #defines this.  :(
58fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
59fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace google {
60fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace protobuf {
61fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
62fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor::CppType
63fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFieldDescriptor::kTypeToCppTypeMap[MAX_TYPE + 1] = {
64fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static_cast<CppType>(0),  // 0 is reserved for errors
65fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
66fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_DOUBLE,   // TYPE_DOUBLE
67fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_FLOAT,    // TYPE_FLOAT
68fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_INT64,    // TYPE_INT64
69fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_UINT64,   // TYPE_UINT64
70fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_INT32,    // TYPE_INT32
71fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_UINT64,   // TYPE_FIXED64
72fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_UINT32,   // TYPE_FIXED32
73fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_BOOL,     // TYPE_BOOL
74fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_STRING,   // TYPE_STRING
75fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_MESSAGE,  // TYPE_GROUP
76fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_MESSAGE,  // TYPE_MESSAGE
77fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_STRING,   // TYPE_BYTES
78fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_UINT32,   // TYPE_UINT32
79fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_ENUM,     // TYPE_ENUM
80fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_INT32,    // TYPE_SFIXED32
81fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_INT64,    // TYPE_SFIXED64
82fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_INT32,    // TYPE_SINT32
83fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CPPTYPE_INT64,    // TYPE_SINT64
84fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
85fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
86fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst char * const FieldDescriptor::kTypeToName[MAX_TYPE + 1] = {
87fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "ERROR",     // 0 is reserved for errors
88fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
89fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "double",    // TYPE_DOUBLE
90fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "float",     // TYPE_FLOAT
91fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "int64",     // TYPE_INT64
92fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "uint64",    // TYPE_UINT64
93fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "int32",     // TYPE_INT32
94fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "fixed64",   // TYPE_FIXED64
95fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "fixed32",   // TYPE_FIXED32
96fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "bool",      // TYPE_BOOL
97fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "string",    // TYPE_STRING
98fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "group",     // TYPE_GROUP
99fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "message",   // TYPE_MESSAGE
100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "bytes",     // TYPE_BYTES
101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "uint32",    // TYPE_UINT32
102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "enum",      // TYPE_ENUM
103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "sfixed32",  // TYPE_SFIXED32
104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "sfixed64",  // TYPE_SFIXED64
105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "sint32",    // TYPE_SINT32
106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "sint64",    // TYPE_SINT64
107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst char * const FieldDescriptor::kLabelToName[MAX_LABEL + 1] = {
110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "ERROR",     // 0 is reserved for errors
111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "optional",  // LABEL_OPTIONAL
113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "required",  // LABEL_REQUIRED
114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  "repeated",  // LABEL_REPEATED
115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#ifndef _MSC_VER  // MSVC doesn't need these and won't even accept them.
118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst int FieldDescriptor::kMaxNumber;
119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst int FieldDescriptor::kFirstReservedNumber;
120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst int FieldDescriptor::kLastReservedNumber;
121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#endif
122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace {
124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst string kEmptyString;
126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring ToCamelCase(const string& input) {
128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool capitalize_next = false;
129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string result;
130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result.reserve(input.size());
131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < input.size(); i++) {
133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (input[i] == '_') {
134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      capitalize_next = true;
135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (capitalize_next) {
136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Note:  I distrust ctype.h due to locales.
137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if ('a' <= input[i] && input[i] <= 'z') {
138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        result.push_back(input[i] - 'a' + 'A');
139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        result.push_back(input[i]);
141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      capitalize_next = false;
143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      result.push_back(input[i]);
145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Lower-case the first letter.
149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.empty() && 'A' <= result[0] && result[0] <= 'Z') {
150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result[0] = result[0] - 'A' + 'a';
151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A DescriptorPool contains a bunch of hash_maps to implement the
157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// various Find*By*() methods.  Since hashtable lookups are O(1), it's
158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// most efficient to construct a fixed set of large hash_maps used by
159fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// all objects in the pool rather than construct one or more small
160fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// hash_maps for each object.
161fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
162fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// The keys to these hash_maps are (parent, name) or (parent, number)
163fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// pairs.  Unfortunately STL doesn't provide hash functions for pair<>,
164fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// so we must invent our own.
165fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
166fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// TODO(kenton):  Use StringPiece rather than const char* in keys?  It would
167fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//   be a lot cleaner but we'd just have to convert it back to const char*
168fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//   for the open source release.
169fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
170fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef pair<const void*, const char*> PointerStringPair;
171fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
172fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestruct PointerStringPairEqual {
173fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline bool operator()(const PointerStringPair& a,
174fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         const PointerStringPair& b) const {
175fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return a.first == b.first && strcmp(a.second, b.second) == 0;
176fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
177fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
178fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
179fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletemplate<typename PairType>
180fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestruct PointerIntegerPairHash {
181fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  size_t operator()(const PairType& p) const {
182fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // FIXME(kenton):  What is the best way to compute this hash?  I have
183fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // no idea!  This seems a bit better than an XOR.
184fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return reinterpret_cast<intptr_t>(p.first) * ((1 << 16) - 1) + p.second;
185fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
186fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
187fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Used only by MSVC and platforms where hash_map is not available.
188fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static const size_t bucket_size = 4;
189fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static const size_t min_buckets = 8;
190fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline bool operator()(const PairType& a, const PairType& b) const {
191fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return a.first < b.first ||
192fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          (a.first == b.first && a.second < b.second);
193fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
194fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
195fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
196fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef pair<const Descriptor*, int> DescriptorIntPair;
197fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef pair<const EnumDescriptor*, int> EnumIntPair;
198fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
199fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestruct PointerStringPairHash {
200fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  size_t operator()(const PointerStringPair& p) const {
201fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // FIXME(kenton):  What is the best way to compute this hash?  I have
202fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // no idea!  This seems a bit better than an XOR.
203fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    hash<const char*> cstring_hash;
204fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return reinterpret_cast<intptr_t>(p.first) * ((1 << 16) - 1) +
205fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville           cstring_hash(p.second);
206fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
207fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
208fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Used only by MSVC and platforms where hash_map is not available.
209fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static const size_t bucket_size = 4;
210fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static const size_t min_buckets = 8;
211fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline bool operator()(const PointerStringPair& a,
212fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         const PointerStringPair& b) const {
213fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (a.first < b.first) return true;
214fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (a.first > b.first) return false;
215fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return strcmp(a.second, b.second) < 0;
216fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
217fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
218fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
219fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
220fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestruct Symbol {
221fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  enum Type {
222fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    NULL_SYMBOL, MESSAGE, FIELD, ENUM, ENUM_VALUE, SERVICE, METHOD, PACKAGE
223fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  };
224fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Type type;
225fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  union {
226fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* descriptor;
227fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* field_descriptor;
228fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const EnumDescriptor* enum_descriptor;
229fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const EnumValueDescriptor* enum_value_descriptor;
230fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const ServiceDescriptor* service_descriptor;
231fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const MethodDescriptor* method_descriptor;
232fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptor* package_file_descriptor;
233fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  };
234fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
235fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline Symbol() : type(NULL_SYMBOL) { descriptor = NULL; }
236fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline bool IsNull() const { return type == NULL_SYMBOL; }
237fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline bool IsType() const {
238fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return type == MESSAGE || type == ENUM;
239fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
240fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline bool IsAggregate() const {
241fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return type == MESSAGE || type == PACKAGE
242fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        || type == ENUM || type == SERVICE;
243fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
244fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
245fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#define CONSTRUCTOR(TYPE, TYPE_CONSTANT, FIELD)  \
246fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline explicit Symbol(const TYPE* value) {    \
247fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    type = TYPE_CONSTANT;                        \
248fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    this->FIELD = value;                         \
249fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
250fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
251fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(Descriptor         , MESSAGE   , descriptor             )
252fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(FieldDescriptor    , FIELD     , field_descriptor       )
253fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(EnumDescriptor     , ENUM      , enum_descriptor        )
254fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(EnumValueDescriptor, ENUM_VALUE, enum_value_descriptor  )
255fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(ServiceDescriptor  , SERVICE   , service_descriptor     )
256fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(MethodDescriptor   , METHOD    , method_descriptor      )
257fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CONSTRUCTOR(FileDescriptor     , PACKAGE   , package_file_descriptor)
258fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#undef CONSTRUCTOR
259fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
260fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* GetFile() const {
261fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    switch (type) {
262fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case NULL_SYMBOL: return NULL;
263fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case MESSAGE    : return descriptor           ->file();
264fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case FIELD      : return field_descriptor     ->file();
265fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case ENUM       : return enum_descriptor      ->file();
266fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case ENUM_VALUE : return enum_value_descriptor->type()->file();
267fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case SERVICE    : return service_descriptor   ->file();
268fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case METHOD     : return method_descriptor    ->service()->file();
269fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case PACKAGE    : return package_file_descriptor;
270fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
271fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
272fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
273fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
274fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
275fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst Symbol kNullSymbol;
276fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
277fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef hash_map<const char*, Symbol,
278fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 hash<const char*>, streq>
279fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  SymbolsByNameMap;
280fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef hash_map<PointerStringPair, Symbol,
281fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 PointerStringPairHash, PointerStringPairEqual>
282fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  SymbolsByParentMap;
283fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef hash_map<const char*, const FileDescriptor*,
284fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 hash<const char*>, streq>
285fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FilesByNameMap;
286fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef hash_map<PointerStringPair, const FieldDescriptor*,
287fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 PointerStringPairHash, PointerStringPairEqual>
288fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FieldsByNameMap;
289fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef hash_map<DescriptorIntPair, const FieldDescriptor*,
290fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 PointerIntegerPairHash<DescriptorIntPair> >
291fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FieldsByNumberMap;
292fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef hash_map<EnumIntPair, const EnumValueDescriptor*,
293fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 PointerIntegerPairHash<EnumIntPair> >
294fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  EnumValuesByNumberMap;
295fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// This is a map rather than a hash_map, since we use it to iterate
296fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// through all the extensions that extend a given Descriptor, and an
297fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// ordered data structure that implements lower_bound is convenient
298fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// for that.
299fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletypedef map<DescriptorIntPair, const FieldDescriptor*>
300fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ExtensionsGroupedByDescriptorMap;
301fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
302fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // anonymous namespace
303fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
304fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// ===================================================================
305fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DescriptorPool::Tables
306fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
307fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleclass DescriptorPool::Tables {
308fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public:
309fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Tables();
310fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ~Tables();
311fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
312fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Checkpoint the state of the tables.  Future calls to Rollback() will
313fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // return the Tables to this state.  This is used when building files, since
314fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // some kinds of validation errors cannot be detected until the file's
315fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptors have already been added to the tables.  BuildFile() calls
316fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Checkpoint() before it starts building and Rollback() if it encounters
317fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // an error.
318fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void Checkpoint();
319fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
320fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Roll back the Tables to the state of the last Checkpoint(), removing
321fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // everything that was added after that point.
322fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void Rollback();
323fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
324fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // The stack of files which are currently being built.  Used to detect
325fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // cyclic dependencies when loading files from a DescriptorDatabase.  Not
326fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // used when fallback_database_ == NULL.
327fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<string> pending_files_;
328fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
329fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // A set of files which we have tried to load from the fallback database
330fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // and encountered errors.  We will not attempt to load them again.
331fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Not used when fallback_database_ == NULL.
332fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  hash_set<string> known_bad_files_;
333fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
334fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // The set of descriptors for which we've already loaded the full
335fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // set of extensions numbers from fallback_database_.
336fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  hash_set<const Descriptor*> extensions_loaded_from_db_;
337fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
338fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // -----------------------------------------------------------------
339fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Finding items.
340fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
341fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find symbols.  This returns a null Symbol (symbol.IsNull() is true)
342fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // if not found.
343fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline Symbol FindSymbol(const string& key) const;
344fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
345fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // This implements the body of DescriptorPool::Find*ByName().  It should
346fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // really be a private method of DescriptorPool, but that would require
347fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // declaring Symbol in descriptor.h, which would drag all kinds of other
348fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // stuff into the header.  Yay C++.
349fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol FindByNameHelper(
350fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const DescriptorPool* pool, const string& name) const;
351fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
352fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // These return NULL if not found.
353fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline const FileDescriptor* FindFile(const string& key) const;
354fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline const FieldDescriptor* FindExtension(const Descriptor* extendee,
355fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                              int number);
356fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline void FindAllExtensions(const Descriptor* extendee,
357fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                vector<const FieldDescriptor*>* out) const;
358fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
359fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // -----------------------------------------------------------------
360fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Adding items.
361fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
362fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // These add items to the corresponding tables.  They return false if
363fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the key already exists in the table.  For AddSymbol(), the string passed
364fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // in must be one that was constructed using AllocateString(), as it will
365fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // be used as a key in the symbols_by_name_ map without copying.
366fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddSymbol(const string& full_name, Symbol symbol);
367fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddFile(const FileDescriptor* file);
368fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddExtension(const FieldDescriptor* field);
369fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
370fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // -----------------------------------------------------------------
371fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocating memory.
372fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
373fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocate an object which will be reclaimed when the pool is
374fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // destroyed.  Note that the object's destructor will never be called,
375fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // so its fields must be plain old data (primitive data types and
376fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // pointers).  All of the descriptor types are such objects.
377fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  template<typename Type> Type* Allocate();
378fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
379fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocate an array of objects which will be reclaimed when the
380fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // pool in destroyed.  Again, destructors are never called.
381fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  template<typename Type> Type* AllocateArray(int count);
382fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
383fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocate a string which will be destroyed when the pool is destroyed.
384fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // The string is initialized to the given value for convenience.
385fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* AllocateString(const string& value);
386fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
387fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocate a protocol message object.  Some older versions of GCC have
388fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // trouble understanding explicit template instantiations in some cases, so
389fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // in those cases we have to pass a dummy pointer of the right type as the
390fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // parameter instead of specifying the type explicitly.
391fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  template<typename Type> Type* AllocateMessage(Type* dummy = NULL);
392fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
393fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocate a FileDescriptorTables object.
394fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorTables* AllocateFileTables();
395fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
396fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private:
397fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<string*> strings_;    // All strings in the pool.
398fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<Message*> messages_;  // All messages in the pool.
399fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<FileDescriptorTables*> file_tables_;  // All file tables in the pool.
400fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<void*> allocations_;  // All other memory allocated in the pool.
401fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
402fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  SymbolsByNameMap      symbols_by_name_;
403fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FilesByNameMap        files_by_name_;
404fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ExtensionsGroupedByDescriptorMap extensions_;
405fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
406fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int strings_before_checkpoint_;
407fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int messages_before_checkpoint_;
408fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int file_tables_before_checkpoint_;
409fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int allocations_before_checkpoint_;
410fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<const char*      > symbols_after_checkpoint_;
411fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<const char*      > files_after_checkpoint_;
412fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<DescriptorIntPair> extensions_after_checkpoint_;
413fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
414fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocate some bytes which will be reclaimed when the pool is
415fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // destroyed.
416fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void* AllocateBytes(int size);
417fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
418fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
419fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Contains tables specific to a particular file.  These tables are not
420fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modified once the file has been constructed, so they need not be
421fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// protected by a mutex.  This makes operations that depend only on the
422fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contents of a single file -- e.g. Descriptor::FindFieldByName() --
423fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// lock-free.
424fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//
425fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// For historical reasons, the definitions of the methods of
426fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// FileDescriptorTables and DescriptorPool::Tables are interleaved below.
427fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// These used to be a single class.
428fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleclass FileDescriptorTables {
429fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public:
430fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorTables();
431fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ~FileDescriptorTables();
432fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
433fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Empty table, used with placeholder files.
434fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static const FileDescriptorTables kEmpty;
435fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
436fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // -----------------------------------------------------------------
437fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Finding items.
438fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
439fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find symbols.  These return a null Symbol (symbol.IsNull() is true)
440fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // if not found.
441fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline Symbol FindNestedSymbol(const void* parent,
442fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 const string& name) const;
443fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline Symbol FindNestedSymbolOfType(const void* parent,
444fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                       const string& name,
445fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                       const Symbol::Type type) const;
446fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
447fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // These return NULL if not found.
448fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline const FieldDescriptor* FindFieldByNumber(
449fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* parent, int number) const;
450fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline const FieldDescriptor* FindFieldByLowercaseName(
451fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& lowercase_name) const;
452fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline const FieldDescriptor* FindFieldByCamelcaseName(
453fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& camelcase_name) const;
454fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline const EnumValueDescriptor* FindEnumValueByNumber(
455fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const EnumDescriptor* parent, int number) const;
456fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
457fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // -----------------------------------------------------------------
458fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Adding items.
459fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
460fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // These add items to the corresponding tables.  They return false if
461fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the key already exists in the table.  For AddAliasUnderParent(), the
462fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // string passed in must be one that was constructed using AllocateString(),
463fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // as it will be used as a key in the symbols_by_parent_ map without copying.
464fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddAliasUnderParent(const void* parent, const string& name,
465fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           Symbol symbol);
466fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddFieldByNumber(const FieldDescriptor* field);
467fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddEnumValueByNumber(const EnumValueDescriptor* value);
468fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
469fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Adds the field to the lowercase_name and camelcase_name maps.  Never
470fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // fails because we allow duplicates; the first field by the name wins.
471fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void AddFieldByStylizedNames(const FieldDescriptor* field);
472fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
473fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private:
474fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  SymbolsByParentMap    symbols_by_parent_;
475fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FieldsByNameMap       fields_by_lowercase_name_;
476fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FieldsByNameMap       fields_by_camelcase_name_;
477fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FieldsByNumberMap     fields_by_number_;       // Not including extensions.
478fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  EnumValuesByNumberMap enum_values_by_number_;
479fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
480fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
481fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::Tables::Tables()
482fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  : strings_before_checkpoint_(0),
483fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    messages_before_checkpoint_(0),
484fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    allocations_before_checkpoint_(0) {}
485fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
486fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::Tables::~Tables() {
487fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note that the deletion order is important, since the destructors of some
488fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // messages may refer to objects in allocations_.
489fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  STLDeleteElements(&messages_);
490fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < allocations_.size(); i++) {
491fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    operator delete(allocations_[i]);
492fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
493fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  STLDeleteElements(&strings_);
494fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  STLDeleteElements(&file_tables_);
495fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
496fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
497fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptorTables::FileDescriptorTables() {}
498fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptorTables::~FileDescriptorTables() {}
499fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
500fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptorTables FileDescriptorTables::kEmpty;
501fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
502fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorPool::Tables::Checkpoint() {
503fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings_before_checkpoint_ = strings_.size();
504fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  messages_before_checkpoint_ = messages_.size();
505fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_tables_before_checkpoint_ = file_tables_.size();
506fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  allocations_before_checkpoint_ = allocations_.size();
507fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
508fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  symbols_after_checkpoint_.clear();
509fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  files_after_checkpoint_.clear();
510fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  extensions_after_checkpoint_.clear();
511fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
512fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
513fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorPool::Tables::Rollback() {
514fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < symbols_after_checkpoint_.size(); i++) {
515fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    symbols_by_name_.erase(symbols_after_checkpoint_[i]);
516fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
517fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < files_after_checkpoint_.size(); i++) {
518fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    files_by_name_.erase(files_after_checkpoint_[i]);
519fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
520fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extensions_after_checkpoint_.size(); i++) {
521fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    extensions_.erase(extensions_after_checkpoint_[i]);
522fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
523fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
524fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  symbols_after_checkpoint_.clear();
525fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  files_after_checkpoint_.clear();
526fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  extensions_after_checkpoint_.clear();
527fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
528fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  STLDeleteContainerPointers(
529fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings_.begin() + strings_before_checkpoint_, strings_.end());
530fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  STLDeleteContainerPointers(
531fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    messages_.begin() + messages_before_checkpoint_, messages_.end());
532fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  STLDeleteContainerPointers(
533fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file_tables_.begin() + file_tables_before_checkpoint_, file_tables_.end());
534fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = allocations_before_checkpoint_; i < allocations_.size(); i++) {
535fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    operator delete(allocations_[i]);
536fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
537fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
538fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings_.resize(strings_before_checkpoint_);
539fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  messages_.resize(messages_before_checkpoint_);
540fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_tables_.resize(file_tables_before_checkpoint_);
541fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  allocations_.resize(allocations_before_checkpoint_);
542fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
543fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
544fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
545fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
546fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline Symbol DescriptorPool::Tables::FindSymbol(const string& key) const {
547fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Symbol* result = FindOrNull(symbols_by_name_, key.c_str());
548fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL) {
549fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return kNullSymbol;
550fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
551fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return *result;
552fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
553fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
554fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
555fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline Symbol FileDescriptorTables::FindNestedSymbol(
556fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& name) const {
557fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Symbol* result =
558fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FindOrNull(symbols_by_parent_, PointerStringPair(parent, name.c_str()));
559fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL) {
560fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return kNullSymbol;
561fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
562fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return *result;
563fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
564fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
565fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
566fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline Symbol FileDescriptorTables::FindNestedSymbolOfType(
567fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& name, const Symbol::Type type) const {
568fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = FindNestedSymbol(parent, name);
569fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.type != type) return kNullSymbol;
570fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
571fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
572fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
573fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleSymbol DescriptorPool::Tables::FindByNameHelper(
574fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const DescriptorPool* pool, const string& name) const {
575fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  MutexLockMaybe lock(pool->mutex_);
576fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = FindSymbol(name);
577fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
578fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.IsNull() && pool->underlay_ != NULL) {
579fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Symbol not found; check the underlay.
580fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result =
581fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      pool->underlay_->tables_->FindByNameHelper(pool->underlay_, name);
582fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
583fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
584fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.IsNull()) {
585fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Symbol still not found, so check fallback database.
586fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (pool->TryFindSymbolInFallbackDatabase(name)) {
587fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      result = FindSymbol(name);
588fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
589fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
590fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
591fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
592fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
593fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
594fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline const FileDescriptor* DescriptorPool::Tables::FindFile(
595fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& key) const {
596fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return FindPtrOrNull(files_by_name_, key.c_str());
597fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
598fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
599fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline const FieldDescriptor* FileDescriptorTables::FindFieldByNumber(
600fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* parent, int number) const {
601fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return FindPtrOrNull(fields_by_number_, make_pair(parent, number));
602fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
603fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
604fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline const FieldDescriptor* FileDescriptorTables::FindFieldByLowercaseName(
605fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& lowercase_name) const {
606fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return FindPtrOrNull(fields_by_lowercase_name_,
607fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       PointerStringPair(parent, lowercase_name.c_str()));
608fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
609fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
610fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline const FieldDescriptor* FileDescriptorTables::FindFieldByCamelcaseName(
611fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& camelcase_name) const {
612fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return FindPtrOrNull(fields_by_camelcase_name_,
613fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       PointerStringPair(parent, camelcase_name.c_str()));
614fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
615fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
616fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline const EnumValueDescriptor* FileDescriptorTables::FindEnumValueByNumber(
617fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const EnumDescriptor* parent, int number) const {
618fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return FindPtrOrNull(enum_values_by_number_, make_pair(parent, number));
619fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
620fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
621fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline const FieldDescriptor* DescriptorPool::Tables::FindExtension(
622fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* extendee, int number) {
623fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return FindPtrOrNull(extensions_, make_pair(extendee, number));
624fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
625fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
626fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline void DescriptorPool::Tables::FindAllExtensions(
627fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* extendee, vector<const FieldDescriptor*>* out) const {
628fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ExtensionsGroupedByDescriptorMap::const_iterator it =
629fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      extensions_.lower_bound(make_pair(extendee, 0));
630fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (; it != extensions_.end() && it->first.first == extendee; ++it) {
631fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    out->push_back(it->second);
632fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
633fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
634fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
635fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
636fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
637fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::Tables::AddSymbol(
638fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& full_name, Symbol symbol) {
639fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (InsertIfNotPresent(&symbols_by_name_, full_name.c_str(), symbol)) {
640fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    symbols_after_checkpoint_.push_back(full_name.c_str());
641fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return true;
642fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
643fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
644fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
645fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
646fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
647fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool FileDescriptorTables::AddAliasUnderParent(
648fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* parent, const string& name, Symbol symbol) {
649fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  PointerStringPair by_parent_key(parent, name.c_str());
650fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return InsertIfNotPresent(&symbols_by_parent_, by_parent_key, symbol);
651fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
652fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
653fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::Tables::AddFile(const FileDescriptor* file) {
654fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (InsertIfNotPresent(&files_by_name_, file->name().c_str(), file)) {
655fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    files_after_checkpoint_.push_back(file->name().c_str());
656fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return true;
657fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
658fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
659fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
660fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
661fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
662fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid FileDescriptorTables::AddFieldByStylizedNames(
663fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* field) {
664fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const void* parent;
665fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (field->is_extension()) {
666fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field->extension_scope() == NULL) {
667fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      parent = field->file();
668fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
669fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      parent = field->extension_scope();
670fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
671fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
672fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    parent = field->containing_type();
673fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
674fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
675fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  PointerStringPair lowercase_key(parent, field->lowercase_name().c_str());
676fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  InsertIfNotPresent(&fields_by_lowercase_name_, lowercase_key, field);
677fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
678fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  PointerStringPair camelcase_key(parent, field->camelcase_name().c_str());
679fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  InsertIfNotPresent(&fields_by_camelcase_name_, camelcase_key, field);
680fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
681fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
682fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool FileDescriptorTables::AddFieldByNumber(const FieldDescriptor* field) {
683fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DescriptorIntPair key(field->containing_type(), field->number());
684fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return InsertIfNotPresent(&fields_by_number_, key, field);
685fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
686fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
687fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool FileDescriptorTables::AddEnumValueByNumber(
688fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const EnumValueDescriptor* value) {
689fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  EnumIntPair key(value->type(), value->number());
690fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return InsertIfNotPresent(&enum_values_by_number_, key, value);
691fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
692fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
693fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::Tables::AddExtension(const FieldDescriptor* field) {
694fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DescriptorIntPair key(field->containing_type(), field->number());
695fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (InsertIfNotPresent(&extensions_, key, field)) {
696fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    extensions_after_checkpoint_.push_back(key);
697fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return true;
698fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
699fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
700fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
701fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
702fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
703fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
704fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
705fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletemplate<typename Type>
706fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleType* DescriptorPool::Tables::Allocate() {
707fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return reinterpret_cast<Type*>(AllocateBytes(sizeof(Type)));
708fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
709fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
710fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletemplate<typename Type>
711fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleType* DescriptorPool::Tables::AllocateArray(int count) {
712fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return reinterpret_cast<Type*>(AllocateBytes(sizeof(Type) * count));
713fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
714fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
715fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring* DescriptorPool::Tables::AllocateString(const string& value) {
716fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* result = new string(value);
717fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings_.push_back(result);
718fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
719fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
720fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
721fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletemplate<typename Type>
722fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleType* DescriptorPool::Tables::AllocateMessage(Type* dummy) {
723fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Type* result = new Type;
724fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  messages_.push_back(result);
725fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
726fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
727fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
728fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptorTables* DescriptorPool::Tables::AllocateFileTables() {
729fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorTables* result = new FileDescriptorTables;
730fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_tables_.push_back(result);
731fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
732fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
733fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
734fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid* DescriptorPool::Tables::AllocateBytes(int size) {
735fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(kenton):  Would it be worthwhile to implement this in some more
736fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // sophisticated way?  Probably not for the open source release, but for
737fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // internal use we could easily plug in one of our existing memory pool
738fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // allocators...
739fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (size == 0) return NULL;
740fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
741fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void* result = operator new(size);
742fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  allocations_.push_back(result);
743fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
744fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
745fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
746fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// ===================================================================
747fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DescriptorPool
748fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
749fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::ErrorCollector::~ErrorCollector() {}
750fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
751fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::DescriptorPool()
752fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  : mutex_(NULL),
753fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    fallback_database_(NULL),
754fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default_error_collector_(NULL),
755fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    underlay_(NULL),
756fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_(new Tables),
757fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enforce_dependencies_(true),
758fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    allow_unknown_(false) {}
759fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
760fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::DescriptorPool(DescriptorDatabase* fallback_database,
761fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               ErrorCollector* error_collector)
762fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  : mutex_(new Mutex),
763fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    fallback_database_(fallback_database),
764fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default_error_collector_(error_collector),
765fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    underlay_(NULL),
766fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_(new Tables),
767fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enforce_dependencies_(true),
768fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    allow_unknown_(false) {
769fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
770fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
771fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::DescriptorPool(const DescriptorPool* underlay)
772fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  : mutex_(NULL),
773fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    fallback_database_(NULL),
774fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default_error_collector_(NULL),
775fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    underlay_(underlay),
776fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_(new Tables),
777fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enforce_dependencies_(true),
778fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    allow_unknown_(false) {}
779fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
780fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool::~DescriptorPool() {
781fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (mutex_ != NULL) delete mutex_;
782fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
783fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
784fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DescriptorPool::BuildFile() defined later.
785fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DescriptorPool::BuildFileCollectingErrors() defined later.
786fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
787fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorPool::InternalDontEnforceDependencies() {
788fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  enforce_dependencies_ = false;
789fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
790fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
791fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::InternalIsFileLoaded(const string& filename) const {
792fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  MutexLockMaybe lock(mutex_);
793fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return tables_->FindFile(filename) != NULL;
794fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
795fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
796fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// generated_pool ====================================================
797fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
798fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace {
799fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
800d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville
801fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleEncodedDescriptorDatabase* generated_database_ = NULL;
802fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool* generated_pool_ = NULL;
803fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleGOOGLE_PROTOBUF_DECLARE_ONCE(generated_pool_init_);
804fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
805fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DeleteGeneratedPool() {
806fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  delete generated_database_;
807fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  generated_database_ = NULL;
808fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  delete generated_pool_;
809fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  generated_pool_ = NULL;
810fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
811fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
812fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid InitGeneratedPool() {
813fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  generated_database_ = new EncodedDescriptorDatabase;
814fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  generated_pool_ = new DescriptorPool(generated_database_);
815d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville
816fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  internal::OnShutdown(&DeleteGeneratedPool);
817fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
818fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
819fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleinline void InitGeneratedPoolOnce() {
820fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ::google::protobuf::GoogleOnceInit(&generated_pool_init_, &InitGeneratedPool);
821fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
822fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
823fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // anonymous namespace
824fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
825fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst DescriptorPool* DescriptorPool::generated_pool() {
826fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  InitGeneratedPoolOnce();
827fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return generated_pool_;
828fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
829fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
830fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorPool* DescriptorPool::internal_generated_pool() {
831fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  InitGeneratedPoolOnce();
832fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return generated_pool_;
833fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
834fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
835fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorPool::InternalAddGeneratedFile(
836fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const void* encoded_file_descriptor, int size) {
837fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // So, this function is called in the process of initializing the
838fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptors for generated proto classes.  Each generated .pb.cc file
839fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // has an internal procedure called AddDescriptors() which is called at
840fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // process startup, and that function calls this one in order to register
841fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the raw bytes of the FileDescriptorProto representing the file.
842fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //
843fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We do not actually construct the descriptor objects right away.  We just
844fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // hang on to the bytes until they are actually needed.  We actually construct
845fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the descriptor the first time one of the following things happens:
846fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // * Someone calls a method like descriptor(), GetDescriptor(), or
847fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   GetReflection() on the generated types, which requires returning the
848fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   descriptor or an object based on it.
849fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // * Someone looks up the descriptor in DescriptorPool::generated_pool().
850fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //
851fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Once one of these happens, the DescriptorPool actually parses the
852fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // FileDescriptorProto and generates a FileDescriptor (and all its children)
853fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // based on it.
854fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //
855fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note that FileDescriptorProto is itself a generated protocol message.
856fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Therefore, when we parse one, we have to be very careful to avoid using
857fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // any descriptor-based operations, since this might cause infinite recursion
858fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // or deadlock.
859fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  InitGeneratedPoolOnce();
860fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(generated_database_->Add(encoded_file_descriptor, size));
861fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
862fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
863fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
864fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Find*By* methods ==================================================
865fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
866fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// TODO(kenton):  There's a lot of repeated code here, but I'm not sure if
867fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//   there's any good way to factor it out.  Think about this some time when
868fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville//   there's nothing more important to do (read: never).
869fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
870fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorPool::FindFileByName(const string& name) const {
871fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  MutexLockMaybe lock(mutex_);
872fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* result = tables_->FindFile(name);
873fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result != NULL) return result;
874fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (underlay_ != NULL) {
875fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptor* result = underlay_->FindFileByName(name);
876fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (result != NULL) return result;
877fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
878fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (TryFindFileInFallbackDatabase(name)) {
879fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptor* result = tables_->FindFile(name);
880fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (result != NULL) return result;
881fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
882fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return NULL;
883fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
884fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
885fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorPool::FindFileContainingSymbol(
886fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& symbol_name) const {
887fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  MutexLockMaybe lock(mutex_);
888fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindSymbol(symbol_name);
889fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) return result.GetFile();
890fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (underlay_ != NULL) {
891fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptor* result =
892fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      underlay_->FindFileContainingSymbol(symbol_name);
893fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (result != NULL) return result;
894fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
895fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (TryFindSymbolInFallbackDatabase(symbol_name)) {
896fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Symbol result = tables_->FindSymbol(symbol_name);
897fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!result.IsNull()) return result.GetFile();
898fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
899fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return NULL;
900fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
901fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
902fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst Descriptor* DescriptorPool::FindMessageTypeByName(
903fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
904fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
905fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return (result.type == Symbol::MESSAGE) ? result.descriptor : NULL;
906fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
907fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
908fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor* DescriptorPool::FindFieldByName(
909fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
910fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
911fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.type == Symbol::FIELD &&
912fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      !result.field_descriptor->is_extension()) {
913fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.field_descriptor;
914fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
915fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
916fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
917fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
918fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
919fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor* DescriptorPool::FindExtensionByName(
920fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
921fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
922fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.type == Symbol::FIELD &&
923fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      result.field_descriptor->is_extension()) {
924fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.field_descriptor;
925fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
926fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
927fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
928fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
929fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
930fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumDescriptor* DescriptorPool::FindEnumTypeByName(
931fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
932fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
933fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return (result.type == Symbol::ENUM) ? result.enum_descriptor : NULL;
934fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
935fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
936fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumValueDescriptor* DescriptorPool::FindEnumValueByName(
937fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
938fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
939fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return (result.type == Symbol::ENUM_VALUE) ?
940fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result.enum_value_descriptor : NULL;
941fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
942fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
943fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst ServiceDescriptor* DescriptorPool::FindServiceByName(
944fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
945fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
946fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return (result.type == Symbol::SERVICE) ? result.service_descriptor : NULL;
947fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
948fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
949fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst MethodDescriptor* DescriptorPool::FindMethodByName(
950fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) const {
951fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindByNameHelper(this, name);
952fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return (result.type == Symbol::METHOD) ? result.method_descriptor : NULL;
953fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
954fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
955fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor* DescriptorPool::FindExtensionByNumber(
956fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* extendee, int number) const {
957fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  MutexLockMaybe lock(mutex_);
958fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result = tables_->FindExtension(extendee, number);
959fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result != NULL) {
960fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
961fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
962fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (underlay_ != NULL) {
963fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* result =
964fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      underlay_->FindExtensionByNumber(extendee, number);
965fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (result != NULL) return result;
966fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
967fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (TryFindExtensionInFallbackDatabase(extendee, number)) {
968fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* result = tables_->FindExtension(extendee, number);
969fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (result != NULL) {
970fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return result;
971fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
972fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
973fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return NULL;
974fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
975fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
976fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorPool::FindAllExtensions(
977fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* extendee, vector<const FieldDescriptor*>* out) const {
978fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  MutexLockMaybe lock(mutex_);
979fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
980fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Initialize tables_->extensions_ from the fallback database first
981fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // (but do this only once per descriptor).
982fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (fallback_database_ != NULL &&
983fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      tables_->extensions_loaded_from_db_.count(extendee) == 0) {
984fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    vector<int> numbers;
985fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (fallback_database_->FindAllExtensionNumbers(extendee->full_name(),
986fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                                    &numbers)) {
987fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      for (int i = 0; i < numbers.size(); ++i) {
988fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        int number = numbers[i];
989fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (tables_->FindExtension(extendee, number) == NULL) {
990fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          TryFindExtensionInFallbackDatabase(extendee, number);
991fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
992fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
993fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      tables_->extensions_loaded_from_db_.insert(extendee);
994fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
995fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
996fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
997fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  tables_->FindAllExtensions(extendee, out);
998fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (underlay_ != NULL) {
999fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    underlay_->FindAllExtensions(extendee, out);
1000fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1001fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1002fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1003fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
1004fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1005fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1006fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindFieldByNumber(int key) const {
1007fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result =
1008fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindFieldByNumber(this, key);
1009fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || result->is_extension()) {
1010fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1011fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1012fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1013fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1014fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1015fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1016fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1017fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindFieldByLowercaseName(const string& key) const {
1018fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result =
1019fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindFieldByLowercaseName(this, key);
1020fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || result->is_extension()) {
1021fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1022fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1023fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1024fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1025fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1026fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1027fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1028fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindFieldByCamelcaseName(const string& key) const {
1029fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result =
1030fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindFieldByCamelcaseName(this, key);
1031fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || result->is_extension()) {
1032fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1033fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1034fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1035fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1036fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1037fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1038fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1039fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindFieldByName(const string& key) const {
1040fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1041fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::FIELD);
1042fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull() && !result.field_descriptor->is_extension()) {
1043fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.field_descriptor;
1044fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1045fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1046fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1047fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1048fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1049fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1050fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindExtensionByName(const string& key) const {
1051fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1052fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::FIELD);
1053fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull() && result.field_descriptor->is_extension()) {
1054fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.field_descriptor;
1055fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1056fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1057fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1058fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1059fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1060fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1061fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindExtensionByLowercaseName(const string& key) const {
1062fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result =
1063fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindFieldByLowercaseName(this, key);
1064fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || !result->is_extension()) {
1065fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1066fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1067fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1068fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1069fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1070fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1071fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1072fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindExtensionByCamelcaseName(const string& key) const {
1073fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result =
1074fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindFieldByCamelcaseName(this, key);
1075fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || !result->is_extension()) {
1076fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1077fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1078fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1079fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1080fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1081fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1082fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst Descriptor*
1083fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindNestedTypeByName(const string& key) const {
1084fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1085fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::MESSAGE);
1086fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1087fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.descriptor;
1088fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1089fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1090fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1091fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1092fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1093fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumDescriptor*
1094fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindEnumTypeByName(const string& key) const {
1095fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1096fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::ENUM);
1097fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1098fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.enum_descriptor;
1099fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumValueDescriptor*
1105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptor::FindEnumValueByName(const string& key) const {
1106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::ENUM_VALUE);
1108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.enum_value_descriptor;
1110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumValueDescriptor*
1116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleEnumDescriptor::FindValueByName(const string& key) const {
1117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::ENUM_VALUE);
1119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.enum_value_descriptor;
1121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumValueDescriptor*
1127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleEnumDescriptor::FindValueByNumber(int key) const {
1128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return file()->tables_->FindEnumValueByNumber(this, key);
1129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst MethodDescriptor*
1132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleServiceDescriptor::FindMethodByName(const string& key) const {
1133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file()->tables_->FindNestedSymbolOfType(this, key, Symbol::METHOD);
1135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.method_descriptor;
1137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst Descriptor*
1143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindMessageTypeByName(const string& key) const {
1144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindNestedSymbolOfType(this, key, Symbol::MESSAGE);
1145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.descriptor;
1147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumDescriptor*
1153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindEnumTypeByName(const string& key) const {
1154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindNestedSymbolOfType(this, key, Symbol::ENUM);
1155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.enum_descriptor;
1157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1159fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1160fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1161fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1162fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst EnumValueDescriptor*
1163fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindEnumValueByName(const string& key) const {
1164fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result =
1165fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->FindNestedSymbolOfType(this, key, Symbol::ENUM_VALUE);
1166fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1167fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.enum_value_descriptor;
1168fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1169fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1170fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1171fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1172fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1173fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst ServiceDescriptor*
1174fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindServiceByName(const string& key) const {
1175fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindNestedSymbolOfType(this, key, Symbol::SERVICE);
1176fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull()) {
1177fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.service_descriptor;
1178fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1179fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1180fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1181fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1182fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1183fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1184fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindExtensionByName(const string& key) const {
1185fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = tables_->FindNestedSymbolOfType(this, key, Symbol::FIELD);
1186fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result.IsNull() && result.field_descriptor->is_extension()) {
1187fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result.field_descriptor;
1188fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1189fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1190fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1191fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1192fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1193fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1194fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindExtensionByLowercaseName(const string& key) const {
1195fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result = tables_->FindFieldByLowercaseName(this, key);
1196fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || !result->is_extension()) {
1197fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1198fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1199fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1200fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1201fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1202fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1203fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FieldDescriptor*
1204fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleFileDescriptor::FindExtensionByCamelcaseName(const string& key) const {
1205fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* result = tables_->FindFieldByCamelcaseName(this, key);
1206fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result == NULL || !result->is_extension()) {
1207fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
1208fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1209fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
1210fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1211fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1212fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1213fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool Descriptor::IsExtensionNumber(int number) const {
1214fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Linear search should be fine because we don't expect a message to have
1215fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // more than a couple extension ranges.
1216fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_range_count(); i++) {
1217fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (number >= extension_range(i)->start &&
1218fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        number <  extension_range(i)->end) {
1219fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return true;
1220fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1221fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1222fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return false;
1223fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1224fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1225fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
1226fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1227fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::TryFindFileInFallbackDatabase(const string& name) const {
1228fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (fallback_database_ == NULL) return false;
1229fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1230fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (tables_->known_bad_files_.count(name) > 0) return false;
1231fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1232fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorProto file_proto;
1233fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!fallback_database_->FindFileByName(name, &file_proto) ||
1234fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      BuildFileFromDatabase(file_proto) == NULL) {
1235fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->known_bad_files_.insert(name);
1236fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1237fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1238fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1239fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return true;
1240fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1241fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1242fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::TryFindSymbolInFallbackDatabase(const string& name) const {
1243fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (fallback_database_ == NULL) return false;
1244fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1245fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorProto file_proto;
1246fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!fallback_database_->FindFileContainingSymbol(name, &file_proto)) {
1247fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1248fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1249fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1250fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (tables_->FindFile(file_proto.name()) != NULL) {
1251fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // We've already loaded this file, and it apparently doesn't contain the
1252fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // symbol we're looking for.  Some DescriptorDatabases return false
1253fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // positives.
1254fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1255fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1256fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1257fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (BuildFileFromDatabase(file_proto) == NULL) {
1258fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1259fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1260fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1261fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return true;
1262fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1263fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1264fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorPool::TryFindExtensionInFallbackDatabase(
1265fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* containing_type, int field_number) const {
1266fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (fallback_database_ == NULL) return false;
1267fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1268fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorProto file_proto;
1269fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!fallback_database_->FindFileContainingExtension(
1270fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        containing_type->full_name(), field_number, &file_proto)) {
1271fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1272fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1273fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1274fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (tables_->FindFile(file_proto.name()) != NULL) {
1275fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // We've already loaded this file, and it apparently doesn't contain the
1276fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // extension we're looking for.  Some DescriptorDatabases return false
1277fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // positives.
1278fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1279fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1280fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1281fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (BuildFileFromDatabase(file_proto) == NULL) {
1282fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
1283fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1284fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1285fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return true;
1286fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1287fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1288fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// ===================================================================
1289fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1290fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring FieldDescriptor::DefaultValueAsString(bool quote_string_type) const {
1291fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(has_default_value()) << "No default value";
1292fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (cpp_type()) {
1293fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_INT32:
1294fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return SimpleItoa(default_value_int32());
1295fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1296fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_INT64:
1297fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return SimpleItoa(default_value_int64());
1298fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1299fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_UINT32:
1300fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return SimpleItoa(default_value_uint32());
1301fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1302fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_UINT64:
1303fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return SimpleItoa(default_value_uint64());
1304fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1305fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_FLOAT:
1306fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return SimpleFtoa(default_value_float());
1307fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1308fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_DOUBLE:
1309fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return SimpleDtoa(default_value_double());
1310fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1311fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_BOOL:
1312fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return default_value_bool() ? "true" : "false";
1313fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1314fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_STRING:
1315fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (quote_string_type) {
1316fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return "\"" + CEscape(default_value_string()) + "\"";
1317fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
1318fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (type() == TYPE_BYTES) {
1319fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return CEscape(default_value_string());
1320fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
1321fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return default_value_string();
1322fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
1323fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
1324fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1325fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_ENUM:
1326fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return default_value_enum()->name();
1327fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1328fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case CPPTYPE_MESSAGE:
1329fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(DFATAL) << "Messages can't have default values!";
1330fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1331fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1332fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_LOG(FATAL) << "Can't get here: failed to get default value as string";
1333fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return "";
1334fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1335fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1336fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// CopyTo methods ====================================================
1337fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1338fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid FileDescriptor::CopyTo(FileDescriptorProto* proto) const {
1339fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1340fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!package().empty()) proto->set_package(package());
1341fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1342fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < dependency_count(); i++) {
1343fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->add_dependency(dependency(i)->name());
1344fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1345fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1346fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < message_type_count(); i++) {
1347fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    message_type(i)->CopyTo(proto->add_message_type());
1348fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1349fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < enum_type_count(); i++) {
1350fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enum_type(i)->CopyTo(proto->add_enum_type());
1351fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1352fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < service_count(); i++) {
1353fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    service(i)->CopyTo(proto->add_service());
1354fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1355fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_count(); i++) {
1356fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    extension(i)->CopyTo(proto->add_extension());
1357fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1358fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1359fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &FileOptions::default_instance()) {
1360fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1361fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1362fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1363fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1364fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid Descriptor::CopyTo(DescriptorProto* proto) const {
1365fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1366fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1367fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < field_count(); i++) {
1368fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    field(i)->CopyTo(proto->add_field());
1369fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1370fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < nested_type_count(); i++) {
1371fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    nested_type(i)->CopyTo(proto->add_nested_type());
1372fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1373fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < enum_type_count(); i++) {
1374fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enum_type(i)->CopyTo(proto->add_enum_type());
1375fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1376fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_range_count(); i++) {
1377fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorProto::ExtensionRange* range = proto->add_extension_range();
1378fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    range->set_start(extension_range(i)->start);
1379fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    range->set_end(extension_range(i)->end);
1380fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1381fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_count(); i++) {
1382fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    extension(i)->CopyTo(proto->add_extension());
1383fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1384fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1385fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &MessageOptions::default_instance()) {
1386fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1387fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1388fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1389fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1390fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid FieldDescriptor::CopyTo(FieldDescriptorProto* proto) const {
1391fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1392fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_number(number());
1393fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1394fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Some compilers do not allow static_cast directly between two enum types,
1395fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // so we must cast to int first.
1396fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_label(static_cast<FieldDescriptorProto::Label>(
1397fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     implicit_cast<int>(label())));
1398fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_type(static_cast<FieldDescriptorProto::Type>(
1399fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    implicit_cast<int>(type())));
1400fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1401fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (is_extension()) {
1402fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!containing_type()->is_unqualified_placeholder_) {
1403fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      proto->set_extendee(".");
1404fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1405fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_extendee()->append(containing_type()->full_name());
1406fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1407fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1408fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (cpp_type() == CPPTYPE_MESSAGE) {
1409fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (message_type()->is_placeholder_) {
1410fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // We don't actually know if the type is a message type.  It could be
1411fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // an enum.
1412fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      proto->clear_type();
1413fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1414fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1415fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!message_type()->is_unqualified_placeholder_) {
1416fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      proto->set_type_name(".");
1417fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1418fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_type_name()->append(message_type()->full_name());
1419fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else if (cpp_type() == CPPTYPE_ENUM) {
1420fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!enum_type()->is_unqualified_placeholder_) {
1421fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      proto->set_type_name(".");
1422fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1423fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_type_name()->append(enum_type()->full_name());
1424fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1425fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1426fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (has_default_value()) {
1427fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->set_default_value(DefaultValueAsString(false));
1428fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1429fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1430fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &FieldOptions::default_instance()) {
1431fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1432fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1433fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1434fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1435fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid EnumDescriptor::CopyTo(EnumDescriptorProto* proto) const {
1436fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1437fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1438fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < value_count(); i++) {
1439fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    value(i)->CopyTo(proto->add_value());
1440fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1441fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1442fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &EnumOptions::default_instance()) {
1443fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1444fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1445fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1446fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1447fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid EnumValueDescriptor::CopyTo(EnumValueDescriptorProto* proto) const {
1448fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1449fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_number(number());
1450fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1451fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &EnumValueOptions::default_instance()) {
1452fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1453fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1454fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1455fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1456fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid ServiceDescriptor::CopyTo(ServiceDescriptorProto* proto) const {
1457fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1458fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1459fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < method_count(); i++) {
1460fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    method(i)->CopyTo(proto->add_method());
1461fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1462fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1463fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &ServiceOptions::default_instance()) {
1464fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1465fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1466fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1467fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1468fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid MethodDescriptor::CopyTo(MethodDescriptorProto* proto) const {
1469fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->set_name(name());
1470fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1471fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!input_type()->is_unqualified_placeholder_) {
1472fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->set_input_type(".");
1473fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1474fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->mutable_input_type()->append(input_type()->full_name());
1475fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1476fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!output_type()->is_unqualified_placeholder_) {
1477fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->set_output_type(".");
1478fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1479fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  proto->mutable_output_type()->append(output_type()->full_name());
1480fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1481fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (&options() != &MethodOptions::default_instance()) {
1482fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    proto->mutable_options()->CopyFrom(options());
1483fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1484fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1485fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1486fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DebugString methods ===============================================
1487fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1488fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace {
1489fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1490fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Used by each of the option formatters.
1491fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool RetrieveOptions(const Message &options, vector<string> *option_entries) {
1492fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  option_entries->clear();
1493fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Reflection* reflection = options.GetReflection();
1494fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<const FieldDescriptor*> fields;
1495fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  reflection->ListFields(options, &fields);
1496fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < fields.size(); i++) {
1497fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Doesn't make sense to have message type fields here
1498fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (fields[i]->cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE) {
1499fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      continue;
1500fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1501fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    int count = 1;
1502fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool repeated = false;
1503fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (fields[i]->is_repeated()) {
1504fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      count = reflection->FieldSize(options, fields[i]);
1505fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      repeated = true;
1506fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1507fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int j = 0; j < count; j++) {
1508fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      string fieldval;
1509fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      TextFormat::PrintFieldValueToString(options, fields[i],
1510fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                          repeated ? count : -1, &fieldval);
1511fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      option_entries->push_back(fields[i]->name() + " = " + fieldval);
1512fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1513fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1514fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return !option_entries->empty();
1515fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1516fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1517fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Formats options that all appear together in brackets. Does not include
1518fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// brackets.
1519fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool FormatBracketedOptions(const Message &options, string *output) {
1520fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<string> all_options;
1521fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (RetrieveOptions(options, &all_options)) {
1522fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    output->append(JoinStrings(all_options, ", "));
1523fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1524fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return !all_options.empty();
1525fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1526fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1527fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Formats options one per line
1528fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool FormatLineOptions(int depth, const Message &options, string *output) {
1529fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string prefix(depth * 2, ' ');
1530fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<string> all_options;
1531fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (RetrieveOptions(options, &all_options)) {
1532fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int i = 0; i < all_options.size(); i++) {
1533fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      strings::SubstituteAndAppend(output, "$0option $1;\n",
1534fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   prefix, all_options[i]);
1535fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1536fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1537fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return !all_options.empty();
1538fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1539fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1540fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // anonymous namespace
1541fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1542fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring FileDescriptor::DebugString() const {
1543fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents = "syntax = \"proto2\";\n\n";
1544fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1545fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < dependency_count(); i++) {
1546fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(&contents, "import \"$0\";\n",
1547fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 dependency(i)->name());
1548fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1549fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1550fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!package().empty()) {
1551fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(&contents, "package $0;\n\n", package());
1552fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1553fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1554fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (FormatLineOptions(0, options(), &contents)) {
1555fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents.append("\n");  // add some space if we had options
1556fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1557fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1558fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < enum_type_count(); i++) {
1559fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enum_type(i)->DebugString(0, &contents);
1560fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents.append("\n");
1561fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1562fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1563fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find all the 'group' type extensions; we will not output their nested
1564fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // definitions (those will be done with their group field descriptor).
1565fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  set<const Descriptor*> groups;
1566fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_count(); i++) {
1567fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (extension(i)->type() == FieldDescriptor::TYPE_GROUP) {
1568fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      groups.insert(extension(i)->message_type());
1569fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1570fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1571fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1572fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < message_type_count(); i++) {
1573fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (groups.count(message_type(i)) == 0) {
1574fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      strings::SubstituteAndAppend(&contents, "message $0",
1575fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   message_type(i)->name());
1576fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      message_type(i)->DebugString(0, &contents);
1577fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      contents.append("\n");
1578fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1579fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1580fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1581fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < service_count(); i++) {
1582fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    service(i)->DebugString(&contents);
1583fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents.append("\n");
1584fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1585fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1586fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Descriptor* containing_type = NULL;
1587fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_count(); i++) {
1588fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (extension(i)->containing_type() != containing_type) {
1589fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (i > 0) contents.append("}\n\n");
1590fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      containing_type = extension(i)->containing_type();
1591fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      strings::SubstituteAndAppend(&contents, "extend .$0 {\n",
1592fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   containing_type->full_name());
1593fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1594fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    extension(i)->DebugString(1, &contents);
1595fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1596fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (extension_count() > 0) contents.append("}\n\n");
1597fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1598fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1599fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1600fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1601fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring Descriptor::DebugString() const {
1602fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents;
1603fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(&contents, "message $0", name());
1604fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DebugString(0, &contents);
1605fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1606fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1607fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1608fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid Descriptor::DebugString(int depth, string *contents) const {
1609fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string prefix(depth * 2, ' ');
1610fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ++depth;
1611fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  contents->append(" {\n");
1612fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1613fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FormatLineOptions(depth, options(), contents);
1614fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1615fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find all the 'group' types for fields and extensions; we will not output
1616fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // their nested definitions (those will be done with their group field
1617fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptor).
1618fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  set<const Descriptor*> groups;
1619fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < field_count(); i++) {
1620fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field(i)->type() == FieldDescriptor::TYPE_GROUP) {
1621fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      groups.insert(field(i)->message_type());
1622fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1623fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1624fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_count(); i++) {
1625fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (extension(i)->type() == FieldDescriptor::TYPE_GROUP) {
1626fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      groups.insert(extension(i)->message_type());
1627fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1628fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1629fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1630fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < nested_type_count(); i++) {
1631fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (groups.count(nested_type(i)) == 0) {
1632fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      strings::SubstituteAndAppend(contents, "$0  message $1",
1633fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   prefix, nested_type(i)->name());
1634fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      nested_type(i)->DebugString(depth, contents);
1635fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1636fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1637fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < enum_type_count(); i++) {
1638fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enum_type(i)->DebugString(depth, contents);
1639fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1640fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < field_count(); i++) {
1641fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    field(i)->DebugString(depth, contents);
1642fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1643fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1644fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_range_count(); i++) {
1645fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(contents, "$0  extensions $1 to $2;\n",
1646fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 prefix,
1647fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 extension_range(i)->start,
1648fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 extension_range(i)->end - 1);
1649fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1650fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1651fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Group extensions by what they extend, so they can be printed out together.
1652fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Descriptor* containing_type = NULL;
1653fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < extension_count(); i++) {
1654fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (extension(i)->containing_type() != containing_type) {
1655fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (i > 0) strings::SubstituteAndAppend(contents, "$0  }\n", prefix);
1656fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      containing_type = extension(i)->containing_type();
1657fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      strings::SubstituteAndAppend(contents, "$0  extend .$1 {\n",
1658fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   prefix, containing_type->full_name());
1659fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
1660fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    extension(i)->DebugString(depth + 1, contents);
1661fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1662fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (extension_count() > 0)
1663fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(contents, "$0  }\n", prefix);
1664fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1665fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "$0}\n", prefix);
1666fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1667fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1668fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring FieldDescriptor::DebugString() const {
1669fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents;
1670fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int depth = 0;
1671fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (is_extension()) {
1672fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(&contents, "extend .$0 {\n",
1673fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 containing_type()->full_name());
1674fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    depth = 1;
1675fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1676fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DebugString(depth, &contents);
1677fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (is_extension()) {
1678fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville     contents.append("}\n");
1679fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1680fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1681fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1682fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1683fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid FieldDescriptor::DebugString(int depth, string *contents) const {
1684fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string prefix(depth * 2, ' ');
1685fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string field_type;
1686fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (type()) {
1687fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case TYPE_MESSAGE:
1688fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field_type = "." + message_type()->full_name();
1689fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1690fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case TYPE_ENUM:
1691fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field_type = "." + enum_type()->full_name();
1692fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
1693fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default:
1694fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field_type = kTypeToName[type()];
1695fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1696fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1697fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "$0$1 $2 $3 = $4",
1698fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               prefix,
1699fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               kLabelToName[label()],
1700fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               field_type,
1701fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               type() == TYPE_GROUP ? message_type()->name() :
1702fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                                      name(),
1703fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               number());
1704fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1705fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool bracketed = false;
1706fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (has_default_value()) {
1707fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bracketed = true;
1708fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(contents, " [default = $0",
1709fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 DefaultValueAsString(true));
1710fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1711fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1712fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string formatted_options;
1713fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (FormatBracketedOptions(options(), &formatted_options)) {
1714fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents->append(bracketed ? ", " : " [");
1715fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bracketed = true;
1716fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents->append(formatted_options);
1717fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1718fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1719fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (bracketed) {
1720fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents->append("]");
1721fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1722fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1723fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (type() == TYPE_GROUP) {
1724fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    message_type()->DebugString(depth, contents);
1725fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1726fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents->append(";\n");
1727fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1728fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1729fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1730fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring EnumDescriptor::DebugString() const {
1731fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents;
1732fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DebugString(0, &contents);
1733fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1734fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1735fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1736fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid EnumDescriptor::DebugString(int depth, string *contents) const {
1737fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string prefix(depth * 2, ' ');
1738fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ++depth;
1739fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "$0enum $1 {\n",
1740fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               prefix, name());
1741fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1742fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FormatLineOptions(depth, options(), contents);
1743fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1744fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < value_count(); i++) {
1745fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    value(i)->DebugString(depth, contents);
1746fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1747fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "$0}\n", prefix);
1748fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1749fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1750fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring EnumValueDescriptor::DebugString() const {
1751fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents;
1752fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DebugString(0, &contents);
1753fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1754fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1755fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1756fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid EnumValueDescriptor::DebugString(int depth, string *contents) const {
1757fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string prefix(depth * 2, ' ');
1758fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "$0$1 = $2",
1759fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               prefix, name(), number());
1760fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1761fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string formatted_options;
1762fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (FormatBracketedOptions(options(), &formatted_options)) {
1763fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(contents, " [$0]", formatted_options);
1764fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1765fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  contents->append(";\n");
1766fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1767fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1768fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring ServiceDescriptor::DebugString() const {
1769fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents;
1770fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DebugString(&contents);
1771fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1772fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1773fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1774fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid ServiceDescriptor::DebugString(string *contents) const {
1775fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "service $0 {\n", name());
1776fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1777fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FormatLineOptions(1, options(), contents);
1778fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1779fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < method_count(); i++) {
1780fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    method(i)->DebugString(1, contents);
1781fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1782fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1783fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  contents->append("}\n");
1784fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1785fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1786fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestring MethodDescriptor::DebugString() const {
1787fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string contents;
1788fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DebugString(0, &contents);
1789fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return contents;
1790fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1791fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1792fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid MethodDescriptor::DebugString(int depth, string *contents) const {
1793fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string prefix(depth * 2, ' ');
1794fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ++depth;
1795fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  strings::SubstituteAndAppend(contents, "$0rpc $1(.$2) returns (.$3)",
1796fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               prefix, name(),
1797fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               input_type()->full_name(),
1798fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               output_type()->full_name());
1799fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1800fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string formatted_options;
1801fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (FormatLineOptions(depth, options(), &formatted_options)) {
1802fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    strings::SubstituteAndAppend(contents, " {\n$0$1}\n",
1803fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 formatted_options, prefix);
1804fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
1805fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    contents->append(";\n");
1806fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1807fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
1808fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// ===================================================================
1809fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1810fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace {
1811fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1812fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Represents an options message to interpret. Extension names in the option
1813fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// name are respolved relative to name_scope. element_name and orig_opt are
1814fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// used only for error reporting (since the parser records locations against
1815fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// pointers in the original options, not the mutable copy). The Message must be
1816fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// one of the Options messages in descriptor.proto.
1817fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestruct OptionsToInterpret {
1818fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  OptionsToInterpret(const string& ns,
1819fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     const string& el,
1820fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     const Message* orig_opt,
1821fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     Message* opt)
1822fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      : name_scope(ns),
1823fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        element_name(el),
1824fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        original_options(orig_opt),
1825fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        options(opt) {
1826fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1827fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string name_scope;
1828fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string element_name;
1829fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Message* original_options;
1830fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Message* options;
1831fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
1832fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1833fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // namespace
1834fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1835fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleclass DescriptorBuilder {
1836fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public:
1837fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DescriptorBuilder(const DescriptorPool* pool,
1838fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    DescriptorPool::Tables* tables,
1839fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    DescriptorPool::ErrorCollector* error_collector);
1840fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ~DescriptorBuilder();
1841fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1842fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* BuildFile(const FileDescriptorProto& proto);
1843fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1844fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private:
1845fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  friend class OptionInterpreter;
1846fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1847fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const DescriptorPool* pool_;
1848fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DescriptorPool::Tables* tables_;  // for convenience
1849fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  DescriptorPool::ErrorCollector* error_collector_;
1850fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1851fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // As we build descriptors we store copies of the options messages in
1852fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // them. We put pointers to those copies in this vector, as we build, so we
1853fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // can later (after cross-linking) interpret those options.
1854fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<OptionsToInterpret> options_to_interpret_;
1855fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1856fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool had_errors_;
1857fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string filename_;
1858fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptor* file_;
1859fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptorTables* file_tables_;
1860fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1861fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // If LookupSymbol() finds a symbol that is in a file which is not a declared
1862fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // dependency of this file, it will fail, but will set
1863fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // possible_undeclared_dependency_ to point at that file.  This is only used
1864fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // by AddNotDefinedError() to report a more useful error message.
1865fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // possible_undeclared_dependency_name_ is the name of the symbol that was
1866fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // actually found in possible_undeclared_dependency_, which may be a parent
1867fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // of the symbol actually looked for.
1868fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* possible_undeclared_dependency_;
1869fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string possible_undeclared_dependency_name_;
1870fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1871fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void AddError(const string& element_name,
1872fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                const Message& descriptor,
1873fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                DescriptorPool::ErrorCollector::ErrorLocation location,
1874fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                const string& error);
1875fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1876fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Adds an error indicating that undefined_symbol was not defined.  Must
1877fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // only be called after LookupSymbol() fails.
1878fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void AddNotDefinedError(
1879fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& element_name,
1880fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Message& descriptor,
1881fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorPool::ErrorCollector::ErrorLocation location,
1882fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& undefined_symbol);
1883fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1884fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Silly helper which determines if the given file is in the given package.
1885fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // I.e., either file->package() == package_name or file->package() is a
1886fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // nested package within package_name.
1887fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool IsInPackage(const FileDescriptor* file, const string& package_name);
1888fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1889fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Like tables_->FindSymbol(), but additionally:
1890fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // - Search the pool's underlay if not found in tables_.
1891fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // - Insure that the resulting Symbol is from one of the file's declared
1892fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   dependencies.
1893fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol FindSymbol(const string& name);
1894fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1895fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Like FindSymbol() but does not require that the symbol is in one of the
1896fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // file's declared dependencies.
1897fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol FindSymbolNotEnforcingDeps(const string& name);
1898fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1899fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Like FindSymbol(), but looks up the name relative to some other symbol
1900fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // name.  This first searches siblings of relative_to, then siblings of its
1901fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // parents, etc.  For example, LookupSymbol("foo.bar", "baz.qux.corge") makes
1902fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the following calls, returning the first non-null result:
1903fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // FindSymbol("baz.qux.foo.bar"), FindSymbol("baz.foo.bar"),
1904fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // FindSymbol("foo.bar").  If AllowUnknownDependencies() has been called
1905fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // on the DescriptorPool, this will generate a placeholder type if
1906fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the name is not found (unless the name itself is malformed).  The
1907fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // placeholder_type parameter indicates what kind of placeholder should be
1908fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // constructed in this case.  The resolve_mode parameter determines whether
1909fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // any symbol is returned, or only symbols that are types.  Note, however,
1910fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // that LookupSymbol may still return a non-type symbol in LOOKUP_TYPES mode,
1911fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // if it believes that's all it could refer to.  The caller should always
1912fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // check that it receives the type of symbol it was expecting.
1913fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  enum PlaceholderType {
1914fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    PLACEHOLDER_MESSAGE,
1915fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    PLACEHOLDER_ENUM,
1916fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    PLACEHOLDER_EXTENDABLE_MESSAGE
1917fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  };
1918fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  enum ResolveMode {
1919fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    LOOKUP_ALL, LOOKUP_TYPES
1920fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  };
1921fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol LookupSymbol(const string& name, const string& relative_to,
1922fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      PlaceholderType placeholder_type = PLACEHOLDER_MESSAGE,
1923fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      ResolveMode resolve_mode = LOOKUP_ALL);
1924fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1925fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Like LookupSymbol() but will not return a placeholder even if
1926fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // AllowUnknownDependencies() has been used.
1927fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol LookupSymbolNoPlaceholder(const string& name,
1928fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   const string& relative_to,
1929fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   ResolveMode resolve_mode = LOOKUP_ALL);
1930fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1931fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Creates a placeholder type suitable for return from LookupSymbol().  May
1932fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // return kNullSymbol if the name is not a valid type name.
1933fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol NewPlaceholder(const string& name, PlaceholderType placeholder_type);
1934fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1935fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Creates a placeholder file.  Never returns NULL.  This is used when an
1936fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // import is not found and AllowUnknownDependencies() is enabled.
1937fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* NewPlaceholderFile(const string& name);
1938fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1939fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Calls tables_->AddSymbol() and records an error if it fails.  Returns
1940fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // true if successful or false if failed, though most callers can ignore
1941fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the return value since an error has already been recorded.
1942fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool AddSymbol(const string& full_name,
1943fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 const void* parent, const string& name,
1944fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 const Message& proto, Symbol symbol);
1945fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1946fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Like AddSymbol(), but succeeds if the symbol is already defined as long
1947fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // as the existing definition is also a package (because it's OK to define
1948fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the same package in two different files).  Also adds all parents of the
1949fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // packgae to the symbol table (e.g. AddPackage("foo.bar", ...) will add
1950fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // "foo.bar" and "foo" to the table).
1951fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void AddPackage(const string& name, const Message& proto,
1952fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  const FileDescriptor* file);
1953fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1954fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Checks that the symbol name contains only alphanumeric characters and
1955fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // underscores.  Records an error otherwise.
1956fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateSymbolName(const string& name, const string& full_name,
1957fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                          const Message& proto);
1958fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1959fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Like ValidateSymbolName(), but the name is allowed to contain periods and
1960fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // an error is indicated by returning false (not recording the error).
1961fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool ValidateQualifiedName(const string& name);
1962fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1963fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Used by BUILD_ARRAY macro (below) to avoid having to have the type
1964fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // specified as a macro parameter.
1965fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  template <typename Type>
1966fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  inline void AllocateArray(int size, Type** output) {
1967fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    *output = tables_->AllocateArray<Type>(size);
1968fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
1969fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1970fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Allocates a copy of orig_options in tables_ and stores it in the
1971fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptor. Remembers its uninterpreted options, to be interpreted
1972fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // later. DescriptorT must be one of the Descriptor messages from
1973fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptor.proto.
1974fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  template<class DescriptorT> void AllocateOptions(
1975fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const typename DescriptorT::OptionsType& orig_options,
1976fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      DescriptorT* descriptor);
1977fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Specialization for FileOptions.
1978fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void AllocateOptions(const FileOptions& orig_options,
1979fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       FileDescriptor* descriptor);
1980fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1981fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Implementation for AllocateOptions(). Don't call this directly.
1982fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  template<class DescriptorT> void AllocateOptionsImpl(
1983fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const string& name_scope,
1984fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const string& element_name,
1985fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const typename DescriptorT::OptionsType& orig_options,
1986fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      DescriptorT* descriptor);
1987fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
1988fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // These methods all have the same signature for the sake of the BUILD_ARRAY
1989fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // macro, below.
1990fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildMessage(const DescriptorProto& proto,
1991fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    const Descriptor* parent,
1992fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    Descriptor* result);
1993fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildFieldOrExtension(const FieldDescriptorProto& proto,
1994fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             const Descriptor* parent,
1995fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             FieldDescriptor* result,
1996fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             bool is_extension);
1997fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildField(const FieldDescriptorProto& proto,
1998fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  const Descriptor* parent,
1999fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  FieldDescriptor* result) {
2000fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    BuildFieldOrExtension(proto, parent, result, false);
2001fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2002fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildExtension(const FieldDescriptorProto& proto,
2003fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      const Descriptor* parent,
2004fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      FieldDescriptor* result) {
2005fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    BuildFieldOrExtension(proto, parent, result, true);
2006fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2007fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildExtensionRange(const DescriptorProto::ExtensionRange& proto,
2008fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           const Descriptor* parent,
2009fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           Descriptor::ExtensionRange* result);
2010fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildEnum(const EnumDescriptorProto& proto,
2011fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 const Descriptor* parent,
2012fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 EnumDescriptor* result);
2013fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildEnumValue(const EnumValueDescriptorProto& proto,
2014fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      const EnumDescriptor* parent,
2015fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      EnumValueDescriptor* result);
2016fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildService(const ServiceDescriptorProto& proto,
2017fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    const void* dummy,
2018fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    ServiceDescriptor* result);
2019fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void BuildMethod(const MethodDescriptorProto& proto,
2020fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   const ServiceDescriptor* parent,
2021fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   MethodDescriptor* result);
2022fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2023fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Must be run only after building.
2024fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //
2025fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // NOTE: Options will not be available during cross-linking, as they
2026fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // have not yet been interpreted. Defer any handling of options to the
2027fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Validate*Options methods.
2028fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkFile(FileDescriptor* file, const FileDescriptorProto& proto);
2029fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkMessage(Descriptor* message, const DescriptorProto& proto);
2030fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkField(FieldDescriptor* field,
2031fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      const FieldDescriptorProto& proto);
2032fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkEnum(EnumDescriptor* enum_type,
2033fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     const EnumDescriptorProto& proto);
2034fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkEnumValue(EnumValueDescriptor* enum_value,
2035fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                          const EnumValueDescriptorProto& proto);
2036fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkService(ServiceDescriptor* service,
2037fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                        const ServiceDescriptorProto& proto);
2038fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void CrossLinkMethod(MethodDescriptor* method,
2039fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       const MethodDescriptorProto& proto);
2040fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2041fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Must be run only after cross-linking.
2042fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void InterpretOptions();
2043fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2044fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // A helper class for interpreting options.
2045fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  class OptionInterpreter {
2046fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville   public:
2047fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Creates an interpreter that operates in the context of the pool of the
2048fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // specified builder, which must not be NULL. We don't take ownership of the
2049fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // builder.
2050fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    explicit OptionInterpreter(DescriptorBuilder* builder);
2051fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2052fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ~OptionInterpreter();
2053fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2054fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Interprets the uninterpreted options in the specified Options message.
2055fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // On error, calls AddError() on the underlying builder and returns false.
2056fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Otherwise returns true.
2057fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool InterpretOptions(OptionsToInterpret* options_to_interpret);
2058fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2059fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville   private:
2060fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Interprets uninterpreted_option_ on the specified message, which
2061fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // must be the mutable copy of the original options message to which
2062fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // uninterpreted_option_ belongs.
2063fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool InterpretSingleOption(Message* options);
2064fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2065fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Adds the uninterpreted_option to the given options message verbatim.
2066fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Used when AllowUnknownDependencies() is in effect and we can't find
2067fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // the option's definition.
2068fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    void AddWithoutInterpreting(const UninterpretedOption& uninterpreted_option,
2069fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                Message* options);
2070fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2071fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // A recursive helper function that drills into the intermediate fields
2072fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // in unknown_fields to check if field innermost_field is set on the
2073fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // innermost message. Returns false and sets an error if so.
2074fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool ExamineIfOptionIsSet(
2075fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        vector<const FieldDescriptor*>::const_iterator intermediate_fields_iter,
2076fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        vector<const FieldDescriptor*>::const_iterator intermediate_fields_end,
2077fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        const FieldDescriptor* innermost_field, const string& debug_msg_name,
2078fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        const UnknownFieldSet& unknown_fields);
2079fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2080fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Validates the value for the option field of the currently interpreted
2081fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // option and then sets it on the unknown_field.
2082fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool SetOptionValue(const FieldDescriptor* option_field,
2083fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                        UnknownFieldSet* unknown_fields);
2084fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2085fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Convenience functions to set an int field the right way, depending on
2086fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // its wire type (a single int CppType can represent multiple wire types).
2087fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    void SetInt32(int number, int32 value, FieldDescriptor::Type type,
2088fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  UnknownFieldSet* unknown_fields);
2089fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    void SetInt64(int number, int64 value, FieldDescriptor::Type type,
2090fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  UnknownFieldSet* unknown_fields);
2091fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    void SetUInt32(int number, uint32 value, FieldDescriptor::Type type,
2092fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   UnknownFieldSet* unknown_fields);
2093fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    void SetUInt64(int number, uint64 value, FieldDescriptor::Type type,
2094fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   UnknownFieldSet* unknown_fields);
2095fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2096fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // A helper function that adds an error at the specified location of the
2097fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // option we're currently interpreting, and returns false.
2098fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool AddOptionError(DescriptorPool::ErrorCollector::ErrorLocation location,
2099fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                        const string& msg) {
2100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      builder_->AddError(options_to_interpret_->element_name,
2101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         *uninterpreted_option_, location, msg);
2102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return false;
2103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // A helper function that adds an error at the location of the option name
2106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // and returns false.
2107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool AddNameError(const string& msg) {
2108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return AddOptionError(DescriptorPool::ErrorCollector::OPTION_NAME, msg);
2109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // A helper function that adds an error at the location of the option name
2112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // and returns false.
2113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool AddValueError(const string& msg) {
2114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return AddOptionError(DescriptorPool::ErrorCollector::OPTION_VALUE, msg);
2115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // We interpret against this builder's pool. Is never NULL. We don't own
2118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // this pointer.
2119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorBuilder* builder_;
2120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // The options we're currently interpreting, or NULL if we're not in a call
2122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // to InterpretOptions.
2123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const OptionsToInterpret* options_to_interpret_;
2124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // The option we're currently interpreting within options_to_interpret_, or
2126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // NULL if we're not in a call to InterpretOptions(). This points to a
2127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // submessage of the original option, not the mutable copy. Therefore we
2128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // can use it to find locations recorded by the parser.
2129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const UninterpretedOption* uninterpreted_option_;
2130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(OptionInterpreter);
2132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  };
2133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Work-around for broken compilers:  According to the C++ standard,
2135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // OptionInterpreter should have access to the private members of any class
2136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // which has declared DescriptorBuilder as a friend.  Unfortunately some old
2137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // versions of GCC and other compilers do not implement this correctly.  So,
2138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // we have to have these intermediate methods to provide access.  We also
2139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // redundantly declare OptionInterpreter a friend just to make things extra
2140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // clear for these bad compilers.
2141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  friend class OptionInterpreter;
2142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static inline bool get_allow_unknown(const DescriptorPool* pool) {
2143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return pool->allow_unknown_;
2144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  static inline bool get_is_placeholder(const Descriptor* descriptor) {
2146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return descriptor->is_placeholder_;
2147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Must be run only after options have been interpreted.
2150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //
2151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // NOTE: Validation code must only reference the options in the mutable
2152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptors, which are the ones that have been interpreted. The const
2153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // proto references are passed in only so they can be provided to calls to
2154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // AddError(). Do not look at their options, which have not been interpreted.
2155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateFileOptions(FileDescriptor* file,
2156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           const FileDescriptorProto& proto);
2157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateMessageOptions(Descriptor* message,
2158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                              const DescriptorProto& proto);
2159fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateFieldOptions(FieldDescriptor* field,
2160fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                            const FieldDescriptorProto& proto);
2161fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateEnumOptions(EnumDescriptor* enm,
2162fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           const EnumDescriptorProto& proto);
2163fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateEnumValueOptions(EnumValueDescriptor* enum_value,
2164fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                const EnumValueDescriptorProto& proto);
2165fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateServiceOptions(ServiceDescriptor* service,
2166fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                              const ServiceDescriptorProto& proto);
2167fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateMethodOptions(MethodDescriptor* method,
2168fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             const MethodDescriptorProto& proto);
2169fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2170fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  void ValidateMapKey(FieldDescriptor* field,
2171fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      const FieldDescriptorProto& proto);
2172fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville};
2173fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2174fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorPool::BuildFile(
2175fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptorProto& proto) {
2176fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(fallback_database_ == NULL)
2177fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    << "Cannot call BuildFile on a DescriptorPool that uses a "
2178fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville       "DescriptorDatabase.  You must instead find a way to get your file "
2179fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville       "into the underlying database.";
2180fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(mutex_ == NULL);   // Implied by the above GOOGLE_CHECK.
2181fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return DescriptorBuilder(this, tables_.get(), NULL).BuildFile(proto);
2182fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2183fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2184fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorPool::BuildFileCollectingErrors(
2185fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptorProto& proto,
2186fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ErrorCollector* error_collector) {
2187fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(fallback_database_ == NULL)
2188fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    << "Cannot call BuildFile on a DescriptorPool that uses a "
2189fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville       "DescriptorDatabase.  You must instead find a way to get your file "
2190fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville       "into the underlying database.";
2191fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(mutex_ == NULL);   // Implied by the above GOOGLE_CHECK.
2192fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return DescriptorBuilder(this, tables_.get(),
2193fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           error_collector).BuildFile(proto);
2194fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2195fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2196fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorPool::BuildFileFromDatabase(
2197fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptorProto& proto) const {
2198fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  mutex_->AssertHeld();
2199fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return DescriptorBuilder(this, tables_.get(),
2200fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           default_error_collector_).BuildFile(proto);
2201fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2202fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2203fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorBuilder::DescriptorBuilder(
2204fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const DescriptorPool* pool,
2205fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorPool::Tables* tables,
2206fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorPool::ErrorCollector* error_collector)
2207fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  : pool_(pool),
2208fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_(tables),
2209fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    error_collector_(error_collector),
2210fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    had_errors_(false),
2211fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    possible_undeclared_dependency_(NULL) {}
2212fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2213fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorBuilder::~DescriptorBuilder() {}
2214fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2215fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::AddError(
2216fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& element_name,
2217fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Message& descriptor,
2218fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorPool::ErrorCollector::ErrorLocation location,
2219fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& error) {
2220fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (error_collector_ == NULL) {
2221fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!had_errors_) {
2222fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(ERROR) << "Invalid proto descriptor for file \"" << filename_
2223fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 << "\":";
2224fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2225fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    GOOGLE_LOG(ERROR) << "  " << element_name << ": " << error;
2226fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2227fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    error_collector_->AddError(filename_, element_name,
2228fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               &descriptor, location, error);
2229fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2230fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  had_errors_ = true;
2231fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2232fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2233fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::AddNotDefinedError(
2234fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& element_name,
2235fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Message& descriptor,
2236fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorPool::ErrorCollector::ErrorLocation location,
2237fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& undefined_symbol) {
2238fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (possible_undeclared_dependency_ == NULL) {
2239fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(element_name, descriptor, location,
2240fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "\"" + undefined_symbol + "\" is not defined.");
2241fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2242fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(element_name, descriptor, location,
2243fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "\"" + possible_undeclared_dependency_name_ +
2244fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "\" seems to be defined in \"" +
2245fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             possible_undeclared_dependency_->name() + "\", which is not "
2246fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "imported by \"" + filename_ + "\".  To use it here, please "
2247fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "add the necessary import.");
2248fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2249fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2250fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2251fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::IsInPackage(const FileDescriptor* file,
2252fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                    const string& package_name) {
2253fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return HasPrefixString(file->package(), package_name) &&
2254fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville           (file->package().size() == package_name.size() ||
2255fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            file->package()[package_name.size()] == '.');
2256fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2257fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2258fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleSymbol DescriptorBuilder::FindSymbolNotEnforcingDeps(const string& name) {
2259fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result;
2260fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2261fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We need to search our pool and all its underlays.
2262fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const DescriptorPool* pool = pool_;
2263fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  while (true) {
2264fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // If we are looking at an underlay, we must lock its mutex_, since we are
2265fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // accessing the underlay's tables_ dircetly.
2266fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    MutexLockMaybe lock((pool == pool_) ? NULL : pool->mutex_);
2267fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2268fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Note that we don't have to check fallback_database_ here because the
2269fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // symbol has to be in one of its file's direct dependencies, and we have
2270fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // already loaded those by the time we get here.
2271fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result = pool->tables_->FindSymbol(name);
2272fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!result.IsNull()) break;
2273fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (pool->underlay_ == NULL) return kNullSymbol;
2274fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    pool = pool->underlay_;
2275fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2276fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2277fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
2278fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2279fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2280fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleSymbol DescriptorBuilder::FindSymbol(const string& name) {
2281fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = FindSymbolNotEnforcingDeps(name);
2282fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2283fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!pool_->enforce_dependencies_) {
2284fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Hack for CompilerUpgrader.
2285fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
2286fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2287fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2288fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Only find symbols which were defined in this file or one of its
2289fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // dependencies.
2290fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* file = result.GetFile();
2291fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (file == file_) return result;
2292fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < file_->dependency_count(); i++) {
2293fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (file == file_->dependency(i)) return result;
2294fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2295fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2296fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.type == Symbol::PACKAGE) {
2297fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Arg, this is overcomplicated.  The symbol is a package name.  It could
2298fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // be that the package was defined in multiple files.  result.GetFile()
2299fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // returns the first file we saw that used this package.  We've determined
2300fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // that that file is not a direct dependency of the file we are currently
2301fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // building, but it could be that some other file which *is* a direct
2302fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // dependency also defines the same package.  We can't really rule out this
2303fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // symbol unless none of the dependencies define it.
2304fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (IsInPackage(file_, name)) return result;
2305fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int i = 0; i < file_->dependency_count(); i++) {
2306fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Note:  A dependency may be NULL if it was not found or had errors.
2307fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (file_->dependency(i) != NULL &&
2308fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          IsInPackage(file_->dependency(i), name)) {
2309fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return result;
2310fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2311fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2312fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2313fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2314fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  possible_undeclared_dependency_ = file;
2315fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  possible_undeclared_dependency_name_ = name;
2316fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return kNullSymbol;
2317fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2318fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2319fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleSymbol DescriptorBuilder::LookupSymbolNoPlaceholder(
2320fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name, const string& relative_to, ResolveMode resolve_mode) {
2321fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  possible_undeclared_dependency_ = NULL;
2322fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2323fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (name.size() > 0 && name[0] == '.') {
2324fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Fully-qualified name.
2325fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return FindSymbol(name.substr(1));
2326fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2327fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2328fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // If name is something like "Foo.Bar.baz", and symbols named "Foo" are
2329fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // defined in multiple parent scopes, we only want to find "Bar.baz" in the
2330fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // innermost one.  E.g., the following should produce an error:
2331fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   message Bar { message Baz {} }
2332fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   message Foo {
2333fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //     message Bar {
2334fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //     }
2335fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //     optional Bar.Baz baz = 1;
2336fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   }
2337fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // So, we look for just "Foo" first, then look for "Bar.baz" within it if
2338fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // found.
2339fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  int name_dot_pos = name.find_first_of('.');
2340fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string first_part_of_name;
2341fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (name_dot_pos == string::npos) {
2342fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    first_part_of_name = name;
2343fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2344fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    first_part_of_name = name.substr(0, name_dot_pos);
2345fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2346fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2347fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string scope_to_try(relative_to);
2348fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2349fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  while (true) {
2350fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Chop off the last component of the scope.
2351fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    string::size_type dot_pos = scope_to_try.find_last_of('.');
2352fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (dot_pos == string::npos) {
2353fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return FindSymbol(name);
2354fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
2355fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      scope_to_try.erase(dot_pos);
2356fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2357fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2358fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Append ".first_part_of_name" and try to find.
2359fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    string::size_type old_size = scope_to_try.size();
2360fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    scope_to_try.append(1, '.');
2361fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    scope_to_try.append(first_part_of_name);
2362fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Symbol result = FindSymbol(scope_to_try);
2363fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!result.IsNull()) {
2364fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (first_part_of_name.size() < name.size()) {
2365fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // name is a compound symbol, of which we only found the first part.
2366fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // Now try to look up the rest of it.
2367fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (result.IsAggregate()) {
2368fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          scope_to_try.append(name, first_part_of_name.size(),
2369fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                              name.size() - first_part_of_name.size());
2370fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return FindSymbol(scope_to_try);
2371fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
2372fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          // We found a symbol but it's not an aggregate.  Continue the loop.
2373fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
2374fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
2375fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (resolve_mode == LOOKUP_TYPES && !result.IsType()) {
2376fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          // We found a symbol but it's not a type.  Continue the loop.
2377fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
2378fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return result;
2379fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
2380fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2381fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2382fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2383fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Not found.  Remove the name so we can try again.
2384fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    scope_to_try.erase(old_size);
2385fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2386fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2387fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2388fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleSymbol DescriptorBuilder::LookupSymbol(
2389fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name, const string& relative_to,
2390fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    PlaceholderType placeholder_type, ResolveMode resolve_mode) {
2391fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol result = LookupSymbolNoPlaceholder(
2392fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      name, relative_to, resolve_mode);
2393fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result.IsNull() && pool_->allow_unknown_) {
2394fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Not found, but AllowUnknownDependencies() is enabled.  Return a
2395fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // placeholder instead.
2396fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result = NewPlaceholder(name, placeholder_type);
2397fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2398fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return result;
2399fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2400fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2401fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleSymbol DescriptorBuilder::NewPlaceholder(const string& name,
2402fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                         PlaceholderType placeholder_type) {
2403fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Compute names.
2404fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string* placeholder_full_name;
2405fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string* placeholder_name;
2406fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string* placeholder_package;
2407fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2408fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!ValidateQualifiedName(name)) return kNullSymbol;
2409fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (name[0] == '.') {
2410fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Fully-qualified.
2411fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_full_name = tables_->AllocateString(name.substr(1));
2412fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2413fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_full_name = tables_->AllocateString(name);
2414fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2415fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2416fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string::size_type dotpos = placeholder_full_name->find_last_of('.');
2417fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (dotpos != string::npos) {
2418fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_package = tables_->AllocateString(
2419fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_full_name->substr(0, dotpos));
2420fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_name = tables_->AllocateString(
2421fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_full_name->substr(dotpos + 1));
2422fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2423fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_package = &kEmptyString;
2424fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_name = placeholder_full_name;
2425fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2426fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2427fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Create the placeholders.
2428fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptor* placeholder_file = tables_->Allocate<FileDescriptor>();
2429fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  memset(placeholder_file, 0, sizeof(*placeholder_file));
2430fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2431fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder_file->name_ =
2432fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->AllocateString(*placeholder_full_name + ".placeholder.proto");
2433fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder_file->package_ = placeholder_package;
2434fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder_file->pool_ = pool_;
2435fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder_file->options_ = &FileOptions::default_instance();
2436fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder_file->tables_ = &FileDescriptorTables::kEmpty;
2437fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // All other fields are zero or NULL.
2438fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2439fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (placeholder_type == PLACEHOLDER_ENUM) {
2440fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_file->enum_type_count_ = 1;
2441fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_file->enum_types_ =
2442fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      tables_->AllocateArray<EnumDescriptor>(1);
2443fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2444fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EnumDescriptor* placeholder_enum = &placeholder_file->enum_types_[0];
2445fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    memset(placeholder_enum, 0, sizeof(*placeholder_enum));
2446fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2447fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->full_name_ = placeholder_full_name;
2448fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->name_ = placeholder_name;
2449fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->file_ = placeholder_file;
2450fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->options_ = &EnumOptions::default_instance();
2451fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->is_placeholder_ = true;
2452fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->is_unqualified_placeholder_ = (name[0] != '.');
2453fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2454fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Enums must have at least one value.
2455fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->value_count_ = 1;
2456fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_enum->values_ = tables_->AllocateArray<EnumValueDescriptor>(1);
2457fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2458fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EnumValueDescriptor* placeholder_value = &placeholder_enum->values_[0];
2459fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    memset(placeholder_value, 0, sizeof(*placeholder_value));
2460fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2461fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_value->name_ = tables_->AllocateString("PLACEHOLDER_VALUE");
2462fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Note that enum value names are siblings of their type, not children.
2463fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_value->full_name_ =
2464fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_package->empty() ? placeholder_value->name_ :
2465fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        tables_->AllocateString(*placeholder_package + ".PLACEHOLDER_VALUE");
2466fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2467fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_value->number_ = 0;
2468fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_value->type_ = placeholder_enum;
2469fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_value->options_ = &EnumValueOptions::default_instance();
2470fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2471fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return Symbol(placeholder_enum);
2472fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2473fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_file->message_type_count_ = 1;
2474fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_file->message_types_ =
2475fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      tables_->AllocateArray<Descriptor>(1);
2476fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2477fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Descriptor* placeholder_message = &placeholder_file->message_types_[0];
2478fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    memset(placeholder_message, 0, sizeof(*placeholder_message));
2479fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2480fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_message->full_name_ = placeholder_full_name;
2481fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_message->name_ = placeholder_name;
2482fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_message->file_ = placeholder_file;
2483fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_message->options_ = &MessageOptions::default_instance();
2484fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_message->is_placeholder_ = true;
2485fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    placeholder_message->is_unqualified_placeholder_ = (name[0] != '.');
2486fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2487fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (placeholder_type == PLACEHOLDER_EXTENDABLE_MESSAGE) {
2488fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_message->extension_range_count_ = 1;
2489fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_message->extension_ranges_ =
2490fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        tables_->AllocateArray<Descriptor::ExtensionRange>(1);
2491fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_message->extension_ranges_->start = 1;
2492fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // kMaxNumber + 1 because ExtensionRange::end is exclusive.
2493fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      placeholder_message->extension_ranges_->end =
2494fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        FieldDescriptor::kMaxNumber + 1;
2495fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2496fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2497fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return Symbol(placeholder_message);
2498fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2499fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2500fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2501fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorBuilder::NewPlaceholderFile(
2502fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name) {
2503fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptor* placeholder = tables_->Allocate<FileDescriptor>();
2504fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  memset(placeholder, 0, sizeof(*placeholder));
2505fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2506fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder->name_ = tables_->AllocateString(name);
2507fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder->package_ = &kEmptyString;
2508fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder->pool_ = pool_;
2509fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder->options_ = &FileOptions::default_instance();
2510fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  placeholder->tables_ = &FileDescriptorTables::kEmpty;
2511fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // All other fields are zero or NULL.
2512fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2513fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return placeholder;
2514fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2515fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2516fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::AddSymbol(
2517fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& full_name, const void* parent, const string& name,
2518fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Message& proto, Symbol symbol) {
2519fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // If the caller passed NULL for the parent, the symbol is at file scope.
2520fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Use its file as the parent instead.
2521fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (parent == NULL) parent = file_;
2522fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2523fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (tables_->AddSymbol(full_name, symbol)) {
2524fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!file_tables_->AddAliasUnderParent(parent, name, symbol)) {
2525fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(DFATAL) << "\"" << full_name << "\" not previously defined in "
2526fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     "symbols_by_name_, but was defined in symbols_by_parent_; "
2527fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     "this shouldn't be possible.";
2528fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return false;
2529fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2530fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return true;
2531fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2532fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptor* other_file = tables_->FindSymbol(full_name).GetFile();
2533fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (other_file == file_) {
2534fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      string::size_type dot_pos = full_name.find_last_of('.');
2535fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (dot_pos == string::npos) {
2536fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(full_name, proto, DescriptorPool::ErrorCollector::NAME,
2537fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\"" + full_name + "\" is already defined.");
2538fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
2539fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(full_name, proto, DescriptorPool::ErrorCollector::NAME,
2540fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\"" + full_name.substr(dot_pos + 1) +
2541fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\" is already defined in \"" +
2542fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 full_name.substr(0, dot_pos) + "\".");
2543fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2544fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
2545fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Symbol seems to have been defined in a different file.
2546fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(full_name, proto, DescriptorPool::ErrorCollector::NAME,
2547fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "\"" + full_name + "\" is already defined in file \"" +
2548fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               other_file->name() + "\".");
2549fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2550fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;
2551fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2552fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2553fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2554fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::AddPackage(
2555fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name, const Message& proto, const FileDescriptor* file) {
2556fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (tables_->AddSymbol(name, Symbol(file))) {
2557fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Success.  Also add parent package, if any.
2558fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    string::size_type dot_pos = name.find_last_of('.');
2559fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (dot_pos == string::npos) {
2560fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // No parents.
2561fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      ValidateSymbolName(name, name, proto);
2562fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
2563fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Has parent.
2564fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      string* parent_name = tables_->AllocateString(name.substr(0, dot_pos));
2565fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddPackage(*parent_name, proto, file);
2566fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      ValidateSymbolName(name.substr(dot_pos + 1), name, proto);
2567fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2568fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2569fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Symbol existing_symbol = tables_->FindSymbol(name);
2570fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // It's OK to redefine a package.
2571fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (existing_symbol.type != Symbol::PACKAGE) {
2572fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Symbol seems to have been defined in a different file.
2573fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(name, proto, DescriptorPool::ErrorCollector::NAME,
2574fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "\"" + name + "\" is already defined (as something other than "
2575fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "a package) in file \"" + existing_symbol.GetFile()->name() +
2576fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "\".");
2577fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2578fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2579fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2580fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2581fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateSymbolName(
2582fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name, const string& full_name, const Message& proto) {
2583fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (name.empty()) {
2584fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(full_name, proto, DescriptorPool::ErrorCollector::NAME,
2585fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Missing name.");
2586fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2587fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int i = 0; i < name.size(); i++) {
2588fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // I don't trust isalnum() due to locales.  :(
2589fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if ((name[i] < 'a' || 'z' < name[i]) &&
2590fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          (name[i] < 'A' || 'Z' < name[i]) &&
2591fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          (name[i] < '0' || '9' < name[i]) &&
2592fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          (name[i] != '_')) {
2593fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(full_name, proto, DescriptorPool::ErrorCollector::NAME,
2594fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\"" + name + "\" is not a valid identifier.");
2595fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2596fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2597fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2598fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2599fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2600fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::ValidateQualifiedName(const string& name) {
2601fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool last_was_period = false;
2602fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2603fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < name.size(); i++) {
2604fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // I don't trust isalnum() due to locales.  :(
2605fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (('a' <= name[i] && name[i] <= 'z') ||
2606fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        ('A' <= name[i] && name[i] <= 'Z') ||
2607fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        ('0' <= name[i] && name[i] <= '9') ||
2608fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        (name[i] == '_')) {
2609fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      last_was_period = false;
2610fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (name[i] == '.') {
2611fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (last_was_period) return false;
2612fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      last_was_period = true;
2613fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
2614fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return false;
2615fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2616fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2617fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2618fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return !name.empty() && !last_was_period;
2619fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2620fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2621fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
2622fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2623fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// This generic implementation is good for all descriptors except
2624fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// FileDescriptor.
2625fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletemplate<class DescriptorT> void DescriptorBuilder::AllocateOptions(
2626fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const typename DescriptorT::OptionsType& orig_options,
2627fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorT* descriptor) {
2628fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AllocateOptionsImpl(descriptor->full_name(), descriptor->full_name(),
2629fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      orig_options, descriptor);
2630fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2631fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2632fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// We specialize for FileDescriptor.
2633fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::AllocateOptions(const FileOptions& orig_options,
2634fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                        FileDescriptor* descriptor) {
2635fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We add the dummy token so that LookupSymbol does the right thing.
2636fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AllocateOptionsImpl(descriptor->package() + ".dummy", descriptor->name(),
2637fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                      orig_options, descriptor);
2638fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2639fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2640fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilletemplate<class DescriptorT> void DescriptorBuilder::AllocateOptionsImpl(
2641fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name_scope,
2642fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& element_name,
2643fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const typename DescriptorT::OptionsType& orig_options,
2644fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorT* descriptor) {
2645fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We need to use a dummy pointer to work around a bug in older versions of
2646fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // GCC.  Otherwise, the following two lines could be replaced with:
2647fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   typename DescriptorT::OptionsType* options =
2648fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //       tables_->AllocateMessage<typename DescriptorT::OptionsType>();
2649fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  typename DescriptorT::OptionsType* const dummy = NULL;
2650fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  typename DescriptorT::OptionsType* options = tables_->AllocateMessage(dummy);
2651fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  options->CopyFrom(orig_options);
2652fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  descriptor->options_ = options;
2653fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2654fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Don't add to options_to_interpret_ unless there were uninterpreted
2655fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // options.  This not only avoids unnecessary work, but prevents a
2656fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // bootstrapping problem when building descriptors for descriptor.proto.
2657fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptor.proto does not contain any uninterpreted options, but
2658fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // attempting to interpret options anyway will cause
2659fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // OptionsType::GetDescriptor() to be called which may then deadlock since
2660fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // we're still trying to build it.
2661fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (options->uninterpreted_option_size() > 0) {
2662fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options_to_interpret_.push_back(
2663fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        OptionsToInterpret(name_scope, element_name, &orig_options, options));
2664fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2665fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2666fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2667fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2668fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A common pattern:  We want to convert a repeated field in the descriptor
2669fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// to an array of values, calling some method to build each value.
2670fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#define BUILD_ARRAY(INPUT, OUTPUT, NAME, METHOD, PARENT)             \
2671fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  OUTPUT->NAME##_count_ = INPUT.NAME##_size();                       \
2672fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AllocateArray(INPUT.NAME##_size(), &OUTPUT->NAME##s_);             \
2673fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < INPUT.NAME##_size(); i++) {                    \
2674fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    METHOD(INPUT.NAME(i), PARENT, OUTPUT->NAME##s_ + i);             \
2675fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2676fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2677fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleconst FileDescriptor* DescriptorBuilder::BuildFile(
2678fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptorProto& proto) {
2679fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  filename_ = proto.name();
2680fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2681fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Check if the file already exists and is identical to the one being built.
2682fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note:  This only works if the input is canonical -- that is, it
2683fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   fully-qualifies all type names, has no UninterpretedOptions, etc.
2684fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   This is fine, because this idempotency "feature" really only exists to
2685fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   accomodate one hack in the proto1->proto2 migration layer.
2686fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FileDescriptor* existing_file = tables_->FindFile(filename_);
2687fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (existing_file != NULL) {
2688fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // File already in pool.  Compare the existing one to the input.
2689fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FileDescriptorProto existing_proto;
2690fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    existing_file->CopyTo(&existing_proto);
2691fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (existing_proto.SerializeAsString() == proto.SerializeAsString()) {
2692fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // They're identical.  Return the existing descriptor.
2693fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return existing_file;
2694fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2695fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2696fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Not a match.  The error will be detected and handled later.
2697fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2698fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2699fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Check to see if this file is already on the pending files list.
2700fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(kenton):  Allow recursive imports?  It may not work with some
2701fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   (most?) programming languages.  E.g., in C++, a forward declaration
2702fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   of a type is not sufficient to allow it to be used even in a
2703fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   generated header file due to inlining.  This could perhaps be
2704fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   worked around using tricks involving inserting #include statements
2705fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   mid-file, but that's pretty ugly, and I'm pretty sure there are
2706fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   some languages out there that do not allow recursive dependencies
2707fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   at all.
2708fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < tables_->pending_files_.size(); i++) {
2709fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (tables_->pending_files_[i] == proto.name()) {
2710fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      string error_message("File recursively imports itself: ");
2711fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      for (; i < tables_->pending_files_.size(); i++) {
2712fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        error_message.append(tables_->pending_files_[i]);
2713fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        error_message.append(" -> ");
2714fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2715fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      error_message.append(proto.name());
2716fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2717fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(proto.name(), proto, DescriptorPool::ErrorCollector::OTHER,
2718fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               error_message);
2719fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return NULL;
2720fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2721fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2722fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2723fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // If we have a fallback_database_, attempt to load all dependencies now,
2724fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // before checkpointing tables_.  This avoids confusion with recursive
2725fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // checkpoints.
2726fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (pool_->fallback_database_ != NULL) {
2727fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->pending_files_.push_back(proto.name());
2728fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int i = 0; i < proto.dependency_size(); i++) {
2729fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (tables_->FindFile(proto.dependency(i)) == NULL &&
2730fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          (pool_->underlay_ == NULL ||
2731fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville           pool_->underlay_->FindFileByName(proto.dependency(i)) == NULL)) {
2732fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // We don't care what this returns since we'll find out below anyway.
2733fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        pool_->TryFindFileInFallbackDatabase(proto.dependency(i));
2734fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2735fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2736fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->pending_files_.pop_back();
2737fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2738fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2739fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Checkpoint the tables so that we can roll back if something goes wrong.
2740fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  tables_->Checkpoint();
2741fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2742fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  FileDescriptor* result = tables_->Allocate<FileDescriptor>();
2743fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_ = result;
2744fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2745fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_tables_ = tables_->AllocateFileTables();
2746fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_->tables_ = file_tables_;
2747fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2748fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_name()) {
2749fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError("", proto, DescriptorPool::ErrorCollector::OTHER,
2750fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Missing field: FileDescriptorProto.name.");
2751fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2752fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2753fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_ = tables_->AllocateString(proto.name());
2754fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (proto.has_package()) {
2755fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->package_ = tables_->AllocateString(proto.package());
2756fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2757fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // We cannot rely on proto.package() returning a valid string if
2758fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // proto.has_package() is false, because we might be running at static
2759fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // initialization time, in which case default values have not yet been
2760fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // initialized.
2761fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->package_ = tables_->AllocateString("");
2762fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2763fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->pool_ = pool_;
2764fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2765fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Add to tables.
2766fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!tables_->AddFile(result)) {
2767fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(proto.name(), proto, DescriptorPool::ErrorCollector::OTHER,
2768fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "A file with this name is already in the pool.");
2769fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Bail out early so that if this is actually the exact same file, we
2770fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // don't end up reporting that every single symbol is already defined.
2771fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->Rollback();
2772fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
2773fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2774fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!result->package().empty()) {
2775fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddPackage(result->package(), proto, result);
2776fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2777fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2778fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Make sure all dependencies are loaded.
2779fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  set<string> seen_dependencies;
2780fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->dependency_count_ = proto.dependency_size();
2781fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->dependencies_ =
2782fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->AllocateArray<const FileDescriptor*>(proto.dependency_size());
2783fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < proto.dependency_size(); i++) {
2784fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!seen_dependencies.insert(proto.dependency(i)).second) {
2785fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(proto.name(), proto,
2786fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::OTHER,
2787fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "Import \"" + proto.dependency(i) + "\" was listed twice.");
2788fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2789fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2790fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FileDescriptor* dependency = tables_->FindFile(proto.dependency(i));
2791fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (dependency == NULL && pool_->underlay_ != NULL) {
2792fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      dependency = pool_->underlay_->FindFileByName(proto.dependency(i));
2793fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2794fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2795fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (dependency == NULL) {
2796fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (pool_->allow_unknown_) {
2797fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        dependency = NewPlaceholderFile(proto.dependency(i));
2798fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
2799fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        string message;
2800fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (pool_->fallback_database_ == NULL) {
2801fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          message = "Import \"" + proto.dependency(i) +
2802fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    "\" has not been loaded.";
2803fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
2804fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          message = "Import \"" + proto.dependency(i) +
2805fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    "\" was not found or had errors.";
2806fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
2807fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(proto.name(), proto,
2808fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::OTHER,
2809fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 message);
2810fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2811fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2812fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2813fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->dependencies_[i] = dependency;
2814fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2815fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2816fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Convert children.
2817fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, message_type, BuildMessage  , NULL);
2818fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, enum_type   , BuildEnum     , NULL);
2819fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, service     , BuildService  , NULL);
2820fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, extension   , BuildExtension, NULL);
2821fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2822fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
2823fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
2824fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
2825fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2826fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
2827fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2828fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2829fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note that the following steps must occur in exactly the specified order.
2830fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2831fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Cross-link.
2832fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  CrossLinkFile(result, proto);
2833fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2834fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Interpret any remaining uninterpreted options gathered into
2835fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // options_to_interpret_ during descriptor building.  Cross-linking has made
2836fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // extension options known, so all interpretations should now succeed.
2837fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!had_errors_) {
2838fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    OptionInterpreter option_interpreter(this);
2839fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (vector<OptionsToInterpret>::iterator iter =
2840fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             options_to_interpret_.begin();
2841fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville         iter != options_to_interpret_.end(); ++iter) {
2842fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      option_interpreter.InterpretOptions(&(*iter));
2843fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2844fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options_to_interpret_.clear();
2845fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2846fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2847fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Validate options.
2848fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!had_errors_) {
2849fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ValidateFileOptions(result, proto);
2850fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2851fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2852fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (had_errors_) {
2853fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->Rollback();
2854fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return NULL;
2855fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2856fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->Checkpoint();
2857fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return result;
2858fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2859fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2860fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2861fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildMessage(const DescriptorProto& proto,
2862fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     const Descriptor* parent,
2863fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     Descriptor* result) {
2864fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string& scope = (parent == NULL) ?
2865fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file_->package() : parent->full_name();
2866fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* full_name = tables_->AllocateString(scope);
2867fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!full_name->empty()) full_name->append(1, '.');
2868fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(proto.name());
2869fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2870fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ValidateSymbolName(proto.name(), *full_name, proto);
2871fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2872fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_            = tables_->AllocateString(proto.name());
2873fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->full_name_       = full_name;
2874fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->file_            = file_;
2875fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->containing_type_ = parent;
2876fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->is_placeholder_  = false;
2877fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->is_unqualified_placeholder_ = false;
2878fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2879fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, field          , BuildField         , result);
2880fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, nested_type    , BuildMessage       , result);
2881fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, enum_type      , BuildEnum          , result);
2882fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, extension_range, BuildExtensionRange, result);
2883fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, extension      , BuildExtension     , result);
2884fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2885fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
2886fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
2887fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
2888fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2889fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
2890fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2891fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2892fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AddSymbol(result->full_name(), parent, result->name(),
2893fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            proto, Symbol(result));
2894fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2895fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Check that no fields have numbers in extension ranges.
2896fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < result->field_count(); i++) {
2897fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* field = result->field(i);
2898fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int j = 0; j < result->extension_range_count(); j++) {
2899fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const Descriptor::ExtensionRange* range = result->extension_range(j);
2900fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (range->start <= field->number() && field->number() < range->end) {
2901fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(field->full_name(), proto.extension_range(j),
2902fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::NUMBER,
2903fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 strings::Substitute(
2904fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   "Extension range $0 to $1 includes field \"$2\" ($3).",
2905fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   range->start, range->end - 1,
2906fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   field->name(), field->number()));
2907fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2908fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2909fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2910fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2911fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Check that extension ranges don't overlap.
2912fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < result->extension_range_count(); i++) {
2913fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor::ExtensionRange* range1 = result->extension_range(i);
2914fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int j = i + 1; j < result->extension_range_count(); j++) {
2915fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const Descriptor::ExtensionRange* range2 = result->extension_range(j);
2916fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (range1->end > range2->start && range2->end > range1->start) {
2917fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(result->full_name(), proto.extension_range(j),
2918fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::NUMBER,
2919fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 strings::Substitute("Extension range $0 to $1 overlaps with "
2920fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     "already-defined range $2 to $3.",
2921fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     range2->start, range2->end - 1,
2922fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     range1->start, range1->end - 1));
2923fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
2924fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
2925fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2926fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
2927fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2928fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildFieldOrExtension(const FieldDescriptorProto& proto,
2929fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                              const Descriptor* parent,
2930fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                              FieldDescriptor* result,
2931fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                              bool is_extension) {
2932fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string& scope = (parent == NULL) ?
2933fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file_->package() : parent->full_name();
2934fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* full_name = tables_->AllocateString(scope);
2935fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!full_name->empty()) full_name->append(1, '.');
2936fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(proto.name());
2937fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2938fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ValidateSymbolName(proto.name(), *full_name, proto);
2939fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2940fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_         = tables_->AllocateString(proto.name());
2941fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->full_name_    = full_name;
2942fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->file_         = file_;
2943fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->number_       = proto.number();
2944fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->is_extension_ = is_extension;
2945fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2946fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // If .proto files follow the style guide then the name should already be
2947fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // lower-cased.  If that's the case we can just reuse the string we already
2948fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // allocated rather than allocate a new one.
2949fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string lowercase_name(proto.name());
2950fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  LowerString(&lowercase_name);
2951fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (lowercase_name == proto.name()) {
2952fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->lowercase_name_ = result->name_;
2953fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
2954fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->lowercase_name_ = tables_->AllocateString(lowercase_name);
2955fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2956fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2957fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Don't bother with the above optimization for camel-case names since
2958fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // .proto files that follow the guide shouldn't be using names in this
2959fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // format, so the optimization wouldn't help much.
2960fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->camelcase_name_ = tables_->AllocateString(ToCamelCase(proto.name()));
2961fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2962fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Some compilers do not allow static_cast directly between two enum types,
2963fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // so we must cast to int first.
2964fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->type_  = static_cast<FieldDescriptor::Type>(
2965fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     implicit_cast<int>(proto.type()));
2966fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->label_ = static_cast<FieldDescriptor::Label>(
2967fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     implicit_cast<int>(proto.label()));
2968fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2969fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Some of these may be filled in when cross-linking.
2970fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->containing_type_ = NULL;
2971fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->extension_scope_ = NULL;
2972fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->experimental_map_key_ = NULL;
2973fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->message_type_ = NULL;
2974fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->enum_type_ = NULL;
2975fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2976fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->has_default_value_ = proto.has_default_value();
2977fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (proto.has_default_value() && result->is_repeated()) {
2978fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(result->full_name(), proto,
2979fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::DEFAULT_VALUE,
2980fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Repeated fields can't have default values.");
2981fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
2982fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
2983fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (proto.has_type()) {
2984fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (proto.has_default_value()) {
2985fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      char* end_pos = NULL;
2986fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      switch (result->cpp_type()) {
2987fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_INT32:
2988fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_int32_ =
2989fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            strtol(proto.default_value().c_str(), &end_pos, 0);
2990fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
2991fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_INT64:
2992fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_int64_ =
2993fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            strto64(proto.default_value().c_str(), &end_pos, 0);
2994fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
2995fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_UINT32:
2996fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_uint32_ =
2997fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            strtoul(proto.default_value().c_str(), &end_pos, 0);
2998fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
2999fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_UINT64:
3000fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_uint64_ =
3001fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            strtou64(proto.default_value().c_str(), &end_pos, 0);
3002fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3003fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_FLOAT:
3004d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          if (proto.default_value() == "inf") {
3005d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_float_ = numeric_limits<float>::infinity();
3006d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          } else if (proto.default_value() == "-inf") {
3007d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_float_ = -numeric_limits<float>::infinity();
3008d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          } else if (proto.default_value() == "nan") {
3009d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_float_ = numeric_limits<float>::quiet_NaN();
3010d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          } else  {
3011d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_float_ =
3012d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville              NoLocaleStrtod(proto.default_value().c_str(), &end_pos);
3013d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          }
3014fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3015fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_DOUBLE:
3016d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          if (proto.default_value() == "inf") {
3017d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_double_ = numeric_limits<double>::infinity();
3018d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          } else if (proto.default_value() == "-inf") {
3019d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_double_ = -numeric_limits<double>::infinity();
3020d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          } else if (proto.default_value() == "nan") {
3021d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_double_ = numeric_limits<double>::quiet_NaN();
3022d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          } else  {
3023d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville            result->default_value_double_ =
3024d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville              NoLocaleStrtod(proto.default_value().c_str(), &end_pos);
3025d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville          }
3026fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3027fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_BOOL:
3028fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          if (proto.default_value() == "true") {
3029fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            result->default_value_bool_ = true;
3030fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          } else if (proto.default_value() == "false") {
3031fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            result->default_value_bool_ = false;
3032fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          } else {
3033fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            AddError(result->full_name(), proto,
3034fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     DescriptorPool::ErrorCollector::DEFAULT_VALUE,
3035fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                     "Boolean default must be true or false.");
3036fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          }
3037fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3038fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_ENUM:
3039fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          // This will be filled in when cross-linking.
3040fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_enum_ = NULL;
3041fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3042fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_STRING:
3043fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          if (result->type() == FieldDescriptor::TYPE_BYTES) {
3044fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            result->default_value_string_ = tables_->AllocateString(
3045fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville              UnescapeCEscapeString(proto.default_value()));
3046fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          } else {
3047fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            result->default_value_string_ =
3048fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville              tables_->AllocateString(proto.default_value());
3049fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          }
3050fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3051fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_MESSAGE:
3052fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          AddError(result->full_name(), proto,
3053fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   DescriptorPool::ErrorCollector::DEFAULT_VALUE,
3054fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   "Messages can't have default values.");
3055fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->has_default_value_ = false;
3056fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3057fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3058fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3059fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (end_pos != NULL) {
3060fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // end_pos is only set non-NULL by the parsers for numeric types, above.
3061fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // This checks that the default was non-empty and had no extra junk
3062fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // after the end of the number.
3063fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (proto.default_value().empty() || *end_pos != '\0') {
3064fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          AddError(result->full_name(), proto,
3065fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   DescriptorPool::ErrorCollector::DEFAULT_VALUE,
3066fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   "Couldn't parse default value.");
3067fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
3068fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3069fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3070fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // No explicit default value
3071fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      switch (result->cpp_type()) {
3072fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_INT32:
3073fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_int32_ = 0;
3074fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3075fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_INT64:
3076fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_int64_ = 0;
3077fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3078fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_UINT32:
3079fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_uint32_ = 0;
3080fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3081fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_UINT64:
3082fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_uint64_ = 0;
3083fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3084fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_FLOAT:
3085fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_float_ = 0.0f;
3086fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3087fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_DOUBLE:
3088fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_double_ = 0.0;
3089fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3090fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_BOOL:
3091fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_bool_ = false;
3092fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3093fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_ENUM:
3094fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          // This will be filled in when cross-linking.
3095fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_enum_ = NULL;
3096fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3097fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_STRING:
3098fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          result->default_value_string_ = &kEmptyString;
3099fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::CPPTYPE_MESSAGE:
3101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
3102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result->number() <= 0) {
3107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(result->full_name(), proto, DescriptorPool::ErrorCollector::NUMBER,
3108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Field numbers must be positive integers.");
3109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else if (result->number() > FieldDescriptor::kMaxNumber) {
3110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(result->full_name(), proto, DescriptorPool::ErrorCollector::NUMBER,
3111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             strings::Substitute("Field numbers cannot be greater than $0.",
3112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 FieldDescriptor::kMaxNumber));
3113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else if (result->number() >= FieldDescriptor::kFirstReservedNumber &&
3114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             result->number() <= FieldDescriptor::kLastReservedNumber) {
3115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(result->full_name(), proto, DescriptorPool::ErrorCollector::NUMBER,
3116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             strings::Substitute(
3117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "Field numbers $0 through $1 are reserved for the protocol "
3118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "buffer library implementation.",
3119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               FieldDescriptor::kFirstReservedNumber,
3120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               FieldDescriptor::kLastReservedNumber));
3121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (is_extension) {
3124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!proto.has_extendee()) {
3125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(result->full_name(), proto,
3126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::EXTENDEE,
3127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "FieldDescriptorProto.extendee not set for extension field.");
3128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->extension_scope_ = parent;
3131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (proto.has_extendee()) {
3133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(result->full_name(), proto,
3134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::EXTENDEE,
3135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "FieldDescriptorProto.extendee set for non-extension field.");
3136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->containing_type_ = parent;
3139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
3142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
3143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
3144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
3146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AddSymbol(result->full_name(), parent, result->name(),
3149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            proto, Symbol(result));
3150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildExtensionRange(
3153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const DescriptorProto::ExtensionRange& proto,
3154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const Descriptor* parent,
3155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Descriptor::ExtensionRange* result) {
3156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->start = proto.start();
3157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->end = proto.end();
3158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result->start <= 0) {
3159fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(parent->full_name(), proto,
3160fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::NUMBER,
3161fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Extension numbers must be positive integers.");
3162fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3163fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3164fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result->end > FieldDescriptor::kMaxNumber + 1) {
3165fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(parent->full_name(), proto,
3166fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::NUMBER,
3167fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             strings::Substitute("Extension numbers cannot be greater than $0.",
3168fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                 FieldDescriptor::kMaxNumber));
3169fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3170fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3171fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (result->start >= result->end) {
3172fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(parent->full_name(), proto,
3173fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::NUMBER,
3174fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Extension range end number must be greater than start number.");
3175fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3176fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3177fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3178fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildEnum(const EnumDescriptorProto& proto,
3179fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                  const Descriptor* parent,
3180fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                  EnumDescriptor* result) {
3181fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string& scope = (parent == NULL) ?
3182fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file_->package() : parent->full_name();
3183fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* full_name = tables_->AllocateString(scope);
3184fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!full_name->empty()) full_name->append(1, '.');
3185fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(proto.name());
3186fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3187fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ValidateSymbolName(proto.name(), *full_name, proto);
3188fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3189fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_            = tables_->AllocateString(proto.name());
3190fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->full_name_       = full_name;
3191fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->file_            = file_;
3192fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->containing_type_ = parent;
3193fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->is_placeholder_  = false;
3194fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->is_unqualified_placeholder_ = false;
3195fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3196fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (proto.value_size() == 0) {
3197fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // We cannot allow enums with no values because this would mean there
3198fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // would be no valid default value for fields of this type.
3199fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(result->full_name(), proto,
3200fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::NAME,
3201fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Enums must contain at least one value.");
3202fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3203fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3204fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, value, BuildEnumValue, result);
3205fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3206fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
3207fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
3208fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
3209fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3210fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
3211fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3212fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3213fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AddSymbol(result->full_name(), parent, result->name(),
3214fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            proto, Symbol(result));
3215fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3216fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3217fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildEnumValue(const EnumValueDescriptorProto& proto,
3218fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                       const EnumDescriptor* parent,
3219fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                       EnumValueDescriptor* result) {
3220fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_   = tables_->AllocateString(proto.name());
3221fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->number_ = proto.number();
3222fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->type_   = parent;
3223fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3224fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note:  full_name for enum values is a sibling to the parent's name, not a
3225fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   child of it.
3226fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* full_name = tables_->AllocateString(*parent->full_name_);
3227fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->resize(full_name->size() - parent->name_->size());
3228fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(*result->name_);
3229fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->full_name_ = full_name;
3230fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3231fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ValidateSymbolName(proto.name(), *full_name, proto);
3232fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3233fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
3234fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
3235fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
3236fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3237fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
3238fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3239fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3240fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Again, enum values are weird because we makes them appear as siblings
3241fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // of the enum type instead of children of it.  So, we use
3242fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // parent->containing_type() as the value's parent.
3243fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool added_to_outer_scope =
3244fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddSymbol(result->full_name(), parent->containing_type(), result->name(),
3245fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville              proto, Symbol(result));
3246fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3247fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // However, we also want to be able to search for values within a single
3248fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // enum type, so we add it as a child of the enum type itself, too.
3249fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note:  This could fail, but if it does, the error has already been
3250fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   reported by the above AddSymbol() call, so we ignore the return code.
3251fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool added_to_inner_scope =
3252fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file_tables_->AddAliasUnderParent(parent, result->name(), Symbol(result));
3253fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3254fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (added_to_inner_scope && !added_to_outer_scope) {
3255fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // This value did not conflict with any values defined in the same enum,
3256fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // but it did conflict with some other symbol defined in the enum type's
3257fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // scope.  Let's print an additional error to explain this.
3258fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    string outer_scope;
3259fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (parent->containing_type() == NULL) {
3260fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      outer_scope = file_->package();
3261fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3262fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      outer_scope = parent->containing_type()->full_name();
3263fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3264fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3265fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (outer_scope.empty()) {
3266fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      outer_scope = "the global scope";
3267fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3268fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      outer_scope = "\"" + outer_scope + "\"";
3269fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3270fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3271fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(result->full_name(), proto,
3272fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::NAME,
3273fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Note that enum values use C++ scoping rules, meaning that "
3274fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "enum values are siblings of their type, not children of it.  "
3275fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Therefore, \"" + result->name() + "\" must be unique within "
3276fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             + outer_scope + ", not just within \"" + parent->name() + "\".");
3277fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3278fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3279fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // An enum is allowed to define two numbers that refer to the same value.
3280fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // FindValueByNumber() should return the first such value, so we simply
3281fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // ignore AddEnumValueByNumber()'s return code.
3282fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_tables_->AddEnumValueByNumber(result);
3283fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3284fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3285fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildService(const ServiceDescriptorProto& proto,
3286fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     const void* dummy,
3287fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                     ServiceDescriptor* result) {
3288fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* full_name = tables_->AllocateString(file_->package());
3289fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!full_name->empty()) full_name->append(1, '.');
3290fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(proto.name());
3291fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3292fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ValidateSymbolName(proto.name(), *full_name, proto);
3293fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3294fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_      = tables_->AllocateString(proto.name());
3295fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->full_name_ = full_name;
3296fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->file_      = file_;
3297fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3298fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  BUILD_ARRAY(proto, result, method, BuildMethod, result);
3299fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3300fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
3301fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
3302fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
3303fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3304fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
3305fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3306fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3307fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AddSymbol(result->full_name(), NULL, result->name(),
3308fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            proto, Symbol(result));
3309fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3310fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3311fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::BuildMethod(const MethodDescriptorProto& proto,
3312fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                    const ServiceDescriptor* parent,
3313fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                    MethodDescriptor* result) {
3314fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->name_    = tables_->AllocateString(proto.name());
3315fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->service_ = parent;
3316fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3317fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string* full_name = tables_->AllocateString(parent->full_name());
3318fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(1, '.');
3319fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  full_name->append(*result->name_);
3320fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->full_name_ = full_name;
3321fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3322fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  ValidateSymbolName(proto.name(), *full_name, proto);
3323fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3324fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // These will be filled in when cross-linking.
3325fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->input_type_ = NULL;
3326fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  result->output_type_ = NULL;
3327fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3328fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Copy options.
3329fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!proto.has_options()) {
3330fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    result->options_ = NULL;  // Will set to default_instance later.
3331fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3332fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AllocateOptions(proto.options(), result);
3333fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3334fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3335fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  AddSymbol(result->full_name(), parent, result->name(),
3336fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            proto, Symbol(result));
3337fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3338fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3339fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#undef BUILD_ARRAY
3340fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3341fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
3342fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3343fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkFile(
3344fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FileDescriptor* file, const FileDescriptorProto& proto) {
3345fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (file->options_ == NULL) {
3346fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    file->options_ = &FileOptions::default_instance();
3347fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3348fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3349fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < file->message_type_count(); i++) {
3350fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkMessage(&file->message_types_[i], proto.message_type(i));
3351fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3352fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3353fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < file->extension_count(); i++) {
3354fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkField(&file->extensions_[i], proto.extension(i));
3355fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3356fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3357fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < file->enum_type_count(); i++) {
3358fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkEnum(&file->enum_types_[i], proto.enum_type(i));
3359fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3360fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3361fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < file->service_count(); i++) {
3362fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkService(&file->services_[i], proto.service(i));
3363fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3364fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3365fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3366fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkMessage(
3367fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Descriptor* message, const DescriptorProto& proto) {
3368fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (message->options_ == NULL) {
3369fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    message->options_ = &MessageOptions::default_instance();
3370fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3371fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3372fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < message->nested_type_count(); i++) {
3373fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkMessage(&message->nested_types_[i], proto.nested_type(i));
3374fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3375fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3376fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < message->enum_type_count(); i++) {
3377fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkEnum(&message->enum_types_[i], proto.enum_type(i));
3378fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3379fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3380fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < message->field_count(); i++) {
3381fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkField(&message->fields_[i], proto.field(i));
3382fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3383fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3384fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < message->extension_count(); i++) {
3385fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkField(&message->extensions_[i], proto.extension(i));
3386fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3387fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3388fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3389fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkField(
3390fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FieldDescriptor* field, const FieldDescriptorProto& proto) {
3391fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (field->options_ == NULL) {
3392fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    field->options_ = &FieldOptions::default_instance();
3393fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3394fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3395fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (proto.has_extendee()) {
3396fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Symbol extendee = LookupSymbol(proto.extendee(), field->full_name(),
3397fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   PLACEHOLDER_EXTENDABLE_MESSAGE);
3398fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (extendee.IsNull()) {
3399fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddNotDefinedError(field->full_name(), proto,
3400fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         DescriptorPool::ErrorCollector::EXTENDEE,
3401fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         proto.extendee());
3402fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return;
3403fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (extendee.type != Symbol::MESSAGE) {
3404fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto,
3405fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::EXTENDEE,
3406fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "\"" + proto.extendee() + "\" is not a message type.");
3407fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return;
3408fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3409fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    field->containing_type_ = extendee.descriptor;
3410fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3411fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!field->containing_type()->IsExtensionNumber(field->number())) {
3412fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto,
3413fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::NUMBER,
3414fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               strings::Substitute("\"$0\" does not declare $1 as an "
3415fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   "extension number.",
3416fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   field->containing_type()->full_name(),
3417fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   field->number()));
3418fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3419fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3420fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3421fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (proto.has_type_name()) {
3422fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // Assume we are expecting a message type unless the proto contains some
3423fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // evidence that it expects an enum type.  This only makes a difference if
3424fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // we end up creating a placeholder.
3425fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    bool expecting_enum = (proto.type() == FieldDescriptorProto::TYPE_ENUM) ||
3426fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                          proto.has_default_value();
3427fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3428fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Symbol type =
3429fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      LookupSymbol(proto.type_name(), field->full_name(),
3430fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   expecting_enum ? PLACEHOLDER_ENUM : PLACEHOLDER_MESSAGE,
3431fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   LOOKUP_TYPES);
3432fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3433fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (type.IsNull()) {
3434fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddNotDefinedError(field->full_name(), proto,
3435fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         DescriptorPool::ErrorCollector::TYPE,
3436fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                         proto.type_name());
3437fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return;
3438fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3439fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3440fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!proto.has_type()) {
3441fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Choose field type based on symbol.
3442fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (type.type == Symbol::MESSAGE) {
3443fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        field->type_ = FieldDescriptor::TYPE_MESSAGE;
3444fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (type.type == Symbol::ENUM) {
3445fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        field->type_ = FieldDescriptor::TYPE_ENUM;
3446fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
3447fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(field->full_name(), proto,
3448fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::TYPE,
3449fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\"" + proto.type_name() + "\" is not a type.");
3450fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return;
3451fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3452fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3453fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3454fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field->cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE) {
3455fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (type.type != Symbol::MESSAGE) {
3456fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(field->full_name(), proto,
3457fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::TYPE,
3458fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\"" + proto.type_name() + "\" is not a message type.");
3459fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return;
3460fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3461fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field->message_type_ = type.descriptor;
3462fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3463fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (field->has_default_value()) {
3464fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(field->full_name(), proto,
3465fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::DEFAULT_VALUE,
3466fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "Messages can't have default values.");
3467fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3468fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (field->cpp_type() == FieldDescriptor::CPPTYPE_ENUM) {
3469fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (type.type != Symbol::ENUM) {
3470fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(field->full_name(), proto,
3471fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::TYPE,
3472fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "\"" + proto.type_name() + "\" is not an enum type.");
3473fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return;
3474fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3475fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field->enum_type_ = type.enum_descriptor;
3476fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3477fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (field->enum_type()->is_placeholder_) {
3478fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // We can't look up default values for placeholder types.  We'll have
3479fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // to just drop them.
3480fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        field->has_default_value_ = false;
3481fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3482fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3483fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (field->has_default_value()) {
3484fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // We can't just use field->enum_type()->FindValueByName() here
3485fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // because that locks the pool's mutex, which we have already locked
3486fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // at this point.
3487fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        Symbol default_value =
3488fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          LookupSymbolNoPlaceholder(proto.default_value(),
3489fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                    field->enum_type()->full_name());
3490fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3491fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (default_value.type == Symbol::ENUM_VALUE &&
3492fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            default_value.enum_value_descriptor->type() == field->enum_type()) {
3493fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          field->default_value_enum_ = default_value.enum_value_descriptor;
3494fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
3495fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          AddError(field->full_name(), proto,
3496fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   DescriptorPool::ErrorCollector::DEFAULT_VALUE,
3497fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   "Enum type \"" + field->enum_type()->full_name() +
3498fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   "\" has no value named \"" + proto.default_value() + "\".");
3499fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
3500fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (field->enum_type()->value_count() > 0) {
3501fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // All enums must have at least one value, or we would have reported
3502fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // an error elsewhere.  We use the first defined value as the default
3503fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // if a default is not explicitly defined.
3504fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        field->default_value_enum_ = field->enum_type()->value(0);
3505fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3506fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3507fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3508fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "Field with primitive type has type_name.");
3509fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3510fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3511fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field->cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE ||
3512fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        field->cpp_type() == FieldDescriptor::CPPTYPE_ENUM) {
3513fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3514fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "Field with message or enum type missing type_name.");
3515fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3516fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3517fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3518fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Add the field to the fields-by-number table.
3519fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note:  We have to do this *after* cross-linking because extensions do not
3520fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   know their containing type until now.
3521fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!file_tables_->AddFieldByNumber(field)) {
3522fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* conflicting_field =
3523fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      file_tables_->FindFieldByNumber(field->containing_type(),
3524fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      field->number());
3525fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field->is_extension()) {
3526fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto,
3527fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::NUMBER,
3528fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               strings::Substitute("Extension number $0 has already been used "
3529fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   "in \"$1\" by extension \"$2\".",
3530fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   field->number(),
3531fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   field->containing_type()->full_name(),
3532fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   conflicting_field->full_name()));
3533fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3534fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto,
3535fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::NUMBER,
3536fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               strings::Substitute("Field number $0 has already been used in "
3537fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   "\"$1\" by field \"$2\".",
3538fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   field->number(),
3539fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   field->containing_type()->full_name(),
3540fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                   conflicting_field->name()));
3541fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3542fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3543fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3544fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (field->is_extension()) {
3545fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // No need for error checking: if the extension number collided,
3546fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // we've already been informed of it by the if() above.
3547fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    tables_->AddExtension(field);
3548fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3549fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3550fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Add the field to the lowercase-name and camelcase-name tables.
3551fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  file_tables_->AddFieldByStylizedNames(field);
3552fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3553fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3554fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkEnum(
3555fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EnumDescriptor* enum_type, const EnumDescriptorProto& proto) {
3556fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (enum_type->options_ == NULL) {
3557fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enum_type->options_ = &EnumOptions::default_instance();
3558fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3559fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3560fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < enum_type->value_count(); i++) {
3561fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkEnumValue(&enum_type->values_[i], proto.value(i));
3562fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3563fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3564fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3565fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkEnumValue(
3566fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EnumValueDescriptor* enum_value, const EnumValueDescriptorProto& proto) {
3567fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (enum_value->options_ == NULL) {
3568fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    enum_value->options_ = &EnumValueOptions::default_instance();
3569fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3570fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3571fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3572fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkService(
3573fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ServiceDescriptor* service, const ServiceDescriptorProto& proto) {
3574fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (service->options_ == NULL) {
3575fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    service->options_ = &ServiceOptions::default_instance();
3576fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3577fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3578fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < service->method_count(); i++) {
3579fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    CrossLinkMethod(&service->methods_[i], proto.method(i));
3580fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3581fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3582fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3583fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::CrossLinkMethod(
3584fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    MethodDescriptor* method, const MethodDescriptorProto& proto) {
3585fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (method->options_ == NULL) {
3586fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    method->options_ = &MethodOptions::default_instance();
3587fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3588fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3589fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol input_type = LookupSymbol(proto.input_type(), method->full_name());
3590fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (input_type.IsNull()) {
3591fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddNotDefinedError(method->full_name(), proto,
3592fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       DescriptorPool::ErrorCollector::INPUT_TYPE,
3593fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       proto.input_type());
3594fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else if (input_type.type != Symbol::MESSAGE) {
3595fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(method->full_name(), proto,
3596fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::INPUT_TYPE,
3597fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "\"" + proto.input_type() + "\" is not a message type.");
3598fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3599fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    method->input_type_ = input_type.descriptor;
3600fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3601fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3602fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol output_type = LookupSymbol(proto.output_type(), method->full_name());
3603fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (output_type.IsNull()) {
3604fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddNotDefinedError(method->full_name(), proto,
3605fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       DescriptorPool::ErrorCollector::OUTPUT_TYPE,
3606fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                       proto.output_type());
3607fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else if (output_type.type != Symbol::MESSAGE) {
3608fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(method->full_name(), proto,
3609fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::OUTPUT_TYPE,
3610fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "\"" + proto.output_type() + "\" is not a message type.");
3611fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3612fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    method->output_type_ = output_type.descriptor;
3613fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3614fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3615fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3616fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
3617fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3618fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#define VALIDATE_OPTIONS_FROM_ARRAY(descriptor, array_name, type)  \
3619fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < descriptor->array_name##_count(); ++i) {     \
3620fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Validate##type##Options(descriptor->array_name##s_ + i,        \
3621fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                            proto.array_name(i));                  \
3622fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3623fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3624fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Determine if the file uses optimize_for = LITE_RUNTIME, being careful to
3625fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// avoid problems that exist at init time.
3626fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillestatic bool IsLite(const FileDescriptor* file) {
3627fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // TODO(kenton):  I don't even remember how many of these conditions are
3628fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   actually possible.  I'm just being super-safe.
3629fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return file != NULL &&
3630fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville         &file->options() != NULL &&
3631fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville         &file->options() != &FileOptions::default_instance() &&
3632fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville         file->options().optimize_for() == FileOptions::LITE_RUNTIME;
3633fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3634fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3635fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateFileOptions(FileDescriptor* file,
3636fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                            const FileDescriptorProto& proto) {
3637fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(file, message_type, Message);
3638fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(file, enum_type, Enum);
3639fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(file, service, Service);
3640fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(file, extension, Field);
3641fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3642fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Lite files can only be imported by other Lite files.
3643fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!IsLite(file)) {
3644fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int i = 0; i < file->dependency_count(); i++) {
3645fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (IsLite(file->dependency(i))) {
3646fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(
3647fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          file->name(), proto,
3648fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          DescriptorPool::ErrorCollector::OTHER,
3649fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          "Files that do not use optimize_for = LITE_RUNTIME cannot import "
3650fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          "files which do use this option.  This file is not lite, but it "
3651fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          "imports \"" + file->dependency(i)->name() + "\" which is.");
3652fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        break;
3653fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3654fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3655fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3656fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3657fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3658fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateMessageOptions(Descriptor* message,
3659fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                               const DescriptorProto& proto) {
3660fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(message, field, Field);
3661fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(message, nested_type, Message);
3662fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(message, enum_type, Enum);
3663fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(message, extension, Field);
3664fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3665fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3666fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateFieldOptions(FieldDescriptor* field,
3667fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptorProto& proto) {
3668fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (field->options().has_experimental_map_key()) {
3669fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ValidateMapKey(field, proto);
3670fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3671fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3672fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Only repeated primitive fields may be packed.
3673d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville  if (field->options().packed() && !field->is_packable()) {
3674d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville    AddError(
3675d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville      field->full_name(), proto,
3676d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville      DescriptorPool::ErrorCollector::TYPE,
3677d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville      "[packed = true] can only be specified for repeated primitive fields.");
3678fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3679fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3680fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note:  Default instance may not yet be initialized here, so we have to
3681fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  //   avoid reading from it.
3682fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (field->containing_type_ != NULL &&
3683fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      &field->containing_type()->options() !=
3684fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      &MessageOptions::default_instance() &&
3685fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field->containing_type()->options().message_set_wire_format()) {
3686fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field->is_extension()) {
3687fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (!field->is_optional() ||
3688fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          field->type() != FieldDescriptor::TYPE_MESSAGE) {
3689fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddError(field->full_name(), proto,
3690fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 DescriptorPool::ErrorCollector::TYPE,
3691fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 "Extensions of MessageSets must be optional messages.");
3692fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3693fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3694fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      AddError(field->full_name(), proto,
3695fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               DescriptorPool::ErrorCollector::NAME,
3696fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville               "MessageSets cannot have fields, only extensions.");
3697fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3698fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3699fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3700fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Lite extensions can only be of Lite types.
3701fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (IsLite(field->file()) &&
3702fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field->containing_type_ != NULL &&
3703fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      !IsLite(field->containing_type()->file())) {
3704fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto,
3705fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::EXTENDEE,
3706fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Extensions to non-lite types can only be declared in non-lite "
3707fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "files.  Note that you cannot extend a non-lite type to contain "
3708fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "a lite type, but the reverse is allowed.");
3709fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3710fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3711fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3712fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateEnumOptions(EnumDescriptor* enm,
3713fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                            const EnumDescriptorProto& proto) {
3714fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(enm, value, EnumValue);
3715fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3716fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3717fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateEnumValueOptions(
3718fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    EnumValueDescriptor* enum_value, const EnumValueDescriptorProto& proto) {
3719fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Nothing to do so far.
3720fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3721fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateServiceOptions(ServiceDescriptor* service,
3722fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const ServiceDescriptorProto& proto) {
3723fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (IsLite(service->file())) {
3724fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(service->full_name(), proto,
3725fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             DescriptorPool::ErrorCollector::NAME,
3726fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Files with optimize_for = LITE_RUNTIME cannot define services.");
3727fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3728fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3729fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  VALIDATE_OPTIONS_FROM_ARRAY(service, method, Method);
3730fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3731fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3732fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateMethodOptions(MethodDescriptor* method,
3733fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const MethodDescriptorProto& proto) {
3734fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Nothing to do so far.
3735fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3736fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3737fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::ValidateMapKey(FieldDescriptor* field,
3738fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                       const FieldDescriptorProto& proto) {
3739fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!field->is_repeated()) {
3740fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3741fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "map type is only allowed for repeated fields.");
3742fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return;
3743fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3744fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3745fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (field->cpp_type() != FieldDescriptor::CPPTYPE_MESSAGE) {
3746fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3747fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "map type is only allowed for fields with a message type.");
3748fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return;
3749fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3750fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3751fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Descriptor* item_type = field->message_type();
3752fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (item_type == NULL) {
3753fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3754fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Could not find field type.");
3755fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return;
3756fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3757fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3758fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find the field in item_type named by "experimental_map_key"
3759fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const string& key_name = field->options().experimental_map_key();
3760fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Symbol key_symbol = LookupSymbol(
3761fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      key_name,
3762fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // We append ".key_name" to the containing type's name since
3763fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // LookupSymbol() searches for peers of the supplied name, not
3764fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // children of the supplied name.
3765fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      item_type->full_name() + "." + key_name);
3766fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3767fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (key_symbol.IsNull() || key_symbol.field_descriptor->is_extension()) {
3768fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3769fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "Could not find field named \"" + key_name + "\" in type \"" +
3770fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             item_type->full_name() + "\".");
3771fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return;
3772fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3773fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* key_field = key_symbol.field_descriptor;
3774fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3775fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (key_field->is_repeated()) {
3776fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3777fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "map_key must not name a repeated field.");
3778fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return;
3779fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3780fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3781fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (key_field->cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE) {
3782fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    AddError(field->full_name(), proto, DescriptorPool::ErrorCollector::TYPE,
3783fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville             "map key must name a scalar or string field.");
3784fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return;
3785fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3786fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3787fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  field->experimental_map_key_ = key_field;
3788fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3789fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3790fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#undef VALIDATE_OPTIONS_FROM_ARRAY
3791fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3792fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// -------------------------------------------------------------------
3793fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3794fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorBuilder::OptionInterpreter::OptionInterpreter(
3795fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    DescriptorBuilder* builder) : builder_(builder) {
3796fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(builder_);
3797fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3798fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3799fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleDescriptorBuilder::OptionInterpreter::~OptionInterpreter() {
3800fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3801fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3802fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::OptionInterpreter::InterpretOptions(
3803fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    OptionsToInterpret* options_to_interpret) {
3804fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note that these may be in different pools, so we can't use the same
3805fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // descriptor and reflection objects on both.
3806fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Message* options = options_to_interpret->options;
3807fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Message* original_options = options_to_interpret->original_options;
3808fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3809fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  bool failed = false;
3810fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  options_to_interpret_ = options_to_interpret;
3811fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3812fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find the uninterpreted_option field in the mutable copy of the options
3813fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // and clear them, since we're about to interpret them.
3814fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* uninterpreted_options_field =
3815fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      options->GetDescriptor()->FindFieldByName("uninterpreted_option");
3816fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(uninterpreted_options_field != NULL)
3817fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      << "No field named \"uninterpreted_option\" in the Options proto.";
3818fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  options->GetReflection()->ClearField(options, uninterpreted_options_field);
3819fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3820fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Find the uninterpreted_option field in the original options.
3821fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* original_uninterpreted_options_field =
3822fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      original_options->GetDescriptor()->
3823fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          FindFieldByName("uninterpreted_option");
3824fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(original_uninterpreted_options_field != NULL)
3825fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      << "No field named \"uninterpreted_option\" in the Options proto.";
3826fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3827fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const int num_uninterpreted_options = original_options->GetReflection()->
3828fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      FieldSize(*original_options, original_uninterpreted_options_field);
3829fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < num_uninterpreted_options; ++i) {
3830fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    uninterpreted_option_ = down_cast<const UninterpretedOption*>(
3831fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        &original_options->GetReflection()->GetRepeatedMessage(
3832fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            *original_options, original_uninterpreted_options_field, i));
3833fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (!InterpretSingleOption(options)) {
3834fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Error already added by InterpretSingleOption().
3835fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      failed = true;
3836fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
3837fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3838fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3839fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Reset these, so we don't have any dangling pointers.
3840fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  uninterpreted_option_ = NULL;
3841fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  options_to_interpret_ = NULL;
3842fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3843fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!failed) {
3844fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // InterpretSingleOption() added the interpreted options in the
3845fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // UnknownFieldSet, in case the option isn't yet known to us.  Now we
3846fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // serialize the options message and deserialize it back.  That way, any
3847fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // option fields that we do happen to know about will get moved from the
3848fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // UnknownFieldSet into the real fields, and thus be available right away.
3849fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // If they are not known, that's OK too. They will get reparsed into the
3850fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // UnknownFieldSet and wait there until the message is parsed by something
3851fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // that does know about the options.
3852fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    string buf;
3853fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options->AppendToString(&buf);
3854fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    GOOGLE_CHECK(options->ParseFromString(buf))
3855fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        << "Protocol message serialized itself in invalid fashion.";
3856fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3857fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3858fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return !failed;
3859fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
3860fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3861fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::OptionInterpreter::InterpretSingleOption(
3862fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    Message* options) {
3863fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // First do some basic validation.
3864fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (uninterpreted_option_->name_size() == 0) {
3865fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // This should never happen unless the parser has gone seriously awry or
3866fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // someone has manually created the uninterpreted option badly.
3867fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return AddNameError("Option must have a name.");
3868fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3869fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (uninterpreted_option_->name(0).name_part() == "uninterpreted_option") {
3870fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return AddNameError("Option must not use reserved name "
3871fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                        "\"uninterpreted_option\".");
3872fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3873fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3874fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Descriptor* options_descriptor = NULL;
3875fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Get the options message's descriptor from the builder's pool, so that we
3876fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // get the version that knows about any extension options declared in the
3877fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // file we're currently building. The descriptor should be there as long as
3878fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the file we're building imported "google/protobuf/descriptors.proto".
3879fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3880fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Note that we use DescriptorBuilder::FindSymbol(), not
3881fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // DescriptorPool::FindMessageTypeByName() because we're already holding the
3882fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // pool's mutex, and the latter method locks it again.
3883fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  Symbol symbol = builder_->FindSymbolNotEnforcingDeps(
3884fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options->GetDescriptor()->full_name());
3885fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!symbol.IsNull() && symbol.type == Symbol::MESSAGE) {
3886fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options_descriptor = symbol.descriptor;
3887fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  } else {
3888fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // The options message's descriptor was not in the builder's pool, so use
3889fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // the standard version from the generated pool. We're not holding the
3890fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // generated pool's mutex, so we can search it the straightforward way.
3891fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options_descriptor = options->GetDescriptor();
3892fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3893fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(options_descriptor);
3894fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3895fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We iterate over the name parts to drill into the submessages until we find
3896fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the leaf field for the option. As we drill down we remember the current
3897fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // submessage's descriptor in |descriptor| and the next field in that
3898fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // submessage in |field|. We also track the fields we're drilling down
3899fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // through in |intermediate_fields|. As we go, we reconstruct the full option
3900fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // name in |debug_msg_name|, for use in error messages.
3901fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const Descriptor* descriptor = options_descriptor;
3902fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* field = NULL;
3903fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  vector<const FieldDescriptor*> intermediate_fields;
3904fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  string debug_msg_name = "";
3905fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3906fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < uninterpreted_option_->name_size(); ++i) {
3907fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const string& name_part = uninterpreted_option_->name(i).name_part();
3908fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (debug_msg_name.size() > 0) {
3909fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      debug_msg_name += ".";
3910fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3911fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (uninterpreted_option_->name(i).is_extension()) {
3912fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      debug_msg_name += "(" + name_part + ")";
3913fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Search for the extension's descriptor as an extension in the builder's
3914fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // pool. Note that we use DescriptorBuilder::LookupSymbol(), not
3915fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // DescriptorPool::FindExtensionByName(), for two reasons: 1) It allows
3916fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // relative lookups, and 2) because we're already holding the pool's
3917fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // mutex, and the latter method locks it again.
3918fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      Symbol symbol = builder_->LookupSymbol(name_part,
3919fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                             options_to_interpret_->name_scope);
3920fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (!symbol.IsNull() && symbol.type == Symbol::FIELD) {
3921fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        field = symbol.field_descriptor;
3922fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3923fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // If we don't find the field then the field's descriptor was not in the
3924fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // builder's pool, but there's no point in looking in the generated
3925fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // pool. We require that you import the file that defines any extensions
3926fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // you use, so they must be present in the builder's pool.
3927fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else {
3928fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      debug_msg_name += name_part;
3929fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Search for the field's descriptor as a regular field.
3930fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      field = descriptor->FindFieldByName(name_part);
3931fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3932fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3933fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (field == NULL) {
3934fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (get_allow_unknown(builder_->pool_)) {
3935fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // We can't find the option, but AllowUnknownDependencies() is enabled,
3936fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // so we will just leave it as uninterpreted.
3937fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddWithoutInterpreting(*uninterpreted_option_, options);
3938fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return true;
3939fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
3940fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddNameError("Option \"" + debug_msg_name + "\" unknown.");
3941fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3942fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (field->containing_type() != descriptor) {
3943fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (get_is_placeholder(field->containing_type())) {
3944fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // The field is an extension of a placeholder type, so we can't
3945fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // reliably verify whether it is a valid extension to use here (e.g.
3946fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // we don't know if it is an extension of the correct *Options message,
3947fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // or if it has a valid field number, etc.).  Just leave it as
3948fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // uninterpreted instead.
3949fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        AddWithoutInterpreting(*uninterpreted_option_, options);
3950fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return true;
3951fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
3952fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // This can only happen if, due to some insane misconfiguration of the
3953fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // pools, we find the options message in one pool but the field in
3954fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // another. This would probably imply a hefty bug somewhere.
3955fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddNameError("Option field \"" + debug_msg_name +
3956fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                            "\" is not a field or extension of message \"" +
3957fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                            descriptor->name() + "\".");
3958fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3959fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (field->is_repeated()) {
3960fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return AddNameError("Option field \"" + debug_msg_name +
3961fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                          "\" is repeated. Repeated options are not "
3962fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                          "supported.");
3963fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (i < uninterpreted_option_->name_size() - 1) {
3964fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (field->cpp_type() != FieldDescriptor::CPPTYPE_MESSAGE) {
3965fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddNameError("Option \"" +  debug_msg_name +
3966fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                            "\" is an atomic type, not a message.");
3967fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
3968fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // Drill down into the submessage.
3969fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        intermediate_fields.push_back(field);
3970fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        descriptor = field->message_type();
3971fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
3972fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    } else if (field->cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE) {
3973fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return AddNameError("Option field \"" + debug_msg_name +
3974fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                          "\" cannot be of message type.");
3975fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
3976fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3977fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3978fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We've found the leaf field. Now we use UnknownFieldSets to set its value
3979fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // on the options message. We do so because the message may not yet know
3980fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // about its extension fields, so we may not be able to set the fields
3981fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // directly. But the UnknownFieldSets will serialize to the same wire-format
3982fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // message, so reading that message back in once the extension fields are
3983fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // known will populate them correctly.
3984fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3985fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // First see if the option is already set.
3986fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!ExamineIfOptionIsSet(
3987fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          intermediate_fields.begin(),
3988fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          intermediate_fields.end(),
3989fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          field, debug_msg_name,
3990fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          options->GetReflection()->GetUnknownFields(*options))) {
3991fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;  // ExamineIfOptionIsSet() already added the error.
3992fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
3993fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3994fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
3995fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // First set the value on the UnknownFieldSet corresponding to the
3996fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // innermost message.
3997fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  scoped_ptr<UnknownFieldSet> unknown_fields(new UnknownFieldSet());
3998fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (!SetOptionValue(field, unknown_fields.get())) {
3999fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return false;  // SetOptionValue() already added the error.
4000fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4001fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4002fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Now wrap the UnknownFieldSet with UnknownFieldSets corresponding to all
4003fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the intermediate messages.
4004fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (vector<const FieldDescriptor*>::reverse_iterator iter =
4005fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville           intermediate_fields.rbegin();
4006fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville       iter != intermediate_fields.rend(); ++iter) {
4007fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    scoped_ptr<UnknownFieldSet> parent_unknown_fields(new UnknownFieldSet());
4008fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    switch ((*iter)->type()) {
4009fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case FieldDescriptor::TYPE_MESSAGE: {
4010fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        io::StringOutputStream outstr(
4011fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            parent_unknown_fields->AddLengthDelimited((*iter)->number()));
4012fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        io::CodedOutputStream out(&outstr);
4013fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        internal::WireFormat::SerializeUnknownFields(*unknown_fields, &out);
4014fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        GOOGLE_CHECK(!out.HadError())
4015fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            << "Unexpected failure while serializing option submessage "
4016fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            << debug_msg_name << "\".";
4017fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        break;
4018fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4019fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4020fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      case FieldDescriptor::TYPE_GROUP: {
4021fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville         parent_unknown_fields->AddGroup((*iter)->number())
4022fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                              ->MergeFrom(*unknown_fields);
4023fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        break;
4024fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4025fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4026fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      default:
4027fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        GOOGLE_LOG(FATAL) << "Invalid wire type for CPPTYPE_MESSAGE: "
4028fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   << (*iter)->type();
4029fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return false;
4030fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
4031fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    unknown_fields.reset(parent_unknown_fields.release());
4032fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4033fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4034fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // Now merge the UnknownFieldSet corresponding to the top-level message into
4035fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // the options message.
4036fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  options->GetReflection()->MutableUnknownFields(options)->MergeFrom(
4037fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      *unknown_fields);
4038fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4039fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return true;
4040fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4041fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4042fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::OptionInterpreter::AddWithoutInterpreting(
4043fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const UninterpretedOption& uninterpreted_option, Message* options) {
4044fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  const FieldDescriptor* field =
4045fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    options->GetDescriptor()->FindFieldByName("uninterpreted_option");
4046fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  GOOGLE_CHECK(field != NULL);
4047fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4048fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  options->GetReflection()->AddMessage(options, field)
4049fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    ->CopyFrom(uninterpreted_option);
4050fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4051fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4052fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::OptionInterpreter::ExamineIfOptionIsSet(
4053fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    vector<const FieldDescriptor*>::const_iterator intermediate_fields_iter,
4054fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    vector<const FieldDescriptor*>::const_iterator intermediate_fields_end,
4055fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* innermost_field, const string& debug_msg_name,
4056fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const UnknownFieldSet& unknown_fields) {
4057fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We do linear searches of the UnknownFieldSet and its sub-groups.  This
4058fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // should be fine since it's unlikely that any one options structure will
4059fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // contain more than a handful of options.
4060fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4061fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  if (intermediate_fields_iter == intermediate_fields_end) {
4062fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    // We're at the innermost submessage.
4063fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    for (int i = 0; i < unknown_fields.field_count(); i++) {
4064fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (unknown_fields.field(i).number() == innermost_field->number()) {
4065fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddNameError("Option \"" + debug_msg_name +
4066fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                            "\" was already set.");
4067fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4068fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
4069fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    return true;
4070fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4071fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4072fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  for (int i = 0; i < unknown_fields.field_count(); i++) {
4073fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    if (unknown_fields.field(i).number() ==
4074fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        (*intermediate_fields_iter)->number()) {
4075fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const UnknownField* unknown_field = &unknown_fields.field(i);
4076fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      FieldDescriptor::Type type = (*intermediate_fields_iter)->type();
4077fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // Recurse into the next submessage.
4078fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      switch (type) {
4079fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::TYPE_MESSAGE:
4080fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          if (unknown_field->type() == UnknownField::TYPE_LENGTH_DELIMITED) {
4081fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            UnknownFieldSet intermediate_unknown_fields;
4082fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            if (intermediate_unknown_fields.ParseFromString(
4083fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    unknown_field->length_delimited()) &&
4084fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                !ExamineIfOptionIsSet(intermediate_fields_iter + 1,
4085fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      intermediate_fields_end,
4086fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      innermost_field, debug_msg_name,
4087fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      intermediate_unknown_fields)) {
4088fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville              return false;  // Error already added.
4089fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            }
4090fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          }
4091fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
4092fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4093fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        case FieldDescriptor::TYPE_GROUP:
4094fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          if (unknown_field->type() == UnknownField::TYPE_GROUP) {
4095fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            if (!ExamineIfOptionIsSet(intermediate_fields_iter + 1,
4096fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      intermediate_fields_end,
4097fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      innermost_field, debug_msg_name,
4098fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                      unknown_field->group())) {
4099fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville              return false;  // Error already added.
4100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            }
4101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          }
4102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          break;
4103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        default:
4105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          GOOGLE_LOG(FATAL) << "Invalid wire type for CPPTYPE_MESSAGE: " << type;
4106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return false;
4107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
4109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return true;
4111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool DescriptorBuilder::OptionInterpreter::SetOptionValue(
4114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    const FieldDescriptor* option_field,
4115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    UnknownFieldSet* unknown_fields) {
4116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  // We switch on the CppType to validate.
4117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (option_field->cpp_type()) {
4118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_INT32:
4120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->has_positive_int_value()) {
4121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (uninterpreted_option_->positive_int_value() >
4122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            static_cast<uint64>(kint32max)) {
4123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return AddValueError("Value out of range for int32 option \"" +
4124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               option_field->full_name() + "\".");
4125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
4126fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          SetInt32(option_field->number(),
4127fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   uninterpreted_option_->positive_int_value(),
4128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   option_field->type(), unknown_fields);
4129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
4130fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->has_negative_int_value()) {
4131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (uninterpreted_option_->negative_int_value() <
4132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            static_cast<int64>(kint32min)) {
4133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return AddValueError("Value out of range for int32 option \"" +
4134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               option_field->full_name() + "\".");
4135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
4136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          SetInt32(option_field->number(),
4137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   uninterpreted_option_->negative_int_value(),
4138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   option_field->type(), unknown_fields);
4139fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
4140fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4141fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be integer for int32 option \"" +
4142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             option_field->full_name() + "\".");
4143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_INT64:
4147fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->has_positive_int_value()) {
4148fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (uninterpreted_option_->positive_int_value() >
4149fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            static_cast<uint64>(kint64max)) {
4150fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return AddValueError("Value out of range for int64 option \"" +
4151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               option_field->full_name() + "\".");
4152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
4153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          SetInt64(option_field->number(),
4154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   uninterpreted_option_->positive_int_value(),
4155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                   option_field->type(), unknown_fields);
4156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
4157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->has_negative_int_value()) {
4158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        SetInt64(option_field->number(),
4159fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 uninterpreted_option_->negative_int_value(),
4160fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                 option_field->type(), unknown_fields);
4161fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4162fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be integer for int64 option \"" +
4163fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             option_field->full_name() + "\".");
4164fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4165fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4166fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4167fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_UINT32:
4168fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->has_positive_int_value()) {
4169fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (uninterpreted_option_->positive_int_value() > kuint32max) {
4170fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          return AddValueError("Value out of range for uint32 option \"" +
4171fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                               option_field->name() + "\".");
4172fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        } else {
4173fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          SetUInt32(option_field->number(),
4174fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    uninterpreted_option_->positive_int_value(),
4175fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                    option_field->type(), unknown_fields);
4176fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
4177fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4178fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be non-negative integer for uint32 "
4179fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "option \"" + option_field->full_name() + "\".");
4180fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4181fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4182fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4183fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_UINT64:
4184fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->has_positive_int_value()) {
4185fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        SetUInt64(option_field->number(),
4186fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  uninterpreted_option_->positive_int_value(),
4187fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                  option_field->type(), unknown_fields);
4188fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4189fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be non-negative integer for uint64 "
4190fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "option \"" + option_field->full_name() + "\".");
4191fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4192fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4193fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4194fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_FLOAT: {
4195fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      float value;
4196fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->has_double_value()) {
4197fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = uninterpreted_option_->double_value();
4198fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->has_positive_int_value()) {
4199fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = uninterpreted_option_->positive_int_value();
4200fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->has_negative_int_value()) {
4201fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = uninterpreted_option_->negative_int_value();
4202fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4203fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be number for float option \"" +
4204fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             option_field->full_name() + "\".");
4205fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4206fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddFixed32(option_field->number(),
4207fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          google::protobuf::internal::WireFormatLite::EncodeFloat(value));
4208fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4209fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
4210fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4211fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_DOUBLE: {
4212fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      double value;
4213fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->has_double_value()) {
4214fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = uninterpreted_option_->double_value();
4215fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->has_positive_int_value()) {
4216fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = uninterpreted_option_->positive_int_value();
4217fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->has_negative_int_value()) {
4218fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = uninterpreted_option_->negative_int_value();
4219fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4220fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be number for double option \"" +
4221fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             option_field->full_name() + "\".");
4222fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4223fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddFixed64(option_field->number(),
4224fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          google::protobuf::internal::WireFormatLite::EncodeDouble(value));
4225fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4226fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
4227fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4228fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_BOOL:
4229fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      uint64 value;
4230fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (!uninterpreted_option_->has_identifier_value()) {
4231fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be identifier for boolean option "
4232fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "\"" + option_field->full_name() + "\".");
4233fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4234fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (uninterpreted_option_->identifier_value() == "true") {
4235fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = 1;
4236fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else if (uninterpreted_option_->identifier_value() == "false") {
4237fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        value = 0;
4238fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4239fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be \"true\" or \"false\" for boolean "
4240fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "option \"" + option_field->full_name() + "\".");
4241fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4242fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(option_field->number(), value);
4243fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4244fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4245fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_ENUM: {
4246fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (!uninterpreted_option_->has_identifier_value()) {
4247fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be identifier for enum-valued option "
4248fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "\"" + option_field->full_name() + "\".");
4249fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4250fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const EnumDescriptor* enum_type = option_field->enum_type();
4251fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const string& value_name = uninterpreted_option_->identifier_value();
4252fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      const EnumValueDescriptor* enum_value = NULL;
4253fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4254fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (enum_type->file()->pool() != DescriptorPool::generated_pool()) {
4255fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // Note that the enum value's fully-qualified name is a sibling of the
4256fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // enum's name, not a child of it.
4257fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        string fully_qualified_name = enum_type->full_name();
4258fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        fully_qualified_name.resize(fully_qualified_name.size() -
4259fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                                    enum_type->name().size());
4260fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        fully_qualified_name += value_name;
4261fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4262fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // Search for the enum value's descriptor in the builder's pool. Note
4263fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // that we use DescriptorBuilder::FindSymbolNotEnforcingDeps(), not
4264fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // DescriptorPool::FindEnumValueByName() because we're already holding
4265fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // the pool's mutex, and the latter method locks it again.
4266fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        Symbol symbol =
4267fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          builder_->FindSymbolNotEnforcingDeps(fully_qualified_name);
4268fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        if (!symbol.IsNull() && symbol.type == Symbol::ENUM_VALUE) {
4269fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          if (symbol.enum_value_descriptor->type() != enum_type) {
4270fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            return AddValueError("Enum type \"" + enum_type->full_name() +
4271fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                "\" has no value named \"" + value_name + "\" for option \"" +
4272fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                option_field->full_name() +
4273fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                "\". This appears to be a value from a sibling type.");
4274fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          } else {
4275fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville            enum_value = symbol.enum_value_descriptor;
4276fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          }
4277fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        }
4278fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4279fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // The enum type is in the generated pool, so we can search for the
4280fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // value there.
4281fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        enum_value = enum_type->FindValueByName(value_name);
4282fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4283fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4284fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (enum_value == NULL) {
4285fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Enum type \"" +
4286fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             option_field->enum_type()->full_name() +
4287fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "\" has no value named \"" + value_name + "\" for "
4288fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "option \"" + option_field->full_name() + "\".");
4289fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      } else {
4290fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // Sign-extension is not a problem, since we cast directly from int32 to
4291fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        // uint64, without first going through uint32.
4292fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        unknown_fields->AddVarint(option_field->number(),
4293fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          static_cast<uint64>(static_cast<int64>(enum_value->number())));
4294fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4295fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4296fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    }
4297fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4298fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_STRING:
4299fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      if (!uninterpreted_option_->has_string_value()) {
4300fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        return AddValueError("Value must be quoted string for string option "
4301fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                             "\"" + option_field->full_name() + "\".");
4302fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      }
4303fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // The string has already been unquoted and unescaped by the parser.
4304fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddLengthDelimited(option_field->number(),
4305fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          uninterpreted_option_->string_value());
4306fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4307fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4308fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::CPPTYPE_MESSAGE:
4309fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // We don't currently support defining a message-typed option, so we
4310fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      // should never actually get here.
4311fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      return AddValueError("Option \"" + option_field->full_name() +
4312fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           "\" is a message.  To set fields within it, use "
4313fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           "syntax like \"" + option_field->name() +
4314fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville                           ".foo = value\".");
4315fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4316fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4317fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4318fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  return true;
4319fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4320fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4321fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::OptionInterpreter::SetInt32(int number, int32 value,
4322fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FieldDescriptor::Type type, UnknownFieldSet* unknown_fields) {
4323fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (type) {
4324fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_INT32:
4325fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(number,
4326fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville        static_cast<uint64>(static_cast<int64>(value)));
4327fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4328fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4329fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_SFIXED32:
4330fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddFixed32(number, static_cast<uint32>(value));
4331fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4332fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4333fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_SINT32:
4334fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(number,
4335fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          google::protobuf::internal::WireFormatLite::ZigZagEncode32(value));
4336fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4337fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4338fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default:
4339fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(FATAL) << "Invalid wire type for CPPTYPE_INT32: " << type;
4340fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4341fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4342fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4343fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4344fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::OptionInterpreter::SetInt64(int number, int64 value,
4345fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FieldDescriptor::Type type, UnknownFieldSet* unknown_fields) {
4346fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (type) {
4347fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_INT64:
4348fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(number, static_cast<uint64>(value));
4349fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4350fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4351fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_SFIXED64:
4352fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddFixed64(number, static_cast<uint64>(value));
4353fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4354fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4355fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_SINT64:
4356fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(number,
4357fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville          google::protobuf::internal::WireFormatLite::ZigZagEncode64(value));
4358fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4359fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4360fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default:
4361fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(FATAL) << "Invalid wire type for CPPTYPE_INT64: " << type;
4362fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4363fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4364fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4365fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4366fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::OptionInterpreter::SetUInt32(int number, uint32 value,
4367fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FieldDescriptor::Type type, UnknownFieldSet* unknown_fields) {
4368fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (type) {
4369fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_UINT32:
4370fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(number, static_cast<uint64>(value));
4371fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4372fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4373fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_FIXED32:
4374fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddFixed32(number, static_cast<uint32>(value));
4375fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4376fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4377fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default:
4378fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(FATAL) << "Invalid wire type for CPPTYPE_UINT32: " << type;
4379fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4380fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4381fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4382fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4383fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillevoid DescriptorBuilder::OptionInterpreter::SetUInt64(int number, uint64 value,
4384fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    FieldDescriptor::Type type, UnknownFieldSet* unknown_fields) {
4385fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  switch (type) {
4386fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_UINT64:
4387fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddVarint(number, value);
4388fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4389fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4390fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    case FieldDescriptor::TYPE_FIXED64:
4391fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      unknown_fields->AddFixed64(number, value);
4392fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4393fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4394fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville    default:
4395fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      GOOGLE_LOG(FATAL) << "Invalid wire type for CPPTYPE_UINT64: " << type;
4396fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville      break;
4397fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville  }
4398fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}
4399fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville
4400fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // namespace protobuf
4401fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}  // namespace google
4402