use crate::binder::AsNative;
use crate::error::{Result, StatusCode, status_result};
use crate::proxy::SpIBinder;
use crate::sys;
use alloc::vec::Vec;
use core::convert::TryInto;
use core::fmt;
use core::marker::PhantomData;
use core::mem::ManuallyDrop;
use core::ptr::{self, NonNull};
mod file_descriptor;
mod parcelable;
mod parcelable_holder;
pub use self::file_descriptor::ParcelFileDescriptor;
pub use self::parcelable::{
Deserialize, DeserializeArray, DeserializeOption, NON_NULL_PARCELABLE_FLAG,
NULL_PARCELABLE_FLAG, Parcelable, Serialize, SerializeArray, SerializeOption,
UnstructuredParcelable,
};
pub use self::parcelable_holder::{ParcelableHolder, ParcelableMetadata};
#[repr(transparent)]
pub struct Parcel {
ptr: NonNull<sys::AParcel>,
}
unsafe impl Send for Parcel {}
#[repr(transparent)]
pub struct BorrowedParcel<'a> {
ptr: NonNull<sys::AParcel>,
_lifetime: PhantomData<&'a mut Parcel>,
}
impl Parcel {
pub fn new() -> Parcel {
let ptr = unsafe { sys::AParcel_create() };
Self { ptr: NonNull::new(ptr).expect("AParcel_create returned null pointer") }
}
pub unsafe fn from_raw(ptr: *mut sys::AParcel) -> Option<Parcel> {
NonNull::new(ptr).map(|ptr| Self { ptr })
}
pub fn unmarshal(buffer: &[u8]) -> Parcel {
let parcel = Self::new();
status_result(unsafe {
sys::AParcel_unmarshal(parcel.ptr.as_ptr(), buffer.as_ptr(), buffer.len())
})
.expect("AParcel_unmarshal failed");
parcel
}
pub(crate) fn into_raw(self) -> *mut sys::AParcel {
let ptr = self.ptr.as_ptr();
let _ = ManuallyDrop::new(self);
ptr
}
pub fn borrowed(&mut self) -> BorrowedParcel<'_> {
BorrowedParcel { ptr: self.ptr, _lifetime: PhantomData }
}
pub fn borrowed_ref(&self) -> &BorrowedParcel<'_> {
unsafe { &*(self as *const Parcel as *const BorrowedParcel<'_>) }
}
}
impl Default for Parcel {
fn default() -> Self {
Self::new()
}
}
impl Clone for Parcel {
fn clone(&self) -> Self {
let mut new_parcel = Self::new();
new_parcel
.borrowed()
.append_all_from(self.borrowed_ref())
.expect("Failed to append from Parcel");
new_parcel
}
}
impl<'a> BorrowedParcel<'a> {
pub unsafe fn from_raw(ptr: *mut sys::AParcel) -> Option<BorrowedParcel<'a>> {
Some(Self { ptr: NonNull::new(ptr)?, _lifetime: PhantomData })
}
pub fn reborrow(&mut self) -> BorrowedParcel<'_> {
BorrowedParcel { ptr: self.ptr, _lifetime: PhantomData }
}
}
unsafe impl AsNative<sys::AParcel> for Parcel {
fn as_native(&self) -> *const sys::AParcel {
self.ptr.as_ptr()
}
fn as_native_mut(&mut self) -> *mut sys::AParcel {
self.ptr.as_ptr()
}
}
unsafe impl AsNative<sys::AParcel> for BorrowedParcel<'_> {
fn as_native(&self) -> *const sys::AParcel {
self.ptr.as_ptr()
}
fn as_native_mut(&mut self) -> *mut sys::AParcel {
self.ptr.as_ptr()
}
}
impl BorrowedParcel<'_> {
#[cfg(not(android_ndk))]
pub fn mark_sensitive(&mut self) {
unsafe { sys::AParcel_markSensitive(self.as_native()) }
}
pub fn write<S: Serialize + ?Sized>(&mut self, parcelable: &S) -> Result<()> {
parcelable.serialize(self)
}
pub fn write_slice_size<T>(&mut self, slice: Option<&[T]>) -> Result<()> {
if let Some(slice) = slice {
let len: i32 = slice.len().try_into().or(Err(StatusCode::BAD_VALUE))?;
self.write(&len)
} else {
self.write(&-1i32)
}
}
pub fn sized_write<F>(&mut self, f: F) -> Result<()>
where
for<'b> F: FnOnce(&'b mut WritableSubParcel<'b>) -> Result<()>,
{
let start = self.get_data_position();
self.write(&0i32)?;
{
let mut subparcel = WritableSubParcel(self.reborrow());
f(&mut subparcel)?;
}
let end = self.get_data_position();
unsafe {
self.set_data_position(start)?;
}
assert!(end >= start);
self.write(&(end - start))?;
unsafe {
self.set_data_position(end)?;
}
Ok(())
}
pub fn get_data_position(&self) -> i32 {
unsafe { sys::AParcel_getDataPosition(self.as_native()) }
}
pub fn get_data_size(&self) -> i32 {
unsafe { sys::AParcel_getDataSize(self.as_native()) }
}
pub unsafe fn set_data_position(&self, pos: i32) -> Result<()> {
status_result(unsafe { sys::AParcel_setDataPosition(self.as_native(), pos) })
}
pub fn append_from(
&mut self,
other: &impl AsNative<sys::AParcel>,
start: i32,
size: i32,
) -> Result<()> {
let status = unsafe {
sys::AParcel_appendFrom(other.as_native(), self.as_native_mut(), start, size)
};
status_result(status)
}
pub fn append_all_from(&mut self, other: &impl AsNative<sys::AParcel>) -> Result<()> {
let size = unsafe { sys::AParcel_getDataSize(other.as_native()) };
self.append_from(other, 0, size)
}
}
pub struct WritableSubParcel<'a>(BorrowedParcel<'a>);
impl WritableSubParcel<'_> {
pub fn write<S: Serialize + ?Sized>(&mut self, parcelable: &S) -> Result<()> {
parcelable.serialize(&mut self.0)
}
}
impl Parcel {
#[cfg(not(android_ndk))]
pub fn mark_sensitive(&mut self) {
self.borrowed().mark_sensitive()
}
pub fn write<S: Serialize + ?Sized>(&mut self, parcelable: &S) -> Result<()> {
self.borrowed().write(parcelable)
}
pub fn write_slice_size<T>(&mut self, slice: Option<&[T]>) -> Result<()> {
self.borrowed().write_slice_size(slice)
}
pub fn sized_write<F>(&mut self, f: F) -> Result<()>
where
for<'b> F: FnOnce(&'b mut WritableSubParcel<'b>) -> Result<()>,
{
self.borrowed().sized_write(f)
}
pub fn get_data_position(&self) -> i32 {
self.borrowed_ref().get_data_position()
}
pub fn get_data_size(&self) -> i32 {
self.borrowed_ref().get_data_size()
}
pub unsafe fn set_data_position(&self, pos: i32) -> Result<()> {
unsafe { self.borrowed_ref().set_data_position(pos) }
}
pub fn append_from(
&mut self,
other: &impl AsNative<sys::AParcel>,
start: i32,
size: i32,
) -> Result<()> {
self.borrowed().append_from(other, start, size)
}
pub fn append_all_from(&mut self, other: &impl AsNative<sys::AParcel>) -> Result<()> {
self.borrowed().append_all_from(other)
}
}
impl BorrowedParcel<'_> {
pub fn read<D: Deserialize>(&self) -> Result<D> {
D::deserialize(self)
}
pub fn read_onto<D: Deserialize>(&self, x: &mut D) -> Result<()> {
x.deserialize_from(self)
}
pub fn sized_read<F>(&self, f: F) -> Result<()>
where
for<'b> F: FnOnce(ReadableSubParcel<'b>) -> Result<()>,
{
let start = self.get_data_position();
let parcelable_size: i32 = self.read()?;
if parcelable_size < 4 {
return Err(StatusCode::BAD_VALUE);
}
let end = start.checked_add(parcelable_size).ok_or(StatusCode::BAD_VALUE)?;
if end > self.get_data_size() {
return Err(StatusCode::NOT_ENOUGH_DATA);
}
let subparcel = ReadableSubParcel {
parcel: BorrowedParcel { ptr: self.ptr, _lifetime: PhantomData },
end_position: end,
};
f(subparcel)?;
unsafe {
self.set_data_position(end)?;
}
Ok(())
}
pub fn resize_out_vec<D: Default + Deserialize>(&self, out_vec: &mut Vec<D>) -> Result<()> {
let len: i32 = self.read()?;
if len < 0 {
return Err(StatusCode::UNEXPECTED_NULL);
}
let len = len.try_into().unwrap();
out_vec.resize_with(len, Default::default);
Ok(())
}
pub fn resize_nullable_out_vec<D: Default + Deserialize>(
&self,
out_vec: &mut Option<Vec<D>>,
) -> Result<()> {
let len: i32 = self.read()?;
if len < 0 {
*out_vec = None;
} else {
let len = len.try_into().unwrap();
let mut vec = Vec::with_capacity(len);
vec.resize_with(len, Default::default);
*out_vec = Some(vec);
}
Ok(())
}
}
pub struct ReadableSubParcel<'a> {
parcel: BorrowedParcel<'a>,
end_position: i32,
}
impl ReadableSubParcel<'_> {
pub fn read<D: Deserialize>(&self) -> Result<D> {
D::deserialize(&self.parcel)
}
pub fn has_more_data(&self) -> bool {
self.parcel.get_data_position() < self.end_position
}
}
impl Parcel {
pub fn read<D: Deserialize>(&self) -> Result<D> {
self.borrowed_ref().read()
}
pub fn read_onto<D: Deserialize>(&self, x: &mut D) -> Result<()> {
self.borrowed_ref().read_onto(x)
}
pub fn sized_read<F>(&self, f: F) -> Result<()>
where
for<'b> F: FnOnce(ReadableSubParcel<'b>) -> Result<()>,
{
self.borrowed_ref().sized_read(f)
}
pub fn resize_out_vec<D: Default + Deserialize>(&self, out_vec: &mut Vec<D>) -> Result<()> {
self.borrowed_ref().resize_out_vec(out_vec)
}
pub fn resize_nullable_out_vec<D: Default + Deserialize>(
&self,
out_vec: &mut Option<Vec<D>>,
) -> Result<()> {
self.borrowed_ref().resize_nullable_out_vec(out_vec)
}
pub fn marshal(&self) -> Vec<u8> {
let mut buffer = vec![0u8; self.get_data_size() as usize];
status_result(unsafe {
sys::AParcel_marshal(self.ptr.as_ptr(), buffer.as_mut_ptr(), 0, buffer.len())
})
.expect("Failed to marshal");
buffer
}
}
impl BorrowedParcel<'_> {
pub(crate) fn write_binder(&mut self, binder: Option<&SpIBinder>) -> Result<()> {
unsafe {
status_result(sys::AParcel_writeStrongBinder(
self.as_native_mut(),
binder.cloned().as_native_mut(),
))
}
}
pub(crate) fn read_binder(&self) -> Result<Option<SpIBinder>> {
let mut binder = ptr::null_mut();
let status = unsafe { sys::AParcel_readStrongBinder(self.as_native(), &mut binder) };
status_result(status)?;
Ok(unsafe { SpIBinder::from_raw(binder) })
}
}
impl Drop for Parcel {
fn drop(&mut self) {
unsafe { sys::AParcel_delete(self.ptr.as_ptr()) }
}
}
impl fmt::Debug for Parcel {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Parcel").finish()
}
}
impl fmt::Debug for BorrowedParcel<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("BorrowedParcel").finish()
}
}
#[cfg(test)]
mod tests {
use super::*;
use alloc::string::String;
#[test]
fn test_read_write() {
let mut parcel = Parcel::new();
let start = parcel.get_data_position();
assert_eq!(parcel.read::<bool>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<i8>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<u16>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<i32>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<u32>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<i64>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<u64>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<f32>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<f64>(), Err(StatusCode::NOT_ENOUGH_DATA));
assert_eq!(parcel.read::<Option<String>>(), Ok(None));
assert_eq!(parcel.read::<String>(), Err(StatusCode::UNEXPECTED_NULL));
assert_eq!(parcel.borrowed_ref().read_binder().err(), Some(StatusCode::BAD_TYPE));
parcel.write(&1i32).unwrap();
unsafe {
parcel.set_data_position(start).unwrap();
}
let i: i32 = parcel.read().unwrap();
assert_eq!(i, 1i32);
}
#[test]
#[allow(clippy::float_cmp)]
fn test_read_data() {
let mut parcel = Parcel::new();
let str_start = parcel.get_data_position();
parcel.write(&b"Hello, Binder!\0"[..]).unwrap();
unsafe {
assert!(parcel.set_data_position(str_start).is_ok());
}
assert_eq!(parcel.read::<i32>().unwrap(), 15);
let start = parcel.get_data_position();
assert!(parcel.read::<bool>().unwrap());
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<i8>().unwrap(), 72i8);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<u16>().unwrap(), 25928);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<i32>().unwrap(), 1819043144);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<u32>().unwrap(), 1819043144);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<i64>().unwrap(), 4764857262830019912);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<u64>().unwrap(), 4764857262830019912);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<f32>().unwrap(), 1143139100000000000000000000.0);
assert_eq!(parcel.read::<f32>().unwrap(), 40.043392);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<f64>().unwrap(), 34732488246.197815);
unsafe {
assert!(parcel.set_data_position(str_start).is_ok());
}
assert_eq!(parcel.read::<Vec<u8>>().unwrap(), b"Hello, Binder!\0");
}
#[test]
fn test_utf8_utf16_conversions() {
let mut parcel = Parcel::new();
let start = parcel.get_data_position();
assert!(parcel.write("Hello, Binder!").is_ok());
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<Option<String>>().unwrap().unwrap(), "Hello, Binder!",);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert!(parcel.write("Embedded null \0 inside a string").is_ok());
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(
parcel.read::<Option<String>>().unwrap().unwrap(),
"Embedded null \0 inside a string",
);
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert!(parcel.write(&["str1", "str2", "str3"][..]).is_ok());
assert!(
parcel
.write(&[String::from("str4"), String::from("str5"), String::from("str6"),][..])
.is_ok()
);
let s1 = "Hello, Binder!";
let s2 = "This is a utf8 string.";
let s3 = "Some more text here.";
assert!(parcel.write(&[s1, s2, s3][..]).is_ok());
unsafe {
assert!(parcel.set_data_position(start).is_ok());
}
assert_eq!(parcel.read::<Vec<String>>().unwrap(), ["str1", "str2", "str3"]);
assert_eq!(parcel.read::<Vec<String>>().unwrap(), ["str4", "str5", "str6"]);
assert_eq!(parcel.read::<Vec<String>>().unwrap(), [s1, s2, s3]);
}
#[test]
fn test_sized_write() {
let mut parcel = Parcel::new();
let start = parcel.get_data_position();
let arr = [1i32, 2i32, 3i32];
parcel
.sized_write(|subparcel| subparcel.write(&arr[..]))
.expect("Could not perform sized write");
let expected_len = 20i32;
assert_eq!(parcel.get_data_position(), start + expected_len);
unsafe {
parcel.set_data_position(start).unwrap();
}
assert_eq!(expected_len, parcel.read().unwrap(),);
assert_eq!(parcel.read::<Vec<i32>>().unwrap(), &arr,);
}
#[test]
fn test_append_from() {
let mut parcel1 = Parcel::new();
parcel1.write(&42i32).expect("Could not perform write");
let mut parcel2 = Parcel::new();
assert_eq!(Ok(()), parcel2.append_all_from(&parcel1));
assert_eq!(4, parcel2.get_data_size());
assert_eq!(Ok(()), parcel2.append_all_from(&parcel1));
assert_eq!(8, parcel2.get_data_size());
unsafe {
parcel2.set_data_position(0).unwrap();
}
assert_eq!(Ok(42), parcel2.read::<i32>());
assert_eq!(Ok(42), parcel2.read::<i32>());
let mut parcel2 = Parcel::new();
assert_eq!(Ok(()), parcel2.append_from(&parcel1, 0, 2));
assert_eq!(Ok(()), parcel2.append_from(&parcel1, 2, 2));
assert_eq!(4, parcel2.get_data_size());
unsafe {
parcel2.set_data_position(0).unwrap();
}
assert_eq!(Ok(42), parcel2.read::<i32>());
let mut parcel2 = Parcel::new();
assert_eq!(Ok(()), parcel2.append_from(&parcel1, 0, 2));
assert_eq!(2, parcel2.get_data_size());
unsafe {
parcel2.set_data_position(0).unwrap();
}
assert_eq!(Err(StatusCode::NOT_ENOUGH_DATA), parcel2.read::<i32>());
let mut parcel2 = Parcel::new();
assert_eq!(Err(StatusCode::BAD_VALUE), parcel2.append_from(&parcel1, 4, 2));
assert_eq!(Err(StatusCode::BAD_VALUE), parcel2.append_from(&parcel1, 2, 4));
assert_eq!(Err(StatusCode::BAD_VALUE), parcel2.append_from(&parcel1, -1, 4));
assert_eq!(Err(StatusCode::BAD_VALUE), parcel2.append_from(&parcel1, 2, -1));
}
#[test]
fn test_marshal() {
let mut parcel1 = Parcel::new();
parcel1.write(&42i32).unwrap();
parcel1.write(&52i32).unwrap();
parcel1.write(&62i32).unwrap();
parcel1.write(&72i32).unwrap();
let buffer = parcel1.marshal();
assert_eq!(16, buffer.len());
assert_eq!(Vec::from("\x2a\0\0\0\x34\0\0\0\x3e\0\0\0\x48\0\0\0"), buffer);
}
#[test]
fn test_unmarshal() {
let mut parcel1 = Parcel::new();
parcel1.write(&42i32).unwrap();
parcel1.write(&52i32).unwrap();
parcel1.write(&62i32).unwrap();
parcel1.write(&72i32).unwrap();
let buffer = parcel1.marshal();
let parcel2 = Parcel::unmarshal(&buffer);
assert_eq!(16, parcel2.get_data_size());
unsafe {
parcel2.set_data_position(0).unwrap();
}
drop(buffer);
assert_eq!(Ok(42i32), parcel2.read::<i32>());
assert_eq!(Ok(52i32), parcel2.read::<i32>());
assert_eq!(Ok(62i32), parcel2.read::<i32>());
assert_eq!(Ok(72i32), parcel2.read::<i32>());
}
}