Index: third_party/protobuf/conformance/conformance_python.py |
diff --git a/third_party/protobuf/conformance/conformance_python.py b/third_party/protobuf/conformance/conformance_python.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..a490c8e8e54bd68e26ef4d10a5f4685dff51b2c4 |
--- /dev/null |
+++ b/third_party/protobuf/conformance/conformance_python.py |
@@ -0,0 +1,130 @@ |
+#!/usr/bin/env python |
+# |
+# Protocol Buffers - Google's data interchange format |
+# Copyright 2008 Google Inc. All rights reserved. |
+# https://developers.google.com/protocol-buffers/ |
+# |
+# 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. |
+ |
+"""A conformance test implementation for the Python protobuf library. |
+ |
+See conformance.proto for more information. |
+""" |
+ |
+import struct |
+import sys |
+import os |
+from google.protobuf import message |
+from google.protobuf import json_format |
+import conformance_pb2 |
+ |
+sys.stdout = os.fdopen(sys.stdout.fileno(), 'wb', 0) |
+sys.stdin = os.fdopen(sys.stdin.fileno(), 'rb', 0) |
+ |
+test_count = 0 |
+verbose = False |
+ |
+class ProtocolError(Exception): |
+ pass |
+ |
+def do_test(request): |
+ test_message = conformance_pb2.TestAllTypes() |
+ response = conformance_pb2.ConformanceResponse() |
+ test_message = conformance_pb2.TestAllTypes() |
+ |
+ try: |
+ if request.WhichOneof('payload') == 'protobuf_payload': |
+ try: |
+ test_message.ParseFromString(request.protobuf_payload) |
+ except message.DecodeError as e: |
+ response.parse_error = str(e) |
+ return response |
+ |
+ elif request.WhichOneof('payload') == 'json_payload': |
+ try: |
+ json_format.Parse(request.json_payload, test_message) |
+ except json_format.ParseError as e: |
+ response.parse_error = str(e) |
+ return response |
+ |
+ else: |
+ raise ProtocolError("Request didn't have payload.") |
+ |
+ if request.requested_output_format == conformance_pb2.UNSPECIFIED: |
+ raise ProtocolError("Unspecified output format") |
+ |
+ elif request.requested_output_format == conformance_pb2.PROTOBUF: |
+ response.protobuf_payload = test_message.SerializeToString() |
+ |
+ elif request.requested_output_format == conformance_pb2.JSON: |
+ response.json_payload = json_format.MessageToJson(test_message) |
+ |
+ except Exception as e: |
+ response.runtime_error = str(e) |
+ |
+ return response |
+ |
+def do_test_io(): |
+ length_bytes = sys.stdin.read(4) |
+ if len(length_bytes) == 0: |
+ return False # EOF |
+ elif len(length_bytes) != 4: |
+ raise IOError("I/O error") |
+ |
+ # "I" is "unsigned int", so this depends on running on a platform with |
+ # 32-bit "unsigned int" type. The Python struct module unfortunately |
+ # has no format specifier for uint32_t. |
+ length = struct.unpack("<I", length_bytes)[0] |
+ serialized_request = sys.stdin.read(length) |
+ if len(serialized_request) != length: |
+ raise IOError("I/O error") |
+ |
+ request = conformance_pb2.ConformanceRequest() |
+ request.ParseFromString(serialized_request) |
+ |
+ response = do_test(request) |
+ |
+ serialized_response = response.SerializeToString() |
+ sys.stdout.write(struct.pack("<I", len(serialized_response))) |
+ sys.stdout.write(serialized_response) |
+ sys.stdout.flush() |
+ |
+ if verbose: |
+ sys.stderr.write("conformance_python: request=%s, response=%s\n" % ( |
+ request.ShortDebugString().c_str(), |
+ response.ShortDebugString().c_str())) |
+ |
+ global test_count |
+ test_count += 1 |
+ |
+ return True |
+ |
+while True: |
+ if not do_test_io(): |
+ sys.stderr.write("conformance_python: received EOF from test runner " + |
+ "after %s tests, exiting\n" % (test_count)) |
+ sys.exit(0) |