95 lines
3.2 KiB
C
95 lines
3.2 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include "unity.h"
|
|
#include "esp_partition.h"
|
|
|
|
|
|
TEST_CASE("Can read partition table", "[partition]")
|
|
{
|
|
|
|
const esp_partition_t *p = esp_partition_find_first(ESP_PARTITION_TYPE_APP, ESP_PARTITION_SUBTYPE_ANY, NULL);
|
|
TEST_ASSERT_NOT_NULL(p);
|
|
TEST_ASSERT_EQUAL(p->address, 0x10000);
|
|
TEST_ASSERT_EQUAL(p->subtype, ESP_PARTITION_SUBTYPE_APP_FACTORY);
|
|
|
|
esp_partition_iterator_t it = esp_partition_find(ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_ANY, NULL);
|
|
TEST_ASSERT_NOT_NULL(it);
|
|
int count = 0;
|
|
for (; it != NULL; it = esp_partition_next(it)) {
|
|
const esp_partition_t *p = esp_partition_get(it);
|
|
TEST_ASSERT_NOT_NULL(p);
|
|
++count;
|
|
}
|
|
esp_partition_iterator_release(it);
|
|
TEST_ASSERT_EQUAL(count, 2);
|
|
|
|
printf("%d\n", __builtin_clz(count));
|
|
}
|
|
|
|
TEST_CASE("Can write, read, mmap partition", "[partition][ignore]")
|
|
{
|
|
const esp_partition_t *p = esp_partition_find_first(ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_ANY, NULL);
|
|
TEST_ASSERT_NOT_NULL(p);
|
|
const size_t max_size = 2 * SPI_FLASH_SEC_SIZE;
|
|
uint8_t *data = (uint8_t *) malloc(max_size);
|
|
TEST_ASSERT_NOT_NULL(data);
|
|
|
|
TEST_ASSERT_EQUAL(ESP_OK, esp_partition_erase_range(p, 0, p->size));
|
|
|
|
srand(0);
|
|
size_t block_size;
|
|
for (size_t offset = 0; offset < p->size; offset += block_size) {
|
|
block_size = ((rand() + 4) % max_size) & (~0x3);
|
|
size_t left = p->size - offset;
|
|
if (block_size > left) {
|
|
block_size = left;
|
|
}
|
|
for (size_t i = 0; i < block_size / 4; ++i) {
|
|
((uint32_t *) (data))[i] = rand();
|
|
if (i == 0 && offset == 0) {
|
|
printf("write: %08x\n", ((uint32_t *) (data))[i]);
|
|
}
|
|
}
|
|
TEST_ASSERT_EQUAL(ESP_OK, esp_partition_write(p, offset, data, block_size));
|
|
}
|
|
|
|
srand(0);
|
|
for (size_t offset = 0; offset < p->size; offset += block_size) {
|
|
block_size = ((rand() + 4) % max_size) & (~0x3);
|
|
size_t left = p->size - offset;
|
|
if (block_size > left) {
|
|
block_size = left;
|
|
}
|
|
TEST_ASSERT_EQUAL(ESP_OK, esp_partition_read(p, offset, data, block_size));
|
|
for (size_t i = 0; i < block_size / 4; ++i) {
|
|
TEST_ASSERT_EQUAL(rand(), ((uint32_t *) data)[i]);
|
|
}
|
|
}
|
|
|
|
free(data);
|
|
|
|
const uint32_t *mmap_data;
|
|
spi_flash_mmap_handle_t mmap_handle;
|
|
size_t begin = 3000;
|
|
size_t size = 12000;
|
|
TEST_ASSERT_EQUAL(ESP_OK, esp_partition_mmap(p, begin, size, SPI_FLASH_MMAP_DATA,
|
|
(const void **)&mmap_data, &mmap_handle));
|
|
srand(0);
|
|
for (size_t offset = 0; offset < p->size; offset += block_size) {
|
|
block_size = ((rand() + 4) % max_size) & (~0x3);
|
|
size_t left = p->size - offset;
|
|
if (block_size > left) {
|
|
block_size = left;
|
|
}
|
|
for (size_t i = 0; i < block_size / 4; ++i) {
|
|
size_t pos = offset + i * 4;
|
|
uint32_t expected = rand();
|
|
if (pos < begin || pos >= (begin + size)) {
|
|
continue;
|
|
}
|
|
TEST_ASSERT_EQUAL(expected, mmap_data[(pos - begin) / 4]);
|
|
}
|
|
}
|
|
|
|
spi_flash_munmap(mmap_handle);
|
|
}
|