1 /* 2 * Copyright (C) 2016 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 17 #ifndef AAPT_FLATTEN_TABLEPROTOSERIALIZER_H 18 #define AAPT_FLATTEN_TABLEPROTOSERIALIZER_H 19 20 #include "Diagnostics.h" 21 #include "ResourceTable.h" 22 #include "Source.h" 23 #include "proto/ProtoHelpers.h" 24 25 #include <android-base/macros.h> 26 #include <google/protobuf/io/coded_stream.h> 27 #include <google/protobuf/io/zero_copy_stream_impl_lite.h> 28 29 namespace aapt { 30 31 std::unique_ptr<pb::ResourceTable> serializeTableToPb(ResourceTable* table); 32 std::unique_ptr<ResourceTable> deserializeTableFromPb(const pb::ResourceTable& pbTable, 33 const Source& source, 34 IDiagnostics* diag); 35 36 std::unique_ptr<pb::CompiledFile> serializeCompiledFileToPb(const ResourceFile& file); 37 std::unique_ptr<ResourceFile> deserializeCompiledFileFromPb(const pb::CompiledFile& pbFile, 38 const Source& source, 39 IDiagnostics* diag); 40 41 class CompiledFileOutputStream : public google::protobuf::io::CopyingOutputStream { 42 public: 43 CompiledFileOutputStream(google::protobuf::io::ZeroCopyOutputStream* out, 44 pb::CompiledFile* pbFile); 45 bool Write(const void* data, int size) override; 46 bool Finish(); 47 48 private: 49 bool ensureFileWritten(); 50 51 google::protobuf::io::CodedOutputStream mOut; 52 pb::CompiledFile* mPbFile; 53 54 DISALLOW_COPY_AND_ASSIGN(CompiledFileOutputStream); 55 }; 56 57 class CompiledFileInputStream { 58 public: 59 CompiledFileInputStream(const void* data, size_t size); 60 61 const pb::CompiledFile* CompiledFile(); 62 63 const void* data(); 64 size_t size(); 65 66 private: 67 google::protobuf::io::CodedInputStream mIn; 68 std::unique_ptr<pb::CompiledFile> mPbFile; 69 const uint8_t* mData; 70 size_t mSize; 71 72 DISALLOW_COPY_AND_ASSIGN(CompiledFileInputStream); 73 }; 74 75 } // namespace aapt 76 77 #endif /* AAPT_FLATTEN_TABLEPROTOSERIALIZER_H */ 78