filesystem_copier_action_unittest.cc - platform/system/update_engine - Gitiles

 // Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.

 #include <fcntl.h>

 #include <set>
 #include <string>
 #include <vector>

 #include <base/posix/eintr_wrapper.h>
 #include <base/strings/string_util.h>
 #include <base/strings/stringprintf.h>
 #include <glib.h>
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>

 #include "update_engine/fake_system_state.h"
 #include "update_engine/filesystem_copier_action.h"
 #include "update_engine/mock_hardware.h"
 #include "update_engine/omaha_hash_calculator.h"
 #include "update_engine/test_utils.h"
 #include "update_engine/utils.h"

 using std::set;
 using std::string;
 using std::vector;

 namespace chromeos_update_engine {

 class FilesystemCopierActionTest : public ::testing::Test {
  protected:
   // |verify_hash|: 0 - no hash verification, 1 -- successful hash verification,
   // 2 -- hash verification failure.
   // Returns true iff test has completed successfully.
   bool DoTest(bool run_out_of_space,
               bool terminate_early,
               bool use_kernel_partition,
               int verify_hash);
   void SetUp() {
   }
   void TearDown() {
   }

   FakeSystemState fake_system_state_;
 };

 class FilesystemCopierActionTestDelegate : public ActionProcessorDelegate {
  public:
   FilesystemCopierActionTestDelegate(GMainLoop* loop,
                                      FilesystemCopierAction* action)
       : loop_(loop), action_(action), ran_(false), code_(ErrorCode::kError) {}
   void ExitMainLoop() {
     GMainContext* context = g_main_loop_get_context(loop_);
     // We cannot use g_main_context_pending() alone to determine if it is safe
     // to quit the main loop here becasuse g_main_context_pending() may return
     // FALSE when g_input_stream_read_async() in FilesystemCopierAction has
     // been cancelled but the callback has not yet been invoked.
     while (g_main_context_pending(context) || action_->IsCleanupPending()) {
       g_main_context_iteration(context, false);
       g_usleep(100);
     }
     g_main_loop_quit(loop_);
   }
   void ProcessingDone(const ActionProcessor* processor, ErrorCode code) {
     ExitMainLoop();
   }
   void ProcessingStopped(const ActionProcessor* processor) {
     ExitMainLoop();
   }
   void ActionCompleted(ActionProcessor* processor,
                        AbstractAction* action,
                        ErrorCode code) {
     if (action->Type() == FilesystemCopierAction::StaticType()) {
       ran_ = true;
       code_ = code;
     }
   }
   bool ran() const { return ran_; }
   ErrorCode code() const { return code_; }

  private:
   GMainLoop* loop_;
   FilesystemCopierAction* action_;
   bool ran_;
   ErrorCode code_;
 };

 struct StartProcessorCallbackArgs {
   ActionProcessor* processor;
   FilesystemCopierAction* filesystem_copier_action;
   bool terminate_early;
 };

 gboolean StartProcessorInRunLoop(gpointer data) {
   StartProcessorCallbackArgs* args =
       reinterpret_cast<StartProcessorCallbackArgs*>(data);
   ActionProcessor* processor = args->processor;
   processor->StartProcessing();
   if (args->terminate_early) {
     EXPECT_TRUE(args->filesystem_copier_action);
     args->processor->StopProcessing();
   }
   return FALSE;
 }

 // TODO(garnold) Temporarily disabling this test, see chromium-os:31082 for
 // details; still trying to track down the root cause for these rare write
 // failures and whether or not they are due to the test setup or an inherent
 // issue with the chroot environiment, library versions we use, etc.
 TEST_F(FilesystemCopierActionTest, DISABLED_RunAsRootSimpleTest) {
   ASSERT_EQ(0, getuid());
   bool test = DoTest(false, false, true, 0);
   EXPECT_TRUE(test);
   if (!test)
     return;
   test = DoTest(false, false, false, 0);
   EXPECT_TRUE(test);
 }

 bool FilesystemCopierActionTest::DoTest(bool run_out_of_space,
                                         bool terminate_early,
                                         bool use_kernel_partition,
                                         int verify_hash) {
   // We need MockHardware to verify MarkUnbootable calls, but don't want
   // warnings about other usages.
   testing::NiceMock<MockHardware> mock_hardware;
   fake_system_state_.set_hardware(&mock_hardware);

   GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE);

   string a_loop_file;
   string b_loop_file;

   if (!(utils::MakeTempFile("a_loop_file.XXXXXX", &a_loop_file, NULL) &&
         utils::MakeTempFile("b_loop_file.XXXXXX", &b_loop_file, NULL))) {
     ADD_FAILURE();
     return false;
   }
   ScopedPathUnlinker a_loop_file_unlinker(a_loop_file);
   ScopedPathUnlinker b_loop_file_unlinker(b_loop_file);

   // Make random data for a, zero filled data for b.
   const size_t kLoopFileSize = 10 * 1024 * 1024 + 512;
   vector<char> a_loop_data(kLoopFileSize);
   FillWithData(&a_loop_data);
   vector<char> b_loop_data(run_out_of_space ?
                            (kLoopFileSize - 1) :
                            kLoopFileSize,
                            '\0');  // Fill with 0s

   // Write data to disk
   if (!(WriteFileVector(a_loop_file, a_loop_data) &&
         WriteFileVector(b_loop_file, b_loop_data))) {
     ADD_FAILURE();
     return false;
   }

   // Attach loop devices to the files
   string a_dev;
   string b_dev;

   ScopedLoopbackDeviceBinder a_dev_releaser(a_loop_file, &a_dev);
   ScopedLoopbackDeviceBinder b_dev_releaser(b_loop_file, &b_dev);
   if (!(a_dev_releaser.is_bound() && b_dev_releaser.is_bound())) {
     ADD_FAILURE();
     return false;
   }

   LOG(INFO) << "copying: "
             << a_loop_file << " (" << a_dev << ") -> "
             << b_loop_file << " (" << b_dev << ", "
             << kLoopFileSize << " bytes";
   bool success = true;

   // Set up the action objects
   InstallPlan install_plan;
   if (verify_hash) {
     if (use_kernel_partition) {
       install_plan.kernel_install_path = a_dev;
       install_plan.kernel_size =
           kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
       if (!OmahaHashCalculator::RawHashOfData(a_loop_data,
                                               &install_plan.kernel_hash)) {
         ADD_FAILURE();
         success = false;
       }
     } else {
       install_plan.install_path = a_dev;
       install_plan.rootfs_size =
           kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
       if (!OmahaHashCalculator::RawHashOfData(a_loop_data,
                                               &install_plan.rootfs_hash)) {
         ADD_FAILURE();
         success = false;
       }
     }
   } else {
     if (use_kernel_partition) {
       install_plan.kernel_install_path = b_dev;
     } else {
       install_plan.install_path = b_dev;
     }
   }

   EXPECT_CALL(mock_hardware,
               MarkKernelUnbootable(a_dev)).Times(use_kernel_partition ? 1 : 0);

   ActionProcessor processor;

   ObjectFeederAction<InstallPlan> feeder_action;
   FilesystemCopierAction copier_action(&fake_system_state_,
                                        use_kernel_partition,
                                        verify_hash != 0);
   ObjectCollectorAction<InstallPlan> collector_action;

   BondActions(&feeder_action, &copier_action);
   BondActions(&copier_action, &collector_action);

   FilesystemCopierActionTestDelegate delegate(loop, &copier_action);
   processor.set_delegate(&delegate);
   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);

   if (!verify_hash) {
     copier_action.set_copy_source(a_dev);
   }
   feeder_action.set_obj(install_plan);

   StartProcessorCallbackArgs start_callback_args;
   start_callback_args.processor = &processor;
   start_callback_args.filesystem_copier_action = &copier_action;
   start_callback_args.terminate_early = terminate_early;

   g_timeout_add(0, &StartProcessorInRunLoop, &start_callback_args);
   g_main_loop_run(loop);
   g_main_loop_unref(loop);

   if (!terminate_early) {
     bool is_delegate_ran = delegate.ran();
     EXPECT_TRUE(is_delegate_ran);
     success = success && is_delegate_ran;
   }
   if (run_out_of_space || terminate_early) {
     EXPECT_EQ(ErrorCode::kError, delegate.code());
     return (ErrorCode::kError == delegate.code());
   }
   if (verify_hash == 2) {
     ErrorCode expected_exit_code =
         (use_kernel_partition ?
          ErrorCode::kNewKernelVerificationError :
          ErrorCode::kNewRootfsVerificationError);
     EXPECT_EQ(expected_exit_code, delegate.code());
     return (expected_exit_code == delegate.code());
   }
   EXPECT_EQ(ErrorCode::kSuccess, delegate.code());

   // Make sure everything in the out_image is there
   vector<char> a_out;
   if (!utils::ReadFile(a_dev, &a_out)) {
     ADD_FAILURE();
     return false;
   }
   const bool is_a_file_reading_eq = ExpectVectorsEq(a_loop_data, a_out);
   EXPECT_TRUE(is_a_file_reading_eq);
   success = success && is_a_file_reading_eq;
   if (!verify_hash) {
     vector<char> b_out;
     if (!utils::ReadFile(b_dev, &b_out)) {
       ADD_FAILURE();
       return false;
     }
     const bool is_b_file_reading_eq = ExpectVectorsEq(a_out, b_out);
     EXPECT_TRUE(is_b_file_reading_eq);
     success = success && is_b_file_reading_eq;
   }

   bool is_install_plan_eq = (collector_action.object() == install_plan);
   EXPECT_TRUE(is_install_plan_eq);
   success = success && is_install_plan_eq;

   LOG(INFO) << "Verifying bootable flag on: " << a_dev;
   bool bootable;
   EXPECT_TRUE(mock_hardware.fake().IsKernelBootable(a_dev, &bootable));
   // We should always mark a partition as unbootable if it's a kernel
   // partition, but never if it's anything else.
   EXPECT_EQ(bootable, !use_kernel_partition);

   return success;
 }

 class FilesystemCopierActionTest2Delegate : public ActionProcessorDelegate {
  public:
   void ActionCompleted(ActionProcessor* processor,
                        AbstractAction* action,
                        ErrorCode code) {
     if (action->Type() == FilesystemCopierAction::StaticType()) {
       ran_ = true;
       code_ = code;
     }
   }
   GMainLoop *loop_;
   bool ran_;
   ErrorCode code_;
 };

 TEST_F(FilesystemCopierActionTest, MissingInputObjectTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   FilesystemCopierAction copier_action(&fake_system_state_, false, false);
   ObjectCollectorAction<InstallPlan> collector_action;

   BondActions(&copier_action, &collector_action);

   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
   processor.StartProcessing();
   EXPECT_FALSE(processor.IsRunning());
   EXPECT_TRUE(delegate.ran_);
   EXPECT_EQ(ErrorCode::kError, delegate.code_);
 }

 TEST_F(FilesystemCopierActionTest, ResumeTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   ObjectFeederAction<InstallPlan> feeder_action;
   const char* kUrl = "http://some/url";
   InstallPlan install_plan(false, true, kUrl, 0, "", 0, "", "", "", "");
   feeder_action.set_obj(install_plan);
   FilesystemCopierAction copier_action(&fake_system_state_, false, false);
   ObjectCollectorAction<InstallPlan> collector_action;

   BondActions(&feeder_action, &copier_action);
   BondActions(&copier_action, &collector_action);

   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
   processor.StartProcessing();
   EXPECT_FALSE(processor.IsRunning());
   EXPECT_TRUE(delegate.ran_);
   EXPECT_EQ(ErrorCode::kSuccess, delegate.code_);
   EXPECT_EQ(kUrl, collector_action.object().download_url);
 }

 TEST_F(FilesystemCopierActionTest, NonExistentDriveTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   ObjectFeederAction<InstallPlan> feeder_action;
   InstallPlan install_plan(false,
                            false,
                            "",
                            0,
                            "",
                            0,
                            "",
                            "/no/such/file",
                            "/no/such/file",
                            "");
   feeder_action.set_obj(install_plan);
   FilesystemCopierAction copier_action(&fake_system_state_, false, false);
   ObjectCollectorAction<InstallPlan> collector_action;

   BondActions(&copier_action, &collector_action);

   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
   processor.StartProcessing();
   EXPECT_FALSE(processor.IsRunning());
   EXPECT_TRUE(delegate.ran_);
   EXPECT_EQ(ErrorCode::kError, delegate.code_);
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashTest) {
   ASSERT_EQ(0, getuid());
   EXPECT_TRUE(DoTest(false, false, false, 1));
   EXPECT_TRUE(DoTest(false, false, true, 1));
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashFailTest) {
   ASSERT_EQ(0, getuid());
   EXPECT_TRUE(DoTest(false, false, false, 2));
   EXPECT_TRUE(DoTest(false, false, true, 2));
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootNoSpaceTest) {
   ASSERT_EQ(0, getuid());
   EXPECT_TRUE(DoTest(true, false, false, 0));
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootTerminateEarlyTest) {
   ASSERT_EQ(0, getuid());
   EXPECT_TRUE(DoTest(false, true, false, 0));
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootDetermineFilesystemSizeTest) {
   string img;
   EXPECT_TRUE(utils::MakeTempFile("img.XXXXXX", &img, NULL));
   ScopedPathUnlinker img_unlinker(img);
   CreateExtImageAtPath(img, NULL);
   // Extend the "partition" holding the file system from 10MiB to 20MiB.
   EXPECT_EQ(0, System(base::StringPrintf(
       "dd if=/dev/zero of=%s seek=20971519 bs=1 count=1",
       img.c_str())));
   EXPECT_EQ(20 * 1024 * 1024, utils::FileSize(img));

   for (int i = 0; i < 2; ++i) {
     bool is_kernel = i == 1;
     FilesystemCopierAction action(&fake_system_state_, is_kernel, false);
     EXPECT_EQ(kint64max, action.filesystem_size_);
     {
       int fd = HANDLE_EINTR(open(img.c_str(), O_RDONLY));
       EXPECT_GT(fd, 0);
       ScopedFdCloser fd_closer(&fd);
       action.DetermineFilesystemSize(fd);
     }
     EXPECT_EQ(is_kernel ? kint64max : 10 * 1024 * 1024,
               action.filesystem_size_);
   }
 }


 }  // namespace chromeos_update_engine
	// Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
	// Use of this source code is governed by a BSD-style license that can be
	// found in the LICENSE file.

	#include <fcntl.h>

	#include <set>
	#include <string>
	#include <vector>

	#include <base/posix/eintr_wrapper.h>
	#include <base/strings/string_util.h>
	#include <base/strings/stringprintf.h>
	#include <glib.h>
	#include <gmock/gmock.h>
	#include <gtest/gtest.h>

	#include "update_engine/fake_system_state.h"
	#include "update_engine/filesystem_copier_action.h"
	#include "update_engine/mock_hardware.h"
	#include "update_engine/omaha_hash_calculator.h"
	#include "update_engine/test_utils.h"
	#include "update_engine/utils.h"

	using std::set;
	using std::string;
	using std::vector;

	namespace chromeos_update_engine {

	class FilesystemCopierActionTest : public ::testing::Test {
	protected:
	// \|verify_hash\|: 0 - no hash verification, 1 -- successful hash verification,
	// 2 -- hash verification failure.
	// Returns true iff test has completed successfully.
	bool DoTest(bool run_out_of_space,
	bool terminate_early,
	bool use_kernel_partition,
	int verify_hash);
	void SetUp() {
	}
	void TearDown() {
	}

	FakeSystemState fake_system_state_;
	};

	class FilesystemCopierActionTestDelegate : public ActionProcessorDelegate {
	public:
	FilesystemCopierActionTestDelegate(GMainLoop* loop,
	FilesystemCopierAction* action)
	: loop_(loop), action_(action), ran_(false), code_(ErrorCode::kError) {}
	void ExitMainLoop() {
	GMainContext* context = g_main_loop_get_context(loop_);
	// We cannot use g_main_context_pending() alone to determine if it is safe
	// to quit the main loop here becasuse g_main_context_pending() may return
	// FALSE when g_input_stream_read_async() in FilesystemCopierAction has
	// been cancelled but the callback has not yet been invoked.
	while (g_main_context_pending(context) \|\| action_->IsCleanupPending()) {
	g_main_context_iteration(context, false);
	g_usleep(100);
	}
	g_main_loop_quit(loop_);
	}
	void ProcessingDone(const ActionProcessor* processor, ErrorCode code) {
	ExitMainLoop();
	}
	void ProcessingStopped(const ActionProcessor* processor) {
	ExitMainLoop();
	}
	void ActionCompleted(ActionProcessor* processor,
	AbstractAction* action,
	ErrorCode code) {
	if (action->Type() == FilesystemCopierAction::StaticType()) {
	ran_ = true;
	code_ = code;
	}
	}
	bool ran() const { return ran_; }
	ErrorCode code() const { return code_; }

	private:
	GMainLoop* loop_;
	FilesystemCopierAction* action_;
	bool ran_;
	ErrorCode code_;
	};

	struct StartProcessorCallbackArgs {
	ActionProcessor* processor;
	FilesystemCopierAction* filesystem_copier_action;
	bool terminate_early;
	};

	gboolean StartProcessorInRunLoop(gpointer data) {
	StartProcessorCallbackArgs* args =
	reinterpret_cast<StartProcessorCallbackArgs*>(data);
	ActionProcessor* processor = args->processor;
	processor->StartProcessing();
	if (args->terminate_early) {
	EXPECT_TRUE(args->filesystem_copier_action);
	args->processor->StopProcessing();
	}
	return FALSE;
	}

	// TODO(garnold) Temporarily disabling this test, see chromium-os:31082 for
	// details; still trying to track down the root cause for these rare write
	// failures and whether or not they are due to the test setup or an inherent
	// issue with the chroot environiment, library versions we use, etc.
	TEST_F(FilesystemCopierActionTest, DISABLED_RunAsRootSimpleTest) {
	ASSERT_EQ(0, getuid());
	bool test = DoTest(false, false, true, 0);
	EXPECT_TRUE(test);
	if (!test)
	return;
	test = DoTest(false, false, false, 0);
	EXPECT_TRUE(test);
	}

	bool FilesystemCopierActionTest::DoTest(bool run_out_of_space,
	bool terminate_early,
	bool use_kernel_partition,
	int verify_hash) {
	// We need MockHardware to verify MarkUnbootable calls, but don't want
	// warnings about other usages.
	testing::NiceMock<MockHardware> mock_hardware;
	fake_system_state_.set_hardware(&mock_hardware);

	GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE);

	string a_loop_file;
	string b_loop_file;

	if (!(utils::MakeTempFile("a_loop_file.XXXXXX", &a_loop_file, NULL) &&
	utils::MakeTempFile("b_loop_file.XXXXXX", &b_loop_file, NULL))) {
	ADD_FAILURE();
	return false;
	}
	ScopedPathUnlinker a_loop_file_unlinker(a_loop_file);
	ScopedPathUnlinker b_loop_file_unlinker(b_loop_file);

	// Make random data for a, zero filled data for b.
	const size_t kLoopFileSize = 10 * 1024 * 1024 + 512;
	vector<char> a_loop_data(kLoopFileSize);
	FillWithData(&a_loop_data);
	vector<char> b_loop_data(run_out_of_space ?
	(kLoopFileSize - 1) :
	kLoopFileSize,
	'\0'); // Fill with 0s

	// Write data to disk
	if (!(WriteFileVector(a_loop_file, a_loop_data) &&
	WriteFileVector(b_loop_file, b_loop_data))) {
	ADD_FAILURE();
	return false;
	}

	// Attach loop devices to the files
	string a_dev;
	string b_dev;

	ScopedLoopbackDeviceBinder a_dev_releaser(a_loop_file, &a_dev);
	ScopedLoopbackDeviceBinder b_dev_releaser(b_loop_file, &b_dev);
	if (!(a_dev_releaser.is_bound() && b_dev_releaser.is_bound())) {
	ADD_FAILURE();
	return false;
	}

	LOG(INFO) << "copying: "
	<< a_loop_file << " (" << a_dev << ") -> "
	<< b_loop_file << " (" << b_dev << ", "
	<< kLoopFileSize << " bytes";
	bool success = true;

	// Set up the action objects
	InstallPlan install_plan;
	if (verify_hash) {
	if (use_kernel_partition) {
	install_plan.kernel_install_path = a_dev;
	install_plan.kernel_size =
	kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
	if (!OmahaHashCalculator::RawHashOfData(a_loop_data,
	&install_plan.kernel_hash)) {
	ADD_FAILURE();
	success = false;
	}
	} else {
	install_plan.install_path = a_dev;
	install_plan.rootfs_size =
	kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
	if (!OmahaHashCalculator::RawHashOfData(a_loop_data,
	&install_plan.rootfs_hash)) {
	ADD_FAILURE();
	success = false;
	}
	}
	} else {
	if (use_kernel_partition) {
	install_plan.kernel_install_path = b_dev;
	} else {
	install_plan.install_path = b_dev;
	}
	}

	EXPECT_CALL(mock_hardware,
	MarkKernelUnbootable(a_dev)).Times(use_kernel_partition ? 1 : 0);

	ActionProcessor processor;

	ObjectFeederAction<InstallPlan> feeder_action;
	FilesystemCopierAction copier_action(&fake_system_state_,
	use_kernel_partition,
	verify_hash != 0);
	ObjectCollectorAction<InstallPlan> collector_action;

	BondActions(&feeder_action, &copier_action);
	BondActions(&copier_action, &collector_action);

	FilesystemCopierActionTestDelegate delegate(loop, &copier_action);
	processor.set_delegate(&delegate);
	processor.EnqueueAction(&feeder_action);
	processor.EnqueueAction(&copier_action);
	processor.EnqueueAction(&collector_action);

	if (!verify_hash) {
	copier_action.set_copy_source(a_dev);
	}
	feeder_action.set_obj(install_plan);

	StartProcessorCallbackArgs start_callback_args;
	start_callback_args.processor = &processor;
	start_callback_args.filesystem_copier_action = &copier_action;
	start_callback_args.terminate_early = terminate_early;

	g_timeout_add(0, &StartProcessorInRunLoop, &start_callback_args);
	g_main_loop_run(loop);
	g_main_loop_unref(loop);

	if (!terminate_early) {
	bool is_delegate_ran = delegate.ran();
	EXPECT_TRUE(is_delegate_ran);
	success = success && is_delegate_ran;
	}
	if (run_out_of_space \|\| terminate_early) {
	EXPECT_EQ(ErrorCode::kError, delegate.code());
	return (ErrorCode::kError == delegate.code());
	}
	if (verify_hash == 2) {
	ErrorCode expected_exit_code =
	(use_kernel_partition ?
	ErrorCode::kNewKernelVerificationError :
	ErrorCode::kNewRootfsVerificationError);
	EXPECT_EQ(expected_exit_code, delegate.code());
	return (expected_exit_code == delegate.code());
	}
	EXPECT_EQ(ErrorCode::kSuccess, delegate.code());

	// Make sure everything in the out_image is there
	vector<char> a_out;
	if (!utils::ReadFile(a_dev, &a_out)) {
	ADD_FAILURE();
	return false;
	}
	const bool is_a_file_reading_eq = ExpectVectorsEq(a_loop_data, a_out);
	EXPECT_TRUE(is_a_file_reading_eq);
	success = success && is_a_file_reading_eq;
	if (!verify_hash) {
	vector<char> b_out;
	if (!utils::ReadFile(b_dev, &b_out)) {
	ADD_FAILURE();
	return false;
	}
	const bool is_b_file_reading_eq = ExpectVectorsEq(a_out, b_out);
	EXPECT_TRUE(is_b_file_reading_eq);
	success = success && is_b_file_reading_eq;
	}

	bool is_install_plan_eq = (collector_action.object() == install_plan);
	EXPECT_TRUE(is_install_plan_eq);
	success = success && is_install_plan_eq;

	LOG(INFO) << "Verifying bootable flag on: " << a_dev;
	bool bootable;
	EXPECT_TRUE(mock_hardware.fake().IsKernelBootable(a_dev, &bootable));
	// We should always mark a partition as unbootable if it's a kernel
	// partition, but never if it's anything else.
	EXPECT_EQ(bootable, !use_kernel_partition);

	return success;
	}

	class FilesystemCopierActionTest2Delegate : public ActionProcessorDelegate {
	public:
	void ActionCompleted(ActionProcessor* processor,
	AbstractAction* action,
	ErrorCode code) {
	if (action->Type() == FilesystemCopierAction::StaticType()) {
	ran_ = true;
	code_ = code;
	}
	}
	GMainLoop *loop_;
	bool ran_;
	ErrorCode code_;
	};

	TEST_F(FilesystemCopierActionTest, MissingInputObjectTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	FilesystemCopierAction copier_action(&fake_system_state_, false, false);
	ObjectCollectorAction<InstallPlan> collector_action;

	BondActions(&copier_action, &collector_action);

	processor.EnqueueAction(&copier_action);
	processor.EnqueueAction(&collector_action);
	processor.StartProcessing();
	EXPECT_FALSE(processor.IsRunning());
	EXPECT_TRUE(delegate.ran_);
	EXPECT_EQ(ErrorCode::kError, delegate.code_);
	}

	TEST_F(FilesystemCopierActionTest, ResumeTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	ObjectFeederAction<InstallPlan> feeder_action;
	const char* kUrl = "http://some/url";
	InstallPlan install_plan(false, true, kUrl, 0, "", 0, "", "", "", "");
	feeder_action.set_obj(install_plan);
	FilesystemCopierAction copier_action(&fake_system_state_, false, false);
	ObjectCollectorAction<InstallPlan> collector_action;

	BondActions(&feeder_action, &copier_action);
	BondActions(&copier_action, &collector_action);

	processor.EnqueueAction(&feeder_action);
	processor.EnqueueAction(&copier_action);
	processor.EnqueueAction(&collector_action);
	processor.StartProcessing();
	EXPECT_FALSE(processor.IsRunning());
	EXPECT_TRUE(delegate.ran_);
	EXPECT_EQ(ErrorCode::kSuccess, delegate.code_);
	EXPECT_EQ(kUrl, collector_action.object().download_url);
	}

	TEST_F(FilesystemCopierActionTest, NonExistentDriveTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	ObjectFeederAction<InstallPlan> feeder_action;
	InstallPlan install_plan(false,
	false,
	"",
	0,
	"",
	0,
	"",
	"/no/such/file",
	"/no/such/file",
	"");
	feeder_action.set_obj(install_plan);
	FilesystemCopierAction copier_action(&fake_system_state_, false, false);
	ObjectCollectorAction<InstallPlan> collector_action;

	BondActions(&copier_action, &collector_action);

	processor.EnqueueAction(&feeder_action);
	processor.EnqueueAction(&copier_action);
	processor.EnqueueAction(&collector_action);
	processor.StartProcessing();
	EXPECT_FALSE(processor.IsRunning());
	EXPECT_TRUE(delegate.ran_);
	EXPECT_EQ(ErrorCode::kError, delegate.code_);
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashTest) {
	ASSERT_EQ(0, getuid());
	EXPECT_TRUE(DoTest(false, false, false, 1));
	EXPECT_TRUE(DoTest(false, false, true, 1));
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashFailTest) {
	ASSERT_EQ(0, getuid());
	EXPECT_TRUE(DoTest(false, false, false, 2));
	EXPECT_TRUE(DoTest(false, false, true, 2));
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootNoSpaceTest) {
	ASSERT_EQ(0, getuid());
	EXPECT_TRUE(DoTest(true, false, false, 0));
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootTerminateEarlyTest) {
	ASSERT_EQ(0, getuid());
	EXPECT_TRUE(DoTest(false, true, false, 0));
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootDetermineFilesystemSizeTest) {
	string img;
	EXPECT_TRUE(utils::MakeTempFile("img.XXXXXX", &img, NULL));
	ScopedPathUnlinker img_unlinker(img);
	CreateExtImageAtPath(img, NULL);
	// Extend the "partition" holding the file system from 10MiB to 20MiB.
	EXPECT_EQ(0, System(base::StringPrintf(
	"dd if=/dev/zero of=%s seek=20971519 bs=1 count=1",
	img.c_str())));
	EXPECT_EQ(20 * 1024 * 1024, utils::FileSize(img));

	for (int i = 0; i < 2; ++i) {
	bool is_kernel = i == 1;
	FilesystemCopierAction action(&fake_system_state_, is_kernel, false);
	EXPECT_EQ(kint64max, action.filesystem_size_);
	{
	int fd = HANDLE_EINTR(open(img.c_str(), O_RDONLY));
	EXPECT_GT(fd, 0);
	ScopedFdCloser fd_closer(&fd);
	action.DetermineFilesystemSize(fd);
	}
	EXPECT_EQ(is_kernel ? kint64max : 10 * 1024 * 1024,
	action.filesystem_size_);
	}
	}


	} // namespace chromeos_update_engine