1 /* 2 * Copyright (C) 2015 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_CHUNKWRITER_H 18 #define AAPT_FLATTEN_CHUNKWRITER_H 19 20 #include "android-base/macros.h" 21 #include "androidfw/ResourceTypes.h" 22 23 #include "util/BigBuffer.h" 24 #include "util/Util.h" 25 26 namespace aapt { 27 28 class ChunkWriter { 29 public: 30 explicit inline ChunkWriter(BigBuffer* buffer) : buffer_(buffer) {} 31 ChunkWriter(ChunkWriter&&) = default; 32 ChunkWriter& operator=(ChunkWriter&&) = default; 33 34 template <typename T> 35 inline T* StartChunk(uint16_t type) { 36 start_size_ = buffer_->size(); 37 T* chunk = buffer_->NextBlock<T>(); 38 header_ = &chunk->header; 39 header_->type = util::HostToDevice16(type); 40 header_->headerSize = util::HostToDevice16(sizeof(T)); 41 return chunk; 42 } 43 44 template <typename T> 45 inline T* NextBlock(size_t count = 1) { 46 return buffer_->NextBlock<T>(count); 47 } 48 49 inline BigBuffer* buffer() { return buffer_; } 50 51 inline android::ResChunk_header* chunk_header() { return header_; } 52 53 inline size_t size() { return buffer_->size() - start_size_; } 54 55 inline android::ResChunk_header* Finish() { 56 buffer_->Align4(); 57 header_->size = util::HostToDevice32(buffer_->size() - start_size_); 58 return header_; 59 } 60 61 private: 62 DISALLOW_COPY_AND_ASSIGN(ChunkWriter); 63 64 BigBuffer* buffer_; 65 size_t start_size_ = 0; 66 android::ResChunk_header* header_ = nullptr; 67 }; 68 69 template <> 70 inline android::ResChunk_header* ChunkWriter::StartChunk(uint16_t type) { 71 start_size_ = buffer_->size(); 72 header_ = buffer_->NextBlock<android::ResChunk_header>(); 73 header_->type = util::HostToDevice16(type); 74 header_->headerSize = util::HostToDevice16(sizeof(android::ResChunk_header)); 75 return header_; 76 } 77 78 } // namespace aapt 79 80 #endif /* AAPT_FLATTEN_CHUNKWRITER_H */ 81