use std::string::String;
use winapi;
use winapi::*;
use winreg::RegKey;
use winreg::types::FromRegValue;
use error::*;
fn get_value<T: FromRegValue>(predef: winapi::HKEY,
subkey: &str,
value_name: &str,
default: T)
-> ::std::io::Result<T> {
let key = RegKey::predef(predef);
let value: T = match key.open_subkey_with_flags(subkey, KEY_READ) {
Ok(key) => {
match key.get_value(value_name) {
Ok(value) => value,
Err(err) => {
match err.kind() {
::std::io::ErrorKind::NotFound => default,
_ => return Err(err),
}
}
}
}
Err(err) => {
match err.kind() {
::std::io::ErrorKind::NotFound => default,
_ => return Err(err),
}
}
};
Ok(value)
}
const TCPIP_PARAMETERS_KEY_PATH: &str = "SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters";
pub fn get_search_list() -> Result<Vec<String>> {
let hklm = RegKey::predef(winreg::HKEY_LOCAL_MACHINE);
let params_key = hklm.open_subkey_with_flags(TCPIP_PARAMETERS_KEY_PATH,
KEY_READ)?;
let search_list: String = params_key.get_value("SearchList")?;
let search_list: Vec<String> = search_list.split(',').map(|s| s.to_string()).collect();
Ok(search_list)
}
pub fn get_domain() -> Result<Option<String>> {
let hklm = RegKey::predef(winreg::HKEY_LOCAL_MACHINE);
let params_key = hklm.open_subkey_with_flags(TCPIP_PARAMETERS_KEY_PATH,
KEY_READ)?;
let domain: String = params_key.get_value("Domain")?;
let domain = if domain.is_empty() {
None
} else {
Some(domain)
};
Ok(domain)
}
pub fn is_round_robin_enabled() -> Result<bool> {
let rotate: u32 = get_value(winreg::HKEY_LOCAL_MACHINE,
"SYSTEM\\CurrentControlSet\\Services\\DNS\\Parameters",
"RoundRobin",
1)?; Ok(rotate != 0)
}