Denis Drakhnia
2 months ago
12 changed files with 711 additions and 639 deletions
@ -0,0 +1,305 @@ |
|||||||
|
// SPDX-License-Identifier: LGPL-3.0-only
|
||||||
|
// SPDX-FileCopyrightText: 2023 Denis Drakhnia <numas13@gmail.com>
|
||||||
|
|
||||||
|
#![cfg_attr(not(feature = "net"), allow(dead_code))] |
||||||
|
|
||||||
|
use core::{mem, str}; |
||||||
|
|
||||||
|
#[cfg(feature = "alloc")] |
||||||
|
use alloc::{borrow::ToOwned, boxed::Box, string::String}; |
||||||
|
|
||||||
|
use crate::{color, wrappers::Str}; |
||||||
|
|
||||||
|
use super::{CursorError, Result}; |
||||||
|
|
||||||
|
pub trait GetKeyValue<'a>: Sized { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self>; |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> GetKeyValue<'a> for &'a [u8] { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
cur.get_key_value_raw() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> GetKeyValue<'a> for Str<&'a [u8]> { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
cur.get_key_value_raw().map(Str) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> GetKeyValue<'a> for &'a str { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
let raw = cur.get_key_value_raw()?; |
||||||
|
str::from_utf8(raw).map_err(|_| CursorError::InvalidString) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#[cfg(feature = "alloc")] |
||||||
|
impl<'a> GetKeyValue<'a> for Box<str> { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
let raw = cur.get_key_value_raw()?; |
||||||
|
str::from_utf8(raw) |
||||||
|
.map(|s| s.to_owned().into_boxed_str()) |
||||||
|
.map_err(|_| CursorError::InvalidString) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#[cfg(feature = "alloc")] |
||||||
|
impl<'a> GetKeyValue<'a> for String { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
let raw = cur.get_key_value_raw()?; |
||||||
|
str::from_utf8(raw) |
||||||
|
.map(|s| s.to_owned()) |
||||||
|
.map_err(|_| CursorError::InvalidString) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> GetKeyValue<'a> for bool { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
match cur.get_key_value_raw()? { |
||||||
|
b"0" => Ok(false), |
||||||
|
b"1" => Ok(true), |
||||||
|
_ => Err(CursorError::InvalidBool), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl GetKeyValue<'_> for crate::server_info::Region { |
||||||
|
fn get_key_value(cur: &mut Cursor) -> Result<Self, CursorError> { |
||||||
|
cur.get_key_value::<u8>()?.try_into() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl GetKeyValue<'_> for crate::server_info::ServerType { |
||||||
|
fn get_key_value(cur: &mut Cursor) -> Result<Self, CursorError> { |
||||||
|
cur.get_key_value_raw()?.try_into() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
macro_rules! impl_get_value { |
||||||
|
($($t:ty),+ $(,)?) => { |
||||||
|
$(impl<'a> GetKeyValue<'a> for $t { |
||||||
|
fn get_key_value(cur: &mut Cursor<'a>) -> Result<Self> { |
||||||
|
let s = cur.get_key_value::<&str>()?; |
||||||
|
// HACK: special case for one asshole
|
||||||
|
let (_, s) = color::trim_start_color(s); |
||||||
|
s.parse().map_err(|_| CursorError::InvalidNumber) |
||||||
|
} |
||||||
|
})+ |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
impl_get_value! { |
||||||
|
u8, |
||||||
|
u16, |
||||||
|
u32, |
||||||
|
u64, |
||||||
|
|
||||||
|
i8, |
||||||
|
i16, |
||||||
|
i32, |
||||||
|
i64, |
||||||
|
} |
||||||
|
|
||||||
|
// TODO: impl GetKeyValue for f32 and f64
|
||||||
|
|
||||||
|
#[derive(Copy, Clone)] |
||||||
|
pub struct Cursor<'a> { |
||||||
|
buffer: &'a [u8], |
||||||
|
} |
||||||
|
|
||||||
|
macro_rules! impl_get { |
||||||
|
($($n:ident: $t:ty = $f:ident),+ $(,)?) => ( |
||||||
|
$(#[inline] |
||||||
|
pub fn $n(&mut self) -> Result<$t> { |
||||||
|
const N: usize = mem::size_of::<$t>(); |
||||||
|
self.get_array::<N>().map(<$t>::$f) |
||||||
|
})+ |
||||||
|
); |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> Cursor<'a> { |
||||||
|
pub fn new(buffer: &'a [u8]) -> Self { |
||||||
|
Self { buffer } |
||||||
|
} |
||||||
|
|
||||||
|
pub fn end(self) -> &'a [u8] { |
||||||
|
self.buffer |
||||||
|
} |
||||||
|
|
||||||
|
pub fn as_slice(&'a self) -> &'a [u8] { |
||||||
|
self.buffer |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn remaining(&self) -> usize { |
||||||
|
self.buffer.len() |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn has_remaining(&self) -> bool { |
||||||
|
self.remaining() != 0 |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_bytes(&mut self, count: usize) -> Result<&'a [u8]> { |
||||||
|
if count <= self.remaining() { |
||||||
|
let (head, tail) = self.buffer.split_at(count); |
||||||
|
self.buffer = tail; |
||||||
|
Ok(head) |
||||||
|
} else { |
||||||
|
Err(CursorError::UnexpectedEnd) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn advance(&mut self, count: usize) -> Result<()> { |
||||||
|
self.get_bytes(count).map(|_| ()) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_array<const N: usize>(&mut self) -> Result<[u8; N]> { |
||||||
|
self.get_bytes(N).map(|s| { |
||||||
|
let mut array = [0; N]; |
||||||
|
array.copy_from_slice(s); |
||||||
|
array |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_str(&mut self, n: usize) -> Result<&'a str> { |
||||||
|
let mut cur = *self; |
||||||
|
let s = cur |
||||||
|
.get_bytes(n) |
||||||
|
.and_then(|s| str::from_utf8(s).map_err(|_| CursorError::InvalidString))?; |
||||||
|
*self = cur; |
||||||
|
Ok(s) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_cstr(&mut self) -> Result<Str<&'a [u8]>> { |
||||||
|
let pos = self |
||||||
|
.buffer |
||||||
|
.iter() |
||||||
|
.position(|&c| c == b'\0') |
||||||
|
.ok_or(CursorError::UnexpectedEnd)?; |
||||||
|
let (head, tail) = self.buffer.split_at(pos); |
||||||
|
self.buffer = &tail[1..]; |
||||||
|
Ok(Str(&head[..pos])) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_cstr_as_str(&mut self) -> Result<&'a str> { |
||||||
|
str::from_utf8(&self.get_cstr()?).map_err(|_| CursorError::InvalidString) |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn get_u8(&mut self) -> Result<u8> { |
||||||
|
self.get_array::<1>().map(|s| s[0]) |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn get_i8(&mut self) -> Result<i8> { |
||||||
|
self.get_array::<1>().map(|s| s[0] as i8) |
||||||
|
} |
||||||
|
|
||||||
|
impl_get! { |
||||||
|
get_u16_le: u16 = from_le_bytes, |
||||||
|
get_u32_le: u32 = from_le_bytes, |
||||||
|
get_u64_le: u64 = from_le_bytes, |
||||||
|
get_i16_le: i16 = from_le_bytes, |
||||||
|
get_i32_le: i32 = from_le_bytes, |
||||||
|
get_i64_le: i64 = from_le_bytes, |
||||||
|
get_f32_le: f32 = from_le_bytes, |
||||||
|
get_f64_le: f64 = from_le_bytes, |
||||||
|
|
||||||
|
get_u16_be: u16 = from_be_bytes, |
||||||
|
get_u32_be: u32 = from_be_bytes, |
||||||
|
get_u64_be: u64 = from_be_bytes, |
||||||
|
get_i16_be: i16 = from_be_bytes, |
||||||
|
get_i32_be: i32 = from_be_bytes, |
||||||
|
get_i64_be: i64 = from_be_bytes, |
||||||
|
get_f32_be: f32 = from_be_bytes, |
||||||
|
get_f64_be: f64 = from_be_bytes, |
||||||
|
|
||||||
|
get_u16_ne: u16 = from_ne_bytes, |
||||||
|
get_u32_ne: u32 = from_ne_bytes, |
||||||
|
get_u64_ne: u64 = from_ne_bytes, |
||||||
|
get_i16_ne: i16 = from_ne_bytes, |
||||||
|
get_i32_ne: i32 = from_ne_bytes, |
||||||
|
get_i64_ne: i64 = from_ne_bytes, |
||||||
|
get_f32_ne: f32 = from_ne_bytes, |
||||||
|
get_f64_ne: f64 = from_ne_bytes, |
||||||
|
} |
||||||
|
|
||||||
|
pub fn expect(&mut self, s: &[u8]) -> Result<()> { |
||||||
|
if self.buffer.starts_with(s) { |
||||||
|
self.advance(s.len())?; |
||||||
|
Ok(()) |
||||||
|
} else { |
||||||
|
Err(CursorError::Expect) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn expect_empty(&self) -> Result<()> { |
||||||
|
if self.has_remaining() { |
||||||
|
Err(CursorError::ExpectEmpty) |
||||||
|
} else { |
||||||
|
Ok(()) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn take_while<F>(&mut self, mut cond: F) -> Result<&'a [u8]> |
||||||
|
where |
||||||
|
F: FnMut(u8) -> bool, |
||||||
|
{ |
||||||
|
self.buffer |
||||||
|
.iter() |
||||||
|
.position(|&i| !cond(i)) |
||||||
|
.ok_or(CursorError::UnexpectedEnd) |
||||||
|
.and_then(|n| self.get_bytes(n)) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn take_while_or_all<F>(&mut self, cond: F) -> &'a [u8] |
||||||
|
where |
||||||
|
F: FnMut(u8) -> bool, |
||||||
|
{ |
||||||
|
self.take_while(cond).unwrap_or_else(|_| { |
||||||
|
let (head, tail) = self.buffer.split_at(self.buffer.len()); |
||||||
|
self.buffer = tail; |
||||||
|
head |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_key_value_raw(&mut self) -> Result<&'a [u8]> { |
||||||
|
let mut cur = *self; |
||||||
|
match cur.get_u8()? { |
||||||
|
b'\\' => { |
||||||
|
let value = cur.take_while_or_all(|c| c != b'\\' && c != b'\n'); |
||||||
|
*self = cur; |
||||||
|
Ok(value) |
||||||
|
} |
||||||
|
_ => Err(CursorError::InvalidTableValue), |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_key_value<T: GetKeyValue<'a>>(&mut self) -> Result<T> { |
||||||
|
T::get_key_value(self) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn skip_key_value<T: GetKeyValue<'a>>(&mut self) -> Result<()> { |
||||||
|
T::get_key_value(self).map(|_| ()) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_key_raw(&mut self) -> Result<&'a [u8]> { |
||||||
|
let mut cur = *self; |
||||||
|
match cur.get_u8() { |
||||||
|
Ok(b'\\') => { |
||||||
|
let value = cur.take_while(|c| c != b'\\' && c != b'\n')?; |
||||||
|
*self = cur; |
||||||
|
Ok(value) |
||||||
|
} |
||||||
|
Ok(b'\n') | Err(CursorError::UnexpectedEnd) => Err(CursorError::TableEnd), |
||||||
|
_ => Err(CursorError::InvalidTableKey), |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn get_key<T: GetKeyValue<'a>>(&mut self) -> Result<(&'a [u8], T)> { |
||||||
|
Ok((self.get_key_raw()?, self.get_key_value()?)) |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,207 @@ |
|||||||
|
// SPDX-License-Identifier: LGPL-3.0-only
|
||||||
|
// SPDX-FileCopyrightText: 2023 Denis Drakhnia <numas13@gmail.com>
|
||||||
|
|
||||||
|
#![cfg_attr(not(feature = "net"), allow(dead_code))] |
||||||
|
|
||||||
|
use core::{ |
||||||
|
fmt::{self, Write}, |
||||||
|
str, |
||||||
|
}; |
||||||
|
|
||||||
|
use super::{CursorError, Result}; |
||||||
|
|
||||||
|
pub trait PutKeyValue { |
||||||
|
fn put_key_value<'a, 'b>(&self, cur: &'b mut CursorMut<'a>) -> Result<&'b mut CursorMut<'a>>; |
||||||
|
} |
||||||
|
|
||||||
|
impl<T> PutKeyValue for &T |
||||||
|
where |
||||||
|
T: PutKeyValue, |
||||||
|
{ |
||||||
|
fn put_key_value<'a, 'b>(&self, cur: &'b mut CursorMut<'a>) -> Result<&'b mut CursorMut<'a>> { |
||||||
|
(*self).put_key_value(cur) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl PutKeyValue for &str { |
||||||
|
fn put_key_value<'a, 'b>(&self, cur: &'b mut CursorMut<'a>) -> Result<&'b mut CursorMut<'a>> { |
||||||
|
cur.put_str(self) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl PutKeyValue for bool { |
||||||
|
fn put_key_value<'a, 'b>(&self, cur: &'b mut CursorMut<'a>) -> Result<&'b mut CursorMut<'a>> { |
||||||
|
cur.put_u8(if *self { b'1' } else { b'0' }) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl PutKeyValue for crate::server_info::ServerType { |
||||||
|
fn put_key_value<'a, 'b>( |
||||||
|
&self, |
||||||
|
cur: &'b mut CursorMut<'a>, |
||||||
|
) -> Result<&'b mut CursorMut<'a>, CursorError> { |
||||||
|
match self { |
||||||
|
Self::Dedicated => cur.put_str("d"), |
||||||
|
Self::Local => cur.put_str("l"), |
||||||
|
Self::Proxy => cur.put_str("p"), |
||||||
|
Self::Unknown => cur.put_str("?"), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
macro_rules! impl_put_key_value { |
||||||
|
($($t:ty),+ $(,)?) => { |
||||||
|
$(impl PutKeyValue for $t { |
||||||
|
fn put_key_value<'a, 'b>(&self, cur: &'b mut CursorMut<'a>) -> Result<&'b mut CursorMut<'a>> { |
||||||
|
cur.put_as_str(self) |
||||||
|
} |
||||||
|
})+ |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
impl_put_key_value! { |
||||||
|
u8, |
||||||
|
u16, |
||||||
|
u32, |
||||||
|
u64, |
||||||
|
|
||||||
|
i8, |
||||||
|
i16, |
||||||
|
i32, |
||||||
|
i64, |
||||||
|
|
||||||
|
f32, |
||||||
|
f64, |
||||||
|
} |
||||||
|
|
||||||
|
pub struct CursorMut<'a> { |
||||||
|
buffer: &'a mut [u8], |
||||||
|
pos: usize, |
||||||
|
} |
||||||
|
|
||||||
|
macro_rules! impl_put { |
||||||
|
($($n:ident: $t:ty = $f:ident),+ $(,)?) => ( |
||||||
|
$(#[inline] |
||||||
|
pub fn $n(&mut self, n: $t) -> Result<&mut Self> { |
||||||
|
self.put_array(&n.$f()) |
||||||
|
})+ |
||||||
|
); |
||||||
|
} |
||||||
|
|
||||||
|
impl<'a> CursorMut<'a> { |
||||||
|
pub fn new(buffer: &'a mut [u8]) -> Self { |
||||||
|
Self { buffer, pos: 0 } |
||||||
|
} |
||||||
|
|
||||||
|
pub fn pos(&mut self) -> usize { |
||||||
|
self.pos |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn available(&self) -> usize { |
||||||
|
self.buffer.len() - self.pos |
||||||
|
} |
||||||
|
|
||||||
|
pub fn advance<F>(&mut self, count: usize, mut f: F) -> Result<&mut Self> |
||||||
|
where |
||||||
|
F: FnMut(&mut [u8]), |
||||||
|
{ |
||||||
|
if count <= self.available() { |
||||||
|
f(&mut self.buffer[self.pos..self.pos + count]); |
||||||
|
self.pos += count; |
||||||
|
Ok(self) |
||||||
|
} else { |
||||||
|
Err(CursorError::UnexpectedEnd) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_bytes(&mut self, s: &[u8]) -> Result<&mut Self> { |
||||||
|
self.advance(s.len(), |i| { |
||||||
|
i.copy_from_slice(s); |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_array<const N: usize>(&mut self, s: &[u8; N]) -> Result<&mut Self> { |
||||||
|
self.advance(N, |i| { |
||||||
|
i.copy_from_slice(s); |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_str(&mut self, s: &str) -> Result<&mut Self> { |
||||||
|
self.put_bytes(s.as_bytes()) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_cstr(&mut self, s: &str) -> Result<&mut Self> { |
||||||
|
self.put_str(s)?.put_u8(0) |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn put_u8(&mut self, n: u8) -> Result<&mut Self> { |
||||||
|
self.put_array(&[n]) |
||||||
|
} |
||||||
|
|
||||||
|
#[inline(always)] |
||||||
|
pub fn put_i8(&mut self, n: i8) -> Result<&mut Self> { |
||||||
|
self.put_u8(n as u8) |
||||||
|
} |
||||||
|
|
||||||
|
impl_put! { |
||||||
|
put_u16_le: u16 = to_le_bytes, |
||||||
|
put_u32_le: u32 = to_le_bytes, |
||||||
|
put_u64_le: u64 = to_le_bytes, |
||||||
|
put_i16_le: i16 = to_le_bytes, |
||||||
|
put_i32_le: i32 = to_le_bytes, |
||||||
|
put_i64_le: i64 = to_le_bytes, |
||||||
|
put_f32_le: f32 = to_le_bytes, |
||||||
|
put_f64_le: f64 = to_le_bytes, |
||||||
|
|
||||||
|
put_u16_be: u16 = to_be_bytes, |
||||||
|
put_u32_be: u32 = to_be_bytes, |
||||||
|
put_u64_be: u64 = to_be_bytes, |
||||||
|
put_i16_be: i16 = to_be_bytes, |
||||||
|
put_i32_be: i32 = to_be_bytes, |
||||||
|
put_i64_be: i64 = to_be_bytes, |
||||||
|
put_f32_be: f32 = to_be_bytes, |
||||||
|
put_f64_be: f64 = to_be_bytes, |
||||||
|
|
||||||
|
put_u16_ne: u16 = to_ne_bytes, |
||||||
|
put_u32_ne: u32 = to_ne_bytes, |
||||||
|
put_u64_ne: u64 = to_ne_bytes, |
||||||
|
put_i16_ne: i16 = to_ne_bytes, |
||||||
|
put_i32_ne: i32 = to_ne_bytes, |
||||||
|
put_i64_ne: i64 = to_ne_bytes, |
||||||
|
put_f32_ne: f32 = to_ne_bytes, |
||||||
|
put_f64_ne: f64 = to_ne_bytes, |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_as_str<T: fmt::Display>(&mut self, value: T) -> Result<&mut Self> { |
||||||
|
write!(self, "{}", value).map_err(|_| CursorError::UnexpectedEnd)?; |
||||||
|
Ok(self) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_key_value<T: PutKeyValue>(&mut self, value: T) -> Result<&mut Self> { |
||||||
|
value.put_key_value(self) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_key_raw(&mut self, key: &str, value: &[u8]) -> Result<&mut Self> { |
||||||
|
self.put_u8(b'\\')? |
||||||
|
.put_str(key)? |
||||||
|
.put_u8(b'\\')? |
||||||
|
.put_bytes(value) |
||||||
|
} |
||||||
|
|
||||||
|
pub fn put_key<T: PutKeyValue>(&mut self, key: &str, value: T) -> Result<&mut Self> { |
||||||
|
self.put_u8(b'\\')? |
||||||
|
.put_str(key)? |
||||||
|
.put_u8(b'\\')? |
||||||
|
.put_key_value(value) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl fmt::Write for CursorMut<'_> { |
||||||
|
fn write_str(&mut self, s: &str) -> fmt::Result { |
||||||
|
self.put_bytes(s.as_bytes()) |
||||||
|
.map(|_| ()) |
||||||
|
.map_err(|_| fmt::Error) |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,6 @@ |
|||||||
|
//! Network packets decoders and encoders.
|
||||||
|
|
||||||
|
pub mod admin; |
||||||
|
pub mod game; |
||||||
|
pub mod master; |
||||||
|
pub mod server; |
Loading…
Reference in new issue