Pavol Rusnak
7 years ago
parent
2d643031ac
commit
f73fc33439
@ -1,36 +0,0 @@
|
||||
# This file is part of the TREZOR project.
|
||||
#
|
||||
# Copyright (C) 2012-2016 Marek Palatinus <slush@satoshilabs.com>
|
||||
# Copyright (C) 2012-2016 Pavol Rusnak <stick@satoshilabs.com>
|
||||
#
|
||||
# This library is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Lesser General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# This library is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Lesser General Public License
|
||||
# along with this library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import common
|
||||
import unittest
|
||||
|
||||
from trezorlib.protobuf_json import pb2json
|
||||
import trezorlib.messages_pb2 as msg
|
||||
|
||||
|
||||
class TestProtobufJson(unittest.TestCase):
|
||||
|
||||
def test_pb2json(self):
|
||||
m = msg.Features()
|
||||
m.device_id = '1234'
|
||||
j = pb2json(m)
|
||||
self.assertEqual(j, {'device_id': u'1234'})
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
@ -1,163 +0,0 @@
|
||||
# JSON serialization support for Google's protobuf Messages
|
||||
# Copyright (c) 2009, Paul Dovbush
|
||||
# All rights reserved.
|
||||
# http://code.google.com/p/protobuf-json/
|
||||
#
|
||||
# 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 <ORGANIZATION> 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.
|
||||
|
||||
'''
|
||||
Provide serialization and de-serialization of Google's protobuf Messages into/from JSON format.
|
||||
'''
|
||||
|
||||
# groups are deprecated and not supported;
|
||||
# Note that preservation of unknown fields is currently not available for Python (c) google docs
|
||||
# extensions is not supported from 0.0.5 (due to gpb2.3 changes)
|
||||
|
||||
__version__ = '0.0.6'
|
||||
__author__ = 'Paul Dovbush <dpp@dpp.su>'
|
||||
|
||||
|
||||
from google.protobuf.descriptor import FieldDescriptor as FD
|
||||
from functools import partial
|
||||
|
||||
|
||||
class ParseError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
def json2pb(pb, js, useFieldNumber=False):
|
||||
''' convert JSON string to google.protobuf.descriptor instance '''
|
||||
for field in pb.DESCRIPTOR.fields:
|
||||
if useFieldNumber:
|
||||
key = field.number
|
||||
else:
|
||||
key = field.name
|
||||
if key not in js:
|
||||
continue
|
||||
if field.type == FD.TYPE_MESSAGE:
|
||||
pass
|
||||
elif field.type in _js2ftype:
|
||||
ftype = _js2ftype[field.type]
|
||||
else:
|
||||
raise ParseError("Field %s.%s of type '%d' is not supported" % (pb.__class__.__name__, field.name, field.type, ))
|
||||
value = js[key]
|
||||
if field.label == FD.LABEL_REPEATED:
|
||||
pb_value = getattr(pb, field.name, None)
|
||||
for v in value:
|
||||
if field.type == FD.TYPE_MESSAGE:
|
||||
json2pb(pb_value.add(), v, useFieldNumber=useFieldNumber)
|
||||
else:
|
||||
pb_value.append(ftype(v))
|
||||
else:
|
||||
if field.type == FD.TYPE_MESSAGE:
|
||||
json2pb(getattr(pb, field.name, None), value, useFieldNumber=useFieldNumber)
|
||||
else:
|
||||
setattr(pb, field.name, ftype(value))
|
||||
return pb
|
||||
|
||||
|
||||
def pb2json(pb, useFieldNumber=False):
|
||||
''' convert google.protobuf.descriptor instance to JSON string '''
|
||||
js = {}
|
||||
# fields = pb.DESCRIPTOR.fields # all fields
|
||||
fields = pb.ListFields() # only filled (including extensions)
|
||||
for field, value in fields:
|
||||
if useFieldNumber:
|
||||
key = field.number
|
||||
else:
|
||||
key = field.name
|
||||
if field.type == FD.TYPE_MESSAGE:
|
||||
ftype = partial(pb2json, useFieldNumber=useFieldNumber)
|
||||
# ---- monkey patching ----
|
||||
elif field.type == FD.TYPE_ENUM:
|
||||
def ftype(x):
|
||||
return field.enum_type.values[x].name
|
||||
# ---- end of monkey patching ----
|
||||
elif field.type in _ftype2js:
|
||||
ftype = _ftype2js[field.type]
|
||||
else:
|
||||
raise ParseError("Field %s.%s of type '%d' is not supported" % (pb.__class__.__name__, field.name, field.type, ))
|
||||
if field.label == FD.LABEL_REPEATED:
|
||||
js_value = []
|
||||
for v in value:
|
||||
js_value.append(ftype(v))
|
||||
else:
|
||||
js_value = ftype(value)
|
||||
js[key] = js_value
|
||||
return js
|
||||
|
||||
|
||||
_ftype2js = {
|
||||
FD.TYPE_DOUBLE: float,
|
||||
FD.TYPE_FLOAT: float,
|
||||
FD.TYPE_INT64: int,
|
||||
FD.TYPE_UINT64: int,
|
||||
FD.TYPE_INT32: int,
|
||||
FD.TYPE_FIXED64: float,
|
||||
FD.TYPE_FIXED32: float,
|
||||
FD.TYPE_BOOL: bool,
|
||||
FD.TYPE_STRING: str,
|
||||
# FD.TYPE_MESSAGE: pb2json, #handled specially
|
||||
FD.TYPE_BYTES: lambda x: x.encode('string_escape'),
|
||||
FD.TYPE_UINT32: int,
|
||||
FD.TYPE_ENUM: int,
|
||||
FD.TYPE_SFIXED32: float,
|
||||
FD.TYPE_SFIXED64: float,
|
||||
FD.TYPE_SINT32: int,
|
||||
FD.TYPE_SINT64: int,
|
||||
}
|
||||
|
||||
_js2ftype = {
|
||||
FD.TYPE_DOUBLE: float,
|
||||
FD.TYPE_FLOAT: float,
|
||||
FD.TYPE_INT64: int,
|
||||
FD.TYPE_UINT64: int,
|
||||
FD.TYPE_INT32: int,
|
||||
FD.TYPE_FIXED64: float,
|
||||
FD.TYPE_FIXED32: float,
|
||||
FD.TYPE_BOOL: bool,
|
||||
FD.TYPE_STRING: str,
|
||||
# FD.TYPE_MESSAGE: json2pb, # handled specially
|
||||
FD.TYPE_BYTES: lambda x: x.decode('string_escape'),
|
||||
FD.TYPE_UINT32: int,
|
||||
FD.TYPE_ENUM: int,
|
||||
FD.TYPE_SFIXED32: float,
|
||||
FD.TYPE_SFIXED64: float,
|
||||
FD.TYPE_SINT32: int,
|
||||
FD.TYPE_SINT64: int,
|
||||
}
|
||||
|
||||
# more monkey patching
|
||||
|
||||
import binascii
|
||||
from . import types_pb2 as types
|
||||
|
||||
_ftype2js[FD.TYPE_BYTES] = lambda x: binascii.hexlify(x)
|
||||
del _ftype2js[FD.TYPE_ENUM] # handled specially
|
||||
|
||||
_js2ftype[FD.TYPE_BYTES] = lambda x: binascii.unhexlify(x)
|
||||
_js2ftype[FD.TYPE_ENUM] = lambda x: getattr(types, x)
|
||||
Loading…
Reference in new issue