parent
7faab2072c
commit
015e696077
|
@ -21,7 +21,7 @@ use std::{
|
||||||
path::Path,
|
path::Path,
|
||||||
process::exit,
|
process::exit,
|
||||||
sync::{
|
sync::{
|
||||||
atomic::{AtomicU64, Ordering},
|
atomic::{AtomicBool, AtomicU64, Ordering},
|
||||||
Arc,
|
Arc,
|
||||||
},
|
},
|
||||||
thread,
|
thread,
|
||||||
|
@ -30,6 +30,12 @@ use std::{
|
||||||
|
|
||||||
const NO_PASSPHRASE: &str = "";
|
const NO_PASSPHRASE: &str = "";
|
||||||
|
|
||||||
|
struct GrindMatch {
|
||||||
|
starts: String,
|
||||||
|
ends: String,
|
||||||
|
count: AtomicU64,
|
||||||
|
}
|
||||||
|
|
||||||
fn check_for_overwrite(outfile: &str, matches: &ArgMatches) {
|
fn check_for_overwrite(outfile: &str, matches: &ArgMatches) {
|
||||||
let force = matches.is_present("force");
|
let force = matches.is_present("force");
|
||||||
if !force && Path::new(outfile).exists() {
|
if !force && Path::new(outfile).exists() {
|
||||||
|
@ -73,6 +79,115 @@ fn output_keypair(
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn grind_validator_starts_with(v: String) -> Result<(), String> {
|
||||||
|
if v.matches(':').count() != 1 || (v.starts_with(':') || v.ends_with(':')) {
|
||||||
|
return Err(String::from("Expected : between PREFIX and COUNT"));
|
||||||
|
}
|
||||||
|
let args: Vec<&str> = v.split(':').collect();
|
||||||
|
bs58::decode(&args[0])
|
||||||
|
.into_vec()
|
||||||
|
.map_err(|err| format!("{}: {:?}", args[0], err))?;
|
||||||
|
let count = args[1].parse::<u64>();
|
||||||
|
if count.is_err() || count.unwrap() == 0 {
|
||||||
|
return Err(String::from("Expected COUNT to be of type u64"));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn grind_validator_ends_with(v: String) -> Result<(), String> {
|
||||||
|
if v.matches(':').count() != 1 || (v.starts_with(':') || v.ends_with(':')) {
|
||||||
|
return Err(String::from("Expected : between SUFFIX and COUNT"));
|
||||||
|
}
|
||||||
|
let args: Vec<&str> = v.split(':').collect();
|
||||||
|
bs58::decode(&args[0])
|
||||||
|
.into_vec()
|
||||||
|
.map_err(|err| format!("{}: {:?}", args[0], err))?;
|
||||||
|
let count = args[1].parse::<u64>();
|
||||||
|
if count.is_err() || count.unwrap() == 0 {
|
||||||
|
return Err(String::from("Expected COUNT to be of type u64"));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn grind_validator_starts_and_ends_with(v: String) -> Result<(), String> {
|
||||||
|
if v.matches(':').count() != 2 || (v.starts_with(':') || v.ends_with(':')) {
|
||||||
|
return Err(String::from(
|
||||||
|
"Expected : between PREFIX and SUFFIX and COUNT",
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let args: Vec<&str> = v.split(':').collect();
|
||||||
|
bs58::decode(&args[0])
|
||||||
|
.into_vec()
|
||||||
|
.map_err(|err| format!("{}: {:?}", args[0], err))?;
|
||||||
|
bs58::decode(&args[1])
|
||||||
|
.into_vec()
|
||||||
|
.map_err(|err| format!("{}: {:?}", args[1], err))?;
|
||||||
|
let count = args[2].parse::<u64>();
|
||||||
|
if count.is_err() || count.unwrap() == 0 {
|
||||||
|
return Err(String::from("Expected COUNT to be a u64"));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn grind_print_info(grind_matches: &[GrindMatch]) {
|
||||||
|
println!("Searching with {} threads for:", num_cpus::get());
|
||||||
|
for gm in grind_matches {
|
||||||
|
let mut msg = Vec::<String>::new();
|
||||||
|
if gm.count.load(Ordering::Relaxed) > 1 {
|
||||||
|
msg.push("pubkeys".to_string());
|
||||||
|
msg.push("start".to_string());
|
||||||
|
msg.push("end".to_string());
|
||||||
|
} else {
|
||||||
|
msg.push("pubkey".to_string());
|
||||||
|
msg.push("starts".to_string());
|
||||||
|
msg.push("ends".to_string());
|
||||||
|
}
|
||||||
|
println!(
|
||||||
|
"\t{} {} that {} with '{}' and {} with '{}'",
|
||||||
|
gm.count.load(Ordering::Relaxed),
|
||||||
|
msg[0],
|
||||||
|
msg[1],
|
||||||
|
gm.starts,
|
||||||
|
msg[2],
|
||||||
|
gm.ends
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn grind_parse_args(
|
||||||
|
starts_with_args: HashSet<String>,
|
||||||
|
ends_with_args: HashSet<String>,
|
||||||
|
starts_and_ends_with_args: HashSet<String>,
|
||||||
|
) -> Vec<GrindMatch> {
|
||||||
|
let mut grind_matches = Vec::<GrindMatch>::new();
|
||||||
|
for sw in starts_with_args {
|
||||||
|
let args: Vec<&str> = sw.split(':').collect();
|
||||||
|
grind_matches.push(GrindMatch {
|
||||||
|
starts: args[0].to_lowercase(),
|
||||||
|
ends: "".to_string(),
|
||||||
|
count: AtomicU64::new(args[1].parse::<u64>().unwrap()),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
for ew in ends_with_args {
|
||||||
|
let args: Vec<&str> = ew.split(':').collect();
|
||||||
|
grind_matches.push(GrindMatch {
|
||||||
|
starts: "".to_string(),
|
||||||
|
ends: args[0].to_lowercase(),
|
||||||
|
count: AtomicU64::new(args[1].parse::<u64>().unwrap()),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
for swew in starts_and_ends_with_args {
|
||||||
|
let args: Vec<&str> = swew.split(':').collect();
|
||||||
|
grind_matches.push(GrindMatch {
|
||||||
|
starts: args[0].to_lowercase(),
|
||||||
|
ends: args[1].to_lowercase(),
|
||||||
|
count: AtomicU64::new(args[2].parse::<u64>().unwrap()),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
grind_print_info(&grind_matches);
|
||||||
|
grind_matches
|
||||||
|
}
|
||||||
|
|
||||||
fn main() -> Result<(), Box<dyn error::Error>> {
|
fn main() -> Result<(), Box<dyn error::Error>> {
|
||||||
let matches = App::new(crate_name!())
|
let matches = App::new(crate_name!())
|
||||||
.about(crate_description!())
|
.about(crate_description!())
|
||||||
|
@ -148,33 +263,37 @@ fn main() -> Result<(), Box<dyn error::Error>> {
|
||||||
.arg(
|
.arg(
|
||||||
Arg::with_name("ignore_case")
|
Arg::with_name("ignore_case")
|
||||||
.long("ignore-case")
|
.long("ignore-case")
|
||||||
.help("Perform case insensitive matches"),
|
.help("Performs case insensitive matches"),
|
||||||
)
|
|
||||||
.arg(
|
|
||||||
Arg::with_name("includes")
|
|
||||||
.long("includes")
|
|
||||||
.value_name("BASE58")
|
|
||||||
.takes_value(true)
|
|
||||||
.multiple(true)
|
|
||||||
.validator(|value| {
|
|
||||||
bs58::decode(&value).into_vec()
|
|
||||||
.map(|_| ())
|
|
||||||
.map_err(|err| format!("{}: {:?}", value, err))
|
|
||||||
})
|
|
||||||
.help("Save keypair if its public key includes this string\n(may be specified multiple times)"),
|
|
||||||
)
|
)
|
||||||
.arg(
|
.arg(
|
||||||
Arg::with_name("starts_with")
|
Arg::with_name("starts_with")
|
||||||
.long("starts-with")
|
.long("starts-with")
|
||||||
.value_name("BASE58 PREFIX")
|
.value_name("PREFIX:COUNT")
|
||||||
|
.number_of_values(1)
|
||||||
.takes_value(true)
|
.takes_value(true)
|
||||||
.multiple(true)
|
.multiple(true)
|
||||||
.validator(|value| {
|
.validator(grind_validator_starts_with)
|
||||||
bs58::decode(&value).into_vec()
|
.help("Saves specified number of keypairs whos public key starts with the indicated prefix\nExample: --starts-with sol:4\nPREFIX type is Base58\nCOUNT type is u64"),
|
||||||
.map(|_| ())
|
)
|
||||||
.map_err(|err| format!("{}: {:?}", value, err))
|
.arg(
|
||||||
})
|
Arg::with_name("ends_with")
|
||||||
.help("Save keypair if its public key starts with this prefix\n(may be specified multiple times)"),
|
.long("ends-with")
|
||||||
|
.value_name("SUFFIX:COUNT")
|
||||||
|
.number_of_values(1)
|
||||||
|
.takes_value(true)
|
||||||
|
.multiple(true)
|
||||||
|
.validator(grind_validator_ends_with)
|
||||||
|
.help("Saves specified number of keypairs whos public key ends with the indicated suffix\nExample: --ends-with ana:4\nSUFFIX type is Base58\nCOUNT type is u64"),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("starts_and_ends_with")
|
||||||
|
.long("starts-and-ends-with")
|
||||||
|
.value_name("PREFIX:SUFFIX:COUNT")
|
||||||
|
.number_of_values(1)
|
||||||
|
.takes_value(true)
|
||||||
|
.multiple(true)
|
||||||
|
.validator(grind_validator_starts_and_ends_with)
|
||||||
|
.help("Saves specified number of keypairs whos public key starts and ends with the indicated perfix and suffix\nExample: --starts-and-ends-with sol:ana:4\nPREFIX and SUFFIX type is Base58\nCOUNT type is u64"),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
.subcommand(
|
.subcommand(
|
||||||
|
@ -311,16 +430,8 @@ fn main() -> Result<(), Box<dyn error::Error>> {
|
||||||
}
|
}
|
||||||
("grind", Some(matches)) => {
|
("grind", Some(matches)) => {
|
||||||
let ignore_case = matches.is_present("ignore_case");
|
let ignore_case = matches.is_present("ignore_case");
|
||||||
let includes = if matches.is_present("includes") {
|
|
||||||
values_t_or_exit!(matches, "includes", String)
|
|
||||||
.into_iter()
|
|
||||||
.map(|s| if ignore_case { s.to_lowercase() } else { s })
|
|
||||||
.collect()
|
|
||||||
} else {
|
|
||||||
HashSet::new()
|
|
||||||
};
|
|
||||||
|
|
||||||
let starts_with = if matches.is_present("starts_with") {
|
let starts_with_args = if matches.is_present("starts_with") {
|
||||||
values_t_or_exit!(matches, "starts_with", String)
|
values_t_or_exit!(matches, "starts_with", String)
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|s| if ignore_case { s.to_lowercase() } else { s })
|
.map(|s| if ignore_case { s.to_lowercase() } else { s })
|
||||||
|
@ -328,65 +439,98 @@ fn main() -> Result<(), Box<dyn error::Error>> {
|
||||||
} else {
|
} else {
|
||||||
HashSet::new()
|
HashSet::new()
|
||||||
};
|
};
|
||||||
|
let ends_with_args = if matches.is_present("ends_with") {
|
||||||
|
values_t_or_exit!(matches, "ends_with", String)
|
||||||
|
.into_iter()
|
||||||
|
.map(|s| if ignore_case { s.to_lowercase() } else { s })
|
||||||
|
.collect()
|
||||||
|
} else {
|
||||||
|
HashSet::new()
|
||||||
|
};
|
||||||
|
let starts_and_ends_with_args = if matches.is_present("starts_and_ends_with") {
|
||||||
|
values_t_or_exit!(matches, "starts_and_ends_with", String)
|
||||||
|
.into_iter()
|
||||||
|
.map(|s| if ignore_case { s.to_lowercase() } else { s })
|
||||||
|
.collect()
|
||||||
|
} else {
|
||||||
|
HashSet::new()
|
||||||
|
};
|
||||||
|
|
||||||
if includes.is_empty() && starts_with.is_empty() {
|
if starts_with_args.is_empty()
|
||||||
|
&& ends_with_args.is_empty()
|
||||||
|
&& starts_and_ends_with_args.is_empty()
|
||||||
|
{
|
||||||
eprintln!(
|
eprintln!(
|
||||||
"Error: No keypair search criteria provided (--includes or --starts-with)"
|
"Error: No keypair search criteria provided (--starts-with or --ends-with or --starts-and-ends-with)"
|
||||||
);
|
);
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
let grind_matches =
|
||||||
|
grind_parse_args(starts_with_args, ends_with_args, starts_and_ends_with_args);
|
||||||
|
|
||||||
|
let grind_matches_thread_safe = Arc::new(grind_matches);
|
||||||
let attempts = Arc::new(AtomicU64::new(1));
|
let attempts = Arc::new(AtomicU64::new(1));
|
||||||
let found = Arc::new(AtomicU64::new(0));
|
let found = Arc::new(AtomicU64::new(0));
|
||||||
let start = Instant::now();
|
let start = Instant::now();
|
||||||
|
let done = Arc::new(AtomicBool::new(false));
|
||||||
|
|
||||||
println!(
|
for _ in 0..num_cpus::get() {
|
||||||
"Searching with {} threads for a pubkey containing {:?} or starting with {:?}",
|
let done = done.clone();
|
||||||
num_cpus::get(),
|
let attempts = attempts.clone();
|
||||||
includes,
|
let found = found.clone();
|
||||||
starts_with
|
let grind_matches_thread_safe = grind_matches_thread_safe.clone();
|
||||||
);
|
|
||||||
|
|
||||||
let _threads = (0..num_cpus::get())
|
let handle = thread::spawn(move || loop {
|
||||||
.map(|_| {
|
if done.load(Ordering::Relaxed) {
|
||||||
let attempts = attempts.clone();
|
break;
|
||||||
let found = found.clone();
|
}
|
||||||
let includes = includes.clone();
|
let attempts = attempts.fetch_add(1, Ordering::Relaxed);
|
||||||
let starts_with = starts_with.clone();
|
if attempts % 1_000_000 == 0 {
|
||||||
|
println!(
|
||||||
thread::spawn(move || loop {
|
"Searched {} keypairs in {}s. {} matches found.",
|
||||||
let attempts = attempts.fetch_add(1, Ordering::Relaxed);
|
attempts,
|
||||||
if attempts % 5_000_000 == 0 {
|
start.elapsed().as_secs(),
|
||||||
println!(
|
found.load(Ordering::Relaxed),
|
||||||
"Searched {} keypairs in {}s. {} matches found",
|
);
|
||||||
attempts,
|
}
|
||||||
start.elapsed().as_secs(),
|
let keypair = Keypair::new();
|
||||||
found.load(Ordering::Relaxed),
|
let mut pubkey = bs58::encode(keypair.pubkey()).into_string();
|
||||||
);
|
if ignore_case {
|
||||||
|
pubkey = pubkey.to_lowercase();
|
||||||
|
}
|
||||||
|
let mut total_matches_found = 0;
|
||||||
|
for i in 0..grind_matches_thread_safe.len() {
|
||||||
|
if grind_matches_thread_safe[i].count.load(Ordering::Relaxed) == 0 {
|
||||||
|
total_matches_found += 1;
|
||||||
|
continue;
|
||||||
}
|
}
|
||||||
|
if (!grind_matches_thread_safe[i].starts.is_empty()
|
||||||
let keypair = Keypair::new();
|
&& grind_matches_thread_safe[i].ends.is_empty()
|
||||||
let mut pubkey = bs58::encode(keypair.pubkey()).into_string();
|
&& pubkey.starts_with(&grind_matches_thread_safe[i].starts))
|
||||||
|
|| (grind_matches_thread_safe[i].starts.is_empty()
|
||||||
if ignore_case {
|
&& !grind_matches_thread_safe[i].ends.is_empty()
|
||||||
pubkey = pubkey.to_lowercase();
|
&& pubkey.ends_with(&grind_matches_thread_safe[i].ends))
|
||||||
}
|
|| (!grind_matches_thread_safe[i].starts.is_empty()
|
||||||
|
&& !grind_matches_thread_safe[i].ends.is_empty()
|
||||||
if starts_with.iter().any(|s| pubkey.starts_with(s))
|
&& pubkey.starts_with(&grind_matches_thread_safe[i].starts)
|
||||||
|| includes.iter().any(|s| pubkey.contains(s))
|
&& pubkey.ends_with(&grind_matches_thread_safe[i].ends))
|
||||||
{
|
{
|
||||||
let found = found.fetch_add(1, Ordering::Relaxed);
|
let _found = found.fetch_add(1, Ordering::Relaxed);
|
||||||
output_keypair(
|
grind_matches_thread_safe[i]
|
||||||
&keypair,
|
.count
|
||||||
&format!("{}.json", keypair.pubkey()),
|
.fetch_sub(1, Ordering::Relaxed);
|
||||||
&format!("{}", found),
|
println!("Wrote keypair to {}", &format!("{}.json", keypair.pubkey()));
|
||||||
)
|
write_keypair_file(&keypair, &format!("{}.json", keypair.pubkey()))
|
||||||
.unwrap();
|
.unwrap();
|
||||||
}
|
}
|
||||||
});
|
}
|
||||||
})
|
if total_matches_found == grind_matches_thread_safe.len() {
|
||||||
.collect::<Vec<_>>();
|
done.store(true, Ordering::Relaxed);
|
||||||
thread::park();
|
}
|
||||||
|
});
|
||||||
|
handle.join().unwrap();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
("verify", Some(matches)) => {
|
("verify", Some(matches)) => {
|
||||||
let keypair = get_keypair_from_matches(matches)?;
|
let keypair = get_keypair_from_matches(matches)?;
|
||||||
|
|
Loading…
Reference in New Issue