1
0
mirror of https://github.com/RIOT-OS/RIOT.git synced 2025-12-25 14:33:52 +01:00

sys/net/gnrc_pktbuf: move CANARY to header file

This commit is contained in:
Benjamin Valentin 2024-11-14 15:27:35 +01:00
parent 278e8f53b7
commit ea1670ab0f
2 changed files with 24 additions and 15 deletions

View File

@ -65,6 +65,23 @@ extern "C" {
#endif
/** @} */
/**
* @brief Enable use-after-free/out of bounds write detection in gnrc_pktbuf
*/
#ifndef CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE
#define CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE (0)
#endif
/**
* @brief Canary value for free pktbuf memory when
* @see CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE is enabled.
*
* Unallocated pktbuf memory is filled with this value and overwritten
* with `~GNRC_PKTBUF_CANARY` when handed out.
* This way we can try to detect when unallocated memory was used.
*/
#define GNRC_PKTBUF_CANARY (0x55)
/**
* @brief Initializes packet buffer module.
*/

View File

@ -38,15 +38,6 @@
#define ENABLE_DEBUG 0
#include "debug.h"
/**
* @brief enable use-after-free/out of bounds write detection
*/
#ifndef CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE
#define CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE (0)
#endif
#define CANARY 0x55
static alignas(sizeof(_unused_t)) uint8_t _static_buf[CONFIG_GNRC_PKTBUF_SIZE];
static_assert((CONFIG_GNRC_PKTBUF_SIZE % sizeof(_unused_t)) == 0,
"CONFIG_GNRC_PKTBUF_SIZE has to be a multiple of 8");
@ -79,7 +70,7 @@ void gnrc_pktbuf_init(void)
{
mutex_lock(&gnrc_pktbuf_mutex);
if (CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE) {
memset(_static_buf, CANARY, sizeof(_static_buf));
memset(_static_buf, GNRC_PKTBUF_CANARY, sizeof(_static_buf));
}
/* Silence false -Wcast-align: _static_buf has qualifier
* `alignas(_unused_t)`, so it is guaranteed to be safe */
@ -426,7 +417,7 @@ static void *_pktbuf_alloc(size_t size)
const void *mismatch;
if (CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE &&
(mismatch = memchk(ptr + 1, CANARY, size - sizeof(_unused_t)))) {
(mismatch = memchk(ptr + 1, GNRC_PKTBUF_CANARY, size - sizeof(_unused_t)))) {
printf("[%p] mismatch at offset %"PRIuPTR"/%" PRIuSIZE
" (ignoring %" PRIuSIZE " initial bytes that were repurposed)\n",
(void *)ptr, (uintptr_t)mismatch - (uintptr_t)ptr, size, sizeof(_unused_t));
@ -437,7 +428,7 @@ static void *_pktbuf_alloc(size_t size)
}
if (CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE) {
/* clear out canary */
memset(ptr, ~CANARY, size);
memset(ptr, ~GNRC_PKTBUF_CANARY, size);
}
return (void *)ptr;
@ -455,7 +446,7 @@ static inline _unused_t *_merge(_unused_t *a, _unused_t *b)
a->next = b->next;
a->size = b->size + ((uint8_t *)b - (uint8_t *)a);
if (CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE) {
memset(b, CANARY, sizeof(*b));
memset(b, GNRC_PKTBUF_CANARY, sizeof(*b));
}
return a;
}
@ -477,12 +468,13 @@ void gnrc_pktbuf_free_internal(void *data, size_t size)
if (CONFIG_GNRC_PKTBUF_CHECK_USE_AFTER_FREE) {
/* check if the data has already been marked as free */
size_t chk_len = _align(size) - sizeof(*new);
if (chk_len && !memchk((uint8_t *)data + sizeof(*new), CANARY, chk_len)) {
if (chk_len &&
!memchk((uint8_t *)data + sizeof(*new), GNRC_PKTBUF_CANARY, chk_len)) {
printf("pktbuf: double free detected! (at %p, len=%u)\n",
data, (unsigned)_align(size));
DEBUG_BREAKPOINT(2);
}
memset(data, CANARY, _align(size));
memset(data, GNRC_PKTBUF_CANARY, _align(size));
}
while (ptr && (((void *)ptr) < data)) {