Compare commits
2 Commits
v2.1.17
...
133-list-a
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c0085ab93b | ||
|
|
66fa9679a4 |
@@ -16,6 +16,7 @@ anyhow = { version = "1.0.94" }
|
|||||||
clap = { version = "4.5.23", features = ["derive"] }
|
clap = { version = "4.5.23", features = ["derive"] }
|
||||||
env_logger = { version = "0.11.5" }
|
env_logger = { version = "0.11.5" }
|
||||||
log = { version = "0.4.26" }
|
log = { version = "0.4.26" }
|
||||||
|
tabwriter = { version = "1.4.1" }
|
||||||
|
|
||||||
[target.'cfg(unix)'.dependencies]
|
[target.'cfg(unix)'.dependencies]
|
||||||
termios = { version = "0.3.3" }
|
termios = { version = "0.3.3" }
|
||||||
|
|||||||
@@ -8,20 +8,22 @@ mod models;
|
|||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use adb_client::{
|
use adb_client::{
|
||||||
ADBDeviceExt, ADBServer, ADBServerDevice, ADBTcpDevice, ADBUSBDevice, MDNSDiscoveryService,
|
ADBDeviceExt, ADBDeviceInfo, ADBServer, ADBServerDevice, ADBTcpDevice, ADBUSBDevice,
|
||||||
|
MDNSDiscoveryService, find_all_connected_adb_devices,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[cfg(any(target_os = "linux", target_os = "macos"))]
|
#[cfg(any(target_os = "linux", target_os = "macos"))]
|
||||||
use adb_termios::ADBTermios;
|
use adb_termios::ADBTermios;
|
||||||
|
|
||||||
use anyhow::Result;
|
use anyhow::{Result, bail};
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use handlers::{handle_emulator_commands, handle_host_commands, handle_local_commands};
|
use handlers::{handle_emulator_commands, handle_host_commands, handle_local_commands};
|
||||||
use models::{DeviceCommands, LocalCommand, MainCommand, Opts};
|
use models::{DeviceCommands, LocalCommand, MainCommand, Opts};
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
use std::io::Write;
|
use std::io::{Write, stdout};
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
|
use tabwriter::TabWriter;
|
||||||
use utils::setup_logger;
|
use utils::setup_logger;
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
@@ -61,6 +63,34 @@ fn main() -> Result<()> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
MainCommand::Usb(usb_command) => {
|
MainCommand::Usb(usb_command) => {
|
||||||
|
if usb_command.list_devices {
|
||||||
|
let devices = find_all_connected_adb_devices()?;
|
||||||
|
|
||||||
|
let mut writer = TabWriter::new(stdout()).alignment(tabwriter::Alignment::Center);
|
||||||
|
writeln!(writer, "Index\tVendor ID\tProduct ID\tDevice Description")?;
|
||||||
|
writeln!(writer, "-----\t---------\t----------\t----------------")?;
|
||||||
|
|
||||||
|
for (
|
||||||
|
index,
|
||||||
|
ADBDeviceInfo {
|
||||||
|
vendor_id,
|
||||||
|
product_id,
|
||||||
|
device_description,
|
||||||
|
},
|
||||||
|
) in devices.iter().enumerate()
|
||||||
|
{
|
||||||
|
writeln!(
|
||||||
|
writer,
|
||||||
|
"#{}\t{:04x}\t{:04x}\t{}",
|
||||||
|
index, vendor_id, product_id, device_description
|
||||||
|
)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
writer.flush()?;
|
||||||
|
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
let device = match (usb_command.vendor_id, usb_command.product_id) {
|
let device = match (usb_command.vendor_id, usb_command.product_id) {
|
||||||
(Some(vid), Some(pid)) => match usb_command.path_to_private_key {
|
(Some(vid), Some(pid)) => match usb_command.path_to_private_key {
|
||||||
Some(pk) => ADBUSBDevice::new_with_custom_private_key(vid, pid, pk)?,
|
Some(pk) => ADBUSBDevice::new_with_custom_private_key(vid, pid, pk)?,
|
||||||
@@ -76,7 +106,11 @@ fn main() -> Result<()> {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
(device.boxed(), usb_command.commands)
|
|
||||||
|
match usb_command.commands {
|
||||||
|
Some(commands) => (device.boxed(), commands),
|
||||||
|
None => bail!("no command provided"),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
MainCommand::Tcp(tcp_command) => {
|
MainCommand::Tcp(tcp_command) => {
|
||||||
let device = ADBTcpDevice::new(tcp_command.address)?;
|
let device = ADBTcpDevice::new(tcp_command.address)?;
|
||||||
|
|||||||
@@ -20,6 +20,9 @@ pub struct UsbCommand {
|
|||||||
/// Path to a custom private key to use for authentication
|
/// Path to a custom private key to use for authentication
|
||||||
#[clap(short = 'k', long = "private-key")]
|
#[clap(short = 'k', long = "private-key")]
|
||||||
pub path_to_private_key: Option<PathBuf>,
|
pub path_to_private_key: Option<PathBuf>,
|
||||||
|
/// List all connected Android devices
|
||||||
|
#[clap(short = 'l', long = "list")]
|
||||||
|
pub list_devices: bool,
|
||||||
#[clap(subcommand)]
|
#[clap(subcommand)]
|
||||||
pub commands: DeviceCommands,
|
pub commands: Option<DeviceCommands>,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -35,28 +35,80 @@ pub fn read_adb_private_key<P: AsRef<Path>>(private_key_path: P) -> Result<Optio
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Search for adb devices with known interface class and subclass values
|
/// Represents an Android device connected via USB
|
||||||
pub fn search_adb_devices() -> Result<Option<(u16, u16)>> {
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct ADBDeviceInfo {
|
||||||
|
/// Vendor ID of the device
|
||||||
|
pub vendor_id: u16,
|
||||||
|
/// Product ID of the device
|
||||||
|
pub product_id: u16,
|
||||||
|
/// Textual description of the device
|
||||||
|
pub device_description: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Find and return a list of all connected Android devices with known interface class and subclass values
|
||||||
|
pub fn find_all_connected_adb_devices() -> Result<Vec<ADBDeviceInfo>> {
|
||||||
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 {
|
||||||
continue;
|
continue;
|
||||||
};
|
};
|
||||||
|
|
||||||
if is_adb_device(&device, &des) {
|
if is_adb_device(&device, &des) {
|
||||||
log::debug!(
|
let device_handle = match device.open() {
|
||||||
"Autodetect device {:04x}:{:04x}",
|
Ok(h) => h,
|
||||||
des.vendor_id(),
|
Err(_) => {
|
||||||
des.product_id()
|
found_devices.push(ADBDeviceInfo {
|
||||||
);
|
vendor_id: des.vendor_id(),
|
||||||
found_devices.push((des.vendor_id(), des.product_id()));
|
product_id: des.product_id(),
|
||||||
|
device_description: "Unknown device".to_string(),
|
||||||
|
});
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let manufacturer = device_handle
|
||||||
|
.read_manufacturer_string_ascii(&des)
|
||||||
|
.unwrap_or_else(|_| "Unknown".to_string());
|
||||||
|
|
||||||
|
let product = device_handle
|
||||||
|
.read_product_string_ascii(&des)
|
||||||
|
.unwrap_or_else(|_| "Unknown".to_string());
|
||||||
|
|
||||||
|
let device_description = format!("{} {}", manufacturer, product);
|
||||||
|
|
||||||
|
found_devices.push(ADBDeviceInfo {
|
||||||
|
vendor_id: des.vendor_id(),
|
||||||
|
product_id: des.product_id(),
|
||||||
|
device_description,
|
||||||
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Ok(found_devices)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Find and return an USB-connected Android device with known interface class and subclass values.
|
||||||
|
///
|
||||||
|
/// Returns the first device found or None if no device is found.
|
||||||
|
/// If multiple devices are found, an error is returned.
|
||||||
|
pub fn get_single_connected_adb_device() -> Result<Option<ADBDeviceInfo>> {
|
||||||
|
let found_devices = find_all_connected_adb_devices()?;
|
||||||
match (found_devices.first(), found_devices.get(1)) {
|
match (found_devices.first(), found_devices.get(1)) {
|
||||||
(None, _) => Ok(None),
|
(None, _) => Ok(None),
|
||||||
(Some(identifiers), None) => Ok(Some(*identifiers)),
|
(Some(device_info), None) => {
|
||||||
(Some((vid1, pid1)), Some((vid2, pid2))) => Err(RustADBError::DeviceNotFound(format!(
|
log::debug!(
|
||||||
"Found two Android devices {vid1:04x}:{pid1:04x} and {vid2:04x}:{pid2:04x}",
|
"Autodetect device {:04x}:{:04x} - {}",
|
||||||
|
device_info.vendor_id,
|
||||||
|
device_info.product_id,
|
||||||
|
device_info.device_description
|
||||||
|
);
|
||||||
|
Ok(Some(device_info.clone()))
|
||||||
|
}
|
||||||
|
(Some(device_1), Some(device_2)) => Err(RustADBError::DeviceNotFound(format!(
|
||||||
|
"Found two Android devices {:04x}:{:04x} and {:04x}:{:04x}",
|
||||||
|
device_1.vendor_id, device_1.product_id, device_2.vendor_id, device_2.product_id
|
||||||
))),
|
))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -167,10 +219,12 @@ impl ADBUSBDevice {
|
|||||||
|
|
||||||
/// autodetect connected ADB devices and establish a connection with the first device found using a custom private key path
|
/// autodetect connected ADB devices and establish a connection with the first device found using a custom private key path
|
||||||
pub fn autodetect_with_custom_private_key(private_key_path: PathBuf) -> Result<Self> {
|
pub fn autodetect_with_custom_private_key(private_key_path: PathBuf) -> Result<Self> {
|
||||||
match search_adb_devices()? {
|
match get_single_connected_adb_device()? {
|
||||||
Some((vendor_id, product_id)) => {
|
Some(device_info) => ADBUSBDevice::new_with_custom_private_key(
|
||||||
ADBUSBDevice::new_with_custom_private_key(vendor_id, product_id, private_key_path)
|
device_info.vendor_id,
|
||||||
}
|
device_info.product_id,
|
||||||
|
private_key_path,
|
||||||
|
),
|
||||||
_ => Err(RustADBError::DeviceNotFound(
|
_ => Err(RustADBError::DeviceNotFound(
|
||||||
"cannot find USB devices matching the signature of an ADB device".into(),
|
"cannot find USB devices matching the signature of an ADB device".into(),
|
||||||
)),
|
)),
|
||||||
|
|||||||
@@ -12,7 +12,8 @@ use adb_message_device::ADBMessageDevice;
|
|||||||
pub use adb_tcp_device::ADBTcpDevice;
|
pub use adb_tcp_device::ADBTcpDevice;
|
||||||
pub use adb_transport_message::{ADBTransportMessage, ADBTransportMessageHeader};
|
pub use adb_transport_message::{ADBTransportMessage, ADBTransportMessageHeader};
|
||||||
pub use adb_usb_device::{
|
pub use adb_usb_device::{
|
||||||
ADBUSBDevice, get_default_adb_key_path, is_adb_device, search_adb_devices,
|
ADBDeviceInfo, ADBUSBDevice, find_all_connected_adb_devices, get_default_adb_key_path,
|
||||||
|
get_single_connected_adb_device, is_adb_device,
|
||||||
};
|
};
|
||||||
pub use message_writer::MessageWriter;
|
pub use message_writer::MessageWriter;
|
||||||
pub use models::{ADBRsaKey, MessageCommand, MessageSubcommand};
|
pub use models::{ADBRsaKey, MessageCommand, MessageSubcommand};
|
||||||
|
|||||||
@@ -17,7 +17,10 @@ mod transports;
|
|||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
pub use adb_device_ext::ADBDeviceExt;
|
pub use adb_device_ext::ADBDeviceExt;
|
||||||
pub use device::{ADBTcpDevice, ADBUSBDevice, is_adb_device, search_adb_devices};
|
pub use device::{
|
||||||
|
ADBDeviceInfo, ADBTcpDevice, ADBUSBDevice, find_all_connected_adb_devices,
|
||||||
|
get_single_connected_adb_device, is_adb_device,
|
||||||
|
};
|
||||||
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::*;
|
||||||
|
|||||||
Reference in New Issue
Block a user