drgn/tests/test_serialize.py
Omar Sandoval 87b7292aa5 Relicense drgn from GPLv3+ to LGPLv2.1+
drgn is currently licensed as GPLv3+. Part of the long term vision for
drgn is that other projects can use it as a library providing
programmatic interfaces for debugger functionality. A more permissive
license is better suited to this goal. We decided on LGPLv2.1+ as a good
balance between software freedom and permissiveness.

All contributors not employed by Meta were contacted via email and
consented to the license change. The only exception was the author of
commit c4fbf7e589 ("libdrgn: fix for compilation error"), who did not
respond. That commit reverted a single line of code to one originally
written by me in commit 640b1c011d ("libdrgn: embed DWARF index in
DWARF info cache").

Signed-off-by: Omar Sandoval <osandov@osandov.com>
2022-11-01 17:05:16 -07:00

64 lines
2.4 KiB
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# SPDX-License-Identifier: LGPL-2.1-or-later
from tests import TestCase
from tests.libdrgn import deserialize_bits, serialize_bits
VALUE = 12345678912345678989
def py_serialize_bits(value, bit_offset, bit_size, little_endian):
bits = bit_offset + bit_size
size = (bits + 7) // 8
if little_endian:
tmp = value << bit_offset
else:
tmp = value << -bits % 8
# Buffer with unused bits set to zero.
buf0 = tmp.to_bytes(size, "little" if little_endian else "big")
# Buffer with unused bits set to one.
buf1 = bytearray(buf0)
if little_endian:
# bit_offset least significant bits.
buf1[0] |= (1 << bit_offset) - 1
# 8 - (bit_offset + bit_size) % 8 most significant bits.
buf1[-1] |= (0xFF00 >> -bits % 8) & 0xFF
else:
# bit_offset most significant bits.
buf1[0] |= (0xFF00 >> bit_offset) & 0xFF
# 8 - (bit_offset + bit_size) % 8 least significant bits.
buf1[-1] |= (1 << -bits % 8) - 1
return buf0, buf1
class TestSerialize(TestCase):
def test_deserialize(self):
for bit_size in range(1, 65):
expected = VALUE & ((1 << bit_size) - 1)
for bit_offset in range(8):
for little_endian in [True, False]:
for buf in py_serialize_bits(
expected, bit_offset, bit_size, little_endian
):
value = deserialize_bits(
buf, bit_offset, bit_size, little_endian
)
self.assertEqual(value, expected)
def test_serialize(self):
for bit_size in range(1, 65):
value = VALUE & ((1 << bit_size) - 1)
for bit_offset in range(8):
for little_endian in [True, False]:
expected0, expected1 = py_serialize_bits(
value, bit_offset, bit_size, little_endian
)
buf = bytearray(len(expected0))
serialize_bits(buf, bit_offset, value, bit_size, little_endian)
self.assertEqual(buf, expected0)
buf = bytearray([0xFF] * len(expected1))
serialize_bits(buf, bit_offset, value, bit_size, little_endian)
self.assertEqual(buf, expected1)