06460ef0d7
This also enables an AAPT behavior that CTS tests have come to depend on. Small files that compress negatively (get larger) are stored uncompressed. Some CTS tests assume this and try to open these files by mmapping them, which is only possible if they are uncompressed. Bug: 35461578 Test: make aapt2_tests Change-Id: Id622a6150fe72477ad65d67d1bad897a8ee2ffb9
82 lines
2.7 KiB
C++
82 lines
2.7 KiB
C++
/*
|
|
* Copyright (C) 2015 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef AAPT_FLATTEN_ARCHIVE_H
|
|
#define AAPT_FLATTEN_ARCHIVE_H
|
|
|
|
#include <fstream>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "androidfw/StringPiece.h"
|
|
#include "google/protobuf/io/zero_copy_stream_impl_lite.h"
|
|
|
|
#include "Diagnostics.h"
|
|
#include "io/Io.h"
|
|
#include "util/BigBuffer.h"
|
|
#include "util/Files.h"
|
|
|
|
namespace aapt {
|
|
|
|
struct ArchiveEntry {
|
|
enum : uint32_t {
|
|
kCompress = 0x01,
|
|
kAlign = 0x02,
|
|
};
|
|
|
|
std::string path;
|
|
uint32_t flags;
|
|
size_t uncompressed_size;
|
|
};
|
|
|
|
class IArchiveWriter : public ::google::protobuf::io::CopyingOutputStream {
|
|
public:
|
|
virtual ~IArchiveWriter() = default;
|
|
|
|
virtual bool WriteFile(const android::StringPiece& path, uint32_t flags, io::InputStream* in) = 0;
|
|
|
|
// Starts a new entry and allows caller to write bytes to it sequentially.
|
|
// Only use StartEntry if code you do not control needs to write to a CopyingOutputStream.
|
|
// Prefer WriteFile instead of manually calling StartEntry/FinishEntry.
|
|
virtual bool StartEntry(const android::StringPiece& path, uint32_t flags) = 0;
|
|
|
|
// Called to finish writing an entry previously started by StartEntry.
|
|
// Prefer WriteFile instead of manually calling StartEntry/FinishEntry.
|
|
virtual bool FinishEntry() = 0;
|
|
|
|
// CopyingOutputStream implementation that allows sequential writes to this archive. Only
|
|
// valid between calls to StartEntry and FinishEntry.
|
|
virtual bool Write(const void* buffer, int size) = 0;
|
|
|
|
// Returns true if there was an error writing to the archive.
|
|
// The resulting error message can be retrieved from GetError().
|
|
virtual bool HadError() const = 0;
|
|
|
|
// Returns the error message if HadError() returns true.
|
|
virtual std::string GetError() const = 0;
|
|
};
|
|
|
|
std::unique_ptr<IArchiveWriter> CreateDirectoryArchiveWriter(IDiagnostics* diag,
|
|
const android::StringPiece& path);
|
|
|
|
std::unique_ptr<IArchiveWriter> CreateZipFileArchiveWriter(IDiagnostics* diag,
|
|
const android::StringPiece& path);
|
|
|
|
} // namespace aapt
|
|
|
|
#endif /* AAPT_FLATTEN_ARCHIVE_H */
|