diff options
| author | 2021-06-20 01:56:35 +0200 | |
|---|---|---|
| committer | 2021-06-20 01:56:35 +0200 | |
| commit | cb50087482cf5a4239ad9c96b7157c45eac12876 (patch) | |
| tree | 1d60de827e4d9c08d7f51f16480d4583ee45ad25 /src/thread.c | |
| parent | eb787a10f06e4470134cff33a81bb872e4268bfa (diff) | |
| download | libirecovery-cb50087482cf5a4239ad9c96b7157c45eac12876.tar.gz libirecovery-cb50087482cf5a4239ad9c96b7157c45eac12876.tar.bz2  | |
thread: Add condition variable support
Diffstat (limited to 'src/thread.c')
| -rw-r--r-- | src/thread.c | 75 | 
1 files changed, 75 insertions, 0 deletions
diff --git a/src/thread.c b/src/thread.c index eb535ab..796ea0b 100644 --- a/src/thread.c +++ b/src/thread.c @@ -67,6 +67,8 @@ int thread_join(THREAD_T thread)  int thread_alive(THREAD_T thread)  { +	if (!thread) +		return 0;  #ifdef WIN32  	return WaitForSingleObject(thread, 0) == WAIT_TIMEOUT;  #else @@ -138,3 +140,76 @@ void thread_once(thread_once_t *once_control, void (*init_routine)(void))  	pthread_once(once_control, init_routine);  #endif  } + +void cond_init(cond_t* cond) +{ +#ifdef WIN32 +	cond->sem = CreateSemaphore(NULL, 0, 32767, NULL); +#else +	pthread_cond_init(cond, NULL); +#endif +} + +void cond_destroy(cond_t* cond) +{ +#ifdef WIN32 +	CloseHandle(cond->sem); +#else +	pthread_cond_destroy(cond); +#endif +} + +int cond_signal(cond_t* cond) +{ +#ifdef WIN32 +	int result = 0; +	if (!ReleaseSemaphore(cond->sem, 1, NULL)) { +		result = -1; +	} +	return result; +#else +	return pthread_cond_signal(cond); +#endif +} + +int cond_wait(cond_t* cond, mutex_t* mutex) +{ +#ifdef WIN32 +	mutex_unlock(mutex); +	DWORD res = WaitForSingleObject(cond->sem, INFINITE); +	switch (res) { +		case WAIT_OBJECT_0: +			return 0; +		default: +			return -1; +	} +#else +	return pthread_cond_wait(cond, mutex); +#endif +} + +int cond_wait_timeout(cond_t* cond, mutex_t* mutex, unsigned int timeout_ms) +{ +#ifdef WIN32 +	mutex_unlock(mutex); +	DWORD res = WaitForSingleObject(cond->sem, timeout_ms); +	switch (res) { +		case WAIT_OBJECT_0: +		case WAIT_TIMEOUT: +			return 0; +		default: +			return -1; +	} +#else +	struct timespec ts; +	struct timeval now; +	gettimeofday(&now, NULL); + +	ts.tv_sec = now.tv_sec + timeout_ms / 1000; +	ts.tv_nsec = now.tv_usec * 1000 + 1000 * 1000 * (timeout_ms % 1000); +	ts.tv_sec += ts.tv_nsec / (1000 * 1000 * 1000); +	ts.tv_nsec %= (1000 * 1000 * 1000); + +	return pthread_cond_timedwait(cond, mutex, &ts); +#endif +}  | 
