Skip to content

Commit

Permalink
feat(thread): Replace pthread to CThreads
Browse files Browse the repository at this point in the history
CThreads can also work in Windows using Windows threads, due to this, pthread was replaced by it.
  • Loading branch information
ThePedroo committed May 16, 2023
1 parent ff6e5a0 commit 8955e85
Show file tree
Hide file tree
Showing 17 changed files with 786 additions and 182 deletions.
3 changes: 2 additions & 1 deletion core/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,8 @@ OBJS = cog-utils.o \
priority_queue.o \
anomap.o \
sha1.o \
threadpool.o
threadpool.o \
cthreads.o

WFLAGS = -Wall -Wextra -Wpedantic
CFLAGS += -std=c99 -pthread -D_XOPEN_SOURCE=600 -DLOG_USE_COLOR \
Expand Down
274 changes: 274 additions & 0 deletions core/cthreads.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,274 @@
#include <stdio.h>
#include <stdlib.h>

#include "cthreads.h"

#if _WIN32
#include <windows.h>
DWORD WINAPI __cthreads_winthreads_function_wrapper(void *data) {
struct cthreads_args *args = data;
args->func(args->data);

free(data);

return TRUE;
}
#else
#include <pthread.h>
void *__cthreads_pthread_function_wrapper(void *data) {
struct cthreads_args *args = data;
args->func(args->data);

free(data);

return NULL;
}
#endif

int cthreads_thread_create(struct cthreads_thread *thread, struct cthreads_thread_attr *attr, void *(*func)(void *data), void *data) {
#ifdef _WIN32
struct cthreads_args *args = malloc(sizeof(struct cthreads_args));
args->func = func;
args->data = data;

if (attr) thread->wThread = CreateThread(NULL, attr->stacksize ? attr->stacksize : 0,
__cthreads_winthreads_function_wrapper, args,
attr->dwCreationFlags ? (DWORD)attr->dwCreationFlags : 0, NULL);
else thread->wThread = CreateThread(NULL, 0, __cthreads_winthreads_function_wrapper, args, 0, NULL);

return 0;
#else
pthread_t pthread;

struct cthreads_args *args = malloc(sizeof(struct cthreads_args));
args->func = func;
args->data = data;

pthread_attr_t pAttr;
if (attr) {
if (attr->detachstate) pthread_attr_setdetachstate(&pAttr, attr->detachstate);
if (attr->guardsize) pthread_attr_setguardsize(&pAttr, attr->guardsize);
if (attr->inheritsched) pthread_attr_setinheritsched(&pAttr, attr->inheritsched);
if (attr->schedpolicy) pthread_attr_setschedpolicy(&pAttr, attr->schedpolicy);
if (attr->scope) pthread_attr_setscope(&pAttr, attr->scope);
if (attr->stack) pthread_attr_setstack(&pAttr, attr->stackaddr, attr->stack);
if (attr->stacksize) pthread_attr_setstacksize(&pAttr, attr->stacksize);
}

int res = pthread_create(&pthread, attr ? &pAttr : NULL, __cthreads_pthread_function_wrapper, (void *)args);

thread->pThread = pthread;

return res;
#endif
}

int cthreads_thread_detach(struct cthreads_thread *thread) {
#ifdef _WIN32
return CloseHandle(thread->wThread);
#else
return pthread_detach(thread->pThread);
#endif
}

void cthreads_thread_close(void *code) {
#ifdef _WIN32
ExitThread((DWORD)(uintptr_t)code);
#else
pthread_exit(code);
#endif
}

int cthreads_mutex_init(struct cthreads_mutex *mutex, struct cthreads_mutex_attr *attr) {
#ifdef _WIN32
HANDLE wMutex;
if (attr) wMutex = CreateMutex(NULL, attr->bInitialOwner ? TRUE : FALSE,
attr->lpName ? (LPCSTR)attr->lpName : NULL);
else wMutex = CreateMutex(NULL, FALSE, NULL);

if (wMutex == NULL) return 1;
else {
mutex->wMutex = wMutex;
return 0;
}
#else
pthread_mutexattr_t pAttr;
if (attr) {
if (attr->pshared) pthread_mutexattr_setpshared(&pAttr, attr->pshared);
if (attr->type) pthread_mutexattr_settype(&pAttr, attr->type);
if (attr->protocol) pthread_mutexattr_setprotocol(&pAttr, attr->protocol);
#ifdef __linux__
if (attr->robust) pthread_mutexattr_setrobust(&pAttr, attr->robust);
#elif
if (attr->robust) pthread_mutexattr_setrobust_np(&pAttr, attr->robust);
#endif
if (attr->prioceiling) pthread_mutexattr_setprioceiling(&pAttr, attr->prioceiling);
}

return pthread_mutex_init(&mutex->pMutex, attr ? &pAttr : NULL);
#endif
}

int cthreads_mutex_lock(struct cthreads_mutex *mutex) {
#ifdef _WIN32
DWORD ret = WaitForSingleObject(mutex->wMutex, INFINITE);

if (ret == WAIT_OBJECT_0) return 0;
return 1;
#else
return pthread_mutex_lock(&mutex->pMutex);
#endif
}

int cthreads_mutex_trylock(struct cthreads_mutex *mutex) {
#ifdef _WIN32
DWORD ret = WaitForSingleObject(mutex->wMutex, 0);

if (ret == WAIT_OBJECT_0) return 0;
return 1;
#else
return pthread_mutex_trylock(&mutex->pMutex);
#endif
}

int cthreads_mutex_unlock(struct cthreads_mutex *mutex) {
#ifdef _WIN32
return ReleaseMutex(mutex->wMutex) == 0 ? 1 : 0;
#else
return pthread_mutex_unlock(&mutex->pMutex);
#endif
}

int cthreads_mutex_destroy(struct cthreads_mutex *mutex) {
#ifdef _WIN32
return CloseHandle(mutex->wMutex) == 0 ? 1 : 0;
#else
return pthread_mutex_destroy(&mutex->pMutex);
#endif
}

int cthreads_cond_init(struct cthreads_cond *cond, struct cthreads_cond_attr *attr) {
#ifdef _WIN32
if (attr) cond->wCond = CreateEvent(NULL, attr->bManualReset ? TRUE : FALSE,
attr->bInitialState ? TRUE : FALSE,
attr->lpName ? (LPTSTR)attr->lpName : NULL);
else cond->wCond = CreateEvent(NULL, FALSE, FALSE, NULL);

if (cond->wCond == NULL) return 1;
return 0;
#else
pthread_condattr_t pAttr;
if (attr) {
if (attr->pshared) pthread_condattr_setpshared(&pAttr, attr->pshared);
if (attr->clock) pthread_condattr_setclock(&pAttr, attr->clock);
}

return pthread_cond_init(&cond->pCond, attr ? &pAttr : NULL);
#endif
}

int cthreads_cond_signal(struct cthreads_cond *cond) {
#ifdef _WIN32
return SetEvent(cond->wCond) == 0 ? 1 : 0;
#else
return pthread_cond_signal(&cond->pCond);
#endif
}

int cthreads_cond_broadcast(struct cthreads_cond *cond) {
#ifdef _WIN32
return SetEvent(cond->wCond) == 0 ? 1 : 0;
#else
return pthread_cond_broadcast(&cond->pCond);
#endif
}

int cthreads_cond_destroy(struct cthreads_cond *cond) {
#ifdef _WIN32
return CloseHandle(cond->wCond) == 0 ? 1 : 0;
#else
return pthread_cond_destroy(&cond->pCond);
#endif
}

int cthreads_cond_wait(struct cthreads_cond *cond, struct cthreads_mutex *mutex) {
#ifdef _WIN32
if (cthreads_mutex_unlock(mutex) == 1) return 1;
if (WaitForSingleObject(cond->wCond, INFINITE) == WAIT_FAILED) return 1;
return cthreads_mutex_lock(mutex) == 1 ? 1 : 0;
#else
return pthread_cond_wait(&cond->pCond, &mutex->pMutex);
#endif
}

int cthreads_join(struct cthreads_thread *thread, void *code) {
#ifdef _WIN32
if (WaitForSingleObject(thread->wThread, INFINITE) == WAIT_FAILED) return 0;
return GetExitCodeThread(thread->wThread, (LPDWORD)&code) == 0 ? 1 : 0;
#else
return pthread_join(thread->pThread, code ? &code : NULL);
#endif
}

int cthreads_rwlock_init(struct cthreads_rwlock *rwlock) {
#ifdef _WIN32
rwlock->wRWLock = CreateMutex(NULL, FALSE, NULL);

if (rwlock->wRWLock == NULL) return 1;
return 0;
#else
return pthread_rwlock_init(&rwlock->pRWLock, NULL);
#endif
}

int cthreads_rwlock_rdlock(struct cthreads_rwlock *rwlock) {
#ifdef _WIN32
return WaitForSingleObject(rwlock->wRWLock, INFINITE) == WAIT_FAILED ? 1 : 0;
#else
return pthread_rwlock_rdlock(&rwlock->pRWLock);
#endif
}

int cthreads_rwlock_unlock(struct cthreads_rwlock *rwlock) {
#ifdef _WIN32
return ReleaseMutex(rwlock->wRWLock) == 0 ? 1 : 0;
#else
return pthread_rwlock_unlock(&rwlock->pRWLock);
#endif
}

int cthreads_rwlock_wrlock(struct cthreads_rwlock *rwlock) {
#ifdef _WIN32
return WaitForSingleObject(rwlock->wRWLock, INFINITE) == WAIT_FAILED ? 1 : 0;
#else
return pthread_rwlock_wrlock(&rwlock->pRWLock);
#endif
}

int cthreads_rwlock_destroy(struct cthreads_rwlock *rwlock) {
#ifdef _WIN32
return CloseHandle(rwlock->wRWLock) == 0 ? 1 : 0;
#else
return pthread_rwlock_destroy(&rwlock->pRWLock);
#endif
}

int cthreads_equal(struct cthreads_thread thread1, struct cthreads_thread thread2) {
#ifdef _WIN32
return thread1.wThread == thread2.wThread;
#else
return pthread_equal(thread1.pThread, thread2.pThread);
#endif
}

struct cthreads_thread cthreads_self() {
struct cthreads_thread t;

#ifdef _WIN32
t.wThread = GetCurrentThread();
#else
t.pThread = pthread_self();
#endif

return t;
}
Loading

0 comments on commit 8955e85

Please sign in to comment.