astatin3/unshell
1
0
Fork 0
mirror of https://github.com/Astatin3/unshell.git synced 2026-06-08 22:38:01 -06:00
Code Issues Packages Projects Releases Wiki Activity

Optimize framed packet encoding

Browse Source
This commit is contained in:
Michael Mikovsky
2026-04-25 19:03:08 -06:00
parent 7bea3e2b6b
commit 25988de1e1
1 changed files with 9 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/protocol/codec.rs
+9 -16
View File
@@ -93,16 +93,15 @@ where
let payload_start = align_up(header_start + header_bytes.len(), SECTION_ALIGN);
let total_len = payload_start + payload_bytes.len();
let mut frame = FrameBytes::with_capacity(total_len);
frame.extend_from_slice(&header_len.to_be_bytes());
frame.extend_from_slice(&payload_len.to_be_bytes());
append_padding(&mut frame, header_start - 8usize);
frame.extend_from_slice(&header_bytes);
append_padding(
&mut frame,
payload_start - (header_start + header_bytes.len()),
);
frame.extend_from_slice(&payload_bytes);
let header_end = header_start + header_bytes.len();
let payload_end = payload_start + payload_bytes.len();
let mut frame = FrameBytes::new();
frame.resize(total_len, 0);
frame[0..4].copy_from_slice(&header_len.to_be_bytes());
frame[4..8].copy_from_slice(&payload_len.to_be_bytes());
frame[header_start..header_end].copy_from_slice(&header_bytes);
frame[payload_start..payload_end].copy_from_slice(&payload_bytes);
Ok(frame)
}
@@ -163,12 +162,6 @@ fn read_u32(bytes: &[u8], start: usize) -> Result<u32, FrameError> {
))
}
fn append_padding(frame: &mut AlignedVec, padding: usize) {
if padding > 0 {
frame.resize(frame.len() + padding, 0);
}
}
fn align_up(offset: usize, alignment: usize) -> usize {
let mask = alignment - 1;
(offset + mask) & !mask