android_frameworks_base/tools/aapt2/io/BigBufferStreams.cpp
Adam Lesinski 06460ef0d7 AAPT2: Fix up file IO
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
2017-03-20 16:53:46 -07:00

87 lines
2.0 KiB
C++

/*
* Copyright (C) 2017 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.
*/
#include "io/BigBufferInputStream.h"
#include "io/BigBufferOutputStream.h"
namespace aapt {
namespace io {
//
// BigBufferInputStream
//
bool BigBufferInputStream::Next(const void** data, size_t* size) {
if (iter_ == buffer_->end()) {
return false;
}
if (offset_ == iter_->size) {
++iter_;
if (iter_ == buffer_->end()) {
return false;
}
offset_ = 0;
}
*data = iter_->buffer.get() + offset_;
*size = iter_->size - offset_;
bytes_read_ += iter_->size - offset_;
offset_ = iter_->size;
return true;
}
void BigBufferInputStream::BackUp(size_t count) {
if (count > offset_) {
bytes_read_ -= offset_;
offset_ = 0;
} else {
offset_ -= count;
bytes_read_ -= count;
}
}
bool BigBufferInputStream::CanRewind() const { return true; }
bool BigBufferInputStream::Rewind() {
iter_ = buffer_->begin();
offset_ = 0;
bytes_read_ = 0;
return true;
}
size_t BigBufferInputStream::ByteCount() const { return bytes_read_; }
bool BigBufferInputStream::HadError() const { return false; }
//
// BigBufferOutputStream
//
bool BigBufferOutputStream::Next(void** data, size_t* size) {
*data = buffer_->NextBlock(size);
return true;
}
void BigBufferOutputStream::BackUp(size_t count) { buffer_->BackUp(count); }
size_t BigBufferOutputStream::ByteCount() const { return buffer_->size(); }
bool BigBufferOutputStream::HadError() const { return false; }
} // namespace io
} // namespace aapt