diff --git a/.gitignore b/.gitignore index ad67955..0728338 100644 --- a/.gitignore +++ b/.gitignore @@ -19,3 +19,8 @@ target # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. #.idea/ + + +# Added by cargo + +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..36ebfc4 --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,533 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 4 + +[[package]] +name = "addr2line" +version = "0.24.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "dfbe277e56a376000877090da837660b4427aad530e3028d44e0bffe4f89a1c1" +dependencies = [ + "gimli", +] + +[[package]] +name = "adler2" +version = "2.0.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "320119579fcad9c21884f5c4861d16174d0e06250625266f50fe6898340abefa" + +[[package]] +name = "anstream" +version = "0.6.19" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "301af1932e46185686725e0fad2f8f2aa7da69dd70bf6ecc44d6b703844a3933" +dependencies = [ + "anstyle", + "anstyle-parse", + "anstyle-query", + "anstyle-wincon", + "colorchoice", + "is_terminal_polyfill", + "utf8parse", +] + +[[package]] +name = "anstyle" +version = "1.0.11" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "862ed96ca487e809f1c8e5a8447f6ee2cf102f846893800b20cebdf541fc6bbd" + +[[package]] +name = "anstyle-parse" +version = "0.2.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4e7644824f0aa2c7b9384579234ef10eb7efb6a0deb83f9630a49594dd9c15c2" +dependencies = [ + "utf8parse", +] + +[[package]] +name = "anstyle-query" +version = "1.1.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6c8bdeb6047d8983be085bab0ba1472e6dc604e7041dbf6fcd5e71523014fae9" +dependencies = [ + "windows-sys 0.59.0", +] + +[[package]] +name = "anstyle-wincon" +version = "3.0.9" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "403f75924867bb1033c59fbf0797484329750cfbe3c4325cd33127941fabc882" +dependencies = [ + "anstyle", + "once_cell_polyfill", + "windows-sys 0.59.0", +] + +[[package]] +name = "autocfg" +version = "1.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8" + +[[package]] +name = "backtrace" +version = "0.3.75" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6806a6321ec58106fea15becdad98371e28d92ccbc7c8f1b3b6dd724fe8f1002" +dependencies = [ + "addr2line", + "cfg-if", + "libc", + "miniz_oxide", + "object", + "rustc-demangle", + "windows-targets", +] + +[[package]] +name = "bitflags" +version = "2.9.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1b8e56985ec62d17e9c1001dc89c88ecd7dc08e47eba5ec7c29c7b5eeecde967" + +[[package]] +name = "bytes" +version = "1.10.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "d71b6127be86fdcfddb610f7182ac57211d4b18a3e9c82eb2d17662f2227ad6a" + +[[package]] +name = "cfg-if" +version = "1.0.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9555578bc9e57714c812a1f84e4fc5b4d21fcb063490c624de019f7464c91268" + +[[package]] +name = "clap" +version = "4.5.41" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "be92d32e80243a54711e5d7ce823c35c41c9d929dc4ab58e1276f625841aadf9" +dependencies = [ + "clap_builder", + "clap_derive", +] + +[[package]] +name = "clap_builder" +version = "4.5.41" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "707eab41e9622f9139419d573eca0900137718000c517d47da73045f54331c3d" +dependencies = [ + "anstream", + "anstyle", + "clap_lex", + "strsim", +] + +[[package]] +name = "clap_derive" +version = "4.5.41" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ef4f52386a59ca4c860f7393bcf8abd8dfd91ecccc0f774635ff68e92eeef491" +dependencies = [ + "heck", + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "clap_lex" +version = "0.7.5" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b94f61472cee1439c0b966b47e3aca9ae07e45d070759512cd390ea2bebc6675" + +[[package]] +name = "colorchoice" +version = "1.0.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b05b61dc5112cbb17e4b6cd61790d9845d13888356391624cbe7e41efeac1e75" + +[[package]] +name = "colored" +version = "3.0.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "fde0e0ec90c9dfb3b4b1a0891a7dcd0e2bffde2f7efed5fe7c9bb00e5bfb915e" +dependencies = [ + "windows-sys 0.59.0", +] + +[[package]] +name = "gimli" +version = "0.31.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "07e28edb80900c19c28f1072f2e8aeca7fa06b23cd4169cefe1af5aa3260783f" + +[[package]] +name = "heck" +version = "0.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea" + +[[package]] +name = "io-uring" +version = "0.7.8" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b86e202f00093dcba4275d4636b93ef9dd75d025ae560d2521b45ea28ab49013" +dependencies = [ + "bitflags", + "cfg-if", + "libc", +] + +[[package]] +name = "is_terminal_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf" + +[[package]] +name = "libc" +version = "0.2.174" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1171693293099992e19cddea4e8b849964e9846f4acee11b3948bcc337be8776" + +[[package]] +name = "lock_api" +version = "0.4.13" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "96936507f153605bddfcda068dd804796c84324ed2510809e5b2a624c81da765" +dependencies = [ + "autocfg", + "scopeguard", +] + +[[package]] +name = "memchr" +version = "2.7.5" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a282da65faaf38286cf3be983213fcf1d2e2a58700e808f83f4ea9a4804bc0" + +[[package]] +name = "miniz_oxide" +version = "0.8.9" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1fa76a2c86f704bdb222d66965fb3d63269ce38518b83cb0575fca855ebb6316" +dependencies = [ + "adler2", +] + +[[package]] +name = "mio" +version = "1.0.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "78bed444cc8a2160f01cbcf811ef18cac863ad68ae8ca62092e8db51d51c761c" +dependencies = [ + "libc", + "wasi", + "windows-sys 0.59.0", +] + +[[package]] +name = "object" +version = "0.36.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "62948e14d923ea95ea2c7c86c71013138b66525b86bdc08d2dcc262bdb497b87" +dependencies = [ + "memchr", +] + +[[package]] +name = "once_cell_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "a4895175b425cb1f87721b59f0f286c2092bd4af812243672510e1ac53e2e0ad" + +[[package]] +name = "parking_lot" +version = "0.12.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "70d58bf43669b5795d1576d0641cfb6fbb2057bf629506267a92807158584a13" +dependencies = [ + "lock_api", + "parking_lot_core", +] + +[[package]] +name = "parking_lot_core" +version = "0.9.11" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "bc838d2a56b5b1a6c25f55575dfc605fabb63bb2365f6c2353ef9159aa69e4a5" +dependencies = [ + "cfg-if", + "libc", + "redox_syscall", + "smallvec", + "windows-targets", +] + +[[package]] +name = "pin-project-lite" +version = "0.2.16" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b3cff922bd51709b605d9ead9aa71031d81447142d828eb4a6eba76fe619f9b" + +[[package]] +name = "proc-macro2" +version = "1.0.95" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "02b3e5e68a3a1a02aad3ec490a98007cbc13c37cbe84a3cd7b8e406d76e7f778" +dependencies = [ + "unicode-ident", +] + +[[package]] +name = "quote" +version = "1.0.40" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1885c039570dc00dcb4ff087a89e185fd56bae234ddc7f056a945bf36467248d" +dependencies = [ + "proc-macro2", +] + +[[package]] +name = "redox_syscall" +version = "0.5.13" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0d04b7d0ee6b4a0207a0a7adb104d23ecb0b47d6beae7152d0fa34b692b29fd6" +dependencies = [ + "bitflags", +] + +[[package]] +name = "rustc-demangle" +version = "0.1.25" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "989e6739f80c4ad5b13e0fd7fe89531180375b18520cc8c82080e4dc4035b84f" + +[[package]] +name = "same-file" +version = "1.0.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "93fc1dc3aaa9bfed95e02e6eadabb4baf7e3078b0bd1b4d7b6b0b68378900502" +dependencies = [ + "winapi-util", +] + +[[package]] +name = "scopeguard" +version = "1.2.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49" + +[[package]] +name = "signal-hook-registry" +version = "1.4.5" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9203b8055f63a2a00e2f593bb0510367fe707d7ff1e5c872de2f537b339e5410" +dependencies = [ + "libc", +] + +[[package]] +name = "slab" +version = "0.4.10" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "04dc19736151f35336d325007ac991178d504a119863a2fcb3758cdb5e52c50d" + +[[package]] +name = "smallvec" +version = "1.15.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03" + +[[package]] +name = "snaferrous" +version = "0.1.0" +dependencies = [ + "clap", + "colored", + "tokio", + "walkdir", +] + +[[package]] +name = "socket2" +version = "0.5.10" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "e22376abed350d73dd1cd119b57ffccad95b4e585a7cda43e286245ce23c0678" +dependencies = [ + "libc", + "windows-sys 0.52.0", +] + +[[package]] +name = "strsim" +version = "0.11.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" + +[[package]] +name = "syn" +version = "2.0.104" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "17b6f705963418cdb9927482fa304bc562ece2fdd4f616084c50b7023b435a40" +dependencies = [ + "proc-macro2", + "quote", + "unicode-ident", +] + +[[package]] +name = "tokio" +version = "1.46.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0cc3a2344dafbe23a245241fe8b09735b521110d30fcefbbd5feb1797ca35d17" +dependencies = [ + "backtrace", + "bytes", + "io-uring", + "libc", + "mio", + "parking_lot", + "pin-project-lite", + "signal-hook-registry", + "slab", + "socket2", + "tokio-macros", + "windows-sys 0.52.0", +] + +[[package]] +name = "tokio-macros" +version = "2.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6e06d43f1345a3bcd39f6a56dbb7dcab2ba47e68e8ac134855e7e2bdbaf8cab8" +dependencies = [ + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "unicode-ident" +version = "1.0.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512" + +[[package]] +name = "utf8parse" +version = "0.2.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821" + +[[package]] +name = "walkdir" +version = "2.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "29790946404f91d9c5d06f9874efddea1dc06c5efe94541a7d6863108e3a5e4b" +dependencies = [ + "same-file", + "winapi-util", +] + +[[package]] +name = "wasi" +version = "0.11.1+wasi-snapshot-preview1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ccf3ec651a847eb01de73ccad15eb7d99f80485de043efb2f370cd654f4ea44b" + +[[package]] +name = "winapi-util" +version = "0.1.9" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "cf221c93e13a30d793f7645a0e7762c55d169dbb0a49671918a2319d289b10bb" +dependencies = [ + "windows-sys 0.59.0", +] + +[[package]] +name = "windows-sys" +version = "0.52.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "282be5f36a8ce781fad8c8ae18fa3f9beff57ec1b52cb3de0789201425d9a33d" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-sys" +version = "0.59.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-targets" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" +dependencies = [ + "windows_aarch64_gnullvm", + "windows_aarch64_msvc", + "windows_i686_gnu", + "windows_i686_gnullvm", + "windows_i686_msvc", + "windows_x86_64_gnu", + "windows_x86_64_gnullvm", + "windows_x86_64_msvc", +] + +[[package]] +name = "windows_aarch64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" + +[[package]] +name = "windows_aarch64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" + +[[package]] +name = "windows_i686_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" + +[[package]] +name = "windows_i686_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" + +[[package]] +name = "windows_i686_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" + +[[package]] +name = "windows_x86_64_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" + +[[package]] +name = "windows_x86_64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" + +[[package]] +name = "windows_x86_64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..5d5dede --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,10 @@ +[package] +name = "snaferrous" +version = "0.1.0" +edition = "2024" + +[dependencies] +clap = { version = "4.5.41", features = ["derive"] } +colored = "3.0.0" +tokio = { version = "1.46.1", features = ["full"] } +walkdir = "2.5.0" diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..b03151b --- /dev/null +++ b/src/main.rs @@ -0,0 +1,429 @@ +/* +Author: Kevin (Kaged Pyro) Gunter +Purpose: I got tired of snaffler getting caught, so I rewrote it in rust, which edrs have trouble detecting. + +*/ +use clap::Parser; +use std::fmt::Debug; +use std::fs; +use std::fs::read_to_string; +use std::fs::OpenOptions; +use std::io::Write; +use std::ops::Index; +use std::path::PathBuf; +use std::process::exit; +use std::process::Command; +use std::thread; +use std::time::Duration; +use tokio; +use tokio::sync::mpsc::{channel, Sender, Receiver}; +use colored::Colorize; + +#[derive(Parser, Debug)] +#[command(version, about, long_about = Some("finds shares, but its written in rust which sometimes gets past EDR!"))] +struct Args{ + #[arg(short, long, help = "path to save output file Defaults to not saving output.")] + outfile: Option, + + #[arg(short, long, help = "number of threads to use, default to 10. \nNote thre thread count will be doubled, one set for share finder tasks, and one set for file and infor finding tasks.")] + threads: Option, + + #[arg(short, long, help = "specific targets. should be comma separated.")] + targets: Option, +} + +struct ShareFinder{ + id: usize, + tx: Sender, +} + +#[derive(Clone)] +struct Message{ + source: MessageType, + destination: MessageType, + content: String, +} + +#[derive(Clone, PartialEq)] +enum MessageType{ + ShareMessage, + InfoMessage, + ControlMessage, +} + + +async fn find_shares(task: ShareFinder, mut rx: Receiver){ + println!("{} share task started!", task.id); + let ping_recv = rx.recv().await; + if ping_recv.is_some(){ + let message = Message{source: MessageType::ShareMessage, destination: MessageType::ControlMessage, content: String::from("pong!")}; + task.tx.send(message).await.unwrap(); + } + loop{ + if rx.capacity() == 0{ + println!("rx is full for share finder {}", task.id); + } + let rx_res = rx.recv().await; + if rx_res.is_some(){ + let computer = rx_res.unwrap().content; + if computer == String::from("||DONE||"){ + let message = Message{source: MessageType::ShareMessage, destination: MessageType::ControlMessage, content: format!("{}:||DONE||", task.id)}; + task.tx.send(message).await.unwrap(); + break; + } + println!("scanning {}", computer); + let share_list_res = Command::new("net").arg("view").arg(computer.clone()).arg("/all").output(); + let mut error_string = String::new(); + let mut success_string = String::new(); + if share_list_res.is_ok(){ + let output = share_list_res.unwrap(); + if output.stdout.len() > 0{ + success_string = String::from_utf8_lossy(&output.stdout).to_string(); + } + + if output.stderr.len() > 0{ + error_string = String::from_utf8_lossy(&output.stderr).to_string(); + } + } + else{ + error_string = share_list_res.err().unwrap().to_string(); + } + if error_string.len() > 0{ + eprintln!("{}", "Error listing shares!".red()); + eprint!("{}", error_string.red()); + } + else if success_string.len() > 0{ + for line in success_string.lines(){ + if line.contains("Disk"){ + let share_name = line.split_whitespace().collect::>()[0]; + let share_path = format!("\\\\{}\\{}", computer, share_name); + let message = Message{source: MessageType::ShareMessage, destination: MessageType::InfoMessage, content: format!("{}:{}", task.id, share_path)}; + task.tx.send(message).await.unwrap(); + } + } + } + } + } +} + +async fn find_info(task: ShareFinder, mut rx: Receiver){ + println!("{} file task started!", task.id); + let ping_recv = rx.recv().await; + if ping_recv.is_some(){ + let message = Message{source: MessageType::ShareMessage, destination: MessageType::ControlMessage, content: String::from("pong!")}; + task.tx.send(message).await.unwrap(); + } + let files_to_read = vec![ + ".txt", + ".ini", + ".xml", + ".json", + ".config", + ".conf", + ".bat", + ".cmd", + ".sql", + ".ps1", + ".py", + ".vbscript" + ]; + + let interesting_info = vec![ + "password", + "pass", + "user", + "api", + "key", + "credit card", + "cc", + "ssn", + "social Security", + "tax", + "i9", + "it", + "identified", + "username", + ]; + loop{ + let rx_res = rx.recv().await; + if rx_res.is_some(){ + let message = rx_res.unwrap(); + let message_vec: Vec<&str> = message.content.split(":").collect(); + let path = message_vec[1]; + if path.contains("||DONE||"){ + let done_message = Message{source: MessageType::InfoMessage, destination: MessageType::ControlMessage, content: format!("{}:||DONE||", task.id)}; + task.tx.send(done_message).await.unwrap(); + } + for entry_res in walkdir::WalkDir::new(path){ + if entry_res.is_ok(){ + let entry = entry_res.unwrap(); + let file_path = entry.into_path(); + let mut file_name = String::new(); + let mut file_content = String::new(); + if file_path.file_name().is_some(){ + file_name = file_path.file_name().unwrap().to_string_lossy().to_string(); + } + for extension in &files_to_read{ + if file_name.contains(extension){ + let file_content_res = read_to_string(&file_path); + if file_content_res.is_ok(){ + file_content = file_content_res.unwrap(); + } + } + } + for thing in &interesting_info{ + if file_name.contains(thing) || file_content.contains(thing){ + let message = Message{source: MessageType::InfoMessage, destination: MessageType::ControlMessage, content: format!("{}:Keyword match at {}", task.id, file_path.display())}; + task.tx.send(message).await.unwrap(); + } + else{ + let message = Message{source: MessageType::InfoMessage, destination: MessageType::ControlMessage, content: format!("{}:file found at {}", task.id, file_path.display())}; + task.tx.send(message).await.unwrap(); + } + } + } + } + } + } +} + + + +#[tokio::main] +async fn main(){ + let args = Args::parse(); + let mut outfile = PathBuf::new(); + let mut file_threads = 5; + let mut share_threads = 5; + let mut save = false; + let mut computers = Vec::new(); + if args.outfile.is_some(){ + outfile = args.outfile.unwrap(); + save = true; + } + if args.threads.is_some(){ + let threads = args.threads.unwrap() / 2; + file_threads = threads; + share_threads = threads; + } + if args.targets.is_some(){ + println!("gathering the targets you gave me."); + let targets = args.targets.unwrap(); + if targets.contains(","){ + let split_targets: Vec<&str> = targets.split(",").collect(); + for target in split_targets{ + computers.push(target.to_string()); + } + } + else{ + computers.push(targets); + } + } + else{ + println!("no targets given, proceeding with domain computer enumeration..."); + println!("finding computers..."); + let command_string = String::from("net group \"domain computers\" /domain"); + let mut temp_file = fs::File::create("./temp.bat").unwrap(); + write!(temp_file, "{}", command_string).unwrap(); + let computer_res = Command::new(".\\temp.bat").output(); + let mut error_string = String::new(); + let mut success_string = String::new(); + fs::remove_file("./temp.bat").unwrap(); + if computer_res.is_ok(){ + let output = computer_res.unwrap(); + if output.stdout.len() > 0{ + success_string = String::from_utf8_lossy(&output.stdout).to_string(); + } + else if output.stderr.len() > 0{ + error_string = String::from_utf8_lossy(&output.stderr).to_string(); + } + } + else{ + error_string = computer_res.err().unwrap().to_string(); + } + if error_string.len() > 0{ + eprintln!("{}", "error getting computers!".red()); + eprintln!("{}", error_string.red()); + exit(1); + } + if success_string.len() > 0{ + for line in success_string.lines(){ + if line.contains("$"){ + let words:Vec<&str> = line.split_whitespace().collect(); + for word in words{ + let mut computer_name = word.to_string(); + computer_name.pop(); + println!("{} {}", "found".green(), computer_name.green()); + computers.push(computer_name); + } + } + } + } + } + if share_threads > computers.len(){ + share_threads = computers.len(); + //file_threads = computers.len(); + } + let mut share_handles = Vec::new(); + let mut file_handles = Vec::new(); + println!("computer enumeration finished, starting task finder threads..."); + let (maintx, mut mainrx) = channel(1024); + let mut share_tasks = Vec::new(); + let mut share_txes = Vec::new(); + let mut file_tasks = Vec::new(); + let mut file_txes = Vec::new(); + for id in 0..share_threads{ + println!("starting share task {}...", id); + let (share_tx,share_rx) = channel(1); + let new_share_task = ShareFinder{id, tx: maintx.clone()}; + share_handles.push(tokio::spawn(find_shares(new_share_task, share_rx))); + share_tasks.push(id); + share_txes.push(share_tx.clone()); + let ping_message = Message{source: MessageType::ControlMessage, destination: MessageType::ShareMessage, content: String::from("ping!")}; + share_tx.send(ping_message).await.unwrap(); + loop{ + let rx_recv = mainrx.recv().await; + if rx_recv.is_some(){ + let message = rx_recv.unwrap(); + if message.content == String::from("pong!"){ + println!("{} ready!", id); + break; + } + } + println!("didn't recieve file pong from {}", id); + } + } + for id in 0..file_threads{ + println!("starting file task {}...", id); + let (file_tx, file_rx) = channel(1); + let new_file_task = ShareFinder{id, tx: maintx.clone()}; + file_handles.push(tokio::spawn(find_info(new_file_task, file_rx))); + file_tasks.push(id); + file_txes.push(file_tx.clone()); + let ping_message = Message{source: MessageType::ControlMessage, destination: MessageType::ShareMessage, content: String::from("ping!")}; + file_tx.send(ping_message).await.unwrap(); + loop{ + let rx_recv = mainrx.recv().await; + if rx_recv.is_some(){ + let message = rx_recv.unwrap(); + if message.content == String::from("pong!"){ + println!("{} ready!", id); + break; + } + } + println!("didn't recieve file pong from {}", id); + } + } + let mut current_computer = 0; + let mut shares_finished = false; + let mut files_finished = false; + let mut file_buffer = Vec::new(); + let mut finished_counter = 0; + let mut empty_counter = 0; + let mut started = true; + let mut handled_lines = Vec::new(); + loop { + if files_finished && shares_finished{ + break; + } + if !mainrx.is_empty(){ + finished_counter = 0; + empty_counter = 0; + started = true; + let rx_res = mainrx.recv().await; + if rx_res.is_some(){ + let message = rx_res.unwrap(); + match message.destination{ + MessageType::ControlMessage => { + let message_vec: Vec<&str> = message.content.split(":").collect(); + let _id = message_vec[0]; + let message_content = message_vec[1].to_string(); + match message_content{ + _ => { + if !handled_lines.contains(&message_content){ + if save{ + let open_res = OpenOptions::new().append(true).create(true).open(&outfile); + if open_res.is_ok(){ + let mut file = open_res.unwrap(); + let write_res = write!(file, "{}\n", message_content); + if write_res.is_err(){ + eprintln!("{}", "error writing to outfile!".red()); + eprintln!("{}", write_res.err().unwrap().to_string().red()); + } + } + } + println!("{}", message_content.green()); + handled_lines.push(message_content); + } + + } + } + } + MessageType::InfoMessage => { + file_buffer.push(message.content); + } + MessageType::ShareMessage => {} + } + } + } + let mut sent = false; + if !shares_finished{ + for tx in &share_txes{ + if tx.capacity() > 0{ + let message = Message{source: MessageType::ShareMessage, destination: MessageType::ShareMessage, content: computers[current_computer].clone()}; + tx.send(message).await.unwrap(); + sent = true; + break; + } + } + if sent{ + current_computer +=1; + if current_computer == computers.len() { + shares_finished = true; + } + } + } + if shares_finished{ + if file_buffer.len() == 0{ + empty_counter += 1; + } + if empty_counter >= 100{ + finished_counter +=1; + thread::sleep(Duration::from_millis(50)); + } + } + if file_buffer.len() > 0{ + let mut sent_index = Vec::new(); + empty_counter = 0; + finished_counter = 0; + let mut current_tx = 0; + for index in 0 .. file_buffer.len() - 1{ + let mut sent = false; + let message = Message{source: MessageType::ControlMessage, destination: MessageType::InfoMessage, content: file_buffer[index].clone()}; + if file_txes[current_tx].capacity()> 0{ + file_txes[current_tx].send(message).await.unwrap(); + sent = true; + } + else{ + current_tx += 1; + if current_tx == file_txes.len(){ + current_tx = 0; + } + } + if sent{ + sent_index.push(index); + } + } + for index in sent_index{ + file_buffer.remove(index); + } + } + if finished_counter == 10{ + for tx in &file_txes{ + let done_message = Message{source: MessageType::ControlMessage, destination: MessageType::InfoMessage, content: String::from("0:||DONE||")}; + tx.send(done_message).await.unwrap(); + } + files_finished = true; + } + } +} \ No newline at end of file