• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2008 Google Inc.  All rights reserved.
3 // https://developers.google.com/protocol-buffers/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 //     * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 //     * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 //     * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 
31 // Author: kenton@google.com (Kenton Varda)
32 //  Based on original Protocol Buffers design by
33 //  Sanjay Ghemawat, Jeff Dean, and others.
34 //
35 // Class for parsing tokenized text from a ZeroCopyInputStream.
36 
37 #ifndef GOOGLE_PROTOBUF_IO_TOKENIZER_H__
38 #define GOOGLE_PROTOBUF_IO_TOKENIZER_H__
39 
40 #include <string>
41 #include <vector>
42 #include <google/protobuf/stubs/common.h>
43 #include <google/protobuf/stubs/logging.h>
44 
45 namespace google {
46 namespace protobuf {
47 namespace io {
48 
49 class ZeroCopyInputStream;     // zero_copy_stream.h
50 
51 // Defined in this file.
52 class ErrorCollector;
53 class Tokenizer;
54 
55 // By "column number", the proto compiler refers to a count of the number
56 // of bytes before a given byte, except that a tab character advances to
57 // the next multiple of 8 bytes.  Note in particular that column numbers
58 // are zero-based, while many user interfaces use one-based column numbers.
59 typedef int ColumnNumber;
60 
61 // Abstract interface for an object which collects the errors that occur
62 // during parsing.  A typical implementation might simply print the errors
63 // to stdout.
64 class LIBPROTOBUF_EXPORT ErrorCollector {
65  public:
ErrorCollector()66   inline ErrorCollector() {}
67   virtual ~ErrorCollector();
68 
69   // Indicates that there was an error in the input at the given line and
70   // column numbers.  The numbers are zero-based, so you may want to add
71   // 1 to each before printing them.
72   virtual void AddError(int line, ColumnNumber column,
73                         const string& message) = 0;
74 
75   // Indicates that there was a warning in the input at the given line and
76   // column numbers.  The numbers are zero-based, so you may want to add
77   // 1 to each before printing them.
AddWarning(int line,ColumnNumber column,const string & message)78   virtual void AddWarning(int line, ColumnNumber column,
79                           const string& message) { }
80 
81  private:
82   GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ErrorCollector);
83 };
84 
85 // This class converts a stream of raw text into a stream of tokens for
86 // the protocol definition parser to parse.  The tokens recognized are
87 // similar to those that make up the C language; see the TokenType enum for
88 // precise descriptions.  Whitespace and comments are skipped.  By default,
89 // C- and C++-style comments are recognized, but other styles can be used by
90 // calling set_comment_style().
91 class LIBPROTOBUF_EXPORT Tokenizer {
92  public:
93   // Construct a Tokenizer that reads and tokenizes text from the given
94   // input stream and writes errors to the given error_collector.
95   // The caller keeps ownership of input and error_collector.
96   Tokenizer(ZeroCopyInputStream* input, ErrorCollector* error_collector);
97   ~Tokenizer();
98 
99   enum TokenType {
100     TYPE_START,       // Next() has not yet been called.
101     TYPE_END,         // End of input reached.  "text" is empty.
102 
103     TYPE_IDENTIFIER,  // A sequence of letters, digits, and underscores, not
104                       // starting with a digit.  It is an error for a number
105                       // to be followed by an identifier with no space in
106                       // between.
107     TYPE_INTEGER,     // A sequence of digits representing an integer.  Normally
108                       // the digits are decimal, but a prefix of "0x" indicates
109                       // a hex number and a leading zero indicates octal, just
110                       // like with C numeric literals.  A leading negative sign
111                       // is NOT included in the token; it's up to the parser to
112                       // interpret the unary minus operator on its own.
113     TYPE_FLOAT,       // A floating point literal, with a fractional part and/or
114                       // an exponent.  Always in decimal.  Again, never
115                       // negative.
116     TYPE_STRING,      // A quoted sequence of escaped characters.  Either single
117                       // or double quotes can be used, but they must match.
118                       // A string literal cannot cross a line break.
119     TYPE_SYMBOL,      // Any other printable character, like '!' or '+'.
120                       // Symbols are always a single character, so "!+$%" is
121                       // four tokens.
122   };
123 
124   // Structure representing a token read from the token stream.
125   struct Token {
126     TokenType type;
127     string text;       // The exact text of the token as it appeared in
128                        // the input.  e.g. tokens of TYPE_STRING will still
129                        // be escaped and in quotes.
130 
131     // "line" and "column" specify the position of the first character of
132     // the token within the input stream.  They are zero-based.
133     int line;
134     ColumnNumber column;
135     ColumnNumber end_column;
136   };
137 
138   // Get the current token.  This is updated when Next() is called.  Before
139   // the first call to Next(), current() has type TYPE_START and no contents.
140   const Token& current();
141 
142   // Return the previous token -- i.e. what current() returned before the
143   // previous call to Next().
144   const Token& previous();
145 
146   // Advance to the next token.  Returns false if the end of the input is
147   // reached.
148   bool Next();
149 
150   // Like Next(), but also collects comments which appear between the previous
151   // and next tokens.
152   //
153   // Comments which appear to be attached to the previous token are stored
154   // in *prev_tailing_comments.  Comments which appear to be attached to the
155   // next token are stored in *next_leading_comments.  Comments appearing in
156   // between which do not appear to be attached to either will be added to
157   // detached_comments.  Any of these parameters can be NULL to simply discard
158   // the comments.
159   //
160   // A series of line comments appearing on consecutive lines, with no other
161   // tokens appearing on those lines, will be treated as a single comment.
162   //
163   // Only the comment content is returned; comment markers (e.g. //) are
164   // stripped out.  For block comments, leading whitespace and an asterisk will
165   // be stripped from the beginning of each line other than the first.  Newlines
166   // are included in the output.
167   //
168   // Examples:
169   //
170   //   optional int32 foo = 1;  // Comment attached to foo.
171   //   // Comment attached to bar.
172   //   optional int32 bar = 2;
173   //
174   //   optional string baz = 3;
175   //   // Comment attached to baz.
176   //   // Another line attached to baz.
177   //
178   //   // Comment attached to qux.
179   //   //
180   //   // Another line attached to qux.
181   //   optional double qux = 4;
182   //
183   //   // Detached comment.  This is not attached to qux or corge
184   //   // because there are blank lines separating it from both.
185   //
186   //   optional string corge = 5;
187   //   /* Block comment attached
188   //    * to corge.  Leading asterisks
189   //    * will be removed. */
190   //   /* Block comment attached to
191   //    * grault. */
192   //   optional int32 grault = 6;
193   bool NextWithComments(string* prev_trailing_comments,
194                         vector<string>* detached_comments,
195                         string* next_leading_comments);
196 
197   // Parse helpers ---------------------------------------------------
198 
199   // Parses a TYPE_FLOAT token.  This never fails, so long as the text actually
200   // comes from a TYPE_FLOAT token parsed by Tokenizer.  If it doesn't, the
201   // result is undefined (possibly an assert failure).
202   static double ParseFloat(const string& text);
203 
204   // Parses a TYPE_STRING token.  This never fails, so long as the text actually
205   // comes from a TYPE_STRING token parsed by Tokenizer.  If it doesn't, the
206   // result is undefined (possibly an assert failure).
207   static void ParseString(const string& text, string* output);
208 
209   // Identical to ParseString, but appends to output.
210   static void ParseStringAppend(const string& text, string* output);
211 
212   // Parses a TYPE_INTEGER token.  Returns false if the result would be
213   // greater than max_value.  Otherwise, returns true and sets *output to the
214   // result.  If the text is not from a Token of type TYPE_INTEGER originally
215   // parsed by a Tokenizer, the result is undefined (possibly an assert
216   // failure).
217   static bool ParseInteger(const string& text, uint64 max_value,
218                            uint64* output);
219 
220   // Options ---------------------------------------------------------
221 
222   // Set true to allow floats to be suffixed with the letter 'f'.  Tokens
223   // which would otherwise be integers but which have the 'f' suffix will be
224   // forced to be interpreted as floats.  For all other purposes, the 'f' is
225   // ignored.
set_allow_f_after_float(bool value)226   void set_allow_f_after_float(bool value) { allow_f_after_float_ = value; }
227 
228   // Valid values for set_comment_style().
229   enum CommentStyle {
230     // Line comments begin with "//", block comments are delimited by "/*" and
231     // "*/".
232     CPP_COMMENT_STYLE,
233     // Line comments begin with "#".  No way to write block comments.
234     SH_COMMENT_STYLE
235   };
236 
237   // Sets the comment style.
set_comment_style(CommentStyle style)238   void set_comment_style(CommentStyle style) { comment_style_ = style; }
239 
240   // Whether to require whitespace between a number and a field name.
241   // Default is true. Do not use this; for Google-internal cleanup only.
set_require_space_after_number(bool require)242   void set_require_space_after_number(bool require) {
243     require_space_after_number_ = require;
244   }
245 
246   // Whether to allow string literals to span multiple lines. Default is false.
247   // Do not use this; for Google-internal cleanup only.
set_allow_multiline_strings(bool allow)248   void set_allow_multiline_strings(bool allow) {
249     allow_multiline_strings_ = allow;
250   }
251 
252   // External helper: validate an identifier.
253   static bool IsIdentifier(const string& text);
254 
255   // -----------------------------------------------------------------
256  private:
257   GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(Tokenizer);
258 
259   Token current_;           // Returned by current().
260   Token previous_;          // Returned by previous().
261 
262   ZeroCopyInputStream* input_;
263   ErrorCollector* error_collector_;
264 
265   char current_char_;       // == buffer_[buffer_pos_], updated by NextChar().
266   const char* buffer_;      // Current buffer returned from input_.
267   int buffer_size_;         // Size of buffer_.
268   int buffer_pos_;          // Current position within the buffer.
269   bool read_error_;         // Did we previously encounter a read error?
270 
271   // Line and column number of current_char_ within the whole input stream.
272   int line_;
273   ColumnNumber column_;
274 
275   // String to which text should be appended as we advance through it.
276   // Call RecordTo(&str) to start recording and StopRecording() to stop.
277   // E.g. StartToken() calls RecordTo(&current_.text).  record_start_ is the
278   // position within the current buffer where recording started.
279   string* record_target_;
280   int record_start_;
281 
282   // Options.
283   bool allow_f_after_float_;
284   CommentStyle comment_style_;
285   bool require_space_after_number_;
286   bool allow_multiline_strings_;
287 
288   // Since we count columns we need to interpret tabs somehow.  We'll take
289   // the standard 8-character definition for lack of any way to do better.
290   // This must match the documentation of ColumnNumber.
291   static const int kTabWidth = 8;
292 
293   // -----------------------------------------------------------------
294   // Helper methods.
295 
296   // Consume this character and advance to the next one.
297   void NextChar();
298 
299   // Read a new buffer from the input.
300   void Refresh();
301 
302   inline void RecordTo(string* target);
303   inline void StopRecording();
304 
305   // Called when the current character is the first character of a new
306   // token (not including whitespace or comments).
307   inline void StartToken();
308   // Called when the current character is the first character after the
309   // end of the last token.  After this returns, current_.text will
310   // contain all text consumed since StartToken() was called.
311   inline void EndToken();
312 
313   // Convenience method to add an error at the current line and column.
AddError(const string & message)314   void AddError(const string& message) {
315     error_collector_->AddError(line_, column_, message);
316   }
317 
318   // -----------------------------------------------------------------
319   // The following four methods are used to consume tokens of specific
320   // types.  They are actually used to consume all characters *after*
321   // the first, since the calling function consumes the first character
322   // in order to decide what kind of token is being read.
323 
324   // Read and consume a string, ending when the given delimiter is
325   // consumed.
326   void ConsumeString(char delimiter);
327 
328   // Read and consume a number, returning TYPE_FLOAT or TYPE_INTEGER
329   // depending on what was read.  This needs to know if the first
330   // character was a zero in order to correctly recognize hex and octal
331   // numbers.
332   // It also needs to know if the first characted was a . to parse floating
333   // point correctly.
334   TokenType ConsumeNumber(bool started_with_zero, bool started_with_dot);
335 
336   // Consume the rest of a line.
337   void ConsumeLineComment(string* content);
338   // Consume until "*/".
339   void ConsumeBlockComment(string* content);
340 
341   enum NextCommentStatus {
342     // Started a line comment.
343     LINE_COMMENT,
344 
345     // Started a block comment.
346     BLOCK_COMMENT,
347 
348     // Consumed a slash, then realized it wasn't a comment.  current_ has
349     // been filled in with a slash token.  The caller should return it.
350     SLASH_NOT_COMMENT,
351 
352     // We do not appear to be starting a comment here.
353     NO_COMMENT
354   };
355 
356   // If we're at the start of a new comment, consume it and return what kind
357   // of comment it is.
358   NextCommentStatus TryConsumeCommentStart();
359 
360   // -----------------------------------------------------------------
361   // These helper methods make the parsing code more readable.  The
362   // "character classes" referred to are defined at the top of the .cc file.
363   // Basically it is a C++ class with one method:
364   //   static bool InClass(char c);
365   // The method returns true if c is a member of this "class", like "Letter"
366   // or "Digit".
367 
368   // Returns true if the current character is of the given character
369   // class, but does not consume anything.
370   template<typename CharacterClass>
371   inline bool LookingAt();
372 
373   // If the current character is in the given class, consume it and return
374   // true.  Otherwise return false.
375   // e.g. TryConsumeOne<Letter>()
376   template<typename CharacterClass>
377   inline bool TryConsumeOne();
378 
379   // Like above, but try to consume the specific character indicated.
380   inline bool TryConsume(char c);
381 
382   // Consume zero or more of the given character class.
383   template<typename CharacterClass>
384   inline void ConsumeZeroOrMore();
385 
386   // Consume one or more of the given character class or log the given
387   // error message.
388   // e.g. ConsumeOneOrMore<Digit>("Expected digits.");
389   template<typename CharacterClass>
390   inline void ConsumeOneOrMore(const char* error);
391 };
392 
393 // inline methods ====================================================
current()394 inline const Tokenizer::Token& Tokenizer::current() {
395   return current_;
396 }
397 
previous()398 inline const Tokenizer::Token& Tokenizer::previous() {
399   return previous_;
400 }
401 
ParseString(const string & text,string * output)402 inline void Tokenizer::ParseString(const string& text, string* output) {
403   output->clear();
404   ParseStringAppend(text, output);
405 }
406 
407 }  // namespace io
408 }  // namespace protobuf
409 
410 }  // namespace google
411 #endif  // GOOGLE_PROTOBUF_IO_TOKENIZER_H__
412