|
|
|
import 'dart:io';
|
|
|
|
|
|
|
|
import 'package:pb_runtime/pb_runtime.dart' as pb;
|
|
|
|
import 'package:third_party.protobuf/test_messages_proto2.upb.dart';
|
|
|
|
import 'package:third_party.protobuf/test_messages_proto3.upb.dart';
|
|
|
|
import 'package:third_party.protobuf.conformance/conformance.upb.dart';
|
|
|
|
|
|
|
|
List<int>? readFromStdin(int numBytes) {
|
|
|
|
final bytesOut = <int>[];
|
|
|
|
for (var i = 0; i < numBytes; i++) {
|
|
|
|
final byte = stdin.readByteSync();
|
|
|
|
if (byte == -1) return null;
|
|
|
|
bytesOut.add(byte);
|
|
|
|
}
|
|
|
|
return bytesOut;
|
|
|
|
}
|
|
|
|
|
|
|
|
int readLittleEndianIntFromStdin() {
|
|
|
|
final buf = readFromStdin(4);
|
|
|
|
if (buf == null) return -1;
|
|
|
|
return (buf[0] & 0xff) |
|
|
|
|
((buf[1] & 0xff) << 8) |
|
|
|
|
((buf[2] & 0xff) << 16) |
|
|
|
|
((buf[3] & 0xff) << 24);
|
|
|
|
}
|
|
|
|
|
|
|
|
void writeLittleEndianIntToStdout(int value) {
|
|
|
|
final buf = <int>[];
|
|
|
|
buf.add(value & 0xff);
|
|
|
|
buf.add((value >> 8) & 0xff);
|
|
|
|
buf.add((value >> 16) & 0xff);
|
|
|
|
buf.add((value >> 24) & 0xff);
|
|
|
|
stdout.add(buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
ConformanceResponse doTest(ConformanceRequest request) {
|
|
|
|
final response = ConformanceResponse();
|
|
|
|
pb.GeneratedMessage? testMessage;
|
|
|
|
final messageType = request.messageType;
|
|
|
|
final isProto3 =
|
|
|
|
messageType == 'protobuf_test_messages.proto3.TestAllTypesProto3';
|
|
|
|
|
|
|
|
if (!isProto3 &&
|
|
|
|
messageType != 'protobuf_test_messages.proto2.TestAllTypesProto2') {
|
|
|
|
throw ArgumentError('Invalid message type $messageType');
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (request.whichPayload) {
|
|
|
|
case ConformanceRequest_payload.protobufPayload:
|
|
|
|
try {
|
|
|
|
testMessage = isProto3
|
|
|
|
? TestAllTypesProto3.fromBuffer(request.protobufPayload)
|
|
|
|
: TestAllTypesProto2.fromBuffer(request.protobufPayload);
|
|
|
|
} catch (e) {
|
|
|
|
final parseErrorResponse = ConformanceResponse();
|
|
|
|
parseErrorResponse.parseError = '$e';
|
|
|
|
return parseErrorResponse;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
response.skipped = 'Only protobuf payload input is supported';
|
|
|
|
return response;
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (request.requestedOutputFormat) {
|
|
|
|
case WireFormat.PROTOBUF:
|
|
|
|
try {
|
|
|
|
response.protobufPayload = pb.GeneratedMessage.toBuffer(testMessage);
|
|
|
|
} catch (e) {
|
|
|
|
response.serializeError = '$e';
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
response.skipped = 'Only protobuf payload output is supported';
|
|
|
|
}
|
|
|
|
|
|
|
|
return response;
|
|
|
|
}
|
|
|
|
|
|
|
|
Future<bool> doTestIo() async {
|
|
|
|
final msgLength = readLittleEndianIntFromStdin();
|
|
|
|
if (msgLength == -1) return false; // EOF
|
|
|
|
final serializedMsg = readFromStdin(msgLength);
|
|
|
|
if (serializedMsg == null) {
|
|
|
|
throw 'Unexpected EOF from test program.';
|
|
|
|
}
|
|
|
|
final request = ConformanceRequest.fromBuffer(serializedMsg);
|
|
|
|
final response = doTest(request);
|
|
|
|
final serializedOutput = pb.GeneratedMessage.toBuffer(response);
|
|
|
|
writeLittleEndianIntToStdout(serializedOutput.length);
|
|
|
|
stdout.add(serializedOutput);
|
|
|
|
await stdout.flush();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
void main() async {
|
|
|
|
await pb.initGlobalInstance();
|
|
|
|
var testCount = 0;
|
|
|
|
while (await doTestIo()) {
|
|
|
|
testCount++;
|
|
|
|
}
|
|
|
|
stderr.writeln(
|
|
|
|
'ConformanceDart: received EOF from test runner after $testCount tests');
|
|
|
|
}
|