blob: f828f73564f99fe2529427b75b5dee40039e521d (
plain) (
blame)
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
|
use std::ffi::OsString;
use std::os::unix::ffi::OsStringExt;
use std::path::PathBuf;
pub struct CircularPathBuffer {
buffer: [u8; 256],
pointer: u8,
}
impl CircularPathBuffer {
pub fn new() -> Self {
Self { buffer: [0; 256] , pointer: 0 }
}
pub fn clear(&mut self) {
self.buffer.fill(0);
self.pointer = 0;
}
pub fn populate(&mut self, bytes: &[u8]) {
self.clear();
if bytes.len() > 255 {
unreachable!(
"Attempted to populate CircularPathBuffer with {} bytes: {:?}",
bytes.len(), String::from_utf8_lossy(bytes)
)
}
let self_slice = &mut self.buffer[..bytes.len()];
self_slice.copy_from_slice(&bytes);
}
pub fn set_pointer(&mut self, value: u8) {
self.pointer = 0;
if value != 0x00 {
for (i, c) in self.buffer.iter().enumerate() {
match c {
b'/' => self.pointer = (i + 1) as u8,
0x00 => break,
_ => continue,
}
}
}
}
pub fn read_byte(&mut self) -> u8 {
let pointer = self.pointer as usize;
self.pointer = self.pointer.wrapping_add(1);
self.buffer[pointer]
}
// Returns an unsanitized relative path.
pub fn push_byte(&mut self, value: u8) -> Option<PathBuf> {
if value == 0x00 {
let pointer = self.pointer as usize;
let vec = self.buffer[..pointer].to_vec();
self.clear();
let os_string: OsString = OsStringExt::from_vec(vec);
Some(os_string.into())
} else {
let pointer = self.pointer as usize;
self.pointer = self.pointer.wrapping_add(1);
self.buffer[pointer] = value;
None
}
}
}
|
