Add test through DNS

This commit is contained in:
Rémi BERTHO 2021-10-05 23:37:30 +02:00
parent 7659338106
commit 94050d4af4
Signed by: dalan
GPG key ID: EE3B917931C07B64
3 changed files with 122 additions and 74 deletions

View file

@ -1,6 +1,6 @@
[package]
name = "check_ip"
version = "1.0.0"
version = "2.0.0"
edition = "2018"
[dependencies]
@ -14,3 +14,4 @@ serde = { version = "1.0", features = ["derive"] }
log = "0.4"
simplelog = "^0.10.0"
clokwerk = "0.3"
trust-dns-resolver = {version = "0.20", features = ["serde-config"] }

View file

@ -1,4 +1,5 @@
# TODO
* [ ] Examples de fichiers de config
* [ ] Test à partir de ladresse IP récupéré par DNS (https://crates.io/crates/trust-dns-resolver)
* [ ] Examples de fichiers de config (config + systemd)
* [ ] Nenvoyer quun seul mail par test =>_test_domains renvoie Option<String>
* [ ] Test de lIPv6 (https://crates.io/crates/query_external_ip)
* [ ] Rustfmt

View file

@ -8,6 +8,7 @@ extern crate native_tls;
extern crate serde;
extern crate simplelog;
extern crate toml;
extern crate trust_dns_resolver;
use self::native_tls::{Protocol, TlsConnector};
use anyhow::{bail, Result};
@ -20,8 +21,8 @@ use serde::Deserialize;
use simplelog::{ConfigBuilder, LevelFilter, WriteLogger};
use std::fs::{self, OpenOptions};
use std::net::IpAddr;
use std::path::Path;
use std::{thread, time::Duration};
use trust_dns_resolver::{config::*, Name, Resolver};
fn main() -> Result<()> {
// Init log
@ -40,48 +41,71 @@ fn main() -> Result<()> {
)
.unwrap();
// Init scheduler
let mut scheduler = Scheduler::new();
info!("Init");
scheduler.every(1.day()).at("06:00").run(|| {
if let Err(e) = check() {
// Init tester
info!("Init tester");
let tester = match Tester::new() {
Err(e) => {
error!("Error {}", e);
bail!("Cannot init tester");
}
});
Ok(tester) => tester,
};
// Initial check
if let Err(e) = check() {
if let Err(e) = tester.check() {
error!("Error {}", e);
bail!("Cannot get initial IP");
}
if let Some(false) = tester.config.test {
// Init scheduler
info!("Init scheduler");
let mut scheduler = Scheduler::new();
scheduler.every(1.day()).at("06:00").run(move || {
if let Err(e) = tester.check() {
error!("Error {}", e);
}
});
// Run
loop {
scheduler.run_pending();
thread::sleep(Duration::from_secs(60));
}
}
Ok(())
}
fn check() -> Result<()> {
info!("Check");
match get() {
Ok((old_ip, new_ip)) => {
if old_ip != new_ip {
let msg = format!("IP changed from {} to {}", old_ip, new_ip);
info!("{}", msg);
send_mail("New IP".to_owned(), msg)?;
struct Tester {
config: Config,
dns_resolver: Resolver,
}
impl Tester {
fn new() -> Result<Tester> {
Ok(Tester {
config: toml::from_str(&fs::read_to_string("config.toml")?)?,
dns_resolver: Resolver::new(ResolverConfig::default(), ResolverOpts::default())?,
})
}
fn check(&self) -> Result<()> {
info!("Check");
match self.get_ip() {
Ok(ipv4) => {
self.test_domains(ipv4)?;
}
Err(err) => {
let msg = format!("Cannot get IP ({})", err);
error!("{}", msg);
send_mail("Error on getting IP".to_owned(), msg)?;
self.send_mail("Error on getting IP".to_owned(), msg)?;
}
}
Ok(())
}
}
fn get() -> Result<(IpAddr, IpAddr)> {
fn get_ip(&self) -> Result<IpAddr> {
// Get new IP
let new_ip = match my_internet_ip::get() {
Ok(ip) => ip,
@ -89,54 +113,76 @@ fn get() -> Result<(IpAddr, IpAddr)> {
};
info!("Current IP is {}", new_ip);
// Get old IP
let path = Path::new("./old_ip.txt");
let old_ip = if path.exists() {
fs::read_to_string(path)?.parse()?
} else {
new_ip.clone()
};
info!("Old IP is {}", old_ip);
Ok(new_ip)
}
// Write new IP
fs::write(path, format!("{}", new_ip))?;
fn test_domains(&self, ipv4: IpAddr) -> Result<()> {
for domain in self.config.server.domains.iter() {
let address = self.dns_resolver.lookup_ip(domain.clone())?;
for dns_ip in address.iter() {
if dns_ip.is_ipv4() {
if dns_ip != ipv4 {
let msg = format!(
"Wrong IPV4 for {} (DNS: {}, current: {})",
domain, dns_ip, ipv4
);
info!("{}", msg);
self.send_mail("Wrong IP".to_owned(), msg)?;
}
}
}
}
Ok(())
}
Ok((old_ip, new_ip))
}
fn send_mail(subject: String, body: String) -> Result<()> {
fn send_mail(&self, subject: String, body: String) -> Result<()> {
info!("Send mail {}", subject);
let config: Config = toml::from_str(&fs::read_to_string("config.toml")?)?;
if let Some(false) = self.config.test {
let email = EmailBuilder::new()
.to(config.mail.from)
.from(config.mail.to)
.to(self.config.mail.from.clone())
.from(self.config.mail.to.clone())
.subject(subject)
.text(body)
.build()?;
// Create transport
let creds = Credentials::new(config.mail.username, config.mail.password);
let creds = Credentials::new(
self.config.mail.username.clone(),
self.config.mail.password.clone(),
);
let mut tls_builder = TlsConnector::builder();
tls_builder.min_protocol_version(Some(Protocol::Sslv3));
let tls_parameters =
ClientTlsParameters::new(config.mail.server.clone(), tls_builder.build().unwrap());
let tls_parameters = ClientTlsParameters::new(
self.config.mail.server.clone(),
tls_builder.build().unwrap(),
);
let mut mailer = SmtpClient::new(
(config.mail.server, config.mail.port),
(self.config.mail.server.clone(), self.config.mail.port),
ClientSecurity::Required(tls_parameters),
)?
.credentials(creds)
.transport();
mailer.send(email.into())?;
} else {
println!("subject: {} - body: {}", subject, body);
};
Ok(())
}
}
#[derive(Debug, Deserialize)]
struct Config {
mail: MailConfig,
server: ServerConfig,
test: Option<bool>,
}
#[derive(Debug, Deserialize)]
struct ServerConfig {
domains: Vec<Name>,
}
#[derive(Debug, Deserialize)]