diff --git a/src/TracingHeap/TracingHeap.c b/src/TracingHeap/TracingHeap.c index bab34e4..5be63b8 100644 --- a/src/TracingHeap/TracingHeap.c +++ b/src/TracingHeap/TracingHeap.c @@ -16,7 +16,6 @@ int TracingHeap_Create(TracingHeap* self, allocator_t* allocator) self->allocator = allocator; - self->objects = NULL; self->black_objects = NULL; self->white_objects = NULL; self->grey_objects = NULL; @@ -29,8 +28,8 @@ int TracingHeap_Create(TracingHeap* self, allocator_t* allocator) static void _TracingHeap_ForceDestroyObject(TracingHeap* self, TracingObject* object) { - self->config.destructor(self->config.destructor_context, object->data); size_t allocated_size = self->config.size_query(self->config.size_query_context, object->data); + self->config.destructor(self->config.destructor_context, object->data); Allocator_Free(self->allocator, object, allocated_size); } @@ -40,7 +39,13 @@ void TracingHeap_Destroy(TracingHeap* self) return; } - for (TracingObject* object = self->objects; object != NULL; object = object->global_next) { + for (TracingObject* object = self->white_objects; object != NULL; object = object->color_next) { + _TracingHeap_ForceDestroyObject(self, object); + } + for (TracingObject* object = self->grey_objects; object != NULL; object = object->color_next) { + _TracingHeap_ForceDestroyObject(self, object); + } + for (TracingObject* object = self->black_objects; object != NULL; object = object->color_next) { _TracingHeap_ForceDestroyObject(self, object); } @@ -153,10 +158,12 @@ int TracingHeap_TraceNext(TracingHeap* self) return EXIT_SUCCESS; } -size_t TracingHeap_TraceNextN(TracingHeap* self, size_t n) +size_t TracingHeap_TraceNextN(TracingHeap* self, size_t n, int* error_code) { for (size_t i = 0; i < n; i++) { - if (TracingHeap_TraceNext(self)) { + int trace_code = TracingHeap_TraceNext(self); + if (trace_code != EXIT_SUCCESS) { + if (error_code != NULL) *error_code = trace_code; return i; } } @@ -166,13 +173,13 @@ size_t TracingHeap_TraceNextN(TracingHeap* self, size_t n) static int _TracingHeap_DestroyObject(TracingHeap* self, TracingObject* unreachable_object) { + size_t allocated_size = self->config.size_query(self->config.size_query_context, unreachable_object->data); + int destructor_code = self->config.destructor(self->config.destructor_context, unreachable_object->data); if (destructor_code) { return destructor_code; } - size_t allocated_size = self->config.size_query(self->config.size_query_context, unreachable_object->data); - Allocator_Free(self->allocator, unreachable_object, allocated_size); return EXIT_SUCCESS; @@ -199,6 +206,11 @@ int TracingHeap_BeginTrace(TracingHeap* self) return EXIT_SUCCESS; } +bool TracingHeap_IsTraceFinished(TracingHeap* self) +{ + return self->grey_objects == NULL; +} + int TracingHeap_EndTrace(TracingHeap* self) { if (self == NULL) { @@ -212,11 +224,11 @@ int TracingHeap_EndTrace(TracingHeap* self) TracingObject* unreachable_object = *unreachable_list; while (unreachable_object != NULL) { - TracingObject* next_unreachable = unreachable_object->global_next; + TracingObject* next_unreachable = unreachable_object->color_next; if (_TracingHeap_DestroyObject(self, unreachable_object)) { *unreachable_list = unreachable_object; - unreachable_object->global_prev = NULL; + unreachable_object->color_prev = NULL; return ECANCELED; } @@ -246,11 +258,5 @@ void* TracingHeap_Allocate(TracingHeap* self, size_t bytes) *reachable_list_start = object; object->color_prev = NULL; - - object->global_next = self->objects; // link before global object list - object->global_prev = NULL; // no previous object - self->objects->global_prev = object; // link first object via previous link - self->objects = object; // insert into linked list - return object->data; } diff --git a/src/TracingHeap/TracingHeap.h b/src/TracingHeap/TracingHeap.h index fbc1c92..cfe4c60 100644 --- a/src/TracingHeap/TracingHeap.h +++ b/src/TracingHeap/TracingHeap.h @@ -4,6 +4,7 @@ #include "../allocator-interface/allocator-interface.h" #include +#include enum TracingColor { TRACINGCOLOR_BLACK, @@ -12,8 +13,6 @@ enum TracingColor { }; typedef struct TracingObject { - struct TracingObject* global_next; - struct TracingObject* global_prev; struct TracingObject* color_next; struct TracingObject* color_prev; enum TracingColor color; @@ -39,7 +38,6 @@ typedef struct TracingHeapConfig_s { } TracingHeapConfig; typedef struct TracingHeap_s { - TracingObject* objects; TracingObject* white_objects; TracingObject* grey_objects; TracingObject* black_objects; @@ -51,15 +49,17 @@ typedef struct TracingHeap_s { allocator_t* allocator; } TracingHeap; -int TracingHeap_Create(TracingHeap* self, allocator_t* allocator); -void TracingHeap_Destroy(TracingHeap* self); +int TracingHeap_Create(TracingHeap* self, allocator_t* allocator); +void TracingHeap_Destroy(TracingHeap* self); -void* TracingHeap_Allocate(TracingHeap* self, size_t bytes); +void* TracingHeap_Allocate(TracingHeap* self, size_t bytes); int TracingHeap_BeginTrace(TracingHeap* self); +bool TracingHeap_IsTraceFinished(TracingHeap* self); int TracingHeap_EndTrace(TracingHeap* self); + int TracingHeap_TraceNext(TracingHeap* self); -size_t TracingHeap_TraceNextN(TracingHeap* self, size_t n); +size_t TracingHeap_TraceNextN(TracingHeap* self, size_t n, int* error_code); int TracingHeap_AddTracingRoot(TracingHeap* self, void* data); #endif // header