mirror of https://github.com/Revertron/Alfis
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
197 lines
5.5 KiB
Rust
197 lines
5.5 KiB
Rust
use std::net::IpAddr;
|
|
use std::num;
|
|
|
|
pub use constants::*;
|
|
use rand::Rng;
|
|
#[cfg(not(target_os = "macos"))]
|
|
use thread_priority::*;
|
|
|
|
use crate::dns::protocol::DnsRecord;
|
|
|
|
pub mod constants;
|
|
pub mod eventbus;
|
|
pub mod simplebus;
|
|
|
|
/// 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 parts: Vec<&str> = name.rsplitn(2, '.').collect();
|
|
if parts.len() == 2 && parts[1].len() < 3 && is_numeric(parts[1]) {
|
|
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_ascii_alphanumeric() {
|
|
return false;
|
|
}
|
|
}
|
|
true
|
|
}
|
|
|
|
pub fn is_numeric(str: &str) -> bool {
|
|
for char in str.chars() {
|
|
if !char.is_numeric() {
|
|
return false;
|
|
}
|
|
}
|
|
true
|
|
}
|
|
|
|
pub fn get_domain_zone(domain: &str) -> String {
|
|
let parts: Vec<&str> = domain.rsplitn(2, '.').collect();
|
|
if !parts.is_empty() {
|
|
parts[0].to_owned()
|
|
} else {
|
|
String::new()
|
|
}
|
|
}
|
|
|
|
fn split_n(s: &str, n: usize) -> Vec<&str> {
|
|
(0..=(s.len() - n + 1) / 2)
|
|
.map(|i| &s[2 * i..2 * i + n])
|
|
.collect()
|
|
}
|
|
|
|
/// 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
|
|
}
|
|
|
|
/// Checks if this IP is from Yggdrasil network
|
|
/// https://yggdrasil-network.github.io
|
|
pub fn is_yggdrasil(addr: &IpAddr) -> bool {
|
|
if let IpAddr::V6(ipv6) = addr {
|
|
let first_byte = ipv6.octets()[0];
|
|
return first_byte == 2 || first_byte == 3;
|
|
}
|
|
false
|
|
}
|
|
|
|
/// Checks if this record has IP from Yggdrasil network
|
|
/// https://yggdrasil-network.github.io
|
|
pub fn is_yggdrasil_record(record: &DnsRecord) -> bool {
|
|
match record {
|
|
DnsRecord::UNKNOWN { .. } => {}
|
|
DnsRecord::A { .. } => return false,
|
|
DnsRecord::NS { .. } => {}
|
|
DnsRecord::CNAME { .. } => {}
|
|
DnsRecord::SOA { .. } => {}
|
|
DnsRecord::PTR { .. } => {}
|
|
DnsRecord::MX { .. } => {}
|
|
DnsRecord::TXT { .. } => {}
|
|
DnsRecord::AAAA { addr, .. } => return is_yggdrasil(&IpAddr::from(*addr)),
|
|
DnsRecord::SRV { .. } => {}
|
|
DnsRecord::OPT { .. } => {}
|
|
DnsRecord::TLSA { .. } => {}
|
|
}
|
|
true
|
|
}
|
|
|
|
#[cfg(target_os = "windows")]
|
|
#[allow(unused_variables)]
|
|
pub fn setup_miner_thread(cpu: u32) {
|
|
let _ = set_current_thread_priority(ThreadPriority::Min);
|
|
//let _ = set_current_thread_ideal_processor(IdealProcessor::from(cpu));
|
|
}
|
|
|
|
#[cfg(target_os = "linux")]
|
|
#[allow(unused_variables)]
|
|
pub fn setup_miner_thread(cpu: u32) {
|
|
let _ = set_current_thread_priority(ThreadPriority::Min);
|
|
}
|
|
|
|
#[cfg(target_os = "macos")]
|
|
#[allow(unused_variables)]
|
|
pub fn setup_miner_thread(cpu: u32) {
|
|
// MacOS is not supported by thread_priority crate
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use std::net::IpAddr;
|
|
|
|
use crate::{check_domain, is_yggdrasil};
|
|
|
|
#[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));
|
|
assert!(check_domain("a.com", true));
|
|
assert!(check_domain("ab.com", true));
|
|
assert!(check_domain("abc.com", true));
|
|
assert!(check_domain("777.com", true));
|
|
assert!(!check_domain("77.com", true));
|
|
assert!(!check_domain("7.com", true));
|
|
}
|
|
|
|
#[test]
|
|
fn test_is_yggdrasil() {
|
|
let addr: IpAddr = "200::1".parse().unwrap();
|
|
assert!(is_yggdrasil(&addr));
|
|
let addr: IpAddr = "226::1".parse().unwrap();
|
|
assert!(is_yggdrasil(&addr));
|
|
let addr: IpAddr = "300::1".parse().unwrap();
|
|
assert!(is_yggdrasil(&addr));
|
|
let addr: IpAddr = "326::1".parse().unwrap();
|
|
assert!(is_yggdrasil(&addr));
|
|
let addr: IpAddr = "2001::1".parse().unwrap();
|
|
assert!(!is_yggdrasil(&addr));
|
|
let addr: IpAddr = "2201::1".parse().unwrap();
|
|
assert!(!is_yggdrasil(&addr));
|
|
}
|
|
}
|