/*
 * SPDX-FileCopyrightText: 2022-2024 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Unlicense OR CC0-1.0
 */
#include "unity.h"
#include "stdio.h"

#include "esp_heap_caps.h"

#define PASS_CODE 0x9876BAAB
#define ALLOC_SIZE 16

static bool block_found = false;
static void heap_walker(walker_heap_into_t heap_info, walker_block_info_t block_info, void* user_data)
{
    TEST_ASSERT(*(uint32_t*)user_data == PASS_CODE);
    TEST_ASSERT_NOT_NULL(block_info.ptr);

    uint32_t metadata_size_head = 0;
    uint32_t metadata_size_tail = 0;
#if !CONFIG_HEAP_POISONING_DISABLED
    metadata_size_head += 8; // sizeof(poison_head_t)
    metadata_size_tail += 4; // sizeof(poison_tail_t)
#endif
#if CONFIG_HEAP_TASK_TRACKING
    metadata_size_head += 4; // sizeof(TaskHandle_t)
#endif

    /* look for the first 4 bytes pass code to identify the memory allocated in the test */
    const uint32_t pass_code = *((uint32_t*)block_info.ptr + (metadata_size_head / sizeof(uint32_t)));
    if (pass_code == PASS_CODE) {
        TEST_ASSERT(block_info.size == ALLOC_SIZE + metadata_size_head + metadata_size_tail);
        TEST_ASSERT_TRUE(block_info.used);
        block_found = true;
    }
}

/* This test assures the proper functioning of heap_caps_walk and heap_caps_walk_all
 */
TEST_CASE("heap walker", "[heap]")
{
    /* Allocate memory using the MALLOC_CAP_DEFAULT capability */
    void *default_ptr = heap_caps_malloc(ALLOC_SIZE, MALLOC_CAP_DEFAULT);
    TEST_ASSERT_NOT_NULL(default_ptr);

    /* Write the pass code in the first word of the allocated memory */
    *((uint32_t*)default_ptr) = (uint32_t)PASS_CODE;

    /* call the heap_caps_walker to make sure the hook function is triggered
     * and check that the allocated memory is found while walking the heap */
    uint32_t user_code = PASS_CODE;
    heap_caps_walk_all(heap_walker, &user_code);
    TEST_ASSERT_TRUE(block_found);
}