use std::net::{SocketAddr, TcpListener, TcpStream, ToSocketAddrs};
use std::ops::Range;
pub fn scan_port(port: u16) -> bool {
match TcpStream::connect(("0.0.0.0", port)) {
Ok(_) => true,
Err(_) => false,
}
}
pub fn scan_port_addr<A: ToSocketAddrs>(addr: A) -> bool {
match TcpStream::connect(addr) {
Ok(_) => true,
Err(_) => false,
}
}
pub fn scan_ports(ports: Vec<u16>) -> Vec<u16> {
let mut open_ports = Vec::new();
for port in ports {
if scan_port(port) {
open_ports.push(port);
}
}
open_ports
}
pub fn scan_ports_addrs<A: ToSocketAddrs>(addrs: Vec<A>) -> Vec<SocketAddr> {
let mut open_addrs = Vec::new();
for addr in addrs {
if scan_port_addr(&addr) {
let addr = addr.to_socket_addrs().unwrap().next().unwrap();
open_addrs.push(addr);
}
}
open_addrs
}
pub fn scan_ports_range(port_range: Range<u16>) -> Vec<u16> {
let mut open_ports = Vec::new();
for port in port_range {
if scan_port(port) {
open_ports.push(port);
}
}
open_ports
}
pub fn request_open_port() -> Option<u16> {
match TcpListener::bind("0.0.0.0:0") {
Ok(a) => {
match a.local_addr() {
Ok(a) => Some(a.port()),
Err(_) => None,
}
}
Err(_) => None,
}
}
pub fn local_port_available(port: u16) -> bool {
match TcpListener::bind(("127.0.0.1", port)) {
Ok(_) => true,
Err(_) => false,
}
}
pub fn local_ports_available(ports: Vec<u16>) -> Vec<u16> {
let mut available = Vec::new();
for port in ports {
if local_port_available(port) {
available.push(port);
}
}
available
}
pub fn local_ports_available_range(port_range: Range<u16>) -> Vec<u16> {
let mut available = Vec::new();
for port in port_range {
if local_port_available(port) {
available.push(port);
}
}
available
}