Bananymous
51fd7a607d
We did not correctly wait until controller is ready to receive data on write command. Also remove possible kernel panic if controller sends unexpected interrupts
64 lines
1.4 KiB
C++
64 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include <BAN/ByteSpan.h>
|
|
#include <BAN/RefPtr.h>
|
|
#include <BAN/Vector.h>
|
|
#include <kernel/Interruptable.h>
|
|
#include <kernel/Lock/Mutex.h>
|
|
|
|
namespace Kernel
|
|
{
|
|
|
|
class ATADevice;
|
|
|
|
class ATABus : public BAN::RefCounted<ATABus>, public Interruptable
|
|
{
|
|
public:
|
|
enum class DeviceType
|
|
{
|
|
ATA,
|
|
ATAPI,
|
|
};
|
|
|
|
public:
|
|
static BAN::ErrorOr<BAN::RefPtr<ATABus>> create(uint16_t base, uint16_t ctrl, uint8_t irq);
|
|
|
|
BAN::ErrorOr<void> read(ATADevice&, uint64_t lba, uint64_t sector_count, BAN::ByteSpan);
|
|
BAN::ErrorOr<void> write(ATADevice&, uint64_t lba, uint64_t sector_count, BAN::ConstByteSpan);
|
|
|
|
virtual void handle_irq() override;
|
|
|
|
private:
|
|
ATABus(uint16_t base, uint16_t ctrl)
|
|
: m_base(base)
|
|
, m_ctrl(ctrl)
|
|
{}
|
|
BAN::ErrorOr<void> initialize();
|
|
|
|
BAN::ErrorOr<void> send_command(ATADevice&, uint64_t lba, uint64_t sector_count, bool write);
|
|
|
|
void select_device(bool is_secondary);
|
|
BAN::ErrorOr<DeviceType> identify(bool is_secondary, BAN::Span<uint16_t> buffer);
|
|
|
|
uint8_t io_read(uint16_t);
|
|
void io_write(uint16_t, uint8_t);
|
|
void read_buffer(uint16_t, uint16_t*, size_t);
|
|
void write_buffer(uint16_t, const uint16_t*, size_t);
|
|
BAN::ErrorOr<void> wait(bool);
|
|
BAN::Error error();
|
|
|
|
private:
|
|
const uint16_t m_base;
|
|
const uint16_t m_ctrl;
|
|
Mutex m_mutex;
|
|
|
|
ThreadBlocker m_thread_blocker;
|
|
|
|
// Non-owning pointers
|
|
BAN::Vector<ATADevice*> m_devices;
|
|
|
|
friend class ATAController;
|
|
};
|
|
|
|
}
|