swh:1:snp:b03b5b26525b84343e26e9b1b952c64cf35589c0
Tip revision: a8ec74d083068cd5e1ebed86fd8254630617b592 authored by Thai Duong on 14 July 2020, 08:36:23 UTC
Merge branch '1.4' of github.com:google/tink into 1.4
Merge branch '1.4' of github.com:google/tink into 1.4
Tip revision: a8ec74d
file_input_stream_test.cc
// Copyright 2018 Google Inc.
//
// 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 "tink/util/file_input_stream.h"
#include "gtest/gtest.h"
#include "absl/memory/memory.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/string_view.h"
#include "tink/util/test_util.h"
namespace crypto {
namespace tink {
namespace {
// Reads the specified 'input_stream' until no more bytes can be read,
// and puts the read bytes into 'contents'.
// Returns the status of the last input_stream->Next()-operation.
util::Status ReadTillEnd(util::FileInputStream* input_stream,
std::string* contents) {
contents->clear();
const void* buffer;
auto next_result = input_stream->Next(&buffer);
while (next_result.ok()) {
contents->append(static_cast<const char*>(buffer),
next_result.ValueOrDie());
next_result = input_stream->Next(&buffer);
}
return next_result.status();
}
class FileInputStreamTest : public ::testing::Test {
};
TEST_F(FileInputStreamTest, testReadingStreams) {
for (auto stream_size : {0, 10, 100, 1000, 10000, 100000, 1000000}) {
SCOPED_TRACE(absl::StrCat("stream_size = ", stream_size));
std::string file_contents;
std::string filename = absl::StrCat(stream_size, "_reading_test.bin");
int input_fd =
test::GetTestFileDescriptor(filename, stream_size, &file_contents);
EXPECT_EQ(stream_size, file_contents.size());
auto input_stream = absl::make_unique<util::FileInputStream>(input_fd);
std::string stream_contents;
auto status = ReadTillEnd(input_stream.get(), &stream_contents);
EXPECT_EQ(util::error::OUT_OF_RANGE, status.error_code());
EXPECT_EQ("EOF", status.error_message());
EXPECT_EQ(file_contents, stream_contents);
}
}
TEST_F(FileInputStreamTest, testCustomBufferSizes) {
int stream_size = 100000;
for (auto buffer_size : {1, 10, 100, 1000, 10000}) {
SCOPED_TRACE(absl::StrCat("buffer_size = ", buffer_size));
std::string file_contents;
std::string filename = absl::StrCat(buffer_size, "_buffer_size_test.bin");
int input_fd =
test::GetTestFileDescriptor(filename, stream_size, &file_contents);
EXPECT_EQ(stream_size, file_contents.size());
auto input_stream =
absl::make_unique<util::FileInputStream>(input_fd, buffer_size);
const void* buffer;
auto next_result = input_stream->Next(&buffer);
EXPECT_TRUE(next_result.ok()) << next_result.status();
EXPECT_EQ(buffer_size, next_result.ValueOrDie());
EXPECT_EQ(file_contents.substr(0, buffer_size),
std::string(static_cast<const char*>(buffer), buffer_size));
}
}
TEST_F(FileInputStreamTest, testBackupAndPosition) {
int stream_size = 100000;
int buffer_size = 1234;
const void* buffer;
std::string file_contents;
std::string filename = absl::StrCat(buffer_size, "_backup_test.bin");
int input_fd =
test::GetTestFileDescriptor(filename, stream_size, &file_contents);
EXPECT_EQ(stream_size, file_contents.size());
// Prepare the stream and do the first call to Next().
auto input_stream =
absl::make_unique<util::FileInputStream>(input_fd, buffer_size);
EXPECT_EQ(0, input_stream->Position());
auto next_result = input_stream->Next(&buffer);
EXPECT_TRUE(next_result.ok()) << next_result.status();
EXPECT_EQ(buffer_size, next_result.ValueOrDie());
EXPECT_EQ(buffer_size, input_stream->Position());
EXPECT_EQ(file_contents.substr(0, buffer_size),
std::string(static_cast<const char*>(buffer), buffer_size));
// BackUp several times, but in total fewer bytes than returned by Next().
int total_backup_size = 0;
for (auto backup_size : {0, 1, 5, 0, 10, 100, -42, 400, 20, -100}) {
SCOPED_TRACE(absl::StrCat("backup_size = ", backup_size));
input_stream->BackUp(backup_size);
total_backup_size += std::max(0, backup_size);
EXPECT_EQ(buffer_size - total_backup_size, input_stream->Position());
}
// Call Next(), it should return exactly the backed up bytes.
next_result = input_stream->Next(&buffer);
EXPECT_TRUE(next_result.ok()) << next_result.status();
EXPECT_EQ(total_backup_size, next_result.ValueOrDie());
EXPECT_EQ(buffer_size, input_stream->Position());
EXPECT_EQ(
file_contents.substr(buffer_size - total_backup_size, total_backup_size),
std::string(static_cast<const char*>(buffer), total_backup_size));
// BackUp() some bytes, again fewer than returned by Next().
total_backup_size = 0;
for (auto backup_size : {0, 72, -94, 37, 82}) {
SCOPED_TRACE(absl::StrCat("backup_size = ", backup_size));
input_stream->BackUp(backup_size);
total_backup_size += std::max(0, backup_size);
EXPECT_EQ(buffer_size - total_backup_size, input_stream->Position());
}
// Call Next(), it should return exactly the backed up bytes.
next_result = input_stream->Next(&buffer);
EXPECT_TRUE(next_result.ok()) << next_result.status();
EXPECT_EQ(total_backup_size, next_result.ValueOrDie());
EXPECT_EQ(buffer_size, input_stream->Position());
EXPECT_EQ(
file_contents.substr(buffer_size - total_backup_size, total_backup_size),
std::string(static_cast<const char*>(buffer), total_backup_size));
// Call Next() again, it should return the second block.
next_result = input_stream->Next(&buffer);
EXPECT_TRUE(next_result.ok()) << next_result.status();
EXPECT_EQ(buffer_size, next_result.ValueOrDie());
EXPECT_EQ(2 * buffer_size, input_stream->Position());
EXPECT_EQ(file_contents.substr(buffer_size, buffer_size),
std::string(static_cast<const char*>(buffer), buffer_size));
// BackUp a few times, with total over the returned buffer_size.
total_backup_size = 0;
for (auto backup_size :
{0, 72, -100, buffer_size/2, 200, -25, buffer_size, 42}) {
SCOPED_TRACE(absl::StrCat("backup_size = ", backup_size));
input_stream->BackUp(backup_size);
total_backup_size = std::min(buffer_size,
total_backup_size + std::max(0, backup_size));
EXPECT_EQ(2 * buffer_size - total_backup_size, input_stream->Position());
}
// Call Next() again, it should return the second block.
next_result = input_stream->Next(&buffer);
EXPECT_TRUE(next_result.ok()) << next_result.status();
EXPECT_EQ(buffer_size, next_result.ValueOrDie());
EXPECT_EQ(2 * buffer_size, input_stream->Position());
EXPECT_EQ(file_contents.substr(buffer_size, buffer_size),
std::string(static_cast<const char*>(buffer), buffer_size));
}
} // namespace
} // namespace tink
} // namespace crypto