vinzenz/servicepoint-simulator
0
0
Fork 0
mirror of https://github.com/kaesaecracker/servicepoint-simulator.git synced 2025-01-18 10:30:14 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

less unsafe code

Browse source
This commit is contained in:
Vinzenz Schroeter 2024-05-09 13:43:28 +02:00
parent 0b03c4d23a
commit 6bca63befd
2 changed files with 27 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/gui.rs
View file

@ -52,18 +52,14 @@ impl ApplicationHandler for App {
let mut i = 0;
for pixel in frame {
unsafe {
if i >= DISPLAY.len() {
break;
}
let color = if DISPLAY[i] {
let is_set = unsafe { DISPLAY[i] };
let color = if is_set {
[255u8, 255, 255, 255]
} else {
[0u8, 0, 0, 255]
};
pixel.copy_from_slice(&color);
}
i += 1;
}

27
src/upd_loop.rs
View file

@ -51,7 +51,6 @@ enum DisplayCommand {
CmdBitmapLinearXor = 0x0016,
}
#[repr(C)]
#[derive(Debug)]
struct HdrWindow {
@ -123,7 +122,7 @@ fn read_hdr_window(buffer: &[u8]) -> Result<HdrWindow, String> {
return Err("received a packet that is too small".into());
}
let command_u16 = u16::from_be(unsafe { std::ptr::read(buffer[0..=1].as_ptr() as *const u16) });
let command_u16 = read_beu16_from_buffer(&buffer[0..=1]);
let maybe_command = num::FromPrimitive::from_u16(command_u16);
if maybe_command.is_none() {
return Err(format!("received invalid command {}", command_u16));
@ -131,13 +130,26 @@ fn read_hdr_window(buffer: &[u8]) -> Result<HdrWindow, String> {
return Ok(HdrWindow {
command: maybe_command.unwrap(),
x: u16::from_be(unsafe { std::ptr::read(buffer[2..=3].as_ptr() as *const u16) }),
y: u16::from_be(unsafe { std::ptr::read(buffer[4..=5].as_ptr() as *const u16) }),
w: u16::from_be(unsafe { std::ptr::read(buffer[6..=7].as_ptr() as *const u16) }),
h: u16::from_be(unsafe { std::ptr::read(buffer[8..=9].as_ptr() as *const u16) }),
x: read_beu16_from_buffer(&buffer[2..=3]),
y: read_beu16_from_buffer(&buffer[4..=5]),
w: read_beu16_from_buffer(&buffer[6..=7]),
h: read_beu16_from_buffer(&buffer[8..=9]),
});
}
fn read_beu16_from_buffer(buffer: &[u8]) -> u16 {
assert_eq!(
buffer.len(),
2,
"cannot read u16 from buffer with size != 2"
);
let ptr = buffer.as_ptr() as *const u16;
let u16 = unsafe { *ptr };
return u16::from_be(u16);
}
fn check_payload_size(buf: &[u8], expected: usize) -> bool {
let actual = buf.len();
if actual == expected {
@ -179,9 +191,10 @@ fn print_bitmap_linear_win(header: &HdrWindow, payload: &[u8]) {
let translated_x = (x + header.x) as usize;
let translated_y = (y + header.y) as usize;
let index = translated_y * PIXEL_WIDTH as usize + translated_x;
unsafe {
DISPLAY[translated_y * PIXEL_WIDTH as usize + translated_x] = is_set;
DISPLAY[index] = is_set;
}
}
}