refator: copy cli into mono

src/services/valve_master_server/service.rs

@@ -1,155 +0,0 @@
-use crate::{
-    buffer::Buffer,
-    socket::{Socket, UdpSocket},
-    valve_master_server::{Region, SearchFilters},
-    GDErrorKind::PacketBad,
-    GDResult,
-};
-
-use std::net::{IpAddr, Ipv4Addr, SocketAddr};
-
-use byteorder::BigEndian;
-
-/// The default master ip, which is the one for Source.
-pub fn default_master_address() -> SocketAddr {
-    SocketAddr::new(IpAddr::V4(Ipv4Addr::new(208, 64, 201, 194)), 27011) // hl2master.steampowered.com
-}
-
-fn construct_payload(region: Region, filters: &Option<SearchFilters>, last_ip: &str, last_port: u16) -> Vec<u8> {
-    let filters_bytes: Vec<u8> = filters
-        .as_ref()
-        .map_or_else(|| vec![0x00], |f| f.to_bytes());
-
-    let region_byte = &[region as u8];
-
-    [
-        // Packet has to begin with the character '1'
-        &[0x31],
-        // The region byte is next
-        region_byte,
-        // The last fetched ip as a string
-        last_ip.as_bytes(),
-        // Followed by an ':'
-        &[b':'],
-        // And the port, as a string
-        last_port.to_string().as_bytes(),
-        // Which needs to end with a NULL byte
-        &[0x00],
-        // Then the filters
-        &filters_bytes,
-    ]
-    .concat()
-}
-
-/// The implementation, use this if you want to keep the same socket.
-pub struct ValveMasterServer {
-    socket: UdpSocket,
-}
-
-impl ValveMasterServer {
-    /// Construct a new struct.
-    pub fn new(master_address: &SocketAddr) -> GDResult<Self> {
-        let socket = UdpSocket::new(master_address)?;
-        socket.apply_timeout(&None)?;
-
-        Ok(Self { socket })
-    }
-
-    /// Make just a single query, providing `0.0.0.0` as the last ip and `0` as
-    /// the last port will give the initial packet.
-    pub fn query_specific(
-        &mut self,
-        region: Region,
-        search_filters: &Option<SearchFilters>,
-        last_address_ip: &str,
-        last_address_port: u16,
-    ) -> GDResult<Vec<(IpAddr, u16)>> {
-        let payload = construct_payload(region, search_filters, last_address_ip, last_address_port);
-        self.socket.send(&payload)?;
-
-        let received_data = self.socket.receive(Some(1400))?;
-        let mut buf = Buffer::<BigEndian>::new(&received_data);
-
-        if buf.read::<u32>()? != u32::MAX || buf.read::<u16>()? != 26122 {
-            return Err(PacketBad.context("Expected 4294967295 followed by 26122"));
-        }
-
-        let mut ips: Vec<(IpAddr, u16)> = Vec::new();
-
-        while buf.remaining_length() > 0 {
-            let ip = IpAddr::V4(Ipv4Addr::new(
-                buf.read::<u8>()?,
-                buf.read::<u8>()?,
-                buf.read::<u8>()?,
-                buf.read::<u8>()?,
-            ));
-            let port = buf.read::<u16>()?;
-
-            ips.push((ip, port));
-        }
-
-        Ok(ips)
-    }
-
-    /// Make a complete query.
-    pub fn query(&mut self, region: Region, search_filters: Option<SearchFilters>) -> GDResult<Vec<(IpAddr, u16)>> {
-        let mut ips: Vec<(IpAddr, u16)> = Vec::new();
-
-        let mut exit_fetching = false;
-        let mut last_ip: String = "0.0.0.0".to_string();
-        let mut last_port: u16 = 0;
-
-        while !exit_fetching {
-            let new_ips = self.query_specific(region, &search_filters, last_ip.as_str(), last_port)?;
-
-            match new_ips.last() {
-                None => exit_fetching = true,
-                Some((latest_ip, latest_port)) => {
-                    let mut remove_last = false;
-
-                    let latest_ip_string = latest_ip.to_string();
-                    if latest_ip_string == "0.0.0.0" && *latest_port == 0 {
-                        exit_fetching = true;
-                        remove_last = true;
-                    } else if latest_ip_string == last_ip && *latest_port == last_port {
-                        exit_fetching = true;
-                    } else {
-                        last_ip = latest_ip_string;
-                        last_port = *latest_port;
-                    }
-
-                    ips.extend(new_ips);
-                    if remove_last {
-                        ips.pop();
-                    }
-                }
-            }
-        }
-
-        Ok(ips)
-    }
-}
-
-/// Take only the first response of (what would be a) complete query. This is
-/// faster as it results in less packets being sent, received and processed but
-/// yields less ips.
-pub fn query_singular(region: Region, search_filters: Option<SearchFilters>) -> GDResult<Vec<(IpAddr, u16)>> {
-    let mut master_server = ValveMasterServer::new(&default_master_address())?;
-
-    let mut ips = master_server.query_specific(region, &search_filters, "0.0.0.0", 0)?;
-
-    if let Some((last_ip, last_port)) = ips.last() {
-        if last_ip.to_string() == "0.0.0.0" && *last_port == 0 {
-            ips.pop();
-        }
-    }
-
-    Ok(ips)
-}
-
-/// Make a complete query.
-pub fn query(region: Region, search_filters: Option<SearchFilters>) -> GDResult<Vec<(IpAddr, u16)>> {
-    let mut master_server = ValveMasterServer::new(&default_master_address())?;
-
-    master_server.query(region, search_filters)
-}