hectic-lab/util.nix
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(libhectic): arena allocators

Browse Source
This commit is contained in:
yukkop
2025-03-21 02:54:47 +00:00
parent 4f2391e9b5
commit 9438280750
7 changed files with 78 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
package/c/hmpl/default.nix
View File

@@ -1,4 +1,4 @@
{ stdenv, gcc, lib, libhectic }: { stdenv, gcc, lib, libhectic, cjson }:
stdenv.mkDerivation { stdenv.mkDerivation {
pname = "hmpl"; pname = "hmpl";
@@ -6,13 +6,13 @@ stdenv.mkDerivation {
src = ./.; src = ./.;
doCheck = true; doCheck = true;
buildInputs = [ libhectic ]; buildInputs = [ libhectic cjson ];
buildPhase = '' buildPhase = ''
mkdir -p target mkdir -p target
${gcc}/bin/cc -Wall -Wextra -g \ ${gcc}/bin/cc -Wall -Wextra -g \
-pedantic -fsanitize=address hmpl.c \ -pedantic -fsanitize=address hmpl.c \
-l:libhectic.a -o target/hmpl -l:libhectic.a -l:cjson -o target/hmpl
''; '';
checkPhase = '' ''; checkPhase = '' '';

24
package/c/hmpl/hmpl.c
View File

@@ -1,6 +1,7 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <unistd.h>
//#include "libhmpl.h" //#include "libhmpl.h"
#include "libhectic.h" #include "libhectic.h"
@@ -80,8 +81,25 @@ void render_template(char *text, char *context) {
render_template_placeholders(text, context, ""); render_template_placeholders(text, context, "");
} }
int main(void) { int main(int argc, char *argv[]) {
render_template(text, context); char *text = NULL;
char *context = strdup(argc > 1 ? argv[1] : "{}");
return 0; if (argc > 2) {
text = strdup(argv[2]);
} else if (!isatty(fileno(stdin))) {
size_t size = 0;
ssize_t len = getdelim(&text, &size, '\0', stdin);
if (len < 0) {
perror("read stdin");
free(context);
return 1;
}
}
render_template(text, context);
free(text);
free(context);
return 0;
} }

5
package/c/libhectic/default.nix
View File

@@ -8,7 +8,10 @@ stdenv.mkDerivation {
buildPhase = '' buildPhase = ''
mkdir -p target mkdir -p target
${gcc}/bin/cc -Wall -Wextra -g -pedantic -fsanitize=address -c libhectic.c -o target/libhectic.o ${gcc}/bin/cc -Wall -Wextra -g \
-std=c99 \
-pedantic -fsanitize=address \
-c libhectic.c -o target/libhectic.o
${gcc}/bin/ar rcs target/libhectic.a target/libhectic.o ${gcc}/bin/ar rcs target/libhectic.a target/libhectic.o
''; '';

25
package/c/libhectic/libhectic.c
View File

@@ -52,7 +52,7 @@ void init_logger(void) {
current_log_level = log_level_from_string(getenv("LOG_LEVEL")); current_log_level = log_level_from_string(getenv("LOG_LEVEL"));
} }
char* log_message(LogLevel level, int line, const char *format, ...) { char* log_message(LogLevel level, char *file, int line, const char *format, ...) {
if (level < current_log_level) { if (level < current_log_level) {
return NULL; return NULL;
} }
@@ -63,7 +63,7 @@ char* log_message(LogLevel level, int line, const char *format, ...) {
static char timeStr[20]; static char timeStr[20];
strftime(timeStr, sizeof(timeStr), "%Y-%m-%d %H:%M:%S", &tm_info); strftime(timeStr, sizeof(timeStr), "%Y-%m-%d %H:%M:%S", &tm_info);
fprintf(stderr, "%s %d %s: ", timeStr, line, log_level_to_string(level)); fprintf(stderr, "%s %s %s:%d ", timeStr, log_level_to_string(level), file, line);
va_list args; va_list args;
va_start(args, format); va_start(args, format);
@@ -89,18 +89,6 @@ Arena arena_init(size_t size) {
return arena; return arena;
} }
void *arena_alloc_or_null(Arena *arena, size_t size) {
if (arena->begin == 0) {
*arena = arena_init(ARENA_DEFAULT_SIZE);
}
size_t current = (size_t)arena->current - (size_t)arena->begin;
// TODO(yukkop): maybe -1
if (arena->capacity <= current && current < size) {
return NULL;
}
return arena;
}
void arena_reset(Arena *arena) { void arena_reset(Arena *arena) {
arena->current = arena->begin; arena->current = arena->begin;
} }
@@ -108,12 +96,3 @@ void arena_reset(Arena *arena) {
void arena_free(Arena *arena) { void arena_free(Arena *arena) {
free(arena->begin); free(arena->begin);
} }
void *arena_alloc(Arena *arena, size_t size) {
void *mem = arena_alloc_or_null(arena, size);
if (!mem) {
raise_exception("Arena out of memory");
exit(1);
}
return mem;
}

110
package/c/libhectic/libhectic.h
View File

@@ -14,13 +14,7 @@
// Helper macros for argument counting // Helper macros for argument counting
// NOTE(yukkop): this ugly macroses for avoid all posible warnings // NOTE(yukkop): this ugly macroses for avoid all posible warnings
#define PP_CAT(a, b) PP_CAT_I(a, b) #define PP_CAT(a, b) a##b
#define PP_CAT_I(a, b) a##b
#define PP_NARG(...) PP_NARG_(__VA_ARGS__, PP_RSEQ_N())
#define PP_NARG_(...) PP_ARG_N(__VA_ARGS__)
#define PP_ARG_N(_1,_2,_3,_4,_5,_6,_7,_8,_9,N,...) N
#define PP_RSEQ_N() 9,8,7,6,5,4,3,2,1,0
// ------------ // ------------
// -- Colors -- // -- Colors --
@@ -51,20 +45,13 @@ void set_output_color_mode(ColorMode mode);
// ------------ // ------------
// Define color macros based on output type // Define color macros based on output type
//#define ERROR_PREFIX PP_CAT_I(COLOR_RED, "Error: ") //#define ERROR_PREFIX PP_CAT(COLOR_RED, "Error: ")
//#define ERROR_SUFFIX PP_CAT_I(COLOR_RESET, "\n") //#define ERROR_SUFFIX PP_CAT(COLOR_RESET, "\n")
#define ERROR_PREFIX (USE_COLOR() ? "\033[1;31mError: " : "Error: ") #define ERROR_PREFIX (USE_COLOR() ? "\033[1;31mError: " : "Error: ")
#define ERROR_SUFFIX (USE_COLOR() ? "\033[0m\n" : "\n") #define ERROR_SUFFIX (USE_COLOR() ? "\033[0m\n" : "\n")
// eprintf handling 1 or more arguments // eprintf handling 1 or more arguments
#define eprintf_1(fmt) \ #define eprintf(fmt, ...) "%s" fmt "%s", ERROR_PREFIX, ##__VA_ARGS__, ERROR_SUFFIX
fprintf(stderr, "%s" fmt "%s", ERROR_PREFIX, ERROR_SUFFIX)
#define eprintf_2(fmt, ...) \
fprintf(stderr, "%s" fmt "%s", ERROR_PREFIX, __VA_ARGS__, ERROR_SUFFIX)
#define eprintf(...) \
PP_CAT(eprintf_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
#define todo fprintf(stderr, "%sNot implimented yet%s", COLOR_RED, COLOR_RESET);exit(1) #define todo fprintf(stderr, "%sNot implimented yet%s", COLOR_RED, COLOR_RESET);exit(1)
@@ -87,55 +74,14 @@ void logger_level(LogLevel level);
LogLevel log_level_from_string(const char *level_str); LogLevel log_level_from_string(const char *level_str);
char* log_message(LogLevel level, int line, const char *format, ...); char* log_message(LogLevel level, char *file, int line, const char *format, ...);
// DEBUG level #define raise_debug(fmt, ...) log_message(LOG_LEVEL_DEBUG, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
#define raise_debug_1(fmt) \ #define raise_log(fmt, ...) log_message(LOG_LEVEL_LOG, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
log_message(LOG_LEVEL_DEBUG, __LINE__, fmt) #define raise_info(fmt, ...) log_message(LOG_LEVEL_INFO, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
#define raise_debug_2(fmt, ...) \ #define raise_notice(fmt, ...) log_message(LOG_LEVEL_NOTICE, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
log_message(LOG_LEVEL_DEBUG, __LINE__, fmt, __VA_ARGS__) #define raise_warn(fmt, ...) log_message(LOG_LEVEL_WARN, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
#define raise_debug(...) \ #define raise_exception(fmt, ...) log_message(LOG_LEVEL_EXCEPTION, __FILE__, __LINE__, fmt, ##__VA_ARGS__)
PP_CAT(raise_debug_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
// LOG level
#define raise_log_1(fmt) \
log_message(LOG_LEVEL_LOG, __LINE__, fmt)
#define raise_log_2(fmt, ...) \
log_message(LOG_LEVEL_LOG, __LINE__, fmt, __VA_ARGS__)
#define raise_log(...) \
PP_CAT(raise_log_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
// INFO level
#define raise_info_1(fmt) \
log_message(LOG_LEVEL_INFO, __LINE__, fmt)
#define raise_info_2(fmt, ...) \
log_message(LOG_LEVEL_INFO, __LINE__, fmt, __VA_ARGS__)
#define raise_info(...) \
PP_CAT(raise_info_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
// NOTICE level
#define raise_notice_1(fmt) \
log_message(LOG_LEVEL_NOTICE, __LINE__, fmt)
#define raise_notice_2(fmt, ...) \
log_message(LOG_LEVEL_NOTICE, __LINE__, fmt, __VA_ARGS__)
#define raise_notice(...) \
PP_CAT(raise_notice_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
// WARN level
#define raise_warn_1(fmt) \
log_message(LOG_LEVEL_WARN, __LINE__, fmt)
#define raise_warn_2(fmt, ...) \
log_message(LOG_LEVEL_WARN, __LINE__, fmt, __VA_ARGS__)
#define raise_warn(...) \
PP_CAT(raise_warn_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
// EXCEPTION level
#define raise_exception_1(fmt) \
log_message(LOG_LEVEL_EXCEPTION, __LINE__, fmt)
#define raise_exception_2(fmt, ...) \
log_message(LOG_LEVEL_EXCEPTION, __LINE__, fmt, __VA_ARGS__)
#define raise_exception(...) \
PP_CAT(raise_exception_, PP_NARG(__VA_ARGS__))(__VA_ARGS__)
#endif // EPRINTF_H #endif // EPRINTF_H
@@ -151,15 +97,43 @@ typedef struct {
size_t capacity; size_t capacity;
} Arena; } Arena;
Arena arena_init(size_t size); // NOTE(yukkop): This macro is used to define procedures so that `__LINE__` and `__FILE__`
// in `raise_debug` reflect the location where the macro is called, not where it's defined.
#define arena_alloc_or_null(arena, size) __extension__ ({ \
void *mem__ = NULL; \
if ((arena)->begin == 0) { \
*(arena) = arena_init(ARENA_DEFAULT_SIZE); \
} \
size_t current__ = (size_t)(arena)->current - (size_t)(arena)->begin; \
if ((arena)->capacity <= current__ || (arena)->capacity - current__ < (size)) {\
raise_debug("Arena from %d with capacity %d allocated on %d cannot be allocated on %d", \
(arena)->begin, (arena)->capacity, \
(arena)->current - (arena)->begin, (size)); \
} else { \
raise_debug("Arena from %d with capacity %d allocated on %d will allocate on %d", \
(arena)->begin, (arena)->capacity, \
(arena)->begin, (arena)->capacity, (size)); \
mem__ = (arena)->current; \
(arena)->current = (char*)(arena)->current + (size); \
} \
mem__; \
})
void *arena_alloc_or_null(Arena *arena, size_t size); Arena arena_init(size_t size);
void arena_reset(Arena *arena); void arena_reset(Arena *arena);
void arena_free(Arena *arena); void arena_free(Arena *arena);
void *arena_alloc(Arena *arena, size_t size); #define arena_alloc(arena, size) __extension__ ({ \
void *mem__ = arena_alloc_or_null((arena), (size)); \
if (!mem__) { \
raise_exception("Arena out of memory"); \
exit(1); \
} \
mem__; \
})
// TODO: mmap // TODO: mmap
// TODO: dynamic array style // TODO: dynamic array style

5
package/c/libhectic/test/arena.c
View File

@@ -18,6 +18,7 @@ void test_arena_alloc() {
assert(ptr1 != NULL); assert(ptr1 != NULL);
void *ptr2 = arena_alloc(&arena, 16); void *ptr2 = arena_alloc(&arena, 16);
assert(ptr2 != NULL); assert(ptr2 != NULL);
raise_debug("%d - %d = %d", (size_t)ptr2, (size_t)ptr1, (char *)ptr2 - (char *)ptr1);
assert((char *)ptr2 - (char *)ptr1 == 16); assert((char *)ptr2 - (char *)ptr1 == 16);
arena_free(&arena); arena_free(&arena);
} }
@@ -25,6 +26,7 @@ void test_arena_alloc() {
void test_arena_alloc_or_null_out_of_memory() { void test_arena_alloc_or_null_out_of_memory() {
Arena arena = arena_init(32); Arena arena = arena_init(32);
void *ptr = arena_alloc_or_null(&arena, 64); void *ptr = arena_alloc_or_null(&arena, 64);
raise_debug("%d %d %d %d", arena.begin, arena.current, arena.capacity, (size_t)ptr);
assert(ptr == NULL); assert(ptr == NULL);
arena_free(&arena); arena_free(&arena);
} }
@@ -46,6 +48,9 @@ void test_arena_null_init() {
} }
int main() { int main() {
set_output_color_mode(COLOR_MODE_DISABLE);
logger_level(LOG_LEVEL_DEBUG); \
test_arena_init(); test_arena_init();
test_arena_alloc(); test_arena_alloc();
test_arena_alloc_or_null_out_of_memory(); test_arena_alloc_or_null_out_of_memory();

2
package/c/libhectic/test/test.c
View File

@@ -20,7 +20,7 @@
stderr = orig_stderr; \ stderr = orig_stderr; \
fclose(temp); \ fclose(temp); \
char expected_buffer[256]; \ char expected_buffer[256]; \
sprintf(expected_buffer, "%s %d " LEVEL_STR ": message\n", time_str, __LINE__); \ sprintf(expected_buffer, "%s " LEVEL_STR " " __FILE__ ":%d message\n", time_str, __LINE__); \
printf("DEBUG: [%s] [%s]\n", result_buffer, expected_buffer); \ printf("DEBUG: [%s] [%s]\n", result_buffer, expected_buffer); \
assert(strcmp(result_buffer, expected_buffer) == 0); \ assert(strcmp(result_buffer, expected_buffer) == 0); \
} while(0) } while(0)