use crate::implementation::types;
use crate::message;
#[derive(Debug)]
pub struct NewObject {
id: u32,
seq: u32,
data: [u8; 12],
}
impl NewObject {
pub fn new(seq: u32, id: u32) -> Self {
let mut data = [0u8; 12];
data[0] = message::MessageType::NewObject as u8;
data[1] = types::MessageMagic::TypeUint as u8;
data[2..6].copy_from_slice(&id.to_le_bytes());
data[6] = types::MessageMagic::TypeUint as u8;
data[7..11].copy_from_slice(&seq.to_le_bytes());
data[11] = types::MessageMagic::End as u8;
Self { id, seq, data }
}
pub fn id(&self) -> u32 {
self.id
}
pub fn seq(&self) -> u32 {
self.seq
}
pub fn from_bytes(data: &[u8], offset: usize) -> super::Result<Self> {
if *data.get(offset).ok_or(message::Error::UnexpectedEof)?
!= message::MessageType::NewObject as u8
{
return Err(message::Error::InvalidMessageType);
}
if *data.get(offset + 1).ok_or(message::Error::UnexpectedEof)?
!= types::MessageMagic::TypeUint as u8
{
return Err(message::Error::InvalidFieldType);
}
let bytes: [u8; 4] = data
.get(offset + 2..offset + 2 + size_of::<u32>())
.ok_or(message::Error::UnexpectedEof)?
.try_into()
.unwrap();
let id = u32::from_le_bytes(bytes);
if *data.get(offset + 6).ok_or(message::Error::UnexpectedEof)?
!= types::MessageMagic::TypeUint as u8
{
return Err(message::Error::InvalidFieldType);
}
let bytes: [u8; 4] = data
.get(offset + 7..offset + 7 + size_of::<u32>())
.ok_or(message::Error::UnexpectedEof)?
.try_into()
.unwrap();
let seq = u32::from_le_bytes(bytes);
if *data.get(offset + 11).ok_or(message::Error::UnexpectedEof)?
!= types::MessageMagic::End as u8
{
return Err(message::Error::MalformedMessage);
}
Ok(Self {
id,
seq,
data: data[offset..offset + 12].try_into().unwrap(),
})
}
}
impl message::Message for NewObject {
fn data(&self) -> &[u8] {
&self.data
}
fn message_type(&self) -> message::MessageType {
message::MessageType::NewObject
}
}
#[cfg(test)]
mod tests {
use super::*;
use message::Message;
#[test]
fn new_object_new() {
let msg = NewObject::new(3, 2);
let parsed = msg.parse_data();
assert_eq!(parsed, "NewObject ( 2, 3 ) ");
}
#[test]
fn new_object_from_bytes() {
let bytes: &[u8] = &[
message::MessageType::NewObject as u8,
types::MessageMagic::TypeUint as u8,
0x02,
0x00,
0x00,
0x00,
types::MessageMagic::TypeUint as u8,
0x03,
0x00,
0x00,
0x00,
types::MessageMagic::End as u8,
];
let msg = NewObject::from_bytes(bytes, 0).unwrap();
let parsed = msg.parse_data();
assert_eq!(parsed, "NewObject ( 2, 3 ) ");
}
#[test]
fn new_object_from_bytes_unexpected_eof() {
let bytes: &[u8] = &[
message::MessageType::NewObject as u8,
types::MessageMagic::TypeUint as u8,
];
let err = NewObject::from_bytes(bytes, 0).unwrap_err();
assert!(matches!(err, message::Error::UnexpectedEof));
}
#[test]
fn new_object_from_bytes_malformed() {
let bytes: &[u8] = &[
message::MessageType::NewObject as u8,
types::MessageMagic::TypeUint as u8,
0x02,
0x00,
0x00,
0x00,
types::MessageMagic::TypeUint as u8,
0x03,
0x00,
0x00,
0x00,
types::MessageMagic::TypeUint as u8,
];
let err = NewObject::from_bytes(bytes, 0).unwrap_err();
assert!(matches!(err, message::Error::MalformedMessage));
}
#[test]
fn new_object_roundtrip_parses_fields() {
let out = NewObject::new(123, 0xBEEF);
let in_msg = NewObject::from_bytes(out.data(), 0).unwrap();
assert_eq!(in_msg.data(), out.data());
assert_eq!(in_msg.seq(), 123);
assert_eq!(in_msg.id(), 0xBEEF);
}
}