From c562d0f9e336251f9ff599a840310835820b0461 Mon Sep 17 00:00:00 2001 From: ame Date: Wed, 8 May 2024 10:55:05 -0500 Subject: work on threads --- src/thread.c | 277 +++++++++++++++++++++++++++++++++++------------------------ 1 file changed, 163 insertions(+), 114 deletions(-) (limited to 'src/thread.c') diff --git a/src/thread.c b/src/thread.c index 25ad500..441793b 100644 --- a/src/thread.c +++ b/src/thread.c @@ -1,114 +1,163 @@ -#include "thread.h" -#include "lua5.4/lauxlib.h" -#include "lua5.4/lua.h" -#include "stdint.h" -#include -#include -#include -#include "types/str.h" -#include "util.h" - -struct thread_info { - str* function; - lua_State* L; - int return_count; - pthread_t tid; - pthread_cond_t cond; - pthread_mutex_t lock; -}; - -#include "io.h" - -int l_res(lua_State* L){ - int return_count = lua_gettop(L) - 1; - lua_pushstring(L, "_"); - lua_gettable(L, 1); - struct thread_info* info = lua_touserdata(L, -1); - info->return_count = return_count; - - for(int i = info->return_count - 1; i != -1; i--){ - int ot = lua_gettop(L); - - lua_pushvalue(L, 2 + i); - i_dcopy(L, info->L, NULL); - - lua_settop(L, ot); - } - - pthread_mutex_unlock(&info->lock); - - pthread_exit(NULL); - p_error("thread did not exit"); - - return 1; -} - -void* handle_thread(void* _args){ - struct thread_info* args = (struct thread_info*)_args; - lua_State* L = args->L; - - lua_newtable(L); - int res_idx = lua_gettop(L); - luaI_tsetcf(L, res_idx, "res", l_res); - luaI_tsetlud(L, res_idx, "_", args); - - luaL_loadbuffer(L, args->function->c, args->function->len, "thread"); - str_free(args->function); - - lua_pushvalue(L, res_idx); - lua_call(L, 1, 0); - - pthread_mutex_unlock(&args->lock); - - return NULL; -} - -int l_await(lua_State* L){ - lua_pushstring(L, "_"); - lua_gettable(L, 1); - struct thread_info* info = lua_touserdata(L, -1); - - pthread_mutex_lock(&info->lock); - - for(int i = 0; i != info->return_count; i++){ - int ot = lua_gettop(info->L); - - lua_pushvalue(info->L, ot - info->return_count + i); - i_dcopy(info->L, L, NULL); - - lua_settop(info->L, ot); - } - - return info->return_count; -} - -int l_async(lua_State* oL){ - lua_State* L = luaL_newstate(); - - luaL_openlibs(L); - - lua_getglobal(oL, "_G"); - i_dcopy(oL, L, NULL); - lua_set_global_table(L); - - struct thread_info* args = calloc(1, sizeof * args); - args->L = L; - args->cond = (pthread_cond_t)PTHREAD_COND_INITIALIZER; - args->lock = (pthread_mutex_t)PTHREAD_MUTEX_INITIALIZER; - pthread_mutex_lock(&args->lock); - args->return_count = 0; - - args->function = str_init(""); - lua_pushvalue(oL, 1); - lua_dump(oL, writer, (void*)args->function, 0); - - pthread_create(&args->tid, NULL, handle_thread, (void*)args); - pthread_detach(args->tid); - - lua_newtable(oL); - int res_idx = lua_gettop(oL); - luaI_tsetcf(oL, res_idx, "await", l_await); - luaI_tsetlud(oL, res_idx, "_", args); - lua_pushvalue(oL, res_idx); - return 1; -} +#include "thread.h" +#include "lua5.4/lauxlib.h" +#include "lua5.4/lua.h" +#include "stdint.h" +#include +#include +#include +#include "types/str.h" +#include "util.h" + +#include "types/larray.h" + +struct thread_info { + str* function; + lua_State* L; + int return_count; + pthread_t tid; + pthread_cond_t cond; + pthread_mutex_t lock; +}; + +#include "io.h" + +pthread_mutex_t thread_lock_lock = (pthread_mutex_t)PTHREAD_MUTEX_INITIALIZER; +larray_t* thread_locks = NULL; + +int l_tlock(lua_State* L){ + int idx = luaL_checkinteger(L, 1); + + pthread_mutex_lock(&thread_lock_lock); + pthread_mutex_t mutex; + if(thread_locks == NULL) thread_locks = larray_init(); + int i = 0; + if((i = larray_geti(thread_locks, idx)) == -1){ + pthread_mutex_init(&mutex, NULL); + pthread_mutex_lock(&mutex); + larray_set(&thread_locks, idx, (void*)mutex); + } else { + pthread_mutex_t m = (pthread_mutex_t)thread_locks->arr[i].value; + pthread_mutex_unlock(&thread_lock_lock); + pthread_mutex_lock(&m); + pthread_mutex_lock(&thread_lock_lock); + thread_locks->arr[i].value = (void*)m; + + } + + pthread_mutex_unlock(&thread_lock_lock); + return 0; +} + +int l_tunlock(lua_State* L){ + int idx = luaL_checkinteger(L, 1); + + pthread_mutex_lock(&thread_lock_lock); + int i = 0; + if(thread_locks != NULL && (i = larray_geti(thread_locks, idx)) != -1){ + pthread_mutex_t m = (pthread_mutex_t)thread_locks->arr[i].value; + pthread_mutex_unlock(&m); + thread_locks->arr[i].value = (void*)m; + } + + pthread_mutex_unlock(&thread_lock_lock); + return 0; +} + +int l_res(lua_State* L){ + int return_count = lua_gettop(L) - 1; + lua_pushstring(L, "_"); + lua_gettable(L, 1); + struct thread_info* info = lua_touserdata(L, -1); + info->return_count = return_count; + + for(int i = info->return_count - 1; i != -1; i--){ + int ot = lua_gettop(L); + + lua_pushvalue(L, 2 + i); + i_dcopy(L, info->L, NULL); + + lua_settop(L, ot); + } + + pthread_mutex_unlock(&info->lock); + + pthread_exit(NULL); + p_error("thread did not exit"); + + return 1; +} + +void* handle_thread(void* _args){ + struct thread_info* args = (struct thread_info*)_args; + lua_State* L = args->L; + + lua_newtable(L); + int res_idx = lua_gettop(L); + luaI_tsetlud(L, res_idx, "_", args); + + lua_newtable(L); + int meta_idx = lua_gettop(L); + luaI_tsetcf(L, meta_idx, "__call", l_res); + lua_pushvalue(L, meta_idx); + lua_setmetatable(L, res_idx); + + luaL_loadbuffer(L, args->function->c, args->function->len, "thread"); + str_free(args->function); + + lua_pushvalue(L, res_idx); + lua_call(L, 1, 0); + + pthread_mutex_unlock(&args->lock); + + return NULL; +} + +int l_await(lua_State* L){ + lua_pushstring(L, "_"); + lua_gettable(L, 1); + struct thread_info* info = lua_touserdata(L, -1); + + pthread_mutex_lock(&info->lock); + + for(int i = 0; i != info->return_count; i++){ + int ot = lua_gettop(info->L); + + lua_pushvalue(info->L, ot - info->return_count + i); + i_dcopy(info->L, L, NULL); + + lua_settop(info->L, ot); + } + + return info->return_count; +} + +int l_async(lua_State* oL){ + lua_State* L = luaL_newstate(); + + luaL_openlibs(L); + + lua_getglobal(oL, "_G"); + i_dcopy(oL, L, NULL); + lua_set_global_table(L); + + struct thread_info* args = calloc(1, sizeof * args); + args->L = L; + args->cond = (pthread_cond_t)PTHREAD_COND_INITIALIZER; + args->lock = (pthread_mutex_t)PTHREAD_MUTEX_INITIALIZER; + pthread_mutex_lock(&args->lock); + args->return_count = 0; + + args->function = str_init(""); + lua_pushvalue(oL, 1); + lua_dump(oL, writer, (void*)args->function, 0); + + pthread_create(&args->tid, NULL, handle_thread, (void*)args); + pthread_detach(args->tid); + + lua_newtable(oL); + int res_idx = lua_gettop(oL); + luaI_tsetcf(oL, res_idx, "await", l_await); + luaI_tsetlud(oL, res_idx, "_", args); + lua_pushvalue(oL, res_idx); + return 1; +} -- cgit v1.2.3