108 lines
2.2 KiB
C
108 lines
2.2 KiB
C
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include <string.h>
|
|
|
|
#define ARENA_ALIGN 8
|
|
|
|
typedef struct{
|
|
unsigned char *buffer;
|
|
size_t size;
|
|
size_t offset;
|
|
}Arena;
|
|
|
|
Arena arena_create(size_t);
|
|
void* arena_alloc(Arena *, size_t);
|
|
void arena_reset(Arena *);
|
|
void arena_destroy(Arena *);
|
|
bool arena_is_aligned(size_t);
|
|
char* arena_strdup(Arena*, const char *);
|
|
|
|
int main(void){
|
|
Arena arena = arena_create(16);
|
|
|
|
char* s1 = arena_strdup(&arena, "arena");
|
|
char* s2 = arena_strdup(&arena, "hola");
|
|
char* s3 = arena_strdup(&arena, "x");
|
|
|
|
arena_destroy(&arena);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
char* arena_strdup(Arena* ref_arena, const char *src){
|
|
char* ref_str = (char*) arena_alloc(ref_arena, strlen(src) + 1);
|
|
if(ref_str == NULL){
|
|
perror("No se ha podido reservar memoria para la cadena.");
|
|
return NULL;
|
|
}
|
|
strcpy(ref_str, src);
|
|
return ref_str;
|
|
}
|
|
|
|
bool arena_is_aligned(size_t pos){
|
|
size_t mask = 7;
|
|
return (pos & mask) == 0;
|
|
}
|
|
|
|
Arena arena_create( size_t bytes){
|
|
Arena new_arena = {
|
|
.buffer = NULL,
|
|
.size = bytes,
|
|
.offset = 0
|
|
};
|
|
|
|
new_arena.buffer = (unsigned char *)malloc(bytes);
|
|
|
|
if(new_arena.buffer == NULL){
|
|
perror("Error asignando memoria");
|
|
exit(1);
|
|
}
|
|
|
|
return new_arena;
|
|
|
|
}
|
|
void* arena_alloc(Arena *ref_arena, size_t bytes){
|
|
|
|
if(ref_arena->buffer == NULL){
|
|
return NULL;
|
|
}
|
|
|
|
size_t aligned = ((ref_arena->offset + (size_t)((ARENA_ALIGN - 1))) & ~(ARENA_ALIGN - 1));
|
|
|
|
if(aligned + bytes > ref_arena->size){
|
|
printf("arena_alloc(%d): NULL \n", (int) bytes);
|
|
return NULL;
|
|
}
|
|
void* ptr_ref = ref_arena->buffer + aligned;
|
|
|
|
printf("arena_alloc(%d) ", (int) bytes);
|
|
|
|
if(arena_is_aligned((size_t)ptr_ref)){
|
|
printf("aligned: true ");
|
|
}else{
|
|
printf("aligned: false ");
|
|
}
|
|
|
|
printf(" base + %d", (int)aligned);
|
|
printf(" offset: %d -> %d \n", (int)ref_arena->offset, (int)aligned);
|
|
|
|
ref_arena->offset = aligned + bytes;
|
|
|
|
return ptr_ref;
|
|
}
|
|
|
|
void arena_reset(Arena *ref_arena){
|
|
ref_arena->offset = 0;
|
|
}
|
|
|
|
void arena_destroy(Arena *ref_arena){
|
|
if(ref_arena->buffer != NULL){
|
|
free(ref_arena->buffer);
|
|
ref_arena->buffer = NULL;
|
|
}
|
|
}
|
|
|