| // Copyright 2021 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_UTILS_H_ |
| #define INIT_UTILS_H_ |
| |
| #include <string> |
| #include <vector> |
| |
| #include <base/files/file.h> |
| #include <base/files/file_path.h> |
| #include <base/values.h> |
| #include <brillo/brillo_export.h> |
| #include <libstorage/platform/platform.h> |
| |
| namespace utils { |
| |
| // Get the root device filepath, optionally removing the |
| // partition number. |
| // Return an empty path if unable to find the root device. |
| base::FilePath BRILLO_EXPORT GetRootDevice(bool strip_partition); |
| |
| // Helper function to read a file to int |
| bool BRILLO_EXPORT ReadFileToInt(const base::FilePath& path, int* value); |
| |
| // Run shutdown. |
| void BRILLO_EXPORT Reboot(); |
| |
| void BRILLO_EXPORT Restorecon(libstorage::Platform* platform_, |
| const base::FilePath& path, |
| const std::vector<base::FilePath>& exclude, |
| bool is_recursive, |
| bool set_digests); |
| |
| // Searches `drive_name` for the partition labeled `partition_label` and |
| // returns its partition number if exactly one partition was found. Returns |
| // -1 on error. |
| int BRILLO_EXPORT GetPartitionNumber(const base::FilePath& drive_name, |
| const std::string& partition_label); |
| |
| // Get the partition number for the given key, |
| // e.g. "PARTITION_NUM_STATE". Fails with a `CHECK()` if any error occurs. |
| int BRILLO_EXPORT GetPartitionNumFromImageVars( |
| const base::Value::Dict& image_dict, std::string_view key); |
| |
| // Splits a device path, for example /dev/mmcblk0p1, /dev/sda3, |
| // into the base device and partition numbers, |
| // which would be respectively /dev/mmcblk0p, 1; and /dev/sda, 3; |
| // Returns true on success. |
| bool BRILLO_EXPORT GetDevicePathComponents(const base::FilePath& device, |
| std::string* base_device_out, |
| int* partition_out); |
| |
| // Reads successful and priority metadata from partition numbered |
| // `partition_number` on `disk`, storing the results in `successful_out` and |
| // `priority_out`, respectively. Returns true on success. |
| // |
| // successful is a 1 bit value indicating if a kernel partition |
| // has been successfully booted, while priority is a 4 bit value |
| // indicating what order the kernel partitions should be booted in, 15 being |
| // the highest, 1 the lowest, and 0 meaning not bootable. |
| // More information on partition metadata is available at. |
| // https://www.chromium.org/chromium-os/chromiumos-design-docs/disk-format |
| bool BRILLO_EXPORT ReadPartitionMetadata(const base::FilePath& disk, |
| int partition_number, |
| bool* successful_out, |
| int* priority_out); |
| |
| // Make sure the kernel partition numbered `kernel_partition` is still |
| // bootable after being wiped. The system may be in AU state that active |
| // kernel does not have "successful" bit set to 1, but the kernel has been |
| // successfully booted. |
| void BRILLO_EXPORT EnsureKernelIsBootable(const base::FilePath root_disk, |
| int kernel_partition); |
| |
| } // namespace utils |
| |
| #endif // INIT_UTILS_H_ |