Compare commits
28 Commits
bump-versi
...
v2.1.18
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
36097fdc07 | ||
|
|
4ab0b01e37 | ||
|
|
ee3d879744 | ||
|
|
31d5388ffa | ||
|
|
f108b5ed5f | ||
|
|
46c8c54751 | ||
|
|
f616373ae2 | ||
|
|
c83df4309f | ||
|
|
88526aa943 | ||
|
|
8cce59193c | ||
|
|
e0b6bb0139 | ||
|
|
373a5265a0 | ||
|
|
060e43590d | ||
|
|
f51ba984ca | ||
|
|
66ebc8a030 | ||
|
|
d39e98695d | ||
|
|
34d5811420 | ||
|
|
86e28a6e25 | ||
|
|
9f113bdb93 | ||
|
|
8670d6db58 | ||
|
|
0732a0bbad | ||
|
|
b5673001ca | ||
|
|
81829c1523 | ||
|
|
b9d2b8374f | ||
|
|
5716784f5d | ||
|
|
b6ddc720d8 | ||
|
|
5438e53361 | ||
|
|
39a7f0a8cf |
12
.github/workflows/rust-build.yml
vendored
12
.github/workflows/rust-build.yml
vendored
@@ -18,6 +18,12 @@ jobs:
|
|||||||
os: [ubuntu-latest, windows-latest, macos-latest]
|
os: [ubuntu-latest, windows-latest, macos-latest]
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
- name: Build project
|
|
||||||
run: cargo build --release --all-features
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: "3.10"
|
||||||
|
|
||||||
|
- name: Build project
|
||||||
|
run: cargo build --release --all-features
|
||||||
|
|||||||
5
.github/workflows/rust-release.yml
vendored
5
.github/workflows/rust-release.yml
vendored
@@ -92,6 +92,11 @@ jobs:
|
|||||||
toolchain: stable
|
toolchain: stable
|
||||||
override: true
|
override: true
|
||||||
|
|
||||||
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: "3.10"
|
||||||
|
|
||||||
- name: Build release
|
- name: Build release
|
||||||
run: cargo build --all-features --release
|
run: cargo build --all-features --release
|
||||||
|
|
||||||
|
|||||||
@@ -9,7 +9,8 @@ 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.13"
|
version = "2.1.18"
|
||||||
|
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]
|
||||||
|
|||||||
@@ -8,6 +8,9 @@
|
|||||||
<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>
|
||||||
|
|||||||
@@ -7,14 +7,15 @@ 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]
|
||||||
adb_client = { version = "^2.0.0" }
|
adb_client = { version = "^2.1.17" }
|
||||||
anyhow = { version = "1.0.94" }
|
anyhow = { version = "1.0.100" }
|
||||||
clap = { version = "4.5.23", features = ["derive"] }
|
clap = { version = "4.5.51", features = ["derive"] }
|
||||||
env_logger = { version = "0.11.5" }
|
env_logger = { version = "0.11.8" }
|
||||||
log = { version = "0.4.26" }
|
log = { version = "0.4.28" }
|
||||||
|
|
||||||
[target.'cfg(unix)'.dependencies]
|
[target.'cfg(unix)'.dependencies]
|
||||||
termios = { version = "0.3.3" }
|
termios = { version = "0.3.3" }
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
[](./LICENSE-MIT)
|
[](./LICENSE-MIT)
|
||||||

|

|
||||||
|

|
||||||
|
|
||||||
Rust binary providing an improved version of `adb` CLI.
|
Rust binary providing an improved version of `adb` CLI.
|
||||||
|
|
||||||
@@ -10,7 +11,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`:
|
||||||
@@ -64,4 +65,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
|
||||||
```
|
```
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ pub struct ADBTermios {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl ADBTermios {
|
impl ADBTermios {
|
||||||
pub fn new(fd: impl AsRawFd) -> Result<Self> {
|
pub fn new(fd: &impl AsRawFd) -> Result<Self> {
|
||||||
let mut new_termios = Termios::from_fd(fd.as_raw_fd())?;
|
let mut new_termios = Termios::from_fd(fd.as_raw_fd())?;
|
||||||
let old_termios = new_termios; // Saves previous state
|
let old_termios = new_termios; // Saves previous state
|
||||||
new_termios.c_lflag = 0;
|
new_termios.c_lflag = 0;
|
||||||
@@ -36,7 +36,7 @@ impl Drop for ADBTermios {
|
|||||||
fn drop(&mut self) {
|
fn drop(&mut self) {
|
||||||
// Custom drop implementation, restores previous termios structure.
|
// Custom drop implementation, restores previous termios structure.
|
||||||
if let Err(e) = tcsetattr(self.fd, TCSANOW, &self.old_termios) {
|
if let Err(e) = tcsetattr(self.fd, TCSANOW, &self.old_termios) {
|
||||||
log::error!("Error while dropping ADBTermios: {e}")
|
log::error!("Error while dropping ADBTermios: {e}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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");
|
||||||
@@ -53,19 +53,22 @@ pub fn handle_host_commands(server_command: ServerCommand<HostCommand>) -> Resul
|
|||||||
let server_status = adb_server.server_status()?;
|
let server_status = adb_server.server_status()?;
|
||||||
match server_status.mdns_backend {
|
match server_status.mdns_backend {
|
||||||
MDNSBackend::Unknown => log::info!("unknown mdns backend..."),
|
MDNSBackend::Unknown => log::info!("unknown mdns backend..."),
|
||||||
MDNSBackend::Bonjour => match check {
|
MDNSBackend::Bonjour => {
|
||||||
true => log::info!("mdns daemon version [Bonjour]"),
|
if check {
|
||||||
false => log::info!("ERROR: mdns daemon unavailable"),
|
log::info!("mdns daemon version [Bonjour]");
|
||||||
},
|
} else {
|
||||||
|
log::info!("ERROR: mdns daemon unavailable");
|
||||||
|
}
|
||||||
|
}
|
||||||
MDNSBackend::OpenScreen => {
|
MDNSBackend::OpenScreen => {
|
||||||
log::info!("mdns daemon version [Openscreen discovery 0.0.0]")
|
log::info!("mdns daemon version [Openscreen discovery 0.0.0]");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
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}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ pub fn handle_local_commands(
|
|||||||
let features = device
|
let features = device
|
||||||
.host_features()?
|
.host_features()?
|
||||||
.iter()
|
.iter()
|
||||||
.map(|v| v.to_string())
|
.map(ToString::to_string)
|
||||||
.reduce(|a, b| format!("{a},{b}"))
|
.reduce(|a, b| format!("{a},{b}"))
|
||||||
.ok_or(anyhow!("cannot list features"))?;
|
.ok_or(anyhow!("cannot list features"))?;
|
||||||
log::info!("Available host features: {features}");
|
log::info!("Available host features: {features}");
|
||||||
|
|||||||
@@ -79,7 +79,10 @@ fn main() -> Result<()> {
|
|||||||
(device.boxed(), usb_command.commands)
|
(device.boxed(), usb_command.commands)
|
||||||
}
|
}
|
||||||
MainCommand::Tcp(tcp_command) => {
|
MainCommand::Tcp(tcp_command) => {
|
||||||
let device = ADBTcpDevice::new(tcp_command.address)?;
|
let device = match tcp_command.path_to_private_key {
|
||||||
|
Some(pk) => ADBTcpDevice::new_with_custom_private_key(tcp_command.address, pk)?,
|
||||||
|
None => ADBTcpDevice::new(tcp_command.address)?,
|
||||||
|
};
|
||||||
(device.boxed(), tcp_command.commands)
|
(device.boxed(), tcp_command.commands)
|
||||||
}
|
}
|
||||||
MainCommand::Mdns => {
|
MainCommand::Mdns => {
|
||||||
@@ -94,7 +97,7 @@ fn main() -> Result<()> {
|
|||||||
"Found device {} with addresses {:?}",
|
"Found device {} with addresses {:?}",
|
||||||
device.fullname,
|
device.fullname,
|
||||||
device.addresses
|
device.addresses
|
||||||
)
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
return Ok(service.shutdown()?);
|
return Ok(service.shutdown()?);
|
||||||
@@ -108,7 +111,7 @@ fn main() -> Result<()> {
|
|||||||
// Using a scope here would call drop() too early..
|
// Using a scope here would call drop() too early..
|
||||||
#[cfg(any(target_os = "linux", target_os = "macos"))]
|
#[cfg(any(target_os = "linux", target_os = "macos"))]
|
||||||
{
|
{
|
||||||
let mut adb_termios = ADBTermios::new(std::io::stdin())?;
|
let mut adb_termios = ADBTermios::new(&std::io::stdin())?;
|
||||||
adb_termios.set_adb_termios()?;
|
adb_termios.set_adb_termios()?;
|
||||||
device.shell(&mut std::io::stdin(), Box::new(std::io::stdout()))?;
|
device.shell(&mut std::io::stdin(), Box::new(std::io::stdout()))?;
|
||||||
}
|
}
|
||||||
@@ -118,7 +121,7 @@ fn main() -> Result<()> {
|
|||||||
device.shell(&mut std::io::stdin(), Box::new(std::io::stdout()))?;
|
device.shell(&mut std::io::stdin(), Box::new(std::io::stdout()))?;
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
let commands: Vec<&str> = commands.iter().map(|v| v.as_str()).collect();
|
let commands: Vec<&str> = commands.iter().map(String::as_str).collect();
|
||||||
device.shell_command(&commands, &mut std::io::stdout())?;
|
device.shell_command(&commands, &mut std::io::stdout())?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -132,11 +135,11 @@ 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 } => {
|
||||||
let mut input = File::open(Path::new(&filename))?;
|
let mut input = File::open(Path::new(&filename))?;
|
||||||
@@ -152,7 +155,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 } => {
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ pub enum DeviceCommands {
|
|||||||
/// The package whose activity is to be invoked
|
/// The package whose activity is to be invoked
|
||||||
#[clap(short = 'p', long = "package")]
|
#[clap(short = 'p', long = "package")]
|
||||||
package: String,
|
package: String,
|
||||||
/// The activity to be invoked itself, Usually it is MainActivity
|
/// The activity to be invoked itself, Usually it is `MainActivity`
|
||||||
#[clap(short = 'a', long = "activity")]
|
#[clap(short = 'a', long = "activity")]
|
||||||
activity: String,
|
activity: String,
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1,11 +1,15 @@
|
|||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use super::DeviceCommands;
|
use super::DeviceCommands;
|
||||||
|
|
||||||
#[derive(Parser, Debug)]
|
#[derive(Parser, Debug)]
|
||||||
pub struct TcpCommand {
|
pub struct TcpCommand {
|
||||||
pub address: SocketAddr,
|
pub address: SocketAddr,
|
||||||
|
/// Path to a custom private key to use for authentication
|
||||||
|
#[clap(short = 'k', long = "private-key")]
|
||||||
|
pub path_to_private_key: Option<PathBuf>,
|
||||||
#[clap(subcommand)]
|
#[clap(subcommand)]
|
||||||
pub commands: DeviceCommands,
|
pub commands: DeviceCommands,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,10 +5,7 @@
|
|||||||
pub unsafe fn setup_logger(debug: bool) {
|
pub unsafe fn setup_logger(debug: bool) {
|
||||||
// RUST_LOG variable has more priority then "--debug" flag
|
// RUST_LOG variable has more priority then "--debug" flag
|
||||||
if std::env::var("RUST_LOG").is_err() {
|
if std::env::var("RUST_LOG").is_err() {
|
||||||
let level = match debug {
|
let level = if debug { "trace" } else { "info" };
|
||||||
true => "trace",
|
|
||||||
false => "info",
|
|
||||||
};
|
|
||||||
|
|
||||||
unsafe { std::env::set_var("RUST_LOG", level) };
|
unsafe { std::env::set_var("RUST_LOG", level) };
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,40 +7,37 @@ 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
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
base64 = { version = "0.22.1" }
|
base64 = { version = "0.22.1" }
|
||||||
bincode = { version = "1.3.3" }
|
bincode = { version = "2.0.1", features = ["serde"] }
|
||||||
byteorder = { version = "1.5.0" }
|
byteorder = { version = "1.5.0" }
|
||||||
chrono = { version = "0.4.40", default-features = false, features = ["std"] }
|
chrono = { version = "0.4.42", default-features = false, features = ["std"] }
|
||||||
homedir = { version = "0.3.4" }
|
image = { version = "0.25.8", default-features = false }
|
||||||
image = { version = "0.25.5", default-features = false }
|
log = { version = "0.4.28" }
|
||||||
log = { version = "0.4.26" }
|
mdns-sd = { version = "0.17.0", default-features = false, features = [
|
||||||
mdns-sd = { version = "0.13.9", default-features = false, features = [
|
|
||||||
"logging",
|
"logging",
|
||||||
] }
|
] }
|
||||||
num-bigint = { version = "0.8.4", package = "num-bigint-dig" }
|
num-bigint = { version = "0.8.5", 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" }
|
||||||
rand = { version = "0.9.0" }
|
rand = { version = "0.9.2" }
|
||||||
rcgen = { version = "0.13.1", default-features = false, features = [
|
rcgen = { version = "0.14.5" }
|
||||||
"aws_lc_rs",
|
regex = { version = "1.12.2", features = ["perf", "std", "unicode"] }
|
||||||
"pem",
|
rsa = { version = "0.9.8" }
|
||||||
] }
|
|
||||||
regex = { version = "1.11.1", features = ["perf", "std", "unicode"] }
|
|
||||||
rsa = { version = "0.9.7" }
|
|
||||||
rusb = { version = "0.9.4", features = ["vendored"] }
|
rusb = { version = "0.9.4", features = ["vendored"] }
|
||||||
rustls = { version = "0.23.27" }
|
rustls = { version = "0.23.35" }
|
||||||
rustls-pki-types = { version = "1.11.0" }
|
rustls-pki-types = { version = "1.13.0" }
|
||||||
serde = { version = "1.0.216", features = ["derive"] }
|
serde = { version = "1.0.228", features = ["derive"] }
|
||||||
serde_repr = { version = "0.1.19" }
|
serde_repr = { version = "0.1.20" }
|
||||||
sha1 = { version = "0.10.6", features = ["oid"] }
|
sha1 = { version = "0.10.6", features = ["oid"] }
|
||||||
thiserror = { version = "2.0.7" }
|
thiserror = { version = "2.0.17" }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
anyhow = { version = "1.0.93" }
|
anyhow = { version = "1.0.100" }
|
||||||
criterion = { version = "0.5.1" } # Used for benchmarks
|
criterion = { version = "0.7.0" } # Used for benchmarks
|
||||||
|
|
||||||
[[bench]]
|
[[bench]]
|
||||||
harness = false
|
harness = false
|
||||||
|
|||||||
@@ -3,6 +3,7 @@
|
|||||||
[](./LICENSE-MIT)
|
[](./LICENSE-MIT)
|
||||||
[](https://docs.rs/adb_client)
|
[](https://docs.rs/adb_client)
|
||||||
[](https://crates.io/crates/adb_client)
|
[](https://crates.io/crates/adb_client)
|
||||||
|

|
||||||
|
|
||||||
Rust library implementing ADB protocol.
|
Rust library implementing ADB protocol.
|
||||||
|
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ pub trait ADBDeviceExt {
|
|||||||
|
|
||||||
/// Starts an interactive shell session on the device.
|
/// Starts an interactive shell session on the device.
|
||||||
/// Input data is read from reader and write to writer.
|
/// Input data is read from reader and write to writer.
|
||||||
fn shell(&mut self, reader: &mut dyn Read, writer: Box<(dyn Write + Send)>) -> Result<()>;
|
fn shell(&mut self, reader: &mut dyn Read, writer: Box<dyn Write + Send>) -> Result<()>;
|
||||||
|
|
||||||
/// Display the stat information for a remote file
|
/// Display the stat information for a remote file
|
||||||
fn stat(&mut self, remote_path: &str) -> Result<AdbStatResponse>;
|
fn stat(&mut self, remote_path: &str) -> Result<AdbStatResponse>;
|
||||||
|
|||||||
@@ -1,10 +1,26 @@
|
|||||||
use byteorder::{LittleEndian, ReadBytesExt};
|
use super::{ADBRsaKey, ADBTransportMessage, MessageCommand, models::MessageSubcommand};
|
||||||
use rand::Rng;
|
use crate::device::adb_transport_message::{AUTH_RSAPUBLICKEY, AUTH_SIGNATURE, AUTH_TOKEN};
|
||||||
use std::io::{Cursor, Read, Seek};
|
|
||||||
|
|
||||||
use crate::{ADBMessageTransport, AdbStatResponse, Result, RustADBError, constants::BUFFER_SIZE};
|
use crate::{ADBMessageTransport, AdbStatResponse, Result, RustADBError, constants::BUFFER_SIZE};
|
||||||
|
use bincode::config::{Configuration, Fixint, LittleEndian, NoLimit};
|
||||||
|
use byteorder::ReadBytesExt;
|
||||||
|
use rand::Rng;
|
||||||
|
use serde::Serialize;
|
||||||
|
use serde::de::DeserializeOwned;
|
||||||
|
use std::io::{Cursor, Read, Seek};
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
use super::{ADBTransportMessage, MessageCommand, models::MessageSubcommand};
|
const BINCODE_CONFIG: Configuration<LittleEndian, Fixint, NoLimit> = bincode::config::legacy();
|
||||||
|
|
||||||
|
pub(crate) fn bincode_serialize_to_vec<E: Serialize>(val: E) -> Result<Vec<u8>> {
|
||||||
|
bincode::serde::encode_to_vec(val, BINCODE_CONFIG).map_err(|_e| RustADBError::ConversionError)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn bincode_deserialize_from_slice<D: DeserializeOwned>(data: &[u8]) -> Result<D> {
|
||||||
|
let (response, _) = bincode::serde::decode_from_slice(data, BINCODE_CONFIG)
|
||||||
|
.map_err(|_e| RustADBError::ConversionError)?;
|
||||||
|
|
||||||
|
Ok(response)
|
||||||
|
}
|
||||||
|
|
||||||
/// Generic structure representing an ADB device reachable over an [`ADBMessageTransport`].
|
/// Generic structure representing an ADB device reachable over an [`ADBMessageTransport`].
|
||||||
/// Structure is totally agnostic over which transport is truly used.
|
/// Structure is totally agnostic over which transport is truly used.
|
||||||
@@ -33,6 +49,66 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
&mut self.transport
|
&mut self.transport
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(crate) fn auth_handshake(
|
||||||
|
&mut self,
|
||||||
|
message: ADBTransportMessage,
|
||||||
|
private_key: &ADBRsaKey,
|
||||||
|
) -> Result<()> {
|
||||||
|
match message.header().command() {
|
||||||
|
MessageCommand::Auth => {
|
||||||
|
log::debug!("Authentication required");
|
||||||
|
}
|
||||||
|
_ => return Ok(()),
|
||||||
|
}
|
||||||
|
|
||||||
|
// At this point, we should have received an AUTH message with arg0 == 1
|
||||||
|
let auth_message = match message.header().arg0() {
|
||||||
|
AUTH_TOKEN => message,
|
||||||
|
v => {
|
||||||
|
return Err(RustADBError::ADBRequestFailed(format!(
|
||||||
|
"Received AUTH message with type != 1 ({v})"
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let sign = private_key.sign(auth_message.into_payload())?;
|
||||||
|
|
||||||
|
let message = ADBTransportMessage::new(MessageCommand::Auth, AUTH_SIGNATURE, 0, &sign);
|
||||||
|
|
||||||
|
self.get_transport_mut().write_message(message)?;
|
||||||
|
|
||||||
|
let received_response = self.get_transport_mut().read_message()?;
|
||||||
|
|
||||||
|
if received_response.header().command() == MessageCommand::Cnxn {
|
||||||
|
log::info!(
|
||||||
|
"Authentication OK, device info {}",
|
||||||
|
String::from_utf8(received_response.into_payload())?
|
||||||
|
);
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut pubkey = private_key.android_pubkey_encode()?.into_bytes();
|
||||||
|
pubkey.push(b'\0');
|
||||||
|
|
||||||
|
let message = ADBTransportMessage::new(MessageCommand::Auth, AUTH_RSAPUBLICKEY, 0, &pubkey);
|
||||||
|
|
||||||
|
self.get_transport_mut().write_message(message)?;
|
||||||
|
|
||||||
|
let response = self
|
||||||
|
.get_transport_mut()
|
||||||
|
.read_message_with_timeout(Duration::from_secs(10))
|
||||||
|
.and_then(|message| {
|
||||||
|
message.assert_command(MessageCommand::Cnxn)?;
|
||||||
|
Ok(message)
|
||||||
|
})?;
|
||||||
|
|
||||||
|
log::info!(
|
||||||
|
"Authentication OK, device info {}",
|
||||||
|
String::from_utf8(response.into_payload())?
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
/// 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) -> Result<ADBTransportMessage> {
|
||||||
let message = self.transport.read_message()?;
|
let message = self.transport.read_message()?;
|
||||||
@@ -70,7 +146,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
match len.take() {
|
match len.take() {
|
||||||
Some(0) | None => {
|
Some(0) | None => {
|
||||||
rdr.seek_relative(4)?;
|
rdr.seek_relative(4)?;
|
||||||
len.replace(rdr.read_u32::<LittleEndian>()? as u64);
|
len.replace(u64::from(rdr.read_u32::<byteorder::LittleEndian>()?));
|
||||||
}
|
}
|
||||||
Some(length) => {
|
Some(length) => {
|
||||||
let remaining_bytes = payload.len() as u64 - rdr.position();
|
let remaining_bytes = payload.len() as u64 - rdr.position();
|
||||||
@@ -86,7 +162,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if Cursor::new(&payload[(payload.len() - 8)..(payload.len() - 4)])
|
if Cursor::new(&payload[(payload.len() - 8)..(payload.len() - 4)])
|
||||||
.read_u32::<LittleEndian>()?
|
.read_u32::<byteorder::LittleEndian>()?
|
||||||
== MessageSubcommand::Done as u32
|
== MessageSubcommand::Done as u32
|
||||||
{
|
{
|
||||||
break;
|
break;
|
||||||
@@ -101,12 +177,11 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
remote_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 = vec![0; BUFFER_SIZE].into_boxed_slice();
|
||||||
let amount_read = reader.read(&mut buffer)?;
|
let amount_read = reader.read(&mut buffer)?;
|
||||||
let subcommand_data = MessageSubcommand::Data.with_arg(amount_read as u32);
|
let subcommand_data = MessageSubcommand::Data.with_arg(u32::try_from(amount_read)?);
|
||||||
|
|
||||||
let mut serialized_message =
|
let mut serialized_message = bincode_serialize_to_vec(&subcommand_data)?;
|
||||||
bincode::serialize(&subcommand_data).map_err(|_e| RustADBError::ConversionError)?;
|
|
||||||
serialized_message.append(&mut buffer[..amount_read].to_vec());
|
serialized_message.append(&mut buffer[..amount_read].to_vec());
|
||||||
|
|
||||||
let message = ADBTransportMessage::new(
|
let message = ADBTransportMessage::new(
|
||||||
@@ -119,16 +194,14 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
self.send_and_expect_okay(message)?;
|
self.send_and_expect_okay(message)?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
let mut buffer = [0; BUFFER_SIZE];
|
let mut buffer = vec![0; BUFFER_SIZE].into_boxed_slice();
|
||||||
|
|
||||||
match reader.read(&mut buffer) {
|
match reader.read(&mut buffer) {
|
||||||
Ok(0) => {
|
Ok(0) => {
|
||||||
// Currently file mtime is not forwarded
|
// Currently file mtime is not forwarded
|
||||||
let subcommand_data = MessageSubcommand::Done.with_arg(0);
|
let subcommand_data = MessageSubcommand::Done.with_arg(0);
|
||||||
|
|
||||||
let serialized_message = bincode::serialize(&subcommand_data)
|
let serialized_message = bincode_serialize_to_vec(&subcommand_data)?;
|
||||||
.map_err(|_e| RustADBError::ConversionError)?;
|
|
||||||
|
|
||||||
let message = ADBTransportMessage::new(
|
let message = ADBTransportMessage::new(
|
||||||
MessageCommand::Write,
|
MessageCommand::Write,
|
||||||
local_id,
|
local_id,
|
||||||
@@ -144,17 +217,15 @@ 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 {}",
|
"Wrong command received {c}"
|
||||||
c
|
|
||||||
)));
|
)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok(size) => {
|
Ok(size) => {
|
||||||
let subcommand_data = MessageSubcommand::Data.with_arg(size as u32);
|
let subcommand_data = MessageSubcommand::Data.with_arg(u32::try_from(size)?);
|
||||||
|
|
||||||
let mut serialized_message = bincode::serialize(&subcommand_data)
|
let mut serialized_message = bincode_serialize_to_vec(&subcommand_data)?;
|
||||||
.map_err(|_e| RustADBError::ConversionError)?;
|
|
||||||
serialized_message.append(&mut buffer[..size].to_vec());
|
serialized_message.append(&mut buffer[..size].to_vec());
|
||||||
|
|
||||||
let message = ADBTransportMessage::new(
|
let message = ADBTransportMessage::new(
|
||||||
@@ -179,12 +250,12 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn stat_with_explicit_ids(&mut self, remote_path: &str) -> Result<AdbStatResponse> {
|
pub(crate) fn stat_with_explicit_ids(&mut self, remote_path: &str) -> Result<AdbStatResponse> {
|
||||||
let stat_buffer = MessageSubcommand::Stat.with_arg(remote_path.len() as u32);
|
let stat_buffer = MessageSubcommand::Stat.with_arg(u32::try_from(remote_path.len())?);
|
||||||
let message = ADBTransportMessage::new(
|
let message = ADBTransportMessage::new(
|
||||||
MessageCommand::Write,
|
MessageCommand::Write,
|
||||||
self.get_local_id()?,
|
self.get_local_id()?,
|
||||||
self.get_remote_id()?,
|
self.get_remote_id()?,
|
||||||
&bincode::serialize(&stat_buffer).map_err(|_e| RustADBError::ConversionError)?,
|
&bincode_serialize_to_vec(&stat_buffer)?,
|
||||||
);
|
);
|
||||||
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(
|
||||||
@@ -196,8 +267,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
let response = self.transport.read_message()?;
|
let response = self.transport.read_message()?;
|
||||||
// Skip first 4 bytes as this is the literal "STAT".
|
// Skip first 4 bytes as this is the literal "STAT".
|
||||||
// Interesting part starts right after
|
// Interesting part starts right after
|
||||||
bincode::deserialize(&response.into_payload()[4..])
|
|
||||||
.map_err(|_e| RustADBError::ConversionError)
|
bincode_deserialize_from_slice(&response.into_payload()[4..])
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn end_transaction(&mut self) -> Result<()> {
|
pub(crate) fn end_transaction(&mut self) -> Result<()> {
|
||||||
@@ -206,7 +277,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
MessageCommand::Write,
|
MessageCommand::Write,
|
||||||
self.get_local_id()?,
|
self.get_local_id()?,
|
||||||
self.get_remote_id()?,
|
self.get_remote_id()?,
|
||||||
&bincode::serialize(&quit_buffer).map_err(|_e| RustADBError::ConversionError)?,
|
&bincode_serialize_to_vec(&quit_buffer)?,
|
||||||
))?;
|
))?;
|
||||||
let _discard_close = self.transport.read_message()?;
|
let _discard_close = self.transport.read_message()?;
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ impl<T: ADBMessageTransport> ADBDeviceExt for ADBMessageDevice<T> {
|
|||||||
self.shell_command(command, output)
|
self.shell_command(command, output)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn shell(&mut self, reader: &mut dyn Read, writer: Box<(dyn Write + Send)>) -> Result<()> {
|
fn shell(&mut self, reader: &mut dyn Read, writer: Box<dyn Write + Send>) -> Result<()> {
|
||||||
self.shell(reader, writer)
|
self.shell(reader, writer)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,22 +1,43 @@
|
|||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
use std::path::Path;
|
use std::path::{Path, PathBuf};
|
||||||
use std::{io::Read, net::SocketAddr};
|
use std::{io::Read, net::SocketAddr};
|
||||||
|
|
||||||
use super::ADBTransportMessage;
|
|
||||||
use super::adb_message_device::ADBMessageDevice;
|
use super::adb_message_device::ADBMessageDevice;
|
||||||
use super::models::MessageCommand;
|
use super::models::MessageCommand;
|
||||||
|
use super::{ADBRsaKey, ADBTransportMessage, get_default_adb_key_path};
|
||||||
|
use crate::device::adb_usb_device::read_adb_private_key;
|
||||||
use crate::{ADBDeviceExt, ADBMessageTransport, ADBTransport, Result, TcpTransport};
|
use crate::{ADBDeviceExt, ADBMessageTransport, ADBTransport, Result, TcpTransport};
|
||||||
|
|
||||||
/// Represent a device reached and available over USB.
|
/// Represent a device reached and available over USB.
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct ADBTcpDevice {
|
pub struct ADBTcpDevice {
|
||||||
|
private_key: ADBRsaKey,
|
||||||
inner: ADBMessageDevice<TcpTransport>,
|
inner: ADBMessageDevice<TcpTransport>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ADBTcpDevice {
|
impl ADBTcpDevice {
|
||||||
/// Instantiate a new [`ADBTcpDevice`]
|
/// Instantiate a new [`ADBTcpDevice`]
|
||||||
pub fn new(address: SocketAddr) -> Result<Self> {
|
pub fn new(address: SocketAddr) -> Result<Self> {
|
||||||
|
Self::new_with_custom_private_key(address, get_default_adb_key_path()?)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Instantiate a new [`ADBTcpDevice`] using a custom private key path
|
||||||
|
pub fn new_with_custom_private_key(
|
||||||
|
address: SocketAddr,
|
||||||
|
private_key_path: PathBuf,
|
||||||
|
) -> Result<Self> {
|
||||||
|
let private_key = if let Some(private_key) = read_adb_private_key(&private_key_path)? {
|
||||||
|
private_key
|
||||||
|
} else {
|
||||||
|
log::warn!(
|
||||||
|
"No private key found at path {}. Using a temporary random one.",
|
||||||
|
private_key_path.display()
|
||||||
|
);
|
||||||
|
ADBRsaKey::new_random()?
|
||||||
|
};
|
||||||
|
|
||||||
let mut device = Self {
|
let mut device = Self {
|
||||||
|
private_key,
|
||||||
inner: ADBMessageDevice::new(TcpTransport::new(address)?),
|
inner: ADBMessageDevice::new(TcpTransport::new(address)?),
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -31,8 +52,8 @@ impl ADBTcpDevice {
|
|||||||
|
|
||||||
let message = ADBTransportMessage::new(
|
let message = ADBTransportMessage::new(
|
||||||
MessageCommand::Cnxn,
|
MessageCommand::Cnxn,
|
||||||
0x01000000,
|
0x0100_0000,
|
||||||
1048576,
|
1_048_576,
|
||||||
format!("host::{}\0", env!("CARGO_PKG_NAME")).as_bytes(),
|
format!("host::{}\0", env!("CARGO_PKG_NAME")).as_bytes(),
|
||||||
);
|
);
|
||||||
|
|
||||||
@@ -40,26 +61,28 @@ impl ADBTcpDevice {
|
|||||||
|
|
||||||
let message = self.get_transport_mut().read_message()?;
|
let message = self.get_transport_mut().read_message()?;
|
||||||
|
|
||||||
// Check if client is requesting a secure connection and upgrade it if necessary
|
// Check if a client is requesting a secure connection and upgrade it if necessary
|
||||||
match message.header().command() {
|
match message.header().command() {
|
||||||
MessageCommand::Stls => {
|
MessageCommand::Stls => {
|
||||||
self.get_transport_mut()
|
self.get_transport_mut()
|
||||||
.write_message(ADBTransportMessage::new(MessageCommand::Stls, 1, 0, &[]))?;
|
.write_message(ADBTransportMessage::new(MessageCommand::Stls, 1, 0, &[]))?;
|
||||||
self.get_transport_mut().upgrade_connection()?;
|
self.get_transport_mut().upgrade_connection()?;
|
||||||
log::debug!("Connection successfully upgraded from TCP to TLS");
|
log::debug!("Connection successfully upgraded from TCP to TLS");
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
MessageCommand::Cnxn => {
|
MessageCommand::Cnxn => {
|
||||||
log::debug!("Unencrypted connection established");
|
log::debug!("Unencrypted connection established");
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
_ => {
|
MessageCommand::Auth => {
|
||||||
return Err(crate::RustADBError::WrongResponseReceived(
|
log::debug!("Authentication required");
|
||||||
"Expected CNXN or STLS command".to_string(),
|
self.inner.auth_handshake(message, &self.private_key)
|
||||||
message.header().command().to_string(),
|
|
||||||
));
|
|
||||||
}
|
}
|
||||||
|
_ => Err(crate::RustADBError::WrongResponseReceived(
|
||||||
|
"Expected CNXN, STLS or AUTH command".to_string(),
|
||||||
|
message.header().command().to_string(),
|
||||||
|
)),
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
@@ -75,7 +98,7 @@ impl ADBDeviceExt for ADBTcpDevice {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn shell(&mut self, reader: &mut dyn Read, writer: Box<(dyn Write + Send)>) -> Result<()> {
|
fn shell(&mut self, reader: &mut dyn Read, writer: Box<dyn Write + Send>) -> Result<()> {
|
||||||
self.inner.shell(reader, writer)
|
self.inner.shell(reader, writer)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
use crate::{Result, RustADBError};
|
use crate::{Result, RustADBError, device::adb_message_device};
|
||||||
|
|
||||||
use super::models::MessageCommand;
|
use super::models::MessageCommand;
|
||||||
|
|
||||||
@@ -58,16 +58,16 @@ impl ADBTransportMessageHeader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn compute_crc32(data: &[u8]) -> u32 {
|
pub(crate) fn compute_crc32(data: &[u8]) -> u32 {
|
||||||
data.iter().map(|&x| x as u32).sum()
|
data.iter().map(|&x| u32::from(x)).sum()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn compute_magic(command: MessageCommand) -> u32 {
|
fn compute_magic(command: MessageCommand) -> u32 {
|
||||||
let command_u32 = command as u32;
|
let command_u32 = command as u32;
|
||||||
command_u32 ^ 0xFFFFFFFF
|
command_u32 ^ 0xFFFF_FFFF
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn as_bytes(&self) -> Result<Vec<u8>> {
|
pub fn as_bytes(&self) -> Result<Vec<u8>> {
|
||||||
bincode::serialize(&self).map_err(|_e| RustADBError::ConversionError)
|
adb_message_device::bincode_serialize_to_vec(self)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -117,6 +117,6 @@ impl TryFrom<[u8; 24]> for ADBTransportMessageHeader {
|
|||||||
type Error = RustADBError;
|
type Error = RustADBError;
|
||||||
|
|
||||||
fn try_from(value: [u8; 24]) -> Result<Self> {
|
fn try_from(value: [u8; 24]) -> Result<Self> {
|
||||||
bincode::deserialize(&value).map_err(|_e| RustADBError::ConversionError)
|
adb_message_device::bincode_deserialize_from_slice(&value)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ use std::io::Read;
|
|||||||
use std::io::Write;
|
use std::io::Write;
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use std::time::Duration;
|
|
||||||
|
|
||||||
use super::adb_message_device::ADBMessageDevice;
|
use super::adb_message_device::ADBMessageDevice;
|
||||||
use super::models::MessageCommand;
|
use super::models::MessageCommand;
|
||||||
@@ -15,23 +14,27 @@ use super::{ADBRsaKey, ADBTransportMessage};
|
|||||||
use crate::ADBDeviceExt;
|
use crate::ADBDeviceExt;
|
||||||
use crate::ADBMessageTransport;
|
use crate::ADBMessageTransport;
|
||||||
use crate::ADBTransport;
|
use crate::ADBTransport;
|
||||||
use crate::device::adb_transport_message::{AUTH_RSAPUBLICKEY, AUTH_SIGNATURE, AUTH_TOKEN};
|
|
||||||
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>> {
|
||||||
Ok(read_to_string(private_key_path.as_ref()).map(|pk| {
|
// Try to read the private key file from given path
|
||||||
match ADBRsaKey::new_from_pkcs8(&pk) {
|
// If the file is not found, return None
|
||||||
Ok(pk) => Some(pk),
|
// If there is another error while reading the file, return this error
|
||||||
Err(e) => {
|
// Else, return the private key content
|
||||||
log::error!("Error while create RSA private key: {e}");
|
let pk = match read_to_string(private_key_path.as_ref()) {
|
||||||
None
|
Ok(pk) => pk,
|
||||||
}
|
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
|
||||||
fn search_adb_devices() -> Result<Option<(u16, u16)>> {
|
pub 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 {
|
||||||
@@ -51,13 +54,13 @@ 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 {:04x}:{:04x} and {:04x}:{:04x}",
|
"Found two Android devices {vid1:04x}:{pid1:04x} and {vid2:04x}:{pid2:04x}",
|
||||||
vid1, pid1, vid2, pid2
|
|
||||||
))),
|
))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn is_adb_device<T: UsbContext>(device: &Device<T>, des: &DeviceDescriptor) -> bool {
|
/// Check whether a device with given descriptor is an ADB device
|
||||||
|
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;
|
||||||
|
|
||||||
@@ -88,9 +91,7 @@ fn is_adb_device<T: UsbContext>(device: &Device<T>, des: &DeviceDescriptor) -> b
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_default_adb_key_path() -> Result<PathBuf> {
|
pub fn get_default_adb_key_path() -> Result<PathBuf> {
|
||||||
homedir::my_home()
|
std::env::home_dir()
|
||||||
.ok()
|
|
||||||
.flatten()
|
|
||||||
.map(|home| home.join(".android").join("adbkey"))
|
.map(|home| home.join(".android").join("adbkey"))
|
||||||
.ok_or(RustADBError::NoHomeDirectory)
|
.ok_or(RustADBError::NoHomeDirectory)
|
||||||
}
|
}
|
||||||
@@ -114,7 +115,7 @@ impl ADBUSBDevice {
|
|||||||
product_id: u16,
|
product_id: u16,
|
||||||
private_key_path: PathBuf,
|
private_key_path: PathBuf,
|
||||||
) -> Result<Self> {
|
) -> Result<Self> {
|
||||||
Self::new_from_transport_inner(USBTransport::new(vendor_id, product_id)?, private_key_path)
|
Self::new_from_transport_inner(USBTransport::new(vendor_id, product_id)?, &private_key_path)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Instantiate a new [`ADBUSBDevice`] from a [`USBTransport`] and an optional private key path.
|
/// Instantiate a new [`ADBUSBDevice`] from a [`USBTransport`] and an optional private key path.
|
||||||
@@ -127,16 +128,21 @@ impl ADBUSBDevice {
|
|||||||
None => get_default_adb_key_path()?,
|
None => get_default_adb_key_path()?,
|
||||||
};
|
};
|
||||||
|
|
||||||
Self::new_from_transport_inner(transport, private_key_path)
|
Self::new_from_transport_inner(transport, &private_key_path)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn new_from_transport_inner(
|
fn new_from_transport_inner(
|
||||||
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 = if let Some(private_key) = read_adb_private_key(private_key_path)? {
|
||||||
Some(pk) => pk,
|
private_key
|
||||||
None => ADBRsaKey::new_random()?,
|
} else {
|
||||||
|
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 {
|
||||||
@@ -172,63 +178,22 @@ impl ADBUSBDevice {
|
|||||||
|
|
||||||
let message = ADBTransportMessage::new(
|
let message = ADBTransportMessage::new(
|
||||||
MessageCommand::Cnxn,
|
MessageCommand::Cnxn,
|
||||||
0x01000000,
|
0x0100_0000,
|
||||||
1048576,
|
1_048_576,
|
||||||
format!("host::{}\0", env!("CARGO_PKG_NAME")).as_bytes(),
|
format!("host::{}\0", env!("CARGO_PKG_NAME")).as_bytes(),
|
||||||
);
|
);
|
||||||
|
|
||||||
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()?;
|
||||||
message.assert_command(MessageCommand::Auth)?;
|
// If the device returned CNXN instead of AUTH it does not require authentication,
|
||||||
|
// so we can skip the auth steps.
|
||||||
// At this point, we should have receive an AUTH message with arg0 == 1
|
if message.header().command() == MessageCommand::Cnxn {
|
||||||
let auth_message = match message.header().arg0() {
|
|
||||||
AUTH_TOKEN => message,
|
|
||||||
v => {
|
|
||||||
return Err(RustADBError::ADBRequestFailed(format!(
|
|
||||||
"Received AUTH message with type != 1 ({v})"
|
|
||||||
)));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let sign = self.private_key.sign(auth_message.into_payload())?;
|
|
||||||
|
|
||||||
let message = ADBTransportMessage::new(MessageCommand::Auth, AUTH_SIGNATURE, 0, &sign);
|
|
||||||
|
|
||||||
self.get_transport_mut().write_message(message)?;
|
|
||||||
|
|
||||||
let received_response = self.get_transport_mut().read_message()?;
|
|
||||||
|
|
||||||
if received_response.header().command() == MessageCommand::Cnxn {
|
|
||||||
log::info!(
|
|
||||||
"Authentication OK, device info {}",
|
|
||||||
String::from_utf8(received_response.into_payload())?
|
|
||||||
);
|
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut pubkey = self.private_key.android_pubkey_encode()?.into_bytes();
|
message.assert_command(MessageCommand::Auth)?;
|
||||||
pubkey.push(b'\0');
|
self.inner.auth_handshake(message, &self.private_key)
|
||||||
|
|
||||||
let message = ADBTransportMessage::new(MessageCommand::Auth, AUTH_RSAPUBLICKEY, 0, &pubkey);
|
|
||||||
|
|
||||||
self.get_transport_mut().write_message(message)?;
|
|
||||||
|
|
||||||
let response = self
|
|
||||||
.get_transport_mut()
|
|
||||||
.read_message_with_timeout(Duration::from_secs(10))
|
|
||||||
.and_then(|message| {
|
|
||||||
message.assert_command(MessageCommand::Cnxn)?;
|
|
||||||
Ok(message)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
log::info!(
|
|
||||||
"Authentication OK, device info {}",
|
|
||||||
String::from_utf8(response.into_payload())?
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
@@ -244,7 +209,7 @@ impl ADBDeviceExt for ADBUSBDevice {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn shell<'a>(&mut self, reader: &mut dyn Read, writer: Box<(dyn Write + Send)>) -> Result<()> {
|
fn shell<'a>(&mut self, reader: &mut dyn Read, writer: Box<dyn Write + Send>) -> Result<()> {
|
||||||
self.inner.shell(reader, writer)
|
self.inner.shell(reader, writer)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
payload_cursor.read_to_end(&mut framebuffer_data)?;
|
payload_cursor.read_to_end(&mut framebuffer_data)?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
if framebuffer_data.len() as u32 == framebuffer_info.size {
|
if u32::try_from(framebuffer_data.len())? == framebuffer_info.size {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -65,7 +65,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
payload_cursor.read_to_end(&mut framebuffer_data)?;
|
payload_cursor.read_to_end(&mut framebuffer_data)?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
if framebuffer_data.len() as u32 == framebuffer_info.size {
|
if u32::try_from(framebuffer_data.len())? == framebuffer_info.size {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ 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 {}\0", file_size).as_bytes())?;
|
self.open_session(format!("exec:cmd package 'install' -S {file_size}\0").as_bytes())?;
|
||||||
|
|
||||||
let transport = self.get_transport().clone();
|
let transport = self.get_transport().clone();
|
||||||
|
|
||||||
|
|||||||
@@ -3,7 +3,8 @@ use std::io::Write;
|
|||||||
use crate::{
|
use crate::{
|
||||||
ADBMessageTransport, Result, RustADBError,
|
ADBMessageTransport, Result, RustADBError,
|
||||||
device::{
|
device::{
|
||||||
ADBTransportMessage, MessageCommand, adb_message_device::ADBMessageDevice,
|
ADBTransportMessage, MessageCommand,
|
||||||
|
adb_message_device::{self, ADBMessageDevice},
|
||||||
models::MessageSubcommand,
|
models::MessageSubcommand,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
@@ -29,9 +30,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
std::time::Duration::from_secs(4),
|
std::time::Duration::from_secs(4),
|
||||||
)?;
|
)?;
|
||||||
|
|
||||||
let recv_buffer = MessageSubcommand::Recv.with_arg(source.len() as u32);
|
let recv_buffer = MessageSubcommand::Recv.with_arg(u32::try_from(source.len())?);
|
||||||
let recv_buffer =
|
let recv_buffer = adb_message_device::bincode_serialize_to_vec(&recv_buffer)?;
|
||||||
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()?,
|
self.get_local_id()?,
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
use std::io::Read;
|
use std::io::Read;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
ADBMessageTransport, Result, RustADBError,
|
ADBMessageTransport, Result,
|
||||||
device::{
|
device::{
|
||||||
ADBTransportMessage, MessageCommand, MessageSubcommand,
|
ADBTransportMessage, MessageCommand, MessageSubcommand,
|
||||||
adb_message_device::ADBMessageDevice,
|
adb_message_device::{self, ADBMessageDevice},
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -14,9 +14,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
|
|
||||||
let path_header = format!("{},0777", path.as_ref());
|
let path_header = format!("{},0777", path.as_ref());
|
||||||
|
|
||||||
let send_buffer = MessageSubcommand::Send.with_arg(path_header.len() as u32);
|
let send_buffer = MessageSubcommand::Send.with_arg(u32::try_from(path_header.len())?);
|
||||||
let mut send_buffer =
|
let mut send_buffer = adb_message_device::bincode_serialize_to_vec(&send_buffer)?;
|
||||||
bincode::serialize(&send_buffer).map_err(|_e| RustADBError::ConversionError)?;
|
|
||||||
send_buffer.append(&mut path_header.as_bytes().to_vec());
|
send_buffer.append(&mut path_header.as_bytes().to_vec());
|
||||||
|
|
||||||
self.send_and_expect_okay(ADBTransportMessage::new(
|
self.send_and_expect_okay(ADBTransportMessage::new(
|
||||||
|
|||||||
@@ -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:{}\0", reboot_type).as_bytes())?;
|
self.open_session(format!("reboot:{reboot_type}\0").as_bytes())?;
|
||||||
|
|
||||||
self.get_transport_mut()
|
self.get_transport_mut()
|
||||||
.read_message()
|
.read_message()
|
||||||
|
|||||||
@@ -36,7 +36,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
pub(crate) fn shell(
|
pub(crate) fn shell(
|
||||||
&mut self,
|
&mut self,
|
||||||
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")?;
|
self.open_session(b"shell:\0")?;
|
||||||
|
|
||||||
@@ -60,7 +60,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
|||||||
writer.write_all(&message.into_payload())?;
|
writer.write_all(&message.into_payload())?;
|
||||||
writer.flush()?;
|
writer.flush()?;
|
||||||
}
|
}
|
||||||
MessageCommand::Okay => continue,
|
MessageCommand::Okay => {}
|
||||||
_ => return Err(RustADBError::ADBShellNotSupported),
|
_ => return Err(RustADBError::ADBShellNotSupported),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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' {}\0", package_name).as_bytes())?;
|
self.open_session(format!("exec:cmd package 'uninstall' {package_name}\0").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 {} successfully uninstalled", package_name);
|
log::info!("Package {package_name} successfully uninstalled");
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(
|
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(
|
||||||
|
|||||||
@@ -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(|e| Error::new(ErrorKind::Other, e))?;
|
.map_err(Error::other)?;
|
||||||
Ok(buf.len())
|
Ok(buf.len())
|
||||||
}
|
}
|
||||||
Err(e) => Err(Error::new(ErrorKind::Other, e)),
|
Err(e) => Err(Error::other(e)),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -11,7 +11,9 @@ 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::{ADBUSBDevice, get_default_adb_key_path};
|
pub use adb_usb_device::{
|
||||||
|
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;
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ use num_traits::cast::ToPrimitive;
|
|||||||
use rsa::pkcs8::DecodePrivateKey;
|
use rsa::pkcs8::DecodePrivateKey;
|
||||||
use rsa::traits::PublicKeyParts;
|
use rsa::traits::PublicKeyParts;
|
||||||
use rsa::{Pkcs1v15Sign, RsaPrivateKey};
|
use rsa::{Pkcs1v15Sign, RsaPrivateKey};
|
||||||
|
use std::fmt::Write;
|
||||||
|
|
||||||
const ADB_PRIVATE_KEY_SIZE: usize = 2048;
|
const ADB_PRIVATE_KEY_SIZE: usize = 2048;
|
||||||
const ANDROID_PUBKEY_MODULUS_SIZE_WORDS: u32 = 64;
|
const ANDROID_PUBKEY_MODULUS_SIZE_WORDS: u32 = 64;
|
||||||
@@ -92,15 +93,16 @@ impl ADBRsaKey {
|
|||||||
.to_u32()
|
.to_u32()
|
||||||
.ok_or(RustADBError::ConversionError)?;
|
.ok_or(RustADBError::ConversionError)?;
|
||||||
|
|
||||||
Ok(self.encode_public_key(adb_rsa_pubkey.into_bytes()))
|
Self::encode_public_key(adb_rsa_pubkey.into_bytes())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn encode_public_key(&self, pub_key: Vec<u8>) -> String {
|
fn encode_public_key(pub_key: Vec<u8>) -> Result<String> {
|
||||||
let mut encoded = STANDARD.encode(pub_key);
|
let mut encoded = STANDARD.encode(pub_key);
|
||||||
encoded.push(' ');
|
encoded.push(' ');
|
||||||
encoded.push_str(&format!("adb_client@{}", env!("CARGO_PKG_VERSION")));
|
write!(encoded, "adb_client@{}", env!("CARGO_PKG_VERSION"))
|
||||||
|
.map_err(|_| RustADBError::ConversionError)?;
|
||||||
|
|
||||||
encoded
|
Ok(encoded)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn sign(&self, msg: impl AsRef<[u8]>) -> Result<Vec<u8>> {
|
pub fn sign(&self, msg: impl AsRef<[u8]>) -> Result<Vec<u8>> {
|
||||||
|
|||||||
@@ -6,32 +6,32 @@ use std::fmt::Display;
|
|||||||
#[repr(u32)]
|
#[repr(u32)]
|
||||||
pub enum MessageCommand {
|
pub enum MessageCommand {
|
||||||
/// Connect to a device
|
/// Connect to a device
|
||||||
Cnxn = 0x4E584E43,
|
Cnxn = 0x4E58_4E43,
|
||||||
/// Close connection to a device
|
/// Close connection to a device
|
||||||
Clse = 0x45534C43,
|
Clse = 0x4553_4C43,
|
||||||
/// Device ask for authentication
|
/// Device ask for authentication
|
||||||
Auth = 0x48545541,
|
Auth = 0x4854_5541,
|
||||||
/// Open a data connection
|
/// Open a data connection
|
||||||
Open = 0x4E45504F,
|
Open = 0x4E45_504F,
|
||||||
/// Write data to connection
|
/// Write data to connection
|
||||||
Write = 0x45545257,
|
Write = 0x4554_5257,
|
||||||
/// Server understood the message
|
/// Server understood the message
|
||||||
Okay = 0x59414B4F,
|
Okay = 0x5941_4B4F,
|
||||||
/// Start a connection using TLS
|
/// Start a connection using TLS
|
||||||
Stls = 0x534C5453,
|
Stls = 0x534C_5453,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Eq, PartialEq, Copy, Clone, Serialize_repr, Deserialize_repr)]
|
#[derive(Debug, Eq, PartialEq, Copy, Clone, Serialize_repr, Deserialize_repr)]
|
||||||
#[repr(u32)]
|
#[repr(u32)]
|
||||||
pub enum MessageSubcommand {
|
pub enum MessageSubcommand {
|
||||||
Stat = 0x54415453,
|
Stat = 0x5441_5453,
|
||||||
Send = 0x444E4553,
|
Send = 0x444E_4553,
|
||||||
Recv = 0x56434552,
|
Recv = 0x5643_4552,
|
||||||
Quit = 0x54495551,
|
Quit = 0x5449_5551,
|
||||||
Fail = 0x4C494146,
|
Fail = 0x4C49_4146,
|
||||||
Done = 0x454E4F44,
|
Done = 0x454E_4F44,
|
||||||
Data = 0x41544144,
|
Data = 0x4154_4144,
|
||||||
List = 0x5453494C,
|
List = 0x5453_494C,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
|
|||||||
@@ -15,23 +15,22 @@ static EMULATOR_REGEX: LazyLock<Regex> = LazyLock::new(|| {
|
|||||||
pub struct ADBEmulatorDevice {
|
pub struct ADBEmulatorDevice {
|
||||||
/// Unique device identifier.
|
/// Unique device identifier.
|
||||||
pub identifier: String,
|
pub identifier: String,
|
||||||
/// Internal [TCPEmulatorTransport]
|
/// Internal [`TCPEmulatorTransport`]
|
||||||
transport: TCPEmulatorTransport,
|
transport: TCPEmulatorTransport,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ADBEmulatorDevice {
|
impl ADBEmulatorDevice {
|
||||||
/// Instantiates a new [ADBEmulatorDevice]
|
/// Instantiates a new [`ADBEmulatorDevice`]
|
||||||
pub fn new(identifier: String, ip_address: Option<Ipv4Addr>) -> Result<Self> {
|
pub fn new(identifier: String, ip_address: Option<Ipv4Addr>) -> Result<Self> {
|
||||||
let ip_address = match ip_address {
|
let ip_address = match ip_address {
|
||||||
Some(ip_address) => ip_address,
|
Some(ip_address) => ip_address,
|
||||||
None => Ipv4Addr::new(127, 0, 0, 1),
|
None => Ipv4Addr::LOCALHOST,
|
||||||
};
|
};
|
||||||
|
|
||||||
let groups = EMULATOR_REGEX
|
let groups = EMULATOR_REGEX
|
||||||
.captures(&identifier)
|
.captures(&identifier)
|
||||||
.ok_or(RustADBError::DeviceNotFound(format!(
|
.ok_or(RustADBError::DeviceNotFound(format!(
|
||||||
"Device {} is likely not an emulator",
|
"Device {identifier} is likely not an emulator"
|
||||||
identifier
|
|
||||||
)))?;
|
)))?;
|
||||||
|
|
||||||
let port = groups
|
let port = groups
|
||||||
|
|||||||
@@ -24,8 +24,7 @@ impl ADBEmulatorCommand {
|
|||||||
pub(crate) fn skip_response_lines(&self) -> u8 {
|
pub(crate) fn skip_response_lines(&self) -> u8 {
|
||||||
match self {
|
match self {
|
||||||
ADBEmulatorCommand::Authenticate(_) => 1,
|
ADBEmulatorCommand::Authenticate(_) => 1,
|
||||||
ADBEmulatorCommand::Sms(_, _) => 0,
|
ADBEmulatorCommand::Sms(_, _) | ADBEmulatorCommand::Rotate => 0,
|
||||||
ADBEmulatorCommand::Rotate => 0,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -42,6 +42,9 @@ pub enum RustADBError {
|
|||||||
/// Indicates that an error occurred when converting a value.
|
/// Indicates that an error occurred when converting a value.
|
||||||
#[error("Conversion error")]
|
#[error("Conversion error")]
|
||||||
ConversionError,
|
ConversionError,
|
||||||
|
/// Indicates an error with the integer conversion.
|
||||||
|
#[error(transparent)]
|
||||||
|
IntegerConversionError(#[from] std::num::TryFromIntError),
|
||||||
/// Remote ADB server does not support shell feature.
|
/// Remote ADB server does not support shell feature.
|
||||||
#[error("Remote ADB server does not support shell feature")]
|
#[error("Remote ADB server does not support shell feature")]
|
||||||
ADBShellNotSupported,
|
ADBShellNotSupported,
|
||||||
@@ -61,8 +64,8 @@ pub enum RustADBError {
|
|||||||
#[error("Unimplemented framebuffer image version: {0}")]
|
#[error("Unimplemented framebuffer image version: {0}")]
|
||||||
UnimplementedFramebufferImageVersion(u32),
|
UnimplementedFramebufferImageVersion(u32),
|
||||||
/// An error occurred while getting user's home directory
|
/// An error occurred while getting user's home directory
|
||||||
#[error(transparent)]
|
#[error("Cannot get user home directory")]
|
||||||
HomeError(#[from] homedir::GetHomeError),
|
HomeError,
|
||||||
/// Cannot get home directory
|
/// Cannot get home directory
|
||||||
#[error("Cannot get home directory")]
|
#[error("Cannot get home directory")]
|
||||||
NoHomeDirectory,
|
NoHomeDirectory,
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ mod transports;
|
|||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
pub use adb_device_ext::ADBDeviceExt;
|
pub use adb_device_ext::ADBDeviceExt;
|
||||||
pub use device::{ADBTcpDevice, ADBUSBDevice};
|
pub use device::{ADBTcpDevice, ADBUSBDevice, is_adb_device, search_adb_devices};
|
||||||
pub use emulator_device::ADBEmulatorDevice;
|
pub use emulator_device::ADBEmulatorDevice;
|
||||||
pub use error::{Result, RustADBError};
|
pub use error::{Result, RustADBError};
|
||||||
pub use mdns::*;
|
pub use mdns::*;
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
use std::{collections::HashSet, net::IpAddr};
|
use std::{collections::HashSet, net::IpAddr};
|
||||||
|
|
||||||
|
use mdns_sd::{ResolvedService, ScopedIp};
|
||||||
|
|
||||||
/// Represent a device found from mdns search
|
/// Represent a device found from mdns search
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct MDNSDevice {
|
pub struct MDNSDevice {
|
||||||
@@ -9,11 +11,11 @@ pub struct MDNSDevice {
|
|||||||
pub addresses: HashSet<IpAddr>,
|
pub addresses: HashSet<IpAddr>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<mdns_sd::ServiceInfo> for MDNSDevice {
|
impl From<Box<ResolvedService>> for MDNSDevice {
|
||||||
fn from(value: mdns_sd::ServiceInfo) -> Self {
|
fn from(value: Box<ResolvedService>) -> Self {
|
||||||
Self {
|
Self {
|
||||||
fullname: value.get_fullname().to_string(),
|
fullname: value.fullname,
|
||||||
addresses: value.get_addresses().to_owned(),
|
addresses: value.addresses.iter().map(ScopedIp::to_ip_addr).collect(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -42,13 +42,15 @@ impl MDNSDiscoveryService {
|
|||||||
| ServiceEvent::ServiceFound(_, _)
|
| ServiceEvent::ServiceFound(_, _)
|
||||||
| ServiceEvent::SearchStopped(_) => {
|
| ServiceEvent::SearchStopped(_) => {
|
||||||
// Ignoring these events. We are only interesting in found devices
|
// Ignoring these events. We are only interesting in found devices
|
||||||
continue;
|
|
||||||
}
|
}
|
||||||
ServiceEvent::ServiceResolved(service_info) => {
|
ServiceEvent::ServiceResolved(service_info) => {
|
||||||
if let Err(e) = sender.send(MDNSDevice::from(service_info)) {
|
if let Err(e) = sender.send(MDNSDevice::from(service_info)) {
|
||||||
return Err(e.into());
|
return Err(e.into());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
e => {
|
||||||
|
log::warn!("received unknown event type {e:?}");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -63,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}")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ use std::process::Command;
|
|||||||
/// Represents an ADB Server
|
/// Represents an ADB Server
|
||||||
#[derive(Debug, Default)]
|
#[derive(Debug, Default)]
|
||||||
pub struct ADBServer {
|
pub struct ADBServer {
|
||||||
/// Internal [TcpStream], lazily initialized
|
/// Internal [`TcpStream`], lazily initialized
|
||||||
pub(crate) transport: Option<TCPServerTransport>,
|
pub(crate) transport: Option<TCPServerTransport>,
|
||||||
/// Address to connect to
|
/// Address to connect to
|
||||||
pub(crate) socket_addr: Option<SocketAddrV4>,
|
pub(crate) socket_addr: Option<SocketAddrV4>,
|
||||||
@@ -21,7 +21,7 @@ pub struct ADBServer {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl ADBServer {
|
impl ADBServer {
|
||||||
/// Instantiates a new [ADBServer]
|
/// Instantiates a new [`ADBServer`]
|
||||||
pub fn new(address: SocketAddrV4) -> Self {
|
pub fn new(address: SocketAddrV4) -> Self {
|
||||||
Self {
|
Self {
|
||||||
transport: None,
|
transport: None,
|
||||||
@@ -31,7 +31,7 @@ impl ADBServer {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Instantiates a new [ADBServer] with a custom adb path
|
/// Instantiates a new [`ADBServer`] with a custom adb path
|
||||||
pub fn new_from_path(address: SocketAddrV4, adb_path: Option<String>) -> Self {
|
pub fn new_from_path(address: SocketAddrV4, adb_path: Option<String>) -> Self {
|
||||||
Self {
|
Self {
|
||||||
transport: None,
|
transport: None,
|
||||||
@@ -46,7 +46,7 @@ impl ADBServer {
|
|||||||
// ADB Server is local, we start it if not already running
|
// ADB Server is local, we start it if not already running
|
||||||
let mut command = Command::new(adb_path.as_deref().unwrap_or("adb"));
|
let mut command = Command::new(adb_path.as_deref().unwrap_or("adb"));
|
||||||
command.arg("start-server");
|
command.arg("start-server");
|
||||||
for (env_k, env_v) in envs.iter() {
|
for (env_k, env_v) in envs {
|
||||||
command.env(env_k, env_v);
|
command.env(env_k, env_v);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -61,7 +61,7 @@ impl ADBServer {
|
|||||||
match child {
|
match child {
|
||||||
Ok(mut child) => {
|
Ok(mut child) => {
|
||||||
if let Err(e) = child.wait() {
|
if let Err(e) = child.wait() {
|
||||||
log::error!("error while starting adb server: {e}")
|
log::error!("error while starting adb server: {e}");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(e) => log::error!("error while starting adb server: {e}"),
|
Err(e) => log::error!("error while starting adb server: {e}"),
|
||||||
|
|||||||
@@ -69,12 +69,12 @@ impl ADBServer {
|
|||||||
.filter(|d| d.identifier.as_str() == name)
|
.filter(|d| d.identifier.as_str() == name)
|
||||||
.collect::<Vec<DeviceShort>>()
|
.collect::<Vec<DeviceShort>>()
|
||||||
.len();
|
.len();
|
||||||
if nb_devices != 1 {
|
if nb_devices == 1 {
|
||||||
|
Ok(ADBServerDevice::new(name.to_string(), self.socket_addr))
|
||||||
|
} else {
|
||||||
Err(RustADBError::DeviceNotFound(format!(
|
Err(RustADBError::DeviceNotFound(format!(
|
||||||
"could not find device {name}"
|
"could not find device {name}"
|
||||||
)))
|
)))
|
||||||
} else {
|
|
||||||
Ok(ADBServerDevice::new(name.to_string(), self.socket_addr))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -32,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}"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ pub struct AdbVersion {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl AdbVersion {
|
impl AdbVersion {
|
||||||
/// Instantiates a new [AdbVersion].
|
/// Instantiates a new [`AdbVersion`].
|
||||||
pub fn new(minor: u32, revision: u32) -> Self {
|
pub fn new(minor: u32, revision: u32) -> Self {
|
||||||
Self {
|
Self {
|
||||||
major: 1,
|
major: 1,
|
||||||
|
|||||||
@@ -86,15 +86,13 @@ impl TryFrom<&[u8]> for DeviceLong {
|
|||||||
None => "Unk".to_string(),
|
None => "Unk".to_string(),
|
||||||
Some(device) => String::from_utf8(device.as_bytes().to_vec())?,
|
Some(device) => String::from_utf8(device.as_bytes().to_vec())?,
|
||||||
},
|
},
|
||||||
transport_id: u32::from_str_radix(
|
transport_id: (str::from_utf8(
|
||||||
str::from_utf8(
|
groups
|
||||||
groups
|
.name("transport_id")
|
||||||
.name("transport_id")
|
.ok_or(RustADBError::RegexParsingError)?
|
||||||
.ok_or(RustADBError::RegexParsingError)?
|
.as_bytes(),
|
||||||
.as_bytes(),
|
)?)
|
||||||
)?,
|
.parse::<u32>()?,
|
||||||
16,
|
|
||||||
)?,
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -48,11 +48,11 @@ impl Display for UsbBackend {
|
|||||||
#[derive(Debug, Clone, PartialEq, Default)]
|
#[derive(Debug, Clone, PartialEq, Default)]
|
||||||
pub enum MDNSBackend {
|
pub enum MDNSBackend {
|
||||||
#[default]
|
#[default]
|
||||||
/// Unknown
|
/// `Unknown`
|
||||||
Unknown = 0,
|
Unknown = 0,
|
||||||
/// Bonjour
|
/// `Bonjour`
|
||||||
Bonjour = 1,
|
Bonjour = 1,
|
||||||
/// OpenScreen
|
/// `OpenScreen`
|
||||||
OpenScreen = 2,
|
OpenScreen = 2,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ use std::fmt::Display;
|
|||||||
|
|
||||||
use crate::RustADBError;
|
use crate::RustADBError;
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug, Default)]
|
||||||
/// List of available transports to wait for.
|
/// List of available transports to wait for.
|
||||||
pub enum WaitForDeviceTransport {
|
pub enum WaitForDeviceTransport {
|
||||||
/// USB transport
|
/// USB transport
|
||||||
@@ -10,15 +10,10 @@ pub enum WaitForDeviceTransport {
|
|||||||
/// Local transport
|
/// Local transport
|
||||||
Local,
|
Local,
|
||||||
/// Any transport (default value)
|
/// Any transport (default value)
|
||||||
|
#[default]
|
||||||
Any,
|
Any,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for WaitForDeviceTransport {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self::Any
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Display for WaitForDeviceTransport {
|
impl Display for WaitForDeviceTransport {
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
|
|||||||
@@ -6,12 +6,12 @@ use std::net::SocketAddrV4;
|
|||||||
pub struct ADBServerDevice {
|
pub struct ADBServerDevice {
|
||||||
/// Unique device identifier.
|
/// Unique device identifier.
|
||||||
pub identifier: Option<String>,
|
pub identifier: Option<String>,
|
||||||
/// Internal [TCPServerTransport]
|
/// Internal [`TCPServerTransport`]
|
||||||
pub(crate) transport: TCPServerTransport,
|
pub(crate) transport: TCPServerTransport,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ADBServerDevice {
|
impl ADBServerDevice {
|
||||||
/// Instantiates a new [ADBServerDevice], knowing its ADB identifier (as returned by `adb devices` command).
|
/// Instantiates a new [`ADBServerDevice`], knowing its ADB identifier (as returned by `adb devices` command).
|
||||||
pub fn new(identifier: String, server_addr: Option<SocketAddrV4>) -> Self {
|
pub fn new(identifier: String, server_addr: Option<SocketAddrV4>) -> Self {
|
||||||
let transport = TCPServerTransport::new_or_default(server_addr);
|
let transport = TCPServerTransport::new_or_default(server_addr);
|
||||||
|
|
||||||
@@ -21,7 +21,7 @@ impl ADBServerDevice {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Instantiates a new [ADBServerDevice], assuming only one is currently connected.
|
/// Instantiates a new [`ADBServerDevice`], assuming only one is currently connected.
|
||||||
pub fn autodetect(server_addr: Option<SocketAddrV4>) -> Self {
|
pub fn autodetect(server_addr: Option<SocketAddrV4>) -> Self {
|
||||||
let transport = TCPServerTransport::new_or_default(server_addr);
|
let transport = TCPServerTransport::new_or_default(server_addr);
|
||||||
|
|
||||||
|
|||||||
@@ -25,15 +25,14 @@ impl ADBDeviceExt for ADBServerDevice {
|
|||||||
self.transport
|
self.transport
|
||||||
.send_adb_request(AdbServerCommand::ShellCommand(command.join(" ")))?;
|
.send_adb_request(AdbServerCommand::ShellCommand(command.join(" ")))?;
|
||||||
|
|
||||||
|
let mut buffer = vec![0; BUFFER_SIZE].into_boxed_slice();
|
||||||
loop {
|
loop {
|
||||||
let mut buffer = [0; BUFFER_SIZE];
|
|
||||||
match self.transport.get_raw_connection()?.read(&mut buffer) {
|
match self.transport.get_raw_connection()?.read(&mut buffer) {
|
||||||
Ok(size) => {
|
Ok(size) => {
|
||||||
if size == 0 {
|
if size == 0 {
|
||||||
return Ok(());
|
return Ok(());
|
||||||
} else {
|
|
||||||
output.write_all(&buffer[..size])?;
|
|
||||||
}
|
}
|
||||||
|
output.write_all(&buffer[..size])?;
|
||||||
}
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
return Err(RustADBError::IOError(e));
|
return Err(RustADBError::IOError(e));
|
||||||
@@ -49,7 +48,7 @@ impl ADBDeviceExt for ADBServerDevice {
|
|||||||
fn shell(
|
fn shell(
|
||||||
&mut self,
|
&mut self,
|
||||||
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 supported_features = self.host_features()?;
|
let supported_features = self.host_features()?;
|
||||||
if !supported_features.contains(&HostFeatures::ShellV2)
|
if !supported_features.contains(&HostFeatures::ShellV2)
|
||||||
@@ -67,8 +66,9 @@ impl ADBDeviceExt for ADBServerDevice {
|
|||||||
|
|
||||||
// Reading thread, reads response from adb-server
|
// Reading thread, reads response from adb-server
|
||||||
std::thread::spawn(move || -> Result<()> {
|
std::thread::spawn(move || -> Result<()> {
|
||||||
|
let mut buffer = vec![0; BUFFER_SIZE].into_boxed_slice();
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
let mut buffer = [0; BUFFER_SIZE];
|
|
||||||
match read_stream.read(&mut buffer) {
|
match read_stream.read(&mut buffer) {
|
||||||
Ok(0) => {
|
Ok(0) => {
|
||||||
read_stream.shutdown(std::net::Shutdown::Both)?;
|
read_stream.shutdown(std::net::Shutdown::Both)?;
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ impl ADBServerDevice {
|
|||||||
// 'DONE' directly without listing anything.
|
// 'DONE' directly without listing anything.
|
||||||
fn handle_list_command<S: AsRef<str>>(&mut self, path: S) -> Result<()> {
|
fn handle_list_command<S: AsRef<str>>(&mut self, path: S) -> Result<()> {
|
||||||
let mut len_buf = [0_u8; 4];
|
let mut len_buf = [0_u8; 4];
|
||||||
LittleEndian::write_u32(&mut len_buf, path.as_ref().len() as u32);
|
LittleEndian::write_u32(&mut len_buf, u32::try_from(path.as_ref().len())?);
|
||||||
|
|
||||||
// 4 bytes of command name is already sent by send_sync_request
|
// 4 bytes of command name is already sent by send_sync_request
|
||||||
self.transport.get_raw_connection()?.write_all(&len_buf)?;
|
self.transport.get_raw_connection()?.write_all(&len_buf)?;
|
||||||
@@ -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}"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,21 +23,26 @@ impl<W: Write> LogFilter<W> {
|
|||||||
|
|
||||||
impl<W: Write> Write for LogFilter<W> {
|
impl<W: Write> Write for LogFilter<W> {
|
||||||
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
||||||
|
// Add newly received bytes to the internal buffer
|
||||||
self.buffer.extend_from_slice(buf);
|
self.buffer.extend_from_slice(buf);
|
||||||
|
|
||||||
let buf_clone = self.buffer.clone();
|
let mut processed = 0;
|
||||||
let mut lines = buf_clone.split_inclusive(|&byte| byte == b'\n').peekable();
|
while let Some(pos) = self.buffer[processed..].iter().position(|&b| b == b'\n') {
|
||||||
|
// Found a newline, need to process it
|
||||||
|
let end = processed + pos + 1; // +1 to include the '\n'
|
||||||
|
let line = &self.buffer[processed..end];
|
||||||
|
|
||||||
while let Some(line) = lines.next() {
|
if self.should_write(line) {
|
||||||
if lines.peek().is_some() {
|
self.writer.write_all(line)?;
|
||||||
if self.should_write(line) {
|
|
||||||
self.writer.write_all(line)?;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// This is the last (unfinished) element, we keep it for next round
|
|
||||||
self.buffer = line.to_vec();
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
processed = end;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Keep only remaining bytes after the last complete line
|
||||||
|
if processed > 0 {
|
||||||
|
self.buffer.copy_within(processed.., 0);
|
||||||
|
self.buffer.truncate(self.buffer.len() - processed);
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(buf.len())
|
Ok(buf.len())
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ use crate::{
|
|||||||
use byteorder::{LittleEndian, ReadBytesExt};
|
use byteorder::{LittleEndian, ReadBytesExt};
|
||||||
use std::io::{BufReader, BufWriter, Read, Write};
|
use std::io::{BufReader, BufWriter, Read, Write};
|
||||||
|
|
||||||
/// Internal structure wrapping a [std::io::Read] and hiding underlying protocol logic.
|
/// Internal structure wrapping a [`std::io::Read`] and hiding underlying protocol logic.
|
||||||
struct ADBRecvCommandReader<R: Read> {
|
struct ADBRecvCommandReader<R: Read> {
|
||||||
inner: R,
|
inner: R,
|
||||||
remaining_data_bytes_to_read: usize,
|
remaining_data_bytes_to_read: usize,
|
||||||
@@ -42,18 +42,14 @@ 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::new(
|
Err(std::io::Error::other(format!(
|
||||||
std::io::ErrorKind::Other,
|
"ADB request failed: {}",
|
||||||
format!(
|
String::from_utf8_lossy(&error_msg)
|
||||||
"ADB request failed: {}",
|
)))
|
||||||
String::from_utf8_lossy(&error_msg)
|
|
||||||
),
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
_ => Err(std::io::Error::new(
|
_ => Err(std::io::Error::other(format!(
|
||||||
std::io::ErrorKind::Other,
|
"Unknown response from device {header:#?}"
|
||||||
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...
|
||||||
@@ -90,7 +86,7 @@ impl ADBServerDevice {
|
|||||||
|
|
||||||
let from_as_bytes = from.as_ref().as_bytes();
|
let from_as_bytes = from.as_ref().as_bytes();
|
||||||
let mut buffer = Vec::with_capacity(4 + from_as_bytes.len());
|
let mut buffer = Vec::with_capacity(4 + from_as_bytes.len());
|
||||||
buffer.extend_from_slice(&(from.as_ref().len() as u32).to_le_bytes());
|
buffer.extend_from_slice(&(u32::try_from(from.as_ref().len())?).to_le_bytes());
|
||||||
buffer.extend_from_slice(from_as_bytes);
|
buffer.extend_from_slice(from_as_bytes);
|
||||||
raw_connection.write_all(&buffer)?;
|
raw_connection.write_all(&buffer)?;
|
||||||
|
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ use std::{
|
|||||||
time::SystemTime,
|
time::SystemTime,
|
||||||
};
|
};
|
||||||
|
|
||||||
/// Internal structure wrapping a [std::io::Write] and hiding underlying protocol logic.
|
/// Internal structure wrapping a [`std::io::Write`] and hiding underlying protocol logic.
|
||||||
struct ADBSendCommandWriter<W: Write> {
|
struct ADBSendCommandWriter<W: Write> {
|
||||||
inner: W,
|
inner: W,
|
||||||
}
|
}
|
||||||
@@ -64,7 +64,7 @@ impl ADBServerDevice {
|
|||||||
// The name of the command is already sent by get_transport()?.send_sync_request
|
// The name of the command is already sent by get_transport()?.send_sync_request
|
||||||
let to_as_bytes = to.as_bytes();
|
let to_as_bytes = to.as_bytes();
|
||||||
let mut buffer = Vec::with_capacity(4 + to_as_bytes.len());
|
let mut buffer = Vec::with_capacity(4 + to_as_bytes.len());
|
||||||
buffer.extend_from_slice(&(to.len() as u32).to_le_bytes());
|
buffer.extend_from_slice(&(u32::try_from(to.len())?).to_le_bytes());
|
||||||
buffer.extend_from_slice(to_as_bytes);
|
buffer.extend_from_slice(to_as_bytes);
|
||||||
raw_connection.write_all(&buffer)?;
|
raw_connection.write_all(&buffer)?;
|
||||||
|
|
||||||
@@ -77,9 +77,10 @@ impl ADBServerDevice {
|
|||||||
|
|
||||||
// Copy is finished, we can now notify as finished
|
// Copy is finished, we can now notify as finished
|
||||||
// Have to send DONE + file mtime
|
// Have to send DONE + file mtime
|
||||||
let last_modified = match SystemTime::now().duration_since(SystemTime::UNIX_EPOCH) {
|
let Ok(last_modified) = SystemTime::now().duration_since(SystemTime::UNIX_EPOCH) else {
|
||||||
Ok(n) => n,
|
return Err(RustADBError::ADBRequestFailed(
|
||||||
Err(_) => panic!("SystemTime before UNIX EPOCH!"),
|
"SystemTime before UNIX EPOCH!".into(),
|
||||||
|
));
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut done_buffer = Vec::with_capacity(8);
|
let mut done_buffer = Vec::with_capacity(8);
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ use crate::{
|
|||||||
impl ADBServerDevice {
|
impl ADBServerDevice {
|
||||||
fn handle_stat_command<S: AsRef<str>>(&mut self, path: S) -> Result<AdbStatResponse> {
|
fn handle_stat_command<S: AsRef<str>>(&mut self, path: S) -> Result<AdbStatResponse> {
|
||||||
let mut len_buf = [0_u8; 4];
|
let mut len_buf = [0_u8; 4];
|
||||||
LittleEndian::write_u32(&mut len_buf, path.as_ref().len() as u32);
|
LittleEndian::write_u32(&mut len_buf, u32::try_from(path.as_ref().len())?);
|
||||||
|
|
||||||
// 4 bytes of command name is already sent by send_sync_request
|
// 4 bytes of command name is already sent by send_sync_request
|
||||||
self.transport.get_raw_connection()?.write_all(&len_buf)?;
|
self.transport.get_raw_connection()?.write_all(&len_buf)?;
|
||||||
@@ -31,8 +31,7 @@ impl ADBServerDevice {
|
|||||||
Ok(data.into())
|
Ok(data.into())
|
||||||
}
|
}
|
||||||
x => Err(RustADBError::UnknownResponseType(format!(
|
x => Err(RustADBError::UnknownResponseType(format!(
|
||||||
"Unknown response {}",
|
"Unknown response {x}"
|
||||||
x
|
|
||||||
))),
|
))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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 {} successfully uninstalled", package_name);
|
log::info!("Package {package_name} successfully uninstalled");
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(
|
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(
|
||||||
|
|||||||
@@ -4,8 +4,6 @@ use std::{
|
|||||||
net::{SocketAddrV4, TcpStream},
|
net::{SocketAddrV4, TcpStream},
|
||||||
};
|
};
|
||||||
|
|
||||||
use homedir::my_home;
|
|
||||||
|
|
||||||
use super::ADBTransport;
|
use super::ADBTransport;
|
||||||
use crate::{Result, RustADBError, emulator_device::ADBEmulatorCommand};
|
use crate::{Result, RustADBError, emulator_device::ADBEmulatorCommand};
|
||||||
|
|
||||||
@@ -17,7 +15,7 @@ pub struct TCPEmulatorTransport {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl TCPEmulatorTransport {
|
impl TCPEmulatorTransport {
|
||||||
/// Instantiates a new instance of [TCPEmulatorTransport]
|
/// Instantiates a new instance of [`TCPEmulatorTransport`]
|
||||||
pub fn new(socket_addr: SocketAddrV4) -> Self {
|
pub fn new(socket_addr: SocketAddrV4) -> Self {
|
||||||
Self {
|
Self {
|
||||||
socket_addr,
|
socket_addr,
|
||||||
@@ -34,11 +32,10 @@ impl TCPEmulatorTransport {
|
|||||||
)))
|
)))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return authentication token stored in $HOME/.emulator_console_auth_token
|
/// Return authentication token stored in `$HOME/.emulator_console_auth_token`
|
||||||
pub fn get_authentication_token(&mut self) -> Result<String> {
|
pub fn get_authentication_token(&mut self) -> Result<String> {
|
||||||
let home = match my_home()? {
|
let Some(home) = std::env::home_dir() else {
|
||||||
Some(home) => home,
|
return Err(RustADBError::NoHomeDirectory);
|
||||||
None => return Err(RustADBError::NoHomeDirectory),
|
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut f = File::open(home.join(".emulator_console_auth_token"))?;
|
let mut f = File::open(home.join(".emulator_console_auth_token"))?;
|
||||||
@@ -54,7 +51,7 @@ impl TCPEmulatorTransport {
|
|||||||
self.send_command(ADBEmulatorCommand::Authenticate(token))
|
self.send_command(ADBEmulatorCommand::Authenticate(token))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send an [ADBEmulatorCommand] to this emulator
|
/// Send an [`ADBEmulatorCommand`] to this emulator
|
||||||
pub(crate) fn send_command(&mut self, command: ADBEmulatorCommand) -> Result<()> {
|
pub(crate) fn send_command(&mut self, command: ADBEmulatorCommand) -> Result<()> {
|
||||||
let mut connection = self.get_raw_connection()?;
|
let mut connection = self.get_raw_connection()?;
|
||||||
|
|
||||||
@@ -80,10 +77,11 @@ impl TCPEmulatorTransport {
|
|||||||
let mut line = String::new();
|
let mut line = String::new();
|
||||||
reader.read_line(&mut line)?;
|
reader.read_line(&mut line)?;
|
||||||
|
|
||||||
match line.starts_with("OK") {
|
if line.starts_with("OK") {
|
||||||
true => Ok(()),
|
return Ok(());
|
||||||
false => Err(RustADBError::ADBRequestFailed(line)),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Err(RustADBError::ADBRequestFailed(line))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ use crate::models::{AdbRequestStatus, SyncCommand};
|
|||||||
use crate::{ADBTransport, models::AdbServerCommand};
|
use crate::{ADBTransport, models::AdbServerCommand};
|
||||||
use crate::{Result, RustADBError};
|
use crate::{Result, RustADBError};
|
||||||
|
|
||||||
const DEFAULT_SERVER_IP: Ipv4Addr = Ipv4Addr::new(127, 0, 0, 1);
|
const DEFAULT_SERVER_IP: Ipv4Addr = Ipv4Addr::LOCALHOST;
|
||||||
const DEFAULT_SERVER_PORT: u16 = 5037;
|
const DEFAULT_SERVER_PORT: u16 = 5037;
|
||||||
|
|
||||||
/// Server transport running on top on TCP
|
/// Server transport running on top on TCP
|
||||||
@@ -25,7 +25,7 @@ impl Default for TCPServerTransport {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl TCPServerTransport {
|
impl TCPServerTransport {
|
||||||
/// Instantiates a new instance of [TCPServerTransport]
|
/// Instantiates a new instance of [`TCPServerTransport`]
|
||||||
pub fn new(socket_addr: SocketAddrV4) -> Self {
|
pub fn new(socket_addr: SocketAddrV4) -> Self {
|
||||||
Self {
|
Self {
|
||||||
socket_addr,
|
socket_addr,
|
||||||
@@ -33,7 +33,7 @@ impl TCPServerTransport {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Instantiate a new instance of [TCPServerTransport] using given address, or default if not specified.
|
/// Instantiate a new instance of [`TCPServerTransport`] using given address, or default if not specified.
|
||||||
pub fn new_or_default(socket_addr: Option<SocketAddrV4>) -> Self {
|
pub fn new_or_default(socket_addr: Option<SocketAddrV4>) -> Self {
|
||||||
match socket_addr {
|
match socket_addr {
|
||||||
Some(s) => Self::new(s),
|
Some(s) => Self::new(s),
|
||||||
@@ -41,7 +41,7 @@ impl TCPServerTransport {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get underlying [SocketAddrV4]
|
/// Get underlying [`SocketAddrV4`]
|
||||||
pub fn get_socketaddr(&self) -> SocketAddrV4 {
|
pub fn get_socketaddr(&self) -> SocketAddrV4 {
|
||||||
self.socket_addr
|
self.socket_addr
|
||||||
}
|
}
|
||||||
@@ -89,7 +89,7 @@ impl TCPServerTransport {
|
|||||||
)?)
|
)?)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send the given [SyncCommand] to ADB server, and checks that the request has been taken in consideration.
|
/// Send the given [`SyncCommand`] to ADB server, and checks that the request has been taken in consideration.
|
||||||
pub(crate) fn send_sync_request(&mut self, command: SyncCommand) -> Result<()> {
|
pub(crate) fn send_sync_request(&mut self, command: SyncCommand) -> Result<()> {
|
||||||
// First 4 bytes are the name of the command we want to send
|
// First 4 bytes are the name of the command we want to send
|
||||||
// (e.g. "SEND", "RECV", "STAT", "LIST")
|
// (e.g. "SEND", "RECV", "STAT", "LIST")
|
||||||
@@ -98,22 +98,22 @@ impl TCPServerTransport {
|
|||||||
.write_all(command.to_string().as_bytes())?)
|
.write_all(command.to_string().as_bytes())?)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Gets the body length from a LittleEndian value
|
/// Gets the body length from a `LittleEndian` value
|
||||||
pub(crate) fn get_body_length(&mut self) -> Result<u32> {
|
pub(crate) fn get_body_length(&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(&mut self) -> Result<[u8; 4]> {
|
fn read_body_length(&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)?;
|
||||||
|
|
||||||
Ok(length_buffer)
|
Ok(length_buffer)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send the given [AdbCommand] to ADB server, and checks that the request has been taken in consideration.
|
/// Send the given [`AdbCommand`] to ADB server, and checks that the request has been taken in consideration.
|
||||||
/// If an error occurred, a [RustADBError] is returned with the response error string.
|
/// If an error occurred, a [`RustADBError`] is returned with the response error string.
|
||||||
pub(crate) fn send_adb_request(&mut self, command: AdbServerCommand) -> Result<()> {
|
pub(crate) fn send_adb_request(&mut self, command: AdbServerCommand) -> Result<()> {
|
||||||
let adb_command_string = command.to_string();
|
let adb_command_string = command.to_string();
|
||||||
let adb_request = format!("{:04x}{}", adb_command_string.len(), adb_command_string);
|
let adb_request = format!("{:04x}{}", adb_command_string.len(), adb_command_string);
|
||||||
|
|||||||
@@ -16,7 +16,6 @@ use std::{
|
|||||||
fs::read_to_string,
|
fs::read_to_string,
|
||||||
io::{Read, Write},
|
io::{Read, Write},
|
||||||
net::{Shutdown, SocketAddr, TcpStream},
|
net::{Shutdown, SocketAddr, TcpStream},
|
||||||
ops::{Deref, DerefMut},
|
|
||||||
path::PathBuf,
|
path::PathBuf,
|
||||||
sync::{Arc, Mutex},
|
sync::{Arc, Mutex},
|
||||||
time::Duration,
|
time::Duration,
|
||||||
@@ -109,7 +108,7 @@ impl TcpTransport {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_current_connection(&mut self) -> Result<Arc<Mutex<CurrentConnection>>> {
|
fn get_current_connection(&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(
|
||||||
@@ -120,18 +119,15 @@ impl TcpTransport {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn upgrade_connection(&mut self) -> Result<()> {
|
pub(crate) fn upgrade_connection(&mut self) -> Result<()> {
|
||||||
let current_connection = match self.current_connection.clone() {
|
let Some(current_connection) = self.current_connection.clone() else {
|
||||||
Some(current_connection) => current_connection,
|
return Err(RustADBError::UpgradeError(
|
||||||
None => {
|
"cannot upgrade a non-existing connection...".into(),
|
||||||
return Err(RustADBError::UpgradeError(
|
));
|
||||||
"cannot upgrade a non-existing connection...".into(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
{
|
{
|
||||||
let mut current_conn_locked = current_connection.lock()?;
|
let mut current_conn_locked = current_connection.lock()?;
|
||||||
match current_conn_locked.deref() {
|
match &*current_conn_locked {
|
||||||
CurrentConnection::Tcp(tcp_stream) => {
|
CurrentConnection::Tcp(tcp_stream) => {
|
||||||
// TODO: Check if we cannot be more precise
|
// TODO: Check if we cannot be more precise
|
||||||
|
|
||||||
@@ -175,8 +171,7 @@ impl TcpTransport {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
c => Err(RustADBError::ADBRequestFailed(format!(
|
c => Err(RustADBError::ADBRequestFailed(format!(
|
||||||
"Wrong command received {}",
|
"Wrong command received {c}"
|
||||||
c
|
|
||||||
))),
|
))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -193,7 +188,7 @@ impl ADBTransport for TcpTransport {
|
|||||||
log::debug!("disconnecting...");
|
log::debug!("disconnecting...");
|
||||||
if let Some(current_connection) = &self.current_connection {
|
if let Some(current_connection) = &self.current_connection {
|
||||||
let mut lock = current_connection.lock()?;
|
let mut lock = current_connection.lock()?;
|
||||||
match lock.deref_mut() {
|
match &mut *lock {
|
||||||
CurrentConnection::Tcp(tcp_stream) => {
|
CurrentConnection::Tcp(tcp_stream) => {
|
||||||
let _ = tcp_stream.shutdown(Shutdown::Both);
|
let _ = tcp_stream.shutdown(Shutdown::Both);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -29,7 +29,7 @@ pub struct USBTransport {
|
|||||||
|
|
||||||
impl USBTransport {
|
impl USBTransport {
|
||||||
/// Instantiate a new [`USBTransport`].
|
/// Instantiate a new [`USBTransport`].
|
||||||
/// Only the first device with given vendor_id and product_id is returned.
|
/// Only the first device with given `vendor_id` and `product_id` is returned.
|
||||||
pub fn new(vendor_id: u16, product_id: u16) -> Result<Self> {
|
pub fn new(vendor_id: u16, product_id: u16) -> Result<Self> {
|
||||||
for device in rusb::devices()?.iter() {
|
for device in rusb::devices()?.iter() {
|
||||||
if let Ok(descriptor) = device.device_descriptor() {
|
if let Ok(descriptor) = device.device_descriptor() {
|
||||||
@@ -40,8 +40,7 @@ impl USBTransport {
|
|||||||
}
|
}
|
||||||
|
|
||||||
Err(RustADBError::DeviceNotFound(format!(
|
Err(RustADBError::DeviceNotFound(format!(
|
||||||
"cannot find USB device with vendor_id={} and product_id={}",
|
"cannot find USB device with vendor_id={vendor_id} and product_id={product_id}",
|
||||||
vendor_id, product_id
|
|
||||||
)))
|
)))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -91,14 +90,13 @@ impl USBTransport {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn find_endpoints(&self, handle: &DeviceHandle<GlobalContext>) -> Result<(Endpoint, Endpoint)> {
|
fn find_endpoints(handle: &DeviceHandle<GlobalContext>) -> Result<(Endpoint, Endpoint)> {
|
||||||
let mut read_endpoint: Option<Endpoint> = None;
|
let mut read_endpoint: Option<Endpoint> = None;
|
||||||
let mut write_endpoint: Option<Endpoint> = None;
|
let mut write_endpoint: Option<Endpoint> = None;
|
||||||
|
|
||||||
for n in 0..handle.device().device_descriptor()?.num_configurations() {
|
for n in 0..handle.device().device_descriptor()?.num_configurations() {
|
||||||
let config_desc = match handle.device().config_descriptor(n) {
|
let Ok(config_desc) = handle.device().config_descriptor(n) else {
|
||||||
Ok(c) => c,
|
continue;
|
||||||
Err(_) => continue,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
for interface in config_desc.interfaces() {
|
for interface in config_desc.interfaces() {
|
||||||
@@ -118,16 +116,14 @@ impl USBTransport {
|
|||||||
Direction::In => {
|
Direction::In => {
|
||||||
if let Some(write_endpoint) = write_endpoint {
|
if let Some(write_endpoint) = write_endpoint {
|
||||||
return Ok((endpoint, write_endpoint));
|
return Ok((endpoint, write_endpoint));
|
||||||
} else {
|
|
||||||
read_endpoint = Some(endpoint);
|
|
||||||
}
|
}
|
||||||
|
read_endpoint = Some(endpoint);
|
||||||
}
|
}
|
||||||
Direction::Out => {
|
Direction::Out => {
|
||||||
if let Some(read_endpoint) = read_endpoint {
|
if let Some(read_endpoint) = read_endpoint {
|
||||||
return Ok((read_endpoint, endpoint));
|
return Ok((read_endpoint, endpoint));
|
||||||
} else {
|
|
||||||
write_endpoint = Some(endpoint);
|
|
||||||
}
|
}
|
||||||
|
write_endpoint = Some(endpoint);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -151,12 +147,7 @@ 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!(
|
log::trace!("wrote chunk of size {write_amount} - {offset}/{data_len}",);
|
||||||
"wrote chunk of size {} - {}/{}",
|
|
||||||
write_amount,
|
|
||||||
offset,
|
|
||||||
data_len
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if offset % max_packet_size == 0 {
|
if offset % max_packet_size == 0 {
|
||||||
@@ -172,14 +163,14 @@ impl ADBTransport for USBTransport {
|
|||||||
fn connect(&mut self) -> crate::Result<()> {
|
fn connect(&mut self) -> crate::Result<()> {
|
||||||
let device = self.device.open()?;
|
let device = self.device.open()?;
|
||||||
|
|
||||||
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));
|
||||||
@@ -197,7 +188,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}"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ 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"]
|
||||||
@@ -20,7 +21,7 @@ name = "stub_gen"
|
|||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
adb_client = { path = "../adb_client" }
|
adb_client = { path = "../adb_client" }
|
||||||
anyhow = { version = "1.0.95" }
|
anyhow = { version = "1.0.100" }
|
||||||
pyo3 = { version = "0.24.1", features = ["abi3-py37", "anyhow"] }
|
pyo3 = { version = "0.27.1", features = ["abi3-py310", "anyhow"] }
|
||||||
pyo3-stub-gen = "0.7.0"
|
pyo3-stub-gen = { version = "0.17.0" }
|
||||||
pyo3-stub-gen-derive = "0.7.0"
|
pyo3-stub-gen-derive = { version = "0.17.0" }
|
||||||
|
|||||||
@@ -47,7 +47,6 @@ usb_device.push("file.txt", "/data/local/tmp/file.txt")
|
|||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Create Python virtual environment
|
# Create Python virtual environment
|
||||||
cd pyadb_client
|
|
||||||
python3 -m venv .venv
|
python3 -m venv .venv
|
||||||
source .venv/bin/activate
|
source .venv/bin/activate
|
||||||
|
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ requires = ["maturin>=1,<2"]
|
|||||||
classifiers = ["Programming Language :: Python", "Programming Language :: Rust"]
|
classifiers = ["Programming Language :: Python", "Programming Language :: Rust"]
|
||||||
dynamic = ["authors", "keywords", "version"]
|
dynamic = ["authors", "keywords", "version"]
|
||||||
name = "pyadb_client"
|
name = "pyadb_client"
|
||||||
requires-python = ">= 3.7"
|
requires-python = ">= 3.10"
|
||||||
|
|
||||||
[tool.maturin]
|
[tool.maturin]
|
||||||
features = ["pyo3/extension-module"]
|
features = ["pyo3/extension-module"]
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ pub struct PyADBServer(ADBServer);
|
|||||||
#[pymethods]
|
#[pymethods]
|
||||||
impl PyADBServer {
|
impl PyADBServer {
|
||||||
#[new]
|
#[new]
|
||||||
/// Instantiate a new PyADBServer instance
|
/// Instantiate a new `PyADBServer` instance
|
||||||
pub fn new(address: String) -> PyResult<Self> {
|
pub fn new(address: String) -> PyResult<Self> {
|
||||||
let address = address.parse::<SocketAddrV4>()?;
|
let address = address.parse::<SocketAddrV4>()?;
|
||||||
Ok(ADBServer::new(address).into())
|
Ok(ADBServer::new(address).into())
|
||||||
@@ -24,7 +24,12 @@ impl PyADBServer {
|
|||||||
|
|
||||||
/// List available devices
|
/// List available devices
|
||||||
pub fn devices(&mut self) -> Result<Vec<PyDeviceShort>> {
|
pub fn devices(&mut self) -> Result<Vec<PyDeviceShort>> {
|
||||||
Ok(self.0.devices()?.into_iter().map(|v| v.into()).collect())
|
Ok(self
|
||||||
|
.0
|
||||||
|
.devices()?
|
||||||
|
.into_iter()
|
||||||
|
.map(std::convert::Into::into)
|
||||||
|
.collect())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get a device, assuming that only one is currently connected
|
/// Get a device, assuming that only one is currently connected
|
||||||
|
|||||||
Reference in New Issue
Block a user