diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/binaryview.py | 41 | ||||
| -rw-r--r-- | python/types.py | 2 |
2 files changed, 43 insertions, 0 deletions
diff --git a/python/binaryview.py b/python/binaryview.py index 33d0b3e8..2446f04f 100644 --- a/python/binaryview.py +++ b/python/binaryview.py @@ -28,6 +28,7 @@ import abc import json import inspect import os +import uuid from typing import Callable, Generator, Optional, Union, Tuple, List, Mapping, Any, \ Iterator, Iterable, KeysView, ItemsView, ValuesView, Dict from dataclasses import dataclass @@ -3691,6 +3692,24 @@ class BinaryView: size = self.arch.address_size return self.read_int(address, size, False, self.endianness) + def read_uuid(self, address: int, ms_format: bool = True) -> uuid.UUID: + """ + Reads a UUID from the specified address in the binary view. + + :param address: The address to read the UUID from. + :param ms_format: Whether to return the UUID in Microsoft format (True) or standard format (False). + :return: A UUID object + :raises ValueError: If 16 bytes couldn't be read from the specified address. + """ + data = self.read(address, 16) + if len(data) != 16: + raise ValueError(f"Couldn't read 16 bytes from address: {address:#x}") + if ms_format: + reordered = data[3::-1] + data[5:3:-1] + data[7:5:-1] + data[8:] + return uuid.UUID(bytes=reordered) + else: + return uuid.UUID(bytes=data) + def write(self, addr: int, data: bytes, except_on_relocation: bool = True) -> int: """ ``write`` writes the bytes in ``data`` to the virtual address ``addr``. @@ -9568,6 +9587,28 @@ class TypedDataAccessor: else: raise TypeError(f"Unhandled `Type` {type(_type)}") + def as_uuid(self, ms_format: bool = True) -> uuid.UUID: + """ + Converts the object to a UUID object using Microsoft byte ordering. + + :param ms_format: Flag indicating whether to use Microsoft byte ordering. Default is True. + :type ms_format: bool + :return: The UUID object representing the byte array. + :rtype: uuid.UUID + :raises ValueError: If the byte array representation of this data is not exactly 16 bytes long. + """ + # Ensure the byte array is 16 bytes long (128 bits) + byte_array = bytes(self) + if len(byte_array) != 16: + raise ValueError("Byte array must be exactly 16 bytes long.") + # Reorder the bytes to match the UUID format + # First 3 components (10 bytes) are little-endian, last 2 components (6 bytes) are big-endian + if ms_format: + reordered = byte_array[3::-1] + byte_array[5:3:-1] + byte_array[7:5:-1] + byte_array[8:] + # Create a UUID object and return its string representation + return uuid.UUID(bytes=reordered) + else: + return uuid.UUID(bytes=byte_array) # for backward compatibility TypedDataReader = TypedDataAccessor diff --git a/python/types.py b/python/types.py index 53454a17..57c0824c 100644 --- a/python/types.py +++ b/python/types.py @@ -1227,6 +1227,8 @@ class BaseStructure: def __init__(self, type: Union['NamedTypeReferenceType', 'StructureType'], offset: int, width: int = 0): if isinstance(type, StructureType): + if type.registered_name is None: + raise ValueError("StructureType must be registered to be used as a base structure") self.type = type.registered_name self.offset = offset self.width = type.width |