48 lines
1.1 KiB
C++
48 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <glacier/memory/unique_ptr.h>
|
|
#include <glacier/status/error_or.h>
|
|
#include <mammoth/thread.h>
|
|
#include <ztypes.h>
|
|
|
|
#include "ahci/ahci.h"
|
|
#include "ahci/ahci_device.h"
|
|
|
|
class AhciDriver {
|
|
public:
|
|
static glcr::ErrorOr<glcr::UniquePtr<AhciDriver>> Init(
|
|
OwnedMemoryRegion&& ahci_phys);
|
|
glcr::ErrorCode RegisterIrq();
|
|
|
|
void InterruptLoop();
|
|
|
|
glcr::ErrorOr<AhciDevice*> GetDevice(uint64_t id);
|
|
|
|
void DumpCapabilities();
|
|
void DumpPorts();
|
|
|
|
private:
|
|
OwnedMemoryRegion pci_region_;
|
|
PciDeviceHeader* pci_device_header_ = nullptr;
|
|
OwnedMemoryRegion ahci_region_;
|
|
AhciHba* ahci_hba_ = nullptr;
|
|
|
|
// TODO: Allocate these dynamically.
|
|
AhciDevice* devices_[32];
|
|
|
|
Thread irq_thread_;
|
|
uint64_t irq_port_cap_ = 0;
|
|
|
|
uint64_t num_ports_;
|
|
uint64_t num_commands_;
|
|
|
|
glcr::ErrorCode LoadCapabilities();
|
|
glcr::ErrorCode LoadHbaRegisters();
|
|
glcr::ErrorCode LoadDevices();
|
|
|
|
AhciDriver(OwnedMemoryRegion&& pci_region)
|
|
: pci_region_(glcr::Move(pci_region)),
|
|
pci_device_header_(
|
|
reinterpret_cast<PciDeviceHeader*>(pci_region_.vaddr())) {}
|
|
};
|