| // Copyright (C) 2023 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| //! Pleasant Rust bindings for libnativewindow, including AHardwareBuffer |
| |
| extern crate nativewindow_bindgen as ffi; |
| |
| pub use ffi::{AHardwareBuffer_Format, AHardwareBuffer_UsageFlags}; |
| |
| use binder::{ |
| binder_impl::{ |
| BorrowedParcel, Deserialize, DeserializeArray, DeserializeOption, Serialize, |
| SerializeArray, SerializeOption, NON_NULL_PARCELABLE_FLAG, NULL_PARCELABLE_FLAG, |
| }, |
| unstable_api::{status_result, AsNative}, |
| StatusCode, |
| }; |
| use ffi::{AHardwareBuffer, AHardwareBuffer_readFromParcel, AHardwareBuffer_writeToParcel}; |
| use std::fmt::{self, Debug, Formatter}; |
| use std::mem::ManuallyDrop; |
| use std::ptr::{self, null_mut, NonNull}; |
| |
| /// Wrapper around an opaque C `AHardwareBuffer`. |
| #[derive(PartialEq, Eq)] |
| pub struct HardwareBuffer(NonNull<AHardwareBuffer>); |
| |
| impl HardwareBuffer { |
| /// Test whether the given format and usage flag combination is allocatable. If this function |
| /// returns true, it means that a buffer with the given description can be allocated on this |
| /// implementation, unless resource exhaustion occurs. If this function returns false, it means |
| /// that the allocation of the given description will never succeed. |
| /// |
| /// Available since API 29 |
| pub fn is_supported( |
| width: u32, |
| height: u32, |
| layers: u32, |
| format: AHardwareBuffer_Format::Type, |
| usage: AHardwareBuffer_UsageFlags, |
| stride: u32, |
| ) -> bool { |
| let buffer_desc = ffi::AHardwareBuffer_Desc { |
| width, |
| height, |
| layers, |
| format, |
| usage: usage.0, |
| stride, |
| rfu0: 0, |
| rfu1: 0, |
| }; |
| // SAFETY: *buffer_desc will never be null. |
| let status = unsafe { ffi::AHardwareBuffer_isSupported(&buffer_desc) }; |
| |
| status == 1 |
| } |
| |
| /// Allocates a buffer that matches the passed AHardwareBuffer_Desc. If allocation succeeds, the |
| /// buffer can be used according to the usage flags specified in its description. If a buffer is |
| /// used in ways not compatible with its usage flags, the results are undefined and may include |
| /// program termination. |
| /// |
| /// Available since API level 26. |
| #[inline] |
| pub fn new( |
| width: u32, |
| height: u32, |
| layers: u32, |
| format: AHardwareBuffer_Format::Type, |
| usage: AHardwareBuffer_UsageFlags, |
| ) -> Option<Self> { |
| let buffer_desc = ffi::AHardwareBuffer_Desc { |
| width, |
| height, |
| layers, |
| format, |
| usage: usage.0, |
| stride: 0, |
| rfu0: 0, |
| rfu1: 0, |
| }; |
| let mut ptr = ptr::null_mut(); |
| // SAFETY: The returned pointer is valid until we drop/deallocate it. The function may fail |
| // and return a status, but we check it later. |
| let status = unsafe { ffi::AHardwareBuffer_allocate(&buffer_desc, &mut ptr) }; |
| |
| if status == 0 { |
| Some(Self(NonNull::new(ptr).expect("Allocated AHardwareBuffer was null"))) |
| } else { |
| None |
| } |
| } |
| |
| /// Adopts the raw pointer and wraps it in a Rust AHardwareBuffer. |
| /// |
| /// # Errors |
| /// |
| /// Will panic if buffer_ptr is null. |
| /// |
| /// # Safety |
| /// |
| /// This function adopts the pointer but does NOT increment the refcount on the buffer. If the |
| /// caller uses the pointer after the created object is dropped it will cause a memory leak. |
| pub unsafe fn from_raw(buffer_ptr: NonNull<AHardwareBuffer>) -> Self { |
| Self(buffer_ptr) |
| } |
| |
| /// Get the internal |AHardwareBuffer| pointer without decrementing the refcount. This can |
| /// be used to provide a pointer to the AHB for a C/C++ API over the FFI. |
| pub fn into_raw(self) -> NonNull<AHardwareBuffer> { |
| let buffer = ManuallyDrop::new(self); |
| buffer.0 |
| } |
| |
| /// Get the system wide unique id for an AHardwareBuffer. This function may panic in extreme |
| /// and undocumented circumstances. |
| /// |
| /// Available since API level 31. |
| pub fn id(&self) -> u64 { |
| let mut out_id = 0; |
| // SAFETY: The AHardwareBuffer pointer we pass is guaranteed to be non-null and valid |
| // because it must have been allocated by `AHardwareBuffer_allocate`, |
| // `AHardwareBuffer_readFromParcel` or the caller of `from_raw` and we have not yet |
| // released it. The id pointer must be valid because it comes from a reference. |
| let status = unsafe { ffi::AHardwareBuffer_getId(self.0.as_ptr(), &mut out_id) }; |
| assert_eq!(status, 0, "id() failed for AHardwareBuffer with error code: {status}"); |
| |
| out_id |
| } |
| |
| /// Get the width of this buffer |
| pub fn width(&self) -> u32 { |
| self.description().width |
| } |
| |
| /// Get the height of this buffer |
| pub fn height(&self) -> u32 { |
| self.description().height |
| } |
| |
| /// Get the number of layers of this buffer |
| pub fn layers(&self) -> u32 { |
| self.description().layers |
| } |
| |
| /// Get the format of this buffer |
| pub fn format(&self) -> AHardwareBuffer_Format::Type { |
| self.description().format |
| } |
| |
| /// Get the usage bitvector of this buffer |
| pub fn usage(&self) -> AHardwareBuffer_UsageFlags { |
| AHardwareBuffer_UsageFlags(self.description().usage) |
| } |
| |
| /// Get the stride of this buffer |
| pub fn stride(&self) -> u32 { |
| self.description().stride |
| } |
| |
| fn description(&self) -> ffi::AHardwareBuffer_Desc { |
| let mut buffer_desc = ffi::AHardwareBuffer_Desc { |
| width: 0, |
| height: 0, |
| layers: 0, |
| format: 0, |
| usage: 0, |
| stride: 0, |
| rfu0: 0, |
| rfu1: 0, |
| }; |
| // SAFETY: neither the buffer nor AHardwareBuffer_Desc pointers will be null. |
| unsafe { ffi::AHardwareBuffer_describe(self.0.as_ref(), &mut buffer_desc) }; |
| buffer_desc |
| } |
| } |
| |
| impl Drop for HardwareBuffer { |
| fn drop(&mut self) { |
| // SAFETY: The AHardwareBuffer pointer we pass is guaranteed to be non-null and valid |
| // because it must have been allocated by `AHardwareBuffer_allocate`, |
| // `AHardwareBuffer_readFromParcel` or the caller of `from_raw` and we have not yet |
| // released it. |
| unsafe { ffi::AHardwareBuffer_release(self.0.as_ptr()) } |
| } |
| } |
| |
| impl Debug for HardwareBuffer { |
| fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result { |
| f.debug_struct("HardwareBuffer").field("id", &self.id()).finish() |
| } |
| } |
| |
| impl Clone for HardwareBuffer { |
| fn clone(&self) -> Self { |
| // SAFETY: ptr is guaranteed to be non-null and the acquire can not fail. |
| unsafe { ffi::AHardwareBuffer_acquire(self.0.as_ptr()) }; |
| Self(self.0) |
| } |
| } |
| |
| impl Serialize for HardwareBuffer { |
| fn serialize(&self, parcel: &mut BorrowedParcel) -> Result<(), StatusCode> { |
| SerializeOption::serialize_option(Some(self), parcel) |
| } |
| } |
| |
| impl SerializeOption for HardwareBuffer { |
| fn serialize_option( |
| this: Option<&Self>, |
| parcel: &mut BorrowedParcel, |
| ) -> Result<(), StatusCode> { |
| if let Some(this) = this { |
| parcel.write(&NON_NULL_PARCELABLE_FLAG)?; |
| |
| let status = |
| // SAFETY: The AHardwareBuffer pointer we pass is guaranteed to be non-null and valid |
| // because it must have been allocated by `AHardwareBuffer_allocate`, |
| // `AHardwareBuffer_readFromParcel` or the caller of `from_raw` and we have not yet |
| // released it. |
| unsafe { AHardwareBuffer_writeToParcel(this.0.as_ptr(), parcel.as_native_mut()) }; |
| status_result(status) |
| } else { |
| parcel.write(&NULL_PARCELABLE_FLAG) |
| } |
| } |
| } |
| |
| impl Deserialize for HardwareBuffer { |
| type UninitType = Option<Self>; |
| |
| fn uninit() -> Option<Self> { |
| None |
| } |
| |
| fn from_init(value: Self) -> Option<Self> { |
| Some(value) |
| } |
| |
| fn deserialize(parcel: &BorrowedParcel) -> Result<Self, StatusCode> { |
| DeserializeOption::deserialize_option(parcel) |
| .transpose() |
| .unwrap_or(Err(StatusCode::UNEXPECTED_NULL)) |
| } |
| } |
| |
| impl DeserializeOption for HardwareBuffer { |
| fn deserialize_option(parcel: &BorrowedParcel) -> Result<Option<Self>, StatusCode> { |
| let present: i32 = parcel.read()?; |
| match present { |
| NULL_PARCELABLE_FLAG => Ok(None), |
| NON_NULL_PARCELABLE_FLAG => { |
| let mut buffer = null_mut(); |
| |
| let status = |
| // SAFETY: Both pointers must be valid because they are obtained from references. |
| // `AHardwareBuffer_readFromParcel` doesn't store them or do anything else special |
| // with them. If it returns success then it will have allocated a new |
| // `AHardwareBuffer` and incremented the reference count, so we can use it until we |
| // release it. |
| unsafe { AHardwareBuffer_readFromParcel(parcel.as_native(), &mut buffer) }; |
| |
| status_result(status)?; |
| |
| Ok(Some(Self(NonNull::new(buffer).expect( |
| "AHardwareBuffer_readFromParcel returned success but didn't allocate buffer", |
| )))) |
| } |
| _ => Err(StatusCode::BAD_VALUE), |
| } |
| } |
| } |
| |
| impl SerializeArray for HardwareBuffer {} |
| |
| impl DeserializeArray for HardwareBuffer {} |
| |
| // SAFETY: The underlying *AHardwareBuffers can be moved between threads. |
| unsafe impl Send for HardwareBuffer {} |
| |
| // SAFETY: The underlying *AHardwareBuffers can be used from multiple threads. |
| // |
| // AHardwareBuffers are backed by C++ GraphicBuffers, which are mostly immutable. The only cases |
| // where they are not immutable are: |
| // |
| // - reallocation (which is never actually done across the codebase and requires special |
| // privileges/platform code access to do) |
| // - "locking" for reading/writing (which is explicitly allowed to be done across multiple threads |
| // according to the docs on the underlying gralloc calls) |
| unsafe impl Sync for HardwareBuffer {} |
| |
| #[cfg(test)] |
| mod test { |
| use super::*; |
| |
| #[test] |
| fn create_valid_buffer_returns_ok() { |
| let buffer = HardwareBuffer::new( |
| 512, |
| 512, |
| 1, |
| AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM, |
| AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN, |
| ); |
| assert!(buffer.is_some()); |
| } |
| |
| #[test] |
| fn create_invalid_buffer_returns_err() { |
| let buffer = HardwareBuffer::new(512, 512, 1, 0, AHardwareBuffer_UsageFlags(0)); |
| assert!(buffer.is_none()); |
| } |
| |
| #[test] |
| fn from_raw_allows_getters() { |
| let buffer_desc = ffi::AHardwareBuffer_Desc { |
| width: 1024, |
| height: 512, |
| layers: 1, |
| format: AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM, |
| usage: AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN.0, |
| stride: 0, |
| rfu0: 0, |
| rfu1: 0, |
| }; |
| let mut raw_buffer_ptr = ptr::null_mut(); |
| |
| // SAFETY: The pointers are valid because they come from references, and |
| // `AHardwareBuffer_allocate` doesn't retain them after it returns. |
| let status = unsafe { ffi::AHardwareBuffer_allocate(&buffer_desc, &mut raw_buffer_ptr) }; |
| assert_eq!(status, 0); |
| |
| // SAFETY: The pointer must be valid because it was just allocated successfully, and we |
| // don't use it after calling this. |
| let buffer = unsafe { HardwareBuffer::from_raw(NonNull::new(raw_buffer_ptr).unwrap()) }; |
| assert_eq!(buffer.width(), 1024); |
| } |
| |
| #[test] |
| fn basic_getters() { |
| let buffer = HardwareBuffer::new( |
| 1024, |
| 512, |
| 1, |
| AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM, |
| AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN, |
| ) |
| .expect("Buffer with some basic parameters was not created successfully"); |
| |
| assert_eq!(buffer.width(), 1024); |
| assert_eq!(buffer.height(), 512); |
| assert_eq!(buffer.layers(), 1); |
| assert_eq!(buffer.format(), AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM); |
| assert_eq!( |
| buffer.usage(), |
| AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN |
| ); |
| } |
| |
| #[test] |
| fn id_getter() { |
| let buffer = HardwareBuffer::new( |
| 1024, |
| 512, |
| 1, |
| AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM, |
| AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN, |
| ) |
| .expect("Buffer with some basic parameters was not created successfully"); |
| |
| assert_ne!(0, buffer.id()); |
| } |
| |
| #[test] |
| fn clone() { |
| let buffer = HardwareBuffer::new( |
| 1024, |
| 512, |
| 1, |
| AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM, |
| AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN, |
| ) |
| .expect("Buffer with some basic parameters was not created successfully"); |
| let buffer2 = buffer.clone(); |
| |
| assert_eq!(buffer, buffer2); |
| } |
| |
| #[test] |
| fn into_raw() { |
| let buffer = HardwareBuffer::new( |
| 1024, |
| 512, |
| 1, |
| AHardwareBuffer_Format::AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM, |
| AHardwareBuffer_UsageFlags::AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN, |
| ) |
| .expect("Buffer with some basic parameters was not created successfully"); |
| let buffer2 = buffer.clone(); |
| |
| let raw_buffer = buffer.into_raw(); |
| // SAFETY: This is the same pointer we had before. |
| let remade_buffer = unsafe { HardwareBuffer::from_raw(raw_buffer) }; |
| |
| assert_eq!(remade_buffer, buffer2); |
| } |
| } |