v/vlib/sync/sync_windows.c.v

161 lines
3.4 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-06-22 21:53:22 +02:00
module sync
2019-07-24 17:36:20 +02:00
2020-07-15 10:22:33 +02:00
import time
// TODO: The suggestion of using CriticalSection instead of mutex
// was discussed. Needs consideration.
2019-07-24 17:36:20 +02:00
// Mutex HANDLE
type MHANDLE = voidptr
2020-07-15 10:22:33 +02:00
// Semaphore HANDLE
type SHANDLE = voidptr
2019-06-22 21:53:22 +02:00
2019-10-25 16:24:40 +02:00
//[init_with=new_mutex] // TODO: implement support for this struct attribute, and disallow Mutex{} from outside the sync.new_mutex() function.
[ref_only]
pub struct Mutex {
2019-07-24 17:36:20 +02:00
mut:
mx MHANDLE // mutex handle
state MutexState // mutex state
cycle_wait i64 // waiting cycles (implemented only with atomic)
cycle_woken i64 // woken cycles ^
reader_sem u32 // reader semarphone
writer_sem u32 // writer semaphones
2019-06-22 21:53:22 +02:00
}
[ref_only]
pub struct RwMutex {
mut:
mx C.SRWLOCK // mutex handle
}
2020-07-15 10:22:33 +02:00
pub struct Semaphore {
mut:
sem SHANDLE
}
enum MutexState {
broken
waiting
released
abandoned
destroyed
}
2019-06-22 21:53:22 +02:00
pub fn new_mutex() &Mutex {
sm := &Mutex{}
2019-10-25 16:24:40 +02:00
unsafe {
2020-04-02 20:58:07 +02:00
mut m := sm
m.mx = MHANDLE(C.CreateMutex(0, false, 0))
2019-10-25 16:24:40 +02:00
if isnil(m.mx) {
m.state = .broken // handle broken and mutex state are broken
return sm
2020-04-02 20:58:07 +02:00
}
2019-10-25 16:24:40 +02:00
}
return sm
2019-10-25 16:24:40 +02:00
}
pub fn new_rwmutex() &RwMutex {
m := &RwMutex{}
C.InitializeSRWLock(&m.mx)
return m
}
pub fn (mut m Mutex) m_lock() {
// if mutex handle not initalized
if isnil(m.mx) {
m.mx = MHANDLE(C.CreateMutex(0, false, 0))
if isnil(m.mx) {
m.state = .broken // handle broken and mutex state are broken
return
}
}
2020-04-02 20:58:07 +02:00
state := C.WaitForSingleObject(m.mx, C.INFINITE) // infinite wait
2019-10-30 15:29:05 +01:00
/* TODO fix match/enum combo
m.state = match state {
2020-04-02 20:58:07 +02:00
C.WAIT_ABANDONED { .abandoned }
C.WAIT_OBJECT_0 { .waiting }
2019-10-30 14:43:40 +01:00
else { .broken }
}
2019-10-30 15:29:05 +01:00
*/
2020-04-02 20:58:07 +02:00
if state == C.WAIT_ABANDONED {
m.state = .abandoned
// FIXME Use C constant instead
} else if state == 0 /* C.WAIT_OBJECT_0 */ {
m.state = .waiting
} else {
m.state = .broken
}
2019-06-22 21:53:22 +02:00
}
2020-05-17 13:51:18 +02:00
pub fn (mut m Mutex) unlock() {
if m.state == .waiting {
2020-04-02 20:58:07 +02:00
if C.ReleaseMutex(m.mx) {
m.state = .broken
return
}
}
m.state = .released
}
2019-07-24 17:36:20 +02:00
// RwMutex has separate read- and write locks
pub fn (mut m RwMutex) r_lock() {
C.AcquireSRWLockShared(&m.mx)
}
pub fn (mut m RwMutex) w_lock() {
C.AcquireSRWLockExclusive(&m.mx)
}
// Windows SRWLocks have different function to unlock
// So provide two functions here, too, to have a common interface
pub fn (mut m RwMutex) r_unlock() {
C.ReleaseSRWLockShared(&m.mx)
}
pub fn (mut m RwMutex) w_unlock() {
C.ReleaseSRWLockExclusive(&m.mx)
}
2020-05-17 13:51:18 +02:00
pub fn (mut m Mutex) destroy() {
if m.state == .waiting {
m.unlock() // unlock mutex before destroying
}
C.CloseHandle(m.mx) // destroy mutex
m.state = .destroyed // setting up reference to invalid state
}
2020-07-15 10:22:33 +02:00
2020-08-06 15:28:19 +02:00
[inline]
2020-07-15 10:22:33 +02:00
pub fn new_semaphore() Semaphore {
2020-08-06 15:28:19 +02:00
return new_semaphore_init(0)
}
pub fn new_semaphore_init(n u32) Semaphore {
2020-07-15 10:22:33 +02:00
return Semaphore{
2020-08-06 15:28:19 +02:00
sem: SHANDLE(C.CreateSemaphore(0, n, C.INT32_MAX, 0))
2020-07-15 10:22:33 +02:00
}
}
pub fn (s Semaphore) post() {
C.ReleaseSemaphore(s.sem, 1, 0)
}
pub fn (s Semaphore) wait() {
C.WaitForSingleObject(s.sem, C.INFINITE)
}
pub fn (s Semaphore) try_wait() bool {
return C.WaitForSingleObject(s.sem, 0) == 0
}
pub fn (s Semaphore) timed_wait(timeout time.Duration) bool {
return C.WaitForSingleObject(s.sem, timeout / time.millisecond) == 0
}
2020-08-06 15:28:19 +02:00
pub fn (s Semaphore) destroy() bool {
return C.CloseHandle(s.sem) != 0
}