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 #include <errno.h>
32 #include <stdarg.h>
33 #include <unistd.h>
34
35 #include "conformance.pb.h"
36 #include <google/protobuf/util/json_util.h>
37 #include <google/protobuf/util/type_resolver_util.h>
38
39 using conformance::ConformanceRequest;
40 using conformance::ConformanceResponse;
41 using conformance::TestAllTypes;
42 using google::protobuf::Descriptor;
43 using google::protobuf::DescriptorPool;
44 using google::protobuf::internal::scoped_ptr;
45 using google::protobuf::util::BinaryToJsonString;
46 using google::protobuf::util::JsonToBinaryString;
47 using google::protobuf::util::NewTypeResolverForDescriptorPool;
48 using google::protobuf::util::Status;
49 using google::protobuf::util::TypeResolver;
50 using std::string;
51
52 static const char kTypeUrlPrefix[] = "type.googleapis.com";
53
GetTypeUrl(const Descriptor * message)54 static string GetTypeUrl(const Descriptor* message) {
55 return string(kTypeUrlPrefix) + "/" + message->full_name();
56 }
57
58 int test_count = 0;
59 bool verbose = false;
60 TypeResolver* type_resolver;
61 string* type_url;
62
63
CheckedRead(int fd,void * buf,size_t len)64 bool CheckedRead(int fd, void *buf, size_t len) {
65 size_t ofs = 0;
66 while (len > 0) {
67 ssize_t bytes_read = read(fd, (char*)buf + ofs, len);
68
69 if (bytes_read == 0) return false;
70
71 if (bytes_read < 0) {
72 GOOGLE_LOG(FATAL) << "Error reading from test runner: " << strerror(errno);
73 }
74
75 len -= bytes_read;
76 ofs += bytes_read;
77 }
78
79 return true;
80 }
81
CheckedWrite(int fd,const void * buf,size_t len)82 void CheckedWrite(int fd, const void *buf, size_t len) {
83 if (write(fd, buf, len) != len) {
84 GOOGLE_LOG(FATAL) << "Error writing to test runner: " << strerror(errno);
85 }
86 }
87
DoTest(const ConformanceRequest & request,ConformanceResponse * response)88 void DoTest(const ConformanceRequest& request, ConformanceResponse* response) {
89 TestAllTypes test_message;
90
91 switch (request.payload_case()) {
92 case ConformanceRequest::kProtobufPayload:
93 if (!test_message.ParseFromString(request.protobuf_payload())) {
94 // Getting parse details would involve something like:
95 // http://stackoverflow.com/questions/22121922/how-can-i-get-more-details-about-errors-generated-during-protobuf-parsing-c
96 response->set_parse_error("Parse error (no more details available).");
97 return;
98 }
99 break;
100
101 case ConformanceRequest::kJsonPayload: {
102 string proto_binary;
103 Status status = JsonToBinaryString(type_resolver, *type_url,
104 request.json_payload(), &proto_binary);
105 if (!status.ok()) {
106 response->set_parse_error(string("Parse error: ") +
107 status.error_message().as_string());
108 return;
109 }
110
111 if (!test_message.ParseFromString(proto_binary)) {
112 response->set_runtime_error(
113 "Parsing JSON generates invalid proto output.");
114 return;
115 }
116 break;
117 }
118
119 case ConformanceRequest::PAYLOAD_NOT_SET:
120 GOOGLE_LOG(FATAL) << "Request didn't have payload.";
121 break;
122 }
123
124 switch (request.requested_output_format()) {
125 case conformance::UNSPECIFIED:
126 GOOGLE_LOG(FATAL) << "Unspecified output format";
127 break;
128
129 case conformance::PROTOBUF:
130 GOOGLE_CHECK(
131 test_message.SerializeToString(response->mutable_protobuf_payload()));
132 break;
133
134 case conformance::JSON: {
135 string proto_binary;
136 GOOGLE_CHECK(test_message.SerializeToString(&proto_binary));
137 Status status = BinaryToJsonString(type_resolver, *type_url, proto_binary,
138 response->mutable_json_payload());
139 if (!status.ok()) {
140 response->set_serialize_error(
141 string("Failed to serialize JSON output: ") +
142 status.error_message().as_string());
143 return;
144 }
145 break;
146 }
147
148 default:
149 GOOGLE_LOG(FATAL) << "Unknown output format: "
150 << request.requested_output_format();
151 }
152 }
153
DoTestIo()154 bool DoTestIo() {
155 string serialized_input;
156 string serialized_output;
157 ConformanceRequest request;
158 ConformanceResponse response;
159 uint32_t bytes;
160
161 if (!CheckedRead(STDIN_FILENO, &bytes, sizeof(uint32_t))) {
162 // EOF.
163 return false;
164 }
165
166 serialized_input.resize(bytes);
167
168 if (!CheckedRead(STDIN_FILENO, (char*)serialized_input.c_str(), bytes)) {
169 GOOGLE_LOG(ERROR) << "Unexpected EOF on stdin. " << strerror(errno);
170 }
171
172 if (!request.ParseFromString(serialized_input)) {
173 GOOGLE_LOG(FATAL) << "Parse of ConformanceRequest proto failed.";
174 return false;
175 }
176
177 DoTest(request, &response);
178
179 response.SerializeToString(&serialized_output);
180
181 bytes = serialized_output.size();
182 CheckedWrite(STDOUT_FILENO, &bytes, sizeof(uint32_t));
183 CheckedWrite(STDOUT_FILENO, serialized_output.c_str(), bytes);
184
185 if (verbose) {
186 fprintf(stderr, "conformance-cpp: request=%s, response=%s\n",
187 request.ShortDebugString().c_str(),
188 response.ShortDebugString().c_str());
189 }
190
191 test_count++;
192
193 return true;
194 }
195
main()196 int main() {
197 type_resolver = NewTypeResolverForDescriptorPool(
198 kTypeUrlPrefix, DescriptorPool::generated_pool());
199 type_url = new string(GetTypeUrl(TestAllTypes::descriptor()));
200 while (1) {
201 if (!DoTestIo()) {
202 fprintf(stderr, "conformance-cpp: received EOF from test runner "
203 "after %d tests, exiting\n", test_count);
204 return 0;
205 }
206 }
207 }
208