blob: 003e4057a8bb7e1ebc9d0e45c0de505e5e5a1ac2 [file] [log] [blame]
// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <stddef.h>
#include <stdint.h>
#include <utility>
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/location.h"
#include "base/path_service.h"
#include "base/run_loop.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/task_traits.h"
#include "base/task/thread_pool.h"
#include "build/build_config.h"
#include "chrome/browser/media_galleries/fileapi/media_file_system_backend.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/test/browser_test.h"
#include "storage/browser/file_system/copy_or_move_file_validator.h"
#include "storage/browser/file_system/copy_or_move_hook_delegate.h"
#include "storage/browser/file_system/file_system_backend.h"
#include "storage/browser/file_system/file_system_context.h"
#include "storage/browser/file_system/file_system_operation_runner.h"
#include "storage/browser/file_system/file_system_url.h"
#include "storage/browser/file_system/isolated_context.h"
#include "storage/browser/quota/quota_manager_proxy.h"
#include "storage/browser/test/test_file_system_backend.h"
#include "storage/browser/test/test_file_system_context.h"
#include "storage/common/file_system/file_system_types.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"
#include "url/gurl.h"
#include "url/origin.h"
namespace {
const char kOrigin[] = "http://foo";
const char kValidImage[] = "RIFF0\0\0\0WEBPVP8 $\0\0\0\xB2\x02\0\x9D\x01\x2A"
"\x01\0\x01\0\x2F\x9D\xCE\xE7s\xA8((((\x01\x9CK(\0"
"\x05\xCE\xB3l\0\0\xFE\xD8\x80\0\0";
const char kInvalidMediaFile[] = "Not a media file";
const int64_t kNoFileSize = -1;
void HandleCheckFileResult(int64_t expected_size,
base::OnceCallback<void(bool success)> callback,
base::File::Error result,
const base::File::Info& file_info) {
if (result == base::File::FILE_OK) {
if (!file_info.is_directory && expected_size != kNoFileSize &&
file_info.size == expected_size) {
std::move(callback).Run(true);
return;
}
}
std::move(callback).Run(expected_size == kNoFileSize);
}
base::FilePath GetMediaTestDir() {
base::FilePath test_file;
if (!base::PathService::Get(base::DIR_SRC_TEST_DATA_ROOT, &test_file)) {
return base::FilePath();
}
return test_file.AppendASCII("media").AppendASCII("test").AppendASCII("data");
}
} // namespace
class MediaFileValidatorTest : public InProcessBrowserTest {
public:
MediaFileValidatorTest() : test_file_size_(0) {}
MediaFileValidatorTest(const MediaFileValidatorTest&) = delete;
MediaFileValidatorTest& operator=(const MediaFileValidatorTest&) = delete;
~MediaFileValidatorTest() override = default;
// Write |content| into |filename| in a test file system and try to move
// it into a media file system. The result is compared to |expected_result|.
void MoveTest(const std::string& filename, const std::string& content,
bool expected_result) {
base::RunLoop run_loop;
quit_closure_ = run_loop.QuitClosure();
base::ThreadPool::PostTask(
FROM_HERE, {base::MayBlock()},
base::BindOnce(&MediaFileValidatorTest::SetupBlocking,
base::Unretained(this), filename, content,
expected_result));
run_loop.Run();
}
// Write |source| into |filename| in a test file system and try to move it
// into a media file system. The result is compared to |expected_result|.
void MoveTestFromFile(const std::string& filename,
const base::FilePath& source, bool expected_result) {
base::RunLoop run_loop;
quit_closure_ = run_loop.QuitClosure();
base::ThreadPool::PostTask(
FROM_HERE, {base::MayBlock()},
base::BindOnce(&MediaFileValidatorTest::SetupFromFileBlocking,
base::Unretained(this), filename, source,
expected_result));
run_loop.Run();
}
private:
// BrowserTestBase interface.
void PostRunTestOnMainThread() override {
// Trigger release of the FileSystemContext before the IO thread is gone,
// so it can teardown there correctly.
file_system_context_ = nullptr;
InProcessBrowserTest::PostRunTestOnMainThread();
}
// Create the test files, filesystem objects, etc.
void SetupBlocking(const std::string& filename,
const std::string& content,
bool expected_result) {
ASSERT_TRUE(base_dir_.CreateUniqueTempDir());
base::FilePath base = base_dir_.GetPath();
base::FilePath src_path = base.AppendASCII("src_fs");
ASSERT_TRUE(base::CreateDirectory(src_path));
std::vector<std::unique_ptr<storage::FileSystemBackend>>
additional_providers;
file_system_runner_ =
base::ThreadPool::CreateSequencedTaskRunner({base::MayBlock()});
additional_providers.push_back(
std::make_unique<storage::TestFileSystemBackend>(
file_system_runner_.get(), src_path));
additional_providers.push_back(
std::make_unique<MediaFileSystemBackend>(base, base::NullCallback()));
file_system_context_ =
storage::CreateFileSystemContextWithAdditionalProvidersForTesting(
content::GetIOThreadTaskRunner({}), file_system_runner_,
/*quota_manager_proxy=*/nullptr, std::move(additional_providers),
base);
move_src_ = file_system_context_->CreateCrackedFileSystemURL(
blink::StorageKey::CreateFromStringForTesting(kOrigin),
storage::kFileSystemTypeTest, base::FilePath::FromUTF8Unsafe(filename));
test_file_size_ = content.size();
base::FilePath test_file = src_path.AppendASCII(filename);
ASSERT_TRUE(base::WriteFile(test_file, content));
base::FilePath dest_path = base.AppendASCII("dest_fs");
ASSERT_TRUE(base::CreateDirectory(dest_path));
dest_fs_ =
storage::IsolatedContext::GetInstance()->RegisterFileSystemForPath(
storage::kFileSystemTypeLocalMedia, std::string(), dest_path,
nullptr);
size_t extension_index = filename.find_last_of(".");
ASSERT_NE(std::string::npos, extension_index);
std::string extension = filename.substr(extension_index);
std::string dest_root_fs_url = storage::GetIsolatedFileSystemRootURIString(
GURL(kOrigin), dest_fs_.id(), "dest_fs/");
const GURL crack_url = GURL(dest_root_fs_url + "move_dest" + extension);
move_dest_ = file_system_context_->CrackURLInFirstPartyContext(crack_url);
content::GetIOThreadTaskRunner({})->PostTask(
FROM_HERE,
base::BindOnce(
&MediaFileValidatorTest::CheckFiles, base::Unretained(this), true,
base::BindOnce(&MediaFileValidatorTest::OnTestFilesReady,
base::Unretained(this), expected_result)));
}
void SetupFromFileBlocking(const std::string& filename,
const base::FilePath& source,
bool expected_result) {
std::string content;
ASSERT_TRUE(base::ReadFileToString(source, &content));
SetupBlocking(filename, content, expected_result);
}
// Check that exactly one of |move_src_| and |move_dest_| exists.
// |src_expected| indicates which one should exist. When complete,
// |callback| is called with success/failure.
void CheckFiles(bool src_expected,
base::OnceCallback<void(bool success)> callback) {
CheckFile(move_src_, src_expected ? test_file_size_ : kNoFileSize,
base::BindOnce(&MediaFileValidatorTest::OnCheckFilesFirstResult,
base::Unretained(this), !src_expected,
std::move(callback)));
}
// Helper that checks a file has the |expected_size|, which may be
// |kNoFileSize| if the file should not exist. |callback| is called
// with success/failure.
void CheckFile(storage::FileSystemURL url,
int64_t expected_size,
base::OnceCallback<void(bool success)> callback) {
operation_runner()->GetMetadata(
url, {storage::FileSystemOperation::GetMetadataField::kSize},
base::BindOnce(&HandleCheckFileResult, expected_size,
std::move(callback)));
}
// Helper that checks the result of |move_src_| lookup and then checks
// |move_dest_| if all is as expected.
void OnCheckFilesFirstResult(bool dest_expected,
base::OnceCallback<void(bool)> callback,
bool src_result) {
EXPECT_TRUE(src_result);
if (!src_result) {
std::move(callback).Run(false);
return;
}
CheckFile(move_dest_, dest_expected ? test_file_size_ : kNoFileSize,
std::move(callback));
}
// Assert |test_files_ready| and then do the actual test of moving
// |move_src_| to |move_dest_|.
void OnTestFilesReady(bool expected_result, bool test_files_ready) {
ASSERT_TRUE(test_files_ready);
operation_runner()->Move(
move_src_, move_dest_,
storage::FileSystemOperation::CopyOrMoveOptionSet(),
storage::FileSystemOperation::ERROR_BEHAVIOR_ABORT,
std::make_unique<storage::CopyOrMoveHookDelegate>(),
base::BindOnce(&MediaFileValidatorTest::OnMoveResult,
base::Unretained(this), expected_result));
}
// Check that the move succeeded/failed based on expectation and then
// check that the right file exists.
void OnMoveResult(bool expected_result, base::File::Error result) {
if (expected_result)
EXPECT_EQ(base::File::FILE_OK, result);
else
EXPECT_EQ(base::File::FILE_ERROR_SECURITY, result);
CheckFiles(!expected_result,
base::BindOnce(&MediaFileValidatorTest::OnTestFilesCheckResult,
base::Unretained(this)));
}
// Check that the correct test file exists and then allow the main-thread
// RunLoop to quit.
void OnTestFilesCheckResult(bool result) {
EXPECT_TRUE(result);
std::move(quit_closure_).Run();
}
storage::FileSystemOperationRunner* operation_runner() {
return file_system_context_->operation_runner();
}
base::ScopedTempDir base_dir_;
scoped_refptr<storage::FileSystemContext> file_system_context_;
int test_file_size_;
storage::FileSystemURL move_src_;
storage::FileSystemURL move_dest_;
storage::IsolatedContext::ScopedFSHandle dest_fs_;
base::OnceClosure quit_closure_;
scoped_refptr<base::SequencedTaskRunner> file_system_runner_;
};
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, UnsupportedExtension) {
MoveTest("a.txt", std::string(kValidImage, std::size(kValidImage)), false);
}
// TODO(crbug.com/40744004): Re-enable. Flaky on Linux.
#if BUILDFLAG(IS_LINUX)
#define MAYBE_ValidImage DISABLED_ValidImage
#else
#define MAYBE_ValidImage ValidImage
#endif
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, MAYBE_ValidImage) {
MoveTest("a.webp", std::string(kValidImage, std::size(kValidImage)), true);
}
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, InvalidImage) {
MoveTest("a.webp",
std::string(kInvalidMediaFile, std::size(kInvalidMediaFile)), false);
}
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, InvalidAudio) {
MoveTest("a.ogg",
std::string(kInvalidMediaFile, std::size(kInvalidMediaFile)), false);
}
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, ValidAudio) {
base::FilePath test_file = GetMediaTestDir();
ASSERT_FALSE(test_file.empty());
test_file = test_file.AppendASCII("sfx.ogg");
MoveTestFromFile("sfx.ogg", test_file, true);
}
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, InvalidVideo) {
base::FilePath test_file = GetMediaTestDir();
ASSERT_FALSE(test_file.empty());
test_file = test_file.AppendASCII("no_streams.webm");
MoveTestFromFile("no_streams.webm", test_file, false);
}
IN_PROC_BROWSER_TEST_F(MediaFileValidatorTest, ValidVideo) {
base::FilePath test_file = GetMediaTestDir();
ASSERT_FALSE(test_file.empty());
test_file = test_file.AppendASCII("bear-320x240-multitrack.webm");
MoveTestFromFile("multitrack.webm", test_file, true);
}