/*
* Copyright (C) 2013 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 <android-base/logging.h>
#include "base/common_art_test.h"
#include "base/macros.h"
#include "base/unix_file/fd_file.h"
#include "stream/buffered_output_stream.h"
#include "stream/file_output_stream.h"
#include "stream/vector_output_stream.h"
namespace art HIDDEN {
namespace linker {
class OutputStreamTest :
public CommonArtTest {
protected:
void CheckOffset(off_t expected) {
off_t actual = output_stream_->Seek(
0, kSeekCurrent);
EXPECT_EQ(expected, actual);
}
void SetOutputStream(OutputStream& output_stream) {
output_stream_ = &output_stream;
}
void GenerateTestOutput() {
EXPECT_EQ(
3, output_stream_->Seek(
3, kSeekCurrent));
CheckOffset(
3);
EXPECT_EQ(
2, output_stream_->Seek(
2, kSeekSet));
CheckOffset(
2);
uint8_t buf[] = {
1,
2,
3,
4,
5,
6,
7,
8,
9 };
EXPECT_TRUE(output_stream_->WriteFully(buf,
2));
CheckOffset(
4);
EXPECT_EQ(
6, output_stream_->Seek(
2, kSeekEnd));
CheckOffset(
6);
EXPECT_TRUE(output_stream_->WriteFully(buf,
4));
CheckOffset(
10);
EXPECT_TRUE(output_stream_->WriteFully(buf,
6));
EXPECT_TRUE(output_stream_->Flush());
}
void CheckTestOutput(
const std::vector<uint8_t>& actual) {
uint8_t expected[] = {
0,
0,
1,
2,
0,
0,
1,
2,
3,
4,
1,
2,
3,
4,
5,
6
};
EXPECT_EQ(
sizeof(expected), actual.size());
EXPECT_EQ(
0, memcmp(expected, &actual[
0], actual.size()));
}
OutputStream* output_stream_;
};
TEST_F(OutputStreamTest, File) {
ScratchFile tmp;
FileOutputStream output_stream(tmp.GetFile());
SetOutputStream(output_stream);
GenerateTestOutput();
std::unique_ptr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str()));
EXPECT_TRUE(in.get() != nullptr);
std::vector<uint8_t> actual(in->GetLength());
bool readSuccess = in->ReadFully(&actual[
0], actual.size());
EXPECT_TRUE(readSuccess);
CheckTestOutput(actual);
}
TEST_F(OutputStreamTest, Buffered) {
ScratchFile tmp;
{
BufferedOutputStream buffered_output_stream(std::make_unique<FileOutputStream>(tmp.
GetFile()));
SetOutputStream(buffered_output_stream);
GenerateTestOutput();
}
std::unique_ptr<File> in(OS::OpenFileForReading(tmp.GetFilename().c_str()));
EXPECT_TRUE(in.get() != nullptr);
std::vector<uint8_t> actual(in->GetLength());
bool readSuccess = in->ReadFully(&actual[0], actual.size());
EXPECT_TRUE(readSuccess);
CheckTestOutput(actual);
}
TEST_F(OutputStreamTest, Vector) {
std::vector<uint8_t> output;
VectorOutputStream output_stream("test vector output", &output);
SetOutputStream(output_stream);
GenerateTestOutput();
CheckTestOutput(output);
}
TEST_F(OutputStreamTest, BufferedFlush) {
struct CheckingOutputStream : OutputStream {
CheckingOutputStream()
: OutputStream("fake-location"),
flush_called(false) { }
~CheckingOutputStream() override {}
bool WriteFully([[maybe_unused]] const void* buffer,
[[maybe_unused]] size_t byte_count) override {
LOG(FATAL) << "UNREACHABLE";
UNREACHABLE();
}
off_t Seek([[maybe_unused]] off_t offset, [[maybe_unused]] Whence whence) override {
LOG(FATAL) << "UNREACHABLE";
UNREACHABLE();
}
bool Flush() override {
flush_called = true;
return true;
}
bool flush_called;
};
std::unique_ptr<CheckingOutputStream> cos = std::make_unique<CheckingOutputStream>();
CheckingOutputStream* checking_output_stream = cos.get();
BufferedOutputStream buffered(std::move(cos));
ASSERT_FALSE(checking_output_stream->flush_called);
bool flush_result = buffered.Flush();
ASSERT_TRUE(flush_result);
ASSERT_TRUE(checking_output_stream->flush_called);
}
} // namespace linker
} // namespace art
