release: v1.0.3

Cargo.toml

@@ -6,7 +6,7 @@ license = "MIT"
 name = "adb_client"
 readme = "README.md"
 repository = "https://github.com/cocool97/adb_client"
-version = "1.0.1"
+version = "1.0.3"
 
 [lib]
 name = "adb_client"
@@ -18,13 +18,15 @@ path = "examples/adb_cli.rs"
 
 [dependencies]
 byteorder = { version = "1.5.0" }
-chrono = { version = "0.4.37" }
-lazy_static = { version = "1.4.0" }
-regex = { version = "1.10.4", features = ["perf", "std", "unicode"] }
+chrono = { version = "0.4.38" }
+lazy_static = { version = "1.5.0" }
+mio = { version = "1.0.0", features = ["os-ext", "os-poll"] }
+regex = { version = "1.10.5", features = ["perf", "std", "unicode"] }
 termios = { version = "0.3.3" }
-thiserror = { version = "1.0.58" }
+thiserror = { version = "1.0.61" }
 
 ## Binary-only dependencies
 ## Marked as optional so that lib users do not depend on them
 [dev-dependencies]
 clap = { version = "4.5.4", features = ["derive"] }
+rand = { version = "0.8.5" }

README.md

@@ -25,7 +25,7 @@ use adb_client::AdbTcpConnection;
 use std::net::Ipv4Addr;
 
 let mut connection = AdbTcpConnection::new(Ipv4Addr::from([127,0,0,1]), 5037).unwrap();
-connection.shell_command(None, vec!["df", "-h"]);
+connection.shell_command(None, ["df", "-h"]);
 ```
 
 ### Get available ADB devices
@@ -47,8 +47,8 @@ use std::fs::File;
 use std::path::Path;
 
 let mut connection = AdbTcpConnection::new(Ipv4Addr::from([127,0,0,1]), 5037).unwrap();
-let mut input = File::open(Path::new(&filename)).unwrap(); 
-connection.send(None, &mut input, &path)?;
+let mut input = File::open(Path::new("/tmp")).unwrap();
+connection.send::<&str,&str>(None, &mut input, "/data/local/tmp");
 ```
 
 ## Rust binary

examples/adb_cli.rs

@@ -111,16 +111,16 @@ fn main() -> Result<(), RustADBError> {
         }
         Command::Pull { path, filename } => {
             let mut output = File::create(Path::new(&filename)).unwrap(); // TODO: Better error handling
-            connection.recv(opt.serial, &path, &mut output)?;
+            connection.recv(opt.serial.as_ref(), &path, &mut output)?;
             println!("Downloaded {path} as {filename}");
         }
         Command::Push { filename, path } => {
             let mut input = File::open(Path::new(&filename)).unwrap(); // TODO: Better error handling
-            connection.send(opt.serial, &mut input, &path)?;
+            connection.send(opt.serial.as_ref(), &mut input, &path)?;
             println!("Uploaded {filename} to {path}");
         }
         Command::List { path } => {
-            connection.list(opt.serial, path)?;
+            connection.list(opt.serial.as_ref(), path)?;
         }
         Command::Stat { path } => {
             let stat_response = connection.stat(opt.serial, path)?;
@@ -128,20 +128,20 @@ fn main() -> Result<(), RustADBError> {
         }
         Command::Shell { command } => {
             if command.is_empty() {
-                connection.shell(&opt.serial)?;
+                connection.shell(opt.serial.as_ref())?;
             } else {
-                connection.shell_command(&opt.serial, command)?;
+                connection.shell_command(opt.serial.as_ref(), command)?;
             }
         }
         Command::HostFeatures => {
             println!("Available host features");
-            for feature in connection.host_features(&opt.serial)? {
+            for feature in connection.host_features(opt.serial.as_ref())? {
                 println!("- {}", feature);
             }
         }
         Command::Reboot { sub_command } => {
             println!("Reboots device");
-            connection.reboot(&opt.serial, sub_command.into())?
+            connection.reboot(opt.serial.as_ref(), sub_command.into())?
         }
     }
 

src/adb_tcp_connection.rs

@@ -5,6 +5,8 @@ use std::{
     str::FromStr,
 };
 
+use byteorder::{ByteOrder, LittleEndian};
+
 use crate::{
     models::{AdbCommand, AdbRequestStatus, SyncCommand},
     Result, RustADBError,
@@ -40,11 +42,12 @@ impl AdbTcpConnection {
         &mut self,
         adb_command: AdbCommand,
         with_response: bool,
+        fresh_connection: bool
     ) -> Result<Vec<u8>> {
-        self.send_adb_request(adb_command)?;
+        self.send_adb_request(adb_command, fresh_connection)?;
 
         if with_response {
-            let length = self.get_body_length()?;
+            let length = self.get_hex_body_length()?;
             let mut body = vec![
                 0;
                 length
@@ -63,7 +66,16 @@ impl AdbTcpConnection {
 
     /// Sends 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.
-    pub(crate) fn send_adb_request(&mut self, command: AdbCommand) -> Result<()> {
+    pub(crate) fn send_adb_request(
+        &mut self,
+        command: AdbCommand,
+        fresh_connection: bool,
+    ) -> Result<()> {
+        if fresh_connection {
+            // Recreate a new connection (likely because command does not need "state" server side)
+            self.new_connection()?;
+        }
+
         let adb_command_string = command.to_string();
         let adb_request = format!("{:04x}{}", adb_command_string.len(), adb_command_string);
 
@@ -101,10 +113,23 @@ impl AdbTcpConnection {
         Ok(self.tcp_stream.write_all(command.to_string().as_bytes())?)
     }
 
-    pub(crate) fn get_body_length(&mut self) -> Result<u32> {
-        let mut length = [0; 4];
-        self.tcp_stream.read_exact(&mut length)?;
+    /// Gets the body length from hexadecimal value
+    pub(crate) fn get_hex_body_length(&mut self) -> Result<u32> {
+        let length_buffer = self.read_body_length()?;
+        Ok(u32::from_str_radix(str::from_utf8(&length_buffer)?, 16)?)
+    }
 
-        Ok(u32::from_str_radix(str::from_utf8(&length)?, 16)?)
+    /// Gets the body length from a LittleEndian value
+    pub(crate) fn get_body_length(&mut self) -> Result<u32> {
+        let length_buffer = self.read_body_length()?;
+        Ok(LittleEndian::read_u32(&length_buffer))
+    }
+
+    /// Read 4 bytes representing body length
+    fn read_body_length(&mut self) -> Result<[u8; 4]> {
+        let mut length_buffer = [0; 4];
+        self.tcp_stream.read_exact(&mut length_buffer)?;
+
+        Ok(length_buffer)
     }
 }

src/commands/devices.rs

@@ -5,7 +5,7 @@ use crate::{models::AdbCommand, AdbTcpConnection, Device, DeviceLong, Result, Ru
 impl AdbTcpConnection {
     /// Gets a list of connected devices.
     pub fn devices(&mut self) -> Result<Vec<Device>> {
-        let devices = self.proxy_connection(AdbCommand::Devices, true)?;
+        let devices = self.proxy_connection(AdbCommand::Devices, true, true)?;
 
         let mut vec_devices: Vec<Device> = vec![];
         for device in devices.split(|x| x.eq(&b'\n')) {
@@ -21,7 +21,7 @@ impl AdbTcpConnection {
 
     /// Gets an extended list of connected devices including the device paths in the state.
     pub fn devices_long(&mut self) -> Result<Vec<DeviceLong>> {
-        let devices_long = self.proxy_connection(AdbCommand::DevicesLong, true)?;
+        let devices_long = self.proxy_connection(AdbCommand::DevicesLong, true, true)?;
 
         let mut vec_devices: Vec<DeviceLong> = vec![];
         for device in devices_long.split(|x| x.eq(&b'\n')) {
@@ -38,10 +38,10 @@ impl AdbTcpConnection {
     /// Tracks new devices showing up.
     // TODO: Change with Generator when feature stabilizes
     pub fn track_devices(&mut self, callback: impl Fn(Device) -> Result<()>) -> Result<()> {
-        self.send_adb_request(AdbCommand::TrackDevices)?;
+        self.send_adb_request(AdbCommand::TrackDevices, true)?;
 
         loop {
-            let length = self.get_body_length()?;
+            let length = self.get_hex_body_length()?;
 
             if length > 0 {
                 let mut body = vec![

src/commands/host_features.rs

@@ -5,15 +5,15 @@ use crate::{
 
 impl AdbTcpConnection {
     /// Lists available ADB server features.
-    pub fn host_features<S: ToString>(&mut self, serial: &Option<S>) -> Result<Vec<HostFeatures>> {
+    pub fn host_features<S: ToString>(&mut self, serial: Option<&S>) -> Result<Vec<HostFeatures>> {
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, true)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), true)?
             }
         }
 
-        let features = self.proxy_connection(AdbCommand::HostFeatures, true)?;
+        let features = self.proxy_connection(AdbCommand::HostFeatures, true, false)?;
 
         Ok(features
             .split(|x| x.eq(&b','))

src/commands/kill.rs

@@ -3,6 +3,6 @@ use crate::{models::AdbCommand, AdbTcpConnection, Result};
 impl AdbTcpConnection {
     /// Asks the ADB server to quit immediately.
     pub fn kill(&mut self) -> Result<()> {
-        self.proxy_connection(AdbCommand::Kill, false).map(|_| ())
+        self.proxy_connection(AdbCommand::Kill, false, true).map(|_| ())
     }
 }

src/commands/list.rs

@@ -10,21 +10,19 @@ use std::{
 
 impl AdbTcpConnection {
     /// Lists files in [path] on the device.
-    pub fn list<S: ToString, A: AsRef<str>>(&mut self, serial: Option<S>, path: A) -> Result<()> {
-        self.new_connection()?;
-
+    pub fn list<S: ToString, A: AsRef<str>>(&mut self, serial: Option<&S>, path: A) -> Result<()> {
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, false)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), false)?
             }
         }
 
         // Set device in SYNC mode
-        self.send_adb_request(AdbCommand::Sync)?;
+        self.send_adb_request(AdbCommand::Sync, false)?;
 
         // Send a list command
-        self.send_sync_request(SyncCommand::List(path.as_ref()))?;
+        self.send_sync_request(SyncCommand::List)?;
 
         self.handle_list_command(path)
     }

src/commands/reboot.rs

@@ -7,17 +7,17 @@ impl AdbTcpConnection {
     /// Reboots the device
     pub fn reboot<S: ToString>(
         &mut self,
-        serial: &Option<S>,
+        serial: Option<&S>,
         reboot_type: RebootType,
     ) -> Result<()> {
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, true)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), true)?
             }
         }
 
-        self.proxy_connection(AdbCommand::Reboot(reboot_type), false)
+        self.proxy_connection(AdbCommand::Reboot(reboot_type), false, false)
             .map(|_| ())
     }
 }

src/commands/recv.rs

@@ -9,24 +9,22 @@ impl AdbTcpConnection {
     /// Receives [path] to [stream] from the device.
     pub fn recv<S: ToString, A: AsRef<str>>(
         &mut self,
-        serial: Option<S>,
+        serial: Option<&S>,
         path: A,
         stream: &mut dyn Write,
     ) -> Result<()> {
-        self.new_connection()?;
-
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, true)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), true)?
             }
         }
 
         // Set device in SYNC mode
-        self.send_adb_request(AdbCommand::Sync)?;
+        self.send_adb_request(AdbCommand::Sync, false)?;
 
         // Send a recv command
-        self.send_sync_request(SyncCommand::Recv(path.as_ref(), stream))?;
+        self.send_sync_request(SyncCommand::Recv)?;
 
         self.handle_recv_command(path, stream)
     }
@@ -46,28 +44,26 @@ impl AdbTcpConnection {
         // Chunk looks like 'DATA' <length> <data>
         let mut buffer = [0_u8; 64 * 1024]; // Should this be Boxed?
         let mut data_header = [0_u8; 4]; // DATA
-        let mut len_header = [0_u8; 4]; // <len>
         loop {
             self.tcp_stream.read_exact(&mut data_header)?;
-            // Check if data_header is DATA or DONE
-            if data_header.eq(b"DATA") {
-                self.tcp_stream.read_exact(&mut len_header)?;
-                let length: usize = LittleEndian::read_u32(&len_header).try_into().unwrap();
-                self.tcp_stream.read_exact(&mut buffer[..length])?;
-                output.write_all(&buffer)?;
-            } else if data_header.eq(b"DONE") {
-                // We're done here
-                break;
-            } else if data_header.eq(b"FAIL") {
-                // Handle fail
-                self.tcp_stream.read_exact(&mut len_header)?;
-                let length: usize = LittleEndian::read_u32(&len_header).try_into().unwrap();
-                self.tcp_stream.read_exact(&mut buffer[..length])?;
-                Err(RustADBError::ADBRequestFailed(String::from_utf8(
-                    buffer[..length].to_vec(),
-                )?))?;
-            } else {
-                panic!("Unknown response from device {:#?}", data_header);
+            // Check if data_header is DATA or DONE or FAIL
+            match &data_header {
+                b"DATA" => {
+                    // Handle received data
+                    let length: usize = self.get_body_length()?.try_into().unwrap();
+                    self.tcp_stream.read_exact(&mut buffer[..length])?;
+                    output.write_all(&buffer[..length])?;
+                }
+                b"DONE" => break, // We're done here
+                b"FAIL" => {
+                    // Handle fail
+                    let length: usize = self.get_body_length()?.try_into().unwrap();
+                    self.tcp_stream.read_exact(&mut buffer[..length])?;
+                    Err(RustADBError::ADBRequestFailed(String::from_utf8(
+                        buffer[..length].to_vec(),
+                    )?))?;
+                }
+                _ => panic!("Unknown response from device {:#?}", data_header),
             }
         }
 

src/commands/send.rs

@@ -14,24 +14,22 @@ impl AdbTcpConnection {
     /// Sends [stream] to [path] on the device.
     pub fn send<S: ToString, A: AsRef<str>>(
         &mut self,
-        serial: Option<S>,
+        serial: Option<&S>,
         stream: &mut dyn Read,
         path: A,
     ) -> Result<()> {
-        self.new_connection()?;
-
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, true)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), true)?
             }
         }
 
         // Set device in SYNC mode
-        self.send_adb_request(AdbCommand::Sync)?;
+        self.send_adb_request(AdbCommand::Sync, false)?;
 
         // Send a send command
-        self.send_sync_request(SyncCommand::Send(stream, path.as_ref()))?;
+        self.send_sync_request(SyncCommand::Send)?;
 
         self.handle_send_command(stream, path)
     }

src/commands/shell.rs

@@ -1,4 +1,10 @@
-use std::io::{ErrorKind, Read, Write};
+use std::{
+    io::{self, Read, Write},
+    sync::mpsc,
+    time::Duration,
+};
+
+use mio::{unix::SourceFd, Events, Interest, Poll, Token};
 
 use crate::{
     adb_termios::ADBTermios,
@@ -6,11 +12,24 @@ use crate::{
     AdbTcpConnection, Result, RustADBError,
 };
 
+const STDIN: Token = Token(0);
+const BUFFER_SIZE: usize = 512;
+const POLL_DURATION: Duration = Duration::from_millis(100);
+
+fn setup_poll_stdin() -> std::result::Result<Poll, io::Error> {
+    let poll = Poll::new()?;
+    let stdin_fd = 0;
+    poll.registry()
+        .register(&mut SourceFd(&stdin_fd), STDIN, Interest::READABLE)?;
+
+    Ok(poll)
+}
+
 impl AdbTcpConnection {
     /// Runs 'command' in a shell on the device, and return its output and error streams.
     pub fn shell_command<S: ToString>(
         &mut self,
-        serial: &Option<S>,
+        serial: Option<&S>,
         command: impl IntoIterator<Item = S>,
     ) -> Result<Vec<u8>> {
         let supported_features = self.host_features(serial)?;
@@ -20,24 +39,25 @@ impl AdbTcpConnection {
             return Err(RustADBError::ADBShellNotSupported);
         }
 
-        self.new_connection()?;
-
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, true)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), true)?
             }
         }
-        self.send_adb_request(AdbCommand::ShellCommand(
-            command
-                .into_iter()
-                .map(|v| v.to_string())
-                .collect::<Vec<_>>()
-                .join(" "),
-        ))?;
+        self.send_adb_request(
+            AdbCommand::ShellCommand(
+                command
+                    .into_iter()
+                    .map(|v| v.to_string())
+                    .collect::<Vec<_>>()
+                    .join(" "),
+            ),
+            false,
+        )?;
 
         const BUFFER_SIZE: usize = 512;
-        let result = (|| {
+        (|| {
             let mut result = Vec::new();
             loop {
                 let mut buffer = [0; BUFFER_SIZE];
@@ -54,20 +74,17 @@ impl AdbTcpConnection {
                     }
                 }
             }
-        })();
-
-        self.new_connection()?;
-        result
+        })()
     }
 
     /// Starts an interactive shell session on the device. Redirects stdin/stdout/stderr as appropriate.
-    pub fn shell<S: ToString>(&mut self, serial: &Option<S>) -> Result<()> {
+    pub fn shell<S: ToString>(&mut self, serial: Option<&S>) -> Result<()> {
         let mut adb_termios = ADBTermios::new(std::io::stdin())?;
         adb_termios.set_adb_termios()?;
 
         self.tcp_stream.set_nodelay(true)?;
 
-        // FORWARD CTRL+C !!
+        // TODO: FORWARD CTRL+C !!
 
         let supported_features = self.host_features(serial)?;
         if !supported_features.contains(&HostFeatures::ShellV2)
@@ -76,37 +93,29 @@ impl AdbTcpConnection {
             return Err(RustADBError::ADBShellNotSupported);
         }
 
-        self.new_connection()?;
-
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, true)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), true)?
             }
         }
-        self.send_adb_request(AdbCommand::Shell)?;
+        self.send_adb_request(AdbCommand::Shell, false)?;
 
         // let read_stream = Arc::new(self.tcp_stream);
         let mut read_stream = self.tcp_stream.try_clone()?;
 
-        // Writing thread
-        let mut write_stream = read_stream.try_clone()?;
-        let writer_t = std::thread::spawn(move || -> Result<()> {
-            let mut buf = [0; 1024];
-            loop {
-                let size = std::io::stdin().read(&mut buf)?;
+        let (tx, rx) = mpsc::channel::<bool>();
 
-                write_stream.write_all(&buf[0..size])?;
-            }
-        });
+        let mut write_stream = read_stream.try_clone()?;
 
         // Reading thread
-        let reader_t = std::thread::spawn(move || -> Result<()> {
-            const BUFFER_SIZE: usize = 512;
+        std::thread::spawn(move || -> Result<()> {
             loop {
                 let mut buffer = [0; BUFFER_SIZE];
                 match read_stream.read(&mut buffer) {
                     Ok(0) => {
+                        let _ = tx.send(true);
+                        read_stream.shutdown(std::net::Shutdown::Both)?;
                         return Ok(());
                     }
                     Ok(size) => {
@@ -120,24 +129,33 @@ impl AdbTcpConnection {
             }
         });
 
-        if let Err(e) = reader_t.join().unwrap() {
-            match e {
-                RustADBError::IOError(e) if e.kind() == ErrorKind::BrokenPipe => {}
-                _ => {
-                    return Err(e);
+        let mut buf = [0; BUFFER_SIZE];
+        let mut events = Events::with_capacity(1);
+
+        let mut poll = setup_poll_stdin()?;
+
+        // Polling either by checking that reading socket hasn't been closed, and if is there is something to read on stdin.
+        loop {
+            poll.poll(&mut events, Some(POLL_DURATION))?;
+            match rx.try_recv() {
+                Ok(_) | Err(mpsc::TryRecvError::Disconnected) => return Ok(()),
+                Err(mpsc::TryRecvError::Empty) => (),
+            }
+
+            for event in events.iter() {
+                match event.token() {
+                    STDIN => {
+                        let size = match std::io::stdin().read(&mut buf) {
+                            Ok(0) => return Ok(()),
+                            Ok(size) => size,
+                            Err(_) => return Ok(()),
+                        };
+
+                        write_stream.write_all(&buf[0..size])?;
+                    }
+                    _ => unreachable!(),
                 }
             }
         }
-
-        if let Err(e) = writer_t.join().unwrap() {
-            match e {
-                RustADBError::IOError(e) if e.kind() == ErrorKind::BrokenPipe => {}
-                _ => {
-                    return Err(e);
-                }
-            }
-        }
-
-        Ok(())
     }
 }

src/commands/stat.rs

@@ -79,20 +79,18 @@ impl AdbTcpConnection {
         serial: Option<S>,
         path: A,
     ) -> Result<AdbStatResponse> {
-        self.new_connection()?;
-
         match serial {
-            None => self.send_adb_request(AdbCommand::TransportAny)?,
+            None => self.send_adb_request(AdbCommand::TransportAny, false)?,
             Some(serial) => {
-                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()))?
+                self.send_adb_request(AdbCommand::TransportSerial(serial.to_string()), false)?
             }
         }
 
         // Set device in SYNC mode
-        self.send_adb_request(AdbCommand::Sync)?;
+        self.send_adb_request(AdbCommand::Sync, false)?;
 
         // Send a "Stat" command
-        self.send_sync_request(SyncCommand::Stat(path.as_ref()))?;
+        self.send_sync_request(SyncCommand::Stat)?;
 
         self.handle_stat_command(path)
     }

src/commands/transport.rs

@@ -3,7 +3,7 @@ use crate::{models::AdbCommand, AdbTcpConnection, Result};
 impl AdbTcpConnection {
     /// Asks ADB server to switch the connection to either the device or emulator connect to/running on the host. Will fail if there is more than one such device/emulator available.
     pub fn transport_any(&mut self) -> Result<()> {
-        self.proxy_connection(AdbCommand::TransportAny, false)
+        self.proxy_connection(AdbCommand::TransportAny, false, true)
             .map(|_| ())
     }
 }

src/commands/version.rs

@@ -3,7 +3,7 @@ use crate::{models::AdbCommand, AdbTcpConnection, AdbVersion, Result};
 impl AdbTcpConnection {
     /// Gets server's internal version number.
     pub fn version(&mut self) -> Result<AdbVersion> {
-        let version = self.proxy_connection(AdbCommand::Version, true)?;
+        let version = self.proxy_connection(AdbCommand::Version, true, true)?;
 
         AdbVersion::try_from(version)
     }

src/models/adb_command.rs

@@ -1,3 +1,5 @@
+use std::fmt::Display;
+
 use super::RebootType;
 
 pub enum AdbCommand {
@@ -45,28 +47,28 @@ pub enum AdbCommand {
     Reboot(RebootType),
 }
 
-impl ToString for AdbCommand {
-    fn to_string(&self) -> String {
+impl Display for AdbCommand {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
-            AdbCommand::Version => "host:version".into(),
-            AdbCommand::Kill => "host:kill".into(),
-            AdbCommand::Devices => "host:devices".into(),
-            AdbCommand::DevicesLong => "host:devices-l".into(),
-            AdbCommand::Sync => "sync:".into(),
-            AdbCommand::TrackDevices => "host:track-devices".into(),
-            AdbCommand::TransportAny => "host:transport-any".into(),
-            AdbCommand::TransportSerial(serial) => format!("host:transport:{serial}"),
+            AdbCommand::Version => write!(f, "host:version"),
+            AdbCommand::Kill => write!(f, "host:kill"),
+            AdbCommand::Devices => write!(f, "host:devices"),
+            AdbCommand::DevicesLong => write!(f, "host:devices-l"),
+            AdbCommand::Sync => write!(f, "sync:"),
+            AdbCommand::TrackDevices => write!(f, "host:track-devices"),
+            AdbCommand::TransportAny => write!(f, "host:transport-any"),
+            AdbCommand::TransportSerial(serial) => write!(f, "host:transport:{serial}"),
             AdbCommand::ShellCommand(command) => match std::env::var("TERM") {
-                Ok(term) => format!("shell,TERM={term},raw:{command}"),
-                Err(_) => format!("shell,raw:{command}"),
+                Ok(term) => write!(f, "shell,TERM={term},raw:{command}"),
+                Err(_) => write!(f, "shell,raw:{command}"),
             },
             AdbCommand::Shell => match std::env::var("TERM") {
-                Ok(term) => format!("shell,TERM={term},raw:"),
-                Err(_) => "shell,raw:".into(),
+                Ok(term) => write!(f, "shell,TERM={term},raw:"),
+                Err(_) => write!(f, "shell,raw:"),
             },
-            AdbCommand::HostFeatures => "host:features".into(),
+            AdbCommand::HostFeatures => write!(f, "host:features"),
             AdbCommand::Reboot(reboot_type) => {
-                format!("reboot:{reboot_type}")
+                write!(f, "reboot:{reboot_type}")
             }
         }
     }

src/models/sync_command.rs

@@ -1,22 +1,23 @@
-pub enum SyncCommand<'a> {
+use std::fmt::Display;
+
+pub enum SyncCommand {
     /// List files in a folder
-    List(&'a str),
+    List,
     /// Receive a file from the device
-    Recv(&'a str, &'a mut dyn std::io::Write),
+    Recv,
     /// Send a file to the device
-    Send(&'a mut dyn std::io::Read, &'a str),
+    Send,
     // Stat a file
-    Stat(&'a str),
+    Stat,
 }
 
-impl ToString for SyncCommand<'_> {
-    fn to_string(&self) -> String {
+impl Display for SyncCommand {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
-            SyncCommand::List(_) => "LIST",
-            SyncCommand::Recv(_, _) => "RECV",
-            SyncCommand::Send(_, _) => "SEND",
-            SyncCommand::Stat(_) => "STAT",
+            SyncCommand::List => write!(f, "LIST"),
+            SyncCommand::Recv => write!(f, "RECV"),
+            SyncCommand::Send => write!(f, "SEND"),
+            SyncCommand::Stat => write!(f, "STAT"),
         }
-        .to_string()
     }
 }

tests/tests.rs

@@ -1,9 +1,11 @@
 #[cfg(test)]
 mod tests {
+    use std::io::Cursor;
     use std::net::Ipv4Addr;
     use std::str::FromStr;
 
     use adb_client::AdbTcpConnection;
+    use rand::Rng;
 
     fn new_client() -> AdbTcpConnection {
         let address = Ipv4Addr::from_str("127.0.0.1").unwrap();
@@ -19,8 +21,8 @@ mod tests {
     #[test]
     fn test_shell() {
         let mut adb = new_client();
-        adb.shell_command(&None, vec!["ls"]).unwrap();
-        adb.shell_command(&None, vec!["pwd"]).unwrap();
+        adb.shell_command(None, vec!["ls"]).unwrap();
+        adb.shell_command(None, vec!["pwd"]).unwrap();
     }
 
     #[test]
@@ -41,4 +43,42 @@ mod tests {
         let address = Ipv4Addr::from_str("127.0.0.300").unwrap();
         let _ = AdbTcpConnection::new(address, 5037).expect("Could not create ADB connection...");
     }
+
+    #[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 connection = new_client();
+
+        const TEST_FILENAME: &'static str = "/data/local/tmp/test_file";
+        // Send it
+        connection
+            .send::<&str, &str>(None, &mut c, TEST_FILENAME)
+            .expect("cannot send file");
+
+        // Pull it to memory
+        let mut res = vec![];
+        connection
+            .recv::<&str, &str>(None, TEST_FILENAME, &mut res)
+            .expect("cannot recv file");
+
+        // diff
+        assert_eq!(c.get_ref(), &res);
+
+        connection
+            .shell_command::<&str>(None, [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");
+        }
+    }
 }