From 503219af891858cba983d9b2ecd22ca50498001f Mon Sep 17 00:00:00 2001 From: Mitya Selivanov Date: Tue, 12 Sep 2023 03:29:53 +0200 Subject: Update header-only --- include/kit.inl.h | 1810 +++++++++++++++++++++++++---------------------------- 1 file changed, 865 insertions(+), 945 deletions(-) (limited to 'include/kit.inl.h') diff --git a/include/kit.inl.h b/include/kit.inl.h index 781de95..e16762b 100644 --- a/include/kit.inl.h +++ b/include/kit.inl.h @@ -41,6 +41,9 @@ *********************************************************************/ #ifndef KIT_TYPES_H #define KIT_TYPES_H +#ifndef _GNU_SOURCE +# define _GNU_SOURCE +#endif typedef signed char i8; typedef signed short i16; typedef signed int i32; @@ -74,7 +77,7 @@ enum { KIT_ERROR_SOCKETS_STARTUP_FAILED, KIT_ERROR_SOCKET_CONTROL_FAILED }; -typedef signed int kit_status_t; +typedef i32 kit_status_t; #ifdef __cplusplus } #endif @@ -86,11 +89,6 @@ typedef signed int kit_status_t; *********************************************************************/ #ifndef KIT_ALLOCATOR_H #define KIT_ALLOCATOR_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include -#include #ifdef __cplusplus extern "C" { #endif @@ -108,11 +106,11 @@ enum { KIT_DEALLOCATE_ALL }; typedef struct { - int type; - ptrdiff_t size; + i32 type; + i64 size; union { - uint8_t *bytes; - void *data; + u8 *bytes; + void *data; }; } kit_allocator_t; // Application should implement this function if custom allocator @@ -120,11 +118,10 @@ typedef struct { // // See KIT_ENABLE_CUSTOM_ALLOC_DISPATCH macro. // -void *kit_alloc_dispatch(kit_allocator_t *alloc, int request, - ptrdiff_t size, ptrdiff_t previous_size, - void *pointer); +void *kit_alloc_dispatch(kit_allocator_t *alloc, i32 request, + i64 size, i64 previous_size, void *poi32er); kit_allocator_t kit_alloc_default(void); -kit_allocator_t kit_alloc_buffer(ptrdiff_t size, void *buffer); +kit_allocator_t kit_alloc_buffer(i64 size, void *buffer); #ifdef __cplusplus } #endif @@ -147,7 +144,7 @@ kit_allocator_t kit_alloc_buffer(ptrdiff_t size, void *buffer); # ifndef WIN32_LEAN_AND_MEAN # define WIN32_LEAN_AND_MEAN 1 # endif -# include +# include # define KIT_TIMESPEC_IMPL_UNIX_EPOCH_IN_TICKS 116444736000000000ull # define KIT_TIMESPEC_IMPL_TICKS_PER_SECONDS 10000000ull static int timespec_get(struct timespec *ts, int base) { @@ -167,107 +164,6 @@ static int timespec_get(struct timespec *ts, int base) { } #endif #endif -/********************************************************************* - * * - * File: source/kit/thread_defs.h * - * * - *********************************************************************/ -#ifndef KIT_THREAD_DEFS_H -#define KIT_THREAD_DEFS_H -#ifdef __cplusplus -extern "C" { -#endif -enum { - thrd_success = 0, - thrd_timedout, - thrd_error, - thrd_busy, - thrd_nomem, - thrd_wrong_stack_size -}; -#ifdef __cplusplus -} -#endif -#endif -/********************************************************************* - * * - * File: source/kit/thread.h * - * * - *********************************************************************/ -#ifndef KIT_THREAD_H -#define KIT_THREAD_H -#ifndef KIT_DISABLE_SYSTEM_THREADS -# ifndef _GNU_SOURCE -# define _GNU_SOURCE -# endif -# include -# include -# if defined(__cplusplus) -# define _Noreturn [[noreturn]] -# elif defined(_MSC_VER) -# define _Noreturn __declspec(noreturn) -# endif -# if !defined(_WIN32) || defined(__CYGWIN__) -# include -# endif -# ifndef _Thread_local -# if defined(__cplusplus) -/* C++11 doesn't need `_Thread_local` keyword or macro */ -# elif !defined(__STDC_NO_THREADS__) -/* threads are optional in C11, _Thread_local present in this - * condition */ -# elif defined(_MSC_VER) -# define _Thread_local __declspec(thread) -# elif defined(__GNUC__) -# define _Thread_local __thread -# else -/* Leave _Thread_local undefined so that use of _Thread_local would - * not promote to a non-thread-local global variable - */ -# endif -# endif -# if !defined(__cplusplus) -/* - * C11 thread_local() macro - * C++11 and above already have thread_local keyword - */ -# ifndef thread_local -# if _MSC_VER -# define thread_local __declspec(thread) -# else -# define thread_local _Thread_local -# endif -# endif -# endif -# ifdef __cplusplus -extern "C" { -# endif -typedef void (*tss_dtor_t)(void *); -typedef int (*thrd_start_t)(void *); -# if defined(_WIN32) && !defined(__CYGWIN__) -typedef struct { - void *handle; -} thrd_t; -typedef unsigned long tss_t; -# else -typedef pthread_t thrd_t; -typedef pthread_key_t tss_t; -# endif -int thrd_create(thrd_t *, thrd_start_t, void *); -int thrd_create_with_stack(thrd_t *, thrd_start_t, void *, - ptrdiff_t stack_size); -thrd_t thrd_current(void); -int thrd_detach(thrd_t); -int thrd_equal(thrd_t, thrd_t); -_Noreturn void thrd_exit(int); -int thrd_join(thrd_t, int *); -int thrd_sleep(const struct timespec *, struct timespec *); -void thrd_yield(void); -# ifdef __cplusplus -} -# endif -#endif -#endif /********************************************************************* * * * File: source/kit/atomic.h * @@ -275,16 +171,13 @@ void thrd_yield(void); *********************************************************************/ #ifndef KIT_ATOMIC_H #define KIT_ATOMIC_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include #ifndef _MSC_VER # include # define KIT_ATOMIC(type_) type_ _Atomic #else # include # define KIT_ATOMIC(type_) type_ volatile +# define _Atomic volatile # ifdef __cplusplus extern "C" { # endif @@ -296,115 +189,160 @@ enum { memory_order_acq_rel, memory_order_seq_cst }; -void kit_atomic_store_explicit_8(uint8_t volatile *var, uint8_t value, - int memory_order); -void kit_atomic_store_explicit_16(uint16_t volatile *var, - uint16_t value, int memory_order); -void kit_atomic_store_explicit_32(uint32_t volatile *var, - uint32_t value, int memory_order); -void kit_atomic_store_explicit_64(uint64_t volatile *var, - uint64_t value, int memory_order); -uint8_t kit_atomic_load_explicit_8(uint8_t volatile *var, - int memory_order); -uint16_t kit_atomic_load_explicit_16(uint16_t volatile *var, - int memory_order); -uint32_t kit_atomic_load_explicit_32(uint32_t volatile *var, - int memory_order); -uint64_t kit_atomic_load_explicit_64(uint64_t volatile *var, - int memory_order); -uint8_t kit_atomic_exchange_explicit_8(uint8_t volatile *var, - uint8_t value, - int memory_order); -uint16_t kit_atomic_exchange_explicit_16(uint16_t volatile *var, - uint16_t value, - int memory_order); -uint32_t kit_atomic_exchange_explicit_32(uint32_t volatile *var, - uint32_t value, - int memory_order); -uint64_t kit_atomic_exchange_explicit_64(uint64_t volatile *var, - uint64_t value, - int memory_order); -uint8_t kit_atomic_fetch_add_explicit_8(uint8_t volatile *var, - uint8_t value, - int memory_order); -uint16_t kit_atomic_fetch_add_explicit_16(uint16_t volatile *var, - uint16_t value, - int memory_order); -uint32_t kit_atomic_fetch_add_explicit_32(uint32_t volatile *var, - uint32_t value, - int memory_order); -uint64_t kit_atomic_fetch_add_explicit_64(uint64_t volatile *var, - uint64_t value, - int memory_order); -# define atomic_store_explicit(var_, value_, memory_order_) \ - do { \ - assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ - sizeof *(var_) == 4 || sizeof *(var_) == 8); \ - if (sizeof *(var_) == 1) \ - kit_atomic_store_explicit_8((uint8_t volatile *) (var_), \ - (uint8_t) (value_), \ - (memory_order_)); \ - if (sizeof *(var_) == 2) \ - kit_atomic_store_explicit_16((uint16_t volatile *) (var_), \ - (uint16_t) (value_), \ - (memory_order_)); \ - if (sizeof *(var_) == 4) \ - kit_atomic_store_explicit_32((uint32_t volatile *) (var_), \ - (uint32_t) (value_), \ - (memory_order_)); \ - if (sizeof *(var_) == 8) \ - kit_atomic_store_explicit_64((uint64_t volatile *) (var_), \ - (uint64_t) (value_), \ - (memory_order_)); \ +void kit_atomic_store_explicit_8(u8 volatile *var, u8 value, + i32 memory_order); +void kit_atomic_store_explicit_16(u16 volatile *var, u16 value, + i32 memory_order); +void kit_atomic_store_explicit_32(u32 volatile *var, u32 value, + i32 memory_order); +void kit_atomic_store_explicit_64(u64 volatile *var, u64 value, + i32 memory_order); +u8 kit_atomic_load_explicit_8(u8 volatile *var, i32 memory_order); +u16 kit_atomic_load_explicit_16(u16 volatile *var, i32 memory_order); +u32 kit_atomic_load_explicit_32(u32 volatile *var, i32 memory_order); +u64 kit_atomic_load_explicit_64(u64 volatile *var, i32 memory_order); +u8 kit_atomic_exchange_explicit_8(u8 volatile *var, u8 value, + i32 memory_order); +u16 kit_atomic_exchange_explicit_16(u16 volatile *var, u16 value, + i32 memory_order); +u32 kit_atomic_exchange_explicit_32(u32 volatile *var, u32 value, + i32 memory_order); +u64 kit_atomic_exchange_explicit_64(u64 volatile *var, u64 value, + i32 memory_order); +i32 kit_atomic_compare_exchange_explicit_8(u8 volatile *var, + u8 *expected, u8 value, + i32 memory_order_succ_, + i32 memory_order_fail_); +i32 kit_atomic_compare_exchange_explicit_16(u16 volatile *var, + u16 *expected, u16 value, + i32 memory_order_succ_, + i32 memory_order_fail_); +i32 kit_atomic_compare_exchange_explicit_32(u32 volatile *var, + u32 *expected, u32 value, + i32 memory_order_succ_, + i32 memory_order_fail_); +i32 kit_atomic_compare_exchange_explicit_64(u64 volatile *var, + u64 *expected, u64 value, + i32 memory_order_succ_, + i32 memory_order_fail_); +u8 kit_atomic_fetch_add_explicit_8(u8 volatile *var, u8 value, + i32 memory_order); +u16 kit_atomic_fetch_add_explicit_16(u16 volatile *var, u16 value, + i32 memory_order); +u32 kit_atomic_fetch_add_explicit_32(u32 volatile *var, u32 value, + i32 memory_order); +u64 kit_atomic_fetch_add_explicit_64(u64 volatile *var, u64 value, + i32 memory_order); +# define atomic_store_explicit(var_, value_, memory_order_) \ + do { \ + assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ + sizeof *(var_) == 4 || sizeof *(var_) == 8); \ + if (sizeof *(var_) == 1) \ + kit_atomic_store_explicit_8((u8 volatile *) (var_), \ + (u8) (value_), (memory_order_)); \ + if (sizeof *(var_) == 2) \ + kit_atomic_store_explicit_16((u16 volatile *) (var_), \ + (u16) (value_), \ + (memory_order_)); \ + if (sizeof *(var_) == 4) \ + kit_atomic_store_explicit_32((u32 volatile *) (var_), \ + (u32) (value_), \ + (memory_order_)); \ + if (sizeof *(var_) == 8) \ + kit_atomic_store_explicit_64((u64 volatile *) (var_), \ + (u64) (value_), \ + (memory_order_)); \ } while (0) -# define atomic_load_explicit(var_, memory_order_) \ - (assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ - sizeof *(var_) == 4 || sizeof *(var_) == 8), \ - (sizeof *(var_) == 1 \ - ? kit_atomic_load_explicit_8((uint8_t volatile *) (var_), \ - (memory_order_)) \ - : sizeof *(var_) == 2 \ - ? kit_atomic_load_explicit_16( \ - (uint16_t volatile *) (var_), (memory_order_)) \ - : sizeof *(var_) == 4 \ - ? kit_atomic_load_explicit_32( \ - (uint32_t volatile *) (var_), (memory_order_)) \ - : kit_atomic_load_explicit_64( \ - (uint64_t volatile *) (var_), (memory_order_)))) +# define atomic_load_explicit(var_, memory_order_) \ + (assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ + sizeof *(var_) == 4 || sizeof *(var_) == 8), \ + (sizeof *(var_) == 1 \ + ? kit_atomic_load_explicit_8((u8 volatile *) (var_), \ + (memory_order_)) \ + : sizeof *(var_) == 2 \ + ? kit_atomic_load_explicit_16((u16 volatile *) (var_), \ + (memory_order_)) \ + : sizeof *(var_) == 4 \ + ? kit_atomic_load_explicit_32((u32 volatile *) (var_), \ + (memory_order_)) \ + : kit_atomic_load_explicit_64((u64 volatile *) (var_), \ + (memory_order_)))) # define atomic_exchange_explicit(var_, value_, memory_order_) \ (assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ sizeof *(var_) == 4 || sizeof *(var_) == 8), \ (sizeof *(var_) == 1 ? kit_atomic_exchange_explicit_8( \ - (uint8_t volatile *) (var_), \ - (uint8_t) (value_), (memory_order_)) \ - : sizeof *(var_) == 2 \ - ? kit_atomic_exchange_explicit_16( \ - (uint16_t volatile *) (var_), (uint16_t) (value_), \ - (memory_order_)) \ + (u8 volatile *) (var_), \ + (u8) (value_), (memory_order_)) \ + : sizeof *(var_) == 2 ? kit_atomic_exchange_explicit_16( \ + (u16 volatile *) (var_), \ + (u16) (value_), (memory_order_)) \ : sizeof *(var_) == 4 \ - ? kit_atomic_exchange_explicit_32( \ - (uint32_t volatile *) (var_), (uint32_t) (value_), \ - (memory_order_)) \ - : kit_atomic_exchange_explicit_64( \ - (uint64_t volatile *) (var_), (uint64_t) (value_), \ - (memory_order_)))) -# define atomic_fetch_add_explicit(var_, value_, memory_order_) \ - (assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ - sizeof *(var_) == 4 || sizeof *(var_) == 8), \ - (sizeof *(var_) == 1 ? kit_atomic_fetch_add_explicit_8( \ - (uint8_t volatile *) (var_), \ - (uint8_t) (value_), (memory_order_)) \ - : sizeof *(var_) == 2 \ - ? kit_atomic_fetch_add_explicit_16( \ - (uint16_t volatile *) (var_), (uint16_t) (value_), \ - (memory_order_)) \ - : sizeof *(var_) == 4 \ - ? kit_atomic_fetch_add_explicit_32( \ - (uint32_t volatile *) (var_), (uint32_t) (value_), \ - (memory_order_)) \ - : kit_atomic_fetch_add_explicit_64( \ - (uint64_t volatile *) (var_), (uint64_t) (value_), \ - (memory_order_)))) + ? kit_atomic_exchange_explicit_32((u32 volatile *) (var_), \ + (u32) (value_), \ + (memory_order_)) \ + : kit_atomic_exchange_explicit_64((u64 volatile *) (var_), \ + (u64) (value_), \ + (memory_order_)))) +# define atomic_compare_exchange_strong_explicit( \ + var_, expected_, value_, memory_order_succ_, \ + memory_order_fail_) \ + (assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ + sizeof *(var_) == 4 || sizeof *(var_) == 8), \ + (sizeof *(var_) == 1 \ + ? kit_atomic_compare_exchange_explicit_8( \ + (u8 volatile *) (var_), (u8 *) (expected_), \ + (u8) (value_), (memory_order_succ_), \ + (memory_order_fail_)) \ + : sizeof *(var_) == 2 \ + ? kit_atomic_compare_exchange_explicit_16( \ + (u16 volatile *) (var_), (u16 *) (expected_), \ + (u16) (value_), (memory_order_succ_), \ + (memory_order_fail_)) \ + : sizeof *(var_) == 4 \ + ? kit_atomic_compare_exchange_explicit_32( \ + (u32 volatile *) (var_), (u32 *) (expected_), \ + (u32) (value_), (memory_order_succ_), \ + (memory_order_fail_)) \ + : kit_atomic_compare_exchange_explicit_64( \ + (u64 volatile *) (var_), (u64 *) (expected_), \ + (u64) (value_), (memory_order_succ_), \ + (memory_order_fail_)))) +# define atomic_compare_exchange_weak_explicit( \ + var_, expected_, value_, memory_order_succ_, \ + memory_order_fail_) \ + atomic_compare_exchange_strong_explicit(var_, expected_, value_, \ + memory_order_succ_, \ + memory_order_fail_) +# define atomic_fetch_add_explicit(var_, value_, memory_order_) \ + (assert(sizeof *(var_) == 1 || sizeof *(var_) == 2 || \ + sizeof *(var_) == 4 || sizeof *(var_) == 8), \ + (sizeof *(var_) == 1 ? kit_atomic_fetch_add_explicit_8( \ + (u8 volatile *) (var_), \ + (u8) (value_), (memory_order_)) \ + : sizeof *(var_) == 2 ? kit_atomic_fetch_add_explicit_16( \ + (u16 volatile *) (var_), \ + (u16) (value_), (memory_order_)) \ + : sizeof *(var_) == 4 ? kit_atomic_fetch_add_explicit_32( \ + (u32 volatile *) (var_), \ + (u32) (value_), (memory_order_)) \ + : kit_atomic_fetch_add_explicit_64( \ + (u64 volatile *) (var_), \ + (u64) (value_), (memory_order_)))) +# define atomic_store(var_, value_) \ + atomic_store(var_, value_, memory_order_seq_cst) +# define atomic_load(var_) atomic_load(var_, memory_order_seq_cst) +# define atomic_exchange(var_, value_) \ + atomic_exchange(var_, value_, memory_order_seq_cst) +# define atomic_compare_exchange_strong(var_, expected_, value_) \ + atomic_compare_exchange_strong_explicit(var_, expected_, value_, \ + memory_order_seq_cst, \ + memory_order_seq_cst) +# define atomic_compare_exchange_weak(var_, expected_, value_) \ + atomic_compare_exchange_weak_explicit(var_, expected_, value_, \ + memory_order_seq_cst, \ + memory_order_seq_cst) +# define atomic_fetch_add(var_, value_) \ + atomic_fetch_add(var_, value_, memory_order_seq_cst) #endif #ifndef KIT_DISABLE_SHORT_NAMES # define ATOMIC KIT_ATOMIC @@ -415,21 +353,64 @@ uint64_t kit_atomic_fetch_add_explicit_64(uint64_t volatile *var, #endif /********************************************************************* * * - * File: source/kit/mutex.h * + * File: source/kit/threads.h * * * *********************************************************************/ -#ifndef KIT_MUTEX_H -#define KIT_MUTEX_H +#ifndef KIT_THREADS_H +#define KIT_THREADS_H #ifndef KIT_DISABLE_SYSTEM_THREADS -# ifndef _GNU_SOURCE -# define _GNU_SOURCE -# endif +# include +# include # if !defined(_WIN32) || defined(__CYGWIN__) # include # endif +# if defined(__cplusplus) +# define _Noreturn [[noreturn]] +# elif defined(_MSC_VER) +# define _Noreturn __declspec(noreturn) +# endif +# ifndef _Thread_local +# if defined(__cplusplus) +// C++11 doesn't need `_Thread_local` keyword or macro +# elif !defined(__STDC_NO_THREADS__) +// threads are optional in C11, _Thread_local present in this +// condition +# elif defined(_MSC_VER) +# define _Thread_local __declspec(thread) +# elif defined(__GNUC__) +# define _Thread_local __thread +# else +// Leave _Thread_local undefined so that use of _Thread_local would +// not promote to a non-thread-local global variable +# endif +# endif +# if !defined(__cplusplus) +// C11 thread_local() macro C++11 and above already have thread_local +// keyword +# ifndef thread_local +# if _MSC_VER +# define thread_local __declspec(thread) +# else +# define thread_local _Thread_local +# endif +# endif +# endif # ifdef __cplusplus extern "C" { # endif +enum { + mtx_plain, + mtx_recursive, + mtx_timed, +}; +enum { + thrd_success, + thrd_timedout, + thrd_error, + thrd_busy, + thrd_nomem, + thrd_wrong_stack_size +}; # if defined(_WIN32) && !defined(__CYGWIN__) typedef struct { void *DebugInfo; @@ -439,56 +420,50 @@ typedef struct { void *LockSemaphore; uintptr_t SpinCount; } mtx_t; -# else -typedef pthread_mutex_t mtx_t; -# endif -enum { - mtx_plain = 0, - mtx_recursive = 1, - mtx_timed = 2, -}; -void mtx_destroy(mtx_t *mtx_); -int mtx_init(mtx_t *mtx_, int); -int mtx_lock(mtx_t *mtx_); -int mtx_timedlock(mtx_t *__restrict mtx_, - struct timespec const *__restrict); -int mtx_trylock(mtx_t *mtx_); -int mtx_unlock(mtx_t *mtx_); -# ifdef __cplusplus -} -# endif -#endif -#endif -/********************************************************************* - * * - * File: source/kit/condition_variable.h * - * * - *********************************************************************/ -#ifndef KIT_CONDITION_VARIABLE_H -#define KIT_CONDITION_VARIABLE_H -#ifndef KIT_DISABLE_SYSTEM_THREADS -# ifdef __cplusplus -extern "C" { -# endif -# if defined(_WIN32) && !defined(__CYGWIN__) typedef struct { void *Ptr; } cnd_t; typedef struct { volatile uintptr_t status; } once_flag; +typedef struct { + void *handle; +} thrd_t; +typedef unsigned long tss_t; # else -typedef pthread_cond_t cnd_t; -typedef pthread_once_t once_flag; +typedef pthread_mutex_t mtx_t; +typedef pthread_cond_t cnd_t; +typedef pthread_once_t once_flag; +typedef pthread_t thrd_t; +typedef pthread_key_t tss_t; # endif -void call_once(once_flag *, void (*)(void)); -int cnd_broadcast(cnd_t *); -void cnd_destroy(cnd_t *); -int cnd_init(cnd_t *); -int cnd_signal(cnd_t *); -int cnd_timedwait(cnd_t *__restrict, mtx_t *__restrict mtx_, - struct timespec const *__restrict); -int cnd_wait(cnd_t *, mtx_t *mtx_); +typedef void (*tss_dtor_t)(void *); +typedef int (*thrd_start_t)(void *); +void mtx_destroy(mtx_t *mtx_); +int mtx_init(mtx_t *mtx_, int); +int mtx_lock(mtx_t *mtx_); +int mtx_timedlock(mtx_t *__restrict mtx_, + struct timespec const *__restrict); +int mtx_trylock(mtx_t *mtx_); +int mtx_unlock(mtx_t *mtx_); +void call_once(once_flag *, void (*)(void)); +int cnd_broadcast(cnd_t *); +void cnd_destroy(cnd_t *); +int cnd_init(cnd_t *); +int cnd_signal(cnd_t *); +int cnd_timedwait(cnd_t *__restrict, mtx_t *__restrict mtx_, + struct timespec const *__restrict); +int cnd_wait(cnd_t *, mtx_t *mtx_); +int thrd_create(thrd_t *, thrd_start_t, void *); +int thrd_create_with_stack(thrd_t *, thrd_start_t, void *, + ptrdiff_t stack_size); +thrd_t thrd_current(void); +int thrd_detach(thrd_t); +int thrd_equal(thrd_t, thrd_t); +_Noreturn void thrd_exit(int); +int thrd_join(thrd_t, int *); +int thrd_sleep(struct timespec const *, struct timespec *); +void thrd_yield(void); # ifdef __cplusplus } # endif @@ -501,31 +476,25 @@ int cnd_wait(cnd_t *, mtx_t *mtx_); *********************************************************************/ #ifndef KIT_ARRAY_REF_H #define KIT_ARRAY_REF_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include -#include #ifdef __cplusplus extern "C" { #endif -typedef int (*kit_ar_compare_fn)(void *left, void *right); -int kit_ar_equal_bytes(ptrdiff_t left_element_size, - ptrdiff_t left_size, void *left_data, - ptrdiff_t right_element_size, - ptrdiff_t right_size, void *right_data); -int kit_ar_compare(ptrdiff_t left_element_size, ptrdiff_t left_size, - void *left_data, ptrdiff_t right_element_size, - ptrdiff_t right_size, void *right_data, - kit_ar_compare_fn compare); +typedef i8 (*kit_ar_compare_fn)(void *left, void *right); +i8 kit_ar_equal_bytes(i64 left_element_size, i64 left_size, + void *left_data, i64 right_element_size, + i64 right_size, void *right_data); +i8 kit_ar_compare(i64 left_element_size, i64 left_size, + void *left_data, i64 right_element_size, + i64 right_size, void *right_data, + kit_ar_compare_fn compare); #define KIT_AR(type_) \ struct { \ - ptrdiff_t size; \ - type_ *values; \ + i64 size; \ + type_ *values; \ } #define KIT_AR_WRAP(name_, element_type_, array_) \ struct { \ - ptrdiff_t size; \ + i64 size; \ element_type_ *values; \ } name_ = { .size = (sizeof(array_) / sizeof((array_)[0])), \ .values = (array_) } @@ -551,67 +520,6 @@ int kit_ar_compare(ptrdiff_t left_element_size, ptrdiff_t left_size, } #endif #endif -/********************************************************************* - * * - * File: source/kit/string_ref.h * - * * - *********************************************************************/ -#ifndef KIT_STRING_REF_H -#define KIT_STRING_REF_H -#include -#ifdef __cplusplus -extern "C" { -#endif -typedef KIT_AR(char) kit_string_ref_t; -typedef kit_string_ref_t kit_str_t; -#ifdef __GNUC__ -# pragma GCC diagnostic push -# pragma GCC diagnostic ignored "-Wunused-function" -# pragma GCC diagnostic ignored "-Wunknown-pragmas" -# pragma GCC push_options -# pragma GCC optimize("O3") -#endif -static kit_str_t kit_str(ptrdiff_t size, char *static_string) { - kit_str_t s = { .size = size, .values = static_string }; - return s; -} -/* Make a barbarian string for C standard library functions. - * Not thread safe. - * Use with caution. - */ -static char *kit_make_bs(kit_str_t s) { - static char buf[8][4096]; - static int index = 0; - ptrdiff_t n = s.size; - if (n > 4095) - n = 4095; - memcpy(buf[index], s.values, n); - buf[index][n] = '\0'; - char *result = buf[index]; - index = (index + 1) % 8; - return result; -} -#ifdef __GNUC__ -# pragma GCC pop_options -# pragma GCC diagnostic pop -#endif -#define KIT_SZ(static_str_) \ - kit_str(sizeof(static_str_) - 1, (static_str_)) -#define KIT_WRAP_BS(string_) kit_str(strlen(string_), (string_)) -#define KIT_WRAP_STR(string_) \ - kit_str((string_).size, (string_).values) -#ifndef KIT_DISABLE_SHORT_NAMES -# define BS(string_) kit_make_bs(KIT_WRAP_STR(string_)) -# define string_ref_t kit_string_ref_t -# define str_t kit_str_t -# define SZ KIT_SZ -# define WRAP_BS KIT_WRAP_BS -# define WRAP_STR KIT_WRAP_STR -#endif -#ifdef __cplusplus -} -#endif -#endif /********************************************************************* * * * File: source/kit/dynamic_array.h * @@ -623,23 +531,22 @@ static char *kit_make_bs(kit_str_t s) { extern "C" { #endif typedef struct { - ptrdiff_t capacity; - ptrdiff_t size; + i64 capacity; + i64 size; void *values; kit_allocator_t *alloc; } kit_da_void_t; -void kit_da_init(kit_da_void_t *array, ptrdiff_t element_size, - ptrdiff_t size, kit_allocator_t *alloc); -void kit_da_resize(kit_da_void_t *array, ptrdiff_t element_size, - ptrdiff_t size); -void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, - ptrdiff_t size); +void kit_da_init(kit_da_void_t *array, i64 element_size, i64 size, + kit_allocator_t *alloc); +void kit_da_resize(kit_da_void_t *array, i64 element_size, i64 size); +void kit_da_resize_exact(kit_da_void_t *array, i64 element_size, + i64 size); /* Dynamic array type definition. */ #define KIT_DA(element_type_) \ struct { \ - ptrdiff_t capacity; \ - ptrdiff_t size; \ + i64 capacity; \ + i64 size; \ element_type_ *values; \ kit_allocator_t *alloc; \ } @@ -675,7 +582,7 @@ void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, */ #define KIT_DA_APPEND(array_, value_) \ do { \ - ptrdiff_t kit_index_back_ = (array_).size; \ + i64 kit_index_back_ = (array_).size; \ KIT_DA_RESIZE((array_), kit_index_back_ + 1); \ if (kit_index_back_ < (array_).size) \ (array_).values[kit_index_back_] = (value_); \ @@ -684,9 +591,9 @@ void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, */ #define KIT_DA_INSERT(array_, index_, value_) \ do { \ - ptrdiff_t kit_i_; \ - ptrdiff_t kit_index_back_ = (array_).size; \ - ptrdiff_t kit_indert_n_ = (index_); \ + i64 kit_i_; \ + i64 kit_index_back_ = (array_).size; \ + i64 kit_indert_n_ = (index_); \ KIT_DA_RESIZE((array_), kit_index_back_ + 1); \ if (kit_index_back_ + 1 == (array_).size) { \ for (kit_i_ = kit_index_back_; kit_i_ > kit_indert_n_; \ @@ -699,17 +606,15 @@ void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, */ #define KIT_DA_ERASE(array_, index_) \ do { \ - ptrdiff_t i_; \ + i64 i_; \ for (i_ = (index_) + 1; i_ < (array_).size; i_++) \ (array_).values[i_ - 1] = (array_).values[i_]; \ KIT_DA_RESIZE((array_), (array_).size - 1); \ } while (0) -typedef KIT_DA(char) kit_string_t; #ifndef KIT_DISABLE_SHORT_NAMES # define da_void_t kit_da_void_t # define da_init kit_da_init # define da_resize kit_da_resize -# define string_t kit_string_t # define DA KIT_DA # define DA_INIT KIT_DA_INIT # define DA_CREATE KIT_DA_CREATE @@ -724,6 +629,77 @@ typedef KIT_DA(char) kit_string_t; } #endif #endif +/********************************************************************* + * * + * File: source/kit/string_ref.h * + * * + *********************************************************************/ +#ifndef KIT_STRING_REF_H +#define KIT_STRING_REF_H +#include +#ifdef __cplusplus +extern "C" { +#endif +typedef KIT_AR(char) kit_str_t; +#ifdef __GNUC__ +# pragma GCC diagnostic push +# pragma GCC diagnostic ignored "-Wunused-function" +# pragma GCC diagnostic ignored "-Wunknown-pragmas" +# pragma GCC push_options +# pragma GCC optimize("O3") +#endif +static kit_str_t kit_str(i64 size, char *static_string) { + kit_str_t s = { .size = size, .values = static_string }; + return s; +} +// Make a barbarian string for C standard library functions. +// Not thread safe. +// Use with caution. +// +static char *kit_make_bs(kit_str_t s) { + static char buf[8][4096]; + static i32 index = 0; + i64 n = s.size; + if (n > 4095) + n = 4095; + memcpy(buf[index], s.values, n); + buf[index][n] = '\0'; + char *result = buf[index]; + index = (index + 1) % 8; + return result; +} +#ifdef __GNUC__ +# pragma GCC pop_options +# pragma GCC diagnostic pop +#endif +#define KIT_SZ(static_str_) \ + kit_str(sizeof(static_str_) - 1, (static_str_)) +#define KIT_WRAP_BS(string_) kit_str(strlen(string_), (string_)) +#define KIT_WRAP_STR(string_) \ + kit_str((string_).size, (string_).values) +#ifndef KIT_DISABLE_SHORT_NAMES +# define BS(string_) kit_make_bs(KIT_WRAP_STR(string_)) +# define str_t kit_str_t +# define SZ KIT_SZ +# define WRAP_BS KIT_WRAP_BS +# define WRAP_STR KIT_WRAP_STR +#endif +#ifdef __cplusplus +} +#endif +#endif +/********************************************************************* + * * + * File: source/kit/string_builder.h * + * * + *********************************************************************/ +#ifndef KIT_STRING_BUILDER_H +#define KIT_STRING_BUILDER_H +typedef KIT_DA(char) kit_str_builder_t; +#ifndef KIT_DISABLE_SHORT_NAMES +# define str_builder_t kit_str_builder_t +#endif +#endif /********************************************************************* * * * File: source/kit/lower_bound.h * @@ -736,11 +712,11 @@ extern "C" { #endif #define KIT_LOWER_BOUND_INL(return_val, size, ...) \ do { \ - ptrdiff_t position_ = 0; \ - ptrdiff_t count_ = (size); \ + i64 position_ = 0; \ + i64 count_ = (size); \ while (count_ > 0) { \ - ptrdiff_t const delta_ = count_ / 2; \ - ptrdiff_t const index_ = position_ + delta_; \ + i64 delta_ = count_ / 2; \ + i64 index_ = position_ + delta_; \ if (__VA_ARGS__) { \ position_ += delta_ + 1; \ count_ -= delta_ + 1; \ @@ -771,18 +747,15 @@ extern "C" { *********************************************************************/ #ifndef KIT_MOVE_BACK_H #define KIT_MOVE_BACK_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif #include #ifdef __cplusplus extern "C" { #endif #define KIT_MOVE_BACK_INL(new_size, array, ...) \ do { \ - ptrdiff_t index_; \ - ptrdiff_t end_ = (array).size; \ - unsigned char temp_[sizeof *(array).values]; \ + i64 index_; \ + i64 end_ = (array).size; \ + u8 temp_[sizeof *(array).values]; \ for (index_ = 0; index_ < end_;) { \ if (__VA_ARGS__) { \ end_--; \ @@ -829,18 +802,15 @@ extern "C" { # define KIT_BIGINT_SIZE 64 #endif #if __STDC_VERSION__ >= 199901L -static_assert(sizeof(uint8_t) == 1, "uint8_t size should be 1 byte"); -static_assert(sizeof(uint32_t) == 4, - "uint32_t size should be 4 bytes"); -static_assert(sizeof(uint64_t) == 8, - "uint64_t size should be 8 bytes"); +static_assert(sizeof(u8) == 1, "u8 size should be 1 byte"); +static_assert(sizeof(u32) == 4, "u32 size should be 4 bytes"); +static_assert(sizeof(u64) == 8, "u64 size should be 8 bytes"); static_assert(KIT_BIGINT_SIZE > 0 && (KIT_BIGINT_SIZE % 8) == 0, "Invalid big integer size"); #endif typedef struct { - uint32_t v[KIT_BIGINT_SIZE / 4]; + u32 v[KIT_BIGINT_SIZE / 4]; } kit_bigint_t; -typedef uint_fast8_t kit_bit_t; #ifdef __GNUC__ # pragma GCC diagnostic push # pragma GCC diagnostic ignored "-Wunused-function" @@ -848,34 +818,34 @@ typedef uint_fast8_t kit_bit_t; # pragma GCC push_options # pragma GCC optimize("O3") #endif -static kit_bigint_t kit_bi_uint32(uint32_t x) { +static kit_bigint_t kit_bi_u32(u32 x) { kit_bigint_t z; memset(&z, 0, sizeof z); z.v[0] = x; return z; } -static kit_bigint_t kit_bi_uint64(uint64_t x) { +static kit_bigint_t kit_bi_u64(u64 x) { kit_bigint_t z; memset(&z, 0, sizeof z); - z.v[0] = (uint32_t) (x & 0xffffffff); - z.v[1] = (uint32_t) (x >> 32); + z.v[0] = (u32) (x & 0xffffffff); + z.v[1] = (u32) (x >> 32); return z; } -static kit_bigint_t kit_bi_int32(int32_t x) { +static kit_bigint_t kit_bi_i32(i32 x) { kit_bigint_t z; memset(&z, x < 0 ? -1 : 0, sizeof z); z.v[0] = x; return z; } -static kit_bigint_t kit_bi_int64(int64_t x) { +static kit_bigint_t kit_bi_i64(i64 x) { kit_bigint_t z; memset(&z, x < 0 ? -1 : 0, sizeof z); - z.v[0] = (uint32_t) (((uint64_t) x) & 0xffffffff); - z.v[1] = (uint32_t) (((uint64_t) x) >> 32); + z.v[0] = (u32) (((u64) x) & 0xffffffff); + z.v[1] = (u32) (((u64) x) >> 32); return z; } static int kit_bi_is_zero(kit_bigint_t x) { - ptrdiff_t i; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) if (x.v[i] != 0) return 0; @@ -889,7 +859,7 @@ static int kit_bi_equal(kit_bigint_t x, kit_bigint_t y) { KIT_BIGINT_SIZE, y.v); } static int kit_bi_compare(kit_bigint_t x, kit_bigint_t y) { - ptrdiff_t i; + i64 i; for (i = KIT_BIGINT_SIZE / 4 - 1; i >= 0; i--) if (x.v[i] < y.v[i]) return -1; @@ -897,70 +867,70 @@ static int kit_bi_compare(kit_bigint_t x, kit_bigint_t y) { return 1; return 0; } -static ptrdiff_t kit_bi_significant_bit_count(kit_bigint_t x) { - ptrdiff_t n = KIT_BIGINT_SIZE / 4 - 1; +static i64 kit_bi_significant_bit_count(kit_bigint_t x) { + i64 n = KIT_BIGINT_SIZE / 4 - 1; while (n > 0 && x.v[n] == 0) n--; - uint32_t val = x.v[n]; + u32 val = x.v[n]; if (val == 0) return 0; - ptrdiff_t bits = (val & 0x80000000u) != 0 ? 32 - : (val & 0x40000000u) != 0 ? 31 - : (val & 0x20000000u) != 0 ? 30 - : (val & 0x10000000u) != 0 ? 29 - : (val & 0x8000000u) != 0 ? 28 - : (val & 0x4000000u) != 0 ? 27 - : (val & 0x2000000u) != 0 ? 26 - : (val & 0x1000000u) != 0 ? 25 - : (val & 0x800000u) != 0 ? 24 - : (val & 0x400000u) != 0 ? 23 - : (val & 0x200000u) != 0 ? 22 - : (val & 0x100000u) != 0 ? 21 - : (val & 0x80000u) != 0 ? 20 - : (val & 0x40000u) != 0 ? 19 - : (val & 0x20000u) != 0 ? 18 - : (val & 0x10000u) != 0 ? 17 - : (val & 0x8000u) != 0 ? 16 - : (val & 0x4000u) != 0 ? 15 - : (val & 0x2000u) != 0 ? 14 - : (val & 0x1000u) != 0 ? 13 - : (val & 0x800u) != 0 ? 12 - : (val & 0x400u) != 0 ? 11 - : (val & 0x200u) != 0 ? 10 - : (val & 0x100u) != 0 ? 9 - : (val & 0x80u) != 0 ? 8 - : (val & 0x40u) != 0 ? 7 - : (val & 0x20u) != 0 ? 6 - : (val & 0x10u) != 0 ? 5 - : (val & 0x08u) != 0 ? 4 - : (val & 0x04u) != 0 ? 3 - : (val & 0x02u) != 0 ? 2 - : 1; + i64 bits = (val & 0x80000000u) != 0 ? 32 + : (val & 0x40000000u) != 0 ? 31 + : (val & 0x20000000u) != 0 ? 30 + : (val & 0x10000000u) != 0 ? 29 + : (val & 0x8000000u) != 0 ? 28 + : (val & 0x4000000u) != 0 ? 27 + : (val & 0x2000000u) != 0 ? 26 + : (val & 0x1000000u) != 0 ? 25 + : (val & 0x800000u) != 0 ? 24 + : (val & 0x400000u) != 0 ? 23 + : (val & 0x200000u) != 0 ? 22 + : (val & 0x100000u) != 0 ? 21 + : (val & 0x80000u) != 0 ? 20 + : (val & 0x40000u) != 0 ? 19 + : (val & 0x20000u) != 0 ? 18 + : (val & 0x10000u) != 0 ? 17 + : (val & 0x8000u) != 0 ? 16 + : (val & 0x4000u) != 0 ? 15 + : (val & 0x2000u) != 0 ? 14 + : (val & 0x1000u) != 0 ? 13 + : (val & 0x800u) != 0 ? 12 + : (val & 0x400u) != 0 ? 11 + : (val & 0x200u) != 0 ? 10 + : (val & 0x100u) != 0 ? 9 + : (val & 0x80u) != 0 ? 8 + : (val & 0x40u) != 0 ? 7 + : (val & 0x20u) != 0 ? 6 + : (val & 0x10u) != 0 ? 5 + : (val & 0x08u) != 0 ? 4 + : (val & 0x04u) != 0 ? 3 + : (val & 0x02u) != 0 ? 2 + : 1; return n * 32 + bits; } static kit_bigint_t kit_bi_and(kit_bigint_t x, kit_bigint_t y) { kit_bigint_t z; - ptrdiff_t i; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) z.v[i] = x.v[i] & y.v[i]; return z; } static kit_bigint_t kit_bi_or(kit_bigint_t x, kit_bigint_t y) { kit_bigint_t z; - ptrdiff_t i; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) z.v[i] = x.v[i] | y.v[i]; return z; } static kit_bigint_t kit_bi_xor(kit_bigint_t x, kit_bigint_t y) { kit_bigint_t z; - ptrdiff_t i; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) z.v[i] = x.v[i] ^ y.v[i]; return z; } -static kit_bigint_t kit_bi_shl_uint(kit_bigint_t x, uint32_t y) { +static kit_bigint_t kit_bi_shl_uint(kit_bigint_t x, u32 y) { kit_bigint_t z; memset(&z, 0, sizeof z); - ptrdiff_t words = (ptrdiff_t) (y / 32); - ptrdiff_t bits = (ptrdiff_t) (y % 32); - ptrdiff_t i; + i64 words = (i64) (y / 32); + i64 bits = (i64) (y % 32); + i64 i; for (i = words; i < KIT_BIGINT_SIZE / 4; i++) { z.v[i] |= x.v[i - words] << bits; if (bits != 0 && i + 1 < KIT_BIGINT_SIZE / 4) @@ -968,12 +938,12 @@ static kit_bigint_t kit_bi_shl_uint(kit_bigint_t x, uint32_t y) { } return z; } -static kit_bigint_t kit_bi_shr_uint(kit_bigint_t x, uint32_t y) { +static kit_bigint_t kit_bi_shr_uint(kit_bigint_t x, u32 y) { kit_bigint_t z; memset(&z, 0, sizeof z); - ptrdiff_t words = (ptrdiff_t) (y / 32); - ptrdiff_t bits = (ptrdiff_t) (y % 32); - ptrdiff_t i; + i64 words = (i64) (y / 32); + i64 bits = (i64) (y % 32); + i64 i; for (i = KIT_BIGINT_SIZE / 4 - words - 1; i >= 0; i--) { z.v[i] |= x.v[i + words] >> bits; if (bits != 0 && i > 0) @@ -981,8 +951,7 @@ static kit_bigint_t kit_bi_shr_uint(kit_bigint_t x, uint32_t y) { } return z; } -static kit_bit_t kit_bi_carry(uint32_t x, uint32_t y, - kit_bit_t carry) { +static i8 kit_bi_carry(u32 x, u32 y, i8 carry) { assert(carry == 0 || carry == 1); return 0xffffffffu - x < y || 0xffffffffu - x - y < carry ? 1 : 0; } @@ -990,8 +959,8 @@ static kit_bit_t kit_bi_carry(uint32_t x, uint32_t y, */ static kit_bigint_t kit_bi_inc(kit_bigint_t x) { kit_bigint_t z; - kit_bit_t carry = 1; - ptrdiff_t i; + i8 carry = 1; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { z.v[i] = x.v[i] + carry; carry = kit_bi_carry(x.v[i], 0, carry); @@ -1002,8 +971,8 @@ static kit_bigint_t kit_bi_inc(kit_bigint_t x) { */ static kit_bigint_t kit_bi_dec(kit_bigint_t x) { kit_bigint_t z; - kit_bit_t carry = 0; - ptrdiff_t i; + i8 carry = 0; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { z.v[i] = x.v[i] + 0xffffffff + carry; carry = kit_bi_carry(x.v[i], 0xffffffff, carry); @@ -1014,8 +983,8 @@ static kit_bigint_t kit_bi_dec(kit_bigint_t x) { */ static kit_bigint_t kit_bi_add(kit_bigint_t x, kit_bigint_t y) { kit_bigint_t z; - kit_bit_t carry = 0; - ptrdiff_t i; + i8 carry = 0; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { z.v[i] = x.v[i] + y.v[i] + carry; carry = kit_bi_carry(x.v[i], y.v[i], carry); @@ -1026,8 +995,8 @@ static kit_bigint_t kit_bi_add(kit_bigint_t x, kit_bigint_t y) { */ static kit_bigint_t kit_bi_neg(kit_bigint_t x) { kit_bigint_t y; - kit_bit_t carry = 1; - ptrdiff_t i; + i8 carry = 1; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { y.v[i] = (x.v[i] ^ 0xffffffff) + carry; carry = kit_bi_carry(x.v[i] ^ 0xffffffff, 0, carry); @@ -1038,27 +1007,27 @@ static kit_bigint_t kit_bi_neg(kit_bigint_t x) { */ static kit_bigint_t kit_bi_sub(kit_bigint_t x, kit_bigint_t y) { kit_bigint_t z; - kit_bit_t carry = 1; - ptrdiff_t i; + i8 carry = 1; + i64 i; for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { z.v[i] = x.v[i] + (y.v[i] ^ 0xffffffff) + carry; carry = kit_bi_carry(x.v[i], (y.v[i] ^ 0xffffffff), carry); } return z; } -static kit_bigint_t kit_bi_mul_uint32(kit_bigint_t x, uint32_t y) { +static kit_bigint_t kit_bi_mul_u32(kit_bigint_t x, u32 y) { kit_bigint_t z; - ptrdiff_t i, k; + i64 i, k; memset(&z, 0, sizeof z); if (y != 0) for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { if (x.v[i] == 0) continue; - uint64_t carry = ((uint64_t) x.v[i]) * ((uint64_t) y); + u64 carry = ((u64) x.v[i]) * ((u64) y); for (k = i; k < KIT_BIGINT_SIZE / 4 && carry != 0; k++) { - uint64_t sum = ((uint64_t) z.v[k]) + carry; - z.v[k] = ((uint32_t) (sum & 0xffffffffull)); - carry = sum >> 32; + u64 sum = ((u64) z.v[k]) + carry; + z.v[k] = ((u32) (sum & 0xffffffffull)); + carry = sum >> 32; } } return z; @@ -1067,7 +1036,7 @@ static kit_bigint_t kit_bi_mul_uint32(kit_bigint_t x, uint32_t y) { */ static kit_bigint_t kit_bi_mul(kit_bigint_t x, kit_bigint_t y) { kit_bigint_t z; - ptrdiff_t i, j, k; + i64 i, j, k; memset(&z, 0, sizeof z); for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { if (x.v[i] == 0) @@ -1075,18 +1044,18 @@ static kit_bigint_t kit_bi_mul(kit_bigint_t x, kit_bigint_t y) { for (j = 0; i + j < KIT_BIGINT_SIZE / 4; j++) { if (y.v[j] == 0) continue; - uint64_t carry = ((uint64_t) x.v[i]) * ((uint64_t) y.v[j]); + u64 carry = ((u64) x.v[i]) * ((u64) y.v[j]); for (k = i + j; k < KIT_BIGINT_SIZE / 4 && carry != 0; k++) { - uint64_t sum = ((uint64_t) z.v[k]) + carry; - z.v[k] = ((uint32_t) (sum & 0xffffffffull)); - carry = sum >> 32; + u64 sum = ((u64) z.v[k]) + carry; + z.v[k] = ((u32) (sum & 0xffffffffull)); + carry = sum >> 32; } } } return z; } typedef struct { - kit_bit_t undefined; + i8 undefined; kit_bigint_t quotient; kit_bigint_t remainder; } kit_bi_division_t; @@ -1095,16 +1064,16 @@ typedef struct { static kit_bi_division_t kit_bi_udiv(kit_bigint_t x, kit_bigint_t y) { kit_bi_division_t z; memset(&z, 0, sizeof z); - ptrdiff_t y_bits = kit_bi_significant_bit_count(y); + i64 y_bits = kit_bi_significant_bit_count(y); if (y_bits == 0) { z.undefined = 1; return z; } - ptrdiff_t x_bits = kit_bi_significant_bit_count(x); - ptrdiff_t shift = x_bits - y_bits; + i64 x_bits = kit_bi_significant_bit_count(x); + i64 shift = x_bits - y_bits; z.remainder = x; - z.quotient = kit_bi_uint32(0); - y = kit_bi_shl_uint(y, (uint32_t) shift); + z.quotient = kit_bi_u32(0); + y = kit_bi_shl_uint(y, (u32) shift); while (shift >= 0) { if (kit_bi_compare(z.remainder, y) >= 0) { z.remainder = kit_bi_sub(z.remainder, y); @@ -1134,55 +1103,55 @@ static kit_bi_division_t kit_bi_div(kit_bigint_t x, kit_bigint_t y) { z.quotient = kit_bi_neg(z.quotient); return z; } -static void kit_bi_serialize(kit_bigint_t in, uint8_t *out) { - ptrdiff_t i; +static void kit_bi_serialize(kit_bigint_t in, u8 *out) { + i64 i; assert(out != NULL); for (i = 0; i < KIT_BIGINT_SIZE / 4; i++) { - out[i * 4] = (uint8_t) (in.v[i] & 0xff); - out[i * 4 + 1] = (uint8_t) ((in.v[i] >> 8) & 0xff); - out[i * 4 + 2] = (uint8_t) ((in.v[i] >> 16) & 0xff); - out[i * 4 + 3] = (uint8_t) ((in.v[i] >> 24) & 0xff); + out[i * 4] = (u8) (in.v[i] & 0xff); + out[i * 4 + 1] = (u8) ((in.v[i] >> 8) & 0xff); + out[i * 4 + 2] = (u8) ((in.v[i] >> 16) & 0xff); + out[i * 4 + 3] = (u8) ((in.v[i] >> 24) & 0xff); } } -static kit_bigint_t kit_bi_deserialize(uint8_t *in) { - ptrdiff_t i; +static kit_bigint_t kit_bi_deserialize(u8 *in) { + i64 i; kit_bigint_t out; assert(in != NULL); memset(&out, 0, sizeof out); for (i = 0; i < KIT_BIGINT_SIZE; i++) - out.v[i / 4] |= ((uint32_t) in[i]) << (8 * (i % 4)); + out.v[i / 4] |= ((u32) in[i]) << (8 * (i % 4)); return out; } -static uint8_t kit_bin_digit(char hex) { +static u8 kit_bin_digit(char hex) { assert(hex == '0' || hex == '1'); return hex == '1' ? 1 : 0; } static kit_bigint_t kit_bi_from_bin(kit_str_t bin) { kit_bigint_t z; - ptrdiff_t i; + i64 i; memset(&z, 0, sizeof z); for (i = 0; i < bin.size && i / 8 < KIT_BIGINT_SIZE; i++) { - uint8_t digit = kit_bin_digit(bin.values[bin.size - i - 1]); + u8 digit = kit_bin_digit(bin.values[bin.size - i - 1]); z.v[i / 32] |= digit << (i % 32); } return z; } -static uint8_t kit_dec_digit(char c) { +static u8 kit_dec_digit(char c) { assert('c' >= '0' && c <= '9'); - return c >= '0' && c <= '9' ? (uint8_t) (c - '0') : 0; + return c >= '0' && c <= '9' ? (u8) (c - '0') : 0; } static kit_bigint_t kit_bi_from_dec(kit_str_t dec) { - kit_bigint_t z = kit_bi_uint32(0); - kit_bigint_t factor = kit_bi_uint32(1); - ptrdiff_t i; + kit_bigint_t z = kit_bi_u32(0); + kit_bigint_t factor = kit_bi_u32(1); + i64 i; for (i = 0; i < dec.size; i++) { - uint32_t digit = kit_dec_digit(dec.values[dec.size - i - 1]); - z = kit_bi_add(z, kit_bi_mul_uint32(factor, digit)); - factor = kit_bi_mul_uint32(factor, 10); + u32 digit = kit_dec_digit(dec.values[dec.size - i - 1]); + z = kit_bi_add(z, kit_bi_mul_u32(factor, digit)); + factor = kit_bi_mul_u32(factor, 10); } return z; } -static uint8_t kit_hex_digit(char hex) { +static u8 kit_hex_digit(char hex) { assert((hex >= '0' && hex <= '9') || (hex >= 'a' && hex <= 'f') || (hex >= 'A' && hex <= 'F')); if (hex >= '0' && hex <= '9') @@ -1195,15 +1164,15 @@ static uint8_t kit_hex_digit(char hex) { } static kit_bigint_t kit_bi_from_hex(kit_str_t hex) { kit_bigint_t z; - ptrdiff_t i; + i64 i; memset(&z, 0, sizeof z); for (i = 0; i < hex.size && i / 2 < KIT_BIGINT_SIZE; i++) { - uint8_t digit = kit_hex_digit(hex.values[hex.size - i - 1]); + u8 digit = kit_hex_digit(hex.values[hex.size - i - 1]); z.v[i / 8] |= digit << (4 * (i % 8)); } return z; } -static uint8_t KIT_BASE32_DIGITS[] = { +static u8 KIT_BASE32_DIGITS[] = { ['1'] = 0, ['2'] = 1, ['3'] = 2, ['4'] = 3, ['5'] = 4, ['6'] = 5, ['7'] = 6, ['8'] = 7, ['9'] = 8, ['a'] = 9, ['b'] = 10, ['c'] = 11, ['d'] = 12, ['e'] = 13, ['f'] = 14, @@ -1212,17 +1181,16 @@ static uint8_t KIT_BASE32_DIGITS[] = { ['t'] = 25, ['u'] = 26, ['v'] = 27, ['w'] = 28, ['x'] = 29, ['y'] = 30, ['z'] = 31 }; -static uint8_t kit_base32_digit(char c) { +static u8 kit_base32_digit(char c) { assert(c >= '\0' && c < sizeof KIT_BASE32_DIGITS); - assert(c == '1' || - KIT_BASE32_DIGITS[(size_t) (unsigned char) c] != 0); + assert(c == '1' || KIT_BASE32_DIGITS[(size_t) (u8) c] != 0); return c >= '\0' && c < sizeof KIT_BASE32_DIGITS - ? KIT_BASE32_DIGITS[(size_t) (unsigned char) c] + ? KIT_BASE32_DIGITS[(size_t) (u8) c] : 0; } static kit_bigint_t kit_bi_from_base32(kit_str_t base32) { kit_bigint_t z; - ptrdiff_t i; + i64 i; memset(&z, 0, sizeof z); for (i = 0; i < base32.size; i++) { z = kit_bi_shl_uint(z, 5 * i); @@ -1230,7 +1198,7 @@ static kit_bigint_t kit_bi_from_base32(kit_str_t base32) { } return z; } -static uint8_t KIT_BASE58_DIGITS[] = { +static u8 KIT_BASE58_DIGITS[] = { ['1'] = 0, ['2'] = 1, ['3'] = 2, ['4'] = 3, ['5'] = 4, ['6'] = 5, ['7'] = 6, ['8'] = 7, ['9'] = 8, ['A'] = 9, ['B'] = 10, ['C'] = 11, ['D'] = 12, ['E'] = 13, ['F'] = 14, @@ -1244,23 +1212,21 @@ static uint8_t KIT_BASE58_DIGITS[] = { ['s'] = 50, ['t'] = 51, ['u'] = 52, ['v'] = 53, ['w'] = 54, ['x'] = 55, ['y'] = 56, ['z'] = 57 }; -static uint8_t kit_base58_digit(char c) { +static u8 kit_base58_digit(char c) { assert(c >= '\0' && c < sizeof KIT_BASE58_DIGITS); - assert(c == '1' || - KIT_BASE58_DIGITS[(size_t) (unsigned char) c] != 0); + assert(c == '1' || KIT_BASE58_DIGITS[(size_t) (u8) c] != 0); return c >= '\0' && c < sizeof KIT_BASE58_DIGITS - ? KIT_BASE58_DIGITS[(size_t) (unsigned char) c] + ? KIT_BASE58_DIGITS[(size_t) (u8) c] : 0; } static kit_bigint_t kit_bi_from_base58(kit_str_t base58) { - kit_bigint_t z = kit_bi_uint32(0); - kit_bigint_t factor = kit_bi_uint32(1); - ptrdiff_t i; + kit_bigint_t z = kit_bi_u32(0); + kit_bigint_t factor = kit_bi_u32(1); + i64 i; for (i = 0; i < base58.size; i++) { - uint32_t digit = kit_base58_digit( - base58.values[base58.size - i - 1]); - z = kit_bi_add(z, kit_bi_mul_uint32(factor, digit)); - factor = kit_bi_mul_uint32(factor, 58); + u32 digit = kit_base58_digit(base58.values[base58.size - i - 1]); + z = kit_bi_add(z, kit_bi_mul_u32(factor, digit)); + factor = kit_bi_mul_u32(factor, 58); } return z; } @@ -1280,10 +1246,10 @@ static kit_bigint_t kit_bi_from_base58(kit_str_t base58) { kit_bi_from_base58(kit_str(sizeof(static_str_) - 1, (static_str_))) #ifndef KIT_DISABLE_SHORT_NAMES # define bigint_t kit_bigint_t -# define bi_uint32 kit_bi_uint32 -# define bi_uint64 kit_bi_uint64 -# define bi_int32 kit_bi_int32 -# define bi_int64 kit_bi_int64 +# define bi_u32 kit_bi_u32 +# define bi_u64 kit_bi_u64 +# define bi_i32 kit_bi_i32 +# define bi_i64 kit_bi_i64 # define bi_is_zero kit_bi_is_zero # define bi_is_neg kit_bi_is_neg # define bi_equal kit_bi_equal @@ -1319,8 +1285,7 @@ static kit_bigint_t kit_bi_from_base58(kit_str_t base58) { #ifdef __cplusplus extern "C" { #endif -typedef ptrdiff_t (*kit_is_read_fn)(void *state, - kit_str_t destination); +typedef i64 (*kit_is_read_fn)(void *state, kit_str_t destination); typedef struct { void *state; kit_is_read_fn read; @@ -1358,14 +1323,14 @@ void kit_is_destroy(kit_is_handle_t in); extern "C" { #endif typedef struct { - kit_status_t status; - ptrdiff_t offset; - void *internal; - kit_string_t data; + kit_status_t status; + i64 offset; + void *internal; + kit_str_builder_t data; } kit_ib_handle_t; kit_ib_handle_t kit_ib_wrap(kit_is_handle_t upstream, kit_allocator_t *alloc); -kit_ib_handle_t kit_ib_read(kit_ib_handle_t buf, ptrdiff_t size); +kit_ib_handle_t kit_ib_read(kit_ib_handle_t buf, i64 size); typedef int (*kit_ib_read_condition_fn)(kit_str_t data); kit_ib_handle_t kit_ib_read_while(kit_ib_handle_t buf, kit_ib_read_condition_fn condition); @@ -1391,10 +1356,6 @@ void kit_ib_destroy(kit_ib_handle_t buf); *********************************************************************/ #ifndef KIT_ASYNC_FUNCTION_H #define KIT_ASYNC_FUNCTION_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include #include #ifdef __cplusplus extern "C" { @@ -1405,13 +1366,13 @@ extern "C" { # pragma GCC diagnostic ignored "-Wunknown-pragmas" #endif typedef struct { - int _; + i32 _; } kit_af_void; typedef void (*kit_af_state_machine)(void *self_void_); #define KIT_AF_STATE_DATA \ struct { \ - int _index; \ - ptrdiff_t _id; \ + i32 _index; \ + i32 _id; \ kit_af_state_machine _state_machine; \ } typedef struct { @@ -1538,7 +1499,7 @@ static void kit_async_function_dispatch(void *promise) { #define KIT_AF_FINISHED(promise_) ((promise_)._index == -1) #define KIT_AF_FINISHED_N(return_, promises_, size_) \ do { \ - int kit_af_index_; \ + i32 kit_af_index_; \ (return_) = 1; \ for (kit_af_index_ = 0; kit_af_index_ < (size_); \ kit_af_index_++) \ @@ -1604,13 +1565,14 @@ extern "C" { # define KIT_PATH_DELIM "/" # define KIT_ENV_HOME "HOME" #endif -kit_string_t kit_path_norm(kit_str_t path, kit_allocator_t *alloc); -kit_string_t kit_path_join(kit_str_t left, kit_str_t right, - kit_allocator_t *alloc); -kit_string_t kit_path_user(kit_allocator_t *alloc); -kit_string_t kit_path_cache(kit_allocator_t *alloc); -kit_str_t kit_path_index(kit_str_t path, ptrdiff_t index); -kit_str_t kit_path_take(kit_str_t path, ptrdiff_t count); +kit_str_builder_t kit_path_norm(kit_str_t path, + kit_allocator_t *alloc); +kit_str_builder_t kit_path_join(kit_str_t left, kit_str_t right, + kit_allocator_t *alloc); +kit_str_builder_t kit_path_user(kit_allocator_t *alloc); +kit_str_builder_t kit_path_cache(kit_allocator_t *alloc); +kit_str_t kit_path_index(kit_str_t path, i64 index); +kit_str_t kit_path_take(kit_str_t path, i64 count); kit_status_t kit_file_create_folder(kit_str_t path); kit_status_t kit_file_create_folder_recursive(kit_str_t path); kit_status_t kit_file_remove(kit_str_t path); @@ -1625,14 +1587,14 @@ typedef enum { kit_path_type_t kit_path_type(kit_str_t path); typedef struct { kit_status_t status; - int64_t time_modified_sec; - int32_t time_modified_nsec; - int64_t size; + i64 time_modified_sec; + i32 time_modified_nsec; + i64 size; } kit_file_info_t; kit_file_info_t kit_file_info(kit_str_t path); typedef struct { kit_status_t status; - KIT_DA(kit_string_t) files; + KIT_DA(kit_str_builder_t) files; } kit_path_list_t; kit_path_list_t kit_file_enum_folder(kit_str_t path, kit_allocator_t *alloc); @@ -1675,11 +1637,6 @@ void kit_path_list_destroy(kit_path_list_t list); *********************************************************************/ #ifndef KIT_MERSENNE_TWISTER_64_H #define KIT_MERSENNE_TWISTER_64_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include -#include #ifdef __cplusplus extern "C" { #endif @@ -1687,14 +1644,14 @@ enum { KIT_MT64_N = 312, }; typedef struct { - uint64_t mt[KIT_MT64_N]; - uint64_t index; + u64 mt[KIT_MT64_N]; + u64 index; } kit_mt64_state_t; -void kit_mt64_init_array(kit_mt64_state_t *state, ptrdiff_t size, - uint64_t *seed); -void kit_mt64_init(kit_mt64_state_t *state, uint64_t seed); -void kit_mt64_rotate(kit_mt64_state_t *state); -uint64_t kit_mt64_generate(kit_mt64_state_t *state); +void kit_mt64_init_array(kit_mt64_state_t *state, i64 size, + u64 *seed); +void kit_mt64_init(kit_mt64_state_t *state, u64 seed); +void kit_mt64_rotate(kit_mt64_state_t *state); +u64 kit_mt64_generate(kit_mt64_state_t *state); #ifndef KIT_DISABLE_SHORT_NAMES # define mt64_state_t kit_mt64_state_t # define mt64_init_array kit_mt64_init_array @@ -1713,15 +1670,10 @@ uint64_t kit_mt64_generate(kit_mt64_state_t *state); *********************************************************************/ #ifndef KIT_SECURE_RANDOM_H #define KIT_SECURE_RANDOM_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include -#include #ifdef __cplusplus extern "C" { #endif -void kit_secure_random(ptrdiff_t size, void *data); +void kit_secure_random(i64 size, void *data); #ifndef KIT_DISABLE_SHORT_NAMES # define secure_random kit_secure_random #endif @@ -1736,19 +1688,14 @@ void kit_secure_random(ptrdiff_t size, void *data); *********************************************************************/ #ifndef KIT_SHA256_H #define KIT_SHA256_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif -#include -#include #ifdef __cplusplus extern "C" { #endif enum { KIT_SHA256_BLOCK_SIZE = 32 }; typedef struct { - uint8_t v[KIT_SHA256_BLOCK_SIZE]; + u8 v[KIT_SHA256_BLOCK_SIZE]; } kit_sha256_hash_t; -kit_sha256_hash_t kit_sha256(ptrdiff_t size, uint8_t *data); +kit_sha256_hash_t kit_sha256(i64 size, u8 *data); #ifdef __cplusplus } #endif @@ -1765,9 +1712,6 @@ kit_sha256_hash_t kit_sha256(ptrdiff_t size, uint8_t *data); *********************************************************************/ #ifndef KIT_SOCKETS_H #define KIT_SOCKETS_H -#ifndef _GNU_SOURCE -# define _GNU_SOURCE -#endif #ifndef KIT_DISABLE_SYSTEM_SOCKETS # ifdef __GNUC__ # pragma GCC diagnostic push @@ -1779,7 +1723,7 @@ kit_sha256_hash_t kit_sha256(ptrdiff_t size, uint8_t *data); # include # include # define socket_t SOCKET -# define socklen_t int +# define socklen_t i32 # ifdef __cplusplus extern "C" { # endif @@ -1795,13 +1739,13 @@ static kit_status_t kit_sockets_cleanup(void) { WSACleanup(); return KIT_OK; } -static int kit_socket_set_blocking(socket_t s) { +static i32 kit_socket_set_blocking(socket_t s) { u_long flag = 0; return ioctlsocket(s, FIONBIO, &flag) == 0 ? KIT_OK : KIT_ERROR_SOCKET_CONTROL_FAILED; } -static int kit_socket_set_nonblocking(socket_t s) { +static i32 kit_socket_set_nonblocking(socket_t s) { u_long flag = 1; return ioctlsocket(s, FIONBIO, &flag) == 0 ? KIT_OK @@ -1821,7 +1765,7 @@ static int kit_socket_set_nonblocking(socket_t s) { # include # include # include -# define socket_t int +# define socket_t i32 # define closesocket close # define INVALID_SOCKET -1 # ifdef __cplusplus @@ -1834,14 +1778,14 @@ static kit_status_t kit_sockets_init(void) { static kit_status_t kit_sockets_cleanup(void) { return KIT_OK; } -static int kit_socket_set_blocking(socket_t s) { - int const flags = fcntl(s, F_GETFL, 0); +static i32 kit_socket_set_blocking(socket_t s) { + i32 flags = fcntl(s, F_GETFL, 0); return fcntl(s, F_SETFL, flags & ~O_NONBLOCK) == 0 ? KIT_OK : KIT_ERROR_SOCKET_CONTROL_FAILED; } -static int kit_socket_set_nonblocking(socket_t s) { - int const flags = fcntl(s, F_GETFL, 0); +static i32 kit_socket_set_nonblocking(socket_t s) { + i32 flags = fcntl(s, F_GETFL, 0); return fcntl(s, F_SETFL, flags | O_NONBLOCK) == 0 ? KIT_OK : KIT_ERROR_SOCKET_CONTROL_FAILED; @@ -1862,14 +1806,12 @@ static int kit_socket_set_nonblocking(socket_t s) { * * *********************************************************************/ #include -#include +#include #ifndef KIT_DISABLE_SYSTEM_MALLOC # include -# include #endif -static void *kit_allocate_default_(int request, ptrdiff_t size, - ptrdiff_t previous_size, - void *pointer) { +static void *kit_allocate_default_(i32 request, i64 size, + i64 previous_size, void *pointer) { #ifndef KIT_DISABLE_SYSTEM_MALLOC switch (request) { case KIT_ALLOCATE: @@ -1895,9 +1837,9 @@ static void *kit_allocate_default_(int request, ptrdiff_t size, return NULL; if (size == previous_size) return pointer; - uint8_t *p = NULL; + u8 *p = NULL; if (size > 0) { - p = (uint8_t *) malloc(size); + p = (u8 *) malloc(size); if (p != NULL) { if (size > 0 && previous_size > 0) memcpy(p, pointer, @@ -1926,9 +1868,9 @@ static void *kit_allocate_default_(int request, ptrdiff_t size, return NULL; } static void *kit_allocate_from_buffer_(kit_allocator_t *alloc, - int request, ptrdiff_t size, - ptrdiff_t previous_size, - void *pointer) { + i32 request, i64 size, + i64 previous_size, + void *pointer) { assert(alloc != NULL); assert(pointer == NULL || pointer < alloc->data); if (alloc == NULL) @@ -1963,7 +1905,7 @@ static void *kit_allocate_from_buffer_(kit_allocator_t *alloc, return NULL; if (previous_size != 0 && pointer == NULL) return NULL; - if ((uint8_t *) pointer + previous_size == alloc->data) { + if ((u8 *) pointer + previous_size == alloc->data) { if (alloc->size < size - previous_size) return NULL; alloc->bytes += size - previous_size; @@ -1972,7 +1914,7 @@ static void *kit_allocate_from_buffer_(kit_allocator_t *alloc, } if (alloc->size < size) return NULL; - uint8_t *p = alloc->bytes; + u8 *p = alloc->bytes; alloc->bytes += size; alloc->size -= size; if (previous_size > 0) @@ -1989,9 +1931,8 @@ static void *kit_allocate_from_buffer_(kit_allocator_t *alloc, return NULL; } #ifndef KIT_ENABLE_CUSTOM_ALLOC_DISPATCH -void *kit_alloc_dispatch(kit_allocator_t *alloc, int request, - ptrdiff_t size, ptrdiff_t previous_size, - void *pointer) { +void *kit_alloc_dispatch(kit_allocator_t *alloc, i32 request, + i64 size, i64 previous_size, void *pointer) { if (alloc == NULL) return kit_allocate_default_(request, size, previous_size, pointer); @@ -2013,7 +1954,7 @@ kit_allocator_t kit_alloc_default(void) { .data = NULL }; return alloc; } -kit_allocator_t kit_alloc_buffer(ptrdiff_t size, void *buffer) { +kit_allocator_t kit_alloc_buffer(i64 size, void *buffer) { kit_allocator_t alloc = { .type = KIT_ALLOC_TYPE_BUFFER, .size = size, .data = buffer }; @@ -2021,7 +1962,194 @@ kit_allocator_t kit_alloc_buffer(ptrdiff_t size, void *buffer) { } /********************************************************************* * * - * File: source/kit/thread.posix.c * + * File: source/kit/atomic.win32.c * + * * + *********************************************************************/ +#ifdef _MSC_VER +static_assert(sizeof(char) == 1, "Wrong char size"); +static_assert(sizeof(short) == 2, "Wrong short size"); +static_assert(sizeof(int) == 4, "Wrong int size"); +# include +void kit_atomic_store_explicit_8(u8 volatile *var, u8 value, + i32 memory_order) { + char volatile *dst = (char volatile *) var; + char src = (char) value; + switch (memory_order) { + case memory_order_relaxed: *dst = src; break; + default: _InterlockedExchange8(dst, src); + } +} +void kit_atomic_store_explicit_16(u16 volatile *var, u16 value, + i32 memory_order) { + short volatile *dst = (short volatile *) var; + short src = (short) value; + switch (memory_order) { + case memory_order_relaxed: *dst = src; break; + default: _InterlockedExchange16(dst, src); + } +} +void kit_atomic_store_explicit_32(u32 volatile *var, u32 value, + i32 memory_order) { + int volatile *dst = (int volatile *) var; + int src = (int) value; + switch (memory_order) { + case memory_order_relaxed: *dst = src; break; + default: _InterlockedExchange(dst, src); + } +} +void kit_atomic_store_explicit_64(u64 volatile *var, u64 value, + i32 memory_order) { + __int64 volatile *dst = (__int64 volatile *) var; + __int64 src = (__int64) value; + switch (memory_order) { + case memory_order_relaxed: *dst = src; break; + default: +# ifdef _WIN64 + _InterlockedExchange64(dst, src); +# else + assert(0); + _InterlockedExchange((int volatile *) dst, (int) src); +# endif + } +} +u8 kit_atomic_load_explicit_8(volatile u8 *var, i32 memory_order) { + char volatile *dst = (char volatile *) var; + if (memory_order == memory_order_relaxed) + return (u8) *dst; + return (u8) _InterlockedOr8(dst, 0); +} +u16 kit_atomic_load_explicit_16(u16 volatile *var, i32 memory_order) { + short volatile *dst = (short volatile *) var; + if (memory_order == memory_order_relaxed) + return (u16) *dst; + return (u16) _InterlockedOr16(dst, 0); +} +u32 kit_atomic_load_explicit_32(u32 volatile *var, i32 memory_order) { + int volatile *dst = (int volatile *) var; + if (memory_order == memory_order_relaxed) + return (u32) *dst; + return (u32) _InterlockedOr(dst, 0); +} +u64 kit_atomic_load_explicit_64(u64 volatile *var, i32 memory_order) { + __int64 volatile *dst = (__int64 volatile *) var; + if (memory_order == memory_order_relaxed) + return (u64) *dst; +# ifdef _WIN64 + return (u64) _InterlockedOr64(dst, 0); +# else + assert(0); + return (u64) _InterlockedOr((int volatile *) dst, 0); +# endif +} +u8 kit_atomic_exchange_explicit_8(volatile u8 *var, u8 value, + i32 memory_order) { + char volatile *dst = (char volatile *) var; + char src = (char) value; + return (u8) _InterlockedExchange8(dst, src); +} +u16 kit_atomic_exchange_explicit_16(u16 volatile *var, u16 value, + i32 memory_order) { + short volatile *dst = (short volatile *) var; + short src = (short) value; + return (u16) _InterlockedExchange16(dst, src); +} +u32 kit_atomic_exchange_explicit_32(u32 volatile *var, u32 value, + i32 memory_order) { + int volatile *dst = (int volatile *) var; + int src = (int) value; + return (u32) _InterlockedExchange(dst, src); +} +u64 kit_atomic_exchange_explicit_64(u64 volatile *var, u64 value, + i32 memory_order) { + __int64 volatile *dst = (__int64 volatile *) var; + __int64 src = (__int64) value; +# ifdef _WIN64 + return (u64) _InterlockedExchange64(dst, src); +# else + assert(0); + return (u64) _InterlockedExchange((int volatile *) dst, (int) src); +# endif +} +int kit_atomic_compare_exchange_explicit_8(volatile u8 *var, + u8 *expected, u8 value, + i32 memory_order_succ_, + i32 memory_order_fail_) { + char volatile *dst = (char volatile *) var; + char src = (char) value; + char exp = (char) *expected; + *expected = (u8) _InterlockedCompareExchange8(dst, src, exp); + return exp == (char) *expected; +} +int kit_atomic_compare_exchange_explicit_16(u16 volatile *var, + u16 *expected, u16 value, + i32 memory_order_succ_, + i32 memory_order_fail_) { + short volatile *dst = (short volatile *) var; + short src = (short) value; + short exp = (short) *expected; + *expected = (u16) _InterlockedCompareExchange16(dst, src, exp); + return exp == (short) *expected; +} +int kit_atomic_compare_exchange_explicit_32(u32 volatile *var, + u32 *expected, u32 value, + i32 memory_order_succ_, + i32 memory_order_fail_) { + int volatile *dst = (int volatile *) var; + int src = (int) value; + int exp = (int) *expected; + *expected = (u32) _InterlockedCompareExchange(dst, src, exp); + return exp == (int) *expected; +} +int kit_atomic_compare_exchange_explicit_64(u64 volatile *var, + u64 *expected, u64 value, + i32 memory_order_succ_, + i32 memory_order_fail_) { + __int64 volatile *dst = (__int64 volatile *) var; + __int64 src = (__int64) value; + __int64 exp = (__int64) *expected; +# ifdef _WIN64 + *expected = (u64) _InterlockedCompareExchange64(dst, src, exp); +# else + assert(0); + *expected = (u64) _InterlockedCompareExchange((int volatile *) dst, + (int) src, (int) exp); +# endif + return exp == (__int64) *expected; +} +u8 kit_atomic_fetch_add_explicit_8(volatile u8 *var, u8 value, + i32 memory_order) { + char volatile *dst = (char volatile *) var; + char src = (char) value; + return (u8) _InterlockedExchangeAdd8(dst, src); +} +u16 kit_atomic_fetch_add_explicit_16(u16 volatile *var, u16 value, + i32 memory_order) { + short volatile *dst = (short volatile *) var; + short src = (short) value; + return (u16) _InterlockedExchangeAdd16(dst, src); +} +u32 kit_atomic_fetch_add_explicit_32(u32 volatile *var, u32 value, + i32 memory_order) { + int volatile *dst = (int volatile *) var; + int src = (int) value; + return (u32) _InterlockedExchangeAdd(dst, src); +} +u64 kit_atomic_fetch_add_explicit_64(u64 volatile *var, u64 value, + i32 memory_order) { + __int64 volatile *dst = (__int64 volatile *) var; + __int64 src = (__int64) value; +# ifdef _WIN64 + return (u64) _InterlockedExchangeAdd64(dst, src); +# else + assert(0); + return (u64) _InterlockedExchangeAdd((int volatile *) dst, + (int) src); +# endif +} +#endif +/********************************************************************* + * * + * File: source/kit/threads.posix.c * * * *********************************************************************/ #ifndef KIT_DISABLE_SYSTEM_THREADS @@ -2030,7 +2158,6 @@ kit_allocator_t kit_alloc_buffer(ptrdiff_t size, void *buffer) { # include # include # include -# include /* intptr_t */ # include # include # ifndef PTHREAD_STACK_MIN @@ -2258,7 +2385,7 @@ void thrd_yield(void) { #endif /********************************************************************* * * - * File: source/kit/thread.win32.c * + * File: source/kit/threads.win32.c * * * *********************************************************************/ #ifndef KIT_DISABLE_SYSTEM_THREADS @@ -2267,12 +2394,11 @@ void thrd_yield(void) { # include # include # include -# include # include # ifndef WIN32_LEAN_AND_MEAN # define WIN32_LEAN_AND_MEAN 1 # endif -# include +# include /* Configuration macro: EMULATED_THREADS_USE_NATIVE_CALL_ONCE @@ -2318,7 +2444,7 @@ typedef struct { } impl_thrd_param_t; struct thrd_state { thrd_t thrd; - bool handle_need_close; + int handle_need_close; }; static thread_local struct thrd_state impl_current_thread = { 0 }; static unsigned __stdcall impl_thrd_routine(void *p) { @@ -2326,7 +2452,7 @@ static unsigned __stdcall impl_thrd_routine(void *p) { impl_thrd_param_t pack; int code; impl_current_thread.thrd = pack_p->thrd; - impl_current_thread.handle_need_close = false; + impl_current_thread.handle_need_close = 0; memcpy(&pack, pack_p, sizeof(impl_thrd_param_t)); kit_alloc_dispatch(NULL, KIT_DEALLOCATE, 0, 0, p); code = pack.func(pack.arg); @@ -2546,7 +2672,7 @@ thrd_t thrd_current(void) { 0, FALSE, DUPLICATE_SAME_ACCESS)) { abort(); } - state->handle_need_close = true; + state->handle_need_close = 1; } return state->thrd; } @@ -2591,166 +2717,6 @@ void thrd_yield(void) { } # endif #endif -/********************************************************************* - * * - * File: source/kit/atomic.win32.c * - * * - *********************************************************************/ -#ifdef _MSC_VER -static_assert(sizeof(char) == 1, "Wrong char size"); -static_assert(sizeof(short) == 2, "Wrong short size"); -static_assert(sizeof(int) == 4, "Wrong int size"); -# include -void kit_atomic_store_explicit_8(uint8_t volatile *var, uint8_t value, - int memory_order) { - char volatile *dst = (char volatile *) var; - char src = (char) value; - switch (memory_order) { - case memory_order_relaxed: *dst = src; break; - default: _InterlockedExchange8(dst, src); - } -} -void kit_atomic_store_explicit_16(uint16_t volatile *var, - uint16_t value, int memory_order) { - short volatile *dst = (short volatile *) var; - short src = (short) value; - switch (memory_order) { - case memory_order_relaxed: *dst = src; break; - default: _InterlockedExchange16(dst, src); - } -} -void kit_atomic_store_explicit_32(uint32_t volatile *var, - uint32_t value, int memory_order) { - int volatile *dst = (int volatile *) var; - int src = (int) value; - switch (memory_order) { - case memory_order_relaxed: *dst = src; break; - default: _InterlockedExchange(dst, src); - } -} -void kit_atomic_store_explicit_64(uint64_t volatile *var, - uint64_t value, int memory_order) { - __int64 volatile *dst = (__int64 volatile *) var; - __int64 src = (__int64) value; - switch (memory_order) { - case memory_order_relaxed: *dst = src; break; - default: -# ifdef _WIN64 - _InterlockedExchange64(dst, src); -# else - _InterlockedExchange((int volatile *) dst, (int) src); -# endif - } -} -uint8_t kit_atomic_load_explicit_8(volatile uint8_t *var, - int memory_order) { - char volatile *dst = (char volatile *) var; - if (memory_order == memory_order_relaxed) - return (uint8_t) *dst; - return (uint8_t) _InterlockedOr8(dst, 0); -} -uint16_t kit_atomic_load_explicit_16(uint16_t volatile *var, - int memory_order) { - short volatile *dst = (short volatile *) var; - if (memory_order == memory_order_relaxed) - return (uint16_t) *dst; - return (uint16_t) _InterlockedOr16(dst, 0); -} -uint32_t kit_atomic_load_explicit_32(uint32_t volatile *var, - int memory_order) { - int volatile *dst = (int volatile *) var; - if (memory_order == memory_order_relaxed) - return (uint32_t) *dst; - return (uint32_t) _InterlockedOr(dst, 0); -} -uint64_t kit_atomic_load_explicit_64(uint64_t volatile *var, - int memory_order) { - __int64 volatile *dst = (__int64 volatile *) var; - if (memory_order == memory_order_relaxed) - return (uint64_t) *dst; -# ifdef _WIN64 - return (uint64_t) _InterlockedOr64(dst, 0); -# else - return (uint64_t) _InterlockedOr((int volatile *) dst, 0); -# endif -} -uint8_t kit_atomic_exchange_explicit_8(volatile uint8_t *var, - uint8_t value, - int memory_order) { - char volatile *dst = (char volatile *) var; - char src = (char) value; - return (uint8_t) _InterlockedExchange8(dst, src); -} -uint16_t kit_atomic_exchange_explicit_16(uint16_t volatile *var, - uint16_t value, - int memory_order) { - short volatile *dst = (short volatile *) var; - short src = (short) value; - return (uint16_t) _InterlockedExchange16(dst, src); -} -uint32_t kit_atomic_exchange_explicit_32(uint32_t volatile *var, - uint32_t value, - int memory_order) { - int volatile *dst = (int volatile *) var; - int src = (int) value; - return (uint32_t) _InterlockedExchange(dst, src); -} -uint64_t kit_atomic_exchange_explicit_64(uint64_t volatile *var, - uint64_t value, - int memory_order) { - __int64 volatile *dst = (__int64 volatile *) var; - __int64 src = (__int64) value; -# ifdef _WIN64 - return (uint64_t) _InterlockedExchange64(dst, src); -# else - return (uint64_t) _InterlockedExchange((int volatile *) dst, - (int) src); -# endif -} -uint8_t kit_atomic_fetch_add_explicit_8(volatile uint8_t *var, - uint8_t value, - int memory_order) { - char volatile *dst = (char volatile *) var; - char src = (char) value; - return (uint8_t) _InterlockedExchangeAdd8(dst, src); -} -uint16_t kit_atomic_fetch_add_explicit_16(uint16_t volatile *var, - uint16_t value, - int memory_order) { - short volatile *dst = (short volatile *) var; - short src = (short) value; - return (uint16_t) _InterlockedExchangeAdd16(dst, src); -} -uint32_t kit_atomic_fetch_add_explicit_32(uint32_t volatile *var, - uint32_t value, - int memory_order) { - int volatile *dst = (int volatile *) var; - int src = (int) value; - return (uint32_t) _InterlockedExchangeAdd(dst, src); -} -uint64_t kit_atomic_fetch_add_explicit_64(uint64_t volatile *var, - uint64_t value, - int memory_order) { - __int64 volatile *dst = (__int64 volatile *) var; - __int64 src = (__int64) value; -# ifdef _WIN64 - return (uint64_t) _InterlockedExchangeAdd64(dst, src); -# else - return (uint64_t) _InterlockedExchangeAdd((int volatile *) dst, - (int) src); -# endif -} -#endif -/********************************************************************* - * * - * File: source/kit/mutex.c * - * * - *********************************************************************/ -/********************************************************************* - * * - * File: source/kit/condition_variable.c * - * * - *********************************************************************/ /********************************************************************* * * * File: source/kit/dynamic_array.c * @@ -2758,8 +2724,8 @@ uint64_t kit_atomic_fetch_add_explicit_64(uint64_t volatile *var, *********************************************************************/ #include #include -void kit_da_init(kit_da_void_t *array, ptrdiff_t element_size, - ptrdiff_t size, kit_allocator_t *alloc) { +void kit_da_init(kit_da_void_t *array, i64 element_size, i64 size, + kit_allocator_t *alloc) { assert(array != NULL); assert(element_size > 0); assert(size >= 0); @@ -2773,23 +2739,21 @@ void kit_da_init(kit_da_void_t *array, ptrdiff_t element_size, } array->alloc = alloc; } -static ptrdiff_t eval_capacity(ptrdiff_t current_cap, - ptrdiff_t required_cap) { +static i64 eval_capacity(i64 current_cap, i64 required_cap) { if (current_cap == 0) return required_cap; - ptrdiff_t cap = current_cap; + i64 cap = current_cap; while (cap < required_cap) cap *= 2; return cap; } -void kit_da_resize(kit_da_void_t *array, ptrdiff_t element_size, - ptrdiff_t size) { +void kit_da_resize(kit_da_void_t *array, i64 element_size, i64 size) { assert(array != NULL); assert(element_size > 0); assert(size >= 0); if (size <= array->capacity) { array->size = size; } else { - ptrdiff_t capacity = eval_capacity(array->capacity, size); + i64 capacity = eval_capacity(array->capacity, size); void *bytes = kit_alloc_dispatch( array->alloc, KIT_ALLOCATE, element_size * capacity, 0, NULL); if (bytes != NULL) { @@ -2804,8 +2768,8 @@ void kit_da_resize(kit_da_void_t *array, ptrdiff_t element_size, } } } -void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, - ptrdiff_t capacity) { +void kit_da_resize_exact(kit_da_void_t *array, i64 element_size, + i64 capacity) { assert(array != NULL); assert(element_size > 0); assert(capacity >= 0); @@ -2825,6 +2789,11 @@ void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, array->values = bytes; } } +/********************************************************************* + * * + * File: source/kit/string_builder.c * + * * + *********************************************************************/ /********************************************************************* * * * File: source/kit/input_stream.c * @@ -2833,43 +2802,42 @@ void kit_da_resize_exact(kit_da_void_t *array, ptrdiff_t element_size, #include enum { KIT_INPUT_STREAM_STR, KIT_INPUT_STREAM_FILE }; typedef struct { - ptrdiff_t type; + i64 type; kit_allocator_t *alloc; } kit_is_state_basic_t; typedef struct { - ptrdiff_t type; + i64 type; kit_allocator_t *alloc; kit_str_t string; } kit_is_state_str_t; typedef struct { - ptrdiff_t type; + i64 type; kit_allocator_t *alloc; FILE *file; } kit_is_state_file_t; -static int kit_is_check_type_(void *state, ptrdiff_t type) { +static int kit_is_check_type_(void *state, i64 type) { kit_is_state_basic_t *basic = (kit_is_state_basic_t *) state; return basic != NULL && basic->type == type; } -static ptrdiff_t kit_read_str_(void *state, kit_str_t destination) { +static i64 kit_read_str_(void *state, kit_str_t destination) { if (!kit_is_check_type_(state, KIT_INPUT_STREAM_STR)) return 0; - kit_is_state_str_t *str = (kit_is_state_str_t *) state; - ptrdiff_t size = destination.size < str->string.size - ? destination.size - : str->string.size; + kit_is_state_str_t *str = (kit_is_state_str_t *) state; + i64 size = destination.size < str->string.size ? destination.size + : str->string.size; memcpy(destination.values, str->string.values, size); str->string.values += size; str->string.size -= size; return size; } -static ptrdiff_t kit_read_file_(void *state, kit_str_t destination) { +static i64 kit_read_file_(void *state, kit_str_t destination) { if (!kit_is_check_type_(state, KIT_INPUT_STREAM_FILE)) return 0; kit_is_state_file_t *f = (kit_is_state_file_t *) state; if (f->file == NULL || feof(f->file)) return 0; - ptrdiff_t size = (ptrdiff_t) fread(destination.values, 1, - destination.size, f->file); + i64 size = (i64) fread(destination.values, 1, destination.size, + f->file); if (size <= 0) return 0; return size; @@ -2920,10 +2888,10 @@ void kit_is_destroy(kit_is_handle_t in) { #include #include typedef struct { - ptrdiff_t ref_count; - kit_is_handle_t upstream; - kit_allocator_t *alloc; - kit_string_t data; + i64 ref_count; + kit_is_handle_t upstream; + kit_allocator_t *alloc; + kit_str_builder_t data; } internal_buffer_t; static internal_buffer_t *kit_buf_init_(kit_is_handle_t upstream, kit_allocator_t *alloc) { @@ -2954,23 +2922,22 @@ static void kit_buf_release_(void *p) { kit_alloc_dispatch(buf->alloc, KIT_DEALLOCATE, 0, 0, buf); } } -static void kit_buf_adjust_(void *p, ptrdiff_t size) { +static void kit_buf_adjust_(void *p, i64 size) { assert(p != NULL); assert(size >= 0); internal_buffer_t *buf = (internal_buffer_t *) p; - ptrdiff_t offset = buf->data.size; + i64 offset = buf->data.size; if (offset < size) { DA_RESIZE(buf->data, size); kit_str_t destination = { .size = size - offset, .values = buf->data.values + offset }; - ptrdiff_t n = KIT_IS_READ(buf->upstream, destination); + i64 n = KIT_IS_READ(buf->upstream, destination); DA_RESIZE(buf->data, offset + n); } } -static ptrdiff_t kit_buf_read_(void *p, ptrdiff_t offset, - kit_str_t destination) { +static i64 kit_buf_read_(void *p, i64 offset, kit_str_t destination) { internal_buffer_t *buf = (internal_buffer_t *) p; - ptrdiff_t n = destination.size < buf->data.size - offset + i64 n = destination.size < buf->data.size - offset ? destination.size : buf->data.size - offset; memcpy(destination.values, buf->data.values + offset, n); @@ -2987,7 +2954,7 @@ kit_ib_handle_t kit_ib_wrap(kit_is_handle_t upstream, buf.status = KIT_ERROR_BAD_ALLOC; return buf; } -kit_ib_handle_t kit_ib_read(kit_ib_handle_t buf, ptrdiff_t size) { +kit_ib_handle_t kit_ib_read(kit_ib_handle_t buf, i64 size) { kit_ib_handle_t next; memset(&next, 0, sizeof next); if (buf.status != KIT_OK) { @@ -3001,7 +2968,7 @@ kit_ib_handle_t kit_ib_read(kit_ib_handle_t buf, ptrdiff_t size) { next.status = KIT_ERROR_BAD_ALLOC; kit_str_t destination = { .size = next.data.size, .values = next.data.values }; - ptrdiff_t n = kit_buf_read_(buf.internal, buf.offset, destination); + i64 n = kit_buf_read_(buf.internal, buf.offset, destination); next.offset = buf.offset + n; next.internal = buf.internal; DA_RESIZE(next.data, n); @@ -3019,7 +2986,7 @@ kit_ib_handle_t kit_ib_read_while( } kit_buf_acquire_(buf.internal); DA_INIT(next.data, 0, kit_buf_alloc_(buf.internal)); - ptrdiff_t size = 0; + i64 size = 0; for (;; ++size) { kit_buf_adjust_(buf.internal, buf.offset + size + 1); DA_RESIZE(next.data, size + 1); @@ -3027,7 +2994,7 @@ kit_ib_handle_t kit_ib_read_while( next.status = KIT_ERROR_BAD_ALLOC; kit_str_t destination = { .size = 1, .values = next.data.values + size }; - ptrdiff_t n = kit_buf_read_(buf.internal, buf.offset + size, + i64 n = kit_buf_read_(buf.internal, buf.offset + size, destination); kit_str_t data = { .size = size + 1, .values = next.data.values }; if (n != 1 || condition == NULL || condition(data) == 0) @@ -3068,13 +3035,14 @@ enum { PATH_BUF_SIZE = 4096 }; #ifdef __APPLE__ # define st_mtim st_mtimespec #endif -static int is_delim(char c) { +static i32 is_delim(char c) { return c == '/' || c == '\\'; } -static kit_string_t kit_get_env_(char *name, kit_allocator_t *alloc) { - char *val = getenv(name); - ptrdiff_t size = val != NULL ? (ptrdiff_t) strlen(val) : 0; - string_t result; +static kit_str_builder_t kit_get_env_(char *name, + kit_allocator_t *alloc) { + char *val = getenv(name); + i64 size = val != NULL ? (i64) strlen(val) : 0; + str_builder_t result; DA_INIT(result, size, alloc); assert(result.size == size); if (result.size == size && size > 0) @@ -3083,10 +3051,11 @@ static kit_string_t kit_get_env_(char *name, kit_allocator_t *alloc) { DA_RESIZE(result, 0); return result; } -kit_string_t kit_path_norm(kit_str_t path, kit_allocator_t *alloc) { - str_t parent = SZ(".."); - ptrdiff_t i, i1, j; - string_t norm; +kit_str_builder_t kit_path_norm(kit_str_t path, + kit_allocator_t *alloc) { + str_t parent = SZ(".."); + i64 i, i1, j; + str_builder_t norm; DA_INIT(norm, path.size, alloc); assert(norm.size == path.size); if (norm.size != path.size) @@ -3097,8 +3066,8 @@ kit_string_t kit_path_norm(kit_str_t path, kit_allocator_t *alloc) { continue; str_t s = { .size = i - i1 - 1, .values = path.values + i1 + 1 }; if (AR_EQUAL(s, parent)) { - int have_parent = 0; - ptrdiff_t i0 = 0; + i32 have_parent = 0; + i64 i0 = 0; for (j = 0; j < i1; j++) { if (norm.values[j] != '\0') have_parent = 1; @@ -3113,7 +3082,7 @@ kit_string_t kit_path_norm(kit_str_t path, kit_allocator_t *alloc) { } i1 = i; } - ptrdiff_t size = 0; + i64 size = 0; for (i = 0; i < norm.size; i++) { if (norm.values[i] != '\0') { if (is_delim(norm.values[i])) @@ -3126,18 +3095,18 @@ kit_string_t kit_path_norm(kit_str_t path, kit_allocator_t *alloc) { norm.size = size; return norm; } -kit_string_t kit_path_join(kit_str_t left, kit_str_t right, - kit_allocator_t *alloc) { - ptrdiff_t left_size = left.size; - ptrdiff_t right_size = right.size; - char *right_values = right.values; +kit_str_builder_t kit_path_join(kit_str_t left, kit_str_t right, + kit_allocator_t *alloc) { + i64 left_size = left.size; + i64 right_size = right.size; + char *right_values = right.values; if (left_size > 0 && is_delim(left.values[left_size - 1])) left_size--; if (right_size > 0 && is_delim(right.values[0])) { right_size--; right_values++; } - kit_string_t joined; + kit_str_builder_t joined; DA_INIT(joined, left_size + right_size + 1, alloc); assert(joined.size == left_size + right_size + 1); if (joined.size != left_size + right_size + 1) @@ -3147,8 +3116,8 @@ kit_string_t kit_path_join(kit_str_t left, kit_str_t right, memcpy(joined.values + left_size + 1, right_values, right_size); return joined; } -kit_string_t kit_path_user(kit_allocator_t *alloc) { - kit_string_t user = kit_get_env_(KIT_ENV_HOME, alloc); +kit_str_builder_t kit_path_user(kit_allocator_t *alloc) { + kit_str_builder_t user = kit_get_env_(KIT_ENV_HOME, alloc); if (user.size == 0) { DA_RESIZE(user, 1); if (user.size == 1) @@ -3156,8 +3125,8 @@ kit_string_t kit_path_user(kit_allocator_t *alloc) { } return user; } -kit_string_t kit_path_cache(kit_allocator_t *alloc) { - kit_string_t cache, user; +kit_str_builder_t kit_path_cache(kit_allocator_t *alloc) { + kit_str_builder_t cache, user; #if defined(_WIN32) && !defined(__CYGWIN__) cache = kit_get_env_("LOCALAPPDATA", alloc); if (cache.size != 0) @@ -3179,11 +3148,11 @@ kit_string_t kit_path_cache(kit_allocator_t *alloc) { DA_DESTROY(user); return cache; } -kit_str_t kit_path_index(kit_str_t path, ptrdiff_t index) { +kit_str_t kit_path_index(kit_str_t path, i64 index) { str_t s = { .size = 0, .values = NULL }; - ptrdiff_t i0 = 0; - ptrdiff_t i = 0; - ptrdiff_t n = 0; + i64 i0 = 0; + i64 i = 0; + i64 n = 0; for (; i < path.size; i++) { if (!is_delim(path.values[i])) continue; @@ -3202,11 +3171,11 @@ kit_str_t kit_path_index(kit_str_t path, ptrdiff_t index) { } return s; } -kit_str_t kit_path_take(kit_str_t path, ptrdiff_t count) { +kit_str_t kit_path_take(kit_str_t path, i64 count) { str_t s = { .size = 0, .values = path.values }; - ptrdiff_t i0 = 0; - ptrdiff_t i = 0; - ptrdiff_t n = 0; + i64 i0 = 0; + i64 i = 0; + i64 n = 0; for (; i < path.size; i++) { if (!is_delim(path.values[i])) continue; @@ -3232,7 +3201,7 @@ static void win32_prepare_path_(WCHAR *buf, kit_str_t path) { buf[2] = L'?'; buf[3] = L'\\'; if (path.size > 0 && path.size + 5 < PATH_BUF_SIZE) - for (ptrdiff_t i = 0; i < path.size; i++) { + for (i64 i = 0; i < path.size; i++) { if (path.values[i] == '/') buf[4 + i] = L'\\'; else @@ -3264,10 +3233,10 @@ kit_status_t kit_file_create_folder(kit_str_t path) { #endif } kit_status_t kit_file_create_folder_recursive(kit_str_t path) { - ptrdiff_t i; + i64 i; for (i = 0;; i++) { str_t part = kit_path_take(path, i); - int type = kit_path_type(part); + i32 type = kit_path_type(part); if (type == KIT_PATH_FILE) return KIT_ERROR_FILE_ALREADY_EXISTS; if (type == KIT_PATH_NONE) { @@ -3298,8 +3267,8 @@ kit_status_t kit_file_remove_folder(kit_str_t path) { } kit_status_t kit_file_remove_recursive(kit_str_t path, kit_allocator_t *alloc) { - int type = kit_path_type(path); - ptrdiff_t i; + i32 type = kit_path_type(path); + i64 i; switch (type) { case KIT_PATH_FILE: return kit_file_remove(path); case KIT_PATH_FOLDER: { @@ -3350,18 +3319,16 @@ kit_file_info_t kit_file_info(kit_str_t path) { if (f != INVALID_HANDLE_VALUE) { FILETIME ft; if (GetFileTime(f, NULL, NULL, &ft) != 0) { - uint64_t nsec100 = (((uint64_t) ft.dwHighDateTime) << 32) | - (uint64_t) ft.dwLowDateTime; - result.time_modified_sec = (int64_t) (nsec100 / 10000000); - result.time_modified_nsec = (int32_t) (100 * - (nsec100 % 10000000)); + ui64 nsec100 = (((ui64) ft.dwHighDateTime) << 32) | + (ui64) ft.dwLowDateTime; + result.time_modified_sec = (i64) (nsec100 / 10000000); + result.time_modified_nsec = (i32) (100 * (nsec100 % 10000000)); } else { assert(0); } DWORD high; DWORD low = GetFileSize(f, &high); - result.size = (int64_t) ((((uint64_t) high) << 32) | - (uint64_t) low); + result.size = (i64) ((((ui64) high) << 32) | (ui64) low); result.status = KIT_OK; CloseHandle(f); return result; @@ -3369,14 +3336,14 @@ kit_file_info_t kit_file_info(kit_str_t path) { #else struct stat info; if (stat(buf, &info) == 0 && S_ISREG(info.st_mode)) { - result.size = (int64_t) info.st_size; + result.size = (i64) info.st_size; # ifndef st_mtime /* No support for nanosecond timestamps. */ - result.time_modified_sec = (int64_t) info.st_mtime; + result.time_modified_sec = (i64) info.st_mtime; # else - result.time_modified_sec = (int64_t) info.st_mtim.tv_sec; - result.time_modified_nsec = (int32_t) info.st_mtim.tv_nsec; + result.time_modified_sec = (i64) info.st_mtim.tv_sec; + result.time_modified_nsec = (i32) info.st_mtim.tv_nsec; # endif result.status = KIT_OK; return result; @@ -3402,13 +3369,13 @@ kit_path_list_t kit_file_enum_folder(kit_str_t path, if (find == INVALID_HANDLE_VALUE) return result; do { - ptrdiff_t n = result.files.size; + i64 n = result.files.size; DA_RESIZE(result.files, n + 1); if (result.files.size != n + 1) { result.status = KIT_ERROR_BAD_ALLOC; break; } - ptrdiff_t size = 0; + i64 size = 0; while (size < MAX_PATH && data.cFileName[size] != L'\0') size++; DA_INIT(result.files.values[n], size, alloc); if (result.files.values[n].size != size) { @@ -3416,7 +3383,7 @@ kit_path_list_t kit_file_enum_folder(kit_str_t path, result.status = KIT_ERROR_BAD_ALLOC; break; } - for (ptrdiff_t i = 0; i < size; i++) + for (i64 i = 0; i < size; i++) result.files.values[n].values[i] = data.cFileName[i]; } while (FindNextFileW(find, &data) != 0); FindClose(find); @@ -3430,13 +3397,13 @@ kit_path_list_t kit_file_enum_folder(kit_str_t path, break; if (entry->d_name[0] == '.') continue; - ptrdiff_t n = result.files.size; + i64 n = result.files.size; DA_RESIZE(result.files, n + 1); if (result.files.size != n + 1) { result.status = KIT_ERROR_BAD_ALLOC; break; } - ptrdiff_t size = (ptrdiff_t) strlen(entry->d_name); + i64 size = (i64) strlen(entry->d_name); DA_INIT(result.files.values[n], size, alloc); if (result.files.values[n].size != size) { DA_RESIZE(result.files, n); @@ -3451,7 +3418,7 @@ kit_path_list_t kit_file_enum_folder(kit_str_t path, return result; } void kit_path_list_destroy(kit_path_list_t list) { - ptrdiff_t i; + i64 i; for (i = 0; i < list.files.size; i++) DA_DESTROY(list.files.values[i]); DA_DESTROY(list.files); @@ -3465,9 +3432,9 @@ void kit_path_list_destroy(kit_path_list_t list) { #define MATRIX_A 0xb5026f5aa96619e9ull #define UM 0xffffffff80000000ull #define LM 0x7fffffffull -void kit_mt64_init_array(kit_mt64_state_t *state, ptrdiff_t size, - uint64_t *seed) { - ptrdiff_t i; +void kit_mt64_init_array(kit_mt64_state_t *state, i64 size, + u64 *seed) { + i64 i; for (i = 0; i < size && i < KIT_MT64_N; i++) state->mt[i] = seed[i]; for (state->index = size; state->index < KIT_MT64_N; state->index++) state->mt[state->index] = (6364136223846793005ull * @@ -3476,38 +3443,42 @@ void kit_mt64_init_array(kit_mt64_state_t *state, ptrdiff_t size, 62u)) + state->index); } -void kit_mt64_init(kit_mt64_state_t *state, uint64_t seed) { +void kit_mt64_init(kit_mt64_state_t *state, u64 seed) { kit_mt64_init_array(state, 1, &seed); } void kit_mt64_rotate(kit_mt64_state_t *state) { - static uint64_t mag01[2] = { 0ull, MATRIX_A }; - uint64_t x; - int i; + static u64 mag01[2] = { 0ull, MATRIX_A }; + u64 x; + i32 i; for (i = 0; i < KIT_MT64_N - MM; i++) { x = (state->mt[i] & UM) | (state->mt[i + 1] & LM); state->mt[i] = state->mt[i + MM] ^ (x >> 1u) ^ - mag01[(int) (x & 1ull)]; + mag01[(i32) (x & 1ull)]; } for (; i < KIT_MT64_N - 1; i++) { x = (state->mt[i] & UM) | (state->mt[i + 1] & LM); state->mt[i] = state->mt[i + (MM - KIT_MT64_N)] ^ (x >> 1u) ^ - mag01[(int) (x & 1ull)]; + mag01[(i32) (x & 1ull)]; } x = (state->mt[KIT_MT64_N - 1] & UM) | (state->mt[0] & LM); state->mt[KIT_MT64_N - 1] = state->mt[MM - 1] ^ (x >> 1u) ^ - mag01[(int) (x & 1ull)]; + mag01[(i32) (x & 1ull)]; state->index = 0; } -uint64_t kit_mt64_generate(kit_mt64_state_t *state) { +u64 kit_mt64_generate(kit_mt64_state_t *state) { if (state->index >= KIT_MT64_N) kit_mt64_rotate(state); - uint64_t x = state->mt[state->index++]; + u64 x = state->mt[state->index++]; x ^= (x >> 29u) & 0x5555555555555555ull; x ^= (x << 17u) & 0x71d67fffeda60000ull; x ^= (x << 37u) & 0xfff7eee000000000ull; x ^= (x >> 43u); return x; } +#undef MM +#undef MATRIX_A +#undef UM +#undef LM /********************************************************************* * * * File: source/kit/secure_random.c * @@ -3515,93 +3486,36 @@ uint64_t kit_mt64_generate(kit_mt64_state_t *state) { *********************************************************************/ #include #include +#include #if defined(_WIN32) && !defined(__CYGWIN__) # ifndef WIN32_LEAN_AND_MEAN # define WIN32_LEAN_AND_MEAN 1 # endif -# include +# include +# include #else # include #endif -static uint64_t get_available_memory(void) { -#if defined(_WIN32) && !defined(__CYGWIN__) - MEMORYSTATUSEX status; - status.dwLength = sizeof(status); - GlobalMemoryStatusEx(&status); - return (uint64_t) status.ullTotalPhys; -#else - uint64_t pages = (uint64_t) sysconf(_SC_PHYS_PAGES); - uint64_t page_size = (uint64_t) sysconf(_SC_PAGE_SIZE); - return pages * page_size; -#endif -} -#ifndef KIT_DISABLE_SYSTEM_THREADS -static once_flag kit_secure_random_fallback_flag; -static mtx_t kit_secure_random_fallback_mutex; -static void secure_random_fallback_init(void) { - mtx_init(&kit_secure_random_fallback_mutex, mtx_plain); -} -#endif -static void secure_random_fallback(ptrdiff_t size, void *data) { -#ifndef KIT_DISABLE_SYSTEM_THREADS - call_once(&kit_secure_random_fallback_flag, - secure_random_fallback_init); - mtx_lock(&kit_secure_random_fallback_mutex); -#endif - /* Try to get some unpredictable system properties and use them to - * seed the pseudo random number generator. - */ - static int8_t first_run = 1; - static uint64_t n = 0; - static uint64_t time_sec = 0; - static uint64_t time_nsec = 0; - struct timespec t; - timespec_get(&t, TIME_UTC); - kit_mt64_state_t state; - if (first_run) { - first_run = 0; - uint64_t seed[] = { n, get_available_memory(), - (uint64_t) t.tv_sec, (uint64_t) t.tv_nsec }; - kit_mt64_init_array(&state, sizeof seed / sizeof *seed, seed); - } else { - uint64_t seed[] = { n, - get_available_memory(), - (uint64_t) t.tv_sec, - (uint64_t) t.tv_nsec, - (uint64_t) t.tv_sec - time_sec, - (uint64_t) t.tv_nsec - time_nsec }; - kit_mt64_init_array(&state, sizeof seed / sizeof *seed, seed); - } - kit_mt64_rotate(&state); - n = kit_mt64_generate(&state); - time_sec = (uint64_t) t.tv_sec; - time_nsec = (uint64_t) t.tv_nsec; - for (ptrdiff_t i = 0; i < size; i++) - ((uint8_t *) data)[i] = (uint8_t) (kit_mt64_generate(&state) >> - 56); -#ifndef KIT_DISABLE_SYSTEM_THREADS - mtx_unlock(&kit_secure_random_fallback_mutex); -#endif -} -void kit_secure_random(ptrdiff_t size, void *data) { - assert(size > 0); +void kit_secure_random(i64 size, void *data) { + assert(size >= 0); assert(data != NULL); if (size <= 0 || data == NULL) return; #if defined(_WIN32) && !defined(__CYGWIN__) - secure_random_fallback(size, data); + HCRYPTPROV prov = NULL; + if (!CryptAcquireContextW(&prov, NULL, NULL, PROV_RSA_FULL, + CRYPT_VERIFYCONTEXT | CRYPT_SILENT) || + !CryptGenRandom(prov, (DWORD) size, (BYTE *) data) || + !CryptReleaseContext(prov, 0)) + abort(); #else FILE *f = fopen("/dev/urandom", "rb"); - assert(f != NULL); - if (f == NULL) { - secure_random_fallback(size, data); - return; - } - size_t n = fread(data, 1, size, f); + if (f == NULL) + abort(); + i64 n = (i64) fread(data, 1, size, f); fclose(f); - assert(n == size); if (n != size) - secure_random_fallback(size, data); + abort(); #endif } /********************************************************************* @@ -3619,7 +3533,7 @@ void kit_secure_random(ptrdiff_t size, void *data) { #define EP1(x) (ROTRIGHT(x, 6) ^ ROTRIGHT(x, 11) ^ ROTRIGHT(x, 25)) #define SIG0(x) (ROTRIGHT(x, 7) ^ ROTRIGHT(x, 18) ^ ((x) >> 3)) #define SIG1(x) (ROTRIGHT(x, 17) ^ ROTRIGHT(x, 19) ^ ((x) >> 10)) -static uint32_t kit_sha256_k[64] = { +static u32 kit_sha256_k[64] = { 0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5, 0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, @@ -3634,14 +3548,13 @@ static uint32_t kit_sha256_k[64] = { 0x682e6ff3, 0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2 }; -static void kit_sha256_transform(uint32_t *state, uint8_t *data) { +static void kit_sha256_transform(u32 *state, u8 *data) { assert(state != NULL); assert(data != NULL); - uint32_t a, b, c, d, e, f, g, h, i, j, t1, t2, m[64]; + u32 a, b, c, d, e, f, g, h, i, j, t1, t2, m[64]; for (i = 0, j = 0; i < 16; ++i, j += 4) - m[i] = ((uint32_t) data[j] << 24) | - ((uint32_t) data[j + 1] << 16) | - ((uint32_t) data[j + 2] << 8) | ((uint32_t) data[j + 3]); + m[i] = ((u32) data[j] << 24) | ((u32) data[j + 1] << 16) | + ((u32) data[j + 2] << 8) | ((u32) data[j + 3]); for (; i < 64; ++i) m[i] = SIG1(m[i - 2]) + m[i - 7] + SIG0(m[i - 15]) + m[i - 16]; a = state[0]; @@ -3673,16 +3586,15 @@ static void kit_sha256_transform(uint32_t *state, uint8_t *data) { state[6] += g; state[7] += h; } -kit_sha256_hash_t kit_sha256(ptrdiff_t in_size, uint8_t *in_data) { +kit_sha256_hash_t kit_sha256(i64 in_size, u8 *in_data) { assert(in_size >= 0); assert(in_data != NULL); - uint32_t state[8] = { 0x6a09e667, 0xbb67ae85, 0x3c6ef372, - 0xa54ff53a, 0x510e527f, 0x9b05688c, - 0x1f83d9ab, 0x5be0cd19 }; - uint8_t data[64]; - ptrdiff_t i; - ptrdiff_t datalen = 0; - uint64_t bitlen = 0; + u32 state[8] = { 0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, + 0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19 }; + u8 data[64]; + i64 i; + i64 datalen = 0; + u64 bitlen = 0; if (in_data != NULL) for (i = 0; i < in_size; ++i) { data[datalen] = in_data[i]; @@ -3727,5 +3639,13 @@ kit_sha256_hash_t kit_sha256(ptrdiff_t in_size, uint8_t *in_data) { } return hash; } +#undef ROTLEFT +#undef ROTRIGHT +#undef CH +#undef MAJ +#undef EP0 +#undef EP1 +#undef SIG0 +#undef SIG1 #endif #endif -- cgit v1.2.3