1 Commits

Author SHA1 Message Date
Sashanoraa
849f7807a9 Track individual sessions (local and remote id) for each operation
This allows multiple operations to be performed simultaneously
2025-06-26 03:29:22 -04:00
35 changed files with 170 additions and 192 deletions

View File

@@ -9,8 +9,7 @@ homepage = "https://github.com/cocool97/adb_client"
keywords = ["adb", "android", "tcp", "usb"] keywords = ["adb", "android", "tcp", "usb"]
license = "MIT" license = "MIT"
repository = "https://github.com/cocool97/adb_client" repository = "https://github.com/cocool97/adb_client"
version = "2.1.16" version = "2.1.13"
rust-version = "1.85.1"
# To build locally when working on a new release # To build locally when working on a new release
[patch.crates-io] [patch.crates-io]

View File

@@ -8,9 +8,6 @@
<a href="https://crates.io/crates/adb_client"> <a href="https://crates.io/crates/adb_client">
<img alt="crates.io" src="https://img.shields.io/crates/v/adb_client.svg"/> <img alt="crates.io" src="https://img.shields.io/crates/v/adb_client.svg"/>
</a> </a>
<a href="https://crates.io/crates/adb_client">
<img alt="msrv" src="https://img.shields.io/crates/msrv/adb_client"/>
</a>
<a href="https://github.com/cocool97/adb_client/actions"> <a href="https://github.com/cocool97/adb_client/actions">
<img alt="ci status" src="https://github.com/cocool97/adb_client/actions/workflows/rust-build.yml/badge.svg"/> <img alt="ci status" src="https://github.com/cocool97/adb_client/actions/workflows/rust-build.yml/badge.svg"/>
</a> </a>

View File

@@ -7,7 +7,6 @@ license.workspace = true
name = "adb_cli" name = "adb_cli"
readme = "README.md" readme = "README.md"
repository.workspace = true repository.workspace = true
rust-version.workspace = true
version.workspace = true version.workspace = true
[dependencies] [dependencies]

View File

@@ -2,7 +2,6 @@
[![MIT licensed](https://img.shields.io/crates/l/adb_cli.svg)](./LICENSE-MIT) [![MIT licensed](https://img.shields.io/crates/l/adb_cli.svg)](./LICENSE-MIT)
![Crates.io Total Downloads](https://img.shields.io/crates/d/adb_cli) ![Crates.io Total Downloads](https://img.shields.io/crates/d/adb_cli)
![MSRV](https://img.shields.io/crates/msrv/adb_cli)
Rust binary providing an improved version of `adb` CLI. Rust binary providing an improved version of `adb` CLI.
@@ -11,7 +10,7 @@ Rust binary providing an improved version of `adb` CLI.
This crate provides a lightweight binary based on the `adb_client` crate. You can install it by running the following command : This crate provides a lightweight binary based on the `adb_client` crate. You can install it by running the following command :
```shell ```shell
cargo install adb_cli cargo install adb_cli
``` ```
Usage is quite simple, and tends to look like `adb`: Usage is quite simple, and tends to look like `adb`:
@@ -65,4 +64,4 @@ Options:
-p, --product-id <PID> Hexadecimal product id of this USB device -p, --product-id <PID> Hexadecimal product id of this USB device
-k, --private-key <PATH_TO_PRIVATE_KEY> Path to a custom private key to use for authentication -k, --private-key <PATH_TO_PRIVATE_KEY> Path to a custom private key to use for authentication
-h, --help Print help -h, --help Print help
``` ```

View File

@@ -8,7 +8,7 @@ pub fn handle_host_commands(server_command: ServerCommand<HostCommand>) -> Resul
match server_command.command { match server_command.command {
HostCommand::Version => { HostCommand::Version => {
let version = adb_server.version()?; let version = adb_server.version()?;
log::info!("Android Debug Bridge version {version}"); log::info!("Android Debug Bridge version {}", version);
log::info!("Package version {}-rust", std::env!("CARGO_PKG_VERSION")); log::info!("Package version {}-rust", std::env!("CARGO_PKG_VERSION"));
} }
HostCommand::Kill => { HostCommand::Kill => {
@@ -18,18 +18,18 @@ pub fn handle_host_commands(server_command: ServerCommand<HostCommand>) -> Resul
if long { if long {
log::info!("List of devices attached (extended)"); log::info!("List of devices attached (extended)");
for device in adb_server.devices_long()? { for device in adb_server.devices_long()? {
log::info!("{device}"); log::info!("{}", device);
} }
} else { } else {
log::info!("List of devices attached"); log::info!("List of devices attached");
for device in adb_server.devices()? { for device in adb_server.devices()? {
log::info!("{device}"); log::info!("{}", device);
} }
} }
} }
HostCommand::TrackDevices => { HostCommand::TrackDevices => {
let callback = |device: DeviceShort| { let callback = |device: DeviceShort| {
log::info!("{device}"); log::info!("{}", device);
Ok(()) Ok(())
}; };
log::info!("Live list of devices attached"); log::info!("Live list of devices attached");
@@ -65,7 +65,7 @@ pub fn handle_host_commands(server_command: ServerCommand<HostCommand>) -> Resul
MdnsCommand::Services => { MdnsCommand::Services => {
log::info!("List of discovered mdns services"); log::info!("List of discovered mdns services");
for service in adb_server.mdns_services()? { for service in adb_server.mdns_services()? {
log::info!("{service}"); log::info!("{}", service);
} }
} }
}, },

View File

@@ -132,10 +132,10 @@ fn main() -> Result<()> {
} }
DeviceCommands::Stat { path } => { DeviceCommands::Stat { path } => {
let stat_response = device.stat(&path)?; let stat_response = device.stat(&path)?;
println!("{stat_response}"); println!("{}", stat_response);
} }
DeviceCommands::Reboot { reboot_type } => { DeviceCommands::Reboot { reboot_type } => {
log::info!("Reboots device in mode {reboot_type:?}"); log::info!("Reboots device in mode {:?}", reboot_type);
device.reboot(reboot_type.into())? device.reboot(reboot_type.into())?
} }
DeviceCommands::Push { filename, path } => { DeviceCommands::Push { filename, path } => {
@@ -152,7 +152,7 @@ fn main() -> Result<()> {
device.install(&path)?; device.install(&path)?;
} }
DeviceCommands::Uninstall { package } => { DeviceCommands::Uninstall { package } => {
log::info!("Uninstalling the package {package}..."); log::info!("Uninstalling the package {}...", package);
device.uninstall(&package)?; device.uninstall(&package)?;
} }
DeviceCommands::Framebuffer { path } => { DeviceCommands::Framebuffer { path } => {

View File

@@ -7,21 +7,19 @@ license.workspace = true
name = "adb_client" name = "adb_client"
readme = "README.md" readme = "README.md"
repository.workspace = true repository.workspace = true
rust-version.workspace = true
version.workspace = true version.workspace = true
[features]
default = ["mdns"]
mdns = ["dep:mdns-sd"]
[dependencies] [dependencies]
base64 = { version = "0.22.1" } base64 = { version = "0.22.1" }
bincode = { version = "1.3.3" } bincode = { version = "1.3.3" }
byteorder = { version = "1.5.0" } byteorder = { version = "1.5.0" }
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" }
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 = [
"logging",
] }
num-bigint = { version = "0.8.4", package = "num-bigint-dig" } num-bigint = { version = "0.8.4", package = "num-bigint-dig" }
num-traits = { version = "0.2.19" } num-traits = { version = "0.2.19" }
quick-protobuf = { version = "0.8.1" } quick-protobuf = { version = "0.8.1" }
@@ -40,11 +38,6 @@ serde_repr = { version = "0.1.19" }
sha1 = { version = "0.10.6", features = ["oid"] } sha1 = { version = "0.10.6", features = ["oid"] }
thiserror = { version = "2.0.7" } thiserror = { version = "2.0.7" }
# MDNS
mdns-sd = { version = "0.13.9", default-features = false, optional = true, features = [
"logging",
] }
[dev-dependencies] [dev-dependencies]
anyhow = { version = "1.0.93" } anyhow = { version = "1.0.93" }
criterion = { version = "0.6.0" } # Used for benchmarks criterion = { version = "0.6.0" } # Used for benchmarks

View File

@@ -3,7 +3,6 @@
[![MIT licensed](https://img.shields.io/crates/l/adb_client.svg)](./LICENSE-MIT) [![MIT licensed](https://img.shields.io/crates/l/adb_client.svg)](./LICENSE-MIT)
[![Documentation](https://docs.rs/adb_client/badge.svg)](https://docs.rs/adb_client) [![Documentation](https://docs.rs/adb_client/badge.svg)](https://docs.rs/adb_client)
[![Crates.io Total Downloads](https://img.shields.io/crates/d/adb_client)](https://crates.io/crates/adb_client) [![Crates.io Total Downloads](https://img.shields.io/crates/d/adb_client)](https://crates.io/crates/adb_client)
![MSRV](https://img.shields.io/crates/msrv/adb_client)
Rust library implementing ADB protocol. Rust library implementing ADB protocol.

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 { Self { transport }
transport,
local_id: None,
remote_id: None,
}
} }
pub(crate) fn get_transport(&mut self) -> &T { pub(crate) fn get_transport(&mut self) -> &T {
@@ -34,12 +34,15 @@ 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(&mut self) -> Result<ADBTransportMessage> { pub(crate) fn recv_and_reply_okay(
&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,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.remote_id,
&[], &[],
))?; ))?;
Ok(message) Ok(message)
@@ -60,11 +63,12 @@ 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()?.into_payload(); let payload = self.recv_and_reply_okay(session)?.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() {
@@ -97,8 +101,7 @@ 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,
local_id: u32, session: ADBSession,
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];
@@ -111,8 +114,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
local_id, session.local_id,
remote_id, session.remote_id,
&serialized_message, &serialized_message,
); );
@@ -131,8 +134,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
local_id, session.local_id,
remote_id, session.remote_id,
&serialized_message, &serialized_message,
); );
@@ -144,7 +147,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
MessageCommand::Write => return Ok(()), MessageCommand::Write => return Ok(()),
c => { c => {
return Err(RustADBError::ADBRequestFailed(format!( return Err(RustADBError::ADBRequestFailed(format!(
"Wrong command received {c}" "Wrong command received {}",
c
))); )));
} }
} }
@@ -158,8 +162,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let message = ADBTransportMessage::new( let message = ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
local_id, session.local_id,
remote_id, session.remote_id,
&serialized_message, &serialized_message,
); );
@@ -172,24 +176,27 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
} }
} }
pub(crate) fn begin_synchronization(&mut self) -> Result<()> { pub(crate) fn begin_synchronization(&mut self) -> Result<ADBSession> {
self.open_session(b"sync:\0")?; self.open_session(b"sync:\0")
Ok(())
} }
pub(crate) fn stat_with_explicit_ids(&mut self, remote_path: &str) -> Result<AdbStatResponse> { pub(crate) fn stat_with_explicit_ids(
&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,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.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,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.remote_id,
remote_path.as_bytes(), remote_path.as_bytes(),
))?; ))?;
let response = self.transport.read_message()?; let response = self.transport.read_message()?;
@@ -199,24 +206,25 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
.map_err(|_e| RustADBError::ConversionError) .map_err(|_e| RustADBError::ConversionError)
} }
pub(crate) fn end_transaction(&mut self) -> Result<()> { pub(crate) fn end_transaction(&mut self, session: ADBSession) -> 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,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.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<ADBTransportMessage> { pub(crate) fn open_session(&mut self, data: &[u8]) -> Result<ADBSession> {
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,
rng.random(), // Our 'local-id' local_id, // Our 'local-id'
0, 0,
data, data,
); );
@@ -224,21 +232,25 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let response = self.get_transport_mut().read_message()?; let response = self.get_transport_mut().read_message()?;
self.local_id = Some(response.header().arg1()); if response.header().command() != MessageCommand::Okay {
self.remote_id = Some(response.header().arg0()); return Err(RustADBError::ADBRequestFailed(format!(
"Open session failed: got {} in respone instead of OKAY",
response.header().command()
)));
}
Ok(response) if response.header().arg1() != local_id {
} return Err(RustADBError::ADBRequestFailed(format!(
"Open session failed: respones used {} for our local_id instead of {local_id}",
response.header().arg1()
)));
}
pub(crate) fn get_local_id(&self) -> Result<u32> { let session = ADBSession {
self.local_id.ok_or(RustADBError::ADBRequestFailed( local_id,
"connection not opened, no local_id".into(), remote_id: response.header().arg0(),
)) };
}
pub(crate) fn get_remote_id(&self) -> Result<u32> { Ok(session)
self.remote_id.ok_or(RustADBError::ADBRequestFailed(
"connection not opened, no remote_id".into(),
))
} }
} }

View File

@@ -19,24 +19,19 @@ use crate::device::adb_transport_message::{AUTH_RSAPUBLICKEY, AUTH_SIGNATURE, AU
use crate::{Result, RustADBError, USBTransport}; use crate::{Result, RustADBError, USBTransport};
pub fn read_adb_private_key<P: AsRef<Path>>(private_key_path: P) -> Result<Option<ADBRsaKey>> { pub fn read_adb_private_key<P: AsRef<Path>>(private_key_path: P) -> Result<Option<ADBRsaKey>> {
// Try to read the private key file from given path Ok(read_to_string(private_key_path.as_ref()).map(|pk| {
// If the file is not found, return None match ADBRsaKey::new_from_pkcs8(&pk) {
// If there is another error while reading the file, return this error Ok(pk) => Some(pk),
// Else, return the private key content Err(e) => {
let pk = match read_to_string(private_key_path.as_ref()) { log::error!("Error while create RSA private key: {e}");
Ok(pk) => pk, None
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(None), }
Err(e) => return Err(e.into()), }
}; })?)
match ADBRsaKey::new_from_pkcs8(&pk) {
Ok(pk) => Ok(Some(pk)),
Err(e) => Err(e),
}
} }
/// Search for adb devices with known interface class and subclass values /// Search for adb devices with known interface class and subclass values
pub fn search_adb_devices() -> Result<Option<(u16, u16)>> { fn search_adb_devices() -> Result<Option<(u16, u16)>> {
let mut found_devices = vec![]; let mut found_devices = vec![];
for device in rusb::devices()?.iter() { for device in rusb::devices()?.iter() {
let Ok(des) = device.device_descriptor() else { let Ok(des) = device.device_descriptor() else {
@@ -56,13 +51,13 @@ pub fn search_adb_devices() -> Result<Option<(u16, u16)>> {
(None, _) => Ok(None), (None, _) => Ok(None),
(Some(identifiers), None) => Ok(Some(*identifiers)), (Some(identifiers), None) => Ok(Some(*identifiers)),
(Some((vid1, pid1)), Some((vid2, pid2))) => Err(RustADBError::DeviceNotFound(format!( (Some((vid1, pid1)), Some((vid2, pid2))) => Err(RustADBError::DeviceNotFound(format!(
"Found two Android devices {vid1:04x}:{pid1:04x} and {vid2:04x}:{pid2:04x}", "Found two Android devices {:04x}:{:04x} and {:04x}:{:04x}",
vid1, pid1, vid2, pid2
))), ))),
} }
} }
/// Check whether a device with given descriptor is an ADB device fn is_adb_device<T: UsbContext>(device: &Device<T>, des: &DeviceDescriptor) -> bool {
pub fn is_adb_device<T: UsbContext>(device: &Device<T>, des: &DeviceDescriptor) -> bool {
const ADB_SUBCLASS: u8 = 0x42; const ADB_SUBCLASS: u8 = 0x42;
const ADB_PROTOCOL: u8 = 0x1; const ADB_PROTOCOL: u8 = 0x1;
@@ -139,15 +134,9 @@ impl ADBUSBDevice {
transport: USBTransport, transport: USBTransport,
private_key_path: PathBuf, private_key_path: PathBuf,
) -> Result<Self> { ) -> Result<Self> {
let private_key = match read_adb_private_key(&private_key_path)? { let private_key = match read_adb_private_key(private_key_path)? {
Some(pk) => pk, Some(pk) => pk,
None => { None => ADBRsaKey::new_random()?,
log::warn!(
"No private key found at path {}. Using a temporary random one.",
private_key_path.display()
);
ADBRsaKey::new_random()?
}
}; };
let mut s = Self { let mut s = Self {
@@ -191,11 +180,6 @@ impl ADBUSBDevice {
self.get_transport_mut().write_message(message)?; self.get_transport_mut().write_message(message)?;
let message = self.get_transport_mut().read_message()?; let message = self.get_transport_mut().read_message()?;
// If the device returned CNXN instead of AUTH it does not require authentication,
// so we can skip the auth steps.
if message.header().command() == MessageCommand::Cnxn {
return Ok(());
}
message.assert_command(MessageCommand::Auth)?; message.assert_command(MessageCommand::Auth)?;
// At this point, we should have receive an AUTH message with arg0 == 1 // At this point, we should have receive an AUTH message with arg0 == 1

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>>> {
self.open_session(b"framebuffer:\0")?; let session = self.open_session(b"framebuffer:\0")?;
let response = self.recv_and_reply_okay()?; let response = self.recv_and_reply_okay(session)?;
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()?; let response = self.recv_and_reply_okay(session)?;
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()?; let response = self.recv_and_reply_okay(session)?;
framebuffer_data.extend_from_slice(&response.into_payload()); framebuffer_data.extend_from_slice(&response.into_payload());

View File

@@ -14,11 +14,12 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
let file_size = apk_file.metadata()?.len(); let file_size = apk_file.metadata()?.len();
self.open_session(format!("exec:cmd package 'install' -S {file_size}\0").as_bytes())?; let session =
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, self.get_local_id()?, self.get_remote_id()?); let mut writer = MessageWriter::new(transport, session.local_id, session.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<()> {
self.begin_synchronization()?; let session = self.begin_synchronization()?;
let source = source.as_ref(); let source = source.as_ref();
let adb_stat_response = self.stat_with_explicit_ids(source)?; let adb_stat_response = self.stat_with_explicit_ids(session, source)?;
if adb_stat_response.file_perm == 0 { if adb_stat_response.file_perm == 0 {
return Err(RustADBError::UnknownResponseType( return Err(RustADBError::UnknownResponseType(
@@ -21,11 +21,13 @@ 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(MessageCommand::Okay, local_id, remote_id, &[]), ADBTransportMessage::new(
MessageCommand::Okay,
session.local_id,
session.remote_id,
&[],
),
std::time::Duration::from_secs(4), std::time::Duration::from_secs(4),
)?; )?;
@@ -34,19 +36,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,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.remote_id,
&recv_buffer, &recv_buffer,
))?; ))?;
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.remote_id,
source.as_bytes(), source.as_bytes(),
))?; ))?;
self.recv_file(output)?; self.recv_file(session, output)?;
self.end_transaction()?; self.end_transaction(session)?;
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<()> {
self.begin_synchronization()?; let session = self.begin_synchronization()?;
let path_header = format!("{},0777", path.as_ref()); let path_header = format!("{},0777", path.as_ref());
@@ -21,14 +21,13 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
self.send_and_expect_okay(ADBTransportMessage::new( self.send_and_expect_okay(ADBTransportMessage::new(
MessageCommand::Write, MessageCommand::Write,
self.get_local_id()?, session.local_id,
self.get_remote_id()?, session.remote_id,
&send_buffer, &send_buffer,
))?; ))?;
self.push_file(self.get_local_id()?, self.get_remote_id()?, stream)?; self.push_file(session, stream)?;
self.end_transaction(session)?;
self.end_transaction()?;
Ok(()) Ok(())
} }

View File

@@ -5,7 +5,7 @@ use crate::{
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn reboot(&mut self, reboot_type: RebootType) -> Result<()> { pub(crate) fn reboot(&mut self, reboot_type: RebootType) -> Result<()> {
self.open_session(format!("reboot:{reboot_type}\0").as_bytes())?; self.open_session(format!("reboot:{}\0", reboot_type).as_bytes())?;
self.get_transport_mut() self.get_transport_mut()
.read_message() .read_message()

View File

@@ -10,14 +10,7 @@ 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 response = self.open_session(format!("shell:{}\0", command.join(" "),).as_bytes())?; let session = 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()?;
@@ -38,21 +31,22 @@ 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<()> {
self.open_session(b"shell:\0")?; let session = 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 = let response = ADBTransportMessage::new(
ADBTransportMessage::new(MessageCommand::Okay, local_id, remote_id, &[]); MessageCommand::Okay,
session.local_id,
session.remote_id,
&[],
);
transport.write_message(response)?; transport.write_message(response)?;
match message.header().command() { match message.header().command() {
@@ -67,7 +61,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
}); });
let transport = self.get_transport().clone(); let transport = self.get_transport().clone();
let mut shell_writer = ShellMessageWriter::new(transport, local_id, remote_id); let mut shell_writer =
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> {
self.begin_synchronization()?; let session = self.begin_synchronization()?;
let adb_stat_response = self.stat_with_explicit_ids(remote_path)?; let adb_stat_response = self.stat_with_explicit_ids(session, remote_path)?;
self.end_transaction()?; self.end_transaction(session)?;
Ok(adb_stat_response) Ok(adb_stat_response)
} }
} }

View File

@@ -2,13 +2,13 @@ use crate::{ADBMessageTransport, Result, device::adb_message_device::ADBMessageD
impl<T: ADBMessageTransport> ADBMessageDevice<T> { impl<T: ADBMessageTransport> ADBMessageDevice<T> {
pub(crate) fn uninstall(&mut self, package_name: &str) -> Result<()> { pub(crate) fn uninstall(&mut self, package_name: &str) -> Result<()> {
self.open_session(format!("exec:cmd package 'uninstall' {package_name}\0").as_bytes())?; self.open_session(format!("exec:cmd package 'uninstall' {}\0", package_name).as_bytes())?;
let final_status = self.get_transport_mut().read_message()?; let final_status = self.get_transport_mut().read_message()?;
match final_status.into_payload().as_slice() { match final_status.into_payload().as_slice() {
b"Success\n" => { b"Success\n" => {
log::info!("Package {package_name} successfully uninstalled"); log::info!("Package {} successfully uninstalled", package_name);
Ok(()) Ok(())
} }
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8( d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(

View File

@@ -35,10 +35,10 @@ impl<T: ADBMessageTransport> Write for MessageWriter<T> {
Ok(response) => { Ok(response) => {
response response
.assert_command(MessageCommand::Okay) .assert_command(MessageCommand::Okay)
.map_err(Error::other)?; .map_err(|e| Error::new(ErrorKind::Other, e))?;
Ok(buf.len()) Ok(buf.len())
} }
Err(e) => Err(Error::other(e)), Err(e) => Err(Error::new(ErrorKind::Other, e)),
} }
} }

View File

@@ -11,9 +11,7 @@ mod shell_message_writer;
use adb_message_device::ADBMessageDevice; use adb_message_device::ADBMessageDevice;
pub use adb_tcp_device::ADBTcpDevice; pub use adb_tcp_device::ADBTcpDevice;
pub use adb_transport_message::{ADBTransportMessage, ADBTransportMessageHeader}; pub use adb_transport_message::{ADBTransportMessage, ADBTransportMessageHeader};
pub use adb_usb_device::{ pub use adb_usb_device::{ADBUSBDevice, get_default_adb_key_path};
ADBUSBDevice, get_default_adb_key_path, is_adb_device, search_adb_devices,
};
pub use message_writer::MessageWriter; pub use message_writer::MessageWriter;
pub use models::{ADBRsaKey, MessageCommand, MessageSubcommand}; pub use models::{ADBRsaKey, MessageCommand, MessageSubcommand};
pub use shell_message_writer::ShellMessageWriter; pub use shell_message_writer::ShellMessageWriter;

View File

@@ -30,7 +30,8 @@ impl ADBEmulatorDevice {
let groups = EMULATOR_REGEX let groups = EMULATOR_REGEX
.captures(&identifier) .captures(&identifier)
.ok_or(RustADBError::DeviceNotFound(format!( .ok_or(RustADBError::DeviceNotFound(format!(
"Device {identifier} is likely not an emulator" "Device {} is likely not an emulator",
identifier
)))?; )))?;
let port = groups let port = groups

View File

@@ -112,7 +112,6 @@ pub enum RustADBError {
#[error("upgrade error: {0}")] #[error("upgrade error: {0}")]
UpgradeError(String), UpgradeError(String),
/// An error occurred while getting mdns devices /// An error occurred while getting mdns devices
#[cfg(feature = "mdns")]
#[error(transparent)] #[error(transparent)]
MDNSError(#[from] mdns_sd::Error), MDNSError(#[from] mdns_sd::Error),
/// An error occurred while sending data to channel /// An error occurred while sending data to channel

View File

@@ -9,7 +9,6 @@ mod constants;
mod device; mod device;
mod emulator_device; mod emulator_device;
mod error; mod error;
#[cfg(feature = "mdns")]
mod mdns; mod mdns;
mod models; mod models;
mod server; mod server;
@@ -18,10 +17,9 @@ mod transports;
mod utils; mod utils;
pub use adb_device_ext::ADBDeviceExt; pub use adb_device_ext::ADBDeviceExt;
pub use device::{ADBTcpDevice, ADBUSBDevice, is_adb_device, search_adb_devices}; pub use device::{ADBTcpDevice, ADBUSBDevice};
pub use emulator_device::ADBEmulatorDevice; pub use emulator_device::ADBEmulatorDevice;
pub use error::{Result, RustADBError}; pub use error::{Result, RustADBError};
#[cfg(feature = "mdns")]
pub use mdns::*; pub use mdns::*;
pub use models::{AdbStatResponse, RebootType}; pub use models::{AdbStatResponse, RebootType};
pub use server::*; pub use server::*;

View File

@@ -18,9 +18,7 @@ pub(crate) enum AdbServerCommand {
Pair(SocketAddrV4, String), Pair(SocketAddrV4, String),
TransportAny, TransportAny,
TransportSerial(String), TransportSerial(String),
#[cfg(feature = "mdns")]
MDNSCheck, MDNSCheck,
#[cfg(feature = "mdns")]
MDNSServices, MDNSServices,
ServerStatus, ServerStatus,
ReconnectOffline, ReconnectOffline,
@@ -65,8 +63,8 @@ impl Display for AdbServerCommand {
AdbServerCommand::Reboot(reboot_type) => { AdbServerCommand::Reboot(reboot_type) => {
write!(f, "reboot:{reboot_type}") write!(f, "reboot:{reboot_type}")
} }
AdbServerCommand::Connect(addr) => write!(f, "host:connect:{addr}"), AdbServerCommand::Connect(addr) => write!(f, "host:connect:{}", addr),
AdbServerCommand::Disconnect(addr) => write!(f, "host:disconnect:{addr}"), AdbServerCommand::Disconnect(addr) => write!(f, "host:disconnect:{}", addr),
AdbServerCommand::Pair(addr, code) => { AdbServerCommand::Pair(addr, code) => {
write!(f, "host:pair:{code}:{addr}") write!(f, "host:pair:{code}:{addr}")
} }
@@ -79,9 +77,7 @@ impl Display for AdbServerCommand {
write!(f, "reverse:forward:{remote};{local}") write!(f, "reverse:forward:{remote};{local}")
} }
AdbServerCommand::ReverseRemoveAll => write!(f, "reverse:killforward-all"), AdbServerCommand::ReverseRemoveAll => write!(f, "reverse:killforward-all"),
#[cfg(feature = "mdns")]
AdbServerCommand::MDNSCheck => write!(f, "host:mdns:check"), AdbServerCommand::MDNSCheck => write!(f, "host:mdns:check"),
#[cfg(feature = "mdns")]
AdbServerCommand::MDNSServices => write!(f, "host:mdns:services"), AdbServerCommand::MDNSServices => write!(f, "host:mdns:services"),
AdbServerCommand::ServerStatus => write!(f, "host:server-status"), AdbServerCommand::ServerStatus => write!(f, "host:server-status"),
AdbServerCommand::Reconnect => write!(f, "reconnect"), AdbServerCommand::Reconnect => write!(f, "reconnect"),

View File

@@ -6,7 +6,6 @@ use crate::{
const OPENSCREEN_MDNS_BACKEND: &str = "ADB_MDNS_OPENSCREEN"; const OPENSCREEN_MDNS_BACKEND: &str = "ADB_MDNS_OPENSCREEN";
#[cfg(feature = "mdns")]
impl ADBServer { impl ADBServer {
/// Check if mdns discovery is available /// Check if mdns discovery is available
pub fn mdns_check(&mut self) -> Result<bool> { pub fn mdns_check(&mut self) -> Result<bool> {
@@ -33,7 +32,7 @@ impl ADBServer {
Ok(service) => { Ok(service) => {
vec_services.push(MDNSServices::try_from(service.as_bytes())?); vec_services.push(MDNSServices::try_from(service.as_bytes())?);
} }
Err(e) => log::error!("{e}"), Err(e) => log::error!("{}", e),
} }
} }

View File

@@ -2,7 +2,6 @@ mod connect;
mod devices; mod devices;
mod disconnect; mod disconnect;
mod kill; mod kill;
#[cfg(feature = "mdns")]
mod mdns; mod mdns;
mod pair; mod pair;
mod reconnect; mod reconnect;

View File

@@ -2,7 +2,6 @@ mod adb_version;
mod device_long; mod device_long;
mod device_short; mod device_short;
mod device_state; mod device_state;
#[cfg(feature = "mdns")]
mod mdns_services; mod mdns_services;
mod server_status; mod server_status;
mod wait_for_device; mod wait_for_device;
@@ -11,7 +10,6 @@ pub use adb_version::AdbVersion;
pub use device_long::DeviceLong; pub use device_long::DeviceLong;
pub use device_short::DeviceShort; pub use device_short::DeviceShort;
pub use device_state::DeviceState; pub use device_state::DeviceState;
#[cfg(feature = "mdns")]
pub use mdns_services::MDNSServices; pub use mdns_services::MDNSServices;
pub use server_status::{MDNSBackend, ServerStatus}; pub use server_status::{MDNSBackend, ServerStatus};
pub use wait_for_device::{WaitForDeviceState, WaitForDeviceTransport}; pub use wait_for_device::{WaitForDeviceState, WaitForDeviceTransport};

View File

@@ -64,7 +64,7 @@ impl ADBServerDevice {
"DONE" => { "DONE" => {
return Ok(()); return Ok(());
} }
x => log::error!("Got an unknown response {x}"), x => log::error!("Got an unknown response {}", x),
} }
} }
} }

View File

@@ -42,14 +42,18 @@ impl<R: Read> Read for ADBRecvCommandReader<R> {
let mut error_msg = vec![0; length]; let mut error_msg = vec![0; length];
self.inner.read_exact(&mut error_msg)?; self.inner.read_exact(&mut error_msg)?;
Err(std::io::Error::other(format!( Err(std::io::Error::new(
"ADB request failed: {}", std::io::ErrorKind::Other,
String::from_utf8_lossy(&error_msg) format!(
))) "ADB request failed: {}",
String::from_utf8_lossy(&error_msg)
),
))
} }
_ => Err(std::io::Error::other(format!( _ => Err(std::io::Error::new(
"Unknown response from device {header:#?}" std::io::ErrorKind::Other,
))), format!("Unknown response from device {:#?}", header),
)),
} }
} else { } else {
// Computing minimum to ensure to stop reading before next header... // Computing minimum to ensure to stop reading before next header...

View File

@@ -31,7 +31,8 @@ impl ADBServerDevice {
Ok(data.into()) Ok(data.into())
} }
x => Err(RustADBError::UnknownResponseType(format!( x => Err(RustADBError::UnknownResponseType(format!(
"Unknown response {x}" "Unknown response {}",
x
))), ))),
} }
} }

View File

@@ -15,7 +15,7 @@ impl ADBServerDevice {
match &data[0..read_amount] { match &data[0..read_amount] {
b"Success\n" => { b"Success\n" => {
log::info!("Package {package_name} successfully uninstalled"); log::info!("Package {} successfully uninstalled", package_name);
Ok(()) Ok(())
} }
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8( d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(

View File

@@ -99,13 +99,13 @@ impl TCPServerTransport {
} }
/// Gets the body length from a LittleEndian value /// Gets the body length from a LittleEndian value
pub(crate) fn get_body_length(&self) -> Result<u32> { pub(crate) fn get_body_length(&mut self) -> Result<u32> {
let length_buffer = self.read_body_length()?; let length_buffer = self.read_body_length()?;
Ok(LittleEndian::read_u32(&length_buffer)) Ok(LittleEndian::read_u32(&length_buffer))
} }
/// Read 4 bytes representing body length /// Read 4 bytes representing body length
fn read_body_length(&self) -> Result<[u8; 4]> { fn read_body_length(&mut self) -> Result<[u8; 4]> {
let mut length_buffer = [0; 4]; let mut length_buffer = [0; 4];
self.get_raw_connection()?.read_exact(&mut length_buffer)?; self.get_raw_connection()?.read_exact(&mut length_buffer)?;

View File

@@ -109,7 +109,7 @@ impl TcpTransport {
}) })
} }
fn get_current_connection(&self) -> Result<Arc<Mutex<CurrentConnection>>> { fn get_current_connection(&mut self) -> Result<Arc<Mutex<CurrentConnection>>> {
self.current_connection self.current_connection
.as_ref() .as_ref()
.ok_or(RustADBError::IOError(std::io::Error::new( .ok_or(RustADBError::IOError(std::io::Error::new(
@@ -175,7 +175,8 @@ impl TcpTransport {
Ok(()) Ok(())
} }
c => Err(RustADBError::ADBRequestFailed(format!( c => Err(RustADBError::ADBRequestFailed(format!(
"Wrong command received {c}" "Wrong command received {}",
c
))), ))),
} }
} }

View File

@@ -40,7 +40,8 @@ impl USBTransport {
} }
Err(RustADBError::DeviceNotFound(format!( Err(RustADBError::DeviceNotFound(format!(
"cannot find USB device with vendor_id={vendor_id} and product_id={product_id}", "cannot find USB device with vendor_id={} and product_id={}",
vendor_id, product_id
))) )))
} }
@@ -150,7 +151,12 @@ impl USBTransport {
let write_amount = handle.write_bulk(endpoint.address, &data[offset..end], timeout)?; let write_amount = handle.write_bulk(endpoint.address, &data[offset..end], timeout)?;
offset += write_amount; offset += write_amount;
log::trace!("wrote chunk of size {write_amount} - {offset}/{data_len}",) log::trace!(
"wrote chunk of size {} - {}/{}",
write_amount,
offset,
data_len
)
} }
if offset % max_packet_size == 0 { if offset % max_packet_size == 0 {
@@ -169,11 +175,11 @@ impl ADBTransport for USBTransport {
let (read_endpoint, write_endpoint) = self.find_endpoints(&device)?; let (read_endpoint, write_endpoint) = self.find_endpoints(&device)?;
Self::configure_endpoint(&device, &read_endpoint)?; Self::configure_endpoint(&device, &read_endpoint)?;
log::debug!("got read endpoint: {read_endpoint:?}"); log::debug!("got read endpoint: {:?}", read_endpoint);
self.read_endpoint = Some(read_endpoint); self.read_endpoint = Some(read_endpoint);
Self::configure_endpoint(&device, &write_endpoint)?; Self::configure_endpoint(&device, &write_endpoint)?;
log::debug!("got write endpoint: {write_endpoint:?}"); log::debug!("got write endpoint: {:?}", write_endpoint);
self.write_endpoint = Some(write_endpoint); self.write_endpoint = Some(write_endpoint);
self.handle = Some(Arc::new(device)); self.handle = Some(Arc::new(device));
@@ -191,7 +197,7 @@ impl ADBTransport for USBTransport {
let endpoint = self.read_endpoint.as_ref().or(self.write_endpoint.as_ref()); let endpoint = self.read_endpoint.as_ref().or(self.write_endpoint.as_ref());
if let Some(endpoint) = &endpoint { if let Some(endpoint) = &endpoint {
match handle.release_interface(endpoint.iface) { match handle.release_interface(endpoint.iface) {
Ok(()) => log::debug!("succesfully released interface"), Ok(_) => log::debug!("succesfully released interface"),
Err(e) => log::error!("error while release interface: {e}"), Err(e) => log::error!("error while release interface: {e}"),
} }
} }

View File

@@ -9,7 +9,6 @@ name = "pyadb_client"
readme = "README.md" readme = "README.md"
repository.workspace = true repository.workspace = true
version.workspace = true version.workspace = true
rust-version.workspace = true
[lib] [lib]
crate-type = ["cdylib", "rlib"] crate-type = ["cdylib", "rlib"]