Revertron
108 lines
2.9 KiB
Rust
108 lines
2.9 KiB
Rust
use std::num;
|
|
use rand::Rng;
|
|
use num_bigint::BigUint;
|
|
use num_traits::One;
|
|
|
|
/// Convert bytes array to HEX format
|
|
pub fn to_hex(buf: &[u8]) -> String {
|
|
let mut result = String::new();
|
|
for x in buf.iter() {
|
|
result.push_str(&format!("{:01$X}", x, 2));
|
|
}
|
|
result
|
|
}
|
|
|
|
pub fn from_hex(string: &str) -> Result<Vec<u8>, num::ParseIntError> {
|
|
split_n(&string.trim()[..], 2)
|
|
.iter()
|
|
.map(|b| u8::from_str_radix(b, 16))
|
|
.collect()
|
|
}
|
|
|
|
pub fn check_domain(name: &str, allow_dots: bool) -> bool {
|
|
if name.starts_with('.') || name.starts_with('-') || name.ends_with('.') || name.ends_with('-') {
|
|
return false;
|
|
}
|
|
let mut last_dot = false;
|
|
let mut last_hyphen = false;
|
|
for char in name.chars() {
|
|
if allow_dots && char == '.' {
|
|
if last_dot {
|
|
return false;
|
|
} else {
|
|
last_dot = true;
|
|
continue;
|
|
}
|
|
}
|
|
if char == '-' {
|
|
if last_hyphen {
|
|
return false;
|
|
} else {
|
|
last_hyphen = true;
|
|
continue;
|
|
}
|
|
}
|
|
last_dot = false;
|
|
last_hyphen = false;
|
|
if !char.is_alphanumeric() {
|
|
return false;
|
|
}
|
|
}
|
|
true
|
|
}
|
|
|
|
fn split_n(s: &str, n: usize) -> Vec<&str> {
|
|
(0..=(s.len() - n + 1) / 2)
|
|
.map(|i| &s[2 * i..2 * i + n])
|
|
.collect()
|
|
}
|
|
|
|
/// There is no default PartialEq implementation for arrays > 32 in size
|
|
pub fn same_hash(left: &[u8], right: &[u8]) -> bool {
|
|
for (x, y) in left.iter().zip(right) {
|
|
if x != y {
|
|
return false;
|
|
}
|
|
}
|
|
true
|
|
}
|
|
|
|
pub fn hash_is_good(hash: &[u8], difficulty: usize) -> bool {
|
|
let target = BigUint::one() << ((hash.len() << 3) - difficulty);
|
|
let hash_int = BigUint::from_bytes_be(&hash);
|
|
|
|
return hash_int < target;
|
|
}
|
|
|
|
/// Generates random string of given length
|
|
pub fn random_string(length: usize) -> String {
|
|
let chars: Vec<char> = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!?".chars().collect();
|
|
let mut rng = rand::thread_rng();
|
|
let mut result = String::with_capacity(length);
|
|
for _ in 0..length {
|
|
let position: usize = rng.gen::<usize>() % chars.len();
|
|
let c: char = *chars.get(position).unwrap();
|
|
result.push(c);
|
|
}
|
|
result
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use crate::check_domain;
|
|
|
|
#[test]
|
|
fn test_check_domain() {
|
|
assert!(check_domain("abc0", false));
|
|
assert!(!check_domain("ab.c", false));
|
|
assert!(check_domain("a.b.c", true));
|
|
assert!(!check_domain("ab..c", true));
|
|
assert!(check_domain("a-b.c", true));
|
|
assert!(!check_domain("a--b.c", true));
|
|
assert!(check_domain("a-0-b.c", true));
|
|
assert!(!check_domain("-ab.c", true));
|
|
assert!(!check_domain("ab.c-", true));
|
|
assert!(!check_domain(".ab.c", true));
|
|
assert!(!check_domain("ab.c-", true));
|
|
}
|
|
} |