OrgZephyr/mcuboot
OrgZephyr
/
mcuboot
mirror of https://github.com/zephyrproject-rtos/mcuboot.git
1
0
Fork 0
Code Issues Releases Wiki Activity

imgtool: Add missing encodings to emitter tests 

Signed-off-by: David Vincze <david.vincze@arm.com>
Change-Id: Iff37cb62514b181a30f5537d6692d55e3d9c73ed
(cherry picked from commit f763c5ffee)
This commit is contained in:
David Vincze 2024-07-30 10:34:34 +02:00 committed by Jamie McCrae
parent 80397e0f28
commit a4800ce0cf
2 changed files with 40 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
scripts/imgtool/keys/ecdsa_test.py
View File

@ -52,17 +52,36 @@ class EcKeyGeneration(unittest.TestCase):
"""Basic sanity check on the code emitters.""" """Basic sanity check on the code emitters."""
k = ECDSA256P1.generate() k = ECDSA256P1.generate()
pubpem = io.StringIO()
k.emit_public_pem(pubpem)
self.assertIn("BEGIN PUBLIC KEY", pubpem.getvalue())
self.assertIn("END PUBLIC KEY", pubpem.getvalue())
ccode = io.StringIO() ccode = io.StringIO()
k.emit_c_public(ccode) k.emit_c_public(ccode)
self.assertIn("ecdsa_pub_key", ccode.getvalue()) self.assertIn("ecdsa_pub_key", ccode.getvalue())
self.assertIn("ecdsa_pub_key_len", ccode.getvalue()) self.assertIn("ecdsa_pub_key_len", ccode.getvalue())
hashccode = io.StringIO()
k.emit_c_public_hash(hashccode)
self.assertIn("ecdsa_pub_key_hash", hashccode.getvalue())
self.assertIn("ecdsa_pub_key_hash_len", hashccode.getvalue())
rustcode = io.StringIO() rustcode = io.StringIO()
k.emit_rust_public(rustcode) k.emit_rust_public(rustcode)
self.assertIn("ECDSA_PUB_KEY", rustcode.getvalue()) self.assertIn("ECDSA_PUB_KEY", rustcode.getvalue())
# raw data - bytes
pubraw = io.BytesIO()
k.emit_raw_public(pubraw)
self.assertTrue(len(pubraw.getvalue()) > 0)
hashraw = io.BytesIO()
k.emit_raw_public_hash(hashraw)
self.assertTrue(len(hashraw.getvalue()) > 0)
def test_emit_pub(self): def test_emit_pub(self):
"""Basic sanity check on the code emitters.""" """Basic sanity check on the code emitters, from public key."""
pubname = self.tname("public.pem") pubname = self.tname("public.pem")
k = ECDSA256P1.generate() k = ECDSA256P1.generate()
k.export_public(pubname) k.export_public(pubname)

21
scripts/imgtool/keys/ed25519_test.py
View File

@ -53,17 +53,36 @@ class Ed25519KeyGeneration(unittest.TestCase):
"""Basic sanity check on the code emitters.""" """Basic sanity check on the code emitters."""
k = Ed25519.generate() k = Ed25519.generate()
pubpem = io.StringIO()
k.emit_public_pem(pubpem)
self.assertIn("BEGIN PUBLIC KEY", pubpem.getvalue())
self.assertIn("END PUBLIC KEY", pubpem.getvalue())
ccode = io.StringIO() ccode = io.StringIO()
k.emit_c_public(ccode) k.emit_c_public(ccode)
self.assertIn("ed25519_pub_key", ccode.getvalue()) self.assertIn("ed25519_pub_key", ccode.getvalue())
self.assertIn("ed25519_pub_key_len", ccode.getvalue()) self.assertIn("ed25519_pub_key_len", ccode.getvalue())
hashccode = io.StringIO()
k.emit_c_public_hash(hashccode)
self.assertIn("ed25519_pub_key_hash", hashccode.getvalue())
self.assertIn("ed25519_pub_key_hash_len", hashccode.getvalue())
rustcode = io.StringIO() rustcode = io.StringIO()
k.emit_rust_public(rustcode) k.emit_rust_public(rustcode)
self.assertIn("ED25519_PUB_KEY", rustcode.getvalue()) self.assertIn("ED25519_PUB_KEY", rustcode.getvalue())
# raw data - bytes
pubraw = io.BytesIO()
k.emit_raw_public(pubraw)
self.assertTrue(len(pubraw.getvalue()) > 0)
hashraw = io.BytesIO()
k.emit_raw_public_hash(hashraw)
self.assertTrue(len(hashraw.getvalue()) > 0)
def test_emit_pub(self): def test_emit_pub(self):
"""Basic sanity check on the code emitters.""" """Basic sanity check on the code emitters, from public key."""
pubname = self.tname("public.pem") pubname = self.tname("public.pem")
k = Ed25519.generate() k = Ed25519.generate()
k.export_public(pubname) k.export_public(pubname)