From 1f22dafe13fbbe98bb7c197a025222407c3ffbca Mon Sep 17 00:00:00 2001
From: Praveen Chandran <PraveenBabu.Chandran@infineon.com>
Date: Mon, 21 Feb 2022 08:50:36 -0800
Subject: [PATCH] Update persistent storage app for P6 platform (#15113)

---
 examples/persistent-storage/p6/main.cpp | 52 ++++++++++++++++---------
 1 file changed, 34 insertions(+), 18 deletions(-)

diff --git a/examples/persistent-storage/p6/main.cpp b/examples/persistent-storage/p6/main.cpp
index 2c5dabc7d09d18..03a5132aa73d99 100644
--- a/examples/persistent-storage/p6/main.cpp
+++ b/examples/persistent-storage/p6/main.cpp
@@ -19,37 +19,53 @@
 #include "AppConfig.h"
 #include "KeyValueStorageTest.h"
 #include "init_p6Platform.h"
+#include <FreeRTOS.h>
 #include <platform/CHIPDeviceLayer.h>
 #include <platform/KeyValueStoreManager.h>
+#include <task.h>
 
 using namespace chip;
 
-int main(int argc, char * argv[])
+static TaskHandle_t sTestTaskHandle;
+void TestTask(void * pvParameter)
 {
     CHIP_ERROR err = CHIP_NO_ERROR;
-    init_p6Platform();
+    err            = chip::DeviceLayer::PersistedStorage::KeyValueStoreMgrImpl().Init();
+    if (err != CHIP_NO_ERROR)
+    {
+        P6_LOG("PersistedStorage::KeyValueStoreMgrImpl().Init() failed");
+        return;
+    }
+    while (1)
+    {
+        P6_LOG("Running Tests:");
+        chip::RunKvsTest();
+        vTaskDelay(60000); // Run every minute
+    }
+}
+
+extern "C" void vApplicationDaemonTaskStartupHook()
+{
+    // Init Chip memory management before the stack
+    chip::Platform::MemoryInit();
 
-    err = chip::Platform::MemoryInit();
-    SuccessOrExit(err);
+    // Run tests
+    xTaskCreate(TestTask, "Test", 2048, NULL, 1, &sTestTaskHandle);
+}
 
-    err = chip::DeviceLayer::PersistedStorage::KeyValueStoreMgrImpl().Init();
-    SuccessOrExit(err);
+int main(int argc, char * argv[])
+{
+    init_p6Platform();
 
     P6_LOG("=============================================\n");
     P6_LOG("chip-p6-persistent-storage-example starting\n");
     P6_LOG("=============================================\n");
 
-    while (1)
-    {
-        P6_LOG("Running Tests:\n");
-        chip::RunKvsTest();
-        cyhal_system_delay_ms(60000); // Run every minute
-    }
+    P6_LOG("Starting FreeRTOS scheduler");
 
-exit:
-    if (err != CHIP_NO_ERROR)
-    {
-        return 1;
-    }
-    return 0;
+    vTaskStartScheduler();
+
+    // Should never get here.
+    P6_LOG("vTaskStartScheduler() failed");
+    return -1;
 }