Compare commits
7 Commits
108-instal
...
up-session
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
849f7807a9 | ||
|
|
39a7f0a8cf | ||
|
|
4129d9d218 | ||
|
|
c23dca61d7 | ||
|
|
2bc338fdf5 | ||
|
|
728fb7da95 | ||
|
|
2e762400e5 |
@@ -9,7 +9,7 @@ homepage = "https://github.com/cocool97/adb_client"
|
||||
keywords = ["adb", "android", "tcp", "usb"]
|
||||
license = "MIT"
|
||||
repository = "https://github.com/cocool97/adb_client"
|
||||
version = "2.1.12"
|
||||
version = "2.1.13"
|
||||
|
||||
# To build locally when working on a new release
|
||||
[patch.crates-io]
|
||||
|
||||
@@ -8,6 +8,7 @@ pub enum RebootTypeCommand {
|
||||
Recovery,
|
||||
Sideload,
|
||||
SideloadAutoReboot,
|
||||
Fastboot,
|
||||
}
|
||||
|
||||
impl From<RebootTypeCommand> for RebootType {
|
||||
@@ -18,6 +19,7 @@ impl From<RebootTypeCommand> for RebootType {
|
||||
RebootTypeCommand::Recovery => RebootType::Recovery,
|
||||
RebootTypeCommand::Sideload => RebootType::Sideload,
|
||||
RebootTypeCommand::SideloadAutoReboot => RebootType::SideloadAutoReboot,
|
||||
RebootTypeCommand::Fastboot => RebootType::Fastboot,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -13,22 +13,26 @@ version.workspace = true
|
||||
base64 = { version = "0.22.1" }
|
||||
bincode = { version = "1.3.3" }
|
||||
byteorder = { version = "1.5.0" }
|
||||
chrono = { version = "0.4.40" }
|
||||
chrono = { version = "0.4.40", default-features = false, features = ["std"] }
|
||||
homedir = { version = "0.3.4" }
|
||||
image = { version = "0.25.5" }
|
||||
lazy_static = { version = "1.5.0" }
|
||||
image = { version = "0.25.5", default-features = false }
|
||||
log = { version = "0.4.26" }
|
||||
mdns-sd = { version = "0.13.2" }
|
||||
mdns-sd = { version = "0.13.9", default-features = false, features = [
|
||||
"logging",
|
||||
] }
|
||||
num-bigint = { version = "0.8.4", package = "num-bigint-dig" }
|
||||
num-traits = { version = "0.2.19" }
|
||||
quick-protobuf = { version = "0.8.1" }
|
||||
rand = { version = "0.9.0" }
|
||||
rcgen = { version = "0.13.1" }
|
||||
rcgen = { version = "0.13.1", default-features = false, features = [
|
||||
"aws_lc_rs",
|
||||
"pem",
|
||||
] }
|
||||
regex = { version = "1.11.1", features = ["perf", "std", "unicode"] }
|
||||
rsa = { version = "0.9.7" }
|
||||
rusb = { version = "0.9.4", features = ["vendored"] }
|
||||
rustls = { version = "0.23.22" }
|
||||
rustls-pki-types = "1.11.0"
|
||||
rustls = { version = "0.23.27" }
|
||||
rustls-pki-types = { version = "1.11.0" }
|
||||
serde = { version = "1.0.216", features = ["derive"] }
|
||||
serde_repr = { version = "0.1.19" }
|
||||
sha1 = { version = "0.10.6", features = ["oid"] }
|
||||
@@ -36,7 +40,7 @@ thiserror = { version = "2.0.7" }
|
||||
|
||||
[dev-dependencies]
|
||||
anyhow = { version = "1.0.93" }
|
||||
criterion = { version = "0.5.1" } # Used for benchmarks
|
||||
criterion = { version = "0.6.0" } # Used for benchmarks
|
||||
|
||||
[[bench]]
|
||||
harness = false
|
||||
|
||||
@@ -11,18 +11,18 @@ use super::{ADBTransportMessage, MessageCommand, models::MessageSubcommand};
|
||||
#[derive(Debug)]
|
||||
pub struct ADBMessageDevice<T: ADBMessageTransport> {
|
||||
transport: T,
|
||||
local_id: Option<u32>,
|
||||
remote_id: Option<u32>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Copy)]
|
||||
pub struct ADBSession {
|
||||
pub local_id: u32,
|
||||
pub remote_id: u32,
|
||||
}
|
||||
|
||||
impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
/// Instantiate a new [`ADBMessageTransport`]
|
||||
pub fn new(transport: T) -> Self {
|
||||
Self {
|
||||
transport,
|
||||
local_id: None,
|
||||
remote_id: None,
|
||||
}
|
||||
Self { transport }
|
||||
}
|
||||
|
||||
pub(crate) fn get_transport(&mut self) -> &T {
|
||||
@@ -34,12 +34,15 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
}
|
||||
|
||||
/// Receive a message and acknowledge it by replying with an `OKAY` command
|
||||
pub(crate) fn recv_and_reply_okay(&mut self) -> Result<ADBTransportMessage> {
|
||||
pub(crate) fn recv_and_reply_okay(
|
||||
&mut self,
|
||||
session: ADBSession,
|
||||
) -> Result<ADBTransportMessage> {
|
||||
let message = self.transport.read_message()?;
|
||||
self.transport.write_message(ADBTransportMessage::new(
|
||||
MessageCommand::Okay,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&[],
|
||||
))?;
|
||||
Ok(message)
|
||||
@@ -60,11 +63,12 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
pub(crate) fn recv_file<W: std::io::Write>(
|
||||
&mut self,
|
||||
session: ADBSession,
|
||||
mut output: W,
|
||||
) -> std::result::Result<(), RustADBError> {
|
||||
let mut len: Option<u64> = None;
|
||||
loop {
|
||||
let payload = self.recv_and_reply_okay()?.into_payload();
|
||||
let payload = self.recv_and_reply_okay(session)?.into_payload();
|
||||
let mut rdr = Cursor::new(&payload);
|
||||
while rdr.position() != payload.len() as u64 {
|
||||
match len.take() {
|
||||
@@ -97,8 +101,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
pub(crate) fn push_file<R: std::io::Read>(
|
||||
&mut self,
|
||||
local_id: u32,
|
||||
remote_id: u32,
|
||||
session: ADBSession,
|
||||
mut reader: R,
|
||||
) -> std::result::Result<(), RustADBError> {
|
||||
let mut buffer = [0; BUFFER_SIZE];
|
||||
@@ -111,8 +114,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
let message = ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
local_id,
|
||||
remote_id,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&serialized_message,
|
||||
);
|
||||
|
||||
@@ -131,8 +134,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
let message = ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
local_id,
|
||||
remote_id,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&serialized_message,
|
||||
);
|
||||
|
||||
@@ -159,8 +162,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
let message = ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
local_id,
|
||||
remote_id,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&serialized_message,
|
||||
);
|
||||
|
||||
@@ -173,24 +176,27 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn begin_synchronization(&mut self) -> Result<()> {
|
||||
self.open_session(b"sync:\0")?;
|
||||
Ok(())
|
||||
pub(crate) fn begin_synchronization(&mut self) -> Result<ADBSession> {
|
||||
self.open_session(b"sync:\0")
|
||||
}
|
||||
|
||||
pub(crate) fn stat_with_explicit_ids(&mut self, remote_path: &str) -> Result<AdbStatResponse> {
|
||||
pub(crate) fn stat_with_explicit_ids(
|
||||
&mut self,
|
||||
session: ADBSession,
|
||||
remote_path: &str,
|
||||
) -> Result<AdbStatResponse> {
|
||||
let stat_buffer = MessageSubcommand::Stat.with_arg(remote_path.len() as u32);
|
||||
let message = ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&bincode::serialize(&stat_buffer).map_err(|_e| RustADBError::ConversionError)?,
|
||||
);
|
||||
self.send_and_expect_okay(message)?;
|
||||
self.send_and_expect_okay(ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
remote_path.as_bytes(),
|
||||
))?;
|
||||
let response = self.transport.read_message()?;
|
||||
@@ -200,24 +206,25 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
.map_err(|_e| RustADBError::ConversionError)
|
||||
}
|
||||
|
||||
pub(crate) fn end_transaction(&mut self) -> Result<()> {
|
||||
pub(crate) fn end_transaction(&mut self, session: ADBSession) -> Result<()> {
|
||||
let quit_buffer = MessageSubcommand::Quit.with_arg(0u32);
|
||||
self.send_and_expect_okay(ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&bincode::serialize(&quit_buffer).map_err(|_e| RustADBError::ConversionError)?,
|
||||
))?;
|
||||
let _discard_close = self.transport.read_message()?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub(crate) fn open_session(&mut self, data: &[u8]) -> Result<ADBTransportMessage> {
|
||||
pub(crate) fn open_session(&mut self, data: &[u8]) -> Result<ADBSession> {
|
||||
let mut rng = rand::rng();
|
||||
let local_id: u32 = rng.random();
|
||||
|
||||
let message = ADBTransportMessage::new(
|
||||
MessageCommand::Open,
|
||||
rng.random(), // Our 'local-id'
|
||||
local_id, // Our 'local-id'
|
||||
0,
|
||||
data,
|
||||
);
|
||||
@@ -225,21 +232,25 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
let response = self.get_transport_mut().read_message()?;
|
||||
|
||||
self.local_id = Some(response.header().arg1());
|
||||
self.remote_id = Some(response.header().arg0());
|
||||
if response.header().command() != MessageCommand::Okay {
|
||||
return Err(RustADBError::ADBRequestFailed(format!(
|
||||
"Open session failed: got {} in respone instead of OKAY",
|
||||
response.header().command()
|
||||
)));
|
||||
}
|
||||
|
||||
Ok(response)
|
||||
}
|
||||
if response.header().arg1() != local_id {
|
||||
return Err(RustADBError::ADBRequestFailed(format!(
|
||||
"Open session failed: respones used {} for our local_id instead of {local_id}",
|
||||
response.header().arg1()
|
||||
)));
|
||||
}
|
||||
|
||||
pub(crate) fn get_local_id(&self) -> Result<u32> {
|
||||
self.local_id.ok_or(RustADBError::ADBRequestFailed(
|
||||
"connection not opened, no local_id".into(),
|
||||
))
|
||||
}
|
||||
let session = ADBSession {
|
||||
local_id,
|
||||
remote_id: response.header().arg0(),
|
||||
};
|
||||
|
||||
pub(crate) fn get_remote_id(&self) -> Result<u32> {
|
||||
self.remote_id.ok_or(RustADBError::ADBRequestFailed(
|
||||
"connection not opened, no remote_id".into(),
|
||||
))
|
||||
Ok(session)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -11,9 +11,9 @@ use crate::{
|
||||
|
||||
impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
pub(crate) fn framebuffer_inner(&mut self) -> Result<ImageBuffer<Rgba<u8>, Vec<u8>>> {
|
||||
self.open_session(b"framebuffer:\0")?;
|
||||
let session = self.open_session(b"framebuffer:\0")?;
|
||||
|
||||
let response = self.recv_and_reply_okay()?;
|
||||
let response = self.recv_and_reply_okay(session)?;
|
||||
|
||||
let mut payload_cursor = Cursor::new(response.payload());
|
||||
|
||||
@@ -36,7 +36,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
break;
|
||||
}
|
||||
|
||||
let response = self.recv_and_reply_okay()?;
|
||||
let response = self.recv_and_reply_okay(session)?;
|
||||
|
||||
framebuffer_data.extend_from_slice(&response.into_payload());
|
||||
|
||||
@@ -69,7 +69,7 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
break;
|
||||
}
|
||||
|
||||
let response = self.recv_and_reply_okay()?;
|
||||
let response = self.recv_and_reply_okay(session)?;
|
||||
|
||||
framebuffer_data.extend_from_slice(&response.into_payload());
|
||||
|
||||
|
||||
@@ -2,7 +2,7 @@ use std::{fs::File, path::Path};
|
||||
|
||||
use crate::{
|
||||
ADBMessageTransport, Result,
|
||||
device::{MessageCommand, MessageWriter, adb_message_device::ADBMessageDevice},
|
||||
device::{MessageWriter, adb_message_device::ADBMessageDevice},
|
||||
utils::check_extension_is_apk,
|
||||
};
|
||||
|
||||
@@ -14,11 +14,12 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
let file_size = apk_file.metadata()?.len();
|
||||
|
||||
self.open_session(format!("exec:cmd package 'install' -S {}\0", file_size).as_bytes())?;
|
||||
let session =
|
||||
self.open_session(format!("exec:cmd package 'install' -S {}\0", file_size).as_bytes())?;
|
||||
|
||||
let transport = self.get_transport().clone();
|
||||
|
||||
let mut writer = MessageWriter::new(transport, self.get_local_id()?, self.get_remote_id()?);
|
||||
let mut writer = MessageWriter::new(transport, session.local_id, session.remote_id);
|
||||
|
||||
std::io::copy(&mut apk_file, &mut writer)?;
|
||||
|
||||
@@ -30,9 +31,6 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
"APK file {} successfully installed",
|
||||
apk_path.as_ref().display()
|
||||
);
|
||||
self.get_transport_mut()
|
||||
.read_message()?
|
||||
.assert_command(MessageCommand::Clse)?;
|
||||
Ok(())
|
||||
}
|
||||
d => Err(crate::RustADBError::ADBRequestFailed(String::from_utf8(
|
||||
|
||||
@@ -10,10 +10,10 @@ use crate::{
|
||||
|
||||
impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
pub(crate) fn pull<A: AsRef<str>, W: Write>(&mut self, source: A, output: W) -> Result<()> {
|
||||
self.begin_synchronization()?;
|
||||
let session = self.begin_synchronization()?;
|
||||
let source = source.as_ref();
|
||||
|
||||
let adb_stat_response = self.stat_with_explicit_ids(source)?;
|
||||
let adb_stat_response = self.stat_with_explicit_ids(session, source)?;
|
||||
|
||||
if adb_stat_response.file_perm == 0 {
|
||||
return Err(RustADBError::UnknownResponseType(
|
||||
@@ -21,11 +21,13 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
));
|
||||
}
|
||||
|
||||
let local_id = self.get_local_id()?;
|
||||
let remote_id = self.get_remote_id()?;
|
||||
|
||||
self.get_transport_mut().write_message_with_timeout(
|
||||
ADBTransportMessage::new(MessageCommand::Okay, local_id, remote_id, &[]),
|
||||
ADBTransportMessage::new(
|
||||
MessageCommand::Okay,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&[],
|
||||
),
|
||||
std::time::Duration::from_secs(4),
|
||||
)?;
|
||||
|
||||
@@ -34,19 +36,19 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
bincode::serialize(&recv_buffer).map_err(|_e| RustADBError::ConversionError)?;
|
||||
self.send_and_expect_okay(ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&recv_buffer,
|
||||
))?;
|
||||
self.send_and_expect_okay(ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
source.as_bytes(),
|
||||
))?;
|
||||
|
||||
self.recv_file(output)?;
|
||||
self.end_transaction()?;
|
||||
self.recv_file(session, output)?;
|
||||
self.end_transaction(session)?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
@@ -10,7 +10,7 @@ use crate::{
|
||||
|
||||
impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
pub(crate) fn push<R: Read, A: AsRef<str>>(&mut self, stream: R, path: A) -> Result<()> {
|
||||
self.begin_synchronization()?;
|
||||
let session = self.begin_synchronization()?;
|
||||
|
||||
let path_header = format!("{},0777", path.as_ref());
|
||||
|
||||
@@ -21,14 +21,13 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
|
||||
self.send_and_expect_okay(ADBTransportMessage::new(
|
||||
MessageCommand::Write,
|
||||
self.get_local_id()?,
|
||||
self.get_remote_id()?,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&send_buffer,
|
||||
))?;
|
||||
|
||||
self.push_file(self.get_local_id()?, self.get_remote_id()?, stream)?;
|
||||
|
||||
self.end_transaction()?;
|
||||
self.push_file(session, stream)?;
|
||||
self.end_transaction(session)?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -10,14 +10,7 @@ use crate::{
|
||||
impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
/// Runs 'command' in a shell on the device, and write its output and error streams into output.
|
||||
pub(crate) fn shell_command(&mut self, command: &[&str], output: &mut dyn Write) -> Result<()> {
|
||||
let response = self.open_session(format!("shell:{}\0", command.join(" "),).as_bytes())?;
|
||||
|
||||
if response.header().command() != MessageCommand::Okay {
|
||||
return Err(RustADBError::ADBRequestFailed(format!(
|
||||
"wrong command {}",
|
||||
response.header().command()
|
||||
)));
|
||||
}
|
||||
let session = self.open_session(format!("shell:{}\0", command.join(" "),).as_bytes())?;
|
||||
|
||||
loop {
|
||||
let response = self.get_transport_mut().read_message()?;
|
||||
@@ -38,21 +31,22 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
mut reader: &mut dyn Read,
|
||||
mut writer: Box<(dyn Write + Send)>,
|
||||
) -> Result<()> {
|
||||
self.open_session(b"shell:\0")?;
|
||||
let session = self.open_session(b"shell:\0")?;
|
||||
|
||||
let mut transport = self.get_transport().clone();
|
||||
|
||||
let local_id = self.get_local_id()?;
|
||||
let remote_id = self.get_remote_id()?;
|
||||
|
||||
// Reading thread, reads response from adbd
|
||||
std::thread::spawn(move || -> Result<()> {
|
||||
loop {
|
||||
let message = transport.read_message()?;
|
||||
|
||||
// Acknowledge for more data
|
||||
let response =
|
||||
ADBTransportMessage::new(MessageCommand::Okay, local_id, remote_id, &[]);
|
||||
let response = ADBTransportMessage::new(
|
||||
MessageCommand::Okay,
|
||||
session.local_id,
|
||||
session.remote_id,
|
||||
&[],
|
||||
);
|
||||
transport.write_message(response)?;
|
||||
|
||||
match message.header().command() {
|
||||
@@ -67,7 +61,8 @@ impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
});
|
||||
|
||||
let transport = self.get_transport().clone();
|
||||
let mut shell_writer = ShellMessageWriter::new(transport, local_id, remote_id);
|
||||
let mut shell_writer =
|
||||
ShellMessageWriter::new(transport, session.local_id, session.remote_id);
|
||||
|
||||
// Read from given reader (that could be stdin e.g), and write content to device adbd
|
||||
if let Err(e) = std::io::copy(&mut reader, &mut shell_writer) {
|
||||
|
||||
@@ -4,9 +4,9 @@ use crate::{
|
||||
|
||||
impl<T: ADBMessageTransport> ADBMessageDevice<T> {
|
||||
pub(crate) fn stat(&mut self, remote_path: &str) -> Result<AdbStatResponse> {
|
||||
self.begin_synchronization()?;
|
||||
let adb_stat_response = self.stat_with_explicit_ids(remote_path)?;
|
||||
self.end_transaction()?;
|
||||
let session = self.begin_synchronization()?;
|
||||
let adb_stat_response = self.stat_with_explicit_ids(session, remote_path)?;
|
||||
self.end_transaction(session)?;
|
||||
Ok(adb_stat_response)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -13,6 +13,8 @@ pub enum RebootType {
|
||||
Sideload,
|
||||
/// Same as `Sideload` but reboots after sideloading
|
||||
SideloadAutoReboot,
|
||||
/// Reboots to fastboot
|
||||
Fastboot,
|
||||
}
|
||||
|
||||
impl Display for RebootType {
|
||||
@@ -23,6 +25,7 @@ impl Display for RebootType {
|
||||
RebootType::Recovery => write!(f, "recovery"),
|
||||
RebootType::Sideload => write!(f, "sideload"),
|
||||
RebootType::SideloadAutoReboot => write!(f, "sideload-auto-reboot"),
|
||||
RebootType::Fastboot => write!(f, "fastboot"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -36,7 +36,7 @@ impl ADBServer {
|
||||
break;
|
||||
}
|
||||
|
||||
vec_devices.push(DeviceLong::try_from(device.to_vec())?);
|
||||
vec_devices.push(DeviceLong::try_from(device)?);
|
||||
}
|
||||
|
||||
Ok(vec_devices)
|
||||
|
||||
@@ -6,7 +6,7 @@ use crate::{DeviceState, RustADBError};
|
||||
use regex::bytes::Regex;
|
||||
|
||||
static DEVICES_LONG_REGEX: LazyLock<Regex> = LazyLock::new(|| {
|
||||
Regex::new(r"^(?P<identifier>\S+)\s+(?P<state>\w+)\s+(usb:(?P<usb1>\S+)|(?P<usb2>\S+))?\s*(product:(?P<product>\w+)\s+model:(?P<model>\w+)\s+device:(?P<device>\w+)\s+)?transport_id:(?P<transport_id>\d+)$").expect("cannot build devices long regex")
|
||||
Regex::new(r"^(?P<identifier>\S+)\s+(?P<state>\w+)\s+(usb:(?P<usb1>\S+)|(?P<usb2>\S+))?\s*(product:(?P<product>\S+)\s+model:(?P<model>\w+)\s+device:(?P<device>\S+)\s+)?transport_id:(?P<transport_id>\d+)$").expect("cannot build devices long regex")
|
||||
});
|
||||
|
||||
/// Represents a new device with more informations.
|
||||
@@ -44,12 +44,12 @@ impl Display for DeviceLong {
|
||||
}
|
||||
}
|
||||
|
||||
impl TryFrom<Vec<u8>> for DeviceLong {
|
||||
impl TryFrom<&[u8]> for DeviceLong {
|
||||
type Error = RustADBError;
|
||||
|
||||
fn try_from(value: Vec<u8>) -> Result<Self, Self::Error> {
|
||||
fn try_from(value: &[u8]) -> Result<Self, Self::Error> {
|
||||
let groups = DEVICES_LONG_REGEX
|
||||
.captures(&value)
|
||||
.captures(value)
|
||||
.ok_or(RustADBError::RegexParsingError)?;
|
||||
|
||||
Ok(DeviceLong {
|
||||
@@ -98,3 +98,17 @@ impl TryFrom<Vec<u8>> for DeviceLong {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_static_devices_long() {
|
||||
let inputs = [
|
||||
"7a5158f05122195aa device 1-5 product:gts210vewifixx model:SM_T813 device:gts210vewifi transport_id:4",
|
||||
"n311r05e device usb:0-1.5 product:alioth model:M2012K11AC device:alioth transport_id:58",
|
||||
"192.168.100.192:5555 device product:alioth model:M2012K11AC device:alioth transport_id:97",
|
||||
"emulator-5554 device product:sdk_gphone64_arm64 model:sdk_gphone64_arm64 device:emu64a transport_id:101",
|
||||
"QQ20131020250511 device 20-4 product:NOH-AN00 model:NOH_AN00 device:HWNOH transport_id:3",
|
||||
];
|
||||
for input in inputs {
|
||||
DeviceLong::try_from(input.as_bytes()).expect(&format!("cannot parse input: '{input}'"));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -21,6 +21,6 @@ name = "stub_gen"
|
||||
[dependencies]
|
||||
adb_client = { path = "../adb_client" }
|
||||
anyhow = { version = "1.0.95" }
|
||||
pyo3 = { version = "0.24.1", features = ["abi3-py37", "anyhow"] }
|
||||
pyo3 = { version = "0.25.0", features = ["abi3-py37", "anyhow"] }
|
||||
pyo3-stub-gen = "0.7.0"
|
||||
pyo3-stub-gen-derive = "0.7.0"
|
||||
|
||||
@@ -47,7 +47,6 @@ usb_device.push("file.txt", "/data/local/tmp/file.txt")
|
||||
|
||||
```bash
|
||||
# Create Python virtual environment
|
||||
cd pyadb_client
|
||||
python3 -m venv .venv
|
||||
source .venv/bin/activate
|
||||
|
||||
|
||||
101
tests/tests.rs
101
tests/tests.rs
@@ -1,101 +0,0 @@
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use std::io::Cursor;
|
||||
|
||||
use adb_client::{ADBServer, ADBServerDevice, DeviceLong};
|
||||
use rand::Rng;
|
||||
|
||||
fn new_client() -> ADBServer {
|
||||
ADBServer::default()
|
||||
}
|
||||
|
||||
fn new_device() -> ADBServerDevice {
|
||||
let mut client = new_client();
|
||||
return client.get_device().expect("cannot get device");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_version() {
|
||||
let mut adb = new_client();
|
||||
adb.version().expect("cannot get adb version");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_shell_commands() {
|
||||
let mut device = new_device();
|
||||
|
||||
device.shell_command(["ls"]).expect("error while executing `ls` command");
|
||||
device.shell_command(["pwd"]).expect("error while executing `pwd` command");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_devices() {
|
||||
let mut adb = new_client();
|
||||
adb.devices().expect("cannot list devices");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_devices_long() {
|
||||
let mut adb = new_client();
|
||||
adb.devices_long().expect("cannot list devices long");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_static_devices_long() {
|
||||
let inputs = ["7a5158f05122195aa device 1-5 product:gts210vewifixx model:SM_T813 device:gts210vewifi transport_id:4",
|
||||
"n311r05e device usb:0-1.5 product:alioth model:M2012K11AC device:alioth transport_id:58",
|
||||
"192.168.100.192:5555 device product:alioth model:M2012K11AC device:alioth transport_id:97",
|
||||
"emulator-5554 device product:sdk_gphone64_arm64 model:sdk_gphone64_arm64 device:emu64a transport_id:101"];
|
||||
for input in inputs {
|
||||
DeviceLong::try_from(input.as_bytes().to_vec())
|
||||
.expect(&format!("cannot parse input: '{input}'"));
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_send_recv() {
|
||||
// Create random "Reader" in memory
|
||||
let mut key = [0u8; 1000];
|
||||
rand::thread_rng().fill(&mut key[..]);
|
||||
let mut c: Cursor<Vec<u8>> = Cursor::new(key.to_vec());
|
||||
|
||||
let mut device = new_device();
|
||||
|
||||
const TEST_FILENAME: &'static str = "/data/local/tmp/test_file";
|
||||
// Send it
|
||||
device
|
||||
.send(&mut c, TEST_FILENAME)
|
||||
.expect("cannot send file");
|
||||
|
||||
// Pull it to memory
|
||||
let mut res = vec![];
|
||||
device
|
||||
.recv(TEST_FILENAME, &mut res)
|
||||
.expect("cannot recv file");
|
||||
|
||||
// diff
|
||||
assert_eq!(c.get_ref(), &res);
|
||||
|
||||
device
|
||||
.shell_command::<&str>([format!("rm {TEST_FILENAME}").as_str()])
|
||||
.expect("cannot remove test file");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn multiple_connexions() {
|
||||
let mut connection = new_client();
|
||||
|
||||
for _ in 0..2 {
|
||||
let _ = connection.devices().expect("cannot get version");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn command_emulator() {
|
||||
let mut connection = new_client();
|
||||
let mut emulator = connection
|
||||
.get_emulator_device()
|
||||
.expect("no emulator running");
|
||||
emulator.hello().expect("cannot hello");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user