diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-30 18:31:44 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-30 18:31:44 +0000 |
commit | c23a457e72abe608715ac76f076f47dc42af07a5 (patch) | |
tree | 2772049aaf84b5c9d0ed12ec8d86812f7a7904b6 /vendor/windows-metadata/src/blob.rs | |
parent | Releasing progress-linux version 1.73.0+dfsg1-1~progress7.99u1. (diff) | |
download | rustc-c23a457e72abe608715ac76f076f47dc42af07a5.tar.xz rustc-c23a457e72abe608715ac76f076f47dc42af07a5.zip |
Merging upstream version 1.74.1+dfsg1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'vendor/windows-metadata/src/blob.rs')
-rw-r--r-- | vendor/windows-metadata/src/blob.rs | 165 |
1 files changed, 165 insertions, 0 deletions
diff --git a/vendor/windows-metadata/src/blob.rs b/vendor/windows-metadata/src/blob.rs new file mode 100644 index 000000000..a54336db1 --- /dev/null +++ b/vendor/windows-metadata/src/blob.rs @@ -0,0 +1,165 @@ +use super::*; + +pub struct Blob<'a> { + pub file: usize, + pub slice: &'a [u8], +} + +impl<'a> std::ops::Deref for Blob<'a> { + type Target = [u8]; + + fn deref(&self) -> &[u8] { + self.slice + } +} + +impl<'a> Blob<'a> { + pub fn new(file: usize, slice: &'a [u8]) -> Self { + Self { file, slice } + } + + pub fn peek_usize(&self) -> (usize, usize) { + if self[0] & 0x80 == 0 { + (self[0] as usize, 1) + } else if self[0] & 0xC0 == 0x80 { + ((((self[0] & 0x3F) as usize) << 8) | self[1] as usize, 2) + } else { + ((((self[0] & 0x1F) as usize) << 24) | (self[1] as usize) << 16 | (self[2] as usize) << 8 | self[3] as usize, 4) + } + } + + pub fn read_usize(&mut self) -> usize { + let (value, offset) = self.peek_usize(); + self.offset(offset); + value + } + + pub fn read_expected(&mut self, expected: usize) -> bool { + let (value, offset) = self.peek_usize(); + if value == expected { + self.offset(offset); + true + } else { + false + } + } + + pub fn read_modifiers(&mut self) -> Vec<TypeDefOrRef> { + let mut mods = vec![]; + loop { + let (value, offset) = self.peek_usize(); + if value != ELEMENT_TYPE_CMOD_OPT as usize && value != ELEMENT_TYPE_CMOD_REQD as usize { + break; + } else { + self.offset(offset); + mods.push(TypeDefOrRef::decode(self.file, self.read_usize())) + } + } + mods + } + + pub fn read_str(&mut self) -> &str { + let len = self.read_usize(); + let value = unsafe { std::str::from_utf8_unchecked(&self.slice[..len]) }; + self.offset(len); + value + } + + pub fn read_string(self) -> String { + let slice = self.slice; + if slice.as_ptr().align_offset(std::mem::align_of::<u16>()) > 0 { + let slice = slice.chunks_exact(2).take(slice.len() / 2).map(|chunk| u16::from_le_bytes(chunk.try_into().unwrap())).collect::<Vec<u16>>(); + String::from_utf16_lossy(&slice) + } else { + let slice = unsafe { std::slice::from_raw_parts(slice.as_ptr() as *const u16, slice.len() / 2) }; + String::from_utf16_lossy(slice) + } + } + + pub fn read_bool(&mut self) -> bool { + // A bool is specified as "a single byte with value 0 (false) or 1 (true)". + match self.read_u8() { + 0 => false, + 1 => true, + _ => panic!("Illegal bool value"), + } + } + + pub fn read_i8(&mut self) -> i8 { + let value = i8::from_le_bytes(self[..1].try_into().unwrap()); + self.offset(1); + value + } + + pub fn read_u8(&mut self) -> u8 { + let value = u8::from_le_bytes(self[..1].try_into().unwrap()); + self.offset(1); + value + } + + pub fn read_i16(&mut self) -> i16 { + let value = i16::from_le_bytes(self[..2].try_into().unwrap()); + self.offset(2); + value + } + + pub fn read_u16(&mut self) -> u16 { + let value = u16::from_le_bytes(self[..2].try_into().unwrap()); + self.offset(2); + value + } + + pub fn read_i32(&mut self) -> i32 { + let value = i32::from_le_bytes(self[..4].try_into().unwrap()); + self.offset(4); + value + } + + pub fn read_u32(&mut self) -> u32 { + let value = u32::from_le_bytes(self[..4].try_into().unwrap()); + self.offset(4); + value + } + + pub fn read_i64(&mut self) -> i64 { + let value = i64::from_le_bytes(self[..8].try_into().unwrap()); + self.offset(8); + value + } + + pub fn read_u64(&mut self) -> u64 { + let value = u64::from_le_bytes(self[..8].try_into().unwrap()); + self.offset(8); + value + } + + pub fn read_f32(&mut self) -> f32 { + let value = f32::from_le_bytes(self[..4].try_into().unwrap()); + self.offset(4); + value + } + + pub fn read_f64(&mut self) -> f64 { + let value = f64::from_le_bytes(self[..8].try_into().unwrap()); + self.offset(8); + value + } + + pub fn read_integer(&mut self, ty: Type) -> Value { + match ty { + Type::I8 => Value::I8(self.read_i8()), + Type::U8 => Value::U8(self.read_u8()), + Type::I16 => Value::I16(self.read_i16()), + Type::U16 => Value::U16(self.read_u16()), + Type::I32 => Value::I32(self.read_i32()), + Type::U32 => Value::U32(self.read_u32()), + Type::I64 => Value::I64(self.read_i64()), + Type::U64 => Value::U64(self.read_u64()), + _ => panic!("Type is not an integer"), + } + } + + fn offset(&mut self, offset: usize) { + self.slice = &self.slice[offset..]; + } +} |