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_FORMAT_BINARY_CHUNKWRITER_H 18 #define AAPT_FORMAT_BINARY_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: ChunkWriter(BigBuffer * buffer)30 explicit inline ChunkWriter(BigBuffer* buffer) : buffer_(buffer) { 31 } 32 ChunkWriter(ChunkWriter&&) = default; 33 ChunkWriter& operator=(ChunkWriter&&) = default; 34 35 template <typename T> StartChunk(uint16_t type)36 inline T* StartChunk(uint16_t type) { 37 start_size_ = buffer_->size(); 38 T* chunk = buffer_->NextBlock<T>(); 39 header_ = &chunk->header; 40 header_->type = util::HostToDevice16(type); 41 header_->headerSize = util::HostToDevice16(sizeof(T)); 42 return chunk; 43 } 44 45 template <typename T> 46 inline T* NextBlock(size_t count = 1) { 47 return buffer_->NextBlock<T>(count); 48 } 49 buffer()50 inline BigBuffer* buffer() { 51 return buffer_; 52 } 53 chunk_header()54 inline android::ResChunk_header* chunk_header() { 55 return header_; 56 } 57 size()58 inline size_t size() { 59 return buffer_->size() - start_size_; 60 } 61 Finish()62 inline android::ResChunk_header* Finish() { 63 buffer_->Align4(); 64 header_->size = util::HostToDevice32(buffer_->size() - start_size_); 65 return header_; 66 } 67 68 private: 69 DISALLOW_COPY_AND_ASSIGN(ChunkWriter); 70 71 BigBuffer* buffer_; 72 size_t start_size_ = 0; 73 android::ResChunk_header* header_ = nullptr; 74 }; 75 76 template <> StartChunk(uint16_t type)77inline android::ResChunk_header* ChunkWriter::StartChunk(uint16_t type) { 78 start_size_ = buffer_->size(); 79 header_ = buffer_->NextBlock<android::ResChunk_header>(); 80 header_->type = util::HostToDevice16(type); 81 header_->headerSize = util::HostToDevice16(sizeof(android::ResChunk_header)); 82 return header_; 83 } 84 85 } // namespace aapt 86 87 #endif /* AAPT_FORMAT_BINARY_CHUNKWRITER_H */ 88