examples/vad_an552/ns_side/main_ns.c - TF-M/tf-m-extras - TrustedFirmware Git Browser

 /*
  * Copyright (c) 2017-2022 Arm Limited
  *
  * 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.
  */

 #include "stdio.h"
 #include "string.h"
 #include "stdbool.h"
 #include "uart_stdout.h"
 #include "print_log.h"

 #include "FreeRTOS.h"
 #include "task.h"
 #include "FreeRTOS_IP.h"

 #include "psa/protected_storage.h"
 #include "psa/crypto.h"
 #include "psa/update.h"
 #include "psa/internal_trusted_storage.h"
 #include "vad_an552.h"

 #include "aws_dev_mode_key_provisioning.h"
 #include "ota_provision.h"

 #include "version/application_version.h"
 #include "aws_demo.h"
 #include "ota.h"

 #define FIRST_BOOT_ITS_UID                  (1U)
 #define BOOT_PATTERN                        (0x55)

 extern uint32_t tfm_ns_interface_init(void);
 extern void vApplicationIPInit(void);
 extern int mbedtls_platform_set_calloc_free(
                                         void* (*calloc_func)(size_t, size_t),
                                         void (*free_func)(void *));
 extern void publishToAWSTopic(const char *msg);

 static void* prvCalloc(size_t xNmemb, size_t xSize);
 static bool is_first_boot(void);
 static void write_boot_pattern(void);
 static void accept_primary_slot_image(void);

 /*
  * Semihosting is a mechanism that enables code running on an ARM target
  * to communicate and use the Input/Output facilities of a host computer
  * that is running a debugger.
  * There is an issue where if you use armclang at -O0 optimisation with
  * no parameters specified in the main function, the initialisation code
  * contains a breakpoint for semihosting by default. This will stop the
  * code from running before main is reached.
  * Semihosting can be disabled by defining __ARM_use_no_argv symbol
  * (or using higher optimization level).
  */
 #if defined (__ARMCC_VERSION) && (__ARMCC_VERSION >= 6010050)
 __asm("  .global __ARM_use_no_argv\n");
 #endif

 /*
  * With current clib settings there is no support for errno in case of Armclang
  * but OTA sources require it.
  */
 #if defined (__ARMCC_VERSION)
 int errno;
 #endif

 psa_key_handle_t xOTACodeVerifyKeyHandle = 0xAA;

 static void accept_primary_slot_image(void)
 {
     psa_image_id_t running_image = \
                 (psa_image_id_t)FWU_CALCULATE_IMAGE_ID(FWU_IMAGE_ID_SLOT_ACTIVE,
                                                     FWU_IMAGE_TYPE_FULL,
                                                     0);
     vLoggingPrintf("Accepting image by setting image_ok flag to 0x1 in MCUBOOT trailer");
     if (psa_fwu_accept(running_image) != PSA_SUCCESS) {
         vLoggingPrintf("Accept failed");
     }
 }

 static bool is_first_boot(void)
 {
     psa_status_t status = PSA_ERROR_GENERIC_ERROR;
     const psa_storage_uid_t uid = FIRST_BOOT_ITS_UID;
     uint8_t boot_pattern_in_its = 0;
     size_t read_data_length = 0;

     status = psa_its_get(uid, 0, 1, &boot_pattern_in_its,
                          &read_data_length);
     if(status != PSA_SUCCESS) {
         vLoggingPrintf("Could not read ITS to determine boot counter");
         vLoggingPrintf("Assuming first boot");
         return true;
     }

     if(boot_pattern_in_its == BOOT_PATTERN) {
         vLoggingPrintf("Boot pattern in ITS matches, not first boot");
         return false;
     }
     else {
         vLoggingPrintf("Boot pattern in ITS doesn't match, first boot");
         return true;
     }
 }

 static void write_boot_pattern(void)
 {
     psa_status_t status = PSA_ERROR_GENERIC_ERROR;
     const psa_storage_uid_t uid = FIRST_BOOT_ITS_UID;
     const psa_storage_create_flags_t flags = PSA_STORAGE_FLAG_NONE;
     uint8_t first_boot_pattern = BOOT_PATTERN;

     /* Write the pattern to ITS */
     status = psa_its_set(uid, 1, &first_boot_pattern, flags);
     if(status == PSA_SUCCESS) {
         vLoggingPrintf("Boot pattern has been written to the ITS");
     }
     else {
         vLoggingPrintf("Couldn't write boot pattern to ITS");
     }
 }

 static void mainTask(void *pvParameters)
 {
     uint32_t vad_status;
     uint32_t vad_freq;
     OtaState_t ota_state;
     TickType_t base_tick, curr_tick;
     char message[256];

 #ifdef VAD_AN552_NO_CONNECTIVITY
     while(1) {
         vad_an552_start_vad();

         do {
             vad_an552_query_vad(&vad_status);
         } while (vad_status != VAD_VOICE_RECORDED);

         vad_an552_get_freq(&vad_freq);

         vLoggingPrintf("Voice detected with most energy at %d Hz", vad_freq);
     }
 #else
     while(1) {
         vLoggingPrintf("==== Start OTA task ====");
         DEMO_RUNNER_RunDemos();

         do {
             vTaskDelay(pdMS_TO_TICKS(10));
             ota_state = OTA_GetState();
         }
         while (ota_state == OtaAgentStateInit ||
                ota_state == OtaAgentStateReady ||
                ota_state == OtaAgentStateStopped ||
                ota_state == OtaAgentStateRequestingJob);

         vTaskDelay(pdMS_TO_TICKS(5000));

         if (OtaAgentStateWaitingForJob == OTA_GetState()) {
             vLoggingPrintf("==== Stop OTA task ====");

             OTA_Shutdown(0, 1);
             while (xTaskGetHandle("iot_thread") != NULL) {
                 vTaskDelay(pdMS_TO_TICKS(100));
             }

             base_tick = xTaskGetTickCount();

             vLoggingPrintf("==== Start listening ====");
             vad_an552_start_vad();

             while(1) {
                 vad_an552_query_vad(&vad_status);
                 if (vad_status == VAD_VOICE_RECORDED) {
                     vad_an552_get_freq(&vad_freq);

                     vLoggingPrintf("Voice detected with most energy at %d Hz",
                                    vad_freq);
                     vLoggingPrintf("==== Send message to cloud ====");
                     sprintf(message, "Voice detected with most energy at %d Hz",
                             vad_freq);
                     publishToAWSTopic(message);

                     /* Message sending takes some time so timeout checked
                      * before restarting the mic algorithm. */
                     curr_tick = xTaskGetTickCount();
                     if ((curr_tick - base_tick) > pdMS_TO_TICKS(60000)) {
                         vad_an552_stop_vad();
                         break;
                     }

                     vLoggingPrintf("==== Start listening ====");
                     vad_an552_start_vad();
                 }

                 curr_tick = xTaskGetTickCount();
                 if ((curr_tick - base_tick) > pdMS_TO_TICKS(60000)) {
                     vLoggingPrintf("==== Stop listening ====");
                     vad_an552_stop_vad();
                     break;
                 }
             }
         } else {
             /* OTA started, nothing to do */
             while (1) {
                 vTaskDelay(pdMS_TO_TICKS(10000));
             }
         }
     }
 #endif
 }

 int main()
 {
     stdio_init();
     vUARTLockInit();
     tfm_ns_interface_init();

     GetImageVersionPSA(FWU_IMAGE_TYPE_FULL);
     vLoggingPrintf("Application firmware version: %d.%d.%d",
                    appFirmwareVersion.u.x.major,
                    appFirmwareVersion.u.x.minor,
                    appFirmwareVersion.u.x.build);

 #ifdef VAD_AN552_NO_CONNECTIVITY
     xTaskCreate(mainTask, "main task", configMINIMAL_STACK_SIZE*2, NULL,
                 configMAX_PRIORITIES-2, NULL);
 #else
     mbedtls_platform_set_calloc_free(prvCalloc, vPortFree);

     if(is_first_boot()) {
         accept_primary_slot_image();
         vDevModeKeyProvisioning();
         ota_privision_code_signing_key(&xOTACodeVerifyKeyHandle);
         write_boot_pattern();
     }

     /* Initialise the RTOS's TCP/IP stack.  The tasks that use the network
     are created in the vApplicationIPNetworkEventHook() hook function
     below.  The hook function is called when the network connects. */
     vApplicationIPInit();
 #endif

     vLoggingPrintf("Starting FreeRTOS scheduler");
 #ifndef VAD_AN552_NO_CONNECTIVITY
     vLoggingPrintf("Waiting for network");
 #endif
     /* Start the scheduler itself. */
     vTaskStartScheduler();

     while (1)
     {
     }
 }

 void vApplicationIPNetworkEventHook(eIPCallbackEvent_t eNetworkEvent)
 {
 #ifndef VAD_AN552_NO_CONNECTIVITY
     if (eNetworkEvent == eNetworkUp) {
         vLoggingPrintf("Network connection established");
         xTaskCreate(mainTask, "main task", configMINIMAL_STACK_SIZE*2, NULL,
         configMAX_PRIORITIES-4, NULL);
     }
 #endif
 }

 /* Functions needed for mbedtls build */
 static void * prvCalloc(size_t xNmemb,
                         size_t xSize)
 {
     void * pvNew = pvPortMalloc(xNmemb * xSize);

     if( NULL != pvNew ) {
         memset(pvNew, 0, xNmemb * xSize);
     }

     return pvNew;
 }

 int mbedtls_hardware_poll(void *data,
                           unsigned char *output, size_t len, size_t *olen)
 {
     psa_status_t status;

     (void) (data);

     if (output == NULL || olen == NULL) {
         return -1;
     }

     status = psa_generate_random(output, len);
     if (status != PSA_SUCCESS) {
         return -1;
     }

     *olen = len;

     return 0;
 }
	/*
	* Copyright (c) 2017-2022 Arm Limited
	*
	* 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.
	*/

	#include "stdio.h"
	#include "string.h"
	#include "stdbool.h"
	#include "uart_stdout.h"
	#include "print_log.h"

	#include "FreeRTOS.h"
	#include "task.h"
	#include "FreeRTOS_IP.h"

	#include "psa/protected_storage.h"
	#include "psa/crypto.h"
	#include "psa/update.h"
	#include "psa/internal_trusted_storage.h"
	#include "vad_an552.h"

	#include "aws_dev_mode_key_provisioning.h"
	#include "ota_provision.h"

	#include "version/application_version.h"
	#include "aws_demo.h"
	#include "ota.h"

	#define FIRST_BOOT_ITS_UID (1U)
	#define BOOT_PATTERN (0x55)

	extern uint32_t tfm_ns_interface_init(void);
	extern void vApplicationIPInit(void);
	extern int mbedtls_platform_set_calloc_free(
	void* (*calloc_func)(size_t, size_t),
	void (free_func)(void ));
	extern void publishToAWSTopic(const char *msg);

	static void* prvCalloc(size_t xNmemb, size_t xSize);
	static bool is_first_boot(void);
	static void write_boot_pattern(void);
	static void accept_primary_slot_image(void);

	/*
	* Semihosting is a mechanism that enables code running on an ARM target
	* to communicate and use the Input/Output facilities of a host computer
	* that is running a debugger.
	* There is an issue where if you use armclang at -O0 optimisation with
	* no parameters specified in the main function, the initialisation code
	* contains a breakpoint for semihosting by default. This will stop the
	* code from running before main is reached.
	* Semihosting can be disabled by defining __ARM_use_no_argv symbol
	* (or using higher optimization level).
	*/
	#if defined (__ARMCC_VERSION) && (__ARMCC_VERSION >= 6010050)
	__asm(" .global __ARM_use_no_argv\n");
	#endif

	/*
	* With current clib settings there is no support for errno in case of Armclang
	* but OTA sources require it.
	*/
	#if defined (__ARMCC_VERSION)
	int errno;
	#endif

	psa_key_handle_t xOTACodeVerifyKeyHandle = 0xAA;

	static void accept_primary_slot_image(void)
	{
	psa_image_id_t running_image = \
	(psa_image_id_t)FWU_CALCULATE_IMAGE_ID(FWU_IMAGE_ID_SLOT_ACTIVE,
	FWU_IMAGE_TYPE_FULL,
	0);
	vLoggingPrintf("Accepting image by setting image_ok flag to 0x1 in MCUBOOT trailer");
	if (psa_fwu_accept(running_image) != PSA_SUCCESS) {
	vLoggingPrintf("Accept failed");
	}
	}

	static bool is_first_boot(void)
	{
	psa_status_t status = PSA_ERROR_GENERIC_ERROR;
	const psa_storage_uid_t uid = FIRST_BOOT_ITS_UID;
	uint8_t boot_pattern_in_its = 0;
	size_t read_data_length = 0;

	status = psa_its_get(uid, 0, 1, &boot_pattern_in_its,
	&read_data_length);
	if(status != PSA_SUCCESS) {
	vLoggingPrintf("Could not read ITS to determine boot counter");
	vLoggingPrintf("Assuming first boot");
	return true;
	}

	if(boot_pattern_in_its == BOOT_PATTERN) {
	vLoggingPrintf("Boot pattern in ITS matches, not first boot");
	return false;
	}
	else {
	vLoggingPrintf("Boot pattern in ITS doesn't match, first boot");
	return true;
	}
	}

	static void write_boot_pattern(void)
	{
	psa_status_t status = PSA_ERROR_GENERIC_ERROR;
	const psa_storage_uid_t uid = FIRST_BOOT_ITS_UID;
	const psa_storage_create_flags_t flags = PSA_STORAGE_FLAG_NONE;
	uint8_t first_boot_pattern = BOOT_PATTERN;

	/* Write the pattern to ITS */
	status = psa_its_set(uid, 1, &first_boot_pattern, flags);
	if(status == PSA_SUCCESS) {
	vLoggingPrintf("Boot pattern has been written to the ITS");
	}
	else {
	vLoggingPrintf("Couldn't write boot pattern to ITS");
	}
	}

	static void mainTask(void *pvParameters)
	{
	uint32_t vad_status;
	uint32_t vad_freq;
	OtaState_t ota_state;
	TickType_t base_tick, curr_tick;
	char message[256];

	#ifdef VAD_AN552_NO_CONNECTIVITY
	while(1) {
	vad_an552_start_vad();

	do {
	vad_an552_query_vad(&vad_status);
	} while (vad_status != VAD_VOICE_RECORDED);

	vad_an552_get_freq(&vad_freq);

	vLoggingPrintf("Voice detected with most energy at %d Hz", vad_freq);
	}
	#else
	while(1) {
	vLoggingPrintf("==== Start OTA task ====");
	DEMO_RUNNER_RunDemos();

	do {
	vTaskDelay(pdMS_TO_TICKS(10));
	ota_state = OTA_GetState();
	}
	while (ota_state == OtaAgentStateInit \|\|
	ota_state == OtaAgentStateReady \|\|
	ota_state == OtaAgentStateStopped \|\|
	ota_state == OtaAgentStateRequestingJob);

	vTaskDelay(pdMS_TO_TICKS(5000));

	if (OtaAgentStateWaitingForJob == OTA_GetState()) {
	vLoggingPrintf("==== Stop OTA task ====");

	OTA_Shutdown(0, 1);
	while (xTaskGetHandle("iot_thread") != NULL) {
	vTaskDelay(pdMS_TO_TICKS(100));
	}

	base_tick = xTaskGetTickCount();

	vLoggingPrintf("==== Start listening ====");
	vad_an552_start_vad();

	while(1) {
	vad_an552_query_vad(&vad_status);
	if (vad_status == VAD_VOICE_RECORDED) {
	vad_an552_get_freq(&vad_freq);

	vLoggingPrintf("Voice detected with most energy at %d Hz",
	vad_freq);
	vLoggingPrintf("==== Send message to cloud ====");
	sprintf(message, "Voice detected with most energy at %d Hz",
	vad_freq);
	publishToAWSTopic(message);

	/* Message sending takes some time so timeout checked
	* before restarting the mic algorithm. */
	curr_tick = xTaskGetTickCount();
	if ((curr_tick - base_tick) > pdMS_TO_TICKS(60000)) {
	vad_an552_stop_vad();
	break;
	}

	vLoggingPrintf("==== Start listening ====");
	vad_an552_start_vad();
	}

	curr_tick = xTaskGetTickCount();
	if ((curr_tick - base_tick) > pdMS_TO_TICKS(60000)) {
	vLoggingPrintf("==== Stop listening ====");
	vad_an552_stop_vad();
	break;
	}
	}
	} else {
	/* OTA started, nothing to do */
	while (1) {
	vTaskDelay(pdMS_TO_TICKS(10000));
	}
	}
	}
	#endif
	}

	int main()
	{
	stdio_init();
	vUARTLockInit();
	tfm_ns_interface_init();

	GetImageVersionPSA(FWU_IMAGE_TYPE_FULL);
	vLoggingPrintf("Application firmware version: %d.%d.%d",
	appFirmwareVersion.u.x.major,
	appFirmwareVersion.u.x.minor,
	appFirmwareVersion.u.x.build);

	#ifdef VAD_AN552_NO_CONNECTIVITY
	xTaskCreate(mainTask, "main task", configMINIMAL_STACK_SIZE*2, NULL,
	configMAX_PRIORITIES-2, NULL);
	#else
	mbedtls_platform_set_calloc_free(prvCalloc, vPortFree);

	if(is_first_boot()) {
	accept_primary_slot_image();
	vDevModeKeyProvisioning();
	ota_privision_code_signing_key(&xOTACodeVerifyKeyHandle);
	write_boot_pattern();
	}

	/* Initialise the RTOS's TCP/IP stack. The tasks that use the network
	are created in the vApplicationIPNetworkEventHook() hook function
	below. The hook function is called when the network connects. */
	vApplicationIPInit();
	#endif

	vLoggingPrintf("Starting FreeRTOS scheduler");
	#ifndef VAD_AN552_NO_CONNECTIVITY
	vLoggingPrintf("Waiting for network");
	#endif
	/* Start the scheduler itself. */
	vTaskStartScheduler();

	while (1)
	{
	}
	}

	void vApplicationIPNetworkEventHook(eIPCallbackEvent_t eNetworkEvent)
	{
	#ifndef VAD_AN552_NO_CONNECTIVITY
	if (eNetworkEvent == eNetworkUp) {
	vLoggingPrintf("Network connection established");
	xTaskCreate(mainTask, "main task", configMINIMAL_STACK_SIZE*2, NULL,
	configMAX_PRIORITIES-4, NULL);
	}
	#endif
	}

	/* Functions needed for mbedtls build */
	static void * prvCalloc(size_t xNmemb,
	size_t xSize)
	{
	void * pvNew = pvPortMalloc(xNmemb * xSize);

	if( NULL != pvNew ) {
	memset(pvNew, 0, xNmemb * xSize);
	}

	return pvNew;
	}

	int mbedtls_hardware_poll(void *data,
	unsigned char output, size_t len, size_t olen)
	{
	psa_status_t status;

	(void) (data);

	if (output == NULL \|\| olen == NULL) {
	return -1;
	}

	status = psa_generate_random(output, len);
	if (status != PSA_SUCCESS) {
	return -1;
	}

	*olen = len;

	return 0;
	}