shenjack/nbt-rust
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

还是直接用 Vec 的 unsafe 方法吧

Browse Source
This commit is contained in:
shenjack 2024-01-13 00:06:43 +08:00
parent d5f80198c6
commit 5737c93568
Signed by: shenjack
GPG Key ID: 7B1134A979775551
1 changed files with 21 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
v2/src/data_struct.rs
View File

@ -96,50 +96,54 @@ pub mod raw_reading {
slice.iter().map(|&x| x as i8).collect::<Vec<i8>>() slice.iter().map(|&x| x as i8).collect::<Vec<i8>>()
} }
/// unsafe 从这里开始 /// unsafe 从这里开始
pub fn slice_as_short_array(slice: &[u8]) -> Option<&[i16]> { pub fn slice_as_short_array(slice: &[u8]) -> Option<Vec<i16>> {
let length = if slice.len() % 2 != 0 { let length = if slice.len() % 2 != 0 {
return None; return None;
} else { } else {
(slice.len() / 2) as usize (slice.len() / 2) as usize
}; };
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const i16, length) }) Some(unsafe {
Vec::from_raw_parts(slice.as_ptr() as *mut i16, length, length)
})
} }
/// 开始 unsafe 了 /// 开始 unsafe 了
/// unsafe rust, 小子! /// unsafe rust, 小子!
pub fn slice_as_int_array(slice: &[u8]) -> Option<&[i32]> { pub fn slice_as_int_array(slice: &[u8]) -> Option<Vec<i32>> {
let length = if slice.len() % 4 != 0 { let length = if slice.len() % 4 != 0 {
return None; return None;
} else { } else {
(slice.len() / 4) as usize (slice.len() / 4) as usize
}; };
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const i32, length) }) Some(unsafe {
Vec::from_raw_parts(slice.as_ptr() as *mut i32, 4, length)
})
} }
/// 这边也是 unsafe 捏 /// 这边也是 unsafe 捏
pub fn slice_as_long_array(slice: &[u8]) -> Option<&[i64]> { pub fn slice_as_long_array(slice: &[u8]) -> Option<Vec<i64>> {
let length = if slice.len() % 8 != 0 { let length = if slice.len() % 8 != 0 {
return None; return None;
} else { } else {
(slice.len() / 8) as usize (slice.len() / 8) as usize
}; };
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const i64, length) }) Some(unsafe { Vec::from_raw_parts(slice.as_ptr() as *mut i64, 8, length) })
} }
/// 这边也是 unsafe 捏 /// 这边也是 unsafe 捏
pub fn slice_as_float_array(slice: &[u8]) -> Option<&[f32]> { pub fn slice_as_float_array(slice: &[u8]) -> Option<Vec<f32>> {
let length = if slice.len() % 4 != 0 { let length = if slice.len() % 4 != 0 {
return None; return None;
} else { } else {
(slice.len() / 4) as usize (slice.len() / 4) as usize
}; };
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const f32, length) }) Some(unsafe { Vec::from_raw_parts(slice.as_ptr() as *mut f32, 4, length) })
} }
/// 这边也是 unsafe 捏 /// 这边也是 unsafe 捏
pub fn slice_as_double_array(slice: &[u8]) -> Option<&[f64]> { pub fn slice_as_double_array(slice: &[u8]) -> Option<Vec<f64>> {
let length = if slice.len() % 8 != 0 { let length = if slice.len() % 8 != 0 {
return None; return None;
} else { } else {
(slice.len() / 8) as usize (slice.len() / 8) as usize
}; };
Some(unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const f64, length) }) Some(unsafe { Vec::from_raw_parts(slice.as_ptr() as *mut f64, 8, length) })
} }
} }
@ -223,28 +227,28 @@ impl<'value> Value<'value> {
} }
2 => { 2 => {
let raw_data = data.read_bytes(length as usize * 2); let raw_data = data.read_bytes(length as usize * 2);
let list = raw_reading::slice_as_short_array(raw_data.as_slice()).unwrap().to_vec(); let list = raw_reading::slice_as_short_array(raw_data.as_slice()).unwrap();
Self::List(ListContent::ShortList(list)) Self::List(ListContent::ShortList(list))
} }
3 => { 3 => {
let raw_data = data.read_bytes(length as usize * 4); let raw_data = data.read_bytes(length as usize * 4);
let list = raw_reading::slice_as_int_array(raw_data.as_slice()).unwrap().to_vec(); let list = raw_reading::slice_as_int_array(raw_data.as_slice()).unwrap();
Self::List(ListContent::IntList(list)) Self::List(ListContent::IntList(list))
} }
4 => { 4 => {
let raw_data = data.read_bytes(length as usize * 8); let raw_data = data.read_bytes(length as usize * 8);
let list = raw_reading::slice_as_long_array(raw_data.as_slice()).unwrap().to_vec(); let list = raw_reading::slice_as_long_array(raw_data.as_slice()).unwrap();
Self::List(ListContent::LongList(list)) Self::List(ListContent::LongList(list))
} }
5 => { 5 => {
let raw_data = data.read_bytes(length as usize * 4); let raw_data = data.read_bytes(length as usize * 4);
let list = raw_reading::slice_as_float_array(raw_data.as_slice()).unwrap().to_vec(); let list = raw_reading::slice_as_float_array(raw_data.as_slice()).unwrap();
Self::List(ListContent::FloatList(list)) Self::List(ListContent::FloatList(list))
} }
6 => { 6 => {
let raw_data = data.read_bytes(length as usize * 8); let raw_data = data.read_bytes(length as usize * 8);
let list = let list =
raw_reading::slice_as_double_array(raw_data.as_slice()).unwrap().to_vec(); raw_reading::slice_as_double_array(raw_data.as_slice()).unwrap();
Self::List(ListContent::DoubleList(list)) Self::List(ListContent::DoubleList(list))
} }
7 => { 7 => {
@ -286,7 +290,7 @@ impl<'value> Value<'value> {
let length = data.read_int(); let length = data.read_int();
let raw_data = data.read_bytes(length as usize * 4); let raw_data = data.read_bytes(length as usize * 4);
let value = let value =
raw_reading::slice_as_int_array(raw_data.as_slice()).unwrap().to_vec(); raw_reading::slice_as_int_array(raw_data.as_slice()).unwrap();
list.push(value); list.push(value);
} }
Self::List(ListContent::IntArrayList(list)) Self::List(ListContent::IntArrayList(list))
@ -297,7 +301,7 @@ impl<'value> Value<'value> {
let length = data.read_int(); let length = data.read_int();
let raw_data = data.read_bytes(length as usize * 8); let raw_data = data.read_bytes(length as usize * 8);
let value = let value =
raw_reading::slice_as_long_array(raw_data.as_slice()).unwrap().to_vec(); raw_reading::slice_as_long_array(raw_data.as_slice()).unwrap();
list.push(value); list.push(value);
} }
Self::List(ListContent::LongArrayList(list)) Self::List(ListContent::LongArrayList(list))