coffee/wepoll
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix reflock test

Browse Source
This commit is contained in:
Raymond Zhao 2024-06-14 14:18:56 -07:00
parent f392daa61f
commit c265e8625b
No known key found for this signature in database
GPG Key ID: 4DA8A81F5F7FCF93
1 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/reflock.c
View File

@ -11,32 +11,31 @@
#define REFLOCK__REF ((long) 0x00000001UL) #define REFLOCK__REF ((long) 0x00000001UL)
#define REFLOCK__REF_MASK ((long) 0x0fffffffUL) #define REFLOCK__REF_MASK ((long) 0x0fffffffUL)
#define REFLOCK__DESTROY ((long) 0x10000000UL) #define REFLOCK__DESTROY ((long) 0x10000000UL)
#define REFLOCK__DESTROY_MASK ((long) 0xf0000000UL) #define REFLOCK__DESTROY_MASK ((long) 0x10000000UL)
#define REFLOCK__SIGNAL ((long) 0x20000000UL)
#define REFLOCK__SIGNAL_MASK ((long) 0x20000000UL)
#define REFLOCK__POISON ((long) 0x300dead0UL) #define REFLOCK__POISON ((long) 0x300dead0UL)
/* clang-format on */ /* clang-format on */
/* Prevents the possibility of exiting early due to spurious wakeups. */
static void* reflock__signal_object = NULL;
void reflock_init(reflock_t* reflock) { void reflock_init(reflock_t* reflock) {
reflock->state = 0; reflock->state = 0;
} }
static void reflock__signal_event(void* address) { static void reflock__signal_event(reflock_t* reflock) {
reflock__signal_object = address; long state = InterlockedAdd(&reflock->state, REFLOCK__SIGNAL);
WakeByAddressSingle(address); unused_var(state);
WakeByAddressSingle(reflock);
} }
static void reflock__await_event(void* address) { static void reflock__await_event(reflock_t* reflock) {
BOOL status = TRUE; BOOL status = TRUE;
if (reflock__signal_object == address) {
reflock__signal_object = NULL;
}
do { do {
status = WaitOnAddress(address, address, sizeof(void*), INFINITE); status = WaitOnAddress(reflock, reflock, sizeof(reflock_t*), INFINITE);
} while (reflock__signal_object != address); } while ((reflock->state & REFLOCK__SIGNAL_MASK) == 0);
long state = InterlockedAdd(&reflock->state, -REFLOCK__SIGNAL);
unused_var(state);
if (status != TRUE) if (status != TRUE)
abort(); abort();