Move everything into features

tools/src/heartbeat_client.rs

@@ -0,0 +1,96 @@
+// Jackson Coxson
+// Heartbeat client
+
+use idevice::{
+    heartbeat::HeartbeatClient,
+    lockdownd::{self, LockdowndClient},
+    pairing_file::PairingFile,
+    Idevice,
+};
+
+use std::{
+    net::{Ipv4Addr, SocketAddrV4},
+    str::FromStr,
+};
+
+#[tokio::main]
+async fn main() {
+    env_logger::init();
+    let mut host = None;
+    let mut pairing_file = None;
+
+    // Loop through args
+    let mut i = 0;
+    while i < std::env::args().len() {
+        match std::env::args().nth(i).unwrap().as_str() {
+            "--host" => {
+                host = Some(std::env::args().nth(i + 1).unwrap().to_string());
+                i += 2;
+            }
+            "--pairing-file" => {
+                pairing_file = Some(std::env::args().nth(i + 1).unwrap().to_string());
+                i += 2;
+            }
+            "-h" | "--help" => {
+                println!("ideviceinfo - get information from the idevice");
+                println!("Usage:");
+                println!("  ideviceinfo [options]");
+                println!("Options:");
+                println!("  --host <host>");
+                println!("  --pairing_file <path>");
+                println!("  -h, --help");
+                println!("  --about");
+                println!("\n\nSet RUST_LOG to info, debug, warn, error, or trace to see more logs. Default is error.");
+                std::process::exit(0);
+            }
+            "--about" => {
+                println!("ideviceinfo - get information from the idevice. Reimplementation of libimobiledevice's binary.");
+                println!("Copyright (c) 2025 Jackson Coxson");
+            }
+            _ => {
+                i += 1;
+            }
+        }
+    }
+    if host.is_none() {
+        println!("Invalid arguments! Pass the IP of the device with --host");
+        return;
+    }
+    if pairing_file.is_none() {
+        println!("Invalid arguments! Pass the path the the pairing file with --pairing-file");
+        return;
+    }
+    let ip = Ipv4Addr::from_str(host.unwrap().as_str()).unwrap();
+    let socket = SocketAddrV4::new(ip, lockdownd::LOCKDOWND_PORT);
+
+    let socket = tokio::net::TcpStream::connect(socket).await.unwrap();
+    let socket = Box::new(socket);
+    let idevice = Idevice::new(socket, "heartbeat_client");
+
+    let p = PairingFile::read_from_file(pairing_file.as_ref().unwrap()).unwrap();
+
+    let mut lockdown_client = LockdowndClient { idevice };
+    lockdown_client.start_session(&p).await.unwrap();
+
+    let (port, _) = lockdown_client
+        .start_service("com.apple.mobile.heartbeat")
+        .await
+        .unwrap();
+
+    let socket = SocketAddrV4::new(ip, port);
+    let socket = tokio::net::TcpStream::connect(socket).await.unwrap();
+    let socket = Box::new(socket);
+    let mut idevice = Idevice::new(socket, "heartbeat_client");
+
+    let p = PairingFile::read_from_file(pairing_file.unwrap()).unwrap();
+
+    idevice.start_session(&p).await.unwrap();
+
+    let mut heartbeat_client = HeartbeatClient { idevice };
+
+    let mut interval = 15;
+    loop {
+        interval = heartbeat_client.get_marco(interval).await.unwrap();
+        heartbeat_client.send_polo().await.unwrap();
+    }
+}