#include <stdio.h>
#include <string.h>
#include "unity.h"
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "freertos/semphr.h"
#include "sdkconfig.h"
#include "test_utils.h"
#include "esp_expression_with_stack.h"

#define SHARED_STACK_SIZE 8192

static StackType_t *shared_stack_sp = NULL;

void external_stack_function(void)
{
    printf("Executing this printf from external stack! sp=%p\n", get_sp());
    shared_stack_sp = (StackType_t *)get_sp();
}

void another_external_stack_function(void)
{
    //We can even use Freertos resources inside of this context.
    printf("We can even use FreeRTOS resources... yielding, sp=%p\n", get_sp());
    taskYIELD();
    shared_stack_sp = (StackType_t *)get_sp();
}

#if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32C3)

TEST_CASE("test printf using shared buffer stack", "[newlib]")
{
    portSTACK_TYPE *shared_stack = malloc(SHARED_STACK_SIZE);

    TEST_ASSERT_NOT_NULL(shared_stack);

    SemaphoreHandle_t printf_lock = xSemaphoreCreateMutex();
    TEST_ASSERT_NOT_NULL(printf_lock);
    printf("current task sp: %p\n", get_sp());
    printf("shared_stack: %p\n", (void *)shared_stack);
    printf("shared_stack expected top: %p\n", (void *)(shared_stack + SHARED_STACK_SIZE));


    esp_execute_shared_stack_function(printf_lock,
                                    shared_stack,
                                    SHARED_STACK_SIZE,
                                    external_stack_function);

    TEST_ASSERT(((shared_stack_sp >= shared_stack) &&
                (shared_stack_sp < (shared_stack + SHARED_STACK_SIZE))));

    esp_execute_shared_stack_function(printf_lock,
                                    shared_stack,
                                    SHARED_STACK_SIZE,
                                    another_external_stack_function);

    TEST_ASSERT(((shared_stack_sp >= shared_stack) &&
                (shared_stack_sp < (shared_stack + SHARED_STACK_SIZE))));

    vSemaphoreDelete(printf_lock);
    free(shared_stack);
}

#endif