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
126
127
128
|
#![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 FooData {}
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Foo {
#[cfg_attr(feature = "serde", serde(flatten))]
foo: FooData,
}
#[derive(Debug)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct FooBuilder {}
impl FooData {
fn conforms(bytes: &[u8]) -> bool {
true
}
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> {
Ok(Self {})
}
fn write_to(&self, buffer: &mut BytesMut) {}
fn get_total_size(&self) -> usize {
self.get_size()
}
fn get_size(&self) -> usize {
0
}
}
impl Packet for Foo {
fn to_bytes(self) -> Bytes {
let mut buffer = BytesMut::with_capacity(self.foo.get_size());
self.foo.write_to(&mut buffer);
buffer.freeze()
}
fn to_vec(self) -> Vec<u8> {
self.to_bytes().to_vec()
}
}
impl From<Foo> for Bytes {
fn from(packet: Foo) -> Self {
packet.to_bytes()
}
}
impl From<Foo> for Vec<u8> {
fn from(packet: Foo) -> Self {
packet.to_vec()
}
}
impl Foo {
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> {
let data = FooData::parse_inner(&mut bytes)?;
Self::new(data)
}
fn new(foo: FooData) -> Result<Self> {
Ok(Self { foo })
}
fn write_to(&self, buffer: &mut BytesMut) {
self.foo.write_to(buffer)
}
pub fn get_size(&self) -> usize {
self.foo.get_size()
}
}
impl FooBuilder {
pub fn build(self) -> Foo {
let foo = FooData {};
Foo::new(foo).unwrap()
}
}
impl From<FooBuilder> for Foo {
fn from(builder: FooBuilder) -> Foo {
builder.build().into()
}
}
|
