splitted tests

dmazzella · Jan 17, 2019 · f50552e · f50552e
1 parent 2018b4a
commit f50552e
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 13 deletions.
diff --git a/tests/ateccX08a/tests_verify.py b/tests/ateccX08a/tests_verify.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # pylint: disable=E0401
 import logging
+import utime
 from ubinascii import hexlify, unhexlify
 
 log = logging.getLogger("ateccX08a.tests_verify")
@@ -15,12 +16,38 @@
         0XDA, 0X3C, 0X14, 0X46, 0X68, 0X2C, 0X25, 0X8C, 0X46, 0X3F, 0XFF, 0XDF, 0X58, 0XDF, 0XD2, 0XFA,
         0X3E, 0X6C, 0X37, 0X8B, 0X53, 0XD7, 0X95, 0XC4, 0XA4, 0XDF, 0XFB, 0X41, 0X99, 0XED, 0XD7, 0X86,
         0X2F, 0X23, 0XAB, 0XAF, 0X02, 0X03, 0XB4, 0XB8, 0X91, 0X1B, 0XA0, 0X56, 0X99, 0X94, 0XE1, 0X01
-    ])
+    ]),
+    "MESSAGE": b'a message to sign via ECDSA     ',
+    "SIGNATURE": {
+        "R": bytes([
+            0X71, 0X07, 0X7D, 0X35, 0X6F, 0XCD, 0X70, 0XD4, 0XCC, 0X47, 0X2A, 0XD0, 0X49, 0X0E, 0X75, 0XAB,
+            0XC5, 0X41, 0X98, 0XEE, 0X6A, 0X96, 0X7B, 0X90, 0XF2, 0XC7, 0XE3, 0XC8, 0X2B, 0XBF, 0X54, 0X96
+        ]),
+        "S": bytes([
+            0X77, 0X8E, 0XFE, 0X0B, 0XF6, 0X9D, 0X15, 0XED, 0XA0, 0X71, 0XBD, 0XD3, 0XFE, 0X46, 0X99, 0X26,
+            0X31, 0XF8, 0X80, 0X01, 0X13, 0X76, 0XCD, 0X45, 0X7C, 0X62, 0X55, 0X43, 0XC9, 0X7F, 0XCC, 0XD9
+        ])
+    }
 }
 
+
 def run(device=None, configuration=None):
     if not device:
         raise ValueError("device")
 
-    log.debug("PRIVATE: %s", _TEST_KEYS["PRIVATE"])
-    log.debug("PUBLIC: %s", _TEST_KEYS["PUBLIC"])
+    slot = 11
+    public_key = _TEST_KEYS["PUBLIC"]
+    message = _TEST_KEYS["MESSAGE"]
+    message_digest = device.atcab_sha(message)[1:-2]
+    signature = _TEST_KEYS["SIGNATURE"]["R"] + _TEST_KEYS["SIGNATURE"]["S"]
+    # # verify the signature extern
+    t = utime.ticks_us()
+    verified = device.atcab_verify_extern(
+        message_digest, signature, public_key)
+    delta = utime.ticks_diff(utime.ticks_us(), t)
+    print('Time={:6.3f}ms '.format(delta/1000))
+    log.info("atcab_verify_extern %r", verified)
+    # # verify the signature stored
+    # device.atcab_nonce_load(ATCA_CONSTANTS.NONCE_MODE_TARGET_TEMPKEY, message_digest)
+    verified = device.atcab_verify_stored(message_digest, signature, slot)
+    log.info("atcab_verify_stored %r", verified)
diff --git a/tests/ateccX08a/tests_write.py b/tests/ateccX08a/tests_write.py
@@ -59,6 +59,7 @@
     }
 }
 
+
 def run(device=None):
     if not device:
         raise ValueError("device")
@@ -82,19 +83,9 @@ def run(device=None):
     slot = 11
     if not device.atcab_is_slot_locked(slot):
         public_key = _TEST_KEYS["PUBLIC"]
-        message = _TEST_KEYS["MESSAGE"]
-        packet = device.atcab_sha(message)
-        message_digest = packet.response_data[1:-2]
-        signature = _TEST_KEYS["SIGNATURE"]["R"] + _TEST_KEYS["SIGNATURE"]["S"]
         # # write public_key to slot
         device.atcab_write_pubkey(slot, public_key)
         # # verify wrote public_key
         assert public_key == device.atcab_read_pubkey(slot)
-        # # verify the signature
-        verified = device.atcab_verify_extern(message_digest, signature, public_key)
-        log.info("atcab_verify_extern %r", verified)
-
-        verified = device.atcab_verify_stored(message, signature, slot)
-        log.info("atcab_verify_stored %r", verified)
     else:
         log.info("slot %d locked", slot)