mirror of
https://github.com/jkcoxson/idevice.git
synced 2026-03-02 14:36:16 +01:00
Refactor FFI bindings
This commit is contained in:
797
ffi/src/mobile_image_mounter.rs
Normal file
797
ffi/src/mobile_image_mounter.rs
Normal file
@@ -0,0 +1,797 @@
|
||||
// Jackson Coxson
|
||||
|
||||
use std::ffi::c_void;
|
||||
|
||||
use idevice::{
|
||||
IdeviceError, IdeviceService, mobile_image_mounter::ImageMounter, provider::IdeviceProvider,
|
||||
};
|
||||
use plist::Value;
|
||||
|
||||
use crate::{IdeviceErrorCode, IdeviceHandle, RUNTIME, provider::IdeviceProviderHandle, util};
|
||||
|
||||
pub struct ImageMounterHandle(pub ImageMounter);
|
||||
|
||||
/// Connects to the Image Mounter service using a provider
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`provider`] - An IdeviceProvider
|
||||
/// * [`client`] - On success, will be set to point to a newly allocated ImageMounter handle
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `provider` must be a valid pointer to a handle allocated by this library
|
||||
/// `client` must be a valid, non-null pointer to a location where the handle will be stored
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_connect(
|
||||
provider: *mut IdeviceProviderHandle,
|
||||
client: *mut *mut ImageMounterHandle,
|
||||
) -> IdeviceErrorCode {
|
||||
if provider.is_null() || client.is_null() {
|
||||
log::error!("Null pointer provided");
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let res: Result<ImageMounter, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let provider_ref: &dyn IdeviceProvider = unsafe { &*(*provider).0 };
|
||||
ImageMounter::connect(provider_ref).await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(r) => {
|
||||
let boxed = Box::new(ImageMounterHandle(r));
|
||||
unsafe { *client = Box::into_raw(boxed) };
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => {
|
||||
let _ = unsafe { Box::from_raw(provider) };
|
||||
e.into()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Creates a new ImageMounter client from an existing Idevice connection
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`socket`] - An IdeviceSocket handle
|
||||
/// * [`client`] - On success, will be set to point to a newly allocated ImageMounter handle
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `socket` must be a valid pointer to a handle allocated by this library
|
||||
/// `client` must be a valid, non-null pointer to a location where the handle will be stored
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_new(
|
||||
socket: *mut IdeviceHandle,
|
||||
client: *mut *mut ImageMounterHandle,
|
||||
) -> IdeviceErrorCode {
|
||||
if socket.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
let socket = unsafe { Box::from_raw(socket) }.0;
|
||||
let r = ImageMounter::new(socket);
|
||||
let boxed = Box::new(ImageMounterHandle(r));
|
||||
unsafe { *client = Box::into_raw(boxed) };
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
|
||||
/// Frees an ImageMounter handle
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`handle`] - The handle to free
|
||||
///
|
||||
/// # Safety
|
||||
/// `handle` must be a valid pointer to the handle that was allocated by this library,
|
||||
/// or NULL (in which case this function does nothing)
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_free(handle: *mut ImageMounterHandle) {
|
||||
if !handle.is_null() {
|
||||
log::debug!("Freeing image_mounter_client");
|
||||
let _ = unsafe { Box::from_raw(handle) };
|
||||
}
|
||||
}
|
||||
|
||||
/// Gets a list of mounted devices
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`devices`] - Will be set to point to a slice of device plists on success
|
||||
/// * [`devices_len`] - Will be set to the number of devices copied
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` must be a valid pointer to a handle allocated by this library
|
||||
/// `devices` must be a valid, non-null pointer to a location where the plist will be stored
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_copy_devices(
|
||||
client: *mut ImageMounterHandle,
|
||||
devices: *mut *mut c_void,
|
||||
devices_len: *mut libc::size_t,
|
||||
) -> IdeviceErrorCode {
|
||||
let res: Result<Vec<Value>, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.copy_devices().await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(devices_list) => {
|
||||
let devices_list = devices_list
|
||||
.into_iter()
|
||||
.map(|x| util::plist_to_libplist(&x))
|
||||
.collect::<Vec<*mut std::ffi::c_void>>();
|
||||
let len = devices_list.len();
|
||||
let boxed_slice = devices_list.into_boxed_slice();
|
||||
let ptr = Box::leak(boxed_slice).as_mut_ptr();
|
||||
|
||||
unsafe {
|
||||
*devices = ptr as *mut c_void;
|
||||
*devices_len = len;
|
||||
}
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Looks up an image and returns its signature
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`image_type`] - The type of image to look up
|
||||
/// * [`signature`] - Will be set to point to the signature data on success
|
||||
/// * [`signature_len`] - Will be set to the length of the signature data
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` must be a valid pointer to a handle allocated by this library
|
||||
/// `image_type` must be a valid null-terminated C string
|
||||
/// `signature` and `signature_len` must be valid pointers
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_lookup_image(
|
||||
client: *mut ImageMounterHandle,
|
||||
image_type: *const libc::c_char,
|
||||
signature: *mut *mut u8,
|
||||
signature_len: *mut libc::size_t,
|
||||
) -> IdeviceErrorCode {
|
||||
if image_type.is_null() || signature.is_null() || signature_len.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_type_cstr = unsafe { std::ffi::CStr::from_ptr(image_type) };
|
||||
let image_type = match image_type_cstr.to_str() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
};
|
||||
|
||||
let res: Result<Vec<u8>, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.lookup_image(image_type).await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(sig) => {
|
||||
let mut boxed = sig.into_boxed_slice();
|
||||
unsafe {
|
||||
*signature = boxed.as_mut_ptr();
|
||||
*signature_len = boxed.len();
|
||||
}
|
||||
std::mem::forget(boxed);
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Uploads an image to the device
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`image_type`] - The type of image being uploaded
|
||||
/// * [`image`] - Pointer to the image data
|
||||
/// * [`image_len`] - Length of the image data
|
||||
/// * [`signature`] - Pointer to the signature data
|
||||
/// * [`signature_len`] - Length of the signature data
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// All pointers must be valid and non-null
|
||||
/// `image_type` must be a valid null-terminated C string
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_upload_image(
|
||||
client: *mut ImageMounterHandle,
|
||||
image_type: *const libc::c_char,
|
||||
image: *const u8,
|
||||
image_len: libc::size_t,
|
||||
signature: *const u8,
|
||||
signature_len: libc::size_t,
|
||||
) -> IdeviceErrorCode {
|
||||
if image_type.is_null() || image.is_null() || signature.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_type_cstr = unsafe { std::ffi::CStr::from_ptr(image_type) };
|
||||
let image_type = match image_type_cstr.to_str() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
};
|
||||
|
||||
let image_slice = unsafe { std::slice::from_raw_parts(image, image_len) };
|
||||
let signature_slice = unsafe { std::slice::from_raw_parts(signature, signature_len) };
|
||||
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref
|
||||
.upload_image(image_type, image_slice, signature_slice.to_vec())
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Mounts an image on the device
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`image_type`] - The type of image being mounted
|
||||
/// * [`signature`] - Pointer to the signature data
|
||||
/// * [`signature_len`] - Length of the signature data
|
||||
/// * [`trust_cache`] - Pointer to trust cache data (optional)
|
||||
/// * [`trust_cache_len`] - Length of trust cache data (0 if none)
|
||||
/// * [`info_plist`] - Pointer to info plist (optional)
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// All pointers must be valid (except optional ones which can be null)
|
||||
/// `image_type` must be a valid null-terminated C string
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_mount_image(
|
||||
client: *mut ImageMounterHandle,
|
||||
image_type: *const libc::c_char,
|
||||
signature: *const u8,
|
||||
signature_len: libc::size_t,
|
||||
trust_cache: *const u8,
|
||||
trust_cache_len: libc::size_t,
|
||||
info_plist: *const c_void,
|
||||
) -> IdeviceErrorCode {
|
||||
if image_type.is_null() || signature.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_type_cstr = unsafe { std::ffi::CStr::from_ptr(image_type) };
|
||||
let image_type = match image_type_cstr.to_str() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
};
|
||||
|
||||
let signature_slice = unsafe { std::slice::from_raw_parts(signature, signature_len) };
|
||||
let trust_cache = if !trust_cache.is_null() && trust_cache_len > 0 {
|
||||
Some(unsafe { std::slice::from_raw_parts(trust_cache, trust_cache_len).to_vec() })
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
let info_plist = if !info_plist.is_null() {
|
||||
Some(
|
||||
unsafe { Box::from_raw(info_plist as *mut Value) }
|
||||
.as_ref()
|
||||
.clone(),
|
||||
)
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref
|
||||
.mount_image(
|
||||
image_type,
|
||||
signature_slice.to_vec(),
|
||||
trust_cache,
|
||||
info_plist,
|
||||
)
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Unmounts an image from the device
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`mount_path`] - The path where the image is mounted
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` must be a valid pointer to a handle allocated by this library
|
||||
/// `mount_path` must be a valid null-terminated C string
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_unmount_image(
|
||||
client: *mut ImageMounterHandle,
|
||||
mount_path: *const libc::c_char,
|
||||
) -> IdeviceErrorCode {
|
||||
if mount_path.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let mount_path_cstr = unsafe { std::ffi::CStr::from_ptr(mount_path) };
|
||||
let mount_path = match mount_path_cstr.to_str() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
};
|
||||
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.unmount_image(mount_path).await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Queries the developer mode status
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`status`] - Will be set to the developer mode status (1 = enabled, 0 = disabled)
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` must be a valid pointer to a handle allocated by this library
|
||||
/// `status` must be a valid pointer
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_query_developer_mode_status(
|
||||
client: *mut ImageMounterHandle,
|
||||
status: *mut libc::c_int,
|
||||
) -> IdeviceErrorCode {
|
||||
if status.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let res: Result<bool, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.query_developer_mode_status().await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(s) => {
|
||||
unsafe { *status = if s { 1 } else { 0 } };
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Mounts a developer image
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`image`] - Pointer to the image data
|
||||
/// * [`image_len`] - Length of the image data
|
||||
/// * [`signature`] - Pointer to the signature data
|
||||
/// * [`signature_len`] - Length of the signature data
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// All pointers must be valid and non-null
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_mount_developer(
|
||||
client: *mut ImageMounterHandle,
|
||||
image: *const u8,
|
||||
image_len: libc::size_t,
|
||||
signature: *const u8,
|
||||
signature_len: libc::size_t,
|
||||
) -> IdeviceErrorCode {
|
||||
if image.is_null() || signature.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_slice = unsafe { std::slice::from_raw_parts(image, image_len) };
|
||||
let signature_slice = unsafe { std::slice::from_raw_parts(signature, signature_len) };
|
||||
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref
|
||||
.mount_developer(image_slice, signature_slice.to_vec())
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Queries the personalization manifest from the device
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`image_type`] - The type of image to query
|
||||
/// * [`signature`] - Pointer to the signature data
|
||||
/// * [`signature_len`] - Length of the signature data
|
||||
/// * [`manifest`] - Will be set to point to the manifest data on success
|
||||
/// * [`manifest_len`] - Will be set to the length of the manifest data
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// All pointers must be valid and non-null
|
||||
/// `image_type` must be a valid null-terminated C string
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_query_personalization_manifest(
|
||||
client: *mut ImageMounterHandle,
|
||||
image_type: *const libc::c_char,
|
||||
signature: *const u8,
|
||||
signature_len: libc::size_t,
|
||||
manifest: *mut *mut u8,
|
||||
manifest_len: *mut libc::size_t,
|
||||
) -> IdeviceErrorCode {
|
||||
if image_type.is_null() || signature.is_null() || manifest.is_null() || manifest_len.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_type_cstr = unsafe { std::ffi::CStr::from_ptr(image_type) };
|
||||
let image_type = match image_type_cstr.to_str() {
|
||||
Ok(s) => s,
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
};
|
||||
|
||||
let signature_slice = unsafe { std::slice::from_raw_parts(signature, signature_len) };
|
||||
|
||||
let res: Result<Vec<u8>, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref
|
||||
.query_personalization_manifest(image_type, signature_slice.to_vec())
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(m) => {
|
||||
let mut boxed = m.into_boxed_slice();
|
||||
unsafe {
|
||||
*manifest = boxed.as_mut_ptr();
|
||||
*manifest_len = boxed.len();
|
||||
}
|
||||
std::mem::forget(boxed);
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Queries the nonce from the device
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`personalized_image_type`] - The type of image to query (optional)
|
||||
/// * [`nonce`] - Will be set to point to the nonce data on success
|
||||
/// * [`nonce_len`] - Will be set to the length of the nonce data
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client`, `nonce`, and `nonce_len` must be valid pointers
|
||||
/// `personalized_image_type` can be NULL
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_query_nonce(
|
||||
client: *mut ImageMounterHandle,
|
||||
personalized_image_type: *const libc::c_char,
|
||||
nonce: *mut *mut u8,
|
||||
nonce_len: *mut libc::size_t,
|
||||
) -> IdeviceErrorCode {
|
||||
if nonce.is_null() || nonce_len.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_type = if !personalized_image_type.is_null() {
|
||||
let image_type_cstr = unsafe { std::ffi::CStr::from_ptr(personalized_image_type) };
|
||||
match image_type_cstr.to_str() {
|
||||
Ok(s) => Some(s.to_string()),
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
}
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
let res: Result<Vec<u8>, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.query_nonce(image_type).await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(n) => {
|
||||
let mut boxed = n.into_boxed_slice();
|
||||
unsafe {
|
||||
*nonce = boxed.as_mut_ptr();
|
||||
*nonce_len = boxed.len();
|
||||
}
|
||||
std::mem::forget(boxed);
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Queries personalization identifiers from the device
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`image_type`] - The type of image to query (optional)
|
||||
/// * [`identifiers`] - Will be set to point to the identifiers plist on success
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` and `identifiers` must be valid pointers
|
||||
/// `image_type` can be NULL
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_query_personalization_identifiers(
|
||||
client: *mut ImageMounterHandle,
|
||||
image_type: *const libc::c_char,
|
||||
identifiers: *mut *mut c_void,
|
||||
) -> IdeviceErrorCode {
|
||||
if identifiers.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_type = if !image_type.is_null() {
|
||||
let image_type_cstr = unsafe { std::ffi::CStr::from_ptr(image_type) };
|
||||
match image_type_cstr.to_str() {
|
||||
Ok(s) => Some(s.to_string()),
|
||||
Err(_) => return IdeviceErrorCode::InvalidArg,
|
||||
}
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
let res: Result<plist::Dictionary, IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref
|
||||
.query_personalization_identifiers(image_type)
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(id) => {
|
||||
let plist = util::plist_to_libplist(&plist::Value::Dictionary(id));
|
||||
unsafe { *identifiers = plist };
|
||||
IdeviceErrorCode::IdeviceSuccess
|
||||
}
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Rolls the personalization nonce
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` must be a valid pointer to a handle allocated by this library
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_roll_personalization_nonce(
|
||||
client: *mut ImageMounterHandle,
|
||||
) -> IdeviceErrorCode {
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.roll_personalization_nonce().await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Rolls the cryptex nonce
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// `client` must be a valid pointer to a handle allocated by this library
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_roll_cryptex_nonce(
|
||||
client: *mut ImageMounterHandle,
|
||||
) -> IdeviceErrorCode {
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
client_ref.roll_cryptex_nonce().await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Mounts a personalized developer image
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`provider`] - A valid provider handle
|
||||
/// * [`image`] - Pointer to the image data
|
||||
/// * [`image_len`] - Length of the image data
|
||||
/// * [`trust_cache`] - Pointer to the trust cache data
|
||||
/// * [`trust_cache_len`] - Length of the trust cache data
|
||||
/// * [`build_manifest`] - Pointer to the build manifest data
|
||||
/// * [`build_manifest_len`] - Length of the build manifest data
|
||||
/// * [`info_plist`] - Pointer to info plist (optional)
|
||||
/// * [`unique_chip_id`] - The device's unique chip ID
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// All pointers must be valid (except optional ones which can be null)
|
||||
#[cfg(feature = "tss")]
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_mount_personalized(
|
||||
client: *mut ImageMounterHandle,
|
||||
provider: *mut IdeviceProviderHandle,
|
||||
image: *const u8,
|
||||
image_len: libc::size_t,
|
||||
trust_cache: *const u8,
|
||||
trust_cache_len: libc::size_t,
|
||||
build_manifest: *const u8,
|
||||
build_manifest_len: libc::size_t,
|
||||
info_plist: *const c_void,
|
||||
unique_chip_id: u64,
|
||||
) -> IdeviceErrorCode {
|
||||
if provider.is_null() || image.is_null() || trust_cache.is_null() || build_manifest.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_slice = unsafe { std::slice::from_raw_parts(image, image_len) };
|
||||
let trust_cache_slice = unsafe { std::slice::from_raw_parts(trust_cache, trust_cache_len) };
|
||||
let build_manifest_slice =
|
||||
unsafe { std::slice::from_raw_parts(build_manifest, build_manifest_len) };
|
||||
|
||||
let info_plist = if !info_plist.is_null() {
|
||||
Some(
|
||||
unsafe { Box::from_raw(info_plist as *mut Value) }
|
||||
.as_ref()
|
||||
.clone(),
|
||||
)
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
let provider_ref: &dyn IdeviceProvider = unsafe { &*(*provider).0 };
|
||||
client_ref
|
||||
.mount_personalized(
|
||||
provider_ref,
|
||||
image_slice.to_vec(),
|
||||
trust_cache_slice.to_vec(),
|
||||
build_manifest_slice,
|
||||
info_plist,
|
||||
unique_chip_id,
|
||||
)
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Mounts a personalized developer image with progress callback
|
||||
///
|
||||
/// # Arguments
|
||||
/// * [`client`] - A valid ImageMounter handle
|
||||
/// * [`provider`] - A valid provider handle
|
||||
/// * [`image`] - Pointer to the image data
|
||||
/// * [`image_len`] - Length of the image data
|
||||
/// * [`trust_cache`] - Pointer to the trust cache data
|
||||
/// * [`trust_cache_len`] - Length of the trust cache data
|
||||
/// * [`build_manifest`] - Pointer to the build manifest data
|
||||
/// * [`build_manifest_len`] - Length of the build manifest data
|
||||
/// * [`info_plist`] - Pointer to info plist (optional)
|
||||
/// * [`unique_chip_id`] - The device's unique chip ID
|
||||
/// * [`callback`] - Progress callback function
|
||||
/// * [`context`] - User context to pass to callback
|
||||
///
|
||||
/// # Returns
|
||||
/// An error code indicating success or failure
|
||||
///
|
||||
/// # Safety
|
||||
/// All pointers must be valid (except optional ones which can be null)
|
||||
#[cfg(feature = "tss")]
|
||||
#[unsafe(no_mangle)]
|
||||
pub unsafe extern "C" fn image_mounter_mount_personalized_with_callback(
|
||||
client: *mut ImageMounterHandle,
|
||||
provider: *mut IdeviceProviderHandle,
|
||||
image: *const u8,
|
||||
image_len: libc::size_t,
|
||||
trust_cache: *const u8,
|
||||
trust_cache_len: libc::size_t,
|
||||
build_manifest: *const u8,
|
||||
build_manifest_len: libc::size_t,
|
||||
info_plist: *const c_void,
|
||||
unique_chip_id: u64,
|
||||
callback: extern "C" fn(progress: libc::size_t, total: libc::size_t, context: *mut c_void),
|
||||
context: *mut c_void,
|
||||
) -> IdeviceErrorCode {
|
||||
if provider.is_null() || image.is_null() || trust_cache.is_null() || build_manifest.is_null() {
|
||||
return IdeviceErrorCode::InvalidArg;
|
||||
}
|
||||
|
||||
let image_slice = unsafe { std::slice::from_raw_parts(image, image_len) };
|
||||
let trust_cache_slice = unsafe { std::slice::from_raw_parts(trust_cache, trust_cache_len) };
|
||||
let build_manifest_slice =
|
||||
unsafe { std::slice::from_raw_parts(build_manifest, build_manifest_len) };
|
||||
|
||||
let info_plist = if !info_plist.is_null() {
|
||||
Some(
|
||||
unsafe { Box::from_raw(info_plist as *mut Value) }
|
||||
.as_ref()
|
||||
.clone(),
|
||||
)
|
||||
} else {
|
||||
None
|
||||
};
|
||||
|
||||
let res: Result<(), IdeviceError> = RUNTIME.block_on(async move {
|
||||
let client_ref = unsafe { &mut (*client).0 };
|
||||
let provider_ref: &dyn IdeviceProvider = unsafe { &*(*provider).0 };
|
||||
|
||||
let callback_wrapper = |((progress, total), context)| async move {
|
||||
callback(progress, total, context);
|
||||
};
|
||||
|
||||
client_ref
|
||||
.mount_personalized_with_callback(
|
||||
provider_ref,
|
||||
image_slice.to_vec(),
|
||||
trust_cache_slice.to_vec(),
|
||||
build_manifest_slice,
|
||||
info_plist,
|
||||
unique_chip_id,
|
||||
callback_wrapper,
|
||||
context,
|
||||
)
|
||||
.await
|
||||
});
|
||||
|
||||
match res {
|
||||
Ok(_) => IdeviceErrorCode::IdeviceSuccess,
|
||||
Err(e) => e.into(),
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user