use std::net::SocketAddrV4;
use std::env;
extern crate ig;
fn main() {
match ig::search_gateway() {
Err(ref err) => match *err {
ig::SearchError::IoError(ref ioe) => println!("IoError: {}", ioe),
_ => println!("{:?}", err),
},
Ok(gateway) => {
let args: Vec<_> = env::args().collect();
if args.len() != 4 {
println!("Usage: add_remove <local_ip> <local_port> <remote_port>");
return;
}
let local_ip = args[1].parse().expect("Invalid IP address");
let local_port = args[2].parse().expect("Invalid local port");
let remote_port = args[3].parse().expect("Invalid remote port");
let local_addr = SocketAddrV4::new(local_ip, local_port);
match gateway.add_port(ig::PortMappingProtocol::TCP, remote_port,
local_addr, 60, "crust") {
Err(ref err) => println!("{:?}", err),
Ok(()) => {
println!("AddPortMapping successful.");
match gateway.remove_port(ig::PortMappingProtocol::TCP, remote_port) {
Err(ref err) => println!("Error removing: {:?}", err),
Ok(_) => println!("DeletePortMapping successful."),
}
},
}
},
}
}