Commander1024
cdc9c7335a
Generate exact-hash SDK source overrides without modifying dependencies. Harden SSH allocation and algorithm policy, tighten web authentication cleanup, and add focused host contract tests and documentation.
69 lines
2.0 KiB
C
69 lines
2.0 KiB
C
/* SPDX-License-Identifier: GPL-3.0-only */
|
|
#include <assert.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
typedef struct { void *p; size_t n; bool wipe; } allocation;
|
|
static allocation allocations[64];
|
|
static unsigned live, calls, fail_at, wiped_frees;
|
|
static size_t secret_size;
|
|
static bool all_secret;
|
|
static void *test_alloc(size_t n, bool clear)
|
|
{
|
|
if (++calls == fail_at) return NULL;
|
|
void *p = clear ? calloc(1, n + 16) : malloc(n + 16);
|
|
assert(p);
|
|
if (!clear) memset(p, 0xa5, n);
|
|
memset((unsigned char *)p + n, 0x7b, 16);
|
|
for (unsigned i = 0; i < 64; ++i) if (!allocations[i].p) {
|
|
allocations[i] = (allocation){p, n, all_secret || n == secret_size};
|
|
++live;
|
|
return p;
|
|
}
|
|
abort();
|
|
}
|
|
static void test_free(void *p)
|
|
{
|
|
if (!p) return;
|
|
for (unsigned i = 0; i < 64; ++i) if (allocations[i].p == p) {
|
|
for (size_t j = 0; j < 16; ++j)
|
|
assert(((unsigned char *)p)[allocations[i].n + j] == 0x7b);
|
|
if (allocations[i].wipe) {
|
|
for (size_t j = 0; j < allocations[i].n; ++j)
|
|
assert(((unsigned char *)p)[j] == 0);
|
|
++wiped_frees;
|
|
}
|
|
memset(p, 0xdd, allocations[i].n);
|
|
allocations[i].p = NULL;
|
|
--live;
|
|
free(p);
|
|
return;
|
|
}
|
|
assert(!"double free or unowned allocation");
|
|
}
|
|
static void mark_secret(void *p)
|
|
{
|
|
for (unsigned i = 0; i < 64; ++i) if (allocations[i].p == p) {
|
|
allocations[i].wipe = true;
|
|
return;
|
|
}
|
|
abort();
|
|
}
|
|
#define malloc(n) test_alloc((n), false)
|
|
#define calloc(n, s) test_alloc((n) * (s), true)
|
|
#define free(p) test_free(p)
|
|
#define ESP_LOGE(...) ((void)0)
|
|
#define ESP_LOGD(...) ((void)0)
|
|
#define ESP_LOGI(...) ((void)0)
|
|
#define ESP_LOGW(...) ((void)0)
|
|
#define ESP_OK 0
|
|
#define ESP_FAIL -1
|
|
#define ESP_ERR_NO_MEM -2
|
|
#define ESP_ERR_INVALID_ARG -3
|
|
#define ESP_ERR_INVALID_STATE -4
|
|
#define ESP_ERR_NOT_SUPPORTED -5
|
|
typedef int esp_err_t;
|