xref: /freebsd-src/contrib/processor-trace/include/windows/threads.h (revision 85f87cf491bec6f90948a85b10f5523ea24db9e3)
174fe6c29SRuslan Bukin /*
2*85f87cf4SRuslan Bukin  * Copyright (c) 2014-2019, Intel Corporation
374fe6c29SRuslan Bukin  *
474fe6c29SRuslan Bukin  * Redistribution and use in source and binary forms, with or without
574fe6c29SRuslan Bukin  * modification, are permitted provided that the following conditions are met:
674fe6c29SRuslan Bukin  *
774fe6c29SRuslan Bukin  *  * Redistributions of source code must retain the above copyright notice,
874fe6c29SRuslan Bukin  *    this list of conditions and the following disclaimer.
974fe6c29SRuslan Bukin  *  * Redistributions in binary form must reproduce the above copyright notice,
1074fe6c29SRuslan Bukin  *    this list of conditions and the following disclaimer in the documentation
1174fe6c29SRuslan Bukin  *    and/or other materials provided with the distribution.
1274fe6c29SRuslan Bukin  *  * Neither the name of Intel Corporation nor the names of its contributors
1374fe6c29SRuslan Bukin  *    may be used to endorse or promote products derived from this software
1474fe6c29SRuslan Bukin  *    without specific prior written permission.
1574fe6c29SRuslan Bukin  *
1674fe6c29SRuslan Bukin  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
1774fe6c29SRuslan Bukin  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
1874fe6c29SRuslan Bukin  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
1974fe6c29SRuslan Bukin  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
2074fe6c29SRuslan Bukin  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
2174fe6c29SRuslan Bukin  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
2274fe6c29SRuslan Bukin  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
2374fe6c29SRuslan Bukin  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
2474fe6c29SRuslan Bukin  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
2574fe6c29SRuslan Bukin  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
2674fe6c29SRuslan Bukin  * POSSIBILITY OF SUCH DAMAGE.
2774fe6c29SRuslan Bukin  *
2874fe6c29SRuslan Bukin  *
2974fe6c29SRuslan Bukin  * It looks like there is still no support for C11's threads.h.
3074fe6c29SRuslan Bukin  *
3174fe6c29SRuslan Bukin  * We implement the few features we actually need hoping that this file will
3274fe6c29SRuslan Bukin  * soon go away.
3374fe6c29SRuslan Bukin  */
3474fe6c29SRuslan Bukin 
3574fe6c29SRuslan Bukin #ifndef THREADS_H
3674fe6c29SRuslan Bukin #define THREADS_H
3774fe6c29SRuslan Bukin 
3874fe6c29SRuslan Bukin #include "windows.h"
3974fe6c29SRuslan Bukin 
4074fe6c29SRuslan Bukin 
4174fe6c29SRuslan Bukin enum {
4274fe6c29SRuslan Bukin 	thrd_success	= 1,
4374fe6c29SRuslan Bukin 	thrd_error
4474fe6c29SRuslan Bukin };
4574fe6c29SRuslan Bukin 
4674fe6c29SRuslan Bukin 
4774fe6c29SRuslan Bukin struct pt_thread {
4874fe6c29SRuslan Bukin 	HANDLE handle;
4974fe6c29SRuslan Bukin };
5074fe6c29SRuslan Bukin typedef struct pt_thread thrd_t;
5174fe6c29SRuslan Bukin 
5274fe6c29SRuslan Bukin typedef int (*thrd_start_t)(void *);
5374fe6c29SRuslan Bukin 
5474fe6c29SRuslan Bukin 
5574fe6c29SRuslan Bukin struct thrd_args {
5674fe6c29SRuslan Bukin 	thrd_start_t fun;
5774fe6c29SRuslan Bukin 	void *arg;
5874fe6c29SRuslan Bukin };
5974fe6c29SRuslan Bukin 
thrd_routine(void * arg)6074fe6c29SRuslan Bukin static DWORD WINAPI thrd_routine(void *arg)
6174fe6c29SRuslan Bukin {
6274fe6c29SRuslan Bukin 	struct thrd_args *args;
6374fe6c29SRuslan Bukin 	int result;
6474fe6c29SRuslan Bukin 
6574fe6c29SRuslan Bukin 	args = (struct thrd_args *) arg;
6674fe6c29SRuslan Bukin 	if (!args)
6774fe6c29SRuslan Bukin 		return (DWORD) -1;
6874fe6c29SRuslan Bukin 
6974fe6c29SRuslan Bukin 	result = -1;
7074fe6c29SRuslan Bukin 	if (args->fun)
7174fe6c29SRuslan Bukin 		result = args->fun(args->arg);
7274fe6c29SRuslan Bukin 
7374fe6c29SRuslan Bukin 	free(args);
7474fe6c29SRuslan Bukin 
7574fe6c29SRuslan Bukin 	return (DWORD) result;
7674fe6c29SRuslan Bukin }
7774fe6c29SRuslan Bukin 
thrd_create(thrd_t * thrd,thrd_start_t fun,void * arg)7874fe6c29SRuslan Bukin static inline int thrd_create(thrd_t *thrd, thrd_start_t fun, void *arg)
7974fe6c29SRuslan Bukin {
8074fe6c29SRuslan Bukin 	struct thrd_args *args;
8174fe6c29SRuslan Bukin 	HANDLE handle;
8274fe6c29SRuslan Bukin 
8374fe6c29SRuslan Bukin 	if (!thrd || !fun)
8474fe6c29SRuslan Bukin 		return thrd_error;
8574fe6c29SRuslan Bukin 
8674fe6c29SRuslan Bukin 	args = malloc(sizeof(*args));
8774fe6c29SRuslan Bukin 	if (!args)
8874fe6c29SRuslan Bukin 		return thrd_error;
8974fe6c29SRuslan Bukin 
9074fe6c29SRuslan Bukin 	args->fun = fun;
9174fe6c29SRuslan Bukin 	args->arg = arg;
9274fe6c29SRuslan Bukin 
9374fe6c29SRuslan Bukin 	handle = CreateThread(NULL, 0, thrd_routine, args, 0, NULL);
9474fe6c29SRuslan Bukin 	if (!handle) {
9574fe6c29SRuslan Bukin 		free(args);
9674fe6c29SRuslan Bukin 		return thrd_error;
9774fe6c29SRuslan Bukin 	}
9874fe6c29SRuslan Bukin 
9974fe6c29SRuslan Bukin 	thrd->handle = handle;
10074fe6c29SRuslan Bukin 	return thrd_success;
10174fe6c29SRuslan Bukin }
10274fe6c29SRuslan Bukin 
thrd_join(thrd_t * thrd,int * res)10374fe6c29SRuslan Bukin static inline int thrd_join(thrd_t *thrd, int *res)
10474fe6c29SRuslan Bukin {
10574fe6c29SRuslan Bukin 	DWORD status;
10674fe6c29SRuslan Bukin 	BOOL success;
10774fe6c29SRuslan Bukin 
10874fe6c29SRuslan Bukin 	if (!thrd)
10974fe6c29SRuslan Bukin 		return thrd_error;
11074fe6c29SRuslan Bukin 
11174fe6c29SRuslan Bukin 	status = WaitForSingleObject(thrd->handle, INFINITE);
11274fe6c29SRuslan Bukin 	if (status)
11374fe6c29SRuslan Bukin 		return thrd_error;
11474fe6c29SRuslan Bukin 
11574fe6c29SRuslan Bukin 	if (res) {
11674fe6c29SRuslan Bukin 		DWORD result;
11774fe6c29SRuslan Bukin 
11874fe6c29SRuslan Bukin 		success = GetExitCodeThread(thrd->handle, &result);
11974fe6c29SRuslan Bukin 		if (!success) {
12074fe6c29SRuslan Bukin 			(void) CloseHandle(thrd->handle);
12174fe6c29SRuslan Bukin 			return thrd_error;
12274fe6c29SRuslan Bukin 		}
12374fe6c29SRuslan Bukin 
12474fe6c29SRuslan Bukin 		*res = (int) result;
12574fe6c29SRuslan Bukin 	}
12674fe6c29SRuslan Bukin 
12774fe6c29SRuslan Bukin 	success = CloseHandle(thrd->handle);
12874fe6c29SRuslan Bukin 	if (!success)
12974fe6c29SRuslan Bukin 		return thrd_error;
13074fe6c29SRuslan Bukin 
13174fe6c29SRuslan Bukin 	return thrd_success;
13274fe6c29SRuslan Bukin }
13374fe6c29SRuslan Bukin 
13474fe6c29SRuslan Bukin struct pt_mutex {
13574fe6c29SRuslan Bukin 	CRITICAL_SECTION cs;
13674fe6c29SRuslan Bukin };
13774fe6c29SRuslan Bukin typedef struct pt_mutex mtx_t;
13874fe6c29SRuslan Bukin 
13974fe6c29SRuslan Bukin enum {
14074fe6c29SRuslan Bukin 	mtx_plain
14174fe6c29SRuslan Bukin };
14274fe6c29SRuslan Bukin 
mtx_init(mtx_t * mtx,int type)14374fe6c29SRuslan Bukin static inline int mtx_init(mtx_t *mtx, int type)
14474fe6c29SRuslan Bukin {
14574fe6c29SRuslan Bukin 	if (!mtx || type != mtx_plain)
14674fe6c29SRuslan Bukin 		return thrd_error;
14774fe6c29SRuslan Bukin 
14874fe6c29SRuslan Bukin 	InitializeCriticalSection(&mtx->cs);
14974fe6c29SRuslan Bukin 
15074fe6c29SRuslan Bukin 	return thrd_success;
15174fe6c29SRuslan Bukin }
15274fe6c29SRuslan Bukin 
mtx_destroy(mtx_t * mtx)15374fe6c29SRuslan Bukin static inline void mtx_destroy(mtx_t *mtx)
15474fe6c29SRuslan Bukin {
15574fe6c29SRuslan Bukin 	if (mtx)
15674fe6c29SRuslan Bukin 		DeleteCriticalSection(&mtx->cs);
15774fe6c29SRuslan Bukin }
15874fe6c29SRuslan Bukin 
mtx_lock(mtx_t * mtx)15974fe6c29SRuslan Bukin static inline int mtx_lock(mtx_t *mtx)
16074fe6c29SRuslan Bukin {
16174fe6c29SRuslan Bukin 	if (!mtx)
16274fe6c29SRuslan Bukin 		return thrd_error;
16374fe6c29SRuslan Bukin 
16474fe6c29SRuslan Bukin 	EnterCriticalSection(&mtx->cs);
16574fe6c29SRuslan Bukin 
16674fe6c29SRuslan Bukin 	return thrd_success;
16774fe6c29SRuslan Bukin }
16874fe6c29SRuslan Bukin 
mtx_unlock(mtx_t * mtx)16974fe6c29SRuslan Bukin static inline int mtx_unlock(mtx_t *mtx)
17074fe6c29SRuslan Bukin {
17174fe6c29SRuslan Bukin 	if (!mtx)
17274fe6c29SRuslan Bukin 		return thrd_error;
17374fe6c29SRuslan Bukin 
17474fe6c29SRuslan Bukin 	LeaveCriticalSection(&mtx->cs);
17574fe6c29SRuslan Bukin 
17674fe6c29SRuslan Bukin 	return thrd_success;
17774fe6c29SRuslan Bukin }
17874fe6c29SRuslan Bukin 
17974fe6c29SRuslan Bukin 
18074fe6c29SRuslan Bukin struct pt_cond {
18174fe6c29SRuslan Bukin 	CONDITION_VARIABLE cond;
18274fe6c29SRuslan Bukin };
18374fe6c29SRuslan Bukin typedef struct pt_cond cnd_t;
18474fe6c29SRuslan Bukin 
cnd_init(cnd_t * cnd)18574fe6c29SRuslan Bukin static inline int cnd_init(cnd_t *cnd)
18674fe6c29SRuslan Bukin {
18774fe6c29SRuslan Bukin 	if (!cnd)
18874fe6c29SRuslan Bukin 		return thrd_error;
18974fe6c29SRuslan Bukin 
19074fe6c29SRuslan Bukin 	InitializeConditionVariable(&cnd->cond);
19174fe6c29SRuslan Bukin 
19274fe6c29SRuslan Bukin 	return thrd_success;
19374fe6c29SRuslan Bukin }
19474fe6c29SRuslan Bukin 
cnd_destroy(cnd_t * cnd)19574fe6c29SRuslan Bukin static inline int cnd_destroy(cnd_t *cnd)
19674fe6c29SRuslan Bukin {
19774fe6c29SRuslan Bukin 	if (!cnd)
19874fe6c29SRuslan Bukin 		return thrd_error;
19974fe6c29SRuslan Bukin 
20074fe6c29SRuslan Bukin 	/* Nothing to do. */
20174fe6c29SRuslan Bukin 
20274fe6c29SRuslan Bukin 	return thrd_success;
20374fe6c29SRuslan Bukin }
20474fe6c29SRuslan Bukin 
cnd_signal(cnd_t * cnd)20574fe6c29SRuslan Bukin static inline int cnd_signal(cnd_t *cnd)
20674fe6c29SRuslan Bukin {
20774fe6c29SRuslan Bukin 	if (!cnd)
20874fe6c29SRuslan Bukin 		return thrd_error;
20974fe6c29SRuslan Bukin 
21074fe6c29SRuslan Bukin 	WakeConditionVariable(&cnd->cond);
21174fe6c29SRuslan Bukin 
21274fe6c29SRuslan Bukin 	return thrd_success;
21374fe6c29SRuslan Bukin }
21474fe6c29SRuslan Bukin 
cnd_broadcast(cnd_t * cnd)21574fe6c29SRuslan Bukin static inline int cnd_broadcast(cnd_t *cnd)
21674fe6c29SRuslan Bukin {
21774fe6c29SRuslan Bukin 	if (!cnd)
21874fe6c29SRuslan Bukin 		return thrd_error;
21974fe6c29SRuslan Bukin 
22074fe6c29SRuslan Bukin 	WakeAllConditionVariable(&cnd->cond);
22174fe6c29SRuslan Bukin 
22274fe6c29SRuslan Bukin 	return thrd_success;
22374fe6c29SRuslan Bukin }
22474fe6c29SRuslan Bukin 
cnd_wait(cnd_t * cnd,mtx_t * mtx)22574fe6c29SRuslan Bukin static inline int cnd_wait(cnd_t *cnd, mtx_t *mtx)
22674fe6c29SRuslan Bukin {
22774fe6c29SRuslan Bukin 	BOOL success;
22874fe6c29SRuslan Bukin 
22974fe6c29SRuslan Bukin 	if (!cnd || !mtx)
23074fe6c29SRuslan Bukin 		return thrd_error;
23174fe6c29SRuslan Bukin 
23274fe6c29SRuslan Bukin 	success = SleepConditionVariableCS(&cnd->cond, &mtx->cs, INFINITE);
23374fe6c29SRuslan Bukin 	if (!success)
23474fe6c29SRuslan Bukin 		return thrd_error;
23574fe6c29SRuslan Bukin 
23674fe6c29SRuslan Bukin 	return thrd_success;
23774fe6c29SRuslan Bukin }
23874fe6c29SRuslan Bukin 
23974fe6c29SRuslan Bukin #endif /* THREADS_H */
240