Ver Fonte

Use `WaitOnAddress` instead of `RtlWaitOnAddress`

gingerBill há 3 anos atrás
pai
commit
8c9299c139
2 ficheiros alterados com 13 adições e 18 exclusões
  1. 9 17
      core/sync/sync2/futex_windows.odin
  2. 4 1
      core/sync/sync2/primitives.odin

+ 9 - 17
core/sync/sync2/futex_windows.odin

@@ -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
+	if duration <= 0 {
+		return bool(WaitOnAddress(f, &expect, size_of(expect), ~u32(0)))
+	}
 	
-	timeout: i64
-	timeout_ptr: ^i64
+	timeout := ~u32(0)
 	if duration > 0 {
-		// In 100 ns units
-		timeout = i64(timeout)/100
-		timeout_ptr = &timeout
+		timeout = u32(timeout/1e6)
 	}
-	\
-	ok := RtlWaitOnAddress(f, &expect, size_of(expect), timeout_ptr)
-	return bool(ok)
+	return bool(WaitOnAddress(f, &expect, size_of(expect), timeout))
 }
 
 _futex_wake_single :: proc(f: ^Futex) {

+ 4 - 1
core/sync/sync2/primitives.odin

@@ -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
 	}