StephenDev0/idevice
1
0
Fork 0
mirror of https://github.com/jkcoxson/idevice.git synced 2026-03-02 14:36:16 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: add screenshotr for iOS versions below iOS17 (#27)

Browse Source 
* feat: add screenshot for iOS versions below iOS17

* "cargo fmt"

* Style cleanup

---------

Co-authored-by: Jackson Coxson <jkcoxson@gmail.com>
This commit is contained in:
ValorBao
2025-09-17 03:44:13 +08:00
committed by GitHub
parent 378c84e6ef
commit 482c9805c4
4 changed files with 162 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
tools/src/screenshot.rs
View File

@@ -2,6 +2,8 @@ use clap::{Arg, Command};
use idevice::{IdeviceService, RsdService, core_device_proxy::CoreDeviceProxy, rsd::RsdHandshake};
use std::fs;
use idevice::screenshotr::ScreenshotService;
mod common;
#[tokio::main]
@@ -62,31 +64,43 @@ async fn main() {
return;
}
};
let proxy = CoreDeviceProxy::connect(&*provider)
let mut res: Vec<u8> = Vec::new();
if let Ok(proxy) = CoreDeviceProxy::connect(&*provider).await {
let rsd_port = proxy.handshake.server_rsd_port;
let adapter = proxy.create_software_tunnel().expect("no software tunnel");
let mut adapter = adapter.to_async_handle();
let stream = adapter.connect(rsd_port).await.expect("no RSD connect");
// Make the connection to RemoteXPC
let mut handshake = RsdHandshake::new(stream).await.unwrap();
let mut ts_client = idevice::dvt::remote_server::RemoteServerClient::connect_rsd(
&mut adapter,
&mut handshake,
)
.await
.expect("no core proxy");
let rsd_port = proxy.handshake.server_rsd_port;
.expect("Failed to connect");
ts_client.read_message(0).await.expect("no read??");
let adapter = proxy.create_software_tunnel().expect("no software tunnel");
let mut adapter = adapter.to_async_handle();
let stream = adapter.connect(rsd_port).await.expect("no RSD connect");
// Make the connection to RemoteXPC
let mut handshake = RsdHandshake::new(stream).await.unwrap();
let mut ts_client =
idevice::dvt::remote_server::RemoteServerClient::connect_rsd(&mut adapter, &mut handshake)
let mut ts_client = idevice::dvt::screenshot::ScreenshotClient::new(&mut ts_client)
.await
.expect("Failed to connect");
ts_client.read_message(0).await.expect("no read??");
let mut ts_client = idevice::dvt::screenshot::ScreenshotClient::new(&mut ts_client)
.await
.expect("Unable to get channel for take screenshot");
let res = ts_client
.take_screenshot()
.await
.expect("Failed to take screenshot");
.expect("Unable to get channel for take screenshot");
res = ts_client
.take_screenshot()
.await
.expect("Failed to take screenshot");
} else {
let mut screenshot_client = match ScreenshotService::connect(&*provider).await {
Ok(client) => client,
Err(e) => {
eprintln!(
"Unable to connect to screenshotr service: {e} Ensure Developer Disk Image is mounted."
);
return;
}
};
res = screenshot_client.take_screenshot().await.unwrap();
}
match fs::write(output_path, &res) {
Ok(_) => println!("Screenshot saved to: {}", output_path),
Err(e) => eprintln!("Failed to write screenshot to file: {}", e),