2020-02-12 18:14:02 +00:00
|
|
|
use std::env;
|
|
|
|
use std::net::SocketAddrV4;
|
|
|
|
|
2020-02-12 17:33:10 +00:00
|
|
|
use igd::aio::search_gateway;
|
2020-02-12 18:14:02 +00:00
|
|
|
use igd::PortMappingProtocol;
|
|
|
|
|
|
|
|
struct DiplonatConfig<'a> {
|
|
|
|
private_ip: &'a str,
|
|
|
|
refresh_time: u16,
|
|
|
|
expiration_time: u16
|
|
|
|
}
|
|
|
|
|
|
|
|
fn fetch_configuration {
|
|
|
|
let env_private_ip = "DIPLONAT_PRIVATE_IP";
|
|
|
|
let private_ip = match env::var(env_private_ip) {
|
|
|
|
Ok(val) => val,
|
|
|
|
Err(e) => return println!("Unable to fetch {} environment variable: {}", env_private_ip, e),
|
|
|
|
};
|
|
|
|
|
|
|
|
}
|
2020-02-12 17:33:10 +00:00
|
|
|
|
|
|
|
#[tokio::main]
|
|
|
|
async fn main() {
|
2020-02-12 18:14:02 +00:00
|
|
|
|
|
|
|
println!("Private IP address: {}", private_ip);
|
|
|
|
|
2020-02-12 17:33:10 +00:00
|
|
|
let gateway = match search_gateway(Default::default()).await {
|
|
|
|
Ok(g) => g,
|
|
|
|
Err(err) => return println!("Faild to find IGD: {}", err),
|
|
|
|
};
|
|
|
|
|
2020-02-12 18:14:02 +00:00
|
|
|
let service = format!("{}:{}", private_ip, 1234);
|
|
|
|
let service: SocketAddrV4 = private_ip.parse().expect("Invalid socket address");
|
|
|
|
match gateway.add_port(PortMappingProtocol::TCP, 1234, service, 120, "diplonat").await {
|
|
|
|
Ok(_) => (),
|
|
|
|
Err(e) => return println!("Unable to insert port 1234: {}", e),
|
2020-02-12 17:33:10 +00:00
|
|
|
};
|
|
|
|
}
|