1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 05:58:23 +00:00
trezor-firmware/core/tests/test_apps.common.cbor.py

197 lines
6.5 KiB
Python
Raw Normal View History

from common import * # isort:skip
2018-06-14 14:28:50 +00:00
from apps.common.cbor import (
2018-06-14 14:28:50 +00:00
IndefiniteLengthArray,
OrderedMap,
Tagged,
create_array_header,
create_embedded_cbor_bytes_header,
create_map_header,
2023-12-08 12:48:53 +00:00
create_tagged_set_header,
decode,
encode,
encode_streamed,
2018-06-14 14:28:50 +00:00
)
2018-06-14 14:28:50 +00:00
class TestCardanoCbor(unittest.TestCase):
def test_create_array_header(self):
test_vectors = [
2023-06-28 10:46:29 +00:00
(0, "80"),
(23, "97"),
((2**8) - 1, "98ff"),
((2**16) - 1, "99ffff"),
((2**32) - 1, "9affffffff"),
((2**64) - 1, "9bffffffffffffffff"),
]
for val, header_hex in test_vectors:
header = unhexlify(header_hex)
self.assertEqual(create_array_header(val), header)
with self.assertRaises(NotImplementedError):
2023-06-28 10:46:29 +00:00
create_array_header(2**64)
2023-12-08 12:48:53 +00:00
def test_create_tagged_set_header(self):
test_vectors = [
(0, 'd9010280'),
(23, 'd9010297'),
((2 ** 8) - 1, 'd9010298ff'),
((2 ** 16) - 1, 'd9010299ffff'),
((2 ** 32) - 1, 'd901029affffffff'),
((2 ** 64) - 1, 'd901029bffffffffffffffff'),
]
for val, header_hex in test_vectors:
header = unhexlify(header_hex)
self.assertEqual(create_tagged_set_header(val), header)
with self.assertRaises(NotImplementedError):
create_tagged_set_header(2 ** 64)
def test_create_map_header(self):
test_vectors = [
2023-06-28 10:46:29 +00:00
(0, "a0"),
(23, "b7"),
((2**8) - 1, "b8ff"),
((2**16) - 1, "b9ffff"),
((2**32) - 1, "baffffffff"),
((2**64) - 1, "bbffffffffffffffff"),
]
for val, header_hex in test_vectors:
header = unhexlify(header_hex)
self.assertEqual(create_map_header(val), header)
with self.assertRaises(NotImplementedError):
2023-06-28 10:46:29 +00:00
create_map_header(2**64)
def test_create_embedded_cbor_bytes_header(self):
test_vectors = [
2023-06-28 10:46:29 +00:00
(0, "d81840"),
(23, "d81857"),
((2**8) - 1, "d81858ff"),
((2**16) - 1, "d81859ffff"),
((2**32) - 1, "d8185affffffff"),
((2**64) - 1, "d8185bffffffffffffffff"),
]
for val, header_hex in test_vectors:
header = unhexlify(header_hex)
self.assertEqual(create_embedded_cbor_bytes_header(val), header)
with self.assertRaises(NotImplementedError):
2023-06-28 10:46:29 +00:00
create_embedded_cbor_bytes_header(2**64)
2018-06-14 14:28:50 +00:00
def test_cbor_encoding(self):
test_vectors = [
# unsigned integers
2023-06-28 10:46:29 +00:00
(0, "00"),
(1, "01"),
(10, "0a"),
(23, "17"),
(24, "1818"),
(25, "1819"),
(100, "1864"),
(1000, "1903e8"),
(1000000, "1a000f4240"),
(1000000000000, "1b000000e8d4a51000"),
# negative integers
2023-06-28 10:46:29 +00:00
(-1, "20"),
(-10, "29"),
(-24, "37"),
(-25, "3818"),
(-26, "3819"),
(-100, "3863"),
(-1000, "3903E7"),
(-1000000, "3A000F423F"),
(-1000000000000, "3B000000E8D4A50FFF"),
2018-06-14 14:28:50 +00:00
# binary strings
2023-06-28 10:46:29 +00:00
(b"", "40"),
(unhexlify("01020304"), "4401020304"),
# text strings
2023-06-28 10:46:29 +00:00
("", "60"),
("Fun", "6346756e"),
(
"P\u0159\xed\u0161ern\u011b \u017elu\u0165ou\u010dk\xfd k\u016f\u0148 \xfap\u011bl \u010f\xe1belsk\xe9 \xf3dy z\xe1ke\u0159n\xfd u\u010de\u0148 b\u011b\u017e\xed pod\xe9l z\xf3ny \xfal\u016f",
"786550c599c3adc5a165726ec49b20c5be6c75c5a56f75c48d6bc3bd206bc5afc58820c3ba70c49b6c20c48fc3a162656c736bc3a920c3b36479207ac3a16b65c5996ec3bd2075c48d65c5882062c49bc5bec3ad20706f64c3a96c207ac3b36e7920c3ba6cc5af",
),
2018-06-14 14:28:50 +00:00
# tags
2023-06-28 10:46:29 +00:00
(Tagged(1, 1363896240), "c11a514b67b0"),
(Tagged(23, unhexlify("01020304")), "d74401020304"),
2018-06-14 14:28:50 +00:00
# arrays
2023-06-28 10:46:29 +00:00
([], "80"),
([1, 2, 3], "83010203"),
([1, [2, 3], [4, 5]], "8301820203820405"),
(
list(range(1, 26)),
"98190102030405060708090a0b0c0d0e0f101112131415161718181819",
),
2018-06-14 14:28:50 +00:00
# maps
2023-06-28 10:46:29 +00:00
({}, "a0"),
({1: 2, 3: 4}, "a201020304"),
({3: 4, 1: 2}, "a201020304"),
2018-06-14 14:28:50 +00:00
# indefinite
2023-06-28 10:46:29 +00:00
(IndefiniteLengthArray([]), "9fff"),
(IndefiniteLengthArray([1, [2, 3], [4, 5]]), "9f01820203820405ff"),
(
IndefiniteLengthArray([1, [2, 3], IndefiniteLengthArray([4, 5])]),
"9f018202039f0405ffff",
),
# boolean
2023-06-28 10:46:29 +00:00
(True, "f5"),
(False, "f4"),
# null
2023-06-28 10:46:29 +00:00
(None, "f6"),
2018-06-14 14:28:50 +00:00
]
for val, encoded_hex in test_vectors:
encoded = unhexlify(encoded_hex)
self.assertEqual(encode(val), encoded)
self.assertEqual(decode(encoded), val)
2018-06-14 14:28:50 +00:00
def test_cbor_tuples(self):
"""
Tuples should be encoded as arrays and decoded back as lists.
"""
test_vectors = [
2023-06-28 10:46:29 +00:00
([], "80"),
([1, 2, 3], "83010203"),
([1, [2, 3], [4, 5]], "8301820203820405"),
(
list(range(1, 26)),
"98190102030405060708090a0b0c0d0e0f101112131415161718181819",
),
]
for val, encoded_hex in test_vectors:
value_tuple = tuple(val)
encoded = unhexlify(encoded_hex)
self.assertEqual(encode(value_tuple), encoded)
self.assertEqual(decode(encoded), val)
def test_cbor_ordered_map(self):
"""
OrderedMaps should be encoded as maps without any ordering and decoded back as dicts.
"""
test_vectors = [
2023-06-28 10:46:29 +00:00
({}, "a0"),
([[1, 2], [3, 4]], "a201020304"),
([[3, 4], [1, 2]], "a203040102"),
]
for val, encoded_hex in test_vectors:
ordered_map = OrderedMap()
for key, value in val:
ordered_map[key] = value
encoded = unhexlify(encoded_hex)
self.assertEqual(encode(ordered_map), encoded)
self.assertEqual(decode(encoded), {k: v for k, v in val})
def test_encode_streamed(self):
large_dict = {i: i for i in range(100)}
encoded = encode(large_dict)
2023-06-28 10:46:29 +00:00
encoded_streamed = [bytes(item) for item in encode_streamed(large_dict)]
2023-06-28 10:46:29 +00:00
self.assertEqual(b"".join(encoded_streamed), encoded)
2023-06-28 10:46:29 +00:00
if __name__ == "__main__":
2018-06-14 14:28:50 +00:00
unittest.main()