Fixed bug

This commit is contained in:
Rémi BERTHO 2021-10-06 23:09:39 +02:00
parent e91093b8f5
commit aa2ce3e0f4
Signed by: dalan
GPG key ID: EE3B917931C07B64
3 changed files with 53 additions and 34 deletions

3
.gitignore vendored
View file

@ -18,3 +18,6 @@ Cargo.lock
/config.toml
/old_ip.txt
/log.txt
# IDE
.idea

View file

@ -1,4 +1,4 @@
# TODO
* [ ] Examples de fichiers de config (config + systemd)
* [ ] Nenvoyer quun seul mail par test =>_test_domains renvoie Option<String>
* [ ] Rustfmt
* [ ] Rustfmt
* [ ] Fonction check asychrone

View file

@ -6,9 +6,9 @@ extern crate log;
extern crate native_tls;
extern crate serde;
extern crate simplelog;
extern crate tokio;
extern crate toml;
extern crate trust_dns_resolver;
extern crate tokio;
use self::native_tls::{Protocol, TlsConnector};
use anyhow::{bail, Result};
@ -17,14 +17,15 @@ use lettre::smtp::authentication::Credentials;
use lettre::{ClientSecurity, ClientTlsParameters, SmtpClient, Transport};
use lettre_email::EmailBuilder;
use log::{error, info};
use query_external_ip::Consensus;
use serde::Deserialize;
use simplelog::{ConfigBuilder, LevelFilter, WriteLogger};
use std::fs::{self, OpenOptions};
use std::net::{Ipv4Addr, Ipv6Addr};
use std::string::String;
use std::{thread, time::Duration};
use trust_dns_resolver::{config::*, Name, Resolver};
use query_external_ip::Consensus;
use tokio::runtime::{Builder, Runtime};
use trust_dns_resolver::{config::*, Name, Resolver};
fn main() -> Result<()> {
// Init log
@ -82,7 +83,7 @@ fn main() -> Result<()> {
struct Tester {
config: Config,
dns_resolver: Resolver,
async_runtime: Runtime
async_runtime: Runtime,
}
impl Tester {
@ -91,11 +92,8 @@ impl Tester {
resolver_opt.ip_strategy = LookupIpStrategy::Ipv4AndIpv6;
Ok(Tester {
config: toml::from_str(&fs::read_to_string("config.toml")?)?,
dns_resolver: Resolver::new(ResolverConfig::default(), resolver_opt)?,
async_runtime: Builder::new_multi_thread()
.enable_all()
.build()
.unwrap()
dns_resolver: Resolver::from_system_conf()?,
async_runtime: Builder::new_multi_thread().enable_all().build().unwrap(),
})
}
@ -103,7 +101,10 @@ impl Tester {
info!("Check");
match self.get_ip() {
Ok((ipv4, ipv6)) => {
self.test_domains(ipv4, ipv6)?;
let msg = self.test_domains(ipv4, ipv6)?;
if !msg.is_empty() {
self.send_mail("IP check error".to_owned(), msg)?;
}
}
Err(err) => {
let msg = format!("Cannot get IP ({})", err);
@ -123,33 +124,48 @@ impl Tester {
Ok((ipv4, ipv6))
}
fn test_domains(&self, ipv4: Ipv4Addr, ipv6: Ipv6Addr) -> Result<()> {
fn test_domains(&self, ipv4: Ipv4Addr, ipv6: Ipv6Addr) -> Result<String> {
macro_rules! msg {
($full_msg:expr, $($arg:tt)*) => {{
let msg = format!($($arg)*);
info!("{}", msg);
$full_msg.push_str(&msg);
$full_msg.push('\n');
}}
}
let mut full_msg = String::new();
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)?;
}
if let Ok(dns_ipv4_lookup) = self.dns_resolver.ipv4_lookup(domain.clone()) {
let dns_ipv4 = dns_ipv4_lookup.iter().next().unwrap();
if *dns_ipv4 != ipv4 {
msg!(
full_msg,
"Wrong IPV4 for {} (DNS: {}, current: {})",
domain,
dns_ipv4,
ipv4
);
}
if dns_ip.is_ipv6() {
if dns_ip != ipv6 {
let msg = format!(
"Wrong IPV6 for {} (DNS: {}, current: {})",
domain, dns_ip, ipv6
);
info!("{}", msg);
self.send_mail("Wrong IP".to_owned(), msg)?;
}
} else {
msg!(full_msg, "Cannot get IPv4 address for {}", domain);
}
if let Ok(dns_ipv6_lookup) = self.dns_resolver.ipv6_lookup(domain.clone()) {
let dns_ipv6 = dns_ipv6_lookup.iter().next().unwrap();
if *dns_ipv6 != ipv6 {
msg!(
full_msg,
"Wrong IPV6 for {} (DNS: {}, current: {})",
domain,
dns_ipv6,
ipv6
);
}
} else {
msg!(full_msg, "Cannot get IPv6 address for {}", domain);
}
}
Ok(())
Ok(full_msg)
}
fn send_mail(&self, subject: String, body: String) -> Result<()> {