mirror of
https://github.com/ARMmbed/mbedtls.git
synced 2025-07-17 21:14:07 +08:00
0bdb6dc079
Asan poisons memory with an 8-byte granularity. We want to make sure that the whole specified region is poisoned (our typical use case is a heap-allocated object, and we want to poison the whole object, and we don't care about the bytes after the end of the object and up to the beginning of the next object). So align the start and end of the region to (un)poison to an 8-byte boundary. Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
55 lines
1.4 KiB
C
55 lines
1.4 KiB
C
/**
|
|
* \file memory.c
|
|
*
|
|
* \brief Helper functions related to testing memory management.
|
|
*/
|
|
|
|
/*
|
|
* Copyright The Mbed TLS Contributors
|
|
* SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
|
|
*/
|
|
|
|
#include <test/helpers.h>
|
|
#include <test/macros.h>
|
|
#include <test/memory.h>
|
|
|
|
#if defined(MBEDTLS_TEST_HAVE_ASAN)
|
|
#include <sanitizer/asan_interface.h>
|
|
#include <stdint.h>
|
|
#endif
|
|
|
|
#if defined(MBEDTLS_TEST_HAVE_ASAN)
|
|
static void align_for_asan(const unsigned char **p_ptr, size_t *p_size)
|
|
{
|
|
uintptr_t start = (uintptr_t) *p_ptr;
|
|
uintptr_t end = start + (uintptr_t) *p_size;
|
|
/* ASan can only poison regions with 8-byte alignment, and only poisons a
|
|
* region if it's fully within the requested range. We want to poison the
|
|
* whole requested region and don't mind a few extra bytes. Therefore,
|
|
* align start down to an 8-byte boundary, and end up to an 8-byte
|
|
* boundary. */
|
|
start = start & ~(uintptr_t) 7;
|
|
end = (end + 7) & ~(uintptr_t) 7;
|
|
*p_ptr = (const unsigned char *) start;
|
|
*p_size = end - start;
|
|
}
|
|
|
|
void mbedtls_test_memory_poison(const unsigned char *ptr, size_t size)
|
|
{
|
|
if (size == 0) {
|
|
return;
|
|
}
|
|
align_for_asan(&ptr, &size);
|
|
__asan_poison_memory_region(ptr, size);
|
|
}
|
|
|
|
void mbedtls_test_memory_unpoison(const unsigned char *ptr, size_t size)
|
|
{
|
|
if (size == 0) {
|
|
return;
|
|
}
|
|
align_for_asan(&ptr, &size);
|
|
__asan_unpoison_memory_region(ptr, size);
|
|
}
|
|
#endif /* Asan */
|