diff --git a/src/config.rs b/src/config.rs index 56c781e..de50543 100644 --- a/src/config.rs +++ b/src/config.rs @@ -3,7 +3,7 @@ use core::fmt; pub struct Config { pub subdomains: Vec, - wildcard_test: String, + pub wildcard_test: String, } impl fmt::Display for Config { @@ -19,7 +19,7 @@ impl Config { let subdomains = config.get("General", "subdomains").unwrap(); let mut subvec: Vec = vec![]; - + for x in subdomains.split(',') { subvec.push(x.to_string()); } diff --git a/src/domain.rs b/src/domain.rs index d32776d..5c0aa5a 100644 --- a/src/domain.rs +++ b/src/domain.rs @@ -80,6 +80,7 @@ impl Domain { for subdomain in &config.subdomains { self.append_subdomain(subdomain.to_string()); } + self.append_subdomain(config.wildcard_test.to_string()) } pub fn lookup_all_records(&mut self) { diff --git a/src/main.rs b/src/main.rs index d704ebb..94b249f 100644 --- a/src/main.rs +++ b/src/main.rs @@ -6,7 +6,7 @@ use crate::config::Config; fn main() { - let mut test = Domain::new("pigandpilgrim.com.au".to_string()); + let mut test = Domain::new("benjamyn.love".to_string()); let config = Config::from_file("test.ini".to_string()); test.apply_config(&config); diff --git a/test.ini b/test.ini index 5996892..19886b3 100644 --- a/test.ini +++ b/test.ini @@ -1,3 +1,3 @@ [General] wildcard_test=dfjgnkdfjngkdfngjkd -subdomains=www,ftp,mail \ No newline at end of file +subdomains=www,ftp,mail,files \ No newline at end of file