2 Commits

Author SHA1 Message Date
LIAUD Corentin
aa3943f21f wip: add wasm target support 2025-05-24 14:57:17 +02:00
LIAUD Corentin
4ebbd99607 chore: bump criterion + pyo3 2025-05-24 13:47:46 +02:00
9 changed files with 133 additions and 118 deletions

View File

@@ -10,32 +10,39 @@ repository.workspace = true
version.workspace = true version.workspace = true
[dependencies] [dependencies]
base64 = { version = "0.22.1" } getrandom = { version = "*", features = ["js"] }
ring = { version = "*", features = ["wasm32_unknown_unknown_js"] }
bincode = { version = "1.3.3" } bincode = { version = "1.3.3" }
byteorder = { version = "1.5.0" } base64 = { version = "0.22.1", optional = true }
byteorder = { version = "1.5.0", optional = true }
chrono = { version = "0.4.40", default-features = false, features = ["std"] } chrono = { version = "0.4.40", default-features = false, features = ["std"] }
homedir = { version = "0.3.4" } homedir = { version = "0.3.4", optional = true }
image = { version = "0.25.5", default-features = false } image = { version = "0.25.5", default-features = false }
log = { version = "0.4.26" } log = { version = "0.4.26" }
mdns-sd = { version = "0.13.9", default-features = false, features = [ mdns-sd = { version = "0.13.9", default-features = false, features = [
"logging", "logging",
] } ], optional = true }
num-bigint = { version = "0.8.4", package = "num-bigint-dig" } num-bigint = { version = "0.8.4", package = "num-bigint-dig", optional = true }
num-traits = { version = "0.2.19" } num-traits = { version = "0.2.19", optional = true }
quick-protobuf = { version = "0.8.1" } quick-protobuf = { version = "0.8.1", optional = true }
rand = { version = "0.9.0" } rand = { version = "0.8.5", optional = true }
rcgen = { version = "0.13.1", default-features = false, features = [ rcgen = { version = "0.13.1", default-features = false, features = [
"aws_lc_rs", "aws_lc_rs",
"pem", "pem",
] } ], optional = true }
regex = { version = "1.11.1", features = ["perf", "std", "unicode"] } regex = { version = "1.11.1", features = [
rsa = { version = "0.9.7" } "perf",
rusb = { version = "0.9.4", features = ["vendored"] } "std",
rustls = { version = "0.23.27" } "unicode",
rustls-pki-types = { version = "1.11.0" } ], optional = true }
rsa = { version = "0.9.7", optional = true }
rusb = { version = "0.9.4", features = ["vendored"], optional = true }
rustls = { version = "0.23.27", optional = true }
rustls-pki-types = { version = "1.11.0", optional = true }
serde = { version = "1.0.216", features = ["derive"] } serde = { version = "1.0.216", features = ["derive"] }
serde_repr = { version = "0.1.19" } serde_repr = { version = "0.1.19", optional = true }
sha1 = { version = "0.10.6", features = ["oid"] } sha1 = { version = "0.10.6", features = ["oid"], optional = true }
thiserror = { version = "2.0.7" } thiserror = { version = "2.0.7" }
[dev-dependencies] [dev-dependencies]

View File

@@ -11,18 +11,18 @@ use super::{ADBTransportMessage, MessageCommand, models::MessageSubcommand};
#[derive(Debug)] #[derive(Debug)]
pub struct ADBMessageDevice<T: ADBMessageTransport> { pub struct ADBMessageDevice<T: ADBMessageTransport> {
transport: T, transport: T,
} local_id: Option<u32>,
remote_id: Option<u32>,
#[derive(Debug, Clone, Copy)]
pub struct ADBSession {
pub local_id: u32,
pub remote_id: u32,
} }
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
/// Instantiate a new [`ADBMessageTransport`] /// Instantiate a new [`ADBMessageTransport`]
pub fn new(transport: T) -> Self { pub fn new(transport: T) -> Self {
Self { transport } Self {
transport,
local_id: None,
remote_id: None,
}
} }
pub(crate) fn get_transport(&mut self) -> &T { pub(crate) fn get_transport(&mut self) -> &T {
@@ -34,15 +34,12 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
} }
/// Receive a message and acknowledge it by replying with an `OKAY` command /// Receive a message and acknowledge it by replying with an `OKAY` command
pub(crate) fn recv_and_reply_okay( pub(crate) fn recv_and_reply_okay(&mut self) -> Result<ADBTransportMessage> {
&mut self,
session: ADBSession,
) -> Result<ADBTransportMessage> {
let message = self.transport.read_message()?; let message = self.transport.read_message()?;
self.transport.write_message(ADBTransportMessage::new( self.transport.write_message(ADBTransportMessage::new(
MessageCommand::Okay, MessageCommand::Okay,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
&[], &[],
))?; ))?;
Ok(message) Ok(message)
@@ -63,12 +60,11 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn recv_file<W: std::io::Write>( pub(crate) fn recv_file<W: std::io::Write>(
&mut self, &mut self,
session: ADBSession,
mut output: W, mut output: W,
) -> std::result::Result<(), RustADBError> { ) -> std::result::Result<(), RustADBError> {
let mut len: Option<u64> = None; let mut len: Option<u64> = None;
loop { loop {
let payload = self.recv_and_reply_okay(session)?.into_payload(); let payload = self.recv_and_reply_okay()?.into_payload();
let mut rdr = Cursor::new(&payload); let mut rdr = Cursor::new(&payload);
while rdr.position() != payload.len() as u64 { while rdr.position() != payload.len() as u64 {
match len.take() { match len.take() {
@@ -101,7 +97,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn push_file<R: std::io::Read>( pub(crate) fn push_file<R: std::io::Read>(
&mut self, &mut self,
session: ADBSession, local_id: u32,
remote_id: u32,
mut reader: R, mut reader: R,
) -> std::result::Result<(), RustADBError> { ) -> std::result::Result<(), RustADBError> {
let mut buffer = [0; BUFFER_SIZE]; let mut buffer = [0; BUFFER_SIZE];
@@ -114,8 +111,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, local_id,
session.remote_id, remote_id,
&serialized_message, &serialized_message,
); );
@@ -134,8 +131,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, local_id,
session.remote_id, remote_id,
&serialized_message, &serialized_message,
); );
@@ -162,8 +159,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, local_id,
session.remote_id, remote_id,
&serialized_message, &serialized_message,
); );
@@ -176,27 +173,24 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
} }
} }
pub(crate) fn begin_synchronization(&mut self) -> Result<ADBSession> { pub(crate) fn begin_synchronization(&mut self) -> Result<()> {
self.open_session(b"sync:\0") self.open_session(b"sync:\0")?;
Ok(())
} }
pub(crate) fn stat_with_explicit_ids( pub(crate) fn stat_with_explicit_ids(&mut self, remote_path: &str) -> Result<AdbStatResponse> {
&mut self,
session: ADBSession,
remote_path: &str,
) -> Result<AdbStatResponse> {
let stat_buffer = MessageSubcommand::Stat.with_arg(remote_path.len() as u32); let stat_buffer = MessageSubcommand::Stat.with_arg(remote_path.len() as u32);
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
&bincode::serialize(&stat_buffer).map_err(|_e| RustADBError::ConversionError)?, &bincode::serialize(&stat_buffer).map_err(|_e| RustADBError::ConversionError)?,
); );
self.send_and_expect_okay(message)?; self.send_and_expect_okay(message)?;
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
remote_path.as_bytes(), remote_path.as_bytes(),
))?; ))?;
let response = self.transport.read_message()?; let response = self.transport.read_message()?;
@@ -206,25 +200,24 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
.map_err(|_e| RustADBError::ConversionError) .map_err(|_e| RustADBError::ConversionError)
} }
pub(crate) fn end_transaction(&mut self, session: ADBSession) -> Result<()> { pub(crate) fn end_transaction(&mut self) -> Result<()> {
let quit_buffer = MessageSubcommand::Quit.with_arg(0u32); let quit_buffer = MessageSubcommand::Quit.with_arg(0u32);
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
&bincode::serialize(&quit_buffer).map_err(|_e| RustADBError::ConversionError)?, &bincode::serialize(&quit_buffer).map_err(|_e| RustADBError::ConversionError)?,
))?; ))?;
let _discard_close = self.transport.read_message()?; let _discard_close = self.transport.read_message()?;
Ok(()) Ok(())
} }
pub(crate) fn open_session(&mut self, data: &[u8]) -> Result<ADBSession> { pub(crate) fn open_session(&mut self, data: &[u8]) -> Result<ADBTransportMessage> {
let mut rng = rand::rng(); let mut rng = rand::rng();
let local_id: u32 = rng.random();
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Open, MessageCommand::Open,
local_id, // Our 'local-id' rng.random(), // Our 'local-id'
0, 0,
data, data,
); );
@@ -232,25 +225,21 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let response = self.get_transport_mut().read_message()?; let response = self.get_transport_mut().read_message()?;
if response.header().command() != MessageCommand::Okay { self.local_id = Some(response.header().arg1());
return Err(RustADBError::ADBRequestFailed(format!( self.remote_id = Some(response.header().arg0());
"Open session failed: got {} in respone instead of OKAY",
response.header().command()
)));
}
if response.header().arg1() != local_id { Ok(response)
return Err(RustADBError::ADBRequestFailed(format!( }
"Open session failed: respones used {} for our local_id instead of {local_id}",
response.header().arg1()
)));
}
let session = ADBSession { pub(crate) fn get_local_id(&self) -> Result<u32> {
local_id, self.local_id.ok_or(RustADBError::ADBRequestFailed(
remote_id: response.header().arg0(), "connection not opened, no local_id".into(),
}; ))
}
Ok(session) pub(crate) fn get_remote_id(&self) -> Result<u32> {
self.remote_id.ok_or(RustADBError::ADBRequestFailed(
"connection not opened, no remote_id".into(),
))
} }
} }

View File

@@ -11,9 +11,9 @@ use crate::{
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn framebuffer_inner(&mut self) -> Result<ImageBuffer<Rgba<u8>, Vec<u8>>> { pub(crate) fn framebuffer_inner(&mut self) -> Result<ImageBuffer<Rgba<u8>, Vec<u8>>> {
let session = self.open_session(b"framebuffer:\0")?; self.open_session(b"framebuffer:\0")?;
let response = self.recv_and_reply_okay(session)?; let response = self.recv_and_reply_okay()?;
let mut payload_cursor = Cursor::new(response.payload()); let mut payload_cursor = Cursor::new(response.payload());
@@ -36,7 +36,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
break; break;
} }
let response = self.recv_and_reply_okay(session)?; let response = self.recv_and_reply_okay()?;
framebuffer_data.extend_from_slice(&response.into_payload()); framebuffer_data.extend_from_slice(&response.into_payload());
@@ -69,7 +69,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
break; break;
} }
let response = self.recv_and_reply_okay(session)?; let response = self.recv_and_reply_okay()?;
framebuffer_data.extend_from_slice(&response.into_payload()); framebuffer_data.extend_from_slice(&response.into_payload());

View File

@@ -14,12 +14,11 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let file_size = apk_file.metadata()?.len(); let file_size = apk_file.metadata()?.len();
let session = self.open_session(format!("exec:cmd package 'install' -S {}\0", file_size).as_bytes())?;
self.open_session(format!("exec:cmd package 'install' -S {}\0", file_size).as_bytes())?;
let transport = self.get_transport().clone(); let transport = self.get_transport().clone();
let mut writer = MessageWriter::new(transport, session.local_id, session.remote_id); let mut writer = MessageWriter::new(transport, self.get_local_id()?, self.get_remote_id()?);
std::io::copy(&mut apk_file, &mut writer)?; std::io::copy(&mut apk_file, &mut writer)?;

View File

@@ -10,10 +10,10 @@ use crate::{
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn pull<A: AsRef<str>, W: Write>(&mut self, source: A, output: W) -> Result<()> { pub(crate) fn pull<A: AsRef<str>, W: Write>(&mut self, source: A, output: W) -> Result<()> {
let session = self.begin_synchronization()?; self.begin_synchronization()?;
let source = source.as_ref(); let source = source.as_ref();
let adb_stat_response = self.stat_with_explicit_ids(session, source)?; let adb_stat_response = self.stat_with_explicit_ids(source)?;
if adb_stat_response.file_perm == 0 { if adb_stat_response.file_perm == 0 {
return Err(RustADBError::UnknownResponseType( return Err(RustADBError::UnknownResponseType(
@@ -21,13 +21,11 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
)); ));
} }
let local_id = self.get_local_id()?;
let remote_id = self.get_remote_id()?;
self.get_transport_mut().write_message_with_timeout( self.get_transport_mut().write_message_with_timeout(
ADBTransportMessage::new( ADBTransportMessage::new(MessageCommand::Okay, local_id, remote_id, &[]),
MessageCommand::Okay,
session.local_id,
session.remote_id,
&[],
),
std::time::Duration::from_secs(4), std::time::Duration::from_secs(4),
)?; )?;
@@ -36,19 +34,19 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
bincode::serialize(&recv_buffer).map_err(|_e| RustADBError::ConversionError)?; bincode::serialize(&recv_buffer).map_err(|_e| RustADBError::ConversionError)?;
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
&recv_buffer, &recv_buffer,
))?; ))?;
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
source.as_bytes(), source.as_bytes(),
))?; ))?;
self.recv_file(session, output)?; self.recv_file(output)?;
self.end_transaction(session)?; self.end_transaction()?;
Ok(()) Ok(())
} }
} }

View File

@@ -10,7 +10,7 @@ use crate::{
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn push<R: Read, A: AsRef<str>>(&mut self, stream: R, path: A) -> Result<()> { pub(crate) fn push<R: Read, A: AsRef<str>>(&mut self, stream: R, path: A) -> Result<()> {
let session = self.begin_synchronization()?; self.begin_synchronization()?;
let path_header = format!("{},0777", path.as_ref()); let path_header = format!("{},0777", path.as_ref());
@@ -21,13 +21,14 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
session.local_id, self.get_local_id()?,
session.remote_id, self.get_remote_id()?,
&send_buffer, &send_buffer,
))?; ))?;
self.push_file(session, stream)?; self.push_file(self.get_local_id()?, self.get_remote_id()?, stream)?;
self.end_transaction(session)?;
self.end_transaction()?;
Ok(()) Ok(())
} }

View File

@@ -10,7 +10,14 @@ use crate::{
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
/// Runs 'command' in a shell on the device, and write its output and error streams into output. /// Runs 'command' in a shell on the device, and write its output and error streams into output.
pub(crate) fn shell_command(&mut self, command: &[&str], output: &mut dyn Write) -> Result<()> { pub(crate) fn shell_command(&mut self, command: &[&str], output: &mut dyn Write) -> Result<()> {
let session = self.open_session(format!("shell:{}\0", command.join(" "),).as_bytes())?; let response = self.open_session(format!("shell:{}\0", command.join(" "),).as_bytes())?;
if response.header().command() != MessageCommand::Okay {
return Err(RustADBError::ADBRequestFailed(format!(
"wrong command {}",
response.header().command()
)));
}
loop { loop {
let response = self.get_transport_mut().read_message()?; let response = self.get_transport_mut().read_message()?;
@@ -31,22 +38,21 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
mut reader: &mut dyn Read, mut reader: &mut dyn Read,
mut writer: Box<(dyn Write + Send)>, mut writer: Box<(dyn Write + Send)>,
) -> Result<()> { ) -> Result<()> {
let session = self.open_session(b"shell:\0")?; self.open_session(b"shell:\0")?;
let mut transport = self.get_transport().clone(); let mut transport = self.get_transport().clone();
let local_id = self.get_local_id()?;
let remote_id = self.get_remote_id()?;
// Reading thread, reads response from adbd // Reading thread, reads response from adbd
std::thread::spawn(move || -> Result<()> { std::thread::spawn(move || -> Result<()> {
loop { loop {
let message = transport.read_message()?; let message = transport.read_message()?;
// Acknowledge for more data // Acknowledge for more data
let response = ADBTransportMessage::new( let response =
MessageCommand::Okay, ADBTransportMessage::new(MessageCommand::Okay, local_id, remote_id, &[]);
session.local_id,
session.remote_id,
&[],
);
transport.write_message(response)?; transport.write_message(response)?;
match message.header().command() { match message.header().command() {
@@ -61,8 +67,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
}); });
let transport = self.get_transport().clone(); let transport = self.get_transport().clone();
let mut shell_writer = let mut shell_writer = ShellMessageWriter::new(transport, local_id, remote_id);
ShellMessageWriter::new(transport, session.local_id, session.remote_id);
// Read from given reader (that could be stdin e.g), and write content to device adbd // Read from given reader (that could be stdin e.g), and write content to device adbd
if let Err(e) = std::io::copy(&mut reader, &mut shell_writer) { if let Err(e) = std::io::copy(&mut reader, &mut shell_writer) {

View File

@@ -4,9 +4,9 @@ use crate::{
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn stat(&mut self, remote_path: &str) -> Result<AdbStatResponse> { pub(crate) fn stat(&mut self, remote_path: &str) -> Result<AdbStatResponse> {
let session = self.begin_synchronization()?; self.begin_synchronization()?;
let adb_stat_response = self.stat_with_explicit_ids(session, remote_path)?; let adb_stat_response = self.stat_with_explicit_ids(remote_path)?;
self.end_transaction(session)?; self.end_transaction()?;
Ok(adb_stat_response) Ok(adb_stat_response)
} }
} }

View File

@@ -7,21 +7,37 @@
mod adb_device_ext; mod adb_device_ext;
mod constants; mod constants;
mod device; mod device;
mod emulator_device;
mod error; mod error;
mod mdns;
mod models; mod models;
mod server;
mod server_device;
mod transports; mod transports;
#[cfg(not(target_arch = "wasm32"))]
mod emulator_device;
#[cfg(not(target_arch = "wasm32"))]
mod mdns;
#[cfg(not(target_arch = "wasm32"))]
mod server;
#[cfg(not(target_arch = "wasm32"))]
mod server_device;
#[cfg(not(target_arch = "wasm32"))]
mod utils; mod utils;
pub use adb_device_ext::ADBDeviceExt; pub use adb_device_ext::ADBDeviceExt;
pub use device::{ADBTcpDevice, ADBUSBDevice}; pub use device::ADBUSBDevice;
pub use emulator_device::ADBEmulatorDevice;
pub use error::{Result, RustADBError}; pub use error::{Result, RustADBError};
pub use mdns::*;
pub use models::{AdbStatResponse, RebootType}; pub use models::{AdbStatResponse, RebootType};
pub use transports::{ADBMessageTransport, ADBTransport};
#[cfg(not(target_arch = "wasm32"))]
pub use device::ADBTcpDevice;
#[cfg(not(target_arch = "wasm32"))]
pub use emulator_device::ADBEmulatorDevice;
#[cfg(not(target_arch = "wasm32"))]
pub use mdns::*;
#[cfg(not(target_arch = "wasm32"))]
pub use server::*; pub use server::*;
#[cfg(not(target_arch = "wasm32"))]
pub use server_device::ADBServerDevice; pub use server_device::ADBServerDevice;
#[cfg(not(target_arch = "wasm32"))]
pub use transports::*; pub use transports::*;