init/startup/test_mode_mount_helper.h

// Copyright 2023 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef INIT_STARTUP_TEST_MODE_MOUNT_HELPER_H_
#define INIT_STARTUP_TEST_MODE_MOUNT_HELPER_H_

#include <memory>
#include <string>

#include <base/files/file_path.h>
#include <base/values.h>
#include <libstorage/platform/platform.h>

#include "init/startup/flags.h"
#include "init/startup/mount_helper.h"
#include "init/startup/mount_var_home_interface.h"
#include "init/startup/startup_dep_impl.h"

namespace startup {

// This class defines the MountHelper behavior we will use when the device
// is running a test image.
class TestModeMountHelper : public MountHelper {
 public:
  TestModeMountHelper(libstorage::Platform* platform,
                      StartupDep* startup_dep,
                      const Flags& flags,
                      const base::FilePath& root,
                      const base::FilePath& stateful,
                      std::unique_ptr<MountVarAndHomeChronosInterface> impl,
                      std::unique_ptr<libstorage::StorageContainerFactory>
                          storage_container_factory);

  bool DoMountVarAndHomeChronos(
      std::optional<encryption::EncryptionKey> key) override;
  base::FilePath GetKeyBackupFile() override;
};

}  // namespace startup

#endif  // INIT_STARTUP_TEST_MODE_MOUNT_HELPER_H_