1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#![rustfmt::skip]
/// @generated rust packets from test.
use bytes::{Buf, BufMut, Bytes, BytesMut};
use std::convert::{TryFrom, TryInto};
use std::cell::Cell;
use std::fmt;
use thiserror::Error;
type Result<T> = std::result::Result<T, Error>;
/// Private prevents users from creating arbitrary scalar values
/// in situations where the value needs to be validated.
/// Users can freely deref the value, but only the backend
/// may create it.
#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq, PartialOrd, Ord)]
pub struct Private<T>(T);
impl<T> std::ops::Deref for Private<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&self.0
}
}
#[derive(Debug, Error)]
pub enum Error {
#[error("Packet parsing failed")]
InvalidPacketError,
#[error("{field} was {value:x}, which is not known")]
ConstraintOutOfBounds { field: String, value: u64 },
#[error("Got {actual:x}, expected {expected:x}")]
InvalidFixedValue { expected: u64, actual: u64 },
#[error("when parsing {obj} needed length of {wanted} but got {got}")]
InvalidLengthError { obj: String, wanted: usize, got: usize },
#[error(
"array size ({array} bytes) is not a multiple of the element size ({element} bytes)"
)]
InvalidArraySize { array: usize, element: usize },
#[error("Due to size restrictions a struct could not be parsed.")]
ImpossibleStructError,
#[error("when parsing field {obj}.{field}, {value} is not a valid {type_} value")]
InvalidEnumValueError { obj: String, field: String, value: u64, type_: String },
#[error("expected child {expected}, got {actual}")]
InvalidChildError { expected: &'static str, actual: String },
}
pub trait Packet {
fn to_bytes(self) -> Bytes;
fn to_vec(self) -> Vec<u8>;
}
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Foo {
pub a: u8,
pub b: u8,
pub c: u8,
pub d: u32,
pub e: u16,
pub f: u8,
}
impl Foo {
fn conforms(bytes: &[u8]) -> bool {
bytes.len() >= 7
}
pub fn parse(bytes: &[u8]) -> Result<Self> {
let mut cell = Cell::new(bytes);
let packet = Self::parse_inner(&mut cell)?;
Ok(packet)
}
fn parse_inner(mut bytes: &mut Cell<&[u8]>) -> Result<Self> {
if bytes.get().remaining() < 2 {
return Err(Error::InvalidLengthError {
obj: "Foo".to_string(),
wanted: 2,
got: bytes.get().remaining(),
});
}
let chunk = bytes.get_mut().get_u16();
let a = (chunk & 0x7) as u8;
let b = (chunk >> 3) as u8;
let c = ((chunk >> 11) & 0x1f) as u8;
if bytes.get().remaining() < 3 {
return Err(Error::InvalidLengthError {
obj: "Foo".to_string(),
wanted: 3,
got: bytes.get().remaining(),
});
}
let d = bytes.get_mut().get_uint(3) as u32;
if bytes.get().remaining() < 2 {
return Err(Error::InvalidLengthError {
obj: "Foo".to_string(),
wanted: 2,
got: bytes.get().remaining(),
});
}
let chunk = bytes.get_mut().get_u16();
let e = (chunk & 0xfff);
let f = ((chunk >> 12) & 0xf) as u8;
Ok(Self { a, b, c, d, e, f })
}
fn write_to(&self, buffer: &mut BytesMut) {
if self.a > 0x7 {
panic!("Invalid value for {}::{}: {} > {}", "Foo", "a", self.a, 0x7);
}
if self.c > 0x1f {
panic!("Invalid value for {}::{}: {} > {}", "Foo", "c", self.c, 0x1f);
}
let value = (self.a as u16) | ((self.b as u16) << 3) | ((self.c as u16) << 11);
buffer.put_u16(value);
if self.d > 0xff_ffff {
panic!("Invalid value for {}::{}: {} > {}", "Foo", "d", self.d, 0xff_ffff);
}
buffer.put_uint(self.d as u64, 3);
if self.e > 0xfff {
panic!("Invalid value for {}::{}: {} > {}", "Foo", "e", self.e, 0xfff);
}
if self.f > 0xf {
panic!("Invalid value for {}::{}: {} > {}", "Foo", "f", self.f, 0xf);
}
let value = self.e | ((self.f as u16) << 12);
buffer.put_u16(value);
}
fn get_total_size(&self) -> usize {
self.get_size()
}
fn get_size(&self) -> usize {
7
}
}
|
