Merge pull request #7 from AndrzejKurek/interactive_app

Make the example interactive

Author: Patater

atecc608a_utils.c

@@ -29,8 +29,7 @@ psa_status_t atecc608a_get_serial_number(uint8_t *buffer,
 {
     psa_status_t status = PSA_ERROR_GENERIC_ERROR;
 
-    if (buffer_size < ATCA_SERIAL_NUM_SIZE)
-    {
+    if (buffer_size < ATCA_SERIAL_NUM_SIZE) {
         return PSA_ERROR_BUFFER_TOO_SMALL;
     }
 
@@ -44,35 +43,52 @@ psa_status_t atecc608a_get_serial_number(uint8_t *buffer,
     return status;
 }
 
-psa_status_t atecc608a_check_config_locked()
+psa_status_t atecc608a_check_zone_locked(uint8_t zone)
 {
-    bool config_locked;
+    bool zone_locked;
     psa_status_t status = PSA_ERROR_GENERIC_ERROR;
 
     ASSERT_SUCCESS_PSA(atecc608a_init());
 
-    ASSERT_SUCCESS(atcab_is_locked(LOCK_ZONE_CONFIG, &config_locked));
+    ASSERT_SUCCESS(atcab_is_locked(zone, &zone_locked));
 
 exit:
     atecc608a_deinit();
-    if (status == PSA_SUCCESS)
-    {
-        status = config_locked? PSA_SUCCESS : PSA_ERROR_HARDWARE_FAILURE;
+    if (status == PSA_SUCCESS) {
+        status = zone_locked ? PSA_SUCCESS : PSA_ERROR_HARDWARE_FAILURE;
     }
     return status;
 }
 
+psa_status_t atecc608a_lock_data_zone()
+{
+    psa_status_t status = PSA_ERROR_GENERIC_ERROR;
+    bool zone_locked;
+
+    ASSERT_SUCCESS_PSA(atecc608a_init());
+    /* atcab_is_locked used instead of atecc608a_check_zone_locked as an
+     * optimization - this way atecc608a_init won't be called again. */
+    ASSERT_SUCCESS(atcab_is_locked(LOCK_ZONE_DATA, &zone_locked));
+    if (zone_locked) {
+        status = PSA_ERROR_HARDWARE_FAILURE;
+        goto exit;
+    }
+    ASSERT_SUCCESS(atcab_lock_data_zone());
+
+exit:
+    atecc608a_deinit();
+    return status;
+}
+
 psa_status_t atecc608a_random_32_bytes(uint8_t *rand_out, size_t buffer_size)
 {
     psa_status_t status = PSA_ERROR_GENERIC_ERROR;
 
-    if (rand_out == NULL)
-    {
+    if (rand_out == NULL) {
         return PSA_ERROR_INVALID_ARGUMENT;
     }
 
-    if (buffer_size < 32)
-    {
+    if (buffer_size < 32) {
         return PSA_ERROR_BUFFER_TOO_SMALL;
     }
 

atecc608a_utils.h

@@ -59,7 +59,9 @@
 psa_status_t atecc608a_get_serial_number(uint8_t *buffer, size_t buffer_size,
                                          size_t *buffer_length);
 
-psa_status_t atecc608a_check_config_locked();
+psa_status_t atecc608a_check_zone_locked(uint8_t zone);
+
+psa_status_t atecc608a_lock_data_zone();
 
 /** Generate a 32 byte random number from the CryptoAuth device. */
 psa_status_t atecc608a_random_32_bytes(uint8_t *rand_out, size_t buffer_size);