tribufu/rust-gamedig
1
0
Fork 0
mirror of https://github.com/tribufu/rust-gamedig synced 2026-05-06 15:27:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Valve PLAYERS and RULES request first implementations

Browse source
This commit is contained in:
CosminPerRam 2022-10-18 01:40:37 +03:00
parent 9ab4b8a7fd
commit 92d9649008
3 changed files with 198 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

62
src/utils.rs
View file

@ -9,14 +9,6 @@ pub fn complete_address(address: &str, port: u16) -> String {
String::from(address.to_owned() + ":").add(&*port.to_string())
}
pub fn combine_two_u8_le(high: u8, low: u8) -> u16 {
u16::from_be_bytes([high, low])
}
pub fn combine_eight_u8_le(a: u8, b: u8, c: u8, d: u8, e: u8, f: u8, g: u8, h: u8) -> u64 {
u64::from_be_bytes([a, b, c, d, e, f, g, h])
}
pub mod buffer {
use super::*;
@ -35,17 +27,37 @@ pub mod buffer {
return Err(GDError::PacketUnderflow("Unexpectedly short packet.".to_string()));
}
let value = combine_two_u8_le(buf[*pos + 1], buf[*pos]);
let value = u16::from_le_bytes([buf[*pos], buf[*pos + 1]]);
*pos += 2;
Ok(value)
}
pub fn get_u32_le(buf: &[u8], pos: &mut usize) -> Result<u32, GDError> {
if buf.len() <= *pos + 3 {
return Err(GDError::PacketUnderflow("Unexpectedly short packet.".to_string()));
}
let value = u32::from_le_bytes([buf[*pos], buf[*pos + 1], buf[*pos + 2], buf[*pos + 3]]);
*pos += 4;
Ok(value)
}
pub fn get_f32_le(buf: &[u8], pos: &mut usize) -> Result<f32, GDError> {
if buf.len() <= *pos + 3 {
return Err(GDError::PacketUnderflow("Unexpectedly short packet.".to_string()));
}
let value = f32::from_le_bytes([buf[*pos], buf[*pos + 1], buf[*pos + 2], buf[*pos + 3]]);
*pos += 4;
Ok(value)
}
pub fn get_u64_le(buf: &[u8], pos: &mut usize) -> Result<u64, GDError> {
if buf.len() <= *pos + 7 {
return Err(GDError::PacketUnderflow("Unexpectedly short packet.".to_string()));
}
let value = combine_eight_u8_le(buf[*pos + 7], buf[*pos + 6], buf[*pos + 5], buf[*pos + 4], buf[*pos + 3], buf[*pos + 2], buf[*pos + 1], buf[*pos]);
let value = u64::from_le_bytes([buf[*pos], buf[*pos + 1], buf[*pos + 2], buf[*pos + 3], buf[*pos + 4], buf[*pos + 5], buf[*pos + 6], buf[*pos + 7]]);
*pos += 8;
Ok(value)
}
@ -81,16 +93,6 @@ mod tests {
assert_eq!(complete_address(address, port), "192.168.0.1:27015");
}
#[test]
fn combine_two_u8_le_test() {
assert_eq!(combine_two_u8_le(49, 123), 12667);
}
#[test]
fn combine_eight_u8_le_test() {
assert_eq!(combine_eight_u8_le(49, 123, 44, 52, 3, 250, 22, 0), 3565492131910522368);
}
#[test]
fn get_u8_test() {
let data = [72];
@ -111,6 +113,26 @@ mod tests {
assert_eq!(pos, 2);
}
#[test]
fn get_u32_le_test() {
let data = [72, 29, 128, 100];
let mut pos = 0;
assert_eq!(buffer::get_u32_le(&data, &mut pos).unwrap(), 1686117704);
assert_eq!(pos, 4);
assert!(buffer::get_u32_le(&data, &mut pos).is_err());
assert_eq!(pos, 4);
}
#[test]
fn get_f32_le_test() {
let data = [72, 29, 128, 100];
let mut pos = 0;
assert_eq!(buffer::get_f32_le(&data, &mut pos).unwrap(), 1.8906345e22);
assert_eq!(pos, 4);
assert!(buffer::get_f32_le(&data, &mut pos).is_err());
assert_eq!(pos, 4);
}
#[test]
fn get_u64_le_test() {
let data = [72, 29, 128, 99, 69, 4, 2, 0];