mbed-os/TESTS/psa/its/main.cpp

/*
* Copyright (c) 2018 ARM Limited. All rights reserved.
*
* SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the License); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an AS IS BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#ifndef TARGET_PSA
#error [NOT_SUPPORTED] ITS tests can run only on PSA-enabled targets.
#endif // TARGET_PSA

#include "greentea-client/test_env.h"
#include "unity/unity.h"
#include "utest/utest.h"
#include "psa/internal_trusted_storage.h"
#include "psa/lifecycle.h"

using namespace utest::v1;

#define TEST_BUFF_SIZE 16

static void pits_test()
{
    psa_its_status_t status = PSA_ITS_SUCCESS;
    uint8_t write_buff[TEST_BUFF_SIZE] = {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F};
    uint8_t read_buff[TEST_BUFF_SIZE] = {0};
    struct psa_its_info_t info = {0, PSA_ITS_FLAG_WRITE_ONCE};
    memset(read_buff, 0, TEST_BUFF_SIZE);

    status = psa_its_get_info(5, &info);
    TEST_ASSERT_EQUAL(PSA_ITS_ERROR_UID_NOT_FOUND, status);

    status = psa_its_set(5, TEST_BUFF_SIZE, write_buff, 0);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);

    status = psa_its_get_info(5, &info);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);
    TEST_ASSERT_EQUAL(TEST_BUFF_SIZE, info.size);
    TEST_ASSERT_EQUAL(0, info.flags);

    status = psa_its_get(5, 0, TEST_BUFF_SIZE, read_buff);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);
    TEST_ASSERT_EQUAL_MEMORY(write_buff, read_buff, TEST_BUFF_SIZE);

    memset(read_buff, 0, TEST_BUFF_SIZE);
    status = psa_its_get(5, 1, TEST_BUFF_SIZE, read_buff);
    TEST_ASSERT_NOT_EQUAL(PSA_ITS_SUCCESS, status);

    status = psa_its_get(5, 1, TEST_BUFF_SIZE - 1, read_buff);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);
    TEST_ASSERT_EQUAL_MEMORY(write_buff + 1, read_buff, TEST_BUFF_SIZE - 1);

    status = psa_its_remove(5);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);

    status = psa_its_get_info(5, &info);
    TEST_ASSERT_EQUAL(PSA_ITS_ERROR_UID_NOT_FOUND, status);
}

static void pits_write_once_test()
{
    psa_its_status_t status = PSA_ITS_SUCCESS;
    uint8_t write_buff[TEST_BUFF_SIZE] = {0x0F, 0x0E, 0x0D, 0x0C, 0x0B, 0x0A, 0x09, 0x08, 0x07, 0x06, 0x05, 0x04, 0x03, 0x02, 0x01, 0x00};
    uint8_t read_buff[TEST_BUFF_SIZE] = {0};
    struct psa_its_info_t info = {0, 0};

    status = psa_its_get_info(5, &info);
    TEST_ASSERT_EQUAL(PSA_ITS_ERROR_UID_NOT_FOUND, status);

    status = psa_its_set(5, TEST_BUFF_SIZE, write_buff, PSA_ITS_FLAG_WRITE_ONCE);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);

    info.size = 0;
    info.flags = PSA_ITS_FLAG_NONE;
    status = psa_its_get_info(5, &info);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);
    TEST_ASSERT_EQUAL(TEST_BUFF_SIZE, info.size);
    TEST_ASSERT_EQUAL(PSA_ITS_FLAG_WRITE_ONCE, info.flags);

    status = psa_its_get(5, 0, TEST_BUFF_SIZE, read_buff);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);
    TEST_ASSERT_EQUAL_MEMORY(write_buff, read_buff, TEST_BUFF_SIZE);

    status = psa_its_set(5, TEST_BUFF_SIZE, write_buff, PSA_ITS_FLAG_WRITE_ONCE);
    TEST_ASSERT_NOT_EQUAL(PSA_ITS_SUCCESS, status);

    status = psa_its_set(5, TEST_BUFF_SIZE, write_buff, 0);
    TEST_ASSERT_NOT_EQUAL(PSA_ITS_SUCCESS, status);

    status = psa_its_remove(5);
    TEST_ASSERT_NOT_EQUAL(PSA_ITS_SUCCESS, status);

    info.size = 0;
    info.flags = PSA_ITS_FLAG_NONE;
    status = psa_its_get_info(5, &info);
    TEST_ASSERT_EQUAL(PSA_ITS_SUCCESS, status);
    TEST_ASSERT_EQUAL(TEST_BUFF_SIZE, info.size);
    TEST_ASSERT_EQUAL(PSA_ITS_FLAG_WRITE_ONCE, info.flags);
}

utest::v1::status_t case_teardown_handler(const Case *const source, const size_t passed, const size_t failed, const failure_t reason)
{
    psa_status_t status;
    status = mbed_psa_reboot_and_request_new_security_state(PSA_LIFECYCLE_ASSEMBLY_AND_TEST);
    TEST_ASSERT_EQUAL(PSA_LIFECYCLE_SUCCESS, status);
    return greentea_case_teardown_handler(source, passed, failed, reason);
}

utest::v1::status_t case_setup_handler(const Case *const source, const size_t index_of_case)
{
    psa_status_t status;
    status = mbed_psa_reboot_and_request_new_security_state(PSA_LIFECYCLE_ASSEMBLY_AND_TEST);
    TEST_ASSERT_EQUAL(PSA_LIFECYCLE_SUCCESS, status);
    return greentea_case_setup_handler(source, index_of_case);
}

Case cases[] = {
    Case("PSA prot internal storage - Basic", case_setup_handler, pits_test, case_teardown_handler),
    Case("PSA prot internal storage - Write-once", case_setup_handler, pits_write_once_test, case_teardown_handler)
};

utest::v1::status_t greentea_test_setup(const size_t number_of_cases)
{
#ifndef NO_GREENTEA
    GREENTEA_SETUP(60, "default_auto");
#endif
    return greentea_test_setup_handler(number_of_cases);
}

Specification specification(greentea_test_setup, cases, greentea_test_teardown_handler);

int main()
{
    return !Harness::run(specification);
}