| Index: third_party/google/protobuf/internal/decoder.py
|
| diff --git a/third_party/google/protobuf/internal/decoder.py b/third_party/google/protobuf/internal/decoder.py
|
| deleted file mode 100644
|
| index 651ee0d4ad469ccf92044242266d188c00d0c6ef..0000000000000000000000000000000000000000
|
| --- a/third_party/google/protobuf/internal/decoder.py
|
| +++ /dev/null
|
| @@ -1,831 +0,0 @@
|
| -# Protocol Buffers - Google's data interchange format
|
| -# Copyright 2008 Google Inc. All rights reserved.
|
| -# http://code.google.com/p/protobuf/
|
| -#
|
| -# Redistribution and use in source and binary forms, with or without
|
| -# modification, are permitted provided that the following conditions are
|
| -# met:
|
| -#
|
| -# * Redistributions of source code must retain the above copyright
|
| -# notice, this list of conditions and the following disclaimer.
|
| -# * Redistributions in binary form must reproduce the above
|
| -# copyright notice, this list of conditions and the following disclaimer
|
| -# in the documentation and/or other materials provided with the
|
| -# distribution.
|
| -# * Neither the name of Google Inc. nor the names of its
|
| -# contributors may be used to endorse or promote products derived from
|
| -# this software without specific prior written permission.
|
| -#
|
| -# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| -# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| -# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| -# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| -# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| -# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| -# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| -# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| -# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| -# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| -# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| -
|
| -#PY25 compatible for GAE.
|
| -#
|
| -# Copyright 2009 Google Inc. All Rights Reserved.
|
| -
|
| -"""Code for decoding protocol buffer primitives.
|
| -
|
| -This code is very similar to encoder.py -- read the docs for that module first.
|
| -
|
| -A "decoder" is a function with the signature:
|
| - Decode(buffer, pos, end, message, field_dict)
|
| -The arguments are:
|
| - buffer: The string containing the encoded message.
|
| - pos: The current position in the string.
|
| - end: The position in the string where the current message ends. May be
|
| - less than len(buffer) if we're reading a sub-message.
|
| - message: The message object into which we're parsing.
|
| - field_dict: message._fields (avoids a hashtable lookup).
|
| -The decoder reads the field and stores it into field_dict, returning the new
|
| -buffer position. A decoder for a repeated field may proactively decode all of
|
| -the elements of that field, if they appear consecutively.
|
| -
|
| -Note that decoders may throw any of the following:
|
| - IndexError: Indicates a truncated message.
|
| - struct.error: Unpacking of a fixed-width field failed.
|
| - message.DecodeError: Other errors.
|
| -
|
| -Decoders are expected to raise an exception if they are called with pos > end.
|
| -This allows callers to be lax about bounds checking: it's fineto read past
|
| -"end" as long as you are sure that someone else will notice and throw an
|
| -exception later on.
|
| -
|
| -Something up the call stack is expected to catch IndexError and struct.error
|
| -and convert them to message.DecodeError.
|
| -
|
| -Decoders are constructed using decoder constructors with the signature:
|
| - MakeDecoder(field_number, is_repeated, is_packed, key, new_default)
|
| -The arguments are:
|
| - field_number: The field number of the field we want to decode.
|
| - is_repeated: Is the field a repeated field? (bool)
|
| - is_packed: Is the field a packed field? (bool)
|
| - key: The key to use when looking up the field within field_dict.
|
| - (This is actually the FieldDescriptor but nothing in this
|
| - file should depend on that.)
|
| - new_default: A function which takes a message object as a parameter and
|
| - returns a new instance of the default value for this field.
|
| - (This is called for repeated fields and sub-messages, when an
|
| - instance does not already exist.)
|
| -
|
| -As with encoders, we define a decoder constructor for every type of field.
|
| -Then, for every field of every message class we construct an actual decoder.
|
| -That decoder goes into a dict indexed by tag, so when we decode a message
|
| -we repeatedly read a tag, look up the corresponding decoder, and invoke it.
|
| -"""
|
| -
|
| -__author__ = 'kenton@google.com (Kenton Varda)'
|
| -
|
| -import struct
|
| -import sys ##PY25
|
| -_PY2 = sys.version_info[0] < 3 ##PY25
|
| -from google.protobuf.internal import encoder
|
| -from google.protobuf.internal import wire_format
|
| -from google.protobuf import message
|
| -
|
| -
|
| -# This will overflow and thus become IEEE-754 "infinity". We would use
|
| -# "float('inf')" but it doesn't work on Windows pre-Python-2.6.
|
| -_POS_INF = 1e10000
|
| -_NEG_INF = -_POS_INF
|
| -_NAN = _POS_INF * 0
|
| -
|
| -
|
| -# This is not for optimization, but rather to avoid conflicts with local
|
| -# variables named "message".
|
| -_DecodeError = message.DecodeError
|
| -
|
| -
|
| -def _VarintDecoder(mask, result_type):
|
| - """Return an encoder for a basic varint value (does not include tag).
|
| -
|
| - Decoded values will be bitwise-anded with the given mask before being
|
| - returned, e.g. to limit them to 32 bits. The returned decoder does not
|
| - take the usual "end" parameter -- the caller is expected to do bounds checking
|
| - after the fact (often the caller can defer such checking until later). The
|
| - decoder returns a (value, new_pos) pair.
|
| - """
|
| -
|
| - local_ord = ord
|
| - py2 = _PY2 ##PY25
|
| -##!PY25 py2 = str is bytes
|
| - def DecodeVarint(buffer, pos):
|
| - result = 0
|
| - shift = 0
|
| - while 1:
|
| - b = local_ord(buffer[pos]) if py2 else buffer[pos]
|
| - result |= ((b & 0x7f) << shift)
|
| - pos += 1
|
| - if not (b & 0x80):
|
| - result &= mask
|
| - result = result_type(result)
|
| - return (result, pos)
|
| - shift += 7
|
| - if shift >= 64:
|
| - raise _DecodeError('Too many bytes when decoding varint.')
|
| - return DecodeVarint
|
| -
|
| -
|
| -def _SignedVarintDecoder(mask, result_type):
|
| - """Like _VarintDecoder() but decodes signed values."""
|
| -
|
| - local_ord = ord
|
| - py2 = _PY2 ##PY25
|
| -##!PY25 py2 = str is bytes
|
| - def DecodeVarint(buffer, pos):
|
| - result = 0
|
| - shift = 0
|
| - while 1:
|
| - b = local_ord(buffer[pos]) if py2 else buffer[pos]
|
| - result |= ((b & 0x7f) << shift)
|
| - pos += 1
|
| - if not (b & 0x80):
|
| - if result > 0x7fffffffffffffff:
|
| - result -= (1 << 64)
|
| - result |= ~mask
|
| - else:
|
| - result &= mask
|
| - result = result_type(result)
|
| - return (result, pos)
|
| - shift += 7
|
| - if shift >= 64:
|
| - raise _DecodeError('Too many bytes when decoding varint.')
|
| - return DecodeVarint
|
| -
|
| -# We force 32-bit values to int and 64-bit values to long to make
|
| -# alternate implementations where the distinction is more significant
|
| -# (e.g. the C++ implementation) simpler.
|
| -
|
| -_DecodeVarint = _VarintDecoder((1 << 64) - 1, long)
|
| -_DecodeSignedVarint = _SignedVarintDecoder((1 << 64) - 1, long)
|
| -
|
| -# Use these versions for values which must be limited to 32 bits.
|
| -_DecodeVarint32 = _VarintDecoder((1 << 32) - 1, int)
|
| -_DecodeSignedVarint32 = _SignedVarintDecoder((1 << 32) - 1, int)
|
| -
|
| -
|
| -def ReadTag(buffer, pos):
|
| - """Read a tag from the buffer, and return a (tag_bytes, new_pos) tuple.
|
| -
|
| - We return the raw bytes of the tag rather than decoding them. The raw
|
| - bytes can then be used to look up the proper decoder. This effectively allows
|
| - us to trade some work that would be done in pure-python (decoding a varint)
|
| - for work that is done in C (searching for a byte string in a hash table).
|
| - In a low-level language it would be much cheaper to decode the varint and
|
| - use that, but not in Python.
|
| - """
|
| -
|
| - py2 = _PY2 ##PY25
|
| -##!PY25 py2 = str is bytes
|
| - start = pos
|
| - while (ord(buffer[pos]) if py2 else buffer[pos]) & 0x80:
|
| - pos += 1
|
| - pos += 1
|
| - return (buffer[start:pos], pos)
|
| -
|
| -
|
| -# --------------------------------------------------------------------
|
| -
|
| -
|
| -def _SimpleDecoder(wire_type, decode_value):
|
| - """Return a constructor for a decoder for fields of a particular type.
|
| -
|
| - Args:
|
| - wire_type: The field's wire type.
|
| - decode_value: A function which decodes an individual value, e.g.
|
| - _DecodeVarint()
|
| - """
|
| -
|
| - def SpecificDecoder(field_number, is_repeated, is_packed, key, new_default):
|
| - if is_packed:
|
| - local_DecodeVarint = _DecodeVarint
|
| - def DecodePackedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - (endpoint, pos) = local_DecodeVarint(buffer, pos)
|
| - endpoint += pos
|
| - if endpoint > end:
|
| - raise _DecodeError('Truncated message.')
|
| - while pos < endpoint:
|
| - (element, pos) = decode_value(buffer, pos)
|
| - value.append(element)
|
| - if pos > endpoint:
|
| - del value[-1] # Discard corrupt value.
|
| - raise _DecodeError('Packed element was truncated.')
|
| - return pos
|
| - return DecodePackedField
|
| - elif is_repeated:
|
| - tag_bytes = encoder.TagBytes(field_number, wire_type)
|
| - tag_len = len(tag_bytes)
|
| - def DecodeRepeatedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - while 1:
|
| - (element, new_pos) = decode_value(buffer, pos)
|
| - value.append(element)
|
| - # Predict that the next tag is another copy of the same repeated
|
| - # field.
|
| - pos = new_pos + tag_len
|
| - if buffer[new_pos:pos] != tag_bytes or new_pos >= end:
|
| - # Prediction failed. Return.
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - return new_pos
|
| - return DecodeRepeatedField
|
| - else:
|
| - def DecodeField(buffer, pos, end, message, field_dict):
|
| - (field_dict[key], pos) = decode_value(buffer, pos)
|
| - if pos > end:
|
| - del field_dict[key] # Discard corrupt value.
|
| - raise _DecodeError('Truncated message.')
|
| - return pos
|
| - return DecodeField
|
| -
|
| - return SpecificDecoder
|
| -
|
| -
|
| -def _ModifiedDecoder(wire_type, decode_value, modify_value):
|
| - """Like SimpleDecoder but additionally invokes modify_value on every value
|
| - before storing it. Usually modify_value is ZigZagDecode.
|
| - """
|
| -
|
| - # Reusing _SimpleDecoder is slightly slower than copying a bunch of code, but
|
| - # not enough to make a significant difference.
|
| -
|
| - def InnerDecode(buffer, pos):
|
| - (result, new_pos) = decode_value(buffer, pos)
|
| - return (modify_value(result), new_pos)
|
| - return _SimpleDecoder(wire_type, InnerDecode)
|
| -
|
| -
|
| -def _StructPackDecoder(wire_type, format):
|
| - """Return a constructor for a decoder for a fixed-width field.
|
| -
|
| - Args:
|
| - wire_type: The field's wire type.
|
| - format: The format string to pass to struct.unpack().
|
| - """
|
| -
|
| - value_size = struct.calcsize(format)
|
| - local_unpack = struct.unpack
|
| -
|
| - # Reusing _SimpleDecoder is slightly slower than copying a bunch of code, but
|
| - # not enough to make a significant difference.
|
| -
|
| - # Note that we expect someone up-stack to catch struct.error and convert
|
| - # it to _DecodeError -- this way we don't have to set up exception-
|
| - # handling blocks every time we parse one value.
|
| -
|
| - def InnerDecode(buffer, pos):
|
| - new_pos = pos + value_size
|
| - result = local_unpack(format, buffer[pos:new_pos])[0]
|
| - return (result, new_pos)
|
| - return _SimpleDecoder(wire_type, InnerDecode)
|
| -
|
| -
|
| -def _FloatDecoder():
|
| - """Returns a decoder for a float field.
|
| -
|
| - This code works around a bug in struct.unpack for non-finite 32-bit
|
| - floating-point values.
|
| - """
|
| -
|
| - local_unpack = struct.unpack
|
| - b = (lambda x:x) if _PY2 else lambda x:x.encode('latin1') ##PY25
|
| -
|
| - def InnerDecode(buffer, pos):
|
| - # We expect a 32-bit value in little-endian byte order. Bit 1 is the sign
|
| - # bit, bits 2-9 represent the exponent, and bits 10-32 are the significand.
|
| - new_pos = pos + 4
|
| - float_bytes = buffer[pos:new_pos]
|
| -
|
| - # If this value has all its exponent bits set, then it's non-finite.
|
| - # In Python 2.4, struct.unpack will convert it to a finite 64-bit value.
|
| - # To avoid that, we parse it specially.
|
| - if ((float_bytes[3:4] in b('\x7F\xFF')) ##PY25
|
| -##!PY25 if ((float_bytes[3:4] in b'\x7F\xFF')
|
| - and (float_bytes[2:3] >= b('\x80'))): ##PY25
|
| -##!PY25 and (float_bytes[2:3] >= b'\x80')):
|
| - # If at least one significand bit is set...
|
| - if float_bytes[0:3] != b('\x00\x00\x80'): ##PY25
|
| -##!PY25 if float_bytes[0:3] != b'\x00\x00\x80':
|
| - return (_NAN, new_pos)
|
| - # If sign bit is set...
|
| - if float_bytes[3:4] == b('\xFF'): ##PY25
|
| -##!PY25 if float_bytes[3:4] == b'\xFF':
|
| - return (_NEG_INF, new_pos)
|
| - return (_POS_INF, new_pos)
|
| -
|
| - # Note that we expect someone up-stack to catch struct.error and convert
|
| - # it to _DecodeError -- this way we don't have to set up exception-
|
| - # handling blocks every time we parse one value.
|
| - result = local_unpack('<f', float_bytes)[0]
|
| - return (result, new_pos)
|
| - return _SimpleDecoder(wire_format.WIRETYPE_FIXED32, InnerDecode)
|
| -
|
| -
|
| -def _DoubleDecoder():
|
| - """Returns a decoder for a double field.
|
| -
|
| - This code works around a bug in struct.unpack for not-a-number.
|
| - """
|
| -
|
| - local_unpack = struct.unpack
|
| - b = (lambda x:x) if _PY2 else lambda x:x.encode('latin1') ##PY25
|
| -
|
| - def InnerDecode(buffer, pos):
|
| - # We expect a 64-bit value in little-endian byte order. Bit 1 is the sign
|
| - # bit, bits 2-12 represent the exponent, and bits 13-64 are the significand.
|
| - new_pos = pos + 8
|
| - double_bytes = buffer[pos:new_pos]
|
| -
|
| - # If this value has all its exponent bits set and at least one significand
|
| - # bit set, it's not a number. In Python 2.4, struct.unpack will treat it
|
| - # as inf or -inf. To avoid that, we treat it specially.
|
| -##!PY25 if ((double_bytes[7:8] in b'\x7F\xFF')
|
| -##!PY25 and (double_bytes[6:7] >= b'\xF0')
|
| -##!PY25 and (double_bytes[0:7] != b'\x00\x00\x00\x00\x00\x00\xF0')):
|
| - if ((double_bytes[7:8] in b('\x7F\xFF')) ##PY25
|
| - and (double_bytes[6:7] >= b('\xF0')) ##PY25
|
| - and (double_bytes[0:7] != b('\x00\x00\x00\x00\x00\x00\xF0'))): ##PY25
|
| - return (_NAN, new_pos)
|
| -
|
| - # Note that we expect someone up-stack to catch struct.error and convert
|
| - # it to _DecodeError -- this way we don't have to set up exception-
|
| - # handling blocks every time we parse one value.
|
| - result = local_unpack('<d', double_bytes)[0]
|
| - return (result, new_pos)
|
| - return _SimpleDecoder(wire_format.WIRETYPE_FIXED64, InnerDecode)
|
| -
|
| -
|
| -def EnumDecoder(field_number, is_repeated, is_packed, key, new_default):
|
| - enum_type = key.enum_type
|
| - if is_packed:
|
| - local_DecodeVarint = _DecodeVarint
|
| - def DecodePackedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - (endpoint, pos) = local_DecodeVarint(buffer, pos)
|
| - endpoint += pos
|
| - if endpoint > end:
|
| - raise _DecodeError('Truncated message.')
|
| - while pos < endpoint:
|
| - value_start_pos = pos
|
| - (element, pos) = _DecodeSignedVarint32(buffer, pos)
|
| - if element in enum_type.values_by_number:
|
| - value.append(element)
|
| - else:
|
| - if not message._unknown_fields:
|
| - message._unknown_fields = []
|
| - tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_VARINT)
|
| - message._unknown_fields.append(
|
| - (tag_bytes, buffer[value_start_pos:pos]))
|
| - if pos > endpoint:
|
| - if element in enum_type.values_by_number:
|
| - del value[-1] # Discard corrupt value.
|
| - else:
|
| - del message._unknown_fields[-1]
|
| - raise _DecodeError('Packed element was truncated.')
|
| - return pos
|
| - return DecodePackedField
|
| - elif is_repeated:
|
| - tag_bytes = encoder.TagBytes(field_number, wire_format.WIRETYPE_VARINT)
|
| - tag_len = len(tag_bytes)
|
| - def DecodeRepeatedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - while 1:
|
| - (element, new_pos) = _DecodeSignedVarint32(buffer, pos)
|
| - if element in enum_type.values_by_number:
|
| - value.append(element)
|
| - else:
|
| - if not message._unknown_fields:
|
| - message._unknown_fields = []
|
| - message._unknown_fields.append(
|
| - (tag_bytes, buffer[pos:new_pos]))
|
| - # Predict that the next tag is another copy of the same repeated
|
| - # field.
|
| - pos = new_pos + tag_len
|
| - if buffer[new_pos:pos] != tag_bytes or new_pos >= end:
|
| - # Prediction failed. Return.
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - return new_pos
|
| - return DecodeRepeatedField
|
| - else:
|
| - def DecodeField(buffer, pos, end, message, field_dict):
|
| - value_start_pos = pos
|
| - (enum_value, pos) = _DecodeSignedVarint32(buffer, pos)
|
| - if pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - if enum_value in enum_type.values_by_number:
|
| - field_dict[key] = enum_value
|
| - else:
|
| - if not message._unknown_fields:
|
| - message._unknown_fields = []
|
| - tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_VARINT)
|
| - message._unknown_fields.append(
|
| - (tag_bytes, buffer[value_start_pos:pos]))
|
| - return pos
|
| - return DecodeField
|
| -
|
| -
|
| -# --------------------------------------------------------------------
|
| -
|
| -
|
| -Int32Decoder = _SimpleDecoder(
|
| - wire_format.WIRETYPE_VARINT, _DecodeSignedVarint32)
|
| -
|
| -Int64Decoder = _SimpleDecoder(
|
| - wire_format.WIRETYPE_VARINT, _DecodeSignedVarint)
|
| -
|
| -UInt32Decoder = _SimpleDecoder(wire_format.WIRETYPE_VARINT, _DecodeVarint32)
|
| -UInt64Decoder = _SimpleDecoder(wire_format.WIRETYPE_VARINT, _DecodeVarint)
|
| -
|
| -SInt32Decoder = _ModifiedDecoder(
|
| - wire_format.WIRETYPE_VARINT, _DecodeVarint32, wire_format.ZigZagDecode)
|
| -SInt64Decoder = _ModifiedDecoder(
|
| - wire_format.WIRETYPE_VARINT, _DecodeVarint, wire_format.ZigZagDecode)
|
| -
|
| -# Note that Python conveniently guarantees that when using the '<' prefix on
|
| -# formats, they will also have the same size across all platforms (as opposed
|
| -# to without the prefix, where their sizes depend on the C compiler's basic
|
| -# type sizes).
|
| -Fixed32Decoder = _StructPackDecoder(wire_format.WIRETYPE_FIXED32, '<I')
|
| -Fixed64Decoder = _StructPackDecoder(wire_format.WIRETYPE_FIXED64, '<Q')
|
| -SFixed32Decoder = _StructPackDecoder(wire_format.WIRETYPE_FIXED32, '<i')
|
| -SFixed64Decoder = _StructPackDecoder(wire_format.WIRETYPE_FIXED64, '<q')
|
| -FloatDecoder = _FloatDecoder()
|
| -DoubleDecoder = _DoubleDecoder()
|
| -
|
| -BoolDecoder = _ModifiedDecoder(
|
| - wire_format.WIRETYPE_VARINT, _DecodeVarint, bool)
|
| -
|
| -
|
| -def StringDecoder(field_number, is_repeated, is_packed, key, new_default):
|
| - """Returns a decoder for a string field."""
|
| -
|
| - local_DecodeVarint = _DecodeVarint
|
| - local_unicode = unicode
|
| -
|
| - def _ConvertToUnicode(byte_str):
|
| - try:
|
| - return local_unicode(byte_str, 'utf-8')
|
| - except UnicodeDecodeError, e:
|
| - # add more information to the error message and re-raise it.
|
| - e.reason = '%s in field: %s' % (e, key.full_name)
|
| - raise
|
| -
|
| - assert not is_packed
|
| - if is_repeated:
|
| - tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_LENGTH_DELIMITED)
|
| - tag_len = len(tag_bytes)
|
| - def DecodeRepeatedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - while 1:
|
| - (size, pos) = local_DecodeVarint(buffer, pos)
|
| - new_pos = pos + size
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated string.')
|
| - value.append(_ConvertToUnicode(buffer[pos:new_pos]))
|
| - # Predict that the next tag is another copy of the same repeated field.
|
| - pos = new_pos + tag_len
|
| - if buffer[new_pos:pos] != tag_bytes or new_pos == end:
|
| - # Prediction failed. Return.
|
| - return new_pos
|
| - return DecodeRepeatedField
|
| - else:
|
| - def DecodeField(buffer, pos, end, message, field_dict):
|
| - (size, pos) = local_DecodeVarint(buffer, pos)
|
| - new_pos = pos + size
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated string.')
|
| - field_dict[key] = _ConvertToUnicode(buffer[pos:new_pos])
|
| - return new_pos
|
| - return DecodeField
|
| -
|
| -
|
| -def BytesDecoder(field_number, is_repeated, is_packed, key, new_default):
|
| - """Returns a decoder for a bytes field."""
|
| -
|
| - local_DecodeVarint = _DecodeVarint
|
| -
|
| - assert not is_packed
|
| - if is_repeated:
|
| - tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_LENGTH_DELIMITED)
|
| - tag_len = len(tag_bytes)
|
| - def DecodeRepeatedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - while 1:
|
| - (size, pos) = local_DecodeVarint(buffer, pos)
|
| - new_pos = pos + size
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated string.')
|
| - value.append(buffer[pos:new_pos])
|
| - # Predict that the next tag is another copy of the same repeated field.
|
| - pos = new_pos + tag_len
|
| - if buffer[new_pos:pos] != tag_bytes or new_pos == end:
|
| - # Prediction failed. Return.
|
| - return new_pos
|
| - return DecodeRepeatedField
|
| - else:
|
| - def DecodeField(buffer, pos, end, message, field_dict):
|
| - (size, pos) = local_DecodeVarint(buffer, pos)
|
| - new_pos = pos + size
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated string.')
|
| - field_dict[key] = buffer[pos:new_pos]
|
| - return new_pos
|
| - return DecodeField
|
| -
|
| -
|
| -def GroupDecoder(field_number, is_repeated, is_packed, key, new_default):
|
| - """Returns a decoder for a group field."""
|
| -
|
| - end_tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_END_GROUP)
|
| - end_tag_len = len(end_tag_bytes)
|
| -
|
| - assert not is_packed
|
| - if is_repeated:
|
| - tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_START_GROUP)
|
| - tag_len = len(tag_bytes)
|
| - def DecodeRepeatedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - while 1:
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - # Read sub-message.
|
| - pos = value.add()._InternalParse(buffer, pos, end)
|
| - # Read end tag.
|
| - new_pos = pos+end_tag_len
|
| - if buffer[pos:new_pos] != end_tag_bytes or new_pos > end:
|
| - raise _DecodeError('Missing group end tag.')
|
| - # Predict that the next tag is another copy of the same repeated field.
|
| - pos = new_pos + tag_len
|
| - if buffer[new_pos:pos] != tag_bytes or new_pos == end:
|
| - # Prediction failed. Return.
|
| - return new_pos
|
| - return DecodeRepeatedField
|
| - else:
|
| - def DecodeField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - # Read sub-message.
|
| - pos = value._InternalParse(buffer, pos, end)
|
| - # Read end tag.
|
| - new_pos = pos+end_tag_len
|
| - if buffer[pos:new_pos] != end_tag_bytes or new_pos > end:
|
| - raise _DecodeError('Missing group end tag.')
|
| - return new_pos
|
| - return DecodeField
|
| -
|
| -
|
| -def MessageDecoder(field_number, is_repeated, is_packed, key, new_default):
|
| - """Returns a decoder for a message field."""
|
| -
|
| - local_DecodeVarint = _DecodeVarint
|
| -
|
| - assert not is_packed
|
| - if is_repeated:
|
| - tag_bytes = encoder.TagBytes(field_number,
|
| - wire_format.WIRETYPE_LENGTH_DELIMITED)
|
| - tag_len = len(tag_bytes)
|
| - def DecodeRepeatedField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - while 1:
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - # Read length.
|
| - (size, pos) = local_DecodeVarint(buffer, pos)
|
| - new_pos = pos + size
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - # Read sub-message.
|
| - if value.add()._InternalParse(buffer, pos, new_pos) != new_pos:
|
| - # The only reason _InternalParse would return early is if it
|
| - # encountered an end-group tag.
|
| - raise _DecodeError('Unexpected end-group tag.')
|
| - # Predict that the next tag is another copy of the same repeated field.
|
| - pos = new_pos + tag_len
|
| - if buffer[new_pos:pos] != tag_bytes or new_pos == end:
|
| - # Prediction failed. Return.
|
| - return new_pos
|
| - return DecodeRepeatedField
|
| - else:
|
| - def DecodeField(buffer, pos, end, message, field_dict):
|
| - value = field_dict.get(key)
|
| - if value is None:
|
| - value = field_dict.setdefault(key, new_default(message))
|
| - # Read length.
|
| - (size, pos) = local_DecodeVarint(buffer, pos)
|
| - new_pos = pos + size
|
| - if new_pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - # Read sub-message.
|
| - if value._InternalParse(buffer, pos, new_pos) != new_pos:
|
| - # The only reason _InternalParse would return early is if it encountered
|
| - # an end-group tag.
|
| - raise _DecodeError('Unexpected end-group tag.')
|
| - return new_pos
|
| - return DecodeField
|
| -
|
| -
|
| -# --------------------------------------------------------------------
|
| -
|
| -MESSAGE_SET_ITEM_TAG = encoder.TagBytes(1, wire_format.WIRETYPE_START_GROUP)
|
| -
|
| -def MessageSetItemDecoder(extensions_by_number):
|
| - """Returns a decoder for a MessageSet item.
|
| -
|
| - The parameter is the _extensions_by_number map for the message class.
|
| -
|
| - The message set message looks like this:
|
| - message MessageSet {
|
| - repeated group Item = 1 {
|
| - required int32 type_id = 2;
|
| - required string message = 3;
|
| - }
|
| - }
|
| - """
|
| -
|
| - type_id_tag_bytes = encoder.TagBytes(2, wire_format.WIRETYPE_VARINT)
|
| - message_tag_bytes = encoder.TagBytes(3, wire_format.WIRETYPE_LENGTH_DELIMITED)
|
| - item_end_tag_bytes = encoder.TagBytes(1, wire_format.WIRETYPE_END_GROUP)
|
| -
|
| - local_ReadTag = ReadTag
|
| - local_DecodeVarint = _DecodeVarint
|
| - local_SkipField = SkipField
|
| -
|
| - def DecodeItem(buffer, pos, end, message, field_dict):
|
| - message_set_item_start = pos
|
| - type_id = -1
|
| - message_start = -1
|
| - message_end = -1
|
| -
|
| - # Technically, type_id and message can appear in any order, so we need
|
| - # a little loop here.
|
| - while 1:
|
| - (tag_bytes, pos) = local_ReadTag(buffer, pos)
|
| - if tag_bytes == type_id_tag_bytes:
|
| - (type_id, pos) = local_DecodeVarint(buffer, pos)
|
| - elif tag_bytes == message_tag_bytes:
|
| - (size, message_start) = local_DecodeVarint(buffer, pos)
|
| - pos = message_end = message_start + size
|
| - elif tag_bytes == item_end_tag_bytes:
|
| - break
|
| - else:
|
| - pos = SkipField(buffer, pos, end, tag_bytes)
|
| - if pos == -1:
|
| - raise _DecodeError('Missing group end tag.')
|
| -
|
| - if pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| -
|
| - if type_id == -1:
|
| - raise _DecodeError('MessageSet item missing type_id.')
|
| - if message_start == -1:
|
| - raise _DecodeError('MessageSet item missing message.')
|
| -
|
| - extension = extensions_by_number.get(type_id)
|
| - if extension is not None:
|
| - value = field_dict.get(extension)
|
| - if value is None:
|
| - value = field_dict.setdefault(
|
| - extension, extension.message_type._concrete_class())
|
| - if value._InternalParse(buffer, message_start,message_end) != message_end:
|
| - # The only reason _InternalParse would return early is if it encountered
|
| - # an end-group tag.
|
| - raise _DecodeError('Unexpected end-group tag.')
|
| - else:
|
| - if not message._unknown_fields:
|
| - message._unknown_fields = []
|
| - message._unknown_fields.append((MESSAGE_SET_ITEM_TAG,
|
| - buffer[message_set_item_start:pos]))
|
| -
|
| - return pos
|
| -
|
| - return DecodeItem
|
| -
|
| -# --------------------------------------------------------------------
|
| -# Optimization is not as heavy here because calls to SkipField() are rare,
|
| -# except for handling end-group tags.
|
| -
|
| -def _SkipVarint(buffer, pos, end):
|
| - """Skip a varint value. Returns the new position."""
|
| - # Previously ord(buffer[pos]) raised IndexError when pos is out of range.
|
| - # With this code, ord(b'') raises TypeError. Both are handled in
|
| - # python_message.py to generate a 'Truncated message' error.
|
| - while ord(buffer[pos:pos+1]) & 0x80:
|
| - pos += 1
|
| - pos += 1
|
| - if pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - return pos
|
| -
|
| -def _SkipFixed64(buffer, pos, end):
|
| - """Skip a fixed64 value. Returns the new position."""
|
| -
|
| - pos += 8
|
| - if pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - return pos
|
| -
|
| -def _SkipLengthDelimited(buffer, pos, end):
|
| - """Skip a length-delimited value. Returns the new position."""
|
| -
|
| - (size, pos) = _DecodeVarint(buffer, pos)
|
| - pos += size
|
| - if pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - return pos
|
| -
|
| -def _SkipGroup(buffer, pos, end):
|
| - """Skip sub-group. Returns the new position."""
|
| -
|
| - while 1:
|
| - (tag_bytes, pos) = ReadTag(buffer, pos)
|
| - new_pos = SkipField(buffer, pos, end, tag_bytes)
|
| - if new_pos == -1:
|
| - return pos
|
| - pos = new_pos
|
| -
|
| -def _EndGroup(buffer, pos, end):
|
| - """Skipping an END_GROUP tag returns -1 to tell the parent loop to break."""
|
| -
|
| - return -1
|
| -
|
| -def _SkipFixed32(buffer, pos, end):
|
| - """Skip a fixed32 value. Returns the new position."""
|
| -
|
| - pos += 4
|
| - if pos > end:
|
| - raise _DecodeError('Truncated message.')
|
| - return pos
|
| -
|
| -def _RaiseInvalidWireType(buffer, pos, end):
|
| - """Skip function for unknown wire types. Raises an exception."""
|
| -
|
| - raise _DecodeError('Tag had invalid wire type.')
|
| -
|
| -def _FieldSkipper():
|
| - """Constructs the SkipField function."""
|
| -
|
| - WIRETYPE_TO_SKIPPER = [
|
| - _SkipVarint,
|
| - _SkipFixed64,
|
| - _SkipLengthDelimited,
|
| - _SkipGroup,
|
| - _EndGroup,
|
| - _SkipFixed32,
|
| - _RaiseInvalidWireType,
|
| - _RaiseInvalidWireType,
|
| - ]
|
| -
|
| - wiretype_mask = wire_format.TAG_TYPE_MASK
|
| -
|
| - def SkipField(buffer, pos, end, tag_bytes):
|
| - """Skips a field with the specified tag.
|
| -
|
| - |pos| should point to the byte immediately after the tag.
|
| -
|
| - Returns:
|
| - The new position (after the tag value), or -1 if the tag is an end-group
|
| - tag (in which case the calling loop should break).
|
| - """
|
| -
|
| - # The wire type is always in the first byte since varints are little-endian.
|
| - wire_type = ord(tag_bytes[0:1]) & wiretype_mask
|
| - return WIRETYPE_TO_SKIPPER[wire_type](buffer, pos, end)
|
| -
|
| - return SkipField
|
| -
|
| -SkipField = _FieldSkipper()
|
|
|