mirror of
https://github.com/RGBCube/serenity
synced 2025-05-31 19:58:11 +00:00
b204da94b0
Storage controllers are initialized during init and are never modified. NonnullRefPtr can be safely used instead of the NonnullLockRefPtr. This also fixes one of the UB issue that was there when using an NVMe device because of NonnullLockRefPtr. We can add proper locking when we need to modify the storage controllers after init.
73 lines
2.4 KiB
C++
73 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) 2020-2022, Liav A. <liavalb@hotmail.co.il>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/IntrusiveList.h>
|
|
#include <AK/Types.h>
|
|
#include <Kernel/FileSystem/FileSystem.h>
|
|
#include <Kernel/Library/NonnullLockRefPtr.h>
|
|
#include <Kernel/Storage/DiskPartition.h>
|
|
#include <Kernel/Storage/StorageController.h>
|
|
#include <Kernel/Storage/StorageDevice.h>
|
|
#include <LibPartition/PartitionTable.h>
|
|
|
|
namespace Kernel {
|
|
|
|
class ATAController;
|
|
class NVMeController;
|
|
class StorageManagement {
|
|
|
|
public:
|
|
StorageManagement();
|
|
void initialize(StringView boot_argument, bool force_pio, bool nvme_poll);
|
|
static StorageManagement& the();
|
|
|
|
NonnullLockRefPtr<FileSystem> root_filesystem() const;
|
|
|
|
static MajorNumber storage_type_major_number();
|
|
static MinorNumber generate_storage_minor_number();
|
|
|
|
static MinorNumber generate_partition_minor_number();
|
|
|
|
static u32 generate_controller_id();
|
|
|
|
static u32 generate_relative_nvme_controller_id(Badge<NVMeController>);
|
|
static u32 generate_relative_ata_controller_id(Badge<ATAController>);
|
|
|
|
void remove_device(StorageDevice&);
|
|
|
|
private:
|
|
void enumerate_pci_controllers(bool force_pio, bool nvme_poll);
|
|
void enumerate_storage_devices();
|
|
void enumerate_disk_partitions();
|
|
|
|
void determine_boot_device();
|
|
void determine_boot_device_with_partition_uuid();
|
|
|
|
void resolve_partition_from_boot_device_parameter(StorageDevice const& chosen_storage_device, StringView boot_device_prefix);
|
|
void determine_boot_device_with_logical_unit_number();
|
|
void determine_block_boot_device();
|
|
void determine_ramdisk_boot_device();
|
|
void determine_nvme_boot_device();
|
|
void determine_ata_boot_device();
|
|
void determine_hardware_relative_boot_device(StringView relative_hardware_prefix, Function<bool(StorageDevice const&)> filter_device_callback);
|
|
Array<unsigned, 3> extract_boot_device_address_parameters(StringView device_prefix);
|
|
Optional<unsigned> extract_boot_device_partition_number_parameter(StringView device_prefix);
|
|
|
|
void dump_storage_devices_and_partitions() const;
|
|
|
|
ErrorOr<NonnullOwnPtr<Partition::PartitionTable>> try_to_initialize_partition_table(StorageDevice&) const;
|
|
|
|
LockRefPtr<BlockDevice> boot_block_device() const;
|
|
|
|
StringView m_boot_argument;
|
|
LockWeakPtr<BlockDevice> m_boot_block_device;
|
|
Vector<NonnullRefPtr<StorageController>> m_controllers;
|
|
IntrusiveList<&StorageDevice::m_list_node> m_storage_devices;
|
|
};
|
|
|
|
}
|