1 /* 2 * Copyright(C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0(the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http: // www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 *distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #include "VtsProfilingUtil.h" 17 18 #include <stdint.h> 19 20 #include "google/protobuf/io/coded_stream.h" 21 22 namespace android { 23 namespace vts { 24 25 bool writeOneDelimited(const google::protobuf::MessageLite& message, 26 google::protobuf::io::ZeroCopyOutputStream* out) { 27 // We create a new coded stream for each message. This is fast. 28 google::protobuf::io::CodedOutputStream output(out); 29 30 // Write the size. 31 const int size = message.ByteSize(); 32 output.WriteVarint32(size); 33 34 uint8_t* buffer = output.GetDirectBufferForNBytesAndAdvance(size); 35 if (buffer) { 36 // Optimization: The message fits in one buffer, so use the faster 37 // direct-to-array serialization path. 38 message.SerializeWithCachedSizesToArray(buffer); 39 } else { 40 // Slightly-slower path when the message is multiple buffers. 41 message.SerializeWithCachedSizes(&output); 42 if (output.HadError()) { 43 return false; 44 } 45 } 46 47 return true; 48 } 49 50 bool readOneDelimited(google::protobuf::MessageLite* message, 51 google::protobuf::io::ZeroCopyInputStream* in) { 52 // We create a new coded stream for each message. This is fast, 53 // and it makes sure the 64MB total size limit is imposed per-message rather 54 // than on the whole stream (See the CodedInputStream interface for more 55 // info on this limit). 56 google::protobuf::io::CodedInputStream input(in); 57 58 // Read the size. 59 uint32_t size; 60 if (!input.ReadVarint32(&size)) { 61 return false; 62 } 63 // Tell the stream not to read beyond that size. 64 const auto limit = input.PushLimit(size); 65 66 // Parse the message. 67 if (!message->MergeFromCodedStream(&input)) { 68 return false; 69 } 70 if (!input.ConsumedEntireMessage()) { 71 return false; 72 } 73 74 // Release the limit. 75 input.PopLimit(limit); 76 77 return true; 78 } 79 80 } // namespace vts 81 } // namespace android 82
