uefi-raw: Add EFI_ATA_PASS_THRU_PROTOCOL bindings

Author: seijikun

uefi-raw/CHANGELOG.md

@@ -10,6 +10,7 @@
 - Added `DiskInfoProtocol`.
 - Added `ExtScsiPassThruProtocol`.
 - Added `NvmExpressPassThruProtocol`.
+- Added `AtaPassThruProtocol`.
 
 
 # uefi-raw - 0.10.0 (2025-02-07)

uefi-raw/src/protocol/ata.rs

@@ -0,0 +1,137 @@
+// SPDX-License-Identifier: MIT OR Apache-2.0
+
+use core::ffi::c_void;
+use crate::Status;
+use uguid::{guid, Guid};
+use super::device_path::DevicePathProtocol;
+
+#[derive(Debug)]
+#[repr(C)]
+pub struct AtaPassThruMode {
+    attributes: u32,
+    io_align: u32,
+}
+
+newtype_enum! {
+    /// Corresponds to the `EFI_ATA_PASS_THRU_PROTOCOL_*` defines.
+    #[derive(Default)]
+    pub enum AtaPassThruCommandProtocol: u8 => {
+        ATA_HARDWARE_RESET = 0x00,
+        ATA_SOFTWARE_RESET = 0x01,
+        ATA_NON_DATA = 0x02,
+        PIO_DATA_IN = 0x04,
+        PIO_DATA_OUT = 0x05,
+        DMA = 0x06,
+        DMA_QUEUED = 0x07,
+        DEVICE_DIAGNOSTIC = 0x08,
+        DEVICE_RESET = 0x09,
+        UDMA_DATA_IN = 0x0A,
+        UDMA_DATA_OUT = 0x0B,
+        FPDMA = 0x0C,
+        RETURN_RESPONSE = 0xFF,
+    }
+}
+
+newtype_enum! {
+    /// Corresponds to the `EFI_ATA_PASS_THRU_LENGTH_*` defines.
+    #[derive(Default)]
+    pub enum AtaPassThruLength: u8 => {
+        BYTES = 0x80,
+        MASK = 0x70,
+        NO_DATA_TRANSFER = 0x00,
+        FEATURES = 0x10,
+        SECTOR_COUNT = 0x20,
+        TPSIU = 0x30,
+        COUNT = 0x0F,
+    }
+}
+
+#[derive(Debug)]
+#[repr(C)]
+pub struct AtaPassThruStatusBlock {
+    reserved1: [u8; 2],
+    ata_status: u8,
+    ata_error: u8,
+    ata_sector_number: u8,
+    ata_cylinder_low: u8,
+    ata_cylinder_high: u8,
+    ata_device_head: u8,
+    ata_sector_number_exp: u8,
+    ata_cylinder_low_exp: u8,
+    ata_cylinder_high_exp: u8,
+    reserved2: u8,
+    ata_sector_count: u8,
+    ata_sector_count_exp: u8,
+    reserved3: [u8; 6],
+}
+
+#[derive(Debug)]
+#[repr(C)]
+pub struct AtaPassThruCommandBlock {
+    reserved1: [u8; 2],
+    ata_command: u8,
+    ata_features: u8,
+    ata_sector_number: u8,
+    ata_cylinder_low: u8,
+    ata_cylinder_high: u8,
+    ata_device_head: u8,
+    ata_sector_number_exp: u8,
+    ata_cylinder_low_exp: u8,
+    ata_cylinder_high_exp: u8,
+    ata_features_exp: u8,
+    ata_sector_count: u8,
+    ata_sector_count_exp: u8,
+    reserved2: [u8; 6],
+}
+
+#[derive(Debug)]
+#[repr(C)]
+pub struct AtaPassThruCommandPacket {
+    asb: *mut AtaPassThruStatusBlock,
+    acb: *const AtaPassThruCommandBlock,
+    timeout: u64,
+    in_data_buffer: *mut c_void,
+    out_data_buffer: *const c_void,
+    protocol: AtaPassThruCommandProtocol,
+    length: AtaPassThruLength,
+}
+
+#[derive(Debug)]
+#[repr(C)]
+pub struct AtaPassThruProtocol {
+    pub mode: *const AtaPassThruMode,
+    pub pass_thru: unsafe extern "efiapi" fn(
+        this: *const Self,
+        port: u16,
+        port_multiplier_port: u16,
+        packet: *mut AtaPassThruCommandPacket,
+        event: *mut c_void,
+    ) -> Status,
+    pub get_next_port: unsafe extern "efiapi" fn(this: *const Self, port: *mut u16) -> Status,
+    pub get_next_device: unsafe extern "efiapi" fn(
+        this: *const Self,
+        port: u16,
+        port_multiplier_port: *mut u16,
+    ) -> Status,
+    pub build_device_path: unsafe extern "efiapi" fn(
+        this: *const Self,
+        port: u16,
+        port_multiplier_port: u16,
+        device_path: *mut *mut DevicePathProtocol,
+    ) -> Status,
+    pub get_device: unsafe extern "efiapi" fn(
+        this: *const Self,
+        device_path: *const DevicePathProtocol,
+        port: *mut u16,
+        port_multiplier_port: *mut u16,
+    ) -> Status,
+    pub reset_port: unsafe extern "efiapi" fn(this: *const Self, port: u16) -> Status,
+    pub reset_device: unsafe extern "efiapi" fn(
+        this: *const Self,
+        port: u16,
+        port_multiplier_port: u16,
+    ) -> Status,
+}
+impl AtaPassThruProtocol {
+    pub const GUID: Guid = guid!("1d3de7f0-0807-424f-aa69-11a54e19a46f");
+}

uefi-raw/src/protocol/mod.rs

@@ -6,6 +6,7 @@
 //! ID. They can be implemented by a UEFI driver or occasionally by a
 //! UEFI application.
 
+pub mod ata;
 pub mod block;
 pub mod console;
 pub mod device_path;

uefi/src/proto/media/disk_info.rs

@@ -48,7 +48,7 @@ pub struct DeviceLocationInfo {
     /// For AHCI, this returns the port.
     pub channel: u32,
     /// For IDE, this contains whether the device is master or slave.
-    /// For AHCI, this returns the port-multiplier.
+    /// For AHCI, this returns the port multiplier port.
     pub device: u32,
 }