mirror of
https://github.com/AdaCore/OpenUxAS.git
synced 2026-02-12 13:07:16 -08:00
041d7cc357
This commit makes significant effort towards refactoring the repository so that it is in a cleaner and more consistent state going forward. Since we are now explicitly expecting multiple languages to be used for OpenUxAS, we have reorganized `src` accordingly. Likewise, we have reorganized `tests`. This is a candidate for the rebaseline of afrl-rq/OpenUxAS.
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
import struct
|
|
import typing
|
|
from typing import Any, Tuple
|
|
|
|
|
|
class Buffer(object):
|
|
|
|
SUPPORTED_TYPES = set(['int32', 'int64',
|
|
'uint16', 'uint32',
|
|
'real32', 'real64',
|
|
'byte', 'bool',
|
|
'string'])
|
|
|
|
BASIC_TYPES = {
|
|
'int32': '>i',
|
|
'int64': '>q',
|
|
'uint16': '>H',
|
|
'uint32': '>I',
|
|
'real32': '>f',
|
|
'real64': '>d',
|
|
'byte': '>B',
|
|
'bool': '>?'}
|
|
|
|
def __init__(self, data: bytes = b'') -> None:
|
|
"""Initialize a buffer.
|
|
|
|
:param data: content of the buffer
|
|
"""
|
|
self.data = data
|
|
self.pos = 0
|
|
|
|
def unpack_struct(self, fmt: str) -> Tuple[Any, ...]:
|
|
"""Unpack a list of elements from the buffer.
|
|
|
|
Position of the buffer is updated.
|
|
|
|
:param fmt: a string compliant with struct.unpack_from fmt argument.
|
|
:return: a list of unpack objects
|
|
"""
|
|
result = struct.unpack_from(fmt, self.data, self.pos)
|
|
self.pos += struct.calcsize(fmt)
|
|
return result
|
|
|
|
def unpack(self, type_name: str) -> Any:
|
|
"""Unpack a simple element from the buffer.
|
|
|
|
Position of the buffer is updated.
|
|
|
|
:param type_name: a type name from SUPPORTED_TYPES
|
|
"""
|
|
assert type_name in self.SUPPORTED_TYPES, \
|
|
'unknown type: %s' % type_name
|
|
if type_name in self.BASIC_TYPES:
|
|
value = self.unpack_struct(self.BASIC_TYPES[type_name])[0]
|
|
elif type_name == 'string':
|
|
string_size = self.unpack('uint16')
|
|
value = self.unpack_struct('%ss' % string_size)[0]
|
|
value = value.decode('utf-8')
|
|
|
|
return value
|