commit 61c340959e7c33f9507e50a470077c6db0991b70
author Imre Kis <imre.kis@arm.com> Thu Apr 10 14:14:38 2025 +0200
committer Imre Kis <imre.kis@arm.com> Mon Apr 14 13:43:10 2025 +0200
tree 5abb77b72bb720e2cedb237e6b9736fb017ef331
parent bbef287e3222d814632e1dc543084b7bff29aa20

Add success type for FFA_PARTITION_INFO_GET

Add specialized success type for converting between
FFA_PARTITION_INFO_GET return arguments and generic success arguments.

Signed-off-by: Imre Kis <imre.kis@arm.com>
Change-Id: Iad79dc3eb44c7199e0715c3e997661bf96332631

src/lib.rs

diff --git a/src/lib.rs b/src/lib.rs
index 174ea9b..616ce02 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -207,6 +207,7 @@
 /// * `FFA_FEATURES` - [`SuccessArgsFeatures`]
 /// * `FFA_ID_GET` - [`SuccessArgsIdGet`]
 /// * `FFA_SPM_ID_GET` - [`SuccessArgsSpmIdGet`]
+/// * `FFA_PARTITION_INFO_GET` - [`partition_info::SuccessArgsPartitionInfoGet`]
 #[derive(Debug, Eq, PartialEq, Clone, Copy)]
 pub enum SuccessArgs {
     Args32([u32; 6]),

src/partition_info.rs

diff --git a/src/partition_info.rs b/src/partition_info.rs
index 23e4c06..d8ce729 100644
--- a/src/partition_info.rs
+++ b/src/partition_info.rs
@@ -10,7 +10,7 @@
 // This module uses FF-A v1.1 types by default.
 // FF-A v1.2 specified some previously reserved bits in the partition info properties field, but
 // this doesn't change the descriptor format.
-use crate::{ffa_v1_1::partition_info_descriptor, Version};
+use crate::{ffa_v1_1::partition_info_descriptor, PartitionInfoGetFlags, SuccessArgs, Version};
 
 // Sanity check to catch if the descriptor format is changed.
 const _: () = assert!(
@@ -261,6 +261,39 @@
     }
 }
 
+/// `FFA_PARTITION_INFO_GET` specific success args structure.
+#[derive(Debug, Eq, PartialEq, Clone, Copy)]
+pub struct SuccessArgsPartitionInfoGet {
+    pub count: u32,
+    pub size: Option<u32>,
+}
+
+impl From<SuccessArgsPartitionInfoGet> for SuccessArgs {
+    fn from(value: SuccessArgsPartitionInfoGet) -> Self {
+        SuccessArgs::Args32([value.count, value.size.unwrap_or(0), 0, 0, 0, 0])
+    }
+}
+
+impl TryFrom<(PartitionInfoGetFlags, SuccessArgs)> for SuccessArgsPartitionInfoGet {
+    type Error = crate::Error;
+
+    fn try_from(value: (PartitionInfoGetFlags, SuccessArgs)) -> Result<Self, Self::Error> {
+        let (flags, value) = value;
+        let args = value.try_get_args32()?;
+
+        let size = if !flags.count_only {
+            Some(args[1])
+        } else {
+            None
+        };
+
+        Ok(Self {
+            count: args[0],
+            size,
+        })
+    }
+}
+
 /// Iterator of partition information descriptors.
 pub struct PartitionInfoIterator<'a> {
     version: Version,