Use WaitOnAddress instead of RtlWaitOnAddress

This commit is contained in:
gingerBill
2021-10-11 15:23:39 +01:00
parent 240b6aab13
commit 8c9299c139
2 changed files with 15 additions and 20 deletions
+11 -19
View File
@@ -5,40 +5,32 @@ package sync2
import "core:time"
foreign import Synchronization "system:Synchronization.lib"
foreign import NtDll "system:NtDll.lib"
@(default_calling_convention="stdcall")
foreign NtDll {
RtlWaitOnAddress :: proc(Address: rawptr, CompareAddress: rawptr, AddressSize: uint, Timeout: ^i64) -> b32 ---
}
@(default_calling_convention="stdcall")
foreign Synchronization {
WaitOnAddress :: proc(Address: rawptr, CompareAddress: rawptr, AddressSize: uint, Timeout: u32) -> b32 ---
WakeByAddressSingle :: proc(Address: rawptr) ---
WakeByAddressAll :: proc(Address: rawptr) ---
WakeByAddressAll :: proc(Address: rawptr) ---
}
_futex_wait :: proc(f: ^Futex, expect: u32) -> bool {
expect := expect
ok := RtlWaitOnAddress(f, &expect, size_of(expect), nil)
return bool(ok)
return bool(WaitOnAddress(f, &expect, size_of(expect), ~u32(0)))
}
_futex_wait_with_timeout :: proc(f: ^Futex, expect: u32, duration: time.Duration) -> bool {
expect := expect
timeout: i64
timeout_ptr: ^i64
if duration > 0 {
// In 100 ns units
timeout = i64(timeout)/100
timeout_ptr = &timeout
if duration <= 0 {
return bool(WaitOnAddress(f, &expect, size_of(expect), ~u32(0)))
}
\
ok := RtlWaitOnAddress(f, &expect, size_of(expect), timeout_ptr)
return bool(ok)
timeout := ~u32(0)
if duration > 0 {
timeout = u32(timeout/1e6)
}
return bool(WaitOnAddress(f, &expect, size_of(expect), timeout))
}
_futex_wake_single :: proc(f: ^Futex) {
+4 -1
View File
@@ -154,6 +154,9 @@ cond_wait :: proc(c: ^Cond, m: ^Mutex) {
}
cond_wait_with_timeout :: proc(c: ^Cond, m: ^Mutex, duration: time.Duration) -> bool {
if duration <= 0 {
return false
}
return _cond_wait_with_timeout(c, m, duration)
}
@@ -215,7 +218,7 @@ futex_wait_with_timeout :: proc(f: ^Futex, expected: u32, duration: time.Duratio
if u32(atomic_load(f)) != expected {
return true
}
if duration == 0 {
if duration <= 0 {
return false
}