From 61c1060a98b9989fe193bdbdcfdf44cad460f50e Mon Sep 17 00:00:00 2001 From: snarmph Date: Sun, 28 Sep 2025 16:31:22 +0200 Subject: [PATCH] . --- colla.c | 64 +- colla.h | 15 +- stb/stb_sprintf.h | 1945 ++++++++++++++++++++++++++++++++++++ tests/arena_tests.c | 235 +++++ tests/core_tests.c | 231 +++++ tests/highlight_tests.c | 178 ++++ tests/net_tests.c | 237 +++++ tests/os_tests.c | 387 +++++++ tests/parsers_tests.c | 369 +++++++ tests/pretty_print_tests.c | 33 + tests/runner.h | 45 + tests/str_tests.c | 456 +++++++++ tests/string_tests.c | 11 + tools/nob.c | 478 +++++++++ tools/noblin.c | 308 ++++++ tools/unit_tests.c | 82 ++ 16 files changed, 5043 insertions(+), 31 deletions(-) create mode 100644 stb/stb_sprintf.h create mode 100644 tests/arena_tests.c create mode 100644 tests/core_tests.c create mode 100644 tests/highlight_tests.c create mode 100644 tests/net_tests.c create mode 100644 tests/os_tests.c create mode 100644 tests/parsers_tests.c create mode 100644 tests/pretty_print_tests.c create mode 100644 tests/runner.h create mode 100644 tests/str_tests.c create mode 100644 tests/string_tests.c create mode 100644 tools/nob.c create mode 100644 tools/noblin.c create mode 100644 tools/unit_tests.c diff --git a/colla.c b/colla.c index e7135a6..0476367 100644 --- a/colla.c +++ b/colla.c @@ -34,13 +34,6 @@ colla_modules_e colla__initialised_modules = 0; -extern void os_init(void); -extern void os_cleanup(void); -#if !COLLA_NO_NET -extern void net_init(void); -extern void net_cleanup(void); -#endif - static char *colla_fmt__stb_callback(const char *buf, void *ud, int len) { // TODO maybe use os_write? fflush(stdout); @@ -154,9 +147,9 @@ tstr_t tstr_init(TCHAR *str, usize optional_len) { }; } -str16_t str16_init(u16 *str, usize optional_len) { +str16_t str16_init(char16_t *str, usize optional_len) { if (str && !optional_len) { - optional_len = wcslen(str); + optional_len = str16_len(str); } return (str16_t){ .buf = str, @@ -173,6 +166,19 @@ str_t str_from_str16(arena_t *arena, str16_t src) { return out; } +usize str16_len(char16_t *str) { +#if COLLA_WIN + return wcslen(str); +#else + usize len = 0; + while (*str) { + str++; + len++; + } + return len; +#endif +} + str_t str_from_tstr(arena_t *arena, tstr_t src) { #if COLLA_UNICODE return str_from_str16(arena, src); @@ -1057,14 +1063,13 @@ static void *arena__alloc_common(const arena_alloc_desc_t *desc) { usize new_cur = allocated + total; if (new_cur > page_end) { - usize extra_mem = os_pad_to_page(new_cur - page_end); usize page_size = os_get_system_info().page_size; - // TODO is this really correct? - usize num_of_pages = (extra_mem / page_size) + 1; - + usize prev_page = os_pad_to_page(allocated - page_size); + usize next_page = os_pad_to_page(new_cur); + usize num_of_pages = (next_page - prev_page) / page_size; colla_assert(num_of_pages > 0); - if (!os_commit(arena->cur, num_of_pages + 1)) { + if (!os_commit(arena->beg + prev_page, num_of_pages)) { if (!soft_fail) { fatal("failed to commit memory for virtual arena, tried to commit %zu pages\n", num_of_pages); } @@ -1214,7 +1219,7 @@ usize os_file_write_buf(oshandle_t handle, buffer_t buf) { } buffer_t os_file_read_all(arena_t *arena, strview_t path) { - oshandle_t fp = os_file_open(path, FILEMODE_READ); + oshandle_t fp = os_file_open(path, OS_FILE_READ); if (!os_handle_valid(fp)) { err("could not open file: %v", path); return (buffer_t){0}; @@ -1242,7 +1247,7 @@ buffer_t os_file_read_all_fp(arena_t *arena, oshandle_t handle) { } str_t os_file_read_all_str(arena_t *arena, strview_t path) { - oshandle_t fp = os_file_open(path, FILEMODE_READ); + oshandle_t fp = os_file_open(path, OS_FILE_READ); if (!os_handle_valid(fp)) { err("could not open file %v: %v", path, os_get_error_string(os_get_last_error())); return STR_EMPTY; @@ -1273,7 +1278,7 @@ str_t os_file_read_all_str_fp(arena_t *arena, oshandle_t handle) { } bool os_file_write_all(strview_t name, buffer_t buffer) { - oshandle_t fp = os_file_open(name, FILEMODE_WRITE); + oshandle_t fp = os_file_open(name, OS_FILE_WRITE); bool result = os_file_write_all_fp(fp, buffer); os_file_close(fp); return result; @@ -1284,7 +1289,7 @@ bool os_file_write_all_fp(oshandle_t handle, buffer_t buffer) { } bool os_file_write_all_str(strview_t name, strview_t data) { - oshandle_t fp = os_file_open(name, FILEMODE_WRITE); + oshandle_t fp = os_file_open(name, OS_FILE_WRITE); bool result = os_file_write_all_str_fp(fp, data); os_file_close(fp); return result; @@ -1295,7 +1300,7 @@ bool os_file_write_all_str_fp(oshandle_t handle, strview_t data) { } u64 os_file_time(strview_t path) { - oshandle_t fp = os_file_open(path, FILEMODE_READ); + oshandle_t fp = os_file_open(path, OS_FILE_READ); u64 result = os_file_time_fp(fp); os_file_close(fp); return result; @@ -1334,7 +1339,7 @@ usize os_pad_to_page(usize byte_count) { void ini__parse(arena_t *arena, ini_t *ini, const iniopt_t *options); ini_t ini_parse(arena_t *arena, strview_t filename, iniopt_t *opt) { - oshandle_t fp = os_file_open(filename, FILEMODE_READ); + oshandle_t fp = os_file_open(filename, OS_FILE_READ); ini_t out = ini_parse_fp(arena, fp, opt); os_file_close(fp); return out; @@ -2735,6 +2740,13 @@ http_url_t http_split_url(strview_t url) { } #if !COLLA_NO_NET + +// HTTP ///////////////////////////// + +http_res_t http_request(http_request_desc_t *req) { + return http_request_cb(req, NULL, NULL); +} + // WEBSOCKETS /////////////////////// #define WEBSOCKET_MAGIC "258EAFA5-E914-47DA-95CA-C5AB0DC85B11" @@ -2772,17 +2784,17 @@ buffer_t websocket_encode(arena_t *arena, strview_t message) { if (message.len > UINT16_MAX) extra += sizeof(u64); else if (message.len > UINT8_MAX) extra += sizeof(u16); u8 *bytes = alloc(arena, u8, message.len + extra); - bytes[0] = 0b10000001; - bytes[1] = 0b10000000; + bytes[0] = 0x81; // 0b10000001 + bytes[1] = 0x80; // 0b10000000; int offset = 2; if (message.len > UINT16_MAX) { - bytes[1] |= 0b01111111; + bytes[1] |= 0x7F; // 0b01111111; u64 len = htonll(message.len); memmove(bytes + 2, &len, sizeof(len)); offset += sizeof(u64); } else if (message.len > UINT8_MAX) { - bytes[1] |= 0b01111110; + bytes[1] |= 0x7E; // 0b01111110; u16 len = htons((u16)message.len); memmove(bytes + 2, &len, sizeof(len)); offset += sizeof(u16); @@ -2803,9 +2815,9 @@ str_t websocket_decode(arena_t *arena, buffer_t message) { str_t out = STR_EMPTY; u8 *bytes = message.data; - bool mask = bytes[1] & 0b10000000; + bool mask = bytes[1] & 0x80; // 0b10000000; int offset = 2; - u64 msglen = bytes[1] & 0b01111111; + u64 msglen = bytes[1] & 0x7F; // 0b01111111; // 16bit msg len if (msglen == 126) { diff --git a/colla.h b/colla.h index 1b22aff..e0a6c30 100644 --- a/colla.h +++ b/colla.h @@ -1,9 +1,13 @@ #ifndef COLLA_HEADER #define COLLA_HEADER +#define _FILE_OFFSET_BITS 1 + #include #include #include +#include +#include // CORE MODULES ///////////////////////////////// @@ -91,7 +95,7 @@ void colla_cleanup(void); #define COLLA_CMT_LIB 0 #endif -#if COLLA_TCC +#if COLLA_TCC || COLLA_CLANG #define alignof __alignof__ #endif @@ -309,7 +313,7 @@ struct str_t { typedef struct str16_t str16_t; struct str16_t { - u16 *buf; + char16_t *buf; usize len; }; @@ -354,11 +358,12 @@ str_t str_fmt(arena_t *arena, const char *fmt, ...); str_t str_fmtv(arena_t *arena, const char *fmt, va_list args); tstr_t tstr_init(TCHAR *str, usize optional_len); -str16_t str16_init(u16 *str, usize optional_len); +str16_t str16_init(char16_t *str, usize optional_len); str_t str_from_str16(arena_t *arena, str16_t src); str_t str_from_tstr(arena_t *arena, tstr_t src); str16_t str16_from_str(arena_t *arena, str_t src); +usize str16_len(char16_t *str); bool str_equals(str_t a, str_t b); int str_compare(str_t a, str_t b); @@ -743,8 +748,8 @@ oshandle_t os_stdin(void); // == FILE ====================================== typedef enum filemode_e { - FILEMODE_READ = 1 << 0, - FILEMODE_WRITE = 1 << 1, + OS_FILE_READ = 1 << 0, + OS_FILE_WRITE = 1 << 1, } filemode_e; bool os_file_exists(strview_t filename); diff --git a/stb/stb_sprintf.h b/stb/stb_sprintf.h new file mode 100644 index 0000000..60a81b2 --- /dev/null +++ b/stb/stb_sprintf.h @@ -0,0 +1,1945 @@ +// stb_sprintf - v1.10 - public domain snprintf() implementation +// originally by Jeff Roberts / RAD Game Tools, 2015/10/20 +// http://github.com/nothings/stb +// +// allowed types: sc uidBboXx p AaGgEef n +// lengths : hh h ll j z t I64 I32 I +// +// Contributors: +// Fabian "ryg" Giesen (reformatting) +// github:aganm (attribute format) +// +// Contributors (bugfixes): +// github:d26435 +// github:trex78 +// github:account-login +// Jari Komppa (SI suffixes) +// Rohit Nirmal +// Marcin Wojdyr +// Leonard Ritter +// Stefano Zanotti +// Adam Allison +// Arvid Gerstmann +// Markus Kolb +// +// LICENSE: +// +// See end of file for license information. + +#define STB_SPRINTF_IMPLEMENTATION + +#ifndef STB_SPRINTF_H_INCLUDE +#define STB_SPRINTF_H_INCLUDE + +/* +Single file sprintf replacement. + +Originally written by Jeff Roberts at RAD Game Tools - 2015/10/20. +Hereby placed in public domain. + +This is a full sprintf replacement that supports everything that +the C runtime sprintfs support, including float/double, 64-bit integers, +hex floats, field parameters (%*.*d stuff), length reads backs, etc. + +Why would you need this if sprintf already exists? Well, first off, +it's *much* faster (see below). It's also much smaller than the CRT +versions code-space-wise. We've also added some simple improvements +that are super handy (commas in thousands, callbacks at buffer full, +for example). Finally, the format strings for MSVC and GCC differ +for 64-bit integers (among other small things), so this lets you use +the same format strings in cross platform code. + +It uses the standard single file trick of being both the header file +and the source itself. If you just include it normally, you just get +the header file function definitions. To get the code, you include +it from a C or C++ file and define STB_SPRINTF_IMPLEMENTATION first. + +It only uses va_args macros from the C runtime to do it's work. It +does cast doubles to S64s and shifts and divides U64s, which does +drag in CRT code on most platforms. + +It compiles to roughly 8K with float support, and 4K without. +As a comparison, when using MSVC static libs, calling sprintf drags +in 16K. + +API: +==== +int stbsp_sprintf( char * buf, char const * fmt, ... ) +int stbsp_snprintf( char * buf, int count, char const * fmt, ... ) + Convert an arg list into a buffer. stbsp_snprintf always returns + a zero-terminated string (unlike regular snprintf). + +int stbsp_vsprintf( char * buf, char const * fmt, va_list va ) +int stbsp_vsnprintf( char * buf, int count, char const * fmt, va_list va ) + Convert a va_list arg list into a buffer. stbsp_vsnprintf always returns + a zero-terminated string (unlike regular snprintf). + +int stbsp_vsprintfcb( STBSP_SPRINTFCB * callback, void * user, char * buf, char const * fmt, va_list va ) + typedef char * STBSP_SPRINTFCB( char const * buf, void * user, int len ); + Convert into a buffer, calling back every STB_SPRINTF_MIN chars. + Your callback can then copy the chars out, print them or whatever. + This function is actually the workhorse for everything else. + The buffer you pass in must hold at least STB_SPRINTF_MIN characters. + // you return the next buffer to use or 0 to stop converting + +void stbsp_set_separators( char comma, char period ) + Set the comma and period characters to use. + +FLOATS/DOUBLES: +=============== +This code uses a internal float->ascii conversion method that uses +doubles with error correction (double-doubles, for ~105 bits of +precision). This conversion is round-trip perfect - that is, an atof +of the values output here will give you the bit-exact double back. + +One difference is that our insignificant digits will be different than +with MSVC or GCC (but they don't match each other either). We also +don't attempt to find the minimum length matching float (pre-MSVC15 +doesn't either). + +If you don't need float or doubles at all, define STB_SPRINTF_NOFLOAT +and you'll save 4K of code space. + +64-BIT INTS: +============ +This library also supports 64-bit integers and you can use MSVC style or +GCC style indicators (%I64d or %lld). It supports the C99 specifiers +for size_t and ptr_diff_t (%jd %zd) as well. + +EXTRAS: +======= +Like some GCCs, for integers and floats, you can use a ' (single quote) +specifier and commas will be inserted on the thousands: "%'d" on 12345 +would print 12,345. + +For integers and floats, you can use a "$" specifier and the number +will be converted to float and then divided to get kilo, mega, giga or +tera and then printed, so "%$d" 1000 is "1.0 k", "%$.2d" 2536000 is +"2.53 M", etc. For byte values, use two $:s, like "%$$d" to turn +2536000 to "2.42 Mi". If you prefer JEDEC suffixes to SI ones, use three +$:s: "%$$$d" -> "2.42 M". To remove the space between the number and the +suffix, add "_" specifier: "%_$d" -> "2.53M". + +In addition to octal and hexadecimal conversions, you can print +integers in binary: "%b" for 256 would print 100. + +PERFORMANCE vs MSVC 2008 32-/64-bit (GCC is even slower than MSVC): +=================================================================== +"%d" across all 32-bit ints (4.8x/4.0x faster than 32-/64-bit MSVC) +"%24d" across all 32-bit ints (4.5x/4.2x faster) +"%x" across all 32-bit ints (4.5x/3.8x faster) +"%08x" across all 32-bit ints (4.3x/3.8x faster) +"%f" across e-10 to e+10 floats (7.3x/6.0x faster) +"%e" across e-10 to e+10 floats (8.1x/6.0x faster) +"%g" across e-10 to e+10 floats (10.0x/7.1x faster) +"%f" for values near e-300 (7.9x/6.5x faster) +"%f" for values near e+300 (10.0x/9.1x faster) +"%e" for values near e-300 (10.1x/7.0x faster) +"%e" for values near e+300 (9.2x/6.0x faster) +"%.320f" for values near e-300 (12.6x/11.2x faster) +"%a" for random values (8.6x/4.3x faster) +"%I64d" for 64-bits with 32-bit values (4.8x/3.4x faster) +"%I64d" for 64-bits > 32-bit values (4.9x/5.5x faster) +"%s%s%s" for 64 char strings (7.1x/7.3x faster) +"...512 char string..." ( 35.0x/32.5x faster!) +*/ + +#if defined(__clang__) + #if defined(__has_feature) && defined(__has_attribute) + #if __has_feature(address_sanitizer) + #if __has_attribute(__no_sanitize__) + #define STBSP__ASAN __attribute__((__no_sanitize__("address"))) + #elif __has_attribute(__no_sanitize_address__) + #define STBSP__ASAN __attribute__((__no_sanitize_address__)) + #elif __has_attribute(__no_address_safety_analysis__) + #define STBSP__ASAN __attribute__((__no_address_safety_analysis__)) + #endif + #endif + #endif +#elif defined(__GNUC__) && (__GNUC__ >= 5 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 8)) + #if defined(__SANITIZE_ADDRESS__) && __SANITIZE_ADDRESS__ + #define STBSP__ASAN __attribute__((__no_sanitize_address__)) + #endif +#endif + +#ifndef STBSP__ASAN +#define STBSP__ASAN +#endif + +#ifdef STB_SPRINTF_STATIC +#define STBSP__PUBLICDEC static +#define STBSP__PUBLICDEF static STBSP__ASAN +#else +#ifdef __cplusplus +#define STBSP__PUBLICDEC extern "C" +#define STBSP__PUBLICDEF extern "C" STBSP__ASAN +#else +#define STBSP__PUBLICDEC extern +#define STBSP__PUBLICDEF STBSP__ASAN +#endif +#endif + +#if defined(__has_attribute) + #if __has_attribute(format) + #define STBSP__ATTRIBUTE_FORMAT(fmt,va) __attribute__((format(printf,fmt,va))) + #endif +#endif + +#ifndef STBSP__ATTRIBUTE_FORMAT +#define STBSP__ATTRIBUTE_FORMAT(fmt,va) +#endif + +#ifdef _MSC_VER +#define STBSP__NOTUSED(v) (void)(v) +#else +#define STBSP__NOTUSED(v) (void)sizeof(v) +#endif + +#include // for va_arg(), va_list() +#include // size_t, ptrdiff_t + +#ifndef STB_SPRINTF_MIN +#define STB_SPRINTF_MIN 512 // how many characters per callback +#endif +typedef char *STBSP_SPRINTFCB(const char *buf, void *user, int len); + +#ifndef STB_SPRINTF_DECORATE +#define STB_SPRINTF_DECORATE(name) stbsp_##name // define this before including if you want to change the names +#endif + +STBSP__PUBLICDEC int STB_SPRINTF_DECORATE(vsprintf)(char *buf, char const *fmt, va_list va); +STBSP__PUBLICDEC int STB_SPRINTF_DECORATE(vsnprintf)(char *buf, int count, char const *fmt, va_list va); +STBSP__PUBLICDEC int STB_SPRINTF_DECORATE(sprintf)(char *buf, char const *fmt, ...) STBSP__ATTRIBUTE_FORMAT(2,3); +STBSP__PUBLICDEC int STB_SPRINTF_DECORATE(snprintf)(char *buf, int count, char const *fmt, ...) STBSP__ATTRIBUTE_FORMAT(3,4); + +STBSP__PUBLICDEC int STB_SPRINTF_DECORATE(vsprintfcb)(STBSP_SPRINTFCB *callback, void *user, char *buf, char const *fmt, va_list va); +STBSP__PUBLICDEC void STB_SPRINTF_DECORATE(set_separators)(char comma, char period); + +#endif // STB_SPRINTF_H_INCLUDE + +#ifdef STB_SPRINTF_IMPLEMENTATION + +#if COLLA_CLANG + +#pragma clang diagnostic push + +#pragma clang diagnostic ignored "-Wextra-semi-stmt" +#pragma clang diagnostic ignored "-Wconditional-uninitialized" +#pragma clang diagnostic ignored "-Wcast-qual" +#pragma clang diagnostic ignored "-Wimplicit-fallthrough" + +#endif + +#define stbsp__uint32 unsigned int +#define stbsp__int32 signed int + +#ifdef _MSC_VER +#define stbsp__uint64 unsigned __int64 +#define stbsp__int64 signed __int64 +#else +#define stbsp__uint64 unsigned long long +#define stbsp__int64 signed long long +#endif +#define stbsp__uint16 unsigned short + +#ifndef stbsp__uintptr +#if defined(__ppc64__) || defined(__powerpc64__) || defined(__aarch64__) || defined(_M_X64) || defined(__x86_64__) || defined(__x86_64) || defined(__s390x__) +#define stbsp__uintptr stbsp__uint64 +#else +#define stbsp__uintptr stbsp__uint32 +#endif +#endif + +#ifndef STB_SPRINTF_MSVC_MODE // used for MSVC2013 and earlier (MSVC2015 matches GCC) +#if defined(_MSC_VER) && (_MSC_VER < 1900) +#define STB_SPRINTF_MSVC_MODE +#endif +#endif + +#ifdef STB_SPRINTF_NOUNALIGNED // define this before inclusion to force stbsp_sprintf to always use aligned accesses +#define STBSP__UNALIGNED(code) +#else +#define STBSP__UNALIGNED(code) code +#endif + +#ifndef STB_SPRINTF_NOFLOAT +// internal float utility functions +static stbsp__int32 stbsp__real_to_str(char const **start, stbsp__uint32 *len, char *out, stbsp__int32 *decimal_pos, double value, stbsp__uint32 frac_digits); +static stbsp__int32 stbsp__real_to_parts(stbsp__int64 *bits, stbsp__int32 *expo, double value); +#define STBSP__SPECIAL 0x7000 +#endif + +static char stbsp__period = '.'; +static char stbsp__comma = ','; +static struct +{ + short temp; // force next field to be 2-byte aligned + char pair[201]; +} stbsp__digitpair = +{ + 0, + "00010203040506070809101112131415161718192021222324" + "25262728293031323334353637383940414243444546474849" + "50515253545556575859606162636465666768697071727374" + "75767778798081828384858687888990919293949596979899" +}; + +STBSP__PUBLICDEF void STB_SPRINTF_DECORATE(set_separators)(char pcomma, char pperiod) +{ + stbsp__period = pperiod; + stbsp__comma = pcomma; +} + +#define STBSP__LEFTJUST 1 +#define STBSP__LEADINGPLUS 2 +#define STBSP__LEADINGSPACE 4 +#define STBSP__LEADING_0X 8 +#define STBSP__LEADINGZERO 16 +#define STBSP__INTMAX 32 +#define STBSP__TRIPLET_COMMA 64 +#define STBSP__NEGATIVE 128 +#define STBSP__METRIC_SUFFIX 256 +#define STBSP__HALFWIDTH 512 +#define STBSP__METRIC_NOSPACE 1024 +#define STBSP__METRIC_1024 2048 +#define STBSP__METRIC_JEDEC 4096 + +static void stbsp__lead_sign(stbsp__uint32 fl, char *sign) +{ + sign[0] = 0; + if (fl & STBSP__NEGATIVE) { + sign[0] = 1; + sign[1] = '-'; + } else if (fl & STBSP__LEADINGSPACE) { + sign[0] = 1; + sign[1] = ' '; + } else if (fl & STBSP__LEADINGPLUS) { + sign[0] = 1; + sign[1] = '+'; + } +} + +static STBSP__ASAN stbsp__uint32 stbsp__strlen_limited(char const *s, stbsp__uint32 limit) +{ + char const * sn = s; + + // get up to 4-byte alignment + for (;;) { + if (((stbsp__uintptr)sn & 3) == 0) + break; + + if (!limit || *sn == 0) + return (stbsp__uint32)(sn - s); + + ++sn; + --limit; + } + + // scan over 4 bytes at a time to find terminating 0 + // this will intentionally scan up to 3 bytes past the end of buffers, + // but becase it works 4B aligned, it will never cross page boundaries + // (hence the STBSP__ASAN markup; the over-read here is intentional + // and harmless) + while (limit >= 4) { + stbsp__uint32 v = *(stbsp__uint32 *)sn; + // bit hack to find if there's a 0 byte in there + if ((v - 0x01010101) & (~v) & 0x80808080UL) + break; + + sn += 4; + limit -= 4; + } + + // handle the last few characters to find actual size + while (limit && *sn) { + ++sn; + --limit; + } + + return (stbsp__uint32)(sn - s); +} + +typedef struct { + char *buf; + size_t len; +} stb__strv_t; + +STBSP__PUBLICDEF int STB_SPRINTF_DECORATE(vsprintfcb)(STBSP_SPRINTFCB *callback, void *user, char *buf, char const *fmt, va_list va) +{ + static char hex[] = "0123456789abcdefxp"; + static char hexu[] = "0123456789ABCDEFXP"; + char *bf; + char const *f; + int tlen = 0; + + bf = buf; + f = fmt; + for (;;) { + stbsp__int32 fw, pr, tz; + stbsp__uint32 fl; + + // macros for the callback buffer stuff + #define stbsp__chk_cb_bufL(bytes) \ + { \ + int len = (int)(bf - buf); \ + if ((len + (bytes)) >= STB_SPRINTF_MIN) { \ + tlen += len; \ + if (0 == (bf = buf = callback(buf, user, len))) \ + goto done; \ + } \ + } + #define stbsp__chk_cb_buf(bytes) \ + { \ + if (callback) { \ + stbsp__chk_cb_bufL(bytes); \ + } \ + } + #define stbsp__flush_cb() \ + { \ + stbsp__chk_cb_bufL(STB_SPRINTF_MIN - 1); \ + } // flush if there is even one byte in the buffer + #define stbsp__cb_buf_clamp(cl, v) \ + cl = v; \ + if (callback) { \ + int lg = STB_SPRINTF_MIN - (int)(bf - buf); \ + if (cl > lg) \ + cl = lg; \ + } + + // fast copy everything up to the next % (or end of string) + for (;;) { + while (((stbsp__uintptr)f) & 3) { + schk1: + if (f[0] == '%') + goto scandd; + schk2: + if (f[0] == 0) + goto endfmt; + stbsp__chk_cb_buf(1); + *bf++ = f[0]; + ++f; + } + for (;;) { + // Check if the next 4 bytes contain %(0x25) or end of string. + // Using the 'hasless' trick: + // https://graphics.stanford.edu/~seander/bithacks.html#HasLessInWord + stbsp__uint32 v, c; + v = *(stbsp__uint32 *)f; + c = (~v) & 0x80808080; + if (((v ^ 0x25252525) - 0x01010101) & c) + goto schk1; + if ((v - 0x01010101) & c) + goto schk2; + if (callback) + if ((STB_SPRINTF_MIN - (int)(bf - buf)) < 4) + goto schk1; + #ifdef STB_SPRINTF_NOUNALIGNED + if(((stbsp__uintptr)bf) & 3) { + bf[0] = f[0]; + bf[1] = f[1]; + bf[2] = f[2]; + bf[3] = f[3]; + } else + #endif + { + *(stbsp__uint32 *)bf = v; + } + bf += 4; + f += 4; + } + } + scandd: + + ++f; + + // ok, we have a percent, read the modifiers first + fw = 0; + pr = -1; + fl = 0; + tz = 0; + + // flags + for (;;) { + switch (f[0]) { + // if we have left justify + case '-': + fl |= STBSP__LEFTJUST; + ++f; + continue; + // if we have leading plus + case '+': + fl |= STBSP__LEADINGPLUS; + ++f; + continue; + // if we have leading space + case ' ': + fl |= STBSP__LEADINGSPACE; + ++f; + continue; + // if we have leading 0x + case '#': + fl |= STBSP__LEADING_0X; + ++f; + continue; + // if we have thousand commas + case '\'': + fl |= STBSP__TRIPLET_COMMA; + ++f; + continue; + // if we have kilo marker (none->kilo->kibi->jedec) + case '$': + if (fl & STBSP__METRIC_SUFFIX) { + if (fl & STBSP__METRIC_1024) { + fl |= STBSP__METRIC_JEDEC; + } else { + fl |= STBSP__METRIC_1024; + } + } else { + fl |= STBSP__METRIC_SUFFIX; + } + ++f; + continue; + // if we don't want space between metric suffix and number + case '_': + fl |= STBSP__METRIC_NOSPACE; + ++f; + continue; + // if we have leading zero + case '0': + fl |= STBSP__LEADINGZERO; + ++f; + goto flags_done; + default: goto flags_done; + } + } + flags_done: + + // get the field width + if (f[0] == '*') { + fw = va_arg(va, stbsp__uint32); + ++f; + } else { + while ((f[0] >= '0') && (f[0] <= '9')) { + fw = fw * 10 + f[0] - '0'; + f++; + } + } + // get the precision + if (f[0] == '.') { + ++f; + if (f[0] == '*') { + pr = va_arg(va, stbsp__uint32); + ++f; + } else { + pr = 0; + while ((f[0] >= '0') && (f[0] <= '9')) { + pr = pr * 10 + f[0] - '0'; + f++; + } + } + } + + // handle integer size overrides + switch (f[0]) { + // are we halfwidth? + case 'h': + fl |= STBSP__HALFWIDTH; + ++f; + if (f[0] == 'h') + ++f; // QUARTERWIDTH + break; + // are we 64-bit (unix style) + case 'l': + fl |= ((sizeof(long) == 8) ? STBSP__INTMAX : 0); + ++f; + if (f[0] == 'l') { + fl |= STBSP__INTMAX; + ++f; + } + break; + // are we 64-bit on intmax? (c99) + case 'j': + fl |= (sizeof(size_t) == 8) ? STBSP__INTMAX : 0; + ++f; + break; + // are we 64-bit on size_t or ptrdiff_t? (c99) + case 'z': + fl |= (sizeof(ptrdiff_t) == 8) ? STBSP__INTMAX : 0; + ++f; + break; + case 't': + fl |= (sizeof(ptrdiff_t) == 8) ? STBSP__INTMAX : 0; + ++f; + break; + // are we 64-bit (msft style) + case 'I': + if ((f[1] == '6') && (f[2] == '4')) { + fl |= STBSP__INTMAX; + f += 3; + } else if ((f[1] == '3') && (f[2] == '2')) { + f += 3; + } else { + fl |= ((sizeof(void *) == 8) ? STBSP__INTMAX : 0); + ++f; + } + break; + default: break; + } + + // handle each replacement + switch (f[0]) { + #define STBSP__NUMSZ 512 // big enough for e308 (with commas) or e-307 + char num[STBSP__NUMSZ]; + char lead[8]; + char tail[8]; + char *s; + char const *h; + stbsp__uint32 l, n, cs; + stbsp__uint64 n64; +#ifndef STB_SPRINTF_NOFLOAT + double fv; +#endif + stbsp__int32 dp; + char const *sn; + + stb__strv_t strv; + + case 's': + // get the string + s = va_arg(va, char *); + if (s == 0) + s = (char *)"null"; + // get the length, limited to desired precision + // always limit to ~0u chars since our counts are 32b + l = stbsp__strlen_limited(s, (pr >= 0) ? pr : ~0u); + lead[0] = 0; + tail[0] = 0; + pr = 0; + dp = 0; + cs = 0; + // copy the string in + goto scopy; + + case 'v': + // get the view + strv = va_arg(va, stb__strv_t); + s = strv.buf; + if (s == 0) + s = (char *)"null"; + l = (unsigned int)strv.len; + lead[0] = 0; + tail[0] = 0; + pr = 0; + dp = 0; + cs = 0; + // copy the string in + goto scopy; + + case 'c': // char + // get the character + s = num + STBSP__NUMSZ - 1; + *s = (char)va_arg(va, int); + l = 1; + lead[0] = 0; + tail[0] = 0; + pr = 0; + dp = 0; + cs = 0; + goto scopy; + + case 'n': // weird write-bytes specifier + { + int *d = va_arg(va, int *); + *d = tlen + (int)(bf - buf); + } break; + +#ifdef STB_SPRINTF_NOFLOAT + case 'A': // float + case 'a': // hex float + case 'G': // float + case 'g': // float + case 'E': // float + case 'e': // float + case 'f': // float + va_arg(va, double); // eat it + s = (char *)"No float"; + l = 8; + lead[0] = 0; + tail[0] = 0; + pr = 0; + cs = 0; + STBSP__NOTUSED(dp); + goto scopy; +#else + case 'A': // hex float + case 'a': // hex float + h = (f[0] == 'A') ? hexu : hex; + fv = va_arg(va, double); + if (pr == -1) + pr = 6; // default is 6 + // read the double into a string + if (stbsp__real_to_parts((stbsp__int64 *)&n64, &dp, fv)) + fl |= STBSP__NEGATIVE; + + s = num + 64; + + stbsp__lead_sign(fl, lead); + + if (dp == -1023) + dp = (n64) ? -1022 : 0; + else + n64 |= (((stbsp__uint64)1) << 52); + n64 <<= (64 - 56); + if (pr < 15) + n64 += ((((stbsp__uint64)8) << 56) >> (pr * 4)); +// add leading chars + +#ifdef STB_SPRINTF_MSVC_MODE + *s++ = '0'; + *s++ = 'x'; +#else + lead[1 + lead[0]] = '0'; + lead[2 + lead[0]] = 'x'; + lead[0] += 2; +#endif + *s++ = h[(n64 >> 60) & 15]; + n64 <<= 4; + if (pr) + *s++ = stbsp__period; + sn = s; + + // print the bits + n = pr; + if (n > 13) + n = 13; + if (pr > (stbsp__int32)n) + tz = pr - n; + pr = 0; + while (n--) { + *s++ = h[(n64 >> 60) & 15]; + n64 <<= 4; + } + + // print the expo + tail[1] = h[17]; + if (dp < 0) { + tail[2] = '-'; + dp = -dp; + } else + tail[2] = '+'; + n = (dp >= 1000) ? 6 : ((dp >= 100) ? 5 : ((dp >= 10) ? 4 : 3)); + tail[0] = (char)n; + for (;;) { + tail[n] = '0' + dp % 10; + if (n <= 3) + break; + --n; + dp /= 10; + } + + dp = (int)(s - sn); + l = (int)(s - (num + 64)); + s = num + 64; + cs = 1 + (3 << 24); + goto scopy; + + case 'G': // float + case 'g': // float + h = (f[0] == 'G') ? hexu : hex; + fv = va_arg(va, double); + if (pr == -1) + pr = 6; + else if (pr == 0) + pr = 1; // default is 6 + // read the double into a string + if (stbsp__real_to_str(&sn, &l, num, &dp, fv, (pr - 1) | 0x80000000)) + fl |= STBSP__NEGATIVE; + + // clamp the precision and delete extra zeros after clamp + n = pr; + if (l > (stbsp__uint32)pr) + l = pr; + while ((l > 1) && (pr) && (sn[l - 1] == '0')) { + --pr; + --l; + } + + // should we use %e + if ((dp <= -4) || (dp > (stbsp__int32)n)) { + if (pr > (stbsp__int32)l) + pr = l - 1; + else if (pr) + --pr; // when using %e, there is one digit before the decimal + goto doexpfromg; + } + // this is the insane action to get the pr to match %g semantics for %f + if (dp > 0) { + pr = (dp < (stbsp__int32)l) ? l - dp : 0; + } else { + pr = -dp + ((pr > (stbsp__int32)l) ? (stbsp__int32) l : pr); + } + goto dofloatfromg; + + case 'E': // float + case 'e': // float + h = (f[0] == 'E') ? hexu : hex; + fv = va_arg(va, double); + if (pr == -1) + pr = 6; // default is 6 + // read the double into a string + if (stbsp__real_to_str(&sn, &l, num, &dp, fv, pr | 0x80000000)) + fl |= STBSP__NEGATIVE; + doexpfromg: + tail[0] = 0; + stbsp__lead_sign(fl, lead); + if (dp == STBSP__SPECIAL) { + s = (char *)sn; + cs = 0; + pr = 0; + goto scopy; + } + s = num + 64; + // handle leading chars + *s++ = sn[0]; + + if (pr) + *s++ = stbsp__period; + + // handle after decimal + if ((l - 1) > (stbsp__uint32)pr) + l = pr + 1; + for (n = 1; n < l; n++) + *s++ = sn[n]; + // trailing zeros + tz = pr - (l - 1); + pr = 0; + // dump expo + tail[1] = h[0xe]; + dp -= 1; + if (dp < 0) { + tail[2] = '-'; + dp = -dp; + } else + tail[2] = '+'; +#ifdef STB_SPRINTF_MSVC_MODE + n = 5; +#else + n = (dp >= 100) ? 5 : 4; +#endif + tail[0] = (char)n; + for (;;) { + tail[n] = '0' + dp % 10; + if (n <= 3) + break; + --n; + dp /= 10; + } + cs = 1 + (3 << 24); // how many tens + goto flt_lead; + + case 'f': // float + fv = va_arg(va, double); + doafloat: + // do kilos + if (fl & STBSP__METRIC_SUFFIX) { + double divisor; + divisor = 1000.0f; + if (fl & STBSP__METRIC_1024) + divisor = 1024.0; + while (fl < 0x4000000) { + if ((fv < divisor) && (fv > -divisor)) + break; + fv /= divisor; + fl += 0x1000000; + } + } + if (pr == -1) + pr = 6; // default is 6 + // read the double into a string + if (stbsp__real_to_str(&sn, &l, num, &dp, fv, pr)) + fl |= STBSP__NEGATIVE; + dofloatfromg: + tail[0] = 0; + stbsp__lead_sign(fl, lead); + if (dp == STBSP__SPECIAL) { + s = (char *)sn; + cs = 0; + pr = 0; + goto scopy; + } + s = num + 64; + + // handle the three decimal varieties + if (dp <= 0) { + stbsp__int32 i; + // handle 0.000*000xxxx + *s++ = '0'; + if (pr) + *s++ = stbsp__period; + n = -dp; + if ((stbsp__int32)n > pr) + n = pr; + i = n; + while (i) { + if ((((stbsp__uintptr)s) & 3) == 0) + break; + *s++ = '0'; + --i; + } + while (i >= 4) { + *(stbsp__uint32 *)s = 0x30303030; + s += 4; + i -= 4; + } + while (i) { + *s++ = '0'; + --i; + } + if ((stbsp__int32)(l + n) > pr) + l = pr - n; + i = l; + while (i) { + *s++ = *sn++; + --i; + } + tz = pr - (n + l); + cs = 1 + (3 << 24); // how many tens did we write (for commas below) + } else { + cs = (fl & STBSP__TRIPLET_COMMA) ? ((600 - (stbsp__uint32)dp) % 3) : 0; + if ((stbsp__uint32)dp >= l) { + // handle xxxx000*000.0 + n = 0; + for (;;) { + if ((fl & STBSP__TRIPLET_COMMA) && (++cs == 4)) { + cs = 0; + *s++ = stbsp__comma; + } else { + *s++ = sn[n]; + ++n; + if (n >= l) + break; + } + } + if (n < (stbsp__uint32)dp) { + n = dp - n; + if ((fl & STBSP__TRIPLET_COMMA) == 0) { + while (n) { + if ((((stbsp__uintptr)s) & 3) == 0) + break; + *s++ = '0'; + --n; + } + while (n >= 4) { + *(stbsp__uint32 *)s = 0x30303030; + s += 4; + n -= 4; + } + } + while (n) { + if ((fl & STBSP__TRIPLET_COMMA) && (++cs == 4)) { + cs = 0; + *s++ = stbsp__comma; + } else { + *s++ = '0'; + --n; + } + } + } + cs = (int)(s - (num + 64)) + (3 << 24); // cs is how many tens + if (pr) { + *s++ = stbsp__period; + tz = pr; + } + } else { + // handle xxxxx.xxxx000*000 + n = 0; + for (;;) { + if ((fl & STBSP__TRIPLET_COMMA) && (++cs == 4)) { + cs = 0; + *s++ = stbsp__comma; + } else { + *s++ = sn[n]; + ++n; + if (n >= (stbsp__uint32)dp) + break; + } + } + cs = (int)(s - (num + 64)) + (3 << 24); // cs is how many tens + if (pr) + *s++ = stbsp__period; + if ((l - dp) > (stbsp__uint32)pr) + l = pr + dp; + while (n < l) { + *s++ = sn[n]; + ++n; + } + tz = pr - (l - dp); + } + } + pr = 0; + + // handle k,m,g,t + if (fl & STBSP__METRIC_SUFFIX) { + char idx; + idx = 1; + if (fl & STBSP__METRIC_NOSPACE) + idx = 0; + tail[0] = idx; + tail[1] = ' '; + { + if (fl >> 24) { // SI kilo is 'k', JEDEC and SI kibits are 'K'. + if (fl & STBSP__METRIC_1024) + tail[idx + 1] = "_KMGT"[fl >> 24]; + else + tail[idx + 1] = "_kMGT"[fl >> 24]; + idx++; + // If printing kibits and not in jedec, add the 'i'. + if (fl & STBSP__METRIC_1024 && !(fl & STBSP__METRIC_JEDEC)) { + tail[idx + 1] = 'i'; + idx++; + } + tail[0] = idx; + } + } + }; + + flt_lead: + // get the length that we copied + l = (stbsp__uint32)(s - (num + 64)); + s = num + 64; + goto scopy; +#endif + + case 'B': // upper binary + case 'b': // lower binary + h = (f[0] == 'B') ? hexu : hex; + lead[0] = 0; + if (fl & STBSP__LEADING_0X) { + lead[0] = 2; + lead[1] = '0'; + lead[2] = h[0xb]; + } + l = (8 << 4) | (1 << 8); + goto radixnum; + + case 'o': // octal + h = hexu; + lead[0] = 0; + if (fl & STBSP__LEADING_0X) { + lead[0] = 1; + lead[1] = '0'; + } + l = (3 << 4) | (3 << 8); + goto radixnum; + + case 'p': // pointer + fl |= (sizeof(void *) == 8) ? STBSP__INTMAX : 0; + pr = sizeof(void *) * 2; + fl &= ~STBSP__LEADINGZERO; // 'p' only prints the pointer with zeros + // fall through - to X + + case 'X': // upper hex + case 'x': // lower hex + h = (f[0] == 'X') ? hexu : hex; + l = (4 << 4) | (4 << 8); + lead[0] = 0; + if (fl & STBSP__LEADING_0X) { + lead[0] = 2; + lead[1] = '0'; + lead[2] = h[16]; + } + radixnum: + // get the number + if (fl & STBSP__INTMAX) + n64 = va_arg(va, stbsp__uint64); + else + n64 = va_arg(va, stbsp__uint32); + + s = num + STBSP__NUMSZ; + dp = 0; + // clear tail, and clear leading if value is zero + tail[0] = 0; + if (n64 == 0) { + lead[0] = 0; + if (pr == 0) { + l = 0; + cs = 0; + goto scopy; + } + } + // convert to string + for (;;) { + *--s = h[n64 & ((1 << (l >> 8)) - 1)]; + n64 >>= (l >> 8); + if (!((n64) || ((stbsp__int32)((num + STBSP__NUMSZ) - s) < pr))) + break; + if (fl & STBSP__TRIPLET_COMMA) { + ++l; + if ((l & 15) == ((l >> 4) & 15)) { + l &= ~15; + *--s = stbsp__comma; + } + } + }; + // get the tens and the comma pos + cs = (stbsp__uint32)((num + STBSP__NUMSZ) - s) + ((((l >> 4) & 15)) << 24); + // get the length that we copied + l = (stbsp__uint32)((num + STBSP__NUMSZ) - s); + // copy it + goto scopy; + + case 'u': // unsigned + case 'i': + case 'd': // integer + // get the integer and abs it + if (fl & STBSP__INTMAX) { + stbsp__int64 i64 = va_arg(va, stbsp__int64); + n64 = (stbsp__uint64)i64; + if ((f[0] != 'u') && (i64 < 0)) { + n64 = (stbsp__uint64)-i64; + fl |= STBSP__NEGATIVE; + } + } else { + stbsp__int32 i = va_arg(va, stbsp__int32); + n64 = (stbsp__uint32)i; + if ((f[0] != 'u') && (i < 0)) { + n64 = (stbsp__uint32)-i; + fl |= STBSP__NEGATIVE; + } + } + +#ifndef STB_SPRINTF_NOFLOAT + if (fl & STBSP__METRIC_SUFFIX) { + if (n64 < 1024) + pr = 0; + else if (pr == -1) + pr = 1; + fv = (double)(stbsp__int64)n64; + goto doafloat; + } +#endif + + // convert to string + s = num + STBSP__NUMSZ; + l = 0; + + for (;;) { + // do in 32-bit chunks (avoid lots of 64-bit divides even with constant denominators) + char *o = s - 8; + if (n64 >= 100000000) { + n = (stbsp__uint32)(n64 % 100000000); + n64 /= 100000000; + } else { + n = (stbsp__uint32)n64; + n64 = 0; + } + if ((fl & STBSP__TRIPLET_COMMA) == 0) { + do { + s -= 2; + *(stbsp__uint16 *)s = *(stbsp__uint16 *)&stbsp__digitpair.pair[(n % 100) * 2]; + n /= 100; + } while (n); + } + while (n) { + if ((fl & STBSP__TRIPLET_COMMA) && (l++ == 3)) { + l = 0; + *--s = stbsp__comma; + --o; + } else { + *--s = (char)(n % 10) + '0'; + n /= 10; + } + } + if (n64 == 0) { + if ((s[0] == '0') && (s != (num + STBSP__NUMSZ))) + ++s; + break; + } + while (s != o) + if ((fl & STBSP__TRIPLET_COMMA) && (l++ == 3)) { + l = 0; + *--s = stbsp__comma; + --o; + } else { + *--s = '0'; + } + } + + tail[0] = 0; + stbsp__lead_sign(fl, lead); + + // get the length that we copied + l = (stbsp__uint32)((num + STBSP__NUMSZ) - s); + if (l == 0) { + *--s = '0'; + l = 1; + } + cs = l + (3 << 24); + if (pr < 0) + pr = 0; + + scopy: + // get fw=leading/trailing space, pr=leading zeros + if (pr < (stbsp__int32)l) + pr = l; + n = pr + lead[0] + tail[0] + tz; + if (fw < (stbsp__int32)n) + fw = n; + fw -= n; + pr -= l; + + // handle right justify and leading zeros + if ((fl & STBSP__LEFTJUST) == 0) { + if (fl & STBSP__LEADINGZERO) // if leading zeros, everything is in pr + { + pr = (fw > pr) ? fw : pr; + fw = 0; + } else { + fl &= ~STBSP__TRIPLET_COMMA; // if no leading zeros, then no commas + } + } + + // copy the spaces and/or zeros + if (fw + pr) { + stbsp__int32 i; + stbsp__uint32 c; + + // copy leading spaces (or when doing %8.4d stuff) + if ((fl & STBSP__LEFTJUST) == 0) + while (fw > 0) { + stbsp__cb_buf_clamp(i, fw); + fw -= i; + while (i) { + if ((((stbsp__uintptr)bf) & 3) == 0) + break; + *bf++ = ' '; + --i; + } + while (i >= 4) { + *(stbsp__uint32 *)bf = 0x20202020; + bf += 4; + i -= 4; + } + while (i) { + *bf++ = ' '; + --i; + } + stbsp__chk_cb_buf(1); + } + + // copy leader + sn = lead + 1; + while (lead[0]) { + stbsp__cb_buf_clamp(i, lead[0]); + lead[0] -= (char)i; + while (i) { + *bf++ = *sn++; + --i; + } + stbsp__chk_cb_buf(1); + } + + // copy leading zeros + c = cs >> 24; + cs &= 0xffffff; + cs = (fl & STBSP__TRIPLET_COMMA) ? ((stbsp__uint32)(c - ((pr + cs) % (c + 1)))) : 0; + while (pr > 0) { + stbsp__cb_buf_clamp(i, pr); + pr -= i; + if ((fl & STBSP__TRIPLET_COMMA) == 0) { + while (i) { + if ((((stbsp__uintptr)bf) & 3) == 0) + break; + *bf++ = '0'; + --i; + } + while (i >= 4) { + *(stbsp__uint32 *)bf = 0x30303030; + bf += 4; + i -= 4; + } + } + while (i) { + if ((fl & STBSP__TRIPLET_COMMA) && (cs++ == c)) { + cs = 0; + *bf++ = stbsp__comma; + } else + *bf++ = '0'; + --i; + } + stbsp__chk_cb_buf(1); + } + } + + // copy leader if there is still one + sn = lead + 1; + while (lead[0]) { + stbsp__int32 i; + stbsp__cb_buf_clamp(i, lead[0]); + lead[0] -= (char)i; + while (i) { + *bf++ = *sn++; + --i; + } + stbsp__chk_cb_buf(1); + } + + // copy the string + n = l; + while (n) { + stbsp__int32 i; + stbsp__cb_buf_clamp(i, n); + n -= i; + STBSP__UNALIGNED(while (i >= 4) { + *(stbsp__uint32 volatile *)bf = *(stbsp__uint32 volatile *)s; + bf += 4; + s += 4; + i -= 4; + }) + while (i) { + *bf++ = *s++; + --i; + } + stbsp__chk_cb_buf(1); + } + + // copy trailing zeros + while (tz) { + stbsp__int32 i; + stbsp__cb_buf_clamp(i, tz); + tz -= i; + while (i) { + if ((((stbsp__uintptr)bf) & 3) == 0) + break; + *bf++ = '0'; + --i; + } + while (i >= 4) { + *(stbsp__uint32 *)bf = 0x30303030; + bf += 4; + i -= 4; + } + while (i) { + *bf++ = '0'; + --i; + } + stbsp__chk_cb_buf(1); + } + + // copy tail if there is one + sn = tail + 1; + while (tail[0]) { + stbsp__int32 i; + stbsp__cb_buf_clamp(i, tail[0]); + tail[0] -= (char)i; + while (i) { + *bf++ = *sn++; + --i; + } + stbsp__chk_cb_buf(1); + } + + // handle the left justify + if (fl & STBSP__LEFTJUST) + if (fw > 0) { + while (fw) { + stbsp__int32 i; + stbsp__cb_buf_clamp(i, fw); + fw -= i; + while (i) { + if ((((stbsp__uintptr)bf) & 3) == 0) + break; + *bf++ = ' '; + --i; + } + while (i >= 4) { + *(stbsp__uint32 *)bf = 0x20202020; + bf += 4; + i -= 4; + } + while (i--) + *bf++ = ' '; + stbsp__chk_cb_buf(1); + } + } + break; + + default: // unknown, just copy code + s = num + STBSP__NUMSZ - 1; + *s = f[0]; + l = 1; + fw = fl = 0; + lead[0] = 0; + tail[0] = 0; + pr = 0; + dp = 0; + cs = 0; + goto scopy; + } + ++f; + } +endfmt: + + if (!callback) + *bf = 0; + else + stbsp__flush_cb(); + +done: + return tlen + (int)(bf - buf); +} + +// cleanup +#undef STBSP__LEFTJUST +#undef STBSP__LEADINGPLUS +#undef STBSP__LEADINGSPACE +#undef STBSP__LEADING_0X +#undef STBSP__LEADINGZERO +#undef STBSP__INTMAX +#undef STBSP__TRIPLET_COMMA +#undef STBSP__NEGATIVE +#undef STBSP__METRIC_SUFFIX +#undef STBSP__NUMSZ +#undef stbsp__chk_cb_bufL +#undef stbsp__chk_cb_buf +#undef stbsp__flush_cb +#undef stbsp__cb_buf_clamp + +// ============================================================================ +// wrapper functions + +STBSP__PUBLICDEF int STB_SPRINTF_DECORATE(sprintf)(char *buf, char const *fmt, ...) +{ + int result; + va_list va; + va_start(va, fmt); + result = STB_SPRINTF_DECORATE(vsprintfcb)(0, 0, buf, fmt, va); + va_end(va); + return result; +} + +typedef struct stbsp__context { + char *buf; + int count; + int length; + char tmp[STB_SPRINTF_MIN]; +} stbsp__context; + +static char *stbsp__clamp_callback(const char *buf, void *user, int len) +{ + stbsp__context *c = (stbsp__context *)user; + c->length += len; + + if (len > c->count) + len = c->count; + + if (len) { + if (buf != c->buf) { + const char *s, *se; + char *d; + d = c->buf; + s = buf; + se = buf + len; + do { + *d++ = *s++; + } while (s < se); + } + c->buf += len; + c->count -= len; + } + + if (c->count <= 0) + return c->tmp; + return (c->count >= STB_SPRINTF_MIN) ? c->buf : c->tmp; // go direct into buffer if you can +} + +static char * stbsp__count_clamp_callback( const char * buf, void * user, int len ) +{ + stbsp__context * c = (stbsp__context*)user; + (void) sizeof(buf); + + c->length += len; + return c->tmp; // go direct into buffer if you can +} + +STBSP__PUBLICDEF int STB_SPRINTF_DECORATE( vsnprintf )( char * buf, int count, char const * fmt, va_list va ) +{ + stbsp__context c; + + if ( (count == 0) && !buf ) + { + c.length = 0; + + STB_SPRINTF_DECORATE( vsprintfcb )( stbsp__count_clamp_callback, &c, c.tmp, fmt, va ); + } + else + { + int l; + + c.buf = buf; + c.count = count; + c.length = 0; + + STB_SPRINTF_DECORATE( vsprintfcb )( stbsp__clamp_callback, &c, stbsp__clamp_callback(0,&c,0), fmt, va ); + + // zero-terminate + l = (int)( c.buf - buf ); + if ( l >= count ) // should never be greater, only equal (or less) than count + l = count - 1; + buf[l] = 0; + } + + return c.length; +} + +STBSP__PUBLICDEF int STB_SPRINTF_DECORATE(snprintf)(char *buf, int count, char const *fmt, ...) +{ + int result; + va_list va; + va_start(va, fmt); + + result = STB_SPRINTF_DECORATE(vsnprintf)(buf, count, fmt, va); + va_end(va); + + return result; +} + +STBSP__PUBLICDEF int STB_SPRINTF_DECORATE(vsprintf)(char *buf, char const *fmt, va_list va) +{ + return STB_SPRINTF_DECORATE(vsprintfcb)(0, 0, buf, fmt, va); +} + +// ======================================================================= +// low level float utility functions + +#ifndef STB_SPRINTF_NOFLOAT + +// copies d to bits w/ strict aliasing (this compiles to nothing on /Ox) +#define STBSP__COPYFP(dest, src) \ + { \ + int cn; \ + for (cn = 0; cn < 8; cn++) \ + ((char *)&dest)[cn] = ((char *)&src)[cn]; \ + } + +// get float info +static stbsp__int32 stbsp__real_to_parts(stbsp__int64 *bits, stbsp__int32 *expo, double value) +{ + double d; + stbsp__int64 b = 0; + + // load value and round at the frac_digits + d = value; + + STBSP__COPYFP(b, d); + + *bits = b & ((((stbsp__uint64)1) << 52) - 1); + *expo = (stbsp__int32)(((b >> 52) & 2047) - 1023); + + return (stbsp__int32)((stbsp__uint64) b >> 63); +} + +static double const stbsp__bot[23] = { + 1e+000, 1e+001, 1e+002, 1e+003, 1e+004, 1e+005, 1e+006, 1e+007, 1e+008, 1e+009, 1e+010, 1e+011, + 1e+012, 1e+013, 1e+014, 1e+015, 1e+016, 1e+017, 1e+018, 1e+019, 1e+020, 1e+021, 1e+022 +}; +static double const stbsp__negbot[22] = { + 1e-001, 1e-002, 1e-003, 1e-004, 1e-005, 1e-006, 1e-007, 1e-008, 1e-009, 1e-010, 1e-011, + 1e-012, 1e-013, 1e-014, 1e-015, 1e-016, 1e-017, 1e-018, 1e-019, 1e-020, 1e-021, 1e-022 +}; +static double const stbsp__negboterr[22] = { + -5.551115123125783e-018, -2.0816681711721684e-019, -2.0816681711721686e-020, -4.7921736023859299e-021, -8.1803053914031305e-022, 4.5251888174113741e-023, + 4.5251888174113739e-024, -2.0922560830128471e-025, -6.2281591457779853e-026, -3.6432197315497743e-027, 6.0503030718060191e-028, 2.0113352370744385e-029, + -3.0373745563400371e-030, 1.1806906454401013e-032, -7.7705399876661076e-032, 2.0902213275965398e-033, -7.1542424054621921e-034, -7.1542424054621926e-035, + 2.4754073164739869e-036, 5.4846728545790429e-037, 9.2462547772103625e-038, -4.8596774326570872e-039 +}; +static double const stbsp__top[13] = { + 1e+023, 1e+046, 1e+069, 1e+092, 1e+115, 1e+138, 1e+161, 1e+184, 1e+207, 1e+230, 1e+253, 1e+276, 1e+299 +}; +static double const stbsp__negtop[13] = { + 1e-023, 1e-046, 1e-069, 1e-092, 1e-115, 1e-138, 1e-161, 1e-184, 1e-207, 1e-230, 1e-253, 1e-276, 1e-299 +}; +static double const stbsp__toperr[13] = { + 8388608, + 6.8601809640529717e+028, + -7.253143638152921e+052, + -4.3377296974619174e+075, + -1.5559416129466825e+098, + -3.2841562489204913e+121, + -3.7745893248228135e+144, + -1.7356668416969134e+167, + -3.8893577551088374e+190, + -9.9566444326005119e+213, + 6.3641293062232429e+236, + -5.2069140800249813e+259, + -5.2504760255204387e+282 +}; +static double const stbsp__negtoperr[13] = { + 3.9565301985100693e-040, -2.299904345391321e-063, 3.6506201437945798e-086, 1.1875228833981544e-109, + -5.0644902316928607e-132, -6.7156837247865426e-155, -2.812077463003139e-178, -5.7778912386589953e-201, + 7.4997100559334532e-224, -4.6439668915134491e-247, -6.3691100762962136e-270, -9.436808465446358e-293, + 8.0970921678014997e-317 +}; + +#if defined(_MSC_VER) && (_MSC_VER <= 1200) +static stbsp__uint64 const stbsp__powten[20] = { + 1, + 10, + 100, + 1000, + 10000, + 100000, + 1000000, + 10000000, + 100000000, + 1000000000, + 10000000000, + 100000000000, + 1000000000000, + 10000000000000, + 100000000000000, + 1000000000000000, + 10000000000000000, + 100000000000000000, + 1000000000000000000, + 10000000000000000000U +}; +#define stbsp__tento19th ((stbsp__uint64)1000000000000000000) +#else +static stbsp__uint64 const stbsp__powten[20] = { + 1, + 10, + 100, + 1000, + 10000, + 100000, + 1000000, + 10000000, + 100000000, + 1000000000, + 10000000000ULL, + 100000000000ULL, + 1000000000000ULL, + 10000000000000ULL, + 100000000000000ULL, + 1000000000000000ULL, + 10000000000000000ULL, + 100000000000000000ULL, + 1000000000000000000ULL, + 10000000000000000000ULL +}; +#define stbsp__tento19th (1000000000000000000ULL) +#endif + +#define stbsp__ddmulthi(oh, ol, xh, yh) \ + { \ + double ahi = 0, alo, bhi = 0, blo; \ + stbsp__int64 bt; \ + oh = xh * yh; \ + STBSP__COPYFP(bt, xh); \ + bt &= ((~(stbsp__uint64)0) << 27); \ + STBSP__COPYFP(ahi, bt); \ + alo = xh - ahi; \ + STBSP__COPYFP(bt, yh); \ + bt &= ((~(stbsp__uint64)0) << 27); \ + STBSP__COPYFP(bhi, bt); \ + blo = yh - bhi; \ + ol = ((ahi * bhi - oh) + ahi * blo + alo * bhi) + alo * blo; \ + } + +#define stbsp__ddtoS64(ob, xh, xl) \ + { \ + double ahi = 0, alo, vh, t; \ + ob = (stbsp__int64)xh; \ + vh = (double)ob; \ + ahi = (xh - vh); \ + t = (ahi - xh); \ + alo = (xh - (ahi - t)) - (vh + t); \ + ob += (stbsp__int64)(ahi + alo + xl); \ + } + +#define stbsp__ddrenorm(oh, ol) \ + { \ + double s; \ + s = oh + ol; \ + ol = ol - (s - oh); \ + oh = s; \ + } + +#define stbsp__ddmultlo(oh, ol, xh, xl, yh, yl) ol = ol + (xh * yl + xl * yh); + +#define stbsp__ddmultlos(oh, ol, xh, yl) ol = ol + (xh * yl); + +static void stbsp__raise_to_power10(double *ohi, double *olo, double d, stbsp__int32 power) // power can be -323 to +350 +{ + double ph, pl; + if ((power >= 0) && (power <= 22)) { + stbsp__ddmulthi(ph, pl, d, stbsp__bot[power]); + } else { + stbsp__int32 e, et, eb; + double p2h, p2l; + + e = power; + if (power < 0) + e = -e; + et = (e * 0x2c9) >> 14; /* %23 */ + if (et > 13) + et = 13; + eb = e - (et * 23); + + ph = d; + pl = 0.0; + if (power < 0) { + if (eb) { + --eb; + stbsp__ddmulthi(ph, pl, d, stbsp__negbot[eb]); + stbsp__ddmultlos(ph, pl, d, stbsp__negboterr[eb]); + } + if (et) { + stbsp__ddrenorm(ph, pl); + --et; + stbsp__ddmulthi(p2h, p2l, ph, stbsp__negtop[et]); + stbsp__ddmultlo(p2h, p2l, ph, pl, stbsp__negtop[et], stbsp__negtoperr[et]); + ph = p2h; + pl = p2l; + } + } else { + if (eb) { + e = eb; + if (eb > 22) + eb = 22; + e -= eb; + stbsp__ddmulthi(ph, pl, d, stbsp__bot[eb]); + if (e) { + stbsp__ddrenorm(ph, pl); + stbsp__ddmulthi(p2h, p2l, ph, stbsp__bot[e]); + stbsp__ddmultlos(p2h, p2l, stbsp__bot[e], pl); + ph = p2h; + pl = p2l; + } + } + if (et) { + stbsp__ddrenorm(ph, pl); + --et; + stbsp__ddmulthi(p2h, p2l, ph, stbsp__top[et]); + stbsp__ddmultlo(p2h, p2l, ph, pl, stbsp__top[et], stbsp__toperr[et]); + ph = p2h; + pl = p2l; + } + } + } + stbsp__ddrenorm(ph, pl); + *ohi = ph; + *olo = pl; +} + +// given a float value, returns the significant bits in bits, and the position of the +// decimal point in decimal_pos. +/-INF and NAN are specified by special values +// returned in the decimal_pos parameter. +// frac_digits is absolute normally, but if you want from first significant digits (got %g and %e), or in 0x80000000 +static stbsp__int32 stbsp__real_to_str(char const **start, stbsp__uint32 *len, char *out, stbsp__int32 *decimal_pos, double value, stbsp__uint32 frac_digits) +{ + double d; + stbsp__int64 bits = 0; + stbsp__int32 expo, e, ng, tens; + + d = value; + STBSP__COPYFP(bits, d); + expo = (stbsp__int32)((bits >> 52) & 2047); + ng = (stbsp__int32)((stbsp__uint64) bits >> 63); + if (ng) + d = -d; + + if (expo == 2047) // is nan or inf? + { + *start = (bits & ((((stbsp__uint64)1) << 52) - 1)) ? "NaN" : "Inf"; + *decimal_pos = STBSP__SPECIAL; + *len = 3; + return ng; + } + + if (expo == 0) // is zero or denormal + { + if (((stbsp__uint64) bits << 1) == 0) // do zero + { + *decimal_pos = 1; + *start = out; + out[0] = '0'; + *len = 1; + return ng; + } + // find the right expo for denormals + { + stbsp__int64 v = ((stbsp__uint64)1) << 51; + while ((bits & v) == 0) { + --expo; + v >>= 1; + } + } + } + + // find the decimal exponent as well as the decimal bits of the value + { + double ph, pl; + + // log10 estimate - very specifically tweaked to hit or undershoot by no more than 1 of log10 of all expos 1..2046 + tens = expo - 1023; + tens = (tens < 0) ? ((tens * 617) / 2048) : (((tens * 1233) / 4096) + 1); + + // move the significant bits into position and stick them into an int + stbsp__raise_to_power10(&ph, &pl, d, 18 - tens); + + // get full as much precision from double-double as possible + stbsp__ddtoS64(bits, ph, pl); + + // check if we undershot + if (((stbsp__uint64)bits) >= stbsp__tento19th) + ++tens; + } + + // now do the rounding in integer land + frac_digits = (frac_digits & 0x80000000) ? ((frac_digits & 0x7ffffff) + 1) : (tens + frac_digits); + if ((frac_digits < 24)) { + stbsp__uint32 dg = 1; + if ((stbsp__uint64)bits >= stbsp__powten[9]) + dg = 10; + while ((stbsp__uint64)bits >= stbsp__powten[dg]) { + ++dg; + if (dg == 20) + goto noround; + } + if (frac_digits < dg) { + stbsp__uint64 r; + // add 0.5 at the right position and round + e = dg - frac_digits; + if ((stbsp__uint32)e >= 24) + goto noround; + r = stbsp__powten[e]; + bits = bits + (r / 2); + if ((stbsp__uint64)bits >= stbsp__powten[dg]) + ++tens; + bits /= r; + } + noround:; + } + + // kill long trailing runs of zeros + if (bits) { + stbsp__uint32 n; + for (;;) { + if (bits <= 0xffffffff) + break; + if (bits % 1000) + goto donez; + bits /= 1000; + } + n = (stbsp__uint32)bits; + while ((n % 1000) == 0) + n /= 1000; + bits = n; + donez:; + } + + // convert to string + out += 64; + e = 0; + for (;;) { + stbsp__uint32 n; + char *o = out - 8; + // do the conversion in chunks of U32s (avoid most 64-bit divides, worth it, constant denomiators be damned) + if (bits >= 100000000) { + n = (stbsp__uint32)(bits % 100000000); + bits /= 100000000; + } else { + n = (stbsp__uint32)bits; + bits = 0; + } + while (n) { + out -= 2; + *(stbsp__uint16 *)out = *(stbsp__uint16 *)&stbsp__digitpair.pair[(n % 100) * 2]; + n /= 100; + e += 2; + } + if (bits == 0) { + if ((e) && (out[0] == '0')) { + ++out; + --e; + } + break; + } + while (out != o) { + *--out = '0'; + ++e; + } + } + + *decimal_pos = tens; + *start = out; + *len = e; + return ng; +} + +#undef stbsp__ddmulthi +#undef stbsp__ddrenorm +#undef stbsp__ddmultlo +#undef stbsp__ddmultlos +#undef STBSP__SPECIAL +#undef STBSP__COPYFP + +#endif // STB_SPRINTF_NOFLOAT + +// clean up +#undef stbsp__uint16 +#undef stbsp__uint32 +#undef stbsp__int32 +#undef stbsp__uint64 +#undef stbsp__int64 +#undef STBSP__UNALIGNED + +#if COLLA_CLANG +#pragma clang diagnostic pop +#endif + +#endif // STB_SPRINTF_IMPLEMENTATION + +/* +------------------------------------------------------------------------------ +This software is available under 2 licenses -- choose whichever you prefer. +------------------------------------------------------------------------------ +ALTERNATIVE A - MIT License +Copyright (c) 2017 Sean Barrett +Permission is hereby granted, free of charge, to any person obtaining a copy of +this software and associated documentation files (the "Software"), to deal in +the Software without restriction, including without limitation the rights to +use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies +of the Software, and to permit persons to whom the Software is furnished to do +so, subject to the following conditions: +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. +------------------------------------------------------------------------------ +ALTERNATIVE B - Public Domain (www.unlicense.org) +This is free and unencumbered software released into the public domain. +Anyone is free to copy, modify, publish, use, compile, sell, or distribute this +software, either in source code form or as a compiled binary, for any purpose, +commercial or non-commercial, and by any means. +In jurisdictions that recognize copyright laws, the author or authors of this +software dedicate any and all copyright interest in the software to the public +domain. We make this dedication for the benefit of the public at large and to +the detriment of our heirs and successors. We intend this dedication to be an +overt act of relinquishment in perpetuity of all present and future rights to +this software under copyright law. +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN +ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION +WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. +------------------------------------------------------------------------------ +*/ diff --git a/tests/arena_tests.c b/tests/arena_tests.c new file mode 100644 index 0000000..032e0e5 --- /dev/null +++ b/tests/arena_tests.c @@ -0,0 +1,235 @@ +#include "../colla.h" + +#include "runner.h" + +UNIT_TEST(arena_init_virtual) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + ASSERT(arena.type == ARENA_VIRTUAL); + ASSERT(arena.beg != NULL); + ASSERT(arena.cur == arena.beg); + ASSERT(arena.end == arena.beg + MB(1)); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_init_malloc) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + ASSERT(arena.type == ARENA_MALLOC); + ASSERT(arena.beg != NULL); + ASSERT(arena.cur == arena.beg); + ASSERT(arena.end == arena.beg + KB(4)); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_init_malloc_always) { + arena_t arena = arena_make(ARENA_MALLOC_ALWAYS, KB(4)); + ASSERT(arena.type == ARENA_MALLOC_ALWAYS); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_init_static) { + u8 buffer[KB(4)]; + arena_t arena = arena_make(ARENA_STATIC, KB(4), buffer); + ASSERT(arena.type == ARENA_STATIC); + ASSERT(arena.beg == buffer); + ASSERT(arena.cur == arena.beg); + ASSERT(arena.end == arena.beg + KB(4)); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_alloc_basic) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + int *ptr = alloc(&arena, int); + ASSERT(ptr != NULL); + *ptr = 42; + ASSERT(*ptr == 42); + ASSERT(arena.cur > arena.beg); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_alloc_array) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + int *arr = alloc(&arena, int, .count = 10); + ASSERT(arr != NULL); + for (int i = 0; i < 10; i++) { + arr[i] = i; + } + for (int i = 0; i < 10; i++) { + ASSERT(arr[i] == i); + } + arena_cleanup(&arena); +} + +UNIT_TEST(arena_alloc_custom_align) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + void *ptr = alloc(&arena, char, .align = 64); + ASSERT(ptr != NULL); + ASSERT(((uintptr_t)ptr & 63) == 0); // Should be 64-byte aligned + arena_cleanup(&arena); +} + +UNIT_TEST(arena_alloc_nozero) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + int *ptr1 = alloc(&arena, int); + ASSERT(*ptr1 == 0); // Default zeroed + + int *ptr2 = alloc(&arena, int, .flags = ALLOC_NOZERO); + // We can't assert on the value as it's uninitialized + ASSERT(ptr2 != NULL); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_alloc_soft_fail) { + u8 buffer[10]; + arena_t arena = arena_make(ARENA_STATIC, 10, buffer); + + void *ptr1 = alloc(&arena, char, .count = 5); + ASSERT(ptr1 != NULL); + + // This would normally fail, but with SOFT_FAIL it returns NULL + void *ptr2 = alloc(&arena, char, .count = 10, .flags = ALLOC_SOFT_FAIL); + ASSERT(ptr2 == NULL); + + arena_cleanup(&arena); +} + +UNIT_TEST(arena_scratch) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + arena_t scratch = arena_scratch(&arena, KB(1)); + + ASSERT(scratch.beg != NULL); + ASSERT(scratch.type == ARENA_STATIC); + + void *ptr = alloc(&scratch, int); + ASSERT(ptr != NULL); + + // Scratch cleanup happens implicitly when parent arena is cleaned up + arena_cleanup(&arena); +} + +UNIT_TEST(arena_tell) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + usize pos1 = arena_tell(&arena); + ASSERT(pos1 == 0); + + alloc(&arena, int); + usize pos2 = arena_tell(&arena); + ASSERT(pos2 > pos1); + + arena_cleanup(&arena); +} + +UNIT_TEST(arena_remaining) { + arena_t arena = arena_make(ARENA_VIRTUAL, KB(64)); + usize initial_remaining = arena_remaining(&arena); + ASSERT(initial_remaining == KB(64)); + + alloc(&arena, char, .count = KB(4)); + usize after_alloc = arena_remaining(&arena); + ASSERT(after_alloc < initial_remaining); + ASSERT(after_alloc >= KB(60)); // Account for possible alignment padding + + arena_cleanup(&arena); +} + +UNIT_TEST(arena_capacity) { + arena_t arena = arena_make(ARENA_VIRTUAL, KB(64)); + usize cap = arena_capacity(&arena); + ASSERT(cap == KB(64)); + arena_cleanup(&arena); +} + +UNIT_TEST(arena_rewind) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + + usize mark = arena_tell(&arena); + + int *ptr1 = alloc(&arena, int); + *ptr1 = 42; + + alloc(&arena, char, .count = 100); + + arena_rewind(&arena, mark); + + int *ptr2 = alloc(&arena, int); + ASSERT(ptr2 == ptr1); // Should reuse the same memory + + // Original value is lost after rewind + *ptr2 = 24; + ASSERT(*ptr2 == 24); + + arena_cleanup(&arena); +} + +UNIT_TEST(arena_pop) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + + alloc(&arena, char, .count = 100); + usize pos = arena_tell(&arena); + + alloc(&arena, char, .count = 50); + + arena_pop(&arena, 50); + ASSERT(arena_tell(&arena) == pos); + + arena_cleanup(&arena); +} + +UNIT_TEST(arena_malloc_arena) { + void *ptr = alloc(&malloc_arena, int); + ASSERT(ptr != NULL); + + // We need to free each allocation from malloc_arena manually + os_free(ptr); +} + +UNIT_TEST(arena_alloc_mixed_types) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(1)); + + int *i = alloc(&arena, int); + float *f = alloc(&arena, float); + char *c = alloc(&arena, char); + + *i = 42; + *f = 3.14f; + *c = 'A'; + + ASSERT(*i == 42); + ASSERT(*f == 3.14f); + ASSERT(*c == 'A'); + + arena_cleanup(&arena); +} + +UNIT_TEST(arena_multiple_arenas) { + arena_t arena1 = arena_make(ARENA_VIRTUAL, KB(4)); + arena_t arena2 = arena_make(ARENA_VIRTUAL, KB(4)); + + int *ptr1 = alloc(&arena1, int); + int *ptr2 = alloc(&arena2, int); + + *ptr1 = 42; + *ptr2 = 24; + + ASSERT(*ptr1 == 42); + ASSERT(*ptr2 == 24); + + arena_cleanup(&arena1); + arena_cleanup(&arena2); +} + +UNIT_TEST(arena_stress_test) { + arena_t arena = arena_make(ARENA_VIRTUAL, MB(10)); + + // Allocate many objects + for (int i = 0; i < 1000; i++) { + int *ptr = alloc(&arena, int, .flags = ALLOC_SOFT_FAIL); + ASSERT(ptr != NULL); + *ptr = i; + } + + // Allocate a large block + void *large = alloc(&arena, char, .count = MB(5), .flags = ALLOC_SOFT_FAIL); + ASSERT(large != NULL); + + arena_cleanup(&arena); +} diff --git a/tests/core_tests.c b/tests/core_tests.c new file mode 100644 index 0000000..ddb89ff --- /dev/null +++ b/tests/core_tests.c @@ -0,0 +1,231 @@ +#include "runner.h" +#include "../colla.h" +#include +#include +#include +#include + +UNIT_TEST(arrlen_macro) { + int array[5] = {1, 2, 3, 4, 5}; + ASSERT(arrlen(array) == 5); + + char str[] = "hello"; + ASSERT(arrlen(str) == 6); // Including null terminator +} + +UNIT_TEST(min_max_macros) { + ASSERT(MIN(5, 10) == 5); + ASSERT(MIN(-5, 10) == -5); + ASSERT(MIN(5.5, 10.1) == 5.5); + + ASSERT(MAX(5, 10) == 10); + ASSERT(MAX(-5, 10) == 10); + ASSERT(MAX(5.5, 10.1) == 10.1); +} + +UNIT_TEST(size_constants) { + ASSERT(KB(1) == 1024); + ASSERT(MB(1) == 1024 * 1024); + ASSERT(GB(1) == 1024 * 1024 * 1024); + ASSERT(TB(1) == 1024ULL * 1024ULL * 1024ULL * 1024ULL); + + ASSERT(KB(2) == 2048); + ASSERT(MB(2) == 2 * 1024 * 1024); +} + +UNIT_TEST(linked_list) { + // Define a simple node structure + typedef struct Node { + int value; + struct Node *next; + } Node; + + // Create some nodes + Node n1 = {1, NULL}; + Node n2 = {2, NULL}; + Node n3 = {3, NULL}; + + // Initialize list + Node *list = NULL; + + // Push nodes onto list + list_push(list, &n3); + list_push(list, &n2); + list_push(list, &n1); + + // Check list order + ASSERT(list == &n1); + ASSERT(list->next == &n2); + ASSERT(list->next->next == &n3); + ASSERT(list->next->next->next == NULL); + + // Pop from list + list_pop(list); + ASSERT(list == &n2); + ASSERT(list->next == &n3); + + list_pop(list); + ASSERT(list == &n3); + + list_pop(list); + ASSERT(list == NULL); +} + +UNIT_TEST(double_linked_list) { + // Define a double linked node + typedef struct DNode { + int value; + struct DNode *next; + struct DNode *prev; + } DNode; + + // Create some nodes + DNode n1 = {1, NULL, NULL}; + DNode n2 = {2, NULL, NULL}; + DNode n3 = {3, NULL, NULL}; + + // Initialize list + DNode *list = NULL; + + // Push nodes + dlist_push(list, &n3); + dlist_push(list, &n2); + dlist_push(list, &n1); + + // Check list structure + ASSERT(list == &n1); + ASSERT(list->next == &n2); + ASSERT(list->next->next == &n3); + ASSERT(list->prev == NULL); + ASSERT(list->next->prev == &n1); + ASSERT(list->next->next->prev == &n2); + + // Pop middle node + dlist_pop(list, &n2); + + // Check updated structure + ASSERT(list == &n1); + ASSERT(list->next == &n3); + ASSERT(list->next->prev == &n1); + + // Pop first node + dlist_pop(list, &n1); + ASSERT(list == &n3); + ASSERT(list->prev == NULL); +} + +UNIT_TEST(ordered_linked_list) { + // Define a simple node + typedef struct ONode { + int value; + struct ONode *next; + } ONode; + + // Create nodes + ONode n1 = {1, NULL}; + ONode n2 = {2, NULL}; + ONode n3 = {3, NULL}; + + // Initialize head and tail + ONode *head = NULL; + ONode *tail = NULL; + + // Push nodes in order + olist_push(head, tail, &n1); + ASSERT(head == &n1); + ASSERT(tail == &n1); + + olist_push(head, tail, &n2); + ASSERT(head == &n1); + ASSERT(tail == &n2); + ASSERT(head->next == &n2); + + olist_push(head, tail, &n3); + ASSERT(head == &n1); + ASSERT(tail == &n3); + ASSERT(head->next == &n2); + ASSERT(head->next->next == &n3); +} + +UNIT_TEST(for_each_macro) { + // Define a simple node + typedef struct Node { + int value; + struct Node *next; + } Node; + + // Create linked list + Node n1 = {1, NULL}; + Node n2 = {2, NULL}; + Node n3 = {3, NULL}; + + n1.next = &n2; + n2.next = &n3; + + Node *list = &n1; + + // Use for_each to sum values + int sum = 0; + for_each(it, list) { + sum += it->value; + } + + ASSERT(sum == 6); // 1 + 2 + 3 +} + +UNIT_TEST(fmt_print) { + // This function outputs to stdout, so we can't easily test its output + // Just verify it doesn't crash and returns a positive value + int result = fmt_print("Test %d %s\n", 42, "hello"); + ASSERT(result > 0); +} + +UNIT_TEST(fmt_buffer) { + char buffer[128]; + + // Basic formatting + int result = fmt_buffer(buffer, sizeof(buffer), "Int: %d", 42); + ASSERT(result > 0); + ASSERT(strcmp(buffer, "Int: 42") == 0); + + // Multiple arguments + result = fmt_buffer(buffer, sizeof(buffer), "%s %d %.2f", "Test", 123, 3.14159); + ASSERT(result > 0); + ASSERT(strcmp(buffer, "Test 123 3.14") == 0); + + // Buffer size limiting + result = fmt_buffer(buffer, 5, "Long text that won't fit"); + ASSERT(result == 24); // fmt_buffer returns the lenght if it did fit + ASSERT(strlen(buffer) == 4); +} + +// Helper function to test variadic function +int test_fmt_printv(const char *fmt, ...) { + va_list args; + va_start(args, fmt); + int result = fmt_printv(fmt, args); + va_end(args); + return result; +} + +UNIT_TEST(fmt_printv) { + // Just verify it doesn't crash and returns positive value + int result = test_fmt_printv("Test %d %s\n", 42, "hello"); + ASSERT(result > 0); +} + +// Helper function to test variadic function +int test_fmt_bufferv(char *buf, usize len, const char *fmt, ...) { + va_list args; + va_start(args, fmt); + int result = fmt_bufferv(buf, len, fmt, args); + va_end(args); + return result; +} + +UNIT_TEST(fmt_bufferv) { + char buffer[128]; + int result = test_fmt_bufferv(buffer, sizeof(buffer), "%d %s", 42, "test"); + ASSERT(result > 0); + ASSERT(strcmp(buffer, "42 test") == 0); +} diff --git a/tests/highlight_tests.c b/tests/highlight_tests.c new file mode 100644 index 0000000..7b47c73 --- /dev/null +++ b/tests/highlight_tests.c @@ -0,0 +1,178 @@ +#include "runner.h" +#include "../colla.h" +#include + +UNIT_TEST(highlight_init) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // Create a basic configuration + hl_config_t config = {0}; + + // Define custom colors + config.colors[HL_COLOR_NORMAL] = strv_init("normal"); + config.colors[HL_COLOR_KEYWORDS] = strv_init("keyword"); + config.colors[HL_COLOR_STRING] = strv_init("string"); + config.colors[HL_COLOR_COMMENT] = strv_init("comment"); + + // Initialize highlighter + hl_ctx_t *ctx = hl_init(&arena, &config); + ASSERT(ctx != NULL); + + arena_cleanup(&arena); +} + +UNIT_TEST(highlight_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // Create a configuration + hl_config_t config = {0}; + + // Define custom colors for HTML output + config.colors[HL_COLOR_NORMAL] = strv_init("color:black"); + config.colors[HL_COLOR_KEYWORDS] = strv_init("color:blue"); + config.colors[HL_COLOR_STRING] = strv_init("color:green"); + config.colors[HL_COLOR_COMMENT] = strv_init("color:gray"); + config.colors[HL_COLOR_NUMBER] = strv_init("color:purple"); + + // Set HTML output flag + config.flags = HL_FLAG_HTML; + + // Initialize highlighter + hl_ctx_t *ctx = hl_init(&arena, &config); + ASSERT(ctx != NULL); + + // Sample C code to highlight + strview_t code = strv_init( + "// This is a comment\n" + "int main() {\n" + " printf(\"Hello, World!\\n\");\n" + " return 0;\n" + "}\n" + ); + + // Highlight the code + str_t highlighted = hl_highlight(&arena, ctx, code); + + // Verify the output + ASSERT(!str_is_empty(highlighted)); + + // We can't easily test the exact output without parsing HTML, + // but we can check that key strings are present + const char *html_start = " + +// Initialization Tests +UNIT_TEST(net_init_cleanup) { + net_init(); + // Simple test to make sure initialization succeeds + ASSERT(net_get_last_error() == 0); + net_cleanup(); +} + +// HTTP Method/Status Tests +UNIT_TEST(http_method_strings) { + ASSERT(strcmp(http_get_method_string(HTTP_GET), "GET") == 0); + ASSERT(strcmp(http_get_method_string(HTTP_POST), "POST") == 0); + ASSERT(strcmp(http_get_method_string(HTTP_HEAD), "HEAD") == 0); + ASSERT(strcmp(http_get_method_string(HTTP_PUT), "PUT") == 0); + ASSERT(strcmp(http_get_method_string(HTTP_DELETE), "DELETE") == 0); +} + +UNIT_TEST(http_status_strings) { + ASSERT(strcmp(http_get_status_string(200), "OK") == 0); + ASSERT(strcmp(http_get_status_string(404), "NOT FOUND") == 0); + ASSERT(strcmp(http_get_status_string(500), "INTERNAL SERVER ERROR") == 0); +} + +// HTTP Headers Tests +UNIT_TEST(http_headers) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // Parse headers + strview_t header_str = strv_init("Content-Type: application/json\r\nUser-Agent: Colla\r\n"); + http_header_t *headers = http_parse_headers(&arena, header_str); + + // Check headers were parsed correctly + ASSERT(headers != NULL); + + // Headers are parsed in reverse order + ASSERT(strv_equals(headers->key, strv_init("User-Agent"))); + ASSERT(strv_equals(headers->value, strv_init("Colla"))); + ASSERT(strv_equals(headers->next->key, strv_init("Content-Type"))); + ASSERT(strv_equals(headers->next->value, strv_init("application/json"))); + + // Test header operations + ASSERT(http_has_header(headers, strv_init("Content-Type"))); + ASSERT(!http_has_header(headers, strv_init("Accept"))); + + strview_t content_type = http_get_header(headers, strv_init("Content-Type")); + ASSERT(strv_equals(content_type, strv_init("application/json"))); + + // Don't try to free headers as they're allocated in the arena + arena_cleanup(&arena); +} + +// HTTP Request/Response Parsing Tests +UNIT_TEST(http_request_parsing) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t req_str = strv_init( + "GET /index.html HTTP/1.1\r\n" + "Host: example.com\r\n" + "User-Agent: Colla\r\n" + "\r\n" + ); + + http_req_t req = http_parse_req(&arena, req_str); + + ASSERT(req.method == HTTP_GET); + ASSERT(req.version.major == 1); + ASSERT(req.version.minor == 1); + ASSERT(strv_equals(req.url, strv("index.html"))); + + ASSERT(http_has_header(req.headers, strv_init("Host"))); + ASSERT(strv_equals(http_get_header(req.headers, strv_init("Host")), strv_init("example.com"))); + + // Convert back to string + str_t req_out = http_req_to_str(&arena, &req); + ASSERT(!str_is_empty(req_out)); + + arena_cleanup(&arena); +} + +UNIT_TEST(http_response_parsing) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t res_str = strv_init( + "HTTP/1.1 200 OK\r\n" + "Content-Type: text/html\r\n" + "Content-Length: 13\r\n" + "\r\n" + "Hello, World!" + ); + + http_res_t res = http_parse_res(&arena, res_str); + + ASSERT(res.status_code == 200); + ASSERT(res.version.major == 1); + ASSERT(res.version.minor == 1); + ASSERT(strv_equals(res.body, strv_init("Hello, World!"))); + + ASSERT(http_has_header(res.headers, strv_init("Content-Type"))); + ASSERT(strv_equals(http_get_header(res.headers, strv_init("Content-Type")), strv_init("text/html"))); + + // Convert back to string + str_t res_out = http_res_to_str(&arena, &res); + ASSERT(!str_is_empty(res_out)); + + arena_cleanup(&arena); +} + +// URL Encoding/Decoding Tests +UNIT_TEST(http_url_encoding) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t original = strv_init("hello world & special chars: ?=&/"); + str_t encoded = http_make_url_safe(&arena, original); + str_t decoded = http_decode_url_safe(&arena, strv_init_str(encoded)); + + ASSERT(!str_is_empty(encoded)); + ASSERT(str_equals(decoded, str_init(&arena, "hello world & special chars: ?=&/"))); + + arena_cleanup(&arena); +} + +UNIT_TEST(http_url_splitting) { + strview_t url = strv_init("http://example.com/path?query=value"); + http_url_t split = http_split_url(url); + + ASSERT(strv_equals(split.host, strv_init("example.com"))); + ASSERT(strv_equals(split.uri, strv_init("/path?query=value"))); +} + +// HTTP Request Tests +// Note: These tests would actually make network requests, so we should mock them +// for real unit tests. Here we'll just test the setup part. +UNIT_TEST(http_request_setup) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // Prepare headers + http_header_t headers[2] = { + { .key = strv_init("Content-Type"), .value = strv_init("application/json"), .next = &headers[1] }, + { .key = strv_init("User-Agent"), .value = strv_init("Colla Test"), .next = NULL } + }; + + // Setup request descriptor + http_request_desc_t desc = { + .arena = &arena, + .url = strv_init("http://example.com"), + .version = { .major = 1, .minor = 1 }, + .request_type = HTTP_GET, + .headers = headers, + .header_count = 2, + .body = strv_init("") + }; + + // We don't actually make the request, just verify the setup is correct + ASSERT(desc.arena == &arena); + ASSERT(strv_equals(desc.url, strv_init("http://example.com"))); + ASSERT(desc.request_type == HTTP_GET); + ASSERT(desc.headers == headers); + ASSERT(desc.header_count == 2); + + arena_cleanup(&arena); +} + +// Socket Tests +UNIT_TEST(socket_basic) { + net_init(); + + // Open a socket + socket_t sock = sk_open(SOCK_TCP); + ASSERT(sk_is_valid(sock)); + + // Close the socket + ASSERT(sk_close(sock)); + + net_cleanup(); +} + +// SHA1 Tests +UNIT_TEST(sha1_hash) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + sha1_t ctx = sha1_init(); + + const char *data = "Hello, World!"; + str_t hash = sha1_str(&arena, &ctx, data, strlen(data)); + + // The SHA1 hash for "Hello, World!" is known + // But we'll just verify it's not empty and has the expected format (40 hex chars) + ASSERT(!str_is_empty(hash)); + ASSERT(hash.len == 40); + + arena_cleanup(&arena); +} + +// Base64 Tests +UNIT_TEST(base64_encoding) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + const char *original = "Hello, World!"; + buffer_t input = { (u8*)original, strlen(original) }; + + // Encode + buffer_t encoded = base64_encode(&arena, input); + ASSERT(encoded.data != NULL); + ASSERT(encoded.len > 0); + + // Decode + buffer_t decoded = base64_decode(&arena, encoded); + + ASSERT(decoded.data != NULL); + ASSERT(decoded.len == input.len); + ASSERT(memcmp(decoded.data, input.data, input.len) == 0); + + arena_cleanup(&arena); +} + +// WebSocket Tests +UNIT_TEST(websocket_encoding) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t message = strv_init("Hello, WebSocket!"); + + // Encode message to WebSocket format + buffer_t encoded = websocket_encode(&arena, message); + ASSERT(encoded.data != NULL); + ASSERT(encoded.len > 0); + + // Decode WebSocket message + str_t decoded = websocket_decode(&arena, encoded); + ASSERT(!str_is_empty(decoded)); + ASSERT(str_equals(decoded, str_init(&arena, "Hello, WebSocket!"))); + + arena_cleanup(&arena); +} diff --git a/tests/os_tests.c b/tests/os_tests.c new file mode 100644 index 0000000..409cdbe --- /dev/null +++ b/tests/os_tests.c @@ -0,0 +1,387 @@ +#include "runner.h" +#include "../colla.h" +#include + +// Handle Tests +UNIT_TEST(os_handle) { + oshandle_t zero = os_handle_zero(); + ASSERT(!os_handle_valid(zero)); + + // Create a handle (using file open) + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + strview_t test_file = strv_init("test_file.txt"); + + // Create test file + oshandle_t h_write = os_file_open(test_file, OS_FILE_WRITE); + ASSERT(os_handle_valid(h_write)); + os_file_puts(h_write, strv_init("test content")); + os_file_close(h_write); + + // Open the file and test handle functions + oshandle_t h_read = os_file_open(test_file, OS_FILE_READ); + ASSERT(os_handle_valid(h_read)); + ASSERT(!os_handle_match(h_read, zero)); + + oshandle_t h_read2 = os_file_open(test_file, OS_FILE_READ); + ASSERT(os_handle_valid(h_read2)); + ASSERT(!os_handle_match(h_read, h_read2)); + + os_file_close(h_read); + os_file_close(h_read2); + os_file_delete(test_file); + + arena_cleanup(&arena); +} + +// File Operations Tests +UNIT_TEST(os_file_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + strview_t test_file = strv_init("test_file.txt"); + + // Delete if exists + if (os_file_exists(test_file)) { + os_file_delete(test_file); + } + + // Check existence + ASSERT(!os_file_exists(test_file)); + + // Create and write + oshandle_t h_write = os_file_open(test_file, OS_FILE_WRITE); + ASSERT(os_handle_valid(h_write)); + + os_file_putc(h_write, 'H'); + os_file_puts(h_write, strv_init("ello World")); + + os_file_close(h_write); + + // Check existence after creation + ASSERT(os_file_exists(test_file)); + + // Read back + oshandle_t h_read = os_file_open(test_file, OS_FILE_READ); + ASSERT(os_handle_valid(h_read)); + + char buffer[12] = {0}; + usize read = os_file_read(h_read, buffer, 11); + ASSERT(read == 11); + ASSERT(strcmp(buffer, "Hello World") == 0); + + os_file_close(h_read); + + // Clean up + os_file_delete(test_file); + ASSERT(!os_file_exists(test_file)); + + arena_cleanup(&arena); +} + +UNIT_TEST(os_file_seek) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + strview_t test_file = strv_init("test_file.txt"); + + // Create and write + oshandle_t h_write = os_file_open(test_file, OS_FILE_WRITE); + ASSERT(os_handle_valid(h_write)); + + os_file_puts(h_write, strv_init("ABCDEFGHIJ")); + os_file_close(h_write); + + // Open for reading + oshandle_t h_read = os_file_open(test_file, OS_FILE_READ); + ASSERT(os_handle_valid(h_read)); + + // Seek to position 5 + ASSERT(os_file_seek(h_read, 5)); + + // Read from position 5 + char buffer[6] = {0}; + usize read = os_file_read(h_read, buffer, 5); + ASSERT(read == 5); + ASSERT(strcmp(buffer, "FGHIJ") == 0); + + // Rewind and read from beginning + os_file_rewind(h_read); + + char buffer2[6] = {0}; + read = os_file_read(h_read, buffer2, 5); + ASSERT(read == 5); + ASSERT(strcmp(buffer2, "ABCDE") == 0); + + // Test file position + ASSERT(os_file_tell(h_read) == 5); + + // Test file size + ASSERT(os_file_size(h_read) == 10); + + // Seek to end + ASSERT(os_file_seek_end(h_read)); + ASSERT(os_file_is_finished(h_read)); + + os_file_close(h_read); + os_file_delete(test_file); + + arena_cleanup(&arena); +} + +UNIT_TEST(os_file_read_write_all) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + strview_t test_file = strv_init("test_file.txt"); + + // Write string + strview_t test_data = strv_init("This is test data for read/write all functions"); + ASSERT(os_file_write_all_str(test_file, test_data)); + + // Read back as string + str_t read_data = os_file_read_all_str(&arena, test_file); + ASSERT(str_equals(read_data, str_init(&arena, "This is test data for read/write all functions"))); + + // Read as buffer + buffer_t buffer = os_file_read_all(&arena, test_file); + ASSERT(buffer.len == test_data.len); + ASSERT(memcmp(buffer.data, test_data.buf, test_data.len) == 0); + + // Write buffer + const char *new_data = "New buffer data"; + buffer_t write_buffer = {(u8*)new_data, strlen(new_data)}; + ASSERT(os_file_write_all(test_file, write_buffer)); + + // Read back after buffer write + str_t read_new = os_file_read_all_str(&arena, test_file); + ASSERT(str_equals(read_new, str_init(&arena, "New buffer data"))); + + // Clean up + os_file_delete(test_file); + arena_cleanup(&arena); +} + +UNIT_TEST(os_file_path) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // Test path splitting + strview_t path = strv_init("/path/to/file.txt"); + strview_t dir, name, ext; + + os_file_split_path(path, &dir, &name, &ext); + + ASSERT(strv_equals(dir, strv_init("/path/to"))); + ASSERT(strv_equals(name, strv_init("file"))); + ASSERT(strv_equals(ext, strv_init(".txt"))); + + // Test full path resolution + strview_t relative_path = strv_init("test_file.txt"); + os_file_write_all_str(relative_path, strv("hello world")); + tstr_t full_path = os_file_fullpath(&arena, relative_path); + + // Can't easily test the exact value, but can verify it's not empty + ASSERT(full_path.len > 0); + + os_file_delete(relative_path); + arena_cleanup(&arena); +} + +// Directory Tests +UNIT_TEST(os_dir_operations) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + strview_t test_dir = strv_init("test_dir"); + + // Delete if exists + if (os_dir_exists(test_dir)) { + os_file_delete(strv("test_dir/test_file.txt")); + os_dir_delete(test_dir); + } + + // Create directory + ASSERT(os_dir_create(test_dir)); + ASSERT(os_dir_exists(test_dir)); + + // Create test file in directory + strview_t test_file_path = strv_init("test_dir/test_file.txt"); + oshandle_t h_write = os_file_open(test_file_path, OS_FILE_WRITE); + ASSERT(os_handle_valid(h_write)); + os_file_puts(h_write, strv_init("test content")); + os_file_close(h_write); + + // Test directory listing + dir_t *dir = os_dir_open(&arena, test_dir); + ASSERT(os_dir_is_valid(dir)); + + bool found_file = false; + dir_foreach(&arena, entry, dir) { + if (str_equals(entry->name, str_init(&arena, "test_file.txt"))) { + found_file = true; + ASSERT(entry->type == DIRTYPE_FILE); + warn(">> %zu", entry->file_size); + ASSERT(entry->file_size == 12); // "test content" + } + } + + ASSERT(found_file); + + // Clean up + os_file_delete(test_file_path); + os_dir_close(dir); + + // Directory should now be empty, so we can delete it + os_dir_delete(test_dir); + ASSERT(!os_dir_exists(test_dir)); + + arena_cleanup(&arena); +} + +// Environment Variable Tests +UNIT_TEST(os_env_vars) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + arena_t scratch = arena_make(ARENA_MALLOC, KB(1)); + + // Set environment variable + strview_t key = strv_init("COLLA_TEST_VAR"); + strview_t value = strv_init("test_value"); + + os_set_env_var(scratch, key, value); + + // Get environment variable + str_t read_value = os_get_env_var(&arena, key); + ASSERT(str_equals(read_value, str_init(&arena, "test_value"))); + + // Get all environment variables + os_env_t *env = os_get_env(&arena); + ASSERT(env != NULL); + + arena_cleanup(&scratch); + arena_cleanup(&arena); +} + +// Virtual Memory Tests +UNIT_TEST(os_virtual_memory) { + usize page_size; + void *memory = os_reserve(MB(1), &page_size); + ASSERT(memory != NULL); + ASSERT(page_size > 0); + + // Commit a page + ASSERT(os_commit(memory, 1)); + + // Write to the committed memory + memset(memory, 0x42, os_get_system_info().page_size); + + // Release the memory + ASSERT(os_release(memory, MB(1))); +} + +// Thread Tests +static int thread_test_func(u64 thread_id, void *userdata) { + int *value = (int*)userdata; + (*value)++; + return 42; +} + +UNIT_TEST(os_thread) { + // Create thread data + int value = 0; + + // Launch thread + oshandle_t thread = os_thread_launch(thread_test_func, &value); + ASSERT(os_handle_valid(thread)); + + // Get thread ID + u64 thread_id = os_thread_get_id(thread); + ASSERT(thread_id != 0); + + // Join thread + int exit_code; + ASSERT(os_thread_join(thread, &exit_code)); + ASSERT(exit_code == 42); + ASSERT(value == 1); +} + +int test_mutex_trylock(u64 id, void *userdata) { + oshandle_t mutex = *((oshandle_t*)userdata); + return os_mutex_try_lock(mutex); +} + + +// Mutex Tests +UNIT_TEST(os_mutex) { + oshandle_t mutex = os_mutex_create(); + ASSERT(os_handle_valid(mutex)); + + oshandle_t thread = os_thread_launch(test_mutex_trylock, &mutex); + + // Lock mutex + os_mutex_lock(mutex); + + int locked = 0; + os_thread_join(thread, &locked); + ASSERT(locked == false); + + // Unlock + os_mutex_unlock(mutex); + + // Try lock should succeed now + ASSERT(os_mutex_try_lock(mutex)); + + // Unlock again + os_mutex_unlock(mutex); + + // Free mutex + os_mutex_free(mutex); +} + +#if !COLLA_NO_CONDITION_VARIABLE +// Condition Variable Tests +typedef struct { + u64 canary_beg; + oshandle_t mutex; + oshandle_t cond_mutex; + oshandle_t cond; + int counter; + u64 canary_end; +} cond_test_data; + +static int cond_test_thread(u64 thread_id, void *userdata) { + cond_test_data *data = (cond_test_data*)userdata; + + info("%zu %zu", data->canary_beg, data->canary_end); + + os_mutex_lock(data->mutex); + data->counter++; + os_mutex_unlock(data->mutex); + + // Signal the condition + os_cond_signal(data->cond); + + return 0; +} + +UNIT_TEST(os_condition_variable) { + cond_test_data data = { + .mutex = os_mutex_create(), + .cond_mutex = os_mutex_create(), + .cond = os_cond_create(), + .counter = 0, + }; + + // Lock mutex before launching thread + os_mutex_lock(data.mutex); + + // Launch thread + oshandle_t thread = os_thread_launch(cond_test_thread, &data); + + // Wait for condition with timeout + os_mutex_lock(data.cond_mutex); + os_cond_wait(data.cond, data.cond_mutex, 1000); + os_mutex_unlock(data.cond_mutex); + + // We should have the lock again, and counter should be 1 + ASSERT(data.counter == 1); + + // Unlock and cleanup + // os_mutex_unlock(data.mutex); + os_thread_join(thread, NULL); + + os_mutex_free(data.mutex); + os_mutex_free(data.cond_mutex); + os_cond_free(data.cond); +} +#endif diff --git a/tests/parsers_tests.c b/tests/parsers_tests.c new file mode 100644 index 0000000..cfbca6e --- /dev/null +++ b/tests/parsers_tests.c @@ -0,0 +1,369 @@ +#include "runner.h" +#include "../colla.h" +#include + +// INI Parser Tests +UNIT_TEST(ini_parse_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t ini_content = strv_init( + "[section1]\n" + "key1=value1\n" + "key2=value2\n" + "\n" + "[section2]\n" + "key3=value3\n" + "key4=value4\n" + ); + + ini_t ini = ini_parse_str(&arena, ini_content, NULL); + ASSERT(ini_is_valid(&ini)); + + // Test section1 + initable_t *section1 = ini_get_table(&ini, strv_init("section1")); + ASSERT(section1 != NULL); + ASSERT(strv_equals(section1->name, strv_init("section1"))); + + // Test section1 values + inivalue_t *key1 = ini_get(section1, strv_init("key1")); + ASSERT(key1 != NULL); + ASSERT(strv_equals(key1->key, strv_init("key1"))); + ASSERT(strv_equals(key1->value, strv_init("value1"))); + + inivalue_t *key2 = ini_get(section1, strv_init("key2")); + ASSERT(key2 != NULL); + ASSERT(strv_equals(key2->key, strv_init("key2"))); + ASSERT(strv_equals(key2->value, strv_init("value2"))); + + // Test section2 + initable_t *section2 = ini_get_table(&ini, strv_init("section2")); + ASSERT(section2 != NULL); + ASSERT(strv_equals(section2->name, strv_init("section2"))); + + // Test section2 values + inivalue_t *key3 = ini_get(section2, strv_init("key3")); + ASSERT(key3 != NULL); + ASSERT(strv_equals(key3->key, strv_init("key3"))); + ASSERT(strv_equals(key3->value, strv_init("value3"))); + + inivalue_t *key4 = ini_get(section2, strv_init("key4")); + ASSERT(key4 != NULL); + ASSERT(strv_equals(key4->key, strv_init("key4"))); + ASSERT(strv_equals(key4->value, strv_init("value4"))); + + arena_cleanup(&arena); +} + +UNIT_TEST(ini_parse_with_options) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t ini_content = strv_init( + "[section1]\n" + "key1:value1\n" + "# This is a comment\n" + "key2:value2\n" + "\n" + "[section1]\n" // Duplicate section + "key3:value3\n" + ); + + iniopt_t options = { + .merge_duplicate_tables = true, + .merge_duplicate_keys = false, + .key_value_divider = ':', + .comment_vals = strv_init("#") + }; + + ini_t ini = ini_parse_str(&arena, ini_content, &options); + ASSERT(ini_is_valid(&ini)); + + // Test section1 (should be merged) + initable_t *section1 = ini_get_table(&ini, strv_init("section1")); + ASSERT(section1 != NULL); + + // Check all keys exist in merged section + inivalue_t *key1 = ini_get(section1, strv_init("key1")); + ASSERT(key1 != NULL); + ASSERT(strv_equals(key1->value, strv_init("value1"))); + + inivalue_t *key2 = ini_get(section1, strv_init("key2")); + ASSERT(key2 != NULL); + ASSERT(strv_equals(key2->value, strv_init("value2"))); + + inivalue_t *key3 = ini_get(section1, strv_init("key3")); + ASSERT(key3 != NULL); + ASSERT(strv_equals(key3->value, strv_init("value3"))); + + arena_cleanup(&arena); +} + +UNIT_TEST(ini_value_conversion) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t ini_content = strv_init( + "[values]\n" + "uint=42\n" + "int=-42\n" + "float=3.14\n" + "bool_true=true\n" + "bool_false=false\n" + "array=item1,item2,item3\n" + ); + + ini_t ini = ini_parse_str(&arena, ini_content, NULL); + initable_t *values = ini_get_table(&ini, strv_init("values")); + ASSERT(values != NULL); + + // Test uint conversion + inivalue_t *uint_val = ini_get(values, strv_init("uint")); + ASSERT(uint_val != NULL); + ASSERT(ini_as_uint(uint_val) == 42); + + // Test int conversion + inivalue_t *int_val = ini_get(values, strv_init("int")); + ASSERT(int_val != NULL); + ASSERT(ini_as_int(int_val) == -42); + + // Test float conversion + inivalue_t *float_val = ini_get(values, strv_init("float")); + ASSERT(float_val != NULL); + ASSERT(ini_as_num(float_val) > 3.13 && ini_as_num(float_val) < 3.15); + + // Test bool conversion + inivalue_t *bool_true = ini_get(values, strv_init("bool_true")); + ASSERT(bool_true != NULL); + ASSERT(ini_as_bool(bool_true) == true); + + inivalue_t *bool_false = ini_get(values, strv_init("bool_false")); + ASSERT(bool_false != NULL); + ASSERT(ini_as_bool(bool_false) == false); + + // Test array conversion + inivalue_t *array_val = ini_get(values, strv_init("array")); + ASSERT(array_val != NULL); + + iniarray_t array = ini_as_arr(&arena, array_val, ','); + ASSERT(array.count == 3); + ASSERT(strv_equals(array.values[0], strv_init("item1"))); + ASSERT(strv_equals(array.values[1], strv_init("item2"))); + ASSERT(strv_equals(array.values[2], strv_init("item3"))); + + arena_cleanup(&arena); +} + +// JSON Parser Tests +UNIT_TEST(json_parse_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t json_content = strv_init( + "{\n" + " \"string\": \"value\",\n" + " \"number\": 42,\n" + " \"bool\": true,\n" + " \"null\": null,\n" + " \"array\": [1, 2, 3],\n" + " \"object\": {\n" + " \"nested\": \"nested_value\"\n" + " }\n" + "}" + ); + + json_t *root = json_parse_str(&arena, json_content, JSON_DEFAULT); + ASSERT(root != NULL); + ASSERT(root->type == JSON_OBJECT); + + // Test string + json_t *string_node = json_get(root, strv_init("string")); + ASSERT(json_check(string_node, JSON_STRING)); + ASSERT(strv_equals(string_node->string, strv_init("value"))); + + // Test number + json_t *number_node = json_get(root, strv_init("number")); + ASSERT(json_check(number_node, JSON_NUMBER)); + ASSERT(number_node->number == 42); + + // Test bool + json_t *bool_node = json_get(root, strv_init("bool")); + ASSERT(json_check(bool_node, JSON_BOOL)); + ASSERT(bool_node->boolean == true); + + // Test null + json_t *null_node = json_get(root, strv_init("null")); + ASSERT(json_check(null_node, JSON_NULL)); + + // Test array + json_t *array_node = json_get(root, strv_init("array")); + ASSERT(json_check(array_node, JSON_ARRAY)); + + // Test array contents + int count = 0; + int sum = 0; + json_for(item, array_node) { + ASSERT(json_check(item, JSON_NUMBER)); + sum += (int)item->number; + count++; + } + ASSERT(count == 3); + ASSERT(sum == 6); // 1 + 2 + 3 + + // Test nested object + json_t *object_node = json_get(root, strv_init("object")); + ASSERT(json_check(object_node, JSON_OBJECT)); + + json_t *nested_node = json_get(object_node, strv_init("nested")); + ASSERT(json_check(nested_node, JSON_STRING)); + ASSERT(strv_equals(nested_node->string, strv_init("nested_value"))); + + arena_cleanup(&arena); +} + +UNIT_TEST(json_parse_with_options) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // JSON with comments and trailing commas + strview_t json_content = strv_init( + "{\n" + " \"key1\": \"value1\",\n" + " // This is a comment\n" + " \"key2\": \"value2\",\n" + " \"array\": [\n" + " 1,\n" + " 2,\n" + " 3,\n" // Trailing comma + " ],\n" // Trailing comma + "}" + ); + + // Test with default flags (should allow comments and trailing commas) + json_t *root1 = json_parse_str(&arena, json_content, JSON_DEFAULT); + ASSERT(root1 != NULL); + ASSERT(json_get(root1, strv_init("key1")) != NULL); + ASSERT(json_get(root1, strv_init("key2")) != NULL); + + // Test with NO_COMMENTS and NO_TRAILING_COMMAS flags + json_t *root2 = json_parse_str(&arena, json_content, JSON_NO_COMMENTS | JSON_NO_TRAILING_COMMAS); + + // This should fail parsing due to the strict flags - but the behavior depends on implementation + // Some parsers might ignore the errors, others might return NULL + // We'll check both possibilities + if (root2 != NULL) { + // If parsing succeeded despite strict flags, ensure the content is correct + ASSERT(json_get(root2, strv_init("key1")) != NULL); + // key2 might be missing if comment handling failed + } + + arena_cleanup(&arena); +} + +// XML Parser Tests +UNIT_TEST(xml_parse_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t xml_content = strv_init( + "\n" + "\n" + " \n" + " Test Title\n" + " Test Author\n" + " \n" + " \n" + " Another Title\n" + " \n" + "" + ); + + xml_t xml = xml_parse_str(&arena, xml_content); + ASSERT(xml.root != NULL); + ASSERT(strv_equals(xml.root->key, strv_init("root"))); + + // Find item tags + xmltag_t *item = xml_get_tag(xml.root, strv_init("item"), false); + ASSERT(item != NULL); + + // Check attributes + strview_t id = xml_get_attribute(item, strv_init("id")); + ASSERT(strv_equals(id, strv_init("1"))); + + strview_t type = xml_get_attribute(item, strv_init("type")); + ASSERT(strv_equals(type, strv_init("book"))); + + // Check nested tags + xmltag_t *title = xml_get_tag(item, strv_init("title"), false); + ASSERT(title != NULL); + ASSERT(strv_equals(title->content, strv_init("Test Title"))); + + xmltag_t *author = xml_get_tag(item, strv_init("author"), false); + ASSERT(author != NULL); + ASSERT(strv_equals(author->content, strv_init("Test Author"))); + + // Check recursive tag finding + xmltag_t *title_recursive = xml_get_tag(xml.root, strv_init("title"), true); + ASSERT(title_recursive != NULL); + ASSERT(strv_equals(title_recursive->content, strv_init("Test Title"))); + + arena_cleanup(&arena); +} + +// HTML Parser Tests +UNIT_TEST(html_parse_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + strview_t html_content = strv_init( + "\n" + "\n" + "\n" + " Test Page\n" + "\n" + "\n" + "

Hello World

\n" + "

This is a test.

\n" + "
\n" + "

More content here.

\n" + "
\n" + "\n" + "" + ); + + html_t html = html_parse_str(&arena, html_content); + ASSERT(html.root != NULL); + ASSERT(str_equals(html.root->key, str_init(&arena, "html"))); + + // Find head and body + htmltag_t *head = html_get_tag(html.root, strv_init("head"), false); + ASSERT(head != NULL); + + htmltag_t *body = html_get_tag(html.root, strv_init("body"), false); + ASSERT(body != NULL); + + // Find title in head + htmltag_t *title = html_get_tag(head, strv_init("title"), false); + ASSERT(title != NULL); + ASSERT(strv_equals(title->content, strv_init("Test Page"))); + + // Find elements in body + htmltag_t *h1 = html_get_tag(body, strv_init("h1"), false); + ASSERT(h1 != NULL); + ASSERT(strv_equals(h1->content, strv_init("Hello World"))); + + // Find paragraph with class + htmltag_t *p = html_get_tag(body, strv_init("p"), false); + ASSERT(p != NULL); + + strview_t p_class = html_get_attribute(p, strv_init("class")); + ASSERT(strv_equals(p_class, strv_init("intro"))); + ASSERT(strv_equals(p->content, strv_init("This is a test."))); + + // Find div by id + htmltag_t *div = html_get_tag(body, strv_init("div"), false); + ASSERT(div != NULL); + + strview_t div_id = html_get_attribute(div, strv_init("id")); + ASSERT(strv_equals(div_id, strv_init("content"))); + + // Find nested paragraph using recursive search + htmltag_t *nested_p = html_get_tag(div, strv_init("p"), false); + ASSERT(nested_p != NULL); + ASSERT(strv_equals(nested_p->content, strv_init("More content here."))); + + arena_cleanup(&arena); +} diff --git a/tests/pretty_print_tests.c b/tests/pretty_print_tests.c new file mode 100644 index 0000000..6b63911 --- /dev/null +++ b/tests/pretty_print_tests.c @@ -0,0 +1,33 @@ +#include "runner.h" +#include "../colla.h" +#include + +UNIT_TEST(pretty_print_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // The pretty_print function outputs to console, so we can't easily verify its exact output + // Instead, we'll just verify it doesn't crash + pretty_print(arena, "Hello, World!"); + + // Test with formatting + pretty_print(arena, "Value: %d, String: %s", 42, "test"); + + arena_cleanup(&arena); +} + +// Helper function to test variadic function +void test_pretty_printv(arena_t arena, const char *fmt, ...) { + va_list args; + va_start(args, fmt); + pretty_printv(arena, fmt, args); + va_end(args); +} + +UNIT_TEST(pretty_printv) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + // Test the helper function + test_pretty_printv(arena, "Test %d %s", 42, "variadic"); + + arena_cleanup(&arena); +} diff --git a/tests/runner.h b/tests/runner.h new file mode 100644 index 0000000..ffc9655 --- /dev/null +++ b/tests/runner.h @@ -0,0 +1,45 @@ +#pragma once + +#include "../colla.h" + +typedef struct unit_test_t unit_test_t; +struct unit_test_t { + strview_t fname; + strview_t name; + void (*fn)(void); + unit_test_t *next; +}; + +extern unit_test_t *test_head; +extern unit_test_t *test_tail; +extern const char *last_fail_reason; +extern bool last_failed; + +void ut_register(const char *file, const char *name, void (*fn)(void)); + +// #pragma data_seg(".CRT$XCU") + +#if COLLA_WIN + #define INITIALIZER(f) \ + static void f(void); \ + __declspec(allocate(".CRT$XCU")) void (*f##_)(void) = f; \ + __pragma(comment(linker,"/include:" #f "_")) \ + static void f(void) +#else + #define INITIALIZER(f) \ + __attribute__((constructor)) static void f(void) +#endif + +#define UNIT_TEST(name) \ + void ut__test_##name(void); \ + INITIALIZER(ut__register_##name) { ut_register(__FILE__, #name, ut__test_##name); } \ + void ut__test_##name(void) + + +#define ASSERT(cond) \ + if (!(cond)) { \ + last_fail_reason = "assert(" COLLA_STRINGIFY(cond) ") at " COLLA_STRINGIFY(__LINE__); \ + last_failed = true; \ + return; \ + } + diff --git a/tests/str_tests.c b/tests/str_tests.c new file mode 100644 index 0000000..3e2576a --- /dev/null +++ b/tests/str_tests.c @@ -0,0 +1,456 @@ +#include "runner.h" +#include "../colla.h" +#include + +// String (str_t) Tests +UNIT_TEST(str_init_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + str_t s = str_init(&arena, "hello"); + + ASSERT(s.len == 5); + ASSERT(s.buf != NULL); + ASSERT(memcmp(s.buf, "hello", 5) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_init_len) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + str_t s = str_init_len(&arena, "hello world", 5); + + ASSERT(s.len == 5); + ASSERT(s.buf != NULL); + ASSERT(memcmp(s.buf, "hello", 5) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_fmt) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + str_t s = str_fmt(&arena, "Number: %d, String: %s", 42, "test"); + + ASSERT(s.buf != NULL); + ASSERT(memcmp(s.buf, "Number: 42, String: test", s.len) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_equals) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s1 = str_init(&arena, "hello"); + str_t s2 = str_init(&arena, "hello"); + str_t s3 = str_init(&arena, "world"); + + ASSERT(str_equals(s1, s2) == true); + ASSERT(str_equals(s1, s3) == false); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_compare) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s1 = str_init(&arena, "abc"); + str_t s2 = str_init(&arena, "abc"); + str_t s3 = str_init(&arena, "abd"); + str_t s4 = str_init(&arena, "abb"); + + ASSERT(str_compare(s1, s2) == 0); + ASSERT(str_compare(s1, s3) < 0); + ASSERT(str_compare(s1, s4) > 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_dup) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s1 = str_init(&arena, "test"); + str_t s2 = str_dup(&arena, s1); + + ASSERT(s1.len == s2.len); + ASSERT(s1.buf != s2.buf); // Different memory locations + ASSERT(memcmp(s1.buf, s2.buf, s1.len) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_cat) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s1 = str_init(&arena, "hello "); + str_t s2 = str_init(&arena, "world"); + str_t s3 = str_cat(&arena, s1, s2); + + ASSERT(s3.len == s1.len + s2.len); + ASSERT(memcmp(s3.buf, "hello world", s3.len) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_is_empty) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s1 = str_init(&arena, "test"); + str_t s2 = STR_EMPTY; + + ASSERT(str_is_empty(s1) == false); + ASSERT(str_is_empty(s2) == true); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_lower_upper) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s1 = str_init(&arena, "TeSt"); + str_lower(&s1); + ASSERT(memcmp(s1.buf, "test", 4) == 0); + + str_t s2 = str_init(&arena, "TeSt"); + str_upper(&s2); + ASSERT(memcmp(s2.buf, "TEST", 4) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_replace) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s = str_init(&arena, "hello"); + str_replace(&s, 'l', 'x'); + ASSERT(memcmp(s.buf, "hexxo", 5) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(str_sub) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s = str_init(&arena, "hello world"); + strview_t sv = str_sub(s, 6, 11); + + ASSERT(sv.len == 5); + ASSERT(memcmp(sv.buf, "world", 5) == 0); + + arena_cleanup(&arena); +} + +// String View (strview_t) Tests +UNIT_TEST(strv_init) { + strview_t sv = strv_init("hello"); + + ASSERT(sv.len == 5); + ASSERT(sv.buf != NULL); + ASSERT(memcmp(sv.buf, "hello", 5) == 0); +} + +UNIT_TEST(strv_init_len) { + strview_t sv = strv_init_len("hello world", 5); + + ASSERT(sv.len == 5); + ASSERT(sv.buf != NULL); + ASSERT(memcmp(sv.buf, "hello", 5) == 0); +} + +UNIT_TEST(strv_init_str) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + + str_t s = str_init(&arena, "hello"); + strview_t sv = strv_init_str(s); + + ASSERT(sv.len == s.len); + ASSERT(sv.buf == s.buf); + + arena_cleanup(&arena); +} + +UNIT_TEST(strv_is_empty) { + strview_t sv1 = strv_init("test"); + strview_t sv2 = STRV_EMPTY; + + ASSERT(strv_is_empty(sv1) == false); + ASSERT(strv_is_empty(sv2) == true); +} + +UNIT_TEST(strv_equals) { + strview_t sv1 = strv_init("hello"); + strview_t sv2 = strv_init("hello"); + strview_t sv3 = strv_init("world"); + + ASSERT(strv_equals(sv1, sv2) == true); + ASSERT(strv_equals(sv1, sv3) == false); +} + +UNIT_TEST(strv_compare) { + strview_t sv1 = strv_init("abc"); + strview_t sv2 = strv_init("abc"); + strview_t sv3 = strv_init("abd"); + strview_t sv4 = strv_init("abb"); + + ASSERT(strv_compare(sv1, sv2) == 0); + ASSERT(strv_compare(sv1, sv3) < 0); + ASSERT(strv_compare(sv1, sv4) > 0); +} + +UNIT_TEST(strv_front_back) { + strview_t sv = strv_init("hello"); + + ASSERT(strv_front(sv) == 'h'); + ASSERT(strv_back(sv) == 'o'); +} + +UNIT_TEST(strv_remove_prefix_suffix) { + strview_t sv = strv_init("hello"); + + strview_t prefix_removed = strv_remove_prefix(sv, 2); + ASSERT(prefix_removed.len == 3); + ASSERT(memcmp(prefix_removed.buf, "llo", 3) == 0); + + strview_t suffix_removed = strv_remove_suffix(sv, 2); + ASSERT(suffix_removed.len == 3); + ASSERT(memcmp(suffix_removed.buf, "hel", 3) == 0); +} + +UNIT_TEST(strv_trim) { + strview_t sv1 = strv_init(" hello "); + strview_t sv2 = strv_init(" hello"); + strview_t sv3 = strv_init("hello "); + + strview_t trimmed1 = strv_trim(sv1); + strview_t trimmed2 = strv_trim_left(sv2); + strview_t trimmed3 = strv_trim_right(sv3); + + ASSERT(trimmed1.len == 5); + ASSERT(memcmp(trimmed1.buf, "hello", 5) == 0); + + ASSERT(trimmed2.len == 5); + ASSERT(memcmp(trimmed2.buf, "hello", 5) == 0); + + ASSERT(trimmed3.len == 5); + ASSERT(memcmp(trimmed3.buf, "hello", 5) == 0); +} + +UNIT_TEST(strv_sub) { + strview_t sv = strv_init("hello world"); + strview_t sub = strv_sub(sv, 6, 11); + + ASSERT(sub.len == 5); + ASSERT(memcmp(sub.buf, "world", 5) == 0); +} + +UNIT_TEST(strv_starts_ends_with) { + strview_t sv = strv_init("hello"); + + ASSERT(strv_starts_with(sv, 'h') == true); + ASSERT(strv_starts_with(sv, 'e') == false); + + ASSERT(strv_ends_with(sv, 'o') == true); + ASSERT(strv_ends_with(sv, 'l') == false); + + strview_t prefix = strv_init("hel"); + strview_t suffix = strv_init("llo"); + + ASSERT(strv_starts_with_view(sv, prefix) == true); + ASSERT(strv_ends_with_view(sv, suffix) == true); +} + +UNIT_TEST(strv_contains) { + strview_t sv = strv_init("hello world"); + + ASSERT(strv_contains(sv, 'e') == true); + ASSERT(strv_contains(sv, 'z') == false); + + strview_t sub = strv_init("world"); + ASSERT(strv_contains_view(sv, sub) == true); + + strview_t chars = strv_init("xyz"); + ASSERT(strv_contains_either(sv, chars) == false); + + strview_t chars2 = strv_init("xyo"); + ASSERT(strv_contains_either(sv, chars2) == true); +} + +UNIT_TEST(strv_find) { + strview_t sv = strv_init("hello world"); + + ASSERT(strv_find(sv, 'o', 0) == 4); + ASSERT(strv_find(sv, 'o', 5) == 7); + ASSERT(strv_find(sv, 'z', 0) == STR_NONE); + + strview_t sub = strv_init("world"); + ASSERT(strv_find_view(sv, sub, 0) == 6); + + strview_t chars = strv_init("xwo"); + ASSERT(strv_find_either(sv, chars, 0) == 4); // 'w' at position 6 +} + +UNIT_TEST(strv_rfind) { + strview_t sv = strv_init("hello world"); + + ASSERT(strv_rfind(sv, 'o', 0) == 7); + ASSERT(strv_rfind(sv, 'o', 5) == 4); + ASSERT(strv_rfind(sv, 'z', 0) == STR_NONE); + + strview_t sub = strv_init("world"); + ASSERT(strv_rfind_view(sv, sub, 0) == 6); +} + +// Character Functions Tests +UNIT_TEST(char_functions) { + ASSERT(char_is_space(' ') == true); + ASSERT(char_is_space('\t') == true); + ASSERT(char_is_space('a') == false); + + ASSERT(char_is_alpha('a') == true); + ASSERT(char_is_alpha('Z') == true); + ASSERT(char_is_alpha('1') == false); + + ASSERT(char_is_num('0') == true); + ASSERT(char_is_num('9') == true); + ASSERT(char_is_num('a') == false); + + ASSERT(char_lower('A') == 'a'); + ASSERT(char_lower('a') == 'a'); + ASSERT(char_lower('1') == '1'); +} + +// Input Stream Tests +UNIT_TEST(instream_basic) { + strview_t sv = strv_init("hello world"); + instream_t is = istr_init(sv); + + ASSERT(istr_get(&is) == 'h'); + ASSERT(istr_get(&is) == 'e'); + ASSERT(istr_peek(&is) == 'l'); + ASSERT(istr_peek_next(&is) == 'l'); + ASSERT(istr_get(&is) == 'l'); + ASSERT(istr_prev(&is) == 'l'); + ASSERT(istr_prev_prev(&is) == 'e'); + + istr_skip(&is, 2); + ASSERT(istr_peek(&is) == ' '); + + istr_skip_whitespace(&is); + ASSERT(istr_peek(&is) == 'w'); + + istr_rewind(&is); + ASSERT(istr_peek(&is) == 'h'); + + ASSERT(istr_tell(&is) == 0); + ASSERT(istr_remaining(&is) == 11); + ASSERT(istr_is_finished(&is) == false); +} + +UNIT_TEST(instream_ignore) { + strview_t sv = strv_init("hello,world"); + instream_t is = istr_init(sv); + + istr_ignore(&is, ','); + ASSERT(istr_peek(&is) == ','); + + istr_ignore_and_skip(&is, ','); + ASSERT(istr_peek(&is) == 'w'); +} + +UNIT_TEST(instream_get_values) { + strview_t sv = strv_init("true 42 3.14 hello"); + instream_t is = istr_init(sv); + + bool b; + ASSERT(istr_get_bool(&is, &b) == true); + ASSERT(b == true); + + istr_skip_whitespace(&is); + + u32 u; + ASSERT(istr_get_u32(&is, &u) == true); + ASSERT(u == 42); + + istr_skip_whitespace(&is); + + double d; + ASSERT(istr_get_num(&is, &d) == true); + ASSERT(d > 3.13 && d < 3.15); + + istr_skip_whitespace(&is); + + strview_t word = istr_get_view(&is, ' '); + ASSERT(word.len == 5); + ASSERT(memcmp(word.buf, "hello", 5) == 0); +} + +// Output Stream Tests +UNIT_TEST(outstream_basic) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + outstream_t os = ostr_init(&arena); + + ostr_putc(&os, 'h'); + ostr_putc(&os, 'i'); + + ASSERT(ostr_tell(&os) == 2); + ASSERT(ostr_back(&os) == 'i'); + + ostr_print(&os, " %d", 42); + + strview_t result = ostr_as_view(&os); + ASSERT(result.len == 5); + ASSERT(memcmp(result.buf, "hi 42", 5) == 0); + + ostr_pop(&os, 3); + result = ostr_as_view(&os); + ASSERT(result.len == 2); + ASSERT(memcmp(result.buf, "hi", 2) == 0); + + ostr_clear(&os); + ASSERT(ostr_tell(&os) == 0); + + arena_cleanup(&arena); +} + +UNIT_TEST(outstream_append) { + arena_t arena = arena_make(ARENA_MALLOC, KB(4)); + outstream_t os = ostr_init(&arena); + + ostr_append_bool(&os, true); + ostr_putc(&os, ' '); + ostr_append_uint(&os, 42); + ostr_putc(&os, ' '); + ostr_append_int(&os, -10); + ostr_putc(&os, ' '); + ostr_append_num(&os, 3.14); + + str_t result = ostr_to_str(&os); + ASSERT(result.len > 0); + + arena_cleanup(&arena); +} + +// Binary Input Stream Tests +UNIT_TEST(binary_stream) { + u8 data[] = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08}; + buffer_t buffer = {data, sizeof(data)}; + + ibstream_t bs = ibstr_init(buffer); + + ASSERT(ibstr_remaining(&bs) == 8); + ASSERT(ibstr_tell(&bs) == 0); + ASSERT(ibstr_is_finished(&bs) == false); + + u8 val8; + ASSERT(ibstr_get_u8(&bs, &val8) == true); + ASSERT(val8 == 0x01); + + u16 val16; + ASSERT(ibstr_get_u16(&bs, &val16) == true); + ASSERT(val16 == 0x0302); // Assuming little-endian + + ibstr_skip(&bs, 1); + + u32 val32; + ASSERT(ibstr_get_u32(&bs, &val32) == true); + ASSERT(val32 == 0x08070605); // Assuming little-endian + + ASSERT(ibstr_is_finished(&bs) == true); +} diff --git a/tests/string_tests.c b/tests/string_tests.c new file mode 100644 index 0000000..b40e67a --- /dev/null +++ b/tests/string_tests.c @@ -0,0 +1,11 @@ +#include "runner.h" +#include "../colla.h" + +UNIT_TEST(str_format) { + arena_t arena = arena_make(ARENA_MALLOC, KB(1)); + str_t s = str_fmt(&arena, "%d %s", 42, "test"); + str_t lit = str(&arena, "42 test"); + ASSERT(str_equals(s, lit)); + arena_cleanup(&arena); +} + diff --git a/tools/nob.c b/tools/nob.c new file mode 100644 index 0000000..f4b0c87 --- /dev/null +++ b/tools/nob.c @@ -0,0 +1,478 @@ +#define COLLA_NO_CONDITION_VARIABLE 1 +#define COLLA_NO_NET 1 + +#include "../colla.c" +#include +#include + +#if COLLA_TCC + +WINBASEAPI LPCH WINAPI GetEnvironmentStringsj(VOID); +WINBASEAPI LPWCH WINAPI GetEnvironmentStringsW(VOID); +WINBASEAPI BOOL WINAPI FreeEnvironmentStringsA(LPCH penv); +WINBASEAPI BOOL WINAPI FreeEnvironmentStringsW(LPWCH penv); +#ifdef UNICODE + #define GetEnvironmentStrings GetEnvironmentStringsW + #define FreeEnvironmentStrings FreeEnvironmentStringsW +#else + #define GetEnvironmentStrings GetEnvironmentStringsA + #define FreeEnvironmentStrings FreeEnvironmentStringsA +#endif + +#endif + +int strv_to_int(strview_t strv) { + instream_t in = istr_init(strv); + i32 value = 0; + if (!istr_get_i32(&in, &value)) { + return 0; + } + return value; +} + +str_t find_vcvars_path(arena_t *arena) { + strview_t base_path = strv("C:/Program Files/Microsoft Visual Studio"); + // find year + int year = 0; + { + arena_t tmp = *arena; + + dir_t *dir = os_dir_open(&tmp, base_path); + if (!os_dir_is_valid(dir)) { + err("couldn't open directory (%v)", base_path); + return STR_EMPTY; + } + + dir_foreach(&tmp, entry, dir) { + if (entry->type != DIRTYPE_DIR) continue; + + int number = strv_to_int(strv(entry->name)); + if (number > year) year = number; + } + } + + if (year == 0) { + err("couldn't find visual studio year version"); + return STR_EMPTY; + } + + str_t path_with_year = str_fmt(arena, "%v/%d", base_path, year); + + // find edition + const char *editions[] = { + "Enterprise", + "Professional", + "Community", + }; + + int edition = 0; + + for (; edition < arrlen(editions); ++edition) { + arena_t tmp = *arena; + str_t path = str_fmt(&tmp, "%v/%s", path_with_year, editions[edition]); + if (os_dir_exists(strv(path))) { + break; + } + } + + if (edition >= arrlen(editions)) { + err("couldn't find visual studio edition"); + return STR_EMPTY; + } + + str_t vcvars = str_fmt(arena, "%v/%s/VC/Auxiliary/Build/vcvars64.bat", path_with_year, editions[edition]); + + return vcvars; +} + +bool load_cache(arena_t *arena) { + if (!os_file_exists(strv("build/cache.ini"))) { + err("build/cache.ini doesn't exist"); + return false; + } + + { + arena_t scratch = *arena; + + ini_t ini = ini_parse(&scratch, strv("build/cache.ini"), &(iniopt_t){ .comment_vals = strv("#") }); + initable_t *root = ini_get_table(&ini, INI_ROOT); + if (!root) fatal("fail"); + + for_each (val, root->values) { + os_set_env_var(scratch, val->key, val->value); + } + } + + return true; +} + +typedef enum optimise_level_e { + OPTIMISE_NONE, + OPTIMISE_FAST, + OPTIMISE_SMALL, + OPTIMISE__COUNT, +} optimise_level_e; + +typedef enum warning_level_e { + WARNING_NONE, + WARNING_DEFAULT, + WARNING_ALL, + WARNING__COUNT, +} warning_level_e; + +typedef enum sanitiser_e { + SANITISER_NONE, + SANITISER_ADDRESS, + SANITISER__COUNT, +} sanitiser_e; + +typedef enum cversion_e { + CVERSION_LATEST, + CVERSION_17, + CVERSION_11, + CVERSION__COUNT +} cversion_e; + +typedef struct options_t options_t; +struct options_t { + strview_t input_fname; + strview_t out_fname; + optimise_level_e optimisation; + warning_level_e warnings; + bool warnings_as_error; + sanitiser_e sanitiser; + bool fast_math; + bool debug; + strv_list_t *defines; + cversion_e cstd; + bool run; + strv_list_t *run_args; + bool is_cpp; +}; + +void print_help_message(void) { + puts("usage:"); + puts(" -r / -run [input.c] [args...] compiles and runs , forwards "); + puts(" -h / -help print this message"); + puts(" -o / -out [filename] output filename (default: build/.exe)"); + puts(" -O / -optimise [fast,small] optimisation level"); + puts(" -w / -warning [default,all] warning level"); + puts(" -werror treat warnings as errors"); + puts(" -fsanitize [address] turn on sanitiser"); + puts(" -fastmath turn on fast math"); + puts(" -g / -debug generate debug information"); + puts(" -D / -define [key=value,key] add a preprocessor define "); + puts(" -std [c11,c17,clatest] select c standard (default: clatest)"); + puts(" -cpp compile c++ instead of c"); + exit(0); +} + +optimise_level_e get_optimisation_level(strview_t arg) { + if (strv_equals(arg, strv("fast"))) { + return OPTIMISE_FAST; + } + else if (strv_equals(arg, strv("small"))) { + return OPTIMISE_SMALL; + } + warn("unrecognised optimisation level: (%v)", arg); + return OPTIMISE_NONE; +} + +warning_level_e get_warning_level(strview_t arg) { + if (strv_equals(arg, strv("default"))) { + return WARNING_DEFAULT; + } + else if (strv_equals(arg, strv("all"))) { + return WARNING_ALL; + } + warn("unrecognised warning level: (%v)", arg); + return WARNING_NONE; +} + +sanitiser_e get_sanitiser(strview_t arg) { + if (strv_equals(arg, strv("address"))) { + return SANITISER_ADDRESS; + } + warn("unrecognised sanitiser: (%v)", arg); + return SANITISER_NONE; +} + +cversion_e get_cversion(strview_t arg) { + if (strv_equals(arg, strv("clatest"))) { + return CVERSION_LATEST; + } + else if (strv_equals(arg, strv("c17"))) { + return CVERSION_17; + } + else if (strv_equals(arg, strv("c11"))) { + return CVERSION_11; + } + warn("unrecognised c std version: (%v)", arg); + return CVERSION_LATEST; +} + +options_t parse_options(arena_t *arena, int argc, char **argv) { + options_t out = {0}; + + for (int i = 1; i < argc; ++i) { + strview_t arg = strv(argv[i]); + +#define CHECK_OPT_BEG() if (false) {} +#define CHECK_OPT1(opt) else if (strv_equals(arg, strv("-" opt))) +#define CHECK_OPT2(small, big) else if (strv_equals(arg, strv("-" small)) || strv_equals(arg, strv("-" big))) + +#define GET_NEXT_ARG() (i + 1) < argc ? strv(argv[++i]) : STRV_EMPTY + + CHECK_OPT_BEG() + CHECK_OPT2("h", "help") { + print_help_message(); + } + CHECK_OPT2("o", "out") { + strview_t out_fname = GET_NEXT_ARG(); + str_t out_fname_str = str_fmt(arena, "build/%v", out_fname); + out.out_fname = strv(out_fname_str); + } + CHECK_OPT2("O", "optimise") { + out.optimisation = get_optimisation_level(GET_NEXT_ARG()); + } + CHECK_OPT2("w", "warning") { + out.warnings = get_warning_level(GET_NEXT_ARG()); + } + CHECK_OPT1("werror") { + out.warnings_as_error = true; + } + CHECK_OPT1("fsanitize") { + out.sanitiser = get_sanitiser(GET_NEXT_ARG()); + } + CHECK_OPT1("fastmath") { + out.fast_math = true; + } + CHECK_OPT2("g", "debug") { + out.debug = true; + } + CHECK_OPT2("D", "define") { + darr_push(arena, out.defines, GET_NEXT_ARG()); + } + CHECK_OPT1("std") { + out.cstd = get_cversion(GET_NEXT_ARG()); + } + CHECK_OPT1("cpp") { + out.is_cpp = true; + } + CHECK_OPT2("r", "run") { + out.run = true; + out.input_fname = GET_NEXT_ARG(); + for (i += 1; i < argc; ++i) { + darr_push(arena, out.run_args, strv(argv[i])); + } + } + else { + out.input_fname = arg; + } + } + +#undef CHECK_OPT_BEG +#undef CHECK_OPT1 +#undef CHECK_OPT2 +#undef GET_NEXT_ARG + + if (strv_is_empty(out.out_fname)) { + strview_t name; + os_file_split_path(out.input_fname, NULL, &name, NULL); + str_t out_fname = str_fmt(arena, "build\\%v", name); + out.out_fname = strv(out_fname); + } + + return out; +} + +void cl_print_line(arena_t scratch, strview_t line, strview_t substr, strview_t col) { + usize sub_beg = strv_find_view(line, substr, 0); + + strview_t file_part = strv_trim(strv_sub(line, 0, sub_beg)); + strview_t sub_part = strv_sub(line, sub_beg, STR_END); + + usize last_beg = strv_find(sub_part, ':', 0) + 1; + + strview_t last_part = strv_trim(strv_sub(sub_part, last_beg, STR_END)); + sub_part = strv_sub(sub_part, 0, last_beg); + + pretty_print(scratch, + "<%v>%v\n" + " %v %v\n", + col, + sub_part, + file_part, + last_part + ); +} + +int main(int argc, char **argv) { + os_init(); + + arena_t arena = arena_make(ARENA_VIRTUAL, GB(1)); + + if (argc < 2) { + print_help_message(); + } + + options_t opt = parse_options(&arena, argc, argv); + + if (!os_dir_exists(strv("build/"))) { + info("creating build folder"); + _mkdir("build"); + } + + if (!os_file_exists(strv("build/cache.ini"))) { + info("couldn't find cache.ini, creating it now"); + + arena_t scratch = arena; + str_t vcvars_path = find_vcvars_path(&scratch); + + os_cmd_t *cmd = NULL; + darr_push(&scratch, cmd, strv(vcvars_path)); + darr_push(&scratch, cmd, strv("&&")); + darr_push(&scratch, cmd, strv("set")); + darr_push(&scratch, cmd, strv(">")); + darr_push(&scratch, cmd, strv("build\\cache.ini")); + + if (!os_run_cmd(scratch, cmd, NULL)) { + fatal("failed to run vcvars64.bat"); + os_abort(1); + } + } + + { + arena_t scratch = arena; + + if (!load_cache(&scratch)) { + os_abort(1); + } + + os_cmd_t *cmd = NULL; + + darr_push(&scratch, cmd, strv("cl")); + darr_push(&scratch, cmd, strv("/nologo")); + darr_push(&scratch, cmd, strv("/utf-8")); + if (!opt.is_cpp) { + darr_push(&scratch, cmd, strv("/TC")); + } + + str_t output = str_fmt(&scratch, "/Fe:%v.exe", opt.out_fname); + str_t object = str_fmt(&scratch, "/Fo:%v.obj", opt.out_fname); + darr_push(&scratch, cmd, strv(output)); + darr_push(&scratch, cmd, strv(object)); + + strview_t optimisations[OPTIMISE__COUNT] = { + strv("/Od"), // disabled + strv("/O2"), // fast code + strv("/O1"), // small code + }; + darr_push(&scratch, cmd, optimisations[opt.optimisation]); + + strview_t warnings[WARNING__COUNT] = { + strv("/W0"), + strv("/W3"), + strv("/W4"), + }; + darr_push(&scratch, cmd, warnings[opt.warnings]); + + if (opt.warnings_as_error) { + darr_push(&scratch, cmd, strv("/WX")); + } + + if (opt.sanitiser) { + strview_t sanitisers[SANITISER__COUNT] = { + strv(""), + strv("/fsanitize=address"), + }; + darr_push(&scratch, cmd, sanitisers[opt.sanitiser]); + } + + if (opt.fast_math) { + darr_push(&scratch, cmd, strv("/fp:fast")); + } + + if (opt.debug) { + darr_push(&scratch, cmd, strv("/Zi")); + darr_push(&scratch, cmd, strv("/D_DEBUG")); + } + + for_each (def, opt.defines) { + for (int i = 0; i < def->count; ++i) { + str_t define = str_fmt(&scratch, "/D%v", def->items[i]); + darr_push(&scratch, cmd, strv(define)); + } + } + + strview_t cversion[CVERSION__COUNT] = { + strv("clatest"), + strv("c17"), + strv("c11"), + }; + + str_t cstd = str_fmt(&scratch, "/std:%v", cversion[opt.cstd]); + darr_push(&scratch, cmd, strv(cstd)); + + darr_push(&scratch, cmd, opt.input_fname); + + // /LD -> create dynamic lib + // /LDd -> create debug dynamic lib + // /link + + oshandle_t hout = os_handle_zero(); + + bool compilation_result = os_run_cmd( + scratch, + cmd, + &(os_cmd_options_t){ .out = &hout, } + ); + + str_t result = os_file_read_all_str_fp(&scratch, hout); + + instream_t in = istr_init(strv(result)); + + while (!istr_is_finished(&in)) { + strview_t line = istr_get_line(&in); + if (strv_contains_view(line, strv("error"))) { + cl_print_line(scratch, line, strv("error"), strv("red")); + } + else if (strv_contains_view(line, strv("warning"))) { + cl_print_line(scratch, line, strv("warning"), strv("yellow")); + } + else { + if (compilation_result) { + pretty_print(scratch, "compiled: %v\n", line); + } + else { + pretty_print(scratch, "while compiling: %v\n", line); + } + } + } + + if (!compilation_result) { + return 1; + } + } + + if (opt.run) { + arena_t scratch = arena; + os_cmd_t *cmd = NULL; + + darr_push(&scratch, cmd, opt.out_fname); + + for_each (arg, opt.run_args) { + for (int i = 0; i < arg->count; ++i) { + darr_push(&scratch, cmd, arg->items[i]); + } + } + + if (!os_run_cmd(scratch, cmd, NULL)) { + return 1; + } + } + + arena_cleanup(&arena); + + os_cleanup(); +} diff --git a/tools/noblin.c b/tools/noblin.c new file mode 100644 index 0000000..40cc4c7 --- /dev/null +++ b/tools/noblin.c @@ -0,0 +1,308 @@ +#define COLLA_NO_CONDITION_VARIABLE 1 +#define COLLA_NO_NET 1 + +#include "../colla.c" + +typedef enum optimise_level_e { + OPTIMISE_NONE, + OPTIMISE_FAST, + OPTIMISE_SMALL, + OPTIMISE__COUNT, +} optimise_level_e; + +typedef enum warning_level_e { + WARNING_NONE, + WARNING_DEFAULT, + WARNING_ALL, + WARNING__COUNT, +} warning_level_e; + +typedef enum sanitiser_e { + SANITISER_NONE, + SANITISER_ADDRESS, + SANITISER__COUNT, +} sanitiser_e; + +typedef enum cversion_e { + CVERSION_LATEST, + CVERSION_17, + CVERSION_11, + CVERSION__COUNT +} cversion_e; + +typedef struct options_t options_t; +struct options_t { + strview_t input_fname; + strview_t out_fname; + optimise_level_e optimisation; + warning_level_e warnings; + bool warnings_as_error; + sanitiser_e sanitiser; + bool fast_math; + bool debug; + strv_list_t *defines; + cversion_e cstd; + bool run; + strv_list_t *run_args; + bool is_cpp; +}; + +void print_help_message(void) { + puts("usage:"); + puts(" -r / -run [input.c] [args...] compiles and runs , forwards "); + puts(" -h / -help print this message"); + puts(" -o / -out [filename] output filename (default: build/.exe)"); + puts(" -O / -optimise [fast,small] optimisation level"); + puts(" -w / -warning [default,all] warning level"); + puts(" -werror treat warnings as errors"); + puts(" -fsanitize [address] turn on sanitiser"); + puts(" -fastmath turn on fast math"); + puts(" -g / -debug generate debug information"); + puts(" -D / -define [key=value,key] add a preprocessor define "); + puts(" -std [c11,c17,clatest] select c standard (default: clatest)"); + puts(" -cpp compile c++ instead of c"); + exit(0); +} + +optimise_level_e get_optimisation_level(strview_t arg) { + if (strv_equals(arg, strv("fast"))) { + return OPTIMISE_FAST; + } + else if (strv_equals(arg, strv("small"))) { + return OPTIMISE_SMALL; + } + warn("unrecognised optimisation level: (%v)", arg); + return OPTIMISE_NONE; +} + +warning_level_e get_warning_level(strview_t arg) { + if (strv_equals(arg, strv("default"))) { + return WARNING_DEFAULT; + } + else if (strv_equals(arg, strv("all"))) { + return WARNING_ALL; + } + warn("unrecognised warning level: (%v)", arg); + return WARNING_NONE; +} + +sanitiser_e get_sanitiser(strview_t arg) { + if (strv_equals(arg, strv("address"))) { + return SANITISER_ADDRESS; + } + warn("unrecognised sanitiser: (%v)", arg); + return SANITISER_NONE; +} + +cversion_e get_cversion(strview_t arg) { + if (strv_equals(arg, strv("clatest"))) { + return CVERSION_LATEST; + } + else if (strv_equals(arg, strv("c17"))) { + return CVERSION_17; + } + else if (strv_equals(arg, strv("c11"))) { + return CVERSION_11; + } + warn("unrecognised c std version: (%v)", arg); + return CVERSION_LATEST; +} + +options_t parse_options(arena_t *arena, int argc, char **argv) { + options_t out = {0}; + + for (int i = 1; i < argc; ++i) { + strview_t arg = strv(argv[i]); + +#define CHECK_OPT_BEG() if (false) {} +#define CHECK_OPT1(opt) else if (strv_equals(arg, strv("-" opt))) +#define CHECK_OPT2(small, big) else if (strv_equals(arg, strv("-" small)) || strv_equals(arg, strv("-" big))) + +#define GET_NEXT_ARG() (i + 1) < argc ? strv(argv[++i]) : STRV_EMPTY + + CHECK_OPT_BEG() + CHECK_OPT2("h", "help") { + print_help_message(); + } + CHECK_OPT2("o", "out") { + strview_t out_fname = GET_NEXT_ARG(); + str_t out_fname_str = str_fmt(arena, "build/%v", out_fname); + out.out_fname = strv(out_fname_str); + } + CHECK_OPT2("O", "optimise") { + out.optimisation = get_optimisation_level(GET_NEXT_ARG()); + } + CHECK_OPT2("w", "warning") { + out.warnings = get_warning_level(GET_NEXT_ARG()); + } + CHECK_OPT1("werror") { + out.warnings_as_error = true; + } + CHECK_OPT1("fsanitize") { + out.sanitiser = get_sanitiser(GET_NEXT_ARG()); + } + CHECK_OPT1("fastmath") { + out.fast_math = true; + } + CHECK_OPT2("g", "debug") { + out.debug = true; + } + CHECK_OPT2("D", "define") { + darr_push(arena, out.defines, GET_NEXT_ARG()); + } + CHECK_OPT1("std") { + out.cstd = get_cversion(GET_NEXT_ARG()); + } + CHECK_OPT1("cpp") { + out.is_cpp = true; + } + CHECK_OPT2("r", "run") { + out.run = true; + out.input_fname = GET_NEXT_ARG(); + for (i += 1; i < argc; ++i) { + darr_push(arena, out.run_args, strv(argv[i])); + } + } + else { + out.input_fname = arg; + } + } + +#undef CHECK_OPT_BEG +#undef CHECK_OPT1 +#undef CHECK_OPT2 +#undef GET_NEXT_ARG + + if (strv_is_empty(out.out_fname)) { + strview_t name; + os_file_split_path(out.input_fname, NULL, &name, NULL); + str_t out_fname = str_fmt(arena, "build\\%v", name); + out.out_fname = strv(out_fname); + } + + return out; +} + +int main(int argc, char **argv) { + os_init(); + + arena_t arena = arena_make(ARENA_VIRTUAL, GB(1)); + + if (argc < 2) { + print_help_message(); + } + + options_t opt = parse_options(&arena, argc, argv); + + if (!os_dir_exists(strv("build/"))) { + info("creating build folder"); + mkdir("build", 755); + } + + { + arena_t scratch = arena; + os_cmd_t *cmd = NULL; + + if (opt.is_cpp) { + darr_push(&scratch, cmd, strv("c++")); + } + else { + darr_push(&scratch, cmd, strv("cc")); + } + + str_t output = str_fmt(&scratch, "-o %v", opt.out_fname); + darr_push(&scratch, cmd, strv(output)); + + strview_t optimisations[OPTIMISE__COUNT] = { + strv("-O0"), // disabled + strv("-O3"), // fast code + strv("-Os"), // small code + }; + darr_push(&scratch, cmd, optimisations[opt.optimisation]); + + strview_t warnings[WARNING__COUNT] = { + strv("-w"), + strv("-Weverything"), + strv("-Wpedantic"), + }; + darr_push(&scratch, cmd, warnings[opt.warnings]); + + if (opt.warnings_as_error) { + darr_push(&scratch, cmd, strv("-Werror")); + } + + if (opt.sanitiser) { + strview_t sanitisers[SANITISER__COUNT] = { + strv(""), + strv("-fsanitize=address"), + }; + darr_push(&scratch, cmd, sanitisers[opt.sanitiser]); + } + + if (opt.fast_math) { + darr_push(&scratch, cmd, strv("-ffast-math")); + } + + if (opt.debug) { + darr_push(&scratch, cmd, strv("-g")); + darr_push(&scratch, cmd, strv("-D_DEBUG")); + } + + for_each (def, opt.defines) { + for (int i = 0; i < def->count; ++i) { + str_t define = str_fmt(&scratch, "-D%v", def->items[i]); + darr_push(&scratch, cmd, strv(define)); + } + } + + strview_t cversion[CVERSION__COUNT] = { + strv("c23"), + strv("c17"), + strv("c11"), + }; + + str_t cstd = str_fmt(&scratch, "-std %v", cversion[opt.cstd]); + darr_push(&scratch, cmd, strv(cstd)); + + darr_push(&scratch, cmd, opt.input_fname); + + oshandle_t hout = os_handle_zero(); + + bool compilation_result = os_run_cmd( + scratch, + cmd, + &(os_cmd_options_t){ .out = &hout, } + ); + + str_t result = os_file_read_all_str_fp(&scratch, hout); + + info("result:\n%v\n", result); + + if (!compilation_result) { + + return 1; + } + } + + if (opt.run) { + arena_t scratch = arena; + os_cmd_t *cmd = NULL; + + darr_push(&scratch, cmd, opt.out_fname); + + for_each (arg, opt.run_args) { + for (int i = 0; i < arg->count; ++i) { + darr_push(&scratch, cmd, arg->items[i]); + } + } + + if (!os_run_cmd(scratch, cmd, NULL)) { + return 1; + } + } + + arena_cleanup(&arena); + + os_cleanup(); + +} diff --git a/tools/unit_tests.c b/tools/unit_tests.c new file mode 100644 index 0000000..9b29c58 --- /dev/null +++ b/tools/unit_tests.c @@ -0,0 +1,82 @@ +#include "../colla.c" + +#if COLLA_WIN +#pragma section(".CRT$XCU", read) +#endif + +#include "../tests/runner.h" + +#include "../tests/arena_tests.c" +#include "../tests/core_tests.c" +#include "../tests/net_tests.c" +#include "../tests/os_tests.c" +// #include "../tests/parsers_tests.c" +// #include "../tests/pretty_print_tests.c" +#include "../tests/str_tests.c" + +unit_test_t *test_head = NULL; +unit_test_t *test_tail = NULL; +const char *last_fail_reason = NULL; +bool last_failed = false; + +void ut_register(const char *file, const char *name, void (*fn)(void)) { + strview_t fname; + os_file_split_path(strv(file), NULL, &fname, NULL); + + fname = strv_remove_suffix(fname, arrlen("_tests") - 1); + + unit_test_t *test = calloc(1, sizeof(unit_test_t)); + test->name = strv(name); + test->fn = fn; + test->fname = fname; + + olist_push(test_head, test_tail, test); +} + +int main() { + colla_init(COLLA_ALL); + arena_t arena = arena_make(ARENA_VIRTUAL, GB(1)); + + strview_t last_file = STRV_EMPTY; + int success = 0; + int total = 0; + + unit_test_t *test = test_head; + while (test) { + if (!strv_equals(test->fname, last_file)) { + last_file = test->fname; + pretty_print(arena, "> %v\n", test->fname); + } + + test->fn(); + + total++; + + if (last_failed) { + pretty_print(arena, "%4s[X] %v: %s\n", "", test->name, last_fail_reason); + } + else { + pretty_print(arena, "%4s[V] %v\n", "", test->name); + success++; + } + + last_failed = false; + + test = test->next; + } + + print("\n"); + + strview_t colors[] = { + cstrv("red"), + cstrv("light_red"), + cstrv("yellow"), + cstrv("light_yellow"), + cstrv("light_green"), + cstrv("green"), + }; + + usize col = success * (arrlen(colors) - 1) / total; + + pretty_print(arena, "<%v>%d/%d tests passed\n", colors[col], success, total); +}