koliseo.c

Go to the documentation of this file.

// jgabaut @ github.com/jgabaut
// SPDX-License-Identifier: GPL-3.0-only
/*
    Copyright (C) 2023-2026  jgabaut
 
    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, version 3 of the License.
 
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
 
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
#ifndef _WIN32
#ifndef _POSIX_C_SOURCE
#define _POSIX_C_SOURCE 200809L
#endif
#endif
#include "koliseo.h"
 
KLS_Conf KLS_DEFAULT_CONF = {
    .collect_stats = 0,
    .verbose_lvl = 0,
    .block_while_has_temp = 1,
    .allow_zerocount_push = 0,
    .log_fp = NULL,
    .growable = 0,
    .log_filepath = "",
    .err_handlers = {
#ifndef KOLISEO_HAS_LOCATE
        .OOM_handler = &KLS_OOM_default_handler__,
        .PTRDIFF_MAX_handler = &KLS_PTRDIFF_MAX_default_handler__,
        .ZEROCOUNT_handler = &KLS_ZEROCOUNT_default_handler__,
#else
        .OOM_handler = &KLS_OOM_default_handler_dbg__,
        .PTRDIFF_MAX_handler = &KLS_PTRDIFF_MAX_default_handler_dbg__,
        .ZEROCOUNT_handler = &KLS_ZEROCOUNT_default_handler_dbg__,
#endif // KOLISEO_HAS_LOCATE
    },
}; 
 
KLS_Stats KLS_STATS_DEFAULT = {
    .tot_pushes = 0,
    .tot_temp_pushes = 0,
    .tot_pops = 0,
    .tot_temp_pops = 0,
    .tot_logcalls = 0,
#ifdef KLS_DEBUG_CORE
    .worst_pushcall_time = -1,
#endif
};
 
static bool kls_set_conf(Koliseo * kls, KLS_Conf conf); //Declare function used internally by kls_new() and kls_new_conf()

const char *string_koliseo_version(void)
{
    return KOLISEO_API_VERSION_STRING;
}

int int_koliseo_version(void)
{
    return KOLISEO_API_VERSION_INT;
}
 
#if defined(__SANITIZE_ADDRESS__)
#include <sanitizer/asan_interface.h>
#define KLS_ASAN_POISON(addr, size)   __asan_poison_memory_region((addr), (size))
#define KLS_ASAN_UNPOISON(addr, size) __asan_unpoison_memory_region((addr), (size))
#else
#define KLS_ASAN_POISON(addr, size)   ((void)0)
#define KLS_ASAN_UNPOISON(addr, size) ((void)0)
#endif // __SANITIZE_ADDRESS__

#ifndef KOLISEO_HAS_LOCATE
void KLS_OOM_default_handler__(Koliseo* kls, ptrdiff_t available, ptrdiff_t padding, ptrdiff_t size, ptrdiff_t count)
#else
void KLS_OOM_default_handler_dbg__(Koliseo* kls, ptrdiff_t available, ptrdiff_t padding, ptrdiff_t size, ptrdiff_t count, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
#ifndef KOLISEO_HAS_LOCATE
    fprintf(stderr,
            "[KLS]  Out of memory. size*count [%td] was bigger than available-padding [%td].\n",
            size * count, available - padding);
#else
    fprintf(stderr,
            "[KLS] " KLS_Loc_Fmt "Out of memory. size*count [%td] was bigger than available-padding [%td].\n",
            KLS_Loc_Arg(loc),
            size * count, available - padding);
#endif // KOLISEO_HAS_LOCATE
    kls_free(kls); // Is it even worth it to try?
    exit(EXIT_FAILURE); // Better than nothing. May change to return NULL instead? Requiring refactor of handler signature
}
 
#ifndef KOLISEO_HAS_LOCATE
void KLS_PTRDIFF_MAX_default_handler__(struct Koliseo* kls, ptrdiff_t size, ptrdiff_t count)
#else
void KLS_PTRDIFF_MAX_default_handler_dbg__(struct Koliseo* kls, ptrdiff_t size, ptrdiff_t count, Koliseo_Loc loc)
#endif
{
#ifndef _WIN32
#ifndef KOLISEO_HAS_LOCATE
    fprintf(stderr,
            "[KLS]  count [%td] was bigger than PTRDIFF_MAX/size [%li].\n",
            count, PTRDIFF_MAX / size);
#else
    fprintf(stderr,
            "[KLS] " KLS_Loc_Fmt "count [%td] was bigger than PTRDIFF_MAX/size [%li].\n",
            KLS_Loc_Arg(loc),
            count, PTRDIFF_MAX / size);
#endif // KOLISEO_HAS_LOCATE
#else
#ifndef KOLISEO_HAS_LOCATE
    fprintf(stderr,
            "[KLS]  count [%td] was bigger than PTRDIFF_MAX/size [%lli].\n",
            count, PTRDIFF_MAX / size);
#else
    fprintf(stderr,
            "[KLS] " KLS_Loc_Fmt "count [%td] was bigger than PTRDIFF_MAX/size [%lli].\n",
            KLS_Loc_Arg(loc),
            count, PTRDIFF_MAX / size);
#endif // KOLISEO_HAS_LOCATE
#endif // _WIN32
    kls_free(kls);
    exit(EXIT_FAILURE);
}

#ifndef KOLISEO_HAS_LOCATE
void KLS_ZEROCOUNT_default_handler__(Koliseo* kls, ptrdiff_t available, ptrdiff_t padding, ptrdiff_t size)
#else
void KLS_ZEROCOUNT_default_handler_dbg__(Koliseo* kls, ptrdiff_t available, ptrdiff_t padding, ptrdiff_t size, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
#ifndef KOLISEO_HAS_LOCATE
    fprintf(stderr,
            "[KLS]  Doing a zero-count push. size [%td] padding [%td] available [%td].\n",
            size, padding, available);
#else
    fprintf(stderr,
            "[KLS] " KLS_Loc_Fmt "Doing a zero-count push. size [%td] padding [%td] available [%td].\n",
            KLS_Loc_Arg(loc),
            size, padding, available);
#endif // KOLISEO_HAS_LOCATE
    kls_free(kls);
    exit(EXIT_FAILURE);
}

KLS_Conf kls_conf_init_handled(int collect_stats, int verbose_lvl, int block_while_has_temp, int allow_zerocount_push, int growable, FILE* log_fp, const char* log_filepath, KLS_Err_Handlers err_handlers)
{
    KLS_Conf res = {0};
    res.collect_stats = collect_stats;
    res.verbose_lvl = verbose_lvl;
    res.block_while_has_temp = block_while_has_temp;
    res.allow_zerocount_push = allow_zerocount_push;
    res.growable = growable;
    res.log_fp = log_fp;
    res.log_filepath = log_filepath;
 
    if (err_handlers.OOM_handler != NULL) {
        res.err_handlers.OOM_handler = err_handlers.OOM_handler;
    } else {
#ifndef KOLISEO_HAS_LOCATE
        res.err_handlers.OOM_handler = &KLS_OOM_default_handler__;
#else
        res.err_handlers.OOM_handler = &KLS_OOM_default_handler_dbg__;
#endif // KOLISEO_HAS_LOCATE
    }
 
    if (err_handlers.PTRDIFF_MAX_handler != NULL) {
        res.err_handlers.PTRDIFF_MAX_handler = err_handlers.PTRDIFF_MAX_handler;
    } else {
#ifndef KOLISEO_HAS_LOCATE
        res.err_handlers.PTRDIFF_MAX_handler = &KLS_PTRDIFF_MAX_default_handler__;
#else
        res.err_handlers.PTRDIFF_MAX_handler = &KLS_PTRDIFF_MAX_default_handler_dbg__;
#endif // KOLISEO_HAS_LOCATE
    }
 
    if (err_handlers.ZEROCOUNT_handler != NULL) {
        res.err_handlers.ZEROCOUNT_handler = err_handlers.ZEROCOUNT_handler;
    } else {
#ifndef KOLISEO_HAS_LOCATE
        res.err_handlers.ZEROCOUNT_handler = &KLS_ZEROCOUNT_default_handler__;
#else
        res.err_handlers.ZEROCOUNT_handler = &KLS_ZEROCOUNT_default_handler_dbg__;
#endif // KOLISEO_HAS_LOCATE
    }
 
    return res;
}

KLS_Conf kls_conf_init(int collect_stats, int verbose_lvl, int block_while_has_temp, int allow_zerocount_push, int growable, FILE* log_fp, const char* log_filepath)
{
    KLS_Err_Handlers err_handlers = KLS_DEFAULT_ERR_HANDLERS;
    return kls_conf_init_handled(collect_stats, verbose_lvl, block_while_has_temp, allow_zerocount_push, growable, log_fp, log_filepath, err_handlers);
}

void kls_dbg_features(void)
{
#ifdef KOLISEO_HAS_LOCATE
    bool kls_locate = true;
#else
    bool kls_locate = false;
#endif
#ifdef KLS_DEBUG_CORE
    bool kls_debug = true;
#else
    bool kls_debug = false;
#endif
#ifdef KOLISEO_HAS_EXPER
    bool kls_exper = true;
#else
    bool kls_exper = false;
#endif
    bool features[3] = {
        [0] = kls_debug,
        [1] = kls_locate,
        [2] = kls_exper,
    };
    int total_enabled = 0;
    for (int i=0; i<3; i++) {
        if (features[i]) {
            total_enabled += 1;
        }
    }
    fprintf(stderr, "[KLS]    Enabled features: {");
    if (total_enabled == 0) {
        fprintf(stderr, "none}\n");
        return;
    } else {
        if (kls_debug) {
            fprintf(stderr, "debug%s", (total_enabled > 1 ? ", " : ""));
            total_enabled -= 1;
        }
        if (kls_locate) {
            fprintf(stderr, "locate%s", (total_enabled > 1 ? ", " : ""));
            total_enabled -= 1;
        }
        if (kls_exper) {
            fprintf(stderr, "exper");
        }
        fprintf(stderr, "}\n");
    }
}

ptrdiff_t kls_get_pos(const Koliseo *kls)
{
    return kls->offset;
}

#ifndef KOLISEO_HAS_LOCATE
void* kls__handle_push_result(Koliseo* kls, KLS_Push_Result r, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, ptrdiff_t padding, const char* caller_name)
#else
void* kls__handle_push_result_dbg(Koliseo* kls, KLS_Push_Result r, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, ptrdiff_t padding, const char* caller_name, Koliseo_Loc loc)
#endif
{
    if (!kls) return NULL;
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
    if (r.error) {
        const ptrdiff_t available = current->size - current->offset;
        switch (r.error) {
        case KLS_PUSH_OK: {
            return r.p;
        }
        break;
        case KLS_PUSH_SIZE_LT1: {
#ifndef KOLISEO_HAS_LOCATE
            fprintf(stderr,
                    "[KLS] %s(): size [%td] was < 1.\n",
                    caller_name,
                    size);
#else
            fprintf(stderr,
                    "[KLS] " KLS_Loc_Fmt "%s(): size [%td] was < 1.\n",
                    KLS_Loc_Arg(loc),
                    caller_name,
                    size);
#endif // KOLISEO_HAS_LOCATE
        }
        break;
        case KLS_PUSH_ALIGN_LT1: {
#ifndef KOLISEO_HAS_LOCATE
            fprintf(stderr,
                    "[KLS] %s(): align [%td] was < 1.\n",
                    caller_name,
                    align);
#else
            fprintf(stderr,
                    "[KLS] " KLS_Loc_Fmt "%s(): align [%td] was < 1.\n",
                    KLS_Loc_Arg(loc),
                    caller_name,
                    align);
#endif // KOLISEO_HAS_LOCATE
        }
        break;
        case KLS_PUSH_ALIGN_NOT_POW2: {
#ifndef KOLISEO_HAS_LOCATE
            fprintf(stderr,
                    "[KLS] %s(): align [%td] was not a power of 2.\n",
                    caller_name,
                    align);
#else
            fprintf(stderr,
                    "[KLS] " KLS_Loc_Fmt "%s(): align [%td] was not a power of 2.\n",
                    KLS_Loc_Arg(loc),
                    caller_name,
                    align);
#endif // KOLISEO_HAS_LOCATE
        }
        break;
        case KLS_PUSH_NEGATIVE_COUNT: {
#ifndef KOLISEO_HAS_LOCATE
            fprintf(stderr,
                    "[KLS] %s(): count [%td] was < 0.\n",
                    caller_name,
                    count);
#else
            fprintf(stderr,
                    "[KLS] " KLS_Loc_Fmt "%s(): count [%td] was < 0.\n",
                    KLS_Loc_Arg(loc),
                    caller_name,
                    count);
#endif // KOLISEO_HAS_LOCATE
        }
        break;
        case KLS_PUSH_ZEROCOUNT: {
            if (current->conf.err_handlers.ZEROCOUNT_handler != NULL) {
#ifndef KOLISEO_HAS_LOCATE
                current->conf.err_handlers.ZEROCOUNT_handler(current, available, padding, size);
#else
                current->conf.err_handlers.ZEROCOUNT_handler(current, available, padding, size, loc);
#endif // KOLISEO_HAS_LOCATE
            } else {
#ifndef KOLISEO_HAS_LOCATE
                fprintf(stderr,
                        "[KLS] %s(): Doing a zero-count push. size [%td] padding [%td] available [%td].\n",
                        caller_name,
                        size, padding, available);
#else
                fprintf(stderr,
                        "[KLS] " KLS_Loc_Fmt "%s(): Doing a zero-count push. size [%td] padding [%td] available [%td].\n",
                        KLS_Loc_Arg(loc),
                        caller_name,
                        size, padding, available);
#endif // KOLISEO_HAS_LOCATE
                kls_free(kls);
                exit(EXIT_FAILURE);
            }
        }
        break;
        case KLS_PUSH_WITH_TEMP_ACTIVE: {
#ifndef KOLISEO_HAS_LOCATE
            fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo has an open Koliseo_Temp session.\n", caller_name);
#else
            fprintf(stderr, "[ERROR] " KLS_Loc_Fmt "[%s()]: Passed Koliseo has an open Koliseo_Temp session.\n", KLS_Loc_Arg(loc), caller_name);
            kls_log(kls, "ERROR", KLS_Loc_Fmt "[%s()]: Passed Koliseo has an open Koliseo_Temp session.", KLS_Loc_Arg(loc), caller_name);
            exit(EXIT_FAILURE);
#endif // KOLISEO_HAS_LOCATE
        }
        break;
        case KLS_PUSH_PTRDIFF_MAX: {
            if (current->conf.err_handlers.PTRDIFF_MAX_handler != NULL) {
#ifndef KOLISEO_HAS_LOCATE
                current->conf.err_handlers.PTRDIFF_MAX_handler(current, size, count);
#else
                current->conf.err_handlers.PTRDIFF_MAX_handler(current, size, count, loc);
#endif // KOLISEO_HAS_LOCATE
            } else { // Let's keep this here for now? It's the original part before adding KLS_PTRDIFF_MAX_default_handler__()
#ifndef _WIN32
#ifndef KOLISEO_HAS_LOCATE
                fprintf(stderr,
                        "[KLS] %s(): count [%td] was bigger than PTRDIFF_MAX/size [%li].\n",
                        caller_name,
                        count, PTRDIFF_MAX / size);
#else
                fprintf(stderr,
                        "[KLS] " KLS_Loc_Fmt "%s(): count [%td] was bigger than PTRDIFF_MAX/size [%li].\n",
                        KLS_Loc_Arg(loc),
                        caller_name,
                        count, PTRDIFF_MAX / size);
#endif // KOLISEO_HAS_LOCATE
#else
#ifndef KOLISEO_HAS_LOCATE
                fprintf(stderr,
                        "[KLS] %s(): count [%td] was bigger than PTRDIFF_MAX/size [%lli].\n",
                        caller_name,
                        count, PTRDIFF_MAX / size);
#else
                fprintf(stderr,
                        "[KLS] " KLS_Loc_Fmt "%s(): count [%td] was bigger than PTRDIFF_MAX/size [%lli].\n",
                        KLS_Loc_Arg(loc),
                        caller_name,
                        count, PTRDIFF_MAX / size);
#endif // KOLISEO_HAS_LOCATE
#endif // _WIN32
            }
        }
        break;
        case KLS_PUSH_OOM: {
            if (current->conf.err_handlers.OOM_handler != NULL) {
#ifndef KOLISEO_HAS_LOCATE
                current->conf.err_handlers.OOM_handler(current, available, padding, size, count);
#else
                current->conf.err_handlers.OOM_handler(current, available, padding, size, count, loc);
#endif // KOLISEO_HAS_LOCATE
            } else { // Let's keep this here for now? It's the original part before adding KLS_OOM_default_handler__()
#ifndef KOLISEO_HAS_LOCATE
                fprintf(stderr,
                        "[KLS] %s(): Out of memory. size*count [%td] was bigger than available-padding [%td].\n",
                        caller_name,
                        size * count, available - padding);
#else
                fprintf(stderr,
                        "[KLS] " KLS_Loc_Fmt "%s(): Out of memory. size*count [%td] was bigger than available-padding [%td].\n",
                        KLS_Loc_Arg(loc),
                        caller_name,
                        size * count, available - padding);
#endif // KOLISEO_HAS_LOCATE
            }
        }
        break;
        default: {
            fprintf(stderr, "[KLS] %s(): unexpected error code [%i]\n", __func__, r.error);
            return NULL;
        }
        break;
        }
    }
    return r.p;
}

void kls_log(Koliseo *kls, const char *tag, const char *format, ...)
{
    if (kls == NULL) {
        fprintf(stderr, "[KLS]    %s(): Passed kls was NULL.\n", __func__);
        return;
    }
    if (kls->conf.verbose_lvl > 0) {
        va_list args;
        FILE *fp = kls->conf.log_fp;
        va_start(args, format);
        if (fp == NULL) {
            fprintf(stderr,
                    "[KLS]    %s(): Failed opening file to print logs.\n",
                    __func__);
        } else {
            time_t now = time(0);
            const struct tm *mytime = localtime(&now);
            char timeheader[500];
            if (strftime(timeheader, sizeof timeheader, "%X", mytime)) {
                fprintf(fp, "[%-10.10s] [%s] [", tag, timeheader);
                vfprintf(fp, format, args);
                fprintf(fp, "]\n");
            }
        }
        va_end(args);
    }
}

#ifndef KOLISEO_HAS_LOCATE
Koliseo *kls_new_alloc_ext(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Hooks ext_handlers, void* user)
#else
Koliseo *kls_new_alloc_ext_dbg(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Hooks ext_handlers, void* user, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
    if (size < (ptrdiff_t)sizeof(Koliseo)) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[ERROR]    at %s():  invalid requested kls size (%td). Min accepted is: (%td).\n",
                __func__, size, (ptrdiff_t)sizeof(Koliseo));
#else
        fprintf(stderr,
                "[ERROR] " KLS_Loc_Fmt "%s():  invalid requested kls size (%td). Min accepted is: (%td).\n",
                KLS_Loc_Arg(loc), __func__, size, (ptrdiff_t)sizeof(Koliseo));
#endif // KOLISEO_HAS_LOCATE
        //TODO Is it better to abort the program?
        return NULL;
    }
    void *p = alloc_func(size);
    if (p) {
        //sprintf(msg,"Allocated (%li) for new KLS.",size);
        //kls_log("KLS",msg);
        char h_size[200];
        kls_formatSize(size, h_size, sizeof(h_size));
        Koliseo *kls = p;
        kls->data = p;
        kls->size = size;
        kls->offset = sizeof(*kls);
        kls->prev_offset = kls->offset;
        kls->has_temp = 0;
        kls->t_kls = NULL;
        kls_set_conf(kls, KLS_DEFAULT_CONF);
        kls->stats = KLS_STATS_DEFAULT;
        kls->conf.log_fp = stderr;
 
        KLS_ASAN_POISON(kls->data + kls->offset, kls->size - kls->offset);
 
        kls->hooks = ext_handlers;
        if (user) {
            kls->extension_data = user;
        } else {
            kls->extension_data = NULL;
        }
        kls->free_func = free_func;
        kls->next = NULL;
#ifdef KLS_DEBUG_CORE
        kls_log(kls, "KLS", "API Level { %i } ->  Allocated (%s) for new KLS.",
                int_koliseo_version(), h_size);
        kls_log(kls, "KLS", "KLS offset: { %p }.", kls);
        kls_log(kls, "KLS", "Allocation begin offset: { %p }.",
                kls + kls->offset);
#endif
 
        if (kls->hooks.on_new_handler != NULL) {
            // Call on_new extension
            kls->hooks.on_new_handler(kls);
        }
    } else {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr, "[KLS] Failed %s() call.\n", __func__);
#else
        fprintf(stderr, "[KLS] " KLS_Loc_Fmt "Failed %s() call.\n", KLS_Loc_Arg(loc), __func__);
#endif // KOLISEO_HAS_LOCATE
        exit(EXIT_FAILURE);
    }
#ifdef KLS_DEBUG_CORE
    Koliseo *kls_ref = p;
    if (kls_ref->conf.verbose_lvl > 0) {
        print_kls_2file(kls_ref->conf.log_fp, p);
    }
#endif
    return p;
}

#ifndef KOLISEO_HAS_LOCATE
Koliseo *kls_new_alloc(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func)
#else
Koliseo *kls_new_alloc_dbg(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
#ifndef KOLISEO_HAS_LOCATE
    return kls_new_alloc_ext(size, alloc_func, free_func, KLS_DEFAULT_HOOKS, KLS_DEFAULT_EXTENSION_DATA);
#else
    return kls_new_alloc_ext_dbg(size, alloc_func, free_func, KLS_DEFAULT_HOOKS, KLS_DEFAULT_EXTENSION_DATA, loc);
#endif // KOLISEO_HAS_LOCATE
}

Koliseo *kls_new(ptrdiff_t size)
{
    return kls_new_alloc(size, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF);
}

Koliseo *kls_new_conf_alloc_ext(ptrdiff_t size, KLS_Conf conf, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Hooks ext_handlers, void* user)
{
    Koliseo *k = kls_new_alloc_ext(size, alloc_func, free_func, ext_handlers, user);
    bool conf_res = kls_set_conf(k, conf);
    if (!conf_res) {
        fprintf(stderr,
                "[ERROR] [%s()]: Failed to set config for new Koliseo.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
    return k;
}

Koliseo *kls_new_conf_alloc(ptrdiff_t size, KLS_Conf conf, kls_alloc_func alloc_func, kls_free_func free_func)
{
    return kls_new_conf_alloc_ext(size, conf, alloc_func, free_func, KLS_DEFAULT_HOOKS, KLS_DEFAULT_EXTENSION_DATA);
}

Koliseo *kls_new_conf_ext(ptrdiff_t size, KLS_Conf conf, KLS_Hooks ext_handlers, void* user)
{
    return kls_new_conf_alloc_ext(size, conf, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF, ext_handlers, user);
}

Koliseo *kls_new_conf(ptrdiff_t size, KLS_Conf conf)
{
    return kls_new_conf_alloc(size, conf, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF);
}

Koliseo *kls_new_traced_alloc_handled_ext(ptrdiff_t size, const char *output_path, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Err_Handlers err_handlers, KLS_Hooks ext_handlers, void* user)
{
 
#ifndef KLS_DEBUG_CORE
    fprintf(stderr,
            "[WARN]    %s(): KLS_DEBUG_CORE is not defined. No tracing allowed.\n",
            __func__);
#endif
    KLS_Conf k = (KLS_Conf) {
        .collect_stats = 1,.verbose_lvl =
                             1,.log_filepath = output_path,
#ifndef KOLISEO_HAS_LOCATE
                               .err_handlers.OOM_handler = (err_handlers.OOM_handler != NULL ? err_handlers.OOM_handler : &KLS_OOM_default_handler__),
                               .err_handlers.PTRDIFF_MAX_handler = ( err_handlers.PTRDIFF_MAX_handler != NULL ? err_handlers.PTRDIFF_MAX_handler : &KLS_PTRDIFF_MAX_default_handler__),
#else
                               .err_handlers.OOM_handler = (err_handlers.OOM_handler != NULL ? err_handlers.OOM_handler : &KLS_OOM_default_handler_dbg__),
                               .err_handlers.PTRDIFF_MAX_handler = ( err_handlers.PTRDIFF_MAX_handler != NULL ? err_handlers.PTRDIFF_MAX_handler : &KLS_PTRDIFF_MAX_default_handler_dbg__),
#endif // KOLISEO_HAS_LOCATE
    };
    return kls_new_conf_alloc_ext(size, k, alloc_func, free_func, ext_handlers, user);
}

Koliseo *kls_new_traced_alloc_handled(ptrdiff_t size, const char *output_path, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Err_Handlers err_handlers)
{
    return kls_new_traced_alloc_handled_ext(size, output_path, alloc_func, free_func, err_handlers, KLS_DEFAULT_HOOKS, KLS_DEFAULT_EXTENSION_DATA);
}

Koliseo *kls_new_traced_ext(ptrdiff_t size, const char *output_path, KLS_Hooks ext_handlers, void* user)
{
    KLS_Err_Handlers err_handlers = KLS_DEFAULT_ERR_HANDLERS;
    return kls_new_traced_alloc_handled_ext(size, output_path, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF, err_handlers, ext_handlers, user);
}

Koliseo *kls_new_traced_alloc(ptrdiff_t size, const char *output_path, kls_alloc_func alloc_func, kls_free_func free_func)
{
    KLS_Err_Handlers err_handlers = KLS_DEFAULT_ERR_HANDLERS;
    return kls_new_traced_alloc_handled(size, output_path, alloc_func, free_func, err_handlers);
}

Koliseo *kls_new_traced(ptrdiff_t size, const char *output_path)
{
    return kls_new_traced_alloc(size, output_path, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF);
}

Koliseo *kls_new_traced_handled(ptrdiff_t size, const char *output_path, KLS_Err_Handlers err_handlers)
{
    return kls_new_traced_alloc_handled(size, output_path, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF, err_handlers);
}

Koliseo *kls_new_dbg_alloc_handled_ext(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Err_Handlers err_handlers, KLS_Hooks ext_handlers, void* user)
{
#ifndef KLS_DEBUG_CORE
    fprintf(stderr,
            "[WARN]    %s(): KLS_DEBUG_CORE is not defined. No debugging support.\n",
            __func__);
#endif
    KLS_Conf k = (KLS_Conf) {
        .collect_stats = 1,.verbose_lvl = 0,
#ifndef KOLISEO_HAS_LOCATE
        .err_handlers.OOM_handler = ( err_handlers.OOM_handler != NULL ? err_handlers.OOM_handler : &KLS_OOM_default_handler__),
        .err_handlers.PTRDIFF_MAX_handler = ( err_handlers.PTRDIFF_MAX_handler != NULL ? err_handlers.PTRDIFF_MAX_handler : &KLS_PTRDIFF_MAX_default_handler__),
#else
        .err_handlers.OOM_handler = ( err_handlers.OOM_handler != NULL ? err_handlers.OOM_handler : &KLS_OOM_default_handler_dbg__),
        .err_handlers.PTRDIFF_MAX_handler = ( err_handlers.PTRDIFF_MAX_handler != NULL ? err_handlers.PTRDIFF_MAX_handler : &KLS_PTRDIFF_MAX_default_handler_dbg__),
#endif // KOLIEO_HAS_LOCATE
    };
    Koliseo * kls = kls_new_conf_alloc_ext(size, k, alloc_func, free_func, ext_handlers, user);
    kls->conf.verbose_lvl = 1;
    return kls;
}

Koliseo *kls_new_dbg_alloc_handled(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func, KLS_Err_Handlers err_handlers)
{
    return kls_new_dbg_alloc_handled_ext(size, alloc_func, free_func, err_handlers, KLS_DEFAULT_HOOKS, KLS_DEFAULT_EXTENSION_DATA);
}

Koliseo *kls_new_dbg_ext(ptrdiff_t size, KLS_Hooks ext_handlers, void* user)
{
    KLS_Err_Handlers err_handlers = KLS_DEFAULT_ERR_HANDLERS;
    return kls_new_dbg_alloc_handled_ext(size, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF, err_handlers, ext_handlers, user);
}

Koliseo *kls_new_dbg_alloc(ptrdiff_t size, kls_alloc_func alloc_func, kls_free_func free_func)
{
    KLS_Err_Handlers err_handlers = KLS_DEFAULT_ERR_HANDLERS;
    return kls_new_dbg_alloc_handled(size, alloc_func, free_func, err_handlers);
}

Koliseo *kls_new_dbg(ptrdiff_t size)
{
    return kls_new_dbg_alloc(size, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF);
}

Koliseo *kls_new_dbg_handled(ptrdiff_t size, KLS_Err_Handlers err_handlers)
{
    return kls_new_dbg_alloc_handled(size, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF, err_handlers);
}

bool kls_set_conf(Koliseo *kls, KLS_Conf conf)
{
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", __func__);
        //TODO: is it better to exit() here?
        return false;
    }
 
    kls->conf = conf;
    if (kls->conf.log_fp == NULL) {
        kls->conf.log_fp = stderr;
#ifdef KLS_DEBUG_CORE
#ifdef KLS_SETCONF_DEBUG
        kls_log(kls, "KLS",
                "[%s()]:  Preliminary set of conf.log_fp to stderr.",
                __func__);
#endif
#endif // KLS_DEBUG_CORE
    }
 
    if (conf.err_handlers.OOM_handler == NULL) {
        fprintf(stderr,
                "[ERROR]    at %s():  passed OOM_handler is NULL. Using default.\n",
                __func__);
#ifdef KLS_DEBUG_CORE
#ifdef KLS_SETCONF_DEBUG
        kls_log(kls, "KLS",
                "[%s()]:  Passed OOM_handler was NULL, using default.",
                __func__);
#endif
#endif // KLS_DEBUG_CORE
#ifndef KOLISEO_HAS_LOCATE
        kls->conf.err_handlers.OOM_handler = &KLS_OOM_default_handler__;
#else
        kls->conf.err_handlers.OOM_handler = &KLS_OOM_default_handler_dbg__;
#endif
    }
 
    if (conf.err_handlers.PTRDIFF_MAX_handler == NULL) {
        fprintf(stderr,
                "[ERROR]    at %s():  passed PTRDIFF_MAX_handler is NULL. Using default.\n",
                __func__);
#ifdef KLS_DEBUG_CORE
#ifdef KLS_SETCONF_DEBUG
        kls_log(kls, "KLS",
                "[%s()]:  Passed PTRDIFF_MAX_handler was NULL, using default.",
                __func__);
#endif
#endif // KLS_DEBUG_CORE
#ifndef KOLISEO_HAS_LOCATE
        kls->conf.err_handlers.PTRDIFF_MAX_handler = &KLS_PTRDIFF_MAX_default_handler__;
#else
        kls->conf.err_handlers.PTRDIFF_MAX_handler = &KLS_PTRDIFF_MAX_default_handler_dbg__;
#endif
    }
 
#ifndef KLS_DEBUG_CORE
    if (kls->conf.collect_stats == 1) {
        fprintf(stderr,
                "[WARN]    [%s()]: KLS_DEBUG_CORE is not defined. Stats may not be collected in full.\n",
                __func__);
    }
#endif
 
    if (kls->conf.verbose_lvl > 0) {
        if (kls->conf.log_fp != NULL) {
#ifdef KLS_DEBUG_CORE
#ifdef KLS_SETCONF_DEBUG
            kls_log(kls, "WARN",
                    "[%s()]: kls->conf.log_fp was not NULL. Overriding it.",
                    __func__);
#endif
#endif
        }
 
        FILE *log_fp = NULL;
        log_fp = fopen(kls->conf.log_filepath, "w");
        if (!log_fp) {
            fprintf(stderr,
                    "[ERROR] [%s()]: Failed opening logfile at {\"%s\"} [write].\n",
                    __func__, kls->conf.log_filepath);
            return false;
        } else {
            fprintf(log_fp, "%s", "");  //Reset log_fp
            fclose(log_fp);
        }
        log_fp = fopen(kls->conf.log_filepath, "a");
        if (!log_fp) {
            fprintf(stderr,
                    "[ERROR] [%s()]: Failed opening logfile at {\"%s\"} [append].\n",
                    __func__, kls->conf.log_filepath);
            return false;
        } else {
            kls->conf.log_fp = log_fp;
        }
    }
    return true;
}
 
static bool kls__try_grow(Koliseo* kls, ptrdiff_t needed);

#ifndef KOLISEO_HAS_LOCATE
KLS_Push_Result kls__advance(Koliseo* kls, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, ptrdiff_t* padding, const char* caller_name)
#else
KLS_Push_Result kls__advance_dbg(Koliseo* kls, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, ptrdiff_t* padding, const char* caller_name, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
#ifdef KLS_DEBUG_CORE
#ifndef _WIN32
    struct timespec start_time, end_time;
    clock_gettime(CLOCK_MONOTONIC, &start_time);
#else
    LARGE_INTEGER start_time, end_time, frequency;
    QueryPerformanceFrequency(&frequency);
    QueryPerformanceCounter(&start_time);
#endif
#endif
 
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", caller_name);
        exit(EXIT_FAILURE);
    }
    if ((kls->has_temp == 1) && (kls->conf.block_while_has_temp == 1)) {
        return (KLS_Push_Result) {
            .p = NULL,
            .error = KLS_PUSH_WITH_TEMP_ACTIVE,
        };
    }
 
#ifndef KOLISEO_HAS_LOCATE
    KLS_Push_Error res = kls__check_available(kls, size, align, count, __func__);
#else
    KLS_Push_Error res = kls__check_available_dbg(kls, size, align, count, __func__, loc);
#endif
    if (res != 0) {
        return (KLS_Push_Result) {
            .p = NULL,
            .error = res,
        };
    }
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
    const ptrdiff_t pad = -current->offset & (align - 1);
    *padding = pad;
    char *p = current->data + current->offset + pad;
    current->prev_offset = current->offset;
    current->offset += pad + size * count;
 
    KLS_ASAN_UNPOISON(p, size * count);
 
    char h_size[200];
    kls_formatSize(size * count, h_size, sizeof(h_size));
    //sprintf(msg,"Pushed zeroes, size (%li) for KLS.",size);
    //kls_log("KLS",msg);
#ifdef KLS_DEBUG_CORE
    kls_log(current, "KLS", "Pushed zeroes on KLS, size (%s). Curr offset: { %p }.", h_size, current->data + current->offset);
    if (current->conf.collect_stats == 1) {
#ifndef _WIN32
        clock_gettime(CLOCK_MONOTONIC, &end_time);  // %.9f
        double elapsed_time =
            (end_time.tv_sec - start_time.tv_sec) + (end_time.tv_nsec -
                start_time.tv_nsec) / 1e9;
#else
        QueryPerformanceCounter(&end_time); // %.7f
        double elapsed_time =
            (double)(end_time.QuadPart -
                     start_time.QuadPart) / frequency.QuadPart;
#endif
        if (elapsed_time > current->stats.worst_pushcall_time) {
            current->stats.worst_pushcall_time = elapsed_time;
        }
    }
#endif
    if (current->conf.collect_stats == 1) {
        current->stats.tot_pushes += 1;
    }
    return (KLS_Push_Result) {
        .p = p,
        .error = KLS_PUSH_OK,
    };
}

#ifndef KOLISEO_HAS_LOCATE
KLS_Push_Error kls__check_available(Koliseo* kls, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, const char* caller_name)
#else
KLS_Push_Error kls__check_available_dbg(Koliseo* kls, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, const char* caller_name, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
    assert(kls != NULL);
    assert(caller_name != NULL);
    if (count < 0) {
        return KLS_PUSH_NEGATIVE_COUNT;
    }
    if (size < 1) {
        return KLS_PUSH_SIZE_LT1;
    }
    if (align < 1) {
        return KLS_PUSH_ALIGN_LT1;
    }
    if (! ((align & (align - 1)) == 0)) {
        return KLS_PUSH_ALIGN_NOT_POW2;
    }
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
    const ptrdiff_t available = current->size - current->offset;
    const ptrdiff_t padding = -current->offset & (align - 1);
    if (count == 0) {
        if (current->conf.allow_zerocount_push != 1) {
            return KLS_PUSH_ZEROCOUNT;
        } else {
#ifdef KLS_DEBUG_CORE
            kls_log(current, "DEBUG", "Accepting zero-count push: conf.allow_zerocount_push was 1");
#endif // KLS_DEBUG_CORE
        }
    }
 
    if (count > PTRDIFF_MAX / size || available - padding < size * count) {
        if (count > PTRDIFF_MAX / size) {
            return KLS_PUSH_PTRDIFF_MAX;
        } else {
            if (current->conf.growable == 1 && kls__try_grow(current, (size * count) + padding)) {
                return KLS_PUSH_OK;
            }
            return KLS_PUSH_OOM;
        }
    }
    return KLS_PUSH_OK;
}

#ifndef KOLISEO_HAS_LOCATE
KLS_Push_Result kls__temp_advance(Koliseo_Temp* kls_t, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, ptrdiff_t* padding, const char* caller_name)
#else
KLS_Push_Result kls__temp_advance_dbg(Koliseo_Temp* kls_t, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count, ptrdiff_t* padding, const char* caller_name, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
#ifdef KLS_DEBUG_CORE
#ifndef _WIN32
    struct timespec start_time, end_time;
    clock_gettime(CLOCK_MONOTONIC, &start_time);
#else
    LARGE_INTEGER start_time, end_time, frequency;
    QueryPerformanceFrequency(&frequency);
    QueryPerformanceCounter(&start_time);
#endif
#endif
 
    if (kls_t == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo_Temp was NULL.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
    Koliseo *kls = kls_t->kls;
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Referred Koliseo was NULL.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
#ifndef KOLISEO_HAS_LOCATE
    KLS_Push_Error res = kls__check_available(kls, size, align, count, __func__);
#else
    KLS_Push_Error res = kls__check_available_dbg(kls, size, align, count, __func__, loc);
#endif
    if (res != 0) {
        return (KLS_Push_Result) {
            .p = NULL,
            .error = res,
        };
    }
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
    ptrdiff_t pad = -current->offset & (align - 1);
    char *p = current->data + current->offset + pad;
    *padding = pad;
    current->prev_offset = current->offset;
    current->offset += pad + size * count;
 
    KLS_ASAN_UNPOISON(p, size * count);
 
    char h_size[200];
    kls_formatSize(size * count, h_size, sizeof(h_size));
    //sprintf(msg,"Pushed zeroes, size (%li) for KLS.",size);
    //kls_log("KLS",msg);
#ifdef KLS_DEBUG_CORE
    if (current->conf.collect_stats == 1) {
#ifndef _WIN32
        clock_gettime(CLOCK_MONOTONIC, &end_time);  // %.9f
        double elapsed_time =
            (end_time.tv_sec - start_time.tv_sec) + (end_time.tv_nsec -
                start_time.tv_nsec) / 1e9;
#else
        QueryPerformanceCounter(&end_time); // %.7f
        double elapsed_time =
            (double)(end_time.QuadPart -
                     start_time.QuadPart) / frequency.QuadPart;
#endif
        if (elapsed_time > current->stats.worst_pushcall_time) {
            current->stats.worst_pushcall_time = elapsed_time;
        }
    }
    kls_log(current, "KLS", "Pushed zeroes on Temp_KLS, size (%s). Curr offset: { %p }.", h_size, current->data + current->offset);
#endif
    if (current->conf.collect_stats == 1) {
        current->stats.tot_temp_pushes += 1;
    }
    return (KLS_Push_Result) {
        .p = p,
        .error = KLS_PUSH_OK,
    };
}
 
bool kls__try_grow(Koliseo* kls, ptrdiff_t needed)
{
    assert(kls->next == NULL);
    ptrdiff_t new_size = KLS_MAX(kls->size * 2, needed);
    KLS_Hooks hooks = kls->hooks;
    hooks.on_new_handler = NULL; // We don't want the extension handler to run again on the tail
    // since the extension_data is shared
    Koliseo* new_kls = kls_new_conf_alloc_ext(new_size, kls->conf, KLS_DEFAULT_ALLOCF, KLS_DEFAULT_FREEF, hooks, kls->extension_data);
    kls_log(kls, "DEBUG", "%s(): growing Koliseo, new size: {%td}", __func__, new_size);
    if (!new_kls) return false;
    kls->next = new_kls;
    return true;
}

void *kls_push(Koliseo *kls, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count)
{
    ptrdiff_t padding = 0;
#ifndef KOLISEO_HAS_LOCATE
    KLS_Push_Result res = kls__advance(kls, size, align, count, &padding, __func__);
    return kls__handle_push_result(kls, res, size, align, count, padding, __func__);
#else
    KLS_Push_Result res = kls__advance_dbg(kls, size, align, count, &padding, __func__, KLS_HERE);
    return kls__handle_push_result_dbg(kls, res, size, align, count, padding, __func__, KLS_HERE);
#endif // KOLISEO_HAS_LOCATE
}

#ifndef KOLISEO_HAS_LOCATE
void *kls_push_zero(Koliseo *kls, ptrdiff_t size, ptrdiff_t align,
                    ptrdiff_t count)
#else
void *kls_push_zero_dbg(Koliseo *kls, ptrdiff_t size, ptrdiff_t align,
                        ptrdiff_t count, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
    ptrdiff_t padding = 0;
#ifndef KOLISEO_HAS_LOCATE
    KLS_Push_Result res = kls__advance(kls, size, align, count, &padding, __func__);
    void* p = kls__handle_push_result(kls, res, size, align, count, padding, __func__);
#else
    KLS_Push_Result res = kls__advance_dbg(kls, size, align, count, &padding, __func__, loc);
    void* p = kls__handle_push_result_dbg(kls, res, size, align, count, padding, __func__, loc);
#endif // KOLISEO_HAS_LOCATE
    if (!p) return NULL;
    //Zero new area
    memset(p, 0, size * count);
    return p;
}

#ifndef KOLISEO_HAS_LOCATE
void *kls_push_zero_ext(Koliseo *kls, ptrdiff_t size, ptrdiff_t align,
                        ptrdiff_t count)
#else
void *kls_push_zero_ext_dbg(Koliseo *kls, ptrdiff_t size, ptrdiff_t align,
                            ptrdiff_t count, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
 
    ptrdiff_t padding = 0;
#ifndef KOLISEO_HAS_LOCATE
    KLS_Push_Result res = kls__advance(kls, size, align, count, &padding, __func__);
    void* p = kls__handle_push_result(kls, res, size, align, count, padding, __func__);
#else
    KLS_Push_Result res = kls__advance_dbg(kls, size, align, count, &padding, __func__, loc);
    void* p = kls__handle_push_result_dbg(kls, res, size, align, count, padding, __func__, loc);
#endif // KOLISEO_HAS_LOCATE
    if (!p) return NULL;
    //Zero new area
    memset(p, 0, size * count);
 
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
 
    if (current->hooks.on_push_handler != NULL) {
        /*
        struct KLS_EXTENSION_AR_DEFAULT_ARGS {
            const char* region_name;
            size_t region_name_len;
            const char* region_desc;
            size_t region_desc_len;
            int region_type;
        };
        struct KLS_EXTENSION_AR_DEFAULT_ARGS ar_args = {
            .region_name = KOLISEO_DEFAULT_REGION_NAME,
            .region_name_len = strlen(KOLISEO_DEFAULT_REGION_NAME),
            .region_desc = KOLISEO_DEFAULT_REGION_DESC,
            .region_desc_len = strlen(KOLISEO_DEFAULT_REGION_DESC),
            .region_type = KLS_None
        };
        kls->hooks.on_push_handler(kls, padding, (void*)&ar_args);
        */
        current->hooks.on_push_handler(current, padding, __func__, NULL);
    }
    return p;
}
 
#ifndef KOLISEO_HAS_LOCATE
char* kls_vsprintf(Koliseo* kls, const char* fmt, va_list args)
#else
char* kls_vsprintf_dbg(Koliseo* kls, Koliseo_Loc loc, const char* fmt, va_list args)
#endif // KOLISEO_HAS_LOCATE
{
    va_list args_copy;
    va_copy(args_copy, args);
    int len = vsnprintf(NULL, 0, fmt, args);
#ifndef KOLISEO_HAS_LOCATE
    char* str = KLS_PUSH_ARR(kls, char, len+1);
#else
    char* str = kls_push_zero_ext_dbg(kls, sizeof(char), KLS_ALIGNOF(char), len+1, loc);
#endif // KOLISEO_HAS_LOCATE
    vsnprintf(str, len+1, fmt, args_copy);
    va_end(args_copy);
    return str;
}

#ifndef KOLISEO_HAS_LOCATE
char* kls_sprintf(Koliseo* kls, const char* fmt, ...)
#else
char* kls_sprintf_dbg(Koliseo* kls, Koliseo_Loc loc, const char* fmt, ...)
#endif // KOLISEO_HAS_LOCATE
{
    va_list args;
    va_start(args, fmt);
#ifndef KOLISEO_HAS_LOCATE
    char* str = kls_vsprintf(kls, fmt, args);
#else
    char* str = kls_vsprintf_dbg(kls, loc, fmt, args);
#endif // KOLISEO_HAS_LOCATE
    va_end(args);
    return str;
}

#ifndef KOLISEO_HAS_LOCATE
void *kls_temp_push_zero_ext(Koliseo_Temp *t_kls, ptrdiff_t size,
                             ptrdiff_t align, ptrdiff_t count)
#else
void *kls_temp_push_zero_ext_dbg(Koliseo_Temp *t_kls, ptrdiff_t size,
                                 ptrdiff_t align, ptrdiff_t count, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
 
    ptrdiff_t padding = 0;
 
#ifndef KOLISEO_HAS_LOCATE
    KLS_Push_Result res = kls__temp_advance(t_kls, size, align, count, &padding, __func__);
#else
    KLS_Push_Result res = kls__temp_advance_dbg(t_kls, size, align, count, &padding, __func__, loc);
#endif // KOLISEO_HAS_LOCATE
    if (res.error) return NULL;
    //Zero new area
    memset(res.p, 0, size * count);
 
    Koliseo* kls = t_kls->kls;
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
 
    if (current->hooks.on_temp_push_handler != NULL) {
        // Call on_temp_push extension with empty user arg
        current->hooks.on_temp_push_handler(t_kls, padding, __func__, NULL);
    }
    return res.p;
}
 
#ifndef KOLISEO_HAS_LOCATE
char* kls_temp_vsprintf(Koliseo_Temp* kls_t, const char* fmt, va_list args)
#else
char* kls_temp_vsprintf_dbg(Koliseo_Temp* kls_t, Koliseo_Loc loc, const char* fmt, va_list args)
#endif // KOLISEO_HAS_LOCATE
{
    va_list args_copy;
    va_copy(args_copy, args);
    int len = vsnprintf(NULL, 0, fmt, args);
#ifndef KOLISEO_HAS_LOCATE
    char* str = KLS_PUSH_ARR_T(kls_t, char, len+1);
#else
    char* str = kls_temp_push_zero_ext_dbg(kls_t, sizeof(char), KLS_ALIGNOF(char), len+1, loc);
#endif // KOLISEO_HAS_LOCATE
    vsnprintf(str, len+1, fmt, args_copy);
    va_end(args_copy);
    return str;
}
 

#ifndef KOLISEO_HAS_LOCATE
char* kls_temp_sprintf(Koliseo_Temp* kls_t, const char* fmt, ...)
#else
char* kls_temp_sprintf_dbg(Koliseo_Temp* kls_t, Koliseo_Loc loc, const char* fmt, ...)
#endif // KOLISEO_HAS_LOCATE
{
    va_list args;
    va_start(args, fmt);
#ifndef KOLISEO_HAS_LOCATE
    char* str = kls_temp_vsprintf(kls_t, fmt, args);
#else
    char* str = kls_temp_vsprintf_dbg(kls_t, loc, fmt, args);
#endif // KOLISEO_HAS_LOCATE
    va_end(args);
    return str;
}

bool kls_contains_pointer(const Koliseo *kls, const void *ptr)
{
    const char *p = ptr;
 
    for (const Koliseo *cur = kls; cur; cur = cur->next) {
        if (p >= cur->data &&
            p < cur->data + cur->offset) {
            return true;
        }
    }
 
    return false;
}

#ifndef KOLISEO_HAS_LOCATE
void *kls_repush(Koliseo *kls, void* old, ptrdiff_t size, ptrdiff_t align,
                 ptrdiff_t old_count, ptrdiff_t new_count)
#else
void *kls_repush_dbg(Koliseo *kls, void* old, ptrdiff_t size, ptrdiff_t align,
                     ptrdiff_t old_count, ptrdiff_t new_count, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
    if (!old) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): old was NULL.\n",
                __func__);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): old was NULL.\n",
                KLS_Loc_Arg(loc),
                __func__);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (!kls_contains_pointer(kls, old)) {
        fprintf(stderr,
                "[KLS] %s(): pointer %p does not belong to this Koliseo.\n",
                __func__,
                old);
        return NULL;
    }
    if (old_count < 0) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): old_count [%td] was < 0.\n",
                __func__,
                old_count);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): old_count [%td] was < 0.\n",
                KLS_Loc_Arg(loc),
                __func__,
                old_count);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (new_count < 0) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): new_count [%td] was < 0.\n",
                __func__,
                new_count);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): new_count [%td] was < 0.\n",
                KLS_Loc_Arg(loc),
                __func__,
                new_count);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (size < 1) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): size [%td] was < 1.\n",
                __func__,
                size);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): size [%td] was < 1.\n",
                KLS_Loc_Arg(loc),
                __func__,
                size);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (align < 1) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): align [%td] was < 1.\n",
                __func__,
                align);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): align [%td] was < 1.\n",
                KLS_Loc_Arg(loc),
                __func__,
                align);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (! ((align & (align - 1)) == 0)) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): align [%td] was not a power of 2.\n",
                __func__,
                align);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): align [%td] was not a power of 2.\n",
                KLS_Loc_Arg(loc),
                __func__,
                align);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (kls->has_temp == 1 && kls->conf.block_while_has_temp == 1) {
        fprintf(stderr, "%s(): kls has an active temp\n", __func__);
        return NULL;
    }
    if (old_count >= new_count) return old;
 
    ptrdiff_t growth = new_count - old_count;
 
    if ((size_t)old_count > SIZE_MAX / (size_t)size) return NULL;
 
    if ((size_t)new_count > SIZE_MAX / (size_t)size) return NULL;
 
    size_t old_size = (size_t)old_count * (size_t)size;
    size_t new_size = (size_t)new_count * (size_t)size;
 
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
 
    ptrdiff_t padding = 0;
    bool is_last_alloc = old == (current->data + current->prev_offset);
 
    void* new_ptr = NULL;
    if (is_last_alloc) {
        // Extend last allocation
 
        // Save prev_offset
        const ptrdiff_t saved_prev_offset = current->prev_offset;
#ifndef KOLISEO_HAS_LOCATE
        KLS_Push_Result res = kls__advance(current, size, align, growth, &padding, __func__);
        void* p = kls__handle_push_result(current, res, size, align, growth, padding, __func__);
#else
        KLS_Push_Result res = kls__advance_dbg(current, size, align, growth, &padding, __func__, loc);
        void* p = kls__handle_push_result_dbg(current, res, size, align, growth, padding, __func__, loc);
#endif // KOLISEO_HAS_LOCATE
        if (!p) return NULL;
        size_t added_size = new_size - old_size;
        //Zero new area
        memset(p, 0, added_size);
        new_ptr = old;
 
        Koliseo* new_current = current;
        while (new_current->next != NULL) {
            new_current = new_current->next;
        }
 
#ifdef KLS_DEBUG_CORE
        const ptrdiff_t new_byte_slice_len = new_current->offset - new_current->prev_offset + padding;
        kls_log(new_current, "KLS", "Extended last allocation on KLS, new_byte_slice_len (%td), padding (%td). Curr offset: { %p } Old: { %p }, New: { %p }.", new_byte_slice_len, padding, new_current->data + new_current->offset, old, new_ptr);
#endif
 
        // Restore saved prev_offset
        current->prev_offset = saved_prev_offset;
    } else {
#ifdef KLS_DEBUG_CORE
        kls_log(current, "KLS", "%s(): pushed new memory", __func__);
#endif
        new_ptr = kls_push_zero_ext(kls, size, align, new_count);
        if (new_ptr && old_size > 0) {
            memcpy(new_ptr, old, old_size);
        }
    }
    return new_ptr;
}

#ifndef KOLISEO_HAS_LOCATE
void *kls_temp_repush(Koliseo_Temp *t_kls, void* old, ptrdiff_t size, ptrdiff_t align,
                      ptrdiff_t old_count, ptrdiff_t new_count)
#else
void *kls_temp_repush_dbg(Koliseo_Temp *t_kls, void* old, ptrdiff_t size, ptrdiff_t align,
                          ptrdiff_t old_count, ptrdiff_t new_count, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
    if (!old) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): old was NULL.\n",
                __func__);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): old was NULL.\n",
                KLS_Loc_Arg(loc),
                __func__);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (!kls_contains_pointer(t_kls->kls, old)) {
        fprintf(stderr,
                "[KLS] %s(): pointer %p does not belong to this Koliseo.\n",
                __func__,
                old);
        return NULL;
    }
    if (old_count < 0) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): old_count [%td] was < 0.\n",
                __func__,
                old_count);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): old_count [%td] was < 0.\n",
                KLS_Loc_Arg(loc),
                __func__,
                old_count);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (new_count < 0) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): new_count [%td] was < 0.\n",
                __func__,
                new_count);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): new_count [%td] was < 0.\n",
                KLS_Loc_Arg(loc),
                __func__,
                new_count);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (size < 1) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): size [%td] was < 1.\n",
                __func__,
                size);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): size [%td] was < 1.\n",
                KLS_Loc_Arg(loc),
                __func__,
                size);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (align < 1) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): align [%td] was < 1.\n",
                __func__,
                align);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): align [%td] was < 1.\n",
                KLS_Loc_Arg(loc),
                __func__,
                align);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
    if (! ((align & (align - 1)) == 0)) {
#ifndef KOLISEO_HAS_LOCATE
        fprintf(stderr,
                "[KLS] %s(): align [%td] was not a power of 2.\n",
                __func__,
                align);
#else
        fprintf(stderr,
                "[KLS] " KLS_Loc_Fmt "%s(): align [%td] was not a power of 2.\n",
                KLS_Loc_Arg(loc),
                __func__,
                align);
#endif // KOLISEO_HAS_LOCATE
        return NULL;
    }
 
    if (old_count >= new_count) return old;
 
    ptrdiff_t growth = new_count - old_count;
 
    if ((size_t)old_count > SIZE_MAX / (size_t)size) return NULL;
 
    if ((size_t)new_count > SIZE_MAX / (size_t)size) return NULL;
 
    size_t old_size = (size_t)old_count * (size_t)size;
    size_t new_size = (size_t)new_count * (size_t)size;
 
    Koliseo* current = t_kls->kls;
    while (current->next != NULL) {
        current = current->next;
    }
 
    ptrdiff_t padding = 0;
    bool is_last_alloc = old == (current->data + current->prev_offset);
 
    void* new_ptr = NULL;
    if (is_last_alloc) {
        // Extend last allocation
 
        // Save prev_offset
        const ptrdiff_t saved_prev_offset = current->prev_offset;
#ifndef KOLISEO_HAS_LOCATE
        KLS_Push_Result res = kls__temp_advance(t_kls, size, align, growth, &padding, __func__);
        void* p = kls__handle_push_result(current, res, size, align, growth, padding, __func__);
#else
        KLS_Push_Result res = kls__temp_advance_dbg(t_kls, size, align, growth, &padding, __func__, loc);
        void* p = kls__handle_push_result_dbg(current, res, size, align, growth, padding, __func__, loc);
#endif // KOLISEO_HAS_LOCATE
        if (!p) return NULL;
        size_t added_size = new_size - old_size;
        //Zero new area
        memset(p, 0, added_size);
        new_ptr = old;
 
        Koliseo* new_current = current;
        while (new_current->next != NULL) {
            new_current = new_current->next;
        }
 
#ifdef KLS_DEBUG_CORE
        const ptrdiff_t new_byte_slice_len = new_current->offset - new_current->prev_offset + padding;
        kls_log(new_current, "KLS", "Extended last allocation on Temp_KLS, new_byte_slice_len (%td), padding (%td). Curr offset: { %p } Old: { %p }, New: { %p }.", new_byte_slice_len, padding, new_current->data + new_current->offset, old, new_ptr);
#endif
 
        // Restore saved prev_offset
        current->prev_offset = saved_prev_offset;
    } else {
#ifdef KLS_DEBUG_CORE
        kls_log(current, "KLS", "%s(): pushed new memory", __func__);
#endif
        new_ptr = kls_temp_push_zero_ext(t_kls, size, align, new_count);
        if (new_ptr && old_size > 0) {
            memcpy(new_ptr, old, old_size);
        }
    }
    return new_ptr;
}

void print_kls_2file(FILE *fp, const Koliseo *kls)
{
    if (fp == NULL) {
        fprintf(stderr, "print_kls_2file():  fp was NULL.\n");
        return;
    }
    if (kls == NULL) {
        fprintf(fp, "[KLS] kls was NULL.\n");
    } else {
        fprintf(fp, "\n[KLS] API Level: { %i }\n", int_koliseo_version());
        fprintf(fp, "\n[INFO] Conf: { " KLS_Conf_Fmt " }\n",
                KLS_Conf_Arg(kls->conf));
        fprintf(fp, "\n[INFO] Stats: { " KLS_Stats_Fmt " }\n",
                KLS_Stats_Arg(kls->stats));
        fprintf(fp, "\n[KLS] Size: { %td }\n", kls->size);
        char human_size[200];
        char curr_size[200];
        kls_formatSize(kls->size, human_size, sizeof(human_size));
        fprintf(fp, "[KLS] Size (Human): { %s }\n", human_size);
        kls_formatSize(kls->offset, curr_size, sizeof(curr_size));
        fprintf(fp, "[KLS] Used (Human): { %s }\n", curr_size);
        fprintf(fp, "[KLS] Offset: { %td }\n", kls->offset);
        fprintf(fp, "[KLS] Prev_Offset: { %td }\n", kls->prev_offset);
        fprintf(fp, "\n");
    }
}

void print_dbg_kls(const Koliseo *kls)
{
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", __func__);
        exit(EXIT_FAILURE);
    }
    print_kls_2file(stderr, kls);
}

void print_temp_kls_2file(FILE *fp, const Koliseo_Temp *t_kls)
{
    if (fp == NULL) {
        fprintf(stderr, "print_temp_kls_2file():  fp was NULL.\n");
        exit(EXIT_FAILURE);
    }
    if (t_kls == NULL) {
        fprintf(fp, "[KLS_T] t_kls was NULL.");
    } else if (t_kls->kls == NULL) {
        fprintf(fp, "[KLS_T] [%s()]: Referred Koliseo was NULL.\n", __func__);
    } else {
        const Koliseo *kls = t_kls->kls;
        fprintf(fp, "\n[KLS_T] API Level: { %i }\n", int_koliseo_version());
        fprintf(fp, "\n[KLS_T] Temp Size: { %td }\n",
                kls->size - t_kls->offset);
        fprintf(fp, "\n[KLS_T] Refer Size: { %td }\n", kls->size);
        char human_size[200];
        char curr_size[200];
        kls_formatSize(kls->size - t_kls->offset, human_size,
                       sizeof(human_size));
        fprintf(fp, "[KLS_T] Temp Size Human: { %s }\n", human_size);
        kls_formatSize(kls->size, human_size, sizeof(human_size));
        fprintf(fp, "[KLS_T] Refer Size Human: { %s }\n", human_size);
        kls_formatSize(kls->offset, curr_size, sizeof(curr_size));
        fprintf(fp, "[KLS_T] Inner Used (Human): { %s }\n", curr_size);
        kls_formatSize(t_kls->offset, curr_size, sizeof(curr_size));
        fprintf(fp, "[KLS_T] Temp Used (Human): { %s }\n", curr_size);
        fprintf(fp, "[KLS_T] Inner Offset: { %td }\n", kls->offset);
        fprintf(fp, "[KLS_T] Temp Offset: { %td }\n", t_kls->offset);
        fprintf(fp, "[KLS_T] Inner Prev_Offset: { %td }\n", kls->prev_offset);
        fprintf(fp, "[KLS_T] Temp Prev_Offset: { %td }\n\n",
                t_kls->prev_offset);
    }
}

void print_dbg_temp_kls(const Koliseo_Temp *t_kls)
{
    print_temp_kls_2file(stderr, t_kls);
}

void kls_formatSize(size_t size, char *outputBuffer, size_t bufferSize)
{
    static const char *units[] = {
        "bytes", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB"
    };
 
    const int numUnits = sizeof(units) / sizeof(units[0]);
 
    size_t unitIndex = 0;
    long double sizeValue = (long double)size;
 
    while (sizeValue >= 1000 && unitIndex < numUnits - 1) {
        sizeValue /= 1000;
        unitIndex++;
    }
 
    snprintf(outputBuffer, bufferSize, "%.2Lf %s", sizeValue, units[unitIndex]);
}

void kls_clear(Koliseo *kls)
{
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", __func__);
        exit(EXIT_FAILURE);
    }
    //Reset pointer
    kls->prev_offset = kls->offset;
    kls->offset = sizeof(*kls);
    if (kls->next) {
        Koliseo *chain = kls->next;
        kls->next = NULL;
        if (chain) {
            kls_free(chain);
            kls->extension_data = NULL; // The chain shares this pointer and we need to restore the extension data
            if (kls->hooks.on_new_handler) {
                kls->hooks.on_new_handler(kls);
            }
        }
    }
    KLS_ASAN_POISON(kls->data + kls->offset, kls->size - kls->offset);
#ifdef KLS_DEBUG_CORE
    kls_log(kls, "KLS", "API Level { %i } -> Cleared offsets for KLS.",
            int_koliseo_version());
#endif
}

void kls_free(Koliseo *kls)
{
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", __func__);
        exit(EXIT_FAILURE);
    }
    Koliseo* current = kls;
    while (current) {
        if (current->has_temp == 1) {
#ifdef KLS_DEBUG_CORE
            kls_log(current, "KLS",
                    "API Level { %i } -> KLS had an active Koliseo_Temp.",
                    int_koliseo_version());
#endif
            kls_temp_end(current->t_kls);
        }
        if (current->hooks.on_free_handler != NULL) {
            // Call on_free() extension
            current->hooks.on_free_handler(current);
            for (Koliseo *k = current; k; k = k->next) {
                // We NULL all subsequent extension_data pointers, since the chain shares it
                // and it's freed in the first hook call
                k->extension_data = NULL;
            }
        }
        Koliseo* next = current->next;
        current->next = NULL;
        //kls_clear(current);
#ifdef KLS_DEBUG_CORE
        kls_log(current, "KLS", "API Level { %i } -> Freeing KLS.",
                int_koliseo_version());
#endif
        if (current->conf.log_fp != NULL && current->conf.log_fp != stdout
            && current->conf.log_fp != stderr) {
#ifdef KLS_DEBUG_CORE
            kls_log(current, "KLS", "Closing kls log file. Path: {\"%s\"}.",
                    kls->conf.log_filepath);
#endif
            int close_res = fclose(current->conf.log_fp);
            if (close_res != 0) {
                fprintf(stderr,
                        "[ERROR]    %s(): Failed fclose() on log_fp. Path: {\"%s\"}.",
                        __func__, current->conf.log_filepath);
            }
        } else if (current->conf.log_fp == stdout || current->conf.log_fp == stderr) {
            if (current->conf.verbose_lvl > 1) {
                fprintf(stderr,
                        "[INFO]    %s(): kls->conf.log_fp is %s. Not closing it.\n",
                        __func__,
                        (current->conf.log_fp == stdout ? "stdout" : "stderr"));
            }
        }
        if (current->free_func == NULL) {
            fprintf(stderr,
                    "[ERROR]    %s(): free function was NULL.\n", __func__);
            return;
        }
        current->free_func(current);
        current = next;
    }
}

#ifndef KOLISEO_HAS_LOCATE
Koliseo_Temp *kls_temp_start(Koliseo *kls)
#else
Koliseo_Temp *kls_temp_start_dbg(Koliseo *kls, Koliseo_Loc loc)
#endif // KOLISEO_HAS_LOCATE
{
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", __func__);
        exit(EXIT_FAILURE);
    }
    Koliseo* current = kls;
    while (current->next != NULL) {
        current = current->next;
    }
    if (current->has_temp != 0) {
        fprintf(stderr,
                "[ERROR] [%s()]: Passed Koliseo->has_temp is not 0. {%i}\n",
                __func__, current->has_temp);
#ifdef KLS_DEBUG_CORE
        kls_log(current, "ERROR", "[%s()]: Passed Koliseo->has_temp != 0 . {%i}",
                __func__, current->has_temp);
#endif
        return NULL;
    }
    ptrdiff_t prev = current->prev_offset;
    ptrdiff_t off = current->offset;
 
    Koliseo_Temp *tmp = KLS_PUSH(current, Koliseo_Temp);
    tmp->kls = current;
    tmp->prev_offset = prev;
    tmp->offset = off;
#ifdef KLS_DEBUG_CORE
    kls_log(current, "INFO", "Passed kls conf: " KLS_Conf_Fmt "\n",
            KLS_Conf_Arg(current->conf));
#endif
 
    current->has_temp = 1;
    current->t_kls = tmp;
    if (current->hooks.on_temp_start_handler != NULL) {
        // Call on_temp_start extension
        current->hooks.on_temp_start_handler(tmp);
    }
#ifdef KLS_DEBUG_CORE
    kls_log(current, "KLS", "Prepared new Temp KLS.");
#endif
    return tmp;
}

void kls_temp_end(Koliseo_Temp *tmp_kls)
{
    if (tmp_kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo_Temp was NULL.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
 
    Koliseo *kls_ref = tmp_kls->kls;
    if (kls_ref == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Referred Koliseo was NULL.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
 
    if (kls_ref->hooks.on_temp_free_handler != NULL) {
        // Call on_temp_free() extension
        kls_ref->hooks.on_temp_free_handler(tmp_kls);
    }
 
#ifdef KLS_DEBUG_CORE
    kls_log(kls_ref, "KLS", "Ended Temp KLS.");
#endif
    tmp_kls->kls->has_temp = 0;
    tmp_kls->kls->t_kls = NULL;
#if defined(__SANITIZE_ADDRESS__)
    ptrdiff_t old_offset = tmp_kls->kls->offset;
    ptrdiff_t new_offset = tmp_kls->offset;
#endif // __SANITIZE_ADDRESS__
    tmp_kls->kls->prev_offset = tmp_kls->prev_offset;
    tmp_kls->kls->offset = tmp_kls->offset;
 
    // Free any Koliseo chained after the current one
    Koliseo* to_free = tmp_kls->kls->next;
    if (to_free != NULL) {
        kls_free(to_free);
        tmp_kls->kls->next = NULL;
    }
 
    KLS_ASAN_POISON(tmp_kls->kls->data + new_offset, old_offset - new_offset);
    tmp_kls = NULL; // statement with no effect TODO: Clear tmp_kls from caller
    if (kls_ref->conf.collect_stats == 1) {
        kls_ref->stats.tot_temp_pushes = 0;
        kls_ref->stats.tot_temp_pops = 0;
    }
}
 
#ifdef KOLISEO_HAS_EXPER
void *kls_pop(Koliseo *kls, ptrdiff_t size, ptrdiff_t align, ptrdiff_t count)
{
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo was NULL.\n", __func__);
        exit(EXIT_FAILURE);
    }
    ptrdiff_t padding = -kls->offset & (align - 1);
    if (count > PTRDIFF_MAX / size
        || (kls->size + kls->offset) < (size * count)) {
        fprintf(stderr, "[KLS] Failed %s() call.\n", __func__);
        kls_free(kls);
        exit(EXIT_FAILURE);
    }
    char *p = kls->data + kls->offset - padding - size * count;
    kls->prev_offset = kls->offset;
    kls->offset -= padding + size * count;
#ifdef KLS_DEBUG_CORE
    kls_log(kls, "KLS", "Popped (%td) for KLS. Curr offset: { %p }",
            size, kls->data + kls->offset);
#endif
    if (kls->conf.collect_stats == 1) {
        kls->stats.tot_pops += 1;
    }
    return p;
}

void *kls_temp_pop(Koliseo_Temp *t_kls, ptrdiff_t size, ptrdiff_t align,
                   ptrdiff_t count)
{
    if (t_kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Passed Koliseo_Temp was NULL.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
    Koliseo *kls = t_kls->kls;
    if (kls == NULL) {
        fprintf(stderr, "[ERROR] [%s()]: Referred Koliseo was NULL.\n",
                __func__);
        exit(EXIT_FAILURE);
    }
    ptrdiff_t padding = -kls->offset & (align - 1);
    if (count > PTRDIFF_MAX / size
        || (kls->size + kls->offset) < (size * count)) {
        fprintf(stderr, "[KLS] Failed %s() call.\n", __func__);
        kls_free(kls);
        exit(EXIT_FAILURE);
    }
    char *p = kls->data + kls->offset - padding - size * count;
    kls->prev_offset = kls->offset;
    kls->offset -= padding + size * count;
#ifdef KLS_DEBUG_CORE
    kls_log(kls, "KLS", "Popped (%td) for Temp_KLS. Curr offset: { %p }.", size, kls->data + kls->offset);
#endif
    if (kls->conf.collect_stats == 1) {
        kls->stats.tot_temp_pops += 1;
    }
    return p;
}

char* kls_strdup(Koliseo* kls, char* source)
{
    char* dest = KLS_PUSH_STR(kls, source);
    KLS__STRCPY(dest, source);
    return dest;
}

char** kls_strdup_arr(Koliseo* kls, size_t count, char** source)
{
    char** strings = NULL;
    strings = KLS_PUSH_ARR(kls, char*, count);
    for (int i=0; i < count; i++) {
        strings[i] = KLS_STRDUP(kls, source[i]);
    }
    return strings;
}

char* kls_t_strdup(Koliseo_Temp* t_kls, char* source)
{
    char* dest = KLS_PUSH_STR_T(t_kls, source);
    KLS__STRCPY(dest, source);
    return dest;
}

char** kls_t_strdup_arr(Koliseo_Temp* t_kls, size_t count, char** source)
{
    char** strings = NULL;
    strings = KLS_PUSH_ARR_T(t_kls, char*, count);
    for (int i=0; i < count; i++) {
        strings[i] = KLS_STRDUP_T(t_kls, source[i]);
    }
    return strings;
}
#endif // KOLISEO_HAS_EXPER