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 26namespace aapt { 27 28class 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 69template <> 70inline 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