mirror of
https://github.com/jrcutler/threadless.io.git
synced 2024-07-07 10:35:49 +00:00
Rewrote allocator/allocation interface to track individual allocations and sizes
This commit is contained in:
parent
682d8484e0
commit
2777a2d44e
@ -2,8 +2,12 @@ cmake_minimum_required(VERSION 3.5)
|
||||
|
||||
set(CMAKE_C_FLAGS "-Wall -Wextra -pedantic -Werror -std=c99")
|
||||
|
||||
add_library (allocation src/allocation.c)
|
||||
target_include_directories (allocation PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})
|
||||
|
||||
add_library (coroutine src/coroutine.c)
|
||||
target_include_directories (coroutine PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})
|
||||
target_link_libraries (coroutine LINK_PUBLIC allocation)
|
||||
|
||||
add_executable (test-coroutine test/coroutine.c src/default_allocator.c)
|
||||
target_include_directories (coroutine PUBLIC ${CMAKE_CURRENT_SOURCE_DIR})
|
||||
|
42
src/allocation.c
Normal file
42
src/allocation.c
Normal file
@ -0,0 +1,42 @@
|
||||
/* threadless.io
|
||||
* Copyright (c) 2016 Justin R. Cutler
|
||||
* Licensed under the MIT License. See LICENSE file in the project root for
|
||||
* full license information.
|
||||
*/
|
||||
/** @file
|
||||
* allocation interface implementation
|
||||
* @author Justin R. Cutler <justin.r.cutler@gmail.com>
|
||||
*/
|
||||
|
||||
/* errno, ENOMEM */
|
||||
#include <errno.h>
|
||||
/* size_t, NULL */
|
||||
#include <stddef.h>
|
||||
|
||||
/* ... */
|
||||
#include <threadless/allocation.h>
|
||||
|
||||
|
||||
#ifndef SIZE_MAX
|
||||
# define SIZE_MAX ((size_t)-1)
|
||||
#endif
|
||||
|
||||
#define SQRT_SIZE_MAX_PLUS_1 ((size_t)1 << (sizeof(size_t) * 4))
|
||||
|
||||
|
||||
int allocation_realloc_array(allocation_t *allocation, size_t nmemb,
|
||||
size_t size)
|
||||
{
|
||||
/* test for multiplication overflow */
|
||||
if (((nmemb >= SQRT_SIZE_MAX_PLUS_1) || (size >= SQRT_SIZE_MAX_PLUS_1)) &&
|
||||
(nmemb != 0) && ((SIZE_MAX / nmemb) < size)) {
|
||||
/* overflow detected */
|
||||
errno = ENOMEM;
|
||||
return -1;
|
||||
}
|
||||
|
||||
size_t alloc_size = nmemb * size;
|
||||
|
||||
/* perform allocator action */
|
||||
return allocation->allocator->allocate(allocation, alloc_size);
|
||||
}
|
@ -19,6 +19,8 @@
|
||||
/* ucontext_t, getcontext, makecontext, swapcontext */
|
||||
#include <ucontext.h>
|
||||
|
||||
/* allocator_t allocation_t, allocation_init, allocation_realloc_array */
|
||||
#include <threadless/allocation.h>
|
||||
/* ... */
|
||||
#include <threadless/coroutine.h>
|
||||
|
||||
@ -29,11 +31,11 @@ enum {
|
||||
|
||||
|
||||
struct coroutine {
|
||||
allocator_t *allocator;
|
||||
ucontext_t context;
|
||||
ucontext_t caller;
|
||||
void *data;
|
||||
int status;
|
||||
allocation_t allocation;
|
||||
ucontext_t context;
|
||||
ucontext_t caller;
|
||||
void *data;
|
||||
int status;
|
||||
};
|
||||
|
||||
|
||||
@ -67,13 +69,15 @@ coroutine_t *coroutine_create(allocator_t *allocator,
|
||||
goto fail;
|
||||
}
|
||||
|
||||
coro = allocator_malloc(allocator, alloc_size);
|
||||
if (NULL == coro) {
|
||||
allocation_t allocation;
|
||||
allocation_init(&allocation, allocator);
|
||||
if (allocation_realloc_array(&allocation, 1, alloc_size)) {
|
||||
goto fail;
|
||||
}
|
||||
|
||||
coro = allocation.memory;
|
||||
memset(coro, 0, alloc_size);
|
||||
coro->allocator = allocator;
|
||||
coro->allocation = allocation;
|
||||
(void) getcontext(&coro->context);
|
||||
coro->context.uc_stack.ss_sp = coro + 1;
|
||||
coro->context.uc_stack.ss_size = stack_size;
|
||||
@ -92,8 +96,9 @@ fail:
|
||||
|
||||
void coroutine_destroy(coroutine_t *coro)
|
||||
{
|
||||
if (NULL != coro && NULL != coro->allocator) {
|
||||
allocator_free(coro->allocator, coro);
|
||||
if (NULL != coro) {
|
||||
allocation_t allocation = coro->allocation;
|
||||
allocation_free(&allocation);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -7,36 +7,31 @@
|
||||
* default allocator implementation
|
||||
* @author Justin R. Cutler <justin.r.cutler@gmail.com>
|
||||
*/
|
||||
/* errno, ENOMEM */
|
||||
#include <errno.h>
|
||||
|
||||
/* realloc */
|
||||
#include <stdlib.h>
|
||||
|
||||
/* ... */
|
||||
#include <threadless/allocator.h>
|
||||
#include <threadless/default_allocator.h>
|
||||
|
||||
|
||||
#ifndef SIZE_MAX
|
||||
# define SIZE_MAX ((size_t)-1)
|
||||
#endif
|
||||
|
||||
#define SQRT_SIZE_MAX_PLUS_1 ((size_t)1 << (sizeof(size_t) * 4))
|
||||
|
||||
|
||||
static void *default_allocate(allocator_t *allocator, void *ptr, size_t nmemb,
|
||||
size_t size)
|
||||
static int default_allocate(allocation_t *allocation, size_t size)
|
||||
{
|
||||
/* ignore allocator */
|
||||
(void) allocator;
|
||||
/* test for multiplication overflow */
|
||||
if (((nmemb >= SQRT_SIZE_MAX_PLUS_1) || (size >= SQRT_SIZE_MAX_PLUS_1)) &&
|
||||
(nmemb != 0) && ((SIZE_MAX / nmemb) < size)) {
|
||||
/* overflow detected */
|
||||
errno = ENOMEM;
|
||||
return NULL;
|
||||
}
|
||||
void *new_memory;
|
||||
|
||||
/* perform allocator action */
|
||||
return realloc(ptr, size * nmemb);
|
||||
new_memory = realloc(allocation->memory, size);
|
||||
|
||||
if ((NULL == new_memory) && (size != 0)) {
|
||||
/* realloc failed */
|
||||
return -1;
|
||||
}
|
||||
|
||||
/* update allocation */
|
||||
allocation->memory = new_memory;
|
||||
allocation->size = size;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
|
@ -14,7 +14,7 @@
|
||||
#include <stdlib.h>
|
||||
|
||||
/* allocator_t, allocator_get_default, allocator_destroy */
|
||||
#include <threadless/allocator.h>
|
||||
#include <threadless/default_allocator.h>
|
||||
/* ... */
|
||||
#include <threadless/coroutine.h>
|
||||
|
||||
|
119
threadless/allocation.h
Normal file
119
threadless/allocation.h
Normal file
@ -0,0 +1,119 @@
|
||||
/* threadless.io
|
||||
* Copyright (c) 2016 Justin R. Cutler
|
||||
* Licensed under the MIT License. See LICENSE file in the project root for
|
||||
* full license information.
|
||||
*/
|
||||
/** @file
|
||||
* allocation interface definition
|
||||
* @author Justin R. Cutler <justin.r.cutler@gmail.com>
|
||||
*/
|
||||
#ifndef THREADLESS_ALLOCATION_H
|
||||
#define THREADLESS_ALLOCATION_H
|
||||
|
||||
/* size_t, NULL */
|
||||
#include <stddef.h>
|
||||
|
||||
/** Memory allocator instance */
|
||||
typedef struct allocator allocator_t;
|
||||
|
||||
/** Memory allocation handle */
|
||||
typedef struct {
|
||||
/** allocator used to manage this memory */
|
||||
allocator_t *allocator;
|
||||
/** pointer to allocated memory */
|
||||
void *memory;
|
||||
/** current size of allocated memory */
|
||||
size_t size;
|
||||
} allocation_t;
|
||||
|
||||
/** Memory allocator function
|
||||
* @param[in,out] allocation memory allocation handle
|
||||
* @param size minimum new size of memory (or 0 to free)
|
||||
* @retval 0 success
|
||||
* @retval -1 error
|
||||
* @pre @p allocation must be initialized
|
||||
* @pre If @p allocation->memory is @c NULL and @p allocation->size is
|
||||
* non-zero, function shall allocate new memory
|
||||
* @pre If @p size is 0, function shall free allocated memory
|
||||
* @post Upon success, @p allocation has been updated to reflect changes
|
||||
* @post Upon failure, @p allocation has not been changed
|
||||
*/
|
||||
typedef int (allocator_function_t)(allocation_t *allocation, size_t size);
|
||||
|
||||
/** Memory allocator destructor function
|
||||
* @param[in,out] allocator allocator instance
|
||||
* @post @p allocator may no longer be used
|
||||
*/
|
||||
typedef void (allocator_destroy_function_t)(allocator_t *allocator);
|
||||
|
||||
/** Memory allocator instance structure */
|
||||
struct allocator {
|
||||
/** Memory (re)allocator function */
|
||||
allocator_function_t *const allocate;
|
||||
/** Memory allocator instance destructor */
|
||||
allocator_destroy_function_t *const destroy;
|
||||
};
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif /* __cplusplus */
|
||||
|
||||
/** Initialize an allocation handle
|
||||
* @param[out] allocation allocation
|
||||
* @param allocator allocator instance
|
||||
* @pre @p allocation must be initialized
|
||||
* @post @p allocation->allocator has been set to @p allocator,
|
||||
* @p allocation->memory has been set to @c NULL, and @p allocation->size
|
||||
* has been set to 0
|
||||
*/
|
||||
static inline void allocation_init(allocation_t *allocation,
|
||||
allocator_t *allocator)
|
||||
{
|
||||
allocation->allocator = allocator;
|
||||
allocation->memory = NULL;
|
||||
allocation->size = 0;
|
||||
}
|
||||
|
||||
/** @c realloc_array() helper function
|
||||
* @param[in,out] allocation memory allocation handle
|
||||
* @param nmemb number of elements to allocate (or 0 to free)
|
||||
* @param size size of each element to allocate (or 0 to free)
|
||||
* @retval 0 success
|
||||
* @retval -1 error
|
||||
* @pre @p allocation must point to an initialized, valid allocation
|
||||
* @pre If @p allocation->memory is @c NULL, function shall allocate new
|
||||
* memory
|
||||
* @pre If @p nmemb or @p size is 0, function shall free allocated memory
|
||||
* @post Upon success, @p allocation->memory and @p allocation->size have been
|
||||
* updated to reflect changes
|
||||
* @post Upon failure, @p allocation has not been changed
|
||||
* @note Upon detection of integer overflow, this function shall return
|
||||
* @c NULL (like @c calloc(3) and FreeBSD's @c realloc_array(3))
|
||||
*/
|
||||
int allocation_realloc_array(allocation_t *allocation, size_t nmemb,
|
||||
size_t size);
|
||||
|
||||
/** @c free() helper function
|
||||
* @param[in,out] allocation memory allocation handle
|
||||
* @pre @p allocation must point to an initialized, valid allocation
|
||||
* @post @p allocation->memory is set to @c NULL and @p allocation->size is set
|
||||
* to 0
|
||||
*/
|
||||
static inline void allocation_free(allocation_t *allocation)
|
||||
{
|
||||
(void) allocation_realloc_array(allocation, 0, 0);
|
||||
}
|
||||
|
||||
/** Memory allocator destructor helper function
|
||||
* @copydetails allocator_destroy_function_t
|
||||
*/
|
||||
static inline void allocator_destroy(allocator_t *allocator)
|
||||
{
|
||||
allocator->destroy(allocator);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif /* __cplusplus */
|
||||
|
||||
#endif /* THREADLESS_ALLOCATOR_H */
|
@ -1,96 +0,0 @@
|
||||
/* threadless.io
|
||||
* Copyright (c) 2016 Justin R. Cutler
|
||||
* Licensed under the MIT License. See LICENSE file in the project root for
|
||||
* full license information.
|
||||
*/
|
||||
/** @file
|
||||
* allocator interface definition
|
||||
* @author Justin R. Cutler <justin.r.cutler@gmail.com>
|
||||
*/
|
||||
#ifndef THREADLESS_ALLOCATOR_H
|
||||
#define THREADLESS_ALLOCATOR_H
|
||||
|
||||
/* size_t, NULL */
|
||||
#include <stddef.h>
|
||||
|
||||
/** Memory allocator instance */
|
||||
typedef struct allocator allocator_t;
|
||||
|
||||
/** Memory allocator function
|
||||
* @param[in,out] allocator allocator instance
|
||||
* @param[in,out] ptr memory to reallocate/free (or @c NULL to allocate)
|
||||
* @param nmemb number of elements to allocate (or 0 to free)
|
||||
* @param size size of each element to allocate (or 0 to free)
|
||||
* @retval non-NULL successful allocation/reallocation
|
||||
* @retval NULL successful free (@p nmemb or @p size was 0) or error
|
||||
* @note Upon detection of integer overflow, this function shall return
|
||||
* @c NULL (like @c calloc(3) and FreeBSD's @c realloc_array(3))
|
||||
*/
|
||||
typedef void *(allocator_function_t)(allocator_t *allocator, void *ptr,
|
||||
size_t nmemb, size_t size);
|
||||
|
||||
/** Memory allocator destructor function
|
||||
* @param[in,out] allocator allocator instance
|
||||
* @post @p allocator may no longer be used
|
||||
*/
|
||||
typedef void (allocator_destroy_function_t)(allocator_t *allocator);
|
||||
|
||||
/** Memory allocator instance structure */
|
||||
struct allocator {
|
||||
/** Memory allocator function */
|
||||
allocator_function_t *const allocate;
|
||||
/** Memory allocator instance destructor */
|
||||
allocator_destroy_function_t *const destroy;
|
||||
};
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif /* __cplusplus */
|
||||
|
||||
/** Get default allocator instance
|
||||
* @returns default allocator
|
||||
*/
|
||||
allocator_t *allocator_get_default(void);
|
||||
|
||||
/** @c malloc() helper function
|
||||
* @param[in,out] allocator allocator
|
||||
* @param size size to allocate
|
||||
* @retval Non-NULL allocated memory
|
||||
* @retval NULL error
|
||||
*/
|
||||
static inline void *allocator_malloc(allocator_t *allocator, size_t size)
|
||||
{
|
||||
return allocator->allocate(allocator, NULL, 1, size);
|
||||
}
|
||||
|
||||
/** @c realloc_array() helper function
|
||||
* @copydetails allocator_function_t
|
||||
*/
|
||||
static inline void *allocator_realloc_array(allocator_t *allocator, void *ptr,
|
||||
size_t nmemb, size_t size)
|
||||
{
|
||||
return allocator->allocate(allocator, ptr, nmemb, size);
|
||||
}
|
||||
|
||||
/** @c free() helper function
|
||||
* @param[in,out] allocator allocator
|
||||
* @param ptr memory to free
|
||||
*/
|
||||
static inline void allocator_free(allocator_t *allocator, void *ptr)
|
||||
{
|
||||
(void) allocator->allocate(allocator, ptr, 0, 0);
|
||||
}
|
||||
|
||||
/** Memory allocator destructor helper function
|
||||
* @copydetails allocator_destroy_function_t
|
||||
*/
|
||||
static inline void allocator_destroy(allocator_t *allocator)
|
||||
{
|
||||
allocator->destroy(allocator);
|
||||
}
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif /* __cplusplus */
|
||||
|
||||
#endif /* THREADLESS_ALLOCATOR_H */
|
@ -16,7 +16,7 @@
|
||||
#include <stddef.h>
|
||||
|
||||
/* allocator_t */
|
||||
#include <threadless/allocator.h>
|
||||
#include <threadless/allocation.h>
|
||||
|
||||
/** Opaque coroutine type */
|
||||
typedef struct coroutine coroutine_t;
|
||||
|
29
threadless/default_allocator.h
Normal file
29
threadless/default_allocator.h
Normal file
@ -0,0 +1,29 @@
|
||||
/* threadless.io
|
||||
* Copyright (c) 2016 Justin R. Cutler
|
||||
* Licensed under the MIT License. See LICENSE file in the project root for
|
||||
* full license information.
|
||||
*/
|
||||
/** @file
|
||||
* allocator interface definition
|
||||
* @author Justin R. Cutler <justin.r.cutler@gmail.com>
|
||||
*/
|
||||
#ifndef THREADLESS_DEFAULT_ALLOCATOR_H
|
||||
#define THREADLESS_DEFAULT_ALLOCATOR_H
|
||||
|
||||
/* allocator_t */
|
||||
#include <threadless/allocation.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif /* __cplusplus */
|
||||
|
||||
/** Get default allocator instance
|
||||
* @returns default allocator
|
||||
*/
|
||||
allocator_t *allocator_get_default(void);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif /* __cplusplus */
|
||||
|
||||
#endif /* THREADLESS_DEFAULT_ALLOCATOR_H */
|
Loading…
x
Reference in New Issue
Block a user