11ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski/* 21ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * Copyright (C) 2015 The Android Open Source Project 31ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * 41ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * Licensed under the Apache License, Version 2.0 (the "License"); 51ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * you may not use this file except in compliance with the License. 61ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * You may obtain a copy of the License at 71ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * 81ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * http://www.apache.org/licenses/LICENSE-2.0 91ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * 101ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * Unless required by applicable law or agreed to in writing, software 111ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * distributed under the License is distributed on an "AS IS" BASIS, 121ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * See the License for the specific language governing permissions and 141ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski * limitations under the License. 151ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski */ 161ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 171ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#ifndef AAPT_FLATTEN_ARCHIVE_H 181ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#define AAPT_FLATTEN_ARCHIVE_H 191ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 20a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski#include "Diagnostics.h" 211ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include "util/BigBuffer.h" 221ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include "util/Files.h" 231ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include "util/StringPiece.h" 241ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 2559e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinski#include <google/protobuf/io/zero_copy_stream_impl_lite.h> 261ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include <fstream> 271ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include <memory> 281ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include <string> 291ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#include <vector> 301ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 311ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinskinamespace aapt { 321ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 331ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinskistruct ArchiveEntry { 341ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski enum : uint32_t { 351ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski kCompress = 0x01, 361ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski kAlign = 0x02, 371ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski }; 381ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 391ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski std::string path; 401ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski uint32_t flags; 411ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski size_t uncompressedSize; 421ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski}; 431ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 4459e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinskistruct IArchiveWriter : public google::protobuf::io::CopyingOutputStream { 451ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski virtual ~IArchiveWriter() = default; 461ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 47a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski virtual bool startEntry(const StringPiece& path, uint32_t flags) = 0; 48a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski virtual bool writeEntry(const BigBuffer& buffer) = 0; 49a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski virtual bool writeEntry(const void* data, size_t len) = 0; 50a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski virtual bool finishEntry() = 0; 5159e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinski 5259e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinski // CopyingOutputStream implementations. 5359e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinski bool Write(const void* buffer, int size) override { 5459e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinski return writeEntry(buffer, size); 5559e04c6f92da584b322c87072f18e6cab4de4c60Adam Lesinski } 561ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski}; 571ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 58a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinskistd::unique_ptr<IArchiveWriter> createDirectoryArchiveWriter(IDiagnostics* diag, 59a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski const StringPiece& path); 601ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 61a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinskistd::unique_ptr<IArchiveWriter> createZipFileArchiveWriter(IDiagnostics* diag, 62a40e972fdaa68fc486ff90a319195515819068b2Adam Lesinski const StringPiece& path); 631ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 641ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski} // namespace aapt 651ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski 661ab598f46c3ff520a67f9d80194847741f3467abAdam Lesinski#endif /* AAPT_FLATTEN_ARCHIVE_H */ 67